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 7f6829af0SVijay Mahadevan /* declare some private DMMoab specific overrides */ 8f6829af0SVijay Mahadevan static PetscErrorCode DMCreateVector_Moab_Private(DM dm,moab::Tag tag,const moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec); 9f6829af0SVijay Mahadevan static PetscErrorCode DMVecUserDestroy_Moab(void *user); 10f6829af0SVijay Mahadevan static PetscErrorCode DMVecDuplicate_Moab(Vec x,Vec *y); 11f6829af0SVijay Mahadevan static PetscErrorCode DMVecCreateTagName_Moab_Private(moab::ParallelComm *pcomm,char** tag_name); 12032b8ab6SVijay Mahadevan 13b8ecf6d3SVijay Mahadevan /*@C 14f6829af0SVijay Mahadevan DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities 15f6829af0SVijay Mahadevan 16f6829af0SVijay Mahadevan Collective on MPI_Comm 17f6829af0SVijay Mahadevan 18f6829af0SVijay Mahadevan Input Parameter: 19b8ecf6d3SVijay Mahadevan + dm - The DMMoab object being set 20f6829af0SVijay Mahadevan . tag - If non-zero, block size will be taken from the tag size 21f6829af0SVijay Mahadevan . range - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab 22f6829af0SVijay Mahadevan . is_global_vec - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one 23f6829af0SVijay Mahadevan . destroy_tag - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB 24f6829af0SVijay Mahadevan 25f6829af0SVijay Mahadevan Output Parameter: 26f6829af0SVijay Mahadevan . vec - The created vector 27f6829af0SVijay Mahadevan 28f6829af0SVijay Mahadevan Level: beginner 29f6829af0SVijay Mahadevan 30b8ecf6d3SVijay Mahadevan .keywords: Vec, create 31b8ecf6d3SVijay Mahadevan 32b8ecf6d3SVijay Mahadevan .seealso: VecCreate() 33f6829af0SVijay Mahadevan @*/ 34f6829af0SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,const moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 35f6829af0SVijay Mahadevan { 36f6829af0SVijay Mahadevan PetscErrorCode ierr; 37f6829af0SVijay Mahadevan 38f6829af0SVijay Mahadevan PetscFunctionBegin; 39f6829af0SVijay Mahadevan if(!tag && (!range || range->empty())) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag and range cannot be null."); 40f6829af0SVijay Mahadevan 41f6829af0SVijay Mahadevan ierr = DMCreateVector_Moab_Private(dm,tag,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr); 42f6829af0SVijay Mahadevan PetscFunctionReturn(0); 43f6829af0SVijay Mahadevan } 44f6829af0SVijay Mahadevan 45f6829af0SVijay Mahadevan 46b8ecf6d3SVijay Mahadevan /*@C 47f6829af0SVijay Mahadevan DMMoabGetVecTag - Get the MOAB tag associated with this Vec 48f6829af0SVijay Mahadevan 49f6829af0SVijay Mahadevan Input Parameter: 50f6829af0SVijay Mahadevan . vec - Vec being queried 51f6829af0SVijay Mahadevan 52f6829af0SVijay Mahadevan Output Parameter: 53f6829af0SVijay Mahadevan . tag - Tag associated with this Vec. NULL if vec is a native PETSc Vec. 54f6829af0SVijay Mahadevan 55f6829af0SVijay Mahadevan Level: beginner 56f6829af0SVijay Mahadevan 57b8ecf6d3SVijay Mahadevan .keywords: DMMoab, MOAB tag 58b8ecf6d3SVijay Mahadevan 59b8ecf6d3SVijay Mahadevan .seealso: DMMoabCreateVector(), DMMoabGetVecRange() 60f6829af0SVijay Mahadevan @*/ 61f6829af0SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag) 62f6829af0SVijay Mahadevan { 63f6829af0SVijay Mahadevan PetscContainer moabdata; 64f6829af0SVijay Mahadevan Vec_MOAB *vmoab; 65f6829af0SVijay Mahadevan PetscErrorCode ierr; 66f6829af0SVijay Mahadevan 67f6829af0SVijay Mahadevan PetscFunctionBegin; 68f6829af0SVijay Mahadevan PetscValidPointer(tag,2); 69f6829af0SVijay Mahadevan 70f6829af0SVijay Mahadevan /* Get the MOAB private data */ 71f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 72f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 73f6829af0SVijay Mahadevan 74f6829af0SVijay Mahadevan *tag = vmoab->tag; 75f6829af0SVijay Mahadevan PetscFunctionReturn(0); 76f6829af0SVijay Mahadevan } 77f6829af0SVijay Mahadevan 78f6829af0SVijay Mahadevan 79b8ecf6d3SVijay Mahadevan /*@C 80f6829af0SVijay Mahadevan DMMoabGetVecRange - Get the MOAB entities associated with this Vec 81f6829af0SVijay Mahadevan 82f6829af0SVijay Mahadevan Input Parameter: 83f6829af0SVijay Mahadevan . vec - Vec being queried 84f6829af0SVijay Mahadevan 85f6829af0SVijay Mahadevan Output Parameter: 86f6829af0SVijay Mahadevan . range - Entities associated with this Vec. NULL if vec is a native PETSc Vec. 87f6829af0SVijay Mahadevan 88f6829af0SVijay Mahadevan Level: beginner 89f6829af0SVijay Mahadevan 90b8ecf6d3SVijay Mahadevan .keywords: DMMoab, MOAB range 91b8ecf6d3SVijay Mahadevan 92b8ecf6d3SVijay Mahadevan .seealso: DMMoabCreateVector(), DMMoabGetVecTag() 93f6829af0SVijay Mahadevan @*/ 94f6829af0SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range) 95f6829af0SVijay Mahadevan { 96f6829af0SVijay Mahadevan PetscContainer moabdata; 97f6829af0SVijay Mahadevan Vec_MOAB *vmoab; 98f6829af0SVijay Mahadevan PetscErrorCode ierr; 99f6829af0SVijay Mahadevan 100f6829af0SVijay Mahadevan PetscFunctionBegin; 101f6829af0SVijay Mahadevan PetscValidPointer(range,2); 102f6829af0SVijay Mahadevan 103f6829af0SVijay Mahadevan /* Get the MOAB private data handle */ 104f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 105f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 106f6829af0SVijay Mahadevan 107f6829af0SVijay Mahadevan *range = *vmoab->tag_range; 108f6829af0SVijay Mahadevan PetscFunctionReturn(0); 109f6829af0SVijay Mahadevan } 110f6829af0SVijay Mahadevan 111f6829af0SVijay Mahadevan 112b8ecf6d3SVijay Mahadevan /*@C 113f6829af0SVijay 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 114f6829af0SVijay Mahadevan 115f6829af0SVijay Mahadevan Collective on MPI_Comm 116f6829af0SVijay Mahadevan 117f6829af0SVijay Mahadevan Input Parameter: 118f6829af0SVijay Mahadevan . dm - The DMMoab object being set 119f6829af0SVijay Mahadevan . vec - The Vector whose underlying data is requested 120f6829af0SVijay Mahadevan 121f6829af0SVijay Mahadevan Output Parameter: 122f6829af0SVijay Mahadevan . array - The local data array 123f6829af0SVijay Mahadevan 124f6829af0SVijay Mahadevan Level: intermediate 125f6829af0SVijay Mahadevan 126f6829af0SVijay Mahadevan .keywords: MOAB, distributed array 127f6829af0SVijay Mahadevan 128f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 129f6829af0SVijay Mahadevan @*/ 130f6829af0SVijay Mahadevan PetscErrorCode DMMoabVecGetArray(DM dm,Vec vec,void* array) 131f6829af0SVijay Mahadevan { 132f6829af0SVijay Mahadevan DM_Moab *dmmoab; 133f6829af0SVijay Mahadevan moab::ErrorCode merr; 134f6829af0SVijay Mahadevan PetscErrorCode ierr; 135f6829af0SVijay Mahadevan PetscInt count,i,f; 136f6829af0SVijay Mahadevan moab::Tag vtag; 137f6829af0SVijay Mahadevan PetscScalar **varray; 138f6829af0SVijay Mahadevan PetscScalar *marray; 139f6829af0SVijay Mahadevan PetscContainer moabdata; 140f6829af0SVijay Mahadevan Vec_MOAB *vmoab,*xmoab; 141f6829af0SVijay Mahadevan 142f6829af0SVijay Mahadevan PetscFunctionBegin; 143f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 144f6829af0SVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 145f6829af0SVijay Mahadevan PetscValidPointer(array,3); 146f6829af0SVijay Mahadevan dmmoab=(DM_Moab*)dm->data; 147f6829af0SVijay Mahadevan 148f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 149f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 150f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 151f6829af0SVijay Mahadevan 152f6829af0SVijay Mahadevan /* Get the real scalar array handle */ 153f6829af0SVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 154f6829af0SVijay Mahadevan 155f6829af0SVijay Mahadevan if (vmoab->is_native_vec) { 156f6829af0SVijay Mahadevan 157f6829af0SVijay Mahadevan /* get the local representation of the arrays from Vectors */ 158f6829af0SVijay Mahadevan ierr = DMGetLocalVector(dm,&vmoab->local);CHKERRQ(ierr); 159f6829af0SVijay Mahadevan ierr = DMGlobalToLocalBegin(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr); 160f6829af0SVijay Mahadevan ierr = DMGlobalToLocalEnd(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr); 161f6829af0SVijay Mahadevan 162f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 163f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 164f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr); 165f6829af0SVijay Mahadevan 166f6829af0SVijay Mahadevan /* get the local representation of the arrays from Vectors */ 167f6829af0SVijay Mahadevan ierr = VecGhostGetLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr); 168f6829af0SVijay Mahadevan ierr = VecGhostUpdateBegin(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 169f6829af0SVijay Mahadevan ierr = VecGhostUpdateEnd(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 170f6829af0SVijay Mahadevan 171f6829af0SVijay Mahadevan ierr = VecGetArray(xmoab->local, varray);CHKERRQ(ierr); 172f6829af0SVijay Mahadevan } 173f6829af0SVijay Mahadevan else { 174f6829af0SVijay Mahadevan 175f6829af0SVijay Mahadevan /* Get the MOAB private data */ 176f6829af0SVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 177f6829af0SVijay Mahadevan 178f6829af0SVijay Mahadevan /* exchange the data into ghost cells first */ 179f6829af0SVijay Mahadevan merr = dmmoab->pcomm->exchange_tags(vtag,*dmmoab->vlocal);MBERRNM(merr); 180f6829af0SVijay Mahadevan 1817ae5e5b6SVijay Mahadevan ierr = PetscMalloc1((dmmoab->nloc+dmmoab->nghost)*dmmoab->numFields,varray);CHKERRQ(ierr); 182f6829af0SVijay Mahadevan 183f6829af0SVijay Mahadevan /* Get the array data for local entities */ 184f6829af0SVijay Mahadevan merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr); 185f6829af0SVijay 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); 186f6829af0SVijay Mahadevan 187f6829af0SVijay Mahadevan i=0; 188f6829af0SVijay Mahadevan for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) { 189f6829af0SVijay Mahadevan for (f=0;f<dmmoab->numFields;f++,i++) 190f6829af0SVijay Mahadevan (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f]=marray[i]; 191f6829af0SVijay Mahadevan } 192f6829af0SVijay Mahadevan } 193f6829af0SVijay Mahadevan PetscFunctionReturn(0); 194f6829af0SVijay Mahadevan } 195f6829af0SVijay Mahadevan 196f6829af0SVijay Mahadevan 197b8ecf6d3SVijay Mahadevan /*@C 198f6829af0SVijay Mahadevan DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray 199f6829af0SVijay Mahadevan 200f6829af0SVijay Mahadevan Collective on MPI_Comm 201f6829af0SVijay Mahadevan 202f6829af0SVijay Mahadevan Input Parameter: 203f6829af0SVijay Mahadevan + dm - The DMMoab object being set 204f6829af0SVijay Mahadevan . vec - The Vector whose underlying data is requested 205b8ecf6d3SVijay Mahadevan . array - The local data array 206f6829af0SVijay Mahadevan 207f6829af0SVijay Mahadevan Level: intermediate 208f6829af0SVijay Mahadevan 209f6829af0SVijay Mahadevan .keywords: MOAB, distributed array 210f6829af0SVijay Mahadevan 211f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 212f6829af0SVijay Mahadevan @*/ 213f6829af0SVijay Mahadevan PetscErrorCode DMMoabVecRestoreArray(DM dm,Vec vec,void* array) 214f6829af0SVijay Mahadevan { 215f6829af0SVijay Mahadevan DM_Moab *dmmoab; 216f6829af0SVijay Mahadevan moab::ErrorCode merr; 217f6829af0SVijay Mahadevan PetscErrorCode ierr; 218f6829af0SVijay Mahadevan moab::Tag vtag; 219f6829af0SVijay Mahadevan PetscInt count,i,f; 220f6829af0SVijay Mahadevan PetscScalar **varray; 221f6829af0SVijay Mahadevan PetscScalar *marray; 222f6829af0SVijay Mahadevan PetscContainer moabdata; 223f6829af0SVijay Mahadevan Vec_MOAB *vmoab,*xmoab; 224f6829af0SVijay Mahadevan 225f6829af0SVijay Mahadevan PetscFunctionBegin; 226f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 227f6829af0SVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 228f6829af0SVijay Mahadevan PetscValidPointer(array,3); 229f6829af0SVijay Mahadevan dmmoab=(DM_Moab*)dm->data; 230f6829af0SVijay Mahadevan 231f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 232f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 233f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 234f6829af0SVijay Mahadevan 235f6829af0SVijay Mahadevan /* Get the real scalar array handle */ 236f6829af0SVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 237f6829af0SVijay Mahadevan 238f6829af0SVijay Mahadevan if (vmoab->is_native_vec) { 239f6829af0SVijay Mahadevan 240f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 241f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 242f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr); 243f6829af0SVijay Mahadevan 244f6829af0SVijay Mahadevan /* get the local representation of the arrays from Vectors */ 245f6829af0SVijay Mahadevan ierr = VecRestoreArray(xmoab->local, varray);CHKERRQ(ierr); 246f6829af0SVijay Mahadevan ierr = VecGhostUpdateBegin(vmoab->local,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 247f6829af0SVijay Mahadevan ierr = VecGhostUpdateEnd(vmoab->local,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 248f6829af0SVijay Mahadevan ierr = VecGhostRestoreLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr); 249f6829af0SVijay Mahadevan 250f6829af0SVijay Mahadevan /* restore local pieces */ 251f6829af0SVijay Mahadevan ierr = DMLocalToGlobalBegin(dm,vmoab->local,INSERT_VALUES,vec);CHKERRQ(ierr); 252f6829af0SVijay Mahadevan ierr = DMLocalToGlobalEnd(dm,vmoab->local,INSERT_VALUES,vec);CHKERRQ(ierr); 253f6829af0SVijay Mahadevan ierr = DMRestoreLocalVector(dm, &vmoab->local);CHKERRQ(ierr); 254f6829af0SVijay Mahadevan } 255f6829af0SVijay Mahadevan else { 256f6829af0SVijay Mahadevan 257f6829af0SVijay Mahadevan /* Get the MOAB private data */ 258f6829af0SVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 259f6829af0SVijay Mahadevan 260f6829af0SVijay Mahadevan /* Get the array data for local entities */ 261f6829af0SVijay Mahadevan merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr); 262f6829af0SVijay 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); 263f6829af0SVijay Mahadevan 264f6829af0SVijay Mahadevan i=0; 265f6829af0SVijay Mahadevan for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) { 266f6829af0SVijay Mahadevan for (f=0;f<dmmoab->numFields;f++,i++) 267f6829af0SVijay Mahadevan marray[i] = (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f]; 268f6829af0SVijay Mahadevan } 269f6829af0SVijay Mahadevan 270f6829af0SVijay Mahadevan /* reduce the tags correctly -> should probably let the user choose how to reduce in the future 271f6829af0SVijay Mahadevan For all FEM residual based assembly calculations, MPI_SUM should serve well */ 272f6829af0SVijay Mahadevan merr = dmmoab->pcomm->reduce_tags(vtag,MPI_SUM,*dmmoab->vlocal);MBERRV(dmmoab->mbiface,merr); 273f6829af0SVijay Mahadevan 274f6829af0SVijay Mahadevan ierr = PetscFree(*varray);CHKERRQ(ierr); 275f6829af0SVijay Mahadevan } 276f6829af0SVijay Mahadevan PetscFunctionReturn(0); 277f6829af0SVijay Mahadevan } 278f6829af0SVijay Mahadevan 279b8ecf6d3SVijay Mahadevan /*@C 280f6829af0SVijay 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 281f6829af0SVijay Mahadevan 282f6829af0SVijay Mahadevan Collective on MPI_Comm 283f6829af0SVijay Mahadevan 284f6829af0SVijay Mahadevan Input Parameter: 285f6829af0SVijay Mahadevan + dm - The DMMoab object being set 286f6829af0SVijay Mahadevan . vec - The Vector whose underlying data is requested 287f6829af0SVijay Mahadevan 288f6829af0SVijay Mahadevan Output Parameter: 289f6829af0SVijay Mahadevan . array - The local data array 290f6829af0SVijay Mahadevan 291f6829af0SVijay Mahadevan Level: intermediate 292f6829af0SVijay Mahadevan 293f6829af0SVijay Mahadevan .keywords: MOAB, distributed array 294f6829af0SVijay Mahadevan 295f6829af0SVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 296f6829af0SVijay Mahadevan @*/ 297f6829af0SVijay Mahadevan PetscErrorCode DMMoabVecGetArrayRead(DM dm,Vec vec,void* array) 298f6829af0SVijay Mahadevan { 299f6829af0SVijay Mahadevan DM_Moab *dmmoab; 300f6829af0SVijay Mahadevan moab::ErrorCode merr; 301f6829af0SVijay Mahadevan PetscErrorCode ierr; 302f6829af0SVijay Mahadevan PetscInt count,i,f; 303f6829af0SVijay Mahadevan moab::Tag vtag; 304f6829af0SVijay Mahadevan PetscScalar **varray; 305f6829af0SVijay Mahadevan PetscScalar *marray; 306f6829af0SVijay Mahadevan PetscContainer moabdata; 307f6829af0SVijay Mahadevan Vec_MOAB *vmoab,*xmoab; 308f6829af0SVijay Mahadevan 309f6829af0SVijay Mahadevan PetscFunctionBegin; 310f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 311f6829af0SVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 312f6829af0SVijay Mahadevan PetscValidPointer(array,3); 313f6829af0SVijay Mahadevan dmmoab=(DM_Moab*)dm->data; 314f6829af0SVijay Mahadevan 315f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 316f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 317f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 318f6829af0SVijay Mahadevan 319f6829af0SVijay Mahadevan /* Get the real scalar array handle */ 320f6829af0SVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 321f6829af0SVijay Mahadevan 322f6829af0SVijay Mahadevan if (vmoab->is_native_vec) { 323f6829af0SVijay Mahadevan /* get the local representation of the arrays from Vectors */ 324f6829af0SVijay Mahadevan ierr = DMGetLocalVector(dm,&vmoab->local);CHKERRQ(ierr); 325f6829af0SVijay Mahadevan ierr = DMGlobalToLocalBegin(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr); 326f6829af0SVijay Mahadevan ierr = DMGlobalToLocalEnd(dm,vec,INSERT_VALUES,vmoab->local);CHKERRQ(ierr); 327f6829af0SVijay Mahadevan 328f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 329f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 330f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr); 331f6829af0SVijay Mahadevan 332f6829af0SVijay Mahadevan /* get the local representation of the arrays from Vectors */ 333f6829af0SVijay Mahadevan ierr = VecGhostGetLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr); 334f6829af0SVijay Mahadevan ierr = VecGhostUpdateBegin(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 335f6829af0SVijay Mahadevan ierr = VecGhostUpdateEnd(vmoab->local,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 336f6829af0SVijay Mahadevan ierr = VecGetArray(xmoab->local, varray);CHKERRQ(ierr); 337f6829af0SVijay Mahadevan } 338f6829af0SVijay Mahadevan else { 339f6829af0SVijay Mahadevan /* Get the MOAB private data */ 340f6829af0SVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 341f6829af0SVijay Mahadevan 342f6829af0SVijay Mahadevan /* exchange the data into ghost cells first */ 343f6829af0SVijay Mahadevan merr = dmmoab->pcomm->exchange_tags(vtag,*dmmoab->vlocal);MBERRNM(merr); 344f6829af0SVijay Mahadevan 3457ae5e5b6SVijay Mahadevan ierr = PetscMalloc1((dmmoab->nloc+dmmoab->nghost)*dmmoab->numFields,varray);CHKERRQ(ierr); 346f6829af0SVijay Mahadevan 347f6829af0SVijay Mahadevan /* Get the array data for local entities */ 348f6829af0SVijay Mahadevan merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(marray),false);MBERRNM(merr); 349f6829af0SVijay 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); 350f6829af0SVijay Mahadevan 351f6829af0SVijay Mahadevan i=0; 352f6829af0SVijay Mahadevan for(moab::Range::iterator iter = dmmoab->vlocal->begin(); iter != dmmoab->vlocal->end(); iter++) { 353f6829af0SVijay Mahadevan for (f=0;f<dmmoab->numFields;f++,i++) 354f6829af0SVijay Mahadevan (*varray)[dmmoab->lidmap[(PetscInt)*iter]*dmmoab->numFields+f]=marray[i]; 355f6829af0SVijay Mahadevan } 356f6829af0SVijay Mahadevan } 357f6829af0SVijay Mahadevan PetscFunctionReturn(0); 358f6829af0SVijay Mahadevan } 359f6829af0SVijay Mahadevan 360f6829af0SVijay Mahadevan 361b8ecf6d3SVijay Mahadevan /*@C 362f6829af0SVijay Mahadevan DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray 363f6829af0SVijay Mahadevan 364f6829af0SVijay Mahadevan Collective on MPI_Comm 365f6829af0SVijay Mahadevan 366f6829af0SVijay Mahadevan Input Parameter: 367f6829af0SVijay Mahadevan + dm - The DMMoab object being set 368f6829af0SVijay Mahadevan . vec - The Vector whose underlying data is requested 369b8ecf6d3SVijay Mahadevan . array - The local data array 370f6829af0SVijay Mahadevan 371f6829af0SVijay Mahadevan Level: intermediate 372f6829af0SVijay Mahadevan 373f6829af0SVijay Mahadevan .keywords: MOAB, distributed array 374f6829af0SVijay Mahadevan 375f6829af0SVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 376f6829af0SVijay Mahadevan @*/ 377f6829af0SVijay Mahadevan PetscErrorCode DMMoabVecRestoreArrayRead(DM dm,Vec vec,void* array) 378f6829af0SVijay Mahadevan { 379f6829af0SVijay Mahadevan PetscErrorCode ierr; 380f6829af0SVijay Mahadevan PetscScalar **varray; 381f6829af0SVijay Mahadevan PetscContainer moabdata; 382f6829af0SVijay Mahadevan Vec_MOAB *vmoab,*xmoab; 383f6829af0SVijay Mahadevan 384f6829af0SVijay Mahadevan PetscFunctionBegin; 385f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 386f6829af0SVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 387f6829af0SVijay Mahadevan PetscValidPointer(array,3); 388f6829af0SVijay Mahadevan 389f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 390f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 391f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 392f6829af0SVijay Mahadevan 393f6829af0SVijay Mahadevan /* Get the real scalar array handle */ 394f6829af0SVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 395f6829af0SVijay Mahadevan 396f6829af0SVijay Mahadevan if (vmoab->is_native_vec) { 397f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 398f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vmoab->local,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 399f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&xmoab);CHKERRQ(ierr); 400f6829af0SVijay Mahadevan 401f6829af0SVijay Mahadevan /* restore the local representation of the arrays from Vectors */ 402f6829af0SVijay Mahadevan ierr = VecRestoreArray(xmoab->local, varray);CHKERRQ(ierr); 403f6829af0SVijay Mahadevan ierr = VecGhostRestoreLocalForm(vmoab->local,&xmoab->local);CHKERRQ(ierr); 404f6829af0SVijay Mahadevan 405f6829af0SVijay Mahadevan /* restore local pieces */ 406f6829af0SVijay Mahadevan ierr = DMRestoreLocalVector(dm, &vmoab->local);CHKERRQ(ierr); 407f6829af0SVijay Mahadevan } 408f6829af0SVijay Mahadevan else { 409f6829af0SVijay Mahadevan /* Nothing to do but just free the memory allocated before */ 410f6829af0SVijay Mahadevan ierr = PetscFree(*varray);CHKERRQ(ierr); 411f6829af0SVijay Mahadevan 412f6829af0SVijay Mahadevan } 413f6829af0SVijay Mahadevan PetscFunctionReturn(0); 414f6829af0SVijay Mahadevan } 415f6829af0SVijay Mahadevan 416f6829af0SVijay Mahadevan 417f6829af0SVijay Mahadevan PetscErrorCode DMCreateVector_Moab_Private(DM dm,moab::Tag tag,const moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 418032b8ab6SVijay Mahadevan { 419032b8ab6SVijay Mahadevan PetscErrorCode ierr; 420032b8ab6SVijay Mahadevan moab::ErrorCode merr; 421032b8ab6SVijay Mahadevan PetscBool is_newtag; 422351b8a77SVijay Mahadevan const moab::Range *range; 423bb8f3634SVijay Mahadevan PetscInt count,lnative_vec,gnative_vec; 4249088682fSVijay Mahadevan std::string ttname; 425c4427235SVijay Mahadevan PetscScalar *data_ptr,*defaultvals; 426032b8ab6SVijay Mahadevan 427db66d124SVijay Mahadevan Vec_MOAB *vmoab; 428032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 429032b8ab6SVijay Mahadevan moab::ParallelComm *pcomm = dmmoab->pcomm; 430032b8ab6SVijay Mahadevan moab::Interface *mbiface = dmmoab->mbiface; 431032b8ab6SVijay Mahadevan 432032b8ab6SVijay Mahadevan PetscFunctionBegin; 433d1839e75SVijay Mahadevan if(sizeof(PetscReal) != sizeof(PetscScalar)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "MOAB tags only support Real types (Complex-type unsupported)"); 434032b8ab6SVijay Mahadevan if(!userrange) range = dmmoab->vowned; 435032b8ab6SVijay Mahadevan else range = userrange; 436db66d124SVijay Mahadevan if(!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first."); 437032b8ab6SVijay Mahadevan 438bb8f3634SVijay Mahadevan /* If the tag data is in a single sequence, use PETSc native vector since tag_iterate isn't useful anymore */ 439bb8f3634SVijay Mahadevan lnative_vec=(range->psize()-1); 440bb8f3634SVijay Mahadevan 441f6829af0SVijay Mahadevan #if 0 4426465f021SVijay Mahadevan // lnative_vec=1; /* NOTE: Testing PETSc vector: will force to create native vector all the time */ 4436465f021SVijay Mahadevan // lnative_vec=0; /* NOTE: Testing MOAB vector: will force to create MOAB tag_iterate based vector all the time */ 444f6829af0SVijay Mahadevan #endif 445b2566f29SBarry Smith ierr = MPIU_Allreduce(&lnative_vec, &gnative_vec, 1, MPI_INT, MPI_MAX, pcomm->comm());CHKERRQ(ierr); 446bb8f3634SVijay Mahadevan 447bb8f3634SVijay Mahadevan /* Create the MOAB internal data object */ 448f90c3b0eSVijay Mahadevan ierr = PetscNew(&vmoab);CHKERRQ(ierr); 449bb8f3634SVijay Mahadevan vmoab->is_native_vec=(gnative_vec>0?PETSC_TRUE:PETSC_FALSE); 450bb8f3634SVijay Mahadevan 451bb8f3634SVijay Mahadevan if (!vmoab->is_native_vec) { 4529088682fSVijay Mahadevan merr = mbiface->tag_get_name(tag, ttname); 4539088682fSVijay Mahadevan if (!ttname.length() && merr !=moab::MB_SUCCESS) { 454db66d124SVijay Mahadevan /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */ 455*c528d872SBarry Smith char *tag_name = NULL; 456f6829af0SVijay Mahadevan ierr = DMVecCreateTagName_Moab_Private(pcomm,&tag_name);CHKERRQ(ierr); 457032b8ab6SVijay Mahadevan is_newtag = PETSC_TRUE; 458032b8ab6SVijay Mahadevan 459db66d124SVijay Mahadevan /* Create the default value for the tag (all zeros) */ 4607ae5e5b6SVijay Mahadevan ierr = PetscCalloc1(dmmoab->numFields,&defaultvals);CHKERRQ(ierr); 461032b8ab6SVijay Mahadevan 462db66d124SVijay Mahadevan /* Create the tag */ 463d1839e75SVijay Mahadevan merr = mbiface->tag_get_handle(tag_name,dmmoab->numFields,moab::MB_TYPE_DOUBLE,tag, 464c4427235SVijay Mahadevan moab::MB_TAG_DENSE|moab::MB_TAG_CREAT,defaultvals);MBERRNM(merr); 465032b8ab6SVijay Mahadevan ierr = PetscFree(tag_name);CHKERRQ(ierr); 466c4427235SVijay Mahadevan ierr = PetscFree(defaultvals);CHKERRQ(ierr); 467032b8ab6SVijay Mahadevan } 468032b8ab6SVijay Mahadevan else { 469db66d124SVijay Mahadevan /* Make sure the tag data is of type "double" */ 470032b8ab6SVijay Mahadevan moab::DataType tag_type; 471032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data_type(tag, tag_type);MBERRNM(merr); 472032b8ab6SVijay Mahadevan if(tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE"); 473032b8ab6SVijay Mahadevan is_newtag = destroy_tag; 474032b8ab6SVijay Mahadevan } 475032b8ab6SVijay Mahadevan 476032b8ab6SVijay Mahadevan vmoab->tag = tag; 477bb8f3634SVijay Mahadevan vmoab->new_tag = is_newtag; 478bb8f3634SVijay Mahadevan } 479032b8ab6SVijay Mahadevan vmoab->mbiface = mbiface; 480032b8ab6SVijay Mahadevan vmoab->pcomm = pcomm; 481032b8ab6SVijay Mahadevan vmoab->is_global_vec = is_global_vec; 482bb8f3634SVijay Mahadevan vmoab->tag_size=dmmoab->bs; 483032b8ab6SVijay Mahadevan 484bb8f3634SVijay Mahadevan if (vmoab->is_native_vec) { 485fc418013SVijay Mahadevan 486bb8f3634SVijay Mahadevan /* Create the PETSc Vector directly and attach our functions accordingly */ 487bb8f3634SVijay Mahadevan if (!is_global_vec) { 488bb8f3634SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 489addae81cSVijay Mahadevan ierr = VecCreateGhostBlock(pcomm->comm(),dmmoab->bs,dmmoab->numFields*dmmoab->nloc, 490addae81cSVijay Mahadevan dmmoab->numFields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],vec);CHKERRQ(ierr); 491bb8f3634SVijay Mahadevan } 492bb8f3634SVijay Mahadevan else { 493bb8f3634SVijay Mahadevan /* This is an MPI/SEQ Vector */ 494bb8f3634SVijay Mahadevan ierr = VecCreate(pcomm->comm(),vec);CHKERRQ(ierr); 495addae81cSVijay Mahadevan ierr = VecSetSizes(*vec,dmmoab->numFields*dmmoab->nloc,PETSC_DECIDE);CHKERRQ(ierr); 496bb8f3634SVijay Mahadevan ierr = VecSetBlockSize(*vec,dmmoab->bs);CHKERRQ(ierr); 497bb8f3634SVijay Mahadevan ierr = VecSetType(*vec, VECMPI);CHKERRQ(ierr); 498bb8f3634SVijay Mahadevan } 499bb8f3634SVijay Mahadevan } 500bb8f3634SVijay Mahadevan else { 501db66d124SVijay Mahadevan /* Call tag_iterate. This will cause MOAB to allocate memory for the 502db66d124SVijay Mahadevan tag data if it hasn't already happened */ 503032b8ab6SVijay Mahadevan merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr); 504032b8ab6SVijay Mahadevan 505bb8f3634SVijay Mahadevan /* set the reference for vector range */ 506bb8f3634SVijay Mahadevan vmoab->tag_range = new moab::Range(*range); 507addae81cSVijay Mahadevan merr = mbiface->tag_get_length(tag,dmmoab->numFields);MBERRNM(merr); 508032b8ab6SVijay Mahadevan 509db66d124SVijay Mahadevan /* Create the PETSc Vector 510db66d124SVijay Mahadevan Query MOAB mesh to check if there are any ghosted entities 511032b8ab6SVijay Mahadevan -> if we do, create a ghosted vector to map correctly to the same layout 512032b8ab6SVijay Mahadevan -> else, create a non-ghosted parallel vector */ 513bb8f3634SVijay Mahadevan if (!is_global_vec) { 514db66d124SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 515addae81cSVijay Mahadevan ierr = VecCreateGhostBlockWithArray(pcomm->comm(),dmmoab->bs,dmmoab->numFields*dmmoab->nloc, 516addae81cSVijay Mahadevan dmmoab->numFields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],data_ptr,vec);CHKERRQ(ierr); 517032b8ab6SVijay Mahadevan } 518032b8ab6SVijay Mahadevan else { 519bb8f3634SVijay Mahadevan /* This is an MPI Vector without ghosted padding */ 520addae81cSVijay Mahadevan ierr = VecCreateMPIWithArray(pcomm->comm(),dmmoab->bs,dmmoab->numFields*range->size(), 521bb8f3634SVijay Mahadevan PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr); 522032b8ab6SVijay Mahadevan } 523bb8f3634SVijay Mahadevan } 524bb8f3634SVijay Mahadevan ierr = VecSetFromOptions(*vec);CHKERRQ(ierr); 525032b8ab6SVijay Mahadevan 526f6829af0SVijay Mahadevan /* create a container and store the internal MOAB data for faster access based on Entities etc */ 527032b8ab6SVijay Mahadevan PetscContainer moabdata; 528bb8f3634SVijay Mahadevan ierr = PetscContainerCreate(PETSC_COMM_WORLD,&moabdata);CHKERRQ(ierr); 529032b8ab6SVijay Mahadevan ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr); 530f6829af0SVijay Mahadevan ierr = PetscContainerSetUserDestroy(moabdata,DMVecUserDestroy_Moab);CHKERRQ(ierr); 531032b8ab6SVijay Mahadevan ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr); 532f6829af0SVijay Mahadevan (*vec)->ops->duplicate = DMVecDuplicate_Moab; 533032b8ab6SVijay Mahadevan ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr); 534032b8ab6SVijay Mahadevan 535db66d124SVijay Mahadevan /* Vector created, manually set local to global mapping */ 5366d9eb265SVijay Mahadevan if (dmmoab->ltog_map) { 537bb8f3634SVijay Mahadevan ierr = VecSetLocalToGlobalMapping(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 5386d9eb265SVijay Mahadevan } 539032b8ab6SVijay Mahadevan 540032b8ab6SVijay Mahadevan /* set the DM reference to the vector */ 541032b8ab6SVijay Mahadevan ierr = VecSetDM(*vec, dm);CHKERRQ(ierr); 542032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 543032b8ab6SVijay Mahadevan } 544032b8ab6SVijay Mahadevan 545032b8ab6SVijay Mahadevan 546f6829af0SVijay Mahadevan /* DMVecCreateTagName_Moab_Private 547032b8ab6SVijay Mahadevan * 548032b8ab6SVijay Mahadevan * Creates a unique tag name that will be shared across processes. If 549032b8ab6SVijay Mahadevan * pcomm is NULL, then this is a serial vector. A unique tag name 550032b8ab6SVijay Mahadevan * will be returned in tag_name in either case. 551032b8ab6SVijay Mahadevan * 552032b8ab6SVijay Mahadevan * The tag names have the format _PETSC_VEC_N where N is some integer. 553032b8ab6SVijay Mahadevan * 554032b8ab6SVijay Mahadevan * NOTE: The tag_name is allocated in this routine; The user needs to free 555032b8ab6SVijay Mahadevan * the character array. 556032b8ab6SVijay Mahadevan */ 557f6829af0SVijay Mahadevan PetscErrorCode DMVecCreateTagName_Moab_Private(moab::ParallelComm *pcomm,char** tag_name) 558032b8ab6SVijay Mahadevan { 559032b8ab6SVijay Mahadevan moab::ErrorCode mberr; 560032b8ab6SVijay Mahadevan PetscErrorCode ierr; 561032b8ab6SVijay Mahadevan PetscInt n,global_n; 562032b8ab6SVijay Mahadevan moab::Tag indexTag; 563032b8ab6SVijay Mahadevan 564032b8ab6SVijay Mahadevan PetscFunctionBegin; 565032b8ab6SVijay Mahadevan const char* PVEC_PREFIX = "__PETSC_VEC_"; 56695dccacaSBarry Smith ierr = PetscMalloc1(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr); 567032b8ab6SVijay Mahadevan 568efd17f3eSVijay Mahadevan /* Check to see if there are any PETSc vectors defined */ 569032b8ab6SVijay Mahadevan moab::Interface *mbiface = pcomm->get_moab(); 570032b8ab6SVijay Mahadevan moab::EntityHandle rootset = mbiface->get_root_set(); 571032b8ab6SVijay Mahadevan 572efd17f3eSVijay Mahadevan /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */ 573032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag, 574032b8ab6SVijay Mahadevan moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr); 575032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n); 576032b8ab6SVijay Mahadevan if (mberr == moab::MB_TAG_NOT_FOUND) n=0; /* this is the first temporary vector */ 577032b8ab6SVijay Mahadevan else MBERRNM(mberr); 578032b8ab6SVijay Mahadevan 579032b8ab6SVijay Mahadevan /* increment the new value of n */ 580032b8ab6SVijay Mahadevan ++n; 581032b8ab6SVijay Mahadevan 582efd17f3eSVijay Mahadevan /* Make sure that n is consistent across all processes */ 583b2566f29SBarry Smith ierr = MPIU_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr); 584032b8ab6SVijay Mahadevan 585efd17f3eSVijay Mahadevan /* Set the new name accordingly and return */ 586032b8ab6SVijay Mahadevan ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr); 587032b8ab6SVijay Mahadevan mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr); 588032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 589032b8ab6SVijay Mahadevan } 590032b8ab6SVijay Mahadevan 591032b8ab6SVijay Mahadevan 592f6829af0SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec) 593f6829af0SVijay Mahadevan { 594f6829af0SVijay Mahadevan PetscErrorCode ierr; 595f6829af0SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 596f6829af0SVijay Mahadevan 597f6829af0SVijay Mahadevan PetscFunctionBegin; 598f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 599f6829af0SVijay Mahadevan PetscValidPointer(gvec,2); 600*c528d872SBarry Smith ierr = DMCreateVector_Moab_Private(dm,NULL,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr); 601f6829af0SVijay Mahadevan PetscFunctionReturn(0); 602f6829af0SVijay Mahadevan } 603f6829af0SVijay Mahadevan 604f6829af0SVijay Mahadevan 605f6829af0SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec) 606f6829af0SVijay Mahadevan { 607f6829af0SVijay Mahadevan PetscErrorCode ierr; 608f6829af0SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 609f6829af0SVijay Mahadevan 610f6829af0SVijay Mahadevan PetscFunctionBegin; 611f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 612f6829af0SVijay Mahadevan PetscValidPointer(lvec,2); 613*c528d872SBarry Smith ierr = DMCreateVector_Moab_Private(dm,NULL,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr); 614f6829af0SVijay Mahadevan PetscFunctionReturn(0); 615f6829af0SVijay Mahadevan } 616f6829af0SVijay Mahadevan 617f6829af0SVijay Mahadevan 618f6829af0SVijay Mahadevan PetscErrorCode DMVecDuplicate_Moab(Vec x,Vec *y) 619f6829af0SVijay Mahadevan { 620f6829af0SVijay Mahadevan PetscErrorCode ierr; 621f6829af0SVijay Mahadevan DM dm; 622f6829af0SVijay Mahadevan PetscContainer moabdata; 623f6829af0SVijay Mahadevan Vec_MOAB *vmoab; 624f6829af0SVijay Mahadevan 625f6829af0SVijay Mahadevan PetscFunctionBegin; 626f6829af0SVijay Mahadevan PetscValidHeaderSpecific(x,VEC_CLASSID,1); 627f6829af0SVijay Mahadevan PetscValidPointer(y,2); 628f6829af0SVijay Mahadevan 629f6829af0SVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 630f6829af0SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 631f6829af0SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 632f6829af0SVijay Mahadevan 633f6829af0SVijay Mahadevan ierr = VecGetDM(x, &dm);CHKERRQ(ierr); 634f6829af0SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 635f6829af0SVijay Mahadevan 636*c528d872SBarry Smith ierr = DMCreateVector_Moab_Private(dm,NULL,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr); 637f6829af0SVijay Mahadevan ierr = VecSetDM(*y, dm);CHKERRQ(ierr); 638f6829af0SVijay Mahadevan PetscFunctionReturn(0); 639f6829af0SVijay Mahadevan } 640f6829af0SVijay Mahadevan 641f6829af0SVijay Mahadevan 642f6829af0SVijay Mahadevan PetscErrorCode DMVecUserDestroy_Moab(void *user) 643032b8ab6SVijay Mahadevan { 644efd17f3eSVijay Mahadevan Vec_MOAB *vmoab=(Vec_MOAB*)user; 645032b8ab6SVijay Mahadevan PetscErrorCode ierr; 646032b8ab6SVijay Mahadevan moab::ErrorCode merr; 647032b8ab6SVijay Mahadevan 648032b8ab6SVijay Mahadevan PetscFunctionBegin; 649032b8ab6SVijay Mahadevan if(vmoab->new_tag && vmoab->tag) { 650efd17f3eSVijay Mahadevan /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */ 651032b8ab6SVijay Mahadevan merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr); 652032b8ab6SVijay Mahadevan } 653032b8ab6SVijay Mahadevan delete vmoab->tag_range; 654*c528d872SBarry Smith vmoab->tag = NULL; 655*c528d872SBarry Smith vmoab->mbiface = NULL; 656*c528d872SBarry Smith vmoab->pcomm = NULL; 657032b8ab6SVijay Mahadevan ierr = PetscFree(vmoab);CHKERRQ(ierr); 658032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 659032b8ab6SVijay Mahadevan } 660032b8ab6SVijay Mahadevan 661efd17f3eSVijay Mahadevan 662db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l) 663db66d124SVijay Mahadevan { 664db66d124SVijay Mahadevan PetscErrorCode ierr; 665db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 666db66d124SVijay Mahadevan 667db66d124SVijay Mahadevan PetscFunctionBegin; 6686d9eb265SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_REVERSE);CHKERRQ(ierr); 669db66d124SVijay Mahadevan PetscFunctionReturn(0); 670db66d124SVijay Mahadevan } 671db66d124SVijay Mahadevan 672db66d124SVijay Mahadevan 673db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l) 674db66d124SVijay Mahadevan { 675db66d124SVijay Mahadevan PetscErrorCode ierr; 676db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 677db66d124SVijay Mahadevan 678db66d124SVijay Mahadevan PetscFunctionBegin; 6796d9eb265SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_REVERSE);CHKERRQ(ierr); 680db66d124SVijay Mahadevan PetscFunctionReturn(0); 681db66d124SVijay Mahadevan } 682db66d124SVijay Mahadevan 683db66d124SVijay Mahadevan 684db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g) 685db66d124SVijay Mahadevan { 686db66d124SVijay Mahadevan PetscErrorCode ierr; 687db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 688db66d124SVijay Mahadevan 689db66d124SVijay Mahadevan PetscFunctionBegin; 6906d9eb265SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_FORWARD);CHKERRQ(ierr); 691db66d124SVijay Mahadevan PetscFunctionReturn(0); 692db66d124SVijay Mahadevan } 693db66d124SVijay Mahadevan 694db66d124SVijay Mahadevan 695db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g) 696db66d124SVijay Mahadevan { 697db66d124SVijay Mahadevan PetscErrorCode ierr; 698db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 699db66d124SVijay Mahadevan 700db66d124SVijay Mahadevan PetscFunctionBegin; 7016d9eb265SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_FORWARD);CHKERRQ(ierr); 702db66d124SVijay Mahadevan PetscFunctionReturn(0); 703db66d124SVijay Mahadevan } 704db66d124SVijay Mahadevan 705