1 #if !defined(__PETSCDMSWARM_H) 2 #define __PETSCDMSWARM_H 3 4 #include <petscdm.h> 5 6 typedef enum { 7 DMSWARM_BASIC=0, 8 DMSWARM_PIC 9 } DMSwarmType; 10 11 typedef enum { 12 DMSWARM_MIGRATE_BASIC=0, 13 DMSWARM_MIGRATE_DMCELLNSCATTER, 14 DMSWARM_MIGRATE_DMCELLEXACT, 15 DMSWARM_MIGRATE_USER 16 } DMSwarmMigrateType; 17 18 typedef enum { 19 DMSWARM_COLLECT_BASIC=0, 20 DMSWARM_COLLECT_DMDABOUNDINGBOX, 21 DMSWARM_COLLECT_GENERAL, 22 DMSWARM_COLLECT_USER 23 } DMSwarmCollectType; 24 25 PETSC_EXTERN const char* DMSwarmTypeNames[]; 26 PETSC_EXTERN const char* DMSwarmMigrateTypeNames[]; 27 PETSC_EXTERN const char* DMSwarmCollectTypeNames[]; 28 29 PETSC_EXTERN const char DMSwarmField_pid[]; 30 PETSC_EXTERN const char DMSwarmField_rank[]; 31 PETSC_EXTERN const char DMSwarmPICField_coor[]; 32 33 PETSC_EXTERN PetscErrorCode DMSwarmCreateGlobalVectorFromField(DM,const char[],Vec*); 34 PETSC_EXTERN PetscErrorCode DMSwarmDestroyGlobalVectorFromField(DM,const char[],Vec*); 35 PETSC_EXTERN PetscErrorCode DMSwarmCreateLocalVectorFromField(DM,const char[],Vec*); 36 PETSC_EXTERN PetscErrorCode DMSwarmDestroyLocalVectorFromField(DM,const char[],Vec*); 37 38 PETSC_EXTERN PetscErrorCode DMSwarmInitializeFieldRegister(DM); 39 PETSC_EXTERN PetscErrorCode DMSwarmFinalizeFieldRegister(DM); 40 PETSC_EXTERN PetscErrorCode DMSwarmSetLocalSizes(DM,PetscInt,PetscInt); 41 PETSC_EXTERN PetscErrorCode DMSwarmRegisterPetscDatatypeField(DM,const char[],PetscInt,PetscDataType); 42 PETSC_EXTERN PetscErrorCode DMSwarmRegisterUserStructField(DM,const char[],size_t); 43 PETSC_EXTERN PetscErrorCode DMSwarmRegisterUserDatatypeField(DM,const char[],size_t,PetscInt); 44 PETSC_EXTERN PetscErrorCode DMSwarmGetField(DM,const char[],PetscInt*,PetscDataType*,void**); 45 PETSC_EXTERN PetscErrorCode DMSwarmRestoreField(DM,const char[],PetscInt*,PetscDataType*,void**); 46 47 PETSC_EXTERN PetscErrorCode DMSwarmVectorDefineField(DM,const char[]); 48 49 PETSC_EXTERN PetscErrorCode DMSwarmAddPoint(DM); 50 PETSC_EXTERN PetscErrorCode DMSwarmAddNPoints(DM,PetscInt); 51 PETSC_EXTERN PetscErrorCode DMSwarmRemovePoint(DM); 52 PETSC_EXTERN PetscErrorCode DMSwarmRemovePointAtIndex(DM,PetscInt); 53 54 PETSC_EXTERN PetscErrorCode DMSwarmGetLocalSize(DM,PetscInt*); 55 PETSC_EXTERN PetscErrorCode DMSwarmGetSize(DM,PetscInt*); 56 PETSC_EXTERN PetscErrorCode DMSwarmMigrate(DM,PetscBool); 57 58 PETSC_EXTERN PetscErrorCode DMSwarmCollectViewCreate(DM); 59 PETSC_EXTERN PetscErrorCode DMSwarmCollectViewDestroy(DM); 60 PETSC_EXTERN PetscErrorCode DMSwarmSetCellDM(DM,DM); 61 PETSC_EXTERN PetscErrorCode DMSwarmGetCellDM(DM,DM*); 62 63 PETSC_EXTERN PetscErrorCode DMSwarmSetType(DM,DMSwarmType); 64 65 #endif 66 67