1032b8ab6SVijay Mahadevan #include <petsc-private/dmmbimpl.h> /*I "petscdm.h" I*/ 2032b8ab6SVijay Mahadevan #include <petsc-private/vecimpl.h> /*I "petscdm.h" I*/ 3032b8ab6SVijay Mahadevan 4032b8ab6SVijay Mahadevan #include <petscdmmoab.h> 5032b8ab6SVijay Mahadevan #include <MBTagConventions.hpp> 6032b8ab6SVijay Mahadevan 7032b8ab6SVijay Mahadevan // declare for use later but before they're defined 8032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecUserDestroy(void *user); 9032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y); 10032b8ab6SVijay Mahadevan static PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name); 11032b8ab6SVijay Mahadevan 12032b8ab6SVijay Mahadevan #undef __FUNCT__ 13032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_CreateVector_Private" 14032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_CreateVector_Private(DM dm,moab::Tag tag,PetscInt tag_size,moab::Range* userrange,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 15032b8ab6SVijay Mahadevan { 16032b8ab6SVijay Mahadevan PetscErrorCode ierr; 17032b8ab6SVijay Mahadevan moab::ErrorCode merr; 18032b8ab6SVijay Mahadevan PetscBool is_newtag; 19032b8ab6SVijay Mahadevan moab::Range *range; 20032b8ab6SVijay Mahadevan PetscInt *gindices,*gsindices; 21032b8ab6SVijay Mahadevan PetscInt i,count,icount,dof; 22032b8ab6SVijay Mahadevan PetscInt size,rank; 23*9088682fSVijay Mahadevan std::string ttname; 24032b8ab6SVijay Mahadevan PetscScalar *data_ptr; 25032b8ab6SVijay Mahadevan 26db66d124SVijay Mahadevan Vec_MOAB *vmoab; 27032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 28032b8ab6SVijay Mahadevan moab::ParallelComm *pcomm = dmmoab->pcomm; 29032b8ab6SVijay Mahadevan moab::Interface *mbiface = dmmoab->mbiface; 30032b8ab6SVijay Mahadevan moab::Tag ltog_tag = dmmoab->ltog_tag; 31032b8ab6SVijay Mahadevan 32032b8ab6SVijay Mahadevan PetscFunctionBegin; 33032b8ab6SVijay Mahadevan if(!userrange) range = dmmoab->vowned; 34032b8ab6SVijay Mahadevan else range = userrange; 35db66d124SVijay Mahadevan if(!range) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_WRONG, "Input range cannot be empty or call DMSetUp first."); 36032b8ab6SVijay Mahadevan 37*9088682fSVijay Mahadevan merr = mbiface->tag_get_name(tag, ttname); 38*9088682fSVijay Mahadevan if (!ttname.length() && merr !=moab::MB_SUCCESS) { 39db66d124SVijay Mahadevan /* get the new name for the anonymous MOABVec -> the tag_name will be destroyed along with Tag */ 40032b8ab6SVijay Mahadevan char *tag_name = PETSC_NULL; 41032b8ab6SVijay Mahadevan ierr = DMMoab_VecCreateTagName_Private(pcomm,&tag_name);CHKERRQ(ierr); 42032b8ab6SVijay Mahadevan is_newtag = PETSC_TRUE; 43032b8ab6SVijay Mahadevan 44db66d124SVijay Mahadevan /* Create the default value for the tag (all zeros) */ 45032b8ab6SVijay Mahadevan std::vector<PetscScalar> default_value(tag_size, 0.0); 46032b8ab6SVijay Mahadevan 47db66d124SVijay Mahadevan /* Create the tag */ 48032b8ab6SVijay Mahadevan merr = mbiface->tag_get_handle(tag_name,tag_size,moab::MB_TYPE_DOUBLE,tag, 49032b8ab6SVijay Mahadevan moab::MB_TAG_DENSE|moab::MB_TAG_CREAT,default_value.data());MBERRNM(merr); 50032b8ab6SVijay Mahadevan ierr = PetscFree(tag_name);CHKERRQ(ierr); 51032b8ab6SVijay Mahadevan } 52032b8ab6SVijay Mahadevan else { 53db66d124SVijay Mahadevan /* Make sure the tag data is of type "double" */ 54032b8ab6SVijay Mahadevan moab::DataType tag_type; 55032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data_type(tag, tag_type);MBERRNM(merr); 56032b8ab6SVijay Mahadevan if(tag_type != moab::MB_TYPE_DOUBLE) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Tag data type must be MB_TYPE_DOUBLE"); 57032b8ab6SVijay Mahadevan is_newtag = destroy_tag; 58032b8ab6SVijay Mahadevan } 59032b8ab6SVijay Mahadevan 60db66d124SVijay Mahadevan /* Create the MOAB internal data object */ 61032b8ab6SVijay Mahadevan ierr = PetscNew(Vec_MOAB,&vmoab);CHKERRQ(ierr); 62032b8ab6SVijay Mahadevan vmoab->tag = tag; 63032b8ab6SVijay Mahadevan vmoab->mbiface = mbiface; 64032b8ab6SVijay Mahadevan vmoab->pcomm = pcomm; 65032b8ab6SVijay Mahadevan vmoab->new_tag = is_newtag; 66032b8ab6SVijay Mahadevan vmoab->is_global_vec = is_global_vec; 67032b8ab6SVijay Mahadevan merr = mbiface->tag_get_length(tag,vmoab->tag_size);MBERR("tag_get_size", merr); 68032b8ab6SVijay Mahadevan 69db66d124SVijay Mahadevan /* set the reference for vector range */ 70032b8ab6SVijay Mahadevan vmoab->tag_range = new moab::Range(*range); 71032b8ab6SVijay Mahadevan 72db66d124SVijay Mahadevan /* Call tag_iterate. This will cause MOAB to allocate memory for the 73db66d124SVijay Mahadevan tag data if it hasn't already happened */ 74032b8ab6SVijay Mahadevan merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr); 75032b8ab6SVijay Mahadevan 76db66d124SVijay Mahadevan /* Check to make sure the tag data is in a single sequence */ 77032b8ab6SVijay Mahadevan if ((unsigned)count != range->size()) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only create MOAB Vector for single sequence"); 78032b8ab6SVijay Mahadevan 79032b8ab6SVijay Mahadevan ierr = MPI_Comm_size(((PetscObject)dm)->comm, &size);CHKERRQ(ierr); 80032b8ab6SVijay Mahadevan ierr = MPI_Comm_rank(((PetscObject)dm)->comm, &rank);CHKERRQ(ierr); 81032b8ab6SVijay Mahadevan 82db66d124SVijay Mahadevan /* Create the PETSc Vector 83db66d124SVijay Mahadevan Query MOAB mesh to check if there are any ghosted entities 84032b8ab6SVijay Mahadevan -> if we do, create a ghosted vector to map correctly to the same layout 85032b8ab6SVijay Mahadevan -> else, create a non-ghosted parallel vector */ 86032b8ab6SVijay Mahadevan if (!is_global_vec && (size>1) && dmmoab->nghost) { 87032b8ab6SVijay Mahadevan moab::Range::iterator iter; 88032b8ab6SVijay Mahadevan ierr = PetscMalloc(dmmoab->nghost*sizeof(PetscInt), &gsindices);CHKERRQ(ierr); 89db66d124SVijay Mahadevan 90032b8ab6SVijay Mahadevan for(iter = dmmoab->vghost->begin(),icount=0; iter != dmmoab->vghost->end(); iter++) { 91032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data(ltog_tag,&(*iter),1,&dof);MBERRNM(merr); 92032b8ab6SVijay Mahadevan gsindices[icount++] = dof; 93032b8ab6SVijay Mahadevan } 94032b8ab6SVijay Mahadevan 95db66d124SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 96032b8ab6SVijay Mahadevan ierr = VecCreateGhostBlockWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*dmmoab->nloc, 97032b8ab6SVijay Mahadevan vmoab->tag_size*dmmoab->n,dmmoab->nghost,gsindices,data_ptr,vec);CHKERRQ(ierr); 98032b8ab6SVijay Mahadevan 99032b8ab6SVijay Mahadevan ierr = PetscFree(gsindices);CHKERRQ(ierr); 100032b8ab6SVijay Mahadevan } 101032b8ab6SVijay Mahadevan else if (size>1) { 102db66d124SVijay Mahadevan /* This is an MPI Vector without ghosted padding */ 103032b8ab6SVijay Mahadevan ierr = VecCreateMPIWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*range->size(), 104032b8ab6SVijay Mahadevan PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr); 105032b8ab6SVijay Mahadevan } 106032b8ab6SVijay Mahadevan else { 107db66d124SVijay Mahadevan /* This is a sequential vector - valid only for the single processor case since MOAB tags are always partitioned 108db66d124SVijay Mahadevan and we cannot define a Vec using the Tag array with size>1 will be of full length */ 109032b8ab6SVijay Mahadevan ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,vmoab->tag_size,vmoab->tag_size*dmmoab->n,data_ptr,vec);CHKERRQ(ierr); 110032b8ab6SVijay Mahadevan } 111032b8ab6SVijay Mahadevan 112032b8ab6SVijay Mahadevan PetscContainer moabdata; 113032b8ab6SVijay Mahadevan ierr = PetscContainerCreate(PETSC_COMM_SELF,&moabdata);CHKERRQ(ierr); 114032b8ab6SVijay Mahadevan ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr); 115032b8ab6SVijay Mahadevan ierr = PetscContainerSetUserDestroy(moabdata,DMMoab_VecUserDestroy);CHKERRQ(ierr); 116032b8ab6SVijay Mahadevan ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr); 117032b8ab6SVijay Mahadevan (*vec)->ops->duplicate = DMMoab_VecDuplicate; 118032b8ab6SVijay Mahadevan ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr); 119032b8ab6SVijay Mahadevan 120032b8ab6SVijay Mahadevan if (!dmmoab->ltog_map) { 121db66d124SVijay Mahadevan /* Vector created, manually set local to global mapping */ 122032b8ab6SVijay Mahadevan ierr = PetscMalloc(range->size()*sizeof(PetscInt)*vmoab->tag_size, &gindices);CHKERRQ(ierr); 123032b8ab6SVijay Mahadevan moab::Range::iterator iter; 124032b8ab6SVijay Mahadevan for(iter = range->begin(),count=0; iter != range->end(); iter++,count+=vmoab->tag_size) { 125032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data(ltog_tag,&(*iter),1,&dof);MBERRNM(merr); 126032b8ab6SVijay Mahadevan for(i=0; i<vmoab->tag_size; ++i) 127032b8ab6SVijay Mahadevan gindices[count+i] = (dof)*vmoab->tag_size+i; 128032b8ab6SVijay Mahadevan } 129032b8ab6SVijay Mahadevan 130032b8ab6SVijay Mahadevan ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_SELF,range->size(),gindices, 131032b8ab6SVijay Mahadevan PETSC_COPY_VALUES,&dmmoab->ltog_map);CHKERRQ(ierr); 132032b8ab6SVijay Mahadevan 133032b8ab6SVijay Mahadevan ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 134032b8ab6SVijay Mahadevan 135db66d124SVijay Mahadevan /* Clean up */ 136032b8ab6SVijay Mahadevan ierr = PetscFree(gindices);CHKERRQ(ierr); 137032b8ab6SVijay Mahadevan } 138032b8ab6SVijay Mahadevan else { 139032b8ab6SVijay Mahadevan ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 140032b8ab6SVijay Mahadevan } 141032b8ab6SVijay Mahadevan 142032b8ab6SVijay Mahadevan /* set the DM reference to the vector */ 143032b8ab6SVijay Mahadevan ierr = VecSetDM(*vec, dm);CHKERRQ(ierr); 144032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 145032b8ab6SVijay Mahadevan } 146032b8ab6SVijay Mahadevan 147032b8ab6SVijay Mahadevan 148032b8ab6SVijay Mahadevan #undef __FUNCT__ 149032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabCreateVector" 150032b8ab6SVijay Mahadevan /*@ 151032b8ab6SVijay Mahadevan DMMoabCreateVector - Create a Vec from either an existing tag, or a specified tag size, and a range of entities 152032b8ab6SVijay Mahadevan 153032b8ab6SVijay Mahadevan Collective on MPI_Comm 154032b8ab6SVijay Mahadevan 155032b8ab6SVijay Mahadevan Input Parameter: 156032b8ab6SVijay Mahadevan . dm - The DMMoab object being set 157032b8ab6SVijay Mahadevan . tag - If non-zero, block size will be taken from the tag size 158032b8ab6SVijay Mahadevan . tag_size - If tag was zero, this parameter specifies the block size; unique tag name will be generated automatically 159032b8ab6SVijay Mahadevan . range - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab 160032b8ab6SVijay Mahadevan . is_global_vec - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one 161032b8ab6SVijay Mahadevan . destroy_tag - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB 162032b8ab6SVijay Mahadevan 163032b8ab6SVijay Mahadevan Output Parameter: 164032b8ab6SVijay Mahadevan . vec - The created vector 165032b8ab6SVijay Mahadevan 166032b8ab6SVijay Mahadevan Level: beginner 167032b8ab6SVijay Mahadevan 168032b8ab6SVijay Mahadevan .keywords: DMMoab, create 169032b8ab6SVijay Mahadevan @*/ 170032b8ab6SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,PetscInt tag_size,moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 171032b8ab6SVijay Mahadevan { 172032b8ab6SVijay Mahadevan PetscErrorCode ierr; 173032b8ab6SVijay Mahadevan 174032b8ab6SVijay Mahadevan PetscFunctionBegin; 175032b8ab6SVijay Mahadevan if(!tag && !tag_size) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag_size and tag cannot be null."); 176032b8ab6SVijay Mahadevan 177032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,tag,tag_size,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr); 178032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 179032b8ab6SVijay Mahadevan } 180032b8ab6SVijay Mahadevan 181032b8ab6SVijay Mahadevan 182032b8ab6SVijay Mahadevan #undef __FUNCT__ 183032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab" 184032b8ab6SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec) 185032b8ab6SVijay Mahadevan { 186032b8ab6SVijay Mahadevan PetscErrorCode ierr; 187032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 188032b8ab6SVijay Mahadevan 189032b8ab6SVijay Mahadevan PetscFunctionBegin; 190032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 191032b8ab6SVijay Mahadevan PetscValidPointer(gvec,2); 192032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr); 193032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 194032b8ab6SVijay Mahadevan } 195032b8ab6SVijay Mahadevan 196032b8ab6SVijay Mahadevan 197032b8ab6SVijay Mahadevan #undef __FUNCT__ 198032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab" 199032b8ab6SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec) 200032b8ab6SVijay Mahadevan { 201032b8ab6SVijay Mahadevan PetscErrorCode ierr; 202032b8ab6SVijay Mahadevan moab::Range vlocal; 203032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 204032b8ab6SVijay Mahadevan 205032b8ab6SVijay Mahadevan PetscFunctionBegin; 206032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 207032b8ab6SVijay Mahadevan PetscValidPointer(lvec,2); 208032b8ab6SVijay Mahadevan vlocal = *dmmoab->vowned; 209032b8ab6SVijay Mahadevan vlocal.merge(*dmmoab->vghost); 210032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vowned,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr); 211032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 212032b8ab6SVijay Mahadevan } 213032b8ab6SVijay Mahadevan 214032b8ab6SVijay Mahadevan 215032b8ab6SVijay Mahadevan #undef __FUNCT__ 216032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecTag" 217032b8ab6SVijay Mahadevan /*@ 218032b8ab6SVijay Mahadevan DMMoabGetVecTag - Get the MOAB tag associated with this Vec 219032b8ab6SVijay Mahadevan 220032b8ab6SVijay Mahadevan Input Parameter: 221032b8ab6SVijay Mahadevan . vec - Vec being queried 222032b8ab6SVijay Mahadevan 223032b8ab6SVijay Mahadevan Output Parameter: 224032b8ab6SVijay Mahadevan . tag - Tag associated with this Vec 225032b8ab6SVijay Mahadevan 226032b8ab6SVijay Mahadevan Level: beginner 227032b8ab6SVijay Mahadevan 228032b8ab6SVijay Mahadevan .keywords: DMMoab, create 229032b8ab6SVijay Mahadevan @*/ 230032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag) 231032b8ab6SVijay Mahadevan { 232032b8ab6SVijay Mahadevan PetscContainer moabdata; 233032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 234032b8ab6SVijay Mahadevan PetscErrorCode ierr; 235032b8ab6SVijay Mahadevan 236032b8ab6SVijay Mahadevan PetscFunctionBegin; 237032b8ab6SVijay Mahadevan PetscValidPointer(tag,2); 238032b8ab6SVijay Mahadevan 239efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 240032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 241032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 242032b8ab6SVijay Mahadevan 243032b8ab6SVijay Mahadevan *tag = vmoab->tag; 244032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 245032b8ab6SVijay Mahadevan } 246032b8ab6SVijay Mahadevan 247032b8ab6SVijay Mahadevan 248032b8ab6SVijay Mahadevan #undef __FUNCT__ 249032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecRange" 250032b8ab6SVijay Mahadevan /*@ 251032b8ab6SVijay Mahadevan DMMoabGetVecRange - Get the MOAB entities associated with this Vec 252032b8ab6SVijay Mahadevan 253032b8ab6SVijay Mahadevan Input Parameter: 254032b8ab6SVijay Mahadevan . vec - Vec being queried 255032b8ab6SVijay Mahadevan 256032b8ab6SVijay Mahadevan Output Parameter: 257032b8ab6SVijay Mahadevan . range - Entities associated with this Vec 258032b8ab6SVijay Mahadevan 259032b8ab6SVijay Mahadevan Level: beginner 260032b8ab6SVijay Mahadevan 261032b8ab6SVijay Mahadevan .keywords: DMMoab, create 262032b8ab6SVijay Mahadevan @*/ 263032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range) 264032b8ab6SVijay Mahadevan { 265032b8ab6SVijay Mahadevan PetscContainer moabdata; 266032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 267032b8ab6SVijay Mahadevan PetscErrorCode ierr; 268032b8ab6SVijay Mahadevan 269032b8ab6SVijay Mahadevan PetscFunctionBegin; 270032b8ab6SVijay Mahadevan PetscValidPointer(range,2); 271032b8ab6SVijay Mahadevan 272efd17f3eSVijay Mahadevan /* Get the MOAB private data handle */ 273032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 274032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 275032b8ab6SVijay Mahadevan 276032b8ab6SVijay Mahadevan *range = *vmoab->tag_range; 277032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 278032b8ab6SVijay Mahadevan } 279032b8ab6SVijay Mahadevan 280032b8ab6SVijay Mahadevan 281032b8ab6SVijay Mahadevan #undef __FUNCT__ 282032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecDuplicate" 283032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y) 284032b8ab6SVijay Mahadevan { 285032b8ab6SVijay Mahadevan PetscErrorCode ierr; 286032b8ab6SVijay Mahadevan DM dm; 287032b8ab6SVijay Mahadevan PetscContainer moabdata; 288032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 289032b8ab6SVijay Mahadevan 290032b8ab6SVijay Mahadevan PetscFunctionBegin; 291032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(x,VEC_CLASSID,1); 292032b8ab6SVijay Mahadevan PetscValidPointer(y,2); 293032b8ab6SVijay Mahadevan 294efd17f3eSVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 295032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 296032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 297032b8ab6SVijay Mahadevan 298032b8ab6SVijay Mahadevan ierr = VecGetDM(x, &dm);CHKERRQ(ierr); 299032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 300032b8ab6SVijay Mahadevan 301032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,vmoab->tag_size,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr); 302032b8ab6SVijay Mahadevan ierr = VecSetDM(*y, dm);CHKERRQ(ierr); 303032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 304032b8ab6SVijay Mahadevan } 305032b8ab6SVijay Mahadevan 306032b8ab6SVijay Mahadevan 307032b8ab6SVijay Mahadevan #undef __FUNCT__ 308032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecCreateTagName_Private" 309032b8ab6SVijay Mahadevan /* DMMoab_VecCreateTagName_Private 310032b8ab6SVijay Mahadevan * 311032b8ab6SVijay Mahadevan * Creates a unique tag name that will be shared across processes. If 312032b8ab6SVijay Mahadevan * pcomm is NULL, then this is a serial vector. A unique tag name 313032b8ab6SVijay Mahadevan * will be returned in tag_name in either case. 314032b8ab6SVijay Mahadevan * 315032b8ab6SVijay Mahadevan * The tag names have the format _PETSC_VEC_N where N is some integer. 316032b8ab6SVijay Mahadevan * 317032b8ab6SVijay Mahadevan * NOTE: The tag_name is allocated in this routine; The user needs to free 318032b8ab6SVijay Mahadevan * the character array. 319032b8ab6SVijay Mahadevan */ 320032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name) 321032b8ab6SVijay Mahadevan { 322032b8ab6SVijay Mahadevan moab::ErrorCode mberr; 323032b8ab6SVijay Mahadevan PetscErrorCode ierr; 324032b8ab6SVijay Mahadevan PetscInt n,global_n; 325032b8ab6SVijay Mahadevan moab::Tag indexTag; 326032b8ab6SVijay Mahadevan 327032b8ab6SVijay Mahadevan PetscFunctionBegin; 328032b8ab6SVijay Mahadevan const char* PVEC_PREFIX = "__PETSC_VEC_"; 329032b8ab6SVijay Mahadevan ierr = PetscMalloc(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr); 330032b8ab6SVijay Mahadevan 331efd17f3eSVijay Mahadevan /* Check to see if there are any PETSc vectors defined */ 332032b8ab6SVijay Mahadevan moab::Interface *mbiface = pcomm->get_moab(); 333032b8ab6SVijay Mahadevan moab::EntityHandle rootset = mbiface->get_root_set(); 334032b8ab6SVijay Mahadevan 335efd17f3eSVijay Mahadevan /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */ 336032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag, 337032b8ab6SVijay Mahadevan moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr); 338032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n); 339032b8ab6SVijay Mahadevan if (mberr == moab::MB_TAG_NOT_FOUND) n=0; /* this is the first temporary vector */ 340032b8ab6SVijay Mahadevan else MBERRNM(mberr); 341032b8ab6SVijay Mahadevan 342032b8ab6SVijay Mahadevan /* increment the new value of n */ 343032b8ab6SVijay Mahadevan ++n; 344032b8ab6SVijay Mahadevan 345efd17f3eSVijay Mahadevan /* Make sure that n is consistent across all processes */ 346032b8ab6SVijay Mahadevan ierr = MPI_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr); 347032b8ab6SVijay Mahadevan 348efd17f3eSVijay Mahadevan /* Set the new name accordingly and return */ 349032b8ab6SVijay Mahadevan ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr); 350032b8ab6SVijay Mahadevan mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr); 351032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 352032b8ab6SVijay Mahadevan } 353032b8ab6SVijay Mahadevan 354032b8ab6SVijay Mahadevan 355032b8ab6SVijay Mahadevan #undef __FUNCT__ 356032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecUserDestroy" 357032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecUserDestroy(void *user) 358032b8ab6SVijay Mahadevan { 359efd17f3eSVijay Mahadevan Vec_MOAB *vmoab=(Vec_MOAB*)user; 360032b8ab6SVijay Mahadevan PetscErrorCode ierr; 361032b8ab6SVijay Mahadevan moab::ErrorCode merr; 362032b8ab6SVijay Mahadevan 363032b8ab6SVijay Mahadevan PetscFunctionBegin; 364032b8ab6SVijay Mahadevan if(vmoab->new_tag && vmoab->tag) { 365efd17f3eSVijay Mahadevan /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */ 366032b8ab6SVijay Mahadevan merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr); 367032b8ab6SVijay Mahadevan } 368032b8ab6SVijay Mahadevan delete vmoab->tag_range; 369032b8ab6SVijay Mahadevan vmoab->tag = PETSC_NULL; 370032b8ab6SVijay Mahadevan vmoab->mbiface = PETSC_NULL; 371032b8ab6SVijay Mahadevan vmoab->pcomm = PETSC_NULL; 372032b8ab6SVijay Mahadevan ierr = PetscFree(vmoab);CHKERRQ(ierr); 373032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 374032b8ab6SVijay Mahadevan } 375032b8ab6SVijay Mahadevan 376db66d124SVijay Mahadevan #undef __FUNCT__ 377efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArray" 378efd17f3eSVijay Mahadevan /*@ 379efd17f3eSVijay 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 380efd17f3eSVijay Mahadevan 381efd17f3eSVijay Mahadevan Collective on MPI_Comm 382efd17f3eSVijay Mahadevan 383efd17f3eSVijay Mahadevan Input Parameter: 384efd17f3eSVijay Mahadevan . dm - The DMMoab object being set 385efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 386efd17f3eSVijay Mahadevan 387efd17f3eSVijay Mahadevan Output Parameter: 388efd17f3eSVijay Mahadevan . array - The local data array 389efd17f3eSVijay Mahadevan 390efd17f3eSVijay Mahadevan Level: intermediate 391efd17f3eSVijay Mahadevan 392efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 393efd17f3eSVijay Mahadevan 394efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 395efd17f3eSVijay Mahadevan @*/ 396efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecGetArray(DM dm,Vec vec,void* array) 397efd17f3eSVijay Mahadevan { 398efd17f3eSVijay Mahadevan DM_Moab *moab; 399efd17f3eSVijay Mahadevan moab::ErrorCode merr; 400efd17f3eSVijay Mahadevan PetscErrorCode ierr; 401efd17f3eSVijay Mahadevan PetscInt count; 402efd17f3eSVijay Mahadevan moab::Tag vtag; 403efd17f3eSVijay Mahadevan PetscScalar **varray; 404efd17f3eSVijay Mahadevan 405efd17f3eSVijay Mahadevan PetscFunctionBegin; 406efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 407efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 408efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 409efd17f3eSVijay Mahadevan 410efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 411efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 412efd17f3eSVijay Mahadevan 413efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 414efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 415efd17f3eSVijay Mahadevan 416efd17f3eSVijay Mahadevan /* Get the array data for local entities */ 417efd17f3eSVijay Mahadevan merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 418efd17f3eSVijay 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()); 419efd17f3eSVijay Mahadevan 420efd17f3eSVijay Mahadevan merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr); 421efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 422efd17f3eSVijay Mahadevan } 423efd17f3eSVijay Mahadevan 424efd17f3eSVijay Mahadevan 425efd17f3eSVijay Mahadevan #undef __FUNCT__ 426efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray" 427efd17f3eSVijay Mahadevan /*@ 428efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray 429efd17f3eSVijay Mahadevan 430efd17f3eSVijay Mahadevan Collective on MPI_Comm 431efd17f3eSVijay Mahadevan 432efd17f3eSVijay Mahadevan Input Parameter: 433efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 434efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 435efd17f3eSVijay Mahadevan - array - The local data array 436efd17f3eSVijay Mahadevan 437efd17f3eSVijay Mahadevan Level: intermediate 438efd17f3eSVijay Mahadevan 439efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 440efd17f3eSVijay Mahadevan 441efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 442efd17f3eSVijay Mahadevan @*/ 443efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArray(DM dm,Vec v,void* array) 444efd17f3eSVijay Mahadevan { 445efd17f3eSVijay Mahadevan DM_Moab *moab; 446efd17f3eSVijay Mahadevan moab::ErrorCode merr; 447efd17f3eSVijay Mahadevan PetscErrorCode ierr; 448efd17f3eSVijay Mahadevan moab::Tag vtag; 449efd17f3eSVijay Mahadevan 450efd17f3eSVijay Mahadevan PetscFunctionBegin; 451efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 452efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(v,VEC_CLASSID,2); 453efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 454efd17f3eSVijay Mahadevan 455efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 456efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(v,&vtag);CHKERRQ(ierr); 457efd17f3eSVijay Mahadevan 458efd17f3eSVijay Mahadevan /* reduce the tags correctly -> should probably let the user choose how to reduce in the future 459efd17f3eSVijay Mahadevan For all FEM residual based assembly calculations, MPI_SUM should serve well 460efd17f3eSVijay Mahadevan */ 461efd17f3eSVijay Mahadevan merr = moab->pcomm->reduce_tags(vtag,MPI_SUM,*moab->vghost);MBERRNM(merr); 462efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 463efd17f3eSVijay Mahadevan } 464efd17f3eSVijay Mahadevan 465efd17f3eSVijay Mahadevan #undef __FUNCT__ 466efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArrayRead" 467efd17f3eSVijay Mahadevan /*@ 468efd17f3eSVijay 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 469efd17f3eSVijay Mahadevan 470efd17f3eSVijay Mahadevan Collective on MPI_Comm 471efd17f3eSVijay Mahadevan 472efd17f3eSVijay Mahadevan Input Parameter: 473efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 474efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 475efd17f3eSVijay Mahadevan 476efd17f3eSVijay Mahadevan Output Parameter: 477efd17f3eSVijay Mahadevan . array - The local data array 478efd17f3eSVijay Mahadevan 479efd17f3eSVijay Mahadevan Level: intermediate 480efd17f3eSVijay Mahadevan 481efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 482efd17f3eSVijay Mahadevan 483efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 484efd17f3eSVijay Mahadevan @*/ 485efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecGetArrayRead(DM dm,Vec vec,void* array) 486efd17f3eSVijay Mahadevan { 487efd17f3eSVijay Mahadevan DM_Moab *moab; 488efd17f3eSVijay Mahadevan moab::ErrorCode merr; 489efd17f3eSVijay Mahadevan PetscErrorCode ierr; 490efd17f3eSVijay Mahadevan PetscInt count; 491efd17f3eSVijay Mahadevan moab::Tag vtag; 492efd17f3eSVijay Mahadevan PetscScalar **varray; 493efd17f3eSVijay Mahadevan 494efd17f3eSVijay Mahadevan PetscFunctionBegin; 495efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 496efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 497efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 498efd17f3eSVijay Mahadevan 499efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 500efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 501efd17f3eSVijay Mahadevan 502efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 503efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 504efd17f3eSVijay Mahadevan 505efd17f3eSVijay Mahadevan /* Get the array data for local entities */ 506efd17f3eSVijay Mahadevan merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 507efd17f3eSVijay 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()); 508efd17f3eSVijay Mahadevan 509efd17f3eSVijay Mahadevan merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr); 510efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 511efd17f3eSVijay Mahadevan } 512efd17f3eSVijay Mahadevan 513efd17f3eSVijay Mahadevan 514efd17f3eSVijay Mahadevan #undef __FUNCT__ 515efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArrayRead" 516efd17f3eSVijay Mahadevan /*@ 517efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray 518efd17f3eSVijay Mahadevan 519efd17f3eSVijay Mahadevan Collective on MPI_Comm 520efd17f3eSVijay Mahadevan 521efd17f3eSVijay Mahadevan Input Parameter: 522efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 523efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 524efd17f3eSVijay Mahadevan - array - The local data array 525efd17f3eSVijay Mahadevan 526efd17f3eSVijay Mahadevan Level: intermediate 527efd17f3eSVijay Mahadevan 528efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 529efd17f3eSVijay Mahadevan 530efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 531efd17f3eSVijay Mahadevan @*/ 532efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArrayRead(DM dm,Vec v,void* array) 533efd17f3eSVijay Mahadevan { 534efd17f3eSVijay Mahadevan PetscFunctionBegin; 535efd17f3eSVijay Mahadevan /* Nothing to do -> do not free the array memory obtained from tag_iterate */ 536efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 537efd17f3eSVijay Mahadevan } 538efd17f3eSVijay Mahadevan 539efd17f3eSVijay Mahadevan 540efd17f3eSVijay Mahadevan #undef __FUNCT__ 541db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab" 542db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l) 543db66d124SVijay Mahadevan { 544db66d124SVijay Mahadevan PetscErrorCode ierr; 545db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 546db66d124SVijay Mahadevan 547db66d124SVijay Mahadevan PetscFunctionBegin; 548db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 549db66d124SVijay Mahadevan PetscFunctionReturn(0); 550db66d124SVijay Mahadevan } 551db66d124SVijay Mahadevan 552db66d124SVijay Mahadevan 553db66d124SVijay Mahadevan #undef __FUNCT__ 554db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab" 555db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l) 556db66d124SVijay Mahadevan { 557db66d124SVijay Mahadevan PetscErrorCode ierr; 558db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 559db66d124SVijay Mahadevan 560db66d124SVijay Mahadevan PetscFunctionBegin; 561db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 562db66d124SVijay Mahadevan PetscFunctionReturn(0); 563db66d124SVijay Mahadevan } 564db66d124SVijay Mahadevan 565db66d124SVijay Mahadevan 566db66d124SVijay Mahadevan #undef __FUNCT__ 567db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab" 568db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g) 569db66d124SVijay Mahadevan { 570db66d124SVijay Mahadevan PetscErrorCode ierr; 571db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 572db66d124SVijay Mahadevan 573db66d124SVijay Mahadevan PetscFunctionBegin; 574db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 575db66d124SVijay Mahadevan PetscFunctionReturn(0); 576db66d124SVijay Mahadevan } 577db66d124SVijay Mahadevan 578db66d124SVijay Mahadevan 579db66d124SVijay Mahadevan #undef __FUNCT__ 580db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab" 581db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g) 582db66d124SVijay Mahadevan { 583db66d124SVijay Mahadevan PetscErrorCode ierr; 584db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 585db66d124SVijay Mahadevan 586db66d124SVijay Mahadevan PetscFunctionBegin; 587db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 588db66d124SVijay Mahadevan PetscFunctionReturn(0); 589db66d124SVijay Mahadevan } 590db66d124SVijay Mahadevan 591db66d124SVijay Mahadevan 592