Actual source code: adios.c

petsc-3.11.3 2019-06-26
Report Typos and Errors
  1:  #include <petsc/private/viewerimpl.h>
  2: #include <adios.h>
  3: #include <adios_read.h>

  5:  #include <petsc/private/vieweradiosimpl.h>

  7: static PetscErrorCode PetscViewerSetFromOptions_ADIOS(PetscOptionItems *PetscOptionsObject,PetscViewer v)
  8: {

 12:   PetscOptionsHead(PetscOptionsObject,"ADIOS PetscViewer Options");
 13:   PetscOptionsTail();
 14:   return(0);
 15: }

 17: static PetscErrorCode PetscViewerFileClose_ADIOS(PetscViewer viewer)
 18: {
 19:   PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*)viewer->data;
 20:   PetscErrorCode    ierr;

 23:   switch (adios->btype) {
 24:   case FILE_MODE_READ:
 25:     adios_read_close(adios->adios_fp);
 26:     break;
 27:   case FILE_MODE_APPEND:
 28:     break;
 29:   case FILE_MODE_WRITE:
 30:      adios_close(adios->adios_handle);
 31:     break;
 32:   default:
 33:     break;
 34:   }
 35:   PetscFree(adios->filename);
 36:   return(0);
 37: }

 39: PetscErrorCode PetscViewerDestroy_ADIOS(PetscViewer viewer)
 40: {
 41:   PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;
 42:   PetscErrorCode    ierr;

 45:   PetscViewerFileClose_ADIOS(viewer);
 46:   PetscFree(adios);
 47:   PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",NULL);
 48:   PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",NULL);
 49:   PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",NULL);
 50:   return(0);
 51: }

 53: PetscErrorCode  PetscViewerFileSetMode_ADIOS(PetscViewer viewer, PetscFileMode type)
 54: {
 55:   PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;

 59:   adios->btype = type;
 60:   return(0);
 61: }

 63: PetscErrorCode  PetscViewerFileSetName_ADIOS(PetscViewer viewer, const char name[])
 64: {
 65:   PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data;
 66:   PetscErrorCode    ierr;

 69:   if (adios->filename) {PetscFree(adios->filename);}
 70:   PetscStrallocpy(name, &adios->filename);
 71:   /* Create or open the file collectively */
 72:   switch (adios->btype) {
 73:   case FILE_MODE_READ:
 74:     adios->adios_fp = adios_read_open_file(adios->filename,ADIOS_READ_METHOD_BP,PetscObjectComm((PetscObject)viewer));
 75:     break;
 76:   case FILE_MODE_APPEND:
 77:     break;
 78:   case FILE_MODE_WRITE:
 79:     adios_open(&adios->adios_handle,"PETSc",adios->filename,"w",PetscObjectComm((PetscObject)viewer));
 80:     break;
 81:   default:
 82:     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER, "Must call PetscViewerFileSetMode() before PetscViewerFileSetName()");
 83:   }
 84:   return(0);
 85: }

 87: static PetscErrorCode PetscViewerFileGetName_ADIOS(PetscViewer viewer,const char **name)
 88: {
 89:   PetscViewer_ADIOS *vadios = (PetscViewer_ADIOS*)viewer->data;

 92:   *name = vadios->filename;
 93:   return(0);
 94: }

 96: /*MC
 97:    PETSCVIEWERADIOS - A viewer that writes to an ADIOS file


100: .seealso:  PetscViewerADIOSOpen(), PetscViewerStringSPrintf(), PetscViewerSocketOpen(), PetscViewerDrawOpen(), PETSCVIEWERSOCKET,
101:            PetscViewerCreate(), PetscViewerASCIIOpen(), PetscViewerBinaryOpen(), PETSCVIEWERBINARY, PETSCVIEWERDRAW, PETSCVIEWERSTRING,
102:            PetscViewerMatlabOpen(), VecView(), DMView(), PetscViewerMatlabPutArray(), PETSCVIEWERASCII, PETSCVIEWERMATLAB,
103:            PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType()

105:   Level: beginner
106: M*/

108: PETSC_EXTERN PetscErrorCode PetscViewerCreate_ADIOS(PetscViewer v)
109: {
110:   PetscViewer_ADIOS *adios;
111:   PetscErrorCode    ierr;

114:   PetscNewLog(v,&adios);

116:   v->data                = (void*) adios;
117:   v->ops->destroy        = PetscViewerDestroy_ADIOS;
118:   v->ops->setfromoptions = PetscViewerSetFromOptions_ADIOS;
119:   v->ops->flush          = 0;
120:   adios->btype            = (PetscFileMode) -1;
121:   adios->filename         = 0;
122:   adios->timestep         = -1;

124:   PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_ADIOS);
125:   PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_ADIOS);
126:   PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ADIOS);
127:   return(0);
128: }

