xref: /petsc/src/dm/impls/plex/plexexodusii.c (revision a1cb98fac0cdf0eb4d3e8a0c8b58f3fe8f800bc6)
1552f7358SJed Brown #define PETSCDM_DLL
2af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h> /*I   "petscdmplex.h"   I*/
3552f7358SJed Brown 
4519f805aSKarl Rupp #if defined(PETSC_HAVE_EXODUSII)
5552f7358SJed Brown   #include <netcdf.h>
6552f7358SJed Brown   #include <exodusII.h>
7552f7358SJed Brown #endif
8552f7358SJed Brown 
91e50132fSMatthew G. Knepley #include <petsc/private/viewerimpl.h>
101e50132fSMatthew G. Knepley #include <petsc/private/viewerexodusiiimpl.h>
11e0266925SMatthew G. Knepley #if defined(PETSC_HAVE_EXODUSII)
12*a1cb98faSBarry Smith /*@C
13*a1cb98faSBarry Smith   PETSC_VIEWER_EXODUSII_ - Creates an `PETSCVIEWEREXODUSII` `PetscViewer` shared by all processors in a communicator.
141e50132fSMatthew G. Knepley 
151e50132fSMatthew G. Knepley   Collective
161e50132fSMatthew G. Knepley 
171e50132fSMatthew G. Knepley   Input Parameter:
18*a1cb98faSBarry Smith . comm - the MPI communicator to share the `PETSCVIEWEREXODUSII` `PetscViewer`
191e50132fSMatthew G. Knepley 
201e50132fSMatthew G. Knepley   Level: intermediate
211e50132fSMatthew G. Knepley 
22*a1cb98faSBarry Smith   Note:
231e50132fSMatthew G. Knepley   Unlike almost all other PETSc routines, PETSC_VIEWER_EXODUSII_ does not return
241e50132fSMatthew G. Knepley   an error code.  The GLVIS PetscViewer is usually used in the form
251e50132fSMatthew G. Knepley $       XXXView(XXX object, PETSC_VIEWER_EXODUSII_(comm));
261e50132fSMatthew G. Knepley 
27*a1cb98faSBarry Smith   Fortran Note:
28*a1cb98faSBarry Smith   No support in Fortran
29*a1cb98faSBarry Smith 
30*a1cb98faSBarry Smith .seealso: `PETSCVIEWEREXODUSII`, `PetscViewer`, `PetscViewer`, `PetscViewerExodusIIOpen()`, `PetscViewerType`, `PetscViewerCreate()`, `PetscViewerDestroy()`
31*a1cb98faSBarry Smith @*/
32d71ae5a4SJacob Faibussowitsch PetscViewer PETSC_VIEWER_EXODUSII_(MPI_Comm comm)
33d71ae5a4SJacob Faibussowitsch {
341e50132fSMatthew G. Knepley   PetscViewer    viewer;
351e50132fSMatthew G. Knepley   PetscErrorCode ierr;
361e50132fSMatthew G. Knepley 
371e50132fSMatthew G. Knepley   PetscFunctionBegin;
381e50132fSMatthew G. Knepley   ierr = PetscViewerExodusIIOpen(comm, "mesh.exo", FILE_MODE_WRITE, &viewer);
399371c9d4SSatish Balay   if (ierr) {
409371c9d4SSatish Balay     PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_EXODUSII_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " ");
4111cc89d2SBarry Smith     PetscFunctionReturn(NULL);
429371c9d4SSatish Balay   }
431e50132fSMatthew G. Knepley   ierr = PetscObjectRegisterDestroy((PetscObject)viewer);
449371c9d4SSatish Balay   if (ierr) {
459371c9d4SSatish Balay     PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_EXODUSII_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " ");
4611cc89d2SBarry Smith     PetscFunctionReturn(NULL);
479371c9d4SSatish Balay   }
481e50132fSMatthew G. Knepley   PetscFunctionReturn(viewer);
491e50132fSMatthew G. Knepley }
501e50132fSMatthew G. Knepley 
51d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerView_ExodusII(PetscViewer v, PetscViewer viewer)
52d71ae5a4SJacob Faibussowitsch {
536823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)v->data;
541e50132fSMatthew G. Knepley 
551e50132fSMatthew G. Knepley   PetscFunctionBegin;
569566063dSJacob Faibussowitsch   if (exo->filename) PetscCall(PetscViewerASCIIPrintf(viewer, "Filename:    %s\n", exo->filename));
579566063dSJacob Faibussowitsch   if (exo->exoid) PetscCall(PetscViewerASCIIPrintf(viewer, "exoid:       %d\n", exo->exoid));
589566063dSJacob Faibussowitsch   if (exo->btype) PetscCall(PetscViewerASCIIPrintf(viewer, "IO Mode:     %d\n", exo->btype));
59197f6770SJed Brown   if (exo->order) PetscCall(PetscViewerASCIIPrintf(viewer, "Mesh order:  %" PetscInt_FMT "\n", exo->order));
601e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
611e50132fSMatthew G. Knepley }
621e50132fSMatthew G. Knepley 
63d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerSetFromOptions_ExodusII(PetscViewer v, PetscOptionItems *PetscOptionsObject)
64d71ae5a4SJacob Faibussowitsch {
651e50132fSMatthew G. Knepley   PetscFunctionBegin;
66d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "ExodusII PetscViewer Options");
67d0609cedSBarry Smith   PetscOptionsHeadEnd();
681e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
691e50132fSMatthew G. Knepley }
701e50132fSMatthew G. Knepley 
71d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerSetUp_ExodusII(PetscViewer viewer)
72d71ae5a4SJacob Faibussowitsch {
731e50132fSMatthew G. Knepley   PetscFunctionBegin;
741e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
751e50132fSMatthew G. Knepley }
761e50132fSMatthew G. Knepley 
77d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerDestroy_ExodusII(PetscViewer viewer)
78d71ae5a4SJacob Faibussowitsch {
791e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
801e50132fSMatthew G. Knepley 
811e50132fSMatthew G. Knepley   PetscFunctionBegin;
8248a46eb9SPierre Jolivet   if (exo->exoid >= 0) PetscCallExternal(ex_close, exo->exoid);
839566063dSJacob Faibussowitsch   PetscCall(PetscFree(exo->filename));
849566063dSJacob Faibussowitsch   PetscCall(PetscFree(exo));
859566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetName_C", NULL));
869566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetName_C", NULL));
879566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetMode_C", NULL));
882e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetMode_C", NULL));
899566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerGetId_C", NULL));
909566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerGetOrder_C", NULL));
919566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerSetOrder_C", NULL));
921e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
931e50132fSMatthew G. Knepley }
941e50132fSMatthew G. Knepley 
95d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetName_ExodusII(PetscViewer viewer, const char name[])
96d71ae5a4SJacob Faibussowitsch {
971e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
981e50132fSMatthew G. Knepley   PetscMPIInt           rank;
991e50132fSMatthew G. Knepley   int                   CPU_word_size, IO_word_size, EXO_mode;
1006823f3c5SBlaise Bourdin   MPI_Info              mpi_info = MPI_INFO_NULL;
1016823f3c5SBlaise Bourdin   float                 EXO_version;
1021e50132fSMatthew G. Knepley 
1039566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
1041e50132fSMatthew G. Knepley   CPU_word_size = sizeof(PetscReal);
1051e50132fSMatthew G. Knepley   IO_word_size  = sizeof(PetscReal);
1061e50132fSMatthew G. Knepley 
1071e50132fSMatthew G. Knepley   PetscFunctionBegin;
1086823f3c5SBlaise Bourdin   if (exo->exoid >= 0) {
109792fecdfSBarry Smith     PetscCallExternal(ex_close, exo->exoid);
1106823f3c5SBlaise Bourdin     exo->exoid = -1;
1116823f3c5SBlaise Bourdin   }
1129566063dSJacob Faibussowitsch   if (exo->filename) PetscCall(PetscFree(exo->filename));
1139566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(name, &exo->filename));
1141e50132fSMatthew G. Knepley   switch (exo->btype) {
115d71ae5a4SJacob Faibussowitsch   case FILE_MODE_READ:
116d71ae5a4SJacob Faibussowitsch     EXO_mode = EX_READ;
117d71ae5a4SJacob Faibussowitsch     break;
1181e50132fSMatthew G. Knepley   case FILE_MODE_APPEND:
1196823f3c5SBlaise Bourdin   case FILE_MODE_UPDATE:
1206823f3c5SBlaise Bourdin   case FILE_MODE_APPEND_UPDATE:
1216823f3c5SBlaise Bourdin     /* Will fail if the file does not already exist */
1226823f3c5SBlaise Bourdin     EXO_mode = EX_WRITE;
1231e50132fSMatthew G. Knepley     break;
1241e50132fSMatthew G. Knepley   case FILE_MODE_WRITE:
1256823f3c5SBlaise Bourdin     /*
1266823f3c5SBlaise Bourdin       exodus only allows writing geometry upon file creation, so we will let DMView create the file.
1276823f3c5SBlaise Bourdin     */
1286823f3c5SBlaise Bourdin     PetscFunctionReturn(0);
1291e50132fSMatthew G. Knepley     break;
130d71ae5a4SJacob Faibussowitsch   default:
131d71ae5a4SJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
1321e50132fSMatthew G. Knepley   }
1331e50132fSMatthew G. Knepley   #if defined(PETSC_USE_64BIT_INDICES)
1341e50132fSMatthew G. Knepley   EXO_mode += EX_ALL_INT64_API;
1351e50132fSMatthew G. Knepley   #endif
1366823f3c5SBlaise Bourdin   exo->exoid = ex_open_par(name, EXO_mode, &CPU_word_size, &IO_word_size, &EXO_version, PETSC_COMM_WORLD, mpi_info);
13708401ef6SPierre Jolivet   PetscCheck(exo->exoid >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open_par failed for %s", name);
1381e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1391e50132fSMatthew G. Knepley }
1401e50132fSMatthew G. Knepley 
141d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetName_ExodusII(PetscViewer viewer, const char **name)
142d71ae5a4SJacob Faibussowitsch {
1431e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1441e50132fSMatthew G. Knepley 
1451e50132fSMatthew G. Knepley   PetscFunctionBegin;
1461e50132fSMatthew G. Knepley   *name = exo->filename;
1471e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1481e50132fSMatthew G. Knepley }
1491e50132fSMatthew G. Knepley 
150d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetMode_ExodusII(PetscViewer viewer, PetscFileMode type)
151d71ae5a4SJacob Faibussowitsch {
1521e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1531e50132fSMatthew G. Knepley 
1541e50132fSMatthew G. Knepley   PetscFunctionBegin;
1551e50132fSMatthew G. Knepley   exo->btype = type;
1561e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1571e50132fSMatthew G. Knepley }
1581e50132fSMatthew G. Knepley 
159d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetMode_ExodusII(PetscViewer viewer, PetscFileMode *type)
160d71ae5a4SJacob Faibussowitsch {
1611e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1621e50132fSMatthew G. Knepley 
1631e50132fSMatthew G. Knepley   PetscFunctionBegin;
1641e50132fSMatthew G. Knepley   *type = exo->btype;
1651e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1661e50132fSMatthew G. Knepley }
1671e50132fSMatthew G. Knepley 
168d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerExodusIIGetId_ExodusII(PetscViewer viewer, int *exoid)
169d71ae5a4SJacob Faibussowitsch {
1701e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1711e50132fSMatthew G. Knepley 
1721e50132fSMatthew G. Knepley   PetscFunctionBegin;
1731e50132fSMatthew G. Knepley   *exoid = exo->exoid;
1741e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1751e50132fSMatthew G. Knepley }
1761e50132fSMatthew G. Knepley 
177d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerExodusIIGetOrder_ExodusII(PetscViewer viewer, PetscInt *order)
178d71ae5a4SJacob Faibussowitsch {
1796823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1801e50132fSMatthew G. Knepley 
1811e50132fSMatthew G. Knepley   PetscFunctionBegin;
1826823f3c5SBlaise Bourdin   *order = exo->order;
1836823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
1846823f3c5SBlaise Bourdin }
1856823f3c5SBlaise Bourdin 
186d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerExodusIISetOrder_ExodusII(PetscViewer viewer, PetscInt order)
187d71ae5a4SJacob Faibussowitsch {
1886823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
1896823f3c5SBlaise Bourdin 
1906823f3c5SBlaise Bourdin   PetscFunctionBegin;
1916823f3c5SBlaise Bourdin   exo->order = order;
1921e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1931e50132fSMatthew G. Knepley }
1941e50132fSMatthew G. Knepley 
1951e50132fSMatthew G. Knepley /*MC
19600373969SVaclav Hapla    PETSCVIEWEREXODUSII - A viewer that writes to an Exodus II file
1971e50132fSMatthew G. Knepley 
198db781477SPatrick Sanan .seealso: `PetscViewerExodusIIOpen()`, `PetscViewerCreate()`, `PETSCVIEWERBINARY`, `PETSCVIEWERHDF5`, `DMView()`,
199db781477SPatrick Sanan           `PetscViewerFileSetName()`, `PetscViewerFileSetMode()`, `PetscViewerFormat`, `PetscViewerType`, `PetscViewerSetType()`
2001e50132fSMatthew G. Knepley 
2011e50132fSMatthew G. Knepley   Level: beginner
2021e50132fSMatthew G. Knepley M*/
2031e50132fSMatthew G. Knepley 
204d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PetscViewerCreate_ExodusII(PetscViewer v)
205d71ae5a4SJacob Faibussowitsch {
2061e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo;
2071e50132fSMatthew G. Knepley 
2081e50132fSMatthew G. Knepley   PetscFunctionBegin;
2094dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&exo));
2101e50132fSMatthew G. Knepley 
2111e50132fSMatthew G. Knepley   v->data                = (void *)exo;
2121e50132fSMatthew G. Knepley   v->ops->destroy        = PetscViewerDestroy_ExodusII;
2131e50132fSMatthew G. Knepley   v->ops->setfromoptions = PetscViewerSetFromOptions_ExodusII;
2141e50132fSMatthew G. Knepley   v->ops->setup          = PetscViewerSetUp_ExodusII;
2151e50132fSMatthew G. Knepley   v->ops->view           = PetscViewerView_ExodusII;
2161e50132fSMatthew G. Knepley   v->ops->flush          = 0;
2171e50132fSMatthew G. Knepley   exo->btype             = (PetscFileMode)-1;
2181e50132fSMatthew G. Knepley   exo->filename          = 0;
2191e50132fSMatthew G. Knepley   exo->exoid             = -1;
2201e50132fSMatthew G. Knepley 
2219566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetName_C", PetscViewerFileSetName_ExodusII));
2229566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetName_C", PetscViewerFileGetName_ExodusII));
2239566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetMode_C", PetscViewerFileSetMode_ExodusII));
2249566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetMode_C", PetscViewerFileGetMode_ExodusII));
2259566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerGetId_C", PetscViewerExodusIIGetId_ExodusII));
2269566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerSetOrder_C", PetscViewerExodusIISetOrder_ExodusII));
2279566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerGetOrder_C", PetscViewerExodusIIGetOrder_ExodusII));
2281e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
2291e50132fSMatthew G. Knepley }
2301e50132fSMatthew G. Knepley 
2311e50132fSMatthew G. Knepley /*
23278569bb4SMatthew G. Knepley   EXOGetVarIndex - Locate a result in an exodus file based on its name
23378569bb4SMatthew G. Knepley 
2346823f3c5SBlaise Bourdin   Collective
23578569bb4SMatthew G. Knepley 
23678569bb4SMatthew G. Knepley   Input Parameters:
23778569bb4SMatthew G. Knepley + exoid    - the exodus id of a file (obtained from ex_open or ex_create for instance)
23878569bb4SMatthew G. Knepley . obj_type - the type of entity for instance EX_NODAL, EX_ELEM_BLOCK
23978569bb4SMatthew G. Knepley - name     - the name of the result
24078569bb4SMatthew G. Knepley 
24178569bb4SMatthew G. Knepley   Output Parameters:
24278569bb4SMatthew G. Knepley . varIndex - the location in the exodus file of the result
24378569bb4SMatthew G. Knepley 
24478569bb4SMatthew G. Knepley   Notes:
24578569bb4SMatthew G. Knepley   The exodus variable index is obtained by comparing name and the
24678569bb4SMatthew G. Knepley   names of zonal variables declared in the exodus file. For instance if name is "V"
24778569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
24878569bb4SMatthew G. Knepley   amongst all variables of type obj_type.
24978569bb4SMatthew G. Knepley 
25078569bb4SMatthew G. Knepley   Level: beginner
25178569bb4SMatthew G. Knepley 
252c2e3fba1SPatrick Sanan .seealso: `EXOGetVarIndex()`, `DMPlexView_ExodusII_Internal()`, `VecViewPlex_ExodusII_Nodal_Internal()`, `VecLoadNodal_PlexEXO()`, `VecLoadZonal_PlexEXO()`
25378569bb4SMatthew G. Knepley */
254d71ae5a4SJacob Faibussowitsch PetscErrorCode EXOGetVarIndex_Internal(int exoid, ex_entity_type obj_type, const char name[], int *varIndex)
255d71ae5a4SJacob Faibussowitsch {
2566de834b4SMatthew G. Knepley   int       num_vars, i, j;
25778569bb4SMatthew G. Knepley   char      ext_name[MAX_STR_LENGTH + 1], var_name[MAX_STR_LENGTH + 1];
25878569bb4SMatthew G. Knepley   const int num_suffix = 5;
25978569bb4SMatthew G. Knepley   char     *suffix[5];
26078569bb4SMatthew G. Knepley   PetscBool flg;
26178569bb4SMatthew G. Knepley 
26278569bb4SMatthew G. Knepley   PetscFunctionBegin;
26378569bb4SMatthew G. Knepley   suffix[0] = (char *)"";
26478569bb4SMatthew G. Knepley   suffix[1] = (char *)"_X";
26578569bb4SMatthew G. Knepley   suffix[2] = (char *)"_XX";
26678569bb4SMatthew G. Knepley   suffix[3] = (char *)"_1";
26778569bb4SMatthew G. Knepley   suffix[4] = (char *)"_11";
26878569bb4SMatthew G. Knepley 
2696823f3c5SBlaise Bourdin   *varIndex = -1;
270792fecdfSBarry Smith   PetscCallExternal(ex_get_variable_param, exoid, obj_type, &num_vars);
27178569bb4SMatthew G. Knepley   for (i = 0; i < num_vars; ++i) {
272792fecdfSBarry Smith     PetscCallExternal(ex_get_variable_name, exoid, obj_type, i + 1, var_name);
27378569bb4SMatthew G. Knepley     for (j = 0; j < num_suffix; ++j) {
2749566063dSJacob Faibussowitsch       PetscCall(PetscStrncpy(ext_name, name, MAX_STR_LENGTH));
2759566063dSJacob Faibussowitsch       PetscCall(PetscStrlcat(ext_name, suffix[j], MAX_STR_LENGTH));
2769566063dSJacob Faibussowitsch       PetscCall(PetscStrcasecmp(ext_name, var_name, &flg));
277ad540459SPierre Jolivet       if (flg) *varIndex = i + 1;
2786823f3c5SBlaise Bourdin     }
2796823f3c5SBlaise Bourdin   }
28078569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
28178569bb4SMatthew G. Knepley }
28278569bb4SMatthew G. Knepley 
28378569bb4SMatthew G. Knepley /*
2846823f3c5SBlaise Bourdin   DMView_PlexExodusII - Write a DM to disk in exodus format
28578569bb4SMatthew G. Knepley 
28678569bb4SMatthew G. Knepley   Collective on dm
28778569bb4SMatthew G. Knepley 
28878569bb4SMatthew G. Knepley   Input Parameters:
28978569bb4SMatthew G. Knepley + dm  - The dm to be written
2906823f3c5SBlaise Bourdin . viewer - an exodusII viewer
29178569bb4SMatthew G. Knepley 
29278569bb4SMatthew G. Knepley   Notes:
29378569bb4SMatthew G. Knepley   Not all DM can be written to disk this way. For instance, exodus assume that element blocks (mapped to "Cell sets" labels)
29478569bb4SMatthew G. Knepley   consists of sequentially numbered cells. If this is not the case, the exodus file will be corrupted.
29578569bb4SMatthew G. Knepley 
2966823f3c5SBlaise Bourdin   If the dm has been distributed, only the part of the DM on MPI rank 0 (including "ghost" cells and vertices)
2976823f3c5SBlaise Bourdin   will be written.
29878569bb4SMatthew G. Knepley 
29978569bb4SMatthew G. Knepley   DMPlex only represents geometry while most post-processing software expect that a mesh also provides information
30078569bb4SMatthew G. Knepley   on the discretization space. This function assumes that the file represents Lagrange finite elements of order 1 or 2.
3016823f3c5SBlaise Bourdin   The order of the mesh shall be set using PetscViewerExodusIISetOrder
30278569bb4SMatthew G. Knepley   It should be extended to use PetscFE objects.
30378569bb4SMatthew G. Knepley 
3046823f3c5SBlaise Bourdin   This function will only handle TRI, TET, QUAD, and HEX cells.
30578569bb4SMatthew G. Knepley   Level: beginner
30678569bb4SMatthew G. Knepley 
3076823f3c5SBlaise Bourdin .seealso:
30878569bb4SMatthew G. Knepley */
309d71ae5a4SJacob Faibussowitsch PetscErrorCode DMView_PlexExodusII(DM dm, PetscViewer viewer)
310d71ae5a4SJacob Faibussowitsch {
3119371c9d4SSatish Balay   enum ElemType {
3129371c9d4SSatish Balay     TRI,
3139371c9d4SSatish Balay     QUAD,
3149371c9d4SSatish Balay     TET,
3159371c9d4SSatish Balay     HEX
3169371c9d4SSatish Balay   };
31778569bb4SMatthew G. Knepley   MPI_Comm comm;
3186823f3c5SBlaise Bourdin   PetscInt degree; /* the order of the mesh */
31978569bb4SMatthew G. Knepley   /* Connectivity Variables */
32078569bb4SMatthew G. Knepley   PetscInt cellsNotInConnectivity;
32178569bb4SMatthew G. Knepley   /* Cell Sets */
32278569bb4SMatthew G. Knepley   DMLabel         csLabel;
32378569bb4SMatthew G. Knepley   IS              csIS;
32478569bb4SMatthew G. Knepley   const PetscInt *csIdx;
32578569bb4SMatthew G. Knepley   PetscInt        num_cs, cs;
32678569bb4SMatthew G. Knepley   enum ElemType  *type;
327fe209ef9SBlaise Bourdin   PetscBool       hasLabel;
32878569bb4SMatthew G. Knepley   /* Coordinate Variables */
32978569bb4SMatthew G. Knepley   DM           cdm;
3306823f3c5SBlaise Bourdin   PetscSection coordSection;
33178569bb4SMatthew G. Knepley   Vec          coord;
33278569bb4SMatthew G. Knepley   PetscInt   **nodes;
333eae8a387SMatthew G. Knepley   PetscInt     depth, d, dim, skipCells = 0;
33478569bb4SMatthew G. Knepley   PetscInt     pStart, pEnd, p, cStart, cEnd, numCells, vStart, vEnd, numVertices, eStart, eEnd, numEdges, fStart, fEnd, numFaces, numNodes;
33578569bb4SMatthew G. Knepley   PetscInt     num_vs, num_fs;
33678569bb4SMatthew G. Knepley   PetscMPIInt  rank, size;
33778569bb4SMatthew G. Knepley   const char  *dmName;
338fe209ef9SBlaise Bourdin   PetscInt     nodesTriP1[4]  = {3, 0, 0, 0};
339fe209ef9SBlaise Bourdin   PetscInt     nodesTriP2[4]  = {3, 3, 0, 0};
340fe209ef9SBlaise Bourdin   PetscInt     nodesQuadP1[4] = {4, 0, 0, 0};
341fe209ef9SBlaise Bourdin   PetscInt     nodesQuadP2[4] = {4, 4, 0, 1};
342fe209ef9SBlaise Bourdin   PetscInt     nodesTetP1[4]  = {4, 0, 0, 0};
343fe209ef9SBlaise Bourdin   PetscInt     nodesTetP2[4]  = {4, 6, 0, 0};
344fe209ef9SBlaise Bourdin   PetscInt     nodesHexP1[4]  = {8, 0, 0, 0};
345fe209ef9SBlaise Bourdin   PetscInt     nodesHexP2[4]  = {8, 12, 6, 1};
3466823f3c5SBlaise Bourdin   int          CPU_word_size, IO_word_size, EXO_mode;
3476823f3c5SBlaise Bourdin   float        EXO_version;
3486823f3c5SBlaise Bourdin 
3496823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *)viewer->data;
3506823f3c5SBlaise Bourdin 
35178569bb4SMatthew G. Knepley   PetscFunctionBegin;
3529566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
3539566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
3549566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
3556823f3c5SBlaise Bourdin 
3566823f3c5SBlaise Bourdin   /*
3576823f3c5SBlaise Bourdin     Creating coordSection is a collective operation so we do it somewhat out of sequence
3586823f3c5SBlaise Bourdin   */
3599566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(comm, &coordSection));
3609566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocalSetUp(dm));
361c5853193SPierre Jolivet   if (rank == 0) {
3626823f3c5SBlaise Bourdin     switch (exo->btype) {
3636823f3c5SBlaise Bourdin     case FILE_MODE_READ:
3646823f3c5SBlaise Bourdin     case FILE_MODE_APPEND:
3656823f3c5SBlaise Bourdin     case FILE_MODE_UPDATE:
3666823f3c5SBlaise Bourdin     case FILE_MODE_APPEND_UPDATE:
3676823f3c5SBlaise Bourdin       /* exodusII does not allow writing geometry to an existing file */
36898921bdaSJacob Faibussowitsch       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "cannot add geometry to existing file %s", exo->filename);
3696823f3c5SBlaise Bourdin     case FILE_MODE_WRITE:
3706823f3c5SBlaise Bourdin       /* Create an empty file if one already exists*/
3716823f3c5SBlaise Bourdin       EXO_mode = EX_CLOBBER;
3726823f3c5SBlaise Bourdin   #if defined(PETSC_USE_64BIT_INDICES)
3736823f3c5SBlaise Bourdin       EXO_mode += EX_ALL_INT64_API;
3746823f3c5SBlaise Bourdin   #endif
3756823f3c5SBlaise Bourdin       CPU_word_size = sizeof(PetscReal);
3766823f3c5SBlaise Bourdin       IO_word_size  = sizeof(PetscReal);
3776823f3c5SBlaise Bourdin       exo->exoid    = ex_create(exo->filename, EXO_mode, &CPU_word_size, &IO_word_size);
37808401ef6SPierre Jolivet       PetscCheck(exo->exoid >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_create failed for %s", exo->filename);
3796823f3c5SBlaise Bourdin 
3806823f3c5SBlaise Bourdin       break;
381d71ae5a4SJacob Faibussowitsch     default:
382d71ae5a4SJacob Faibussowitsch       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
3836823f3c5SBlaise Bourdin     }
3846823f3c5SBlaise Bourdin 
38578569bb4SMatthew G. Knepley     /* --- Get DM info --- */
3869566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)dm, &dmName));
3879566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepth(dm, &depth));
3889566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &dim));
3899566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
3909566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
3919566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
3929566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, 1, &eStart, &eEnd));
3939566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
39478569bb4SMatthew G. Knepley     numCells    = cEnd - cStart;
39578569bb4SMatthew G. Knepley     numEdges    = eEnd - eStart;
39678569bb4SMatthew G. Knepley     numVertices = vEnd - vStart;
3979371c9d4SSatish Balay     if (depth == 3) {
3989371c9d4SSatish Balay       numFaces = fEnd - fStart;
3999371c9d4SSatish Balay     } else {
4009371c9d4SSatish Balay       numFaces = 0;
4019371c9d4SSatish Balay     }
4029566063dSJacob Faibussowitsch     PetscCall(DMGetLabelSize(dm, "Cell Sets", &num_cs));
4039566063dSJacob Faibussowitsch     PetscCall(DMGetLabelSize(dm, "Vertex Sets", &num_vs));
4049566063dSJacob Faibussowitsch     PetscCall(DMGetLabelSize(dm, "Face Sets", &num_fs));
4059566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &coord));
4069566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
40778569bb4SMatthew G. Knepley     if (num_cs > 0) {
4089566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm, "Cell Sets", &csLabel));
4099566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValueIS(csLabel, &csIS));
4109566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(csIS, &csIdx));
41178569bb4SMatthew G. Knepley     }
4129566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(num_cs, &nodes));
41378569bb4SMatthew G. Knepley     /* Set element type for each block and compute total number of nodes */
4149566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(num_cs, &type));
41578569bb4SMatthew G. Knepley     numNodes = numVertices;
4166823f3c5SBlaise Bourdin 
4179566063dSJacob Faibussowitsch     PetscCall(PetscViewerExodusIIGetOrder(viewer, &degree));
418ad540459SPierre Jolivet     if (degree == 2) numNodes += numEdges;
41978569bb4SMatthew G. Knepley     cellsNotInConnectivity = numCells;
42078569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
42178569bb4SMatthew G. Knepley       IS              stratumIS;
42278569bb4SMatthew G. Knepley       const PetscInt *cells;
42378569bb4SMatthew G. Knepley       PetscScalar    *xyz = NULL;
42478569bb4SMatthew G. Knepley       PetscInt        csSize, closureSize;
42578569bb4SMatthew G. Knepley 
4269566063dSJacob Faibussowitsch       PetscCall(DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS));
4279566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(stratumIS, &cells));
4289566063dSJacob Faibussowitsch       PetscCall(ISGetSize(stratumIS, &csSize));
4299566063dSJacob Faibussowitsch       PetscCall(DMPlexVecGetClosure(cdm, NULL, coord, cells[0], &closureSize, &xyz));
43078569bb4SMatthew G. Knepley       switch (dim) {
43178569bb4SMatthew G. Knepley       case 2:
4329371c9d4SSatish Balay         if (closureSize == 3 * dim) {
4339371c9d4SSatish Balay           type[cs] = TRI;
4349371c9d4SSatish Balay         } else if (closureSize == 4 * dim) {
4359371c9d4SSatish Balay           type[cs] = QUAD;
4369371c9d4SSatish Balay         } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of vertices %" PetscInt_FMT " in dimension %" PetscInt_FMT " has no ExodusII type", closureSize / dim, dim);
43778569bb4SMatthew G. Knepley         break;
43878569bb4SMatthew G. Knepley       case 3:
4399371c9d4SSatish Balay         if (closureSize == 4 * dim) {
4409371c9d4SSatish Balay           type[cs] = TET;
4419371c9d4SSatish Balay         } else if (closureSize == 8 * dim) {
4429371c9d4SSatish Balay           type[cs] = HEX;
4439371c9d4SSatish Balay         } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of vertices %" PetscInt_FMT " in dimension %" PetscInt_FMT " has no ExodusII type", closureSize / dim, dim);
44478569bb4SMatthew G. Knepley         break;
445d71ae5a4SJacob Faibussowitsch       default:
446d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Dimension %" PetscInt_FMT " not handled by ExodusII viewer", dim);
44778569bb4SMatthew G. Knepley       }
448ad540459SPierre Jolivet       if ((degree == 2) && (type[cs] == QUAD)) numNodes += csSize;
4499371c9d4SSatish Balay       if ((degree == 2) && (type[cs] == HEX)) {
4509371c9d4SSatish Balay         numNodes += csSize;
4519371c9d4SSatish Balay         numNodes += numFaces;
4529371c9d4SSatish Balay       }
4539566063dSJacob Faibussowitsch       PetscCall(DMPlexVecRestoreClosure(cdm, NULL, coord, cells[0], &closureSize, &xyz));
45478569bb4SMatthew G. Knepley       /* Set nodes and Element type */
45578569bb4SMatthew G. Knepley       if (type[cs] == TRI) {
45678569bb4SMatthew G. Knepley         if (degree == 1) nodes[cs] = nodesTriP1;
45778569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesTriP2;
45878569bb4SMatthew G. Knepley       } else if (type[cs] == QUAD) {
45978569bb4SMatthew G. Knepley         if (degree == 1) nodes[cs] = nodesQuadP1;
46078569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesQuadP2;
46178569bb4SMatthew G. Knepley       } else if (type[cs] == TET) {
46278569bb4SMatthew G. Knepley         if (degree == 1) nodes[cs] = nodesTetP1;
46378569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesTetP2;
46478569bb4SMatthew G. Knepley       } else if (type[cs] == HEX) {
46578569bb4SMatthew G. Knepley         if (degree == 1) nodes[cs] = nodesHexP1;
46678569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesHexP2;
46778569bb4SMatthew G. Knepley       }
46878569bb4SMatthew G. Knepley       /* Compute the number of cells not in the connectivity table */
46978569bb4SMatthew G. Knepley       cellsNotInConnectivity -= nodes[cs][3] * csSize;
47078569bb4SMatthew G. Knepley 
4719566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(stratumIS, &cells));
4729566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&stratumIS));
47378569bb4SMatthew G. Knepley     }
47448a46eb9SPierre Jolivet     if (num_cs > 0) PetscCallExternal(ex_put_init, exo->exoid, dmName, dim, numNodes, numCells, num_cs, num_vs, num_fs);
47578569bb4SMatthew G. Knepley     /* --- Connectivity --- */
47678569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
47778569bb4SMatthew G. Knepley       IS              stratumIS;
47878569bb4SMatthew G. Knepley       const PetscInt *cells;
479fe209ef9SBlaise Bourdin       PetscInt       *connect, off = 0;
480eae8a387SMatthew G. Knepley       PetscInt        edgesInClosure = 0, facesInClosure = 0, verticesInClosure = 0;
48178569bb4SMatthew G. Knepley       PetscInt        csSize, c, connectSize, closureSize;
48278569bb4SMatthew G. Knepley       char           *elem_type        = NULL;
48378569bb4SMatthew G. Knepley       char            elem_type_tri3[] = "TRI3", elem_type_quad4[] = "QUAD4";
48478569bb4SMatthew G. Knepley       char            elem_type_tri6[] = "TRI6", elem_type_quad9[] = "QUAD9";
48578569bb4SMatthew G. Knepley       char            elem_type_tet4[] = "TET4", elem_type_hex8[] = "HEX8";
48678569bb4SMatthew G. Knepley       char            elem_type_tet10[] = "TET10", elem_type_hex27[] = "HEX27";
48778569bb4SMatthew G. Knepley 
4889566063dSJacob Faibussowitsch       PetscCall(DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS));
4899566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(stratumIS, &cells));
4909566063dSJacob Faibussowitsch       PetscCall(ISGetSize(stratumIS, &csSize));
49178569bb4SMatthew G. Knepley       /* Set Element type */
49278569bb4SMatthew G. Knepley       if (type[cs] == TRI) {
49378569bb4SMatthew G. Knepley         if (degree == 1) elem_type = elem_type_tri3;
49478569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_tri6;
49578569bb4SMatthew G. Knepley       } else if (type[cs] == QUAD) {
49678569bb4SMatthew G. Knepley         if (degree == 1) elem_type = elem_type_quad4;
49778569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_quad9;
49878569bb4SMatthew G. Knepley       } else if (type[cs] == TET) {
49978569bb4SMatthew G. Knepley         if (degree == 1) elem_type = elem_type_tet4;
50078569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_tet10;
50178569bb4SMatthew G. Knepley       } else if (type[cs] == HEX) {
50278569bb4SMatthew G. Knepley         if (degree == 1) elem_type = elem_type_hex8;
50378569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_hex27;
50478569bb4SMatthew G. Knepley       }
50578569bb4SMatthew G. Knepley       connectSize = nodes[cs][0] + nodes[cs][1] + nodes[cs][2] + nodes[cs][3];
5069566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(PetscMax(27, connectSize) * csSize, &connect));
507792fecdfSBarry Smith       PetscCallExternal(ex_put_block, exo->exoid, EX_ELEM_BLOCK, csIdx[cs], elem_type, csSize, connectSize, 0, 0, 1);
50878569bb4SMatthew G. Knepley       /* Find number of vertices, edges, and faces in the closure */
50978569bb4SMatthew G. Knepley       verticesInClosure = nodes[cs][0];
51078569bb4SMatthew G. Knepley       if (depth > 1) {
51178569bb4SMatthew G. Knepley         if (dim == 2) {
5129566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm, cells[0], &edgesInClosure));
51378569bb4SMatthew G. Knepley         } else if (dim == 3) {
51478569bb4SMatthew G. Knepley           PetscInt *closure = NULL;
51578569bb4SMatthew G. Knepley 
5169566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm, cells[0], &facesInClosure));
5179566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTransitiveClosure(dm, cells[0], PETSC_TRUE, &closureSize, &closure));
51878569bb4SMatthew G. Knepley           edgesInClosure = closureSize - facesInClosure - 1 - verticesInClosure;
5199566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreTransitiveClosure(dm, cells[0], PETSC_TRUE, &closureSize, &closure));
52078569bb4SMatthew G. Knepley         }
52178569bb4SMatthew G. Knepley       }
52278569bb4SMatthew G. Knepley       /* Get connectivity for each cell */
52378569bb4SMatthew G. Knepley       for (c = 0; c < csSize; ++c) {
52478569bb4SMatthew G. Knepley         PetscInt *closure = NULL;
52578569bb4SMatthew G. Knepley         PetscInt  temp, i;
52678569bb4SMatthew G. Knepley 
5279566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(dm, cells[c], PETSC_TRUE, &closureSize, &closure));
52878569bb4SMatthew G. Knepley         for (i = 0; i < connectSize; ++i) {
52978569bb4SMatthew G. Knepley           if (i < nodes[cs][0]) { /* Vertices */
530fe209ef9SBlaise Bourdin             connect[i + off] = closure[(i + edgesInClosure + facesInClosure + 1) * 2] + 1;
531fe209ef9SBlaise Bourdin             connect[i + off] -= cellsNotInConnectivity;
53278569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0] + nodes[cs][1]) { /* Edges */
533fe209ef9SBlaise Bourdin             connect[i + off] = closure[(i - verticesInClosure + facesInClosure + 1) * 2] + 1;
534fe209ef9SBlaise Bourdin             if (nodes[cs][2] == 0) connect[i + off] -= numFaces;
535fe209ef9SBlaise Bourdin             connect[i + off] -= cellsNotInConnectivity;
53678569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0] + nodes[cs][1] + nodes[cs][3]) { /* Cells */
537fe209ef9SBlaise Bourdin             connect[i + off] = closure[0] + 1;
538fe209ef9SBlaise Bourdin             connect[i + off] -= skipCells;
53978569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0] + nodes[cs][1] + nodes[cs][3] + nodes[cs][2]) { /* Faces */
540fe209ef9SBlaise Bourdin             connect[i + off] = closure[(i - edgesInClosure - verticesInClosure) * 2] + 1;
541fe209ef9SBlaise Bourdin             connect[i + off] -= cellsNotInConnectivity;
54278569bb4SMatthew G. Knepley           } else {
543fe209ef9SBlaise Bourdin             connect[i + off] = -1;
54478569bb4SMatthew G. Knepley           }
54578569bb4SMatthew G. Knepley         }
54678569bb4SMatthew G. Knepley         /* Tetrahedra are inverted */
54778569bb4SMatthew G. Knepley         if (type[cs] == TET) {
5489371c9d4SSatish Balay           temp             = connect[0 + off];
5499371c9d4SSatish Balay           connect[0 + off] = connect[1 + off];
5509371c9d4SSatish Balay           connect[1 + off] = temp;
55178569bb4SMatthew G. Knepley           if (degree == 2) {
5529371c9d4SSatish Balay             temp             = connect[5 + off];
5539371c9d4SSatish Balay             connect[5 + off] = connect[6 + off];
5549371c9d4SSatish Balay             connect[6 + off] = temp;
5559371c9d4SSatish Balay             temp             = connect[7 + off];
5569371c9d4SSatish Balay             connect[7 + off] = connect[8 + off];
5579371c9d4SSatish Balay             connect[8 + off] = temp;
55878569bb4SMatthew G. Knepley           }
55978569bb4SMatthew G. Knepley         }
56078569bb4SMatthew G. Knepley         /* Hexahedra are inverted */
56178569bb4SMatthew G. Knepley         if (type[cs] == HEX) {
5629371c9d4SSatish Balay           temp             = connect[1 + off];
5639371c9d4SSatish Balay           connect[1 + off] = connect[3 + off];
5649371c9d4SSatish Balay           connect[3 + off] = temp;
56578569bb4SMatthew G. Knepley           if (degree == 2) {
5669371c9d4SSatish Balay             temp              = connect[8 + off];
5679371c9d4SSatish Balay             connect[8 + off]  = connect[11 + off];
5689371c9d4SSatish Balay             connect[11 + off] = temp;
5699371c9d4SSatish Balay             temp              = connect[9 + off];
5709371c9d4SSatish Balay             connect[9 + off]  = connect[10 + off];
5719371c9d4SSatish Balay             connect[10 + off] = temp;
5729371c9d4SSatish Balay             temp              = connect[16 + off];
5739371c9d4SSatish Balay             connect[16 + off] = connect[17 + off];
5749371c9d4SSatish Balay             connect[17 + off] = temp;
5759371c9d4SSatish Balay             temp              = connect[18 + off];
5769371c9d4SSatish Balay             connect[18 + off] = connect[19 + off];
5779371c9d4SSatish Balay             connect[19 + off] = temp;
57878569bb4SMatthew G. Knepley 
5799371c9d4SSatish Balay             temp              = connect[12 + off];
5809371c9d4SSatish Balay             connect[12 + off] = connect[16 + off];
5819371c9d4SSatish Balay             connect[16 + off] = temp;
5829371c9d4SSatish Balay             temp              = connect[13 + off];
5839371c9d4SSatish Balay             connect[13 + off] = connect[17 + off];
5849371c9d4SSatish Balay             connect[17 + off] = temp;
5859371c9d4SSatish Balay             temp              = connect[14 + off];
5869371c9d4SSatish Balay             connect[14 + off] = connect[18 + off];
5879371c9d4SSatish Balay             connect[18 + off] = temp;
5889371c9d4SSatish Balay             temp              = connect[15 + off];
5899371c9d4SSatish Balay             connect[15 + off] = connect[19 + off];
5909371c9d4SSatish Balay             connect[19 + off] = temp;
59178569bb4SMatthew G. Knepley 
5929371c9d4SSatish Balay             temp              = connect[23 + off];
5939371c9d4SSatish Balay             connect[23 + off] = connect[26 + off];
5949371c9d4SSatish Balay             connect[26 + off] = temp;
5959371c9d4SSatish Balay             temp              = connect[24 + off];
5969371c9d4SSatish Balay             connect[24 + off] = connect[25 + off];
5979371c9d4SSatish Balay             connect[25 + off] = temp;
5989371c9d4SSatish Balay             temp              = connect[25 + off];
5999371c9d4SSatish Balay             connect[25 + off] = connect[26 + off];
6009371c9d4SSatish Balay             connect[26 + off] = temp;
60178569bb4SMatthew G. Knepley           }
60278569bb4SMatthew G. Knepley         }
603fe209ef9SBlaise Bourdin         off += connectSize;
6049566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(dm, cells[c], PETSC_TRUE, &closureSize, &closure));
60578569bb4SMatthew G. Knepley       }
606792fecdfSBarry Smith       PetscCallExternal(ex_put_conn, exo->exoid, EX_ELEM_BLOCK, csIdx[cs], connect, 0, 0);
60778569bb4SMatthew G. Knepley       skipCells += (nodes[cs][3] == 0) * csSize;
6089566063dSJacob Faibussowitsch       PetscCall(PetscFree(connect));
6099566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(stratumIS, &cells));
6109566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&stratumIS));
61178569bb4SMatthew G. Knepley     }
6129566063dSJacob Faibussowitsch     PetscCall(PetscFree(type));
61378569bb4SMatthew G. Knepley     /* --- Coordinates --- */
6149566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, pStart, pEnd));
6151e50132fSMatthew G. Knepley     if (num_cs) {
61678569bb4SMatthew G. Knepley       for (d = 0; d < depth; ++d) {
6179566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthStratum(dm, d, &pStart, &pEnd));
61848a46eb9SPierre Jolivet         for (p = pStart; p < pEnd; ++p) PetscCall(PetscSectionSetDof(coordSection, p, nodes[0][d] > 0));
61978569bb4SMatthew G. Knepley       }
6201e50132fSMatthew G. Knepley     }
62178569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
62278569bb4SMatthew G. Knepley       IS              stratumIS;
62378569bb4SMatthew G. Knepley       const PetscInt *cells;
62478569bb4SMatthew G. Knepley       PetscInt        csSize, c;
62578569bb4SMatthew G. Knepley 
6269566063dSJacob Faibussowitsch       PetscCall(DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS));
6279566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(stratumIS, &cells));
6289566063dSJacob Faibussowitsch       PetscCall(ISGetSize(stratumIS, &csSize));
62948a46eb9SPierre Jolivet       for (c = 0; c < csSize; ++c) PetscCall(PetscSectionSetDof(coordSection, cells[c], nodes[cs][3] > 0));
6309566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(stratumIS, &cells));
6319566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&stratumIS));
63278569bb4SMatthew G. Knepley     }
63378569bb4SMatthew G. Knepley     if (num_cs > 0) {
6349566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(csIS, &csIdx));
6359566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&csIS));
63678569bb4SMatthew G. Knepley     }
6379566063dSJacob Faibussowitsch     PetscCall(PetscFree(nodes));
6389566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
63978569bb4SMatthew G. Knepley     if (numNodes > 0) {
64078569bb4SMatthew G. Knepley       const char  *coordNames[3] = {"x", "y", "z"};
641233c95e0SFrancesco Ballarin       PetscScalar *closure, *cval;
642233c95e0SFrancesco Ballarin       PetscReal   *coords;
64378569bb4SMatthew G. Knepley       PetscInt     hasDof, n = 0;
64478569bb4SMatthew G. Knepley 
6456823f3c5SBlaise Bourdin       /* There can't be more than 24 values in the closure of a point for the coord coordSection */
6469566063dSJacob Faibussowitsch       PetscCall(PetscCalloc3(numNodes * 3, &coords, dim, &cval, 24, &closure));
6479566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocalNoncollective(dm, &coord));
6489566063dSJacob Faibussowitsch       PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
64978569bb4SMatthew G. Knepley       for (p = pStart; p < pEnd; ++p) {
6509566063dSJacob Faibussowitsch         PetscCall(PetscSectionGetDof(coordSection, p, &hasDof));
65178569bb4SMatthew G. Knepley         if (hasDof) {
65278569bb4SMatthew G. Knepley           PetscInt closureSize = 24, j;
65378569bb4SMatthew G. Knepley 
6549566063dSJacob Faibussowitsch           PetscCall(DMPlexVecGetClosure(cdm, NULL, coord, p, &closureSize, &closure));
65578569bb4SMatthew G. Knepley           for (d = 0; d < dim; ++d) {
65678569bb4SMatthew G. Knepley             cval[d] = 0.0;
65778569bb4SMatthew G. Knepley             for (j = 0; j < closureSize / dim; j++) cval[d] += closure[j * dim + d];
658233c95e0SFrancesco Ballarin             coords[d * numNodes + n] = PetscRealPart(cval[d]) * dim / closureSize;
65978569bb4SMatthew G. Knepley           }
66078569bb4SMatthew G. Knepley           ++n;
66178569bb4SMatthew G. Knepley         }
66278569bb4SMatthew G. Knepley       }
663792fecdfSBarry Smith       PetscCallExternal(ex_put_coord, exo->exoid, &coords[0 * numNodes], &coords[1 * numNodes], &coords[2 * numNodes]);
6649566063dSJacob Faibussowitsch       PetscCall(PetscFree3(coords, cval, closure));
665792fecdfSBarry Smith       PetscCallExternal(ex_put_coord_names, exo->exoid, (char **)coordNames);
66678569bb4SMatthew G. Knepley     }
6676823f3c5SBlaise Bourdin 
668fe209ef9SBlaise Bourdin     /* --- Node Sets/Vertex Sets --- */
669fe209ef9SBlaise Bourdin     DMHasLabel(dm, "Vertex Sets", &hasLabel);
670fe209ef9SBlaise Bourdin     if (hasLabel) {
671fe209ef9SBlaise Bourdin       PetscInt        i, vs, vsSize;
672fe209ef9SBlaise Bourdin       const PetscInt *vsIdx, *vertices;
673fe209ef9SBlaise Bourdin       PetscInt       *nodeList;
674fe209ef9SBlaise Bourdin       IS              vsIS, stratumIS;
675fe209ef9SBlaise Bourdin       DMLabel         vsLabel;
6769566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm, "Vertex Sets", &vsLabel));
6779566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValueIS(vsLabel, &vsIS));
6789566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(vsIS, &vsIdx));
679fe209ef9SBlaise Bourdin       for (vs = 0; vs < num_vs; ++vs) {
6809566063dSJacob Faibussowitsch         PetscCall(DMLabelGetStratumIS(vsLabel, vsIdx[vs], &stratumIS));
6819566063dSJacob Faibussowitsch         PetscCall(ISGetIndices(stratumIS, &vertices));
6829566063dSJacob Faibussowitsch         PetscCall(ISGetSize(stratumIS, &vsSize));
6839566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(vsSize, &nodeList));
684ad540459SPierre Jolivet         for (i = 0; i < vsSize; ++i) nodeList[i] = vertices[i] - skipCells + 1;
685792fecdfSBarry Smith         PetscCallExternal(ex_put_set_param, exo->exoid, EX_NODE_SET, vsIdx[vs], vsSize, 0);
686792fecdfSBarry Smith         PetscCallExternal(ex_put_set, exo->exoid, EX_NODE_SET, vsIdx[vs], nodeList, NULL);
6879566063dSJacob Faibussowitsch         PetscCall(ISRestoreIndices(stratumIS, &vertices));
6889566063dSJacob Faibussowitsch         PetscCall(ISDestroy(&stratumIS));
6899566063dSJacob Faibussowitsch         PetscCall(PetscFree(nodeList));
690fe209ef9SBlaise Bourdin       }
6919566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(vsIS, &vsIdx));
6929566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&vsIS));
693fe209ef9SBlaise Bourdin     }
694fe209ef9SBlaise Bourdin     /* --- Side Sets/Face Sets --- */
6959566063dSJacob Faibussowitsch     PetscCall(DMHasLabel(dm, "Face Sets", &hasLabel));
696fe209ef9SBlaise Bourdin     if (hasLabel) {
697fe209ef9SBlaise Bourdin       PetscInt        i, j, fs, fsSize;
698fe209ef9SBlaise Bourdin       const PetscInt *fsIdx, *faces;
699fe209ef9SBlaise Bourdin       IS              fsIS, stratumIS;
700fe209ef9SBlaise Bourdin       DMLabel         fsLabel;
701fe209ef9SBlaise Bourdin       PetscInt        numPoints, *points;
702fe209ef9SBlaise Bourdin       PetscInt        elem_list_size = 0;
703fe209ef9SBlaise Bourdin       PetscInt       *elem_list, *elem_ind, *side_list;
704fe209ef9SBlaise Bourdin 
7059566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm, "Face Sets", &fsLabel));
706fe209ef9SBlaise Bourdin       /* Compute size of Node List and Element List */
7079566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValueIS(fsLabel, &fsIS));
7089566063dSJacob Faibussowitsch       PetscCall(ISGetIndices(fsIS, &fsIdx));
709fe209ef9SBlaise Bourdin       for (fs = 0; fs < num_fs; ++fs) {
7109566063dSJacob Faibussowitsch         PetscCall(DMLabelGetStratumIS(fsLabel, fsIdx[fs], &stratumIS));
7119566063dSJacob Faibussowitsch         PetscCall(ISGetSize(stratumIS, &fsSize));
712fe209ef9SBlaise Bourdin         elem_list_size += fsSize;
7139566063dSJacob Faibussowitsch         PetscCall(ISDestroy(&stratumIS));
714fe209ef9SBlaise Bourdin       }
715d0609cedSBarry Smith       PetscCall(PetscMalloc3(num_fs, &elem_ind, elem_list_size, &elem_list, elem_list_size, &side_list));
716fe209ef9SBlaise Bourdin       elem_ind[0] = 0;
717fe209ef9SBlaise Bourdin       for (fs = 0; fs < num_fs; ++fs) {
7189566063dSJacob Faibussowitsch         PetscCall(DMLabelGetStratumIS(fsLabel, fsIdx[fs], &stratumIS));
7199566063dSJacob Faibussowitsch         PetscCall(ISGetIndices(stratumIS, &faces));
7209566063dSJacob Faibussowitsch         PetscCall(ISGetSize(stratumIS, &fsSize));
721fe209ef9SBlaise Bourdin         /* Set Parameters */
722792fecdfSBarry Smith         PetscCallExternal(ex_put_set_param, exo->exoid, EX_SIDE_SET, fsIdx[fs], fsSize, 0);
723fe209ef9SBlaise Bourdin         /* Indices */
724ad540459SPierre Jolivet         if (fs < num_fs - 1) elem_ind[fs + 1] = elem_ind[fs] + fsSize;
725fe209ef9SBlaise Bourdin 
726fe209ef9SBlaise Bourdin         for (i = 0; i < fsSize; ++i) {
727fe209ef9SBlaise Bourdin           /* Element List */
728fe209ef9SBlaise Bourdin           points = NULL;
7299566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTransitiveClosure(dm, faces[i], PETSC_FALSE, &numPoints, &points));
730fe209ef9SBlaise Bourdin           elem_list[elem_ind[fs] + i] = points[2] + 1;
7319566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreTransitiveClosure(dm, faces[i], PETSC_FALSE, &numPoints, &points));
732fe209ef9SBlaise Bourdin 
733fe209ef9SBlaise Bourdin           /* Side List */
734fe209ef9SBlaise Bourdin           points = NULL;
7359566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTransitiveClosure(dm, elem_list[elem_ind[fs] + i] - 1, PETSC_TRUE, &numPoints, &points));
736fe209ef9SBlaise Bourdin           for (j = 1; j < numPoints; ++j) {
737ad540459SPierre Jolivet             if (points[j * 2] == faces[i]) break;
738fe209ef9SBlaise Bourdin           }
739fe209ef9SBlaise Bourdin           /* Convert HEX sides */
740fe209ef9SBlaise Bourdin           if (numPoints == 27) {
7419371c9d4SSatish Balay             if (j == 1) {
7429371c9d4SSatish Balay               j = 5;
7439371c9d4SSatish Balay             } else if (j == 2) {
7449371c9d4SSatish Balay               j = 6;
7459371c9d4SSatish Balay             } else if (j == 3) {
7469371c9d4SSatish Balay               j = 1;
7479371c9d4SSatish Balay             } else if (j == 4) {
7489371c9d4SSatish Balay               j = 3;
7499371c9d4SSatish Balay             } else if (j == 5) {
7509371c9d4SSatish Balay               j = 2;
7519371c9d4SSatish Balay             } else if (j == 6) {
7529371c9d4SSatish Balay               j = 4;
7539371c9d4SSatish Balay             }
754fe209ef9SBlaise Bourdin           }
755fe209ef9SBlaise Bourdin           /* Convert TET sides */
756fe209ef9SBlaise Bourdin           if (numPoints == 15) {
757fe209ef9SBlaise Bourdin             --j;
758ad540459SPierre Jolivet             if (j == 0) j = 4;
759fe209ef9SBlaise Bourdin           }
760fe209ef9SBlaise Bourdin           side_list[elem_ind[fs] + i] = j;
7619566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreTransitiveClosure(dm, elem_list[elem_ind[fs] + i] - 1, PETSC_TRUE, &numPoints, &points));
762fe209ef9SBlaise Bourdin         }
7639566063dSJacob Faibussowitsch         PetscCall(ISRestoreIndices(stratumIS, &faces));
7649566063dSJacob Faibussowitsch         PetscCall(ISDestroy(&stratumIS));
765fe209ef9SBlaise Bourdin       }
7669566063dSJacob Faibussowitsch       PetscCall(ISRestoreIndices(fsIS, &fsIdx));
7679566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&fsIS));
768fe209ef9SBlaise Bourdin 
769fe209ef9SBlaise Bourdin       /* Put side sets */
77048a46eb9SPierre Jolivet       for (fs = 0; fs < num_fs; ++fs) PetscCallExternal(ex_put_set, exo->exoid, EX_SIDE_SET, fsIdx[fs], &elem_list[elem_ind[fs]], &side_list[elem_ind[fs]]);
7719566063dSJacob Faibussowitsch       PetscCall(PetscFree3(elem_ind, elem_list, side_list));
772fe209ef9SBlaise Bourdin     }
7736823f3c5SBlaise Bourdin     /*
7746823f3c5SBlaise Bourdin       close the exodus file
7756823f3c5SBlaise Bourdin     */
7766823f3c5SBlaise Bourdin     ex_close(exo->exoid);
7776823f3c5SBlaise Bourdin     exo->exoid = -1;
7786823f3c5SBlaise Bourdin   }
7799566063dSJacob Faibussowitsch   PetscCall(PetscSectionDestroy(&coordSection));
7806823f3c5SBlaise Bourdin 
7816823f3c5SBlaise Bourdin   /*
7826823f3c5SBlaise Bourdin     reopen the file in parallel
7836823f3c5SBlaise Bourdin   */
7846823f3c5SBlaise Bourdin   EXO_mode = EX_WRITE;
7856823f3c5SBlaise Bourdin   #if defined(PETSC_USE_64BIT_INDICES)
7866823f3c5SBlaise Bourdin   EXO_mode += EX_ALL_INT64_API;
7876823f3c5SBlaise Bourdin   #endif
7886823f3c5SBlaise Bourdin   CPU_word_size = sizeof(PetscReal);
7896823f3c5SBlaise Bourdin   IO_word_size  = sizeof(PetscReal);
7906823f3c5SBlaise Bourdin   exo->exoid    = ex_open_par(exo->filename, EXO_mode, &CPU_word_size, &IO_word_size, &EXO_version, comm, MPI_INFO_NULL);
79108401ef6SPierre Jolivet   PetscCheck(exo->exoid >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open_par failed for %s", exo->filename);
7926823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
7936823f3c5SBlaise Bourdin }
7946823f3c5SBlaise Bourdin 
7956823f3c5SBlaise Bourdin /*
7966823f3c5SBlaise Bourdin   VecView_PlexExodusII_Internal - Write a Vec corresponding in an exodus file
7976823f3c5SBlaise Bourdin 
7986823f3c5SBlaise Bourdin   Collective on v
7996823f3c5SBlaise Bourdin 
8006823f3c5SBlaise Bourdin   Input Parameters:
8016823f3c5SBlaise Bourdin + v  - The vector to be written
8026823f3c5SBlaise Bourdin - viewer - The PetscViewerExodusII viewer associate to an exodus file
8036823f3c5SBlaise Bourdin 
8046823f3c5SBlaise Bourdin   Notes:
8056823f3c5SBlaise Bourdin   The exodus result variable index is obtained by comparing the Vec name and the
8066823f3c5SBlaise Bourdin   names of variables declared in the exodus file. For instance for a Vec named "V"
8076823f3c5SBlaise Bourdin   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
8086823f3c5SBlaise Bourdin   amongst all variables.
8096823f3c5SBlaise Bourdin   In the event where a nodal and zonal variable both match, the function will return an error instead of
8106823f3c5SBlaise Bourdin   possibly corrupting the file
8116823f3c5SBlaise Bourdin 
8126823f3c5SBlaise Bourdin   Level: beginner
8136823f3c5SBlaise Bourdin 
814c2e3fba1SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII()`, `VecView_PlexExodusII()`
8156823f3c5SBlaise Bourdin @*/
816d71ae5a4SJacob Faibussowitsch PetscErrorCode VecView_PlexExodusII_Internal(Vec v, PetscViewer viewer)
817d71ae5a4SJacob Faibussowitsch {
8186823f3c5SBlaise Bourdin   DM          dm;
8196823f3c5SBlaise Bourdin   MPI_Comm    comm;
8206823f3c5SBlaise Bourdin   PetscMPIInt rank;
8216823f3c5SBlaise Bourdin 
8226823f3c5SBlaise Bourdin   int         exoid, offsetN = 0, offsetZ = 0;
8236823f3c5SBlaise Bourdin   const char *vecname;
8246823f3c5SBlaise Bourdin   PetscInt    step;
8256823f3c5SBlaise Bourdin 
8266823f3c5SBlaise Bourdin   PetscFunctionBegin;
8279566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
8289566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
8299566063dSJacob Faibussowitsch   PetscCall(PetscViewerExodusIIGetId(viewer, &exoid));
8309566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
8319566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetName((PetscObject)v, &vecname));
8326823f3c5SBlaise Bourdin 
8339566063dSJacob Faibussowitsch   PetscCall(DMGetOutputSequenceNumber(dm, &step, NULL));
8349566063dSJacob Faibussowitsch   PetscCall(EXOGetVarIndex_Internal(exoid, EX_NODAL, vecname, &offsetN));
8359566063dSJacob Faibussowitsch   PetscCall(EXOGetVarIndex_Internal(exoid, EX_ELEM_BLOCK, vecname, &offsetZ));
8361dca8a05SBarry Smith   PetscCheck(offsetN > 0 || offsetZ > 0, comm, PETSC_ERR_FILE_UNEXPECTED, "Found both nodal and zonal variable %s in exodus file. ", vecname);
8376823f3c5SBlaise Bourdin   if (offsetN > 0) {
8389566063dSJacob Faibussowitsch     PetscCall(VecViewPlex_ExodusII_Nodal_Internal(v, exoid, (int)step + 1, offsetN));
8396823f3c5SBlaise Bourdin   } else if (offsetZ > 0) {
8409566063dSJacob Faibussowitsch     PetscCall(VecViewPlex_ExodusII_Zonal_Internal(v, exoid, (int)step + 1, offsetZ));
84198921bdaSJacob Faibussowitsch   } else SETERRQ(comm, PETSC_ERR_FILE_UNEXPECTED, "Could not find nodal or zonal variable %s in exodus file. ", vecname);
8426823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
8436823f3c5SBlaise Bourdin }
8446823f3c5SBlaise Bourdin 
8456823f3c5SBlaise Bourdin /*
8466823f3c5SBlaise Bourdin   VecLoad_PlexExodusII_Internal - Write a Vec corresponding in an exodus file
8476823f3c5SBlaise Bourdin 
8486823f3c5SBlaise Bourdin   Collective on v
8496823f3c5SBlaise Bourdin 
8506823f3c5SBlaise Bourdin   Input Parameters:
8516823f3c5SBlaise Bourdin + v  - The vector to be written
8526823f3c5SBlaise Bourdin - viewer - The PetscViewerExodusII viewer associate to an exodus file
8536823f3c5SBlaise Bourdin 
8546823f3c5SBlaise Bourdin   Notes:
8556823f3c5SBlaise Bourdin   The exodus result variable index is obtained by comparing the Vec name and the
8566823f3c5SBlaise Bourdin   names of variables declared in the exodus file. For instance for a Vec named "V"
8576823f3c5SBlaise Bourdin   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
8586823f3c5SBlaise Bourdin   amongst all variables.
8596823f3c5SBlaise Bourdin   In the event where a nodal and zonal variable both match, the function will return an error instead of
8606823f3c5SBlaise Bourdin   possibly corrupting the file
8616823f3c5SBlaise Bourdin 
8626823f3c5SBlaise Bourdin   Level: beginner
8636823f3c5SBlaise Bourdin 
864c2e3fba1SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII()`, `VecView_PlexExodusII()`
8656823f3c5SBlaise Bourdin @*/
866d71ae5a4SJacob Faibussowitsch PetscErrorCode VecLoad_PlexExodusII_Internal(Vec v, PetscViewer viewer)
867d71ae5a4SJacob Faibussowitsch {
8686823f3c5SBlaise Bourdin   DM          dm;
8696823f3c5SBlaise Bourdin   MPI_Comm    comm;
8706823f3c5SBlaise Bourdin   PetscMPIInt rank;
8716823f3c5SBlaise Bourdin 
8726823f3c5SBlaise Bourdin   int         exoid, offsetN = 0, offsetZ = 0;
8736823f3c5SBlaise Bourdin   const char *vecname;
8746823f3c5SBlaise Bourdin   PetscInt    step;
8756823f3c5SBlaise Bourdin 
8766823f3c5SBlaise Bourdin   PetscFunctionBegin;
8779566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
8789566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
8799566063dSJacob Faibussowitsch   PetscCall(PetscViewerExodusIIGetId(viewer, &exoid));
8809566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
8819566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetName((PetscObject)v, &vecname));
8826823f3c5SBlaise Bourdin 
8839566063dSJacob Faibussowitsch   PetscCall(DMGetOutputSequenceNumber(dm, &step, NULL));
8849566063dSJacob Faibussowitsch   PetscCall(EXOGetVarIndex_Internal(exoid, EX_NODAL, vecname, &offsetN));
8859566063dSJacob Faibussowitsch   PetscCall(EXOGetVarIndex_Internal(exoid, EX_ELEM_BLOCK, vecname, &offsetZ));
8861dca8a05SBarry Smith   PetscCheck(offsetN > 0 || offsetZ > 0, comm, PETSC_ERR_FILE_UNEXPECTED, "Found both nodal and zonal variable %s in exodus file. ", vecname);
8871dca8a05SBarry Smith   if (offsetN > 0) PetscCall(VecLoadPlex_ExodusII_Nodal_Internal(v, exoid, (int)step + 1, offsetN));
8881dca8a05SBarry Smith   else if (offsetZ > 0) PetscCall(VecLoadPlex_ExodusII_Zonal_Internal(v, exoid, (int)step + 1, offsetZ));
8891dca8a05SBarry Smith   else SETERRQ(comm, PETSC_ERR_FILE_UNEXPECTED, "Could not find nodal or zonal variable %s in exodus file. ", vecname);
89078569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
89178569bb4SMatthew G. Knepley }
89278569bb4SMatthew G. Knepley 
89378569bb4SMatthew G. Knepley /*
89478569bb4SMatthew G. Knepley   VecViewPlex_ExodusII_Nodal_Internal - Write a Vec corresponding to a nodal field to an exodus file
89578569bb4SMatthew G. Knepley 
89678569bb4SMatthew G. Knepley   Collective on v
89778569bb4SMatthew G. Knepley 
89878569bb4SMatthew G. Knepley   Input Parameters:
89978569bb4SMatthew G. Knepley + v  - The vector to be written
90078569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
9016823f3c5SBlaise Bourdin . step - the time step to write at (exodus steps are numbered starting from 1)
9026823f3c5SBlaise Bourdin - offset - the location of the variable in the file
90378569bb4SMatthew G. Knepley 
90478569bb4SMatthew G. Knepley   Notes:
90578569bb4SMatthew G. Knepley   The exodus result nodal variable index is obtained by comparing the Vec name and the
90678569bb4SMatthew G. Knepley   names of nodal variables declared in the exodus file. For instance for a Vec named "V"
90778569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
90878569bb4SMatthew G. Knepley   amongst all nodal variables.
90978569bb4SMatthew G. Knepley 
91078569bb4SMatthew G. Knepley   Level: beginner
91178569bb4SMatthew G. Knepley 
912c2e3fba1SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII_Internal()`, `VecLoadNodal_PlexEXO()`, `VecViewZonal_PlexEXO()`, `VecLoadZonal_PlexEXO()`
91378569bb4SMatthew G. Knepley @*/
914d71ae5a4SJacob Faibussowitsch PetscErrorCode VecViewPlex_ExodusII_Nodal_Internal(Vec v, int exoid, int step, int offset)
915d71ae5a4SJacob Faibussowitsch {
91678569bb4SMatthew G. Knepley   MPI_Comm           comm;
91778569bb4SMatthew G. Knepley   PetscMPIInt        size;
91878569bb4SMatthew G. Knepley   DM                 dm;
91978569bb4SMatthew G. Knepley   Vec                vNatural, vComp;
92022a7544eSVaclav Hapla   const PetscScalar *varray;
92178569bb4SMatthew G. Knepley   PetscInt           xs, xe, bs;
92278569bb4SMatthew G. Knepley   PetscBool          useNatural;
92378569bb4SMatthew G. Knepley 
92478569bb4SMatthew G. Knepley   PetscFunctionBegin;
9259566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
9269566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
9279566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
9289566063dSJacob Faibussowitsch   PetscCall(DMGetUseNatural(dm, &useNatural));
92978569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
93078569bb4SMatthew G. Knepley   if (useNatural) {
931b7352c5cSAlexis Marboeuf     PetscCall(DMPlexCreateNaturalVector(dm, &vNatural));
9329566063dSJacob Faibussowitsch     PetscCall(DMPlexGlobalToNaturalBegin(dm, v, vNatural));
9339566063dSJacob Faibussowitsch     PetscCall(DMPlexGlobalToNaturalEnd(dm, v, vNatural));
93478569bb4SMatthew G. Knepley   } else {
93578569bb4SMatthew G. Knepley     vNatural = v;
93678569bb4SMatthew G. Knepley   }
9376823f3c5SBlaise Bourdin 
93878569bb4SMatthew G. Knepley   /* Write local chunk of the result in the exodus file
93978569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
94078569bb4SMatthew G. Knepley      We assume that they are stored sequentially */
9419566063dSJacob Faibussowitsch   PetscCall(VecGetOwnershipRange(vNatural, &xs, &xe));
9429566063dSJacob Faibussowitsch   PetscCall(VecGetBlockSize(vNatural, &bs));
94378569bb4SMatthew G. Knepley   if (bs == 1) {
9449566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(vNatural, &varray));
945792fecdfSBarry Smith     PetscCallExternal(ex_put_partial_var, exoid, step, EX_NODAL, offset, 1, xs + 1, xe - xs, varray);
9469566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(vNatural, &varray));
94778569bb4SMatthew G. Knepley   } else {
94878569bb4SMatthew G. Knepley     IS       compIS;
94978569bb4SMatthew G. Knepley     PetscInt c;
95078569bb4SMatthew G. Knepley 
9519566063dSJacob Faibussowitsch     PetscCall(ISCreateStride(comm, (xe - xs) / bs, xs, bs, &compIS));
95278569bb4SMatthew G. Knepley     for (c = 0; c < bs; ++c) {
9539566063dSJacob Faibussowitsch       PetscCall(ISStrideSetStride(compIS, (xe - xs) / bs, xs + c, bs));
9549566063dSJacob Faibussowitsch       PetscCall(VecGetSubVector(vNatural, compIS, &vComp));
9559566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(vComp, &varray));
956792fecdfSBarry Smith       PetscCallExternal(ex_put_partial_var, exoid, step, EX_NODAL, offset + c, 1, xs / bs + 1, (xe - xs) / bs, varray);
9579566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(vComp, &varray));
9589566063dSJacob Faibussowitsch       PetscCall(VecRestoreSubVector(vNatural, compIS, &vComp));
95978569bb4SMatthew G. Knepley     }
9609566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&compIS));
96178569bb4SMatthew G. Knepley   }
962b7352c5cSAlexis Marboeuf   if (useNatural) PetscCall(VecDestroy(&vNatural));
96378569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
96478569bb4SMatthew G. Knepley }
96578569bb4SMatthew G. Knepley 
96678569bb4SMatthew G. Knepley /*
96778569bb4SMatthew G. Knepley   VecLoadPlex_ExodusII_Nodal_Internal - Read a Vec corresponding to a nodal field from an exodus file
96878569bb4SMatthew G. Knepley 
96978569bb4SMatthew G. Knepley   Collective on v
97078569bb4SMatthew G. Knepley 
97178569bb4SMatthew G. Knepley   Input Parameters:
97278569bb4SMatthew G. Knepley + v  - The vector to be written
97378569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
9746823f3c5SBlaise Bourdin . step - the time step to read at (exodus steps are numbered starting from 1)
9756823f3c5SBlaise Bourdin - offset - the location of the variable in the file
97678569bb4SMatthew G. Knepley 
97778569bb4SMatthew G. Knepley   Notes:
97878569bb4SMatthew G. Knepley   The exodus result nodal variable index is obtained by comparing the Vec name and the
97978569bb4SMatthew G. Knepley   names of nodal variables declared in the exodus file. For instance for a Vec named "V"
98078569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
98178569bb4SMatthew G. Knepley   amongst all nodal variables.
98278569bb4SMatthew G. Knepley 
98378569bb4SMatthew G. Knepley   Level: beginner
98478569bb4SMatthew G. Knepley 
985db781477SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII_Internal()`, `VecViewPlex_ExodusII_Nodal_Internal()`, `VecViewZonal_PlexEXO()`, `VecLoadZonal_PlexEXO()`
98678569bb4SMatthew G. Knepley */
987d71ae5a4SJacob Faibussowitsch PetscErrorCode VecLoadPlex_ExodusII_Nodal_Internal(Vec v, int exoid, int step, int offset)
988d71ae5a4SJacob Faibussowitsch {
98978569bb4SMatthew G. Knepley   MPI_Comm     comm;
99078569bb4SMatthew G. Knepley   PetscMPIInt  size;
99178569bb4SMatthew G. Knepley   DM           dm;
99278569bb4SMatthew G. Knepley   Vec          vNatural, vComp;
99322a7544eSVaclav Hapla   PetscScalar *varray;
99478569bb4SMatthew G. Knepley   PetscInt     xs, xe, bs;
99578569bb4SMatthew G. Knepley   PetscBool    useNatural;
99678569bb4SMatthew G. Knepley 
99778569bb4SMatthew G. Knepley   PetscFunctionBegin;
9989566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
9999566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
10009566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
10019566063dSJacob Faibussowitsch   PetscCall(DMGetUseNatural(dm, &useNatural));
100278569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
1003b7352c5cSAlexis Marboeuf   if (useNatural) PetscCall(DMPlexCreateNaturalVector(dm, &vNatural));
1004ad540459SPierre Jolivet   else vNatural = v;
10056823f3c5SBlaise Bourdin 
100678569bb4SMatthew G. Knepley   /* Read local chunk from the file */
10079566063dSJacob Faibussowitsch   PetscCall(VecGetOwnershipRange(vNatural, &xs, &xe));
10089566063dSJacob Faibussowitsch   PetscCall(VecGetBlockSize(vNatural, &bs));
100978569bb4SMatthew G. Knepley   if (bs == 1) {
10109566063dSJacob Faibussowitsch     PetscCall(VecGetArray(vNatural, &varray));
1011792fecdfSBarry Smith     PetscCallExternal(ex_get_partial_var, exoid, step, EX_NODAL, offset, 1, xs + 1, xe - xs, varray);
10129566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(vNatural, &varray));
101378569bb4SMatthew G. Knepley   } else {
101478569bb4SMatthew G. Knepley     IS       compIS;
101578569bb4SMatthew G. Knepley     PetscInt c;
101678569bb4SMatthew G. Knepley 
10179566063dSJacob Faibussowitsch     PetscCall(ISCreateStride(comm, (xe - xs) / bs, xs, bs, &compIS));
101878569bb4SMatthew G. Knepley     for (c = 0; c < bs; ++c) {
10199566063dSJacob Faibussowitsch       PetscCall(ISStrideSetStride(compIS, (xe - xs) / bs, xs + c, bs));
10209566063dSJacob Faibussowitsch       PetscCall(VecGetSubVector(vNatural, compIS, &vComp));
10219566063dSJacob Faibussowitsch       PetscCall(VecGetArray(vComp, &varray));
1022792fecdfSBarry Smith       PetscCallExternal(ex_get_partial_var, exoid, step, EX_NODAL, offset + c, 1, xs / bs + 1, (xe - xs) / bs, varray);
10239566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(vComp, &varray));
10249566063dSJacob Faibussowitsch       PetscCall(VecRestoreSubVector(vNatural, compIS, &vComp));
102578569bb4SMatthew G. Knepley     }
10269566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&compIS));
102778569bb4SMatthew G. Knepley   }
102878569bb4SMatthew G. Knepley   if (useNatural) {
10299566063dSJacob Faibussowitsch     PetscCall(DMPlexNaturalToGlobalBegin(dm, vNatural, v));
10309566063dSJacob Faibussowitsch     PetscCall(DMPlexNaturalToGlobalEnd(dm, vNatural, v));
1031b7352c5cSAlexis Marboeuf     PetscCall(VecDestroy(&vNatural));
103278569bb4SMatthew G. Knepley   }
103378569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
103478569bb4SMatthew G. Knepley }
103578569bb4SMatthew G. Knepley 
103678569bb4SMatthew G. Knepley /*
103778569bb4SMatthew G. Knepley   VecViewPlex_ExodusII_Zonal_Internal - Write a Vec corresponding to a zonal (cell based) field to an exodus file
103878569bb4SMatthew G. Knepley 
103978569bb4SMatthew G. Knepley   Collective on v
104078569bb4SMatthew G. Knepley 
104178569bb4SMatthew G. Knepley   Input Parameters:
104278569bb4SMatthew G. Knepley + v  - The vector to be written
104378569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
10446823f3c5SBlaise Bourdin . step - the time step to write at (exodus steps are numbered starting from 1)
10456823f3c5SBlaise Bourdin - offset - the location of the variable in the file
104678569bb4SMatthew G. Knepley 
104778569bb4SMatthew G. Knepley   Notes:
104878569bb4SMatthew G. Knepley   The exodus result zonal variable index is obtained by comparing the Vec name and the
104978569bb4SMatthew G. Knepley   names of zonal variables declared in the exodus file. For instance for a Vec named "V"
105078569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
105178569bb4SMatthew G. Knepley   amongst all zonal variables.
105278569bb4SMatthew G. Knepley 
105378569bb4SMatthew G. Knepley   Level: beginner
105478569bb4SMatthew G. Knepley 
1055c2e3fba1SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII_Internal()`, `VecViewPlex_ExodusII_Nodal_Internal()`, `VecLoadPlex_ExodusII_Nodal_Internal()`, `VecLoadPlex_ExodusII_Zonal_Internal()`
105678569bb4SMatthew G. Knepley */
1057d71ae5a4SJacob Faibussowitsch PetscErrorCode VecViewPlex_ExodusII_Zonal_Internal(Vec v, int exoid, int step, int offset)
1058d71ae5a4SJacob Faibussowitsch {
105978569bb4SMatthew G. Knepley   MPI_Comm           comm;
106078569bb4SMatthew G. Knepley   PetscMPIInt        size;
106178569bb4SMatthew G. Knepley   DM                 dm;
106278569bb4SMatthew G. Knepley   Vec                vNatural, vComp;
106322a7544eSVaclav Hapla   const PetscScalar *varray;
106478569bb4SMatthew G. Knepley   PetscInt           xs, xe, bs;
106578569bb4SMatthew G. Knepley   PetscBool          useNatural;
106678569bb4SMatthew G. Knepley   IS                 compIS;
106778569bb4SMatthew G. Knepley   PetscInt          *csSize, *csID;
106878569bb4SMatthew G. Knepley   PetscInt           numCS, set, csxs = 0;
106978569bb4SMatthew G. Knepley 
107078569bb4SMatthew G. Knepley   PetscFunctionBegin;
10719566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
10729566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
10739566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
10749566063dSJacob Faibussowitsch   PetscCall(DMGetUseNatural(dm, &useNatural));
107578569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
107678569bb4SMatthew G. Knepley   if (useNatural) {
1077b7352c5cSAlexis Marboeuf     PetscCall(DMPlexCreateNaturalVector(dm, &vNatural));
10789566063dSJacob Faibussowitsch     PetscCall(DMPlexGlobalToNaturalBegin(dm, v, vNatural));
10799566063dSJacob Faibussowitsch     PetscCall(DMPlexGlobalToNaturalEnd(dm, v, vNatural));
108078569bb4SMatthew G. Knepley   } else {
108178569bb4SMatthew G. Knepley     vNatural = v;
108278569bb4SMatthew G. Knepley   }
10836823f3c5SBlaise Bourdin 
108478569bb4SMatthew G. Knepley   /* Write local chunk of the result in the exodus file
108578569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
108678569bb4SMatthew G. Knepley      We assume that they are stored sequentially
108778569bb4SMatthew G. Knepley      Zonal variables are accessed one element block at a time, so we loop through the cell sets,
1088a5b23f4aSJose E. Roman      but once the vector has been reordered to natural size, we cannot use the label information
108978569bb4SMatthew G. Knepley      to figure out what to save where. */
109078569bb4SMatthew G. Knepley   numCS = ex_inquire_int(exoid, EX_INQ_ELEM_BLK);
10919566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numCS, &csID, numCS, &csSize));
1092792fecdfSBarry Smith   PetscCallExternal(ex_get_ids, exoid, EX_ELEM_BLOCK, csID);
109378569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
109478569bb4SMatthew G. Knepley     ex_block block;
109578569bb4SMatthew G. Knepley 
109678569bb4SMatthew G. Knepley     block.id   = csID[set];
109778569bb4SMatthew G. Knepley     block.type = EX_ELEM_BLOCK;
1098792fecdfSBarry Smith     PetscCallExternal(ex_get_block_param, exoid, &block);
109978569bb4SMatthew G. Knepley     csSize[set] = block.num_entry;
110078569bb4SMatthew G. Knepley   }
11019566063dSJacob Faibussowitsch   PetscCall(VecGetOwnershipRange(vNatural, &xs, &xe));
11029566063dSJacob Faibussowitsch   PetscCall(VecGetBlockSize(vNatural, &bs));
11039566063dSJacob Faibussowitsch   if (bs > 1) PetscCall(ISCreateStride(comm, (xe - xs) / bs, xs, bs, &compIS));
110478569bb4SMatthew G. Knepley   for (set = 0; set < numCS; set++) {
110578569bb4SMatthew G. Knepley     PetscInt csLocalSize, c;
110678569bb4SMatthew G. Knepley 
110778569bb4SMatthew G. Knepley     /* range of indices for set setID[set]: csxs:csxs + csSize[set]-1
110878569bb4SMatthew G. Knepley        local slice of zonal values:         xs/bs,xm/bs-1
110978569bb4SMatthew G. Knepley        intersection:                        max(xs/bs,csxs),min(xm/bs-1,csxs + csSize[set]-1) */
111078569bb4SMatthew G. Knepley     csLocalSize = PetscMax(0, PetscMin(xe / bs, csxs + csSize[set]) - PetscMax(xs / bs, csxs));
111178569bb4SMatthew G. Knepley     if (bs == 1) {
11129566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(vNatural, &varray));
1113792fecdfSBarry Smith       PetscCallExternal(ex_put_partial_var, exoid, step, EX_ELEM_BLOCK, offset, csID[set], PetscMax(xs - csxs, 0) + 1, csLocalSize, &varray[PetscMax(0, csxs - xs)]);
11149566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(vNatural, &varray));
111578569bb4SMatthew G. Knepley     } else {
111678569bb4SMatthew G. Knepley       for (c = 0; c < bs; ++c) {
11179566063dSJacob Faibussowitsch         PetscCall(ISStrideSetStride(compIS, (xe - xs) / bs, xs + c, bs));
11189566063dSJacob Faibussowitsch         PetscCall(VecGetSubVector(vNatural, compIS, &vComp));
11199566063dSJacob Faibussowitsch         PetscCall(VecGetArrayRead(vComp, &varray));
1120792fecdfSBarry Smith         PetscCallExternal(ex_put_partial_var, exoid, step, EX_ELEM_BLOCK, offset + c, csID[set], PetscMax(xs / bs - csxs, 0) + 1, csLocalSize, &varray[PetscMax(0, csxs - xs / bs)]);
11219566063dSJacob Faibussowitsch         PetscCall(VecRestoreArrayRead(vComp, &varray));
11229566063dSJacob Faibussowitsch         PetscCall(VecRestoreSubVector(vNatural, compIS, &vComp));
112378569bb4SMatthew G. Knepley       }
112478569bb4SMatthew G. Knepley     }
112578569bb4SMatthew G. Knepley     csxs += csSize[set];
112678569bb4SMatthew G. Knepley   }
11279566063dSJacob Faibussowitsch   PetscCall(PetscFree2(csID, csSize));
11289566063dSJacob Faibussowitsch   if (bs > 1) PetscCall(ISDestroy(&compIS));
1129b7352c5cSAlexis Marboeuf   if (useNatural) PetscCall(VecDestroy(&vNatural));
113078569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
113178569bb4SMatthew G. Knepley }
113278569bb4SMatthew G. Knepley 
113378569bb4SMatthew G. Knepley /*
113478569bb4SMatthew G. Knepley   VecLoadPlex_ExodusII_Zonal_Internal - Read a Vec corresponding to a zonal (cell based) field from an exodus file
113578569bb4SMatthew G. Knepley 
113678569bb4SMatthew G. Knepley   Collective on v
113778569bb4SMatthew G. Knepley 
113878569bb4SMatthew G. Knepley   Input Parameters:
113978569bb4SMatthew G. Knepley + v  - The vector to be written
114078569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
11416823f3c5SBlaise Bourdin . step - the time step to read at (exodus steps are numbered starting from 1)
11426823f3c5SBlaise Bourdin - offset - the location of the variable in the file
114378569bb4SMatthew G. Knepley 
114478569bb4SMatthew G. Knepley   Notes:
114578569bb4SMatthew G. Knepley   The exodus result zonal variable index is obtained by comparing the Vec name and the
114678569bb4SMatthew G. Knepley   names of zonal variables declared in the exodus file. For instance for a Vec named "V"
114778569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
114878569bb4SMatthew G. Knepley   amongst all zonal variables.
114978569bb4SMatthew G. Knepley 
115078569bb4SMatthew G. Knepley   Level: beginner
115178569bb4SMatthew G. Knepley 
1152db781477SPatrick Sanan .seealso: `EXOGetVarIndex_Internal()`, `DMPlexView_ExodusII_Internal()`, `VecViewPlex_ExodusII_Nodal_Internal()`, `VecLoadPlex_ExodusII_Nodal_Internal()`, `VecLoadPlex_ExodusII_Zonal_Internal()`
115378569bb4SMatthew G. Knepley */
1154d71ae5a4SJacob Faibussowitsch PetscErrorCode VecLoadPlex_ExodusII_Zonal_Internal(Vec v, int exoid, int step, int offset)
1155d71ae5a4SJacob Faibussowitsch {
115678569bb4SMatthew G. Knepley   MPI_Comm     comm;
115778569bb4SMatthew G. Knepley   PetscMPIInt  size;
115878569bb4SMatthew G. Knepley   DM           dm;
115978569bb4SMatthew G. Knepley   Vec          vNatural, vComp;
116022a7544eSVaclav Hapla   PetscScalar *varray;
116178569bb4SMatthew G. Knepley   PetscInt     xs, xe, bs;
116278569bb4SMatthew G. Knepley   PetscBool    useNatural;
116378569bb4SMatthew G. Knepley   IS           compIS;
116478569bb4SMatthew G. Knepley   PetscInt    *csSize, *csID;
116578569bb4SMatthew G. Knepley   PetscInt     numCS, set, csxs = 0;
116678569bb4SMatthew G. Knepley 
116778569bb4SMatthew G. Knepley   PetscFunctionBegin;
11689566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)v, &comm));
11699566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
11709566063dSJacob Faibussowitsch   PetscCall(VecGetDM(v, &dm));
11719566063dSJacob Faibussowitsch   PetscCall(DMGetUseNatural(dm, &useNatural));
117278569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
1173b7352c5cSAlexis Marboeuf   if (useNatural) PetscCall(DMPlexCreateNaturalVector(dm, &vNatural));
1174ad540459SPierre Jolivet   else vNatural = v;
11756823f3c5SBlaise Bourdin 
117678569bb4SMatthew G. Knepley   /* Read local chunk of the result in the exodus file
117778569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
117878569bb4SMatthew G. Knepley      We assume that they are stored sequentially
117978569bb4SMatthew G. Knepley      Zonal variables are accessed one element block at a time, so we loop through the cell sets,
1180a5b23f4aSJose E. Roman      but once the vector has been reordered to natural size, we cannot use the label information
118178569bb4SMatthew G. Knepley      to figure out what to save where. */
118278569bb4SMatthew G. Knepley   numCS = ex_inquire_int(exoid, EX_INQ_ELEM_BLK);
11839566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numCS, &csID, numCS, &csSize));
1184792fecdfSBarry Smith   PetscCallExternal(ex_get_ids, exoid, EX_ELEM_BLOCK, csID);
118578569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
118678569bb4SMatthew G. Knepley     ex_block block;
118778569bb4SMatthew G. Knepley 
118878569bb4SMatthew G. Knepley     block.id   = csID[set];
118978569bb4SMatthew G. Knepley     block.type = EX_ELEM_BLOCK;
1190792fecdfSBarry Smith     PetscCallExternal(ex_get_block_param, exoid, &block);
119178569bb4SMatthew G. Knepley     csSize[set] = block.num_entry;
119278569bb4SMatthew G. Knepley   }
11939566063dSJacob Faibussowitsch   PetscCall(VecGetOwnershipRange(vNatural, &xs, &xe));
11949566063dSJacob Faibussowitsch   PetscCall(VecGetBlockSize(vNatural, &bs));
11959566063dSJacob Faibussowitsch   if (bs > 1) PetscCall(ISCreateStride(comm, (xe - xs) / bs, xs, bs, &compIS));
119678569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
119778569bb4SMatthew G. Knepley     PetscInt csLocalSize, c;
119878569bb4SMatthew G. Knepley 
119978569bb4SMatthew G. Knepley     /* range of indices for set setID[set]: csxs:csxs + csSize[set]-1
120078569bb4SMatthew G. Knepley        local slice of zonal values:         xs/bs,xm/bs-1
120178569bb4SMatthew G. Knepley        intersection:                        max(xs/bs,csxs),min(xm/bs-1,csxs + csSize[set]-1) */
120278569bb4SMatthew G. Knepley     csLocalSize = PetscMax(0, PetscMin(xe / bs, csxs + csSize[set]) - PetscMax(xs / bs, csxs));
120378569bb4SMatthew G. Knepley     if (bs == 1) {
12049566063dSJacob Faibussowitsch       PetscCall(VecGetArray(vNatural, &varray));
1205792fecdfSBarry Smith       PetscCallExternal(ex_get_partial_var, exoid, step, EX_ELEM_BLOCK, offset, csID[set], PetscMax(xs - csxs, 0) + 1, csLocalSize, &varray[PetscMax(0, csxs - xs)]);
12069566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(vNatural, &varray));
120778569bb4SMatthew G. Knepley     } else {
120878569bb4SMatthew G. Knepley       for (c = 0; c < bs; ++c) {
12099566063dSJacob Faibussowitsch         PetscCall(ISStrideSetStride(compIS, (xe - xs) / bs, xs + c, bs));
12109566063dSJacob Faibussowitsch         PetscCall(VecGetSubVector(vNatural, compIS, &vComp));
12119566063dSJacob Faibussowitsch         PetscCall(VecGetArray(vComp, &varray));
1212792fecdfSBarry Smith         PetscCallExternal(ex_get_partial_var, exoid, step, EX_ELEM_BLOCK, offset + c, csID[set], PetscMax(xs / bs - csxs, 0) + 1, csLocalSize, &varray[PetscMax(0, csxs - xs / bs)]);
12139566063dSJacob Faibussowitsch         PetscCall(VecRestoreArray(vComp, &varray));
12149566063dSJacob Faibussowitsch         PetscCall(VecRestoreSubVector(vNatural, compIS, &vComp));
121578569bb4SMatthew G. Knepley       }
121678569bb4SMatthew G. Knepley     }
121778569bb4SMatthew G. Knepley     csxs += csSize[set];
121878569bb4SMatthew G. Knepley   }
12199566063dSJacob Faibussowitsch   PetscCall(PetscFree2(csID, csSize));
12209566063dSJacob Faibussowitsch   if (bs > 1) PetscCall(ISDestroy(&compIS));
122178569bb4SMatthew G. Knepley   if (useNatural) {
12229566063dSJacob Faibussowitsch     PetscCall(DMPlexNaturalToGlobalBegin(dm, vNatural, v));
12239566063dSJacob Faibussowitsch     PetscCall(DMPlexNaturalToGlobalEnd(dm, vNatural, v));
1224b7352c5cSAlexis Marboeuf     PetscCall(VecDestroy(&vNatural));
122578569bb4SMatthew G. Knepley   }
122678569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
122778569bb4SMatthew G. Knepley }
1228b53c8456SSatish Balay #endif
122978569bb4SMatthew G. Knepley 
12301e50132fSMatthew G. Knepley /*@
1231*a1cb98faSBarry Smith   PetscViewerExodusIIGetId - Get the file id of the `PETSCVIEWEREXODUSII` file
12321e50132fSMatthew G. Knepley 
1233*a1cb98faSBarry Smith   Logically Collective on viewer
12341e50132fSMatthew G. Knepley 
12351e50132fSMatthew G. Knepley   Input Parameter:
1236*a1cb98faSBarry Smith .  viewer - the `PetscViewer`
12371e50132fSMatthew G. Knepley 
12381e50132fSMatthew G. Knepley   Output Parameter:
1239d8d19677SJose E. Roman .  exoid - The ExodusII file id
12401e50132fSMatthew G. Knepley 
12411e50132fSMatthew G. Knepley   Level: intermediate
12421e50132fSMatthew G. Knepley 
1243*a1cb98faSBarry Smith .seealso: `PETSCVIEWEREXODUSII`, `PetscViewer`, `PetscViewerFileSetMode()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerBinaryOpen()`
12441e50132fSMatthew G. Knepley @*/
1245d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerExodusIIGetId(PetscViewer viewer, int *exoid)
1246d71ae5a4SJacob Faibussowitsch {
12471e50132fSMatthew G. Knepley   PetscFunctionBegin;
12481e50132fSMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
1249cac4c232SBarry Smith   PetscTryMethod(viewer, "PetscViewerGetId_C", (PetscViewer, int *), (viewer, exoid));
12506823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12516823f3c5SBlaise Bourdin }
12526823f3c5SBlaise Bourdin 
12536823f3c5SBlaise Bourdin /*@
12546823f3c5SBlaise Bourdin    PetscViewerExodusIISetOrder - Set the elements order in the exodusII file.
12556823f3c5SBlaise Bourdin 
12566823f3c5SBlaise Bourdin    Collective
12576823f3c5SBlaise Bourdin 
12586823f3c5SBlaise Bourdin    Input Parameters:
1259*a1cb98faSBarry Smith +  viewer - the `PETSCVIEWEREXODUSII` viewer
126098a6a78aSPierre Jolivet -  order - elements order
12616823f3c5SBlaise Bourdin 
12626823f3c5SBlaise Bourdin    Output Parameter:
12636823f3c5SBlaise Bourdin 
12646823f3c5SBlaise Bourdin    Level: beginner
12656823f3c5SBlaise Bourdin 
1266*a1cb98faSBarry Smith .seealso: `PETSCVIEWEREXODUSII`, `PetscViewer`, `PetscViewerExodusIIGetId()`, `PetscViewerExodusIIGetOrder()`, `PetscViewerExodusIISetOrder()`
12676823f3c5SBlaise Bourdin @*/
1268d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerExodusIISetOrder(PetscViewer viewer, PetscInt order)
1269d71ae5a4SJacob Faibussowitsch {
12706823f3c5SBlaise Bourdin   PetscFunctionBegin;
12716823f3c5SBlaise Bourdin   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
1272cac4c232SBarry Smith   PetscTryMethod(viewer, "PetscViewerSetOrder_C", (PetscViewer, PetscInt), (viewer, order));
12736823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12746823f3c5SBlaise Bourdin }
12756823f3c5SBlaise Bourdin 
12766823f3c5SBlaise Bourdin /*@
12776823f3c5SBlaise Bourdin    PetscViewerExodusIIGetOrder - Get the elements order in the exodusII file.
12786823f3c5SBlaise Bourdin 
12796823f3c5SBlaise Bourdin    Collective
12806823f3c5SBlaise Bourdin 
12816823f3c5SBlaise Bourdin    Input Parameters:
1282*a1cb98faSBarry Smith +  viewer - the `PETSCVIEWEREXODUSII` viewer
128398a6a78aSPierre Jolivet -  order - elements order
12846823f3c5SBlaise Bourdin 
12856823f3c5SBlaise Bourdin    Output Parameter:
12866823f3c5SBlaise Bourdin 
12876823f3c5SBlaise Bourdin    Level: beginner
12886823f3c5SBlaise Bourdin 
1289*a1cb98faSBarry Smith .seealso: `PETSCVIEWEREXODUSII`, `PetscViewer`, `PetscViewerExodusIIGetId()`, `PetscViewerExodusIIGetOrder()`, `PetscViewerExodusIISetOrder()`
12906823f3c5SBlaise Bourdin @*/
1291d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerExodusIIGetOrder(PetscViewer viewer, PetscInt *order)
1292d71ae5a4SJacob Faibussowitsch {
12936823f3c5SBlaise Bourdin   PetscFunctionBegin;
12946823f3c5SBlaise Bourdin   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
1295cac4c232SBarry Smith   PetscTryMethod(viewer, "PetscViewerGetOrder_C", (PetscViewer, PetscInt *), (viewer, order));
12966823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12976823f3c5SBlaise Bourdin }
12986823f3c5SBlaise Bourdin 
12996823f3c5SBlaise Bourdin /*@C
13006823f3c5SBlaise Bourdin    PetscViewerExodusIIOpen - Opens a file for ExodusII input/output.
13016823f3c5SBlaise Bourdin 
13026823f3c5SBlaise Bourdin    Collective
13036823f3c5SBlaise Bourdin 
13046823f3c5SBlaise Bourdin    Input Parameters:
13056823f3c5SBlaise Bourdin +  comm - MPI communicator
13066823f3c5SBlaise Bourdin .  name - name of file
13076823f3c5SBlaise Bourdin -  type - type of file
1308*a1cb98faSBarry Smith .vb
1309*a1cb98faSBarry Smith     FILE_MODE_WRITE - create new file for binary output
1310*a1cb98faSBarry Smith     FILE_MODE_READ - open existing file for binary input
1311*a1cb98faSBarry Smith     FILE_MODE_APPEND - open existing file for binary output
1312*a1cb98faSBarry Smith .ve
13136823f3c5SBlaise Bourdin 
13146823f3c5SBlaise Bourdin    Output Parameter:
1315*a1cb98faSBarry Smith .  exo - `PETSCVIEWEREXODUSII` `PetscViewer` for Exodus II input/output to use with the specified file
13166823f3c5SBlaise Bourdin 
13176823f3c5SBlaise Bourdin    Level: beginner
13186823f3c5SBlaise Bourdin 
1319*a1cb98faSBarry Smith .seealso: `PETSCVIEWEREXODUSII`, `PetscViewer`, `PetscViewerPushFormat()`, `PetscViewerDestroy()`,
1320db781477SPatrick Sanan           `DMLoad()`, `PetscFileMode`, `PetscViewer`, `PetscViewerSetType()`, `PetscViewerFileSetMode()`, `PetscViewerFileSetName()`
13216823f3c5SBlaise Bourdin @*/
1322d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerExodusIIOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *exo)
1323d71ae5a4SJacob Faibussowitsch {
13246823f3c5SBlaise Bourdin   PetscFunctionBegin;
13259566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(comm, exo));
13269566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(*exo, PETSCVIEWEREXODUSII));
13279566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetMode(*exo, type));
13289566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetName(*exo, name));
13299566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions(*exo));
13301e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
13311e50132fSMatthew G. Knepley }
13321e50132fSMatthew G. Knepley 
133333751fbdSMichael Lange /*@C
1334*a1cb98faSBarry Smith   DMPlexCreateExodusFromFile - Create a `DMPLEX` mesh from an ExodusII file.
133533751fbdSMichael Lange 
1336d083f849SBarry Smith   Collective
133733751fbdSMichael Lange 
133833751fbdSMichael Lange   Input Parameters:
133933751fbdSMichael Lange + comm  - The MPI communicator
134033751fbdSMichael Lange . filename - The name of the ExodusII file
134133751fbdSMichael Lange - interpolate - Create faces and edges in the mesh
134233751fbdSMichael Lange 
134333751fbdSMichael Lange   Output Parameter:
1344*a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
134533751fbdSMichael Lange 
134633751fbdSMichael Lange   Level: beginner
134733751fbdSMichael Lange 
1348*a1cb98faSBarry Smith .seealso: [](chapter_unstructured), `DM`, `PETSCVIEWEREXODUSII`, `DMPLEX`, `DMCreate()`, `DMPlexCreateExodus()`
134933751fbdSMichael Lange @*/
1350d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateExodusFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
1351d71ae5a4SJacob Faibussowitsch {
135233751fbdSMichael Lange   PetscMPIInt rank;
135333751fbdSMichael Lange #if defined(PETSC_HAVE_EXODUSII)
1354e84f7031SBlaise Bourdin   int   CPU_word_size = sizeof(PetscReal), IO_word_size = 0, exoid = -1;
135533751fbdSMichael Lange   float version;
135633751fbdSMichael Lange #endif
135733751fbdSMichael Lange 
135833751fbdSMichael Lange   PetscFunctionBegin;
135933751fbdSMichael Lange   PetscValidCharPointer(filename, 2);
13609566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
136133751fbdSMichael Lange #if defined(PETSC_HAVE_EXODUSII)
1362dd400576SPatrick Sanan   if (rank == 0) {
136333751fbdSMichael Lange     exoid = ex_open(filename, EX_READ, &CPU_word_size, &IO_word_size, &version);
136408401ef6SPierre Jolivet     PetscCheck(exoid > 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open(\"%s\",...) did not return a valid file ID", filename);
136533751fbdSMichael Lange   }
13669566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateExodus(comm, exoid, interpolate, dm));
136748a46eb9SPierre Jolivet   if (rank == 0) PetscCallExternal(ex_close, exoid);
1368b458e8f1SJose E. Roman   PetscFunctionReturn(0);
136933751fbdSMichael Lange #else
137033751fbdSMichael Lange   SETERRQ(comm, PETSC_ERR_SUP, "This method requires ExodusII support. Reconfigure using --download-exodusii");
137133751fbdSMichael Lange #endif
137233751fbdSMichael Lange }
137333751fbdSMichael Lange 
13748f861fbcSMatthew G. Knepley #if defined(PETSC_HAVE_EXODUSII)
1375d71ae5a4SJacob Faibussowitsch static PetscErrorCode ExodusGetCellType_Internal(const char *elem_type, DMPolytopeType *ct)
1376d71ae5a4SJacob Faibussowitsch {
13778f861fbcSMatthew G. Knepley   PetscBool flg;
13788f861fbcSMatthew G. Knepley 
13798f861fbcSMatthew G. Knepley   PetscFunctionBegin;
13808f861fbcSMatthew G. Knepley   *ct = DM_POLYTOPE_UNKNOWN;
13819566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "TRI", &flg));
13829371c9d4SSatish Balay   if (flg) {
13839371c9d4SSatish Balay     *ct = DM_POLYTOPE_TRIANGLE;
13849371c9d4SSatish Balay     goto done;
13859371c9d4SSatish Balay   }
13869566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "TRI3", &flg));
13879371c9d4SSatish Balay   if (flg) {
13889371c9d4SSatish Balay     *ct = DM_POLYTOPE_TRIANGLE;
13899371c9d4SSatish Balay     goto done;
13909371c9d4SSatish Balay   }
13919566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "QUAD", &flg));
13929371c9d4SSatish Balay   if (flg) {
13939371c9d4SSatish Balay     *ct = DM_POLYTOPE_QUADRILATERAL;
13949371c9d4SSatish Balay     goto done;
13959371c9d4SSatish Balay   }
13969566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "QUAD4", &flg));
13979371c9d4SSatish Balay   if (flg) {
13989371c9d4SSatish Balay     *ct = DM_POLYTOPE_QUADRILATERAL;
13999371c9d4SSatish Balay     goto done;
14009371c9d4SSatish Balay   }
14019566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "SHELL4", &flg));
14029371c9d4SSatish Balay   if (flg) {
14039371c9d4SSatish Balay     *ct = DM_POLYTOPE_QUADRILATERAL;
14049371c9d4SSatish Balay     goto done;
14059371c9d4SSatish Balay   }
14069566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "TETRA", &flg));
14079371c9d4SSatish Balay   if (flg) {
14089371c9d4SSatish Balay     *ct = DM_POLYTOPE_TETRAHEDRON;
14099371c9d4SSatish Balay     goto done;
14109371c9d4SSatish Balay   }
14119566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "TET4", &flg));
14129371c9d4SSatish Balay   if (flg) {
14139371c9d4SSatish Balay     *ct = DM_POLYTOPE_TETRAHEDRON;
14149371c9d4SSatish Balay     goto done;
14159371c9d4SSatish Balay   }
14169566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "WEDGE", &flg));
14179371c9d4SSatish Balay   if (flg) {
14189371c9d4SSatish Balay     *ct = DM_POLYTOPE_TRI_PRISM;
14199371c9d4SSatish Balay     goto done;
14209371c9d4SSatish Balay   }
14219566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "HEX", &flg));
14229371c9d4SSatish Balay   if (flg) {
14239371c9d4SSatish Balay     *ct = DM_POLYTOPE_HEXAHEDRON;
14249371c9d4SSatish Balay     goto done;
14259371c9d4SSatish Balay   }
14269566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "HEX8", &flg));
14279371c9d4SSatish Balay   if (flg) {
14289371c9d4SSatish Balay     *ct = DM_POLYTOPE_HEXAHEDRON;
14299371c9d4SSatish Balay     goto done;
14309371c9d4SSatish Balay   }
14319566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(elem_type, "HEXAHEDRON", &flg));
14329371c9d4SSatish Balay   if (flg) {
14339371c9d4SSatish Balay     *ct = DM_POLYTOPE_HEXAHEDRON;
14349371c9d4SSatish Balay     goto done;
14359371c9d4SSatish Balay   }
143698921bdaSJacob Faibussowitsch   SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unrecognized element type %s", elem_type);
14378f861fbcSMatthew G. Knepley done:
14388f861fbcSMatthew G. Knepley   PetscFunctionReturn(0);
14398f861fbcSMatthew G. Knepley }
14408f861fbcSMatthew G. Knepley #endif
14418f861fbcSMatthew G. Knepley 
1442552f7358SJed Brown /*@
1443*a1cb98faSBarry Smith   DMPlexCreateExodus - Create a `DMPLEX` mesh from an ExodusII file ID.
1444552f7358SJed Brown 
1445d083f849SBarry Smith   Collective
1446552f7358SJed Brown 
1447552f7358SJed Brown   Input Parameters:
1448552f7358SJed Brown + comm  - The MPI communicator
1449552f7358SJed Brown . exoid - The ExodusII id associated with a exodus file and obtained using ex_open
1450552f7358SJed Brown - interpolate - Create faces and edges in the mesh
1451552f7358SJed Brown 
1452552f7358SJed Brown   Output Parameter:
1453*a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
1454552f7358SJed Brown 
1455552f7358SJed Brown   Level: beginner
1456552f7358SJed Brown 
1457*a1cb98faSBarry Smith .seealso: [](chapter_unstructured), `DM`, `PETSCVIEWEREXODUSII`, `DMPLEX`, `DMPLEX`, `DMCreate()`
1458552f7358SJed Brown @*/
1459d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateExodus(MPI_Comm comm, PetscInt exoid, PetscBool interpolate, DM *dm)
1460d71ae5a4SJacob Faibussowitsch {
1461552f7358SJed Brown #if defined(PETSC_HAVE_EXODUSII)
1462552f7358SJed Brown   PetscMPIInt  num_proc, rank;
1463ae9eebabSLisandro Dalcin   DMLabel      cellSets = NULL, faceSets = NULL, vertSets = NULL;
1464552f7358SJed Brown   PetscSection coordSection;
1465552f7358SJed Brown   Vec          coordinates;
1466552f7358SJed Brown   PetscScalar *coords;
1467552f7358SJed Brown   PetscInt     coordSize, v;
1468552f7358SJed Brown   /* Read from ex_get_init() */
1469552f7358SJed Brown   char title[PETSC_MAX_PATH_LEN + 1];
14705f80ce2aSJacob Faibussowitsch   int  dim = 0, dimEmbed = 0, numVertices = 0, numCells = 0;
1471552f7358SJed Brown   int  num_cs = 0, num_vs = 0, num_fs = 0;
1472552f7358SJed Brown #endif
1473552f7358SJed Brown 
1474552f7358SJed Brown   PetscFunctionBegin;
1475552f7358SJed Brown #if defined(PETSC_HAVE_EXODUSII)
14769566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
14779566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &num_proc));
14789566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
14799566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
1480a5b23f4aSJose E. Roman   /* Open EXODUS II file and read basic information on rank 0, then broadcast to all processors */
1481dd400576SPatrick Sanan   if (rank == 0) {
14829566063dSJacob Faibussowitsch     PetscCall(PetscMemzero(title, PETSC_MAX_PATH_LEN + 1));
1483792fecdfSBarry Smith     PetscCallExternal(ex_get_init, exoid, title, &dimEmbed, &numVertices, &numCells, &num_cs, &num_vs, &num_fs);
148428b400f6SJacob Faibussowitsch     PetscCheck(num_cs, PETSC_COMM_SELF, PETSC_ERR_SUP, "Exodus file does not contain any cell set");
1485552f7358SJed Brown   }
14869566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Bcast(title, PETSC_MAX_PATH_LEN + 1, MPI_CHAR, 0, comm));
14879566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm));
14889566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)*dm, title));
14899566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(*dm, 0, numCells + numVertices));
1490412e9a14SMatthew G. Knepley   /*   We do not want this label automatically computed, instead we compute it here */
14919566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(*dm, "celltype"));
1492552f7358SJed Brown 
1493552f7358SJed Brown   /* Read cell sets information */
1494dd400576SPatrick Sanan   if (rank == 0) {
1495552f7358SJed Brown     PetscInt *cone;
1496e8f6893fSMatthew G. Knepley     int       c, cs, ncs, c_loc, v, v_loc;
1497552f7358SJed Brown     /* Read from ex_get_elem_blk_ids() */
1498e8f6893fSMatthew G. Knepley     int *cs_id, *cs_order;
1499552f7358SJed Brown     /* Read from ex_get_elem_block() */
1500552f7358SJed Brown     char buffer[PETSC_MAX_PATH_LEN + 1];
1501e8f6893fSMatthew G. Knepley     int  num_cell_in_set, num_vertex_per_cell, num_hybrid, num_attr;
1502552f7358SJed Brown     /* Read from ex_get_elem_conn() */
1503552f7358SJed Brown     int *cs_connect;
1504552f7358SJed Brown 
1505552f7358SJed Brown     /* Get cell sets IDs */
15069566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(num_cs, &cs_id, num_cs, &cs_order));
1507792fecdfSBarry Smith     PetscCallExternal(ex_get_ids, exoid, EX_ELEM_BLOCK, cs_id);
1508552f7358SJed Brown     /* Read the cell set connectivity table and build mesh topology
1509552f7358SJed Brown        EXO standard requires that cells in cell sets be numbered sequentially and be pairwise disjoint. */
1510e8f6893fSMatthew G. Knepley     /* Check for a hybrid mesh */
1511e8f6893fSMatthew G. Knepley     for (cs = 0, num_hybrid = 0; cs < num_cs; ++cs) {
15128f861fbcSMatthew G. Knepley       DMPolytopeType ct;
15138f861fbcSMatthew G. Knepley       char           elem_type[PETSC_MAX_PATH_LEN];
15148f861fbcSMatthew G. Knepley 
15159566063dSJacob Faibussowitsch       PetscCall(PetscArrayzero(elem_type, sizeof(elem_type)));
1516792fecdfSBarry Smith       PetscCallExternal(ex_get_elem_type, exoid, cs_id[cs], elem_type);
15179566063dSJacob Faibussowitsch       PetscCall(ExodusGetCellType_Internal(elem_type, &ct));
15188f861fbcSMatthew G. Knepley       dim = PetscMax(dim, DMPolytopeTypeGetDim(ct));
1519792fecdfSBarry Smith       PetscCallExternal(ex_get_block, exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set, &num_vertex_per_cell, 0, 0, &num_attr);
15208f861fbcSMatthew G. Knepley       switch (ct) {
15218f861fbcSMatthew G. Knepley       case DM_POLYTOPE_TRI_PRISM:
1522e8f6893fSMatthew G. Knepley         cs_order[cs] = cs;
1523e8f6893fSMatthew G. Knepley         ++num_hybrid;
1524e8f6893fSMatthew G. Knepley         break;
1525e8f6893fSMatthew G. Knepley       default:
1526e8f6893fSMatthew G. Knepley         for (c = cs; c > cs - num_hybrid; --c) cs_order[c] = cs_order[c - 1];
1527e8f6893fSMatthew G. Knepley         cs_order[cs - num_hybrid] = cs;
1528e8f6893fSMatthew G. Knepley       }
1529e8f6893fSMatthew G. Knepley     }
1530552f7358SJed Brown     /* First set sizes */
1531e8f6893fSMatthew G. Knepley     for (ncs = 0, c = 0; ncs < num_cs; ++ncs) {
15328f861fbcSMatthew G. Knepley       DMPolytopeType ct;
15338f861fbcSMatthew G. Knepley       char           elem_type[PETSC_MAX_PATH_LEN];
1534e8f6893fSMatthew G. Knepley       const PetscInt cs = cs_order[ncs];
15358f861fbcSMatthew G. Knepley 
15369566063dSJacob Faibussowitsch       PetscCall(PetscArrayzero(elem_type, sizeof(elem_type)));
1537792fecdfSBarry Smith       PetscCallExternal(ex_get_elem_type, exoid, cs_id[cs], elem_type);
15389566063dSJacob Faibussowitsch       PetscCall(ExodusGetCellType_Internal(elem_type, &ct));
1539792fecdfSBarry Smith       PetscCallExternal(ex_get_block, exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set, &num_vertex_per_cell, 0, 0, &num_attr);
1540552f7358SJed Brown       for (c_loc = 0; c_loc < num_cell_in_set; ++c_loc, ++c) {
15419566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeSize(*dm, c, num_vertex_per_cell));
15429566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCellType(*dm, c, ct));
1543552f7358SJed Brown       }
1544552f7358SJed Brown     }
15459566063dSJacob Faibussowitsch     for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(*dm, v, DM_POLYTOPE_POINT));
15469566063dSJacob Faibussowitsch     PetscCall(DMSetUp(*dm));
1547e8f6893fSMatthew G. Knepley     for (ncs = 0, c = 0; ncs < num_cs; ++ncs) {
1548e8f6893fSMatthew G. Knepley       const PetscInt cs = cs_order[ncs];
1549792fecdfSBarry Smith       PetscCallExternal(ex_get_block, exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set, &num_vertex_per_cell, 0, 0, &num_attr);
15509566063dSJacob Faibussowitsch       PetscCall(PetscMalloc2(num_vertex_per_cell * num_cell_in_set, &cs_connect, num_vertex_per_cell, &cone));
1551792fecdfSBarry Smith       PetscCallExternal(ex_get_conn, exoid, EX_ELEM_BLOCK, cs_id[cs], cs_connect, NULL, NULL);
1552eaf898f9SPatrick Sanan       /* EXO uses Fortran-based indexing, DMPlex uses C-style and numbers cell first then vertices. */
1553552f7358SJed Brown       for (c_loc = 0, v = 0; c_loc < num_cell_in_set; ++c_loc, ++c) {
1554636e64ffSStefano Zampini         DMPolytopeType ct;
1555636e64ffSStefano Zampini 
1556ad540459SPierre Jolivet         for (v_loc = 0; v_loc < num_vertex_per_cell; ++v_loc, ++v) cone[v_loc] = cs_connect[v] + numCells - 1;
15579566063dSJacob Faibussowitsch         PetscCall(DMPlexGetCellType(*dm, c, &ct));
15589566063dSJacob Faibussowitsch         PetscCall(DMPlexInvertCell(ct, cone));
15599566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(*dm, c, cone));
15609566063dSJacob Faibussowitsch         PetscCall(DMSetLabelValue_Fast(*dm, &cellSets, "Cell Sets", c, cs_id[cs]));
1561552f7358SJed Brown       }
15629566063dSJacob Faibussowitsch       PetscCall(PetscFree2(cs_connect, cone));
1563552f7358SJed Brown     }
15649566063dSJacob Faibussowitsch     PetscCall(PetscFree2(cs_id, cs_order));
1565552f7358SJed Brown   }
15668f861fbcSMatthew G. Knepley   {
15678f861fbcSMatthew G. Knepley     PetscInt ints[] = {dim, dimEmbed};
15688f861fbcSMatthew G. Knepley 
15699566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Bcast(ints, 2, MPIU_INT, 0, comm));
15709566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(*dm, ints[0]));
15719566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinateDim(*dm, ints[1]));
15728f861fbcSMatthew G. Knepley     dim      = ints[0];
15738f861fbcSMatthew G. Knepley     dimEmbed = ints[1];
15748f861fbcSMatthew G. Knepley   }
15759566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(*dm));
15769566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(*dm));
1577552f7358SJed Brown   if (interpolate) {
15785fd9971aSMatthew G. Knepley     DM idm;
1579552f7358SJed Brown 
15809566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
15819566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
1582552f7358SJed Brown     *dm = idm;
1583552f7358SJed Brown   }
1584552f7358SJed Brown 
1585552f7358SJed Brown   /* Create vertex set label */
1586dd400576SPatrick Sanan   if (rank == 0 && (num_vs > 0)) {
1587552f7358SJed Brown     int vs, v;
1588552f7358SJed Brown     /* Read from ex_get_node_set_ids() */
1589552f7358SJed Brown     int *vs_id;
1590552f7358SJed Brown     /* Read from ex_get_node_set_param() */
1591f958083aSBlaise Bourdin     int num_vertex_in_set;
1592552f7358SJed Brown     /* Read from ex_get_node_set() */
1593552f7358SJed Brown     int *vs_vertex_list;
1594552f7358SJed Brown 
1595552f7358SJed Brown     /* Get vertex set ids */
15969566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(num_vs, &vs_id));
1597792fecdfSBarry Smith     PetscCallExternal(ex_get_ids, exoid, EX_NODE_SET, vs_id);
1598552f7358SJed Brown     for (vs = 0; vs < num_vs; ++vs) {
1599792fecdfSBarry Smith       PetscCallExternal(ex_get_set_param, exoid, EX_NODE_SET, vs_id[vs], &num_vertex_in_set, NULL);
16009566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(num_vertex_in_set, &vs_vertex_list));
1601792fecdfSBarry Smith       PetscCallExternal(ex_get_set, exoid, EX_NODE_SET, vs_id[vs], vs_vertex_list, NULL);
160248a46eb9SPierre Jolivet       for (v = 0; v < num_vertex_in_set; ++v) PetscCall(DMSetLabelValue_Fast(*dm, &vertSets, "Vertex Sets", vs_vertex_list[v] + numCells - 1, vs_id[vs]));
16039566063dSJacob Faibussowitsch       PetscCall(PetscFree(vs_vertex_list));
1604552f7358SJed Brown     }
16059566063dSJacob Faibussowitsch     PetscCall(PetscFree(vs_id));
1606552f7358SJed Brown   }
1607552f7358SJed Brown   /* Read coordinates */
16089566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(*dm, &coordSection));
16099566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
16109566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed));
16119566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
1612552f7358SJed Brown   for (v = numCells; v < numCells + numVertices; ++v) {
16139566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed));
16149566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed));
1615552f7358SJed Brown   }
16169566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
16179566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
16189566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
16199566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
16209566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
16219566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, dimEmbed));
16229566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
16239566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
1624dd400576SPatrick Sanan   if (rank == 0) {
1625e84f7031SBlaise Bourdin     PetscReal *x, *y, *z;
1626552f7358SJed Brown 
16279566063dSJacob Faibussowitsch     PetscCall(PetscMalloc3(numVertices, &x, numVertices, &y, numVertices, &z));
1628792fecdfSBarry Smith     PetscCallExternal(ex_get_coord, exoid, x, y, z);
16298f861fbcSMatthew G. Knepley     if (dimEmbed > 0) {
16308f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v * dimEmbed + 0] = x[v];
16310d644c17SKarl Rupp     }
16328f861fbcSMatthew G. Knepley     if (dimEmbed > 1) {
16338f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v * dimEmbed + 1] = y[v];
16340d644c17SKarl Rupp     }
16358f861fbcSMatthew G. Knepley     if (dimEmbed > 2) {
16368f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v * dimEmbed + 2] = z[v];
16370d644c17SKarl Rupp     }
16389566063dSJacob Faibussowitsch     PetscCall(PetscFree3(x, y, z));
1639552f7358SJed Brown   }
16409566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
16419566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
16429566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
1643552f7358SJed Brown 
1644552f7358SJed Brown   /* Create side set label */
1645dd400576SPatrick Sanan   if (rank == 0 && interpolate && (num_fs > 0)) {
1646552f7358SJed Brown     int fs, f, voff;
1647552f7358SJed Brown     /* Read from ex_get_side_set_ids() */
1648552f7358SJed Brown     int *fs_id;
1649552f7358SJed Brown     /* Read from ex_get_side_set_param() */
1650f958083aSBlaise Bourdin     int num_side_in_set;
1651552f7358SJed Brown     /* Read from ex_get_side_set_node_list() */
1652552f7358SJed Brown     int *fs_vertex_count_list, *fs_vertex_list;
1653ef073283Smichael_afanasiev     /* Read side set labels */
1654ef073283Smichael_afanasiev     char   fs_name[MAX_STR_LENGTH + 1];
1655034d25ccSMatthew G. Knepley     size_t fs_name_len;
1656552f7358SJed Brown 
1657552f7358SJed Brown     /* Get side set ids */
16589566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(num_fs, &fs_id));
1659792fecdfSBarry Smith     PetscCallExternal(ex_get_ids, exoid, EX_SIDE_SET, fs_id);
1660552f7358SJed Brown     for (fs = 0; fs < num_fs; ++fs) {
1661792fecdfSBarry Smith       PetscCallExternal(ex_get_set_param, exoid, EX_SIDE_SET, fs_id[fs], &num_side_in_set, NULL);
16629566063dSJacob Faibussowitsch       PetscCall(PetscMalloc2(num_side_in_set, &fs_vertex_count_list, num_side_in_set * 4, &fs_vertex_list));
1663792fecdfSBarry Smith       PetscCallExternal(ex_get_side_set_node_list, exoid, fs_id[fs], fs_vertex_count_list, fs_vertex_list);
1664ef073283Smichael_afanasiev       /* Get the specific name associated with this side set ID. */
1665ef073283Smichael_afanasiev       int fs_name_err = ex_get_name(exoid, EX_SIDE_SET, fs_id[fs], fs_name);
1666034d25ccSMatthew G. Knepley       if (!fs_name_err) {
1667034d25ccSMatthew G. Knepley         PetscCall(PetscStrlen(fs_name, &fs_name_len));
1668034d25ccSMatthew G. Knepley         if (fs_name_len == 0) PetscCall(PetscStrncpy(fs_name, "Face Sets", MAX_STR_LENGTH));
1669034d25ccSMatthew G. Knepley       }
1670552f7358SJed Brown       for (f = 0, voff = 0; f < num_side_in_set; ++f) {
16710298fd71SBarry Smith         const PetscInt *faces    = NULL;
1672552f7358SJed Brown         PetscInt        faceSize = fs_vertex_count_list[f], numFaces;
1673552f7358SJed Brown         PetscInt        faceVertices[4], v;
1674552f7358SJed Brown 
1675197f6770SJed Brown         PetscCheck(faceSize <= 4, comm, PETSC_ERR_ARG_WRONG, "ExodusII side cannot have %" PetscInt_FMT " > 4 vertices", faceSize);
1676ad540459SPierre Jolivet         for (v = 0; v < faceSize; ++v, ++voff) faceVertices[v] = fs_vertex_list[voff] + numCells - 1;
16779566063dSJacob Faibussowitsch         PetscCall(DMPlexGetFullJoin(*dm, faceSize, faceVertices, &numFaces, &faces));
1678197f6770SJed Brown         PetscCheck(numFaces == 1, comm, PETSC_ERR_ARG_WRONG, "Invalid ExodusII side %d in set %d maps to %" PetscInt_FMT " faces", f, fs, numFaces);
16799566063dSJacob Faibussowitsch         PetscCall(DMSetLabelValue_Fast(*dm, &faceSets, "Face Sets", faces[0], fs_id[fs]));
1680ef073283Smichael_afanasiev         /* Only add the label if one has been detected for this side set. */
1681034d25ccSMatthew G. Knepley         if (!fs_name_err) PetscCall(DMSetLabelValue(*dm, fs_name, faces[0], fs_id[fs]));
16829566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreJoin(*dm, faceSize, faceVertices, &numFaces, &faces));
1683552f7358SJed Brown       }
16849566063dSJacob Faibussowitsch       PetscCall(PetscFree2(fs_vertex_count_list, fs_vertex_list));
1685552f7358SJed Brown     }
16869566063dSJacob Faibussowitsch     PetscCall(PetscFree(fs_id));
1687552f7358SJed Brown   }
1688ae9eebabSLisandro Dalcin 
1689ae9eebabSLisandro Dalcin   { /* Create Cell/Face/Vertex Sets labels at all processes */
16909371c9d4SSatish Balay     enum {
16919371c9d4SSatish Balay       n = 3
16929371c9d4SSatish Balay     };
1693ae9eebabSLisandro Dalcin     PetscBool flag[n];
1694ae9eebabSLisandro Dalcin 
1695ae9eebabSLisandro Dalcin     flag[0] = cellSets ? PETSC_TRUE : PETSC_FALSE;
1696ae9eebabSLisandro Dalcin     flag[1] = faceSets ? PETSC_TRUE : PETSC_FALSE;
1697ae9eebabSLisandro Dalcin     flag[2] = vertSets ? PETSC_TRUE : PETSC_FALSE;
16989566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Bcast(flag, n, MPIU_BOOL, 0, comm));
16999566063dSJacob Faibussowitsch     if (flag[0]) PetscCall(DMCreateLabel(*dm, "Cell Sets"));
17009566063dSJacob Faibussowitsch     if (flag[1]) PetscCall(DMCreateLabel(*dm, "Face Sets"));
17019566063dSJacob Faibussowitsch     if (flag[2]) PetscCall(DMCreateLabel(*dm, "Vertex Sets"));
1702ae9eebabSLisandro Dalcin   }
1703b458e8f1SJose E. Roman   PetscFunctionReturn(0);
1704552f7358SJed Brown #else
1705552f7358SJed Brown   SETERRQ(comm, PETSC_ERR_SUP, "This method requires ExodusII support. Reconfigure using --download-exodusii");
1706552f7358SJed Brown #endif
1707552f7358SJed Brown }
1708