ReadWriteTest.cpp

Go to the documentation of this file.

/** @example ReadWriteTest.cpp \n
 * \brief Read mesh into MOAB and write some back \n
 *
 * <b>To run</b>: mpiexec -np 4 ReadWriteTest [input] [output] -O [read_opts] -o [write_opts]\n
 *
 * used for stress test of reader/writer
 * report times to read and write
 *
 * example ReadWriteTest ../MeshFiles/io/fv3x46x72.t.3.nc out.nc \
 * -O PARALLEL=READ_PART;PARTITION_METHOD=SQIJ;PARALLEL_RESOLVE_SHARED_ENTS;VARIABLE=T,U; \
 * -o PARALLEL=WRITE_PART;VARIABLE=T,U
 */
 
#include "moab/Core.hpp"
#ifdef MOAB_HAVE_MPI
#include "moab/ParallelComm.hpp"
#include "MBParallelConventions.h"
#endif
#include <iostream>
#include <ctime>
 
using namespace moab;
using namespace std;
 
int main( int argc, char** argv )
{
#ifdef MOAB_HAVE_MPI
 string input_file, output_file, read_opts, write_opts;
 
 MPI_Init( &argc, &argv );
 
 // Need option handling here for input filename
 if( argc < 3 )
 {
#ifdef MOAB_HAVE_NETCDF
 input_file = string( MESH_DIR ) + string( "/io/fv3x46x72.t.3.nc" );
 output_file = "ReadWriteTestOut.h5m";
 read_opts = "PARALLEL=READ_PART;PARTITION_METHOD=SQIJ;PARALLEL_RESOLVE_SHARED_ENTS;VARIABLE=T,U";
 write_opts = "PARALLEL=WRITE_PART";
#else
 cout << "Usage: mpiexec -n $NP ReadWriteTest [input] [output] -O <read_opts> -o "
 "<write_opts>\n";
 return 0;
#endif
 }
 else
 {
 input_file = argv[1];
 output_file = argv[2];
 }
 
 if( argc > 3 )
 {
 int index = 3;
 while( index < argc )
 {
 if( !strcmp( argv[index], "-O" ) ) // This is for reading options, optional
 read_opts = argv[++index];
 if( !strcmp( argv[index], "-o" ) ) write_opts = argv[++index];
 index++;
 }
 }
 
 // Get MOAB instance
 Interface* mb = new( std::nothrow ) Core;
 if( NULL == mb ) return 1;
 
 // Get the ParallelComm instance
 ParallelComm* pcomm = new ParallelComm( mb, MPI_COMM_WORLD );
 int nprocs = pcomm->proc_config().proc_size();
 int rank = pcomm->proc_config().proc_rank();
 
 EntityHandle set;
 ErrorCode rval = mb->create_meshset( MESHSET_SET, set );MB_CHK_ERR( rval );
 
 clock_t tt = clock();
 
 if( 0 == rank )
 cout << "Reading file " << input_file << "\n with options: " << read_opts << "\n on " << nprocs
 << " processors\n";
 
 // Read the file with the specified options
 rval = mb->load_file( input_file.c_str(), &set, read_opts.c_str() );MB_CHK_ERR( rval );
 
 if( 0 == rank )
 {
 cout << "Time: " << ( clock() - tt ) / (double)CLOCKS_PER_SEC << " seconds" << endl;
 tt = clock();
 }
 
 // Write back the file with the specified options
 rval = mb->write_file( output_file.c_str(), 0, write_opts.c_str(), &set, 1 );MB_CHK_ERR( rval );
 
 if( 0 == rank )
 {
 cout << "Writing file " << output_file << "\n with options: " << write_opts << endl;
 cout << "Time: " << ( clock() - tt ) / (double)CLOCKS_PER_SEC << " seconds" << endl;
 tt = clock();
 }
 
 delete mb;
 
 MPI_Finalize();
#else
 std::cout << " compile MOAB with mpi for this example to work\n";
#endif
 
 return 0;
}