Actual source code: ex2.cxx
petsc-3.14.6 2021-03-30
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*/