#include "moab/Core.hpp"
#include <iostream>
#include <fstream>

Include dependency graph for ReadPartFile.cpp:

Macros
#define	MESH_DIR "."

Functions
int	main (int argc, char **argv)

Variables
string	test_file_name = string( MESH_DIR ) + string( "/3k-tri-sphere.vtk" )

string	part_file_name

int	nparts

Macro Definition Documentation

◆ MESH_DIR

#define MESH_DIR "."

Examples: ReadPartFile.cpp.

Definition at line 15 of file ReadPartFile.cpp.

Function Documentation

◆ main()

int main	(	int	argc,
		char **	argv
	)

Examples: ReadPartFile.cpp.

Definition at line 22 of file ReadPartFile.cpp.

 {
     // Get MOAB instance
     Interface* mb = new( std::nothrow ) Core;
     if( NULL == mb ) return 1;
  
     // Need option handling here for input filename
     if( argc > 4 )
     {
         // User has input a mesh file
         test_file_name = argv[1];
         part_file_name = argv[2];
         nparts         = atoi( argv[3] );
     }
     else
     {
         cerr << " usage is " << argv[0] << " <input file> <part file> <#parts> <output file> \n";
         exit( 0 );
     }
     ifstream inFile;
     inFile.open( part_file_name.c_str() );
     if( !inFile )
     {
         cerr << "Unable to open file " << part_file_name << "\n";
         exit( 1 );  // call system to stop
     }
     // Load the mesh from file
     ErrorCode rval = mb->load_mesh( test_file_name.c_str() );MB_CHK_ERR( rval );
  
     // Get sets entities, by type
     Range sets;
     rval = mb->get_entities_by_type( 0, MBENTITYSET, sets );MB_CHK_ERR( rval );
  
     // Output the number of sets
     cout << "Number of sets is " << sets.size() << endl;
     // remove the sets that have a PARALLEL_PARTITION tag
  
     Tag tag;
     rval = mb->tag_get_handle( "PARALLEL_PARTITION", tag );MB_CHK_ERR( rval );
  
     int i                = 0;
     int num_deleted_sets = 0;
     for( Range::iterator it = sets.begin(); it != sets.end(); it++, i++ )
     {
         EntityHandle eh = *it;
         // cout << " set :" << mb->id_from_handle(eh) <<"\n";
         int val = -1;
         rval    = mb->tag_get_data( tag, &eh, 1, &val );
         if( val != -1 )
         {
             num_deleted_sets++;
             rval = mb->delete_entities( &eh, 1 );  // delete the set, we will have a new partition soon
         }
     }
     if( num_deleted_sets ) cout << "delete " << num_deleted_sets << " existing  partition sets, and create new ones \n";
     Range cells;  // get them by dimension 2!
     rval = mb->get_entities_by_dimension( 0, 2, cells );MB_CHK_ERR( rval );
     EntityHandle* psets = new EntityHandle[nparts];
     for( int i = 0; i < nparts; i++ )
     {
         rval = mb->create_meshset( MESHSET_SET, psets[i] );MB_CHK_ERR( rval );
         rval = mb->tag_set_data( tag, &( psets[i] ), 1, &i );MB_CHK_ERR( rval );
     }
  
     for( Range::iterator it = cells.begin(); it != cells.end(); it++ )
     {
         int part;
         EntityHandle eh = *it;
         inFile >> part;
         rval = mb->add_entities( psets[part], &eh, 1 );MB_CHK_ERR( rval );
     }
     mb->write_file( argv[4] );
  
     delete mb;
  
     return 0;
 }