imoab_ptest2.cpp File Reference

#include "moab/MOABConfig.h"
#include "moab_mpi.h"
#include "moab/iMOAB.h"
#include <cstdio>
#include <cstring>
#include <vector>

Include dependency graph for imoab_ptest2.cpp:

Go to the source code of this file.

Macros
#define	ERROR(rc, msg)

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

Macro Definition Documentation

ERROR

#define ERROR	(		rc,
			msg
	)

Value:

 if( 0 != ( rc ) ) \
 { \
 printf( "error %s", msg ); \
 return 1; \
 }

Definition at line 72 of file imoab_ptest2.cpp.

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 79 of file imoab_ptest2.cpp.

{
 int my_id, num_procs, ix, iy, numv, nume;
 int dime, lco, mbtype, blockid, npe;
 char appname[10] = "IMTEST";
 /* coordinates for 9 vertices */
 double coordinates[27], deltax, deltay;
 int ids[9] = { 1, 2, 3, 6, 7, 8, 11, 12, 13 };
 int connec[16] = { 1, 2, 5, 4, 2, 3, 6, 5, 4, 5, 8, 7, 5, 6, 9, 8 };
 /* used for ghosting */
 int dimgh, bridge, num_layers;
 double coords_core[27] = { 0., 0., 0., 0., 1., 0., 0., 2., 0., 1., 0., 0., 1., 1.,
 0., 1., 2., 0., 2., 0., 0., 2., 1., 0., 2., 2., 0. };
 int appID, compid;
 iMOAB_AppID pid = &appID;
 
 MPI_Init( &argc, &argv );
 MPI_Comm comm = MPI_COMM_WORLD;
 
 MPI_Comm_size( comm, &num_procs );
 MPI_Comm_rank( comm, &my_id );
 
 ErrCode rc = iMOAB_Initialize( argc, argv );
 ERROR( rc, "can't initialize" );
 if( !my_id ) printf( " I'm process %d out of %d\n", my_id, num_procs );
 
 appname[7] = 0; // make sure it is NULL
 compid = 8; // given number, external application number
 rc = iMOAB_RegisterApplication( appname, &comm, &compid, pid );
 ERROR( rc, " can't register app" );
 /* create first 9 vertices, in a square 3x3; */
 deltax = ( my_id / 2 ) * 2.;
 deltay = ( my_id % 2 ) * 2.;
 ix = ( my_id / 2 ) * 10;
 iy = my_id % 2 * 2;
 for( int i = 0; i < 9; i++ )
 {
 coordinates[3 * i] = coords_core[3 * i] + deltax;
 coordinates[3 * i + 1] = coords_core[3 * i + 1] + deltay;
 coordinates[3 * i + 2] = coords_core[3 * i + 2];
 
 /* translate the ids too, by multiples of 10 or add 2, depending on my_id*/
 ids[i] = ids[i] + ix + iy;
 }
 numv = 9;
 nume = 4;
 lco = numv * 3;
 dime = 3;
 rc = iMOAB_CreateVertices( pid, &lco, &dime, coordinates );
 ERROR( rc, "can't create vertices" );
 /* create now 4 quads with those 9 vertices*/
 mbtype = 3;
 blockid = 100;
 npe = 4;
 rc = iMOAB_CreateElements( pid, &nume, &mbtype, &npe, connec, &blockid );
 ERROR( rc, "can't create elements" );
 
 rc = iMOAB_ResolveSharedEntities( pid, &numv, ids );
 ERROR( rc, "can't resolve shared ents" );
 
 rc = iMOAB_UpdateMeshInfo( pid );
 ERROR( rc, "can't update mesh info" );
 
 int num_global_verts, num_global_cells;
 iMOAB_GetGlobalInfo( pid, &num_global_verts, &num_global_cells );
 
 // test iMOAB_ReduceTagsMax on a tag
 int tagType = DENSE_DOUBLE;
 int num_components = 1;
 int tagIndex = 0; // output
 
 rc = iMOAB_DefineTagStorage( pid, "afrac:ifrac", &tagType, &num_components, &tagIndex );
 ERROR( rc, "failed to create tags afrac:ifrac " );
 double af[4] = { 1., 1., 1., 1. };
 int entType = 1; // cells
 rc = iMOAB_SetDoubleTagStorage( pid, "afrac", &nume, &entType, af );
 ERROR( rc, "failed to set tag afrac " );
 /* write out the mesh file to disk, in parallel, if h5m*/
#ifdef MOAB_HAVE_HDF5_PARALLEL
 char wopts[100] = "PARALLEL=WRITE_PART";
 char outfile[32] = "whole.h5m";
 rc = iMOAB_WriteMesh( pid, outfile, wopts );
 ERROR( rc, "can't write mesh" );
#endif
 
 // define repeat tags
 rc = iMOAB_DefineTagStorage( pid, "ofrac:afrac:ofrad", &tagType, &num_components, &tagIndex );
 ERROR( rc, "failed to create tags ofrac:afrac:ofrad " );
 
 // write again
#ifdef MOAB_HAVE_HDF5_PARALLEL
 rc = iMOAB_WriteMesh( pid, outfile, wopts );
 ERROR( rc, "can't write mesh 3\n" );
#endif
 
 tagType = DENSE_INTEGER;
 rc = iMOAB_DefineTagStorage( pid, "INTFIELD", &tagType, &num_components, &tagIndex );
 ERROR( rc, "failed to get tag INTFIELD " );
 // set some values
 std::vector< int > valstest( numv );
 for( int k = 0; k < numv; k++ )
 {
 valstest[k] = my_id + k;
 }
 int num_tag_storage_length = numv * num_components;
 entType = 0; // vertex
 // create an integer tag
 
 rc = iMOAB_SetIntTagStorage( pid, "INTFIELD", &num_tag_storage_length, &entType, &valstest[0] );
 ERROR( rc, "failed to set tag INTFIELD " );
 
 rc = iMOAB_ReduceTagsMax( pid, &tagIndex, &entType );
 ERROR( rc, "failed reduce tags max " );
 
 /* see ghost elements */
 dimgh = 2; /* will ghost quads, topological dim 2 */
 bridge = 0; /* use vertex as bridge */
 num_layers = 1; /* so far, one layer only */
 rc = iMOAB_DetermineGhostEntities( pid, &dimgh, &num_layers, &bridge );
 ERROR( rc, "can't determine ghosts" );
 
 /* write out the mesh file to disk, in parallel, if h5m*/
#ifdef MOAB_HAVE_HDF5_PARALLEL
 rc = iMOAB_WriteMesh( pid, outfile, wopts );
 ERROR( rc, "can't write mesh" );
#endif
 /* all done. de-register and finalize */
 rc = iMOAB_DeregisterApplication( pid );
 ERROR( rc, "can't de-register app" );
 
 rc = iMOAB_Finalize();
 ERROR( rc, "can't finalize" );
 
 MPI_Finalize();
 return 0;
}

References DENSE_DOUBLE, DENSE_INTEGER, ErrCode, ERROR, iMOAB_AppID, iMOAB_CreateElements(), iMOAB_CreateVertices(), iMOAB_DefineTagStorage(), iMOAB_DeregisterApplication(), iMOAB_DetermineGhostEntities(), iMOAB_Finalize(), iMOAB_GetGlobalInfo(), iMOAB_Initialize(), iMOAB_ReduceTagsMax(), iMOAB_RegisterApplication(), iMOAB_ResolveSharedEntities(), iMOAB_SetDoubleTagStorage(), iMOAB_SetIntTagStorage(), iMOAB_UpdateMeshInfo(), iMOAB_WriteMesh(), MPI_COMM_WORLD, and outfile.