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