impls/adios/adios.c

4c02969dSBarry Smith#include <petsc/private/viewerimpl.h>    /*I   "petscsys.h"   I*/
4c02969dSBarry Smith#include <adios.h>
4c02969dSBarry Smith
*a56f64adSBarry Smithint64_t Petsc_adios_group;
*a56f64adSBarry Smith
4c02969dSBarry Smithtypedef struct GroupList {
4c02969dSBarry Smith  const char       *name;
4c02969dSBarry Smith  struct GroupList *next;
4c02969dSBarry Smith} GroupList;
4c02969dSBarry Smith
4c02969dSBarry Smithtypedef struct {
4c02969dSBarry Smith  char          *filename;
4c02969dSBarry Smith  PetscFileMode btype;
4c02969dSBarry Smith  PetscInt      timestep;
4c02969dSBarry Smith  GroupList     *groups;
*a56f64adSBarry Smith  int64_t       adios_handle;
4c02969dSBarry Smith} PetscViewer_ADIOS;
4c02969dSBarry Smith
4c02969dSBarry Smithstatic PetscErrorCode PetscViewerSetFromOptions_ADIOS(PetscOptionItems *PetscOptionsObject,PetscViewer v)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscErrorCode    ierr;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  ierr = PetscOptionsHead(PetscOptionsObject,"ADIOS PetscViewer Options");CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscOptionsTail();CHKERRQ(ierr);
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smithstatic PetscErrorCode PetscViewerFileClose_ADIOS(PetscViewer viewer)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*)viewer->data;
4c02969dSBarry Smith  PetscErrorCode   ierr;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  ierr = PetscFree(adios->filename);CHKERRQ(ierr);
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry SmithPetscErrorCode PetscViewerDestroy_ADIOS(PetscViewer viewer)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;
4c02969dSBarry Smith  PetscErrorCode   ierr;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  ierr = PetscViewerFileClose_ADIOS(viewer);CHKERRQ(ierr);
4c02969dSBarry Smith  while (adios->groups) {
4c02969dSBarry Smith    GroupList *tmp = adios->groups->next;
4c02969dSBarry Smith
4c02969dSBarry Smith    ierr         = PetscFree(adios->groups->name);CHKERRQ(ierr);
4c02969dSBarry Smith    ierr         = PetscFree(adios->groups);CHKERRQ(ierr);
4c02969dSBarry Smith    adios->groups = tmp;
4c02969dSBarry Smith  }
4c02969dSBarry Smith  ierr = PetscFree(adios);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",NULL);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",NULL);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",NULL);CHKERRQ(ierr);
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry SmithPetscErrorCode  PetscViewerFileSetMode_ADIOS(PetscViewer viewer, PetscFileMode type)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
4c02969dSBarry Smith  adios->btype = type;
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry SmithPetscErrorCode  PetscViewerFileSetName_ADIOS(PetscViewer viewer, const char name[])
4c02969dSBarry Smith{
*a56f64adSBarry Smith  PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;
*a56f64adSBarry Smith  PetscErrorCode    ierr;
*a56f64adSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
*a56f64adSBarry Smith  if (adios->filename) {ierr = PetscFree(adios->filename);CHKERRQ(ierr);}
*a56f64adSBarry Smith  ierr = PetscStrallocpy(name, &adios->filename);CHKERRQ(ierr);
*a56f64adSBarry Smith  /* Create or open the file collectively */
*a56f64adSBarry Smith  switch (adios->btype) {
*a56f64adSBarry Smith    adios_open(&adios->adios_handle,"PETSc",adios->filename,"r",PetscObjectComm((PetscObject)viewer));
*a56f64adSBarry Smith  case FILE_MODE_READ:
*a56f64adSBarry Smith    break;
*a56f64adSBarry Smith  case FILE_MODE_APPEND:
*a56f64adSBarry Smith    break;
*a56f64adSBarry Smith  case FILE_MODE_WRITE:
*a56f64adSBarry Smith    adios_open(&adios->adios_handle,"PETSc",adios->filename,"w",PetscObjectComm((PetscObject)viewer));
*a56f64adSBarry Smith    break;
*a56f64adSBarry Smith  default:
*a56f64adSBarry Smith    SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
*a56f64adSBarry Smith  }
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smithstatic PetscErrorCode PetscViewerFileGetName_ADIOS(PetscViewer viewer,const char **name)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscViewer_ADIOS *vadios = (PetscViewer_ADIOS*)viewer->data;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  *name = vadios->filename;
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smith/*MC
4c02969dSBarry Smith   PETSCVIEWERADIOS - A viewer that writes to an ADIOS file
4c02969dSBarry Smith
4c02969dSBarry Smith
4c02969dSBarry Smith.seealso:  PetscViewerADIOSOpen(), PetscViewerStringSPrintf(), PetscViewerSocketOpen(), PetscViewerDrawOpen(), PETSCVIEWERSOCKET,
4c02969dSBarry Smith           PetscViewerCreate(), PetscViewerASCIIOpen(), PetscViewerBinaryOpen(), PETSCVIEWERBINARY, PETSCVIEWERDRAW, PETSCVIEWERSTRING,
4c02969dSBarry Smith           PetscViewerMatlabOpen(), VecView(), DMView(), PetscViewerMatlabPutArray(), PETSCVIEWERASCII, PETSCVIEWERMATLAB,
4c02969dSBarry Smith           PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType()
4c02969dSBarry Smith
4c02969dSBarry Smith  Level: beginner
4c02969dSBarry SmithM*/
4c02969dSBarry Smith
4c02969dSBarry SmithPETSC_EXTERN PetscErrorCode PetscViewerCreate_ADIOS(PetscViewer v)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscViewer_ADIOS *adios;
4c02969dSBarry Smith  PetscErrorCode   ierr;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  ierr = PetscNewLog(v,&adios);CHKERRQ(ierr);
4c02969dSBarry Smith
4c02969dSBarry Smith  v->data                = (void*) adios;
4c02969dSBarry Smith  v->ops->destroy        = PetscViewerDestroy_ADIOS;
4c02969dSBarry Smith  v->ops->setfromoptions = PetscViewerSetFromOptions_ADIOS;
4c02969dSBarry Smith  v->ops->flush          = 0;
4c02969dSBarry Smith  adios->btype            = (PetscFileMode) -1;
4c02969dSBarry Smith  adios->filename         = 0;
4c02969dSBarry Smith  adios->timestep         = -1;
4c02969dSBarry Smith  adios->groups           = NULL;
4c02969dSBarry Smith
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_ADIOS);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_ADIOS);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ADIOS);CHKERRQ(ierr);
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smith/*@C
4c02969dSBarry Smith   PetscViewerADIOSOpen - Opens a file for ADIOS input/output.
4c02969dSBarry Smith
4c02969dSBarry Smith   Collective on MPI_Comm
4c02969dSBarry Smith
4c02969dSBarry Smith   Input Parameters:
4c02969dSBarry Smith+  comm - MPI communicator
4c02969dSBarry Smith.  name - name of file
4c02969dSBarry Smith-  type - type of file
4c02969dSBarry Smith$    FILE_MODE_WRITE - create new file for binary output
4c02969dSBarry Smith$    FILE_MODE_READ - open existing file for binary input
4c02969dSBarry Smith$    FILE_MODE_APPEND - open existing file for binary output
4c02969dSBarry Smith
4c02969dSBarry Smith   Output Parameter:
4c02969dSBarry Smith.  adiosv - PetscViewer for ADIOS input/output to use with the specified file
4c02969dSBarry Smith
4c02969dSBarry Smith   Level: beginner
4c02969dSBarry Smith
4c02969dSBarry Smith   Note:
4c02969dSBarry Smith   This PetscViewer should be destroyed with PetscViewerDestroy().
4c02969dSBarry Smith
4c02969dSBarry Smith   Concepts: ADIOS files
4c02969dSBarry Smith   Concepts: PetscViewerADIOS^creating
4c02969dSBarry Smith
*a56f64adSBarry Smith.seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(), PetscViewerHDF5Open(),
*a56f64adSBarry Smith          VecView(), MatView(), VecLoad(), PetscViewerSetType(), PetscViewerFileSetMode(), PetscViewerFileSetName()
4c02969dSBarry Smith          MatLoad(), PetscFileMode, PetscViewer
4c02969dSBarry Smith@*/
4c02969dSBarry SmithPetscErrorCode  PetscViewerADIOSOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *adiosv)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscErrorCode ierr;
4c02969dSBarry Smith
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  ierr = PetscViewerCreate(comm, adiosv);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscViewerSetType(*adiosv, PETSCVIEWERADIOS);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscViewerFileSetMode(*adiosv, type);CHKERRQ(ierr);
4c02969dSBarry Smith  ierr = PetscViewerFileSetName(*adiosv, name);CHKERRQ(ierr);
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smith/*@C
4c02969dSBarry Smith  PetscDataTypeToADIOSDataType - Converts the PETSc name of a datatype to its ADIOS name.
4c02969dSBarry Smith
4c02969dSBarry Smith  Not collective
4c02969dSBarry Smith
4c02969dSBarry Smith  Input Parameter:
4c02969dSBarry Smith. ptype - the PETSc datatype name (for example PETSC_DOUBLE)
4c02969dSBarry Smith
4c02969dSBarry Smith  Output Parameter:
4c02969dSBarry Smith. mtype - the MPI datatype (for example MPI_DOUBLE, ...)
4c02969dSBarry Smith
4c02969dSBarry Smith  Level: advanced
4c02969dSBarry Smith
4c02969dSBarry Smith  Developer Notes: These have not been verified
4c02969dSBarry Smith
4c02969dSBarry Smith.seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType()
4c02969dSBarry Smith@*/
4c02969dSBarry SmithPetscErrorCode PetscDataTypeToADIOSDataType(PetscDataType ptype, enum ADIOS_DATATYPES *htype)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith  if (ptype == PETSC_INT)
4c02969dSBarry Smith#if defined(PETSC_USE_64BIT_INDICES)
4c02969dSBarry Smith                                       *htype = adios_long;
4c02969dSBarry Smith#else
4c02969dSBarry Smith                                       *htype = adios_integer;
4c02969dSBarry Smith#endif
4c02969dSBarry Smith  else if (ptype == PETSC_ENUM)        *htype = adios_integer;
4c02969dSBarry Smith  else if (ptype == PETSC_DOUBLE)      *htype = adios_double;
4c02969dSBarry Smith  else if (ptype == PETSC_LONG)        *htype = adios_long;
4c02969dSBarry Smith  else if (ptype == PETSC_SHORT)       *htype = adios_short;
4c02969dSBarry Smith  else if (ptype == PETSC_FLOAT)       *htype = adios_real;
4c02969dSBarry Smith  else if (ptype == PETSC_CHAR)        *htype = adios_string_array;
4c02969dSBarry Smith  else if (ptype == PETSC_STRING)      *htype = adios_string;
4c02969dSBarry Smith  else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported PETSc datatype");
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
4c02969dSBarry Smith
4c02969dSBarry Smith/*@C
4c02969dSBarry Smith  PetscADIOSDataTypeToPetscDataType - Finds the PETSc name of a datatype from its ADIOS name
4c02969dSBarry Smith
4c02969dSBarry Smith  Not collective
4c02969dSBarry Smith
4c02969dSBarry Smith  Input Parameter:
4c02969dSBarry Smith. htype - the ADIOS datatype (for example H5T_NATIVE_DOUBLE, ...)
4c02969dSBarry Smith
4c02969dSBarry Smith  Output Parameter:
4c02969dSBarry Smith. ptype - the PETSc datatype name (for example PETSC_DOUBLE)
4c02969dSBarry Smith
4c02969dSBarry Smith  Level: advanced
4c02969dSBarry Smith
4c02969dSBarry Smith  Developer Notes: These have not been verified
4c02969dSBarry Smith
4c02969dSBarry Smith.seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType()
4c02969dSBarry Smith@*/
4c02969dSBarry SmithPetscErrorCode PetscADIOSDataTypeToPetscDataType(enum ADIOS_DATATYPES htype, PetscDataType *ptype)
4c02969dSBarry Smith{
4c02969dSBarry Smith  PetscFunctionBegin;
4c02969dSBarry Smith#if defined(PETSC_USE_64BIT_INDICES)
4c02969dSBarry Smith  if      (htype == adios_integer)     *ptype = PETSC_ENUM;
4c02969dSBarry Smith  else if (htype == adios_long)        *ptype = PETSC_INT;
4c02969dSBarry Smith#else
4c02969dSBarry Smith  if      (htype == adios_integer)     *ptype = PETSC_INT;
4c02969dSBarry Smith#endif
4c02969dSBarry Smith  else if (htype == adios_double)      *ptype = PETSC_DOUBLE;
4c02969dSBarry Smith  else if (htype == adios_long)        *ptype = PETSC_LONG;
4c02969dSBarry Smith  else if (htype == adios_short)       *ptype = PETSC_SHORT;
4c02969dSBarry Smith  else if (htype == adios_real)        *ptype = PETSC_FLOAT;
4c02969dSBarry Smith  else if (htype == adios_string_array) *ptype = PETSC_CHAR;
4c02969dSBarry Smith  else if (htype == adios_string)       *ptype = PETSC_STRING;
4c02969dSBarry Smith  else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported ADIOS datatype");
4c02969dSBarry Smith  PetscFunctionReturn(0);
4c02969dSBarry Smith}
*a56f64adSBarry Smith