137045ce4SJed Brown /* Discretization tools */ 237045ce4SJed Brown 3*a6fc04d9SSatish Balay #include <petscconf.h> 4*a6fc04d9SSatish Balay #if defined(PETSC_HAVE_MATHIMF_H) 5*a6fc04d9SSatish Balay #include <mathimf.h> /* this needs to be included before math.h */ 6*a6fc04d9SSatish Balay #endif 7*a6fc04d9SSatish Balay 837045ce4SJed Brown #include <petscdt.h> /*I "petscdt.h" I*/ 937045ce4SJed Brown #include <petscblaslapack.h> 10194825f6SJed Brown #include <petsc-private/petscimpl.h> 11665c2dedSJed Brown #include <petscviewer.h> 1237045ce4SJed Brown 1337045ce4SJed Brown #undef __FUNCT__ 1437045ce4SJed Brown #define __FUNCT__ "PetscDTLegendreEval" 1537045ce4SJed Brown /*@ 1637045ce4SJed Brown PetscDTLegendreEval - evaluate Legendre polynomial at points 1737045ce4SJed Brown 1837045ce4SJed Brown Not Collective 1937045ce4SJed Brown 2037045ce4SJed Brown Input Arguments: 2137045ce4SJed Brown + npoints - number of spatial points to evaluate at 2237045ce4SJed Brown . points - array of locations to evaluate at 2337045ce4SJed Brown . ndegree - number of basis degrees to evaluate 2437045ce4SJed Brown - degrees - sorted array of degrees to evaluate 2537045ce4SJed Brown 2637045ce4SJed Brown Output Arguments: 270298fd71SBarry Smith + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or NULL) 280298fd71SBarry Smith . D - row-oriented derivative evaluation matrix (or NULL) 290298fd71SBarry Smith - D2 - row-oriented second derivative evaluation matrix (or NULL) 3037045ce4SJed Brown 3137045ce4SJed Brown Level: intermediate 3237045ce4SJed Brown 3337045ce4SJed Brown .seealso: PetscDTGaussQuadrature() 3437045ce4SJed Brown @*/ 3537045ce4SJed Brown PetscErrorCode PetscDTLegendreEval(PetscInt npoints,const PetscReal *points,PetscInt ndegree,const PetscInt *degrees,PetscReal *B,PetscReal *D,PetscReal *D2) 3637045ce4SJed Brown { 3737045ce4SJed Brown PetscInt i,maxdegree; 3837045ce4SJed Brown 3937045ce4SJed Brown PetscFunctionBegin; 4037045ce4SJed Brown if (!npoints || !ndegree) PetscFunctionReturn(0); 4137045ce4SJed Brown maxdegree = degrees[ndegree-1]; 4237045ce4SJed Brown for (i=0; i<npoints; i++) { 4337045ce4SJed Brown PetscReal pm1,pm2,pd1,pd2,pdd1,pdd2,x; 4437045ce4SJed Brown PetscInt j,k; 4537045ce4SJed Brown x = points[i]; 4637045ce4SJed Brown pm2 = 0; 4737045ce4SJed Brown pm1 = 1; 4837045ce4SJed Brown pd2 = 0; 4937045ce4SJed Brown pd1 = 0; 5037045ce4SJed Brown pdd2 = 0; 5137045ce4SJed Brown pdd1 = 0; 5237045ce4SJed Brown k = 0; 5337045ce4SJed Brown if (degrees[k] == 0) { 5437045ce4SJed Brown if (B) B[i*ndegree+k] = pm1; 5537045ce4SJed Brown if (D) D[i*ndegree+k] = pd1; 5637045ce4SJed Brown if (D2) D2[i*ndegree+k] = pdd1; 5737045ce4SJed Brown k++; 5837045ce4SJed Brown } 5937045ce4SJed Brown for (j=1; j<=maxdegree; j++,k++) { 6037045ce4SJed Brown PetscReal p,d,dd; 6137045ce4SJed Brown p = ((2*j-1)*x*pm1 - (j-1)*pm2)/j; 6237045ce4SJed Brown d = pd2 + (2*j-1)*pm1; 6337045ce4SJed Brown dd = pdd2 + (2*j-1)*pd1; 6437045ce4SJed Brown pm2 = pm1; 6537045ce4SJed Brown pm1 = p; 6637045ce4SJed Brown pd2 = pd1; 6737045ce4SJed Brown pd1 = d; 6837045ce4SJed Brown pdd2 = pdd1; 6937045ce4SJed Brown pdd1 = dd; 7037045ce4SJed Brown if (degrees[k] == j) { 7137045ce4SJed Brown if (B) B[i*ndegree+k] = p; 7237045ce4SJed Brown if (D) D[i*ndegree+k] = d; 7337045ce4SJed Brown if (D2) D2[i*ndegree+k] = dd; 7437045ce4SJed Brown } 7537045ce4SJed Brown } 7637045ce4SJed Brown } 7737045ce4SJed Brown PetscFunctionReturn(0); 7837045ce4SJed Brown } 7937045ce4SJed Brown 8037045ce4SJed Brown #undef __FUNCT__ 8137045ce4SJed Brown #define __FUNCT__ "PetscDTGaussQuadrature" 8237045ce4SJed Brown /*@ 8337045ce4SJed Brown PetscDTGaussQuadrature - create Gauss quadrature 8437045ce4SJed Brown 8537045ce4SJed Brown Not Collective 8637045ce4SJed Brown 8737045ce4SJed Brown Input Arguments: 8837045ce4SJed Brown + npoints - number of points 8937045ce4SJed Brown . a - left end of interval (often-1) 9037045ce4SJed Brown - b - right end of interval (often +1) 9137045ce4SJed Brown 9237045ce4SJed Brown Output Arguments: 9337045ce4SJed Brown + x - quadrature points 9437045ce4SJed Brown - w - quadrature weights 9537045ce4SJed Brown 9637045ce4SJed Brown Level: intermediate 9737045ce4SJed Brown 9837045ce4SJed Brown References: 9937045ce4SJed Brown Golub and Welsch, Calculation of Quadrature Rules, Math. Comp. 23(106), 221--230, 1969. 10037045ce4SJed Brown 10137045ce4SJed Brown .seealso: PetscDTLegendreEval() 10237045ce4SJed Brown @*/ 10337045ce4SJed Brown PetscErrorCode PetscDTGaussQuadrature(PetscInt npoints,PetscReal a,PetscReal b,PetscReal *x,PetscReal *w) 10437045ce4SJed Brown { 10537045ce4SJed Brown PetscErrorCode ierr; 10637045ce4SJed Brown PetscInt i; 10737045ce4SJed Brown PetscReal *work; 10837045ce4SJed Brown PetscScalar *Z; 10937045ce4SJed Brown PetscBLASInt N,LDZ,info; 11037045ce4SJed Brown 11137045ce4SJed Brown PetscFunctionBegin; 11237045ce4SJed Brown /* Set up the Golub-Welsch system */ 11337045ce4SJed Brown for (i=0; i<npoints; i++) { 11437045ce4SJed Brown x[i] = 0; /* diagonal is 0 */ 11537045ce4SJed Brown if (i) w[i-1] = 0.5 / PetscSqrtReal(1 - 1./PetscSqr(2*i)); 11637045ce4SJed Brown } 11737045ce4SJed Brown ierr = PetscRealView(npoints-1,w,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); 11837045ce4SJed Brown ierr = PetscMalloc2(npoints*npoints,PetscScalar,&Z,PetscMax(1,2*npoints-2),PetscReal,&work);CHKERRQ(ierr); 119c5df96a5SBarry Smith ierr = PetscBLASIntCast(npoints,&N);CHKERRQ(ierr); 12037045ce4SJed Brown LDZ = N; 12137045ce4SJed Brown ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1228b83055fSJed Brown PetscStackCallBLAS("LAPACKsteqr",LAPACKsteqr_("I",&N,x,w,Z,&LDZ,work,&info)); 12337045ce4SJed Brown ierr = PetscFPTrapPop();CHKERRQ(ierr); 1241c3d6f74SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"xSTEQR error"); 12537045ce4SJed Brown 12637045ce4SJed Brown for (i=0; i<(npoints+1)/2; i++) { 12737045ce4SJed Brown PetscReal y = 0.5 * (-x[i] + x[npoints-i-1]); /* enforces symmetry */ 12837045ce4SJed Brown x[i] = (a+b)/2 - y*(b-a)/2; 12937045ce4SJed Brown x[npoints-i-1] = (a+b)/2 + y*(b-a)/2; 1300d644c17SKarl Rupp 13137045ce4SJed Brown w[i] = w[npoints-1-i] = (b-a)*PetscSqr(0.5*PetscAbsScalar(Z[i*npoints] + Z[(npoints-i-1)*npoints])); 13237045ce4SJed Brown } 13337045ce4SJed Brown ierr = PetscFree2(Z,work);CHKERRQ(ierr); 13437045ce4SJed Brown PetscFunctionReturn(0); 13537045ce4SJed Brown } 136194825f6SJed Brown 137194825f6SJed Brown #undef __FUNCT__ 138494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTFactorial_Internal" 139494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 140494e7359SMatthew G. Knepley Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 141494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTFactorial_Internal(PetscInt n, PetscReal *factorial) 142494e7359SMatthew G. Knepley { 143494e7359SMatthew G. Knepley PetscReal f = 1.0; 144494e7359SMatthew G. Knepley PetscInt i; 145494e7359SMatthew G. Knepley 146494e7359SMatthew G. Knepley PetscFunctionBegin; 147494e7359SMatthew G. Knepley for (i = 1; i < n+1; ++i) f *= i; 148494e7359SMatthew G. Knepley *factorial = f; 149494e7359SMatthew G. Knepley PetscFunctionReturn(0); 150494e7359SMatthew G. Knepley } 151494e7359SMatthew G. Knepley 152494e7359SMatthew G. Knepley #undef __FUNCT__ 153494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTComputeJacobi" 154494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 155494e7359SMatthew G. Knepley Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 156494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 157494e7359SMatthew G. Knepley { 158494e7359SMatthew G. Knepley PetscReal apb, pn1, pn2; 159494e7359SMatthew G. Knepley PetscInt k; 160494e7359SMatthew G. Knepley 161494e7359SMatthew G. Knepley PetscFunctionBegin; 162494e7359SMatthew G. Knepley if (!n) {*P = 1.0; PetscFunctionReturn(0);} 163494e7359SMatthew G. Knepley if (n == 1) {*P = 0.5 * (a - b + (a + b + 2.0) * x); PetscFunctionReturn(0);} 164494e7359SMatthew G. Knepley apb = a + b; 165494e7359SMatthew G. Knepley pn2 = 1.0; 166494e7359SMatthew G. Knepley pn1 = 0.5 * (a - b + (apb + 2.0) * x); 167494e7359SMatthew G. Knepley *P = 0.0; 168494e7359SMatthew G. Knepley for (k = 2; k < n+1; ++k) { 169494e7359SMatthew G. Knepley PetscReal a1 = 2.0 * k * (k + apb) * (2.0*k + apb - 2.0); 170494e7359SMatthew G. Knepley PetscReal a2 = (2.0 * k + apb - 1.0) * (a*a - b*b); 171494e7359SMatthew G. Knepley PetscReal a3 = (2.0 * k + apb - 2.0) * (2.0 * k + apb - 1.0) * (2.0 * k + apb); 172494e7359SMatthew G. Knepley PetscReal a4 = 2.0 * (k + a - 1.0) * (k + b - 1.0) * (2.0 * k + apb); 173494e7359SMatthew G. Knepley 174494e7359SMatthew G. Knepley a2 = a2 / a1; 175494e7359SMatthew G. Knepley a3 = a3 / a1; 176494e7359SMatthew G. Knepley a4 = a4 / a1; 177494e7359SMatthew G. Knepley *P = (a2 + a3 * x) * pn1 - a4 * pn2; 178494e7359SMatthew G. Knepley pn2 = pn1; 179494e7359SMatthew G. Knepley pn1 = *P; 180494e7359SMatthew G. Knepley } 181494e7359SMatthew G. Knepley PetscFunctionReturn(0); 182494e7359SMatthew G. Knepley } 183494e7359SMatthew G. Knepley 184494e7359SMatthew G. Knepley #undef __FUNCT__ 185494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTComputeJacobiDerivative" 186494e7359SMatthew G. Knepley /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */ 187494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 188494e7359SMatthew G. Knepley { 189494e7359SMatthew G. Knepley PetscReal nP; 190494e7359SMatthew G. Knepley PetscErrorCode ierr; 191494e7359SMatthew G. Knepley 192494e7359SMatthew G. Knepley PetscFunctionBegin; 193494e7359SMatthew G. Knepley if (!n) {*P = 0.0; PetscFunctionReturn(0);} 194494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobi(a+1, b+1, n-1, x, &nP);CHKERRQ(ierr); 195494e7359SMatthew G. Knepley *P = 0.5 * (a + b + n + 1) * nP; 196494e7359SMatthew G. Knepley PetscFunctionReturn(0); 197494e7359SMatthew G. Knepley } 198494e7359SMatthew G. Knepley 199494e7359SMatthew G. Knepley #undef __FUNCT__ 200494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTMapSquareToTriangle_Internal" 201494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */ 202494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapSquareToTriangle_Internal(PetscReal x, PetscReal y, PetscReal *xi, PetscReal *eta) 203494e7359SMatthew G. Knepley { 204494e7359SMatthew G. Knepley PetscFunctionBegin; 205494e7359SMatthew G. Knepley *xi = 0.5 * (1.0 + x) * (1.0 - y) - 1.0; 206494e7359SMatthew G. Knepley *eta = y; 207494e7359SMatthew G. Knepley PetscFunctionReturn(0); 208494e7359SMatthew G. Knepley } 209494e7359SMatthew G. Knepley 210494e7359SMatthew G. Knepley #undef __FUNCT__ 211494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTMapCubeToTetrahedron_Internal" 212494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */ 213494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapCubeToTetrahedron_Internal(PetscReal x, PetscReal y, PetscReal z, PetscReal *xi, PetscReal *eta, PetscReal *zeta) 214494e7359SMatthew G. Knepley { 215494e7359SMatthew G. Knepley PetscFunctionBegin; 216494e7359SMatthew G. Knepley *xi = 0.25 * (1.0 + x) * (1.0 - y) * (1.0 - z) - 1.0; 217494e7359SMatthew G. Knepley *eta = 0.5 * (1.0 + y) * (1.0 - z) - 1.0; 218494e7359SMatthew G. Knepley *zeta = z; 219494e7359SMatthew G. Knepley PetscFunctionReturn(0); 220494e7359SMatthew G. Knepley } 221494e7359SMatthew G. Knepley 222494e7359SMatthew G. Knepley #undef __FUNCT__ 223494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTGaussJacobiQuadrature1D_Internal" 224494e7359SMatthew G. Knepley static PetscErrorCode PetscDTGaussJacobiQuadrature1D_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 225494e7359SMatthew G. Knepley { 226494e7359SMatthew G. Knepley PetscInt maxIter = 100; 227494e7359SMatthew G. Knepley PetscReal eps = 1.0e-8; 228a8291ba1SSatish Balay PetscReal a1, a2, a3, a4, a5, a6; 229494e7359SMatthew G. Knepley PetscInt k; 230494e7359SMatthew G. Knepley PetscErrorCode ierr; 231494e7359SMatthew G. Knepley 232494e7359SMatthew G. Knepley PetscFunctionBegin; 233a8291ba1SSatish Balay 234a8291ba1SSatish Balay a1 = pow(2, a+b+1); 235a8291ba1SSatish Balay #if defined(PETSC_HAVE_TGAMMA) 236a8291ba1SSatish Balay a2 = tgamma(a + npoints + 1); 237a8291ba1SSatish Balay a3 = tgamma(b + npoints + 1); 238a8291ba1SSatish Balay a4 = tgamma(a + b + npoints + 1); 239a8291ba1SSatish Balay #else 240a8291ba1SSatish Balay SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"tgamma() - math routine is unavailable."); 241a8291ba1SSatish Balay #endif 242a8291ba1SSatish Balay 243494e7359SMatthew G. Knepley ierr = PetscDTFactorial_Internal(npoints, &a5);CHKERRQ(ierr); 244494e7359SMatthew G. Knepley a6 = a1 * a2 * a3 / a4 / a5; 245494e7359SMatthew G. Knepley /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses. 246494e7359SMatthew G. Knepley Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */ 247494e7359SMatthew G. Knepley for (k = 0; k < npoints; ++k) { 2487f1c68b3SMatthew G. Knepley PetscReal r = -cos((2.0*k + 1.0) * PETSC_PI / (2.0 * npoints)), dP; 249494e7359SMatthew G. Knepley PetscInt j; 250494e7359SMatthew G. Knepley 251494e7359SMatthew G. Knepley if (k > 0) r = 0.5 * (r + x[k-1]); 252494e7359SMatthew G. Knepley for (j = 0; j < maxIter; ++j) { 253494e7359SMatthew G. Knepley PetscReal s = 0.0, delta, f, fp; 254494e7359SMatthew G. Knepley PetscInt i; 255494e7359SMatthew G. Knepley 256494e7359SMatthew G. Knepley for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]); 257494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobi(a, b, npoints, r, &f);CHKERRQ(ierr); 258494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobiDerivative(a, b, npoints, r, &fp);CHKERRQ(ierr); 259494e7359SMatthew G. Knepley delta = f / (fp - f * s); 260494e7359SMatthew G. Knepley r = r - delta; 261494e7359SMatthew G. Knepley if (fabs(delta) < eps) break; 262494e7359SMatthew G. Knepley } 263494e7359SMatthew G. Knepley x[k] = r; 264494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobiDerivative(a, b, npoints, x[k], &dP);CHKERRQ(ierr); 265494e7359SMatthew G. Knepley w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP); 266494e7359SMatthew G. Knepley } 267494e7359SMatthew G. Knepley PetscFunctionReturn(0); 268494e7359SMatthew G. Knepley } 269494e7359SMatthew G. Knepley 270494e7359SMatthew G. Knepley #undef __FUNCT__ 271494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTGaussJacobiQuadrature" 272fd9d31fbSMatthew G. Knepley /*@C 273494e7359SMatthew G. Knepley PetscDTGaussJacobiQuadrature - create Gauss-Jacobi quadrature for a simplex 274494e7359SMatthew G. Knepley 275494e7359SMatthew G. Knepley Not Collective 276494e7359SMatthew G. Knepley 277494e7359SMatthew G. Knepley Input Arguments: 278494e7359SMatthew G. Knepley + dim - The simplex dimension 279494e7359SMatthew G. Knepley . npoints - number of points 280494e7359SMatthew G. Knepley . a - left end of interval (often-1) 281494e7359SMatthew G. Knepley - b - right end of interval (often +1) 282494e7359SMatthew G. Knepley 283494e7359SMatthew G. Knepley Output Arguments: 284494e7359SMatthew G. Knepley + points - quadrature points 285494e7359SMatthew G. Knepley - weights - quadrature weights 286494e7359SMatthew G. Knepley 287494e7359SMatthew G. Knepley Level: intermediate 288494e7359SMatthew G. Knepley 289494e7359SMatthew G. Knepley References: 290494e7359SMatthew G. Knepley Karniadakis and Sherwin. 291494e7359SMatthew G. Knepley FIAT 292494e7359SMatthew G. Knepley 293494e7359SMatthew G. Knepley .seealso: PetscDTGaussQuadrature() 294494e7359SMatthew G. Knepley @*/ 295494e7359SMatthew G. Knepley PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt dim, PetscInt npoints, PetscReal a, PetscReal b, PetscReal *points[], PetscReal *weights[]) 296494e7359SMatthew G. Knepley { 297494e7359SMatthew G. Knepley PetscReal *px, *wx, *py, *wy, *pz, *wz, *x, *w; 298494e7359SMatthew G. Knepley PetscInt i, j, k; 299494e7359SMatthew G. Knepley PetscErrorCode ierr; 300494e7359SMatthew G. Knepley 301494e7359SMatthew G. Knepley PetscFunctionBegin; 302494e7359SMatthew G. Knepley if ((a != -1.0) || (b != 1.0)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now"); 303494e7359SMatthew G. Knepley switch (dim) { 304494e7359SMatthew G. Knepley case 1: 305494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints * sizeof(PetscReal), &x);CHKERRQ(ierr); 306494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints * sizeof(PetscReal), &w);CHKERRQ(ierr); 307494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, x, w);CHKERRQ(ierr); 308494e7359SMatthew G. Knepley break; 309494e7359SMatthew G. Knepley case 2: 310494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints*npoints*2 * sizeof(PetscReal), &x);CHKERRQ(ierr); 311494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints*npoints * sizeof(PetscReal), &w);CHKERRQ(ierr); 312494e7359SMatthew G. Knepley ierr = PetscMalloc4(npoints,PetscReal,&px,npoints,PetscReal,&wx,npoints,PetscReal,&py,npoints,PetscReal,&wy);CHKERRQ(ierr); 313494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr); 314494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr); 315494e7359SMatthew G. Knepley for (i = 0; i < npoints; ++i) { 316494e7359SMatthew G. Knepley for (j = 0; j < npoints; ++j) { 317494e7359SMatthew G. Knepley ierr = PetscDTMapSquareToTriangle_Internal(px[i], py[j], &x[(i*npoints+j)*2+0], &x[(i*npoints+j)*2+1]);CHKERRQ(ierr); 318494e7359SMatthew G. Knepley w[i*npoints+j] = 0.5 * wx[i] * wy[j]; 319494e7359SMatthew G. Knepley } 320494e7359SMatthew G. Knepley } 321494e7359SMatthew G. Knepley ierr = PetscFree4(px,wx,py,wy);CHKERRQ(ierr); 322494e7359SMatthew G. Knepley break; 323494e7359SMatthew G. Knepley case 3: 324494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints*npoints*3 * sizeof(PetscReal), &x);CHKERRQ(ierr); 325494e7359SMatthew G. Knepley ierr = PetscMalloc(npoints*npoints * sizeof(PetscReal), &w);CHKERRQ(ierr); 326494e7359SMatthew G. Knepley ierr = PetscMalloc6(npoints,PetscReal,&px,npoints,PetscReal,&wx,npoints,PetscReal,&py,npoints,PetscReal,&wy,npoints,PetscReal,&pz,npoints,PetscReal,&wz);CHKERRQ(ierr); 327494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr); 328494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr); 329494e7359SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 2.0, 0.0, pz, wz);CHKERRQ(ierr); 330494e7359SMatthew G. Knepley for (i = 0; i < npoints; ++i) { 331494e7359SMatthew G. Knepley for (j = 0; j < npoints; ++j) { 332494e7359SMatthew G. Knepley for (k = 0; k < npoints; ++k) { 333494e7359SMatthew G. Knepley ierr = PetscDTMapCubeToTetrahedron_Internal(px[i], py[j], pz[k], &x[((i*npoints+j)*npoints+k)*3+0], &x[((i*npoints+j)*npoints+k)*3+1], &x[((i*npoints+j)*npoints+k)*3+2]);CHKERRQ(ierr); 334494e7359SMatthew G. Knepley w[(i*npoints+j)*npoints+k] = 0.125 * wx[i] * wy[j] * wz[k]; 335494e7359SMatthew G. Knepley } 336494e7359SMatthew G. Knepley } 337494e7359SMatthew G. Knepley } 338494e7359SMatthew G. Knepley ierr = PetscFree6(px,wx,py,wy,pz,wz);CHKERRQ(ierr); 339494e7359SMatthew G. Knepley break; 340494e7359SMatthew G. Knepley default: 341494e7359SMatthew G. Knepley SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim); 342494e7359SMatthew G. Knepley } 343494e7359SMatthew G. Knepley if (points) *points = x; 344494e7359SMatthew G. Knepley if (weights) *weights = w; 345494e7359SMatthew G. Knepley PetscFunctionReturn(0); 346494e7359SMatthew G. Knepley } 347494e7359SMatthew G. Knepley 348494e7359SMatthew G. Knepley #undef __FUNCT__ 349194825f6SJed Brown #define __FUNCT__ "PetscDTPseudoInverseQR" 350194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n 351194825f6SJed Brown * A in column-major format 352194825f6SJed Brown * Ainv in row-major format 353194825f6SJed Brown * tau has length m 354194825f6SJed Brown * worksize must be >= max(1,n) 355194825f6SJed Brown */ 356194825f6SJed Brown static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m,PetscInt mstride,PetscInt n,PetscReal *A_in,PetscReal *Ainv_out,PetscScalar *tau,PetscInt worksize,PetscScalar *work) 357194825f6SJed Brown { 358194825f6SJed Brown PetscErrorCode ierr; 359194825f6SJed Brown PetscBLASInt M,N,K,lda,ldb,ldwork,info; 360194825f6SJed Brown PetscScalar *A,*Ainv,*R,*Q,Alpha; 361194825f6SJed Brown 362194825f6SJed Brown PetscFunctionBegin; 363194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 364194825f6SJed Brown { 365194825f6SJed Brown PetscInt i,j; 366194825f6SJed Brown ierr = PetscMalloc2(m*n,PetscScalar,&A,m*n,PetscScalar,&Ainv);CHKERRQ(ierr); 367194825f6SJed Brown for (j=0; j<n; j++) { 368194825f6SJed Brown for (i=0; i<m; i++) A[i+m*j] = A_in[i+mstride*j]; 369194825f6SJed Brown } 370194825f6SJed Brown mstride = m; 371194825f6SJed Brown } 372194825f6SJed Brown #else 373194825f6SJed Brown A = A_in; 374194825f6SJed Brown Ainv = Ainv_out; 375194825f6SJed Brown #endif 376194825f6SJed Brown 377194825f6SJed Brown ierr = PetscBLASIntCast(m,&M);CHKERRQ(ierr); 378194825f6SJed Brown ierr = PetscBLASIntCast(n,&N);CHKERRQ(ierr); 379194825f6SJed Brown ierr = PetscBLASIntCast(mstride,&lda);CHKERRQ(ierr); 380194825f6SJed Brown ierr = PetscBLASIntCast(worksize,&ldwork);CHKERRQ(ierr); 381194825f6SJed Brown ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 382194825f6SJed Brown LAPACKgeqrf_(&M,&N,A,&lda,tau,work,&ldwork,&info); 383194825f6SJed Brown ierr = PetscFPTrapPop();CHKERRQ(ierr); 384194825f6SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xGEQRF error"); 385194825f6SJed Brown R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */ 386194825f6SJed Brown 387194825f6SJed Brown /* Extract an explicit representation of Q */ 388194825f6SJed Brown Q = Ainv; 389194825f6SJed Brown ierr = PetscMemcpy(Q,A,mstride*n*sizeof(PetscScalar));CHKERRQ(ierr); 390194825f6SJed Brown K = N; /* full rank */ 391194825f6SJed Brown LAPACKungqr_(&M,&N,&K,Q,&lda,tau,work,&ldwork,&info); 392194825f6SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xORGQR/xUNGQR error"); 393194825f6SJed Brown 394194825f6SJed Brown /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */ 395194825f6SJed Brown Alpha = 1.0; 396194825f6SJed Brown ldb = lda; 397194825f6SJed Brown BLAStrsm_("Right","Upper","ConjugateTranspose","NotUnitTriangular",&M,&N,&Alpha,R,&lda,Q,&ldb); 398194825f6SJed Brown /* Ainv is Q, overwritten with inverse */ 399194825f6SJed Brown 400194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 401194825f6SJed Brown { 402194825f6SJed Brown PetscInt i; 403194825f6SJed Brown for (i=0; i<m*n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]); 404194825f6SJed Brown ierr = PetscFree2(A,Ainv);CHKERRQ(ierr); 405194825f6SJed Brown } 406194825f6SJed Brown #endif 407194825f6SJed Brown PetscFunctionReturn(0); 408194825f6SJed Brown } 409194825f6SJed Brown 410194825f6SJed Brown #undef __FUNCT__ 411194825f6SJed Brown #define __FUNCT__ "PetscDTLegendreIntegrate" 412194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */ 413194825f6SJed Brown static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval,const PetscReal *x,PetscInt ndegree,const PetscInt *degrees,PetscBool Transpose,PetscReal *B) 414194825f6SJed Brown { 415194825f6SJed Brown PetscErrorCode ierr; 416194825f6SJed Brown PetscReal *Bv; 417194825f6SJed Brown PetscInt i,j; 418194825f6SJed Brown 419194825f6SJed Brown PetscFunctionBegin; 420194825f6SJed Brown ierr = PetscMalloc((ninterval+1)*ndegree*sizeof(PetscReal),&Bv);CHKERRQ(ierr); 421194825f6SJed Brown /* Point evaluation of L_p on all the source vertices */ 422194825f6SJed Brown ierr = PetscDTLegendreEval(ninterval+1,x,ndegree,degrees,Bv,NULL,NULL);CHKERRQ(ierr); 423194825f6SJed Brown /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */ 424194825f6SJed Brown for (i=0; i<ninterval; i++) { 425194825f6SJed Brown for (j=0; j<ndegree; j++) { 426194825f6SJed Brown if (Transpose) B[i+ninterval*j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j]; 427194825f6SJed Brown else B[i*ndegree+j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j]; 428194825f6SJed Brown } 429194825f6SJed Brown } 430194825f6SJed Brown ierr = PetscFree(Bv);CHKERRQ(ierr); 431194825f6SJed Brown PetscFunctionReturn(0); 432194825f6SJed Brown } 433194825f6SJed Brown 434194825f6SJed Brown #undef __FUNCT__ 435194825f6SJed Brown #define __FUNCT__ "PetscDTReconstructPoly" 436194825f6SJed Brown /*@ 437194825f6SJed Brown PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals 438194825f6SJed Brown 439194825f6SJed Brown Not Collective 440194825f6SJed Brown 441194825f6SJed Brown Input Arguments: 442194825f6SJed Brown + degree - degree of reconstruction polynomial 443194825f6SJed Brown . nsource - number of source intervals 444194825f6SJed Brown . sourcex - sorted coordinates of source cell boundaries (length nsource+1) 445194825f6SJed Brown . ntarget - number of target intervals 446194825f6SJed Brown - targetx - sorted coordinates of target cell boundaries (length ntarget+1) 447194825f6SJed Brown 448194825f6SJed Brown Output Arguments: 449194825f6SJed Brown . R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s] 450194825f6SJed Brown 451194825f6SJed Brown Level: advanced 452194825f6SJed Brown 453194825f6SJed Brown .seealso: PetscDTLegendreEval() 454194825f6SJed Brown @*/ 455194825f6SJed Brown PetscErrorCode PetscDTReconstructPoly(PetscInt degree,PetscInt nsource,const PetscReal *sourcex,PetscInt ntarget,const PetscReal *targetx,PetscReal *R) 456194825f6SJed Brown { 457194825f6SJed Brown PetscErrorCode ierr; 458194825f6SJed Brown PetscInt i,j,k,*bdegrees,worksize; 459194825f6SJed Brown PetscReal xmin,xmax,center,hscale,*sourcey,*targety,*Bsource,*Bsinv,*Btarget; 460194825f6SJed Brown PetscScalar *tau,*work; 461194825f6SJed Brown 462194825f6SJed Brown PetscFunctionBegin; 463194825f6SJed Brown PetscValidRealPointer(sourcex,3); 464194825f6SJed Brown PetscValidRealPointer(targetx,5); 465194825f6SJed Brown PetscValidRealPointer(R,6); 466194825f6SJed Brown if (degree >= nsource) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Reconstruction degree %D must be less than number of source intervals %D",degree,nsource); 467194825f6SJed Brown #if defined(PETSC_USE_DEBUG) 468194825f6SJed Brown for (i=0; i<nsource; i++) { 469194825f6SJed Brown if (sourcex[i] >= sourcex[i+1]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_CORRUPT,"Source interval %D has negative orientation (%G,%G)",i,sourcex[i],sourcex[i+1]); 470194825f6SJed Brown } 471194825f6SJed Brown for (i=0; i<ntarget; i++) { 472194825f6SJed Brown if (targetx[i] >= targetx[i+1]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_CORRUPT,"Target interval %D has negative orientation (%G,%G)",i,targetx[i],targetx[i+1]); 473194825f6SJed Brown } 474194825f6SJed Brown #endif 475194825f6SJed Brown xmin = PetscMin(sourcex[0],targetx[0]); 476194825f6SJed Brown xmax = PetscMax(sourcex[nsource],targetx[ntarget]); 477194825f6SJed Brown center = (xmin + xmax)/2; 478194825f6SJed Brown hscale = (xmax - xmin)/2; 479194825f6SJed Brown worksize = nsource; 480194825f6SJed Brown ierr = PetscMalloc4(degree+1,PetscInt,&bdegrees,nsource+1,PetscReal,&sourcey,nsource*(degree+1),PetscReal,&Bsource,worksize,PetscScalar,&work);CHKERRQ(ierr); 48182772646SJed Brown ierr = PetscMalloc4(nsource,PetscScalar,&tau,nsource*(degree+1),PetscReal,&Bsinv,ntarget+1,PetscReal,&targety,ntarget*(degree+1),PetscReal,&Btarget);CHKERRQ(ierr); 482194825f6SJed Brown for (i=0; i<=nsource; i++) sourcey[i] = (sourcex[i]-center)/hscale; 483194825f6SJed Brown for (i=0; i<=degree; i++) bdegrees[i] = i+1; 484194825f6SJed Brown ierr = PetscDTLegendreIntegrate(nsource,sourcey,degree+1,bdegrees,PETSC_TRUE,Bsource);CHKERRQ(ierr); 485194825f6SJed Brown ierr = PetscDTPseudoInverseQR(nsource,nsource,degree+1,Bsource,Bsinv,tau,nsource,work);CHKERRQ(ierr); 486194825f6SJed Brown for (i=0; i<=ntarget; i++) targety[i] = (targetx[i]-center)/hscale; 487194825f6SJed Brown ierr = PetscDTLegendreIntegrate(ntarget,targety,degree+1,bdegrees,PETSC_FALSE,Btarget);CHKERRQ(ierr); 488194825f6SJed Brown for (i=0; i<ntarget; i++) { 489194825f6SJed Brown PetscReal rowsum = 0; 490194825f6SJed Brown for (j=0; j<nsource; j++) { 491194825f6SJed Brown PetscReal sum = 0; 492194825f6SJed Brown for (k=0; k<degree+1; k++) { 493194825f6SJed Brown sum += Btarget[i*(degree+1)+k] * Bsinv[k*nsource+j]; 494194825f6SJed Brown } 495194825f6SJed Brown R[i*nsource+j] = sum; 496194825f6SJed Brown rowsum += sum; 497194825f6SJed Brown } 498194825f6SJed Brown for (j=0; j<nsource; j++) R[i*nsource+j] /= rowsum; /* normalize each row */ 499194825f6SJed Brown } 500194825f6SJed Brown ierr = PetscFree4(bdegrees,sourcey,Bsource,work);CHKERRQ(ierr); 501194825f6SJed Brown ierr = PetscFree4(tau,Bsinv,targety,Btarget);CHKERRQ(ierr); 502194825f6SJed Brown PetscFunctionReturn(0); 503194825f6SJed Brown } 504