xref: /petsc/src/sys/classes/viewer/impls/cgns/cgnsv.c (revision ec76947a4f8d19eaaeee63981d78f56521ad4f2e)
15f34f2dcSJed Brown #include <petsc/private/viewercgnsimpl.h> /*I "petscviewer.h" I*/
2472b9844SJames Wright #include <petsc/private/dmpleximpl.h>     /*I   "petscdmplex.h"   I*/
35f34f2dcSJed Brown #if defined(PETSC_HDF5_HAVE_PARALLEL)
45f34f2dcSJed Brown   #include <pcgnslib.h>
55f34f2dcSJed Brown #else
65f34f2dcSJed Brown   #include <cgnslib.h>
75f34f2dcSJed Brown #endif
8472b9844SJames Wright #include <cgns_io.h>
9472b9844SJames Wright #include <ctype.h>
105f34f2dcSJed Brown 
11d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerSetFromOptions_CGNS(PetscViewer v, PetscOptionItems *PetscOptionsObject)
12d71ae5a4SJacob Faibussowitsch {
1325760affSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)v->data;
144d86920dSPierre Jolivet 
155f34f2dcSJed Brown   PetscFunctionBegin;
165f34f2dcSJed Brown   PetscOptionsHeadBegin(PetscOptionsObject, "CGNS Viewer Options");
17da0802e2SStefano Zampini   PetscCall(PetscOptionsInt("-viewer_cgns_batch_size", "Max number of steps to store in single file when using a template cgns:name-\%d.cgns", "", cgv->batch_size, &cgv->batch_size, NULL));
185f34f2dcSJed Brown   PetscOptionsHeadEnd();
193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
205f34f2dcSJed Brown }
215f34f2dcSJed Brown 
22d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerView_CGNS(PetscViewer v, PetscViewer viewer)
23d71ae5a4SJacob Faibussowitsch {
245f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)v->data;
255f34f2dcSJed Brown 
265f34f2dcSJed Brown   PetscFunctionBegin;
2748a46eb9SPierre Jolivet   if (cgv->filename) PetscCall(PetscViewerASCIIPrintf(viewer, "Filename: %s\n", cgv->filename));
283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
295f34f2dcSJed Brown }
305f34f2dcSJed Brown 
31d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileClose_CGNS(PetscViewer viewer)
32d71ae5a4SJacob Faibussowitsch {
335f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
345f34f2dcSJed Brown 
355f34f2dcSJed Brown   PetscFunctionBegin;
365f34f2dcSJed Brown   if (cgv->output_times) {
376497c311SBarry Smith     PetscCount size, width = 32, *steps;
385f34f2dcSJed Brown     char      *solnames;
395f34f2dcSJed Brown     PetscReal *times;
405f34f2dcSJed Brown     cgsize_t   num_times;
415f34f2dcSJed Brown     PetscCall(PetscSegBufferGetSize(cgv->output_times, &size));
425f34f2dcSJed Brown     PetscCall(PetscSegBufferExtractInPlace(cgv->output_times, &times));
435f34f2dcSJed Brown     num_times = size;
445f34f2dcSJed Brown     PetscCallCGNS(cg_biter_write(cgv->file_num, cgv->base, "TimeIterValues", num_times));
455f34f2dcSJed Brown     PetscCallCGNS(cg_goto(cgv->file_num, cgv->base, "BaseIterativeData_t", 1, NULL));
465f34f2dcSJed Brown     PetscCallCGNS(cg_array_write("TimeValues", CGNS_ENUMV(RealDouble), 1, &num_times, times));
475f34f2dcSJed Brown     PetscCall(PetscSegBufferDestroy(&cgv->output_times));
485f34f2dcSJed Brown     PetscCallCGNS(cg_ziter_write(cgv->file_num, cgv->base, cgv->zone, "ZoneIterativeData"));
495f34f2dcSJed Brown     PetscCallCGNS(cg_goto(cgv->file_num, cgv->base, "Zone_t", cgv->zone, "ZoneIterativeData_t", 1, NULL));
505f34f2dcSJed Brown     PetscCall(PetscMalloc(size * width + 1, &solnames));
519812b6beSJed Brown     PetscCall(PetscSegBufferExtractInPlace(cgv->output_steps, &steps));
526497c311SBarry Smith     for (PetscCount i = 0; i < size; i++) PetscCall(PetscSNPrintf(&solnames[i * width], width + 1, "FlowSolution%-20zu", (size_t)steps[i]));
539812b6beSJed Brown     PetscCall(PetscSegBufferDestroy(&cgv->output_steps));
545f34f2dcSJed Brown     cgsize_t shape[2] = {(cgsize_t)width, (cgsize_t)size};
555f34f2dcSJed Brown     PetscCallCGNS(cg_array_write("FlowSolutionPointers", CGNS_ENUMV(Character), 2, shape, solnames));
565f34f2dcSJed Brown     // The VTK reader looks for names like FlowSolution*Pointers.
576497c311SBarry Smith     for (PetscCount i = 0; i < size; i++) PetscCall(PetscSNPrintf(&solnames[i * width], width + 1, "%-32s", "CellInfo"));
585f34f2dcSJed Brown     PetscCallCGNS(cg_array_write("FlowSolutionCellInfoPointers", CGNS_ENUMV(Character), 2, shape, solnames));
595f34f2dcSJed Brown     PetscCall(PetscFree(solnames));
605f34f2dcSJed Brown 
615f34f2dcSJed Brown     PetscCallCGNS(cg_simulation_type_write(cgv->file_num, cgv->base, CGNS_ENUMV(TimeAccurate)));
625f34f2dcSJed Brown   }
635f34f2dcSJed Brown   PetscCall(PetscFree(cgv->filename));
645f34f2dcSJed Brown #if defined(PETSC_HDF5_HAVE_PARALLEL)
65753e5362SJed Brown   if (cgv->file_num) PetscCallCGNS(cgp_close(cgv->file_num));
665f34f2dcSJed Brown #else
675f34f2dcSJed Brown   if (cgv->file_num) PetscCallCGNS(cg_close(cgv->file_num));
685f34f2dcSJed Brown #endif
695f34f2dcSJed Brown   cgv->file_num = 0;
7025760affSJed Brown   PetscCall(PetscFree(cgv->node_l2g));
7125760affSJed Brown   PetscCall(PetscFree(cgv->nodal_field));
723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7325760affSJed Brown }
7425760affSJed Brown 
7525760affSJed Brown PetscErrorCode PetscViewerCGNSFileOpen_Internal(PetscViewer viewer, PetscInt sequence_number)
7625760affSJed Brown {
7725760affSJed Brown   PetscViewer_CGNS *cgv          = (PetscViewer_CGNS *)viewer->data;
78*ec76947aSJames Wright   int               cg_file_mode = -1;
794d86920dSPierre Jolivet 
8025760affSJed Brown   PetscFunctionBegin;
8125760affSJed Brown   PetscCheck((cgv->filename == NULL) ^ (sequence_number < 0), PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_INCOMP, "Expect either a template filename or non-negative sequence number");
8225760affSJed Brown   if (!cgv->filename) {
8325760affSJed Brown     char filename_numbered[PETSC_MAX_PATH_LEN];
8425760affSJed Brown     // Cast sequence_number so %d can be used also when PetscInt is 64-bit. We could upgrade the format string if users
8525760affSJed Brown     // run more than 2B time steps.
8625760affSJed Brown     PetscCall(PetscSNPrintf(filename_numbered, sizeof filename_numbered, cgv->filename_template, (int)sequence_number));
8725760affSJed Brown     PetscCall(PetscStrallocpy(filename_numbered, &cgv->filename));
8825760affSJed Brown   }
8925760affSJed Brown   switch (cgv->btype) {
9025760affSJed Brown   case FILE_MODE_READ:
91*ec76947aSJames Wright     cg_file_mode = CG_MODE_READ;
9225760affSJed Brown     break;
9325760affSJed Brown   case FILE_MODE_WRITE:
94*ec76947aSJames Wright     cg_file_mode = CG_MODE_WRITE;
9525760affSJed Brown     break;
9625760affSJed Brown   case FILE_MODE_UNDEFINED:
9725760affSJed Brown     SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
9825760affSJed Brown   default:
9925760affSJed Brown     SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Unsupported file mode %s", PetscFileModes[cgv->btype]);
10025760affSJed Brown   }
101*ec76947aSJames Wright #if defined(PETSC_HDF5_HAVE_PARALLEL)
102*ec76947aSJames Wright   PetscCallCGNS(cgp_mpi_comm(PetscObjectComm((PetscObject)viewer)));
103*ec76947aSJames Wright   PetscCallCGNS(cgp_open(cgv->filename, cg_file_mode, &cgv->file_num));
104*ec76947aSJames Wright #else
105*ec76947aSJames Wright   PetscCallCGNS(cg_open(filename, cg_file_mode, &cgv->file_num));
106*ec76947aSJames Wright #endif
1073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
10825760affSJed Brown }
10925760affSJed Brown 
11025760affSJed Brown PetscErrorCode PetscViewerCGNSCheckBatch_Internal(PetscViewer viewer)
11125760affSJed Brown {
11225760affSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
1136497c311SBarry Smith   PetscCount        num_steps;
11425760affSJed Brown 
11525760affSJed Brown   PetscFunctionBegin;
1163ba16761SJacob Faibussowitsch   if (!cgv->filename_template) PetscFunctionReturn(PETSC_SUCCESS); // Batches are closed when viewer is destroyed
11725760affSJed Brown   PetscCall(PetscSegBufferGetSize(cgv->output_times, &num_steps));
1186497c311SBarry Smith   if (num_steps >= (PetscCount)cgv->batch_size) PetscCall(PetscViewerFileClose_CGNS(viewer));
1193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1205f34f2dcSJed Brown }
1215f34f2dcSJed Brown 
122d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerDestroy_CGNS(PetscViewer viewer)
123d71ae5a4SJacob Faibussowitsch {
1245f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
1255f34f2dcSJed Brown 
1265f34f2dcSJed Brown   PetscFunctionBegin;
1275f34f2dcSJed Brown   PetscCall(PetscViewerFileClose_CGNS(viewer));
128472b9844SJames Wright   PetscCall(PetscFree(cgv->solution_name));
1295f34f2dcSJed Brown   PetscCall(PetscFree(cgv));
1305f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetName_C", NULL));
1315f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetName_C", NULL));
1325f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetMode_C", NULL));
1335f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetMode_C", NULL));
1343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1355f34f2dcSJed Brown }
1365f34f2dcSJed Brown 
137d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetMode_CGNS(PetscViewer viewer, PetscFileMode type)
138d71ae5a4SJacob Faibussowitsch {
1395f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
1405f34f2dcSJed Brown 
1415f34f2dcSJed Brown   PetscFunctionBegin;
1425f34f2dcSJed Brown   cgv->btype = type;
1433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1445f34f2dcSJed Brown }
1455f34f2dcSJed Brown 
146d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetMode_CGNS(PetscViewer viewer, PetscFileMode *type)
147d71ae5a4SJacob Faibussowitsch {
1485f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
1495f34f2dcSJed Brown 
1505f34f2dcSJed Brown   PetscFunctionBegin;
1515f34f2dcSJed Brown   *type = cgv->btype;
1523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1535f34f2dcSJed Brown }
1545f34f2dcSJed Brown 
155d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetName_CGNS(PetscViewer viewer, const char *filename)
156d71ae5a4SJacob Faibussowitsch {
1575f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
15825760affSJed Brown   char             *has_pattern;
1595f34f2dcSJed Brown 
1605f34f2dcSJed Brown   PetscFunctionBegin;
1615f34f2dcSJed Brown   if (cgv->file_num) PetscCallCGNS(cg_close(cgv->file_num));
1625f34f2dcSJed Brown   PetscCall(PetscFree(cgv->filename));
16325760affSJed Brown   PetscCall(PetscFree(cgv->filename_template));
16425760affSJed Brown   PetscCall(PetscStrchr(filename, '%', &has_pattern));
16525760affSJed Brown   if (has_pattern) {
16625760affSJed Brown     PetscCall(PetscStrallocpy(filename, &cgv->filename_template));
16725760affSJed Brown     // Templated file names open lazily, once we know DMGetOutputSequenceNumber()
16825760affSJed Brown   } else {
1695f34f2dcSJed Brown     PetscCall(PetscStrallocpy(filename, &cgv->filename));
17025760affSJed Brown     PetscCall(PetscViewerCGNSFileOpen_Internal(viewer, -1));
1715f34f2dcSJed Brown   }
1723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1735f34f2dcSJed Brown }
1745f34f2dcSJed Brown 
175d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetName_CGNS(PetscViewer viewer, const char **filename)
176d71ae5a4SJacob Faibussowitsch {
1775f34f2dcSJed Brown   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
1785f34f2dcSJed Brown 
1795f34f2dcSJed Brown   PetscFunctionBegin;
1805f34f2dcSJed Brown   *filename = cgv->filename;
1813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1825f34f2dcSJed Brown }
1835f34f2dcSJed Brown 
1845f34f2dcSJed Brown /*MC
1855f34f2dcSJed Brown    PETSCVIEWERCGNS - A viewer for CGNS files
1865f34f2dcSJed Brown 
1875f34f2dcSJed Brown   Level: beginner
188811af0c4SBarry Smith 
1893f423023SBarry Smith   Options Database Key:
1903f423023SBarry Smith . -viewer_cgns_batch_size SIZE - set max number of output sequence times to write per batch
19125760affSJed Brown 
1923f423023SBarry Smith   Note:
19325760affSJed Brown   If the filename contains an integer format character, the CGNS viewer will created a batched output sequence. For
19425760affSJed Brown   example, one could use `-ts_monitor_solution cgns:flow-%d.cgns`. This is desirable if one wants to limit file sizes or
19525760affSJed Brown   if the job might crash/be killed by a resource manager before exiting cleanly.
19625760affSJed Brown 
19725760affSJed Brown .seealso: [](sec_viewers), `PetscViewer`, `PetscViewerCreate()`, `VecView()`, `DMView()`, `PetscViewerFileSetName()`, `PetscViewerFileSetMode()`, `TSSetFromOptions()`
1985f34f2dcSJed Brown M*/
19914640894SJames Wright PetscErrorCode PetscViewerCreate_CGNS(PetscViewer v)
200d71ae5a4SJacob Faibussowitsch {
2015f34f2dcSJed Brown   PetscViewer_CGNS *cgv;
2025f34f2dcSJed Brown 
2035f34f2dcSJed Brown   PetscFunctionBegin;
2044dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&cgv));
2055f34f2dcSJed Brown 
2065f34f2dcSJed Brown   v->data                = cgv;
2075f34f2dcSJed Brown   v->ops->destroy        = PetscViewerDestroy_CGNS;
2085f34f2dcSJed Brown   v->ops->setfromoptions = PetscViewerSetFromOptions_CGNS;
2095f34f2dcSJed Brown   v->ops->view           = PetscViewerView_CGNS;
2105f34f2dcSJed Brown   cgv->btype             = FILE_MODE_UNDEFINED;
2115f34f2dcSJed Brown   cgv->filename          = NULL;
21225760affSJed Brown   cgv->batch_size        = 20;
213472b9844SJames Wright   cgv->solution_index    = -1; // Default to use the "last" solution
214472b9844SJames Wright   cgv->base              = 1;
215472b9844SJames Wright   cgv->zone              = 1;
2165f34f2dcSJed Brown 
2175f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetName_C", PetscViewerFileSetName_CGNS));
2185f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetName_C", PetscViewerFileGetName_CGNS));
2195f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileSetMode_C", PetscViewerFileSetMode_CGNS));
2205f34f2dcSJed Brown   PetscCall(PetscObjectComposeFunction((PetscObject)v, "PetscViewerFileGetMode_C", PetscViewerFileGetMode_CGNS));
2213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2225f34f2dcSJed Brown }
223472b9844SJames Wright 
224472b9844SJames Wright // Find DataArray_t node under the current node (determined by `cg_goto` and friends) that matches `name`
225472b9844SJames Wright // Return the index of that array and (optionally) other data about the array
226472b9844SJames Wright static inline PetscErrorCode CGNS_Find_Array(MPI_Comm comm, const char name[], int *A_index, CGNS_ENUMT(DataType_t) * data_type, int *dim, cgsize_t size[])
227472b9844SJames Wright {
228472b9844SJames Wright   int  narrays; // number of arrays under the current node
229472b9844SJames Wright   char array_name[CGIO_MAX_NAME_LENGTH + 1];
230472b9844SJames Wright   CGNS_ENUMT(DataType_t) data_type_local;
231472b9844SJames Wright   int       _dim;
232472b9844SJames Wright   cgsize_t  _size[12];
233472b9844SJames Wright   PetscBool matches_name = PETSC_FALSE;
234472b9844SJames Wright 
235472b9844SJames Wright   PetscFunctionBeginUser;
236472b9844SJames Wright   PetscCallCGNS(cg_narrays(&narrays));
237472b9844SJames Wright   for (int i = 1; i <= narrays; i++) {
238472b9844SJames Wright     PetscCallCGNS(cg_array_info(i, array_name, &data_type_local, &_dim, _size));
239472b9844SJames Wright     PetscCall(PetscStrcmp(name, array_name, &matches_name));
240472b9844SJames Wright     if (matches_name) {
241472b9844SJames Wright       *A_index = i;
242472b9844SJames Wright       if (data_type) *data_type = data_type_local;
243472b9844SJames Wright       if (dim) *dim = _dim;
244472b9844SJames Wright       if (size) PetscArraycpy(size, _size, _dim);
245472b9844SJames Wright       PetscFunctionReturn(PETSC_SUCCESS);
246472b9844SJames Wright     }
247472b9844SJames Wright   }
248472b9844SJames Wright   SETERRQ(comm, PETSC_ERR_SUP, "Cannot find %s array under current CGNS node", name);
249472b9844SJames Wright }
250472b9844SJames Wright 
251472b9844SJames Wright /*@
252472b9844SJames Wright   PetscViewerCGNSOpen - Opens a file for CGNS input/output.
253472b9844SJames Wright 
254472b9844SJames Wright   Collective
255472b9844SJames Wright 
256472b9844SJames Wright   Input Parameters:
257472b9844SJames Wright + comm - MPI communicator
258472b9844SJames Wright . name - name of file
259472b9844SJames Wright - type - type of file
260472b9844SJames Wright .vb
261472b9844SJames Wright     FILE_MODE_WRITE - create new file for binary output
262472b9844SJames Wright     FILE_MODE_READ - open existing file for binary input
263472b9844SJames Wright     FILE_MODE_APPEND - open existing file for binary output
264472b9844SJames Wright .ve
265472b9844SJames Wright 
266472b9844SJames Wright   Output Parameter:
267472b9844SJames Wright . viewer - `PETSCVIEWERCGNS` `PetscViewer` for CGNS input/output to use with the specified file
268472b9844SJames Wright 
269472b9844SJames Wright   Level: beginner
270472b9844SJames Wright 
271472b9844SJames Wright .seealso: `PETSCVIEWERCGNS`, `PetscViewer`, `PetscViewerPushFormat()`, `PetscViewerDestroy()`,
272472b9844SJames Wright           `DMLoad()`, `PetscFileMode`, `PetscViewerSetType()`, `PetscViewerFileSetMode()`, `PetscViewerFileSetName()`
273472b9844SJames Wright @*/
274472b9844SJames Wright PetscErrorCode PetscViewerCGNSOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *viewer)
275472b9844SJames Wright {
276472b9844SJames Wright   PetscFunctionBegin;
27714640894SJames Wright   PetscAssertPointer(name, 2);
27814640894SJames Wright   PetscAssertPointer(viewer, 4);
279472b9844SJames Wright   PetscCall(PetscViewerCreate(comm, viewer));
280472b9844SJames Wright   PetscCall(PetscViewerSetType(*viewer, PETSCVIEWERCGNS));
281472b9844SJames Wright   PetscCall(PetscViewerFileSetMode(*viewer, type));
282472b9844SJames Wright   PetscCall(PetscViewerFileSetName(*viewer, name));
283472b9844SJames Wright   PetscCall(PetscViewerSetFromOptions(*viewer));
284472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
285472b9844SJames Wright }
286472b9844SJames Wright 
287472b9844SJames Wright /*@
288472b9844SJames Wright   PetscViewerCGNSSetSolutionIndex - Set index of solution
289472b9844SJames Wright 
290472b9844SJames Wright   Not Collective
291472b9844SJames Wright 
292472b9844SJames Wright   Input Parameters:
293472b9844SJames Wright + viewer      - `PETSCVIEWERCGNS` `PetscViewer` for CGNS input/output to use with the specified file
294472b9844SJames Wright - solution_id - Index of the solution id, or `-1` for the last solution on the file
295472b9844SJames Wright 
296472b9844SJames Wright   Level: intermediate
297472b9844SJames Wright 
298472b9844SJames Wright   Notes:
299472b9844SJames Wright   By default, `solution_id` is set to `-1` to mean the last solution available in the file.
300472b9844SJames Wright   If the file contains a FlowSolutionPointers node, then that array is indexed to determine which FlowSolution_t node to read from.
301472b9844SJames Wright   Otherwise, `solution_id` indexes the total available FlowSolution_t nodes in the file.
302472b9844SJames Wright 
303472b9844SJames Wright   This solution index is used by `VecLoad()` to determine which solution to load from the file
304472b9844SJames Wright 
305472b9844SJames Wright .seealso: `PETSCVIEWERCGNS`, `PetscViewerCGNSGetSolutionIndex()`, `PetscViewerCGNSGetSolutionInfo()`
306472b9844SJames Wright 
307472b9844SJames Wright @*/
308472b9844SJames Wright PetscErrorCode PetscViewerCGNSSetSolutionIndex(PetscViewer viewer, PetscInt solution_id)
309472b9844SJames Wright {
310472b9844SJames Wright   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
311472b9844SJames Wright 
312472b9844SJames Wright   PetscFunctionBeginUser;
31314640894SJames Wright   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
31414640894SJames Wright   PetscValidLogicalCollectiveInt(viewer, solution_id, 2);
315472b9844SJames Wright   PetscCheck((solution_id != 0) && (solution_id > -2), PetscObjectComm((PetscObject)viewer), PETSC_ERR_USER_INPUT, "Solution index must be either -1 or greater than 0, not %" PetscInt_FMT, solution_id);
316472b9844SJames Wright   cgv->solution_index      = solution_id;
317472b9844SJames Wright   cgv->solution_file_index = 0; // Reset sol_index when solution_id changes (0 is invalid)
318472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
319472b9844SJames Wright }
320472b9844SJames Wright 
321472b9844SJames Wright /*@
322472b9844SJames Wright   PetscViewerCGNSGetSolutionIndex - Get index of solution
323472b9844SJames Wright 
324472b9844SJames Wright   Not Collective
325472b9844SJames Wright 
326472b9844SJames Wright   Input Parameter:
327472b9844SJames Wright . viewer - `PETSCVIEWERCGNS` `PetscViewer` for CGNS input/output to use with the specified file
328472b9844SJames Wright 
329472b9844SJames Wright   Output Parameter:
330472b9844SJames Wright . solution_id - Index of the solution id
331472b9844SJames Wright 
332472b9844SJames Wright   Level: intermediate
333472b9844SJames Wright 
334472b9844SJames Wright   Notes:
335472b9844SJames Wright   By default, solution_id is set to `-1` to mean the last solution available in the file
336472b9844SJames Wright 
337472b9844SJames Wright .seealso: `PETSCVIEWERCGNS`, `PetscViewerCGNSSetSolutionIndex()`, `PetscViewerCGNSGetSolutionInfo()`
338472b9844SJames Wright 
339472b9844SJames Wright @*/
340472b9844SJames Wright PetscErrorCode PetscViewerCGNSGetSolutionIndex(PetscViewer viewer, PetscInt *solution_id)
341472b9844SJames Wright {
342472b9844SJames Wright   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
343472b9844SJames Wright 
344472b9844SJames Wright   PetscFunctionBeginUser;
34514640894SJames Wright   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
34614640894SJames Wright   PetscAssertPointer(solution_id, 2);
347472b9844SJames Wright   *solution_id = cgv->solution_index;
348472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
349472b9844SJames Wright }
350472b9844SJames Wright 
351472b9844SJames Wright // Gets the index for the solution in this CGNS file
352472b9844SJames Wright PetscErrorCode PetscViewerCGNSGetSolutionFileIndex_Internal(PetscViewer viewer, int *sol_index)
353472b9844SJames Wright {
354472b9844SJames Wright   PetscViewer_CGNS *cgv  = (PetscViewer_CGNS *)viewer->data;
355472b9844SJames Wright   MPI_Comm          comm = PetscObjectComm((PetscObject)viewer);
356472b9844SJames Wright   int               nsols, cgns_ier;
357472b9844SJames Wright   char              buffer[CGIO_MAX_NAME_LENGTH + 1];
358472b9844SJames Wright   CGNS_ENUMT(GridLocation_t) gridloc; // Throwaway
359472b9844SJames Wright 
360472b9844SJames Wright   PetscFunctionBeginUser;
361472b9844SJames Wright   if (cgv->solution_file_index > 0) {
362472b9844SJames Wright     *sol_index = cgv->solution_file_index;
363472b9844SJames Wright     PetscFunctionReturn(PETSC_SUCCESS);
364472b9844SJames Wright   }
365472b9844SJames Wright 
366472b9844SJames Wright   PetscCallCGNS(cg_nsols(cgv->file_num, cgv->base, cgv->zone, &nsols));
367472b9844SJames Wright   cgns_ier = cg_goto(cgv->file_num, cgv->base, "Zone_t", cgv->zone, "ZoneIterativeData_t", 1, "FlowSolutionPointers", 0, NULL);
368472b9844SJames Wright   if (cgns_ier == CG_NODE_NOT_FOUND) {
369472b9844SJames Wright     // If FlowSolutionPointers does not exist, then just index off of nsols (which can include non-solution data)
370472b9844SJames Wright     PetscCheck(cgv->solution_index == -1 || cgv->solution_index <= nsols, comm, PETSC_ERR_ARG_OUTOFRANGE, "CGNS Solution index (%" PetscInt_FMT ") not in [1, %d]", cgv->solution_index, nsols);
371472b9844SJames Wright 
372472b9844SJames Wright     cgv->solution_file_index = cgv->solution_index == -1 ? nsols : cgv->solution_index;
373472b9844SJames Wright   } else {
374472b9844SJames Wright     // If FlowSolutionPointers exists, then solution_id should index that array of FlowSolutions
375472b9844SJames Wright     char     *pointer_id_name;
376472b9844SJames Wright     PetscBool matches_name = PETSC_FALSE;
377472b9844SJames Wright     int       sol_id;
378472b9844SJames Wright 
379472b9844SJames Wright     PetscCheck(cgns_ier == CG_OK, PETSC_COMM_SELF, PETSC_ERR_LIB, "CGNS error %d %s", cgns_ier, cg_get_error());
380472b9844SJames Wright     cgns_ier = cg_goto(cgv->file_num, cgv->base, "Zone_t", cgv->zone, "ZoneIterativeData_t", 1, NULL);
381472b9844SJames Wright 
382472b9844SJames Wright     { // Get FlowSolutionPointer name corresponding to solution_id
383472b9844SJames Wright       cgsize_t size[12];
384472b9844SJames Wright       int      dim, A_index, pointer_id;
385472b9844SJames Wright       char    *pointer_names, *pointer_id_name_ref;
386472b9844SJames Wright 
387472b9844SJames Wright       PetscCall(CGNS_Find_Array(comm, "FlowSolutionPointers", &A_index, NULL, &dim, size));
388472b9844SJames Wright       PetscCheck(cgv->solution_index == -1 || cgv->solution_index <= size[1], comm, PETSC_ERR_ARG_OUTOFRANGE, "CGNS Solution index (%" PetscInt_FMT ") not in range of FlowSolutionPointers [1, %" PRIdCGSIZE "]", cgv->solution_index, size[1]);
389472b9844SJames Wright       PetscCall(PetscCalloc1(size[0] * size[1] + 1, &pointer_names)); // Need the +1 for (possibly) setting \0 for the last pointer name if it's full
390472b9844SJames Wright       PetscCallCGNS(cg_array_read_as(1, CGNS_ENUMV(Character), pointer_names));
391472b9844SJames Wright       pointer_id          = cgv->solution_index == -1 ? size[1] : cgv->solution_index;
392472b9844SJames Wright       pointer_id_name_ref = &pointer_names[size[0] * (pointer_id - 1)];
393472b9844SJames Wright       { // Set last non-whitespace character of the pointer name to \0 (CGNS pads with spaces)
394472b9844SJames Wright         int str_idx;
395472b9844SJames Wright         for (str_idx = size[0] - 1; str_idx > 0; str_idx--) {
396472b9844SJames Wright           if (!isspace((unsigned char)pointer_id_name_ref[str_idx])) break;
397472b9844SJames Wright         }
398472b9844SJames Wright         pointer_id_name_ref[str_idx + 1] = '\0';
399472b9844SJames Wright       }
400472b9844SJames Wright       PetscCall(PetscStrallocpy(pointer_id_name_ref, &pointer_id_name));
401472b9844SJames Wright       PetscCall(PetscFree(pointer_names));
402472b9844SJames Wright     }
403472b9844SJames Wright 
404472b9844SJames Wright     // Find FlowSolution_t node that matches pointer_id_name
405472b9844SJames Wright     for (sol_id = 1; sol_id <= nsols; sol_id++) {
406472b9844SJames Wright       PetscCallCGNS(cg_sol_info(cgv->file_num, cgv->base, cgv->zone, sol_id, buffer, &gridloc));
407472b9844SJames Wright       PetscCall(PetscStrcmp(pointer_id_name, buffer, &matches_name));
408472b9844SJames Wright       if (matches_name) break;
409472b9844SJames Wright     }
410472b9844SJames Wright     PetscCheck(matches_name, comm, PETSC_ERR_LIB, "Cannot find FlowSolution_t node %s in file", pointer_id_name);
411472b9844SJames Wright     cgv->solution_file_index = sol_id;
412472b9844SJames Wright     PetscCall(PetscFree(pointer_id_name));
413472b9844SJames Wright   }
414472b9844SJames Wright 
415472b9844SJames Wright   *sol_index = cgv->solution_file_index;
416472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
417472b9844SJames Wright }
418472b9844SJames Wright 
419472b9844SJames Wright /*@
420472b9844SJames Wright   PetscViewerCGNSGetSolutionTime - Gets the solution time for the FlowSolution of the viewer
421472b9844SJames Wright 
422472b9844SJames Wright   Collective
423472b9844SJames Wright 
424472b9844SJames Wright   Input Parameter:
425472b9844SJames Wright . viewer - `PETSCVIEWERCGNS` `PetscViewer` for CGNS input/output to use with the specified file
426472b9844SJames Wright 
4277dcfde4dSJose E. Roman   Output Parameters:
428472b9844SJames Wright + time - Solution time of the FlowSolution_t node
429472b9844SJames Wright - set  - Whether the time data is in the file
430472b9844SJames Wright 
431472b9844SJames Wright   Level: intermediate
432472b9844SJames Wright 
433472b9844SJames Wright   Notes:
434472b9844SJames Wright   Reads data from a DataArray named `TimeValues` under a `BaseIterativeData_t` node
435472b9844SJames Wright 
436472b9844SJames Wright .seealso: `PETSCVIEWERCGNS`, `PetscViewer`, `PetscViewerCGNSSetSolutionIndex()`, `PetscViewerCGNSGetSolutionIndex()`, `PetscViewerCGNSGetSolutionName()`
437472b9844SJames Wright @*/
438472b9844SJames Wright PetscErrorCode PetscViewerCGNSGetSolutionTime(PetscViewer viewer, PetscReal *time, PetscBool *set)
439472b9844SJames Wright {
440472b9844SJames Wright   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
441472b9844SJames Wright   int               cgns_ier, A_index = 0;
442472b9844SJames Wright   PetscReal        *times;
443472b9844SJames Wright   cgsize_t          size[12];
444472b9844SJames Wright 
445472b9844SJames Wright   PetscFunctionBeginUser;
44614640894SJames Wright   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
44714640894SJames Wright   PetscAssertPointer(time, 2);
44814640894SJames Wright   PetscAssertPointer(set, 3);
449472b9844SJames Wright   cgns_ier = cg_goto(cgv->file_num, cgv->base, "BaseIterativeData_t", 1, NULL);
450472b9844SJames Wright   if (cgns_ier == CG_NODE_NOT_FOUND) {
451472b9844SJames Wright     *set = PETSC_FALSE;
452472b9844SJames Wright     PetscFunctionReturn(PETSC_SUCCESS);
453472b9844SJames Wright   } else PetscCallCGNS(cgns_ier);
454472b9844SJames Wright   PetscCall(CGNS_Find_Array(PetscObjectComm((PetscObject)viewer), "TimeValues", &A_index, NULL, NULL, size));
455472b9844SJames Wright   PetscCall(PetscMalloc1(size[0], &times));
456472b9844SJames Wright   PetscCallCGNS(cg_array_read_as(A_index, CGNS_ENUMV(RealDouble), times));
457472b9844SJames Wright   *time = times[cgv->solution_index - 1];
458472b9844SJames Wright   *set  = PETSC_TRUE;
459472b9844SJames Wright   PetscCall(PetscFree(times));
460472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
461472b9844SJames Wright }
462472b9844SJames Wright 
463472b9844SJames Wright /*@
464472b9844SJames Wright   PetscViewerCGNSGetSolutionName - Gets name of FlowSolution of the viewer
465472b9844SJames Wright 
466472b9844SJames Wright   Collective
467472b9844SJames Wright 
468472b9844SJames Wright   Input Parameter:
469472b9844SJames Wright . viewer - `PETSCVIEWERCGNS` `PetscViewer` for CGNS input/output to use with the specified file
470472b9844SJames Wright 
471472b9844SJames Wright   Output Parameter:
472472b9844SJames Wright . name - Name of the FlowSolution_t node corresponding to the solution index
473472b9844SJames Wright 
474472b9844SJames Wright   Level: intermediate
475472b9844SJames Wright 
476472b9844SJames Wright   Notes:
477472b9844SJames Wright   Currently assumes there is only one Zone in the CGNS file
478472b9844SJames Wright 
479472b9844SJames Wright .seealso: `PETSCVIEWERCGNS`, `PetscViewer`, `PetscViewerCGNSSetSolutionIndex()`, `PetscViewerCGNSGetSolutionIndex()`, `PetscViewerCGNSGetSolutionTime()`
480472b9844SJames Wright @*/
481472b9844SJames Wright PetscErrorCode PetscViewerCGNSGetSolutionName(PetscViewer viewer, const char *name[])
482472b9844SJames Wright {
483472b9844SJames Wright   PetscViewer_CGNS *cgv = (PetscViewer_CGNS *)viewer->data;
484472b9844SJames Wright   int               sol_id;
485472b9844SJames Wright   char              buffer[CGIO_MAX_NAME_LENGTH + 1];
486472b9844SJames Wright   CGNS_ENUMT(GridLocation_t) gridloc; // Throwaway
487472b9844SJames Wright 
488472b9844SJames Wright   PetscFunctionBeginUser;
48914640894SJames Wright   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
49014640894SJames Wright   PetscAssertPointer(name, 2);
491472b9844SJames Wright   PetscCall(PetscViewerCGNSGetSolutionFileIndex_Internal(viewer, &sol_id));
492472b9844SJames Wright 
493472b9844SJames Wright   PetscCallCGNS(cg_sol_info(cgv->file_num, cgv->base, cgv->zone, sol_id, buffer, &gridloc));
494472b9844SJames Wright   if (cgv->solution_name) PetscCall(PetscFree(cgv->solution_name));
495472b9844SJames Wright   PetscCall(PetscStrallocpy(buffer, &cgv->solution_name));
496472b9844SJames Wright   *name = cgv->solution_name;
497472b9844SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
498472b9844SJames Wright }
499