130: /*@C
131:    PetscViewerADIOSOpen - Opens a file for ADIOS input/output.

133:    Collective on MPI_Comm

135:    Input Parameters:
136: +  comm - MPI communicator
137: .  name - name of file
138: -  type - type of file
139: $    FILE_MODE_WRITE - create new file for binary output
140: $    FILE_MODE_READ - open existing file for binary input
141: $    FILE_MODE_APPEND - open existing file for binary output

143:    Output Parameter:
144: .  adiosv - PetscViewer for ADIOS input/output to use with the specified file

146:    Level: beginner

148:    Note:
149:    This PetscViewer should be destroyed with PetscViewerDestroy().

151:    Concepts: ADIOS files
152:    Concepts: PetscViewerADIOS^creating

154: .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(), PetscViewerHDF5Open(),
155:           VecView(), MatView(), VecLoad(), PetscViewerSetType(), PetscViewerFileSetMode(), PetscViewerFileSetName()
156:           MatLoad(), PetscFileMode, PetscViewer
157: @*/
158: PetscErrorCode  PetscViewerADIOSOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *adiosv)
159: {

163:   PetscViewerCreate(comm, adiosv);
164:   PetscViewerSetType(*adiosv, PETSCVIEWERADIOS);
165:   PetscViewerFileSetMode(*adiosv, type);
166:   PetscViewerFileSetName(*adiosv, name);
167:   return(0);
168: }

170: /*@C
171:   PetscDataTypeToADIOSDataType - Converts the PETSc name of a datatype to its ADIOS name.

173:   Not collective

175:   Input Parameter:
176: . ptype - the PETSc datatype name (for example PETSC_DOUBLE)

178:   Output Parameter:
179: . mtype - the MPI datatype (for example MPI_DOUBLE, ...)

181:   Level: advanced

183:   Developer Notes: These have not been verified

185: .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType()
186: @*/
187: PetscErrorCode PetscDataTypeToADIOSDataType(PetscDataType ptype, enum ADIOS_DATATYPES *htype)
188: {
190:   if (ptype == PETSC_INT)
191: #if defined(PETSC_USE_64BIT_INDICES)
192:                                        *htype = adios_long;
193: #else
194:                                        *htype = adios_integer;
195: #endif
196:   else if (ptype == PETSC_ENUM)        *htype = adios_integer;
197:   else if (ptype == PETSC_DOUBLE)      *htype = adios_double;
198:   else if (ptype == PETSC_LONG)        *htype = adios_long;
199:   else if (ptype == PETSC_SHORT)       *htype = adios_short;
200:   else if (ptype == PETSC_FLOAT)       *htype = adios_real;
201:   else if (ptype == PETSC_CHAR)        *htype = adios_string_array;
202:   else if (ptype == PETSC_STRING)      *htype = adios_string;
203:   else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported PETSc datatype");
204:   return(0);
205: }

207: /*@C
208:   PetscADIOSDataTypeToPetscDataType - Finds the PETSc name of a datatype from its ADIOS name

210:   Not collective

212:   Input Parameter:
213: . htype - the ADIOS datatype (for example H5T_NATIVE_DOUBLE, ...)

215:   Output Parameter:
216: . ptype - the PETSc datatype name (for example PETSC_DOUBLE)

218:   Level: advanced

220:   Developer Notes: These have not been verified

222: .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType()
223: @*/
224: PetscErrorCode PetscADIOSDataTypeToPetscDataType(enum ADIOS_DATATYPES htype, PetscDataType *ptype)
225: {
227: #if defined(PETSC_USE_64BIT_INDICES)
228:   if      (htype == adios_integer)     *ptype = PETSC_ENUM;
229:   else if (htype == adios_long)        *ptype = PETSC_INT;
230: #else
231:   if      (htype == adios_integer)     *ptype = PETSC_INT;
232: #endif
233:   else if (htype == adios_double)      *ptype = PETSC_DOUBLE;
234:   else if (htype == adios_long)        *ptype = PETSC_LONG;
235:   else if (htype == adios_short)       *ptype = PETSC_SHORT;
236:   else if (htype == adios_real)        *ptype = PETSC_FLOAT;
237:   else if (htype == adios_string_array) *ptype = PETSC_CHAR;
238:   else if (htype == adios_string)       *ptype = PETSC_STRING;
239:   else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported ADIOS datatype");
240:   return(0);
241: }