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