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