xref: /petsc/src/dm/impls/plex/plexdistribute.c (revision cc71bff1308b5c843157f5fa098d26a9289e3ffb)
170034214SMatthew G. Knepley #include <petsc-private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
270034214SMatthew G. Knepley #include <petscsf.h>
370034214SMatthew G. Knepley 
470034214SMatthew G. Knepley #undef __FUNCT__
570034214SMatthew G. Knepley #define __FUNCT__ "DMPlexSetAdjacencyUseCone"
670034214SMatthew G. Knepley /*@
770034214SMatthew G. Knepley   DMPlexSetAdjacencyUseCone - Define adjacency in the mesh using either the cone or the support first
870034214SMatthew G. Knepley 
970034214SMatthew G. Knepley   Input Parameters:
1070034214SMatthew G. Knepley + dm      - The DM object
1170034214SMatthew G. Knepley - useCone - Flag to use the cone first
1270034214SMatthew G. Knepley 
1370034214SMatthew G. Knepley   Level: intermediate
1470034214SMatthew G. Knepley 
1570034214SMatthew G. Knepley   Notes:
1670034214SMatthew G. Knepley $     FEM:   Two points p and q are adjacent if q \in closure(star(p)), useCone = PETSC_FALSE, useClosure = PETSC_TRUE
1770034214SMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in star(cone(p)),    useCone = PETSC_TRUE,  useClosure = PETSC_FALSE
1870034214SMatthew G. Knepley $     FVM++: Two points p and q are adjacent if q \in star(closure(p)), useCone = PETSC_TRUE,  useClosure = PETSC_TRUE
1970034214SMatthew G. Knepley 
2070034214SMatthew G. Knepley .seealso: DMPlexGetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure(), DMPlexGetAdjacencyUseClosure(), DMPlexDistribute(), DMPlexPreallocateOperator()
2170034214SMatthew G. Knepley @*/
2270034214SMatthew G. Knepley PetscErrorCode DMPlexSetAdjacencyUseCone(DM dm, PetscBool useCone)
2370034214SMatthew G. Knepley {
2470034214SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *) dm->data;
2570034214SMatthew G. Knepley 
2670034214SMatthew G. Knepley   PetscFunctionBegin;
2770034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2870034214SMatthew G. Knepley   mesh->useCone = useCone;
2970034214SMatthew G. Knepley   PetscFunctionReturn(0);
3070034214SMatthew G. Knepley }
3170034214SMatthew G. Knepley 
3270034214SMatthew G. Knepley #undef __FUNCT__
3370034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacencyUseCone"
3470034214SMatthew G. Knepley /*@
3570034214SMatthew G. Knepley   DMPlexGetAdjacencyUseCone - Query whether adjacency in the mesh uses the cone or the support first
3670034214SMatthew G. Knepley 
3770034214SMatthew G. Knepley   Input Parameter:
3870034214SMatthew G. Knepley . dm      - The DM object
3970034214SMatthew G. Knepley 
4070034214SMatthew G. Knepley   Output Parameter:
4170034214SMatthew G. Knepley . useCone - Flag to use the cone first
4270034214SMatthew G. Knepley 
4370034214SMatthew G. Knepley   Level: intermediate
4470034214SMatthew G. Knepley 
4570034214SMatthew G. Knepley   Notes:
4670034214SMatthew G. Knepley $     FEM:   Two points p and q are adjacent if q \in closure(star(p)), useCone = PETSC_FALSE, useClosure = PETSC_TRUE
4770034214SMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in star(cone(p)),    useCone = PETSC_TRUE,  useClosure = PETSC_FALSE
4870034214SMatthew G. Knepley $     FVM++: Two points p and q are adjacent if q \in star(closure(p)), useCone = PETSC_TRUE,  useClosure = PETSC_TRUE
4970034214SMatthew G. Knepley 
5070034214SMatthew G. Knepley .seealso: DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure(), DMPlexGetAdjacencyUseClosure(), DMPlexDistribute(), DMPlexPreallocateOperator()
5170034214SMatthew G. Knepley @*/
5270034214SMatthew G. Knepley PetscErrorCode DMPlexGetAdjacencyUseCone(DM dm, PetscBool *useCone)
5370034214SMatthew G. Knepley {
5470034214SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *) dm->data;
5570034214SMatthew G. Knepley 
5670034214SMatthew G. Knepley   PetscFunctionBegin;
5770034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5870034214SMatthew G. Knepley   PetscValidIntPointer(useCone, 2);
5970034214SMatthew G. Knepley   *useCone = mesh->useCone;
6070034214SMatthew G. Knepley   PetscFunctionReturn(0);
6170034214SMatthew G. Knepley }
6270034214SMatthew G. Knepley 
6370034214SMatthew G. Knepley #undef __FUNCT__
6470034214SMatthew G. Knepley #define __FUNCT__ "DMPlexSetAdjacencyUseClosure"
6570034214SMatthew G. Knepley /*@
6670034214SMatthew G. Knepley   DMPlexSetAdjacencyUseClosure - Define adjacency in the mesh using the transitive closure
6770034214SMatthew G. Knepley 
6870034214SMatthew G. Knepley   Input Parameters:
6970034214SMatthew G. Knepley + dm      - The DM object
7070034214SMatthew G. Knepley - useClosure - Flag to use the closure
7170034214SMatthew G. Knepley 
7270034214SMatthew G. Knepley   Level: intermediate
7370034214SMatthew G. Knepley 
7470034214SMatthew G. Knepley   Notes:
7570034214SMatthew G. Knepley $     FEM:   Two points p and q are adjacent if q \in closure(star(p)), useCone = PETSC_FALSE, useClosure = PETSC_TRUE
7670034214SMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in star(cone(p)),    useCone = PETSC_TRUE,  useClosure = PETSC_FALSE
7770034214SMatthew G. Knepley $     FVM++: Two points p and q are adjacent if q \in star(closure(p)), useCone = PETSC_TRUE,  useClosure = PETSC_TRUE
7870034214SMatthew G. Knepley 
7970034214SMatthew G. Knepley .seealso: DMPlexGetAdjacencyUseClosure(), DMPlexSetAdjacencyUseCone(), DMPlexGetAdjacencyUseCone(), DMPlexDistribute(), DMPlexPreallocateOperator()
8070034214SMatthew G. Knepley @*/
8170034214SMatthew G. Knepley PetscErrorCode DMPlexSetAdjacencyUseClosure(DM dm, PetscBool useClosure)
8270034214SMatthew G. Knepley {
8370034214SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *) dm->data;
8470034214SMatthew G. Knepley 
8570034214SMatthew G. Knepley   PetscFunctionBegin;
8670034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8770034214SMatthew G. Knepley   mesh->useClosure = useClosure;
8870034214SMatthew G. Knepley   PetscFunctionReturn(0);
8970034214SMatthew G. Knepley }
9070034214SMatthew G. Knepley 
9170034214SMatthew G. Knepley #undef __FUNCT__
9270034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacencyUseClosure"
9370034214SMatthew G. Knepley /*@
9470034214SMatthew G. Knepley   DMPlexGetAdjacencyUseClosure - Query whether adjacency in the mesh uses the transitive closure
9570034214SMatthew G. Knepley 
9670034214SMatthew G. Knepley   Input Parameter:
9770034214SMatthew G. Knepley . dm      - The DM object
9870034214SMatthew G. Knepley 
9970034214SMatthew G. Knepley   Output Parameter:
10070034214SMatthew G. Knepley . useClosure - Flag to use the closure
10170034214SMatthew G. Knepley 
10270034214SMatthew G. Knepley   Level: intermediate
10370034214SMatthew G. Knepley 
10470034214SMatthew G. Knepley   Notes:
10570034214SMatthew G. Knepley $     FEM:   Two points p and q are adjacent if q \in closure(star(p)), useCone = PETSC_FALSE, useClosure = PETSC_TRUE
10670034214SMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in star(cone(p)),    useCone = PETSC_TRUE,  useClosure = PETSC_FALSE
10770034214SMatthew G. Knepley $     FVM++: Two points p and q are adjacent if q \in star(closure(p)), useCone = PETSC_TRUE,  useClosure = PETSC_TRUE
10870034214SMatthew G. Knepley 
10970034214SMatthew G. Knepley .seealso: DMPlexSetAdjacencyUseClosure(), DMPlexSetAdjacencyUseCone(), DMPlexGetAdjacencyUseCone(), DMPlexDistribute(), DMPlexPreallocateOperator()
11070034214SMatthew G. Knepley @*/
11170034214SMatthew G. Knepley PetscErrorCode DMPlexGetAdjacencyUseClosure(DM dm, PetscBool *useClosure)
11270034214SMatthew G. Knepley {
11370034214SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *) dm->data;
11470034214SMatthew G. Knepley 
11570034214SMatthew G. Knepley   PetscFunctionBegin;
11670034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
11770034214SMatthew G. Knepley   PetscValidIntPointer(useClosure, 2);
11870034214SMatthew G. Knepley   *useClosure = mesh->useClosure;
11970034214SMatthew G. Knepley   PetscFunctionReturn(0);
12070034214SMatthew G. Knepley }
12170034214SMatthew G. Knepley 
12270034214SMatthew G. Knepley #undef __FUNCT__
123a17985deSToby Isaac #define __FUNCT__ "DMPlexSetAdjacencyUseAnchors"
1248b0b4c70SToby Isaac /*@
125a17985deSToby Isaac   DMPlexSetAdjacencyUseAnchors - Define adjacency in the mesh using the point-to-point constraints.
1268b0b4c70SToby Isaac 
1278b0b4c70SToby Isaac   Input Parameters:
1288b0b4c70SToby Isaac + dm      - The DM object
1295b317d89SToby Isaac - useAnchors - Flag to use the constraints.  If PETSC_TRUE, then constrained points are omitted from DMPlexGetAdjacency(), and their anchor points appear in their place.
1308b0b4c70SToby Isaac 
1318b0b4c70SToby Isaac   Level: intermediate
1328b0b4c70SToby Isaac 
133a17985deSToby Isaac .seealso: DMPlexGetAdjacencyUseClosure(), DMPlexSetAdjacencyUseCone(), DMPlexGetAdjacencyUseCone(), DMPlexDistribute(), DMPlexPreallocateOperator(), DMPlexSetAnchors()
1348b0b4c70SToby Isaac @*/
1355b317d89SToby Isaac PetscErrorCode DMPlexSetAdjacencyUseAnchors(DM dm, PetscBool useAnchors)
1368b0b4c70SToby Isaac {
1378b0b4c70SToby Isaac   DM_Plex *mesh = (DM_Plex *) dm->data;
1388b0b4c70SToby Isaac 
1398b0b4c70SToby Isaac   PetscFunctionBegin;
1408b0b4c70SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1415b317d89SToby Isaac   mesh->useAnchors = useAnchors;
1428b0b4c70SToby Isaac   PetscFunctionReturn(0);
1438b0b4c70SToby Isaac }
1448b0b4c70SToby Isaac 
1458b0b4c70SToby Isaac #undef __FUNCT__
146a17985deSToby Isaac #define __FUNCT__ "DMPlexGetAdjacencyUseAnchors"
1478b0b4c70SToby Isaac /*@
148a17985deSToby Isaac   DMPlexGetAdjacencyUseAnchors - Query whether adjacency in the mesh uses the point-to-point constraints.
1498b0b4c70SToby Isaac 
1508b0b4c70SToby Isaac   Input Parameter:
1518b0b4c70SToby Isaac . dm      - The DM object
1528b0b4c70SToby Isaac 
1538b0b4c70SToby Isaac   Output Parameter:
1545b317d89SToby Isaac . useAnchors - Flag to use the closure.  If PETSC_TRUE, then constrained points are omitted from DMPlexGetAdjacency(), and their anchor points appear in their place.
1558b0b4c70SToby Isaac 
1568b0b4c70SToby Isaac   Level: intermediate
1578b0b4c70SToby Isaac 
158a17985deSToby Isaac .seealso: DMPlexSetAdjacencyUseAnchors(), DMPlexSetAdjacencyUseCone(), DMPlexGetAdjacencyUseCone(), DMPlexDistribute(), DMPlexPreallocateOperator(), DMPlexSetAnchors()
1598b0b4c70SToby Isaac @*/
1605b317d89SToby Isaac PetscErrorCode DMPlexGetAdjacencyUseAnchors(DM dm, PetscBool *useAnchors)
1618b0b4c70SToby Isaac {
1628b0b4c70SToby Isaac   DM_Plex *mesh = (DM_Plex *) dm->data;
1638b0b4c70SToby Isaac 
1648b0b4c70SToby Isaac   PetscFunctionBegin;
1658b0b4c70SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1665b317d89SToby Isaac   PetscValidIntPointer(useAnchors, 2);
1675b317d89SToby Isaac   *useAnchors = mesh->useAnchors;
1688b0b4c70SToby Isaac   PetscFunctionReturn(0);
1698b0b4c70SToby Isaac }
1708b0b4c70SToby Isaac 
1718b0b4c70SToby Isaac #undef __FUNCT__
17270034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Cone_Internal"
17370034214SMatthew G. Knepley static PetscErrorCode DMPlexGetAdjacency_Cone_Internal(DM dm, PetscInt p, PetscInt *adjSize, PetscInt adj[])
17470034214SMatthew G. Knepley {
17570034214SMatthew G. Knepley   const PetscInt *cone = NULL;
17670034214SMatthew G. Knepley   PetscInt        numAdj = 0, maxAdjSize = *adjSize, coneSize, c;
17770034214SMatthew G. Knepley   PetscErrorCode  ierr;
17870034214SMatthew G. Knepley 
17970034214SMatthew G. Knepley   PetscFunctionBeginHot;
18070034214SMatthew G. Knepley   ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
18170034214SMatthew G. Knepley   ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
18270034214SMatthew G. Knepley   for (c = 0; c < coneSize; ++c) {
18370034214SMatthew G. Knepley     const PetscInt *support = NULL;
18470034214SMatthew G. Knepley     PetscInt        supportSize, s, q;
18570034214SMatthew G. Knepley 
18670034214SMatthew G. Knepley     ierr = DMPlexGetSupportSize(dm, cone[c], &supportSize);CHKERRQ(ierr);
18770034214SMatthew G. Knepley     ierr = DMPlexGetSupport(dm, cone[c], &support);CHKERRQ(ierr);
18870034214SMatthew G. Knepley     for (s = 0; s < supportSize; ++s) {
18970034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = support[s],0); ++q) {
19070034214SMatthew G. Knepley         if (support[s] == adj[q]) break;
19170034214SMatthew G. Knepley       }
19270034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
19370034214SMatthew G. Knepley     }
19470034214SMatthew G. Knepley   }
19570034214SMatthew G. Knepley   *adjSize = numAdj;
19670034214SMatthew G. Knepley   PetscFunctionReturn(0);
19770034214SMatthew G. Knepley }
19870034214SMatthew G. Knepley 
19970034214SMatthew G. Knepley #undef __FUNCT__
20070034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Support_Internal"
20170034214SMatthew G. Knepley static PetscErrorCode DMPlexGetAdjacency_Support_Internal(DM dm, PetscInt p, PetscInt *adjSize, PetscInt adj[])
20270034214SMatthew G. Knepley {
20370034214SMatthew G. Knepley   const PetscInt *support = NULL;
20470034214SMatthew G. Knepley   PetscInt        numAdj   = 0, maxAdjSize = *adjSize, supportSize, s;
20570034214SMatthew G. Knepley   PetscErrorCode  ierr;
20670034214SMatthew G. Knepley 
20770034214SMatthew G. Knepley   PetscFunctionBeginHot;
20870034214SMatthew G. Knepley   ierr = DMPlexGetSupportSize(dm, p, &supportSize);CHKERRQ(ierr);
20970034214SMatthew G. Knepley   ierr = DMPlexGetSupport(dm, p, &support);CHKERRQ(ierr);
21070034214SMatthew G. Knepley   for (s = 0; s < supportSize; ++s) {
21170034214SMatthew G. Knepley     const PetscInt *cone = NULL;
21270034214SMatthew G. Knepley     PetscInt        coneSize, c, q;
21370034214SMatthew G. Knepley 
21470034214SMatthew G. Knepley     ierr = DMPlexGetConeSize(dm, support[s], &coneSize);CHKERRQ(ierr);
21570034214SMatthew G. Knepley     ierr = DMPlexGetCone(dm, support[s], &cone);CHKERRQ(ierr);
21670034214SMatthew G. Knepley     for (c = 0; c < coneSize; ++c) {
21770034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = cone[c],0); ++q) {
21870034214SMatthew G. Knepley         if (cone[c] == adj[q]) break;
21970034214SMatthew G. Knepley       }
22070034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
22170034214SMatthew G. Knepley     }
22270034214SMatthew G. Knepley   }
22370034214SMatthew G. Knepley   *adjSize = numAdj;
22470034214SMatthew G. Knepley   PetscFunctionReturn(0);
22570034214SMatthew G. Knepley }
22670034214SMatthew G. Knepley 
22770034214SMatthew G. Knepley #undef __FUNCT__
22870034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Transitive_Internal"
22970034214SMatthew G. Knepley static PetscErrorCode DMPlexGetAdjacency_Transitive_Internal(DM dm, PetscInt p, PetscBool useClosure, PetscInt *adjSize, PetscInt adj[])
23070034214SMatthew G. Knepley {
23170034214SMatthew G. Knepley   PetscInt      *star = NULL;
23270034214SMatthew G. Knepley   PetscInt       numAdj = 0, maxAdjSize = *adjSize, starSize, s;
23370034214SMatthew G. Knepley   PetscErrorCode ierr;
23470034214SMatthew G. Knepley 
23570034214SMatthew G. Knepley   PetscFunctionBeginHot;
23670034214SMatthew G. Knepley   ierr = DMPlexGetTransitiveClosure(dm, p, useClosure, &starSize, &star);CHKERRQ(ierr);
23770034214SMatthew G. Knepley   for (s = 0; s < starSize*2; s += 2) {
23870034214SMatthew G. Knepley     const PetscInt *closure = NULL;
23970034214SMatthew G. Knepley     PetscInt        closureSize, c, q;
24070034214SMatthew G. Knepley 
24170034214SMatthew G. Knepley     ierr = DMPlexGetTransitiveClosure(dm, star[s], (PetscBool)!useClosure, &closureSize, (PetscInt**) &closure);CHKERRQ(ierr);
24270034214SMatthew G. Knepley     for (c = 0; c < closureSize*2; c += 2) {
24370034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = closure[c],0); ++q) {
24470034214SMatthew G. Knepley         if (closure[c] == adj[q]) break;
24570034214SMatthew G. Knepley       }
24670034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
24770034214SMatthew G. Knepley     }
24870034214SMatthew G. Knepley     ierr = DMPlexRestoreTransitiveClosure(dm, star[s], (PetscBool)!useClosure, &closureSize, (PetscInt**) &closure);CHKERRQ(ierr);
24970034214SMatthew G. Knepley   }
25070034214SMatthew G. Knepley   ierr = DMPlexRestoreTransitiveClosure(dm, p, useClosure, &starSize, &star);CHKERRQ(ierr);
25170034214SMatthew G. Knepley   *adjSize = numAdj;
25270034214SMatthew G. Knepley   PetscFunctionReturn(0);
25370034214SMatthew G. Knepley }
25470034214SMatthew G. Knepley 
25570034214SMatthew G. Knepley #undef __FUNCT__
25670034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Internal"
2575b317d89SToby Isaac PetscErrorCode DMPlexGetAdjacency_Internal(DM dm, PetscInt p, PetscBool useCone, PetscBool useTransitiveClosure, PetscBool useAnchors, PetscInt *adjSize, PetscInt *adj[])
25870034214SMatthew G. Knepley {
25979bad331SMatthew G. Knepley   static PetscInt asiz = 0;
2608b0b4c70SToby Isaac   PetscInt maxAnchors = 1;
2618b0b4c70SToby Isaac   PetscInt aStart = -1, aEnd = -1;
2628b0b4c70SToby Isaac   PetscInt maxAdjSize;
2638b0b4c70SToby Isaac   PetscSection aSec = NULL;
2648b0b4c70SToby Isaac   IS aIS = NULL;
2658b0b4c70SToby Isaac   const PetscInt *anchors;
26670034214SMatthew G. Knepley   PetscErrorCode  ierr;
26770034214SMatthew G. Knepley 
26870034214SMatthew G. Knepley   PetscFunctionBeginHot;
2695b317d89SToby Isaac   if (useAnchors) {
270a17985deSToby Isaac     ierr = DMPlexGetAnchors(dm,&aSec,&aIS);CHKERRQ(ierr);
2718b0b4c70SToby Isaac     if (aSec) {
2728b0b4c70SToby Isaac       ierr = PetscSectionGetMaxDof(aSec,&maxAnchors);CHKERRQ(ierr);
27324c766afSToby Isaac       maxAnchors = PetscMax(1,maxAnchors);
2748b0b4c70SToby Isaac       ierr = PetscSectionGetChart(aSec,&aStart,&aEnd);CHKERRQ(ierr);
2758b0b4c70SToby Isaac       ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
2768b0b4c70SToby Isaac     }
2778b0b4c70SToby Isaac   }
27879bad331SMatthew G. Knepley   if (!*adj) {
27924c766afSToby Isaac     PetscInt depth, coneSeries, supportSeries, maxC, maxS, pStart, pEnd;
28079bad331SMatthew G. Knepley 
28124c766afSToby Isaac     ierr  = DMPlexGetChart(dm, &pStart,&pEnd);CHKERRQ(ierr);
28279bad331SMatthew G. Knepley     ierr  = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
28324c766afSToby Isaac     ierr  = DMPlexGetMaxSizes(dm, &maxC, &maxS);CHKERRQ(ierr);
28424c766afSToby Isaac     coneSeries    = (maxC > 1) ? ((PetscPowInt(maxC,depth+1)-1)/(maxC-1)) : depth+1;
28524c766afSToby Isaac     supportSeries = (maxS > 1) ? ((PetscPowInt(maxS,depth+1)-1)/(maxS-1)) : depth+1;
28624c766afSToby Isaac     asiz  = PetscMax(PetscPowInt(maxS,depth)*coneSeries,PetscPowInt(maxC,depth)*supportSeries);
2878b0b4c70SToby Isaac     asiz *= maxAnchors;
28824c766afSToby Isaac     asiz  = PetscMin(asiz,pEnd-pStart);
28979bad331SMatthew G. Knepley     ierr  = PetscMalloc1(asiz,adj);CHKERRQ(ierr);
29079bad331SMatthew G. Knepley   }
29179bad331SMatthew G. Knepley   if (*adjSize < 0) *adjSize = asiz;
2928b0b4c70SToby Isaac   maxAdjSize = *adjSize;
29370034214SMatthew G. Knepley   if (useTransitiveClosure) {
29479bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Transitive_Internal(dm, p, useCone, adjSize, *adj);CHKERRQ(ierr);
29570034214SMatthew G. Knepley   } else if (useCone) {
29679bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Cone_Internal(dm, p, adjSize, *adj);CHKERRQ(ierr);
29770034214SMatthew G. Knepley   } else {
29879bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Support_Internal(dm, p, adjSize, *adj);CHKERRQ(ierr);
29970034214SMatthew G. Knepley   }
3005b317d89SToby Isaac   if (useAnchors && aSec) {
3018b0b4c70SToby Isaac     PetscInt origSize = *adjSize;
3028b0b4c70SToby Isaac     PetscInt numAdj = origSize;
3038b0b4c70SToby Isaac     PetscInt i = 0, j;
3048b0b4c70SToby Isaac     PetscInt *orig = *adj;
3058b0b4c70SToby Isaac 
3068b0b4c70SToby Isaac     while (i < origSize) {
3078b0b4c70SToby Isaac       PetscInt p = orig[i];
3088b0b4c70SToby Isaac       PetscInt aDof = 0;
3098b0b4c70SToby Isaac 
3108b0b4c70SToby Isaac       if (p >= aStart && p < aEnd) {
3118b0b4c70SToby Isaac         ierr = PetscSectionGetDof(aSec,p,&aDof);CHKERRQ(ierr);
3128b0b4c70SToby Isaac       }
3138b0b4c70SToby Isaac       if (aDof) {
3148b0b4c70SToby Isaac         PetscInt aOff;
3158b0b4c70SToby Isaac         PetscInt s, q;
3168b0b4c70SToby Isaac 
3178b0b4c70SToby Isaac         for (j = i + 1; j < numAdj; j++) {
3188b0b4c70SToby Isaac           orig[j - 1] = orig[j];
3198b0b4c70SToby Isaac         }
3208b0b4c70SToby Isaac         origSize--;
3218b0b4c70SToby Isaac         numAdj--;
3228b0b4c70SToby Isaac         ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
3238b0b4c70SToby Isaac         for (s = 0; s < aDof; ++s) {
3248b0b4c70SToby Isaac           for (q = 0; q < numAdj || (orig[numAdj++] = anchors[aOff+s],0); ++q) {
3258b0b4c70SToby Isaac             if (anchors[aOff+s] == orig[q]) break;
3268b0b4c70SToby Isaac           }
3278b0b4c70SToby Isaac           if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
3288b0b4c70SToby Isaac         }
3298b0b4c70SToby Isaac       }
3308b0b4c70SToby Isaac       else {
3318b0b4c70SToby Isaac         i++;
3328b0b4c70SToby Isaac       }
3338b0b4c70SToby Isaac     }
3348b0b4c70SToby Isaac     *adjSize = numAdj;
3358b0b4c70SToby Isaac     ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
3368b0b4c70SToby Isaac   }
33770034214SMatthew G. Knepley   PetscFunctionReturn(0);
33870034214SMatthew G. Knepley }
33970034214SMatthew G. Knepley 
34070034214SMatthew G. Knepley #undef __FUNCT__
34170034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency"
34270034214SMatthew G. Knepley /*@
34370034214SMatthew G. Knepley   DMPlexGetAdjacency - Return all points adjacent to the given point
34470034214SMatthew G. Knepley 
34570034214SMatthew G. Knepley   Input Parameters:
34670034214SMatthew G. Knepley + dm - The DM object
34770034214SMatthew G. Knepley . p  - The point
34870034214SMatthew G. Knepley . adjSize - The maximum size of adj if it is non-NULL, or PETSC_DETERMINE
34970034214SMatthew G. Knepley - adj - Either NULL so that the array is allocated, or an existing array with size adjSize
35070034214SMatthew G. Knepley 
35170034214SMatthew G. Knepley   Output Parameters:
35270034214SMatthew G. Knepley + adjSize - The number of adjacent points
35370034214SMatthew G. Knepley - adj - The adjacent points
35470034214SMatthew G. Knepley 
35570034214SMatthew G. Knepley   Level: advanced
35670034214SMatthew G. Knepley 
35770034214SMatthew G. Knepley   Notes: The user must PetscFree the adj array if it was not passed in.
35870034214SMatthew G. Knepley 
35970034214SMatthew G. Knepley .seealso: DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure(), DMPlexDistribute(), DMCreateMatrix(), DMPlexPreallocateOperator()
36070034214SMatthew G. Knepley @*/
36170034214SMatthew G. Knepley PetscErrorCode DMPlexGetAdjacency(DM dm, PetscInt p, PetscInt *adjSize, PetscInt *adj[])
36270034214SMatthew G. Knepley {
36370034214SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *) dm->data;
36470034214SMatthew G. Knepley   PetscErrorCode ierr;
36570034214SMatthew G. Knepley 
36670034214SMatthew G. Knepley   PetscFunctionBeginHot;
36770034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
36870034214SMatthew G. Knepley   PetscValidPointer(adjSize,3);
36970034214SMatthew G. Knepley   PetscValidPointer(adj,4);
3705b317d89SToby Isaac   ierr = DMPlexGetAdjacency_Internal(dm, p, mesh->useCone, mesh->useClosure, mesh->useAnchors, adjSize, adj);CHKERRQ(ierr);
37170034214SMatthew G. Knepley   PetscFunctionReturn(0);
37270034214SMatthew G. Knepley }
373b0a623aaSMatthew G. Knepley #undef __FUNCT__
374b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexCreateTwoSidedProcessSF"
375b0a623aaSMatthew G. Knepley /*@
376b0a623aaSMatthew G. Knepley   DMPlexCreateTwoSidedProcessSF - Create an SF which just has process connectivity
377b0a623aaSMatthew G. Knepley 
378b0a623aaSMatthew G. Knepley   Collective on DM
379b0a623aaSMatthew G. Knepley 
380b0a623aaSMatthew G. Knepley   Input Parameters:
381b0a623aaSMatthew G. Knepley + dm      - The DM
382b0a623aaSMatthew G. Knepley - sfPoint - The PetscSF which encodes point connectivity
383b0a623aaSMatthew G. Knepley 
384b0a623aaSMatthew G. Knepley   Output Parameters:
385b0a623aaSMatthew G. Knepley + processRanks - A list of process neighbors, or NULL
386b0a623aaSMatthew G. Knepley - sfProcess    - An SF encoding the two-sided process connectivity, or NULL
387b0a623aaSMatthew G. Knepley 
388b0a623aaSMatthew G. Knepley   Level: developer
389b0a623aaSMatthew G. Knepley 
390b0a623aaSMatthew G. Knepley .seealso: PetscSFCreate(), DMPlexCreateProcessSF()
391b0a623aaSMatthew G. Knepley @*/
392b0a623aaSMatthew G. Knepley PetscErrorCode DMPlexCreateTwoSidedProcessSF(DM dm, PetscSF sfPoint, PetscSection rootRankSection, IS rootRanks, PetscSection leafRankSection, IS leafRanks, IS *processRanks, PetscSF *sfProcess)
393b0a623aaSMatthew G. Knepley {
394b0a623aaSMatthew G. Knepley   const PetscSFNode *remotePoints;
395b0a623aaSMatthew G. Knepley   PetscInt          *localPointsNew;
396b0a623aaSMatthew G. Knepley   PetscSFNode       *remotePointsNew;
397b0a623aaSMatthew G. Knepley   const PetscInt    *nranks;
398b0a623aaSMatthew G. Knepley   PetscInt          *ranksNew;
399b0a623aaSMatthew G. Knepley   PetscBT            neighbors;
400b0a623aaSMatthew G. Knepley   PetscInt           pStart, pEnd, p, numLeaves, l, numNeighbors, n;
401b0a623aaSMatthew G. Knepley   PetscMPIInt        numProcs, proc, rank;
402b0a623aaSMatthew G. Knepley   PetscErrorCode     ierr;
403b0a623aaSMatthew G. Knepley 
404b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
405b0a623aaSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
406b0a623aaSMatthew G. Knepley   PetscValidHeaderSpecific(sfPoint, PETSCSF_CLASSID, 2);
407b0a623aaSMatthew G. Knepley   if (processRanks) {PetscValidPointer(processRanks, 3);}
408b0a623aaSMatthew G. Knepley   if (sfProcess)    {PetscValidPointer(sfProcess, 4);}
409b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm), &numProcs);CHKERRQ(ierr);
410b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
411b0a623aaSMatthew G. Knepley   ierr = PetscSFGetGraph(sfPoint, NULL, &numLeaves, NULL, &remotePoints);CHKERRQ(ierr);
412b0a623aaSMatthew G. Knepley   ierr = PetscBTCreate(numProcs, &neighbors);CHKERRQ(ierr);
413b0a623aaSMatthew G. Knepley   ierr = PetscBTMemzero(numProcs, neighbors);CHKERRQ(ierr);
414b0a623aaSMatthew G. Knepley   /* Compute root-to-leaf process connectivity */
415b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(rootRankSection, &pStart, &pEnd);CHKERRQ(ierr);
416b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(rootRanks, &nranks);CHKERRQ(ierr);
417b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
418b0a623aaSMatthew G. Knepley     PetscInt ndof, noff, n;
419b0a623aaSMatthew G. Knepley 
420b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(rootRankSection, p, &ndof);CHKERRQ(ierr);
421b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(rootRankSection, p, &noff);CHKERRQ(ierr);
422b0a623aaSMatthew G. Knepley     for (n = 0; n < ndof; ++n) {ierr = PetscBTSet(neighbors, nranks[noff+n]);}
423b0a623aaSMatthew G. Knepley   }
424b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(rootRanks, &nranks);CHKERRQ(ierr);
425b0a623aaSMatthew G. Knepley   /* Compute leaf-to-neighbor process connectivity */
426b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(leafRankSection, &pStart, &pEnd);CHKERRQ(ierr);
427b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(leafRanks, &nranks);CHKERRQ(ierr);
428b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
429b0a623aaSMatthew G. Knepley     PetscInt ndof, noff, n;
430b0a623aaSMatthew G. Knepley 
431b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(leafRankSection, p, &ndof);CHKERRQ(ierr);
432b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(leafRankSection, p, &noff);CHKERRQ(ierr);
433b0a623aaSMatthew G. Knepley     for (n = 0; n < ndof; ++n) {ierr = PetscBTSet(neighbors, nranks[noff+n]);}
434b0a623aaSMatthew G. Knepley   }
435b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(leafRanks, &nranks);CHKERRQ(ierr);
436b0a623aaSMatthew G. Knepley   /* Compute leaf-to-root process connectivity */
437b0a623aaSMatthew G. Knepley   for (l = 0; l < numLeaves; ++l) {PetscBTSet(neighbors, remotePoints[l].rank);}
438b0a623aaSMatthew G. Knepley   /* Calculate edges */
439b0a623aaSMatthew G. Knepley   PetscBTClear(neighbors, rank);
440b0a623aaSMatthew G. Knepley   for(proc = 0, numNeighbors = 0; proc < numProcs; ++proc) {if (PetscBTLookup(neighbors, proc)) ++numNeighbors;}
441b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &ranksNew);CHKERRQ(ierr);
442b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &localPointsNew);CHKERRQ(ierr);
443b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &remotePointsNew);CHKERRQ(ierr);
444b0a623aaSMatthew G. Knepley   for(proc = 0, n = 0; proc < numProcs; ++proc) {
445b0a623aaSMatthew G. Knepley     if (PetscBTLookup(neighbors, proc)) {
446b0a623aaSMatthew G. Knepley       ranksNew[n]              = proc;
447b0a623aaSMatthew G. Knepley       localPointsNew[n]        = proc;
448b0a623aaSMatthew G. Knepley       remotePointsNew[n].index = rank;
449b0a623aaSMatthew G. Knepley       remotePointsNew[n].rank  = proc;
450b0a623aaSMatthew G. Knepley       ++n;
451b0a623aaSMatthew G. Knepley     }
452b0a623aaSMatthew G. Knepley   }
453b0a623aaSMatthew G. Knepley   ierr = PetscBTDestroy(&neighbors);CHKERRQ(ierr);
454b0a623aaSMatthew G. Knepley   if (processRanks) {ierr = ISCreateGeneral(PetscObjectComm((PetscObject)dm), numNeighbors, ranksNew, PETSC_OWN_POINTER, processRanks);CHKERRQ(ierr);}
455b0a623aaSMatthew G. Knepley   else              {ierr = PetscFree(ranksNew);CHKERRQ(ierr);}
456b0a623aaSMatthew G. Knepley   if (sfProcess) {
457b0a623aaSMatthew G. Knepley     ierr = PetscSFCreate(PetscObjectComm((PetscObject)dm), sfProcess);CHKERRQ(ierr);
458b0a623aaSMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject) *sfProcess, "Two-Sided Process SF");CHKERRQ(ierr);
459b0a623aaSMatthew G. Knepley     ierr = PetscSFSetFromOptions(*sfProcess);CHKERRQ(ierr);
460b0a623aaSMatthew G. Knepley     ierr = PetscSFSetGraph(*sfProcess, numProcs, numNeighbors, localPointsNew, PETSC_OWN_POINTER, remotePointsNew, PETSC_OWN_POINTER);CHKERRQ(ierr);
461b0a623aaSMatthew G. Knepley   }
462b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
463b0a623aaSMatthew G. Knepley }
464b0a623aaSMatthew G. Knepley 
465b0a623aaSMatthew G. Knepley #undef __FUNCT__
466b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeOwnership"
467b0a623aaSMatthew G. Knepley /*@
468b0a623aaSMatthew G. Knepley   DMPlexDistributeOwnership - Compute owner information for shared points. This basically gets two-sided for an SF.
469b0a623aaSMatthew G. Knepley 
470b0a623aaSMatthew G. Knepley   Collective on DM
471b0a623aaSMatthew G. Knepley 
472b0a623aaSMatthew G. Knepley   Input Parameter:
473b0a623aaSMatthew G. Knepley . dm - The DM
474b0a623aaSMatthew G. Knepley 
475b0a623aaSMatthew G. Knepley   Output Parameters:
476b0a623aaSMatthew G. Knepley + rootSection - The number of leaves for a given root point
477b0a623aaSMatthew G. Knepley . rootrank    - The rank of each edge into the root point
478b0a623aaSMatthew G. Knepley . leafSection - The number of processes sharing a given leaf point
479b0a623aaSMatthew G. Knepley - leafrank    - The rank of each process sharing a leaf point
480b0a623aaSMatthew G. Knepley 
481b0a623aaSMatthew G. Knepley   Level: developer
482b0a623aaSMatthew G. Knepley 
483b0a623aaSMatthew G. Knepley .seealso: DMPlexCreateOverlap()
484b0a623aaSMatthew G. Knepley @*/
485b0a623aaSMatthew G. Knepley PetscErrorCode DMPlexDistributeOwnership(DM dm, PetscSection rootSection, IS *rootrank, PetscSection leafSection, IS *leafrank)
486b0a623aaSMatthew G. Knepley {
487b0a623aaSMatthew G. Knepley   MPI_Comm        comm;
488b0a623aaSMatthew G. Knepley   PetscSF         sfPoint;
489b0a623aaSMatthew G. Knepley   const PetscInt *rootdegree;
490b0a623aaSMatthew G. Knepley   PetscInt       *myrank, *remoterank;
491b0a623aaSMatthew G. Knepley   PetscInt        pStart, pEnd, p, nedges;
492b0a623aaSMatthew G. Knepley   PetscMPIInt     rank;
493b0a623aaSMatthew G. Knepley   PetscErrorCode  ierr;
494b0a623aaSMatthew G. Knepley 
495b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
496b0a623aaSMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
497b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
498b0a623aaSMatthew G. Knepley   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
499b0a623aaSMatthew G. Knepley   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
500b0a623aaSMatthew G. Knepley   /* Compute number of leaves for each root */
501b0a623aaSMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) rootSection, "Root Section");CHKERRQ(ierr);
502b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetChart(rootSection, pStart, pEnd);CHKERRQ(ierr);
503b0a623aaSMatthew G. Knepley   ierr = PetscSFComputeDegreeBegin(sfPoint, &rootdegree);CHKERRQ(ierr);
504b0a623aaSMatthew G. Knepley   ierr = PetscSFComputeDegreeEnd(sfPoint, &rootdegree);CHKERRQ(ierr);
505b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {ierr = PetscSectionSetDof(rootSection, p, rootdegree[p-pStart]);CHKERRQ(ierr);}
506b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetUp(rootSection);CHKERRQ(ierr);
507b0a623aaSMatthew G. Knepley   /* Gather rank of each leaf to root */
508b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetStorageSize(rootSection, &nedges);CHKERRQ(ierr);
509b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(pEnd-pStart, &myrank);CHKERRQ(ierr);
510b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(nedges,  &remoterank);CHKERRQ(ierr);
511b0a623aaSMatthew G. Knepley   for (p = 0; p < pEnd-pStart; ++p) myrank[p] = rank;
512b0a623aaSMatthew G. Knepley   ierr = PetscSFGatherBegin(sfPoint, MPIU_INT, myrank, remoterank);CHKERRQ(ierr);
513b0a623aaSMatthew G. Knepley   ierr = PetscSFGatherEnd(sfPoint, MPIU_INT, myrank, remoterank);CHKERRQ(ierr);
514b0a623aaSMatthew G. Knepley   ierr = PetscFree(myrank);CHKERRQ(ierr);
515b0a623aaSMatthew G. Knepley   ierr = ISCreateGeneral(comm, nedges, remoterank, PETSC_OWN_POINTER, rootrank);CHKERRQ(ierr);
516b0a623aaSMatthew G. Knepley   /* Distribute remote ranks to leaves */
517b0a623aaSMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) leafSection, "Leaf Section");CHKERRQ(ierr);
518b0a623aaSMatthew G. Knepley   ierr = DMPlexDistributeFieldIS(dm, sfPoint, rootSection, *rootrank, leafSection, leafrank);CHKERRQ(ierr);
519b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
520b0a623aaSMatthew G. Knepley }
521b0a623aaSMatthew G. Knepley 
522b0a623aaSMatthew G. Knepley #undef __FUNCT__
523b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexCreateOverlap"
524278397a0SMatthew G. Knepley /*@C
525b0a623aaSMatthew G. Knepley   DMPlexCreateOverlap - Compute owner information for shared points. This basically gets two-sided for an SF.
526b0a623aaSMatthew G. Knepley 
527b0a623aaSMatthew G. Knepley   Collective on DM
528b0a623aaSMatthew G. Knepley 
529b0a623aaSMatthew G. Knepley   Input Parameters:
530b0a623aaSMatthew G. Knepley + dm          - The DM
531b0a623aaSMatthew G. Knepley . rootSection - The number of leaves for a given root point
532b0a623aaSMatthew G. Knepley . rootrank    - The rank of each edge into the root point
533b0a623aaSMatthew G. Knepley . leafSection - The number of processes sharing a given leaf point
534b0a623aaSMatthew G. Knepley - leafrank    - The rank of each process sharing a leaf point
535b0a623aaSMatthew G. Knepley 
536b0a623aaSMatthew G. Knepley   Output Parameters:
537b0a623aaSMatthew G. Knepley + ovRootSection - The number of new overlap points for each neighboring process
538b0a623aaSMatthew G. Knepley . ovRootPoints  - The new overlap points for each neighboring process
539b0a623aaSMatthew G. Knepley . ovLeafSection - The number of new overlap points from each neighboring process
540b0a623aaSMatthew G. Knepley - ovLeafPoints  - The new overlap points from each neighboring process
541b0a623aaSMatthew G. Knepley 
542b0a623aaSMatthew G. Knepley   Level: developer
543b0a623aaSMatthew G. Knepley 
544b0a623aaSMatthew G. Knepley .seealso: DMPlexDistributeOwnership()
545b0a623aaSMatthew G. Knepley @*/
546b0a623aaSMatthew G. Knepley PetscErrorCode DMPlexCreateOverlap(DM dm, PetscSection rootSection, IS rootrank, PetscSection leafSection, IS leafrank, PetscSection ovRootSection, PetscSFNode **ovRootPoints, PetscSection ovLeafSection, PetscSFNode **ovLeafPoints)
547b0a623aaSMatthew G. Knepley {
548b0a623aaSMatthew G. Knepley   DMLabel            ovAdjByRank; /* A DMLabel containing all points adjacent to shared points, separated by rank (value in label) */
549b0a623aaSMatthew G. Knepley   PetscSF            sfPoint, sfProc;
550b0a623aaSMatthew G. Knepley   IS                 valueIS;
551b0a623aaSMatthew G. Knepley   const PetscSFNode *remote;
552b0a623aaSMatthew G. Knepley   const PetscInt    *local;
553b0a623aaSMatthew G. Knepley   const PetscInt    *nrank, *rrank, *neighbors;
554b0a623aaSMatthew G. Knepley   PetscInt          *adj = NULL;
555b0a623aaSMatthew G. Knepley   PetscInt           pStart, pEnd, p, sStart, sEnd, nleaves, l, numNeighbors, n, ovSize;
556b0a623aaSMatthew G. Knepley   PetscMPIInt        rank, numProcs;
557b0a623aaSMatthew G. Knepley   PetscErrorCode     ierr;
558b0a623aaSMatthew G. Knepley 
559b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
560b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm), &numProcs);CHKERRQ(ierr);
561b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
562b0a623aaSMatthew G. Knepley   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
563b0a623aaSMatthew G. Knepley   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
564b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(leafSection, &sStart, &sEnd);CHKERRQ(ierr);
565b0a623aaSMatthew G. Knepley   ierr = PetscSFGetGraph(sfPoint, NULL, &nleaves, &local, &remote);CHKERRQ(ierr);
566b0a623aaSMatthew G. Knepley   ierr = DMLabelCreate("Overlap adjacency", &ovAdjByRank);CHKERRQ(ierr);
567b0a623aaSMatthew G. Knepley   /* Handle leaves: shared with the root point */
568b0a623aaSMatthew G. Knepley   for (l = 0; l < nleaves; ++l) {
569b0a623aaSMatthew G. Knepley     PetscInt adjSize = PETSC_DETERMINE, a;
570b0a623aaSMatthew G. Knepley 
571b0a623aaSMatthew G. Knepley     ierr = DMPlexGetAdjacency(dm, local[l], &adjSize, &adj);CHKERRQ(ierr);
572b0a623aaSMatthew G. Knepley     for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remote[l].rank);CHKERRQ(ierr);}
573b0a623aaSMatthew G. Knepley   }
574b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(rootrank, &rrank);CHKERRQ(ierr);
575b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(leafrank, &nrank);CHKERRQ(ierr);
576b0a623aaSMatthew G. Knepley   /* Handle roots */
577b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
578b0a623aaSMatthew G. Knepley     PetscInt adjSize = PETSC_DETERMINE, neighbors = 0, noff, n, a;
579b0a623aaSMatthew G. Knepley 
580b0a623aaSMatthew G. Knepley     if ((p >= sStart) && (p < sEnd)) {
581b0a623aaSMatthew G. Knepley       /* Some leaves share a root with other leaves on different processes */
582b0a623aaSMatthew G. Knepley       ierr = PetscSectionGetDof(leafSection, p, &neighbors);CHKERRQ(ierr);
583b0a623aaSMatthew G. Knepley       if (neighbors) {
584b0a623aaSMatthew G. Knepley         ierr = PetscSectionGetOffset(leafSection, p, &noff);CHKERRQ(ierr);
585b0a623aaSMatthew G. Knepley         ierr = DMPlexGetAdjacency(dm, p, &adjSize, &adj);CHKERRQ(ierr);
586b0a623aaSMatthew G. Knepley         for (n = 0; n < neighbors; ++n) {
587b0a623aaSMatthew G. Knepley           const PetscInt remoteRank = nrank[noff+n];
588b0a623aaSMatthew G. Knepley 
589b0a623aaSMatthew G. Knepley           if (remoteRank == rank) continue;
590b0a623aaSMatthew G. Knepley           for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remoteRank);CHKERRQ(ierr);}
591b0a623aaSMatthew G. Knepley         }
592b0a623aaSMatthew G. Knepley       }
593b0a623aaSMatthew G. Knepley     }
594b0a623aaSMatthew G. Knepley     /* Roots are shared with leaves */
595b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(rootSection, p, &neighbors);CHKERRQ(ierr);
596b0a623aaSMatthew G. Knepley     if (!neighbors) continue;
597b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(rootSection, p, &noff);CHKERRQ(ierr);
598b0a623aaSMatthew G. Knepley     ierr = DMPlexGetAdjacency(dm, p, &adjSize, &adj);CHKERRQ(ierr);
599b0a623aaSMatthew G. Knepley     for (n = 0; n < neighbors; ++n) {
600b0a623aaSMatthew G. Knepley       const PetscInt remoteRank = rrank[noff+n];
601b0a623aaSMatthew G. Knepley 
602b0a623aaSMatthew G. Knepley       if (remoteRank == rank) continue;
603b0a623aaSMatthew G. Knepley       for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remoteRank);CHKERRQ(ierr);}
604b0a623aaSMatthew G. Knepley     }
605b0a623aaSMatthew G. Knepley   }
606b0a623aaSMatthew G. Knepley   ierr = PetscFree(adj);CHKERRQ(ierr);
607b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(rootrank, &rrank);CHKERRQ(ierr);
608b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(leafrank, &nrank);CHKERRQ(ierr);
609b0a623aaSMatthew G. Knepley   {
610b0a623aaSMatthew G. Knepley     ierr = PetscViewerASCIISynchronizedAllow(PETSC_VIEWER_STDOUT_WORLD, PETSC_TRUE);CHKERRQ(ierr);
611b0a623aaSMatthew G. Knepley     ierr = DMLabelView(ovAdjByRank, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
612b0a623aaSMatthew G. Knepley   }
613b0a623aaSMatthew G. Knepley   /* Convert to (point, rank) and use actual owners */
614b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetChart(ovRootSection, 0, numProcs);CHKERRQ(ierr);
615b0a623aaSMatthew G. Knepley   ierr = DMLabelGetValueIS(ovAdjByRank, &valueIS);CHKERRQ(ierr);
616b0a623aaSMatthew G. Knepley   ierr = ISGetLocalSize(valueIS, &numNeighbors);CHKERRQ(ierr);
617b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(valueIS, &neighbors);CHKERRQ(ierr);
618b0a623aaSMatthew G. Knepley   for (n = 0; n < numNeighbors; ++n) {
619b0a623aaSMatthew G. Knepley     PetscInt numPoints;
620b0a623aaSMatthew G. Knepley 
621b0a623aaSMatthew G. Knepley     ierr = DMLabelGetStratumSize(ovAdjByRank, neighbors[n], &numPoints);CHKERRQ(ierr);
622b0a623aaSMatthew G. Knepley     ierr = PetscSectionAddDof(ovRootSection, neighbors[n], numPoints);CHKERRQ(ierr);
623b0a623aaSMatthew G. Knepley   }
624b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetUp(ovRootSection);CHKERRQ(ierr);
625b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetStorageSize(ovRootSection, &ovSize);CHKERRQ(ierr);
626b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(ovSize, ovRootPoints);CHKERRQ(ierr);
627b0a623aaSMatthew G. Knepley   ierr = PetscSFGetGraph(sfPoint, NULL, &nleaves, &local, &remote);CHKERRQ(ierr);
628b0a623aaSMatthew G. Knepley   for (n = 0; n < numNeighbors; ++n) {
629b0a623aaSMatthew G. Knepley     IS              pointIS;
630b0a623aaSMatthew G. Knepley     const PetscInt *points;
631b0a623aaSMatthew G. Knepley     PetscInt        off, numPoints, p;
632b0a623aaSMatthew G. Knepley 
633b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(ovRootSection, neighbors[n], &off);CHKERRQ(ierr);
634b0a623aaSMatthew G. Knepley     ierr = DMLabelGetStratumIS(ovAdjByRank, neighbors[n], &pointIS);CHKERRQ(ierr);
635b0a623aaSMatthew G. Knepley     ierr = ISGetLocalSize(pointIS, &numPoints);CHKERRQ(ierr);
636b0a623aaSMatthew G. Knepley     ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr);
637b0a623aaSMatthew G. Knepley     for (p = 0; p < numPoints; ++p) {
638b0a623aaSMatthew G. Knepley       ierr = PetscFindInt(points[p], nleaves, local, &l);CHKERRQ(ierr);
639b0a623aaSMatthew G. Knepley       if (l >= 0) {(*ovRootPoints)[off+p] = remote[l];}
640b0a623aaSMatthew G. Knepley       else        {(*ovRootPoints)[off+p].index = points[p]; (*ovRootPoints)[off+p].rank = rank;}
641b0a623aaSMatthew G. Knepley     }
642b0a623aaSMatthew G. Knepley     ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr);
643b0a623aaSMatthew G. Knepley     ierr = ISDestroy(&pointIS);CHKERRQ(ierr);
644b0a623aaSMatthew G. Knepley   }
645b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(valueIS, &neighbors);CHKERRQ(ierr);
646b0a623aaSMatthew G. Knepley   ierr = ISDestroy(&valueIS);CHKERRQ(ierr);
647b0a623aaSMatthew G. Knepley   ierr = DMLabelDestroy(&ovAdjByRank);CHKERRQ(ierr);
648b0a623aaSMatthew G. Knepley   /* Make process SF */
649b0a623aaSMatthew G. Knepley   ierr = DMPlexCreateTwoSidedProcessSF(dm, sfPoint, rootSection, rootrank, leafSection, leafrank, NULL, &sfProc);CHKERRQ(ierr);
650b0a623aaSMatthew G. Knepley   {
651b0a623aaSMatthew G. Knepley     ierr = PetscSFView(sfProc, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
652b0a623aaSMatthew G. Knepley   }
653b0a623aaSMatthew G. Knepley   /* Communicate overlap */
654b0a623aaSMatthew G. Knepley   ierr = DMPlexDistributeData(dm, sfProc, ovRootSection, MPIU_2INT, (void *) *ovRootPoints, ovLeafSection, (void **) ovLeafPoints);CHKERRQ(ierr);
655b0a623aaSMatthew G. Knepley   ierr = PetscSFDestroy(&sfProc);CHKERRQ(ierr);
656b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
657b0a623aaSMatthew G. Knepley }
65870034214SMatthew G. Knepley 
65970034214SMatthew G. Knepley #undef __FUNCT__
66070034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeField"
66170034214SMatthew G. Knepley /*@
66270034214SMatthew G. Knepley   DMPlexDistributeField - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
66370034214SMatthew G. Knepley 
66470034214SMatthew G. Knepley   Collective on DM
66570034214SMatthew G. Knepley 
66670034214SMatthew G. Knepley   Input Parameters:
66770034214SMatthew G. Knepley + dm - The DMPlex object
66870034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
66970034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout
67070034214SMatthew G. Knepley - originalVec - The existing data
67170034214SMatthew G. Knepley 
67270034214SMatthew G. Knepley   Output Parameters:
67370034214SMatthew G. Knepley + newSection - The PetscSF describing the new data layout
67470034214SMatthew G. Knepley - newVec - The new data
67570034214SMatthew G. Knepley 
67670034214SMatthew G. Knepley   Level: developer
67770034214SMatthew G. Knepley 
6781e8fc25dSMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeFieldIS(), DMPlexDistributeData()
67970034214SMatthew G. Knepley @*/
68070034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeField(DM dm, PetscSF pointSF, PetscSection originalSection, Vec originalVec, PetscSection newSection, Vec newVec)
68170034214SMatthew G. Knepley {
68270034214SMatthew G. Knepley   PetscSF        fieldSF;
68370034214SMatthew G. Knepley   PetscInt      *remoteOffsets, fieldSize;
68470034214SMatthew G. Knepley   PetscScalar   *originalValues, *newValues;
68570034214SMatthew G. Knepley   PetscErrorCode ierr;
68670034214SMatthew G. Knepley 
68770034214SMatthew G. Knepley   PetscFunctionBegin;
68870034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
68970034214SMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
69070034214SMatthew G. Knepley 
69170034214SMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
69270034214SMatthew G. Knepley   ierr = VecSetSizes(newVec, fieldSize, PETSC_DETERMINE);CHKERRQ(ierr);
69370034214SMatthew G. Knepley   ierr = VecSetType(newVec,dm->vectype);CHKERRQ(ierr);
69470034214SMatthew G. Knepley 
69570034214SMatthew G. Knepley   ierr = VecGetArray(originalVec, &originalValues);CHKERRQ(ierr);
69670034214SMatthew G. Knepley   ierr = VecGetArray(newVec, &newValues);CHKERRQ(ierr);
69770034214SMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
69870034214SMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr);
69970034214SMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr);
70070034214SMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
70170034214SMatthew G. Knepley   ierr = VecRestoreArray(newVec, &newValues);CHKERRQ(ierr);
70270034214SMatthew G. Knepley   ierr = VecRestoreArray(originalVec, &originalValues);CHKERRQ(ierr);
70370034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
70470034214SMatthew G. Knepley   PetscFunctionReturn(0);
70570034214SMatthew G. Knepley }
70670034214SMatthew G. Knepley 
70770034214SMatthew G. Knepley #undef __FUNCT__
7081e8fc25dSMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeFieldIS"
7091e8fc25dSMatthew G. Knepley /*@
7101e8fc25dSMatthew G. Knepley   DMPlexDistributeFieldIS - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
7111e8fc25dSMatthew G. Knepley 
7121e8fc25dSMatthew G. Knepley   Collective on DM
7131e8fc25dSMatthew G. Knepley 
7141e8fc25dSMatthew G. Knepley   Input Parameters:
7151e8fc25dSMatthew G. Knepley + dm - The DMPlex object
7161e8fc25dSMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
7171e8fc25dSMatthew G. Knepley . originalSection - The PetscSection for existing data layout
7181e8fc25dSMatthew G. Knepley - originalIS - The existing data
7191e8fc25dSMatthew G. Knepley 
7201e8fc25dSMatthew G. Knepley   Output Parameters:
7211e8fc25dSMatthew G. Knepley + newSection - The PetscSF describing the new data layout
7221e8fc25dSMatthew G. Knepley - newIS - The new data
7231e8fc25dSMatthew G. Knepley 
7241e8fc25dSMatthew G. Knepley   Level: developer
7251e8fc25dSMatthew G. Knepley 
7261e8fc25dSMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField(), DMPlexDistributeData()
7271e8fc25dSMatthew G. Knepley @*/
7281e8fc25dSMatthew G. Knepley PetscErrorCode DMPlexDistributeFieldIS(DM dm, PetscSF pointSF, PetscSection originalSection, IS originalIS, PetscSection newSection, IS *newIS)
7291e8fc25dSMatthew G. Knepley {
7301e8fc25dSMatthew G. Knepley   PetscSF         fieldSF;
7311e8fc25dSMatthew G. Knepley   PetscInt       *newValues, *remoteOffsets, fieldSize;
7321e8fc25dSMatthew G. Knepley   const PetscInt *originalValues;
7331e8fc25dSMatthew G. Knepley   PetscErrorCode  ierr;
7341e8fc25dSMatthew G. Knepley 
7351e8fc25dSMatthew G. Knepley   PetscFunctionBegin;
7361e8fc25dSMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
7371e8fc25dSMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
7381e8fc25dSMatthew G. Knepley 
7391e8fc25dSMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
7401e8fc25dSMatthew G. Knepley   ierr = PetscMalloc(fieldSize * sizeof(PetscInt), &newValues);CHKERRQ(ierr);
7411e8fc25dSMatthew G. Knepley 
7421e8fc25dSMatthew G. Knepley   ierr = ISGetIndices(originalIS, &originalValues);CHKERRQ(ierr);
7431e8fc25dSMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
7441e8fc25dSMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, MPIU_INT, originalValues, newValues);CHKERRQ(ierr);
7451e8fc25dSMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, MPIU_INT, originalValues, newValues);CHKERRQ(ierr);
7461e8fc25dSMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
7471e8fc25dSMatthew G. Knepley   ierr = ISRestoreIndices(originalIS, &originalValues);CHKERRQ(ierr);
7481e8fc25dSMatthew G. Knepley   ierr = ISCreateGeneral(PetscObjectComm((PetscObject) pointSF), fieldSize, newValues, PETSC_OWN_POINTER, newIS);CHKERRQ(ierr);
7491e8fc25dSMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
7501e8fc25dSMatthew G. Knepley   PetscFunctionReturn(0);
7511e8fc25dSMatthew G. Knepley }
7521e8fc25dSMatthew G. Knepley 
7531e8fc25dSMatthew G. Knepley #undef __FUNCT__
75470034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeData"
75570034214SMatthew G. Knepley /*@
75670034214SMatthew G. Knepley   DMPlexDistributeData - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
75770034214SMatthew G. Knepley 
75870034214SMatthew G. Knepley   Collective on DM
75970034214SMatthew G. Knepley 
76070034214SMatthew G. Knepley   Input Parameters:
76170034214SMatthew G. Knepley + dm - The DMPlex object
76270034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
76370034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout
76470034214SMatthew G. Knepley . datatype - The type of data
76570034214SMatthew G. Knepley - originalData - The existing data
76670034214SMatthew G. Knepley 
76770034214SMatthew G. Knepley   Output Parameters:
7681e8fc25dSMatthew G. Knepley + newSection - The PetscSection describing the new data layout
76970034214SMatthew G. Knepley - newData - The new data
77070034214SMatthew G. Knepley 
77170034214SMatthew G. Knepley   Level: developer
77270034214SMatthew G. Knepley 
77370034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField()
77470034214SMatthew G. Knepley @*/
77570034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeData(DM dm, PetscSF pointSF, PetscSection originalSection, MPI_Datatype datatype, void *originalData, PetscSection newSection, void **newData)
77670034214SMatthew G. Knepley {
77770034214SMatthew G. Knepley   PetscSF        fieldSF;
77870034214SMatthew G. Knepley   PetscInt      *remoteOffsets, fieldSize;
77970034214SMatthew G. Knepley   PetscMPIInt    dataSize;
78070034214SMatthew G. Knepley   PetscErrorCode ierr;
78170034214SMatthew G. Knepley 
78270034214SMatthew G. Knepley   PetscFunctionBegin;
78370034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr);
78470034214SMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
78570034214SMatthew G. Knepley 
78670034214SMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
78770034214SMatthew G. Knepley   ierr = MPI_Type_size(datatype, &dataSize);CHKERRQ(ierr);
78870034214SMatthew G. Knepley   ierr = PetscMalloc(fieldSize * dataSize, newData);CHKERRQ(ierr);
78970034214SMatthew G. Knepley 
79070034214SMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
79170034214SMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr);
79270034214SMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr);
79370034214SMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
79470034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr);
79570034214SMatthew G. Knepley   PetscFunctionReturn(0);
79670034214SMatthew G. Knepley }
79770034214SMatthew G. Knepley 
79870034214SMatthew G. Knepley #undef __FUNCT__
799*cc71bff1SMichael Lange #define __FUNCT__ "DMPlexDistributeCones"
800*cc71bff1SMichael Lange PetscErrorCode DMPlexDistributeCones(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping renumbering, DM dmParallel)
801*cc71bff1SMichael Lange {
802*cc71bff1SMichael Lange   DM_Plex               *pmesh   = (DM_Plex*) (dmParallel)->data;
803*cc71bff1SMichael Lange   MPI_Comm               comm;
804*cc71bff1SMichael Lange   PetscSF                coneSF;
805*cc71bff1SMichael Lange   PetscSection           originalConeSection, newConeSection;
806*cc71bff1SMichael Lange   PetscInt              *remoteOffsets, *cones, *newCones, newConesSize;
807*cc71bff1SMichael Lange   PetscBool              flg;
808*cc71bff1SMichael Lange   PetscErrorCode         ierr;
809*cc71bff1SMichael Lange 
810*cc71bff1SMichael Lange   PetscFunctionBegin;
811*cc71bff1SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
812*cc71bff1SMichael Lange   PetscValidPointer(dmParallel,4);
813*cc71bff1SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr);
814*cc71bff1SMichael Lange 
815*cc71bff1SMichael Lange   /* Distribute cone section */
816*cc71bff1SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
817*cc71bff1SMichael Lange   ierr = DMPlexGetConeSection(dm, &originalConeSection);CHKERRQ(ierr);
818*cc71bff1SMichael Lange   ierr = DMPlexGetConeSection(dmParallel, &newConeSection);CHKERRQ(ierr);
819*cc71bff1SMichael Lange   ierr = PetscSFDistributeSection(migrationSF, originalConeSection, &remoteOffsets, newConeSection);CHKERRQ(ierr);
820*cc71bff1SMichael Lange   ierr = DMSetUp(dmParallel);CHKERRQ(ierr);
821*cc71bff1SMichael Lange   {
822*cc71bff1SMichael Lange     PetscInt pStart, pEnd, p;
823*cc71bff1SMichael Lange 
824*cc71bff1SMichael Lange     ierr = PetscSectionGetChart(newConeSection, &pStart, &pEnd);CHKERRQ(ierr);
825*cc71bff1SMichael Lange     for (p = pStart; p < pEnd; ++p) {
826*cc71bff1SMichael Lange       PetscInt coneSize;
827*cc71bff1SMichael Lange       ierr               = PetscSectionGetDof(newConeSection, p, &coneSize);CHKERRQ(ierr);
828*cc71bff1SMichael Lange       pmesh->maxConeSize = PetscMax(pmesh->maxConeSize, coneSize);
829*cc71bff1SMichael Lange     }
830*cc71bff1SMichael Lange   }
831*cc71bff1SMichael Lange   /* Communicate and renumber cones */
832*cc71bff1SMichael Lange   ierr = PetscSFCreateSectionSF(migrationSF, originalConeSection, remoteOffsets, newConeSection, &coneSF);CHKERRQ(ierr);
833*cc71bff1SMichael Lange   ierr = DMPlexGetCones(dm, &cones);CHKERRQ(ierr);
834*cc71bff1SMichael Lange   ierr = DMPlexGetCones(dmParallel, &newCones);CHKERRQ(ierr);
835*cc71bff1SMichael Lange   ierr = PetscSFBcastBegin(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
836*cc71bff1SMichael Lange   ierr = PetscSFBcastEnd(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
837*cc71bff1SMichael Lange   ierr = PetscSectionGetStorageSize(newConeSection, &newConesSize);CHKERRQ(ierr);
838*cc71bff1SMichael Lange   ierr = ISGlobalToLocalMappingApplyBlock(renumbering, IS_GTOLM_MASK, newConesSize, newCones, NULL, newCones);CHKERRQ(ierr);
839*cc71bff1SMichael Lange   ierr = PetscOptionsHasName(((PetscObject) dm)->prefix, "-cones_view", &flg);CHKERRQ(ierr);
840*cc71bff1SMichael Lange   if (flg) {
841*cc71bff1SMichael Lange     ierr = PetscPrintf(comm, "Serial Cone Section:\n");CHKERRQ(ierr);
842*cc71bff1SMichael Lange     ierr = PetscSectionView(originalConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
843*cc71bff1SMichael Lange     ierr = PetscPrintf(comm, "Parallel Cone Section:\n");CHKERRQ(ierr);
844*cc71bff1SMichael Lange     ierr = PetscSectionView(newConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
845*cc71bff1SMichael Lange     ierr = PetscSFView(coneSF, NULL);CHKERRQ(ierr);
846*cc71bff1SMichael Lange   }
847*cc71bff1SMichael Lange   ierr = DMPlexGetConeOrientations(dm, &cones);CHKERRQ(ierr);
848*cc71bff1SMichael Lange   ierr = DMPlexGetConeOrientations(dmParallel, &newCones);CHKERRQ(ierr);
849*cc71bff1SMichael Lange   ierr = PetscSFBcastBegin(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
850*cc71bff1SMichael Lange   ierr = PetscSFBcastEnd(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
851*cc71bff1SMichael Lange   ierr = PetscSFDestroy(&coneSF);CHKERRQ(ierr);
852*cc71bff1SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr);
853*cc71bff1SMichael Lange   /* Create supports and stratify sieve */
854*cc71bff1SMichael Lange   {
855*cc71bff1SMichael Lange     PetscInt pStart, pEnd;
856*cc71bff1SMichael Lange 
857*cc71bff1SMichael Lange     ierr = PetscSectionGetChart(pmesh->coneSection, &pStart, &pEnd);CHKERRQ(ierr);
858*cc71bff1SMichael Lange     ierr = PetscSectionSetChart(pmesh->supportSection, pStart, pEnd);CHKERRQ(ierr);
859*cc71bff1SMichael Lange   }
860*cc71bff1SMichael Lange   ierr = DMPlexSymmetrize(dmParallel);CHKERRQ(ierr);
861*cc71bff1SMichael Lange   ierr = DMPlexStratify(dmParallel);CHKERRQ(ierr);
862*cc71bff1SMichael Lange   PetscFunctionReturn(0);
863*cc71bff1SMichael Lange }
864*cc71bff1SMichael Lange 
865*cc71bff1SMichael Lange #undef __FUNCT__
86670034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistribute"
86770034214SMatthew G. Knepley /*@C
86870034214SMatthew G. Knepley   DMPlexDistribute - Distributes the mesh and any associated sections.
86970034214SMatthew G. Knepley 
87070034214SMatthew G. Knepley   Not Collective
87170034214SMatthew G. Knepley 
87270034214SMatthew G. Knepley   Input Parameter:
87370034214SMatthew G. Knepley + dm  - The original DMPlex object
87470034214SMatthew G. Knepley . partitioner - The partitioning package, or NULL for the default
87570034214SMatthew G. Knepley - overlap - The overlap of partitions, 0 is the default
87670034214SMatthew G. Knepley 
87770034214SMatthew G. Knepley   Output Parameter:
87870034214SMatthew G. Knepley + sf - The PetscSF used for point distribution
87970034214SMatthew G. Knepley - parallelMesh - The distributed DMPlex object, or NULL
88070034214SMatthew G. Knepley 
881a9c22940SMatthew G. Knepley   Note: If the mesh was not distributed, the return value is NULL.
882a9c22940SMatthew G. Knepley 
883a9c22940SMatthew G. Knepley   The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and
884a9c22940SMatthew G. Knepley   DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function
885a9c22940SMatthew G. Knepley   representation on the mesh.
88670034214SMatthew G. Knepley 
88770034214SMatthew G. Knepley   Level: intermediate
88870034214SMatthew G. Knepley 
88970034214SMatthew G. Knepley .keywords: mesh, elements
890a9c22940SMatthew G. Knepley .seealso: DMPlexCreate(), DMPlexDistributeByFace(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure()
89170034214SMatthew G. Knepley @*/
89270034214SMatthew G. Knepley PetscErrorCode DMPlexDistribute(DM dm, const char partitioner[], PetscInt overlap, PetscSF *sf, DM *dmParallel)
89370034214SMatthew G. Knepley {
89470034214SMatthew G. Knepley   DM_Plex               *mesh   = (DM_Plex*) dm->data, *pmesh;
89570034214SMatthew G. Knepley   MPI_Comm               comm;
89670034214SMatthew G. Knepley   PetscInt               dim, numRemoteRanks;
89770034214SMatthew G. Knepley   IS                     origCellPart,        origPart,        cellPart,        part;
89870034214SMatthew G. Knepley   PetscSection           origCellPartSection, origPartSection, cellPartSection, partSection;
89970034214SMatthew G. Knepley   PetscSFNode           *remoteRanks;
900*cc71bff1SMichael Lange   PetscSF                partSF, pointSF;
90170034214SMatthew G. Knepley   ISLocalToGlobalMapping renumbering;
90270034214SMatthew G. Knepley   PetscBool              flg;
90370034214SMatthew G. Knepley   PetscMPIInt            rank, numProcs, p;
90470034214SMatthew G. Knepley   PetscErrorCode         ierr;
90570034214SMatthew G. Knepley 
90670034214SMatthew G. Knepley   PetscFunctionBegin;
90770034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
90870034214SMatthew G. Knepley   if (sf) PetscValidPointer(sf,4);
90970034214SMatthew G. Knepley   PetscValidPointer(dmParallel,5);
91070034214SMatthew G. Knepley 
91170034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr);
91270034214SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
91370034214SMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
91470034214SMatthew G. Knepley   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
91570034214SMatthew G. Knepley 
91670034214SMatthew G. Knepley   *dmParallel = NULL;
91770034214SMatthew G. Knepley   if (numProcs == 1) PetscFunctionReturn(0);
91870034214SMatthew G. Knepley 
919c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
92070034214SMatthew G. Knepley   /* Create cell partition - We need to rewrite to use IS, use the MatPartition stuff */
92177623264SMatthew G. Knepley   ierr = PetscLogEventBegin(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
92277623264SMatthew G. Knepley   if (overlap > 1) SETERRQ(comm, PETSC_ERR_SUP, "Overlap > 1 not yet implemented");
92377623264SMatthew G. Knepley   ierr = PetscSectionCreate(comm, &cellPartSection);CHKERRQ(ierr);
92477623264SMatthew G. Knepley   ierr = PetscSectionCreate(comm, &origCellPartSection);CHKERRQ(ierr);
92577623264SMatthew G. Knepley   ierr = PetscPartitionerPartition(mesh->partitioner, dm, overlap > 0 ? PETSC_TRUE : PETSC_FALSE, cellPartSection, &cellPart, origCellPartSection, &origCellPart);CHKERRQ(ierr);
92670034214SMatthew G. Knepley   /* Create SF assuming a serial partition for all processes: Could check for IS length here */
92770034214SMatthew G. Knepley   if (!rank) numRemoteRanks = numProcs;
92870034214SMatthew G. Knepley   else       numRemoteRanks = 0;
92970034214SMatthew G. Knepley   ierr = PetscMalloc1(numRemoteRanks, &remoteRanks);CHKERRQ(ierr);
93070034214SMatthew G. Knepley   for (p = 0; p < numRemoteRanks; ++p) {
93170034214SMatthew G. Knepley     remoteRanks[p].rank  = p;
93270034214SMatthew G. Knepley     remoteRanks[p].index = 0;
93370034214SMatthew G. Knepley   }
93470034214SMatthew G. Knepley   ierr = PetscSFCreate(comm, &partSF);CHKERRQ(ierr);
93570034214SMatthew G. Knepley   ierr = PetscSFSetGraph(partSF, 1, numRemoteRanks, NULL, PETSC_OWN_POINTER, remoteRanks, PETSC_OWN_POINTER);CHKERRQ(ierr);
93670034214SMatthew G. Knepley   ierr = PetscOptionsHasName(((PetscObject) dm)->prefix, "-partition_view", &flg);CHKERRQ(ierr);
93770034214SMatthew G. Knepley   if (flg) {
93870034214SMatthew G. Knepley     ierr = PetscPrintf(comm, "Cell Partition:\n");CHKERRQ(ierr);
93970034214SMatthew G. Knepley     ierr = PetscSectionView(cellPartSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
94070034214SMatthew G. Knepley     ierr = ISView(cellPart, NULL);CHKERRQ(ierr);
94170034214SMatthew G. Knepley     if (origCellPart) {
94270034214SMatthew G. Knepley       ierr = PetscPrintf(comm, "Original Cell Partition:\n");CHKERRQ(ierr);
94370034214SMatthew G. Knepley       ierr = PetscSectionView(origCellPartSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
94470034214SMatthew G. Knepley       ierr = ISView(origCellPart, NULL);CHKERRQ(ierr);
94570034214SMatthew G. Knepley     }
94670034214SMatthew G. Knepley     ierr = PetscSFView(partSF, NULL);CHKERRQ(ierr);
94770034214SMatthew G. Knepley   }
94870034214SMatthew G. Knepley   /* Close the partition over the mesh */
94970034214SMatthew G. Knepley   ierr = DMPlexCreatePartitionClosure(dm, cellPartSection, cellPart, &partSection, &part);CHKERRQ(ierr);
95070034214SMatthew G. Knepley   ierr = ISDestroy(&cellPart);CHKERRQ(ierr);
95170034214SMatthew G. Knepley   ierr = PetscSectionDestroy(&cellPartSection);CHKERRQ(ierr);
95270034214SMatthew G. Knepley   /* Create new mesh */
95370034214SMatthew G. Knepley   ierr  = DMPlexCreate(comm, dmParallel);CHKERRQ(ierr);
954c73cfb54SMatthew G. Knepley   ierr  = DMSetDimension(*dmParallel, dim);CHKERRQ(ierr);
95570034214SMatthew G. Knepley   ierr  = PetscObjectSetName((PetscObject) *dmParallel, "Parallel Mesh");CHKERRQ(ierr);
95670034214SMatthew G. Knepley   pmesh = (DM_Plex*) (*dmParallel)->data;
95770034214SMatthew G. Knepley   /* Distribute sieve points and the global point numbering (replaces creating remote bases) */
95870034214SMatthew G. Knepley   ierr = PetscSFConvertPartition(partSF, partSection, part, &renumbering, &pointSF);CHKERRQ(ierr);
95970034214SMatthew G. Knepley   if (flg) {
96070034214SMatthew G. Knepley     ierr = PetscPrintf(comm, "Point Partition:\n");CHKERRQ(ierr);
96170034214SMatthew G. Knepley     ierr = PetscSectionView(partSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
96270034214SMatthew G. Knepley     ierr = ISView(part, NULL);CHKERRQ(ierr);
96370034214SMatthew G. Knepley     ierr = PetscSFView(pointSF, NULL);CHKERRQ(ierr);
96470034214SMatthew G. Knepley     ierr = PetscPrintf(comm, "Point Renumbering after partition:\n");CHKERRQ(ierr);
96570034214SMatthew G. Knepley     ierr = ISLocalToGlobalMappingView(renumbering, NULL);CHKERRQ(ierr);
96670034214SMatthew G. Knepley   }
96777623264SMatthew G. Knepley   ierr = PetscLogEventEnd(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
96870034214SMatthew G. Knepley 
969*cc71bff1SMichael Lange   ierr = DMPlexDistributeCones(dm, pointSF, renumbering, *dmParallel);CHKERRQ(ierr);
97070034214SMatthew G. Knepley 
97170034214SMatthew G. Knepley   /* Create point SF for parallel mesh */
97270034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr);
97370034214SMatthew G. Knepley   {
97470034214SMatthew G. Knepley     const PetscInt *leaves;
97570034214SMatthew G. Knepley     PetscSFNode    *remotePoints, *rowners, *lowners;
97670034214SMatthew G. Knepley     PetscInt        numRoots, numLeaves, numGhostPoints = 0, p, gp, *ghostPoints;
97770034214SMatthew G. Knepley     PetscInt        pStart, pEnd;
97870034214SMatthew G. Knepley 
97970034214SMatthew G. Knepley     ierr = DMPlexGetChart(*dmParallel, &pStart, &pEnd);CHKERRQ(ierr);
98070034214SMatthew G. Knepley     ierr = PetscSFGetGraph(pointSF, &numRoots, &numLeaves, &leaves, NULL);CHKERRQ(ierr);
98170034214SMatthew G. Knepley     ierr = PetscMalloc2(numRoots,&rowners,numLeaves,&lowners);CHKERRQ(ierr);
98270034214SMatthew G. Knepley     for (p=0; p<numRoots; p++) {
98370034214SMatthew G. Knepley       rowners[p].rank  = -1;
98470034214SMatthew G. Knepley       rowners[p].index = -1;
98570034214SMatthew G. Knepley     }
98670034214SMatthew G. Knepley     if (origCellPart) {
98770034214SMatthew G. Knepley       /* Make sure points in the original partition are not assigned to other procs */
98870034214SMatthew G. Knepley       const PetscInt *origPoints;
98970034214SMatthew G. Knepley 
99070034214SMatthew G. Knepley       ierr = DMPlexCreatePartitionClosure(dm, origCellPartSection, origCellPart, &origPartSection, &origPart);CHKERRQ(ierr);
99170034214SMatthew G. Knepley       ierr = ISGetIndices(origPart, &origPoints);CHKERRQ(ierr);
99270034214SMatthew G. Knepley       for (p = 0; p < numProcs; ++p) {
99370034214SMatthew G. Knepley         PetscInt dof, off, d;
99470034214SMatthew G. Knepley 
99570034214SMatthew G. Knepley         ierr = PetscSectionGetDof(origPartSection, p, &dof);CHKERRQ(ierr);
99670034214SMatthew G. Knepley         ierr = PetscSectionGetOffset(origPartSection, p, &off);CHKERRQ(ierr);
99770034214SMatthew G. Knepley         for (d = off; d < off+dof; ++d) {
99870034214SMatthew G. Knepley           rowners[origPoints[d]].rank = p;
99970034214SMatthew G. Knepley         }
100070034214SMatthew G. Knepley       }
100170034214SMatthew G. Knepley       ierr = ISRestoreIndices(origPart, &origPoints);CHKERRQ(ierr);
100270034214SMatthew G. Knepley       ierr = ISDestroy(&origPart);CHKERRQ(ierr);
100370034214SMatthew G. Knepley       ierr = PetscSectionDestroy(&origPartSection);CHKERRQ(ierr);
100470034214SMatthew G. Knepley     }
100570034214SMatthew G. Knepley     ierr = ISDestroy(&origCellPart);CHKERRQ(ierr);
100670034214SMatthew G. Knepley     ierr = PetscSectionDestroy(&origCellPartSection);CHKERRQ(ierr);
100770034214SMatthew G. Knepley 
100870034214SMatthew G. Knepley     ierr = PetscSFBcastBegin(pointSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
100970034214SMatthew G. Knepley     ierr = PetscSFBcastEnd(pointSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
101070034214SMatthew G. Knepley     for (p = 0; p < numLeaves; ++p) {
101170034214SMatthew G. Knepley       if (lowners[p].rank < 0 || lowners[p].rank == rank) { /* Either put in a bid or we know we own it */
101270034214SMatthew G. Knepley         lowners[p].rank  = rank;
101370034214SMatthew G. Knepley         lowners[p].index = leaves ? leaves[p] : p;
101470034214SMatthew G. Knepley       } else if (lowners[p].rank >= 0) { /* Point already claimed so flag so that MAXLOC does not listen to us */
101570034214SMatthew G. Knepley         lowners[p].rank  = -2;
101670034214SMatthew G. Knepley         lowners[p].index = -2;
101770034214SMatthew G. Knepley       }
101870034214SMatthew G. Knepley     }
101970034214SMatthew G. Knepley     for (p=0; p<numRoots; p++) { /* Root must not participate in the rediction, flag so that MAXLOC does not use */
102070034214SMatthew G. Knepley       rowners[p].rank  = -3;
102170034214SMatthew G. Knepley       rowners[p].index = -3;
102270034214SMatthew G. Knepley     }
102370034214SMatthew G. Knepley     ierr = PetscSFReduceBegin(pointSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr);
102470034214SMatthew G. Knepley     ierr = PetscSFReduceEnd(pointSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr);
102570034214SMatthew G. Knepley     ierr = PetscSFBcastBegin(pointSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
102670034214SMatthew G. Knepley     ierr = PetscSFBcastEnd(pointSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
102770034214SMatthew G. Knepley     for (p = 0; p < numLeaves; ++p) {
102870034214SMatthew G. Knepley       if (lowners[p].rank < 0 || lowners[p].index < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Cell partition corrupt: point not claimed");
102970034214SMatthew G. Knepley       if (lowners[p].rank != rank) ++numGhostPoints;
103070034214SMatthew G. Knepley     }
103170034214SMatthew G. Knepley     ierr = PetscMalloc1(numGhostPoints,    &ghostPoints);CHKERRQ(ierr);
103270034214SMatthew G. Knepley     ierr = PetscMalloc1(numGhostPoints, &remotePoints);CHKERRQ(ierr);
103370034214SMatthew G. Knepley     for (p = 0, gp = 0; p < numLeaves; ++p) {
103470034214SMatthew G. Knepley       if (lowners[p].rank != rank) {
103570034214SMatthew G. Knepley         ghostPoints[gp]        = leaves ? leaves[p] : p;
103670034214SMatthew G. Knepley         remotePoints[gp].rank  = lowners[p].rank;
103770034214SMatthew G. Knepley         remotePoints[gp].index = lowners[p].index;
103870034214SMatthew G. Knepley         ++gp;
103970034214SMatthew G. Knepley       }
104070034214SMatthew G. Knepley     }
104170034214SMatthew G. Knepley     ierr = PetscFree2(rowners,lowners);CHKERRQ(ierr);
104270034214SMatthew G. Knepley     ierr = PetscSFSetGraph((*dmParallel)->sf, pEnd - pStart, numGhostPoints, ghostPoints, PETSC_OWN_POINTER, remotePoints, PETSC_OWN_POINTER);CHKERRQ(ierr);
104370034214SMatthew G. Knepley     ierr = PetscSFSetFromOptions((*dmParallel)->sf);CHKERRQ(ierr);
104470034214SMatthew G. Knepley   }
1045a42b08eeSMatthew G. Knepley   pmesh->useCone    = mesh->useCone;
1046a42b08eeSMatthew G. Knepley   pmesh->useClosure = mesh->useClosure;
10475b317d89SToby Isaac   pmesh->useAnchors = mesh->useAnchors;
104870034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr);
1049bf5244c7SMatthew G. Knepley   /* Distribute Coordinates */
1050bf5244c7SMatthew G. Knepley   {
1051bf5244c7SMatthew G. Knepley     PetscSection     originalCoordSection, newCoordSection;
1052bf5244c7SMatthew G. Knepley     Vec              originalCoordinates, newCoordinates;
1053bf5244c7SMatthew G. Knepley     PetscInt         bs;
1054bf5244c7SMatthew G. Knepley     const char      *name;
1055bf5244c7SMatthew G. Knepley     const PetscReal *maxCell, *L;
1056bf5244c7SMatthew G. Knepley 
1057bf5244c7SMatthew G. Knepley     ierr = DMGetCoordinateSection(dm, &originalCoordSection);CHKERRQ(ierr);
1058bf5244c7SMatthew G. Knepley     ierr = DMGetCoordinateSection(*dmParallel, &newCoordSection);CHKERRQ(ierr);
1059bf5244c7SMatthew G. Knepley     ierr = DMGetCoordinatesLocal(dm, &originalCoordinates);CHKERRQ(ierr);
10601c57f3caSMatthew G. Knepley     if (originalCoordinates) {
1061bf5244c7SMatthew G. Knepley       ierr = VecCreate(comm, &newCoordinates);CHKERRQ(ierr);
1062bf5244c7SMatthew G. Knepley       ierr = PetscObjectGetName((PetscObject) originalCoordinates, &name);CHKERRQ(ierr);
1063bf5244c7SMatthew G. Knepley       ierr = PetscObjectSetName((PetscObject) newCoordinates, name);CHKERRQ(ierr);
1064bf5244c7SMatthew G. Knepley 
1065bf5244c7SMatthew G. Knepley       ierr = DMPlexDistributeField(dm, pointSF, originalCoordSection, originalCoordinates, newCoordSection, newCoordinates);CHKERRQ(ierr);
1066bf5244c7SMatthew G. Knepley       ierr = DMSetCoordinatesLocal(*dmParallel, newCoordinates);CHKERRQ(ierr);
1067bf5244c7SMatthew G. Knepley       ierr = VecGetBlockSize(originalCoordinates, &bs);CHKERRQ(ierr);
1068bf5244c7SMatthew G. Knepley       ierr = VecSetBlockSize(newCoordinates, bs);CHKERRQ(ierr);
1069bf5244c7SMatthew G. Knepley       ierr = VecDestroy(&newCoordinates);CHKERRQ(ierr);
10701c57f3caSMatthew G. Knepley     }
1071bf5244c7SMatthew G. Knepley     ierr = DMGetPeriodicity(dm, &maxCell, &L);CHKERRQ(ierr);
1072bf5244c7SMatthew G. Knepley     if (L) {ierr = DMSetPeriodicity(*dmParallel, maxCell, L);CHKERRQ(ierr);}
1073bf5244c7SMatthew G. Knepley   }
1074bf5244c7SMatthew G. Knepley   /* Distribute labels */
1075bf5244c7SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr);
1076bf5244c7SMatthew G. Knepley   {
1077bf5244c7SMatthew G. Knepley     DMLabel  next      = mesh->labels, newNext = pmesh->labels;
1078bf5244c7SMatthew G. Knepley     PetscInt numLabels = 0, l;
1079bf5244c7SMatthew G. Knepley 
1080bf5244c7SMatthew G. Knepley     /* Bcast number of labels */
1081bf5244c7SMatthew G. Knepley     while (next) {++numLabels; next = next->next;}
1082bf5244c7SMatthew G. Knepley     ierr = MPI_Bcast(&numLabels, 1, MPIU_INT, 0, comm);CHKERRQ(ierr);
1083bf5244c7SMatthew G. Knepley     next = mesh->labels;
1084bf5244c7SMatthew G. Knepley     for (l = 0; l < numLabels; ++l) {
1085bf5244c7SMatthew G. Knepley       DMLabel   labelNew;
1086bf5244c7SMatthew G. Knepley       PetscBool isdepth;
1087bf5244c7SMatthew G. Knepley 
1088bf5244c7SMatthew G. Knepley       /* Skip "depth" because it is recreated */
1089bf5244c7SMatthew G. Knepley       if (!rank) {ierr = PetscStrcmp(next->name, "depth", &isdepth);CHKERRQ(ierr);}
1090bf5244c7SMatthew G. Knepley       ierr = MPI_Bcast(&isdepth, 1, MPIU_BOOL, 0, comm);CHKERRQ(ierr);
1091bf5244c7SMatthew G. Knepley       if (isdepth) {if (!rank) next = next->next; continue;}
1092bf5244c7SMatthew G. Knepley       ierr = DMLabelDistribute(next, partSection, part, renumbering, &labelNew);CHKERRQ(ierr);
1093bf5244c7SMatthew G. Knepley       /* Insert into list */
1094bf5244c7SMatthew G. Knepley       if (newNext) newNext->next = labelNew;
1095bf5244c7SMatthew G. Knepley       else         pmesh->labels = labelNew;
1096bf5244c7SMatthew G. Knepley       newNext = labelNew;
1097bf5244c7SMatthew G. Knepley       if (!rank) next = next->next;
1098bf5244c7SMatthew G. Knepley     }
1099bf5244c7SMatthew G. Knepley   }
1100bf5244c7SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr);
1101bf5244c7SMatthew G. Knepley   /* Setup hybrid structure */
1102bf5244c7SMatthew G. Knepley   {
1103bf5244c7SMatthew G. Knepley     const PetscInt *gpoints;
1104bf5244c7SMatthew G. Knepley     PetscInt        depth, n, d;
1105bf5244c7SMatthew G. Knepley 
1106bf5244c7SMatthew G. Knepley     for (d = 0; d <= dim; ++d) {pmesh->hybridPointMax[d] = mesh->hybridPointMax[d];}
1107bf5244c7SMatthew G. Knepley     ierr = MPI_Bcast(pmesh->hybridPointMax, dim+1, MPIU_INT, 0, comm);CHKERRQ(ierr);
1108bf5244c7SMatthew G. Knepley     ierr = ISLocalToGlobalMappingGetSize(renumbering, &n);CHKERRQ(ierr);
1109bf5244c7SMatthew G. Knepley     ierr = ISLocalToGlobalMappingGetIndices(renumbering, &gpoints);CHKERRQ(ierr);
1110bf5244c7SMatthew G. Knepley     ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
1111bf5244c7SMatthew G. Knepley     for (d = 0; d <= dim; ++d) {
1112bf5244c7SMatthew G. Knepley       PetscInt pmax = pmesh->hybridPointMax[d], newmax = 0, pEnd, stratum[2], p;
1113bf5244c7SMatthew G. Knepley 
1114bf5244c7SMatthew G. Knepley       if (pmax < 0) continue;
1115bf5244c7SMatthew G. Knepley       ierr = DMPlexGetDepthStratum(dm, d > depth ? depth : d, &stratum[0], &stratum[1]);CHKERRQ(ierr);
1116bf5244c7SMatthew G. Knepley       ierr = DMPlexGetDepthStratum(*dmParallel, d, NULL, &pEnd);CHKERRQ(ierr);
1117bf5244c7SMatthew G. Knepley       ierr = MPI_Bcast(stratum, 2, MPIU_INT, 0, comm);CHKERRQ(ierr);
1118bf5244c7SMatthew G. Knepley       for (p = 0; p < n; ++p) {
1119bf5244c7SMatthew G. Knepley         const PetscInt point = gpoints[p];
1120bf5244c7SMatthew G. Knepley 
1121bf5244c7SMatthew G. Knepley         if ((point >= stratum[0]) && (point < stratum[1]) && (point >= pmax)) ++newmax;
1122bf5244c7SMatthew G. Knepley       }
1123bf5244c7SMatthew G. Knepley       if (newmax > 0) pmesh->hybridPointMax[d] = pEnd - newmax;
1124bf5244c7SMatthew G. Knepley       else            pmesh->hybridPointMax[d] = -1;
1125bf5244c7SMatthew G. Knepley     }
1126bf5244c7SMatthew G. Knepley     ierr = ISLocalToGlobalMappingRestoreIndices(renumbering, &gpoints);CHKERRQ(ierr);
1127bf5244c7SMatthew G. Knepley   }
1128c389ea9fSToby Isaac   /* Set up tree */
1129c389ea9fSToby Isaac   {
1130c389ea9fSToby Isaac     DM              refTree;
1131c389ea9fSToby Isaac     PetscSection    origParentSection, newParentSection;
1132c389ea9fSToby Isaac     PetscInt        *origParents, *origChildIDs;
1133c389ea9fSToby Isaac 
1134c389ea9fSToby Isaac     ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
1135c389ea9fSToby Isaac     ierr = DMPlexSetReferenceTree(*dmParallel,refTree);CHKERRQ(ierr);
1136c389ea9fSToby Isaac     ierr = DMPlexGetTree(dm,&origParentSection,&origParents,&origChildIDs,NULL,NULL);CHKERRQ(ierr);
1137c389ea9fSToby Isaac     if (origParentSection) {
1138c389ea9fSToby Isaac       PetscInt        pStart, pEnd;
1139c389ea9fSToby Isaac       PetscInt        *newParents, *newChildIDs;
1140c389ea9fSToby Isaac       PetscInt        *remoteOffsetsParents, newParentSize;
1141c389ea9fSToby Isaac       PetscSF         parentSF;
1142c389ea9fSToby Isaac 
1143c389ea9fSToby Isaac       ierr = DMPlexGetChart(*dmParallel, &pStart, &pEnd);CHKERRQ(ierr);
1144c389ea9fSToby Isaac       ierr = PetscSectionCreate(PetscObjectComm((PetscObject)*dmParallel),&newParentSection);CHKERRQ(ierr);
1145c389ea9fSToby Isaac       ierr = PetscSectionSetChart(newParentSection,pStart,pEnd);CHKERRQ(ierr);
1146c389ea9fSToby Isaac       ierr = PetscSFDistributeSection(pointSF, origParentSection, &remoteOffsetsParents, newParentSection);CHKERRQ(ierr);
1147c389ea9fSToby Isaac       ierr = PetscSFCreateSectionSF(pointSF, origParentSection, remoteOffsetsParents, newParentSection, &parentSF);CHKERRQ(ierr);
1148c389ea9fSToby Isaac       ierr = PetscSectionGetStorageSize(newParentSection,&newParentSize);CHKERRQ(ierr);
1149c389ea9fSToby Isaac       ierr = PetscMalloc2(newParentSize,&newParents,newParentSize,&newChildIDs);CHKERRQ(ierr);
1150c389ea9fSToby Isaac       ierr = PetscSFBcastBegin(parentSF, MPIU_INT, origParents, newParents);CHKERRQ(ierr);
1151c389ea9fSToby Isaac       ierr = PetscSFBcastEnd(parentSF, MPIU_INT, origParents, newParents);CHKERRQ(ierr);
1152c389ea9fSToby Isaac       ierr = PetscSFBcastBegin(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr);
1153c389ea9fSToby Isaac       ierr = PetscSFBcastEnd(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr);
1154c389ea9fSToby Isaac       ierr = ISGlobalToLocalMappingApplyBlock(renumbering,IS_GTOLM_MASK, newParentSize, newParents, NULL, newParents);CHKERRQ(ierr);
1155c389ea9fSToby Isaac       ierr = PetscOptionsHasName(((PetscObject) dm)->prefix, "-parents_view", &flg);CHKERRQ(ierr);
1156c389ea9fSToby Isaac       if (flg) {
1157c389ea9fSToby Isaac         ierr = PetscPrintf(comm, "Serial Parent Section: \n");CHKERRQ(ierr);
1158c389ea9fSToby Isaac         ierr = PetscSectionView(origParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1159c389ea9fSToby Isaac         ierr = PetscPrintf(comm, "Parallel Parent Section: \n");CHKERRQ(ierr);
1160c389ea9fSToby Isaac         ierr = PetscSectionView(newParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1161c389ea9fSToby Isaac         ierr = PetscSFView(parentSF, NULL);CHKERRQ(ierr);
1162c389ea9fSToby Isaac       }
1163c389ea9fSToby Isaac       ierr = DMPlexSetTree(*dmParallel,newParentSection,newParents,newChildIDs);CHKERRQ(ierr);
1164c389ea9fSToby Isaac       ierr = PetscSectionDestroy(&newParentSection);CHKERRQ(ierr);
1165c389ea9fSToby Isaac       ierr = PetscFree2(newParents,newChildIDs);CHKERRQ(ierr);
1166c389ea9fSToby Isaac       ierr = PetscSFDestroy(&parentSF);CHKERRQ(ierr);
1167c389ea9fSToby Isaac     }
1168c389ea9fSToby Isaac   }
1169bf5244c7SMatthew G. Knepley   /* Cleanup Partition */
1170bf5244c7SMatthew G. Knepley   ierr = ISLocalToGlobalMappingDestroy(&renumbering);CHKERRQ(ierr);
1171bf5244c7SMatthew G. Knepley   ierr = PetscSFDestroy(&partSF);CHKERRQ(ierr);
1172bf5244c7SMatthew G. Knepley   ierr = PetscSectionDestroy(&partSection);CHKERRQ(ierr);
1173bf5244c7SMatthew G. Knepley   ierr = ISDestroy(&part);CHKERRQ(ierr);
117486719b60SMatthew G. Knepley   /* Copy BC */
117586719b60SMatthew G. Knepley   ierr = DMPlexCopyBoundary(dm, *dmParallel);CHKERRQ(ierr);
117670034214SMatthew G. Knepley   /* Cleanup */
117770034214SMatthew G. Knepley   if (sf) {*sf = pointSF;}
117870034214SMatthew G. Knepley   else    {ierr = PetscSFDestroy(&pointSF);CHKERRQ(ierr);}
117970034214SMatthew G. Knepley   ierr = DMSetFromOptions(*dmParallel);CHKERRQ(ierr);
118070034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr);
118170034214SMatthew G. Knepley   PetscFunctionReturn(0);
118270034214SMatthew G. Knepley }
1183