xref: /petsc/src/dm/impls/moab/dmmbvec.cxx (revision fc4180132621dce802f5cfc3099d116b2d2b65bb)
1032b8ab6SVijay Mahadevan #include <petsc-private/dmmbimpl.h> /*I  "petscdm.h"   I*/
2032b8ab6SVijay Mahadevan #include <petsc-private/vecimpl.h> /*I  "petscdm.h"   I*/
3032b8ab6SVijay Mahadevan 
4032b8ab6SVijay Mahadevan #include <petscdmmoab.h>
5032b8ab6SVijay Mahadevan #include <MBTagConventions.hpp>
6032b8ab6SVijay Mahadevan 
7032b8ab6SVijay Mahadevan // declare for use later but before they're defined
8032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecUserDestroy(void *user);
9032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y);
10032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name);
11032b8ab6SVijay Mahadevan 
12032b8ab6SVijay Mahadevan #undef __FUNCT__
13032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_CreateVector_Private"
14032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_CreateVector_Private(DM dm,moab::Tag tag,PetscInt tag_size,moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec)
15032b8ab6SVijay Mahadevan {
16032b8ab6SVijay Mahadevan   PetscErrorCode         ierr;
17032b8ab6SVijay Mahadevan   moab::ErrorCode        merr;
18032b8ab6SVijay Mahadevan   PetscBool              is_newtag;
19032b8ab6SVijay Mahadevan   moab::Range           *range;
20*fc418013SVijay Mahadevan   PetscInt               *gindices;
21032b8ab6SVijay Mahadevan   PetscInt               i,count,icount,dof;
22032b8ab6SVijay Mahadevan   PetscInt               size,rank;
239088682fSVijay Mahadevan   std::string ttname;
24032b8ab6SVijay Mahadevan   PetscScalar *data_ptr;
25032b8ab6SVijay Mahadevan 
26db66d124SVijay Mahadevan   Vec_MOAB *vmoab;
27032b8ab6SVijay Mahadevan   DM_Moab *dmmoab = (DM_Moab*)dm->data;
28032b8ab6SVijay Mahadevan   moab::ParallelComm *pcomm = dmmoab->pcomm;
29032b8ab6SVijay Mahadevan   moab::Interface *mbiface = dmmoab->mbiface;
30032b8ab6SVijay Mahadevan   moab::Tag ltog_tag = dmmoab->ltog_tag;
31032b8ab6SVijay Mahadevan 
32032b8ab6SVijay Mahadevan   PetscFunctionBegin;
33032b8ab6SVijay Mahadevan   if(!userrange) range = dmmoab->vowned;
34032b8ab6SVijay Mahadevan   else range = userrange;
35db66d124SVijay Mahadevan   if(!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first.");
36032b8ab6SVijay Mahadevan 
379088682fSVijay Mahadevan   merr = mbiface->tag_get_name(tag, ttname);
389088682fSVijay Mahadevan   if (!ttname.length() && merr !=moab::MB_SUCCESS) {
39db66d124SVijay Mahadevan     /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */
40032b8ab6SVijay Mahadevan     char *tag_name = PETSC_NULL;
41032b8ab6SVijay Mahadevan     ierr = DMMoab_VecCreateTagName_Private(pcomm,&tag_name);CHKERRQ(ierr);
42032b8ab6SVijay Mahadevan     is_newtag = PETSC_TRUE;
43032b8ab6SVijay Mahadevan 
44db66d124SVijay Mahadevan     /* Create the default value for the tag (all zeros) */
45032b8ab6SVijay Mahadevan     std::vector<PetscScalar> default_value(tag_size, 0.0);
46032b8ab6SVijay Mahadevan 
47db66d124SVijay Mahadevan     /* Create the tag */
48032b8ab6SVijay Mahadevan     merr = mbiface->tag_get_handle(tag_name,tag_size,moab::MB_TYPE_DOUBLE,tag,
49032b8ab6SVijay Mahadevan                                    moab::MB_TAG_DENSE|moab::MB_TAG_CREAT,default_value.data());MBERRNM(merr);
50032b8ab6SVijay Mahadevan     ierr = PetscFree(tag_name);CHKERRQ(ierr);
51032b8ab6SVijay Mahadevan   }
52032b8ab6SVijay Mahadevan   else {
53db66d124SVijay Mahadevan     /* Make sure the tag data is of type "double" */
54032b8ab6SVijay Mahadevan     moab::DataType tag_type;
55032b8ab6SVijay Mahadevan     merr = mbiface->tag_get_data_type(tag, tag_type);MBERRNM(merr);
56032b8ab6SVijay Mahadevan     if(tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE");
57032b8ab6SVijay Mahadevan     is_newtag = destroy_tag;
58032b8ab6SVijay Mahadevan   }
59032b8ab6SVijay Mahadevan 
60db66d124SVijay Mahadevan   /* Create the MOAB internal data object */
61032b8ab6SVijay Mahadevan   ierr = PetscNew(Vec_MOAB,&vmoab);CHKERRQ(ierr);
62032b8ab6SVijay Mahadevan   vmoab->tag = tag;
63032b8ab6SVijay Mahadevan   vmoab->mbiface = mbiface;
64032b8ab6SVijay Mahadevan   vmoab->pcomm = pcomm;
65032b8ab6SVijay Mahadevan   vmoab->new_tag = is_newtag;
66032b8ab6SVijay Mahadevan   vmoab->is_global_vec = is_global_vec;
67032b8ab6SVijay Mahadevan   merr = mbiface->tag_get_length(tag,vmoab->tag_size);MBERR("tag_get_size", merr);
68032b8ab6SVijay Mahadevan 
69*fc418013SVijay Mahadevan   size = pcomm->size();
70*fc418013SVijay Mahadevan   rank = pcomm->rank();
71*fc418013SVijay Mahadevan 
72db66d124SVijay Mahadevan   /* set the reference for vector range */
73032b8ab6SVijay Mahadevan   vmoab->tag_range = new moab::Range(*range);
74032b8ab6SVijay Mahadevan 
75db66d124SVijay Mahadevan   /* Call tag_iterate. This will cause MOAB to allocate memory for the
76db66d124SVijay Mahadevan      tag data if it hasn't already happened */
77032b8ab6SVijay Mahadevan   merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr);
78032b8ab6SVijay Mahadevan 
79*fc418013SVijay Mahadevan   if (rank) range->print();
80db66d124SVijay Mahadevan   /* Check to make sure the tag data is in a single sequence */
81*fc418013SVijay Mahadevan   if ((unsigned)count != range->size()) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only create MOAB Vector for single sequence: %D != %D",count,range->size());
82032b8ab6SVijay Mahadevan 
83db66d124SVijay Mahadevan   /* Create the PETSc Vector
84db66d124SVijay Mahadevan      Query MOAB mesh to check if there are any ghosted entities
85032b8ab6SVijay Mahadevan        -> if we do, create a ghosted vector to map correctly to the same layout
86032b8ab6SVijay Mahadevan        -> else, create a non-ghosted parallel vector */
87*fc418013SVijay Mahadevan   if (!is_global_vec && (size>1)) {
88db66d124SVijay Mahadevan     /* This is an MPI Vector with ghosted padding */
89032b8ab6SVijay Mahadevan     ierr = VecCreateGhostBlockWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*dmmoab->nloc,
90*fc418013SVijay Mahadevan                               vmoab->tag_size*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],data_ptr,vec);CHKERRQ(ierr);
91032b8ab6SVijay Mahadevan   }
92032b8ab6SVijay Mahadevan   else if (size>1) {
93db66d124SVijay Mahadevan     /* This is an MPI Vector without ghosted padding */
94032b8ab6SVijay Mahadevan     ierr = VecCreateMPIWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*range->size(),
95032b8ab6SVijay Mahadevan                               PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr);
96032b8ab6SVijay Mahadevan   }
97032b8ab6SVijay Mahadevan   else {
98db66d124SVijay Mahadevan     /* This is a sequential vector - valid only for the single processor case since MOAB tags are always partitioned
99db66d124SVijay Mahadevan        and we cannot define a Vec using the Tag array with size>1 will be of full length */
100032b8ab6SVijay Mahadevan     ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,vmoab->tag_size,vmoab->tag_size*dmmoab->n,data_ptr,vec);CHKERRQ(ierr);
101032b8ab6SVijay Mahadevan   }
102032b8ab6SVijay Mahadevan 
103032b8ab6SVijay Mahadevan   PetscContainer moabdata;
104032b8ab6SVijay Mahadevan   ierr = PetscContainerCreate(PETSC_COMM_SELF,&moabdata);CHKERRQ(ierr);
105032b8ab6SVijay Mahadevan   ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr);
106032b8ab6SVijay Mahadevan   ierr = PetscContainerSetUserDestroy(moabdata,DMMoab_VecUserDestroy);CHKERRQ(ierr);
107032b8ab6SVijay Mahadevan   ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr);
108032b8ab6SVijay Mahadevan   (*vec)->ops->duplicate = DMMoab_VecDuplicate;
109032b8ab6SVijay Mahadevan   ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr);
110032b8ab6SVijay Mahadevan 
111032b8ab6SVijay Mahadevan   if (!dmmoab->ltog_map) {
112db66d124SVijay Mahadevan     /* Vector created, manually set local to global mapping */
113032b8ab6SVijay Mahadevan     ierr = PetscMalloc(range->size()*sizeof(PetscInt)*vmoab->tag_size, &gindices);CHKERRQ(ierr);
114032b8ab6SVijay Mahadevan     moab::Range::iterator  iter;
115032b8ab6SVijay Mahadevan     for(iter = range->begin(),count=0; iter != range->end(); iter++,count+=vmoab->tag_size) {
116032b8ab6SVijay Mahadevan       merr = mbiface->tag_get_data(ltog_tag,&(*iter),1,&dof);MBERRNM(merr);
117032b8ab6SVijay Mahadevan       for(i=0; i<vmoab->tag_size; ++i)
118032b8ab6SVijay Mahadevan         gindices[count+i] = (dof)*vmoab->tag_size+i;
119032b8ab6SVijay Mahadevan     }
120032b8ab6SVijay Mahadevan 
121032b8ab6SVijay Mahadevan     ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_SELF,range->size(),gindices,
122032b8ab6SVijay Mahadevan                                         PETSC_COPY_VALUES,&dmmoab->ltog_map);CHKERRQ(ierr);
123032b8ab6SVijay Mahadevan 
124032b8ab6SVijay Mahadevan     ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr);
125032b8ab6SVijay Mahadevan 
126db66d124SVijay Mahadevan     /* Clean up */
127032b8ab6SVijay Mahadevan     ierr = PetscFree(gindices);CHKERRQ(ierr);
128032b8ab6SVijay Mahadevan   }
129032b8ab6SVijay Mahadevan   else {
130032b8ab6SVijay Mahadevan     ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr);
131032b8ab6SVijay Mahadevan   }
132032b8ab6SVijay Mahadevan 
133032b8ab6SVijay Mahadevan   /* set the DM reference to the vector */
134032b8ab6SVijay Mahadevan   ierr = VecSetDM(*vec, dm);CHKERRQ(ierr);
135032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
136032b8ab6SVijay Mahadevan }
137032b8ab6SVijay Mahadevan 
138032b8ab6SVijay Mahadevan 
139032b8ab6SVijay Mahadevan #undef __FUNCT__
140032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabCreateVector"
141032b8ab6SVijay Mahadevan /*@
142032b8ab6SVijay Mahadevan   DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities
143032b8ab6SVijay Mahadevan 
144032b8ab6SVijay Mahadevan   Collective on MPI_Comm
145032b8ab6SVijay Mahadevan 
146032b8ab6SVijay Mahadevan   Input Parameter:
147032b8ab6SVijay Mahadevan . dm              - The DMMoab object being set
148032b8ab6SVijay Mahadevan . tag             - If non-zero, block size will be taken from the tag size
149032b8ab6SVijay Mahadevan . tag_size        - If tag was zero, this parameter specifies the block size; unique tag name will be generated automatically
150032b8ab6SVijay Mahadevan . range           - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab
151032b8ab6SVijay Mahadevan . is_global_vec   - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one
152032b8ab6SVijay Mahadevan . destroy_tag     - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB
153032b8ab6SVijay Mahadevan 
154032b8ab6SVijay Mahadevan   Output Parameter:
155032b8ab6SVijay Mahadevan . vec             - The created vector
156032b8ab6SVijay Mahadevan 
157032b8ab6SVijay Mahadevan   Level: beginner
158032b8ab6SVijay Mahadevan 
159032b8ab6SVijay Mahadevan .keywords: DMMoab, create
160032b8ab6SVijay Mahadevan @*/
161032b8ab6SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,PetscInt tag_size,moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec)
162032b8ab6SVijay Mahadevan {
163032b8ab6SVijay Mahadevan   PetscErrorCode     ierr;
164032b8ab6SVijay Mahadevan 
165032b8ab6SVijay Mahadevan   PetscFunctionBegin;
166032b8ab6SVijay Mahadevan   if(!tag && !tag_size) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag_size and tag cannot be null.");
167032b8ab6SVijay Mahadevan 
168032b8ab6SVijay Mahadevan   ierr = DMMoab_CreateVector_Private(dm,tag,tag_size,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr);
169032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
170032b8ab6SVijay Mahadevan }
171032b8ab6SVijay Mahadevan 
172032b8ab6SVijay Mahadevan 
173032b8ab6SVijay Mahadevan #undef __FUNCT__
174032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab"
175032b8ab6SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec)
176032b8ab6SVijay Mahadevan {
177032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
178032b8ab6SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
179032b8ab6SVijay Mahadevan 
180032b8ab6SVijay Mahadevan   PetscFunctionBegin;
181032b8ab6SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
182032b8ab6SVijay Mahadevan   PetscValidPointer(gvec,2);
183032b8ab6SVijay Mahadevan   ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr);
184032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
185032b8ab6SVijay Mahadevan }
186032b8ab6SVijay Mahadevan 
187032b8ab6SVijay Mahadevan 
188032b8ab6SVijay Mahadevan #undef __FUNCT__
189032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab"
190032b8ab6SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec)
191032b8ab6SVijay Mahadevan {
192032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
193032b8ab6SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
194032b8ab6SVijay Mahadevan 
195032b8ab6SVijay Mahadevan   PetscFunctionBegin;
196032b8ab6SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
197032b8ab6SVijay Mahadevan   PetscValidPointer(lvec,2);
198*fc418013SVijay Mahadevan   ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr);
199032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
200032b8ab6SVijay Mahadevan }
201032b8ab6SVijay Mahadevan 
202032b8ab6SVijay Mahadevan 
203032b8ab6SVijay Mahadevan #undef __FUNCT__
204032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecTag"
205032b8ab6SVijay Mahadevan /*@
206032b8ab6SVijay Mahadevan   DMMoabGetVecTag - Get the MOAB tag associated with this Vec
207032b8ab6SVijay Mahadevan 
208032b8ab6SVijay Mahadevan   Input Parameter:
209032b8ab6SVijay Mahadevan . vec - Vec being queried
210032b8ab6SVijay Mahadevan 
211032b8ab6SVijay Mahadevan   Output Parameter:
212032b8ab6SVijay Mahadevan . tag - Tag associated with this Vec
213032b8ab6SVijay Mahadevan 
214032b8ab6SVijay Mahadevan   Level: beginner
215032b8ab6SVijay Mahadevan 
216032b8ab6SVijay Mahadevan .keywords: DMMoab, create
217032b8ab6SVijay Mahadevan @*/
218032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag)
219032b8ab6SVijay Mahadevan {
220032b8ab6SVijay Mahadevan   PetscContainer  moabdata;
221032b8ab6SVijay Mahadevan   Vec_MOAB        *vmoab;
222032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
223032b8ab6SVijay Mahadevan 
224032b8ab6SVijay Mahadevan   PetscFunctionBegin;
225032b8ab6SVijay Mahadevan   PetscValidPointer(tag,2);
226032b8ab6SVijay Mahadevan 
227efd17f3eSVijay Mahadevan   /* Get the MOAB private data */
228032b8ab6SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
229032b8ab6SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr);
230032b8ab6SVijay Mahadevan 
231032b8ab6SVijay Mahadevan   *tag = vmoab->tag;
232032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
233032b8ab6SVijay Mahadevan }
234032b8ab6SVijay Mahadevan 
235032b8ab6SVijay Mahadevan 
236032b8ab6SVijay Mahadevan #undef __FUNCT__
237032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecRange"
238032b8ab6SVijay Mahadevan /*@
239032b8ab6SVijay Mahadevan   DMMoabGetVecRange - Get the MOAB entities associated with this Vec
240032b8ab6SVijay Mahadevan 
241032b8ab6SVijay Mahadevan   Input Parameter:
242032b8ab6SVijay Mahadevan . vec   - Vec being queried
243032b8ab6SVijay Mahadevan 
244032b8ab6SVijay Mahadevan   Output Parameter:
245032b8ab6SVijay Mahadevan . range - Entities associated with this Vec
246032b8ab6SVijay Mahadevan 
247032b8ab6SVijay Mahadevan   Level: beginner
248032b8ab6SVijay Mahadevan 
249032b8ab6SVijay Mahadevan .keywords: DMMoab, create
250032b8ab6SVijay Mahadevan @*/
251032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range)
252032b8ab6SVijay Mahadevan {
253032b8ab6SVijay Mahadevan   PetscContainer  moabdata;
254032b8ab6SVijay Mahadevan   Vec_MOAB        *vmoab;
255032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
256032b8ab6SVijay Mahadevan 
257032b8ab6SVijay Mahadevan   PetscFunctionBegin;
258032b8ab6SVijay Mahadevan   PetscValidPointer(range,2);
259032b8ab6SVijay Mahadevan 
260efd17f3eSVijay Mahadevan   /* Get the MOAB private data handle */
261032b8ab6SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
262032b8ab6SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr);
263032b8ab6SVijay Mahadevan 
264032b8ab6SVijay Mahadevan   *range = *vmoab->tag_range;
265032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
266032b8ab6SVijay Mahadevan }
267032b8ab6SVijay Mahadevan 
268032b8ab6SVijay Mahadevan 
269032b8ab6SVijay Mahadevan #undef __FUNCT__
270032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecDuplicate"
271032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y)
272032b8ab6SVijay Mahadevan {
273032b8ab6SVijay Mahadevan   PetscErrorCode ierr;
274032b8ab6SVijay Mahadevan   DM             dm;
275032b8ab6SVijay Mahadevan   PetscContainer  moabdata;
276032b8ab6SVijay Mahadevan   Vec_MOAB        *vmoab;
277032b8ab6SVijay Mahadevan 
278032b8ab6SVijay Mahadevan   PetscFunctionBegin;
279032b8ab6SVijay Mahadevan   PetscValidHeaderSpecific(x,VEC_CLASSID,1);
280032b8ab6SVijay Mahadevan   PetscValidPointer(y,2);
281032b8ab6SVijay Mahadevan 
282efd17f3eSVijay Mahadevan   /* Get the Vec_MOAB struct for the original vector */
283032b8ab6SVijay Mahadevan   ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr);
284032b8ab6SVijay Mahadevan   ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr);
285032b8ab6SVijay Mahadevan 
286032b8ab6SVijay Mahadevan   ierr = VecGetDM(x, &dm);CHKERRQ(ierr);
287032b8ab6SVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
288032b8ab6SVijay Mahadevan 
289032b8ab6SVijay Mahadevan   ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,vmoab->tag_size,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr);
290032b8ab6SVijay Mahadevan   ierr = VecSetDM(*y, dm);CHKERRQ(ierr);
291032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
292032b8ab6SVijay Mahadevan }
293032b8ab6SVijay Mahadevan 
294032b8ab6SVijay Mahadevan 
295032b8ab6SVijay Mahadevan #undef __FUNCT__
296032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecCreateTagName_Private"
297032b8ab6SVijay Mahadevan /*  DMMoab_VecCreateTagName_Private
298032b8ab6SVijay Mahadevan  *
299032b8ab6SVijay Mahadevan  *  Creates a unique tag name that will be shared across processes. If
300032b8ab6SVijay Mahadevan  *  pcomm is NULL, then this is a serial vector. A unique tag name
301032b8ab6SVijay Mahadevan  *  will be returned in tag_name in either case.
302032b8ab6SVijay Mahadevan  *
303032b8ab6SVijay Mahadevan  *  The tag names have the format _PETSC_VEC_N where N is some integer.
304032b8ab6SVijay Mahadevan  *
305032b8ab6SVijay Mahadevan  *  NOTE: The tag_name is allocated in this routine; The user needs to free
306032b8ab6SVijay Mahadevan  *        the character array.
307032b8ab6SVijay Mahadevan  */
308032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name)
309032b8ab6SVijay Mahadevan {
310032b8ab6SVijay Mahadevan   moab::ErrorCode mberr;
311032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
312032b8ab6SVijay Mahadevan   PetscInt        n,global_n;
313032b8ab6SVijay Mahadevan   moab::Tag indexTag;
314032b8ab6SVijay Mahadevan 
315032b8ab6SVijay Mahadevan   PetscFunctionBegin;
316032b8ab6SVijay Mahadevan   const char*       PVEC_PREFIX      = "__PETSC_VEC_";
317032b8ab6SVijay Mahadevan   ierr = PetscMalloc(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr);
318032b8ab6SVijay Mahadevan 
319efd17f3eSVijay Mahadevan   /* Check to see if there are any PETSc vectors defined */
320032b8ab6SVijay Mahadevan   moab::Interface  *mbiface = pcomm->get_moab();
321032b8ab6SVijay Mahadevan   moab::EntityHandle rootset = mbiface->get_root_set();
322032b8ab6SVijay Mahadevan 
323efd17f3eSVijay Mahadevan   /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */
324032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag,
325032b8ab6SVijay Mahadevan                                   moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr);
326032b8ab6SVijay Mahadevan   mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n);
327032b8ab6SVijay Mahadevan   if (mberr == moab::MB_TAG_NOT_FOUND) n=0;  /* this is the first temporary vector */
328032b8ab6SVijay Mahadevan   else MBERRNM(mberr);
329032b8ab6SVijay Mahadevan 
330032b8ab6SVijay Mahadevan   /* increment the new value of n */
331032b8ab6SVijay Mahadevan   ++n;
332032b8ab6SVijay Mahadevan 
333efd17f3eSVijay Mahadevan   /* Make sure that n is consistent across all processes */
334032b8ab6SVijay Mahadevan   ierr = MPI_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr);
335032b8ab6SVijay Mahadevan 
336efd17f3eSVijay Mahadevan   /* Set the new name accordingly and return */
337032b8ab6SVijay Mahadevan   ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr);
338032b8ab6SVijay Mahadevan   mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr);
339032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
340032b8ab6SVijay Mahadevan }
341032b8ab6SVijay Mahadevan 
342032b8ab6SVijay Mahadevan 
343032b8ab6SVijay Mahadevan #undef __FUNCT__
344032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecUserDestroy"
345032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecUserDestroy(void *user)
346032b8ab6SVijay Mahadevan {
347efd17f3eSVijay Mahadevan   Vec_MOAB        *vmoab=(Vec_MOAB*)user;
348032b8ab6SVijay Mahadevan   PetscErrorCode  ierr;
349032b8ab6SVijay Mahadevan   moab::ErrorCode merr;
350032b8ab6SVijay Mahadevan 
351032b8ab6SVijay Mahadevan   PetscFunctionBegin;
352032b8ab6SVijay Mahadevan   if(vmoab->new_tag && vmoab->tag) {
353efd17f3eSVijay Mahadevan     /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */
354032b8ab6SVijay Mahadevan     merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr);
355032b8ab6SVijay Mahadevan   }
356032b8ab6SVijay Mahadevan   delete vmoab->tag_range;
357032b8ab6SVijay Mahadevan   vmoab->tag = PETSC_NULL;
358032b8ab6SVijay Mahadevan   vmoab->mbiface = PETSC_NULL;
359032b8ab6SVijay Mahadevan   vmoab->pcomm = PETSC_NULL;
360032b8ab6SVijay Mahadevan   ierr = PetscFree(vmoab);CHKERRQ(ierr);
361032b8ab6SVijay Mahadevan   PetscFunctionReturn(0);
362032b8ab6SVijay Mahadevan }
363032b8ab6SVijay Mahadevan 
364db66d124SVijay Mahadevan #undef __FUNCT__
365efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArray"
366efd17f3eSVijay Mahadevan /*@
367efd17f3eSVijay 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
368efd17f3eSVijay Mahadevan 
369efd17f3eSVijay Mahadevan   Collective on MPI_Comm
370efd17f3eSVijay Mahadevan 
371efd17f3eSVijay Mahadevan   Input Parameter:
372efd17f3eSVijay Mahadevan . dm              - The DMMoab object being set
373efd17f3eSVijay Mahadevan . vec             - The Vector whose underlying data is requested
374efd17f3eSVijay Mahadevan 
375efd17f3eSVijay Mahadevan   Output Parameter:
376efd17f3eSVijay Mahadevan . array           - The local data array
377efd17f3eSVijay Mahadevan 
378efd17f3eSVijay Mahadevan   Level: intermediate
379efd17f3eSVijay Mahadevan 
380efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array
381efd17f3eSVijay Mahadevan 
382efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
383efd17f3eSVijay Mahadevan @*/
384efd17f3eSVijay Mahadevan PetscErrorCode  DMMoabVecGetArray(DM dm,Vec vec,void* array)
385efd17f3eSVijay Mahadevan {
386efd17f3eSVijay Mahadevan   DM_Moab        *moab;
387efd17f3eSVijay Mahadevan   moab::ErrorCode merr;
388efd17f3eSVijay Mahadevan   PetscErrorCode  ierr;
389efd17f3eSVijay Mahadevan   PetscInt        count;
390efd17f3eSVijay Mahadevan   moab::Tag       vtag;
391efd17f3eSVijay Mahadevan   PetscScalar    **varray;
392efd17f3eSVijay Mahadevan 
393efd17f3eSVijay Mahadevan   PetscFunctionBegin;
394efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
395efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
396efd17f3eSVijay Mahadevan   moab=(DM_Moab*)dm->data;
397efd17f3eSVijay Mahadevan 
398efd17f3eSVijay Mahadevan   /* Get the MOAB private data */
399efd17f3eSVijay Mahadevan   ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr);
400efd17f3eSVijay Mahadevan 
401efd17f3eSVijay Mahadevan   /* Get the real scalar array handle */
402efd17f3eSVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
403efd17f3eSVijay Mahadevan 
404efd17f3eSVijay Mahadevan   /* Get the array data for local entities */
405efd17f3eSVijay Mahadevan   merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr);
406efd17f3eSVijay Mahadevan   if (count!=(PetscInt)moab->vlocal->size()) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,moab->vlocal->size());
407efd17f3eSVijay Mahadevan 
408efd17f3eSVijay Mahadevan   merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr);
409efd17f3eSVijay Mahadevan   PetscFunctionReturn(0);
410efd17f3eSVijay Mahadevan }
411efd17f3eSVijay Mahadevan 
412efd17f3eSVijay Mahadevan 
413efd17f3eSVijay Mahadevan #undef __FUNCT__
414efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray"
415efd17f3eSVijay Mahadevan /*@
416efd17f3eSVijay Mahadevan   DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray
417efd17f3eSVijay Mahadevan 
418efd17f3eSVijay Mahadevan   Collective on MPI_Comm
419efd17f3eSVijay Mahadevan 
420efd17f3eSVijay Mahadevan   Input Parameter:
421efd17f3eSVijay Mahadevan + dm              - The DMMoab object being set
422efd17f3eSVijay Mahadevan . vec             - The Vector whose underlying data is requested
423efd17f3eSVijay Mahadevan - array           - The local data array
424efd17f3eSVijay Mahadevan 
425efd17f3eSVijay Mahadevan   Level: intermediate
426efd17f3eSVijay Mahadevan 
427efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array
428efd17f3eSVijay Mahadevan 
429efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead()
430efd17f3eSVijay Mahadevan @*/
431efd17f3eSVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArray(DM dm,Vec v,void* array)
432efd17f3eSVijay Mahadevan {
433efd17f3eSVijay Mahadevan   DM_Moab        *moab;
434efd17f3eSVijay Mahadevan   moab::ErrorCode merr;
435efd17f3eSVijay Mahadevan   PetscErrorCode  ierr;
436efd17f3eSVijay Mahadevan   moab::Tag       vtag;
437efd17f3eSVijay Mahadevan 
438efd17f3eSVijay Mahadevan   PetscFunctionBegin;
439efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
440efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
441efd17f3eSVijay Mahadevan   moab=(DM_Moab*)dm->data;
442efd17f3eSVijay Mahadevan 
443efd17f3eSVijay Mahadevan   /* Get the MOAB private data */
444efd17f3eSVijay Mahadevan   ierr = DMMoabGetVecTag(v,&vtag);CHKERRQ(ierr);
445efd17f3eSVijay Mahadevan 
446efd17f3eSVijay Mahadevan   /* reduce the tags correctly -> should probably let the user choose how to reduce in the future
447efd17f3eSVijay Mahadevan      For all FEM residual based assembly calculations, MPI_SUM should serve well
448efd17f3eSVijay Mahadevan   */
449efd17f3eSVijay Mahadevan   merr = moab->pcomm->reduce_tags(vtag,MPI_SUM,*moab->vghost);MBERRNM(merr);
450efd17f3eSVijay Mahadevan   PetscFunctionReturn(0);
451efd17f3eSVijay Mahadevan }
452efd17f3eSVijay Mahadevan 
453efd17f3eSVijay Mahadevan #undef __FUNCT__
454efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArrayRead"
455efd17f3eSVijay Mahadevan /*@
456efd17f3eSVijay 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
457efd17f3eSVijay Mahadevan 
458efd17f3eSVijay Mahadevan   Collective on MPI_Comm
459efd17f3eSVijay Mahadevan 
460efd17f3eSVijay Mahadevan   Input Parameter:
461efd17f3eSVijay Mahadevan + dm              - The DMMoab object being set
462efd17f3eSVijay Mahadevan . vec             - The Vector whose underlying data is requested
463efd17f3eSVijay Mahadevan 
464efd17f3eSVijay Mahadevan   Output Parameter:
465efd17f3eSVijay Mahadevan . array           - The local data array
466efd17f3eSVijay Mahadevan 
467efd17f3eSVijay Mahadevan   Level: intermediate
468efd17f3eSVijay Mahadevan 
469efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array
470efd17f3eSVijay Mahadevan 
471efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
472efd17f3eSVijay Mahadevan @*/
473efd17f3eSVijay Mahadevan PetscErrorCode  DMMoabVecGetArrayRead(DM dm,Vec vec,void* array)
474efd17f3eSVijay Mahadevan {
475efd17f3eSVijay Mahadevan   DM_Moab        *moab;
476efd17f3eSVijay Mahadevan   moab::ErrorCode merr;
477efd17f3eSVijay Mahadevan   PetscErrorCode  ierr;
478efd17f3eSVijay Mahadevan   PetscInt        count;
479efd17f3eSVijay Mahadevan   moab::Tag       vtag;
480efd17f3eSVijay Mahadevan   PetscScalar    **varray;
481efd17f3eSVijay Mahadevan 
482efd17f3eSVijay Mahadevan   PetscFunctionBegin;
483efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
484efd17f3eSVijay Mahadevan   PetscValidHeaderSpecific(vec,VEC_CLASSID,2);
485efd17f3eSVijay Mahadevan   moab=(DM_Moab*)dm->data;
486efd17f3eSVijay Mahadevan 
487efd17f3eSVijay Mahadevan   /* Get the MOAB private data */
488efd17f3eSVijay Mahadevan   ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr);
489efd17f3eSVijay Mahadevan 
490efd17f3eSVijay Mahadevan   /* Get the real scalar array handle */
491efd17f3eSVijay Mahadevan   varray = reinterpret_cast<PetscScalar**>(array);
492efd17f3eSVijay Mahadevan 
493efd17f3eSVijay Mahadevan   /* Get the array data for local entities */
494efd17f3eSVijay Mahadevan   merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr);
495efd17f3eSVijay Mahadevan   if (count!=(PetscInt)moab->vlocal->size()) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,moab->vlocal->size());
496efd17f3eSVijay Mahadevan 
497efd17f3eSVijay Mahadevan   merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr);
498efd17f3eSVijay Mahadevan   PetscFunctionReturn(0);
499efd17f3eSVijay Mahadevan }
500efd17f3eSVijay Mahadevan 
501efd17f3eSVijay Mahadevan 
502efd17f3eSVijay Mahadevan #undef __FUNCT__
503efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArrayRead"
504efd17f3eSVijay Mahadevan /*@
505efd17f3eSVijay Mahadevan   DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray
506efd17f3eSVijay Mahadevan 
507efd17f3eSVijay Mahadevan   Collective on MPI_Comm
508efd17f3eSVijay Mahadevan 
509efd17f3eSVijay Mahadevan   Input Parameter:
510efd17f3eSVijay Mahadevan + dm              - The DMMoab object being set
511efd17f3eSVijay Mahadevan . vec             - The Vector whose underlying data is requested
512efd17f3eSVijay Mahadevan - array           - The local data array
513efd17f3eSVijay Mahadevan 
514efd17f3eSVijay Mahadevan   Level: intermediate
515efd17f3eSVijay Mahadevan 
516efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array
517efd17f3eSVijay Mahadevan 
518efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray()
519efd17f3eSVijay Mahadevan @*/
520efd17f3eSVijay Mahadevan PetscErrorCode  DMMoabVecRestoreArrayRead(DM dm,Vec v,void* array)
521efd17f3eSVijay Mahadevan {
522efd17f3eSVijay Mahadevan   PetscFunctionBegin;
523efd17f3eSVijay Mahadevan   /* Nothing to do -> do not free the array memory obtained from tag_iterate */
524efd17f3eSVijay Mahadevan   PetscFunctionReturn(0);
525efd17f3eSVijay Mahadevan }
526efd17f3eSVijay Mahadevan 
527efd17f3eSVijay Mahadevan 
528efd17f3eSVijay Mahadevan #undef __FUNCT__
529db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab"
530db66d124SVijay Mahadevan PetscErrorCode  DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l)
531db66d124SVijay Mahadevan {
532db66d124SVijay Mahadevan   PetscErrorCode    ierr;
533db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
534db66d124SVijay Mahadevan 
535db66d124SVijay Mahadevan   PetscFunctionBegin;
536db66d124SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr);
537db66d124SVijay Mahadevan   PetscFunctionReturn(0);
538db66d124SVijay Mahadevan }
539db66d124SVijay Mahadevan 
540db66d124SVijay Mahadevan 
541db66d124SVijay Mahadevan #undef __FUNCT__
542db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab"
543db66d124SVijay Mahadevan PetscErrorCode  DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l)
544db66d124SVijay Mahadevan {
545db66d124SVijay Mahadevan   PetscErrorCode    ierr;
546db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
547db66d124SVijay Mahadevan 
548db66d124SVijay Mahadevan   PetscFunctionBegin;
549db66d124SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr);
550db66d124SVijay Mahadevan   PetscFunctionReturn(0);
551db66d124SVijay Mahadevan }
552db66d124SVijay Mahadevan 
553db66d124SVijay Mahadevan 
554db66d124SVijay Mahadevan #undef __FUNCT__
555db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab"
556db66d124SVijay Mahadevan PetscErrorCode  DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g)
557db66d124SVijay Mahadevan {
558db66d124SVijay Mahadevan   PetscErrorCode    ierr;
559db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
560db66d124SVijay Mahadevan 
561db66d124SVijay Mahadevan   PetscFunctionBegin;
562db66d124SVijay Mahadevan   ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr);
563db66d124SVijay Mahadevan   PetscFunctionReturn(0);
564db66d124SVijay Mahadevan }
565db66d124SVijay Mahadevan 
566db66d124SVijay Mahadevan 
567db66d124SVijay Mahadevan #undef __FUNCT__
568db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab"
569db66d124SVijay Mahadevan PetscErrorCode  DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g)
570db66d124SVijay Mahadevan {
571db66d124SVijay Mahadevan   PetscErrorCode    ierr;
572db66d124SVijay Mahadevan   DM_Moab         *dmmoab = (DM_Moab*)dm->data;
573db66d124SVijay Mahadevan 
574db66d124SVijay Mahadevan   PetscFunctionBegin;
575db66d124SVijay Mahadevan   ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr);
576db66d124SVijay Mahadevan   PetscFunctionReturn(0);
577db66d124SVijay Mahadevan }
578db66d124SVijay Mahadevan 
579db66d124SVijay Mahadevan 
580