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