xref: /petsc/src/vec/is/utils/isltog.c (revision 5fd668637986a8d8518383a9159eebc368e1d5b4)
12362add9SBarry Smith 
2b45d2f2cSJed Brown #include <petsc-private/isimpl.h>    /*I "petscis.h"  I*/
32362add9SBarry Smith 
47087cfbeSBarry Smith PetscClassId  IS_LTOGM_CLASSID;
58e58c17dSMatthew Knepley 
64a2ae208SSatish Balay #undef __FUNCT__
74a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetSize"
83b9aefa3SBarry Smith /*@C
93b9aefa3SBarry Smith     ISLocalToGlobalMappingGetSize - Gets the local size of a local to global mapping.
103b9aefa3SBarry Smith 
113b9aefa3SBarry Smith     Not Collective
123b9aefa3SBarry Smith 
133b9aefa3SBarry Smith     Input Parameter:
143b9aefa3SBarry Smith .   ltog - local to global mapping
153b9aefa3SBarry Smith 
163b9aefa3SBarry Smith     Output Parameter:
173b9aefa3SBarry Smith .   n - the number of entries in the local mapping
183b9aefa3SBarry Smith 
193b9aefa3SBarry Smith     Level: advanced
203b9aefa3SBarry Smith 
21273d9f13SBarry Smith     Concepts: mapping^local to global
223b9aefa3SBarry Smith 
233b9aefa3SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
243b9aefa3SBarry Smith @*/
257087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping mapping,PetscInt *n)
263b9aefa3SBarry Smith {
273b9aefa3SBarry Smith   PetscFunctionBegin;
280700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
294482741eSBarry Smith   PetscValidIntPointer(n,2);
303b9aefa3SBarry Smith   *n = mapping->n;
313b9aefa3SBarry Smith   PetscFunctionReturn(0);
323b9aefa3SBarry Smith }
333b9aefa3SBarry Smith 
344a2ae208SSatish Balay #undef __FUNCT__
354a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingView"
365a5d4f66SBarry Smith /*@C
375a5d4f66SBarry Smith     ISLocalToGlobalMappingView - View a local to global mapping
385a5d4f66SBarry Smith 
39b9cd556bSLois Curfman McInnes     Not Collective
40b9cd556bSLois Curfman McInnes 
415a5d4f66SBarry Smith     Input Parameters:
423b9aefa3SBarry Smith +   ltog - local to global mapping
433b9aefa3SBarry Smith -   viewer - viewer
445a5d4f66SBarry Smith 
45a997ad1aSLois Curfman McInnes     Level: advanced
46a997ad1aSLois Curfman McInnes 
47273d9f13SBarry Smith     Concepts: mapping^local to global
485a5d4f66SBarry Smith 
495a5d4f66SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
505a5d4f66SBarry Smith @*/
517087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingView(ISLocalToGlobalMapping mapping,PetscViewer viewer)
525a5d4f66SBarry Smith {
5332dcc486SBarry Smith   PetscInt        i;
5432dcc486SBarry Smith   PetscMPIInt     rank;
55ace3abfcSBarry Smith   PetscBool       iascii;
566849ba73SBarry Smith   PetscErrorCode  ierr;
575a5d4f66SBarry Smith 
585a5d4f66SBarry Smith   PetscFunctionBegin;
590700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
603050cee2SBarry Smith   if (!viewer) {
617adad957SLisandro Dalcin     ierr = PetscViewerASCIIGetStdout(((PetscObject)mapping)->comm,&viewer);CHKERRQ(ierr);
623050cee2SBarry Smith   }
630700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
645a5d4f66SBarry Smith 
657adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mapping)->comm,&rank);CHKERRQ(ierr);
66251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
6732077d6dSBarry Smith   if (iascii) {
687b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
695a5d4f66SBarry Smith     for (i=0; i<mapping->n; i++) {
70b0a32e0cSBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %d %d\n",rank,i,mapping->indices[i]);CHKERRQ(ierr);
716831982aSBarry Smith     }
72b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
737b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
747b23a99aSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name);
755a5d4f66SBarry Smith   PetscFunctionReturn(0);
765a5d4f66SBarry Smith }
775a5d4f66SBarry Smith 
784a2ae208SSatish Balay #undef __FUNCT__
794a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS"
801f428162SBarry Smith /*@
812bdab257SBarry Smith     ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n)
822bdab257SBarry Smith     ordering and a global parallel ordering.
832bdab257SBarry Smith 
840f5bd95cSBarry Smith     Not collective
85b9cd556bSLois Curfman McInnes 
86a997ad1aSLois Curfman McInnes     Input Parameter:
878c03b21aSDmitry Karpeev .   is - index set containing the global numbers for each local number
882bdab257SBarry Smith 
89a997ad1aSLois Curfman McInnes     Output Parameter:
902bdab257SBarry Smith .   mapping - new mapping data structure
912bdab257SBarry Smith 
92a997ad1aSLois Curfman McInnes     Level: advanced
93a997ad1aSLois Curfman McInnes 
94273d9f13SBarry Smith     Concepts: mapping^local to global
952bdab257SBarry Smith 
962bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
972bdab257SBarry Smith @*/
987087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping)
992bdab257SBarry Smith {
1006849ba73SBarry Smith   PetscErrorCode ierr;
1015d0c19d7SBarry Smith   PetscInt       n;
1025d0c19d7SBarry Smith   const PetscInt *indices;
1032bdab257SBarry Smith   MPI_Comm       comm;
1043a40ed3dSBarry Smith 
1053a40ed3dSBarry Smith   PetscFunctionBegin;
1060700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,1);
1074482741eSBarry Smith   PetscValidPointer(mapping,2);
1082bdab257SBarry Smith 
1092bdab257SBarry Smith   ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr);
1103b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
1112bdab257SBarry Smith   ierr = ISGetIndices(is,&indices);CHKERRQ(ierr);
112d5ad8652SBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
1132bdab257SBarry Smith   ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr);
1143a40ed3dSBarry Smith   PetscFunctionReturn(0);
1152bdab257SBarry Smith }
1165a5d4f66SBarry Smith 
117a4d96a55SJed Brown #undef __FUNCT__
118a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF"
119a4d96a55SJed Brown /*@C
120a4d96a55SJed Brown     ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n)
121a4d96a55SJed Brown     ordering and a global parallel ordering.
122a4d96a55SJed Brown 
123a4d96a55SJed Brown     Collective
124a4d96a55SJed Brown 
125a4d96a55SJed Brown     Input Parameter:
126a4d96a55SJed Brown +   sf - star forest mapping contiguous local indices to (rank, offset)
127a4d96a55SJed Brown -   start - first global index on this process
128a4d96a55SJed Brown 
129a4d96a55SJed Brown     Output Parameter:
130a4d96a55SJed Brown .   mapping - new mapping data structure
131a4d96a55SJed Brown 
132a4d96a55SJed Brown     Level: advanced
133a4d96a55SJed Brown 
134a4d96a55SJed Brown     Concepts: mapping^local to global
135a4d96a55SJed Brown 
136a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
137a4d96a55SJed Brown @*/
138a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping)
139a4d96a55SJed Brown {
140a4d96a55SJed Brown   PetscErrorCode ierr;
141a4d96a55SJed Brown   PetscInt       i,maxlocal,nroots,nleaves,*globals,*ltog;
142a4d96a55SJed Brown   const PetscInt *ilocal;
143a4d96a55SJed Brown   MPI_Comm       comm;
144a4d96a55SJed Brown 
145a4d96a55SJed Brown   PetscFunctionBegin;
146a4d96a55SJed Brown   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
147a4d96a55SJed Brown   PetscValidPointer(mapping,3);
148a4d96a55SJed Brown 
149a4d96a55SJed Brown   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
150a4d96a55SJed Brown   ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,PETSC_NULL);CHKERRQ(ierr);
151a4d96a55SJed Brown   if (ilocal) {for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1);}
152a4d96a55SJed Brown   else maxlocal = nleaves;
153a4d96a55SJed Brown   ierr = PetscMalloc(nroots*sizeof(PetscInt),&globals);CHKERRQ(ierr);
154a4d96a55SJed Brown   ierr = PetscMalloc(maxlocal*sizeof(PetscInt),&ltog);CHKERRQ(ierr);
155a4d96a55SJed Brown   for (i=0; i<nroots; i++) globals[i] = start + i;
156a4d96a55SJed Brown   for (i=0; i<maxlocal; i++) ltog[i] = -1;
157a4d96a55SJed Brown   ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
158a4d96a55SJed Brown   ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
159a4d96a55SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
160a4d96a55SJed Brown   ierr = PetscFree(globals);CHKERRQ(ierr);
161a4d96a55SJed Brown   PetscFunctionReturn(0);
162a4d96a55SJed Brown }
163b46b645bSBarry Smith 
1644a2ae208SSatish Balay #undef __FUNCT__
1654a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate"
166ba5bb76aSSatish Balay /*@
16790f02eecSBarry Smith     ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n)
16890f02eecSBarry Smith     ordering and a global parallel ordering.
1692362add9SBarry Smith 
17089d82c54SBarry Smith     Not Collective, but communicator may have more than one process
171b9cd556bSLois Curfman McInnes 
1722362add9SBarry Smith     Input Parameters:
17389d82c54SBarry Smith +   comm - MPI communicator
17490f02eecSBarry Smith .   n - the number of local elements
1759669e4d8SBarry Smith .   indices - the global index for each local element, these do not need to be in increasing order (sorted)
176d5ad8652SBarry Smith -   mode - see PetscCopyMode
1772362add9SBarry Smith 
178a997ad1aSLois Curfman McInnes     Output Parameter:
17990f02eecSBarry Smith .   mapping - new mapping data structure
1802362add9SBarry Smith 
181a997ad1aSLois Curfman McInnes     Level: advanced
182a997ad1aSLois Curfman McInnes 
183273d9f13SBarry Smith     Concepts: mapping^local to global
1842362add9SBarry Smith 
185d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
1862362add9SBarry Smith @*/
1877087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping)
1882362add9SBarry Smith {
1896849ba73SBarry Smith   PetscErrorCode ierr;
19032dcc486SBarry Smith   PetscInt       *in;
191b46b645bSBarry Smith 
192b46b645bSBarry Smith   PetscFunctionBegin;
19373911063SBarry Smith   if (n) PetscValidIntPointer(indices,3);
1944482741eSBarry Smith   PetscValidPointer(mapping,4);
195b46b645bSBarry Smith 
1968e58c17dSMatthew Knepley   *mapping = PETSC_NULL;
197519f805aSKarl Rupp #if !defined(PETSC_USE_DYNAMIC_LIBRARIES)
1982b6de112SBarry Smith   ierr = ISInitializePackage(PETSC_NULL);CHKERRQ(ierr);
1998e58c17dSMatthew Knepley #endif
2002362add9SBarry Smith 
2013194b578SJed Brown   ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,0,"ISLocalToGlobalMapping","Local to global mapping","IS",
20252e6d16bSBarry Smith                            cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr);
203d4bb536fSBarry Smith   (*mapping)->n       = n;
204d4bb536fSBarry Smith   /*
205d4bb536fSBarry Smith     Do not create the global to local mapping. This is only created if
206d4bb536fSBarry Smith     ISGlobalToLocalMapping() is called
207d4bb536fSBarry Smith   */
208d4bb536fSBarry Smith   (*mapping)->globals = 0;
209d5ad8652SBarry Smith   if (mode == PETSC_COPY_VALUES) {
210d5ad8652SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&in);CHKERRQ(ierr);
211d5ad8652SBarry Smith     ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr);
212d5ad8652SBarry Smith     ierr = PetscLogObjectMemory(*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
213d5ad8652SBarry Smith     (*mapping)->indices = in;
214d5ad8652SBarry Smith   } else if (mode == PETSC_OWN_POINTER) {
215d5ad8652SBarry Smith     (*mapping)->indices = (PetscInt*)indices;
216d5ad8652SBarry Smith   } else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER");
2173a40ed3dSBarry Smith   PetscFunctionReturn(0);
2182362add9SBarry Smith }
2192362add9SBarry Smith 
2204a2ae208SSatish Balay #undef __FUNCT__
2214a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingBlock"
222bce096a4SSatish Balay /*@
223323b833fSBarry Smith     ISLocalToGlobalMappingBlock - Creates a blocked index version of an
224323b833fSBarry Smith        ISLocalToGlobalMapping that is appropriate for MatSetLocalToGlobalMappingBlock()
225323b833fSBarry Smith        and VecSetLocalToGlobalMappingBlock().
226323b833fSBarry Smith 
227323b833fSBarry Smith     Not Collective, but communicator may have more than one process
228323b833fSBarry Smith 
229323b833fSBarry Smith     Input Parameters:
230323b833fSBarry Smith +    inmap - original point-wise mapping
231323b833fSBarry Smith -    bs - block size
232323b833fSBarry Smith 
233323b833fSBarry Smith     Output Parameter:
23469eb54c3SBarry Smith .   outmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
235323b833fSBarry Smith 
236323b833fSBarry Smith     Level: advanced
237323b833fSBarry Smith 
238323b833fSBarry Smith     Concepts: mapping^local to global
239323b833fSBarry Smith 
240323b833fSBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
241323b833fSBarry Smith @*/
2427087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
243323b833fSBarry Smith {
2446849ba73SBarry Smith   PetscErrorCode ierr;
24532dcc486SBarry Smith   PetscInt       *ii,i,n;
246323b833fSBarry Smith 
247323b833fSBarry Smith   PetscFunctionBegin;
2480700a824SBarry Smith   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
249b2beed0aSJed Brown   PetscValidPointer(outmap,3);
250323b833fSBarry Smith   if (bs > 1) {
251323b833fSBarry Smith     n    = inmap->n/bs;
252e32f2f54SBarry Smith     if (n*bs != inmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Pointwise mapping length is not divisible by block size");
25332dcc486SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
254323b833fSBarry Smith     for (i=0; i<n; i++) {
255db032c9dSBarry Smith       ii[i] = inmap->indices[bs*i]/bs;
256323b833fSBarry Smith     }
257d5ad8652SBarry Smith     ierr = ISLocalToGlobalMappingCreate(((PetscObject)inmap)->comm,n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
258323b833fSBarry Smith   } else {
259323b833fSBarry Smith     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
260c3122656SLisandro Dalcin     *outmap = inmap;
261323b833fSBarry Smith   }
262323b833fSBarry Smith   PetscFunctionReturn(0);
263323b833fSBarry Smith }
264323b833fSBarry Smith 
2654a2ae208SSatish Balay #undef __FUNCT__
2668ab951edSJed Brown #define __FUNCT__ "ISLocalToGlobalMappingUnBlock"
267b2beed0aSJed Brown /*@
268b2beed0aSJed Brown     ISLocalToGlobalMappingUnBlock - Creates a scalar index version of a blocked
269b2beed0aSJed Brown        ISLocalToGlobalMapping
270b2beed0aSJed Brown 
271b2beed0aSJed Brown     Not Collective, but communicator may have more than one process
272b2beed0aSJed Brown 
273b2beed0aSJed Brown     Input Parameter:
274b2beed0aSJed Brown + inmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
275b2beed0aSJed Brown - bs - block size
276b2beed0aSJed Brown 
277b2beed0aSJed Brown     Output Parameter:
278b2beed0aSJed Brown .   outmap - pointwise mapping
279b2beed0aSJed Brown 
280b2beed0aSJed Brown     Level: advanced
281b2beed0aSJed Brown 
282b2beed0aSJed Brown     Concepts: mapping^local to global
283b2beed0aSJed Brown 
284b2beed0aSJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingBlock()
285b2beed0aSJed Brown @*/
2867087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
287b2beed0aSJed Brown {
288b2beed0aSJed Brown   PetscErrorCode ierr;
289b2beed0aSJed Brown   PetscInt       *ii,i,n;
290b2beed0aSJed Brown 
291b2beed0aSJed Brown   PetscFunctionBegin;
292b2beed0aSJed Brown   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
293b2beed0aSJed Brown   PetscValidPointer(outmap,2);
294b2beed0aSJed Brown   if (bs > 1) {
295b2beed0aSJed Brown     n    = inmap->n*bs;
296b2beed0aSJed Brown     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
297b2beed0aSJed Brown     for (i=0; i<n; i++) {
29894f4c46bSJed Brown       ii[i] = inmap->indices[i/bs]*bs + (i%bs);
299b2beed0aSJed Brown     }
300b2beed0aSJed Brown     ierr = ISLocalToGlobalMappingCreate(((PetscObject)inmap)->comm,n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
301b2beed0aSJed Brown   } else {
302b2beed0aSJed Brown     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
303b2beed0aSJed Brown     *outmap = inmap;
304b2beed0aSJed Brown   }
305b2beed0aSJed Brown   PetscFunctionReturn(0);
306b2beed0aSJed Brown }
307b2beed0aSJed Brown 
308b2beed0aSJed Brown #undef __FUNCT__
3094a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy"
31090f02eecSBarry Smith /*@
31190f02eecSBarry Smith    ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n)
31290f02eecSBarry Smith    ordering and a global parallel ordering.
31390f02eecSBarry Smith 
3140f5bd95cSBarry Smith    Note Collective
315b9cd556bSLois Curfman McInnes 
31690f02eecSBarry Smith    Input Parameters:
31790f02eecSBarry Smith .  mapping - mapping data structure
31890f02eecSBarry Smith 
319a997ad1aSLois Curfman McInnes    Level: advanced
320a997ad1aSLois Curfman McInnes 
3213acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate()
32290f02eecSBarry Smith @*/
3236bf464f9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping)
32490f02eecSBarry Smith {
325dfbe8321SBarry Smith   PetscErrorCode ierr;
326*5fd66863SKarl Rupp 
3273a40ed3dSBarry Smith   PetscFunctionBegin;
3286bf464f9SBarry Smith   if (!*mapping) PetscFunctionReturn(0);
3296bf464f9SBarry Smith   PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1);
330997056adSBarry Smith   if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);}
3316bf464f9SBarry Smith   ierr = PetscFree((*mapping)->indices);CHKERRQ(ierr);
3326bf464f9SBarry Smith   ierr = PetscFree((*mapping)->globals);CHKERRQ(ierr);
333d38fa0fbSBarry Smith   ierr = PetscHeaderDestroy(mapping);CHKERRQ(ierr);
334992144d0SBarry Smith   *mapping = 0;
3353a40ed3dSBarry Smith   PetscFunctionReturn(0);
33690f02eecSBarry Smith }
33790f02eecSBarry Smith 
3384a2ae208SSatish Balay #undef __FUNCT__
3394a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS"
34090f02eecSBarry Smith /*@
3413acfe500SLois Curfman McInnes     ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering
3423acfe500SLois Curfman McInnes     a new index set using the global numbering defined in an ISLocalToGlobalMapping
3433acfe500SLois Curfman McInnes     context.
34490f02eecSBarry Smith 
345b9cd556bSLois Curfman McInnes     Not collective
346b9cd556bSLois Curfman McInnes 
34790f02eecSBarry Smith     Input Parameters:
348b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
349b9cd556bSLois Curfman McInnes -   is - index set in local numbering
35090f02eecSBarry Smith 
35190f02eecSBarry Smith     Output Parameters:
35290f02eecSBarry Smith .   newis - index set in global numbering
35390f02eecSBarry Smith 
354a997ad1aSLois Curfman McInnes     Level: advanced
355a997ad1aSLois Curfman McInnes 
356273d9f13SBarry Smith     Concepts: mapping^local to global
3573acfe500SLois Curfman McInnes 
35890f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
359d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply()
36090f02eecSBarry Smith @*/
3617087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis)
36290f02eecSBarry Smith {
3636849ba73SBarry Smith   PetscErrorCode ierr;
3645d0c19d7SBarry Smith   PetscInt       n,i,*idxmap,*idxout,Nmax = mapping->n;
3655d0c19d7SBarry Smith   const PetscInt *idxin;
3663a40ed3dSBarry Smith 
3673a40ed3dSBarry Smith   PetscFunctionBegin;
3680700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
3690700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,2);
3704482741eSBarry Smith   PetscValidPointer(newis,3);
37190f02eecSBarry Smith 
3723b9aefa3SBarry Smith   ierr   = ISGetLocalSize(is,&n);CHKERRQ(ierr);
37390f02eecSBarry Smith   ierr   = ISGetIndices(is,&idxin);CHKERRQ(ierr);
37490f02eecSBarry Smith   idxmap = mapping->indices;
37590f02eecSBarry Smith 
3767c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&idxout);CHKERRQ(ierr);
37790f02eecSBarry Smith   for (i=0; i<n; i++) {
378e32f2f54SBarry Smith     if (idxin[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %d too large %d (max) at %d",idxin[i],Nmax-1,i);
37990f02eecSBarry Smith     idxout[i] = idxmap[idxin[i]];
38090f02eecSBarry Smith   }
3813b9aefa3SBarry Smith   ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr);
38270b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr);
3833a40ed3dSBarry Smith   PetscFunctionReturn(0);
38490f02eecSBarry Smith }
38590f02eecSBarry Smith 
38689d82c54SBarry Smith /*MC
3873acfe500SLois Curfman McInnes    ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering
3883acfe500SLois Curfman McInnes    and converts them to the global numbering.
38990f02eecSBarry Smith 
390eca87e8dSBarry Smith    Synopsis:
391f2ba6396SBarry Smith    #include "petscis.h"
392eca87e8dSBarry Smith    PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,int N,int in[],int out[])
393eca87e8dSBarry Smith 
394b9cd556bSLois Curfman McInnes    Not collective
395b9cd556bSLois Curfman McInnes 
396bb25748dSBarry Smith    Input Parameters:
397b9cd556bSLois Curfman McInnes +  mapping - the local to global mapping context
398bb25748dSBarry Smith .  N - number of integers
399b9cd556bSLois Curfman McInnes -  in - input indices in local numbering
400bb25748dSBarry Smith 
401bb25748dSBarry Smith    Output Parameter:
402bb25748dSBarry Smith .  out - indices in global numbering
403bb25748dSBarry Smith 
404b9cd556bSLois Curfman McInnes    Notes:
405b9cd556bSLois Curfman McInnes    The in and out array parameters may be identical.
406d4bb536fSBarry Smith 
407a997ad1aSLois Curfman McInnes    Level: advanced
408a997ad1aSLois Curfman McInnes 
409bb25748dSBarry Smith .seealso: ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
4100752156aSBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
411d4bb536fSBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
412bb25748dSBarry Smith 
413273d9f13SBarry Smith     Concepts: mapping^local to global
414d4bb536fSBarry Smith 
41589d82c54SBarry Smith M*/
416d4bb536fSBarry Smith 
417d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/
418d4bb536fSBarry Smith 
4194a2ae208SSatish Balay #undef __FUNCT__
4204a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private"
421d4bb536fSBarry Smith /*
422d4bb536fSBarry Smith     Creates the global fields in the ISLocalToGlobalMapping structure
423d4bb536fSBarry Smith */
4246849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping)
425d4bb536fSBarry Smith {
4266849ba73SBarry Smith   PetscErrorCode ierr;
42732dcc486SBarry Smith   PetscInt       i,*idx = mapping->indices,n = mapping->n,end,start,*globals;
428d4bb536fSBarry Smith 
4293a40ed3dSBarry Smith   PetscFunctionBegin;
430d4bb536fSBarry Smith   end   = 0;
431ec268f7cSJed Brown   start = PETSC_MAX_INT;
432d4bb536fSBarry Smith 
433d4bb536fSBarry Smith   for (i=0; i<n; i++) {
434d4bb536fSBarry Smith     if (idx[i] < 0) continue;
435d4bb536fSBarry Smith     if (idx[i] < start) start = idx[i];
436d4bb536fSBarry Smith     if (idx[i] > end)   end   = idx[i];
437d4bb536fSBarry Smith   }
438d4bb536fSBarry Smith   if (start > end) {start = 0; end = -1;}
439d4bb536fSBarry Smith   mapping->globalstart = start;
440d4bb536fSBarry Smith   mapping->globalend   = end;
441d4bb536fSBarry Smith 
44232dcc486SBarry Smith   ierr             = PetscMalloc((end-start+2)*sizeof(PetscInt),&globals);CHKERRQ(ierr);
443b0a32e0cSBarry Smith   mapping->globals = globals;
444d4bb536fSBarry Smith   for (i=0; i<end-start+1; i++) {
445d4bb536fSBarry Smith     globals[i] = -1;
446d4bb536fSBarry Smith   }
447d4bb536fSBarry Smith   for (i=0; i<n; i++) {
448d4bb536fSBarry Smith     if (idx[i] < 0) continue;
449d4bb536fSBarry Smith     globals[idx[i] - start] = i;
450d4bb536fSBarry Smith   }
451d4bb536fSBarry Smith 
45252e6d16bSBarry Smith   ierr = PetscLogObjectMemory(mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr);
4533a40ed3dSBarry Smith   PetscFunctionReturn(0);
454d4bb536fSBarry Smith }
455d4bb536fSBarry Smith 
4564a2ae208SSatish Balay #undef __FUNCT__
4574a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply"
458d4bb536fSBarry Smith /*@
459a997ad1aSLois Curfman McInnes     ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers
460a997ad1aSLois Curfman McInnes     specified with a global numbering.
461d4bb536fSBarry Smith 
462b9cd556bSLois Curfman McInnes     Not collective
463b9cd556bSLois Curfman McInnes 
464d4bb536fSBarry Smith     Input Parameters:
465b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
466d4bb536fSBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
467d4bb536fSBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
468d4bb536fSBarry Smith .   n - number of global indices to map
469b9cd556bSLois Curfman McInnes -   idx - global indices to map
470d4bb536fSBarry Smith 
471d4bb536fSBarry Smith     Output Parameters:
472b9cd556bSLois Curfman McInnes +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
473b9cd556bSLois Curfman McInnes -   idxout - local index of each global index, one must pass in an array long enough
474e182c471SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApply() with
475e182c471SBarry Smith              idxout == PETSC_NULL to determine the required length (returned in nout)
476e182c471SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApply()
477e182c471SBarry Smith              a second time to set the values.
478d4bb536fSBarry Smith 
479b9cd556bSLois Curfman McInnes     Notes:
480b9cd556bSLois Curfman McInnes     Either nout or idxout may be PETSC_NULL. idx and idxout may be identical.
481d4bb536fSBarry Smith 
4820f5bd95cSBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
4830f5bd95cSBarry Smith     array to compute these.
4840f5bd95cSBarry Smith 
485a997ad1aSLois Curfman McInnes     Level: advanced
486a997ad1aSLois Curfman McInnes 
48732fd6b96SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
48832fd6b96SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
48932fd6b96SBarry Smith 
490273d9f13SBarry Smith     Concepts: mapping^global to local
491d4bb536fSBarry Smith 
492d4bb536fSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
493d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy()
494d4bb536fSBarry Smith @*/
4957087cfbeSBarry Smith PetscErrorCode  ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
49632dcc486SBarry Smith                                   PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
497d4bb536fSBarry Smith {
49832dcc486SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end;
4996849ba73SBarry Smith   PetscErrorCode ierr;
500d4bb536fSBarry Smith 
5013a40ed3dSBarry Smith   PetscFunctionBegin;
5020700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
503d4bb536fSBarry Smith   if (!mapping->globals) {
504d4bb536fSBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
505d4bb536fSBarry Smith   }
506d4bb536fSBarry Smith   globals = mapping->globals;
507d4bb536fSBarry Smith   start   = mapping->globalstart;
508d4bb536fSBarry Smith   end     = mapping->globalend;
509d4bb536fSBarry Smith 
510d4bb536fSBarry Smith   if (type == IS_GTOLM_MASK) {
511d4bb536fSBarry Smith     if (idxout) {
512d4bb536fSBarry Smith       for (i=0; i<n; i++) {
513d4bb536fSBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
514d4bb536fSBarry Smith         else if (idx[i] < start) idxout[i] = -1;
515d4bb536fSBarry Smith         else if (idx[i] > end)   idxout[i] = -1;
516d4bb536fSBarry Smith         else                     idxout[i] = globals[idx[i] - start];
517d4bb536fSBarry Smith       }
518d4bb536fSBarry Smith     }
519d4bb536fSBarry Smith     if (nout) *nout = n;
520d4bb536fSBarry Smith   } else {
521d4bb536fSBarry Smith     if (idxout) {
522d4bb536fSBarry Smith       for (i=0; i<n; i++) {
523d4bb536fSBarry Smith         if (idx[i] < 0) continue;
524d4bb536fSBarry Smith         if (idx[i] < start) continue;
525d4bb536fSBarry Smith         if (idx[i] > end) continue;
526d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
527d4bb536fSBarry Smith         if (tmp < 0) continue;
528d4bb536fSBarry Smith         idxout[nf++] = tmp;
529d4bb536fSBarry Smith       }
530d4bb536fSBarry Smith     } else {
531d4bb536fSBarry Smith       for (i=0; i<n; i++) {
532d4bb536fSBarry Smith         if (idx[i] < 0) continue;
533d4bb536fSBarry Smith         if (idx[i] < start) continue;
534d4bb536fSBarry Smith         if (idx[i] > end) continue;
535d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
536d4bb536fSBarry Smith         if (tmp < 0) continue;
537d4bb536fSBarry Smith         nf++;
538d4bb536fSBarry Smith       }
539d4bb536fSBarry Smith     }
540d4bb536fSBarry Smith     if (nout) *nout = nf;
541d4bb536fSBarry Smith   }
542d4bb536fSBarry Smith 
5433a40ed3dSBarry Smith   PetscFunctionReturn(0);
544d4bb536fSBarry Smith }
54590f02eecSBarry Smith 
5464a2ae208SSatish Balay #undef __FUNCT__
5474a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetInfo"
54889d82c54SBarry Smith /*@C
54989d82c54SBarry Smith     ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and
55089d82c54SBarry Smith      each index shared by more than one processor
55189d82c54SBarry Smith 
55289d82c54SBarry Smith     Collective on ISLocalToGlobalMapping
55389d82c54SBarry Smith 
55489d82c54SBarry Smith     Input Parameters:
55589d82c54SBarry Smith .   mapping - the mapping from local to global indexing
55689d82c54SBarry Smith 
55789d82c54SBarry Smith     Output Parameter:
55889d82c54SBarry Smith +   nproc - number of processors that are connected to this one
55989d82c54SBarry Smith .   proc - neighboring processors
56007b52d57SBarry Smith .   numproc - number of indices for each subdomain (processor)
5613463a7baSJed Brown -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
56289d82c54SBarry Smith 
56389d82c54SBarry Smith     Level: advanced
56489d82c54SBarry Smith 
565273d9f13SBarry Smith     Concepts: mapping^local to global
56689d82c54SBarry Smith 
5672cfcea29SBarry Smith     Fortran Usage:
5682cfcea29SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
5692cfcea29SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
5702cfcea29SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
5712cfcea29SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
5722cfcea29SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
5732cfcea29SBarry Smith 
5742cfcea29SBarry Smith 
57507b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
57607b52d57SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
57789d82c54SBarry Smith @*/
5787087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
57989d82c54SBarry Smith {
5806849ba73SBarry Smith   PetscErrorCode ierr;
58197f1f81fSBarry Smith   PetscMPIInt    size,rank,tag1,tag2,tag3,*len,*source,imdex;
58232dcc486SBarry Smith   PetscInt       i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices;
58332dcc486SBarry Smith   PetscInt       *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc;
58497f1f81fSBarry Smith   PetscInt       cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned;
58532dcc486SBarry Smith   PetscInt       node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp;
58632dcc486SBarry Smith   PetscInt       first_procs,first_numprocs,*first_indices;
58789d82c54SBarry Smith   MPI_Request    *recv_waits,*send_waits;
58830dcb7c9SBarry Smith   MPI_Status     recv_status,*send_status,*recv_statuses;
5897adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mapping)->comm;
590ace3abfcSBarry Smith   PetscBool      debug = PETSC_FALSE;
59189d82c54SBarry Smith 
59289d82c54SBarry Smith   PetscFunctionBegin;
5930700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
59424cf384cSBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
59524cf384cSBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
59624cf384cSBarry Smith   if (size == 1) {
59724cf384cSBarry Smith     *nproc         = 0;
59824cf384cSBarry Smith     *procs         = PETSC_NULL;
59932dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr);
6001e2105dcSBarry Smith     (*numprocs)[0] = 0;
60132dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr);
6021e2105dcSBarry Smith     (*indices)[0]  = PETSC_NULL;
60324cf384cSBarry Smith     PetscFunctionReturn(0);
60424cf384cSBarry Smith   }
60524cf384cSBarry Smith 
606acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,PETSC_NULL);CHKERRQ(ierr);
60707b52d57SBarry Smith 
6083677ff5aSBarry Smith   /*
6093677ff5aSBarry Smith     Notes on ISLocalToGlobalMappingGetInfo
6103677ff5aSBarry Smith 
6113677ff5aSBarry Smith     globally owned node - the nodes that have been assigned to this processor in global
6123677ff5aSBarry Smith            numbering, just for this routine.
6133677ff5aSBarry Smith 
6143677ff5aSBarry Smith     nontrivial globally owned node - node assigned to this processor that is on a subdomain
6153677ff5aSBarry Smith            boundary (i.e. is has more than one local owner)
6163677ff5aSBarry Smith 
6173677ff5aSBarry Smith     locally owned node - node that exists on this processors subdomain
6183677ff5aSBarry Smith 
6193677ff5aSBarry Smith     nontrivial locally owned node - node that is not in the interior (i.e. has more than one
6203677ff5aSBarry Smith            local subdomain
6213677ff5aSBarry Smith   */
62224cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr);
62324cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr);
62424cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr);
62589d82c54SBarry Smith 
62689d82c54SBarry Smith   for (i=0; i<n; i++) {
62789d82c54SBarry Smith     if (lindices[i] > max) max = lindices[i];
62889d82c54SBarry Smith   }
62932dcc486SBarry Smith   ierr   = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
63078058e43SBarry Smith   Ng++;
63189d82c54SBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
63289d82c54SBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
633bc8ff85bSBarry Smith   scale  = Ng/size + 1;
634a2e34c3dSBarry Smith   ng     = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng);
635caba0dd0SBarry Smith   rstart = scale*rank;
63689d82c54SBarry Smith 
63789d82c54SBarry Smith   /* determine ownership ranges of global indices */
6387c334f02SBarry Smith   ierr = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
63932dcc486SBarry Smith   ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
64089d82c54SBarry Smith 
64189d82c54SBarry Smith   /* determine owners of each local node  */
6427c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&owner);CHKERRQ(ierr);
64389d82c54SBarry Smith   for (i=0; i<n; i++) {
6443677ff5aSBarry Smith     proc             = lindices[i]/scale; /* processor that globally owns this index */
64527c402fcSBarry Smith     nprocs[2*proc+1] = 1;                 /* processor globally owns at least one of ours */
6463677ff5aSBarry Smith     owner[i]         = proc;
64727c402fcSBarry Smith     nprocs[2*proc]++;                     /* count of how many that processor globally owns of ours */
64889d82c54SBarry Smith   }
64927c402fcSBarry Smith   nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1];
6501e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of global owners for my local data %d\n",nsends);CHKERRQ(ierr);
65189d82c54SBarry Smith 
65289d82c54SBarry Smith   /* inform other processors of number of messages and max length*/
65327c402fcSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
6541e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of local owners for my global data %d\n",nrecvs);CHKERRQ(ierr);
65589d82c54SBarry Smith 
65689d82c54SBarry Smith   /* post receives for owned rows */
65732dcc486SBarry Smith   ierr = PetscMalloc((2*nrecvs+1)*(nmax+1)*sizeof(PetscInt),&recvs);CHKERRQ(ierr);
658b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
65989d82c54SBarry Smith   for (i=0; i<nrecvs; i++) {
66032dcc486SBarry Smith     ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr);
66189d82c54SBarry Smith   }
66289d82c54SBarry Smith 
66389d82c54SBarry Smith   /* pack messages containing lists of local nodes to owners */
66432dcc486SBarry Smith   ierr       = PetscMalloc((2*n+1)*sizeof(PetscInt),&sends);CHKERRQ(ierr);
66532dcc486SBarry Smith   ierr       = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
66689d82c54SBarry Smith   starts[0]  = 0;
66727c402fcSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + 2*nprocs[2*i-2];}
66889d82c54SBarry Smith   for (i=0; i<n; i++) {
66989d82c54SBarry Smith     sends[starts[owner[i]]++] = lindices[i];
67030dcb7c9SBarry Smith     sends[starts[owner[i]]++] = i;
67189d82c54SBarry Smith   }
67289d82c54SBarry Smith   ierr = PetscFree(owner);CHKERRQ(ierr);
67389d82c54SBarry Smith   starts[0]  = 0;
67427c402fcSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + 2*nprocs[2*i-2];}
67589d82c54SBarry Smith 
67689d82c54SBarry Smith   /* send the messages */
677b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
67832dcc486SBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(PetscInt),&dest);CHKERRQ(ierr);
67989d82c54SBarry Smith   cnt = 0;
68089d82c54SBarry Smith   for (i=0; i<size; i++) {
68127c402fcSBarry Smith     if (nprocs[2*i]) {
68232dcc486SBarry Smith       ierr      = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr);
68330dcb7c9SBarry Smith       dest[cnt] = i;
68489d82c54SBarry Smith       cnt++;
68589d82c54SBarry Smith     }
68689d82c54SBarry Smith   }
68789d82c54SBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
68889d82c54SBarry Smith 
68989d82c54SBarry Smith   /* wait on receives */
69097f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&source);CHKERRQ(ierr);
69197f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&len);CHKERRQ(ierr);
69289d82c54SBarry Smith   cnt  = nrecvs;
69332dcc486SBarry Smith   ierr = PetscMalloc((ng+1)*sizeof(PetscInt),&nownedsenders);CHKERRQ(ierr);
69432dcc486SBarry Smith   ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr);
69589d82c54SBarry Smith   while (cnt) {
69689d82c54SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
69789d82c54SBarry Smith     /* unpack receives into our local space */
69832dcc486SBarry Smith     ierr           = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr);
69989d82c54SBarry Smith     source[imdex]  = recv_status.MPI_SOURCE;
70030dcb7c9SBarry Smith     len[imdex]     = len[imdex]/2;
701caba0dd0SBarry Smith     /* count how many local owners for each of my global owned indices */
70230dcb7c9SBarry Smith     for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++;
70389d82c54SBarry Smith     cnt--;
70489d82c54SBarry Smith   }
70589d82c54SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
70689d82c54SBarry Smith 
70730dcb7c9SBarry Smith   /* count how many globally owned indices are on an edge multiplied by how many processors own them. */
708bc8ff85bSBarry Smith   nowned  = 0;
709bc8ff85bSBarry Smith   nownedm = 0;
710bc8ff85bSBarry Smith   for (i=0; i<ng; i++) {
711bc8ff85bSBarry Smith     if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;}
712bc8ff85bSBarry Smith   }
713bc8ff85bSBarry Smith 
714bc8ff85bSBarry Smith   /* create single array to contain rank of all local owners of each globally owned index */
71532dcc486SBarry Smith   ierr      = PetscMalloc((nownedm+1)*sizeof(PetscInt),&ownedsenders);CHKERRQ(ierr);
71632dcc486SBarry Smith   ierr      = PetscMalloc((ng+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
717bc8ff85bSBarry Smith   starts[0] = 0;
718bc8ff85bSBarry Smith   for (i=1; i<ng; i++) {
719bc8ff85bSBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
720bc8ff85bSBarry Smith     else starts[i] = starts[i-1];
721bc8ff85bSBarry Smith   }
722bc8ff85bSBarry Smith 
72330dcb7c9SBarry Smith   /* for each nontrival globally owned node list all arriving processors */
724bc8ff85bSBarry Smith   for (i=0; i<nrecvs; i++) {
725bc8ff85bSBarry Smith     for (j=0; j<len[i]; j++) {
72630dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
727bc8ff85bSBarry Smith       if (nownedsenders[node] > 1) {
728bc8ff85bSBarry Smith         ownedsenders[starts[node]++] = source[i];
729bc8ff85bSBarry Smith       }
730bc8ff85bSBarry Smith     }
731bc8ff85bSBarry Smith   }
732bc8ff85bSBarry Smith 
73307b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
73430dcb7c9SBarry Smith     starts[0]    = 0;
73530dcb7c9SBarry Smith     for (i=1; i<ng; i++) {
73630dcb7c9SBarry Smith       if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
73730dcb7c9SBarry Smith       else starts[i] = starts[i-1];
73830dcb7c9SBarry Smith     }
73930dcb7c9SBarry Smith     for (i=0; i<ng; i++) {
74030dcb7c9SBarry Smith       if (nownedsenders[i] > 1) {
74130dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] global node %d local owner processors: ",rank,i+rstart);CHKERRQ(ierr);
74230dcb7c9SBarry Smith         for (j=0; j<nownedsenders[i]; j++) {
74330dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",ownedsenders[starts[i]+j]);CHKERRQ(ierr);
74430dcb7c9SBarry Smith         }
74530dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
74630dcb7c9SBarry Smith       }
74730dcb7c9SBarry Smith     }
74830dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
74907b52d57SBarry Smith   }/* -----------------------------------  */
75030dcb7c9SBarry Smith 
7513677ff5aSBarry Smith   /* wait on original sends */
7523a96401aSBarry Smith   if (nsends) {
753b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
7543a96401aSBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
7553a96401aSBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
7563a96401aSBarry Smith   }
75789d82c54SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
7583a96401aSBarry Smith   ierr = PetscFree(sends);CHKERRQ(ierr);
7593677ff5aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
7603677ff5aSBarry Smith 
7613677ff5aSBarry Smith   /* pack messages to send back to local owners */
76230dcb7c9SBarry Smith   starts[0]    = 0;
76330dcb7c9SBarry Smith   for (i=1; i<ng; i++) {
76430dcb7c9SBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
76530dcb7c9SBarry Smith     else starts[i] = starts[i-1];
76630dcb7c9SBarry Smith   }
76730dcb7c9SBarry Smith   nsends2 = nrecvs;
76832dcc486SBarry Smith   ierr    = PetscMalloc((nsends2+1)*sizeof(PetscInt),&nprocs);CHKERRQ(ierr); /* length of each message */
76930dcb7c9SBarry Smith   for (i=0; i<nrecvs; i++) {
77030dcb7c9SBarry Smith     nprocs[i] = 1;
77130dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
77230dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
77330dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
77430dcb7c9SBarry Smith         nprocs[i] += 2 + nownedsenders[node];
77530dcb7c9SBarry Smith       }
77630dcb7c9SBarry Smith     }
77730dcb7c9SBarry Smith   }
77830dcb7c9SBarry Smith   nt = 0; for (i=0; i<nsends2; i++) nt += nprocs[i];
77932dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&sends2);CHKERRQ(ierr);
78032dcc486SBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(PetscInt),&starts2);CHKERRQ(ierr);
78130dcb7c9SBarry Smith   starts2[0] = 0; for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1];
78230dcb7c9SBarry Smith   /*
78330dcb7c9SBarry Smith      Each message is 1 + nprocs[i] long, and consists of
78430dcb7c9SBarry Smith        (0) the number of nodes being sent back
78530dcb7c9SBarry Smith        (1) the local node number,
78630dcb7c9SBarry Smith        (2) the number of processors sharing it,
78730dcb7c9SBarry Smith        (3) the processors sharing it
78830dcb7c9SBarry Smith   */
78930dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
79030dcb7c9SBarry Smith     cnt = 1;
79130dcb7c9SBarry Smith     sends2[starts2[i]] = 0;
79230dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
79330dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
79430dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
79530dcb7c9SBarry Smith         sends2[starts2[i]]++;
79630dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1];
79730dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = nownedsenders[node];
79832dcc486SBarry Smith         ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr);
79930dcb7c9SBarry Smith         cnt += nownedsenders[node];
80030dcb7c9SBarry Smith       }
80130dcb7c9SBarry Smith     }
80230dcb7c9SBarry Smith   }
80330dcb7c9SBarry Smith 
80430dcb7c9SBarry Smith   /* receive the message lengths */
80530dcb7c9SBarry Smith   nrecvs2 = nsends;
80632dcc486SBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&lens2);CHKERRQ(ierr);
80732dcc486SBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&starts3);CHKERRQ(ierr);
808d44834fbSBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
80930dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
810d44834fbSBarry Smith     ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr);
81130dcb7c9SBarry Smith   }
812d44834fbSBarry Smith 
8138a8e0b3aSBarry Smith   /* send the message lengths */
8148a8e0b3aSBarry Smith   for (i=0; i<nsends2; i++) {
8158a8e0b3aSBarry Smith     ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr);
8168a8e0b3aSBarry Smith   }
8178a8e0b3aSBarry Smith 
818d44834fbSBarry Smith   /* wait on receives of lens */
8190c468ba9SBarry Smith   if (nrecvs2) {
8200c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
821d44834fbSBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
822d44834fbSBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8230c468ba9SBarry Smith   }
824a2ea699eSBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
825d44834fbSBarry Smith 
82630dcb7c9SBarry Smith   starts3[0] = 0;
827d44834fbSBarry Smith   nt         = 0;
82830dcb7c9SBarry Smith   for (i=0; i<nrecvs2-1; i++) {
82930dcb7c9SBarry Smith     starts3[i+1] = starts3[i] + lens2[i];
830d44834fbSBarry Smith     nt          += lens2[i];
83130dcb7c9SBarry Smith   }
83276466f69SStefano Zampini   if (nrecvs2) nt += lens2[nrecvs2-1];
833d44834fbSBarry Smith 
83432dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&recvs2);CHKERRQ(ierr);
835b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
83652b72c4aSBarry Smith   for (i=0; i<nrecvs2; i++) {
83732dcc486SBarry Smith     ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr);
83830dcb7c9SBarry Smith   }
83930dcb7c9SBarry Smith 
84030dcb7c9SBarry Smith   /* send the messages */
841b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
84230dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
84332dcc486SBarry Smith     ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr);
84430dcb7c9SBarry Smith   }
84530dcb7c9SBarry Smith 
84630dcb7c9SBarry Smith   /* wait on receives */
8470c468ba9SBarry Smith   if (nrecvs2) {
8480c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
84930dcb7c9SBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
85030dcb7c9SBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8510c468ba9SBarry Smith   }
85230dcb7c9SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
85330dcb7c9SBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
85430dcb7c9SBarry Smith 
85507b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
85630dcb7c9SBarry Smith     cnt = 0;
85730dcb7c9SBarry Smith     for (i=0; i<nrecvs2; i++) {
85830dcb7c9SBarry Smith       nt = recvs2[cnt++];
85930dcb7c9SBarry Smith       for (j=0; j<nt; j++) {
86030dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] local node %d number of subdomains %d: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr);
86130dcb7c9SBarry Smith         for (k=0; k<recvs2[cnt+1]; k++) {
86230dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",recvs2[cnt+2+k]);CHKERRQ(ierr);
86330dcb7c9SBarry Smith         }
86430dcb7c9SBarry Smith         cnt += 2 + recvs2[cnt+1];
86530dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
86630dcb7c9SBarry Smith       }
86730dcb7c9SBarry Smith     }
86830dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
86907b52d57SBarry Smith   } /* -----------------------------------  */
87030dcb7c9SBarry Smith 
87130dcb7c9SBarry Smith   /* count number subdomains for each local node */
87232dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
87332dcc486SBarry Smith   ierr = PetscMemzero(nprocs,size*sizeof(PetscInt));CHKERRQ(ierr);
87430dcb7c9SBarry Smith   cnt  = 0;
87530dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
87630dcb7c9SBarry Smith     nt = recvs2[cnt++];
87730dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
87830dcb7c9SBarry Smith       for (k=0; k<recvs2[cnt+1]; k++) {
87930dcb7c9SBarry Smith         nprocs[recvs2[cnt+2+k]]++;
88030dcb7c9SBarry Smith       }
88130dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
88230dcb7c9SBarry Smith     }
88330dcb7c9SBarry Smith   }
88430dcb7c9SBarry Smith   nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0);
88530dcb7c9SBarry Smith   *nproc    = nt;
88632dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),procs);CHKERRQ(ierr);
88732dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),numprocs);CHKERRQ(ierr);
88832dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt*),indices);CHKERRQ(ierr);
88976466f69SStefano Zampini   for (i=0;i<nt+1;i++) { (*indices)[i]=PETSC_NULL; }
89032dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&bprocs);CHKERRQ(ierr);
89130dcb7c9SBarry Smith   cnt       = 0;
89230dcb7c9SBarry Smith   for (i=0; i<size; i++) {
89330dcb7c9SBarry Smith     if (nprocs[i] > 0) {
89430dcb7c9SBarry Smith       bprocs[i]        = cnt;
89530dcb7c9SBarry Smith       (*procs)[cnt]    = i;
89630dcb7c9SBarry Smith       (*numprocs)[cnt] = nprocs[i];
89732dcc486SBarry Smith       ierr             = PetscMalloc(nprocs[i]*sizeof(PetscInt),&(*indices)[cnt]);CHKERRQ(ierr);
89830dcb7c9SBarry Smith       cnt++;
89930dcb7c9SBarry Smith     }
90030dcb7c9SBarry Smith   }
90130dcb7c9SBarry Smith 
90230dcb7c9SBarry Smith   /* make the list of subdomains for each nontrivial local node */
90332dcc486SBarry Smith   ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr);
90430dcb7c9SBarry Smith   cnt  = 0;
90530dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
90630dcb7c9SBarry Smith     nt = recvs2[cnt++];
90730dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
90830dcb7c9SBarry Smith       for (k=0; k<recvs2[cnt+1]; k++) {
90930dcb7c9SBarry Smith         (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt];
91030dcb7c9SBarry Smith       }
91130dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
91230dcb7c9SBarry Smith     }
91330dcb7c9SBarry Smith   }
91430dcb7c9SBarry Smith   ierr = PetscFree(bprocs);CHKERRQ(ierr);
91507b52d57SBarry Smith   ierr = PetscFree(recvs2);CHKERRQ(ierr);
91630dcb7c9SBarry Smith 
91707b52d57SBarry Smith   /* sort the node indexing by their global numbers */
91807b52d57SBarry Smith   nt = *nproc;
91907b52d57SBarry Smith   for (i=0; i<nt; i++) {
92032dcc486SBarry Smith     ierr = PetscMalloc(((*numprocs)[i])*sizeof(PetscInt),&tmp);CHKERRQ(ierr);
92107b52d57SBarry Smith     for (j=0; j<(*numprocs)[i]; j++) {
92207b52d57SBarry Smith       tmp[j] = lindices[(*indices)[i][j]];
92307b52d57SBarry Smith     }
92407b52d57SBarry Smith     ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr);
92507b52d57SBarry Smith     ierr = PetscFree(tmp);CHKERRQ(ierr);
92607b52d57SBarry Smith   }
92707b52d57SBarry Smith 
92807b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
92930dcb7c9SBarry Smith     nt = *nproc;
93030dcb7c9SBarry Smith     for (i=0; i<nt; i++) {
93130dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %d number of indices %d: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr);
93230dcb7c9SBarry Smith       for (j=0; j<(*numprocs)[i]; j++) {
93330dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"%d ",(*indices)[i][j]);CHKERRQ(ierr);
93430dcb7c9SBarry Smith       }
93530dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
93630dcb7c9SBarry Smith     }
93730dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
93807b52d57SBarry Smith   } /* -----------------------------------  */
93930dcb7c9SBarry Smith 
94030dcb7c9SBarry Smith   /* wait on sends */
94130dcb7c9SBarry Smith   if (nsends2) {
942b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends2*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
94330dcb7c9SBarry Smith     ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr);
94430dcb7c9SBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
94530dcb7c9SBarry Smith   }
94630dcb7c9SBarry Smith 
94730dcb7c9SBarry Smith   ierr = PetscFree(starts3);CHKERRQ(ierr);
94830dcb7c9SBarry Smith   ierr = PetscFree(dest);CHKERRQ(ierr);
94930dcb7c9SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
9503677ff5aSBarry Smith 
951bc8ff85bSBarry Smith   ierr = PetscFree(nownedsenders);CHKERRQ(ierr);
952bc8ff85bSBarry Smith   ierr = PetscFree(ownedsenders);CHKERRQ(ierr);
953bc8ff85bSBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
95430dcb7c9SBarry Smith   ierr = PetscFree(starts2);CHKERRQ(ierr);
95530dcb7c9SBarry Smith   ierr = PetscFree(lens2);CHKERRQ(ierr);
95689d82c54SBarry Smith 
95789d82c54SBarry Smith   ierr = PetscFree(source);CHKERRQ(ierr);
95897f1f81fSBarry Smith   ierr = PetscFree(len);CHKERRQ(ierr);
95989d82c54SBarry Smith   ierr = PetscFree(recvs);CHKERRQ(ierr);
9603a96401aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
96130dcb7c9SBarry Smith   ierr = PetscFree(sends2);CHKERRQ(ierr);
96224cf384cSBarry Smith 
96324cf384cSBarry Smith   /* put the information about myself as the first entry in the list */
96424cf384cSBarry Smith   first_procs    = (*procs)[0];
96524cf384cSBarry Smith   first_numprocs = (*numprocs)[0];
96624cf384cSBarry Smith   first_indices  = (*indices)[0];
96724cf384cSBarry Smith   for (i=0; i<*nproc; i++) {
96824cf384cSBarry Smith     if ((*procs)[i] == rank) {
96924cf384cSBarry Smith       (*procs)[0]    = (*procs)[i];
97024cf384cSBarry Smith       (*numprocs)[0] = (*numprocs)[i];
97124cf384cSBarry Smith       (*indices)[0]  = (*indices)[i];
97224cf384cSBarry Smith       (*procs)[i]    = first_procs;
97324cf384cSBarry Smith       (*numprocs)[i] = first_numprocs;
97424cf384cSBarry Smith       (*indices)[i]  = first_indices;
97524cf384cSBarry Smith       break;
97624cf384cSBarry Smith     }
97724cf384cSBarry Smith   }
97889d82c54SBarry Smith   PetscFunctionReturn(0);
97989d82c54SBarry Smith }
98089d82c54SBarry Smith 
9814a2ae208SSatish Balay #undef __FUNCT__
9824a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo"
98307b52d57SBarry Smith /*@C
98407b52d57SBarry Smith     ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo()
98589d82c54SBarry Smith 
98607b52d57SBarry Smith     Collective on ISLocalToGlobalMapping
98707b52d57SBarry Smith 
98807b52d57SBarry Smith     Input Parameters:
98907b52d57SBarry Smith .   mapping - the mapping from local to global indexing
99007b52d57SBarry Smith 
99107b52d57SBarry Smith     Output Parameter:
99207b52d57SBarry Smith +   nproc - number of processors that are connected to this one
99307b52d57SBarry Smith .   proc - neighboring processors
99407b52d57SBarry Smith .   numproc - number of indices for each processor
99507b52d57SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
99607b52d57SBarry Smith 
99707b52d57SBarry Smith     Level: advanced
99807b52d57SBarry Smith 
99907b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
100007b52d57SBarry Smith           ISLocalToGlobalMappingGetInfo()
100107b52d57SBarry Smith @*/
10027087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
100307b52d57SBarry Smith {
10046849ba73SBarry Smith   PetscErrorCode ierr;
100532dcc486SBarry Smith   PetscInt       i;
100607b52d57SBarry Smith 
100707b52d57SBarry Smith   PetscFunctionBegin;
100805b42c5fSBarry Smith   ierr = PetscFree(*procs);CHKERRQ(ierr);
100905b42c5fSBarry Smith   ierr = PetscFree(*numprocs);CHKERRQ(ierr);
101000ff320aSBarry Smith   if (*indices) {
101105b42c5fSBarry Smith     ierr = PetscFree((*indices)[0]);CHKERRQ(ierr);
101200ff320aSBarry Smith     for (i=1; i<*nproc; i++) {
101305b42c5fSBarry Smith       ierr = PetscFree((*indices)[i]);CHKERRQ(ierr);
101407b52d57SBarry Smith     }
101507b52d57SBarry Smith     ierr = PetscFree(*indices);CHKERRQ(ierr);
101624cf384cSBarry Smith   }
101707b52d57SBarry Smith   PetscFunctionReturn(0);
101807b52d57SBarry Smith }
101986994e45SJed Brown 
102086994e45SJed Brown #undef __FUNCT__
102186994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices"
102286994e45SJed Brown /*@C
102386994e45SJed Brown    ISLocalToGlobalMappingGetIndices - Get global indices for every local point
102486994e45SJed Brown 
102586994e45SJed Brown    Not Collective
102686994e45SJed Brown 
102786994e45SJed Brown    Input Arguments:
102886994e45SJed Brown . ltog - local to global mapping
102986994e45SJed Brown 
103086994e45SJed Brown    Output Arguments:
103186994e45SJed Brown . array - array of indices
103286994e45SJed Brown 
103386994e45SJed Brown    Level: advanced
103486994e45SJed Brown 
103586994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices()
103686994e45SJed Brown @*/
10377087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
103886994e45SJed Brown {
103986994e45SJed Brown   PetscFunctionBegin;
104086994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
104186994e45SJed Brown   PetscValidPointer(array,2);
104286994e45SJed Brown   *array = ltog->indices;
104386994e45SJed Brown   PetscFunctionReturn(0);
104486994e45SJed Brown }
104586994e45SJed Brown 
104686994e45SJed Brown #undef __FUNCT__
104786994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices"
104886994e45SJed Brown /*@C
104986994e45SJed Brown    ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices()
105086994e45SJed Brown 
105186994e45SJed Brown    Not Collective
105286994e45SJed Brown 
105386994e45SJed Brown    Input Arguments:
105486994e45SJed Brown + ltog - local to global mapping
105586994e45SJed Brown - array - array of indices
105686994e45SJed Brown 
105786994e45SJed Brown    Level: advanced
105886994e45SJed Brown 
105986994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
106086994e45SJed Brown @*/
10617087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
106286994e45SJed Brown {
106386994e45SJed Brown   PetscFunctionBegin;
106486994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
106586994e45SJed Brown   PetscValidPointer(array,2);
106686994e45SJed Brown   if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
106786994e45SJed Brown   *array = PETSC_NULL;
106886994e45SJed Brown   PetscFunctionReturn(0);
106986994e45SJed Brown }
1070f7efa3c7SJed Brown 
1071f7efa3c7SJed Brown #undef __FUNCT__
1072f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate"
1073f7efa3c7SJed Brown /*@C
1074f7efa3c7SJed Brown    ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings
1075f7efa3c7SJed Brown 
1076f7efa3c7SJed Brown    Not Collective
1077f7efa3c7SJed Brown 
1078f7efa3c7SJed Brown    Input Arguments:
1079f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate
1080f7efa3c7SJed Brown . n - number of mappings to concatenate
1081f7efa3c7SJed Brown - ltogs - local to global mappings
1082f7efa3c7SJed Brown 
1083f7efa3c7SJed Brown    Output Arguments:
1084f7efa3c7SJed Brown . ltogcat - new mapping
1085f7efa3c7SJed Brown 
1086f7efa3c7SJed Brown    Level: advanced
1087f7efa3c7SJed Brown 
1088f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate()
1089f7efa3c7SJed Brown @*/
1090f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat)
1091f7efa3c7SJed Brown {
1092f7efa3c7SJed Brown   PetscInt       i,cnt,m,*idx;
1093f7efa3c7SJed Brown   PetscErrorCode ierr;
1094f7efa3c7SJed Brown 
1095f7efa3c7SJed Brown   PetscFunctionBegin;
1096f7efa3c7SJed Brown   if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n);
1097f7efa3c7SJed Brown   if (n > 0) PetscValidPointer(ltogs,3);
1098f7efa3c7SJed Brown   for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3);
1099f7efa3c7SJed Brown   PetscValidPointer(ltogcat,4);
1100f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1101f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1102f7efa3c7SJed Brown     cnt += m;
1103f7efa3c7SJed Brown   }
1104f7efa3c7SJed Brown   ierr = PetscMalloc(cnt*sizeof(PetscInt),&idx);CHKERRQ(ierr);
1105f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1106f7efa3c7SJed Brown     const PetscInt *subidx;
1107f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1108f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1109f7efa3c7SJed Brown     ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr);
1110f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1111f7efa3c7SJed Brown     cnt += m;
1112f7efa3c7SJed Brown   }
1113f7efa3c7SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr);
1114f7efa3c7SJed Brown   PetscFunctionReturn(0);
1115f7efa3c7SJed Brown }
1116