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 */ 613be200f8dSMichael Lange for (l = 1; l < levels; l++) { 614be200f8dSMichael Lange /* Propagate point donations over SF to capture remote connections */ 615be200f8dSMichael Lange ierr = DMPlexPartitionLabelPropagate(dm, ovAdjByRank);CHKERRQ(ierr); 616be200f8dSMichael Lange /* Add next level of point donations to the label */ 617be200f8dSMichael Lange ierr = DMPlexPartitionLabelAdjacency(dm, ovAdjByRank);CHKERRQ(ierr); 618be200f8dSMichael 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 } 629*71a8c5fcSMichael Lange /* Make global process SF and invert sender to receiver label */ 630*71a8c5fcSMichael Lange { 631*71a8c5fcSMichael Lange /* Build a global process SF */ 632*71a8c5fcSMichael Lange PetscSFNode *remoteProc; 633*71a8c5fcSMichael Lange ierr = PetscMalloc1(numProcs, &remoteProc);CHKERRQ(ierr); 634*71a8c5fcSMichael Lange for (p = 0; p < numProcs; ++p) { 635*71a8c5fcSMichael Lange remoteProc[p].rank = p; 636*71a8c5fcSMichael Lange remoteProc[p].index = rank; 637*71a8c5fcSMichael Lange } 638*71a8c5fcSMichael Lange ierr = PetscSFCreate(comm, &sfProc);CHKERRQ(ierr); 639*71a8c5fcSMichael Lange ierr = PetscObjectSetName((PetscObject) sfProc, "Process SF");CHKERRQ(ierr); 640*71a8c5fcSMichael Lange ierr = PetscSFSetGraph(sfProc, numProcs, numProcs, NULL, PETSC_OWN_POINTER, remoteProc, PETSC_OWN_POINTER);CHKERRQ(ierr); 641*71a8c5fcSMichael Lange } 642a9f1d5b2SMichael Lange ierr = DMLabelCreate("Overlap label", ovLabel);CHKERRQ(ierr); 643a9f1d5b2SMichael Lange ierr = DMPlexPartitionLabelInvert(dm, ovAdjByRank, sfProc, *ovLabel);CHKERRQ(ierr); 644a9f1d5b2SMichael Lange /* Add owned points, except for shared local points */ 645a9f1d5b2SMichael Lange for (p = pStart; p < pEnd; ++p) {ierr = DMLabelSetValue(*ovLabel, p, rank);CHKERRQ(ierr);} 646e540f424SMichael Lange for (l = 0; l < nleaves; ++l) { 647a9f1d5b2SMichael Lange ierr = DMLabelClearValue(*ovLabel, local[l], rank);CHKERRQ(ierr); 648a9f1d5b2SMichael Lange ierr = DMLabelSetValue(*ovLabel, remote[l].index, remote[l].rank);CHKERRQ(ierr); 649e540f424SMichael Lange } 650e540f424SMichael Lange /* Clean up */ 6511fd9873aSMichael Lange ierr = DMLabelDestroy(&ovAdjByRank);CHKERRQ(ierr); 652b0a623aaSMatthew G. Knepley ierr = PetscSFDestroy(&sfProc);CHKERRQ(ierr); 653b0a623aaSMatthew G. Knepley PetscFunctionReturn(0); 654b0a623aaSMatthew G. Knepley } 65570034214SMatthew G. Knepley 65670034214SMatthew G. Knepley #undef __FUNCT__ 65746f9b1c3SMichael Lange #define __FUNCT__ "DMPlexCreateOverlapMigrationSF" 65846f9b1c3SMichael Lange PetscErrorCode DMPlexCreateOverlapMigrationSF(DM dm, PetscSF overlapSF, PetscSF *migrationSF) 65946f9b1c3SMichael Lange { 66046f9b1c3SMichael Lange MPI_Comm comm; 66146f9b1c3SMichael Lange PetscMPIInt rank, numProcs; 66246f9b1c3SMichael Lange PetscInt d, dim, p, pStart, pEnd, nroots, nleaves, newLeaves, point, numSharedPoints; 66346f9b1c3SMichael Lange PetscInt *pointDepths, *remoteDepths, *ilocal; 66446f9b1c3SMichael Lange PetscInt *depthRecv, *depthShift, *depthIdx; 66546f9b1c3SMichael Lange PetscSFNode *iremote; 66646f9b1c3SMichael Lange PetscSF pointSF; 66746f9b1c3SMichael Lange const PetscInt *sharedLocal; 66846f9b1c3SMichael Lange const PetscSFNode *overlapRemote, *sharedRemote; 66946f9b1c3SMichael Lange PetscErrorCode ierr; 67046f9b1c3SMichael Lange 67146f9b1c3SMichael Lange PetscFunctionBegin; 67246f9b1c3SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 67346f9b1c3SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 67446f9b1c3SMichael Lange ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 67546f9b1c3SMichael Lange ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr); 67646f9b1c3SMichael Lange ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 67746f9b1c3SMichael Lange 67846f9b1c3SMichael Lange /* Before building the migration SF we need to know the new stratum offsets */ 67946f9b1c3SMichael Lange ierr = PetscSFGetGraph(overlapSF, &nroots, &nleaves, NULL, &overlapRemote);CHKERRQ(ierr); 68046f9b1c3SMichael Lange ierr = PetscMalloc2(nroots, &pointDepths, nleaves, &remoteDepths);CHKERRQ(ierr); 68146f9b1c3SMichael Lange for (d=0; d<dim+1; d++) { 68246f9b1c3SMichael Lange ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr); 68346f9b1c3SMichael Lange for (p=pStart; p<pEnd; p++) pointDepths[p] = d; 68446f9b1c3SMichael Lange } 68546f9b1c3SMichael Lange for (p=0; p<nleaves; p++) remoteDepths[p] = -1; 68646f9b1c3SMichael Lange ierr = PetscSFBcastBegin(overlapSF, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr); 68746f9b1c3SMichael Lange ierr = PetscSFBcastEnd(overlapSF, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr); 68846f9b1c3SMichael Lange 68946f9b1c3SMichael Lange /* Count recevied points in each stratum and compute the internal strata shift */ 69046f9b1c3SMichael Lange ierr = PetscMalloc3(dim+1, &depthRecv, dim+1, &depthShift, dim+1, &depthIdx);CHKERRQ(ierr); 69146f9b1c3SMichael Lange for (d=0; d<dim+1; d++) depthRecv[d]=0; 69246f9b1c3SMichael Lange for (p=0; p<nleaves; p++) depthRecv[remoteDepths[p]]++; 69346f9b1c3SMichael Lange depthShift[dim] = 0; 69446f9b1c3SMichael Lange for (d=0; d<dim; d++) depthShift[d] = depthRecv[dim]; 69546f9b1c3SMichael Lange for (d=1; d<dim; d++) depthShift[d] += depthRecv[0]; 69646f9b1c3SMichael Lange for (d=dim-2; d>0; d--) depthShift[d] += depthRecv[d+1]; 69746f9b1c3SMichael Lange for (d=0; d<dim+1; d++) { 69846f9b1c3SMichael Lange ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr); 69946f9b1c3SMichael Lange depthIdx[d] = pStart + depthShift[d]; 70046f9b1c3SMichael Lange } 70146f9b1c3SMichael Lange 70246f9b1c3SMichael Lange /* Form the overlap SF build an SF that describes the full overlap migration SF */ 70346f9b1c3SMichael Lange ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr); 70446f9b1c3SMichael Lange newLeaves = pEnd - pStart + nleaves; 70509b7985cSMatthew G. Knepley ierr = PetscMalloc1(newLeaves, &ilocal);CHKERRQ(ierr); 70609b7985cSMatthew G. Knepley ierr = PetscMalloc1(newLeaves, &iremote);CHKERRQ(ierr); 70746f9b1c3SMichael Lange /* First map local points to themselves */ 70846f9b1c3SMichael Lange for (d=0; d<dim+1; d++) { 70946f9b1c3SMichael Lange ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr); 71046f9b1c3SMichael Lange for (p=pStart; p<pEnd; p++) { 71146f9b1c3SMichael Lange point = p + depthShift[d]; 71246f9b1c3SMichael Lange ilocal[point] = point; 71346f9b1c3SMichael Lange iremote[point].index = p; 71446f9b1c3SMichael Lange iremote[point].rank = rank; 71546f9b1c3SMichael Lange depthIdx[d]++; 71646f9b1c3SMichael Lange } 71746f9b1c3SMichael Lange } 71846f9b1c3SMichael Lange 71946f9b1c3SMichael Lange /* Add in the remote roots for currently shared points */ 72046f9b1c3SMichael Lange ierr = DMGetPointSF(dm, &pointSF);CHKERRQ(ierr); 72146f9b1c3SMichael Lange ierr = PetscSFGetGraph(pointSF, NULL, &numSharedPoints, &sharedLocal, &sharedRemote);CHKERRQ(ierr); 72246f9b1c3SMichael Lange for (d=0; d<dim+1; d++) { 72346f9b1c3SMichael Lange ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr); 72446f9b1c3SMichael Lange for (p=0; p<numSharedPoints; p++) { 72546f9b1c3SMichael Lange if (pStart <= sharedLocal[p] && sharedLocal[p] < pEnd) { 72646f9b1c3SMichael Lange point = sharedLocal[p] + depthShift[d]; 72746f9b1c3SMichael Lange iremote[point].index = sharedRemote[p].index; 72846f9b1c3SMichael Lange iremote[point].rank = sharedRemote[p].rank; 72946f9b1c3SMichael Lange } 73046f9b1c3SMichael Lange } 73146f9b1c3SMichael Lange } 73246f9b1c3SMichael Lange 73346f9b1c3SMichael Lange /* Now add the incoming overlap points */ 73446f9b1c3SMichael Lange for (p=0; p<nleaves; p++) { 73546f9b1c3SMichael Lange point = depthIdx[remoteDepths[p]]; 73646f9b1c3SMichael Lange ilocal[point] = point; 73746f9b1c3SMichael Lange iremote[point].index = overlapRemote[p].index; 73846f9b1c3SMichael Lange iremote[point].rank = overlapRemote[p].rank; 73946f9b1c3SMichael Lange depthIdx[remoteDepths[p]]++; 74046f9b1c3SMichael Lange } 74115fff7beSMatthew G. Knepley ierr = PetscFree2(pointDepths,remoteDepths);CHKERRQ(ierr); 74246f9b1c3SMichael Lange 74346f9b1c3SMichael Lange ierr = PetscSFCreate(comm, migrationSF);CHKERRQ(ierr); 74446f9b1c3SMichael Lange ierr = PetscObjectSetName((PetscObject) *migrationSF, "Overlap Migration SF");CHKERRQ(ierr); 74546f9b1c3SMichael Lange ierr = PetscSFSetFromOptions(*migrationSF);CHKERRQ(ierr); 74646f9b1c3SMichael Lange ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr); 74746f9b1c3SMichael Lange ierr = PetscSFSetGraph(*migrationSF, pEnd-pStart, newLeaves, ilocal, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER);CHKERRQ(ierr); 74846f9b1c3SMichael Lange 74946f9b1c3SMichael Lange ierr = PetscFree3(depthRecv, depthShift, depthIdx);CHKERRQ(ierr); 75070034214SMatthew G. Knepley PetscFunctionReturn(0); 75170034214SMatthew G. Knepley } 75270034214SMatthew G. Knepley 75370034214SMatthew G. Knepley #undef __FUNCT__ 754a9f1d5b2SMichael Lange #define __FUNCT__ "DMPlexStratifyMigrationSF" 755a9f1d5b2SMichael Lange /*@ 756a9f1d5b2SMichael Lange DMPlexStratifyMigrationSF - Add partition overlap to a distributed non-overlapping DM. 757a9f1d5b2SMichael Lange 758a9f1d5b2SMichael Lange Input Parameter: 759a9f1d5b2SMichael Lange + dm - The DM 760a9f1d5b2SMichael Lange - sf - A star forest with non-ordered leaves, usually defining a DM point migration 761a9f1d5b2SMichael Lange 762a9f1d5b2SMichael Lange Output Parameter: 763a9f1d5b2SMichael Lange . migrationSF - A star forest with added leaf indirection that ensures the resulting DM is stratified 764a9f1d5b2SMichael Lange 765a9f1d5b2SMichael Lange Level: developer 766a9f1d5b2SMichael Lange 767a9f1d5b2SMichael Lange .seealso: DMPlexPartitionLabelCreateSF(), DMPlexDistribute(), DMPlexDistributeOverlap() 768a9f1d5b2SMichael Lange @*/ 769a9f1d5b2SMichael Lange PetscErrorCode DMPlexStratifyMigrationSF(DM dm, PetscSF sf, PetscSF *migrationSF) 770a9f1d5b2SMichael Lange { 771a9f1d5b2SMichael Lange MPI_Comm comm; 772a9f1d5b2SMichael Lange PetscMPIInt rank, numProcs; 7737fab53ddSMatthew G. Knepley PetscInt d, ldepth, depth, p, pStart, pEnd, nroots, nleaves; 774a9f1d5b2SMichael Lange PetscInt *pointDepths, *remoteDepths, *ilocal; 775a9f1d5b2SMichael Lange PetscInt *depthRecv, *depthShift, *depthIdx; 776a9f1d5b2SMichael Lange const PetscSFNode *iremote; 777a9f1d5b2SMichael Lange PetscErrorCode ierr; 778a9f1d5b2SMichael Lange 779a9f1d5b2SMichael Lange PetscFunctionBegin; 780a9f1d5b2SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 781a9f1d5b2SMichael Lange ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr); 782a9f1d5b2SMichael Lange ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 783a9f1d5b2SMichael Lange ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr); 7847fab53ddSMatthew G. Knepley ierr = DMPlexGetDepth(dm, &ldepth);CHKERRQ(ierr); 785b2566f29SBarry Smith ierr = MPIU_Allreduce(&ldepth, &depth, 1, MPIU_INT, MPI_MAX, comm);CHKERRQ(ierr); 7867fab53ddSMatthew G. Knepley if ((ldepth >= 0) && (depth != ldepth)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Inconsistent Plex depth %d != %d", ldepth, depth); 787a9f1d5b2SMichael Lange 788a9f1d5b2SMichael Lange /* Before building the migration SF we need to know the new stratum offsets */ 789a9f1d5b2SMichael Lange ierr = PetscSFGetGraph(sf, &nroots, &nleaves, NULL, &iremote);CHKERRQ(ierr); 790a9f1d5b2SMichael Lange ierr = PetscMalloc2(nroots, &pointDepths, nleaves, &remoteDepths);CHKERRQ(ierr); 7917fab53ddSMatthew G. Knepley for (d = 0; d < depth+1; ++d) { 792a9f1d5b2SMichael Lange ierr = DMPlexGetDepthStratum(dm, d, &pStart, &pEnd);CHKERRQ(ierr); 7937fab53ddSMatthew G. Knepley for (p = pStart; p < pEnd; ++p) pointDepths[p] = d; 794a9f1d5b2SMichael Lange } 7957fab53ddSMatthew G. Knepley for (p = 0; p < nleaves; ++p) remoteDepths[p] = -1; 796a9f1d5b2SMichael Lange ierr = PetscSFBcastBegin(sf, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr); 797a9f1d5b2SMichael Lange ierr = PetscSFBcastEnd(sf, MPIU_INT, pointDepths, remoteDepths);CHKERRQ(ierr); 798a9f1d5b2SMichael Lange /* Count recevied points in each stratum and compute the internal strata shift */ 7997fab53ddSMatthew G. Knepley ierr = PetscMalloc3(depth+1, &depthRecv, depth+1, &depthShift, depth+1, &depthIdx);CHKERRQ(ierr); 8007fab53ddSMatthew G. Knepley for (d = 0; d < depth+1; ++d) depthRecv[d] = 0; 8017fab53ddSMatthew G. Knepley for (p = 0; p < nleaves; ++p) depthRecv[remoteDepths[p]]++; 8027fab53ddSMatthew G. Knepley depthShift[depth] = 0; 8037fab53ddSMatthew G. Knepley for (d = 0; d < depth; ++d) depthShift[d] = depthRecv[depth]; 8047fab53ddSMatthew G. Knepley for (d = 1; d < depth; ++d) depthShift[d] += depthRecv[0]; 8057fab53ddSMatthew G. Knepley for (d = depth-2; d > 0; --d) depthShift[d] += depthRecv[d+1]; 8067fab53ddSMatthew G. Knepley for (d = 0; d < depth+1; ++d) {depthIdx[d] = 0;} 807a9f1d5b2SMichael Lange /* Derive a new local permutation based on stratified indices */ 808a9f1d5b2SMichael Lange ierr = PetscMalloc1(nleaves, &ilocal);CHKERRQ(ierr); 8097fab53ddSMatthew G. Knepley for (p = 0; p < nleaves; ++p) { 8107fab53ddSMatthew G. Knepley const PetscInt dep = remoteDepths[p]; 8117fab53ddSMatthew G. Knepley 8127fab53ddSMatthew G. Knepley ilocal[p] = depthShift[dep] + depthIdx[dep]; 8137fab53ddSMatthew G. Knepley depthIdx[dep]++; 814a9f1d5b2SMichael Lange } 815a9f1d5b2SMichael Lange ierr = PetscSFCreate(comm, migrationSF);CHKERRQ(ierr); 816a9f1d5b2SMichael Lange ierr = PetscObjectSetName((PetscObject) *migrationSF, "Migration SF");CHKERRQ(ierr); 817a9f1d5b2SMichael Lange ierr = PetscSFSetGraph(*migrationSF, nroots, nleaves, ilocal, PETSC_OWN_POINTER, iremote, PETSC_COPY_VALUES);CHKERRQ(ierr); 818a9f1d5b2SMichael Lange ierr = PetscFree2(pointDepths,remoteDepths);CHKERRQ(ierr); 819a9f1d5b2SMichael Lange ierr = PetscFree3(depthRecv, depthShift, depthIdx);CHKERRQ(ierr); 820a9f1d5b2SMichael Lange PetscFunctionReturn(0); 821a9f1d5b2SMichael Lange } 822a9f1d5b2SMichael Lange 823a9f1d5b2SMichael Lange #undef __FUNCT__ 82470034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeField" 82570034214SMatthew G. Knepley /*@ 82670034214SMatthew G. Knepley DMPlexDistributeField - Distribute field data to match a given PetscSF, usually the SF from mesh distribution 82770034214SMatthew G. Knepley 82870034214SMatthew G. Knepley Collective on DM 82970034214SMatthew G. Knepley 83070034214SMatthew G. Knepley Input Parameters: 83170034214SMatthew G. Knepley + dm - The DMPlex object 83270034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern 83370034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout 83470034214SMatthew G. Knepley - originalVec - The existing data 83570034214SMatthew G. Knepley 83670034214SMatthew G. Knepley Output Parameters: 83770034214SMatthew G. Knepley + newSection - The PetscSF describing the new data layout 83870034214SMatthew G. Knepley - newVec - The new data 83970034214SMatthew G. Knepley 84070034214SMatthew G. Knepley Level: developer 84170034214SMatthew G. Knepley 84270034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeFieldIS(), DMPlexDistributeData() 84370034214SMatthew G. Knepley @*/ 84470034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeField(DM dm, PetscSF pointSF, PetscSection originalSection, Vec originalVec, PetscSection newSection, Vec newVec) 84570034214SMatthew G. Knepley { 84670034214SMatthew G. Knepley PetscSF fieldSF; 84770034214SMatthew G. Knepley PetscInt *remoteOffsets, fieldSize; 84870034214SMatthew G. Knepley PetscScalar *originalValues, *newValues; 84970034214SMatthew G. Knepley PetscErrorCode ierr; 85070034214SMatthew G. Knepley 85170034214SMatthew G. Knepley PetscFunctionBegin; 85270034214SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr); 85370034214SMatthew G. Knepley ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr); 85470034214SMatthew G. Knepley 85570034214SMatthew G. Knepley ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr); 85670034214SMatthew G. Knepley ierr = VecSetSizes(newVec, fieldSize, PETSC_DETERMINE);CHKERRQ(ierr); 85770034214SMatthew G. Knepley ierr = VecSetType(newVec,dm->vectype);CHKERRQ(ierr); 85870034214SMatthew G. Knepley 85970034214SMatthew G. Knepley ierr = VecGetArray(originalVec, &originalValues);CHKERRQ(ierr); 86070034214SMatthew G. Knepley ierr = VecGetArray(newVec, &newValues);CHKERRQ(ierr); 86170034214SMatthew G. Knepley ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr); 8628a713f3bSLawrence Mitchell ierr = PetscFree(remoteOffsets);CHKERRQ(ierr); 86370034214SMatthew G. Knepley ierr = PetscSFBcastBegin(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr); 86470034214SMatthew G. Knepley ierr = PetscSFBcastEnd(fieldSF, MPIU_SCALAR, originalValues, newValues);CHKERRQ(ierr); 86570034214SMatthew G. Knepley ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr); 86670034214SMatthew G. Knepley ierr = VecRestoreArray(newVec, &newValues);CHKERRQ(ierr); 86770034214SMatthew G. Knepley ierr = VecRestoreArray(originalVec, &originalValues);CHKERRQ(ierr); 86870034214SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr); 86970034214SMatthew G. Knepley PetscFunctionReturn(0); 87070034214SMatthew G. Knepley } 87170034214SMatthew G. Knepley 87270034214SMatthew G. Knepley #undef __FUNCT__ 87370034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeFieldIS" 87470034214SMatthew G. Knepley /*@ 87570034214SMatthew G. Knepley DMPlexDistributeFieldIS - Distribute field data to match a given PetscSF, usually the SF from mesh distribution 87670034214SMatthew G. Knepley 87770034214SMatthew G. Knepley Collective on DM 87870034214SMatthew G. Knepley 87970034214SMatthew G. Knepley Input Parameters: 88070034214SMatthew G. Knepley + dm - The DMPlex object 88170034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern 88270034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout 88370034214SMatthew G. Knepley - originalIS - The existing data 88470034214SMatthew G. Knepley 88570034214SMatthew G. Knepley Output Parameters: 88670034214SMatthew G. Knepley + newSection - The PetscSF describing the new data layout 88770034214SMatthew G. Knepley - newIS - The new data 88870034214SMatthew G. Knepley 88970034214SMatthew G. Knepley Level: developer 89070034214SMatthew G. Knepley 89170034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField(), DMPlexDistributeData() 89270034214SMatthew G. Knepley @*/ 89370034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeFieldIS(DM dm, PetscSF pointSF, PetscSection originalSection, IS originalIS, PetscSection newSection, IS *newIS) 89470034214SMatthew G. Knepley { 89570034214SMatthew G. Knepley PetscSF fieldSF; 89670034214SMatthew G. Knepley PetscInt *newValues, *remoteOffsets, fieldSize; 89770034214SMatthew G. Knepley const PetscInt *originalValues; 89870034214SMatthew G. Knepley PetscErrorCode ierr; 89970034214SMatthew G. Knepley 90070034214SMatthew G. Knepley PetscFunctionBegin; 90170034214SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr); 90270034214SMatthew G. Knepley ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr); 90370034214SMatthew G. Knepley 90470034214SMatthew G. Knepley ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr); 905854ce69bSBarry Smith ierr = PetscMalloc1(fieldSize, &newValues);CHKERRQ(ierr); 90670034214SMatthew G. Knepley 90770034214SMatthew G. Knepley ierr = ISGetIndices(originalIS, &originalValues);CHKERRQ(ierr); 90870034214SMatthew G. Knepley ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr); 9098a713f3bSLawrence Mitchell ierr = PetscFree(remoteOffsets);CHKERRQ(ierr); 910aaf8c182SMatthew G. Knepley ierr = PetscSFBcastBegin(fieldSF, MPIU_INT, (PetscInt *) originalValues, newValues);CHKERRQ(ierr); 911aaf8c182SMatthew G. Knepley ierr = PetscSFBcastEnd(fieldSF, MPIU_INT, (PetscInt *) originalValues, newValues);CHKERRQ(ierr); 91270034214SMatthew G. Knepley ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr); 91370034214SMatthew G. Knepley ierr = ISRestoreIndices(originalIS, &originalValues);CHKERRQ(ierr); 91470034214SMatthew G. Knepley ierr = ISCreateGeneral(PetscObjectComm((PetscObject) pointSF), fieldSize, newValues, PETSC_OWN_POINTER, newIS);CHKERRQ(ierr); 91570034214SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_DistributeField,dm,0,0,0);CHKERRQ(ierr); 91670034214SMatthew G. Knepley PetscFunctionReturn(0); 91770034214SMatthew G. Knepley } 91870034214SMatthew G. Knepley 91970034214SMatthew G. Knepley #undef __FUNCT__ 92070034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistributeData" 92170034214SMatthew G. Knepley /*@ 92270034214SMatthew G. Knepley DMPlexDistributeData - Distribute field data to match a given PetscSF, usually the SF from mesh distribution 92370034214SMatthew G. Knepley 92470034214SMatthew G. Knepley Collective on DM 92570034214SMatthew G. Knepley 92670034214SMatthew G. Knepley Input Parameters: 92770034214SMatthew G. Knepley + dm - The DMPlex object 92870034214SMatthew G. Knepley . pointSF - The PetscSF describing the communication pattern 92970034214SMatthew G. Knepley . originalSection - The PetscSection for existing data layout 93070034214SMatthew G. Knepley . datatype - The type of data 93170034214SMatthew G. Knepley - originalData - The existing data 93270034214SMatthew G. Knepley 93370034214SMatthew G. Knepley Output Parameters: 93470034214SMatthew G. Knepley + newSection - The PetscSection describing the new data layout 93570034214SMatthew G. Knepley - newData - The new data 93670034214SMatthew G. Knepley 93770034214SMatthew G. Knepley Level: developer 93870034214SMatthew G. Knepley 93970034214SMatthew G. Knepley .seealso: DMPlexDistribute(), DMPlexDistributeField() 94070034214SMatthew G. Knepley @*/ 94170034214SMatthew G. Knepley PetscErrorCode DMPlexDistributeData(DM dm, PetscSF pointSF, PetscSection originalSection, MPI_Datatype datatype, void *originalData, PetscSection newSection, void **newData) 94270034214SMatthew G. Knepley { 94370034214SMatthew G. Knepley PetscSF fieldSF; 94470034214SMatthew G. Knepley PetscInt *remoteOffsets, fieldSize; 94570034214SMatthew G. Knepley PetscMPIInt dataSize; 94670034214SMatthew G. Knepley PetscErrorCode ierr; 94770034214SMatthew G. Knepley 94870034214SMatthew G. Knepley PetscFunctionBegin; 94970034214SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr); 95070034214SMatthew G. Knepley ierr = PetscSFDistributeSection(pointSF, originalSection, &remoteOffsets, newSection);CHKERRQ(ierr); 95170034214SMatthew G. Knepley 95270034214SMatthew G. Knepley ierr = PetscSectionGetStorageSize(newSection, &fieldSize);CHKERRQ(ierr); 95370034214SMatthew G. Knepley ierr = MPI_Type_size(datatype, &dataSize);CHKERRQ(ierr); 95470034214SMatthew G. Knepley ierr = PetscMalloc(fieldSize * dataSize, newData);CHKERRQ(ierr); 95570034214SMatthew G. Knepley 95670034214SMatthew G. Knepley ierr = PetscSFCreateSectionSF(pointSF, originalSection, remoteOffsets, newSection, &fieldSF);CHKERRQ(ierr); 9578a713f3bSLawrence Mitchell ierr = PetscFree(remoteOffsets);CHKERRQ(ierr); 95870034214SMatthew G. Knepley ierr = PetscSFBcastBegin(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr); 95970034214SMatthew G. Knepley ierr = PetscSFBcastEnd(fieldSF, datatype, originalData, *newData);CHKERRQ(ierr); 96070034214SMatthew G. Knepley ierr = PetscSFDestroy(&fieldSF);CHKERRQ(ierr); 96170034214SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_DistributeData,dm,0,0,0);CHKERRQ(ierr); 96270034214SMatthew G. Knepley PetscFunctionReturn(0); 96370034214SMatthew G. Knepley } 96470034214SMatthew G. Knepley 96570034214SMatthew G. Knepley #undef __FUNCT__ 966cc71bff1SMichael Lange #define __FUNCT__ "DMPlexDistributeCones" 967ac04eaf7SToby Isaac PetscErrorCode DMPlexDistributeCones(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping original, ISLocalToGlobalMapping renumbering, DM dmParallel) 968cc71bff1SMichael Lange { 969f5bf2dbfSMichael Lange DM_Plex *mesh = (DM_Plex*) dm->data; 970cc71bff1SMichael Lange DM_Plex *pmesh = (DM_Plex*) (dmParallel)->data; 971cc71bff1SMichael Lange MPI_Comm comm; 972cc71bff1SMichael Lange PetscSF coneSF; 973cc71bff1SMichael Lange PetscSection originalConeSection, newConeSection; 974ac04eaf7SToby Isaac PetscInt *remoteOffsets, *cones, *globCones, *newCones, newConesSize; 975cc71bff1SMichael Lange PetscBool flg; 976cc71bff1SMichael Lange PetscErrorCode ierr; 977cc71bff1SMichael Lange 978cc71bff1SMichael Lange PetscFunctionBegin; 979cc71bff1SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 980cc71bff1SMichael Lange PetscValidPointer(dmParallel,4); 981cc71bff1SMichael Lange ierr = PetscLogEventBegin(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr); 982cc71bff1SMichael Lange 983cc71bff1SMichael Lange /* Distribute cone section */ 984cc71bff1SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 985cc71bff1SMichael Lange ierr = DMPlexGetConeSection(dm, &originalConeSection);CHKERRQ(ierr); 986cc71bff1SMichael Lange ierr = DMPlexGetConeSection(dmParallel, &newConeSection);CHKERRQ(ierr); 987cc71bff1SMichael Lange ierr = PetscSFDistributeSection(migrationSF, originalConeSection, &remoteOffsets, newConeSection);CHKERRQ(ierr); 988cc71bff1SMichael Lange ierr = DMSetUp(dmParallel);CHKERRQ(ierr); 989cc71bff1SMichael Lange { 990cc71bff1SMichael Lange PetscInt pStart, pEnd, p; 991cc71bff1SMichael Lange 992cc71bff1SMichael Lange ierr = PetscSectionGetChart(newConeSection, &pStart, &pEnd);CHKERRQ(ierr); 993cc71bff1SMichael Lange for (p = pStart; p < pEnd; ++p) { 994cc71bff1SMichael Lange PetscInt coneSize; 995cc71bff1SMichael Lange ierr = PetscSectionGetDof(newConeSection, p, &coneSize);CHKERRQ(ierr); 996cc71bff1SMichael Lange pmesh->maxConeSize = PetscMax(pmesh->maxConeSize, coneSize); 997cc71bff1SMichael Lange } 998cc71bff1SMichael Lange } 999cc71bff1SMichael Lange /* Communicate and renumber cones */ 1000cc71bff1SMichael Lange ierr = PetscSFCreateSectionSF(migrationSF, originalConeSection, remoteOffsets, newConeSection, &coneSF);CHKERRQ(ierr); 10018a713f3bSLawrence Mitchell ierr = PetscFree(remoteOffsets);CHKERRQ(ierr); 1002cc71bff1SMichael Lange ierr = DMPlexGetCones(dm, &cones);CHKERRQ(ierr); 1003ac04eaf7SToby Isaac if (original) { 1004ac04eaf7SToby Isaac PetscInt numCones; 1005ac04eaf7SToby Isaac 1006ac04eaf7SToby Isaac ierr = PetscSectionGetStorageSize(originalConeSection,&numCones);CHKERRQ(ierr); ierr = PetscMalloc1(numCones,&globCones);CHKERRQ(ierr); 1007ac04eaf7SToby Isaac ierr = ISLocalToGlobalMappingApplyBlock(original, numCones, cones, globCones);CHKERRQ(ierr); 1008ac04eaf7SToby Isaac } 1009ac04eaf7SToby Isaac else { 1010ac04eaf7SToby Isaac globCones = cones; 1011ac04eaf7SToby Isaac } 1012cc71bff1SMichael Lange ierr = DMPlexGetCones(dmParallel, &newCones);CHKERRQ(ierr); 1013ac04eaf7SToby Isaac ierr = PetscSFBcastBegin(coneSF, MPIU_INT, globCones, newCones);CHKERRQ(ierr); 1014ac04eaf7SToby Isaac ierr = PetscSFBcastEnd(coneSF, MPIU_INT, globCones, newCones);CHKERRQ(ierr); 1015ac04eaf7SToby Isaac if (original) { 1016ac04eaf7SToby Isaac ierr = PetscFree(globCones);CHKERRQ(ierr); 1017ac04eaf7SToby Isaac } 1018cc71bff1SMichael Lange ierr = PetscSectionGetStorageSize(newConeSection, &newConesSize);CHKERRQ(ierr); 1019cc71bff1SMichael Lange ierr = ISGlobalToLocalMappingApplyBlock(renumbering, IS_GTOLM_MASK, newConesSize, newCones, NULL, newCones);CHKERRQ(ierr); 10203533c52bSMatthew G. Knepley #if PETSC_USE_DEBUG 10213533c52bSMatthew G. Knepley { 10223533c52bSMatthew G. Knepley PetscInt p; 10233533c52bSMatthew G. Knepley PetscBool valid = PETSC_TRUE; 10243533c52bSMatthew G. Knepley for (p = 0; p < newConesSize; ++p) { 10253533c52bSMatthew G. Knepley if (newCones[p] < 0) {valid = PETSC_FALSE; ierr = PetscPrintf(PETSC_COMM_SELF, "Point %d not in overlap SF\n", p);CHKERRQ(ierr);} 10263533c52bSMatthew G. Knepley } 10273533c52bSMatthew G. Knepley if (!valid) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid global to local map"); 10283533c52bSMatthew G. Knepley } 10293533c52bSMatthew G. Knepley #endif 1030c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-cones_view", &flg);CHKERRQ(ierr); 1031cc71bff1SMichael Lange if (flg) { 1032cc71bff1SMichael Lange ierr = PetscPrintf(comm, "Serial Cone Section:\n");CHKERRQ(ierr); 1033cc71bff1SMichael Lange ierr = PetscSectionView(originalConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 1034cc71bff1SMichael Lange ierr = PetscPrintf(comm, "Parallel Cone Section:\n");CHKERRQ(ierr); 1035cc71bff1SMichael Lange ierr = PetscSectionView(newConeSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 1036cc71bff1SMichael Lange ierr = PetscSFView(coneSF, NULL);CHKERRQ(ierr); 1037cc71bff1SMichael Lange } 1038cc71bff1SMichael Lange ierr = DMPlexGetConeOrientations(dm, &cones);CHKERRQ(ierr); 1039cc71bff1SMichael Lange ierr = DMPlexGetConeOrientations(dmParallel, &newCones);CHKERRQ(ierr); 1040cc71bff1SMichael Lange ierr = PetscSFBcastBegin(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr); 1041cc71bff1SMichael Lange ierr = PetscSFBcastEnd(coneSF, MPIU_INT, cones, newCones);CHKERRQ(ierr); 1042cc71bff1SMichael Lange ierr = PetscSFDestroy(&coneSF);CHKERRQ(ierr); 1043cc71bff1SMichael Lange ierr = PetscLogEventEnd(DMPLEX_DistributeCones,dm,0,0,0);CHKERRQ(ierr); 1044cc71bff1SMichael Lange /* Create supports and stratify sieve */ 1045cc71bff1SMichael Lange { 1046cc71bff1SMichael Lange PetscInt pStart, pEnd; 1047cc71bff1SMichael Lange 1048cc71bff1SMichael Lange ierr = PetscSectionGetChart(pmesh->coneSection, &pStart, &pEnd);CHKERRQ(ierr); 1049cc71bff1SMichael Lange ierr = PetscSectionSetChart(pmesh->supportSection, pStart, pEnd);CHKERRQ(ierr); 1050cc71bff1SMichael Lange } 1051cc71bff1SMichael Lange ierr = DMPlexSymmetrize(dmParallel);CHKERRQ(ierr); 1052cc71bff1SMichael Lange ierr = DMPlexStratify(dmParallel);CHKERRQ(ierr); 1053f5bf2dbfSMichael Lange pmesh->useCone = mesh->useCone; 1054f5bf2dbfSMichael Lange pmesh->useClosure = mesh->useClosure; 1055cc71bff1SMichael Lange PetscFunctionReturn(0); 1056cc71bff1SMichael Lange } 1057cc71bff1SMichael Lange 1058cc71bff1SMichael Lange #undef __FUNCT__ 10590df0e737SMichael Lange #define __FUNCT__ "DMPlexDistributeCoordinates" 10600df0e737SMichael Lange PetscErrorCode DMPlexDistributeCoordinates(DM dm, PetscSF migrationSF, DM dmParallel) 10610df0e737SMichael Lange { 10620df0e737SMichael Lange MPI_Comm comm; 10630df0e737SMichael Lange PetscSection originalCoordSection, newCoordSection; 10640df0e737SMichael Lange Vec originalCoordinates, newCoordinates; 10650df0e737SMichael Lange PetscInt bs; 10660df0e737SMichael Lange const char *name; 10670df0e737SMichael Lange const PetscReal *maxCell, *L; 10685dc8c3f7SMatthew G. Knepley const DMBoundaryType *bd; 10690df0e737SMichael Lange PetscErrorCode ierr; 10700df0e737SMichael Lange 10710df0e737SMichael Lange PetscFunctionBegin; 10720df0e737SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 10730df0e737SMichael Lange PetscValidPointer(dmParallel, 3); 10740df0e737SMichael Lange 10750df0e737SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 10760df0e737SMichael Lange ierr = DMGetCoordinateSection(dm, &originalCoordSection);CHKERRQ(ierr); 10770df0e737SMichael Lange ierr = DMGetCoordinateSection(dmParallel, &newCoordSection);CHKERRQ(ierr); 10780df0e737SMichael Lange ierr = DMGetCoordinatesLocal(dm, &originalCoordinates);CHKERRQ(ierr); 10790df0e737SMichael Lange if (originalCoordinates) { 10800df0e737SMichael Lange ierr = VecCreate(comm, &newCoordinates);CHKERRQ(ierr); 10810df0e737SMichael Lange ierr = PetscObjectGetName((PetscObject) originalCoordinates, &name);CHKERRQ(ierr); 10820df0e737SMichael Lange ierr = PetscObjectSetName((PetscObject) newCoordinates, name);CHKERRQ(ierr); 10830df0e737SMichael Lange 10840df0e737SMichael Lange ierr = DMPlexDistributeField(dm, migrationSF, originalCoordSection, originalCoordinates, newCoordSection, newCoordinates);CHKERRQ(ierr); 10850df0e737SMichael Lange ierr = DMSetCoordinatesLocal(dmParallel, newCoordinates);CHKERRQ(ierr); 10860df0e737SMichael Lange ierr = VecGetBlockSize(originalCoordinates, &bs);CHKERRQ(ierr); 10870df0e737SMichael Lange ierr = VecSetBlockSize(newCoordinates, bs);CHKERRQ(ierr); 10880df0e737SMichael Lange ierr = VecDestroy(&newCoordinates);CHKERRQ(ierr); 10890df0e737SMichael Lange } 10905dc8c3f7SMatthew G. Knepley ierr = DMGetPeriodicity(dm, &maxCell, &L, &bd);CHKERRQ(ierr); 10915dc8c3f7SMatthew G. Knepley if (L) {ierr = DMSetPeriodicity(dmParallel, maxCell, L, bd);CHKERRQ(ierr);} 10920df0e737SMichael Lange PetscFunctionReturn(0); 10930df0e737SMichael Lange } 10940df0e737SMichael Lange 10950df0e737SMichael Lange #undef __FUNCT__ 10960df0e737SMichael Lange #define __FUNCT__ "DMPlexDistributeLabels" 1097d995df53SMatthew G. Knepley /* Here we are assuming that process 0 always has everything */ 10982eb1fa14SMichael Lange PetscErrorCode DMPlexDistributeLabels(DM dm, PetscSF migrationSF, DM dmParallel) 10990df0e737SMichael Lange { 1100df0420ecSMatthew G. Knepley DM_Plex *mesh = (DM_Plex*) dm->data; 11010df0e737SMichael Lange MPI_Comm comm; 1102df0420ecSMatthew G. Knepley DMLabel depth; 11030df0e737SMichael Lange PetscMPIInt rank; 1104d995df53SMatthew G. Knepley PetscInt numLabels, numLocalLabels, l; 1105df0420ecSMatthew G. Knepley PetscBool hasLabels = PETSC_FALSE, lsendDepth, sendDepth; 1106df0420ecSMatthew G. Knepley PetscObjectState depthState = -1; 11070df0e737SMichael Lange PetscErrorCode ierr; 11080df0e737SMichael Lange 11090df0e737SMichael Lange PetscFunctionBegin; 11100df0e737SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 11112eb1fa14SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 3); 11120df0e737SMichael Lange ierr = PetscLogEventBegin(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr); 11130df0e737SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 11140df0e737SMichael Lange ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 11150df0e737SMichael Lange 1116df0420ecSMatthew G. Knepley /* If the user has changed the depth label, communicate it instead */ 1117df0420ecSMatthew G. Knepley ierr = DMPlexGetDepthLabel(dm, &depth);CHKERRQ(ierr); 1118df0420ecSMatthew G. Knepley if (depth) {ierr = DMLabelGetState(depth, &depthState);CHKERRQ(ierr);} 1119df0420ecSMatthew G. Knepley lsendDepth = mesh->depthState != depthState ? PETSC_TRUE : PETSC_FALSE; 1120b2566f29SBarry Smith ierr = MPIU_Allreduce(&lsendDepth, &sendDepth, 1, MPIU_BOOL, MPI_LOR, comm);CHKERRQ(ierr); 1121df0420ecSMatthew G. Knepley if (sendDepth) { 1122c58f1c22SToby Isaac ierr = DMRemoveLabel(dmParallel, "depth", &depth);CHKERRQ(ierr); 1123df0420ecSMatthew G. Knepley ierr = DMLabelDestroy(&depth);CHKERRQ(ierr); 1124df0420ecSMatthew G. Knepley } 1125d995df53SMatthew G. Knepley /* Everyone must have either the same number of labels, or none */ 1126c58f1c22SToby Isaac ierr = DMGetNumLabels(dm, &numLocalLabels);CHKERRQ(ierr); 1127d995df53SMatthew G. Knepley numLabels = numLocalLabels; 11280df0e737SMichael Lange ierr = MPI_Bcast(&numLabels, 1, MPIU_INT, 0, comm);CHKERRQ(ierr); 1129627847f0SMatthew G. Knepley if (numLabels == numLocalLabels) hasLabels = PETSC_TRUE; 1130bdd2d751SMichael Lange for (l = numLabels-1; l >= 0; --l) { 1131bdd2d751SMichael Lange DMLabel label = NULL, labelNew = NULL; 11320df0e737SMichael Lange PetscBool isdepth; 11330df0e737SMichael Lange 1134d995df53SMatthew G. Knepley if (hasLabels) { 1135c58f1c22SToby Isaac ierr = DMGetLabelByNum(dm, l, &label);CHKERRQ(ierr); 11360df0e737SMichael Lange /* Skip "depth" because it is recreated */ 1137bdd2d751SMichael Lange ierr = PetscStrcmp(label->name, "depth", &isdepth);CHKERRQ(ierr); 1138bdd2d751SMichael Lange } 11390df0e737SMichael Lange ierr = MPI_Bcast(&isdepth, 1, MPIU_BOOL, 0, comm);CHKERRQ(ierr); 1140df0420ecSMatthew G. Knepley if (isdepth && !sendDepth) continue; 1141bdd2d751SMichael Lange ierr = DMLabelDistribute(label, migrationSF, &labelNew);CHKERRQ(ierr); 1142c58f1c22SToby Isaac ierr = DMAddLabel(dmParallel, labelNew);CHKERRQ(ierr); 11430df0e737SMichael Lange } 11440df0e737SMichael Lange ierr = PetscLogEventEnd(DMPLEX_DistributeLabels,dm,0,0,0);CHKERRQ(ierr); 11450df0e737SMichael Lange PetscFunctionReturn(0); 11460df0e737SMichael Lange } 11470df0e737SMichael Lange 11489734c634SMichael Lange #undef __FUNCT__ 11499734c634SMichael Lange #define __FUNCT__ "DMPlexDistributeSetupHybrid" 11509734c634SMichael Lange PetscErrorCode DMPlexDistributeSetupHybrid(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping renumbering, DM dmParallel) 11519734c634SMichael Lange { 11529734c634SMichael Lange DM_Plex *mesh = (DM_Plex*) dm->data; 11539734c634SMichael Lange DM_Plex *pmesh = (DM_Plex*) (dmParallel)->data; 11549734c634SMichael Lange MPI_Comm comm; 11559734c634SMichael Lange const PetscInt *gpoints; 11569734c634SMichael Lange PetscInt dim, depth, n, d; 11579734c634SMichael Lange PetscErrorCode ierr; 11589734c634SMichael Lange 11599734c634SMichael Lange PetscFunctionBegin; 11609734c634SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 11619734c634SMichael Lange PetscValidPointer(dmParallel, 4); 11629734c634SMichael Lange 11639734c634SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 11649734c634SMichael Lange ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 11659734c634SMichael Lange 11669734c634SMichael Lange /* Setup hybrid structure */ 11679734c634SMichael Lange for (d = 0; d <= dim; ++d) {pmesh->hybridPointMax[d] = mesh->hybridPointMax[d];} 11689734c634SMichael Lange ierr = MPI_Bcast(pmesh->hybridPointMax, dim+1, MPIU_INT, 0, comm);CHKERRQ(ierr); 11699734c634SMichael Lange ierr = ISLocalToGlobalMappingGetSize(renumbering, &n);CHKERRQ(ierr); 11709734c634SMichael Lange ierr = ISLocalToGlobalMappingGetIndices(renumbering, &gpoints);CHKERRQ(ierr); 11719734c634SMichael Lange ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 11729734c634SMichael Lange for (d = 0; d <= dim; ++d) { 11739734c634SMichael Lange PetscInt pmax = pmesh->hybridPointMax[d], newmax = 0, pEnd, stratum[2], p; 11749734c634SMichael Lange 11759734c634SMichael Lange if (pmax < 0) continue; 11769734c634SMichael Lange ierr = DMPlexGetDepthStratum(dm, d > depth ? depth : d, &stratum[0], &stratum[1]);CHKERRQ(ierr); 11779734c634SMichael Lange ierr = DMPlexGetDepthStratum(dmParallel, d, NULL, &pEnd);CHKERRQ(ierr); 11789734c634SMichael Lange ierr = MPI_Bcast(stratum, 2, MPIU_INT, 0, comm);CHKERRQ(ierr); 11799734c634SMichael Lange for (p = 0; p < n; ++p) { 11809734c634SMichael Lange const PetscInt point = gpoints[p]; 11819734c634SMichael Lange 11829734c634SMichael Lange if ((point >= stratum[0]) && (point < stratum[1]) && (point >= pmax)) ++newmax; 11839734c634SMichael Lange } 11849734c634SMichael Lange if (newmax > 0) pmesh->hybridPointMax[d] = pEnd - newmax; 11859734c634SMichael Lange else pmesh->hybridPointMax[d] = -1; 11869734c634SMichael Lange } 11879734c634SMichael Lange ierr = ISLocalToGlobalMappingRestoreIndices(renumbering, &gpoints);CHKERRQ(ierr); 11889734c634SMichael Lange PetscFunctionReturn(0); 11899734c634SMichael Lange } 11900df0e737SMichael Lange 1191a6f36705SMichael Lange #undef __FUNCT__ 1192a6f36705SMichael Lange #define __FUNCT__ "DMPlexDistributeSetupTree" 119308633170SToby Isaac PetscErrorCode DMPlexDistributeSetupTree(DM dm, PetscSF migrationSF, ISLocalToGlobalMapping original, ISLocalToGlobalMapping renumbering, DM dmParallel) 1194a6f36705SMichael Lange { 119515078cd4SMichael Lange DM_Plex *mesh = (DM_Plex*) dm->data; 119615078cd4SMichael Lange DM_Plex *pmesh = (DM_Plex*) (dmParallel)->data; 1197a6f36705SMichael Lange MPI_Comm comm; 1198a6f36705SMichael Lange DM refTree; 1199a6f36705SMichael Lange PetscSection origParentSection, newParentSection; 1200a6f36705SMichael Lange PetscInt *origParents, *origChildIDs; 1201a6f36705SMichael Lange PetscBool flg; 1202a6f36705SMichael Lange PetscErrorCode ierr; 1203a6f36705SMichael Lange 1204a6f36705SMichael Lange PetscFunctionBegin; 1205a6f36705SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 1206a6f36705SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 4); 1207a6f36705SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 1208a6f36705SMichael Lange 1209a6f36705SMichael Lange /* Set up tree */ 1210a6f36705SMichael Lange ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr); 1211a6f36705SMichael Lange ierr = DMPlexSetReferenceTree(dmParallel,refTree);CHKERRQ(ierr); 1212a6f36705SMichael Lange ierr = DMPlexGetTree(dm,&origParentSection,&origParents,&origChildIDs,NULL,NULL);CHKERRQ(ierr); 1213a6f36705SMichael Lange if (origParentSection) { 1214a6f36705SMichael Lange PetscInt pStart, pEnd; 121508633170SToby Isaac PetscInt *newParents, *newChildIDs, *globParents; 1216a6f36705SMichael Lange PetscInt *remoteOffsetsParents, newParentSize; 1217a6f36705SMichael Lange PetscSF parentSF; 1218a6f36705SMichael Lange 1219a6f36705SMichael Lange ierr = DMPlexGetChart(dmParallel, &pStart, &pEnd);CHKERRQ(ierr); 1220a6f36705SMichael Lange ierr = PetscSectionCreate(PetscObjectComm((PetscObject)dmParallel),&newParentSection);CHKERRQ(ierr); 1221a6f36705SMichael Lange ierr = PetscSectionSetChart(newParentSection,pStart,pEnd);CHKERRQ(ierr); 1222a6f36705SMichael Lange ierr = PetscSFDistributeSection(migrationSF, origParentSection, &remoteOffsetsParents, newParentSection);CHKERRQ(ierr); 1223a6f36705SMichael Lange ierr = PetscSFCreateSectionSF(migrationSF, origParentSection, remoteOffsetsParents, newParentSection, &parentSF);CHKERRQ(ierr); 12248a713f3bSLawrence Mitchell ierr = PetscFree(remoteOffsetsParents);CHKERRQ(ierr); 1225a6f36705SMichael Lange ierr = PetscSectionGetStorageSize(newParentSection,&newParentSize);CHKERRQ(ierr); 1226a6f36705SMichael Lange ierr = PetscMalloc2(newParentSize,&newParents,newParentSize,&newChildIDs);CHKERRQ(ierr); 122708633170SToby Isaac if (original) { 122808633170SToby Isaac PetscInt numParents; 122908633170SToby Isaac 123008633170SToby Isaac ierr = PetscSectionGetStorageSize(origParentSection,&numParents);CHKERRQ(ierr); 123108633170SToby Isaac ierr = PetscMalloc1(numParents,&globParents);CHKERRQ(ierr); 123208633170SToby Isaac ierr = ISLocalToGlobalMappingApplyBlock(original, numParents, origParents, globParents);CHKERRQ(ierr); 123308633170SToby Isaac } 123408633170SToby Isaac else { 123508633170SToby Isaac globParents = origParents; 123608633170SToby Isaac } 123708633170SToby Isaac ierr = PetscSFBcastBegin(parentSF, MPIU_INT, globParents, newParents);CHKERRQ(ierr); 123808633170SToby Isaac ierr = PetscSFBcastEnd(parentSF, MPIU_INT, globParents, newParents);CHKERRQ(ierr); 123908633170SToby Isaac if (original) { 124008633170SToby Isaac ierr = PetscFree(globParents);CHKERRQ(ierr); 124108633170SToby Isaac } 1242a6f36705SMichael Lange ierr = PetscSFBcastBegin(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr); 1243a6f36705SMichael Lange ierr = PetscSFBcastEnd(parentSF, MPIU_INT, origChildIDs, newChildIDs);CHKERRQ(ierr); 1244a6f36705SMichael Lange ierr = ISGlobalToLocalMappingApplyBlock(renumbering,IS_GTOLM_MASK, newParentSize, newParents, NULL, newParents);CHKERRQ(ierr); 12454a54e071SToby Isaac #if PETSC_USE_DEBUG 12464a54e071SToby Isaac { 12474a54e071SToby Isaac PetscInt p; 12484a54e071SToby Isaac PetscBool valid = PETSC_TRUE; 12494a54e071SToby Isaac for (p = 0; p < newParentSize; ++p) { 12504a54e071SToby Isaac if (newParents[p] < 0) {valid = PETSC_FALSE; ierr = PetscPrintf(PETSC_COMM_SELF, "Point %d not in overlap SF\n", p);CHKERRQ(ierr);} 12514a54e071SToby Isaac } 12524a54e071SToby Isaac if (!valid) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid global to local map"); 12534a54e071SToby Isaac } 12544a54e071SToby Isaac #endif 1255c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-parents_view", &flg);CHKERRQ(ierr); 1256a6f36705SMichael Lange if (flg) { 1257a6f36705SMichael Lange ierr = PetscPrintf(comm, "Serial Parent Section: \n");CHKERRQ(ierr); 1258a6f36705SMichael Lange ierr = PetscSectionView(origParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 1259a6f36705SMichael Lange ierr = PetscPrintf(comm, "Parallel Parent Section: \n");CHKERRQ(ierr); 1260a6f36705SMichael Lange ierr = PetscSectionView(newParentSection, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 1261a6f36705SMichael Lange ierr = PetscSFView(parentSF, NULL);CHKERRQ(ierr); 1262a6f36705SMichael Lange } 1263a6f36705SMichael Lange ierr = DMPlexSetTree(dmParallel,newParentSection,newParents,newChildIDs);CHKERRQ(ierr); 1264a6f36705SMichael Lange ierr = PetscSectionDestroy(&newParentSection);CHKERRQ(ierr); 1265a6f36705SMichael Lange ierr = PetscFree2(newParents,newChildIDs);CHKERRQ(ierr); 1266a6f36705SMichael Lange ierr = PetscSFDestroy(&parentSF);CHKERRQ(ierr); 1267a6f36705SMichael Lange } 126815078cd4SMichael Lange pmesh->useAnchors = mesh->useAnchors; 1269a6f36705SMichael Lange PetscFunctionReturn(0); 1270a6f36705SMichael Lange } 12710df0e737SMichael Lange 12720df0e737SMichael Lange #undef __FUNCT__ 12734eca1733SMichael Lange #define __FUNCT__ "DMPlexDistributeSF" 1274f8987ae8SMichael Lange PetscErrorCode DMPlexDistributeSF(DM dm, PetscSF migrationSF, DM dmParallel) 12754eca1733SMichael Lange { 12764eca1733SMichael Lange DM_Plex *mesh = (DM_Plex*) dm->data; 12774eca1733SMichael Lange DM_Plex *pmesh = (DM_Plex*) (dmParallel)->data; 12784eca1733SMichael Lange PetscMPIInt rank, numProcs; 12794eca1733SMichael Lange MPI_Comm comm; 12804eca1733SMichael Lange PetscErrorCode ierr; 12814eca1733SMichael Lange 12824eca1733SMichael Lange PetscFunctionBegin; 12834eca1733SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 12844eca1733SMichael Lange PetscValidPointer(dmParallel,7); 12854eca1733SMichael Lange 12864eca1733SMichael Lange /* Create point SF for parallel mesh */ 12874eca1733SMichael Lange ierr = PetscLogEventBegin(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr); 12884eca1733SMichael Lange ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr); 12894eca1733SMichael Lange ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 12904eca1733SMichael Lange ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr); 12914eca1733SMichael Lange { 12924eca1733SMichael Lange const PetscInt *leaves; 12934eca1733SMichael Lange PetscSFNode *remotePoints, *rowners, *lowners; 12944eca1733SMichael Lange PetscInt numRoots, numLeaves, numGhostPoints = 0, p, gp, *ghostPoints; 12954eca1733SMichael Lange PetscInt pStart, pEnd; 12964eca1733SMichael Lange 12974eca1733SMichael Lange ierr = DMPlexGetChart(dmParallel, &pStart, &pEnd);CHKERRQ(ierr); 12984eca1733SMichael Lange ierr = PetscSFGetGraph(migrationSF, &numRoots, &numLeaves, &leaves, NULL);CHKERRQ(ierr); 12994eca1733SMichael Lange ierr = PetscMalloc2(numRoots,&rowners,numLeaves,&lowners);CHKERRQ(ierr); 13004eca1733SMichael Lange for (p=0; p<numRoots; p++) { 13014eca1733SMichael Lange rowners[p].rank = -1; 13024eca1733SMichael Lange rowners[p].index = -1; 13034eca1733SMichael Lange } 13044eca1733SMichael Lange ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr); 13054eca1733SMichael Lange ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr); 13064eca1733SMichael Lange for (p = 0; p < numLeaves; ++p) { 13074eca1733SMichael Lange if (lowners[p].rank < 0 || lowners[p].rank == rank) { /* Either put in a bid or we know we own it */ 13084eca1733SMichael Lange lowners[p].rank = rank; 13094eca1733SMichael Lange lowners[p].index = leaves ? leaves[p] : p; 13104eca1733SMichael Lange } else if (lowners[p].rank >= 0) { /* Point already claimed so flag so that MAXLOC does not listen to us */ 13114eca1733SMichael Lange lowners[p].rank = -2; 13124eca1733SMichael Lange lowners[p].index = -2; 13134eca1733SMichael Lange } 13144eca1733SMichael Lange } 13154eca1733SMichael Lange for (p=0; p<numRoots; p++) { /* Root must not participate in the rediction, flag so that MAXLOC does not use */ 13164eca1733SMichael Lange rowners[p].rank = -3; 13174eca1733SMichael Lange rowners[p].index = -3; 13184eca1733SMichael Lange } 13194eca1733SMichael Lange ierr = PetscSFReduceBegin(migrationSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr); 13204eca1733SMichael Lange ierr = PetscSFReduceEnd(migrationSF, MPIU_2INT, lowners, rowners, MPI_MAXLOC);CHKERRQ(ierr); 13214eca1733SMichael Lange ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr); 13224eca1733SMichael Lange ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rowners, lowners);CHKERRQ(ierr); 13234eca1733SMichael Lange for (p = 0; p < numLeaves; ++p) { 13244eca1733SMichael Lange if (lowners[p].rank < 0 || lowners[p].index < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Cell partition corrupt: point not claimed"); 13254eca1733SMichael Lange if (lowners[p].rank != rank) ++numGhostPoints; 13264eca1733SMichael Lange } 13274eca1733SMichael Lange ierr = PetscMalloc1(numGhostPoints, &ghostPoints);CHKERRQ(ierr); 13284eca1733SMichael Lange ierr = PetscMalloc1(numGhostPoints, &remotePoints);CHKERRQ(ierr); 13294eca1733SMichael Lange for (p = 0, gp = 0; p < numLeaves; ++p) { 13304eca1733SMichael Lange if (lowners[p].rank != rank) { 13314eca1733SMichael Lange ghostPoints[gp] = leaves ? leaves[p] : p; 13324eca1733SMichael Lange remotePoints[gp].rank = lowners[p].rank; 13334eca1733SMichael Lange remotePoints[gp].index = lowners[p].index; 13344eca1733SMichael Lange ++gp; 13354eca1733SMichael Lange } 13364eca1733SMichael Lange } 13374eca1733SMichael Lange ierr = PetscFree2(rowners,lowners);CHKERRQ(ierr); 13384eca1733SMichael Lange ierr = PetscSFSetGraph((dmParallel)->sf, pEnd - pStart, numGhostPoints, ghostPoints, PETSC_OWN_POINTER, remotePoints, PETSC_OWN_POINTER);CHKERRQ(ierr); 13394eca1733SMichael Lange ierr = PetscSFSetFromOptions((dmParallel)->sf);CHKERRQ(ierr); 13404eca1733SMichael Lange } 13414eca1733SMichael Lange pmesh->useCone = mesh->useCone; 13424eca1733SMichael Lange pmesh->useClosure = mesh->useClosure; 13434eca1733SMichael Lange ierr = PetscLogEventEnd(DMPLEX_DistributeSF,dm,0,0,0);CHKERRQ(ierr); 13444eca1733SMichael Lange PetscFunctionReturn(0); 13454eca1733SMichael Lange } 13464eca1733SMichael Lange 13474eca1733SMichael Lange #undef __FUNCT__ 1348f5bf2dbfSMichael Lange #define __FUNCT__ "DMPlexCreatePointSF" 1349f5bf2dbfSMichael Lange /*@C 1350f5bf2dbfSMichael Lange DMPlexDerivePointSF - Build a point SF from an SF describing a point migration 1351f5bf2dbfSMichael Lange 1352f5bf2dbfSMichael Lange Input Parameter: 1353f5bf2dbfSMichael Lange + dm - The source DMPlex object 1354f5bf2dbfSMichael Lange . migrationSF - The star forest that describes the parallel point remapping 13551627f6ccSMichael Lange . ownership - Flag causing a vote to determine point ownership 1356f5bf2dbfSMichael Lange 1357f5bf2dbfSMichael Lange Output Parameter: 1358f5bf2dbfSMichael Lange - pointSF - The star forest describing the point overlap in the remapped DM 1359f5bf2dbfSMichael Lange 1360f5bf2dbfSMichael Lange Level: developer 1361f5bf2dbfSMichael Lange 1362f5bf2dbfSMichael Lange .seealso: DMPlexDistribute(), DMPlexDistributeOverlap() 1363f5bf2dbfSMichael Lange @*/ 1364f5bf2dbfSMichael Lange PetscErrorCode DMPlexCreatePointSF(DM dm, PetscSF migrationSF, PetscBool ownership, PetscSF *pointSF) 1365f5bf2dbfSMichael Lange { 1366f5bf2dbfSMichael Lange PetscMPIInt rank; 13671627f6ccSMichael Lange PetscInt p, nroots, nleaves, idx, npointLeaves; 1368f5bf2dbfSMichael Lange PetscInt *pointLocal; 1369f5bf2dbfSMichael Lange const PetscInt *leaves; 1370f5bf2dbfSMichael Lange const PetscSFNode *roots; 1371f5bf2dbfSMichael Lange PetscSFNode *rootNodes, *leafNodes, *pointRemote; 1372f5bf2dbfSMichael Lange PetscErrorCode ierr; 1373f5bf2dbfSMichael Lange 1374f5bf2dbfSMichael Lange PetscFunctionBegin; 1375f5bf2dbfSMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 1376f5bf2dbfSMichael Lange ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr); 1377f5bf2dbfSMichael Lange 1378f5bf2dbfSMichael Lange ierr = PetscSFGetGraph(migrationSF, &nroots, &nleaves, &leaves, &roots);CHKERRQ(ierr); 1379f5bf2dbfSMichael Lange ierr = PetscMalloc2(nroots, &rootNodes, nleaves, &leafNodes);CHKERRQ(ierr); 1380f5bf2dbfSMichael Lange if (ownership) { 13811627f6ccSMichael Lange /* Point ownership vote: Process with highest rank ownes shared points */ 1382f5bf2dbfSMichael Lange for (p = 0; p < nleaves; ++p) { 1383f5bf2dbfSMichael Lange /* Either put in a bid or we know we own it */ 1384f5bf2dbfSMichael Lange leafNodes[p].rank = rank; 138543f7d02bSMichael Lange leafNodes[p].index = p; 1386f5bf2dbfSMichael Lange } 1387f5bf2dbfSMichael Lange for (p = 0; p < nroots; p++) { 13881627f6ccSMichael Lange /* Root must not participate in the reduction, flag so that MAXLOC does not use */ 1389f5bf2dbfSMichael Lange rootNodes[p].rank = -3; 1390f5bf2dbfSMichael Lange rootNodes[p].index = -3; 1391f5bf2dbfSMichael Lange } 1392f5bf2dbfSMichael Lange ierr = PetscSFReduceBegin(migrationSF, MPIU_2INT, leafNodes, rootNodes, MPI_MAXLOC);CHKERRQ(ierr); 1393f5bf2dbfSMichael Lange ierr = PetscSFReduceEnd(migrationSF, MPIU_2INT, leafNodes, rootNodes, MPI_MAXLOC);CHKERRQ(ierr); 1394f5bf2dbfSMichael Lange } else { 1395f5bf2dbfSMichael Lange for (p = 0; p < nroots; p++) { 1396f5bf2dbfSMichael Lange rootNodes[p].index = -1; 1397f5bf2dbfSMichael Lange rootNodes[p].rank = rank; 1398f5bf2dbfSMichael Lange }; 1399f5bf2dbfSMichael Lange for (p = 0; p < nleaves; p++) { 1400f5bf2dbfSMichael Lange /* Write new local id into old location */ 1401f5bf2dbfSMichael Lange if (roots[p].rank == rank) { 1402f5bf2dbfSMichael Lange rootNodes[roots[p].index].index = leaves[p]; 1403f5bf2dbfSMichael Lange } 1404f5bf2dbfSMichael Lange } 1405f5bf2dbfSMichael Lange } 1406f5bf2dbfSMichael Lange ierr = PetscSFBcastBegin(migrationSF, MPIU_2INT, rootNodes, leafNodes);CHKERRQ(ierr); 1407f5bf2dbfSMichael Lange ierr = PetscSFBcastEnd(migrationSF, MPIU_2INT, rootNodes, leafNodes);CHKERRQ(ierr); 1408f5bf2dbfSMichael Lange 1409f5bf2dbfSMichael Lange for (npointLeaves = 0, p = 0; p < nleaves; p++) {if (leafNodes[p].rank != rank) npointLeaves++;} 14101627f6ccSMichael Lange ierr = PetscMalloc1(npointLeaves, &pointLocal);CHKERRQ(ierr); 14111627f6ccSMichael Lange ierr = PetscMalloc1(npointLeaves, &pointRemote);CHKERRQ(ierr); 1412f5bf2dbfSMichael Lange for (idx = 0, p = 0; p < nleaves; p++) { 1413f5bf2dbfSMichael Lange if (leafNodes[p].rank != rank) { 1414f5bf2dbfSMichael Lange pointLocal[idx] = p; 1415f5bf2dbfSMichael Lange pointRemote[idx] = leafNodes[p]; 1416f5bf2dbfSMichael Lange idx++; 1417f5bf2dbfSMichael Lange } 1418f5bf2dbfSMichael Lange } 1419f5bf2dbfSMichael Lange ierr = PetscSFCreate(PetscObjectComm((PetscObject) dm), pointSF);CHKERRQ(ierr); 14201627f6ccSMichael Lange ierr = PetscSFSetFromOptions(*pointSF);CHKERRQ(ierr); 1421f5bf2dbfSMichael Lange ierr = PetscSFSetGraph(*pointSF, nleaves, npointLeaves, pointLocal, PETSC_OWN_POINTER, pointRemote, PETSC_OWN_POINTER);CHKERRQ(ierr); 1422f5bf2dbfSMichael Lange ierr = PetscFree2(rootNodes, leafNodes);CHKERRQ(ierr); 1423f5bf2dbfSMichael Lange PetscFunctionReturn(0); 1424f5bf2dbfSMichael Lange } 1425f5bf2dbfSMichael Lange 1426f5bf2dbfSMichael Lange #undef __FUNCT__ 142715078cd4SMichael Lange #define __FUNCT__ "DMPlexMigrate" 142815078cd4SMichael Lange /*@C 142915078cd4SMichael Lange DMPlexMigrate - Migrates internal DM data over the supplied star forest 143015078cd4SMichael Lange 143115078cd4SMichael Lange Input Parameter: 143215078cd4SMichael Lange + dm - The source DMPlex object 14331627f6ccSMichael Lange . sf - The star forest communication context describing the migration pattern 143415078cd4SMichael Lange 143515078cd4SMichael Lange Output Parameter: 143615078cd4SMichael Lange - targetDM - The target DMPlex object 143715078cd4SMichael Lange 1438b9f40539SMichael Lange Level: intermediate 143915078cd4SMichael Lange 144015078cd4SMichael Lange .seealso: DMPlexDistribute(), DMPlexDistributeOverlap() 144115078cd4SMichael Lange @*/ 1442b9f40539SMichael Lange PetscErrorCode DMPlexMigrate(DM dm, PetscSF sf, DM targetDM) 144315078cd4SMichael Lange { 1444b9f40539SMichael Lange MPI_Comm comm; 1445b9f40539SMichael Lange PetscInt dim, nroots; 1446b9f40539SMichael Lange PetscSF sfPoint; 144715078cd4SMichael Lange ISLocalToGlobalMapping ltogMigration; 1448ac04eaf7SToby Isaac ISLocalToGlobalMapping ltogOriginal = NULL; 1449b9f40539SMichael Lange PetscBool flg; 145015078cd4SMichael Lange PetscErrorCode ierr; 145115078cd4SMichael Lange 145215078cd4SMichael Lange PetscFunctionBegin; 145315078cd4SMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 14541b858b30SMichael Lange ierr = PetscLogEventBegin(DMPLEX_Migrate, dm, 0, 0, 0);CHKERRQ(ierr); 1455b9f40539SMichael Lange ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr); 145615078cd4SMichael Lange ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 145715078cd4SMichael Lange ierr = DMSetDimension(targetDM, dim);CHKERRQ(ierr); 145815078cd4SMichael Lange 1459bfb0467fSMichael Lange /* Check for a one-to-all distribution pattern */ 1460b9f40539SMichael Lange ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr); 1461b9f40539SMichael Lange ierr = PetscSFGetGraph(sfPoint, &nroots, NULL, NULL, NULL);CHKERRQ(ierr); 1462bfb0467fSMichael Lange if (nroots >= 0) { 1463b9f40539SMichael Lange IS isOriginal; 1464ac04eaf7SToby Isaac PetscInt n, size, nleaves; 1465ac04eaf7SToby Isaac PetscInt *numbering_orig, *numbering_new; 1466b9f40539SMichael Lange /* Get the original point numbering */ 1467b9f40539SMichael Lange ierr = DMPlexCreatePointNumbering(dm, &isOriginal);CHKERRQ(ierr); 1468b9f40539SMichael Lange ierr = ISLocalToGlobalMappingCreateIS(isOriginal, <ogOriginal);CHKERRQ(ierr); 1469b9f40539SMichael Lange ierr = ISLocalToGlobalMappingGetSize(ltogOriginal, &size);CHKERRQ(ierr); 1470b9f40539SMichael Lange ierr = ISLocalToGlobalMappingGetBlockIndices(ltogOriginal, (const PetscInt**)&numbering_orig);CHKERRQ(ierr); 1471b9f40539SMichael Lange /* Convert to positive global numbers */ 1472b9f40539SMichael Lange for (n=0; n<size; n++) {if (numbering_orig[n] < 0) numbering_orig[n] = -(numbering_orig[n]+1);} 1473b9f40539SMichael Lange /* Derive the new local-to-global mapping from the old one */ 1474b9f40539SMichael Lange ierr = PetscSFGetGraph(sf, NULL, &nleaves, NULL, NULL);CHKERRQ(ierr); 1475b9f40539SMichael Lange ierr = PetscMalloc1(nleaves, &numbering_new);CHKERRQ(ierr); 1476b9f40539SMichael Lange ierr = PetscSFBcastBegin(sf, MPIU_INT, (PetscInt *) numbering_orig, numbering_new);CHKERRQ(ierr); 1477b9f40539SMichael Lange ierr = PetscSFBcastEnd(sf, MPIU_INT, (PetscInt *) numbering_orig, numbering_new);CHKERRQ(ierr); 1478b9f40539SMichael Lange ierr = ISLocalToGlobalMappingCreate(comm, 1, nleaves, (const PetscInt*) numbering_new, PETSC_OWN_POINTER, <ogMigration);CHKERRQ(ierr); 1479b9f40539SMichael Lange ierr = ISLocalToGlobalMappingRestoreIndices(ltogOriginal, (const PetscInt**)&numbering_orig);CHKERRQ(ierr); 1480b9f40539SMichael Lange ierr = ISDestroy(&isOriginal);CHKERRQ(ierr); 148115078cd4SMichael Lange } else { 1482bfb0467fSMichael Lange /* One-to-all distribution pattern: We can derive LToG from SF */ 148315078cd4SMichael Lange ierr = ISLocalToGlobalMappingCreateSF(sf, 0, <ogMigration);CHKERRQ(ierr); 148415078cd4SMichael Lange } 1485c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-partition_view", &flg);CHKERRQ(ierr); 1486b9f40539SMichael Lange if (flg) { 1487b9f40539SMichael Lange ierr = PetscPrintf(comm, "Point renumbering for DM migration:\n");CHKERRQ(ierr); 1488b9f40539SMichael Lange ierr = ISLocalToGlobalMappingView(ltogMigration, NULL);CHKERRQ(ierr); 1489b9f40539SMichael Lange } 149015078cd4SMichael Lange /* Migrate DM data to target DM */ 1491ac04eaf7SToby Isaac ierr = DMPlexDistributeCones(dm, sf, ltogOriginal, ltogMigration, targetDM);CHKERRQ(ierr); 149215078cd4SMichael Lange ierr = DMPlexDistributeCoordinates(dm, sf, targetDM);CHKERRQ(ierr); 149315078cd4SMichael Lange ierr = DMPlexDistributeLabels(dm, sf, targetDM);CHKERRQ(ierr); 149415078cd4SMichael Lange ierr = DMPlexDistributeSetupHybrid(dm, sf, ltogMigration, targetDM);CHKERRQ(ierr); 149508633170SToby Isaac ierr = DMPlexDistributeSetupTree(dm, sf, ltogOriginal, ltogMigration, targetDM);CHKERRQ(ierr); 1496ac04eaf7SToby Isaac ierr = ISLocalToGlobalMappingDestroy(<ogOriginal);CHKERRQ(ierr); 1497302440fdSBarry Smith ierr = ISLocalToGlobalMappingDestroy(<ogMigration);CHKERRQ(ierr); 14981b858b30SMichael Lange ierr = PetscLogEventEnd(DMPLEX_Migrate, dm, 0, 0, 0);CHKERRQ(ierr); 149915078cd4SMichael Lange PetscFunctionReturn(0); 150015078cd4SMichael Lange } 150115078cd4SMichael Lange 150215078cd4SMichael Lange #undef __FUNCT__ 150370034214SMatthew G. Knepley #define __FUNCT__ "DMPlexDistribute" 15043b8f15a2SMatthew G. Knepley /*@C 150570034214SMatthew G. Knepley DMPlexDistribute - Distributes the mesh and any associated sections. 150670034214SMatthew G. Knepley 150770034214SMatthew G. Knepley Not Collective 150870034214SMatthew G. Knepley 150970034214SMatthew G. Knepley Input Parameter: 151070034214SMatthew G. Knepley + dm - The original DMPlex object 151170034214SMatthew G. Knepley - overlap - The overlap of partitions, 0 is the default 151270034214SMatthew G. Knepley 151370034214SMatthew G. Knepley Output Parameter: 151470034214SMatthew G. Knepley + sf - The PetscSF used for point distribution 151570034214SMatthew G. Knepley - parallelMesh - The distributed DMPlex object, or NULL 151670034214SMatthew G. Knepley 151770034214SMatthew G. Knepley Note: If the mesh was not distributed, the return value is NULL. 151870034214SMatthew G. Knepley 151970034214SMatthew G. Knepley The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and 152070034214SMatthew G. Knepley DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function 152170034214SMatthew G. Knepley representation on the mesh. 152270034214SMatthew G. Knepley 152370034214SMatthew G. Knepley Level: intermediate 152470034214SMatthew G. Knepley 152570034214SMatthew G. Knepley .keywords: mesh, elements 152670034214SMatthew G. Knepley .seealso: DMPlexCreate(), DMPlexDistributeByFace(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure() 152770034214SMatthew G. Knepley @*/ 152880cf41d5SMatthew G. Knepley PetscErrorCode DMPlexDistribute(DM dm, PetscInt overlap, PetscSF *sf, DM *dmParallel) 152970034214SMatthew G. Knepley { 153070034214SMatthew G. Knepley MPI_Comm comm; 153115078cd4SMichael Lange PetscPartitioner partitioner; 1532f8987ae8SMichael Lange IS cellPart; 1533f8987ae8SMichael Lange PetscSection cellPartSection; 1534cf86098cSMatthew G. Knepley DM dmCoord; 1535f8987ae8SMichael Lange DMLabel lblPartition, lblMigration; 153643f7d02bSMichael Lange PetscSF sfProcess, sfMigration, sfStratified, sfPoint; 153770034214SMatthew G. Knepley PetscBool flg; 153870034214SMatthew G. Knepley PetscMPIInt rank, numProcs, p; 153970034214SMatthew G. Knepley PetscErrorCode ierr; 154070034214SMatthew G. Knepley 154170034214SMatthew G. Knepley PetscFunctionBegin; 154270034214SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 154370034214SMatthew G. Knepley if (sf) PetscValidPointer(sf,4); 154470034214SMatthew G. Knepley PetscValidPointer(dmParallel,5); 154570034214SMatthew G. Knepley 154670034214SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr); 154770034214SMatthew G. Knepley ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr); 154870034214SMatthew G. Knepley ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 154970034214SMatthew G. Knepley ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr); 155070034214SMatthew G. Knepley 155170034214SMatthew G. Knepley *dmParallel = NULL; 155270034214SMatthew G. Knepley if (numProcs == 1) PetscFunctionReturn(0); 155370034214SMatthew G. Knepley 155415078cd4SMichael Lange /* Create cell partition */ 155577623264SMatthew G. Knepley ierr = PetscLogEventBegin(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr); 155677623264SMatthew G. Knepley ierr = PetscSectionCreate(comm, &cellPartSection);CHKERRQ(ierr); 155715078cd4SMichael Lange ierr = DMPlexGetPartitioner(dm, &partitioner);CHKERRQ(ierr); 155815078cd4SMichael Lange ierr = PetscPartitionerPartition(partitioner, dm, cellPartSection, &cellPart);CHKERRQ(ierr); 1559f8987ae8SMichael Lange { 1560f8987ae8SMichael Lange /* Convert partition to DMLabel */ 1561f8987ae8SMichael Lange PetscInt proc, pStart, pEnd, npoints, poffset; 1562f8987ae8SMichael Lange const PetscInt *points; 1563f8987ae8SMichael Lange ierr = DMLabelCreate("Point Partition", &lblPartition);CHKERRQ(ierr); 1564f8987ae8SMichael Lange ierr = ISGetIndices(cellPart, &points);CHKERRQ(ierr); 1565f8987ae8SMichael Lange ierr = PetscSectionGetChart(cellPartSection, &pStart, &pEnd);CHKERRQ(ierr); 1566f8987ae8SMichael Lange for (proc = pStart; proc < pEnd; proc++) { 1567f8987ae8SMichael Lange ierr = PetscSectionGetDof(cellPartSection, proc, &npoints);CHKERRQ(ierr); 1568f8987ae8SMichael Lange ierr = PetscSectionGetOffset(cellPartSection, proc, &poffset);CHKERRQ(ierr); 1569f8987ae8SMichael Lange for (p = poffset; p < poffset+npoints; p++) { 1570f8987ae8SMichael Lange ierr = DMLabelSetValue(lblPartition, points[p], proc);CHKERRQ(ierr); 157170034214SMatthew G. Knepley } 1572f8987ae8SMichael Lange } 1573f8987ae8SMichael Lange ierr = ISRestoreIndices(cellPart, &points);CHKERRQ(ierr); 1574f8987ae8SMichael Lange } 1575f8987ae8SMichael Lange ierr = DMPlexPartitionLabelClosure(dm, lblPartition);CHKERRQ(ierr); 1576f8987ae8SMichael Lange { 1577f8987ae8SMichael Lange /* Build a global process SF */ 1578f8987ae8SMichael Lange PetscSFNode *remoteProc; 1579f8987ae8SMichael Lange ierr = PetscMalloc1(numProcs, &remoteProc);CHKERRQ(ierr); 1580f8987ae8SMichael Lange for (p = 0; p < numProcs; ++p) { 1581f8987ae8SMichael Lange remoteProc[p].rank = p; 1582f8987ae8SMichael Lange remoteProc[p].index = rank; 1583f8987ae8SMichael Lange } 1584f8987ae8SMichael Lange ierr = PetscSFCreate(comm, &sfProcess);CHKERRQ(ierr); 1585f8987ae8SMichael Lange ierr = PetscObjectSetName((PetscObject) sfProcess, "Process SF");CHKERRQ(ierr); 1586f8987ae8SMichael Lange ierr = PetscSFSetGraph(sfProcess, numProcs, numProcs, NULL, PETSC_OWN_POINTER, remoteProc, PETSC_OWN_POINTER);CHKERRQ(ierr); 1587f8987ae8SMichael Lange } 1588f8987ae8SMichael Lange ierr = DMLabelCreate("Point migration", &lblMigration);CHKERRQ(ierr); 1589f8987ae8SMichael Lange ierr = DMPlexPartitionLabelInvert(dm, lblPartition, sfProcess, lblMigration);CHKERRQ(ierr); 1590302440fdSBarry Smith ierr = DMPlexPartitionLabelCreateSF(dm, lblMigration, &sfMigration);CHKERRQ(ierr); 159143f7d02bSMichael Lange /* Stratify the SF in case we are migrating an already parallel plex */ 159243f7d02bSMichael Lange ierr = DMPlexStratifyMigrationSF(dm, sfMigration, &sfStratified);CHKERRQ(ierr); 159343f7d02bSMichael Lange ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr); 159443f7d02bSMichael Lange sfMigration = sfStratified; 1595f8987ae8SMichael Lange ierr = PetscLogEventEnd(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr); 1596c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject) dm)->options,((PetscObject) dm)->prefix, "-partition_view", &flg);CHKERRQ(ierr); 159770034214SMatthew G. Knepley if (flg) { 1598f8987ae8SMichael Lange ierr = DMLabelView(lblPartition, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 1599f8987ae8SMichael Lange ierr = PetscSFView(sfMigration, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 160070034214SMatthew G. Knepley } 1601f8987ae8SMichael Lange 160215078cd4SMichael Lange /* Create non-overlapping parallel DM and migrate internal data */ 160370034214SMatthew G. Knepley ierr = DMPlexCreate(comm, dmParallel);CHKERRQ(ierr); 160470034214SMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) *dmParallel, "Parallel Mesh");CHKERRQ(ierr); 1605b9f40539SMichael Lange ierr = DMPlexMigrate(dm, sfMigration, *dmParallel);CHKERRQ(ierr); 160670034214SMatthew G. Knepley 1607a157612eSMichael Lange /* Build the point SF without overlap */ 16081627f6ccSMichael Lange ierr = DMPlexCreatePointSF(*dmParallel, sfMigration, PETSC_TRUE, &sfPoint);CHKERRQ(ierr); 1609f5bf2dbfSMichael Lange ierr = DMSetPointSF(*dmParallel, sfPoint);CHKERRQ(ierr); 1610cf86098cSMatthew G. Knepley ierr = DMGetCoordinateDM(*dmParallel, &dmCoord);CHKERRQ(ierr); 1611cf86098cSMatthew G. Knepley if (dmCoord) {ierr = DMSetPointSF(dmCoord, sfPoint);CHKERRQ(ierr);} 1612f5bf2dbfSMichael Lange if (flg) {ierr = PetscSFView(sfPoint, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);} 161370034214SMatthew G. Knepley 1614a157612eSMichael Lange if (overlap > 0) { 161515078cd4SMichael Lange DM dmOverlap; 161615078cd4SMichael Lange PetscInt nroots, nleaves; 161715078cd4SMichael Lange PetscSFNode *newRemote; 161815078cd4SMichael Lange const PetscSFNode *oldRemote; 161915078cd4SMichael Lange PetscSF sfOverlap, sfOverlapPoint; 1620a157612eSMichael Lange /* Add the partition overlap to the distributed DM */ 1621b9f40539SMichael Lange ierr = DMPlexDistributeOverlap(*dmParallel, overlap, &sfOverlap, &dmOverlap);CHKERRQ(ierr); 1622a157612eSMichael Lange ierr = DMDestroy(dmParallel);CHKERRQ(ierr); 1623a157612eSMichael Lange *dmParallel = dmOverlap; 1624c389ea9fSToby Isaac if (flg) { 16253d822a50SMichael Lange ierr = PetscPrintf(comm, "Overlap Migration SF:\n");CHKERRQ(ierr); 162615078cd4SMichael Lange ierr = PetscSFView(sfOverlap, NULL);CHKERRQ(ierr); 1627c389ea9fSToby Isaac } 162843331d4aSMichael Lange 1629f8987ae8SMichael Lange /* Re-map the migration SF to establish the full migration pattern */ 1630f8987ae8SMichael Lange ierr = PetscSFGetGraph(sfMigration, &nroots, NULL, NULL, &oldRemote);CHKERRQ(ierr); 163115078cd4SMichael Lange ierr = PetscSFGetGraph(sfOverlap, NULL, &nleaves, NULL, NULL);CHKERRQ(ierr); 163243331d4aSMichael Lange ierr = PetscMalloc1(nleaves, &newRemote);CHKERRQ(ierr); 163315078cd4SMichael Lange ierr = PetscSFBcastBegin(sfOverlap, MPIU_2INT, oldRemote, newRemote);CHKERRQ(ierr); 163415078cd4SMichael Lange ierr = PetscSFBcastEnd(sfOverlap, MPIU_2INT, oldRemote, newRemote);CHKERRQ(ierr); 163515078cd4SMichael Lange ierr = PetscSFCreate(comm, &sfOverlapPoint);CHKERRQ(ierr); 163615078cd4SMichael Lange ierr = PetscSFSetGraph(sfOverlapPoint, nroots, nleaves, NULL, PETSC_OWN_POINTER, newRemote, PETSC_OWN_POINTER);CHKERRQ(ierr); 163715078cd4SMichael Lange ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr); 1638f8987ae8SMichael Lange ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr); 163915078cd4SMichael Lange sfMigration = sfOverlapPoint; 1640c389ea9fSToby Isaac } 1641bf5244c7SMatthew G. Knepley /* Cleanup Partition */ 1642f8987ae8SMichael Lange ierr = PetscSFDestroy(&sfProcess);CHKERRQ(ierr); 1643f8987ae8SMichael Lange ierr = DMLabelDestroy(&lblPartition);CHKERRQ(ierr); 1644f8987ae8SMichael Lange ierr = DMLabelDestroy(&lblMigration);CHKERRQ(ierr); 16454eca1733SMichael Lange ierr = PetscSectionDestroy(&cellPartSection);CHKERRQ(ierr); 16464eca1733SMichael Lange ierr = ISDestroy(&cellPart);CHKERRQ(ierr); 1647721cbd76SMatthew G. Knepley /* Copy BC */ 1648721cbd76SMatthew G. Knepley ierr = DMPlexCopyBoundary(dm, *dmParallel);CHKERRQ(ierr); 164966fe0bfeSMatthew G. Knepley /* Create sfNatural */ 165066fe0bfeSMatthew G. Knepley if (dm->useNatural) { 165166fe0bfeSMatthew G. Knepley PetscSection section; 165266fe0bfeSMatthew G. Knepley 165366fe0bfeSMatthew G. Knepley ierr = DMGetDefaultSection(dm, §ion);CHKERRQ(ierr); 16540c1f158bSMatthew G. Knepley ierr = DMPlexCreateGlobalToNaturalSF(*dmParallel, section, sfMigration, &(*dmParallel)->sfNatural);CHKERRQ(ierr); 165566fe0bfeSMatthew G. Knepley } 1656721cbd76SMatthew G. Knepley /* Cleanup */ 1657f8987ae8SMichael Lange if (sf) {*sf = sfMigration;} 1658f8987ae8SMichael Lange else {ierr = PetscSFDestroy(&sfMigration);CHKERRQ(ierr);} 1659f5bf2dbfSMichael Lange ierr = PetscSFDestroy(&sfPoint);CHKERRQ(ierr); 166070034214SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_Distribute,dm,0,0,0);CHKERRQ(ierr); 166170034214SMatthew G. Knepley PetscFunctionReturn(0); 166270034214SMatthew G. Knepley } 1663a157612eSMichael Lange 1664a157612eSMichael Lange #undef __FUNCT__ 1665a157612eSMichael Lange #define __FUNCT__ "DMPlexDistributeOverlap" 1666a157612eSMichael Lange /*@C 1667a157612eSMichael Lange DMPlexDistribute - Add partition overlap to a distributed non-overlapping DM. 1668a157612eSMichael Lange 1669a157612eSMichael Lange Not Collective 1670a157612eSMichael Lange 1671a157612eSMichael Lange Input Parameter: 1672a157612eSMichael Lange + dm - The non-overlapping distrbuted DMPlex object 1673a157612eSMichael Lange - overlap - The overlap of partitions, 0 is the default 1674a157612eSMichael Lange 1675a157612eSMichael Lange Output Parameter: 1676a157612eSMichael Lange + sf - The PetscSF used for point distribution 1677a157612eSMichael Lange - dmOverlap - The overlapping distributed DMPlex object, or NULL 1678a157612eSMichael Lange 1679a157612eSMichael Lange Note: If the mesh was not distributed, the return value is NULL. 1680a157612eSMichael Lange 1681a157612eSMichael Lange The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and 1682a157612eSMichael Lange DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function 1683a157612eSMichael Lange representation on the mesh. 1684a157612eSMichael Lange 1685a157612eSMichael Lange Level: intermediate 1686a157612eSMichael Lange 1687a157612eSMichael Lange .keywords: mesh, elements 1688a157612eSMichael Lange .seealso: DMPlexCreate(), DMPlexDistributeByFace(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure() 1689a157612eSMichael Lange @*/ 1690b9f40539SMichael Lange PetscErrorCode DMPlexDistributeOverlap(DM dm, PetscInt overlap, PetscSF *sf, DM *dmOverlap) 1691a157612eSMichael Lange { 1692a157612eSMichael Lange MPI_Comm comm; 1693a157612eSMichael Lange PetscMPIInt rank; 1694a157612eSMichael Lange PetscSection rootSection, leafSection; 1695a157612eSMichael Lange IS rootrank, leafrank; 1696cf86098cSMatthew G. Knepley DM dmCoord; 1697a9f1d5b2SMichael Lange DMLabel lblOverlap; 1698f5bf2dbfSMichael Lange PetscSF sfOverlap, sfStratified, sfPoint; 1699a157612eSMichael Lange PetscErrorCode ierr; 1700a157612eSMichael Lange 1701a157612eSMichael Lange PetscFunctionBegin; 1702a157612eSMichael Lange PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 1703a157612eSMichael Lange if (sf) PetscValidPointer(sf, 3); 1704a157612eSMichael Lange PetscValidPointer(dmOverlap, 4); 1705a157612eSMichael Lange 17061b858b30SMichael Lange ierr = PetscLogEventBegin(DMPLEX_DistributeOverlap, dm, 0, 0, 0);CHKERRQ(ierr); 1707a157612eSMichael Lange ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr); 1708a157612eSMichael Lange ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr); 1709a157612eSMichael Lange 1710a157612eSMichael Lange /* Compute point overlap with neighbouring processes on the distributed DM */ 1711a157612eSMichael Lange ierr = PetscLogEventBegin(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr); 1712a157612eSMichael Lange ierr = PetscSectionCreate(comm, &rootSection);CHKERRQ(ierr); 1713a157612eSMichael Lange ierr = PetscSectionCreate(comm, &leafSection);CHKERRQ(ierr); 1714a157612eSMichael Lange ierr = DMPlexDistributeOwnership(dm, rootSection, &rootrank, leafSection, &leafrank);CHKERRQ(ierr); 1715a9f1d5b2SMichael Lange ierr = DMPlexCreateOverlap(dm, overlap, rootSection, rootrank, leafSection, leafrank, &lblOverlap);CHKERRQ(ierr); 1716a9f1d5b2SMichael Lange /* Convert overlap label to stratified migration SF */ 1717a9f1d5b2SMichael Lange ierr = DMPlexPartitionLabelCreateSF(dm, lblOverlap, &sfOverlap);CHKERRQ(ierr); 1718a9f1d5b2SMichael Lange ierr = DMPlexStratifyMigrationSF(dm, sfOverlap, &sfStratified);CHKERRQ(ierr); 1719a9f1d5b2SMichael Lange ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr); 1720a9f1d5b2SMichael Lange sfOverlap = sfStratified; 1721a9f1d5b2SMichael Lange ierr = PetscObjectSetName((PetscObject) sfOverlap, "Overlap SF");CHKERRQ(ierr); 1722a9f1d5b2SMichael Lange ierr = PetscSFSetFromOptions(sfOverlap);CHKERRQ(ierr); 1723a9f1d5b2SMichael Lange 172415fff7beSMatthew G. Knepley ierr = PetscSectionDestroy(&rootSection);CHKERRQ(ierr); 172515fff7beSMatthew G. Knepley ierr = PetscSectionDestroy(&leafSection);CHKERRQ(ierr); 172615fff7beSMatthew G. Knepley ierr = ISDestroy(&rootrank);CHKERRQ(ierr); 172715fff7beSMatthew G. Knepley ierr = ISDestroy(&leafrank);CHKERRQ(ierr); 1728a157612eSMichael Lange ierr = PetscLogEventEnd(PETSCPARTITIONER_Partition,dm,0,0,0);CHKERRQ(ierr); 1729a157612eSMichael Lange 1730a157612eSMichael Lange /* Build the overlapping DM */ 1731a157612eSMichael Lange ierr = DMPlexCreate(comm, dmOverlap);CHKERRQ(ierr); 1732a157612eSMichael Lange ierr = PetscObjectSetName((PetscObject) *dmOverlap, "Parallel Mesh");CHKERRQ(ierr); 1733a9f1d5b2SMichael Lange ierr = DMPlexMigrate(dm, sfOverlap, *dmOverlap);CHKERRQ(ierr); 1734f5bf2dbfSMichael Lange /* Build the new point SF */ 17351627f6ccSMichael Lange ierr = DMPlexCreatePointSF(*dmOverlap, sfOverlap, PETSC_FALSE, &sfPoint);CHKERRQ(ierr); 1736f5bf2dbfSMichael Lange ierr = DMSetPointSF(*dmOverlap, sfPoint);CHKERRQ(ierr); 1737cf86098cSMatthew G. Knepley ierr = DMGetCoordinateDM(*dmOverlap, &dmCoord);CHKERRQ(ierr); 1738cf86098cSMatthew G. Knepley if (dmCoord) {ierr = DMSetPointSF(dmCoord, sfPoint);CHKERRQ(ierr);} 1739f5bf2dbfSMichael Lange ierr = PetscSFDestroy(&sfPoint);CHKERRQ(ierr); 1740a157612eSMichael Lange /* Cleanup overlap partition */ 1741a9f1d5b2SMichael Lange ierr = DMLabelDestroy(&lblOverlap);CHKERRQ(ierr); 1742a9f1d5b2SMichael Lange if (sf) *sf = sfOverlap; 1743a9f1d5b2SMichael Lange else {ierr = PetscSFDestroy(&sfOverlap);CHKERRQ(ierr);} 17441b858b30SMichael Lange ierr = PetscLogEventEnd(DMPLEX_DistributeOverlap, dm, 0, 0, 0);CHKERRQ(ierr); 1745a157612eSMichael Lange PetscFunctionReturn(0); 1746a157612eSMichael Lange } 1747