1 #include <petsc/private/viewerimpl.h> /*I "petscsys.h" I*/ 2 #include <adios.h> 3 #include <adios_read.h> 4 5 #include <petsc/private/vieweradiosimpl.h> 6 7 static PetscErrorCode PetscViewerSetFromOptions_ADIOS(PetscOptionItems *PetscOptionsObject,PetscViewer v) 8 { 9 PetscErrorCode ierr; 10 11 PetscFunctionBegin; 12 ierr = PetscOptionsHead(PetscOptionsObject,"ADIOS PetscViewer Options");CHKERRQ(ierr); 13 ierr = PetscOptionsTail();CHKERRQ(ierr); 14 PetscFunctionReturn(0); 15 } 16 17 static PetscErrorCode PetscViewerFileClose_ADIOS(PetscViewer viewer) 18 { 19 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*)viewer->data; 20 PetscErrorCode ierr; 21 22 PetscFunctionBegin; 23 switch (adios->btype) { 24 case FILE_MODE_READ: 25 ierr = adios_read_close(adios->adios_fp);CHKERRQ(ierr); 26 break; 27 case FILE_MODE_WRITE: 28 ierr = adios_close(adios->adios_handle);CHKERRQ(ierr); 29 break; 30 default: 31 break; 32 } 33 ierr = PetscFree(adios->filename);CHKERRQ(ierr); 34 PetscFunctionReturn(0); 35 } 36 37 PetscErrorCode PetscViewerDestroy_ADIOS(PetscViewer viewer) 38 { 39 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 40 PetscErrorCode ierr; 41 42 PetscFunctionBegin; 43 ierr = PetscViewerFileClose_ADIOS(viewer);CHKERRQ(ierr); 44 ierr = PetscFree(adios);CHKERRQ(ierr); 45 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",NULL);CHKERRQ(ierr); 46 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",NULL);CHKERRQ(ierr); 47 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",NULL);CHKERRQ(ierr); 48 PetscFunctionReturn(0); 49 } 50 51 PetscErrorCode PetscViewerFileSetMode_ADIOS(PetscViewer viewer, PetscFileMode type) 52 { 53 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 54 55 PetscFunctionBegin; 56 adios->btype = type; 57 PetscFunctionReturn(0); 58 } 59 60 PetscErrorCode PetscViewerFileSetName_ADIOS(PetscViewer viewer, const char name[]) 61 { 62 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 63 PetscErrorCode ierr; 64 65 PetscFunctionBegin; 66 if (adios->filename) {ierr = PetscFree(adios->filename);CHKERRQ(ierr);} 67 ierr = PetscStrallocpy(name, &adios->filename);CHKERRQ(ierr); 68 /* Create or open the file collectively */ 69 switch (adios->btype) { 70 case FILE_MODE_READ: 71 adios->adios_fp = adios_read_open_file(adios->filename,ADIOS_READ_METHOD_BP,PetscObjectComm((PetscObject)viewer)); 72 break; 73 case FILE_MODE_WRITE: 74 adios_open(&adios->adios_handle,"PETSc",adios->filename,"w",PetscObjectComm((PetscObject)viewer)); 75 break; 76 case FILE_MODE_UNDEFINED: 77 SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_ORDER,"Must call PetscViewerFileSetMode() before PetscViewerFileSetName()"); 78 default: 79 SETERRQ1(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Unsupported file mode %s",PetscFileModes[adios->btype]); 80 } 81 PetscFunctionReturn(0); 82 } 83 84 static PetscErrorCode PetscViewerFileGetName_ADIOS(PetscViewer viewer,const char **name) 85 { 86 PetscViewer_ADIOS *vadios = (PetscViewer_ADIOS*)viewer->data; 87 88 PetscFunctionBegin; 89 *name = vadios->filename; 90 PetscFunctionReturn(0); 91 } 92 93 /*MC 94 PETSCVIEWERADIOS - A viewer that writes to an ADIOS file 95 96 97 .seealso: PetscViewerADIOSOpen(), PetscViewerStringSPrintf(), PetscViewerSocketOpen(), PetscViewerDrawOpen(), PETSCVIEWERSOCKET, 98 PetscViewerCreate(), PetscViewerASCIIOpen(), PetscViewerBinaryOpen(), PETSCVIEWERBINARY, PETSCVIEWERDRAW, PETSCVIEWERSTRING, 99 PetscViewerMatlabOpen(), VecView(), DMView(), PetscViewerMatlabPutArray(), PETSCVIEWERASCII, PETSCVIEWERMATLAB, 100 PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType() 101 102 Level: beginner 103 M*/ 104 105 PETSC_EXTERN PetscErrorCode PetscViewerCreate_ADIOS(PetscViewer v) 106 { 107 PetscViewer_ADIOS *adios; 108 PetscErrorCode ierr; 109 110 PetscFunctionBegin; 111 ierr = PetscNewLog(v,&adios);CHKERRQ(ierr); 112 113 v->data = (void*) adios; 114 v->ops->destroy = PetscViewerDestroy_ADIOS; 115 v->ops->setfromoptions = PetscViewerSetFromOptions_ADIOS; 116 v->ops->flush = NULL; 117 adios->btype = FILE_MODE_UNDEFINED; 118 adios->filename = NULL; 119 adios->timestep = -1; 120 121 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_ADIOS);CHKERRQ(ierr); 122 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_ADIOS);CHKERRQ(ierr); 123 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ADIOS);CHKERRQ(ierr); 124 PetscFunctionReturn(0); 125 } 126 127 /*@C 128 PetscViewerADIOSOpen - Opens a file for ADIOS input/output. 129 130 Collective 131 132 Input Parameters: 133 + comm - MPI communicator 134 . name - name of file 135 - type - type of file 136 $ FILE_MODE_WRITE - create new file for binary output 137 $ FILE_MODE_READ - open existing file for binary input 138 $ FILE_MODE_APPEND - open existing file for binary output 139 140 Output Parameter: 141 . adiosv - PetscViewer for ADIOS input/output to use with the specified file 142 143 Level: beginner 144 145 Note: 146 This PetscViewer should be destroyed with PetscViewerDestroy(). 147 148 149 .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(), PetscViewerHDF5Open(), 150 VecView(), MatView(), VecLoad(), PetscViewerSetType(), PetscViewerFileSetMode(), PetscViewerFileSetName() 151 MatLoad(), PetscFileMode, PetscViewer 152 @*/ 153 PetscErrorCode PetscViewerADIOSOpen(MPI_Comm comm, const char name[], PetscFileMode type, PetscViewer *adiosv) 154 { 155 PetscErrorCode ierr; 156 157 PetscFunctionBegin; 158 ierr = PetscViewerCreate(comm, adiosv);CHKERRQ(ierr); 159 ierr = PetscViewerSetType(*adiosv, PETSCVIEWERADIOS);CHKERRQ(ierr); 160 ierr = PetscViewerFileSetMode(*adiosv, type);CHKERRQ(ierr); 161 ierr = PetscViewerFileSetName(*adiosv, name);CHKERRQ(ierr); 162 PetscFunctionReturn(0); 163 } 164 165 /*@C 166 PetscDataTypeToADIOSDataType - Converts the PETSc name of a datatype to its ADIOS name. 167 168 Not collective 169 170 Input Parameter: 171 . ptype - the PETSc datatype name (for example PETSC_DOUBLE) 172 173 Output Parameter: 174 . mtype - the MPI datatype (for example MPI_DOUBLE, ...) 175 176 Level: advanced 177 178 Developer Notes: These have not been verified 179 180 .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType() 181 @*/ 182 PetscErrorCode PetscDataTypeToADIOSDataType(PetscDataType ptype, enum ADIOS_DATATYPES *htype) 183 { 184 PetscFunctionBegin; 185 if (ptype == PETSC_INT) 186 #if defined(PETSC_USE_64BIT_INDICES) 187 *htype = adios_long; 188 #else 189 *htype = adios_integer; 190 #endif 191 else if (ptype == PETSC_ENUM) *htype = adios_integer; 192 else if (ptype == PETSC_DOUBLE) *htype = adios_double; 193 else if (ptype == PETSC_LONG) *htype = adios_long; 194 else if (ptype == PETSC_SHORT) *htype = adios_short; 195 else if (ptype == PETSC_FLOAT) *htype = adios_real; 196 else if (ptype == PETSC_CHAR) *htype = adios_string_array; 197 else if (ptype == PETSC_STRING) *htype = adios_string; 198 else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported PETSc datatype"); 199 PetscFunctionReturn(0); 200 } 201 202 /*@C 203 PetscADIOSDataTypeToPetscDataType - Finds the PETSc name of a datatype from its ADIOS name 204 205 Not collective 206 207 Input Parameter: 208 . htype - the ADIOS datatype (for example H5T_NATIVE_DOUBLE, ...) 209 210 Output Parameter: 211 . ptype - the PETSc datatype name (for example PETSC_DOUBLE) 212 213 Level: advanced 214 215 Developer Notes: These have not been verified 216 217 .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType() 218 @*/ 219 PetscErrorCode PetscADIOSDataTypeToPetscDataType(enum ADIOS_DATATYPES htype, PetscDataType *ptype) 220 { 221 PetscFunctionBegin; 222 #if defined(PETSC_USE_64BIT_INDICES) 223 if (htype == adios_integer) *ptype = PETSC_ENUM; 224 else if (htype == adios_long) *ptype = PETSC_INT; 225 #else 226 if (htype == adios_integer) *ptype = PETSC_INT; 227 #endif 228 else if (htype == adios_double) *ptype = PETSC_DOUBLE; 229 else if (htype == adios_long) *ptype = PETSC_LONG; 230 else if (htype == adios_short) *ptype = PETSC_SHORT; 231 else if (htype == adios_real) *ptype = PETSC_FLOAT; 232 else if (htype == adios_string_array) *ptype = PETSC_CHAR; 233 else if (htype == adios_string) *ptype = PETSC_STRING; 234 else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unsupported ADIOS datatype"); 235 PetscFunctionReturn(0); 236 } 237 238