xref: /petsc/src/dm/dt/interface/dt.c (revision 88393a60d6280223acaa9f85de006825764cc7ec)
137045ce4SJed Brown /* Discretization tools */
237045ce4SJed Brown 
3a6fc04d9SSatish Balay #include <petscconf.h>
4a6fc04d9SSatish Balay #if defined(PETSC_HAVE_MATHIMF_H)
5a6fc04d9SSatish Balay #include <mathimf.h>           /* this needs to be included before math.h */
6a6fc04d9SSatish Balay #endif
7a6fc04d9SSatish 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 = PetscMalloc2(npoints*npoints,PetscScalar,&Z,PetscMax(1,2*npoints-2),PetscReal,&work);CHKERRQ(ierr);
118c5df96a5SBarry Smith   ierr = PetscBLASIntCast(npoints,&N);CHKERRQ(ierr);
11937045ce4SJed Brown   LDZ  = N;
12037045ce4SJed Brown   ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1218b83055fSJed Brown   PetscStackCallBLAS("LAPACKsteqr",LAPACKsteqr_("I",&N,x,w,Z,&LDZ,work,&info));
12237045ce4SJed Brown   ierr = PetscFPTrapPop();CHKERRQ(ierr);
1231c3d6f74SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"xSTEQR error");
12437045ce4SJed Brown 
12537045ce4SJed Brown   for (i=0; i<(npoints+1)/2; i++) {
12637045ce4SJed Brown     PetscReal y = 0.5 * (-x[i] + x[npoints-i-1]); /* enforces symmetry */
12737045ce4SJed Brown     x[i]           = (a+b)/2 - y*(b-a)/2;
12837045ce4SJed Brown     x[npoints-i-1] = (a+b)/2 + y*(b-a)/2;
1290d644c17SKarl Rupp 
130*88393a60SJed Brown     w[i] = w[npoints-1-i] = 0.5*(b-a)*(PetscSqr(PetscAbsScalar(Z[i*npoints])) + PetscSqr(PetscAbsScalar(Z[(npoints-i-1)*npoints])));
13137045ce4SJed Brown   }
13237045ce4SJed Brown   ierr = PetscFree2(Z,work);CHKERRQ(ierr);
13337045ce4SJed Brown   PetscFunctionReturn(0);
13437045ce4SJed Brown }
135194825f6SJed Brown 
136194825f6SJed Brown #undef __FUNCT__
137494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTFactorial_Internal"
138494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x.
139494e7359SMatthew G. Knepley    Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */
140494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTFactorial_Internal(PetscInt n, PetscReal *factorial)
141494e7359SMatthew G. Knepley {
142494e7359SMatthew G. Knepley   PetscReal f = 1.0;
143494e7359SMatthew G. Knepley   PetscInt  i;
144494e7359SMatthew G. Knepley 
145494e7359SMatthew G. Knepley   PetscFunctionBegin;
146494e7359SMatthew G. Knepley   for (i = 1; i < n+1; ++i) f *= i;
147494e7359SMatthew G. Knepley   *factorial = f;
148494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
149494e7359SMatthew G. Knepley }
150494e7359SMatthew G. Knepley 
151494e7359SMatthew G. Knepley #undef __FUNCT__
152494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTComputeJacobi"
153494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x.
154494e7359SMatthew G. Knepley    Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */
155494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P)
156494e7359SMatthew G. Knepley {
157494e7359SMatthew G. Knepley   PetscReal apb, pn1, pn2;
158494e7359SMatthew G. Knepley   PetscInt  k;
159494e7359SMatthew G. Knepley 
160494e7359SMatthew G. Knepley   PetscFunctionBegin;
161494e7359SMatthew G. Knepley   if (!n) {*P = 1.0; PetscFunctionReturn(0);}
162494e7359SMatthew G. Knepley   if (n == 1) {*P = 0.5 * (a - b + (a + b + 2.0) * x); PetscFunctionReturn(0);}
163494e7359SMatthew G. Knepley   apb = a + b;
164494e7359SMatthew G. Knepley   pn2 = 1.0;
165494e7359SMatthew G. Knepley   pn1 = 0.5 * (a - b + (apb + 2.0) * x);
166494e7359SMatthew G. Knepley   *P  = 0.0;
167494e7359SMatthew G. Knepley   for (k = 2; k < n+1; ++k) {
168494e7359SMatthew G. Knepley     PetscReal a1 = 2.0 * k * (k + apb) * (2.0*k + apb - 2.0);
169494e7359SMatthew G. Knepley     PetscReal a2 = (2.0 * k + apb - 1.0) * (a*a - b*b);
170494e7359SMatthew G. Knepley     PetscReal a3 = (2.0 * k + apb - 2.0) * (2.0 * k + apb - 1.0) * (2.0 * k + apb);
171494e7359SMatthew G. Knepley     PetscReal a4 = 2.0 * (k + a - 1.0) * (k + b - 1.0) * (2.0 * k + apb);
172494e7359SMatthew G. Knepley 
173494e7359SMatthew G. Knepley     a2  = a2 / a1;
174494e7359SMatthew G. Knepley     a3  = a3 / a1;
175494e7359SMatthew G. Knepley     a4  = a4 / a1;
176494e7359SMatthew G. Knepley     *P  = (a2 + a3 * x) * pn1 - a4 * pn2;
177494e7359SMatthew G. Knepley     pn2 = pn1;
178494e7359SMatthew G. Knepley     pn1 = *P;
179494e7359SMatthew G. Knepley   }
180494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
181494e7359SMatthew G. Knepley }
182494e7359SMatthew G. Knepley 
183494e7359SMatthew G. Knepley #undef __FUNCT__
184494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTComputeJacobiDerivative"
185494e7359SMatthew G. Knepley /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */
186494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P)
187494e7359SMatthew G. Knepley {
188494e7359SMatthew G. Knepley   PetscReal      nP;
189494e7359SMatthew G. Knepley   PetscErrorCode ierr;
190494e7359SMatthew G. Knepley 
191494e7359SMatthew G. Knepley   PetscFunctionBegin;
192494e7359SMatthew G. Knepley   if (!n) {*P = 0.0; PetscFunctionReturn(0);}
193494e7359SMatthew G. Knepley   ierr = PetscDTComputeJacobi(a+1, b+1, n-1, x, &nP);CHKERRQ(ierr);
194494e7359SMatthew G. Knepley   *P   = 0.5 * (a + b + n + 1) * nP;
195494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
196494e7359SMatthew G. Knepley }
197494e7359SMatthew G. Knepley 
198494e7359SMatthew G. Knepley #undef __FUNCT__
199494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTMapSquareToTriangle_Internal"
200494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */
201494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapSquareToTriangle_Internal(PetscReal x, PetscReal y, PetscReal *xi, PetscReal *eta)
202494e7359SMatthew G. Knepley {
203494e7359SMatthew G. Knepley   PetscFunctionBegin;
204494e7359SMatthew G. Knepley   *xi  = 0.5 * (1.0 + x) * (1.0 - y) - 1.0;
205494e7359SMatthew G. Knepley   *eta = y;
206494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
207494e7359SMatthew G. Knepley }
208494e7359SMatthew G. Knepley 
209494e7359SMatthew G. Knepley #undef __FUNCT__
210494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTMapCubeToTetrahedron_Internal"
211494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */
212494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapCubeToTetrahedron_Internal(PetscReal x, PetscReal y, PetscReal z, PetscReal *xi, PetscReal *eta, PetscReal *zeta)
213494e7359SMatthew G. Knepley {
214494e7359SMatthew G. Knepley   PetscFunctionBegin;
215494e7359SMatthew G. Knepley   *xi   = 0.25 * (1.0 + x) * (1.0 - y) * (1.0 - z) - 1.0;
216494e7359SMatthew G. Knepley   *eta  = 0.5  * (1.0 + y) * (1.0 - z) - 1.0;
217494e7359SMatthew G. Knepley   *zeta = z;
218494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
219494e7359SMatthew G. Knepley }
220494e7359SMatthew G. Knepley 
221494e7359SMatthew G. Knepley #undef __FUNCT__
222494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTGaussJacobiQuadrature1D_Internal"
223494e7359SMatthew G. Knepley static PetscErrorCode PetscDTGaussJacobiQuadrature1D_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w)
224494e7359SMatthew G. Knepley {
225494e7359SMatthew G. Knepley   PetscInt       maxIter = 100;
226494e7359SMatthew G. Knepley   PetscReal      eps     = 1.0e-8;
227a8291ba1SSatish Balay   PetscReal      a1, a2, a3, a4, a5, a6;
228494e7359SMatthew G. Knepley   PetscInt       k;
229494e7359SMatthew G. Knepley   PetscErrorCode ierr;
230494e7359SMatthew G. Knepley 
231494e7359SMatthew G. Knepley   PetscFunctionBegin;
232a8291ba1SSatish Balay 
233a8291ba1SSatish Balay   a1      = pow(2, a+b+1);
234a8291ba1SSatish Balay #if defined(PETSC_HAVE_TGAMMA)
235a8291ba1SSatish Balay   a2      = tgamma(a + npoints + 1);
236a8291ba1SSatish Balay   a3      = tgamma(b + npoints + 1);
237a8291ba1SSatish Balay   a4      = tgamma(a + b + npoints + 1);
238a8291ba1SSatish Balay #else
239a8291ba1SSatish Balay   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"tgamma() - math routine is unavailable.");
240a8291ba1SSatish Balay #endif
241a8291ba1SSatish Balay 
242494e7359SMatthew G. Knepley   ierr = PetscDTFactorial_Internal(npoints, &a5);CHKERRQ(ierr);
243494e7359SMatthew G. Knepley   a6   = a1 * a2 * a3 / a4 / a5;
244494e7359SMatthew G. Knepley   /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses.
245494e7359SMatthew G. Knepley    Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */
246494e7359SMatthew G. Knepley   for (k = 0; k < npoints; ++k) {
2477f1c68b3SMatthew G. Knepley     PetscReal r = -cos((2.0*k + 1.0) * PETSC_PI / (2.0 * npoints)), dP;
248494e7359SMatthew G. Knepley     PetscInt  j;
249494e7359SMatthew G. Knepley 
250494e7359SMatthew G. Knepley     if (k > 0) r = 0.5 * (r + x[k-1]);
251494e7359SMatthew G. Knepley     for (j = 0; j < maxIter; ++j) {
252494e7359SMatthew G. Knepley       PetscReal s = 0.0, delta, f, fp;
253494e7359SMatthew G. Knepley       PetscInt  i;
254494e7359SMatthew G. Knepley 
255494e7359SMatthew G. Knepley       for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]);
256494e7359SMatthew G. Knepley       ierr = PetscDTComputeJacobi(a, b, npoints, r, &f);CHKERRQ(ierr);
257494e7359SMatthew G. Knepley       ierr = PetscDTComputeJacobiDerivative(a, b, npoints, r, &fp);CHKERRQ(ierr);
258494e7359SMatthew G. Knepley       delta = f / (fp - f * s);
259494e7359SMatthew G. Knepley       r     = r - delta;
260494e7359SMatthew G. Knepley       if (fabs(delta) < eps) break;
261494e7359SMatthew G. Knepley     }
262494e7359SMatthew G. Knepley     x[k] = r;
263494e7359SMatthew G. Knepley     ierr = PetscDTComputeJacobiDerivative(a, b, npoints, x[k], &dP);CHKERRQ(ierr);
264494e7359SMatthew G. Knepley     w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP);
265494e7359SMatthew G. Knepley   }
266494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
267494e7359SMatthew G. Knepley }
268494e7359SMatthew G. Knepley 
269494e7359SMatthew G. Knepley #undef __FUNCT__
270494e7359SMatthew G. Knepley #define __FUNCT__ "PetscDTGaussJacobiQuadrature"
271fd9d31fbSMatthew G. Knepley /*@C
272494e7359SMatthew G. Knepley   PetscDTGaussJacobiQuadrature - create Gauss-Jacobi quadrature for a simplex
273494e7359SMatthew G. Knepley 
274494e7359SMatthew G. Knepley   Not Collective
275494e7359SMatthew G. Knepley 
276494e7359SMatthew G. Knepley   Input Arguments:
277494e7359SMatthew G. Knepley + dim - The simplex dimension
278494e7359SMatthew G. Knepley . npoints - number of points
279494e7359SMatthew G. Knepley . a - left end of interval (often-1)
280494e7359SMatthew G. Knepley - b - right end of interval (often +1)
281494e7359SMatthew G. Knepley 
282494e7359SMatthew G. Knepley   Output Arguments:
283494e7359SMatthew G. Knepley + points - quadrature points
284494e7359SMatthew G. Knepley - weights - quadrature weights
285494e7359SMatthew G. Knepley 
286494e7359SMatthew G. Knepley   Level: intermediate
287494e7359SMatthew G. Knepley 
288494e7359SMatthew G. Knepley   References:
289494e7359SMatthew G. Knepley   Karniadakis and Sherwin.
290494e7359SMatthew G. Knepley   FIAT
291494e7359SMatthew G. Knepley 
292494e7359SMatthew G. Knepley .seealso: PetscDTGaussQuadrature()
293494e7359SMatthew G. Knepley @*/
294494e7359SMatthew G. Knepley PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt dim, PetscInt npoints, PetscReal a, PetscReal b, PetscReal *points[], PetscReal *weights[])
295494e7359SMatthew G. Knepley {
296494e7359SMatthew G. Knepley   PetscReal     *px, *wx, *py, *wy, *pz, *wz, *x, *w;
297494e7359SMatthew G. Knepley   PetscInt       i, j, k;
298494e7359SMatthew G. Knepley   PetscErrorCode ierr;
299494e7359SMatthew G. Knepley 
300494e7359SMatthew G. Knepley   PetscFunctionBegin;
301494e7359SMatthew G. Knepley   if ((a != -1.0) || (b != 1.0)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now");
302494e7359SMatthew G. Knepley   switch (dim) {
303494e7359SMatthew G. Knepley   case 1:
304494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints * sizeof(PetscReal), &x);CHKERRQ(ierr);
305494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints * sizeof(PetscReal), &w);CHKERRQ(ierr);
306494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, x, w);CHKERRQ(ierr);
307494e7359SMatthew G. Knepley     break;
308494e7359SMatthew G. Knepley   case 2:
309494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints*npoints*2 * sizeof(PetscReal), &x);CHKERRQ(ierr);
310494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints*npoints   * sizeof(PetscReal), &w);CHKERRQ(ierr);
311494e7359SMatthew G. Knepley     ierr = PetscMalloc4(npoints,PetscReal,&px,npoints,PetscReal,&wx,npoints,PetscReal,&py,npoints,PetscReal,&wy);CHKERRQ(ierr);
312494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr);
313494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr);
314494e7359SMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
315494e7359SMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
316494e7359SMatthew G. Knepley         ierr = PetscDTMapSquareToTriangle_Internal(px[i], py[j], &x[(i*npoints+j)*2+0], &x[(i*npoints+j)*2+1]);CHKERRQ(ierr);
317494e7359SMatthew G. Knepley         w[i*npoints+j] = 0.5 * wx[i] * wy[j];
318494e7359SMatthew G. Knepley       }
319494e7359SMatthew G. Knepley     }
320494e7359SMatthew G. Knepley     ierr = PetscFree4(px,wx,py,wy);CHKERRQ(ierr);
321494e7359SMatthew G. Knepley     break;
322494e7359SMatthew G. Knepley   case 3:
323494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints*npoints*3 * sizeof(PetscReal), &x);CHKERRQ(ierr);
324494e7359SMatthew G. Knepley     ierr = PetscMalloc(npoints*npoints   * sizeof(PetscReal), &w);CHKERRQ(ierr);
325494e7359SMatthew G. Knepley     ierr = PetscMalloc6(npoints,PetscReal,&px,npoints,PetscReal,&wx,npoints,PetscReal,&py,npoints,PetscReal,&wy,npoints,PetscReal,&pz,npoints,PetscReal,&wz);CHKERRQ(ierr);
326494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr);
327494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr);
328494e7359SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 2.0, 0.0, pz, wz);CHKERRQ(ierr);
329494e7359SMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
330494e7359SMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
331494e7359SMatthew G. Knepley         for (k = 0; k < npoints; ++k) {
332494e7359SMatthew 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);
333494e7359SMatthew G. Knepley           w[(i*npoints+j)*npoints+k] = 0.125 * wx[i] * wy[j] * wz[k];
334494e7359SMatthew G. Knepley         }
335494e7359SMatthew G. Knepley       }
336494e7359SMatthew G. Knepley     }
337494e7359SMatthew G. Knepley     ierr = PetscFree6(px,wx,py,wy,pz,wz);CHKERRQ(ierr);
338494e7359SMatthew G. Knepley     break;
339494e7359SMatthew G. Knepley   default:
340494e7359SMatthew G. Knepley     SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim);
341494e7359SMatthew G. Knepley   }
342494e7359SMatthew G. Knepley   if (points)  *points  = x;
343494e7359SMatthew G. Knepley   if (weights) *weights = w;
344494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
345494e7359SMatthew G. Knepley }
346494e7359SMatthew G. Knepley 
347494e7359SMatthew G. Knepley #undef __FUNCT__
348194825f6SJed Brown #define __FUNCT__ "PetscDTPseudoInverseQR"
349194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n
350194825f6SJed Brown  * A in column-major format
351194825f6SJed Brown  * Ainv in row-major format
352194825f6SJed Brown  * tau has length m
353194825f6SJed Brown  * worksize must be >= max(1,n)
354194825f6SJed Brown  */
355194825f6SJed Brown static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m,PetscInt mstride,PetscInt n,PetscReal *A_in,PetscReal *Ainv_out,PetscScalar *tau,PetscInt worksize,PetscScalar *work)
356194825f6SJed Brown {
357194825f6SJed Brown   PetscErrorCode ierr;
358194825f6SJed Brown   PetscBLASInt M,N,K,lda,ldb,ldwork,info;
359194825f6SJed Brown   PetscScalar *A,*Ainv,*R,*Q,Alpha;
360194825f6SJed Brown 
361194825f6SJed Brown   PetscFunctionBegin;
362194825f6SJed Brown #if defined(PETSC_USE_COMPLEX)
363194825f6SJed Brown   {
364194825f6SJed Brown     PetscInt i,j;
365194825f6SJed Brown     ierr = PetscMalloc2(m*n,PetscScalar,&A,m*n,PetscScalar,&Ainv);CHKERRQ(ierr);
366194825f6SJed Brown     for (j=0; j<n; j++) {
367194825f6SJed Brown       for (i=0; i<m; i++) A[i+m*j] = A_in[i+mstride*j];
368194825f6SJed Brown     }
369194825f6SJed Brown     mstride = m;
370194825f6SJed Brown   }
371194825f6SJed Brown #else
372194825f6SJed Brown   A = A_in;
373194825f6SJed Brown   Ainv = Ainv_out;
374194825f6SJed Brown #endif
375194825f6SJed Brown 
376194825f6SJed Brown   ierr = PetscBLASIntCast(m,&M);CHKERRQ(ierr);
377194825f6SJed Brown   ierr = PetscBLASIntCast(n,&N);CHKERRQ(ierr);
378194825f6SJed Brown   ierr = PetscBLASIntCast(mstride,&lda);CHKERRQ(ierr);
379194825f6SJed Brown   ierr = PetscBLASIntCast(worksize,&ldwork);CHKERRQ(ierr);
380194825f6SJed Brown   ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
381194825f6SJed Brown   LAPACKgeqrf_(&M,&N,A,&lda,tau,work,&ldwork,&info);
382194825f6SJed Brown   ierr = PetscFPTrapPop();CHKERRQ(ierr);
383194825f6SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xGEQRF error");
384194825f6SJed Brown   R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */
385194825f6SJed Brown 
386194825f6SJed Brown   /* Extract an explicit representation of Q */
387194825f6SJed Brown   Q = Ainv;
388194825f6SJed Brown   ierr = PetscMemcpy(Q,A,mstride*n*sizeof(PetscScalar));CHKERRQ(ierr);
389194825f6SJed Brown   K = N;                        /* full rank */
390194825f6SJed Brown   LAPACKungqr_(&M,&N,&K,Q,&lda,tau,work,&ldwork,&info);
391194825f6SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xORGQR/xUNGQR error");
392194825f6SJed Brown 
393194825f6SJed Brown   /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */
394194825f6SJed Brown   Alpha = 1.0;
395194825f6SJed Brown   ldb = lda;
396194825f6SJed Brown   BLAStrsm_("Right","Upper","ConjugateTranspose","NotUnitTriangular",&M,&N,&Alpha,R,&lda,Q,&ldb);
397194825f6SJed Brown   /* Ainv is Q, overwritten with inverse */
398194825f6SJed Brown 
399194825f6SJed Brown #if defined(PETSC_USE_COMPLEX)
400194825f6SJed Brown   {
401194825f6SJed Brown     PetscInt i;
402194825f6SJed Brown     for (i=0; i<m*n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]);
403194825f6SJed Brown     ierr = PetscFree2(A,Ainv);CHKERRQ(ierr);
404194825f6SJed Brown   }
405194825f6SJed Brown #endif
406194825f6SJed Brown   PetscFunctionReturn(0);
407194825f6SJed Brown }
408194825f6SJed Brown 
409194825f6SJed Brown #undef __FUNCT__
410194825f6SJed Brown #define __FUNCT__ "PetscDTLegendreIntegrate"
411194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */
412194825f6SJed Brown static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval,const PetscReal *x,PetscInt ndegree,const PetscInt *degrees,PetscBool Transpose,PetscReal *B)
413194825f6SJed Brown {
414194825f6SJed Brown   PetscErrorCode ierr;
415194825f6SJed Brown   PetscReal *Bv;
416194825f6SJed Brown   PetscInt i,j;
417194825f6SJed Brown 
418194825f6SJed Brown   PetscFunctionBegin;
419194825f6SJed Brown   ierr = PetscMalloc((ninterval+1)*ndegree*sizeof(PetscReal),&Bv);CHKERRQ(ierr);
420194825f6SJed Brown   /* Point evaluation of L_p on all the source vertices */
421194825f6SJed Brown   ierr = PetscDTLegendreEval(ninterval+1,x,ndegree,degrees,Bv,NULL,NULL);CHKERRQ(ierr);
422194825f6SJed Brown   /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */
423194825f6SJed Brown   for (i=0; i<ninterval; i++) {
424194825f6SJed Brown     for (j=0; j<ndegree; j++) {
425194825f6SJed Brown       if (Transpose) B[i+ninterval*j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j];
426194825f6SJed Brown       else           B[i*ndegree+j]   = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j];
427194825f6SJed Brown     }
428194825f6SJed Brown   }
429194825f6SJed Brown   ierr = PetscFree(Bv);CHKERRQ(ierr);
430194825f6SJed Brown   PetscFunctionReturn(0);
431194825f6SJed Brown }
432194825f6SJed Brown 
433194825f6SJed Brown #undef __FUNCT__
434194825f6SJed Brown #define __FUNCT__ "PetscDTReconstructPoly"
435194825f6SJed Brown /*@
436194825f6SJed Brown    PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals
437194825f6SJed Brown 
438194825f6SJed Brown    Not Collective
439194825f6SJed Brown 
440194825f6SJed Brown    Input Arguments:
441194825f6SJed Brown +  degree - degree of reconstruction polynomial
442194825f6SJed Brown .  nsource - number of source intervals
443194825f6SJed Brown .  sourcex - sorted coordinates of source cell boundaries (length nsource+1)
444194825f6SJed Brown .  ntarget - number of target intervals
445194825f6SJed Brown -  targetx - sorted coordinates of target cell boundaries (length ntarget+1)
446194825f6SJed Brown 
447194825f6SJed Brown    Output Arguments:
448194825f6SJed Brown .  R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s]
449194825f6SJed Brown 
450194825f6SJed Brown    Level: advanced
451194825f6SJed Brown 
452194825f6SJed Brown .seealso: PetscDTLegendreEval()
453194825f6SJed Brown @*/
454194825f6SJed Brown PetscErrorCode PetscDTReconstructPoly(PetscInt degree,PetscInt nsource,const PetscReal *sourcex,PetscInt ntarget,const PetscReal *targetx,PetscReal *R)
455194825f6SJed Brown {
456194825f6SJed Brown   PetscErrorCode ierr;
457194825f6SJed Brown   PetscInt i,j,k,*bdegrees,worksize;
458194825f6SJed Brown   PetscReal xmin,xmax,center,hscale,*sourcey,*targety,*Bsource,*Bsinv,*Btarget;
459194825f6SJed Brown   PetscScalar *tau,*work;
460194825f6SJed Brown 
461194825f6SJed Brown   PetscFunctionBegin;
462194825f6SJed Brown   PetscValidRealPointer(sourcex,3);
463194825f6SJed Brown   PetscValidRealPointer(targetx,5);
464194825f6SJed Brown   PetscValidRealPointer(R,6);
465194825f6SJed 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);
466194825f6SJed Brown #if defined(PETSC_USE_DEBUG)
467194825f6SJed Brown   for (i=0; i<nsource; i++) {
468194825f6SJed 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]);
469194825f6SJed Brown   }
470194825f6SJed Brown   for (i=0; i<ntarget; i++) {
471194825f6SJed 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]);
472194825f6SJed Brown   }
473194825f6SJed Brown #endif
474194825f6SJed Brown   xmin = PetscMin(sourcex[0],targetx[0]);
475194825f6SJed Brown   xmax = PetscMax(sourcex[nsource],targetx[ntarget]);
476194825f6SJed Brown   center = (xmin + xmax)/2;
477194825f6SJed Brown   hscale = (xmax - xmin)/2;
478194825f6SJed Brown   worksize = nsource;
479194825f6SJed Brown   ierr = PetscMalloc4(degree+1,PetscInt,&bdegrees,nsource+1,PetscReal,&sourcey,nsource*(degree+1),PetscReal,&Bsource,worksize,PetscScalar,&work);CHKERRQ(ierr);
48082772646SJed Brown   ierr = PetscMalloc4(nsource,PetscScalar,&tau,nsource*(degree+1),PetscReal,&Bsinv,ntarget+1,PetscReal,&targety,ntarget*(degree+1),PetscReal,&Btarget);CHKERRQ(ierr);
481194825f6SJed Brown   for (i=0; i<=nsource; i++) sourcey[i] = (sourcex[i]-center)/hscale;
482194825f6SJed Brown   for (i=0; i<=degree; i++) bdegrees[i] = i+1;
483194825f6SJed Brown   ierr = PetscDTLegendreIntegrate(nsource,sourcey,degree+1,bdegrees,PETSC_TRUE,Bsource);CHKERRQ(ierr);
484194825f6SJed Brown   ierr = PetscDTPseudoInverseQR(nsource,nsource,degree+1,Bsource,Bsinv,tau,nsource,work);CHKERRQ(ierr);
485194825f6SJed Brown   for (i=0; i<=ntarget; i++) targety[i] = (targetx[i]-center)/hscale;
486194825f6SJed Brown   ierr = PetscDTLegendreIntegrate(ntarget,targety,degree+1,bdegrees,PETSC_FALSE,Btarget);CHKERRQ(ierr);
487194825f6SJed Brown   for (i=0; i<ntarget; i++) {
488194825f6SJed Brown     PetscReal rowsum = 0;
489194825f6SJed Brown     for (j=0; j<nsource; j++) {
490194825f6SJed Brown       PetscReal sum = 0;
491194825f6SJed Brown       for (k=0; k<degree+1; k++) {
492194825f6SJed Brown         sum += Btarget[i*(degree+1)+k] * Bsinv[k*nsource+j];
493194825f6SJed Brown       }
494194825f6SJed Brown       R[i*nsource+j] = sum;
495194825f6SJed Brown       rowsum += sum;
496194825f6SJed Brown     }
497194825f6SJed Brown     for (j=0; j<nsource; j++) R[i*nsource+j] /= rowsum; /* normalize each row */
498194825f6SJed Brown   }
499194825f6SJed Brown   ierr = PetscFree4(bdegrees,sourcey,Bsource,work);CHKERRQ(ierr);
500194825f6SJed Brown   ierr = PetscFree4(tau,Bsinv,targety,Btarget);CHKERRQ(ierr);
501194825f6SJed Brown   PetscFunctionReturn(0);
502194825f6SJed Brown }
503