xref: /petsc/src/dm/impls/plex/plexdistribute.c (revision be200f8d7b9f2c03ad91139b9b00a1aef64bd73e)
1af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h>    /*I      "petscdmplex.h"   I*/
2c58f1c22SToby Isaac #include <petsc/private/dmlabelimpl.h>   /*I      "petscdmlabel.h"  I*/
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
174b6b44bdSMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in support(p+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
474b6b44bdSMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in support(p+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
764b6b44bdSMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in support(p+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
1064b6b44bdSMatthew G. Knepley $     FVM:   Two points p and q are adjacent if q \in support(p+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);
1824b6b44bdSMatthew G. Knepley   for (c = 0; c <= coneSize; ++c) {
1834b6b44bdSMatthew G. Knepley     const PetscInt  point   = !c ? p : cone[c-1];
18470034214SMatthew G. Knepley     const PetscInt *support = NULL;
18570034214SMatthew G. Knepley     PetscInt        supportSize, s, q;
18670034214SMatthew G. Knepley 
1874b6b44bdSMatthew G. Knepley     ierr = DMPlexGetSupportSize(dm, point, &supportSize);CHKERRQ(ierr);
1884b6b44bdSMatthew G. Knepley     ierr = DMPlexGetSupport(dm, point, &support);CHKERRQ(ierr);
18970034214SMatthew G. Knepley     for (s = 0; s < supportSize; ++s) {
19070034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = support[s],0); ++q) {
19170034214SMatthew G. Knepley         if (support[s] == adj[q]) break;
19270034214SMatthew G. Knepley       }
19370034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
19470034214SMatthew G. Knepley     }
19570034214SMatthew G. Knepley   }
19670034214SMatthew G. Knepley   *adjSize = numAdj;
19770034214SMatthew G. Knepley   PetscFunctionReturn(0);
19870034214SMatthew G. Knepley }
19970034214SMatthew G. Knepley 
20070034214SMatthew G. Knepley #undef __FUNCT__
20170034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Support_Internal"
20270034214SMatthew G. Knepley static PetscErrorCode DMPlexGetAdjacency_Support_Internal(DM dm, PetscInt p, PetscInt *adjSize, PetscInt adj[])
20370034214SMatthew G. Knepley {
20470034214SMatthew G. Knepley   const PetscInt *support = NULL;
20570034214SMatthew G. Knepley   PetscInt        numAdj   = 0, maxAdjSize = *adjSize, supportSize, s;
20670034214SMatthew G. Knepley   PetscErrorCode  ierr;
20770034214SMatthew G. Knepley 
20870034214SMatthew G. Knepley   PetscFunctionBeginHot;
20970034214SMatthew G. Knepley   ierr = DMPlexGetSupportSize(dm, p, &supportSize);CHKERRQ(ierr);
21070034214SMatthew G. Knepley   ierr = DMPlexGetSupport(dm, p, &support);CHKERRQ(ierr);
2114b6b44bdSMatthew G. Knepley   for (s = 0; s <= supportSize; ++s) {
2124b6b44bdSMatthew G. Knepley     const PetscInt  point = !s ? p : support[s-1];
21370034214SMatthew G. Knepley     const PetscInt *cone  = NULL;
21470034214SMatthew G. Knepley     PetscInt        coneSize, c, q;
21570034214SMatthew G. Knepley 
2164b6b44bdSMatthew G. Knepley     ierr = DMPlexGetConeSize(dm, point, &coneSize);CHKERRQ(ierr);
2174b6b44bdSMatthew G. Knepley     ierr = DMPlexGetCone(dm, point, &cone);CHKERRQ(ierr);
21870034214SMatthew G. Knepley     for (c = 0; c < coneSize; ++c) {
21970034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = cone[c],0); ++q) {
22070034214SMatthew G. Knepley         if (cone[c] == adj[q]) break;
22170034214SMatthew G. Knepley       }
22270034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
22370034214SMatthew G. Knepley     }
22470034214SMatthew G. Knepley   }
22570034214SMatthew G. Knepley   *adjSize = numAdj;
22670034214SMatthew G. Knepley   PetscFunctionReturn(0);
22770034214SMatthew G. Knepley }
22870034214SMatthew G. Knepley 
22970034214SMatthew G. Knepley #undef __FUNCT__
23070034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Transitive_Internal"
23170034214SMatthew G. Knepley static PetscErrorCode DMPlexGetAdjacency_Transitive_Internal(DM dm, PetscInt p, PetscBool useClosure, PetscInt *adjSize, PetscInt adj[])
23270034214SMatthew G. Knepley {
23370034214SMatthew G. Knepley   PetscInt      *star = NULL;
23470034214SMatthew G. Knepley   PetscInt       numAdj = 0, maxAdjSize = *adjSize, starSize, s;
23570034214SMatthew G. Knepley   PetscErrorCode ierr;
23670034214SMatthew G. Knepley 
23770034214SMatthew G. Knepley   PetscFunctionBeginHot;
23870034214SMatthew G. Knepley   ierr = DMPlexGetTransitiveClosure(dm, p, useClosure, &starSize, &star);CHKERRQ(ierr);
23970034214SMatthew G. Knepley   for (s = 0; s < starSize*2; s += 2) {
24070034214SMatthew G. Knepley     const PetscInt *closure = NULL;
24170034214SMatthew G. Knepley     PetscInt        closureSize, c, q;
24270034214SMatthew G. Knepley 
24370034214SMatthew G. Knepley     ierr = DMPlexGetTransitiveClosure(dm, star[s], (PetscBool)!useClosure, &closureSize, (PetscInt**) &closure);CHKERRQ(ierr);
24470034214SMatthew G. Knepley     for (c = 0; c < closureSize*2; c += 2) {
24570034214SMatthew G. Knepley       for (q = 0; q < numAdj || (adj[numAdj++] = closure[c],0); ++q) {
24670034214SMatthew G. Knepley         if (closure[c] == adj[q]) break;
24770034214SMatthew G. Knepley       }
24870034214SMatthew G. Knepley       if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
24970034214SMatthew G. Knepley     }
25070034214SMatthew G. Knepley     ierr = DMPlexRestoreTransitiveClosure(dm, star[s], (PetscBool)!useClosure, &closureSize, (PetscInt**) &closure);CHKERRQ(ierr);
25170034214SMatthew G. Knepley   }
25270034214SMatthew G. Knepley   ierr = DMPlexRestoreTransitiveClosure(dm, p, useClosure, &starSize, &star);CHKERRQ(ierr);
25370034214SMatthew G. Knepley   *adjSize = numAdj;
25470034214SMatthew G. Knepley   PetscFunctionReturn(0);
25570034214SMatthew G. Knepley }
25670034214SMatthew G. Knepley 
25770034214SMatthew G. Knepley #undef __FUNCT__
25870034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency_Internal"
2595b317d89SToby Isaac PetscErrorCode DMPlexGetAdjacency_Internal(DM dm, PetscInt p, PetscBool useCone, PetscBool useTransitiveClosure, PetscBool useAnchors, PetscInt *adjSize, PetscInt *adj[])
26070034214SMatthew G. Knepley {
26179bad331SMatthew G. Knepley   static PetscInt asiz = 0;
2628b0b4c70SToby Isaac   PetscInt maxAnchors = 1;
2638b0b4c70SToby Isaac   PetscInt aStart = -1, aEnd = -1;
2648b0b4c70SToby Isaac   PetscInt maxAdjSize;
2658b0b4c70SToby Isaac   PetscSection aSec = NULL;
2668b0b4c70SToby Isaac   IS aIS = NULL;
2678b0b4c70SToby Isaac   const PetscInt *anchors;
26870034214SMatthew G. Knepley   PetscErrorCode  ierr;
26970034214SMatthew G. Knepley 
27070034214SMatthew G. Knepley   PetscFunctionBeginHot;
2715b317d89SToby Isaac   if (useAnchors) {
272a17985deSToby Isaac     ierr = DMPlexGetAnchors(dm,&aSec,&aIS);CHKERRQ(ierr);
2738b0b4c70SToby Isaac     if (aSec) {
2748b0b4c70SToby Isaac       ierr = PetscSectionGetMaxDof(aSec,&maxAnchors);CHKERRQ(ierr);
27524c766afSToby Isaac       maxAnchors = PetscMax(1,maxAnchors);
2768b0b4c70SToby Isaac       ierr = PetscSectionGetChart(aSec,&aStart,&aEnd);CHKERRQ(ierr);
2778b0b4c70SToby Isaac       ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
2788b0b4c70SToby Isaac     }
2798b0b4c70SToby Isaac   }
28079bad331SMatthew G. Knepley   if (!*adj) {
28124c766afSToby Isaac     PetscInt depth, coneSeries, supportSeries, maxC, maxS, pStart, pEnd;
28279bad331SMatthew G. Knepley 
28324c766afSToby Isaac     ierr  = DMPlexGetChart(dm, &pStart,&pEnd);CHKERRQ(ierr);
28479bad331SMatthew G. Knepley     ierr  = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
28524c766afSToby Isaac     ierr  = DMPlexGetMaxSizes(dm, &maxC, &maxS);CHKERRQ(ierr);
28624c766afSToby Isaac     coneSeries    = (maxC > 1) ? ((PetscPowInt(maxC,depth+1)-1)/(maxC-1)) : depth+1;
28724c766afSToby Isaac     supportSeries = (maxS > 1) ? ((PetscPowInt(maxS,depth+1)-1)/(maxS-1)) : depth+1;
28824c766afSToby Isaac     asiz  = PetscMax(PetscPowInt(maxS,depth)*coneSeries,PetscPowInt(maxC,depth)*supportSeries);
2898b0b4c70SToby Isaac     asiz *= maxAnchors;
29024c766afSToby Isaac     asiz  = PetscMin(asiz,pEnd-pStart);
29179bad331SMatthew G. Knepley     ierr  = PetscMalloc1(asiz,adj);CHKERRQ(ierr);
29279bad331SMatthew G. Knepley   }
29379bad331SMatthew G. Knepley   if (*adjSize < 0) *adjSize = asiz;
2948b0b4c70SToby Isaac   maxAdjSize = *adjSize;
29570034214SMatthew G. Knepley   if (useTransitiveClosure) {
29679bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Transitive_Internal(dm, p, useCone, adjSize, *adj);CHKERRQ(ierr);
29770034214SMatthew G. Knepley   } else if (useCone) {
29879bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Cone_Internal(dm, p, adjSize, *adj);CHKERRQ(ierr);
29970034214SMatthew G. Knepley   } else {
30079bad331SMatthew G. Knepley     ierr = DMPlexGetAdjacency_Support_Internal(dm, p, adjSize, *adj);CHKERRQ(ierr);
30170034214SMatthew G. Knepley   }
3025b317d89SToby Isaac   if (useAnchors && aSec) {
3038b0b4c70SToby Isaac     PetscInt origSize = *adjSize;
3048b0b4c70SToby Isaac     PetscInt numAdj = origSize;
3058b0b4c70SToby Isaac     PetscInt i = 0, j;
3068b0b4c70SToby Isaac     PetscInt *orig = *adj;
3078b0b4c70SToby Isaac 
3088b0b4c70SToby Isaac     while (i < origSize) {
3098b0b4c70SToby Isaac       PetscInt p = orig[i];
3108b0b4c70SToby Isaac       PetscInt aDof = 0;
3118b0b4c70SToby Isaac 
3128b0b4c70SToby Isaac       if (p >= aStart && p < aEnd) {
3138b0b4c70SToby Isaac         ierr = PetscSectionGetDof(aSec,p,&aDof);CHKERRQ(ierr);
3148b0b4c70SToby Isaac       }
3158b0b4c70SToby Isaac       if (aDof) {
3168b0b4c70SToby Isaac         PetscInt aOff;
3178b0b4c70SToby Isaac         PetscInt s, q;
3188b0b4c70SToby Isaac 
3198b0b4c70SToby Isaac         for (j = i + 1; j < numAdj; j++) {
3208b0b4c70SToby Isaac           orig[j - 1] = orig[j];
3218b0b4c70SToby Isaac         }
3228b0b4c70SToby Isaac         origSize--;
3238b0b4c70SToby Isaac         numAdj--;
3248b0b4c70SToby Isaac         ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
3258b0b4c70SToby Isaac         for (s = 0; s < aDof; ++s) {
3268b0b4c70SToby Isaac           for (q = 0; q < numAdj || (orig[numAdj++] = anchors[aOff+s],0); ++q) {
3278b0b4c70SToby Isaac             if (anchors[aOff+s] == orig[q]) break;
3288b0b4c70SToby Isaac           }
3298b0b4c70SToby Isaac           if (numAdj > maxAdjSize) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid mesh exceeded adjacency allocation (%D)", maxAdjSize);
3308b0b4c70SToby Isaac         }
3318b0b4c70SToby Isaac       }
3328b0b4c70SToby Isaac       else {
3338b0b4c70SToby Isaac         i++;
3348b0b4c70SToby Isaac       }
3358b0b4c70SToby Isaac     }
3368b0b4c70SToby Isaac     *adjSize = numAdj;
3378b0b4c70SToby Isaac     ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
3388b0b4c70SToby Isaac   }
33970034214SMatthew G. Knepley   PetscFunctionReturn(0);
34070034214SMatthew G. Knepley }
34170034214SMatthew G. Knepley 
34270034214SMatthew G. Knepley #undef __FUNCT__
34370034214SMatthew G. Knepley #define __FUNCT__ "DMPlexGetAdjacency"
34470034214SMatthew G. Knepley /*@
34570034214SMatthew G. Knepley   DMPlexGetAdjacency - Return all points adjacent to the given point
34670034214SMatthew G. Knepley 
34770034214SMatthew G. Knepley   Input Parameters:
34870034214SMatthew G. Knepley + dm - The DM object
34970034214SMatthew G. Knepley . p  - The point
35070034214SMatthew G. Knepley . adjSize - The maximum size of adj if it is non-NULL, or PETSC_DETERMINE
35170034214SMatthew G. Knepley - adj - Either NULL so that the array is allocated, or an existing array with size adjSize
35270034214SMatthew G. Knepley 
35370034214SMatthew G. Knepley   Output Parameters:
35470034214SMatthew G. Knepley + adjSize - The number of adjacent points
35570034214SMatthew G. Knepley - adj - The adjacent points
35670034214SMatthew G. Knepley 
35770034214SMatthew G. Knepley   Level: advanced
35870034214SMatthew G. Knepley 
35970034214SMatthew G. Knepley   Notes: The user must PetscFree the adj array if it was not passed in.
36070034214SMatthew G. Knepley 
36170034214SMatthew G. Knepley .seealso: DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure(), DMPlexDistribute(), DMCreateMatrix(), DMPlexPreallocateOperator()
36270034214SMatthew G. Knepley @*/
36370034214SMatthew G. Knepley PetscErrorCode DMPlexGetAdjacency(DM dm, PetscInt p, PetscInt *adjSize, PetscInt *adj[])
36470034214SMatthew G. Knepley {
36570034214SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *) dm->data;
36670034214SMatthew G. Knepley   PetscErrorCode ierr;
36770034214SMatthew G. Knepley 
36870034214SMatthew G. Knepley   PetscFunctionBeginHot;
36970034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
37070034214SMatthew G. Knepley   PetscValidPointer(adjSize,3);
37170034214SMatthew G. Knepley   PetscValidPointer(adj,4);
3725b317d89SToby Isaac   ierr = DMPlexGetAdjacency_Internal(dm, p, mesh->useCone, mesh->useClosure, mesh->useAnchors, adjSize, adj);CHKERRQ(ierr);
37370034214SMatthew G. Knepley   PetscFunctionReturn(0);
37470034214SMatthew G. Knepley }
37508633170SToby Isaac 
376b0a623aaSMatthew G. Knepley #undef __FUNCT__
377b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexCreateTwoSidedProcessSF"
378b0a623aaSMatthew G. Knepley /*@
379b0a623aaSMatthew G. Knepley   DMPlexCreateTwoSidedProcessSF - Create an SF which just has process connectivity
380b0a623aaSMatthew G. Knepley 
381b0a623aaSMatthew G. Knepley   Collective on DM
382b0a623aaSMatthew G. Knepley 
383b0a623aaSMatthew G. Knepley   Input Parameters:
384b0a623aaSMatthew G. Knepley + dm      - The DM
385b0a623aaSMatthew G. Knepley - sfPoint - The PetscSF which encodes point connectivity
386b0a623aaSMatthew G. Knepley 
387b0a623aaSMatthew G. Knepley   Output Parameters:
388b0a623aaSMatthew G. Knepley + processRanks - A list of process neighbors, or NULL
389b0a623aaSMatthew G. Knepley - sfProcess    - An SF encoding the two-sided process connectivity, or NULL
390b0a623aaSMatthew G. Knepley 
391b0a623aaSMatthew G. Knepley   Level: developer
392b0a623aaSMatthew G. Knepley 
393b0a623aaSMatthew G. Knepley .seealso: PetscSFCreate(), DMPlexCreateProcessSF()
394b0a623aaSMatthew G. Knepley @*/
395b0a623aaSMatthew G. Knepley PetscErrorCode DMPlexCreateTwoSidedProcessSF(DM dm, PetscSF sfPoint, PetscSection rootRankSection, IS rootRanks, PetscSection leafRankSection, IS leafRanks, IS *processRanks, PetscSF *sfProcess)
396b0a623aaSMatthew G. Knepley {
397b0a623aaSMatthew G. Knepley   const PetscSFNode *remotePoints;
398b0a623aaSMatthew G. Knepley   PetscInt          *localPointsNew;
399b0a623aaSMatthew G. Knepley   PetscSFNode       *remotePointsNew;
400b0a623aaSMatthew G. Knepley   const PetscInt    *nranks;
401b0a623aaSMatthew G. Knepley   PetscInt          *ranksNew;
402b0a623aaSMatthew G. Knepley   PetscBT            neighbors;
403b0a623aaSMatthew G. Knepley   PetscInt           pStart, pEnd, p, numLeaves, l, numNeighbors, n;
404b0a623aaSMatthew G. Knepley   PetscMPIInt        numProcs, proc, rank;
405b0a623aaSMatthew G. Knepley   PetscErrorCode     ierr;
406b0a623aaSMatthew G. Knepley 
407b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
408b0a623aaSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
409b0a623aaSMatthew G. Knepley   PetscValidHeaderSpecific(sfPoint, PETSCSF_CLASSID, 2);
410b0a623aaSMatthew G. Knepley   if (processRanks) {PetscValidPointer(processRanks, 3);}
411b0a623aaSMatthew G. Knepley   if (sfProcess)    {PetscValidPointer(sfProcess, 4);}
412b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm), &numProcs);CHKERRQ(ierr);
413b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
414b0a623aaSMatthew G. Knepley   ierr = PetscSFGetGraph(sfPoint, NULL, &numLeaves, NULL, &remotePoints);CHKERRQ(ierr);
415b0a623aaSMatthew G. Knepley   ierr = PetscBTCreate(numProcs, &neighbors);CHKERRQ(ierr);
416b0a623aaSMatthew G. Knepley   ierr = PetscBTMemzero(numProcs, neighbors);CHKERRQ(ierr);
417b0a623aaSMatthew G. Knepley   /* Compute root-to-leaf process connectivity */
418b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(rootRankSection, &pStart, &pEnd);CHKERRQ(ierr);
419b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(rootRanks, &nranks);CHKERRQ(ierr);
420b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
421b0a623aaSMatthew G. Knepley     PetscInt ndof, noff, n;
422b0a623aaSMatthew G. Knepley 
423b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(rootRankSection, p, &ndof);CHKERRQ(ierr);
424b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(rootRankSection, p, &noff);CHKERRQ(ierr);
425302440fdSBarry Smith     for (n = 0; n < ndof; ++n) {ierr = PetscBTSet(neighbors, nranks[noff+n]);CHKERRQ(ierr);}
426b0a623aaSMatthew G. Knepley   }
427b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(rootRanks, &nranks);CHKERRQ(ierr);
428b0a623aaSMatthew G. Knepley   /* Compute leaf-to-neighbor process connectivity */
429b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(leafRankSection, &pStart, &pEnd);CHKERRQ(ierr);
430b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(leafRanks, &nranks);CHKERRQ(ierr);
431b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
432b0a623aaSMatthew G. Knepley     PetscInt ndof, noff, n;
433b0a623aaSMatthew G. Knepley 
434b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(leafRankSection, p, &ndof);CHKERRQ(ierr);
435b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(leafRankSection, p, &noff);CHKERRQ(ierr);
436302440fdSBarry Smith     for (n = 0; n < ndof; ++n) {ierr = PetscBTSet(neighbors, nranks[noff+n]);CHKERRQ(ierr);}
437b0a623aaSMatthew G. Knepley   }
438b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(leafRanks, &nranks);CHKERRQ(ierr);
439b0a623aaSMatthew G. Knepley   /* Compute leaf-to-root process connectivity */
440b0a623aaSMatthew G. Knepley   for (l = 0; l < numLeaves; ++l) {PetscBTSet(neighbors, remotePoints[l].rank);}
441b0a623aaSMatthew G. Knepley   /* Calculate edges */
442b0a623aaSMatthew G. Knepley   PetscBTClear(neighbors, rank);
443b0a623aaSMatthew G. Knepley   for(proc = 0, numNeighbors = 0; proc < numProcs; ++proc) {if (PetscBTLookup(neighbors, proc)) ++numNeighbors;}
444b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &ranksNew);CHKERRQ(ierr);
445b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &localPointsNew);CHKERRQ(ierr);
446b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(numNeighbors, &remotePointsNew);CHKERRQ(ierr);
447b0a623aaSMatthew G. Knepley   for(proc = 0, n = 0; proc < numProcs; ++proc) {
448b0a623aaSMatthew G. Knepley     if (PetscBTLookup(neighbors, proc)) {
449b0a623aaSMatthew G. Knepley       ranksNew[n]              = proc;
450b0a623aaSMatthew G. Knepley       localPointsNew[n]        = proc;
451b0a623aaSMatthew G. Knepley       remotePointsNew[n].index = rank;
452b0a623aaSMatthew G. Knepley       remotePointsNew[n].rank  = proc;
453b0a623aaSMatthew G. Knepley       ++n;
454b0a623aaSMatthew G. Knepley     }
455b0a623aaSMatthew G. Knepley   }
456b0a623aaSMatthew G. Knepley   ierr = PetscBTDestroy(&neighbors);CHKERRQ(ierr);
457b0a623aaSMatthew G. Knepley   if (processRanks) {ierr = ISCreateGeneral(PetscObjectComm((PetscObject)dm), numNeighbors, ranksNew, PETSC_OWN_POINTER, processRanks);CHKERRQ(ierr);}
458b0a623aaSMatthew G. Knepley   else              {ierr = PetscFree(ranksNew);CHKERRQ(ierr);}
459b0a623aaSMatthew G. Knepley   if (sfProcess) {
460b0a623aaSMatthew G. Knepley     ierr = PetscSFCreate(PetscObjectComm((PetscObject)dm), sfProcess);CHKERRQ(ierr);
461b0a623aaSMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject) *sfProcess, "Two-Sided Process SF");CHKERRQ(ierr);
462b0a623aaSMatthew G. Knepley     ierr = PetscSFSetFromOptions(*sfProcess);CHKERRQ(ierr);
463b0a623aaSMatthew G. Knepley     ierr = PetscSFSetGraph(*sfProcess, numProcs, numNeighbors, localPointsNew, PETSC_OWN_POINTER, remotePointsNew, PETSC_OWN_POINTER);CHKERRQ(ierr);
464b0a623aaSMatthew G. Knepley   }
465b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
466b0a623aaSMatthew G. Knepley }
467b0a623aaSMatthew G. Knepley 
468b0a623aaSMatthew G. Knepley #undef __FUNCT__
469b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeOwnership"
470b0a623aaSMatthew G. Knepley /*@
471b0a623aaSMatthew G. Knepley   DMPlexDistributeOwnership - Compute owner information for shared points. This basically gets two-sided for an SF.
472b0a623aaSMatthew G. Knepley 
473b0a623aaSMatthew G. Knepley   Collective on DM
474b0a623aaSMatthew G. Knepley 
475b0a623aaSMatthew G. Knepley   Input Parameter:
476b0a623aaSMatthew G. Knepley . dm - The DM
477b0a623aaSMatthew G. Knepley 
478b0a623aaSMatthew G. Knepley   Output Parameters:
479b0a623aaSMatthew G. Knepley + rootSection - The number of leaves for a given root point
480b0a623aaSMatthew G. Knepley . rootrank    - The rank of each edge into the root point
481b0a623aaSMatthew G. Knepley . leafSection - The number of processes sharing a given leaf point
482b0a623aaSMatthew G. Knepley - leafrank    - The rank of each process sharing a leaf point
483b0a623aaSMatthew G. Knepley 
484b0a623aaSMatthew G. Knepley   Level: developer
485b0a623aaSMatthew G. Knepley 
486b0a623aaSMatthew G. Knepley .seealso: DMPlexCreateOverlap()
487b0a623aaSMatthew G. Knepley @*/
488b0a623aaSMatthew G. Knepley PetscErrorCode DMPlexDistributeOwnership(DM dm, PetscSection rootSection, IS *rootrank, PetscSection leafSection, IS *leafrank)
489b0a623aaSMatthew G. Knepley {
490b0a623aaSMatthew G. Knepley   MPI_Comm        comm;
491b0a623aaSMatthew G. Knepley   PetscSF         sfPoint;
492b0a623aaSMatthew G. Knepley   const PetscInt *rootdegree;
493b0a623aaSMatthew G. Knepley   PetscInt       *myrank, *remoterank;
494b0a623aaSMatthew G. Knepley   PetscInt        pStart, pEnd, p, nedges;
495b0a623aaSMatthew G. Knepley   PetscMPIInt     rank;
496b0a623aaSMatthew G. Knepley   PetscErrorCode  ierr;
497b0a623aaSMatthew G. Knepley 
498b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
499b0a623aaSMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
500b0a623aaSMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
501b0a623aaSMatthew G. Knepley   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
502b0a623aaSMatthew G. Knepley   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
503b0a623aaSMatthew G. Knepley   /* Compute number of leaves for each root */
504b0a623aaSMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) rootSection, "Root Section");CHKERRQ(ierr);
505b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetChart(rootSection, pStart, pEnd);CHKERRQ(ierr);
506b0a623aaSMatthew G. Knepley   ierr = PetscSFComputeDegreeBegin(sfPoint, &rootdegree);CHKERRQ(ierr);
507b0a623aaSMatthew G. Knepley   ierr = PetscSFComputeDegreeEnd(sfPoint, &rootdegree);CHKERRQ(ierr);
508b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {ierr = PetscSectionSetDof(rootSection, p, rootdegree[p-pStart]);CHKERRQ(ierr);}
509b0a623aaSMatthew G. Knepley   ierr = PetscSectionSetUp(rootSection);CHKERRQ(ierr);
510b0a623aaSMatthew G. Knepley   /* Gather rank of each leaf to root */
511b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetStorageSize(rootSection, &nedges);CHKERRQ(ierr);
512b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(pEnd-pStart, &myrank);CHKERRQ(ierr);
513b0a623aaSMatthew G. Knepley   ierr = PetscMalloc1(nedges,  &remoterank);CHKERRQ(ierr);
514b0a623aaSMatthew G. Knepley   for (p = 0; p < pEnd-pStart; ++p) myrank[p] = rank;
515b0a623aaSMatthew G. Knepley   ierr = PetscSFGatherBegin(sfPoint, MPIU_INT, myrank, remoterank);CHKERRQ(ierr);
516b0a623aaSMatthew G. Knepley   ierr = PetscSFGatherEnd(sfPoint, MPIU_INT, myrank, remoterank);CHKERRQ(ierr);
517b0a623aaSMatthew G. Knepley   ierr = PetscFree(myrank);CHKERRQ(ierr);
518b0a623aaSMatthew G. Knepley   ierr = ISCreateGeneral(comm, nedges, remoterank, PETSC_OWN_POINTER, rootrank);CHKERRQ(ierr);
519b0a623aaSMatthew G. Knepley   /* Distribute remote ranks to leaves */
520b0a623aaSMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) leafSection, "Leaf Section");CHKERRQ(ierr);
521b0a623aaSMatthew G. Knepley   ierr = DMPlexDistributeFieldIS(dm, sfPoint, rootSection, *rootrank, leafSection, leafrank);CHKERRQ(ierr);
522b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
523b0a623aaSMatthew G. Knepley }
524b0a623aaSMatthew G. Knepley 
525b0a623aaSMatthew G. Knepley #undef __FUNCT__
526b0a623aaSMatthew G. Knepley #define __FUNCT__ "DMPlexCreateOverlap"
527278397a0SMatthew G. Knepley /*@C
528b0a623aaSMatthew G. Knepley   DMPlexCreateOverlap - Compute owner information for shared points. This basically gets two-sided for an SF.
529b0a623aaSMatthew G. Knepley 
530b0a623aaSMatthew G. Knepley   Collective on DM
531b0a623aaSMatthew G. Knepley 
532b0a623aaSMatthew G. Knepley   Input Parameters:
533b0a623aaSMatthew G. Knepley + dm          - The DM
53424d039d7SMichael Lange . levels      - Number of overlap levels
535b0a623aaSMatthew G. Knepley . rootSection - The number of leaves for a given root point
536b0a623aaSMatthew G. Knepley . rootrank    - The rank of each edge into the root point
537b0a623aaSMatthew G. Knepley . leafSection - The number of processes sharing a given leaf point
538b0a623aaSMatthew G. Knepley - leafrank    - The rank of each process sharing a leaf point
539b0a623aaSMatthew G. Knepley 
540b0a623aaSMatthew G. Knepley   Output Parameters:
541a9f1d5b2SMichael Lange + ovLabel     - DMLabel containing remote overlap contributions as point/rank pairings
542b0a623aaSMatthew G. Knepley 
543b0a623aaSMatthew G. Knepley   Level: developer
544b0a623aaSMatthew G. Knepley 
5451fd9873aSMichael Lange .seealso: DMPlexDistributeOwnership(), DMPlexDistribute()
546b0a623aaSMatthew G. Knepley @*/
547a9f1d5b2SMichael Lange PetscErrorCode DMPlexCreateOverlap(DM dm, PetscInt levels, PetscSection rootSection, IS rootrank, PetscSection leafSection, IS leafrank, DMLabel *ovLabel)
548b0a623aaSMatthew G. Knepley {
549e540f424SMichael Lange   MPI_Comm           comm;
550b0a623aaSMatthew G. Knepley   DMLabel            ovAdjByRank; /* A DMLabel containing all points adjacent to shared points, separated by rank (value in label) */
551b0a623aaSMatthew G. Knepley   PetscSF            sfPoint, sfProc;
552b0a623aaSMatthew G. Knepley   const PetscSFNode *remote;
553b0a623aaSMatthew G. Knepley   const PetscInt    *local;
5541fd9873aSMichael Lange   const PetscInt    *nrank, *rrank;
555b0a623aaSMatthew G. Knepley   PetscInt          *adj = NULL;
5561fd9873aSMichael Lange   PetscInt           pStart, pEnd, p, sStart, sEnd, nleaves, l;
557b0a623aaSMatthew G. Knepley   PetscMPIInt        rank, numProcs;
55826a7d390SMatthew G. Knepley   PetscBool          useCone, useClosure, flg;
559b0a623aaSMatthew G. Knepley   PetscErrorCode     ierr;
560b0a623aaSMatthew G. Knepley 
561b0a623aaSMatthew G. Knepley   PetscFunctionBegin;
562e540f424SMichael Lange   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
563e540f424SMichael Lange   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
564e540f424SMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
565b0a623aaSMatthew G. Knepley   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
566b0a623aaSMatthew G. Knepley   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
567b0a623aaSMatthew G. Knepley   ierr = PetscSectionGetChart(leafSection, &sStart, &sEnd);CHKERRQ(ierr);
568b0a623aaSMatthew G. Knepley   ierr = PetscSFGetGraph(sfPoint, NULL, &nleaves, &local, &remote);CHKERRQ(ierr);
569b0a623aaSMatthew G. Knepley   ierr = DMLabelCreate("Overlap adjacency", &ovAdjByRank);CHKERRQ(ierr);
570b0a623aaSMatthew G. Knepley   /* Handle leaves: shared with the root point */
571b0a623aaSMatthew G. Knepley   for (l = 0; l < nleaves; ++l) {
572b0a623aaSMatthew G. Knepley     PetscInt adjSize = PETSC_DETERMINE, a;
573b0a623aaSMatthew G. Knepley 
574b0a623aaSMatthew G. Knepley     ierr = DMPlexGetAdjacency(dm, local[l], &adjSize, &adj);CHKERRQ(ierr);
575b0a623aaSMatthew G. Knepley     for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remote[l].rank);CHKERRQ(ierr);}
576b0a623aaSMatthew G. Knepley   }
577b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(rootrank, &rrank);CHKERRQ(ierr);
578b0a623aaSMatthew G. Knepley   ierr = ISGetIndices(leafrank, &nrank);CHKERRQ(ierr);
579b0a623aaSMatthew G. Knepley   /* Handle roots */
580b0a623aaSMatthew G. Knepley   for (p = pStart; p < pEnd; ++p) {
581b0a623aaSMatthew G. Knepley     PetscInt adjSize = PETSC_DETERMINE, neighbors = 0, noff, n, a;
582b0a623aaSMatthew G. Knepley 
583b0a623aaSMatthew G. Knepley     if ((p >= sStart) && (p < sEnd)) {
584b0a623aaSMatthew G. Knepley       /* Some leaves share a root with other leaves on different processes */
585b0a623aaSMatthew G. Knepley       ierr = PetscSectionGetDof(leafSection, p, &neighbors);CHKERRQ(ierr);
586b0a623aaSMatthew G. Knepley       if (neighbors) {
587b0a623aaSMatthew G. Knepley         ierr = PetscSectionGetOffset(leafSection, p, &noff);CHKERRQ(ierr);
588b0a623aaSMatthew G. Knepley         ierr = DMPlexGetAdjacency(dm, p, &adjSize, &adj);CHKERRQ(ierr);
589b0a623aaSMatthew G. Knepley         for (n = 0; n < neighbors; ++n) {
590b0a623aaSMatthew G. Knepley           const PetscInt remoteRank = nrank[noff+n];
591b0a623aaSMatthew G. Knepley 
592b0a623aaSMatthew G. Knepley           if (remoteRank == rank) continue;
593b0a623aaSMatthew G. Knepley           for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remoteRank);CHKERRQ(ierr);}
594b0a623aaSMatthew G. Knepley         }
595b0a623aaSMatthew G. Knepley       }
596b0a623aaSMatthew G. Knepley     }
597b0a623aaSMatthew G. Knepley     /* Roots are shared with leaves */
598b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetDof(rootSection, p, &neighbors);CHKERRQ(ierr);
599b0a623aaSMatthew G. Knepley     if (!neighbors) continue;
600b0a623aaSMatthew G. Knepley     ierr = PetscSectionGetOffset(rootSection, p, &noff);CHKERRQ(ierr);
601b0a623aaSMatthew G. Knepley     ierr = DMPlexGetAdjacency(dm, p, &adjSize, &adj);CHKERRQ(ierr);
602b0a623aaSMatthew G. Knepley     for (n = 0; n < neighbors; ++n) {
603b0a623aaSMatthew G. Knepley       const PetscInt remoteRank = rrank[noff+n];
604b0a623aaSMatthew G. Knepley 
605b0a623aaSMatthew G. Knepley       if (remoteRank == rank) continue;
606b0a623aaSMatthew G. Knepley       for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(ovAdjByRank, adj[a], remoteRank);CHKERRQ(ierr);}
607b0a623aaSMatthew G. Knepley     }
608b0a623aaSMatthew G. Knepley   }
609b0a623aaSMatthew G. Knepley   ierr = PetscFree(adj);CHKERRQ(ierr);
610b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(rootrank, &rrank);CHKERRQ(ierr);
611b0a623aaSMatthew G. Knepley   ierr = ISRestoreIndices(leafrank, &nrank);CHKERRQ(ierr);
61224d039d7SMichael Lange   /* Add additional overlap levels */
613*be200f8dSMichael Lange   for (l = 1; l < levels; l++) {
614*be200f8dSMichael Lange     /* Propagate point donations over SF to capture remote connections */
615*be200f8dSMichael Lange     ierr = DMPlexPartitionLabelPropagate(dm, ovAdjByRank);CHKERRQ(ierr);
616*be200f8dSMichael Lange     /* Add next level of point donations to the label */
617*be200f8dSMichael Lange     ierr = DMPlexPartitionLabelAdjacency(dm, ovAdjByRank);CHKERRQ(ierr);
618*be200f8dSMichael Lange   }
61926a7d390SMatthew G. Knepley   /* We require the closure in the overlap */
62026a7d390SMatthew G. Knepley   ierr = DMPlexGetAdjacencyUseCone(dm, &useCone);CHKERRQ(ierr);
62126a7d390SMatthew G. Knepley   ierr = DMPlexGetAdjacencyUseClosure(dm, &useClosure);CHKERRQ(ierr);
62226a7d390SMatthew G. Knepley   if (useCone || !useClosure) {
6235abbe4feSMichael Lange     ierr = DMPlexPartitionLabelClosure(dm, ovAdjByRank);CHKERRQ(ierr);
62426a7d390SMatthew G. Knepley   }
625c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-overlap_view", &flg);CHKERRQ(ierr);
626e540f424SMichael Lange   if (flg) {
627b0a623aaSMatthew G. Knepley     ierr = DMLabelView(ovAdjByRank, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
628b0a623aaSMatthew G. Knepley   }
6291fd9873aSMichael Lange   /* Make process SF and invert sender to receiver label */
630b0a623aaSMatthew G. Knepley   ierr = DMPlexCreateTwoSidedProcessSF(dm, sfPoint, rootSection, rootrank, leafSection, leafrank, NULL, &sfProc);CHKERRQ(ierr);
631a9f1d5b2SMichael Lange   ierr = DMLabelCreate("Overlap label", ovLabel);CHKERRQ(ierr);
632a9f1d5b2SMichael Lange   ierr = DMPlexPartitionLabelInvert(dm, ovAdjByRank, sfProc, *ovLabel);CHKERRQ(ierr);
633a9f1d5b2SMichael Lange   /* Add owned points, except for shared local points */
634a9f1d5b2SMichael Lange   for (p = pStart; p < pEnd; ++p) {ierr = DMLabelSetValue(*ovLabel, p, rank);CHKERRQ(ierr);}
635e540f424SMichael Lange   for (l = 0; l < nleaves; ++l) {
636a9f1d5b2SMichael Lange     ierr = DMLabelClearValue(*ovLabel, local[l], rank);CHKERRQ(ierr);
637a9f1d5b2SMichael Lange     ierr = DMLabelSetValue(*ovLabel, remote[l].index, remote[l].rank);CHKERRQ(ierr);
638e540f424SMichael Lange   }
639e540f424SMichael Lange   /* Clean up */
6401fd9873aSMichael Lange   ierr = DMLabelDestroy(&ovAdjByRank);CHKERRQ(ierr);
641b0a623aaSMatthew G. Knepley   ierr = PetscSFDestroy(&sfProc);CHKERRQ(ierr);
642b0a623aaSMatthew G. Knepley   PetscFunctionReturn(0);
643b0a623aaSMatthew G. Knepley }
64470034214SMatthew G. Knepley 
64570034214SMatthew G. Knepley #undef __FUNCT__
64646f9b1c3SMichael Lange #define __FUNCT__ "DMPlexCreateOverlapMigrationSF"
64746f9b1c3SMichael Lange PetscErrorCode DMPlexCreateOverlapMigrationSF(DM dm, PetscSF overlapSF, PetscSF *migrationSF)
64846f9b1c3SMichael Lange {
64946f9b1c3SMichael Lange   MPI_Comm           comm;
65046f9b1c3SMichael Lange   PetscMPIInt        rank, numProcs;
65146f9b1c3SMichael Lange   PetscInt           d, dim, p, pStart, pEnd, nroots, nleaves, newLeaves, point, numSharedPoints;
65246f9b1c3SMichael Lange   PetscInt          *pointDepths, *remoteDepths, *ilocal;
65346f9b1c3SMichael Lange   PetscInt          *depthRecv, *depthShift, *depthIdx;
65446f9b1c3SMichael Lange   PetscSFNode       *iremote;
65546f9b1c3SMichael Lange   PetscSF            pointSF;
65646f9b1c3SMichael Lange   const PetscInt    *sharedLocal;
65746f9b1c3SMichael Lange   const PetscSFNode *overlapRemote, *sharedRemote;
65846f9b1c3SMichael Lange   PetscErrorCode     ierr;
65946f9b1c3SMichael Lange 
66046f9b1c3SMichael Lange   PetscFunctionBegin;
66146f9b1c3SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
66246f9b1c3SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
66346f9b1c3SMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
66446f9b1c3SMichael Lange   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
66546f9b1c3SMichael Lange   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
66646f9b1c3SMichael Lange 
66746f9b1c3SMichael Lange   /* Before building the migration SF we need to know the new stratum offsets */
66846f9b1c3SMichael Lange   ierr = PetscSFGetGraph(overlapSF, &nroots, &nleaves, NULL, &overlapRemote);CHKERRQ(ierr);
66946f9b1c3SMichael Lange   ierr = PetscMalloc2(nroots, &pointDepths, nleaves, &remoteDepths);CHKERRQ(ierr);
67046f9b1c3SMichael Lange   for (d=0; d<dim+1; d++) {
67146f9b1c3SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
67246f9b1c3SMichael Lange     for (p=pStart; p<pEnd; p++) pointDepths[p] = d;
67346f9b1c3SMichael Lange   }
67446f9b1c3SMichael Lange   for (p=0; p<nleaves; p++) remoteDepths[p] = -1;
67546f9b1c3SMichael Lange   ierr = PetscSFBcastBegin(overlapSF, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr);
67646f9b1c3SMichael Lange   ierr = PetscSFBcastEnd(overlapSF, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr);
67746f9b1c3SMichael Lange 
67846f9b1c3SMichael Lange   /* Count recevied points in each stratum and compute the internal strata shift */
67946f9b1c3SMichael Lange   ierr = PetscMalloc3(dim+1, &depthRecv, dim+1, &depthShift, dim+1, &depthIdx);CHKERRQ(ierr);
68046f9b1c3SMichael Lange   for (d=0; d<dim+1; d++) depthRecv[d]=0;
68146f9b1c3SMichael Lange   for (p=0; p<nleaves; p++) depthRecv[remoteDepths[p]]++;
68246f9b1c3SMichael Lange   depthShift[dim] = 0;
68346f9b1c3SMichael Lange   for (d=0; d<dim; d++) depthShift[d] = depthRecv[dim];
68446f9b1c3SMichael Lange   for (d=1; d<dim; d++) depthShift[d] += depthRecv[0];
68546f9b1c3SMichael Lange   for (d=dim-2; d>0; d--) depthShift[d] += depthRecv[d+1];
68646f9b1c3SMichael Lange   for (d=0; d<dim+1; d++) {
68746f9b1c3SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
68846f9b1c3SMichael Lange     depthIdx[d] = pStart + depthShift[d];
68946f9b1c3SMichael Lange   }
69046f9b1c3SMichael Lange 
69146f9b1c3SMichael Lange   /* Form the overlap SF build an SF that describes the full overlap migration SF */
69246f9b1c3SMichael Lange   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
69346f9b1c3SMichael Lange   newLeaves = pEnd - pStart + nleaves;
69409b7985cSMatthew G. Knepley   ierr = PetscMalloc1(newLeaves, &ilocal);CHKERRQ(ierr);
69509b7985cSMatthew G. Knepley   ierr = PetscMalloc1(newLeaves, &iremote);CHKERRQ(ierr);
69646f9b1c3SMichael Lange   /* First map local points to themselves */
69746f9b1c3SMichael Lange   for (d=0; d<dim+1; d++) {
69846f9b1c3SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
69946f9b1c3SMichael Lange     for (p=pStart; p<pEnd; p++) {
70046f9b1c3SMichael Lange       point = p + depthShift[d];
70146f9b1c3SMichael Lange       ilocal[point] = point;
70246f9b1c3SMichael Lange       iremote[point].index = p;
70346f9b1c3SMichael Lange       iremote[point].rank = rank;
70446f9b1c3SMichael Lange       depthIdx[d]++;
70546f9b1c3SMichael Lange     }
70646f9b1c3SMichael Lange   }
70746f9b1c3SMichael Lange 
70846f9b1c3SMichael Lange   /* Add in the remote roots for currently shared points */
70946f9b1c3SMichael Lange   ierr = DMGetPointSF(dm, &pointSF);CHKERRQ(ierr);
71046f9b1c3SMichael Lange   ierr = PetscSFGetGraph(pointSF, NULL, &numSharedPoints, &sharedLocal, &sharedRemote);CHKERRQ(ierr);
71146f9b1c3SMichael Lange   for (d=0; d<dim+1; d++) {
71246f9b1c3SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
71346f9b1c3SMichael Lange     for (p=0; p<numSharedPoints; p++) {
71446f9b1c3SMichael Lange       if (pStart <= sharedLocal[p] && sharedLocal[p] < pEnd) {
71546f9b1c3SMichael Lange         point = sharedLocal[p] + depthShift[d];
71646f9b1c3SMichael Lange         iremote[point].index = sharedRemote[p].index;
71746f9b1c3SMichael Lange         iremote[point].rank = sharedRemote[p].rank;
71846f9b1c3SMichael Lange       }
71946f9b1c3SMichael Lange     }
72046f9b1c3SMichael Lange   }
72146f9b1c3SMichael Lange 
72246f9b1c3SMichael Lange   /* Now add the incoming overlap points */
72346f9b1c3SMichael Lange   for (p=0; p<nleaves; p++) {
72446f9b1c3SMichael Lange     point = depthIdx[remoteDepths[p]];
72546f9b1c3SMichael Lange     ilocal[point] = point;
72646f9b1c3SMichael Lange     iremote[point].index = overlapRemote[p].index;
72746f9b1c3SMichael Lange     iremote[point].rank = overlapRemote[p].rank;
72846f9b1c3SMichael Lange     depthIdx[remoteDepths[p]]++;
72946f9b1c3SMichael Lange   }
73015fff7beSMatthew G. Knepley   ierr = PetscFree2(pointDepths,remoteDepths);CHKERRQ(ierr);
73146f9b1c3SMichael Lange 
73246f9b1c3SMichael Lange   ierr = PetscSFCreate(comm, migrationSF);CHKERRQ(ierr);
73346f9b1c3SMichael Lange   ierr = PetscObjectSetName((PetscObject) *migrationSF, "Overlap Migration SF");CHKERRQ(ierr);
73446f9b1c3SMichael Lange   ierr = PetscSFSetFromOptions(*migrationSF);CHKERRQ(ierr);
73546f9b1c3SMichael Lange   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
73646f9b1c3SMichael Lange   ierr = PetscSFSetGraph(*migrationSF, pEnd-pStart, newLeaves, ilocal, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER);CHKERRQ(ierr);
73746f9b1c3SMichael Lange 
73846f9b1c3SMichael Lange   ierr = PetscFree3(depthRecv, depthShift, depthIdx);CHKERRQ(ierr);
73970034214SMatthew G. Knepley   PetscFunctionReturn(0);
74070034214SMatthew G. Knepley }
74170034214SMatthew G. Knepley 
74270034214SMatthew G. Knepley #undef __FUNCT__
743a9f1d5b2SMichael Lange #define __FUNCT__ "DMPlexStratifyMigrationSF"
744a9f1d5b2SMichael Lange /*@
745a9f1d5b2SMichael Lange   DMPlexStratifyMigrationSF - Add partition overlap to a distributed non-overlapping DM.
746a9f1d5b2SMichael Lange 
747a9f1d5b2SMichael Lange   Input Parameter:
748a9f1d5b2SMichael Lange + dm          - The DM
749a9f1d5b2SMichael Lange - sf          - A star forest with non-ordered leaves, usually defining a DM point migration
750a9f1d5b2SMichael Lange 
751a9f1d5b2SMichael Lange   Output Parameter:
752a9f1d5b2SMichael Lange . migrationSF - A star forest with added leaf indirection that ensures the resulting DM is stratified
753a9f1d5b2SMichael Lange 
754a9f1d5b2SMichael Lange   Level: developer
755a9f1d5b2SMichael Lange 
756a9f1d5b2SMichael Lange .seealso: DMPlexPartitionLabelCreateSF(), DMPlexDistribute(), DMPlexDistributeOverlap()
757a9f1d5b2SMichael Lange @*/
758a9f1d5b2SMichael Lange PetscErrorCode DMPlexStratifyMigrationSF(DM dm, PetscSF sf, PetscSF *migrationSF)
759a9f1d5b2SMichael Lange {
760a9f1d5b2SMichael Lange   MPI_Comm           comm;
761a9f1d5b2SMichael Lange   PetscMPIInt        rank, numProcs;
7627fab53ddSMatthew G. Knepley   PetscInt           d, ldepth, depth, p, pStart, pEnd, nroots, nleaves;
763a9f1d5b2SMichael Lange   PetscInt          *pointDepths, *remoteDepths, *ilocal;
764a9f1d5b2SMichael Lange   PetscInt          *depthRecv, *depthShift, *depthIdx;
765a9f1d5b2SMichael Lange   const PetscSFNode *iremote;
766a9f1d5b2SMichael Lange   PetscErrorCode     ierr;
767a9f1d5b2SMichael Lange 
768a9f1d5b2SMichael Lange   PetscFunctionBegin;
769a9f1d5b2SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
770a9f1d5b2SMichael Lange   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
771a9f1d5b2SMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
772a9f1d5b2SMichael Lange   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
7737fab53ddSMatthew G. Knepley   ierr = DMPlexGetDepth(dm, &ldepth);CHKERRQ(ierr);
774b2566f29SBarry Smith   ierr = MPIU_Allreduce(&ldepth, &depth, 1, MPIU_INT, MPI_MAX, comm);CHKERRQ(ierr);
7757fab53ddSMatthew G. Knepley   if ((ldepth >= 0) && (depth != ldepth)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Inconsistent Plex depth %d != %d", ldepth, depth);
776a9f1d5b2SMichael Lange 
777a9f1d5b2SMichael Lange   /* Before building the migration SF we need to know the new stratum offsets */
778a9f1d5b2SMichael Lange   ierr = PetscSFGetGraph(sf, &nroots, &nleaves, NULL, &iremote);CHKERRQ(ierr);
779a9f1d5b2SMichael Lange   ierr = PetscMalloc2(nroots, &pointDepths, nleaves, &remoteDepths);CHKERRQ(ierr);
7807fab53ddSMatthew G. Knepley   for (d = 0; d < depth+1; ++d) {
781a9f1d5b2SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr);
7827fab53ddSMatthew G. Knepley     for (p = pStart; p < pEnd; ++p) pointDepths[p] = d;
783a9f1d5b2SMichael Lange   }
7847fab53ddSMatthew G. Knepley   for (p = 0; p < nleaves; ++p) remoteDepths[p] = -1;
785a9f1d5b2SMichael Lange   ierr = PetscSFBcastBegin(sf, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr);
786a9f1d5b2SMichael Lange   ierr = PetscSFBcastEnd(sf, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr);
787a9f1d5b2SMichael Lange   /* Count recevied points in each stratum and compute the internal strata shift */
7887fab53ddSMatthew G. Knepley   ierr = PetscMalloc3(depth+1, &depthRecv, depth+1, &depthShift, depth+1, &depthIdx);CHKERRQ(ierr);
7897fab53ddSMatthew G. Knepley   for (d = 0; d < depth+1; ++d) depthRecv[d] = 0;
7907fab53ddSMatthew G. Knepley   for (p = 0; p < nleaves; ++p) depthRecv[remoteDepths[p]]++;
7917fab53ddSMatthew G. Knepley   depthShift[depth] = 0;
7927fab53ddSMatthew G. Knepley   for (d = 0; d < depth; ++d) depthShift[d] = depthRecv[depth];
7937fab53ddSMatthew G. Knepley   for (d = 1; d < depth; ++d) depthShift[d] += depthRecv[0];
7947fab53ddSMatthew G. Knepley   for (d = depth-2; d > 0; --d) depthShift[d] += depthRecv[d+1];
7957fab53ddSMatthew G. Knepley   for (d = 0; d < depth+1; ++d) {depthIdx[d] = 0;}
796a9f1d5b2SMichael Lange   /* Derive a new local permutation based on stratified indices */
797a9f1d5b2SMichael Lange   ierr = PetscMalloc1(nleaves, &ilocal);CHKERRQ(ierr);
7987fab53ddSMatthew G. Knepley   for (p = 0; p < nleaves; ++p) {
7997fab53ddSMatthew G. Knepley     const PetscInt dep = remoteDepths[p];
8007fab53ddSMatthew G. Knepley 
8017fab53ddSMatthew G. Knepley     ilocal[p] = depthShift[dep] + depthIdx[dep];
8027fab53ddSMatthew G. Knepley     depthIdx[dep]++;
803a9f1d5b2SMichael Lange   }
804a9f1d5b2SMichael Lange   ierr = PetscSFCreate(comm, migrationSF);CHKERRQ(ierr);
805a9f1d5b2SMichael Lange   ierr = PetscObjectSetName((PetscObject) *migrationSF, "Migration SF");CHKERRQ(ierr);
806a9f1d5b2SMichael Lange   ierr = PetscSFSetGraph(*migrationSF, nroots, nleaves, ilocal, PETSC_OWN_POINTER, iremote, PETSC_COPY_VALUES);CHKERRQ(ierr);
807a9f1d5b2SMichael Lange   ierr = PetscFree2(pointDepths,remoteDepths);CHKERRQ(ierr);
808a9f1d5b2SMichael Lange   ierr = PetscFree3(depthRecv, depthShift, depthIdx);CHKERRQ(ierr);
809a9f1d5b2SMichael Lange   PetscFunctionReturn(0);
810a9f1d5b2SMichael Lange }
811a9f1d5b2SMichael Lange 
812a9f1d5b2SMichael Lange #undef __FUNCT__
81370034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeField"
81470034214SMatthew G. Knepley /*@
81570034214SMatthew G. Knepley   DMPlexDistributeField - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
81670034214SMatthew G. Knepley 
81770034214SMatthew G. Knepley   Collective on DM
81870034214SMatthew G. Knepley 
81970034214SMatthew G. Knepley   Input Parameters:
82070034214SMatthew G. Knepley + dm - The DMPlex object
82170034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
82270034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout
82370034214SMatthew G. Knepley - originalVec - The existing data
82470034214SMatthew G. Knepley 
82570034214SMatthew G. Knepley   Output Parameters:
82670034214SMatthew G. Knepley + newSection - The PetscSF describing the new data layout
82770034214SMatthew G. Knepley - newVec - The new data
82870034214SMatthew G. Knepley 
82970034214SMatthew G. Knepley   Level: developer
83070034214SMatthew G. Knepley 
83170034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeFieldIS(), DMPlexDistributeData()
83270034214SMatthew G. Knepley @*/
83370034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeField(DM dm, PetscSF pointSF, PetscSection originalSection, Vec originalVec, PetscSection newSection, Vec newVec)
83470034214SMatthew G. Knepley {
83570034214SMatthew G. Knepley   PetscSF        fieldSF;
83670034214SMatthew G. Knepley   PetscInt      *remoteOffsets, fieldSize;
83770034214SMatthew G. Knepley   PetscScalar   *originalValues, *newValues;
83870034214SMatthew G. Knepley   PetscErrorCode ierr;
83970034214SMatthew G. Knepley 
84070034214SMatthew G. Knepley   PetscFunctionBegin;
84170034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
84270034214SMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
84370034214SMatthew G. Knepley 
84470034214SMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
84570034214SMatthew G. Knepley   ierr = VecSetSizes(newVec, fieldSize, PETSC_DETERMINE);CHKERRQ(ierr);
84670034214SMatthew G. Knepley   ierr = VecSetType(newVec,dm->vectype);CHKERRQ(ierr);
84770034214SMatthew G. Knepley 
84870034214SMatthew G. Knepley   ierr = VecGetArray(originalVec, &originalValues);CHKERRQ(ierr);
84970034214SMatthew G. Knepley   ierr = VecGetArray(newVec, &newValues);CHKERRQ(ierr);
85070034214SMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
8518a713f3bSLawrence Mitchell   ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
85270034214SMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr);
85370034214SMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr);
85470034214SMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
85570034214SMatthew G. Knepley   ierr = VecRestoreArray(newVec, &newValues);CHKERRQ(ierr);
85670034214SMatthew G. Knepley   ierr = VecRestoreArray(originalVec, &originalValues);CHKERRQ(ierr);
85770034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
85870034214SMatthew G. Knepley   PetscFunctionReturn(0);
85970034214SMatthew G. Knepley }
86070034214SMatthew G. Knepley 
86170034214SMatthew G. Knepley #undef __FUNCT__
86270034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeFieldIS"
86370034214SMatthew G. Knepley /*@
86470034214SMatthew G. Knepley   DMPlexDistributeFieldIS - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
86570034214SMatthew G. Knepley 
86670034214SMatthew G. Knepley   Collective on DM
86770034214SMatthew G. Knepley 
86870034214SMatthew G. Knepley   Input Parameters:
86970034214SMatthew G. Knepley + dm - The DMPlex object
87070034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
87170034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout
87270034214SMatthew G. Knepley - originalIS - The existing data
87370034214SMatthew G. Knepley 
87470034214SMatthew G. Knepley   Output Parameters:
87570034214SMatthew G. Knepley + newSection - The PetscSF describing the new data layout
87670034214SMatthew G. Knepley - newIS - The new data
87770034214SMatthew G. Knepley 
87870034214SMatthew G. Knepley   Level: developer
87970034214SMatthew G. Knepley 
88070034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField(), DMPlexDistributeData()
88170034214SMatthew G. Knepley @*/
88270034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeFieldIS(DM dm, PetscSF pointSF, PetscSection originalSection, IS originalIS, PetscSection newSection, IS *newIS)
88370034214SMatthew G. Knepley {
88470034214SMatthew G. Knepley   PetscSF         fieldSF;
88570034214SMatthew G. Knepley   PetscInt       *newValues, *remoteOffsets, fieldSize;
88670034214SMatthew G. Knepley   const PetscInt *originalValues;
88770034214SMatthew G. Knepley   PetscErrorCode  ierr;
88870034214SMatthew G. Knepley 
88970034214SMatthew G. Knepley   PetscFunctionBegin;
89070034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
89170034214SMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
89270034214SMatthew G. Knepley 
89370034214SMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
894854ce69bSBarry Smith   ierr = PetscMalloc1(fieldSize, &newValues);CHKERRQ(ierr);
89570034214SMatthew G. Knepley 
89670034214SMatthew G. Knepley   ierr = ISGetIndices(originalIS, &originalValues);CHKERRQ(ierr);
89770034214SMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
8988a713f3bSLawrence Mitchell   ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
899aaf8c182SMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, MPIU_INT, (PetscInt *) originalValues, newValues);CHKERRQ(ierr);
900aaf8c182SMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, MPIU_INT, (PetscInt *) originalValues, newValues);CHKERRQ(ierr);
90170034214SMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
90270034214SMatthew G. Knepley   ierr = ISRestoreIndices(originalIS, &originalValues);CHKERRQ(ierr);
90370034214SMatthew G. Knepley   ierr = ISCreateGeneral(PetscObjectComm((PetscObject) pointSF), fieldSize, newValues, PETSC_OWN_POINTER, newIS);CHKERRQ(ierr);
90470034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr);
90570034214SMatthew G. Knepley   PetscFunctionReturn(0);
90670034214SMatthew G. Knepley }
90770034214SMatthew G. Knepley 
90870034214SMatthew G. Knepley #undef __FUNCT__
90970034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeData"
91070034214SMatthew G. Knepley /*@
91170034214SMatthew G. Knepley   DMPlexDistributeData - Distribute field data to match a given PetscSF, usually the SF from mesh distribution
91270034214SMatthew G. Knepley 
91370034214SMatthew G. Knepley   Collective on DM
91470034214SMatthew G. Knepley 
91570034214SMatthew G. Knepley   Input Parameters:
91670034214SMatthew G. Knepley + dm - The DMPlex object
91770034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern
91870034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout
91970034214SMatthew G. Knepley . datatype - The type of data
92070034214SMatthew G. Knepley - originalData - The existing data
92170034214SMatthew G. Knepley 
92270034214SMatthew G. Knepley   Output Parameters:
92370034214SMatthew G. Knepley + newSection - The PetscSection describing the new data layout
92470034214SMatthew G. Knepley - newData - The new data
92570034214SMatthew G. Knepley 
92670034214SMatthew G. Knepley   Level: developer
92770034214SMatthew G. Knepley 
92870034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField()
92970034214SMatthew G. Knepley @*/
93070034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeData(DM dm, PetscSF pointSF, PetscSection originalSection, MPI_Datatype datatype, void *originalData, PetscSection newSection, void **newData)
93170034214SMatthew G. Knepley {
93270034214SMatthew G. Knepley   PetscSF        fieldSF;
93370034214SMatthew G. Knepley   PetscInt      *remoteOffsets, fieldSize;
93470034214SMatthew G. Knepley   PetscMPIInt    dataSize;
93570034214SMatthew G. Knepley   PetscErrorCode ierr;
93670034214SMatthew G. Knepley 
93770034214SMatthew G. Knepley   PetscFunctionBegin;
93870034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr);
93970034214SMatthew G. Knepley   ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr);
94070034214SMatthew G. Knepley 
94170034214SMatthew G. Knepley   ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr);
94270034214SMatthew G. Knepley   ierr = MPI_Type_size(datatype, &dataSize);CHKERRQ(ierr);
94370034214SMatthew G. Knepley   ierr = PetscMalloc(fieldSize * dataSize, newData);CHKERRQ(ierr);
94470034214SMatthew G. Knepley 
94570034214SMatthew G. Knepley   ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr);
9468a713f3bSLawrence Mitchell   ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
94770034214SMatthew G. Knepley   ierr = PetscSFBcastBegin(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr);
94870034214SMatthew G. Knepley   ierr = PetscSFBcastEnd(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr);
94970034214SMatthew G. Knepley   ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr);
95070034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr);
95170034214SMatthew G. Knepley   PetscFunctionReturn(0);
95270034214SMatthew G. Knepley }
95370034214SMatthew G. Knepley 
95470034214SMatthew G. Knepley #undef __FUNCT__
955cc71bff1SMichael Lange #define __FUNCT__ "DMPlexDistributeCones"
956ac04eaf7SToby Isaac PetscErrorCode DMPlexDistributeCones(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping original, ISLocalToGlobalMapping renumbering, DM dmParallel)
957cc71bff1SMichael Lange {
958f5bf2dbfSMichael Lange   DM_Plex               *mesh  = (DM_Plex*) dm->data;
959cc71bff1SMichael Lange   DM_Plex               *pmesh = (DM_Plex*) (dmParallel)->data;
960cc71bff1SMichael Lange   MPI_Comm               comm;
961cc71bff1SMichael Lange   PetscSF                coneSF;
962cc71bff1SMichael Lange   PetscSection           originalConeSection, newConeSection;
963ac04eaf7SToby Isaac   PetscInt              *remoteOffsets, *cones, *globCones, *newCones, newConesSize;
964cc71bff1SMichael Lange   PetscBool              flg;
965cc71bff1SMichael Lange   PetscErrorCode         ierr;
966cc71bff1SMichael Lange 
967cc71bff1SMichael Lange   PetscFunctionBegin;
968cc71bff1SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
969cc71bff1SMichael Lange   PetscValidPointer(dmParallel,4);
970cc71bff1SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr);
971cc71bff1SMichael Lange 
972cc71bff1SMichael Lange   /* Distribute cone section */
973cc71bff1SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
974cc71bff1SMichael Lange   ierr = DMPlexGetConeSection(dm, &originalConeSection);CHKERRQ(ierr);
975cc71bff1SMichael Lange   ierr = DMPlexGetConeSection(dmParallel, &newConeSection);CHKERRQ(ierr);
976cc71bff1SMichael Lange   ierr = PetscSFDistributeSection(migrationSF, originalConeSection, &remoteOffsets, newConeSection);CHKERRQ(ierr);
977cc71bff1SMichael Lange   ierr = DMSetUp(dmParallel);CHKERRQ(ierr);
978cc71bff1SMichael Lange   {
979cc71bff1SMichael Lange     PetscInt pStart, pEnd, p;
980cc71bff1SMichael Lange 
981cc71bff1SMichael Lange     ierr = PetscSectionGetChart(newConeSection, &pStart, &pEnd);CHKERRQ(ierr);
982cc71bff1SMichael Lange     for (p = pStart; p < pEnd; ++p) {
983cc71bff1SMichael Lange       PetscInt coneSize;
984cc71bff1SMichael Lange       ierr               = PetscSectionGetDof(newConeSection, p, &coneSize);CHKERRQ(ierr);
985cc71bff1SMichael Lange       pmesh->maxConeSize = PetscMax(pmesh->maxConeSize, coneSize);
986cc71bff1SMichael Lange     }
987cc71bff1SMichael Lange   }
988cc71bff1SMichael Lange   /* Communicate and renumber cones */
989cc71bff1SMichael Lange   ierr = PetscSFCreateSectionSF(migrationSF, originalConeSection, remoteOffsets, newConeSection, &coneSF);CHKERRQ(ierr);
9908a713f3bSLawrence Mitchell   ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
991cc71bff1SMichael Lange   ierr = DMPlexGetCones(dm, &cones);CHKERRQ(ierr);
992ac04eaf7SToby Isaac   if (original) {
993ac04eaf7SToby Isaac     PetscInt numCones;
994ac04eaf7SToby Isaac 
995ac04eaf7SToby Isaac     ierr = PetscSectionGetStorageSize(originalConeSection,&numCones);CHKERRQ(ierr); ierr = PetscMalloc1(numCones,&globCones);CHKERRQ(ierr);
996ac04eaf7SToby Isaac     ierr = ISLocalToGlobalMappingApplyBlock(original, numCones, cones, globCones);CHKERRQ(ierr);
997ac04eaf7SToby Isaac   }
998ac04eaf7SToby Isaac   else {
999ac04eaf7SToby Isaac     globCones = cones;
1000ac04eaf7SToby Isaac   }
1001cc71bff1SMichael Lange   ierr = DMPlexGetCones(dmParallel, &newCones);CHKERRQ(ierr);
1002ac04eaf7SToby Isaac   ierr = PetscSFBcastBegin(coneSF, MPIU_INT, globCones, newCones);CHKERRQ(ierr);
1003ac04eaf7SToby Isaac   ierr = PetscSFBcastEnd(coneSF, MPIU_INT, globCones, newCones);CHKERRQ(ierr);
1004ac04eaf7SToby Isaac   if (original) {
1005ac04eaf7SToby Isaac     ierr = PetscFree(globCones);CHKERRQ(ierr);
1006ac04eaf7SToby Isaac   }
1007cc71bff1SMichael Lange   ierr = PetscSectionGetStorageSize(newConeSection, &newConesSize);CHKERRQ(ierr);
1008cc71bff1SMichael Lange   ierr = ISGlobalToLocalMappingApplyBlock(renumbering, IS_GTOLM_MASK, newConesSize, newCones, NULL, newCones);CHKERRQ(ierr);
10093533c52bSMatthew G. Knepley #if PETSC_USE_DEBUG
10103533c52bSMatthew G. Knepley   {
10113533c52bSMatthew G. Knepley     PetscInt  p;
10123533c52bSMatthew G. Knepley     PetscBool valid = PETSC_TRUE;
10133533c52bSMatthew G. Knepley     for (p = 0; p < newConesSize; ++p) {
10143533c52bSMatthew G. Knepley       if (newCones[p] < 0) {valid = PETSC_FALSE; ierr = PetscPrintf(PETSC_COMM_SELF, "Point %d not in overlap SF\n", p);CHKERRQ(ierr);}
10153533c52bSMatthew G. Knepley     }
10163533c52bSMatthew G. Knepley     if (!valid) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid global to local map");
10173533c52bSMatthew G. Knepley   }
10183533c52bSMatthew G. Knepley #endif
1019c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-cones_view", &flg);CHKERRQ(ierr);
1020cc71bff1SMichael Lange   if (flg) {
1021cc71bff1SMichael Lange     ierr = PetscPrintf(comm, "Serial Cone Section:\n");CHKERRQ(ierr);
1022cc71bff1SMichael Lange     ierr = PetscSectionView(originalConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1023cc71bff1SMichael Lange     ierr = PetscPrintf(comm, "Parallel Cone Section:\n");CHKERRQ(ierr);
1024cc71bff1SMichael Lange     ierr = PetscSectionView(newConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1025cc71bff1SMichael Lange     ierr = PetscSFView(coneSF, NULL);CHKERRQ(ierr);
1026cc71bff1SMichael Lange   }
1027cc71bff1SMichael Lange   ierr = DMPlexGetConeOrientations(dm, &cones);CHKERRQ(ierr);
1028cc71bff1SMichael Lange   ierr = DMPlexGetConeOrientations(dmParallel, &newCones);CHKERRQ(ierr);
1029cc71bff1SMichael Lange   ierr = PetscSFBcastBegin(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
1030cc71bff1SMichael Lange   ierr = PetscSFBcastEnd(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr);
1031cc71bff1SMichael Lange   ierr = PetscSFDestroy(&coneSF);CHKERRQ(ierr);
1032cc71bff1SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr);
1033cc71bff1SMichael Lange   /* Create supports and stratify sieve */
1034cc71bff1SMichael Lange   {
1035cc71bff1SMichael Lange     PetscInt pStart, pEnd;
1036cc71bff1SMichael Lange 
1037cc71bff1SMichael Lange     ierr = PetscSectionGetChart(pmesh->coneSection, &pStart, &pEnd);CHKERRQ(ierr);
1038cc71bff1SMichael Lange     ierr = PetscSectionSetChart(pmesh->supportSection, pStart, pEnd);CHKERRQ(ierr);
1039cc71bff1SMichael Lange   }
1040cc71bff1SMichael Lange   ierr = DMPlexSymmetrize(dmParallel);CHKERRQ(ierr);
1041cc71bff1SMichael Lange   ierr = DMPlexStratify(dmParallel);CHKERRQ(ierr);
1042f5bf2dbfSMichael Lange   pmesh->useCone    = mesh->useCone;
1043f5bf2dbfSMichael Lange   pmesh->useClosure = mesh->useClosure;
1044cc71bff1SMichael Lange   PetscFunctionReturn(0);
1045cc71bff1SMichael Lange }
1046cc71bff1SMichael Lange 
1047cc71bff1SMichael Lange #undef __FUNCT__
10480df0e737SMichael Lange #define __FUNCT__ "DMPlexDistributeCoordinates"
10490df0e737SMichael Lange PetscErrorCode DMPlexDistributeCoordinates(DM dm, PetscSF migrationSF, DM dmParallel)
10500df0e737SMichael Lange {
10510df0e737SMichael Lange   MPI_Comm         comm;
10520df0e737SMichael Lange   PetscSection     originalCoordSection, newCoordSection;
10530df0e737SMichael Lange   Vec              originalCoordinates, newCoordinates;
10540df0e737SMichael Lange   PetscInt         bs;
10550df0e737SMichael Lange   const char      *name;
10560df0e737SMichael Lange   const PetscReal *maxCell, *L;
10575dc8c3f7SMatthew G. Knepley   const DMBoundaryType *bd;
10580df0e737SMichael Lange   PetscErrorCode   ierr;
10590df0e737SMichael Lange 
10600df0e737SMichael Lange   PetscFunctionBegin;
10610df0e737SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10620df0e737SMichael Lange   PetscValidPointer(dmParallel, 3);
10630df0e737SMichael Lange 
10640df0e737SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
10650df0e737SMichael Lange   ierr = DMGetCoordinateSection(dm, &originalCoordSection);CHKERRQ(ierr);
10660df0e737SMichael Lange   ierr = DMGetCoordinateSection(dmParallel, &newCoordSection);CHKERRQ(ierr);
10670df0e737SMichael Lange   ierr = DMGetCoordinatesLocal(dm, &originalCoordinates);CHKERRQ(ierr);
10680df0e737SMichael Lange   if (originalCoordinates) {
10690df0e737SMichael Lange     ierr = VecCreate(comm, &newCoordinates);CHKERRQ(ierr);
10700df0e737SMichael Lange     ierr = PetscObjectGetName((PetscObject) originalCoordinates, &name);CHKERRQ(ierr);
10710df0e737SMichael Lange     ierr = PetscObjectSetName((PetscObject) newCoordinates, name);CHKERRQ(ierr);
10720df0e737SMichael Lange 
10730df0e737SMichael Lange     ierr = DMPlexDistributeField(dm, migrationSF, originalCoordSection, originalCoordinates, newCoordSection, newCoordinates);CHKERRQ(ierr);
10740df0e737SMichael Lange     ierr = DMSetCoordinatesLocal(dmParallel, newCoordinates);CHKERRQ(ierr);
10750df0e737SMichael Lange     ierr = VecGetBlockSize(originalCoordinates, &bs);CHKERRQ(ierr);
10760df0e737SMichael Lange     ierr = VecSetBlockSize(newCoordinates, bs);CHKERRQ(ierr);
10770df0e737SMichael Lange     ierr = VecDestroy(&newCoordinates);CHKERRQ(ierr);
10780df0e737SMichael Lange   }
10795dc8c3f7SMatthew G. Knepley   ierr = DMGetPeriodicity(dm, &maxCell, &L, &bd);CHKERRQ(ierr);
10805dc8c3f7SMatthew G. Knepley   if (L) {ierr = DMSetPeriodicity(dmParallel, maxCell, L, bd);CHKERRQ(ierr);}
10810df0e737SMichael Lange   PetscFunctionReturn(0);
10820df0e737SMichael Lange }
10830df0e737SMichael Lange 
10840df0e737SMichael Lange #undef __FUNCT__
10850df0e737SMichael Lange #define __FUNCT__ "DMPlexDistributeLabels"
1086d995df53SMatthew G. Knepley /* Here we are assuming that process 0 always has everything */
10872eb1fa14SMichael Lange PetscErrorCode DMPlexDistributeLabels(DM dm, PetscSF migrationSF, DM dmParallel)
10880df0e737SMichael Lange {
1089df0420ecSMatthew G. Knepley   DM_Plex         *mesh = (DM_Plex*) dm->data;
10900df0e737SMichael Lange   MPI_Comm         comm;
1091df0420ecSMatthew G. Knepley   DMLabel          depth;
10920df0e737SMichael Lange   PetscMPIInt      rank;
1093d995df53SMatthew G. Knepley   PetscInt         numLabels, numLocalLabels, l;
1094df0420ecSMatthew G. Knepley   PetscBool        hasLabels = PETSC_FALSE, lsendDepth, sendDepth;
1095df0420ecSMatthew G. Knepley   PetscObjectState depthState = -1;
10960df0e737SMichael Lange   PetscErrorCode   ierr;
10970df0e737SMichael Lange 
10980df0e737SMichael Lange   PetscFunctionBegin;
10990df0e737SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
11002eb1fa14SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 3);
11010df0e737SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr);
11020df0e737SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
11030df0e737SMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
11040df0e737SMichael Lange 
1105df0420ecSMatthew G. Knepley   /* If the user has changed the depth label, communicate it instead */
1106df0420ecSMatthew G. Knepley   ierr = DMPlexGetDepthLabel(dm, &depth);CHKERRQ(ierr);
1107df0420ecSMatthew G. Knepley   if (depth) {ierr = DMLabelGetState(depth, &depthState);CHKERRQ(ierr);}
1108df0420ecSMatthew G. Knepley   lsendDepth = mesh->depthState != depthState ? PETSC_TRUE : PETSC_FALSE;
1109b2566f29SBarry Smith   ierr = MPIU_Allreduce(&lsendDepth, &sendDepth, 1, MPIU_BOOL, MPI_LOR, comm);CHKERRQ(ierr);
1110df0420ecSMatthew G. Knepley   if (sendDepth) {
1111c58f1c22SToby Isaac     ierr = DMRemoveLabel(dmParallel, "depth", &depth);CHKERRQ(ierr);
1112df0420ecSMatthew G. Knepley     ierr = DMLabelDestroy(&depth);CHKERRQ(ierr);
1113df0420ecSMatthew G. Knepley   }
1114d995df53SMatthew G. Knepley   /* Everyone must have either the same number of labels, or none */
1115c58f1c22SToby Isaac   ierr = DMGetNumLabels(dm, &numLocalLabels);CHKERRQ(ierr);
1116d995df53SMatthew G. Knepley   numLabels = numLocalLabels;
11170df0e737SMichael Lange   ierr = MPI_Bcast(&numLabels, 1, MPIU_INT, 0, comm);CHKERRQ(ierr);
1118627847f0SMatthew G. Knepley   if (numLabels == numLocalLabels) hasLabels = PETSC_TRUE;
1119bdd2d751SMichael Lange   for (l = numLabels-1; l >= 0; --l) {
1120bdd2d751SMichael Lange     DMLabel     label = NULL, labelNew = NULL;
11210df0e737SMichael Lange     PetscBool   isdepth;
11220df0e737SMichael Lange 
1123d995df53SMatthew G. Knepley     if (hasLabels) {
1124c58f1c22SToby Isaac       ierr = DMGetLabelByNum(dm, l, &label);CHKERRQ(ierr);
11250df0e737SMichael Lange       /* Skip "depth" because it is recreated */
1126bdd2d751SMichael Lange       ierr = PetscStrcmp(label->name, "depth", &isdepth);CHKERRQ(ierr);
1127bdd2d751SMichael Lange     }
11280df0e737SMichael Lange     ierr = MPI_Bcast(&isdepth, 1, MPIU_BOOL, 0, comm);CHKERRQ(ierr);
1129df0420ecSMatthew G. Knepley     if (isdepth && !sendDepth) continue;
1130bdd2d751SMichael Lange     ierr = DMLabelDistribute(label, migrationSF, &labelNew);CHKERRQ(ierr);
1131c58f1c22SToby Isaac     ierr = DMAddLabel(dmParallel, labelNew);CHKERRQ(ierr);
11320df0e737SMichael Lange   }
11330df0e737SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr);
11340df0e737SMichael Lange   PetscFunctionReturn(0);
11350df0e737SMichael Lange }
11360df0e737SMichael Lange 
11379734c634SMichael Lange #undef __FUNCT__
11389734c634SMichael Lange #define __FUNCT__ "DMPlexDistributeSetupHybrid"
11399734c634SMichael Lange PetscErrorCode DMPlexDistributeSetupHybrid(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping renumbering, DM dmParallel)
11409734c634SMichael Lange {
11419734c634SMichael Lange   DM_Plex        *mesh  = (DM_Plex*) dm->data;
11429734c634SMichael Lange   DM_Plex        *pmesh = (DM_Plex*) (dmParallel)->data;
11439734c634SMichael Lange   MPI_Comm        comm;
11449734c634SMichael Lange   const PetscInt *gpoints;
11459734c634SMichael Lange   PetscInt        dim, depth, n, d;
11469734c634SMichael Lange   PetscErrorCode  ierr;
11479734c634SMichael Lange 
11489734c634SMichael Lange   PetscFunctionBegin;
11499734c634SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
11509734c634SMichael Lange   PetscValidPointer(dmParallel, 4);
11519734c634SMichael Lange 
11529734c634SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
11539734c634SMichael Lange   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
11549734c634SMichael Lange 
11559734c634SMichael Lange   /* Setup hybrid structure */
11569734c634SMichael Lange   for (d = 0; d <= dim; ++d) {pmesh->hybridPointMax[d] = mesh->hybridPointMax[d];}
11579734c634SMichael Lange   ierr = MPI_Bcast(pmesh->hybridPointMax, dim+1, MPIU_INT, 0, comm);CHKERRQ(ierr);
11589734c634SMichael Lange   ierr = ISLocalToGlobalMappingGetSize(renumbering, &n);CHKERRQ(ierr);
11599734c634SMichael Lange   ierr = ISLocalToGlobalMappingGetIndices(renumbering, &gpoints);CHKERRQ(ierr);
11609734c634SMichael Lange   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
11619734c634SMichael Lange   for (d = 0; d <= dim; ++d) {
11629734c634SMichael Lange     PetscInt pmax = pmesh->hybridPointMax[d], newmax = 0, pEnd, stratum[2], p;
11639734c634SMichael Lange 
11649734c634SMichael Lange     if (pmax < 0) continue;
11659734c634SMichael Lange     ierr = DMPlexGetDepthStratum(dm, d > depth ? depth : d, &stratum[0], &stratum[1]);CHKERRQ(ierr);
11669734c634SMichael Lange     ierr = DMPlexGetDepthStratum(dmParallel, d, NULL, &pEnd);CHKERRQ(ierr);
11679734c634SMichael Lange     ierr = MPI_Bcast(stratum, 2, MPIU_INT, 0, comm);CHKERRQ(ierr);
11689734c634SMichael Lange     for (p = 0; p < n; ++p) {
11699734c634SMichael Lange       const PetscInt point = gpoints[p];
11709734c634SMichael Lange 
11719734c634SMichael Lange       if ((point >= stratum[0]) && (point < stratum[1]) && (point >= pmax)) ++newmax;
11729734c634SMichael Lange     }
11739734c634SMichael Lange     if (newmax > 0) pmesh->hybridPointMax[d] = pEnd - newmax;
11749734c634SMichael Lange     else            pmesh->hybridPointMax[d] = -1;
11759734c634SMichael Lange   }
11769734c634SMichael Lange   ierr = ISLocalToGlobalMappingRestoreIndices(renumbering, &gpoints);CHKERRQ(ierr);
11779734c634SMichael Lange   PetscFunctionReturn(0);
11789734c634SMichael Lange }
11790df0e737SMichael Lange 
1180a6f36705SMichael Lange #undef __FUNCT__
1181a6f36705SMichael Lange #define __FUNCT__ "DMPlexDistributeSetupTree"
118208633170SToby Isaac PetscErrorCode DMPlexDistributeSetupTree(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping original, ISLocalToGlobalMapping renumbering, DM dmParallel)
1183a6f36705SMichael Lange {
118415078cd4SMichael Lange   DM_Plex        *mesh  = (DM_Plex*) dm->data;
118515078cd4SMichael Lange   DM_Plex        *pmesh = (DM_Plex*) (dmParallel)->data;
1186a6f36705SMichael Lange   MPI_Comm        comm;
1187a6f36705SMichael Lange   DM              refTree;
1188a6f36705SMichael Lange   PetscSection    origParentSection, newParentSection;
1189a6f36705SMichael Lange   PetscInt        *origParents, *origChildIDs;
1190a6f36705SMichael Lange   PetscBool       flg;
1191a6f36705SMichael Lange   PetscErrorCode  ierr;
1192a6f36705SMichael Lange 
1193a6f36705SMichael Lange   PetscFunctionBegin;
1194a6f36705SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1195a6f36705SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 4);
1196a6f36705SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
1197a6f36705SMichael Lange 
1198a6f36705SMichael Lange   /* Set up tree */
1199a6f36705SMichael Lange   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
1200a6f36705SMichael Lange   ierr = DMPlexSetReferenceTree(dmParallel,refTree);CHKERRQ(ierr);
1201a6f36705SMichael Lange   ierr = DMPlexGetTree(dm,&origParentSection,&origParents,&origChildIDs,NULL,NULL);CHKERRQ(ierr);
1202a6f36705SMichael Lange   if (origParentSection) {
1203a6f36705SMichael Lange     PetscInt        pStart, pEnd;
120408633170SToby Isaac     PetscInt        *newParents, *newChildIDs, *globParents;
1205a6f36705SMichael Lange     PetscInt        *remoteOffsetsParents, newParentSize;
1206a6f36705SMichael Lange     PetscSF         parentSF;
1207a6f36705SMichael Lange 
1208a6f36705SMichael Lange     ierr = DMPlexGetChart(dmParallel, &pStart, &pEnd);CHKERRQ(ierr);
1209a6f36705SMichael Lange     ierr = PetscSectionCreate(PetscObjectComm((PetscObject)dmParallel),&newParentSection);CHKERRQ(ierr);
1210a6f36705SMichael Lange     ierr = PetscSectionSetChart(newParentSection,pStart,pEnd);CHKERRQ(ierr);
1211a6f36705SMichael Lange     ierr = PetscSFDistributeSection(migrationSF, origParentSection, &remoteOffsetsParents, newParentSection);CHKERRQ(ierr);
1212a6f36705SMichael Lange     ierr = PetscSFCreateSectionSF(migrationSF, origParentSection, remoteOffsetsParents, newParentSection, &parentSF);CHKERRQ(ierr);
12138a713f3bSLawrence Mitchell     ierr = PetscFree(remoteOffsetsParents);CHKERRQ(ierr);
1214a6f36705SMichael Lange     ierr = PetscSectionGetStorageSize(newParentSection,&newParentSize);CHKERRQ(ierr);
1215a6f36705SMichael Lange     ierr = PetscMalloc2(newParentSize,&newParents,newParentSize,&newChildIDs);CHKERRQ(ierr);
121608633170SToby Isaac     if (original) {
121708633170SToby Isaac       PetscInt numParents;
121808633170SToby Isaac 
121908633170SToby Isaac       ierr = PetscSectionGetStorageSize(origParentSection,&numParents);CHKERRQ(ierr);
122008633170SToby Isaac       ierr = PetscMalloc1(numParents,&globParents);CHKERRQ(ierr);
122108633170SToby Isaac       ierr = ISLocalToGlobalMappingApplyBlock(original, numParents, origParents, globParents);CHKERRQ(ierr);
122208633170SToby Isaac     }
122308633170SToby Isaac     else {
122408633170SToby Isaac       globParents = origParents;
122508633170SToby Isaac     }
122608633170SToby Isaac     ierr = PetscSFBcastBegin(parentSF, MPIU_INT, globParents, newParents);CHKERRQ(ierr);
122708633170SToby Isaac     ierr = PetscSFBcastEnd(parentSF, MPIU_INT, globParents, newParents);CHKERRQ(ierr);
122808633170SToby Isaac     if (original) {
122908633170SToby Isaac       ierr = PetscFree(globParents);CHKERRQ(ierr);
123008633170SToby Isaac     }
1231a6f36705SMichael Lange     ierr = PetscSFBcastBegin(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr);
1232a6f36705SMichael Lange     ierr = PetscSFBcastEnd(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr);
1233a6f36705SMichael Lange     ierr = ISGlobalToLocalMappingApplyBlock(renumbering,IS_GTOLM_MASK, newParentSize, newParents, NULL, newParents);CHKERRQ(ierr);
12344a54e071SToby Isaac #if PETSC_USE_DEBUG
12354a54e071SToby Isaac     {
12364a54e071SToby Isaac       PetscInt  p;
12374a54e071SToby Isaac       PetscBool valid = PETSC_TRUE;
12384a54e071SToby Isaac       for (p = 0; p < newParentSize; ++p) {
12394a54e071SToby Isaac         if (newParents[p] < 0) {valid = PETSC_FALSE; ierr = PetscPrintf(PETSC_COMM_SELF, "Point %d not in overlap SF\n", p);CHKERRQ(ierr);}
12404a54e071SToby Isaac       }
12414a54e071SToby Isaac       if (!valid) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid global to local map");
12424a54e071SToby Isaac     }
12434a54e071SToby Isaac #endif
1244c5929fdfSBarry Smith     ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-parents_view", &flg);CHKERRQ(ierr);
1245a6f36705SMichael Lange     if (flg) {
1246a6f36705SMichael Lange       ierr = PetscPrintf(comm, "Serial Parent Section: \n");CHKERRQ(ierr);
1247a6f36705SMichael Lange       ierr = PetscSectionView(origParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1248a6f36705SMichael Lange       ierr = PetscPrintf(comm, "Parallel Parent Section: \n");CHKERRQ(ierr);
1249a6f36705SMichael Lange       ierr = PetscSectionView(newParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1250a6f36705SMichael Lange       ierr = PetscSFView(parentSF, NULL);CHKERRQ(ierr);
1251a6f36705SMichael Lange     }
1252a6f36705SMichael Lange     ierr = DMPlexSetTree(dmParallel,newParentSection,newParents,newChildIDs);CHKERRQ(ierr);
1253a6f36705SMichael Lange     ierr = PetscSectionDestroy(&newParentSection);CHKERRQ(ierr);
1254a6f36705SMichael Lange     ierr = PetscFree2(newParents,newChildIDs);CHKERRQ(ierr);
1255a6f36705SMichael Lange     ierr = PetscSFDestroy(&parentSF);CHKERRQ(ierr);
1256a6f36705SMichael Lange   }
125715078cd4SMichael Lange   pmesh->useAnchors = mesh->useAnchors;
1258a6f36705SMichael Lange   PetscFunctionReturn(0);
1259a6f36705SMichael Lange }
12600df0e737SMichael Lange 
12610df0e737SMichael Lange #undef __FUNCT__
12624eca1733SMichael Lange #define __FUNCT__ "DMPlexDistributeSF"
1263f8987ae8SMichael Lange PetscErrorCode DMPlexDistributeSF(DM dm, PetscSF migrationSF, DM dmParallel)
12644eca1733SMichael Lange {
12654eca1733SMichael Lange   DM_Plex               *mesh  = (DM_Plex*) dm->data;
12664eca1733SMichael Lange   DM_Plex               *pmesh = (DM_Plex*) (dmParallel)->data;
12674eca1733SMichael Lange   PetscMPIInt            rank, numProcs;
12684eca1733SMichael Lange   MPI_Comm               comm;
12694eca1733SMichael Lange   PetscErrorCode         ierr;
12704eca1733SMichael Lange 
12714eca1733SMichael Lange   PetscFunctionBegin;
12724eca1733SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
12734eca1733SMichael Lange   PetscValidPointer(dmParallel,7);
12744eca1733SMichael Lange 
12754eca1733SMichael Lange   /* Create point SF for parallel mesh */
12764eca1733SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr);
12774eca1733SMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
12784eca1733SMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
12794eca1733SMichael Lange   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
12804eca1733SMichael Lange   {
12814eca1733SMichael Lange     const PetscInt *leaves;
12824eca1733SMichael Lange     PetscSFNode    *remotePoints, *rowners, *lowners;
12834eca1733SMichael Lange     PetscInt        numRoots, numLeaves, numGhostPoints = 0, p, gp, *ghostPoints;
12844eca1733SMichael Lange     PetscInt        pStart, pEnd;
12854eca1733SMichael Lange 
12864eca1733SMichael Lange     ierr = DMPlexGetChart(dmParallel, &pStart, &pEnd);CHKERRQ(ierr);
12874eca1733SMichael Lange     ierr = PetscSFGetGraph(migrationSF, &numRoots, &numLeaves, &leaves, NULL);CHKERRQ(ierr);
12884eca1733SMichael Lange     ierr = PetscMalloc2(numRoots,&rowners,numLeaves,&lowners);CHKERRQ(ierr);
12894eca1733SMichael Lange     for (p=0; p<numRoots; p++) {
12904eca1733SMichael Lange       rowners[p].rank  = -1;
12914eca1733SMichael Lange       rowners[p].index = -1;
12924eca1733SMichael Lange     }
12934eca1733SMichael Lange     ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
12944eca1733SMichael Lange     ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
12954eca1733SMichael Lange     for (p = 0; p < numLeaves; ++p) {
12964eca1733SMichael Lange       if (lowners[p].rank < 0 || lowners[p].rank == rank) { /* Either put in a bid or we know we own it */
12974eca1733SMichael Lange         lowners[p].rank  = rank;
12984eca1733SMichael Lange         lowners[p].index = leaves ? leaves[p] : p;
12994eca1733SMichael Lange       } else if (lowners[p].rank >= 0) { /* Point already claimed so flag so that MAXLOC does not listen to us */
13004eca1733SMichael Lange         lowners[p].rank  = -2;
13014eca1733SMichael Lange         lowners[p].index = -2;
13024eca1733SMichael Lange       }
13034eca1733SMichael Lange     }
13044eca1733SMichael Lange     for (p=0; p<numRoots; p++) { /* Root must not participate in the rediction, flag so that MAXLOC does not use */
13054eca1733SMichael Lange       rowners[p].rank  = -3;
13064eca1733SMichael Lange       rowners[p].index = -3;
13074eca1733SMichael Lange     }
13084eca1733SMichael Lange     ierr = PetscSFReduceBegin(migrationSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr);
13094eca1733SMichael Lange     ierr = PetscSFReduceEnd(migrationSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr);
13104eca1733SMichael Lange     ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
13114eca1733SMichael Lange     ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr);
13124eca1733SMichael Lange     for (p = 0; p < numLeaves; ++p) {
13134eca1733SMichael Lange       if (lowners[p].rank < 0 || lowners[p].index < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Cell partition corrupt: point not claimed");
13144eca1733SMichael Lange       if (lowners[p].rank != rank) ++numGhostPoints;
13154eca1733SMichael Lange     }
13164eca1733SMichael Lange     ierr = PetscMalloc1(numGhostPoints, &ghostPoints);CHKERRQ(ierr);
13174eca1733SMichael Lange     ierr = PetscMalloc1(numGhostPoints, &remotePoints);CHKERRQ(ierr);
13184eca1733SMichael Lange     for (p = 0, gp = 0; p < numLeaves; ++p) {
13194eca1733SMichael Lange       if (lowners[p].rank != rank) {
13204eca1733SMichael Lange         ghostPoints[gp]        = leaves ? leaves[p] : p;
13214eca1733SMichael Lange         remotePoints[gp].rank  = lowners[p].rank;
13224eca1733SMichael Lange         remotePoints[gp].index = lowners[p].index;
13234eca1733SMichael Lange         ++gp;
13244eca1733SMichael Lange       }
13254eca1733SMichael Lange     }
13264eca1733SMichael Lange     ierr = PetscFree2(rowners,lowners);CHKERRQ(ierr);
13274eca1733SMichael Lange     ierr = PetscSFSetGraph((dmParallel)->sf, pEnd - pStart, numGhostPoints, ghostPoints, PETSC_OWN_POINTER, remotePoints, PETSC_OWN_POINTER);CHKERRQ(ierr);
13284eca1733SMichael Lange     ierr = PetscSFSetFromOptions((dmParallel)->sf);CHKERRQ(ierr);
13294eca1733SMichael Lange   }
13304eca1733SMichael Lange   pmesh->useCone    = mesh->useCone;
13314eca1733SMichael Lange   pmesh->useClosure = mesh->useClosure;
13324eca1733SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr);
13334eca1733SMichael Lange   PetscFunctionReturn(0);
13344eca1733SMichael Lange }
13354eca1733SMichael Lange 
13364eca1733SMichael Lange #undef __FUNCT__
1337f5bf2dbfSMichael Lange #define __FUNCT__ "DMPlexCreatePointSF"
1338f5bf2dbfSMichael Lange /*@C
1339f5bf2dbfSMichael Lange   DMPlexDerivePointSF - Build a point SF from an SF describing a point migration
1340f5bf2dbfSMichael Lange 
1341f5bf2dbfSMichael Lange   Input Parameter:
1342f5bf2dbfSMichael Lange + dm          - The source DMPlex object
1343f5bf2dbfSMichael Lange . migrationSF - The star forest that describes the parallel point remapping
13441627f6ccSMichael Lange . ownership   - Flag causing a vote to determine point ownership
1345f5bf2dbfSMichael Lange 
1346f5bf2dbfSMichael Lange   Output Parameter:
1347f5bf2dbfSMichael Lange - pointSF     - The star forest describing the point overlap in the remapped DM
1348f5bf2dbfSMichael Lange 
1349f5bf2dbfSMichael Lange   Level: developer
1350f5bf2dbfSMichael Lange 
1351f5bf2dbfSMichael Lange .seealso: DMPlexDistribute(), DMPlexDistributeOverlap()
1352f5bf2dbfSMichael Lange @*/
1353f5bf2dbfSMichael Lange PetscErrorCode DMPlexCreatePointSF(DM dm, PetscSF migrationSF, PetscBool ownership, PetscSF *pointSF)
1354f5bf2dbfSMichael Lange {
1355f5bf2dbfSMichael Lange   PetscMPIInt        rank;
13561627f6ccSMichael Lange   PetscInt           p, nroots, nleaves, idx, npointLeaves;
1357f5bf2dbfSMichael Lange   PetscInt          *pointLocal;
1358f5bf2dbfSMichael Lange   const PetscInt    *leaves;
1359f5bf2dbfSMichael Lange   const PetscSFNode *roots;
1360f5bf2dbfSMichael Lange   PetscSFNode       *rootNodes, *leafNodes, *pointRemote;
1361f5bf2dbfSMichael Lange   PetscErrorCode     ierr;
1362f5bf2dbfSMichael Lange 
1363f5bf2dbfSMichael Lange   PetscFunctionBegin;
1364f5bf2dbfSMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1365f5bf2dbfSMichael Lange   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
1366f5bf2dbfSMichael Lange 
1367f5bf2dbfSMichael Lange   ierr = PetscSFGetGraph(migrationSF, &nroots, &nleaves, &leaves, &roots);CHKERRQ(ierr);
1368f5bf2dbfSMichael Lange   ierr = PetscMalloc2(nroots, &rootNodes, nleaves, &leafNodes);CHKERRQ(ierr);
1369f5bf2dbfSMichael Lange   if (ownership) {
13701627f6ccSMichael Lange     /* Point ownership vote: Process with highest rank ownes shared points */
1371f5bf2dbfSMichael Lange     for (p = 0; p < nleaves; ++p) {
1372f5bf2dbfSMichael Lange       /* Either put in a bid or we know we own it */
1373f5bf2dbfSMichael Lange       leafNodes[p].rank  = rank;
137443f7d02bSMichael Lange       leafNodes[p].index = p;
1375f5bf2dbfSMichael Lange     }
1376f5bf2dbfSMichael Lange     for (p = 0; p < nroots; p++) {
13771627f6ccSMichael Lange       /* Root must not participate in the reduction, flag so that MAXLOC does not use */
1378f5bf2dbfSMichael Lange       rootNodes[p].rank  = -3;
1379f5bf2dbfSMichael Lange       rootNodes[p].index = -3;
1380f5bf2dbfSMichael Lange     }
1381f5bf2dbfSMichael Lange     ierr = PetscSFReduceBegin(migrationSF, MPIU_2INT, leafNodes, rootNodes, MPI_MAXLOC);CHKERRQ(ierr);
1382f5bf2dbfSMichael Lange     ierr = PetscSFReduceEnd(migrationSF, MPIU_2INT, leafNodes, rootNodes, MPI_MAXLOC);CHKERRQ(ierr);
1383f5bf2dbfSMichael Lange   } else {
1384f5bf2dbfSMichael Lange     for (p = 0; p < nroots; p++) {
1385f5bf2dbfSMichael Lange       rootNodes[p].index = -1;
1386f5bf2dbfSMichael Lange       rootNodes[p].rank = rank;
1387f5bf2dbfSMichael Lange     };
1388f5bf2dbfSMichael Lange     for (p = 0; p < nleaves; p++) {
1389f5bf2dbfSMichael Lange       /* Write new local id into old location */
1390f5bf2dbfSMichael Lange       if (roots[p].rank == rank) {
1391f5bf2dbfSMichael Lange         rootNodes[roots[p].index].index = leaves[p];
1392f5bf2dbfSMichael Lange       }
1393f5bf2dbfSMichael Lange     }
1394f5bf2dbfSMichael Lange   }
1395f5bf2dbfSMichael Lange   ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rootNodes, leafNodes);CHKERRQ(ierr);
1396f5bf2dbfSMichael Lange   ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rootNodes, leafNodes);CHKERRQ(ierr);
1397f5bf2dbfSMichael Lange 
1398f5bf2dbfSMichael Lange   for (npointLeaves = 0, p = 0; p < nleaves; p++) {if (leafNodes[p].rank != rank) npointLeaves++;}
13991627f6ccSMichael Lange   ierr = PetscMalloc1(npointLeaves, &pointLocal);CHKERRQ(ierr);
14001627f6ccSMichael Lange   ierr = PetscMalloc1(npointLeaves, &pointRemote);CHKERRQ(ierr);
1401f5bf2dbfSMichael Lange   for (idx = 0, p = 0; p < nleaves; p++) {
1402f5bf2dbfSMichael Lange     if (leafNodes[p].rank != rank) {
1403f5bf2dbfSMichael Lange       pointLocal[idx] = p;
1404f5bf2dbfSMichael Lange       pointRemote[idx] = leafNodes[p];
1405f5bf2dbfSMichael Lange       idx++;
1406f5bf2dbfSMichael Lange     }
1407f5bf2dbfSMichael Lange   }
1408f5bf2dbfSMichael Lange   ierr = PetscSFCreate(PetscObjectComm((PetscObject) dm), pointSF);CHKERRQ(ierr);
14091627f6ccSMichael Lange   ierr = PetscSFSetFromOptions(*pointSF);CHKERRQ(ierr);
1410f5bf2dbfSMichael Lange   ierr = PetscSFSetGraph(*pointSF, nleaves, npointLeaves, pointLocal, PETSC_OWN_POINTER, pointRemote, PETSC_OWN_POINTER);CHKERRQ(ierr);
1411f5bf2dbfSMichael Lange   ierr = PetscFree2(rootNodes, leafNodes);CHKERRQ(ierr);
1412f5bf2dbfSMichael Lange   PetscFunctionReturn(0);
1413f5bf2dbfSMichael Lange }
1414f5bf2dbfSMichael Lange 
1415f5bf2dbfSMichael Lange #undef __FUNCT__
141615078cd4SMichael Lange #define __FUNCT__ "DMPlexMigrate"
141715078cd4SMichael Lange /*@C
141815078cd4SMichael Lange   DMPlexMigrate  - Migrates internal DM data over the supplied star forest
141915078cd4SMichael Lange 
142015078cd4SMichael Lange   Input Parameter:
142115078cd4SMichael Lange + dm       - The source DMPlex object
14221627f6ccSMichael Lange . sf       - The star forest communication context describing the migration pattern
142315078cd4SMichael Lange 
142415078cd4SMichael Lange   Output Parameter:
142515078cd4SMichael Lange - targetDM - The target DMPlex object
142615078cd4SMichael Lange 
1427b9f40539SMichael Lange   Level: intermediate
142815078cd4SMichael Lange 
142915078cd4SMichael Lange .seealso: DMPlexDistribute(), DMPlexDistributeOverlap()
143015078cd4SMichael Lange @*/
1431b9f40539SMichael Lange PetscErrorCode DMPlexMigrate(DM dm, PetscSF sf, DM targetDM)
143215078cd4SMichael Lange {
1433b9f40539SMichael Lange   MPI_Comm               comm;
1434b9f40539SMichael Lange   PetscInt               dim, nroots;
1435b9f40539SMichael Lange   PetscSF                sfPoint;
143615078cd4SMichael Lange   ISLocalToGlobalMapping ltogMigration;
1437ac04eaf7SToby Isaac   ISLocalToGlobalMapping ltogOriginal = NULL;
1438b9f40539SMichael Lange   PetscBool              flg;
143915078cd4SMichael Lange   PetscErrorCode         ierr;
144015078cd4SMichael Lange 
144115078cd4SMichael Lange   PetscFunctionBegin;
144215078cd4SMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
14431b858b30SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_Migrate, dm, 0, 0, 0);CHKERRQ(ierr);
1444b9f40539SMichael Lange   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
144515078cd4SMichael Lange   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
144615078cd4SMichael Lange   ierr = DMSetDimension(targetDM, dim);CHKERRQ(ierr);
144715078cd4SMichael Lange 
1448bfb0467fSMichael Lange   /* Check for a one-to-all distribution pattern */
1449b9f40539SMichael Lange   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
1450b9f40539SMichael Lange   ierr = PetscSFGetGraph(sfPoint, &nroots, NULL, NULL, NULL);CHKERRQ(ierr);
1451bfb0467fSMichael Lange   if (nroots >= 0) {
1452b9f40539SMichael Lange     IS                     isOriginal;
1453ac04eaf7SToby Isaac     PetscInt               n, size, nleaves;
1454ac04eaf7SToby Isaac     PetscInt              *numbering_orig, *numbering_new;
1455b9f40539SMichael Lange     /* Get the original point numbering */
1456b9f40539SMichael Lange     ierr = DMPlexCreatePointNumbering(dm, &isOriginal);CHKERRQ(ierr);
1457b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingCreateIS(isOriginal, &ltogOriginal);CHKERRQ(ierr);
1458b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingGetSize(ltogOriginal, &size);CHKERRQ(ierr);
1459b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingGetBlockIndices(ltogOriginal, (const PetscInt**)&numbering_orig);CHKERRQ(ierr);
1460b9f40539SMichael Lange     /* Convert to positive global numbers */
1461b9f40539SMichael Lange     for (n=0; n<size; n++) {if (numbering_orig[n] < 0) numbering_orig[n] = -(numbering_orig[n]+1);}
1462b9f40539SMichael Lange     /* Derive the new local-to-global mapping from the old one */
1463b9f40539SMichael Lange     ierr = PetscSFGetGraph(sf, NULL, &nleaves, NULL, NULL);CHKERRQ(ierr);
1464b9f40539SMichael Lange     ierr = PetscMalloc1(nleaves, &numbering_new);CHKERRQ(ierr);
1465b9f40539SMichael Lange     ierr = PetscSFBcastBegin(sf, MPIU_INT, (PetscInt *) numbering_orig, numbering_new);CHKERRQ(ierr);
1466b9f40539SMichael Lange     ierr = PetscSFBcastEnd(sf, MPIU_INT, (PetscInt *) numbering_orig, numbering_new);CHKERRQ(ierr);
1467b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingCreate(comm, 1, nleaves, (const PetscInt*) numbering_new, PETSC_OWN_POINTER, &ltogMigration);CHKERRQ(ierr);
1468b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingRestoreIndices(ltogOriginal, (const PetscInt**)&numbering_orig);CHKERRQ(ierr);
1469b9f40539SMichael Lange     ierr = ISDestroy(&isOriginal);CHKERRQ(ierr);
147015078cd4SMichael Lange   } else {
1471bfb0467fSMichael Lange     /* One-to-all distribution pattern: We can derive LToG from SF */
147215078cd4SMichael Lange     ierr = ISLocalToGlobalMappingCreateSF(sf, 0, &ltogMigration);CHKERRQ(ierr);
147315078cd4SMichael Lange   }
1474c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-partition_view", &flg);CHKERRQ(ierr);
1475b9f40539SMichael Lange   if (flg) {
1476b9f40539SMichael Lange     ierr = PetscPrintf(comm, "Point renumbering for DM migration:\n");CHKERRQ(ierr);
1477b9f40539SMichael Lange     ierr = ISLocalToGlobalMappingView(ltogMigration, NULL);CHKERRQ(ierr);
1478b9f40539SMichael Lange   }
147915078cd4SMichael Lange   /* Migrate DM data to target DM */
1480ac04eaf7SToby Isaac   ierr = DMPlexDistributeCones(dm, sf, ltogOriginal, ltogMigration, targetDM);CHKERRQ(ierr);
148115078cd4SMichael Lange   ierr = DMPlexDistributeCoordinates(dm, sf, targetDM);CHKERRQ(ierr);
148215078cd4SMichael Lange   ierr = DMPlexDistributeLabels(dm, sf, targetDM);CHKERRQ(ierr);
148315078cd4SMichael Lange   ierr = DMPlexDistributeSetupHybrid(dm, sf, ltogMigration, targetDM);CHKERRQ(ierr);
148408633170SToby Isaac   ierr = DMPlexDistributeSetupTree(dm, sf, ltogOriginal, ltogMigration, targetDM);CHKERRQ(ierr);
1485ac04eaf7SToby Isaac   ierr = ISLocalToGlobalMappingDestroy(&ltogOriginal);CHKERRQ(ierr);
1486302440fdSBarry Smith   ierr = ISLocalToGlobalMappingDestroy(&ltogMigration);CHKERRQ(ierr);
14871b858b30SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_Migrate, dm, 0, 0, 0);CHKERRQ(ierr);
148815078cd4SMichael Lange   PetscFunctionReturn(0);
148915078cd4SMichael Lange }
149015078cd4SMichael Lange 
149115078cd4SMichael Lange #undef __FUNCT__
149270034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistribute"
14933b8f15a2SMatthew G. Knepley /*@C
149470034214SMatthew G. Knepley   DMPlexDistribute - Distributes the mesh and any associated sections.
149570034214SMatthew G. Knepley 
149670034214SMatthew G. Knepley   Not Collective
149770034214SMatthew G. Knepley 
149870034214SMatthew G. Knepley   Input Parameter:
149970034214SMatthew G. Knepley + dm  - The original DMPlex object
150070034214SMatthew G. Knepley - overlap - The overlap of partitions, 0 is the default
150170034214SMatthew G. Knepley 
150270034214SMatthew G. Knepley   Output Parameter:
150370034214SMatthew G. Knepley + sf - The PetscSF used for point distribution
150470034214SMatthew G. Knepley - parallelMesh - The distributed DMPlex object, or NULL
150570034214SMatthew G. Knepley 
150670034214SMatthew G. Knepley   Note: If the mesh was not distributed, the return value is NULL.
150770034214SMatthew G. Knepley 
150870034214SMatthew G. Knepley   The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and
150970034214SMatthew G. Knepley   DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function
151070034214SMatthew G. Knepley   representation on the mesh.
151170034214SMatthew G. Knepley 
151270034214SMatthew G. Knepley   Level: intermediate
151370034214SMatthew G. Knepley 
151470034214SMatthew G. Knepley .keywords: mesh, elements
151570034214SMatthew G. Knepley .seealso: DMPlexCreate(), DMPlexDistributeByFace(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure()
151670034214SMatthew G. Knepley @*/
151780cf41d5SMatthew G. Knepley PetscErrorCode DMPlexDistribute(DM dm, PetscInt overlap, PetscSF *sf, DM *dmParallel)
151870034214SMatthew G. Knepley {
151970034214SMatthew G. Knepley   MPI_Comm               comm;
152015078cd4SMichael Lange   PetscPartitioner       partitioner;
1521f8987ae8SMichael Lange   IS                     cellPart;
1522f8987ae8SMichael Lange   PetscSection           cellPartSection;
1523cf86098cSMatthew G. Knepley   DM                     dmCoord;
1524f8987ae8SMichael Lange   DMLabel                lblPartition, lblMigration;
152543f7d02bSMichael Lange   PetscSF                sfProcess, sfMigration, sfStratified, sfPoint;
152670034214SMatthew G. Knepley   PetscBool              flg;
152770034214SMatthew G. Knepley   PetscMPIInt            rank, numProcs, p;
152870034214SMatthew G. Knepley   PetscErrorCode         ierr;
152970034214SMatthew G. Knepley 
153070034214SMatthew G. Knepley   PetscFunctionBegin;
153170034214SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
153270034214SMatthew G. Knepley   if (sf) PetscValidPointer(sf,4);
153370034214SMatthew G. Knepley   PetscValidPointer(dmParallel,5);
153470034214SMatthew G. Knepley 
153570034214SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr);
153670034214SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
153770034214SMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
153870034214SMatthew G. Knepley   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
153970034214SMatthew G. Knepley 
154070034214SMatthew G. Knepley   *dmParallel = NULL;
154170034214SMatthew G. Knepley   if (numProcs == 1) PetscFunctionReturn(0);
154270034214SMatthew G. Knepley 
154315078cd4SMichael Lange   /* Create cell partition */
154477623264SMatthew G. Knepley   ierr = PetscLogEventBegin(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
154577623264SMatthew G. Knepley   ierr = PetscSectionCreate(comm, &cellPartSection);CHKERRQ(ierr);
154615078cd4SMichael Lange   ierr = DMPlexGetPartitioner(dm, &partitioner);CHKERRQ(ierr);
154715078cd4SMichael Lange   ierr = PetscPartitionerPartition(partitioner, dm, cellPartSection, &cellPart);CHKERRQ(ierr);
1548f8987ae8SMichael Lange   {
1549f8987ae8SMichael Lange     /* Convert partition to DMLabel */
1550f8987ae8SMichael Lange     PetscInt proc, pStart, pEnd, npoints, poffset;
1551f8987ae8SMichael Lange     const PetscInt *points;
1552f8987ae8SMichael Lange     ierr = DMLabelCreate("Point Partition", &lblPartition);CHKERRQ(ierr);
1553f8987ae8SMichael Lange     ierr = ISGetIndices(cellPart, &points);CHKERRQ(ierr);
1554f8987ae8SMichael Lange     ierr = PetscSectionGetChart(cellPartSection, &pStart, &pEnd);CHKERRQ(ierr);
1555f8987ae8SMichael Lange     for (proc = pStart; proc < pEnd; proc++) {
1556f8987ae8SMichael Lange       ierr = PetscSectionGetDof(cellPartSection, proc, &npoints);CHKERRQ(ierr);
1557f8987ae8SMichael Lange       ierr = PetscSectionGetOffset(cellPartSection, proc, &poffset);CHKERRQ(ierr);
1558f8987ae8SMichael Lange       for (p = poffset; p < poffset+npoints; p++) {
1559f8987ae8SMichael Lange         ierr = DMLabelSetValue(lblPartition, points[p], proc);CHKERRQ(ierr);
156070034214SMatthew G. Knepley       }
1561f8987ae8SMichael Lange     }
1562f8987ae8SMichael Lange     ierr = ISRestoreIndices(cellPart, &points);CHKERRQ(ierr);
1563f8987ae8SMichael Lange   }
1564f8987ae8SMichael Lange   ierr = DMPlexPartitionLabelClosure(dm, lblPartition);CHKERRQ(ierr);
1565f8987ae8SMichael Lange   {
1566f8987ae8SMichael Lange     /* Build a global process SF */
1567f8987ae8SMichael Lange     PetscSFNode *remoteProc;
1568f8987ae8SMichael Lange     ierr = PetscMalloc1(numProcs, &remoteProc);CHKERRQ(ierr);
1569f8987ae8SMichael Lange     for (p = 0; p < numProcs; ++p) {
1570f8987ae8SMichael Lange       remoteProc[p].rank  = p;
1571f8987ae8SMichael Lange       remoteProc[p].index = rank;
1572f8987ae8SMichael Lange     }
1573f8987ae8SMichael Lange     ierr = PetscSFCreate(comm, &sfProcess);CHKERRQ(ierr);
1574f8987ae8SMichael Lange     ierr = PetscObjectSetName((PetscObject) sfProcess, "Process SF");CHKERRQ(ierr);
1575f8987ae8SMichael Lange     ierr = PetscSFSetGraph(sfProcess, numProcs, numProcs, NULL, PETSC_OWN_POINTER, remoteProc, PETSC_OWN_POINTER);CHKERRQ(ierr);
1576f8987ae8SMichael Lange   }
1577f8987ae8SMichael Lange   ierr = DMLabelCreate("Point migration", &lblMigration);CHKERRQ(ierr);
1578f8987ae8SMichael Lange   ierr = DMPlexPartitionLabelInvert(dm, lblPartition, sfProcess, lblMigration);CHKERRQ(ierr);
1579302440fdSBarry Smith   ierr = DMPlexPartitionLabelCreateSF(dm, lblMigration, &sfMigration);CHKERRQ(ierr);
158043f7d02bSMichael Lange   /* Stratify the SF in case we are migrating an already parallel plex */
158143f7d02bSMichael Lange   ierr = DMPlexStratifyMigrationSF(dm, sfMigration, &sfStratified);CHKERRQ(ierr);
158243f7d02bSMichael Lange   ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr);
158343f7d02bSMichael Lange   sfMigration = sfStratified;
1584f8987ae8SMichael Lange   ierr = PetscLogEventEnd(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
1585c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-partition_view", &flg);CHKERRQ(ierr);
158670034214SMatthew G. Knepley   if (flg) {
1587f8987ae8SMichael Lange     ierr = DMLabelView(lblPartition, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1588f8987ae8SMichael Lange     ierr = PetscSFView(sfMigration, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
158970034214SMatthew G. Knepley   }
1590f8987ae8SMichael Lange 
159115078cd4SMichael Lange   /* Create non-overlapping parallel DM and migrate internal data */
159270034214SMatthew G. Knepley   ierr = DMPlexCreate(comm, dmParallel);CHKERRQ(ierr);
159370034214SMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) *dmParallel, "Parallel Mesh");CHKERRQ(ierr);
1594b9f40539SMichael Lange   ierr = DMPlexMigrate(dm, sfMigration, *dmParallel);CHKERRQ(ierr);
159570034214SMatthew G. Knepley 
1596a157612eSMichael Lange   /* Build the point SF without overlap */
15971627f6ccSMichael Lange   ierr = DMPlexCreatePointSF(*dmParallel, sfMigration, PETSC_TRUE, &sfPoint);CHKERRQ(ierr);
1598f5bf2dbfSMichael Lange   ierr = DMSetPointSF(*dmParallel, sfPoint);CHKERRQ(ierr);
1599cf86098cSMatthew G. Knepley   ierr = DMGetCoordinateDM(*dmParallel, &dmCoord);CHKERRQ(ierr);
1600cf86098cSMatthew G. Knepley   if (dmCoord) {ierr = DMSetPointSF(dmCoord, sfPoint);CHKERRQ(ierr);}
1601f5bf2dbfSMichael Lange   if (flg) {ierr = PetscSFView(sfPoint, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);}
160270034214SMatthew G. Knepley 
1603a157612eSMichael Lange   if (overlap > 0) {
160415078cd4SMichael Lange     DM                 dmOverlap;
160515078cd4SMichael Lange     PetscInt           nroots, nleaves;
160615078cd4SMichael Lange     PetscSFNode       *newRemote;
160715078cd4SMichael Lange     const PetscSFNode *oldRemote;
160815078cd4SMichael Lange     PetscSF            sfOverlap, sfOverlapPoint;
1609a157612eSMichael Lange     /* Add the partition overlap to the distributed DM */
1610b9f40539SMichael Lange     ierr = DMPlexDistributeOverlap(*dmParallel, overlap, &sfOverlap, &dmOverlap);CHKERRQ(ierr);
1611a157612eSMichael Lange     ierr = DMDestroy(dmParallel);CHKERRQ(ierr);
1612a157612eSMichael Lange     *dmParallel = dmOverlap;
1613c389ea9fSToby Isaac     if (flg) {
16143d822a50SMichael Lange       ierr = PetscPrintf(comm, "Overlap Migration SF:\n");CHKERRQ(ierr);
161515078cd4SMichael Lange       ierr = PetscSFView(sfOverlap, NULL);CHKERRQ(ierr);
1616c389ea9fSToby Isaac     }
161743331d4aSMichael Lange 
1618f8987ae8SMichael Lange     /* Re-map the migration SF to establish the full migration pattern */
1619f8987ae8SMichael Lange     ierr = PetscSFGetGraph(sfMigration, &nroots, NULL, NULL, &oldRemote);CHKERRQ(ierr);
162015078cd4SMichael Lange     ierr = PetscSFGetGraph(sfOverlap, NULL, &nleaves, NULL, NULL);CHKERRQ(ierr);
162143331d4aSMichael Lange     ierr = PetscMalloc1(nleaves, &newRemote);CHKERRQ(ierr);
162215078cd4SMichael Lange     ierr = PetscSFBcastBegin(sfOverlap, MPIU_2INT, oldRemote, newRemote);CHKERRQ(ierr);
162315078cd4SMichael Lange     ierr = PetscSFBcastEnd(sfOverlap, MPIU_2INT, oldRemote, newRemote);CHKERRQ(ierr);
162415078cd4SMichael Lange     ierr = PetscSFCreate(comm, &sfOverlapPoint);CHKERRQ(ierr);
162515078cd4SMichael Lange     ierr = PetscSFSetGraph(sfOverlapPoint, nroots, nleaves, NULL, PETSC_OWN_POINTER, newRemote, PETSC_OWN_POINTER);CHKERRQ(ierr);
162615078cd4SMichael Lange     ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr);
1627f8987ae8SMichael Lange     ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr);
162815078cd4SMichael Lange     sfMigration = sfOverlapPoint;
1629c389ea9fSToby Isaac   }
1630bf5244c7SMatthew G. Knepley   /* Cleanup Partition */
1631f8987ae8SMichael Lange   ierr = PetscSFDestroy(&sfProcess);CHKERRQ(ierr);
1632f8987ae8SMichael Lange   ierr = DMLabelDestroy(&lblPartition);CHKERRQ(ierr);
1633f8987ae8SMichael Lange   ierr = DMLabelDestroy(&lblMigration);CHKERRQ(ierr);
16344eca1733SMichael Lange   ierr = PetscSectionDestroy(&cellPartSection);CHKERRQ(ierr);
16354eca1733SMichael Lange   ierr = ISDestroy(&cellPart);CHKERRQ(ierr);
1636721cbd76SMatthew G. Knepley   /* Copy BC */
1637721cbd76SMatthew G. Knepley   ierr = DMPlexCopyBoundary(dm, *dmParallel);CHKERRQ(ierr);
163866fe0bfeSMatthew G. Knepley   /* Create sfNatural */
163966fe0bfeSMatthew G. Knepley   if (dm->useNatural) {
164066fe0bfeSMatthew G. Knepley     PetscSection section;
164166fe0bfeSMatthew G. Knepley 
164266fe0bfeSMatthew G. Knepley     ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
16430c1f158bSMatthew G. Knepley     ierr = DMPlexCreateGlobalToNaturalSF(*dmParallel, section, sfMigration, &(*dmParallel)->sfNatural);CHKERRQ(ierr);
164466fe0bfeSMatthew G. Knepley   }
1645721cbd76SMatthew G. Knepley   /* Cleanup */
1646f8987ae8SMichael Lange   if (sf) {*sf = sfMigration;}
1647f8987ae8SMichael Lange   else    {ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr);}
1648f5bf2dbfSMichael Lange   ierr = PetscSFDestroy(&sfPoint);CHKERRQ(ierr);
164970034214SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr);
165070034214SMatthew G. Knepley   PetscFunctionReturn(0);
165170034214SMatthew G. Knepley }
1652a157612eSMichael Lange 
1653a157612eSMichael Lange #undef __FUNCT__
1654a157612eSMichael Lange #define __FUNCT__ "DMPlexDistributeOverlap"
1655a157612eSMichael Lange /*@C
1656a157612eSMichael Lange   DMPlexDistribute - Add partition overlap to a distributed non-overlapping DM.
1657a157612eSMichael Lange 
1658a157612eSMichael Lange   Not Collective
1659a157612eSMichael Lange 
1660a157612eSMichael Lange   Input Parameter:
1661a157612eSMichael Lange + dm  - The non-overlapping distrbuted DMPlex object
1662a157612eSMichael Lange - overlap - The overlap of partitions, 0 is the default
1663a157612eSMichael Lange 
1664a157612eSMichael Lange   Output Parameter:
1665a157612eSMichael Lange + sf - The PetscSF used for point distribution
1666a157612eSMichael Lange - dmOverlap - The overlapping distributed DMPlex object, or NULL
1667a157612eSMichael Lange 
1668a157612eSMichael Lange   Note: If the mesh was not distributed, the return value is NULL.
1669a157612eSMichael Lange 
1670a157612eSMichael Lange   The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and
1671a157612eSMichael Lange   DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function
1672a157612eSMichael Lange   representation on the mesh.
1673a157612eSMichael Lange 
1674a157612eSMichael Lange   Level: intermediate
1675a157612eSMichael Lange 
1676a157612eSMichael Lange .keywords: mesh, elements
1677a157612eSMichael Lange .seealso: DMPlexCreate(), DMPlexDistributeByFace(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure()
1678a157612eSMichael Lange @*/
1679b9f40539SMichael Lange PetscErrorCode DMPlexDistributeOverlap(DM dm, PetscInt overlap, PetscSF *sf, DM *dmOverlap)
1680a157612eSMichael Lange {
1681a157612eSMichael Lange   MPI_Comm               comm;
1682a157612eSMichael Lange   PetscMPIInt            rank;
1683a157612eSMichael Lange   PetscSection           rootSection, leafSection;
1684a157612eSMichael Lange   IS                     rootrank, leafrank;
1685cf86098cSMatthew G. Knepley   DM                     dmCoord;
1686a9f1d5b2SMichael Lange   DMLabel                lblOverlap;
1687f5bf2dbfSMichael Lange   PetscSF                sfOverlap, sfStratified, sfPoint;
1688a157612eSMichael Lange   PetscErrorCode         ierr;
1689a157612eSMichael Lange 
1690a157612eSMichael Lange   PetscFunctionBegin;
1691a157612eSMichael Lange   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1692a157612eSMichael Lange   if (sf) PetscValidPointer(sf, 3);
1693a157612eSMichael Lange   PetscValidPointer(dmOverlap, 4);
1694a157612eSMichael Lange 
16951b858b30SMichael Lange   ierr = PetscLogEventBegin(DMPLEX_DistributeOverlap, dm, 0, 0, 0);CHKERRQ(ierr);
1696a157612eSMichael Lange   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
1697a157612eSMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
1698a157612eSMichael Lange 
1699a157612eSMichael Lange   /* Compute point overlap with neighbouring processes on the distributed DM */
1700a157612eSMichael Lange   ierr = PetscLogEventBegin(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
1701a157612eSMichael Lange   ierr = PetscSectionCreate(comm, &rootSection);CHKERRQ(ierr);
1702a157612eSMichael Lange   ierr = PetscSectionCreate(comm, &leafSection);CHKERRQ(ierr);
1703a157612eSMichael Lange   ierr = DMPlexDistributeOwnership(dm, rootSection, &rootrank, leafSection, &leafrank);CHKERRQ(ierr);
1704a9f1d5b2SMichael Lange   ierr = DMPlexCreateOverlap(dm, overlap, rootSection, rootrank, leafSection, leafrank, &lblOverlap);CHKERRQ(ierr);
1705a9f1d5b2SMichael Lange   /* Convert overlap label to stratified migration SF */
1706a9f1d5b2SMichael Lange   ierr = DMPlexPartitionLabelCreateSF(dm, lblOverlap, &sfOverlap);CHKERRQ(ierr);
1707a9f1d5b2SMichael Lange   ierr = DMPlexStratifyMigrationSF(dm, sfOverlap, &sfStratified);CHKERRQ(ierr);
1708a9f1d5b2SMichael Lange   ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr);
1709a9f1d5b2SMichael Lange   sfOverlap = sfStratified;
1710a9f1d5b2SMichael Lange   ierr = PetscObjectSetName((PetscObject) sfOverlap, "Overlap SF");CHKERRQ(ierr);
1711a9f1d5b2SMichael Lange   ierr = PetscSFSetFromOptions(sfOverlap);CHKERRQ(ierr);
1712a9f1d5b2SMichael Lange 
171315fff7beSMatthew G. Knepley   ierr = PetscSectionDestroy(&rootSection);CHKERRQ(ierr);
171415fff7beSMatthew G. Knepley   ierr = PetscSectionDestroy(&leafSection);CHKERRQ(ierr);
171515fff7beSMatthew G. Knepley   ierr = ISDestroy(&rootrank);CHKERRQ(ierr);
171615fff7beSMatthew G. Knepley   ierr = ISDestroy(&leafrank);CHKERRQ(ierr);
1717a157612eSMichael Lange   ierr = PetscLogEventEnd(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr);
1718a157612eSMichael Lange 
1719a157612eSMichael Lange   /* Build the overlapping DM */
1720a157612eSMichael Lange   ierr = DMPlexCreate(comm, dmOverlap);CHKERRQ(ierr);
1721a157612eSMichael Lange   ierr = PetscObjectSetName((PetscObject) *dmOverlap, "Parallel Mesh");CHKERRQ(ierr);
1722a9f1d5b2SMichael Lange   ierr = DMPlexMigrate(dm, sfOverlap, *dmOverlap);CHKERRQ(ierr);
1723f5bf2dbfSMichael Lange   /* Build the new point SF */
17241627f6ccSMichael Lange   ierr = DMPlexCreatePointSF(*dmOverlap, sfOverlap, PETSC_FALSE, &sfPoint);CHKERRQ(ierr);
1725f5bf2dbfSMichael Lange   ierr = DMSetPointSF(*dmOverlap, sfPoint);CHKERRQ(ierr);
1726cf86098cSMatthew G. Knepley   ierr = DMGetCoordinateDM(*dmOverlap, &dmCoord);CHKERRQ(ierr);
1727cf86098cSMatthew G. Knepley   if (dmCoord) {ierr = DMSetPointSF(dmCoord, sfPoint);CHKERRQ(ierr);}
1728f5bf2dbfSMichael Lange   ierr = PetscSFDestroy(&sfPoint);CHKERRQ(ierr);
1729a157612eSMichael Lange   /* Cleanup overlap partition */
1730a9f1d5b2SMichael Lange   ierr = DMLabelDestroy(&lblOverlap);CHKERRQ(ierr);
1731a9f1d5b2SMichael Lange   if (sf) *sf = sfOverlap;
1732a9f1d5b2SMichael Lange   else    {ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr);}
17331b858b30SMichael Lange   ierr = PetscLogEventEnd(DMPLEX_DistributeOverlap, dm, 0, 0, 0);CHKERRQ(ierr);
1734a157612eSMichael Lange   PetscFunctionReturn(0);
1735a157612eSMichael Lange }
1736