Actual source code: ex2.cxx

petsc-3.14.6 2021-03-30
Report Typos and Errors
  1: static char help[] = "Create a box mesh with DMMoab and test defining a tag on the mesh\n\n";

  3: #include <petscdmmoab.h>

  5: typedef struct {
  6:   DM            dm;                /* DM implementation using the MOAB interface */
  7:   PetscBool     debug;             /* The debugging level */
  8:   PetscLogEvent createMeshEvent;
  9:   /* Domain and mesh definition */
 10:   PetscInt      dim;                            /* The topological mesh dimension */
 11:   PetscInt      nele;                           /* Elements in each dimension */
 12:   PetscBool     simplex;                        /* Use simplex elements */
 13:   PetscBool     interlace;
 14:   char          input_file[PETSC_MAX_PATH_LEN];   /* Import mesh from file */
 15:   char          output_file[PETSC_MAX_PATH_LEN];   /* Output mesh file name */
 16:   PetscBool     write_output;                        /* Write output mesh and data to file */
 17:   PetscInt      nfields;         /* Number of fields */
 18:   char          *fieldnames[PETSC_MAX_PATH_LEN]; /* Name of a defined field on the mesh */
 19: } AppCtx;

 21: PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options)
 22: {
 24:   PetscBool      flg;

 27:   options->debug             = PETSC_FALSE;
 28:   options->dim               = 2;
 29:   options->nele              = 5;
 30:   options->nfields           = 256;
 31:   options->simplex           = PETSC_FALSE;
 32:   options->write_output      = PETSC_FALSE;
 33:   options->interlace         = PETSC_FALSE;
 34:   options->input_file[0]     = '\0';
 35:   PetscStrcpy(options->output_file,"ex2.h5m");

 37:   PetscOptionsBegin(comm, "", "Meshing Problem Options", "DMMOAB");
 38:   PetscOptionsBool("-debug", "Enable debug messages", "ex2.cxx", options->debug, &options->debug, NULL);
 39:   PetscOptionsBool("-interlace", "Use interlaced arrangement for the field data", "ex2.cxx", options->interlace, &options->interlace, NULL);
 40:   PetscOptionsBool("-simplex", "Create simplices instead of tensor product elements", "ex2.cxx", options->simplex, &options->simplex, NULL);
 41:   PetscOptionsRangeInt("-dim", "The topological mesh dimension", "ex2.cxx", options->dim, &options->dim, NULL,1,3);
 42:   PetscOptionsBoundedInt("-n", "The number of elements in each dimension", "ex2.cxx", options->nele, &options->nele, NULL,1);
 43:   PetscOptionsString("-meshfile", "The input mesh file", "ex2.cxx", options->input_file, options->input_file, sizeof(options->input_file), NULL);
 44:   PetscOptionsString("-io", "Write out the mesh and solution that is defined on it (Default H5M format)", "ex2.cxx", options->output_file, options->output_file, sizeof(options->output_file), &options->write_output);
 45:   PetscOptionsStringArray("-fields", "The list of names of the field variables", "ex2.cxx", options->fieldnames,&options->nfields, &flg);
 46:   PetscOptionsEnd();

 48:   if (options->debug) PetscPrintf(comm, "Total number of fields: %D.\n",options->nfields);
 49:   if (!flg) { /* if no field names were given by user, assign a default */
 50:     options->nfields = 1;
 51:     PetscStrallocpy("TestEX2Var",&options->fieldnames[0]);
 52:   }

 54:   PetscLogEventRegister("CreateMesh",          DM_CLASSID,   &options->createMeshEvent);
 55:   return(0);
 56: }

 58: PetscErrorCode CreateMesh(MPI_Comm comm, AppCtx *user)
 59: {
 60:   PetscInt       i;
 61:   size_t         len;
 62:   PetscMPIInt    rank;

 66:   PetscLogEventBegin(user->createMeshEvent,0,0,0,0);
 67:   MPI_Comm_rank(comm, &rank);
 68:   PetscStrlen(user->input_file, &len);
 69:   if (len) {
 70:     if (user->debug) PetscPrintf(comm, "Loading mesh from file: %s and creating a DM object.\n",user->input_file);
 71:     DMMoabLoadFromFile(comm, user->dim, 1, user->input_file, "", &user->dm);
 72:   }
 73:   else {
 74:     if (user->debug) {
 75:       PetscPrintf(comm, "Creating a %D-dimensional structured %s mesh of %Dx%Dx%D in memory and creating a DM object.\n",user->dim,(user->simplex?"simplex":"regular"),user->nele,user->nele,user->nele);
 76:     }
 77:     DMMoabCreateBoxMesh(comm, user->dim, user->simplex, NULL, user->nele, 1, &user->dm);
 78:   }

 80:   if (user->debug) {
 81:     PetscPrintf(comm, "Setting field names to DM: \n");
 82:     for (i=0; i<user->nfields; i++)
 83:       PetscPrintf(comm, "\t Field{%D} = %s.\n",i,user->fieldnames[i]);
 84:   }
 85:   DMMoabSetFieldNames(user->dm, user->nfields, (const char**)user->fieldnames);
 86:   PetscObjectSetName((PetscObject)user->dm, "Structured Mesh");
 87:   PetscLogEventEnd(user->createMeshEvent,0,0,0,0);
 88:   return(0);
 89: }

 91: int main(int argc, char **argv)
 92: {
 93:   AppCtx         user;                 /* user-defined work context */
 94:   PetscRandom    rctx;
 95:   Vec            solution;
 96:   Mat            system;
 97:   MPI_Comm       comm;
 98:   PetscInt       i;

101:   PetscInitialize(&argc, &argv, NULL, help);if (ierr) return ierr;
102:   comm = PETSC_COMM_WORLD;
103:   ProcessOptions(comm, &user);
104:   CreateMesh(comm, &user);

106:   /* set block size */
107:   DMMoabSetBlockSize(user.dm, (user.interlace?user.nfields:1));
108:   DMSetMatType(user.dm,MATAIJ);

110:   DMSetFromOptions(user.dm);

112:   /* SetUp the data structures for DMMOAB */
113:   DMSetUp(user.dm);

115:   if (user.debug) PetscPrintf(comm, "Creating a global vector defined on DM and setting random data.\n");
116:   DMCreateGlobalVector(user.dm,&solution);
117:   PetscRandomCreate(comm,&rctx);
118:   VecSetRandom(solution,rctx);

120:   /* test if matrix allocation for the prescribed matrix type is done correctly */
121:   if (user.debug) PetscPrintf(comm, "Creating a global matrix defined on DM with the right block structure.\n");
122:   DMCreateMatrix(user.dm,&system);

124:   if (user.write_output) {
125:     DMMoabSetGlobalFieldVector(user.dm, solution);
126:     if (user.debug) PetscPrintf(comm, "Output mesh and associated field data to file: %s.\n",user.output_file);
127:     DMMoabOutput(user.dm,(const char*)user.output_file,"");
128:   }

130:   if (user.nfields) {
131:     for (i=0; i<user.nfields; i++) {
132:       PetscFree(user.fieldnames[i]);
133:     }
134:   }
135:   PetscRandomDestroy(&rctx);
136:   VecDestroy(&solution);
137:   MatDestroy(&system);
138:   DMDestroy(&user.dm);
139:   PetscFinalize();
140:   return ierr;
141: }

143: /*TEST

145:    build:
146:      requires: moab

148:    test:
149:      args: -debug -fields v1,v2,v3

151: TEST*/