1 #if !defined(__PETSCMOAB_H) 2 #define __PETSCMOAB_H 3 4 #include <petscvec.h> 5 #include <petscmat.h> 6 #include <petscdm.h> 7 8 #include <string> 9 #include <moab/Core.hpp> 10 #include <moab/ParallelComm.hpp> 11 12 /* The MBERR macro is used to save typing. It checks a MOAB error code 13 * (rval) and calls SETERRQ if not MB_SUCCESS. A message (msg) can 14 * also be passed in. */ 15 #define MBERR(msg,rval) do{if(rval != moab::MB_SUCCESS) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"MOAB ERROR (%i): %s",(PetscErrorCode)rval,msg);} while(0) 16 #define MBERRNM(rval) do{if(rval != moab::MB_SUCCESS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"MOAB ERROR (%i)",rval);} while(0) 17 #define MBERRV(mbif,rval) do{if(rval != moab::MB_SUCCESS) { std::string emsg; mbif->get_last_error(emsg); SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"MOAB ERROR (%i): %s",(PetscErrorCode)rval,emsg.c_str());} } while(0) 18 #define MBERRVM(mbif,msg,rval) do{if(rval != moab::MB_SUCCESS) { std::string emsg; mbif->get_last_error(emsg); SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_LIB,"MOAB ERROR (%i): %s :: %s",(PetscErrorCode)rval,msg,emsg.c_str());} } while(0) 19 20 #define MOAB_PARROPTS_READ_PART "READ_PART" 21 #define MOAB_PARROPTS_READ_DELETE "READ_DELETE" 22 #define MOAB_PARROPTS_BCAST_DELETE "BCAST_DELETE" 23 24 #define MOAB_PARWOPTS_WRITE_PART "WRITE_PART" 25 #define MOAB_PARWOPTS_WRITE_FORMAT "FORMAT" 26 27 typedef const char* MoabReadMode; 28 typedef const char* MoabWriteMode; 29 30 PETSC_EXTERN PetscErrorCode DMMoabCreate(MPI_Comm comm, DM *moab); 31 PETSC_EXTERN PetscErrorCode DMMoabCreateMoab(MPI_Comm comm, moab::Interface *mbiface, moab::ParallelComm *pcomm, moab::Tag *ltog_tag, moab::Range *range, DM *moab); 32 PETSC_EXTERN PetscErrorCode DMMoabOutput(DM,const char*,const char*); 33 34 PETSC_EXTERN PetscErrorCode DMMoabSetParallelComm(DM dm,moab::ParallelComm *pcomm); 35 PETSC_EXTERN PetscErrorCode DMMoabGetParallelComm(DM dm,moab::ParallelComm **pcomm); 36 PETSC_EXTERN PetscErrorCode DMMoabSetInterface(DM dm,moab::Interface *iface); 37 PETSC_EXTERN PetscErrorCode DMMoabGetInterface(DM dm,moab::Interface **iface); 38 PETSC_EXTERN PetscErrorCode DMMoabSetLocalVertices(DM dm,moab::Range *range); 39 PETSC_EXTERN PetscErrorCode DMMoabGetLocalVertices(DM dm,moab::Range *owned,moab::Range *ghost); 40 PETSC_EXTERN PetscErrorCode DMMoabSetLocalElements(DM dm,moab::Range *range); 41 PETSC_EXTERN PetscErrorCode DMMoabGetLocalElements(DM dm,moab::Range *range); 42 PETSC_EXTERN PetscErrorCode DMMoabSetLocalToGlobalTag(DM dm,moab::Tag ltogtag); 43 PETSC_EXTERN PetscErrorCode DMMoabGetLocalToGlobalTag(DM dm,moab::Tag *ltog_tag); 44 PETSC_EXTERN PetscErrorCode DMMoabSetBlockSize(DM dm,PetscInt bs); 45 PETSC_EXTERN PetscErrorCode DMMoabGetBlockSize(DM dm,PetscInt *bs); 46 PETSC_EXTERN PetscErrorCode DMMoabGetBoundaryEntities(DM dm,moab::Range *bdvtx,moab::Range* bdfaces); 47 48 PETSC_EXTERN PetscErrorCode DMMoabVecGetArrayRead(DM,Vec,void*); 49 PETSC_EXTERN PetscErrorCode DMMoabVecRestoreArrayRead(DM,Vec,void*); 50 PETSC_EXTERN PetscErrorCode DMMoabVecGetArray(DM,Vec,void*); 51 PETSC_EXTERN PetscErrorCode DMMoabVecRestoreArray(DM,Vec,void*); 52 53 PETSC_EXTERN PetscErrorCode DMMoabCreateVector(DM dm, moab::Tag tag,PetscInt tag_size,moab::Range *range,PetscBool serial, PetscBool destroy_tag,Vec *X); 54 PETSC_EXTERN PetscErrorCode DMMoabCreateMatrix(DM dm, MatType mtype,Mat *J); 55 PETSC_EXTERN PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag); 56 PETSC_EXTERN PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range); 57 PETSC_EXTERN PetscErrorCode DMMoabSetFieldVector(DM, PetscInt, Vec); 58 59 PETSC_EXTERN PetscErrorCode DMMoabSetFields(DM dm,PetscInt nfields,const char** fields); 60 PETSC_EXTERN PetscErrorCode DMMoabGetFieldDof(DM dm,moab::EntityHandle point, PetscInt field,PetscInt* dof); 61 PETSC_EXTERN PetscErrorCode DMMoabGetFieldDofs(DM dm,PetscInt npoints,const moab::EntityHandle* points, PetscInt field,PetscInt* dof); 62 63 /* discretization and assembly specific DMMoab interface functions */ 64 PETSC_EXTERN PetscErrorCode DMMoabGetElementConnectivity(DM dm,moab::EntityHandle ehandle,PetscInt* nconn,const moab::EntityHandle **conn); 65 PETSC_EXTERN PetscErrorCode DMMoabGetVertexCoordinates(DM dm,PetscInt nconn,const moab::EntityHandle *conn,PetscScalar *vpos); 66 PETSC_EXTERN PetscErrorCode DMMoabCheckBoundaryVertices(DM dm,PetscInt nconn,const moab::EntityHandle *cnt,PetscBool* isbdvtx,PetscBool* elem_on_boundary); 67 68 /* DM utility creation interface */ 69 PETSC_EXTERN PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm,PetscInt,PetscInt,PetscInt,DM*); 70 PETSC_EXTERN PetscErrorCode DMMoabLoadFromFile(MPI_Comm,PetscInt,const char*,const char*,DM*); 71 72 #endif 73