zcy
/
openEMS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

hdf5 writer: allow data write for double data 

Signed-off-by: Thorsten Liebig <Thorsten.Liebig@gmx.de>
pull/1/head
Thorsten Liebig 2012-09-17 11:46:15 +02:00
parent 3f5e22b7f0
commit 4818e836b7
2 changed files with 117 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
tools/hdf5_file_writer.cpp
View File

@ -212,6 +212,23 @@ bool HDF5_File_Writer::WriteScalarField(std::string dataSetName, float const* co
return success;
}
bool HDF5_File_Writer::WriteScalarField(std::string dataSetName, double const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
size_t size = datasize[0]*datasize[1]*datasize[2];
size_t n_size[3]={datasize[2],datasize[1],datasize[0]};
double* buffer = new double[size];
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=field[i][j][k];
}
bool success = WriteData(dataSetName,buffer,3,n_size);
delete[] buffer;
return success;
}
bool HDF5_File_Writer::WriteScalarField(std::string dataSetName, std::complex<float> const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
@ -239,6 +256,33 @@ bool HDF5_File_Writer::WriteScalarField(std::string dataSetName, std::complex<fl
return success;
}
bool HDF5_File_Writer::WriteScalarField(std::string dataSetName, std::complex<double> const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
size_t size = datasize[0]*datasize[1]*datasize[2];
size_t n_size[3]={datasize[2],datasize[1],datasize[0]};
double* buffer = new double[size];
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=real(field[i][j][k]);
}
bool success = WriteData(dataSetName + "_real",buffer,3,n_size);
pos = 0;
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=imag(field[i][j][k]);
}
success &= WriteData(dataSetName + "_imag",buffer,3,n_size);
delete[] buffer;
return success;
}
bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, float const* const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
@ -257,6 +301,24 @@ bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, float const* co
return success;
}
bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, double const* const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
size_t size = datasize[0]*datasize[1]*datasize[2]*3;
double* buffer = new double[size];
for (int n=0;n<3;++n)
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=field[n][i][j][k];
}
size_t n_size[4]={3,datasize[2],datasize[1],datasize[0]};
bool success = WriteData(dataSetName,buffer,4,n_size);
delete[] buffer;
return success;
}
bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, std::complex<float> const* const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
@ -286,8 +348,46 @@ bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, std::complex<fl
return success;
}
bool HDF5_File_Writer::WriteVectorField(std::string dataSetName, std::complex<double> const* const* const* const* field, size_t datasize[3])
{
size_t pos = 0;
size_t size = datasize[0]*datasize[1]*datasize[2]*3;
size_t n_size[4]={3,datasize[2],datasize[1],datasize[0]};
double* buffer = new double[size];
for (int n=0;n<3;++n)
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=real(field[n][i][j][k]);
}
bool success = WriteData(dataSetName + "_real",buffer,4,n_size);
pos = 0;
for (int n=0;n<3;++n)
for (size_t k=0;k<datasize[2];++k)
for (size_t j=0;j<datasize[1];++j)
for (size_t i=0;i<datasize[0];++i)
{
buffer[pos++]=imag(field[n][i][j][k]);
}
success &= WriteData(dataSetName + "_imag",buffer,4,n_size);
delete[] buffer;
return success;
}
bool HDF5_File_Writer::WriteData(std::string dataSetName, float const* field_buf, size_t dim, size_t* datasize)
{
return WriteData(dataSetName, H5T_NATIVE_FLOAT, field_buf,dim, datasize);
}
bool HDF5_File_Writer::WriteData(std::string dataSetName, double const* field_buf, size_t dim, size_t* datasize)
{
return WriteData(dataSetName, H5T_NATIVE_DOUBLE, field_buf,dim, datasize);
}
bool HDF5_File_Writer::WriteData(std::string dataSetName, hid_t mem_type, void const* field_buf, size_t dim, size_t* datasize)
{
hid_t hdf5_file = H5Fopen( m_filename.c_str(), H5F_ACC_RDWR, H5P_DEFAULT );
if (hdf5_file<0)
@ -308,8 +408,8 @@ bool HDF5_File_Writer::WriteData(std::string dataSetName, float const* field_buf
for (size_t n=0;n<dim;++n)
dims[n]=datasize[n];
hid_t space = H5Screate_simple(dim, dims, NULL);
hid_t dataset = H5Dcreate(group, dataSetName.c_str(), H5T_NATIVE_FLOAT, space, H5P_DEFAULT);
if (H5Dwrite(dataset, H5T_NATIVE_FLOAT, space, H5P_DEFAULT, H5P_DEFAULT, field_buf))
hid_t dataset = H5Dcreate(group, dataSetName.c_str(), mem_type, space, H5P_DEFAULT);
if (H5Dwrite(dataset, mem_type, space, H5P_DEFAULT, H5P_DEFAULT, field_buf))
{
cerr << "HDF5_File_Writer::WriteData: Error, writing to dataset failed" << endl;
H5Dclose(dataset);
@ -383,6 +483,11 @@ bool HDF5_File_Writer::WriteAtrribute(std::string locName, std::string attr_name
return WriteAtrribute(locName,attr_name, value, size, H5T_NATIVE_FLOAT);
}
bool HDF5_File_Writer::WriteAtrribute(std::string locName, std::string attr_name, double const* value, hsize_t size)
{
return WriteAtrribute(locName,attr_name, value, size, H5T_NATIVE_DOUBLE);
}
bool HDF5_File_Writer::WriteAtrribute(std::string locName, std::string attr_name, vector<float> values)
{
float val[values.size()];

10
tools/hdf5_file_writer.h
View File

@ -33,14 +33,23 @@ public:
bool WriteRectMesh(unsigned int const* numLines, float const* const* discLines, int MeshType=0, float scaling=1);
bool WriteScalarField(std::string dataSetName, float const* const* const* field, size_t datasize[3]);
bool WriteScalarField(std::string dataSetName, double const* const* const* field, size_t datasize[3]);
bool WriteScalarField(std::string dataSetName, std::complex<float> const* const* const* field, size_t datasize[3]);
bool WriteScalarField(std::string dataSetName, std::complex<double> const* const* const* field, size_t datasize[3]);
bool WriteVectorField(std::string dataSetName, float const* const* const* const* field, size_t datasize[3]);
bool WriteVectorField(std::string dataSetName, double const* const* const* const* field, size_t datasize[3]);
bool WriteVectorField(std::string dataSetName, std::complex<float> const* const* const* const* field, size_t datasize[3]);
bool WriteVectorField(std::string dataSetName, std::complex<double> const* const* const* const* field, size_t datasize[3]);
bool WriteData(std::string dataSetName, float const* field_buf, size_t dim, size_t* datasize);
bool WriteData(std::string dataSetName, double const* field_buf, size_t dim, size_t* datasize);
bool WriteAtrribute(std::string locName, std::string attr_name, void const* value, hsize_t size, hid_t mem_type);
bool WriteAtrribute(std::string locName, std::string attr_name, float const* value, hsize_t size);
bool WriteAtrribute(std::string locName, std::string attr_name, double const* value, hsize_t size);
bool WriteAtrribute(std::string locName, std::string attr_name, std::vector<float> values);
bool WriteAtrribute(std::string locName, std::string attr_name, std::vector<double> values);
bool WriteAtrribute(std::string locName, std::string attr_name, float value);
@ -53,6 +62,7 @@ protected:
std::string m_Group;
hid_t OpenGroup(hid_t hdf5_file, std::string group);
bool WriteData(std::string dataSetName, hid_t mem_type, void const* field_buf, size_t dim, size_t* datasize);
};
#endif // HDF5_FILE_WRITER_H