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