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*/