147c6ae99SBarry Smith 23c48a1e8SJed Brown #include "packimpl.h" /*I "petscdmcomposite.h" I*/ 347c6ae99SBarry Smith 447c6ae99SBarry Smith #undef __FUNCT__ 547c6ae99SBarry Smith #define __FUNCT__ "DMCompositeSetCoupling" 647c6ae99SBarry Smith /*@C 747c6ae99SBarry Smith DMCompositeSetCoupling - Sets user provided routines that compute the coupling between the 8aa219208SBarry Smith seperate components (DMDA's and arrays) in a DMto build the correct matrix nonzero structure. 947c6ae99SBarry Smith 1047c6ae99SBarry Smith 1147c6ae99SBarry Smith Logically Collective on MPI_Comm 1247c6ae99SBarry Smith 1347c6ae99SBarry Smith Input Parameter: 1447c6ae99SBarry Smith + dm - the composite object 1547c6ae99SBarry Smith - formcouplelocations - routine to set the nonzero locations in the matrix 1647c6ae99SBarry Smith 1747c6ae99SBarry Smith Level: advanced 1847c6ae99SBarry Smith 19065723b2SJed Brown Notes: See DMSetContext() and DMGetContext() for how to get user information into 2047c6ae99SBarry Smith this routine 2147c6ae99SBarry Smith 2247c6ae99SBarry Smith @*/ 237087cfbeSBarry Smith PetscErrorCode DMCompositeSetCoupling(DM dm,PetscErrorCode (*FormCoupleLocations)(DM,Mat,PetscInt*,PetscInt*,PetscInt,PetscInt,PetscInt,PetscInt)) 2447c6ae99SBarry Smith { 2547c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 2647c6ae99SBarry Smith 2747c6ae99SBarry Smith PetscFunctionBegin; 2847c6ae99SBarry Smith com->FormCoupleLocations = FormCoupleLocations; 2947c6ae99SBarry Smith PetscFunctionReturn(0); 3047c6ae99SBarry Smith } 3147c6ae99SBarry Smith 3247c6ae99SBarry Smith #undef __FUNCT__ 330c010503SBarry Smith #define __FUNCT__ "DMDestroy_Composite" 346bf464f9SBarry Smith PetscErrorCode DMDestroy_Composite(DM dm) 3547c6ae99SBarry Smith { 3647c6ae99SBarry Smith PetscErrorCode ierr; 3747c6ae99SBarry Smith struct DMCompositeLink *next, *prev; 3847c6ae99SBarry Smith DM_Composite *com = (DM_Composite *)dm->data; 3947c6ae99SBarry Smith 4047c6ae99SBarry Smith PetscFunctionBegin; 4147c6ae99SBarry Smith next = com->next; 4247c6ae99SBarry Smith while (next) { 4347c6ae99SBarry Smith prev = next; 4447c6ae99SBarry Smith next = next->next; 4547c6ae99SBarry Smith if (prev->type == DMCOMPOSITE_DM) { 46fcfd50ebSBarry Smith ierr = DMDestroy(&prev->dm);CHKERRQ(ierr); 4747c6ae99SBarry Smith } 4847c6ae99SBarry Smith ierr = PetscFree(prev->grstarts);CHKERRQ(ierr); 4947c6ae99SBarry Smith ierr = PetscFree(prev);CHKERRQ(ierr); 5047c6ae99SBarry Smith } 5147c6ae99SBarry Smith PetscFunctionReturn(0); 5247c6ae99SBarry Smith } 5347c6ae99SBarry Smith 5447c6ae99SBarry Smith #undef __FUNCT__ 550c010503SBarry Smith #define __FUNCT__ "DMView_Composite" 567087cfbeSBarry Smith PetscErrorCode DMView_Composite(DM dm,PetscViewer v) 5747c6ae99SBarry Smith { 5847c6ae99SBarry Smith PetscErrorCode ierr; 5947c6ae99SBarry Smith PetscBool iascii; 6047c6ae99SBarry Smith DM_Composite *com = (DM_Composite *)dm->data; 6147c6ae99SBarry Smith 6247c6ae99SBarry Smith PetscFunctionBegin; 6347c6ae99SBarry Smith ierr = PetscTypeCompare((PetscObject)v,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 6447c6ae99SBarry Smith if (iascii) { 6547c6ae99SBarry Smith struct DMCompositeLink *lnk = com->next; 6647c6ae99SBarry Smith PetscInt i; 6747c6ae99SBarry Smith 6847c6ae99SBarry Smith ierr = PetscViewerASCIIPrintf(v,"DM (%s)\n",((PetscObject)dm)->prefix?((PetscObject)dm)->prefix:"no prefix");CHKERRQ(ierr); 6947c6ae99SBarry Smith ierr = PetscViewerASCIIPrintf(v," contains %d DMs and %d redundant arrays\n",com->nDM,com->nredundant);CHKERRQ(ierr); 7047c6ae99SBarry Smith ierr = PetscViewerASCIIPushTab(v);CHKERRQ(ierr); 7147c6ae99SBarry Smith for (i=0; lnk; lnk=lnk->next,i++) { 7247c6ae99SBarry Smith if (lnk->dm) { 7347c6ae99SBarry Smith ierr = PetscViewerASCIIPrintf(v,"Link %d: DM of type %s\n",i,((PetscObject)lnk->dm)->type_name);CHKERRQ(ierr); 7447c6ae99SBarry Smith ierr = PetscViewerASCIIPushTab(v);CHKERRQ(ierr); 7547c6ae99SBarry Smith ierr = DMView(lnk->dm,v);CHKERRQ(ierr); 7647c6ae99SBarry Smith ierr = PetscViewerASCIIPopTab(v);CHKERRQ(ierr); 7747c6ae99SBarry Smith } else { 7806ebdd98SJed Brown ierr = PetscViewerASCIIPrintf(v,"Link %d: Redundant array of size %d owned by rank %d\n",i,lnk->nlocal,lnk->rank);CHKERRQ(ierr); 7947c6ae99SBarry Smith } 8047c6ae99SBarry Smith } 8147c6ae99SBarry Smith ierr = PetscViewerASCIIPopTab(v);CHKERRQ(ierr); 8247c6ae99SBarry Smith } 8347c6ae99SBarry Smith PetscFunctionReturn(0); 8447c6ae99SBarry Smith } 8547c6ae99SBarry Smith 8647c6ae99SBarry Smith /* --------------------------------------------------------------------------------------*/ 8747c6ae99SBarry Smith #undef __FUNCT__ 88d7bf68aeSBarry Smith #define __FUNCT__ "DMSetUp_Composite" 897087cfbeSBarry Smith PetscErrorCode DMSetUp_Composite(DM dm) 9047c6ae99SBarry Smith { 9147c6ae99SBarry Smith PetscErrorCode ierr; 9247c6ae99SBarry Smith PetscInt nprev = 0; 9347c6ae99SBarry Smith PetscMPIInt rank,size; 9447c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 9547c6ae99SBarry Smith struct DMCompositeLink *next = com->next; 9647c6ae99SBarry Smith PetscLayout map; 9747c6ae99SBarry Smith 9847c6ae99SBarry Smith PetscFunctionBegin; 9947c6ae99SBarry Smith if (com->setup) SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_ARG_WRONGSTATE,"Packer has already been setup"); 10047c6ae99SBarry Smith ierr = PetscLayoutCreate(((PetscObject)dm)->comm,&map);CHKERRQ(ierr); 10147c6ae99SBarry Smith ierr = PetscLayoutSetLocalSize(map,com->n);CHKERRQ(ierr); 10247c6ae99SBarry Smith ierr = PetscLayoutSetSize(map,PETSC_DETERMINE);CHKERRQ(ierr); 10347c6ae99SBarry Smith ierr = PetscLayoutSetBlockSize(map,1);CHKERRQ(ierr); 10447c6ae99SBarry Smith ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 10547c6ae99SBarry Smith ierr = PetscLayoutGetSize(map,&com->N);CHKERRQ(ierr); 10647c6ae99SBarry Smith ierr = PetscLayoutGetRange(map,&com->rstart,PETSC_NULL);CHKERRQ(ierr); 107fcfd50ebSBarry Smith ierr = PetscLayoutDestroy(&map);CHKERRQ(ierr); 10847c6ae99SBarry Smith 10947c6ae99SBarry Smith /* now set the rstart for each linked array/vector */ 11047c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 11147c6ae99SBarry Smith ierr = MPI_Comm_size(((PetscObject)dm)->comm,&size);CHKERRQ(ierr); 11247c6ae99SBarry Smith while (next) { 11347c6ae99SBarry Smith next->rstart = nprev; 11406ebdd98SJed Brown nprev += next->n; 11547c6ae99SBarry Smith next->grstart = com->rstart + next->rstart; 11647c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 11747c6ae99SBarry Smith ierr = MPI_Bcast(&next->grstart,1,MPIU_INT,next->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 11847c6ae99SBarry Smith } else { 11947c6ae99SBarry Smith ierr = PetscMalloc(size*sizeof(PetscInt),&next->grstarts);CHKERRQ(ierr); 12047c6ae99SBarry Smith ierr = MPI_Allgather(&next->grstart,1,MPIU_INT,next->grstarts,1,MPIU_INT,((PetscObject)dm)->comm);CHKERRQ(ierr); 12147c6ae99SBarry Smith } 12247c6ae99SBarry Smith next = next->next; 12347c6ae99SBarry Smith } 12447c6ae99SBarry Smith com->setup = PETSC_TRUE; 12547c6ae99SBarry Smith PetscFunctionReturn(0); 12647c6ae99SBarry Smith } 12747c6ae99SBarry Smith 12847c6ae99SBarry Smith 12947c6ae99SBarry Smith #undef __FUNCT__ 13047c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetAccess_Array" 13147c6ae99SBarry Smith PetscErrorCode DMCompositeGetAccess_Array(DM dm,struct DMCompositeLink *mine,Vec vec,PetscScalar **array) 13247c6ae99SBarry Smith { 13347c6ae99SBarry Smith PetscErrorCode ierr; 13447c6ae99SBarry Smith PetscScalar *varray; 13547c6ae99SBarry Smith PetscMPIInt rank; 13647c6ae99SBarry Smith 13747c6ae99SBarry Smith PetscFunctionBegin; 13847c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 13947c6ae99SBarry Smith if (array) { 14047c6ae99SBarry Smith if (rank == mine->rank) { 14147c6ae99SBarry Smith ierr = VecGetArray(vec,&varray);CHKERRQ(ierr); 14247c6ae99SBarry Smith *array = varray + mine->rstart; 14347c6ae99SBarry Smith ierr = VecRestoreArray(vec,&varray);CHKERRQ(ierr); 14447c6ae99SBarry Smith } else { 14547c6ae99SBarry Smith *array = 0; 14647c6ae99SBarry Smith } 14747c6ae99SBarry Smith } 14847c6ae99SBarry Smith PetscFunctionReturn(0); 14947c6ae99SBarry Smith } 15047c6ae99SBarry Smith 15147c6ae99SBarry Smith #undef __FUNCT__ 15247c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetAccess_DM" 15347c6ae99SBarry Smith PetscErrorCode DMCompositeGetAccess_DM(DM dm,struct DMCompositeLink *mine,Vec vec,Vec *global) 15447c6ae99SBarry Smith { 15547c6ae99SBarry Smith PetscErrorCode ierr; 15647c6ae99SBarry Smith PetscScalar *array; 15747c6ae99SBarry Smith 15847c6ae99SBarry Smith PetscFunctionBegin; 15947c6ae99SBarry Smith if (global) { 16047c6ae99SBarry Smith ierr = DMGetGlobalVector(mine->dm,global);CHKERRQ(ierr); 16147c6ae99SBarry Smith ierr = VecGetArray(vec,&array);CHKERRQ(ierr); 16247c6ae99SBarry Smith ierr = VecPlaceArray(*global,array+mine->rstart);CHKERRQ(ierr); 16347c6ae99SBarry Smith ierr = VecRestoreArray(vec,&array);CHKERRQ(ierr); 16447c6ae99SBarry Smith } 16547c6ae99SBarry Smith PetscFunctionReturn(0); 16647c6ae99SBarry Smith } 16747c6ae99SBarry Smith 16847c6ae99SBarry Smith #undef __FUNCT__ 16947c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreAccess_Array" 17047c6ae99SBarry Smith PetscErrorCode DMCompositeRestoreAccess_Array(DM dm,struct DMCompositeLink *mine,Vec vec,PetscScalar **array) 17147c6ae99SBarry Smith { 17247c6ae99SBarry Smith PetscFunctionBegin; 17347c6ae99SBarry Smith PetscFunctionReturn(0); 17447c6ae99SBarry Smith } 17547c6ae99SBarry Smith 17647c6ae99SBarry Smith #undef __FUNCT__ 17747c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreAccess_DM" 17847c6ae99SBarry Smith PetscErrorCode DMCompositeRestoreAccess_DM(DM dm,struct DMCompositeLink *mine,Vec vec,Vec *global) 17947c6ae99SBarry Smith { 18047c6ae99SBarry Smith PetscErrorCode ierr; 18147c6ae99SBarry Smith 18247c6ae99SBarry Smith PetscFunctionBegin; 18347c6ae99SBarry Smith if (global) { 18447c6ae99SBarry Smith ierr = VecResetArray(*global);CHKERRQ(ierr); 18547c6ae99SBarry Smith ierr = DMRestoreGlobalVector(mine->dm,global);CHKERRQ(ierr); 18647c6ae99SBarry Smith } 18747c6ae99SBarry Smith PetscFunctionReturn(0); 18847c6ae99SBarry Smith } 18947c6ae99SBarry Smith 19047c6ae99SBarry Smith #undef __FUNCT__ 19147c6ae99SBarry Smith #define __FUNCT__ "DMCompositeScatter_Array" 19247c6ae99SBarry Smith PetscErrorCode DMCompositeScatter_Array(DM dm,struct DMCompositeLink *mine,Vec vec,PetscScalar *array) 19347c6ae99SBarry Smith { 19447c6ae99SBarry Smith PetscErrorCode ierr; 19547c6ae99SBarry Smith PetscScalar *varray; 19647c6ae99SBarry Smith PetscMPIInt rank; 19747c6ae99SBarry Smith 19847c6ae99SBarry Smith PetscFunctionBegin; 19947c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 20047c6ae99SBarry Smith if (rank == mine->rank) { 20147c6ae99SBarry Smith ierr = VecGetArray(vec,&varray);CHKERRQ(ierr); 20247c6ae99SBarry Smith ierr = PetscMemcpy(array,varray+mine->rstart,mine->n*sizeof(PetscScalar));CHKERRQ(ierr); 20347c6ae99SBarry Smith ierr = VecRestoreArray(vec,&varray);CHKERRQ(ierr); 20447c6ae99SBarry Smith } 20506ebdd98SJed Brown ierr = MPI_Bcast(array,mine->nlocal,MPIU_SCALAR,mine->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 20647c6ae99SBarry Smith PetscFunctionReturn(0); 20747c6ae99SBarry Smith } 20847c6ae99SBarry Smith 20947c6ae99SBarry Smith #undef __FUNCT__ 21047c6ae99SBarry Smith #define __FUNCT__ "DMCompositeScatter_DM" 21147c6ae99SBarry Smith PetscErrorCode DMCompositeScatter_DM(DM dm,struct DMCompositeLink *mine,Vec vec,Vec local) 21247c6ae99SBarry Smith { 21347c6ae99SBarry Smith PetscErrorCode ierr; 21447c6ae99SBarry Smith PetscScalar *array; 21547c6ae99SBarry Smith Vec global; 21647c6ae99SBarry Smith 21747c6ae99SBarry Smith PetscFunctionBegin; 21847c6ae99SBarry Smith ierr = DMGetGlobalVector(mine->dm,&global);CHKERRQ(ierr); 21947c6ae99SBarry Smith ierr = VecGetArray(vec,&array);CHKERRQ(ierr); 22047c6ae99SBarry Smith ierr = VecPlaceArray(global,array+mine->rstart);CHKERRQ(ierr); 22147c6ae99SBarry Smith ierr = DMGlobalToLocalBegin(mine->dm,global,INSERT_VALUES,local);CHKERRQ(ierr); 22247c6ae99SBarry Smith ierr = DMGlobalToLocalEnd(mine->dm,global,INSERT_VALUES,local);CHKERRQ(ierr); 22347c6ae99SBarry Smith ierr = VecRestoreArray(vec,&array);CHKERRQ(ierr); 22447c6ae99SBarry Smith ierr = VecResetArray(global);CHKERRQ(ierr); 22547c6ae99SBarry Smith ierr = DMRestoreGlobalVector(mine->dm,&global);CHKERRQ(ierr); 22647c6ae99SBarry Smith PetscFunctionReturn(0); 22747c6ae99SBarry Smith } 22847c6ae99SBarry Smith 22947c6ae99SBarry Smith #undef __FUNCT__ 23047c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGather_Array" 231acc1e270SJed Brown PetscErrorCode DMCompositeGather_Array(DM dm,struct DMCompositeLink *mine,Vec vec,InsertMode imode,const PetscScalar *array) 23247c6ae99SBarry Smith { 23347c6ae99SBarry Smith PetscErrorCode ierr; 23447c6ae99SBarry Smith PetscScalar *varray; 23547c6ae99SBarry Smith PetscMPIInt rank; 23647c6ae99SBarry Smith 23747c6ae99SBarry Smith PetscFunctionBegin; 23847c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 23947c6ae99SBarry Smith if (rank == mine->rank) { 24047c6ae99SBarry Smith ierr = VecGetArray(vec,&varray);CHKERRQ(ierr); 24147c6ae99SBarry Smith if (varray+mine->rstart == array) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"You need not DMCompositeGather() into objects obtained via DMCompositeGetAccess()"); 242acc1e270SJed Brown } 243df0c820aSJed Brown switch (imode) { 244df0c820aSJed Brown case INSERT_VALUES: 245acc1e270SJed Brown if (rank == mine->rank) { 24647c6ae99SBarry Smith ierr = PetscMemcpy(varray+mine->rstart,array,mine->n*sizeof(PetscScalar));CHKERRQ(ierr); 247acc1e270SJed Brown } 248df0c820aSJed Brown break; 249df0c820aSJed Brown case ADD_VALUES: { 250df0c820aSJed Brown PetscInt i; 251760fd489SMatthew G Knepley void *source; 252acc1e270SJed Brown PetscScalar *buffer,*dest; 253acc1e270SJed Brown if (rank == mine->rank) { 254acc1e270SJed Brown dest = &varray[mine->rstart]; 255acc1e270SJed Brown #if defined(PETSC_HAVE_MPI_IN_PLACE) 256acc1e270SJed Brown buffer = dest; 257acc1e270SJed Brown source = MPI_IN_PLACE; 258acc1e270SJed Brown #else 25906ebdd98SJed Brown ierr = PetscMalloc(mine->nlocal*sizeof(PetscScalar),&buffer);CHKERRQ(ierr); 260760fd489SMatthew G Knepley source = (void *) buffer; 261acc1e270SJed Brown #endif 26206ebdd98SJed Brown for (i=0; i<mine->nlocal; i++) buffer[i] = varray[mine->rstart+i] + array[i]; 263acc1e270SJed Brown } else { 264760fd489SMatthew G Knepley source = (void *) array; 265acc1e270SJed Brown dest = PETSC_NULL; 266acc1e270SJed Brown } 26706ebdd98SJed Brown ierr = MPI_Reduce(source,dest,mine->nlocal,MPIU_SCALAR,MPI_SUM,mine->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 268acc1e270SJed Brown #if !defined(PETSC_HAVE_MPI_IN_PLACE) 269acc1e270SJed Brown if (rank == mine->rank) {ierr = PetscFree(source);CHKERRQ(ierr);} 270acc1e270SJed Brown #endif 271df0c820aSJed Brown } break; 272df0c820aSJed Brown default: SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"imode"); 273df0c820aSJed Brown } 274acc1e270SJed Brown if (rank == mine->rank) {ierr = VecRestoreArray(vec,&varray);CHKERRQ(ierr);} 27547c6ae99SBarry Smith PetscFunctionReturn(0); 27647c6ae99SBarry Smith } 27747c6ae99SBarry Smith 27847c6ae99SBarry Smith #undef __FUNCT__ 27947c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGather_DM" 280df0c820aSJed Brown PetscErrorCode DMCompositeGather_DM(DM dm,struct DMCompositeLink *mine,Vec vec,InsertMode imode,Vec local) 28147c6ae99SBarry Smith { 28247c6ae99SBarry Smith PetscErrorCode ierr; 28347c6ae99SBarry Smith PetscScalar *array; 28447c6ae99SBarry Smith Vec global; 28547c6ae99SBarry Smith 28647c6ae99SBarry Smith PetscFunctionBegin; 28747c6ae99SBarry Smith ierr = DMGetGlobalVector(mine->dm,&global);CHKERRQ(ierr); 28847c6ae99SBarry Smith ierr = VecGetArray(vec,&array);CHKERRQ(ierr); 28947c6ae99SBarry Smith ierr = VecPlaceArray(global,array+mine->rstart);CHKERRQ(ierr); 290df0c820aSJed Brown ierr = DMLocalToGlobalBegin(mine->dm,local,imode,global);CHKERRQ(ierr); 291df0c820aSJed Brown ierr = DMLocalToGlobalEnd(mine->dm,local,imode,global);CHKERRQ(ierr); 29247c6ae99SBarry Smith ierr = VecRestoreArray(vec,&array);CHKERRQ(ierr); 29347c6ae99SBarry Smith ierr = VecResetArray(global);CHKERRQ(ierr); 29447c6ae99SBarry Smith ierr = DMRestoreGlobalVector(mine->dm,&global);CHKERRQ(ierr); 29547c6ae99SBarry Smith PetscFunctionReturn(0); 29647c6ae99SBarry Smith } 29747c6ae99SBarry Smith 29847c6ae99SBarry Smith /* ----------------------------------------------------------------------------------*/ 29947c6ae99SBarry Smith 30047c6ae99SBarry Smith #include <stdarg.h> 30147c6ae99SBarry Smith 30247c6ae99SBarry Smith #undef __FUNCT__ 30347c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetNumberDM" 30447c6ae99SBarry Smith /*@C 30547c6ae99SBarry Smith DMCompositeGetNumberDM - Get's the number of DM objects in the DMComposite 30647c6ae99SBarry Smith representation. 30747c6ae99SBarry Smith 30847c6ae99SBarry Smith Not Collective 30947c6ae99SBarry Smith 31047c6ae99SBarry Smith Input Parameter: 31147c6ae99SBarry Smith . dm - the packer object 31247c6ae99SBarry Smith 31347c6ae99SBarry Smith Output Parameter: 31447c6ae99SBarry Smith . nDM - the number of DMs 31547c6ae99SBarry Smith 31647c6ae99SBarry Smith Level: beginner 31747c6ae99SBarry Smith 31847c6ae99SBarry Smith @*/ 3197087cfbeSBarry Smith PetscErrorCode DMCompositeGetNumberDM(DM dm,PetscInt *nDM) 32047c6ae99SBarry Smith { 32147c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 32247c6ae99SBarry Smith PetscFunctionBegin; 32347c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 32447c6ae99SBarry Smith *nDM = com->nDM; 32547c6ae99SBarry Smith PetscFunctionReturn(0); 32647c6ae99SBarry Smith } 32747c6ae99SBarry Smith 32847c6ae99SBarry Smith 32947c6ae99SBarry Smith #undef __FUNCT__ 33047c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetAccess" 33147c6ae99SBarry Smith /*@C 33247c6ae99SBarry Smith DMCompositeGetAccess - Allows one to access the individual packed vectors in their global 33347c6ae99SBarry Smith representation. 33447c6ae99SBarry Smith 33547c6ae99SBarry Smith Collective on DMComposite 33647c6ae99SBarry Smith 33747c6ae99SBarry Smith Input Parameter: 33847c6ae99SBarry Smith + dm - the packer object 33947c6ae99SBarry Smith . gvec - the global vector 34047c6ae99SBarry Smith - ... - the individual sequential or parallel objects (arrays or vectors) 34147c6ae99SBarry Smith 34247c6ae99SBarry Smith Notes: Use DMCompositeRestoreAccess() to return the vectors when you no longer need them 34347c6ae99SBarry Smith 34447c6ae99SBarry Smith Level: advanced 34547c6ae99SBarry Smith 34647c6ae99SBarry Smith @*/ 3477087cfbeSBarry Smith PetscErrorCode DMCompositeGetAccess(DM dm,Vec gvec,...) 34847c6ae99SBarry Smith { 34947c6ae99SBarry Smith va_list Argp; 35047c6ae99SBarry Smith PetscErrorCode ierr; 35147c6ae99SBarry Smith struct DMCompositeLink *next; 35247c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 35347c6ae99SBarry Smith 35447c6ae99SBarry Smith PetscFunctionBegin; 35547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 35647c6ae99SBarry Smith PetscValidHeaderSpecific(gvec,VEC_CLASSID,2); 35747c6ae99SBarry Smith next = com->next; 35847c6ae99SBarry Smith if (!com->setup) { 359d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 36047c6ae99SBarry Smith } 36147c6ae99SBarry Smith 36247c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 36347c6ae99SBarry Smith va_start(Argp,gvec); 36447c6ae99SBarry Smith while (next) { 36547c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 36647c6ae99SBarry Smith PetscScalar **array; 36747c6ae99SBarry Smith array = va_arg(Argp, PetscScalar**); 36847c6ae99SBarry Smith ierr = DMCompositeGetAccess_Array(dm,next,gvec,array);CHKERRQ(ierr); 36947c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 37047c6ae99SBarry Smith Vec *vec; 37147c6ae99SBarry Smith vec = va_arg(Argp, Vec*); 37247c6ae99SBarry Smith ierr = DMCompositeGetAccess_DM(dm,next,gvec,vec);CHKERRQ(ierr); 37347c6ae99SBarry Smith } else { 37447c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 37547c6ae99SBarry Smith } 37647c6ae99SBarry Smith next = next->next; 37747c6ae99SBarry Smith } 37847c6ae99SBarry Smith va_end(Argp); 37947c6ae99SBarry Smith PetscFunctionReturn(0); 38047c6ae99SBarry Smith } 38147c6ae99SBarry Smith 38247c6ae99SBarry Smith #undef __FUNCT__ 38347c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreAccess" 38447c6ae99SBarry Smith /*@C 385aa219208SBarry Smith DMCompositeRestoreAccess - Returns the vectors obtained with DMCompositeGetAccess() 38647c6ae99SBarry Smith representation. 38747c6ae99SBarry Smith 38847c6ae99SBarry Smith Collective on DMComposite 38947c6ae99SBarry Smith 39047c6ae99SBarry Smith Input Parameter: 39147c6ae99SBarry Smith + dm - the packer object 39247c6ae99SBarry Smith . gvec - the global vector 39347c6ae99SBarry Smith - ... - the individual sequential or parallel objects (arrays or vectors) 39447c6ae99SBarry Smith 39547c6ae99SBarry Smith Level: advanced 39647c6ae99SBarry Smith 3970c010503SBarry Smith .seealso DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 3986eb61c8cSJed Brown DMCompositeGather(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeScatter(), 399aa219208SBarry Smith DMCompositeRestoreAccess(), DMCompositeGetAccess() 40047c6ae99SBarry Smith 40147c6ae99SBarry Smith @*/ 4027087cfbeSBarry Smith PetscErrorCode DMCompositeRestoreAccess(DM dm,Vec gvec,...) 40347c6ae99SBarry Smith { 40447c6ae99SBarry Smith va_list Argp; 40547c6ae99SBarry Smith PetscErrorCode ierr; 40647c6ae99SBarry Smith struct DMCompositeLink *next; 40747c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 40847c6ae99SBarry Smith 40947c6ae99SBarry Smith PetscFunctionBegin; 41047c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 41147c6ae99SBarry Smith PetscValidHeaderSpecific(gvec,VEC_CLASSID,2); 41247c6ae99SBarry Smith next = com->next; 41347c6ae99SBarry Smith if (!com->setup) { 414d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 41547c6ae99SBarry Smith } 41647c6ae99SBarry Smith 41747c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 41847c6ae99SBarry Smith va_start(Argp,gvec); 41947c6ae99SBarry Smith while (next) { 42047c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 42147c6ae99SBarry Smith PetscScalar **array; 42247c6ae99SBarry Smith array = va_arg(Argp, PetscScalar**); 42347c6ae99SBarry Smith ierr = DMCompositeRestoreAccess_Array(dm,next,gvec,array);CHKERRQ(ierr); 42447c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 42547c6ae99SBarry Smith Vec *vec; 42647c6ae99SBarry Smith vec = va_arg(Argp, Vec*); 42747c6ae99SBarry Smith ierr = DMCompositeRestoreAccess_DM(dm,next,gvec,vec);CHKERRQ(ierr); 42847c6ae99SBarry Smith } else { 42947c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 43047c6ae99SBarry Smith } 43147c6ae99SBarry Smith next = next->next; 43247c6ae99SBarry Smith } 43347c6ae99SBarry Smith va_end(Argp); 43447c6ae99SBarry Smith PetscFunctionReturn(0); 43547c6ae99SBarry Smith } 43647c6ae99SBarry Smith 43747c6ae99SBarry Smith #undef __FUNCT__ 43847c6ae99SBarry Smith #define __FUNCT__ "DMCompositeScatter" 43947c6ae99SBarry Smith /*@C 44047c6ae99SBarry Smith DMCompositeScatter - Scatters from a global packed vector into its individual local vectors 44147c6ae99SBarry Smith 44247c6ae99SBarry Smith Collective on DMComposite 44347c6ae99SBarry Smith 44447c6ae99SBarry Smith Input Parameter: 44547c6ae99SBarry Smith + dm - the packer object 44647c6ae99SBarry Smith . gvec - the global vector 4478fd8f222SJed Brown - ... - the individual sequential objects (arrays or vectors), PETSC_NULL for those that are not needed 44847c6ae99SBarry Smith 44947c6ae99SBarry Smith Level: advanced 45047c6ae99SBarry Smith 4510c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 4526eb61c8cSJed Brown DMCompositeGather(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 45347c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(), DMCompositeGetEntries() 45447c6ae99SBarry Smith 45547c6ae99SBarry Smith @*/ 4567087cfbeSBarry Smith PetscErrorCode DMCompositeScatter(DM dm,Vec gvec,...) 45747c6ae99SBarry Smith { 45847c6ae99SBarry Smith va_list Argp; 45947c6ae99SBarry Smith PetscErrorCode ierr; 46047c6ae99SBarry Smith struct DMCompositeLink *next; 4618fd8f222SJed Brown PetscInt cnt; 46247c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 46347c6ae99SBarry Smith 46447c6ae99SBarry Smith PetscFunctionBegin; 46547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 46647c6ae99SBarry Smith PetscValidHeaderSpecific(gvec,VEC_CLASSID,2); 46747c6ae99SBarry Smith if (!com->setup) { 468d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 46947c6ae99SBarry Smith } 47047c6ae99SBarry Smith 47147c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 47247c6ae99SBarry Smith va_start(Argp,gvec); 4738fd8f222SJed Brown for (cnt=3,next=com->next; next; cnt++,next=next->next) { 47447c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 47547c6ae99SBarry Smith PetscScalar *array; 47647c6ae99SBarry Smith array = va_arg(Argp, PetscScalar*); 477c72eef85SJed Brown if (array) PetscValidScalarPointer(array,cnt); 478c72eef85SJed Brown PetscValidLogicalCollectiveBool(dm,!!array,cnt); 4798fd8f222SJed Brown if (!array) continue; 48047c6ae99SBarry Smith ierr = DMCompositeScatter_Array(dm,next,gvec,array);CHKERRQ(ierr); 48147c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 48247c6ae99SBarry Smith Vec vec; 48347c6ae99SBarry Smith vec = va_arg(Argp, Vec); 4848fd8f222SJed Brown if (!vec) continue; 48547c6ae99SBarry Smith PetscValidHeaderSpecific(vec,VEC_CLASSID,cnt); 48647c6ae99SBarry Smith ierr = DMCompositeScatter_DM(dm,next,gvec,vec);CHKERRQ(ierr); 48747c6ae99SBarry Smith } else { 48847c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 48947c6ae99SBarry Smith } 49047c6ae99SBarry Smith } 49147c6ae99SBarry Smith va_end(Argp); 49247c6ae99SBarry Smith PetscFunctionReturn(0); 49347c6ae99SBarry Smith } 49447c6ae99SBarry Smith 49547c6ae99SBarry Smith #undef __FUNCT__ 49647c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGather" 49747c6ae99SBarry Smith /*@C 49847c6ae99SBarry Smith DMCompositeGather - Gathers into a global packed vector from its individual local vectors 49947c6ae99SBarry Smith 50047c6ae99SBarry Smith Collective on DMComposite 50147c6ae99SBarry Smith 50247c6ae99SBarry Smith Input Parameter: 50347c6ae99SBarry Smith + dm - the packer object 50447c6ae99SBarry Smith . gvec - the global vector 5058fd8f222SJed Brown - ... - the individual sequential objects (arrays or vectors), PETSC_NULL for any that are not needed 50647c6ae99SBarry Smith 50747c6ae99SBarry Smith Level: advanced 50847c6ae99SBarry Smith 5090c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 5106eb61c8cSJed Brown DMCompositeScatter(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 51147c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(), DMCompositeGetEntries() 51247c6ae99SBarry Smith 51347c6ae99SBarry Smith @*/ 5147087cfbeSBarry Smith PetscErrorCode DMCompositeGather(DM dm,Vec gvec,InsertMode imode,...) 51547c6ae99SBarry Smith { 51647c6ae99SBarry Smith va_list Argp; 51747c6ae99SBarry Smith PetscErrorCode ierr; 51847c6ae99SBarry Smith struct DMCompositeLink *next; 51947c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 5208fd8f222SJed Brown PetscInt cnt; 52147c6ae99SBarry Smith 52247c6ae99SBarry Smith PetscFunctionBegin; 52347c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 52447c6ae99SBarry Smith PetscValidHeaderSpecific(gvec,VEC_CLASSID,2); 52547c6ae99SBarry Smith if (!com->setup) { 526d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 52747c6ae99SBarry Smith } 52847c6ae99SBarry Smith 52947c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 530df0c820aSJed Brown va_start(Argp,imode); 5318fd8f222SJed Brown for (cnt=3,next=com->next; next; cnt++,next=next->next) { 53247c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 53347c6ae99SBarry Smith PetscScalar *array; 53447c6ae99SBarry Smith array = va_arg(Argp, PetscScalar*); 5358fd8f222SJed Brown if (!array) continue; 5368fd8f222SJed Brown PetscValidScalarPointer(array,cnt); 537df0c820aSJed Brown ierr = DMCompositeGather_Array(dm,next,gvec,imode,array);CHKERRQ(ierr); 53847c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 53947c6ae99SBarry Smith Vec vec; 54047c6ae99SBarry Smith vec = va_arg(Argp, Vec); 5418fd8f222SJed Brown if (!vec) continue; 5428fd8f222SJed Brown PetscValidHeaderSpecific(vec,VEC_CLASSID,cnt); 543df0c820aSJed Brown ierr = DMCompositeGather_DM(dm,next,gvec,imode,vec);CHKERRQ(ierr); 54447c6ae99SBarry Smith } else { 54547c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 54647c6ae99SBarry Smith } 54747c6ae99SBarry Smith } 54847c6ae99SBarry Smith va_end(Argp); 54947c6ae99SBarry Smith PetscFunctionReturn(0); 55047c6ae99SBarry Smith } 55147c6ae99SBarry Smith 55247c6ae99SBarry Smith #undef __FUNCT__ 55347c6ae99SBarry Smith #define __FUNCT__ "DMCompositeAddArray" 55447c6ae99SBarry Smith /*@C 55547c6ae99SBarry Smith DMCompositeAddArray - adds an "redundant" array to a DMComposite. The array values will 55647c6ae99SBarry Smith be stored in part of the array on process orank. 55747c6ae99SBarry Smith 55847c6ae99SBarry Smith Collective on DMComposite 55947c6ae99SBarry Smith 56047c6ae99SBarry Smith Input Parameter: 56147c6ae99SBarry Smith + dm - the packer object 56247c6ae99SBarry Smith . orank - the process on which the array entries officially live, this number must be 56347c6ae99SBarry Smith the same on all processes. 56447c6ae99SBarry Smith - n - the length of the array 56547c6ae99SBarry Smith 56647c6ae99SBarry Smith Level: advanced 56747c6ae99SBarry Smith 5680c010503SBarry Smith .seealso DMDestroy(), DMCompositeGather(), DMCompositeAddDM(), DMCreateGlobalVector(), 5696eb61c8cSJed Brown DMCompositeScatter(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 57047c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(), DMCompositeGetEntries() 57147c6ae99SBarry Smith 57247c6ae99SBarry Smith @*/ 5737087cfbeSBarry Smith PetscErrorCode DMCompositeAddArray(DM dm,PetscMPIInt orank,PetscInt n) 57447c6ae99SBarry Smith { 57547c6ae99SBarry Smith struct DMCompositeLink *mine,*next; 57647c6ae99SBarry Smith PetscErrorCode ierr; 57747c6ae99SBarry Smith PetscMPIInt rank; 57847c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 57947c6ae99SBarry Smith 58047c6ae99SBarry Smith PetscFunctionBegin; 58147c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 58247c6ae99SBarry Smith next = com->next; 58347c6ae99SBarry Smith if (com->setup) { 58447c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_ARG_WRONGSTATE,"Cannot add an array once you have used the DMComposite"); 58547c6ae99SBarry Smith } 58647c6ae99SBarry Smith #if defined(PETSC_USE_DEBUG) 58747c6ae99SBarry Smith { 58847c6ae99SBarry Smith PetscMPIInt orankmax; 58947c6ae99SBarry Smith ierr = MPI_Allreduce(&orank,&orankmax,1,MPI_INT,MPI_MAX,((PetscObject)dm)->comm);CHKERRQ(ierr); 59047c6ae99SBarry Smith if (orank != orankmax) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"orank %d must be equal on all processes, another process has value %d",orank,orankmax); 59147c6ae99SBarry Smith } 59247c6ae99SBarry Smith #endif 59347c6ae99SBarry Smith 59447c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 59547c6ae99SBarry Smith /* create new link */ 59647c6ae99SBarry Smith ierr = PetscNew(struct DMCompositeLink,&mine);CHKERRQ(ierr); 59706ebdd98SJed Brown mine->nlocal = n; 59806ebdd98SJed Brown mine->n = (rank == orank) ? n : 0; 59947c6ae99SBarry Smith mine->rank = orank; 60047c6ae99SBarry Smith mine->dm = PETSC_NULL; 60147c6ae99SBarry Smith mine->type = DMCOMPOSITE_ARRAY; 60247c6ae99SBarry Smith mine->next = PETSC_NULL; 60347c6ae99SBarry Smith if (rank == mine->rank) {com->n += n;com->nmine++;} 60447c6ae99SBarry Smith 60547c6ae99SBarry Smith /* add to end of list */ 60647c6ae99SBarry Smith if (!next) { 60747c6ae99SBarry Smith com->next = mine; 60847c6ae99SBarry Smith } else { 60947c6ae99SBarry Smith while (next->next) next = next->next; 61047c6ae99SBarry Smith next->next = mine; 61147c6ae99SBarry Smith } 61247c6ae99SBarry Smith com->nredundant++; 61347c6ae99SBarry Smith PetscFunctionReturn(0); 61447c6ae99SBarry Smith } 61547c6ae99SBarry Smith 61647c6ae99SBarry Smith #undef __FUNCT__ 61747c6ae99SBarry Smith #define __FUNCT__ "DMCompositeAddDM" 61847c6ae99SBarry Smith /*@C 619aa219208SBarry Smith DMCompositeAddDM - adds a DM vector to a DMComposite 62047c6ae99SBarry Smith 62147c6ae99SBarry Smith Collective on DMComposite 62247c6ae99SBarry Smith 62347c6ae99SBarry Smith Input Parameter: 62447c6ae99SBarry Smith + dm - the packer object 62547c6ae99SBarry Smith - dm - the DM object, if the DM is a da you will need to caste it with a (DM) 62647c6ae99SBarry Smith 62747c6ae99SBarry Smith Level: advanced 62847c6ae99SBarry Smith 6290c010503SBarry Smith .seealso DMDestroy(), DMCompositeGather(), DMCompositeAddDM(), DMCreateGlobalVector(), 6306eb61c8cSJed Brown DMCompositeScatter(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 63147c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(), DMCompositeGetEntries() 63247c6ae99SBarry Smith 63347c6ae99SBarry Smith @*/ 6347087cfbeSBarry Smith PetscErrorCode DMCompositeAddDM(DM dmc,DM dm) 63547c6ae99SBarry Smith { 63647c6ae99SBarry Smith PetscErrorCode ierr; 63706ebdd98SJed Brown PetscInt n,nlocal; 63847c6ae99SBarry Smith struct DMCompositeLink *mine,*next; 63906ebdd98SJed Brown Vec global,local; 64047c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dmc->data; 64147c6ae99SBarry Smith 64247c6ae99SBarry Smith PetscFunctionBegin; 64347c6ae99SBarry Smith PetscValidHeaderSpecific(dmc,DM_CLASSID,1); 64447c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 64547c6ae99SBarry Smith next = com->next; 646aa219208SBarry Smith if (com->setup) SETERRQ(((PetscObject)dmc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Cannot add a DM once you have used the DMComposite"); 64747c6ae99SBarry Smith 64847c6ae99SBarry Smith /* create new link */ 64947c6ae99SBarry Smith ierr = PetscNew(struct DMCompositeLink,&mine);CHKERRQ(ierr); 65047c6ae99SBarry Smith ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr); 65147c6ae99SBarry Smith ierr = DMGetGlobalVector(dm,&global);CHKERRQ(ierr); 65247c6ae99SBarry Smith ierr = VecGetLocalSize(global,&n);CHKERRQ(ierr); 65347c6ae99SBarry Smith ierr = DMRestoreGlobalVector(dm,&global);CHKERRQ(ierr); 65406ebdd98SJed Brown ierr = DMGetLocalVector(dm,&local);CHKERRQ(ierr); 65506ebdd98SJed Brown ierr = VecGetSize(local,&nlocal);CHKERRQ(ierr); 65606ebdd98SJed Brown ierr = DMRestoreLocalVector(dm,&local);CHKERRQ(ierr); 65747c6ae99SBarry Smith mine->n = n; 65806ebdd98SJed Brown mine->nlocal = nlocal; 65947c6ae99SBarry Smith mine->dm = dm; 66047c6ae99SBarry Smith mine->type = DMCOMPOSITE_DM; 66147c6ae99SBarry Smith mine->next = PETSC_NULL; 66247c6ae99SBarry Smith com->n += n; 66347c6ae99SBarry Smith 66447c6ae99SBarry Smith /* add to end of list */ 66547c6ae99SBarry Smith if (!next) { 66647c6ae99SBarry Smith com->next = mine; 66747c6ae99SBarry Smith } else { 66847c6ae99SBarry Smith while (next->next) next = next->next; 66947c6ae99SBarry Smith next->next = mine; 67047c6ae99SBarry Smith } 67147c6ae99SBarry Smith com->nDM++; 67247c6ae99SBarry Smith com->nmine++; 67347c6ae99SBarry Smith PetscFunctionReturn(0); 67447c6ae99SBarry Smith } 67547c6ae99SBarry Smith 6767087cfbeSBarry Smith extern PetscErrorCode VecView_MPI(Vec,PetscViewer); 67747c6ae99SBarry Smith EXTERN_C_BEGIN 67847c6ae99SBarry Smith #undef __FUNCT__ 67947c6ae99SBarry Smith #define __FUNCT__ "VecView_DMComposite" 6807087cfbeSBarry Smith PetscErrorCode VecView_DMComposite(Vec gvec,PetscViewer viewer) 68147c6ae99SBarry Smith { 68247c6ae99SBarry Smith DM dm; 68347c6ae99SBarry Smith PetscErrorCode ierr; 68447c6ae99SBarry Smith struct DMCompositeLink *next; 68547c6ae99SBarry Smith PetscBool isdraw; 686cef07954SSatish Balay DM_Composite *com; 68747c6ae99SBarry Smith 68847c6ae99SBarry Smith PetscFunctionBegin; 6893c0c59f3SBarry Smith ierr = PetscObjectQuery((PetscObject)gvec,"DM",(PetscObject*)&dm);CHKERRQ(ierr); 69047c6ae99SBarry Smith if (!dm) SETERRQ(((PetscObject)gvec)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMComposite"); 69147c6ae99SBarry Smith com = (DM_Composite*)dm->data; 69247c6ae99SBarry Smith next = com->next; 69347c6ae99SBarry Smith 69447c6ae99SBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 69547c6ae99SBarry Smith if (!isdraw) { 69647c6ae99SBarry Smith /* do I really want to call this? */ 69747c6ae99SBarry Smith ierr = VecView_MPI(gvec,viewer);CHKERRQ(ierr); 69847c6ae99SBarry Smith } else { 69947c6ae99SBarry Smith PetscInt cnt = 0; 70047c6ae99SBarry Smith 70147c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 70247c6ae99SBarry Smith while (next) { 70347c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 70447c6ae99SBarry Smith PetscScalar *array; 70547c6ae99SBarry Smith ierr = DMCompositeGetAccess_Array(dm,next,gvec,&array);CHKERRQ(ierr); 70647c6ae99SBarry Smith 70747c6ae99SBarry Smith /*skip it for now */ 70847c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 70947c6ae99SBarry Smith Vec vec; 71047c6ae99SBarry Smith PetscInt bs; 71147c6ae99SBarry Smith 71247c6ae99SBarry Smith ierr = DMCompositeGetAccess_DM(dm,next,gvec,&vec);CHKERRQ(ierr); 71347c6ae99SBarry Smith ierr = VecView(vec,viewer);CHKERRQ(ierr); 71447c6ae99SBarry Smith ierr = VecGetBlockSize(vec,&bs);CHKERRQ(ierr); 71547c6ae99SBarry Smith ierr = DMCompositeRestoreAccess_DM(dm,next,gvec,&vec);CHKERRQ(ierr); 71647c6ae99SBarry Smith ierr = PetscViewerDrawBaseAdd(viewer,bs);CHKERRQ(ierr); 71747c6ae99SBarry Smith cnt += bs; 71847c6ae99SBarry Smith } else { 71947c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 72047c6ae99SBarry Smith } 72147c6ae99SBarry Smith next = next->next; 72247c6ae99SBarry Smith } 72347c6ae99SBarry Smith ierr = PetscViewerDrawBaseAdd(viewer,-cnt);CHKERRQ(ierr); 72447c6ae99SBarry Smith } 72547c6ae99SBarry Smith PetscFunctionReturn(0); 72647c6ae99SBarry Smith } 72747c6ae99SBarry Smith EXTERN_C_END 72847c6ae99SBarry Smith 72947c6ae99SBarry Smith 73047c6ae99SBarry Smith #undef __FUNCT__ 7310c010503SBarry Smith #define __FUNCT__ "DMCreateGlobalVector_Composite" 7327087cfbeSBarry Smith PetscErrorCode DMCreateGlobalVector_Composite(DM dm,Vec *gvec) 73347c6ae99SBarry Smith { 73447c6ae99SBarry Smith PetscErrorCode ierr; 73547c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 73647c6ae99SBarry Smith 73747c6ae99SBarry Smith PetscFunctionBegin; 73847c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 73947c6ae99SBarry Smith if (!com->setup) { 740d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 74147c6ae99SBarry Smith } 74247c6ae99SBarry Smith ierr = VecCreateMPI(((PetscObject)dm)->comm,com->n,com->N,gvec);CHKERRQ(ierr); 7433c0c59f3SBarry Smith ierr = PetscObjectCompose((PetscObject)*gvec,"DM",(PetscObject)dm);CHKERRQ(ierr); 74447c6ae99SBarry Smith ierr = VecSetOperation(*gvec,VECOP_VIEW,(void(*)(void))VecView_DMComposite);CHKERRQ(ierr); 74547c6ae99SBarry Smith PetscFunctionReturn(0); 74647c6ae99SBarry Smith } 74747c6ae99SBarry Smith 74847c6ae99SBarry Smith #undef __FUNCT__ 7490c010503SBarry Smith #define __FUNCT__ "DMCreateLocalVector_Composite" 7507087cfbeSBarry Smith PetscErrorCode DMCreateLocalVector_Composite(DM dm,Vec *lvec) 75147c6ae99SBarry Smith { 75247c6ae99SBarry Smith PetscErrorCode ierr; 75347c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 75447c6ae99SBarry Smith 75547c6ae99SBarry Smith PetscFunctionBegin; 75647c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 75747c6ae99SBarry Smith if (!com->setup) { 758d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 75947c6ae99SBarry Smith } 76047c6ae99SBarry Smith ierr = VecCreateSeq(((PetscObject)dm)->comm,com->nghost,lvec);CHKERRQ(ierr); 7613c0c59f3SBarry Smith ierr = PetscObjectCompose((PetscObject)*lvec,"DM",(PetscObject)dm);CHKERRQ(ierr); 76247c6ae99SBarry Smith PetscFunctionReturn(0); 76347c6ae99SBarry Smith } 76447c6ae99SBarry Smith 76547c6ae99SBarry Smith #undef __FUNCT__ 7666eb61c8cSJed Brown #define __FUNCT__ "DMCompositeGetISLocalToGlobalMappings" 76747c6ae99SBarry Smith /*@C 7686eb61c8cSJed Brown DMCompositeGetISLocalToGlobalMappings - gets an ISLocalToGlobalMapping for each DM/array in the DMComposite, maps to the composite global space 76947c6ae99SBarry Smith 77006ebdd98SJed Brown Collective on DM 77147c6ae99SBarry Smith 77247c6ae99SBarry Smith Input Parameter: 77347c6ae99SBarry Smith . dm - the packer object 77447c6ae99SBarry Smith 77547c6ae99SBarry Smith Output Parameters: 7766eb61c8cSJed Brown . ltogs - the individual mappings for each packed vector/array. Note that this includes 777aa219208SBarry Smith all the ghost points that individual ghosted DMDA's may have. Also each process has an 7786eb61c8cSJed Brown mapping for EACH redundant array (not just the local redundant arrays). 77947c6ae99SBarry Smith 78047c6ae99SBarry Smith Level: advanced 78147c6ae99SBarry Smith 78247c6ae99SBarry Smith Notes: 7836eb61c8cSJed Brown Each entry of ltogs should be destroyed with ISLocalToGlobalMappingDestroy(), the ltogs array should be freed with PetscFree(). 78447c6ae99SBarry Smith 7850c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 78647c6ae99SBarry Smith DMCompositeGather(), DMCompositeCreate(), DMCompositeGetAccess(), DMCompositeScatter(), 78747c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(),DMCompositeGetEntries() 78847c6ae99SBarry Smith 78947c6ae99SBarry Smith @*/ 7907087cfbeSBarry Smith PetscErrorCode DMCompositeGetISLocalToGlobalMappings(DM dm,ISLocalToGlobalMapping **ltogs) 79147c6ae99SBarry Smith { 79247c6ae99SBarry Smith PetscErrorCode ierr; 79347c6ae99SBarry Smith PetscInt i,*idx,n,cnt; 79447c6ae99SBarry Smith struct DMCompositeLink *next; 79547c6ae99SBarry Smith PetscMPIInt rank; 79647c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 79747c6ae99SBarry Smith 79847c6ae99SBarry Smith PetscFunctionBegin; 79947c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 800acc1e270SJed Brown ierr = PetscMalloc((com->nDM+com->nredundant)*sizeof(ISLocalToGlobalMapping),ltogs);CHKERRQ(ierr); 80147c6ae99SBarry Smith next = com->next; 80247c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 80347c6ae99SBarry Smith 80447c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 80547c6ae99SBarry Smith cnt = 0; 80647c6ae99SBarry Smith while (next) { 80747c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 80806ebdd98SJed Brown ierr = PetscMalloc(next->nlocal*sizeof(PetscInt),&idx);CHKERRQ(ierr); 80947c6ae99SBarry Smith if (rank == next->rank) { 81006ebdd98SJed Brown for (i=0; i<next->nlocal; i++) idx[i] = next->grstart + i; 81147c6ae99SBarry Smith } 81206ebdd98SJed Brown ierr = MPI_Bcast(idx,next->nlocal,MPIU_INT,next->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 81306ebdd98SJed Brown ierr = ISLocalToGlobalMappingCreate(((PetscObject)dm)->comm,next->nlocal,idx,PETSC_OWN_POINTER,&(*ltogs)[cnt]);CHKERRQ(ierr); 81447c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 8156eb61c8cSJed Brown ISLocalToGlobalMapping ltog; 8166eb61c8cSJed Brown PetscMPIInt size; 81786994e45SJed Brown const PetscInt *suboff,*indices; 8186eb61c8cSJed Brown Vec global; 81947c6ae99SBarry Smith 8206eb61c8cSJed Brown /* Get sub-DM global indices for each local dof */ 8211411c6eeSJed Brown ierr = DMGetLocalToGlobalMapping(next->dm,<og);CHKERRQ(ierr); 8226eb61c8cSJed Brown ierr = ISLocalToGlobalMappingGetSize(ltog,&n);CHKERRQ(ierr); 82386994e45SJed Brown ierr = ISLocalToGlobalMappingGetIndices(ltog,&indices);CHKERRQ(ierr); 82447c6ae99SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&idx);CHKERRQ(ierr); 82547c6ae99SBarry Smith 8266eb61c8cSJed Brown /* Get the offsets for the sub-DM global vector */ 8276eb61c8cSJed Brown ierr = DMGetGlobalVector(next->dm,&global);CHKERRQ(ierr); 8286eb61c8cSJed Brown ierr = VecGetOwnershipRanges(global,&suboff);CHKERRQ(ierr); 8296eb61c8cSJed Brown ierr = MPI_Comm_size(((PetscObject)global)->comm,&size);CHKERRQ(ierr); 8306eb61c8cSJed Brown 8316eb61c8cSJed Brown /* Shift the sub-DM definition of the global space to the composite global space */ 8326eb61c8cSJed Brown for (i=0; i<n; i++) { 83386994e45SJed Brown PetscInt subi = indices[i],lo = 0,hi = size,t; 8346eb61c8cSJed Brown /* Binary search to find which rank owns subi */ 8356eb61c8cSJed Brown while (hi-lo > 1) { 8366eb61c8cSJed Brown t = lo + (hi-lo)/2; 8376eb61c8cSJed Brown if (suboff[t] > subi) hi = t; 8386eb61c8cSJed Brown else lo = t; 8396eb61c8cSJed Brown } 8406eb61c8cSJed Brown idx[i] = subi - suboff[lo] + next->grstarts[lo]; 8416eb61c8cSJed Brown } 84286994e45SJed Brown ierr = ISLocalToGlobalMappingRestoreIndices(ltog,&indices);CHKERRQ(ierr); 8436eb61c8cSJed Brown ierr = ISLocalToGlobalMappingCreate(((PetscObject)dm)->comm,n,idx,PETSC_OWN_POINTER,&(*ltogs)[cnt]);CHKERRQ(ierr); 8446eb61c8cSJed Brown ierr = DMRestoreGlobalVector(next->dm,&global);CHKERRQ(ierr); 8456eb61c8cSJed Brown } else SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 84647c6ae99SBarry Smith next = next->next; 84747c6ae99SBarry Smith cnt++; 84847c6ae99SBarry Smith } 84947c6ae99SBarry Smith PetscFunctionReturn(0); 85047c6ae99SBarry Smith } 85147c6ae99SBarry Smith 85247c6ae99SBarry Smith #undef __FUNCT__ 85387c85e80SJed Brown #define __FUNCT__ "DMCompositeGetLocalISs" 85487c85e80SJed Brown /*@C 85587c85e80SJed Brown DMCompositeGetLocalISs - Gets index sets for each DM/array component of a composite local vector 85687c85e80SJed Brown 85787c85e80SJed Brown Not Collective 85887c85e80SJed Brown 85987c85e80SJed Brown Input Arguments: 86087c85e80SJed Brown . dm - composite DM 86187c85e80SJed Brown 86287c85e80SJed Brown Output Arguments: 86387c85e80SJed Brown . is - array of serial index sets for each each component of the DMComposite 86487c85e80SJed Brown 86587c85e80SJed Brown Level: intermediate 86687c85e80SJed Brown 86787c85e80SJed Brown Notes: 86887c85e80SJed Brown At present, a composite local vector does not normally exist. This function is used to provide index sets for 86987c85e80SJed Brown MatGetLocalSubMatrix(). In the future, the scatters for each entry in the DMComposite may be be merged into a single 87087c85e80SJed Brown scatter to a composite local vector. 87187c85e80SJed Brown 87287c85e80SJed Brown To get the composite global indices at all local points (including ghosts), use DMCompositeGetISLocalToGlobalMappings(). 87387c85e80SJed Brown 87487c85e80SJed Brown To get index sets for pieces of the composite global vector, use DMCompositeGetGlobalISs(). 87587c85e80SJed Brown 87687c85e80SJed Brown Each returned IS should be destroyed with ISDestroy(), the array should be freed with PetscFree(). 87787c85e80SJed Brown 87887c85e80SJed Brown .seealso: DMCompositeGetGlobalISs(), DMCompositeGetISLocalToGlobalMappings(), MatGetLocalSubMatrix(), MatCreateLocalRef() 87987c85e80SJed Brown @*/ 8807087cfbeSBarry Smith PetscErrorCode DMCompositeGetLocalISs(DM dm,IS **is) 88187c85e80SJed Brown { 88287c85e80SJed Brown PetscErrorCode ierr; 88387c85e80SJed Brown DM_Composite *com = (DM_Composite*)dm->data; 88487c85e80SJed Brown struct DMCompositeLink *link; 88587c85e80SJed Brown PetscInt cnt,start; 88687c85e80SJed Brown 88787c85e80SJed Brown PetscFunctionBegin; 88887c85e80SJed Brown PetscValidHeaderSpecific(dm,DM_CLASSID,1); 88987c85e80SJed Brown PetscValidPointer(is,2); 89087c85e80SJed Brown ierr = PetscMalloc(com->nmine*sizeof(IS),is);CHKERRQ(ierr); 89106ebdd98SJed Brown for (cnt=0,start=0,link=com->next; link; start+=link->nlocal,cnt++,link=link->next) { 89206ebdd98SJed Brown ierr = ISCreateStride(PETSC_COMM_SELF,link->nlocal,start,1,&(*is)[cnt]);CHKERRQ(ierr); 893520db06cSJed Brown if (link->type == DMCOMPOSITE_DM) { 894520db06cSJed Brown PetscInt bs; 8951411c6eeSJed Brown ierr = DMGetBlockSize(link->dm,&bs);CHKERRQ(ierr); 896520db06cSJed Brown ierr = ISSetBlockSize((*is)[cnt],bs);CHKERRQ(ierr); 897520db06cSJed Brown } 89887c85e80SJed Brown } 89987c85e80SJed Brown PetscFunctionReturn(0); 90087c85e80SJed Brown } 90187c85e80SJed Brown 90287c85e80SJed Brown #undef __FUNCT__ 90347c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetGlobalISs" 90447c6ae99SBarry Smith /*@C 90547c6ae99SBarry Smith DMCompositeGetGlobalISs - Gets the index sets for each composed object 90647c6ae99SBarry Smith 90747c6ae99SBarry Smith Collective on DMComposite 90847c6ae99SBarry Smith 90947c6ae99SBarry Smith Input Parameter: 91047c6ae99SBarry Smith . dm - the packer object 91147c6ae99SBarry Smith 91247c6ae99SBarry Smith Output Parameters: 91347c6ae99SBarry Smith . is - the array of index sets 91447c6ae99SBarry Smith 91547c6ae99SBarry Smith Level: advanced 91647c6ae99SBarry Smith 91747c6ae99SBarry Smith Notes: 91847c6ae99SBarry Smith The is entries should be destroyed with ISDestroy(), the is array should be freed with PetscFree() 91947c6ae99SBarry Smith 92047c6ae99SBarry Smith These could be used to extract a subset of vector entries for a "multi-physics" preconditioner 92147c6ae99SBarry Smith 9226eb61c8cSJed Brown Use DMCompositeGetLocalISs() for index sets in the packed local numbering, and 9236eb61c8cSJed Brown DMCompositeGetISLocalToGlobalMappings() for to map local sub-DM (including ghost) indices to packed global 9246eb61c8cSJed Brown indices. 92547c6ae99SBarry Smith 9260c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 92747c6ae99SBarry Smith DMCompositeGather(), DMCompositeCreate(), DMCompositeGetAccess(), DMCompositeScatter(), 92847c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(),DMCompositeGetEntries() 92947c6ae99SBarry Smith 93047c6ae99SBarry Smith @*/ 9316eb61c8cSJed Brown 9327087cfbeSBarry Smith PetscErrorCode DMCompositeGetGlobalISs(DM dm,IS *is[]) 93347c6ae99SBarry Smith { 93447c6ae99SBarry Smith PetscErrorCode ierr; 93547c6ae99SBarry Smith PetscInt cnt = 0,*idx,i; 93647c6ae99SBarry Smith struct DMCompositeLink *next; 93747c6ae99SBarry Smith PetscMPIInt rank; 93847c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 93947c6ae99SBarry Smith 94047c6ae99SBarry Smith PetscFunctionBegin; 94147c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 94206ebdd98SJed Brown ierr = PetscMalloc((com->nDM+com->nredundant)*sizeof(IS),is);CHKERRQ(ierr); 94347c6ae99SBarry Smith next = com->next; 94447c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 94547c6ae99SBarry Smith 94647c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 94747c6ae99SBarry Smith while (next) { 94847c6ae99SBarry Smith ierr = PetscMalloc(next->n*sizeof(PetscInt),&idx);CHKERRQ(ierr); 94947c6ae99SBarry Smith for (i=0; i<next->n; i++) idx[i] = next->grstart + i; 95047c6ae99SBarry Smith ierr = ISCreateGeneral(((PetscObject)dm)->comm,next->n,idx,PETSC_OWN_POINTER,&(*is)[cnt]);CHKERRQ(ierr); 95147c6ae99SBarry Smith cnt++; 95247c6ae99SBarry Smith next = next->next; 95347c6ae99SBarry Smith } 95447c6ae99SBarry Smith PetscFunctionReturn(0); 95547c6ae99SBarry Smith } 95647c6ae99SBarry Smith 95747c6ae99SBarry Smith /* -------------------------------------------------------------------------------------*/ 95847c6ae99SBarry Smith #undef __FUNCT__ 95947c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetLocalVectors_Array" 96047c6ae99SBarry Smith PetscErrorCode DMCompositeGetLocalVectors_Array(DM dm,struct DMCompositeLink *mine,PetscScalar **array) 96147c6ae99SBarry Smith { 96247c6ae99SBarry Smith PetscErrorCode ierr; 96347c6ae99SBarry Smith PetscFunctionBegin; 96447c6ae99SBarry Smith if (array) { 965c72eef85SJed Brown ierr = PetscMalloc(mine->nlocal*sizeof(PetscScalar),array);CHKERRQ(ierr); 96647c6ae99SBarry Smith } 96747c6ae99SBarry Smith PetscFunctionReturn(0); 96847c6ae99SBarry Smith } 96947c6ae99SBarry Smith 97047c6ae99SBarry Smith #undef __FUNCT__ 97147c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetLocalVectors_DM" 97247c6ae99SBarry Smith PetscErrorCode DMCompositeGetLocalVectors_DM(DM dm,struct DMCompositeLink *mine,Vec *local) 97347c6ae99SBarry Smith { 97447c6ae99SBarry Smith PetscErrorCode ierr; 97547c6ae99SBarry Smith PetscFunctionBegin; 97647c6ae99SBarry Smith if (local) { 97747c6ae99SBarry Smith ierr = DMGetLocalVector(mine->dm,local);CHKERRQ(ierr); 97847c6ae99SBarry Smith } 97947c6ae99SBarry Smith PetscFunctionReturn(0); 98047c6ae99SBarry Smith } 98147c6ae99SBarry Smith 98247c6ae99SBarry Smith #undef __FUNCT__ 98347c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreLocalVectors_Array" 98447c6ae99SBarry Smith PetscErrorCode DMCompositeRestoreLocalVectors_Array(DM dm,struct DMCompositeLink *mine,PetscScalar **array) 98547c6ae99SBarry Smith { 98647c6ae99SBarry Smith PetscErrorCode ierr; 98747c6ae99SBarry Smith PetscFunctionBegin; 98847c6ae99SBarry Smith if (array) { 98947c6ae99SBarry Smith ierr = PetscFree(*array);CHKERRQ(ierr); 99047c6ae99SBarry Smith } 99147c6ae99SBarry Smith PetscFunctionReturn(0); 99247c6ae99SBarry Smith } 99347c6ae99SBarry Smith 99447c6ae99SBarry Smith #undef __FUNCT__ 99547c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreLocalVectors_DM" 99647c6ae99SBarry Smith PetscErrorCode DMCompositeRestoreLocalVectors_DM(DM dm,struct DMCompositeLink *mine,Vec *local) 99747c6ae99SBarry Smith { 99847c6ae99SBarry Smith PetscErrorCode ierr; 99947c6ae99SBarry Smith PetscFunctionBegin; 100047c6ae99SBarry Smith if (local) { 100147c6ae99SBarry Smith ierr = DMRestoreLocalVector(mine->dm,local);CHKERRQ(ierr); 100247c6ae99SBarry Smith } 100347c6ae99SBarry Smith PetscFunctionReturn(0); 100447c6ae99SBarry Smith } 100547c6ae99SBarry Smith 100647c6ae99SBarry Smith #undef __FUNCT__ 100747c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetLocalVectors" 100847c6ae99SBarry Smith /*@C 100947c6ae99SBarry Smith DMCompositeGetLocalVectors - Gets local vectors and arrays for each part of a DMComposite. 101047c6ae99SBarry Smith Use DMCompositeRestoreLocalVectors() to return them. 101147c6ae99SBarry Smith 101247c6ae99SBarry Smith Not Collective 101347c6ae99SBarry Smith 101447c6ae99SBarry Smith Input Parameter: 101547c6ae99SBarry Smith . dm - the packer object 101647c6ae99SBarry Smith 101747c6ae99SBarry Smith Output Parameter: 101847c6ae99SBarry Smith . ... - the individual sequential objects (arrays or vectors) 101947c6ae99SBarry Smith 102047c6ae99SBarry Smith Level: advanced 102147c6ae99SBarry Smith 10220c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 10236eb61c8cSJed Brown DMCompositeGather(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 102447c6ae99SBarry Smith DMCompositeRestoreLocalVectors(), DMCompositeScatter(), DMCompositeGetEntries() 102547c6ae99SBarry Smith 102647c6ae99SBarry Smith @*/ 10277087cfbeSBarry Smith PetscErrorCode DMCompositeGetLocalVectors(DM dm,...) 102847c6ae99SBarry Smith { 102947c6ae99SBarry Smith va_list Argp; 103047c6ae99SBarry Smith PetscErrorCode ierr; 103147c6ae99SBarry Smith struct DMCompositeLink *next; 103247c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 103347c6ae99SBarry Smith 103447c6ae99SBarry Smith PetscFunctionBegin; 103547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 103647c6ae99SBarry Smith next = com->next; 103747c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 103847c6ae99SBarry Smith va_start(Argp,dm); 103947c6ae99SBarry Smith while (next) { 104047c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 104147c6ae99SBarry Smith PetscScalar **array; 104247c6ae99SBarry Smith array = va_arg(Argp, PetscScalar**); 104347c6ae99SBarry Smith ierr = DMCompositeGetLocalVectors_Array(dm,next,array);CHKERRQ(ierr); 104447c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 104547c6ae99SBarry Smith Vec *vec; 104647c6ae99SBarry Smith vec = va_arg(Argp, Vec*); 104747c6ae99SBarry Smith ierr = DMCompositeGetLocalVectors_DM(dm,next,vec);CHKERRQ(ierr); 104847c6ae99SBarry Smith } else { 104947c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 105047c6ae99SBarry Smith } 105147c6ae99SBarry Smith next = next->next; 105247c6ae99SBarry Smith } 105347c6ae99SBarry Smith va_end(Argp); 105447c6ae99SBarry Smith PetscFunctionReturn(0); 105547c6ae99SBarry Smith } 105647c6ae99SBarry Smith 105747c6ae99SBarry Smith #undef __FUNCT__ 105847c6ae99SBarry Smith #define __FUNCT__ "DMCompositeRestoreLocalVectors" 105947c6ae99SBarry Smith /*@C 106047c6ae99SBarry Smith DMCompositeRestoreLocalVectors - Restores local vectors and arrays for each part of a DMComposite. 106147c6ae99SBarry Smith 106247c6ae99SBarry Smith Not Collective 106347c6ae99SBarry Smith 106447c6ae99SBarry Smith Input Parameter: 106547c6ae99SBarry Smith . dm - the packer object 106647c6ae99SBarry Smith 106747c6ae99SBarry Smith Output Parameter: 106847c6ae99SBarry Smith . ... - the individual sequential objects (arrays or vectors) 106947c6ae99SBarry Smith 107047c6ae99SBarry Smith Level: advanced 107147c6ae99SBarry Smith 10720c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 10736eb61c8cSJed Brown DMCompositeGather(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 107447c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeScatter(), DMCompositeGetEntries() 107547c6ae99SBarry Smith 107647c6ae99SBarry Smith @*/ 10777087cfbeSBarry Smith PetscErrorCode DMCompositeRestoreLocalVectors(DM dm,...) 107847c6ae99SBarry Smith { 107947c6ae99SBarry Smith va_list Argp; 108047c6ae99SBarry Smith PetscErrorCode ierr; 108147c6ae99SBarry Smith struct DMCompositeLink *next; 108247c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 108347c6ae99SBarry Smith 108447c6ae99SBarry Smith PetscFunctionBegin; 108547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 108647c6ae99SBarry Smith next = com->next; 108747c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 108847c6ae99SBarry Smith va_start(Argp,dm); 108947c6ae99SBarry Smith while (next) { 109047c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 109147c6ae99SBarry Smith PetscScalar **array; 109247c6ae99SBarry Smith array = va_arg(Argp, PetscScalar**); 109347c6ae99SBarry Smith ierr = DMCompositeRestoreLocalVectors_Array(dm,next,array);CHKERRQ(ierr); 109447c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 109547c6ae99SBarry Smith Vec *vec; 109647c6ae99SBarry Smith vec = va_arg(Argp, Vec*); 109747c6ae99SBarry Smith ierr = DMCompositeRestoreLocalVectors_DM(dm,next,vec);CHKERRQ(ierr); 109847c6ae99SBarry Smith } else { 109947c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 110047c6ae99SBarry Smith } 110147c6ae99SBarry Smith next = next->next; 110247c6ae99SBarry Smith } 110347c6ae99SBarry Smith va_end(Argp); 110447c6ae99SBarry Smith PetscFunctionReturn(0); 110547c6ae99SBarry Smith } 110647c6ae99SBarry Smith 110747c6ae99SBarry Smith /* -------------------------------------------------------------------------------------*/ 110847c6ae99SBarry Smith #undef __FUNCT__ 110947c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetEntries_Array" 111047c6ae99SBarry Smith PetscErrorCode DMCompositeGetEntries_Array(DM dm,struct DMCompositeLink *mine,PetscInt *n) 111147c6ae99SBarry Smith { 111247c6ae99SBarry Smith PetscFunctionBegin; 111306ebdd98SJed Brown if (n) *n = mine->nlocal; 111447c6ae99SBarry Smith PetscFunctionReturn(0); 111547c6ae99SBarry Smith } 111647c6ae99SBarry Smith 111747c6ae99SBarry Smith #undef __FUNCT__ 111847c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetEntries_DM" 111947c6ae99SBarry Smith PetscErrorCode DMCompositeGetEntries_DM(DM dmi,struct DMCompositeLink *mine,DM *dm) 112047c6ae99SBarry Smith { 112147c6ae99SBarry Smith PetscFunctionBegin; 112247c6ae99SBarry Smith if (dm) *dm = mine->dm; 112347c6ae99SBarry Smith PetscFunctionReturn(0); 112447c6ae99SBarry Smith } 112547c6ae99SBarry Smith 112647c6ae99SBarry Smith #undef __FUNCT__ 112747c6ae99SBarry Smith #define __FUNCT__ "DMCompositeGetEntries" 112847c6ae99SBarry Smith /*@C 1129aa219208SBarry Smith DMCompositeGetEntries - Gets the DM, redundant size, etc for each entry in a DMComposite. 113047c6ae99SBarry Smith 113147c6ae99SBarry Smith Not Collective 113247c6ae99SBarry Smith 113347c6ae99SBarry Smith Input Parameter: 113447c6ae99SBarry Smith . dm - the packer object 113547c6ae99SBarry Smith 113647c6ae99SBarry Smith Output Parameter: 1137aa219208SBarry Smith . ... - the individual entries, DMs or integer sizes) 113847c6ae99SBarry Smith 113947c6ae99SBarry Smith Level: advanced 114047c6ae99SBarry Smith 11410c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCreateGlobalVector(), 11426eb61c8cSJed Brown DMCompositeGather(), DMCompositeCreate(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess(), 114347c6ae99SBarry Smith DMCompositeRestoreLocalVectors(), DMCompositeGetLocalVectors(), DMCompositeScatter(), 114447c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors() 114547c6ae99SBarry Smith 114647c6ae99SBarry Smith @*/ 11477087cfbeSBarry Smith PetscErrorCode DMCompositeGetEntries(DM dm,...) 114847c6ae99SBarry Smith { 114947c6ae99SBarry Smith va_list Argp; 115047c6ae99SBarry Smith PetscErrorCode ierr; 115147c6ae99SBarry Smith struct DMCompositeLink *next; 115247c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 115347c6ae99SBarry Smith 115447c6ae99SBarry Smith PetscFunctionBegin; 115547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 115647c6ae99SBarry Smith next = com->next; 115747c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 115847c6ae99SBarry Smith va_start(Argp,dm); 115947c6ae99SBarry Smith while (next) { 116047c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 116147c6ae99SBarry Smith PetscInt *n; 116247c6ae99SBarry Smith n = va_arg(Argp, PetscInt*); 116347c6ae99SBarry Smith ierr = DMCompositeGetEntries_Array(dm,next,n);CHKERRQ(ierr); 116447c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 116547c6ae99SBarry Smith DM *dmn; 116647c6ae99SBarry Smith dmn = va_arg(Argp, DM*); 116747c6ae99SBarry Smith ierr = DMCompositeGetEntries_DM(dm,next,dmn);CHKERRQ(ierr); 116847c6ae99SBarry Smith } else { 116947c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 117047c6ae99SBarry Smith } 117147c6ae99SBarry Smith next = next->next; 117247c6ae99SBarry Smith } 117347c6ae99SBarry Smith va_end(Argp); 117447c6ae99SBarry Smith PetscFunctionReturn(0); 117547c6ae99SBarry Smith } 117647c6ae99SBarry Smith 117747c6ae99SBarry Smith #undef __FUNCT__ 11780c010503SBarry Smith #define __FUNCT__ "DMRefine_Composite" 11797087cfbeSBarry Smith PetscErrorCode DMRefine_Composite(DM dmi,MPI_Comm comm,DM *fine) 118047c6ae99SBarry Smith { 118147c6ae99SBarry Smith PetscErrorCode ierr; 118247c6ae99SBarry Smith struct DMCompositeLink *next; 118347c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dmi->data; 118447c6ae99SBarry Smith DM dm; 118547c6ae99SBarry Smith 118647c6ae99SBarry Smith PetscFunctionBegin; 118747c6ae99SBarry Smith PetscValidHeaderSpecific(dmi,DM_CLASSID,1); 118847c6ae99SBarry Smith next = com->next; 118947c6ae99SBarry Smith ierr = DMCompositeCreate(comm,fine);CHKERRQ(ierr); 119047c6ae99SBarry Smith 119147c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 119247c6ae99SBarry Smith while (next) { 119347c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 119406ebdd98SJed Brown ierr = DMCompositeAddArray(*fine,next->rank,next->nlocal);CHKERRQ(ierr); 119547c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 119647c6ae99SBarry Smith ierr = DMRefine(next->dm,comm,&dm);CHKERRQ(ierr); 119747c6ae99SBarry Smith ierr = DMCompositeAddDM(*fine,dm);CHKERRQ(ierr); 119847c6ae99SBarry Smith ierr = PetscObjectDereference((PetscObject)dm);CHKERRQ(ierr); 119947c6ae99SBarry Smith } else { 120047c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 120147c6ae99SBarry Smith } 120247c6ae99SBarry Smith next = next->next; 120347c6ae99SBarry Smith } 120447c6ae99SBarry Smith PetscFunctionReturn(0); 120547c6ae99SBarry Smith } 120647c6ae99SBarry Smith 1207c6db04a5SJed Brown #include <petscmat.h> 120847c6ae99SBarry Smith 120947c6ae99SBarry Smith struct MatPackLink { 121047c6ae99SBarry Smith Mat A; 121147c6ae99SBarry Smith struct MatPackLink *next; 121247c6ae99SBarry Smith }; 121347c6ae99SBarry Smith 121447c6ae99SBarry Smith struct MatPack { 121547c6ae99SBarry Smith DM right,left; 121647c6ae99SBarry Smith struct MatPackLink *next; 121747c6ae99SBarry Smith }; 121847c6ae99SBarry Smith 121947c6ae99SBarry Smith #undef __FUNCT__ 122047c6ae99SBarry Smith #define __FUNCT__ "MatMultBoth_Shell_Pack" 122147c6ae99SBarry Smith PetscErrorCode MatMultBoth_Shell_Pack(Mat A,Vec x,Vec y,PetscBool add) 122247c6ae99SBarry Smith { 122347c6ae99SBarry Smith struct MatPack *mpack; 122447c6ae99SBarry Smith struct DMCompositeLink *xnext,*ynext; 122547c6ae99SBarry Smith struct MatPackLink *anext; 122647c6ae99SBarry Smith PetscScalar *xarray,*yarray; 122747c6ae99SBarry Smith PetscErrorCode ierr; 122847c6ae99SBarry Smith PetscInt i; 122947c6ae99SBarry Smith Vec xglobal,yglobal; 123047c6ae99SBarry Smith PetscMPIInt rank; 123147c6ae99SBarry Smith DM_Composite *comright; 123247c6ae99SBarry Smith DM_Composite *comleft; 123347c6ae99SBarry Smith 123447c6ae99SBarry Smith PetscFunctionBegin; 123547c6ae99SBarry Smith ierr = MatShellGetContext(A,(void**)&mpack);CHKERRQ(ierr); 123647c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)mpack->right)->comm,&rank);CHKERRQ(ierr); 123747c6ae99SBarry Smith comright = (DM_Composite*)mpack->right->data; 123847c6ae99SBarry Smith comleft = (DM_Composite*)mpack->left->data; 123947c6ae99SBarry Smith xnext = comright->next; 124047c6ae99SBarry Smith ynext = comleft->next; 124147c6ae99SBarry Smith anext = mpack->next; 124247c6ae99SBarry Smith 124347c6ae99SBarry Smith while (xnext) { 124447c6ae99SBarry Smith if (xnext->type == DMCOMPOSITE_ARRAY) { 124547c6ae99SBarry Smith if (rank == xnext->rank) { 124647c6ae99SBarry Smith ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 124747c6ae99SBarry Smith ierr = VecGetArray(y,&yarray);CHKERRQ(ierr); 124847c6ae99SBarry Smith if (add) { 124947c6ae99SBarry Smith for (i=0; i<xnext->n; i++) { 125047c6ae99SBarry Smith yarray[ynext->rstart+i] += xarray[xnext->rstart+i]; 125147c6ae99SBarry Smith } 125247c6ae99SBarry Smith } else { 125347c6ae99SBarry Smith ierr = PetscMemcpy(yarray+ynext->rstart,xarray+xnext->rstart,xnext->n*sizeof(PetscScalar));CHKERRQ(ierr); 125447c6ae99SBarry Smith } 125547c6ae99SBarry Smith ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 125647c6ae99SBarry Smith ierr = VecRestoreArray(y,&yarray);CHKERRQ(ierr); 125747c6ae99SBarry Smith } 125847c6ae99SBarry Smith } else if (xnext->type == DMCOMPOSITE_DM) { 125947c6ae99SBarry Smith ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 126047c6ae99SBarry Smith ierr = VecGetArray(y,&yarray);CHKERRQ(ierr); 126147c6ae99SBarry Smith ierr = DMGetGlobalVector(xnext->dm,&xglobal);CHKERRQ(ierr); 126247c6ae99SBarry Smith ierr = DMGetGlobalVector(ynext->dm,&yglobal);CHKERRQ(ierr); 126347c6ae99SBarry Smith ierr = VecPlaceArray(xglobal,xarray+xnext->rstart);CHKERRQ(ierr); 126447c6ae99SBarry Smith ierr = VecPlaceArray(yglobal,yarray+ynext->rstart);CHKERRQ(ierr); 126547c6ae99SBarry Smith if (add) { 126647c6ae99SBarry Smith ierr = MatMultAdd(anext->A,xglobal,yglobal,yglobal);CHKERRQ(ierr); 126747c6ae99SBarry Smith } else { 126847c6ae99SBarry Smith ierr = MatMult(anext->A,xglobal,yglobal);CHKERRQ(ierr); 126947c6ae99SBarry Smith } 127047c6ae99SBarry Smith ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 127147c6ae99SBarry Smith ierr = VecRestoreArray(y,&yarray);CHKERRQ(ierr); 127247c6ae99SBarry Smith ierr = VecResetArray(xglobal);CHKERRQ(ierr); 127347c6ae99SBarry Smith ierr = VecResetArray(yglobal);CHKERRQ(ierr); 127447c6ae99SBarry Smith ierr = DMRestoreGlobalVector(xnext->dm,&xglobal);CHKERRQ(ierr); 127547c6ae99SBarry Smith ierr = DMRestoreGlobalVector(ynext->dm,&yglobal);CHKERRQ(ierr); 127647c6ae99SBarry Smith anext = anext->next; 127747c6ae99SBarry Smith } else { 127847c6ae99SBarry Smith SETERRQ(((PetscObject)A)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 127947c6ae99SBarry Smith } 128047c6ae99SBarry Smith xnext = xnext->next; 128147c6ae99SBarry Smith ynext = ynext->next; 128247c6ae99SBarry Smith } 128347c6ae99SBarry Smith PetscFunctionReturn(0); 128447c6ae99SBarry Smith } 128547c6ae99SBarry Smith 128647c6ae99SBarry Smith #undef __FUNCT__ 128747c6ae99SBarry Smith #define __FUNCT__ "MatMultAdd_Shell_Pack" 128847c6ae99SBarry Smith PetscErrorCode MatMultAdd_Shell_Pack(Mat A,Vec x,Vec y,Vec z) 128947c6ae99SBarry Smith { 129047c6ae99SBarry Smith PetscErrorCode ierr; 129147c6ae99SBarry Smith PetscFunctionBegin; 129247c6ae99SBarry Smith if (z != y) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_SUP,"Handles y == z only"); 129347c6ae99SBarry Smith ierr = MatMultBoth_Shell_Pack(A,x,y,PETSC_TRUE);CHKERRQ(ierr); 129447c6ae99SBarry Smith PetscFunctionReturn(0); 129547c6ae99SBarry Smith } 129647c6ae99SBarry Smith 129747c6ae99SBarry Smith #undef __FUNCT__ 129847c6ae99SBarry Smith #define __FUNCT__ "MatMult_Shell_Pack" 129947c6ae99SBarry Smith PetscErrorCode MatMult_Shell_Pack(Mat A,Vec x,Vec y) 130047c6ae99SBarry Smith { 130147c6ae99SBarry Smith PetscErrorCode ierr; 130247c6ae99SBarry Smith PetscFunctionBegin; 130347c6ae99SBarry Smith ierr = MatMultBoth_Shell_Pack(A,x,y,PETSC_FALSE);CHKERRQ(ierr); 130447c6ae99SBarry Smith PetscFunctionReturn(0); 130547c6ae99SBarry Smith } 130647c6ae99SBarry Smith 130747c6ae99SBarry Smith #undef __FUNCT__ 130847c6ae99SBarry Smith #define __FUNCT__ "MatMultTranspose_Shell_Pack" 130947c6ae99SBarry Smith PetscErrorCode MatMultTranspose_Shell_Pack(Mat A,Vec x,Vec y) 131047c6ae99SBarry Smith { 131147c6ae99SBarry Smith struct MatPack *mpack; 131247c6ae99SBarry Smith struct DMCompositeLink *xnext,*ynext; 131347c6ae99SBarry Smith struct MatPackLink *anext; 131447c6ae99SBarry Smith PetscScalar *xarray,*yarray; 131547c6ae99SBarry Smith PetscErrorCode ierr; 131647c6ae99SBarry Smith Vec xglobal,yglobal; 131747c6ae99SBarry Smith PetscMPIInt rank; 131847c6ae99SBarry Smith DM_Composite *comright; 131947c6ae99SBarry Smith DM_Composite *comleft; 132047c6ae99SBarry Smith 132147c6ae99SBarry Smith PetscFunctionBegin; 132247c6ae99SBarry Smith ierr = MatShellGetContext(A,(void**)&mpack);CHKERRQ(ierr); 132347c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)mpack->right)->comm,&rank);CHKERRQ(ierr); 132447c6ae99SBarry Smith comright = (DM_Composite*)mpack->right->data; 132547c6ae99SBarry Smith comleft = (DM_Composite*)mpack->left->data; 132647c6ae99SBarry Smith ynext = comright->next; 132747c6ae99SBarry Smith xnext = comleft->next; 132847c6ae99SBarry Smith anext = mpack->next; 132947c6ae99SBarry Smith 133047c6ae99SBarry Smith while (xnext) { 133147c6ae99SBarry Smith if (xnext->type == DMCOMPOSITE_ARRAY) { 133247c6ae99SBarry Smith if (rank == ynext->rank) { 133347c6ae99SBarry Smith ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 133447c6ae99SBarry Smith ierr = VecGetArray(y,&yarray);CHKERRQ(ierr); 133547c6ae99SBarry Smith ierr = PetscMemcpy(yarray+ynext->rstart,xarray+xnext->rstart,xnext->n*sizeof(PetscScalar));CHKERRQ(ierr); 133647c6ae99SBarry Smith ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 133747c6ae99SBarry Smith ierr = VecRestoreArray(y,&yarray);CHKERRQ(ierr); 133847c6ae99SBarry Smith } 133947c6ae99SBarry Smith } else if (xnext->type == DMCOMPOSITE_DM) { 134047c6ae99SBarry Smith ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 134147c6ae99SBarry Smith ierr = VecGetArray(y,&yarray);CHKERRQ(ierr); 134247c6ae99SBarry Smith ierr = DMGetGlobalVector(xnext->dm,&xglobal);CHKERRQ(ierr); 134347c6ae99SBarry Smith ierr = DMGetGlobalVector(ynext->dm,&yglobal);CHKERRQ(ierr); 134447c6ae99SBarry Smith ierr = VecPlaceArray(xglobal,xarray+xnext->rstart);CHKERRQ(ierr); 134547c6ae99SBarry Smith ierr = VecPlaceArray(yglobal,yarray+ynext->rstart);CHKERRQ(ierr); 134647c6ae99SBarry Smith ierr = MatMultTranspose(anext->A,xglobal,yglobal);CHKERRQ(ierr); 134747c6ae99SBarry Smith ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 134847c6ae99SBarry Smith ierr = VecRestoreArray(y,&yarray);CHKERRQ(ierr); 134947c6ae99SBarry Smith ierr = VecResetArray(xglobal);CHKERRQ(ierr); 135047c6ae99SBarry Smith ierr = VecResetArray(yglobal);CHKERRQ(ierr); 135147c6ae99SBarry Smith ierr = DMRestoreGlobalVector(xnext->dm,&xglobal);CHKERRQ(ierr); 135247c6ae99SBarry Smith ierr = DMRestoreGlobalVector(ynext->dm,&yglobal);CHKERRQ(ierr); 135347c6ae99SBarry Smith anext = anext->next; 135447c6ae99SBarry Smith } else { 135547c6ae99SBarry Smith SETERRQ(((PetscObject)A)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 135647c6ae99SBarry Smith } 135747c6ae99SBarry Smith xnext = xnext->next; 135847c6ae99SBarry Smith ynext = ynext->next; 135947c6ae99SBarry Smith } 136047c6ae99SBarry Smith PetscFunctionReturn(0); 136147c6ae99SBarry Smith } 136247c6ae99SBarry Smith 136347c6ae99SBarry Smith #undef __FUNCT__ 136447c6ae99SBarry Smith #define __FUNCT__ "MatDestroy_Shell_Pack" 136547c6ae99SBarry Smith PetscErrorCode MatDestroy_Shell_Pack(Mat A) 136647c6ae99SBarry Smith { 136747c6ae99SBarry Smith struct MatPack *mpack; 136847c6ae99SBarry Smith struct MatPackLink *anext,*oldanext; 136947c6ae99SBarry Smith PetscErrorCode ierr; 137047c6ae99SBarry Smith 137147c6ae99SBarry Smith PetscFunctionBegin; 137247c6ae99SBarry Smith ierr = MatShellGetContext(A,(void**)&mpack);CHKERRQ(ierr); 137347c6ae99SBarry Smith anext = mpack->next; 137447c6ae99SBarry Smith 137547c6ae99SBarry Smith while (anext) { 1376fcfd50ebSBarry Smith ierr = MatDestroy(&anext->A);CHKERRQ(ierr); 137747c6ae99SBarry Smith oldanext = anext; 137847c6ae99SBarry Smith anext = anext->next; 137947c6ae99SBarry Smith ierr = PetscFree(oldanext);CHKERRQ(ierr); 138047c6ae99SBarry Smith } 138147c6ae99SBarry Smith ierr = PetscFree(mpack);CHKERRQ(ierr); 138247c6ae99SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 138347c6ae99SBarry Smith PetscFunctionReturn(0); 138447c6ae99SBarry Smith } 138547c6ae99SBarry Smith 138647c6ae99SBarry Smith #undef __FUNCT__ 13870c010503SBarry Smith #define __FUNCT__ "DMGetInterpolation_Composite" 13887087cfbeSBarry Smith PetscErrorCode DMGetInterpolation_Composite(DM coarse,DM fine,Mat *A,Vec *v) 138947c6ae99SBarry Smith { 139047c6ae99SBarry Smith PetscErrorCode ierr; 139147c6ae99SBarry Smith PetscInt m,n,M,N; 139247c6ae99SBarry Smith struct DMCompositeLink *nextc; 139347c6ae99SBarry Smith struct DMCompositeLink *nextf; 139447c6ae99SBarry Smith struct MatPackLink *nextmat,*pnextmat = 0; 139547c6ae99SBarry Smith struct MatPack *mpack; 139647c6ae99SBarry Smith Vec gcoarse,gfine; 139747c6ae99SBarry Smith DM_Composite *comcoarse = (DM_Composite*)coarse->data; 139847c6ae99SBarry Smith DM_Composite *comfine = (DM_Composite*)fine->data; 139947c6ae99SBarry Smith 140047c6ae99SBarry Smith PetscFunctionBegin; 140147c6ae99SBarry Smith PetscValidHeaderSpecific(coarse,DM_CLASSID,1); 140247c6ae99SBarry Smith PetscValidHeaderSpecific(fine,DM_CLASSID,2); 140347c6ae99SBarry Smith nextc = comcoarse->next; 140447c6ae99SBarry Smith nextf = comfine->next; 140547c6ae99SBarry Smith /* use global vectors only for determining matrix layout */ 14060c010503SBarry Smith ierr = DMCreateGlobalVector(coarse,&gcoarse);CHKERRQ(ierr); 14070c010503SBarry Smith ierr = DMCreateGlobalVector(fine,&gfine);CHKERRQ(ierr); 140847c6ae99SBarry Smith ierr = VecGetLocalSize(gcoarse,&n);CHKERRQ(ierr); 140947c6ae99SBarry Smith ierr = VecGetLocalSize(gfine,&m);CHKERRQ(ierr); 141047c6ae99SBarry Smith ierr = VecGetSize(gcoarse,&N);CHKERRQ(ierr); 141147c6ae99SBarry Smith ierr = VecGetSize(gfine,&M);CHKERRQ(ierr); 1412fcfd50ebSBarry Smith ierr = VecDestroy(&gcoarse);CHKERRQ(ierr); 1413fcfd50ebSBarry Smith ierr = VecDestroy(&gfine);CHKERRQ(ierr); 141447c6ae99SBarry Smith 141547c6ae99SBarry Smith ierr = PetscNew(struct MatPack,&mpack);CHKERRQ(ierr); 141647c6ae99SBarry Smith mpack->right = coarse; 141747c6ae99SBarry Smith mpack->left = fine; 141847c6ae99SBarry Smith ierr = MatCreate(((PetscObject)fine)->comm,A);CHKERRQ(ierr); 141947c6ae99SBarry Smith ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 142047c6ae99SBarry Smith ierr = MatSetType(*A,MATSHELL);CHKERRQ(ierr); 142147c6ae99SBarry Smith ierr = MatShellSetContext(*A,mpack);CHKERRQ(ierr); 142247c6ae99SBarry Smith ierr = MatShellSetOperation(*A,MATOP_MULT,(void(*)(void))MatMult_Shell_Pack);CHKERRQ(ierr); 142347c6ae99SBarry Smith ierr = MatShellSetOperation(*A,MATOP_MULT_TRANSPOSE,(void(*)(void))MatMultTranspose_Shell_Pack);CHKERRQ(ierr); 142447c6ae99SBarry Smith ierr = MatShellSetOperation(*A,MATOP_MULT_ADD,(void(*)(void))MatMultAdd_Shell_Pack);CHKERRQ(ierr); 142547c6ae99SBarry Smith ierr = MatShellSetOperation(*A,MATOP_DESTROY,(void(*)(void))MatDestroy_Shell_Pack);CHKERRQ(ierr); 142647c6ae99SBarry Smith 142747c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 142847c6ae99SBarry Smith while (nextc) { 142947c6ae99SBarry Smith if (nextc->type != nextf->type) SETERRQ(((PetscObject)fine)->comm,PETSC_ERR_ARG_INCOMP,"Two DM have different layout"); 143047c6ae99SBarry Smith 143147c6ae99SBarry Smith if (nextc->type == DMCOMPOSITE_ARRAY) { 143247c6ae99SBarry Smith ; 143347c6ae99SBarry Smith } else if (nextc->type == DMCOMPOSITE_DM) { 143447c6ae99SBarry Smith ierr = PetscNew(struct MatPackLink,&nextmat);CHKERRQ(ierr); 143547c6ae99SBarry Smith nextmat->next = 0; 143647c6ae99SBarry Smith if (pnextmat) { 143747c6ae99SBarry Smith pnextmat->next = nextmat; 143847c6ae99SBarry Smith pnextmat = nextmat; 143947c6ae99SBarry Smith } else { 144047c6ae99SBarry Smith pnextmat = nextmat; 144147c6ae99SBarry Smith mpack->next = nextmat; 144247c6ae99SBarry Smith } 144347c6ae99SBarry Smith ierr = DMGetInterpolation(nextc->dm,nextf->dm,&nextmat->A,PETSC_NULL);CHKERRQ(ierr); 144447c6ae99SBarry Smith } else { 144547c6ae99SBarry Smith SETERRQ(((PetscObject)fine)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 144647c6ae99SBarry Smith } 144747c6ae99SBarry Smith nextc = nextc->next; 144847c6ae99SBarry Smith nextf = nextf->next; 144947c6ae99SBarry Smith } 145047c6ae99SBarry Smith PetscFunctionReturn(0); 145147c6ae99SBarry Smith } 145247c6ae99SBarry Smith 145347c6ae99SBarry Smith #undef __FUNCT__ 14541411c6eeSJed Brown #define __FUNCT__ "DMCreateLocalToGlobalMapping_Composite" 14551411c6eeSJed Brown static PetscErrorCode DMCreateLocalToGlobalMapping_Composite(DM dm) 14561411c6eeSJed Brown { 14571411c6eeSJed Brown DM_Composite *com = (DM_Composite*)dm->data; 14581411c6eeSJed Brown ISLocalToGlobalMapping *ltogs; 1459*f7efa3c7SJed Brown PetscInt i; 14601411c6eeSJed Brown PetscErrorCode ierr; 14611411c6eeSJed Brown 14621411c6eeSJed Brown PetscFunctionBegin; 14631411c6eeSJed Brown /* Set the ISLocalToGlobalMapping on the new matrix */ 14641411c6eeSJed Brown ierr = DMCompositeGetISLocalToGlobalMappings(dm,<ogs);CHKERRQ(ierr); 1465*f7efa3c7SJed Brown ierr = ISLocalToGlobalMappingConcatenate(((PetscObject)dm)->comm,com->nDM+com->nredundant,ltogs,&dm->ltogmap);CHKERRQ(ierr); 1466fcfd50ebSBarry Smith for (i=0; i<com->nDM+com->nredundant; i++) {ierr = ISLocalToGlobalMappingDestroy(<ogs[i]);CHKERRQ(ierr);} 14671411c6eeSJed Brown ierr = PetscFree(ltogs);CHKERRQ(ierr); 14681411c6eeSJed Brown PetscFunctionReturn(0); 14691411c6eeSJed Brown } 14701411c6eeSJed Brown 14711411c6eeSJed Brown 14721411c6eeSJed Brown #undef __FUNCT__ 14730c010503SBarry Smith #define __FUNCT__ "DMGetColoring_Composite" 14747087cfbeSBarry Smith PetscErrorCode DMGetColoring_Composite(DM dm,ISColoringType ctype,const MatType mtype,ISColoring *coloring) 147547c6ae99SBarry Smith { 147647c6ae99SBarry Smith PetscErrorCode ierr; 147747c6ae99SBarry Smith PetscInt n,i,cnt; 147847c6ae99SBarry Smith ISColoringValue *colors; 147947c6ae99SBarry Smith PetscBool dense = PETSC_FALSE; 148047c6ae99SBarry Smith ISColoringValue maxcol = 0; 148147c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 148247c6ae99SBarry Smith 148347c6ae99SBarry Smith PetscFunctionBegin; 148447c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 148547c6ae99SBarry Smith if (ctype == IS_COLORING_GHOSTED) { 148647c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Currently you must use -dmmg_iscoloring_type global" ); 148747c6ae99SBarry Smith } else if (ctype == IS_COLORING_GLOBAL) { 148847c6ae99SBarry Smith n = com->n; 148947c6ae99SBarry Smith } else SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Unknown ISColoringType"); 149047c6ae99SBarry Smith ierr = PetscMalloc(n*sizeof(ISColoringValue),&colors);CHKERRQ(ierr); /* freed in ISColoringDestroy() */ 149147c6ae99SBarry Smith 1492671f6225SBarry Smith ierr = PetscOptionsGetBool(PETSC_NULL,"-dmcomposite_dense_jacobian",&dense,PETSC_NULL);CHKERRQ(ierr); 149347c6ae99SBarry Smith if (dense) { 149447c6ae99SBarry Smith for (i=0; i<n; i++) { 149547c6ae99SBarry Smith colors[i] = (ISColoringValue)(com->rstart + i); 149647c6ae99SBarry Smith } 149747c6ae99SBarry Smith maxcol = com->N; 149847c6ae99SBarry Smith } else { 149947c6ae99SBarry Smith struct DMCompositeLink *next = com->next; 150047c6ae99SBarry Smith PetscMPIInt rank; 150147c6ae99SBarry Smith 150247c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 150347c6ae99SBarry Smith cnt = 0; 150447c6ae99SBarry Smith while (next) { 150547c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 150647c6ae99SBarry Smith if (rank == next->rank) { /* each column gets is own color */ 150747c6ae99SBarry Smith for (i=com->rstart+next->rstart; i<com->rstart+next->rstart+next->n; i++) { 150847c6ae99SBarry Smith colors[cnt++] = maxcol++; 150947c6ae99SBarry Smith } 151047c6ae99SBarry Smith } 151147c6ae99SBarry Smith ierr = MPI_Bcast(&maxcol,1,MPIU_COLORING_VALUE,next->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 151247c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 151347c6ae99SBarry Smith ISColoring lcoloring; 151447c6ae99SBarry Smith 151547c6ae99SBarry Smith ierr = DMGetColoring(next->dm,IS_COLORING_GLOBAL,mtype,&lcoloring);CHKERRQ(ierr); 151647c6ae99SBarry Smith for (i=0; i<lcoloring->N; i++) { 151747c6ae99SBarry Smith colors[cnt++] = maxcol + lcoloring->colors[i]; 151847c6ae99SBarry Smith } 151947c6ae99SBarry Smith maxcol += lcoloring->n; 1520fcfd50ebSBarry Smith ierr = ISColoringDestroy(&lcoloring);CHKERRQ(ierr); 152196e147daSBarry Smith } else SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 152247c6ae99SBarry Smith next = next->next; 152347c6ae99SBarry Smith } 152447c6ae99SBarry Smith } 152547c6ae99SBarry Smith ierr = ISColoringCreate(((PetscObject)dm)->comm,maxcol,n,colors,coloring);CHKERRQ(ierr); 152647c6ae99SBarry Smith PetscFunctionReturn(0); 152747c6ae99SBarry Smith } 152847c6ae99SBarry Smith 152947c6ae99SBarry Smith #undef __FUNCT__ 15300c010503SBarry Smith #define __FUNCT__ "DMGlobalToLocalBegin_Composite" 15317087cfbeSBarry Smith PetscErrorCode DMGlobalToLocalBegin_Composite(DM dm,Vec gvec,InsertMode mode,Vec lvec) 153247c6ae99SBarry Smith { 153347c6ae99SBarry Smith PetscErrorCode ierr; 153447c6ae99SBarry Smith struct DMCompositeLink *next; 153547c6ae99SBarry Smith PetscInt cnt = 3; 153647c6ae99SBarry Smith PetscMPIInt rank; 153747c6ae99SBarry Smith PetscScalar *garray,*larray; 153847c6ae99SBarry Smith DM_Composite *com = (DM_Composite*)dm->data; 153947c6ae99SBarry Smith 154047c6ae99SBarry Smith PetscFunctionBegin; 154147c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 154247c6ae99SBarry Smith PetscValidHeaderSpecific(gvec,VEC_CLASSID,2); 154347c6ae99SBarry Smith next = com->next; 154447c6ae99SBarry Smith if (!com->setup) { 1545d7bf68aeSBarry Smith ierr = DMSetUp(dm);CHKERRQ(ierr); 154647c6ae99SBarry Smith } 154747c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)dm)->comm,&rank);CHKERRQ(ierr); 154847c6ae99SBarry Smith ierr = VecGetArray(gvec,&garray);CHKERRQ(ierr); 154947c6ae99SBarry Smith ierr = VecGetArray(lvec,&larray);CHKERRQ(ierr); 155047c6ae99SBarry Smith 155147c6ae99SBarry Smith /* loop over packed objects, handling one at at time */ 155247c6ae99SBarry Smith while (next) { 155347c6ae99SBarry Smith if (next->type == DMCOMPOSITE_ARRAY) { 155447c6ae99SBarry Smith if (rank == next->rank) { 155547c6ae99SBarry Smith ierr = PetscMemcpy(larray,garray,next->n*sizeof(PetscScalar));CHKERRQ(ierr); 155647c6ae99SBarry Smith garray += next->n; 155747c6ae99SBarry Smith } 155847c6ae99SBarry Smith /* does not handle ADD_VALUES */ 155906ebdd98SJed Brown ierr = MPI_Bcast(larray,next->nlocal,MPIU_SCALAR,next->rank,((PetscObject)dm)->comm);CHKERRQ(ierr); 156047c6ae99SBarry Smith } else if (next->type == DMCOMPOSITE_DM) { 156147c6ae99SBarry Smith Vec local,global; 156247c6ae99SBarry Smith PetscInt N; 156347c6ae99SBarry Smith 156447c6ae99SBarry Smith ierr = DMGetGlobalVector(next->dm,&global);CHKERRQ(ierr); 156547c6ae99SBarry Smith ierr = VecGetLocalSize(global,&N);CHKERRQ(ierr); 156647c6ae99SBarry Smith ierr = VecPlaceArray(global,garray);CHKERRQ(ierr); 156747c6ae99SBarry Smith ierr = DMGetLocalVector(next->dm,&local);CHKERRQ(ierr); 156847c6ae99SBarry Smith ierr = VecPlaceArray(local,larray);CHKERRQ(ierr); 156947c6ae99SBarry Smith ierr = DMGlobalToLocalBegin(next->dm,global,mode,local);CHKERRQ(ierr); 157047c6ae99SBarry Smith ierr = DMGlobalToLocalEnd(next->dm,global,mode,local);CHKERRQ(ierr); 157147c6ae99SBarry Smith ierr = VecResetArray(global);CHKERRQ(ierr); 157247c6ae99SBarry Smith ierr = VecResetArray(local);CHKERRQ(ierr); 157347c6ae99SBarry Smith ierr = DMRestoreGlobalVector(next->dm,&global);CHKERRQ(ierr); 157447c6ae99SBarry Smith ierr = DMRestoreGlobalVector(next->dm,&local);CHKERRQ(ierr); 157547c6ae99SBarry Smith } else { 157647c6ae99SBarry Smith SETERRQ(((PetscObject)dm)->comm,PETSC_ERR_SUP,"Cannot handle that object type yet"); 157747c6ae99SBarry Smith } 157847c6ae99SBarry Smith cnt++; 157906ebdd98SJed Brown larray += next->nlocal; 158047c6ae99SBarry Smith next = next->next; 158147c6ae99SBarry Smith } 158247c6ae99SBarry Smith 158347c6ae99SBarry Smith ierr = VecRestoreArray(gvec,PETSC_NULL);CHKERRQ(ierr); 158447c6ae99SBarry Smith ierr = VecRestoreArray(lvec,PETSC_NULL);CHKERRQ(ierr); 158547c6ae99SBarry Smith PetscFunctionReturn(0); 158647c6ae99SBarry Smith } 158747c6ae99SBarry Smith 158847c6ae99SBarry Smith #undef __FUNCT__ 15890c010503SBarry Smith #define __FUNCT__ "DMGlobalToLocalEnd_Composite" 15907087cfbeSBarry Smith PetscErrorCode DMGlobalToLocalEnd_Composite(DM dm,Vec gvec,InsertMode mode,Vec lvec) 15910c010503SBarry Smith { 15920c010503SBarry Smith PetscFunctionBegin; 15930c010503SBarry Smith PetscFunctionReturn(0); 15940c010503SBarry Smith } 159547c6ae99SBarry Smith 1596a4121054SBarry Smith EXTERN_C_BEGIN 1597a4121054SBarry Smith #undef __FUNCT__ 1598a4121054SBarry Smith #define __FUNCT__ "DMCreate_Composite" 15997087cfbeSBarry Smith PetscErrorCode DMCreate_Composite(DM p) 1600a4121054SBarry Smith { 1601a4121054SBarry Smith PetscErrorCode ierr; 1602a4121054SBarry Smith DM_Composite *com; 1603a4121054SBarry Smith 1604a4121054SBarry Smith PetscFunctionBegin; 1605a4121054SBarry Smith ierr = PetscNewLog(p,DM_Composite,&com);CHKERRQ(ierr); 1606a4121054SBarry Smith p->data = com; 1607a4121054SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)p,"DMComposite");CHKERRQ(ierr); 1608a4121054SBarry Smith com->n = 0; 1609a4121054SBarry Smith com->next = PETSC_NULL; 1610a4121054SBarry Smith com->nredundant = 0; 1611a4121054SBarry Smith com->nDM = 0; 1612a4121054SBarry Smith 1613a4121054SBarry Smith p->ops->createglobalvector = DMCreateGlobalVector_Composite; 1614a4121054SBarry Smith p->ops->createlocalvector = DMCreateLocalVector_Composite; 16151411c6eeSJed Brown p->ops->createlocaltoglobalmapping = DMCreateLocalToGlobalMapping_Composite; 16161411c6eeSJed Brown p->ops->createlocaltoglobalmappingblock = 0; 1617a4121054SBarry Smith p->ops->refine = DMRefine_Composite; 1618a4121054SBarry Smith p->ops->getinterpolation = DMGetInterpolation_Composite; 1619a4121054SBarry Smith p->ops->getmatrix = DMGetMatrix_Composite; 1620a4121054SBarry Smith p->ops->getcoloring = DMGetColoring_Composite; 1621a4121054SBarry Smith p->ops->globaltolocalbegin = DMGlobalToLocalBegin_Composite; 1622a4121054SBarry Smith p->ops->globaltolocalend = DMGlobalToLocalEnd_Composite; 1623a4121054SBarry Smith p->ops->destroy = DMDestroy_Composite; 1624a4121054SBarry Smith p->ops->view = DMView_Composite; 1625a4121054SBarry Smith p->ops->setup = DMSetUp_Composite; 1626a4121054SBarry Smith PetscFunctionReturn(0); 1627a4121054SBarry Smith } 1628a4121054SBarry Smith EXTERN_C_END 1629a4121054SBarry Smith 16300c010503SBarry Smith #undef __FUNCT__ 16310c010503SBarry Smith #define __FUNCT__ "DMCompositeCreate" 16320c010503SBarry Smith /*@C 16330c010503SBarry Smith DMCompositeCreate - Creates a vector packer, used to generate "composite" 16340c010503SBarry Smith vectors made up of several subvectors. 16350c010503SBarry Smith 16360c010503SBarry Smith Collective on MPI_Comm 163747c6ae99SBarry Smith 163847c6ae99SBarry Smith Input Parameter: 16390c010503SBarry Smith . comm - the processors that will share the global vector 16400c010503SBarry Smith 16410c010503SBarry Smith Output Parameters: 16420c010503SBarry Smith . packer - the packer object 164347c6ae99SBarry Smith 164447c6ae99SBarry Smith Level: advanced 164547c6ae99SBarry Smith 16460c010503SBarry Smith .seealso DMDestroy(), DMCompositeAddArray(), DMCompositeAddDM(), DMCompositeScatter(), 16476eb61c8cSJed Brown DMCompositeGather(), DMCreateGlobalVector(), DMCompositeGetISLocalToGlobalMappings(), DMCompositeGetAccess() 164847c6ae99SBarry Smith DMCompositeGetLocalVectors(), DMCompositeRestoreLocalVectors(), DMCompositeGetEntries() 164947c6ae99SBarry Smith 165047c6ae99SBarry Smith @*/ 16517087cfbeSBarry Smith PetscErrorCode DMCompositeCreate(MPI_Comm comm,DM *packer) 165247c6ae99SBarry Smith { 16530c010503SBarry Smith PetscErrorCode ierr; 16540c010503SBarry Smith 165547c6ae99SBarry Smith PetscFunctionBegin; 16560c010503SBarry Smith PetscValidPointer(packer,2); 1657a4121054SBarry Smith ierr = DMCreate(comm,packer);CHKERRQ(ierr); 1658a4121054SBarry Smith ierr = DMSetType(*packer,DMCOMPOSITE);CHKERRQ(ierr); 165947c6ae99SBarry Smith PetscFunctionReturn(0); 166047c6ae99SBarry Smith } 1661