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