xref: /petsc/src/dm/dt/interface/dt.c (revision dcce0ee293c059df9846add1252b299925e2c6bc)
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
729f144ccSMatthew G. Knepley #ifdef PETSC_HAVE_MPFR
829f144ccSMatthew G. Knepley #include <mpfr.h>
929f144ccSMatthew G. Knepley #endif
10a6fc04d9SSatish Balay 
110c35b76eSJed Brown #include <petscdt.h>            /*I "petscdt.h" I*/
1237045ce4SJed Brown #include <petscblaslapack.h>
13af0996ceSBarry Smith #include <petsc/private/petscimpl.h>
14af0996ceSBarry Smith #include <petsc/private/dtimpl.h>
15665c2dedSJed Brown #include <petscviewer.h>
1659804f93SMatthew G. Knepley #include <petscdmplex.h>
1759804f93SMatthew G. Knepley #include <petscdmshell.h>
1837045ce4SJed Brown 
190bfcf5a5SMatthew G. Knepley static PetscBool GaussCite       = PETSC_FALSE;
200bfcf5a5SMatthew G. Knepley const char       GaussCitation[] = "@article{GolubWelsch1969,\n"
210bfcf5a5SMatthew G. Knepley                                    "  author  = {Golub and Welsch},\n"
220bfcf5a5SMatthew G. Knepley                                    "  title   = {Calculation of Quadrature Rules},\n"
230bfcf5a5SMatthew G. Knepley                                    "  journal = {Math. Comp.},\n"
240bfcf5a5SMatthew G. Knepley                                    "  volume  = {23},\n"
250bfcf5a5SMatthew G. Knepley                                    "  number  = {106},\n"
260bfcf5a5SMatthew G. Knepley                                    "  pages   = {221--230},\n"
270bfcf5a5SMatthew G. Knepley                                    "  year    = {1969}\n}\n";
280bfcf5a5SMatthew G. Knepley 
2940d8ff71SMatthew G. Knepley /*@
3040d8ff71SMatthew G. Knepley   PetscQuadratureCreate - Create a PetscQuadrature object
3140d8ff71SMatthew G. Knepley 
3240d8ff71SMatthew G. Knepley   Collective on MPI_Comm
3340d8ff71SMatthew G. Knepley 
3440d8ff71SMatthew G. Knepley   Input Parameter:
3540d8ff71SMatthew G. Knepley . comm - The communicator for the PetscQuadrature object
3640d8ff71SMatthew G. Knepley 
3740d8ff71SMatthew G. Knepley   Output Parameter:
3840d8ff71SMatthew G. Knepley . q  - The PetscQuadrature object
3940d8ff71SMatthew G. Knepley 
4040d8ff71SMatthew G. Knepley   Level: beginner
4140d8ff71SMatthew G. Knepley 
4240d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature, create
4340d8ff71SMatthew G. Knepley .seealso: PetscQuadratureDestroy(), PetscQuadratureGetData()
4440d8ff71SMatthew G. Knepley @*/
4521454ff5SMatthew G. Knepley PetscErrorCode PetscQuadratureCreate(MPI_Comm comm, PetscQuadrature *q)
4621454ff5SMatthew G. Knepley {
4721454ff5SMatthew G. Knepley   PetscErrorCode ierr;
4821454ff5SMatthew G. Knepley 
4921454ff5SMatthew G. Knepley   PetscFunctionBegin;
5021454ff5SMatthew G. Knepley   PetscValidPointer(q, 2);
51623436dcSMatthew G. Knepley   ierr = PetscSysInitializePackage();CHKERRQ(ierr);
5273107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(*q,PETSC_OBJECT_CLASSID,"PetscQuadrature","Quadrature","DT",comm,PetscQuadratureDestroy,PetscQuadratureView);CHKERRQ(ierr);
5321454ff5SMatthew G. Knepley   (*q)->dim       = -1;
54a6b92713SMatthew G. Knepley   (*q)->Nc        =  1;
55bcede257SMatthew G. Knepley   (*q)->order     = -1;
5621454ff5SMatthew G. Knepley   (*q)->numPoints = 0;
5721454ff5SMatthew G. Knepley   (*q)->points    = NULL;
5821454ff5SMatthew G. Knepley   (*q)->weights   = NULL;
5921454ff5SMatthew G. Knepley   PetscFunctionReturn(0);
6021454ff5SMatthew G. Knepley }
6121454ff5SMatthew G. Knepley 
62c9638911SMatthew G. Knepley /*@
63c9638911SMatthew G. Knepley   PetscQuadratureDuplicate - Create a deep copy of the PetscQuadrature object
64c9638911SMatthew G. Knepley 
65c9638911SMatthew G. Knepley   Collective on PetscQuadrature
66c9638911SMatthew G. Knepley 
67c9638911SMatthew G. Knepley   Input Parameter:
68c9638911SMatthew G. Knepley . q  - The PetscQuadrature object
69c9638911SMatthew G. Knepley 
70c9638911SMatthew G. Knepley   Output Parameter:
71c9638911SMatthew G. Knepley . r  - The new PetscQuadrature object
72c9638911SMatthew G. Knepley 
73c9638911SMatthew G. Knepley   Level: beginner
74c9638911SMatthew G. Knepley 
75c9638911SMatthew G. Knepley .keywords: PetscQuadrature, quadrature, clone
76c9638911SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureDestroy(), PetscQuadratureGetData()
77c9638911SMatthew G. Knepley @*/
78c9638911SMatthew G. Knepley PetscErrorCode PetscQuadratureDuplicate(PetscQuadrature q, PetscQuadrature *r)
79c9638911SMatthew G. Knepley {
80a6b92713SMatthew G. Knepley   PetscInt         order, dim, Nc, Nq;
81c9638911SMatthew G. Knepley   const PetscReal *points, *weights;
82c9638911SMatthew G. Knepley   PetscReal       *p, *w;
83c9638911SMatthew G. Knepley   PetscErrorCode   ierr;
84c9638911SMatthew G. Knepley 
85c9638911SMatthew G. Knepley   PetscFunctionBegin;
86c9638911SMatthew G. Knepley   PetscValidPointer(q, 2);
87c9638911SMatthew G. Knepley   ierr = PetscQuadratureCreate(PetscObjectComm((PetscObject) q), r);CHKERRQ(ierr);
88c9638911SMatthew G. Knepley   ierr = PetscQuadratureGetOrder(q, &order);CHKERRQ(ierr);
89c9638911SMatthew G. Knepley   ierr = PetscQuadratureSetOrder(*r, order);CHKERRQ(ierr);
90a6b92713SMatthew G. Knepley   ierr = PetscQuadratureGetData(q, &dim, &Nc, &Nq, &points, &weights);CHKERRQ(ierr);
91c9638911SMatthew G. Knepley   ierr = PetscMalloc1(Nq*dim, &p);CHKERRQ(ierr);
92c9638911SMatthew G. Knepley   ierr = PetscMalloc1(Nq, &w);CHKERRQ(ierr);
93c9638911SMatthew G. Knepley   ierr = PetscMemcpy(p, points, Nq*dim * sizeof(PetscReal));CHKERRQ(ierr);
94a6b92713SMatthew G. Knepley   ierr = PetscMemcpy(w, weights, Nc * Nq * sizeof(PetscReal));CHKERRQ(ierr);
95a6b92713SMatthew G. Knepley   ierr = PetscQuadratureSetData(*r, dim, Nc, Nq, p, w);CHKERRQ(ierr);
96c9638911SMatthew G. Knepley   PetscFunctionReturn(0);
97c9638911SMatthew G. Knepley }
98c9638911SMatthew G. Knepley 
9940d8ff71SMatthew G. Knepley /*@
10040d8ff71SMatthew G. Knepley   PetscQuadratureDestroy - Destroys a PetscQuadrature object
10140d8ff71SMatthew G. Knepley 
10240d8ff71SMatthew G. Knepley   Collective on PetscQuadrature
10340d8ff71SMatthew G. Knepley 
10440d8ff71SMatthew G. Knepley   Input Parameter:
10540d8ff71SMatthew G. Knepley . q  - The PetscQuadrature object
10640d8ff71SMatthew G. Knepley 
10740d8ff71SMatthew G. Knepley   Level: beginner
10840d8ff71SMatthew G. Knepley 
10940d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature, destroy
11040d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureGetData()
11140d8ff71SMatthew G. Knepley @*/
112bfa639d9SMatthew G. Knepley PetscErrorCode PetscQuadratureDestroy(PetscQuadrature *q)
113bfa639d9SMatthew G. Knepley {
114bfa639d9SMatthew G. Knepley   PetscErrorCode ierr;
115bfa639d9SMatthew G. Knepley 
116bfa639d9SMatthew G. Knepley   PetscFunctionBegin;
11721454ff5SMatthew G. Knepley   if (!*q) PetscFunctionReturn(0);
11821454ff5SMatthew G. Knepley   PetscValidHeaderSpecific((*q),PETSC_OBJECT_CLASSID,1);
11921454ff5SMatthew G. Knepley   if (--((PetscObject)(*q))->refct > 0) {
12021454ff5SMatthew G. Knepley     *q = NULL;
12121454ff5SMatthew G. Knepley     PetscFunctionReturn(0);
12221454ff5SMatthew G. Knepley   }
12321454ff5SMatthew G. Knepley   ierr = PetscFree((*q)->points);CHKERRQ(ierr);
12421454ff5SMatthew G. Knepley   ierr = PetscFree((*q)->weights);CHKERRQ(ierr);
12521454ff5SMatthew G. Knepley   ierr = PetscHeaderDestroy(q);CHKERRQ(ierr);
12621454ff5SMatthew G. Knepley   PetscFunctionReturn(0);
12721454ff5SMatthew G. Knepley }
12821454ff5SMatthew G. Knepley 
129bcede257SMatthew G. Knepley /*@
130a6b92713SMatthew G. Knepley   PetscQuadratureGetOrder - Return the order of the method
131bcede257SMatthew G. Knepley 
132bcede257SMatthew G. Knepley   Not collective
133bcede257SMatthew G. Knepley 
134bcede257SMatthew G. Knepley   Input Parameter:
135bcede257SMatthew G. Knepley . q - The PetscQuadrature object
136bcede257SMatthew G. Knepley 
137bcede257SMatthew G. Knepley   Output Parameter:
138bcede257SMatthew G. Knepley . order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated
139bcede257SMatthew G. Knepley 
140bcede257SMatthew G. Knepley   Level: intermediate
141bcede257SMatthew G. Knepley 
142bcede257SMatthew G. Knepley .seealso: PetscQuadratureSetOrder(), PetscQuadratureGetData(), PetscQuadratureSetData()
143bcede257SMatthew G. Knepley @*/
144bcede257SMatthew G. Knepley PetscErrorCode PetscQuadratureGetOrder(PetscQuadrature q, PetscInt *order)
145bcede257SMatthew G. Knepley {
146bcede257SMatthew G. Knepley   PetscFunctionBegin;
147bcede257SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
148bcede257SMatthew G. Knepley   PetscValidPointer(order, 2);
149bcede257SMatthew G. Knepley   *order = q->order;
150bcede257SMatthew G. Knepley   PetscFunctionReturn(0);
151bcede257SMatthew G. Knepley }
152bcede257SMatthew G. Knepley 
153bcede257SMatthew G. Knepley /*@
154a6b92713SMatthew G. Knepley   PetscQuadratureSetOrder - Return the order of the method
155bcede257SMatthew G. Knepley 
156bcede257SMatthew G. Knepley   Not collective
157bcede257SMatthew G. Knepley 
158bcede257SMatthew G. Knepley   Input Parameters:
159bcede257SMatthew G. Knepley + q - The PetscQuadrature object
160bcede257SMatthew G. Knepley - order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated
161bcede257SMatthew G. Knepley 
162bcede257SMatthew G. Knepley   Level: intermediate
163bcede257SMatthew G. Knepley 
164bcede257SMatthew G. Knepley .seealso: PetscQuadratureGetOrder(), PetscQuadratureGetData(), PetscQuadratureSetData()
165bcede257SMatthew G. Knepley @*/
166bcede257SMatthew G. Knepley PetscErrorCode PetscQuadratureSetOrder(PetscQuadrature q, PetscInt order)
167bcede257SMatthew G. Knepley {
168bcede257SMatthew G. Knepley   PetscFunctionBegin;
169bcede257SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
170bcede257SMatthew G. Knepley   q->order = order;
171bcede257SMatthew G. Knepley   PetscFunctionReturn(0);
172bcede257SMatthew G. Knepley }
173bcede257SMatthew G. Knepley 
174a6b92713SMatthew G. Knepley /*@
175a6b92713SMatthew G. Knepley   PetscQuadratureGetNumComponents - Return the number of components for functions to be integrated
176a6b92713SMatthew G. Knepley 
177a6b92713SMatthew G. Knepley   Not collective
178a6b92713SMatthew G. Knepley 
179a6b92713SMatthew G. Knepley   Input Parameter:
180a6b92713SMatthew G. Knepley . q - The PetscQuadrature object
181a6b92713SMatthew G. Knepley 
182a6b92713SMatthew G. Knepley   Output Parameter:
183a6b92713SMatthew G. Knepley . Nc - The number of components
184a6b92713SMatthew G. Knepley 
185a6b92713SMatthew G. Knepley   Note: We are performing an integral int f(x) . w(x) dx, where both f and w (the weight) have Nc components.
186a6b92713SMatthew G. Knepley 
187a6b92713SMatthew G. Knepley   Level: intermediate
188a6b92713SMatthew G. Knepley 
189a6b92713SMatthew G. Knepley .seealso: PetscQuadratureSetNumComponents(), PetscQuadratureGetData(), PetscQuadratureSetData()
190a6b92713SMatthew G. Knepley @*/
191a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureGetNumComponents(PetscQuadrature q, PetscInt *Nc)
192a6b92713SMatthew G. Knepley {
193a6b92713SMatthew G. Knepley   PetscFunctionBegin;
194a6b92713SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
195a6b92713SMatthew G. Knepley   PetscValidPointer(Nc, 2);
196a6b92713SMatthew G. Knepley   *Nc = q->Nc;
197a6b92713SMatthew G. Knepley   PetscFunctionReturn(0);
198a6b92713SMatthew G. Knepley }
199a6b92713SMatthew G. Knepley 
200a6b92713SMatthew G. Knepley /*@
201a6b92713SMatthew G. Knepley   PetscQuadratureSetNumComponents - Return the number of components for functions to be integrated
202a6b92713SMatthew G. Knepley 
203a6b92713SMatthew G. Knepley   Not collective
204a6b92713SMatthew G. Knepley 
205a6b92713SMatthew G. Knepley   Input Parameters:
206a6b92713SMatthew G. Knepley + q  - The PetscQuadrature object
207a6b92713SMatthew G. Knepley - Nc - The number of components
208a6b92713SMatthew G. Knepley 
209a6b92713SMatthew G. Knepley   Note: We are performing an integral int f(x) . w(x) dx, where both f and w (the weight) have Nc components.
210a6b92713SMatthew G. Knepley 
211a6b92713SMatthew G. Knepley   Level: intermediate
212a6b92713SMatthew G. Knepley 
213a6b92713SMatthew G. Knepley .seealso: PetscQuadratureGetNumComponents(), PetscQuadratureGetData(), PetscQuadratureSetData()
214a6b92713SMatthew G. Knepley @*/
215a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureSetNumComponents(PetscQuadrature q, PetscInt Nc)
216a6b92713SMatthew G. Knepley {
217a6b92713SMatthew G. Knepley   PetscFunctionBegin;
218a6b92713SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
219a6b92713SMatthew G. Knepley   q->Nc = Nc;
220a6b92713SMatthew G. Knepley   PetscFunctionReturn(0);
221a6b92713SMatthew G. Knepley }
222a6b92713SMatthew G. Knepley 
22340d8ff71SMatthew G. Knepley /*@C
22440d8ff71SMatthew G. Knepley   PetscQuadratureGetData - Returns the data defining the quadrature
22540d8ff71SMatthew G. Knepley 
22640d8ff71SMatthew G. Knepley   Not collective
22740d8ff71SMatthew G. Knepley 
22840d8ff71SMatthew G. Knepley   Input Parameter:
22940d8ff71SMatthew G. Knepley . q  - The PetscQuadrature object
23040d8ff71SMatthew G. Knepley 
23140d8ff71SMatthew G. Knepley   Output Parameters:
23240d8ff71SMatthew G. Knepley + dim - The spatial dimension
233a6b92713SMatthew G. Knepley , Nc - The number of components
23440d8ff71SMatthew G. Knepley . npoints - The number of quadrature points
23540d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point
23640d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point
23740d8ff71SMatthew G. Knepley 
23840d8ff71SMatthew G. Knepley   Level: intermediate
23940d8ff71SMatthew G. Knepley 
24040d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature
24140d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureSetData()
24240d8ff71SMatthew G. Knepley @*/
243a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureGetData(PetscQuadrature q, PetscInt *dim, PetscInt *Nc, PetscInt *npoints, const PetscReal *points[], const PetscReal *weights[])
24421454ff5SMatthew G. Knepley {
24521454ff5SMatthew G. Knepley   PetscFunctionBegin;
24621454ff5SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
24721454ff5SMatthew G. Knepley   if (dim) {
24821454ff5SMatthew G. Knepley     PetscValidPointer(dim, 2);
24921454ff5SMatthew G. Knepley     *dim = q->dim;
25021454ff5SMatthew G. Knepley   }
251a6b92713SMatthew G. Knepley   if (Nc) {
252a6b92713SMatthew G. Knepley     PetscValidPointer(Nc, 3);
253a6b92713SMatthew G. Knepley     *Nc = q->Nc;
254a6b92713SMatthew G. Knepley   }
25521454ff5SMatthew G. Knepley   if (npoints) {
256a6b92713SMatthew G. Knepley     PetscValidPointer(npoints, 4);
25721454ff5SMatthew G. Knepley     *npoints = q->numPoints;
25821454ff5SMatthew G. Knepley   }
25921454ff5SMatthew G. Knepley   if (points) {
260a6b92713SMatthew G. Knepley     PetscValidPointer(points, 5);
26121454ff5SMatthew G. Knepley     *points = q->points;
26221454ff5SMatthew G. Knepley   }
26321454ff5SMatthew G. Knepley   if (weights) {
264a6b92713SMatthew G. Knepley     PetscValidPointer(weights, 6);
26521454ff5SMatthew G. Knepley     *weights = q->weights;
26621454ff5SMatthew G. Knepley   }
26721454ff5SMatthew G. Knepley   PetscFunctionReturn(0);
26821454ff5SMatthew G. Knepley }
26921454ff5SMatthew G. Knepley 
27040d8ff71SMatthew G. Knepley /*@C
27140d8ff71SMatthew G. Knepley   PetscQuadratureSetData - Sets the data defining the quadrature
27240d8ff71SMatthew G. Knepley 
27340d8ff71SMatthew G. Knepley   Not collective
27440d8ff71SMatthew G. Knepley 
27540d8ff71SMatthew G. Knepley   Input Parameters:
27640d8ff71SMatthew G. Knepley + q  - The PetscQuadrature object
27740d8ff71SMatthew G. Knepley . dim - The spatial dimension
278a6b92713SMatthew G. Knepley , Nc - The number of components
27940d8ff71SMatthew G. Knepley . npoints - The number of quadrature points
28040d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point
28140d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point
28240d8ff71SMatthew G. Knepley 
28340d8ff71SMatthew G. Knepley   Level: intermediate
28440d8ff71SMatthew G. Knepley 
28540d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature
28640d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureGetData()
28740d8ff71SMatthew G. Knepley @*/
288a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureSetData(PetscQuadrature q, PetscInt dim, PetscInt Nc, PetscInt npoints, const PetscReal points[], const PetscReal weights[])
28921454ff5SMatthew G. Knepley {
29021454ff5SMatthew G. Knepley   PetscFunctionBegin;
29121454ff5SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
29221454ff5SMatthew G. Knepley   if (dim >= 0)     q->dim       = dim;
293a6b92713SMatthew G. Knepley   if (Nc >= 0)      q->Nc        = Nc;
29421454ff5SMatthew G. Knepley   if (npoints >= 0) q->numPoints = npoints;
29521454ff5SMatthew G. Knepley   if (points) {
29621454ff5SMatthew G. Knepley     PetscValidPointer(points, 4);
29721454ff5SMatthew G. Knepley     q->points = points;
29821454ff5SMatthew G. Knepley   }
29921454ff5SMatthew G. Knepley   if (weights) {
30021454ff5SMatthew G. Knepley     PetscValidPointer(weights, 5);
30121454ff5SMatthew G. Knepley     q->weights = weights;
30221454ff5SMatthew G. Knepley   }
303f9fd7fdbSMatthew G. Knepley   PetscFunctionReturn(0);
304f9fd7fdbSMatthew G. Knepley }
305f9fd7fdbSMatthew G. Knepley 
30640d8ff71SMatthew G. Knepley /*@C
30740d8ff71SMatthew G. Knepley   PetscQuadratureView - Views a PetscQuadrature object
30840d8ff71SMatthew G. Knepley 
30940d8ff71SMatthew G. Knepley   Collective on PetscQuadrature
31040d8ff71SMatthew G. Knepley 
31140d8ff71SMatthew G. Knepley   Input Parameters:
31240d8ff71SMatthew G. Knepley + q  - The PetscQuadrature object
31340d8ff71SMatthew G. Knepley - viewer - The PetscViewer object
31440d8ff71SMatthew G. Knepley 
31540d8ff71SMatthew G. Knepley   Level: beginner
31640d8ff71SMatthew G. Knepley 
31740d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature, view
31840d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureGetData()
31940d8ff71SMatthew G. Knepley @*/
320f9fd7fdbSMatthew G. Knepley PetscErrorCode PetscQuadratureView(PetscQuadrature quad, PetscViewer viewer)
321f9fd7fdbSMatthew G. Knepley {
322a6b92713SMatthew G. Knepley   PetscInt       q, d, c;
323f9fd7fdbSMatthew G. Knepley   PetscErrorCode ierr;
324f9fd7fdbSMatthew G. Knepley 
325f9fd7fdbSMatthew G. Knepley   PetscFunctionBegin;
32698c3331eSBarry Smith   ierr = PetscObjectPrintClassNamePrefixType((PetscObject)quad,viewer);CHKERRQ(ierr);
327a6b92713SMatthew G. Knepley   if (quad->Nc > 1) {ierr = PetscViewerASCIIPrintf(viewer, "Quadrature on %D points with %D components\n  (", quad->numPoints, quad->Nc);CHKERRQ(ierr);}
328a6b92713SMatthew G. Knepley   else              {ierr = PetscViewerASCIIPrintf(viewer, "Quadrature on %D points\n  (", quad->numPoints);CHKERRQ(ierr);}
32921454ff5SMatthew G. Knepley   for (q = 0; q < quad->numPoints; ++q) {
33021454ff5SMatthew G. Knepley     for (d = 0; d < quad->dim; ++d) {
331f9fd7fdbSMatthew G. Knepley       if (d) ierr = PetscViewerASCIIPrintf(viewer, ", ");CHKERRQ(ierr);
332ab15ae43SMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, "%g\n", (double)quad->points[q*quad->dim+d]);CHKERRQ(ierr);
333f9fd7fdbSMatthew G. Knepley     }
334a6b92713SMatthew G. Knepley     if (quad->Nc > 1) {
335a6b92713SMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, ") (");CHKERRQ(ierr);
336a6b92713SMatthew G. Knepley       for (c = 0; c < quad->Nc; ++c) {
337a6b92713SMatthew G. Knepley         if (c) ierr = PetscViewerASCIIPrintf(viewer, ", ");CHKERRQ(ierr);
338a6b92713SMatthew G. Knepley         ierr = PetscViewerASCIIPrintf(viewer, "%g", (double)quad->weights[q*quad->Nc+c]);CHKERRQ(ierr);
339a6b92713SMatthew G. Knepley       }
340a6b92713SMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, ")\n");CHKERRQ(ierr);
341a6b92713SMatthew G. Knepley     } else {
342ab15ae43SMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, ") %g\n", (double)quad->weights[q]);CHKERRQ(ierr);
343f9fd7fdbSMatthew G. Knepley     }
344a6b92713SMatthew G. Knepley   }
345bfa639d9SMatthew G. Knepley   PetscFunctionReturn(0);
346bfa639d9SMatthew G. Knepley }
347bfa639d9SMatthew G. Knepley 
34889710940SMatthew G. Knepley /*@C
34989710940SMatthew G. Knepley   PetscQuadratureExpandComposite - Return a quadrature over the composite element, which has the original quadrature in each subelement
35089710940SMatthew G. Knepley 
35189710940SMatthew G. Knepley   Not collective
35289710940SMatthew G. Knepley 
35389710940SMatthew G. Knepley   Input Parameter:
35489710940SMatthew G. Knepley + q - The original PetscQuadrature
35589710940SMatthew G. Knepley . numSubelements - The number of subelements the original element is divided into
35689710940SMatthew G. Knepley . v0 - An array of the initial points for each subelement
35789710940SMatthew G. Knepley - jac - An array of the Jacobian mappings from the reference to each subelement
35889710940SMatthew G. Knepley 
35989710940SMatthew G. Knepley   Output Parameters:
36089710940SMatthew G. Knepley . dim - The dimension
36189710940SMatthew G. Knepley 
36289710940SMatthew G. Knepley   Note: Together v0 and jac define an affine mapping from the original reference element to each subelement
36389710940SMatthew G. Knepley 
36489710940SMatthew G. Knepley   Level: intermediate
36589710940SMatthew G. Knepley 
36689710940SMatthew G. Knepley .seealso: PetscFECreate(), PetscSpaceGetDimension(), PetscDualSpaceGetDimension()
36789710940SMatthew G. Knepley @*/
36889710940SMatthew G. Knepley PetscErrorCode PetscQuadratureExpandComposite(PetscQuadrature q, PetscInt numSubelements, const PetscReal v0[], const PetscReal jac[], PetscQuadrature *qref)
36989710940SMatthew G. Knepley {
37089710940SMatthew G. Knepley   const PetscReal *points,    *weights;
37189710940SMatthew G. Knepley   PetscReal       *pointsRef, *weightsRef;
372a6b92713SMatthew G. Knepley   PetscInt         dim, Nc, order, npoints, npointsRef, c, p, cp, d, e;
37389710940SMatthew G. Knepley   PetscErrorCode   ierr;
37489710940SMatthew G. Knepley 
37589710940SMatthew G. Knepley   PetscFunctionBegin;
37689710940SMatthew G. Knepley   PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1);
37789710940SMatthew G. Knepley   PetscValidPointer(v0, 3);
37889710940SMatthew G. Knepley   PetscValidPointer(jac, 4);
37989710940SMatthew G. Knepley   PetscValidPointer(qref, 5);
38089710940SMatthew G. Knepley   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, qref);CHKERRQ(ierr);
38189710940SMatthew G. Knepley   ierr = PetscQuadratureGetOrder(q, &order);CHKERRQ(ierr);
382a6b92713SMatthew G. Knepley   ierr = PetscQuadratureGetData(q, &dim, &Nc, &npoints, &points, &weights);CHKERRQ(ierr);
38389710940SMatthew G. Knepley   npointsRef = npoints*numSubelements;
38489710940SMatthew G. Knepley   ierr = PetscMalloc1(npointsRef*dim,&pointsRef);CHKERRQ(ierr);
385a6b92713SMatthew G. Knepley   ierr = PetscMalloc1(npointsRef*Nc, &weightsRef);CHKERRQ(ierr);
38689710940SMatthew G. Knepley   for (c = 0; c < numSubelements; ++c) {
38789710940SMatthew G. Knepley     for (p = 0; p < npoints; ++p) {
38889710940SMatthew G. Knepley       for (d = 0; d < dim; ++d) {
38989710940SMatthew G. Knepley         pointsRef[(c*npoints + p)*dim+d] = v0[c*dim+d];
39089710940SMatthew G. Knepley         for (e = 0; e < dim; ++e) {
39189710940SMatthew G. Knepley           pointsRef[(c*npoints + p)*dim+d] += jac[(c*dim + d)*dim+e]*(points[p*dim+e] + 1.0);
39289710940SMatthew G. Knepley         }
39389710940SMatthew G. Knepley       }
39489710940SMatthew G. Knepley       /* Could also use detJ here */
395a6b92713SMatthew G. Knepley       for (cp = 0; cp < Nc; ++cp) weightsRef[(c*npoints+p)*Nc+cp] = weights[p*Nc+cp]/numSubelements;
39689710940SMatthew G. Knepley     }
39789710940SMatthew G. Knepley   }
39889710940SMatthew G. Knepley   ierr = PetscQuadratureSetOrder(*qref, order);CHKERRQ(ierr);
399a6b92713SMatthew G. Knepley   ierr = PetscQuadratureSetData(*qref, dim, Nc, npointsRef, pointsRef, weightsRef);CHKERRQ(ierr);
40089710940SMatthew G. Knepley   PetscFunctionReturn(0);
40189710940SMatthew G. Knepley }
40289710940SMatthew G. Knepley 
40337045ce4SJed Brown /*@
40437045ce4SJed Brown    PetscDTLegendreEval - evaluate Legendre polynomial at points
40537045ce4SJed Brown 
40637045ce4SJed Brown    Not Collective
40737045ce4SJed Brown 
40837045ce4SJed Brown    Input Arguments:
40937045ce4SJed Brown +  npoints - number of spatial points to evaluate at
41037045ce4SJed Brown .  points - array of locations to evaluate at
41137045ce4SJed Brown .  ndegree - number of basis degrees to evaluate
41237045ce4SJed Brown -  degrees - sorted array of degrees to evaluate
41337045ce4SJed Brown 
41437045ce4SJed Brown    Output Arguments:
4150298fd71SBarry Smith +  B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or NULL)
4160298fd71SBarry Smith .  D - row-oriented derivative evaluation matrix (or NULL)
4170298fd71SBarry Smith -  D2 - row-oriented second derivative evaluation matrix (or NULL)
41837045ce4SJed Brown 
41937045ce4SJed Brown    Level: intermediate
42037045ce4SJed Brown 
42137045ce4SJed Brown .seealso: PetscDTGaussQuadrature()
42237045ce4SJed Brown @*/
42337045ce4SJed Brown PetscErrorCode PetscDTLegendreEval(PetscInt npoints,const PetscReal *points,PetscInt ndegree,const PetscInt *degrees,PetscReal *B,PetscReal *D,PetscReal *D2)
42437045ce4SJed Brown {
42537045ce4SJed Brown   PetscInt i,maxdegree;
42637045ce4SJed Brown 
42737045ce4SJed Brown   PetscFunctionBegin;
42837045ce4SJed Brown   if (!npoints || !ndegree) PetscFunctionReturn(0);
42937045ce4SJed Brown   maxdegree = degrees[ndegree-1];
43037045ce4SJed Brown   for (i=0; i<npoints; i++) {
43137045ce4SJed Brown     PetscReal pm1,pm2,pd1,pd2,pdd1,pdd2,x;
43237045ce4SJed Brown     PetscInt  j,k;
43337045ce4SJed Brown     x    = points[i];
43437045ce4SJed Brown     pm2  = 0;
43537045ce4SJed Brown     pm1  = 1;
43637045ce4SJed Brown     pd2  = 0;
43737045ce4SJed Brown     pd1  = 0;
43837045ce4SJed Brown     pdd2 = 0;
43937045ce4SJed Brown     pdd1 = 0;
44037045ce4SJed Brown     k    = 0;
44137045ce4SJed Brown     if (degrees[k] == 0) {
44237045ce4SJed Brown       if (B) B[i*ndegree+k] = pm1;
44337045ce4SJed Brown       if (D) D[i*ndegree+k] = pd1;
44437045ce4SJed Brown       if (D2) D2[i*ndegree+k] = pdd1;
44537045ce4SJed Brown       k++;
44637045ce4SJed Brown     }
44737045ce4SJed Brown     for (j=1; j<=maxdegree; j++,k++) {
44837045ce4SJed Brown       PetscReal p,d,dd;
44937045ce4SJed Brown       p    = ((2*j-1)*x*pm1 - (j-1)*pm2)/j;
45037045ce4SJed Brown       d    = pd2 + (2*j-1)*pm1;
45137045ce4SJed Brown       dd   = pdd2 + (2*j-1)*pd1;
45237045ce4SJed Brown       pm2  = pm1;
45337045ce4SJed Brown       pm1  = p;
45437045ce4SJed Brown       pd2  = pd1;
45537045ce4SJed Brown       pd1  = d;
45637045ce4SJed Brown       pdd2 = pdd1;
45737045ce4SJed Brown       pdd1 = dd;
45837045ce4SJed Brown       if (degrees[k] == j) {
45937045ce4SJed Brown         if (B) B[i*ndegree+k] = p;
46037045ce4SJed Brown         if (D) D[i*ndegree+k] = d;
46137045ce4SJed Brown         if (D2) D2[i*ndegree+k] = dd;
46237045ce4SJed Brown       }
46337045ce4SJed Brown     }
46437045ce4SJed Brown   }
46537045ce4SJed Brown   PetscFunctionReturn(0);
46637045ce4SJed Brown }
46737045ce4SJed Brown 
46837045ce4SJed Brown /*@
46937045ce4SJed Brown    PetscDTGaussQuadrature - create Gauss quadrature
47037045ce4SJed Brown 
47137045ce4SJed Brown    Not Collective
47237045ce4SJed Brown 
47337045ce4SJed Brown    Input Arguments:
47437045ce4SJed Brown +  npoints - number of points
47537045ce4SJed Brown .  a - left end of interval (often-1)
47637045ce4SJed Brown -  b - right end of interval (often +1)
47737045ce4SJed Brown 
47837045ce4SJed Brown    Output Arguments:
47937045ce4SJed Brown +  x - quadrature points
48037045ce4SJed Brown -  w - quadrature weights
48137045ce4SJed Brown 
48237045ce4SJed Brown    Level: intermediate
48337045ce4SJed Brown 
48437045ce4SJed Brown    References:
48596a0c994SBarry Smith .   1. - Golub and Welsch, Calculation of Quadrature Rules, Math. Comp. 23(106), 1969.
48637045ce4SJed Brown 
48737045ce4SJed Brown .seealso: PetscDTLegendreEval()
48837045ce4SJed Brown @*/
48937045ce4SJed Brown PetscErrorCode PetscDTGaussQuadrature(PetscInt npoints,PetscReal a,PetscReal b,PetscReal *x,PetscReal *w)
49037045ce4SJed Brown {
49137045ce4SJed Brown   PetscErrorCode ierr;
49237045ce4SJed Brown   PetscInt       i;
49337045ce4SJed Brown   PetscReal      *work;
49437045ce4SJed Brown   PetscScalar    *Z;
49537045ce4SJed Brown   PetscBLASInt   N,LDZ,info;
49637045ce4SJed Brown 
49737045ce4SJed Brown   PetscFunctionBegin;
4980bfcf5a5SMatthew G. Knepley   ierr = PetscCitationsRegister(GaussCitation, &GaussCite);CHKERRQ(ierr);
49937045ce4SJed Brown   /* Set up the Golub-Welsch system */
50037045ce4SJed Brown   for (i=0; i<npoints; i++) {
50137045ce4SJed Brown     x[i] = 0;                   /* diagonal is 0 */
50237045ce4SJed Brown     if (i) w[i-1] = 0.5 / PetscSqrtReal(1 - 1./PetscSqr(2*i));
50337045ce4SJed Brown   }
504dcca6d9dSJed Brown   ierr = PetscMalloc2(npoints*npoints,&Z,PetscMax(1,2*npoints-2),&work);CHKERRQ(ierr);
505c5df96a5SBarry Smith   ierr = PetscBLASIntCast(npoints,&N);CHKERRQ(ierr);
50637045ce4SJed Brown   LDZ  = N;
50737045ce4SJed Brown   ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
5088b83055fSJed Brown   PetscStackCallBLAS("LAPACKsteqr",LAPACKsteqr_("I",&N,x,w,Z,&LDZ,work,&info));
50937045ce4SJed Brown   ierr = PetscFPTrapPop();CHKERRQ(ierr);
5101c3d6f74SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"xSTEQR error");
51137045ce4SJed Brown 
51237045ce4SJed Brown   for (i=0; i<(npoints+1)/2; i++) {
51337045ce4SJed Brown     PetscReal y = 0.5 * (-x[i] + x[npoints-i-1]); /* enforces symmetry */
51437045ce4SJed Brown     x[i]           = (a+b)/2 - y*(b-a)/2;
51519a57d60SBarry Smith     if (x[i] == -0.0) x[i] = 0.0;
51637045ce4SJed Brown     x[npoints-i-1] = (a+b)/2 + y*(b-a)/2;
5170d644c17SKarl Rupp 
51888393a60SJed Brown     w[i] = w[npoints-1-i] = 0.5*(b-a)*(PetscSqr(PetscAbsScalar(Z[i*npoints])) + PetscSqr(PetscAbsScalar(Z[(npoints-i-1)*npoints])));
51937045ce4SJed Brown   }
52037045ce4SJed Brown   ierr = PetscFree2(Z,work);CHKERRQ(ierr);
52137045ce4SJed Brown   PetscFunctionReturn(0);
52237045ce4SJed Brown }
523194825f6SJed Brown 
524744bafbcSMatthew G. Knepley /*@
525744bafbcSMatthew G. Knepley   PetscDTGaussTensorQuadrature - creates a tensor-product Gauss quadrature
526744bafbcSMatthew G. Knepley 
527744bafbcSMatthew G. Knepley   Not Collective
528744bafbcSMatthew G. Knepley 
529744bafbcSMatthew G. Knepley   Input Arguments:
530744bafbcSMatthew G. Knepley + dim     - The spatial dimension
531a6b92713SMatthew G. Knepley . Nc      - The number of components
532744bafbcSMatthew G. Knepley . npoints - number of points in one dimension
533744bafbcSMatthew G. Knepley . a       - left end of interval (often-1)
534744bafbcSMatthew G. Knepley - b       - right end of interval (often +1)
535744bafbcSMatthew G. Knepley 
536744bafbcSMatthew G. Knepley   Output Argument:
537744bafbcSMatthew G. Knepley . q - A PetscQuadrature object
538744bafbcSMatthew G. Knepley 
539744bafbcSMatthew G. Knepley   Level: intermediate
540744bafbcSMatthew G. Knepley 
541744bafbcSMatthew G. Knepley .seealso: PetscDTGaussQuadrature(), PetscDTLegendreEval()
542744bafbcSMatthew G. Knepley @*/
543a6b92713SMatthew G. Knepley PetscErrorCode PetscDTGaussTensorQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q)
544744bafbcSMatthew G. Knepley {
545a6b92713SMatthew G. Knepley   PetscInt       totpoints = dim > 1 ? dim > 2 ? npoints*PetscSqr(npoints) : PetscSqr(npoints) : npoints, i, j, k, c;
546744bafbcSMatthew G. Knepley   PetscReal     *x, *w, *xw, *ww;
547744bafbcSMatthew G. Knepley   PetscErrorCode ierr;
548744bafbcSMatthew G. Knepley 
549744bafbcSMatthew G. Knepley   PetscFunctionBegin;
550744bafbcSMatthew G. Knepley   ierr = PetscMalloc1(totpoints*dim,&x);CHKERRQ(ierr);
551a6b92713SMatthew G. Knepley   ierr = PetscMalloc1(totpoints*Nc,&w);CHKERRQ(ierr);
552744bafbcSMatthew G. Knepley   /* Set up the Golub-Welsch system */
553744bafbcSMatthew G. Knepley   switch (dim) {
554744bafbcSMatthew G. Knepley   case 0:
555744bafbcSMatthew G. Knepley     ierr = PetscFree(x);CHKERRQ(ierr);
556744bafbcSMatthew G. Knepley     ierr = PetscFree(w);CHKERRQ(ierr);
557744bafbcSMatthew G. Knepley     ierr = PetscMalloc1(1, &x);CHKERRQ(ierr);
558a6b92713SMatthew G. Knepley     ierr = PetscMalloc1(Nc, &w);CHKERRQ(ierr);
559744bafbcSMatthew G. Knepley     x[0] = 0.0;
560a6b92713SMatthew G. Knepley     for (c = 0; c < Nc; ++c) w[c] = 1.0;
561744bafbcSMatthew G. Knepley     break;
562744bafbcSMatthew G. Knepley   case 1:
563a6b92713SMatthew G. Knepley     ierr = PetscMalloc1(npoints,&ww);CHKERRQ(ierr);
564a6b92713SMatthew G. Knepley     ierr = PetscDTGaussQuadrature(npoints, a, b, x, ww);CHKERRQ(ierr);
565a6b92713SMatthew G. Knepley     for (i = 0; i < npoints; ++i) for (c = 0; c < Nc; ++c) w[i*Nc+c] = ww[i];
566a6b92713SMatthew G. Knepley     ierr = PetscFree(ww);CHKERRQ(ierr);
567744bafbcSMatthew G. Knepley     break;
568744bafbcSMatthew G. Knepley   case 2:
569744bafbcSMatthew G. Knepley     ierr = PetscMalloc2(npoints,&xw,npoints,&ww);CHKERRQ(ierr);
570744bafbcSMatthew G. Knepley     ierr = PetscDTGaussQuadrature(npoints, a, b, xw, ww);CHKERRQ(ierr);
571744bafbcSMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
572744bafbcSMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
573744bafbcSMatthew G. Knepley         x[(i*npoints+j)*dim+0] = xw[i];
574744bafbcSMatthew G. Knepley         x[(i*npoints+j)*dim+1] = xw[j];
575a6b92713SMatthew G. Knepley         for (c = 0; c < Nc; ++c) w[(i*npoints+j)*Nc+c] = ww[i] * ww[j];
576744bafbcSMatthew G. Knepley       }
577744bafbcSMatthew G. Knepley     }
578744bafbcSMatthew G. Knepley     ierr = PetscFree2(xw,ww);CHKERRQ(ierr);
579744bafbcSMatthew G. Knepley     break;
580744bafbcSMatthew G. Knepley   case 3:
581744bafbcSMatthew G. Knepley     ierr = PetscMalloc2(npoints,&xw,npoints,&ww);CHKERRQ(ierr);
582744bafbcSMatthew G. Knepley     ierr = PetscDTGaussQuadrature(npoints, a, b, xw, ww);CHKERRQ(ierr);
583744bafbcSMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
584744bafbcSMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
585744bafbcSMatthew G. Knepley         for (k = 0; k < npoints; ++k) {
586744bafbcSMatthew G. Knepley           x[((i*npoints+j)*npoints+k)*dim+0] = xw[i];
587744bafbcSMatthew G. Knepley           x[((i*npoints+j)*npoints+k)*dim+1] = xw[j];
588744bafbcSMatthew G. Knepley           x[((i*npoints+j)*npoints+k)*dim+2] = xw[k];
589a6b92713SMatthew G. Knepley           for (c = 0; c < Nc; ++c) w[((i*npoints+j)*npoints+k)*Nc+c] = ww[i] * ww[j] * ww[k];
590744bafbcSMatthew G. Knepley         }
591744bafbcSMatthew G. Knepley       }
592744bafbcSMatthew G. Knepley     }
593744bafbcSMatthew G. Knepley     ierr = PetscFree2(xw,ww);CHKERRQ(ierr);
594744bafbcSMatthew G. Knepley     break;
595744bafbcSMatthew G. Knepley   default:
596744bafbcSMatthew G. Knepley     SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim);
597744bafbcSMatthew G. Knepley   }
598744bafbcSMatthew G. Knepley   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr);
599*dcce0ee2SMatthew G. Knepley   ierr = PetscQuadratureSetOrder(*q, npoints-1);CHKERRQ(ierr);
600a6b92713SMatthew G. Knepley   ierr = PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w);CHKERRQ(ierr);
601744bafbcSMatthew G. Knepley   PetscFunctionReturn(0);
602744bafbcSMatthew G. Knepley }
603744bafbcSMatthew G. Knepley 
604494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x.
605494e7359SMatthew G. Knepley    Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */
606494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTFactorial_Internal(PetscInt n, PetscReal *factorial)
607494e7359SMatthew G. Knepley {
608494e7359SMatthew G. Knepley   PetscReal f = 1.0;
609494e7359SMatthew G. Knepley   PetscInt  i;
610494e7359SMatthew G. Knepley 
611494e7359SMatthew G. Knepley   PetscFunctionBegin;
612494e7359SMatthew G. Knepley   for (i = 1; i < n+1; ++i) f *= i;
613494e7359SMatthew G. Knepley   *factorial = f;
614494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
615494e7359SMatthew G. Knepley }
616494e7359SMatthew G. Knepley 
617494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x.
618494e7359SMatthew G. Knepley    Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */
619494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P)
620494e7359SMatthew G. Knepley {
621494e7359SMatthew G. Knepley   PetscReal apb, pn1, pn2;
622494e7359SMatthew G. Knepley   PetscInt  k;
623494e7359SMatthew G. Knepley 
624494e7359SMatthew G. Knepley   PetscFunctionBegin;
625494e7359SMatthew G. Knepley   if (!n) {*P = 1.0; PetscFunctionReturn(0);}
626494e7359SMatthew G. Knepley   if (n == 1) {*P = 0.5 * (a - b + (a + b + 2.0) * x); PetscFunctionReturn(0);}
627494e7359SMatthew G. Knepley   apb = a + b;
628494e7359SMatthew G. Knepley   pn2 = 1.0;
629494e7359SMatthew G. Knepley   pn1 = 0.5 * (a - b + (apb + 2.0) * x);
630494e7359SMatthew G. Knepley   *P  = 0.0;
631494e7359SMatthew G. Knepley   for (k = 2; k < n+1; ++k) {
632494e7359SMatthew G. Knepley     PetscReal a1 = 2.0 * k * (k + apb) * (2.0*k + apb - 2.0);
633494e7359SMatthew G. Knepley     PetscReal a2 = (2.0 * k + apb - 1.0) * (a*a - b*b);
634494e7359SMatthew G. Knepley     PetscReal a3 = (2.0 * k + apb - 2.0) * (2.0 * k + apb - 1.0) * (2.0 * k + apb);
635494e7359SMatthew G. Knepley     PetscReal a4 = 2.0 * (k + a - 1.0) * (k + b - 1.0) * (2.0 * k + apb);
636494e7359SMatthew G. Knepley 
637494e7359SMatthew G. Knepley     a2  = a2 / a1;
638494e7359SMatthew G. Knepley     a3  = a3 / a1;
639494e7359SMatthew G. Knepley     a4  = a4 / a1;
640494e7359SMatthew G. Knepley     *P  = (a2 + a3 * x) * pn1 - a4 * pn2;
641494e7359SMatthew G. Knepley     pn2 = pn1;
642494e7359SMatthew G. Knepley     pn1 = *P;
643494e7359SMatthew G. Knepley   }
644494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
645494e7359SMatthew G. Knepley }
646494e7359SMatthew G. Knepley 
647494e7359SMatthew G. Knepley /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */
648494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P)
649494e7359SMatthew G. Knepley {
650494e7359SMatthew G. Knepley   PetscReal      nP;
651494e7359SMatthew G. Knepley   PetscErrorCode ierr;
652494e7359SMatthew G. Knepley 
653494e7359SMatthew G. Knepley   PetscFunctionBegin;
654494e7359SMatthew G. Knepley   if (!n) {*P = 0.0; PetscFunctionReturn(0);}
655494e7359SMatthew G. Knepley   ierr = PetscDTComputeJacobi(a+1, b+1, n-1, x, &nP);CHKERRQ(ierr);
656494e7359SMatthew G. Knepley   *P   = 0.5 * (a + b + n + 1) * nP;
657494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
658494e7359SMatthew G. Knepley }
659494e7359SMatthew G. Knepley 
660494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */
661494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapSquareToTriangle_Internal(PetscReal x, PetscReal y, PetscReal *xi, PetscReal *eta)
662494e7359SMatthew G. Knepley {
663494e7359SMatthew G. Knepley   PetscFunctionBegin;
664494e7359SMatthew G. Knepley   *xi  = 0.5 * (1.0 + x) * (1.0 - y) - 1.0;
665494e7359SMatthew G. Knepley   *eta = y;
666494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
667494e7359SMatthew G. Knepley }
668494e7359SMatthew G. Knepley 
669494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */
670494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapCubeToTetrahedron_Internal(PetscReal x, PetscReal y, PetscReal z, PetscReal *xi, PetscReal *eta, PetscReal *zeta)
671494e7359SMatthew G. Knepley {
672494e7359SMatthew G. Knepley   PetscFunctionBegin;
673494e7359SMatthew G. Knepley   *xi   = 0.25 * (1.0 + x) * (1.0 - y) * (1.0 - z) - 1.0;
674494e7359SMatthew G. Knepley   *eta  = 0.5  * (1.0 + y) * (1.0 - z) - 1.0;
675494e7359SMatthew G. Knepley   *zeta = z;
676494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
677494e7359SMatthew G. Knepley }
678494e7359SMatthew G. Knepley 
679494e7359SMatthew G. Knepley static PetscErrorCode PetscDTGaussJacobiQuadrature1D_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w)
680494e7359SMatthew G. Knepley {
681494e7359SMatthew G. Knepley   PetscInt       maxIter = 100;
682494e7359SMatthew G. Knepley   PetscReal      eps     = 1.0e-8;
683a8291ba1SSatish Balay   PetscReal      a1, a2, a3, a4, a5, a6;
684494e7359SMatthew G. Knepley   PetscInt       k;
685494e7359SMatthew G. Knepley   PetscErrorCode ierr;
686494e7359SMatthew G. Knepley 
687494e7359SMatthew G. Knepley   PetscFunctionBegin;
688a8291ba1SSatish Balay 
6898b49ba18SBarry Smith   a1      = PetscPowReal(2.0, a+b+1);
690a8291ba1SSatish Balay #if defined(PETSC_HAVE_TGAMMA)
6910646a658SBarry Smith   a2      = PetscTGamma(a + npoints + 1);
6920646a658SBarry Smith   a3      = PetscTGamma(b + npoints + 1);
6930646a658SBarry Smith   a4      = PetscTGamma(a + b + npoints + 1);
694a8291ba1SSatish Balay #else
69529bcbfd0SToby Isaac   {
696d24bbb91SToby Isaac     PetscInt ia, ib;
69729bcbfd0SToby Isaac 
698d24bbb91SToby Isaac     ia = (PetscInt) a;
699d24bbb91SToby Isaac     ib = (PetscInt) b;
700d24bbb91SToby Isaac     if (ia == a && ib == b && ia + npoints + 1 > 0 && ib + npoints + 1 > 0 && ia + ib + npoints + 1 > 0) { /* All gamma(x) terms are (x-1)! terms */
701d24bbb91SToby Isaac       ierr = PetscDTFactorial_Internal(ia + npoints, &a2);CHKERRQ(ierr);
702d24bbb91SToby Isaac       ierr = PetscDTFactorial_Internal(ib + npoints, &a3);CHKERRQ(ierr);
703d24bbb91SToby Isaac       ierr = PetscDTFactorial_Internal(ia + ib + npoints, &a4);CHKERRQ(ierr);
70429bcbfd0SToby Isaac     } else {
705a8291ba1SSatish Balay       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"tgamma() - math routine is unavailable.");
70629bcbfd0SToby Isaac     }
70729bcbfd0SToby Isaac   }
708a8291ba1SSatish Balay #endif
709a8291ba1SSatish Balay 
710494e7359SMatthew G. Knepley   ierr = PetscDTFactorial_Internal(npoints, &a5);CHKERRQ(ierr);
711494e7359SMatthew G. Knepley   a6   = a1 * a2 * a3 / a4 / a5;
712494e7359SMatthew G. Knepley   /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses.
713494e7359SMatthew G. Knepley    Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */
714494e7359SMatthew G. Knepley   for (k = 0; k < npoints; ++k) {
7158b49ba18SBarry Smith     PetscReal r = -PetscCosReal((2.0*k + 1.0) * PETSC_PI / (2.0 * npoints)), dP;
716494e7359SMatthew G. Knepley     PetscInt  j;
717494e7359SMatthew G. Knepley 
718494e7359SMatthew G. Knepley     if (k > 0) r = 0.5 * (r + x[k-1]);
719494e7359SMatthew G. Knepley     for (j = 0; j < maxIter; ++j) {
720494e7359SMatthew G. Knepley       PetscReal s = 0.0, delta, f, fp;
721494e7359SMatthew G. Knepley       PetscInt  i;
722494e7359SMatthew G. Knepley 
723494e7359SMatthew G. Knepley       for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]);
724494e7359SMatthew G. Knepley       ierr = PetscDTComputeJacobi(a, b, npoints, r, &f);CHKERRQ(ierr);
725494e7359SMatthew G. Knepley       ierr = PetscDTComputeJacobiDerivative(a, b, npoints, r, &fp);CHKERRQ(ierr);
726494e7359SMatthew G. Knepley       delta = f / (fp - f * s);
727494e7359SMatthew G. Knepley       r     = r - delta;
72877b4d14cSPeter Brune       if (PetscAbsReal(delta) < eps) break;
729494e7359SMatthew G. Knepley     }
730494e7359SMatthew G. Knepley     x[k] = r;
731494e7359SMatthew G. Knepley     ierr = PetscDTComputeJacobiDerivative(a, b, npoints, x[k], &dP);CHKERRQ(ierr);
732494e7359SMatthew G. Knepley     w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP);
733494e7359SMatthew G. Knepley   }
734494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
735494e7359SMatthew G. Knepley }
736494e7359SMatthew G. Knepley 
737fd9d31fbSMatthew G. Knepley /*@C
738494e7359SMatthew G. Knepley   PetscDTGaussJacobiQuadrature - create Gauss-Jacobi quadrature for a simplex
739494e7359SMatthew G. Knepley 
740494e7359SMatthew G. Knepley   Not Collective
741494e7359SMatthew G. Knepley 
742494e7359SMatthew G. Knepley   Input Arguments:
743494e7359SMatthew G. Knepley + dim     - The simplex dimension
744a6b92713SMatthew G. Knepley . Nc      - The number of components
745*dcce0ee2SMatthew G. Knepley . npoints - The number of points in one dimension
746494e7359SMatthew G. Knepley . a       - left end of interval (often-1)
747494e7359SMatthew G. Knepley - b       - right end of interval (often +1)
748494e7359SMatthew G. Knepley 
749744bafbcSMatthew G. Knepley   Output Argument:
750552aa4f7SMatthew G. Knepley . q - A PetscQuadrature object
751494e7359SMatthew G. Knepley 
752494e7359SMatthew G. Knepley   Level: intermediate
753494e7359SMatthew G. Knepley 
754494e7359SMatthew G. Knepley   References:
75596a0c994SBarry Smith .  1. - Karniadakis and Sherwin.  FIAT
756494e7359SMatthew G. Knepley 
757744bafbcSMatthew G. Knepley .seealso: PetscDTGaussTensorQuadrature(), PetscDTGaussQuadrature()
758494e7359SMatthew G. Knepley @*/
759*dcce0ee2SMatthew G. Knepley PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q)
760494e7359SMatthew G. Knepley {
761*dcce0ee2SMatthew G. Knepley   PetscInt       totpoints = dim > 1 ? dim > 2 ? npoints*PetscSqr(npoints) : PetscSqr(npoints) : npoints;
762494e7359SMatthew G. Knepley   PetscReal     *px, *wx, *py, *wy, *pz, *wz, *x, *w;
763a6b92713SMatthew G. Knepley   PetscInt       i, j, k, c;
764494e7359SMatthew G. Knepley   PetscErrorCode ierr;
765494e7359SMatthew G. Knepley 
766494e7359SMatthew G. Knepley   PetscFunctionBegin;
767494e7359SMatthew G. Knepley   if ((a != -1.0) || (b != 1.0)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now");
768*dcce0ee2SMatthew G. Knepley   ierr = PetscMalloc1(totpoints*dim, &x);CHKERRQ(ierr);
769*dcce0ee2SMatthew G. Knepley   ierr = PetscMalloc1(totpoints*Nc, &w);CHKERRQ(ierr);
770494e7359SMatthew G. Knepley   switch (dim) {
771707aa5c5SMatthew G. Knepley   case 0:
772707aa5c5SMatthew G. Knepley     ierr = PetscFree(x);CHKERRQ(ierr);
773707aa5c5SMatthew G. Knepley     ierr = PetscFree(w);CHKERRQ(ierr);
774785e854fSJed Brown     ierr = PetscMalloc1(1, &x);CHKERRQ(ierr);
775a6b92713SMatthew G. Knepley     ierr = PetscMalloc1(Nc, &w);CHKERRQ(ierr);
776707aa5c5SMatthew G. Knepley     x[0] = 0.0;
777a6b92713SMatthew G. Knepley     for (c = 0; c < Nc; ++c) w[c] = 1.0;
778707aa5c5SMatthew G. Knepley     break;
779494e7359SMatthew G. Knepley   case 1:
780*dcce0ee2SMatthew G. Knepley     ierr = PetscMalloc1(npoints,&wx);CHKERRQ(ierr);
781*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, x, wx);CHKERRQ(ierr);
782*dcce0ee2SMatthew G. Knepley     for (i = 0; i < npoints; ++i) for (c = 0; c < Nc; ++c) w[i*Nc+c] = wx[i];
783a6b92713SMatthew G. Knepley     ierr = PetscFree(wx);CHKERRQ(ierr);
784494e7359SMatthew G. Knepley     break;
785494e7359SMatthew G. Knepley   case 2:
786*dcce0ee2SMatthew G. Knepley     ierr = PetscMalloc4(npoints,&px,npoints,&wx,npoints,&py,npoints,&wy);CHKERRQ(ierr);
787*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr);
788*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr);
789*dcce0ee2SMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
790*dcce0ee2SMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
791*dcce0ee2SMatthew G. Knepley         ierr = PetscDTMapSquareToTriangle_Internal(px[i], py[j], &x[(i*npoints+j)*2+0], &x[(i*npoints+j)*2+1]);CHKERRQ(ierr);
792*dcce0ee2SMatthew G. Knepley         for (c = 0; c < Nc; ++c) w[(i*npoints+j)*Nc+c] = 0.5 * wx[i] * wy[j];
793494e7359SMatthew G. Knepley       }
794494e7359SMatthew G. Knepley     }
795494e7359SMatthew G. Knepley     ierr = PetscFree4(px,wx,py,wy);CHKERRQ(ierr);
796494e7359SMatthew G. Knepley     break;
797494e7359SMatthew G. Knepley   case 3:
798*dcce0ee2SMatthew G. Knepley     ierr = PetscMalloc6(npoints,&px,npoints,&wx,npoints,&py,npoints,&wy,npoints,&pz,npoints,&wz);CHKERRQ(ierr);
799*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr);
800*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr);
801*dcce0ee2SMatthew G. Knepley     ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 2.0, 0.0, pz, wz);CHKERRQ(ierr);
802*dcce0ee2SMatthew G. Knepley     for (i = 0; i < npoints; ++i) {
803*dcce0ee2SMatthew G. Knepley       for (j = 0; j < npoints; ++j) {
804*dcce0ee2SMatthew G. Knepley         for (k = 0; k < npoints; ++k) {
805*dcce0ee2SMatthew 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);
806*dcce0ee2SMatthew G. Knepley           for (c = 0; c < Nc; ++c) w[((i*npoints+j)*npoints+k)*Nc+c] = 0.125 * wx[i] * wy[j] * wz[k];
807494e7359SMatthew G. Knepley         }
808494e7359SMatthew G. Knepley       }
809494e7359SMatthew G. Knepley     }
810494e7359SMatthew G. Knepley     ierr = PetscFree6(px,wx,py,wy,pz,wz);CHKERRQ(ierr);
811494e7359SMatthew G. Knepley     break;
812494e7359SMatthew G. Knepley   default:
813494e7359SMatthew G. Knepley     SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim);
814494e7359SMatthew G. Knepley   }
81521454ff5SMatthew G. Knepley   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr);
816*dcce0ee2SMatthew G. Knepley   ierr = PetscQuadratureSetOrder(*q, npoints-1);CHKERRQ(ierr);
817*dcce0ee2SMatthew G. Knepley   ierr = PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w);CHKERRQ(ierr);
818494e7359SMatthew G. Knepley   PetscFunctionReturn(0);
819494e7359SMatthew G. Knepley }
820494e7359SMatthew G. Knepley 
821b3c0f97bSTom Klotz /*@C
822b3c0f97bSTom Klotz   PetscDTTanhSinhTensorQuadrature - create tanh-sinh quadrature for a tensor product cell
823b3c0f97bSTom Klotz 
824b3c0f97bSTom Klotz   Not Collective
825b3c0f97bSTom Klotz 
826b3c0f97bSTom Klotz   Input Arguments:
827b3c0f97bSTom Klotz + dim   - The cell dimension
828b3c0f97bSTom Klotz . level - The number of points in one dimension, 2^l
829b3c0f97bSTom Klotz . a     - left end of interval (often-1)
830b3c0f97bSTom Klotz - b     - right end of interval (often +1)
831b3c0f97bSTom Klotz 
832b3c0f97bSTom Klotz   Output Argument:
833b3c0f97bSTom Klotz . q - A PetscQuadrature object
834b3c0f97bSTom Klotz 
835b3c0f97bSTom Klotz   Level: intermediate
836b3c0f97bSTom Klotz 
837b3c0f97bSTom Klotz .seealso: PetscDTGaussTensorQuadrature()
838b3c0f97bSTom Klotz @*/
839b3c0f97bSTom Klotz PetscErrorCode PetscDTTanhSinhTensorQuadrature(PetscInt dim, PetscInt level, PetscReal a, PetscReal b, PetscQuadrature *q)
840b3c0f97bSTom Klotz {
841b3c0f97bSTom Klotz   const PetscInt  p     = 16;                        /* Digits of precision in the evaluation */
842b3c0f97bSTom Klotz   const PetscReal alpha = (b-a)/2.;                  /* Half-width of the integration interval */
843b3c0f97bSTom Klotz   const PetscReal beta  = (b+a)/2.;                  /* Center of the integration interval */
844b3c0f97bSTom Klotz   const PetscReal h     = PetscPowReal(2.0, -level); /* Step size, length between x_k */
845d84b4d08SMatthew G. Knepley   PetscReal       xk;                                /* Quadrature point x_k on reference domain [-1, 1] */
846b3c0f97bSTom Klotz   PetscReal       wk    = 0.5*PETSC_PI;              /* Quadrature weight at x_k */
847b3c0f97bSTom Klotz   PetscReal      *x, *w;
848b3c0f97bSTom Klotz   PetscInt        K, k, npoints;
849b3c0f97bSTom Klotz   PetscErrorCode  ierr;
850b3c0f97bSTom Klotz 
851b3c0f97bSTom Klotz   PetscFunctionBegin;
852b3c0f97bSTom Klotz   if (dim > 1) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %d not yet implemented", dim);
853b3c0f97bSTom Klotz   if (!level) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a number of significant digits");
854b3c0f97bSTom Klotz   /* Find K such that the weights are < 32 digits of precision */
855b3c0f97bSTom Klotz   for (K = 1; PetscAbsReal(PetscLog10Real(wk)) < 2*p; ++K) {
8569add2064SThomas Klotz     wk = 0.5*h*PETSC_PI*PetscCoshReal(K*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(K*h)));
857b3c0f97bSTom Klotz   }
858b3c0f97bSTom Klotz   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr);
859b3c0f97bSTom Klotz   ierr = PetscQuadratureSetOrder(*q, 2*K+1);CHKERRQ(ierr);
860b3c0f97bSTom Klotz   npoints = 2*K-1;
861b3c0f97bSTom Klotz   ierr = PetscMalloc1(npoints*dim, &x);CHKERRQ(ierr);
862b3c0f97bSTom Klotz   ierr = PetscMalloc1(npoints, &w);CHKERRQ(ierr);
863b3c0f97bSTom Klotz   /* Center term */
864b3c0f97bSTom Klotz   x[0] = beta;
865b3c0f97bSTom Klotz   w[0] = 0.5*alpha*PETSC_PI;
866b3c0f97bSTom Klotz   for (k = 1; k < K; ++k) {
8679add2064SThomas Klotz     wk = 0.5*alpha*h*PETSC_PI*PetscCoshReal(k*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h)));
8689add2064SThomas Klotz     xk = tanh(0.5*PETSC_PI*PetscSinhReal(k*h));
869b3c0f97bSTom Klotz     x[2*k-1] = -alpha*xk+beta;
870b3c0f97bSTom Klotz     w[2*k-1] = wk;
871b3c0f97bSTom Klotz     x[2*k+0] =  alpha*xk+beta;
872b3c0f97bSTom Klotz     w[2*k+0] = wk;
873b3c0f97bSTom Klotz   }
874a6b92713SMatthew G. Knepley   ierr = PetscQuadratureSetData(*q, dim, 1, npoints, x, w);CHKERRQ(ierr);
875b3c0f97bSTom Klotz   PetscFunctionReturn(0);
876b3c0f97bSTom Klotz }
877b3c0f97bSTom Klotz 
878b3c0f97bSTom Klotz PetscErrorCode PetscDTTanhSinhIntegrate(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol)
879b3c0f97bSTom Klotz {
880b3c0f97bSTom Klotz   const PetscInt  p     = 16;        /* Digits of precision in the evaluation */
881b3c0f97bSTom Klotz   const PetscReal alpha = (b-a)/2.;  /* Half-width of the integration interval */
882b3c0f97bSTom Klotz   const PetscReal beta  = (b+a)/2.;  /* Center of the integration interval */
883b3c0f97bSTom Klotz   PetscReal       h     = 1.0;       /* Step size, length between x_k */
884b3c0f97bSTom Klotz   PetscInt        l     = 0;         /* Level of refinement, h = 2^{-l} */
885b3c0f97bSTom Klotz   PetscReal       osum  = 0.0;       /* Integral on last level */
886b3c0f97bSTom Klotz   PetscReal       psum  = 0.0;       /* Integral on the level before the last level */
887b3c0f97bSTom Klotz   PetscReal       sum;               /* Integral on current level */
888446c295cSMatthew G. Knepley   PetscReal       yk;                /* Quadrature point 1 - x_k on reference domain [-1, 1] */
889b3c0f97bSTom Klotz   PetscReal       lx, rx;            /* Quadrature points to the left and right of 0 on the real domain [a, b] */
890b3c0f97bSTom Klotz   PetscReal       wk;                /* Quadrature weight at x_k */
891b3c0f97bSTom Klotz   PetscReal       lval, rval;        /* Terms in the quadature sum to the left and right of 0 */
892b3c0f97bSTom Klotz   PetscInt        d;                 /* Digits of precision in the integral */
893b3c0f97bSTom Klotz 
894b3c0f97bSTom Klotz   PetscFunctionBegin;
895b3c0f97bSTom Klotz   if (digits <= 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits");
896b3c0f97bSTom Klotz   /* Center term */
897b3c0f97bSTom Klotz   func(beta, &lval);
898b3c0f97bSTom Klotz   sum = 0.5*alpha*PETSC_PI*lval;
899b3c0f97bSTom Klotz   /* */
900b3c0f97bSTom Klotz   do {
901b3c0f97bSTom Klotz     PetscReal lterm, rterm, maxTerm = 0.0, d1, d2, d3, d4;
902b3c0f97bSTom Klotz     PetscInt  k = 1;
903b3c0f97bSTom Klotz 
904b3c0f97bSTom Klotz     ++l;
905b3c0f97bSTom Klotz     /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %D sum: %15.15f\n", l, sum); */
906b3c0f97bSTom Klotz     /* At each level of refinement, h --> h/2 and sum --> sum/2 */
907b3c0f97bSTom Klotz     psum = osum;
908b3c0f97bSTom Klotz     osum = sum;
909b3c0f97bSTom Klotz     h   *= 0.5;
910b3c0f97bSTom Klotz     sum *= 0.5;
911b3c0f97bSTom Klotz     do {
9129add2064SThomas Klotz       wk = 0.5*h*PETSC_PI*PetscCoshReal(k*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h)));
913446c295cSMatthew G. Knepley       yk = 1.0/(PetscExpReal(0.5*PETSC_PI*PetscSinhReal(k*h)) * PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h)));
914446c295cSMatthew G. Knepley       lx = -alpha*(1.0 - yk)+beta;
915446c295cSMatthew G. Knepley       rx =  alpha*(1.0 - yk)+beta;
916b3c0f97bSTom Klotz       func(lx, &lval);
917b3c0f97bSTom Klotz       func(rx, &rval);
918b3c0f97bSTom Klotz       lterm   = alpha*wk*lval;
919b3c0f97bSTom Klotz       maxTerm = PetscMax(PetscAbsReal(lterm), maxTerm);
920b3c0f97bSTom Klotz       sum    += lterm;
921b3c0f97bSTom Klotz       rterm   = alpha*wk*rval;
922b3c0f97bSTom Klotz       maxTerm = PetscMax(PetscAbsReal(rterm), maxTerm);
923b3c0f97bSTom Klotz       sum    += rterm;
924b3c0f97bSTom Klotz       ++k;
925b3c0f97bSTom Klotz       /* Only need to evaluate every other point on refined levels */
926b3c0f97bSTom Klotz       if (l != 1) ++k;
9279add2064SThomas Klotz     } while (PetscAbsReal(PetscLog10Real(wk)) < p); /* Only need to evaluate sum until weights are < 32 digits of precision */
928b3c0f97bSTom Klotz 
929b3c0f97bSTom Klotz     d1 = PetscLog10Real(PetscAbsReal(sum - osum));
930b3c0f97bSTom Klotz     d2 = PetscLog10Real(PetscAbsReal(sum - psum));
931b3c0f97bSTom Klotz     d3 = PetscLog10Real(maxTerm) - p;
93209d48545SBarry Smith     if (PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm)) == 0.0) d4 = 0.0;
93309d48545SBarry Smith     else d4 = PetscLog10Real(PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm)));
934b3c0f97bSTom Klotz     d  = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1)/d2, 2*d1), d3), d4)));
9359add2064SThomas Klotz   } while (d < digits && l < 12);
936b3c0f97bSTom Klotz   *sol = sum;
937e510cb1fSThomas Klotz 
938b3c0f97bSTom Klotz   PetscFunctionReturn(0);
939b3c0f97bSTom Klotz }
940b3c0f97bSTom Klotz 
94129f144ccSMatthew G. Knepley #ifdef PETSC_HAVE_MPFR
94229f144ccSMatthew G. Knepley PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol)
94329f144ccSMatthew G. Knepley {
944e510cb1fSThomas Klotz   const PetscInt  safetyFactor = 2;  /* Calculate abcissa until 2*p digits */
94529f144ccSMatthew G. Knepley   PetscInt        l            = 0;  /* Level of refinement, h = 2^{-l} */
94629f144ccSMatthew G. Knepley   mpfr_t          alpha;             /* Half-width of the integration interval */
94729f144ccSMatthew G. Knepley   mpfr_t          beta;              /* Center of the integration interval */
94829f144ccSMatthew G. Knepley   mpfr_t          h;                 /* Step size, length between x_k */
94929f144ccSMatthew G. Knepley   mpfr_t          osum;              /* Integral on last level */
95029f144ccSMatthew G. Knepley   mpfr_t          psum;              /* Integral on the level before the last level */
95129f144ccSMatthew G. Knepley   mpfr_t          sum;               /* Integral on current level */
95229f144ccSMatthew G. Knepley   mpfr_t          yk;                /* Quadrature point 1 - x_k on reference domain [-1, 1] */
95329f144ccSMatthew G. Knepley   mpfr_t          lx, rx;            /* Quadrature points to the left and right of 0 on the real domain [a, b] */
95429f144ccSMatthew G. Knepley   mpfr_t          wk;                /* Quadrature weight at x_k */
95529f144ccSMatthew G. Knepley   PetscReal       lval, rval;        /* Terms in the quadature sum to the left and right of 0 */
95629f144ccSMatthew G. Knepley   PetscInt        d;                 /* Digits of precision in the integral */
95729f144ccSMatthew G. Knepley   mpfr_t          pi2, kh, msinh, mcosh, maxTerm, curTerm, tmp;
95829f144ccSMatthew G. Knepley 
95929f144ccSMatthew G. Knepley   PetscFunctionBegin;
96029f144ccSMatthew G. Knepley   if (digits <= 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits");
96129f144ccSMatthew G. Knepley   /* Create high precision storage */
962c9f744b5SMatthew G. Knepley   mpfr_inits2(PetscCeilReal(safetyFactor*digits*PetscLogReal(10.)/PetscLogReal(2.)), alpha, beta, h, sum, osum, psum, yk, wk, lx, rx, tmp, maxTerm, curTerm, pi2, kh, msinh, mcosh, NULL);
96329f144ccSMatthew G. Knepley   /* Initialization */
96429f144ccSMatthew G. Knepley   mpfr_set_d(alpha, 0.5*(b-a), MPFR_RNDN);
96529f144ccSMatthew G. Knepley   mpfr_set_d(beta,  0.5*(b+a), MPFR_RNDN);
96629f144ccSMatthew G. Knepley   mpfr_set_d(osum,  0.0,       MPFR_RNDN);
96729f144ccSMatthew G. Knepley   mpfr_set_d(psum,  0.0,       MPFR_RNDN);
96829f144ccSMatthew G. Knepley   mpfr_set_d(h,     1.0,       MPFR_RNDN);
96929f144ccSMatthew G. Knepley   mpfr_const_pi(pi2, MPFR_RNDN);
97029f144ccSMatthew G. Knepley   mpfr_mul_d(pi2, pi2, 0.5, MPFR_RNDN);
97129f144ccSMatthew G. Knepley   /* Center term */
97229f144ccSMatthew G. Knepley   func(0.5*(b+a), &lval);
97329f144ccSMatthew G. Knepley   mpfr_set(sum, pi2, MPFR_RNDN);
97429f144ccSMatthew G. Knepley   mpfr_mul(sum, sum, alpha, MPFR_RNDN);
97529f144ccSMatthew G. Knepley   mpfr_mul_d(sum, sum, lval, MPFR_RNDN);
97629f144ccSMatthew G. Knepley   /* */
97729f144ccSMatthew G. Knepley   do {
97829f144ccSMatthew G. Knepley     PetscReal d1, d2, d3, d4;
97929f144ccSMatthew G. Knepley     PetscInt  k = 1;
98029f144ccSMatthew G. Knepley 
98129f144ccSMatthew G. Knepley     ++l;
98229f144ccSMatthew G. Knepley     mpfr_set_d(maxTerm, 0.0, MPFR_RNDN);
98329f144ccSMatthew G. Knepley     /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %D sum: %15.15f\n", l, sum); */
98429f144ccSMatthew G. Knepley     /* At each level of refinement, h --> h/2 and sum --> sum/2 */
98529f144ccSMatthew G. Knepley     mpfr_set(psum, osum, MPFR_RNDN);
98629f144ccSMatthew G. Knepley     mpfr_set(osum,  sum, MPFR_RNDN);
98729f144ccSMatthew G. Knepley     mpfr_mul_d(h,   h,   0.5, MPFR_RNDN);
98829f144ccSMatthew G. Knepley     mpfr_mul_d(sum, sum, 0.5, MPFR_RNDN);
98929f144ccSMatthew G. Knepley     do {
99029f144ccSMatthew G. Knepley       mpfr_set_si(kh, k, MPFR_RNDN);
99129f144ccSMatthew G. Knepley       mpfr_mul(kh, kh, h, MPFR_RNDN);
99229f144ccSMatthew G. Knepley       /* Weight */
99329f144ccSMatthew G. Knepley       mpfr_set(wk, h, MPFR_RNDN);
99429f144ccSMatthew G. Knepley       mpfr_sinh_cosh(msinh, mcosh, kh, MPFR_RNDN);
99529f144ccSMatthew G. Knepley       mpfr_mul(msinh, msinh, pi2, MPFR_RNDN);
99629f144ccSMatthew G. Knepley       mpfr_mul(mcosh, mcosh, pi2, MPFR_RNDN);
99729f144ccSMatthew G. Knepley       mpfr_cosh(tmp, msinh, MPFR_RNDN);
99829f144ccSMatthew G. Knepley       mpfr_sqr(tmp, tmp, MPFR_RNDN);
99929f144ccSMatthew G. Knepley       mpfr_mul(wk, wk, mcosh, MPFR_RNDN);
100029f144ccSMatthew G. Knepley       mpfr_div(wk, wk, tmp, MPFR_RNDN);
100129f144ccSMatthew G. Knepley       /* Abscissa */
100229f144ccSMatthew G. Knepley       mpfr_set_d(yk, 1.0, MPFR_RNDZ);
100329f144ccSMatthew G. Knepley       mpfr_cosh(tmp, msinh, MPFR_RNDN);
100429f144ccSMatthew G. Knepley       mpfr_div(yk, yk, tmp, MPFR_RNDZ);
100529f144ccSMatthew G. Knepley       mpfr_exp(tmp, msinh, MPFR_RNDN);
100629f144ccSMatthew G. Knepley       mpfr_div(yk, yk, tmp, MPFR_RNDZ);
100729f144ccSMatthew G. Knepley       /* Quadrature points */
100829f144ccSMatthew G. Knepley       mpfr_sub_d(lx, yk, 1.0, MPFR_RNDZ);
100929f144ccSMatthew G. Knepley       mpfr_mul(lx, lx, alpha, MPFR_RNDU);
101029f144ccSMatthew G. Knepley       mpfr_add(lx, lx, beta, MPFR_RNDU);
101129f144ccSMatthew G. Knepley       mpfr_d_sub(rx, 1.0, yk, MPFR_RNDZ);
101229f144ccSMatthew G. Knepley       mpfr_mul(rx, rx, alpha, MPFR_RNDD);
101329f144ccSMatthew G. Knepley       mpfr_add(rx, rx, beta, MPFR_RNDD);
101429f144ccSMatthew G. Knepley       /* Evaluation */
101529f144ccSMatthew G. Knepley       func(mpfr_get_d(lx, MPFR_RNDU), &lval);
101629f144ccSMatthew G. Knepley       func(mpfr_get_d(rx, MPFR_RNDD), &rval);
101729f144ccSMatthew G. Knepley       /* Update */
101829f144ccSMatthew G. Knepley       mpfr_mul(tmp, wk, alpha, MPFR_RNDN);
101929f144ccSMatthew G. Knepley       mpfr_mul_d(tmp, tmp, lval, MPFR_RNDN);
102029f144ccSMatthew G. Knepley       mpfr_add(sum, sum, tmp, MPFR_RNDN);
102129f144ccSMatthew G. Knepley       mpfr_abs(tmp, tmp, MPFR_RNDN);
102229f144ccSMatthew G. Knepley       mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN);
102329f144ccSMatthew G. Knepley       mpfr_set(curTerm, tmp, MPFR_RNDN);
102429f144ccSMatthew G. Knepley       mpfr_mul(tmp, wk, alpha, MPFR_RNDN);
102529f144ccSMatthew G. Knepley       mpfr_mul_d(tmp, tmp, rval, MPFR_RNDN);
102629f144ccSMatthew G. Knepley       mpfr_add(sum, sum, tmp, MPFR_RNDN);
102729f144ccSMatthew G. Knepley       mpfr_abs(tmp, tmp, MPFR_RNDN);
102829f144ccSMatthew G. Knepley       mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN);
102929f144ccSMatthew G. Knepley       mpfr_max(curTerm, curTerm, tmp, MPFR_RNDN);
103029f144ccSMatthew G. Knepley       ++k;
103129f144ccSMatthew G. Knepley       /* Only need to evaluate every other point on refined levels */
103229f144ccSMatthew G. Knepley       if (l != 1) ++k;
103329f144ccSMatthew G. Knepley       mpfr_log10(tmp, wk, MPFR_RNDN);
103429f144ccSMatthew G. Knepley       mpfr_abs(tmp, tmp, MPFR_RNDN);
1035c9f744b5SMatthew G. Knepley     } while (mpfr_get_d(tmp, MPFR_RNDN) < safetyFactor*digits); /* Only need to evaluate sum until weights are < 32 digits of precision */
103629f144ccSMatthew G. Knepley     mpfr_sub(tmp, sum, osum, MPFR_RNDN);
103729f144ccSMatthew G. Knepley     mpfr_abs(tmp, tmp, MPFR_RNDN);
103829f144ccSMatthew G. Knepley     mpfr_log10(tmp, tmp, MPFR_RNDN);
103929f144ccSMatthew G. Knepley     d1 = mpfr_get_d(tmp, MPFR_RNDN);
104029f144ccSMatthew G. Knepley     mpfr_sub(tmp, sum, psum, MPFR_RNDN);
104129f144ccSMatthew G. Knepley     mpfr_abs(tmp, tmp, MPFR_RNDN);
104229f144ccSMatthew G. Knepley     mpfr_log10(tmp, tmp, MPFR_RNDN);
104329f144ccSMatthew G. Knepley     d2 = mpfr_get_d(tmp, MPFR_RNDN);
104429f144ccSMatthew G. Knepley     mpfr_log10(tmp, maxTerm, MPFR_RNDN);
1045c9f744b5SMatthew G. Knepley     d3 = mpfr_get_d(tmp, MPFR_RNDN) - digits;
104629f144ccSMatthew G. Knepley     mpfr_log10(tmp, curTerm, MPFR_RNDN);
104729f144ccSMatthew G. Knepley     d4 = mpfr_get_d(tmp, MPFR_RNDN);
104829f144ccSMatthew G. Knepley     d  = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1)/d2, 2*d1), d3), d4)));
1049b0649871SThomas Klotz   } while (d < digits && l < 8);
105029f144ccSMatthew G. Knepley   *sol = mpfr_get_d(sum, MPFR_RNDN);
105129f144ccSMatthew G. Knepley   /* Cleanup */
105229f144ccSMatthew G. Knepley   mpfr_clears(alpha, beta, h, sum, osum, psum, yk, wk, lx, rx, tmp, maxTerm, curTerm, pi2, kh, msinh, mcosh, NULL);
105329f144ccSMatthew G. Knepley   PetscFunctionReturn(0);
105429f144ccSMatthew G. Knepley }
1055d525116cSMatthew G. Knepley #else
1056fbfcfee5SBarry Smith 
1057d525116cSMatthew G. Knepley PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol)
1058d525116cSMatthew G. Knepley {
1059d525116cSMatthew G. Knepley   SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "This method will not work without MPFR. Reconfigure using --download-mpfr --download-gmp");
1060d525116cSMatthew G. Knepley }
106129f144ccSMatthew G. Knepley #endif
106229f144ccSMatthew G. Knepley 
1063194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n
1064194825f6SJed Brown  * A in column-major format
1065194825f6SJed Brown  * Ainv in row-major format
1066194825f6SJed Brown  * tau has length m
1067194825f6SJed Brown  * worksize must be >= max(1,n)
1068194825f6SJed Brown  */
1069194825f6SJed Brown static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m,PetscInt mstride,PetscInt n,PetscReal *A_in,PetscReal *Ainv_out,PetscScalar *tau,PetscInt worksize,PetscScalar *work)
1070194825f6SJed Brown {
1071194825f6SJed Brown   PetscErrorCode ierr;
1072194825f6SJed Brown   PetscBLASInt   M,N,K,lda,ldb,ldwork,info;
1073194825f6SJed Brown   PetscScalar    *A,*Ainv,*R,*Q,Alpha;
1074194825f6SJed Brown 
1075194825f6SJed Brown   PetscFunctionBegin;
1076194825f6SJed Brown #if defined(PETSC_USE_COMPLEX)
1077194825f6SJed Brown   {
1078194825f6SJed Brown     PetscInt i,j;
1079dcca6d9dSJed Brown     ierr = PetscMalloc2(m*n,&A,m*n,&Ainv);CHKERRQ(ierr);
1080194825f6SJed Brown     for (j=0; j<n; j++) {
1081194825f6SJed Brown       for (i=0; i<m; i++) A[i+m*j] = A_in[i+mstride*j];
1082194825f6SJed Brown     }
1083194825f6SJed Brown     mstride = m;
1084194825f6SJed Brown   }
1085194825f6SJed Brown #else
1086194825f6SJed Brown   A = A_in;
1087194825f6SJed Brown   Ainv = Ainv_out;
1088194825f6SJed Brown #endif
1089194825f6SJed Brown 
1090194825f6SJed Brown   ierr = PetscBLASIntCast(m,&M);CHKERRQ(ierr);
1091194825f6SJed Brown   ierr = PetscBLASIntCast(n,&N);CHKERRQ(ierr);
1092194825f6SJed Brown   ierr = PetscBLASIntCast(mstride,&lda);CHKERRQ(ierr);
1093194825f6SJed Brown   ierr = PetscBLASIntCast(worksize,&ldwork);CHKERRQ(ierr);
1094194825f6SJed Brown   ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1095001a771dSBarry Smith   PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&M,&N,A,&lda,tau,work,&ldwork,&info));
1096194825f6SJed Brown   ierr = PetscFPTrapPop();CHKERRQ(ierr);
1097194825f6SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xGEQRF error");
1098194825f6SJed Brown   R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */
1099194825f6SJed Brown 
1100194825f6SJed Brown   /* Extract an explicit representation of Q */
1101194825f6SJed Brown   Q = Ainv;
1102194825f6SJed Brown   ierr = PetscMemcpy(Q,A,mstride*n*sizeof(PetscScalar));CHKERRQ(ierr);
1103194825f6SJed Brown   K = N;                        /* full rank */
1104001a771dSBarry Smith   PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&M,&N,&K,Q,&lda,tau,work,&ldwork,&info));
1105194825f6SJed Brown   if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xORGQR/xUNGQR error");
1106194825f6SJed Brown 
1107194825f6SJed Brown   /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */
1108194825f6SJed Brown   Alpha = 1.0;
1109194825f6SJed Brown   ldb = lda;
1110001a771dSBarry Smith   PetscStackCallBLAS("BLAStrsm",BLAStrsm_("Right","Upper","ConjugateTranspose","NotUnitTriangular",&M,&N,&Alpha,R,&lda,Q,&ldb));
1111194825f6SJed Brown   /* Ainv is Q, overwritten with inverse */
1112194825f6SJed Brown 
1113194825f6SJed Brown #if defined(PETSC_USE_COMPLEX)
1114194825f6SJed Brown   {
1115194825f6SJed Brown     PetscInt i;
1116194825f6SJed Brown     for (i=0; i<m*n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]);
1117194825f6SJed Brown     ierr = PetscFree2(A,Ainv);CHKERRQ(ierr);
1118194825f6SJed Brown   }
1119194825f6SJed Brown #endif
1120194825f6SJed Brown   PetscFunctionReturn(0);
1121194825f6SJed Brown }
1122194825f6SJed Brown 
1123194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */
1124194825f6SJed Brown static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval,const PetscReal *x,PetscInt ndegree,const PetscInt *degrees,PetscBool Transpose,PetscReal *B)
1125194825f6SJed Brown {
1126194825f6SJed Brown   PetscErrorCode ierr;
1127194825f6SJed Brown   PetscReal      *Bv;
1128194825f6SJed Brown   PetscInt       i,j;
1129194825f6SJed Brown 
1130194825f6SJed Brown   PetscFunctionBegin;
1131785e854fSJed Brown   ierr = PetscMalloc1((ninterval+1)*ndegree,&Bv);CHKERRQ(ierr);
1132194825f6SJed Brown   /* Point evaluation of L_p on all the source vertices */
1133194825f6SJed Brown   ierr = PetscDTLegendreEval(ninterval+1,x,ndegree,degrees,Bv,NULL,NULL);CHKERRQ(ierr);
1134194825f6SJed Brown   /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */
1135194825f6SJed Brown   for (i=0; i<ninterval; i++) {
1136194825f6SJed Brown     for (j=0; j<ndegree; j++) {
1137194825f6SJed Brown       if (Transpose) B[i+ninterval*j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j];
1138194825f6SJed Brown       else           B[i*ndegree+j]   = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j];
1139194825f6SJed Brown     }
1140194825f6SJed Brown   }
1141194825f6SJed Brown   ierr = PetscFree(Bv);CHKERRQ(ierr);
1142194825f6SJed Brown   PetscFunctionReturn(0);
1143194825f6SJed Brown }
1144194825f6SJed Brown 
1145194825f6SJed Brown /*@
1146194825f6SJed Brown    PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals
1147194825f6SJed Brown 
1148194825f6SJed Brown    Not Collective
1149194825f6SJed Brown 
1150194825f6SJed Brown    Input Arguments:
1151194825f6SJed Brown +  degree - degree of reconstruction polynomial
1152194825f6SJed Brown .  nsource - number of source intervals
1153194825f6SJed Brown .  sourcex - sorted coordinates of source cell boundaries (length nsource+1)
1154194825f6SJed Brown .  ntarget - number of target intervals
1155194825f6SJed Brown -  targetx - sorted coordinates of target cell boundaries (length ntarget+1)
1156194825f6SJed Brown 
1157194825f6SJed Brown    Output Arguments:
1158194825f6SJed Brown .  R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s]
1159194825f6SJed Brown 
1160194825f6SJed Brown    Level: advanced
1161194825f6SJed Brown 
1162194825f6SJed Brown .seealso: PetscDTLegendreEval()
1163194825f6SJed Brown @*/
1164194825f6SJed Brown PetscErrorCode PetscDTReconstructPoly(PetscInt degree,PetscInt nsource,const PetscReal *sourcex,PetscInt ntarget,const PetscReal *targetx,PetscReal *R)
1165194825f6SJed Brown {
1166194825f6SJed Brown   PetscErrorCode ierr;
1167194825f6SJed Brown   PetscInt       i,j,k,*bdegrees,worksize;
1168194825f6SJed Brown   PetscReal      xmin,xmax,center,hscale,*sourcey,*targety,*Bsource,*Bsinv,*Btarget;
1169194825f6SJed Brown   PetscScalar    *tau,*work;
1170194825f6SJed Brown 
1171194825f6SJed Brown   PetscFunctionBegin;
1172194825f6SJed Brown   PetscValidRealPointer(sourcex,3);
1173194825f6SJed Brown   PetscValidRealPointer(targetx,5);
1174194825f6SJed Brown   PetscValidRealPointer(R,6);
1175194825f6SJed 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);
1176194825f6SJed Brown #if defined(PETSC_USE_DEBUG)
1177194825f6SJed Brown   for (i=0; i<nsource; i++) {
117857622a8eSBarry Smith     if (sourcex[i] >= sourcex[i+1]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_CORRUPT,"Source interval %D has negative orientation (%g,%g)",i,(double)sourcex[i],(double)sourcex[i+1]);
1179194825f6SJed Brown   }
1180194825f6SJed Brown   for (i=0; i<ntarget; i++) {
118157622a8eSBarry Smith     if (targetx[i] >= targetx[i+1]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_CORRUPT,"Target interval %D has negative orientation (%g,%g)",i,(double)targetx[i],(double)targetx[i+1]);
1182194825f6SJed Brown   }
1183194825f6SJed Brown #endif
1184194825f6SJed Brown   xmin = PetscMin(sourcex[0],targetx[0]);
1185194825f6SJed Brown   xmax = PetscMax(sourcex[nsource],targetx[ntarget]);
1186194825f6SJed Brown   center = (xmin + xmax)/2;
1187194825f6SJed Brown   hscale = (xmax - xmin)/2;
1188194825f6SJed Brown   worksize = nsource;
1189dcca6d9dSJed Brown   ierr = PetscMalloc4(degree+1,&bdegrees,nsource+1,&sourcey,nsource*(degree+1),&Bsource,worksize,&work);CHKERRQ(ierr);
1190dcca6d9dSJed Brown   ierr = PetscMalloc4(nsource,&tau,nsource*(degree+1),&Bsinv,ntarget+1,&targety,ntarget*(degree+1),&Btarget);CHKERRQ(ierr);
1191194825f6SJed Brown   for (i=0; i<=nsource; i++) sourcey[i] = (sourcex[i]-center)/hscale;
1192194825f6SJed Brown   for (i=0; i<=degree; i++) bdegrees[i] = i+1;
1193194825f6SJed Brown   ierr = PetscDTLegendreIntegrate(nsource,sourcey,degree+1,bdegrees,PETSC_TRUE,Bsource);CHKERRQ(ierr);
1194194825f6SJed Brown   ierr = PetscDTPseudoInverseQR(nsource,nsource,degree+1,Bsource,Bsinv,tau,nsource,work);CHKERRQ(ierr);
1195194825f6SJed Brown   for (i=0; i<=ntarget; i++) targety[i] = (targetx[i]-center)/hscale;
1196194825f6SJed Brown   ierr = PetscDTLegendreIntegrate(ntarget,targety,degree+1,bdegrees,PETSC_FALSE,Btarget);CHKERRQ(ierr);
1197194825f6SJed Brown   for (i=0; i<ntarget; i++) {
1198194825f6SJed Brown     PetscReal rowsum = 0;
1199194825f6SJed Brown     for (j=0; j<nsource; j++) {
1200194825f6SJed Brown       PetscReal sum = 0;
1201194825f6SJed Brown       for (k=0; k<degree+1; k++) {
1202194825f6SJed Brown         sum += Btarget[i*(degree+1)+k] * Bsinv[k*nsource+j];
1203194825f6SJed Brown       }
1204194825f6SJed Brown       R[i*nsource+j] = sum;
1205194825f6SJed Brown       rowsum += sum;
1206194825f6SJed Brown     }
1207194825f6SJed Brown     for (j=0; j<nsource; j++) R[i*nsource+j] /= rowsum; /* normalize each row */
1208194825f6SJed Brown   }
1209194825f6SJed Brown   ierr = PetscFree4(bdegrees,sourcey,Bsource,work);CHKERRQ(ierr);
1210194825f6SJed Brown   ierr = PetscFree4(tau,Bsinv,targety,Btarget);CHKERRQ(ierr);
1211194825f6SJed Brown   PetscFunctionReturn(0);
1212194825f6SJed Brown }
1213