xref: /petsc/src/dm/impls/swarm/swarm.c (revision 095059a4a9da3c077576069a1cce327c884bb767)
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 }