Actual source code: ptype.c
petsc-3.14.6 2021-03-30
2: /*
3: Provides utility routines for manipulating any type of PETSc object.
4: */
5: #include <petscsys.h>
7: const char *const PetscDataTypes[] = {"UNKNOWN",
8: "DOUBLE","COMPLEX","LONG","SHORT","FLOAT",
9: "CHAR","LOGICAL","ENUM","BOOL","LONGDOUBLE",
10: "OBJECT","FUNCTION","STRING","FP16","STRUCT",
11: "INT","INT64",
12: "PetscDataType","PETSC_",NULL};
14: /*@C
15: PetscDataTypeToMPIDataType - Converts the PETSc name of a datatype to its MPI name.
17: Not collective
19: Input Parameter:
20: . ptype - the PETSc datatype name (for example PETSC_DOUBLE)
22: Output Parameter:
23: . mtype - the MPI datatype (for example MPI_DOUBLE, ...)
25: Level: advanced
27: .seealso: PetscDataType, PetscMPIDataTypeToPetscDataType()
28: @*/
29: PetscErrorCode PetscDataTypeToMPIDataType(PetscDataType ptype,MPI_Datatype *mtype)
30: {
32: if (ptype == PETSC_INT) *mtype = MPIU_INT;
33: else if (ptype == PETSC_DOUBLE) *mtype = MPI_DOUBLE;
34: #if defined(PETSC_HAVE_COMPLEX)
35: #if defined(PETSC_USE_REAL_SINGLE)
36: else if (ptype == PETSC_COMPLEX) *mtype = MPIU_C_COMPLEX;
37: #elif defined(PETSC_USE_REAL___FLOAT128)
38: else if (ptype == PETSC_COMPLEX) *mtype = MPIU___COMPLEX128;
39: #else
40: else if (ptype == PETSC_COMPLEX) *mtype = MPIU_C_DOUBLE_COMPLEX;
41: #endif
42: #endif
43: else if (ptype == PETSC_LONG) *mtype = MPI_LONG;
44: else if (ptype == PETSC_SHORT) *mtype = MPI_SHORT;
45: else if (ptype == PETSC_ENUM) *mtype = MPI_INT;
46: else if (ptype == PETSC_BOOL) *mtype = MPI_INT;
47: else if (ptype == PETSC_INT64) *mtype = MPIU_INT64;
48: else if (ptype == PETSC_FLOAT) *mtype = MPI_FLOAT;
49: else if (ptype == PETSC_CHAR) *mtype = MPI_CHAR;
50: else if (ptype == PETSC_BIT_LOGICAL) *mtype = MPI_BYTE;
51: #if defined(PETSC_USE_REAL___FLOAT128)
52: else if (ptype == PETSC___FLOAT128) *mtype = MPIU___FLOAT128;
53: #elif defined(PETSC_USE_REAL___FP16)
54: else if (ptype == PETSC___FP16) *mtype = MPIU___FP16;
55: #endif
56: else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown PETSc datatype");
57: return(0);
58: }
60: /*@C
61: PetscMPIDataTypeToPetscDataType Finds the PETSc name of a datatype from its MPI name
63: Not collective
65: Input Parameter:
66: . mtype - the MPI datatype (for example MPI_DOUBLE, ...)
68: Output Parameter:
69: . ptype - the PETSc datatype name (for example PETSC_DOUBLE)
71: Level: advanced
73: .seealso: PetscDataType, PetscMPIDataTypeToPetscDataType()
74: @*/
75: PetscErrorCode PetscMPIDataTypeToPetscDataType(MPI_Datatype mtype,PetscDataType *ptype)
76: {
78: if (mtype == MPIU_INT) *ptype = PETSC_INT;
79: #if defined(PETSC_USE_64BIT_INDICES)
80: else if (mtype == MPI_INT) *ptype = PETSC_ENUM;
81: #endif
82: else if (mtype == MPIU_INT64) *ptype = PETSC_INT64;
83: else if (mtype == MPI_DOUBLE) *ptype = PETSC_DOUBLE;
84: #if defined(PETSC_HAVE_COMPLEX)
85: #if defined(PETSC_USE_REAL_SINGLE)
86: else if (mtype == MPIU_C_COMPLEX) *ptype = PETSC_COMPLEX;
87: #elif defined(PETSC_USE_REAL___FLOAT128)
88: else if (mtype == MPIU___COMPLEX128) *ptype = PETSC_COMPLEX;
89: #else
90: else if (mtype == MPIU_C_DOUBLE_COMPLEX) *ptype = PETSC_COMPLEX;
91: #endif
92: #endif
93: else if (mtype == MPI_LONG) *ptype = PETSC_LONG;
94: else if (mtype == MPI_SHORT) *ptype = PETSC_SHORT;
95: else if (mtype == MPI_FLOAT) *ptype = PETSC_FLOAT;
96: else if (mtype == MPI_CHAR) *ptype = PETSC_CHAR;
97: #if defined(PETSC_USE_REAL___FLOAT128)
98: else if (mtype == MPIU___FLOAT128) *ptype = PETSC___FLOAT128;
99: #elif defined(PETSC_USE_REAL___FP16)
100: else if (mtype == MPIU___FP16) *ptype = PETSC___FP16;
101: #endif
102: else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled MPI datatype");
103: return(0);
104: }
106: typedef enum {PETSC_INT_SIZE = sizeof(PetscInt),
107: PETSC_DOUBLE_SIZE = sizeof(double),
108: #if defined(PETSC_HAVE_COMPLEX)
109: PETSC_COMPLEX_SIZE = sizeof(PetscComplex),
110: #else
111: PETSC_COMPLEX_SIZE = 2*sizeof(PetscReal),
112: #endif
113: PETSC_LONG_SIZE = sizeof(long),
114: PETSC_SHORT_SIZE = sizeof(short),
115: PETSC_FLOAT_SIZE = sizeof(float),
116: PETSC_CHAR_SIZE = sizeof(char),
117: PETSC_ENUM_SIZE = sizeof(PetscEnum),
118: PETSC_BOOL_SIZE = sizeof(PetscBool),
119: PETSC_INT64_SIZE = sizeof(PetscInt64),
120: PETSC_BIT_LOGICAL_SIZE = sizeof(char)
121: #if defined(PETSC_USE_REAL___FLOAT128)
122: ,PETSC___FLOAT128_SIZE = sizeof(__float128)
123: #elif defined(PETSC_USE_REAL___FP16)
124: ,PETSC___FP16_SIZE = sizeof(__fp16)
125: #endif
126: } PetscDataTypeSize;
128: /*@C
129: PetscDataTypeGetSize - Gets the size (in bytes) of a PETSc datatype
131: Not collective
133: Input Parameter:
134: . ptype - the PETSc datatype name (for example PETSC_DOUBLE)
136: Output Parameter:
137: . size - the size in bytes (for example the size of PETSC_DOUBLE is 8)
139: Level: advanced
141: .seealso: PetscDataType, PetscDataTypeToMPIDataType()
142: @*/
143: PetscErrorCode PetscDataTypeGetSize(PetscDataType ptype,size_t *size)
144: {
146: if ((int) ptype < 0) *size = -(int)ptype;
147: else if (ptype == PETSC_INT) *size = PETSC_INT_SIZE;
148: else if (ptype == PETSC_DOUBLE) *size = PETSC_DOUBLE_SIZE;
149: else if (ptype == PETSC_COMPLEX) *size = PETSC_COMPLEX_SIZE;
150: else if (ptype == PETSC_LONG) *size = PETSC_LONG_SIZE;
151: else if (ptype == PETSC_SHORT) *size = PETSC_SHORT_SIZE;
152: else if (ptype == PETSC_FLOAT) *size = PETSC_FLOAT_SIZE;
153: else if (ptype == PETSC_CHAR) *size = PETSC_CHAR_SIZE;
154: else if (ptype == PETSC_ENUM) *size = PETSC_ENUM_SIZE;
155: else if (ptype == PETSC_BOOL) *size = PETSC_BOOL_SIZE;
156: else if (ptype == PETSC_INT64) *size = PETSC_INT64_SIZE;
157: else if (ptype == PETSC_BIT_LOGICAL) *size = PETSC_BIT_LOGICAL_SIZE;
158: #if defined(PETSC_USE_REAL___FLOAT128)
159: else if (ptype == PETSC___FLOAT128) *size = PETSC___FLOAT128_SIZE;
160: #elif defined(PETSC_USE_REAL___FP16)
161: else if (ptype == PETSC___FP16) *size = PETSC___FP16_SIZE;
162: #endif
163: else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown PETSc datatype");
164: return(0);
165: }
167: /*@C
168: PetscDataTypeFromString - Gets the enum value of a PETSc datatype represented as a string
170: Not collective
172: Input Parameter:
173: . name - the PETSc datatype name (for example, DOUBLE or real or Scalar)
175: Output Parameter:
176: + ptype - the enum value
177: - found - the string matches one of the data types
179: Level: advanced
181: .seealso: PetscDataType, PetscDataTypeToMPIDataType(), PetscDataTypeGetSize()
182: @*/
183: PetscErrorCode PetscDataTypeFromString(const char*name, PetscDataType *ptype,PetscBool *found)
184: {
188: PetscEnumFind(PetscDataTypes,name,(PetscEnum*)ptype,found);
189: if (!*found) {
190: char formatted[16];
192: PetscStrncpy(formatted,name,16);
193: PetscStrtolower(formatted);
194: PetscStrcmp(formatted,"scalar",found);
195: if (*found) {
196: *ptype = PETSC_SCALAR;
197: } else {
198: PetscStrcmp(formatted,"real",found);
199: if (*found) {
200: *ptype = PETSC_REAL;
201: }
202: }
203: }
204: return(0);
205: }