xref: /petsc/src/dm/impls/plex/plexexodusii.c (revision 98a6a78a7d031693dfb853baf868f3a083ee603d)
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)
1278569bb4SMatthew G. Knepley /*
131e50132fSMatthew G. Knepley   PETSC_VIEWER_EXODUSII_ - Creates an ExodusII PetscViewer shared by all processors in a communicator.
141e50132fSMatthew G. Knepley 
151e50132fSMatthew G. Knepley   Collective
161e50132fSMatthew G. Knepley 
171e50132fSMatthew G. Knepley   Input Parameter:
181e50132fSMatthew G. Knepley . comm - the MPI communicator to share the ExodusII PetscViewer
191e50132fSMatthew G. Knepley 
201e50132fSMatthew G. Knepley   Level: intermediate
211e50132fSMatthew G. Knepley 
221e50132fSMatthew G. Knepley   Notes:
231e50132fSMatthew G. Knepley     misses Fortran bindings
241e50132fSMatthew G. Knepley 
251e50132fSMatthew G. Knepley   Notes:
261e50132fSMatthew G. Knepley   Unlike almost all other PETSc routines, PETSC_VIEWER_EXODUSII_ does not return
271e50132fSMatthew G. Knepley   an error code.  The GLVIS PetscViewer is usually used in the form
281e50132fSMatthew G. Knepley $       XXXView(XXX object, PETSC_VIEWER_EXODUSII_(comm));
291e50132fSMatthew G. Knepley 
3000373969SVaclav Hapla .seealso: PetscViewerExodusIIOpen(), PetscViewerType, PetscViewerCreate(), PetscViewerDestroy()
311e50132fSMatthew G. Knepley */
321e50132fSMatthew G. Knepley PetscViewer PETSC_VIEWER_EXODUSII_(MPI_Comm comm)
331e50132fSMatthew G. Knepley {
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);
3900373969SVaclav Hapla   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_EXODUSII_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
401e50132fSMatthew G. Knepley   ierr = PetscObjectRegisterDestroy((PetscObject) viewer);
4100373969SVaclav Hapla   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_EXODUSII_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
421e50132fSMatthew G. Knepley   PetscFunctionReturn(viewer);
431e50132fSMatthew G. Knepley }
441e50132fSMatthew G. Knepley 
451e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerView_ExodusII(PetscViewer v, PetscViewer viewer)
461e50132fSMatthew G. Knepley {
476823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) v->data;
481e50132fSMatthew G. Knepley   PetscErrorCode        ierr;
491e50132fSMatthew G. Knepley 
501e50132fSMatthew G. Knepley   PetscFunctionBegin;
511e50132fSMatthew G. Knepley   if (exo->filename) {ierr = PetscViewerASCIIPrintf(viewer, "Filename:    %s\n", exo->filename);CHKERRQ(ierr);}
526823f3c5SBlaise Bourdin   if (exo->exoid)    {ierr = PetscViewerASCIIPrintf(viewer, "exoid:       %d\n", exo->exoid)   ;CHKERRQ(ierr);}
536823f3c5SBlaise Bourdin   if (exo->btype)    {ierr = PetscViewerASCIIPrintf(viewer, "IO Mode:     %d\n", exo->btype)   ;CHKERRQ(ierr);}
546823f3c5SBlaise Bourdin   if (exo->order)    {ierr = PetscViewerASCIIPrintf(viewer, "Mesh order:  %d\n", exo->order)   ;CHKERRQ(ierr);}
551e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
561e50132fSMatthew G. Knepley }
571e50132fSMatthew G. Knepley 
581e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerSetFromOptions_ExodusII(PetscOptionItems *PetscOptionsObject, PetscViewer v)
591e50132fSMatthew G. Knepley {
601e50132fSMatthew G. Knepley   PetscErrorCode ierr;
611e50132fSMatthew G. Knepley 
621e50132fSMatthew G. Knepley   PetscFunctionBegin;
631e50132fSMatthew G. Knepley   ierr = PetscOptionsHead(PetscOptionsObject, "ExodusII PetscViewer Options");CHKERRQ(ierr);
641e50132fSMatthew G. Knepley   ierr = PetscOptionsTail();CHKERRQ(ierr);
651e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
661e50132fSMatthew G. Knepley }
671e50132fSMatthew G. Knepley 
681e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerSetUp_ExodusII(PetscViewer viewer)
691e50132fSMatthew G. Knepley {
701e50132fSMatthew G. Knepley   PetscFunctionBegin;
711e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
721e50132fSMatthew G. Knepley }
731e50132fSMatthew G. Knepley 
741e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerDestroy_ExodusII(PetscViewer viewer)
751e50132fSMatthew G. Knepley {
761e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
771e50132fSMatthew G. Knepley   PetscErrorCode        ierr;
781e50132fSMatthew G. Knepley 
791e50132fSMatthew G. Knepley   PetscFunctionBegin;
806823f3c5SBlaise Bourdin   if (exo->exoid >= 0) {PetscStackCallStandard(ex_close,(exo->exoid));}
816823f3c5SBlaise Bourdin   ierr = PetscFree(exo->filename);CHKERRQ(ierr);
821e50132fSMatthew G. Knepley   ierr = PetscFree(exo);CHKERRQ(ierr);
831e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",NULL);CHKERRQ(ierr);
841e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",NULL);CHKERRQ(ierr);
851e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",NULL);CHKERRQ(ierr);
866823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerGetId_C",NULL);CHKERRQ(ierr);
876823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerGetOrder_C",NULL);CHKERRQ(ierr);
886823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerSetOrder_C",NULL);CHKERRQ(ierr);
891e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
901e50132fSMatthew G. Knepley }
911e50132fSMatthew G. Knepley 
921e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerFileSetName_ExodusII(PetscViewer viewer, const char name[])
931e50132fSMatthew G. Knepley {
941e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
951e50132fSMatthew G. Knepley   PetscMPIInt           rank;
961e50132fSMatthew G. Knepley   int                   CPU_word_size, IO_word_size, EXO_mode;
971e50132fSMatthew G. Knepley   PetscErrorCode        ierr;
986823f3c5SBlaise Bourdin   MPI_Info              mpi_info = MPI_INFO_NULL;
996823f3c5SBlaise Bourdin   float                 EXO_version;
1001e50132fSMatthew G. Knepley 
101ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) viewer), &rank);CHKERRMPI(ierr);
1021e50132fSMatthew G. Knepley   CPU_word_size = sizeof(PetscReal);
1031e50132fSMatthew G. Knepley   IO_word_size  = sizeof(PetscReal);
1041e50132fSMatthew G. Knepley 
1051e50132fSMatthew G. Knepley   PetscFunctionBegin;
1066823f3c5SBlaise Bourdin   if (exo->exoid >= 0) {
1076823f3c5SBlaise Bourdin     PetscStackCallStandard(ex_close,(exo->exoid));
1086823f3c5SBlaise Bourdin     exo->exoid = -1;
1096823f3c5SBlaise Bourdin   }
1101e50132fSMatthew G. Knepley   if (exo->filename) {ierr = PetscFree(exo->filename);CHKERRQ(ierr);}
1111e50132fSMatthew G. Knepley   ierr = PetscStrallocpy(name, &exo->filename);CHKERRQ(ierr);
1121e50132fSMatthew G. Knepley   switch (exo->btype) {
1131e50132fSMatthew G. Knepley   case FILE_MODE_READ:
1146823f3c5SBlaise Bourdin     EXO_mode = EX_READ;
1151e50132fSMatthew G. Knepley     break;
1161e50132fSMatthew G. Knepley   case FILE_MODE_APPEND:
1176823f3c5SBlaise Bourdin   case FILE_MODE_UPDATE:
1186823f3c5SBlaise Bourdin   case FILE_MODE_APPEND_UPDATE:
1196823f3c5SBlaise Bourdin     /* Will fail if the file does not already exist */
1206823f3c5SBlaise Bourdin     EXO_mode = EX_WRITE;
1211e50132fSMatthew G. Knepley     break;
1221e50132fSMatthew G. Knepley   case FILE_MODE_WRITE:
1236823f3c5SBlaise Bourdin     /*
1246823f3c5SBlaise Bourdin       exodus only allows writing geometry upon file creation, so we will let DMView create the file.
1256823f3c5SBlaise Bourdin     */
1266823f3c5SBlaise Bourdin     PetscFunctionReturn(0);
1271e50132fSMatthew G. Knepley     break;
1281e50132fSMatthew G. Knepley   default:
1291e50132fSMatthew G. Knepley     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
1301e50132fSMatthew G. Knepley   }
1311e50132fSMatthew G. Knepley   #if defined(PETSC_USE_64BIT_INDICES)
1321e50132fSMatthew G. Knepley   EXO_mode += EX_ALL_INT64_API;
1331e50132fSMatthew G. Knepley   #endif
1346823f3c5SBlaise Bourdin   exo->exoid = ex_open_par(name,EXO_mode,&CPU_word_size,&IO_word_size,&EXO_version,PETSC_COMM_WORLD,mpi_info);
1356823f3c5SBlaise Bourdin   if (exo->exoid < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open_par failed for %s", name);
1361e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1371e50132fSMatthew G. Knepley }
1381e50132fSMatthew G. Knepley 
1391e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerFileGetName_ExodusII(PetscViewer viewer, const char **name)
1401e50132fSMatthew G. Knepley {
1411e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1421e50132fSMatthew G. Knepley 
1431e50132fSMatthew G. Knepley   PetscFunctionBegin;
1441e50132fSMatthew G. Knepley   *name = exo->filename;
1451e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1461e50132fSMatthew G. Knepley }
1471e50132fSMatthew G. Knepley 
1481e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerFileSetMode_ExodusII(PetscViewer viewer, PetscFileMode type)
1491e50132fSMatthew G. Knepley {
1501e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1511e50132fSMatthew G. Knepley 
1521e50132fSMatthew G. Knepley   PetscFunctionBegin;
1531e50132fSMatthew G. Knepley   exo->btype = type;
1541e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1551e50132fSMatthew G. Knepley }
1561e50132fSMatthew G. Knepley 
1571e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerFileGetMode_ExodusII(PetscViewer viewer, PetscFileMode *type)
1581e50132fSMatthew G. Knepley {
1591e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1601e50132fSMatthew G. Knepley 
1611e50132fSMatthew G. Knepley   PetscFunctionBegin;
1621e50132fSMatthew G. Knepley   *type = exo->btype;
1631e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1641e50132fSMatthew G. Knepley }
1651e50132fSMatthew G. Knepley 
1661e50132fSMatthew G. Knepley static PetscErrorCode PetscViewerExodusIIGetId_ExodusII(PetscViewer viewer, int *exoid)
1671e50132fSMatthew G. Knepley {
1681e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1691e50132fSMatthew G. Knepley 
1701e50132fSMatthew G. Knepley   PetscFunctionBegin;
1711e50132fSMatthew G. Knepley   *exoid = exo->exoid;
1721e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1731e50132fSMatthew G. Knepley }
1741e50132fSMatthew G. Knepley 
1756823f3c5SBlaise Bourdin static PetscErrorCode PetscViewerExodusIIGetOrder_ExodusII(PetscViewer viewer, PetscInt *order)
1761e50132fSMatthew G. Knepley {
1776823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1781e50132fSMatthew G. Knepley 
1791e50132fSMatthew G. Knepley   PetscFunctionBegin;
1806823f3c5SBlaise Bourdin   *order = exo->order;
1816823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
1826823f3c5SBlaise Bourdin }
1836823f3c5SBlaise Bourdin 
1846823f3c5SBlaise Bourdin static PetscErrorCode PetscViewerExodusIISetOrder_ExodusII(PetscViewer viewer, PetscInt order)
1856823f3c5SBlaise Bourdin {
1866823f3c5SBlaise Bourdin   PetscViewer_ExodusII *exo = (PetscViewer_ExodusII *) viewer->data;
1876823f3c5SBlaise Bourdin 
1886823f3c5SBlaise Bourdin   PetscFunctionBegin;
1896823f3c5SBlaise Bourdin   exo->order = order;
1901e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
1911e50132fSMatthew G. Knepley }
1921e50132fSMatthew G. Knepley 
1931e50132fSMatthew G. Knepley /*MC
19400373969SVaclav Hapla    PETSCVIEWEREXODUSII - A viewer that writes to an Exodus II file
1951e50132fSMatthew G. Knepley 
1961e50132fSMatthew G. Knepley 
19700373969SVaclav Hapla .seealso:  PetscViewerExodusIIOpen(), PetscViewerCreate(), PETSCVIEWERBINARY, PETSCVIEWERHDF5, DMView(),
1981e50132fSMatthew G. Knepley            PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType()
1991e50132fSMatthew G. Knepley 
2001e50132fSMatthew G. Knepley   Level: beginner
2011e50132fSMatthew G. Knepley M*/
2021e50132fSMatthew G. Knepley 
2031e50132fSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscViewerCreate_ExodusII(PetscViewer v)
2041e50132fSMatthew G. Knepley {
2051e50132fSMatthew G. Knepley   PetscViewer_ExodusII *exo;
2061e50132fSMatthew G. Knepley   PetscErrorCode        ierr;
2071e50132fSMatthew G. Knepley 
2081e50132fSMatthew G. Knepley   PetscFunctionBegin;
2091e50132fSMatthew G. Knepley   ierr = PetscNewLog(v,&exo);CHKERRQ(ierr);
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 
2211e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_ExodusII);CHKERRQ(ierr);
2221e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_ExodusII);CHKERRQ(ierr);
2231e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ExodusII);CHKERRQ(ierr);
2241e50132fSMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetMode_C",PetscViewerFileGetMode_ExodusII);CHKERRQ(ierr);
2256823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerGetId_C",PetscViewerExodusIIGetId_ExodusII);CHKERRQ(ierr);
2266823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerSetOrder_C",PetscViewerExodusIISetOrder_ExodusII);CHKERRQ(ierr);
2276823f3c5SBlaise Bourdin   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerGetOrder_C",PetscViewerExodusIIGetOrder_ExodusII);CHKERRQ(ierr);
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 
252cd921712SStefano Zampini .seealso: EXOGetVarIndex(),DMPlexView_ExodusII_Internal(),VecViewPlex_ExodusII_Nodal_Internal(),VecLoadNodal_PlexEXO(),VecLoadZonal_PlexEXO()
25378569bb4SMatthew G. Knepley */
2546823f3c5SBlaise Bourdin PetscErrorCode EXOGetVarIndex_Internal(int exoid, ex_entity_type obj_type, const char name[], int *varIndex)
25578569bb4SMatthew G. Knepley {
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   PetscErrorCode ierr;
26278569bb4SMatthew G. Knepley 
26378569bb4SMatthew G. Knepley   PetscFunctionBegin;
26478569bb4SMatthew G. Knepley   suffix[0] = (char *) "";
26578569bb4SMatthew G. Knepley   suffix[1] = (char *) "_X";
26678569bb4SMatthew G. Knepley   suffix[2] = (char *) "_XX";
26778569bb4SMatthew G. Knepley   suffix[3] = (char *) "_1";
26878569bb4SMatthew G. Knepley   suffix[4] = (char *) "_11";
26978569bb4SMatthew G. Knepley 
2706823f3c5SBlaise Bourdin   *varIndex = -1;
2716de834b4SMatthew G. Knepley   PetscStackCallStandard(ex_get_variable_param,(exoid, obj_type, &num_vars));
27278569bb4SMatthew G. Knepley   for (i = 0; i < num_vars; ++i) {
2736de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_variable_name,(exoid, obj_type, i+1, var_name));
27478569bb4SMatthew G. Knepley     for (j = 0; j < num_suffix; ++j){
27578569bb4SMatthew G. Knepley       ierr = PetscStrncpy(ext_name, name, MAX_STR_LENGTH);CHKERRQ(ierr);
276a126751eSBarry Smith       ierr = PetscStrlcat(ext_name, suffix[j], MAX_STR_LENGTH);CHKERRQ(ierr);
27778569bb4SMatthew G. Knepley       ierr = PetscStrcasecmp(ext_name, var_name, &flg);
27878569bb4SMatthew G. Knepley       if (flg) {
27978569bb4SMatthew G. Knepley         *varIndex = i+1;
2806823f3c5SBlaise Bourdin       }
2816823f3c5SBlaise Bourdin     }
2826823f3c5SBlaise Bourdin   }
28378569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
28478569bb4SMatthew G. Knepley }
28578569bb4SMatthew G. Knepley 
28678569bb4SMatthew G. Knepley /*
2876823f3c5SBlaise Bourdin   DMView_PlexExodusII - Write a DM to disk in exodus format
28878569bb4SMatthew G. Knepley 
28978569bb4SMatthew G. Knepley   Collective on dm
29078569bb4SMatthew G. Knepley 
29178569bb4SMatthew G. Knepley   Input Parameters:
29278569bb4SMatthew G. Knepley + dm  - The dm to be written
2936823f3c5SBlaise Bourdin . viewer - an exodusII viewer
29478569bb4SMatthew G. Knepley 
29578569bb4SMatthew G. Knepley   Notes:
29678569bb4SMatthew G. Knepley   Not all DM can be written to disk this way. For instance, exodus assume that element blocks (mapped to "Cell sets" labels)
29778569bb4SMatthew G. Knepley   consists of sequentially numbered cells. If this is not the case, the exodus file will be corrupted.
29878569bb4SMatthew G. Knepley 
2996823f3c5SBlaise Bourdin   If the dm has been distributed, only the part of the DM on MPI rank 0 (including "ghost" cells and vertices)
3006823f3c5SBlaise Bourdin   will be written.
30178569bb4SMatthew G. Knepley 
30278569bb4SMatthew G. Knepley   DMPlex only represents geometry while most post-processing software expect that a mesh also provides information
30378569bb4SMatthew G. Knepley   on the discretization space. This function assumes that the file represents Lagrange finite elements of order 1 or 2.
3046823f3c5SBlaise Bourdin   The order of the mesh shall be set using PetscViewerExodusIISetOrder
30578569bb4SMatthew G. Knepley   It should be extended to use PetscFE objects.
30678569bb4SMatthew G. Knepley 
3076823f3c5SBlaise Bourdin   This function will only handle TRI, TET, QUAD, and HEX cells.
30878569bb4SMatthew G. Knepley   Level: beginner
30978569bb4SMatthew G. Knepley 
3106823f3c5SBlaise Bourdin .seealso:
31178569bb4SMatthew G. Knepley */
3126823f3c5SBlaise Bourdin PetscErrorCode DMView_PlexExodusII(DM dm, PetscViewer viewer)
31378569bb4SMatthew G. Knepley {
31478569bb4SMatthew G. Knepley   enum ElemType {TRI, QUAD, TET, HEX};
31578569bb4SMatthew G. Knepley   MPI_Comm        comm;
3166823f3c5SBlaise Bourdin   PetscInt        degree; /* the order of the mesh */
31778569bb4SMatthew G. Knepley   /* Connectivity Variables */
31878569bb4SMatthew G. Knepley   PetscInt        cellsNotInConnectivity;
31978569bb4SMatthew G. Knepley   /* Cell Sets */
32078569bb4SMatthew G. Knepley   DMLabel         csLabel;
32178569bb4SMatthew G. Knepley   IS              csIS;
32278569bb4SMatthew G. Knepley   const PetscInt *csIdx;
32378569bb4SMatthew G. Knepley   PetscInt        num_cs, cs;
32478569bb4SMatthew G. Knepley   enum ElemType  *type;
325fe209ef9SBlaise Bourdin   PetscBool       hasLabel;
32678569bb4SMatthew G. Knepley   /* Coordinate Variables */
32778569bb4SMatthew G. Knepley   DM              cdm;
3286823f3c5SBlaise Bourdin   PetscSection    coordSection;
32978569bb4SMatthew G. Knepley   Vec             coord;
33078569bb4SMatthew G. Knepley   PetscInt      **nodes;
331eae8a387SMatthew G. Knepley   PetscInt        depth, d, dim, skipCells = 0;
33278569bb4SMatthew G. Knepley   PetscInt        pStart, pEnd, p, cStart, cEnd, numCells, vStart, vEnd, numVertices, eStart, eEnd, numEdges, fStart, fEnd, numFaces, numNodes;
33378569bb4SMatthew G. Knepley   PetscInt        num_vs, num_fs;
33478569bb4SMatthew G. Knepley   PetscMPIInt     rank, size;
33578569bb4SMatthew G. Knepley   const char     *dmName;
336fe209ef9SBlaise Bourdin   PetscInt        nodesTriP1[4]  = {3,0,0,0};
337fe209ef9SBlaise Bourdin   PetscInt        nodesTriP2[4]  = {3,3,0,0};
338fe209ef9SBlaise Bourdin   PetscInt        nodesQuadP1[4] = {4,0,0,0};
339fe209ef9SBlaise Bourdin   PetscInt        nodesQuadP2[4] = {4,4,0,1};
340fe209ef9SBlaise Bourdin   PetscInt        nodesTetP1[4]  = {4,0,0,0};
341fe209ef9SBlaise Bourdin   PetscInt        nodesTetP2[4]  = {4,6,0,0};
342fe209ef9SBlaise Bourdin   PetscInt        nodesHexP1[4]  = {8,0,0,0};
343fe209ef9SBlaise Bourdin   PetscInt        nodesHexP2[4]  = {8,12,6,1};
34478569bb4SMatthew G. Knepley   PetscErrorCode  ierr;
34578569bb4SMatthew G. Knepley 
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;
35278569bb4SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
353ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr);
354ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &size);CHKERRMPI(ierr);
3556823f3c5SBlaise Bourdin 
3566823f3c5SBlaise Bourdin   /*
3576823f3c5SBlaise Bourdin     Creating coordSection is a collective operation so we do it somewhat out of sequence
3586823f3c5SBlaise Bourdin   */
3596823f3c5SBlaise Bourdin   ierr = PetscSectionCreate(comm, &coordSection);CHKERRQ(ierr);
3606823f3c5SBlaise Bourdin   ierr = DMGetCoordinatesLocalSetUp(dm);CHKERRQ(ierr);
3616823f3c5SBlaise Bourdin   if (!rank) {
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 */
3686823f3c5SBlaise Bourdin       SETERRQ1(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);
3786823f3c5SBlaise Bourdin         if (exo->exoid < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_create failed for %s", exo->filename);
3796823f3c5SBlaise Bourdin 
3806823f3c5SBlaise Bourdin       break;
3816823f3c5SBlaise Bourdin     default:
3826823f3c5SBlaise Bourdin       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
3836823f3c5SBlaise Bourdin     }
3846823f3c5SBlaise Bourdin 
38578569bb4SMatthew G. Knepley     /* --- Get DM info --- */
38678569bb4SMatthew G. Knepley     ierr = PetscObjectGetName((PetscObject) dm, &dmName);CHKERRQ(ierr);
38778569bb4SMatthew G. Knepley     ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
38878569bb4SMatthew G. Knepley     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
38978569bb4SMatthew G. Knepley     ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
39078569bb4SMatthew G. Knepley     ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
39178569bb4SMatthew G. Knepley     ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr);
39278569bb4SMatthew G. Knepley     ierr = DMPlexGetDepthStratum(dm, 1, &eStart, &eEnd);CHKERRQ(ierr);
39378569bb4SMatthew G. Knepley     ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
39478569bb4SMatthew G. Knepley     numCells    = cEnd - cStart;
39578569bb4SMatthew G. Knepley     numEdges    = eEnd - eStart;
39678569bb4SMatthew G. Knepley     numVertices = vEnd - vStart;
39778569bb4SMatthew G. Knepley     if (depth == 3) {numFaces = fEnd - fStart;}
39878569bb4SMatthew G. Knepley     else            {numFaces = 0;}
39978569bb4SMatthew G. Knepley     ierr = DMGetLabelSize(dm, "Cell Sets", &num_cs);CHKERRQ(ierr);
40078569bb4SMatthew G. Knepley     ierr = DMGetLabelSize(dm, "Vertex Sets", &num_vs);CHKERRQ(ierr);
40178569bb4SMatthew G. Knepley     ierr = DMGetLabelSize(dm, "Face Sets", &num_fs);CHKERRQ(ierr);
40278569bb4SMatthew G. Knepley     ierr = DMGetCoordinatesLocal(dm, &coord);CHKERRQ(ierr);
40378569bb4SMatthew G. Knepley     ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
40478569bb4SMatthew G. Knepley     if (num_cs > 0) {
40578569bb4SMatthew G. Knepley       ierr = DMGetLabel(dm, "Cell Sets", &csLabel);CHKERRQ(ierr);
40678569bb4SMatthew G. Knepley       ierr = DMLabelGetValueIS(csLabel, &csIS);CHKERRQ(ierr);
40778569bb4SMatthew G. Knepley       ierr = ISGetIndices(csIS, &csIdx);CHKERRQ(ierr);
40878569bb4SMatthew G. Knepley     }
40978569bb4SMatthew G. Knepley     ierr = PetscMalloc1(num_cs, &nodes);CHKERRQ(ierr);
41078569bb4SMatthew G. Knepley     /* Set element type for each block and compute total number of nodes */
41178569bb4SMatthew G. Knepley     ierr = PetscMalloc1(num_cs, &type);CHKERRQ(ierr);
41278569bb4SMatthew G. Knepley     numNodes = numVertices;
4136823f3c5SBlaise Bourdin 
4146823f3c5SBlaise Bourdin     ierr = PetscViewerExodusIIGetOrder(viewer, &degree);CHKERRQ(ierr);
41578569bb4SMatthew G. Knepley     if (degree == 2) {numNodes += numEdges;}
41678569bb4SMatthew G. Knepley     cellsNotInConnectivity = numCells;
41778569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
41878569bb4SMatthew G. Knepley       IS              stratumIS;
41978569bb4SMatthew G. Knepley       const PetscInt *cells;
42078569bb4SMatthew G. Knepley       PetscScalar    *xyz = NULL;
42178569bb4SMatthew G. Knepley       PetscInt        csSize, closureSize;
42278569bb4SMatthew G. Knepley 
42378569bb4SMatthew G. Knepley       ierr = DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS);CHKERRQ(ierr);
42478569bb4SMatthew G. Knepley       ierr = ISGetIndices(stratumIS, &cells);CHKERRQ(ierr);
42578569bb4SMatthew G. Knepley       ierr = ISGetSize(stratumIS, &csSize);CHKERRQ(ierr);
42678569bb4SMatthew G. Knepley       ierr = DMPlexVecGetClosure(cdm, NULL, coord, cells[0], &closureSize, &xyz);CHKERRQ(ierr);
42778569bb4SMatthew G. Knepley       switch (dim) {
42878569bb4SMatthew G. Knepley         case 2:
42978569bb4SMatthew G. Knepley           if      (closureSize == 3*dim) {type[cs] = TRI;}
43078569bb4SMatthew G. Knepley           else if (closureSize == 4*dim) {type[cs] = QUAD;}
43178569bb4SMatthew G. Knepley           else SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of vertices %D in dimension %D has no ExodusII type", closureSize/dim, dim);
43278569bb4SMatthew G. Knepley           break;
43378569bb4SMatthew G. Knepley         case 3:
43478569bb4SMatthew G. Knepley           if      (closureSize == 4*dim) {type[cs] = TET;}
43578569bb4SMatthew G. Knepley           else if (closureSize == 8*dim) {type[cs] = HEX;}
43678569bb4SMatthew G. Knepley           else SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of vertices %D in dimension %D has no ExodusII type", closureSize/dim, dim);
43778569bb4SMatthew G. Knepley           break;
43878569bb4SMatthew G. Knepley         default: SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Dimension %D not handled by ExodusII viewer", dim);
43978569bb4SMatthew G. Knepley       }
44078569bb4SMatthew G. Knepley       if ((degree == 2) && (type[cs] == QUAD)) {numNodes += csSize;}
44178569bb4SMatthew G. Knepley       if ((degree == 2) && (type[cs] == HEX))  {numNodes += csSize; numNodes += numFaces;}
44278569bb4SMatthew G. Knepley       ierr = DMPlexVecRestoreClosure(cdm, NULL, coord, cells[0], &closureSize, &xyz);CHKERRQ(ierr);
44378569bb4SMatthew G. Knepley       /* Set nodes and Element type */
44478569bb4SMatthew G. Knepley       if (type[cs] == TRI) {
44578569bb4SMatthew G. Knepley         if      (degree == 1) nodes[cs] = nodesTriP1;
44678569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesTriP2;
44778569bb4SMatthew G. Knepley       } else if (type[cs] == QUAD) {
44878569bb4SMatthew G. Knepley         if      (degree == 1) nodes[cs] = nodesQuadP1;
44978569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesQuadP2;
45078569bb4SMatthew G. Knepley       } else if (type[cs] == TET) {
45178569bb4SMatthew G. Knepley         if      (degree == 1) nodes[cs] = nodesTetP1;
45278569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesTetP2;
45378569bb4SMatthew G. Knepley       } else if (type[cs] == HEX) {
45478569bb4SMatthew G. Knepley         if      (degree == 1) nodes[cs] = nodesHexP1;
45578569bb4SMatthew G. Knepley         else if (degree == 2) nodes[cs] = nodesHexP2;
45678569bb4SMatthew G. Knepley       }
45778569bb4SMatthew G. Knepley       /* Compute the number of cells not in the connectivity table */
45878569bb4SMatthew G. Knepley       cellsNotInConnectivity -= nodes[cs][3]*csSize;
45978569bb4SMatthew G. Knepley 
46078569bb4SMatthew G. Knepley       ierr = ISRestoreIndices(stratumIS, &cells);CHKERRQ(ierr);
46178569bb4SMatthew G. Knepley       ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
46278569bb4SMatthew G. Knepley     }
4636823f3c5SBlaise Bourdin     if (num_cs > 0) {PetscStackCallStandard(ex_put_init,(exo->exoid, dmName, dim, numNodes, numCells, num_cs, num_vs, num_fs));}
46478569bb4SMatthew G. Knepley     /* --- Connectivity --- */
46578569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
46678569bb4SMatthew G. Knepley       IS              stratumIS;
46778569bb4SMatthew G. Knepley       const PetscInt *cells;
468fe209ef9SBlaise Bourdin       PetscInt       *connect, off = 0;
469eae8a387SMatthew G. Knepley       PetscInt        edgesInClosure = 0, facesInClosure = 0, verticesInClosure = 0;
47078569bb4SMatthew G. Knepley       PetscInt        csSize, c, connectSize, closureSize;
47178569bb4SMatthew G. Knepley       char           *elem_type = NULL;
47278569bb4SMatthew G. Knepley       char            elem_type_tri3[]  = "TRI3",  elem_type_quad4[] = "QUAD4";
47378569bb4SMatthew G. Knepley       char            elem_type_tri6[]  = "TRI6",  elem_type_quad9[] = "QUAD9";
47478569bb4SMatthew G. Knepley       char            elem_type_tet4[]  = "TET4",  elem_type_hex8[]  = "HEX8";
47578569bb4SMatthew G. Knepley       char            elem_type_tet10[] = "TET10", elem_type_hex27[] = "HEX27";
47678569bb4SMatthew G. Knepley 
47778569bb4SMatthew G. Knepley       ierr = DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS);CHKERRQ(ierr);
47878569bb4SMatthew G. Knepley       ierr = ISGetIndices(stratumIS, &cells);CHKERRQ(ierr);
47978569bb4SMatthew G. Knepley       ierr = ISGetSize(stratumIS, &csSize);CHKERRQ(ierr);
48078569bb4SMatthew G. Knepley       /* Set Element type */
48178569bb4SMatthew G. Knepley       if (type[cs] == TRI) {
48278569bb4SMatthew G. Knepley         if      (degree == 1) elem_type = elem_type_tri3;
48378569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_tri6;
48478569bb4SMatthew G. Knepley       } else if (type[cs] == QUAD) {
48578569bb4SMatthew G. Knepley         if      (degree == 1) elem_type = elem_type_quad4;
48678569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_quad9;
48778569bb4SMatthew G. Knepley       } else if (type[cs] == TET) {
48878569bb4SMatthew G. Knepley         if      (degree == 1) elem_type = elem_type_tet4;
48978569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_tet10;
49078569bb4SMatthew G. Knepley       } else if (type[cs] == HEX) {
49178569bb4SMatthew G. Knepley         if      (degree == 1) elem_type = elem_type_hex8;
49278569bb4SMatthew G. Knepley         else if (degree == 2) elem_type = elem_type_hex27;
49378569bb4SMatthew G. Knepley       }
49478569bb4SMatthew G. Knepley       connectSize = nodes[cs][0] + nodes[cs][1] + nodes[cs][2] + nodes[cs][3];
495fe209ef9SBlaise Bourdin       ierr = PetscMalloc1(PetscMax(27,connectSize)*csSize, &connect);CHKERRQ(ierr);
4966823f3c5SBlaise Bourdin       PetscStackCallStandard(ex_put_block,(exo->exoid, EX_ELEM_BLOCK, csIdx[cs], elem_type, csSize, connectSize, 0, 0, 1));
49778569bb4SMatthew G. Knepley       /* Find number of vertices, edges, and faces in the closure */
49878569bb4SMatthew G. Knepley       verticesInClosure = nodes[cs][0];
49978569bb4SMatthew G. Knepley       if (depth > 1) {
50078569bb4SMatthew G. Knepley         if (dim == 2) {
50178569bb4SMatthew G. Knepley           ierr = DMPlexGetConeSize(dm, cells[0], &edgesInClosure);CHKERRQ(ierr);
50278569bb4SMatthew G. Knepley         } else if (dim == 3) {
50378569bb4SMatthew G. Knepley           PetscInt *closure = NULL;
50478569bb4SMatthew G. Knepley 
50578569bb4SMatthew G. Knepley           ierr = DMPlexGetConeSize(dm, cells[0], &facesInClosure);CHKERRQ(ierr);
50678569bb4SMatthew G. Knepley           ierr = DMPlexGetTransitiveClosure(dm, cells[0], PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
50778569bb4SMatthew G. Knepley           edgesInClosure = closureSize - facesInClosure - 1 - verticesInClosure;
50878569bb4SMatthew G. Knepley           ierr = DMPlexRestoreTransitiveClosure(dm, cells[0], PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
50978569bb4SMatthew G. Knepley         }
51078569bb4SMatthew G. Knepley       }
51178569bb4SMatthew G. Knepley       /* Get connectivity for each cell */
51278569bb4SMatthew G. Knepley       for (c = 0; c < csSize; ++c) {
51378569bb4SMatthew G. Knepley         PetscInt *closure = NULL;
51478569bb4SMatthew G. Knepley         PetscInt  temp, i;
51578569bb4SMatthew G. Knepley 
51678569bb4SMatthew G. Knepley         ierr = DMPlexGetTransitiveClosure(dm, cells[c], PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
51778569bb4SMatthew G. Knepley         for (i = 0; i < connectSize; ++i) {
51878569bb4SMatthew G. Knepley           if (i < nodes[cs][0]) {/* Vertices */
519fe209ef9SBlaise Bourdin             connect[i+off] = closure[(i+edgesInClosure+facesInClosure+1)*2] + 1;
520fe209ef9SBlaise Bourdin             connect[i+off] -= cellsNotInConnectivity;
52178569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0]+nodes[cs][1]) { /* Edges */
522fe209ef9SBlaise Bourdin             connect[i+off] = closure[(i-verticesInClosure+facesInClosure+1)*2] + 1;
523fe209ef9SBlaise Bourdin             if (nodes[cs][2] == 0) connect[i+off] -= numFaces;
524fe209ef9SBlaise Bourdin             connect[i+off] -= cellsNotInConnectivity;
52578569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0]+nodes[cs][1]+nodes[cs][3]){ /* Cells */
526fe209ef9SBlaise Bourdin             connect[i+off] = closure[0] + 1;
527fe209ef9SBlaise Bourdin             connect[i+off] -= skipCells;
52878569bb4SMatthew G. Knepley           } else if (i < nodes[cs][0]+nodes[cs][1]+nodes[cs][3]+nodes[cs][2]){ /* Faces */
529fe209ef9SBlaise Bourdin             connect[i+off] = closure[(i-edgesInClosure-verticesInClosure)*2] + 1;
530fe209ef9SBlaise Bourdin             connect[i+off] -= cellsNotInConnectivity;
53178569bb4SMatthew G. Knepley           } else {
532fe209ef9SBlaise Bourdin             connect[i+off] = -1;
53378569bb4SMatthew G. Knepley           }
53478569bb4SMatthew G. Knepley         }
53578569bb4SMatthew G. Knepley         /* Tetrahedra are inverted */
53678569bb4SMatthew G. Knepley         if (type[cs] == TET) {
537fe209ef9SBlaise Bourdin           temp = connect[0+off]; connect[0+off] = connect[1+off]; connect[1+off] = temp;
53878569bb4SMatthew G. Knepley           if (degree == 2) {
539e2c9586dSBlaise Bourdin             temp = connect[5+off]; connect[5+off] = connect[6+off]; connect[6+off] = temp;
540e2c9586dSBlaise Bourdin             temp = connect[7+off]; connect[7+off] = connect[8+off]; connect[8+off] = temp;
54178569bb4SMatthew G. Knepley           }
54278569bb4SMatthew G. Knepley         }
54378569bb4SMatthew G. Knepley         /* Hexahedra are inverted */
54478569bb4SMatthew G. Knepley         if (type[cs] == HEX) {
545fe209ef9SBlaise Bourdin           temp = connect[1+off]; connect[1+off] = connect[3+off]; connect[3+off] = temp;
54678569bb4SMatthew G. Knepley           if (degree == 2) {
547fe209ef9SBlaise Bourdin             temp = connect[8+off];  connect[8+off]  = connect[11+off]; connect[11+off] = temp;
548fe209ef9SBlaise Bourdin             temp = connect[9+off];  connect[9+off]  = connect[10+off]; connect[10+off] = temp;
549fe209ef9SBlaise Bourdin             temp = connect[16+off]; connect[16+off] = connect[17+off]; connect[17+off] = temp;
550fe209ef9SBlaise Bourdin             temp = connect[18+off]; connect[18+off] = connect[19+off]; connect[19+off] = temp;
55178569bb4SMatthew G. Knepley 
552fe209ef9SBlaise Bourdin             temp = connect[12+off]; connect[12+off] = connect[16+off]; connect[16+off] = temp;
553fe209ef9SBlaise Bourdin             temp = connect[13+off]; connect[13+off] = connect[17+off]; connect[17+off] = temp;
554fe209ef9SBlaise Bourdin             temp = connect[14+off]; connect[14+off] = connect[18+off]; connect[18+off] = temp;
555fe209ef9SBlaise Bourdin             temp = connect[15+off]; connect[15+off] = connect[19+off]; connect[19+off] = temp;
55678569bb4SMatthew G. Knepley 
557fe209ef9SBlaise Bourdin             temp = connect[23+off]; connect[23+off] = connect[26+off]; connect[26+off] = temp;
558fe209ef9SBlaise Bourdin             temp = connect[24+off]; connect[24+off] = connect[25+off]; connect[25+off] = temp;
559fe209ef9SBlaise Bourdin             temp = connect[25+off]; connect[25+off] = connect[26+off]; connect[26+off] = temp;
56078569bb4SMatthew G. Knepley           }
56178569bb4SMatthew G. Knepley         }
562fe209ef9SBlaise Bourdin         off += connectSize;
56378569bb4SMatthew G. Knepley         ierr = DMPlexRestoreTransitiveClosure(dm, cells[c], PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
56478569bb4SMatthew G. Knepley       }
5656823f3c5SBlaise Bourdin       PetscStackCallStandard(ex_put_conn,(exo->exoid, EX_ELEM_BLOCK, csIdx[cs], connect, 0, 0));
56678569bb4SMatthew G. Knepley       skipCells += (nodes[cs][3] == 0)*csSize;
56778569bb4SMatthew G. Knepley       ierr = PetscFree(connect);CHKERRQ(ierr);
56878569bb4SMatthew G. Knepley       ierr = ISRestoreIndices(stratumIS, &cells);CHKERRQ(ierr);
56978569bb4SMatthew G. Knepley       ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
57078569bb4SMatthew G. Knepley     }
57178569bb4SMatthew G. Knepley     ierr = PetscFree(type);CHKERRQ(ierr);
57278569bb4SMatthew G. Knepley     /* --- Coordinates --- */
5736823f3c5SBlaise Bourdin     ierr = PetscSectionSetChart(coordSection, pStart, pEnd);CHKERRQ(ierr);
5741e50132fSMatthew G. Knepley     if (num_cs) {
57578569bb4SMatthew G. Knepley       for (d = 0; d < depth; ++d) {
57678569bb4SMatthew G. Knepley         ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
57778569bb4SMatthew G. Knepley         for (p = pStart; p < pEnd; ++p) {
5786823f3c5SBlaise Bourdin           ierr = PetscSectionSetDof(coordSection, p, nodes[0][d] > 0);CHKERRQ(ierr);
57978569bb4SMatthew G. Knepley         }
58078569bb4SMatthew G. Knepley       }
5811e50132fSMatthew G. Knepley     }
58278569bb4SMatthew G. Knepley     for (cs = 0; cs < num_cs; ++cs) {
58378569bb4SMatthew G. Knepley       IS              stratumIS;
58478569bb4SMatthew G. Knepley       const PetscInt *cells;
58578569bb4SMatthew G. Knepley       PetscInt        csSize, c;
58678569bb4SMatthew G. Knepley 
58778569bb4SMatthew G. Knepley       ierr = DMLabelGetStratumIS(csLabel, csIdx[cs], &stratumIS);CHKERRQ(ierr);
58878569bb4SMatthew G. Knepley       ierr = ISGetIndices(stratumIS, &cells);CHKERRQ(ierr);
58978569bb4SMatthew G. Knepley       ierr = ISGetSize(stratumIS, &csSize);CHKERRQ(ierr);
59078569bb4SMatthew G. Knepley       for (c = 0; c < csSize; ++c) {
5916823f3c5SBlaise Bourdin         ierr = PetscSectionSetDof(coordSection, cells[c], nodes[cs][3] > 0);CHKERRQ(ierr);
59278569bb4SMatthew G. Knepley       }
59378569bb4SMatthew G. Knepley       ierr = ISRestoreIndices(stratumIS, &cells);CHKERRQ(ierr);
59478569bb4SMatthew G. Knepley       ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
59578569bb4SMatthew G. Knepley     }
59678569bb4SMatthew G. Knepley     if (num_cs > 0) {
59778569bb4SMatthew G. Knepley       ierr = ISRestoreIndices(csIS, &csIdx);CHKERRQ(ierr);
59878569bb4SMatthew G. Knepley       ierr = ISDestroy(&csIS);CHKERRQ(ierr);
59978569bb4SMatthew G. Knepley     }
60078569bb4SMatthew G. Knepley     ierr = PetscFree(nodes);CHKERRQ(ierr);
6016823f3c5SBlaise Bourdin     ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
60278569bb4SMatthew G. Knepley     if (numNodes > 0) {
60378569bb4SMatthew G. Knepley       const char  *coordNames[3] = {"x", "y", "z"};
60478569bb4SMatthew G. Knepley       PetscScalar *coords, *closure;
60578569bb4SMatthew G. Knepley       PetscReal   *cval;
60678569bb4SMatthew G. Knepley       PetscInt     hasDof, n = 0;
60778569bb4SMatthew G. Knepley 
6086823f3c5SBlaise Bourdin       /* There can't be more than 24 values in the closure of a point for the coord coordSection */
6096de834b4SMatthew G. Knepley       ierr = PetscCalloc3(numNodes*3, &coords, dim, &cval, 24, &closure);CHKERRQ(ierr);
6106823f3c5SBlaise Bourdin       ierr = DMGetCoordinatesLocalNoncollective(dm, &coord);CHKERRQ(ierr);
61178569bb4SMatthew G. Knepley       ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
61278569bb4SMatthew G. Knepley       for (p = pStart; p < pEnd; ++p) {
6136823f3c5SBlaise Bourdin         ierr = PetscSectionGetDof(coordSection, p, &hasDof);
61478569bb4SMatthew G. Knepley         if (hasDof) {
61578569bb4SMatthew G. Knepley           PetscInt closureSize = 24, j;
61678569bb4SMatthew G. Knepley 
61778569bb4SMatthew G. Knepley           ierr = DMPlexVecGetClosure(cdm, NULL, coord, p, &closureSize, &closure);CHKERRQ(ierr);
61878569bb4SMatthew G. Knepley           for (d = 0; d < dim; ++d) {
61978569bb4SMatthew G. Knepley             cval[d] = 0.0;
62078569bb4SMatthew G. Knepley             for (j = 0; j < closureSize/dim; j++) cval[d] += closure[j*dim+d];
62178569bb4SMatthew G. Knepley             coords[d*numNodes+n] = cval[d] * dim / closureSize;
62278569bb4SMatthew G. Knepley           }
62378569bb4SMatthew G. Knepley           ++n;
62478569bb4SMatthew G. Knepley         }
62578569bb4SMatthew G. Knepley       }
6266823f3c5SBlaise Bourdin       PetscStackCallStandard(ex_put_coord,(exo->exoid, &coords[0*numNodes], &coords[1*numNodes], &coords[2*numNodes]));
62778569bb4SMatthew G. Knepley       ierr = PetscFree3(coords, cval, closure);CHKERRQ(ierr);
6286823f3c5SBlaise Bourdin       PetscStackCallStandard(ex_put_coord_names,(exo->exoid, (char **) coordNames));
62978569bb4SMatthew G. Knepley     }
6306823f3c5SBlaise Bourdin 
631fe209ef9SBlaise Bourdin     /* --- Node Sets/Vertex Sets --- */
632fe209ef9SBlaise Bourdin     DMHasLabel(dm, "Vertex Sets", &hasLabel);
633fe209ef9SBlaise Bourdin     if (hasLabel) {
634fe209ef9SBlaise Bourdin       PetscInt        i, vs, vsSize;
635fe209ef9SBlaise Bourdin       const PetscInt *vsIdx, *vertices;
636fe209ef9SBlaise Bourdin       PetscInt       *nodeList;
637fe209ef9SBlaise Bourdin       IS              vsIS, stratumIS;
638fe209ef9SBlaise Bourdin       DMLabel         vsLabel;
639fe209ef9SBlaise Bourdin       ierr = DMGetLabel(dm, "Vertex Sets", &vsLabel);CHKERRQ(ierr);
640fe209ef9SBlaise Bourdin       ierr = DMLabelGetValueIS(vsLabel, &vsIS);CHKERRQ(ierr);
641fe209ef9SBlaise Bourdin       ierr = ISGetIndices(vsIS, &vsIdx);CHKERRQ(ierr);
642fe209ef9SBlaise Bourdin       for (vs=0; vs<num_vs; ++vs) {
643fe209ef9SBlaise Bourdin         ierr = DMLabelGetStratumIS(vsLabel, vsIdx[vs], &stratumIS);CHKERRQ(ierr);
644fe209ef9SBlaise Bourdin         ierr = ISGetIndices(stratumIS, &vertices);CHKERRQ(ierr);
645fe209ef9SBlaise Bourdin         ierr = ISGetSize(stratumIS, &vsSize);CHKERRQ(ierr);
646fe209ef9SBlaise Bourdin         ierr = PetscMalloc1(vsSize, &nodeList);
647fe209ef9SBlaise Bourdin         for (i=0; i<vsSize; ++i) {
648fe209ef9SBlaise Bourdin           nodeList[i] = vertices[i] - skipCells + 1;
649fe209ef9SBlaise Bourdin         }
6506823f3c5SBlaise Bourdin         PetscStackCallStandard(ex_put_set_param,(exo->exoid, EX_NODE_SET, vsIdx[vs], vsSize, 0));
6516823f3c5SBlaise Bourdin         PetscStackCallStandard(ex_put_set,(exo->exoid, EX_NODE_SET, vsIdx[vs], nodeList, NULL));
652fe209ef9SBlaise Bourdin         ierr = ISRestoreIndices(stratumIS, &vertices);CHKERRQ(ierr);
653fe209ef9SBlaise Bourdin         ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
654fe209ef9SBlaise Bourdin         ierr = PetscFree(nodeList);CHKERRQ(ierr);
655fe209ef9SBlaise Bourdin       }
656fe209ef9SBlaise Bourdin       ierr = ISRestoreIndices(vsIS, &vsIdx);CHKERRQ(ierr);
657fe209ef9SBlaise Bourdin       ierr = ISDestroy(&vsIS);CHKERRQ(ierr);
658fe209ef9SBlaise Bourdin     }
659fe209ef9SBlaise Bourdin     /* --- Side Sets/Face Sets --- */
660fe209ef9SBlaise Bourdin     ierr = DMHasLabel(dm, "Face Sets", &hasLabel);CHKERRQ(ierr);
661fe209ef9SBlaise Bourdin     if (hasLabel) {
662fe209ef9SBlaise Bourdin       PetscInt        i, j, fs, fsSize;
663fe209ef9SBlaise Bourdin       const PetscInt *fsIdx, *faces;
664fe209ef9SBlaise Bourdin       IS              fsIS, stratumIS;
665fe209ef9SBlaise Bourdin       DMLabel         fsLabel;
666fe209ef9SBlaise Bourdin       PetscInt        numPoints, *points;
667fe209ef9SBlaise Bourdin       PetscInt        elem_list_size = 0;
668fe209ef9SBlaise Bourdin       PetscInt       *elem_list, *elem_ind, *side_list;
669fe209ef9SBlaise Bourdin 
670fe209ef9SBlaise Bourdin       ierr = DMGetLabel(dm, "Face Sets", &fsLabel);CHKERRQ(ierr);
671fe209ef9SBlaise Bourdin       /* Compute size of Node List and Element List */
672fe209ef9SBlaise Bourdin       ierr = DMLabelGetValueIS(fsLabel, &fsIS);CHKERRQ(ierr);
673fe209ef9SBlaise Bourdin       ierr = ISGetIndices(fsIS, &fsIdx);CHKERRQ(ierr);
674fe209ef9SBlaise Bourdin       for (fs=0; fs<num_fs; ++fs) {
675feb237baSPierre Jolivet         ierr = DMLabelGetStratumIS(fsLabel, fsIdx[fs], &stratumIS);CHKERRQ(ierr);
676fe209ef9SBlaise Bourdin         ierr = ISGetSize(stratumIS, &fsSize);CHKERRQ(ierr);
677fe209ef9SBlaise Bourdin         elem_list_size += fsSize;
678fe209ef9SBlaise Bourdin         ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
679fe209ef9SBlaise Bourdin       }
680fe209ef9SBlaise Bourdin       ierr = PetscMalloc3(num_fs, &elem_ind,
681fe209ef9SBlaise Bourdin                           elem_list_size, &elem_list,
682fe209ef9SBlaise Bourdin                           elem_list_size, &side_list);CHKERRQ(ierr);
683fe209ef9SBlaise Bourdin       elem_ind[0] = 0;
684fe209ef9SBlaise Bourdin       for (fs=0; fs<num_fs; ++fs) {
685fe209ef9SBlaise Bourdin         ierr = DMLabelGetStratumIS(fsLabel, fsIdx[fs], &stratumIS);CHKERRQ(ierr);
686fe209ef9SBlaise Bourdin         ierr = ISGetIndices(stratumIS, &faces);CHKERRQ(ierr);
687fe209ef9SBlaise Bourdin         ierr = ISGetSize(stratumIS, &fsSize);CHKERRQ(ierr);
688fe209ef9SBlaise Bourdin         /* Set Parameters */
6896823f3c5SBlaise Bourdin         PetscStackCallStandard(ex_put_set_param,(exo->exoid, EX_SIDE_SET, fsIdx[fs], fsSize, 0));
690fe209ef9SBlaise Bourdin         /* Indices */
691fe209ef9SBlaise Bourdin         if (fs<num_fs-1) {
692fe209ef9SBlaise Bourdin           elem_ind[fs+1] = elem_ind[fs] + fsSize;
693fe209ef9SBlaise Bourdin         }
694fe209ef9SBlaise Bourdin 
695fe209ef9SBlaise Bourdin         for (i=0; i<fsSize; ++i) {
696fe209ef9SBlaise Bourdin           /* Element List */
697fe209ef9SBlaise Bourdin           points = NULL;
698fe209ef9SBlaise Bourdin           ierr = DMPlexGetTransitiveClosure(dm, faces[i], PETSC_FALSE, &numPoints, &points);CHKERRQ(ierr);
699fe209ef9SBlaise Bourdin           elem_list[elem_ind[fs] + i] = points[2] +1;
700fe209ef9SBlaise Bourdin           ierr = DMPlexRestoreTransitiveClosure(dm, faces[i], PETSC_FALSE, &numPoints, &points);CHKERRQ(ierr);
701fe209ef9SBlaise Bourdin 
702fe209ef9SBlaise Bourdin           /* Side List */
703fe209ef9SBlaise Bourdin           points = NULL;
704fe209ef9SBlaise Bourdin           ierr = DMPlexGetTransitiveClosure(dm, elem_list[elem_ind[fs] + i]-1, PETSC_TRUE, &numPoints, &points);CHKERRQ(ierr);
705fe209ef9SBlaise Bourdin           for (j=1; j<numPoints; ++j) {
706fe209ef9SBlaise Bourdin             if (points[j*2]==faces[i]) {break;}
707fe209ef9SBlaise Bourdin           }
708fe209ef9SBlaise Bourdin           /* Convert HEX sides */
709fe209ef9SBlaise Bourdin           if (numPoints == 27) {
710fe209ef9SBlaise Bourdin             if      (j == 1) {j = 5;}
711fe209ef9SBlaise Bourdin             else if (j == 2) {j = 6;}
712fe209ef9SBlaise Bourdin             else if (j == 3) {j = 1;}
713fe209ef9SBlaise Bourdin             else if (j == 4) {j = 3;}
714fe209ef9SBlaise Bourdin             else if (j == 5) {j = 2;}
715fe209ef9SBlaise Bourdin             else if (j == 6) {j = 4;}
716fe209ef9SBlaise Bourdin           }
717fe209ef9SBlaise Bourdin           /* Convert TET sides */
718fe209ef9SBlaise Bourdin           if (numPoints == 15) {
719fe209ef9SBlaise Bourdin             --j;
720fe209ef9SBlaise Bourdin             if (j == 0) {j = 4;}
721fe209ef9SBlaise Bourdin           }
722fe209ef9SBlaise Bourdin           side_list[elem_ind[fs] + i] = j;
723fe209ef9SBlaise Bourdin           ierr = DMPlexRestoreTransitiveClosure(dm, elem_list[elem_ind[fs] + i]-1, PETSC_TRUE, &numPoints, &points);CHKERRQ(ierr);
724fe209ef9SBlaise Bourdin 
725fe209ef9SBlaise Bourdin         }
726fe209ef9SBlaise Bourdin         ierr = ISRestoreIndices(stratumIS, &faces);CHKERRQ(ierr);
727fe209ef9SBlaise Bourdin         ierr = ISDestroy(&stratumIS);CHKERRQ(ierr);
728fe209ef9SBlaise Bourdin       }
729fe209ef9SBlaise Bourdin       ierr = ISRestoreIndices(fsIS, &fsIdx);CHKERRQ(ierr);
730fe209ef9SBlaise Bourdin       ierr = ISDestroy(&fsIS);CHKERRQ(ierr);
731fe209ef9SBlaise Bourdin 
732fe209ef9SBlaise Bourdin       /* Put side sets */
733fe209ef9SBlaise Bourdin       for (fs=0; fs<num_fs; ++fs) {
7346823f3c5SBlaise Bourdin         PetscStackCallStandard(ex_put_set,(exo->exoid, EX_SIDE_SET, fsIdx[fs], &elem_list[elem_ind[fs]], &side_list[elem_ind[fs]]));
735fe209ef9SBlaise Bourdin       }
736fe209ef9SBlaise Bourdin       ierr = PetscFree3(elem_ind,elem_list,side_list);CHKERRQ(ierr);
737fe209ef9SBlaise Bourdin     }
7386823f3c5SBlaise Bourdin     /*
7396823f3c5SBlaise Bourdin       close the exodus file
7406823f3c5SBlaise Bourdin     */
7416823f3c5SBlaise Bourdin     ex_close(exo->exoid);
7426823f3c5SBlaise Bourdin     exo->exoid = -1;
7436823f3c5SBlaise Bourdin   }
7446823f3c5SBlaise Bourdin   ierr = PetscSectionDestroy(&coordSection);CHKERRQ(ierr);
7456823f3c5SBlaise Bourdin 
7466823f3c5SBlaise Bourdin   /*
7476823f3c5SBlaise Bourdin     reopen the file in parallel
7486823f3c5SBlaise Bourdin   */
7496823f3c5SBlaise Bourdin   EXO_mode = EX_WRITE;
7506823f3c5SBlaise Bourdin #if defined(PETSC_USE_64BIT_INDICES)
7516823f3c5SBlaise Bourdin   EXO_mode += EX_ALL_INT64_API;
7526823f3c5SBlaise Bourdin #endif
7536823f3c5SBlaise Bourdin   CPU_word_size = sizeof(PetscReal);
7546823f3c5SBlaise Bourdin   IO_word_size  = sizeof(PetscReal);
7556823f3c5SBlaise Bourdin   exo->exoid = ex_open_par(exo->filename,EXO_mode,&CPU_word_size,&IO_word_size,&EXO_version,comm,MPI_INFO_NULL);
7566823f3c5SBlaise Bourdin   if (exo->exoid < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open_par failed for %s", exo->filename);
7576823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
7586823f3c5SBlaise Bourdin }
7596823f3c5SBlaise Bourdin 
7606823f3c5SBlaise Bourdin /*
7616823f3c5SBlaise Bourdin   VecView_PlexExodusII_Internal - Write a Vec corresponding in an exodus file
7626823f3c5SBlaise Bourdin 
7636823f3c5SBlaise Bourdin   Collective on v
7646823f3c5SBlaise Bourdin 
7656823f3c5SBlaise Bourdin   Input Parameters:
7666823f3c5SBlaise Bourdin + v  - The vector to be written
7676823f3c5SBlaise Bourdin - viewer - The PetscViewerExodusII viewer associate to an exodus file
7686823f3c5SBlaise Bourdin 
7696823f3c5SBlaise Bourdin   Notes:
7706823f3c5SBlaise Bourdin   The exodus result variable index is obtained by comparing the Vec name and the
7716823f3c5SBlaise Bourdin   names of variables declared in the exodus file. For instance for a Vec named "V"
7726823f3c5SBlaise Bourdin   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
7736823f3c5SBlaise Bourdin   amongst all variables.
7746823f3c5SBlaise Bourdin   In the event where a nodal and zonal variable both match, the function will return an error instead of
7756823f3c5SBlaise Bourdin   possibly corrupting the file
7766823f3c5SBlaise Bourdin 
7776823f3c5SBlaise Bourdin   Level: beginner
7786823f3c5SBlaise Bourdin 
7796823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(),DMPlexView_ExodusII(),VecView_PlexExodusII()
7806823f3c5SBlaise Bourdin @*/
7816823f3c5SBlaise Bourdin PetscErrorCode VecView_PlexExodusII_Internal(Vec v, PetscViewer viewer)
7826823f3c5SBlaise Bourdin {
7836823f3c5SBlaise Bourdin   DM                 dm;
7846823f3c5SBlaise Bourdin   MPI_Comm           comm;
7856823f3c5SBlaise Bourdin   PetscMPIInt        rank;
7866823f3c5SBlaise Bourdin 
7876823f3c5SBlaise Bourdin 
7886823f3c5SBlaise Bourdin   int                exoid,offsetN = 0, offsetZ = 0;
7896823f3c5SBlaise Bourdin   const char        *vecname;
7906823f3c5SBlaise Bourdin   PetscInt           step;
7916823f3c5SBlaise Bourdin   PetscErrorCode     ierr;
7926823f3c5SBlaise Bourdin 
7936823f3c5SBlaise Bourdin   PetscFunctionBegin;
7946823f3c5SBlaise Bourdin   ierr = PetscObjectGetComm((PetscObject) v, &comm);CHKERRQ(ierr);
7956823f3c5SBlaise Bourdin   ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr);
7966823f3c5SBlaise Bourdin   ierr = PetscViewerExodusIIGetId(viewer,&exoid);CHKERRQ(ierr);
7976823f3c5SBlaise Bourdin   ierr = VecGetDM(v, &dm);CHKERRQ(ierr);
7986823f3c5SBlaise Bourdin   ierr = PetscObjectGetName((PetscObject) v, &vecname);CHKERRQ(ierr);
7996823f3c5SBlaise Bourdin 
8006823f3c5SBlaise Bourdin   ierr = DMGetOutputSequenceNumber(dm,&step,NULL);CHKERRQ(ierr);
8016823f3c5SBlaise Bourdin   ierr = EXOGetVarIndex_Internal(exoid,EX_NODAL,vecname,&offsetN);CHKERRQ(ierr);
8026823f3c5SBlaise Bourdin   ierr = EXOGetVarIndex_Internal(exoid,EX_ELEM_BLOCK,vecname,&offsetZ);CHKERRQ(ierr);
8036823f3c5SBlaise Bourdin   if (offsetN <= 0 && offsetZ <= 0) SETERRQ1(comm, PETSC_ERR_FILE_UNEXPECTED, "Found both nodal and zonal variable %s in exodus file. \n", vecname);
8046823f3c5SBlaise Bourdin   if (offsetN > 0) {
8056823f3c5SBlaise Bourdin     ierr = VecViewPlex_ExodusII_Nodal_Internal(v,exoid,(int) step+1,offsetN);CHKERRQ(ierr);
8066823f3c5SBlaise Bourdin   } else if (offsetZ > 0) {
8076823f3c5SBlaise Bourdin     ierr = VecViewPlex_ExodusII_Zonal_Internal(v,exoid,(int) step+1,offsetZ);CHKERRQ(ierr);
8086823f3c5SBlaise Bourdin   } else SETERRQ1(comm, PETSC_ERR_FILE_UNEXPECTED, "Could not find nodal or zonal variable %s in exodus file. \n", vecname);
8096823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
8106823f3c5SBlaise Bourdin }
8116823f3c5SBlaise Bourdin 
8126823f3c5SBlaise Bourdin /*
8136823f3c5SBlaise Bourdin   VecLoad_PlexExodusII_Internal - Write a Vec corresponding in an exodus file
8146823f3c5SBlaise Bourdin 
8156823f3c5SBlaise Bourdin   Collective on v
8166823f3c5SBlaise Bourdin 
8176823f3c5SBlaise Bourdin   Input Parameters:
8186823f3c5SBlaise Bourdin + v  - The vector to be written
8196823f3c5SBlaise Bourdin - viewer - The PetscViewerExodusII viewer associate to an exodus file
8206823f3c5SBlaise Bourdin 
8216823f3c5SBlaise Bourdin   Notes:
8226823f3c5SBlaise Bourdin   The exodus result variable index is obtained by comparing the Vec name and the
8236823f3c5SBlaise Bourdin   names of variables declared in the exodus file. For instance for a Vec named "V"
8246823f3c5SBlaise Bourdin   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
8256823f3c5SBlaise Bourdin   amongst all variables.
8266823f3c5SBlaise Bourdin   In the event where a nodal and zonal variable both match, the function will return an error instead of
8276823f3c5SBlaise Bourdin   possibly corrupting the file
8286823f3c5SBlaise Bourdin 
8296823f3c5SBlaise Bourdin   Level: beginner
8306823f3c5SBlaise Bourdin 
8316823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(),DMPlexView_ExodusII(),VecView_PlexExodusII()
8326823f3c5SBlaise Bourdin @*/
8336823f3c5SBlaise Bourdin PetscErrorCode VecLoad_PlexExodusII_Internal(Vec v, PetscViewer viewer)
8346823f3c5SBlaise Bourdin {
8356823f3c5SBlaise Bourdin   DM                 dm;
8366823f3c5SBlaise Bourdin   MPI_Comm           comm;
8376823f3c5SBlaise Bourdin   PetscMPIInt        rank;
8386823f3c5SBlaise Bourdin 
8396823f3c5SBlaise Bourdin 
8406823f3c5SBlaise Bourdin   int                exoid,offsetN = 0, offsetZ = 0;
8416823f3c5SBlaise Bourdin   const char        *vecname;
8426823f3c5SBlaise Bourdin   PetscInt           step;
8436823f3c5SBlaise Bourdin   PetscErrorCode     ierr;
8446823f3c5SBlaise Bourdin 
8456823f3c5SBlaise Bourdin   PetscFunctionBegin;
8466823f3c5SBlaise Bourdin   ierr = PetscObjectGetComm((PetscObject) v, &comm);CHKERRQ(ierr);
8476823f3c5SBlaise Bourdin   ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr);
8486823f3c5SBlaise Bourdin   ierr = PetscViewerExodusIIGetId(viewer,&exoid);CHKERRQ(ierr);
8496823f3c5SBlaise Bourdin   ierr = VecGetDM(v, &dm);CHKERRQ(ierr);
8506823f3c5SBlaise Bourdin   ierr = PetscObjectGetName((PetscObject) v, &vecname);CHKERRQ(ierr);
8516823f3c5SBlaise Bourdin 
8526823f3c5SBlaise Bourdin   ierr = DMGetOutputSequenceNumber(dm,&step,NULL);CHKERRQ(ierr);
8536823f3c5SBlaise Bourdin   ierr = EXOGetVarIndex_Internal(exoid,EX_NODAL,vecname,&offsetN);CHKERRQ(ierr);
8546823f3c5SBlaise Bourdin   ierr = EXOGetVarIndex_Internal(exoid,EX_ELEM_BLOCK,vecname,&offsetZ);CHKERRQ(ierr);
8556823f3c5SBlaise Bourdin   if (offsetN <= 0 && offsetZ <= 0) SETERRQ1(comm, PETSC_ERR_FILE_UNEXPECTED, "Found both nodal and zonal variable %s in exodus file. \n", vecname);
8566823f3c5SBlaise Bourdin   if (offsetN > 0) {
8576823f3c5SBlaise Bourdin     ierr = VecLoadPlex_ExodusII_Nodal_Internal(v,exoid,(int) step+1,offsetN);CHKERRQ(ierr);
8586823f3c5SBlaise Bourdin   } else if (offsetZ > 0) {
8596823f3c5SBlaise Bourdin     ierr = VecLoadPlex_ExodusII_Zonal_Internal(v,exoid,(int) step+1,offsetZ);CHKERRQ(ierr);
8606823f3c5SBlaise Bourdin   } else SETERRQ1(comm, PETSC_ERR_FILE_UNEXPECTED, "Could not find nodal or zonal variable %s in exodus file. \n", vecname);
86178569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
86278569bb4SMatthew G. Knepley }
86378569bb4SMatthew G. Knepley 
86478569bb4SMatthew G. Knepley /*
86578569bb4SMatthew G. Knepley   VecViewPlex_ExodusII_Nodal_Internal - Write a Vec corresponding to a nodal field to an exodus file
86678569bb4SMatthew G. Knepley 
86778569bb4SMatthew G. Knepley   Collective on v
86878569bb4SMatthew G. Knepley 
86978569bb4SMatthew G. Knepley   Input Parameters:
87078569bb4SMatthew G. Knepley + v  - The vector to be written
87178569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
8726823f3c5SBlaise Bourdin . step - the time step to write at (exodus steps are numbered starting from 1)
8736823f3c5SBlaise Bourdin - offset - the location of the variable in the file
87478569bb4SMatthew G. Knepley 
87578569bb4SMatthew G. Knepley   Notes:
87678569bb4SMatthew G. Knepley   The exodus result nodal variable index is obtained by comparing the Vec name and the
87778569bb4SMatthew G. Knepley   names of nodal variables declared in the exodus file. For instance for a Vec named "V"
87878569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
87978569bb4SMatthew G. Knepley   amongst all nodal variables.
88078569bb4SMatthew G. Knepley 
88178569bb4SMatthew G. Knepley   Level: beginner
88278569bb4SMatthew G. Knepley 
8836823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(),DMPlexView_ExodusII_Internal(),VecLoadNodal_PlexEXO(),VecViewZonal_PlexEXO(),VecLoadZonal_PlexEXO()
88478569bb4SMatthew G. Knepley @*/
8856823f3c5SBlaise Bourdin PetscErrorCode VecViewPlex_ExodusII_Nodal_Internal(Vec v, int exoid, int step, int offset)
88678569bb4SMatthew G. Knepley {
88778569bb4SMatthew G. Knepley   MPI_Comm           comm;
88878569bb4SMatthew G. Knepley   PetscMPIInt        size;
88978569bb4SMatthew G. Knepley   DM                 dm;
89078569bb4SMatthew G. Knepley   Vec                vNatural, vComp;
89122a7544eSVaclav Hapla   const PetscScalar *varray;
89278569bb4SMatthew G. Knepley   PetscInt           xs, xe, bs;
89378569bb4SMatthew G. Knepley   PetscBool          useNatural;
89478569bb4SMatthew G. Knepley   PetscErrorCode     ierr;
89578569bb4SMatthew G. Knepley 
89678569bb4SMatthew G. Knepley   PetscFunctionBegin;
89778569bb4SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) v, &comm);CHKERRQ(ierr);
898ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &size);CHKERRMPI(ierr);
89978569bb4SMatthew G. Knepley   ierr = VecGetDM(v, &dm);CHKERRQ(ierr);
90078569bb4SMatthew G. Knepley   ierr = DMGetUseNatural(dm, &useNatural);CHKERRQ(ierr);
90178569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
90278569bb4SMatthew G. Knepley   if (useNatural) {
90378569bb4SMatthew G. Knepley     ierr = DMGetGlobalVector(dm, &vNatural);CHKERRQ(ierr);
90478569bb4SMatthew G. Knepley     ierr = DMPlexGlobalToNaturalBegin(dm, v, vNatural);CHKERRQ(ierr);
90578569bb4SMatthew G. Knepley     ierr = DMPlexGlobalToNaturalEnd(dm, v, vNatural);CHKERRQ(ierr);
90678569bb4SMatthew G. Knepley   } else {
90778569bb4SMatthew G. Knepley     vNatural = v;
90878569bb4SMatthew G. Knepley   }
9096823f3c5SBlaise Bourdin 
91078569bb4SMatthew G. Knepley   /* Write local chunk of the result in the exodus file
91178569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
91278569bb4SMatthew G. Knepley      We assume that they are stored sequentially */
91378569bb4SMatthew G. Knepley   ierr = VecGetOwnershipRange(vNatural, &xs, &xe);CHKERRQ(ierr);
91478569bb4SMatthew G. Knepley   ierr = VecGetBlockSize(vNatural, &bs);CHKERRQ(ierr);
91578569bb4SMatthew G. Knepley   if (bs == 1) {
91678569bb4SMatthew G. Knepley     ierr = VecGetArrayRead(vNatural, &varray);CHKERRQ(ierr);
9176de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_put_partial_var,(exoid, step, EX_NODAL, offset, 1, xs+1, xe-xs, varray));
91878569bb4SMatthew G. Knepley     ierr = VecRestoreArrayRead(vNatural, &varray);CHKERRQ(ierr);
91978569bb4SMatthew G. Knepley   } else {
92078569bb4SMatthew G. Knepley     IS       compIS;
92178569bb4SMatthew G. Knepley     PetscInt c;
92278569bb4SMatthew G. Knepley 
92394a2d30dSStefano Zampini     ierr = ISCreateStride(comm, (xe-xs)/bs, xs, bs, &compIS);CHKERRQ(ierr);
92478569bb4SMatthew G. Knepley     for (c = 0; c < bs; ++c) {
92578569bb4SMatthew G. Knepley       ierr = ISStrideSetStride(compIS, (xe-xs)/bs, xs+c, bs);CHKERRQ(ierr);
92678569bb4SMatthew G. Knepley       ierr = VecGetSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
92778569bb4SMatthew G. Knepley       ierr = VecGetArrayRead(vComp, &varray);CHKERRQ(ierr);
9286de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_put_partial_var,(exoid, step, EX_NODAL, offset+c, 1, xs/bs+1, (xe-xs)/bs, varray));
92978569bb4SMatthew G. Knepley       ierr = VecRestoreArrayRead(vComp, &varray);CHKERRQ(ierr);
93078569bb4SMatthew G. Knepley       ierr = VecRestoreSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
93178569bb4SMatthew G. Knepley     }
93278569bb4SMatthew G. Knepley     ierr = ISDestroy(&compIS);CHKERRQ(ierr);
93378569bb4SMatthew G. Knepley   }
93478569bb4SMatthew G. Knepley   if (useNatural) {ierr = DMRestoreGlobalVector(dm, &vNatural);CHKERRQ(ierr);}
93578569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
93678569bb4SMatthew G. Knepley }
93778569bb4SMatthew G. Knepley 
93878569bb4SMatthew G. Knepley /*
93978569bb4SMatthew G. Knepley   VecLoadPlex_ExodusII_Nodal_Internal - Read a Vec corresponding to a nodal field from an exodus file
94078569bb4SMatthew G. Knepley 
94178569bb4SMatthew G. Knepley   Collective on v
94278569bb4SMatthew G. Knepley 
94378569bb4SMatthew G. Knepley   Input Parameters:
94478569bb4SMatthew G. Knepley + v  - The vector to be written
94578569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
9466823f3c5SBlaise Bourdin . step - the time step to read at (exodus steps are numbered starting from 1)
9476823f3c5SBlaise Bourdin - offset - the location of the variable in the file
94878569bb4SMatthew G. Knepley 
94978569bb4SMatthew G. Knepley   Notes:
95078569bb4SMatthew G. Knepley   The exodus result nodal variable index is obtained by comparing the Vec name and the
95178569bb4SMatthew G. Knepley   names of nodal variables declared in the exodus file. For instance for a Vec named "V"
95278569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
95378569bb4SMatthew G. Knepley   amongst all nodal variables.
95478569bb4SMatthew G. Knepley 
95578569bb4SMatthew G. Knepley   Level: beginner
95678569bb4SMatthew G. Knepley 
9576823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(), DMPlexView_ExodusII_Internal(), VecViewPlex_ExodusII_Nodal_Internal(), VecViewZonal_PlexEXO(), VecLoadZonal_PlexEXO()
95878569bb4SMatthew G. Knepley */
9596823f3c5SBlaise Bourdin PetscErrorCode VecLoadPlex_ExodusII_Nodal_Internal(Vec v, int exoid, int step, int offset)
96078569bb4SMatthew G. Knepley {
96178569bb4SMatthew G. Knepley   MPI_Comm       comm;
96278569bb4SMatthew G. Knepley   PetscMPIInt    size;
96378569bb4SMatthew G. Knepley   DM             dm;
96478569bb4SMatthew G. Knepley   Vec            vNatural, vComp;
96522a7544eSVaclav Hapla   PetscScalar   *varray;
96678569bb4SMatthew G. Knepley   PetscInt       xs, xe, bs;
96778569bb4SMatthew G. Knepley   PetscBool      useNatural;
96878569bb4SMatthew G. Knepley   PetscErrorCode ierr;
96978569bb4SMatthew G. Knepley 
97078569bb4SMatthew G. Knepley   PetscFunctionBegin;
97178569bb4SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) v, &comm);CHKERRQ(ierr);
972ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &size);CHKERRMPI(ierr);
97378569bb4SMatthew G. Knepley   ierr = VecGetDM(v,&dm);CHKERRQ(ierr);
97478569bb4SMatthew G. Knepley   ierr = DMGetUseNatural(dm, &useNatural);CHKERRQ(ierr);
97578569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
97678569bb4SMatthew G. Knepley   if (useNatural) {ierr = DMGetGlobalVector(dm,&vNatural);CHKERRQ(ierr);}
97778569bb4SMatthew G. Knepley   else            {vNatural = v;}
9786823f3c5SBlaise Bourdin 
97978569bb4SMatthew G. Knepley   /* Read local chunk from the file */
98078569bb4SMatthew G. Knepley   ierr = VecGetOwnershipRange(vNatural, &xs, &xe);CHKERRQ(ierr);
98178569bb4SMatthew G. Knepley   ierr = VecGetBlockSize(vNatural, &bs);CHKERRQ(ierr);
98278569bb4SMatthew G. Knepley   if (bs == 1) {
98378569bb4SMatthew G. Knepley     ierr = VecGetArray(vNatural, &varray);CHKERRQ(ierr);
9846de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_partial_var,(exoid, step, EX_NODAL, offset, 1, xs+1, xe-xs, varray));
98578569bb4SMatthew G. Knepley     ierr = VecRestoreArray(vNatural, &varray);CHKERRQ(ierr);
98678569bb4SMatthew G. Knepley   } else {
98778569bb4SMatthew G. Knepley     IS       compIS;
98878569bb4SMatthew G. Knepley     PetscInt c;
98978569bb4SMatthew G. Knepley 
99094a2d30dSStefano Zampini     ierr = ISCreateStride(comm, (xe-xs)/bs, xs, bs, &compIS);CHKERRQ(ierr);
99178569bb4SMatthew G. Knepley     for (c = 0; c < bs; ++c) {
99278569bb4SMatthew G. Knepley       ierr = ISStrideSetStride(compIS, (xe-xs)/bs, xs+c, bs);CHKERRQ(ierr);
99378569bb4SMatthew G. Knepley       ierr = VecGetSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
99478569bb4SMatthew G. Knepley       ierr = VecGetArray(vComp, &varray);CHKERRQ(ierr);
9956de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_partial_var,(exoid, step, EX_NODAL, offset+c, 1, xs/bs+1, (xe-xs)/bs, varray));
99678569bb4SMatthew G. Knepley       ierr = VecRestoreArray(vComp, &varray);CHKERRQ(ierr);
99778569bb4SMatthew G. Knepley       ierr = VecRestoreSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
99878569bb4SMatthew G. Knepley     }
99978569bb4SMatthew G. Knepley     ierr = ISDestroy(&compIS);CHKERRQ(ierr);
100078569bb4SMatthew G. Knepley   }
100178569bb4SMatthew G. Knepley   if (useNatural) {
100278569bb4SMatthew G. Knepley     ierr = DMPlexNaturalToGlobalBegin(dm, vNatural, v);CHKERRQ(ierr);
100378569bb4SMatthew G. Knepley     ierr = DMPlexNaturalToGlobalEnd(dm, vNatural, v);CHKERRQ(ierr);
100478569bb4SMatthew G. Knepley     ierr = DMRestoreGlobalVector(dm, &vNatural);CHKERRQ(ierr);
100578569bb4SMatthew G. Knepley   }
100678569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
100778569bb4SMatthew G. Knepley }
100878569bb4SMatthew G. Knepley 
100978569bb4SMatthew G. Knepley /*
101078569bb4SMatthew G. Knepley   VecViewPlex_ExodusII_Zonal_Internal - Write a Vec corresponding to a zonal (cell based) field to an exodus file
101178569bb4SMatthew G. Knepley 
101278569bb4SMatthew G. Knepley   Collective on v
101378569bb4SMatthew G. Knepley 
101478569bb4SMatthew G. Knepley   Input Parameters:
101578569bb4SMatthew G. Knepley + v  - The vector to be written
101678569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
10176823f3c5SBlaise Bourdin . step - the time step to write at (exodus steps are numbered starting from 1)
10186823f3c5SBlaise Bourdin - offset - the location of the variable in the file
101978569bb4SMatthew G. Knepley 
102078569bb4SMatthew G. Knepley   Notes:
102178569bb4SMatthew G. Knepley   The exodus result zonal variable index is obtained by comparing the Vec name and the
102278569bb4SMatthew G. Knepley   names of zonal variables declared in the exodus file. For instance for a Vec named "V"
102378569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
102478569bb4SMatthew G. Knepley   amongst all zonal variables.
102578569bb4SMatthew G. Knepley 
102678569bb4SMatthew G. Knepley   Level: beginner
102778569bb4SMatthew G. Knepley 
10286823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(),DMPlexView_ExodusII_Internal(),VecViewPlex_ExodusII_Nodal_Internal(),VecLoadPlex_ExodusII_Nodal_Internal(),VecLoadPlex_ExodusII_Zonal_Internal()
102978569bb4SMatthew G. Knepley */
10306823f3c5SBlaise Bourdin PetscErrorCode VecViewPlex_ExodusII_Zonal_Internal(Vec v, int exoid, int step, int offset)
103178569bb4SMatthew G. Knepley {
103278569bb4SMatthew G. Knepley   MPI_Comm          comm;
103378569bb4SMatthew G. Knepley   PetscMPIInt       size;
103478569bb4SMatthew G. Knepley   DM                dm;
103578569bb4SMatthew G. Knepley   Vec               vNatural, vComp;
103622a7544eSVaclav Hapla   const PetscScalar *varray;
103778569bb4SMatthew G. Knepley   PetscInt          xs, xe, bs;
103878569bb4SMatthew G. Knepley   PetscBool         useNatural;
103978569bb4SMatthew G. Knepley   IS                compIS;
104078569bb4SMatthew G. Knepley   PetscInt         *csSize, *csID;
104178569bb4SMatthew G. Knepley   PetscInt          numCS, set, csxs = 0;
104278569bb4SMatthew G. Knepley   PetscErrorCode    ierr;
104378569bb4SMatthew G. Knepley 
104478569bb4SMatthew G. Knepley   PetscFunctionBegin;
104578569bb4SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)v, &comm);CHKERRQ(ierr);
1046ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &size);CHKERRMPI(ierr);
104778569bb4SMatthew G. Knepley   ierr = VecGetDM(v, &dm);CHKERRQ(ierr);
104878569bb4SMatthew G. Knepley   ierr = DMGetUseNatural(dm, &useNatural);CHKERRQ(ierr);
104978569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
105078569bb4SMatthew G. Knepley   if (useNatural) {
105178569bb4SMatthew G. Knepley     ierr = DMGetGlobalVector(dm, &vNatural);CHKERRQ(ierr);
105278569bb4SMatthew G. Knepley     ierr = DMPlexGlobalToNaturalBegin(dm, v, vNatural);CHKERRQ(ierr);
105378569bb4SMatthew G. Knepley     ierr = DMPlexGlobalToNaturalEnd(dm, v, vNatural);CHKERRQ(ierr);
105478569bb4SMatthew G. Knepley   } else {
105578569bb4SMatthew G. Knepley     vNatural = v;
105678569bb4SMatthew G. Knepley   }
10576823f3c5SBlaise Bourdin 
105878569bb4SMatthew G. Knepley   /* Write local chunk of the result in the exodus file
105978569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
106078569bb4SMatthew G. Knepley      We assume that they are stored sequentially
106178569bb4SMatthew G. Knepley      Zonal variables are accessed one element block at a time, so we loop through the cell sets,
106278569bb4SMatthew G. Knepley      but once the vector has been reordered to natural size, we cannot use the label informations
106378569bb4SMatthew G. Knepley      to figure out what to save where. */
106478569bb4SMatthew G. Knepley   numCS = ex_inquire_int(exoid, EX_INQ_ELEM_BLK);
106578569bb4SMatthew G. Knepley   ierr = PetscMalloc2(numCS, &csID, numCS, &csSize);CHKERRQ(ierr);
10666de834b4SMatthew G. Knepley   PetscStackCallStandard(ex_get_ids,(exoid, EX_ELEM_BLOCK, csID));
106778569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
106878569bb4SMatthew G. Knepley     ex_block block;
106978569bb4SMatthew G. Knepley 
107078569bb4SMatthew G. Knepley     block.id   = csID[set];
107178569bb4SMatthew G. Knepley     block.type = EX_ELEM_BLOCK;
10726de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_block_param,(exoid, &block));
107378569bb4SMatthew G. Knepley     csSize[set] = block.num_entry;
107478569bb4SMatthew G. Knepley   }
107578569bb4SMatthew G. Knepley   ierr = VecGetOwnershipRange(vNatural, &xs, &xe);CHKERRQ(ierr);
107678569bb4SMatthew G. Knepley   ierr = VecGetBlockSize(vNatural, &bs);CHKERRQ(ierr);
107778569bb4SMatthew G. Knepley   if (bs > 1) {ierr = ISCreateStride(comm, (xe-xs)/bs, xs, bs, &compIS);CHKERRQ(ierr);}
107878569bb4SMatthew G. Knepley   for (set = 0; set < numCS; set++) {
107978569bb4SMatthew G. Knepley     PetscInt csLocalSize, c;
108078569bb4SMatthew G. Knepley 
108178569bb4SMatthew G. Knepley     /* range of indices for set setID[set]: csxs:csxs + csSize[set]-1
108278569bb4SMatthew G. Knepley        local slice of zonal values:         xs/bs,xm/bs-1
108378569bb4SMatthew G. Knepley        intersection:                        max(xs/bs,csxs),min(xm/bs-1,csxs + csSize[set]-1) */
108478569bb4SMatthew G. Knepley     csLocalSize = PetscMax(0, PetscMin(xe/bs, csxs+csSize[set]) - PetscMax(xs/bs, csxs));
108578569bb4SMatthew G. Knepley     if (bs == 1) {
108678569bb4SMatthew G. Knepley       ierr = VecGetArrayRead(vNatural, &varray);CHKERRQ(ierr);
10876de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_put_partial_var,(exoid, step, EX_ELEM_BLOCK, offset, csID[set], PetscMax(xs-csxs, 0)+1, csLocalSize, &varray[PetscMax(0, csxs-xs)]));
108878569bb4SMatthew G. Knepley       ierr = VecRestoreArrayRead(vNatural, &varray);CHKERRQ(ierr);
108978569bb4SMatthew G. Knepley     } else {
109078569bb4SMatthew G. Knepley       for (c = 0; c < bs; ++c) {
109178569bb4SMatthew G. Knepley         ierr = ISStrideSetStride(compIS, (xe-xs)/bs, xs+c, bs);CHKERRQ(ierr);
109278569bb4SMatthew G. Knepley         ierr = VecGetSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
109378569bb4SMatthew G. Knepley         ierr = VecGetArrayRead(vComp, &varray);CHKERRQ(ierr);
10946de834b4SMatthew G. Knepley         PetscStackCallStandard(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)]));
109578569bb4SMatthew G. Knepley         ierr = VecRestoreArrayRead(vComp, &varray);CHKERRQ(ierr);
109678569bb4SMatthew G. Knepley         ierr = VecRestoreSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
109778569bb4SMatthew G. Knepley       }
109878569bb4SMatthew G. Knepley     }
109978569bb4SMatthew G. Knepley     csxs += csSize[set];
110078569bb4SMatthew G. Knepley   }
110178569bb4SMatthew G. Knepley   ierr = PetscFree2(csID, csSize);CHKERRQ(ierr);
110278569bb4SMatthew G. Knepley   if (bs > 1) {ierr = ISDestroy(&compIS);CHKERRQ(ierr);}
110378569bb4SMatthew G. Knepley   if (useNatural) {ierr = DMRestoreGlobalVector(dm,&vNatural);CHKERRQ(ierr);}
110478569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
110578569bb4SMatthew G. Knepley }
110678569bb4SMatthew G. Knepley 
110778569bb4SMatthew G. Knepley /*
110878569bb4SMatthew G. Knepley   VecLoadPlex_ExodusII_Zonal_Internal - Read a Vec corresponding to a zonal (cell based) field from an exodus file
110978569bb4SMatthew G. Knepley 
111078569bb4SMatthew G. Knepley   Collective on v
111178569bb4SMatthew G. Knepley 
111278569bb4SMatthew G. Knepley   Input Parameters:
111378569bb4SMatthew G. Knepley + v  - The vector to be written
111478569bb4SMatthew G. Knepley . exoid - the exodus id of a file (obtained from ex_open or ex_create for instance)
11156823f3c5SBlaise Bourdin . step - the time step to read at (exodus steps are numbered starting from 1)
11166823f3c5SBlaise Bourdin - offset - the location of the variable in the file
111778569bb4SMatthew G. Knepley 
111878569bb4SMatthew G. Knepley   Notes:
111978569bb4SMatthew G. Knepley   The exodus result zonal variable index is obtained by comparing the Vec name and the
112078569bb4SMatthew G. Knepley   names of zonal variables declared in the exodus file. For instance for a Vec named "V"
112178569bb4SMatthew G. Knepley   the location in the exodus file will be the first match of "V", "V_X", "V_XX", "V_1", or "V_11"
112278569bb4SMatthew G. Knepley   amongst all zonal variables.
112378569bb4SMatthew G. Knepley 
112478569bb4SMatthew G. Knepley   Level: beginner
112578569bb4SMatthew G. Knepley 
11266823f3c5SBlaise Bourdin .seealso: EXOGetVarIndex_Internal(), DMPlexView_ExodusII_Internal(), VecViewPlex_ExodusII_Nodal_Internal(), VecLoadPlex_ExodusII_Nodal_Internal(), VecLoadPlex_ExodusII_Zonal_Internal()
112778569bb4SMatthew G. Knepley */
11286823f3c5SBlaise Bourdin PetscErrorCode VecLoadPlex_ExodusII_Zonal_Internal(Vec v, int exoid, int step, int offset)
112978569bb4SMatthew G. Knepley {
113078569bb4SMatthew G. Knepley   MPI_Comm          comm;
113178569bb4SMatthew G. Knepley   PetscMPIInt       size;
113278569bb4SMatthew G. Knepley   DM                dm;
113378569bb4SMatthew G. Knepley   Vec               vNatural, vComp;
113422a7544eSVaclav Hapla   PetscScalar      *varray;
113578569bb4SMatthew G. Knepley   PetscInt          xs, xe, bs;
113678569bb4SMatthew G. Knepley   PetscBool         useNatural;
113778569bb4SMatthew G. Knepley   IS                compIS;
113878569bb4SMatthew G. Knepley   PetscInt         *csSize, *csID;
113978569bb4SMatthew G. Knepley   PetscInt          numCS, set, csxs = 0;
114078569bb4SMatthew G. Knepley   PetscErrorCode    ierr;
114178569bb4SMatthew G. Knepley 
114278569bb4SMatthew G. Knepley   PetscFunctionBegin;
114378569bb4SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)v,&comm);CHKERRQ(ierr);
1144ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &size);CHKERRMPI(ierr);
114578569bb4SMatthew G. Knepley   ierr = VecGetDM(v, &dm);CHKERRQ(ierr);
114678569bb4SMatthew G. Knepley   ierr = DMGetUseNatural(dm, &useNatural);CHKERRQ(ierr);
114778569bb4SMatthew G. Knepley   useNatural = useNatural && size > 1 ? PETSC_TRUE : PETSC_FALSE;
114878569bb4SMatthew G. Knepley   if (useNatural) {ierr = DMGetGlobalVector(dm,&vNatural);CHKERRQ(ierr);}
114978569bb4SMatthew G. Knepley   else            {vNatural = v;}
11506823f3c5SBlaise Bourdin 
115178569bb4SMatthew G. Knepley   /* Read local chunk of the result in the exodus file
115278569bb4SMatthew G. Knepley      exodus stores each component of a vector-valued field as a separate variable.
115378569bb4SMatthew G. Knepley      We assume that they are stored sequentially
115478569bb4SMatthew G. Knepley      Zonal variables are accessed one element block at a time, so we loop through the cell sets,
115578569bb4SMatthew G. Knepley      but once the vector has been reordered to natural size, we cannot use the label informations
115678569bb4SMatthew G. Knepley      to figure out what to save where. */
115778569bb4SMatthew G. Knepley   numCS = ex_inquire_int(exoid, EX_INQ_ELEM_BLK);
115878569bb4SMatthew G. Knepley   ierr = PetscMalloc2(numCS, &csID, numCS, &csSize);CHKERRQ(ierr);
11596de834b4SMatthew G. Knepley   PetscStackCallStandard(ex_get_ids,(exoid, EX_ELEM_BLOCK, csID));
116078569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
116178569bb4SMatthew G. Knepley     ex_block block;
116278569bb4SMatthew G. Knepley 
116378569bb4SMatthew G. Knepley     block.id   = csID[set];
116478569bb4SMatthew G. Knepley     block.type = EX_ELEM_BLOCK;
11656de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_block_param,(exoid, &block));
116678569bb4SMatthew G. Knepley     csSize[set] = block.num_entry;
116778569bb4SMatthew G. Knepley   }
116878569bb4SMatthew G. Knepley   ierr = VecGetOwnershipRange(vNatural, &xs, &xe);CHKERRQ(ierr);
116978569bb4SMatthew G. Knepley   ierr = VecGetBlockSize(vNatural, &bs);CHKERRQ(ierr);
117078569bb4SMatthew G. Knepley   if (bs > 1) {ierr = ISCreateStride(comm, (xe-xs)/bs, xs, bs, &compIS);CHKERRQ(ierr);}
117178569bb4SMatthew G. Knepley   for (set = 0; set < numCS; ++set) {
117278569bb4SMatthew G. Knepley     PetscInt csLocalSize, c;
117378569bb4SMatthew G. Knepley 
117478569bb4SMatthew G. Knepley     /* range of indices for set setID[set]: csxs:csxs + csSize[set]-1
117578569bb4SMatthew G. Knepley        local slice of zonal values:         xs/bs,xm/bs-1
117678569bb4SMatthew G. Knepley        intersection:                        max(xs/bs,csxs),min(xm/bs-1,csxs + csSize[set]-1) */
117778569bb4SMatthew G. Knepley     csLocalSize = PetscMax(0, PetscMin(xe/bs, csxs+csSize[set]) - PetscMax(xs/bs, csxs));
117878569bb4SMatthew G. Knepley     if (bs == 1) {
117978569bb4SMatthew G. Knepley       ierr = VecGetArray(vNatural, &varray);CHKERRQ(ierr);
11806de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_partial_var,(exoid, step, EX_ELEM_BLOCK, offset, csID[set], PetscMax(xs-csxs, 0)+1, csLocalSize, &varray[PetscMax(0, csxs-xs)]));
118178569bb4SMatthew G. Knepley       ierr = VecRestoreArray(vNatural, &varray);CHKERRQ(ierr);
118278569bb4SMatthew G. Knepley     } else {
118378569bb4SMatthew G. Knepley       for (c = 0; c < bs; ++c) {
118478569bb4SMatthew G. Knepley         ierr = ISStrideSetStride(compIS, (xe-xs)/bs, xs+c, bs);CHKERRQ(ierr);
118578569bb4SMatthew G. Knepley         ierr = VecGetSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
118678569bb4SMatthew G. Knepley         ierr = VecGetArray(vComp, &varray);CHKERRQ(ierr);
11876de834b4SMatthew G. Knepley         PetscStackCallStandard(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)]));
118878569bb4SMatthew G. Knepley         ierr = VecRestoreArray(vComp, &varray);CHKERRQ(ierr);
118978569bb4SMatthew G. Knepley         ierr = VecRestoreSubVector(vNatural, compIS, &vComp);CHKERRQ(ierr);
119078569bb4SMatthew G. Knepley       }
119178569bb4SMatthew G. Knepley     }
119278569bb4SMatthew G. Knepley     csxs += csSize[set];
119378569bb4SMatthew G. Knepley   }
119478569bb4SMatthew G. Knepley   ierr = PetscFree2(csID, csSize);CHKERRQ(ierr);
119578569bb4SMatthew G. Knepley   if (bs > 1) {ierr = ISDestroy(&compIS);CHKERRQ(ierr);}
119678569bb4SMatthew G. Knepley   if (useNatural) {
119778569bb4SMatthew G. Knepley     ierr = DMPlexNaturalToGlobalBegin(dm, vNatural, v);CHKERRQ(ierr);
119878569bb4SMatthew G. Knepley     ierr = DMPlexNaturalToGlobalEnd(dm, vNatural, v);CHKERRQ(ierr);
119978569bb4SMatthew G. Knepley     ierr = DMRestoreGlobalVector(dm, &vNatural);CHKERRQ(ierr);
120078569bb4SMatthew G. Knepley   }
120178569bb4SMatthew G. Knepley   PetscFunctionReturn(0);
120278569bb4SMatthew G. Knepley }
1203b53c8456SSatish Balay #endif
120478569bb4SMatthew G. Knepley 
12051e50132fSMatthew G. Knepley /*@
12061e50132fSMatthew G. Knepley   PetscViewerExodusIIGetId - Get the file id of the ExodusII file
12071e50132fSMatthew G. Knepley 
12081e50132fSMatthew G. Knepley   Logically Collective on PetscViewer
12091e50132fSMatthew G. Knepley 
12101e50132fSMatthew G. Knepley   Input Parameter:
12111e50132fSMatthew G. Knepley .  viewer - the PetscViewer
12121e50132fSMatthew G. Knepley 
12131e50132fSMatthew G. Knepley   Output Parameter:
12141e50132fSMatthew G. Knepley -  exoid - The ExodusII file id
12151e50132fSMatthew G. Knepley 
12161e50132fSMatthew G. Knepley   Level: intermediate
12171e50132fSMatthew G. Knepley 
12181e50132fSMatthew G. Knepley .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen()
12191e50132fSMatthew G. Knepley @*/
12201e50132fSMatthew G. Knepley PetscErrorCode PetscViewerExodusIIGetId(PetscViewer viewer, int *exoid)
12211e50132fSMatthew G. Knepley {
12221e50132fSMatthew G. Knepley   PetscErrorCode ierr;
12231e50132fSMatthew G. Knepley 
12241e50132fSMatthew G. Knepley   PetscFunctionBegin;
12251e50132fSMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
12266823f3c5SBlaise Bourdin   ierr = PetscTryMethod(viewer, "PetscViewerGetId_C",(PetscViewer,int*),(viewer,exoid));CHKERRQ(ierr);
12276823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12286823f3c5SBlaise Bourdin }
12296823f3c5SBlaise Bourdin 
12306823f3c5SBlaise Bourdin 
12316823f3c5SBlaise Bourdin 
12326823f3c5SBlaise Bourdin /*@
12336823f3c5SBlaise Bourdin    PetscViewerExodusIISetOrder - Set the elements order in the exodusII file.
12346823f3c5SBlaise Bourdin 
12356823f3c5SBlaise Bourdin    Collective
12366823f3c5SBlaise Bourdin 
12376823f3c5SBlaise Bourdin    Input Parameters:
12386823f3c5SBlaise Bourdin +  viewer - the viewer
1239*98a6a78aSPierre Jolivet -  order - elements order
12406823f3c5SBlaise Bourdin 
12416823f3c5SBlaise Bourdin    Output Parameter:
12426823f3c5SBlaise Bourdin 
12436823f3c5SBlaise Bourdin    Level: beginner
12446823f3c5SBlaise Bourdin 
12456823f3c5SBlaise Bourdin    Note:
12466823f3c5SBlaise Bourdin 
12476823f3c5SBlaise Bourdin 
12486823f3c5SBlaise Bourdin .seealso: PetscViewerExodusIIGetId(), PetscViewerExodusIIGetOrder(), PetscViewerExodusIISetOrder()
12496823f3c5SBlaise Bourdin @*/
12506823f3c5SBlaise Bourdin PetscErrorCode PetscViewerExodusIISetOrder(PetscViewer viewer, PetscInt order)
12516823f3c5SBlaise Bourdin {
12526823f3c5SBlaise Bourdin   PetscErrorCode ierr;
12536823f3c5SBlaise Bourdin 
12546823f3c5SBlaise Bourdin   PetscFunctionBegin;
12556823f3c5SBlaise Bourdin   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
12566823f3c5SBlaise Bourdin   ierr = PetscTryMethod(viewer, "PetscViewerSetOrder_C",(PetscViewer,PetscInt),(viewer,order));CHKERRQ(ierr);
12576823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12586823f3c5SBlaise Bourdin }
12596823f3c5SBlaise Bourdin 
12606823f3c5SBlaise Bourdin /*@
12616823f3c5SBlaise Bourdin    PetscViewerExodusIIGetOrder - Get the elements order in the exodusII file.
12626823f3c5SBlaise Bourdin 
12636823f3c5SBlaise Bourdin    Collective
12646823f3c5SBlaise Bourdin 
12656823f3c5SBlaise Bourdin    Input Parameters:
12666823f3c5SBlaise Bourdin +  viewer - the viewer
1267*98a6a78aSPierre Jolivet -  order - elements order
12686823f3c5SBlaise Bourdin 
12696823f3c5SBlaise Bourdin    Output Parameter:
12706823f3c5SBlaise Bourdin 
12716823f3c5SBlaise Bourdin    Level: beginner
12726823f3c5SBlaise Bourdin 
12736823f3c5SBlaise Bourdin    Note:
12746823f3c5SBlaise Bourdin 
12756823f3c5SBlaise Bourdin 
12766823f3c5SBlaise Bourdin .seealso: PetscViewerExodusIIGetId(), PetscViewerExodusIIGetOrder(), PetscViewerExodusIISetOrder()
12776823f3c5SBlaise Bourdin @*/
12786823f3c5SBlaise Bourdin PetscErrorCode PetscViewerExodusIIGetOrder(PetscViewer viewer, PetscInt *order)
12796823f3c5SBlaise Bourdin {
12806823f3c5SBlaise Bourdin   PetscErrorCode ierr;
12816823f3c5SBlaise Bourdin 
12826823f3c5SBlaise Bourdin   PetscFunctionBegin;
12836823f3c5SBlaise Bourdin   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
12846823f3c5SBlaise Bourdin   ierr = PetscTryMethod(viewer, "PetscViewerGetOrder_C",(PetscViewer,PetscInt*),(viewer,order));CHKERRQ(ierr);
12856823f3c5SBlaise Bourdin   PetscFunctionReturn(0);
12866823f3c5SBlaise Bourdin }
12876823f3c5SBlaise Bourdin 
12886823f3c5SBlaise Bourdin /*@C
12896823f3c5SBlaise Bourdin    PetscViewerExodusIIOpen - Opens a file for ExodusII input/output.
12906823f3c5SBlaise Bourdin 
12916823f3c5SBlaise Bourdin    Collective
12926823f3c5SBlaise Bourdin 
12936823f3c5SBlaise Bourdin    Input Parameters:
12946823f3c5SBlaise Bourdin +  comm - MPI communicator
12956823f3c5SBlaise Bourdin .  name - name of file
12966823f3c5SBlaise Bourdin -  type - type of file
12976823f3c5SBlaise Bourdin $    FILE_MODE_WRITE - create new file for binary output
12986823f3c5SBlaise Bourdin $    FILE_MODE_READ - open existing file for binary input
12996823f3c5SBlaise Bourdin $    FILE_MODE_APPEND - open existing file for binary output
13006823f3c5SBlaise Bourdin 
13016823f3c5SBlaise Bourdin    Output Parameter:
13026823f3c5SBlaise Bourdin .  exo - PetscViewer for Exodus II input/output to use with the specified file
13036823f3c5SBlaise Bourdin 
13046823f3c5SBlaise Bourdin    Level: beginner
13056823f3c5SBlaise Bourdin 
13066823f3c5SBlaise Bourdin    Note:
13076823f3c5SBlaise Bourdin    This PetscViewer should be destroyed with PetscViewerDestroy().
13086823f3c5SBlaise Bourdin 
13096823f3c5SBlaise Bourdin 
13106823f3c5SBlaise Bourdin .seealso: PetscViewerPushFormat(), PetscViewerDestroy(),
13116823f3c5SBlaise Bourdin           DMLoad(), PetscFileMode, PetscViewer, PetscViewerSetType(), PetscViewerFileSetMode(), PetscViewerFileSetName()
13126823f3c5SBlaise Bourdin @*/
13136823f3c5SBlaise Bourdin PetscErrorCode PetscViewerExodusIIOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *exo)
13146823f3c5SBlaise Bourdin {
13156823f3c5SBlaise Bourdin   PetscErrorCode ierr;
13166823f3c5SBlaise Bourdin 
13176823f3c5SBlaise Bourdin   PetscFunctionBegin;
13186823f3c5SBlaise Bourdin   ierr = PetscViewerCreate(comm, exo);CHKERRQ(ierr);
13196823f3c5SBlaise Bourdin   ierr = PetscViewerSetType(*exo, PETSCVIEWEREXODUSII);CHKERRQ(ierr);
13206823f3c5SBlaise Bourdin   ierr = PetscViewerFileSetMode(*exo, type);CHKERRQ(ierr);
13216823f3c5SBlaise Bourdin   ierr = PetscViewerFileSetName(*exo, name);CHKERRQ(ierr);
13226823f3c5SBlaise Bourdin   ierr = PetscViewerSetFromOptions(*exo);CHKERRQ(ierr);
13231e50132fSMatthew G. Knepley   PetscFunctionReturn(0);
13241e50132fSMatthew G. Knepley }
13251e50132fSMatthew G. Knepley 
132633751fbdSMichael Lange /*@C
1327eaf898f9SPatrick Sanan   DMPlexCreateExodusFromFile - Create a DMPlex mesh from an ExodusII file.
132833751fbdSMichael Lange 
1329d083f849SBarry Smith   Collective
133033751fbdSMichael Lange 
133133751fbdSMichael Lange   Input Parameters:
133233751fbdSMichael Lange + comm  - The MPI communicator
133333751fbdSMichael Lange . filename - The name of the ExodusII file
133433751fbdSMichael Lange - interpolate - Create faces and edges in the mesh
133533751fbdSMichael Lange 
133633751fbdSMichael Lange   Output Parameter:
133733751fbdSMichael Lange . dm  - The DM object representing the mesh
133833751fbdSMichael Lange 
133933751fbdSMichael Lange   Level: beginner
134033751fbdSMichael Lange 
134133751fbdSMichael Lange .seealso: DMPLEX, DMCreate(), DMPlexCreateExodus()
134233751fbdSMichael Lange @*/
134333751fbdSMichael Lange PetscErrorCode DMPlexCreateExodusFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
134433751fbdSMichael Lange {
134533751fbdSMichael Lange   PetscMPIInt    rank;
134633751fbdSMichael Lange   PetscErrorCode ierr;
134733751fbdSMichael Lange #if defined(PETSC_HAVE_EXODUSII)
1348e84f7031SBlaise Bourdin   int   CPU_word_size = sizeof(PetscReal), IO_word_size = 0, exoid = -1;
134933751fbdSMichael Lange   float version;
135033751fbdSMichael Lange #endif
135133751fbdSMichael Lange 
135233751fbdSMichael Lange   PetscFunctionBegin;
135333751fbdSMichael Lange   PetscValidCharPointer(filename, 2);
1354ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr);
135533751fbdSMichael Lange #if defined(PETSC_HAVE_EXODUSII)
135633751fbdSMichael Lange   if (!rank) {
135733751fbdSMichael Lange     exoid = ex_open(filename, EX_READ, &CPU_word_size, &IO_word_size, &version);
135833751fbdSMichael Lange     if (exoid <= 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_LIB, "ex_open(\"%s\",...) did not return a valid file ID", filename);
135933751fbdSMichael Lange   }
136033751fbdSMichael Lange   ierr = DMPlexCreateExodus(comm, exoid, interpolate, dm);CHKERRQ(ierr);
13616de834b4SMatthew G. Knepley   if (!rank) {PetscStackCallStandard(ex_close,(exoid));}
1362b458e8f1SJose E. Roman   PetscFunctionReturn(0);
136333751fbdSMichael Lange #else
136433751fbdSMichael Lange   SETERRQ(comm, PETSC_ERR_SUP, "This method requires ExodusII support. Reconfigure using --download-exodusii");
136533751fbdSMichael Lange #endif
136633751fbdSMichael Lange }
136733751fbdSMichael Lange 
13688f861fbcSMatthew G. Knepley #if defined(PETSC_HAVE_EXODUSII)
13696823f3c5SBlaise Bourdin static PetscErrorCode ExodusGetCellType_Internal(const char *elem_type, DMPolytopeType *ct)
13708f861fbcSMatthew G. Knepley {
13718f861fbcSMatthew G. Knepley   PetscBool      flg;
13728f861fbcSMatthew G. Knepley   PetscErrorCode ierr;
13738f861fbcSMatthew G. Knepley 
13748f861fbcSMatthew G. Knepley   PetscFunctionBegin;
13758f861fbcSMatthew G. Knepley   *ct = DM_POLYTOPE_UNKNOWN;
13768f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "TRI", &flg);CHKERRQ(ierr);
13778f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_TRIANGLE; goto done;}
13788f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "TRI3", &flg);CHKERRQ(ierr);
13798f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_TRIANGLE; goto done;}
13808f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "QUAD", &flg);CHKERRQ(ierr);
13818f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_QUADRILATERAL; goto done;}
13828f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "QUAD4", &flg);CHKERRQ(ierr);
13838f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_QUADRILATERAL; goto done;}
13848f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "SHELL4", &flg);CHKERRQ(ierr);
13858f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_QUADRILATERAL; goto done;}
13868f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "TETRA", &flg);CHKERRQ(ierr);
13878f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_TETRAHEDRON; goto done;}
13888f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "TET4", &flg);CHKERRQ(ierr);
13898f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_TETRAHEDRON; goto done;}
13908f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "WEDGE", &flg);CHKERRQ(ierr);
13918f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_TRI_PRISM; goto done;}
13928f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "HEX", &flg);CHKERRQ(ierr);
13938f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_HEXAHEDRON; goto done;}
13948f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "HEX8", &flg);CHKERRQ(ierr);
13958f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_HEXAHEDRON; goto done;}
13968f861fbcSMatthew G. Knepley   ierr = PetscStrcmp(elem_type, "HEXAHEDRON", &flg);CHKERRQ(ierr);
13978f861fbcSMatthew G. Knepley   if (flg) {*ct = DM_POLYTOPE_HEXAHEDRON; goto done;}
13988f861fbcSMatthew G. Knepley   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unrecognized element type %s", elem_type);
13998f861fbcSMatthew G. Knepley   done:
14008f861fbcSMatthew G. Knepley   PetscFunctionReturn(0);
14018f861fbcSMatthew G. Knepley }
14028f861fbcSMatthew G. Knepley #endif
14038f861fbcSMatthew G. Knepley 
1404552f7358SJed Brown /*@
140533751fbdSMichael Lange   DMPlexCreateExodus - Create a DMPlex mesh from an ExodusII file ID.
1406552f7358SJed Brown 
1407d083f849SBarry Smith   Collective
1408552f7358SJed Brown 
1409552f7358SJed Brown   Input Parameters:
1410552f7358SJed Brown + comm  - The MPI communicator
1411552f7358SJed Brown . exoid - The ExodusII id associated with a exodus file and obtained using ex_open
1412552f7358SJed Brown - interpolate - Create faces and edges in the mesh
1413552f7358SJed Brown 
1414552f7358SJed Brown   Output Parameter:
1415552f7358SJed Brown . dm  - The DM object representing the mesh
1416552f7358SJed Brown 
1417552f7358SJed Brown   Level: beginner
1418552f7358SJed Brown 
1419e84e7769SJed Brown .seealso: DMPLEX, DMCreate()
1420552f7358SJed Brown @*/
1421552f7358SJed Brown PetscErrorCode DMPlexCreateExodus(MPI_Comm comm, PetscInt exoid, PetscBool interpolate, DM *dm)
1422552f7358SJed Brown {
1423552f7358SJed Brown #if defined(PETSC_HAVE_EXODUSII)
1424552f7358SJed Brown   PetscMPIInt    num_proc, rank;
1425552f7358SJed Brown   PetscSection   coordSection;
1426552f7358SJed Brown   Vec            coordinates;
1427552f7358SJed Brown   PetscScalar    *coords;
1428552f7358SJed Brown   PetscInt       coordSize, v;
1429552f7358SJed Brown   PetscErrorCode ierr;
1430552f7358SJed Brown   /* Read from ex_get_init() */
1431552f7358SJed Brown   char title[PETSC_MAX_PATH_LEN+1];
14328f861fbcSMatthew G. Knepley   int  dim    = 0, dimEmbed = 0, numVertices = 0, numCells = 0, numHybridCells = 0;
1433552f7358SJed Brown   int  num_cs = 0, num_vs = 0, num_fs = 0;
1434552f7358SJed Brown #endif
1435552f7358SJed Brown 
1436552f7358SJed Brown   PetscFunctionBegin;
1437552f7358SJed Brown #if defined(PETSC_HAVE_EXODUSII)
1438ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr);
1439ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm, &num_proc);CHKERRMPI(ierr);
1440552f7358SJed Brown   ierr = DMCreate(comm, dm);CHKERRQ(ierr);
1441552f7358SJed Brown   ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
1442552f7358SJed Brown   /* Open EXODUS II file and read basic informations on rank 0, then broadcast to all processors */
1443552f7358SJed Brown   if (!rank) {
1444580bdb30SBarry Smith     ierr = PetscMemzero(title,PETSC_MAX_PATH_LEN+1);CHKERRQ(ierr);
14458f861fbcSMatthew G. Knepley     PetscStackCallStandard(ex_get_init,(exoid, title, &dimEmbed, &numVertices, &numCells, &num_cs, &num_vs, &num_fs));
1446552f7358SJed Brown     if (!num_cs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Exodus file does not contain any cell set\n");
1447552f7358SJed Brown   }
1448ffc4695bSBarry Smith   ierr = MPI_Bcast(title, PETSC_MAX_PATH_LEN+1, MPI_CHAR, 0, comm);CHKERRMPI(ierr);
1449ffc4695bSBarry Smith   ierr = MPI_Bcast(&dim, 1, MPI_INT, 0, comm);CHKERRMPI(ierr);
1450552f7358SJed Brown   ierr = PetscObjectSetName((PetscObject) *dm, title);CHKERRQ(ierr);
1451552f7358SJed Brown   ierr = DMPlexSetChart(*dm, 0, numCells+numVertices);CHKERRQ(ierr);
1452412e9a14SMatthew G. Knepley   /*   We do not want this label automatically computed, instead we compute it here */
1453412e9a14SMatthew G. Knepley   ierr = DMCreateLabel(*dm, "celltype");CHKERRQ(ierr);
1454552f7358SJed Brown 
1455552f7358SJed Brown   /* Read cell sets information */
1456552f7358SJed Brown   if (!rank) {
1457552f7358SJed Brown     PetscInt *cone;
1458e8f6893fSMatthew G. Knepley     int      c, cs, ncs, c_loc, v, v_loc;
1459552f7358SJed Brown     /* Read from ex_get_elem_blk_ids() */
1460e8f6893fSMatthew G. Knepley     int *cs_id, *cs_order;
1461552f7358SJed Brown     /* Read from ex_get_elem_block() */
1462552f7358SJed Brown     char buffer[PETSC_MAX_PATH_LEN+1];
1463e8f6893fSMatthew G. Knepley     int  num_cell_in_set, num_vertex_per_cell, num_hybrid, num_attr;
1464552f7358SJed Brown     /* Read from ex_get_elem_conn() */
1465552f7358SJed Brown     int *cs_connect;
1466552f7358SJed Brown 
1467552f7358SJed Brown     /* Get cell sets IDs */
1468e8f6893fSMatthew G. Knepley     ierr = PetscMalloc2(num_cs, &cs_id, num_cs, &cs_order);CHKERRQ(ierr);
14696de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_ids,(exoid, EX_ELEM_BLOCK, cs_id));
1470552f7358SJed Brown     /* Read the cell set connectivity table and build mesh topology
1471552f7358SJed Brown        EXO standard requires that cells in cell sets be numbered sequentially and be pairwise disjoint. */
1472e8f6893fSMatthew G. Knepley     /* Check for a hybrid mesh */
1473e8f6893fSMatthew G. Knepley     for (cs = 0, num_hybrid = 0; cs < num_cs; ++cs) {
14748f861fbcSMatthew G. Knepley       DMPolytopeType ct;
14758f861fbcSMatthew G. Knepley       char           elem_type[PETSC_MAX_PATH_LEN];
14768f861fbcSMatthew G. Knepley 
14778f861fbcSMatthew G. Knepley       ierr = PetscArrayzero(elem_type, sizeof(elem_type));CHKERRQ(ierr);
14788f861fbcSMatthew G. Knepley       PetscStackCallStandard(ex_get_elem_type,(exoid, cs_id[cs], elem_type));
14796823f3c5SBlaise Bourdin       ierr = ExodusGetCellType_Internal(elem_type, &ct);CHKERRQ(ierr);
14808f861fbcSMatthew G. Knepley       dim  = PetscMax(dim, DMPolytopeTypeGetDim(ct));
1481e8f6893fSMatthew G. Knepley       PetscStackCallStandard(ex_get_block,(exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set,&num_vertex_per_cell, 0, 0, &num_attr));
14828f861fbcSMatthew G. Knepley       switch (ct) {
14838f861fbcSMatthew G. Knepley         case DM_POLYTOPE_TRI_PRISM:
1484e8f6893fSMatthew G. Knepley           cs_order[cs] = cs;
1485e8f6893fSMatthew G. Knepley           numHybridCells += num_cell_in_set;
1486e8f6893fSMatthew G. Knepley           ++num_hybrid;
1487e8f6893fSMatthew G. Knepley           break;
1488e8f6893fSMatthew G. Knepley         default:
1489e8f6893fSMatthew G. Knepley           for (c = cs; c > cs-num_hybrid; --c) cs_order[c] = cs_order[c-1];
1490e8f6893fSMatthew G. Knepley           cs_order[cs-num_hybrid] = cs;
1491e8f6893fSMatthew G. Knepley       }
1492e8f6893fSMatthew G. Knepley     }
1493552f7358SJed Brown     /* First set sizes */
1494e8f6893fSMatthew G. Knepley     for (ncs = 0, c = 0; ncs < num_cs; ++ncs) {
14958f861fbcSMatthew G. Knepley       DMPolytopeType ct;
14968f861fbcSMatthew G. Knepley       char           elem_type[PETSC_MAX_PATH_LEN];
1497e8f6893fSMatthew G. Knepley       const PetscInt cs = cs_order[ncs];
14988f861fbcSMatthew G. Knepley 
14998f861fbcSMatthew G. Knepley       ierr = PetscArrayzero(elem_type, sizeof(elem_type));CHKERRQ(ierr);
15008f861fbcSMatthew G. Knepley       PetscStackCallStandard(ex_get_elem_type,(exoid, cs_id[cs], elem_type));
15016823f3c5SBlaise Bourdin       ierr = ExodusGetCellType_Internal(elem_type, &ct);CHKERRQ(ierr);
15026de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_block,(exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set,&num_vertex_per_cell, 0, 0, &num_attr));
1503552f7358SJed Brown       for (c_loc = 0; c_loc < num_cell_in_set; ++c_loc, ++c) {
1504552f7358SJed Brown         ierr = DMPlexSetConeSize(*dm, c, num_vertex_per_cell);CHKERRQ(ierr);
1505412e9a14SMatthew G. Knepley         ierr = DMPlexSetCellType(*dm, c, ct);CHKERRQ(ierr);
1506552f7358SJed Brown       }
1507552f7358SJed Brown     }
1508412e9a14SMatthew G. Knepley     for (v = numCells; v < numCells+numVertices; ++v) {ierr = DMPlexSetCellType(*dm, v, DM_POLYTOPE_POINT);CHKERRQ(ierr);}
1509552f7358SJed Brown     ierr = DMSetUp(*dm);CHKERRQ(ierr);
1510e8f6893fSMatthew G. Knepley     for (ncs = 0, c = 0; ncs < num_cs; ++ncs) {
1511e8f6893fSMatthew G. Knepley       const PetscInt cs = cs_order[ncs];
15126de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_block,(exoid, EX_ELEM_BLOCK, cs_id[cs], buffer, &num_cell_in_set, &num_vertex_per_cell, 0, 0, &num_attr));
1513dcca6d9dSJed Brown       ierr = PetscMalloc2(num_vertex_per_cell*num_cell_in_set,&cs_connect,num_vertex_per_cell,&cone);CHKERRQ(ierr);
15146de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_conn,(exoid, EX_ELEM_BLOCK, cs_id[cs], cs_connect,NULL,NULL));
1515eaf898f9SPatrick Sanan       /* EXO uses Fortran-based indexing, DMPlex uses C-style and numbers cell first then vertices. */
1516552f7358SJed Brown       for (c_loc = 0, v = 0; c_loc < num_cell_in_set; ++c_loc, ++c) {
1517636e64ffSStefano Zampini         DMPolytopeType ct;
1518636e64ffSStefano Zampini 
1519552f7358SJed Brown         for (v_loc = 0; v_loc < num_vertex_per_cell; ++v_loc, ++v) {
1520552f7358SJed Brown           cone[v_loc] = cs_connect[v]+numCells-1;
1521552f7358SJed Brown         }
1522636e64ffSStefano Zampini         ierr = DMPlexGetCellType(*dm, c, &ct);CHKERRQ(ierr);
1523636e64ffSStefano Zampini         ierr = DMPlexInvertCell(ct, cone);CHKERRQ(ierr);
1524552f7358SJed Brown         ierr = DMPlexSetCone(*dm, c, cone);CHKERRQ(ierr);
1525c58f1c22SToby Isaac         ierr = DMSetLabelValue(*dm, "Cell Sets", c, cs_id[cs]);CHKERRQ(ierr);
1526552f7358SJed Brown       }
1527552f7358SJed Brown       ierr = PetscFree2(cs_connect,cone);CHKERRQ(ierr);
1528552f7358SJed Brown     }
1529e8f6893fSMatthew G. Knepley     ierr = PetscFree2(cs_id, cs_order);CHKERRQ(ierr);
1530552f7358SJed Brown   }
15318f861fbcSMatthew G. Knepley   {
15328f861fbcSMatthew G. Knepley     PetscInt ints[] = {dim, dimEmbed};
15338f861fbcSMatthew G. Knepley 
1534ffc4695bSBarry Smith     ierr = MPI_Bcast(ints, 2, MPIU_INT, 0, comm);CHKERRMPI(ierr);
15358f861fbcSMatthew G. Knepley     ierr = DMSetDimension(*dm, ints[0]);CHKERRQ(ierr);
15368f861fbcSMatthew G. Knepley     ierr = DMSetCoordinateDim(*dm, ints[1]);CHKERRQ(ierr);
15378f861fbcSMatthew G. Knepley     dim      = ints[0];
15388f861fbcSMatthew G. Knepley     dimEmbed = ints[1];
15398f861fbcSMatthew G. Knepley   }
1540552f7358SJed Brown   ierr = DMPlexSymmetrize(*dm);CHKERRQ(ierr);
1541552f7358SJed Brown   ierr = DMPlexStratify(*dm);CHKERRQ(ierr);
1542552f7358SJed Brown   if (interpolate) {
15435fd9971aSMatthew G. Knepley     DM idm;
1544552f7358SJed Brown 
15459f074e33SMatthew G Knepley     ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
1546552f7358SJed Brown     ierr = DMDestroy(dm);CHKERRQ(ierr);
1547552f7358SJed Brown     *dm  = idm;
1548552f7358SJed Brown   }
1549552f7358SJed Brown 
1550552f7358SJed Brown   /* Create vertex set label */
1551552f7358SJed Brown   if (!rank && (num_vs > 0)) {
1552552f7358SJed Brown     int vs, v;
1553552f7358SJed Brown     /* Read from ex_get_node_set_ids() */
1554552f7358SJed Brown     int *vs_id;
1555552f7358SJed Brown     /* Read from ex_get_node_set_param() */
1556f958083aSBlaise Bourdin     int num_vertex_in_set;
1557552f7358SJed Brown     /* Read from ex_get_node_set() */
1558552f7358SJed Brown     int *vs_vertex_list;
1559552f7358SJed Brown 
1560552f7358SJed Brown     /* Get vertex set ids */
1561785e854fSJed Brown     ierr = PetscMalloc1(num_vs, &vs_id);CHKERRQ(ierr);
15626de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_ids,(exoid, EX_NODE_SET, vs_id));
1563552f7358SJed Brown     for (vs = 0; vs < num_vs; ++vs) {
15646de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_set_param,(exoid, EX_NODE_SET, vs_id[vs], &num_vertex_in_set, NULL));
1565785e854fSJed Brown       ierr = PetscMalloc1(num_vertex_in_set, &vs_vertex_list);CHKERRQ(ierr);
15666de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_set,(exoid, EX_NODE_SET, vs_id[vs], vs_vertex_list, NULL));
1567552f7358SJed Brown       for (v = 0; v < num_vertex_in_set; ++v) {
1568c58f1c22SToby Isaac         ierr = DMSetLabelValue(*dm, "Vertex Sets", vs_vertex_list[v]+numCells-1, vs_id[vs]);CHKERRQ(ierr);
1569552f7358SJed Brown       }
1570552f7358SJed Brown       ierr = PetscFree(vs_vertex_list);CHKERRQ(ierr);
1571552f7358SJed Brown     }
1572552f7358SJed Brown     ierr = PetscFree(vs_id);CHKERRQ(ierr);
1573552f7358SJed Brown   }
1574552f7358SJed Brown   /* Read coordinates */
157569d8a9ceSMatthew G. Knepley   ierr = DMGetCoordinateSection(*dm, &coordSection);CHKERRQ(ierr);
1576552f7358SJed Brown   ierr = PetscSectionSetNumFields(coordSection, 1);CHKERRQ(ierr);
15778f861fbcSMatthew G. Knepley   ierr = PetscSectionSetFieldComponents(coordSection, 0, dimEmbed);CHKERRQ(ierr);
1578552f7358SJed Brown   ierr = PetscSectionSetChart(coordSection, numCells, numCells + numVertices);CHKERRQ(ierr);
1579552f7358SJed Brown   for (v = numCells; v < numCells+numVertices; ++v) {
15808f861fbcSMatthew G. Knepley     ierr = PetscSectionSetDof(coordSection, v, dimEmbed);CHKERRQ(ierr);
15818f861fbcSMatthew G. Knepley     ierr = PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed);CHKERRQ(ierr);
1582552f7358SJed Brown   }
1583552f7358SJed Brown   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
1584552f7358SJed Brown   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
15858b9ced59SLisandro Dalcin   ierr = VecCreate(PETSC_COMM_SELF, &coordinates);CHKERRQ(ierr);
1586552f7358SJed Brown   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
1587552f7358SJed Brown   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
15888f861fbcSMatthew G. Knepley   ierr = VecSetBlockSize(coordinates, dimEmbed);CHKERRQ(ierr);
15892eb5907fSJed Brown   ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
1590552f7358SJed Brown   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
15910aba08f6SKarl Rupp   if (!rank) {
1592e84f7031SBlaise Bourdin     PetscReal *x, *y, *z;
1593552f7358SJed Brown 
1594dcca6d9dSJed Brown     ierr = PetscMalloc3(numVertices,&x,numVertices,&y,numVertices,&z);CHKERRQ(ierr);
15956de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_coord,(exoid, x, y, z));
15968f861fbcSMatthew G. Knepley     if (dimEmbed > 0) {
15978f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v*dimEmbed+0] = x[v];
15980d644c17SKarl Rupp     }
15998f861fbcSMatthew G. Knepley     if (dimEmbed > 1) {
16008f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v*dimEmbed+1] = y[v];
16010d644c17SKarl Rupp     }
16028f861fbcSMatthew G. Knepley     if (dimEmbed > 2) {
16038f861fbcSMatthew G. Knepley       for (v = 0; v < numVertices; ++v) coords[v*dimEmbed+2] = z[v];
16040d644c17SKarl Rupp     }
1605552f7358SJed Brown     ierr = PetscFree3(x,y,z);CHKERRQ(ierr);
1606552f7358SJed Brown   }
1607552f7358SJed Brown   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
1608552f7358SJed Brown   ierr = DMSetCoordinatesLocal(*dm, coordinates);CHKERRQ(ierr);
1609552f7358SJed Brown   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
1610552f7358SJed Brown 
1611552f7358SJed Brown   /* Create side set label */
1612552f7358SJed Brown   if (!rank && interpolate && (num_fs > 0)) {
1613552f7358SJed Brown     int fs, f, voff;
1614552f7358SJed Brown     /* Read from ex_get_side_set_ids() */
1615552f7358SJed Brown     int *fs_id;
1616552f7358SJed Brown     /* Read from ex_get_side_set_param() */
1617f958083aSBlaise Bourdin     int num_side_in_set;
1618552f7358SJed Brown     /* Read from ex_get_side_set_node_list() */
1619552f7358SJed Brown     int *fs_vertex_count_list, *fs_vertex_list;
1620ef073283Smichael_afanasiev     /* Read side set labels */
1621ef073283Smichael_afanasiev     char fs_name[MAX_STR_LENGTH+1];
1622552f7358SJed Brown 
1623552f7358SJed Brown     /* Get side set ids */
1624785e854fSJed Brown     ierr = PetscMalloc1(num_fs, &fs_id);CHKERRQ(ierr);
16256de834b4SMatthew G. Knepley     PetscStackCallStandard(ex_get_ids,(exoid, EX_SIDE_SET, fs_id));
1626552f7358SJed Brown     for (fs = 0; fs < num_fs; ++fs) {
16276de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_set_param,(exoid, EX_SIDE_SET, fs_id[fs], &num_side_in_set, NULL));
1628dcca6d9dSJed Brown       ierr = PetscMalloc2(num_side_in_set,&fs_vertex_count_list,num_side_in_set*4,&fs_vertex_list);CHKERRQ(ierr);
16296de834b4SMatthew G. Knepley       PetscStackCallStandard(ex_get_side_set_node_list,(exoid, fs_id[fs], fs_vertex_count_list, fs_vertex_list));
1630ef073283Smichael_afanasiev       /* Get the specific name associated with this side set ID. */
1631ef073283Smichael_afanasiev       int fs_name_err = ex_get_name(exoid, EX_SIDE_SET, fs_id[fs], fs_name);
1632552f7358SJed Brown       for (f = 0, voff = 0; f < num_side_in_set; ++f) {
16330298fd71SBarry Smith         const PetscInt *faces   = NULL;
1634552f7358SJed Brown         PetscInt       faceSize = fs_vertex_count_list[f], numFaces;
1635552f7358SJed Brown         PetscInt       faceVertices[4], v;
1636552f7358SJed Brown 
1637552f7358SJed Brown         if (faceSize > 4) SETERRQ1(comm, PETSC_ERR_ARG_WRONG, "ExodusII side cannot have %d > 4 vertices", faceSize);
1638552f7358SJed Brown         for (v = 0; v < faceSize; ++v, ++voff) {
1639552f7358SJed Brown           faceVertices[v] = fs_vertex_list[voff]+numCells-1;
1640552f7358SJed Brown         }
1641552f7358SJed Brown         ierr = DMPlexGetFullJoin(*dm, faceSize, faceVertices, &numFaces, &faces);CHKERRQ(ierr);
1642552f7358SJed Brown         if (numFaces != 1) SETERRQ3(comm, PETSC_ERR_ARG_WRONG, "Invalid ExodusII side %d in set %d maps to %d faces", f, fs, numFaces);
1643c58f1c22SToby Isaac         ierr = DMSetLabelValue(*dm, "Face Sets", faces[0], fs_id[fs]);CHKERRQ(ierr);
1644ef073283Smichael_afanasiev         /* Only add the label if one has been detected for this side set. */
1645ef073283Smichael_afanasiev         if (!fs_name_err) {
1646ef073283Smichael_afanasiev           ierr = DMSetLabelValue(*dm, fs_name, faces[0], fs_id[fs]);CHKERRQ(ierr);
1647ef073283Smichael_afanasiev         }
1648552f7358SJed Brown         ierr = DMPlexRestoreJoin(*dm, faceSize, faceVertices, &numFaces, &faces);CHKERRQ(ierr);
1649552f7358SJed Brown       }
1650552f7358SJed Brown       ierr = PetscFree2(fs_vertex_count_list,fs_vertex_list);CHKERRQ(ierr);
1651552f7358SJed Brown     }
1652552f7358SJed Brown     ierr = PetscFree(fs_id);CHKERRQ(ierr);
1653552f7358SJed Brown   }
1654b458e8f1SJose E. Roman   PetscFunctionReturn(0);
1655552f7358SJed Brown #else
1656552f7358SJed Brown   SETERRQ(comm, PETSC_ERR_SUP, "This method requires ExodusII support. Reconfigure using --download-exodusii");
1657552f7358SJed Brown #endif
1658552f7358SJed Brown }
1659