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