1 /* 2 Objects which encapsulate finite element spaces and operations 3 */ 4 #if !defined(__PETSCFE_H) 5 #define __PETSCFE_H 6 #include <petscdm.h> 7 #include <petscdt.h> 8 #include <petscfetypes.h> 9 #include <petscdstypes.h> 10 11 /* Assuming dim <= 3 */ 12 typedef struct { 13 PetscReal v0[3]; 14 PetscReal J[9]; 15 PetscReal invJ[9]; 16 PetscReal detJ; 17 PetscReal n[3]; 18 } PetscFECellGeom; 19 20 PETSC_EXTERN PetscErrorCode PetscFEInitializePackage(void); 21 22 PETSC_EXTERN PetscClassId PETSCSPACE_CLASSID; 23 24 /*J 25 PetscSpaceType - String with the name of a PETSc linear space 26 27 Level: beginner 28 29 .seealso: PetscSpaceSetType(), PetscSpace 30 J*/ 31 typedef const char *PetscSpaceType; 32 #define PETSCSPACEPOLYNOMIAL "poly" 33 #define PETSCSPACEDG "dg" 34 35 PETSC_EXTERN PetscFunctionList PetscSpaceList; 36 PETSC_EXTERN PetscBool PetscSpaceRegisterAllCalled; 37 PETSC_EXTERN PetscErrorCode PetscSpaceCreate(MPI_Comm, PetscSpace *); 38 PETSC_EXTERN PetscErrorCode PetscSpaceDestroy(PetscSpace *); 39 PETSC_EXTERN PetscErrorCode PetscSpaceSetType(PetscSpace, PetscSpaceType); 40 PETSC_EXTERN PetscErrorCode PetscSpaceGetType(PetscSpace, PetscSpaceType *); 41 PETSC_EXTERN PetscErrorCode PetscSpaceSetUp(PetscSpace); 42 PETSC_EXTERN PetscErrorCode PetscSpaceSetFromOptions(PetscSpace); 43 PETSC_EXTERN PetscErrorCode PetscSpaceViewFromOptions(PetscSpace,const char[],const char[]); 44 PETSC_EXTERN PetscErrorCode PetscSpaceView(PetscSpace,PetscViewer); 45 PETSC_EXTERN PetscErrorCode PetscSpaceRegister(const char [], PetscErrorCode (*)(PetscSpace)); 46 PETSC_EXTERN PetscErrorCode PetscSpaceRegisterAll(void); 47 PETSC_EXTERN PetscErrorCode PetscSpaceRegisterDestroy(void); 48 49 PETSC_EXTERN PetscErrorCode PetscSpaceGetDimension(PetscSpace, PetscInt *); 50 PETSC_EXTERN PetscErrorCode PetscSpaceSetOrder(PetscSpace, PetscInt); 51 PETSC_EXTERN PetscErrorCode PetscSpaceGetOrder(PetscSpace, PetscInt *); 52 PETSC_EXTERN PetscErrorCode PetscSpaceEvaluate(PetscSpace, PetscInt, const PetscReal[], PetscReal[], PetscReal[], PetscReal[]); 53 54 PETSC_EXTERN PetscErrorCode PetscSpacePolynomialSetNumVariables(PetscSpace, PetscInt); 55 PETSC_EXTERN PetscErrorCode PetscSpacePolynomialGetNumVariables(PetscSpace, PetscInt *); 56 PETSC_EXTERN PetscErrorCode PetscSpacePolynomialSetSymmetric(PetscSpace, PetscBool); 57 PETSC_EXTERN PetscErrorCode PetscSpacePolynomialGetSymmetric(PetscSpace, PetscBool *); 58 59 PETSC_EXTERN PetscErrorCode PetscSpaceDGSetQuadrature(PetscSpace, PetscQuadrature); 60 PETSC_EXTERN PetscErrorCode PetscSpaceDGGetQuadrature(PetscSpace, PetscQuadrature *); 61 62 PETSC_EXTERN PetscClassId PETSCDUALSPACE_CLASSID; 63 64 /*J 65 PetscDualSpaceType - String with the name of a PETSc dual space 66 67 Level: beginner 68 69 .seealso: PetscDualSpaceSetType(), PetscDualSpace 70 J*/ 71 typedef const char *PetscDualSpaceType; 72 #define PETSCDUALSPACELAGRANGE "lagrange" 73 #define PETSCDUALSPACESIMPLE "simple" 74 75 PETSC_EXTERN PetscFunctionList PetscDualSpaceList; 76 PETSC_EXTERN PetscBool PetscDualSpaceRegisterAllCalled; 77 PETSC_EXTERN PetscErrorCode PetscDualSpaceCreate(MPI_Comm, PetscDualSpace *); 78 PETSC_EXTERN PetscErrorCode PetscDualSpaceDestroy(PetscDualSpace *); 79 PETSC_EXTERN PetscErrorCode PetscDualSpaceDuplicate(PetscDualSpace, PetscDualSpace *); 80 PETSC_EXTERN PetscErrorCode PetscDualSpaceSetType(PetscDualSpace, PetscDualSpaceType); 81 PETSC_EXTERN PetscErrorCode PetscDualSpaceGetType(PetscDualSpace, PetscDualSpaceType *); 82 PETSC_EXTERN PetscErrorCode PetscDualSpaceSetUp(PetscDualSpace); 83 PETSC_EXTERN PetscErrorCode PetscDualSpaceSetFromOptions(PetscDualSpace); 84 PETSC_EXTERN PetscErrorCode PetscDualSpaceViewFromOptions(PetscDualSpace,const char[],const char[]); 85 PETSC_EXTERN PetscErrorCode PetscDualSpaceView(PetscDualSpace,PetscViewer); 86 PETSC_EXTERN PetscErrorCode PetscDualSpaceRegister(const char [], PetscErrorCode (*)(PetscDualSpace)); 87 PETSC_EXTERN PetscErrorCode PetscDualSpaceRegisterAll(void); 88 PETSC_EXTERN PetscErrorCode PetscDualSpaceRegisterDestroy(void); 89 90 PETSC_EXTERN PetscErrorCode PetscDualSpaceGetDimension(PetscDualSpace, PetscInt *); 91 PETSC_EXTERN PetscErrorCode PetscDualSpaceSetOrder(PetscDualSpace, PetscInt); 92 PETSC_EXTERN PetscErrorCode PetscDualSpaceGetOrder(PetscDualSpace, PetscInt *); 93 PETSC_EXTERN PetscErrorCode PetscDualSpaceSetDM(PetscDualSpace, DM); 94 PETSC_EXTERN PetscErrorCode PetscDualSpaceGetDM(PetscDualSpace, DM *); 95 PETSC_EXTERN PetscErrorCode PetscDualSpaceGetFunctional(PetscDualSpace, PetscInt, PetscQuadrature *); 96 PETSC_EXTERN PetscErrorCode PetscDualSpaceCreateReferenceCell(PetscDualSpace, PetscInt, PetscBool, DM *); 97 98 PETSC_EXTERN PetscErrorCode PetscDualSpaceApply(PetscDualSpace, PetscInt, PetscFECellGeom *, PetscInt, void (*)(const PetscReal [], PetscScalar *, void *), void *, PetscScalar *); 99 100 PETSC_EXTERN PetscErrorCode PetscDualSpaceLagrangeGetContinuity(PetscDualSpace, PetscBool *); 101 PETSC_EXTERN PetscErrorCode PetscDualSpaceLagrangeSetContinuity(PetscDualSpace, PetscBool); 102 103 PETSC_EXTERN PetscErrorCode PetscDualSpaceSimpleSetDimension(PetscDualSpace, PetscInt); 104 PETSC_EXTERN PetscErrorCode PetscDualSpaceSimpleSetFunctional(PetscDualSpace, PetscInt, PetscQuadrature); 105 106 PETSC_EXTERN PetscClassId PETSCFE_CLASSID; 107 108 /*J 109 PetscFEType - String with the name of a PETSc finite element space 110 111 Level: beginner 112 113 Note: Currently, the classes are concerned with the implementation of element integration 114 115 .seealso: PetscFESetType(), PetscFE 116 J*/ 117 typedef const char *PetscFEType; 118 #define PETSCFEBASIC "basic" 119 #define PETSCFENONAFFINE "nonaffine" 120 #define PETSCFEOPENCL "opencl" 121 #define PETSCFECOMPOSITE "composite" 122 123 PETSC_EXTERN PetscFunctionList PetscFEList; 124 PETSC_EXTERN PetscBool PetscFERegisterAllCalled; 125 PETSC_EXTERN PetscErrorCode PetscFECreate(MPI_Comm, PetscFE *); 126 PETSC_EXTERN PetscErrorCode PetscFEDestroy(PetscFE *); 127 PETSC_EXTERN PetscErrorCode PetscFESetType(PetscFE, PetscFEType); 128 PETSC_EXTERN PetscErrorCode PetscFEGetType(PetscFE, PetscFEType *); 129 PETSC_EXTERN PetscErrorCode PetscFESetUp(PetscFE); 130 PETSC_EXTERN PetscErrorCode PetscFESetFromOptions(PetscFE); 131 PETSC_EXTERN PetscErrorCode PetscFEViewFromOptions(PetscFE,const char[],const char[]); 132 PETSC_EXTERN PetscErrorCode PetscFEView(PetscFE,PetscViewer); 133 PETSC_EXTERN PetscErrorCode PetscFERegister(const char [], PetscErrorCode (*)(PetscFE)); 134 PETSC_EXTERN PetscErrorCode PetscFERegisterAll(void); 135 PETSC_EXTERN PetscErrorCode PetscFERegisterDestroy(void); 136 PETSC_EXTERN PetscErrorCode PetscFECreateDefault(DM, PetscInt, PetscInt, PetscBool, const char [], PetscInt, PetscFE *); 137 138 PETSC_EXTERN PetscErrorCode PetscFEGetDimension(PetscFE, PetscInt *); 139 PETSC_EXTERN PetscErrorCode PetscFEGetSpatialDimension(PetscFE, PetscInt *); 140 PETSC_EXTERN PetscErrorCode PetscFESetNumComponents(PetscFE, PetscInt); 141 PETSC_EXTERN PetscErrorCode PetscFEGetNumComponents(PetscFE, PetscInt *); 142 PETSC_EXTERN PetscErrorCode PetscFEGetTileSizes(PetscFE, PetscInt *, PetscInt *, PetscInt *, PetscInt *); 143 PETSC_EXTERN PetscErrorCode PetscFESetTileSizes(PetscFE, PetscInt, PetscInt, PetscInt, PetscInt); 144 PETSC_EXTERN PetscErrorCode PetscFESetBasisSpace(PetscFE, PetscSpace); 145 PETSC_EXTERN PetscErrorCode PetscFEGetBasisSpace(PetscFE, PetscSpace *); 146 PETSC_EXTERN PetscErrorCode PetscFESetDualSpace(PetscFE, PetscDualSpace); 147 PETSC_EXTERN PetscErrorCode PetscFEGetDualSpace(PetscFE, PetscDualSpace *); 148 PETSC_EXTERN PetscErrorCode PetscFESetQuadrature(PetscFE, PetscQuadrature); 149 PETSC_EXTERN PetscErrorCode PetscFEGetQuadrature(PetscFE, PetscQuadrature *); 150 PETSC_EXTERN PetscErrorCode PetscFEGetNumDof(PetscFE, const PetscInt **); 151 PETSC_EXTERN PetscErrorCode PetscFEGetDefaultTabulation(PetscFE, PetscReal **, PetscReal **, PetscReal **); 152 PETSC_EXTERN PetscErrorCode PetscFEGetTabulation(PetscFE, PetscInt, const PetscReal[], PetscReal **, PetscReal **, PetscReal **); 153 PETSC_EXTERN PetscErrorCode PetscFERestoreTabulation(PetscFE, PetscInt, const PetscReal[], PetscReal **, PetscReal **, PetscReal **); 154 PETSC_EXTERN PetscErrorCode PetscFERefine(PetscFE, PetscFE *); 155 156 PETSC_EXTERN PetscErrorCode PetscFEIntegrate(PetscFE, PetscDS, PetscInt, PetscInt, PetscFECellGeom *, const PetscScalar[], PetscDS, const PetscScalar[], PetscReal[]); 157 PETSC_EXTERN PetscErrorCode PetscFEIntegrateResidual(PetscFE, PetscDS, PetscInt, PetscInt, PetscFECellGeom *, const PetscScalar[], const PetscScalar[], PetscDS, const PetscScalar[], PetscScalar[]); 158 PETSC_EXTERN PetscErrorCode PetscFEIntegrateBdResidual(PetscFE, PetscDS, PetscInt, PetscInt, PetscFECellGeom *, const PetscScalar[], const PetscScalar[], PetscDS, const PetscScalar[], PetscScalar[]); 159 PETSC_EXTERN PetscErrorCode PetscFEIntegrateJacobian(PetscFE, PetscDS, PetscInt, PetscInt, PetscInt, PetscFECellGeom *, const PetscScalar[], const PetscScalar[], PetscDS, const PetscScalar[], PetscScalar[]); 160 PETSC_EXTERN PetscErrorCode PetscFEIntegrateBdJacobian(PetscFE, PetscDS, PetscInt, PetscInt, PetscInt, PetscFECellGeom *, const PetscScalar[], const PetscScalar[], PetscDS, const PetscScalar[], PetscScalar[]); 161 162 PETSC_EXTERN PetscErrorCode PetscFEOpenCLSetRealType(PetscFE, PetscDataType); 163 PETSC_EXTERN PetscErrorCode PetscFEOpenCLGetRealType(PetscFE, PetscDataType *); 164 165 #endif 166