xref: /petsc/src/vec/is/sf/utils/sfutils.c (revision 38a251981c7f9aceca5179b53bf56d5ac9cf2dfa)
1b0c7db22SLisandro Dalcin #include <petsc/private/sfimpl.h>       /*I  "petscsf.h"   I*/
2b0c7db22SLisandro Dalcin #include <petsc/private/sectionimpl.h>
3b0c7db22SLisandro Dalcin 
4b0c7db22SLisandro Dalcin /*@C
5b0c7db22SLisandro Dalcin    PetscSFSetGraphLayout - Set a parallel star forest via global indices and a PetscLayout
6b0c7db22SLisandro Dalcin 
7b0c7db22SLisandro Dalcin    Collective
8b0c7db22SLisandro Dalcin 
9b0c7db22SLisandro Dalcin    Input Arguments:
10b0c7db22SLisandro Dalcin +  sf - star forest
11b0c7db22SLisandro Dalcin .  layout - PetscLayout defining the global space
12b0c7db22SLisandro Dalcin .  nleaves - number of leaf vertices on the current process, each of these references a root on any process
13b0c7db22SLisandro Dalcin .  ilocal - locations of leaves in leafdata buffers, pass NULL for contiguous storage
14b0c7db22SLisandro Dalcin .  localmode - copy mode for ilocal
15b0c7db22SLisandro Dalcin -  iremote - remote locations of root vertices for each leaf on the current process
16b0c7db22SLisandro Dalcin 
17b0c7db22SLisandro Dalcin    Level: intermediate
18b0c7db22SLisandro Dalcin 
19b0c7db22SLisandro Dalcin    Developers Note: Local indices which are the identity permutation in the range [0,nleaves) are discarded as they
20b0c7db22SLisandro Dalcin    encode contiguous storage. In such case, if localmode is PETSC_OWN_POINTER, the memory is deallocated as it is not
21b0c7db22SLisandro Dalcin    needed
22b0c7db22SLisandro Dalcin 
23b0c7db22SLisandro Dalcin .seealso: PetscSFCreate(), PetscSFView(), PetscSFSetGraph(), PetscSFGetGraph()
24b0c7db22SLisandro Dalcin @*/
25b0c7db22SLisandro Dalcin PetscErrorCode PetscSFSetGraphLayout(PetscSF sf,PetscLayout layout,PetscInt nleaves,const PetscInt *ilocal,PetscCopyMode localmode,const PetscInt *iremote)
26b0c7db22SLisandro Dalcin {
27b0c7db22SLisandro Dalcin   PetscErrorCode ierr;
28*38a25198SStefano Zampini   const PetscInt *range;
29*38a25198SStefano Zampini   PetscInt       i, nroots, ls = -1, ln = -1;
30*38a25198SStefano Zampini   PetscMPIInt    lr = -1;
31b0c7db22SLisandro Dalcin   PetscSFNode    *remote;
32b0c7db22SLisandro Dalcin 
33b0c7db22SLisandro Dalcin   PetscFunctionBegin;
34b0c7db22SLisandro Dalcin   ierr = PetscLayoutGetLocalSize(layout,&nroots);CHKERRQ(ierr);
35*38a25198SStefano Zampini   ierr = PetscLayoutGetRanges(layout,&range);CHKERRQ(ierr);
36b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(nleaves,&remote);CHKERRQ(ierr);
37*38a25198SStefano Zampini   if (nleaves) { ls = iremote[0] + 1; }
38b0c7db22SLisandro Dalcin   for (i=0; i<nleaves; i++) {
39*38a25198SStefano Zampini     const PetscInt idx = iremote[i] - ls;
40*38a25198SStefano Zampini     if (idx < 0 || idx >= ln) { /* short-circuit the search */
41*38a25198SStefano Zampini       ierr = PetscLayoutFindOwnerIndex(layout,iremote[i],&lr,&remote[i].index);CHKERRQ(ierr);
42*38a25198SStefano Zampini       remote[i].rank = lr;
43*38a25198SStefano Zampini       ls = range[lr];
44*38a25198SStefano Zampini       ln = range[lr+1] - ls;
45*38a25198SStefano Zampini     } else {
46*38a25198SStefano Zampini       remote[i].rank  = lr;
47*38a25198SStefano Zampini       remote[i].index = idx;
48*38a25198SStefano Zampini     }
49b0c7db22SLisandro Dalcin   }
50b0c7db22SLisandro Dalcin   ierr = PetscSFSetGraph(sf,nroots,nleaves,ilocal,localmode,remote,PETSC_OWN_POINTER);CHKERRQ(ierr);
51b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
52b0c7db22SLisandro Dalcin }
53b0c7db22SLisandro Dalcin 
54b0c7db22SLisandro Dalcin /*@
55b0c7db22SLisandro Dalcin   PetscSFSetGraphSection - Sets the PetscSF graph encoding the parallel dof overlap based upon the PetscSections
56b0c7db22SLisandro Dalcin   describing the data layout.
57b0c7db22SLisandro Dalcin 
58b0c7db22SLisandro Dalcin   Input Parameters:
59b0c7db22SLisandro Dalcin + sf - The SF
60b0c7db22SLisandro Dalcin . localSection - PetscSection describing the local data layout
61b0c7db22SLisandro Dalcin - globalSection - PetscSection describing the global data layout
62b0c7db22SLisandro Dalcin 
63b0c7db22SLisandro Dalcin   Level: developer
64b0c7db22SLisandro Dalcin 
65b0c7db22SLisandro Dalcin .seealso: PetscSFSetGraph(), PetscSFSetGraphLayout()
66b0c7db22SLisandro Dalcin @*/
67b0c7db22SLisandro Dalcin PetscErrorCode PetscSFSetGraphSection(PetscSF sf, PetscSection localSection, PetscSection globalSection)
68b0c7db22SLisandro Dalcin {
69b0c7db22SLisandro Dalcin   MPI_Comm       comm;
70b0c7db22SLisandro Dalcin   PetscLayout    layout;
71b0c7db22SLisandro Dalcin   const PetscInt *ranges;
72b0c7db22SLisandro Dalcin   PetscInt       *local;
73b0c7db22SLisandro Dalcin   PetscSFNode    *remote;
74b0c7db22SLisandro Dalcin   PetscInt       pStart, pEnd, p, nroots, nleaves = 0, l;
75b0c7db22SLisandro Dalcin   PetscMPIInt    size, rank;
76b0c7db22SLisandro Dalcin   PetscErrorCode ierr;
77b0c7db22SLisandro Dalcin 
78b0c7db22SLisandro Dalcin   PetscFunctionBegin;
79b0c7db22SLisandro Dalcin   PetscValidHeaderSpecific(sf, PETSCSF_CLASSID, 1);
80b0c7db22SLisandro Dalcin   PetscValidHeaderSpecific(localSection, PETSC_SECTION_CLASSID, 2);
81b0c7db22SLisandro Dalcin   PetscValidHeaderSpecific(globalSection, PETSC_SECTION_CLASSID, 3);
82b0c7db22SLisandro Dalcin 
83b0c7db22SLisandro Dalcin   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
84b0c7db22SLisandro Dalcin   ierr = MPI_Comm_size(comm, &size);CHKERRQ(ierr);
85b0c7db22SLisandro Dalcin   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
86b0c7db22SLisandro Dalcin   ierr = PetscSectionGetChart(globalSection, &pStart, &pEnd);CHKERRQ(ierr);
87b0c7db22SLisandro Dalcin   ierr = PetscSectionGetConstrainedStorageSize(globalSection, &nroots);CHKERRQ(ierr);
88b0c7db22SLisandro Dalcin   ierr = PetscLayoutCreate(comm, &layout);CHKERRQ(ierr);
89b0c7db22SLisandro Dalcin   ierr = PetscLayoutSetBlockSize(layout, 1);CHKERRQ(ierr);
90b0c7db22SLisandro Dalcin   ierr = PetscLayoutSetLocalSize(layout, nroots);CHKERRQ(ierr);
91b0c7db22SLisandro Dalcin   ierr = PetscLayoutSetUp(layout);CHKERRQ(ierr);
92b0c7db22SLisandro Dalcin   ierr = PetscLayoutGetRanges(layout, &ranges);CHKERRQ(ierr);
93b0c7db22SLisandro Dalcin   for (p = pStart; p < pEnd; ++p) {
94b0c7db22SLisandro Dalcin     PetscInt gdof, gcdof;
95b0c7db22SLisandro Dalcin 
96b0c7db22SLisandro Dalcin     ierr = PetscSectionGetDof(globalSection, p, &gdof);CHKERRQ(ierr);
97b0c7db22SLisandro Dalcin     ierr = PetscSectionGetConstraintDof(globalSection, p, &gcdof);CHKERRQ(ierr);
98b0c7db22SLisandro Dalcin     if (gcdof > (gdof < 0 ? -(gdof+1) : gdof)) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %D has %D constraints > %D dof", p, gcdof, (gdof < 0 ? -(gdof+1) : gdof));
99b0c7db22SLisandro Dalcin     nleaves += gdof < 0 ? -(gdof+1)-gcdof : gdof-gcdof;
100b0c7db22SLisandro Dalcin   }
101b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(nleaves, &local);CHKERRQ(ierr);
102b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(nleaves, &remote);CHKERRQ(ierr);
103b0c7db22SLisandro Dalcin   for (p = pStart, l = 0; p < pEnd; ++p) {
104b0c7db22SLisandro Dalcin     const PetscInt *cind;
105b0c7db22SLisandro Dalcin     PetscInt       dof, cdof, off, gdof, gcdof, goff, gsize, d, c;
106b0c7db22SLisandro Dalcin 
107b0c7db22SLisandro Dalcin     ierr = PetscSectionGetDof(localSection, p, &dof);CHKERRQ(ierr);
108b0c7db22SLisandro Dalcin     ierr = PetscSectionGetOffset(localSection, p, &off);CHKERRQ(ierr);
109b0c7db22SLisandro Dalcin     ierr = PetscSectionGetConstraintDof(localSection, p, &cdof);CHKERRQ(ierr);
110b0c7db22SLisandro Dalcin     ierr = PetscSectionGetConstraintIndices(localSection, p, &cind);CHKERRQ(ierr);
111b0c7db22SLisandro Dalcin     ierr = PetscSectionGetDof(globalSection, p, &gdof);CHKERRQ(ierr);
112b0c7db22SLisandro Dalcin     ierr = PetscSectionGetConstraintDof(globalSection, p, &gcdof);CHKERRQ(ierr);
113b0c7db22SLisandro Dalcin     ierr = PetscSectionGetOffset(globalSection, p, &goff);CHKERRQ(ierr);
114b0c7db22SLisandro Dalcin     if (!gdof) continue; /* Censored point */
115b0c7db22SLisandro Dalcin     gsize = gdof < 0 ? -(gdof+1)-gcdof : gdof-gcdof;
116b0c7db22SLisandro Dalcin     if (gsize != dof-cdof) {
117b0c7db22SLisandro Dalcin       if (gsize != dof) SETERRQ4(comm, PETSC_ERR_ARG_WRONG, "Global dof %D for point %D is neither the constrained size %D, nor the unconstrained %D", gsize, p, dof-cdof, dof);
118b0c7db22SLisandro Dalcin       cdof = 0; /* Ignore constraints */
119b0c7db22SLisandro Dalcin     }
120b0c7db22SLisandro Dalcin     for (d = 0, c = 0; d < dof; ++d) {
121b0c7db22SLisandro Dalcin       if ((c < cdof) && (cind[c] == d)) {++c; continue;}
122b0c7db22SLisandro Dalcin       local[l+d-c] = off+d;
123b0c7db22SLisandro Dalcin     }
124b0c7db22SLisandro Dalcin     if (gdof < 0) {
125b0c7db22SLisandro Dalcin       for (d = 0; d < gsize; ++d, ++l) {
126b0c7db22SLisandro Dalcin         PetscInt offset = -(goff+1) + d, r;
127b0c7db22SLisandro Dalcin 
128b0c7db22SLisandro Dalcin         ierr = PetscFindInt(offset,size+1,ranges,&r);CHKERRQ(ierr);
129b0c7db22SLisandro Dalcin         if (r < 0) r = -(r+2);
130b0c7db22SLisandro Dalcin         if ((r < 0) || (r >= size)) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %D mapped to invalid process %D (%D, %D)", p, r, gdof, goff);
131b0c7db22SLisandro Dalcin         remote[l].rank  = r;
132b0c7db22SLisandro Dalcin         remote[l].index = offset - ranges[r];
133b0c7db22SLisandro Dalcin       }
134b0c7db22SLisandro Dalcin     } else {
135b0c7db22SLisandro Dalcin       for (d = 0; d < gsize; ++d, ++l) {
136b0c7db22SLisandro Dalcin         remote[l].rank  = rank;
137b0c7db22SLisandro Dalcin         remote[l].index = goff+d - ranges[rank];
138b0c7db22SLisandro Dalcin       }
139b0c7db22SLisandro Dalcin     }
140b0c7db22SLisandro Dalcin   }
141b0c7db22SLisandro Dalcin   if (l != nleaves) SETERRQ2(comm, PETSC_ERR_PLIB, "Iteration error, l %D != nleaves %D", l, nleaves);
142b0c7db22SLisandro Dalcin   ierr = PetscLayoutDestroy(&layout);CHKERRQ(ierr);
143b0c7db22SLisandro Dalcin   ierr = PetscSFSetGraph(sf, nroots, nleaves, local, PETSC_OWN_POINTER, remote, PETSC_OWN_POINTER);CHKERRQ(ierr);
144b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
145b0c7db22SLisandro Dalcin }
146b0c7db22SLisandro Dalcin 
147b0c7db22SLisandro Dalcin static PetscErrorCode PetscSectionCheckConstraints_Static(PetscSection s)
148b0c7db22SLisandro Dalcin {
149b0c7db22SLisandro Dalcin   PetscErrorCode ierr;
150b0c7db22SLisandro Dalcin 
151b0c7db22SLisandro Dalcin   PetscFunctionBegin;
152b0c7db22SLisandro Dalcin   if (!s->bc) {
153b0c7db22SLisandro Dalcin     ierr = PetscSectionCreate(PETSC_COMM_SELF, &s->bc);CHKERRQ(ierr);
154b0c7db22SLisandro Dalcin     ierr = PetscSectionSetChart(s->bc, s->pStart, s->pEnd);CHKERRQ(ierr);
155b0c7db22SLisandro Dalcin   }
156b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
157b0c7db22SLisandro Dalcin }
158b0c7db22SLisandro Dalcin 
159b0c7db22SLisandro Dalcin /*@C
160b0c7db22SLisandro Dalcin   PetscSFDistributeSection - Create a new PetscSection reorganized, moving from the root to the leaves of the SF
161b0c7db22SLisandro Dalcin 
162b0c7db22SLisandro Dalcin   Collective on sf
163b0c7db22SLisandro Dalcin 
164b0c7db22SLisandro Dalcin   Input Parameters:
165b0c7db22SLisandro Dalcin + sf - The SF
166b0c7db22SLisandro Dalcin - rootSection - Section defined on root space
167b0c7db22SLisandro Dalcin 
168b0c7db22SLisandro Dalcin   Output Parameters:
169b0c7db22SLisandro Dalcin + remoteOffsets - root offsets in leaf storage, or NULL
170b0c7db22SLisandro Dalcin - leafSection - Section defined on the leaf space
171b0c7db22SLisandro Dalcin 
172b0c7db22SLisandro Dalcin   Level: advanced
173b0c7db22SLisandro Dalcin 
174b0c7db22SLisandro Dalcin .seealso: PetscSFCreate()
175b0c7db22SLisandro Dalcin @*/
176b0c7db22SLisandro Dalcin PetscErrorCode PetscSFDistributeSection(PetscSF sf, PetscSection rootSection, PetscInt **remoteOffsets, PetscSection leafSection)
177b0c7db22SLisandro Dalcin {
178b0c7db22SLisandro Dalcin   PetscSF        embedSF;
179b0c7db22SLisandro Dalcin   const PetscInt *indices;
180b0c7db22SLisandro Dalcin   IS             selected;
181b0c7db22SLisandro Dalcin   PetscInt       numFields, nroots, rpStart, rpEnd, lpStart = PETSC_MAX_INT, lpEnd = -1, f, c;
182b0c7db22SLisandro Dalcin   PetscBool      *sub, hasc;
183b0c7db22SLisandro Dalcin   PetscErrorCode ierr;
184b0c7db22SLisandro Dalcin 
185b0c7db22SLisandro Dalcin   PetscFunctionBegin;
186b0c7db22SLisandro Dalcin   ierr = PetscLogEventBegin(PETSCSF_DistSect,sf,0,0,0);CHKERRQ(ierr);
187b0c7db22SLisandro Dalcin   ierr = PetscSectionGetNumFields(rootSection, &numFields);CHKERRQ(ierr);
188b0c7db22SLisandro Dalcin   if (numFields) {ierr = PetscSectionSetNumFields(leafSection, numFields);CHKERRQ(ierr);}
189b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(numFields+2, &sub);CHKERRQ(ierr);
190b0c7db22SLisandro Dalcin   sub[1] = rootSection->bc ? PETSC_TRUE : PETSC_FALSE;
191b0c7db22SLisandro Dalcin   for (f = 0; f < numFields; ++f) {
192b0c7db22SLisandro Dalcin     PetscSectionSym sym;
193b0c7db22SLisandro Dalcin     const char      *name   = NULL;
194b0c7db22SLisandro Dalcin     PetscInt        numComp = 0;
195b0c7db22SLisandro Dalcin 
196b0c7db22SLisandro Dalcin     sub[2 + f] = rootSection->field[f]->bc ? PETSC_TRUE : PETSC_FALSE;
197b0c7db22SLisandro Dalcin     ierr = PetscSectionGetFieldComponents(rootSection, f, &numComp);CHKERRQ(ierr);
198b0c7db22SLisandro Dalcin     ierr = PetscSectionGetFieldName(rootSection, f, &name);CHKERRQ(ierr);
199b0c7db22SLisandro Dalcin     ierr = PetscSectionGetFieldSym(rootSection, f, &sym);CHKERRQ(ierr);
200b0c7db22SLisandro Dalcin     ierr = PetscSectionSetFieldComponents(leafSection, f, numComp);CHKERRQ(ierr);
201b0c7db22SLisandro Dalcin     ierr = PetscSectionSetFieldName(leafSection, f, name);CHKERRQ(ierr);
202b0c7db22SLisandro Dalcin     ierr = PetscSectionSetFieldSym(leafSection, f, sym);CHKERRQ(ierr);
203b0c7db22SLisandro Dalcin     for (c = 0; c < rootSection->numFieldComponents[f]; ++c) {
204b0c7db22SLisandro Dalcin       ierr = PetscSectionGetComponentName(rootSection, f, c, &name);CHKERRQ(ierr);
205b0c7db22SLisandro Dalcin       ierr = PetscSectionSetComponentName(leafSection, f, c, name);CHKERRQ(ierr);
206b0c7db22SLisandro Dalcin     }
207b0c7db22SLisandro Dalcin   }
208b0c7db22SLisandro Dalcin   ierr = PetscSectionGetChart(rootSection, &rpStart, &rpEnd);CHKERRQ(ierr);
209b0c7db22SLisandro Dalcin   ierr = PetscSFGetGraph(sf,&nroots,NULL,NULL,NULL);CHKERRQ(ierr);
210b0c7db22SLisandro Dalcin   rpEnd = PetscMin(rpEnd,nroots);
211b0c7db22SLisandro Dalcin   rpEnd = PetscMax(rpStart,rpEnd);
212b0c7db22SLisandro Dalcin   /* see if we can avoid creating the embedded SF, since it can cost more than an allreduce */
213b0c7db22SLisandro Dalcin   sub[0] = (PetscBool)(nroots != rpEnd - rpStart);
214b0c7db22SLisandro Dalcin   ierr = MPIU_Allreduce(MPI_IN_PLACE, sub, 2+numFields, MPIU_BOOL, MPI_LOR, PetscObjectComm((PetscObject)sf));CHKERRQ(ierr);
215b0c7db22SLisandro Dalcin   if (sub[0]) {
216b0c7db22SLisandro Dalcin     ierr = ISCreateStride(PETSC_COMM_SELF, rpEnd - rpStart, rpStart, 1, &selected);CHKERRQ(ierr);
217b0c7db22SLisandro Dalcin     ierr = ISGetIndices(selected, &indices);CHKERRQ(ierr);
218b0c7db22SLisandro Dalcin     ierr = PetscSFCreateEmbeddedSF(sf, rpEnd - rpStart, indices, &embedSF);CHKERRQ(ierr);
219b0c7db22SLisandro Dalcin     ierr = ISRestoreIndices(selected, &indices);CHKERRQ(ierr);
220b0c7db22SLisandro Dalcin     ierr = ISDestroy(&selected);CHKERRQ(ierr);
221b0c7db22SLisandro Dalcin   } else {
222b0c7db22SLisandro Dalcin     ierr = PetscObjectReference((PetscObject)sf);CHKERRQ(ierr);
223b0c7db22SLisandro Dalcin     embedSF = sf;
224b0c7db22SLisandro Dalcin   }
225b0c7db22SLisandro Dalcin   ierr = PetscSFGetLeafRange(embedSF, &lpStart, &lpEnd);CHKERRQ(ierr);
226b0c7db22SLisandro Dalcin   lpEnd++;
227b0c7db22SLisandro Dalcin 
228b0c7db22SLisandro Dalcin   ierr = PetscSectionSetChart(leafSection, lpStart, lpEnd);CHKERRQ(ierr);
229b0c7db22SLisandro Dalcin 
230b0c7db22SLisandro Dalcin   /* Constrained dof section */
231b0c7db22SLisandro Dalcin   hasc = sub[1];
232b0c7db22SLisandro Dalcin   for (f = 0; f < numFields; ++f) hasc = (PetscBool)(hasc || sub[2+f]);
233b0c7db22SLisandro Dalcin 
234b0c7db22SLisandro Dalcin   /* Could fuse these at the cost of copies and extra allocation */
235b0c7db22SLisandro Dalcin   ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->atlasDof[-rpStart], &leafSection->atlasDof[-lpStart]);CHKERRQ(ierr);
236b0c7db22SLisandro Dalcin   ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->atlasDof[-rpStart], &leafSection->atlasDof[-lpStart]);CHKERRQ(ierr);
237b0c7db22SLisandro Dalcin   if (sub[1]) {
238b0c7db22SLisandro Dalcin     ierr = PetscSectionCheckConstraints_Static(rootSection);CHKERRQ(ierr);
239b0c7db22SLisandro Dalcin     ierr = PetscSectionCheckConstraints_Static(leafSection);CHKERRQ(ierr);
240b0c7db22SLisandro Dalcin     ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->bc->atlasDof[-rpStart], &leafSection->bc->atlasDof[-lpStart]);CHKERRQ(ierr);
241b0c7db22SLisandro Dalcin     ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->bc->atlasDof[-rpStart], &leafSection->bc->atlasDof[-lpStart]);CHKERRQ(ierr);
242b0c7db22SLisandro Dalcin   }
243b0c7db22SLisandro Dalcin   for (f = 0; f < numFields; ++f) {
244b0c7db22SLisandro Dalcin     ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->field[f]->atlasDof[-rpStart], &leafSection->field[f]->atlasDof[-lpStart]);CHKERRQ(ierr);
245b0c7db22SLisandro Dalcin     ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->field[f]->atlasDof[-rpStart], &leafSection->field[f]->atlasDof[-lpStart]);CHKERRQ(ierr);
246b0c7db22SLisandro Dalcin     if (sub[2+f]) {
247b0c7db22SLisandro Dalcin       ierr = PetscSectionCheckConstraints_Static(rootSection->field[f]);CHKERRQ(ierr);
248b0c7db22SLisandro Dalcin       ierr = PetscSectionCheckConstraints_Static(leafSection->field[f]);CHKERRQ(ierr);
249b0c7db22SLisandro Dalcin       ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->field[f]->bc->atlasDof[-rpStart], &leafSection->field[f]->bc->atlasDof[-lpStart]);CHKERRQ(ierr);
250b0c7db22SLisandro Dalcin       ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->field[f]->bc->atlasDof[-rpStart], &leafSection->field[f]->bc->atlasDof[-lpStart]);CHKERRQ(ierr);
251b0c7db22SLisandro Dalcin     }
252b0c7db22SLisandro Dalcin   }
253b0c7db22SLisandro Dalcin   if (remoteOffsets) {
254b0c7db22SLisandro Dalcin     ierr = PetscMalloc1(lpEnd - lpStart, remoteOffsets);CHKERRQ(ierr);
255b0c7db22SLisandro Dalcin     ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->atlasOff[-rpStart], &(*remoteOffsets)[-lpStart]);CHKERRQ(ierr);
256b0c7db22SLisandro Dalcin     ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->atlasOff[-rpStart], &(*remoteOffsets)[-lpStart]);CHKERRQ(ierr);
257b0c7db22SLisandro Dalcin   }
258b0c7db22SLisandro Dalcin   ierr = PetscSectionSetUp(leafSection);CHKERRQ(ierr);
259b0c7db22SLisandro Dalcin   if (hasc) { /* need to communicate bcIndices */
260b0c7db22SLisandro Dalcin     PetscSF  bcSF;
261b0c7db22SLisandro Dalcin     PetscInt *rOffBc;
262b0c7db22SLisandro Dalcin 
263b0c7db22SLisandro Dalcin     ierr = PetscMalloc1(lpEnd - lpStart, &rOffBc);CHKERRQ(ierr);
264b0c7db22SLisandro Dalcin     if (sub[1]) {
265b0c7db22SLisandro Dalcin       ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->bc->atlasOff[-rpStart], &rOffBc[-lpStart]);CHKERRQ(ierr);
266b0c7db22SLisandro Dalcin       ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->bc->atlasOff[-rpStart], &rOffBc[-lpStart]);CHKERRQ(ierr);
267b0c7db22SLisandro Dalcin       ierr = PetscSFCreateSectionSF(embedSF, rootSection->bc, rOffBc, leafSection->bc, &bcSF);CHKERRQ(ierr);
268b0c7db22SLisandro Dalcin       ierr = PetscSFBcastBegin(bcSF, MPIU_INT, rootSection->bcIndices, leafSection->bcIndices);CHKERRQ(ierr);
269b0c7db22SLisandro Dalcin       ierr = PetscSFBcastEnd(bcSF, MPIU_INT, rootSection->bcIndices, leafSection->bcIndices);CHKERRQ(ierr);
270b0c7db22SLisandro Dalcin       ierr = PetscSFDestroy(&bcSF);CHKERRQ(ierr);
271b0c7db22SLisandro Dalcin     }
272b0c7db22SLisandro Dalcin     for (f = 0; f < numFields; ++f) {
273b0c7db22SLisandro Dalcin       if (sub[2+f]) {
274b0c7db22SLisandro Dalcin         ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->field[f]->bc->atlasOff[-rpStart], &rOffBc[-lpStart]);CHKERRQ(ierr);
275b0c7db22SLisandro Dalcin         ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->field[f]->bc->atlasOff[-rpStart], &rOffBc[-lpStart]);CHKERRQ(ierr);
276b0c7db22SLisandro Dalcin         ierr = PetscSFCreateSectionSF(embedSF, rootSection->field[f]->bc, rOffBc, leafSection->field[f]->bc, &bcSF);CHKERRQ(ierr);
277b0c7db22SLisandro Dalcin         ierr = PetscSFBcastBegin(bcSF, MPIU_INT, rootSection->field[f]->bcIndices, leafSection->field[f]->bcIndices);CHKERRQ(ierr);
278b0c7db22SLisandro Dalcin         ierr = PetscSFBcastEnd(bcSF, MPIU_INT, rootSection->field[f]->bcIndices, leafSection->field[f]->bcIndices);CHKERRQ(ierr);
279b0c7db22SLisandro Dalcin         ierr = PetscSFDestroy(&bcSF);CHKERRQ(ierr);
280b0c7db22SLisandro Dalcin       }
281b0c7db22SLisandro Dalcin     }
282b0c7db22SLisandro Dalcin     ierr = PetscFree(rOffBc);CHKERRQ(ierr);
283b0c7db22SLisandro Dalcin   }
284b0c7db22SLisandro Dalcin   ierr = PetscSFDestroy(&embedSF);CHKERRQ(ierr);
285b0c7db22SLisandro Dalcin   ierr = PetscFree(sub);CHKERRQ(ierr);
286b0c7db22SLisandro Dalcin   ierr = PetscLogEventEnd(PETSCSF_DistSect,sf,0,0,0);CHKERRQ(ierr);
287b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
288b0c7db22SLisandro Dalcin }
289b0c7db22SLisandro Dalcin 
290b0c7db22SLisandro Dalcin /*@C
291b0c7db22SLisandro Dalcin   PetscSFCreateRemoteOffsets - Create offsets for point data on remote processes
292b0c7db22SLisandro Dalcin 
293b0c7db22SLisandro Dalcin   Collective on sf
294b0c7db22SLisandro Dalcin 
295b0c7db22SLisandro Dalcin   Input Parameters:
296b0c7db22SLisandro Dalcin + sf          - The SF
297b0c7db22SLisandro Dalcin . rootSection - Data layout of remote points for outgoing data (this is layout for SF roots)
298b0c7db22SLisandro Dalcin - leafSection - Data layout of local points for incoming data  (this is layout for SF leaves)
299b0c7db22SLisandro Dalcin 
300b0c7db22SLisandro Dalcin   Output Parameter:
301b0c7db22SLisandro Dalcin . remoteOffsets - Offsets for point data on remote processes (these are offsets from the root section), or NULL
302b0c7db22SLisandro Dalcin 
303b0c7db22SLisandro Dalcin   Level: developer
304b0c7db22SLisandro Dalcin 
305b0c7db22SLisandro Dalcin .seealso: PetscSFCreate()
306b0c7db22SLisandro Dalcin @*/
307b0c7db22SLisandro Dalcin PetscErrorCode PetscSFCreateRemoteOffsets(PetscSF sf, PetscSection rootSection, PetscSection leafSection, PetscInt **remoteOffsets)
308b0c7db22SLisandro Dalcin {
309b0c7db22SLisandro Dalcin   PetscSF         embedSF;
310b0c7db22SLisandro Dalcin   const PetscInt *indices;
311b0c7db22SLisandro Dalcin   IS              selected;
312b0c7db22SLisandro Dalcin   PetscInt        numRoots, rpStart = 0, rpEnd = 0, lpStart = 0, lpEnd = 0;
313b0c7db22SLisandro Dalcin   PetscErrorCode  ierr;
314b0c7db22SLisandro Dalcin 
315b0c7db22SLisandro Dalcin   PetscFunctionBegin;
316b0c7db22SLisandro Dalcin   *remoteOffsets = NULL;
317b0c7db22SLisandro Dalcin   ierr = PetscSFGetGraph(sf, &numRoots, NULL, NULL, NULL);CHKERRQ(ierr);
318b0c7db22SLisandro Dalcin   if (numRoots < 0) PetscFunctionReturn(0);
319b0c7db22SLisandro Dalcin   ierr = PetscLogEventBegin(PETSCSF_RemoteOff,sf,0,0,0);CHKERRQ(ierr);
320b0c7db22SLisandro Dalcin   ierr = PetscSectionGetChart(rootSection, &rpStart, &rpEnd);CHKERRQ(ierr);
321b0c7db22SLisandro Dalcin   ierr = PetscSectionGetChart(leafSection, &lpStart, &lpEnd);CHKERRQ(ierr);
322b0c7db22SLisandro Dalcin   ierr = ISCreateStride(PETSC_COMM_SELF, rpEnd - rpStart, rpStart, 1, &selected);CHKERRQ(ierr);
323b0c7db22SLisandro Dalcin   ierr = ISGetIndices(selected, &indices);CHKERRQ(ierr);
324b0c7db22SLisandro Dalcin   ierr = PetscSFCreateEmbeddedSF(sf, rpEnd - rpStart, indices, &embedSF);CHKERRQ(ierr);
325b0c7db22SLisandro Dalcin   ierr = ISRestoreIndices(selected, &indices);CHKERRQ(ierr);
326b0c7db22SLisandro Dalcin   ierr = ISDestroy(&selected);CHKERRQ(ierr);
327b0c7db22SLisandro Dalcin   ierr = PetscCalloc1(lpEnd - lpStart, remoteOffsets);CHKERRQ(ierr);
328b0c7db22SLisandro Dalcin   ierr = PetscSFBcastBegin(embedSF, MPIU_INT, &rootSection->atlasOff[-rpStart], &(*remoteOffsets)[-lpStart]);CHKERRQ(ierr);
329b0c7db22SLisandro Dalcin   ierr = PetscSFBcastEnd(embedSF, MPIU_INT, &rootSection->atlasOff[-rpStart], &(*remoteOffsets)[-lpStart]);CHKERRQ(ierr);
330b0c7db22SLisandro Dalcin   ierr = PetscSFDestroy(&embedSF);CHKERRQ(ierr);
331b0c7db22SLisandro Dalcin   ierr = PetscLogEventEnd(PETSCSF_RemoteOff,sf,0,0,0);CHKERRQ(ierr);
332b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
333b0c7db22SLisandro Dalcin }
334b0c7db22SLisandro Dalcin 
335b0c7db22SLisandro Dalcin /*@C
336b0c7db22SLisandro Dalcin   PetscSFCreateSectionSF - Create an expanded SF of dofs, assuming the input SF relates points
337b0c7db22SLisandro Dalcin 
338b0c7db22SLisandro Dalcin   Collective on sf
339b0c7db22SLisandro Dalcin 
340b0c7db22SLisandro Dalcin   Input Parameters:
341b0c7db22SLisandro Dalcin + sf - The SF
342b0c7db22SLisandro Dalcin . rootSection - Data layout of remote points for outgoing data (this is usually the serial section)
343b0c7db22SLisandro Dalcin . remoteOffsets - Offsets for point data on remote processes (these are offsets from the root section), or NULL
344b0c7db22SLisandro Dalcin - leafSection - Data layout of local points for incoming data  (this is the distributed section)
345b0c7db22SLisandro Dalcin 
346b0c7db22SLisandro Dalcin   Output Parameters:
347b0c7db22SLisandro Dalcin - sectionSF - The new SF
348b0c7db22SLisandro Dalcin 
349b0c7db22SLisandro Dalcin   Note: Either rootSection or remoteOffsets can be specified
350b0c7db22SLisandro Dalcin 
351b0c7db22SLisandro Dalcin   Level: advanced
352b0c7db22SLisandro Dalcin 
353b0c7db22SLisandro Dalcin .seealso: PetscSFCreate()
354b0c7db22SLisandro Dalcin @*/
355b0c7db22SLisandro Dalcin PetscErrorCode PetscSFCreateSectionSF(PetscSF sf, PetscSection rootSection, PetscInt remoteOffsets[], PetscSection leafSection, PetscSF *sectionSF)
356b0c7db22SLisandro Dalcin {
357b0c7db22SLisandro Dalcin   MPI_Comm          comm;
358b0c7db22SLisandro Dalcin   const PetscInt    *localPoints;
359b0c7db22SLisandro Dalcin   const PetscSFNode *remotePoints;
360b0c7db22SLisandro Dalcin   PetscInt          lpStart, lpEnd;
361b0c7db22SLisandro Dalcin   PetscInt          numRoots, numSectionRoots, numPoints, numIndices = 0;
362b0c7db22SLisandro Dalcin   PetscInt          *localIndices;
363b0c7db22SLisandro Dalcin   PetscSFNode       *remoteIndices;
364b0c7db22SLisandro Dalcin   PetscInt          i, ind;
365b0c7db22SLisandro Dalcin   PetscErrorCode    ierr;
366b0c7db22SLisandro Dalcin 
367b0c7db22SLisandro Dalcin   PetscFunctionBegin;
368b0c7db22SLisandro Dalcin   PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1);
369b0c7db22SLisandro Dalcin   PetscValidPointer(rootSection,2);
370b0c7db22SLisandro Dalcin   /* Cannot check PetscValidIntPointer(remoteOffsets,3) because it can be NULL if sf does not reference any points in leafSection */
371b0c7db22SLisandro Dalcin   PetscValidPointer(leafSection,4);
372b0c7db22SLisandro Dalcin   PetscValidPointer(sectionSF,5);
373b0c7db22SLisandro Dalcin   ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr);
374b0c7db22SLisandro Dalcin   ierr = PetscSFCreate(comm, sectionSF);CHKERRQ(ierr);
375b0c7db22SLisandro Dalcin   ierr = PetscSectionGetChart(leafSection, &lpStart, &lpEnd);CHKERRQ(ierr);
376b0c7db22SLisandro Dalcin   ierr = PetscSectionGetStorageSize(rootSection, &numSectionRoots);CHKERRQ(ierr);
377b0c7db22SLisandro Dalcin   ierr = PetscSFGetGraph(sf, &numRoots, &numPoints, &localPoints, &remotePoints);CHKERRQ(ierr);
378b0c7db22SLisandro Dalcin   if (numRoots < 0) PetscFunctionReturn(0);
379b0c7db22SLisandro Dalcin   ierr = PetscLogEventBegin(PETSCSF_SectSF,sf,0,0,0);CHKERRQ(ierr);
380b0c7db22SLisandro Dalcin   for (i = 0; i < numPoints; ++i) {
381b0c7db22SLisandro Dalcin     PetscInt localPoint = localPoints ? localPoints[i] : i;
382b0c7db22SLisandro Dalcin     PetscInt dof;
383b0c7db22SLisandro Dalcin 
384b0c7db22SLisandro Dalcin     if ((localPoint >= lpStart) && (localPoint < lpEnd)) {
385b0c7db22SLisandro Dalcin       ierr = PetscSectionGetDof(leafSection, localPoint, &dof);CHKERRQ(ierr);
386b0c7db22SLisandro Dalcin       numIndices += dof;
387b0c7db22SLisandro Dalcin     }
388b0c7db22SLisandro Dalcin   }
389b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(numIndices, &localIndices);CHKERRQ(ierr);
390b0c7db22SLisandro Dalcin   ierr = PetscMalloc1(numIndices, &remoteIndices);CHKERRQ(ierr);
391b0c7db22SLisandro Dalcin   /* Create new index graph */
392b0c7db22SLisandro Dalcin   for (i = 0, ind = 0; i < numPoints; ++i) {
393b0c7db22SLisandro Dalcin     PetscInt localPoint = localPoints ? localPoints[i] : i;
394b0c7db22SLisandro Dalcin     PetscInt rank       = remotePoints[i].rank;
395b0c7db22SLisandro Dalcin 
396b0c7db22SLisandro Dalcin     if ((localPoint >= lpStart) && (localPoint < lpEnd)) {
397b0c7db22SLisandro Dalcin       PetscInt remoteOffset = remoteOffsets[localPoint-lpStart];
398b0c7db22SLisandro Dalcin       PetscInt loff, dof, d;
399b0c7db22SLisandro Dalcin 
400b0c7db22SLisandro Dalcin       ierr = PetscSectionGetOffset(leafSection, localPoint, &loff);CHKERRQ(ierr);
401b0c7db22SLisandro Dalcin       ierr = PetscSectionGetDof(leafSection, localPoint, &dof);CHKERRQ(ierr);
402b0c7db22SLisandro Dalcin       for (d = 0; d < dof; ++d, ++ind) {
403b0c7db22SLisandro Dalcin         localIndices[ind]        = loff+d;
404b0c7db22SLisandro Dalcin         remoteIndices[ind].rank  = rank;
405b0c7db22SLisandro Dalcin         remoteIndices[ind].index = remoteOffset+d;
406b0c7db22SLisandro Dalcin       }
407b0c7db22SLisandro Dalcin     }
408b0c7db22SLisandro Dalcin   }
409b0c7db22SLisandro Dalcin   if (numIndices != ind) SETERRQ2(comm, PETSC_ERR_PLIB, "Inconsistency in indices, %D should be %D", ind, numIndices);
410b0c7db22SLisandro Dalcin   ierr = PetscSFSetGraph(*sectionSF, numSectionRoots, numIndices, localIndices, PETSC_OWN_POINTER, remoteIndices, PETSC_OWN_POINTER);CHKERRQ(ierr);
411b0c7db22SLisandro Dalcin   ierr = PetscSFSetUp(*sectionSF);CHKERRQ(ierr);
412b0c7db22SLisandro Dalcin   ierr = PetscLogEventEnd(PETSCSF_SectSF,sf,0,0,0);CHKERRQ(ierr);
413b0c7db22SLisandro Dalcin   PetscFunctionReturn(0);
414b0c7db22SLisandro Dalcin }
415