1af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h> /*I "petscdmplex.h" I*/ 29d150b73SToby Isaac #include <petsc/private/petscfeimpl.h> /*I "petscfe.h" I*/ 39d150b73SToby Isaac #include <petscblaslapack.h> 4af74b616SDave May #include <petsctime.h> 5ccd2543fSMatthew G Knepley 6fea14342SMatthew G. Knepley static PetscErrorCode DMPlexGetLineIntersection_2D_Internal(const PetscReal segmentA[], const PetscReal segmentB[], PetscReal intersection[], PetscBool *hasIntersection) 7fea14342SMatthew G. Knepley { 8fea14342SMatthew G. Knepley const PetscReal p0_x = segmentA[0*2+0]; 9fea14342SMatthew G. Knepley const PetscReal p0_y = segmentA[0*2+1]; 10fea14342SMatthew G. Knepley const PetscReal p1_x = segmentA[1*2+0]; 11fea14342SMatthew G. Knepley const PetscReal p1_y = segmentA[1*2+1]; 12fea14342SMatthew G. Knepley const PetscReal p2_x = segmentB[0*2+0]; 13fea14342SMatthew G. Knepley const PetscReal p2_y = segmentB[0*2+1]; 14fea14342SMatthew G. Knepley const PetscReal p3_x = segmentB[1*2+0]; 15fea14342SMatthew G. Knepley const PetscReal p3_y = segmentB[1*2+1]; 16fea14342SMatthew G. Knepley const PetscReal s1_x = p1_x - p0_x; 17fea14342SMatthew G. Knepley const PetscReal s1_y = p1_y - p0_y; 18fea14342SMatthew G. Knepley const PetscReal s2_x = p3_x - p2_x; 19fea14342SMatthew G. Knepley const PetscReal s2_y = p3_y - p2_y; 20fea14342SMatthew G. Knepley const PetscReal denom = (-s2_x * s1_y + s1_x * s2_y); 21fea14342SMatthew G. Knepley 22fea14342SMatthew G. Knepley PetscFunctionBegin; 23fea14342SMatthew G. Knepley *hasIntersection = PETSC_FALSE; 24fea14342SMatthew G. Knepley /* Non-parallel lines */ 25fea14342SMatthew G. Knepley if (denom != 0.0) { 26fea14342SMatthew G. Knepley const PetscReal s = (-s1_y * (p0_x - p2_x) + s1_x * (p0_y - p2_y)) / denom; 27fea14342SMatthew G. Knepley const PetscReal t = ( s2_x * (p0_y - p2_y) - s2_y * (p0_x - p2_x)) / denom; 28fea14342SMatthew G. Knepley 29fea14342SMatthew G. Knepley if (s >= 0 && s <= 1 && t >= 0 && t <= 1) { 30fea14342SMatthew G. Knepley *hasIntersection = PETSC_TRUE; 31fea14342SMatthew G. Knepley if (intersection) { 32fea14342SMatthew G. Knepley intersection[0] = p0_x + (t * s1_x); 33fea14342SMatthew G. Knepley intersection[1] = p0_y + (t * s1_y); 34fea14342SMatthew G. Knepley } 35fea14342SMatthew G. Knepley } 36fea14342SMatthew G. Knepley } 37fea14342SMatthew G. Knepley PetscFunctionReturn(0); 38fea14342SMatthew G. Knepley } 39fea14342SMatthew G. Knepley 40ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexLocatePoint_Simplex_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell) 41ccd2543fSMatthew G Knepley { 42ccd2543fSMatthew G Knepley const PetscInt embedDim = 2; 43f5ebc837SMatthew G. Knepley const PetscReal eps = PETSC_SQRT_MACHINE_EPSILON; 44ccd2543fSMatthew G Knepley PetscReal x = PetscRealPart(point[0]); 45ccd2543fSMatthew G Knepley PetscReal y = PetscRealPart(point[1]); 46ccd2543fSMatthew G Knepley PetscReal v0[2], J[4], invJ[4], detJ; 47ccd2543fSMatthew G Knepley PetscReal xi, eta; 48ccd2543fSMatthew G Knepley PetscErrorCode ierr; 49ccd2543fSMatthew G Knepley 50ccd2543fSMatthew G Knepley PetscFunctionBegin; 518e0841e0SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 52ccd2543fSMatthew G Knepley xi = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]); 53ccd2543fSMatthew G Knepley eta = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]); 54ccd2543fSMatthew G Knepley 55f5ebc837SMatthew G. Knepley if ((xi >= -eps) && (eta >= -eps) && (xi + eta <= 2.0+eps)) *cell = c; 56c1496c66SMatthew G. Knepley else *cell = DMLOCATEPOINT_POINT_NOT_FOUND; 57ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 58ccd2543fSMatthew G Knepley } 59ccd2543fSMatthew G Knepley 6062a38674SMatthew G. Knepley static PetscErrorCode DMPlexClosestPoint_Simplex_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscReal cpoint[]) 6162a38674SMatthew G. Knepley { 6262a38674SMatthew G. Knepley const PetscInt embedDim = 2; 6362a38674SMatthew G. Knepley PetscReal x = PetscRealPart(point[0]); 6462a38674SMatthew G. Knepley PetscReal y = PetscRealPart(point[1]); 6562a38674SMatthew G. Knepley PetscReal v0[2], J[4], invJ[4], detJ; 6662a38674SMatthew G. Knepley PetscReal xi, eta, r; 6762a38674SMatthew G. Knepley PetscErrorCode ierr; 6862a38674SMatthew G. Knepley 6962a38674SMatthew G. Knepley PetscFunctionBegin; 7062a38674SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 7162a38674SMatthew G. Knepley xi = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]); 7262a38674SMatthew G. Knepley eta = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]); 7362a38674SMatthew G. Knepley 7462a38674SMatthew G. Knepley xi = PetscMax(xi, 0.0); 7562a38674SMatthew G. Knepley eta = PetscMax(eta, 0.0); 7662a38674SMatthew G. Knepley if (xi + eta > 2.0) { 7762a38674SMatthew G. Knepley r = (xi + eta)/2.0; 7862a38674SMatthew G. Knepley xi /= r; 7962a38674SMatthew G. Knepley eta /= r; 8062a38674SMatthew G. Knepley } 8162a38674SMatthew G. Knepley cpoint[0] = J[0*embedDim+0]*xi + J[0*embedDim+1]*eta + v0[0]; 8262a38674SMatthew G. Knepley cpoint[1] = J[1*embedDim+0]*xi + J[1*embedDim+1]*eta + v0[1]; 8362a38674SMatthew G. Knepley PetscFunctionReturn(0); 8462a38674SMatthew G. Knepley } 8562a38674SMatthew G. Knepley 86ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexLocatePoint_General_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell) 87ccd2543fSMatthew G Knepley { 88ccd2543fSMatthew G Knepley PetscSection coordSection; 89ccd2543fSMatthew G Knepley Vec coordsLocal; 90a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 91ccd2543fSMatthew G Knepley const PetscInt faces[8] = {0, 1, 1, 2, 2, 3, 3, 0}; 92ccd2543fSMatthew G Knepley PetscReal x = PetscRealPart(point[0]); 93ccd2543fSMatthew G Knepley PetscReal y = PetscRealPart(point[1]); 94ccd2543fSMatthew G Knepley PetscInt crossings = 0, f; 95ccd2543fSMatthew G Knepley PetscErrorCode ierr; 96ccd2543fSMatthew G Knepley 97ccd2543fSMatthew G Knepley PetscFunctionBegin; 98ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr); 9969d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 100ccd2543fSMatthew G Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr); 101ccd2543fSMatthew G Knepley for (f = 0; f < 4; ++f) { 102ccd2543fSMatthew G Knepley PetscReal x_i = PetscRealPart(coords[faces[2*f+0]*2+0]); 103ccd2543fSMatthew G Knepley PetscReal y_i = PetscRealPart(coords[faces[2*f+0]*2+1]); 104ccd2543fSMatthew G Knepley PetscReal x_j = PetscRealPart(coords[faces[2*f+1]*2+0]); 105ccd2543fSMatthew G Knepley PetscReal y_j = PetscRealPart(coords[faces[2*f+1]*2+1]); 106ccd2543fSMatthew G Knepley PetscReal slope = (y_j - y_i) / (x_j - x_i); 107ccd2543fSMatthew G Knepley PetscBool cond1 = (x_i <= x) && (x < x_j) ? PETSC_TRUE : PETSC_FALSE; 108ccd2543fSMatthew G Knepley PetscBool cond2 = (x_j <= x) && (x < x_i) ? PETSC_TRUE : PETSC_FALSE; 109ccd2543fSMatthew G Knepley PetscBool above = (y < slope * (x - x_i) + y_i) ? PETSC_TRUE : PETSC_FALSE; 110ccd2543fSMatthew G Knepley if ((cond1 || cond2) && above) ++crossings; 111ccd2543fSMatthew G Knepley } 112ccd2543fSMatthew G Knepley if (crossings % 2) *cell = c; 113c1496c66SMatthew G. Knepley else *cell = DMLOCATEPOINT_POINT_NOT_FOUND; 114ccd2543fSMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr); 115ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 116ccd2543fSMatthew G Knepley } 117ccd2543fSMatthew G Knepley 118ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexLocatePoint_Simplex_3D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell) 119ccd2543fSMatthew G Knepley { 120ccd2543fSMatthew G Knepley const PetscInt embedDim = 3; 121ccd2543fSMatthew G Knepley PetscReal v0[3], J[9], invJ[9], detJ; 122ccd2543fSMatthew G Knepley PetscReal x = PetscRealPart(point[0]); 123ccd2543fSMatthew G Knepley PetscReal y = PetscRealPart(point[1]); 124ccd2543fSMatthew G Knepley PetscReal z = PetscRealPart(point[2]); 125ccd2543fSMatthew G Knepley PetscReal xi, eta, zeta; 126ccd2543fSMatthew G Knepley PetscErrorCode ierr; 127ccd2543fSMatthew G Knepley 128ccd2543fSMatthew G Knepley PetscFunctionBegin; 1298e0841e0SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 130ccd2543fSMatthew G Knepley xi = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]) + invJ[0*embedDim+2]*(z - v0[2]); 131ccd2543fSMatthew G Knepley eta = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]) + invJ[1*embedDim+2]*(z - v0[2]); 132ccd2543fSMatthew G Knepley zeta = invJ[2*embedDim+0]*(x - v0[0]) + invJ[2*embedDim+1]*(y - v0[1]) + invJ[2*embedDim+2]*(z - v0[2]); 133ccd2543fSMatthew G Knepley 134ccd2543fSMatthew G Knepley if ((xi >= 0.0) && (eta >= 0.0) && (zeta >= 0.0) && (xi + eta + zeta <= 2.0)) *cell = c; 135c1496c66SMatthew G. Knepley else *cell = DMLOCATEPOINT_POINT_NOT_FOUND; 136ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 137ccd2543fSMatthew G Knepley } 138ccd2543fSMatthew G Knepley 139ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexLocatePoint_General_3D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell) 140ccd2543fSMatthew G Knepley { 141ccd2543fSMatthew G Knepley PetscSection coordSection; 142ccd2543fSMatthew G Knepley Vec coordsLocal; 143872a9804SMatthew G. Knepley PetscScalar *coords = NULL; 144fb150da6SMatthew G. Knepley const PetscInt faces[24] = {0, 3, 2, 1, 5, 4, 7, 6, 3, 0, 4, 5, 145fb150da6SMatthew G. Knepley 1, 2, 6, 7, 3, 5, 6, 2, 0, 1, 7, 4}; 146ccd2543fSMatthew G Knepley PetscBool found = PETSC_TRUE; 147ccd2543fSMatthew G Knepley PetscInt f; 148ccd2543fSMatthew G Knepley PetscErrorCode ierr; 149ccd2543fSMatthew G Knepley 150ccd2543fSMatthew G Knepley PetscFunctionBegin; 151ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr); 15269d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 153ccd2543fSMatthew G Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr); 154ccd2543fSMatthew G Knepley for (f = 0; f < 6; ++f) { 155ccd2543fSMatthew G Knepley /* Check the point is under plane */ 156ccd2543fSMatthew G Knepley /* Get face normal */ 157ccd2543fSMatthew G Knepley PetscReal v_i[3]; 158ccd2543fSMatthew G Knepley PetscReal v_j[3]; 159ccd2543fSMatthew G Knepley PetscReal normal[3]; 160ccd2543fSMatthew G Knepley PetscReal pp[3]; 161ccd2543fSMatthew G Knepley PetscReal dot; 162ccd2543fSMatthew G Knepley 163ccd2543fSMatthew G Knepley v_i[0] = PetscRealPart(coords[faces[f*4+3]*3+0]-coords[faces[f*4+0]*3+0]); 164ccd2543fSMatthew G Knepley v_i[1] = PetscRealPart(coords[faces[f*4+3]*3+1]-coords[faces[f*4+0]*3+1]); 165ccd2543fSMatthew G Knepley v_i[2] = PetscRealPart(coords[faces[f*4+3]*3+2]-coords[faces[f*4+0]*3+2]); 166ccd2543fSMatthew G Knepley v_j[0] = PetscRealPart(coords[faces[f*4+1]*3+0]-coords[faces[f*4+0]*3+0]); 167ccd2543fSMatthew G Knepley v_j[1] = PetscRealPart(coords[faces[f*4+1]*3+1]-coords[faces[f*4+0]*3+1]); 168ccd2543fSMatthew G Knepley v_j[2] = PetscRealPart(coords[faces[f*4+1]*3+2]-coords[faces[f*4+0]*3+2]); 169ccd2543fSMatthew G Knepley normal[0] = v_i[1]*v_j[2] - v_i[2]*v_j[1]; 170ccd2543fSMatthew G Knepley normal[1] = v_i[2]*v_j[0] - v_i[0]*v_j[2]; 171ccd2543fSMatthew G Knepley normal[2] = v_i[0]*v_j[1] - v_i[1]*v_j[0]; 172ccd2543fSMatthew G Knepley pp[0] = PetscRealPart(coords[faces[f*4+0]*3+0] - point[0]); 173ccd2543fSMatthew G Knepley pp[1] = PetscRealPart(coords[faces[f*4+0]*3+1] - point[1]); 174ccd2543fSMatthew G Knepley pp[2] = PetscRealPart(coords[faces[f*4+0]*3+2] - point[2]); 175ccd2543fSMatthew G Knepley dot = normal[0]*pp[0] + normal[1]*pp[1] + normal[2]*pp[2]; 176ccd2543fSMatthew G Knepley 177ccd2543fSMatthew G Knepley /* Check that projected point is in face (2D location problem) */ 178ccd2543fSMatthew G Knepley if (dot < 0.0) { 179ccd2543fSMatthew G Knepley found = PETSC_FALSE; 180ccd2543fSMatthew G Knepley break; 181ccd2543fSMatthew G Knepley } 182ccd2543fSMatthew G Knepley } 183ccd2543fSMatthew G Knepley if (found) *cell = c; 184c1496c66SMatthew G. Knepley else *cell = DMLOCATEPOINT_POINT_NOT_FOUND; 185ccd2543fSMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr); 186ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 187ccd2543fSMatthew G Knepley } 188ccd2543fSMatthew G Knepley 189c4eade1cSMatthew G. Knepley static PetscErrorCode PetscGridHashInitialize_Internal(PetscGridHash box, PetscInt dim, const PetscScalar point[]) 190c4eade1cSMatthew G. Knepley { 191c4eade1cSMatthew G. Knepley PetscInt d; 192c4eade1cSMatthew G. Knepley 193c4eade1cSMatthew G. Knepley PetscFunctionBegin; 194c4eade1cSMatthew G. Knepley box->dim = dim; 195c4eade1cSMatthew G. Knepley for (d = 0; d < dim; ++d) box->lower[d] = box->upper[d] = PetscRealPart(point[d]); 196c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 197c4eade1cSMatthew G. Knepley } 198c4eade1cSMatthew G. Knepley 199c4eade1cSMatthew G. Knepley PetscErrorCode PetscGridHashCreate(MPI_Comm comm, PetscInt dim, const PetscScalar point[], PetscGridHash *box) 200c4eade1cSMatthew G. Knepley { 201c4eade1cSMatthew G. Knepley PetscErrorCode ierr; 202c4eade1cSMatthew G. Knepley 203c4eade1cSMatthew G. Knepley PetscFunctionBegin; 204c4eade1cSMatthew G. Knepley ierr = PetscMalloc1(1, box);CHKERRQ(ierr); 205c4eade1cSMatthew G. Knepley ierr = PetscGridHashInitialize_Internal(*box, dim, point);CHKERRQ(ierr); 206c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 207c4eade1cSMatthew G. Knepley } 208c4eade1cSMatthew G. Knepley 209c4eade1cSMatthew G. Knepley PetscErrorCode PetscGridHashEnlarge(PetscGridHash box, const PetscScalar point[]) 210c4eade1cSMatthew G. Knepley { 211c4eade1cSMatthew G. Knepley PetscInt d; 212c4eade1cSMatthew G. Knepley 213c4eade1cSMatthew G. Knepley PetscFunctionBegin; 214c4eade1cSMatthew G. Knepley for (d = 0; d < box->dim; ++d) { 215c4eade1cSMatthew G. Knepley box->lower[d] = PetscMin(box->lower[d], PetscRealPart(point[d])); 216c4eade1cSMatthew G. Knepley box->upper[d] = PetscMax(box->upper[d], PetscRealPart(point[d])); 217c4eade1cSMatthew G. Knepley } 218c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 219c4eade1cSMatthew G. Knepley } 220c4eade1cSMatthew G. Knepley 22162a38674SMatthew G. Knepley /* 22262a38674SMatthew G. Knepley PetscGridHashSetGrid - Divide the grid into boxes 22362a38674SMatthew G. Knepley 22462a38674SMatthew G. Knepley Not collective 22562a38674SMatthew G. Knepley 22662a38674SMatthew G. Knepley Input Parameters: 22762a38674SMatthew G. Knepley + box - The grid hash object 22862a38674SMatthew G. Knepley . n - The number of boxes in each dimension, or PETSC_DETERMINE 22962a38674SMatthew G. Knepley - h - The box size in each dimension, only used if n[d] == PETSC_DETERMINE 23062a38674SMatthew G. Knepley 23162a38674SMatthew G. Knepley Level: developer 23262a38674SMatthew G. Knepley 23362a38674SMatthew G. Knepley .seealso: PetscGridHashCreate() 23462a38674SMatthew G. Knepley */ 235c4eade1cSMatthew G. Knepley PetscErrorCode PetscGridHashSetGrid(PetscGridHash box, const PetscInt n[], const PetscReal h[]) 236c4eade1cSMatthew G. Knepley { 237c4eade1cSMatthew G. Knepley PetscInt d; 238c4eade1cSMatthew G. Knepley 239c4eade1cSMatthew G. Knepley PetscFunctionBegin; 240c4eade1cSMatthew G. Knepley for (d = 0; d < box->dim; ++d) { 241c4eade1cSMatthew G. Knepley box->extent[d] = box->upper[d] - box->lower[d]; 242c4eade1cSMatthew G. Knepley if (n[d] == PETSC_DETERMINE) { 243c4eade1cSMatthew G. Knepley box->h[d] = h[d]; 244c4eade1cSMatthew G. Knepley box->n[d] = PetscCeilReal(box->extent[d]/h[d]); 245c4eade1cSMatthew G. Knepley } else { 246c4eade1cSMatthew G. Knepley box->n[d] = n[d]; 247c4eade1cSMatthew G. Knepley box->h[d] = box->extent[d]/n[d]; 248c4eade1cSMatthew G. Knepley } 249c4eade1cSMatthew G. Knepley } 250c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 251c4eade1cSMatthew G. Knepley } 252c4eade1cSMatthew G. Knepley 25362a38674SMatthew G. Knepley /* 25462a38674SMatthew G. Knepley PetscGridHashGetEnclosingBox - Find the grid boxes containing each input point 25562a38674SMatthew G. Knepley 25662a38674SMatthew G. Knepley Not collective 25762a38674SMatthew G. Knepley 25862a38674SMatthew G. Knepley Input Parameters: 25962a38674SMatthew G. Knepley + box - The grid hash object 26062a38674SMatthew G. Knepley . numPoints - The number of input points 26162a38674SMatthew G. Knepley - points - The input point coordinates 26262a38674SMatthew G. Knepley 26362a38674SMatthew G. Knepley Output Parameters: 26462a38674SMatthew G. Knepley + dboxes - An array of numPoints*dim integers expressing the enclosing box as (i_0, i_1, ..., i_dim) 26562a38674SMatthew G. Knepley - boxes - An array of numPoints integers expressing the enclosing box as single number, or NULL 26662a38674SMatthew G. Knepley 26762a38674SMatthew G. Knepley Level: developer 26862a38674SMatthew G. Knepley 26962a38674SMatthew G. Knepley .seealso: PetscGridHashCreate() 27062a38674SMatthew G. Knepley */ 2711c6dfc3eSMatthew G. Knepley PetscErrorCode PetscGridHashGetEnclosingBox(PetscGridHash box, PetscInt numPoints, const PetscScalar points[], PetscInt dboxes[], PetscInt boxes[]) 272c4eade1cSMatthew G. Knepley { 273c4eade1cSMatthew G. Knepley const PetscReal *lower = box->lower; 274c4eade1cSMatthew G. Knepley const PetscReal *upper = box->upper; 275c4eade1cSMatthew G. Knepley const PetscReal *h = box->h; 276c4eade1cSMatthew G. Knepley const PetscInt *n = box->n; 277c4eade1cSMatthew G. Knepley const PetscInt dim = box->dim; 278c4eade1cSMatthew G. Knepley PetscInt d, p; 279c4eade1cSMatthew G. Knepley 280c4eade1cSMatthew G. Knepley PetscFunctionBegin; 281c4eade1cSMatthew G. Knepley for (p = 0; p < numPoints; ++p) { 282c4eade1cSMatthew G. Knepley for (d = 0; d < dim; ++d) { 2831c6dfc3eSMatthew G. Knepley PetscInt dbox = PetscFloorReal((PetscRealPart(points[p*dim+d]) - lower[d])/h[d]); 284c4eade1cSMatthew G. Knepley 2851c6dfc3eSMatthew G. Knepley if (dbox == n[d] && PetscAbsReal(PetscRealPart(points[p*dim+d]) - upper[d]) < 1.0e-9) dbox = n[d]-1; 2862a705cacSMatthew G. Knepley if (dbox == -1 && PetscAbsReal(PetscRealPart(points[p*dim+d]) - lower[d]) < 1.0e-9) dbox = 0; 287c4eade1cSMatthew G. Knepley if (dbox < 0 || dbox >= n[d]) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Input point %d (%g, %g, %g) is outside of our bounding box", 2881c6dfc3eSMatthew G. Knepley p, PetscRealPart(points[p*dim+0]), dim > 1 ? PetscRealPart(points[p*dim+1]) : 0.0, dim > 2 ? PetscRealPart(points[p*dim+2]) : 0.0); 289c4eade1cSMatthew G. Knepley dboxes[p*dim+d] = dbox; 290c4eade1cSMatthew G. Knepley } 291c4eade1cSMatthew G. Knepley if (boxes) for (d = 1, boxes[p] = dboxes[p*dim]; d < dim; ++d) boxes[p] += dboxes[p*dim+d]*n[d-1]; 292c4eade1cSMatthew G. Knepley } 293c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 294c4eade1cSMatthew G. Knepley } 295c4eade1cSMatthew G. Knepley 296af74b616SDave May /* 297af74b616SDave May PetscGridHashGetEnclosingBoxQuery - Find the grid boxes containing each input point 298af74b616SDave May 299af74b616SDave May Not collective 300af74b616SDave May 301af74b616SDave May Input Parameters: 302af74b616SDave May + box - The grid hash object 303af74b616SDave May . numPoints - The number of input points 304af74b616SDave May - points - The input point coordinates 305af74b616SDave May 306af74b616SDave May Output Parameters: 307af74b616SDave May + dboxes - An array of numPoints*dim integers expressing the enclosing box as (i_0, i_1, ..., i_dim) 308af74b616SDave May . boxes - An array of numPoints integers expressing the enclosing box as single number, or NULL 309af74b616SDave May - found - Flag indicating if point was located within a box 310af74b616SDave May 311af74b616SDave May Level: developer 312af74b616SDave May 313af74b616SDave May .seealso: PetscGridHashGetEnclosingBox() 314af74b616SDave May */ 315af74b616SDave May PetscErrorCode PetscGridHashGetEnclosingBoxQuery(PetscGridHash box, PetscInt numPoints, const PetscScalar points[], PetscInt dboxes[], PetscInt boxes[],PetscBool *found) 316af74b616SDave May { 317af74b616SDave May const PetscReal *lower = box->lower; 318af74b616SDave May const PetscReal *upper = box->upper; 319af74b616SDave May const PetscReal *h = box->h; 320af74b616SDave May const PetscInt *n = box->n; 321af74b616SDave May const PetscInt dim = box->dim; 322af74b616SDave May PetscInt d, p; 323af74b616SDave May 324af74b616SDave May PetscFunctionBegin; 325af74b616SDave May *found = PETSC_FALSE; 326af74b616SDave May for (p = 0; p < numPoints; ++p) { 327af74b616SDave May for (d = 0; d < dim; ++d) { 328af74b616SDave May PetscInt dbox = PetscFloorReal((PetscRealPart(points[p*dim+d]) - lower[d])/h[d]); 329af74b616SDave May 330af74b616SDave May if (dbox == n[d] && PetscAbsReal(PetscRealPart(points[p*dim+d]) - upper[d]) < 1.0e-9) dbox = n[d]-1; 331af74b616SDave May if (dbox < 0 || dbox >= n[d]) { 332af74b616SDave May PetscFunctionReturn(0); 333af74b616SDave May } 334af74b616SDave May dboxes[p*dim+d] = dbox; 335af74b616SDave May } 336af74b616SDave May if (boxes) for (d = 1, boxes[p] = dboxes[p*dim]; d < dim; ++d) boxes[p] += dboxes[p*dim+d]*n[d-1]; 337af74b616SDave May } 338af74b616SDave May *found = PETSC_TRUE; 339af74b616SDave May PetscFunctionReturn(0); 340af74b616SDave May } 341af74b616SDave May 342c4eade1cSMatthew G. Knepley PetscErrorCode PetscGridHashDestroy(PetscGridHash *box) 343c4eade1cSMatthew G. Knepley { 344c4eade1cSMatthew G. Knepley PetscErrorCode ierr; 345c4eade1cSMatthew G. Knepley 346c4eade1cSMatthew G. Knepley PetscFunctionBegin; 347c4eade1cSMatthew G. Knepley if (*box) { 348c4eade1cSMatthew G. Knepley ierr = PetscSectionDestroy(&(*box)->cellSection);CHKERRQ(ierr); 349c4eade1cSMatthew G. Knepley ierr = ISDestroy(&(*box)->cells);CHKERRQ(ierr); 350c4eade1cSMatthew G. Knepley ierr = DMLabelDestroy(&(*box)->cellsSparse);CHKERRQ(ierr); 351c4eade1cSMatthew G. Knepley } 352c4eade1cSMatthew G. Knepley ierr = PetscFree(*box);CHKERRQ(ierr); 353c4eade1cSMatthew G. Knepley PetscFunctionReturn(0); 354c4eade1cSMatthew G. Knepley } 355c4eade1cSMatthew G. Knepley 356cafe43deSMatthew G. Knepley PetscErrorCode DMPlexLocatePoint_Internal(DM dm, PetscInt dim, const PetscScalar point[], PetscInt cellStart, PetscInt *cell) 357cafe43deSMatthew G. Knepley { 358cafe43deSMatthew G. Knepley PetscInt coneSize; 359cafe43deSMatthew G. Knepley PetscErrorCode ierr; 360cafe43deSMatthew G. Knepley 361cafe43deSMatthew G. Knepley PetscFunctionBegin; 362cafe43deSMatthew G. Knepley switch (dim) { 363cafe43deSMatthew G. Knepley case 2: 364cafe43deSMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cellStart, &coneSize);CHKERRQ(ierr); 365cafe43deSMatthew G. Knepley switch (coneSize) { 366cafe43deSMatthew G. Knepley case 3: 367cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_Simplex_2D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr); 368cafe43deSMatthew G. Knepley break; 369cafe43deSMatthew G. Knepley case 4: 370cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_General_2D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr); 371cafe43deSMatthew G. Knepley break; 372cafe43deSMatthew G. Knepley default: 373cafe43deSMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for cell with cone size %D", coneSize); 374cafe43deSMatthew G. Knepley } 375cafe43deSMatthew G. Knepley break; 376cafe43deSMatthew G. Knepley case 3: 377cafe43deSMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cellStart, &coneSize);CHKERRQ(ierr); 378cafe43deSMatthew G. Knepley switch (coneSize) { 379cafe43deSMatthew G. Knepley case 4: 380cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_Simplex_3D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr); 381cafe43deSMatthew G. Knepley break; 382cafe43deSMatthew G. Knepley case 6: 383cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_General_3D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr); 384cafe43deSMatthew G. Knepley break; 385cafe43deSMatthew G. Knepley default: 386cafe43deSMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for cell with cone size %D", coneSize); 387cafe43deSMatthew G. Knepley } 388cafe43deSMatthew G. Knepley break; 389cafe43deSMatthew G. Knepley default: 390cafe43deSMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for mesh dimension %D", dim); 391cafe43deSMatthew G. Knepley } 392cafe43deSMatthew G. Knepley PetscFunctionReturn(0); 393cafe43deSMatthew G. Knepley } 394cafe43deSMatthew G. Knepley 39562a38674SMatthew G. Knepley /* 39662a38674SMatthew G. Knepley DMPlexClosestPoint_Internal - Returns the closest point in the cell to the given point 39762a38674SMatthew G. Knepley */ 39862a38674SMatthew G. Knepley PetscErrorCode DMPlexClosestPoint_Internal(DM dm, PetscInt dim, const PetscScalar point[], PetscInt cell, PetscReal cpoint[]) 39962a38674SMatthew G. Knepley { 40062a38674SMatthew G. Knepley PetscInt coneSize; 40162a38674SMatthew G. Knepley PetscErrorCode ierr; 40262a38674SMatthew G. Knepley 40362a38674SMatthew G. Knepley PetscFunctionBegin; 40462a38674SMatthew G. Knepley switch (dim) { 40562a38674SMatthew G. Knepley case 2: 40662a38674SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr); 40762a38674SMatthew G. Knepley switch (coneSize) { 40862a38674SMatthew G. Knepley case 3: 40962a38674SMatthew G. Knepley ierr = DMPlexClosestPoint_Simplex_2D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr); 41062a38674SMatthew G. Knepley break; 41162a38674SMatthew G. Knepley #if 0 41262a38674SMatthew G. Knepley case 4: 41362a38674SMatthew G. Knepley ierr = DMPlexClosestPoint_General_2D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr); 41462a38674SMatthew G. Knepley break; 41562a38674SMatthew G. Knepley #endif 41662a38674SMatthew G. Knepley default: 41762a38674SMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for cell with cone size %D", coneSize); 41862a38674SMatthew G. Knepley } 41962a38674SMatthew G. Knepley break; 42062a38674SMatthew G. Knepley #if 0 42162a38674SMatthew G. Knepley case 3: 42262a38674SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr); 42362a38674SMatthew G. Knepley switch (coneSize) { 42462a38674SMatthew G. Knepley case 4: 42562a38674SMatthew G. Knepley ierr = DMPlexClosestPoint_Simplex_3D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr); 42662a38674SMatthew G. Knepley break; 42762a38674SMatthew G. Knepley case 6: 42862a38674SMatthew G. Knepley ierr = DMPlexClosestPoint_General_3D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr); 42962a38674SMatthew G. Knepley break; 43062a38674SMatthew G. Knepley default: 43162a38674SMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for cell with cone size %D", coneSize); 43262a38674SMatthew G. Knepley } 43362a38674SMatthew G. Knepley break; 43462a38674SMatthew G. Knepley #endif 43562a38674SMatthew G. Knepley default: 43662a38674SMatthew G. Knepley SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for mesh dimension %D", dim); 43762a38674SMatthew G. Knepley } 43862a38674SMatthew G. Knepley PetscFunctionReturn(0); 43962a38674SMatthew G. Knepley } 44062a38674SMatthew G. Knepley 44162a38674SMatthew G. Knepley /* 44262a38674SMatthew G. Knepley DMPlexComputeGridHash_Internal - Create a grid hash structure covering the Plex 44362a38674SMatthew G. Knepley 44462a38674SMatthew G. Knepley Collective on DM 44562a38674SMatthew G. Knepley 44662a38674SMatthew G. Knepley Input Parameter: 44762a38674SMatthew G. Knepley . dm - The Plex 44862a38674SMatthew G. Knepley 44962a38674SMatthew G. Knepley Output Parameter: 45062a38674SMatthew G. Knepley . localBox - The grid hash object 45162a38674SMatthew G. Knepley 45262a38674SMatthew G. Knepley Level: developer 45362a38674SMatthew G. Knepley 45462a38674SMatthew G. Knepley .seealso: PetscGridHashCreate(), PetscGridHashGetEnclosingBox() 45562a38674SMatthew G. Knepley */ 456cafe43deSMatthew G. Knepley PetscErrorCode DMPlexComputeGridHash_Internal(DM dm, PetscGridHash *localBox) 457cafe43deSMatthew G. Knepley { 458cafe43deSMatthew G. Knepley MPI_Comm comm; 459cafe43deSMatthew G. Knepley PetscGridHash lbox; 460cafe43deSMatthew G. Knepley Vec coordinates; 461cafe43deSMatthew G. Knepley PetscSection coordSection; 462cafe43deSMatthew G. Knepley Vec coordsLocal; 463cafe43deSMatthew G. Knepley const PetscScalar *coords; 464722d0f5cSMatthew G. Knepley PetscInt *dboxes, *boxes; 465cafe43deSMatthew G. Knepley PetscInt n[3] = {10, 10, 10}; 4661d0c6c94SMatthew G. Knepley PetscInt dim, N, cStart, cEnd, cMax, c, i; 467cafe43deSMatthew G. Knepley PetscErrorCode ierr; 468cafe43deSMatthew G. Knepley 469cafe43deSMatthew G. Knepley PetscFunctionBegin; 470cafe43deSMatthew G. Knepley ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr); 471cafe43deSMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 472cafe43deSMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr); 4735b3353d8SMatthew G. Knepley if (dim != 2) SETERRQ(comm, PETSC_ERR_SUP, "I have only coded this for 2D"); 474cafe43deSMatthew G. Knepley ierr = VecGetLocalSize(coordinates, &N);CHKERRQ(ierr); 475cafe43deSMatthew G. Knepley ierr = VecGetArrayRead(coordinates, &coords);CHKERRQ(ierr); 476cafe43deSMatthew G. Knepley ierr = PetscGridHashCreate(comm, dim, coords, &lbox);CHKERRQ(ierr); 477cafe43deSMatthew G. Knepley for (i = 0; i < N; i += dim) {ierr = PetscGridHashEnlarge(lbox, &coords[i]);CHKERRQ(ierr);} 478cafe43deSMatthew G. Knepley ierr = VecRestoreArrayRead(coordinates, &coords);CHKERRQ(ierr); 4799cb35068SDave May ierr = PetscOptionsGetInt(NULL,NULL,"-dm_plex_hash_box_nijk",&n[0],NULL);CHKERRQ(ierr); 4809cb35068SDave May n[1] = n[0]; 4819cb35068SDave May n[2] = n[0]; 482cafe43deSMatthew G. Knepley ierr = PetscGridHashSetGrid(lbox, n, NULL);CHKERRQ(ierr); 483cafe43deSMatthew G. Knepley #if 0 484cafe43deSMatthew G. Knepley /* Could define a custom reduction to merge these */ 485b2566f29SBarry Smith ierr = MPIU_Allreduce(lbox->lower, gbox->lower, 3, MPIU_REAL, MPI_MIN, comm);CHKERRQ(ierr); 486b2566f29SBarry Smith ierr = MPIU_Allreduce(lbox->upper, gbox->upper, 3, MPIU_REAL, MPI_MAX, comm);CHKERRQ(ierr); 487cafe43deSMatthew G. Knepley #endif 488cafe43deSMatthew G. Knepley /* Is there a reason to snap the local bounding box to a division of the global box? */ 489cafe43deSMatthew G. Knepley /* Should we compute all overlaps of local boxes? We could do this with a rendevouz scheme partitioning the global box */ 490cafe43deSMatthew G. Knepley /* Create label */ 491cafe43deSMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 4921d0c6c94SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr); 4931d0c6c94SMatthew G. Knepley if (cMax >= 0) cEnd = PetscMin(cEnd, cMax); 494d67d17b1SMatthew G. Knepley ierr = DMLabelCreate(PETSC_COMM_SELF, "cells", &lbox->cellsSparse);CHKERRQ(ierr); 495cafe43deSMatthew G. Knepley ierr = DMLabelCreateIndex(lbox->cellsSparse, cStart, cEnd);CHKERRQ(ierr); 496722d0f5cSMatthew G. Knepley /* Compute boxes which overlap each cell: http://stackoverflow.com/questions/13790208/triangle-square-intersection-test-in-2d */ 497cafe43deSMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr); 498cafe43deSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 49938353de4SMatthew G. Knepley ierr = PetscCalloc2(16 * dim, &dboxes, 16, &boxes);CHKERRQ(ierr); 500cafe43deSMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 501cafe43deSMatthew G. Knepley const PetscReal *h = lbox->h; 502cafe43deSMatthew G. Knepley PetscScalar *ccoords = NULL; 50338353de4SMatthew G. Knepley PetscInt csize = 0; 504cafe43deSMatthew G. Knepley PetscScalar point[3]; 505cafe43deSMatthew G. Knepley PetscInt dlim[6], d, e, i, j, k; 506cafe43deSMatthew G. Knepley 507cafe43deSMatthew G. Knepley /* Find boxes enclosing each vertex */ 50838353de4SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, &csize, &ccoords);CHKERRQ(ierr); 50938353de4SMatthew G. Knepley ierr = PetscGridHashGetEnclosingBox(lbox, csize/dim, ccoords, dboxes, boxes);CHKERRQ(ierr); 510722d0f5cSMatthew G. Knepley /* Mark cells containing the vertices */ 51138353de4SMatthew G. Knepley for (e = 0; e < csize/dim; ++e) {ierr = DMLabelSetValue(lbox->cellsSparse, c, boxes[e]);CHKERRQ(ierr);} 512cafe43deSMatthew G. Knepley /* Get grid of boxes containing these */ 513cafe43deSMatthew G. Knepley for (d = 0; d < dim; ++d) {dlim[d*2+0] = dlim[d*2+1] = dboxes[d];} 5142291669eSMatthew G. Knepley for (d = dim; d < 3; ++d) {dlim[d*2+0] = dlim[d*2+1] = 0;} 515cafe43deSMatthew G. Knepley for (e = 1; e < dim+1; ++e) { 516cafe43deSMatthew G. Knepley for (d = 0; d < dim; ++d) { 517cafe43deSMatthew G. Knepley dlim[d*2+0] = PetscMin(dlim[d*2+0], dboxes[e*dim+d]); 518cafe43deSMatthew G. Knepley dlim[d*2+1] = PetscMax(dlim[d*2+1], dboxes[e*dim+d]); 519cafe43deSMatthew G. Knepley } 520cafe43deSMatthew G. Knepley } 521fea14342SMatthew G. Knepley /* Check for intersection of box with cell */ 522cafe43deSMatthew G. Knepley for (k = dlim[2*2+0], point[2] = lbox->lower[2] + k*h[2]; k <= dlim[2*2+1]; ++k, point[2] += h[2]) { 523cafe43deSMatthew G. Knepley for (j = dlim[1*2+0], point[1] = lbox->lower[1] + j*h[1]; j <= dlim[1*2+1]; ++j, point[1] += h[1]) { 524cafe43deSMatthew G. Knepley for (i = dlim[0*2+0], point[0] = lbox->lower[0] + i*h[0]; i <= dlim[0*2+1]; ++i, point[0] += h[0]) { 525cafe43deSMatthew G. Knepley const PetscInt box = (k*lbox->n[1] + j)*lbox->n[0] + i; 526cafe43deSMatthew G. Knepley PetscScalar cpoint[3]; 527fea14342SMatthew G. Knepley PetscInt cell, edge, ii, jj, kk; 528cafe43deSMatthew G. Knepley 529fea14342SMatthew G. Knepley /* Check whether cell contains any vertex of these subboxes TODO vectorize this */ 530cafe43deSMatthew G. Knepley for (kk = 0, cpoint[2] = point[2]; kk < (dim > 2 ? 2 : 1); ++kk, cpoint[2] += h[2]) { 531cafe43deSMatthew G. Knepley for (jj = 0, cpoint[1] = point[1]; jj < (dim > 1 ? 2 : 1); ++jj, cpoint[1] += h[1]) { 532cafe43deSMatthew G. Knepley for (ii = 0, cpoint[0] = point[0]; ii < 2; ++ii, cpoint[0] += h[0]) { 533cafe43deSMatthew G. Knepley 534cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_Internal(dm, dim, cpoint, c, &cell);CHKERRQ(ierr); 535367003a6SStefano Zampini if (cell >= 0) { ierr = DMLabelSetValue(lbox->cellsSparse, c, box);CHKERRQ(ierr); ii = jj = kk = 2;} 536cafe43deSMatthew G. Knepley } 537cafe43deSMatthew G. Knepley } 538cafe43deSMatthew G. Knepley } 539fea14342SMatthew G. Knepley /* Check whether cell edge intersects any edge of these subboxes TODO vectorize this */ 540fea14342SMatthew G. Knepley for (edge = 0; edge < dim+1; ++edge) { 541fea14342SMatthew G. Knepley PetscReal segA[6], segB[6]; 542fea14342SMatthew G. Knepley 543aab5bcd8SJed Brown if (PetscUnlikely(dim > 3)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected dim %d > 3",dim); 544fea14342SMatthew G. Knepley for (d = 0; d < dim; ++d) {segA[d] = PetscRealPart(ccoords[edge*dim+d]); segA[dim+d] = PetscRealPart(ccoords[((edge+1)%(dim+1))*dim+d]);} 545fea14342SMatthew G. Knepley for (kk = 0; kk < (dim > 2 ? 2 : 1); ++kk) { 5469a128ed2SMatthew G. Knepley if (dim > 2) {segB[2] = PetscRealPart(point[2]); 5479a128ed2SMatthew G. Knepley segB[dim+2] = PetscRealPart(point[2]) + kk*h[2];} 548fea14342SMatthew G. Knepley for (jj = 0; jj < (dim > 1 ? 2 : 1); ++jj) { 5499a128ed2SMatthew G. Knepley if (dim > 1) {segB[1] = PetscRealPart(point[1]); 5509a128ed2SMatthew G. Knepley segB[dim+1] = PetscRealPart(point[1]) + jj*h[1];} 551fea14342SMatthew G. Knepley for (ii = 0; ii < 2; ++ii) { 552fea14342SMatthew G. Knepley PetscBool intersects; 553fea14342SMatthew G. Knepley 5549a128ed2SMatthew G. Knepley segB[0] = PetscRealPart(point[0]); 5559a128ed2SMatthew G. Knepley segB[dim+0] = PetscRealPart(point[0]) + ii*h[0]; 556fea14342SMatthew G. Knepley ierr = DMPlexGetLineIntersection_2D_Internal(segA, segB, NULL, &intersects);CHKERRQ(ierr); 557367003a6SStefano Zampini if (intersects) { ierr = DMLabelSetValue(lbox->cellsSparse, c, box);CHKERRQ(ierr); edge = ii = jj = kk = dim+1;} 558cafe43deSMatthew G. Knepley } 559cafe43deSMatthew G. Knepley } 560cafe43deSMatthew G. Knepley } 561cafe43deSMatthew G. Knepley } 562fea14342SMatthew G. Knepley } 563fea14342SMatthew G. Knepley } 564fea14342SMatthew G. Knepley } 565fea14342SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &ccoords);CHKERRQ(ierr); 566fea14342SMatthew G. Knepley } 567722d0f5cSMatthew G. Knepley ierr = PetscFree2(dboxes, boxes);CHKERRQ(ierr); 568cafe43deSMatthew G. Knepley ierr = DMLabelConvertToSection(lbox->cellsSparse, &lbox->cellSection, &lbox->cells);CHKERRQ(ierr); 569cafe43deSMatthew G. Knepley ierr = DMLabelDestroy(&lbox->cellsSparse);CHKERRQ(ierr); 570cafe43deSMatthew G. Knepley *localBox = lbox; 571cafe43deSMatthew G. Knepley PetscFunctionReturn(0); 572cafe43deSMatthew G. Knepley } 573cafe43deSMatthew G. Knepley 57462a38674SMatthew G. Knepley PetscErrorCode DMLocatePoints_Plex(DM dm, Vec v, DMPointLocationType ltype, PetscSF cellSF) 575ccd2543fSMatthew G Knepley { 576cafe43deSMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 577af74b616SDave May PetscBool hash = mesh->useHashLocation, reuse = PETSC_FALSE; 5783a93e3b7SToby Isaac PetscInt bs, numPoints, p, numFound, *found = NULL; 5799cb35068SDave May PetscInt dim, cStart, cEnd, cMax, numCells, c, d; 580cafe43deSMatthew G. Knepley const PetscInt *boxCells; 5813a93e3b7SToby Isaac PetscSFNode *cells; 582ccd2543fSMatthew G Knepley PetscScalar *a; 5833a93e3b7SToby Isaac PetscMPIInt result; 584af74b616SDave May PetscLogDouble t0,t1; 5859cb35068SDave May PetscReal gmin[3],gmax[3]; 5869cb35068SDave May PetscInt terminating_query_type[] = { 0, 0, 0 }; 587ccd2543fSMatthew G Knepley PetscErrorCode ierr; 588ccd2543fSMatthew G Knepley 589ccd2543fSMatthew G Knepley PetscFunctionBegin; 590af74b616SDave May ierr = PetscTime(&t0);CHKERRQ(ierr); 591080342d1SMatthew G. Knepley if (ltype == DM_POINTLOCATION_NEAREST && !hash) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_SUP, "Nearest point location only supported with grid hashing. Use -dm_plex_hash_location to enable it."); 592cafe43deSMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr); 593cafe43deSMatthew G. Knepley ierr = VecGetBlockSize(v, &bs);CHKERRQ(ierr); 5943a93e3b7SToby Isaac ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)cellSF),PETSC_COMM_SELF,&result);CHKERRQ(ierr); 5953a93e3b7SToby Isaac if (result != MPI_IDENT && result != MPI_CONGRUENT) SETERRQ(PetscObjectComm((PetscObject)cellSF),PETSC_ERR_SUP, "Trying parallel point location: only local point location supported"); 596cafe43deSMatthew G. Knepley if (bs != dim) SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Block size for point vector %D must be the mesh coordinate dimension %D", bs, dim); 597ccd2543fSMatthew G Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 598ccd2543fSMatthew G Knepley ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr); 599ccd2543fSMatthew G Knepley if (cMax >= 0) cEnd = PetscMin(cEnd, cMax); 600ccd2543fSMatthew G Knepley ierr = VecGetLocalSize(v, &numPoints);CHKERRQ(ierr); 601ccd2543fSMatthew G Knepley ierr = VecGetArray(v, &a);CHKERRQ(ierr); 602ccd2543fSMatthew G Knepley numPoints /= bs; 603af74b616SDave May { 604af74b616SDave May const PetscSFNode *sf_cells; 605af74b616SDave May 606af74b616SDave May ierr = PetscSFGetGraph(cellSF,NULL,NULL,NULL,&sf_cells);CHKERRQ(ierr); 607af74b616SDave May if (sf_cells) { 608af74b616SDave May ierr = PetscInfo(dm,"[DMLocatePoints_Plex] Re-using existing StarForest node list\n");CHKERRQ(ierr); 609af74b616SDave May cells = (PetscSFNode*)sf_cells; 610af74b616SDave May reuse = PETSC_TRUE; 611af74b616SDave May } else { 612af74b616SDave May ierr = PetscInfo(dm,"[DMLocatePoints_Plex] Creating and initializing new StarForest node list\n");CHKERRQ(ierr); 613785e854fSJed Brown ierr = PetscMalloc1(numPoints, &cells);CHKERRQ(ierr); 614af74b616SDave May /* initialize cells if created */ 615af74b616SDave May for (p=0; p<numPoints; p++) { 616af74b616SDave May cells[p].rank = 0; 617af74b616SDave May cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND; 618af74b616SDave May } 619af74b616SDave May } 620af74b616SDave May } 6219cb35068SDave May /* define domain bounding box */ 6229cb35068SDave May { 6239cb35068SDave May Vec coorglobal; 6249cb35068SDave May 6259cb35068SDave May ierr = DMGetCoordinates(dm,&coorglobal);CHKERRQ(ierr); 6269cb35068SDave May ierr = VecStrideMaxAll(coorglobal,NULL,gmax);CHKERRQ(ierr); 6279cb35068SDave May ierr = VecStrideMinAll(coorglobal,NULL,gmin);CHKERRQ(ierr); 6289cb35068SDave May } 629953fc75cSMatthew G. Knepley if (hash) { 630ac6ec2abSMatthew G. Knepley if (!mesh->lbox) {ierr = PetscInfo(dm, "Initializing grid hashing");CHKERRQ(ierr);ierr = DMPlexComputeGridHash_Internal(dm, &mesh->lbox);CHKERRQ(ierr);} 631cafe43deSMatthew G. Knepley /* Designate the local box for each point */ 632cafe43deSMatthew G. Knepley /* Send points to correct process */ 633cafe43deSMatthew G. Knepley /* Search cells that lie in each subbox */ 634cafe43deSMatthew G. Knepley /* Should we bin points before doing search? */ 635cafe43deSMatthew G. Knepley ierr = ISGetIndices(mesh->lbox->cells, &boxCells);CHKERRQ(ierr); 636953fc75cSMatthew G. Knepley } 6373a93e3b7SToby Isaac for (p = 0, numFound = 0; p < numPoints; ++p) { 638ccd2543fSMatthew G Knepley const PetscScalar *point = &a[p*bs]; 639e56f9228SJed Brown PetscInt dbin[3] = {-1,-1,-1}, bin, cell = -1, cellOffset; 6409cb35068SDave May PetscBool point_outside_domain = PETSC_FALSE; 641ccd2543fSMatthew G Knepley 6429cb35068SDave May /* check bounding box of domain */ 6439cb35068SDave May for (d=0; d<dim; d++) { 644a5f152d1SDave May if (PetscRealPart(point[d]) < gmin[d]) { point_outside_domain = PETSC_TRUE; break; } 645a5f152d1SDave May if (PetscRealPart(point[d]) > gmax[d]) { point_outside_domain = PETSC_TRUE; break; } 6469cb35068SDave May } 6479cb35068SDave May if (point_outside_domain) { 648e9b685f5SMatthew G. Knepley cells[p].rank = 0; 649e9b685f5SMatthew G. Knepley cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND; 6509cb35068SDave May terminating_query_type[0]++; 6519cb35068SDave May continue; 6529cb35068SDave May } 653ccd2543fSMatthew G Knepley 654af74b616SDave May /* check initial values in cells[].index - abort early if found */ 655af74b616SDave May if (cells[p].index != DMLOCATEPOINT_POINT_NOT_FOUND) { 656af74b616SDave May c = cells[p].index; 6573a93e3b7SToby Isaac cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND; 658af74b616SDave May ierr = DMPlexLocatePoint_Internal(dm, dim, point, c, &cell);CHKERRQ(ierr); 659af74b616SDave May if (cell >= 0) { 660af74b616SDave May cells[p].rank = 0; 661af74b616SDave May cells[p].index = cell; 662af74b616SDave May numFound++; 663af74b616SDave May } 664af74b616SDave May } 6659cb35068SDave May if (cells[p].index != DMLOCATEPOINT_POINT_NOT_FOUND) { 6669cb35068SDave May terminating_query_type[1]++; 6679cb35068SDave May continue; 6689cb35068SDave May } 669af74b616SDave May 670953fc75cSMatthew G. Knepley if (hash) { 671af74b616SDave May PetscBool found_box; 672af74b616SDave May 673af74b616SDave May /* allow for case that point is outside box - abort early */ 674af74b616SDave May ierr = PetscGridHashGetEnclosingBoxQuery(mesh->lbox, 1, point, dbin, &bin,&found_box);CHKERRQ(ierr); 675af74b616SDave May if (found_box) { 676cafe43deSMatthew G. Knepley /* TODO Lay an interface over this so we can switch between Section (dense) and Label (sparse) */ 677cafe43deSMatthew G. Knepley ierr = PetscSectionGetDof(mesh->lbox->cellSection, bin, &numCells);CHKERRQ(ierr); 678cafe43deSMatthew G. Knepley ierr = PetscSectionGetOffset(mesh->lbox->cellSection, bin, &cellOffset);CHKERRQ(ierr); 679cafe43deSMatthew G. Knepley for (c = cellOffset; c < cellOffset + numCells; ++c) { 680cafe43deSMatthew G. Knepley ierr = DMPlexLocatePoint_Internal(dm, dim, point, boxCells[c], &cell);CHKERRQ(ierr); 6813a93e3b7SToby Isaac if (cell >= 0) { 6823a93e3b7SToby Isaac cells[p].rank = 0; 6833a93e3b7SToby Isaac cells[p].index = cell; 6843a93e3b7SToby Isaac numFound++; 6859cb35068SDave May terminating_query_type[2]++; 6863a93e3b7SToby Isaac break; 687ccd2543fSMatthew G Knepley } 6883a93e3b7SToby Isaac } 689af74b616SDave May } 690953fc75cSMatthew G. Knepley } else { 691953fc75cSMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 692953fc75cSMatthew G. Knepley ierr = DMPlexLocatePoint_Internal(dm, dim, point, c, &cell);CHKERRQ(ierr); 6933a93e3b7SToby Isaac if (cell >= 0) { 6943a93e3b7SToby Isaac cells[p].rank = 0; 6953a93e3b7SToby Isaac cells[p].index = cell; 6963a93e3b7SToby Isaac numFound++; 6979cb35068SDave May terminating_query_type[2]++; 6983a93e3b7SToby Isaac break; 699953fc75cSMatthew G. Knepley } 700953fc75cSMatthew G. Knepley } 7013a93e3b7SToby Isaac } 702ccd2543fSMatthew G Knepley } 703953fc75cSMatthew G. Knepley if (hash) {ierr = ISRestoreIndices(mesh->lbox->cells, &boxCells);CHKERRQ(ierr);} 70462a38674SMatthew G. Knepley if (ltype == DM_POINTLOCATION_NEAREST && hash && numFound < numPoints) { 70562a38674SMatthew G. Knepley for (p = 0; p < numPoints; p++) { 70662a38674SMatthew G. Knepley const PetscScalar *point = &a[p*bs]; 70762a38674SMatthew G. Knepley PetscReal cpoint[3], diff[3], dist, distMax = PETSC_MAX_REAL; 708e56f9228SJed Brown PetscInt dbin[3] = {-1,-1,-1}, bin, cellOffset, d; 70962a38674SMatthew G. Knepley 710e9b685f5SMatthew G. Knepley if (cells[p].index < 0) { 71162a38674SMatthew G. Knepley ++numFound; 71262a38674SMatthew G. Knepley ierr = PetscGridHashGetEnclosingBox(mesh->lbox, 1, point, dbin, &bin);CHKERRQ(ierr); 71362a38674SMatthew G. Knepley ierr = PetscSectionGetDof(mesh->lbox->cellSection, bin, &numCells);CHKERRQ(ierr); 71462a38674SMatthew G. Knepley ierr = PetscSectionGetOffset(mesh->lbox->cellSection, bin, &cellOffset);CHKERRQ(ierr); 71562a38674SMatthew G. Knepley for (c = cellOffset; c < cellOffset + numCells; ++c) { 71662a38674SMatthew G. Knepley ierr = DMPlexClosestPoint_Internal(dm, dim, point, boxCells[c], cpoint);CHKERRQ(ierr); 717b716b415SMatthew G. Knepley for (d = 0; d < dim; ++d) diff[d] = cpoint[d] - PetscRealPart(point[d]); 71862a38674SMatthew G. Knepley dist = DMPlex_NormD_Internal(dim, diff); 71962a38674SMatthew G. Knepley if (dist < distMax) { 72062a38674SMatthew G. Knepley for (d = 0; d < dim; ++d) a[p*bs+d] = cpoint[d]; 72162a38674SMatthew G. Knepley cells[p].rank = 0; 72262a38674SMatthew G. Knepley cells[p].index = boxCells[c]; 72362a38674SMatthew G. Knepley distMax = dist; 72462a38674SMatthew G. Knepley break; 72562a38674SMatthew G. Knepley } 72662a38674SMatthew G. Knepley } 72762a38674SMatthew G. Knepley } 72862a38674SMatthew G. Knepley } 72962a38674SMatthew G. Knepley } 73062a38674SMatthew G. Knepley /* This code is only be relevant when interfaced to parallel point location */ 731cafe43deSMatthew G. Knepley /* Check for highest numbered proc that claims a point (do we care?) */ 7322d1fa6caSMatthew G. Knepley if (ltype == DM_POINTLOCATION_REMOVE && numFound < numPoints) { 7333a93e3b7SToby Isaac ierr = PetscMalloc1(numFound,&found);CHKERRQ(ierr); 7343a93e3b7SToby Isaac for (p = 0, numFound = 0; p < numPoints; p++) { 7353a93e3b7SToby Isaac if (cells[p].rank >= 0 && cells[p].index >= 0) { 7363a93e3b7SToby Isaac if (numFound < p) { 7373a93e3b7SToby Isaac cells[numFound] = cells[p]; 7383a93e3b7SToby Isaac } 7393a93e3b7SToby Isaac found[numFound++] = p; 7403a93e3b7SToby Isaac } 7413a93e3b7SToby Isaac } 7423a93e3b7SToby Isaac } 74362a38674SMatthew G. Knepley ierr = VecRestoreArray(v, &a);CHKERRQ(ierr); 744af74b616SDave May if (!reuse) { 7453a93e3b7SToby Isaac ierr = PetscSFSetGraph(cellSF, cEnd - cStart, numFound, found, PETSC_OWN_POINTER, cells, PETSC_OWN_POINTER);CHKERRQ(ierr); 746af74b616SDave May } 747af74b616SDave May ierr = PetscTime(&t1);CHKERRQ(ierr); 7489cb35068SDave May if (hash) { 749d6e0b8ebSSatish Balay ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] terminating_query_type : %D [outside domain] : %D [inside intial cell] : %D [hash]\n",terminating_query_type[0],terminating_query_type[1],terminating_query_type[2]);CHKERRQ(ierr); 7509cb35068SDave May } else { 751d6e0b8ebSSatish Balay ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] terminating_query_type : %D [outside domain] : %D [inside intial cell] : %D [brute-force]\n",terminating_query_type[0],terminating_query_type[1],terminating_query_type[2]);CHKERRQ(ierr); 7529cb35068SDave May } 753af74b616SDave May ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] npoints %D : time(rank0) %1.2e (sec): points/sec %1.4e\n",numPoints,t1-t0,(double)((double)numPoints/(t1-t0)));CHKERRQ(ierr); 754ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 755ccd2543fSMatthew G Knepley } 756ccd2543fSMatthew G Knepley 757741bfc07SMatthew G. Knepley /*@C 758741bfc07SMatthew G. Knepley DMPlexComputeProjection2Dto1D - Rewrite coordinates to be the 1D projection of the 2D coordinates 759741bfc07SMatthew G. Knepley 760741bfc07SMatthew G. Knepley Not collective 761741bfc07SMatthew G. Knepley 762741bfc07SMatthew G. Knepley Input Parameter: 763741bfc07SMatthew G. Knepley . coords - The coordinates of a segment 764741bfc07SMatthew G. Knepley 765741bfc07SMatthew G. Knepley Output Parameters: 766741bfc07SMatthew G. Knepley + coords - The new y-coordinate, and 0 for x 767741bfc07SMatthew G. Knepley - R - The rotation which accomplishes the projection 768741bfc07SMatthew G. Knepley 769741bfc07SMatthew G. Knepley Level: developer 770741bfc07SMatthew G. Knepley 771741bfc07SMatthew G. Knepley .seealso: DMPlexComputeProjection3Dto1D(), DMPlexComputeProjection3Dto2D() 772741bfc07SMatthew G. Knepley @*/ 773741bfc07SMatthew G. Knepley PetscErrorCode DMPlexComputeProjection2Dto1D(PetscScalar coords[], PetscReal R[]) 77417fe8556SMatthew G. Knepley { 77517fe8556SMatthew G. Knepley const PetscReal x = PetscRealPart(coords[2] - coords[0]); 77617fe8556SMatthew G. Knepley const PetscReal y = PetscRealPart(coords[3] - coords[1]); 7778b49ba18SBarry Smith const PetscReal r = PetscSqrtReal(x*x + y*y), c = x/r, s = y/r; 77817fe8556SMatthew G. Knepley 77917fe8556SMatthew G. Knepley PetscFunctionBegin; 7801c99cf0cSGeoffrey Irving R[0] = c; R[1] = -s; 7811c99cf0cSGeoffrey Irving R[2] = s; R[3] = c; 78217fe8556SMatthew G. Knepley coords[0] = 0.0; 7837f07f362SMatthew G. Knepley coords[1] = r; 78417fe8556SMatthew G. Knepley PetscFunctionReturn(0); 78517fe8556SMatthew G. Knepley } 78617fe8556SMatthew G. Knepley 787741bfc07SMatthew G. Knepley /*@C 788741bfc07SMatthew G. Knepley DMPlexComputeProjection3Dto1D - Rewrite coordinates to be the 1D projection of the 3D coordinates 78928dbe442SToby Isaac 790741bfc07SMatthew G. Knepley Not collective 79128dbe442SToby Isaac 792741bfc07SMatthew G. Knepley Input Parameter: 793741bfc07SMatthew G. Knepley . coords - The coordinates of a segment 794741bfc07SMatthew G. Knepley 795741bfc07SMatthew G. Knepley Output Parameters: 796741bfc07SMatthew G. Knepley + coords - The new y-coordinate, and 0 for x and z 797741bfc07SMatthew G. Knepley - R - The rotation which accomplishes the projection 798741bfc07SMatthew G. Knepley 799741bfc07SMatthew G. Knepley Note: This uses the basis completion described by Frisvad in http://www.imm.dtu.dk/~jerf/papers/abstracts/onb.html, DOI:10.1080/2165347X.2012.689606 800741bfc07SMatthew G. Knepley 801741bfc07SMatthew G. Knepley Level: developer 802741bfc07SMatthew G. Knepley 803741bfc07SMatthew G. Knepley .seealso: DMPlexComputeProjection2Dto1D(), DMPlexComputeProjection3Dto2D() 804741bfc07SMatthew G. Knepley @*/ 805741bfc07SMatthew G. Knepley PetscErrorCode DMPlexComputeProjection3Dto1D(PetscScalar coords[], PetscReal R[]) 80628dbe442SToby Isaac { 80728dbe442SToby Isaac PetscReal x = PetscRealPart(coords[3] - coords[0]); 80828dbe442SToby Isaac PetscReal y = PetscRealPart(coords[4] - coords[1]); 80928dbe442SToby Isaac PetscReal z = PetscRealPart(coords[5] - coords[2]); 81028dbe442SToby Isaac PetscReal r = PetscSqrtReal(x*x + y*y + z*z); 81128dbe442SToby Isaac PetscReal rinv = 1. / r; 81228dbe442SToby Isaac PetscFunctionBegin; 81328dbe442SToby Isaac 81428dbe442SToby Isaac x *= rinv; y *= rinv; z *= rinv; 81528dbe442SToby Isaac if (x > 0.) { 81628dbe442SToby Isaac PetscReal inv1pX = 1./ (1. + x); 81728dbe442SToby Isaac 81828dbe442SToby Isaac R[0] = x; R[1] = -y; R[2] = -z; 81928dbe442SToby Isaac R[3] = y; R[4] = 1. - y*y*inv1pX; R[5] = -y*z*inv1pX; 82028dbe442SToby Isaac R[6] = z; R[7] = -y*z*inv1pX; R[8] = 1. - z*z*inv1pX; 82128dbe442SToby Isaac } 82228dbe442SToby Isaac else { 82328dbe442SToby Isaac PetscReal inv1mX = 1./ (1. - x); 82428dbe442SToby Isaac 82528dbe442SToby Isaac R[0] = x; R[1] = z; R[2] = y; 82628dbe442SToby Isaac R[3] = y; R[4] = -y*z*inv1mX; R[5] = 1. - y*y*inv1mX; 82728dbe442SToby Isaac R[6] = z; R[7] = 1. - z*z*inv1mX; R[8] = -y*z*inv1mX; 82828dbe442SToby Isaac } 82928dbe442SToby Isaac coords[0] = 0.0; 83028dbe442SToby Isaac coords[1] = r; 83128dbe442SToby Isaac PetscFunctionReturn(0); 83228dbe442SToby Isaac } 83328dbe442SToby Isaac 834741bfc07SMatthew G. Knepley /*@ 835741bfc07SMatthew G. Knepley DMPlexComputeProjection3Dto2D - Rewrite coordinates to be the 2D projection of the 3D coordinates 836741bfc07SMatthew G. Knepley 837741bfc07SMatthew G. Knepley Not collective 838741bfc07SMatthew G. Knepley 839741bfc07SMatthew G. Knepley Input Parameter: 840741bfc07SMatthew G. Knepley . coords - The coordinates of a segment 841741bfc07SMatthew G. Knepley 842741bfc07SMatthew G. Knepley Output Parameters: 843741bfc07SMatthew G. Knepley + coords - The new y- and z-coordinates, and 0 for x 844741bfc07SMatthew G. Knepley - R - The rotation which accomplishes the projection 845741bfc07SMatthew G. Knepley 846741bfc07SMatthew G. Knepley Level: developer 847741bfc07SMatthew G. Knepley 848741bfc07SMatthew G. Knepley .seealso: DMPlexComputeProjection2Dto1D(), DMPlexComputeProjection3Dto1D() 849741bfc07SMatthew G. Knepley @*/ 850741bfc07SMatthew G. Knepley PetscErrorCode DMPlexComputeProjection3Dto2D(PetscInt coordSize, PetscScalar coords[], PetscReal R[]) 851ccd2543fSMatthew G Knepley { 8521ee9d5ecSMatthew G. Knepley PetscReal x1[3], x2[3], n[3], norm; 85399dec3a6SMatthew G. Knepley PetscReal x1p[3], x2p[3], xnp[3]; 8544a217a95SMatthew G. Knepley PetscReal sqrtz, alpha; 855ccd2543fSMatthew G Knepley const PetscInt dim = 3; 85699dec3a6SMatthew G. Knepley PetscInt d, e, p; 857ccd2543fSMatthew G Knepley 858ccd2543fSMatthew G Knepley PetscFunctionBegin; 859ccd2543fSMatthew G Knepley /* 0) Calculate normal vector */ 860ccd2543fSMatthew G Knepley for (d = 0; d < dim; ++d) { 8611ee9d5ecSMatthew G. Knepley x1[d] = PetscRealPart(coords[1*dim+d] - coords[0*dim+d]); 8621ee9d5ecSMatthew G. Knepley x2[d] = PetscRealPart(coords[2*dim+d] - coords[0*dim+d]); 863ccd2543fSMatthew G Knepley } 864ccd2543fSMatthew G Knepley n[0] = x1[1]*x2[2] - x1[2]*x2[1]; 865ccd2543fSMatthew G Knepley n[1] = x1[2]*x2[0] - x1[0]*x2[2]; 866ccd2543fSMatthew G Knepley n[2] = x1[0]*x2[1] - x1[1]*x2[0]; 8678b49ba18SBarry Smith norm = PetscSqrtReal(n[0]*n[0] + n[1]*n[1] + n[2]*n[2]); 868ccd2543fSMatthew G Knepley n[0] /= norm; 869ccd2543fSMatthew G Knepley n[1] /= norm; 870ccd2543fSMatthew G Knepley n[2] /= norm; 871ccd2543fSMatthew G Knepley /* 1) Take the normal vector and rotate until it is \hat z 872ccd2543fSMatthew G Knepley 873ccd2543fSMatthew G Knepley Let the normal vector be <nx, ny, nz> and alpha = 1/sqrt(1 - nz^2), then 874ccd2543fSMatthew G Knepley 875ccd2543fSMatthew G Knepley R = / alpha nx nz alpha ny nz -1/alpha \ 876ccd2543fSMatthew G Knepley | -alpha ny alpha nx 0 | 877ccd2543fSMatthew G Knepley \ nx ny nz / 878ccd2543fSMatthew G Knepley 879ccd2543fSMatthew G Knepley will rotate the normal vector to \hat z 880ccd2543fSMatthew G Knepley */ 8818b49ba18SBarry Smith sqrtz = PetscSqrtReal(1.0 - n[2]*n[2]); 88273868372SMatthew G. Knepley /* Check for n = z */ 88373868372SMatthew G. Knepley if (sqrtz < 1.0e-10) { 8847df32b8bSSanderA const PetscInt s = PetscSign(n[2]); 8857df32b8bSSanderA /* If nz < 0, rotate 180 degrees around x-axis */ 88699dec3a6SMatthew G. Knepley for (p = 3; p < coordSize/3; ++p) { 88799dec3a6SMatthew G. Knepley coords[p*2+0] = PetscRealPart(coords[p*dim+0] - coords[0*dim+0]); 8887df32b8bSSanderA coords[p*2+1] = (PetscRealPart(coords[p*dim+1] - coords[0*dim+1])) * s; 88973868372SMatthew G. Knepley } 89099dec3a6SMatthew G. Knepley coords[0] = 0.0; 89199dec3a6SMatthew G. Knepley coords[1] = 0.0; 8927df32b8bSSanderA coords[2] = x1[0]; 8937df32b8bSSanderA coords[3] = x1[1] * s; 8947df32b8bSSanderA coords[4] = x2[0]; 8957df32b8bSSanderA coords[5] = x2[1] * s; 8967df32b8bSSanderA R[0] = 1.0; R[1] = 0.0; R[2] = 0.0; 8977df32b8bSSanderA R[3] = 0.0; R[4] = 1.0 * s; R[5] = 0.0; 8987df32b8bSSanderA R[6] = 0.0; R[7] = 0.0; R[8] = 1.0 * s; 89973868372SMatthew G. Knepley PetscFunctionReturn(0); 90073868372SMatthew G. Knepley } 901da18b5e6SMatthew G Knepley alpha = 1.0/sqrtz; 902ccd2543fSMatthew G Knepley R[0] = alpha*n[0]*n[2]; R[1] = alpha*n[1]*n[2]; R[2] = -sqrtz; 903ccd2543fSMatthew G Knepley R[3] = -alpha*n[1]; R[4] = alpha*n[0]; R[5] = 0.0; 904ccd2543fSMatthew G Knepley R[6] = n[0]; R[7] = n[1]; R[8] = n[2]; 905ccd2543fSMatthew G Knepley for (d = 0; d < dim; ++d) { 906ccd2543fSMatthew G Knepley x1p[d] = 0.0; 907ccd2543fSMatthew G Knepley x2p[d] = 0.0; 908ccd2543fSMatthew G Knepley for (e = 0; e < dim; ++e) { 909ccd2543fSMatthew G Knepley x1p[d] += R[d*dim+e]*x1[e]; 910ccd2543fSMatthew G Knepley x2p[d] += R[d*dim+e]*x2[e]; 911ccd2543fSMatthew G Knepley } 912ccd2543fSMatthew G Knepley } 91348120919SToby Isaac if (PetscAbsReal(x1p[2]) > 10. * PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid rotation calculated"); 91448120919SToby Isaac if (PetscAbsReal(x2p[2]) > 10. * PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid rotation calculated"); 915ccd2543fSMatthew G Knepley /* 2) Project to (x, y) */ 91699dec3a6SMatthew G. Knepley for (p = 3; p < coordSize/3; ++p) { 91799dec3a6SMatthew G. Knepley for (d = 0; d < dim; ++d) { 91899dec3a6SMatthew G. Knepley xnp[d] = 0.0; 91999dec3a6SMatthew G. Knepley for (e = 0; e < dim; ++e) { 92099dec3a6SMatthew G. Knepley xnp[d] += R[d*dim+e]*PetscRealPart(coords[p*dim+e] - coords[0*dim+e]); 92199dec3a6SMatthew G. Knepley } 92299dec3a6SMatthew G. Knepley if (d < dim-1) coords[p*2+d] = xnp[d]; 92399dec3a6SMatthew G. Knepley } 92499dec3a6SMatthew G. Knepley } 925ccd2543fSMatthew G Knepley coords[0] = 0.0; 926ccd2543fSMatthew G Knepley coords[1] = 0.0; 927ccd2543fSMatthew G Knepley coords[2] = x1p[0]; 928ccd2543fSMatthew G Knepley coords[3] = x1p[1]; 929ccd2543fSMatthew G Knepley coords[4] = x2p[0]; 930ccd2543fSMatthew G Knepley coords[5] = x2p[1]; 9317f07f362SMatthew G. Knepley /* Output R^T which rotates \hat z to the input normal */ 9327f07f362SMatthew G. Knepley for (d = 0; d < dim; ++d) { 9337f07f362SMatthew G. Knepley for (e = d+1; e < dim; ++e) { 9347f07f362SMatthew G. Knepley PetscReal tmp; 9357f07f362SMatthew G. Knepley 9367f07f362SMatthew G. Knepley tmp = R[d*dim+e]; 9377f07f362SMatthew G. Knepley R[d*dim+e] = R[e*dim+d]; 9387f07f362SMatthew G. Knepley R[e*dim+d] = tmp; 9397f07f362SMatthew G. Knepley } 9407f07f362SMatthew G. Knepley } 941ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 942ccd2543fSMatthew G Knepley } 943ccd2543fSMatthew G Knepley 9446322fe33SJed Brown PETSC_UNUSED 945834e62ceSMatthew G. Knepley PETSC_STATIC_INLINE void Volume_Triangle_Internal(PetscReal *vol, PetscReal coords[]) 946834e62ceSMatthew G. Knepley { 947834e62ceSMatthew G. Knepley /* Signed volume is 1/2 the determinant 948834e62ceSMatthew G. Knepley 949834e62ceSMatthew G. Knepley | 1 1 1 | 950834e62ceSMatthew G. Knepley | x0 x1 x2 | 951834e62ceSMatthew G. Knepley | y0 y1 y2 | 952834e62ceSMatthew G. Knepley 953834e62ceSMatthew G. Knepley but if x0,y0 is the origin, we have 954834e62ceSMatthew G. Knepley 955834e62ceSMatthew G. Knepley | x1 x2 | 956834e62ceSMatthew G. Knepley | y1 y2 | 957834e62ceSMatthew G. Knepley */ 958834e62ceSMatthew G. Knepley const PetscReal x1 = coords[2] - coords[0], y1 = coords[3] - coords[1]; 959834e62ceSMatthew G. Knepley const PetscReal x2 = coords[4] - coords[0], y2 = coords[5] - coords[1]; 960834e62ceSMatthew G. Knepley PetscReal M[4], detM; 961834e62ceSMatthew G. Knepley M[0] = x1; M[1] = x2; 96286623015SMatthew G. Knepley M[2] = y1; M[3] = y2; 963923591dfSMatthew G. Knepley DMPlex_Det2D_Internal(&detM, M); 964834e62ceSMatthew G. Knepley *vol = 0.5*detM; 9653bc0b13bSBarry Smith (void)PetscLogFlops(5.0); 966834e62ceSMatthew G. Knepley } 967834e62ceSMatthew G. Knepley 968834e62ceSMatthew G. Knepley PETSC_STATIC_INLINE void Volume_Triangle_Origin_Internal(PetscReal *vol, PetscReal coords[]) 969834e62ceSMatthew G. Knepley { 970923591dfSMatthew G. Knepley DMPlex_Det2D_Internal(vol, coords); 971834e62ceSMatthew G. Knepley *vol *= 0.5; 972834e62ceSMatthew G. Knepley } 973834e62ceSMatthew G. Knepley 9746322fe33SJed Brown PETSC_UNUSED 975834e62ceSMatthew G. Knepley PETSC_STATIC_INLINE void Volume_Tetrahedron_Internal(PetscReal *vol, PetscReal coords[]) 976834e62ceSMatthew G. Knepley { 977834e62ceSMatthew G. Knepley /* Signed volume is 1/6th of the determinant 978834e62ceSMatthew G. Knepley 979834e62ceSMatthew G. Knepley | 1 1 1 1 | 980834e62ceSMatthew G. Knepley | x0 x1 x2 x3 | 981834e62ceSMatthew G. Knepley | y0 y1 y2 y3 | 982834e62ceSMatthew G. Knepley | z0 z1 z2 z3 | 983834e62ceSMatthew G. Knepley 984834e62ceSMatthew G. Knepley but if x0,y0,z0 is the origin, we have 985834e62ceSMatthew G. Knepley 986834e62ceSMatthew G. Knepley | x1 x2 x3 | 987834e62ceSMatthew G. Knepley | y1 y2 y3 | 988834e62ceSMatthew G. Knepley | z1 z2 z3 | 989834e62ceSMatthew G. Knepley */ 990834e62ceSMatthew G. Knepley const PetscReal x1 = coords[3] - coords[0], y1 = coords[4] - coords[1], z1 = coords[5] - coords[2]; 991834e62ceSMatthew G. Knepley const PetscReal x2 = coords[6] - coords[0], y2 = coords[7] - coords[1], z2 = coords[8] - coords[2]; 992834e62ceSMatthew G. Knepley const PetscReal x3 = coords[9] - coords[0], y3 = coords[10] - coords[1], z3 = coords[11] - coords[2]; 9930a3da2c2SToby Isaac const PetscReal onesixth = ((PetscReal)1./(PetscReal)6.); 994834e62ceSMatthew G. Knepley PetscReal M[9], detM; 995834e62ceSMatthew G. Knepley M[0] = x1; M[1] = x2; M[2] = x3; 996834e62ceSMatthew G. Knepley M[3] = y1; M[4] = y2; M[5] = y3; 997834e62ceSMatthew G. Knepley M[6] = z1; M[7] = z2; M[8] = z3; 998923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(&detM, M); 9990a3da2c2SToby Isaac *vol = -onesixth*detM; 10003bc0b13bSBarry Smith (void)PetscLogFlops(10.0); 1001834e62ceSMatthew G. Knepley } 1002834e62ceSMatthew G. Knepley 10030ec8681fSMatthew G. Knepley PETSC_STATIC_INLINE void Volume_Tetrahedron_Origin_Internal(PetscReal *vol, PetscReal coords[]) 10040ec8681fSMatthew G. Knepley { 10050a3da2c2SToby Isaac const PetscReal onesixth = ((PetscReal)1./(PetscReal)6.); 1006923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(vol, coords); 10070a3da2c2SToby Isaac *vol *= -onesixth; 10080ec8681fSMatthew G. Knepley } 10090ec8681fSMatthew G. Knepley 1010cb92db44SToby Isaac static PetscErrorCode DMPlexComputePointGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 1011cb92db44SToby Isaac { 1012cb92db44SToby Isaac PetscSection coordSection; 1013cb92db44SToby Isaac Vec coordinates; 1014cb92db44SToby Isaac const PetscScalar *coords; 1015cb92db44SToby Isaac PetscInt dim, d, off; 1016cb92db44SToby Isaac PetscErrorCode ierr; 1017cb92db44SToby Isaac 1018cb92db44SToby Isaac PetscFunctionBegin; 1019cb92db44SToby Isaac ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 1020cb92db44SToby Isaac ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 1021cb92db44SToby Isaac ierr = PetscSectionGetDof(coordSection,e,&dim);CHKERRQ(ierr); 1022cb92db44SToby Isaac if (!dim) PetscFunctionReturn(0); 1023cb92db44SToby Isaac ierr = PetscSectionGetOffset(coordSection,e,&off);CHKERRQ(ierr); 1024cb92db44SToby Isaac ierr = VecGetArrayRead(coordinates,&coords);CHKERRQ(ierr); 1025cb92db44SToby Isaac if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[off + d]);} 1026cb92db44SToby Isaac ierr = VecRestoreArrayRead(coordinates,&coords);CHKERRQ(ierr); 1027cb92db44SToby Isaac *detJ = 1.; 1028cb92db44SToby Isaac if (J) { 1029cb92db44SToby Isaac for (d = 0; d < dim * dim; d++) J[d] = 0.; 1030cb92db44SToby Isaac for (d = 0; d < dim; d++) J[d * dim + d] = 1.; 1031cb92db44SToby Isaac if (invJ) { 1032cb92db44SToby Isaac for (d = 0; d < dim * dim; d++) invJ[d] = 0.; 1033cb92db44SToby Isaac for (d = 0; d < dim; d++) invJ[d * dim + d] = 1.; 1034cb92db44SToby Isaac } 1035cb92db44SToby Isaac } 1036cb92db44SToby Isaac PetscFunctionReturn(0); 1037cb92db44SToby Isaac } 1038cb92db44SToby Isaac 103917fe8556SMatthew G. Knepley static PetscErrorCode DMPlexComputeLineGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 104017fe8556SMatthew G. Knepley { 104117fe8556SMatthew G. Knepley PetscSection coordSection; 104217fe8556SMatthew G. Knepley Vec coordinates; 1043a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 10448bf5c034SToby Isaac PetscInt numCoords, d, pStart, pEnd, numSelfCoords = 0; 104517fe8556SMatthew G. Knepley PetscErrorCode ierr; 104617fe8556SMatthew G. Knepley 104717fe8556SMatthew G. Knepley PetscFunctionBegin; 104817fe8556SMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 104969d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 10508bf5c034SToby Isaac ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr); 10518bf5c034SToby Isaac if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);} 105217fe8556SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 10538bf5c034SToby Isaac numCoords = numSelfCoords ? numSelfCoords : numCoords; 1054adac9986SMatthew G. Knepley if (invJ && !J) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "In order to compute invJ, J must not be NULL"); 10557f07f362SMatthew G. Knepley *detJ = 0.0; 105628dbe442SToby Isaac if (numCoords == 6) { 105728dbe442SToby Isaac const PetscInt dim = 3; 105828dbe442SToby Isaac PetscReal R[9], J0; 105928dbe442SToby Isaac 106028dbe442SToby Isaac if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 1061741bfc07SMatthew G. Knepley ierr = DMPlexComputeProjection3Dto1D(coords, R);CHKERRQ(ierr); 106228dbe442SToby Isaac if (J) { 106328dbe442SToby Isaac J0 = 0.5*PetscRealPart(coords[1]); 106428dbe442SToby Isaac J[0] = R[0]*J0; J[1] = R[1]; J[2] = R[2]; 106528dbe442SToby Isaac J[3] = R[3]*J0; J[4] = R[4]; J[5] = R[5]; 106628dbe442SToby Isaac J[6] = R[6]*J0; J[7] = R[7]; J[8] = R[8]; 106728dbe442SToby Isaac DMPlex_Det3D_Internal(detJ, J); 106828dbe442SToby Isaac if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);} 1069adac9986SMatthew G. Knepley } 107028dbe442SToby Isaac } else if (numCoords == 4) { 10717f07f362SMatthew G. Knepley const PetscInt dim = 2; 10727f07f362SMatthew G. Knepley PetscReal R[4], J0; 10737f07f362SMatthew G. Knepley 10747f07f362SMatthew G. Knepley if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 1075741bfc07SMatthew G. Knepley ierr = DMPlexComputeProjection2Dto1D(coords, R);CHKERRQ(ierr); 107617fe8556SMatthew G. Knepley if (J) { 10777f07f362SMatthew G. Knepley J0 = 0.5*PetscRealPart(coords[1]); 10787f07f362SMatthew G. Knepley J[0] = R[0]*J0; J[1] = R[1]; 10797f07f362SMatthew G. Knepley J[2] = R[2]*J0; J[3] = R[3]; 1080923591dfSMatthew G. Knepley DMPlex_Det2D_Internal(detJ, J); 1081923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);} 1082adac9986SMatthew G. Knepley } 10837f07f362SMatthew G. Knepley } else if (numCoords == 2) { 10847f07f362SMatthew G. Knepley const PetscInt dim = 1; 10857f07f362SMatthew G. Knepley 10867f07f362SMatthew G. Knepley if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 10877f07f362SMatthew G. Knepley if (J) { 10887f07f362SMatthew G. Knepley J[0] = 0.5*(PetscRealPart(coords[1]) - PetscRealPart(coords[0])); 108917fe8556SMatthew G. Knepley *detJ = J[0]; 10903bc0b13bSBarry Smith ierr = PetscLogFlops(2.0);CHKERRQ(ierr); 10913bc0b13bSBarry Smith if (invJ) {invJ[0] = 1.0/J[0]; ierr = PetscLogFlops(1.0);CHKERRQ(ierr);} 1092adac9986SMatthew G. Knepley } 1093796f034aSJed Brown } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this segment is %D != 2", numCoords); 109417fe8556SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 109517fe8556SMatthew G. Knepley PetscFunctionReturn(0); 109617fe8556SMatthew G. Knepley } 109717fe8556SMatthew G. Knepley 1098ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexComputeTriangleGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 1099ccd2543fSMatthew G Knepley { 1100ccd2543fSMatthew G Knepley PetscSection coordSection; 1101ccd2543fSMatthew G Knepley Vec coordinates; 1102a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 110303d7ed2eSStefano Zampini PetscInt numCoords, numSelfCoords = 0, d, f, g, pStart, pEnd; 1104ccd2543fSMatthew G Knepley PetscErrorCode ierr; 1105ccd2543fSMatthew G Knepley 1106ccd2543fSMatthew G Knepley PetscFunctionBegin; 1107ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 110869d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 110903d7ed2eSStefano Zampini ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr); 111003d7ed2eSStefano Zampini if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);} 1111ccd2543fSMatthew G Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 111203d7ed2eSStefano Zampini numCoords = numSelfCoords ? numSelfCoords : numCoords; 11137f07f362SMatthew G. Knepley *detJ = 0.0; 1114ccd2543fSMatthew G Knepley if (numCoords == 9) { 11157f07f362SMatthew G. Knepley const PetscInt dim = 3; 11167f07f362SMatthew G. Knepley PetscReal R[9], J0[9] = {1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0}; 11177f07f362SMatthew G. Knepley 11187f07f362SMatthew G. Knepley if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 1119741bfc07SMatthew G. Knepley ierr = DMPlexComputeProjection3Dto2D(numCoords, coords, R);CHKERRQ(ierr); 11207f07f362SMatthew G. Knepley if (J) { 1121b7ad821dSMatthew G. Knepley const PetscInt pdim = 2; 1122b7ad821dSMatthew G. Knepley 1123b7ad821dSMatthew G. Knepley for (d = 0; d < pdim; d++) { 1124b7ad821dSMatthew G. Knepley for (f = 0; f < pdim; f++) { 1125b7ad821dSMatthew G. Knepley J0[d*dim+f] = 0.5*(PetscRealPart(coords[(f+1)*pdim+d]) - PetscRealPart(coords[0*pdim+d])); 1126ccd2543fSMatthew G Knepley } 11277f07f362SMatthew G. Knepley } 11283bc0b13bSBarry Smith ierr = PetscLogFlops(8.0);CHKERRQ(ierr); 1129923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(detJ, J0); 11307f07f362SMatthew G. Knepley for (d = 0; d < dim; d++) { 11317f07f362SMatthew G. Knepley for (f = 0; f < dim; f++) { 11327f07f362SMatthew G. Knepley J[d*dim+f] = 0.0; 11337f07f362SMatthew G. Knepley for (g = 0; g < dim; g++) { 11347f07f362SMatthew G. Knepley J[d*dim+f] += R[d*dim+g]*J0[g*dim+f]; 11357f07f362SMatthew G. Knepley } 11367f07f362SMatthew G. Knepley } 11377f07f362SMatthew G. Knepley } 11383bc0b13bSBarry Smith ierr = PetscLogFlops(18.0);CHKERRQ(ierr); 11397f07f362SMatthew G. Knepley } 1140923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);} 11417f07f362SMatthew G. Knepley } else if (numCoords == 6) { 11427f07f362SMatthew G. Knepley const PetscInt dim = 2; 11437f07f362SMatthew G. Knepley 11447f07f362SMatthew G. Knepley if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 1145ccd2543fSMatthew G Knepley if (J) { 1146ccd2543fSMatthew G Knepley for (d = 0; d < dim; d++) { 1147ccd2543fSMatthew G Knepley for (f = 0; f < dim; f++) { 1148ccd2543fSMatthew G Knepley J[d*dim+f] = 0.5*(PetscRealPart(coords[(f+1)*dim+d]) - PetscRealPart(coords[0*dim+d])); 1149ccd2543fSMatthew G Knepley } 1150ccd2543fSMatthew G Knepley } 11513bc0b13bSBarry Smith ierr = PetscLogFlops(8.0);CHKERRQ(ierr); 1152923591dfSMatthew G. Knepley DMPlex_Det2D_Internal(detJ, J); 1153ccd2543fSMatthew G Knepley } 1154923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);} 1155796f034aSJed Brown } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this triangle is %D != 6 or 9", numCoords); 1156ccd2543fSMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 1157ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 1158ccd2543fSMatthew G Knepley } 1159ccd2543fSMatthew G Knepley 1160dfccc68fSToby Isaac static PetscErrorCode DMPlexComputeRectangleGeometry_Internal(DM dm, PetscInt e, PetscInt Nq, const PetscReal points[], PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 1161ccd2543fSMatthew G Knepley { 1162ccd2543fSMatthew G Knepley PetscSection coordSection; 1163ccd2543fSMatthew G Knepley Vec coordinates; 1164a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 11650d29256aSToby Isaac PetscInt numCoords, numSelfCoords = 0, d, f, g, pStart, pEnd; 1166ccd2543fSMatthew G Knepley PetscErrorCode ierr; 1167ccd2543fSMatthew G Knepley 1168ccd2543fSMatthew G Knepley PetscFunctionBegin; 1169ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 117069d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 11710d29256aSToby Isaac ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr); 11720d29256aSToby Isaac if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);} 117399dec3a6SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 117471f58de1SToby Isaac numCoords = numSelfCoords ? numSelfCoords : numCoords; 1175dfccc68fSToby Isaac if (!Nq) { 11767f07f362SMatthew G. Knepley *detJ = 0.0; 117799dec3a6SMatthew G. Knepley if (numCoords == 12) { 117899dec3a6SMatthew G. Knepley const PetscInt dim = 3; 117999dec3a6SMatthew G. Knepley PetscReal R[9], J0[9] = {1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0}; 118099dec3a6SMatthew G. Knepley 1181dfccc68fSToby Isaac if (v) {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);} 1182741bfc07SMatthew G. Knepley ierr = DMPlexComputeProjection3Dto2D(numCoords, coords, R);CHKERRQ(ierr); 118399dec3a6SMatthew G. Knepley if (J) { 118499dec3a6SMatthew G. Knepley const PetscInt pdim = 2; 118599dec3a6SMatthew G. Knepley 118699dec3a6SMatthew G. Knepley for (d = 0; d < pdim; d++) { 118799dec3a6SMatthew G. Knepley J0[d*dim+0] = 0.5*(PetscRealPart(coords[1*pdim+d]) - PetscRealPart(coords[0*pdim+d])); 11880892ce5fSStefano Zampini J0[d*dim+1] = 0.5*(PetscRealPart(coords[2*pdim+d]) - PetscRealPart(coords[1*pdim+d])); 118999dec3a6SMatthew G. Knepley } 11903bc0b13bSBarry Smith ierr = PetscLogFlops(8.0);CHKERRQ(ierr); 1191923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(detJ, J0); 119299dec3a6SMatthew G. Knepley for (d = 0; d < dim; d++) { 119399dec3a6SMatthew G. Knepley for (f = 0; f < dim; f++) { 119499dec3a6SMatthew G. Knepley J[d*dim+f] = 0.0; 119599dec3a6SMatthew G. Knepley for (g = 0; g < dim; g++) { 119699dec3a6SMatthew G. Knepley J[d*dim+f] += R[d*dim+g]*J0[g*dim+f]; 119799dec3a6SMatthew G. Knepley } 119899dec3a6SMatthew G. Knepley } 119999dec3a6SMatthew G. Knepley } 12003bc0b13bSBarry Smith ierr = PetscLogFlops(18.0);CHKERRQ(ierr); 120199dec3a6SMatthew G. Knepley } 1202923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);} 120371f58de1SToby Isaac } else if (numCoords == 8) { 120499dec3a6SMatthew G. Knepley const PetscInt dim = 2; 120599dec3a6SMatthew G. Knepley 1206dfccc68fSToby Isaac if (v) {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);} 1207ccd2543fSMatthew G Knepley if (J) { 1208ccd2543fSMatthew G Knepley for (d = 0; d < dim; d++) { 120999dec3a6SMatthew G. Knepley J[d*dim+0] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d])); 121099dec3a6SMatthew G. Knepley J[d*dim+1] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d])); 1211ccd2543fSMatthew G Knepley } 12123bc0b13bSBarry Smith ierr = PetscLogFlops(8.0);CHKERRQ(ierr); 1213923591dfSMatthew G. Knepley DMPlex_Det2D_Internal(detJ, J); 1214ccd2543fSMatthew G Knepley } 1215923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);} 1216796f034aSJed Brown } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this quadrilateral is %D != 8 or 12", numCoords); 1217dfccc68fSToby Isaac } else { 1218dfccc68fSToby Isaac const PetscInt Nv = 4; 1219dfccc68fSToby Isaac const PetscInt dimR = 2; 1220dfccc68fSToby Isaac const PetscInt zToPlex[4] = {0, 1, 3, 2}; 1221dfccc68fSToby Isaac PetscReal zOrder[12]; 1222dfccc68fSToby Isaac PetscReal zCoeff[12]; 1223dfccc68fSToby Isaac PetscInt i, j, k, l, dim; 1224dfccc68fSToby Isaac 1225dfccc68fSToby Isaac if (numCoords == 12) { 1226dfccc68fSToby Isaac dim = 3; 1227dfccc68fSToby Isaac } else if (numCoords == 8) { 1228dfccc68fSToby Isaac dim = 2; 1229dfccc68fSToby Isaac } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this quadrilateral is %D != 8 or 12", numCoords); 1230dfccc68fSToby Isaac for (i = 0; i < Nv; i++) { 1231dfccc68fSToby Isaac PetscInt zi = zToPlex[i]; 1232dfccc68fSToby Isaac 1233dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1234dfccc68fSToby Isaac zOrder[dim * i + j] = PetscRealPart(coords[dim * zi + j]); 1235dfccc68fSToby Isaac } 1236dfccc68fSToby Isaac } 1237dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1238dfccc68fSToby Isaac zCoeff[dim * 0 + j] = 0.25 * ( zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j]); 1239dfccc68fSToby Isaac zCoeff[dim * 1 + j] = 0.25 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j]); 1240dfccc68fSToby Isaac zCoeff[dim * 2 + j] = 0.25 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j]); 1241dfccc68fSToby Isaac zCoeff[dim * 3 + j] = 0.25 * ( zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j]); 1242dfccc68fSToby Isaac } 1243dfccc68fSToby Isaac for (i = 0; i < Nq; i++) { 1244dfccc68fSToby Isaac PetscReal xi = points[dimR * i], eta = points[dimR * i + 1]; 1245dfccc68fSToby Isaac 1246dfccc68fSToby Isaac if (v) { 1247dfccc68fSToby Isaac PetscReal extPoint[4]; 1248dfccc68fSToby Isaac 1249dfccc68fSToby Isaac extPoint[0] = 1.; 1250dfccc68fSToby Isaac extPoint[1] = xi; 1251dfccc68fSToby Isaac extPoint[2] = eta; 1252dfccc68fSToby Isaac extPoint[3] = xi * eta; 1253dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1254dfccc68fSToby Isaac PetscReal val = 0.; 1255dfccc68fSToby Isaac 1256dfccc68fSToby Isaac for (k = 0; k < Nv; k++) { 1257dfccc68fSToby Isaac val += extPoint[k] * zCoeff[dim * k + j]; 1258dfccc68fSToby Isaac } 1259dfccc68fSToby Isaac v[i * dim + j] = val; 1260dfccc68fSToby Isaac } 1261dfccc68fSToby Isaac } 1262dfccc68fSToby Isaac if (J) { 1263dfccc68fSToby Isaac PetscReal extJ[8]; 1264dfccc68fSToby Isaac 1265dfccc68fSToby Isaac extJ[0] = 0.; 1266dfccc68fSToby Isaac extJ[1] = 0.; 1267dfccc68fSToby Isaac extJ[2] = 1.; 1268dfccc68fSToby Isaac extJ[3] = 0.; 1269dfccc68fSToby Isaac extJ[4] = 0.; 1270dfccc68fSToby Isaac extJ[5] = 1.; 1271dfccc68fSToby Isaac extJ[6] = eta; 1272dfccc68fSToby Isaac extJ[7] = xi; 1273dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1274dfccc68fSToby Isaac for (k = 0; k < dimR; k++) { 1275dfccc68fSToby Isaac PetscReal val = 0.; 1276dfccc68fSToby Isaac 1277dfccc68fSToby Isaac for (l = 0; l < Nv; l++) { 1278dfccc68fSToby Isaac val += zCoeff[dim * l + j] * extJ[dimR * l + k]; 1279dfccc68fSToby Isaac } 1280dfccc68fSToby Isaac J[i * dim * dim + dim * j + k] = val; 1281dfccc68fSToby Isaac } 1282dfccc68fSToby Isaac } 1283dfccc68fSToby Isaac if (dim == 3) { /* put the cross product in the third component of the Jacobian */ 1284dfccc68fSToby Isaac PetscReal x, y, z; 1285dfccc68fSToby Isaac PetscReal *iJ = &J[i * dim * dim]; 1286dfccc68fSToby Isaac PetscReal norm; 1287dfccc68fSToby Isaac 1288dfccc68fSToby Isaac x = iJ[1 * dim + 0] * iJ[2 * dim + 1] - iJ[1 * dim + 1] * iJ[2 * dim + 0]; 1289dfccc68fSToby Isaac y = iJ[0 * dim + 1] * iJ[2 * dim + 0] - iJ[0 * dim + 0] * iJ[2 * dim + 1]; 1290dfccc68fSToby Isaac z = iJ[0 * dim + 0] * iJ[1 * dim + 1] - iJ[0 * dim + 1] * iJ[1 * dim + 0]; 1291dfccc68fSToby Isaac norm = PetscSqrtReal(x * x + y * y + z * z); 1292dfccc68fSToby Isaac iJ[2] = x / norm; 1293dfccc68fSToby Isaac iJ[5] = y / norm; 1294dfccc68fSToby Isaac iJ[8] = z / norm; 1295dfccc68fSToby Isaac DMPlex_Det3D_Internal(&detJ[i], &J[i * dim * dim]); 1296dfccc68fSToby Isaac if (invJ) {DMPlex_Invert3D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);} 1297dfccc68fSToby Isaac } else { 1298dfccc68fSToby Isaac DMPlex_Det2D_Internal(&detJ[i], &J[i * dim * dim]); 1299dfccc68fSToby Isaac if (invJ) {DMPlex_Invert2D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);} 1300dfccc68fSToby Isaac } 1301dfccc68fSToby Isaac } 1302dfccc68fSToby Isaac } 1303dfccc68fSToby Isaac } 130499dec3a6SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr); 1305ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 1306ccd2543fSMatthew G Knepley } 1307ccd2543fSMatthew G Knepley 1308ccd2543fSMatthew G Knepley static PetscErrorCode DMPlexComputeTetrahedronGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 1309ccd2543fSMatthew G Knepley { 1310ccd2543fSMatthew G Knepley PetscSection coordSection; 1311ccd2543fSMatthew G Knepley Vec coordinates; 1312a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 1313ccd2543fSMatthew G Knepley const PetscInt dim = 3; 131499dec3a6SMatthew G. Knepley PetscInt d; 1315ccd2543fSMatthew G Knepley PetscErrorCode ierr; 1316ccd2543fSMatthew G Knepley 1317ccd2543fSMatthew G Knepley PetscFunctionBegin; 1318ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 131969d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 1320ccd2543fSMatthew G Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr); 13217f07f362SMatthew G. Knepley *detJ = 0.0; 13227f07f362SMatthew G. Knepley if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);} 1323ccd2543fSMatthew G Knepley if (J) { 1324ccd2543fSMatthew G Knepley for (d = 0; d < dim; d++) { 1325f0df753eSMatthew G. Knepley /* I orient with outward face normals */ 1326f0df753eSMatthew G. Knepley J[d*dim+0] = 0.5*(PetscRealPart(coords[2*dim+d]) - PetscRealPart(coords[0*dim+d])); 1327f0df753eSMatthew G. Knepley J[d*dim+1] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d])); 1328f0df753eSMatthew G. Knepley J[d*dim+2] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d])); 1329ccd2543fSMatthew G Knepley } 13303bc0b13bSBarry Smith ierr = PetscLogFlops(18.0);CHKERRQ(ierr); 1331923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(detJ, J); 1332ccd2543fSMatthew G Knepley } 1333923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);} 1334ccd2543fSMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr); 1335ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 1336ccd2543fSMatthew G Knepley } 1337ccd2543fSMatthew G Knepley 1338dfccc68fSToby Isaac static PetscErrorCode DMPlexComputeHexahedronGeometry_Internal(DM dm, PetscInt e, PetscInt Nq, const PetscReal points[], PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 1339ccd2543fSMatthew G Knepley { 1340ccd2543fSMatthew G Knepley PetscSection coordSection; 1341ccd2543fSMatthew G Knepley Vec coordinates; 1342a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 1343ccd2543fSMatthew G Knepley const PetscInt dim = 3; 1344ccd2543fSMatthew G Knepley PetscInt d; 1345ccd2543fSMatthew G Knepley PetscErrorCode ierr; 1346ccd2543fSMatthew G Knepley 1347ccd2543fSMatthew G Knepley PetscFunctionBegin; 1348ccd2543fSMatthew G Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 134969d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 1350ccd2543fSMatthew G Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr); 1351dfccc68fSToby Isaac if (!Nq) { 13527f07f362SMatthew G. Knepley *detJ = 0.0; 1353dfccc68fSToby Isaac if (v) {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);} 1354ccd2543fSMatthew G Knepley if (J) { 1355ccd2543fSMatthew G Knepley for (d = 0; d < dim; d++) { 1356f0df753eSMatthew G. Knepley J[d*dim+0] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d])); 1357f0df753eSMatthew G. Knepley J[d*dim+1] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d])); 1358f0df753eSMatthew G. Knepley J[d*dim+2] = 0.5*(PetscRealPart(coords[4*dim+d]) - PetscRealPart(coords[0*dim+d])); 1359ccd2543fSMatthew G Knepley } 13603bc0b13bSBarry Smith ierr = PetscLogFlops(18.0);CHKERRQ(ierr); 1361923591dfSMatthew G. Knepley DMPlex_Det3D_Internal(detJ, J); 1362ccd2543fSMatthew G Knepley } 1363923591dfSMatthew G. Knepley if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);} 1364dfccc68fSToby Isaac } else { 1365dfccc68fSToby Isaac const PetscInt Nv = 8; 1366dfccc68fSToby Isaac const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6}; 1367dfccc68fSToby Isaac const PetscInt dim = 3; 1368dfccc68fSToby Isaac const PetscInt dimR = 3; 1369dfccc68fSToby Isaac PetscReal zOrder[24]; 1370dfccc68fSToby Isaac PetscReal zCoeff[24]; 1371dfccc68fSToby Isaac PetscInt i, j, k, l; 1372dfccc68fSToby Isaac 1373dfccc68fSToby Isaac for (i = 0; i < Nv; i++) { 1374dfccc68fSToby Isaac PetscInt zi = zToPlex[i]; 1375dfccc68fSToby Isaac 1376dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1377dfccc68fSToby Isaac zOrder[dim * i + j] = PetscRealPart(coords[dim * zi + j]); 1378dfccc68fSToby Isaac } 1379dfccc68fSToby Isaac } 1380dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1381dfccc68fSToby Isaac zCoeff[dim * 0 + j] = 0.125 * ( zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j] + zOrder[dim * 4 + j] + zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1382dfccc68fSToby Isaac zCoeff[dim * 1 + j] = 0.125 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j] - zOrder[dim * 4 + j] + zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1383dfccc68fSToby Isaac zCoeff[dim * 2 + j] = 0.125 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j] - zOrder[dim * 4 + j] - zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1384dfccc68fSToby Isaac zCoeff[dim * 3 + j] = 0.125 * ( zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j] + zOrder[dim * 4 + j] - zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1385dfccc68fSToby Isaac zCoeff[dim * 4 + j] = 0.125 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] - zOrder[dim * 3 + j] + zOrder[dim * 4 + j] + zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1386dfccc68fSToby Isaac zCoeff[dim * 5 + j] = 0.125 * (+ zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] - zOrder[dim * 3 + j] - zOrder[dim * 4 + j] + zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1387dfccc68fSToby Isaac zCoeff[dim * 6 + j] = 0.125 * (+ zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] - zOrder[dim * 3 + j] - zOrder[dim * 4 + j] - zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1388dfccc68fSToby Isaac zCoeff[dim * 7 + j] = 0.125 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] - zOrder[dim * 3 + j] + zOrder[dim * 4 + j] - zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]); 1389dfccc68fSToby Isaac } 1390dfccc68fSToby Isaac for (i = 0; i < Nq; i++) { 1391dfccc68fSToby Isaac PetscReal xi = points[dimR * i], eta = points[dimR * i + 1], theta = points[dimR * i + 2]; 1392dfccc68fSToby Isaac 1393dfccc68fSToby Isaac if (v) { 139491d2b7ceSToby Isaac PetscReal extPoint[8]; 1395dfccc68fSToby Isaac 1396dfccc68fSToby Isaac extPoint[0] = 1.; 1397dfccc68fSToby Isaac extPoint[1] = xi; 1398dfccc68fSToby Isaac extPoint[2] = eta; 1399dfccc68fSToby Isaac extPoint[3] = xi * eta; 1400dfccc68fSToby Isaac extPoint[4] = theta; 1401dfccc68fSToby Isaac extPoint[5] = theta * xi; 1402dfccc68fSToby Isaac extPoint[6] = theta * eta; 1403dfccc68fSToby Isaac extPoint[7] = theta * eta * xi; 1404dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1405dfccc68fSToby Isaac PetscReal val = 0.; 1406dfccc68fSToby Isaac 1407dfccc68fSToby Isaac for (k = 0; k < Nv; k++) { 1408dfccc68fSToby Isaac val += extPoint[k] * zCoeff[dim * k + j]; 1409dfccc68fSToby Isaac } 1410dfccc68fSToby Isaac v[i * dim + j] = val; 1411dfccc68fSToby Isaac } 1412dfccc68fSToby Isaac } 1413dfccc68fSToby Isaac if (J) { 1414dfccc68fSToby Isaac PetscReal extJ[24]; 1415dfccc68fSToby Isaac 1416dfccc68fSToby Isaac extJ[0] = 0. ; extJ[1] = 0. ; extJ[2] = 0. ; 1417dfccc68fSToby Isaac extJ[3] = 1. ; extJ[4] = 0. ; extJ[5] = 0. ; 1418dfccc68fSToby Isaac extJ[6] = 0. ; extJ[7] = 1. ; extJ[8] = 0. ; 1419dfccc68fSToby Isaac extJ[9] = eta ; extJ[10] = xi ; extJ[11] = 0. ; 1420dfccc68fSToby Isaac extJ[12] = 0. ; extJ[13] = 0. ; extJ[14] = 1. ; 1421dfccc68fSToby Isaac extJ[15] = theta ; extJ[16] = 0. ; extJ[17] = xi ; 1422dfccc68fSToby Isaac extJ[18] = 0. ; extJ[19] = theta ; extJ[20] = eta ; 1423dfccc68fSToby Isaac extJ[21] = theta * eta; extJ[22] = theta * xi; extJ[23] = eta * xi; 1424dfccc68fSToby Isaac 1425dfccc68fSToby Isaac for (j = 0; j < dim; j++) { 1426dfccc68fSToby Isaac for (k = 0; k < dimR; k++) { 1427dfccc68fSToby Isaac PetscReal val = 0.; 1428dfccc68fSToby Isaac 1429dfccc68fSToby Isaac for (l = 0; l < Nv; l++) { 1430dfccc68fSToby Isaac val += zCoeff[dim * l + j] * extJ[dimR * l + k]; 1431dfccc68fSToby Isaac } 1432dfccc68fSToby Isaac J[i * dim * dim + dim * j + k] = val; 1433dfccc68fSToby Isaac } 1434dfccc68fSToby Isaac } 1435dfccc68fSToby Isaac DMPlex_Det3D_Internal(&detJ[i], &J[i * dim * dim]); 1436dfccc68fSToby Isaac if (invJ) {DMPlex_Invert3D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);} 1437dfccc68fSToby Isaac } 1438dfccc68fSToby Isaac } 1439dfccc68fSToby Isaac } 1440ccd2543fSMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr); 1441ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 1442ccd2543fSMatthew G Knepley } 1443ccd2543fSMatthew G Knepley 1444dfccc68fSToby Isaac static PetscErrorCode DMPlexComputeCellGeometryFEM_Implicit(DM dm, PetscInt cell, PetscQuadrature quad, PetscReal *v, PetscReal *J, PetscReal *invJ, PetscReal *detJ) 1445dfccc68fSToby Isaac { 1446dfccc68fSToby Isaac PetscInt depth, dim, coordDim, coneSize, i; 1447dfccc68fSToby Isaac PetscInt Nq = 0; 1448dfccc68fSToby Isaac const PetscReal *points = NULL; 1449dfccc68fSToby Isaac DMLabel depthLabel; 1450c330f8ffSToby Isaac PetscReal xi0[3] = {-1.,-1.,-1.}, v0[3], J0[9], detJ0; 1451dfccc68fSToby Isaac PetscBool isAffine = PETSC_TRUE; 1452dfccc68fSToby Isaac PetscErrorCode ierr; 1453dfccc68fSToby Isaac 1454dfccc68fSToby Isaac PetscFunctionBegin; 1455dfccc68fSToby Isaac ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 1456dfccc68fSToby Isaac ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr); 1457dfccc68fSToby Isaac ierr = DMPlexGetDepthLabel(dm, &depthLabel);CHKERRQ(ierr); 1458dfccc68fSToby Isaac ierr = DMLabelGetValue(depthLabel, cell, &dim);CHKERRQ(ierr); 1459dfccc68fSToby Isaac if (depth == 1 && dim == 1) { 1460dfccc68fSToby Isaac ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 1461dfccc68fSToby Isaac } 1462dfccc68fSToby Isaac ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 1463dfccc68fSToby Isaac if (coordDim > 3) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported coordinate dimension %D > 3", coordDim); 14649c3cf19fSMatthew G. Knepley if (quad) {ierr = PetscQuadratureGetData(quad, NULL, NULL, &Nq, &points, NULL);CHKERRQ(ierr);} 1465dfccc68fSToby Isaac switch (dim) { 1466dfccc68fSToby Isaac case 0: 1467dfccc68fSToby Isaac ierr = DMPlexComputePointGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr); 1468dfccc68fSToby Isaac isAffine = PETSC_FALSE; 1469dfccc68fSToby Isaac break; 1470dfccc68fSToby Isaac case 1: 14717318780aSToby Isaac if (Nq) { 14727318780aSToby Isaac ierr = DMPlexComputeLineGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr); 14737318780aSToby Isaac } else { 14747318780aSToby Isaac ierr = DMPlexComputeLineGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr); 14757318780aSToby Isaac } 1476dfccc68fSToby Isaac break; 1477dfccc68fSToby Isaac case 2: 1478dfccc68fSToby Isaac switch (coneSize) { 1479dfccc68fSToby Isaac case 3: 14807318780aSToby Isaac if (Nq) { 14817318780aSToby Isaac ierr = DMPlexComputeTriangleGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr); 14827318780aSToby Isaac } else { 14837318780aSToby Isaac ierr = DMPlexComputeTriangleGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr); 14847318780aSToby Isaac } 1485dfccc68fSToby Isaac break; 1486dfccc68fSToby Isaac case 4: 1487dfccc68fSToby Isaac ierr = DMPlexComputeRectangleGeometry_Internal(dm, cell, Nq, points, v, J, invJ, detJ);CHKERRQ(ierr); 1488dfccc68fSToby Isaac isAffine = PETSC_FALSE; 1489dfccc68fSToby Isaac break; 1490dfccc68fSToby Isaac default: 1491dfccc68fSToby Isaac SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported number of faces %D in cell %D for element geometry computation", coneSize, cell); 1492dfccc68fSToby Isaac } 1493dfccc68fSToby Isaac break; 1494dfccc68fSToby Isaac case 3: 1495dfccc68fSToby Isaac switch (coneSize) { 1496dfccc68fSToby Isaac case 4: 14977318780aSToby Isaac if (Nq) { 14987318780aSToby Isaac ierr = DMPlexComputeTetrahedronGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr); 14997318780aSToby Isaac } else { 15007318780aSToby Isaac ierr = DMPlexComputeTetrahedronGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr); 15017318780aSToby Isaac } 1502dfccc68fSToby Isaac break; 1503dfccc68fSToby Isaac case 6: /* Faces */ 1504dfccc68fSToby Isaac case 8: /* Vertices */ 1505dfccc68fSToby Isaac ierr = DMPlexComputeHexahedronGeometry_Internal(dm, cell, Nq, points, v, J, invJ, detJ);CHKERRQ(ierr); 1506dfccc68fSToby Isaac isAffine = PETSC_FALSE; 1507dfccc68fSToby Isaac break; 1508dfccc68fSToby Isaac default: 1509dfccc68fSToby Isaac SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported number of faces %D in cell %D for element geometry computation", coneSize, cell); 1510dfccc68fSToby Isaac } 1511dfccc68fSToby Isaac break; 1512dfccc68fSToby Isaac default: 1513dfccc68fSToby Isaac SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension %D for element geometry computation", dim); 1514dfccc68fSToby Isaac } 15157318780aSToby Isaac if (isAffine && Nq) { 1516dfccc68fSToby Isaac if (v) { 1517dfccc68fSToby Isaac for (i = 0; i < Nq; i++) { 1518c330f8ffSToby Isaac CoordinatesRefToReal(coordDim, dim, xi0, v0, J0, &points[dim * i], &v[coordDim * i]); 1519dfccc68fSToby Isaac } 1520dfccc68fSToby Isaac } 15217318780aSToby Isaac if (detJ) { 15227318780aSToby Isaac for (i = 0; i < Nq; i++) { 15237318780aSToby Isaac detJ[i] = detJ0; 1524dfccc68fSToby Isaac } 15257318780aSToby Isaac } 15267318780aSToby Isaac if (J) { 15277318780aSToby Isaac PetscInt k; 15287318780aSToby Isaac 15297318780aSToby Isaac for (i = 0, k = 0; i < Nq; i++) { 1530dfccc68fSToby Isaac PetscInt j; 1531dfccc68fSToby Isaac 15327318780aSToby Isaac for (j = 0; j < coordDim * coordDim; j++, k++) { 15337318780aSToby Isaac J[k] = J0[j]; 15347318780aSToby Isaac } 15357318780aSToby Isaac } 15367318780aSToby Isaac } 15377318780aSToby Isaac if (invJ) { 15387318780aSToby Isaac PetscInt k; 15397318780aSToby Isaac switch (coordDim) { 15407318780aSToby Isaac case 0: 15417318780aSToby Isaac break; 15427318780aSToby Isaac case 1: 15437318780aSToby Isaac invJ[0] = 1./J0[0]; 15447318780aSToby Isaac break; 15457318780aSToby Isaac case 2: 15467318780aSToby Isaac DMPlex_Invert2D_Internal(invJ, J0, detJ0); 15477318780aSToby Isaac break; 15487318780aSToby Isaac case 3: 15497318780aSToby Isaac DMPlex_Invert3D_Internal(invJ, J0, detJ0); 15507318780aSToby Isaac break; 15517318780aSToby Isaac } 15527318780aSToby Isaac for (i = 1, k = coordDim * coordDim; i < Nq; i++) { 15537318780aSToby Isaac PetscInt j; 15547318780aSToby Isaac 15557318780aSToby Isaac for (j = 0; j < coordDim * coordDim; j++, k++) { 15567318780aSToby Isaac invJ[k] = invJ[j]; 15577318780aSToby Isaac } 1558dfccc68fSToby Isaac } 1559dfccc68fSToby Isaac } 1560dfccc68fSToby Isaac } 1561dfccc68fSToby Isaac PetscFunctionReturn(0); 1562dfccc68fSToby Isaac } 1563dfccc68fSToby Isaac 1564ccd2543fSMatthew G Knepley /*@C 15658e0841e0SMatthew G. Knepley DMPlexComputeCellGeometryAffineFEM - Assuming an affine map, compute the Jacobian, inverse Jacobian, and Jacobian determinant for a given cell 1566ccd2543fSMatthew G Knepley 1567ccd2543fSMatthew G Knepley Collective on DM 1568ccd2543fSMatthew G Knepley 1569ccd2543fSMatthew G Knepley Input Arguments: 1570ccd2543fSMatthew G Knepley + dm - the DM 1571ccd2543fSMatthew G Knepley - cell - the cell 1572ccd2543fSMatthew G Knepley 1573ccd2543fSMatthew G Knepley Output Arguments: 1574ccd2543fSMatthew G Knepley + v0 - the translation part of this affine transform 1575ccd2543fSMatthew G Knepley . J - the Jacobian of the transform from the reference element 1576ccd2543fSMatthew G Knepley . invJ - the inverse of the Jacobian 1577ccd2543fSMatthew G Knepley - detJ - the Jacobian determinant 1578ccd2543fSMatthew G Knepley 1579ccd2543fSMatthew G Knepley Level: advanced 1580ccd2543fSMatthew G Knepley 1581ccd2543fSMatthew G Knepley Fortran Notes: 1582ccd2543fSMatthew G Knepley Since it returns arrays, this routine is only available in Fortran 90, and you must 1583ccd2543fSMatthew G Knepley include petsc.h90 in your code. 1584ccd2543fSMatthew G Knepley 1585e8964c0aSStefano Zampini .seealso: DMPlexComputeCellGeometryFEM(), DMGetCoordinateSection(), DMGetCoordinates() 1586ccd2543fSMatthew G Knepley @*/ 15878e0841e0SMatthew G. Knepley PetscErrorCode DMPlexComputeCellGeometryAffineFEM(DM dm, PetscInt cell, PetscReal *v0, PetscReal *J, PetscReal *invJ, PetscReal *detJ) 1588ccd2543fSMatthew G Knepley { 1589ccd2543fSMatthew G Knepley PetscErrorCode ierr; 1590ccd2543fSMatthew G Knepley 1591ccd2543fSMatthew G Knepley PetscFunctionBegin; 1592dfccc68fSToby Isaac ierr = DMPlexComputeCellGeometryFEM_Implicit(dm,cell,NULL,v0,J,invJ,detJ);CHKERRQ(ierr); 15938e0841e0SMatthew G. Knepley PetscFunctionReturn(0); 15948e0841e0SMatthew G. Knepley } 15958e0841e0SMatthew G. Knepley 1596dfccc68fSToby Isaac static PetscErrorCode DMPlexComputeCellGeometryFEM_FE(DM dm, PetscFE fe, PetscInt point, PetscQuadrature quad, PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ) 15978e0841e0SMatthew G. Knepley { 1598dfccc68fSToby Isaac PetscQuadrature feQuad; 15998e0841e0SMatthew G. Knepley PetscSection coordSection; 16008e0841e0SMatthew G. Knepley Vec coordinates; 16018e0841e0SMatthew G. Knepley PetscScalar *coords = NULL; 16028e0841e0SMatthew G. Knepley const PetscReal *quadPoints; 1603f960e424SToby Isaac PetscReal *basisDer, *basis, detJt; 1604f960e424SToby Isaac PetscInt dim, cdim, pdim, qdim, Nq, numCoords, q; 16058e0841e0SMatthew G. Knepley PetscErrorCode ierr; 16068e0841e0SMatthew G. Knepley 16078e0841e0SMatthew G. Knepley PetscFunctionBegin; 16088e0841e0SMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 16098e0841e0SMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 16108e0841e0SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, point, &numCoords, &coords);CHKERRQ(ierr); 16118e0841e0SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 16128e0841e0SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr); 1613dfccc68fSToby Isaac if (!quad) { /* use the first point of the first functional of the dual space */ 1614dfccc68fSToby Isaac PetscDualSpace dsp; 1615dfccc68fSToby Isaac 1616dfccc68fSToby Isaac ierr = PetscFEGetDualSpace(fe, &dsp);CHKERRQ(ierr); 1617dfccc68fSToby Isaac ierr = PetscDualSpaceGetFunctional(dsp, 0, &quad);CHKERRQ(ierr); 16189c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, &qdim, NULL, &Nq, &quadPoints, NULL);CHKERRQ(ierr); 1619dfccc68fSToby Isaac Nq = 1; 1620dfccc68fSToby Isaac } else { 16219c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, &qdim, NULL, &Nq, &quadPoints, NULL);CHKERRQ(ierr); 1622dfccc68fSToby Isaac } 162391d2b7ceSToby Isaac ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr); 1624dfccc68fSToby Isaac ierr = PetscFEGetQuadrature(fe, &feQuad);CHKERRQ(ierr); 1625dfccc68fSToby Isaac if (feQuad == quad) { 16268135c375SStefano Zampini ierr = PetscFEGetDefaultTabulation(fe, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr); 16278e0841e0SMatthew G. Knepley if (numCoords != pdim*cdim) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "There are %d coordinates for point %d != %d*%d", numCoords, point, pdim, cdim); 1628dfccc68fSToby Isaac } else { 16298135c375SStefano Zampini ierr = PetscFEGetTabulation(fe, Nq, quadPoints, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr); 1630dfccc68fSToby Isaac } 1631dfccc68fSToby Isaac if (qdim != dim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Point dimension %d != quadrature dimension %d", dim, qdim); 1632dfccc68fSToby Isaac if (v) { 1633dfccc68fSToby Isaac ierr = PetscMemzero(v, Nq*cdim*sizeof(PetscReal));CHKERRQ(ierr); 1634f960e424SToby Isaac for (q = 0; q < Nq; ++q) { 1635f960e424SToby Isaac PetscInt i, k; 1636f960e424SToby Isaac 1637f960e424SToby Isaac for (k = 0; k < pdim; ++k) 1638f960e424SToby Isaac for (i = 0; i < cdim; ++i) 1639dfccc68fSToby Isaac v[q*cdim + i] += basis[q*pdim + k] * PetscRealPart(coords[k*cdim + i]); 1640f960e424SToby Isaac ierr = PetscLogFlops(2.0*pdim*cdim);CHKERRQ(ierr); 1641f960e424SToby Isaac } 1642f960e424SToby Isaac } 16438e0841e0SMatthew G. Knepley if (J) { 1644dfccc68fSToby Isaac ierr = PetscMemzero(J, Nq*cdim*cdim*sizeof(PetscReal));CHKERRQ(ierr); 16458e0841e0SMatthew G. Knepley for (q = 0; q < Nq; ++q) { 16468e0841e0SMatthew G. Knepley PetscInt i, j, k, c, r; 16478e0841e0SMatthew G. Knepley 16488e0841e0SMatthew G. Knepley /* J = dx_i/d\xi_j = sum[k=0,n-1] dN_k/d\xi_j * x_i(k) */ 16498e0841e0SMatthew G. Knepley for (k = 0; k < pdim; ++k) 16508e0841e0SMatthew G. Knepley for (j = 0; j < dim; ++j) 16518e0841e0SMatthew G. Knepley for (i = 0; i < cdim; ++i) 1652dfccc68fSToby Isaac J[(q*cdim + i)*cdim + j] += basisDer[(q*pdim + k)*dim + j] * PetscRealPart(coords[k*cdim + i]); 16533bc0b13bSBarry Smith ierr = PetscLogFlops(2.0*pdim*dim*cdim);CHKERRQ(ierr); 16548e0841e0SMatthew G. Knepley if (cdim > dim) { 16558e0841e0SMatthew G. Knepley for (c = dim; c < cdim; ++c) 16568e0841e0SMatthew G. Knepley for (r = 0; r < cdim; ++r) 16578e0841e0SMatthew G. Knepley J[r*cdim+c] = r == c ? 1.0 : 0.0; 16588e0841e0SMatthew G. Knepley } 1659f960e424SToby Isaac if (!detJ && !invJ) continue; 1660a63b72c6SToby Isaac detJt = 0.; 16618e0841e0SMatthew G. Knepley switch (cdim) { 16628e0841e0SMatthew G. Knepley case 3: 1663037dc194SToby Isaac DMPlex_Det3D_Internal(&detJt, &J[q*cdim*dim]); 1664037dc194SToby Isaac if (invJ) {DMPlex_Invert3D_Internal(&invJ[q*cdim*dim], &J[q*cdim*dim], detJt);} 166517fe8556SMatthew G. Knepley break; 166649dc4407SMatthew G. Knepley case 2: 16679f328543SToby Isaac DMPlex_Det2D_Internal(&detJt, &J[q*cdim*dim]); 1668037dc194SToby Isaac if (invJ) {DMPlex_Invert2D_Internal(&invJ[q*cdim*dim], &J[q*cdim*dim], detJt);} 166949dc4407SMatthew G. Knepley break; 16708e0841e0SMatthew G. Knepley case 1: 1671037dc194SToby Isaac detJt = J[q*cdim*dim]; 1672037dc194SToby Isaac if (invJ) invJ[q*cdim*dim] = 1.0/detJt; 167349dc4407SMatthew G. Knepley } 1674f960e424SToby Isaac if (detJ) detJ[q] = detJt; 167549dc4407SMatthew G. Knepley } 167649dc4407SMatthew G. Knepley } 1677037dc194SToby Isaac else if (detJ || invJ) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Need J to compute invJ or detJ"); 1678dfccc68fSToby Isaac if (feQuad != quad) { 16798135c375SStefano Zampini ierr = PetscFERestoreTabulation(fe, Nq, quadPoints, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr); 1680dfccc68fSToby Isaac } 16818e0841e0SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, point, &numCoords, &coords);CHKERRQ(ierr); 16828e0841e0SMatthew G. Knepley PetscFunctionReturn(0); 16838e0841e0SMatthew G. Knepley } 16848e0841e0SMatthew G. Knepley 16858e0841e0SMatthew G. Knepley /*@C 16868e0841e0SMatthew G. Knepley DMPlexComputeCellGeometryFEM - Compute the Jacobian, inverse Jacobian, and Jacobian determinant at each quadrature point in the given cell 16878e0841e0SMatthew G. Knepley 16888e0841e0SMatthew G. Knepley Collective on DM 16898e0841e0SMatthew G. Knepley 16908e0841e0SMatthew G. Knepley Input Arguments: 16918e0841e0SMatthew G. Knepley + dm - the DM 16928e0841e0SMatthew G. Knepley . cell - the cell 1693dfccc68fSToby Isaac - quad - the quadrature containing the points in the reference element where the geometry will be evaluated. If quad == NULL, geometry will be 1694dfccc68fSToby Isaac evaluated at the first vertex of the reference element 16958e0841e0SMatthew G. Knepley 16968e0841e0SMatthew G. Knepley Output Arguments: 1697dfccc68fSToby Isaac + v - the image of the transformed quadrature points, otherwise the image of the first vertex in the closure of the reference element 16988e0841e0SMatthew G. Knepley . J - the Jacobian of the transform from the reference element at each quadrature point 16998e0841e0SMatthew G. Knepley . invJ - the inverse of the Jacobian at each quadrature point 17008e0841e0SMatthew G. Knepley - detJ - the Jacobian determinant at each quadrature point 17018e0841e0SMatthew G. Knepley 17028e0841e0SMatthew G. Knepley Level: advanced 17038e0841e0SMatthew G. Knepley 17048e0841e0SMatthew G. Knepley Fortran Notes: 17058e0841e0SMatthew G. Knepley Since it returns arrays, this routine is only available in Fortran 90, and you must 17068e0841e0SMatthew G. Knepley include petsc.h90 in your code. 17078e0841e0SMatthew G. Knepley 1708e8964c0aSStefano Zampini .seealso: DMGetCoordinateSection(), DMGetCoordinates() 17098e0841e0SMatthew G. Knepley @*/ 1710dfccc68fSToby Isaac PetscErrorCode DMPlexComputeCellGeometryFEM(DM dm, PetscInt cell, PetscQuadrature quad, PetscReal *v, PetscReal *J, PetscReal *invJ, PetscReal *detJ) 17118e0841e0SMatthew G. Knepley { 1712bb4a5db5SMatthew G. Knepley DM cdm; 1713dfccc68fSToby Isaac PetscFE fe = NULL; 17148e0841e0SMatthew G. Knepley PetscErrorCode ierr; 17158e0841e0SMatthew G. Knepley 17168e0841e0SMatthew G. Knepley PetscFunctionBegin; 17172d89661fSMatthew G. Knepley PetscValidPointer(detJ, 7); 1718bb4a5db5SMatthew G. Knepley ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr); 1719bb4a5db5SMatthew G. Knepley if (cdm) { 1720dfccc68fSToby Isaac PetscClassId id; 1721dfccc68fSToby Isaac PetscInt numFields; 1722e5e52638SMatthew G. Knepley PetscDS prob; 1723dfccc68fSToby Isaac PetscObject disc; 1724dfccc68fSToby Isaac 1725bb4a5db5SMatthew G. Knepley ierr = DMGetNumFields(cdm, &numFields);CHKERRQ(ierr); 1726dfccc68fSToby Isaac if (numFields) { 1727bb4a5db5SMatthew G. Knepley ierr = DMGetDS(cdm, &prob);CHKERRQ(ierr); 1728dfccc68fSToby Isaac ierr = PetscDSGetDiscretization(prob,0,&disc);CHKERRQ(ierr); 1729dfccc68fSToby Isaac ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr); 1730dfccc68fSToby Isaac if (id == PETSCFE_CLASSID) { 1731dfccc68fSToby Isaac fe = (PetscFE) disc; 1732dfccc68fSToby Isaac } 1733dfccc68fSToby Isaac } 1734dfccc68fSToby Isaac } 1735dfccc68fSToby Isaac if (!fe) {ierr = DMPlexComputeCellGeometryFEM_Implicit(dm, cell, quad, v, J, invJ, detJ);CHKERRQ(ierr);} 1736dfccc68fSToby Isaac else {ierr = DMPlexComputeCellGeometryFEM_FE(dm, fe, cell, quad, v, J, invJ, detJ);CHKERRQ(ierr);} 1737ccd2543fSMatthew G Knepley PetscFunctionReturn(0); 1738ccd2543fSMatthew G Knepley } 1739834e62ceSMatthew G. Knepley 1740011ea5d8SMatthew G. Knepley static PetscErrorCode DMPlexComputeGeometryFVM_1D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[]) 1741cc08537eSMatthew G. Knepley { 1742cc08537eSMatthew G. Knepley PetscSection coordSection; 1743cc08537eSMatthew G. Knepley Vec coordinates; 1744a1e44745SMatthew G. Knepley PetscScalar *coords = NULL; 174506e2781eSMatthew G. Knepley PetscScalar tmp[2]; 1746cc08537eSMatthew G. Knepley PetscInt coordSize; 1747cc08537eSMatthew G. Knepley PetscErrorCode ierr; 1748cc08537eSMatthew G. Knepley 1749cc08537eSMatthew G. Knepley PetscFunctionBegin; 1750cc08537eSMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 175169d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 1752cc08537eSMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr); 1753011ea5d8SMatthew G. Knepley if (dim != 2) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "We only support 2D edges right now"); 17542e17dfb7SMatthew G. Knepley ierr = DMLocalizeCoordinate_Internal(dm, dim, coords, &coords[dim], tmp);CHKERRQ(ierr); 1755cc08537eSMatthew G. Knepley if (centroid) { 175606e2781eSMatthew G. Knepley centroid[0] = 0.5*PetscRealPart(coords[0] + tmp[0]); 175706e2781eSMatthew G. Knepley centroid[1] = 0.5*PetscRealPart(coords[1] + tmp[1]); 1758cc08537eSMatthew G. Knepley } 1759cc08537eSMatthew G. Knepley if (normal) { 1760a60a936bSMatthew G. Knepley PetscReal norm; 1761a60a936bSMatthew G. Knepley 176206e2781eSMatthew G. Knepley normal[0] = -PetscRealPart(coords[1] - tmp[1]); 176306e2781eSMatthew G. Knepley normal[1] = PetscRealPart(coords[0] - tmp[0]); 1764a60a936bSMatthew G. Knepley norm = PetscSqrtReal(normal[0]*normal[0] + normal[1]*normal[1]); 1765a60a936bSMatthew G. Knepley normal[0] /= norm; 1766a60a936bSMatthew G. Knepley normal[1] /= norm; 1767cc08537eSMatthew G. Knepley } 1768cc08537eSMatthew G. Knepley if (vol) { 176906e2781eSMatthew G. Knepley *vol = PetscSqrtReal(PetscSqr(PetscRealPart(coords[0] - tmp[0])) + PetscSqr(PetscRealPart(coords[1] - tmp[1]))); 1770cc08537eSMatthew G. Knepley } 1771cc08537eSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr); 1772cc08537eSMatthew G. Knepley PetscFunctionReturn(0); 1773cc08537eSMatthew G. Knepley } 1774cc08537eSMatthew G. Knepley 1775cc08537eSMatthew G. Knepley /* Centroid_i = (\sum_n A_n Cn_i ) / A */ 1776011ea5d8SMatthew G. Knepley static PetscErrorCode DMPlexComputeGeometryFVM_2D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[]) 1777cc08537eSMatthew G. Knepley { 1778*793a2a13SMatthew G. Knepley DMLabel depth; 1779cc08537eSMatthew G. Knepley PetscSection coordSection; 1780cc08537eSMatthew G. Knepley Vec coordinates; 1781cc08537eSMatthew G. Knepley PetscScalar *coords = NULL; 17820a1d6728SMatthew G. Knepley PetscReal vsum = 0.0, csum[3] = {0.0, 0.0, 0.0}, vtmp, ctmp[4], v0[3], R[9]; 1783*793a2a13SMatthew G. Knepley PetscBool isHybrid = PETSC_FALSE; 1784*793a2a13SMatthew G. Knepley PetscInt fv[4] = {0, 1, 2, 3}; 1785*793a2a13SMatthew G. Knepley PetscInt cdepth, pEndInterior[4], tdim = 2, coordSize, numCorners, p, d, e; 1786cc08537eSMatthew G. Knepley PetscErrorCode ierr; 1787cc08537eSMatthew G. Knepley 1788cc08537eSMatthew G. Knepley PetscFunctionBegin; 1789*793a2a13SMatthew G. Knepley /* Must check for hybrid cells because prisms have a different orientation scheme */ 1790*793a2a13SMatthew G. Knepley ierr = DMPlexGetDepthLabel(dm, &depth);CHKERRQ(ierr); 1791*793a2a13SMatthew G. Knepley ierr = DMLabelGetValue(depth, cell, &cdepth);CHKERRQ(ierr); 1792*793a2a13SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &pEndInterior[3], &pEndInterior[2], &pEndInterior[1], &pEndInterior[0]); 1793*793a2a13SMatthew G. Knepley if ((pEndInterior[cdepth]) >=0 && (cell >= pEndInterior[cdepth])) isHybrid = PETSC_TRUE; 1794*793a2a13SMatthew G. Knepley 1795cc08537eSMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 17960a1d6728SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cell, &numCorners);CHKERRQ(ierr); 179769d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 1798cc08537eSMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr); 17990bce18caSMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr); 1800*793a2a13SMatthew G. Knepley /* Side faces for hybrid cells are are stored as tensor products */ 1801*793a2a13SMatthew G. Knepley if (isHybrid && numCorners == 4) {fv[2] = 3; fv[3] = 2;} 1802*793a2a13SMatthew G. Knepley 1803ceee4971SMatthew G. Knepley if (dim > 2 && centroid) { 1804ceee4971SMatthew G. Knepley v0[0] = PetscRealPart(coords[0]); 1805ceee4971SMatthew G. Knepley v0[1] = PetscRealPart(coords[1]); 1806ceee4971SMatthew G. Knepley v0[2] = PetscRealPart(coords[2]); 1807ceee4971SMatthew G. Knepley } 1808011ea5d8SMatthew G. Knepley if (normal) { 1809011ea5d8SMatthew G. Knepley if (dim > 2) { 1810*793a2a13SMatthew G. Knepley const PetscReal x0 = PetscRealPart(coords[dim*fv[1]+0] - coords[0]), x1 = PetscRealPart(coords[dim*fv[2]+0] - coords[0]); 1811*793a2a13SMatthew G. Knepley const PetscReal y0 = PetscRealPart(coords[dim*fv[1]+1] - coords[1]), y1 = PetscRealPart(coords[dim*fv[2]+1] - coords[1]); 1812*793a2a13SMatthew G. Knepley const PetscReal z0 = PetscRealPart(coords[dim*fv[1]+2] - coords[2]), z1 = PetscRealPart(coords[dim*fv[2]+2] - coords[2]); 18130a1d6728SMatthew G. Knepley PetscReal norm; 18140a1d6728SMatthew G. Knepley 18150a1d6728SMatthew G. Knepley normal[0] = y0*z1 - z0*y1; 18160a1d6728SMatthew G. Knepley normal[1] = z0*x1 - x0*z1; 18170a1d6728SMatthew G. Knepley normal[2] = x0*y1 - y0*x1; 18188b49ba18SBarry Smith norm = PetscSqrtReal(normal[0]*normal[0] + normal[1]*normal[1] + normal[2]*normal[2]); 18190a1d6728SMatthew G. Knepley normal[0] /= norm; 18200a1d6728SMatthew G. Knepley normal[1] /= norm; 18210a1d6728SMatthew G. Knepley normal[2] /= norm; 1822011ea5d8SMatthew G. Knepley } else { 1823011ea5d8SMatthew G. Knepley for (d = 0; d < dim; ++d) normal[d] = 0.0; 1824011ea5d8SMatthew G. Knepley } 1825011ea5d8SMatthew G. Knepley } 1826741bfc07SMatthew G. Knepley if (dim == 3) {ierr = DMPlexComputeProjection3Dto2D(coordSize, coords, R);CHKERRQ(ierr);} 18270a1d6728SMatthew G. Knepley for (p = 0; p < numCorners; ++p) { 1828*793a2a13SMatthew G. Knepley const PetscInt pi = p < 4 ? fv[p] : p; 1829*793a2a13SMatthew G. Knepley const PetscInt pin = p < 3 ? fv[(p+1)%numCorners] : (p+1)%numCorners; 18300a1d6728SMatthew G. Knepley /* Need to do this copy to get types right */ 18310a1d6728SMatthew G. Knepley for (d = 0; d < tdim; ++d) { 1832*793a2a13SMatthew G. Knepley ctmp[d] = PetscRealPart(coords[pi*tdim+d]); 1833*793a2a13SMatthew G. Knepley ctmp[tdim+d] = PetscRealPart(coords[pin*tdim+d]); 18340a1d6728SMatthew G. Knepley } 18350a1d6728SMatthew G. Knepley Volume_Triangle_Origin_Internal(&vtmp, ctmp); 18360a1d6728SMatthew G. Knepley vsum += vtmp; 18370a1d6728SMatthew G. Knepley for (d = 0; d < tdim; ++d) { 18380a1d6728SMatthew G. Knepley csum[d] += (ctmp[d] + ctmp[tdim+d])*vtmp; 18390a1d6728SMatthew G. Knepley } 18400a1d6728SMatthew G. Knepley } 18410a1d6728SMatthew G. Knepley for (d = 0; d < tdim; ++d) { 18420a1d6728SMatthew G. Knepley csum[d] /= (tdim+1)*vsum; 18430a1d6728SMatthew G. Knepley } 18440a1d6728SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr); 1845ee6bbdb2SSatish Balay if (vol) *vol = PetscAbsReal(vsum); 18460a1d6728SMatthew G. Knepley if (centroid) { 18470a1d6728SMatthew G. Knepley if (dim > 2) { 18480a1d6728SMatthew G. Knepley for (d = 0; d < dim; ++d) { 18490a1d6728SMatthew G. Knepley centroid[d] = v0[d]; 18500a1d6728SMatthew G. Knepley for (e = 0; e < dim; ++e) { 18510a1d6728SMatthew G. Knepley centroid[d] += R[d*dim+e]*csum[e]; 18520a1d6728SMatthew G. Knepley } 18530a1d6728SMatthew G. Knepley } 18540a1d6728SMatthew G. Knepley } else for (d = 0; d < dim; ++d) centroid[d] = csum[d]; 18550a1d6728SMatthew G. Knepley } 1856cc08537eSMatthew G. Knepley PetscFunctionReturn(0); 1857cc08537eSMatthew G. Knepley } 1858cc08537eSMatthew G. Knepley 18590ec8681fSMatthew G. Knepley /* Centroid_i = (\sum_n V_n Cn_i ) / V */ 1860011ea5d8SMatthew G. Knepley static PetscErrorCode DMPlexComputeGeometryFVM_3D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[]) 18610ec8681fSMatthew G. Knepley { 1862*793a2a13SMatthew G. Knepley DMLabel depth; 18630ec8681fSMatthew G. Knepley PetscSection coordSection; 18640ec8681fSMatthew G. Knepley Vec coordinates; 18650ec8681fSMatthew G. Knepley PetscScalar *coords = NULL; 186686623015SMatthew G. Knepley PetscReal vsum = 0.0, vtmp, coordsTmp[3*3]; 1867a7df9edeSMatthew G. Knepley const PetscInt *faces, *facesO; 1868*793a2a13SMatthew G. Knepley PetscBool isHybrid = PETSC_FALSE; 1869*793a2a13SMatthew G. Knepley PetscInt pEndInterior[4], cdepth, numFaces, f, coordSize, numCorners, p, d; 18700ec8681fSMatthew G. Knepley PetscErrorCode ierr; 18710ec8681fSMatthew G. Knepley 18720ec8681fSMatthew G. Knepley PetscFunctionBegin; 1873f6dae198SJed Brown if (PetscUnlikely(dim > 3)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"No support for dim %D > 3",dim); 1874*793a2a13SMatthew G. Knepley /* Must check for hybrid cells because prisms have a different orientation scheme */ 1875*793a2a13SMatthew G. Knepley ierr = DMPlexGetDepthLabel(dm, &depth);CHKERRQ(ierr); 1876*793a2a13SMatthew G. Knepley ierr = DMLabelGetValue(depth, cell, &cdepth);CHKERRQ(ierr); 1877*793a2a13SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &pEndInterior[3], &pEndInterior[2], &pEndInterior[1], &pEndInterior[0]); 1878*793a2a13SMatthew G. Knepley if ((pEndInterior[cdepth]) >=0 && (cell >= pEndInterior[cdepth])) isHybrid = PETSC_TRUE; 1879*793a2a13SMatthew G. Knepley 18800ec8681fSMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 188169d8a9ceSMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 18820ec8681fSMatthew G. Knepley 1883d9a81ebdSMatthew G. Knepley if (centroid) for (d = 0; d < dim; ++d) centroid[d] = 0.0; 18840ec8681fSMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cell, &numFaces);CHKERRQ(ierr); 18850ec8681fSMatthew G. Knepley ierr = DMPlexGetCone(dm, cell, &faces);CHKERRQ(ierr); 1886a7df9edeSMatthew G. Knepley ierr = DMPlexGetConeOrientation(dm, cell, &facesO);CHKERRQ(ierr); 18870ec8681fSMatthew G. Knepley for (f = 0; f < numFaces; ++f) { 1888*793a2a13SMatthew G. Knepley PetscBool flip = isHybrid && f == 0 ? PETSC_TRUE : PETSC_FALSE; /* The first hybrid face is reversed */ 1889*793a2a13SMatthew G. Knepley 1890011ea5d8SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, faces[f], &coordSize, &coords);CHKERRQ(ierr); 18910ec8681fSMatthew G. Knepley numCorners = coordSize/dim; 18920ec8681fSMatthew G. Knepley switch (numCorners) { 18930ec8681fSMatthew G. Knepley case 3: 18940ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 18951ee9d5ecSMatthew G. Knepley coordsTmp[0*dim+d] = PetscRealPart(coords[0*dim+d]); 18961ee9d5ecSMatthew G. Knepley coordsTmp[1*dim+d] = PetscRealPart(coords[1*dim+d]); 18971ee9d5ecSMatthew G. Knepley coordsTmp[2*dim+d] = PetscRealPart(coords[2*dim+d]); 18980ec8681fSMatthew G. Knepley } 18990ec8681fSMatthew G. Knepley Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp); 1900*793a2a13SMatthew G. Knepley if (facesO[f] < 0 || flip) vtmp = -vtmp; 19010ec8681fSMatthew G. Knepley vsum += vtmp; 19024f25033aSJed Brown if (centroid) { /* Centroid of OABC = (a+b+c)/4 */ 19030ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 19041ee9d5ecSMatthew G. Knepley for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp; 19050ec8681fSMatthew G. Knepley } 19060ec8681fSMatthew G. Knepley } 19070ec8681fSMatthew G. Knepley break; 19080ec8681fSMatthew G. Knepley case 4: 1909*793a2a13SMatthew G. Knepley { 1910*793a2a13SMatthew G. Knepley PetscInt fv[4] = {0, 1, 2, 3}; 1911*793a2a13SMatthew G. Knepley 1912*793a2a13SMatthew G. Knepley /* Side faces for hybrid cells are are stored as tensor products */ 1913*793a2a13SMatthew G. Knepley if (isHybrid && f > 1) {fv[2] = 3; fv[3] = 2;} 19140ec8681fSMatthew G. Knepley /* DO FOR PYRAMID */ 19150ec8681fSMatthew G. Knepley /* First tet */ 19160ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 1917*793a2a13SMatthew G. Knepley coordsTmp[0*dim+d] = PetscRealPart(coords[fv[0]*dim+d]); 1918*793a2a13SMatthew G. Knepley coordsTmp[1*dim+d] = PetscRealPart(coords[fv[1]*dim+d]); 1919*793a2a13SMatthew G. Knepley coordsTmp[2*dim+d] = PetscRealPart(coords[fv[3]*dim+d]); 19200ec8681fSMatthew G. Knepley } 19210ec8681fSMatthew G. Knepley Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp); 1922*793a2a13SMatthew G. Knepley if (facesO[f] < 0 || flip) vtmp = -vtmp; 19230ec8681fSMatthew G. Knepley vsum += vtmp; 19240ec8681fSMatthew G. Knepley if (centroid) { 19250ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 19260ec8681fSMatthew G. Knepley for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp; 19270ec8681fSMatthew G. Knepley } 19280ec8681fSMatthew G. Knepley } 19290ec8681fSMatthew G. Knepley /* Second tet */ 19300ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 1931*793a2a13SMatthew G. Knepley coordsTmp[0*dim+d] = PetscRealPart(coords[fv[1]*dim+d]); 1932*793a2a13SMatthew G. Knepley coordsTmp[1*dim+d] = PetscRealPart(coords[fv[2]*dim+d]); 1933*793a2a13SMatthew G. Knepley coordsTmp[2*dim+d] = PetscRealPart(coords[fv[3]*dim+d]); 19340ec8681fSMatthew G. Knepley } 19350ec8681fSMatthew G. Knepley Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp); 1936*793a2a13SMatthew G. Knepley if (facesO[f] < 0 || flip) vtmp = -vtmp; 19370ec8681fSMatthew G. Knepley vsum += vtmp; 19380ec8681fSMatthew G. Knepley if (centroid) { 19390ec8681fSMatthew G. Knepley for (d = 0; d < dim; ++d) { 19400ec8681fSMatthew G. Knepley for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp; 19410ec8681fSMatthew G. Knepley } 19420ec8681fSMatthew G. Knepley } 19430ec8681fSMatthew G. Knepley break; 1944*793a2a13SMatthew G. Knepley } 19450ec8681fSMatthew G. Knepley default: 1946796f034aSJed Brown SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle faces with %D vertices", numCorners); 19470ec8681fSMatthew G. Knepley } 19484f25033aSJed Brown ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, faces[f], &coordSize, &coords);CHKERRQ(ierr); 19490ec8681fSMatthew G. Knepley } 19508763be8eSMatthew G. Knepley if (vol) *vol = PetscAbsReal(vsum); 19510ec8681fSMatthew G. Knepley if (normal) for (d = 0; d < dim; ++d) normal[d] = 0.0; 1952d9a81ebdSMatthew G. Knepley if (centroid) for (d = 0; d < dim; ++d) centroid[d] /= (vsum*4); 19530ec8681fSMatthew G. Knepley PetscFunctionReturn(0); 19540ec8681fSMatthew G. Knepley } 19550ec8681fSMatthew G. Knepley 1956834e62ceSMatthew G. Knepley /*@C 1957834e62ceSMatthew G. Knepley DMPlexComputeCellGeometryFVM - Compute the volume for a given cell 1958834e62ceSMatthew G. Knepley 1959834e62ceSMatthew G. Knepley Collective on DM 1960834e62ceSMatthew G. Knepley 1961834e62ceSMatthew G. Knepley Input Arguments: 1962834e62ceSMatthew G. Knepley + dm - the DM 1963834e62ceSMatthew G. Knepley - cell - the cell 1964834e62ceSMatthew G. Knepley 1965834e62ceSMatthew G. Knepley Output Arguments: 1966834e62ceSMatthew G. Knepley + volume - the cell volume 1967cc08537eSMatthew G. Knepley . centroid - the cell centroid 1968cc08537eSMatthew G. Knepley - normal - the cell normal, if appropriate 1969834e62ceSMatthew G. Knepley 1970834e62ceSMatthew G. Knepley Level: advanced 1971834e62ceSMatthew G. Knepley 1972834e62ceSMatthew G. Knepley Fortran Notes: 1973834e62ceSMatthew G. Knepley Since it returns arrays, this routine is only available in Fortran 90, and you must 1974834e62ceSMatthew G. Knepley include petsc.h90 in your code. 1975834e62ceSMatthew G. Knepley 1976e8964c0aSStefano Zampini .seealso: DMGetCoordinateSection(), DMGetCoordinates() 1977834e62ceSMatthew G. Knepley @*/ 1978cc08537eSMatthew G. Knepley PetscErrorCode DMPlexComputeCellGeometryFVM(DM dm, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[]) 1979834e62ceSMatthew G. Knepley { 19800ec8681fSMatthew G. Knepley PetscInt depth, dim; 1981834e62ceSMatthew G. Knepley PetscErrorCode ierr; 1982834e62ceSMatthew G. Knepley 1983834e62ceSMatthew G. Knepley PetscFunctionBegin; 1984834e62ceSMatthew G. Knepley ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 1985c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 1986834e62ceSMatthew G. Knepley if (depth != dim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh must be interpolated"); 1987834e62ceSMatthew G. Knepley /* We need to keep a pointer to the depth label */ 1988c58f1c22SToby Isaac ierr = DMGetLabelValue(dm, "depth", cell, &depth);CHKERRQ(ierr); 1989834e62ceSMatthew G. Knepley /* Cone size is now the number of faces */ 1990011ea5d8SMatthew G. Knepley switch (depth) { 1991cc08537eSMatthew G. Knepley case 1: 1992011ea5d8SMatthew G. Knepley ierr = DMPlexComputeGeometryFVM_1D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr); 1993cc08537eSMatthew G. Knepley break; 1994834e62ceSMatthew G. Knepley case 2: 1995011ea5d8SMatthew G. Knepley ierr = DMPlexComputeGeometryFVM_2D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr); 1996834e62ceSMatthew G. Knepley break; 1997834e62ceSMatthew G. Knepley case 3: 1998011ea5d8SMatthew G. Knepley ierr = DMPlexComputeGeometryFVM_3D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr); 1999834e62ceSMatthew G. Knepley break; 2000834e62ceSMatthew G. Knepley default: 2001b81cf158SMatthew G. Knepley SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension %D (depth %D) for element geometry computation", dim, depth); 2002834e62ceSMatthew G. Knepley } 2003834e62ceSMatthew G. Knepley PetscFunctionReturn(0); 2004834e62ceSMatthew G. Knepley } 2005113c68e6SMatthew G. Knepley 2006c501906fSMatthew G. Knepley /*@ 2007c501906fSMatthew G. Knepley DMPlexComputeGeometryFEM - Precompute cell geometry for the entire mesh 2008c501906fSMatthew G. Knepley 2009c501906fSMatthew G. Knepley Collective on dm 2010c501906fSMatthew G. Knepley 2011c501906fSMatthew G. Knepley Input Parameter: 2012c501906fSMatthew G. Knepley . dm - The DMPlex 2013c501906fSMatthew G. Knepley 2014c501906fSMatthew G. Knepley Output Parameter: 2015c501906fSMatthew G. Knepley . cellgeom - A vector with the cell geometry data for each cell 2016c501906fSMatthew G. Knepley 2017c501906fSMatthew G. Knepley Level: beginner 2018c501906fSMatthew G. Knepley 2019c501906fSMatthew G. Knepley .keywords: DMPlexComputeCellGeometryFEM() 2020c501906fSMatthew G. Knepley @*/ 2021c0d900a5SMatthew G. Knepley PetscErrorCode DMPlexComputeGeometryFEM(DM dm, Vec *cellgeom) 2022c0d900a5SMatthew G. Knepley { 2023c0d900a5SMatthew G. Knepley DM dmCell; 2024c0d900a5SMatthew G. Knepley Vec coordinates; 2025c0d900a5SMatthew G. Knepley PetscSection coordSection, sectionCell; 2026c0d900a5SMatthew G. Knepley PetscScalar *cgeom; 2027c0d900a5SMatthew G. Knepley PetscInt cStart, cEnd, cMax, c; 2028c0d900a5SMatthew G. Knepley PetscErrorCode ierr; 2029c0d900a5SMatthew G. Knepley 2030c0d900a5SMatthew G. Knepley PetscFunctionBegin; 2031c0d900a5SMatthew G. Knepley ierr = DMClone(dm, &dmCell);CHKERRQ(ierr); 2032c0d900a5SMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 2033c0d900a5SMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 2034c0d900a5SMatthew G. Knepley ierr = DMSetCoordinateSection(dmCell, PETSC_DETERMINE, coordSection);CHKERRQ(ierr); 2035c0d900a5SMatthew G. Knepley ierr = DMSetCoordinatesLocal(dmCell, coordinates);CHKERRQ(ierr); 2036c0d900a5SMatthew G. Knepley ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), §ionCell);CHKERRQ(ierr); 2037c0d900a5SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 2038c0d900a5SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr); 2039c0d900a5SMatthew G. Knepley cEnd = cMax < 0 ? cEnd : cMax; 2040c0d900a5SMatthew G. Knepley ierr = PetscSectionSetChart(sectionCell, cStart, cEnd);CHKERRQ(ierr); 2041c0d900a5SMatthew G. Knepley /* TODO This needs to be multiplied by Nq for non-affine */ 2042cf0b7c11SKarl Rupp for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionCell, c, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFEGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);} 2043c0d900a5SMatthew G. Knepley ierr = PetscSectionSetUp(sectionCell);CHKERRQ(ierr); 2044e87a4003SBarry Smith ierr = DMSetSection(dmCell, sectionCell);CHKERRQ(ierr); 2045c0d900a5SMatthew G. Knepley ierr = PetscSectionDestroy(§ionCell);CHKERRQ(ierr); 2046c0d900a5SMatthew G. Knepley ierr = DMCreateLocalVector(dmCell, cellgeom);CHKERRQ(ierr); 2047c0d900a5SMatthew G. Knepley ierr = VecGetArray(*cellgeom, &cgeom);CHKERRQ(ierr); 2048c0d900a5SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 2049cf0b7c11SKarl Rupp PetscFEGeom *cg; 2050c0d900a5SMatthew G. Knepley 2051c0d900a5SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmCell, c, cgeom, &cg);CHKERRQ(ierr); 2052c0d900a5SMatthew G. Knepley ierr = PetscMemzero(cg, sizeof(*cg));CHKERRQ(ierr); 2053cf0b7c11SKarl Rupp ierr = DMPlexComputeCellGeometryFEM(dmCell, c, NULL, cg->v, cg->J, cg->invJ, cg->detJ);CHKERRQ(ierr); 2054cf0b7c11SKarl Rupp if (*cg->detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", cg->detJ, c); 2055c0d900a5SMatthew G. Knepley } 2056c0d900a5SMatthew G. Knepley PetscFunctionReturn(0); 2057c0d900a5SMatthew G. Knepley } 2058c0d900a5SMatthew G. Knepley 2059891a9168SMatthew G. Knepley /*@ 2060891a9168SMatthew G. Knepley DMPlexComputeGeometryFVM - Computes the cell and face geometry for a finite volume method 2061891a9168SMatthew G. Knepley 2062891a9168SMatthew G. Knepley Input Parameter: 2063891a9168SMatthew G. Knepley . dm - The DM 2064891a9168SMatthew G. Knepley 2065891a9168SMatthew G. Knepley Output Parameters: 2066891a9168SMatthew G. Knepley + cellgeom - A Vec of PetscFVCellGeom data 2067891a9168SMatthew G. Knepley . facegeom - A Vec of PetscFVFaceGeom data 2068891a9168SMatthew G. Knepley 2069891a9168SMatthew G. Knepley Level: developer 2070891a9168SMatthew G. Knepley 2071891a9168SMatthew G. Knepley .seealso: PetscFVFaceGeom, PetscFVCellGeom, DMPlexComputeGeometryFEM() 2072891a9168SMatthew G. Knepley @*/ 2073113c68e6SMatthew G. Knepley PetscErrorCode DMPlexComputeGeometryFVM(DM dm, Vec *cellgeom, Vec *facegeom) 2074113c68e6SMatthew G. Knepley { 2075113c68e6SMatthew G. Knepley DM dmFace, dmCell; 2076113c68e6SMatthew G. Knepley DMLabel ghostLabel; 2077113c68e6SMatthew G. Knepley PetscSection sectionFace, sectionCell; 2078113c68e6SMatthew G. Knepley PetscSection coordSection; 2079113c68e6SMatthew G. Knepley Vec coordinates; 2080113c68e6SMatthew G. Knepley PetscScalar *fgeom, *cgeom; 2081113c68e6SMatthew G. Knepley PetscReal minradius, gminradius; 2082113c68e6SMatthew G. Knepley PetscInt dim, cStart, cEnd, cEndInterior, c, fStart, fEnd, f; 2083113c68e6SMatthew G. Knepley PetscErrorCode ierr; 2084113c68e6SMatthew G. Knepley 2085113c68e6SMatthew G. Knepley PetscFunctionBegin; 2086113c68e6SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 2087113c68e6SMatthew G. Knepley ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr); 2088113c68e6SMatthew G. Knepley ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr); 2089113c68e6SMatthew G. Knepley /* Make cell centroids and volumes */ 2090113c68e6SMatthew G. Knepley ierr = DMClone(dm, &dmCell);CHKERRQ(ierr); 2091113c68e6SMatthew G. Knepley ierr = DMSetCoordinateSection(dmCell, PETSC_DETERMINE, coordSection);CHKERRQ(ierr); 2092113c68e6SMatthew G. Knepley ierr = DMSetCoordinatesLocal(dmCell, coordinates);CHKERRQ(ierr); 2093113c68e6SMatthew G. Knepley ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), §ionCell);CHKERRQ(ierr); 2094113c68e6SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 2095113c68e6SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr); 2096113c68e6SMatthew G. Knepley ierr = PetscSectionSetChart(sectionCell, cStart, cEnd);CHKERRQ(ierr); 20979e5edeeeSMatthew G. Knepley for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionCell, c, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFVCellGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);} 2098113c68e6SMatthew G. Knepley ierr = PetscSectionSetUp(sectionCell);CHKERRQ(ierr); 2099e87a4003SBarry Smith ierr = DMSetSection(dmCell, sectionCell);CHKERRQ(ierr); 2100113c68e6SMatthew G. Knepley ierr = PetscSectionDestroy(§ionCell);CHKERRQ(ierr); 2101113c68e6SMatthew G. Knepley ierr = DMCreateLocalVector(dmCell, cellgeom);CHKERRQ(ierr); 210206348e87SToby Isaac if (cEndInterior < 0) { 210306348e87SToby Isaac cEndInterior = cEnd; 210406348e87SToby Isaac } 2105113c68e6SMatthew G. Knepley ierr = VecGetArray(*cellgeom, &cgeom);CHKERRQ(ierr); 2106113c68e6SMatthew G. Knepley for (c = cStart; c < cEndInterior; ++c) { 2107113c68e6SMatthew G. Knepley PetscFVCellGeom *cg; 2108113c68e6SMatthew G. Knepley 2109113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmCell, c, cgeom, &cg);CHKERRQ(ierr); 2110113c68e6SMatthew G. Knepley ierr = PetscMemzero(cg, sizeof(*cg));CHKERRQ(ierr); 2111113c68e6SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFVM(dmCell, c, &cg->volume, cg->centroid, NULL);CHKERRQ(ierr); 2112113c68e6SMatthew G. Knepley } 2113113c68e6SMatthew G. Knepley /* Compute face normals and minimum cell radius */ 2114113c68e6SMatthew G. Knepley ierr = DMClone(dm, &dmFace);CHKERRQ(ierr); 2115113c68e6SMatthew G. Knepley ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), §ionFace);CHKERRQ(ierr); 2116113c68e6SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr); 2117113c68e6SMatthew G. Knepley ierr = PetscSectionSetChart(sectionFace, fStart, fEnd);CHKERRQ(ierr); 21189e5edeeeSMatthew G. Knepley for (f = fStart; f < fEnd; ++f) {ierr = PetscSectionSetDof(sectionFace, f, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFVFaceGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);} 2119113c68e6SMatthew G. Knepley ierr = PetscSectionSetUp(sectionFace);CHKERRQ(ierr); 2120e87a4003SBarry Smith ierr = DMSetSection(dmFace, sectionFace);CHKERRQ(ierr); 2121113c68e6SMatthew G. Knepley ierr = PetscSectionDestroy(§ionFace);CHKERRQ(ierr); 2122113c68e6SMatthew G. Knepley ierr = DMCreateLocalVector(dmFace, facegeom);CHKERRQ(ierr); 2123113c68e6SMatthew G. Knepley ierr = VecGetArray(*facegeom, &fgeom);CHKERRQ(ierr); 2124c58f1c22SToby Isaac ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 2125113c68e6SMatthew G. Knepley minradius = PETSC_MAX_REAL; 2126113c68e6SMatthew G. Knepley for (f = fStart; f < fEnd; ++f) { 2127113c68e6SMatthew G. Knepley PetscFVFaceGeom *fg; 2128113c68e6SMatthew G. Knepley PetscReal area; 212950d63984SToby Isaac PetscInt ghost = -1, d, numChildren; 2130113c68e6SMatthew G. Knepley 21319ac3fadcSMatthew G. Knepley if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);} 213250d63984SToby Isaac ierr = DMPlexGetTreeChildren(dm,f,&numChildren,NULL);CHKERRQ(ierr); 213350d63984SToby Isaac if (ghost >= 0 || numChildren) continue; 2134113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmFace, f, fgeom, &fg);CHKERRQ(ierr); 2135113c68e6SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFVM(dm, f, &area, fg->centroid, fg->normal);CHKERRQ(ierr); 2136113c68e6SMatthew G. Knepley for (d = 0; d < dim; ++d) fg->normal[d] *= area; 2137113c68e6SMatthew G. Knepley /* Flip face orientation if necessary to match ordering in support, and Update minimum radius */ 2138113c68e6SMatthew G. Knepley { 2139113c68e6SMatthew G. Knepley PetscFVCellGeom *cL, *cR; 214006348e87SToby Isaac PetscInt ncells; 2141113c68e6SMatthew G. Knepley const PetscInt *cells; 2142113c68e6SMatthew G. Knepley PetscReal *lcentroid, *rcentroid; 21430453c0cdSMatthew G. Knepley PetscReal l[3], r[3], v[3]; 2144113c68e6SMatthew G. Knepley 2145113c68e6SMatthew G. Knepley ierr = DMPlexGetSupport(dm, f, &cells);CHKERRQ(ierr); 214606348e87SToby Isaac ierr = DMPlexGetSupportSize(dm, f, &ncells);CHKERRQ(ierr); 2147113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[0], cgeom, &cL);CHKERRQ(ierr); 2148113c68e6SMatthew G. Knepley lcentroid = cells[0] >= cEndInterior ? fg->centroid : cL->centroid; 214906348e87SToby Isaac if (ncells > 1) { 215006348e87SToby Isaac ierr = DMPlexPointLocalRead(dmCell, cells[1], cgeom, &cR);CHKERRQ(ierr); 2151113c68e6SMatthew G. Knepley rcentroid = cells[1] >= cEndInterior ? fg->centroid : cR->centroid; 215206348e87SToby Isaac } 215306348e87SToby Isaac else { 215406348e87SToby Isaac rcentroid = fg->centroid; 215506348e87SToby Isaac } 21562e17dfb7SMatthew G. Knepley ierr = DMLocalizeCoordinateReal_Internal(dm, dim, fg->centroid, lcentroid, l);CHKERRQ(ierr); 21572e17dfb7SMatthew G. Knepley ierr = DMLocalizeCoordinateReal_Internal(dm, dim, fg->centroid, rcentroid, r);CHKERRQ(ierr); 21580453c0cdSMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, l, r, v); 2159113c68e6SMatthew G. Knepley if (DMPlex_DotRealD_Internal(dim, fg->normal, v) < 0) { 2160113c68e6SMatthew G. Knepley for (d = 0; d < dim; ++d) fg->normal[d] = -fg->normal[d]; 2161113c68e6SMatthew G. Knepley } 2162113c68e6SMatthew G. Knepley if (DMPlex_DotRealD_Internal(dim, fg->normal, v) <= 0) { 2163113c68e6SMatthew G. Knepley if (dim == 2) SETERRQ5(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed, normal (%g,%g) v (%g,%g)", f, (double) fg->normal[0], (double) fg->normal[1], (double) v[0], (double) v[1]); 2164113c68e6SMatthew G. Knepley if (dim == 3) SETERRQ7(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed, normal (%g,%g,%g) v (%g,%g,%g)", f, (double) fg->normal[0], (double) fg->normal[1], (double) fg->normal[2], (double) v[0], (double) v[1], (double) v[2]); 2165113c68e6SMatthew G. Knepley SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed", f); 2166113c68e6SMatthew G. Knepley } 2167113c68e6SMatthew G. Knepley if (cells[0] < cEndInterior) { 2168113c68e6SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, fg->centroid, cL->centroid, v); 2169113c68e6SMatthew G. Knepley minradius = PetscMin(minradius, DMPlex_NormD_Internal(dim, v)); 2170113c68e6SMatthew G. Knepley } 217106348e87SToby Isaac if (ncells > 1 && cells[1] < cEndInterior) { 2172113c68e6SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, fg->centroid, cR->centroid, v); 2173113c68e6SMatthew G. Knepley minradius = PetscMin(minradius, DMPlex_NormD_Internal(dim, v)); 2174113c68e6SMatthew G. Knepley } 2175113c68e6SMatthew G. Knepley } 2176113c68e6SMatthew G. Knepley } 2177b2566f29SBarry Smith ierr = MPIU_Allreduce(&minradius, &gminradius, 1, MPIU_REAL, MPIU_MIN, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr); 2178113c68e6SMatthew G. Knepley ierr = DMPlexSetMinRadius(dm, gminradius);CHKERRQ(ierr); 2179113c68e6SMatthew G. Knepley /* Compute centroids of ghost cells */ 2180113c68e6SMatthew G. Knepley for (c = cEndInterior; c < cEnd; ++c) { 2181113c68e6SMatthew G. Knepley PetscFVFaceGeom *fg; 2182113c68e6SMatthew G. Knepley const PetscInt *cone, *support; 2183113c68e6SMatthew G. Knepley PetscInt coneSize, supportSize, s; 2184113c68e6SMatthew G. Knepley 2185113c68e6SMatthew G. Knepley ierr = DMPlexGetConeSize(dmCell, c, &coneSize);CHKERRQ(ierr); 2186113c68e6SMatthew G. Knepley if (coneSize != 1) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Ghost cell %d has cone size %d != 1", c, coneSize); 2187113c68e6SMatthew G. Knepley ierr = DMPlexGetCone(dmCell, c, &cone);CHKERRQ(ierr); 2188113c68e6SMatthew G. Knepley ierr = DMPlexGetSupportSize(dmCell, cone[0], &supportSize);CHKERRQ(ierr); 218950d63984SToby Isaac if (supportSize != 2) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %d has support size %d != 2", cone[0], supportSize); 2190113c68e6SMatthew G. Knepley ierr = DMPlexGetSupport(dmCell, cone[0], &support);CHKERRQ(ierr); 2191113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmFace, cone[0], fgeom, &fg);CHKERRQ(ierr); 2192113c68e6SMatthew G. Knepley for (s = 0; s < 2; ++s) { 2193113c68e6SMatthew G. Knepley /* Reflect ghost centroid across plane of face */ 2194113c68e6SMatthew G. Knepley if (support[s] == c) { 2195640bce14SSatish Balay PetscFVCellGeom *ci; 2196113c68e6SMatthew G. Knepley PetscFVCellGeom *cg; 2197113c68e6SMatthew G. Knepley PetscReal c2f[3], a; 2198113c68e6SMatthew G. Knepley 2199113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, support[(s+1)%2], cgeom, &ci);CHKERRQ(ierr); 2200113c68e6SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, ci->centroid, fg->centroid, c2f); /* cell to face centroid */ 2201113c68e6SMatthew G. Knepley a = DMPlex_DotRealD_Internal(dim, c2f, fg->normal)/DMPlex_DotRealD_Internal(dim, fg->normal, fg->normal); 2202113c68e6SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmCell, support[s], cgeom, &cg);CHKERRQ(ierr); 2203113c68e6SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, 2*a, fg->normal, ci->centroid, cg->centroid); 2204113c68e6SMatthew G. Knepley cg->volume = ci->volume; 2205113c68e6SMatthew G. Knepley } 2206113c68e6SMatthew G. Knepley } 2207113c68e6SMatthew G. Knepley } 2208113c68e6SMatthew G. Knepley ierr = VecRestoreArray(*facegeom, &fgeom);CHKERRQ(ierr); 2209113c68e6SMatthew G. Knepley ierr = VecRestoreArray(*cellgeom, &cgeom);CHKERRQ(ierr); 2210113c68e6SMatthew G. Knepley ierr = DMDestroy(&dmCell);CHKERRQ(ierr); 2211113c68e6SMatthew G. Knepley ierr = DMDestroy(&dmFace);CHKERRQ(ierr); 2212113c68e6SMatthew G. Knepley PetscFunctionReturn(0); 2213113c68e6SMatthew G. Knepley } 2214113c68e6SMatthew G. Knepley 2215113c68e6SMatthew G. Knepley /*@C 2216113c68e6SMatthew G. Knepley DMPlexGetMinRadius - Returns the minimum distance from any cell centroid to a face 2217113c68e6SMatthew G. Knepley 2218113c68e6SMatthew G. Knepley Not collective 2219113c68e6SMatthew G. Knepley 2220113c68e6SMatthew G. Knepley Input Argument: 2221113c68e6SMatthew G. Knepley . dm - the DM 2222113c68e6SMatthew G. Knepley 2223113c68e6SMatthew G. Knepley Output Argument: 2224113c68e6SMatthew G. Knepley . minradius - the minium cell radius 2225113c68e6SMatthew G. Knepley 2226113c68e6SMatthew G. Knepley Level: developer 2227113c68e6SMatthew G. Knepley 2228113c68e6SMatthew G. Knepley .seealso: DMGetCoordinates() 2229113c68e6SMatthew G. Knepley @*/ 2230113c68e6SMatthew G. Knepley PetscErrorCode DMPlexGetMinRadius(DM dm, PetscReal *minradius) 2231113c68e6SMatthew G. Knepley { 2232113c68e6SMatthew G. Knepley PetscFunctionBegin; 2233113c68e6SMatthew G. Knepley PetscValidHeaderSpecific(dm,DM_CLASSID,1); 2234113c68e6SMatthew G. Knepley PetscValidPointer(minradius,2); 2235113c68e6SMatthew G. Knepley *minradius = ((DM_Plex*) dm->data)->minradius; 2236113c68e6SMatthew G. Knepley PetscFunctionReturn(0); 2237113c68e6SMatthew G. Knepley } 2238113c68e6SMatthew G. Knepley 2239113c68e6SMatthew G. Knepley /*@C 2240113c68e6SMatthew G. Knepley DMPlexSetMinRadius - Sets the minimum distance from the cell centroid to a face 2241113c68e6SMatthew G. Knepley 2242113c68e6SMatthew G. Knepley Logically collective 2243113c68e6SMatthew G. Knepley 2244113c68e6SMatthew G. Knepley Input Arguments: 2245113c68e6SMatthew G. Knepley + dm - the DM 2246113c68e6SMatthew G. Knepley - minradius - the minium cell radius 2247113c68e6SMatthew G. Knepley 2248113c68e6SMatthew G. Knepley Level: developer 2249113c68e6SMatthew G. Knepley 2250113c68e6SMatthew G. Knepley .seealso: DMSetCoordinates() 2251113c68e6SMatthew G. Knepley @*/ 2252113c68e6SMatthew G. Knepley PetscErrorCode DMPlexSetMinRadius(DM dm, PetscReal minradius) 2253113c68e6SMatthew G. Knepley { 2254113c68e6SMatthew G. Knepley PetscFunctionBegin; 2255113c68e6SMatthew G. Knepley PetscValidHeaderSpecific(dm,DM_CLASSID,1); 2256113c68e6SMatthew G. Knepley ((DM_Plex*) dm->data)->minradius = minradius; 2257113c68e6SMatthew G. Knepley PetscFunctionReturn(0); 2258113c68e6SMatthew G. Knepley } 2259856ac710SMatthew G. Knepley 2260856ac710SMatthew G. Knepley static PetscErrorCode BuildGradientReconstruction_Internal(DM dm, PetscFV fvm, DM dmFace, PetscScalar *fgeom, DM dmCell, PetscScalar *cgeom) 2261856ac710SMatthew G. Knepley { 2262856ac710SMatthew G. Knepley DMLabel ghostLabel; 2263856ac710SMatthew G. Knepley PetscScalar *dx, *grad, **gref; 2264856ac710SMatthew G. Knepley PetscInt dim, cStart, cEnd, c, cEndInterior, maxNumFaces; 2265856ac710SMatthew G. Knepley PetscErrorCode ierr; 2266856ac710SMatthew G. Knepley 2267856ac710SMatthew G. Knepley PetscFunctionBegin; 2268856ac710SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 2269856ac710SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 2270856ac710SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr); 2271856ac710SMatthew G. Knepley ierr = DMPlexGetMaxSizes(dm, &maxNumFaces, NULL);CHKERRQ(ierr); 2272856ac710SMatthew G. Knepley ierr = PetscFVLeastSquaresSetMaxFaces(fvm, maxNumFaces);CHKERRQ(ierr); 2273c58f1c22SToby Isaac ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 2274856ac710SMatthew G. Knepley ierr = PetscMalloc3(maxNumFaces*dim, &dx, maxNumFaces*dim, &grad, maxNumFaces, &gref);CHKERRQ(ierr); 2275856ac710SMatthew G. Knepley for (c = cStart; c < cEndInterior; c++) { 2276856ac710SMatthew G. Knepley const PetscInt *faces; 2277856ac710SMatthew G. Knepley PetscInt numFaces, usedFaces, f, d; 2278640bce14SSatish Balay PetscFVCellGeom *cg; 2279856ac710SMatthew G. Knepley PetscBool boundary; 2280856ac710SMatthew G. Knepley PetscInt ghost; 2281856ac710SMatthew G. Knepley 2282856ac710SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr); 2283856ac710SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, c, &numFaces);CHKERRQ(ierr); 2284856ac710SMatthew G. Knepley ierr = DMPlexGetCone(dm, c, &faces);CHKERRQ(ierr); 2285856ac710SMatthew G. Knepley if (numFaces < dim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Cell %D has only %D faces, not enough for gradient reconstruction", c, numFaces); 2286856ac710SMatthew G. Knepley for (f = 0, usedFaces = 0; f < numFaces; ++f) { 2287640bce14SSatish Balay PetscFVCellGeom *cg1; 2288856ac710SMatthew G. Knepley PetscFVFaceGeom *fg; 2289856ac710SMatthew G. Knepley const PetscInt *fcells; 2290856ac710SMatthew G. Knepley PetscInt ncell, side; 2291856ac710SMatthew G. Knepley 2292856ac710SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel, faces[f], &ghost);CHKERRQ(ierr); 2293a6ba4734SToby Isaac ierr = DMIsBoundaryPoint(dm, faces[f], &boundary);CHKERRQ(ierr); 2294856ac710SMatthew G. Knepley if ((ghost >= 0) || boundary) continue; 2295856ac710SMatthew G. Knepley ierr = DMPlexGetSupport(dm, faces[f], &fcells);CHKERRQ(ierr); 2296856ac710SMatthew G. Knepley side = (c != fcells[0]); /* c is on left=0 or right=1 of face */ 2297856ac710SMatthew G. Knepley ncell = fcells[!side]; /* the neighbor */ 2298856ac710SMatthew G. Knepley ierr = DMPlexPointLocalRef(dmFace, faces[f], fgeom, &fg);CHKERRQ(ierr); 2299856ac710SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, ncell, cgeom, &cg1);CHKERRQ(ierr); 2300856ac710SMatthew G. Knepley for (d = 0; d < dim; ++d) dx[usedFaces*dim+d] = cg1->centroid[d] - cg->centroid[d]; 2301856ac710SMatthew G. Knepley gref[usedFaces++] = fg->grad[side]; /* Gradient reconstruction term will go here */ 2302856ac710SMatthew G. Knepley } 2303856ac710SMatthew G. Knepley if (!usedFaces) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_USER, "Mesh contains isolated cell (no neighbors). Is it intentional?"); 2304856ac710SMatthew G. Knepley ierr = PetscFVComputeGradient(fvm, usedFaces, dx, grad);CHKERRQ(ierr); 2305856ac710SMatthew G. Knepley for (f = 0, usedFaces = 0; f < numFaces; ++f) { 2306856ac710SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel, faces[f], &ghost);CHKERRQ(ierr); 2307a6ba4734SToby Isaac ierr = DMIsBoundaryPoint(dm, faces[f], &boundary);CHKERRQ(ierr); 2308856ac710SMatthew G. Knepley if ((ghost >= 0) || boundary) continue; 2309856ac710SMatthew G. Knepley for (d = 0; d < dim; ++d) gref[usedFaces][d] = grad[usedFaces*dim+d]; 2310856ac710SMatthew G. Knepley ++usedFaces; 2311856ac710SMatthew G. Knepley } 2312856ac710SMatthew G. Knepley } 2313856ac710SMatthew G. Knepley ierr = PetscFree3(dx, grad, gref);CHKERRQ(ierr); 2314856ac710SMatthew G. Knepley PetscFunctionReturn(0); 2315856ac710SMatthew G. Knepley } 2316856ac710SMatthew G. Knepley 2317b81db932SToby Isaac static PetscErrorCode BuildGradientReconstruction_Internal_Tree(DM dm, PetscFV fvm, DM dmFace, PetscScalar *fgeom, DM dmCell, PetscScalar *cgeom) 2318b81db932SToby Isaac { 2319b81db932SToby Isaac DMLabel ghostLabel; 2320b81db932SToby Isaac PetscScalar *dx, *grad, **gref; 2321b81db932SToby Isaac PetscInt dim, cStart, cEnd, c, cEndInterior, fStart, fEnd, f, nStart, nEnd, maxNumFaces = 0; 2322b81db932SToby Isaac PetscSection neighSec; 2323b81db932SToby Isaac PetscInt (*neighbors)[2]; 2324b81db932SToby Isaac PetscInt *counter; 2325b81db932SToby Isaac PetscErrorCode ierr; 2326b81db932SToby Isaac 2327b81db932SToby Isaac PetscFunctionBegin; 2328b81db932SToby Isaac ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 2329b81db932SToby Isaac ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 2330b81db932SToby Isaac ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr); 23315bc680faSToby Isaac if (cEndInterior < 0) { 23325bc680faSToby Isaac cEndInterior = cEnd; 23335bc680faSToby Isaac } 2334b81db932SToby Isaac ierr = PetscSectionCreate(PetscObjectComm((PetscObject)dm),&neighSec);CHKERRQ(ierr); 2335b81db932SToby Isaac ierr = PetscSectionSetChart(neighSec,cStart,cEndInterior);CHKERRQ(ierr); 2336b81db932SToby Isaac ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr); 2337c58f1c22SToby Isaac ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 2338b81db932SToby Isaac for (f = fStart; f < fEnd; f++) { 2339b81db932SToby Isaac const PetscInt *fcells; 2340b81db932SToby Isaac PetscBool boundary; 23415bc680faSToby Isaac PetscInt ghost = -1; 2342b81db932SToby Isaac PetscInt numChildren, numCells, c; 2343b81db932SToby Isaac 234406348e87SToby Isaac if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);} 2345a6ba4734SToby Isaac ierr = DMIsBoundaryPoint(dm, f, &boundary);CHKERRQ(ierr); 2346b81db932SToby Isaac ierr = DMPlexGetTreeChildren(dm, f, &numChildren, NULL);CHKERRQ(ierr); 2347b81db932SToby Isaac if ((ghost >= 0) || boundary || numChildren) continue; 2348b81db932SToby Isaac ierr = DMPlexGetSupportSize(dm, f, &numCells);CHKERRQ(ierr); 234906348e87SToby Isaac if (numCells == 2) { 2350b81db932SToby Isaac ierr = DMPlexGetSupport(dm, f, &fcells);CHKERRQ(ierr); 2351b81db932SToby Isaac for (c = 0; c < 2; c++) { 2352b81db932SToby Isaac PetscInt cell = fcells[c]; 2353b81db932SToby Isaac 2354e6885bbbSToby Isaac if (cell >= cStart && cell < cEndInterior) { 2355b81db932SToby Isaac ierr = PetscSectionAddDof(neighSec,cell,1);CHKERRQ(ierr); 2356b81db932SToby Isaac } 2357b81db932SToby Isaac } 2358b81db932SToby Isaac } 235906348e87SToby Isaac } 2360b81db932SToby Isaac ierr = PetscSectionSetUp(neighSec);CHKERRQ(ierr); 2361b81db932SToby Isaac ierr = PetscSectionGetMaxDof(neighSec,&maxNumFaces);CHKERRQ(ierr); 2362b81db932SToby Isaac ierr = PetscFVLeastSquaresSetMaxFaces(fvm, maxNumFaces);CHKERRQ(ierr); 2363b81db932SToby Isaac nStart = 0; 2364b81db932SToby Isaac ierr = PetscSectionGetStorageSize(neighSec,&nEnd);CHKERRQ(ierr); 2365b81db932SToby Isaac ierr = PetscMalloc1((nEnd-nStart),&neighbors);CHKERRQ(ierr); 2366b81db932SToby Isaac ierr = PetscCalloc1((cEndInterior-cStart),&counter);CHKERRQ(ierr); 2367b81db932SToby Isaac for (f = fStart; f < fEnd; f++) { 2368b81db932SToby Isaac const PetscInt *fcells; 2369b81db932SToby Isaac PetscBool boundary; 23705bc680faSToby Isaac PetscInt ghost = -1; 2371b81db932SToby Isaac PetscInt numChildren, numCells, c; 2372b81db932SToby Isaac 237306348e87SToby Isaac if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);} 2374a6ba4734SToby Isaac ierr = DMIsBoundaryPoint(dm, f, &boundary);CHKERRQ(ierr); 2375b81db932SToby Isaac ierr = DMPlexGetTreeChildren(dm, f, &numChildren, NULL);CHKERRQ(ierr); 2376b81db932SToby Isaac if ((ghost >= 0) || boundary || numChildren) continue; 2377b81db932SToby Isaac ierr = DMPlexGetSupportSize(dm, f, &numCells);CHKERRQ(ierr); 237806348e87SToby Isaac if (numCells == 2) { 2379b81db932SToby Isaac ierr = DMPlexGetSupport(dm, f, &fcells);CHKERRQ(ierr); 2380b81db932SToby Isaac for (c = 0; c < 2; c++) { 2381b81db932SToby Isaac PetscInt cell = fcells[c], off; 2382b81db932SToby Isaac 2383e6885bbbSToby Isaac if (cell >= cStart && cell < cEndInterior) { 2384b81db932SToby Isaac ierr = PetscSectionGetOffset(neighSec,cell,&off);CHKERRQ(ierr); 2385b81db932SToby Isaac off += counter[cell - cStart]++; 2386b81db932SToby Isaac neighbors[off][0] = f; 2387b81db932SToby Isaac neighbors[off][1] = fcells[1 - c]; 2388b81db932SToby Isaac } 2389b81db932SToby Isaac } 2390b81db932SToby Isaac } 239106348e87SToby Isaac } 2392b81db932SToby Isaac ierr = PetscFree(counter);CHKERRQ(ierr); 2393b81db932SToby Isaac ierr = PetscMalloc3(maxNumFaces*dim, &dx, maxNumFaces*dim, &grad, maxNumFaces, &gref);CHKERRQ(ierr); 2394b81db932SToby Isaac for (c = cStart; c < cEndInterior; c++) { 2395317218b9SToby Isaac PetscInt numFaces, f, d, off, ghost = -1; 2396640bce14SSatish Balay PetscFVCellGeom *cg; 2397b81db932SToby Isaac 2398b81db932SToby Isaac ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr); 2399b81db932SToby Isaac ierr = PetscSectionGetDof(neighSec, c, &numFaces);CHKERRQ(ierr); 2400b81db932SToby Isaac ierr = PetscSectionGetOffset(neighSec, c, &off);CHKERRQ(ierr); 2401317218b9SToby Isaac if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, c, &ghost);CHKERRQ(ierr);} 2402317218b9SToby Isaac if (ghost < 0 && numFaces < dim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Cell %D has only %D faces, not enough for gradient reconstruction", c, numFaces); 2403b81db932SToby Isaac for (f = 0; f < numFaces; ++f) { 2404640bce14SSatish Balay PetscFVCellGeom *cg1; 2405b81db932SToby Isaac PetscFVFaceGeom *fg; 2406b81db932SToby Isaac const PetscInt *fcells; 2407b81db932SToby Isaac PetscInt ncell, side, nface; 2408b81db932SToby Isaac 2409b81db932SToby Isaac nface = neighbors[off + f][0]; 2410b81db932SToby Isaac ncell = neighbors[off + f][1]; 2411b81db932SToby Isaac ierr = DMPlexGetSupport(dm,nface,&fcells);CHKERRQ(ierr); 2412b81db932SToby Isaac side = (c != fcells[0]); 2413b81db932SToby Isaac ierr = DMPlexPointLocalRef(dmFace, nface, fgeom, &fg);CHKERRQ(ierr); 2414b81db932SToby Isaac ierr = DMPlexPointLocalRead(dmCell, ncell, cgeom, &cg1);CHKERRQ(ierr); 2415b81db932SToby Isaac for (d = 0; d < dim; ++d) dx[f*dim+d] = cg1->centroid[d] - cg->centroid[d]; 2416b81db932SToby Isaac gref[f] = fg->grad[side]; /* Gradient reconstruction term will go here */ 2417b81db932SToby Isaac } 2418b81db932SToby Isaac ierr = PetscFVComputeGradient(fvm, numFaces, dx, grad);CHKERRQ(ierr); 2419b81db932SToby Isaac for (f = 0; f < numFaces; ++f) { 2420b81db932SToby Isaac for (d = 0; d < dim; ++d) gref[f][d] = grad[f*dim+d]; 2421b81db932SToby Isaac } 2422b81db932SToby Isaac } 2423b81db932SToby Isaac ierr = PetscFree3(dx, grad, gref);CHKERRQ(ierr); 24245fe94518SToby Isaac ierr = PetscSectionDestroy(&neighSec);CHKERRQ(ierr); 2425b81db932SToby Isaac ierr = PetscFree(neighbors);CHKERRQ(ierr); 2426b81db932SToby Isaac PetscFunctionReturn(0); 2427b81db932SToby Isaac } 2428b81db932SToby Isaac 2429856ac710SMatthew G. Knepley /*@ 2430856ac710SMatthew G. Knepley DMPlexComputeGradientFVM - Compute geometric factors for gradient reconstruction, which are stored in the geometry data, and compute layout for gradient data 2431856ac710SMatthew G. Knepley 2432856ac710SMatthew G. Knepley Collective on DM 2433856ac710SMatthew G. Knepley 2434856ac710SMatthew G. Knepley Input Arguments: 2435856ac710SMatthew G. Knepley + dm - The DM 2436856ac710SMatthew G. Knepley . fvm - The PetscFV 24378f9f38e3SMatthew G. Knepley . faceGeometry - The face geometry from DMPlexComputeFaceGeometryFVM() 24388f9f38e3SMatthew G. Knepley - cellGeometry - The face geometry from DMPlexComputeCellGeometryFVM() 2439856ac710SMatthew G. Knepley 2440856ac710SMatthew G. Knepley Output Parameters: 2441856ac710SMatthew G. Knepley + faceGeometry - The geometric factors for gradient calculation are inserted 2442856ac710SMatthew G. Knepley - dmGrad - The DM describing the layout of gradient data 2443856ac710SMatthew G. Knepley 2444856ac710SMatthew G. Knepley Level: developer 2445856ac710SMatthew G. Knepley 2446856ac710SMatthew G. Knepley .seealso: DMPlexGetFaceGeometryFVM(), DMPlexGetCellGeometryFVM() 2447856ac710SMatthew G. Knepley @*/ 2448856ac710SMatthew G. Knepley PetscErrorCode DMPlexComputeGradientFVM(DM dm, PetscFV fvm, Vec faceGeometry, Vec cellGeometry, DM *dmGrad) 2449856ac710SMatthew G. Knepley { 2450856ac710SMatthew G. Knepley DM dmFace, dmCell; 2451856ac710SMatthew G. Knepley PetscScalar *fgeom, *cgeom; 2452b81db932SToby Isaac PetscSection sectionGrad, parentSection; 2453856ac710SMatthew G. Knepley PetscInt dim, pdim, cStart, cEnd, cEndInterior, c; 2454856ac710SMatthew G. Knepley PetscErrorCode ierr; 2455856ac710SMatthew G. Knepley 2456856ac710SMatthew G. Knepley PetscFunctionBegin; 2457856ac710SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 2458856ac710SMatthew G. Knepley ierr = PetscFVGetNumComponents(fvm, &pdim);CHKERRQ(ierr); 2459856ac710SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 2460856ac710SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr); 2461856ac710SMatthew G. Knepley /* Construct the interpolant corresponding to each face from the least-square solution over the cell neighborhood */ 2462856ac710SMatthew G. Knepley ierr = VecGetDM(faceGeometry, &dmFace);CHKERRQ(ierr); 2463856ac710SMatthew G. Knepley ierr = VecGetDM(cellGeometry, &dmCell);CHKERRQ(ierr); 2464856ac710SMatthew G. Knepley ierr = VecGetArray(faceGeometry, &fgeom);CHKERRQ(ierr); 2465856ac710SMatthew G. Knepley ierr = VecGetArray(cellGeometry, &cgeom);CHKERRQ(ierr); 2466b81db932SToby Isaac ierr = DMPlexGetTree(dm,&parentSection,NULL,NULL,NULL,NULL);CHKERRQ(ierr); 2467b81db932SToby Isaac if (!parentSection) { 2468856ac710SMatthew G. Knepley ierr = BuildGradientReconstruction_Internal(dm, fvm, dmFace, fgeom, dmCell, cgeom);CHKERRQ(ierr); 2469b5a3613cSMatthew G. Knepley } else { 2470b81db932SToby Isaac ierr = BuildGradientReconstruction_Internal_Tree(dm, fvm, dmFace, fgeom, dmCell, cgeom);CHKERRQ(ierr); 2471b81db932SToby Isaac } 2472856ac710SMatthew G. Knepley ierr = VecRestoreArray(faceGeometry, &fgeom);CHKERRQ(ierr); 2473856ac710SMatthew G. Knepley ierr = VecRestoreArray(cellGeometry, &cgeom);CHKERRQ(ierr); 2474856ac710SMatthew G. Knepley /* Create storage for gradients */ 2475856ac710SMatthew G. Knepley ierr = DMClone(dm, dmGrad);CHKERRQ(ierr); 2476856ac710SMatthew G. Knepley ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), §ionGrad);CHKERRQ(ierr); 2477856ac710SMatthew G. Knepley ierr = PetscSectionSetChart(sectionGrad, cStart, cEnd);CHKERRQ(ierr); 2478856ac710SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionGrad, c, pdim*dim);CHKERRQ(ierr);} 2479856ac710SMatthew G. Knepley ierr = PetscSectionSetUp(sectionGrad);CHKERRQ(ierr); 2480e87a4003SBarry Smith ierr = DMSetSection(*dmGrad, sectionGrad);CHKERRQ(ierr); 2481856ac710SMatthew G. Knepley ierr = PetscSectionDestroy(§ionGrad);CHKERRQ(ierr); 2482856ac710SMatthew G. Knepley PetscFunctionReturn(0); 2483856ac710SMatthew G. Knepley } 2484b27d5b9eSToby Isaac 2485c501906fSMatthew G. Knepley /*@ 2486c501906fSMatthew G. Knepley DMPlexGetDataFVM - Retrieve precomputed cell geometry 2487c501906fSMatthew G. Knepley 2488c501906fSMatthew G. Knepley Collective on DM 2489c501906fSMatthew G. Knepley 2490c501906fSMatthew G. Knepley Input Arguments: 2491c501906fSMatthew G. Knepley + dm - The DM 2492c501906fSMatthew G. Knepley - fvm - The PetscFV 2493c501906fSMatthew G. Knepley 2494c501906fSMatthew G. Knepley Output Parameters: 2495c501906fSMatthew G. Knepley + cellGeometry - The cell geometry 2496c501906fSMatthew G. Knepley . faceGeometry - The face geometry 2497c501906fSMatthew G. Knepley - dmGrad - The gradient matrices 2498c501906fSMatthew G. Knepley 2499c501906fSMatthew G. Knepley Level: developer 2500c501906fSMatthew G. Knepley 2501c501906fSMatthew G. Knepley .seealso: DMPlexComputeGeometryFVM() 2502c501906fSMatthew G. Knepley @*/ 2503b27d5b9eSToby Isaac PetscErrorCode DMPlexGetDataFVM(DM dm, PetscFV fv, Vec *cellgeom, Vec *facegeom, DM *gradDM) 2504b27d5b9eSToby Isaac { 2505b27d5b9eSToby Isaac PetscObject cellgeomobj, facegeomobj; 2506b27d5b9eSToby Isaac PetscErrorCode ierr; 2507b27d5b9eSToby Isaac 2508b27d5b9eSToby Isaac PetscFunctionBegin; 2509b27d5b9eSToby Isaac ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_cellgeom_fvm", &cellgeomobj);CHKERRQ(ierr); 2510b27d5b9eSToby Isaac if (!cellgeomobj) { 2511b27d5b9eSToby Isaac Vec cellgeomInt, facegeomInt; 2512b27d5b9eSToby Isaac 2513b27d5b9eSToby Isaac ierr = DMPlexComputeGeometryFVM(dm, &cellgeomInt, &facegeomInt);CHKERRQ(ierr); 2514b27d5b9eSToby Isaac ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_cellgeom_fvm",(PetscObject)cellgeomInt);CHKERRQ(ierr); 2515b27d5b9eSToby Isaac ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_facegeom_fvm",(PetscObject)facegeomInt);CHKERRQ(ierr); 2516b27d5b9eSToby Isaac ierr = VecDestroy(&cellgeomInt);CHKERRQ(ierr); 2517b27d5b9eSToby Isaac ierr = VecDestroy(&facegeomInt);CHKERRQ(ierr); 2518b27d5b9eSToby Isaac ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_cellgeom_fvm", &cellgeomobj);CHKERRQ(ierr); 2519b27d5b9eSToby Isaac } 2520b27d5b9eSToby Isaac ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_facegeom_fvm", &facegeomobj);CHKERRQ(ierr); 2521b27d5b9eSToby Isaac if (cellgeom) *cellgeom = (Vec) cellgeomobj; 2522b27d5b9eSToby Isaac if (facegeom) *facegeom = (Vec) facegeomobj; 2523b27d5b9eSToby Isaac if (gradDM) { 2524b27d5b9eSToby Isaac PetscObject gradobj; 2525b27d5b9eSToby Isaac PetscBool computeGradients; 2526b27d5b9eSToby Isaac 2527b27d5b9eSToby Isaac ierr = PetscFVGetComputeGradients(fv,&computeGradients);CHKERRQ(ierr); 2528b27d5b9eSToby Isaac if (!computeGradients) { 2529b27d5b9eSToby Isaac *gradDM = NULL; 2530b27d5b9eSToby Isaac PetscFunctionReturn(0); 2531b27d5b9eSToby Isaac } 2532b27d5b9eSToby Isaac ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_dmgrad_fvm", &gradobj);CHKERRQ(ierr); 2533b27d5b9eSToby Isaac if (!gradobj) { 2534b27d5b9eSToby Isaac DM dmGradInt; 2535b27d5b9eSToby Isaac 2536b27d5b9eSToby Isaac ierr = DMPlexComputeGradientFVM(dm,fv,(Vec) facegeomobj,(Vec) cellgeomobj,&dmGradInt);CHKERRQ(ierr); 2537b27d5b9eSToby Isaac ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_dmgrad_fvm", (PetscObject)dmGradInt);CHKERRQ(ierr); 2538b27d5b9eSToby Isaac ierr = DMDestroy(&dmGradInt);CHKERRQ(ierr); 2539b27d5b9eSToby Isaac ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_dmgrad_fvm", &gradobj);CHKERRQ(ierr); 2540b27d5b9eSToby Isaac } 2541b27d5b9eSToby Isaac *gradDM = (DM) gradobj; 2542b27d5b9eSToby Isaac } 2543b27d5b9eSToby Isaac PetscFunctionReturn(0); 2544b27d5b9eSToby Isaac } 2545d6143a4eSToby Isaac 25469d150b73SToby Isaac static PetscErrorCode DMPlexCoordinatesToReference_NewtonUpdate(PetscInt dimC, PetscInt dimR, PetscScalar *J, PetscScalar *invJ, PetscScalar *work, PetscReal *resNeg, PetscReal *guess) 25479d150b73SToby Isaac { 25489d150b73SToby Isaac PetscInt l, m; 25499d150b73SToby Isaac 2550cd345991SToby Isaac PetscFunctionBeginHot; 25519d150b73SToby Isaac if (dimC == dimR && dimR <= 3) { 25529d150b73SToby Isaac /* invert Jacobian, multiply */ 25539d150b73SToby Isaac PetscScalar det, idet; 25549d150b73SToby Isaac 25559d150b73SToby Isaac switch (dimR) { 25569d150b73SToby Isaac case 1: 25579d150b73SToby Isaac invJ[0] = 1./ J[0]; 25589d150b73SToby Isaac break; 25599d150b73SToby Isaac case 2: 25609d150b73SToby Isaac det = J[0] * J[3] - J[1] * J[2]; 25619d150b73SToby Isaac idet = 1./det; 25629d150b73SToby Isaac invJ[0] = J[3] * idet; 25639d150b73SToby Isaac invJ[1] = -J[1] * idet; 25649d150b73SToby Isaac invJ[2] = -J[2] * idet; 25659d150b73SToby Isaac invJ[3] = J[0] * idet; 25669d150b73SToby Isaac break; 25679d150b73SToby Isaac case 3: 25689d150b73SToby Isaac { 25699d150b73SToby Isaac invJ[0] = J[4] * J[8] - J[5] * J[7]; 25709d150b73SToby Isaac invJ[1] = J[2] * J[7] - J[1] * J[8]; 25719d150b73SToby Isaac invJ[2] = J[1] * J[5] - J[2] * J[4]; 25729d150b73SToby Isaac det = invJ[0] * J[0] + invJ[1] * J[3] + invJ[2] * J[6]; 25739d150b73SToby Isaac idet = 1./det; 25749d150b73SToby Isaac invJ[0] *= idet; 25759d150b73SToby Isaac invJ[1] *= idet; 25769d150b73SToby Isaac invJ[2] *= idet; 25779d150b73SToby Isaac invJ[3] = idet * (J[5] * J[6] - J[3] * J[8]); 25789d150b73SToby Isaac invJ[4] = idet * (J[0] * J[8] - J[2] * J[6]); 25799d150b73SToby Isaac invJ[5] = idet * (J[2] * J[3] - J[0] * J[5]); 25809d150b73SToby Isaac invJ[6] = idet * (J[3] * J[7] - J[4] * J[6]); 25819d150b73SToby Isaac invJ[7] = idet * (J[1] * J[6] - J[0] * J[7]); 25829d150b73SToby Isaac invJ[8] = idet * (J[0] * J[4] - J[1] * J[3]); 25839d150b73SToby Isaac } 25849d150b73SToby Isaac break; 25859d150b73SToby Isaac } 25869d150b73SToby Isaac for (l = 0; l < dimR; l++) { 25879d150b73SToby Isaac for (m = 0; m < dimC; m++) { 2588c6e120d1SToby Isaac guess[l] += PetscRealPart(invJ[l * dimC + m]) * resNeg[m]; 25899d150b73SToby Isaac } 25909d150b73SToby Isaac } 25919d150b73SToby Isaac } else { 25929d150b73SToby Isaac #if defined(PETSC_USE_COMPLEX) 25939d150b73SToby Isaac char transpose = 'C'; 25949d150b73SToby Isaac #else 25959d150b73SToby Isaac char transpose = 'T'; 25969d150b73SToby Isaac #endif 25979d150b73SToby Isaac PetscBLASInt m = dimR; 25989d150b73SToby Isaac PetscBLASInt n = dimC; 25999d150b73SToby Isaac PetscBLASInt one = 1; 26009d150b73SToby Isaac PetscBLASInt worksize = dimR * dimC, info; 26019d150b73SToby Isaac 26029d150b73SToby Isaac for (l = 0; l < dimC; l++) {invJ[l] = resNeg[l];} 26039d150b73SToby Isaac 26049d150b73SToby Isaac PetscStackCallBLAS("LAPACKgels",LAPACKgels_(&transpose,&m,&n,&one,J,&m,invJ,&n,work,&worksize, &info)); 26059d150b73SToby Isaac if (info != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad argument to GELS"); 26069d150b73SToby Isaac 2607c6e120d1SToby Isaac for (l = 0; l < dimR; l++) {guess[l] += PetscRealPart(invJ[l]);} 26089d150b73SToby Isaac } 26099d150b73SToby Isaac PetscFunctionReturn(0); 26109d150b73SToby Isaac } 26119d150b73SToby Isaac 26129d150b73SToby Isaac static PetscErrorCode DMPlexCoordinatesToReference_Tensor(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[], Vec coords, PetscInt dimC, PetscInt dimR) 26139d150b73SToby Isaac { 2614c0cbe899SToby Isaac PetscInt coordSize, i, j, k, l, m, maxIts = 7, numV = (1 << dimR); 26159d150b73SToby Isaac PetscScalar *coordsScalar = NULL; 26169d150b73SToby Isaac PetscReal *cellData, *cellCoords, *cellCoeffs, *extJ, *resNeg; 26179d150b73SToby Isaac PetscScalar *J, *invJ, *work; 26189d150b73SToby Isaac PetscErrorCode ierr; 26199d150b73SToby Isaac 26209d150b73SToby Isaac PetscFunctionBegin; 26219d150b73SToby Isaac PetscValidHeaderSpecific(dm,DM_CLASSID,1); 26229d150b73SToby Isaac ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr); 262313903a91SSatish Balay if (coordSize < dimC * numV) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expecting at least %D coordinates, got %D",dimC * (1 << dimR), coordSize); 262469291d52SBarry Smith ierr = DMGetWorkArray(dm, 2 * coordSize + dimR + dimC, MPIU_REAL, &cellData);CHKERRQ(ierr); 262569291d52SBarry Smith ierr = DMGetWorkArray(dm, 3 * dimR * dimC, MPIU_SCALAR, &J);CHKERRQ(ierr); 26269d150b73SToby Isaac cellCoords = &cellData[0]; 26279d150b73SToby Isaac cellCoeffs = &cellData[coordSize]; 26289d150b73SToby Isaac extJ = &cellData[2 * coordSize]; 26299d150b73SToby Isaac resNeg = &cellData[2 * coordSize + dimR]; 26309d150b73SToby Isaac invJ = &J[dimR * dimC]; 26319d150b73SToby Isaac work = &J[2 * dimR * dimC]; 26329d150b73SToby Isaac if (dimR == 2) { 26339d150b73SToby Isaac const PetscInt zToPlex[4] = {0, 1, 3, 2}; 26349d150b73SToby Isaac 26359d150b73SToby Isaac for (i = 0; i < 4; i++) { 26369d150b73SToby Isaac PetscInt plexI = zToPlex[i]; 26379d150b73SToby Isaac 26389d150b73SToby Isaac for (j = 0; j < dimC; j++) { 26399d150b73SToby Isaac cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]); 26409d150b73SToby Isaac } 26419d150b73SToby Isaac } 26429d150b73SToby Isaac } else if (dimR == 3) { 26439d150b73SToby Isaac const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6}; 26449d150b73SToby Isaac 26459d150b73SToby Isaac for (i = 0; i < 8; i++) { 26469d150b73SToby Isaac PetscInt plexI = zToPlex[i]; 26479d150b73SToby Isaac 26489d150b73SToby Isaac for (j = 0; j < dimC; j++) { 26499d150b73SToby Isaac cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]); 26509d150b73SToby Isaac } 26519d150b73SToby Isaac } 26529d150b73SToby Isaac } else { 26539d150b73SToby Isaac for (i = 0; i < coordSize; i++) {cellCoords[i] = PetscRealPart(coordsScalar[i]);} 26549d150b73SToby Isaac } 26559d150b73SToby Isaac /* Perform the shuffling transform that converts values at the corners of [-1,1]^d to coefficients */ 26569d150b73SToby Isaac for (i = 0; i < dimR; i++) { 26579d150b73SToby Isaac PetscReal *swap; 26589d150b73SToby Isaac 26599d150b73SToby Isaac for (j = 0; j < (numV / 2); j++) { 26609d150b73SToby Isaac for (k = 0; k < dimC; k++) { 26619d150b73SToby Isaac cellCoeffs[dimC * j + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] + cellCoords[dimC * 2 * j + k]); 26629d150b73SToby Isaac cellCoeffs[dimC * (j + (numV / 2)) + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] - cellCoords[dimC * 2 * j + k]); 26639d150b73SToby Isaac } 26649d150b73SToby Isaac } 26659d150b73SToby Isaac 26669d150b73SToby Isaac if (i < dimR - 1) { 26679d150b73SToby Isaac swap = cellCoeffs; 26689d150b73SToby Isaac cellCoeffs = cellCoords; 26699d150b73SToby Isaac cellCoords = swap; 26709d150b73SToby Isaac } 26719d150b73SToby Isaac } 26729d150b73SToby Isaac ierr = PetscMemzero(refCoords,numPoints * dimR * sizeof (PetscReal));CHKERRQ(ierr); 26739d150b73SToby Isaac for (j = 0; j < numPoints; j++) { 26749d150b73SToby Isaac for (i = 0; i < maxIts; i++) { 26759d150b73SToby Isaac PetscReal *guess = &refCoords[dimR * j]; 26769d150b73SToby Isaac 26779d150b73SToby Isaac /* compute -residual and Jacobian */ 26789d150b73SToby Isaac for (k = 0; k < dimC; k++) {resNeg[k] = realCoords[dimC * j + k];} 26799d150b73SToby Isaac for (k = 0; k < dimC * dimR; k++) {J[k] = 0.;} 26809d150b73SToby Isaac for (k = 0; k < numV; k++) { 26819d150b73SToby Isaac PetscReal extCoord = 1.; 26829d150b73SToby Isaac for (l = 0; l < dimR; l++) { 26839d150b73SToby Isaac PetscReal coord = guess[l]; 26849d150b73SToby Isaac PetscInt dep = (k & (1 << l)) >> l; 26859d150b73SToby Isaac 26869d150b73SToby Isaac extCoord *= dep * coord + !dep; 26879d150b73SToby Isaac extJ[l] = dep; 26889d150b73SToby Isaac 26899d150b73SToby Isaac for (m = 0; m < dimR; m++) { 26909d150b73SToby Isaac PetscReal coord = guess[m]; 26919d150b73SToby Isaac PetscInt dep = ((k & (1 << m)) >> m) && (m != l); 26929d150b73SToby Isaac PetscReal mult = dep * coord + !dep; 26939d150b73SToby Isaac 26949d150b73SToby Isaac extJ[l] *= mult; 26959d150b73SToby Isaac } 26969d150b73SToby Isaac } 26979d150b73SToby Isaac for (l = 0; l < dimC; l++) { 26989d150b73SToby Isaac PetscReal coeff = cellCoeffs[dimC * k + l]; 26999d150b73SToby Isaac 27009d150b73SToby Isaac resNeg[l] -= coeff * extCoord; 27019d150b73SToby Isaac for (m = 0; m < dimR; m++) { 27029d150b73SToby Isaac J[dimR * l + m] += coeff * extJ[m]; 27039d150b73SToby Isaac } 27049d150b73SToby Isaac } 27059d150b73SToby Isaac } 27060611203eSToby Isaac #if 0 && defined(PETSC_USE_DEBUG) 27070611203eSToby Isaac { 27080611203eSToby Isaac PetscReal maxAbs = 0.; 27090611203eSToby Isaac 27100611203eSToby Isaac for (l = 0; l < dimC; l++) { 27110611203eSToby Isaac maxAbs = PetscMax(maxAbs,PetscAbsReal(resNeg[l])); 27120611203eSToby Isaac } 27130611203eSToby Isaac ierr = PetscInfo4(dm,"cell %D, point %D, iter %D: res %g\n",cell,j,i,maxAbs);CHKERRQ(ierr); 27140611203eSToby Isaac } 27150611203eSToby Isaac #endif 27169d150b73SToby Isaac 27179d150b73SToby Isaac ierr = DMPlexCoordinatesToReference_NewtonUpdate(dimC,dimR,J,invJ,work,resNeg,guess);CHKERRQ(ierr); 27189d150b73SToby Isaac } 27199d150b73SToby Isaac } 272069291d52SBarry Smith ierr = DMRestoreWorkArray(dm, 3 * dimR * dimC, MPIU_SCALAR, &J);CHKERRQ(ierr); 272169291d52SBarry Smith ierr = DMRestoreWorkArray(dm, 2 * coordSize + dimR + dimC, MPIU_REAL, &cellData);CHKERRQ(ierr); 27229d150b73SToby Isaac ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr); 27239d150b73SToby Isaac PetscFunctionReturn(0); 27249d150b73SToby Isaac } 27259d150b73SToby Isaac 27269d150b73SToby Isaac static PetscErrorCode DMPlexReferenceToCoordinates_Tensor(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[], Vec coords, PetscInt dimC, PetscInt dimR) 27279d150b73SToby Isaac { 27289d150b73SToby Isaac PetscInt coordSize, i, j, k, l, numV = (1 << dimR); 27299d150b73SToby Isaac PetscScalar *coordsScalar = NULL; 27309d150b73SToby Isaac PetscReal *cellData, *cellCoords, *cellCoeffs; 27319d150b73SToby Isaac PetscErrorCode ierr; 27329d150b73SToby Isaac 27339d150b73SToby Isaac PetscFunctionBegin; 27349d150b73SToby Isaac PetscValidHeaderSpecific(dm,DM_CLASSID,1); 27359d150b73SToby Isaac ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr); 273613903a91SSatish Balay if (coordSize < dimC * numV) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expecting at least %D coordinates, got %D",dimC * (1 << dimR), coordSize); 273769291d52SBarry Smith ierr = DMGetWorkArray(dm, 2 * coordSize, MPIU_REAL, &cellData);CHKERRQ(ierr); 27389d150b73SToby Isaac cellCoords = &cellData[0]; 27399d150b73SToby Isaac cellCoeffs = &cellData[coordSize]; 27409d150b73SToby Isaac if (dimR == 2) { 27419d150b73SToby Isaac const PetscInt zToPlex[4] = {0, 1, 3, 2}; 27429d150b73SToby Isaac 27439d150b73SToby Isaac for (i = 0; i < 4; i++) { 27449d150b73SToby Isaac PetscInt plexI = zToPlex[i]; 27459d150b73SToby Isaac 27469d150b73SToby Isaac for (j = 0; j < dimC; j++) { 27479d150b73SToby Isaac cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]); 27489d150b73SToby Isaac } 27499d150b73SToby Isaac } 27509d150b73SToby Isaac } else if (dimR == 3) { 27519d150b73SToby Isaac const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6}; 27529d150b73SToby Isaac 27539d150b73SToby Isaac for (i = 0; i < 8; i++) { 27549d150b73SToby Isaac PetscInt plexI = zToPlex[i]; 27559d150b73SToby Isaac 27569d150b73SToby Isaac for (j = 0; j < dimC; j++) { 27579d150b73SToby Isaac cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]); 27589d150b73SToby Isaac } 27599d150b73SToby Isaac } 27609d150b73SToby Isaac } else { 27619d150b73SToby Isaac for (i = 0; i < coordSize; i++) {cellCoords[i] = PetscRealPart(coordsScalar[i]);} 27629d150b73SToby Isaac } 27639d150b73SToby Isaac /* Perform the shuffling transform that converts values at the corners of [-1,1]^d to coefficients */ 27649d150b73SToby Isaac for (i = 0; i < dimR; i++) { 27659d150b73SToby Isaac PetscReal *swap; 27669d150b73SToby Isaac 27679d150b73SToby Isaac for (j = 0; j < (numV / 2); j++) { 27689d150b73SToby Isaac for (k = 0; k < dimC; k++) { 27699d150b73SToby Isaac cellCoeffs[dimC * j + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] + cellCoords[dimC * 2 * j + k]); 27709d150b73SToby Isaac cellCoeffs[dimC * (j + (numV / 2)) + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] - cellCoords[dimC * 2 * j + k]); 27719d150b73SToby Isaac } 27729d150b73SToby Isaac } 27739d150b73SToby Isaac 27749d150b73SToby Isaac if (i < dimR - 1) { 27759d150b73SToby Isaac swap = cellCoeffs; 27769d150b73SToby Isaac cellCoeffs = cellCoords; 27779d150b73SToby Isaac cellCoords = swap; 27789d150b73SToby Isaac } 27799d150b73SToby Isaac } 27809d150b73SToby Isaac ierr = PetscMemzero(realCoords,numPoints * dimC * sizeof (PetscReal));CHKERRQ(ierr); 27819d150b73SToby Isaac for (j = 0; j < numPoints; j++) { 27829d150b73SToby Isaac const PetscReal *guess = &refCoords[dimR * j]; 27839d150b73SToby Isaac PetscReal *mapped = &realCoords[dimC * j]; 27849d150b73SToby Isaac 27859d150b73SToby Isaac for (k = 0; k < numV; k++) { 27869d150b73SToby Isaac PetscReal extCoord = 1.; 27879d150b73SToby Isaac for (l = 0; l < dimR; l++) { 27889d150b73SToby Isaac PetscReal coord = guess[l]; 27899d150b73SToby Isaac PetscInt dep = (k & (1 << l)) >> l; 27909d150b73SToby Isaac 27919d150b73SToby Isaac extCoord *= dep * coord + !dep; 27929d150b73SToby Isaac } 27939d150b73SToby Isaac for (l = 0; l < dimC; l++) { 27949d150b73SToby Isaac PetscReal coeff = cellCoeffs[dimC * k + l]; 27959d150b73SToby Isaac 27969d150b73SToby Isaac mapped[l] += coeff * extCoord; 27979d150b73SToby Isaac } 27989d150b73SToby Isaac } 27999d150b73SToby Isaac } 280069291d52SBarry Smith ierr = DMRestoreWorkArray(dm, 2 * coordSize, MPIU_REAL, &cellData);CHKERRQ(ierr); 28019d150b73SToby Isaac ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr); 28029d150b73SToby Isaac PetscFunctionReturn(0); 28039d150b73SToby Isaac } 28049d150b73SToby Isaac 28059c3cf19fSMatthew G. Knepley /* TODO: TOBY please fix this for Nc > 1 */ 28069c3cf19fSMatthew G. Knepley static PetscErrorCode DMPlexCoordinatesToReference_FE(DM dm, PetscFE fe, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[], Vec coords, PetscInt Nc, PetscInt dimR) 28079d150b73SToby Isaac { 28089c3cf19fSMatthew G. Knepley PetscInt numComp, pdim, i, j, k, l, m, maxIter = 7, coordSize; 2809c6e120d1SToby Isaac PetscScalar *nodes = NULL; 2810c6e120d1SToby Isaac PetscReal *invV, *modes; 2811c6e120d1SToby Isaac PetscReal *B, *D, *resNeg; 2812c6e120d1SToby Isaac PetscScalar *J, *invJ, *work; 28139d150b73SToby Isaac PetscErrorCode ierr; 28149d150b73SToby Isaac 28159d150b73SToby Isaac PetscFunctionBegin; 28169c3cf19fSMatthew G. Knepley ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr); 28179d150b73SToby Isaac ierr = PetscFEGetNumComponents(fe, &numComp);CHKERRQ(ierr); 28189c3cf19fSMatthew G. Knepley if (numComp != Nc) SETERRQ2(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"coordinate discretization must have as many components (%D) as embedding dimension (!= %D)",numComp,Nc); 28199d150b73SToby Isaac ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr); 28209d150b73SToby Isaac /* convert nodes to values in the stable evaluation basis */ 282169291d52SBarry Smith ierr = DMGetWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr); 28229d150b73SToby Isaac invV = fe->invV; 2823012b7cc6SMatthew G. Knepley for (i = 0; i < pdim; ++i) { 2824012b7cc6SMatthew G. Knepley modes[i] = 0.; 2825012b7cc6SMatthew G. Knepley for (j = 0; j < pdim; ++j) { 2826012b7cc6SMatthew G. Knepley modes[i] += invV[i * pdim + j] * PetscRealPart(nodes[j]); 28279d150b73SToby Isaac } 28289d150b73SToby Isaac } 282969291d52SBarry Smith ierr = DMGetWorkArray(dm,pdim * Nc + pdim * Nc * dimR + Nc,MPIU_REAL,&B);CHKERRQ(ierr); 28309c3cf19fSMatthew G. Knepley D = &B[pdim*Nc]; 28319c3cf19fSMatthew G. Knepley resNeg = &D[pdim*Nc * dimR]; 283269291d52SBarry Smith ierr = DMGetWorkArray(dm,3 * Nc * dimR,MPIU_SCALAR,&J);CHKERRQ(ierr); 28339c3cf19fSMatthew G. Knepley invJ = &J[Nc * dimR]; 28349c3cf19fSMatthew G. Knepley work = &invJ[Nc * dimR]; 28359d150b73SToby Isaac for (i = 0; i < numPoints * dimR; i++) {refCoords[i] = 0.;} 28369d150b73SToby Isaac for (j = 0; j < numPoints; j++) { 28379b1f03cbSToby Isaac for (i = 0; i < maxIter; i++) { /* we could batch this so that we're not making big B and D arrays all the time */ 28389d150b73SToby Isaac PetscReal *guess = &refCoords[j * dimR]; 28399d150b73SToby Isaac ierr = PetscSpaceEvaluate(fe->basisSpace, 1, guess, B, D, NULL);CHKERRQ(ierr); 28409c3cf19fSMatthew G. Knepley for (k = 0; k < Nc; k++) {resNeg[k] = realCoords[j * Nc + k];} 28419c3cf19fSMatthew G. Knepley for (k = 0; k < Nc * dimR; k++) {J[k] = 0.;} 28429c3cf19fSMatthew G. Knepley for (k = 0; k < pdim; k++) { 28439c3cf19fSMatthew G. Knepley for (l = 0; l < Nc; l++) { 2844012b7cc6SMatthew G. Knepley resNeg[l] -= modes[k] * B[k * Nc + l]; 28459d150b73SToby Isaac for (m = 0; m < dimR; m++) { 2846012b7cc6SMatthew G. Knepley J[l * dimR + m] += modes[k] * D[(k * Nc + l) * dimR + m]; 28479d150b73SToby Isaac } 28489d150b73SToby Isaac } 28499d150b73SToby Isaac } 28500611203eSToby Isaac #if 0 && defined(PETSC_USE_DEBUG) 28510611203eSToby Isaac { 28520611203eSToby Isaac PetscReal maxAbs = 0.; 28530611203eSToby Isaac 28549c3cf19fSMatthew G. Knepley for (l = 0; l < Nc; l++) { 28550611203eSToby Isaac maxAbs = PetscMax(maxAbs,PetscAbsReal(resNeg[l])); 28560611203eSToby Isaac } 28570611203eSToby Isaac ierr = PetscInfo4(dm,"cell %D, point %D, iter %D: res %g\n",cell,j,i,maxAbs);CHKERRQ(ierr); 28580611203eSToby Isaac } 28590611203eSToby Isaac #endif 28609c3cf19fSMatthew G. Knepley ierr = DMPlexCoordinatesToReference_NewtonUpdate(Nc,dimR,J,invJ,work,resNeg,guess);CHKERRQ(ierr); 28619d150b73SToby Isaac } 28629d150b73SToby Isaac } 286369291d52SBarry Smith ierr = DMRestoreWorkArray(dm,3 * Nc * dimR,MPIU_SCALAR,&J);CHKERRQ(ierr); 286469291d52SBarry Smith ierr = DMRestoreWorkArray(dm,pdim * Nc + pdim * Nc * dimR + Nc,MPIU_REAL,&B);CHKERRQ(ierr); 286569291d52SBarry Smith ierr = DMRestoreWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr); 28669d150b73SToby Isaac ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr); 28679d150b73SToby Isaac PetscFunctionReturn(0); 28689d150b73SToby Isaac } 28699d150b73SToby Isaac 28709c3cf19fSMatthew G. Knepley /* TODO: TOBY please fix this for Nc > 1 */ 28719c3cf19fSMatthew G. Knepley static PetscErrorCode DMPlexReferenceToCoordinates_FE(DM dm, PetscFE fe, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[], Vec coords, PetscInt Nc, PetscInt dimR) 28729d150b73SToby Isaac { 28739c3cf19fSMatthew G. Knepley PetscInt numComp, pdim, i, j, k, l, coordSize; 2874c6e120d1SToby Isaac PetscScalar *nodes = NULL; 2875c6e120d1SToby Isaac PetscReal *invV, *modes; 28769d150b73SToby Isaac PetscReal *B; 28779d150b73SToby Isaac PetscErrorCode ierr; 28789d150b73SToby Isaac 28799d150b73SToby Isaac PetscFunctionBegin; 28809c3cf19fSMatthew G. Knepley ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr); 28819d150b73SToby Isaac ierr = PetscFEGetNumComponents(fe, &numComp);CHKERRQ(ierr); 28829c3cf19fSMatthew G. Knepley if (numComp != Nc) SETERRQ2(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"coordinate discretization must have as many components (%D) as embedding dimension (!= %D)",numComp,Nc); 28839d150b73SToby Isaac ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr); 28849d150b73SToby Isaac /* convert nodes to values in the stable evaluation basis */ 288569291d52SBarry Smith ierr = DMGetWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr); 28869d150b73SToby Isaac invV = fe->invV; 2887012b7cc6SMatthew G. Knepley for (i = 0; i < pdim; ++i) { 2888012b7cc6SMatthew G. Knepley modes[i] = 0.; 2889012b7cc6SMatthew G. Knepley for (j = 0; j < pdim; ++j) { 2890012b7cc6SMatthew G. Knepley modes[i] += invV[i * pdim + j] * PetscRealPart(nodes[j]); 28919d150b73SToby Isaac } 28929d150b73SToby Isaac } 289369291d52SBarry Smith ierr = DMGetWorkArray(dm,numPoints * pdim * Nc,MPIU_REAL,&B);CHKERRQ(ierr); 2894012b7cc6SMatthew G. Knepley ierr = PetscSpaceEvaluate(fe->basisSpace, numPoints, refCoords, B, NULL, NULL);CHKERRQ(ierr); 28959c3cf19fSMatthew G. Knepley for (i = 0; i < numPoints * Nc; i++) {realCoords[i] = 0.;} 28969d150b73SToby Isaac for (j = 0; j < numPoints; j++) { 28979c3cf19fSMatthew G. Knepley PetscReal *mapped = &realCoords[j * Nc]; 28989d150b73SToby Isaac 28999c3cf19fSMatthew G. Knepley for (k = 0; k < pdim; k++) { 29009c3cf19fSMatthew G. Knepley for (l = 0; l < Nc; l++) { 290140cf36b3SToby Isaac mapped[l] += modes[k] * B[(j * pdim + k) * Nc + l]; 29029d150b73SToby Isaac } 29039d150b73SToby Isaac } 29049d150b73SToby Isaac } 290569291d52SBarry Smith ierr = DMRestoreWorkArray(dm,numPoints * pdim * Nc,MPIU_REAL,&B);CHKERRQ(ierr); 290669291d52SBarry Smith ierr = DMRestoreWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr); 29079d150b73SToby Isaac ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr); 29089d150b73SToby Isaac PetscFunctionReturn(0); 29099d150b73SToby Isaac } 29109d150b73SToby Isaac 2911d6143a4eSToby Isaac /*@ 2912d6143a4eSToby Isaac DMPlexCoordinatesToReference - Pull coordinates back from the mesh to the reference element using a single element 2913d6143a4eSToby Isaac map. This inversion will be accurate inside the reference element, but may be inaccurate for mappings that do not 2914d6143a4eSToby Isaac extend uniquely outside the reference cell (e.g, most non-affine maps) 2915d6143a4eSToby Isaac 2916d6143a4eSToby Isaac Not collective 2917d6143a4eSToby Isaac 2918d6143a4eSToby Isaac Input Parameters: 2919d6143a4eSToby Isaac + dm - The mesh, with coordinate maps defined either by a PetscDS for the coordinate DM (see DMGetCoordinateDM()) or 2920d6143a4eSToby Isaac implicitly by the coordinates of the corner vertices of the cell: as an affine map for simplicial elements, or 2921d6143a4eSToby Isaac as a multilinear map for tensor-product elements 2922d6143a4eSToby Isaac . cell - the cell whose map is used. 2923d6143a4eSToby Isaac . numPoints - the number of points to locate 29241b266c99SBarry Smith - realCoords - (numPoints x coordinate dimension) array of coordinates (see DMGetCoordinateDim()) 2925d6143a4eSToby Isaac 2926d6143a4eSToby Isaac Output Parameters: 2927d6143a4eSToby Isaac . refCoords - (numPoints x dimension) array of reference coordinates (see DMGetDimension()) 29281b266c99SBarry Smith 29291b266c99SBarry Smith Level: intermediate 293073c9229bSMatthew Knepley 293173c9229bSMatthew Knepley .seealso: DMPlexReferenceToCoordinates() 2932d6143a4eSToby Isaac @*/ 2933d6143a4eSToby Isaac PetscErrorCode DMPlexCoordinatesToReference(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[]) 2934d6143a4eSToby Isaac { 29359d150b73SToby Isaac PetscInt dimC, dimR, depth, cStart, cEnd, cEndInterior, i; 29369d150b73SToby Isaac DM coordDM = NULL; 29379d150b73SToby Isaac Vec coords; 29389d150b73SToby Isaac PetscFE fe = NULL; 29399d150b73SToby Isaac PetscErrorCode ierr; 29409d150b73SToby Isaac 2941d6143a4eSToby Isaac PetscFunctionBegin; 29429d150b73SToby Isaac PetscValidHeaderSpecific(dm,DM_CLASSID,1); 29439d150b73SToby Isaac ierr = DMGetDimension(dm,&dimR);CHKERRQ(ierr); 29449d150b73SToby Isaac ierr = DMGetCoordinateDim(dm,&dimC);CHKERRQ(ierr); 29459d150b73SToby Isaac if (dimR <= 0 || dimC <= 0 || numPoints <= 0) PetscFunctionReturn(0); 29469d150b73SToby Isaac ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr); 29479d150b73SToby Isaac ierr = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr); 29489d150b73SToby Isaac ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr); 29499d150b73SToby Isaac if (coordDM) { 29509d150b73SToby Isaac PetscInt coordFields; 29519d150b73SToby Isaac 29529d150b73SToby Isaac ierr = DMGetNumFields(coordDM,&coordFields);CHKERRQ(ierr); 29539d150b73SToby Isaac if (coordFields) { 29549d150b73SToby Isaac PetscClassId id; 29559d150b73SToby Isaac PetscObject disc; 29569d150b73SToby Isaac 295744a7f3ddSMatthew G. Knepley ierr = DMGetField(coordDM,0,NULL,&disc);CHKERRQ(ierr); 29589d150b73SToby Isaac ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr); 29599d150b73SToby Isaac if (id == PETSCFE_CLASSID) { 29609d150b73SToby Isaac fe = (PetscFE) disc; 29619d150b73SToby Isaac } 29629d150b73SToby Isaac } 29639d150b73SToby Isaac } 29649d150b73SToby Isaac ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr); 29659d150b73SToby Isaac ierr = DMPlexGetHybridBounds(dm,&cEndInterior,NULL,NULL,NULL);CHKERRQ(ierr); 29669d150b73SToby Isaac cEnd = cEndInterior > 0 ? cEndInterior : cEnd; 296713903a91SSatish Balay if (cell < cStart || cell >= cEnd) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"point %D not in cell range [%D,%D)",cell,cStart,cEnd); 29689d150b73SToby Isaac if (!fe) { /* implicit discretization: affine or multilinear */ 29699d150b73SToby Isaac PetscInt coneSize; 29709d150b73SToby Isaac PetscBool isSimplex, isTensor; 29719d150b73SToby Isaac 29729d150b73SToby Isaac ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr); 29739d150b73SToby Isaac isSimplex = (coneSize == (dimR + 1)) ? PETSC_TRUE : PETSC_FALSE; 29749d150b73SToby Isaac isTensor = (coneSize == ((depth == 1) ? (1 << dimR) : (2 * dimR))) ? PETSC_TRUE : PETSC_FALSE; 29759d150b73SToby Isaac if (isSimplex) { 29769d150b73SToby Isaac PetscReal detJ, *v0, *J, *invJ; 29779d150b73SToby Isaac 297869291d52SBarry Smith ierr = DMGetWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr); 29799d150b73SToby Isaac J = &v0[dimC]; 29809d150b73SToby Isaac invJ = &J[dimC * dimC]; 29819d150b73SToby Isaac ierr = DMPlexComputeCellGeometryAffineFEM(dm, cell, v0, J, invJ, &detJ);CHKERRQ(ierr); 29829d150b73SToby Isaac for (i = 0; i < numPoints; i++) { /* Apply the inverse affine transformation for each point */ 2983c330f8ffSToby Isaac const PetscReal x0[3] = {-1.,-1.,-1.}; 2984c330f8ffSToby Isaac 2985c330f8ffSToby Isaac CoordinatesRealToRef(dimC, dimR, x0, v0, invJ, &realCoords[dimC * i], &refCoords[dimR * i]); 29869d150b73SToby Isaac } 298769291d52SBarry Smith ierr = DMRestoreWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr); 29889d150b73SToby Isaac } else if (isTensor) { 29899d150b73SToby Isaac ierr = DMPlexCoordinatesToReference_Tensor(coordDM, cell, numPoints, realCoords, refCoords, coords, dimC, dimR);CHKERRQ(ierr); 29909d150b73SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unrecognized cone size %D",coneSize); 29919d150b73SToby Isaac } else { 29929d150b73SToby Isaac ierr = DMPlexCoordinatesToReference_FE(coordDM, fe, cell, numPoints, realCoords, refCoords, coords, dimC, dimR);CHKERRQ(ierr); 29939d150b73SToby Isaac } 29949d150b73SToby Isaac PetscFunctionReturn(0); 29959d150b73SToby Isaac } 29969d150b73SToby Isaac 29979d150b73SToby Isaac /*@ 29989d150b73SToby Isaac DMPlexReferenceToCoordinates - Map references coordinates to coordinates in the the mesh for a single element map. 29999d150b73SToby Isaac 30009d150b73SToby Isaac Not collective 30019d150b73SToby Isaac 30029d150b73SToby Isaac Input Parameters: 30039d150b73SToby Isaac + dm - The mesh, with coordinate maps defined either by a PetscDS for the coordinate DM (see DMGetCoordinateDM()) or 30049d150b73SToby Isaac implicitly by the coordinates of the corner vertices of the cell: as an affine map for simplicial elements, or 30059d150b73SToby Isaac as a multilinear map for tensor-product elements 30069d150b73SToby Isaac . cell - the cell whose map is used. 30079d150b73SToby Isaac . numPoints - the number of points to locate 30089d150b73SToby Isaac + refCoords - (numPoints x dimension) array of reference coordinates (see DMGetDimension()) 30099d150b73SToby Isaac 30109d150b73SToby Isaac Output Parameters: 30119d150b73SToby Isaac . realCoords - (numPoints x coordinate dimension) array of coordinates (see DMGetCoordinateDim()) 30121b266c99SBarry Smith 30131b266c99SBarry Smith Level: intermediate 301473c9229bSMatthew Knepley 301573c9229bSMatthew Knepley .seealso: DMPlexCoordinatesToReference() 30169d150b73SToby Isaac @*/ 30179d150b73SToby Isaac PetscErrorCode DMPlexReferenceToCoordinates(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[]) 30189d150b73SToby Isaac { 30199d150b73SToby Isaac PetscInt dimC, dimR, depth, cStart, cEnd, cEndInterior, i; 30209d150b73SToby Isaac DM coordDM = NULL; 30219d150b73SToby Isaac Vec coords; 30229d150b73SToby Isaac PetscFE fe = NULL; 30239d150b73SToby Isaac PetscErrorCode ierr; 30249d150b73SToby Isaac 30259d150b73SToby Isaac PetscFunctionBegin; 30269d150b73SToby Isaac PetscValidHeaderSpecific(dm,DM_CLASSID,1); 30279d150b73SToby Isaac ierr = DMGetDimension(dm,&dimR);CHKERRQ(ierr); 30289d150b73SToby Isaac ierr = DMGetCoordinateDim(dm,&dimC);CHKERRQ(ierr); 30299d150b73SToby Isaac if (dimR <= 0 || dimC <= 0 || numPoints <= 0) PetscFunctionReturn(0); 30309d150b73SToby Isaac ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr); 30319d150b73SToby Isaac ierr = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr); 30329d150b73SToby Isaac ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr); 30339d150b73SToby Isaac if (coordDM) { 30349d150b73SToby Isaac PetscInt coordFields; 30359d150b73SToby Isaac 30369d150b73SToby Isaac ierr = DMGetNumFields(coordDM,&coordFields);CHKERRQ(ierr); 30379d150b73SToby Isaac if (coordFields) { 30389d150b73SToby Isaac PetscClassId id; 30399d150b73SToby Isaac PetscObject disc; 30409d150b73SToby Isaac 304144a7f3ddSMatthew G. Knepley ierr = DMGetField(coordDM,0,NULL,&disc);CHKERRQ(ierr); 30429d150b73SToby Isaac ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr); 30439d150b73SToby Isaac if (id == PETSCFE_CLASSID) { 30449d150b73SToby Isaac fe = (PetscFE) disc; 30459d150b73SToby Isaac } 30469d150b73SToby Isaac } 30479d150b73SToby Isaac } 30489d150b73SToby Isaac ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr); 30499d150b73SToby Isaac ierr = DMPlexGetHybridBounds(dm,&cEndInterior,NULL,NULL,NULL);CHKERRQ(ierr); 30509d150b73SToby Isaac cEnd = cEndInterior > 0 ? cEndInterior : cEnd; 305113903a91SSatish Balay if (cell < cStart || cell >= cEnd) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"point %D not in cell range [%D,%D)",cell,cStart,cEnd); 30529d150b73SToby Isaac if (!fe) { /* implicit discretization: affine or multilinear */ 30539d150b73SToby Isaac PetscInt coneSize; 30549d150b73SToby Isaac PetscBool isSimplex, isTensor; 30559d150b73SToby Isaac 30569d150b73SToby Isaac ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr); 30579d150b73SToby Isaac isSimplex = (coneSize == (dimR + 1)) ? PETSC_TRUE : PETSC_FALSE; 30589d150b73SToby Isaac isTensor = (coneSize == ((depth == 1) ? (1 << dimR) : (2 * dimR))) ? PETSC_TRUE : PETSC_FALSE; 30599d150b73SToby Isaac if (isSimplex) { 30609d150b73SToby Isaac PetscReal detJ, *v0, *J; 30619d150b73SToby Isaac 306269291d52SBarry Smith ierr = DMGetWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr); 30639d150b73SToby Isaac J = &v0[dimC]; 30649d150b73SToby Isaac ierr = DMPlexComputeCellGeometryAffineFEM(dm, cell, v0, J, NULL, &detJ);CHKERRQ(ierr); 3065c330f8ffSToby Isaac for (i = 0; i < numPoints; i++) { /* Apply the affine transformation for each point */ 3066c330f8ffSToby Isaac const PetscReal xi0[3] = {-1.,-1.,-1.}; 3067c330f8ffSToby Isaac 3068c330f8ffSToby Isaac CoordinatesRefToReal(dimC, dimR, xi0, v0, J, &refCoords[dimR * i], &realCoords[dimC * i]); 30699d150b73SToby Isaac } 307069291d52SBarry Smith ierr = DMRestoreWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr); 30719d150b73SToby Isaac } else if (isTensor) { 30729d150b73SToby Isaac ierr = DMPlexReferenceToCoordinates_Tensor(coordDM, cell, numPoints, refCoords, realCoords, coords, dimC, dimR);CHKERRQ(ierr); 30739d150b73SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unrecognized cone size %D",coneSize); 30749d150b73SToby Isaac } else { 30759d150b73SToby Isaac ierr = DMPlexReferenceToCoordinates_FE(coordDM, fe, cell, numPoints, refCoords, realCoords, coords, dimC, dimR);CHKERRQ(ierr); 30769d150b73SToby Isaac } 3077d6143a4eSToby Isaac PetscFunctionReturn(0); 3078d6143a4eSToby Isaac } 3079