xref: /petsc/src/vec/is/utils/isltog.c (revision 543f30987b8111c12be6c48a4b9ae9791d966951)
12362add9SBarry Smith 
2b45d2f2cSJed Brown #include <petsc-private/isimpl.h>    /*I "petscis.h"  I*/
30c312b8eSJed Brown #include <petscsf.h>
4665c2dedSJed Brown #include <petscviewer.h>
52362add9SBarry Smith 
67087cfbeSBarry Smith PetscClassId IS_LTOGM_CLASSID;
78e58c17dSMatthew Knepley 
804a59952SBarry Smith 
94a2ae208SSatish Balay #undef __FUNCT__
10743c9b42SStefano Zampini #define __FUNCT__ "ISG2LMapApply"
11186d4ecdSBarry Smith PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping mapping,PetscInt n,const PetscInt in[],PetscInt out[])
12186d4ecdSBarry Smith {
13186d4ecdSBarry Smith   PetscErrorCode ierr;
14743c9b42SStefano Zampini   PetscInt       i,start,end;
15186d4ecdSBarry Smith 
16186d4ecdSBarry Smith   PetscFunctionBegin;
17186d4ecdSBarry Smith   if (!mapping->globals) {
18186d4ecdSBarry Smith     ierr = ISGlobalToLocalMappingApply(mapping,IS_GTOLM_MASK,0,0,0,0);CHKERRQ(ierr);
19186d4ecdSBarry Smith   }
20743c9b42SStefano Zampini   start = mapping->globalstart;
21743c9b42SStefano Zampini   end = mapping->globalend;
22186d4ecdSBarry Smith   for (i=0; i<n; i++) {
23186d4ecdSBarry Smith     if (in[i] < 0)          out[i] = in[i];
24186d4ecdSBarry Smith     else if (in[i] < start) out[i] = -1;
25186d4ecdSBarry Smith     else if (in[i] > end)   out[i] = -1;
26743c9b42SStefano Zampini     else                    out[i] = mapping->globals[in[i] - start];
27186d4ecdSBarry Smith   }
28186d4ecdSBarry Smith   PetscFunctionReturn(0);
29186d4ecdSBarry Smith }
30186d4ecdSBarry Smith 
31186d4ecdSBarry Smith 
32186d4ecdSBarry Smith #undef __FUNCT__
334a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetSize"
343b9aefa3SBarry Smith /*@C
353b9aefa3SBarry Smith     ISLocalToGlobalMappingGetSize - Gets the local size of a local to global mapping.
363b9aefa3SBarry Smith 
373b9aefa3SBarry Smith     Not Collective
383b9aefa3SBarry Smith 
393b9aefa3SBarry Smith     Input Parameter:
403b9aefa3SBarry Smith .   ltog - local to global mapping
413b9aefa3SBarry Smith 
423b9aefa3SBarry Smith     Output Parameter:
433b9aefa3SBarry Smith .   n - the number of entries in the local mapping
443b9aefa3SBarry Smith 
453b9aefa3SBarry Smith     Level: advanced
463b9aefa3SBarry Smith 
47273d9f13SBarry Smith     Concepts: mapping^local to global
483b9aefa3SBarry Smith 
493b9aefa3SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
503b9aefa3SBarry Smith @*/
517087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping mapping,PetscInt *n)
523b9aefa3SBarry Smith {
533b9aefa3SBarry Smith   PetscFunctionBegin;
540700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
554482741eSBarry Smith   PetscValidIntPointer(n,2);
563b9aefa3SBarry Smith   *n = mapping->n;
573b9aefa3SBarry Smith   PetscFunctionReturn(0);
583b9aefa3SBarry Smith }
593b9aefa3SBarry Smith 
604a2ae208SSatish Balay #undef __FUNCT__
614a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingView"
625a5d4f66SBarry Smith /*@C
635a5d4f66SBarry Smith     ISLocalToGlobalMappingView - View a local to global mapping
645a5d4f66SBarry Smith 
65b9cd556bSLois Curfman McInnes     Not Collective
66b9cd556bSLois Curfman McInnes 
675a5d4f66SBarry Smith     Input Parameters:
683b9aefa3SBarry Smith +   ltog - local to global mapping
693b9aefa3SBarry Smith -   viewer - viewer
705a5d4f66SBarry Smith 
71a997ad1aSLois Curfman McInnes     Level: advanced
72a997ad1aSLois Curfman McInnes 
73273d9f13SBarry Smith     Concepts: mapping^local to global
745a5d4f66SBarry Smith 
755a5d4f66SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
765a5d4f66SBarry Smith @*/
777087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingView(ISLocalToGlobalMapping mapping,PetscViewer viewer)
785a5d4f66SBarry Smith {
7932dcc486SBarry Smith   PetscInt       i;
8032dcc486SBarry Smith   PetscMPIInt    rank;
81ace3abfcSBarry Smith   PetscBool      iascii;
826849ba73SBarry Smith   PetscErrorCode ierr;
835a5d4f66SBarry Smith 
845a5d4f66SBarry Smith   PetscFunctionBegin;
850700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
863050cee2SBarry Smith   if (!viewer) {
87ce94432eSBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mapping),&viewer);CHKERRQ(ierr);
883050cee2SBarry Smith   }
890700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
905a5d4f66SBarry Smith 
91ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)mapping),&rank);CHKERRQ(ierr);
92251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
9332077d6dSBarry Smith   if (iascii) {
947b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
955a5d4f66SBarry Smith     for (i=0; i<mapping->n; i++) {
967904a332SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D %D\n",rank,i,mapping->indices[i]);CHKERRQ(ierr);
976831982aSBarry Smith     }
98b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
997b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
1007b23a99aSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name);
1015a5d4f66SBarry Smith   PetscFunctionReturn(0);
1025a5d4f66SBarry Smith }
1035a5d4f66SBarry Smith 
1044a2ae208SSatish Balay #undef __FUNCT__
1054a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS"
1061f428162SBarry Smith /*@
1072bdab257SBarry Smith     ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n)
1082bdab257SBarry Smith     ordering and a global parallel ordering.
1092bdab257SBarry Smith 
1100f5bd95cSBarry Smith     Not collective
111b9cd556bSLois Curfman McInnes 
112a997ad1aSLois Curfman McInnes     Input Parameter:
1138c03b21aSDmitry Karpeev .   is - index set containing the global numbers for each local number
1142bdab257SBarry Smith 
115a997ad1aSLois Curfman McInnes     Output Parameter:
1162bdab257SBarry Smith .   mapping - new mapping data structure
1172bdab257SBarry Smith 
118a997ad1aSLois Curfman McInnes     Level: advanced
119a997ad1aSLois Curfman McInnes 
120273d9f13SBarry Smith     Concepts: mapping^local to global
1212bdab257SBarry Smith 
1222bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
1232bdab257SBarry Smith @*/
1247087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping)
1252bdab257SBarry Smith {
1266849ba73SBarry Smith   PetscErrorCode ierr;
1273bbf0e92SBarry Smith   PetscInt       n,bs;
1285d0c19d7SBarry Smith   const PetscInt *indices;
1292bdab257SBarry Smith   MPI_Comm       comm;
1303bbf0e92SBarry Smith   PetscBool      isblock;
1313a40ed3dSBarry Smith 
1323a40ed3dSBarry Smith   PetscFunctionBegin;
1330700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,1);
1344482741eSBarry Smith   PetscValidPointer(mapping,2);
1352bdab257SBarry Smith 
1362bdab257SBarry Smith   ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr);
1373b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
1382bdab257SBarry Smith   ierr = ISGetIndices(is,&indices);CHKERRQ(ierr);
1393bbf0e92SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)is,ISBLOCK,&isblock);CHKERRQ(ierr);
1403bbf0e92SBarry Smith   ierr = ISGetBlockSize(is,&bs);CHKERRQ(ierr);
1413bbf0e92SBarry Smith   if (!isblock || bs == 1) {
142d5ad8652SBarry Smith     ierr = ISLocalToGlobalMappingCreate(comm,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
1432bdab257SBarry Smith     ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr);
1443bbf0e92SBarry Smith   } else {
1453bbf0e92SBarry Smith     ierr = ISLocalToGlobalMappingCreate(comm,n,indices,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
1463bbf0e92SBarry Smith   }
1473a40ed3dSBarry Smith   PetscFunctionReturn(0);
1482bdab257SBarry Smith }
1495a5d4f66SBarry Smith 
150a4d96a55SJed Brown #undef __FUNCT__
151a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF"
152a4d96a55SJed Brown /*@C
153a4d96a55SJed Brown     ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n)
154a4d96a55SJed Brown     ordering and a global parallel ordering.
155a4d96a55SJed Brown 
156a4d96a55SJed Brown     Collective
157a4d96a55SJed Brown 
158a4d96a55SJed Brown     Input Parameter:
159a4d96a55SJed Brown +   sf - star forest mapping contiguous local indices to (rank, offset)
160a4d96a55SJed Brown -   start - first global index on this process
161a4d96a55SJed Brown 
162a4d96a55SJed Brown     Output Parameter:
163a4d96a55SJed Brown .   mapping - new mapping data structure
164a4d96a55SJed Brown 
165a4d96a55SJed Brown     Level: advanced
166a4d96a55SJed Brown 
167a4d96a55SJed Brown     Concepts: mapping^local to global
168a4d96a55SJed Brown 
169a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
170a4d96a55SJed Brown @*/
171a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping)
172a4d96a55SJed Brown {
173a4d96a55SJed Brown   PetscErrorCode ierr;
174a4d96a55SJed Brown   PetscInt       i,maxlocal,nroots,nleaves,*globals,*ltog;
175a4d96a55SJed Brown   const PetscInt *ilocal;
176a4d96a55SJed Brown   MPI_Comm       comm;
177a4d96a55SJed Brown 
178a4d96a55SJed Brown   PetscFunctionBegin;
179a4d96a55SJed Brown   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
180a4d96a55SJed Brown   PetscValidPointer(mapping,3);
181a4d96a55SJed Brown 
182a4d96a55SJed Brown   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
1830298fd71SBarry Smith   ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,NULL);CHKERRQ(ierr);
184f6e5521dSKarl Rupp   if (ilocal) {
185f6e5521dSKarl Rupp     for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1);
186f6e5521dSKarl Rupp   }
187a4d96a55SJed Brown   else maxlocal = nleaves;
188785e854fSJed Brown   ierr = PetscMalloc1(nroots,&globals);CHKERRQ(ierr);
189785e854fSJed Brown   ierr = PetscMalloc1(maxlocal,&ltog);CHKERRQ(ierr);
190a4d96a55SJed Brown   for (i=0; i<nroots; i++) globals[i] = start + i;
191a4d96a55SJed Brown   for (i=0; i<maxlocal; i++) ltog[i] = -1;
192a4d96a55SJed Brown   ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
193a4d96a55SJed Brown   ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
194a4d96a55SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
195a4d96a55SJed Brown   ierr = PetscFree(globals);CHKERRQ(ierr);
196a4d96a55SJed Brown   PetscFunctionReturn(0);
197a4d96a55SJed Brown }
198b46b645bSBarry Smith 
1994a2ae208SSatish Balay #undef __FUNCT__
2004a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate"
201ba5bb76aSSatish Balay /*@
20290f02eecSBarry Smith     ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n)
20390f02eecSBarry Smith     ordering and a global parallel ordering.
2042362add9SBarry Smith 
20589d82c54SBarry Smith     Not Collective, but communicator may have more than one process
206b9cd556bSLois Curfman McInnes 
2072362add9SBarry Smith     Input Parameters:
20889d82c54SBarry Smith +   comm - MPI communicator
20990f02eecSBarry Smith .   n - the number of local elements
2109669e4d8SBarry Smith .   indices - the global index for each local element, these do not need to be in increasing order (sorted)
211d5ad8652SBarry Smith -   mode - see PetscCopyMode
2122362add9SBarry Smith 
213a997ad1aSLois Curfman McInnes     Output Parameter:
21490f02eecSBarry Smith .   mapping - new mapping data structure
2152362add9SBarry Smith 
216a997ad1aSLois Curfman McInnes     Level: advanced
217a997ad1aSLois Curfman McInnes 
218273d9f13SBarry Smith     Concepts: mapping^local to global
2192362add9SBarry Smith 
220d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
2212362add9SBarry Smith @*/
2227087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping)
2232362add9SBarry Smith {
2246849ba73SBarry Smith   PetscErrorCode ierr;
22532dcc486SBarry Smith   PetscInt       *in;
226b46b645bSBarry Smith 
227b46b645bSBarry Smith   PetscFunctionBegin;
22873911063SBarry Smith   if (n) PetscValidIntPointer(indices,3);
2294482741eSBarry Smith   PetscValidPointer(mapping,4);
230b46b645bSBarry Smith 
2310298fd71SBarry Smith   *mapping = NULL;
232607a6623SBarry Smith   ierr = ISInitializePackage();CHKERRQ(ierr);
2332362add9SBarry Smith 
23467c2884eSBarry Smith   ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,"ISLocalToGlobalMapping","Local to global mapping","IS",
23552e6d16bSBarry Smith                            cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr);
236d4bb536fSBarry Smith   (*mapping)->n = n;
237d4bb536fSBarry Smith   /*
238d4bb536fSBarry Smith     Do not create the global to local mapping. This is only created if
239d4bb536fSBarry Smith     ISGlobalToLocalMapping() is called
240d4bb536fSBarry Smith   */
241d4bb536fSBarry Smith   (*mapping)->globals = 0;
242d5ad8652SBarry Smith   if (mode == PETSC_COPY_VALUES) {
243785e854fSJed Brown     ierr = PetscMalloc1(n,&in);CHKERRQ(ierr);
244d5ad8652SBarry Smith     ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr);
2453bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
246d5ad8652SBarry Smith     (*mapping)->indices = in;
247f6e5521dSKarl Rupp   } else if (mode == PETSC_OWN_POINTER) (*mapping)->indices = (PetscInt*)indices;
248f6e5521dSKarl Rupp   else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER");
2493a40ed3dSBarry Smith   PetscFunctionReturn(0);
2502362add9SBarry Smith }
2512362add9SBarry Smith 
2524a2ae208SSatish Balay #undef __FUNCT__
2534a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingBlock"
254bce096a4SSatish Balay /*@
255323b833fSBarry Smith     ISLocalToGlobalMappingBlock - Creates a blocked index version of an
256323b833fSBarry Smith        ISLocalToGlobalMapping that is appropriate for MatSetLocalToGlobalMappingBlock()
257323b833fSBarry Smith        and VecSetLocalToGlobalMappingBlock().
258323b833fSBarry Smith 
259323b833fSBarry Smith     Not Collective, but communicator may have more than one process
260323b833fSBarry Smith 
261323b833fSBarry Smith     Input Parameters:
262323b833fSBarry Smith +    inmap - original point-wise mapping
263323b833fSBarry Smith -    bs - block size
264323b833fSBarry Smith 
265323b833fSBarry Smith     Output Parameter:
26669eb54c3SBarry Smith .   outmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
267323b833fSBarry Smith 
268323b833fSBarry Smith     Level: advanced
269323b833fSBarry Smith 
270323b833fSBarry Smith     Concepts: mapping^local to global
271323b833fSBarry Smith 
272323b833fSBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
273323b833fSBarry Smith @*/
2747087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
275323b833fSBarry Smith {
2766849ba73SBarry Smith   PetscErrorCode ierr;
27732dcc486SBarry Smith   PetscInt       *ii,i,n;
278323b833fSBarry Smith 
279323b833fSBarry Smith   PetscFunctionBegin;
2800700a824SBarry Smith   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
281b2beed0aSJed Brown   PetscValidPointer(outmap,3);
282323b833fSBarry Smith   if (bs > 1) {
283323b833fSBarry Smith     n = inmap->n/bs;
284e32f2f54SBarry Smith     if (n*bs != inmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Pointwise mapping length is not divisible by block size");
285785e854fSJed Brown     ierr = PetscMalloc1(n,&ii);CHKERRQ(ierr);
286f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[bs*i]/bs;
287ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
288323b833fSBarry Smith   } else {
289323b833fSBarry Smith     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
290c3122656SLisandro Dalcin     *outmap = inmap;
291323b833fSBarry Smith   }
292323b833fSBarry Smith   PetscFunctionReturn(0);
293323b833fSBarry Smith }
294323b833fSBarry Smith 
2954a2ae208SSatish Balay #undef __FUNCT__
2968ab951edSJed Brown #define __FUNCT__ "ISLocalToGlobalMappingUnBlock"
297b2beed0aSJed Brown /*@
298b2beed0aSJed Brown     ISLocalToGlobalMappingUnBlock - Creates a scalar index version of a blocked
299b2beed0aSJed Brown        ISLocalToGlobalMapping
300b2beed0aSJed Brown 
301b2beed0aSJed Brown     Not Collective, but communicator may have more than one process
302b2beed0aSJed Brown 
303b2beed0aSJed Brown     Input Parameter:
304b2beed0aSJed Brown + inmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
305b2beed0aSJed Brown - bs - block size
306b2beed0aSJed Brown 
307b2beed0aSJed Brown     Output Parameter:
308b2beed0aSJed Brown .   outmap - pointwise mapping
309b2beed0aSJed Brown 
310b2beed0aSJed Brown     Level: advanced
311b2beed0aSJed Brown 
312b2beed0aSJed Brown     Concepts: mapping^local to global
313b2beed0aSJed Brown 
314b2beed0aSJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingBlock()
315b2beed0aSJed Brown @*/
3167087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
317b2beed0aSJed Brown {
318b2beed0aSJed Brown   PetscErrorCode ierr;
319b2beed0aSJed Brown   PetscInt       *ii,i,n;
320b2beed0aSJed Brown 
321b2beed0aSJed Brown   PetscFunctionBegin;
322b2beed0aSJed Brown   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
323b2beed0aSJed Brown   PetscValidPointer(outmap,2);
324b2beed0aSJed Brown   if (bs > 1) {
325b2beed0aSJed Brown     n    = inmap->n*bs;
326785e854fSJed Brown     ierr = PetscMalloc1(n,&ii);CHKERRQ(ierr);
327f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[i/bs]*bs + (i%bs);
328ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
329b2beed0aSJed Brown   } else {
330b2beed0aSJed Brown     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
331b2beed0aSJed Brown     *outmap = inmap;
332b2beed0aSJed Brown   }
333b2beed0aSJed Brown   PetscFunctionReturn(0);
334b2beed0aSJed Brown }
335b2beed0aSJed Brown 
336b2beed0aSJed Brown #undef __FUNCT__
3374a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy"
33890f02eecSBarry Smith /*@
33990f02eecSBarry Smith    ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n)
34090f02eecSBarry Smith    ordering and a global parallel ordering.
34190f02eecSBarry Smith 
3420f5bd95cSBarry Smith    Note Collective
343b9cd556bSLois Curfman McInnes 
34490f02eecSBarry Smith    Input Parameters:
34590f02eecSBarry Smith .  mapping - mapping data structure
34690f02eecSBarry Smith 
347a997ad1aSLois Curfman McInnes    Level: advanced
348a997ad1aSLois Curfman McInnes 
3493acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate()
35090f02eecSBarry Smith @*/
3516bf464f9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping)
35290f02eecSBarry Smith {
353dfbe8321SBarry Smith   PetscErrorCode ierr;
3545fd66863SKarl Rupp 
3553a40ed3dSBarry Smith   PetscFunctionBegin;
3566bf464f9SBarry Smith   if (!*mapping) PetscFunctionReturn(0);
3576bf464f9SBarry Smith   PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1);
358997056adSBarry Smith   if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);}
3596bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->indices);CHKERRQ(ierr);
3606bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->globals);CHKERRQ(ierr);
361d38fa0fbSBarry Smith   ierr     = PetscHeaderDestroy(mapping);CHKERRQ(ierr);
362992144d0SBarry Smith   *mapping = 0;
3633a40ed3dSBarry Smith   PetscFunctionReturn(0);
36490f02eecSBarry Smith }
36590f02eecSBarry Smith 
3664a2ae208SSatish Balay #undef __FUNCT__
3674a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS"
36890f02eecSBarry Smith /*@
3693acfe500SLois Curfman McInnes     ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering
3703acfe500SLois Curfman McInnes     a new index set using the global numbering defined in an ISLocalToGlobalMapping
3713acfe500SLois Curfman McInnes     context.
37290f02eecSBarry Smith 
373b9cd556bSLois Curfman McInnes     Not collective
374b9cd556bSLois Curfman McInnes 
37590f02eecSBarry Smith     Input Parameters:
376b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
377b9cd556bSLois Curfman McInnes -   is - index set in local numbering
37890f02eecSBarry Smith 
37990f02eecSBarry Smith     Output Parameters:
38090f02eecSBarry Smith .   newis - index set in global numbering
38190f02eecSBarry Smith 
382a997ad1aSLois Curfman McInnes     Level: advanced
383a997ad1aSLois Curfman McInnes 
384273d9f13SBarry Smith     Concepts: mapping^local to global
3853acfe500SLois Curfman McInnes 
38690f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
387d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply()
38890f02eecSBarry Smith @*/
3897087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis)
39090f02eecSBarry Smith {
3916849ba73SBarry Smith   PetscErrorCode ierr;
3925d0c19d7SBarry Smith   PetscInt       n,i,*idxmap,*idxout,Nmax = mapping->n;
3935d0c19d7SBarry Smith   const PetscInt *idxin;
3943a40ed3dSBarry Smith 
3953a40ed3dSBarry Smith   PetscFunctionBegin;
3960700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
3970700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,2);
3984482741eSBarry Smith   PetscValidPointer(newis,3);
39990f02eecSBarry Smith 
4003b9aefa3SBarry Smith   ierr   = ISGetLocalSize(is,&n);CHKERRQ(ierr);
40190f02eecSBarry Smith   ierr   = ISGetIndices(is,&idxin);CHKERRQ(ierr);
40290f02eecSBarry Smith   idxmap = mapping->indices;
40390f02eecSBarry Smith 
404785e854fSJed Brown   ierr = PetscMalloc1(n,&idxout);CHKERRQ(ierr);
40590f02eecSBarry Smith   for (i=0; i<n; i++) {
4067904a332SBarry 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);
40790f02eecSBarry Smith     idxout[i] = idxmap[idxin[i]];
40890f02eecSBarry Smith   }
4093b9aefa3SBarry Smith   ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr);
410*543f3098SMatthew G. Knepley   ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is),n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr);
4113a40ed3dSBarry Smith   PetscFunctionReturn(0);
41290f02eecSBarry Smith }
41390f02eecSBarry Smith 
414afcb2eb5SJed Brown #undef __FUNCT__
415afcb2eb5SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingApply"
416b89cb25eSSatish Balay /*@
4173acfe500SLois Curfman McInnes    ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering
4183acfe500SLois Curfman McInnes    and converts them to the global numbering.
41990f02eecSBarry Smith 
420b9cd556bSLois Curfman McInnes    Not collective
421b9cd556bSLois Curfman McInnes 
422bb25748dSBarry Smith    Input Parameters:
423b9cd556bSLois Curfman McInnes +  mapping - the local to global mapping context
424bb25748dSBarry Smith .  N - number of integers
425b9cd556bSLois Curfman McInnes -  in - input indices in local numbering
426bb25748dSBarry Smith 
427bb25748dSBarry Smith    Output Parameter:
428bb25748dSBarry Smith .  out - indices in global numbering
429bb25748dSBarry Smith 
430b9cd556bSLois Curfman McInnes    Notes:
431b9cd556bSLois Curfman McInnes    The in and out array parameters may be identical.
432d4bb536fSBarry Smith 
433a997ad1aSLois Curfman McInnes    Level: advanced
434a997ad1aSLois Curfman McInnes 
435bb25748dSBarry Smith .seealso: ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
4360752156aSBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
437d4bb536fSBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
438bb25748dSBarry Smith 
439273d9f13SBarry Smith     Concepts: mapping^local to global
440afcb2eb5SJed Brown @*/
441afcb2eb5SJed Brown PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[])
442afcb2eb5SJed Brown {
443afcb2eb5SJed Brown   PetscInt       i,Nmax = mapping->n;
444afcb2eb5SJed Brown   const PetscInt *idx = mapping->indices;
445d4bb536fSBarry Smith 
446afcb2eb5SJed Brown   PetscFunctionBegin;
447afcb2eb5SJed Brown   for (i=0; i<N; i++) {
448afcb2eb5SJed Brown     if (in[i] < 0) {
449afcb2eb5SJed Brown       out[i] = in[i];
450afcb2eb5SJed Brown       continue;
451afcb2eb5SJed Brown     }
452afcb2eb5SJed 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);
453afcb2eb5SJed Brown     out[i] = idx[in[i]];
454afcb2eb5SJed Brown   }
455afcb2eb5SJed Brown   PetscFunctionReturn(0);
456afcb2eb5SJed Brown }
457d4bb536fSBarry Smith 
458d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/
459d4bb536fSBarry Smith 
4604a2ae208SSatish Balay #undef __FUNCT__
4614a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private"
462d4bb536fSBarry Smith /*
463d4bb536fSBarry Smith     Creates the global fields in the ISLocalToGlobalMapping structure
464d4bb536fSBarry Smith */
4656849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping)
466d4bb536fSBarry Smith {
4676849ba73SBarry Smith   PetscErrorCode ierr;
46832dcc486SBarry Smith   PetscInt       i,*idx = mapping->indices,n = mapping->n,end,start,*globals;
469d4bb536fSBarry Smith 
4703a40ed3dSBarry Smith   PetscFunctionBegin;
471d4bb536fSBarry Smith   end   = 0;
472ec268f7cSJed Brown   start = PETSC_MAX_INT;
473d4bb536fSBarry Smith 
474d4bb536fSBarry Smith   for (i=0; i<n; i++) {
475d4bb536fSBarry Smith     if (idx[i] < 0) continue;
476d4bb536fSBarry Smith     if (idx[i] < start) start = idx[i];
477d4bb536fSBarry Smith     if (idx[i] > end)   end   = idx[i];
478d4bb536fSBarry Smith   }
479d4bb536fSBarry Smith   if (start > end) {start = 0; end = -1;}
480d4bb536fSBarry Smith   mapping->globalstart = start;
481d4bb536fSBarry Smith   mapping->globalend   = end;
482d4bb536fSBarry Smith 
483785e854fSJed Brown   ierr             = PetscMalloc1((end-start+2),&globals);CHKERRQ(ierr);
484b0a32e0cSBarry Smith   mapping->globals = globals;
485f6e5521dSKarl Rupp   for (i=0; i<end-start+1; i++) globals[i] = -1;
486d4bb536fSBarry Smith   for (i=0; i<n; i++) {
487d4bb536fSBarry Smith     if (idx[i] < 0) continue;
488d4bb536fSBarry Smith     globals[idx[i] - start] = i;
489d4bb536fSBarry Smith   }
490d4bb536fSBarry Smith 
4913bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr);
4923a40ed3dSBarry Smith   PetscFunctionReturn(0);
493d4bb536fSBarry Smith }
494d4bb536fSBarry Smith 
4954a2ae208SSatish Balay #undef __FUNCT__
4964a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply"
497d4bb536fSBarry Smith /*@
498a997ad1aSLois Curfman McInnes     ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers
499a997ad1aSLois Curfman McInnes     specified with a global numbering.
500d4bb536fSBarry Smith 
501b9cd556bSLois Curfman McInnes     Not collective
502b9cd556bSLois Curfman McInnes 
503d4bb536fSBarry Smith     Input Parameters:
504b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
505d4bb536fSBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
506d4bb536fSBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
507d4bb536fSBarry Smith .   n - number of global indices to map
508b9cd556bSLois Curfman McInnes -   idx - global indices to map
509d4bb536fSBarry Smith 
510d4bb536fSBarry Smith     Output Parameters:
511b9cd556bSLois Curfman McInnes +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
512b9cd556bSLois Curfman McInnes -   idxout - local index of each global index, one must pass in an array long enough
513e182c471SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApply() with
5140298fd71SBarry Smith              idxout == NULL to determine the required length (returned in nout)
515e182c471SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApply()
516e182c471SBarry Smith              a second time to set the values.
517d4bb536fSBarry Smith 
518b9cd556bSLois Curfman McInnes     Notes:
5190298fd71SBarry Smith     Either nout or idxout may be NULL. idx and idxout may be identical.
520d4bb536fSBarry Smith 
5210f5bd95cSBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
5220f5bd95cSBarry Smith     array to compute these.
5230f5bd95cSBarry Smith 
524a997ad1aSLois Curfman McInnes     Level: advanced
525a997ad1aSLois Curfman McInnes 
52632fd6b96SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
52732fd6b96SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
52832fd6b96SBarry Smith 
529273d9f13SBarry Smith     Concepts: mapping^global to local
530d4bb536fSBarry Smith 
531d4bb536fSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
532d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy()
533d4bb536fSBarry Smith @*/
5347087cfbeSBarry Smith PetscErrorCode  ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
53532dcc486SBarry Smith                                   PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
536d4bb536fSBarry Smith {
53732dcc486SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end;
5386849ba73SBarry Smith   PetscErrorCode ierr;
539d4bb536fSBarry Smith 
5403a40ed3dSBarry Smith   PetscFunctionBegin;
5410700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
542d4bb536fSBarry Smith   if (!mapping->globals) {
543d4bb536fSBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
544d4bb536fSBarry Smith   }
545d4bb536fSBarry Smith   globals = mapping->globals;
546d4bb536fSBarry Smith   start   = mapping->globalstart;
547d4bb536fSBarry Smith   end     = mapping->globalend;
548d4bb536fSBarry Smith 
549d4bb536fSBarry Smith   if (type == IS_GTOLM_MASK) {
550d4bb536fSBarry Smith     if (idxout) {
551d4bb536fSBarry Smith       for (i=0; i<n; i++) {
552d4bb536fSBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
553d4bb536fSBarry Smith         else if (idx[i] < start) idxout[i] = -1;
554d4bb536fSBarry Smith         else if (idx[i] > end)   idxout[i] = -1;
555d4bb536fSBarry Smith         else                     idxout[i] = globals[idx[i] - start];
556d4bb536fSBarry Smith       }
557d4bb536fSBarry Smith     }
558d4bb536fSBarry Smith     if (nout) *nout = n;
559d4bb536fSBarry Smith   } else {
560d4bb536fSBarry Smith     if (idxout) {
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         idxout[nf++] = tmp;
568d4bb536fSBarry Smith       }
569d4bb536fSBarry Smith     } else {
570d4bb536fSBarry Smith       for (i=0; i<n; i++) {
571d4bb536fSBarry Smith         if (idx[i] < 0) continue;
572d4bb536fSBarry Smith         if (idx[i] < start) continue;
573d4bb536fSBarry Smith         if (idx[i] > end) continue;
574d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
575d4bb536fSBarry Smith         if (tmp < 0) continue;
576d4bb536fSBarry Smith         nf++;
577d4bb536fSBarry Smith       }
578d4bb536fSBarry Smith     }
579d4bb536fSBarry Smith     if (nout) *nout = nf;
580d4bb536fSBarry Smith   }
5813a40ed3dSBarry Smith   PetscFunctionReturn(0);
582d4bb536fSBarry Smith }
58390f02eecSBarry Smith 
5844a2ae208SSatish Balay #undef __FUNCT__
5854a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetInfo"
58689d82c54SBarry Smith /*@C
58789d82c54SBarry Smith     ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and
58889d82c54SBarry Smith      each index shared by more than one processor
58989d82c54SBarry Smith 
59089d82c54SBarry Smith     Collective on ISLocalToGlobalMapping
59189d82c54SBarry Smith 
59289d82c54SBarry Smith     Input Parameters:
59389d82c54SBarry Smith .   mapping - the mapping from local to global indexing
59489d82c54SBarry Smith 
59589d82c54SBarry Smith     Output Parameter:
59689d82c54SBarry Smith +   nproc - number of processors that are connected to this one
59789d82c54SBarry Smith .   proc - neighboring processors
59807b52d57SBarry Smith .   numproc - number of indices for each subdomain (processor)
5993463a7baSJed Brown -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
60089d82c54SBarry Smith 
60189d82c54SBarry Smith     Level: advanced
60289d82c54SBarry Smith 
603273d9f13SBarry Smith     Concepts: mapping^local to global
60489d82c54SBarry Smith 
6052cfcea29SBarry Smith     Fortran Usage:
6062cfcea29SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
6072cfcea29SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
6082cfcea29SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
6092cfcea29SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
6102cfcea29SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
6112cfcea29SBarry Smith 
6122cfcea29SBarry Smith 
61307b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
61407b52d57SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
61589d82c54SBarry Smith @*/
6167087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
61789d82c54SBarry Smith {
6186849ba73SBarry Smith   PetscErrorCode ierr;
61997f1f81fSBarry Smith   PetscMPIInt    size,rank,tag1,tag2,tag3,*len,*source,imdex;
62032dcc486SBarry Smith   PetscInt       i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices;
62132dcc486SBarry Smith   PetscInt       *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc;
62297f1f81fSBarry Smith   PetscInt       cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned;
62332dcc486SBarry Smith   PetscInt       node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp;
62432dcc486SBarry Smith   PetscInt       first_procs,first_numprocs,*first_indices;
62589d82c54SBarry Smith   MPI_Request    *recv_waits,*send_waits;
62630dcb7c9SBarry Smith   MPI_Status     recv_status,*send_status,*recv_statuses;
627ce94432eSBarry Smith   MPI_Comm       comm;
628ace3abfcSBarry Smith   PetscBool      debug = PETSC_FALSE;
62989d82c54SBarry Smith 
63089d82c54SBarry Smith   PetscFunctionBegin;
6310700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
632ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)mapping,&comm);CHKERRQ(ierr);
63324cf384cSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
63424cf384cSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
63524cf384cSBarry Smith   if (size == 1) {
63624cf384cSBarry Smith     *nproc         = 0;
6370298fd71SBarry Smith     *procs         = NULL;
63832dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr);
6391e2105dcSBarry Smith     (*numprocs)[0] = 0;
64032dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr);
6410298fd71SBarry Smith     (*indices)[0]  = NULL;
64224cf384cSBarry Smith     PetscFunctionReturn(0);
64324cf384cSBarry Smith   }
64424cf384cSBarry Smith 
6450298fd71SBarry Smith   ierr = PetscOptionsGetBool(NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,NULL);CHKERRQ(ierr);
64607b52d57SBarry Smith 
6473677ff5aSBarry Smith   /*
6483677ff5aSBarry Smith     Notes on ISLocalToGlobalMappingGetInfo
6493677ff5aSBarry Smith 
6503677ff5aSBarry Smith     globally owned node - the nodes that have been assigned to this processor in global
6513677ff5aSBarry Smith            numbering, just for this routine.
6523677ff5aSBarry Smith 
6533677ff5aSBarry Smith     nontrivial globally owned node - node assigned to this processor that is on a subdomain
6543677ff5aSBarry Smith            boundary (i.e. is has more than one local owner)
6553677ff5aSBarry Smith 
6563677ff5aSBarry Smith     locally owned node - node that exists on this processors subdomain
6573677ff5aSBarry Smith 
6583677ff5aSBarry Smith     nontrivial locally owned node - node that is not in the interior (i.e. has more than one
6593677ff5aSBarry Smith            local subdomain
6603677ff5aSBarry Smith   */
66124cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr);
66224cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr);
66324cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr);
66489d82c54SBarry Smith 
66589d82c54SBarry Smith   for (i=0; i<n; i++) {
66689d82c54SBarry Smith     if (lindices[i] > max) max = lindices[i];
66789d82c54SBarry Smith   }
66832dcc486SBarry Smith   ierr   = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
66978058e43SBarry Smith   Ng++;
67089d82c54SBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
67189d82c54SBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
672bc8ff85bSBarry Smith   scale  = Ng/size + 1;
673a2e34c3dSBarry Smith   ng     = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng);
674caba0dd0SBarry Smith   rstart = scale*rank;
67589d82c54SBarry Smith 
67689d82c54SBarry Smith   /* determine ownership ranges of global indices */
677785e854fSJed Brown   ierr = PetscMalloc1(2*size,&nprocs);CHKERRQ(ierr);
67832dcc486SBarry Smith   ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
67989d82c54SBarry Smith 
68089d82c54SBarry Smith   /* determine owners of each local node  */
681785e854fSJed Brown   ierr = PetscMalloc1(n,&owner);CHKERRQ(ierr);
68289d82c54SBarry Smith   for (i=0; i<n; i++) {
6833677ff5aSBarry Smith     proc             = lindices[i]/scale; /* processor that globally owns this index */
68427c402fcSBarry Smith     nprocs[2*proc+1] = 1;                 /* processor globally owns at least one of ours */
6853677ff5aSBarry Smith     owner[i]         = proc;
68627c402fcSBarry Smith     nprocs[2*proc]++;                     /* count of how many that processor globally owns of ours */
68789d82c54SBarry Smith   }
68827c402fcSBarry Smith   nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1];
6897904a332SBarry Smith   ierr = PetscInfo1(mapping,"Number of global owners for my local data %D\n",nsends);CHKERRQ(ierr);
69089d82c54SBarry Smith 
69189d82c54SBarry Smith   /* inform other processors of number of messages and max length*/
69227c402fcSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
6937904a332SBarry Smith   ierr = PetscInfo1(mapping,"Number of local owners for my global data %D\n",nrecvs);CHKERRQ(ierr);
69489d82c54SBarry Smith 
69589d82c54SBarry Smith   /* post receives for owned rows */
696785e854fSJed Brown   ierr = PetscMalloc1((2*nrecvs+1)*(nmax+1),&recvs);CHKERRQ(ierr);
697785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&recv_waits);CHKERRQ(ierr);
69889d82c54SBarry Smith   for (i=0; i<nrecvs; i++) {
69932dcc486SBarry Smith     ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr);
70089d82c54SBarry Smith   }
70189d82c54SBarry Smith 
70289d82c54SBarry Smith   /* pack messages containing lists of local nodes to owners */
703785e854fSJed Brown   ierr      = PetscMalloc1((2*n+1),&sends);CHKERRQ(ierr);
704785e854fSJed Brown   ierr      = PetscMalloc1((size+1),&starts);CHKERRQ(ierr);
70589d82c54SBarry Smith   starts[0] = 0;
706f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
70789d82c54SBarry Smith   for (i=0; i<n; i++) {
70889d82c54SBarry Smith     sends[starts[owner[i]]++] = lindices[i];
70930dcb7c9SBarry Smith     sends[starts[owner[i]]++] = i;
71089d82c54SBarry Smith   }
71189d82c54SBarry Smith   ierr = PetscFree(owner);CHKERRQ(ierr);
71289d82c54SBarry Smith   starts[0] = 0;
713f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
71489d82c54SBarry Smith 
71589d82c54SBarry Smith   /* send the messages */
716785e854fSJed Brown   ierr = PetscMalloc1((nsends+1),&send_waits);CHKERRQ(ierr);
717785e854fSJed Brown   ierr = PetscMalloc1((nsends+1),&dest);CHKERRQ(ierr);
71889d82c54SBarry Smith   cnt = 0;
71989d82c54SBarry Smith   for (i=0; i<size; i++) {
72027c402fcSBarry Smith     if (nprocs[2*i]) {
72132dcc486SBarry Smith       ierr      = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr);
72230dcb7c9SBarry Smith       dest[cnt] = i;
72389d82c54SBarry Smith       cnt++;
72489d82c54SBarry Smith     }
72589d82c54SBarry Smith   }
72689d82c54SBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
72789d82c54SBarry Smith 
72889d82c54SBarry Smith   /* wait on receives */
729785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&source);CHKERRQ(ierr);
730785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&len);CHKERRQ(ierr);
73189d82c54SBarry Smith   cnt  = nrecvs;
732785e854fSJed Brown   ierr = PetscMalloc1((ng+1),&nownedsenders);CHKERRQ(ierr);
73332dcc486SBarry Smith   ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr);
73489d82c54SBarry Smith   while (cnt) {
73589d82c54SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
73689d82c54SBarry Smith     /* unpack receives into our local space */
73732dcc486SBarry Smith     ierr          = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr);
73889d82c54SBarry Smith     source[imdex] = recv_status.MPI_SOURCE;
73930dcb7c9SBarry Smith     len[imdex]    = len[imdex]/2;
740caba0dd0SBarry Smith     /* count how many local owners for each of my global owned indices */
74130dcb7c9SBarry Smith     for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++;
74289d82c54SBarry Smith     cnt--;
74389d82c54SBarry Smith   }
74489d82c54SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
74589d82c54SBarry Smith 
74630dcb7c9SBarry Smith   /* count how many globally owned indices are on an edge multiplied by how many processors own them. */
747bc8ff85bSBarry Smith   nowned  = 0;
748bc8ff85bSBarry Smith   nownedm = 0;
749bc8ff85bSBarry Smith   for (i=0; i<ng; i++) {
750bc8ff85bSBarry Smith     if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;}
751bc8ff85bSBarry Smith   }
752bc8ff85bSBarry Smith 
753bc8ff85bSBarry Smith   /* create single array to contain rank of all local owners of each globally owned index */
754785e854fSJed Brown   ierr      = PetscMalloc1((nownedm+1),&ownedsenders);CHKERRQ(ierr);
755785e854fSJed Brown   ierr      = PetscMalloc1((ng+1),&starts);CHKERRQ(ierr);
756bc8ff85bSBarry Smith   starts[0] = 0;
757bc8ff85bSBarry Smith   for (i=1; i<ng; i++) {
758bc8ff85bSBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
759bc8ff85bSBarry Smith     else starts[i] = starts[i-1];
760bc8ff85bSBarry Smith   }
761bc8ff85bSBarry Smith 
76230dcb7c9SBarry Smith   /* for each nontrival globally owned node list all arriving processors */
763bc8ff85bSBarry Smith   for (i=0; i<nrecvs; i++) {
764bc8ff85bSBarry Smith     for (j=0; j<len[i]; j++) {
76530dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
766f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) ownedsenders[starts[node]++] = source[i];
767bc8ff85bSBarry Smith     }
768bc8ff85bSBarry Smith   }
769bc8ff85bSBarry Smith 
77007b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
77130dcb7c9SBarry Smith     starts[0] = 0;
77230dcb7c9SBarry Smith     for (i=1; i<ng; i++) {
77330dcb7c9SBarry Smith       if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
77430dcb7c9SBarry Smith       else starts[i] = starts[i-1];
77530dcb7c9SBarry Smith     }
77630dcb7c9SBarry Smith     for (i=0; i<ng; i++) {
77730dcb7c9SBarry Smith       if (nownedsenders[i] > 1) {
7787904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] global node %D local owner processors: ",rank,i+rstart);CHKERRQ(ierr);
77930dcb7c9SBarry Smith         for (j=0; j<nownedsenders[i]; j++) {
7807904a332SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%D ",ownedsenders[starts[i]+j]);CHKERRQ(ierr);
78130dcb7c9SBarry Smith         }
78230dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
78330dcb7c9SBarry Smith       }
78430dcb7c9SBarry Smith     }
7850ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
78607b52d57SBarry Smith   } /* -----------------------------------  */
78730dcb7c9SBarry Smith 
7883677ff5aSBarry Smith   /* wait on original sends */
7893a96401aSBarry Smith   if (nsends) {
790785e854fSJed Brown     ierr = PetscMalloc1(nsends,&send_status);CHKERRQ(ierr);
7913a96401aSBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
7923a96401aSBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
7933a96401aSBarry Smith   }
79489d82c54SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
7953a96401aSBarry Smith   ierr = PetscFree(sends);CHKERRQ(ierr);
7963677ff5aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
7973677ff5aSBarry Smith 
7983677ff5aSBarry Smith   /* pack messages to send back to local owners */
79930dcb7c9SBarry Smith   starts[0] = 0;
80030dcb7c9SBarry Smith   for (i=1; i<ng; i++) {
80130dcb7c9SBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
80230dcb7c9SBarry Smith     else starts[i] = starts[i-1];
80330dcb7c9SBarry Smith   }
80430dcb7c9SBarry Smith   nsends2 = nrecvs;
805785e854fSJed Brown   ierr    = PetscMalloc1((nsends2+1),&nprocs);CHKERRQ(ierr); /* length of each message */
80630dcb7c9SBarry Smith   for (i=0; i<nrecvs; i++) {
80730dcb7c9SBarry Smith     nprocs[i] = 1;
80830dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
80930dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
810f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) nprocs[i] += 2 + nownedsenders[node];
81130dcb7c9SBarry Smith     }
81230dcb7c9SBarry Smith   }
813f6e5521dSKarl Rupp   nt = 0;
814f6e5521dSKarl Rupp   for (i=0; i<nsends2; i++) nt += nprocs[i];
815f6e5521dSKarl Rupp 
816785e854fSJed Brown   ierr = PetscMalloc1((nt+1),&sends2);CHKERRQ(ierr);
817785e854fSJed Brown   ierr = PetscMalloc1((nsends2+1),&starts2);CHKERRQ(ierr);
818f6e5521dSKarl Rupp 
819f6e5521dSKarl Rupp   starts2[0] = 0;
820f6e5521dSKarl Rupp   for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1];
82130dcb7c9SBarry Smith   /*
82230dcb7c9SBarry Smith      Each message is 1 + nprocs[i] long, and consists of
82330dcb7c9SBarry Smith        (0) the number of nodes being sent back
82430dcb7c9SBarry Smith        (1) the local node number,
82530dcb7c9SBarry Smith        (2) the number of processors sharing it,
82630dcb7c9SBarry Smith        (3) the processors sharing it
82730dcb7c9SBarry Smith   */
82830dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
82930dcb7c9SBarry Smith     cnt = 1;
83030dcb7c9SBarry Smith     sends2[starts2[i]] = 0;
83130dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
83230dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
83330dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
83430dcb7c9SBarry Smith         sends2[starts2[i]]++;
83530dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1];
83630dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = nownedsenders[node];
83732dcc486SBarry Smith         ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr);
83830dcb7c9SBarry Smith         cnt += nownedsenders[node];
83930dcb7c9SBarry Smith       }
84030dcb7c9SBarry Smith     }
84130dcb7c9SBarry Smith   }
84230dcb7c9SBarry Smith 
84330dcb7c9SBarry Smith   /* receive the message lengths */
84430dcb7c9SBarry Smith   nrecvs2 = nsends;
845785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&lens2);CHKERRQ(ierr);
846785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&starts3);CHKERRQ(ierr);
847785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr);
84830dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
849d44834fbSBarry Smith     ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr);
85030dcb7c9SBarry Smith   }
851d44834fbSBarry Smith 
8528a8e0b3aSBarry Smith   /* send the message lengths */
8538a8e0b3aSBarry Smith   for (i=0; i<nsends2; i++) {
8548a8e0b3aSBarry Smith     ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr);
8558a8e0b3aSBarry Smith   }
8568a8e0b3aSBarry Smith 
857d44834fbSBarry Smith   /* wait on receives of lens */
8580c468ba9SBarry Smith   if (nrecvs2) {
859785e854fSJed Brown     ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr);
860d44834fbSBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
861d44834fbSBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8620c468ba9SBarry Smith   }
863a2ea699eSBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
864d44834fbSBarry Smith 
86530dcb7c9SBarry Smith   starts3[0] = 0;
866d44834fbSBarry Smith   nt         = 0;
86730dcb7c9SBarry Smith   for (i=0; i<nrecvs2-1; i++) {
86830dcb7c9SBarry Smith     starts3[i+1] = starts3[i] + lens2[i];
869d44834fbSBarry Smith     nt          += lens2[i];
87030dcb7c9SBarry Smith   }
87176466f69SStefano Zampini   if (nrecvs2) nt += lens2[nrecvs2-1];
872d44834fbSBarry Smith 
873785e854fSJed Brown   ierr = PetscMalloc1((nt+1),&recvs2);CHKERRQ(ierr);
874785e854fSJed Brown   ierr = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr);
87552b72c4aSBarry Smith   for (i=0; i<nrecvs2; i++) {
87632dcc486SBarry Smith     ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr);
87730dcb7c9SBarry Smith   }
87830dcb7c9SBarry Smith 
87930dcb7c9SBarry Smith   /* send the messages */
880785e854fSJed Brown   ierr = PetscMalloc1((nsends2+1),&send_waits);CHKERRQ(ierr);
88130dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
88232dcc486SBarry Smith     ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr);
88330dcb7c9SBarry Smith   }
88430dcb7c9SBarry Smith 
88530dcb7c9SBarry Smith   /* wait on receives */
8860c468ba9SBarry Smith   if (nrecvs2) {
887785e854fSJed Brown     ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr);
88830dcb7c9SBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
88930dcb7c9SBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8900c468ba9SBarry Smith   }
89130dcb7c9SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
89230dcb7c9SBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
89330dcb7c9SBarry Smith 
89407b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
89530dcb7c9SBarry Smith     cnt = 0;
89630dcb7c9SBarry Smith     for (i=0; i<nrecvs2; i++) {
89730dcb7c9SBarry Smith       nt = recvs2[cnt++];
89830dcb7c9SBarry Smith       for (j=0; j<nt; j++) {
8997904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] local node %D number of subdomains %D: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr);
90030dcb7c9SBarry Smith         for (k=0; k<recvs2[cnt+1]; k++) {
9017904a332SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%D ",recvs2[cnt+2+k]);CHKERRQ(ierr);
90230dcb7c9SBarry Smith         }
90330dcb7c9SBarry Smith         cnt += 2 + recvs2[cnt+1];
90430dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
90530dcb7c9SBarry Smith       }
90630dcb7c9SBarry Smith     }
9070ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
90807b52d57SBarry Smith   } /* -----------------------------------  */
90930dcb7c9SBarry Smith 
91030dcb7c9SBarry Smith   /* count number subdomains for each local node */
911785e854fSJed Brown   ierr = PetscMalloc1(size,&nprocs);CHKERRQ(ierr);
91232dcc486SBarry Smith   ierr = PetscMemzero(nprocs,size*sizeof(PetscInt));CHKERRQ(ierr);
91330dcb7c9SBarry Smith   cnt  = 0;
91430dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
91530dcb7c9SBarry Smith     nt = recvs2[cnt++];
91630dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
917f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) nprocs[recvs2[cnt+2+k]]++;
91830dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
91930dcb7c9SBarry Smith     }
92030dcb7c9SBarry Smith   }
92130dcb7c9SBarry Smith   nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0);
92230dcb7c9SBarry Smith   *nproc    = nt;
923785e854fSJed Brown   ierr = PetscMalloc1((nt+1),procs);CHKERRQ(ierr);
924785e854fSJed Brown   ierr = PetscMalloc1((nt+1),numprocs);CHKERRQ(ierr);
925785e854fSJed Brown   ierr = PetscMalloc1((nt+1),indices);CHKERRQ(ierr);
9260298fd71SBarry Smith   for (i=0;i<nt+1;i++) (*indices)[i]=NULL;
927785e854fSJed Brown   ierr = PetscMalloc1(size,&bprocs);CHKERRQ(ierr);
92830dcb7c9SBarry Smith   cnt       = 0;
92930dcb7c9SBarry Smith   for (i=0; i<size; i++) {
93030dcb7c9SBarry Smith     if (nprocs[i] > 0) {
93130dcb7c9SBarry Smith       bprocs[i]        = cnt;
93230dcb7c9SBarry Smith       (*procs)[cnt]    = i;
93330dcb7c9SBarry Smith       (*numprocs)[cnt] = nprocs[i];
934785e854fSJed Brown       ierr             = PetscMalloc1(nprocs[i],&(*indices)[cnt]);CHKERRQ(ierr);
93530dcb7c9SBarry Smith       cnt++;
93630dcb7c9SBarry Smith     }
93730dcb7c9SBarry Smith   }
93830dcb7c9SBarry Smith 
93930dcb7c9SBarry Smith   /* make the list of subdomains for each nontrivial local node */
94032dcc486SBarry Smith   ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr);
94130dcb7c9SBarry Smith   cnt  = 0;
94230dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
94330dcb7c9SBarry Smith     nt = recvs2[cnt++];
94430dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
945f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt];
94630dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
94730dcb7c9SBarry Smith     }
94830dcb7c9SBarry Smith   }
94930dcb7c9SBarry Smith   ierr = PetscFree(bprocs);CHKERRQ(ierr);
95007b52d57SBarry Smith   ierr = PetscFree(recvs2);CHKERRQ(ierr);
95130dcb7c9SBarry Smith 
95207b52d57SBarry Smith   /* sort the node indexing by their global numbers */
95307b52d57SBarry Smith   nt = *nproc;
95407b52d57SBarry Smith   for (i=0; i<nt; i++) {
955785e854fSJed Brown     ierr = PetscMalloc1(((*numprocs)[i]),&tmp);CHKERRQ(ierr);
956f6e5521dSKarl Rupp     for (j=0; j<(*numprocs)[i]; j++) tmp[j] = lindices[(*indices)[i][j]];
95707b52d57SBarry Smith     ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr);
95807b52d57SBarry Smith     ierr = PetscFree(tmp);CHKERRQ(ierr);
95907b52d57SBarry Smith   }
96007b52d57SBarry Smith 
96107b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
96230dcb7c9SBarry Smith     nt = *nproc;
96330dcb7c9SBarry Smith     for (i=0; i<nt; i++) {
9647904a332SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %D number of indices %D: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr);
96530dcb7c9SBarry Smith       for (j=0; j<(*numprocs)[i]; j++) {
9667904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"%D ",(*indices)[i][j]);CHKERRQ(ierr);
96730dcb7c9SBarry Smith       }
96830dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
96930dcb7c9SBarry Smith     }
9700ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
97107b52d57SBarry Smith   } /* -----------------------------------  */
97230dcb7c9SBarry Smith 
97330dcb7c9SBarry Smith   /* wait on sends */
97430dcb7c9SBarry Smith   if (nsends2) {
975785e854fSJed Brown     ierr = PetscMalloc1(nsends2,&send_status);CHKERRQ(ierr);
97630dcb7c9SBarry Smith     ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr);
97730dcb7c9SBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
97830dcb7c9SBarry Smith   }
97930dcb7c9SBarry Smith 
98030dcb7c9SBarry Smith   ierr = PetscFree(starts3);CHKERRQ(ierr);
98130dcb7c9SBarry Smith   ierr = PetscFree(dest);CHKERRQ(ierr);
98230dcb7c9SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
9833677ff5aSBarry Smith 
984bc8ff85bSBarry Smith   ierr = PetscFree(nownedsenders);CHKERRQ(ierr);
985bc8ff85bSBarry Smith   ierr = PetscFree(ownedsenders);CHKERRQ(ierr);
986bc8ff85bSBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
98730dcb7c9SBarry Smith   ierr = PetscFree(starts2);CHKERRQ(ierr);
98830dcb7c9SBarry Smith   ierr = PetscFree(lens2);CHKERRQ(ierr);
98989d82c54SBarry Smith 
99089d82c54SBarry Smith   ierr = PetscFree(source);CHKERRQ(ierr);
99197f1f81fSBarry Smith   ierr = PetscFree(len);CHKERRQ(ierr);
99289d82c54SBarry Smith   ierr = PetscFree(recvs);CHKERRQ(ierr);
9933a96401aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
99430dcb7c9SBarry Smith   ierr = PetscFree(sends2);CHKERRQ(ierr);
99524cf384cSBarry Smith 
99624cf384cSBarry Smith   /* put the information about myself as the first entry in the list */
99724cf384cSBarry Smith   first_procs    = (*procs)[0];
99824cf384cSBarry Smith   first_numprocs = (*numprocs)[0];
99924cf384cSBarry Smith   first_indices  = (*indices)[0];
100024cf384cSBarry Smith   for (i=0; i<*nproc; i++) {
100124cf384cSBarry Smith     if ((*procs)[i] == rank) {
100224cf384cSBarry Smith       (*procs)[0]    = (*procs)[i];
100324cf384cSBarry Smith       (*numprocs)[0] = (*numprocs)[i];
100424cf384cSBarry Smith       (*indices)[0]  = (*indices)[i];
100524cf384cSBarry Smith       (*procs)[i]    = first_procs;
100624cf384cSBarry Smith       (*numprocs)[i] = first_numprocs;
100724cf384cSBarry Smith       (*indices)[i]  = first_indices;
100824cf384cSBarry Smith       break;
100924cf384cSBarry Smith     }
101024cf384cSBarry Smith   }
101189d82c54SBarry Smith   PetscFunctionReturn(0);
101289d82c54SBarry Smith }
101389d82c54SBarry Smith 
10144a2ae208SSatish Balay #undef __FUNCT__
10154a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo"
101607b52d57SBarry Smith /*@C
101707b52d57SBarry Smith     ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo()
101889d82c54SBarry Smith 
101907b52d57SBarry Smith     Collective on ISLocalToGlobalMapping
102007b52d57SBarry Smith 
102107b52d57SBarry Smith     Input Parameters:
102207b52d57SBarry Smith .   mapping - the mapping from local to global indexing
102307b52d57SBarry Smith 
102407b52d57SBarry Smith     Output Parameter:
102507b52d57SBarry Smith +   nproc - number of processors that are connected to this one
102607b52d57SBarry Smith .   proc - neighboring processors
102707b52d57SBarry Smith .   numproc - number of indices for each processor
102807b52d57SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
102907b52d57SBarry Smith 
103007b52d57SBarry Smith     Level: advanced
103107b52d57SBarry Smith 
103207b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
103307b52d57SBarry Smith           ISLocalToGlobalMappingGetInfo()
103407b52d57SBarry Smith @*/
10357087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
103607b52d57SBarry Smith {
10376849ba73SBarry Smith   PetscErrorCode ierr;
103832dcc486SBarry Smith   PetscInt       i;
103907b52d57SBarry Smith 
104007b52d57SBarry Smith   PetscFunctionBegin;
104105b42c5fSBarry Smith   ierr = PetscFree(*procs);CHKERRQ(ierr);
104205b42c5fSBarry Smith   ierr = PetscFree(*numprocs);CHKERRQ(ierr);
104300ff320aSBarry Smith   if (*indices) {
104405b42c5fSBarry Smith     ierr = PetscFree((*indices)[0]);CHKERRQ(ierr);
104500ff320aSBarry Smith     for (i=1; i<*nproc; i++) {
104605b42c5fSBarry Smith       ierr = PetscFree((*indices)[i]);CHKERRQ(ierr);
104707b52d57SBarry Smith     }
104807b52d57SBarry Smith     ierr = PetscFree(*indices);CHKERRQ(ierr);
104924cf384cSBarry Smith   }
105007b52d57SBarry Smith   PetscFunctionReturn(0);
105107b52d57SBarry Smith }
105286994e45SJed Brown 
105386994e45SJed Brown #undef __FUNCT__
105486994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices"
105586994e45SJed Brown /*@C
105686994e45SJed Brown    ISLocalToGlobalMappingGetIndices - Get global indices for every local point
105786994e45SJed Brown 
105886994e45SJed Brown    Not Collective
105986994e45SJed Brown 
106086994e45SJed Brown    Input Arguments:
106186994e45SJed Brown . ltog - local to global mapping
106286994e45SJed Brown 
106386994e45SJed Brown    Output Arguments:
106486994e45SJed Brown . array - array of indices
106586994e45SJed Brown 
106686994e45SJed Brown    Level: advanced
106786994e45SJed Brown 
106886994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices()
106986994e45SJed Brown @*/
10707087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
107186994e45SJed Brown {
107286994e45SJed Brown   PetscFunctionBegin;
107386994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
107486994e45SJed Brown   PetscValidPointer(array,2);
107586994e45SJed Brown   *array = ltog->indices;
107686994e45SJed Brown   PetscFunctionReturn(0);
107786994e45SJed Brown }
107886994e45SJed Brown 
107986994e45SJed Brown #undef __FUNCT__
108086994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices"
108186994e45SJed Brown /*@C
108286994e45SJed Brown    ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices()
108386994e45SJed Brown 
108486994e45SJed Brown    Not Collective
108586994e45SJed Brown 
108686994e45SJed Brown    Input Arguments:
108786994e45SJed Brown + ltog - local to global mapping
108886994e45SJed Brown - array - array of indices
108986994e45SJed Brown 
109086994e45SJed Brown    Level: advanced
109186994e45SJed Brown 
109286994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
109386994e45SJed Brown @*/
10947087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
109586994e45SJed Brown {
109686994e45SJed Brown   PetscFunctionBegin;
109786994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
109886994e45SJed Brown   PetscValidPointer(array,2);
109986994e45SJed Brown   if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
11000298fd71SBarry Smith   *array = NULL;
110186994e45SJed Brown   PetscFunctionReturn(0);
110286994e45SJed Brown }
1103f7efa3c7SJed Brown 
1104f7efa3c7SJed Brown #undef __FUNCT__
1105f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate"
1106f7efa3c7SJed Brown /*@C
1107f7efa3c7SJed Brown    ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings
1108f7efa3c7SJed Brown 
1109f7efa3c7SJed Brown    Not Collective
1110f7efa3c7SJed Brown 
1111f7efa3c7SJed Brown    Input Arguments:
1112f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate
1113f7efa3c7SJed Brown . n - number of mappings to concatenate
1114f7efa3c7SJed Brown - ltogs - local to global mappings
1115f7efa3c7SJed Brown 
1116f7efa3c7SJed Brown    Output Arguments:
1117f7efa3c7SJed Brown . ltogcat - new mapping
1118f7efa3c7SJed Brown 
1119f7efa3c7SJed Brown    Level: advanced
1120f7efa3c7SJed Brown 
1121f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate()
1122f7efa3c7SJed Brown @*/
1123f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat)
1124f7efa3c7SJed Brown {
1125f7efa3c7SJed Brown   PetscInt       i,cnt,m,*idx;
1126f7efa3c7SJed Brown   PetscErrorCode ierr;
1127f7efa3c7SJed Brown 
1128f7efa3c7SJed Brown   PetscFunctionBegin;
1129f7efa3c7SJed Brown   if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n);
1130f7efa3c7SJed Brown   if (n > 0) PetscValidPointer(ltogs,3);
1131f7efa3c7SJed Brown   for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3);
1132f7efa3c7SJed Brown   PetscValidPointer(ltogcat,4);
1133f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1134f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1135f7efa3c7SJed Brown     cnt += m;
1136f7efa3c7SJed Brown   }
1137785e854fSJed Brown   ierr = PetscMalloc1(cnt,&idx);CHKERRQ(ierr);
1138f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1139f7efa3c7SJed Brown     const PetscInt *subidx;
1140f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1141f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1142f7efa3c7SJed Brown     ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr);
1143f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1144f7efa3c7SJed Brown     cnt += m;
1145f7efa3c7SJed Brown   }
1146f7efa3c7SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr);
1147f7efa3c7SJed Brown   PetscFunctionReturn(0);
1148f7efa3c7SJed Brown }
114904a59952SBarry Smith 
115004a59952SBarry Smith 
1151