16b9fdc0aSVaclav Hapla #include <petsc/private/viewerhdf5impl.h> /*I "petscviewerhdf5.h" I*/ 25c6c1daeSBarry Smith 3bb286ee1SVaclav Hapla static PetscErrorCode PetscViewerHDF5Traverse_Internal(PetscViewer, const char[], PetscBool, PetscBool *, H5O_type_t *); 406db490cSVaclav Hapla static PetscErrorCode PetscViewerHDF5HasAttribute_Internal(PetscViewer, const char[], const char[], PetscBool *); 534e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5GetGroup_Internal(PetscViewer, const char *[]); 606db490cSVaclav Hapla 777717648SVaclav Hapla /*@C 877717648SVaclav Hapla PetscViewerHDF5GetGroup - Get the current HDF5 group name (full path), set with `PetscViewerHDF5PushGroup()`/`PetscViewerHDF5PopGroup()`. 977717648SVaclav Hapla 1020f4b53cSBarry Smith Not Collective 1177717648SVaclav Hapla 1277717648SVaclav Hapla Input Parameters: 1377717648SVaclav Hapla + viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 1477717648SVaclav Hapla - path - (Optional) The path relative to the pushed group 1577717648SVaclav Hapla 1677717648SVaclav Hapla Output Parameter: 1777717648SVaclav Hapla . abspath - The absolute HDF5 path (group) 1877717648SVaclav Hapla 1977717648SVaclav Hapla Level: intermediate 2077717648SVaclav Hapla 2177717648SVaclav Hapla Notes: 2277717648SVaclav Hapla If path starts with '/', it is taken as an absolute path overriding currently pushed group, else path is relative to the current pushed group. 233f423023SBarry Smith `NULL` or empty path means the current pushed group. 2477717648SVaclav Hapla 2577717648SVaclav Hapla The output abspath is newly allocated so needs to be freed. 2677717648SVaclav Hapla 27d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5OpenGroup()`, `PetscViewerHDF5WriteGroup()` 2877717648SVaclav Hapla @*/ 29ce78bad3SBarry Smith PetscErrorCode PetscViewerHDF5GetGroup(PetscViewer viewer, const char path[], const char *abspath[]) 30d71ae5a4SJacob Faibussowitsch { 3177717648SVaclav Hapla size_t len; 324302210dSVaclav Hapla PetscBool relative = PETSC_FALSE; 336c132bc1SVaclav Hapla const char *group; 346c132bc1SVaclav Hapla char buf[PETSC_MAX_PATH_LEN] = ""; 356c132bc1SVaclav Hapla 366c132bc1SVaclav Hapla PetscFunctionBegin; 3777717648SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 384f572ea9SToby Isaac if (path) PetscAssertPointer(path, 2); 394f572ea9SToby Isaac PetscAssertPointer(abspath, 3); 4077717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup_Internal(viewer, &group)); 4177717648SVaclav Hapla PetscCall(PetscStrlen(path, &len)); 4277717648SVaclav Hapla relative = (PetscBool)(!len || path[0] != '/'); 434302210dSVaclav Hapla if (relative) { 44c6a7a370SJeremy L Thompson PetscCall(PetscStrncpy(buf, group, sizeof(buf))); 45c6a7a370SJeremy L Thompson if (!group || len) PetscCall(PetscStrlcat(buf, "/", sizeof(buf))); 46c6a7a370SJeremy L Thompson PetscCall(PetscStrlcat(buf, path, sizeof(buf))); 47ce78bad3SBarry Smith PetscCall(PetscStrallocpy(buf, (char **)abspath)); 484302210dSVaclav Hapla } else { 49ce78bad3SBarry Smith PetscCall(PetscStrallocpy(path, (char **)abspath)); 504302210dSVaclav Hapla } 513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 526c132bc1SVaclav Hapla } 536c132bc1SVaclav Hapla 54d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5CheckNamedObject_Internal(PetscViewer viewer, PetscObject obj) 55d71ae5a4SJacob Faibussowitsch { 56577e0e04SVaclav Hapla PetscBool has; 57577e0e04SVaclav Hapla 58577e0e04SVaclav Hapla PetscFunctionBegin; 599566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5HasObject(viewer, obj, &has)); 60577e0e04SVaclav Hapla if (!has) { 61ce78bad3SBarry Smith const char *group; 6277717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, NULL, &group)); 6377717648SVaclav Hapla SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Object (dataset) \"%s\" not stored in group %s", obj->name, group); 64577e0e04SVaclav Hapla } 653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 66577e0e04SVaclav Hapla } 67577e0e04SVaclav Hapla 68ce78bad3SBarry Smith static PetscErrorCode PetscViewerSetFromOptions_HDF5(PetscViewer v, PetscOptionItems PetscOptionsObject) 69d71ae5a4SJacob Faibussowitsch { 70ee8b9732SVaclav Hapla PetscBool flg = PETSC_FALSE, set; 7182ea9b62SBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)v->data; 7282ea9b62SBarry Smith 7382ea9b62SBarry Smith PetscFunctionBegin; 74d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "HDF5 PetscViewer Options"); 759566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-viewer_hdf5_base_dimension2", "1d Vectors get 2 dimensions in HDF5", "PetscViewerHDF5SetBaseDimension2", hdf5->basedimension2, &hdf5->basedimension2, NULL)); 769566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-viewer_hdf5_sp_output", "Force data to be written in single precision", "PetscViewerHDF5SetSPOutput", hdf5->spoutput, &hdf5->spoutput, NULL)); 779566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-viewer_hdf5_collective", "Enable collective transfer mode", "PetscViewerHDF5SetCollective", flg, &flg, &set)); 789566063dSJacob Faibussowitsch if (set) PetscCall(PetscViewerHDF5SetCollective(v, flg)); 7919a20e4cSMatthew G. Knepley flg = PETSC_FALSE; 809566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-viewer_hdf5_default_timestepping", "Set default timestepping state", "PetscViewerHDF5SetDefaultTimestepping", flg, &flg, &set)); 819566063dSJacob Faibussowitsch if (set) PetscCall(PetscViewerHDF5SetDefaultTimestepping(v, flg)); 82*ea87367fSMatthew G. Knepley PetscCall(PetscOptionsBool("-viewer_hdf5_compress", "Enable compression", "PetscViewerHDF5SetCompress", hdf5->compress, &hdf5->compress, NULL)); 83d0609cedSBarry Smith PetscOptionsHeadEnd(); 843ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 8582ea9b62SBarry Smith } 8682ea9b62SBarry Smith 87d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerView_HDF5(PetscViewer v, PetscViewer viewer) 88d71ae5a4SJacob Faibussowitsch { 891b793a25SVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)v->data; 9003fa8834SVaclav Hapla PetscBool flg; 911b793a25SVaclav Hapla 921b793a25SVaclav Hapla PetscFunctionBegin; 9348a46eb9SPierre Jolivet if (hdf5->filename) PetscCall(PetscViewerASCIIPrintf(viewer, "Filename: %s\n", hdf5->filename)); 949566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Vectors with blocksize 1 saved as 2D datasets: %s\n", PetscBools[hdf5->basedimension2])); 959566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Enforce single precision storage: %s\n", PetscBools[hdf5->spoutput])); 969566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetCollective(v, &flg)); 979566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "MPI-IO transfer mode: %s\n", flg ? "collective" : "independent")); 989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Default timestepping: %s\n", PetscBools[hdf5->defTimestepping])); 99*ea87367fSMatthew G. Knepley PetscCall(PetscViewerASCIIPrintf(viewer, "Compression: %s\n", PetscBools[hdf5->compress])); 1003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1011b793a25SVaclav Hapla } 1021b793a25SVaclav Hapla 103d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileClose_HDF5(PetscViewer viewer) 104d71ae5a4SJacob Faibussowitsch { 1055c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1065c6c1daeSBarry Smith 1075c6c1daeSBarry Smith PetscFunctionBegin; 1089566063dSJacob Faibussowitsch PetscCall(PetscFree(hdf5->filename)); 109792fecdfSBarry Smith if (hdf5->file_id) PetscCallHDF5(H5Fclose, (hdf5->file_id)); 1103ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1115c6c1daeSBarry Smith } 1125c6c1daeSBarry Smith 113d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFlush_HDF5(PetscViewer viewer) 114d71ae5a4SJacob Faibussowitsch { 1156226335aSVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1166226335aSVaclav Hapla 1176226335aSVaclav Hapla PetscFunctionBegin; 118792fecdfSBarry Smith if (hdf5->file_id) PetscCallHDF5(H5Fflush, (hdf5->file_id, H5F_SCOPE_LOCAL)); 1193ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1206226335aSVaclav Hapla } 1216226335aSVaclav Hapla 122d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerDestroy_HDF5(PetscViewer viewer) 123d71ae5a4SJacob Faibussowitsch { 1245c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1255c6c1daeSBarry Smith 1265c6c1daeSBarry Smith PetscFunctionBegin; 127792fecdfSBarry Smith PetscCallHDF5(H5Pclose, (hdf5->dxpl_id)); 1289566063dSJacob Faibussowitsch PetscCall(PetscViewerFileClose_HDF5(viewer)); 1295c6c1daeSBarry Smith while (hdf5->groups) { 1307d964842SVaclav Hapla PetscViewerHDF5GroupList *tmp = hdf5->groups->next; 1315c6c1daeSBarry Smith 1329566063dSJacob Faibussowitsch PetscCall(PetscFree(hdf5->groups->name)); 1339566063dSJacob Faibussowitsch PetscCall(PetscFree(hdf5->groups)); 1345c6c1daeSBarry Smith hdf5->groups = tmp; 1355c6c1daeSBarry Smith } 1369566063dSJacob Faibussowitsch PetscCall(PetscFree(hdf5)); 1379566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetName_C", NULL)); 1389566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetName_C", NULL)); 1399566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetMode_C", NULL)); 1402e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetMode_C", NULL)); 1419566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5SetBaseDimension2_C", NULL)); 1429566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5SetSPOutput_C", NULL)); 1432e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5SetCollective_C", NULL)); 1442e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5GetCollective_C", NULL)); 1459566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5GetDefaultTimestepping_C", NULL)); 1469566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5SetDefaultTimestepping_C", NULL)); 147*ea87367fSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5SetCompress_C", NULL)); 148*ea87367fSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerHDF5GetCompress_C", NULL)); 1493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1505c6c1daeSBarry Smith } 1515c6c1daeSBarry Smith 152d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetMode_HDF5(PetscViewer viewer, PetscFileMode type) 153d71ae5a4SJacob Faibussowitsch { 1545c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1555c6c1daeSBarry Smith 1565c6c1daeSBarry Smith PetscFunctionBegin; 1575c6c1daeSBarry Smith hdf5->btype = type; 1583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1595c6c1daeSBarry Smith } 1605c6c1daeSBarry Smith 161d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetMode_HDF5(PetscViewer viewer, PetscFileMode *type) 162d71ae5a4SJacob Faibussowitsch { 1630b62783dSVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1640b62783dSVaclav Hapla 1650b62783dSVaclav Hapla PetscFunctionBegin; 1660b62783dSVaclav Hapla *type = hdf5->btype; 1673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1680b62783dSVaclav Hapla } 1690b62783dSVaclav Hapla 170d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5SetBaseDimension2_HDF5(PetscViewer viewer, PetscBool flg) 171d71ae5a4SJacob Faibussowitsch { 17282ea9b62SBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 17382ea9b62SBarry Smith 17482ea9b62SBarry Smith PetscFunctionBegin; 17582ea9b62SBarry Smith hdf5->basedimension2 = flg; 1763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 17782ea9b62SBarry Smith } 17882ea9b62SBarry Smith 179df863907SAlex Fikl /*@ 18082ea9b62SBarry Smith PetscViewerHDF5SetBaseDimension2 - Vectors of 1 dimension (i.e. bs/dof is 1) will be saved in the HDF5 file with a 18182ea9b62SBarry Smith dimension of 2. 18282ea9b62SBarry Smith 183c3339decSBarry Smith Logically Collective 18482ea9b62SBarry Smith 18582ea9b62SBarry Smith Input Parameters: 186811af0c4SBarry Smith + viewer - the `PetscViewer`; if it is a `PETSCVIEWERHDF5` then this command is ignored 187811af0c4SBarry Smith - flg - if `PETSC_TRUE` the vector will always have at least a dimension of 2 even if that first dimension is of size 1 18882ea9b62SBarry Smith 189811af0c4SBarry Smith Options Database Key: 19082ea9b62SBarry Smith . -viewer_hdf5_base_dimension2 - turns on (true) or off (false) using a dimension of 2 in the HDF5 file even if the bs/dof of the vector is 1 19182ea9b62SBarry Smith 1923f423023SBarry Smith Level: intermediate 1933f423023SBarry Smith 194811af0c4SBarry Smith Note: 19595452b02SPatrick Sanan Setting this option allegedly makes code that reads the HDF5 in easier since they do not have a "special case" of a bs/dof 19682ea9b62SBarry Smith of one when the dimension is lower. Others think the option is crazy. 19782ea9b62SBarry Smith 198aec76313SJacob Faibussowitsch .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerFileSetMode()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerBinaryOpen()` 19982ea9b62SBarry Smith @*/ 200d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5SetBaseDimension2(PetscViewer viewer, PetscBool flg) 201d71ae5a4SJacob Faibussowitsch { 20282ea9b62SBarry Smith PetscFunctionBegin; 20382ea9b62SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 204cac4c232SBarry Smith PetscTryMethod(viewer, "PetscViewerHDF5SetBaseDimension2_C", (PetscViewer, PetscBool), (viewer, flg)); 2053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 20682ea9b62SBarry Smith } 20782ea9b62SBarry Smith 208df863907SAlex Fikl /*@ 20982ea9b62SBarry Smith PetscViewerHDF5GetBaseDimension2 - Vectors of 1 dimension (i.e. bs/dof is 1) will be saved in the HDF5 file with a 21082ea9b62SBarry Smith dimension of 2. 21182ea9b62SBarry Smith 212c3339decSBarry Smith Logically Collective 21382ea9b62SBarry Smith 21482ea9b62SBarry Smith Input Parameter: 215811af0c4SBarry Smith . viewer - the `PetscViewer`, must be `PETSCVIEWERHDF5` 21682ea9b62SBarry Smith 21782ea9b62SBarry Smith Output Parameter: 218811af0c4SBarry Smith . flg - if `PETSC_TRUE` the vector will always have at least a dimension of 2 even if that first dimension is of size 1 21982ea9b62SBarry Smith 2203f423023SBarry Smith Level: intermediate 2213f423023SBarry Smith 222811af0c4SBarry Smith Note: 22395452b02SPatrick Sanan Setting this option allegedly makes code that reads the HDF5 in easier since they do not have a "special case" of a bs/dof 22482ea9b62SBarry Smith of one when the dimension is lower. Others think the option is crazy. 22582ea9b62SBarry Smith 226d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerFileSetMode()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerBinaryOpen()` 22782ea9b62SBarry Smith @*/ 228d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetBaseDimension2(PetscViewer viewer, PetscBool *flg) 229d71ae5a4SJacob Faibussowitsch { 23082ea9b62SBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 23182ea9b62SBarry Smith 23282ea9b62SBarry Smith PetscFunctionBegin; 23382ea9b62SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 23482ea9b62SBarry Smith *flg = hdf5->basedimension2; 2353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23682ea9b62SBarry Smith } 23782ea9b62SBarry Smith 238d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5SetSPOutput_HDF5(PetscViewer viewer, PetscBool flg) 239d71ae5a4SJacob Faibussowitsch { 2409a0502c6SHåkon Strandenes PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 2419a0502c6SHåkon Strandenes 2429a0502c6SHåkon Strandenes PetscFunctionBegin; 2439a0502c6SHåkon Strandenes hdf5->spoutput = flg; 2443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2459a0502c6SHåkon Strandenes } 2469a0502c6SHåkon Strandenes 247df863907SAlex Fikl /*@ 2489a0502c6SHåkon Strandenes PetscViewerHDF5SetSPOutput - Data is written to disk in single precision even if PETSc is 249811af0c4SBarry Smith compiled with double precision `PetscReal`. 2509a0502c6SHåkon Strandenes 251c3339decSBarry Smith Logically Collective 2529a0502c6SHåkon Strandenes 2539a0502c6SHåkon Strandenes Input Parameters: 254811af0c4SBarry Smith + viewer - the PetscViewer; if it is a `PETSCVIEWERHDF5` then this command is ignored 255811af0c4SBarry Smith - flg - if `PETSC_TRUE` the data will be written to disk with single precision 2569a0502c6SHåkon Strandenes 257811af0c4SBarry Smith Options Database Key: 2589a0502c6SHåkon Strandenes . -viewer_hdf5_sp_output - turns on (true) or off (false) output in single precision 2599a0502c6SHåkon Strandenes 2603f423023SBarry Smith Level: intermediate 2613f423023SBarry Smith 262811af0c4SBarry Smith Note: 26395452b02SPatrick Sanan Setting this option does not make any difference if PETSc is compiled with single precision 2649a0502c6SHåkon Strandenes in the first place. It does not affect reading datasets (HDF5 handle this internally). 2659a0502c6SHåkon Strandenes 266d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerFileSetMode()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerBinaryOpen()`, 267811af0c4SBarry Smith `PetscReal`, `PetscViewerHDF5GetSPOutput()` 2689a0502c6SHåkon Strandenes @*/ 269d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5SetSPOutput(PetscViewer viewer, PetscBool flg) 270d71ae5a4SJacob Faibussowitsch { 2719a0502c6SHåkon Strandenes PetscFunctionBegin; 2729a0502c6SHåkon Strandenes PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 273cac4c232SBarry Smith PetscTryMethod(viewer, "PetscViewerHDF5SetSPOutput_C", (PetscViewer, PetscBool), (viewer, flg)); 2743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2759a0502c6SHåkon Strandenes } 2769a0502c6SHåkon Strandenes 277df863907SAlex Fikl /*@ 2789a0502c6SHåkon Strandenes PetscViewerHDF5GetSPOutput - Data is written to disk in single precision even if PETSc is 279811af0c4SBarry Smith compiled with double precision `PetscReal`. 2809a0502c6SHåkon Strandenes 281c3339decSBarry Smith Logically Collective 2829a0502c6SHåkon Strandenes 2839a0502c6SHåkon Strandenes Input Parameter: 284811af0c4SBarry Smith . viewer - the PetscViewer, must be of type `PETSCVIEWERHDF5` 2859a0502c6SHåkon Strandenes 2869a0502c6SHåkon Strandenes Output Parameter: 287811af0c4SBarry Smith . flg - if `PETSC_TRUE` the data will be written to disk with single precision 2889a0502c6SHåkon Strandenes 2899a0502c6SHåkon Strandenes Level: intermediate 2909a0502c6SHåkon Strandenes 291d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerFileSetMode()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerBinaryOpen()`, 292811af0c4SBarry Smith `PetscReal`, `PetscViewerHDF5SetSPOutput()` 2939a0502c6SHåkon Strandenes @*/ 294d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetSPOutput(PetscViewer viewer, PetscBool *flg) 295d71ae5a4SJacob Faibussowitsch { 2969a0502c6SHåkon Strandenes PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 2979a0502c6SHåkon Strandenes 2989a0502c6SHåkon Strandenes PetscFunctionBegin; 2999a0502c6SHåkon Strandenes PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 3009a0502c6SHåkon Strandenes *flg = hdf5->spoutput; 3013ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3029a0502c6SHåkon Strandenes } 3039a0502c6SHåkon Strandenes 304d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5SetCollective_HDF5(PetscViewer viewer, PetscBool flg) 305d71ae5a4SJacob Faibussowitsch { 306ee8b9732SVaclav Hapla PetscFunctionBegin; 307ee8b9732SVaclav Hapla /* H5FD_MPIO_COLLECTIVE is wrong in hdf5 1.10.2, and is the same as H5FD_MPIO_INDEPENDENT in earlier versions 308ee8b9732SVaclav Hapla - see e.g. https://gitlab.cosma.dur.ac.uk/swift/swiftsim/issues/431 */ 309c796909eSBarry Smith #if H5_VERSION_GE(1, 10, 3) && defined(H5_HAVE_PARALLEL) 310ee8b9732SVaclav Hapla { 311ee8b9732SVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 312792fecdfSBarry Smith PetscCallHDF5(H5Pset_dxpl_mpio, (hdf5->dxpl_id, flg ? H5FD_MPIO_COLLECTIVE : H5FD_MPIO_INDEPENDENT)); 313ee8b9732SVaclav Hapla } 314ee8b9732SVaclav Hapla #else 3159566063dSJacob Faibussowitsch if (flg) PetscCall(PetscPrintf(PetscObjectComm((PetscObject)viewer), "Warning: PetscViewerHDF5SetCollective(viewer,PETSC_TRUE) is ignored for HDF5 versions prior to 1.10.3 or if built without MPI support\n")); 316ee8b9732SVaclav Hapla #endif 3173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 318ee8b9732SVaclav Hapla } 319ee8b9732SVaclav Hapla 320ee8b9732SVaclav Hapla /*@ 321ee8b9732SVaclav Hapla PetscViewerHDF5SetCollective - Use collective MPI-IO transfer mode for HDF5 reads and writes. 322ee8b9732SVaclav Hapla 323ee8b9732SVaclav Hapla Logically Collective; flg must contain common value 324ee8b9732SVaclav Hapla 325ee8b9732SVaclav Hapla Input Parameters: 326811af0c4SBarry Smith + viewer - the `PetscViewer`; if it is not `PETSCVIEWERHDF5` then this command is ignored 327811af0c4SBarry Smith - flg - `PETSC_TRUE` for collective mode; `PETSC_FALSE` for independent mode (default) 328ee8b9732SVaclav Hapla 329811af0c4SBarry Smith Options Database Key: 330ee8b9732SVaclav Hapla . -viewer_hdf5_collective - turns on (true) or off (false) collective transfers 331ee8b9732SVaclav Hapla 3323f423023SBarry Smith Level: intermediate 3333f423023SBarry Smith 334811af0c4SBarry Smith Note: 335ee8b9732SVaclav Hapla Collective mode gives the MPI-IO layer underneath HDF5 a chance to do some additional collective optimizations and hence can perform better. 33653effed7SBarry Smith However, this works correctly only since HDF5 1.10.3 and if HDF5 is installed for MPI; hence, we ignore this setting for older versions. 337ee8b9732SVaclav Hapla 338aec76313SJacob Faibussowitsch Developer Notes: 339811af0c4SBarry Smith In the HDF5 layer, `PETSC_TRUE` / `PETSC_FALSE` means `H5Pset_dxpl_mpio()` is called with `H5FD_MPIO_COLLECTIVE` / `H5FD_MPIO_INDEPENDENT`, respectively. 340ee8b9732SVaclav Hapla This in turn means use of MPI_File_{read,write}_all / MPI_File_{read,write} in the MPI-IO layer, respectively. 341ee8b9732SVaclav Hapla See HDF5 documentation and MPI-IO documentation for details. 342ee8b9732SVaclav Hapla 343d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5GetCollective()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerHDF5Open()` 344ee8b9732SVaclav Hapla @*/ 345d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5SetCollective(PetscViewer viewer, PetscBool flg) 346d71ae5a4SJacob Faibussowitsch { 347ee8b9732SVaclav Hapla PetscFunctionBegin; 348ee8b9732SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 349ee8b9732SVaclav Hapla PetscValidLogicalCollectiveBool(viewer, flg, 2); 350cac4c232SBarry Smith PetscTryMethod(viewer, "PetscViewerHDF5SetCollective_C", (PetscViewer, PetscBool), (viewer, flg)); 3513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 352ee8b9732SVaclav Hapla } 353ee8b9732SVaclav Hapla 354d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5GetCollective_HDF5(PetscViewer viewer, PetscBool *flg) 355d71ae5a4SJacob Faibussowitsch { 356c796909eSBarry Smith #if defined(H5_HAVE_PARALLEL) 357ee8b9732SVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 358ee8b9732SVaclav Hapla H5FD_mpio_xfer_t mode; 359c796909eSBarry Smith #endif 360ee8b9732SVaclav Hapla 361ee8b9732SVaclav Hapla PetscFunctionBegin; 362c796909eSBarry Smith #if !defined(H5_HAVE_PARALLEL) 363c796909eSBarry Smith *flg = PETSC_FALSE; 364c796909eSBarry Smith #else 365792fecdfSBarry Smith PetscCallHDF5(H5Pget_dxpl_mpio, (hdf5->dxpl_id, &mode)); 366ee8b9732SVaclav Hapla *flg = (mode == H5FD_MPIO_COLLECTIVE) ? PETSC_TRUE : PETSC_FALSE; 367c796909eSBarry Smith #endif 3683ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 369ee8b9732SVaclav Hapla } 370ee8b9732SVaclav Hapla 371ee8b9732SVaclav Hapla /*@ 372ee8b9732SVaclav Hapla PetscViewerHDF5GetCollective - Return flag whether collective MPI-IO transfer mode is used for HDF5 reads and writes. 373ee8b9732SVaclav Hapla 374ee8b9732SVaclav Hapla Not Collective 375ee8b9732SVaclav Hapla 37620f4b53cSBarry Smith Input Parameter: 377811af0c4SBarry Smith . viewer - the `PETSCVIEWERHDF5` `PetscViewer` 378ee8b9732SVaclav Hapla 37920f4b53cSBarry Smith Output Parameter: 380ee8b9732SVaclav Hapla . flg - the flag 381ee8b9732SVaclav Hapla 382ee8b9732SVaclav Hapla Level: intermediate 383ee8b9732SVaclav Hapla 384811af0c4SBarry Smith Note: 385811af0c4SBarry Smith This setting works correctly only since HDF5 1.10.3 and if HDF5 was installed for MPI. For older versions, `PETSC_FALSE` will be always returned. 386811af0c4SBarry Smith For more details, see `PetscViewerHDF5SetCollective()`. 387ee8b9732SVaclav Hapla 388d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5SetCollective()`, `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerHDF5Open()` 389ee8b9732SVaclav Hapla @*/ 390d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetCollective(PetscViewer viewer, PetscBool *flg) 391d71ae5a4SJacob Faibussowitsch { 392ee8b9732SVaclav Hapla PetscFunctionBegin; 393ee8b9732SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 3944f572ea9SToby Isaac PetscAssertPointer(flg, 2); 395ee8b9732SVaclav Hapla 396cac4c232SBarry Smith PetscUseMethod(viewer, "PetscViewerHDF5GetCollective_C", (PetscViewer, PetscBool *), (viewer, flg)); 3973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 398ee8b9732SVaclav Hapla } 399ee8b9732SVaclav Hapla 400d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetName_HDF5(PetscViewer viewer, const char name[]) 401d71ae5a4SJacob Faibussowitsch { 4025c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 4035c6c1daeSBarry Smith hid_t plist_id; 4045c6c1daeSBarry Smith 4055c6c1daeSBarry Smith PetscFunctionBegin; 406792fecdfSBarry Smith if (hdf5->file_id) PetscCallHDF5(H5Fclose, (hdf5->file_id)); 4079566063dSJacob Faibussowitsch if (hdf5->filename) PetscCall(PetscFree(hdf5->filename)); 4089566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(name, &hdf5->filename)); 4095c6c1daeSBarry Smith /* Set up file access property list with parallel I/O access */ 410792fecdfSBarry Smith PetscCallHDF5Return(plist_id, H5Pcreate, (H5P_FILE_ACCESS)); 411c796909eSBarry Smith #if defined(H5_HAVE_PARALLEL) 412792fecdfSBarry Smith PetscCallHDF5(H5Pset_fapl_mpio, (plist_id, PetscObjectComm((PetscObject)viewer), MPI_INFO_NULL)); 413c796909eSBarry Smith #endif 4145c6c1daeSBarry Smith /* Create or open the file collectively */ 4155c6c1daeSBarry Smith switch (hdf5->btype) { 4165c6c1daeSBarry Smith case FILE_MODE_READ: 4178a2871f6SBarry Smith if (PetscDefined(USE_DEBUG)) { 4188a2871f6SBarry Smith PetscMPIInt rank; 4198a2871f6SBarry Smith PetscBool flg; 4208a2871f6SBarry Smith 4219566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank)); 4228a2871f6SBarry Smith if (rank == 0) { 4239566063dSJacob Faibussowitsch PetscCall(PetscTestFile(hdf5->filename, 'r', &flg)); 4245f80ce2aSJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "File %s requested for reading does not exist", hdf5->filename); 4258a2871f6SBarry Smith } 4269566063dSJacob Faibussowitsch PetscCallMPI(MPI_Barrier(PetscObjectComm((PetscObject)viewer))); 4278a2871f6SBarry Smith } 428792fecdfSBarry Smith PetscCallHDF5Return(hdf5->file_id, H5Fopen, (name, H5F_ACC_RDONLY, plist_id)); 4295c6c1daeSBarry Smith break; 4305c6c1daeSBarry Smith case FILE_MODE_APPEND: 4319371c9d4SSatish Balay case FILE_MODE_UPDATE: { 4327e4fd573SVaclav Hapla PetscBool flg; 4339566063dSJacob Faibussowitsch PetscCall(PetscTestFile(hdf5->filename, 'r', &flg)); 434792fecdfSBarry Smith if (flg) PetscCallHDF5Return(hdf5->file_id, H5Fopen, (name, H5F_ACC_RDWR, plist_id)); 435792fecdfSBarry Smith else PetscCallHDF5Return(hdf5->file_id, H5Fcreate, (name, H5F_ACC_EXCL, H5P_DEFAULT, plist_id)); 4365c6c1daeSBarry Smith break; 4377e4fd573SVaclav Hapla } 438d71ae5a4SJacob Faibussowitsch case FILE_MODE_WRITE: 439d71ae5a4SJacob Faibussowitsch PetscCallHDF5Return(hdf5->file_id, H5Fcreate, (name, H5F_ACC_TRUNC, H5P_DEFAULT, plist_id)); 440d71ae5a4SJacob Faibussowitsch break; 441d71ae5a4SJacob Faibussowitsch case FILE_MODE_UNDEFINED: 442d71ae5a4SJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()"); 443d71ae5a4SJacob Faibussowitsch default: 444d71ae5a4SJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Unsupported file mode %s", PetscFileModes[hdf5->btype]); 4455c6c1daeSBarry Smith } 4465f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->file_id >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "H5Fcreate failed for %s", name); 447792fecdfSBarry Smith PetscCallHDF5(H5Pclose, (plist_id)); 448671abe24SVaclav Hapla PetscCall(PetscViewerHDF5ResetAttachedDMPlexStorageVersion(viewer)); 4493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4505c6c1daeSBarry Smith } 4515c6c1daeSBarry Smith 452d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetName_HDF5(PetscViewer viewer, const char **name) 453d71ae5a4SJacob Faibussowitsch { 454d1232d7fSToby Isaac PetscViewer_HDF5 *vhdf5 = (PetscViewer_HDF5 *)viewer->data; 455d1232d7fSToby Isaac 456d1232d7fSToby Isaac PetscFunctionBegin; 457d1232d7fSToby Isaac *name = vhdf5->filename; 4583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 459d1232d7fSToby Isaac } 460d1232d7fSToby Isaac 461d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerSetUp_HDF5(PetscViewer viewer) 462d71ae5a4SJacob Faibussowitsch { 4635dc64a97SVaclav Hapla /* 464b723ab35SVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5*) viewer->data; 4655dc64a97SVaclav Hapla */ 466b723ab35SVaclav Hapla 467b723ab35SVaclav Hapla PetscFunctionBegin; 4683ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 469b723ab35SVaclav Hapla } 470b723ab35SVaclav Hapla 471d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5SetDefaultTimestepping_HDF5(PetscViewer viewer, PetscBool flg) 472d71ae5a4SJacob Faibussowitsch { 47319a20e4cSMatthew G. Knepley PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 47419a20e4cSMatthew G. Knepley 47519a20e4cSMatthew G. Knepley PetscFunctionBegin; 47619a20e4cSMatthew G. Knepley hdf5->defTimestepping = flg; 4773ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 47819a20e4cSMatthew G. Knepley } 47919a20e4cSMatthew G. Knepley 480d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5GetDefaultTimestepping_HDF5(PetscViewer viewer, PetscBool *flg) 481d71ae5a4SJacob Faibussowitsch { 48219a20e4cSMatthew G. Knepley PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 48319a20e4cSMatthew G. Knepley 48419a20e4cSMatthew G. Knepley PetscFunctionBegin; 48519a20e4cSMatthew G. Knepley *flg = hdf5->defTimestepping; 4863ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 48719a20e4cSMatthew G. Knepley } 48819a20e4cSMatthew G. Knepley 48919a20e4cSMatthew G. Knepley /*@ 49019a20e4cSMatthew G. Knepley PetscViewerHDF5SetDefaultTimestepping - Set the flag for default timestepping 49119a20e4cSMatthew G. Knepley 492c3339decSBarry Smith Logically Collective 49319a20e4cSMatthew G. Knepley 49419a20e4cSMatthew G. Knepley Input Parameters: 495811af0c4SBarry Smith + viewer - the `PetscViewer`; if it is not `PETSCVIEWERHDF5` then this command is ignored 496811af0c4SBarry Smith - flg - if `PETSC_TRUE` we will assume that timestepping is on 49719a20e4cSMatthew G. Knepley 498811af0c4SBarry Smith Options Database Key: 49919a20e4cSMatthew G. Knepley . -viewer_hdf5_default_timestepping - turns on (true) or off (false) default timestepping 50019a20e4cSMatthew G. Knepley 5013f423023SBarry Smith Level: intermediate 5023f423023SBarry Smith 503811af0c4SBarry Smith Note: 50419a20e4cSMatthew G. Knepley If the timestepping attribute is not found for an object, then the default timestepping is used 50519a20e4cSMatthew G. Knepley 506d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5GetDefaultTimestepping()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5GetTimestep()` 50719a20e4cSMatthew G. Knepley @*/ 508d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5SetDefaultTimestepping(PetscViewer viewer, PetscBool flg) 509d71ae5a4SJacob Faibussowitsch { 51019a20e4cSMatthew G. Knepley PetscFunctionBegin; 51119a20e4cSMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 512cac4c232SBarry Smith PetscTryMethod(viewer, "PetscViewerHDF5SetDefaultTimestepping_C", (PetscViewer, PetscBool), (viewer, flg)); 5133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 51419a20e4cSMatthew G. Knepley } 51519a20e4cSMatthew G. Knepley 51619a20e4cSMatthew G. Knepley /*@ 51719a20e4cSMatthew G. Knepley PetscViewerHDF5GetDefaultTimestepping - Get the flag for default timestepping 51819a20e4cSMatthew G. Knepley 51920f4b53cSBarry Smith Not Collective 52019a20e4cSMatthew G. Knepley 52119a20e4cSMatthew G. Knepley Input Parameter: 522811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 52319a20e4cSMatthew G. Knepley 52419a20e4cSMatthew G. Knepley Output Parameter: 525811af0c4SBarry Smith . flg - if `PETSC_TRUE` we will assume that timestepping is on 52619a20e4cSMatthew G. Knepley 52719a20e4cSMatthew G. Knepley Level: intermediate 52819a20e4cSMatthew G. Knepley 529d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5SetDefaultTimestepping()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5GetTimestep()` 53019a20e4cSMatthew G. Knepley @*/ 531d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetDefaultTimestepping(PetscViewer viewer, PetscBool *flg) 532d71ae5a4SJacob Faibussowitsch { 53319a20e4cSMatthew G. Knepley PetscFunctionBegin; 53419a20e4cSMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 535cac4c232SBarry Smith PetscUseMethod(viewer, "PetscViewerHDF5GetDefaultTimestepping_C", (PetscViewer, PetscBool *), (viewer, flg)); 5363ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 53719a20e4cSMatthew G. Knepley } 53819a20e4cSMatthew G. Knepley 539*ea87367fSMatthew G. Knepley static PetscErrorCode PetscViewerHDF5SetCompress_HDF5(PetscViewer viewer, PetscBool flg) 540*ea87367fSMatthew G. Knepley { 541*ea87367fSMatthew G. Knepley PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 542*ea87367fSMatthew G. Knepley 543*ea87367fSMatthew G. Knepley PetscFunctionBegin; 544*ea87367fSMatthew G. Knepley hdf5->compress = flg; 545*ea87367fSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 546*ea87367fSMatthew G. Knepley } 547*ea87367fSMatthew G. Knepley 548*ea87367fSMatthew G. Knepley static PetscErrorCode PetscViewerHDF5GetCompress_HDF5(PetscViewer viewer, PetscBool *flg) 549*ea87367fSMatthew G. Knepley { 550*ea87367fSMatthew G. Knepley PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 551*ea87367fSMatthew G. Knepley 552*ea87367fSMatthew G. Knepley PetscFunctionBegin; 553*ea87367fSMatthew G. Knepley *flg = hdf5->compress; 554*ea87367fSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 555*ea87367fSMatthew G. Knepley } 556*ea87367fSMatthew G. Knepley 557*ea87367fSMatthew G. Knepley /*@ 558*ea87367fSMatthew G. Knepley PetscViewerHDF5SetCompress - Set the flag for compression 559*ea87367fSMatthew G. Knepley 560*ea87367fSMatthew G. Knepley Logically Collective 561*ea87367fSMatthew G. Knepley 562*ea87367fSMatthew G. Knepley Input Parameters: 563*ea87367fSMatthew G. Knepley + viewer - the `PetscViewer`; if it is not `PETSCVIEWERHDF5` then this command is ignored 564*ea87367fSMatthew G. Knepley - flg - if `PETSC_TRUE` we will turn on compression 565*ea87367fSMatthew G. Knepley 566*ea87367fSMatthew G. Knepley Options Database Key: 567*ea87367fSMatthew G. Knepley . -viewer_hdf5_compress - turns on (true) or off (false) compression 568*ea87367fSMatthew G. Knepley 569*ea87367fSMatthew G. Knepley Level: intermediate 570*ea87367fSMatthew G. Knepley 571*ea87367fSMatthew G. Knepley .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5GetCompress()` 572*ea87367fSMatthew G. Knepley @*/ 573*ea87367fSMatthew G. Knepley PetscErrorCode PetscViewerHDF5SetCompress(PetscViewer viewer, PetscBool flg) 574*ea87367fSMatthew G. Knepley { 575*ea87367fSMatthew G. Knepley PetscFunctionBegin; 576*ea87367fSMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 577*ea87367fSMatthew G. Knepley PetscTryMethod(viewer, "PetscViewerHDF5SetCompress_C", (PetscViewer, PetscBool), (viewer, flg)); 578*ea87367fSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 579*ea87367fSMatthew G. Knepley } 580*ea87367fSMatthew G. Knepley 581*ea87367fSMatthew G. Knepley /*@ 582*ea87367fSMatthew G. Knepley PetscViewerHDF5GetCompress - Get the flag for compression 583*ea87367fSMatthew G. Knepley 584*ea87367fSMatthew G. Knepley Not Collective 585*ea87367fSMatthew G. Knepley 586*ea87367fSMatthew G. Knepley Input Parameter: 587*ea87367fSMatthew G. Knepley . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 588*ea87367fSMatthew G. Knepley 589*ea87367fSMatthew G. Knepley Output Parameter: 590*ea87367fSMatthew G. Knepley . flg - if `PETSC_TRUE` we will turn on compression 591*ea87367fSMatthew G. Knepley 592*ea87367fSMatthew G. Knepley Level: intermediate 593*ea87367fSMatthew G. Knepley 594*ea87367fSMatthew G. Knepley .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5SetCompress()` 595*ea87367fSMatthew G. Knepley @*/ 596*ea87367fSMatthew G. Knepley PetscErrorCode PetscViewerHDF5GetCompress(PetscViewer viewer, PetscBool *flg) 597*ea87367fSMatthew G. Knepley { 598*ea87367fSMatthew G. Knepley PetscFunctionBegin; 599*ea87367fSMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 600*ea87367fSMatthew G. Knepley PetscUseMethod(viewer, "PetscViewerHDF5GetCompress_C", (PetscViewer, PetscBool *), (viewer, flg)); 601*ea87367fSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 602*ea87367fSMatthew G. Knepley } 603*ea87367fSMatthew G. Knepley 6048556b5ebSBarry Smith /*MC 6058556b5ebSBarry Smith PETSCVIEWERHDF5 - A viewer that writes to an HDF5 file 6068556b5ebSBarry Smith 607811af0c4SBarry Smith Level: beginner 608811af0c4SBarry Smith 609d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerHDF5Open()`, `PetscViewerStringSPrintf()`, `PetscViewerSocketOpen()`, `PetscViewerDrawOpen()`, `PETSCVIEWERSOCKET`, 610db781477SPatrick Sanan `PetscViewerCreate()`, `PetscViewerASCIIOpen()`, `PetscViewerBinaryOpen()`, `PETSCVIEWERBINARY`, `PETSCVIEWERDRAW`, `PETSCVIEWERSTRING`, 611db781477SPatrick Sanan `PetscViewerMatlabOpen()`, `VecView()`, `DMView()`, `PetscViewerMatlabPutArray()`, `PETSCVIEWERASCII`, `PETSCVIEWERMATLAB`, 612db781477SPatrick Sanan `PetscViewerFileSetName()`, `PetscViewerFileSetMode()`, `PetscViewerFormat`, `PetscViewerType`, `PetscViewerSetType()` 6138556b5ebSBarry Smith M*/ 614d1232d7fSToby Isaac 615d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PetscViewerCreate_HDF5(PetscViewer v) 616d71ae5a4SJacob Faibussowitsch { 6175c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5; 6185c6c1daeSBarry Smith 6195c6c1daeSBarry Smith PetscFunctionBegin; 62099335c34SVaclav Hapla #if !defined(H5_HAVE_PARALLEL) 62199335c34SVaclav Hapla { 62299335c34SVaclav Hapla PetscMPIInt size; 6239566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)v), &size)); 6245f80ce2aSJacob Faibussowitsch PetscCheck(size <= 1, PetscObjectComm((PetscObject)v), PETSC_ERR_SUP, "Cannot use parallel HDF5 viewer since the given HDF5 does not support parallel I/O (H5_HAVE_PARALLEL is unset)"); 62599335c34SVaclav Hapla } 62699335c34SVaclav Hapla #endif 62799335c34SVaclav Hapla 6284dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&hdf5)); 6295c6c1daeSBarry Smith 6305c6c1daeSBarry Smith v->data = (void *)hdf5; 6315c6c1daeSBarry Smith v->ops->destroy = PetscViewerDestroy_HDF5; 63282ea9b62SBarry Smith v->ops->setfromoptions = PetscViewerSetFromOptions_HDF5; 633b723ab35SVaclav Hapla v->ops->setup = PetscViewerSetUp_HDF5; 6341b793a25SVaclav Hapla v->ops->view = PetscViewerView_HDF5; 6356226335aSVaclav Hapla v->ops->flush = PetscViewerFlush_HDF5; 6367e4fd573SVaclav Hapla hdf5->btype = FILE_MODE_UNDEFINED; 637908793a3SLisandro Dalcin hdf5->filename = NULL; 6385c6c1daeSBarry Smith hdf5->timestep = -1; 6390298fd71SBarry Smith hdf5->groups = NULL; 6405c6c1daeSBarry Smith 641792fecdfSBarry Smith PetscCallHDF5Return(hdf5->dxpl_id, H5Pcreate, (H5P_DATASET_XFER)); 6429c5072fbSVaclav Hapla 6439566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetName_C", PetscViewerFileSetName_HDF5)); 6449566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetName_C", PetscViewerFileGetName_HDF5)); 6459566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetMode_C", PetscViewerFileSetMode_HDF5)); 6469566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetMode_C", PetscViewerFileGetMode_HDF5)); 6479566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5SetBaseDimension2_C", PetscViewerHDF5SetBaseDimension2_HDF5)); 6489566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5SetSPOutput_C", PetscViewerHDF5SetSPOutput_HDF5)); 6499566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5SetCollective_C", PetscViewerHDF5SetCollective_HDF5)); 6509566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5GetCollective_C", PetscViewerHDF5GetCollective_HDF5)); 6519566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5GetDefaultTimestepping_C", PetscViewerHDF5GetDefaultTimestepping_HDF5)); 6529566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5SetDefaultTimestepping_C", PetscViewerHDF5SetDefaultTimestepping_HDF5)); 653*ea87367fSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5GetCompress_C", PetscViewerHDF5GetCompress_HDF5)); 654*ea87367fSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerHDF5SetCompress_C", PetscViewerHDF5SetCompress_HDF5)); 6553ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6565c6c1daeSBarry Smith } 6575c6c1daeSBarry Smith 6585c6c1daeSBarry Smith /*@C 659811af0c4SBarry Smith PetscViewerHDF5Open - Opens a file for HDF5 input/output as a `PETSCVIEWERHDF5` `PetscViewer` 6605c6c1daeSBarry Smith 661d083f849SBarry Smith Collective 6625c6c1daeSBarry Smith 6635c6c1daeSBarry Smith Input Parameters: 6645c6c1daeSBarry Smith + comm - MPI communicator 6655c6c1daeSBarry Smith . name - name of file 6665c6c1daeSBarry Smith - type - type of file 6675c6c1daeSBarry Smith 6685c6c1daeSBarry Smith Output Parameter: 669811af0c4SBarry Smith . hdf5v - `PetscViewer` for HDF5 input/output to use with the specified file 6705c6c1daeSBarry Smith 671811af0c4SBarry Smith Options Database Keys: 672a2b725a8SWilliam Gropp + -viewer_hdf5_base_dimension2 - turns on (true) or off (false) using a dimension of 2 in the HDF5 file even if the bs/dof of the vector is 1 673a2b725a8SWilliam Gropp - -viewer_hdf5_sp_output - forces (if true) the viewer to write data in single precision independent on the precision of PetscReal 67482ea9b62SBarry Smith 6755c6c1daeSBarry Smith Level: beginner 6765c6c1daeSBarry Smith 6777e4fd573SVaclav Hapla Notes: 6787e4fd573SVaclav Hapla Reading is always available, regardless of the mode. Available modes are 679811af0c4SBarry Smith .vb 680811af0c4SBarry Smith FILE_MODE_READ - open existing HDF5 file for read only access, fail if file does not exist [H5Fopen() with H5F_ACC_RDONLY] 681811af0c4SBarry Smith FILE_MODE_WRITE - if file exists, fully overwrite it, else create new HDF5 file [H5FcreateH5Fcreate() with H5F_ACC_TRUNC] 682811af0c4SBarry Smith FILE_MODE_APPEND - if file exists, keep existing contents [H5Fopen() with H5F_ACC_RDWR], else create new HDF5 file [H5FcreateH5Fcreate() with H5F_ACC_EXCL] 683811af0c4SBarry Smith FILE_MODE_UPDATE - same as FILE_MODE_APPEND 684811af0c4SBarry Smith .ve 6857e4fd573SVaclav Hapla 686da81f932SPierre Jolivet In case of `FILE_MODE_APPEND` / `FILE_MODE_UPDATE`, any stored object (dataset, attribute) can be selectively overwritten if the same fully qualified name (/group/path/to/object) is specified. 6877e4fd573SVaclav Hapla 6885c6c1daeSBarry Smith This PetscViewer should be destroyed with PetscViewerDestroy(). 6895c6c1daeSBarry Smith 690d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerASCIIOpen()`, `PetscViewerPushFormat()`, `PetscViewerDestroy()`, `PetscViewerHDF5SetBaseDimension2()`, 691db781477SPatrick Sanan `PetscViewerHDF5SetSPOutput()`, `PetscViewerHDF5GetBaseDimension2()`, `VecView()`, `MatView()`, `VecLoad()`, 692db781477SPatrick Sanan `MatLoad()`, `PetscFileMode`, `PetscViewer`, `PetscViewerSetType()`, `PetscViewerFileSetMode()`, `PetscViewerFileSetName()` 6935c6c1daeSBarry Smith @*/ 694d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5Open(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *hdf5v) 695d71ae5a4SJacob Faibussowitsch { 6965c6c1daeSBarry Smith PetscFunctionBegin; 6979566063dSJacob Faibussowitsch PetscCall(PetscViewerCreate(comm, hdf5v)); 6989566063dSJacob Faibussowitsch PetscCall(PetscViewerSetType(*hdf5v, PETSCVIEWERHDF5)); 6999566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetMode(*hdf5v, type)); 7009566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetName(*hdf5v, name)); 7019566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions(*hdf5v)); 7023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 7035c6c1daeSBarry Smith } 7045c6c1daeSBarry Smith 7055c6c1daeSBarry Smith /*@C 7065c6c1daeSBarry Smith PetscViewerHDF5GetFileId - Retrieve the file id, this file ID then can be used in direct HDF5 calls 7075c6c1daeSBarry Smith 70820f4b53cSBarry Smith Not Collective 7095c6c1daeSBarry Smith 7105c6c1daeSBarry Smith Input Parameter: 711811af0c4SBarry Smith . viewer - the `PetscViewer` 7125c6c1daeSBarry Smith 7135c6c1daeSBarry Smith Output Parameter: 7145c6c1daeSBarry Smith . file_id - The file id 7155c6c1daeSBarry Smith 7165c6c1daeSBarry Smith Level: intermediate 7175c6c1daeSBarry Smith 718d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()` 7195c6c1daeSBarry Smith @*/ 720d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetFileId(PetscViewer viewer, hid_t *file_id) 721d71ae5a4SJacob Faibussowitsch { 7225c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 7235c6c1daeSBarry Smith 7245c6c1daeSBarry Smith PetscFunctionBegin; 7255c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 7265c6c1daeSBarry Smith if (file_id) *file_id = hdf5->file_id; 7273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 7285c6c1daeSBarry Smith } 7295c6c1daeSBarry Smith 7305c6c1daeSBarry Smith /*@C 7315c6c1daeSBarry Smith PetscViewerHDF5PushGroup - Set the current HDF5 group for output 7325c6c1daeSBarry Smith 73320f4b53cSBarry Smith Not Collective 7345c6c1daeSBarry Smith 7355c6c1daeSBarry Smith Input Parameters: 736811af0c4SBarry Smith + viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 7375c6c1daeSBarry Smith - name - The group name 7385c6c1daeSBarry Smith 7395c6c1daeSBarry Smith Level: intermediate 7405c6c1daeSBarry Smith 7412e361344SVaclav Hapla Notes: 7422e361344SVaclav Hapla This is designed to mnemonically resemble the Unix cd command. 743811af0c4SBarry Smith .vb 744811af0c4SBarry Smith If name begins with '/', it is interpreted as an absolute path fully replacing current group, otherwise it is taken as relative to the current group. 7453f423023SBarry Smith `NULL`, empty string, or any sequence of all slashes (e.g. "///") is interpreted as the root group "/". 746811af0c4SBarry Smith "." means the current group is pushed again. 747811af0c4SBarry Smith .ve 7482e361344SVaclav Hapla 7492e361344SVaclav Hapla Example: 7502e361344SVaclav Hapla Suppose the current group is "/a". 7513f423023SBarry Smith .vb 7523f423023SBarry Smith If name is `NULL`, empty string, or a sequence of all slashes (e.g. "///"), then the new group will be "/". 7533f423023SBarry Smith If name is ".", then the new group will be "/a". 7543f423023SBarry Smith If name is "b", then the new group will be "/a/b". 7553f423023SBarry Smith If name is "/b", then the new group will be "/b". 7563f423023SBarry Smith .ve 7572e361344SVaclav Hapla 758aec76313SJacob Faibussowitsch Developer Notes: 7593f423023SBarry Smith The root group "/" is internally stored as `NULL`. 760820fc2d1SVaclav Hapla 761d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()`, `PetscViewerHDF5OpenGroup()`, `PetscViewerHDF5WriteGroup()` 7625c6c1daeSBarry Smith @*/ 763d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5PushGroup(PetscViewer viewer, const char name[]) 764d71ae5a4SJacob Faibussowitsch { 7655c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 7667d964842SVaclav Hapla PetscViewerHDF5GroupList *groupNode; 7672e361344SVaclav Hapla size_t i, len; 7682e361344SVaclav Hapla char buf[PETSC_MAX_PATH_LEN]; 7692e361344SVaclav Hapla const char *gname; 7705c6c1daeSBarry Smith 7715c6c1daeSBarry Smith PetscFunctionBegin; 7725c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 7734f572ea9SToby Isaac if (name) PetscAssertPointer(name, 2); 7749566063dSJacob Faibussowitsch PetscCall(PetscStrlen(name, &len)); 7752e361344SVaclav Hapla gname = NULL; 7762e361344SVaclav Hapla if (len) { 7772e361344SVaclav Hapla if (len == 1 && name[0] == '.') { 7782e361344SVaclav Hapla /* use current name */ 7792e361344SVaclav Hapla gname = (hdf5->groups && hdf5->groups->name) ? hdf5->groups->name : NULL; 7802e361344SVaclav Hapla } else if (name[0] == '/') { 7812e361344SVaclav Hapla /* absolute */ 7822e361344SVaclav Hapla for (i = 1; i < len; i++) { 7832e361344SVaclav Hapla if (name[i] != '/') { 7842e361344SVaclav Hapla gname = name; 7852e361344SVaclav Hapla break; 7862e361344SVaclav Hapla } 7872e361344SVaclav Hapla } 7882e361344SVaclav Hapla } else { 7892e361344SVaclav Hapla /* relative */ 7902e361344SVaclav Hapla const char *parent = (hdf5->groups && hdf5->groups->name) ? hdf5->groups->name : ""; 7919566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(buf, sizeof(buf), "%s/%s", parent, name)); 7922e361344SVaclav Hapla gname = buf; 7932e361344SVaclav Hapla } 7942e361344SVaclav Hapla } 7959566063dSJacob Faibussowitsch PetscCall(PetscNew(&groupNode)); 7969566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(gname, (char **)&groupNode->name)); 7975c6c1daeSBarry Smith groupNode->next = hdf5->groups; 7985c6c1daeSBarry Smith hdf5->groups = groupNode; 7993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 8005c6c1daeSBarry Smith } 8015c6c1daeSBarry Smith 8023ef9c667SSatish Balay /*@ 8035c6c1daeSBarry Smith PetscViewerHDF5PopGroup - Return the current HDF5 group for output to the previous value 8045c6c1daeSBarry Smith 80520f4b53cSBarry Smith Not Collective 8065c6c1daeSBarry Smith 8075c6c1daeSBarry Smith Input Parameter: 808811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 8095c6c1daeSBarry Smith 8105c6c1daeSBarry Smith Level: intermediate 8115c6c1daeSBarry Smith 812d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5GetGroup()`, `PetscViewerHDF5OpenGroup()`, `PetscViewerHDF5WriteGroup()` 8135c6c1daeSBarry Smith @*/ 814d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5PopGroup(PetscViewer viewer) 815d71ae5a4SJacob Faibussowitsch { 8165c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 8177d964842SVaclav Hapla PetscViewerHDF5GroupList *groupNode; 8185c6c1daeSBarry Smith 8195c6c1daeSBarry Smith PetscFunctionBegin; 8205c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 8215f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->groups, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "HDF5 group stack is empty, cannot pop"); 8225c6c1daeSBarry Smith groupNode = hdf5->groups; 8235c6c1daeSBarry Smith hdf5->groups = hdf5->groups->next; 8249566063dSJacob Faibussowitsch PetscCall(PetscFree(groupNode->name)); 8259566063dSJacob Faibussowitsch PetscCall(PetscFree(groupNode)); 8263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 8275c6c1daeSBarry Smith } 8285c6c1daeSBarry Smith 82934e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5GetGroup_Internal(PetscViewer viewer, const char *name[]) 830d71ae5a4SJacob Faibussowitsch { 8315c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 8325c6c1daeSBarry Smith 8335c6c1daeSBarry Smith PetscFunctionBegin; 8345c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 8354f572ea9SToby Isaac PetscAssertPointer(name, 2); 836a297a907SKarl Rupp if (hdf5->groups) *name = hdf5->groups->name; 8370298fd71SBarry Smith else *name = NULL; 8383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 8395c6c1daeSBarry Smith } 8405c6c1daeSBarry Smith 8413014b61aSVaclav Hapla /*@C 842811af0c4SBarry Smith PetscViewerHDF5OpenGroup - Open the HDF5 group with the name (full path) returned by `PetscViewerHDF5GetGroup()`, 843874270d9SVaclav Hapla and return this group's ID and file ID. 844811af0c4SBarry Smith If `PetscViewerHDF5GetGroup()` yields NULL, then group ID is file ID. 845874270d9SVaclav Hapla 84620f4b53cSBarry Smith Not Collective 847874270d9SVaclav Hapla 8483014b61aSVaclav Hapla Input Parameters: 8493014b61aSVaclav Hapla + viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 8503014b61aSVaclav Hapla - path - (Optional) The path relative to the pushed group 851874270d9SVaclav Hapla 852d8d19677SJose E. Roman Output Parameters: 853874270d9SVaclav Hapla + fileId - The HDF5 file ID 854874270d9SVaclav Hapla - groupId - The HDF5 group ID 855874270d9SVaclav Hapla 8563f423023SBarry Smith Level: intermediate 8573f423023SBarry Smith 858811af0c4SBarry Smith Note: 8593014b61aSVaclav Hapla If path starts with '/', it is taken as an absolute path overriding currently pushed group, else path is relative to the current pushed group. 8603f423023SBarry Smith `NULL` or empty path means the current pushed group. 8613014b61aSVaclav Hapla 862e74616beSVaclav Hapla If the viewer is writable, the group is created if it doesn't exist yet. 863e74616beSVaclav Hapla 864d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()`, `PetscViewerHDF5WriteGroup()` 865874270d9SVaclav Hapla @*/ 866d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5OpenGroup(PetscViewer viewer, const char path[], hid_t *fileId, hid_t *groupId) 867d71ae5a4SJacob Faibussowitsch { 86890e03537SVaclav Hapla hid_t file_id; 86990e03537SVaclav Hapla H5O_type_t type; 8703014b61aSVaclav Hapla const char *fileName = NULL; 871ce78bad3SBarry Smith const char *groupName = NULL; 87276d59af2SVaclav Hapla PetscBool writable, has; 87354dbf706SVaclav Hapla 87454dbf706SVaclav Hapla PetscFunctionBegin; 8759566063dSJacob Faibussowitsch PetscCall(PetscViewerWritable(viewer, &writable)); 8769566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetFileId(viewer, &file_id)); 8779566063dSJacob Faibussowitsch PetscCall(PetscViewerFileGetName(viewer, &fileName)); 8783014b61aSVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, path, &groupName)); 8799566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, groupName, writable, &has, &type)); 88076d59af2SVaclav Hapla if (!has) { 8815f80ce2aSJacob Faibussowitsch PetscCheck(writable, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Group %s does not exist and file %s is not open for writing", groupName, fileName); 882f7d195e4SLawrence Mitchell SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_LIB, "HDF5 failed to create group %s although file %s is open for writing", groupName, fileName); 88376d59af2SVaclav Hapla } 8845f80ce2aSJacob Faibussowitsch PetscCheck(type == H5O_TYPE_GROUP, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Path %s in file %s resolves to something which is not a group", groupName, fileName); 8853014b61aSVaclav Hapla PetscCallHDF5Return(*groupId, H5Gopen2, (file_id, groupName, H5P_DEFAULT)); 8863014b61aSVaclav Hapla PetscCall(PetscFree(groupName)); 88754dbf706SVaclav Hapla *fileId = file_id; 8883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 88954dbf706SVaclav Hapla } 89054dbf706SVaclav Hapla 89163cb69f5SVaclav Hapla /*@C 89263cb69f5SVaclav Hapla PetscViewerHDF5WriteGroup - Ensure the HDF5 group exists in the HDF5 file 89363cb69f5SVaclav Hapla 89420f4b53cSBarry Smith Not Collective 89563cb69f5SVaclav Hapla 89663cb69f5SVaclav Hapla Input Parameters: 89763cb69f5SVaclav Hapla + viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 89863cb69f5SVaclav Hapla - path - (Optional) The path relative to the pushed group 89963cb69f5SVaclav Hapla 9003f423023SBarry Smith Level: intermediate 9013f423023SBarry Smith 90263cb69f5SVaclav Hapla Note: 90363cb69f5SVaclav Hapla If path starts with '/', it is taken as an absolute path overriding currently pushed group, else path is relative to the current pushed group. 9043f423023SBarry Smith `NULL` or empty path means the current pushed group. 90563cb69f5SVaclav Hapla 90663cb69f5SVaclav Hapla This will fail if the viewer is not writable. 90763cb69f5SVaclav Hapla 908d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()`, `PetscViewerHDF5OpenGroup()` 90963cb69f5SVaclav Hapla @*/ 910d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5WriteGroup(PetscViewer viewer, const char path[]) 911d71ae5a4SJacob Faibussowitsch { 91263cb69f5SVaclav Hapla hid_t fileId, groupId; 91363cb69f5SVaclav Hapla 91463cb69f5SVaclav Hapla PetscFunctionBegin; 91563cb69f5SVaclav Hapla PetscCall(PetscViewerHDF5OpenGroup(viewer, path, &fileId, &groupId)); // make sure group is actually created 91663cb69f5SVaclav Hapla PetscCallHDF5(H5Gclose, (groupId)); 9173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 91863cb69f5SVaclav Hapla } 91963cb69f5SVaclav Hapla 9203ef9c667SSatish Balay /*@ 921d7dd068bSVaclav Hapla PetscViewerHDF5PushTimestepping - Activate timestepping mode for subsequent HDF5 reading and writing. 9225c6c1daeSBarry Smith 92320f4b53cSBarry Smith Not Collective 9245c6c1daeSBarry Smith 9255c6c1daeSBarry Smith Input Parameter: 926811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 9275c6c1daeSBarry Smith 9285c6c1daeSBarry Smith Level: intermediate 9295c6c1daeSBarry Smith 930d7dd068bSVaclav Hapla Notes: 931811af0c4SBarry Smith On first `PetscViewerHDF5PushTimestepping()`, the initial time step is set to 0. 932811af0c4SBarry Smith Next timesteps can then be set using `PetscViewerHDF5IncrementTimestep()` or `PetscViewerHDF5SetTimestep()`. 933811af0c4SBarry Smith Current timestep value determines which timestep is read from or written to any dataset on the next HDF5 I/O operation [e.g. `VecView()`]. 934811af0c4SBarry Smith Use `PetscViewerHDF5PopTimestepping()` to deactivate timestepping mode; calling it by the end of the program is NOT mandatory. 935811af0c4SBarry Smith Current timestep is remembered between `PetscViewerHDF5PopTimestepping()` and the next `PetscViewerHDF5PushTimestepping()`. 936d7dd068bSVaclav Hapla 937d7dd068bSVaclav Hapla If a dataset was stored with timestepping, it can be loaded only in the timestepping mode again. 938d7dd068bSVaclav Hapla Loading a timestepped dataset with timestepping disabled, or vice-versa results in an error. 939d7dd068bSVaclav Hapla 940aec76313SJacob Faibussowitsch Developer Notes: 941d7dd068bSVaclav Hapla Timestepped HDF5 dataset has an extra dimension and attribute "timestepping" set to true. 942d7dd068bSVaclav Hapla 943d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PopTimestepping()`, `PetscViewerHDF5IsTimestepping()`, `PetscViewerHDF5SetTimestep()`, `PetscViewerHDF5IncrementTimestep()`, `PetscViewerHDF5GetTimestep()` 944d7dd068bSVaclav Hapla @*/ 945d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5PushTimestepping(PetscViewer viewer) 946d71ae5a4SJacob Faibussowitsch { 947d7dd068bSVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 948d7dd068bSVaclav Hapla 949d7dd068bSVaclav Hapla PetscFunctionBegin; 950d7dd068bSVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 9515f80ce2aSJacob Faibussowitsch PetscCheck(!hdf5->timestepping, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestepping is already pushed"); 952d7dd068bSVaclav Hapla hdf5->timestepping = PETSC_TRUE; 953d7dd068bSVaclav Hapla if (hdf5->timestep < 0) hdf5->timestep = 0; 9543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 955d7dd068bSVaclav Hapla } 956d7dd068bSVaclav Hapla 957d7dd068bSVaclav Hapla /*@ 958d7dd068bSVaclav Hapla PetscViewerHDF5PopTimestepping - Deactivate timestepping mode for subsequent HDF5 reading and writing. 959d7dd068bSVaclav Hapla 96020f4b53cSBarry Smith Not Collective 961d7dd068bSVaclav Hapla 962d7dd068bSVaclav Hapla Input Parameter: 963811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 964d7dd068bSVaclav Hapla 965d7dd068bSVaclav Hapla Level: intermediate 966d7dd068bSVaclav Hapla 967811af0c4SBarry Smith Note: 968811af0c4SBarry Smith See `PetscViewerHDF5PushTimestepping()` for details. 969d7dd068bSVaclav Hapla 970d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5IsTimestepping()`, `PetscViewerHDF5SetTimestep()`, `PetscViewerHDF5IncrementTimestep()`, `PetscViewerHDF5GetTimestep()` 971d7dd068bSVaclav Hapla @*/ 972d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5PopTimestepping(PetscViewer viewer) 973d71ae5a4SJacob Faibussowitsch { 974d7dd068bSVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 975d7dd068bSVaclav Hapla 976d7dd068bSVaclav Hapla PetscFunctionBegin; 977d7dd068bSVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 9785f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->timestepping, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestepping has not been pushed yet. Call PetscViewerHDF5PushTimestepping() first"); 979d7dd068bSVaclav Hapla hdf5->timestepping = PETSC_FALSE; 9803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 981d7dd068bSVaclav Hapla } 982d7dd068bSVaclav Hapla 983d7dd068bSVaclav Hapla /*@ 984d7dd068bSVaclav Hapla PetscViewerHDF5IsTimestepping - Ask the viewer whether it is in timestepping mode currently. 985d7dd068bSVaclav Hapla 98620f4b53cSBarry Smith Not Collective 987d7dd068bSVaclav Hapla 988d7dd068bSVaclav Hapla Input Parameter: 989811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 990d7dd068bSVaclav Hapla 991d7dd068bSVaclav Hapla Output Parameter: 992d7dd068bSVaclav Hapla . flg - is timestepping active? 993d7dd068bSVaclav Hapla 994d7dd068bSVaclav Hapla Level: intermediate 995d7dd068bSVaclav Hapla 996811af0c4SBarry Smith Note: 997811af0c4SBarry Smith See `PetscViewerHDF5PushTimestepping()` for details. 998d7dd068bSVaclav Hapla 999d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5PopTimestepping()`, `PetscViewerHDF5SetTimestep()`, `PetscViewerHDF5IncrementTimestep()`, `PetscViewerHDF5GetTimestep()` 1000d7dd068bSVaclav Hapla @*/ 1001d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5IsTimestepping(PetscViewer viewer, PetscBool *flg) 1002d71ae5a4SJacob Faibussowitsch { 1003d7dd068bSVaclav Hapla PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 1004d7dd068bSVaclav Hapla 1005d7dd068bSVaclav Hapla PetscFunctionBegin; 1006d7dd068bSVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1007d7dd068bSVaclav Hapla *flg = hdf5->timestepping; 10083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1009d7dd068bSVaclav Hapla } 1010d7dd068bSVaclav Hapla 1011d7dd068bSVaclav Hapla /*@ 1012d7dd068bSVaclav Hapla PetscViewerHDF5IncrementTimestep - Increments current timestep for the HDF5 output. Fields are stacked in time. 1013d7dd068bSVaclav Hapla 101420f4b53cSBarry Smith Not Collective 1015d7dd068bSVaclav Hapla 1016d7dd068bSVaclav Hapla Input Parameter: 1017811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 1018d7dd068bSVaclav Hapla 1019d7dd068bSVaclav Hapla Level: intermediate 1020d7dd068bSVaclav Hapla 1021811af0c4SBarry Smith Note: 1022811af0c4SBarry Smith This can be called only if the viewer is in timestepping mode. See `PetscViewerHDF5PushTimestepping()` for details. 1023d7dd068bSVaclav Hapla 1024d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5SetTimestep()`, `PetscViewerHDF5GetTimestep()` 10255c6c1daeSBarry Smith @*/ 1026d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5IncrementTimestep(PetscViewer viewer) 1027d71ae5a4SJacob Faibussowitsch { 10285c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 10295c6c1daeSBarry Smith 10305c6c1daeSBarry Smith PetscFunctionBegin; 10315c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 10325f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->timestepping, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestepping has not been pushed yet. Call PetscViewerHDF5PushTimestepping() first"); 10335c6c1daeSBarry Smith ++hdf5->timestep; 10343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 10355c6c1daeSBarry Smith } 10365c6c1daeSBarry Smith 10373ef9c667SSatish Balay /*@ 1038d7dd068bSVaclav Hapla PetscViewerHDF5SetTimestep - Set the current timestep for the HDF5 output. Fields are stacked in time. 10395c6c1daeSBarry Smith 104020f4b53cSBarry Smith Logically Collective 10415c6c1daeSBarry Smith 10425c6c1daeSBarry Smith Input Parameters: 1043811af0c4SBarry Smith + viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 1044d7dd068bSVaclav Hapla - timestep - The timestep 10455c6c1daeSBarry Smith 10465c6c1daeSBarry Smith Level: intermediate 10475c6c1daeSBarry Smith 1048811af0c4SBarry Smith Note: 1049811af0c4SBarry Smith This can be called only if the viewer is in timestepping mode. See `PetscViewerHDF5PushTimestepping()` for details. 1050d7dd068bSVaclav Hapla 1051d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5IncrementTimestep()`, `PetscViewerHDF5GetTimestep()` 10525c6c1daeSBarry Smith @*/ 1053d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5SetTimestep(PetscViewer viewer, PetscInt timestep) 1054d71ae5a4SJacob Faibussowitsch { 10555c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 10565c6c1daeSBarry Smith 10575c6c1daeSBarry Smith PetscFunctionBegin; 10585c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1059d7dd068bSVaclav Hapla PetscValidLogicalCollectiveInt(viewer, timestep, 2); 10605f80ce2aSJacob Faibussowitsch PetscCheck(timestep >= 0, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestep %" PetscInt_FMT " is negative", timestep); 10615f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->timestepping, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestepping has not been pushed yet. Call PetscViewerHDF5PushTimestepping() first"); 10625c6c1daeSBarry Smith hdf5->timestep = timestep; 10633ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 10645c6c1daeSBarry Smith } 10655c6c1daeSBarry Smith 10663ef9c667SSatish Balay /*@ 10675c6c1daeSBarry Smith PetscViewerHDF5GetTimestep - Get the current timestep for the HDF5 output. Fields are stacked in time. 10685c6c1daeSBarry Smith 106920f4b53cSBarry Smith Not Collective 10705c6c1daeSBarry Smith 10715c6c1daeSBarry Smith Input Parameter: 1072811af0c4SBarry Smith . viewer - the `PetscViewer` of type `PETSCVIEWERHDF5` 10735c6c1daeSBarry Smith 10745c6c1daeSBarry Smith Output Parameter: 1075d7dd068bSVaclav Hapla . timestep - The timestep 10765c6c1daeSBarry Smith 10775c6c1daeSBarry Smith Level: intermediate 10785c6c1daeSBarry Smith 1079811af0c4SBarry Smith Note: 1080811af0c4SBarry Smith This can be called only if the viewer is in the timestepping mode. See `PetscViewerHDF5PushTimestepping()` for details. 1081d7dd068bSVaclav Hapla 1082d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5PushTimestepping()`, `PetscViewerHDF5IncrementTimestep()`, `PetscViewerHDF5SetTimestep()` 10835c6c1daeSBarry Smith @*/ 1084d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5GetTimestep(PetscViewer viewer, PetscInt *timestep) 1085d71ae5a4SJacob Faibussowitsch { 10865c6c1daeSBarry Smith PetscViewer_HDF5 *hdf5 = (PetscViewer_HDF5 *)viewer->data; 10875c6c1daeSBarry Smith 10885c6c1daeSBarry Smith PetscFunctionBegin; 10895c6c1daeSBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 10904f572ea9SToby Isaac PetscAssertPointer(timestep, 2); 10915f80ce2aSJacob Faibussowitsch PetscCheck(hdf5->timestepping, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Timestepping has not been pushed yet. Call PetscViewerHDF5PushTimestepping() first"); 10925c6c1daeSBarry Smith *timestep = hdf5->timestep; 10933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 10945c6c1daeSBarry Smith } 10955c6c1daeSBarry Smith 109636bce6e8SMatthew G. Knepley /*@C 109736bce6e8SMatthew G. Knepley PetscDataTypeToHDF5DataType - Converts the PETSc name of a datatype to its HDF5 name. 109836bce6e8SMatthew G. Knepley 109920f4b53cSBarry Smith Not Collective 110036bce6e8SMatthew G. Knepley 110136bce6e8SMatthew G. Knepley Input Parameter: 1102811af0c4SBarry Smith . ptype - the PETSc datatype name (for example `PETSC_DOUBLE`) 110336bce6e8SMatthew G. Knepley 110436bce6e8SMatthew G. Knepley Output Parameter: 1105aec76313SJacob Faibussowitsch . htype - the HDF5 datatype 110636bce6e8SMatthew G. Knepley 110736bce6e8SMatthew G. Knepley Level: advanced 110836bce6e8SMatthew G. Knepley 1109d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscDataType`, `PetscHDF5DataTypeToPetscDataType()` 111036bce6e8SMatthew G. Knepley @*/ 1111d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDataTypeToHDF5DataType(PetscDataType ptype, hid_t *htype) 1112d71ae5a4SJacob Faibussowitsch { 111336bce6e8SMatthew G. Knepley PetscFunctionBegin; 111436bce6e8SMatthew G. Knepley if (ptype == PETSC_INT) 111536bce6e8SMatthew G. Knepley #if defined(PETSC_USE_64BIT_INDICES) 111636bce6e8SMatthew G. Knepley *htype = H5T_NATIVE_LLONG; 111736bce6e8SMatthew G. Knepley #else 111836bce6e8SMatthew G. Knepley *htype = H5T_NATIVE_INT; 111936bce6e8SMatthew G. Knepley #endif 112036bce6e8SMatthew G. Knepley else if (ptype == PETSC_DOUBLE) *htype = H5T_NATIVE_DOUBLE; 112136bce6e8SMatthew G. Knepley else if (ptype == PETSC_LONG) *htype = H5T_NATIVE_LONG; 112236bce6e8SMatthew G. Knepley else if (ptype == PETSC_SHORT) *htype = H5T_NATIVE_SHORT; 1123c9450970SBarry Smith else if (ptype == PETSC_ENUM) *htype = H5T_NATIVE_INT; 1124de3ba262SVaclav Hapla else if (ptype == PETSC_BOOL) *htype = H5T_NATIVE_INT; 112536bce6e8SMatthew G. Knepley else if (ptype == PETSC_FLOAT) *htype = H5T_NATIVE_FLOAT; 112636bce6e8SMatthew G. Knepley else if (ptype == PETSC_CHAR) *htype = H5T_NATIVE_CHAR; 112736bce6e8SMatthew G. Knepley else if (ptype == PETSC_BIT_LOGICAL) *htype = H5T_NATIVE_UCHAR; 11287e97c476SMatthew G. Knepley else if (ptype == PETSC_STRING) *htype = H5Tcopy(H5T_C_S1); 112936bce6e8SMatthew G. Knepley else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported PETSc datatype"); 11303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 113136bce6e8SMatthew G. Knepley } 113236bce6e8SMatthew G. Knepley 113336bce6e8SMatthew G. Knepley /*@C 113436bce6e8SMatthew G. Knepley PetscHDF5DataTypeToPetscDataType - Finds the PETSc name of a datatype from its HDF5 name 113536bce6e8SMatthew G. Knepley 113620f4b53cSBarry Smith Not Collective 113736bce6e8SMatthew G. Knepley 113836bce6e8SMatthew G. Knepley Input Parameter: 1139811af0c4SBarry Smith . htype - the HDF5 datatype (for example `H5T_NATIVE_DOUBLE`, ...) 114036bce6e8SMatthew G. Knepley 114136bce6e8SMatthew G. Knepley Output Parameter: 1142811af0c4SBarry Smith . ptype - the PETSc datatype name (for example `PETSC_DOUBLE`) 114336bce6e8SMatthew G. Knepley 114436bce6e8SMatthew G. Knepley Level: advanced 114536bce6e8SMatthew G. Knepley 114642747ad1SJacob Faibussowitsch .seealso: [](sec_viewers), `PetscDataType` 114736bce6e8SMatthew G. Knepley @*/ 1148d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscHDF5DataTypeToPetscDataType(hid_t htype, PetscDataType *ptype) 1149d71ae5a4SJacob Faibussowitsch { 115036bce6e8SMatthew G. Knepley PetscFunctionBegin; 115136bce6e8SMatthew G. Knepley #if defined(PETSC_USE_64BIT_INDICES) 115236bce6e8SMatthew G. Knepley if (htype == H5T_NATIVE_INT) *ptype = PETSC_LONG; 115336bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_LLONG) *ptype = PETSC_INT; 115436bce6e8SMatthew G. Knepley #else 115536bce6e8SMatthew G. Knepley if (htype == H5T_NATIVE_INT) *ptype = PETSC_INT; 115636bce6e8SMatthew G. Knepley #endif 115736bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_DOUBLE) *ptype = PETSC_DOUBLE; 115836bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_LONG) *ptype = PETSC_LONG; 115936bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_SHORT) *ptype = PETSC_SHORT; 116036bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_FLOAT) *ptype = PETSC_FLOAT; 116136bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_CHAR) *ptype = PETSC_CHAR; 116236bce6e8SMatthew G. Knepley else if (htype == H5T_NATIVE_UCHAR) *ptype = PETSC_CHAR; 11637e97c476SMatthew G. Knepley else if (htype == H5T_C_S1) *ptype = PETSC_STRING; 116436bce6e8SMatthew G. Knepley else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported HDF5 datatype"); 11653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 116636bce6e8SMatthew G. Knepley } 116736bce6e8SMatthew G. Knepley 1168df863907SAlex Fikl /*@C 1169b8ee85a1SVaclav Hapla PetscViewerHDF5WriteAttribute - Write an attribute 117036bce6e8SMatthew G. Knepley 1171343a20b2SVaclav Hapla Collective 1172343a20b2SVaclav Hapla 117336bce6e8SMatthew G. Knepley Input Parameters: 1174811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 11754302210dSVaclav Hapla . parent - The parent dataset/group name 117636bce6e8SMatthew G. Knepley . name - The attribute name 117736bce6e8SMatthew G. Knepley . datatype - The attribute type 117836bce6e8SMatthew G. Knepley - value - The attribute value 117936bce6e8SMatthew G. Knepley 118036bce6e8SMatthew G. Knepley Level: advanced 118136bce6e8SMatthew G. Knepley 1182811af0c4SBarry Smith Note: 1183343a20b2SVaclav Hapla If parent starts with '/', it is taken as an absolute path overriding currently pushed group, else parent is relative to the current pushed group. NULL means the current pushed group. 11844302210dSVaclav Hapla 1185d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5WriteObjectAttribute()`, `PetscViewerHDF5ReadAttribute()`, `PetscViewerHDF5HasAttribute()`, 1186811af0c4SBarry Smith `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 118736bce6e8SMatthew G. Knepley @*/ 1188d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5WriteAttribute(PetscViewer viewer, const char parent[], const char name[], PetscDataType datatype, const void *value) 1189d71ae5a4SJacob Faibussowitsch { 1190ce78bad3SBarry Smith const char *parentAbsPath; 119160568592SMatthew G. Knepley hid_t h5, dataspace, obj, attribute, dtype; 1192080f144cSVaclav Hapla PetscBool has; 119336bce6e8SMatthew G. Knepley 119436bce6e8SMatthew G. Knepley PetscFunctionBegin; 11955cdeb108SMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 11964f572ea9SToby Isaac if (parent) PetscAssertPointer(parent, 2); 11974f572ea9SToby Isaac PetscAssertPointer(name, 3); 11984302210dSVaclav Hapla PetscValidLogicalCollectiveEnum(viewer, datatype, 4); 11994f572ea9SToby Isaac PetscAssertPointer(value, 5); 120077717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, parent, &parentAbsPath)); 12019566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, parentAbsPath, PETSC_TRUE, NULL, NULL)); 12029566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5HasAttribute_Internal(viewer, parentAbsPath, name, &has)); 12039566063dSJacob Faibussowitsch PetscCall(PetscDataTypeToHDF5DataType(datatype, &dtype)); 12047e97c476SMatthew G. Knepley if (datatype == PETSC_STRING) { 12057e97c476SMatthew G. Knepley size_t len; 12069566063dSJacob Faibussowitsch PetscCall(PetscStrlen((const char *)value, &len)); 1207792fecdfSBarry Smith PetscCallHDF5(H5Tset_size, (dtype, len + 1)); 12087e97c476SMatthew G. Knepley } 12099566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetFileId(viewer, &h5)); 1210792fecdfSBarry Smith PetscCallHDF5Return(dataspace, H5Screate, (H5S_SCALAR)); 1211792fecdfSBarry Smith PetscCallHDF5Return(obj, H5Oopen, (h5, parentAbsPath, H5P_DEFAULT)); 1212080f144cSVaclav Hapla if (has) { 1213792fecdfSBarry Smith PetscCallHDF5Return(attribute, H5Aopen_name, (obj, name)); 1214080f144cSVaclav Hapla } else { 1215792fecdfSBarry Smith PetscCallHDF5Return(attribute, H5Acreate2, (obj, name, dtype, dataspace, H5P_DEFAULT, H5P_DEFAULT)); 1216080f144cSVaclav Hapla } 1217792fecdfSBarry Smith PetscCallHDF5(H5Awrite, (attribute, dtype, value)); 1218792fecdfSBarry Smith if (datatype == PETSC_STRING) PetscCallHDF5(H5Tclose, (dtype)); 1219792fecdfSBarry Smith PetscCallHDF5(H5Aclose, (attribute)); 1220792fecdfSBarry Smith PetscCallHDF5(H5Oclose, (obj)); 1221792fecdfSBarry Smith PetscCallHDF5(H5Sclose, (dataspace)); 12229566063dSJacob Faibussowitsch PetscCall(PetscFree(parentAbsPath)); 12233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 122436bce6e8SMatthew G. Knepley } 122536bce6e8SMatthew G. Knepley 1226df863907SAlex Fikl /*@C 1227811af0c4SBarry Smith PetscViewerHDF5WriteObjectAttribute - Write an attribute to the dataset matching the given `PetscObject` by name 1228577e0e04SVaclav Hapla 1229343a20b2SVaclav Hapla Collective 1230343a20b2SVaclav Hapla 1231577e0e04SVaclav Hapla Input Parameters: 1232811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 1233577e0e04SVaclav Hapla . obj - The object whose name is used to lookup the parent dataset, relative to the current group. 1234577e0e04SVaclav Hapla . name - The attribute name 1235577e0e04SVaclav Hapla . datatype - The attribute type 1236577e0e04SVaclav Hapla - value - The attribute value 1237577e0e04SVaclav Hapla 12383f423023SBarry Smith Level: advanced 12393f423023SBarry Smith 1240811af0c4SBarry Smith Note: 1241343a20b2SVaclav Hapla This fails if the path current_group/object_name doesn't resolve to a dataset (the path doesn't exist or is not a dataset). 1242811af0c4SBarry Smith You might want to check first if it does using `PetscViewerHDF5HasObject()`. 1243577e0e04SVaclav Hapla 1244d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5WriteAttribute()`, `PetscViewerHDF5ReadObjectAttribute()`, `PetscViewerHDF5HasObjectAttribute()`, 1245811af0c4SBarry Smith `PetscViewerHDF5HasObject()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1246577e0e04SVaclav Hapla @*/ 1247d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5WriteObjectAttribute(PetscViewer viewer, PetscObject obj, const char name[], PetscDataType datatype, const void *value) 1248d71ae5a4SJacob Faibussowitsch { 1249577e0e04SVaclav Hapla PetscFunctionBegin; 1250577e0e04SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1251577e0e04SVaclav Hapla PetscValidHeader(obj, 2); 12524f572ea9SToby Isaac PetscAssertPointer(name, 3); 12534f572ea9SToby Isaac PetscAssertPointer(value, 5); 12549566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5CheckNamedObject_Internal(viewer, obj)); 12559566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5WriteAttribute(viewer, obj->name, name, datatype, value)); 12563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1257577e0e04SVaclav Hapla } 1258577e0e04SVaclav Hapla 1259577e0e04SVaclav Hapla /*@C 1260b8ee85a1SVaclav Hapla PetscViewerHDF5ReadAttribute - Read an attribute 1261ad0c61feSMatthew G. Knepley 1262343a20b2SVaclav Hapla Collective 1263343a20b2SVaclav Hapla 1264ad0c61feSMatthew G. Knepley Input Parameters: 1265811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 12664302210dSVaclav Hapla . parent - The parent dataset/group name 1267ad0c61feSMatthew G. Knepley . name - The attribute name 1268a2d6be1bSVaclav Hapla . datatype - The attribute type 1269a2d6be1bSVaclav Hapla - defaultValue - The pointer to the default value 1270ad0c61feSMatthew G. Knepley 1271ad0c61feSMatthew G. Knepley Output Parameter: 1272a2d6be1bSVaclav Hapla . value - The pointer to the read HDF5 attribute value 1273ad0c61feSMatthew G. Knepley 12743f423023SBarry Smith Level: advanced 12753f423023SBarry Smith 1276a2d6be1bSVaclav Hapla Notes: 12773f423023SBarry Smith If defaultValue is `NULL` and the attribute is not found, an error occurs. 12783f423023SBarry Smith 12793f423023SBarry Smith If defaultValue is not `NULL` and the attribute is not found, `defaultValue` is copied to value. 12803f423023SBarry Smith 12813f423023SBarry Smith The pointers `defaultValue` and value can be the same; for instance 12823f423023SBarry Smith .vb 12833f423023SBarry Smith flg = PETSC_FALSE; 12843f423023SBarry Smith PetscCall(`PetscViewerHDF5ReadAttribute`(viewer,name,"attr",PETSC_BOOL,&flg,&flg)); 12853f423023SBarry Smith .ve 1286a2d6be1bSVaclav Hapla is valid, but make sure the default value is initialized. 1287a2d6be1bSVaclav Hapla 1288811af0c4SBarry Smith If the datatype is `PETSC_STRING`, the output string is newly allocated so one must `PetscFree()` it when no longer needed. 1289708d4cb3SBarry Smith 12903f423023SBarry Smith If parent starts with '/', it is taken as an absolute path overriding currently pushed group, else parent is relative to the current pushed group. `NULL` means the current pushed group. 12914302210dSVaclav Hapla 1292d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5ReadObjectAttribute()`, `PetscViewerHDF5WriteAttribute()`, `PetscViewerHDF5HasAttribute()`, `PetscViewerHDF5HasObject()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1293ad0c61feSMatthew G. Knepley @*/ 1294d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5ReadAttribute(PetscViewer viewer, const char parent[], const char name[], PetscDataType datatype, const void *defaultValue, void *value) 1295d71ae5a4SJacob Faibussowitsch { 1296ce78bad3SBarry Smith const char *parentAbsPath; 1297a2d6be1bSVaclav Hapla hid_t h5, obj, attribute, dtype; 1298969748fdSVaclav Hapla PetscBool has; 1299ad0c61feSMatthew G. Knepley 1300ad0c61feSMatthew G. Knepley PetscFunctionBegin; 13015cdeb108SMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 13024f572ea9SToby Isaac if (parent) PetscAssertPointer(parent, 2); 13034f572ea9SToby Isaac PetscAssertPointer(name, 3); 13044f572ea9SToby Isaac if (defaultValue) PetscAssertPointer(defaultValue, 5); 13054f572ea9SToby Isaac PetscAssertPointer(value, 6); 13069566063dSJacob Faibussowitsch PetscCall(PetscDataTypeToHDF5DataType(datatype, &dtype)); 130777717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, parent, &parentAbsPath)); 13089566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, parentAbsPath, PETSC_FALSE, &has, NULL)); 13099566063dSJacob Faibussowitsch if (has) PetscCall(PetscViewerHDF5HasAttribute_Internal(viewer, parentAbsPath, name, &has)); 1310a2d6be1bSVaclav Hapla if (!has) { 1311a2d6be1bSVaclav Hapla if (defaultValue) { 1312a2d6be1bSVaclav Hapla if (defaultValue != value) { 1313a2d6be1bSVaclav Hapla if (datatype == PETSC_STRING) { 13149566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(*(char **)defaultValue, (char **)value)); 1315a2d6be1bSVaclav Hapla } else { 1316a2d6be1bSVaclav Hapla size_t len; 1317792fecdfSBarry Smith PetscCallHDF5ReturnNoCheck(len, H5Tget_size, (dtype)); 13189566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(value, defaultValue, len)); 1319a2d6be1bSVaclav Hapla } 1320a2d6be1bSVaclav Hapla } 13219566063dSJacob Faibussowitsch PetscCall(PetscFree(parentAbsPath)); 13223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 132398921bdaSJacob Faibussowitsch } else SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Attribute %s/%s does not exist and default value not provided", parentAbsPath, name); 1324a2d6be1bSVaclav Hapla } 13259566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetFileId(viewer, &h5)); 1326792fecdfSBarry Smith PetscCallHDF5Return(obj, H5Oopen, (h5, parentAbsPath, H5P_DEFAULT)); 1327792fecdfSBarry Smith PetscCallHDF5Return(attribute, H5Aopen_name, (obj, name)); 1328f0b58479SMatthew G. Knepley if (datatype == PETSC_STRING) { 1329f0b58479SMatthew G. Knepley size_t len; 1330a2d6be1bSVaclav Hapla hid_t atype; 1331792fecdfSBarry Smith PetscCallHDF5Return(atype, H5Aget_type, (attribute)); 1332792fecdfSBarry Smith PetscCallHDF5ReturnNoCheck(len, H5Tget_size, (atype)); 13339566063dSJacob Faibussowitsch PetscCall(PetscMalloc((len + 1) * sizeof(char), value)); 1334792fecdfSBarry Smith PetscCallHDF5(H5Tset_size, (dtype, len + 1)); 1335792fecdfSBarry Smith PetscCallHDF5(H5Aread, (attribute, dtype, *(char **)value)); 1336708d4cb3SBarry Smith } else { 1337792fecdfSBarry Smith PetscCallHDF5(H5Aread, (attribute, dtype, value)); 1338708d4cb3SBarry Smith } 1339792fecdfSBarry Smith PetscCallHDF5(H5Aclose, (attribute)); 1340e90c5075SMarek Pecha /* H5Oclose can be used to close groups, datasets, or committed datatypes */ 1341792fecdfSBarry Smith PetscCallHDF5(H5Oclose, (obj)); 13429566063dSJacob Faibussowitsch PetscCall(PetscFree(parentAbsPath)); 13433ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1344ad0c61feSMatthew G. Knepley } 1345ad0c61feSMatthew G. Knepley 1346577e0e04SVaclav Hapla /*@C 1347811af0c4SBarry Smith PetscViewerHDF5ReadObjectAttribute - Read an attribute from the dataset matching the given `PetscObject` by name 1348577e0e04SVaclav Hapla 1349343a20b2SVaclav Hapla Collective 1350343a20b2SVaclav Hapla 1351577e0e04SVaclav Hapla Input Parameters: 1352811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 1353577e0e04SVaclav Hapla . obj - The object whose name is used to lookup the parent dataset, relative to the current group. 1354577e0e04SVaclav Hapla . name - The attribute name 135510450e9eSJacob Faibussowitsch . datatype - The attribute type 135610450e9eSJacob Faibussowitsch - defaultValue - The default attribute value 1357577e0e04SVaclav Hapla 1358577e0e04SVaclav Hapla Output Parameter: 1359577e0e04SVaclav Hapla . value - The attribute value 1360577e0e04SVaclav Hapla 13613f423023SBarry Smith Level: advanced 13623f423023SBarry Smith 1363811af0c4SBarry Smith Note: 1364577e0e04SVaclav Hapla This fails if current_group/object_name doesn't resolve to a dataset (the path doesn't exist or is not a dataset). 1365811af0c4SBarry Smith You might want to check first if it does using `PetscViewerHDF5HasObject()`. 1366577e0e04SVaclav Hapla 1367d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5ReadAttribute()` `PetscViewerHDF5WriteObjectAttribute()`, `PetscViewerHDF5HasObjectAttribute()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1368577e0e04SVaclav Hapla @*/ 1369d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5ReadObjectAttribute(PetscViewer viewer, PetscObject obj, const char name[], PetscDataType datatype, void *defaultValue, void *value) 1370d71ae5a4SJacob Faibussowitsch { 1371577e0e04SVaclav Hapla PetscFunctionBegin; 1372577e0e04SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1373577e0e04SVaclav Hapla PetscValidHeader(obj, 2); 13744f572ea9SToby Isaac PetscAssertPointer(name, 3); 13754f572ea9SToby Isaac PetscAssertPointer(value, 6); 13769566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5CheckNamedObject_Internal(viewer, obj)); 13779566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5ReadAttribute(viewer, obj->name, name, datatype, defaultValue, value)); 13783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1379577e0e04SVaclav Hapla } 1380577e0e04SVaclav Hapla 138134e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5Traverse_Inner_Internal(hid_t h5, const char name[], PetscBool createGroup, PetscBool *exists_) 1382d71ae5a4SJacob Faibussowitsch { 1383a75c3fd4SVaclav Hapla htri_t exists; 1384a75c3fd4SVaclav Hapla hid_t group; 1385a75c3fd4SVaclav Hapla 1386a75c3fd4SVaclav Hapla PetscFunctionBegin; 1387792fecdfSBarry Smith PetscCallHDF5Return(exists, H5Lexists, (h5, name, H5P_DEFAULT)); 1388792fecdfSBarry Smith if (exists) PetscCallHDF5Return(exists, H5Oexists_by_name, (h5, name, H5P_DEFAULT)); 1389a75c3fd4SVaclav Hapla if (!exists && createGroup) { 1390792fecdfSBarry Smith PetscCallHDF5Return(group, H5Gcreate2, (h5, name, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT)); 1391792fecdfSBarry Smith PetscCallHDF5(H5Gclose, (group)); 1392a75c3fd4SVaclav Hapla exists = PETSC_TRUE; 1393a75c3fd4SVaclav Hapla } 1394a75c3fd4SVaclav Hapla *exists_ = (PetscBool)exists; 13953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1396a75c3fd4SVaclav Hapla } 1397a75c3fd4SVaclav Hapla 1398d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5Traverse_Internal(PetscViewer viewer, const char name[], PetscBool createGroup, PetscBool *has, H5O_type_t *otype) 1399d71ae5a4SJacob Faibussowitsch { 140090e03537SVaclav Hapla const char rootGroupName[] = "/"; 14015cdeb108SMatthew G. Knepley hid_t h5; 1402e5bf9ebcSVaclav Hapla PetscBool exists = PETSC_FALSE; 140315b861d2SVaclav Hapla PetscInt i; 140415b861d2SVaclav Hapla int n; 140585688ae2SVaclav Hapla char **hierarchy; 140685688ae2SVaclav Hapla char buf[PETSC_MAX_PATH_LEN] = ""; 14075cdeb108SMatthew G. Knepley 14085cdeb108SMatthew G. Knepley PetscFunctionBegin; 14095cdeb108SMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 14104f572ea9SToby Isaac if (name) PetscAssertPointer(name, 2); 141190e03537SVaclav Hapla else name = rootGroupName; 1412ccd66a83SVaclav Hapla if (has) { 14134f572ea9SToby Isaac PetscAssertPointer(has, 4); 14145cdeb108SMatthew G. Knepley *has = PETSC_FALSE; 1415ccd66a83SVaclav Hapla } 1416ccd66a83SVaclav Hapla if (otype) { 14174f572ea9SToby Isaac PetscAssertPointer(otype, 5); 141856cc0592SVaclav Hapla *otype = H5O_TYPE_UNKNOWN; 1419ccd66a83SVaclav Hapla } 14209566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetFileId(viewer, &h5)); 142185688ae2SVaclav Hapla 142285688ae2SVaclav Hapla /* 142385688ae2SVaclav Hapla Unfortunately, H5Oexists_by_name() fails if any object in hierarchy is missing. 142485688ae2SVaclav Hapla Hence, each of them needs to be tested separately: 142585688ae2SVaclav Hapla 1) whether it's a valid link 142685688ae2SVaclav Hapla 2) whether this link resolves to an object 142785688ae2SVaclav Hapla See H5Oexists_by_name() documentation. 142885688ae2SVaclav Hapla */ 14299566063dSJacob Faibussowitsch PetscCall(PetscStrToArray(name, '/', &n, &hierarchy)); 143085688ae2SVaclav Hapla if (!n) { 143185688ae2SVaclav Hapla /* Assume group "/" always exists in accordance with HDF5 >= 1.10.0. See H5Lexists() documentation. */ 1432ccd66a83SVaclav Hapla if (has) *has = PETSC_TRUE; 1433ccd66a83SVaclav Hapla if (otype) *otype = H5O_TYPE_GROUP; 14349566063dSJacob Faibussowitsch PetscCall(PetscStrToArrayDestroy(n, hierarchy)); 14353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 143685688ae2SVaclav Hapla } 143785688ae2SVaclav Hapla for (i = 0; i < n; i++) { 1438c6a7a370SJeremy L Thompson PetscCall(PetscStrlcat(buf, "/", sizeof(buf))); 1439c6a7a370SJeremy L Thompson PetscCall(PetscStrlcat(buf, hierarchy[i], sizeof(buf))); 14409566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Inner_Internal(h5, buf, createGroup, &exists)); 1441a75c3fd4SVaclav Hapla if (!exists) break; 144285688ae2SVaclav Hapla } 14439566063dSJacob Faibussowitsch PetscCall(PetscStrToArrayDestroy(n, hierarchy)); 144485688ae2SVaclav Hapla 144585688ae2SVaclav Hapla /* If the object exists, get its type */ 1446ccd66a83SVaclav Hapla if (exists && otype) { 14475cdeb108SMatthew G. Knepley H5O_info_t info; 14485cdeb108SMatthew G. Knepley 144976276748SVaclav Hapla /* We could use H5Iget_type() here but that would require opening the object. This way we only need its name. */ 1450792fecdfSBarry Smith PetscCallHDF5(H5Oget_info_by_name, (h5, name, &info, H5P_DEFAULT)); 145156cc0592SVaclav Hapla *otype = info.type; 14525cdeb108SMatthew G. Knepley } 1453ccd66a83SVaclav Hapla if (has) *has = exists; 14543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14555cdeb108SMatthew G. Knepley } 14565cdeb108SMatthew G. Knepley 14574302210dSVaclav Hapla /*@C 14580a9f38efSVaclav Hapla PetscViewerHDF5HasGroup - Check whether the current (pushed) group exists in the HDF5 file 14590a9f38efSVaclav Hapla 1460343a20b2SVaclav Hapla Collective 1461343a20b2SVaclav Hapla 14620a9f38efSVaclav Hapla Input Parameters: 1463811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 1464a0558868SVaclav Hapla - path - (Optional) The path relative to the pushed group 14650a9f38efSVaclav Hapla 14660a9f38efSVaclav Hapla Output Parameter: 14670a9f38efSVaclav Hapla . has - Flag for group existence 14680a9f38efSVaclav Hapla 14690a9f38efSVaclav Hapla Level: advanced 14700a9f38efSVaclav Hapla 14714302210dSVaclav Hapla Notes: 1472785c443eSVaclav Hapla If path starts with '/', it is taken as an absolute path overriding currently pushed group, else path is relative to the current pushed group. 14733f423023SBarry Smith `NULL` or empty path means the current pushed group. 14744302210dSVaclav Hapla 1475811af0c4SBarry Smith If path exists but is not a group, `PETSC_FALSE` is returned. 1476811af0c4SBarry Smith 1477d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5HasAttribute()`, `PetscViewerHDF5HasDataset()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()`, `PetscViewerHDF5OpenGroup()` 14780a9f38efSVaclav Hapla @*/ 1479d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5HasGroup(PetscViewer viewer, const char path[], PetscBool *has) 1480d71ae5a4SJacob Faibussowitsch { 14810a9f38efSVaclav Hapla H5O_type_t type; 1482ce78bad3SBarry Smith const char *abspath; 14830a9f38efSVaclav Hapla 14840a9f38efSVaclav Hapla PetscFunctionBegin; 14850a9f38efSVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 14864f572ea9SToby Isaac if (path) PetscAssertPointer(path, 2); 14874f572ea9SToby Isaac PetscAssertPointer(has, 3); 148877717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, path, &abspath)); 14899566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, abspath, PETSC_FALSE, NULL, &type)); 14904302210dSVaclav Hapla *has = (PetscBool)(type == H5O_TYPE_GROUP); 14919566063dSJacob Faibussowitsch PetscCall(PetscFree(abspath)); 14923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14930a9f38efSVaclav Hapla } 14940a9f38efSVaclav Hapla 149589e0ef10SVaclav Hapla /*@C 149689e0ef10SVaclav Hapla PetscViewerHDF5HasDataset - Check whether a given dataset exists in the HDF5 file 149789e0ef10SVaclav Hapla 1498343a20b2SVaclav Hapla Collective 1499343a20b2SVaclav Hapla 150089e0ef10SVaclav Hapla Input Parameters: 1501811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 150289e0ef10SVaclav Hapla - path - The dataset path 150389e0ef10SVaclav Hapla 150489e0ef10SVaclav Hapla Output Parameter: 150589e0ef10SVaclav Hapla . has - Flag whether dataset exists 150689e0ef10SVaclav Hapla 150789e0ef10SVaclav Hapla Level: advanced 150889e0ef10SVaclav Hapla 150989e0ef10SVaclav Hapla Notes: 1510343a20b2SVaclav Hapla If path starts with '/', it is taken as an absolute path overriding currently pushed group, else path is relative to the current pushed group. 151189e0ef10SVaclav Hapla 15123f423023SBarry Smith If `path` is `NULL` or empty, has is set to `PETSC_FALSE`. 1513811af0c4SBarry Smith 15143f423023SBarry Smith If `path` exists but is not a dataset, has is set to `PETSC_FALSE` as well. 1515811af0c4SBarry Smith 1516d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5HasObject()`, `PetscViewerHDF5HasAttribute()`, `PetscViewerHDF5HasGroup()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 151789e0ef10SVaclav Hapla @*/ 1518d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5HasDataset(PetscViewer viewer, const char path[], PetscBool *has) 1519d71ae5a4SJacob Faibussowitsch { 152089e0ef10SVaclav Hapla H5O_type_t type; 1521ce78bad3SBarry Smith const char *abspath; 152289e0ef10SVaclav Hapla 152389e0ef10SVaclav Hapla PetscFunctionBegin; 152489e0ef10SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 15254f572ea9SToby Isaac if (path) PetscAssertPointer(path, 2); 15264f572ea9SToby Isaac PetscAssertPointer(has, 3); 152777717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, path, &abspath)); 15289566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, abspath, PETSC_FALSE, NULL, &type)); 152989e0ef10SVaclav Hapla *has = (PetscBool)(type == H5O_TYPE_DATASET); 15309566063dSJacob Faibussowitsch PetscCall(PetscFree(abspath)); 15313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 153289e0ef10SVaclav Hapla } 153389e0ef10SVaclav Hapla 15340a9f38efSVaclav Hapla /*@ 1535e3f143f7SVaclav Hapla PetscViewerHDF5HasObject - Check whether a dataset with the same name as given object exists in the HDF5 file under current group 1536ecce1506SVaclav Hapla 1537343a20b2SVaclav Hapla Collective 1538343a20b2SVaclav Hapla 1539ecce1506SVaclav Hapla Input Parameters: 1540811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 1541ecce1506SVaclav Hapla - obj - The named object 1542ecce1506SVaclav Hapla 1543ecce1506SVaclav Hapla Output Parameter: 154489e0ef10SVaclav Hapla . has - Flag for dataset existence 1545ecce1506SVaclav Hapla 15463f423023SBarry Smith Level: advanced 15473f423023SBarry Smith 1548e3f143f7SVaclav Hapla Notes: 154989e0ef10SVaclav Hapla If the object is unnamed, an error occurs. 1550811af0c4SBarry Smith 1551811af0c4SBarry Smith If the path current_group/object_name exists but is not a dataset, has is set to `PETSC_FALSE` as well. 1552e3f143f7SVaclav Hapla 1553d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5HasDataset()`, `PetscViewerHDF5HasAttribute()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1554ecce1506SVaclav Hapla @*/ 1555d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5HasObject(PetscViewer viewer, PetscObject obj, PetscBool *has) 1556d71ae5a4SJacob Faibussowitsch { 155789e0ef10SVaclav Hapla size_t len; 1558ecce1506SVaclav Hapla 1559ecce1506SVaclav Hapla PetscFunctionBegin; 1560c57a1a9aSVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1561c57a1a9aSVaclav Hapla PetscValidHeader(obj, 2); 15624f572ea9SToby Isaac PetscAssertPointer(has, 3); 15639566063dSJacob Faibussowitsch PetscCall(PetscStrlen(obj->name, &len)); 15645f80ce2aSJacob Faibussowitsch PetscCheck(len, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONG, "Object must be named"); 15659566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5HasDataset(viewer, obj->name, has)); 15663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1567ecce1506SVaclav Hapla } 1568ecce1506SVaclav Hapla 1569df863907SAlex Fikl /*@C 1570ece83b6cSVaclav Hapla PetscViewerHDF5HasAttribute - Check whether an attribute exists 1571e7bdbf8eSMatthew G. Knepley 1572343a20b2SVaclav Hapla Collective 1573343a20b2SVaclav Hapla 1574e7bdbf8eSMatthew G. Knepley Input Parameters: 1575811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 15764302210dSVaclav Hapla . parent - The parent dataset/group name 1577e7bdbf8eSMatthew G. Knepley - name - The attribute name 1578e7bdbf8eSMatthew G. Knepley 1579e7bdbf8eSMatthew G. Knepley Output Parameter: 1580e7bdbf8eSMatthew G. Knepley . has - Flag for attribute existence 1581e7bdbf8eSMatthew G. Knepley 1582e7bdbf8eSMatthew G. Knepley Level: advanced 1583e7bdbf8eSMatthew G. Knepley 1584811af0c4SBarry Smith Note: 15853f423023SBarry Smith If parent starts with '/', it is taken as an absolute path overriding currently pushed group, else parent is relative to the current pushed group. `NULL` means the current pushed group. 15864302210dSVaclav Hapla 1587d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5HasObjectAttribute()`, `PetscViewerHDF5WriteAttribute()`, `PetscViewerHDF5ReadAttribute()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1588e7bdbf8eSMatthew G. Knepley @*/ 1589d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5HasAttribute(PetscViewer viewer, const char parent[], const char name[], PetscBool *has) 1590d71ae5a4SJacob Faibussowitsch { 1591ce78bad3SBarry Smith const char *parentAbsPath; 1592e7bdbf8eSMatthew G. Knepley 1593e7bdbf8eSMatthew G. Knepley PetscFunctionBegin; 15945cdeb108SMatthew G. Knepley PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 15954f572ea9SToby Isaac if (parent) PetscAssertPointer(parent, 2); 15964f572ea9SToby Isaac PetscAssertPointer(name, 3); 15974f572ea9SToby Isaac PetscAssertPointer(has, 4); 159877717648SVaclav Hapla PetscCall(PetscViewerHDF5GetGroup(viewer, parent, &parentAbsPath)); 15999566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5Traverse_Internal(viewer, parentAbsPath, PETSC_FALSE, has, NULL)); 16009566063dSJacob Faibussowitsch if (*has) PetscCall(PetscViewerHDF5HasAttribute_Internal(viewer, parentAbsPath, name, has)); 16019566063dSJacob Faibussowitsch PetscCall(PetscFree(parentAbsPath)); 16023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 160306db490cSVaclav Hapla } 160406db490cSVaclav Hapla 1605577e0e04SVaclav Hapla /*@C 1606811af0c4SBarry Smith PetscViewerHDF5HasObjectAttribute - Check whether an attribute is attached to the dataset matching the given `PetscObject` by name 1607577e0e04SVaclav Hapla 1608343a20b2SVaclav Hapla Collective 1609343a20b2SVaclav Hapla 1610577e0e04SVaclav Hapla Input Parameters: 1611811af0c4SBarry Smith + viewer - The `PETSCVIEWERHDF5` viewer 1612577e0e04SVaclav Hapla . obj - The object whose name is used to lookup the parent dataset, relative to the current group. 1613577e0e04SVaclav Hapla - name - The attribute name 1614577e0e04SVaclav Hapla 1615577e0e04SVaclav Hapla Output Parameter: 1616577e0e04SVaclav Hapla . has - Flag for attribute existence 1617577e0e04SVaclav Hapla 16183f423023SBarry Smith Level: advanced 16193f423023SBarry Smith 1620811af0c4SBarry Smith Note: 1621577e0e04SVaclav Hapla This fails if current_group/object_name doesn't resolve to a dataset (the path doesn't exist or is not a dataset). 1622811af0c4SBarry Smith You might want to check first if it does using `PetscViewerHDF5HasObject()`. 1623577e0e04SVaclav Hapla 1624d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerHDF5HasAttribute()`, `PetscViewerHDF5WriteObjectAttribute()`, `PetscViewerHDF5ReadObjectAttribute()`, `PetscViewerHDF5HasObject()`, `PetscViewerHDF5PushGroup()`, `PetscViewerHDF5PopGroup()`, `PetscViewerHDF5GetGroup()` 1625577e0e04SVaclav Hapla @*/ 1626d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerHDF5HasObjectAttribute(PetscViewer viewer, PetscObject obj, const char name[], PetscBool *has) 1627d71ae5a4SJacob Faibussowitsch { 1628577e0e04SVaclav Hapla PetscFunctionBegin; 1629577e0e04SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 1630577e0e04SVaclav Hapla PetscValidHeader(obj, 2); 16314f572ea9SToby Isaac PetscAssertPointer(name, 3); 16324f572ea9SToby Isaac PetscAssertPointer(has, 4); 16339566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5CheckNamedObject_Internal(viewer, obj)); 16349566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5HasAttribute(viewer, obj->name, name, has)); 16353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1636577e0e04SVaclav Hapla } 1637577e0e04SVaclav Hapla 1638d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerHDF5HasAttribute_Internal(PetscViewer viewer, const char parent[], const char name[], PetscBool *has) 1639d71ae5a4SJacob Faibussowitsch { 164006db490cSVaclav Hapla hid_t h5; 164106db490cSVaclav Hapla htri_t hhas; 164206db490cSVaclav Hapla 164306db490cSVaclav Hapla PetscFunctionBegin; 16449566063dSJacob Faibussowitsch PetscCall(PetscViewerHDF5GetFileId(viewer, &h5)); 1645792fecdfSBarry Smith PetscCallHDF5Return(hhas, H5Aexists_by_name, (h5, parent, name, H5P_DEFAULT)); 16465cdeb108SMatthew G. Knepley *has = hhas ? PETSC_TRUE : PETSC_FALSE; 16473ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1648e7bdbf8eSMatthew G. Knepley } 1649e7bdbf8eSMatthew G. Knepley 1650a75e6a4aSMatthew G. Knepley /* 1651a75e6a4aSMatthew G. Knepley The variable Petsc_Viewer_HDF5_keyval is used to indicate an MPI attribute that 1652a75e6a4aSMatthew G. Knepley is attached to a communicator, in this case the attribute is a PetscViewer. 1653a75e6a4aSMatthew G. Knepley */ 1654d4c7638eSBarry Smith PetscMPIInt Petsc_Viewer_HDF5_keyval = MPI_KEYVAL_INVALID; 1655a75e6a4aSMatthew G. Knepley 1656a75e6a4aSMatthew G. Knepley /*@C 1657811af0c4SBarry Smith PETSC_VIEWER_HDF5_ - Creates an `PETSCVIEWERHDF5` `PetscViewer` shared by all processors in a communicator. 1658a75e6a4aSMatthew G. Knepley 1659d083f849SBarry Smith Collective 1660a75e6a4aSMatthew G. Knepley 1661a75e6a4aSMatthew G. Knepley Input Parameter: 16623f423023SBarry Smith . comm - the MPI communicator to share the `PETSCVIEWERHDF5` `PetscViewer` 1663a75e6a4aSMatthew G. Knepley 1664811af0c4SBarry Smith Options Database Key: 166510699b91SBarry Smith . -viewer_hdf5_filename <name> - name of the HDF5 file 1666a75e6a4aSMatthew G. Knepley 1667811af0c4SBarry Smith Environmental variable: 1668811af0c4SBarry Smith . `PETSC_VIEWER_HDF5_FILENAME` - name of the HDF5 file 1669a75e6a4aSMatthew G. Knepley 167020f4b53cSBarry Smith Level: intermediate 167120f4b53cSBarry Smith 1672811af0c4SBarry Smith Note: 1673811af0c4SBarry Smith Unlike almost all other PETSc routines, `PETSC_VIEWER_HDF5_()` does not return 1674811af0c4SBarry Smith an error code. The HDF5 `PetscViewer` is usually used in the form 1675a75e6a4aSMatthew G. Knepley $ XXXView(XXX object, PETSC_VIEWER_HDF5_(comm)); 1676a75e6a4aSMatthew G. Knepley 1677d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERHDF5`, `PetscViewerHDF5Open()`, `PetscViewerCreate()`, `PetscViewerDestroy()` 1678a75e6a4aSMatthew G. Knepley @*/ 1679d71ae5a4SJacob Faibussowitsch PetscViewer PETSC_VIEWER_HDF5_(MPI_Comm comm) 1680d71ae5a4SJacob Faibussowitsch { 1681a75e6a4aSMatthew G. Knepley PetscErrorCode ierr; 16823ba16761SJacob Faibussowitsch PetscMPIInt mpi_ierr; 1683a75e6a4aSMatthew G. Knepley PetscBool flg; 1684a75e6a4aSMatthew G. Knepley PetscViewer viewer; 1685a75e6a4aSMatthew G. Knepley char fname[PETSC_MAX_PATH_LEN]; 1686a75e6a4aSMatthew G. Knepley MPI_Comm ncomm; 1687a75e6a4aSMatthew G. Knepley 1688a75e6a4aSMatthew G. Knepley PetscFunctionBegin; 16899371c9d4SSatish Balay ierr = PetscCommDuplicate(comm, &ncomm, NULL); 16909371c9d4SSatish Balay if (ierr) { 16913ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " "); 16929371c9d4SSatish Balay PetscFunctionReturn(NULL); 16939371c9d4SSatish Balay } 1694a75e6a4aSMatthew G. Knepley if (Petsc_Viewer_HDF5_keyval == MPI_KEYVAL_INVALID) { 16953ba16761SJacob Faibussowitsch mpi_ierr = MPI_Comm_create_keyval(MPI_COMM_NULL_COPY_FN, MPI_COMM_NULL_DELETE_FN, &Petsc_Viewer_HDF5_keyval, NULL); 16963ba16761SJacob Faibussowitsch if (mpi_ierr) { 16973ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " "); 16989371c9d4SSatish Balay PetscFunctionReturn(NULL); 16999371c9d4SSatish Balay } 1700a75e6a4aSMatthew G. Knepley } 17013ba16761SJacob Faibussowitsch mpi_ierr = MPI_Comm_get_attr(ncomm, Petsc_Viewer_HDF5_keyval, (void **)&viewer, (int *)&flg); 17023ba16761SJacob Faibussowitsch if (mpi_ierr) { 17033ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " "); 17049371c9d4SSatish Balay PetscFunctionReturn(NULL); 17059371c9d4SSatish Balay } 1706a75e6a4aSMatthew G. Knepley if (!flg) { /* PetscViewer not yet created */ 1707a75e6a4aSMatthew G. Knepley ierr = PetscOptionsGetenv(ncomm, "PETSC_VIEWER_HDF5_FILENAME", fname, PETSC_MAX_PATH_LEN, &flg); 17089371c9d4SSatish Balay if (ierr) { 17093ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_REPEAT, " "); 17109371c9d4SSatish Balay PetscFunctionReturn(NULL); 17119371c9d4SSatish Balay } 1712a75e6a4aSMatthew G. Knepley if (!flg) { 1713c6a7a370SJeremy L Thompson ierr = PetscStrncpy(fname, "output.h5", sizeof(fname)); 17149371c9d4SSatish Balay if (ierr) { 17153ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_REPEAT, " "); 17169371c9d4SSatish Balay PetscFunctionReturn(NULL); 17179371c9d4SSatish Balay } 1718a75e6a4aSMatthew G. Knepley } 1719a75e6a4aSMatthew G. Knepley ierr = PetscViewerHDF5Open(ncomm, fname, FILE_MODE_WRITE, &viewer); 17209371c9d4SSatish Balay if (ierr) { 17213ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_REPEAT, " "); 17229371c9d4SSatish Balay PetscFunctionReturn(NULL); 17239371c9d4SSatish Balay } 1724a75e6a4aSMatthew G. Knepley ierr = PetscObjectRegisterDestroy((PetscObject)viewer); 17259371c9d4SSatish Balay if (ierr) { 17263ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_REPEAT, " "); 17279371c9d4SSatish Balay PetscFunctionReturn(NULL); 17289371c9d4SSatish Balay } 17293ba16761SJacob Faibussowitsch mpi_ierr = MPI_Comm_set_attr(ncomm, Petsc_Viewer_HDF5_keyval, (void *)viewer); 17303ba16761SJacob Faibussowitsch if (mpi_ierr) { 17313ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_INITIAL, " "); 17329371c9d4SSatish Balay PetscFunctionReturn(NULL); 17339371c9d4SSatish Balay } 1734a75e6a4aSMatthew G. Knepley } 1735a75e6a4aSMatthew G. Knepley ierr = PetscCommDestroy(&ncomm); 17369371c9d4SSatish Balay if (ierr) { 17373ba16761SJacob Faibussowitsch ierr = PetscError(PETSC_COMM_SELF, __LINE__, "PETSC_VIEWER_HDF5_", __FILE__, PETSC_ERR_PLIB, PETSC_ERROR_REPEAT, " "); 17389371c9d4SSatish Balay PetscFunctionReturn(NULL); 17399371c9d4SSatish Balay } 1740a75e6a4aSMatthew G. Knepley PetscFunctionReturn(viewer); 1741a75e6a4aSMatthew G. Knepley } 1742