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), §ion);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, <ogCells);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(<ogCells);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(§ion);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