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; 20*fc418013SVijay Mahadevan PetscInt *gindices; 21032b8ab6SVijay Mahadevan PetscInt i,count,icount,dof; 22032b8ab6SVijay Mahadevan PetscInt size,rank; 239088682fSVijay 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 379088682fSVijay Mahadevan merr = mbiface->tag_get_name(tag, ttname); 389088682fSVijay 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 69*fc418013SVijay Mahadevan size = pcomm->size(); 70*fc418013SVijay Mahadevan rank = pcomm->rank(); 71*fc418013SVijay Mahadevan 72db66d124SVijay Mahadevan /* set the reference for vector range */ 73032b8ab6SVijay Mahadevan vmoab->tag_range = new moab::Range(*range); 74032b8ab6SVijay Mahadevan 75db66d124SVijay Mahadevan /* Call tag_iterate. This will cause MOAB to allocate memory for the 76db66d124SVijay Mahadevan tag data if it hasn't already happened */ 77032b8ab6SVijay Mahadevan merr = mbiface->tag_iterate(tag,range->begin(),range->end(),count,(void*&)data_ptr);MBERRNM(merr); 78032b8ab6SVijay Mahadevan 79*fc418013SVijay Mahadevan if (rank) range->print(); 80db66d124SVijay Mahadevan /* Check to make sure the tag data is in a single sequence */ 81*fc418013SVijay Mahadevan if ((unsigned)count != range->size()) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only create MOAB Vector for single sequence: %D != %D",count,range->size()); 82032b8ab6SVijay Mahadevan 83db66d124SVijay Mahadevan /* Create the PETSc Vector 84db66d124SVijay Mahadevan Query MOAB mesh to check if there are any ghosted entities 85032b8ab6SVijay Mahadevan -> if we do, create a ghosted vector to map correctly to the same layout 86032b8ab6SVijay Mahadevan -> else, create a non-ghosted parallel vector */ 87*fc418013SVijay Mahadevan if (!is_global_vec && (size>1)) { 88db66d124SVijay Mahadevan /* This is an MPI Vector with ghosted padding */ 89032b8ab6SVijay Mahadevan ierr = VecCreateGhostBlockWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*dmmoab->nloc, 90*fc418013SVijay Mahadevan vmoab->tag_size*dmmoab->n,dmmoab->nghost,&dmmoab->gsindices[dmmoab->nloc],data_ptr,vec);CHKERRQ(ierr); 91032b8ab6SVijay Mahadevan } 92032b8ab6SVijay Mahadevan else if (size>1) { 93db66d124SVijay Mahadevan /* This is an MPI Vector without ghosted padding */ 94032b8ab6SVijay Mahadevan ierr = VecCreateMPIWithArray(vmoab->pcomm->comm(),vmoab->tag_size,vmoab->tag_size*range->size(), 95032b8ab6SVijay Mahadevan PETSC_DECIDE,data_ptr,vec);CHKERRQ(ierr); 96032b8ab6SVijay Mahadevan } 97032b8ab6SVijay Mahadevan else { 98db66d124SVijay Mahadevan /* This is a sequential vector - valid only for the single processor case since MOAB tags are always partitioned 99db66d124SVijay Mahadevan and we cannot define a Vec using the Tag array with size>1 will be of full length */ 100032b8ab6SVijay Mahadevan ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,vmoab->tag_size,vmoab->tag_size*dmmoab->n,data_ptr,vec);CHKERRQ(ierr); 101032b8ab6SVijay Mahadevan } 102032b8ab6SVijay Mahadevan 103032b8ab6SVijay Mahadevan PetscContainer moabdata; 104032b8ab6SVijay Mahadevan ierr = PetscContainerCreate(PETSC_COMM_SELF,&moabdata);CHKERRQ(ierr); 105032b8ab6SVijay Mahadevan ierr = PetscContainerSetPointer(moabdata,vmoab);CHKERRQ(ierr); 106032b8ab6SVijay Mahadevan ierr = PetscContainerSetUserDestroy(moabdata,DMMoab_VecUserDestroy);CHKERRQ(ierr); 107032b8ab6SVijay Mahadevan ierr = PetscObjectCompose((PetscObject)*vec,"MOABData",(PetscObject)moabdata);CHKERRQ(ierr); 108032b8ab6SVijay Mahadevan (*vec)->ops->duplicate = DMMoab_VecDuplicate; 109032b8ab6SVijay Mahadevan ierr = PetscContainerDestroy(&moabdata);CHKERRQ(ierr); 110032b8ab6SVijay Mahadevan 111032b8ab6SVijay Mahadevan if (!dmmoab->ltog_map) { 112db66d124SVijay Mahadevan /* Vector created, manually set local to global mapping */ 113032b8ab6SVijay Mahadevan ierr = PetscMalloc(range->size()*sizeof(PetscInt)*vmoab->tag_size, &gindices);CHKERRQ(ierr); 114032b8ab6SVijay Mahadevan moab::Range::iterator iter; 115032b8ab6SVijay Mahadevan for(iter = range->begin(),count=0; iter != range->end(); iter++,count+=vmoab->tag_size) { 116032b8ab6SVijay Mahadevan merr = mbiface->tag_get_data(ltog_tag,&(*iter),1,&dof);MBERRNM(merr); 117032b8ab6SVijay Mahadevan for(i=0; i<vmoab->tag_size; ++i) 118032b8ab6SVijay Mahadevan gindices[count+i] = (dof)*vmoab->tag_size+i; 119032b8ab6SVijay Mahadevan } 120032b8ab6SVijay Mahadevan 121032b8ab6SVijay Mahadevan ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_SELF,range->size(),gindices, 122032b8ab6SVijay Mahadevan PETSC_COPY_VALUES,&dmmoab->ltog_map);CHKERRQ(ierr); 123032b8ab6SVijay Mahadevan 124032b8ab6SVijay Mahadevan ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 125032b8ab6SVijay Mahadevan 126db66d124SVijay Mahadevan /* Clean up */ 127032b8ab6SVijay Mahadevan ierr = PetscFree(gindices);CHKERRQ(ierr); 128032b8ab6SVijay Mahadevan } 129032b8ab6SVijay Mahadevan else { 130032b8ab6SVijay Mahadevan ierr = VecSetLocalToGlobalMappingBlock(*vec,dmmoab->ltog_map);CHKERRQ(ierr); 131032b8ab6SVijay Mahadevan } 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 . tag_size - If tag was zero, this parameter specifies the block size; unique tag name will be generated automatically 150032b8ab6SVijay Mahadevan . range - If non-empty, Vec corresponds to these entities, otherwise to the entities set on the DMMoab 151032b8ab6SVijay Mahadevan . is_global_vec - If true, this is a local representation of the Vec (including ghosts in parallel), otherwise a truly parallel one 152032b8ab6SVijay Mahadevan . destroy_tag - If true, MOAB tag is destroyed with Vec, otherwise it is left on MOAB 153032b8ab6SVijay Mahadevan 154032b8ab6SVijay Mahadevan Output Parameter: 155032b8ab6SVijay Mahadevan . vec - The created vector 156032b8ab6SVijay Mahadevan 157032b8ab6SVijay Mahadevan Level: beginner 158032b8ab6SVijay Mahadevan 159032b8ab6SVijay Mahadevan .keywords: DMMoab, create 160032b8ab6SVijay Mahadevan @*/ 161032b8ab6SVijay Mahadevan PetscErrorCode DMMoabCreateVector(DM dm,moab::Tag tag,PetscInt tag_size,moab::Range* range,PetscBool is_global_vec,PetscBool destroy_tag,Vec *vec) 162032b8ab6SVijay Mahadevan { 163032b8ab6SVijay Mahadevan PetscErrorCode ierr; 164032b8ab6SVijay Mahadevan 165032b8ab6SVijay Mahadevan PetscFunctionBegin; 166032b8ab6SVijay Mahadevan if(!tag && !tag_size) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Both tag_size and tag cannot be null."); 167032b8ab6SVijay Mahadevan 168032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,tag,tag_size,range,is_global_vec,destroy_tag,vec);CHKERRQ(ierr); 169032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 170032b8ab6SVijay Mahadevan } 171032b8ab6SVijay Mahadevan 172032b8ab6SVijay Mahadevan 173032b8ab6SVijay Mahadevan #undef __FUNCT__ 174032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateGlobalVector_Moab" 175032b8ab6SVijay Mahadevan PetscErrorCode DMCreateGlobalVector_Moab(DM dm,Vec *gvec) 176032b8ab6SVijay Mahadevan { 177032b8ab6SVijay Mahadevan PetscErrorCode ierr; 178032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 179032b8ab6SVijay Mahadevan 180032b8ab6SVijay Mahadevan PetscFunctionBegin; 181032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 182032b8ab6SVijay Mahadevan PetscValidPointer(gvec,2); 183032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vowned,PETSC_TRUE,PETSC_TRUE,gvec);CHKERRQ(ierr); 184032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 185032b8ab6SVijay Mahadevan } 186032b8ab6SVijay Mahadevan 187032b8ab6SVijay Mahadevan 188032b8ab6SVijay Mahadevan #undef __FUNCT__ 189032b8ab6SVijay Mahadevan #define __FUNCT__ "DMCreateLocalVector_Moab" 190032b8ab6SVijay Mahadevan PetscErrorCode DMCreateLocalVector_Moab(DM dm,Vec *lvec) 191032b8ab6SVijay Mahadevan { 192032b8ab6SVijay Mahadevan PetscErrorCode ierr; 193032b8ab6SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 194032b8ab6SVijay Mahadevan 195032b8ab6SVijay Mahadevan PetscFunctionBegin; 196032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 197032b8ab6SVijay Mahadevan PetscValidPointer(lvec,2); 198*fc418013SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,dmmoab->bs,dmmoab->vlocal,PETSC_FALSE,PETSC_TRUE,lvec);CHKERRQ(ierr); 199032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 200032b8ab6SVijay Mahadevan } 201032b8ab6SVijay Mahadevan 202032b8ab6SVijay Mahadevan 203032b8ab6SVijay Mahadevan #undef __FUNCT__ 204032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecTag" 205032b8ab6SVijay Mahadevan /*@ 206032b8ab6SVijay Mahadevan DMMoabGetVecTag - Get the MOAB tag associated with this Vec 207032b8ab6SVijay Mahadevan 208032b8ab6SVijay Mahadevan Input Parameter: 209032b8ab6SVijay Mahadevan . vec - Vec being queried 210032b8ab6SVijay Mahadevan 211032b8ab6SVijay Mahadevan Output Parameter: 212032b8ab6SVijay Mahadevan . tag - Tag associated with this Vec 213032b8ab6SVijay Mahadevan 214032b8ab6SVijay Mahadevan Level: beginner 215032b8ab6SVijay Mahadevan 216032b8ab6SVijay Mahadevan .keywords: DMMoab, create 217032b8ab6SVijay Mahadevan @*/ 218032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecTag(Vec vec,moab::Tag *tag) 219032b8ab6SVijay Mahadevan { 220032b8ab6SVijay Mahadevan PetscContainer moabdata; 221032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 222032b8ab6SVijay Mahadevan PetscErrorCode ierr; 223032b8ab6SVijay Mahadevan 224032b8ab6SVijay Mahadevan PetscFunctionBegin; 225032b8ab6SVijay Mahadevan PetscValidPointer(tag,2); 226032b8ab6SVijay Mahadevan 227efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 228032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 229032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 230032b8ab6SVijay Mahadevan 231032b8ab6SVijay Mahadevan *tag = vmoab->tag; 232032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 233032b8ab6SVijay Mahadevan } 234032b8ab6SVijay Mahadevan 235032b8ab6SVijay Mahadevan 236032b8ab6SVijay Mahadevan #undef __FUNCT__ 237032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoabGetVecRange" 238032b8ab6SVijay Mahadevan /*@ 239032b8ab6SVijay Mahadevan DMMoabGetVecRange - Get the MOAB entities associated with this Vec 240032b8ab6SVijay Mahadevan 241032b8ab6SVijay Mahadevan Input Parameter: 242032b8ab6SVijay Mahadevan . vec - Vec being queried 243032b8ab6SVijay Mahadevan 244032b8ab6SVijay Mahadevan Output Parameter: 245032b8ab6SVijay Mahadevan . range - Entities associated with this Vec 246032b8ab6SVijay Mahadevan 247032b8ab6SVijay Mahadevan Level: beginner 248032b8ab6SVijay Mahadevan 249032b8ab6SVijay Mahadevan .keywords: DMMoab, create 250032b8ab6SVijay Mahadevan @*/ 251032b8ab6SVijay Mahadevan PetscErrorCode DMMoabGetVecRange(Vec vec,moab::Range *range) 252032b8ab6SVijay Mahadevan { 253032b8ab6SVijay Mahadevan PetscContainer moabdata; 254032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 255032b8ab6SVijay Mahadevan PetscErrorCode ierr; 256032b8ab6SVijay Mahadevan 257032b8ab6SVijay Mahadevan PetscFunctionBegin; 258032b8ab6SVijay Mahadevan PetscValidPointer(range,2); 259032b8ab6SVijay Mahadevan 260efd17f3eSVijay Mahadevan /* Get the MOAB private data handle */ 261032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)vec,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 262032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**) &vmoab);CHKERRQ(ierr); 263032b8ab6SVijay Mahadevan 264032b8ab6SVijay Mahadevan *range = *vmoab->tag_range; 265032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 266032b8ab6SVijay Mahadevan } 267032b8ab6SVijay Mahadevan 268032b8ab6SVijay Mahadevan 269032b8ab6SVijay Mahadevan #undef __FUNCT__ 270032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecDuplicate" 271032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecDuplicate(Vec x,Vec *y) 272032b8ab6SVijay Mahadevan { 273032b8ab6SVijay Mahadevan PetscErrorCode ierr; 274032b8ab6SVijay Mahadevan DM dm; 275032b8ab6SVijay Mahadevan PetscContainer moabdata; 276032b8ab6SVijay Mahadevan Vec_MOAB *vmoab; 277032b8ab6SVijay Mahadevan 278032b8ab6SVijay Mahadevan PetscFunctionBegin; 279032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(x,VEC_CLASSID,1); 280032b8ab6SVijay Mahadevan PetscValidPointer(y,2); 281032b8ab6SVijay Mahadevan 282efd17f3eSVijay Mahadevan /* Get the Vec_MOAB struct for the original vector */ 283032b8ab6SVijay Mahadevan ierr = PetscObjectQuery((PetscObject)x,"MOABData", (PetscObject*) &moabdata);CHKERRQ(ierr); 284032b8ab6SVijay Mahadevan ierr = PetscContainerGetPointer(moabdata, (void**)&vmoab);CHKERRQ(ierr); 285032b8ab6SVijay Mahadevan 286032b8ab6SVijay Mahadevan ierr = VecGetDM(x, &dm);CHKERRQ(ierr); 287032b8ab6SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 288032b8ab6SVijay Mahadevan 289032b8ab6SVijay Mahadevan ierr = DMMoab_CreateVector_Private(dm,PETSC_NULL,vmoab->tag_size,vmoab->tag_range,vmoab->is_global_vec,PETSC_TRUE,y);CHKERRQ(ierr); 290032b8ab6SVijay Mahadevan ierr = VecSetDM(*y, dm);CHKERRQ(ierr); 291032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 292032b8ab6SVijay Mahadevan } 293032b8ab6SVijay Mahadevan 294032b8ab6SVijay Mahadevan 295032b8ab6SVijay Mahadevan #undef __FUNCT__ 296032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecCreateTagName_Private" 297032b8ab6SVijay Mahadevan /* DMMoab_VecCreateTagName_Private 298032b8ab6SVijay Mahadevan * 299032b8ab6SVijay Mahadevan * Creates a unique tag name that will be shared across processes. If 300032b8ab6SVijay Mahadevan * pcomm is NULL, then this is a serial vector. A unique tag name 301032b8ab6SVijay Mahadevan * will be returned in tag_name in either case. 302032b8ab6SVijay Mahadevan * 303032b8ab6SVijay Mahadevan * The tag names have the format _PETSC_VEC_N where N is some integer. 304032b8ab6SVijay Mahadevan * 305032b8ab6SVijay Mahadevan * NOTE: The tag_name is allocated in this routine; The user needs to free 306032b8ab6SVijay Mahadevan * the character array. 307032b8ab6SVijay Mahadevan */ 308032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecCreateTagName_Private(moab::ParallelComm *pcomm,char** tag_name) 309032b8ab6SVijay Mahadevan { 310032b8ab6SVijay Mahadevan moab::ErrorCode mberr; 311032b8ab6SVijay Mahadevan PetscErrorCode ierr; 312032b8ab6SVijay Mahadevan PetscInt n,global_n; 313032b8ab6SVijay Mahadevan moab::Tag indexTag; 314032b8ab6SVijay Mahadevan 315032b8ab6SVijay Mahadevan PetscFunctionBegin; 316032b8ab6SVijay Mahadevan const char* PVEC_PREFIX = "__PETSC_VEC_"; 317032b8ab6SVijay Mahadevan ierr = PetscMalloc(PETSC_MAX_PATH_LEN, tag_name);CHKERRQ(ierr); 318032b8ab6SVijay Mahadevan 319efd17f3eSVijay Mahadevan /* Check to see if there are any PETSc vectors defined */ 320032b8ab6SVijay Mahadevan moab::Interface *mbiface = pcomm->get_moab(); 321032b8ab6SVijay Mahadevan moab::EntityHandle rootset = mbiface->get_root_set(); 322032b8ab6SVijay Mahadevan 323efd17f3eSVijay Mahadevan /* Create a tag in MOAB mesh to index and keep track of number of Petsc vec tags */ 324032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_handle("__PETSC_VECS__",1,moab::MB_TYPE_INTEGER,indexTag, 325032b8ab6SVijay Mahadevan moab::MB_TAG_SPARSE | moab::MB_TAG_CREAT,0);MBERRNM(mberr); 326032b8ab6SVijay Mahadevan mberr = mbiface->tag_get_data(indexTag, &rootset, 1, &n); 327032b8ab6SVijay Mahadevan if (mberr == moab::MB_TAG_NOT_FOUND) n=0; /* this is the first temporary vector */ 328032b8ab6SVijay Mahadevan else MBERRNM(mberr); 329032b8ab6SVijay Mahadevan 330032b8ab6SVijay Mahadevan /* increment the new value of n */ 331032b8ab6SVijay Mahadevan ++n; 332032b8ab6SVijay Mahadevan 333efd17f3eSVijay Mahadevan /* Make sure that n is consistent across all processes */ 334032b8ab6SVijay Mahadevan ierr = MPI_Allreduce(&n,&global_n,1,MPI_INT,MPI_MAX,pcomm->comm());CHKERRQ(ierr); 335032b8ab6SVijay Mahadevan 336efd17f3eSVijay Mahadevan /* Set the new name accordingly and return */ 337032b8ab6SVijay Mahadevan ierr = PetscSNPrintf(*tag_name, PETSC_MAX_PATH_LEN-1, "%s_%D", PVEC_PREFIX, global_n);CHKERRQ(ierr); 338032b8ab6SVijay Mahadevan mberr = mbiface->tag_set_data(indexTag, &rootset, 1, (const void*)&global_n);MBERRNM(mberr); 339032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 340032b8ab6SVijay Mahadevan } 341032b8ab6SVijay Mahadevan 342032b8ab6SVijay Mahadevan 343032b8ab6SVijay Mahadevan #undef __FUNCT__ 344032b8ab6SVijay Mahadevan #define __FUNCT__ "DMMoab_VecUserDestroy" 345032b8ab6SVijay Mahadevan PetscErrorCode DMMoab_VecUserDestroy(void *user) 346032b8ab6SVijay Mahadevan { 347efd17f3eSVijay Mahadevan Vec_MOAB *vmoab=(Vec_MOAB*)user; 348032b8ab6SVijay Mahadevan PetscErrorCode ierr; 349032b8ab6SVijay Mahadevan moab::ErrorCode merr; 350032b8ab6SVijay Mahadevan 351032b8ab6SVijay Mahadevan PetscFunctionBegin; 352032b8ab6SVijay Mahadevan if(vmoab->new_tag && vmoab->tag) { 353efd17f3eSVijay Mahadevan /* Tag was created via a call to VecDuplicate, delete the underlying tag in MOAB */ 354032b8ab6SVijay Mahadevan merr = vmoab->mbiface->tag_delete(vmoab->tag);MBERRNM(merr); 355032b8ab6SVijay Mahadevan } 356032b8ab6SVijay Mahadevan delete vmoab->tag_range; 357032b8ab6SVijay Mahadevan vmoab->tag = PETSC_NULL; 358032b8ab6SVijay Mahadevan vmoab->mbiface = PETSC_NULL; 359032b8ab6SVijay Mahadevan vmoab->pcomm = PETSC_NULL; 360032b8ab6SVijay Mahadevan ierr = PetscFree(vmoab);CHKERRQ(ierr); 361032b8ab6SVijay Mahadevan PetscFunctionReturn(0); 362032b8ab6SVijay Mahadevan } 363032b8ab6SVijay 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 { 386efd17f3eSVijay Mahadevan DM_Moab *moab; 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); 396efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 397efd17f3eSVijay Mahadevan 398efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 399efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 400efd17f3eSVijay Mahadevan 401efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 402efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 403efd17f3eSVijay Mahadevan 404efd17f3eSVijay Mahadevan /* Get the array data for local entities */ 405efd17f3eSVijay Mahadevan merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 406efd17f3eSVijay 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()); 407efd17f3eSVijay Mahadevan 408efd17f3eSVijay Mahadevan merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr); 409efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 410efd17f3eSVijay Mahadevan } 411efd17f3eSVijay Mahadevan 412efd17f3eSVijay Mahadevan 413efd17f3eSVijay Mahadevan #undef __FUNCT__ 414efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArray" 415efd17f3eSVijay Mahadevan /*@ 416efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the writable direct access array obtained via DMMoabVecGetArray 417efd17f3eSVijay Mahadevan 418efd17f3eSVijay Mahadevan Collective on MPI_Comm 419efd17f3eSVijay Mahadevan 420efd17f3eSVijay Mahadevan Input Parameter: 421efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 422efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 423efd17f3eSVijay Mahadevan - array - The local data array 424efd17f3eSVijay Mahadevan 425efd17f3eSVijay Mahadevan Level: intermediate 426efd17f3eSVijay Mahadevan 427efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 428efd17f3eSVijay Mahadevan 429efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArray(), DMMoabVecGetArrayRead(), DMMoabVecRestoreArrayRead() 430efd17f3eSVijay Mahadevan @*/ 431efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArray(DM dm,Vec v,void* array) 432efd17f3eSVijay Mahadevan { 433efd17f3eSVijay Mahadevan DM_Moab *moab; 434efd17f3eSVijay Mahadevan moab::ErrorCode merr; 435efd17f3eSVijay Mahadevan PetscErrorCode ierr; 436efd17f3eSVijay Mahadevan moab::Tag vtag; 437efd17f3eSVijay Mahadevan 438efd17f3eSVijay Mahadevan PetscFunctionBegin; 439efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 440efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(v,VEC_CLASSID,2); 441efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 442efd17f3eSVijay Mahadevan 443efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 444efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(v,&vtag);CHKERRQ(ierr); 445efd17f3eSVijay Mahadevan 446efd17f3eSVijay Mahadevan /* reduce the tags correctly -> should probably let the user choose how to reduce in the future 447efd17f3eSVijay Mahadevan For all FEM residual based assembly calculations, MPI_SUM should serve well 448efd17f3eSVijay Mahadevan */ 449efd17f3eSVijay Mahadevan merr = moab->pcomm->reduce_tags(vtag,MPI_SUM,*moab->vghost);MBERRNM(merr); 450efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 451efd17f3eSVijay Mahadevan } 452efd17f3eSVijay Mahadevan 453efd17f3eSVijay Mahadevan #undef __FUNCT__ 454efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecGetArrayRead" 455efd17f3eSVijay Mahadevan /*@ 456efd17f3eSVijay 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 457efd17f3eSVijay Mahadevan 458efd17f3eSVijay Mahadevan Collective on MPI_Comm 459efd17f3eSVijay Mahadevan 460efd17f3eSVijay Mahadevan Input Parameter: 461efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 462efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 463efd17f3eSVijay Mahadevan 464efd17f3eSVijay Mahadevan Output Parameter: 465efd17f3eSVijay Mahadevan . array - The local data array 466efd17f3eSVijay Mahadevan 467efd17f3eSVijay Mahadevan Level: intermediate 468efd17f3eSVijay Mahadevan 469efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 470efd17f3eSVijay Mahadevan 471efd17f3eSVijay Mahadevan .seealso: DMMoabVecRestoreArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 472efd17f3eSVijay Mahadevan @*/ 473efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecGetArrayRead(DM dm,Vec vec,void* array) 474efd17f3eSVijay Mahadevan { 475efd17f3eSVijay Mahadevan DM_Moab *moab; 476efd17f3eSVijay Mahadevan moab::ErrorCode merr; 477efd17f3eSVijay Mahadevan PetscErrorCode ierr; 478efd17f3eSVijay Mahadevan PetscInt count; 479efd17f3eSVijay Mahadevan moab::Tag vtag; 480efd17f3eSVijay Mahadevan PetscScalar **varray; 481efd17f3eSVijay Mahadevan 482efd17f3eSVijay Mahadevan PetscFunctionBegin; 483efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 484efd17f3eSVijay Mahadevan PetscValidHeaderSpecific(vec,VEC_CLASSID,2); 485efd17f3eSVijay Mahadevan moab=(DM_Moab*)dm->data; 486efd17f3eSVijay Mahadevan 487efd17f3eSVijay Mahadevan /* Get the MOAB private data */ 488efd17f3eSVijay Mahadevan ierr = DMMoabGetVecTag(vec,&vtag);CHKERRQ(ierr); 489efd17f3eSVijay Mahadevan 490efd17f3eSVijay Mahadevan /* Get the real scalar array handle */ 491efd17f3eSVijay Mahadevan varray = reinterpret_cast<PetscScalar**>(array); 492efd17f3eSVijay Mahadevan 493efd17f3eSVijay Mahadevan /* Get the array data for local entities */ 494efd17f3eSVijay Mahadevan merr = moab->mbiface->tag_iterate(vtag,moab->vlocal->begin(),moab->vlocal->end(),count,reinterpret_cast<void*&>(*varray),true);MBERRNM(merr); 495efd17f3eSVijay 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()); 496efd17f3eSVijay Mahadevan 497efd17f3eSVijay Mahadevan merr = moab->pcomm->exchange_tags(vtag,*moab->vlocal);MBERRNM(merr); 498efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 499efd17f3eSVijay Mahadevan } 500efd17f3eSVijay Mahadevan 501efd17f3eSVijay Mahadevan 502efd17f3eSVijay Mahadevan #undef __FUNCT__ 503efd17f3eSVijay Mahadevan #define __FUNCT__ "DMMoabVecRestoreArrayRead" 504efd17f3eSVijay Mahadevan /*@ 505efd17f3eSVijay Mahadevan DMMoabVecRestoreArray - Restores the read-only direct access array obtained via DMMoabVecGetArray 506efd17f3eSVijay Mahadevan 507efd17f3eSVijay Mahadevan Collective on MPI_Comm 508efd17f3eSVijay Mahadevan 509efd17f3eSVijay Mahadevan Input Parameter: 510efd17f3eSVijay Mahadevan + dm - The DMMoab object being set 511efd17f3eSVijay Mahadevan . vec - The Vector whose underlying data is requested 512efd17f3eSVijay Mahadevan - array - The local data array 513efd17f3eSVijay Mahadevan 514efd17f3eSVijay Mahadevan Level: intermediate 515efd17f3eSVijay Mahadevan 516efd17f3eSVijay Mahadevan .keywords: MOAB, distributed array 517efd17f3eSVijay Mahadevan 518efd17f3eSVijay Mahadevan .seealso: DMMoabVecGetArrayRead(), DMMoabVecGetArray(), DMMoabVecRestoreArray() 519efd17f3eSVijay Mahadevan @*/ 520efd17f3eSVijay Mahadevan PetscErrorCode DMMoabVecRestoreArrayRead(DM dm,Vec v,void* array) 521efd17f3eSVijay Mahadevan { 522efd17f3eSVijay Mahadevan PetscFunctionBegin; 523efd17f3eSVijay Mahadevan /* Nothing to do -> do not free the array memory obtained from tag_iterate */ 524efd17f3eSVijay Mahadevan PetscFunctionReturn(0); 525efd17f3eSVijay Mahadevan } 526efd17f3eSVijay Mahadevan 527efd17f3eSVijay Mahadevan 528efd17f3eSVijay Mahadevan #undef __FUNCT__ 529db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalBegin_Moab" 530db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalBegin_Moab(DM dm,Vec g,InsertMode mode,Vec l) 531db66d124SVijay Mahadevan { 532db66d124SVijay Mahadevan PetscErrorCode ierr; 533db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 534db66d124SVijay Mahadevan 535db66d124SVijay Mahadevan PetscFunctionBegin; 536db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 537db66d124SVijay Mahadevan PetscFunctionReturn(0); 538db66d124SVijay Mahadevan } 539db66d124SVijay Mahadevan 540db66d124SVijay Mahadevan 541db66d124SVijay Mahadevan #undef __FUNCT__ 542db66d124SVijay Mahadevan #define __FUNCT__ "DMGlobalToLocalEnd_Moab" 543db66d124SVijay Mahadevan PetscErrorCode DMGlobalToLocalEnd_Moab(DM dm,Vec g,InsertMode mode,Vec l) 544db66d124SVijay Mahadevan { 545db66d124SVijay Mahadevan PetscErrorCode ierr; 546db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 547db66d124SVijay Mahadevan 548db66d124SVijay Mahadevan PetscFunctionBegin; 549db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,g,l,mode,SCATTER_FORWARD);CHKERRQ(ierr); 550db66d124SVijay Mahadevan PetscFunctionReturn(0); 551db66d124SVijay Mahadevan } 552db66d124SVijay Mahadevan 553db66d124SVijay Mahadevan 554db66d124SVijay Mahadevan #undef __FUNCT__ 555db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalBegin_Moab" 556db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalBegin_Moab(DM dm,Vec l,InsertMode mode,Vec g) 557db66d124SVijay Mahadevan { 558db66d124SVijay Mahadevan PetscErrorCode ierr; 559db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 560db66d124SVijay Mahadevan 561db66d124SVijay Mahadevan PetscFunctionBegin; 562db66d124SVijay Mahadevan ierr = VecScatterBegin(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 563db66d124SVijay Mahadevan PetscFunctionReturn(0); 564db66d124SVijay Mahadevan } 565db66d124SVijay Mahadevan 566db66d124SVijay Mahadevan 567db66d124SVijay Mahadevan #undef __FUNCT__ 568db66d124SVijay Mahadevan #define __FUNCT__ "DMLocalToGlobalEnd_Moab" 569db66d124SVijay Mahadevan PetscErrorCode DMLocalToGlobalEnd_Moab(DM dm,Vec l,InsertMode mode,Vec g) 570db66d124SVijay Mahadevan { 571db66d124SVijay Mahadevan PetscErrorCode ierr; 572db66d124SVijay Mahadevan DM_Moab *dmmoab = (DM_Moab*)dm->data; 573db66d124SVijay Mahadevan 574db66d124SVijay Mahadevan PetscFunctionBegin; 575db66d124SVijay Mahadevan ierr = VecScatterEnd(dmmoab->ltog_sendrecv,l,g,mode,SCATTER_REVERSE);CHKERRQ(ierr); 576db66d124SVijay Mahadevan PetscFunctionReturn(0); 577db66d124SVijay Mahadevan } 578db66d124SVijay Mahadevan 579db66d124SVijay Mahadevan 580