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_APPEND: 28 break; 29 case FILE_MODE_WRITE: 30 ierr = adios_close(adios->adios_handle);CHKERRQ(ierr); 31 break; 32 default: 33 break; 34 } 35 ierr = PetscFree(adios->filename);CHKERRQ(ierr); 36 PetscFunctionReturn(0); 37 } 38 39 PetscErrorCode PetscViewerDestroy_ADIOS(PetscViewer viewer) 40 { 41 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 42 PetscErrorCode ierr; 43 44 PetscFunctionBegin; 45 ierr = PetscViewerFileClose_ADIOS(viewer);CHKERRQ(ierr); 46 ierr = PetscFree(adios);CHKERRQ(ierr); 47 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",NULL);CHKERRQ(ierr); 48 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",NULL);CHKERRQ(ierr); 49 ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",NULL);CHKERRQ(ierr); 50 PetscFunctionReturn(0); 51 } 52 53 PetscErrorCode PetscViewerFileSetMode_ADIOS(PetscViewer viewer, PetscFileMode type) 54 { 55 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 56 57 PetscFunctionBegin; 58 PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1); 59 adios->btype = type; 60 PetscFunctionReturn(0); 61 } 62 63 PetscErrorCode PetscViewerFileSetName_ADIOS(PetscViewer viewer, const char name[]) 64 { 65 PetscViewer_ADIOS *adios = (PetscViewer_ADIOS*) viewer->data; 66 PetscErrorCode ierr; 67 68 PetscFunctionBegin; 69 if (adios->filename) {ierr = PetscFree(adios->filename);CHKERRQ(ierr);} 70 ierr = PetscStrallocpy(name, &adios->filename);CHKERRQ(ierr); 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 PetscFunctionReturn(0); 85 } 86 87 static PetscErrorCode PetscViewerFileGetName_ADIOS(PetscViewer viewer,const char **name) 88 { 89 PetscViewer_ADIOS *vadios = (PetscViewer_ADIOS*)viewer->data; 90 91 PetscFunctionBegin; 92 *name = vadios->filename; 93 PetscFunctionReturn(0); 94 } 95 96 /*MC 97 PETSCVIEWERADIOS - A viewer that writes to an ADIOS file 98 99 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() 104 105 Level: beginner 106 M*/ 107 108 PETSC_EXTERN PetscErrorCode PetscViewerCreate_ADIOS(PetscViewer v) 109 { 110 PetscViewer_ADIOS *adios; 111 PetscErrorCode ierr; 112 113 PetscFunctionBegin; 114 ierr = PetscNewLog(v,&adios);CHKERRQ(ierr); 115 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; 123 124 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_ADIOS);CHKERRQ(ierr); 125 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_ADIOS);CHKERRQ(ierr); 126 ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ADIOS);CHKERRQ(ierr); 127 PetscFunctionReturn(0); 128 } 129 130 /*@C 131 PetscViewerADIOSOpen - Opens a file for ADIOS input/output. 132 133 Collective on MPI_Comm 134 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 142 143 Output Parameter: 144 . adiosv - PetscViewer for ADIOS input/output to use with the specified file 145 146 Level: beginner 147 148 Note: 149 This PetscViewer should be destroyed with PetscViewerDestroy(). 150 151 Concepts: ADIOS files 152 Concepts: PetscViewerADIOS^creating 153 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 { 160 PetscErrorCode ierr; 161 162 PetscFunctionBegin; 163 ierr = PetscViewerCreate(comm, adiosv);CHKERRQ(ierr); 164 ierr = PetscViewerSetType(*adiosv, PETSCVIEWERADIOS);CHKERRQ(ierr); 165 ierr = PetscViewerFileSetMode(*adiosv, type);CHKERRQ(ierr); 166 ierr = PetscViewerFileSetName(*adiosv, name);CHKERRQ(ierr); 167 PetscFunctionReturn(0); 168 } 169 170 /*@C 171 PetscDataTypeToADIOSDataType - Converts the PETSc name of a datatype to its ADIOS name. 172 173 Not collective 174 175 Input Parameter: 176 . ptype - the PETSc datatype name (for example PETSC_DOUBLE) 177 178 Output Parameter: 179 . mtype - the MPI datatype (for example MPI_DOUBLE, ...) 180 181 Level: advanced 182 183 Developer Notes: These have not been verified 184 185 .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType() 186 @*/ 187 PetscErrorCode PetscDataTypeToADIOSDataType(PetscDataType ptype, enum ADIOS_DATATYPES *htype) 188 { 189 PetscFunctionBegin; 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 PetscFunctionReturn(0); 205 } 206 207 /*@C 208 PetscADIOSDataTypeToPetscDataType - Finds the PETSc name of a datatype from its ADIOS name 209 210 Not collective 211 212 Input Parameter: 213 . htype - the ADIOS datatype (for example H5T_NATIVE_DOUBLE, ...) 214 215 Output Parameter: 216 . ptype - the PETSc datatype name (for example PETSC_DOUBLE) 217 218 Level: advanced 219 220 Developer Notes: These have not been verified 221 222 .seealso: PetscDataType, PetscADIOSDataTypeToPetscDataType() 223 @*/ 224 PetscErrorCode PetscADIOSDataTypeToPetscDataType(enum ADIOS_DATATYPES htype, PetscDataType *ptype) 225 { 226 PetscFunctionBegin; 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 PetscFunctionReturn(0); 241 } 242 243