19f074e33SMatthew G Knepley #include <petsc-private/dmpleximpl.h> /*I "petscdmplex.h" I*/ 29f074e33SMatthew G Knepley #include <../src/sys/utils/hash.h> 39f074e33SMatthew G Knepley 49f074e33SMatthew G Knepley #undef __FUNCT__ 59a5b13a2SMatthew G Knepley #define __FUNCT__ "DMPlexGetFaces_Internal" 69f074e33SMatthew G Knepley /* 79a5b13a2SMatthew G Knepley DMPlexGetFaces_Internal - Gets groups of vertices that correspond to faces for the given cell 89f074e33SMatthew G Knepley */ 9439ece16SMatthew G. Knepley PetscErrorCode DMPlexGetFaces_Internal(DM dm, PetscInt dim, PetscInt p, PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[]) 109f074e33SMatthew G Knepley { 119f074e33SMatthew G Knepley const PetscInt *cone = NULL; 129a5b13a2SMatthew G Knepley PetscInt maxConeSize, maxSupportSize, coneSize; 139f074e33SMatthew G Knepley PetscErrorCode ierr; 149f074e33SMatthew G Knepley 159f074e33SMatthew G Knepley PetscFunctionBegin; 169f074e33SMatthew G Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 179a5b13a2SMatthew G Knepley ierr = DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize);CHKERRQ(ierr); 189f074e33SMatthew G Knepley ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr); 199f074e33SMatthew G Knepley ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr); 20439ece16SMatthew G. Knepley ierr = DMPlexGetRawFaces_Internal(dm, dim, coneSize, cone, numFaces, faceSize, faces);CHKERRQ(ierr); 21439ece16SMatthew G. Knepley PetscFunctionReturn(0); 22439ece16SMatthew G. Knepley } 23439ece16SMatthew G. Knepley 24439ece16SMatthew G. Knepley #undef __FUNCT__ 25439ece16SMatthew G. Knepley #define __FUNCT__ "DMPlexRestoreFaces_Internal" 26439ece16SMatthew G. Knepley /* 27439ece16SMatthew G. Knepley DMPlexRestoreFaces_Internal - Restores the array 28439ece16SMatthew G. Knepley */ 29439ece16SMatthew G. Knepley PetscErrorCode DMPlexRestoreFaces_Internal(DM dm, PetscInt dim, PetscInt p, PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[]) 30439ece16SMatthew G. Knepley { 31439ece16SMatthew G. Knepley PetscErrorCode ierr; 32439ece16SMatthew G. Knepley 33439ece16SMatthew G. Knepley PetscFunctionBegin; 3406acb219SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, 0, PETSC_INT, (void *) faces);CHKERRQ(ierr); 35439ece16SMatthew G. Knepley PetscFunctionReturn(0); 36439ece16SMatthew G. Knepley } 37439ece16SMatthew G. Knepley 38439ece16SMatthew G. Knepley #undef __FUNCT__ 39439ece16SMatthew G. Knepley #define __FUNCT__ "DMPlexGetRawFaces_Internal" 40439ece16SMatthew G. Knepley /* 41439ece16SMatthew G. Knepley DMPlexGetRawFaces_Internal - Gets groups of vertices that correspond to faces for the given cone 42439ece16SMatthew G. Knepley */ 43439ece16SMatthew G. Knepley PetscErrorCode DMPlexGetRawFaces_Internal(DM dm, PetscInt dim, PetscInt coneSize, const PetscInt cone[], PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[]) 44439ece16SMatthew G. Knepley { 45439ece16SMatthew G. Knepley PetscInt *facesTmp; 46439ece16SMatthew G. Knepley PetscInt maxConeSize, maxSupportSize; 47439ece16SMatthew G. Knepley PetscErrorCode ierr; 48439ece16SMatthew G. Knepley 49439ece16SMatthew G. Knepley PetscFunctionBegin; 50439ece16SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 51439ece16SMatthew G. Knepley ierr = DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize);CHKERRQ(ierr); 528496a43fSMatthew G. Knepley if (faces) {ierr = DMGetWorkArray(dm, PetscSqr(PetscMax(maxConeSize, maxSupportSize)), PETSC_INT, &facesTmp);CHKERRQ(ierr);} 539f074e33SMatthew G Knepley switch (dim) { 549f074e33SMatthew G Knepley case 2: 559f074e33SMatthew G Knepley switch (coneSize) { 569f074e33SMatthew G Knepley case 3: 579a5b13a2SMatthew G Knepley if (faces) { 589a5b13a2SMatthew G Knepley facesTmp[0] = cone[0]; facesTmp[1] = cone[1]; 599a5b13a2SMatthew G Knepley facesTmp[2] = cone[1]; facesTmp[3] = cone[2]; 609a5b13a2SMatthew G Knepley facesTmp[4] = cone[2]; facesTmp[5] = cone[0]; 619a5b13a2SMatthew G Knepley *faces = facesTmp; 629a5b13a2SMatthew G Knepley } 639a5b13a2SMatthew G Knepley if (numFaces) *numFaces = 3; 649a5b13a2SMatthew G Knepley if (faceSize) *faceSize = 2; 659f074e33SMatthew G Knepley break; 669f074e33SMatthew G Knepley case 4: 679a5b13a2SMatthew G Knepley /* Vertices follow right hand rule */ 689a5b13a2SMatthew G Knepley if (faces) { 699a5b13a2SMatthew G Knepley facesTmp[0] = cone[0]; facesTmp[1] = cone[1]; 709a5b13a2SMatthew G Knepley facesTmp[2] = cone[1]; facesTmp[3] = cone[2]; 719a5b13a2SMatthew G Knepley facesTmp[4] = cone[2]; facesTmp[5] = cone[3]; 729a5b13a2SMatthew G Knepley facesTmp[6] = cone[3]; facesTmp[7] = cone[0]; 739a5b13a2SMatthew G Knepley *faces = facesTmp; 749a5b13a2SMatthew G Knepley } 759a5b13a2SMatthew G Knepley if (numFaces) *numFaces = 4; 769a5b13a2SMatthew G Knepley if (faceSize) *faceSize = 2; 779f074e33SMatthew G Knepley break; 789f074e33SMatthew G Knepley default: 799f074e33SMatthew G Knepley SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim); 809f074e33SMatthew G Knepley } 819f074e33SMatthew G Knepley break; 829f074e33SMatthew G Knepley case 3: 839f074e33SMatthew G Knepley switch (coneSize) { 849f074e33SMatthew G Knepley case 3: 859a5b13a2SMatthew G Knepley if (faces) { 869a5b13a2SMatthew G Knepley facesTmp[0] = cone[0]; facesTmp[1] = cone[1]; 879a5b13a2SMatthew G Knepley facesTmp[2] = cone[1]; facesTmp[3] = cone[2]; 889a5b13a2SMatthew G Knepley facesTmp[4] = cone[2]; facesTmp[5] = cone[0]; 899a5b13a2SMatthew G Knepley *faces = facesTmp; 909a5b13a2SMatthew G Knepley } 919a5b13a2SMatthew G Knepley if (numFaces) *numFaces = 3; 929a5b13a2SMatthew G Knepley if (faceSize) *faceSize = 2; 939f074e33SMatthew G Knepley break; 949f074e33SMatthew G Knepley case 4: 952e1b13c2SMatthew G. Knepley /* Vertices of first face follow right hand rule and normal points away from last vertex */ 969a5b13a2SMatthew G Knepley if (faces) { 972e1b13c2SMatthew G. Knepley facesTmp[0] = cone[0]; facesTmp[1] = cone[1]; facesTmp[2] = cone[2]; 982e1b13c2SMatthew G. Knepley facesTmp[3] = cone[0]; facesTmp[4] = cone[3]; facesTmp[5] = cone[1]; 992e1b13c2SMatthew G. Knepley facesTmp[6] = cone[0]; facesTmp[7] = cone[2]; facesTmp[8] = cone[3]; 1002e1b13c2SMatthew G. Knepley facesTmp[9] = cone[2]; facesTmp[10] = cone[1]; facesTmp[11] = cone[3]; 1019a5b13a2SMatthew G Knepley *faces = facesTmp; 1029a5b13a2SMatthew G Knepley } 1039a5b13a2SMatthew G Knepley if (numFaces) *numFaces = 4; 1049a5b13a2SMatthew G Knepley if (faceSize) *faceSize = 3; 1059a5b13a2SMatthew G Knepley break; 1069a5b13a2SMatthew G Knepley case 8: 1079a5b13a2SMatthew G Knepley if (faces) { 1082e1b13c2SMatthew G. Knepley facesTmp[0] = cone[0]; facesTmp[1] = cone[1]; facesTmp[2] = cone[2]; facesTmp[3] = cone[3]; 109a014044eSMatthew G Knepley facesTmp[4] = cone[4]; facesTmp[5] = cone[5]; facesTmp[6] = cone[6]; facesTmp[7] = cone[7]; 1102e1b13c2SMatthew G. Knepley facesTmp[8] = cone[0]; facesTmp[9] = cone[3]; facesTmp[10] = cone[5]; facesTmp[11] = cone[4]; 1112e1b13c2SMatthew G. Knepley facesTmp[12] = cone[2]; facesTmp[13] = cone[1]; facesTmp[14] = cone[7]; facesTmp[15] = cone[6]; 1122e1b13c2SMatthew G. Knepley facesTmp[16] = cone[3]; facesTmp[17] = cone[2]; facesTmp[18] = cone[6]; facesTmp[19] = cone[5]; 1132e1b13c2SMatthew G. Knepley facesTmp[20] = cone[0]; facesTmp[21] = cone[4]; facesTmp[22] = cone[7]; facesTmp[23] = cone[1]; 1149a5b13a2SMatthew G Knepley *faces = facesTmp; 1159a5b13a2SMatthew G Knepley } 1169a5b13a2SMatthew G Knepley if (numFaces) *numFaces = 6; 1179a5b13a2SMatthew G Knepley if (faceSize) *faceSize = 4; 1189f074e33SMatthew G Knepley break; 1199f074e33SMatthew G Knepley default: 1209f074e33SMatthew G Knepley SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim); 1219f074e33SMatthew G Knepley } 1229f074e33SMatthew G Knepley break; 1239f074e33SMatthew G Knepley default: 1249f074e33SMatthew G Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Dimension %D not supported", dim); 1259f074e33SMatthew G Knepley } 1269f074e33SMatthew G Knepley PetscFunctionReturn(0); 1279f074e33SMatthew G Knepley } 1289f074e33SMatthew G Knepley 1299f074e33SMatthew G Knepley #undef __FUNCT__ 1309a5b13a2SMatthew G Knepley #define __FUNCT__ "DMPlexInterpolateFaces_Internal" 1319a5b13a2SMatthew G Knepley /* This interpolates faces for cells at some stratum */ 1329a5b13a2SMatthew G Knepley static PetscErrorCode DMPlexInterpolateFaces_Internal(DM dm, PetscInt cellDepth, DM idm) 1339f074e33SMatthew G Knepley { 134d4efc6f1SMatthew G. Knepley DMLabel subpointMap; 1359a5b13a2SMatthew G Knepley PetscHashIJKL faceTable; 1369a5b13a2SMatthew G Knepley PetscInt *pStart, *pEnd; 1379a5b13a2SMatthew G Knepley PetscInt cellDim, depth, faceDepth = cellDepth, numPoints = 0, faceSizeAll = 0, face, c, d; 1389f074e33SMatthew G Knepley PetscErrorCode ierr; 1399f074e33SMatthew G Knepley 1409f074e33SMatthew G Knepley PetscFunctionBegin; 1419a5b13a2SMatthew G Knepley ierr = DMPlexGetDimension(dm, &cellDim);CHKERRQ(ierr); 142d4efc6f1SMatthew G. Knepley /* HACK: I need a better way to determine face dimension, or an alternative to GetFaces() */ 143d4efc6f1SMatthew G. Knepley ierr = DMPlexGetSubpointMap(dm, &subpointMap);CHKERRQ(ierr); 144d4efc6f1SMatthew G. Knepley if (subpointMap) ++cellDim; 1459a5b13a2SMatthew G Knepley ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 1469a5b13a2SMatthew G Knepley ++depth; 1479a5b13a2SMatthew G Knepley ++cellDepth; 1489a5b13a2SMatthew G Knepley cellDim -= depth - cellDepth; 149*dcca6d9dSJed Brown ierr = PetscMalloc2(depth+1,&pStart,depth+1,&pEnd);CHKERRQ(ierr); 1509a5b13a2SMatthew G Knepley for (d = depth-1; d >= faceDepth; --d) { 1519a5b13a2SMatthew G Knepley ierr = DMPlexGetDepthStratum(dm, d, &pStart[d+1], &pEnd[d+1]);CHKERRQ(ierr); 1529f074e33SMatthew G Knepley } 1539a5b13a2SMatthew G Knepley ierr = DMPlexGetDepthStratum(dm, -1, NULL, &pStart[faceDepth]);CHKERRQ(ierr); 1549a5b13a2SMatthew G Knepley pEnd[faceDepth] = pStart[faceDepth]; 1559a5b13a2SMatthew G Knepley for (d = faceDepth-1; d >= 0; --d) { 1569a5b13a2SMatthew G Knepley ierr = DMPlexGetDepthStratum(dm, d, &pStart[d], &pEnd[d]);CHKERRQ(ierr); 1579f074e33SMatthew G Knepley } 1589a5b13a2SMatthew G Knepley if (pEnd[cellDepth] > pStart[cellDepth]) {ierr = DMPlexGetFaces_Internal(dm, cellDim, pStart[cellDepth], NULL, &faceSizeAll, NULL);CHKERRQ(ierr);} 1599a5b13a2SMatthew G Knepley if (faceSizeAll > 4) SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll); 1609a5b13a2SMatthew G Knepley ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr); 1619a5b13a2SMatthew G Knepley for (c = pStart[cellDepth], face = pStart[faceDepth]; c < pEnd[cellDepth]; ++c) { 1629f074e33SMatthew G Knepley const PetscInt *cellFaces; 163735a0255SMatthew G. Knepley PetscInt numCellFaces, faceSize, cf; 1649f074e33SMatthew G Knepley 1659a5b13a2SMatthew G Knepley ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr); 1669a5b13a2SMatthew G Knepley if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll); 1679f074e33SMatthew G Knepley for (cf = 0; cf < numCellFaces; ++cf) { 1689a5b13a2SMatthew G Knepley const PetscInt *cellFace = &cellFaces[cf*faceSize]; 1699a5b13a2SMatthew G Knepley PetscHashIJKLKey key; 170735a0255SMatthew G. Knepley PetscHashIJKLIter missing, iter; 1719f074e33SMatthew G Knepley 1729a5b13a2SMatthew G Knepley if (faceSize == 2) { 1739a5b13a2SMatthew G Knepley key.i = PetscMin(cellFace[0], cellFace[1]); 1749a5b13a2SMatthew G Knepley key.j = PetscMax(cellFace[0], cellFace[1]); 1757fcd4ef0SJed Brown key.k = 0; 1767fcd4ef0SJed Brown key.l = 0; 1779a5b13a2SMatthew G Knepley } else { 1789a5b13a2SMatthew G Knepley key.i = cellFace[0]; key.j = cellFace[1]; key.k = cellFace[2]; key.l = faceSize > 3 ? cellFace[3] : 0; 1799a5b13a2SMatthew G Knepley ierr = PetscSortInt(faceSize, (PetscInt *) &key); 1809f074e33SMatthew G Knepley } 181735a0255SMatthew G. Knepley ierr = PetscHashIJKLPut(faceTable, key, &missing, &iter);CHKERRQ(ierr); 182735a0255SMatthew G. Knepley if (missing) {ierr = PetscHashIJKLSet(faceTable, iter, face++);CHKERRQ(ierr);} 1839a5b13a2SMatthew G Knepley } 184439ece16SMatthew G. Knepley ierr = DMPlexRestoreFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr); 1859a5b13a2SMatthew G Knepley } 1869a5b13a2SMatthew G Knepley pEnd[faceDepth] = face; 1879a5b13a2SMatthew G Knepley ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr); 1889a5b13a2SMatthew G Knepley /* Count new points */ 1899a5b13a2SMatthew G Knepley for (d = 0; d <= depth; ++d) { 1909a5b13a2SMatthew G Knepley numPoints += pEnd[d]-pStart[d]; 1919a5b13a2SMatthew G Knepley } 1929a5b13a2SMatthew G Knepley ierr = DMPlexSetChart(idm, 0, numPoints);CHKERRQ(ierr); 1939a5b13a2SMatthew G Knepley /* Set cone sizes */ 1949a5b13a2SMatthew G Knepley for (d = 0; d <= depth; ++d) { 1959a5b13a2SMatthew G Knepley PetscInt coneSize, p; 1969f074e33SMatthew G Knepley 1979a5b13a2SMatthew G Knepley if (d == faceDepth) { 1989a5b13a2SMatthew G Knepley for (p = pStart[d]; p < pEnd[d]; ++p) { 1999a5b13a2SMatthew G Knepley /* I see no way to do this if we admit faces of different shapes */ 2009a5b13a2SMatthew G Knepley ierr = DMPlexSetConeSize(idm, p, faceSizeAll);CHKERRQ(ierr); 2019a5b13a2SMatthew G Knepley } 202a014044eSMatthew G Knepley } else if (d == cellDepth) { 203a014044eSMatthew G Knepley for (p = pStart[d]; p < pEnd[d]; ++p) { 204a014044eSMatthew G Knepley /* Number of cell faces may be different from number of cell vertices*/ 205a014044eSMatthew G Knepley ierr = DMPlexGetFaces_Internal(dm, cellDim, p, &coneSize, NULL, NULL);CHKERRQ(ierr); 206a014044eSMatthew G Knepley ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr); 207a014044eSMatthew G Knepley } 2089a5b13a2SMatthew G Knepley } else { 2099a5b13a2SMatthew G Knepley for (p = pStart[d]; p < pEnd[d]; ++p) { 2109a5b13a2SMatthew G Knepley ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr); 2119a5b13a2SMatthew G Knepley ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr); 2129f074e33SMatthew G Knepley } 2139f074e33SMatthew G Knepley } 2149f074e33SMatthew G Knepley } 2159f074e33SMatthew G Knepley ierr = DMSetUp(idm);CHKERRQ(ierr); 2169f074e33SMatthew G Knepley /* Get face cones from subsets of cell vertices */ 2179a5b13a2SMatthew G Knepley if (faceSizeAll > 4) SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll); 2189a5b13a2SMatthew G Knepley ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr); 2199a5b13a2SMatthew G Knepley for (d = depth; d > cellDepth; --d) { 2209a5b13a2SMatthew G Knepley const PetscInt *cone; 2219a5b13a2SMatthew G Knepley PetscInt p; 2229a5b13a2SMatthew G Knepley 2239a5b13a2SMatthew G Knepley for (p = pStart[d]; p < pEnd[d]; ++p) { 2249a5b13a2SMatthew G Knepley ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr); 2259a5b13a2SMatthew G Knepley ierr = DMPlexSetCone(idm, p, cone);CHKERRQ(ierr); 2269a5b13a2SMatthew G Knepley ierr = DMPlexGetConeOrientation(dm, p, &cone);CHKERRQ(ierr); 2279a5b13a2SMatthew G Knepley ierr = DMPlexSetConeOrientation(idm, p, cone);CHKERRQ(ierr); 2289f074e33SMatthew G Knepley } 2299a5b13a2SMatthew G Knepley } 2309a5b13a2SMatthew G Knepley for (c = pStart[cellDepth], face = pStart[faceDepth]; c < pEnd[cellDepth]; ++c) { 2319f074e33SMatthew G Knepley const PetscInt *cellFaces; 232735a0255SMatthew G. Knepley PetscInt numCellFaces, faceSize, cf; 2339f074e33SMatthew G Knepley 2349a5b13a2SMatthew G Knepley ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr); 2359a5b13a2SMatthew G Knepley if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll); 2369f074e33SMatthew G Knepley for (cf = 0; cf < numCellFaces; ++cf) { 2379a5b13a2SMatthew G Knepley const PetscInt *cellFace = &cellFaces[cf*faceSize]; 2389a5b13a2SMatthew G Knepley PetscHashIJKLKey key; 239735a0255SMatthew G. Knepley PetscHashIJKLIter missing, iter; 2409f074e33SMatthew G Knepley 2419a5b13a2SMatthew G Knepley if (faceSize == 2) { 2429a5b13a2SMatthew G Knepley key.i = PetscMin(cellFace[0], cellFace[1]); 2439a5b13a2SMatthew G Knepley key.j = PetscMax(cellFace[0], cellFace[1]); 2447fcd4ef0SJed Brown key.k = 0; 2457fcd4ef0SJed Brown key.l = 0; 2469a5b13a2SMatthew G Knepley } else { 2479a5b13a2SMatthew G Knepley key.i = cellFace[0]; key.j = cellFace[1]; key.k = cellFace[2]; key.l = faceSize > 3 ? cellFace[3] : 0; 2489a5b13a2SMatthew G Knepley ierr = PetscSortInt(faceSize, (PetscInt *) &key); 2499f074e33SMatthew G Knepley } 250735a0255SMatthew G. Knepley ierr = PetscHashIJKLPut(faceTable, key, &missing, &iter);CHKERRQ(ierr); 251735a0255SMatthew G. Knepley if (missing) { 2529a5b13a2SMatthew G Knepley ierr = DMPlexSetCone(idm, face, cellFace);CHKERRQ(ierr); 253735a0255SMatthew G. Knepley ierr = PetscHashIJKLSet(faceTable, iter, face);CHKERRQ(ierr); 254735a0255SMatthew G. Knepley ierr = DMPlexInsertCone(idm, c, cf, face++);CHKERRQ(ierr); 2559a5b13a2SMatthew G Knepley } else { 2569a5b13a2SMatthew G Knepley const PetscInt *cone; 257735a0255SMatthew G. Knepley PetscInt coneSize, ornt, i, j, f; 2589f074e33SMatthew G Knepley 259735a0255SMatthew G. Knepley ierr = PetscHashIJKLGet(faceTable, iter, &f);CHKERRQ(ierr); 2609a5b13a2SMatthew G Knepley ierr = DMPlexInsertCone(idm, c, cf, f);CHKERRQ(ierr); 2612e1b13c2SMatthew G. Knepley /* Orient face: Do not allow reverse orientation at the first vertex */ 2629f074e33SMatthew G Knepley ierr = DMPlexGetConeSize(idm, f, &coneSize);CHKERRQ(ierr); 2639f074e33SMatthew G Knepley ierr = DMPlexGetCone(idm, f, &cone);CHKERRQ(ierr); 2649a5b13a2SMatthew G Knepley if (coneSize != faceSize) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of face vertices %D for face %D should be %D", coneSize, f, faceSize); 2659a5b13a2SMatthew G Knepley /* - First find the initial vertex */ 2669a5b13a2SMatthew G Knepley for (i = 0; i < faceSize; ++i) if (cellFace[0] == cone[i]) break; 2679a5b13a2SMatthew G Knepley /* - Try forward comparison */ 2689a5b13a2SMatthew G Knepley for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+j)%faceSize]) break; 2699a5b13a2SMatthew G Knepley if (j == faceSize) { 2709a5b13a2SMatthew G Knepley if ((faceSize == 2) && (i == 1)) ornt = -2; 2719a5b13a2SMatthew G Knepley else ornt = i; 2729a5b13a2SMatthew G Knepley } else { 2739a5b13a2SMatthew G Knepley /* - Try backward comparison */ 2749a5b13a2SMatthew G Knepley for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+faceSize-j)%faceSize]) break; 2752e1b13c2SMatthew G. Knepley if (j == faceSize) { 2762e1b13c2SMatthew G. Knepley if (i == 0) ornt = -faceSize; 277dc1a705cSMatthew G. Knepley else ornt = -i; 2782e1b13c2SMatthew G. Knepley } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not determine face orientation"); 2799a5b13a2SMatthew G Knepley } 2809a5b13a2SMatthew G Knepley ierr = DMPlexInsertConeOrientation(idm, c, cf, ornt);CHKERRQ(ierr); 2819f074e33SMatthew G Knepley } 2829f074e33SMatthew G Knepley } 283439ece16SMatthew G. Knepley ierr = DMPlexRestoreFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr); 2849f074e33SMatthew G Knepley } 2859a5b13a2SMatthew G Knepley if (face != pEnd[faceDepth]) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_PLIB, "Invalid number of faces %D should be %D", face-pStart[faceDepth], pEnd[faceDepth]-pStart[faceDepth]); 286c907b753SJed Brown ierr = PetscFree2(pStart,pEnd);CHKERRQ(ierr); 2879a5b13a2SMatthew G Knepley ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr); 2886551a8c7SMatthew G. Knepley ierr = PetscFree2(pStart,pEnd);CHKERRQ(ierr); 2899a5b13a2SMatthew G Knepley ierr = DMPlexSymmetrize(idm);CHKERRQ(ierr); 2909a5b13a2SMatthew G Knepley ierr = DMPlexStratify(idm);CHKERRQ(ierr); 2919f074e33SMatthew G Knepley PetscFunctionReturn(0); 2929f074e33SMatthew G Knepley } 2939f074e33SMatthew G Knepley 2949f074e33SMatthew G Knepley #undef __FUNCT__ 2959f074e33SMatthew G Knepley #define __FUNCT__ "DMPlexInterpolate" 29680330477SMatthew G. Knepley /*@ 29780330477SMatthew G. Knepley DMPlexInterpolate - Take in a cell-vertex mesh and return one with all intermediate faces, edges, etc. 29880330477SMatthew G. Knepley 29980330477SMatthew G. Knepley Collective on DM 30080330477SMatthew G. Knepley 30180330477SMatthew G. Knepley Input Parameter: 3024e3744c5SMatthew G. Knepley . dm - The DMPlex object with only cells and vertices 30380330477SMatthew G. Knepley 30480330477SMatthew G. Knepley Output Parameter: 3054e3744c5SMatthew G. Knepley . dmInt - The complete DMPlex object 30680330477SMatthew G. Knepley 30780330477SMatthew G. Knepley Level: intermediate 30880330477SMatthew G. Knepley 30980330477SMatthew G. Knepley .keywords: mesh 3104e3744c5SMatthew G. Knepley .seealso: DMPlexUninterpolate(), DMPlexCreateFromCellList() 31180330477SMatthew G. Knepley @*/ 3129f074e33SMatthew G Knepley PetscErrorCode DMPlexInterpolate(DM dm, DM *dmInt) 3139f074e33SMatthew G Knepley { 3149a5b13a2SMatthew G Knepley DM idm, odm = dm; 3152e1b13c2SMatthew G. Knepley PetscInt depth, dim, d; 3169f074e33SMatthew G Knepley PetscErrorCode ierr; 3179f074e33SMatthew G Knepley 3189f074e33SMatthew G Knepley PetscFunctionBegin; 319a72f3261SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_Interpolate,dm,0,0,0);CHKERRQ(ierr); 3202e1b13c2SMatthew G. Knepley ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 3219f074e33SMatthew G Knepley ierr = DMPlexGetDimension(dm, &dim);CHKERRQ(ierr); 32276b791aaSMatthew G. Knepley if (dim <= 1) { 32376b791aaSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr); 32476b791aaSMatthew G. Knepley idm = dm; 32576b791aaSMatthew G. Knepley } 3269a5b13a2SMatthew G Knepley for (d = 1; d < dim; ++d) { 3279a5b13a2SMatthew G Knepley /* Create interpolated mesh */ 3289a5b13a2SMatthew G Knepley ierr = DMCreate(PetscObjectComm((PetscObject)dm), &idm);CHKERRQ(ierr); 3299a5b13a2SMatthew G Knepley ierr = DMSetType(idm, DMPLEX);CHKERRQ(ierr); 3309a5b13a2SMatthew G Knepley ierr = DMPlexSetDimension(idm, dim);CHKERRQ(ierr); 3312e1b13c2SMatthew G. Knepley if (depth > 0) {ierr = DMPlexInterpolateFaces_Internal(odm, 1, idm);CHKERRQ(ierr);} 3329a5b13a2SMatthew G Knepley if (odm != dm) {ierr = DMDestroy(&odm);CHKERRQ(ierr);} 3339a5b13a2SMatthew G Knepley odm = idm; 3349f074e33SMatthew G Knepley } 3359a5b13a2SMatthew G Knepley *dmInt = idm; 336a72f3261SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_Interpolate,dm,0,0,0);CHKERRQ(ierr); 3379f074e33SMatthew G Knepley PetscFunctionReturn(0); 3389f074e33SMatthew G Knepley } 33907b0a1fcSMatthew G Knepley 34007b0a1fcSMatthew G Knepley #undef __FUNCT__ 34107b0a1fcSMatthew G Knepley #define __FUNCT__ "DMPlexCopyCoordinates" 34280330477SMatthew G. Knepley /*@ 34380330477SMatthew G. Knepley DMPlexCopyCoordinates - Copy coordinates from one mesh to another with the same vertices 34480330477SMatthew G. Knepley 34580330477SMatthew G. Knepley Collective on DM 34680330477SMatthew G. Knepley 34780330477SMatthew G. Knepley Input Parameter: 34880330477SMatthew G. Knepley . dmA - The DMPlex object with initial coordinates 34980330477SMatthew G. Knepley 35080330477SMatthew G. Knepley Output Parameter: 35180330477SMatthew G. Knepley . dmB - The DMPlex object with copied coordinates 35280330477SMatthew G. Knepley 35380330477SMatthew G. Knepley Level: intermediate 35480330477SMatthew G. Knepley 35580330477SMatthew G. Knepley Note: This is typically used when adding pieces other than vertices to a mesh 35680330477SMatthew G. Knepley 35780330477SMatthew G. Knepley .keywords: mesh 3585c386225SMatthew G. Knepley .seealso: DMCopyLabels(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM(), DMPlexGetCoordinateSection() 35980330477SMatthew G. Knepley @*/ 36007b0a1fcSMatthew G Knepley PetscErrorCode DMPlexCopyCoordinates(DM dmA, DM dmB) 36107b0a1fcSMatthew G Knepley { 36207b0a1fcSMatthew G Knepley Vec coordinatesA, coordinatesB; 36307b0a1fcSMatthew G Knepley PetscSection coordSectionA, coordSectionB; 36407b0a1fcSMatthew G Knepley PetscScalar *coordsA, *coordsB; 36507b0a1fcSMatthew G Knepley PetscInt spaceDim, vStartA, vStartB, vEndA, vEndB, coordSizeB, v, d; 36607b0a1fcSMatthew G Knepley PetscErrorCode ierr; 36707b0a1fcSMatthew G Knepley 36807b0a1fcSMatthew G Knepley PetscFunctionBegin; 36976b791aaSMatthew G. Knepley if (dmA == dmB) PetscFunctionReturn(0); 37007b0a1fcSMatthew G Knepley ierr = DMPlexGetDepthStratum(dmA, 0, &vStartA, &vEndA);CHKERRQ(ierr); 37107b0a1fcSMatthew G Knepley ierr = DMPlexGetDepthStratum(dmB, 0, &vStartB, &vEndB);CHKERRQ(ierr); 37207b0a1fcSMatthew G Knepley if ((vEndA-vStartA) != (vEndB-vStartB)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The number of vertices in first DM %d != %d in the second DM", vEndA-vStartA, vEndB-vStartB); 37307b0a1fcSMatthew G Knepley ierr = DMPlexGetCoordinateSection(dmA, &coordSectionA);CHKERRQ(ierr); 37407b0a1fcSMatthew G Knepley ierr = DMPlexGetCoordinateSection(dmB, &coordSectionB);CHKERRQ(ierr); 37507b0a1fcSMatthew G Knepley ierr = PetscSectionSetNumFields(coordSectionB, 1);CHKERRQ(ierr); 37607b0a1fcSMatthew G Knepley ierr = PetscSectionGetFieldComponents(coordSectionA, 0, &spaceDim);CHKERRQ(ierr); 37707b0a1fcSMatthew G Knepley ierr = PetscSectionSetFieldComponents(coordSectionB, 0, spaceDim);CHKERRQ(ierr); 37807b0a1fcSMatthew G Knepley ierr = PetscSectionSetChart(coordSectionB, vStartB, vEndB);CHKERRQ(ierr); 37907b0a1fcSMatthew G Knepley for (v = vStartB; v < vEndB; ++v) { 38007b0a1fcSMatthew G Knepley ierr = PetscSectionSetDof(coordSectionB, v, spaceDim);CHKERRQ(ierr); 38107b0a1fcSMatthew G Knepley ierr = PetscSectionSetFieldDof(coordSectionB, v, 0, spaceDim);CHKERRQ(ierr); 38207b0a1fcSMatthew G Knepley } 38307b0a1fcSMatthew G Knepley ierr = PetscSectionSetUp(coordSectionB);CHKERRQ(ierr); 38407b0a1fcSMatthew G Knepley ierr = PetscSectionGetStorageSize(coordSectionB, &coordSizeB);CHKERRQ(ierr); 38507b0a1fcSMatthew G Knepley ierr = DMGetCoordinatesLocal(dmA, &coordinatesA);CHKERRQ(ierr); 38607b0a1fcSMatthew G Knepley ierr = VecCreate(PetscObjectComm((PetscObject) dmB), &coordinatesB);CHKERRQ(ierr); 38707b0a1fcSMatthew G Knepley ierr = PetscObjectSetName((PetscObject) coordinatesB, "coordinates");CHKERRQ(ierr); 38807b0a1fcSMatthew G Knepley ierr = VecSetSizes(coordinatesB, coordSizeB, PETSC_DETERMINE);CHKERRQ(ierr); 3892eb5907fSJed Brown ierr = VecSetType(coordinatesB,VECSTANDARD);CHKERRQ(ierr); 39007b0a1fcSMatthew G Knepley ierr = VecGetArray(coordinatesA, &coordsA);CHKERRQ(ierr); 39107b0a1fcSMatthew G Knepley ierr = VecGetArray(coordinatesB, &coordsB);CHKERRQ(ierr); 39207b0a1fcSMatthew G Knepley for (v = 0; v < vEndB-vStartB; ++v) { 39307b0a1fcSMatthew G Knepley for (d = 0; d < spaceDim; ++d) { 39407b0a1fcSMatthew G Knepley coordsB[v*spaceDim+d] = coordsA[v*spaceDim+d]; 39507b0a1fcSMatthew G Knepley } 39607b0a1fcSMatthew G Knepley } 39707b0a1fcSMatthew G Knepley ierr = VecRestoreArray(coordinatesA, &coordsA);CHKERRQ(ierr); 39807b0a1fcSMatthew G Knepley ierr = VecRestoreArray(coordinatesB, &coordsB);CHKERRQ(ierr); 39907b0a1fcSMatthew G Knepley ierr = DMSetCoordinatesLocal(dmB, coordinatesB);CHKERRQ(ierr); 40007b0a1fcSMatthew G Knepley ierr = VecDestroy(&coordinatesB);CHKERRQ(ierr); 40107b0a1fcSMatthew G Knepley PetscFunctionReturn(0); 40207b0a1fcSMatthew G Knepley } 4035c386225SMatthew G. Knepley 4045c386225SMatthew G. Knepley #undef __FUNCT__ 4055c386225SMatthew G. Knepley #define __FUNCT__ "DMPlexCopyLabels" 4065c386225SMatthew G. Knepley /*@ 4075c386225SMatthew G. Knepley DMPlexCopyLabels - Copy labels from one mesh to another with a superset of the points 4085c386225SMatthew G. Knepley 4095c386225SMatthew G. Knepley Collective on DM 4105c386225SMatthew G. Knepley 4115c386225SMatthew G. Knepley Input Parameter: 4125c386225SMatthew G. Knepley . dmA - The DMPlex object with initial labels 4135c386225SMatthew G. Knepley 4145c386225SMatthew G. Knepley Output Parameter: 4155c386225SMatthew G. Knepley . dmB - The DMPlex object with copied labels 4165c386225SMatthew G. Knepley 4175c386225SMatthew G. Knepley Level: intermediate 4185c386225SMatthew G. Knepley 4195c386225SMatthew G. Knepley Note: This is typically used when interpolating or otherwise adding to a mesh 4205c386225SMatthew G. Knepley 4215c386225SMatthew G. Knepley .keywords: mesh 4225c386225SMatthew G. Knepley .seealso: DMCopyCoordinates(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM(), DMPlexGetCoordinateSection() 4235c386225SMatthew G. Knepley @*/ 4245c386225SMatthew G. Knepley PetscErrorCode DMPlexCopyLabels(DM dmA, DM dmB) 4255c386225SMatthew G. Knepley { 4265c386225SMatthew G. Knepley PetscInt numLabels, l; 4275c386225SMatthew G. Knepley PetscErrorCode ierr; 4285c386225SMatthew G. Knepley 4295c386225SMatthew G. Knepley PetscFunctionBegin; 4305c386225SMatthew G. Knepley if (dmA == dmB) PetscFunctionReturn(0); 4315c386225SMatthew G. Knepley ierr = DMPlexGetNumLabels(dmA, &numLabels);CHKERRQ(ierr); 4325c386225SMatthew G. Knepley for (l = 0; l < numLabels; ++l) { 4335c386225SMatthew G. Knepley DMLabel label, labelNew; 4345c386225SMatthew G. Knepley const char *name; 4355c386225SMatthew G. Knepley PetscBool flg; 4365c386225SMatthew G. Knepley 4375c386225SMatthew G. Knepley ierr = DMPlexGetLabelName(dmA, l, &name);CHKERRQ(ierr); 4385c386225SMatthew G. Knepley ierr = PetscStrcmp(name, "depth", &flg);CHKERRQ(ierr); 4395c386225SMatthew G. Knepley if (flg) continue; 4405c386225SMatthew G. Knepley ierr = DMPlexGetLabel(dmA, name, &label);CHKERRQ(ierr); 4415c386225SMatthew G. Knepley ierr = DMLabelDuplicate(label, &labelNew);CHKERRQ(ierr); 4425c386225SMatthew G. Knepley ierr = DMPlexAddLabel(dmB, labelNew);CHKERRQ(ierr); 4435c386225SMatthew G. Knepley } 4445c386225SMatthew G. Knepley PetscFunctionReturn(0); 4455c386225SMatthew G. Knepley } 446f5cedc29SMatthew G. Knepley 447f5cedc29SMatthew G. Knepley #undef __FUNCT__ 4484e3744c5SMatthew G. Knepley #define __FUNCT__ "DMPlexUninterpolate" 4494e3744c5SMatthew G. Knepley /*@ 4504e3744c5SMatthew G. Knepley DMPlexUninterpolate - Take in a mesh with all intermediate faces, edges, etc. and return a cell-vertex mesh 4514e3744c5SMatthew G. Knepley 4524e3744c5SMatthew G. Knepley Collective on DM 4534e3744c5SMatthew G. Knepley 4544e3744c5SMatthew G. Knepley Input Parameter: 4554e3744c5SMatthew G. Knepley . dm - The complete DMPlex object 4564e3744c5SMatthew G. Knepley 4574e3744c5SMatthew G. Knepley Output Parameter: 4584e3744c5SMatthew G. Knepley . dmUnint - The DMPlex object with only cells and vertices 4594e3744c5SMatthew G. Knepley 4604e3744c5SMatthew G. Knepley Level: intermediate 4614e3744c5SMatthew G. Knepley 4624e3744c5SMatthew G. Knepley .keywords: mesh 4634e3744c5SMatthew G. Knepley .seealso: DMPlexInterpolate(), DMPlexCreateFromCellList() 4644e3744c5SMatthew G. Knepley @*/ 4654e3744c5SMatthew G. Knepley PetscErrorCode DMPlexUninterpolate(DM dm, DM *dmUnint) 4664e3744c5SMatthew G. Knepley { 4674e3744c5SMatthew G. Knepley DM udm; 468595d4abbSMatthew G. Knepley PetscInt dim, vStart, vEnd, cStart, cEnd, c, maxConeSize = 0, *cone; 4694e3744c5SMatthew G. Knepley PetscErrorCode ierr; 4704e3744c5SMatthew G. Knepley 4714e3744c5SMatthew G. Knepley PetscFunctionBegin; 4724e3744c5SMatthew G. Knepley ierr = DMPlexGetDimension(dm, &dim);CHKERRQ(ierr); 4734e3744c5SMatthew G. Knepley if (dim <= 1) { 4744e3744c5SMatthew G. Knepley ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr); 475595d4abbSMatthew G. Knepley *dmUnint = dm; 476595d4abbSMatthew G. Knepley PetscFunctionReturn(0); 4774e3744c5SMatthew G. Knepley } 478595d4abbSMatthew G. Knepley ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr); 479595d4abbSMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 4804e3744c5SMatthew G. Knepley ierr = DMCreate(PetscObjectComm((PetscObject) dm), &udm);CHKERRQ(ierr); 4814e3744c5SMatthew G. Knepley ierr = DMSetType(udm, DMPLEX);CHKERRQ(ierr); 4824e3744c5SMatthew G. Knepley ierr = DMPlexSetDimension(udm, dim);CHKERRQ(ierr); 4834e3744c5SMatthew G. Knepley ierr = DMPlexSetChart(udm, cStart, vEnd);CHKERRQ(ierr); 4844e3744c5SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 485595d4abbSMatthew G. Knepley PetscInt *closure = NULL, closureSize, cl, coneSize = 0; 4864e3744c5SMatthew G. Knepley 4874e3744c5SMatthew G. Knepley ierr = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr); 4884e3744c5SMatthew G. Knepley for (cl = 0; cl < closureSize*2; cl += 2) { 4894e3744c5SMatthew G. Knepley const PetscInt p = closure[cl]; 4904e3744c5SMatthew G. Knepley 4914e3744c5SMatthew G. Knepley if ((p >= vStart) && (p < vEnd)) ++coneSize; 4924e3744c5SMatthew G. Knepley } 4934e3744c5SMatthew G. Knepley ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr); 4944e3744c5SMatthew G. Knepley ierr = DMPlexSetConeSize(udm, c, coneSize);CHKERRQ(ierr); 495595d4abbSMatthew G. Knepley maxConeSize = PetscMax(maxConeSize, coneSize); 4964e3744c5SMatthew G. Knepley } 4974e3744c5SMatthew G. Knepley ierr = DMSetUp(udm);CHKERRQ(ierr); 4984e3744c5SMatthew G. Knepley ierr = PetscMalloc(maxConeSize * sizeof(PetscInt), &cone);CHKERRQ(ierr); 4994e3744c5SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 500595d4abbSMatthew G. Knepley PetscInt *closure = NULL, closureSize, cl, coneSize = 0; 5014e3744c5SMatthew G. Knepley 5024e3744c5SMatthew G. Knepley ierr = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr); 5034e3744c5SMatthew G. Knepley for (cl = 0; cl < closureSize*2; cl += 2) { 5044e3744c5SMatthew G. Knepley const PetscInt p = closure[cl]; 5054e3744c5SMatthew G. Knepley 5064e3744c5SMatthew G. Knepley if ((p >= vStart) && (p < vEnd)) cone[coneSize++] = p; 5074e3744c5SMatthew G. Knepley } 5084e3744c5SMatthew G. Knepley ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr); 5094e3744c5SMatthew G. Knepley ierr = DMPlexSetCone(udm, c, cone);CHKERRQ(ierr); 5104e3744c5SMatthew G. Knepley } 5114e3744c5SMatthew G. Knepley ierr = PetscFree(cone);CHKERRQ(ierr); 5124e3744c5SMatthew G. Knepley ierr = DMPlexSymmetrize(udm);CHKERRQ(ierr); 5134e3744c5SMatthew G. Knepley ierr = DMPlexStratify(udm);CHKERRQ(ierr); 5144e3744c5SMatthew G. Knepley *dmUnint = udm; 5154e3744c5SMatthew G. Knepley PetscFunctionReturn(0); 5164e3744c5SMatthew G. Knepley } 517