mirror of
https://github.com/MartinOpat/cuda-based-raytrace.git
synced 2025-06-07 02:13:10 +02:00
still need to modify datareader to use netcdf on only one thread
This commit is contained in:
1 parent
a7ad248c38
commit
7b3c87c656
11 files changed
+487
-77
No files matched your search
@@ -0,0 +1,58 @@
|
||||
#include "datareader.h"
|
||||
|
||||
#include <netcdf>
|
||||
#include <optional>
|
||||
|
||||
using namespace std;
|
||||
using namespace netCDF;
|
||||
|
||||
DataReader::DataReader(const std::string &path, std::string variableName):
|
||||
filePathManager(path), variableName(variableName) { }
|
||||
|
||||
size_t DataReader::fileLength(size_t fileIndex) {
|
||||
cout << "filePathMan = " << filePathManager.getPath(fileIndex) << " bla = " << fileIndex << "\n";
|
||||
|
||||
NcFile data(filePathManager.getPath(fileIndex), NcFile::read);
|
||||
|
||||
multimap<string, NcVar> vars = data.getVars();
|
||||
|
||||
NcVar var = vars.find(variableName)->second;
|
||||
|
||||
size_t length = 1;
|
||||
for (NcDim dim: var.getDims()) {
|
||||
length *= dim.getSize();
|
||||
}
|
||||
|
||||
// TODO: Turns out c-NetCDF is not thread safe :( https://github.com/Unidata/netcdf-c/issues/1373
|
||||
// size_t length = 34933248;
|
||||
std::cout << "return length" << length << "\n";
|
||||
return length;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void DataReader::loadFile(T* dataOut, size_t fileIndex) {
|
||||
std::cout << "loading file" << fileIndex <<"\n";
|
||||
NcFile data(filePathManager.getPath(fileIndex), NcFile::read);
|
||||
|
||||
// multimap<string, NcVar> vars = data.getVars();
|
||||
|
||||
// NcVar var = vars.find(variableName)->second;
|
||||
|
||||
// var.getVar(dataOut);
|
||||
}
|
||||
|
||||
template void DataReader::loadFile<float>(float* dataOut, size_t fileIndex);
|
||||
template void DataReader::loadFile<int>(int* dataOut, size_t fileIndex);
|
||||
|
||||
DataReader::~DataReader() {
|
||||
|
||||
}
|
||||
|
||||
// template <typename T>
|
||||
// void DataReader::readAndAllocateAxis(T** axis_ptr, size_t *size, const string &varName) {
|
||||
// assert(var.getDimCount() == 1);
|
||||
// netCDF::NcDim dim = var.getDim(0);
|
||||
// *size = dim.getSize();
|
||||
// cudaError_t status = cudaMallocManaged(axis_ptr, *size*sizeof(T));
|
||||
// var.getVar(*axis_ptr);
|
||||
// }
|
||||