xref: /petsc/src/vec/is/utils/isltog.c (revision 743c9b42af3adbbf486f9d714e008338c9fbe0c4)
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 
84a2ae208SSatish Balay #undef __FUNCT__
9*743c9b42SStefano Zampini #define __FUNCT__ "ISG2LMapApply"
10186d4ecdSBarry Smith PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping mapping,PetscInt n,const PetscInt in[],PetscInt out[])
11186d4ecdSBarry Smith {
12186d4ecdSBarry Smith   PetscErrorCode ierr;
13*743c9b42SStefano Zampini   PetscInt       i,start,end;
14186d4ecdSBarry Smith 
15186d4ecdSBarry Smith   PetscFunctionBegin;
16186d4ecdSBarry Smith   if (!mapping->globals) {
17186d4ecdSBarry Smith     ierr = ISGlobalToLocalMappingApply(mapping,IS_GTOLM_MASK,0,0,0,0);CHKERRQ(ierr);
18186d4ecdSBarry Smith   }
19*743c9b42SStefano Zampini   start = mapping->globalstart;
20*743c9b42SStefano Zampini   end = mapping->globalend;
21186d4ecdSBarry Smith   for (i=0; i<n; i++) {
22186d4ecdSBarry Smith     if (in[i] < 0)          out[i] = in[i];
23186d4ecdSBarry Smith     else if (in[i] < start) out[i] = -1;
24186d4ecdSBarry Smith     else if (in[i] > end)   out[i] = -1;
25*743c9b42SStefano Zampini     else                    out[i] = mapping->globals[in[i] - start];
26186d4ecdSBarry Smith   }
27186d4ecdSBarry Smith   PetscFunctionReturn(0);
28186d4ecdSBarry Smith }
29186d4ecdSBarry Smith 
30186d4ecdSBarry Smith 
31186d4ecdSBarry Smith #undef __FUNCT__
324a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetSize"
333b9aefa3SBarry Smith /*@C
343b9aefa3SBarry Smith     ISLocalToGlobalMappingGetSize - Gets the local size of a local to global mapping.
353b9aefa3SBarry Smith 
363b9aefa3SBarry Smith     Not Collective
373b9aefa3SBarry Smith 
383b9aefa3SBarry Smith     Input Parameter:
393b9aefa3SBarry Smith .   ltog - local to global mapping
403b9aefa3SBarry Smith 
413b9aefa3SBarry Smith     Output Parameter:
423b9aefa3SBarry Smith .   n - the number of entries in the local mapping
433b9aefa3SBarry Smith 
443b9aefa3SBarry Smith     Level: advanced
453b9aefa3SBarry Smith 
46273d9f13SBarry Smith     Concepts: mapping^local to global
473b9aefa3SBarry Smith 
483b9aefa3SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
493b9aefa3SBarry Smith @*/
507087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping mapping,PetscInt *n)
513b9aefa3SBarry Smith {
523b9aefa3SBarry Smith   PetscFunctionBegin;
530700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
544482741eSBarry Smith   PetscValidIntPointer(n,2);
553b9aefa3SBarry Smith   *n = mapping->n;
563b9aefa3SBarry Smith   PetscFunctionReturn(0);
573b9aefa3SBarry Smith }
583b9aefa3SBarry Smith 
594a2ae208SSatish Balay #undef __FUNCT__
604a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingView"
615a5d4f66SBarry Smith /*@C
625a5d4f66SBarry Smith     ISLocalToGlobalMappingView - View a local to global mapping
635a5d4f66SBarry Smith 
64b9cd556bSLois Curfman McInnes     Not Collective
65b9cd556bSLois Curfman McInnes 
665a5d4f66SBarry Smith     Input Parameters:
673b9aefa3SBarry Smith +   ltog - local to global mapping
683b9aefa3SBarry Smith -   viewer - viewer
695a5d4f66SBarry Smith 
70a997ad1aSLois Curfman McInnes     Level: advanced
71a997ad1aSLois Curfman McInnes 
72273d9f13SBarry Smith     Concepts: mapping^local to global
735a5d4f66SBarry Smith 
745a5d4f66SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
755a5d4f66SBarry Smith @*/
767087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingView(ISLocalToGlobalMapping mapping,PetscViewer viewer)
775a5d4f66SBarry Smith {
7832dcc486SBarry Smith   PetscInt       i;
7932dcc486SBarry Smith   PetscMPIInt    rank;
80ace3abfcSBarry Smith   PetscBool      iascii;
816849ba73SBarry Smith   PetscErrorCode ierr;
825a5d4f66SBarry Smith 
835a5d4f66SBarry Smith   PetscFunctionBegin;
840700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
853050cee2SBarry Smith   if (!viewer) {
86ce94432eSBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mapping),&viewer);CHKERRQ(ierr);
873050cee2SBarry Smith   }
880700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
895a5d4f66SBarry Smith 
90ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)mapping),&rank);CHKERRQ(ierr);
91251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
9232077d6dSBarry Smith   if (iascii) {
937b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
945a5d4f66SBarry Smith     for (i=0; i<mapping->n; i++) {
95b0a32e0cSBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %d %d\n",rank,i,mapping->indices[i]);CHKERRQ(ierr);
966831982aSBarry Smith     }
97b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
987b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
997b23a99aSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name);
1005a5d4f66SBarry Smith   PetscFunctionReturn(0);
1015a5d4f66SBarry Smith }
1025a5d4f66SBarry Smith 
1034a2ae208SSatish Balay #undef __FUNCT__
1044a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS"
1051f428162SBarry Smith /*@
1062bdab257SBarry Smith     ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n)
1072bdab257SBarry Smith     ordering and a global parallel ordering.
1082bdab257SBarry Smith 
1090f5bd95cSBarry Smith     Not collective
110b9cd556bSLois Curfman McInnes 
111a997ad1aSLois Curfman McInnes     Input Parameter:
1128c03b21aSDmitry Karpeev .   is - index set containing the global numbers for each local number
1132bdab257SBarry Smith 
114a997ad1aSLois Curfman McInnes     Output Parameter:
1152bdab257SBarry Smith .   mapping - new mapping data structure
1162bdab257SBarry Smith 
117a997ad1aSLois Curfman McInnes     Level: advanced
118a997ad1aSLois Curfman McInnes 
119273d9f13SBarry Smith     Concepts: mapping^local to global
1202bdab257SBarry Smith 
1212bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
1222bdab257SBarry Smith @*/
1237087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping)
1242bdab257SBarry Smith {
1256849ba73SBarry Smith   PetscErrorCode ierr;
1265d0c19d7SBarry Smith   PetscInt       n;
1275d0c19d7SBarry Smith   const PetscInt *indices;
1282bdab257SBarry Smith   MPI_Comm       comm;
1293a40ed3dSBarry Smith 
1303a40ed3dSBarry Smith   PetscFunctionBegin;
1310700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,1);
1324482741eSBarry Smith   PetscValidPointer(mapping,2);
1332bdab257SBarry Smith 
1342bdab257SBarry Smith   ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr);
1353b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
1362bdab257SBarry Smith   ierr = ISGetIndices(is,&indices);CHKERRQ(ierr);
137d5ad8652SBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
1382bdab257SBarry Smith   ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr);
1393a40ed3dSBarry Smith   PetscFunctionReturn(0);
1402bdab257SBarry Smith }
1415a5d4f66SBarry Smith 
142a4d96a55SJed Brown #undef __FUNCT__
143a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF"
144a4d96a55SJed Brown /*@C
145a4d96a55SJed Brown     ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n)
146a4d96a55SJed Brown     ordering and a global parallel ordering.
147a4d96a55SJed Brown 
148a4d96a55SJed Brown     Collective
149a4d96a55SJed Brown 
150a4d96a55SJed Brown     Input Parameter:
151a4d96a55SJed Brown +   sf - star forest mapping contiguous local indices to (rank, offset)
152a4d96a55SJed Brown -   start - first global index on this process
153a4d96a55SJed Brown 
154a4d96a55SJed Brown     Output Parameter:
155a4d96a55SJed Brown .   mapping - new mapping data structure
156a4d96a55SJed Brown 
157a4d96a55SJed Brown     Level: advanced
158a4d96a55SJed Brown 
159a4d96a55SJed Brown     Concepts: mapping^local to global
160a4d96a55SJed Brown 
161a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
162a4d96a55SJed Brown @*/
163a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping)
164a4d96a55SJed Brown {
165a4d96a55SJed Brown   PetscErrorCode ierr;
166a4d96a55SJed Brown   PetscInt       i,maxlocal,nroots,nleaves,*globals,*ltog;
167a4d96a55SJed Brown   const PetscInt *ilocal;
168a4d96a55SJed Brown   MPI_Comm       comm;
169a4d96a55SJed Brown 
170a4d96a55SJed Brown   PetscFunctionBegin;
171a4d96a55SJed Brown   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
172a4d96a55SJed Brown   PetscValidPointer(mapping,3);
173a4d96a55SJed Brown 
174a4d96a55SJed Brown   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
1750298fd71SBarry Smith   ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,NULL);CHKERRQ(ierr);
176f6e5521dSKarl Rupp   if (ilocal) {
177f6e5521dSKarl Rupp     for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1);
178f6e5521dSKarl Rupp   }
179a4d96a55SJed Brown   else maxlocal = nleaves;
180a4d96a55SJed Brown   ierr = PetscMalloc(nroots*sizeof(PetscInt),&globals);CHKERRQ(ierr);
181a4d96a55SJed Brown   ierr = PetscMalloc(maxlocal*sizeof(PetscInt),&ltog);CHKERRQ(ierr);
182a4d96a55SJed Brown   for (i=0; i<nroots; i++) globals[i] = start + i;
183a4d96a55SJed Brown   for (i=0; i<maxlocal; i++) ltog[i] = -1;
184a4d96a55SJed Brown   ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
185a4d96a55SJed Brown   ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
186a4d96a55SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
187a4d96a55SJed Brown   ierr = PetscFree(globals);CHKERRQ(ierr);
188a4d96a55SJed Brown   PetscFunctionReturn(0);
189a4d96a55SJed Brown }
190b46b645bSBarry Smith 
1914a2ae208SSatish Balay #undef __FUNCT__
1924a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate"
193ba5bb76aSSatish Balay /*@
19490f02eecSBarry Smith     ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n)
19590f02eecSBarry Smith     ordering and a global parallel ordering.
1962362add9SBarry Smith 
19789d82c54SBarry Smith     Not Collective, but communicator may have more than one process
198b9cd556bSLois Curfman McInnes 
1992362add9SBarry Smith     Input Parameters:
20089d82c54SBarry Smith +   comm - MPI communicator
20190f02eecSBarry Smith .   n - the number of local elements
2029669e4d8SBarry Smith .   indices - the global index for each local element, these do not need to be in increasing order (sorted)
203d5ad8652SBarry Smith -   mode - see PetscCopyMode
2042362add9SBarry Smith 
205a997ad1aSLois Curfman McInnes     Output Parameter:
20690f02eecSBarry Smith .   mapping - new mapping data structure
2072362add9SBarry Smith 
208a997ad1aSLois Curfman McInnes     Level: advanced
209a997ad1aSLois Curfman McInnes 
210273d9f13SBarry Smith     Concepts: mapping^local to global
2112362add9SBarry Smith 
212d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
2132362add9SBarry Smith @*/
2147087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping)
2152362add9SBarry Smith {
2166849ba73SBarry Smith   PetscErrorCode ierr;
21732dcc486SBarry Smith   PetscInt       *in;
218b46b645bSBarry Smith 
219b46b645bSBarry Smith   PetscFunctionBegin;
22073911063SBarry Smith   if (n) PetscValidIntPointer(indices,3);
2214482741eSBarry Smith   PetscValidPointer(mapping,4);
222b46b645bSBarry Smith 
2230298fd71SBarry Smith   *mapping = NULL;
224519f805aSKarl Rupp #if !defined(PETSC_USE_DYNAMIC_LIBRARIES)
225607a6623SBarry Smith   ierr = ISInitializePackage();CHKERRQ(ierr);
2268e58c17dSMatthew Knepley #endif
2272362add9SBarry Smith 
22867c2884eSBarry Smith   ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,"ISLocalToGlobalMapping","Local to global mapping","IS",
22952e6d16bSBarry Smith                            cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr);
230d4bb536fSBarry Smith   (*mapping)->n = n;
231d4bb536fSBarry Smith   /*
232d4bb536fSBarry Smith     Do not create the global to local mapping. This is only created if
233d4bb536fSBarry Smith     ISGlobalToLocalMapping() is called
234d4bb536fSBarry Smith   */
235d4bb536fSBarry Smith   (*mapping)->globals = 0;
236d5ad8652SBarry Smith   if (mode == PETSC_COPY_VALUES) {
237d5ad8652SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&in);CHKERRQ(ierr);
238d5ad8652SBarry Smith     ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr);
239d5ad8652SBarry Smith     ierr = PetscLogObjectMemory(*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
240d5ad8652SBarry Smith     (*mapping)->indices = in;
241f6e5521dSKarl Rupp   } else if (mode == PETSC_OWN_POINTER) (*mapping)->indices = (PetscInt*)indices;
242f6e5521dSKarl Rupp   else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER");
2433a40ed3dSBarry Smith   PetscFunctionReturn(0);
2442362add9SBarry Smith }
2452362add9SBarry Smith 
2464a2ae208SSatish Balay #undef __FUNCT__
2474a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingBlock"
248bce096a4SSatish Balay /*@
249323b833fSBarry Smith     ISLocalToGlobalMappingBlock - Creates a blocked index version of an
250323b833fSBarry Smith        ISLocalToGlobalMapping that is appropriate for MatSetLocalToGlobalMappingBlock()
251323b833fSBarry Smith        and VecSetLocalToGlobalMappingBlock().
252323b833fSBarry Smith 
253323b833fSBarry Smith     Not Collective, but communicator may have more than one process
254323b833fSBarry Smith 
255323b833fSBarry Smith     Input Parameters:
256323b833fSBarry Smith +    inmap - original point-wise mapping
257323b833fSBarry Smith -    bs - block size
258323b833fSBarry Smith 
259323b833fSBarry Smith     Output Parameter:
26069eb54c3SBarry Smith .   outmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
261323b833fSBarry Smith 
262323b833fSBarry Smith     Level: advanced
263323b833fSBarry Smith 
264323b833fSBarry Smith     Concepts: mapping^local to global
265323b833fSBarry Smith 
266323b833fSBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
267323b833fSBarry Smith @*/
2687087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
269323b833fSBarry Smith {
2706849ba73SBarry Smith   PetscErrorCode ierr;
27132dcc486SBarry Smith   PetscInt       *ii,i,n;
272323b833fSBarry Smith 
273323b833fSBarry Smith   PetscFunctionBegin;
2740700a824SBarry Smith   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
275b2beed0aSJed Brown   PetscValidPointer(outmap,3);
276323b833fSBarry Smith   if (bs > 1) {
277323b833fSBarry Smith     n = inmap->n/bs;
278e32f2f54SBarry Smith     if (n*bs != inmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Pointwise mapping length is not divisible by block size");
27932dcc486SBarry Smith     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
280f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[bs*i]/bs;
281ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
282323b833fSBarry Smith   } else {
283323b833fSBarry Smith     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
284c3122656SLisandro Dalcin     *outmap = inmap;
285323b833fSBarry Smith   }
286323b833fSBarry Smith   PetscFunctionReturn(0);
287323b833fSBarry Smith }
288323b833fSBarry Smith 
2894a2ae208SSatish Balay #undef __FUNCT__
2908ab951edSJed Brown #define __FUNCT__ "ISLocalToGlobalMappingUnBlock"
291b2beed0aSJed Brown /*@
292b2beed0aSJed Brown     ISLocalToGlobalMappingUnBlock - Creates a scalar index version of a blocked
293b2beed0aSJed Brown        ISLocalToGlobalMapping
294b2beed0aSJed Brown 
295b2beed0aSJed Brown     Not Collective, but communicator may have more than one process
296b2beed0aSJed Brown 
297b2beed0aSJed Brown     Input Parameter:
298b2beed0aSJed Brown + inmap - block based mapping; the indices are relative to BLOCKS, not individual vector or matrix entries.
299b2beed0aSJed Brown - bs - block size
300b2beed0aSJed Brown 
301b2beed0aSJed Brown     Output Parameter:
302b2beed0aSJed Brown .   outmap - pointwise mapping
303b2beed0aSJed Brown 
304b2beed0aSJed Brown     Level: advanced
305b2beed0aSJed Brown 
306b2beed0aSJed Brown     Concepts: mapping^local to global
307b2beed0aSJed Brown 
308b2beed0aSJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingBlock()
309b2beed0aSJed Brown @*/
3107087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping inmap,PetscInt bs,ISLocalToGlobalMapping *outmap)
311b2beed0aSJed Brown {
312b2beed0aSJed Brown   PetscErrorCode ierr;
313b2beed0aSJed Brown   PetscInt       *ii,i,n;
314b2beed0aSJed Brown 
315b2beed0aSJed Brown   PetscFunctionBegin;
316b2beed0aSJed Brown   PetscValidHeaderSpecific(inmap,IS_LTOGM_CLASSID,1);
317b2beed0aSJed Brown   PetscValidPointer(outmap,2);
318b2beed0aSJed Brown   if (bs > 1) {
319b2beed0aSJed Brown     n    = inmap->n*bs;
320b2beed0aSJed Brown     ierr = PetscMalloc(n*sizeof(PetscInt),&ii);CHKERRQ(ierr);
321f6e5521dSKarl Rupp     for (i=0; i<n; i++) ii[i] = inmap->indices[i/bs]*bs + (i%bs);
322ce94432eSBarry Smith     ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)inmap),n,ii,PETSC_OWN_POINTER,outmap);CHKERRQ(ierr);
323b2beed0aSJed Brown   } else {
324b2beed0aSJed Brown     ierr    = PetscObjectReference((PetscObject)inmap);CHKERRQ(ierr);
325b2beed0aSJed Brown     *outmap = inmap;
326b2beed0aSJed Brown   }
327b2beed0aSJed Brown   PetscFunctionReturn(0);
328b2beed0aSJed Brown }
329b2beed0aSJed Brown 
330b2beed0aSJed Brown #undef __FUNCT__
3314a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy"
33290f02eecSBarry Smith /*@
33390f02eecSBarry Smith    ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n)
33490f02eecSBarry Smith    ordering and a global parallel ordering.
33590f02eecSBarry Smith 
3360f5bd95cSBarry Smith    Note Collective
337b9cd556bSLois Curfman McInnes 
33890f02eecSBarry Smith    Input Parameters:
33990f02eecSBarry Smith .  mapping - mapping data structure
34090f02eecSBarry Smith 
341a997ad1aSLois Curfman McInnes    Level: advanced
342a997ad1aSLois Curfman McInnes 
3433acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate()
34490f02eecSBarry Smith @*/
3456bf464f9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping)
34690f02eecSBarry Smith {
347dfbe8321SBarry Smith   PetscErrorCode ierr;
3485fd66863SKarl Rupp 
3493a40ed3dSBarry Smith   PetscFunctionBegin;
3506bf464f9SBarry Smith   if (!*mapping) PetscFunctionReturn(0);
3516bf464f9SBarry Smith   PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1);
352997056adSBarry Smith   if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);}
3536bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->indices);CHKERRQ(ierr);
3546bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->globals);CHKERRQ(ierr);
355d38fa0fbSBarry Smith   ierr     = PetscHeaderDestroy(mapping);CHKERRQ(ierr);
356992144d0SBarry Smith   *mapping = 0;
3573a40ed3dSBarry Smith   PetscFunctionReturn(0);
35890f02eecSBarry Smith }
35990f02eecSBarry Smith 
3604a2ae208SSatish Balay #undef __FUNCT__
3614a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS"
36290f02eecSBarry Smith /*@
3633acfe500SLois Curfman McInnes     ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering
3643acfe500SLois Curfman McInnes     a new index set using the global numbering defined in an ISLocalToGlobalMapping
3653acfe500SLois Curfman McInnes     context.
36690f02eecSBarry Smith 
367b9cd556bSLois Curfman McInnes     Not collective
368b9cd556bSLois Curfman McInnes 
36990f02eecSBarry Smith     Input Parameters:
370b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
371b9cd556bSLois Curfman McInnes -   is - index set in local numbering
37290f02eecSBarry Smith 
37390f02eecSBarry Smith     Output Parameters:
37490f02eecSBarry Smith .   newis - index set in global numbering
37590f02eecSBarry Smith 
376a997ad1aSLois Curfman McInnes     Level: advanced
377a997ad1aSLois Curfman McInnes 
378273d9f13SBarry Smith     Concepts: mapping^local to global
3793acfe500SLois Curfman McInnes 
38090f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
381d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply()
38290f02eecSBarry Smith @*/
3837087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis)
38490f02eecSBarry Smith {
3856849ba73SBarry Smith   PetscErrorCode ierr;
3865d0c19d7SBarry Smith   PetscInt       n,i,*idxmap,*idxout,Nmax = mapping->n;
3875d0c19d7SBarry Smith   const PetscInt *idxin;
3883a40ed3dSBarry Smith 
3893a40ed3dSBarry Smith   PetscFunctionBegin;
3900700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
3910700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,2);
3924482741eSBarry Smith   PetscValidPointer(newis,3);
39390f02eecSBarry Smith 
3943b9aefa3SBarry Smith   ierr   = ISGetLocalSize(is,&n);CHKERRQ(ierr);
39590f02eecSBarry Smith   ierr   = ISGetIndices(is,&idxin);CHKERRQ(ierr);
39690f02eecSBarry Smith   idxmap = mapping->indices;
39790f02eecSBarry Smith 
3987c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&idxout);CHKERRQ(ierr);
39990f02eecSBarry Smith   for (i=0; i<n; i++) {
400e32f2f54SBarry 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);
40190f02eecSBarry Smith     idxout[i] = idxmap[idxin[i]];
40290f02eecSBarry Smith   }
4033b9aefa3SBarry Smith   ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr);
40470b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr);
4053a40ed3dSBarry Smith   PetscFunctionReturn(0);
40690f02eecSBarry Smith }
40790f02eecSBarry Smith 
408afcb2eb5SJed Brown #undef __FUNCT__
409afcb2eb5SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingApply"
410b89cb25eSSatish Balay /*@
4113acfe500SLois Curfman McInnes    ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering
4123acfe500SLois Curfman McInnes    and converts them to the global numbering.
41390f02eecSBarry Smith 
414b9cd556bSLois Curfman McInnes    Not collective
415b9cd556bSLois Curfman McInnes 
416bb25748dSBarry Smith    Input Parameters:
417b9cd556bSLois Curfman McInnes +  mapping - the local to global mapping context
418bb25748dSBarry Smith .  N - number of integers
419b9cd556bSLois Curfman McInnes -  in - input indices in local numbering
420bb25748dSBarry Smith 
421bb25748dSBarry Smith    Output Parameter:
422bb25748dSBarry Smith .  out - indices in global numbering
423bb25748dSBarry Smith 
424b9cd556bSLois Curfman McInnes    Notes:
425b9cd556bSLois Curfman McInnes    The in and out array parameters may be identical.
426d4bb536fSBarry Smith 
427a997ad1aSLois Curfman McInnes    Level: advanced
428a997ad1aSLois Curfman McInnes 
429bb25748dSBarry Smith .seealso: ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
4300752156aSBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
431d4bb536fSBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
432bb25748dSBarry Smith 
433273d9f13SBarry Smith     Concepts: mapping^local to global
434afcb2eb5SJed Brown @*/
435afcb2eb5SJed Brown PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[])
436afcb2eb5SJed Brown {
437afcb2eb5SJed Brown   PetscInt       i,Nmax = mapping->n;
438afcb2eb5SJed Brown   const PetscInt *idx = mapping->indices;
439d4bb536fSBarry Smith 
440afcb2eb5SJed Brown   PetscFunctionBegin;
441afcb2eb5SJed Brown   for (i=0; i<N; i++) {
442afcb2eb5SJed Brown     if (in[i] < 0) {
443afcb2eb5SJed Brown       out[i] = in[i];
444afcb2eb5SJed Brown       continue;
445afcb2eb5SJed Brown     }
446afcb2eb5SJed 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);
447afcb2eb5SJed Brown     out[i] = idx[in[i]];
448afcb2eb5SJed Brown   }
449afcb2eb5SJed Brown   PetscFunctionReturn(0);
450afcb2eb5SJed Brown }
451d4bb536fSBarry Smith 
452d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/
453d4bb536fSBarry Smith 
4544a2ae208SSatish Balay #undef __FUNCT__
4554a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private"
456d4bb536fSBarry Smith /*
457d4bb536fSBarry Smith     Creates the global fields in the ISLocalToGlobalMapping structure
458d4bb536fSBarry Smith */
4596849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping)
460d4bb536fSBarry Smith {
4616849ba73SBarry Smith   PetscErrorCode ierr;
46232dcc486SBarry Smith   PetscInt       i,*idx = mapping->indices,n = mapping->n,end,start,*globals;
463d4bb536fSBarry Smith 
4643a40ed3dSBarry Smith   PetscFunctionBegin;
465d4bb536fSBarry Smith   end   = 0;
466ec268f7cSJed Brown   start = PETSC_MAX_INT;
467d4bb536fSBarry Smith 
468d4bb536fSBarry Smith   for (i=0; i<n; i++) {
469d4bb536fSBarry Smith     if (idx[i] < 0) continue;
470d4bb536fSBarry Smith     if (idx[i] < start) start = idx[i];
471d4bb536fSBarry Smith     if (idx[i] > end)   end   = idx[i];
472d4bb536fSBarry Smith   }
473d4bb536fSBarry Smith   if (start > end) {start = 0; end = -1;}
474d4bb536fSBarry Smith   mapping->globalstart = start;
475d4bb536fSBarry Smith   mapping->globalend   = end;
476d4bb536fSBarry Smith 
47732dcc486SBarry Smith   ierr             = PetscMalloc((end-start+2)*sizeof(PetscInt),&globals);CHKERRQ(ierr);
478b0a32e0cSBarry Smith   mapping->globals = globals;
479f6e5521dSKarl Rupp   for (i=0; i<end-start+1; i++) globals[i] = -1;
480d4bb536fSBarry Smith   for (i=0; i<n; i++) {
481d4bb536fSBarry Smith     if (idx[i] < 0) continue;
482d4bb536fSBarry Smith     globals[idx[i] - start] = i;
483d4bb536fSBarry Smith   }
484d4bb536fSBarry Smith 
48552e6d16bSBarry Smith   ierr = PetscLogObjectMemory(mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr);
4863a40ed3dSBarry Smith   PetscFunctionReturn(0);
487d4bb536fSBarry Smith }
488d4bb536fSBarry Smith 
4894a2ae208SSatish Balay #undef __FUNCT__
4904a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply"
491d4bb536fSBarry Smith /*@
492a997ad1aSLois Curfman McInnes     ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers
493a997ad1aSLois Curfman McInnes     specified with a global numbering.
494d4bb536fSBarry Smith 
495b9cd556bSLois Curfman McInnes     Not collective
496b9cd556bSLois Curfman McInnes 
497d4bb536fSBarry Smith     Input Parameters:
498b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
499d4bb536fSBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
500d4bb536fSBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
501d4bb536fSBarry Smith .   n - number of global indices to map
502b9cd556bSLois Curfman McInnes -   idx - global indices to map
503d4bb536fSBarry Smith 
504d4bb536fSBarry Smith     Output Parameters:
505b9cd556bSLois Curfman McInnes +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
506b9cd556bSLois Curfman McInnes -   idxout - local index of each global index, one must pass in an array long enough
507e182c471SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApply() with
5080298fd71SBarry Smith              idxout == NULL to determine the required length (returned in nout)
509e182c471SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApply()
510e182c471SBarry Smith              a second time to set the values.
511d4bb536fSBarry Smith 
512b9cd556bSLois Curfman McInnes     Notes:
5130298fd71SBarry Smith     Either nout or idxout may be NULL. idx and idxout may be identical.
514d4bb536fSBarry Smith 
5150f5bd95cSBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
5160f5bd95cSBarry Smith     array to compute these.
5170f5bd95cSBarry Smith 
518a997ad1aSLois Curfman McInnes     Level: advanced
519a997ad1aSLois Curfman McInnes 
52032fd6b96SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
52132fd6b96SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
52232fd6b96SBarry Smith 
523273d9f13SBarry Smith     Concepts: mapping^global to local
524d4bb536fSBarry Smith 
525d4bb536fSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
526d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy()
527d4bb536fSBarry Smith @*/
5287087cfbeSBarry Smith PetscErrorCode  ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
52932dcc486SBarry Smith                                   PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
530d4bb536fSBarry Smith {
53132dcc486SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end;
5326849ba73SBarry Smith   PetscErrorCode ierr;
533d4bb536fSBarry Smith 
5343a40ed3dSBarry Smith   PetscFunctionBegin;
5350700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
536d4bb536fSBarry Smith   if (!mapping->globals) {
537d4bb536fSBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
538d4bb536fSBarry Smith   }
539d4bb536fSBarry Smith   globals = mapping->globals;
540d4bb536fSBarry Smith   start   = mapping->globalstart;
541d4bb536fSBarry Smith   end     = mapping->globalend;
542d4bb536fSBarry Smith 
543d4bb536fSBarry Smith   if (type == IS_GTOLM_MASK) {
544d4bb536fSBarry Smith     if (idxout) {
545d4bb536fSBarry Smith       for (i=0; i<n; i++) {
546d4bb536fSBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
547d4bb536fSBarry Smith         else if (idx[i] < start) idxout[i] = -1;
548d4bb536fSBarry Smith         else if (idx[i] > end)   idxout[i] = -1;
549d4bb536fSBarry Smith         else                     idxout[i] = globals[idx[i] - start];
550d4bb536fSBarry Smith       }
551d4bb536fSBarry Smith     }
552d4bb536fSBarry Smith     if (nout) *nout = n;
553d4bb536fSBarry Smith   } else {
554d4bb536fSBarry Smith     if (idxout) {
555d4bb536fSBarry Smith       for (i=0; i<n; i++) {
556d4bb536fSBarry Smith         if (idx[i] < 0) continue;
557d4bb536fSBarry Smith         if (idx[i] < start) continue;
558d4bb536fSBarry Smith         if (idx[i] > end) continue;
559d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
560d4bb536fSBarry Smith         if (tmp < 0) continue;
561d4bb536fSBarry Smith         idxout[nf++] = tmp;
562d4bb536fSBarry Smith       }
563d4bb536fSBarry Smith     } else {
564d4bb536fSBarry Smith       for (i=0; i<n; i++) {
565d4bb536fSBarry Smith         if (idx[i] < 0) continue;
566d4bb536fSBarry Smith         if (idx[i] < start) continue;
567d4bb536fSBarry Smith         if (idx[i] > end) continue;
568d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
569d4bb536fSBarry Smith         if (tmp < 0) continue;
570d4bb536fSBarry Smith         nf++;
571d4bb536fSBarry Smith       }
572d4bb536fSBarry Smith     }
573d4bb536fSBarry Smith     if (nout) *nout = nf;
574d4bb536fSBarry Smith   }
5753a40ed3dSBarry Smith   PetscFunctionReturn(0);
576d4bb536fSBarry Smith }
57790f02eecSBarry Smith 
5784a2ae208SSatish Balay #undef __FUNCT__
5794a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetInfo"
58089d82c54SBarry Smith /*@C
58189d82c54SBarry Smith     ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and
58289d82c54SBarry Smith      each index shared by more than one processor
58389d82c54SBarry Smith 
58489d82c54SBarry Smith     Collective on ISLocalToGlobalMapping
58589d82c54SBarry Smith 
58689d82c54SBarry Smith     Input Parameters:
58789d82c54SBarry Smith .   mapping - the mapping from local to global indexing
58889d82c54SBarry Smith 
58989d82c54SBarry Smith     Output Parameter:
59089d82c54SBarry Smith +   nproc - number of processors that are connected to this one
59189d82c54SBarry Smith .   proc - neighboring processors
59207b52d57SBarry Smith .   numproc - number of indices for each subdomain (processor)
5933463a7baSJed Brown -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
59489d82c54SBarry Smith 
59589d82c54SBarry Smith     Level: advanced
59689d82c54SBarry Smith 
597273d9f13SBarry Smith     Concepts: mapping^local to global
59889d82c54SBarry Smith 
5992cfcea29SBarry Smith     Fortran Usage:
6002cfcea29SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
6012cfcea29SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
6022cfcea29SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
6032cfcea29SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
6042cfcea29SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
6052cfcea29SBarry Smith 
6062cfcea29SBarry Smith 
60707b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
60807b52d57SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
60989d82c54SBarry Smith @*/
6107087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
61189d82c54SBarry Smith {
6126849ba73SBarry Smith   PetscErrorCode ierr;
61397f1f81fSBarry Smith   PetscMPIInt    size,rank,tag1,tag2,tag3,*len,*source,imdex;
61432dcc486SBarry Smith   PetscInt       i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices;
61532dcc486SBarry Smith   PetscInt       *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc;
61697f1f81fSBarry Smith   PetscInt       cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned;
61732dcc486SBarry Smith   PetscInt       node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp;
61832dcc486SBarry Smith   PetscInt       first_procs,first_numprocs,*first_indices;
61989d82c54SBarry Smith   MPI_Request    *recv_waits,*send_waits;
62030dcb7c9SBarry Smith   MPI_Status     recv_status,*send_status,*recv_statuses;
621ce94432eSBarry Smith   MPI_Comm       comm;
622ace3abfcSBarry Smith   PetscBool      debug = PETSC_FALSE;
62389d82c54SBarry Smith 
62489d82c54SBarry Smith   PetscFunctionBegin;
6250700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
626ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)mapping,&comm);CHKERRQ(ierr);
62724cf384cSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
62824cf384cSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
62924cf384cSBarry Smith   if (size == 1) {
63024cf384cSBarry Smith     *nproc         = 0;
6310298fd71SBarry Smith     *procs         = NULL;
63232dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr);
6331e2105dcSBarry Smith     (*numprocs)[0] = 0;
63432dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr);
6350298fd71SBarry Smith     (*indices)[0]  = NULL;
63624cf384cSBarry Smith     PetscFunctionReturn(0);
63724cf384cSBarry Smith   }
63824cf384cSBarry Smith 
6390298fd71SBarry Smith   ierr = PetscOptionsGetBool(NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,NULL);CHKERRQ(ierr);
64007b52d57SBarry Smith 
6413677ff5aSBarry Smith   /*
6423677ff5aSBarry Smith     Notes on ISLocalToGlobalMappingGetInfo
6433677ff5aSBarry Smith 
6443677ff5aSBarry Smith     globally owned node - the nodes that have been assigned to this processor in global
6453677ff5aSBarry Smith            numbering, just for this routine.
6463677ff5aSBarry Smith 
6473677ff5aSBarry Smith     nontrivial globally owned node - node assigned to this processor that is on a subdomain
6483677ff5aSBarry Smith            boundary (i.e. is has more than one local owner)
6493677ff5aSBarry Smith 
6503677ff5aSBarry Smith     locally owned node - node that exists on this processors subdomain
6513677ff5aSBarry Smith 
6523677ff5aSBarry Smith     nontrivial locally owned node - node that is not in the interior (i.e. has more than one
6533677ff5aSBarry Smith            local subdomain
6543677ff5aSBarry Smith   */
65524cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr);
65624cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr);
65724cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr);
65889d82c54SBarry Smith 
65989d82c54SBarry Smith   for (i=0; i<n; i++) {
66089d82c54SBarry Smith     if (lindices[i] > max) max = lindices[i];
66189d82c54SBarry Smith   }
66232dcc486SBarry Smith   ierr   = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
66378058e43SBarry Smith   Ng++;
66489d82c54SBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
66589d82c54SBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
666bc8ff85bSBarry Smith   scale  = Ng/size + 1;
667a2e34c3dSBarry Smith   ng     = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng);
668caba0dd0SBarry Smith   rstart = scale*rank;
66989d82c54SBarry Smith 
67089d82c54SBarry Smith   /* determine ownership ranges of global indices */
6717c334f02SBarry Smith   ierr = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
67232dcc486SBarry Smith   ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
67389d82c54SBarry Smith 
67489d82c54SBarry Smith   /* determine owners of each local node  */
6757c334f02SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&owner);CHKERRQ(ierr);
67689d82c54SBarry Smith   for (i=0; i<n; i++) {
6773677ff5aSBarry Smith     proc             = lindices[i]/scale; /* processor that globally owns this index */
67827c402fcSBarry Smith     nprocs[2*proc+1] = 1;                 /* processor globally owns at least one of ours */
6793677ff5aSBarry Smith     owner[i]         = proc;
68027c402fcSBarry Smith     nprocs[2*proc]++;                     /* count of how many that processor globally owns of ours */
68189d82c54SBarry Smith   }
68227c402fcSBarry Smith   nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1];
6831e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of global owners for my local data %d\n",nsends);CHKERRQ(ierr);
68489d82c54SBarry Smith 
68589d82c54SBarry Smith   /* inform other processors of number of messages and max length*/
68627c402fcSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
6871e2582c4SBarry Smith   ierr = PetscInfo1(mapping,"Number of local owners for my global data %d\n",nrecvs);CHKERRQ(ierr);
68889d82c54SBarry Smith 
68989d82c54SBarry Smith   /* post receives for owned rows */
69032dcc486SBarry Smith   ierr = PetscMalloc((2*nrecvs+1)*(nmax+1)*sizeof(PetscInt),&recvs);CHKERRQ(ierr);
691b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
69289d82c54SBarry Smith   for (i=0; i<nrecvs; i++) {
69332dcc486SBarry Smith     ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr);
69489d82c54SBarry Smith   }
69589d82c54SBarry Smith 
69689d82c54SBarry Smith   /* pack messages containing lists of local nodes to owners */
69732dcc486SBarry Smith   ierr      = PetscMalloc((2*n+1)*sizeof(PetscInt),&sends);CHKERRQ(ierr);
69832dcc486SBarry Smith   ierr      = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
69989d82c54SBarry Smith   starts[0] = 0;
700f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
70189d82c54SBarry Smith   for (i=0; i<n; i++) {
70289d82c54SBarry Smith     sends[starts[owner[i]]++] = lindices[i];
70330dcb7c9SBarry Smith     sends[starts[owner[i]]++] = i;
70489d82c54SBarry Smith   }
70589d82c54SBarry Smith   ierr = PetscFree(owner);CHKERRQ(ierr);
70689d82c54SBarry Smith   starts[0] = 0;
707f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
70889d82c54SBarry Smith 
70989d82c54SBarry Smith   /* send the messages */
710b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
71132dcc486SBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(PetscInt),&dest);CHKERRQ(ierr);
71289d82c54SBarry Smith   cnt = 0;
71389d82c54SBarry Smith   for (i=0; i<size; i++) {
71427c402fcSBarry Smith     if (nprocs[2*i]) {
71532dcc486SBarry Smith       ierr      = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr);
71630dcb7c9SBarry Smith       dest[cnt] = i;
71789d82c54SBarry Smith       cnt++;
71889d82c54SBarry Smith     }
71989d82c54SBarry Smith   }
72089d82c54SBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
72189d82c54SBarry Smith 
72289d82c54SBarry Smith   /* wait on receives */
72397f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&source);CHKERRQ(ierr);
72497f1f81fSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(PetscMPIInt),&len);CHKERRQ(ierr);
72589d82c54SBarry Smith   cnt  = nrecvs;
72632dcc486SBarry Smith   ierr = PetscMalloc((ng+1)*sizeof(PetscInt),&nownedsenders);CHKERRQ(ierr);
72732dcc486SBarry Smith   ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr);
72889d82c54SBarry Smith   while (cnt) {
72989d82c54SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
73089d82c54SBarry Smith     /* unpack receives into our local space */
73132dcc486SBarry Smith     ierr          = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr);
73289d82c54SBarry Smith     source[imdex] = recv_status.MPI_SOURCE;
73330dcb7c9SBarry Smith     len[imdex]    = len[imdex]/2;
734caba0dd0SBarry Smith     /* count how many local owners for each of my global owned indices */
73530dcb7c9SBarry Smith     for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++;
73689d82c54SBarry Smith     cnt--;
73789d82c54SBarry Smith   }
73889d82c54SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
73989d82c54SBarry Smith 
74030dcb7c9SBarry Smith   /* count how many globally owned indices are on an edge multiplied by how many processors own them. */
741bc8ff85bSBarry Smith   nowned  = 0;
742bc8ff85bSBarry Smith   nownedm = 0;
743bc8ff85bSBarry Smith   for (i=0; i<ng; i++) {
744bc8ff85bSBarry Smith     if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;}
745bc8ff85bSBarry Smith   }
746bc8ff85bSBarry Smith 
747bc8ff85bSBarry Smith   /* create single array to contain rank of all local owners of each globally owned index */
74832dcc486SBarry Smith   ierr      = PetscMalloc((nownedm+1)*sizeof(PetscInt),&ownedsenders);CHKERRQ(ierr);
74932dcc486SBarry Smith   ierr      = PetscMalloc((ng+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
750bc8ff85bSBarry Smith   starts[0] = 0;
751bc8ff85bSBarry Smith   for (i=1; i<ng; i++) {
752bc8ff85bSBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
753bc8ff85bSBarry Smith     else starts[i] = starts[i-1];
754bc8ff85bSBarry Smith   }
755bc8ff85bSBarry Smith 
75630dcb7c9SBarry Smith   /* for each nontrival globally owned node list all arriving processors */
757bc8ff85bSBarry Smith   for (i=0; i<nrecvs; i++) {
758bc8ff85bSBarry Smith     for (j=0; j<len[i]; j++) {
75930dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
760f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) ownedsenders[starts[node]++] = source[i];
761bc8ff85bSBarry Smith     }
762bc8ff85bSBarry Smith   }
763bc8ff85bSBarry Smith 
76407b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
76530dcb7c9SBarry Smith     starts[0] = 0;
76630dcb7c9SBarry Smith     for (i=1; i<ng; i++) {
76730dcb7c9SBarry Smith       if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
76830dcb7c9SBarry Smith       else starts[i] = starts[i-1];
76930dcb7c9SBarry Smith     }
77030dcb7c9SBarry Smith     for (i=0; i<ng; i++) {
77130dcb7c9SBarry Smith       if (nownedsenders[i] > 1) {
77230dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] global node %d local owner processors: ",rank,i+rstart);CHKERRQ(ierr);
77330dcb7c9SBarry Smith         for (j=0; j<nownedsenders[i]; j++) {
77430dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",ownedsenders[starts[i]+j]);CHKERRQ(ierr);
77530dcb7c9SBarry Smith         }
77630dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
77730dcb7c9SBarry Smith       }
77830dcb7c9SBarry Smith     }
77930dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
78007b52d57SBarry Smith   } /* -----------------------------------  */
78130dcb7c9SBarry Smith 
7823677ff5aSBarry Smith   /* wait on original sends */
7833a96401aSBarry Smith   if (nsends) {
784b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
7853a96401aSBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
7863a96401aSBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
7873a96401aSBarry Smith   }
78889d82c54SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
7893a96401aSBarry Smith   ierr = PetscFree(sends);CHKERRQ(ierr);
7903677ff5aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
7913677ff5aSBarry Smith 
7923677ff5aSBarry Smith   /* pack messages to send back to local owners */
79330dcb7c9SBarry Smith   starts[0] = 0;
79430dcb7c9SBarry Smith   for (i=1; i<ng; i++) {
79530dcb7c9SBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
79630dcb7c9SBarry Smith     else starts[i] = starts[i-1];
79730dcb7c9SBarry Smith   }
79830dcb7c9SBarry Smith   nsends2 = nrecvs;
79932dcc486SBarry Smith   ierr    = PetscMalloc((nsends2+1)*sizeof(PetscInt),&nprocs);CHKERRQ(ierr); /* length of each message */
80030dcb7c9SBarry Smith   for (i=0; i<nrecvs; i++) {
80130dcb7c9SBarry Smith     nprocs[i] = 1;
80230dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
80330dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
804f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) nprocs[i] += 2 + nownedsenders[node];
80530dcb7c9SBarry Smith     }
80630dcb7c9SBarry Smith   }
807f6e5521dSKarl Rupp   nt = 0;
808f6e5521dSKarl Rupp   for (i=0; i<nsends2; i++) nt += nprocs[i];
809f6e5521dSKarl Rupp 
81032dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&sends2);CHKERRQ(ierr);
81132dcc486SBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(PetscInt),&starts2);CHKERRQ(ierr);
812f6e5521dSKarl Rupp 
813f6e5521dSKarl Rupp   starts2[0] = 0;
814f6e5521dSKarl Rupp   for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1];
81530dcb7c9SBarry Smith   /*
81630dcb7c9SBarry Smith      Each message is 1 + nprocs[i] long, and consists of
81730dcb7c9SBarry Smith        (0) the number of nodes being sent back
81830dcb7c9SBarry Smith        (1) the local node number,
81930dcb7c9SBarry Smith        (2) the number of processors sharing it,
82030dcb7c9SBarry Smith        (3) the processors sharing it
82130dcb7c9SBarry Smith   */
82230dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
82330dcb7c9SBarry Smith     cnt = 1;
82430dcb7c9SBarry Smith     sends2[starts2[i]] = 0;
82530dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
82630dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
82730dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
82830dcb7c9SBarry Smith         sends2[starts2[i]]++;
82930dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1];
83030dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = nownedsenders[node];
83132dcc486SBarry Smith         ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr);
83230dcb7c9SBarry Smith         cnt += nownedsenders[node];
83330dcb7c9SBarry Smith       }
83430dcb7c9SBarry Smith     }
83530dcb7c9SBarry Smith   }
83630dcb7c9SBarry Smith 
83730dcb7c9SBarry Smith   /* receive the message lengths */
83830dcb7c9SBarry Smith   nrecvs2 = nsends;
83932dcc486SBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&lens2);CHKERRQ(ierr);
84032dcc486SBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(PetscInt),&starts3);CHKERRQ(ierr);
841d44834fbSBarry Smith   ierr    = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
84230dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
843d44834fbSBarry Smith     ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr);
84430dcb7c9SBarry Smith   }
845d44834fbSBarry Smith 
8468a8e0b3aSBarry Smith   /* send the message lengths */
8478a8e0b3aSBarry Smith   for (i=0; i<nsends2; i++) {
8488a8e0b3aSBarry Smith     ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr);
8498a8e0b3aSBarry Smith   }
8508a8e0b3aSBarry Smith 
851d44834fbSBarry Smith   /* wait on receives of lens */
8520c468ba9SBarry Smith   if (nrecvs2) {
8530c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
854d44834fbSBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
855d44834fbSBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8560c468ba9SBarry Smith   }
857a2ea699eSBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
858d44834fbSBarry Smith 
85930dcb7c9SBarry Smith   starts3[0] = 0;
860d44834fbSBarry Smith   nt         = 0;
86130dcb7c9SBarry Smith   for (i=0; i<nrecvs2-1; i++) {
86230dcb7c9SBarry Smith     starts3[i+1] = starts3[i] + lens2[i];
863d44834fbSBarry Smith     nt          += lens2[i];
86430dcb7c9SBarry Smith   }
86576466f69SStefano Zampini   if (nrecvs2) nt += lens2[nrecvs2-1];
866d44834fbSBarry Smith 
86732dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),&recvs2);CHKERRQ(ierr);
868b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs2+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
86952b72c4aSBarry Smith   for (i=0; i<nrecvs2; i++) {
87032dcc486SBarry Smith     ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr);
87130dcb7c9SBarry Smith   }
87230dcb7c9SBarry Smith 
87330dcb7c9SBarry Smith   /* send the messages */
874b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends2+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
87530dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
87632dcc486SBarry Smith     ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr);
87730dcb7c9SBarry Smith   }
87830dcb7c9SBarry Smith 
87930dcb7c9SBarry Smith   /* wait on receives */
8800c468ba9SBarry Smith   if (nrecvs2) {
8810c468ba9SBarry Smith     ierr = PetscMalloc(nrecvs2*sizeof(MPI_Status),&recv_statuses);CHKERRQ(ierr);
88230dcb7c9SBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
88330dcb7c9SBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
8840c468ba9SBarry Smith   }
88530dcb7c9SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
88630dcb7c9SBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
88730dcb7c9SBarry Smith 
88807b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
88930dcb7c9SBarry Smith     cnt = 0;
89030dcb7c9SBarry Smith     for (i=0; i<nrecvs2; i++) {
89130dcb7c9SBarry Smith       nt = recvs2[cnt++];
89230dcb7c9SBarry Smith       for (j=0; j<nt; j++) {
89330dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] local node %d number of subdomains %d: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr);
89430dcb7c9SBarry Smith         for (k=0; k<recvs2[cnt+1]; k++) {
89530dcb7c9SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%d ",recvs2[cnt+2+k]);CHKERRQ(ierr);
89630dcb7c9SBarry Smith         }
89730dcb7c9SBarry Smith         cnt += 2 + recvs2[cnt+1];
89830dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
89930dcb7c9SBarry Smith       }
90030dcb7c9SBarry Smith     }
90130dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
90207b52d57SBarry Smith   } /* -----------------------------------  */
90330dcb7c9SBarry Smith 
90430dcb7c9SBarry Smith   /* count number subdomains for each local node */
90532dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
90632dcc486SBarry Smith   ierr = PetscMemzero(nprocs,size*sizeof(PetscInt));CHKERRQ(ierr);
90730dcb7c9SBarry Smith   cnt  = 0;
90830dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
90930dcb7c9SBarry Smith     nt = recvs2[cnt++];
91030dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
911f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) nprocs[recvs2[cnt+2+k]]++;
91230dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
91330dcb7c9SBarry Smith     }
91430dcb7c9SBarry Smith   }
91530dcb7c9SBarry Smith   nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0);
91630dcb7c9SBarry Smith   *nproc    = nt;
91732dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),procs);CHKERRQ(ierr);
91832dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt),numprocs);CHKERRQ(ierr);
91932dcc486SBarry Smith   ierr = PetscMalloc((nt+1)*sizeof(PetscInt*),indices);CHKERRQ(ierr);
9200298fd71SBarry Smith   for (i=0;i<nt+1;i++) (*indices)[i]=NULL;
92132dcc486SBarry Smith   ierr = PetscMalloc(size*sizeof(PetscInt),&bprocs);CHKERRQ(ierr);
92230dcb7c9SBarry Smith   cnt       = 0;
92330dcb7c9SBarry Smith   for (i=0; i<size; i++) {
92430dcb7c9SBarry Smith     if (nprocs[i] > 0) {
92530dcb7c9SBarry Smith       bprocs[i]        = cnt;
92630dcb7c9SBarry Smith       (*procs)[cnt]    = i;
92730dcb7c9SBarry Smith       (*numprocs)[cnt] = nprocs[i];
92832dcc486SBarry Smith       ierr             = PetscMalloc(nprocs[i]*sizeof(PetscInt),&(*indices)[cnt]);CHKERRQ(ierr);
92930dcb7c9SBarry Smith       cnt++;
93030dcb7c9SBarry Smith     }
93130dcb7c9SBarry Smith   }
93230dcb7c9SBarry Smith 
93330dcb7c9SBarry Smith   /* make the list of subdomains for each nontrivial local node */
93432dcc486SBarry Smith   ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr);
93530dcb7c9SBarry Smith   cnt  = 0;
93630dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
93730dcb7c9SBarry Smith     nt = recvs2[cnt++];
93830dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
939f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt];
94030dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
94130dcb7c9SBarry Smith     }
94230dcb7c9SBarry Smith   }
94330dcb7c9SBarry Smith   ierr = PetscFree(bprocs);CHKERRQ(ierr);
94407b52d57SBarry Smith   ierr = PetscFree(recvs2);CHKERRQ(ierr);
94530dcb7c9SBarry Smith 
94607b52d57SBarry Smith   /* sort the node indexing by their global numbers */
94707b52d57SBarry Smith   nt = *nproc;
94807b52d57SBarry Smith   for (i=0; i<nt; i++) {
94932dcc486SBarry Smith     ierr = PetscMalloc(((*numprocs)[i])*sizeof(PetscInt),&tmp);CHKERRQ(ierr);
950f6e5521dSKarl Rupp     for (j=0; j<(*numprocs)[i]; j++) tmp[j] = lindices[(*indices)[i][j]];
95107b52d57SBarry Smith     ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr);
95207b52d57SBarry Smith     ierr = PetscFree(tmp);CHKERRQ(ierr);
95307b52d57SBarry Smith   }
95407b52d57SBarry Smith 
95507b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
95630dcb7c9SBarry Smith     nt = *nproc;
95730dcb7c9SBarry Smith     for (i=0; i<nt; i++) {
95830dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %d number of indices %d: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr);
95930dcb7c9SBarry Smith       for (j=0; j<(*numprocs)[i]; j++) {
96030dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"%d ",(*indices)[i][j]);CHKERRQ(ierr);
96130dcb7c9SBarry Smith       }
96230dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
96330dcb7c9SBarry Smith     }
96430dcb7c9SBarry Smith     ierr = PetscSynchronizedFlush(comm);CHKERRQ(ierr);
96507b52d57SBarry Smith   } /* -----------------------------------  */
96630dcb7c9SBarry Smith 
96730dcb7c9SBarry Smith   /* wait on sends */
96830dcb7c9SBarry Smith   if (nsends2) {
969b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends2*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
97030dcb7c9SBarry Smith     ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr);
97130dcb7c9SBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
97230dcb7c9SBarry Smith   }
97330dcb7c9SBarry Smith 
97430dcb7c9SBarry Smith   ierr = PetscFree(starts3);CHKERRQ(ierr);
97530dcb7c9SBarry Smith   ierr = PetscFree(dest);CHKERRQ(ierr);
97630dcb7c9SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
9773677ff5aSBarry Smith 
978bc8ff85bSBarry Smith   ierr = PetscFree(nownedsenders);CHKERRQ(ierr);
979bc8ff85bSBarry Smith   ierr = PetscFree(ownedsenders);CHKERRQ(ierr);
980bc8ff85bSBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
98130dcb7c9SBarry Smith   ierr = PetscFree(starts2);CHKERRQ(ierr);
98230dcb7c9SBarry Smith   ierr = PetscFree(lens2);CHKERRQ(ierr);
98389d82c54SBarry Smith 
98489d82c54SBarry Smith   ierr = PetscFree(source);CHKERRQ(ierr);
98597f1f81fSBarry Smith   ierr = PetscFree(len);CHKERRQ(ierr);
98689d82c54SBarry Smith   ierr = PetscFree(recvs);CHKERRQ(ierr);
9873a96401aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
98830dcb7c9SBarry Smith   ierr = PetscFree(sends2);CHKERRQ(ierr);
98924cf384cSBarry Smith 
99024cf384cSBarry Smith   /* put the information about myself as the first entry in the list */
99124cf384cSBarry Smith   first_procs    = (*procs)[0];
99224cf384cSBarry Smith   first_numprocs = (*numprocs)[0];
99324cf384cSBarry Smith   first_indices  = (*indices)[0];
99424cf384cSBarry Smith   for (i=0; i<*nproc; i++) {
99524cf384cSBarry Smith     if ((*procs)[i] == rank) {
99624cf384cSBarry Smith       (*procs)[0]    = (*procs)[i];
99724cf384cSBarry Smith       (*numprocs)[0] = (*numprocs)[i];
99824cf384cSBarry Smith       (*indices)[0]  = (*indices)[i];
99924cf384cSBarry Smith       (*procs)[i]    = first_procs;
100024cf384cSBarry Smith       (*numprocs)[i] = first_numprocs;
100124cf384cSBarry Smith       (*indices)[i]  = first_indices;
100224cf384cSBarry Smith       break;
100324cf384cSBarry Smith     }
100424cf384cSBarry Smith   }
100589d82c54SBarry Smith   PetscFunctionReturn(0);
100689d82c54SBarry Smith }
100789d82c54SBarry Smith 
10084a2ae208SSatish Balay #undef __FUNCT__
10094a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo"
101007b52d57SBarry Smith /*@C
101107b52d57SBarry Smith     ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo()
101289d82c54SBarry Smith 
101307b52d57SBarry Smith     Collective on ISLocalToGlobalMapping
101407b52d57SBarry Smith 
101507b52d57SBarry Smith     Input Parameters:
101607b52d57SBarry Smith .   mapping - the mapping from local to global indexing
101707b52d57SBarry Smith 
101807b52d57SBarry Smith     Output Parameter:
101907b52d57SBarry Smith +   nproc - number of processors that are connected to this one
102007b52d57SBarry Smith .   proc - neighboring processors
102107b52d57SBarry Smith .   numproc - number of indices for each processor
102207b52d57SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
102307b52d57SBarry Smith 
102407b52d57SBarry Smith     Level: advanced
102507b52d57SBarry Smith 
102607b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
102707b52d57SBarry Smith           ISLocalToGlobalMappingGetInfo()
102807b52d57SBarry Smith @*/
10297087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
103007b52d57SBarry Smith {
10316849ba73SBarry Smith   PetscErrorCode ierr;
103232dcc486SBarry Smith   PetscInt       i;
103307b52d57SBarry Smith 
103407b52d57SBarry Smith   PetscFunctionBegin;
103505b42c5fSBarry Smith   ierr = PetscFree(*procs);CHKERRQ(ierr);
103605b42c5fSBarry Smith   ierr = PetscFree(*numprocs);CHKERRQ(ierr);
103700ff320aSBarry Smith   if (*indices) {
103805b42c5fSBarry Smith     ierr = PetscFree((*indices)[0]);CHKERRQ(ierr);
103900ff320aSBarry Smith     for (i=1; i<*nproc; i++) {
104005b42c5fSBarry Smith       ierr = PetscFree((*indices)[i]);CHKERRQ(ierr);
104107b52d57SBarry Smith     }
104207b52d57SBarry Smith     ierr = PetscFree(*indices);CHKERRQ(ierr);
104324cf384cSBarry Smith   }
104407b52d57SBarry Smith   PetscFunctionReturn(0);
104507b52d57SBarry Smith }
104686994e45SJed Brown 
104786994e45SJed Brown #undef __FUNCT__
104886994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices"
104986994e45SJed Brown /*@C
105086994e45SJed Brown    ISLocalToGlobalMappingGetIndices - Get global indices for every local point
105186994e45SJed Brown 
105286994e45SJed Brown    Not Collective
105386994e45SJed Brown 
105486994e45SJed Brown    Input Arguments:
105586994e45SJed Brown . ltog - local to global mapping
105686994e45SJed Brown 
105786994e45SJed Brown    Output Arguments:
105886994e45SJed Brown . array - array of indices
105986994e45SJed Brown 
106086994e45SJed Brown    Level: advanced
106186994e45SJed Brown 
106286994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices()
106386994e45SJed Brown @*/
10647087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
106586994e45SJed Brown {
106686994e45SJed Brown   PetscFunctionBegin;
106786994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
106886994e45SJed Brown   PetscValidPointer(array,2);
106986994e45SJed Brown   *array = ltog->indices;
107086994e45SJed Brown   PetscFunctionReturn(0);
107186994e45SJed Brown }
107286994e45SJed Brown 
107386994e45SJed Brown #undef __FUNCT__
107486994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices"
107586994e45SJed Brown /*@C
107686994e45SJed Brown    ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices()
107786994e45SJed Brown 
107886994e45SJed Brown    Not Collective
107986994e45SJed Brown 
108086994e45SJed Brown    Input Arguments:
108186994e45SJed Brown + ltog - local to global mapping
108286994e45SJed Brown - array - array of indices
108386994e45SJed Brown 
108486994e45SJed Brown    Level: advanced
108586994e45SJed Brown 
108686994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
108786994e45SJed Brown @*/
10887087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
108986994e45SJed Brown {
109086994e45SJed Brown   PetscFunctionBegin;
109186994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
109286994e45SJed Brown   PetscValidPointer(array,2);
109386994e45SJed Brown   if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
10940298fd71SBarry Smith   *array = NULL;
109586994e45SJed Brown   PetscFunctionReturn(0);
109686994e45SJed Brown }
1097f7efa3c7SJed Brown 
1098f7efa3c7SJed Brown #undef __FUNCT__
1099f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate"
1100f7efa3c7SJed Brown /*@C
1101f7efa3c7SJed Brown    ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings
1102f7efa3c7SJed Brown 
1103f7efa3c7SJed Brown    Not Collective
1104f7efa3c7SJed Brown 
1105f7efa3c7SJed Brown    Input Arguments:
1106f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate
1107f7efa3c7SJed Brown . n - number of mappings to concatenate
1108f7efa3c7SJed Brown - ltogs - local to global mappings
1109f7efa3c7SJed Brown 
1110f7efa3c7SJed Brown    Output Arguments:
1111f7efa3c7SJed Brown . ltogcat - new mapping
1112f7efa3c7SJed Brown 
1113f7efa3c7SJed Brown    Level: advanced
1114f7efa3c7SJed Brown 
1115f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate()
1116f7efa3c7SJed Brown @*/
1117f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat)
1118f7efa3c7SJed Brown {
1119f7efa3c7SJed Brown   PetscInt       i,cnt,m,*idx;
1120f7efa3c7SJed Brown   PetscErrorCode ierr;
1121f7efa3c7SJed Brown 
1122f7efa3c7SJed Brown   PetscFunctionBegin;
1123f7efa3c7SJed Brown   if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n);
1124f7efa3c7SJed Brown   if (n > 0) PetscValidPointer(ltogs,3);
1125f7efa3c7SJed Brown   for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3);
1126f7efa3c7SJed Brown   PetscValidPointer(ltogcat,4);
1127f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1128f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1129f7efa3c7SJed Brown     cnt += m;
1130f7efa3c7SJed Brown   }
1131f7efa3c7SJed Brown   ierr = PetscMalloc(cnt*sizeof(PetscInt),&idx);CHKERRQ(ierr);
1132f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1133f7efa3c7SJed Brown     const PetscInt *subidx;
1134f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1135f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1136f7efa3c7SJed Brown     ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr);
1137f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1138f7efa3c7SJed Brown     cnt += m;
1139f7efa3c7SJed Brown   }
1140f7efa3c7SJed Brown   ierr = ISLocalToGlobalMappingCreate(comm,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr);
1141f7efa3c7SJed Brown   PetscFunctionReturn(0);
1142f7efa3c7SJed Brown }
1143