xref: /petsc/src/dm/impls/moab/dmmbvec.cxx (revision 7ae5e5b63e531d1144854fb663ae16bda2a50fcf)
1b8ecf6d3SVijay Mahadevan #include <petsc-private/dmmbimpl.h> /*I  "petscdmmoab.h"   I*/
2b8ecf6d3SVijay Mahadevan #include <petsc-private/vecimpl.h>
3032b8ab6SVijay Mahadevan 
4032b8ab6SVijay Mahadevan #include <petscdmmoab.h>
5032b8ab6SVijay Mahadevan #include <MBTagConventions.hpp>
6032b8ab6SVijay Mahadevan 
7f6829af0SVijay Mahadevan /* declare some private DMMoab specific overrides */
8f6829af0SVijay Mahadevan static PetscErrorCode DMCreateVector_Moab_Private(DM dm,moab::Tag tag,const moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec);
9f6829af0SVijay Mahadevan static PetscErrorCode DMVecUserDestroy_Moab(void *user);
10f6829af0SVijay Mahadevan static PetscErrorCode DMVecDuplicate_Moab(Vec x,Vec *y);
11f6829af0SVijay Mahadevan static PetscErrorCode DMVecCreateTagName_Moab_Private(moab::ParallelComm *pcomm,char** tag_name);
12032b8ab6SVijay Mahadevan 
13032b8ab6SVijay Mahadevan #undef __FUNCT__
14f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabCreateVector"
15b8ecf6d3SVijay Mahadevan /*@C
16f6829af0SVijay Mahadevan   DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities
17f6829af0SVijay Mahadevan 
18f6829af0SVijay Mahadevan   Collective on MPI_Comm
19f6829af0SVijay Mahadevan 
20f6829af0SVijay Mahadevan   Input Parameter:
21b8ecf6d3SVijay Mahadevan + dm              - The DMMoab object being set
22f6829af0SVijay Mahadevan . tag             - If non-zero, block size will be taken from the tag size
23f6829af0SVijay Mahadevan . range           - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab
24f6829af0SVijay Mahadevan . is_global_vec   - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one
25f6829af0SVijay Mahadevan . destroy_tag     - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB
26f6829af0SVijay Mahadevan 
27f6829af0SVijay Mahadevan   Output Parameter:
28f6829af0SVijay Mahadevan . vec             - The created vector
29f6829af0SVijay Mahadevan 
30f6829af0SVijay Mahadevan   Level: beginner
31f6829af0SVijay Mahadevan 
32b8ecf6d3SVijay Mahadevan .keywords: Vec, create
33b8ecf6d3SVijay Mahadevan 
34b8ecf6d3SVijay Mahadevan .seealso: VecCreate()
35f6829af0SVijay Mahadevan @*/
36f6829af0SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,const moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec)
37f6829af0SVijay Mahadevan {
38f6829af0SVijay Mahadevan   PetscErrorCode     ierr;
39f6829af0SVijay Mahadevan 
40f6829af0SVijay Mahadevan   PetscFunctionBegin;
41f6829af0SVijay Mahadevan   if(!tag && (!range || range->empty())) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag and range cannot be null.");
42f6829af0SVijay Mahadevan 
43f6829af0SVijay Mahadevan   ierr = DMCreateVector_Moab_Private(dm,tag,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr);
44f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
45f6829af0SVijay Mahadevan }
46f6829af0SVijay Mahadevan 
47f6829af0SVijay Mahadevan 
48f6829af0SVijay Mahadevan #undef __FUNCT__
49f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecTag"
50b8ecf6d3SVijay Mahadevan /*@C
51f6829af0SVijay Mahadevan   DMMoabGetVecTag - Get the MOAB tag associated with this Vec
52f6829af0SVijay Mahadevan 
53f6829af0SVijay Mahadevan   Input Parameter:
54f6829af0SVijay Mahadevan . vec - Vec being queried
55f6829af0SVijay Mahadevan 
56f6829af0SVijay Mahadevan   Output Parameter:
57f6829af0SVijay Mahadevan . tag - Tag associated with this Vec. NULL if vec is a native PETSc Vec.
58f6829af0SVijay Mahadevan 
59f6829af0SVijay Mahadevan   Level: beginner
60f6829af0SVijay Mahadevan 
61b8ecf6d3SVijay Mahadevan .keywords: DMMoab, MOAB tag
62b8ecf6d3SVijay Mahadevan 
63b8ecf6d3SVijay Mahadevan .seealso: DMMoabCreateVector(), DMMoabGetVecRange()
64f6829af0SVijay Mahadevan @*/
65f6829af0SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag)
66f6829af0SVijay Mahadevan {
67f6829af0SVijay Mahadevan   PetscContainer  moabdata;
68f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab;
69f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
70f6829af0SVijay Mahadevan 
71f6829af0SVijay Mahadevan   PetscFunctionBegin;
72f6829af0SVijay Mahadevan   PetscValidPointer(tag,2);
73f6829af0SVijay Mahadevan 
74f6829af0SVijay Mahadevan   /* Get the MOAB private data */
75f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
76f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr);
77f6829af0SVijay Mahadevan 
78f6829af0SVijay Mahadevan   *tag = vmoab->tag;
79f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
80f6829af0SVijay Mahadevan }
81f6829af0SVijay Mahadevan 
82f6829af0SVijay Mahadevan 
83f6829af0SVijay Mahadevan #undef __FUNCT__
84f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecRange"
85b8ecf6d3SVijay Mahadevan /*@C
86f6829af0SVijay Mahadevan   DMMoabGetVecRange - Get the MOAB entities associated with this Vec
87f6829af0SVijay Mahadevan 
88f6829af0SVijay Mahadevan   Input Parameter:
89f6829af0SVijay Mahadevan . vec   - Vec being queried
90f6829af0SVijay Mahadevan 
91f6829af0SVijay Mahadevan   Output Parameter:
92f6829af0SVijay Mahadevan . range - Entities associated with this Vec. NULL if vec is a native PETSc Vec.
93f6829af0SVijay Mahadevan 
94f6829af0SVijay Mahadevan   Level: beginner
95f6829af0SVijay Mahadevan 
96b8ecf6d3SVijay Mahadevan .keywords: DMMoab, MOAB range
97b8ecf6d3SVijay Mahadevan 
98b8ecf6d3SVijay Mahadevan .seealso: DMMoabCreateVector(), DMMoabGetVecTag()
99f6829af0SVijay Mahadevan @*/
100f6829af0SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range)
101f6829af0SVijay Mahadevan {
102f6829af0SVijay Mahadevan   PetscContainer  moabdata;
103f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab;
104f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
105f6829af0SVijay Mahadevan 
106f6829af0SVijay Mahadevan   PetscFunctionBegin;
107f6829af0SVijay Mahadevan   PetscValidPointer(range,2);
108f6829af0SVijay Mahadevan 
109f6829af0SVijay Mahadevan   /* Get the MOAB private data handle */
110f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
111f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr);
112f6829af0SVijay Mahadevan 
113f6829af0SVijay Mahadevan   *range = *vmoab->tag_range;
114f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
115f6829af0SVijay Mahadevan }
116f6829af0SVijay Mahadevan 
117f6829af0SVijay Mahadevan 
118f6829af0SVijay Mahadevan #undef __FUNCT__
119f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArray"
120b8ecf6d3SVijay Mahadevan /*@C
121f6829af0SVijay Mahadevan   DMMoabVecGetArray - Returns the writable direct access array to the local representation of MOAB tag data for the underlying vector using locally owned+ghosted range of entities
122f6829af0SVijay Mahadevan 
123f6829af0SVijay Mahadevan   Collective on MPI_Comm
124f6829af0SVijay Mahadevan 
125f6829af0SVijay Mahadevan   Input Parameter:
126f6829af0SVijay Mahadevan . dm              - The DMMoab object being set
127f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
128f6829af0SVijay Mahadevan 
129f6829af0SVijay Mahadevan   Output Parameter:
130f6829af0SVijay Mahadevan . array           - The local data array
131f6829af0SVijay Mahadevan 
132f6829af0SVijay Mahadevan   Level: intermediate
133f6829af0SVijay Mahadevan 
134f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
135f6829af0SVijay Mahadevan 
136f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
137f6829af0SVijay Mahadevan @*/
138f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecGetArray(DM dm,Vec vec,void* array)
139f6829af0SVijay Mahadevan {
140f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
141f6829af0SVijay Mahadevan   moab::ErrorCode merr;
142f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
143f6829af0SVijay Mahadevan   PetscInt        count,i,f;
144f6829af0SVijay Mahadevan   moab::Tag       vtag;
145f6829af0SVijay Mahadevan   PetscScalar     **varray;
146f6829af0SVijay Mahadevan   PetscScalar     *marray;
147f6829af0SVijay Mahadevan   PetscContainer  moabdata;
148f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab,*xmoab;
149f6829af0SVijay Mahadevan 
150f6829af0SVijay Mahadevan   PetscFunctionBegin;
151f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
152f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
153f6829af0SVijay Mahadevan   PetscValidPointer(array,3);
154f6829af0SVijay Mahadevan   dmmoab=(DM_Moab*)dm->data;
155f6829af0SVijay Mahadevan 
156f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
157f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
158f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
159f6829af0SVijay Mahadevan 
160f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
161f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
162f6829af0SVijay Mahadevan 
163f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
164f6829af0SVijay Mahadevan 
165f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
166f6829af0SVijay Mahadevan     ierr = DMGetLocalVector(dm,&vmoab->local);CHKERRQ(ierr);
167f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalBegin(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr);
168f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalEnd(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr);
169f6829af0SVijay Mahadevan 
170f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
171f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
172f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr);
173f6829af0SVijay Mahadevan 
174f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
175f6829af0SVijay Mahadevan     ierr = VecGhostGetLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr);
176f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
177f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
178f6829af0SVijay Mahadevan 
179f6829af0SVijay Mahadevan     ierr = VecGetArray(xmoab->local, varray);CHKERRQ(ierr);
180f6829af0SVijay Mahadevan   }
181f6829af0SVijay Mahadevan   else {
182f6829af0SVijay Mahadevan 
183f6829af0SVijay Mahadevan     /* Get the MOAB private data */
184f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr);
185f6829af0SVijay Mahadevan 
186f6829af0SVijay Mahadevan     /* exchange the data into ghost cells first */
187f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->exchange_tags(vtag,*dmmoab->vlocal);MBERRNM(merr);
188f6829af0SVijay Mahadevan 
189*7ae5e5b6SVijay Mahadevan     ierr = PetscMalloc1((dmmoab->nloc+dmmoab->nghost)*dmmoab->numFields,varray);CHKERRQ(ierr);
190f6829af0SVijay Mahadevan 
191f6829af0SVijay Mahadevan     /* Get the array data for local entities */
192f6829af0SVijay Mahadevan     merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr);
193f6829af0SVijay Mahadevan     if (count!=dmmoab->nloc+dmmoab->nghost) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,dmmoab->nloc+dmmoab->nghost);
194f6829af0SVijay Mahadevan 
195f6829af0SVijay Mahadevan     i=0;
196f6829af0SVijay Mahadevan     for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) {
197f6829af0SVijay Mahadevan       for (f=0;f<dmmoab->numFields;f++,i++)
198f6829af0SVijay Mahadevan         (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f]=marray[i];
199f6829af0SVijay Mahadevan     }
200f6829af0SVijay Mahadevan   }
201f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
202f6829af0SVijay Mahadevan }
203f6829af0SVijay Mahadevan 
204f6829af0SVijay Mahadevan 
205f6829af0SVijay Mahadevan #undef __FUNCT__
206f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray"
207b8ecf6d3SVijay Mahadevan /*@C
208f6829af0SVijay Mahadevan   DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray
209f6829af0SVijay Mahadevan 
210f6829af0SVijay Mahadevan   Collective on MPI_Comm
211f6829af0SVijay Mahadevan 
212f6829af0SVijay Mahadevan   Input Parameter:
213f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
214f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
215b8ecf6d3SVijay Mahadevan . array           - The local data array
216f6829af0SVijay Mahadevan 
217f6829af0SVijay Mahadevan   Level: intermediate
218f6829af0SVijay Mahadevan 
219f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
220f6829af0SVijay Mahadevan 
221f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
222f6829af0SVijay Mahadevan @*/
223f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArray(DM dm,Vec vec,void* array)
224f6829af0SVijay Mahadevan {
225f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
226f6829af0SVijay Mahadevan   moab::ErrorCode merr;
227f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
228f6829af0SVijay Mahadevan   moab::Tag       vtag;
229f6829af0SVijay Mahadevan   PetscInt        count,i,f;
230f6829af0SVijay Mahadevan   PetscScalar     **varray;
231f6829af0SVijay Mahadevan   PetscScalar     *marray;
232f6829af0SVijay Mahadevan   PetscContainer  moabdata;
233f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab,*xmoab;
234f6829af0SVijay Mahadevan 
235f6829af0SVijay Mahadevan   PetscFunctionBegin;
236f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
237f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
238f6829af0SVijay Mahadevan   PetscValidPointer(array,3);
239f6829af0SVijay Mahadevan   dmmoab=(DM_Moab*)dm->data;
240f6829af0SVijay Mahadevan 
241f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
242f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
243f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
244f6829af0SVijay Mahadevan 
245f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
246f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
247f6829af0SVijay Mahadevan 
248f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
249f6829af0SVijay Mahadevan 
250f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
251f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
252f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr);
253f6829af0SVijay Mahadevan 
254f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
255f6829af0SVijay Mahadevan     ierr = VecRestoreArray(xmoab->local, varray);CHKERRQ(ierr);
256f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
257f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
258f6829af0SVijay Mahadevan     ierr = VecGhostRestoreLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr);
259f6829af0SVijay Mahadevan 
260f6829af0SVijay Mahadevan     /* restore local pieces */
261f6829af0SVijay Mahadevan     ierr = DMLocalToGlobalBegin(dm,vmoab->local,INSERT_VALUES,vec);CHKERRQ(ierr);
262f6829af0SVijay Mahadevan     ierr = DMLocalToGlobalEnd(dm,vmoab->local,INSERT_VALUES,vec);CHKERRQ(ierr);
263f6829af0SVijay Mahadevan     ierr = DMRestoreLocalVector(dm, &vmoab->local);CHKERRQ(ierr);
264f6829af0SVijay Mahadevan   }
265f6829af0SVijay Mahadevan   else {
266f6829af0SVijay Mahadevan 
267f6829af0SVijay Mahadevan     /* Get the MOAB private data */
268f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr);
269f6829af0SVijay Mahadevan 
270f6829af0SVijay Mahadevan     /* Get the array data for local entities */
271f6829af0SVijay Mahadevan     merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr);
272f6829af0SVijay Mahadevan     if (count!=dmmoab->nloc+dmmoab->nghost) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,dmmoab->nloc+dmmoab->nghost);
273f6829af0SVijay Mahadevan 
274f6829af0SVijay Mahadevan     i=0;
275f6829af0SVijay Mahadevan     for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) {
276f6829af0SVijay Mahadevan       for (f=0;f<dmmoab->numFields;f++,i++)
277f6829af0SVijay Mahadevan       marray[i] = (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f];
278f6829af0SVijay Mahadevan     }
279f6829af0SVijay Mahadevan 
280f6829af0SVijay Mahadevan     /* reduce the tags correctly -> should probably let the user choose how to reduce in the future
281f6829af0SVijay Mahadevan       For all FEM residual based assembly calculations, MPI_SUM should serve well */
282f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->reduce_tags(vtag,MPI_SUM,*dmmoab->vlocal);MBERRV(dmmoab->mbiface,merr);
283f6829af0SVijay Mahadevan 
284f6829af0SVijay Mahadevan     ierr = PetscFree(*varray);CHKERRQ(ierr);
285f6829af0SVijay Mahadevan   }
286f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
287f6829af0SVijay Mahadevan }
288f6829af0SVijay Mahadevan 
289f6829af0SVijay Mahadevan #undef __FUNCT__
290f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArrayRead"
291b8ecf6d3SVijay Mahadevan /*@C
292f6829af0SVijay Mahadevan   DMMoabVecGetArrayRead - Returns the read-only direct access array to the local representation of MOAB tag data for the underlying vector using locally owned+ghosted range of entities
293f6829af0SVijay Mahadevan 
294f6829af0SVijay Mahadevan   Collective on MPI_Comm
295f6829af0SVijay Mahadevan 
296f6829af0SVijay Mahadevan   Input Parameter:
297f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
298f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
299f6829af0SVijay Mahadevan 
300f6829af0SVijay Mahadevan   Output Parameter:
301f6829af0SVijay Mahadevan . array           - The local data array
302f6829af0SVijay Mahadevan 
303f6829af0SVijay Mahadevan   Level: intermediate
304f6829af0SVijay Mahadevan 
305f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
306f6829af0SVijay Mahadevan 
307f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
308f6829af0SVijay Mahadevan @*/
309f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecGetArrayRead(DM dm,Vec vec,void* array)
310f6829af0SVijay Mahadevan {
311f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
312f6829af0SVijay Mahadevan   moab::ErrorCode merr;
313f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
314f6829af0SVijay Mahadevan   PetscInt        count,i,f;
315f6829af0SVijay Mahadevan   moab::Tag       vtag;
316f6829af0SVijay Mahadevan   PetscScalar     **varray;
317f6829af0SVijay Mahadevan   PetscScalar     *marray;
318f6829af0SVijay Mahadevan   PetscContainer  moabdata;
319f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab,*xmoab;
320f6829af0SVijay Mahadevan 
321f6829af0SVijay Mahadevan   PetscFunctionBegin;
322f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
323f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
324f6829af0SVijay Mahadevan   PetscValidPointer(array,3);
325f6829af0SVijay Mahadevan   dmmoab=(DM_Moab*)dm->data;
326f6829af0SVijay Mahadevan 
327f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
328f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
329f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
330f6829af0SVijay Mahadevan 
331f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
332f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
333f6829af0SVijay Mahadevan 
334f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
335f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
336f6829af0SVijay Mahadevan     ierr = DMGetLocalVector(dm,&vmoab->local);CHKERRQ(ierr);
337f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalBegin(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr);
338f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalEnd(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr);
339f6829af0SVijay Mahadevan 
340f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
341f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
342f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr);
343f6829af0SVijay Mahadevan 
344f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
345f6829af0SVijay Mahadevan     ierr = VecGhostGetLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr);
346f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
347f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
348f6829af0SVijay Mahadevan     ierr = VecGetArray(xmoab->local, varray);CHKERRQ(ierr);
349f6829af0SVijay Mahadevan   }
350f6829af0SVijay Mahadevan   else {
351f6829af0SVijay Mahadevan     /* Get the MOAB private data */
352f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr);
353f6829af0SVijay Mahadevan 
354f6829af0SVijay Mahadevan     /* exchange the data into ghost cells first */
355f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->exchange_tags(vtag,*dmmoab->vlocal);MBERRNM(merr);
356f6829af0SVijay Mahadevan 
357*7ae5e5b6SVijay Mahadevan     ierr = PetscMalloc1((dmmoab->nloc+dmmoab->nghost)*dmmoab->numFields,varray);CHKERRQ(ierr);
358f6829af0SVijay Mahadevan 
359f6829af0SVijay Mahadevan     /* Get the array data for local entities */
360f6829af0SVijay Mahadevan     merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr);
361f6829af0SVijay Mahadevan     if (count!=dmmoab->nloc+dmmoab->nghost) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,dmmoab->nloc+dmmoab->nghost);
362f6829af0SVijay Mahadevan 
363f6829af0SVijay Mahadevan     i=0;
364f6829af0SVijay Mahadevan     for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) {
365f6829af0SVijay Mahadevan       for (f=0;f<dmmoab->numFields;f++,i++)
366f6829af0SVijay Mahadevan         (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f]=marray[i];
367f6829af0SVijay Mahadevan     }
368f6829af0SVijay Mahadevan   }
369f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
370f6829af0SVijay Mahadevan }
371f6829af0SVijay Mahadevan 
372f6829af0SVijay Mahadevan 
373f6829af0SVijay Mahadevan #undef __FUNCT__
374f6829af0SVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArrayRead"
375b8ecf6d3SVijay Mahadevan /*@C
376f6829af0SVijay Mahadevan   DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray
377f6829af0SVijay Mahadevan 
378f6829af0SVijay Mahadevan   Collective on MPI_Comm
379f6829af0SVijay Mahadevan 
380f6829af0SVijay Mahadevan   Input Parameter:
381f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
382f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
383b8ecf6d3SVijay Mahadevan . array           - The local data array
384f6829af0SVijay Mahadevan 
385f6829af0SVijay Mahadevan   Level: intermediate
386f6829af0SVijay Mahadevan 
387f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
388f6829af0SVijay Mahadevan 
389f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
390f6829af0SVijay Mahadevan @*/
391f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArrayRead(DM dm,Vec vec,void* array)
392f6829af0SVijay Mahadevan {
393f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
394f6829af0SVijay Mahadevan   PetscScalar     **varray;
395f6829af0SVijay Mahadevan   PetscContainer  moabdata;
396f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab,*xmoab;
397f6829af0SVijay Mahadevan 
398f6829af0SVijay Mahadevan   PetscFunctionBegin;
399f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
400f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
401f6829af0SVijay Mahadevan   PetscValidPointer(array,3);
402f6829af0SVijay Mahadevan 
403f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
404f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
405f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
406f6829af0SVijay Mahadevan 
407f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
408f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
409f6829af0SVijay Mahadevan 
410f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
411f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
412f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
413f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr);
414f6829af0SVijay Mahadevan 
415f6829af0SVijay Mahadevan     /* restore the local representation of the arrays from Vectors */
416f6829af0SVijay Mahadevan     ierr = VecRestoreArray(xmoab->local, varray);CHKERRQ(ierr);
417f6829af0SVijay Mahadevan     ierr = VecGhostRestoreLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr);
418f6829af0SVijay Mahadevan 
419f6829af0SVijay Mahadevan     /* restore local pieces */
420f6829af0SVijay Mahadevan     ierr = DMRestoreLocalVector(dm, &vmoab->local);CHKERRQ(ierr);
421f6829af0SVijay Mahadevan   }
422f6829af0SVijay Mahadevan   else {
423f6829af0SVijay Mahadevan     /* Nothing to do but just free the memory allocated before */
424f6829af0SVijay Mahadevan     ierr = PetscFree(*varray);CHKERRQ(ierr);
425f6829af0SVijay Mahadevan 
426f6829af0SVijay Mahadevan   }
427f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
428f6829af0SVijay Mahadevan }
429f6829af0SVijay Mahadevan 
430f6829af0SVijay Mahadevan 
431f6829af0SVijay Mahadevan #undef __FUNCT__
432f6829af0SVijay Mahadevan #define __FUNCT__ "DMCreateVector_Moab_Private"
433f6829af0SVijay Mahadevan PetscErrorCode DMCreateVector_Moab_Private(DM dm,moab::Tag tag,const moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec)
434032b8ab6SVijay Mahadevan {
435032b8ab6SVijay Mahadevan   PetscErrorCode    ierr;
436032b8ab6SVijay Mahadevan   moab::ErrorCode   merr;
437032b8ab6SVijay Mahadevan   PetscBool         is_newtag;
438351b8a77SVijay Mahadevan   const moab::Range *range;
439bb8f3634SVijay Mahadevan   PetscInt          count,lnative_vec,gnative_vec;
4409088682fSVijay Mahadevan   std::string       ttname;
441c4427235SVijay Mahadevan   PetscScalar       *data_ptr,*defaultvals;
442032b8ab6SVijay Mahadevan 
443db66d124SVijay Mahadevan   Vec_MOAB *vmoab;
444032b8ab6SVijay Mahadevan   DM_Moab *dmmoab = (DM_Moab*)dm->data;
445032b8ab6SVijay Mahadevan   moab::ParallelComm *pcomm = dmmoab->pcomm;
446032b8ab6SVijay Mahadevan   moab::Interface *mbiface = dmmoab->mbiface;
447032b8ab6SVijay Mahadevan 
448032b8ab6SVijay Mahadevan   PetscFunctionBegin;
449d1839e75SVijay Mahadevan   if(sizeof(PetscReal) != sizeof(PetscScalar)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "MOAB tags only support Real types (Complex-type unsupported)");
450032b8ab6SVijay Mahadevan   if(!userrange) range = dmmoab->vowned;
451032b8ab6SVijay Mahadevan   else range = userrange;
452db66d124SVijay Mahadevan   if(!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first.");
453032b8ab6SVijay Mahadevan 
454bb8f3634SVijay Mahadevan   /* If the tag data is in a single sequence, use PETSc native vector since tag_iterate isn't useful anymore */
455bb8f3634SVijay Mahadevan   lnative_vec=(range->psize()-1);
456bb8f3634SVijay Mahadevan 
457f6829af0SVijay Mahadevan #if 0
4586465f021SVijay Mahadevan //  lnative_vec=1; /* NOTE: Testing PETSc vector: will force to create native vector all the time */
4596465f021SVijay Mahadevan //  lnative_vec=0; /* NOTE: Testing MOAB vector: will force to create MOAB tag_iterate based vector all the time */
460f6829af0SVijay Mahadevan #endif
461bb8f3634SVijay Mahadevan   ierr = MPI_Allreduce(&lnative_vec, &gnative_vec, 1, MPI_INT, MPI_MAX, pcomm->comm());CHKERRQ(ierr);
462bb8f3634SVijay Mahadevan 
463bb8f3634SVijay Mahadevan   /* Create the MOAB internal data object */
464f90c3b0eSVijay Mahadevan   ierr = PetscNew(&vmoab);CHKERRQ(ierr);
465bb8f3634SVijay Mahadevan   vmoab->is_native_vec=(gnative_vec>0?PETSC_TRUE:PETSC_FALSE);
466bb8f3634SVijay Mahadevan 
467bb8f3634SVijay Mahadevan   if (!vmoab->is_native_vec) {
4689088682fSVijay Mahadevan     merr = mbiface->tag_get_name(tag, ttname);
4699088682fSVijay Mahadevan     if (!ttname.length() && merr !=moab::MB_SUCCESS) {
470db66d124SVijay Mahadevan       /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */
471032b8ab6SVijay Mahadevan       char *tag_name = PETSC_NULL;
472f6829af0SVijay Mahadevan       ierr = DMVecCreateTagName_Moab_Private(pcomm,&tag_name);CHKERRQ(ierr);
473032b8ab6SVijay Mahadevan       is_newtag = PETSC_TRUE;
474032b8ab6SVijay Mahadevan 
475db66d124SVijay Mahadevan       /* Create the default value for the tag (all zeros) */
476*7ae5e5b6SVijay Mahadevan       ierr = PetscCalloc1(dmmoab->numFields,&defaultvals);CHKERRQ(ierr);
477032b8ab6SVijay Mahadevan 
478db66d124SVijay Mahadevan       /* Create the tag */
479d1839e75SVijay Mahadevan       merr = mbiface->tag_get_handle(tag_name,dmmoab->numFields,moab::MB_TYPE_DOUBLE,tag,
480c4427235SVijay Mahadevan                                     moab::MB_TAG_DENSE|moab::MB_TAG_CREAT,defaultvals);MBERRNM(merr);
481032b8ab6SVijay Mahadevan       ierr = PetscFree(tag_name);CHKERRQ(ierr);
482c4427235SVijay Mahadevan       ierr = PetscFree(defaultvals);CHKERRQ(ierr);
483032b8ab6SVijay Mahadevan     }
484032b8ab6SVijay Mahadevan     else {
485db66d124SVijay Mahadevan       /* Make sure the tag data is of type "double" */
486032b8ab6SVijay Mahadevan       moab::DataType tag_type;
487032b8ab6SVijay Mahadevan       merr = mbiface->tag_get_data_type(tag, tag_type);MBERRNM(merr);
488032b8ab6SVijay Mahadevan       if(tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE");
489032b8ab6SVijay Mahadevan       is_newtag = destroy_tag;
490032b8ab6SVijay Mahadevan     }
491032b8ab6SVijay Mahadevan 
492032b8ab6SVijay Mahadevan     vmoab->tag = tag;
493bb8f3634SVijay Mahadevan     vmoab->new_tag = is_newtag;
494bb8f3634SVijay Mahadevan   }
495032b8ab6SVijay Mahadevan   vmoab->mbiface = mbiface;
496032b8ab6SVijay Mahadevan   vmoab->pcomm = pcomm;
497032b8ab6SVijay Mahadevan   vmoab->is_global_vec = is_global_vec;
498bb8f3634SVijay Mahadevan   vmoab->tag_size=dmmoab->bs;
499032b8ab6SVijay Mahadevan 
500bb8f3634SVijay Mahadevan   if (vmoab->is_native_vec) {
501fc418013SVijay Mahadevan 
502bb8f3634SVijay Mahadevan     /* Create the PETSc Vector directly and attach our functions accordingly */
503bb8f3634SVijay Mahadevan     if (!is_global_vec) {
504bb8f3634SVijay Mahadevan       /* This is an MPI Vector with ghosted padding */
505addae81cSVijay Mahadevan       ierr = VecCreateGhostBlock(pcomm->comm(),dmmoab->bs,dmmoab->numFields*dmmoab->nloc,
506addae81cSVijay Mahadevan                                  dmmoab->numFields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],vec);CHKERRQ(ierr);
507bb8f3634SVijay Mahadevan     }
508bb8f3634SVijay Mahadevan     else {
509bb8f3634SVijay Mahadevan       /* This is an MPI/SEQ Vector */
510bb8f3634SVijay Mahadevan       ierr = VecCreate(pcomm->comm(),vec);CHKERRQ(ierr);
511addae81cSVijay Mahadevan       ierr = VecSetSizes(*vec,dmmoab->numFields*dmmoab->nloc,PETSC_DECIDE);CHKERRQ(ierr);
512bb8f3634SVijay Mahadevan       ierr = VecSetBlockSize(*vec,dmmoab->bs);CHKERRQ(ierr);
513bb8f3634SVijay Mahadevan       ierr = VecSetType(*vec, VECMPI);CHKERRQ(ierr);
514bb8f3634SVijay Mahadevan     }
515bb8f3634SVijay Mahadevan   }
516bb8f3634SVijay Mahadevan   else {
517db66d124SVijay Mahadevan     /* Call tag_iterate. This will cause MOAB to allocate memory for the
518db66d124SVijay Mahadevan        tag data if it hasn't already happened */
519032b8ab6SVijay Mahadevan     merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr);
520032b8ab6SVijay Mahadevan 
521bb8f3634SVijay Mahadevan     /* set the reference for vector range */
522bb8f3634SVijay Mahadevan     vmoab->tag_range = new moab::Range(*range);
523addae81cSVijay Mahadevan     merr = mbiface->tag_get_length(tag,dmmoab->numFields);MBERRNM(merr);
524032b8ab6SVijay Mahadevan 
525db66d124SVijay Mahadevan     /* Create the PETSc Vector
526db66d124SVijay Mahadevan       Query MOAB mesh to check if there are any ghosted entities
527032b8ab6SVijay Mahadevan         -> if we do, create a ghosted vector to map correctly to the same layout
528032b8ab6SVijay Mahadevan         -> else, create a non-ghosted parallel vector */
529bb8f3634SVijay Mahadevan     if (!is_global_vec) {
530db66d124SVijay Mahadevan       /* This is an MPI Vector with ghosted padding */
531addae81cSVijay Mahadevan       ierr = VecCreateGhostBlockWithArray(pcomm->comm(),dmmoab->bs,dmmoab->numFields*dmmoab->nloc,
532addae81cSVijay Mahadevan                                 dmmoab->numFields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],data_ptr,vec);CHKERRQ(ierr);
533032b8ab6SVijay Mahadevan     }
534032b8ab6SVijay Mahadevan     else {
535bb8f3634SVijay Mahadevan       /* This is an MPI Vector without ghosted padding */
536addae81cSVijay Mahadevan       ierr = VecCreateMPIWithArray(pcomm->comm(),dmmoab->bs,dmmoab->numFields*range->size(),
537bb8f3634SVijay Mahadevan                                 PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr);
538032b8ab6SVijay Mahadevan     }
539bb8f3634SVijay Mahadevan   }
540bb8f3634SVijay Mahadevan   ierr = VecSetFromOptions(*vec);CHKERRQ(ierr);
541032b8ab6SVijay Mahadevan 
542f6829af0SVijay Mahadevan   /* create a container and store the internal MOAB data for faster access based on Entities etc */
543032b8ab6SVijay Mahadevan   PetscContainer moabdata;
544bb8f3634SVijay Mahadevan   ierr = PetscContainerCreate(PETSC_COMM_WORLD,&moabdata);CHKERRQ(ierr);
545032b8ab6SVijay Mahadevan   ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr);
546f6829af0SVijay Mahadevan   ierr = PetscContainerSetUserDestroy(moabdata,DMVecUserDestroy_Moab);CHKERRQ(ierr);
547032b8ab6SVijay Mahadevan   ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr);
548f6829af0SVijay Mahadevan   (*vec)->ops->duplicate = DMVecDuplicate_Moab;
549032b8ab6SVijay Mahadevan   ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr);
550032b8ab6SVijay Mahadevan 
551db66d124SVijay Mahadevan   /* Vector created, manually set local to global mapping */
5526d9eb265SVijay Mahadevan   if (dmmoab->ltog_map) {
553bb8f3634SVijay Mahadevan     ierr = VecSetLocalToGlobalMapping(*vec,dmmoab->ltog_map);CHKERRQ(ierr);
5546d9eb265SVijay Mahadevan   }
555032b8ab6SVijay Mahadevan 
556032b8ab6SVijay Mahadevan   /* set the DM reference to the vector */
557032b8ab6SVijay Mahadevan   ierr = VecSetDM(*vec, dm);CHKERRQ(ierr);
558032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
559032b8ab6SVijay Mahadevan }
560032b8ab6SVijay Mahadevan 
561032b8ab6SVijay Mahadevan 
562032b8ab6SVijay Mahadevan #undef __FUNCT__
563f6829af0SVijay Mahadevan #define __FUNCT__ "DMVecCreateTagName_Moab_Private"
564f6829af0SVijay Mahadevan /*  DMVecCreateTagName_Moab_Private
565032b8ab6SVijay Mahadevan  *
566032b8ab6SVijay Mahadevan  *  Creates a unique tag name that will be shared across processes. If
567032b8ab6SVijay Mahadevan  *  pcomm is NULL, then this is a serial vector. A unique tag name
568032b8ab6SVijay Mahadevan  *  will be returned in tag_name in either case.
569032b8ab6SVijay Mahadevan  *
570032b8ab6SVijay Mahadevan  *  The tag names have the format _PETSC_VEC_N where N is some integer.
571032b8ab6SVijay Mahadevan  *
572032b8ab6SVijay Mahadevan  *  NOTE: The tag_name is allocated in this routine; The user needs to free
573032b8ab6SVijay Mahadevan  *        the character array.
574032b8ab6SVijay Mahadevan  */
575f6829af0SVijay Mahadevan PetscErrorCode DMVecCreateTagName_Moab_Private(moab::ParallelComm *pcomm,char** tag_name)
576032b8ab6SVijay Mahadevan {
577032b8ab6SVijay Mahadevan   moab::ErrorCode mberr;
578032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
579032b8ab6SVijay Mahadevan   PetscInt        n,global_n;
580032b8ab6SVijay Mahadevan   moab::Tag indexTag;
581032b8ab6SVijay Mahadevan 
582032b8ab6SVijay Mahadevan   PetscFunctionBegin;
583032b8ab6SVijay Mahadevan   const char*       PVEC_PREFIX      = "__PETSC_VEC_";
584032b8ab6SVijay Mahadevan   ierr = PetscMalloc(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr);
585032b8ab6SVijay Mahadevan 
586efd17f3eSVijay Mahadevan   /* Check to see if there are any PETSc vectors defined */
587032b8ab6SVijay Mahadevan   moab::Interface  *mbiface = pcomm->get_moab();
588032b8ab6SVijay Mahadevan   moab::EntityHandle rootset = mbiface->get_root_set();
589032b8ab6SVijay Mahadevan 
590efd17f3eSVijay Mahadevan   /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */
591032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag,
592032b8ab6SVijay Mahadevan                                   moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr);
593032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n);
594032b8ab6SVijay Mahadevan   if (mberr == moab::MB_TAG_NOT_FOUND) n=0;  /* this is the first temporary vector */
595032b8ab6SVijay Mahadevan   else MBERRNM(mberr);
596032b8ab6SVijay Mahadevan 
597032b8ab6SVijay Mahadevan   /* increment the new value of n */
598032b8ab6SVijay Mahadevan   ++n;
599032b8ab6SVijay Mahadevan 
600efd17f3eSVijay Mahadevan   /* Make sure that n is consistent across all processes */
601032b8ab6SVijay Mahadevan   ierr = MPI_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr);
602032b8ab6SVijay Mahadevan 
603efd17f3eSVijay Mahadevan   /* Set the new name accordingly and return */
604032b8ab6SVijay Mahadevan   ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr);
605032b8ab6SVijay Mahadevan   mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr);
606032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
607032b8ab6SVijay Mahadevan }
608032b8ab6SVijay Mahadevan 
609032b8ab6SVijay Mahadevan 
610032b8ab6SVijay Mahadevan #undef __FUNCT__
611f6829af0SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab"
612f6829af0SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec)
613f6829af0SVijay Mahadevan {
614f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
615f6829af0SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
616f6829af0SVijay Mahadevan 
617f6829af0SVijay Mahadevan   PetscFunctionBegin;
618f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
619f6829af0SVijay Mahadevan   PetscValidPointer(gvec,2);
620f6829af0SVijay Mahadevan   ierr = DMCreateVector_Moab_Private(dm,PETSC_NULL,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr);
621f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
622f6829af0SVijay Mahadevan }
623f6829af0SVijay Mahadevan 
624f6829af0SVijay Mahadevan 
625f6829af0SVijay Mahadevan #undef __FUNCT__
626f6829af0SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab"
627f6829af0SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec)
628f6829af0SVijay Mahadevan {
629f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
630f6829af0SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
631f6829af0SVijay Mahadevan 
632f6829af0SVijay Mahadevan   PetscFunctionBegin;
633f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
634f6829af0SVijay Mahadevan   PetscValidPointer(lvec,2);
635f6829af0SVijay Mahadevan   ierr = DMCreateVector_Moab_Private(dm,PETSC_NULL,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr);
636f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
637f6829af0SVijay Mahadevan }
638f6829af0SVijay Mahadevan 
639f6829af0SVijay Mahadevan 
640f6829af0SVijay Mahadevan #undef __FUNCT__
641f6829af0SVijay Mahadevan #define __FUNCT__ "DMVecDuplicate_Moab"
642f6829af0SVijay Mahadevan PetscErrorCode DMVecDuplicate_Moab(Vec x,Vec *y)
643f6829af0SVijay Mahadevan {
644f6829af0SVijay Mahadevan   PetscErrorCode ierr;
645f6829af0SVijay Mahadevan   DM             dm;
646f6829af0SVijay Mahadevan   PetscContainer  moabdata;
647f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab;
648f6829af0SVijay Mahadevan 
649f6829af0SVijay Mahadevan   PetscFunctionBegin;
650f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(x,VEC_CLASSID,1);
651f6829af0SVijay Mahadevan   PetscValidPointer(y,2);
652f6829af0SVijay Mahadevan 
653f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
654f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
655f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
656f6829af0SVijay Mahadevan 
657f6829af0SVijay Mahadevan   ierr = VecGetDM(x, &dm);CHKERRQ(ierr);
658f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
659f6829af0SVijay Mahadevan 
660f6829af0SVijay Mahadevan   ierr = DMCreateVector_Moab_Private(dm,PETSC_NULL,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr);
661f6829af0SVijay Mahadevan   ierr = VecSetDM(*y, dm);CHKERRQ(ierr);
662f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
663f6829af0SVijay Mahadevan }
664f6829af0SVijay Mahadevan 
665f6829af0SVijay Mahadevan 
666f6829af0SVijay Mahadevan #undef __FUNCT__
667f6829af0SVijay Mahadevan #define __FUNCT__ "DMVecUserDestroy_Moab"
668f6829af0SVijay Mahadevan PetscErrorCode DMVecUserDestroy_Moab(void *user)
669032b8ab6SVijay Mahadevan {
670efd17f3eSVijay Mahadevan   Vec_MOAB        *vmoab=(Vec_MOAB*)user;
671032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
672032b8ab6SVijay Mahadevan   moab::ErrorCode merr;
673032b8ab6SVijay Mahadevan 
674032b8ab6SVijay Mahadevan   PetscFunctionBegin;
675032b8ab6SVijay Mahadevan   if(vmoab->new_tag && vmoab->tag) {
676efd17f3eSVijay Mahadevan     /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */
677032b8ab6SVijay Mahadevan     merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr);
678032b8ab6SVijay Mahadevan   }
679032b8ab6SVijay Mahadevan   delete vmoab->tag_range;
680032b8ab6SVijay Mahadevan   vmoab->tag = PETSC_NULL;
681032b8ab6SVijay Mahadevan   vmoab->mbiface = PETSC_NULL;
682032b8ab6SVijay Mahadevan   vmoab->pcomm = PETSC_NULL;
683032b8ab6SVijay Mahadevan   ierr = PetscFree(vmoab);CHKERRQ(ierr);
684032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
685032b8ab6SVijay Mahadevan }
686032b8ab6SVijay Mahadevan 
687efd17f3eSVijay Mahadevan 
688efd17f3eSVijay Mahadevan #undef __FUNCT__
689db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab"
690db66d124SVijay Mahadevan PetscErrorCode  DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l)
691db66d124SVijay Mahadevan {
692db66d124SVijay Mahadevan   PetscErrorCode    ierr;
693db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
694db66d124SVijay Mahadevan 
695db66d124SVijay Mahadevan   PetscFunctionBegin;
6966d9eb265SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_REVERSE);CHKERRQ(ierr);
697db66d124SVijay Mahadevan   PetscFunctionReturn(0);
698db66d124SVijay Mahadevan }
699db66d124SVijay Mahadevan 
700db66d124SVijay Mahadevan 
701db66d124SVijay Mahadevan #undef __FUNCT__
702db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab"
703db66d124SVijay Mahadevan PetscErrorCode  DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l)
704db66d124SVijay Mahadevan {
705db66d124SVijay Mahadevan   PetscErrorCode    ierr;
706db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
707db66d124SVijay Mahadevan 
708db66d124SVijay Mahadevan   PetscFunctionBegin;
7096d9eb265SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_REVERSE);CHKERRQ(ierr);
710db66d124SVijay Mahadevan   PetscFunctionReturn(0);
711db66d124SVijay Mahadevan }
712db66d124SVijay Mahadevan 
713db66d124SVijay Mahadevan 
714db66d124SVijay Mahadevan #undef __FUNCT__
715db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab"
716db66d124SVijay Mahadevan PetscErrorCode  DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g)
717db66d124SVijay Mahadevan {
718db66d124SVijay Mahadevan   PetscErrorCode    ierr;
719db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
720db66d124SVijay Mahadevan 
721db66d124SVijay Mahadevan   PetscFunctionBegin;
7226d9eb265SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_FORWARD);CHKERRQ(ierr);
723db66d124SVijay Mahadevan   PetscFunctionReturn(0);
724db66d124SVijay Mahadevan }
725db66d124SVijay Mahadevan 
726db66d124SVijay Mahadevan 
727db66d124SVijay Mahadevan #undef __FUNCT__
728db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab"
729db66d124SVijay Mahadevan PetscErrorCode  DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g)
730db66d124SVijay Mahadevan {
731db66d124SVijay Mahadevan   PetscErrorCode    ierr;
732db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
733db66d124SVijay Mahadevan 
734db66d124SVijay Mahadevan   PetscFunctionBegin;
7356d9eb265SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_FORWARD);CHKERRQ(ierr);
736db66d124SVijay Mahadevan   PetscFunctionReturn(0);
737db66d124SVijay Mahadevan }
738db66d124SVijay Mahadevan 
739