110 lines
3.4 KiB
C++
110 lines
3.4 KiB
C++
#include "IO/silo.h"
|
|
#include "common/MPI.h"
|
|
#include "common/Utilities.h"
|
|
|
|
#include "ProfilerApp.h"
|
|
|
|
|
|
#ifdef USE_SILO
|
|
|
|
#include <silo.h>
|
|
|
|
|
|
namespace IO {
|
|
namespace silo {
|
|
|
|
|
|
/****************************************************
|
|
* Open/close a file *
|
|
****************************************************/
|
|
DBfile *open( const std::string &filename, FileMode mode )
|
|
{
|
|
DBfile *fid = nullptr;
|
|
if ( mode == CREATE ) {
|
|
fid = DBCreate( filename.c_str(), DB_CLOBBER, DB_LOCAL, nullptr, DB_HDF5 );
|
|
} else if ( mode == WRITE ) {
|
|
fid = DBOpen( filename.c_str(), DB_HDF5, DB_APPEND );
|
|
} else if ( mode == READ ) {
|
|
fid = DBOpen( filename.c_str(), DB_HDF5, DB_READ );
|
|
}
|
|
return fid;
|
|
}
|
|
void close( DBfile *fid ) { DBClose( fid ); }
|
|
|
|
|
|
/****************************************************
|
|
* Helper functions *
|
|
****************************************************/
|
|
DataType varDataType( DBfile *fid, const std::string &name )
|
|
{
|
|
auto type = DBGetVarType( fid, name.c_str() );
|
|
DataType type2 = DataType::Null;
|
|
if ( type == DB_DOUBLE )
|
|
type2 = DataType::Double;
|
|
else if ( type == DB_FLOAT )
|
|
type2 = DataType::Float;
|
|
else if ( type == DB_INT )
|
|
type2 = DataType::Int;
|
|
return type2;
|
|
}
|
|
|
|
|
|
/****************************************************
|
|
* Write/read a uniform mesh to silo *
|
|
****************************************************/
|
|
void readUniformMesh(
|
|
DBfile *fid, const std::string &meshname, std::vector<double> &range, std::vector<int> &N )
|
|
{
|
|
DBquadmesh *mesh = DBGetQuadmesh( fid, meshname.c_str() );
|
|
int ndim = mesh->ndims;
|
|
range.resize( 2 * ndim );
|
|
N.resize( ndim );
|
|
for ( int d = 0; d < ndim; d++ ) {
|
|
N[d] = mesh->dims[d] - 1;
|
|
range[2 * d + 0] = mesh->min_extents[d];
|
|
range[2 * d + 1] = mesh->max_extents[d];
|
|
}
|
|
DBFreeQuadmesh( mesh );
|
|
}
|
|
|
|
|
|
/****************************************************
|
|
* Write a multimesh *
|
|
****************************************************/
|
|
void writeMultiMesh( DBfile *fid, const std::string &meshname,
|
|
const std::vector<std::string> &meshNames, const std::vector<int> &meshTypes )
|
|
{
|
|
std::vector<char *> meshnames( meshNames.size() );
|
|
for ( size_t i = 0; i < meshNames.size(); ++i )
|
|
meshnames[i] = (char *) meshNames[i].c_str();
|
|
std::string tree_name = meshname + "_tree";
|
|
DBoptlist *optList = DBMakeOptlist( 1 );
|
|
DBAddOption( optList, DBOPT_MRGTREE_NAME, (char *) tree_name.c_str() );
|
|
DBPutMultimesh( fid, meshname.c_str(), meshNames.size(), meshnames.data(),
|
|
(int *) meshTypes.data(), nullptr );
|
|
DBFreeOptlist( optList );
|
|
}
|
|
|
|
|
|
/****************************************************
|
|
* Write a multivariable *
|
|
****************************************************/
|
|
void writeMultiVar( DBfile *fid, const std::string &varname,
|
|
const std::vector<std::string> &varNames, const std::vector<int> &varTypes )
|
|
{
|
|
std::vector<char *> varnames( varNames.size(), nullptr );
|
|
for ( size_t j = 0; j < varNames.size(); j++ )
|
|
varnames[j] = const_cast<char *>( varNames[j].c_str() );
|
|
DBPutMultivar(
|
|
fid, varname.c_str(), varNames.size(), varnames.data(), (int *) varTypes.data(), nullptr );
|
|
}
|
|
|
|
|
|
} // namespace silo
|
|
} // namespace IO
|
|
|
|
|
|
#else
|
|
|
|
#endif
|