xref: /petsc/src/dm/impls/composite/pack.c (revision 1b2093e4d1d741eb25c3aec22d4fd75e2597f056)
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 
19*1b2093e4SBarry Smith     Notes: See DMSetApplicationContext() and DMGetApplicationContext() 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,&ltog);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;
1459f7efa3c7SJed 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,&ltogs);CHKERRQ(ierr);
1465f7efa3c7SJed 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(&ltogs[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