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.

Defines
#define	ERROR(rc, msg)
Functions
int	main (int argc, char **argv)

---

Define Documentation

#define ERROR	(		rc,
			msg
	)

Value:

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

Definition at line 72 of file imoab_ptest2.cpp.

Referenced by main().

---

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 79 of file imoab_ptest2.cpp.

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

{
    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" );

    // 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;
}