xref: /petsc/src/vec/is/utils/isltog.c (revision 665c2ded495bb9782a7454dcfef3abf1536c3670)
12362add9SBarry Smith 
2b45d2f2cSJed Brown #include <petsc-private/isimpl.h>    /*I "petscis.h"  I*/
3*665c2dedSJed Brown #include <petscviewer.h>
42362add9SBarry Smith 
57087cfbeSBarry Smith PetscClassId IS_LTOGM_CLASSID;
68e58c17dSMatthew Knepley 
74a2ae208SSatish Balay #undef __FUNCT__
8186d4ecdSBarry Smith #define __FUNCT__ "ISL2GMapApply"
9186d4ecdSBarry Smith PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping mapping,PetscInt n,const PetscInt in[],PetscInt out[])
10186d4ecdSBarry Smith {
11186d4ecdSBarry Smith   PetscErrorCode ierr;
12186d4ecdSBarry Smith   PetscInt       i,*globals = mapping->globals,start = mapping->globalstart,end = mapping->globalend;
13186d4ecdSBarry Smith 
14186d4ecdSBarry Smith   PetscFunctionBegin;
15186d4ecdSBarry Smith   if (!mapping->globals) {
16186d4ecdSBarry Smith     ierr = ISGlobalToLocalMappingApply(mapping,IS_GTOLM_MASK,0,0,0,0);CHKERRQ(ierr);
17186d4ecdSBarry Smith   }
18186d4ecdSBarry Smith   for (i=0; i<n; i++) {
19186d4ecdSBarry Smith     if (in[i] < 0)          out[i] = in[i];
20186d4ecdSBarry Smith     else if (in[i] < start) out[i] = -1;
21186d4ecdSBarry Smith     else if (in[i] > end)   out[i] = -1;
22186d4ecdSBarry Smith     else                    out[i] = globals[in[i] - start];
23186d4ecdSBarry Smith   }
24186d4ecdSBarry Smith   PetscFunctionReturn(0);
25186d4ecdSBarry Smith }
26186d4ecdSBarry Smith 
27186d4ecdSBarry Smith 
28186d4ecdSBarry Smith #undef __FUNCT__
294a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetSize"
303b9aefa3SBarry Smith /*@C
313b9aefa3SBarry Smith     ISLocalToGlobalMappingGetSize - Gets the local size of a local to global mapping.
323b9aefa3SBarry Smith 
333b9aefa3SBarry Smith     Not Collective
343b9aefa3SBarry Smith 
353b9aefa3SBarry Smith     Input Parameter:
363b9aefa3SBarry Smith .   ltog - local to global mapping
373b9aefa3SBarry Smith 
383b9aefa3SBarry Smith     Output Parameter:
393b9aefa3SBarry Smith .   n - the number of entries in the local mapping
403b9aefa3SBarry Smith 
413b9aefa3SBarry Smith     Level: advanced
423b9aefa3SBarry Smith 
43273d9f13SBarry Smith     Concepts: mapping^local to global
443b9aefa3SBarry Smith 
453b9aefa3SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
463b9aefa3SBarry Smith @*/
477087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping mapping,PetscInt *n)
483b9aefa3SBarry Smith {
493b9aefa3SBarry Smith   PetscFunctionBegin;
500700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
514482741eSBarry Smith   PetscValidIntPointer(n,2);
523b9aefa3SBarry Smith   *n = mapping->n;
533b9aefa3SBarry Smith   PetscFunctionReturn(0);
543b9aefa3SBarry Smith }
553b9aefa3SBarry Smith 
564a2ae208SSatish Balay #undef __FUNCT__
574a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingView"
585a5d4f66SBarry Smith /*@C
595a5d4f66SBarry Smith     ISLocalToGlobalMappingView - View a local to global mapping
605a5d4f66SBarry Smith 
61b9cd556bSLois Curfman McInnes     Not Collective
62b9cd556bSLois Curfman McInnes 
635a5d4f66SBarry Smith     Input Parameters:
643b9aefa3SBarry Smith +   ltog - local to global mapping
653b9aefa3SBarry Smith -   viewer - viewer
665a5d4f66SBarry Smith 
67a997ad1aSLois Curfman McInnes     Level: advanced
68a997ad1aSLois Curfman McInnes 
69273d9f13SBarry Smith     Concepts: mapping^local to global
705a5d4f66SBarry Smith 
715a5d4f66SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
725a5d4f66SBarry Smith @*/
737087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingView(ISLocalToGlobalMapping mapping,PetscViewer viewer)
745a5d4f66SBarry Smith {
7532dcc486SBarry Smith   PetscInt       i;
7632dcc486SBarry Smith   PetscMPIInt    rank;
77ace3abfcSBarry Smith   PetscBool      iascii;
786849ba73SBarry Smith   PetscErrorCode ierr;
795a5d4f66SBarry Smith 
805a5d4f66SBarry Smith   PetscFunctionBegin;
810700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
823050cee2SBarry Smith   if (!viewer) {
83ce94432eSBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mapping),&viewer);CHKERRQ(ierr);
843050cee2SBarry Smith   }
850700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
865a5d4f66SBarry Smith 
87ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)mapping),&rank);CHKERRQ(ierr);
88251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
8932077d6dSBarry Smith   if (iascii) {
907b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
915a5d4f66SBarry Smith     for (i=0; i<mapping->n; i++) {
92b0a32e0cSBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %d %d\n",rank,i,mapping->indices[i]);CHKERRQ(ierr);
936831982aSBarry Smith     }
94b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
957b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
967b23a99aSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name);
975a5d4f66SBarry Smith   PetscFunctionReturn(0);
985a5d4f66SBarry Smith }
995a5d4f66SBarry Smith 
1004a2ae208SSatish Balay #undef __FUNCT__
1014a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS"
1021f428162SBarry Smith /*@
1032bdab257SBarry Smith     ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n)
1042bdab257SBarry Smith     ordering and a global parallel ordering.
1052bdab257SBarry Smith 
1060f5bd95cSBarry Smith     Not collective
107b9cd556bSLois Curfman McInnes 
108a997ad1aSLois Curfman McInnes     Input Parameter:
1098c03b21aSDmitry Karpeev .   is - index set containing the global numbers for each local number
1102bdab257SBarry Smith 
111a997ad1aSLois Curfman McInnes     Output Parameter:
1122bdab257SBarry Smith .   mapping - new mapping data structure
1132bdab257SBarry Smith 
114a997ad1aSLois Curfman McInnes     Level: advanced
115a997ad1aSLois Curfman McInnes 
116273d9f13SBarry Smith     Concepts: mapping^local to global
1172bdab257SBarry Smith 
1182bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
1192bdab257SBarry Smith @*/
1207087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping)
1212bdab257SBarry Smith {
1226849ba73SBarry Smith   PetscErrorCode ierr;
1235d0c19d7SBarry Smith   PetscInt       n;
1245d0c19d7SBarry Smith   const PetscInt *indices;
1252bdab257SBarry Smith   MPI_Comm       comm;
1263a40ed3dSBarry Smith 
1273a40ed3dSBarry Smith   PetscFunctionBegin;
1280700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,1);
1294482741eSBarry Smith   PetscValidPointer(mapping,2);
1302bdab257SBarry Smith 
1312bdab257SBarry Smith   ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr);
1323b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
1332bdab257SBarry Smith   ierr = ISGetIndices(is,&indices);CHKERRQ(ierr);
134d5ad8652SBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
1352bdab257SBarry Smith   ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr);
1363a40ed3dSBarry Smith   PetscFunctionReturn(0);
1372bdab257SBarry Smith }
1385a5d4f66SBarry Smith 
139a4d96a55SJed Brown #undef __FUNCT__
140a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF"
141a4d96a55SJed Brown /*@C
142a4d96a55SJed Brown     ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n)
143a4d96a55SJed Brown     ordering and a global parallel ordering.
144a4d96a55SJed Brown 
145a4d96a55SJed Brown     Collective
146a4d96a55SJed Brown 
147a4d96a55SJed Brown     Input Parameter:
148a4d96a55SJed Brown +   sf - star forest mapping contiguous local indices to (rank, offset)
149a4d96a55SJed Brown -   start - first global index on this process
150a4d96a55SJed Brown 
151a4d96a55SJed Brown     Output Parameter:
152a4d96a55SJed Brown .   mapping - new mapping data structure
153a4d96a55SJed Brown 
154a4d96a55SJed Brown     Level: advanced
155a4d96a55SJed Brown 
156a4d96a55SJed Brown     Concepts: mapping^local to global
157a4d96a55SJed Brown 
158a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
159a4d96a55SJed Brown @*/
160a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping)
161a4d96a55SJed Brown {
162a4d96a55SJed Brown   PetscErrorCode ierr;
163a4d96a55SJed Brown   PetscInt       i,maxlocal,nroots,nleaves,*globals,*ltog;
164a4d96a55SJed Brown   const PetscInt *ilocal;
165a4d96a55SJed Brown   MPI_Comm       comm;
166a4d96a55SJed Brown 
167a4d96a55SJed Brown   PetscFunctionBegin;
168a4d96a55SJed Brown   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
169a4d96a55SJed Brown   PetscValidPointer(mapping,3);
170a4d96a55SJed Brown 
171a4d96a55SJed Brown   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
1720298fd71SBarry Smith   ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,NULL);CHKERRQ(ierr);
173f6e5521dSKarl Rupp   if (ilocal) {
174f6e5521dSKarl Rupp     for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1);
175f6e5521dSKarl Rupp   }
176a4d96a55SJed Brown   else maxlocal = nleaves;
177a4d96a55SJed Brown   ierr = PetscMalloc(nroots*sizeof(PetscInt),&globals);CHKERRQ(ierr);
178a4d96a55SJed Brown   ierr = PetscMalloc(maxlocal*sizeof(PetscInt),&ltog);CHKERRQ(ierr);
179a4d96a55SJed Brown   for (i=0; i<nroots; i++) globals[i] = start + i;
180a4d96a55SJed Brown   for (i=0; i<maxlocal; i++) ltog[i] = -1;
181a4d96a55SJed Brown   ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
182a4d96a55SJed Brown   ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
183a4d96a55SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
184a4d96a55SJed Brown   ierr = PetscFree(globals);CHKERRQ(ierr);
185a4d96a55SJed Brown   PetscFunctionReturn(0);
186a4d96a55SJed Brown }
187b46b645bSBarry Smith 
1884a2ae208SSatish Balay #undef __FUNCT__
1894a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate"
190ba5bb76aSSatish Balay /*@
19190f02eecSBarry Smith     ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n)
19290f02eecSBarry Smith     ordering and a global parallel ordering.
1932362add9SBarry Smith 
19489d82c54SBarry Smith     Not Collective, but communicator may have more than one process
195b9cd556bSLois Curfman McInnes 
1962362add9SBarry Smith     Input Parameters:
19789d82c54SBarry Smith +   comm - MPI communicator
19890f02eecSBarry Smith .   n - the number of local elements
1999669e4d8SBarry Smith .   indices - the global index for each local element, these do not need to be in increasing order (sorted)
200d5ad8652SBarry Smith -   mode - see PetscCopyMode
2012362add9SBarry Smith 
202a997ad1aSLois Curfman McInnes     Output Parameter:
20390f02eecSBarry Smith .   mapping - new mapping data structure
2042362add9SBarry Smith 
205a997ad1aSLois Curfman McInnes     Level: advanced
206a997ad1aSLois Curfman McInnes 
207273d9f13SBarry Smith     Concepts: mapping^local to global
2082362add9SBarry Smith 
209d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
2102362add9SBarry Smith @*/
2117087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping)
2122362add9SBarry Smith {
2136849ba73SBarry Smith   PetscErrorCode ierr;
21432dcc486SBarry Smith   PetscInt       *in;
215b46b645bSBarry Smith 
216b46b645bSBarry Smith   PetscFunctionBegin;
21773911063SBarry Smith   if (n) PetscValidIntPointer(indices,3);
2184482741eSBarry Smith   PetscValidPointer(mapping,4);
219b46b645bSBarry Smith 
2200298fd71SBarry Smith   *mapping = NULL;
221519f805aSKarl Rupp #if !defined(PETSC_USE_DYNAMIC_LIBRARIES)
2220298fd71SBarry Smith   ierr = ISInitializePackage(NULL);CHKERRQ(ierr);
2238e58c17dSMatthew Knepley #endif
2242362add9SBarry Smith 
22567c2884eSBarry Smith   ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,"ISLocalToGlobalMapping","Local to global mapping","IS",
22652e6d16bSBarry Smith                            cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr);
227d4bb536fSBarry Smith   (*mapping)->n = n;
228d4bb536fSBarry Smith   /*
229d4bb536fSBarry Smith     Do not create the global to local mapping. This is only created if
230d4bb536fSBarry Smith     ISGlobalToLocalMapping() is called
231d4bb536fSBarry Smith   */
232d4bb536fSBarry Smith   (*mapping)->globals = 0;
233d5ad8652SBarry Smith   if (mode == PETSC_COPY_VALUES) {
234d5ad8652SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&in);CHKERRQ(ierr);
235d5ad8652SBarry Smith     ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr);
236d5ad8652SBarry Smith     ierr = PetscLogObjectMemory(*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
237d5ad8652SBarry Smith     (*mapping)->indices = in;
238f6e5521dSKarl Rupp   } else if (mode == PETSC_OWN_POINTER) (*mapping)->indices = (PetscInt*)indices;
239f6e5521dSKarl Rupp   else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER");
2403a40ed3dSBarry Smith   PetscFunctionReturn(0);
2412362add9SBarry Smith }
2422362add9SBarry Smith 
2434a2ae208SSatish Balay #undef __FUNCT__
2444a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingBlock"
245bce096a4SSatish Balay /*@
246323b833fSBarry Smith     ISLocalToGlobalMappingBlock - Creates a blocked index version of an
247323b833fSBarry Smith        ISLocalToGlobalMapping that is appropriate for MatSetLocalToGlobalMappingBlock()
248323b833fSBarry Smith        and VecSetLocalToGlobalMappingBlock().
249323b833fSBarry Smith 
250323b833fSBarry Smith     Not Collective, but communicator may have more than one process
251323b833fSBarry Smith 
252323b833fSBarry Smith     Input Parameters:
253323b833fSBarry Smith +    inmap - original point-wise mapping
254323b833fSBarry Smith -    bs - block size
255323b833fSBarry Smith 
256323b833fSBarry Smith     Output Parameter:
25769eb54c3SBarry Smith .   outmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
258323b833fSBarry Smith 
259323b833fSBarry Smith     Level: advanced
260323b833fSBarry Smith 
261323b833fSBarry Smith     Concepts: mapping^local to global
262323b833fSBarry Smith 
263323b833fSBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
264323b833fSBarry Smith @*/
2657087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
266323b833fSBarry Smith {
2676849ba73SBarry Smith   PetscErrorCode ierr;
26832dcc486SBarry Smith   PetscInt       *ii,i,n;
269323b833fSBarry Smith 
270323b833fSBarry Smith   PetscFunctionBegin;
2710700a824SBarry Smith   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
272b2beed0aSJed Brown   PetscValidPointer(outmap,3);
273323b833fSBarry Smith   if (bs > 1) {
274323b833fSBarry Smith     n = inmap->n/bs;
275e32f2f54SBarry Smith     if (n*bs != inmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Pointwise mapping length is not divisible by block size");
27632dcc486SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
277f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[bs*i]/bs;
278ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
279323b833fSBarry Smith   } else {
280323b833fSBarry Smith     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
281c3122656SLisandro Dalcin     *outmap = inmap;
282323b833fSBarry Smith   }
283323b833fSBarry Smith   PetscFunctionReturn(0);
284323b833fSBarry Smith }
285323b833fSBarry Smith 
2864a2ae208SSatish Balay #undef __FUNCT__
2878ab951edSJed Brown #define __FUNCT__ "ISLocalToGlobalMappingUnBlock"
288b2beed0aSJed Brown /*@
289b2beed0aSJed Brown     ISLocalToGlobalMappingUnBlock - Creates a scalar index version of a blocked
290b2beed0aSJed Brown        ISLocalToGlobalMapping
291b2beed0aSJed Brown 
292b2beed0aSJed Brown     Not Collective, but communicator may have more than one process
293b2beed0aSJed Brown 
294b2beed0aSJed Brown     Input Parameter:
295b2beed0aSJed Brown + inmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
296b2beed0aSJed Brown - bs - block size
297b2beed0aSJed Brown 
298b2beed0aSJed Brown     Output Parameter:
299b2beed0aSJed Brown .   outmap - pointwise mapping
300b2beed0aSJed Brown 
301b2beed0aSJed Brown     Level: advanced
302b2beed0aSJed Brown 
303b2beed0aSJed Brown     Concepts: mapping^local to global
304b2beed0aSJed Brown 
305b2beed0aSJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingBlock()
306b2beed0aSJed Brown @*/
3077087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
308b2beed0aSJed Brown {
309b2beed0aSJed Brown   PetscErrorCode ierr;
310b2beed0aSJed Brown   PetscInt       *ii,i,n;
311b2beed0aSJed Brown 
312b2beed0aSJed Brown   PetscFunctionBegin;
313b2beed0aSJed Brown   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
314b2beed0aSJed Brown   PetscValidPointer(outmap,2);
315b2beed0aSJed Brown   if (bs > 1) {
316b2beed0aSJed Brown     n    = inmap->n*bs;
317b2beed0aSJed Brown     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
318f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[i/bs]*bs + (i%bs);
319ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
320b2beed0aSJed Brown   } else {
321b2beed0aSJed Brown     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
322b2beed0aSJed Brown     *outmap = inmap;
323b2beed0aSJed Brown   }
324b2beed0aSJed Brown   PetscFunctionReturn(0);
325b2beed0aSJed Brown }
326b2beed0aSJed Brown 
327b2beed0aSJed Brown #undef __FUNCT__
3284a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy"
32990f02eecSBarry Smith /*@
33090f02eecSBarry Smith    ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n)
33190f02eecSBarry Smith    ordering and a global parallel ordering.
33290f02eecSBarry Smith 
3330f5bd95cSBarry Smith    Note Collective
334b9cd556bSLois Curfman McInnes 
33590f02eecSBarry Smith    Input Parameters:
33690f02eecSBarry Smith .  mapping - mapping data structure
33790f02eecSBarry Smith 
338a997ad1aSLois Curfman McInnes    Level: advanced
339a997ad1aSLois Curfman McInnes 
3403acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate()
34190f02eecSBarry Smith @*/
3426bf464f9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping)
34390f02eecSBarry Smith {
344dfbe8321SBarry Smith   PetscErrorCode ierr;
3455fd66863SKarl Rupp 
3463a40ed3dSBarry Smith   PetscFunctionBegin;
3476bf464f9SBarry Smith   if (!*mapping) PetscFunctionReturn(0);
3486bf464f9SBarry Smith   PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1);
349997056adSBarry Smith   if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);}
3506bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->indices);CHKERRQ(ierr);
3516bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->globals);CHKERRQ(ierr);
352d38fa0fbSBarry Smith   ierr     = PetscHeaderDestroy(mapping);CHKERRQ(ierr);
353992144d0SBarry Smith   *mapping = 0;
3543a40ed3dSBarry Smith   PetscFunctionReturn(0);
35590f02eecSBarry Smith }
35690f02eecSBarry Smith 
3574a2ae208SSatish Balay #undef __FUNCT__
3584a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS"
35990f02eecSBarry Smith /*@
3603acfe500SLois Curfman McInnes     ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering
3613acfe500SLois Curfman McInnes     a new index set using the global numbering defined in an ISLocalToGlobalMapping
3623acfe500SLois Curfman McInnes     context.
36390f02eecSBarry Smith 
364b9cd556bSLois Curfman McInnes     Not collective
365b9cd556bSLois Curfman McInnes 
36690f02eecSBarry Smith     Input Parameters:
367b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
368b9cd556bSLois Curfman McInnes -   is - index set in local numbering
36990f02eecSBarry Smith 
37090f02eecSBarry Smith     Output Parameters:
37190f02eecSBarry Smith .   newis - index set in global numbering
37290f02eecSBarry Smith 
373a997ad1aSLois Curfman McInnes     Level: advanced
374a997ad1aSLois Curfman McInnes 
375273d9f13SBarry Smith     Concepts: mapping^local to global
3763acfe500SLois Curfman McInnes 
37790f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
378d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply()
37990f02eecSBarry Smith @*/
3807087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis)
38190f02eecSBarry Smith {
3826849ba73SBarry Smith   PetscErrorCode ierr;
3835d0c19d7SBarry Smith   PetscInt       n,i,*idxmap,*idxout,Nmax = mapping->n;
3845d0c19d7SBarry Smith   const PetscInt *idxin;
3853a40ed3dSBarry Smith 
3863a40ed3dSBarry Smith   PetscFunctionBegin;
3870700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
3880700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,2);
3894482741eSBarry Smith   PetscValidPointer(newis,3);
39090f02eecSBarry Smith 
3913b9aefa3SBarry Smith   ierr   = ISGetLocalSize(is,&n);CHKERRQ(ierr);
39290f02eecSBarry Smith   ierr   = ISGetIndices(is,&idxin);CHKERRQ(ierr);
39390f02eecSBarry Smith   idxmap = mapping->indices;
39490f02eecSBarry Smith 
3957c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&idxout);CHKERRQ(ierr);
39690f02eecSBarry Smith   for (i=0; i<n; i++) {
397e32f2f54SBarry 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);
39890f02eecSBarry Smith     idxout[i] = idxmap[idxin[i]];
39990f02eecSBarry Smith   }
4003b9aefa3SBarry Smith   ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr);
40170b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr);
4023a40ed3dSBarry Smith   PetscFunctionReturn(0);
40390f02eecSBarry Smith }
40490f02eecSBarry Smith 
405afcb2eb5SJed Brown #undef __FUNCT__
406afcb2eb5SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingApply"
407b89cb25eSSatish Balay /*@
4083acfe500SLois Curfman McInnes    ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering
4093acfe500SLois Curfman McInnes    and converts them to the global numbering.
41090f02eecSBarry Smith 
411b9cd556bSLois Curfman McInnes    Not collective
412b9cd556bSLois Curfman McInnes 
413bb25748dSBarry Smith    Input Parameters:
414b9cd556bSLois Curfman McInnes +  mapping - the local to global mapping context
415bb25748dSBarry Smith .  N - number of integers
416b9cd556bSLois Curfman McInnes -  in - input indices in local numbering
417bb25748dSBarry Smith 
418bb25748dSBarry Smith    Output Parameter:
419bb25748dSBarry Smith .  out - indices in global numbering
420bb25748dSBarry Smith 
421b9cd556bSLois Curfman McInnes    Notes:
422b9cd556bSLois Curfman McInnes    The in and out array parameters may be identical.
423d4bb536fSBarry Smith 
424a997ad1aSLois Curfman McInnes    Level: advanced
425a997ad1aSLois Curfman McInnes 
426bb25748dSBarry Smith .seealso: ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
4270752156aSBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
428d4bb536fSBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
429bb25748dSBarry Smith 
430273d9f13SBarry Smith     Concepts: mapping^local to global
431afcb2eb5SJed Brown @*/
432afcb2eb5SJed Brown PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[])
433afcb2eb5SJed Brown {
434afcb2eb5SJed Brown   PetscInt       i,Nmax = mapping->n;
435afcb2eb5SJed Brown   const PetscInt *idx = mapping->indices;
436d4bb536fSBarry Smith 
437afcb2eb5SJed Brown   PetscFunctionBegin;
438afcb2eb5SJed Brown   for (i=0; i<N; i++) {
439afcb2eb5SJed Brown     if (in[i] < 0) {
440afcb2eb5SJed Brown       out[i] = in[i];
441afcb2eb5SJed Brown       continue;
442afcb2eb5SJed Brown     }
443afcb2eb5SJed Brown     if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %D too large %D (max) at %D",in[i],Nmax,i);
444afcb2eb5SJed Brown     out[i] = idx[in[i]];
445afcb2eb5SJed Brown   }
446afcb2eb5SJed Brown   PetscFunctionReturn(0);
447afcb2eb5SJed Brown }
448d4bb536fSBarry Smith 
449d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/
450d4bb536fSBarry Smith 
4514a2ae208SSatish Balay #undef __FUNCT__
4524a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private"
453d4bb536fSBarry Smith /*
454d4bb536fSBarry Smith     Creates the global fields in the ISLocalToGlobalMapping structure
455d4bb536fSBarry Smith */
4566849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping)
457d4bb536fSBarry Smith {
4586849ba73SBarry Smith   PetscErrorCode ierr;
45932dcc486SBarry Smith   PetscInt       i,*idx = mapping->indices,n = mapping->n,end,start,*globals;
460d4bb536fSBarry Smith 
4613a40ed3dSBarry Smith   PetscFunctionBegin;
462d4bb536fSBarry Smith   end   = 0;
463ec268f7cSJed Brown   start = PETSC_MAX_INT;
464d4bb536fSBarry Smith 
465d4bb536fSBarry Smith   for (i=0; i<n; i++) {
466d4bb536fSBarry Smith     if (idx[i] < 0) continue;
467d4bb536fSBarry Smith     if (idx[i] < start) start = idx[i];
468d4bb536fSBarry Smith     if (idx[i] > end)   end   = idx[i];
469d4bb536fSBarry Smith   }
470d4bb536fSBarry Smith   if (start > end) {start = 0; end = -1;}
471d4bb536fSBarry Smith   mapping->globalstart = start;
472d4bb536fSBarry Smith   mapping->globalend   = end;
473d4bb536fSBarry Smith 
47432dcc486SBarry Smith   ierr             = PetscMalloc((end-start+2)*sizeof(PetscInt),&globals);CHKERRQ(ierr);
475b0a32e0cSBarry Smith   mapping->globals = globals;
476f6e5521dSKarl Rupp   for (i=0; i<end-start+1; i++) globals[i] = -1;
477d4bb536fSBarry Smith   for (i=0; i<n; i++) {
478d4bb536fSBarry Smith     if (idx[i] < 0) continue;
479d4bb536fSBarry Smith     globals[idx[i] - start] = i;
480d4bb536fSBarry Smith   }
481d4bb536fSBarry Smith 
48252e6d16bSBarry Smith   ierr = PetscLogObjectMemory(mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr);
4833a40ed3dSBarry Smith   PetscFunctionReturn(0);
484d4bb536fSBarry Smith }
485d4bb536fSBarry Smith 
4864a2ae208SSatish Balay #undef __FUNCT__
4874a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply"
488d4bb536fSBarry Smith /*@
489a997ad1aSLois Curfman McInnes     ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers
490a997ad1aSLois Curfman McInnes     specified with a global numbering.
491d4bb536fSBarry Smith 
492b9cd556bSLois Curfman McInnes     Not collective
493b9cd556bSLois Curfman McInnes 
494d4bb536fSBarry Smith     Input Parameters:
495b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
496d4bb536fSBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
497d4bb536fSBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
498d4bb536fSBarry Smith .   n - number of global indices to map
499b9cd556bSLois Curfman McInnes -   idx - global indices to map
500d4bb536fSBarry Smith 
501d4bb536fSBarry Smith     Output Parameters:
502b9cd556bSLois Curfman McInnes +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
503b9cd556bSLois Curfman McInnes -   idxout - local index of each global index, one must pass in an array long enough
504e182c471SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApply() with
5050298fd71SBarry Smith              idxout == NULL to determine the required length (returned in nout)
506e182c471SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApply()
507e182c471SBarry Smith              a second time to set the values.
508d4bb536fSBarry Smith 
509b9cd556bSLois Curfman McInnes     Notes:
5100298fd71SBarry Smith     Either nout or idxout may be NULL. idx and idxout may be identical.
511d4bb536fSBarry Smith 
5120f5bd95cSBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
5130f5bd95cSBarry Smith     array to compute these.
5140f5bd95cSBarry Smith 
515a997ad1aSLois Curfman McInnes     Level: advanced
516a997ad1aSLois Curfman McInnes 
51732fd6b96SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
51832fd6b96SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
51932fd6b96SBarry Smith 
520273d9f13SBarry Smith     Concepts: mapping^global to local
521d4bb536fSBarry Smith 
522d4bb536fSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
523d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy()
524d4bb536fSBarry Smith @*/
5257087cfbeSBarry Smith PetscErrorCode  ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
52632dcc486SBarry Smith                                   PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
527d4bb536fSBarry Smith {
52832dcc486SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end;
5296849ba73SBarry Smith   PetscErrorCode ierr;
530d4bb536fSBarry Smith 
5313a40ed3dSBarry Smith   PetscFunctionBegin;
5320700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
533d4bb536fSBarry Smith   if (!mapping->globals) {
534d4bb536fSBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
535d4bb536fSBarry Smith   }
536d4bb536fSBarry Smith   globals = mapping->globals;
537d4bb536fSBarry Smith   start   = mapping->globalstart;
538d4bb536fSBarry Smith   end     = mapping->globalend;
539d4bb536fSBarry Smith 
540d4bb536fSBarry Smith   if (type == IS_GTOLM_MASK) {
541d4bb536fSBarry Smith     if (idxout) {
542d4bb536fSBarry Smith       for (i=0; i<n; i++) {
543d4bb536fSBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
544d4bb536fSBarry Smith         else if (idx[i] < start) idxout[i] = -1;
545d4bb536fSBarry Smith         else if (idx[i] > end)   idxout[i] = -1;
546d4bb536fSBarry Smith         else                     idxout[i] = globals[idx[i] - start];
547d4bb536fSBarry Smith       }
548d4bb536fSBarry Smith     }
549d4bb536fSBarry Smith     if (nout) *nout = n;
550d4bb536fSBarry Smith   } else {
551d4bb536fSBarry Smith     if (idxout) {
552d4bb536fSBarry Smith       for (i=0; i<n; i++) {
553d4bb536fSBarry Smith         if (idx[i] < 0) continue;
554d4bb536fSBarry Smith         if (idx[i] < start) continue;
555d4bb536fSBarry Smith         if (idx[i] > end) continue;
556d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
557d4bb536fSBarry Smith         if (tmp < 0) continue;
558d4bb536fSBarry Smith         idxout[nf++] = tmp;
559d4bb536fSBarry Smith       }
560d4bb536fSBarry Smith     } else {
561d4bb536fSBarry Smith       for (i=0; i<n; i++) {
562d4bb536fSBarry Smith         if (idx[i] < 0) continue;
563d4bb536fSBarry Smith         if (idx[i] < start) continue;
564d4bb536fSBarry Smith         if (idx[i] > end) continue;
565d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
566d4bb536fSBarry Smith         if (tmp < 0) continue;
567d4bb536fSBarry Smith         nf++;
568d4bb536fSBarry Smith       }
569d4bb536fSBarry Smith     }
570d4bb536fSBarry Smith     if (nout) *nout = nf;
571d4bb536fSBarry Smith   }
5723a40ed3dSBarry Smith   PetscFunctionReturn(0);
573d4bb536fSBarry Smith }
57490f02eecSBarry Smith 
5754a2ae208SSatish Balay #undef __FUNCT__
5764a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetInfo"
57789d82c54SBarry Smith /*@C
57889d82c54SBarry Smith     ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and
57989d82c54SBarry Smith      each index shared by more than one processor
58089d82c54SBarry Smith 
58189d82c54SBarry Smith     Collective on ISLocalToGlobalMapping
58289d82c54SBarry Smith 
58389d82c54SBarry Smith     Input Parameters:
58489d82c54SBarry Smith .   mapping - the mapping from local to global indexing
58589d82c54SBarry Smith 
58689d82c54SBarry Smith     Output Parameter:
58789d82c54SBarry Smith +   nproc - number of processors that are connected to this one
58889d82c54SBarry Smith .   proc - neighboring processors
58907b52d57SBarry Smith .   numproc - number of indices for each subdomain (processor)
5903463a7baSJed Brown -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
59189d82c54SBarry Smith 
59289d82c54SBarry Smith     Level: advanced
59389d82c54SBarry Smith 
594273d9f13SBarry Smith     Concepts: mapping^local to global
59589d82c54SBarry Smith 
5962cfcea29SBarry Smith     Fortran Usage:
5972cfcea29SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
5982cfcea29SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
5992cfcea29SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
6002cfcea29SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
6012cfcea29SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
6022cfcea29SBarry Smith 
6032cfcea29SBarry Smith 
60407b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
60507b52d57SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
60689d82c54SBarry Smith @*/
6077087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
60889d82c54SBarry Smith {
6096849ba73SBarry Smith   PetscErrorCode ierr;
61097f1f81fSBarry Smith   PetscMPIInt    size,rank,tag1,tag2,tag3,*len,*source,imdex;
61132dcc486SBarry Smith   PetscInt       i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices;
61232dcc486SBarry Smith   PetscInt       *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc;
61397f1f81fSBarry Smith   PetscInt       cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned;
61432dcc486SBarry Smith   PetscInt       node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp;
61532dcc486SBarry Smith   PetscInt       first_procs,first_numprocs,*first_indices;
61689d82c54SBarry Smith   MPI_Request    *recv_waits,*send_waits;
61730dcb7c9SBarry Smith   MPI_Status     recv_status,*send_status,*recv_statuses;
618ce94432eSBarry Smith   MPI_Comm       comm;
619ace3abfcSBarry Smith   PetscBool      debug = PETSC_FALSE;
62089d82c54SBarry Smith 
62189d82c54SBarry Smith   PetscFunctionBegin;
6220700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
623ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)mapping,&comm);CHKERRQ(ierr);
62424cf384cSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
62524cf384cSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
62624cf384cSBarry Smith   if (size == 1) {
62724cf384cSBarry Smith     *nproc         = 0;
6280298fd71SBarry Smith     *procs         = NULL;
62932dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr);
6301e2105dcSBarry Smith     (*numprocs)[0] = 0;
63132dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr);
6320298fd71SBarry Smith     (*indices)[0]  = NULL;
63324cf384cSBarry Smith     PetscFunctionReturn(0);
63424cf384cSBarry Smith   }
63524cf384cSBarry Smith 
6360298fd71SBarry Smith   ierr = PetscOptionsGetBool(NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,NULL);CHKERRQ(ierr);
63707b52d57SBarry Smith 
6383677ff5aSBarry Smith   /*
6393677ff5aSBarry Smith     Notes on ISLocalToGlobalMappingGetInfo
6403677ff5aSBarry Smith 
6413677ff5aSBarry Smith     globally owned node - the nodes that have been assigned to this processor in global
6423677ff5aSBarry Smith            numbering, just for this routine.
6433677ff5aSBarry Smith 
6443677ff5aSBarry Smith     nontrivial globally owned node - node assigned to this processor that is on a subdomain
6453677ff5aSBarry Smith            boundary (i.e. is has more than one local owner)
6463677ff5aSBarry Smith 
6473677ff5aSBarry Smith     locally owned node - node that exists on this processors subdomain
6483677ff5aSBarry Smith 
6493677ff5aSBarry Smith     nontrivial locally owned node - node that is not in the interior (i.e. has more than one
6503677ff5aSBarry Smith            local subdomain
6513677ff5aSBarry Smith   */
65224cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr);
65324cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr);
65424cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr);
65589d82c54SBarry Smith 
65689d82c54SBarry Smith   for (i=0; i<n; i++) {
65789d82c54SBarry Smith     if (lindices[i] > max) max = lindices[i];
65889d82c54SBarry Smith   }
65932dcc486SBarry Smith   ierr   = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
66078058e43SBarry Smith   Ng++;
66189d82c54SBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
66289d82c54SBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
663bc8ff85bSBarry Smith   scale  = Ng/size + 1;
664a2e34c3dSBarry Smith   ng     = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng);
665caba0dd0SBarry Smith   rstart = scale*rank;
66689d82c54SBarry Smith 
66789d82c54SBarry Smith   /* determine ownership ranges of global indices */
6687c334f02SBarry Smith   ierr = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
66932dcc486SBarry Smith   ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
67089d82c54SBarry Smith 
67189d82c54SBarry Smith   /* determine owners of each local node  */
6727c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&owner);CHKERRQ(ierr);
67389d82c54SBarry Smith   for (i=0; i<n; i++) {
6743677ff5aSBarry Smith     proc             = lindices[i]/scale; /* processor that globally owns this index */
67527c402fcSBarry Smith     nprocs[2*proc+1] = 1;                 /* processor globally owns at least one of ours */
6763677ff5aSBarry Smith     owner[i]         = proc;
67727c402fcSBarry Smith     nprocs[2*proc]++;                     /* count of how many that processor globally owns of ours */
67889d82c54SBarry Smith   }
67927c402fcSBarry Smith   nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1];
6801e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of global owners for my local data %d\n",nsends);CHKERRQ(ierr);
68189d82c54SBarry Smith 
68289d82c54SBarry Smith   /* inform other processors of number of messages and max length*/
68327c402fcSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
6841e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of local owners for my global data %d\n",nrecvs);CHKERRQ(ierr);
68589d82c54SBarry Smith 
68689d82c54SBarry Smith   /* post receives for owned rows */
68732dcc486SBarry Smith   ierr = PetscMalloc((2*nrecvs+1)*(nmax+1)*sizeof(PetscInt),&recvs);CHKERRQ(ierr);
688b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
68989d82c54SBarry Smith   for (i=0; i<nrecvs; i++) {
69032dcc486SBarry Smith     ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr);
69189d82c54SBarry Smith   }
69289d82c54SBarry Smith 
69389d82c54SBarry Smith   /* pack messages containing lists of local nodes to owners */
69432dcc486SBarry Smith   ierr      = PetscMalloc((2*n+1)*sizeof(PetscInt),&sends);CHKERRQ(ierr);
69532dcc486SBarry Smith   ierr      = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
69689d82c54SBarry Smith   starts[0] = 0;
697f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
69889d82c54SBarry Smith   for (i=0; i<n; i++) {
69989d82c54SBarry Smith     sends[starts[owner[i]]++] = lindices[i];
70030dcb7c9SBarry Smith     sends[starts[owner[i]]++] = i;
70189d82c54SBarry Smith   }
70289d82c54SBarry Smith   ierr = PetscFree(owner);CHKERRQ(ierr);
70389d82c54SBarry Smith   starts[0] = 0;
704f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
70589d82c54SBarry Smith 
70689d82c54SBarry Smith   /* send the messages */
707b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
70832dcc486SBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(PetscInt),&dest);CHKERRQ(ierr);
70989d82c54SBarry Smith   cnt = 0;
71089d82c54SBarry Smith   for (i=0; i<size; i++) {
71127c402fcSBarry Smith     if (nprocs[2*i]) {
71232dcc486SBarry Smith       ierr      = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr);
71330dcb7c9SBarry Smith       dest[cnt] = i;
71489d82c54SBarry Smith       cnt++;
71589d82c54SBarry Smith     }
71689d82c54SBarry Smith   }
71789d82c54SBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
71889d82c54SBarry Smith 
71989d82c54SBarry Smith   /* wait on receives */
72097f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&source);CHKERRQ(ierr);
72197f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&len);CHKERRQ(ierr);
72289d82c54SBarry Smith   cnt  = nrecvs;
72332dcc486SBarry Smith   ierr = PetscMalloc((ng+1)*sizeof(PetscInt),&nownedsenders);CHKERRQ(ierr);
72432dcc486SBarry Smith   ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr);
72589d82c54SBarry Smith   while (cnt) {
72689d82c54SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
72789d82c54SBarry Smith     /* unpack receives into our local space */
72832dcc486SBarry Smith     ierr          = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr);
72989d82c54SBarry Smith     source[imdex] = recv_status.MPI_SOURCE;
73030dcb7c9SBarry Smith     len[imdex]    = len[imdex]/2;
731caba0dd0SBarry Smith     /* count how many local owners for each of my global owned indices */
73230dcb7c9SBarry Smith     for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++;
73389d82c54SBarry Smith     cnt--;
73489d82c54SBarry Smith   }
73589d82c54SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
73689d82c54SBarry Smith 
73730dcb7c9SBarry Smith   /* count how many globally owned indices are on an edge multiplied by how many processors own them. */
738bc8ff85bSBarry Smith   nowned  = 0;
739bc8ff85bSBarry Smith   nownedm = 0;
740bc8ff85bSBarry Smith   for (i=0; i<ng; i++) {
741bc8ff85bSBarry Smith     if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;}
742bc8ff85bSBarry Smith   }
743bc8ff85bSBarry Smith 
744bc8ff85bSBarry Smith   /* create single array to contain rank of all local owners of each globally owned index */
74532dcc486SBarry Smith   ierr      = PetscMalloc((nownedm+1)*sizeof(PetscInt),&ownedsenders);CHKERRQ(ierr);
74632dcc486SBarry Smith   ierr      = PetscMalloc((ng+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
747bc8ff85bSBarry Smith   starts[0] = 0;
748bc8ff85bSBarry Smith   for (i=1; i<ng; i++) {
749bc8ff85bSBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
750bc8ff85bSBarry Smith     else starts[i] = starts[i-1];
751bc8ff85bSBarry Smith   }
752bc8ff85bSBarry Smith 
75330dcb7c9SBarry Smith   /* for each nontrival globally owned node list all arriving processors */
754bc8ff85bSBarry Smith   for (i=0; i<nrecvs; i++) {
755bc8ff85bSBarry Smith     for (j=0; j<len[i]; j++) {
75630dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
757f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) ownedsenders[starts[node]++] = source[i];
758bc8ff85bSBarry Smith     }
759bc8ff85bSBarry Smith   }
760bc8ff85bSBarry Smith 
76107b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
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     for (i=0; i<ng; i++) {
76830dcb7c9SBarry Smith       if (nownedsenders[i] > 1) {
76930dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] global node %d local owner processors: ",rank,i+rstart);CHKERRQ(ierr);
77030dcb7c9SBarry Smith         for (j=0; j<nownedsenders[i]; j++) {
77130dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",ownedsenders[starts[i]+j]);CHKERRQ(ierr);
77230dcb7c9SBarry Smith         }
77330dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
77430dcb7c9SBarry Smith       }
77530dcb7c9SBarry Smith     }
77630dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
77707b52d57SBarry Smith   } /* -----------------------------------  */
77830dcb7c9SBarry Smith 
7793677ff5aSBarry Smith   /* wait on original sends */
7803a96401aSBarry Smith   if (nsends) {
781b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
7823a96401aSBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
7833a96401aSBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
7843a96401aSBarry Smith   }
78589d82c54SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
7863a96401aSBarry Smith   ierr = PetscFree(sends);CHKERRQ(ierr);
7873677ff5aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
7883677ff5aSBarry Smith 
7893677ff5aSBarry Smith   /* pack messages to send back to local owners */
79030dcb7c9SBarry Smith   starts[0] = 0;
79130dcb7c9SBarry Smith   for (i=1; i<ng; i++) {
79230dcb7c9SBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
79330dcb7c9SBarry Smith     else starts[i] = starts[i-1];
79430dcb7c9SBarry Smith   }
79530dcb7c9SBarry Smith   nsends2 = nrecvs;
79632dcc486SBarry Smith   ierr    = PetscMalloc((nsends2+1)*sizeof(PetscInt),&nprocs);CHKERRQ(ierr); /* length of each message */
79730dcb7c9SBarry Smith   for (i=0; i<nrecvs; i++) {
79830dcb7c9SBarry Smith     nprocs[i] = 1;
79930dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
80030dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
801f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) nprocs[i] += 2 + nownedsenders[node];
80230dcb7c9SBarry Smith     }
80330dcb7c9SBarry Smith   }
804f6e5521dSKarl Rupp   nt = 0;
805f6e5521dSKarl Rupp   for (i=0; i<nsends2; i++) nt += nprocs[i];
806f6e5521dSKarl Rupp 
80732dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&sends2);CHKERRQ(ierr);
80832dcc486SBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(PetscInt),&starts2);CHKERRQ(ierr);
809f6e5521dSKarl Rupp 
810f6e5521dSKarl Rupp   starts2[0] = 0;
811f6e5521dSKarl Rupp   for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1];
81230dcb7c9SBarry Smith   /*
81330dcb7c9SBarry Smith      Each message is 1 + nprocs[i] long, and consists of
81430dcb7c9SBarry Smith        (0) the number of nodes being sent back
81530dcb7c9SBarry Smith        (1) the local node number,
81630dcb7c9SBarry Smith        (2) the number of processors sharing it,
81730dcb7c9SBarry Smith        (3) the processors sharing it
81830dcb7c9SBarry Smith   */
81930dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
82030dcb7c9SBarry Smith     cnt = 1;
82130dcb7c9SBarry Smith     sends2[starts2[i]] = 0;
82230dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
82330dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
82430dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
82530dcb7c9SBarry Smith         sends2[starts2[i]]++;
82630dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1];
82730dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = nownedsenders[node];
82832dcc486SBarry Smith         ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr);
82930dcb7c9SBarry Smith         cnt += nownedsenders[node];
83030dcb7c9SBarry Smith       }
83130dcb7c9SBarry Smith     }
83230dcb7c9SBarry Smith   }
83330dcb7c9SBarry Smith 
83430dcb7c9SBarry Smith   /* receive the message lengths */
83530dcb7c9SBarry Smith   nrecvs2 = nsends;
83632dcc486SBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&lens2);CHKERRQ(ierr);
83732dcc486SBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&starts3);CHKERRQ(ierr);
838d44834fbSBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
83930dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
840d44834fbSBarry Smith     ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr);
84130dcb7c9SBarry Smith   }
842d44834fbSBarry Smith 
8438a8e0b3aSBarry Smith   /* send the message lengths */
8448a8e0b3aSBarry Smith   for (i=0; i<nsends2; i++) {
8458a8e0b3aSBarry Smith     ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr);
8468a8e0b3aSBarry Smith   }
8478a8e0b3aSBarry Smith 
848d44834fbSBarry Smith   /* wait on receives of lens */
8490c468ba9SBarry Smith   if (nrecvs2) {
8500c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
851d44834fbSBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
852d44834fbSBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8530c468ba9SBarry Smith   }
854a2ea699eSBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
855d44834fbSBarry Smith 
85630dcb7c9SBarry Smith   starts3[0] = 0;
857d44834fbSBarry Smith   nt         = 0;
85830dcb7c9SBarry Smith   for (i=0; i<nrecvs2-1; i++) {
85930dcb7c9SBarry Smith     starts3[i+1] = starts3[i] + lens2[i];
860d44834fbSBarry Smith     nt          += lens2[i];
86130dcb7c9SBarry Smith   }
86276466f69SStefano Zampini   if (nrecvs2) nt += lens2[nrecvs2-1];
863d44834fbSBarry Smith 
86432dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&recvs2);CHKERRQ(ierr);
865b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
86652b72c4aSBarry Smith   for (i=0; i<nrecvs2; i++) {
86732dcc486SBarry Smith     ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr);
86830dcb7c9SBarry Smith   }
86930dcb7c9SBarry Smith 
87030dcb7c9SBarry Smith   /* send the messages */
871b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
87230dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
87332dcc486SBarry Smith     ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr);
87430dcb7c9SBarry Smith   }
87530dcb7c9SBarry Smith 
87630dcb7c9SBarry Smith   /* wait on receives */
8770c468ba9SBarry Smith   if (nrecvs2) {
8780c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
87930dcb7c9SBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
88030dcb7c9SBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8810c468ba9SBarry Smith   }
88230dcb7c9SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
88330dcb7c9SBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
88430dcb7c9SBarry Smith 
88507b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
88630dcb7c9SBarry Smith     cnt = 0;
88730dcb7c9SBarry Smith     for (i=0; i<nrecvs2; i++) {
88830dcb7c9SBarry Smith       nt = recvs2[cnt++];
88930dcb7c9SBarry Smith       for (j=0; j<nt; j++) {
89030dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] local node %d number of subdomains %d: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr);
89130dcb7c9SBarry Smith         for (k=0; k<recvs2[cnt+1]; k++) {
89230dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",recvs2[cnt+2+k]);CHKERRQ(ierr);
89330dcb7c9SBarry Smith         }
89430dcb7c9SBarry Smith         cnt += 2 + recvs2[cnt+1];
89530dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
89630dcb7c9SBarry Smith       }
89730dcb7c9SBarry Smith     }
89830dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
89907b52d57SBarry Smith   } /* -----------------------------------  */
90030dcb7c9SBarry Smith 
90130dcb7c9SBarry Smith   /* count number subdomains for each local node */
90232dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
90332dcc486SBarry Smith   ierr = PetscMemzero(nprocs,size*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++) {
908f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) nprocs[recvs2[cnt+2+k]]++;
90930dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
91030dcb7c9SBarry Smith     }
91130dcb7c9SBarry Smith   }
91230dcb7c9SBarry Smith   nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0);
91330dcb7c9SBarry Smith   *nproc    = nt;
91432dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),procs);CHKERRQ(ierr);
91532dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),numprocs);CHKERRQ(ierr);
91632dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt*),indices);CHKERRQ(ierr);
9170298fd71SBarry Smith   for (i=0;i<nt+1;i++) (*indices)[i]=NULL;
91832dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&bprocs);CHKERRQ(ierr);
91930dcb7c9SBarry Smith   cnt       = 0;
92030dcb7c9SBarry Smith   for (i=0; i<size; i++) {
92130dcb7c9SBarry Smith     if (nprocs[i] > 0) {
92230dcb7c9SBarry Smith       bprocs[i]        = cnt;
92330dcb7c9SBarry Smith       (*procs)[cnt]    = i;
92430dcb7c9SBarry Smith       (*numprocs)[cnt] = nprocs[i];
92532dcc486SBarry Smith       ierr             = PetscMalloc(nprocs[i]*sizeof(PetscInt),&(*indices)[cnt]);CHKERRQ(ierr);
92630dcb7c9SBarry Smith       cnt++;
92730dcb7c9SBarry Smith     }
92830dcb7c9SBarry Smith   }
92930dcb7c9SBarry Smith 
93030dcb7c9SBarry Smith   /* make the list of subdomains for each nontrivial local node */
93132dcc486SBarry Smith   ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr);
93230dcb7c9SBarry Smith   cnt  = 0;
93330dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
93430dcb7c9SBarry Smith     nt = recvs2[cnt++];
93530dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
936f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt];
93730dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
93830dcb7c9SBarry Smith     }
93930dcb7c9SBarry Smith   }
94030dcb7c9SBarry Smith   ierr = PetscFree(bprocs);CHKERRQ(ierr);
94107b52d57SBarry Smith   ierr = PetscFree(recvs2);CHKERRQ(ierr);
94230dcb7c9SBarry Smith 
94307b52d57SBarry Smith   /* sort the node indexing by their global numbers */
94407b52d57SBarry Smith   nt = *nproc;
94507b52d57SBarry Smith   for (i=0; i<nt; i++) {
94632dcc486SBarry Smith     ierr = PetscMalloc(((*numprocs)[i])*sizeof(PetscInt),&tmp);CHKERRQ(ierr);
947f6e5521dSKarl Rupp     for (j=0; j<(*numprocs)[i]; j++) tmp[j] = lindices[(*indices)[i][j]];
94807b52d57SBarry Smith     ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr);
94907b52d57SBarry Smith     ierr = PetscFree(tmp);CHKERRQ(ierr);
95007b52d57SBarry Smith   }
95107b52d57SBarry Smith 
95207b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
95330dcb7c9SBarry Smith     nt = *nproc;
95430dcb7c9SBarry Smith     for (i=0; i<nt; i++) {
95530dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %d number of indices %d: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr);
95630dcb7c9SBarry Smith       for (j=0; j<(*numprocs)[i]; j++) {
95730dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"%d ",(*indices)[i][j]);CHKERRQ(ierr);
95830dcb7c9SBarry Smith       }
95930dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
96030dcb7c9SBarry Smith     }
96130dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
96207b52d57SBarry Smith   } /* -----------------------------------  */
96330dcb7c9SBarry Smith 
96430dcb7c9SBarry Smith   /* wait on sends */
96530dcb7c9SBarry Smith   if (nsends2) {
966b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends2*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
96730dcb7c9SBarry Smith     ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr);
96830dcb7c9SBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
96930dcb7c9SBarry Smith   }
97030dcb7c9SBarry Smith 
97130dcb7c9SBarry Smith   ierr = PetscFree(starts3);CHKERRQ(ierr);
97230dcb7c9SBarry Smith   ierr = PetscFree(dest);CHKERRQ(ierr);
97330dcb7c9SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
9743677ff5aSBarry Smith 
975bc8ff85bSBarry Smith   ierr = PetscFree(nownedsenders);CHKERRQ(ierr);
976bc8ff85bSBarry Smith   ierr = PetscFree(ownedsenders);CHKERRQ(ierr);
977bc8ff85bSBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
97830dcb7c9SBarry Smith   ierr = PetscFree(starts2);CHKERRQ(ierr);
97930dcb7c9SBarry Smith   ierr = PetscFree(lens2);CHKERRQ(ierr);
98089d82c54SBarry Smith 
98189d82c54SBarry Smith   ierr = PetscFree(source);CHKERRQ(ierr);
98297f1f81fSBarry Smith   ierr = PetscFree(len);CHKERRQ(ierr);
98389d82c54SBarry Smith   ierr = PetscFree(recvs);CHKERRQ(ierr);
9843a96401aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
98530dcb7c9SBarry Smith   ierr = PetscFree(sends2);CHKERRQ(ierr);
98624cf384cSBarry Smith 
98724cf384cSBarry Smith   /* put the information about myself as the first entry in the list */
98824cf384cSBarry Smith   first_procs    = (*procs)[0];
98924cf384cSBarry Smith   first_numprocs = (*numprocs)[0];
99024cf384cSBarry Smith   first_indices  = (*indices)[0];
99124cf384cSBarry Smith   for (i=0; i<*nproc; i++) {
99224cf384cSBarry Smith     if ((*procs)[i] == rank) {
99324cf384cSBarry Smith       (*procs)[0]    = (*procs)[i];
99424cf384cSBarry Smith       (*numprocs)[0] = (*numprocs)[i];
99524cf384cSBarry Smith       (*indices)[0]  = (*indices)[i];
99624cf384cSBarry Smith       (*procs)[i]    = first_procs;
99724cf384cSBarry Smith       (*numprocs)[i] = first_numprocs;
99824cf384cSBarry Smith       (*indices)[i]  = first_indices;
99924cf384cSBarry Smith       break;
100024cf384cSBarry Smith     }
100124cf384cSBarry Smith   }
100289d82c54SBarry Smith   PetscFunctionReturn(0);
100389d82c54SBarry Smith }
100489d82c54SBarry Smith 
10054a2ae208SSatish Balay #undef __FUNCT__
10064a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo"
100707b52d57SBarry Smith /*@C
100807b52d57SBarry Smith     ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo()
100989d82c54SBarry Smith 
101007b52d57SBarry Smith     Collective on ISLocalToGlobalMapping
101107b52d57SBarry Smith 
101207b52d57SBarry Smith     Input Parameters:
101307b52d57SBarry Smith .   mapping - the mapping from local to global indexing
101407b52d57SBarry Smith 
101507b52d57SBarry Smith     Output Parameter:
101607b52d57SBarry Smith +   nproc - number of processors that are connected to this one
101707b52d57SBarry Smith .   proc - neighboring processors
101807b52d57SBarry Smith .   numproc - number of indices for each processor
101907b52d57SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
102007b52d57SBarry Smith 
102107b52d57SBarry Smith     Level: advanced
102207b52d57SBarry Smith 
102307b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
102407b52d57SBarry Smith           ISLocalToGlobalMappingGetInfo()
102507b52d57SBarry Smith @*/
10267087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
102707b52d57SBarry Smith {
10286849ba73SBarry Smith   PetscErrorCode ierr;
102932dcc486SBarry Smith   PetscInt       i;
103007b52d57SBarry Smith 
103107b52d57SBarry Smith   PetscFunctionBegin;
103205b42c5fSBarry Smith   ierr = PetscFree(*procs);CHKERRQ(ierr);
103305b42c5fSBarry Smith   ierr = PetscFree(*numprocs);CHKERRQ(ierr);
103400ff320aSBarry Smith   if (*indices) {
103505b42c5fSBarry Smith     ierr = PetscFree((*indices)[0]);CHKERRQ(ierr);
103600ff320aSBarry Smith     for (i=1; i<*nproc; i++) {
103705b42c5fSBarry Smith       ierr = PetscFree((*indices)[i]);CHKERRQ(ierr);
103807b52d57SBarry Smith     }
103907b52d57SBarry Smith     ierr = PetscFree(*indices);CHKERRQ(ierr);
104024cf384cSBarry Smith   }
104107b52d57SBarry Smith   PetscFunctionReturn(0);
104207b52d57SBarry Smith }
104386994e45SJed Brown 
104486994e45SJed Brown #undef __FUNCT__
104586994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices"
104686994e45SJed Brown /*@C
104786994e45SJed Brown    ISLocalToGlobalMappingGetIndices - Get global indices for every local point
104886994e45SJed Brown 
104986994e45SJed Brown    Not Collective
105086994e45SJed Brown 
105186994e45SJed Brown    Input Arguments:
105286994e45SJed Brown . ltog - local to global mapping
105386994e45SJed Brown 
105486994e45SJed Brown    Output Arguments:
105586994e45SJed Brown . array - array of indices
105686994e45SJed Brown 
105786994e45SJed Brown    Level: advanced
105886994e45SJed Brown 
105986994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices()
106086994e45SJed Brown @*/
10617087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetIndices(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   *array = ltog->indices;
106786994e45SJed Brown   PetscFunctionReturn(0);
106886994e45SJed Brown }
106986994e45SJed Brown 
107086994e45SJed Brown #undef __FUNCT__
107186994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices"
107286994e45SJed Brown /*@C
107386994e45SJed Brown    ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices()
107486994e45SJed Brown 
107586994e45SJed Brown    Not Collective
107686994e45SJed Brown 
107786994e45SJed Brown    Input Arguments:
107886994e45SJed Brown + ltog - local to global mapping
107986994e45SJed Brown - array - array of indices
108086994e45SJed Brown 
108186994e45SJed Brown    Level: advanced
108286994e45SJed Brown 
108386994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
108486994e45SJed Brown @*/
10857087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
108686994e45SJed Brown {
108786994e45SJed Brown   PetscFunctionBegin;
108886994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
108986994e45SJed Brown   PetscValidPointer(array,2);
109086994e45SJed Brown   if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
10910298fd71SBarry Smith   *array = NULL;
109286994e45SJed Brown   PetscFunctionReturn(0);
109386994e45SJed Brown }
1094f7efa3c7SJed Brown 
1095f7efa3c7SJed Brown #undef __FUNCT__
1096f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate"
1097f7efa3c7SJed Brown /*@C
1098f7efa3c7SJed Brown    ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings
1099f7efa3c7SJed Brown 
1100f7efa3c7SJed Brown    Not Collective
1101f7efa3c7SJed Brown 
1102f7efa3c7SJed Brown    Input Arguments:
1103f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate
1104f7efa3c7SJed Brown . n - number of mappings to concatenate
1105f7efa3c7SJed Brown - ltogs - local to global mappings
1106f7efa3c7SJed Brown 
1107f7efa3c7SJed Brown    Output Arguments:
1108f7efa3c7SJed Brown . ltogcat - new mapping
1109f7efa3c7SJed Brown 
1110f7efa3c7SJed Brown    Level: advanced
1111f7efa3c7SJed Brown 
1112f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate()
1113f7efa3c7SJed Brown @*/
1114f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat)
1115f7efa3c7SJed Brown {
1116f7efa3c7SJed Brown   PetscInt       i,cnt,m,*idx;
1117f7efa3c7SJed Brown   PetscErrorCode ierr;
1118f7efa3c7SJed Brown 
1119f7efa3c7SJed Brown   PetscFunctionBegin;
1120f7efa3c7SJed Brown   if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n);
1121f7efa3c7SJed Brown   if (n > 0) PetscValidPointer(ltogs,3);
1122f7efa3c7SJed Brown   for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3);
1123f7efa3c7SJed Brown   PetscValidPointer(ltogcat,4);
1124f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1125f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1126f7efa3c7SJed Brown     cnt += m;
1127f7efa3c7SJed Brown   }
1128f7efa3c7SJed Brown   ierr = PetscMalloc(cnt*sizeof(PetscInt),&idx);CHKERRQ(ierr);
1129f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1130f7efa3c7SJed Brown     const PetscInt *subidx;
1131f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1132f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1133f7efa3c7SJed Brown     ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr);
1134f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1135f7efa3c7SJed Brown     cnt += m;
1136f7efa3c7SJed Brown   }
1137f7efa3c7SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr);
1138f7efa3c7SJed Brown   PetscFunctionReturn(0);
1139f7efa3c7SJed Brown }
1140