xref: /petsc/src/dm/impls/moab/dmmbvec.cxx (revision 9c36898578a91ec77675797fbfe6e623f4be5691)
1af0996ceSBarry Smith #include <petsc/private/dmmbimpl.h> /*I  "petscdmmoab.h"   I*/
2af0996ceSBarry Smith #include <petsc/private/vecimpl.h>
3032b8ab6SVijay Mahadevan 
4032b8ab6SVijay Mahadevan #include <petscdmmoab.h>
5032b8ab6SVijay Mahadevan #include <MBTagConventions.hpp>
6032b8ab6SVijay Mahadevan 
7*9c368985SVijay Mahadevan #define USE_NATIVE_PETSCVEC
8*9c368985SVijay Mahadevan 
9f6829af0SVijay Mahadevan /* declare some private DMMoab specific overrides */
10f6829af0SVijay Mahadevan static PetscErrorCode DMCreateVector_Moab_Private(DM dm,moab::Tag tag,const moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec);
11f6829af0SVijay Mahadevan static PetscErrorCode DMVecUserDestroy_Moab(void *user);
12f6829af0SVijay Mahadevan static PetscErrorCode DMVecDuplicate_Moab(Vec x,Vec *y);
139daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
149daf19fdSVijay Mahadevan static PetscErrorCode DMVecCreateTagName_Moab_Private(moab::Interface *mbiface,moab::ParallelComm *pcomm,char** tag_name);
159daf19fdSVijay Mahadevan #else
169daf19fdSVijay Mahadevan static PetscErrorCode DMVecCreateTagName_Moab_Private(moab::Interface *mbiface,char** tag_name);
179daf19fdSVijay Mahadevan #endif
18032b8ab6SVijay Mahadevan 
19b8ecf6d3SVijay Mahadevan /*@C
20f6829af0SVijay Mahadevan   DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities
21f6829af0SVijay Mahadevan 
22f6829af0SVijay Mahadevan   Collective on MPI_Comm
23f6829af0SVijay Mahadevan 
24f6829af0SVijay Mahadevan   Input Parameter:
25b8ecf6d3SVijay Mahadevan + dm              - The DMMoab object being set
26f6829af0SVijay Mahadevan . tag             - If non-zero, block size will be taken from the tag size
27f6829af0SVijay Mahadevan . range           - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab
28f6829af0SVijay Mahadevan . is_global_vec   - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one
29f6829af0SVijay Mahadevan . destroy_tag     - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB
30f6829af0SVijay Mahadevan 
31f6829af0SVijay Mahadevan   Output Parameter:
32f6829af0SVijay Mahadevan . vec             - The created vector
33f6829af0SVijay Mahadevan 
34f6829af0SVijay Mahadevan   Level: beginner
35f6829af0SVijay Mahadevan 
36b8ecf6d3SVijay Mahadevan .keywords: Vec, create
37b8ecf6d3SVijay Mahadevan 
38b8ecf6d3SVijay Mahadevan .seealso: VecCreate()
39f6829af0SVijay Mahadevan @*/
40f6829af0SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm, moab::Tag tag, const moab::Range* range, PetscBool is_global_vec, PetscBool destroy_tag, Vec *vec)
41f6829af0SVijay Mahadevan {
42f6829af0SVijay Mahadevan   PetscErrorCode     ierr;
43f6829af0SVijay Mahadevan 
44f6829af0SVijay Mahadevan   PetscFunctionBegin;
45f6829af0SVijay Mahadevan   if (!tag && (!range || range->empty())) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag and range cannot be null.");
46f6829af0SVijay Mahadevan 
47f6829af0SVijay Mahadevan   ierr = DMCreateVector_Moab_Private(dm, tag, range, is_global_vec, destroy_tag, vec); CHKERRQ(ierr);
48f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
49f6829af0SVijay Mahadevan }
50f6829af0SVijay Mahadevan 
51f6829af0SVijay Mahadevan 
52b8ecf6d3SVijay Mahadevan /*@C
53f6829af0SVijay Mahadevan   DMMoabGetVecTag - Get the MOAB tag associated with this Vec
54f6829af0SVijay Mahadevan 
55f6829af0SVijay Mahadevan   Input Parameter:
56f6829af0SVijay Mahadevan . vec - Vec being queried
57f6829af0SVijay Mahadevan 
58f6829af0SVijay Mahadevan   Output Parameter:
59f6829af0SVijay Mahadevan . tag - Tag associated with this Vec. NULL if vec is a native PETSc Vec.
60f6829af0SVijay Mahadevan 
61f6829af0SVijay Mahadevan   Level: beginner
62f6829af0SVijay Mahadevan 
63b8ecf6d3SVijay Mahadevan .keywords: DMMoab, MOAB tag
64b8ecf6d3SVijay Mahadevan 
65b8ecf6d3SVijay Mahadevan .seealso: DMMoabCreateVector(), DMMoabGetVecRange()
66f6829af0SVijay Mahadevan @*/
67f6829af0SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec, moab::Tag *tag)
68f6829af0SVijay Mahadevan {
69f6829af0SVijay Mahadevan   PetscContainer  moabdata;
70f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab;
71f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
72f6829af0SVijay Mahadevan 
73f6829af0SVijay Mahadevan   PetscFunctionBegin;
74f6829af0SVijay Mahadevan   PetscValidPointer(tag, 2);
75f6829af0SVijay Mahadevan 
76f6829af0SVijay Mahadevan   /* Get the MOAB private data */
77f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
78f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab); CHKERRQ(ierr);
79f6829af0SVijay Mahadevan 
80f6829af0SVijay Mahadevan   *tag = vmoab->tag;
81f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
82f6829af0SVijay Mahadevan }
83f6829af0SVijay Mahadevan 
84f6829af0SVijay Mahadevan 
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 
118b8ecf6d3SVijay Mahadevan /*@C
119f6829af0SVijay 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
120f6829af0SVijay Mahadevan 
121f6829af0SVijay Mahadevan   Collective on MPI_Comm
122f6829af0SVijay Mahadevan 
123f6829af0SVijay Mahadevan   Input Parameter:
124f6829af0SVijay Mahadevan . dm              - The DMMoab object being set
125f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
126f6829af0SVijay Mahadevan 
127f6829af0SVijay Mahadevan   Output Parameter:
128f6829af0SVijay Mahadevan . array           - The local data array
129f6829af0SVijay Mahadevan 
130f6829af0SVijay Mahadevan   Level: intermediate
131f6829af0SVijay Mahadevan 
132f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
133f6829af0SVijay Mahadevan 
134f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
135f6829af0SVijay Mahadevan @*/
136f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecGetArray(DM dm, Vec vec, void* array)
137f6829af0SVijay Mahadevan {
138f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
139f6829af0SVijay Mahadevan   moab::ErrorCode merr;
140f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
141f6829af0SVijay Mahadevan   PetscInt        count, i, f;
142f6829af0SVijay Mahadevan   moab::Tag       vtag;
143f6829af0SVijay Mahadevan   PetscScalar     **varray;
144f6829af0SVijay Mahadevan   PetscScalar     *marray;
145f6829af0SVijay Mahadevan   PetscContainer  moabdata;
146f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab, *xmoab;
147f6829af0SVijay Mahadevan 
148f6829af0SVijay Mahadevan   PetscFunctionBegin;
149f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
150f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec, VEC_CLASSID, 2);
151f6829af0SVijay Mahadevan   PetscValidPointer(array, 3);
152f6829af0SVijay Mahadevan   dmmoab = (DM_Moab*)dm->data;
153f6829af0SVijay Mahadevan 
154f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
155f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
156f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab); CHKERRQ(ierr);
157f6829af0SVijay Mahadevan 
158f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
159f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
160f6829af0SVijay Mahadevan 
161f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
162f6829af0SVijay Mahadevan 
163f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
164f6829af0SVijay Mahadevan     ierr = DMGetLocalVector(dm, &vmoab->local); CHKERRQ(ierr);
165f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalBegin(dm, vec, INSERT_VALUES, vmoab->local); CHKERRQ(ierr);
166f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalEnd(dm, vec, INSERT_VALUES, vmoab->local); CHKERRQ(ierr);
167f6829af0SVijay Mahadevan 
168f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
169f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
170f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab); CHKERRQ(ierr);
171f6829af0SVijay Mahadevan 
172f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
173f6829af0SVijay Mahadevan     ierr = VecGhostGetLocalForm(vmoab->local, &xmoab->local); CHKERRQ(ierr);
174f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local, INSERT_VALUES, SCATTER_FORWARD); CHKERRQ(ierr);
175f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local, INSERT_VALUES, SCATTER_FORWARD); CHKERRQ(ierr);
176f6829af0SVijay Mahadevan 
177f6829af0SVijay Mahadevan     ierr = VecGetArray(xmoab->local, varray); CHKERRQ(ierr);
178f6829af0SVijay Mahadevan   }
179f6829af0SVijay Mahadevan   else {
180f6829af0SVijay Mahadevan 
181f6829af0SVijay Mahadevan     /* Get the MOAB private data */
182f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec, &vtag); CHKERRQ(ierr);
183f6829af0SVijay Mahadevan 
1849daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
185f6829af0SVijay Mahadevan     /* exchange the data into ghost cells first */
186f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->exchange_tags(vtag, *dmmoab->vlocal); MBERRNM(merr);
1879daf19fdSVijay Mahadevan #endif
188f6829af0SVijay Mahadevan 
1897ae5e5b6SVijay 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++)
1983f1c6e43SVijay Mahadevan         (*varray)[dmmoab->lidmap[(PetscInt)*iter - dmmoab->seqstart]*dmmoab->numFields + f] = marray[i];
1993f1c6e43SVijay Mahadevan       //(*varray)[dmmoab->llmap[dmmoab->lidmap[((PetscInt)*iter-dmmoab->seqstart)]*dmmoab->numFields+f]]=marray[i];
200f6829af0SVijay Mahadevan     }
201f6829af0SVijay Mahadevan   }
202f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
203f6829af0SVijay Mahadevan }
204f6829af0SVijay Mahadevan 
205f6829af0SVijay Mahadevan 
2063f1c6e43SVijay Mahadevan #undef __FUNCT__
2073f1c6e43SVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray"
208b8ecf6d3SVijay Mahadevan /*@C
209f6829af0SVijay Mahadevan   DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray
210f6829af0SVijay Mahadevan 
211f6829af0SVijay Mahadevan   Collective on MPI_Comm
212f6829af0SVijay Mahadevan 
213f6829af0SVijay Mahadevan   Input Parameter:
214f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
215f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
216b8ecf6d3SVijay Mahadevan . array           - The local data array
217f6829af0SVijay Mahadevan 
218f6829af0SVijay Mahadevan   Level: intermediate
219f6829af0SVijay Mahadevan 
220f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
221f6829af0SVijay Mahadevan 
222f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
223f6829af0SVijay Mahadevan @*/
224f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArray(DM dm, Vec vec, void* array)
225f6829af0SVijay Mahadevan {
226f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
227f6829af0SVijay Mahadevan   moab::ErrorCode merr;
228f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
229f6829af0SVijay Mahadevan   moab::Tag       vtag;
230f6829af0SVijay Mahadevan   PetscInt        count, i, f;
231f6829af0SVijay Mahadevan   PetscScalar     **varray;
232f6829af0SVijay Mahadevan   PetscScalar     *marray;
233f6829af0SVijay Mahadevan   PetscContainer  moabdata;
234f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab, *xmoab;
235f6829af0SVijay Mahadevan 
236f6829af0SVijay Mahadevan   PetscFunctionBegin;
237f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
238f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec, VEC_CLASSID, 2);
239f6829af0SVijay Mahadevan   PetscValidPointer(array, 3);
240f6829af0SVijay Mahadevan   dmmoab = (DM_Moab*)dm->data;
241f6829af0SVijay Mahadevan 
242f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
243f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
244f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab); CHKERRQ(ierr);
245f6829af0SVijay Mahadevan 
246f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
247f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
248f6829af0SVijay Mahadevan 
249f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
250f6829af0SVijay Mahadevan 
251f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
252f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
253f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab); CHKERRQ(ierr);
254f6829af0SVijay Mahadevan 
255f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
256f6829af0SVijay Mahadevan     ierr = VecRestoreArray(xmoab->local, varray); CHKERRQ(ierr);
257f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local, ADD_VALUES, SCATTER_REVERSE); CHKERRQ(ierr);
258f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local, ADD_VALUES, SCATTER_REVERSE); CHKERRQ(ierr);
259f6829af0SVijay Mahadevan     ierr = VecGhostRestoreLocalForm(vmoab->local, &xmoab->local); CHKERRQ(ierr);
260f6829af0SVijay Mahadevan 
261f6829af0SVijay Mahadevan     /* restore local pieces */
262f6829af0SVijay Mahadevan     ierr = DMLocalToGlobalBegin(dm, vmoab->local, INSERT_VALUES, vec); CHKERRQ(ierr);
263f6829af0SVijay Mahadevan     ierr = DMLocalToGlobalEnd(dm, vmoab->local, INSERT_VALUES, vec); CHKERRQ(ierr);
264f6829af0SVijay Mahadevan     ierr = DMRestoreLocalVector(dm, &vmoab->local); CHKERRQ(ierr);
265f6829af0SVijay Mahadevan   }
266f6829af0SVijay Mahadevan   else {
267f6829af0SVijay Mahadevan 
268f6829af0SVijay Mahadevan     /* Get the MOAB private data */
269f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec, &vtag); CHKERRQ(ierr);
270f6829af0SVijay Mahadevan 
271f6829af0SVijay Mahadevan     /* Get the array data for local entities */
272f6829af0SVijay Mahadevan     merr = dmmoab->mbiface->tag_iterate(vtag, dmmoab->vlocal->begin(), dmmoab->vlocal->end(), count, reinterpret_cast<void*&>(marray), false); MBERRNM(merr);
273f6829af0SVijay 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);
274f6829af0SVijay Mahadevan 
275f6829af0SVijay Mahadevan     i = 0;
276f6829af0SVijay Mahadevan     for (moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) {
277f6829af0SVijay Mahadevan       for (f = 0; f < dmmoab->numFields; f++, i++)
2783f1c6e43SVijay Mahadevan         marray[i] = (*varray)[dmmoab->lidmap[(PetscInt) * iter - dmmoab->seqstart] * dmmoab->numFields + f];
2793f1c6e43SVijay Mahadevan       //marray[i] = (*varray)[dmmoab->llmap[dmmoab->lidmap[((PetscInt)*iter-dmmoab->seqstart)]*dmmoab->numFields+f]];
280f6829af0SVijay Mahadevan     }
281f6829af0SVijay Mahadevan 
2829daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
283f6829af0SVijay Mahadevan     /* reduce the tags correctly -> should probably let the user choose how to reduce in the future
284f6829af0SVijay Mahadevan       For all FEM residual based assembly calculations, MPI_SUM should serve well */
285f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->reduce_tags(vtag, MPI_SUM, *dmmoab->vlocal); MBERRV(dmmoab->mbiface, merr);
2869daf19fdSVijay Mahadevan #endif
287f6829af0SVijay Mahadevan     ierr = PetscFree(*varray); CHKERRQ(ierr);
288f6829af0SVijay Mahadevan   }
289f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
290f6829af0SVijay Mahadevan }
291f6829af0SVijay Mahadevan 
292b8ecf6d3SVijay Mahadevan /*@C
293f6829af0SVijay 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
294f6829af0SVijay Mahadevan 
295f6829af0SVijay Mahadevan   Collective on MPI_Comm
296f6829af0SVijay Mahadevan 
297f6829af0SVijay Mahadevan   Input Parameter:
298f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
299f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
300f6829af0SVijay Mahadevan 
301f6829af0SVijay Mahadevan   Output Parameter:
302f6829af0SVijay Mahadevan . array           - The local data array
303f6829af0SVijay Mahadevan 
304f6829af0SVijay Mahadevan   Level: intermediate
305f6829af0SVijay Mahadevan 
306f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
307f6829af0SVijay Mahadevan 
308f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
309f6829af0SVijay Mahadevan @*/
310f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecGetArrayRead(DM dm, Vec vec, void* array)
311f6829af0SVijay Mahadevan {
312f6829af0SVijay Mahadevan   DM_Moab        *dmmoab;
313f6829af0SVijay Mahadevan   moab::ErrorCode merr;
314f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
315f6829af0SVijay Mahadevan   PetscInt        count, i, f;
316f6829af0SVijay Mahadevan   moab::Tag       vtag;
317f6829af0SVijay Mahadevan   PetscScalar     **varray;
318f6829af0SVijay Mahadevan   PetscScalar     *marray;
319f6829af0SVijay Mahadevan   PetscContainer  moabdata;
320f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab, *xmoab;
321f6829af0SVijay Mahadevan 
322f6829af0SVijay Mahadevan   PetscFunctionBegin;
323f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
324f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec, VEC_CLASSID, 2);
325f6829af0SVijay Mahadevan   PetscValidPointer(array, 3);
326f6829af0SVijay Mahadevan   dmmoab = (DM_Moab*)dm->data;
327f6829af0SVijay Mahadevan 
328f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
329f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
330f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab); CHKERRQ(ierr);
331f6829af0SVijay Mahadevan 
332f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
333f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
334f6829af0SVijay Mahadevan 
335f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
336f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
337f6829af0SVijay Mahadevan     ierr = DMGetLocalVector(dm, &vmoab->local); CHKERRQ(ierr);
338f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalBegin(dm, vec, INSERT_VALUES, vmoab->local); CHKERRQ(ierr);
339f6829af0SVijay Mahadevan     ierr = DMGlobalToLocalEnd(dm, vec, INSERT_VALUES, vmoab->local); CHKERRQ(ierr);
340f6829af0SVijay Mahadevan 
341f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
342f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
343f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab); CHKERRQ(ierr);
344f6829af0SVijay Mahadevan 
345f6829af0SVijay Mahadevan     /* get the local representation of the arrays from Vectors */
346f6829af0SVijay Mahadevan     ierr = VecGhostGetLocalForm(vmoab->local, &xmoab->local); CHKERRQ(ierr);
347f6829af0SVijay Mahadevan     ierr = VecGhostUpdateBegin(vmoab->local, INSERT_VALUES, SCATTER_FORWARD); CHKERRQ(ierr);
348f6829af0SVijay Mahadevan     ierr = VecGhostUpdateEnd(vmoab->local, INSERT_VALUES, SCATTER_FORWARD); CHKERRQ(ierr);
349f6829af0SVijay Mahadevan     ierr = VecGetArray(xmoab->local, varray); CHKERRQ(ierr);
350f6829af0SVijay Mahadevan   }
351f6829af0SVijay Mahadevan   else {
352f6829af0SVijay Mahadevan     /* Get the MOAB private data */
353f6829af0SVijay Mahadevan     ierr = DMMoabGetVecTag(vec, &vtag); CHKERRQ(ierr);
354f6829af0SVijay Mahadevan 
3559daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
356f6829af0SVijay Mahadevan     /* exchange the data into ghost cells first */
357f6829af0SVijay Mahadevan     merr = dmmoab->pcomm->exchange_tags(vtag, *dmmoab->vlocal); MBERRNM(merr);
3589daf19fdSVijay Mahadevan #endif
3597ae5e5b6SVijay Mahadevan     ierr = PetscMalloc1((dmmoab->nloc + dmmoab->nghost) * dmmoab->numFields, varray); CHKERRQ(ierr);
360f6829af0SVijay Mahadevan 
361f6829af0SVijay Mahadevan     /* Get the array data for local entities */
362f6829af0SVijay Mahadevan     merr = dmmoab->mbiface->tag_iterate(vtag, dmmoab->vlocal->begin(), dmmoab->vlocal->end(), count, reinterpret_cast<void*&>(marray), false); MBERRNM(merr);
363f6829af0SVijay 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);
364f6829af0SVijay Mahadevan 
365f6829af0SVijay Mahadevan     i = 0;
366f6829af0SVijay Mahadevan     for (moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) {
367f6829af0SVijay Mahadevan       for (f = 0; f < dmmoab->numFields; f++, i++)
3683f1c6e43SVijay Mahadevan         (*varray)[dmmoab->lidmap[(PetscInt)*iter - dmmoab->seqstart]*dmmoab->numFields + f] = marray[i];
3693f1c6e43SVijay Mahadevan       //(*varray)[dmmoab->llmap[dmmoab->lidmap[((PetscInt)*iter-dmmoab->seqstart)]*dmmoab->numFields+f]]=marray[i];
370f6829af0SVijay Mahadevan     }
371f6829af0SVijay Mahadevan   }
372f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
373f6829af0SVijay Mahadevan }
374f6829af0SVijay Mahadevan 
375f6829af0SVijay Mahadevan 
376b8ecf6d3SVijay Mahadevan /*@C
377f6829af0SVijay Mahadevan   DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray
378f6829af0SVijay Mahadevan 
379f6829af0SVijay Mahadevan   Collective on MPI_Comm
380f6829af0SVijay Mahadevan 
381f6829af0SVijay Mahadevan   Input Parameter:
382f6829af0SVijay Mahadevan + dm              - The DMMoab object being set
383f6829af0SVijay Mahadevan . vec             - The Vector whose underlying data is requested
384b8ecf6d3SVijay Mahadevan . array           - The local data array
385f6829af0SVijay Mahadevan 
386f6829af0SVijay Mahadevan   Level: intermediate
387f6829af0SVijay Mahadevan 
388f6829af0SVijay Mahadevan .keywords: MOAB, distributed array
389f6829af0SVijay Mahadevan 
390f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
391f6829af0SVijay Mahadevan @*/
392f6829af0SVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArrayRead(DM dm, Vec vec, void* array)
393f6829af0SVijay Mahadevan {
394f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
395f6829af0SVijay Mahadevan   PetscScalar     **varray;
396f6829af0SVijay Mahadevan   PetscContainer  moabdata;
397f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab, *xmoab;
398f6829af0SVijay Mahadevan 
399f6829af0SVijay Mahadevan   PetscFunctionBegin;
400f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
401f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(vec, VEC_CLASSID, 2);
402f6829af0SVijay Mahadevan   PetscValidPointer(array, 3);
403f6829af0SVijay Mahadevan 
404f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
405f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
406f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab); CHKERRQ(ierr);
407f6829af0SVijay Mahadevan 
408f6829af0SVijay Mahadevan   /* Get the real scalar array handle */
409f6829af0SVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
410f6829af0SVijay Mahadevan 
411f6829af0SVijay Mahadevan   if (vmoab->is_native_vec) {
412f6829af0SVijay Mahadevan     /* Get the Vec_MOAB struct for the original vector */
413f6829af0SVijay Mahadevan     ierr = PetscObjectQuery((PetscObject)vmoab->local, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
414f6829af0SVijay Mahadevan     ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab); CHKERRQ(ierr);
415f6829af0SVijay Mahadevan 
416f6829af0SVijay Mahadevan     /* restore the local representation of the arrays from Vectors */
417f6829af0SVijay Mahadevan     ierr = VecRestoreArray(xmoab->local, varray); CHKERRQ(ierr);
418f6829af0SVijay Mahadevan     ierr = VecGhostRestoreLocalForm(vmoab->local, &xmoab->local); CHKERRQ(ierr);
419f6829af0SVijay Mahadevan 
420f6829af0SVijay Mahadevan     /* restore local pieces */
421f6829af0SVijay Mahadevan     ierr = DMRestoreLocalVector(dm, &vmoab->local); CHKERRQ(ierr);
422f6829af0SVijay Mahadevan   }
423f6829af0SVijay Mahadevan   else {
424f6829af0SVijay Mahadevan     /* Nothing to do but just free the memory allocated before */
425f6829af0SVijay Mahadevan     ierr = PetscFree(*varray); CHKERRQ(ierr);
426f6829af0SVijay Mahadevan 
427f6829af0SVijay Mahadevan   }
428f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
429f6829af0SVijay Mahadevan }
430f6829af0SVijay Mahadevan 
431f6829af0SVijay Mahadevan 
43263d025dbSVijay Mahadevan #undef __FUNCT__
43363d025dbSVijay Mahadevan #define __FUNCT__ "DMCreateVector_Moab_Private"
434f6829af0SVijay Mahadevan PetscErrorCode DMCreateVector_Moab_Private(DM dm, moab::Tag tag, const moab::Range* userrange, PetscBool is_global_vec, PetscBool destroy_tag, Vec *vec)
435032b8ab6SVijay Mahadevan {
436032b8ab6SVijay Mahadevan   PetscErrorCode    ierr;
437032b8ab6SVijay Mahadevan   moab::ErrorCode   merr;
438032b8ab6SVijay Mahadevan   PetscBool         is_newtag;
439351b8a77SVijay Mahadevan   const moab::Range *range;
440bb8f3634SVijay Mahadevan   PetscInt          count, lnative_vec, gnative_vec;
4419088682fSVijay Mahadevan   std::string       ttname;
442c4427235SVijay Mahadevan   PetscScalar       *data_ptr, *defaultvals;
443032b8ab6SVijay Mahadevan 
444db66d124SVijay Mahadevan   Vec_MOAB *vmoab;
445032b8ab6SVijay Mahadevan   DM_Moab *dmmoab = (DM_Moab*)dm->data;
4469daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
447032b8ab6SVijay Mahadevan   moab::ParallelComm *pcomm = dmmoab->pcomm;
4489daf19fdSVijay Mahadevan #endif
449032b8ab6SVijay Mahadevan   moab::Interface *mbiface = dmmoab->mbiface;
450032b8ab6SVijay Mahadevan 
451032b8ab6SVijay Mahadevan   PetscFunctionBegin;
452d1839e75SVijay Mahadevan   if (sizeof(PetscReal) != sizeof(PetscScalar)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "MOAB tags only support Real types (Complex-type unsupported)");
453032b8ab6SVijay Mahadevan   if (!userrange) range = dmmoab->vowned;
454032b8ab6SVijay Mahadevan   else range = userrange;
455db66d124SVijay Mahadevan   if (!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first.");
456032b8ab6SVijay Mahadevan 
4573ac71f70SVijay Mahadevan #ifndef USE_NATIVE_PETSCVEC
458bb8f3634SVijay Mahadevan   /* If the tag data is in a single sequence, use PETSc native vector since tag_iterate isn't useful anymore */
459bb8f3634SVijay Mahadevan   lnative_vec = (range->psize() - 1);
4603ac71f70SVijay Mahadevan #else
4613ac71f70SVijay Mahadevan   lnative_vec = 1; /* NOTE: Testing PETSc vector: will force to create native vector all the time */
4626465f021SVijay Mahadevan //  lnative_vec=0; /* NOTE: Testing MOAB vector: will force to create MOAB tag_iterate based vector all the time */
463f6829af0SVijay Mahadevan #endif
4643ac71f70SVijay Mahadevan 
4659daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
4669daf19fdSVijay Mahadevan   ierr = MPIU_Allreduce(&lnative_vec, &gnative_vec, 1, MPI_INT, MPI_MAX, (((PetscObject)dm)->comm)); CHKERRQ(ierr);
4679daf19fdSVijay Mahadevan #else
4689daf19fdSVijay Mahadevan   gnative_vec = lnative_vec;
4699daf19fdSVijay Mahadevan #endif
470bb8f3634SVijay Mahadevan 
471bb8f3634SVijay Mahadevan   /* Create the MOAB internal data object */
472f90c3b0eSVijay Mahadevan   ierr = PetscNew(&vmoab); CHKERRQ(ierr);
473bb8f3634SVijay Mahadevan   vmoab->is_native_vec = (gnative_vec > 0 ? PETSC_TRUE : PETSC_FALSE);
474bb8f3634SVijay Mahadevan 
475bb8f3634SVijay Mahadevan   if (!vmoab->is_native_vec) {
4763ac71f70SVijay Mahadevan     merr = moab::MB_SUCCESS;
47764e1c140SVijay Mahadevan     if (tag != 0) merr = mbiface->tag_get_name(tag, ttname);
47864e1c140SVijay Mahadevan     if (!ttname.length() || merr != moab::MB_SUCCESS) {
479db66d124SVijay Mahadevan       /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */
480c528d872SBarry Smith       char *tag_name = NULL;
4819daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
4829daf19fdSVijay Mahadevan       ierr = DMVecCreateTagName_Moab_Private(mbiface,pcomm,&tag_name); CHKERRQ(ierr);
4839daf19fdSVijay Mahadevan #else
4849daf19fdSVijay Mahadevan       ierr = DMVecCreateTagName_Moab_Private(mbiface,&tag_name); CHKERRQ(ierr);
4859daf19fdSVijay Mahadevan #endif
486032b8ab6SVijay Mahadevan       is_newtag = PETSC_TRUE;
487032b8ab6SVijay Mahadevan 
488db66d124SVijay Mahadevan       /* Create the default value for the tag (all zeros) */
4897ae5e5b6SVijay Mahadevan       ierr = PetscCalloc1(dmmoab->numFields, &defaultvals); CHKERRQ(ierr);
490032b8ab6SVijay Mahadevan 
491db66d124SVijay Mahadevan       /* Create the tag */
492d1839e75SVijay Mahadevan       merr = mbiface->tag_get_handle(tag_name, dmmoab->numFields, moab::MB_TYPE_DOUBLE, tag,
493c4427235SVijay Mahadevan                                      moab::MB_TAG_DENSE | moab::MB_TAG_CREAT, defaultvals); MBERRNM(merr);
494032b8ab6SVijay Mahadevan       ierr = PetscFree(tag_name); CHKERRQ(ierr);
495c4427235SVijay Mahadevan       ierr = PetscFree(defaultvals); CHKERRQ(ierr);
496032b8ab6SVijay Mahadevan     }
497032b8ab6SVijay Mahadevan     else {
498db66d124SVijay Mahadevan       /* Make sure the tag data is of type "double" */
499032b8ab6SVijay Mahadevan       moab::DataType tag_type;
500032b8ab6SVijay Mahadevan       merr = mbiface->tag_get_data_type(tag, tag_type); MBERRNM(merr);
501032b8ab6SVijay Mahadevan       if (tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE");
502032b8ab6SVijay Mahadevan       is_newtag = destroy_tag;
503032b8ab6SVijay Mahadevan     }
504032b8ab6SVijay Mahadevan 
505032b8ab6SVijay Mahadevan     vmoab->tag = tag;
506bb8f3634SVijay Mahadevan     vmoab->new_tag = is_newtag;
507bb8f3634SVijay Mahadevan   }
508032b8ab6SVijay Mahadevan   vmoab->mbiface = mbiface;
5099daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
510032b8ab6SVijay Mahadevan   vmoab->pcomm = pcomm;
5119daf19fdSVijay Mahadevan #endif
512032b8ab6SVijay Mahadevan   vmoab->is_global_vec = is_global_vec;
513bb8f3634SVijay Mahadevan   vmoab->tag_size = dmmoab->bs;
514032b8ab6SVijay Mahadevan 
515bb8f3634SVijay Mahadevan   if (vmoab->is_native_vec) {
516fc418013SVijay Mahadevan 
517bb8f3634SVijay Mahadevan     /* Create the PETSc Vector directly and attach our functions accordingly */
518bb8f3634SVijay Mahadevan     if (!is_global_vec) {
519bb8f3634SVijay Mahadevan       /* This is an MPI Vector with ghosted padding */
5209daf19fdSVijay Mahadevan       ierr = VecCreateGhostBlock((((PetscObject)dm)->comm), dmmoab->bs, dmmoab->numFields * dmmoab->nloc,
521addae81cSVijay Mahadevan                                  dmmoab->numFields * dmmoab->n, dmmoab->nghost, &dmmoab->gsindices[dmmoab->nloc], vec); CHKERRQ(ierr);
522bb8f3634SVijay Mahadevan     }
523bb8f3634SVijay Mahadevan     else {
524bb8f3634SVijay Mahadevan       /* This is an MPI/SEQ Vector */
5259daf19fdSVijay Mahadevan       ierr = VecCreate((((PetscObject)dm)->comm), vec); CHKERRQ(ierr);
526addae81cSVijay Mahadevan       ierr = VecSetSizes(*vec, dmmoab->numFields * dmmoab->nloc, PETSC_DECIDE); CHKERRQ(ierr);
527bb8f3634SVijay Mahadevan       ierr = VecSetBlockSize(*vec, dmmoab->bs); CHKERRQ(ierr);
528bb8f3634SVijay Mahadevan       ierr = VecSetType(*vec, VECMPI); CHKERRQ(ierr);
529bb8f3634SVijay Mahadevan     }
530bb8f3634SVijay Mahadevan   }
531bb8f3634SVijay Mahadevan   else {
532db66d124SVijay Mahadevan     /* Call tag_iterate. This will cause MOAB to allocate memory for the
533db66d124SVijay Mahadevan        tag data if it hasn't already happened */
534032b8ab6SVijay Mahadevan     merr = mbiface->tag_iterate(tag, range->begin(), range->end(), count, (void*&)data_ptr); MBERRNM(merr);
535032b8ab6SVijay Mahadevan 
536bb8f3634SVijay Mahadevan     /* set the reference for vector range */
537bb8f3634SVijay Mahadevan     vmoab->tag_range = new moab::Range(*range);
538addae81cSVijay Mahadevan     merr = mbiface->tag_get_length(tag, dmmoab->numFields); MBERRNM(merr);
539032b8ab6SVijay Mahadevan 
540db66d124SVijay Mahadevan     /* Create the PETSc Vector
541db66d124SVijay Mahadevan       Query MOAB mesh to check if there are any ghosted entities
542032b8ab6SVijay Mahadevan         -> if we do, create a ghosted vector to map correctly to the same layout
543032b8ab6SVijay Mahadevan         -> else, create a non-ghosted parallel vector */
544bb8f3634SVijay Mahadevan     if (!is_global_vec) {
545db66d124SVijay Mahadevan       /* This is an MPI Vector with ghosted padding */
5469daf19fdSVijay Mahadevan       ierr = VecCreateGhostBlockWithArray((((PetscObject)dm)->comm), dmmoab->bs, dmmoab->numFields * dmmoab->nloc,
547addae81cSVijay Mahadevan                                           dmmoab->numFields * dmmoab->n, dmmoab->nghost, &dmmoab->gsindices[dmmoab->nloc], data_ptr, vec); CHKERRQ(ierr);
548032b8ab6SVijay Mahadevan     }
549032b8ab6SVijay Mahadevan     else {
550bb8f3634SVijay Mahadevan       /* This is an MPI Vector without ghosted padding */
5519daf19fdSVijay Mahadevan       ierr = VecCreateMPIWithArray((((PetscObject)dm)->comm), dmmoab->bs, dmmoab->numFields * range->size(),
552bb8f3634SVijay Mahadevan                                    PETSC_DECIDE, data_ptr, vec); CHKERRQ(ierr);
553032b8ab6SVijay Mahadevan     }
554bb8f3634SVijay Mahadevan   }
555bb8f3634SVijay Mahadevan   ierr = VecSetFromOptions(*vec); CHKERRQ(ierr);
556032b8ab6SVijay Mahadevan 
557f6829af0SVijay Mahadevan   /* create a container and store the internal MOAB data for faster access based on Entities etc */
558032b8ab6SVijay Mahadevan   PetscContainer moabdata;
559bb8f3634SVijay Mahadevan   ierr = PetscContainerCreate(PETSC_COMM_WORLD, &moabdata); CHKERRQ(ierr);
560032b8ab6SVijay Mahadevan   ierr = PetscContainerSetPointer(moabdata, vmoab); CHKERRQ(ierr);
561f6829af0SVijay Mahadevan   ierr = PetscContainerSetUserDestroy(moabdata, DMVecUserDestroy_Moab); CHKERRQ(ierr);
562032b8ab6SVijay Mahadevan   ierr = PetscObjectCompose((PetscObject) * vec, "MOABData", (PetscObject)moabdata); CHKERRQ(ierr);
563f6829af0SVijay Mahadevan   (*vec)->ops->duplicate = DMVecDuplicate_Moab;
564032b8ab6SVijay Mahadevan   ierr = PetscContainerDestroy(&moabdata); CHKERRQ(ierr);
565032b8ab6SVijay Mahadevan 
566db66d124SVijay Mahadevan   /* Vector created, manually set local to global mapping */
5676d9eb265SVijay Mahadevan   if (dmmoab->ltog_map) {
568bb8f3634SVijay Mahadevan     ierr = VecSetLocalToGlobalMapping(*vec, dmmoab->ltog_map); CHKERRQ(ierr);
5696d9eb265SVijay Mahadevan   }
570032b8ab6SVijay Mahadevan 
571032b8ab6SVijay Mahadevan   /* set the DM reference to the vector */
572032b8ab6SVijay Mahadevan   ierr = VecSetDM(*vec, dm); CHKERRQ(ierr);
573032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
574032b8ab6SVijay Mahadevan }
575032b8ab6SVijay Mahadevan 
576032b8ab6SVijay Mahadevan 
577f6829af0SVijay Mahadevan /*  DMVecCreateTagName_Moab_Private
578032b8ab6SVijay Mahadevan  *
579032b8ab6SVijay Mahadevan  *  Creates a unique tag name that will be shared across processes. If
580032b8ab6SVijay Mahadevan  *  pcomm is NULL, then this is a serial vector. A unique tag name
581032b8ab6SVijay Mahadevan  *  will be returned in tag_name in either case.
582032b8ab6SVijay Mahadevan  *
583032b8ab6SVijay Mahadevan  *  The tag names have the format _PETSC_VEC_N where N is some integer.
584032b8ab6SVijay Mahadevan  *
585032b8ab6SVijay Mahadevan  *  NOTE: The tag_name is allocated in this routine; The user needs to free
586032b8ab6SVijay Mahadevan  *        the character array.
587032b8ab6SVijay Mahadevan  */
5889daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
5899daf19fdSVijay Mahadevan PetscErrorCode DMVecCreateTagName_Moab_Private(moab::Interface *mbiface, moab::ParallelComm *pcomm, char** tag_name)
5909daf19fdSVijay Mahadevan #else
5919daf19fdSVijay Mahadevan PetscErrorCode DMVecCreateTagName_Moab_Private(moab::Interface *mbiface, char** tag_name)
5929daf19fdSVijay Mahadevan #endif
593032b8ab6SVijay Mahadevan {
594032b8ab6SVijay Mahadevan   moab::ErrorCode mberr;
595032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
596032b8ab6SVijay Mahadevan   PetscInt        n, global_n;
597032b8ab6SVijay Mahadevan   moab::Tag indexTag;
598032b8ab6SVijay Mahadevan 
599032b8ab6SVijay Mahadevan   PetscFunctionBegin;
600032b8ab6SVijay Mahadevan   const char*       PVEC_PREFIX      = "__PETSC_VEC_";
60195dccacaSBarry Smith   ierr = PetscMalloc1(PETSC_MAX_PATH_LEN, tag_name); CHKERRQ(ierr);
602032b8ab6SVijay Mahadevan 
603032b8ab6SVijay Mahadevan   moab::EntityHandle rootset = mbiface->get_root_set();
604032b8ab6SVijay Mahadevan 
6059daf19fdSVijay Mahadevan   /* Check to see if there are any PETSc vectors defined */
606efd17f3eSVijay Mahadevan   /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */
607032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_handle("__PETSC_VECS__", 1, moab::MB_TYPE_INTEGER, indexTag,
608032b8ab6SVijay Mahadevan                                   moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT, 0); MBERRNM(mberr);
609032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n);
610032b8ab6SVijay Mahadevan   if (mberr == moab::MB_TAG_NOT_FOUND) n = 0; /* this is the first temporary vector */
611032b8ab6SVijay Mahadevan   else MBERRNM(mberr);
612032b8ab6SVijay Mahadevan 
613032b8ab6SVijay Mahadevan   /* increment the new value of n */
614032b8ab6SVijay Mahadevan   ++n;
615032b8ab6SVijay Mahadevan 
6169daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
617efd17f3eSVijay Mahadevan   /* Make sure that n is consistent across all processes */
618b2566f29SBarry Smith   ierr = MPIU_Allreduce(&n, &global_n, 1, MPI_INT, MPI_MAX, pcomm->comm()); CHKERRQ(ierr);
6199daf19fdSVijay Mahadevan #else
6209daf19fdSVijay Mahadevan   global_n = n;
6219daf19fdSVijay Mahadevan #endif
622032b8ab6SVijay Mahadevan 
623efd17f3eSVijay Mahadevan   /* Set the new name accordingly and return */
624032b8ab6SVijay Mahadevan   ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN - 1, "%s_%D", PVEC_PREFIX, global_n); CHKERRQ(ierr);
625032b8ab6SVijay Mahadevan   mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n); MBERRNM(mberr);
626032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
627032b8ab6SVijay Mahadevan }
628032b8ab6SVijay Mahadevan 
629032b8ab6SVijay Mahadevan 
630304006b3SVijay Mahadevan #undef __FUNCT__
631304006b3SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab"
632304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode DMCreateGlobalVector_Moab(DM dm, Vec *gvec)
633f6829af0SVijay Mahadevan {
634f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
635f6829af0SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
636f6829af0SVijay Mahadevan 
637f6829af0SVijay Mahadevan   PetscFunctionBegin;
638f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
639f6829af0SVijay Mahadevan   PetscValidPointer(gvec,2);
640c528d872SBarry Smith   ierr = DMCreateVector_Moab_Private(dm,NULL,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr);
641f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
642f6829af0SVijay Mahadevan }
643f6829af0SVijay Mahadevan 
644f6829af0SVijay Mahadevan 
645304006b3SVijay Mahadevan #undef __FUNCT__
646304006b3SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab"
647304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode DMCreateLocalVector_Moab(DM dm, Vec *lvec)
648f6829af0SVijay Mahadevan {
649f6829af0SVijay Mahadevan   PetscErrorCode  ierr;
650f6829af0SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
651f6829af0SVijay Mahadevan 
652f6829af0SVijay Mahadevan   PetscFunctionBegin;
653f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
654f6829af0SVijay Mahadevan   PetscValidPointer(lvec,2);
655c528d872SBarry Smith   ierr = DMCreateVector_Moab_Private(dm,NULL,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr);
656f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
657f6829af0SVijay Mahadevan }
658f6829af0SVijay Mahadevan 
659f6829af0SVijay Mahadevan 
66063d025dbSVijay Mahadevan #undef __FUNCT__
66163d025dbSVijay Mahadevan #define __FUNCT__ "DMVecDuplicate_Moab"
662f6829af0SVijay Mahadevan PetscErrorCode DMVecDuplicate_Moab(Vec x, Vec *y)
663f6829af0SVijay Mahadevan {
664f6829af0SVijay Mahadevan   PetscErrorCode ierr;
665f6829af0SVijay Mahadevan   DM             dm;
666f6829af0SVijay Mahadevan   PetscContainer  moabdata;
667f6829af0SVijay Mahadevan   Vec_MOAB        *vmoab;
668f6829af0SVijay Mahadevan 
669f6829af0SVijay Mahadevan   PetscFunctionBegin;
670f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(x, VEC_CLASSID, 1);
671f6829af0SVijay Mahadevan   PetscValidPointer(y, 2);
672f6829af0SVijay Mahadevan 
673f6829af0SVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
674f6829af0SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)x, "MOABData", (PetscObject*) &moabdata); CHKERRQ(ierr);
675f6829af0SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab); CHKERRQ(ierr);
676f6829af0SVijay Mahadevan 
677f6829af0SVijay Mahadevan   ierr = VecGetDM(x, &dm); CHKERRQ(ierr);
678f6829af0SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
679f6829af0SVijay Mahadevan 
680c528d872SBarry Smith   ierr = DMCreateVector_Moab_Private(dm,NULL,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr);
681f6829af0SVijay Mahadevan   ierr = VecSetDM(*y, dm);CHKERRQ(ierr);
682f6829af0SVijay Mahadevan   PetscFunctionReturn(0);
683f6829af0SVijay Mahadevan }
684f6829af0SVijay Mahadevan 
685f6829af0SVijay Mahadevan 
686f6829af0SVijay Mahadevan PetscErrorCode DMVecUserDestroy_Moab(void *user)
687032b8ab6SVijay Mahadevan {
688efd17f3eSVijay Mahadevan   Vec_MOAB        *vmoab = (Vec_MOAB*)user;
689032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
690032b8ab6SVijay Mahadevan   moab::ErrorCode merr;
691032b8ab6SVijay Mahadevan 
692032b8ab6SVijay Mahadevan   PetscFunctionBegin;
693032b8ab6SVijay Mahadevan   if (vmoab->new_tag && vmoab->tag) {
694efd17f3eSVijay Mahadevan     /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */
695032b8ab6SVijay Mahadevan     merr = vmoab->mbiface->tag_delete(vmoab->tag); MBERRNM(merr);
696032b8ab6SVijay Mahadevan   }
697032b8ab6SVijay Mahadevan   delete vmoab->tag_range;
698c528d872SBarry Smith   vmoab->tag = NULL;
699c528d872SBarry Smith   vmoab->mbiface = NULL;
7009daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
701c528d872SBarry Smith   vmoab->pcomm = NULL;
7029daf19fdSVijay Mahadevan #endif
703032b8ab6SVijay Mahadevan   ierr = PetscFree(vmoab); CHKERRQ(ierr);
704032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
705032b8ab6SVijay Mahadevan }
706032b8ab6SVijay Mahadevan 
707efd17f3eSVijay Mahadevan 
708304006b3SVijay Mahadevan #undef __FUNCT__
709304006b3SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab"
710304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode  DMGlobalToLocalBegin_Moab(DM dm, Vec g, InsertMode mode, Vec l)
711db66d124SVijay Mahadevan {
712db66d124SVijay Mahadevan   PetscErrorCode    ierr;
713db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
714db66d124SVijay Mahadevan 
715db66d124SVijay Mahadevan   PetscFunctionBegin;
7166d9eb265SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv, g, l, mode, SCATTER_REVERSE); CHKERRQ(ierr);
717db66d124SVijay Mahadevan   PetscFunctionReturn(0);
718db66d124SVijay Mahadevan }
719db66d124SVijay Mahadevan 
720db66d124SVijay Mahadevan 
721304006b3SVijay Mahadevan #undef __FUNCT__
722304006b3SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab"
723304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode  DMGlobalToLocalEnd_Moab(DM dm, Vec g, InsertMode mode, Vec l)
724db66d124SVijay Mahadevan {
725db66d124SVijay Mahadevan   PetscErrorCode    ierr;
726db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
727db66d124SVijay Mahadevan 
728db66d124SVijay Mahadevan   PetscFunctionBegin;
7296d9eb265SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv, g, l, mode, SCATTER_REVERSE); CHKERRQ(ierr);
730db66d124SVijay Mahadevan   PetscFunctionReturn(0);
731db66d124SVijay Mahadevan }
732db66d124SVijay Mahadevan 
733db66d124SVijay Mahadevan 
734304006b3SVijay Mahadevan #undef __FUNCT__
735304006b3SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab"
736304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode  DMLocalToGlobalBegin_Moab(DM dm, Vec l, InsertMode mode, Vec g)
737db66d124SVijay Mahadevan {
738db66d124SVijay Mahadevan   PetscErrorCode    ierr;
739db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
740db66d124SVijay Mahadevan 
741db66d124SVijay Mahadevan   PetscFunctionBegin;
7426d9eb265SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv, l, g, mode, SCATTER_FORWARD); CHKERRQ(ierr);
743db66d124SVijay Mahadevan   PetscFunctionReturn(0);
744db66d124SVijay Mahadevan }
745db66d124SVijay Mahadevan 
746db66d124SVijay Mahadevan 
747304006b3SVijay Mahadevan #undef __FUNCT__
748304006b3SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab"
749304006b3SVijay Mahadevan PETSC_EXTERN PetscErrorCode  DMLocalToGlobalEnd_Moab(DM dm, Vec l, InsertMode mode, Vec g)
750db66d124SVijay Mahadevan {
751db66d124SVijay Mahadevan   PetscErrorCode    ierr;
752db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
753db66d124SVijay Mahadevan 
754db66d124SVijay Mahadevan   PetscFunctionBegin;
7556d9eb265SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv, l, g, mode, SCATTER_FORWARD); CHKERRQ(ierr);
756db66d124SVijay Mahadevan   PetscFunctionReturn(0);
757db66d124SVijay Mahadevan }
758db66d124SVijay Mahadevan 
759