xref: /petsc/src/vec/is/utils/isltog.c (revision cbc1caf078fb2bf42b82e0b5ac811b1101900405)
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"
34565245c5SBarry Smith /*@
35107e9a97SBarry 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:
43107e9a97SBarry Smith .   n - the number of entries in the local mapping, ISLocalToGlobalMappingGetIndices() returns an array of this length
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);
56107e9a97SBarry Smith   *n = mapping->bs*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) {
9498c3331eSBarry Smith     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)mapping,viewer);CHKERRQ(ierr);
957b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
965a5d4f66SBarry Smith     for (i=0; i<mapping->n; i++) {
977904a332SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D %D\n",rank,i,mapping->indices[i]);CHKERRQ(ierr);
986831982aSBarry Smith     }
99b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
1007b23a99aSBarry Smith     ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
1017b23a99aSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name);
1025a5d4f66SBarry Smith   PetscFunctionReturn(0);
1035a5d4f66SBarry Smith }
1045a5d4f66SBarry Smith 
1054a2ae208SSatish Balay #undef __FUNCT__
1064a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS"
1071f428162SBarry Smith /*@
1082bdab257SBarry Smith     ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n)
1092bdab257SBarry Smith     ordering and a global parallel ordering.
1102bdab257SBarry Smith 
1110f5bd95cSBarry Smith     Not collective
112b9cd556bSLois Curfman McInnes 
113a997ad1aSLois Curfman McInnes     Input Parameter:
1148c03b21aSDmitry Karpeev .   is - index set containing the global numbers for each local number
1152bdab257SBarry Smith 
116a997ad1aSLois Curfman McInnes     Output Parameter:
1172bdab257SBarry Smith .   mapping - new mapping data structure
1182bdab257SBarry Smith 
119f0413b6fSBarry Smith     Notes: the block size of the IS determines the block size of the mapping
120a997ad1aSLois Curfman McInnes     Level: advanced
121a997ad1aSLois Curfman McInnes 
122273d9f13SBarry Smith     Concepts: mapping^local to global
1232bdab257SBarry Smith 
1242bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate()
1252bdab257SBarry Smith @*/
1267087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping)
1272bdab257SBarry Smith {
1286849ba73SBarry Smith   PetscErrorCode ierr;
1293bbf0e92SBarry Smith   PetscInt       n,bs;
1305d0c19d7SBarry Smith   const PetscInt *indices;
1312bdab257SBarry Smith   MPI_Comm       comm;
1323bbf0e92SBarry Smith   PetscBool      isblock;
1333a40ed3dSBarry Smith 
1343a40ed3dSBarry Smith   PetscFunctionBegin;
1350700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,1);
1364482741eSBarry Smith   PetscValidPointer(mapping,2);
1372bdab257SBarry Smith 
1382bdab257SBarry Smith   ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr);
1393b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
1403bbf0e92SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)is,ISBLOCK,&isblock);CHKERRQ(ierr);
1416006e8d2SBarry Smith   if (!isblock) {
142f0413b6fSBarry Smith     ierr = ISGetIndices(is,&indices);CHKERRQ(ierr);
143f0413b6fSBarry Smith     ierr = ISLocalToGlobalMappingCreate(comm,1,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
1442bdab257SBarry Smith     ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr);
1456006e8d2SBarry Smith   } else {
1466006e8d2SBarry Smith     ierr = ISGetBlockSize(is,&bs);CHKERRQ(ierr);
147f0413b6fSBarry Smith     ierr = ISBlockGetIndices(is,&indices);CHKERRQ(ierr);
14828bc9809SBarry Smith     ierr = ISLocalToGlobalMappingCreate(comm,bs,n/bs,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr);
149f0413b6fSBarry Smith     ierr = ISBlockRestoreIndices(is,&indices);CHKERRQ(ierr);
1506006e8d2SBarry Smith   }
1513a40ed3dSBarry Smith   PetscFunctionReturn(0);
1522bdab257SBarry Smith }
1535a5d4f66SBarry Smith 
154a4d96a55SJed Brown #undef __FUNCT__
155a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF"
156a4d96a55SJed Brown /*@C
157a4d96a55SJed Brown     ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n)
158a4d96a55SJed Brown     ordering and a global parallel ordering.
159a4d96a55SJed Brown 
160a4d96a55SJed Brown     Collective
161a4d96a55SJed Brown 
162a4d96a55SJed Brown     Input Parameter:
163a4d96a55SJed Brown +   sf - star forest mapping contiguous local indices to (rank, offset)
164a4d96a55SJed Brown -   start - first global index on this process
165a4d96a55SJed Brown 
166a4d96a55SJed Brown     Output Parameter:
167a4d96a55SJed Brown .   mapping - new mapping data structure
168a4d96a55SJed Brown 
169a4d96a55SJed Brown     Level: advanced
170a4d96a55SJed Brown 
171a4d96a55SJed Brown     Concepts: mapping^local to global
172a4d96a55SJed Brown 
173a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS()
174a4d96a55SJed Brown @*/
175a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping)
176a4d96a55SJed Brown {
177a4d96a55SJed Brown   PetscErrorCode ierr;
178a4d96a55SJed Brown   PetscInt       i,maxlocal,nroots,nleaves,*globals,*ltog;
179a4d96a55SJed Brown   const PetscInt *ilocal;
180a4d96a55SJed Brown   MPI_Comm       comm;
181a4d96a55SJed Brown 
182a4d96a55SJed Brown   PetscFunctionBegin;
183a4d96a55SJed Brown   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
184a4d96a55SJed Brown   PetscValidPointer(mapping,3);
185a4d96a55SJed Brown 
186a4d96a55SJed Brown   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
1870298fd71SBarry Smith   ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,NULL);CHKERRQ(ierr);
188f6e5521dSKarl Rupp   if (ilocal) {
189f6e5521dSKarl Rupp     for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1);
190f6e5521dSKarl Rupp   }
191a4d96a55SJed Brown   else maxlocal = nleaves;
192785e854fSJed Brown   ierr = PetscMalloc1(nroots,&globals);CHKERRQ(ierr);
193785e854fSJed Brown   ierr = PetscMalloc1(maxlocal,&ltog);CHKERRQ(ierr);
194a4d96a55SJed Brown   for (i=0; i<nroots; i++) globals[i] = start + i;
195a4d96a55SJed Brown   for (i=0; i<maxlocal; i++) ltog[i] = -1;
196a4d96a55SJed Brown   ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
197a4d96a55SJed Brown   ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr);
198f0413b6fSBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,1,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr);
199a4d96a55SJed Brown   ierr = PetscFree(globals);CHKERRQ(ierr);
200a4d96a55SJed Brown   PetscFunctionReturn(0);
201a4d96a55SJed Brown }
202b46b645bSBarry Smith 
2034a2ae208SSatish Balay #undef __FUNCT__
20445b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetBlockSize"
20545b6f7e9SBarry Smith /*@
20645b6f7e9SBarry Smith     ISLocalToGlobalMappingGetBlockSize - Gets the blocksize of the mapping
20745b6f7e9SBarry Smith     ordering and a global parallel ordering.
20845b6f7e9SBarry Smith 
20945b6f7e9SBarry Smith     Not Collective
21045b6f7e9SBarry Smith 
21145b6f7e9SBarry Smith     Input Parameters:
21245b6f7e9SBarry Smith .   mapping - mapping data structure
21345b6f7e9SBarry Smith 
21445b6f7e9SBarry Smith     Output Parameter:
21545b6f7e9SBarry Smith .   bs - the blocksize
21645b6f7e9SBarry Smith 
21745b6f7e9SBarry Smith     Level: advanced
21845b6f7e9SBarry Smith 
21945b6f7e9SBarry Smith     Concepts: mapping^local to global
22045b6f7e9SBarry Smith 
22145b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
22245b6f7e9SBarry Smith @*/
22345b6f7e9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetBlockSize(ISLocalToGlobalMapping mapping,PetscInt *bs)
22445b6f7e9SBarry Smith {
22545b6f7e9SBarry Smith   PetscFunctionBegin;
226*cbc1caf0SMatthew G. Knepley   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
22745b6f7e9SBarry Smith   *bs = mapping->bs;
22845b6f7e9SBarry Smith   PetscFunctionReturn(0);
22945b6f7e9SBarry Smith }
23045b6f7e9SBarry Smith 
23145b6f7e9SBarry Smith #undef __FUNCT__
2324a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate"
233ba5bb76aSSatish Balay /*@
23490f02eecSBarry Smith     ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n)
23590f02eecSBarry Smith     ordering and a global parallel ordering.
2362362add9SBarry Smith 
23789d82c54SBarry Smith     Not Collective, but communicator may have more than one process
238b9cd556bSLois Curfman McInnes 
2392362add9SBarry Smith     Input Parameters:
24089d82c54SBarry Smith +   comm - MPI communicator
241f0413b6fSBarry Smith .   bs - the block size
24228bc9809SBarry Smith .   n - the number of local elements divided by the block size, or equivalently the number of block indices
24328bc9809SBarry Smith .   indices - the global index for each local element, these do not need to be in increasing order (sorted), these values should not be scaled (i.e. multiplied) by the blocksize bs
244d5ad8652SBarry Smith -   mode - see PetscCopyMode
2452362add9SBarry Smith 
246a997ad1aSLois Curfman McInnes     Output Parameter:
24790f02eecSBarry Smith .   mapping - new mapping data structure
2482362add9SBarry Smith 
249f0413b6fSBarry Smith     Notes: There is one integer value in indices per block and it represents the actual indices bs*idx + j, where j=0,..,bs-1
250a997ad1aSLois Curfman McInnes     Level: advanced
251a997ad1aSLois Curfman McInnes 
252273d9f13SBarry Smith     Concepts: mapping^local to global
2532362add9SBarry Smith 
254d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS()
2552362add9SBarry Smith @*/
256f0413b6fSBarry Smith PetscErrorCode  ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt bs,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping)
2572362add9SBarry Smith {
2586849ba73SBarry Smith   PetscErrorCode ierr;
25932dcc486SBarry Smith   PetscInt       *in;
260b46b645bSBarry Smith 
261b46b645bSBarry Smith   PetscFunctionBegin;
26273911063SBarry Smith   if (n) PetscValidIntPointer(indices,3);
2634482741eSBarry Smith   PetscValidPointer(mapping,4);
264b46b645bSBarry Smith 
2650298fd71SBarry Smith   *mapping = NULL;
266607a6623SBarry Smith   ierr = ISInitializePackage();CHKERRQ(ierr);
2672362add9SBarry Smith 
26867c2884eSBarry Smith   ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,"ISLocalToGlobalMapping","Local to global mapping","IS",
26952e6d16bSBarry Smith                            cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr);
270d4bb536fSBarry Smith   (*mapping)->n  = n;
271f0413b6fSBarry Smith   (*mapping)->bs = bs;
272d4bb536fSBarry Smith   /*
273d4bb536fSBarry Smith     Do not create the global to local mapping. This is only created if
274d4bb536fSBarry Smith     ISGlobalToLocalMapping() is called
275d4bb536fSBarry Smith   */
276d4bb536fSBarry Smith   (*mapping)->globals = 0;
277d5ad8652SBarry Smith   if (mode == PETSC_COPY_VALUES) {
278785e854fSJed Brown     ierr = PetscMalloc1(n,&in);CHKERRQ(ierr);
279d5ad8652SBarry Smith     ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr);
280d5ad8652SBarry Smith     (*mapping)->indices = in;
2816389a1a1SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
2826389a1a1SBarry Smith   } else if (mode == PETSC_OWN_POINTER) {
2836389a1a1SBarry Smith     (*mapping)->indices = (PetscInt*)indices;
2846389a1a1SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)*mapping,n*sizeof(PetscInt));CHKERRQ(ierr);
2856389a1a1SBarry Smith   }
286f6e5521dSKarl Rupp   else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER");
2873a40ed3dSBarry Smith   PetscFunctionReturn(0);
2882362add9SBarry Smith }
2892362add9SBarry Smith 
2904a2ae208SSatish Balay #undef __FUNCT__
2914a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy"
29290f02eecSBarry Smith /*@
29390f02eecSBarry Smith    ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n)
29490f02eecSBarry Smith    ordering and a global parallel ordering.
29590f02eecSBarry Smith 
2960f5bd95cSBarry Smith    Note Collective
297b9cd556bSLois Curfman McInnes 
29890f02eecSBarry Smith    Input Parameters:
29990f02eecSBarry Smith .  mapping - mapping data structure
30090f02eecSBarry Smith 
301a997ad1aSLois Curfman McInnes    Level: advanced
302a997ad1aSLois Curfman McInnes 
3033acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate()
30490f02eecSBarry Smith @*/
3056bf464f9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping)
30690f02eecSBarry Smith {
307dfbe8321SBarry Smith   PetscErrorCode ierr;
3085fd66863SKarl Rupp 
3093a40ed3dSBarry Smith   PetscFunctionBegin;
3106bf464f9SBarry Smith   if (!*mapping) PetscFunctionReturn(0);
3116bf464f9SBarry Smith   PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1);
312997056adSBarry Smith   if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);}
3136bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->indices);CHKERRQ(ierr);
3146bf464f9SBarry Smith   ierr     = PetscFree((*mapping)->globals);CHKERRQ(ierr);
315d38fa0fbSBarry Smith   ierr     = PetscHeaderDestroy(mapping);CHKERRQ(ierr);
316992144d0SBarry Smith   *mapping = 0;
3173a40ed3dSBarry Smith   PetscFunctionReturn(0);
31890f02eecSBarry Smith }
31990f02eecSBarry Smith 
3204a2ae208SSatish Balay #undef __FUNCT__
3214a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS"
32290f02eecSBarry Smith /*@
3233acfe500SLois Curfman McInnes     ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering
3243acfe500SLois Curfman McInnes     a new index set using the global numbering defined in an ISLocalToGlobalMapping
3253acfe500SLois Curfman McInnes     context.
32690f02eecSBarry Smith 
327b9cd556bSLois Curfman McInnes     Not collective
328b9cd556bSLois Curfman McInnes 
32990f02eecSBarry Smith     Input Parameters:
330b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
331b9cd556bSLois Curfman McInnes -   is - index set in local numbering
33290f02eecSBarry Smith 
33390f02eecSBarry Smith     Output Parameters:
33490f02eecSBarry Smith .   newis - index set in global numbering
33590f02eecSBarry Smith 
336a997ad1aSLois Curfman McInnes     Level: advanced
337a997ad1aSLois Curfman McInnes 
338273d9f13SBarry Smith     Concepts: mapping^local to global
3393acfe500SLois Curfman McInnes 
34090f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),
341d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply()
34290f02eecSBarry Smith @*/
3437087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis)
34490f02eecSBarry Smith {
3456849ba73SBarry Smith   PetscErrorCode ierr;
346e24637baSBarry Smith   PetscInt       n,*idxout;
3475d0c19d7SBarry Smith   const PetscInt *idxin;
3483a40ed3dSBarry Smith 
3493a40ed3dSBarry Smith   PetscFunctionBegin;
3500700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
3510700a824SBarry Smith   PetscValidHeaderSpecific(is,IS_CLASSID,2);
3524482741eSBarry Smith   PetscValidPointer(newis,3);
35390f02eecSBarry Smith 
3543b9aefa3SBarry Smith   ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr);
35590f02eecSBarry Smith   ierr = ISGetIndices(is,&idxin);CHKERRQ(ierr);
356785e854fSJed Brown   ierr = PetscMalloc1(n,&idxout);CHKERRQ(ierr);
357e24637baSBarry Smith   ierr = ISLocalToGlobalMappingApply(mapping,n,idxin,idxout);CHKERRQ(ierr);
3583b9aefa3SBarry Smith   ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr);
359543f3098SMatthew G. Knepley   ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is),n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr);
3603a40ed3dSBarry Smith   PetscFunctionReturn(0);
36190f02eecSBarry Smith }
36290f02eecSBarry Smith 
363afcb2eb5SJed Brown #undef __FUNCT__
364afcb2eb5SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingApply"
365b89cb25eSSatish Balay /*@
3663acfe500SLois Curfman McInnes    ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering
3673acfe500SLois Curfman McInnes    and converts them to the global numbering.
36890f02eecSBarry Smith 
369b9cd556bSLois Curfman McInnes    Not collective
370b9cd556bSLois Curfman McInnes 
371bb25748dSBarry Smith    Input Parameters:
372b9cd556bSLois Curfman McInnes +  mapping - the local to global mapping context
373bb25748dSBarry Smith .  N - number of integers
374b9cd556bSLois Curfman McInnes -  in - input indices in local numbering
375bb25748dSBarry Smith 
376bb25748dSBarry Smith    Output Parameter:
377bb25748dSBarry Smith .  out - indices in global numbering
378bb25748dSBarry Smith 
379b9cd556bSLois Curfman McInnes    Notes:
380b9cd556bSLois Curfman McInnes    The in and out array parameters may be identical.
381d4bb536fSBarry Smith 
382a997ad1aSLois Curfman McInnes    Level: advanced
383a997ad1aSLois Curfman McInnes 
38445b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingApplyBlock(), ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
3850752156aSBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
386d4bb536fSBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
387bb25748dSBarry Smith 
388273d9f13SBarry Smith     Concepts: mapping^local to global
389afcb2eb5SJed Brown @*/
390afcb2eb5SJed Brown PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[])
391afcb2eb5SJed Brown {
392*cbc1caf0SMatthew G. Knepley   PetscInt i,bs,Nmax;
39345b6f7e9SBarry Smith 
39445b6f7e9SBarry Smith   PetscFunctionBegin;
395*cbc1caf0SMatthew G. Knepley   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
396*cbc1caf0SMatthew G. Knepley   bs   = mapping->bs;
397*cbc1caf0SMatthew G. Knepley   Nmax = bs*mapping->n;
39845b6f7e9SBarry Smith   if (bs == 1) {
399*cbc1caf0SMatthew G. Knepley     const PetscInt *idx = mapping->indices;
40045b6f7e9SBarry Smith     for (i=0; i<N; i++) {
40145b6f7e9SBarry Smith       if (in[i] < 0) {
40245b6f7e9SBarry Smith         out[i] = in[i];
40345b6f7e9SBarry Smith         continue;
40445b6f7e9SBarry Smith       }
405e24637baSBarry Smith       if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %D too large %D (max) at %D",in[i],Nmax-1,i);
40645b6f7e9SBarry Smith       out[i] = idx[in[i]];
40745b6f7e9SBarry Smith     }
40845b6f7e9SBarry Smith   } else {
409*cbc1caf0SMatthew G. Knepley     const PetscInt *idx = mapping->indices;
41045b6f7e9SBarry Smith     for (i=0; i<N; i++) {
41145b6f7e9SBarry Smith       if (in[i] < 0) {
41245b6f7e9SBarry Smith         out[i] = in[i];
41345b6f7e9SBarry Smith         continue;
41445b6f7e9SBarry Smith       }
415e24637baSBarry Smith       if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %D too large %D (max) at %D",in[i],Nmax-1,i);
41645b6f7e9SBarry Smith       out[i] = idx[in[i]/bs]*bs + (in[i] % bs);
41745b6f7e9SBarry Smith     }
41845b6f7e9SBarry Smith   }
41945b6f7e9SBarry Smith   PetscFunctionReturn(0);
42045b6f7e9SBarry Smith }
42145b6f7e9SBarry Smith 
42245b6f7e9SBarry Smith #undef __FUNCT__
42345b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingApplyBlock"
42445b6f7e9SBarry Smith /*@
4256006e8d2SBarry Smith    ISLocalToGlobalMappingApplyBlock - Takes a list of integers in a local block numbering  and converts them to the global block numbering
42645b6f7e9SBarry Smith 
42745b6f7e9SBarry Smith    Not collective
42845b6f7e9SBarry Smith 
42945b6f7e9SBarry Smith    Input Parameters:
43045b6f7e9SBarry Smith +  mapping - the local to global mapping context
43145b6f7e9SBarry Smith .  N - number of integers
4326006e8d2SBarry Smith -  in - input indices in local block numbering
43345b6f7e9SBarry Smith 
43445b6f7e9SBarry Smith    Output Parameter:
4356006e8d2SBarry Smith .  out - indices in global block numbering
43645b6f7e9SBarry Smith 
43745b6f7e9SBarry Smith    Notes:
43845b6f7e9SBarry Smith    The in and out array parameters may be identical.
43945b6f7e9SBarry Smith 
4406006e8d2SBarry Smith    Example:
4416006e8d2SBarry Smith      If the index values are {0,1,6,7} set with a call to ISLocalToGlobalMappingCreate(PETSC_COMM_SELF,2,2,{0,3}) then the mapping applied to 0
4426006e8d2SBarry Smith      (the first block) would produce 0 and the mapping applied to 1 (the second block) would produce 3.
4436006e8d2SBarry Smith 
44445b6f7e9SBarry Smith    Level: advanced
44545b6f7e9SBarry Smith 
44645b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(),
44745b6f7e9SBarry Smith           ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(),
44845b6f7e9SBarry Smith           AOPetscToApplication(), ISGlobalToLocalMappingApply()
44945b6f7e9SBarry Smith 
45045b6f7e9SBarry Smith     Concepts: mapping^local to global
45145b6f7e9SBarry Smith @*/
45245b6f7e9SBarry Smith PetscErrorCode ISLocalToGlobalMappingApplyBlock(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[])
45345b6f7e9SBarry Smith {
454*cbc1caf0SMatthew G. Knepley 
455*cbc1caf0SMatthew G. Knepley   PetscFunctionBegin;
456*cbc1caf0SMatthew G. Knepley   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
457*cbc1caf0SMatthew G. Knepley   {
458afcb2eb5SJed Brown     PetscInt i,Nmax = mapping->n;
459afcb2eb5SJed Brown     const PetscInt *idx = mapping->indices;
460d4bb536fSBarry Smith 
461afcb2eb5SJed Brown     for (i=0; i<N; i++) {
462afcb2eb5SJed Brown       if (in[i] < 0) {
463afcb2eb5SJed Brown         out[i] = in[i];
464afcb2eb5SJed Brown         continue;
465afcb2eb5SJed Brown       }
466e24637baSBarry Smith       if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local block index %D too large %D (max) at %D",in[i],Nmax-1,i);
467afcb2eb5SJed Brown       out[i] = idx[in[i]];
468afcb2eb5SJed Brown     }
469*cbc1caf0SMatthew G. Knepley   }
470afcb2eb5SJed Brown   PetscFunctionReturn(0);
471afcb2eb5SJed Brown }
472d4bb536fSBarry Smith 
473d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/
474d4bb536fSBarry Smith 
4754a2ae208SSatish Balay #undef __FUNCT__
4764a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private"
477d4bb536fSBarry Smith /*
478d4bb536fSBarry Smith     Creates the global fields in the ISLocalToGlobalMapping structure
479d4bb536fSBarry Smith */
4806849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping)
481d4bb536fSBarry Smith {
4826849ba73SBarry Smith   PetscErrorCode ierr;
48332dcc486SBarry Smith   PetscInt       i,*idx = mapping->indices,n = mapping->n,end,start,*globals;
484d4bb536fSBarry Smith 
4853a40ed3dSBarry Smith   PetscFunctionBegin;
486d4bb536fSBarry Smith   end   = 0;
487ec268f7cSJed Brown   start = PETSC_MAX_INT;
488d4bb536fSBarry Smith 
489d4bb536fSBarry Smith   for (i=0; i<n; i++) {
490d4bb536fSBarry Smith     if (idx[i] < 0) continue;
491d4bb536fSBarry Smith     if (idx[i] < start) start = idx[i];
492d4bb536fSBarry Smith     if (idx[i] > end)   end   = idx[i];
493d4bb536fSBarry Smith   }
494d4bb536fSBarry Smith   if (start > end) {start = 0; end = -1;}
495d4bb536fSBarry Smith   mapping->globalstart = start;
496d4bb536fSBarry Smith   mapping->globalend   = end;
497d4bb536fSBarry Smith 
498785e854fSJed Brown   ierr             = PetscMalloc1((end-start+2),&globals);CHKERRQ(ierr);
499b0a32e0cSBarry Smith   mapping->globals = globals;
500f6e5521dSKarl Rupp   for (i=0; i<end-start+1; i++) globals[i] = -1;
501d4bb536fSBarry Smith   for (i=0; i<n; i++) {
502d4bb536fSBarry Smith     if (idx[i] < 0) continue;
503d4bb536fSBarry Smith     globals[idx[i] - start] = i;
504d4bb536fSBarry Smith   }
505d4bb536fSBarry Smith 
5063bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr);
5073a40ed3dSBarry Smith   PetscFunctionReturn(0);
508d4bb536fSBarry Smith }
509d4bb536fSBarry Smith 
5104a2ae208SSatish Balay #undef __FUNCT__
5114a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply"
512d4bb536fSBarry Smith /*@
513a997ad1aSLois Curfman McInnes     ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers
514a997ad1aSLois Curfman McInnes     specified with a global numbering.
515d4bb536fSBarry Smith 
516b9cd556bSLois Curfman McInnes     Not collective
517b9cd556bSLois Curfman McInnes 
518d4bb536fSBarry Smith     Input Parameters:
519b9cd556bSLois Curfman McInnes +   mapping - mapping between local and global numbering
520d4bb536fSBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
521d4bb536fSBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
522d4bb536fSBarry Smith .   n - number of global indices to map
523b9cd556bSLois Curfman McInnes -   idx - global indices to map
524d4bb536fSBarry Smith 
525d4bb536fSBarry Smith     Output Parameters:
526b9cd556bSLois Curfman McInnes +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
527b9cd556bSLois Curfman McInnes -   idxout - local index of each global index, one must pass in an array long enough
528e182c471SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApply() with
5290298fd71SBarry Smith              idxout == NULL to determine the required length (returned in nout)
530e182c471SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApply()
531e182c471SBarry Smith              a second time to set the values.
532d4bb536fSBarry Smith 
533b9cd556bSLois Curfman McInnes     Notes:
5340298fd71SBarry Smith     Either nout or idxout may be NULL. idx and idxout may be identical.
535d4bb536fSBarry Smith 
5360f5bd95cSBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
5370f5bd95cSBarry Smith     array to compute these.
5380f5bd95cSBarry Smith 
539a997ad1aSLois Curfman McInnes     Level: advanced
540a997ad1aSLois Curfman McInnes 
54132fd6b96SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
54232fd6b96SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
54332fd6b96SBarry Smith 
544273d9f13SBarry Smith     Concepts: mapping^global to local
545d4bb536fSBarry Smith 
5469d90f715SBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISGlobalToLocalMappingApplyBlock(), ISLocalToGlobalMappingCreate(),
547d4bb536fSBarry Smith           ISLocalToGlobalMappingDestroy()
548d4bb536fSBarry Smith @*/
5497087cfbeSBarry Smith PetscErrorCode  ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
55032dcc486SBarry Smith                                             PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
551d4bb536fSBarry Smith {
5529d90f715SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end,bs;
5539d90f715SBarry Smith   PetscErrorCode ierr;
5549d90f715SBarry Smith 
5559d90f715SBarry Smith   PetscFunctionBegin;
5569d90f715SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
5579d90f715SBarry Smith   if (!mapping->globals) {
5589d90f715SBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
5599d90f715SBarry Smith   }
5609d90f715SBarry Smith   globals = mapping->globals;
5619d90f715SBarry Smith   start   = mapping->globalstart;
5629d90f715SBarry Smith   end     = mapping->globalend;
5639d90f715SBarry Smith   bs      = mapping->bs;
5649d90f715SBarry Smith 
5659d90f715SBarry Smith   if (type == IS_GTOLM_MASK) {
5669d90f715SBarry Smith     if (idxout) {
5679d90f715SBarry Smith       for (i=0; i<n; i++) {
5689d90f715SBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
5699d90f715SBarry Smith         else if (idx[i] < bs*start) idxout[i] = -1;
5709d90f715SBarry Smith         else if (idx[i] > bs*end)   idxout[i] = -1;
5719d90f715SBarry Smith         else                        idxout[i] = bs*globals[idx[i]/bs - start] + (idx[i] % bs);
5729d90f715SBarry Smith       }
5739d90f715SBarry Smith     }
5749d90f715SBarry Smith     if (nout) *nout = n;
5759d90f715SBarry Smith   } else {
5769d90f715SBarry Smith     if (idxout) {
5779d90f715SBarry Smith       for (i=0; i<n; i++) {
5789d90f715SBarry Smith         if (idx[i] < 0) continue;
5799d90f715SBarry Smith         if (idx[i] < bs*start) continue;
5809d90f715SBarry Smith         if (idx[i] > bs*end) continue;
5819d90f715SBarry Smith         tmp = bs*globals[idx[i]/bs - start] + (idx[i] % bs);
5829d90f715SBarry Smith         if (tmp < 0) continue;
5839d90f715SBarry Smith         idxout[nf++] = tmp;
5849d90f715SBarry Smith       }
5859d90f715SBarry Smith     } else {
5869d90f715SBarry Smith       for (i=0; i<n; i++) {
5879d90f715SBarry Smith         if (idx[i] < 0) continue;
5889d90f715SBarry Smith         if (idx[i] < bs*start) continue;
5899d90f715SBarry Smith         if (idx[i] > bs*end) continue;
5909d90f715SBarry Smith         tmp = bs*globals[idx[i]/bs - start] + (idx[i] % bs);
5919d90f715SBarry Smith         if (tmp < 0) continue;
5929d90f715SBarry Smith         nf++;
5939d90f715SBarry Smith       }
5949d90f715SBarry Smith     }
5959d90f715SBarry Smith     if (nout) *nout = nf;
5969d90f715SBarry Smith   }
5979d90f715SBarry Smith   PetscFunctionReturn(0);
5989d90f715SBarry Smith }
5999d90f715SBarry Smith 
6009d90f715SBarry Smith #undef __FUNCT__
6019d90f715SBarry Smith #define __FUNCT__ "ISGlobalToLocalMappingApplyBlock"
6029d90f715SBarry Smith /*@
6039d90f715SBarry Smith     ISGlobalToLocalMappingApplyBlock - Provides the local block numbering for a list of integers
6049d90f715SBarry Smith     specified with a block global numbering.
6059d90f715SBarry Smith 
6069d90f715SBarry Smith     Not collective
6079d90f715SBarry Smith 
6089d90f715SBarry Smith     Input Parameters:
6099d90f715SBarry Smith +   mapping - mapping between local and global numbering
6109d90f715SBarry Smith .   type - IS_GTOLM_MASK - replaces global indices with no local value with -1
6119d90f715SBarry Smith            IS_GTOLM_DROP - drops the indices with no local value from the output list
6129d90f715SBarry Smith .   n - number of global indices to map
6139d90f715SBarry Smith -   idx - global indices to map
6149d90f715SBarry Smith 
6159d90f715SBarry Smith     Output Parameters:
6169d90f715SBarry Smith +   nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n)
6179d90f715SBarry Smith -   idxout - local index of each global index, one must pass in an array long enough
6189d90f715SBarry Smith              to hold all the indices. You can call ISGlobalToLocalMappingApplyBlock() with
6199d90f715SBarry Smith              idxout == NULL to determine the required length (returned in nout)
6209d90f715SBarry Smith              and then allocate the required space and call ISGlobalToLocalMappingApplyBlock()
6219d90f715SBarry Smith              a second time to set the values.
6229d90f715SBarry Smith 
6239d90f715SBarry Smith     Notes:
6249d90f715SBarry Smith     Either nout or idxout may be NULL. idx and idxout may be identical.
6259d90f715SBarry Smith 
6269d90f715SBarry Smith     This is not scalable in memory usage. Each processor requires O(Nglobal) size
6279d90f715SBarry Smith     array to compute these.
6289d90f715SBarry Smith 
6299d90f715SBarry Smith     Level: advanced
6309d90f715SBarry Smith 
6319d90f715SBarry Smith     Developer Note: The manual page states that idx and idxout may be identical but the calling
6329d90f715SBarry Smith        sequence declares idx as const so it cannot be the same as idxout.
6339d90f715SBarry Smith 
6349d90f715SBarry Smith     Concepts: mapping^global to local
6359d90f715SBarry Smith 
6369d90f715SBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISGlobalToLocalMappingApply(), ISLocalToGlobalMappingCreate(),
6379d90f715SBarry Smith           ISLocalToGlobalMappingDestroy()
6389d90f715SBarry Smith @*/
6399d90f715SBarry Smith PetscErrorCode  ISGlobalToLocalMappingApplyBlock(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type,
6409d90f715SBarry Smith                                   PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[])
6419d90f715SBarry Smith {
64232dcc486SBarry Smith   PetscInt       i,*globals,nf = 0,tmp,start,end;
6436849ba73SBarry Smith   PetscErrorCode ierr;
644d4bb536fSBarry Smith 
6453a40ed3dSBarry Smith   PetscFunctionBegin;
6460700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
647d4bb536fSBarry Smith   if (!mapping->globals) {
648d4bb536fSBarry Smith     ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr);
649d4bb536fSBarry Smith   }
650d4bb536fSBarry Smith   globals = mapping->globals;
651d4bb536fSBarry Smith   start   = mapping->globalstart;
652d4bb536fSBarry Smith   end     = mapping->globalend;
653d4bb536fSBarry Smith 
654d4bb536fSBarry Smith   if (type == IS_GTOLM_MASK) {
655d4bb536fSBarry Smith     if (idxout) {
656d4bb536fSBarry Smith       for (i=0; i<n; i++) {
657d4bb536fSBarry Smith         if (idx[i] < 0) idxout[i] = idx[i];
658d4bb536fSBarry Smith         else if (idx[i] < start) idxout[i] = -1;
659d4bb536fSBarry Smith         else if (idx[i] > end)   idxout[i] = -1;
660d4bb536fSBarry Smith         else                     idxout[i] = globals[idx[i] - start];
661d4bb536fSBarry Smith       }
662d4bb536fSBarry Smith     }
663d4bb536fSBarry Smith     if (nout) *nout = n;
664d4bb536fSBarry Smith   } else {
665d4bb536fSBarry Smith     if (idxout) {
666d4bb536fSBarry Smith       for (i=0; i<n; i++) {
667d4bb536fSBarry Smith         if (idx[i] < 0) continue;
668d4bb536fSBarry Smith         if (idx[i] < start) continue;
669d4bb536fSBarry Smith         if (idx[i] > end) continue;
670d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
671d4bb536fSBarry Smith         if (tmp < 0) continue;
672d4bb536fSBarry Smith         idxout[nf++] = tmp;
673d4bb536fSBarry Smith       }
674d4bb536fSBarry Smith     } else {
675d4bb536fSBarry Smith       for (i=0; i<n; i++) {
676d4bb536fSBarry Smith         if (idx[i] < 0) continue;
677d4bb536fSBarry Smith         if (idx[i] < start) continue;
678d4bb536fSBarry Smith         if (idx[i] > end) continue;
679d4bb536fSBarry Smith         tmp = globals[idx[i] - start];
680d4bb536fSBarry Smith         if (tmp < 0) continue;
681d4bb536fSBarry Smith         nf++;
682d4bb536fSBarry Smith       }
683d4bb536fSBarry Smith     }
684d4bb536fSBarry Smith     if (nout) *nout = nf;
685d4bb536fSBarry Smith   }
6863a40ed3dSBarry Smith   PetscFunctionReturn(0);
687d4bb536fSBarry Smith }
68890f02eecSBarry Smith 
6894a2ae208SSatish Balay #undef __FUNCT__
6906a818285SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetBlockInfo"
69189d82c54SBarry Smith /*@C
6926a818285SBarry Smith     ISLocalToGlobalMappingGetBlockInfo - Gets the neighbor information for each processor and
69389d82c54SBarry Smith      each index shared by more than one processor
69489d82c54SBarry Smith 
69589d82c54SBarry Smith     Collective on ISLocalToGlobalMapping
69689d82c54SBarry Smith 
69789d82c54SBarry Smith     Input Parameters:
69889d82c54SBarry Smith .   mapping - the mapping from local to global indexing
69989d82c54SBarry Smith 
70089d82c54SBarry Smith     Output Parameter:
70189d82c54SBarry Smith +   nproc - number of processors that are connected to this one
70289d82c54SBarry Smith .   proc - neighboring processors
70307b52d57SBarry Smith .   numproc - number of indices for each subdomain (processor)
7043463a7baSJed Brown -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
70589d82c54SBarry Smith 
70689d82c54SBarry Smith     Level: advanced
70789d82c54SBarry Smith 
708273d9f13SBarry Smith     Concepts: mapping^local to global
70989d82c54SBarry Smith 
7102cfcea29SBarry Smith     Fortran Usage:
7112cfcea29SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
7122cfcea29SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
7132cfcea29SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
7142cfcea29SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
7152cfcea29SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
7162cfcea29SBarry Smith 
7172cfcea29SBarry Smith 
71807b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
71907b52d57SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
72089d82c54SBarry Smith @*/
7216a818285SBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetBlockInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
72289d82c54SBarry Smith {
7236849ba73SBarry Smith   PetscErrorCode ierr;
72497f1f81fSBarry Smith   PetscMPIInt    size,rank,tag1,tag2,tag3,*len,*source,imdex;
72532dcc486SBarry Smith   PetscInt       i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices;
72632dcc486SBarry Smith   PetscInt       *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc;
72797f1f81fSBarry Smith   PetscInt       cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned;
72832dcc486SBarry Smith   PetscInt       node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp;
72932dcc486SBarry Smith   PetscInt       first_procs,first_numprocs,*first_indices;
73089d82c54SBarry Smith   MPI_Request    *recv_waits,*send_waits;
73130dcb7c9SBarry Smith   MPI_Status     recv_status,*send_status,*recv_statuses;
732ce94432eSBarry Smith   MPI_Comm       comm;
733ace3abfcSBarry Smith   PetscBool      debug = PETSC_FALSE;
73489d82c54SBarry Smith 
73589d82c54SBarry Smith   PetscFunctionBegin;
7360700a824SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
737ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)mapping,&comm);CHKERRQ(ierr);
73824cf384cSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
73924cf384cSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
74024cf384cSBarry Smith   if (size == 1) {
74124cf384cSBarry Smith     *nproc         = 0;
7420298fd71SBarry Smith     *procs         = NULL;
74332dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr);
7441e2105dcSBarry Smith     (*numprocs)[0] = 0;
74532dcc486SBarry Smith     ierr           = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr);
7460298fd71SBarry Smith     (*indices)[0]  = NULL;
74724cf384cSBarry Smith     PetscFunctionReturn(0);
74824cf384cSBarry Smith   }
74924cf384cSBarry Smith 
7500298fd71SBarry Smith   ierr = PetscOptionsGetBool(NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,NULL);CHKERRQ(ierr);
75107b52d57SBarry Smith 
7523677ff5aSBarry Smith   /*
7536a818285SBarry Smith     Notes on ISLocalToGlobalMappingGetBlockInfo
7543677ff5aSBarry Smith 
7553677ff5aSBarry Smith     globally owned node - the nodes that have been assigned to this processor in global
7563677ff5aSBarry Smith            numbering, just for this routine.
7573677ff5aSBarry Smith 
7583677ff5aSBarry Smith     nontrivial globally owned node - node assigned to this processor that is on a subdomain
7593677ff5aSBarry Smith            boundary (i.e. is has more than one local owner)
7603677ff5aSBarry Smith 
7613677ff5aSBarry Smith     locally owned node - node that exists on this processors subdomain
7623677ff5aSBarry Smith 
7633677ff5aSBarry Smith     nontrivial locally owned node - node that is not in the interior (i.e. has more than one
7643677ff5aSBarry Smith            local subdomain
7653677ff5aSBarry Smith   */
76624cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr);
76724cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr);
76824cf384cSBarry Smith   ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr);
76989d82c54SBarry Smith 
77089d82c54SBarry Smith   for (i=0; i<n; i++) {
77189d82c54SBarry Smith     if (lindices[i] > max) max = lindices[i];
77289d82c54SBarry Smith   }
77332dcc486SBarry Smith   ierr   = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
77478058e43SBarry Smith   Ng++;
77589d82c54SBarry Smith   ierr   = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
77689d82c54SBarry Smith   ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
777bc8ff85bSBarry Smith   scale  = Ng/size + 1;
778a2e34c3dSBarry Smith   ng     = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng);
779caba0dd0SBarry Smith   rstart = scale*rank;
78089d82c54SBarry Smith 
78189d82c54SBarry Smith   /* determine ownership ranges of global indices */
782785e854fSJed Brown   ierr = PetscMalloc1(2*size,&nprocs);CHKERRQ(ierr);
78332dcc486SBarry Smith   ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
78489d82c54SBarry Smith 
78589d82c54SBarry Smith   /* determine owners of each local node  */
786785e854fSJed Brown   ierr = PetscMalloc1(n,&owner);CHKERRQ(ierr);
78789d82c54SBarry Smith   for (i=0; i<n; i++) {
7883677ff5aSBarry Smith     proc             = lindices[i]/scale; /* processor that globally owns this index */
78927c402fcSBarry Smith     nprocs[2*proc+1] = 1;                 /* processor globally owns at least one of ours */
7903677ff5aSBarry Smith     owner[i]         = proc;
79127c402fcSBarry Smith     nprocs[2*proc]++;                     /* count of how many that processor globally owns of ours */
79289d82c54SBarry Smith   }
79327c402fcSBarry Smith   nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1];
7947904a332SBarry Smith   ierr = PetscInfo1(mapping,"Number of global owners for my local data %D\n",nsends);CHKERRQ(ierr);
79589d82c54SBarry Smith 
79689d82c54SBarry Smith   /* inform other processors of number of messages and max length*/
79727c402fcSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
7987904a332SBarry Smith   ierr = PetscInfo1(mapping,"Number of local owners for my global data %D\n",nrecvs);CHKERRQ(ierr);
79989d82c54SBarry Smith 
80089d82c54SBarry Smith   /* post receives for owned rows */
801785e854fSJed Brown   ierr = PetscMalloc1((2*nrecvs+1)*(nmax+1),&recvs);CHKERRQ(ierr);
802785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&recv_waits);CHKERRQ(ierr);
80389d82c54SBarry Smith   for (i=0; i<nrecvs; i++) {
80432dcc486SBarry Smith     ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr);
80589d82c54SBarry Smith   }
80689d82c54SBarry Smith 
80789d82c54SBarry Smith   /* pack messages containing lists of local nodes to owners */
808785e854fSJed Brown   ierr      = PetscMalloc1((2*n+1),&sends);CHKERRQ(ierr);
809785e854fSJed Brown   ierr      = PetscMalloc1((size+1),&starts);CHKERRQ(ierr);
81089d82c54SBarry Smith   starts[0] = 0;
811f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
81289d82c54SBarry Smith   for (i=0; i<n; i++) {
81389d82c54SBarry Smith     sends[starts[owner[i]]++] = lindices[i];
81430dcb7c9SBarry Smith     sends[starts[owner[i]]++] = i;
81589d82c54SBarry Smith   }
81689d82c54SBarry Smith   ierr = PetscFree(owner);CHKERRQ(ierr);
81789d82c54SBarry Smith   starts[0] = 0;
818f6e5521dSKarl Rupp   for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2];
81989d82c54SBarry Smith 
82089d82c54SBarry Smith   /* send the messages */
821785e854fSJed Brown   ierr = PetscMalloc1((nsends+1),&send_waits);CHKERRQ(ierr);
822785e854fSJed Brown   ierr = PetscMalloc1((nsends+1),&dest);CHKERRQ(ierr);
82389d82c54SBarry Smith   cnt = 0;
82489d82c54SBarry Smith   for (i=0; i<size; i++) {
82527c402fcSBarry Smith     if (nprocs[2*i]) {
82632dcc486SBarry Smith       ierr      = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr);
82730dcb7c9SBarry Smith       dest[cnt] = i;
82889d82c54SBarry Smith       cnt++;
82989d82c54SBarry Smith     }
83089d82c54SBarry Smith   }
83189d82c54SBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
83289d82c54SBarry Smith 
83389d82c54SBarry Smith   /* wait on receives */
834785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&source);CHKERRQ(ierr);
835785e854fSJed Brown   ierr = PetscMalloc1((nrecvs+1),&len);CHKERRQ(ierr);
83689d82c54SBarry Smith   cnt  = nrecvs;
837785e854fSJed Brown   ierr = PetscMalloc1((ng+1),&nownedsenders);CHKERRQ(ierr);
83832dcc486SBarry Smith   ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr);
83989d82c54SBarry Smith   while (cnt) {
84089d82c54SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
84189d82c54SBarry Smith     /* unpack receives into our local space */
84232dcc486SBarry Smith     ierr          = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr);
84389d82c54SBarry Smith     source[imdex] = recv_status.MPI_SOURCE;
84430dcb7c9SBarry Smith     len[imdex]    = len[imdex]/2;
845caba0dd0SBarry Smith     /* count how many local owners for each of my global owned indices */
84630dcb7c9SBarry Smith     for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++;
84789d82c54SBarry Smith     cnt--;
84889d82c54SBarry Smith   }
84989d82c54SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
85089d82c54SBarry Smith 
85130dcb7c9SBarry Smith   /* count how many globally owned indices are on an edge multiplied by how many processors own them. */
852bc8ff85bSBarry Smith   nowned  = 0;
853bc8ff85bSBarry Smith   nownedm = 0;
854bc8ff85bSBarry Smith   for (i=0; i<ng; i++) {
855bc8ff85bSBarry Smith     if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;}
856bc8ff85bSBarry Smith   }
857bc8ff85bSBarry Smith 
858bc8ff85bSBarry Smith   /* create single array to contain rank of all local owners of each globally owned index */
859785e854fSJed Brown   ierr      = PetscMalloc1((nownedm+1),&ownedsenders);CHKERRQ(ierr);
860785e854fSJed Brown   ierr      = PetscMalloc1((ng+1),&starts);CHKERRQ(ierr);
861bc8ff85bSBarry Smith   starts[0] = 0;
862bc8ff85bSBarry Smith   for (i=1; i<ng; i++) {
863bc8ff85bSBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
864bc8ff85bSBarry Smith     else starts[i] = starts[i-1];
865bc8ff85bSBarry Smith   }
866bc8ff85bSBarry Smith 
86730dcb7c9SBarry Smith   /* for each nontrival globally owned node list all arriving processors */
868bc8ff85bSBarry Smith   for (i=0; i<nrecvs; i++) {
869bc8ff85bSBarry Smith     for (j=0; j<len[i]; j++) {
87030dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
871f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) ownedsenders[starts[node]++] = source[i];
872bc8ff85bSBarry Smith     }
873bc8ff85bSBarry Smith   }
874bc8ff85bSBarry Smith 
87507b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
87630dcb7c9SBarry Smith     starts[0] = 0;
87730dcb7c9SBarry Smith     for (i=1; i<ng; i++) {
87830dcb7c9SBarry Smith       if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
87930dcb7c9SBarry Smith       else starts[i] = starts[i-1];
88030dcb7c9SBarry Smith     }
88130dcb7c9SBarry Smith     for (i=0; i<ng; i++) {
88230dcb7c9SBarry Smith       if (nownedsenders[i] > 1) {
8837904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] global node %D local owner processors: ",rank,i+rstart);CHKERRQ(ierr);
88430dcb7c9SBarry Smith         for (j=0; j<nownedsenders[i]; j++) {
8857904a332SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%D ",ownedsenders[starts[i]+j]);CHKERRQ(ierr);
88630dcb7c9SBarry Smith         }
88730dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
88830dcb7c9SBarry Smith       }
88930dcb7c9SBarry Smith     }
8900ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
89107b52d57SBarry Smith   } /* -----------------------------------  */
89230dcb7c9SBarry Smith 
8933677ff5aSBarry Smith   /* wait on original sends */
8943a96401aSBarry Smith   if (nsends) {
895785e854fSJed Brown     ierr = PetscMalloc1(nsends,&send_status);CHKERRQ(ierr);
8963a96401aSBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
8973a96401aSBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
8983a96401aSBarry Smith   }
89989d82c54SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
9003a96401aSBarry Smith   ierr = PetscFree(sends);CHKERRQ(ierr);
9013677ff5aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
9023677ff5aSBarry Smith 
9033677ff5aSBarry Smith   /* pack messages to send back to local owners */
90430dcb7c9SBarry Smith   starts[0] = 0;
90530dcb7c9SBarry Smith   for (i=1; i<ng; i++) {
90630dcb7c9SBarry Smith     if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1];
90730dcb7c9SBarry Smith     else starts[i] = starts[i-1];
90830dcb7c9SBarry Smith   }
90930dcb7c9SBarry Smith   nsends2 = nrecvs;
910785e854fSJed Brown   ierr    = PetscMalloc1((nsends2+1),&nprocs);CHKERRQ(ierr); /* length of each message */
91130dcb7c9SBarry Smith   for (i=0; i<nrecvs; i++) {
91230dcb7c9SBarry Smith     nprocs[i] = 1;
91330dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
91430dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
915f6e5521dSKarl Rupp       if (nownedsenders[node] > 1) nprocs[i] += 2 + nownedsenders[node];
91630dcb7c9SBarry Smith     }
91730dcb7c9SBarry Smith   }
918f6e5521dSKarl Rupp   nt = 0;
919f6e5521dSKarl Rupp   for (i=0; i<nsends2; i++) nt += nprocs[i];
920f6e5521dSKarl Rupp 
921785e854fSJed Brown   ierr = PetscMalloc1((nt+1),&sends2);CHKERRQ(ierr);
922785e854fSJed Brown   ierr = PetscMalloc1((nsends2+1),&starts2);CHKERRQ(ierr);
923f6e5521dSKarl Rupp 
924f6e5521dSKarl Rupp   starts2[0] = 0;
925f6e5521dSKarl Rupp   for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1];
92630dcb7c9SBarry Smith   /*
92730dcb7c9SBarry Smith      Each message is 1 + nprocs[i] long, and consists of
92830dcb7c9SBarry Smith        (0) the number of nodes being sent back
92930dcb7c9SBarry Smith        (1) the local node number,
93030dcb7c9SBarry Smith        (2) the number of processors sharing it,
93130dcb7c9SBarry Smith        (3) the processors sharing it
93230dcb7c9SBarry Smith   */
93330dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
93430dcb7c9SBarry Smith     cnt = 1;
93530dcb7c9SBarry Smith     sends2[starts2[i]] = 0;
93630dcb7c9SBarry Smith     for (j=0; j<len[i]; j++) {
93730dcb7c9SBarry Smith       node = recvs[2*i*nmax+2*j]-rstart;
93830dcb7c9SBarry Smith       if (nownedsenders[node] > 1) {
93930dcb7c9SBarry Smith         sends2[starts2[i]]++;
94030dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1];
94130dcb7c9SBarry Smith         sends2[starts2[i]+cnt++] = nownedsenders[node];
94232dcc486SBarry Smith         ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr);
94330dcb7c9SBarry Smith         cnt += nownedsenders[node];
94430dcb7c9SBarry Smith       }
94530dcb7c9SBarry Smith     }
94630dcb7c9SBarry Smith   }
94730dcb7c9SBarry Smith 
94830dcb7c9SBarry Smith   /* receive the message lengths */
94930dcb7c9SBarry Smith   nrecvs2 = nsends;
950785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&lens2);CHKERRQ(ierr);
951785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&starts3);CHKERRQ(ierr);
952785e854fSJed Brown   ierr    = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr);
95330dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
954d44834fbSBarry Smith     ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr);
95530dcb7c9SBarry Smith   }
956d44834fbSBarry Smith 
9578a8e0b3aSBarry Smith   /* send the message lengths */
9588a8e0b3aSBarry Smith   for (i=0; i<nsends2; i++) {
9598a8e0b3aSBarry Smith     ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr);
9608a8e0b3aSBarry Smith   }
9618a8e0b3aSBarry Smith 
962d44834fbSBarry Smith   /* wait on receives of lens */
9630c468ba9SBarry Smith   if (nrecvs2) {
964785e854fSJed Brown     ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr);
965d44834fbSBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
966d44834fbSBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
9670c468ba9SBarry Smith   }
968a2ea699eSBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
969d44834fbSBarry Smith 
97030dcb7c9SBarry Smith   starts3[0] = 0;
971d44834fbSBarry Smith   nt         = 0;
97230dcb7c9SBarry Smith   for (i=0; i<nrecvs2-1; i++) {
97330dcb7c9SBarry Smith     starts3[i+1] = starts3[i] + lens2[i];
974d44834fbSBarry Smith     nt          += lens2[i];
97530dcb7c9SBarry Smith   }
97676466f69SStefano Zampini   if (nrecvs2) nt += lens2[nrecvs2-1];
977d44834fbSBarry Smith 
978785e854fSJed Brown   ierr = PetscMalloc1((nt+1),&recvs2);CHKERRQ(ierr);
979785e854fSJed Brown   ierr = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr);
98052b72c4aSBarry Smith   for (i=0; i<nrecvs2; i++) {
98132dcc486SBarry Smith     ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr);
98230dcb7c9SBarry Smith   }
98330dcb7c9SBarry Smith 
98430dcb7c9SBarry Smith   /* send the messages */
985785e854fSJed Brown   ierr = PetscMalloc1((nsends2+1),&send_waits);CHKERRQ(ierr);
98630dcb7c9SBarry Smith   for (i=0; i<nsends2; i++) {
98732dcc486SBarry Smith     ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr);
98830dcb7c9SBarry Smith   }
98930dcb7c9SBarry Smith 
99030dcb7c9SBarry Smith   /* wait on receives */
9910c468ba9SBarry Smith   if (nrecvs2) {
992785e854fSJed Brown     ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr);
99330dcb7c9SBarry Smith     ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr);
99430dcb7c9SBarry Smith     ierr = PetscFree(recv_statuses);CHKERRQ(ierr);
9950c468ba9SBarry Smith   }
99630dcb7c9SBarry Smith   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
99730dcb7c9SBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
99830dcb7c9SBarry Smith 
99907b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
100030dcb7c9SBarry Smith     cnt = 0;
100130dcb7c9SBarry Smith     for (i=0; i<nrecvs2; i++) {
100230dcb7c9SBarry Smith       nt = recvs2[cnt++];
100330dcb7c9SBarry Smith       for (j=0; j<nt; j++) {
10047904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"[%d] local node %D number of subdomains %D: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr);
100530dcb7c9SBarry Smith         for (k=0; k<recvs2[cnt+1]; k++) {
10067904a332SBarry Smith           ierr = PetscSynchronizedPrintf(comm,"%D ",recvs2[cnt+2+k]);CHKERRQ(ierr);
100730dcb7c9SBarry Smith         }
100830dcb7c9SBarry Smith         cnt += 2 + recvs2[cnt+1];
100930dcb7c9SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
101030dcb7c9SBarry Smith       }
101130dcb7c9SBarry Smith     }
10120ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
101307b52d57SBarry Smith   } /* -----------------------------------  */
101430dcb7c9SBarry Smith 
101530dcb7c9SBarry Smith   /* count number subdomains for each local node */
1016785e854fSJed Brown   ierr = PetscMalloc1(size,&nprocs);CHKERRQ(ierr);
101732dcc486SBarry Smith   ierr = PetscMemzero(nprocs,size*sizeof(PetscInt));CHKERRQ(ierr);
101830dcb7c9SBarry Smith   cnt  = 0;
101930dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
102030dcb7c9SBarry Smith     nt = recvs2[cnt++];
102130dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
1022f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) nprocs[recvs2[cnt+2+k]]++;
102330dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
102430dcb7c9SBarry Smith     }
102530dcb7c9SBarry Smith   }
102630dcb7c9SBarry Smith   nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0);
102730dcb7c9SBarry Smith   *nproc    = nt;
1028785e854fSJed Brown   ierr = PetscMalloc1((nt+1),procs);CHKERRQ(ierr);
1029785e854fSJed Brown   ierr = PetscMalloc1((nt+1),numprocs);CHKERRQ(ierr);
1030785e854fSJed Brown   ierr = PetscMalloc1((nt+1),indices);CHKERRQ(ierr);
10310298fd71SBarry Smith   for (i=0;i<nt+1;i++) (*indices)[i]=NULL;
1032785e854fSJed Brown   ierr = PetscMalloc1(size,&bprocs);CHKERRQ(ierr);
103330dcb7c9SBarry Smith   cnt       = 0;
103430dcb7c9SBarry Smith   for (i=0; i<size; i++) {
103530dcb7c9SBarry Smith     if (nprocs[i] > 0) {
103630dcb7c9SBarry Smith       bprocs[i]        = cnt;
103730dcb7c9SBarry Smith       (*procs)[cnt]    = i;
103830dcb7c9SBarry Smith       (*numprocs)[cnt] = nprocs[i];
1039785e854fSJed Brown       ierr             = PetscMalloc1(nprocs[i],&(*indices)[cnt]);CHKERRQ(ierr);
104030dcb7c9SBarry Smith       cnt++;
104130dcb7c9SBarry Smith     }
104230dcb7c9SBarry Smith   }
104330dcb7c9SBarry Smith 
104430dcb7c9SBarry Smith   /* make the list of subdomains for each nontrivial local node */
104532dcc486SBarry Smith   ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr);
104630dcb7c9SBarry Smith   cnt  = 0;
104730dcb7c9SBarry Smith   for (i=0; i<nrecvs2; i++) {
104830dcb7c9SBarry Smith     nt = recvs2[cnt++];
104930dcb7c9SBarry Smith     for (j=0; j<nt; j++) {
1050f6e5521dSKarl Rupp       for (k=0; k<recvs2[cnt+1]; k++) (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt];
105130dcb7c9SBarry Smith       cnt += 2 + recvs2[cnt+1];
105230dcb7c9SBarry Smith     }
105330dcb7c9SBarry Smith   }
105430dcb7c9SBarry Smith   ierr = PetscFree(bprocs);CHKERRQ(ierr);
105507b52d57SBarry Smith   ierr = PetscFree(recvs2);CHKERRQ(ierr);
105630dcb7c9SBarry Smith 
105707b52d57SBarry Smith   /* sort the node indexing by their global numbers */
105807b52d57SBarry Smith   nt = *nproc;
105907b52d57SBarry Smith   for (i=0; i<nt; i++) {
1060785e854fSJed Brown     ierr = PetscMalloc1(((*numprocs)[i]),&tmp);CHKERRQ(ierr);
1061f6e5521dSKarl Rupp     for (j=0; j<(*numprocs)[i]; j++) tmp[j] = lindices[(*indices)[i][j]];
106207b52d57SBarry Smith     ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr);
106307b52d57SBarry Smith     ierr = PetscFree(tmp);CHKERRQ(ierr);
106407b52d57SBarry Smith   }
106507b52d57SBarry Smith 
106607b52d57SBarry Smith   if (debug) { /* -----------------------------------  */
106730dcb7c9SBarry Smith     nt = *nproc;
106830dcb7c9SBarry Smith     for (i=0; i<nt; i++) {
10697904a332SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %D number of indices %D: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr);
107030dcb7c9SBarry Smith       for (j=0; j<(*numprocs)[i]; j++) {
10717904a332SBarry Smith         ierr = PetscSynchronizedPrintf(comm,"%D ",(*indices)[i][j]);CHKERRQ(ierr);
107230dcb7c9SBarry Smith       }
107330dcb7c9SBarry Smith       ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr);
107430dcb7c9SBarry Smith     }
10750ec8b6e3SBarry Smith     ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr);
107607b52d57SBarry Smith   } /* -----------------------------------  */
107730dcb7c9SBarry Smith 
107830dcb7c9SBarry Smith   /* wait on sends */
107930dcb7c9SBarry Smith   if (nsends2) {
1080785e854fSJed Brown     ierr = PetscMalloc1(nsends2,&send_status);CHKERRQ(ierr);
108130dcb7c9SBarry Smith     ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr);
108230dcb7c9SBarry Smith     ierr = PetscFree(send_status);CHKERRQ(ierr);
108330dcb7c9SBarry Smith   }
108430dcb7c9SBarry Smith 
108530dcb7c9SBarry Smith   ierr = PetscFree(starts3);CHKERRQ(ierr);
108630dcb7c9SBarry Smith   ierr = PetscFree(dest);CHKERRQ(ierr);
108730dcb7c9SBarry Smith   ierr = PetscFree(send_waits);CHKERRQ(ierr);
10883677ff5aSBarry Smith 
1089bc8ff85bSBarry Smith   ierr = PetscFree(nownedsenders);CHKERRQ(ierr);
1090bc8ff85bSBarry Smith   ierr = PetscFree(ownedsenders);CHKERRQ(ierr);
1091bc8ff85bSBarry Smith   ierr = PetscFree(starts);CHKERRQ(ierr);
109230dcb7c9SBarry Smith   ierr = PetscFree(starts2);CHKERRQ(ierr);
109330dcb7c9SBarry Smith   ierr = PetscFree(lens2);CHKERRQ(ierr);
109489d82c54SBarry Smith 
109589d82c54SBarry Smith   ierr = PetscFree(source);CHKERRQ(ierr);
109697f1f81fSBarry Smith   ierr = PetscFree(len);CHKERRQ(ierr);
109789d82c54SBarry Smith   ierr = PetscFree(recvs);CHKERRQ(ierr);
10983a96401aSBarry Smith   ierr = PetscFree(nprocs);CHKERRQ(ierr);
109930dcb7c9SBarry Smith   ierr = PetscFree(sends2);CHKERRQ(ierr);
110024cf384cSBarry Smith 
110124cf384cSBarry Smith   /* put the information about myself as the first entry in the list */
110224cf384cSBarry Smith   first_procs    = (*procs)[0];
110324cf384cSBarry Smith   first_numprocs = (*numprocs)[0];
110424cf384cSBarry Smith   first_indices  = (*indices)[0];
110524cf384cSBarry Smith   for (i=0; i<*nproc; i++) {
110624cf384cSBarry Smith     if ((*procs)[i] == rank) {
110724cf384cSBarry Smith       (*procs)[0]    = (*procs)[i];
110824cf384cSBarry Smith       (*numprocs)[0] = (*numprocs)[i];
110924cf384cSBarry Smith       (*indices)[0]  = (*indices)[i];
111024cf384cSBarry Smith       (*procs)[i]    = first_procs;
111124cf384cSBarry Smith       (*numprocs)[i] = first_numprocs;
111224cf384cSBarry Smith       (*indices)[i]  = first_indices;
111324cf384cSBarry Smith       break;
111424cf384cSBarry Smith     }
111524cf384cSBarry Smith   }
111689d82c54SBarry Smith   PetscFunctionReturn(0);
111789d82c54SBarry Smith }
111889d82c54SBarry Smith 
11194a2ae208SSatish Balay #undef __FUNCT__
11206a818285SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingRestoreBlockInfo"
11216a818285SBarry Smith /*@C
11226a818285SBarry Smith     ISLocalToGlobalMappingRestoreBlockInfo - Frees the memory allocated by ISLocalToGlobalMappingGetBlockInfo()
11236a818285SBarry Smith 
11246a818285SBarry Smith     Collective on ISLocalToGlobalMapping
11256a818285SBarry Smith 
11266a818285SBarry Smith     Input Parameters:
11276a818285SBarry Smith .   mapping - the mapping from local to global indexing
11286a818285SBarry Smith 
11296a818285SBarry Smith     Output Parameter:
11306a818285SBarry Smith +   nproc - number of processors that are connected to this one
11316a818285SBarry Smith .   proc - neighboring processors
11326a818285SBarry Smith .   numproc - number of indices for each processor
11336a818285SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
11346a818285SBarry Smith 
11356a818285SBarry Smith     Level: advanced
11366a818285SBarry Smith 
11376a818285SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
11386a818285SBarry Smith           ISLocalToGlobalMappingGetInfo()
11396a818285SBarry Smith @*/
11406a818285SBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreBlockInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
11416a818285SBarry Smith {
11426a818285SBarry Smith   PetscErrorCode ierr;
11436a818285SBarry Smith   PetscInt       i;
11446a818285SBarry Smith 
11456a818285SBarry Smith   PetscFunctionBegin;
1146*cbc1caf0SMatthew G. Knepley   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
11476a818285SBarry Smith   ierr = PetscFree(*procs);CHKERRQ(ierr);
11486a818285SBarry Smith   ierr = PetscFree(*numprocs);CHKERRQ(ierr);
11496a818285SBarry Smith   if (*indices) {
11506a818285SBarry Smith     ierr = PetscFree((*indices)[0]);CHKERRQ(ierr);
11516a818285SBarry Smith     for (i=1; i<*nproc; i++) {
11526a818285SBarry Smith       ierr = PetscFree((*indices)[i]);CHKERRQ(ierr);
11536a818285SBarry Smith     }
11546a818285SBarry Smith     ierr = PetscFree(*indices);CHKERRQ(ierr);
11556a818285SBarry Smith   }
11566a818285SBarry Smith   PetscFunctionReturn(0);
11576a818285SBarry Smith }
11586a818285SBarry Smith 
11596a818285SBarry Smith #undef __FUNCT__
11606a818285SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetInfo"
11616a818285SBarry Smith /*@C
11626a818285SBarry Smith     ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and
11636a818285SBarry Smith      each index shared by more than one processor
11646a818285SBarry Smith 
11656a818285SBarry Smith     Collective on ISLocalToGlobalMapping
11666a818285SBarry Smith 
11676a818285SBarry Smith     Input Parameters:
11686a818285SBarry Smith .   mapping - the mapping from local to global indexing
11696a818285SBarry Smith 
11706a818285SBarry Smith     Output Parameter:
11716a818285SBarry Smith +   nproc - number of processors that are connected to this one
11726a818285SBarry Smith .   proc - neighboring processors
11736a818285SBarry Smith .   numproc - number of indices for each subdomain (processor)
11746a818285SBarry Smith -   indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering)
11756a818285SBarry Smith 
11766a818285SBarry Smith     Level: advanced
11776a818285SBarry Smith 
11786a818285SBarry Smith     Concepts: mapping^local to global
11796a818285SBarry Smith 
11806a818285SBarry Smith     Fortran Usage:
11816a818285SBarry Smith $        ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by
11826a818285SBarry Smith $        ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc],
11836a818285SBarry Smith           PetscInt indices[nproc][numprocmax],ierr)
11846a818285SBarry Smith         There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and
11856a818285SBarry Smith         indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough.
11866a818285SBarry Smith 
11876a818285SBarry Smith 
11886a818285SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
11896a818285SBarry Smith           ISLocalToGlobalMappingRestoreInfo()
11906a818285SBarry Smith @*/
11916a818285SBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
11926a818285SBarry Smith {
11936a818285SBarry Smith   PetscErrorCode ierr;
11946a818285SBarry Smith   PetscInt       **bindices = NULL,bs = mapping->bs,i,j,k;
11956a818285SBarry Smith 
11966a818285SBarry Smith   PetscFunctionBegin;
11976a818285SBarry Smith   PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1);
11986a818285SBarry Smith   ierr = ISLocalToGlobalMappingGetBlockInfo(mapping,nproc,procs,numprocs,&bindices);CHKERRQ(ierr);
1199732f65e3SBarry Smith   ierr = PetscCalloc1(*nproc,&*indices);CHKERRQ(ierr);
12006a818285SBarry Smith   for (i=0; i<*nproc; i++) {
12016a818285SBarry Smith     ierr = PetscMalloc1(bs*(*numprocs)[i],&(*indices)[i]);CHKERRQ(ierr);
12026a818285SBarry Smith     for (j=0; j<(*numprocs)[i]; j++) {
12036a818285SBarry Smith       for (k=0; k<bs; k++) {
12046a818285SBarry Smith         (*indices)[i][j*bs+k] = bs*bindices[i][j] + k;
12056a818285SBarry Smith       }
12066a818285SBarry Smith     }
12076a818285SBarry Smith     (*numprocs)[i] *= bs;
12086a818285SBarry Smith   }
12096a818285SBarry Smith   if (bindices) {
12106a818285SBarry Smith     ierr = PetscFree(bindices[0]);CHKERRQ(ierr);
12116a818285SBarry Smith     for (i=1; i<*nproc; i++) {
12126a818285SBarry Smith       ierr = PetscFree(bindices[i]);CHKERRQ(ierr);
12136a818285SBarry Smith     }
12146a818285SBarry Smith     ierr = PetscFree(bindices);CHKERRQ(ierr);
12156a818285SBarry Smith   }
12166a818285SBarry Smith   PetscFunctionReturn(0);
12176a818285SBarry Smith }
12186a818285SBarry Smith 
12196a818285SBarry Smith #undef __FUNCT__
12204a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo"
122107b52d57SBarry Smith /*@C
122207b52d57SBarry Smith     ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo()
122389d82c54SBarry Smith 
122407b52d57SBarry Smith     Collective on ISLocalToGlobalMapping
122507b52d57SBarry Smith 
122607b52d57SBarry Smith     Input Parameters:
122707b52d57SBarry Smith .   mapping - the mapping from local to global indexing
122807b52d57SBarry Smith 
122907b52d57SBarry Smith     Output Parameter:
123007b52d57SBarry Smith +   nproc - number of processors that are connected to this one
123107b52d57SBarry Smith .   proc - neighboring processors
123207b52d57SBarry Smith .   numproc - number of indices for each processor
123307b52d57SBarry Smith -   indices - indices of local nodes shared with neighbor (sorted by global numbering)
123407b52d57SBarry Smith 
123507b52d57SBarry Smith     Level: advanced
123607b52d57SBarry Smith 
123707b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(),
123807b52d57SBarry Smith           ISLocalToGlobalMappingGetInfo()
123907b52d57SBarry Smith @*/
12407087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[])
124107b52d57SBarry Smith {
12426849ba73SBarry Smith   PetscErrorCode ierr;
124307b52d57SBarry Smith 
124407b52d57SBarry Smith   PetscFunctionBegin;
12456a818285SBarry Smith   ierr = ISLocalToGlobalMappingRestoreBlockInfo(mapping,nproc,procs,numprocs,indices);CHKERRQ(ierr);
124607b52d57SBarry Smith   PetscFunctionReturn(0);
124707b52d57SBarry Smith }
124886994e45SJed Brown 
124986994e45SJed Brown #undef __FUNCT__
125086994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices"
125186994e45SJed Brown /*@C
1252107e9a97SBarry Smith    ISLocalToGlobalMappingGetIndices - Get global indices for every local point that is mapped
125386994e45SJed Brown 
125486994e45SJed Brown    Not Collective
125586994e45SJed Brown 
125686994e45SJed Brown    Input Arguments:
125786994e45SJed Brown . ltog - local to global mapping
125886994e45SJed Brown 
125986994e45SJed Brown    Output Arguments:
1260565245c5SBarry Smith . array - array of indices, the length of this array may be obtained with ISLocalToGlobalMappingGetSize()
126186994e45SJed Brown 
126286994e45SJed Brown    Level: advanced
126386994e45SJed Brown 
1264107e9a97SBarry Smith    Notes: ISLocalToGlobalMappingGetSize() returns the length the this array
1265107e9a97SBarry Smith 
1266107e9a97SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices(), ISLocalToGlobalMappingGetBlockIndices(), ISLocalToGlobalMappingRestoreBlockIndices()
126786994e45SJed Brown @*/
12687087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
126986994e45SJed Brown {
127086994e45SJed Brown   PetscFunctionBegin;
127186994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
127286994e45SJed Brown   PetscValidPointer(array,2);
127345b6f7e9SBarry Smith   if (ltog->bs == 1) {
127486994e45SJed Brown     *array = ltog->indices;
127545b6f7e9SBarry Smith   } else {
127645b6f7e9SBarry Smith     PetscInt       *jj,k,i,j,n = ltog->n, bs = ltog->bs;
127745b6f7e9SBarry Smith     const PetscInt *ii;
127845b6f7e9SBarry Smith     PetscErrorCode ierr;
127945b6f7e9SBarry Smith 
128045b6f7e9SBarry Smith     ierr = PetscMalloc1(bs*n,&jj);CHKERRQ(ierr);
128145b6f7e9SBarry Smith     *array = jj;
128245b6f7e9SBarry Smith     k    = 0;
128345b6f7e9SBarry Smith     ii   = ltog->indices;
128445b6f7e9SBarry Smith     for (i=0; i<n; i++)
128545b6f7e9SBarry Smith       for (j=0; j<bs; j++)
128645b6f7e9SBarry Smith         jj[k++] = bs*ii[i] + j;
128745b6f7e9SBarry Smith   }
128886994e45SJed Brown   PetscFunctionReturn(0);
128986994e45SJed Brown }
129086994e45SJed Brown 
129186994e45SJed Brown #undef __FUNCT__
129286994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices"
129386994e45SJed Brown /*@C
129486994e45SJed Brown    ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices()
129586994e45SJed Brown 
129686994e45SJed Brown    Not Collective
129786994e45SJed Brown 
129886994e45SJed Brown    Input Arguments:
129986994e45SJed Brown + ltog - local to global mapping
130086994e45SJed Brown - array - array of indices
130186994e45SJed Brown 
130286994e45SJed Brown    Level: advanced
130386994e45SJed Brown 
130486994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
130586994e45SJed Brown @*/
13067087cfbeSBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
130786994e45SJed Brown {
130886994e45SJed Brown   PetscFunctionBegin;
130986994e45SJed Brown   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
131086994e45SJed Brown   PetscValidPointer(array,2);
131145b6f7e9SBarry Smith   if (ltog->bs == 1 && *array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
131245b6f7e9SBarry Smith 
131345b6f7e9SBarry Smith   if (ltog->bs > 1) {
131445b6f7e9SBarry Smith     PetscErrorCode ierr;
131545b6f7e9SBarry Smith     ierr = PetscFree(*(void**)array);CHKERRQ(ierr);
131645b6f7e9SBarry Smith   }
131745b6f7e9SBarry Smith   PetscFunctionReturn(0);
131845b6f7e9SBarry Smith }
131945b6f7e9SBarry Smith 
132045b6f7e9SBarry Smith #undef __FUNCT__
132145b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetBlockIndices"
132245b6f7e9SBarry Smith /*@C
132345b6f7e9SBarry Smith    ISLocalToGlobalMappingGetBlockIndices - Get global indices for every local block
132445b6f7e9SBarry Smith 
132545b6f7e9SBarry Smith    Not Collective
132645b6f7e9SBarry Smith 
132745b6f7e9SBarry Smith    Input Arguments:
132845b6f7e9SBarry Smith . ltog - local to global mapping
132945b6f7e9SBarry Smith 
133045b6f7e9SBarry Smith    Output Arguments:
133145b6f7e9SBarry Smith . array - array of indices
133245b6f7e9SBarry Smith 
133345b6f7e9SBarry Smith    Level: advanced
133445b6f7e9SBarry Smith 
133545b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreBlockIndices()
133645b6f7e9SBarry Smith @*/
133745b6f7e9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingGetBlockIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
133845b6f7e9SBarry Smith {
133945b6f7e9SBarry Smith   PetscFunctionBegin;
134045b6f7e9SBarry Smith   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
134145b6f7e9SBarry Smith   PetscValidPointer(array,2);
134245b6f7e9SBarry Smith   *array = ltog->indices;
134345b6f7e9SBarry Smith   PetscFunctionReturn(0);
134445b6f7e9SBarry Smith }
134545b6f7e9SBarry Smith 
134645b6f7e9SBarry Smith #undef __FUNCT__
134745b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingRestoreBlockIndices"
134845b6f7e9SBarry Smith /*@C
134945b6f7e9SBarry Smith    ISLocalToGlobalMappingRestoreBlockIndices - Restore indices obtained with ISLocalToGlobalMappingGetBlockIndices()
135045b6f7e9SBarry Smith 
135145b6f7e9SBarry Smith    Not Collective
135245b6f7e9SBarry Smith 
135345b6f7e9SBarry Smith    Input Arguments:
135445b6f7e9SBarry Smith + ltog - local to global mapping
135545b6f7e9SBarry Smith - array - array of indices
135645b6f7e9SBarry Smith 
135745b6f7e9SBarry Smith    Level: advanced
135845b6f7e9SBarry Smith 
135945b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices()
136045b6f7e9SBarry Smith @*/
136145b6f7e9SBarry Smith PetscErrorCode  ISLocalToGlobalMappingRestoreBlockIndices(ISLocalToGlobalMapping ltog,const PetscInt **array)
136245b6f7e9SBarry Smith {
136345b6f7e9SBarry Smith   PetscFunctionBegin;
136445b6f7e9SBarry Smith   PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1);
136545b6f7e9SBarry Smith   PetscValidPointer(array,2);
136686994e45SJed Brown   if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer");
13670298fd71SBarry Smith   *array = NULL;
136886994e45SJed Brown   PetscFunctionReturn(0);
136986994e45SJed Brown }
1370f7efa3c7SJed Brown 
1371f7efa3c7SJed Brown #undef __FUNCT__
1372f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate"
1373f7efa3c7SJed Brown /*@C
1374f7efa3c7SJed Brown    ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings
1375f7efa3c7SJed Brown 
1376f7efa3c7SJed Brown    Not Collective
1377f7efa3c7SJed Brown 
1378f7efa3c7SJed Brown    Input Arguments:
1379f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate
1380f7efa3c7SJed Brown . n - number of mappings to concatenate
1381f7efa3c7SJed Brown - ltogs - local to global mappings
1382f7efa3c7SJed Brown 
1383f7efa3c7SJed Brown    Output Arguments:
1384f7efa3c7SJed Brown . ltogcat - new mapping
1385f7efa3c7SJed Brown 
13869d90f715SBarry Smith    Note: this currently always returns a mapping with block size of 1
13879d90f715SBarry Smith 
13889d90f715SBarry Smith    Developer Note: If all the input mapping have the same block size we could easily handle that as a special case
13899d90f715SBarry Smith 
1390f7efa3c7SJed Brown    Level: advanced
1391f7efa3c7SJed Brown 
1392f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate()
1393f7efa3c7SJed Brown @*/
1394f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat)
1395f7efa3c7SJed Brown {
1396f7efa3c7SJed Brown   PetscInt       i,cnt,m,*idx;
1397f7efa3c7SJed Brown   PetscErrorCode ierr;
1398f7efa3c7SJed Brown 
1399f7efa3c7SJed Brown   PetscFunctionBegin;
1400f7efa3c7SJed Brown   if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n);
1401f7efa3c7SJed Brown   if (n > 0) PetscValidPointer(ltogs,3);
1402f7efa3c7SJed Brown   for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3);
1403f7efa3c7SJed Brown   PetscValidPointer(ltogcat,4);
1404f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1405f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1406f7efa3c7SJed Brown     cnt += m;
1407f7efa3c7SJed Brown   }
1408785e854fSJed Brown   ierr = PetscMalloc1(cnt,&idx);CHKERRQ(ierr);
1409f7efa3c7SJed Brown   for (cnt=0,i=0; i<n; i++) {
1410f7efa3c7SJed Brown     const PetscInt *subidx;
1411f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr);
1412f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1413f7efa3c7SJed Brown     ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr);
1414f7efa3c7SJed Brown     ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr);
1415f7efa3c7SJed Brown     cnt += m;
1416f7efa3c7SJed Brown   }
1417f0413b6fSBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,1,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr);
1418f7efa3c7SJed Brown   PetscFunctionReturn(0);
1419f7efa3c7SJed Brown }
142004a59952SBarry Smith 
142104a59952SBarry Smith 
1422