157795646SDave May 257795646SDave May #define PETSCDM_DLL 357795646SDave May #include <petsc/private/dmswarmimpl.h> /*I "petscdmswarm.h" I*/ 4b62e03f8SDave May #include "data_bucket.h" 557795646SDave May 657795646SDave May //typedef PetscErrorCode (*swarm_project)(DM,DM,Vec) DMSwarmProjectMethod; /* swarm, geometry, result */ 757795646SDave May 857795646SDave May //typedef enum { PROJECT_DMDA_AQ1=0, PROJECT_DMDA_P0 } DMSwarmDMDAProjectionType; 957795646SDave May 1057795646SDave May #if 0 1157795646SDave May 1257795646SDave May /* Defines what the local space will be */ 13b62e03f8SDave May PetscErrorCode DMSwarmSetOverlap(void) 1457795646SDave May { 1557795646SDave May 16b62e03f8SDave May PetscFunctionReturn(0); 1757795646SDave May } 1857795646SDave May 1957795646SDave May 2057795646SDave May /* coordinates */ 2157795646SDave May /* 2257795646SDave May DMGetCoordinateDM returns self 2357795646SDave May DMGetCoordinates and DMGetCoordinatesLocal return same thing 2457795646SDave May Local view could be used to define overlapping information 2557795646SDave May */ 2657795646SDave May 2757795646SDave May #endif 2857795646SDave May 2957795646SDave May #undef __FUNCT__ 30b5bcf523SDave May #define __FUNCT__ "DMSwarmVectorDefineField" 31b5bcf523SDave May PETSC_EXTERN PetscErrorCode DMSwarmVectorDefineField(DM dm,const char fieldname[]) 32b5bcf523SDave May { 33b5bcf523SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 34b5bcf523SDave May PetscErrorCode ierr; 35b5bcf523SDave May PetscInt bs,n; 36b5bcf523SDave May PetscScalar *array; 37b5bcf523SDave May PetscDataType type; 38b5bcf523SDave May 393454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 406845f8f5SDave May ierr = DataBucketGetSizes(swarm->db,&n,NULL,NULL);CHKERRQ(ierr); 41b5bcf523SDave May ierr = DMSwarmGetField(dm,fieldname,&bs,&type,(void**)&array);CHKERRQ(ierr); 42b5bcf523SDave May 43b5bcf523SDave May /* Check all fields are of type PETSC_REAL or PETSC_SCALAR */ 44b5bcf523SDave May if (type != PETSC_REAL) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Only valid for PETSC_REAL"); 45b5bcf523SDave May 46b5bcf523SDave May PetscSNPrintf(swarm->vec_field_name,PETSC_MAX_PATH_LEN-1,"%s",fieldname); 47b5bcf523SDave May swarm->vec_field_set = PETSC_TRUE; 48b5bcf523SDave May swarm->vec_field_bs = 1;//bs; 49b5bcf523SDave May swarm->vec_field_nlocal = n; 50b5bcf523SDave May 51b5bcf523SDave May PetscFunctionReturn(0); 52b5bcf523SDave May } 53b5bcf523SDave May 54b5bcf523SDave May /* requires DMSwarmDefineFieldVector has been called */ 55b5bcf523SDave May #undef __FUNCT__ 56b5bcf523SDave May #define __FUNCT__ "DMCreateGlobalVector_Swarm" 57b5bcf523SDave May PetscErrorCode DMCreateGlobalVector_Swarm(DM dm,Vec *vec) 58b5bcf523SDave May { 59b5bcf523SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 60b5bcf523SDave May PetscErrorCode ierr; 61b5bcf523SDave May Vec x; 62b5bcf523SDave May char name[PETSC_MAX_PATH_LEN]; 63b5bcf523SDave May 643454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 65b5bcf523SDave May if (!swarm->vec_field_set) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Must call DMSwarmVectorDefineField first"); 66b5bcf523SDave May PetscSNPrintf(name,PETSC_MAX_PATH_LEN-1,"DMSwarmField_%s",swarm->vec_field_name); 67b5bcf523SDave May ierr = VecCreate(PetscObjectComm((PetscObject)dm),&x);CHKERRQ(ierr); 68b5bcf523SDave May ierr = PetscObjectSetName((PetscObject)x,name);CHKERRQ(ierr); 69b5bcf523SDave May ierr = VecSetSizes(x,swarm->vec_field_nlocal,PETSC_DETERMINE);CHKERRQ(ierr); 70b5bcf523SDave May ierr = VecSetBlockSize(x,swarm->vec_field_bs);CHKERRQ(ierr); 71b5bcf523SDave May ierr = VecSetFromOptions(x);CHKERRQ(ierr); 72b5bcf523SDave May *vec = x; 73b5bcf523SDave May 74b5bcf523SDave May PetscFunctionReturn(0); 75b5bcf523SDave May } 76b5bcf523SDave May 77b5bcf523SDave May /* requires DMSwarmDefineFieldVector has been called */ 78b5bcf523SDave May #undef __FUNCT__ 79b5bcf523SDave May #define __FUNCT__ "DMCreateLocalVector_Swarm" 80b5bcf523SDave May PetscErrorCode DMCreateLocalVector_Swarm(DM dm,Vec *vec) 81b5bcf523SDave May { 82b5bcf523SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 83b5bcf523SDave May PetscErrorCode ierr; 84b5bcf523SDave May Vec x; 85b5bcf523SDave May char name[PETSC_MAX_PATH_LEN]; 86b5bcf523SDave May 873454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 88b5bcf523SDave May if (!swarm->vec_field_set) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Must call DMSwarmVectorDefineField first"); 89b5bcf523SDave May PetscSNPrintf(name,PETSC_MAX_PATH_LEN-1,"DMSwarmField_%s",swarm->vec_field_name); 90b5bcf523SDave May ierr = VecCreate(PETSC_COMM_SELF,&x);CHKERRQ(ierr); 91b5bcf523SDave May ierr = PetscObjectSetName((PetscObject)x,name);CHKERRQ(ierr); 92b5bcf523SDave May ierr = VecSetSizes(x,swarm->vec_field_nlocal,swarm->vec_field_nlocal);CHKERRQ(ierr); 93b5bcf523SDave May ierr = VecSetBlockSize(x,swarm->vec_field_bs);CHKERRQ(ierr); 94b5bcf523SDave May ierr = VecSetFromOptions(x);CHKERRQ(ierr); 95b5bcf523SDave May *vec = x; 96b5bcf523SDave May 97b5bcf523SDave May PetscFunctionReturn(0); 98b5bcf523SDave May } 99b5bcf523SDave May 100b5bcf523SDave May #undef __FUNCT__ 101b5bcf523SDave May #define __FUNCT__ "DMSwarmCreateGlobalVectorFromField" 102b5bcf523SDave May PETSC_EXTERN PetscErrorCode DMSwarmCreateGlobalVectorFromField(DM dm,const char fieldname[],Vec *vec) 103b5bcf523SDave May { 104b5bcf523SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 105b5bcf523SDave May PetscErrorCode ierr; 106b5bcf523SDave May PetscInt bs,n; 107b5bcf523SDave May PetscScalar *array; 108b5bcf523SDave May Vec x; 109b5bcf523SDave May PetscDataType type; 110b5bcf523SDave May char name[PETSC_MAX_PATH_LEN]; 1113454631fSDave May PetscMPIInt commsize; 112b5bcf523SDave May 1133454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 1146845f8f5SDave May ierr = DataBucketGetSizes(swarm->db,&n,NULL,NULL);CHKERRQ(ierr); 115b5bcf523SDave May ierr = DMSwarmGetField(dm,fieldname,&bs,&type,(void**)&array);CHKERRQ(ierr); 116b5bcf523SDave May 117b5bcf523SDave May /* Check all fields are of type PETSC_REAL or PETSC_SCALAR */ 118b5bcf523SDave May if (type != PETSC_REAL) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Only valid for PETSC_REAL"); 119b5bcf523SDave May 1203454631fSDave May ierr = MPI_Comm_size(PetscObjectComm((PetscObject)dm),&commsize);CHKERRQ(ierr); 1213454631fSDave May if (commsize == 1) { 122b5bcf523SDave May ierr = VecCreateSeqWithArray(PetscObjectComm((PetscObject)dm),1,n,array,&x);CHKERRQ(ierr); 1233454631fSDave May } else { 1243454631fSDave May ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)dm),1,n,PETSC_DETERMINE,array,&x);CHKERRQ(ierr); 1253454631fSDave May } 126b5bcf523SDave May PetscSNPrintf(name,PETSC_MAX_PATH_LEN-1,"DMSwarm_VecFieldInPlace_%s",fieldname); 127b5bcf523SDave May ierr = PetscObjectComposeFunction((PetscObject)x,name,DMSwarmDestroyGlobalVectorFromField);CHKERRQ(ierr); 128b5bcf523SDave May 129b5bcf523SDave May /* Set guard */ 130b5bcf523SDave May 131b5bcf523SDave May *vec = x; 132b5bcf523SDave May PetscFunctionReturn(0); 133b5bcf523SDave May } 134b5bcf523SDave May 135b5bcf523SDave May #undef __FUNCT__ 136b5bcf523SDave May #define __FUNCT__ "DMSwarmDestroyGlobalVectorFromField" 137b5bcf523SDave May PETSC_EXTERN PetscErrorCode DMSwarmDestroyGlobalVectorFromField(DM dm,const char fieldname[],Vec *vec) 138b5bcf523SDave May { 139b5bcf523SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 140b5bcf523SDave May PetscErrorCode ierr; 141b5bcf523SDave May DataField gfield; 142b5bcf523SDave May char name[PETSC_MAX_PATH_LEN]; 143b5bcf523SDave May void (*fptr)(void); 144b5bcf523SDave May 145b5bcf523SDave May /* get data field */ 1462eac95f8SDave May ierr = DataBucketGetDataFieldByName(swarm->db,fieldname,&gfield);CHKERRQ(ierr); 147b5bcf523SDave May 148b5bcf523SDave May /* check vector is an inplace array */ 149b5bcf523SDave May PetscSNPrintf(name,PETSC_MAX_PATH_LEN-1,"DMSwarm_VecFieldInPlace_%s",fieldname); 150b5bcf523SDave May ierr = PetscObjectQueryFunction((PetscObject)(*vec),name,&fptr);CHKERRQ(ierr); 151b5bcf523SDave May if (!fptr) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Vector being destroyed was not created from DMSwarm field(%s)",fieldname); 152b5bcf523SDave May 153b5bcf523SDave May /* restore data field */ 1546845f8f5SDave May ierr = DataFieldRestoreAccess(gfield);CHKERRQ(ierr); 155b5bcf523SDave May 156b5bcf523SDave May ierr = VecDestroy(vec);CHKERRQ(ierr); 157b5bcf523SDave May 158b5bcf523SDave May PetscFunctionReturn(0); 159b5bcf523SDave May } 160b5bcf523SDave May 161b5bcf523SDave May /* 162b5bcf523SDave May PETSC_EXTERN PetscErrorCode DMSwarmCreateGlobalVectorFromFields(DM dm,const PetscInt nf,const char *fieldnames[],Vec *vec) 163b5bcf523SDave May { 164b5bcf523SDave May PetscFunctionReturn(0); 165b5bcf523SDave May } 166b5bcf523SDave May 167b5bcf523SDave May PETSC_EXTERN PetscErrorCode DMSwarmRestoreGlobalVectorFromFields(DM dm,Vec *vec) 168b5bcf523SDave May { 169b5bcf523SDave May PetscFunctionReturn(0); 170b5bcf523SDave May } 171b5bcf523SDave May */ 172b5bcf523SDave May 173b5bcf523SDave May #undef __FUNCT__ 1745f50eb2eSDave May #define __FUNCT__ "DMSwarmInitializeFieldRegister" 1755f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmInitializeFieldRegister(DM dm) 1765f50eb2eSDave May { 1775f50eb2eSDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 1783454631fSDave May PetscErrorCode ierr; 1793454631fSDave May 1805f50eb2eSDave May swarm->field_registration_initialized = PETSC_TRUE; 1813454631fSDave May 1823454631fSDave May ierr = DMSwarmRegisterPetscDatatypeField(dm,"DMSwarm_pid",1,PETSC_LONG);CHKERRQ(ierr); /* unique identifer */ 1833454631fSDave May ierr = DMSwarmRegisterPetscDatatypeField(dm,"DMSwarm_rank",1,PETSC_INT);CHKERRQ(ierr); /* used for communication */ 1843454631fSDave May 1855f50eb2eSDave May PetscFunctionReturn(0); 1865f50eb2eSDave May } 1875f50eb2eSDave May 1885f50eb2eSDave May #undef __FUNCT__ 1895f50eb2eSDave May #define __FUNCT__ "DMSwarmFinalizeFieldRegister" 1905f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmFinalizeFieldRegister(DM dm) 1915f50eb2eSDave May { 1925f50eb2eSDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 1936845f8f5SDave May PetscErrorCode ierr; 1946845f8f5SDave May 1955f50eb2eSDave May swarm->field_registration_finalized = PETSC_TRUE; 1966845f8f5SDave May ierr = DataBucketFinalize(swarm->db);CHKERRQ(ierr); 1975f50eb2eSDave May PetscFunctionReturn(0); 1985f50eb2eSDave May } 1995f50eb2eSDave May 2005f50eb2eSDave May #undef __FUNCT__ 2015f50eb2eSDave May #define __FUNCT__ "DMSwarmSetLocalSizes" 2025f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmSetLocalSizes(DM dm,PetscInt nlocal,PetscInt buffer) 2035f50eb2eSDave May { 2045f50eb2eSDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 2056845f8f5SDave May PetscErrorCode ierr; 2065f50eb2eSDave May 2076845f8f5SDave May ierr = DataBucketSetSizes(swarm->db,nlocal,buffer);CHKERRQ(ierr); 2085f50eb2eSDave May 2095f50eb2eSDave May PetscFunctionReturn(0); 2105f50eb2eSDave May } 2115f50eb2eSDave May 2125f50eb2eSDave May #undef __FUNCT__ 213b62e03f8SDave May #define __FUNCT__ "DMSwarmRegisterPetscDatatypeField" 2145f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmRegisterPetscDatatypeField(DM dm,const char fieldname[],PetscInt blocksize,PetscDataType type) 215b62e03f8SDave May { 2162eac95f8SDave May PetscErrorCode ierr; 217b62e03f8SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 218b62e03f8SDave May size_t size; 219b62e03f8SDave May 2205f50eb2eSDave May if (!swarm->field_registration_initialized) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Must call DMSwarmInitializeFieldRegister() first"); 2215f50eb2eSDave May if (swarm->field_registration_finalized) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Cannot register additional fields after calling DMSwarmFinalizeFieldRegister() first"); 2225f50eb2eSDave May 2235f50eb2eSDave May if (type == PETSC_OBJECT) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 2245f50eb2eSDave May if (type == PETSC_FUNCTION) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 2255f50eb2eSDave May if (type == PETSC_STRING) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 2265f50eb2eSDave May if (type == PETSC_STRUCT) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 2275f50eb2eSDave May if (type == PETSC_DATATYPE_UNKNOWN) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 228b62e03f8SDave May 229b62e03f8SDave May switch (type) { 230b62e03f8SDave May case PETSC_CHAR: 231b62e03f8SDave May size = sizeof(PetscChar); 232b62e03f8SDave May break; 233b62e03f8SDave May case PETSC_SHORT: 234b62e03f8SDave May size = sizeof(PetscShort); 235b62e03f8SDave May break; 236b62e03f8SDave May case PETSC_INT: 237b62e03f8SDave May size = sizeof(PetscInt); 238b62e03f8SDave May break; 239b62e03f8SDave May case PETSC_LONG: 240b62e03f8SDave May size = sizeof(Petsc64bitInt); 241b62e03f8SDave May break; 242b62e03f8SDave May case PETSC_FLOAT: 243b62e03f8SDave May size = sizeof(PetscFloat); 244b62e03f8SDave May break; 245b62e03f8SDave May case PETSC_DOUBLE: 246b62e03f8SDave May size = sizeof(PetscReal); 247b62e03f8SDave May break; 248b62e03f8SDave May 249b62e03f8SDave May default: 2505f50eb2eSDave May SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Valid for {char,short,int,long,float,double}"); 251b62e03f8SDave May break; 252b62e03f8SDave May } 253b62e03f8SDave May 254b62e03f8SDave May /* Load a specific data type into data bucket, specifying textual name and its size in bytes */ 2552eac95f8SDave May ierr = DataBucketRegisterField(swarm->db,"DMSwarmRegisterPetscDatatypeField",fieldname,size,NULL);CHKERRQ(ierr); 256b62e03f8SDave May swarm->db->field[swarm->db->nfields-1]->petsc_type = type; 257b62e03f8SDave May 258b62e03f8SDave May PetscFunctionReturn(0); 259b62e03f8SDave May } 260b62e03f8SDave May 261b62e03f8SDave May #undef __FUNCT__ 262b62e03f8SDave May #define __FUNCT__ "DMSwarmRegisterUserStructField" 2635f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmRegisterUserStructField(DM dm,const char fieldname[],size_t size) 264b62e03f8SDave May { 2652eac95f8SDave May PetscErrorCode ierr; 266b62e03f8SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 267b62e03f8SDave May 2682eac95f8SDave May ierr = DataBucketRegisterField(swarm->db,"DMSwarmRegisterUserStructField",fieldname,size,NULL);CHKERRQ(ierr); 269b62e03f8SDave May swarm->db->field[swarm->db->nfields-1]->petsc_type = PETSC_STRUCT ; 270b62e03f8SDave May 271b62e03f8SDave May PetscFunctionReturn(0); 272b62e03f8SDave May } 273b62e03f8SDave May 274b62e03f8SDave May #undef __FUNCT__ 275b62e03f8SDave May #define __FUNCT__ "DMSwarmRegisterUserDatatypeField" 2765f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmRegisterUserDatatypeField(DM dm,const char fieldname[],size_t size) 277b62e03f8SDave May { 278b62e03f8SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 2796845f8f5SDave May PetscErrorCode ierr; 280b62e03f8SDave May 2816845f8f5SDave May ierr = DataBucketRegisterField(swarm->db,"DMSwarmRegisterUserDatatypeField",fieldname,size,NULL);CHKERRQ(ierr); 282b62e03f8SDave May swarm->db->field[swarm->db->nfields-1]->petsc_type = PETSC_DATATYPE_UNKNOWN; 283b62e03f8SDave May 284b62e03f8SDave May PetscFunctionReturn(0); 285b62e03f8SDave May } 286b62e03f8SDave May 287b62e03f8SDave May #undef __FUNCT__ 288b62e03f8SDave May #define __FUNCT__ "DMSwarmGetField" 2895f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmGetField(DM dm,const char fieldname[],PetscInt *blocksize,PetscDataType *type,void **data) 290b62e03f8SDave May { 291b62e03f8SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 292b62e03f8SDave May DataField gfield; 2932eac95f8SDave May PetscErrorCode ierr; 294b62e03f8SDave May 2953454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 2963454631fSDave May 2972eac95f8SDave May ierr = DataBucketGetDataFieldByName(swarm->db,fieldname,&gfield);CHKERRQ(ierr); 2986845f8f5SDave May ierr = DataFieldGetAccess(gfield);CHKERRQ(ierr); 2996845f8f5SDave May ierr = DataFieldGetEntries(gfield,data);CHKERRQ(ierr); 300b5bcf523SDave May if (blocksize) {} 301b5bcf523SDave May if (type) { *type = gfield->petsc_type; } 302b62e03f8SDave May 303b62e03f8SDave May PetscFunctionReturn(0); 304b62e03f8SDave May } 305b62e03f8SDave May 306b62e03f8SDave May #undef __FUNCT__ 307b62e03f8SDave May #define __FUNCT__ "DMSwarmRestoreField" 3085f50eb2eSDave May PETSC_EXTERN PetscErrorCode DMSwarmRestoreField(DM dm,const char fieldname[],PetscInt *blocksize,PetscDataType *type,void **data) 309b62e03f8SDave May { 310b62e03f8SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 311b62e03f8SDave May DataField gfield; 3122eac95f8SDave May PetscErrorCode ierr; 313b62e03f8SDave May 3142eac95f8SDave May ierr = DataBucketGetDataFieldByName(swarm->db,fieldname,&gfield);CHKERRQ(ierr); 3156845f8f5SDave May ierr = DataFieldRestoreAccess(gfield);CHKERRQ(ierr); 316b62e03f8SDave May if (data) *data = NULL; 317b62e03f8SDave May 318b62e03f8SDave May PetscFunctionReturn(0); 319b62e03f8SDave May } 320b62e03f8SDave May 321cb1d1399SDave May #undef __FUNCT__ 322cb1d1399SDave May #define __FUNCT__ "DMSwarmAddPoint" 323cb1d1399SDave May PETSC_EXTERN PetscErrorCode DMSwarmAddPoint(DM dm) 324cb1d1399SDave May { 325cb1d1399SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 326cb1d1399SDave May PetscErrorCode ierr; 327cb1d1399SDave May 3283454631fSDave May if (!swarm->issetup) { ierr = DMSetUp(dm);CHKERRQ(ierr); } 329cb1d1399SDave May ierr = DataBucketAddPoint(swarm->db);CHKERRQ(ierr); 330cb1d1399SDave May PetscFunctionReturn(0); 331cb1d1399SDave May } 332cb1d1399SDave May 333cb1d1399SDave May #undef __FUNCT__ 334cb1d1399SDave May #define __FUNCT__ "DMSwarmAddNPoints" 335cb1d1399SDave May PETSC_EXTERN PetscErrorCode DMSwarmAddNPoints(DM dm,PetscInt npoints) 336cb1d1399SDave May { 337cb1d1399SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 338cb1d1399SDave May PetscErrorCode ierr; 339cb1d1399SDave May PetscInt nlocal; 340cb1d1399SDave May 341cb1d1399SDave May ierr = DataBucketGetSizes(swarm->db,&nlocal,NULL,NULL);CHKERRQ(ierr); 342cb1d1399SDave May nlocal = nlocal + npoints; 343cb1d1399SDave May ierr = DataBucketSetSizes(swarm->db,nlocal,-1);CHKERRQ(ierr); 344cb1d1399SDave May PetscFunctionReturn(0); 345cb1d1399SDave May } 346cb1d1399SDave May 347cb1d1399SDave May #undef __FUNCT__ 348cb1d1399SDave May #define __FUNCT__ "DMSwarmRemovePoint" 349cb1d1399SDave May PETSC_EXTERN PetscErrorCode DMSwarmRemovePoint(DM dm) 350cb1d1399SDave May { 351cb1d1399SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 352cb1d1399SDave May PetscErrorCode ierr; 353cb1d1399SDave May 354cb1d1399SDave May ierr = DataBucketRemovePoint(swarm->db);CHKERRQ(ierr); 355cb1d1399SDave May PetscFunctionReturn(0); 356cb1d1399SDave May } 357cb1d1399SDave May 358cb1d1399SDave May #undef __FUNCT__ 359cb1d1399SDave May #define __FUNCT__ "DMSwarmRemovePointAtIndex" 360cb1d1399SDave May PETSC_EXTERN PetscErrorCode DMSwarmRemovePointAtIndex(DM dm,PetscInt idx) 361cb1d1399SDave May { 362cb1d1399SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 363cb1d1399SDave May PetscErrorCode ierr; 364cb1d1399SDave May 365cb1d1399SDave May ierr = DataBucketRemovePointAtIndex(swarm->db,idx);CHKERRQ(ierr); 366cb1d1399SDave May PetscFunctionReturn(0); 367cb1d1399SDave May } 368b62e03f8SDave May 3693454631fSDave May #undef __FUNCT__ 370*095059a4SDave May #define __FUNCT__ "DMSwarmMigrate_Basic" 371*095059a4SDave May PetscErrorCode DMSwarmMigrate_Basic(DM dm,PetscBool remove_sent_points) 3723454631fSDave May { 3733454631fSDave May PetscFunctionReturn(0); 3743454631fSDave May } 3753454631fSDave May 3763454631fSDave May #undef __FUNCT__ 377*095059a4SDave May #define __FUNCT__ "DMSwarmMigrate" 378*095059a4SDave May PETSC_EXTERN PetscErrorCode DMSwarmMigrate(DM dm,PetscBool remove_sent_points) 3793454631fSDave May { 3803454631fSDave May PetscErrorCode ierr; 381*095059a4SDave May ierr = DMSwarmMigrate_Basic(dm,remove_sent_points);CHKERRQ(ierr); 3823454631fSDave May PetscFunctionReturn(0); 3833454631fSDave May } 3843454631fSDave May 3853454631fSDave May 3863454631fSDave May #undef __FUNCT__ 3873454631fSDave May #define __FUNCT__ "DMSetup_Swarm" 3883454631fSDave May PetscErrorCode DMSetup_Swarm(DM dm) 3893454631fSDave May { 3903454631fSDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 3913454631fSDave May PetscErrorCode ierr; 3923454631fSDave May PetscMPIInt rank; 3933454631fSDave May PetscInt p,npoints,*rankval; 3943454631fSDave May 3953454631fSDave May if (swarm->issetup) PetscFunctionReturn(0); 3963454631fSDave May 3973454631fSDave May PetscPrintf(PETSC_COMM_SELF,"Swarm setup \n"); 3983454631fSDave May swarm->issetup = PETSC_TRUE; 3993454631fSDave May 4003454631fSDave May /* check some fields were registered */ 4013454631fSDave May if (swarm->db->nfields <= 2) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"At least one field user must be registered via DMSwarmRegisterXXX()"); 4023454631fSDave May 4033454631fSDave May /* check local sizes were set */ 4043454631fSDave May if (swarm->db->L == -1) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_USER,"Local sizes must be set via DMSwarmSetLocalSizes()"); 4053454631fSDave May 4063454631fSDave May /* initialize values in pid and rank placeholders */ 4073454631fSDave May /* TODO: [pid - use MPI_Scan] */ 4083454631fSDave May 4093454631fSDave May ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm),&rank);CHKERRQ(ierr); 4103454631fSDave May ierr = DataBucketGetSizes(swarm->db,&npoints,NULL,NULL);CHKERRQ(ierr); 4113454631fSDave May ierr = DMSwarmGetField(dm,"DMSwarm_rank",NULL,NULL,(void**)&rankval);CHKERRQ(ierr); 4123454631fSDave May for (p=0; p<npoints; p++) { 4133454631fSDave May rankval[p] = (PetscInt)rank; 4143454631fSDave May } 4153454631fSDave May ierr = DMSwarmRestoreField(dm,"DMSwarm_rank",NULL,NULL,(void**)&rankval);CHKERRQ(ierr); 4163454631fSDave May 4173454631fSDave May PetscFunctionReturn(0); 4183454631fSDave May } 4193454631fSDave May 420b62e03f8SDave May #undef __FUNCT__ 42157795646SDave May #define __FUNCT__ "DMDestroy_Swarm" 42257795646SDave May PetscErrorCode DMDestroy_Swarm(DM dm) 42357795646SDave May { 42457795646SDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 42557795646SDave May PetscErrorCode ierr; 42657795646SDave May 42757795646SDave May PetscFunctionBegin; 4286845f8f5SDave May ierr = DataBucketDestroy(&swarm->db);CHKERRQ(ierr); 42957795646SDave May ierr = PetscFree(swarm);CHKERRQ(ierr); 43057795646SDave May PetscFunctionReturn(0); 43157795646SDave May } 43257795646SDave May 43357795646SDave May #undef __FUNCT__ 4345f50eb2eSDave May #define __FUNCT__ "DMView_Swarm" 4355f50eb2eSDave May PetscErrorCode DMView_Swarm(DM dm, PetscViewer viewer) 4365f50eb2eSDave May { 4375f50eb2eSDave May DM_Swarm *swarm = (DM_Swarm*)dm->data; 4385f50eb2eSDave May PetscBool iascii,ibinary,ishdf5,isvtk; 4395f50eb2eSDave May PetscErrorCode ierr; 4405f50eb2eSDave May 4415f50eb2eSDave May PetscFunctionBegin; 4425f50eb2eSDave May PetscValidHeaderSpecific(dm,DM_CLASSID,1); 4435f50eb2eSDave May PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 4445f50eb2eSDave May ierr = PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr); 4455f50eb2eSDave May ierr = PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERBINARY,&ibinary);CHKERRQ(ierr); 4465f50eb2eSDave May ierr = PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk);CHKERRQ(ierr); 4475f50eb2eSDave May ierr = PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5);CHKERRQ(ierr); 4485f50eb2eSDave May if (iascii) { 4496845f8f5SDave May ierr = DataBucketView(PetscObjectComm((PetscObject)dm),swarm->db,NULL,DATABUCKET_VIEW_STDOUT);CHKERRQ(ierr); 4505f50eb2eSDave May } else if (ibinary) { 4515f50eb2eSDave May SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"NO VTK support"); 4525f50eb2eSDave May } else if (ishdf5) { 4535f50eb2eSDave May #if defined(PETSC_HAVE_HDF5) 4545f50eb2eSDave May SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"NO HDF5 support"); 4555f50eb2eSDave May #else 4565f50eb2eSDave May SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"HDF5 not supported. Please reconfigure using --download-hdf5"); 4575f50eb2eSDave May #endif 4585f50eb2eSDave May } else if (isvtk) { 4595f50eb2eSDave May SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"NO VTK support"); 4605f50eb2eSDave May } 4615f50eb2eSDave May PetscFunctionReturn(0); 4625f50eb2eSDave May } 4635f50eb2eSDave May 4645f50eb2eSDave May #undef __FUNCT__ 46557795646SDave May #define __FUNCT__ "DMCreate_Swarm" 46657795646SDave May PETSC_EXTERN PetscErrorCode DMCreate_Swarm(DM dm) 46757795646SDave May { 46857795646SDave May DM_Swarm *swarm; 46957795646SDave May PetscErrorCode ierr; 47057795646SDave May 47157795646SDave May PetscFunctionBegin; 47257795646SDave May PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 47357795646SDave May ierr = PetscNewLog(dm,&swarm);CHKERRQ(ierr); 47457795646SDave May 4756845f8f5SDave May ierr = DataBucketCreate(&swarm->db);CHKERRQ(ierr); 476b5bcf523SDave May swarm->vec_field_set = PETSC_FALSE; 4773454631fSDave May swarm->issetup = PETSC_FALSE; 478b62e03f8SDave May 47957795646SDave May dm->dim = 0; 48057795646SDave May dm->data = swarm; 48157795646SDave May 4825f50eb2eSDave May dm->ops->view = DMView_Swarm; 48357795646SDave May dm->ops->load = NULL; 48457795646SDave May dm->ops->setfromoptions = NULL; 48557795646SDave May dm->ops->clone = NULL; 4863454631fSDave May dm->ops->setup = DMSetup_Swarm; 48757795646SDave May dm->ops->createdefaultsection = NULL; 48857795646SDave May dm->ops->createdefaultconstraints = NULL; 489b5bcf523SDave May dm->ops->createglobalvector = DMCreateGlobalVector_Swarm; 490b5bcf523SDave May dm->ops->createlocalvector = DMCreateLocalVector_Swarm; 49157795646SDave May dm->ops->getlocaltoglobalmapping = NULL; 49257795646SDave May dm->ops->createfieldis = NULL; 49357795646SDave May dm->ops->createcoordinatedm = NULL; 49457795646SDave May dm->ops->getcoloring = NULL; 49557795646SDave May dm->ops->creatematrix = NULL; 49657795646SDave May dm->ops->createinterpolation = NULL; 49757795646SDave May dm->ops->getaggregates = NULL; 49857795646SDave May dm->ops->getinjection = NULL; 49957795646SDave May dm->ops->refine = NULL; 50057795646SDave May dm->ops->coarsen = NULL; 50157795646SDave May dm->ops->refinehierarchy = NULL; 50257795646SDave May dm->ops->coarsenhierarchy = NULL; 50357795646SDave May dm->ops->globaltolocalbegin = NULL; 50457795646SDave May dm->ops->globaltolocalend = NULL; 50557795646SDave May dm->ops->localtoglobalbegin = NULL; 50657795646SDave May dm->ops->localtoglobalend = NULL; 50757795646SDave May dm->ops->destroy = DMDestroy_Swarm; 50857795646SDave May dm->ops->createsubdm = NULL; 50957795646SDave May dm->ops->getdimpoints = NULL; 51057795646SDave May dm->ops->locatepoints = NULL; 51157795646SDave May 51257795646SDave May PetscFunctionReturn(0); 51357795646SDave May }