xref: /petsc/src/dm/impls/plex/plexpartition.c (revision 555a9cf8977bc36650a8aebd5df49b86389a659d)
170034214SMatthew G. Knepley #include <petsc-private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
270034214SMatthew G. Knepley 
377623264SMatthew G. Knepley PetscClassId PETSCPARTITIONER_CLASSID = 0;
477623264SMatthew G. Knepley 
577623264SMatthew G. Knepley PetscFunctionList PetscPartitionerList              = NULL;
677623264SMatthew G. Knepley PetscBool         PetscPartitionerRegisterAllCalled = PETSC_FALSE;
777623264SMatthew G. Knepley 
877623264SMatthew G. Knepley PetscBool ChacoPartitionercite = PETSC_FALSE;
977623264SMatthew G. Knepley const char ChacoPartitionerCitation[] = "@inproceedings{Chaco95,\n"
1077623264SMatthew G. Knepley                                "  author    = {Bruce Hendrickson and Robert Leland},\n"
1177623264SMatthew G. Knepley                                "  title     = {A multilevel algorithm for partitioning graphs},\n"
1277623264SMatthew G. Knepley                                "  booktitle = {Supercomputing '95: Proceedings of the 1995 ACM/IEEE Conference on Supercomputing (CDROM)},"
1377623264SMatthew G. Knepley                                "  isbn      = {0-89791-816-9},\n"
1477623264SMatthew G. Knepley                                "  pages     = {28},\n"
1577623264SMatthew G. Knepley                                "  doi       = {http://doi.acm.org/10.1145/224170.224228},\n"
1677623264SMatthew G. Knepley                                "  publisher = {ACM Press},\n"
1777623264SMatthew G. Knepley                                "  address   = {New York},\n"
1877623264SMatthew G. Knepley                                "  year      = {1995}\n}\n";
1977623264SMatthew G. Knepley 
2077623264SMatthew G. Knepley PetscBool ParMetisPartitionercite = PETSC_FALSE;
2177623264SMatthew G. Knepley const char ParMetisPartitionerCitation[] = "@article{KarypisKumar98,\n"
2277623264SMatthew G. Knepley                                "  author  = {George Karypis and Vipin Kumar},\n"
2377623264SMatthew G. Knepley                                "  title   = {A Parallel Algorithm for Multilevel Graph Partitioning and Sparse Matrix Ordering},\n"
2477623264SMatthew G. Knepley                                "  journal = {Journal of Parallel and Distributed Computing},\n"
2577623264SMatthew G. Knepley                                "  volume  = {48},\n"
2677623264SMatthew G. Knepley                                "  pages   = {71--85},\n"
2777623264SMatthew G. Knepley                                "  year    = {1998}\n}\n";
2877623264SMatthew G. Knepley 
2970034214SMatthew G. Knepley #undef __FUNCT__
30532c4e7dSMichael Lange #define __FUNCT__ "DMPlexCreatePartitionerGraph"
31532c4e7dSMichael Lange /*@C
32532c4e7dSMichael Lange   DMPlexCreatePartitionerGraph - Create a CSR graph of point connections for the partitioner
33532c4e7dSMichael Lange 
34532c4e7dSMichael Lange   Input Parameters:
35532c4e7dSMichael Lange + dm      - The mesh DM dm
36532c4e7dSMichael Lange - height  - Height of the strata from which to construct the graph
37532c4e7dSMichael Lange 
38532c4e7dSMichael Lange   Output Parameter:
39532c4e7dSMichael Lange + numVertices - Number of vertices in the graph
40532c4e7dSMichael Lange - offsets     - Point offsets in the graph
41532c4e7dSMichael Lange - adjacency   - Point connectivity in the graph
42532c4e7dSMichael Lange 
43532c4e7dSMichael Lange   The user can control the definition of adjacency for the mesh using DMPlexGetAdjacencyUseCone() and
44532c4e7dSMichael Lange   DMPlexSetAdjacencyUseClosure(). They should choose the combination appropriate for the function
45532c4e7dSMichael Lange   representation on the mesh.
46532c4e7dSMichael Lange 
47532c4e7dSMichael Lange   Level: developer
48532c4e7dSMichael Lange 
49532c4e7dSMichael Lange .seealso: PetscPartitionerGetType(), PetscPartitionerCreate(), DMPlexSetAdjacencyUseCone(), DMPlexSetAdjacencyUseClosure()
50532c4e7dSMichael Lange @*/
51532c4e7dSMichael Lange PetscErrorCode DMPlexCreatePartitionerGraph(DM dm, PetscInt height, PetscInt *numVertices, PetscInt **offsets, PetscInt **adjacency)
52532c4e7dSMichael Lange {
5343f7d02bSMichael Lange   PetscInt       p, pStart, pEnd, a, adjSize, idx, size, nroots;
54389e55d8SMichael Lange   PetscInt      *adj = NULL, *vOffsets = NULL, *graph = NULL;
558cfe4c1fSMichael Lange   IS             cellNumbering;
568cfe4c1fSMichael Lange   const PetscInt *cellNum;
57532c4e7dSMichael Lange   PetscBool      useCone, useClosure;
58532c4e7dSMichael Lange   PetscSection   section;
59532c4e7dSMichael Lange   PetscSegBuffer adjBuffer;
608cfe4c1fSMichael Lange   PetscSF        sfPoint;
61532c4e7dSMichael Lange   PetscMPIInt    rank;
62532c4e7dSMichael Lange   PetscErrorCode ierr;
63532c4e7dSMichael Lange 
64532c4e7dSMichael Lange   PetscFunctionBegin;
65532c4e7dSMichael Lange   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
66532c4e7dSMichael Lange   ierr = DMPlexGetHeightStratum(dm, height, &pStart, &pEnd);CHKERRQ(ierr);
678cfe4c1fSMichael Lange   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
688cfe4c1fSMichael Lange   ierr = PetscSFGetGraph(sfPoint, &nroots, NULL, NULL, NULL);CHKERRQ(ierr);
69532c4e7dSMichael Lange   /* Build adjacency graph via a section/segbuffer */
70532c4e7dSMichael Lange   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &section);CHKERRQ(ierr);
71532c4e7dSMichael Lange   ierr = PetscSectionSetChart(section, pStart, pEnd);CHKERRQ(ierr);
72532c4e7dSMichael Lange   ierr = PetscSegBufferCreate(sizeof(PetscInt),1000,&adjBuffer);CHKERRQ(ierr);
73532c4e7dSMichael Lange   /* Always use FVM adjacency to create partitioner graph */
74532c4e7dSMichael Lange   ierr = DMPlexGetAdjacencyUseCone(dm, &useCone);CHKERRQ(ierr);
75532c4e7dSMichael Lange   ierr = DMPlexGetAdjacencyUseClosure(dm, &useClosure);CHKERRQ(ierr);
76532c4e7dSMichael Lange   ierr = DMPlexSetAdjacencyUseCone(dm, PETSC_TRUE);CHKERRQ(ierr);
77532c4e7dSMichael Lange   ierr = DMPlexSetAdjacencyUseClosure(dm, PETSC_FALSE);CHKERRQ(ierr);
788cfe4c1fSMichael Lange   if (nroots > 0) {
798cfe4c1fSMichael Lange     ierr = DMPlexGetCellNumbering(dm, &cellNumbering);CHKERRQ(ierr);
808cfe4c1fSMichael Lange     ierr = ISGetIndices(cellNumbering, &cellNum);CHKERRQ(ierr);
818cfe4c1fSMichael Lange   }
828cfe4c1fSMichael Lange   for (*numVertices = 0, p = pStart; p < pEnd; p++) {
838cfe4c1fSMichael Lange     /* Skip non-owned cells in parallel (ParMetis expects no overlap) */
848cfe4c1fSMichael Lange     if (nroots > 0) {if (cellNum[p] < 0) continue;}
85532c4e7dSMichael Lange     adjSize = PETSC_DETERMINE;
86532c4e7dSMichael Lange     ierr = DMPlexGetAdjacency(dm, p, &adjSize, &adj);CHKERRQ(ierr);
87532c4e7dSMichael Lange     for (a = 0; a < adjSize; ++a) {
88532c4e7dSMichael Lange       const PetscInt point = adj[a];
89532c4e7dSMichael Lange       if (point != p && pStart <= point && point < pEnd) {
90532c4e7dSMichael Lange         PetscInt *PETSC_RESTRICT pBuf;
91532c4e7dSMichael Lange         ierr = PetscSectionAddDof(section, p, 1);CHKERRQ(ierr);
92532c4e7dSMichael Lange         ierr = PetscSegBufferGetInts(adjBuffer, 1, &pBuf);CHKERRQ(ierr);
93532c4e7dSMichael Lange         *pBuf = point;
94532c4e7dSMichael Lange       }
95532c4e7dSMichael Lange     }
968cfe4c1fSMichael Lange     (*numVertices)++;
97532c4e7dSMichael Lange   }
98532c4e7dSMichael Lange   ierr = DMPlexSetAdjacencyUseCone(dm, useCone);CHKERRQ(ierr);
99532c4e7dSMichael Lange   ierr = DMPlexSetAdjacencyUseClosure(dm, useClosure);CHKERRQ(ierr);
100532c4e7dSMichael Lange   /* Derive CSR graph from section/segbuffer */
101532c4e7dSMichael Lange   ierr = PetscSectionSetUp(section);CHKERRQ(ierr);
102532c4e7dSMichael Lange   ierr = PetscSectionGetStorageSize(section, &size);CHKERRQ(ierr);
103389e55d8SMichael Lange   ierr = PetscMalloc1(*numVertices+1, &vOffsets);CHKERRQ(ierr);
10443f7d02bSMichael Lange   for (idx = 0, p = pStart; p < pEnd; p++) {
10543f7d02bSMichael Lange     if (nroots > 0) {if (cellNum[p] < 0) continue;}
10643f7d02bSMichael Lange     ierr = PetscSectionGetOffset(section, p, &(vOffsets[idx++]));CHKERRQ(ierr);
10743f7d02bSMichael Lange   }
108532c4e7dSMichael Lange   vOffsets[*numVertices] = size;
109532c4e7dSMichael Lange   if (offsets) *offsets = vOffsets;
110389e55d8SMichael Lange   ierr = PetscSegBufferExtractAlloc(adjBuffer, &graph);CHKERRQ(ierr);
1118cfe4c1fSMichael Lange   if (nroots > 0) {
1128cfe4c1fSMichael Lange     ISLocalToGlobalMapping ltogCells;
1138cfe4c1fSMichael Lange     PetscInt n, size, *cells_arr;
1148cfe4c1fSMichael Lange     /* In parallel, apply a global cell numbering to the graph */
1158cfe4c1fSMichael Lange     ierr = ISRestoreIndices(cellNumbering, &cellNum);CHKERRQ(ierr);
1168cfe4c1fSMichael Lange     ierr = ISLocalToGlobalMappingCreateIS(cellNumbering, &ltogCells);CHKERRQ(ierr);
1178cfe4c1fSMichael Lange     ierr = ISLocalToGlobalMappingGetSize(ltogCells, &size);CHKERRQ(ierr);
1188cfe4c1fSMichael Lange     ierr = ISLocalToGlobalMappingGetIndices(ltogCells, (const PetscInt**)&cells_arr);CHKERRQ(ierr);
1198cfe4c1fSMichael Lange     /* Convert to positive global cell numbers */
1208cfe4c1fSMichael Lange     for (n=0; n<size; n++) {if (cells_arr[n] < 0) cells_arr[n] = -(cells_arr[n]+1);}
1218cfe4c1fSMichael Lange     ierr = ISLocalToGlobalMappingRestoreIndices(ltogCells, (const PetscInt**)&cells_arr);CHKERRQ(ierr);
122389e55d8SMichael Lange     ierr = ISLocalToGlobalMappingApplyBlock(ltogCells, vOffsets[*numVertices], graph, graph);CHKERRQ(ierr);
1238cfe4c1fSMichael Lange     ierr = ISLocalToGlobalMappingDestroy(&ltogCells);CHKERRQ(ierr);
1248cfe4c1fSMichael Lange   }
125389e55d8SMichael Lange   if (adjacency) *adjacency = graph;
126532c4e7dSMichael Lange   /* Clean up */
127532c4e7dSMichael Lange   ierr = PetscSegBufferDestroy(&adjBuffer);CHKERRQ(ierr);
128532c4e7dSMichael Lange   ierr = PetscSectionDestroy(&section);CHKERRQ(ierr);
129532c4e7dSMichael Lange   ierr = PetscFree(adj);CHKERRQ(ierr);
130532c4e7dSMichael Lange   PetscFunctionReturn(0);
131532c4e7dSMichael Lange }
132532c4e7dSMichael Lange 
133532c4e7dSMichael Lange #undef __FUNCT__
13470034214SMatthew G. Knepley #define __FUNCT__ "DMPlexCreateNeighborCSR"
13570034214SMatthew G. Knepley PetscErrorCode DMPlexCreateNeighborCSR(DM dm, PetscInt cellHeight, PetscInt *numVertices, PetscInt **offsets, PetscInt **adjacency)
13670034214SMatthew G. Knepley {
13770034214SMatthew G. Knepley   const PetscInt maxFaceCases = 30;
13870034214SMatthew G. Knepley   PetscInt       numFaceCases = 0;
13970034214SMatthew G. Knepley   PetscInt       numFaceVertices[30]; /* maxFaceCases, C89 sucks sucks sucks */
14070034214SMatthew G. Knepley   PetscInt      *off, *adj;
14170034214SMatthew G. Knepley   PetscInt      *neighborCells = NULL;
14270034214SMatthew G. Knepley   PetscInt       dim, cellDim, depth = 0, faceDepth, cStart, cEnd, c, numCells, cell;
14370034214SMatthew G. Knepley   PetscErrorCode ierr;
14470034214SMatthew G. Knepley 
14570034214SMatthew G. Knepley   PetscFunctionBegin;
14670034214SMatthew G. Knepley   /* For parallel partitioning, I think you have to communicate supports */
147c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
14870034214SMatthew G. Knepley   cellDim = dim - cellHeight;
14970034214SMatthew G. Knepley   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
15070034214SMatthew G. Knepley   ierr = DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr);
15170034214SMatthew G. Knepley   if (cEnd - cStart == 0) {
15270034214SMatthew G. Knepley     if (numVertices) *numVertices = 0;
15370034214SMatthew G. Knepley     if (offsets)   *offsets   = NULL;
15470034214SMatthew G. Knepley     if (adjacency) *adjacency = NULL;
15570034214SMatthew G. Knepley     PetscFunctionReturn(0);
15670034214SMatthew G. Knepley   }
15770034214SMatthew G. Knepley   numCells  = cEnd - cStart;
15870034214SMatthew G. Knepley   faceDepth = depth - cellHeight;
15970034214SMatthew G. Knepley   if (dim == depth) {
16070034214SMatthew G. Knepley     PetscInt f, fStart, fEnd;
16170034214SMatthew G. Knepley 
16270034214SMatthew G. Knepley     ierr = PetscCalloc1(numCells+1, &off);CHKERRQ(ierr);
16370034214SMatthew G. Knepley     /* Count neighboring cells */
16470034214SMatthew G. Knepley     ierr = DMPlexGetHeightStratum(dm, cellHeight+1, &fStart, &fEnd);CHKERRQ(ierr);
16570034214SMatthew G. Knepley     for (f = fStart; f < fEnd; ++f) {
16670034214SMatthew G. Knepley       const PetscInt *support;
16770034214SMatthew G. Knepley       PetscInt        supportSize;
16870034214SMatthew G. Knepley       ierr = DMPlexGetSupportSize(dm, f, &supportSize);CHKERRQ(ierr);
16970034214SMatthew G. Knepley       ierr = DMPlexGetSupport(dm, f, &support);CHKERRQ(ierr);
17070034214SMatthew G. Knepley       if (supportSize == 2) {
1719ffc88e4SToby Isaac         PetscInt numChildren;
1729ffc88e4SToby Isaac 
1739ffc88e4SToby Isaac         ierr = DMPlexGetTreeChildren(dm,f,&numChildren,NULL);CHKERRQ(ierr);
1749ffc88e4SToby Isaac         if (!numChildren) {
17570034214SMatthew G. Knepley           ++off[support[0]-cStart+1];
17670034214SMatthew G. Knepley           ++off[support[1]-cStart+1];
17770034214SMatthew G. Knepley         }
17870034214SMatthew G. Knepley       }
1799ffc88e4SToby Isaac     }
18070034214SMatthew G. Knepley     /* Prefix sum */
18170034214SMatthew G. Knepley     for (c = 1; c <= numCells; ++c) off[c] += off[c-1];
18270034214SMatthew G. Knepley     if (adjacency) {
18370034214SMatthew G. Knepley       PetscInt *tmp;
18470034214SMatthew G. Knepley 
18570034214SMatthew G. Knepley       ierr = PetscMalloc1(off[numCells], &adj);CHKERRQ(ierr);
186854ce69bSBarry Smith       ierr = PetscMalloc1(numCells+1, &tmp);CHKERRQ(ierr);
18770034214SMatthew G. Knepley       ierr = PetscMemcpy(tmp, off, (numCells+1) * sizeof(PetscInt));CHKERRQ(ierr);
18870034214SMatthew G. Knepley       /* Get neighboring cells */
18970034214SMatthew G. Knepley       for (f = fStart; f < fEnd; ++f) {
19070034214SMatthew G. Knepley         const PetscInt *support;
19170034214SMatthew G. Knepley         PetscInt        supportSize;
19270034214SMatthew G. Knepley         ierr = DMPlexGetSupportSize(dm, f, &supportSize);CHKERRQ(ierr);
19370034214SMatthew G. Knepley         ierr = DMPlexGetSupport(dm, f, &support);CHKERRQ(ierr);
19470034214SMatthew G. Knepley         if (supportSize == 2) {
1959ffc88e4SToby Isaac           PetscInt numChildren;
1969ffc88e4SToby Isaac 
1979ffc88e4SToby Isaac           ierr = DMPlexGetTreeChildren(dm,f,&numChildren,NULL);CHKERRQ(ierr);
1989ffc88e4SToby Isaac           if (!numChildren) {
19970034214SMatthew G. Knepley             adj[tmp[support[0]-cStart]++] = support[1];
20070034214SMatthew G. Knepley             adj[tmp[support[1]-cStart]++] = support[0];
20170034214SMatthew G. Knepley           }
20270034214SMatthew G. Knepley         }
2039ffc88e4SToby Isaac       }
20470034214SMatthew G. Knepley #if defined(PETSC_USE_DEBUG)
20570034214SMatthew G. Knepley       for (c = 0; c < cEnd-cStart; ++c) if (tmp[c] != off[c+1]) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Offset %d != %d for cell %d", tmp[c], off[c], c+cStart);
20670034214SMatthew G. Knepley #endif
20770034214SMatthew G. Knepley       ierr = PetscFree(tmp);CHKERRQ(ierr);
20870034214SMatthew G. Knepley     }
20970034214SMatthew G. Knepley     if (numVertices) *numVertices = numCells;
21070034214SMatthew G. Knepley     if (offsets)   *offsets   = off;
21170034214SMatthew G. Knepley     if (adjacency) *adjacency = adj;
21270034214SMatthew G. Knepley     PetscFunctionReturn(0);
21370034214SMatthew G. Knepley   }
21470034214SMatthew G. Knepley   /* Setup face recognition */
21570034214SMatthew G. Knepley   if (faceDepth == 1) {
21670034214SMatthew G. Knepley     PetscInt cornersSeen[30] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0}; /* Could use PetscBT */
21770034214SMatthew G. Knepley 
21870034214SMatthew G. Knepley     for (c = cStart; c < cEnd; ++c) {
21970034214SMatthew G. Knepley       PetscInt corners;
22070034214SMatthew G. Knepley 
22170034214SMatthew G. Knepley       ierr = DMPlexGetConeSize(dm, c, &corners);CHKERRQ(ierr);
22270034214SMatthew G. Knepley       if (!cornersSeen[corners]) {
22370034214SMatthew G. Knepley         PetscInt nFV;
22470034214SMatthew G. Knepley 
22570034214SMatthew G. Knepley         if (numFaceCases >= maxFaceCases) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Exceeded maximum number of face recognition cases");
22670034214SMatthew G. Knepley         cornersSeen[corners] = 1;
22770034214SMatthew G. Knepley 
22870034214SMatthew G. Knepley         ierr = DMPlexGetNumFaceVertices(dm, cellDim, corners, &nFV);CHKERRQ(ierr);
22970034214SMatthew G. Knepley 
23070034214SMatthew G. Knepley         numFaceVertices[numFaceCases++] = nFV;
23170034214SMatthew G. Knepley       }
23270034214SMatthew G. Knepley     }
23370034214SMatthew G. Knepley   }
23470034214SMatthew G. Knepley   ierr = PetscCalloc1(numCells+1, &off);CHKERRQ(ierr);
23570034214SMatthew G. Knepley   /* Count neighboring cells */
23670034214SMatthew G. Knepley   for (cell = cStart; cell < cEnd; ++cell) {
23770034214SMatthew G. Knepley     PetscInt numNeighbors = PETSC_DETERMINE, n;
23870034214SMatthew G. Knepley 
2398b0b4c70SToby Isaac     ierr = DMPlexGetAdjacency_Internal(dm, cell, PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, &numNeighbors, &neighborCells);CHKERRQ(ierr);
24070034214SMatthew G. Knepley     /* Get meet with each cell, and check with recognizer (could optimize to check each pair only once) */
24170034214SMatthew G. Knepley     for (n = 0; n < numNeighbors; ++n) {
24270034214SMatthew G. Knepley       PetscInt        cellPair[2];
24370034214SMatthew G. Knepley       PetscBool       found    = faceDepth > 1 ? PETSC_TRUE : PETSC_FALSE;
24470034214SMatthew G. Knepley       PetscInt        meetSize = 0;
24570034214SMatthew G. Knepley       const PetscInt *meet    = NULL;
24670034214SMatthew G. Knepley 
24770034214SMatthew G. Knepley       cellPair[0] = cell; cellPair[1] = neighborCells[n];
24870034214SMatthew G. Knepley       if (cellPair[0] == cellPair[1]) continue;
24970034214SMatthew G. Knepley       if (!found) {
25070034214SMatthew G. Knepley         ierr = DMPlexGetMeet(dm, 2, cellPair, &meetSize, &meet);CHKERRQ(ierr);
25170034214SMatthew G. Knepley         if (meetSize) {
25270034214SMatthew G. Knepley           PetscInt f;
25370034214SMatthew G. Knepley 
25470034214SMatthew G. Knepley           for (f = 0; f < numFaceCases; ++f) {
25570034214SMatthew G. Knepley             if (numFaceVertices[f] == meetSize) {
25670034214SMatthew G. Knepley               found = PETSC_TRUE;
25770034214SMatthew G. Knepley               break;
25870034214SMatthew G. Knepley             }
25970034214SMatthew G. Knepley           }
26070034214SMatthew G. Knepley         }
26170034214SMatthew G. Knepley         ierr = DMPlexRestoreMeet(dm, 2, cellPair, &meetSize, &meet);CHKERRQ(ierr);
26270034214SMatthew G. Knepley       }
26370034214SMatthew G. Knepley       if (found) ++off[cell-cStart+1];
26470034214SMatthew G. Knepley     }
26570034214SMatthew G. Knepley   }
26670034214SMatthew G. Knepley   /* Prefix sum */
26770034214SMatthew G. Knepley   for (cell = 1; cell <= numCells; ++cell) off[cell] += off[cell-1];
26870034214SMatthew G. Knepley 
26970034214SMatthew G. Knepley   if (adjacency) {
27070034214SMatthew G. Knepley     ierr = PetscMalloc1(off[numCells], &adj);CHKERRQ(ierr);
27170034214SMatthew G. Knepley     /* Get neighboring cells */
27270034214SMatthew G. Knepley     for (cell = cStart; cell < cEnd; ++cell) {
27370034214SMatthew G. Knepley       PetscInt numNeighbors = PETSC_DETERMINE, n;
27470034214SMatthew G. Knepley       PetscInt cellOffset   = 0;
27570034214SMatthew G. Knepley 
2768b0b4c70SToby Isaac       ierr = DMPlexGetAdjacency_Internal(dm, cell, PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, &numNeighbors, &neighborCells);CHKERRQ(ierr);
27770034214SMatthew G. Knepley       /* Get meet with each cell, and check with recognizer (could optimize to check each pair only once) */
27870034214SMatthew G. Knepley       for (n = 0; n < numNeighbors; ++n) {
27970034214SMatthew G. Knepley         PetscInt        cellPair[2];
28070034214SMatthew G. Knepley         PetscBool       found    = faceDepth > 1 ? PETSC_TRUE : PETSC_FALSE;
28170034214SMatthew G. Knepley         PetscInt        meetSize = 0;
28270034214SMatthew G. Knepley         const PetscInt *meet    = NULL;
28370034214SMatthew G. Knepley 
28470034214SMatthew G. Knepley         cellPair[0] = cell; cellPair[1] = neighborCells[n];
28570034214SMatthew G. Knepley         if (cellPair[0] == cellPair[1]) continue;
28670034214SMatthew G. Knepley         if (!found) {
28770034214SMatthew G. Knepley           ierr = DMPlexGetMeet(dm, 2, cellPair, &meetSize, &meet);CHKERRQ(ierr);
28870034214SMatthew G. Knepley           if (meetSize) {
28970034214SMatthew G. Knepley             PetscInt f;
29070034214SMatthew G. Knepley 
29170034214SMatthew G. Knepley             for (f = 0; f < numFaceCases; ++f) {
29270034214SMatthew G. Knepley               if (numFaceVertices[f] == meetSize) {
29370034214SMatthew G. Knepley                 found = PETSC_TRUE;
29470034214SMatthew G. Knepley                 break;
29570034214SMatthew G. Knepley               }
29670034214SMatthew G. Knepley             }
29770034214SMatthew G. Knepley           }
29870034214SMatthew G. Knepley           ierr = DMPlexRestoreMeet(dm, 2, cellPair, &meetSize, &meet);CHKERRQ(ierr);
29970034214SMatthew G. Knepley         }
30070034214SMatthew G. Knepley         if (found) {
30170034214SMatthew G. Knepley           adj[off[cell-cStart]+cellOffset] = neighborCells[n];
30270034214SMatthew G. Knepley           ++cellOffset;
30370034214SMatthew G. Knepley         }
30470034214SMatthew G. Knepley       }
30570034214SMatthew G. Knepley     }
30670034214SMatthew G. Knepley   }
30770034214SMatthew G. Knepley   ierr = PetscFree(neighborCells);CHKERRQ(ierr);
30870034214SMatthew G. Knepley   if (numVertices) *numVertices = numCells;
30970034214SMatthew G. Knepley   if (offsets)   *offsets   = off;
31070034214SMatthew G. Knepley   if (adjacency) *adjacency = adj;
31170034214SMatthew G. Knepley   PetscFunctionReturn(0);
31270034214SMatthew G. Knepley }
31370034214SMatthew G. Knepley 
31477623264SMatthew G. Knepley #undef __FUNCT__
31577623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerRegister"
31677623264SMatthew G. Knepley /*@C
31777623264SMatthew G. Knepley   PetscPartitionerRegister - Adds a new PetscPartitioner implementation
31877623264SMatthew G. Knepley 
31977623264SMatthew G. Knepley   Not Collective
32077623264SMatthew G. Knepley 
32177623264SMatthew G. Knepley   Input Parameters:
32277623264SMatthew G. Knepley + name        - The name of a new user-defined creation routine
32377623264SMatthew G. Knepley - create_func - The creation routine itself
32477623264SMatthew G. Knepley 
32577623264SMatthew G. Knepley   Notes:
32677623264SMatthew G. Knepley   PetscPartitionerRegister() may be called multiple times to add several user-defined PetscPartitioners
32777623264SMatthew G. Knepley 
32877623264SMatthew G. Knepley   Sample usage:
32977623264SMatthew G. Knepley .vb
33077623264SMatthew G. Knepley     PetscPartitionerRegister("my_part", MyPetscPartitionerCreate);
33177623264SMatthew G. Knepley .ve
33277623264SMatthew G. Knepley 
33377623264SMatthew G. Knepley   Then, your PetscPartitioner type can be chosen with the procedural interface via
33477623264SMatthew G. Knepley .vb
33577623264SMatthew G. Knepley     PetscPartitionerCreate(MPI_Comm, PetscPartitioner *);
33677623264SMatthew G. Knepley     PetscPartitionerSetType(PetscPartitioner, "my_part");
33777623264SMatthew G. Knepley .ve
33877623264SMatthew G. Knepley    or at runtime via the option
33977623264SMatthew G. Knepley .vb
34077623264SMatthew G. Knepley     -petscpartitioner_type my_part
34177623264SMatthew G. Knepley .ve
34277623264SMatthew G. Knepley 
34377623264SMatthew G. Knepley   Level: advanced
34477623264SMatthew G. Knepley 
34577623264SMatthew G. Knepley .keywords: PetscPartitioner, register
34677623264SMatthew G. Knepley .seealso: PetscPartitionerRegisterAll(), PetscPartitionerRegisterDestroy()
34777623264SMatthew G. Knepley 
34877623264SMatthew G. Knepley @*/
34977623264SMatthew G. Knepley PetscErrorCode PetscPartitionerRegister(const char sname[], PetscErrorCode (*function)(PetscPartitioner))
35077623264SMatthew G. Knepley {
35177623264SMatthew G. Knepley   PetscErrorCode ierr;
35277623264SMatthew G. Knepley 
35377623264SMatthew G. Knepley   PetscFunctionBegin;
35477623264SMatthew G. Knepley   ierr = PetscFunctionListAdd(&PetscPartitionerList, sname, function);CHKERRQ(ierr);
35577623264SMatthew G. Knepley   PetscFunctionReturn(0);
35677623264SMatthew G. Knepley }
35777623264SMatthew G. Knepley 
35877623264SMatthew G. Knepley #undef __FUNCT__
35977623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerSetType"
36077623264SMatthew G. Knepley /*@C
36177623264SMatthew G. Knepley   PetscPartitionerSetType - Builds a particular PetscPartitioner
36277623264SMatthew G. Knepley 
36377623264SMatthew G. Knepley   Collective on PetscPartitioner
36477623264SMatthew G. Knepley 
36577623264SMatthew G. Knepley   Input Parameters:
36677623264SMatthew G. Knepley + part - The PetscPartitioner object
36777623264SMatthew G. Knepley - name - The kind of partitioner
36877623264SMatthew G. Knepley 
36977623264SMatthew G. Knepley   Options Database Key:
37077623264SMatthew G. Knepley . -petscpartitioner_type <type> - Sets the PetscPartitioner type; use -help for a list of available types
37177623264SMatthew G. Knepley 
37277623264SMatthew G. Knepley   Level: intermediate
37377623264SMatthew G. Knepley 
37477623264SMatthew G. Knepley .keywords: PetscPartitioner, set, type
37577623264SMatthew G. Knepley .seealso: PetscPartitionerGetType(), PetscPartitionerCreate()
37677623264SMatthew G. Knepley @*/
37777623264SMatthew G. Knepley PetscErrorCode PetscPartitionerSetType(PetscPartitioner part, PetscPartitionerType name)
37877623264SMatthew G. Knepley {
37977623264SMatthew G. Knepley   PetscErrorCode (*r)(PetscPartitioner);
38077623264SMatthew G. Knepley   PetscBool      match;
38177623264SMatthew G. Knepley   PetscErrorCode ierr;
38277623264SMatthew G. Knepley 
38377623264SMatthew G. Knepley   PetscFunctionBegin;
38477623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
38577623264SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) part, name, &match);CHKERRQ(ierr);
38677623264SMatthew G. Knepley   if (match) PetscFunctionReturn(0);
38777623264SMatthew G. Knepley 
38877623264SMatthew G. Knepley   if (!PetscPartitionerRegisterAllCalled) {ierr = PetscPartitionerRegisterAll();CHKERRQ(ierr);}
38977623264SMatthew G. Knepley   ierr = PetscFunctionListFind(PetscPartitionerList, name, &r);CHKERRQ(ierr);
39077623264SMatthew G. Knepley   if (!r) SETERRQ1(PetscObjectComm((PetscObject) part), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown PetscPartitioner type: %s", name);
39177623264SMatthew G. Knepley 
39277623264SMatthew G. Knepley   if (part->ops->destroy) {
39377623264SMatthew G. Knepley     ierr              = (*part->ops->destroy)(part);CHKERRQ(ierr);
39477623264SMatthew G. Knepley     part->ops->destroy = NULL;
39577623264SMatthew G. Knepley   }
39677623264SMatthew G. Knepley   ierr = (*r)(part);CHKERRQ(ierr);
39777623264SMatthew G. Knepley   ierr = PetscObjectChangeTypeName((PetscObject) part, name);CHKERRQ(ierr);
39877623264SMatthew G. Knepley   PetscFunctionReturn(0);
39977623264SMatthew G. Knepley }
40077623264SMatthew G. Knepley 
40177623264SMatthew G. Knepley #undef __FUNCT__
40277623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerGetType"
40377623264SMatthew G. Knepley /*@C
40477623264SMatthew G. Knepley   PetscPartitionerGetType - Gets the PetscPartitioner type name (as a string) from the object.
40577623264SMatthew G. Knepley 
40677623264SMatthew G. Knepley   Not Collective
40777623264SMatthew G. Knepley 
40877623264SMatthew G. Knepley   Input Parameter:
40977623264SMatthew G. Knepley . part - The PetscPartitioner
41077623264SMatthew G. Knepley 
41177623264SMatthew G. Knepley   Output Parameter:
41277623264SMatthew G. Knepley . name - The PetscPartitioner type name
41377623264SMatthew G. Knepley 
41477623264SMatthew G. Knepley   Level: intermediate
41577623264SMatthew G. Knepley 
41677623264SMatthew G. Knepley .keywords: PetscPartitioner, get, type, name
41777623264SMatthew G. Knepley .seealso: PetscPartitionerSetType(), PetscPartitionerCreate()
41877623264SMatthew G. Knepley @*/
41977623264SMatthew G. Knepley PetscErrorCode PetscPartitionerGetType(PetscPartitioner part, PetscPartitionerType *name)
42077623264SMatthew G. Knepley {
42177623264SMatthew G. Knepley   PetscErrorCode ierr;
42277623264SMatthew G. Knepley 
42377623264SMatthew G. Knepley   PetscFunctionBegin;
42477623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
42577623264SMatthew G. Knepley   PetscValidPointer(name, 2);
42677623264SMatthew G. Knepley   if (!PetscPartitionerRegisterAllCalled) {ierr = PetscPartitionerRegisterAll();CHKERRQ(ierr);}
42777623264SMatthew G. Knepley   *name = ((PetscObject) part)->type_name;
42877623264SMatthew G. Knepley   PetscFunctionReturn(0);
42977623264SMatthew G. Knepley }
43077623264SMatthew G. Knepley 
43177623264SMatthew G. Knepley #undef __FUNCT__
43277623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView"
43377623264SMatthew G. Knepley /*@C
43477623264SMatthew G. Knepley   PetscPartitionerView - Views a PetscPartitioner
43577623264SMatthew G. Knepley 
43677623264SMatthew G. Knepley   Collective on PetscPartitioner
43777623264SMatthew G. Knepley 
43877623264SMatthew G. Knepley   Input Parameter:
43977623264SMatthew G. Knepley + part - the PetscPartitioner object to view
44077623264SMatthew G. Knepley - v    - the viewer
44177623264SMatthew G. Knepley 
44277623264SMatthew G. Knepley   Level: developer
44377623264SMatthew G. Knepley 
44477623264SMatthew G. Knepley .seealso: PetscPartitionerDestroy()
44577623264SMatthew G. Knepley @*/
44677623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView(PetscPartitioner part, PetscViewer v)
44777623264SMatthew G. Knepley {
44877623264SMatthew G. Knepley   PetscErrorCode ierr;
44977623264SMatthew G. Knepley 
45077623264SMatthew G. Knepley   PetscFunctionBegin;
45177623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
45277623264SMatthew G. Knepley   if (!v) {ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject) part), &v);CHKERRQ(ierr);}
45377623264SMatthew G. Knepley   if (part->ops->view) {ierr = (*part->ops->view)(part, v);CHKERRQ(ierr);}
45477623264SMatthew G. Knepley   PetscFunctionReturn(0);
45577623264SMatthew G. Knepley }
45677623264SMatthew G. Knepley 
45777623264SMatthew G. Knepley #undef __FUNCT__
45877623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerViewFromOptions"
45977623264SMatthew G. Knepley /*
46077623264SMatthew G. Knepley   PetscPartitionerViewFromOptions - Processes command line options to determine if/how a PetscPartitioner is to be viewed.
46177623264SMatthew G. Knepley 
46277623264SMatthew G. Knepley   Collective on PetscPartitioner
46377623264SMatthew G. Knepley 
46477623264SMatthew G. Knepley   Input Parameters:
46577623264SMatthew G. Knepley + part   - the PetscPartitioner
46677623264SMatthew G. Knepley . prefix - prefix to use for viewing, or NULL
46777623264SMatthew G. Knepley - optionname - option to activate viewing
46877623264SMatthew G. Knepley 
46977623264SMatthew G. Knepley   Level: intermediate
47077623264SMatthew G. Knepley 
47177623264SMatthew G. Knepley .keywords: PetscPartitioner, view, options, database
47277623264SMatthew G. Knepley .seealso: VecViewFromOptions(), MatViewFromOptions()
47377623264SMatthew G. Knepley */
47477623264SMatthew G. Knepley PetscErrorCode PetscPartitionerViewFromOptions(PetscPartitioner part, const char prefix[], const char optionname[])
47577623264SMatthew G. Knepley {
47677623264SMatthew G. Knepley   PetscViewer       viewer;
47777623264SMatthew G. Knepley   PetscViewerFormat format;
47877623264SMatthew G. Knepley   PetscBool         flg;
47977623264SMatthew G. Knepley   PetscErrorCode    ierr;
48077623264SMatthew G. Knepley 
48177623264SMatthew G. Knepley   PetscFunctionBegin;
48277623264SMatthew G. Knepley   if (prefix) {ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) part), prefix,                       optionname, &viewer, &format, &flg);CHKERRQ(ierr);}
48377623264SMatthew G. Knepley   else        {ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) part), ((PetscObject) part)->prefix, optionname, &viewer, &format, &flg);CHKERRQ(ierr);}
48477623264SMatthew G. Knepley   if (flg) {
48577623264SMatthew G. Knepley     ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr);
48677623264SMatthew G. Knepley     ierr = PetscPartitionerView(part, viewer);CHKERRQ(ierr);
48777623264SMatthew G. Knepley     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
48877623264SMatthew G. Knepley     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
48977623264SMatthew G. Knepley   }
49077623264SMatthew G. Knepley   PetscFunctionReturn(0);
49177623264SMatthew G. Knepley }
49277623264SMatthew G. Knepley #undef __FUNCT__
49377623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerSetTypeFromOptions_Internal"
49477623264SMatthew G. Knepley PetscErrorCode PetscPartitionerSetTypeFromOptions_Internal(PetscPartitioner part)
49577623264SMatthew G. Knepley {
49677623264SMatthew G. Knepley   const char    *defaultType;
49777623264SMatthew G. Knepley   char           name[256];
49877623264SMatthew G. Knepley   PetscBool      flg;
49977623264SMatthew G. Knepley   PetscErrorCode ierr;
50077623264SMatthew G. Knepley 
50177623264SMatthew G. Knepley   PetscFunctionBegin;
50277623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
50377623264SMatthew G. Knepley   if (!((PetscObject) part)->type_name) defaultType = PETSCPARTITIONERCHACO;
50477623264SMatthew G. Knepley   else                                  defaultType = ((PetscObject) part)->type_name;
50577623264SMatthew G. Knepley   if (!PetscPartitionerRegisterAllCalled) {ierr = PetscPartitionerRegisterAll();CHKERRQ(ierr);}
50677623264SMatthew G. Knepley 
50777623264SMatthew G. Knepley   ierr = PetscObjectOptionsBegin((PetscObject) part);CHKERRQ(ierr);
50877623264SMatthew G. Knepley   ierr = PetscOptionsFList("-petscpartitioner_type", "Graph partitioner", "PetscPartitionerSetType", PetscPartitionerList, defaultType, name, 256, &flg);CHKERRQ(ierr);
50977623264SMatthew G. Knepley   if (flg) {
51077623264SMatthew G. Knepley     ierr = PetscPartitionerSetType(part, name);CHKERRQ(ierr);
51177623264SMatthew G. Knepley   } else if (!((PetscObject) part)->type_name) {
51277623264SMatthew G. Knepley     ierr = PetscPartitionerSetType(part, defaultType);CHKERRQ(ierr);
51377623264SMatthew G. Knepley   }
51477623264SMatthew G. Knepley   ierr = PetscOptionsEnd();CHKERRQ(ierr);
51577623264SMatthew G. Knepley   PetscFunctionReturn(0);
51677623264SMatthew G. Knepley }
51777623264SMatthew G. Knepley 
51877623264SMatthew G. Knepley #undef __FUNCT__
51977623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerSetFromOptions"
52077623264SMatthew G. Knepley /*@
52177623264SMatthew G. Knepley   PetscPartitionerSetFromOptions - sets parameters in a PetscPartitioner from the options database
52277623264SMatthew G. Knepley 
52377623264SMatthew G. Knepley   Collective on PetscPartitioner
52477623264SMatthew G. Knepley 
52577623264SMatthew G. Knepley   Input Parameter:
52677623264SMatthew G. Knepley . part - the PetscPartitioner object to set options for
52777623264SMatthew G. Knepley 
52877623264SMatthew G. Knepley   Level: developer
52977623264SMatthew G. Knepley 
53077623264SMatthew G. Knepley .seealso: PetscPartitionerView()
53177623264SMatthew G. Knepley @*/
53277623264SMatthew G. Knepley PetscErrorCode PetscPartitionerSetFromOptions(PetscPartitioner part)
53377623264SMatthew G. Knepley {
53477623264SMatthew G. Knepley   PetscErrorCode ierr;
53577623264SMatthew G. Knepley 
53677623264SMatthew G. Knepley   PetscFunctionBegin;
53777623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
53877623264SMatthew G. Knepley   ierr = PetscPartitionerSetTypeFromOptions_Internal(part);CHKERRQ(ierr);
53977623264SMatthew G. Knepley 
54077623264SMatthew G. Knepley   ierr = PetscObjectOptionsBegin((PetscObject) part);CHKERRQ(ierr);
54177623264SMatthew G. Knepley   if (part->ops->setfromoptions) {ierr = (*part->ops->setfromoptions)(part);CHKERRQ(ierr);}
54277623264SMatthew G. Knepley   /* process any options handlers added with PetscObjectAddOptionsHandler() */
54377623264SMatthew G. Knepley   ierr = PetscObjectProcessOptionsHandlers((PetscObject) part);CHKERRQ(ierr);
54477623264SMatthew G. Knepley   ierr = PetscOptionsEnd();CHKERRQ(ierr);
54577623264SMatthew G. Knepley   ierr = PetscPartitionerViewFromOptions(part, NULL, "-petscpartitioner_view");CHKERRQ(ierr);
54677623264SMatthew G. Knepley   PetscFunctionReturn(0);
54777623264SMatthew G. Knepley }
54877623264SMatthew G. Knepley 
54977623264SMatthew G. Knepley #undef __FUNCT__
55077623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerSetUp"
55177623264SMatthew G. Knepley /*@C
55277623264SMatthew G. Knepley   PetscPartitionerSetUp - Construct data structures for the PetscPartitioner
55377623264SMatthew G. Knepley 
55477623264SMatthew G. Knepley   Collective on PetscPartitioner
55577623264SMatthew G. Knepley 
55677623264SMatthew G. Knepley   Input Parameter:
55777623264SMatthew G. Knepley . part - the PetscPartitioner object to setup
55877623264SMatthew G. Knepley 
55977623264SMatthew G. Knepley   Level: developer
56077623264SMatthew G. Knepley 
56177623264SMatthew G. Knepley .seealso: PetscPartitionerView(), PetscPartitionerDestroy()
56277623264SMatthew G. Knepley @*/
56377623264SMatthew G. Knepley PetscErrorCode PetscPartitionerSetUp(PetscPartitioner part)
56477623264SMatthew G. Knepley {
56577623264SMatthew G. Knepley   PetscErrorCode ierr;
56677623264SMatthew G. Knepley 
56777623264SMatthew G. Knepley   PetscFunctionBegin;
56877623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
56977623264SMatthew G. Knepley   if (part->ops->setup) {ierr = (*part->ops->setup)(part);CHKERRQ(ierr);}
57077623264SMatthew G. Knepley   PetscFunctionReturn(0);
57177623264SMatthew G. Knepley }
57277623264SMatthew G. Knepley 
57377623264SMatthew G. Knepley #undef __FUNCT__
57477623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerDestroy"
57577623264SMatthew G. Knepley /*@
57677623264SMatthew G. Knepley   PetscPartitionerDestroy - Destroys a PetscPartitioner object
57777623264SMatthew G. Knepley 
57877623264SMatthew G. Knepley   Collective on PetscPartitioner
57977623264SMatthew G. Knepley 
58077623264SMatthew G. Knepley   Input Parameter:
58177623264SMatthew G. Knepley . part - the PetscPartitioner object to destroy
58277623264SMatthew G. Knepley 
58377623264SMatthew G. Knepley   Level: developer
58477623264SMatthew G. Knepley 
58577623264SMatthew G. Knepley .seealso: PetscPartitionerView()
58677623264SMatthew G. Knepley @*/
58777623264SMatthew G. Knepley PetscErrorCode PetscPartitionerDestroy(PetscPartitioner *part)
58877623264SMatthew G. Knepley {
58977623264SMatthew G. Knepley   PetscErrorCode ierr;
59077623264SMatthew G. Knepley 
59177623264SMatthew G. Knepley   PetscFunctionBegin;
59277623264SMatthew G. Knepley   if (!*part) PetscFunctionReturn(0);
59377623264SMatthew G. Knepley   PetscValidHeaderSpecific((*part), PETSCPARTITIONER_CLASSID, 1);
59477623264SMatthew G. Knepley 
59577623264SMatthew G. Knepley   if (--((PetscObject)(*part))->refct > 0) {*part = 0; PetscFunctionReturn(0);}
59677623264SMatthew G. Knepley   ((PetscObject) (*part))->refct = 0;
59777623264SMatthew G. Knepley 
59877623264SMatthew G. Knepley   if ((*part)->ops->destroy) {ierr = (*(*part)->ops->destroy)(*part);CHKERRQ(ierr);}
59977623264SMatthew G. Knepley   ierr = PetscHeaderDestroy(part);CHKERRQ(ierr);
60077623264SMatthew G. Knepley   PetscFunctionReturn(0);
60177623264SMatthew G. Knepley }
60277623264SMatthew G. Knepley 
60377623264SMatthew G. Knepley #undef __FUNCT__
60477623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerCreate"
60577623264SMatthew G. Knepley /*@
60677623264SMatthew G. Knepley   PetscPartitionerCreate - Creates an empty PetscPartitioner object. The type can then be set with PetscPartitionerSetType().
60777623264SMatthew G. Knepley 
60877623264SMatthew G. Knepley   Collective on MPI_Comm
60977623264SMatthew G. Knepley 
61077623264SMatthew G. Knepley   Input Parameter:
61177623264SMatthew G. Knepley . comm - The communicator for the PetscPartitioner object
61277623264SMatthew G. Knepley 
61377623264SMatthew G. Knepley   Output Parameter:
61477623264SMatthew G. Knepley . part - The PetscPartitioner object
61577623264SMatthew G. Knepley 
61677623264SMatthew G. Knepley   Level: beginner
61777623264SMatthew G. Knepley 
618*555a9cf8SMatthew G. Knepley .seealso: PetscPartitionerSetType(), PETSCPARTITIONERCHACO, PETSCPARTITIONERPARMETIS, PETSCPARTITIONERSHELL, PETSCPARTITIONERSIMPLE
61977623264SMatthew G. Knepley @*/
62077623264SMatthew G. Knepley PetscErrorCode PetscPartitionerCreate(MPI_Comm comm, PetscPartitioner *part)
62177623264SMatthew G. Knepley {
62277623264SMatthew G. Knepley   PetscPartitioner p;
62377623264SMatthew G. Knepley   PetscErrorCode   ierr;
62477623264SMatthew G. Knepley 
62577623264SMatthew G. Knepley   PetscFunctionBegin;
62677623264SMatthew G. Knepley   PetscValidPointer(part, 2);
62777623264SMatthew G. Knepley   *part = NULL;
62877623264SMatthew G. Knepley   ierr = PetscFVInitializePackage();CHKERRQ(ierr);
62977623264SMatthew G. Knepley 
63077623264SMatthew G. Knepley   ierr = PetscHeaderCreate(p, _p_PetscPartitioner, struct _PetscPartitionerOps, PETSCPARTITIONER_CLASSID, "PetscPartitioner", "Graph Partitioner", "PetscPartitioner", comm, PetscPartitionerDestroy, PetscPartitionerView);CHKERRQ(ierr);
63177623264SMatthew G. Knepley   ierr = PetscMemzero(p->ops, sizeof(struct _PetscPartitionerOps));CHKERRQ(ierr);
63277623264SMatthew G. Knepley 
63377623264SMatthew G. Knepley   *part = p;
63477623264SMatthew G. Knepley   PetscFunctionReturn(0);
63577623264SMatthew G. Knepley }
63677623264SMatthew G. Knepley 
63777623264SMatthew G. Knepley #undef __FUNCT__
63877623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerPartition"
63977623264SMatthew G. Knepley /*@
64077623264SMatthew G. Knepley   PetscPartitionerPartition - Create a non-overlapping partition of the cells in the mesh
64177623264SMatthew G. Knepley 
64277623264SMatthew G. Knepley   Collective on DM
64377623264SMatthew G. Knepley 
64477623264SMatthew G. Knepley   Input Parameters:
64577623264SMatthew G. Knepley + part    - The PetscPartitioner
646f8987ae8SMichael Lange - dm      - The mesh DM
64777623264SMatthew G. Knepley 
64877623264SMatthew G. Knepley   Output Parameters:
64977623264SMatthew G. Knepley + partSection     - The PetscSection giving the division of points by partition
650f8987ae8SMichael Lange - partition       - The list of points by partition
65177623264SMatthew G. Knepley 
65277623264SMatthew G. Knepley   Note: Instead of cells, points at a given height can be partitioned by calling PetscPartitionerSetPointHeight()
65377623264SMatthew G. Knepley 
65477623264SMatthew G. Knepley   Level: developer
65577623264SMatthew G. Knepley 
65677623264SMatthew G. Knepley .seealso DMPlexDistribute(), PetscPartitionerSetPointHeight(), PetscPartitionerCreate()
6574b15ede2SMatthew G. Knepley @*/
658f8987ae8SMichael Lange PetscErrorCode PetscPartitionerPartition(PetscPartitioner part, DM dm, PetscSection partSection, IS *partition)
65977623264SMatthew G. Knepley {
66077623264SMatthew G. Knepley   PetscMPIInt    size;
66177623264SMatthew G. Knepley   PetscErrorCode ierr;
66277623264SMatthew G. Knepley 
66377623264SMatthew G. Knepley   PetscFunctionBegin;
66477623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
66577623264SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 2);
66677623264SMatthew G. Knepley   PetscValidHeaderSpecific(partSection, PETSC_SECTION_CLASSID, 4);
66777623264SMatthew G. Knepley   PetscValidPointer(partition, 5);
66877623264SMatthew G. Knepley   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) part), &size);CHKERRQ(ierr);
66977623264SMatthew G. Knepley   if (size == 1) {
67077623264SMatthew G. Knepley     PetscInt *points;
67177623264SMatthew G. Knepley     PetscInt  cStart, cEnd, c;
67277623264SMatthew G. Knepley 
67377623264SMatthew G. Knepley     ierr = DMPlexGetHeightStratum(dm, part->height, &cStart, &cEnd);CHKERRQ(ierr);
67477623264SMatthew G. Knepley     ierr = PetscSectionSetChart(partSection, 0, size);CHKERRQ(ierr);
67577623264SMatthew G. Knepley     ierr = PetscSectionSetDof(partSection, 0, cEnd-cStart);CHKERRQ(ierr);
67677623264SMatthew G. Knepley     ierr = PetscSectionSetUp(partSection);CHKERRQ(ierr);
67777623264SMatthew G. Knepley     ierr = PetscMalloc1(cEnd-cStart, &points);CHKERRQ(ierr);
67877623264SMatthew G. Knepley     for (c = cStart; c < cEnd; ++c) points[c] = c;
67977623264SMatthew G. Knepley     ierr = ISCreateGeneral(PetscObjectComm((PetscObject) part), cEnd-cStart, points, PETSC_OWN_POINTER, partition);CHKERRQ(ierr);
68077623264SMatthew G. Knepley     PetscFunctionReturn(0);
68177623264SMatthew G. Knepley   }
68277623264SMatthew G. Knepley   if (part->height == 0) {
68377623264SMatthew G. Knepley     PetscInt  numVertices;
68477623264SMatthew G. Knepley     PetscInt *start     = NULL;
68577623264SMatthew G. Knepley     PetscInt *adjacency = NULL;
68677623264SMatthew G. Knepley 
687532c4e7dSMichael Lange     ierr = DMPlexCreatePartitionerGraph(dm, 0, &numVertices, &start, &adjacency);CHKERRQ(ierr);
68877623264SMatthew G. Knepley     if (!part->ops->partition) SETERRQ(PetscObjectComm((PetscObject) part), PETSC_ERR_ARG_WRONGSTATE, "PetscPartitioner has no type");
68977623264SMatthew G. Knepley     ierr = (*part->ops->partition)(part, dm, size, numVertices, start, adjacency, partSection, partition);CHKERRQ(ierr);
69077623264SMatthew G. Knepley     ierr = PetscFree(start);CHKERRQ(ierr);
69177623264SMatthew G. Knepley     ierr = PetscFree(adjacency);CHKERRQ(ierr);
69277623264SMatthew G. Knepley   } else SETERRQ1(PetscObjectComm((PetscObject) part), PETSC_ERR_ARG_OUTOFRANGE, "Invalid height %D for points to partition", part->height);
69377623264SMatthew G. Knepley   PetscFunctionReturn(0);
69477623264SMatthew G. Knepley }
69577623264SMatthew G. Knepley 
69677623264SMatthew G. Knepley #undef __FUNCT__
69777623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerDestroy_Shell"
69877623264SMatthew G. Knepley PetscErrorCode PetscPartitionerDestroy_Shell(PetscPartitioner part)
69977623264SMatthew G. Knepley {
70077623264SMatthew G. Knepley   PetscPartitioner_Shell *p = (PetscPartitioner_Shell *) part->data;
70177623264SMatthew G. Knepley   PetscErrorCode          ierr;
70277623264SMatthew G. Knepley 
70377623264SMatthew G. Knepley   PetscFunctionBegin;
70477623264SMatthew G. Knepley   ierr = PetscSectionDestroy(&p->section);CHKERRQ(ierr);
70577623264SMatthew G. Knepley   ierr = ISDestroy(&p->partition);CHKERRQ(ierr);
70677623264SMatthew G. Knepley   ierr = PetscFree(p);CHKERRQ(ierr);
70777623264SMatthew G. Knepley   PetscFunctionReturn(0);
70877623264SMatthew G. Knepley }
70977623264SMatthew G. Knepley 
71077623264SMatthew G. Knepley #undef __FUNCT__
71177623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Shell_Ascii"
71277623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Shell_Ascii(PetscPartitioner part, PetscViewer viewer)
71377623264SMatthew G. Knepley {
71477623264SMatthew G. Knepley   PetscViewerFormat format;
71577623264SMatthew G. Knepley   PetscErrorCode    ierr;
71677623264SMatthew G. Knepley 
71777623264SMatthew G. Knepley   PetscFunctionBegin;
71877623264SMatthew G. Knepley   ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
71977623264SMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "Shell Graph Partitioner:\n");CHKERRQ(ierr);
72077623264SMatthew G. Knepley   PetscFunctionReturn(0);
72177623264SMatthew G. Knepley }
72277623264SMatthew G. Knepley 
72377623264SMatthew G. Knepley #undef __FUNCT__
72477623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Shell"
72577623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Shell(PetscPartitioner part, PetscViewer viewer)
72677623264SMatthew G. Knepley {
72777623264SMatthew G. Knepley   PetscBool      iascii;
72877623264SMatthew G. Knepley   PetscErrorCode ierr;
72977623264SMatthew G. Knepley 
73077623264SMatthew G. Knepley   PetscFunctionBegin;
73177623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
73277623264SMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
73377623264SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
73477623264SMatthew G. Knepley   if (iascii) {ierr = PetscPartitionerView_Shell_Ascii(part, viewer);CHKERRQ(ierr);}
73577623264SMatthew G. Knepley   PetscFunctionReturn(0);
73677623264SMatthew G. Knepley }
73777623264SMatthew G. Knepley 
73877623264SMatthew G. Knepley #undef __FUNCT__
73977623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerPartition_Shell"
74077623264SMatthew G. Knepley PetscErrorCode PetscPartitionerPartition_Shell(PetscPartitioner part, DM dm, PetscInt nparts, PetscInt numVertices, PetscInt start[], PetscInt adjacency[], PetscSection partSection, IS *partition)
74177623264SMatthew G. Knepley {
74277623264SMatthew G. Knepley   PetscPartitioner_Shell *p = (PetscPartitioner_Shell *) part->data;
74377623264SMatthew G. Knepley   PetscInt                np;
74477623264SMatthew G. Knepley   PetscErrorCode          ierr;
74577623264SMatthew G. Knepley 
74677623264SMatthew G. Knepley   PetscFunctionBegin;
74777623264SMatthew G. Knepley   ierr = PetscSectionGetChart(p->section, NULL, &np);CHKERRQ(ierr);
74877623264SMatthew G. Knepley   if (nparts != np) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of requested partitions %d != configured partitions %d", nparts, np);
74977623264SMatthew G. Knepley   ierr = ISGetLocalSize(p->partition, &np);CHKERRQ(ierr);
75077623264SMatthew G. Knepley   if (numVertices != np) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of input vertices %d != configured vertices %d", numVertices, np);
7515680f57bSMatthew G. Knepley   ierr = PetscSectionCopy(p->section, partSection);CHKERRQ(ierr);
75277623264SMatthew G. Knepley   *partition = p->partition;
75377623264SMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) p->partition);CHKERRQ(ierr);
75477623264SMatthew G. Knepley   PetscFunctionReturn(0);
75577623264SMatthew G. Knepley }
75677623264SMatthew G. Knepley 
75777623264SMatthew G. Knepley #undef __FUNCT__
75877623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerInitialize_Shell"
75977623264SMatthew G. Knepley PetscErrorCode PetscPartitionerInitialize_Shell(PetscPartitioner part)
76077623264SMatthew G. Knepley {
76177623264SMatthew G. Knepley   PetscFunctionBegin;
76277623264SMatthew G. Knepley   part->ops->view      = PetscPartitionerView_Shell;
76377623264SMatthew G. Knepley   part->ops->destroy   = PetscPartitionerDestroy_Shell;
76477623264SMatthew G. Knepley   part->ops->partition = PetscPartitionerPartition_Shell;
76577623264SMatthew G. Knepley   PetscFunctionReturn(0);
76677623264SMatthew G. Knepley }
76777623264SMatthew G. Knepley 
76877623264SMatthew G. Knepley /*MC
76977623264SMatthew G. Knepley   PETSCPARTITIONERSHELL = "shell" - A PetscPartitioner object
77077623264SMatthew G. Knepley 
77177623264SMatthew G. Knepley   Level: intermediate
77277623264SMatthew G. Knepley 
77377623264SMatthew G. Knepley .seealso: PetscPartitionerType, PetscPartitionerCreate(), PetscPartitionerSetType()
77477623264SMatthew G. Knepley M*/
77577623264SMatthew G. Knepley 
77677623264SMatthew G. Knepley #undef __FUNCT__
77777623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerCreate_Shell"
77877623264SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscPartitionerCreate_Shell(PetscPartitioner part)
77977623264SMatthew G. Knepley {
78077623264SMatthew G. Knepley   PetscPartitioner_Shell *p;
78177623264SMatthew G. Knepley   PetscErrorCode          ierr;
78277623264SMatthew G. Knepley 
78377623264SMatthew G. Knepley   PetscFunctionBegin;
78477623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
78577623264SMatthew G. Knepley   ierr       = PetscNewLog(part, &p);CHKERRQ(ierr);
78677623264SMatthew G. Knepley   part->data = p;
78777623264SMatthew G. Knepley 
78877623264SMatthew G. Knepley   ierr = PetscPartitionerInitialize_Shell(part);CHKERRQ(ierr);
78977623264SMatthew G. Knepley   PetscFunctionReturn(0);
79077623264SMatthew G. Knepley }
79177623264SMatthew G. Knepley 
79277623264SMatthew G. Knepley #undef __FUNCT__
7935680f57bSMatthew G. Knepley #define __FUNCT__ "PetscPartitionerShellSetPartition"
7945680f57bSMatthew G. Knepley /*@C
7955680f57bSMatthew G. Knepley   PetscPartitionerShellSetPartition - Set an artifical partition for a mesh
7965680f57bSMatthew G. Knepley 
797b7e49471SLawrence Mitchell   Collective on PART
7985680f57bSMatthew G. Knepley 
7995680f57bSMatthew G. Knepley   Input Parameters:
8005680f57bSMatthew G. Knepley + part     - The PetscPartitioner
801b7e49471SLawrence Mitchell . numProcs - The number of partitions
802b7e49471SLawrence Mitchell . sizes    - array of size numProcs (or NULL) providing the number of points in each partition
803b7e49471SLawrence Mitchell - points   - array of size sum(sizes) (may be NULL iff sizes is NULL) providing the partition each point belongs to
8045680f57bSMatthew G. Knepley 
8055680f57bSMatthew G. Knepley   Level: developer
8065680f57bSMatthew G. Knepley 
807b7e49471SLawrence Mitchell   Notes:
808b7e49471SLawrence Mitchell 
809b7e49471SLawrence Mitchell     It is safe to free the sizes and points arrays after use in this routine.
810b7e49471SLawrence Mitchell 
8115680f57bSMatthew G. Knepley .seealso DMPlexDistribute(), PetscPartitionerCreate()
8125680f57bSMatthew G. Knepley @*/
8135680f57bSMatthew G. Knepley PetscErrorCode PetscPartitionerShellSetPartition(PetscPartitioner part, PetscInt numProcs, const PetscInt sizes[], const PetscInt points[])
8145680f57bSMatthew G. Knepley {
8155680f57bSMatthew G. Knepley   PetscPartitioner_Shell *p = (PetscPartitioner_Shell *) part->data;
8165680f57bSMatthew G. Knepley   PetscInt                proc, numPoints;
8175680f57bSMatthew G. Knepley   PetscErrorCode          ierr;
8185680f57bSMatthew G. Knepley 
8195680f57bSMatthew G. Knepley   PetscFunctionBegin;
8205680f57bSMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
8215680f57bSMatthew G. Knepley   if (sizes) {PetscValidPointer(sizes, 3);}
8225680f57bSMatthew G. Knepley   if (sizes) {PetscValidPointer(points, 4);}
8235680f57bSMatthew G. Knepley   ierr = PetscSectionDestroy(&p->section);CHKERRQ(ierr);
8245680f57bSMatthew G. Knepley   ierr = ISDestroy(&p->partition);CHKERRQ(ierr);
8255680f57bSMatthew G. Knepley   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) part), &p->section);CHKERRQ(ierr);
8265680f57bSMatthew G. Knepley   ierr = PetscSectionSetChart(p->section, 0, numProcs);CHKERRQ(ierr);
8275680f57bSMatthew G. Knepley   if (sizes) {
8285680f57bSMatthew G. Knepley     for (proc = 0; proc < numProcs; ++proc) {
8295680f57bSMatthew G. Knepley       ierr = PetscSectionSetDof(p->section, proc, sizes[proc]);CHKERRQ(ierr);
8305680f57bSMatthew G. Knepley     }
8315680f57bSMatthew G. Knepley   }
8325680f57bSMatthew G. Knepley   ierr = PetscSectionSetUp(p->section);CHKERRQ(ierr);
8335680f57bSMatthew G. Knepley   ierr = PetscSectionGetStorageSize(p->section, &numPoints);CHKERRQ(ierr);
8345680f57bSMatthew G. Knepley   ierr = ISCreateGeneral(PetscObjectComm((PetscObject) part), numPoints, points, PETSC_COPY_VALUES, &p->partition);CHKERRQ(ierr);
8355680f57bSMatthew G. Knepley   PetscFunctionReturn(0);
8365680f57bSMatthew G. Knepley }
8375680f57bSMatthew G. Knepley 
8385680f57bSMatthew G. Knepley #undef __FUNCT__
839*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerDestroy_Simple"
840*555a9cf8SMatthew G. Knepley PetscErrorCode PetscPartitionerDestroy_Simple(PetscPartitioner part)
841*555a9cf8SMatthew G. Knepley {
842*555a9cf8SMatthew G. Knepley   PetscPartitioner_Simple *p = (PetscPartitioner_Simple *) part->data;
843*555a9cf8SMatthew G. Knepley   PetscErrorCode          ierr;
844*555a9cf8SMatthew G. Knepley 
845*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
846*555a9cf8SMatthew G. Knepley   ierr = PetscFree(p);CHKERRQ(ierr);
847*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
848*555a9cf8SMatthew G. Knepley }
849*555a9cf8SMatthew G. Knepley 
850*555a9cf8SMatthew G. Knepley #undef __FUNCT__
851*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Simple_Ascii"
852*555a9cf8SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Simple_Ascii(PetscPartitioner part, PetscViewer viewer)
853*555a9cf8SMatthew G. Knepley {
854*555a9cf8SMatthew G. Knepley   PetscViewerFormat format;
855*555a9cf8SMatthew G. Knepley   PetscErrorCode    ierr;
856*555a9cf8SMatthew G. Knepley 
857*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
858*555a9cf8SMatthew G. Knepley   ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
859*555a9cf8SMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "Simple Graph Partitioner:\n");CHKERRQ(ierr);
860*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
861*555a9cf8SMatthew G. Knepley }
862*555a9cf8SMatthew G. Knepley 
863*555a9cf8SMatthew G. Knepley #undef __FUNCT__
864*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Simple"
865*555a9cf8SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Simple(PetscPartitioner part, PetscViewer viewer)
866*555a9cf8SMatthew G. Knepley {
867*555a9cf8SMatthew G. Knepley   PetscBool      iascii;
868*555a9cf8SMatthew G. Knepley   PetscErrorCode ierr;
869*555a9cf8SMatthew G. Knepley 
870*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
871*555a9cf8SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
872*555a9cf8SMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
873*555a9cf8SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
874*555a9cf8SMatthew G. Knepley   if (iascii) {ierr = PetscPartitionerView_Simple_Ascii(part, viewer);CHKERRQ(ierr);}
875*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
876*555a9cf8SMatthew G. Knepley }
877*555a9cf8SMatthew G. Knepley 
878*555a9cf8SMatthew G. Knepley #undef __FUNCT__
879*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerPartition_Simple"
880*555a9cf8SMatthew G. Knepley PetscErrorCode PetscPartitionerPartition_Simple(PetscPartitioner part, DM dm, PetscInt nparts, PetscInt numVertices, PetscInt start[], PetscInt adjacency[], PetscSection partSection, IS *partition)
881*555a9cf8SMatthew G. Knepley {
882*555a9cf8SMatthew G. Knepley   PetscPartitioner_Simple *p = (PetscPartitioner_Simple *) part->data;
883*555a9cf8SMatthew G. Knepley   PetscInt                 np;
884*555a9cf8SMatthew G. Knepley   PetscErrorCode           ierr;
885*555a9cf8SMatthew G. Knepley 
886*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
887*555a9cf8SMatthew G. Knepley   ierr = PetscSectionSetChart(partSection, 0, nparts);CHKERRQ(ierr);
888*555a9cf8SMatthew G. Knepley   for (np = 0; np < nparts; ++np) {ierr = PetscSectionSetDof(partSection, np, numVertices/nparts + ((numVertices % nparts) > np));CHKERRQ(ierr);}
889*555a9cf8SMatthew G. Knepley   ierr = PetscSectionSetUp(partSection);CHKERRQ(ierr);
890*555a9cf8SMatthew G. Knepley   ierr = ISCreateStride(PETSC_COMM_SELF, numVertices, 0, 1, partition);CHKERRQ(ierr);
891*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
892*555a9cf8SMatthew G. Knepley }
893*555a9cf8SMatthew G. Knepley 
894*555a9cf8SMatthew G. Knepley #undef __FUNCT__
895*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerInitialize_Simple"
896*555a9cf8SMatthew G. Knepley PetscErrorCode PetscPartitionerInitialize_Simple(PetscPartitioner part)
897*555a9cf8SMatthew G. Knepley {
898*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
899*555a9cf8SMatthew G. Knepley   part->ops->view      = PetscPartitionerView_Simple;
900*555a9cf8SMatthew G. Knepley   part->ops->destroy   = PetscPartitionerDestroy_Simple;
901*555a9cf8SMatthew G. Knepley   part->ops->partition = PetscPartitionerPartition_Simple;
902*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
903*555a9cf8SMatthew G. Knepley }
904*555a9cf8SMatthew G. Knepley 
905*555a9cf8SMatthew G. Knepley /*MC
906*555a9cf8SMatthew G. Knepley   PETSCPARTITIONERSIMPLE = "simple" - A PetscPartitioner object
907*555a9cf8SMatthew G. Knepley 
908*555a9cf8SMatthew G. Knepley   Level: intermediate
909*555a9cf8SMatthew G. Knepley 
910*555a9cf8SMatthew G. Knepley .seealso: PetscPartitionerType, PetscPartitionerCreate(), PetscPartitionerSetType()
911*555a9cf8SMatthew G. Knepley M*/
912*555a9cf8SMatthew G. Knepley 
913*555a9cf8SMatthew G. Knepley #undef __FUNCT__
914*555a9cf8SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerCreate_Simple"
915*555a9cf8SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscPartitionerCreate_Simple(PetscPartitioner part)
916*555a9cf8SMatthew G. Knepley {
917*555a9cf8SMatthew G. Knepley   PetscPartitioner_Simple *p;
918*555a9cf8SMatthew G. Knepley   PetscErrorCode           ierr;
919*555a9cf8SMatthew G. Knepley 
920*555a9cf8SMatthew G. Knepley   PetscFunctionBegin;
921*555a9cf8SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
922*555a9cf8SMatthew G. Knepley   ierr       = PetscNewLog(part, &p);CHKERRQ(ierr);
923*555a9cf8SMatthew G. Knepley   part->data = p;
924*555a9cf8SMatthew G. Knepley 
925*555a9cf8SMatthew G. Knepley   ierr = PetscPartitionerInitialize_Simple(part);CHKERRQ(ierr);
926*555a9cf8SMatthew G. Knepley   PetscFunctionReturn(0);
927*555a9cf8SMatthew G. Knepley }
928*555a9cf8SMatthew G. Knepley 
929*555a9cf8SMatthew G. Knepley #undef __FUNCT__
93077623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerDestroy_Chaco"
93177623264SMatthew G. Knepley PetscErrorCode PetscPartitionerDestroy_Chaco(PetscPartitioner part)
93277623264SMatthew G. Knepley {
93377623264SMatthew G. Knepley   PetscPartitioner_Chaco *p = (PetscPartitioner_Chaco *) part->data;
93477623264SMatthew G. Knepley   PetscErrorCode          ierr;
93577623264SMatthew G. Knepley 
93677623264SMatthew G. Knepley   PetscFunctionBegin;
93777623264SMatthew G. Knepley   ierr = PetscFree(p);CHKERRQ(ierr);
93877623264SMatthew G. Knepley   PetscFunctionReturn(0);
93977623264SMatthew G. Knepley }
94077623264SMatthew G. Knepley 
94177623264SMatthew G. Knepley #undef __FUNCT__
94277623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Chaco_Ascii"
94377623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Chaco_Ascii(PetscPartitioner part, PetscViewer viewer)
94477623264SMatthew G. Knepley {
94577623264SMatthew G. Knepley   PetscViewerFormat format;
94677623264SMatthew G. Knepley   PetscErrorCode    ierr;
94777623264SMatthew G. Knepley 
94877623264SMatthew G. Knepley   PetscFunctionBegin;
94977623264SMatthew G. Knepley   ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
95077623264SMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "Chaco Graph Partitioner:\n");CHKERRQ(ierr);
95177623264SMatthew G. Knepley   PetscFunctionReturn(0);
95277623264SMatthew G. Knepley }
95377623264SMatthew G. Knepley 
95477623264SMatthew G. Knepley #undef __FUNCT__
95577623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_Chaco"
95677623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_Chaco(PetscPartitioner part, PetscViewer viewer)
95777623264SMatthew G. Knepley {
95877623264SMatthew G. Knepley   PetscBool      iascii;
95977623264SMatthew G. Knepley   PetscErrorCode ierr;
96077623264SMatthew G. Knepley 
96177623264SMatthew G. Knepley   PetscFunctionBegin;
96277623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
96377623264SMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
96477623264SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
96577623264SMatthew G. Knepley   if (iascii) {ierr = PetscPartitionerView_Chaco_Ascii(part, viewer);CHKERRQ(ierr);}
96677623264SMatthew G. Knepley   PetscFunctionReturn(0);
96777623264SMatthew G. Knepley }
96877623264SMatthew G. Knepley 
96970034214SMatthew G. Knepley #if defined(PETSC_HAVE_CHACO)
97070034214SMatthew G. Knepley #if defined(PETSC_HAVE_UNISTD_H)
97170034214SMatthew G. Knepley #include <unistd.h>
97270034214SMatthew G. Knepley #endif
97370034214SMatthew G. Knepley /* Chaco does not have an include file */
97470034214SMatthew G. Knepley PETSC_EXTERN int interface(int nvtxs, int *start, int *adjacency, int *vwgts,
97570034214SMatthew G. Knepley                        float *ewgts, float *x, float *y, float *z, char *outassignname,
97670034214SMatthew G. Knepley                        char *outfilename, short *assignment, int architecture, int ndims_tot,
97770034214SMatthew G. Knepley                        int mesh_dims[3], double *goal, int global_method, int local_method,
97870034214SMatthew G. Knepley                        int rqi_flag, int vmax, int ndims, double eigtol, long seed);
97970034214SMatthew G. Knepley 
98070034214SMatthew G. Knepley extern int FREE_GRAPH;
98177623264SMatthew G. Knepley #endif
98270034214SMatthew G. Knepley 
98370034214SMatthew G. Knepley #undef __FUNCT__
98477623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerPartition_Chaco"
98577623264SMatthew G. Knepley PetscErrorCode PetscPartitionerPartition_Chaco(PetscPartitioner part, DM dm, PetscInt nparts, PetscInt numVertices, PetscInt start[], PetscInt adjacency[], PetscSection partSection, IS *partition)
98670034214SMatthew G. Knepley {
98777623264SMatthew G. Knepley #if defined(PETSC_HAVE_CHACO)
98870034214SMatthew G. Knepley   enum {DEFAULT_METHOD = 1, INERTIAL_METHOD = 3};
98970034214SMatthew G. Knepley   MPI_Comm       comm;
99070034214SMatthew G. Knepley   int            nvtxs          = numVertices; /* number of vertices in full graph */
99170034214SMatthew G. Knepley   int           *vwgts          = NULL;   /* weights for all vertices */
99270034214SMatthew G. Knepley   float         *ewgts          = NULL;   /* weights for all edges */
99370034214SMatthew G. Knepley   float         *x              = NULL, *y = NULL, *z = NULL; /* coordinates for inertial method */
99470034214SMatthew G. Knepley   char          *outassignname  = NULL;   /*  name of assignment output file */
99570034214SMatthew G. Knepley   char          *outfilename    = NULL;   /* output file name */
99670034214SMatthew G. Knepley   int            architecture   = 1;      /* 0 => hypercube, d => d-dimensional mesh */
99770034214SMatthew G. Knepley   int            ndims_tot      = 0;      /* total number of cube dimensions to divide */
99870034214SMatthew G. Knepley   int            mesh_dims[3];            /* dimensions of mesh of processors */
99970034214SMatthew G. Knepley   double        *goal          = NULL;    /* desired set sizes for each set */
100070034214SMatthew G. Knepley   int            global_method = 1;       /* global partitioning algorithm */
100170034214SMatthew G. Knepley   int            local_method  = 1;       /* local partitioning algorithm */
100270034214SMatthew G. Knepley   int            rqi_flag      = 0;       /* should I use RQI/Symmlq eigensolver? */
100370034214SMatthew G. Knepley   int            vmax          = 200;     /* how many vertices to coarsen down to? */
100470034214SMatthew G. Knepley   int            ndims         = 1;       /* number of eigenvectors (2^d sets) */
100570034214SMatthew G. Knepley   double         eigtol        = 0.001;   /* tolerance on eigenvectors */
100670034214SMatthew G. Knepley   long           seed          = 123636512; /* for random graph mutations */
100770034214SMatthew G. Knepley   short int     *assignment;              /* Output partition */
100870034214SMatthew G. Knepley   int            fd_stdout, fd_pipe[2];
100970034214SMatthew G. Knepley   PetscInt      *points;
101070034214SMatthew G. Knepley   int            i, v, p;
101170034214SMatthew G. Knepley   PetscErrorCode ierr;
101270034214SMatthew G. Knepley 
101370034214SMatthew G. Knepley   PetscFunctionBegin;
101470034214SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
101570034214SMatthew G. Knepley   if (!numVertices) {
101677623264SMatthew G. Knepley     ierr = PetscSectionSetChart(partSection, 0, nparts);CHKERRQ(ierr);
101777623264SMatthew G. Knepley     ierr = PetscSectionSetUp(partSection);CHKERRQ(ierr);
101870034214SMatthew G. Knepley     ierr = ISCreateGeneral(comm, 0, NULL, PETSC_OWN_POINTER, partition);CHKERRQ(ierr);
101970034214SMatthew G. Knepley     PetscFunctionReturn(0);
102070034214SMatthew G. Knepley   }
102170034214SMatthew G. Knepley   FREE_GRAPH = 0;                         /* Do not let Chaco free my memory */
102270034214SMatthew G. Knepley   for (i = 0; i < start[numVertices]; ++i) ++adjacency[i];
102370034214SMatthew G. Knepley 
102470034214SMatthew G. Knepley   if (global_method == INERTIAL_METHOD) {
102570034214SMatthew G. Knepley     /* manager.createCellCoordinates(nvtxs, &x, &y, &z); */
102670034214SMatthew G. Knepley     SETERRQ(comm, PETSC_ERR_SUP, "Inertial partitioning not yet supported");
102770034214SMatthew G. Knepley   }
102877623264SMatthew G. Knepley   mesh_dims[0] = nparts;
102970034214SMatthew G. Knepley   mesh_dims[1] = 1;
103070034214SMatthew G. Knepley   mesh_dims[2] = 1;
103170034214SMatthew G. Knepley   ierr = PetscMalloc1(nvtxs, &assignment);CHKERRQ(ierr);
103270034214SMatthew G. Knepley   /* Chaco outputs to stdout. We redirect this to a buffer. */
103370034214SMatthew G. Knepley   /* TODO: check error codes for UNIX calls */
103470034214SMatthew G. Knepley #if defined(PETSC_HAVE_UNISTD_H)
103570034214SMatthew G. Knepley   {
103670034214SMatthew G. Knepley     int piperet;
103770034214SMatthew G. Knepley     piperet = pipe(fd_pipe);
103870034214SMatthew G. Knepley     if (piperet) SETERRQ(comm,PETSC_ERR_SYS,"Could not create pipe");
103970034214SMatthew G. Knepley     fd_stdout = dup(1);
104070034214SMatthew G. Knepley     close(1);
104170034214SMatthew G. Knepley     dup2(fd_pipe[1], 1);
104270034214SMatthew G. Knepley   }
104370034214SMatthew G. Knepley #endif
104470034214SMatthew G. Knepley   ierr = interface(nvtxs, (int*) start, (int*) adjacency, vwgts, ewgts, x, y, z, outassignname, outfilename,
104570034214SMatthew G. Knepley                    assignment, architecture, ndims_tot, mesh_dims, goal, global_method, local_method, rqi_flag,
104670034214SMatthew G. Knepley                    vmax, ndims, eigtol, seed);
104770034214SMatthew G. Knepley #if defined(PETSC_HAVE_UNISTD_H)
104870034214SMatthew G. Knepley   {
104970034214SMatthew G. Knepley     char msgLog[10000];
105070034214SMatthew G. Knepley     int  count;
105170034214SMatthew G. Knepley 
105270034214SMatthew G. Knepley     fflush(stdout);
105370034214SMatthew G. Knepley     count = read(fd_pipe[0], msgLog, (10000-1)*sizeof(char));
105470034214SMatthew G. Knepley     if (count < 0) count = 0;
105570034214SMatthew G. Knepley     msgLog[count] = 0;
105670034214SMatthew G. Knepley     close(1);
105770034214SMatthew G. Knepley     dup2(fd_stdout, 1);
105870034214SMatthew G. Knepley     close(fd_stdout);
105970034214SMatthew G. Knepley     close(fd_pipe[0]);
106070034214SMatthew G. Knepley     close(fd_pipe[1]);
106170034214SMatthew G. Knepley     if (ierr) SETERRQ1(comm, PETSC_ERR_LIB, "Error in Chaco library: %s", msgLog);
106270034214SMatthew G. Knepley   }
106370034214SMatthew G. Knepley #endif
106470034214SMatthew G. Knepley   /* Convert to PetscSection+IS */
106577623264SMatthew G. Knepley   ierr = PetscSectionSetChart(partSection, 0, nparts);CHKERRQ(ierr);
106670034214SMatthew G. Knepley   for (v = 0; v < nvtxs; ++v) {
106777623264SMatthew G. Knepley     ierr = PetscSectionAddDof(partSection, assignment[v], 1);CHKERRQ(ierr);
106870034214SMatthew G. Knepley   }
106977623264SMatthew G. Knepley   ierr = PetscSectionSetUp(partSection);CHKERRQ(ierr);
107070034214SMatthew G. Knepley   ierr = PetscMalloc1(nvtxs, &points);CHKERRQ(ierr);
107177623264SMatthew G. Knepley   for (p = 0, i = 0; p < nparts; ++p) {
107270034214SMatthew G. Knepley     for (v = 0; v < nvtxs; ++v) {
107370034214SMatthew G. Knepley       if (assignment[v] == p) points[i++] = v;
107470034214SMatthew G. Knepley     }
107570034214SMatthew G. Knepley   }
107670034214SMatthew G. Knepley   if (i != nvtxs) SETERRQ2(comm, PETSC_ERR_PLIB, "Number of points %D should be %D", i, nvtxs);
107770034214SMatthew G. Knepley   ierr = ISCreateGeneral(comm, nvtxs, points, PETSC_OWN_POINTER, partition);CHKERRQ(ierr);
107870034214SMatthew G. Knepley   if (global_method == INERTIAL_METHOD) {
107970034214SMatthew G. Knepley     /* manager.destroyCellCoordinates(nvtxs, &x, &y, &z); */
108070034214SMatthew G. Knepley   }
108170034214SMatthew G. Knepley   ierr = PetscFree(assignment);CHKERRQ(ierr);
108270034214SMatthew G. Knepley   for (i = 0; i < start[numVertices]; ++i) --adjacency[i];
108370034214SMatthew G. Knepley   PetscFunctionReturn(0);
108477623264SMatthew G. Knepley #else
108577623264SMatthew G. Knepley   SETERRQ(PetscObjectComm((PetscObject) part), PETSC_ERR_SUP, "Mesh partitioning needs external package support.\nPlease reconfigure with --download-chaco.");
108670034214SMatthew G. Knepley #endif
108777623264SMatthew G. Knepley }
108877623264SMatthew G. Knepley 
108977623264SMatthew G. Knepley #undef __FUNCT__
109077623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerInitialize_Chaco"
109177623264SMatthew G. Knepley PetscErrorCode PetscPartitionerInitialize_Chaco(PetscPartitioner part)
109277623264SMatthew G. Knepley {
109377623264SMatthew G. Knepley   PetscFunctionBegin;
109477623264SMatthew G. Knepley   part->ops->view      = PetscPartitionerView_Chaco;
109577623264SMatthew G. Knepley   part->ops->destroy   = PetscPartitionerDestroy_Chaco;
109677623264SMatthew G. Knepley   part->ops->partition = PetscPartitionerPartition_Chaco;
109777623264SMatthew G. Knepley   PetscFunctionReturn(0);
109877623264SMatthew G. Knepley }
109977623264SMatthew G. Knepley 
110077623264SMatthew G. Knepley /*MC
110177623264SMatthew G. Knepley   PETSCPARTITIONERCHACO = "chaco" - A PetscPartitioner object using the Chaco library
110277623264SMatthew G. Knepley 
110377623264SMatthew G. Knepley   Level: intermediate
110477623264SMatthew G. Knepley 
110577623264SMatthew G. Knepley .seealso: PetscPartitionerType, PetscPartitionerCreate(), PetscPartitionerSetType()
110677623264SMatthew G. Knepley M*/
110777623264SMatthew G. Knepley 
110877623264SMatthew G. Knepley #undef __FUNCT__
110977623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerCreate_Chaco"
111077623264SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscPartitionerCreate_Chaco(PetscPartitioner part)
111177623264SMatthew G. Knepley {
111277623264SMatthew G. Knepley   PetscPartitioner_Chaco *p;
111377623264SMatthew G. Knepley   PetscErrorCode          ierr;
111477623264SMatthew G. Knepley 
111577623264SMatthew G. Knepley   PetscFunctionBegin;
111677623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
111777623264SMatthew G. Knepley   ierr       = PetscNewLog(part, &p);CHKERRQ(ierr);
111877623264SMatthew G. Knepley   part->data = p;
111977623264SMatthew G. Knepley 
112077623264SMatthew G. Knepley   ierr = PetscPartitionerInitialize_Chaco(part);CHKERRQ(ierr);
112177623264SMatthew G. Knepley   ierr = PetscCitationsRegister(ChacoPartitionerCitation, &ChacoPartitionercite);CHKERRQ(ierr);
112277623264SMatthew G. Knepley   PetscFunctionReturn(0);
112377623264SMatthew G. Knepley }
112477623264SMatthew G. Knepley 
112577623264SMatthew G. Knepley #undef __FUNCT__
112677623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerDestroy_ParMetis"
112777623264SMatthew G. Knepley PetscErrorCode PetscPartitionerDestroy_ParMetis(PetscPartitioner part)
112877623264SMatthew G. Knepley {
112977623264SMatthew G. Knepley   PetscPartitioner_ParMetis *p = (PetscPartitioner_ParMetis *) part->data;
113077623264SMatthew G. Knepley   PetscErrorCode             ierr;
113177623264SMatthew G. Knepley 
113277623264SMatthew G. Knepley   PetscFunctionBegin;
113377623264SMatthew G. Knepley   ierr = PetscFree(p);CHKERRQ(ierr);
113477623264SMatthew G. Knepley   PetscFunctionReturn(0);
113577623264SMatthew G. Knepley }
113677623264SMatthew G. Knepley 
113777623264SMatthew G. Knepley #undef __FUNCT__
113877623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_ParMetis_Ascii"
113977623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_ParMetis_Ascii(PetscPartitioner part, PetscViewer viewer)
114077623264SMatthew G. Knepley {
114177623264SMatthew G. Knepley   PetscViewerFormat format;
114277623264SMatthew G. Knepley   PetscErrorCode    ierr;
114377623264SMatthew G. Knepley 
114477623264SMatthew G. Knepley   PetscFunctionBegin;
114577623264SMatthew G. Knepley   ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
114677623264SMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "ParMetis Graph Partitioner:\n");CHKERRQ(ierr);
114777623264SMatthew G. Knepley   PetscFunctionReturn(0);
114877623264SMatthew G. Knepley }
114977623264SMatthew G. Knepley 
115077623264SMatthew G. Knepley #undef __FUNCT__
115177623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerView_ParMetis"
115277623264SMatthew G. Knepley PetscErrorCode PetscPartitionerView_ParMetis(PetscPartitioner part, PetscViewer viewer)
115377623264SMatthew G. Knepley {
115477623264SMatthew G. Knepley   PetscBool      iascii;
115577623264SMatthew G. Knepley   PetscErrorCode ierr;
115677623264SMatthew G. Knepley 
115777623264SMatthew G. Knepley   PetscFunctionBegin;
115877623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
115977623264SMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
116077623264SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
116177623264SMatthew G. Knepley   if (iascii) {ierr = PetscPartitionerView_ParMetis_Ascii(part, viewer);CHKERRQ(ierr);}
116277623264SMatthew G. Knepley   PetscFunctionReturn(0);
116377623264SMatthew G. Knepley }
116470034214SMatthew G. Knepley 
116570034214SMatthew G. Knepley #if defined(PETSC_HAVE_PARMETIS)
116670034214SMatthew G. Knepley #include <parmetis.h>
116777623264SMatthew G. Knepley #endif
116870034214SMatthew G. Knepley 
116970034214SMatthew G. Knepley #undef __FUNCT__
117077623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerPartition_ParMetis"
117177623264SMatthew G. Knepley PetscErrorCode PetscPartitionerPartition_ParMetis(PetscPartitioner part, DM dm, PetscInt nparts, PetscInt numVertices, PetscInt start[], PetscInt adjacency[], PetscSection partSection, IS *partition)
117270034214SMatthew G. Knepley {
117377623264SMatthew G. Knepley #if defined(PETSC_HAVE_PARMETIS)
117470034214SMatthew G. Knepley   MPI_Comm       comm;
117570034214SMatthew G. Knepley   PetscInt       nvtxs      = numVertices; /* The number of vertices in full graph */
117670034214SMatthew G. Knepley   PetscInt      *vtxdist;                  /* Distribution of vertices across processes */
117770034214SMatthew G. Knepley   PetscInt      *xadj       = start;       /* Start of edge list for each vertex */
117870034214SMatthew G. Knepley   PetscInt      *adjncy     = adjacency;   /* Edge lists for all vertices */
117970034214SMatthew G. Knepley   PetscInt      *vwgt       = NULL;        /* Vertex weights */
118070034214SMatthew G. Knepley   PetscInt      *adjwgt     = NULL;        /* Edge weights */
118170034214SMatthew G. Knepley   PetscInt       wgtflag    = 0;           /* Indicates which weights are present */
118270034214SMatthew G. Knepley   PetscInt       numflag    = 0;           /* Indicates initial offset (0 or 1) */
118370034214SMatthew G. Knepley   PetscInt       ncon       = 1;           /* The number of weights per vertex */
118470034214SMatthew G. Knepley   PetscReal     *tpwgts;                   /* The fraction of vertex weights assigned to each partition */
118570034214SMatthew G. Knepley   PetscReal     *ubvec;                    /* The balance intolerance for vertex weights */
118670034214SMatthew G. Knepley   PetscInt       options[5];               /* Options */
118770034214SMatthew G. Knepley   /* Outputs */
118870034214SMatthew G. Knepley   PetscInt       edgeCut;                  /* The number of edges cut by the partition */
118970034214SMatthew G. Knepley   PetscInt      *assignment, *points;
119077623264SMatthew G. Knepley   PetscMPIInt    rank, p, v, i;
119170034214SMatthew G. Knepley   PetscErrorCode ierr;
119270034214SMatthew G. Knepley 
119370034214SMatthew G. Knepley   PetscFunctionBegin;
119477623264SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) part, &comm);CHKERRQ(ierr);
119570034214SMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
119670034214SMatthew G. Knepley   options[0] = 0; /* Use all defaults */
119770034214SMatthew G. Knepley   /* Calculate vertex distribution */
119870034214SMatthew G. Knepley   ierr = PetscMalloc4(nparts+1,&vtxdist,nparts*ncon,&tpwgts,ncon,&ubvec,nvtxs,&assignment);CHKERRQ(ierr);
119970034214SMatthew G. Knepley   vtxdist[0] = 0;
120070034214SMatthew G. Knepley   ierr = MPI_Allgather(&nvtxs, 1, MPIU_INT, &vtxdist[1], 1, MPIU_INT, comm);CHKERRQ(ierr);
120170034214SMatthew G. Knepley   for (p = 2; p <= nparts; ++p) {
120270034214SMatthew G. Knepley     vtxdist[p] += vtxdist[p-1];
120370034214SMatthew G. Knepley   }
120470034214SMatthew G. Knepley   /* Calculate weights */
120570034214SMatthew G. Knepley   for (p = 0; p < nparts; ++p) {
120670034214SMatthew G. Knepley     tpwgts[p] = 1.0/nparts;
120770034214SMatthew G. Knepley   }
120870034214SMatthew G. Knepley   ubvec[0] = 1.05;
120970034214SMatthew G. Knepley 
121070034214SMatthew G. Knepley   if (nparts == 1) {
121170034214SMatthew G. Knepley     ierr = PetscMemzero(assignment, nvtxs * sizeof(PetscInt));
121270034214SMatthew G. Knepley   } else {
121370034214SMatthew G. Knepley     if (vtxdist[1] == vtxdist[nparts]) {
121470034214SMatthew G. Knepley       if (!rank) {
121570034214SMatthew G. Knepley         PetscStackPush("METIS_PartGraphKway");
121670034214SMatthew G. Knepley         ierr = METIS_PartGraphKway(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, &nparts, tpwgts, ubvec, NULL, &edgeCut, assignment);
121770034214SMatthew G. Knepley         PetscStackPop;
121870034214SMatthew G. Knepley         if (ierr != METIS_OK) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "Error in METIS_PartGraphKway()");
121970034214SMatthew G. Knepley       }
122070034214SMatthew G. Knepley     } else {
122170034214SMatthew G. Knepley       PetscStackPush("ParMETIS_V3_PartKway");
122270034214SMatthew G. Knepley       ierr = ParMETIS_V3_PartKway(vtxdist, xadj, adjncy, vwgt, adjwgt, &wgtflag, &numflag, &ncon, &nparts, tpwgts, ubvec, options, &edgeCut, assignment, &comm);
122370034214SMatthew G. Knepley       PetscStackPop;
122470034214SMatthew G. Knepley       if (ierr != METIS_OK) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "Error in ParMETIS_V3_PartKway()");
122570034214SMatthew G. Knepley     }
122670034214SMatthew G. Knepley   }
122770034214SMatthew G. Knepley   /* Convert to PetscSection+IS */
122877623264SMatthew G. Knepley   ierr = PetscSectionSetChart(partSection, 0, nparts);CHKERRQ(ierr);
122977623264SMatthew G. Knepley   for (v = 0; v < nvtxs; ++v) {ierr = PetscSectionAddDof(partSection, assignment[v], 1);CHKERRQ(ierr);}
123077623264SMatthew G. Knepley   ierr = PetscSectionSetUp(partSection);CHKERRQ(ierr);
123170034214SMatthew G. Knepley   ierr = PetscMalloc1(nvtxs, &points);CHKERRQ(ierr);
123277623264SMatthew G. Knepley   for (p = 0, i = 0; p < nparts; ++p) {
123370034214SMatthew G. Knepley     for (v = 0; v < nvtxs; ++v) {
123470034214SMatthew G. Knepley       if (assignment[v] == p) points[i++] = v;
123570034214SMatthew G. Knepley     }
123670034214SMatthew G. Knepley   }
123770034214SMatthew G. Knepley   if (i != nvtxs) SETERRQ2(comm, PETSC_ERR_PLIB, "Number of points %D should be %D", i, nvtxs);
123870034214SMatthew G. Knepley   ierr = ISCreateGeneral(comm, nvtxs, points, PETSC_OWN_POINTER, partition);CHKERRQ(ierr);
123970034214SMatthew G. Knepley   ierr = PetscFree4(vtxdist,tpwgts,ubvec,assignment);CHKERRQ(ierr);
124070034214SMatthew G. Knepley #else
124177623264SMatthew G. Knepley   SETERRQ(PetscObjectComm((PetscObject) part), PETSC_ERR_SUP, "Mesh partitioning needs external package support.\nPlease reconfigure with --download-parmetis.");
124270034214SMatthew G. Knepley #endif
124377623264SMatthew G. Knepley   PetscFunctionReturn(0);
124470034214SMatthew G. Knepley }
124570034214SMatthew G. Knepley 
124677623264SMatthew G. Knepley #undef __FUNCT__
124777623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerInitialize_ParMetis"
124877623264SMatthew G. Knepley PetscErrorCode PetscPartitionerInitialize_ParMetis(PetscPartitioner part)
124977623264SMatthew G. Knepley {
125077623264SMatthew G. Knepley   PetscFunctionBegin;
125177623264SMatthew G. Knepley   part->ops->view      = PetscPartitionerView_ParMetis;
125277623264SMatthew G. Knepley   part->ops->destroy   = PetscPartitionerDestroy_ParMetis;
125377623264SMatthew G. Knepley   part->ops->partition = PetscPartitionerPartition_ParMetis;
125477623264SMatthew G. Knepley   PetscFunctionReturn(0);
125577623264SMatthew G. Knepley }
125677623264SMatthew G. Knepley 
125777623264SMatthew G. Knepley /*MC
125877623264SMatthew G. Knepley   PETSCPARTITIONERPARMETIS = "parmetis" - A PetscPartitioner object using the ParMetis library
125977623264SMatthew G. Knepley 
126077623264SMatthew G. Knepley   Level: intermediate
126177623264SMatthew G. Knepley 
126277623264SMatthew G. Knepley .seealso: PetscPartitionerType, PetscPartitionerCreate(), PetscPartitionerSetType()
126377623264SMatthew G. Knepley M*/
126477623264SMatthew G. Knepley 
126577623264SMatthew G. Knepley #undef __FUNCT__
126677623264SMatthew G. Knepley #define __FUNCT__ "PetscPartitionerCreate_ParMetis"
126777623264SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscPartitionerCreate_ParMetis(PetscPartitioner part)
126877623264SMatthew G. Knepley {
126977623264SMatthew G. Knepley   PetscPartitioner_ParMetis *p;
127077623264SMatthew G. Knepley   PetscErrorCode          ierr;
127177623264SMatthew G. Knepley 
127277623264SMatthew G. Knepley   PetscFunctionBegin;
127377623264SMatthew G. Knepley   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 1);
127477623264SMatthew G. Knepley   ierr       = PetscNewLog(part, &p);CHKERRQ(ierr);
127577623264SMatthew G. Knepley   part->data = p;
127677623264SMatthew G. Knepley 
127777623264SMatthew G. Knepley   ierr = PetscPartitionerInitialize_ParMetis(part);CHKERRQ(ierr);
127877623264SMatthew G. Knepley   ierr = PetscCitationsRegister(ParMetisPartitionerCitation, &ParMetisPartitionercite);CHKERRQ(ierr);
127970034214SMatthew G. Knepley   PetscFunctionReturn(0);
128070034214SMatthew G. Knepley }
128170034214SMatthew G. Knepley 
128270034214SMatthew G. Knepley #undef __FUNCT__
12835680f57bSMatthew G. Knepley #define __FUNCT__ "DMPlexGetPartitioner"
12845680f57bSMatthew G. Knepley /*@
12855680f57bSMatthew G. Knepley   DMPlexGetPartitioner - Get the mesh partitioner
12865680f57bSMatthew G. Knepley 
12875680f57bSMatthew G. Knepley   Not collective
12885680f57bSMatthew G. Knepley 
12895680f57bSMatthew G. Knepley   Input Parameter:
12905680f57bSMatthew G. Knepley . dm - The DM
12915680f57bSMatthew G. Knepley 
12925680f57bSMatthew G. Knepley   Output Parameter:
12935680f57bSMatthew G. Knepley . part - The PetscPartitioner
12945680f57bSMatthew G. Knepley 
12955680f57bSMatthew G. Knepley   Level: developer
12965680f57bSMatthew G. Knepley 
129798599a47SLawrence Mitchell   Note: This gets a borrowed reference, so the user should not destroy this PetscPartitioner.
129898599a47SLawrence Mitchell 
129998599a47SLawrence Mitchell .seealso DMPlexDistribute(), DMPlexSetPartitioner(), PetscPartitionerCreate()
13005680f57bSMatthew G. Knepley @*/
13015680f57bSMatthew G. Knepley PetscErrorCode DMPlexGetPartitioner(DM dm, PetscPartitioner *part)
13025680f57bSMatthew G. Knepley {
13035680f57bSMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *) dm->data;
13045680f57bSMatthew G. Knepley 
13055680f57bSMatthew G. Knepley   PetscFunctionBegin;
13065680f57bSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
13075680f57bSMatthew G. Knepley   PetscValidPointer(part, 2);
13085680f57bSMatthew G. Knepley   *part = mesh->partitioner;
13095680f57bSMatthew G. Knepley   PetscFunctionReturn(0);
13105680f57bSMatthew G. Knepley }
13115680f57bSMatthew G. Knepley 
13125680f57bSMatthew G. Knepley #undef __FUNCT__
131371bb2955SLawrence Mitchell #define __FUNCT__ "DMPlexSetPartitioner"
131471bb2955SLawrence Mitchell /*@
131571bb2955SLawrence Mitchell   DMPlexSetPartitioner - Set the mesh partitioner
131671bb2955SLawrence Mitchell 
131771bb2955SLawrence Mitchell   logically collective on dm and part
131871bb2955SLawrence Mitchell 
131971bb2955SLawrence Mitchell   Input Parameters:
132071bb2955SLawrence Mitchell + dm - The DM
132171bb2955SLawrence Mitchell - part - The partitioner
132271bb2955SLawrence Mitchell 
132371bb2955SLawrence Mitchell   Level: developer
132471bb2955SLawrence Mitchell 
132571bb2955SLawrence Mitchell   Note: Any existing PetscPartitioner will be destroyed.
132671bb2955SLawrence Mitchell 
132771bb2955SLawrence Mitchell .seealso DMPlexDistribute(), DMPlexGetPartitioner(), PetscPartitionerCreate()
132871bb2955SLawrence Mitchell @*/
132971bb2955SLawrence Mitchell PetscErrorCode DMPlexSetPartitioner(DM dm, PetscPartitioner part)
133071bb2955SLawrence Mitchell {
133171bb2955SLawrence Mitchell   DM_Plex       *mesh = (DM_Plex *) dm->data;
133271bb2955SLawrence Mitchell   PetscErrorCode ierr;
133371bb2955SLawrence Mitchell 
133471bb2955SLawrence Mitchell   PetscFunctionBegin;
133571bb2955SLawrence Mitchell   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
133671bb2955SLawrence Mitchell   PetscValidHeaderSpecific(part, PETSCPARTITIONER_CLASSID, 2);
133771bb2955SLawrence Mitchell   ierr = PetscObjectReference((PetscObject)part);CHKERRQ(ierr);
133871bb2955SLawrence Mitchell   ierr = PetscPartitionerDestroy(&mesh->partitioner);CHKERRQ(ierr);
133971bb2955SLawrence Mitchell   mesh->partitioner = part;
134071bb2955SLawrence Mitchell   PetscFunctionReturn(0);
134171bb2955SLawrence Mitchell }
134271bb2955SLawrence Mitchell 
134371bb2955SLawrence Mitchell #undef __FUNCT__
13445abbe4feSMichael Lange #define __FUNCT__ "DMPlexPartitionLabelClosure"
13455abbe4feSMichael Lange /*@
13465abbe4feSMichael Lange   DMPlexPartitionLabelClosure - Add the closure of all points to the partition label
13475abbe4feSMichael Lange 
13485abbe4feSMichael Lange   Input Parameters:
13495abbe4feSMichael Lange + dm     - The DM
13505abbe4feSMichael Lange - label  - DMLabel assinging ranks to remote roots
13515abbe4feSMichael Lange 
13525abbe4feSMichael Lange   Level: developer
13535abbe4feSMichael Lange 
13545abbe4feSMichael Lange .seealso: DMPlexPartitionLabelCreateSF, DMPlexDistribute(), DMPlexCreateOverlap
13555abbe4feSMichael Lange @*/
13565abbe4feSMichael Lange PetscErrorCode DMPlexPartitionLabelClosure(DM dm, DMLabel label)
13579ffc88e4SToby Isaac {
13585abbe4feSMichael Lange   IS              rankIS,   pointIS;
13595abbe4feSMichael Lange   const PetscInt *ranks,   *points;
13605abbe4feSMichael Lange   PetscInt        numRanks, numPoints, r, p, c, closureSize;
13615abbe4feSMichael Lange   PetscInt       *closure = NULL;
13629ffc88e4SToby Isaac   PetscErrorCode  ierr;
13639ffc88e4SToby Isaac 
13649ffc88e4SToby Isaac   PetscFunctionBegin;
13655abbe4feSMichael Lange   ierr = DMLabelGetValueIS(label, &rankIS);CHKERRQ(ierr);
13665abbe4feSMichael Lange   ierr = ISGetLocalSize(rankIS, &numRanks);CHKERRQ(ierr);
13675abbe4feSMichael Lange   ierr = ISGetIndices(rankIS, &ranks);CHKERRQ(ierr);
13685abbe4feSMichael Lange   for (r = 0; r < numRanks; ++r) {
13695abbe4feSMichael Lange     const PetscInt rank = ranks[r];
13709ffc88e4SToby Isaac 
13715abbe4feSMichael Lange     ierr = DMLabelGetStratumIS(label, rank, &pointIS);CHKERRQ(ierr);
13725abbe4feSMichael Lange     ierr = ISGetLocalSize(pointIS, &numPoints);CHKERRQ(ierr);
13735abbe4feSMichael Lange     ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr);
13745abbe4feSMichael Lange     for (p = 0; p < numPoints; ++p) {
13755abbe4feSMichael Lange       ierr = DMPlexGetTransitiveClosure(dm, points[p], PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
13765abbe4feSMichael Lange       for (c = 0; c < closureSize*2; c += 2) {ierr = DMLabelSetValue(label, closure[c], rank);CHKERRQ(ierr);}
13779ffc88e4SToby Isaac     }
13785abbe4feSMichael Lange     ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr);
13795abbe4feSMichael Lange     ierr = ISDestroy(&pointIS);CHKERRQ(ierr);
13809ffc88e4SToby Isaac   }
13817de78196SMichael Lange   if (closure) {ierr = DMPlexRestoreTransitiveClosure(dm, 0, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);}
13825abbe4feSMichael Lange   ierr = ISRestoreIndices(rankIS, &ranks);CHKERRQ(ierr);
13835abbe4feSMichael Lange   ierr = ISDestroy(&rankIS);CHKERRQ(ierr);
13849ffc88e4SToby Isaac   PetscFunctionReturn(0);
13859ffc88e4SToby Isaac }
13869ffc88e4SToby Isaac 
13879ffc88e4SToby Isaac #undef __FUNCT__
138824d039d7SMichael Lange #define __FUNCT__ "DMPlexPartitionLabelAdjacency"
138924d039d7SMichael Lange /*@
139024d039d7SMichael Lange   DMPlexPartitionLabelAdjacency - Add one level of adjacent points to the partition label
139124d039d7SMichael Lange 
139224d039d7SMichael Lange   Input Parameters:
139324d039d7SMichael Lange + dm     - The DM
139424d039d7SMichael Lange - label  - DMLabel assinging ranks to remote roots
139524d039d7SMichael Lange 
139624d039d7SMichael Lange   Level: developer
139724d039d7SMichael Lange 
139824d039d7SMichael Lange .seealso: DMPlexPartitionLabelCreateSF, DMPlexDistribute(), DMPlexCreateOverlap
139924d039d7SMichael Lange @*/
140024d039d7SMichael Lange PetscErrorCode DMPlexPartitionLabelAdjacency(DM dm, DMLabel label)
140170034214SMatthew G. Knepley {
140224d039d7SMichael Lange   IS              rankIS,   pointIS;
140324d039d7SMichael Lange   const PetscInt *ranks,   *points;
140424d039d7SMichael Lange   PetscInt        numRanks, numPoints, r, p, a, adjSize;
140524d039d7SMichael Lange   PetscInt       *adj = NULL;
140670034214SMatthew G. Knepley   PetscErrorCode  ierr;
140770034214SMatthew G. Knepley 
140870034214SMatthew G. Knepley   PetscFunctionBegin;
140924d039d7SMichael Lange   ierr = DMLabelGetValueIS(label, &rankIS);CHKERRQ(ierr);
141024d039d7SMichael Lange   ierr = ISGetLocalSize(rankIS, &numRanks);CHKERRQ(ierr);
141124d039d7SMichael Lange   ierr = ISGetIndices(rankIS, &ranks);CHKERRQ(ierr);
141224d039d7SMichael Lange   for (r = 0; r < numRanks; ++r) {
141324d039d7SMichael Lange     const PetscInt rank = ranks[r];
141470034214SMatthew G. Knepley 
141524d039d7SMichael Lange     ierr = DMLabelGetStratumIS(label, rank, &pointIS);CHKERRQ(ierr);
141624d039d7SMichael Lange     ierr = ISGetLocalSize(pointIS, &numPoints);CHKERRQ(ierr);
141724d039d7SMichael Lange     ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr);
141870034214SMatthew G. Knepley     for (p = 0; p < numPoints; ++p) {
141924d039d7SMichael Lange       adjSize = PETSC_DETERMINE;
142024d039d7SMichael Lange       ierr = DMPlexGetAdjacency(dm, points[p], &adjSize, &adj);CHKERRQ(ierr);
142124d039d7SMichael Lange       for (a = 0; a < adjSize; ++a) {ierr = DMLabelSetValue(label, adj[a], rank);CHKERRQ(ierr);}
142270034214SMatthew G. Knepley     }
142324d039d7SMichael Lange     ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr);
142424d039d7SMichael Lange     ierr = ISDestroy(&pointIS);CHKERRQ(ierr);
142570034214SMatthew G. Knepley   }
142624d039d7SMichael Lange   ierr = ISRestoreIndices(rankIS, &ranks);CHKERRQ(ierr);
142724d039d7SMichael Lange   ierr = ISDestroy(&rankIS);CHKERRQ(ierr);
142824d039d7SMichael Lange   ierr = PetscFree(adj);CHKERRQ(ierr);
142924d039d7SMichael Lange   PetscFunctionReturn(0);
143070034214SMatthew G. Knepley }
143170034214SMatthew G. Knepley 
143224d039d7SMichael Lange #undef __FUNCT__
14331fd9873aSMichael Lange #define __FUNCT__ "DMPlexPartitionLabelInvert"
14341fd9873aSMichael Lange /*@
14351fd9873aSMichael Lange   DMPlexPartitionLabelInvert - Create a partition label of remote roots from a local root label
14361fd9873aSMichael Lange 
14371fd9873aSMichael Lange   Input Parameters:
14381fd9873aSMichael Lange + dm        - The DM
14391fd9873aSMichael Lange . rootLabel - DMLabel assinging ranks to local roots
14401fd9873aSMichael Lange . processSF - A star forest mapping into the local index on each remote rank
14411fd9873aSMichael Lange 
14421fd9873aSMichael Lange   Output Parameter:
14431fd9873aSMichael Lange - leafLabel - DMLabel assinging ranks to remote roots
14441fd9873aSMichael Lange 
14451fd9873aSMichael Lange   Note: The rootLabel defines a send pattern by mapping local points to remote target ranks. The
14461fd9873aSMichael Lange   resulting leafLabel is a receiver mapping of remote roots to their parent rank.
14471fd9873aSMichael Lange 
14481fd9873aSMichael Lange   Level: developer
14491fd9873aSMichael Lange 
14501fd9873aSMichael Lange .seealso: DMPlexPartitionLabelCreateSF, DMPlexDistribute(), DMPlexCreateOverlap
14511fd9873aSMichael Lange @*/
14521fd9873aSMichael Lange PetscErrorCode DMPlexPartitionLabelInvert(DM dm, DMLabel rootLabel, PetscSF processSF, DMLabel leafLabel)
14531fd9873aSMichael Lange {
14541fd9873aSMichael Lange   MPI_Comm           comm;
14551fd9873aSMichael Lange   PetscMPIInt        rank, numProcs;
14561fd9873aSMichael Lange   PetscInt           p, n, numNeighbors, size, l, nleaves;
14571fd9873aSMichael Lange   PetscSF            sfPoint;
14581fd9873aSMichael Lange   PetscSFNode       *rootPoints, *leafPoints;
14591fd9873aSMichael Lange   PetscSection       rootSection, leafSection;
14601fd9873aSMichael Lange   const PetscSFNode *remote;
14611fd9873aSMichael Lange   const PetscInt    *local, *neighbors;
14621fd9873aSMichael Lange   IS                 valueIS;
14631fd9873aSMichael Lange   PetscErrorCode     ierr;
14641fd9873aSMichael Lange 
14651fd9873aSMichael Lange   PetscFunctionBegin;
14661fd9873aSMichael Lange   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
14671fd9873aSMichael Lange   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
14681fd9873aSMichael Lange   ierr = MPI_Comm_size(comm, &numProcs);CHKERRQ(ierr);
14691fd9873aSMichael Lange   ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
14701fd9873aSMichael Lange 
14711fd9873aSMichael Lange   /* Convert to (point, rank) and use actual owners */
14721fd9873aSMichael Lange   ierr = PetscSectionCreate(comm, &rootSection);CHKERRQ(ierr);
14731fd9873aSMichael Lange   ierr = PetscSectionSetChart(rootSection, 0, numProcs);CHKERRQ(ierr);
14741fd9873aSMichael Lange   ierr = DMLabelGetValueIS(rootLabel, &valueIS);CHKERRQ(ierr);
14751fd9873aSMichael Lange   ierr = ISGetLocalSize(valueIS, &numNeighbors);CHKERRQ(ierr);
14761fd9873aSMichael Lange   ierr = ISGetIndices(valueIS, &neighbors);CHKERRQ(ierr);
14771fd9873aSMichael Lange   for (n = 0; n < numNeighbors; ++n) {
14781fd9873aSMichael Lange     PetscInt numPoints;
14791fd9873aSMichael Lange 
14801fd9873aSMichael Lange     ierr = DMLabelGetStratumSize(rootLabel, neighbors[n], &numPoints);CHKERRQ(ierr);
14811fd9873aSMichael Lange     ierr = PetscSectionAddDof(rootSection, neighbors[n], numPoints);CHKERRQ(ierr);
14821fd9873aSMichael Lange   }
14831fd9873aSMichael Lange   ierr = PetscSectionSetUp(rootSection);CHKERRQ(ierr);
14841fd9873aSMichael Lange   ierr = PetscSectionGetStorageSize(rootSection, &size);CHKERRQ(ierr);
14851fd9873aSMichael Lange   ierr = PetscMalloc1(size, &rootPoints);CHKERRQ(ierr);
14861fd9873aSMichael Lange   ierr = PetscSFGetGraph(sfPoint, NULL, &nleaves, &local, &remote);CHKERRQ(ierr);
14871fd9873aSMichael Lange   for (n = 0; n < numNeighbors; ++n) {
14881fd9873aSMichael Lange     IS              pointIS;
14891fd9873aSMichael Lange     const PetscInt *points;
14901fd9873aSMichael Lange     PetscInt        off, numPoints, p;
14911fd9873aSMichael Lange 
14921fd9873aSMichael Lange     ierr = PetscSectionGetOffset(rootSection, neighbors[n], &off);CHKERRQ(ierr);
14931fd9873aSMichael Lange     ierr = DMLabelGetStratumIS(rootLabel, neighbors[n], &pointIS);CHKERRQ(ierr);
14941fd9873aSMichael Lange     ierr = ISGetLocalSize(pointIS, &numPoints);CHKERRQ(ierr);
14951fd9873aSMichael Lange     ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr);
14961fd9873aSMichael Lange     for (p = 0; p < numPoints; ++p) {
1497f8987ae8SMichael Lange       if (local) {ierr = PetscFindInt(points[p], nleaves, local, &l);CHKERRQ(ierr);}
1498f8987ae8SMichael Lange       else       {l = -1;}
14991fd9873aSMichael Lange       if (l >= 0) {rootPoints[off+p] = remote[l];}
15001fd9873aSMichael Lange       else        {rootPoints[off+p].index = points[p]; rootPoints[off+p].rank = rank;}
15011fd9873aSMichael Lange     }
15021fd9873aSMichael Lange     ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr);
15031fd9873aSMichael Lange     ierr = ISDestroy(&pointIS);CHKERRQ(ierr);
15041fd9873aSMichael Lange   }
15051fd9873aSMichael Lange   ierr = ISRestoreIndices(valueIS, &neighbors);CHKERRQ(ierr);
15061fd9873aSMichael Lange   ierr = ISDestroy(&valueIS);CHKERRQ(ierr);
15071fd9873aSMichael Lange   /* Communicate overlap */
15081fd9873aSMichael Lange   ierr = PetscSectionCreate(comm, &leafSection);CHKERRQ(ierr);
15091fd9873aSMichael Lange   ierr = DMPlexDistributeData(dm, processSF, rootSection, MPIU_2INT, rootPoints, leafSection, (void**) &leafPoints);CHKERRQ(ierr);
15101fd9873aSMichael Lange   /* Filter remote contributions (ovLeafPoints) into the overlapSF */
15111fd9873aSMichael Lange   ierr = PetscSectionGetStorageSize(leafSection, &size);CHKERRQ(ierr);
15121fd9873aSMichael Lange   for (p = 0; p < size; p++) {
15131fd9873aSMichael Lange     ierr = DMLabelSetValue(leafLabel, leafPoints[p].index, leafPoints[p].rank);CHKERRQ(ierr);
15141fd9873aSMichael Lange   }
15151fd9873aSMichael Lange   ierr = PetscFree(rootPoints);CHKERRQ(ierr);
15161fd9873aSMichael Lange   ierr = PetscSectionDestroy(&rootSection);CHKERRQ(ierr);
15171fd9873aSMichael Lange   ierr = PetscFree(leafPoints);CHKERRQ(ierr);
15181fd9873aSMichael Lange   ierr = PetscSectionDestroy(&leafSection);CHKERRQ(ierr);
15191fd9873aSMichael Lange   PetscFunctionReturn(0);
15201fd9873aSMichael Lange }
15211fd9873aSMichael Lange 
15221fd9873aSMichael Lange #undef __FUNCT__
1523aa3148a8SMichael Lange #define __FUNCT__ "DMPlexPartitionLabelCreateSF"
1524aa3148a8SMichael Lange /*@
1525aa3148a8SMichael Lange   DMPlexPartitionLabelCreateSF - Create a star forest from a label that assigns ranks to points
1526aa3148a8SMichael Lange 
1527aa3148a8SMichael Lange   Input Parameters:
1528aa3148a8SMichael Lange + dm    - The DM
1529aa3148a8SMichael Lange . label - DMLabel assinging ranks to remote roots
1530aa3148a8SMichael Lange 
1531aa3148a8SMichael Lange   Output Parameter:
1532aa3148a8SMichael Lange - sf    - The star forest communication context encapsulating the defined mapping
1533aa3148a8SMichael Lange 
1534aa3148a8SMichael Lange   Note: The incoming label is a receiver mapping of remote points to their parent rank.
1535aa3148a8SMichael Lange 
1536aa3148a8SMichael Lange   Level: developer
1537aa3148a8SMichael Lange 
1538aa3148a8SMichael Lange .seealso: DMPlexDistribute(), DMPlexCreateOverlap
1539aa3148a8SMichael Lange @*/
1540aa3148a8SMichael Lange PetscErrorCode DMPlexPartitionLabelCreateSF(DM dm, DMLabel label, PetscSF *sf)
1541aa3148a8SMichael Lange {
154243f7d02bSMichael Lange   PetscMPIInt     rank, numProcs;
154343f7d02bSMichael Lange   PetscInt        n, numRemote, p, numPoints, pStart, pEnd, idx = 0;
1544aa3148a8SMichael Lange   PetscSFNode    *remotePoints;
154543f7d02bSMichael Lange   IS              remoteRootIS;
154643f7d02bSMichael Lange   const PetscInt *remoteRoots;
1547aa3148a8SMichael Lange   PetscErrorCode ierr;
1548aa3148a8SMichael Lange 
1549aa3148a8SMichael Lange   PetscFunctionBegin;
155043f7d02bSMichael Lange   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
1551aa3148a8SMichael Lange   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm), &numProcs);CHKERRQ(ierr);
1552aa3148a8SMichael Lange 
1553aa3148a8SMichael Lange   for (numRemote = 0, n = 0; n < numProcs; ++n) {
1554aa3148a8SMichael Lange     ierr = DMLabelGetStratumSize(label, n, &numPoints);CHKERRQ(ierr);
1555aa3148a8SMichael Lange     numRemote += numPoints;
1556aa3148a8SMichael Lange   }
1557aa3148a8SMichael Lange   ierr = PetscMalloc1(numRemote, &remotePoints);CHKERRQ(ierr);
155843f7d02bSMichael Lange   /* Put owned points first */
155943f7d02bSMichael Lange   ierr = DMLabelGetStratumSize(label, rank, &numPoints);CHKERRQ(ierr);
156043f7d02bSMichael Lange   if (numPoints > 0) {
156143f7d02bSMichael Lange     ierr = DMLabelGetStratumIS(label, rank, &remoteRootIS);CHKERRQ(ierr);
156243f7d02bSMichael Lange     ierr = ISGetIndices(remoteRootIS, &remoteRoots);CHKERRQ(ierr);
156343f7d02bSMichael Lange     for (p = 0; p < numPoints; p++) {
156443f7d02bSMichael Lange       remotePoints[idx].index = remoteRoots[p];
156543f7d02bSMichael Lange       remotePoints[idx].rank = rank;
156643f7d02bSMichael Lange       idx++;
156743f7d02bSMichael Lange     }
156843f7d02bSMichael Lange     ierr = ISRestoreIndices(remoteRootIS, &remoteRoots);CHKERRQ(ierr);
156943f7d02bSMichael Lange     ierr = ISDestroy(&remoteRootIS);CHKERRQ(ierr);
157043f7d02bSMichael Lange   }
157143f7d02bSMichael Lange   /* Now add remote points */
157243f7d02bSMichael Lange   for (n = 0; n < numProcs; ++n) {
1573aa3148a8SMichael Lange     ierr = DMLabelGetStratumSize(label, n, &numPoints);CHKERRQ(ierr);
157443f7d02bSMichael Lange     if (numPoints <= 0 || n == rank) continue;
1575aa3148a8SMichael Lange     ierr = DMLabelGetStratumIS(label, n, &remoteRootIS);CHKERRQ(ierr);
1576aa3148a8SMichael Lange     ierr = ISGetIndices(remoteRootIS, &remoteRoots);CHKERRQ(ierr);
1577aa3148a8SMichael Lange     for (p = 0; p < numPoints; p++) {
1578aa3148a8SMichael Lange       remotePoints[idx].index = remoteRoots[p];
1579aa3148a8SMichael Lange       remotePoints[idx].rank = n;
1580aa3148a8SMichael Lange       idx++;
1581aa3148a8SMichael Lange     }
1582aa3148a8SMichael Lange     ierr = ISRestoreIndices(remoteRootIS, &remoteRoots);CHKERRQ(ierr);
1583aa3148a8SMichael Lange     ierr = ISDestroy(&remoteRootIS);CHKERRQ(ierr);
1584aa3148a8SMichael Lange   }
1585aa3148a8SMichael Lange   ierr = PetscSFCreate(PetscObjectComm((PetscObject) dm), sf);CHKERRQ(ierr);
1586aa3148a8SMichael Lange   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
1587aa3148a8SMichael Lange   ierr = PetscSFSetGraph(*sf, pEnd-pStart, numRemote, NULL, PETSC_OWN_POINTER, remotePoints, PETSC_OWN_POINTER);CHKERRQ(ierr);
158870034214SMatthew G. Knepley   PetscFunctionReturn(0);
158970034214SMatthew G. Knepley }
1590