Actual source code: ex2.cxx

  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: {
 23:   PetscBool flg;

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

 35:   PetscOptionsBegin(comm, "", "Meshing Problem Options", "DMMOAB");
 36:   PetscOptionsBool("-debug", "Enable debug messages", "ex2.cxx", options->debug, &options->debug, NULL);
 37:   PetscOptionsBool("-interlace", "Use interlaced arrangement for the field data", "ex2.cxx", options->interlace, &options->interlace, NULL);
 38:   PetscOptionsBool("-simplex", "Create simplices instead of tensor product elements", "ex2.cxx", options->simplex, &options->simplex, NULL);
 39:   PetscOptionsRangeInt("-dim", "The topological mesh dimension", "ex2.cxx", options->dim, &options->dim, NULL, 1, 3);
 40:   PetscOptionsBoundedInt("-n", "The number of elements in each dimension", "ex2.cxx", options->nele, &options->nele, NULL, 1);
 41:   PetscOptionsString("-meshfile", "The input mesh file", "ex2.cxx", options->input_file, options->input_file, sizeof(options->input_file), NULL);
 42:   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);
 43:   PetscOptionsStringArray("-fields", "The list of names of the field variables", "ex2.cxx", options->fieldnames, &options->nfields, &flg);
 44:   PetscOptionsEnd();

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

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

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

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

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

 86: int main(int argc, char **argv)
 87: {
 88:   AppCtx      user; /* user-defined work context */
 89:   PetscRandom rctx;
 90:   Vec         solution;
 91:   Mat         system;
 92:   MPI_Comm    comm;
 93:   PetscInt    i;

 96:   PetscInitialize(&argc, &argv, NULL, help);
 97:   comm = PETSC_COMM_WORLD;
 98:   ProcessOptions(comm, &user);
 99:   CreateMesh(comm, &user);

101:   /* set block size */
102:   DMMoabSetBlockSize(user.dm, (user.interlace ? user.nfields : 1));
103:   DMSetMatType(user.dm, MATAIJ);

105:   DMSetFromOptions(user.dm);

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

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

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

119:   if (user.write_output) {
120:     DMMoabSetGlobalFieldVector(user.dm, solution);
121:     if (user.debug) PetscPrintf(comm, "Output mesh and associated field data to file: %s.\n", user.output_file);
122:     DMMoabOutput(user.dm, (const char *)user.output_file, "");
123:   }

125:   if (user.nfields) {
126:     for (i = 0; i < user.nfields; i++) PetscFree(user.fieldnames[i]);
127:   }
128:   PetscRandomDestroy(&rctx);
129:   VecDestroy(&solution);
130:   MatDestroy(&system);
131:   DMDestroy(&user.dm);
132:   PetscFinalize();
133:   return 0;
134: }

136: /*TEST

138:    build:
139:      requires: moab

141:    test:
142:      args: -debug -fields v1,v2,v3

144: TEST*/