1032b8ab6SVijay Mahadevan #include <petsc-private/dmmbimpl.h> /*I "petscdm.h" I*/ 2032b8ab6SVijay Mahadevan #include <petsc-private/vecimpl.h> /*I "petscdm.h" I*/ 3032b8ab6SVijay Mahadevan 4032b8ab6SVijay Mahadevan #include <petscdmmoab.h> 5032b8ab6SVijay Mahadevan #include <MBTagConventions.hpp> 6032b8ab6SVijay Mahadevan 7032b8ab6SVijay Mahadevan // declare for use later but before they're defined 8032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecUserDestroy(void *user); 9032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y); 10032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name); 11032b8ab6SVijay Mahadevan 12032b8ab6SVijay Mahadevan #undef __FUNCT__ 13032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_CreateVector_Private" 14*bb8f3634SVijay Mahadevan PetscErrorCode DMMoab_CreateVector_Private(DM dm,moab::Tag tag,moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 15032b8ab6SVijay Mahadevan { 16032b8ab6SVijay Mahadevan PetscErrorCode ierr; 17032b8ab6SVijay Mahadevan moab::ErrorCode merr; 18032b8ab6SVijay Mahadevan PetscBool is_newtag; 19032b8ab6SVijay Mahadevan moab::Range *range; 20*bb8f3634SVijay Mahadevan PetscInt count,lnative_vec,gnative_vec; 219088682fSVijay Mahadevan std::string ttname; 22032b8ab6SVijay Mahadevan PetscScalar *data_ptr; 23*bb8f3634SVijay Mahadevan ISLocalToGlobalMapping ltogb; 24032b8ab6SVijay Mahadevan 25db66d124SVijay Mahadevan Vec_MOAB *vmoab; 26032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 27032b8ab6SVijay Mahadevan moab::ParallelComm *pcomm = dmmoab->pcomm; 28032b8ab6SVijay Mahadevan moab::Interface *mbiface = dmmoab->mbiface; 29032b8ab6SVijay Mahadevan 30032b8ab6SVijay Mahadevan PetscFunctionBegin; 31032b8ab6SVijay Mahadevan if(!userrange) range = dmmoab->vowned; 32032b8ab6SVijay Mahadevan else range = userrange; 33db66d124SVijay Mahadevan if(!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first."); 34032b8ab6SVijay Mahadevan 35*bb8f3634SVijay Mahadevan /* If the tag data is in a single sequence, use PETSc native vector since tag_iterate isn't useful anymore */ 36*bb8f3634SVijay Mahadevan lnative_vec=(range->psize()-1); 37*bb8f3634SVijay Mahadevan 38*bb8f3634SVijay Mahadevan // lnative_vec=1; /* NOTE: Testing PETSc vector: will force to create native vector all the time */ 39*bb8f3634SVijay Mahadevan ierr = MPI_Allreduce(&lnative_vec, &gnative_vec, 1, MPI_INT, MPI_MAX, pcomm->comm());CHKERRQ(ierr); 40*bb8f3634SVijay Mahadevan 41*bb8f3634SVijay Mahadevan /* Create the MOAB internal data object */ 42*bb8f3634SVijay Mahadevan ierr = PetscNew(Vec_MOAB,&vmoab);CHKERRQ(ierr); 43*bb8f3634SVijay Mahadevan vmoab->is_native_vec=(gnative_vec>0?PETSC_TRUE:PETSC_FALSE); 44*bb8f3634SVijay Mahadevan 45*bb8f3634SVijay Mahadevan if (!vmoab->is_native_vec) { 469088682fSVijay Mahadevan merr = mbiface->tag_get_name(tag, ttname); 479088682fSVijay Mahadevan if (!ttname.length() && merr !=moab::MB_SUCCESS) { 48db66d124SVijay Mahadevan /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */ 49032b8ab6SVijay Mahadevan char *tag_name = PETSC_NULL; 50032b8ab6SVijay Mahadevan ierr = DMMoab_VecCreateTagName_Private(pcomm,&tag_name);CHKERRQ(ierr); 51032b8ab6SVijay Mahadevan is_newtag = PETSC_TRUE; 52032b8ab6SVijay Mahadevan 53db66d124SVijay Mahadevan /* Create the default value for the tag (all zeros) */ 54*bb8f3634SVijay Mahadevan std::vector<PetscScalar> default_value(dmmoab->nfields, 0.0); 55032b8ab6SVijay Mahadevan 56db66d124SVijay Mahadevan /* Create the tag */ 57*bb8f3634SVijay Mahadevan merr = mbiface->tag_get_handle(tag_name,dmmoab->nfields,moab::MB_TYPE_DOUBLE,tag, 58032b8ab6SVijay Mahadevan moab::MB_TAG_DENSE|moab::MB_TAG_CREAT,default_value.data());MBERRNM(merr); 59032b8ab6SVijay Mahadevan ierr = PetscFree(tag_name);CHKERRQ(ierr); 60032b8ab6SVijay Mahadevan } 61032b8ab6SVijay Mahadevan else { 62db66d124SVijay Mahadevan /* Make sure the tag data is of type "double" */ 63032b8ab6SVijay Mahadevan moab::DataType tag_type; 64032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data_type(tag, tag_type);MBERRNM(merr); 65032b8ab6SVijay Mahadevan if(tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE"); 66032b8ab6SVijay Mahadevan is_newtag = destroy_tag; 67032b8ab6SVijay Mahadevan } 68032b8ab6SVijay Mahadevan 69032b8ab6SVijay Mahadevan vmoab->tag = tag; 70*bb8f3634SVijay Mahadevan vmoab->new_tag = is_newtag; 71*bb8f3634SVijay Mahadevan } 72032b8ab6SVijay Mahadevan vmoab->mbiface = mbiface; 73032b8ab6SVijay Mahadevan vmoab->pcomm = pcomm; 74032b8ab6SVijay Mahadevan vmoab->is_global_vec = is_global_vec; 75*bb8f3634SVijay Mahadevan vmoab->tag_size=dmmoab->bs; 76032b8ab6SVijay Mahadevan 77*bb8f3634SVijay Mahadevan if (vmoab->is_native_vec) { 78fc418013SVijay Mahadevan 79*bb8f3634SVijay Mahadevan /* Create the PETSc Vector directly and attach our functions accordingly */ 80*bb8f3634SVijay Mahadevan if (!is_global_vec) { 81*bb8f3634SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 82*bb8f3634SVijay Mahadevan ierr = VecCreateGhostBlock(pcomm->comm(),dmmoab->bs,dmmoab->nfields*dmmoab->nloc, 83*bb8f3634SVijay Mahadevan dmmoab->nfields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],vec);CHKERRQ(ierr); 84*bb8f3634SVijay Mahadevan } 85*bb8f3634SVijay Mahadevan else { 86*bb8f3634SVijay Mahadevan /* This is an MPI/SEQ Vector */ 87*bb8f3634SVijay Mahadevan ierr = VecCreate(pcomm->comm(),vec);CHKERRQ(ierr); 88*bb8f3634SVijay Mahadevan ierr = VecSetSizes(*vec,dmmoab->nfields*dmmoab->nloc,PETSC_DECIDE);CHKERRQ(ierr); 89*bb8f3634SVijay Mahadevan ierr = VecSetBlockSize(*vec,dmmoab->bs);CHKERRQ(ierr); 90*bb8f3634SVijay Mahadevan ierr = VecSetType(*vec, VECMPI);CHKERRQ(ierr); 91*bb8f3634SVijay Mahadevan } 92*bb8f3634SVijay Mahadevan } 93*bb8f3634SVijay Mahadevan else { 94db66d124SVijay Mahadevan /* Call tag_iterate. This will cause MOAB to allocate memory for the 95db66d124SVijay Mahadevan tag data if it hasn't already happened */ 96032b8ab6SVijay Mahadevan merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr); 97032b8ab6SVijay Mahadevan 98*bb8f3634SVijay Mahadevan /* set the reference for vector range */ 99*bb8f3634SVijay Mahadevan vmoab->tag_range = new moab::Range(*range); 100*bb8f3634SVijay Mahadevan merr = mbiface->tag_get_length(tag,dmmoab->nfields);MBERRNM(merr); 101032b8ab6SVijay Mahadevan 102db66d124SVijay Mahadevan /* Create the PETSc Vector 103db66d124SVijay Mahadevan Query MOAB mesh to check if there are any ghosted entities 104032b8ab6SVijay Mahadevan -> if we do, create a ghosted vector to map correctly to the same layout 105032b8ab6SVijay Mahadevan -> else, create a non-ghosted parallel vector */ 106*bb8f3634SVijay Mahadevan if (!is_global_vec) { 107db66d124SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 108*bb8f3634SVijay Mahadevan ierr = VecCreateGhostBlockWithArray(pcomm->comm(),dmmoab->bs,dmmoab->nfields*dmmoab->nloc, 109*bb8f3634SVijay Mahadevan dmmoab->nfields*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],data_ptr,vec);CHKERRQ(ierr); 110032b8ab6SVijay Mahadevan } 111032b8ab6SVijay Mahadevan else { 112*bb8f3634SVijay Mahadevan /* This is an MPI Vector without ghosted padding */ 113*bb8f3634SVijay Mahadevan ierr = VecCreateMPIWithArray(pcomm->comm(),dmmoab->bs,dmmoab->nfields*range->size(), 114*bb8f3634SVijay Mahadevan PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr); 115032b8ab6SVijay Mahadevan } 116*bb8f3634SVijay Mahadevan } 117*bb8f3634SVijay Mahadevan ierr = VecSetFromOptions(*vec);CHKERRQ(ierr); 118032b8ab6SVijay Mahadevan 119032b8ab6SVijay Mahadevan PetscContainer moabdata; 120*bb8f3634SVijay Mahadevan ierr = PetscContainerCreate(PETSC_COMM_WORLD,&moabdata);CHKERRQ(ierr); 121032b8ab6SVijay Mahadevan ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr); 122032b8ab6SVijay Mahadevan ierr = PetscContainerSetUserDestroy(moabdata,DMMoab_VecUserDestroy);CHKERRQ(ierr); 123032b8ab6SVijay Mahadevan ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr); 124032b8ab6SVijay Mahadevan (*vec)->ops->duplicate = DMMoab_VecDuplicate; 125032b8ab6SVijay Mahadevan ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr); 126032b8ab6SVijay Mahadevan 127db66d124SVijay Mahadevan /* Vector created, manually set local to global mapping */ 128*bb8f3634SVijay Mahadevan ierr = VecSetLocalToGlobalMapping(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 129*bb8f3634SVijay Mahadevan ierr = ISLocalToGlobalMappingBlock(dmmoab->ltog_map,dmmoab->bs,<ogb); 130*bb8f3634SVijay Mahadevan ierr = VecSetLocalToGlobalMappingBlock(*vec,ltogb);CHKERRQ(ierr); 131*bb8f3634SVijay Mahadevan ierr = ISLocalToGlobalMappingDestroy(<ogb);CHKERRQ(ierr); 132032b8ab6SVijay Mahadevan 133032b8ab6SVijay Mahadevan /* set the DM reference to the vector */ 134032b8ab6SVijay Mahadevan ierr = VecSetDM(*vec, dm);CHKERRQ(ierr); 135032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 136032b8ab6SVijay Mahadevan } 137032b8ab6SVijay Mahadevan 138032b8ab6SVijay Mahadevan 139032b8ab6SVijay Mahadevan #undef __FUNCT__ 140032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabCreateVector" 141032b8ab6SVijay Mahadevan /*@ 142032b8ab6SVijay Mahadevan DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities 143032b8ab6SVijay Mahadevan 144032b8ab6SVijay Mahadevan Collective on MPI_Comm 145032b8ab6SVijay Mahadevan 146032b8ab6SVijay Mahadevan Input Parameter: 147032b8ab6SVijay Mahadevan . dm - The DMMoab object being set 148032b8ab6SVijay Mahadevan . tag - If non-zero, block size will be taken from the tag size 149032b8ab6SVijay Mahadevan . range - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab 150032b8ab6SVijay Mahadevan . is_global_vec - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one 151032b8ab6SVijay Mahadevan . destroy_tag - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB 152032b8ab6SVijay Mahadevan 153032b8ab6SVijay Mahadevan Output Parameter: 154032b8ab6SVijay Mahadevan . vec - The created vector 155032b8ab6SVijay Mahadevan 156032b8ab6SVijay Mahadevan Level: beginner 157032b8ab6SVijay Mahadevan 158032b8ab6SVijay Mahadevan .keywords: DMMoab, create 159032b8ab6SVijay Mahadevan @*/ 160*bb8f3634SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 161032b8ab6SVijay Mahadevan { 162032b8ab6SVijay Mahadevan PetscErrorCode ierr; 163032b8ab6SVijay Mahadevan 164032b8ab6SVijay Mahadevan PetscFunctionBegin; 165*bb8f3634SVijay Mahadevan if(!tag && (!range || range->empty())) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag and range cannot be null."); 166032b8ab6SVijay Mahadevan 167*bb8f3634SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,tag,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr); 168032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 169032b8ab6SVijay Mahadevan } 170032b8ab6SVijay Mahadevan 171032b8ab6SVijay Mahadevan 172032b8ab6SVijay Mahadevan #undef __FUNCT__ 173032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab" 174032b8ab6SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec) 175032b8ab6SVijay Mahadevan { 176032b8ab6SVijay Mahadevan PetscErrorCode ierr; 177032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 178032b8ab6SVijay Mahadevan 179032b8ab6SVijay Mahadevan PetscFunctionBegin; 180032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 181032b8ab6SVijay Mahadevan PetscValidPointer(gvec,2); 182*bb8f3634SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr); 183032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 184032b8ab6SVijay Mahadevan } 185032b8ab6SVijay Mahadevan 186032b8ab6SVijay Mahadevan 187032b8ab6SVijay Mahadevan #undef __FUNCT__ 188032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab" 189032b8ab6SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec) 190032b8ab6SVijay Mahadevan { 191032b8ab6SVijay Mahadevan PetscErrorCode ierr; 192032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 193032b8ab6SVijay Mahadevan 194032b8ab6SVijay Mahadevan PetscFunctionBegin; 195032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 196032b8ab6SVijay Mahadevan PetscValidPointer(lvec,2); 197*bb8f3634SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr); 198032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 199032b8ab6SVijay Mahadevan } 200032b8ab6SVijay Mahadevan 201032b8ab6SVijay Mahadevan 202032b8ab6SVijay Mahadevan #undef __FUNCT__ 203032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecTag" 204032b8ab6SVijay Mahadevan /*@ 205032b8ab6SVijay Mahadevan DMMoabGetVecTag - Get the MOAB tag associated with this Vec 206032b8ab6SVijay Mahadevan 207032b8ab6SVijay Mahadevan Input Parameter: 208032b8ab6SVijay Mahadevan . vec - Vec being queried 209032b8ab6SVijay Mahadevan 210032b8ab6SVijay Mahadevan Output Parameter: 211032b8ab6SVijay Mahadevan . tag - Tag associated with this Vec 212032b8ab6SVijay Mahadevan 213032b8ab6SVijay Mahadevan Level: beginner 214032b8ab6SVijay Mahadevan 215032b8ab6SVijay Mahadevan .keywords: DMMoab, create 216032b8ab6SVijay Mahadevan @*/ 217032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag) 218032b8ab6SVijay Mahadevan { 219032b8ab6SVijay Mahadevan PetscContainer moabdata; 220032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 221032b8ab6SVijay Mahadevan PetscErrorCode ierr; 222032b8ab6SVijay Mahadevan 223032b8ab6SVijay Mahadevan PetscFunctionBegin; 224032b8ab6SVijay Mahadevan PetscValidPointer(tag,2); 225032b8ab6SVijay Mahadevan 226efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 227032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 228032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 229032b8ab6SVijay Mahadevan 230032b8ab6SVijay Mahadevan *tag = vmoab->tag; 231032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 232032b8ab6SVijay Mahadevan } 233032b8ab6SVijay Mahadevan 234032b8ab6SVijay Mahadevan 235032b8ab6SVijay Mahadevan #undef __FUNCT__ 236032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecRange" 237032b8ab6SVijay Mahadevan /*@ 238032b8ab6SVijay Mahadevan DMMoabGetVecRange - Get the MOAB entities associated with this Vec 239032b8ab6SVijay Mahadevan 240032b8ab6SVijay Mahadevan Input Parameter: 241032b8ab6SVijay Mahadevan . vec - Vec being queried 242032b8ab6SVijay Mahadevan 243032b8ab6SVijay Mahadevan Output Parameter: 244032b8ab6SVijay Mahadevan . range - Entities associated with this Vec 245032b8ab6SVijay Mahadevan 246032b8ab6SVijay Mahadevan Level: beginner 247032b8ab6SVijay Mahadevan 248032b8ab6SVijay Mahadevan .keywords: DMMoab, create 249032b8ab6SVijay Mahadevan @*/ 250032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range) 251032b8ab6SVijay Mahadevan { 252032b8ab6SVijay Mahadevan PetscContainer moabdata; 253032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 254032b8ab6SVijay Mahadevan PetscErrorCode ierr; 255032b8ab6SVijay Mahadevan 256032b8ab6SVijay Mahadevan PetscFunctionBegin; 257032b8ab6SVijay Mahadevan PetscValidPointer(range,2); 258032b8ab6SVijay Mahadevan 259efd17f3eSVijay Mahadevan /* Get the MOAB private data handle */ 260032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 261032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 262032b8ab6SVijay Mahadevan 263032b8ab6SVijay Mahadevan *range = *vmoab->tag_range; 264032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 265032b8ab6SVijay Mahadevan } 266032b8ab6SVijay Mahadevan 267032b8ab6SVijay Mahadevan 268032b8ab6SVijay Mahadevan #undef __FUNCT__ 269032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecDuplicate" 270032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y) 271032b8ab6SVijay Mahadevan { 272032b8ab6SVijay Mahadevan PetscErrorCode ierr; 273032b8ab6SVijay Mahadevan DM dm; 274032b8ab6SVijay Mahadevan PetscContainer moabdata; 275032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 276032b8ab6SVijay Mahadevan 277032b8ab6SVijay Mahadevan PetscFunctionBegin; 278032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(x,VEC_CLASSID,1); 279032b8ab6SVijay Mahadevan PetscValidPointer(y,2); 280032b8ab6SVijay Mahadevan 281efd17f3eSVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 282032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 283032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 284032b8ab6SVijay Mahadevan 285032b8ab6SVijay Mahadevan ierr = VecGetDM(x, &dm);CHKERRQ(ierr); 286032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 287032b8ab6SVijay Mahadevan 288*bb8f3634SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr); 289032b8ab6SVijay Mahadevan ierr = VecSetDM(*y, dm);CHKERRQ(ierr); 290032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 291032b8ab6SVijay Mahadevan } 292032b8ab6SVijay Mahadevan 293032b8ab6SVijay Mahadevan 294032b8ab6SVijay Mahadevan #undef __FUNCT__ 295032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecCreateTagName_Private" 296032b8ab6SVijay Mahadevan /* DMMoab_VecCreateTagName_Private 297032b8ab6SVijay Mahadevan * 298032b8ab6SVijay Mahadevan * Creates a unique tag name that will be shared across processes. If 299032b8ab6SVijay Mahadevan * pcomm is NULL, then this is a serial vector. A unique tag name 300032b8ab6SVijay Mahadevan * will be returned in tag_name in either case. 301032b8ab6SVijay Mahadevan * 302032b8ab6SVijay Mahadevan * The tag names have the format _PETSC_VEC_N where N is some integer. 303032b8ab6SVijay Mahadevan * 304032b8ab6SVijay Mahadevan * NOTE: The tag_name is allocated in this routine; The user needs to free 305032b8ab6SVijay Mahadevan * the character array. 306032b8ab6SVijay Mahadevan */ 307032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name) 308032b8ab6SVijay Mahadevan { 309032b8ab6SVijay Mahadevan moab::ErrorCode mberr; 310032b8ab6SVijay Mahadevan PetscErrorCode ierr; 311032b8ab6SVijay Mahadevan PetscInt n,global_n; 312032b8ab6SVijay Mahadevan moab::Tag indexTag; 313032b8ab6SVijay Mahadevan 314032b8ab6SVijay Mahadevan PetscFunctionBegin; 315032b8ab6SVijay Mahadevan const char* PVEC_PREFIX = "__PETSC_VEC_"; 316032b8ab6SVijay Mahadevan ierr = PetscMalloc(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr); 317032b8ab6SVijay Mahadevan 318efd17f3eSVijay Mahadevan /* Check to see if there are any PETSc vectors defined */ 319032b8ab6SVijay Mahadevan moab::Interface *mbiface = pcomm->get_moab(); 320032b8ab6SVijay Mahadevan moab::EntityHandle rootset = mbiface->get_root_set(); 321032b8ab6SVijay Mahadevan 322efd17f3eSVijay Mahadevan /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */ 323032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag, 324032b8ab6SVijay Mahadevan moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr); 325032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n); 326032b8ab6SVijay Mahadevan if (mberr == moab::MB_TAG_NOT_FOUND) n=0; /* this is the first temporary vector */ 327032b8ab6SVijay Mahadevan else MBERRNM(mberr); 328032b8ab6SVijay Mahadevan 329032b8ab6SVijay Mahadevan /* increment the new value of n */ 330032b8ab6SVijay Mahadevan ++n; 331032b8ab6SVijay Mahadevan 332efd17f3eSVijay Mahadevan /* Make sure that n is consistent across all processes */ 333032b8ab6SVijay Mahadevan ierr = MPI_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr); 334032b8ab6SVijay Mahadevan 335efd17f3eSVijay Mahadevan /* Set the new name accordingly and return */ 336032b8ab6SVijay Mahadevan ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr); 337032b8ab6SVijay Mahadevan mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr); 338032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 339032b8ab6SVijay Mahadevan } 340032b8ab6SVijay Mahadevan 341032b8ab6SVijay Mahadevan 342032b8ab6SVijay Mahadevan #undef __FUNCT__ 343032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecUserDestroy" 344032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecUserDestroy(void *user) 345032b8ab6SVijay Mahadevan { 346efd17f3eSVijay Mahadevan Vec_MOAB *vmoab=(Vec_MOAB*)user; 347032b8ab6SVijay Mahadevan PetscErrorCode ierr; 348032b8ab6SVijay Mahadevan moab::ErrorCode merr; 349032b8ab6SVijay Mahadevan 350032b8ab6SVijay Mahadevan PetscFunctionBegin; 351032b8ab6SVijay Mahadevan if(vmoab->new_tag && vmoab->tag) { 352efd17f3eSVijay Mahadevan /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */ 353032b8ab6SVijay Mahadevan merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr); 354032b8ab6SVijay Mahadevan } 355032b8ab6SVijay Mahadevan delete vmoab->tag_range; 356032b8ab6SVijay Mahadevan vmoab->tag = PETSC_NULL; 357032b8ab6SVijay Mahadevan vmoab->mbiface = PETSC_NULL; 358032b8ab6SVijay Mahadevan vmoab->pcomm = PETSC_NULL; 359032b8ab6SVijay Mahadevan ierr = PetscFree(vmoab);CHKERRQ(ierr); 360032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 361032b8ab6SVijay Mahadevan } 362032b8ab6SVijay Mahadevan 363*bb8f3634SVijay Mahadevan 364db66d124SVijay Mahadevan #undef __FUNCT__ 365efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArray" 366efd17f3eSVijay Mahadevan /*@ 367efd17f3eSVijay Mahadevan DMMoabVecGetArray - Returns the writable direct access array to the local representation of MOAB tag data for the underlying vector using locally owned+ghosted range of entities 368efd17f3eSVijay Mahadevan 369efd17f3eSVijay Mahadevan Collective on MPI_Comm 370efd17f3eSVijay Mahadevan 371efd17f3eSVijay Mahadevan Input Parameter: 372efd17f3eSVijay Mahadevan . dm - The DMMoab object being set 373efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 374efd17f3eSVijay Mahadevan 375efd17f3eSVijay Mahadevan Output Parameter: 376efd17f3eSVijay Mahadevan . array - The local data array 377efd17f3eSVijay Mahadevan 378efd17f3eSVijay Mahadevan Level: intermediate 379efd17f3eSVijay Mahadevan 380efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 381efd17f3eSVijay Mahadevan 382efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 383efd17f3eSVijay Mahadevan @*/ 384efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecGetArray(DM dm,Vec vec,void* array) 385efd17f3eSVijay Mahadevan { 386a2eadca2SVijay Mahadevan DM_Moab *dmmoab; 387efd17f3eSVijay Mahadevan moab::ErrorCode merr; 388efd17f3eSVijay Mahadevan PetscErrorCode ierr; 389efd17f3eSVijay Mahadevan PetscInt count; 390efd17f3eSVijay Mahadevan moab::Tag vtag; 391efd17f3eSVijay Mahadevan PetscScalar **varray; 392efd17f3eSVijay Mahadevan 393efd17f3eSVijay Mahadevan PetscFunctionBegin; 394efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 395efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 396a2eadca2SVijay Mahadevan PetscValidPointer(array,3); 397a2eadca2SVijay Mahadevan dmmoab=(DM_Moab*)dm->data; 398efd17f3eSVijay Mahadevan 399efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 400efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 401efd17f3eSVijay Mahadevan 402efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 403efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 404efd17f3eSVijay Mahadevan 405a2eadca2SVijay Mahadevan /* exchange the data into ghost cells first */ 406a2eadca2SVijay Mahadevan merr = dmmoab->pcomm->exchange_tags(vtag,*dmmoab->vlocal);MBERRNM(merr); 407efd17f3eSVijay Mahadevan 408a2eadca2SVijay Mahadevan /* Get the array data for local entities */ 409a2eadca2SVijay Mahadevan merr = dmmoab->mbiface->tag_iterate(vtag,dmmoab->vlocal->begin(),dmmoab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 410a2eadca2SVijay Mahadevan if (count!=(PetscInt)dmmoab->vlocal->size()) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,dmmoab->vlocal->size()); 411a2eadca2SVijay Mahadevan 412efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 413efd17f3eSVijay Mahadevan } 414efd17f3eSVijay Mahadevan 415efd17f3eSVijay Mahadevan 416efd17f3eSVijay Mahadevan #undef __FUNCT__ 417efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray" 418efd17f3eSVijay Mahadevan /*@ 419efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray 420efd17f3eSVijay Mahadevan 421efd17f3eSVijay Mahadevan Collective on MPI_Comm 422efd17f3eSVijay Mahadevan 423efd17f3eSVijay Mahadevan Input Parameter: 424efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 425efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 426efd17f3eSVijay Mahadevan - array - The local data array 427efd17f3eSVijay Mahadevan 428efd17f3eSVijay Mahadevan Level: intermediate 429efd17f3eSVijay Mahadevan 430efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 431efd17f3eSVijay Mahadevan 432efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 433efd17f3eSVijay Mahadevan @*/ 434efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArray(DM dm,Vec v,void* array) 435efd17f3eSVijay Mahadevan { 436efd17f3eSVijay Mahadevan DM_Moab *moab; 437efd17f3eSVijay Mahadevan moab::ErrorCode merr; 438efd17f3eSVijay Mahadevan PetscErrorCode ierr; 439efd17f3eSVijay Mahadevan moab::Tag vtag; 440a2eadca2SVijay Mahadevan PetscScalar **varray; 441efd17f3eSVijay Mahadevan 442efd17f3eSVijay Mahadevan PetscFunctionBegin; 443efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 444efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(v,VEC_CLASSID,2); 445a2eadca2SVijay Mahadevan PetscValidPointer(array,3); 446efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 447efd17f3eSVijay Mahadevan 448efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 449efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(v,&vtag);CHKERRQ(ierr); 450efd17f3eSVijay Mahadevan 451a2eadca2SVijay Mahadevan /* Get the real scalar array handle */ 452a2eadca2SVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 453a2eadca2SVijay Mahadevan merr = moab->mbiface->tag_set_data(vtag,*moab->vlocal,reinterpret_cast<void*&>(*varray));MBERRNM(merr); 454a2eadca2SVijay Mahadevan 455efd17f3eSVijay Mahadevan /* reduce the tags correctly -> should probably let the user choose how to reduce in the future 456a2eadca2SVijay Mahadevan For all FEM residual based assembly calculations, MPI_SUM should serve well */ 457ac5d1045SVijay Mahadevan merr = moab->pcomm->reduce_tags(vtag,MPI_SUM,*moab->vlocal);MBERRNM(merr); 458efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 459efd17f3eSVijay Mahadevan } 460efd17f3eSVijay Mahadevan 461efd17f3eSVijay Mahadevan #undef __FUNCT__ 462efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArrayRead" 463efd17f3eSVijay Mahadevan /*@ 464efd17f3eSVijay 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 465efd17f3eSVijay Mahadevan 466efd17f3eSVijay Mahadevan Collective on MPI_Comm 467efd17f3eSVijay Mahadevan 468efd17f3eSVijay Mahadevan Input Parameter: 469efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 470efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 471efd17f3eSVijay Mahadevan 472efd17f3eSVijay Mahadevan Output Parameter: 473efd17f3eSVijay Mahadevan . array - The local data array 474efd17f3eSVijay Mahadevan 475efd17f3eSVijay Mahadevan Level: intermediate 476efd17f3eSVijay Mahadevan 477efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 478efd17f3eSVijay Mahadevan 479efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 480efd17f3eSVijay Mahadevan @*/ 481efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecGetArrayRead(DM dm,Vec vec,void* array) 482efd17f3eSVijay Mahadevan { 483efd17f3eSVijay Mahadevan DM_Moab *moab; 484efd17f3eSVijay Mahadevan moab::ErrorCode merr; 485efd17f3eSVijay Mahadevan PetscErrorCode ierr; 486efd17f3eSVijay Mahadevan PetscInt count; 487efd17f3eSVijay Mahadevan moab::Tag vtag; 488efd17f3eSVijay Mahadevan PetscScalar **varray; 489efd17f3eSVijay Mahadevan 490efd17f3eSVijay Mahadevan PetscFunctionBegin; 491efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 492efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 493efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 494efd17f3eSVijay Mahadevan 495efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 496efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 497efd17f3eSVijay Mahadevan 498efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 499efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 500efd17f3eSVijay Mahadevan 501a2eadca2SVijay Mahadevan /* exchange the data into ghost cells first */ 502a2eadca2SVijay Mahadevan merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr); 503a2eadca2SVijay Mahadevan 504efd17f3eSVijay Mahadevan /* Get the array data for local entities */ 505efd17f3eSVijay Mahadevan merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 506efd17f3eSVijay Mahadevan if (count!=(PetscInt)moab->vlocal->size()) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and tag partition for Vec. %D != %D.",count,moab->vlocal->size()); 507efd17f3eSVijay Mahadevan 508efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 509efd17f3eSVijay Mahadevan } 510efd17f3eSVijay Mahadevan 511efd17f3eSVijay Mahadevan 512efd17f3eSVijay Mahadevan #undef __FUNCT__ 513efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArrayRead" 514efd17f3eSVijay Mahadevan /*@ 515efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray 516efd17f3eSVijay Mahadevan 517efd17f3eSVijay Mahadevan Collective on MPI_Comm 518efd17f3eSVijay Mahadevan 519efd17f3eSVijay Mahadevan Input Parameter: 520efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 521efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 522efd17f3eSVijay Mahadevan - array - The local data array 523efd17f3eSVijay Mahadevan 524efd17f3eSVijay Mahadevan Level: intermediate 525efd17f3eSVijay Mahadevan 526efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 527efd17f3eSVijay Mahadevan 528efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 529efd17f3eSVijay Mahadevan @*/ 530efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArrayRead(DM dm,Vec v,void* array) 531efd17f3eSVijay Mahadevan { 532efd17f3eSVijay Mahadevan PetscFunctionBegin; 533efd17f3eSVijay Mahadevan /* Nothing to do -> do not free the array memory obtained from tag_iterate */ 534efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 535efd17f3eSVijay Mahadevan } 536efd17f3eSVijay Mahadevan 537efd17f3eSVijay Mahadevan 538efd17f3eSVijay Mahadevan #undef __FUNCT__ 539db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab" 540db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l) 541db66d124SVijay Mahadevan { 542db66d124SVijay Mahadevan PetscErrorCode ierr; 543db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 544db66d124SVijay Mahadevan 545db66d124SVijay Mahadevan PetscFunctionBegin; 546db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 547db66d124SVijay Mahadevan PetscFunctionReturn(0); 548db66d124SVijay Mahadevan } 549db66d124SVijay Mahadevan 550db66d124SVijay Mahadevan 551db66d124SVijay Mahadevan #undef __FUNCT__ 552db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab" 553db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l) 554db66d124SVijay Mahadevan { 555db66d124SVijay Mahadevan PetscErrorCode ierr; 556db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 557db66d124SVijay Mahadevan 558db66d124SVijay Mahadevan PetscFunctionBegin; 559db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 560db66d124SVijay Mahadevan PetscFunctionReturn(0); 561db66d124SVijay Mahadevan } 562db66d124SVijay Mahadevan 563db66d124SVijay Mahadevan 564db66d124SVijay Mahadevan #undef __FUNCT__ 565db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab" 566db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g) 567db66d124SVijay Mahadevan { 568db66d124SVijay Mahadevan PetscErrorCode ierr; 569db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 570db66d124SVijay Mahadevan 571db66d124SVijay Mahadevan PetscFunctionBegin; 572db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 573db66d124SVijay Mahadevan PetscFunctionReturn(0); 574db66d124SVijay Mahadevan } 575db66d124SVijay Mahadevan 576db66d124SVijay Mahadevan 577db66d124SVijay Mahadevan #undef __FUNCT__ 578db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab" 579db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g) 580db66d124SVijay Mahadevan { 581db66d124SVijay Mahadevan PetscErrorCode ierr; 582db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 583db66d124SVijay Mahadevan 584db66d124SVijay Mahadevan PetscFunctionBegin; 585db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 586db66d124SVijay Mahadevan PetscFunctionReturn(0); 587db66d124SVijay Mahadevan } 588db66d124SVijay Mahadevan 589db66d124SVijay Mahadevan 590