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); 92f0a0bfafSMatthew G. Knepley ierr = PetscMalloc1(Nq*Nc, &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 2401fd49c25SBarry Smith Fortran Notes: From Fortran you must call PetscQuadratureRestoreData() when you are done with the data 2411fd49c25SBarry Smith 24240d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature 24340d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureSetData() 24440d8ff71SMatthew G. Knepley @*/ 245a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureGetData(PetscQuadrature q, PetscInt *dim, PetscInt *Nc, PetscInt *npoints, const PetscReal *points[], const PetscReal *weights[]) 24621454ff5SMatthew G. Knepley { 24721454ff5SMatthew G. Knepley PetscFunctionBegin; 24821454ff5SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1); 24921454ff5SMatthew G. Knepley if (dim) { 25021454ff5SMatthew G. Knepley PetscValidPointer(dim, 2); 25121454ff5SMatthew G. Knepley *dim = q->dim; 25221454ff5SMatthew G. Knepley } 253a6b92713SMatthew G. Knepley if (Nc) { 254a6b92713SMatthew G. Knepley PetscValidPointer(Nc, 3); 255a6b92713SMatthew G. Knepley *Nc = q->Nc; 256a6b92713SMatthew G. Knepley } 25721454ff5SMatthew G. Knepley if (npoints) { 258a6b92713SMatthew G. Knepley PetscValidPointer(npoints, 4); 25921454ff5SMatthew G. Knepley *npoints = q->numPoints; 26021454ff5SMatthew G. Knepley } 26121454ff5SMatthew G. Knepley if (points) { 262a6b92713SMatthew G. Knepley PetscValidPointer(points, 5); 26321454ff5SMatthew G. Knepley *points = q->points; 26421454ff5SMatthew G. Knepley } 26521454ff5SMatthew G. Knepley if (weights) { 266a6b92713SMatthew G. Knepley PetscValidPointer(weights, 6); 26721454ff5SMatthew G. Knepley *weights = q->weights; 26821454ff5SMatthew G. Knepley } 26921454ff5SMatthew G. Knepley PetscFunctionReturn(0); 27021454ff5SMatthew G. Knepley } 27121454ff5SMatthew G. Knepley 27240d8ff71SMatthew G. Knepley /*@C 27340d8ff71SMatthew G. Knepley PetscQuadratureSetData - Sets the data defining the quadrature 27440d8ff71SMatthew G. Knepley 27540d8ff71SMatthew G. Knepley Not collective 27640d8ff71SMatthew G. Knepley 27740d8ff71SMatthew G. Knepley Input Parameters: 27840d8ff71SMatthew G. Knepley + q - The PetscQuadrature object 27940d8ff71SMatthew G. Knepley . dim - The spatial dimension 280a6b92713SMatthew G. Knepley , Nc - The number of components 28140d8ff71SMatthew G. Knepley . npoints - The number of quadrature points 28240d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point 28340d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point 28440d8ff71SMatthew G. Knepley 2851fd49c25SBarry Smith Notes: The arrays are not copieed so you should not free arrays, the arrays must be optained with a version of PetscMalloc() not the system malloc 2861fd49c25SBarry Smith 28740d8ff71SMatthew G. Knepley Level: intermediate 28840d8ff71SMatthew G. Knepley 28940d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature 29040d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureGetData() 29140d8ff71SMatthew G. Knepley @*/ 292a6b92713SMatthew G. Knepley PetscErrorCode PetscQuadratureSetData(PetscQuadrature q, PetscInt dim, PetscInt Nc, PetscInt npoints, const PetscReal points[], const PetscReal weights[]) 29321454ff5SMatthew G. Knepley { 29421454ff5SMatthew G. Knepley PetscFunctionBegin; 29521454ff5SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1); 29621454ff5SMatthew G. Knepley if (dim >= 0) q->dim = dim; 297a6b92713SMatthew G. Knepley if (Nc >= 0) q->Nc = Nc; 29821454ff5SMatthew G. Knepley if (npoints >= 0) q->numPoints = npoints; 29921454ff5SMatthew G. Knepley if (points) { 30021454ff5SMatthew G. Knepley PetscValidPointer(points, 4); 30121454ff5SMatthew G. Knepley q->points = points; 30221454ff5SMatthew G. Knepley } 30321454ff5SMatthew G. Knepley if (weights) { 30421454ff5SMatthew G. Knepley PetscValidPointer(weights, 5); 30521454ff5SMatthew G. Knepley q->weights = weights; 30621454ff5SMatthew G. Knepley } 307f9fd7fdbSMatthew G. Knepley PetscFunctionReturn(0); 308f9fd7fdbSMatthew G. Knepley } 309f9fd7fdbSMatthew G. Knepley 31040d8ff71SMatthew G. Knepley /*@C 31140d8ff71SMatthew G. Knepley PetscQuadratureView - Views a PetscQuadrature object 31240d8ff71SMatthew G. Knepley 31340d8ff71SMatthew G. Knepley Collective on PetscQuadrature 31440d8ff71SMatthew G. Knepley 31540d8ff71SMatthew G. Knepley Input Parameters: 31640d8ff71SMatthew G. Knepley + q - The PetscQuadrature object 31740d8ff71SMatthew G. Knepley - viewer - The PetscViewer object 31840d8ff71SMatthew G. Knepley 31940d8ff71SMatthew G. Knepley Level: beginner 32040d8ff71SMatthew G. Knepley 32140d8ff71SMatthew G. Knepley .keywords: PetscQuadrature, quadrature, view 32240d8ff71SMatthew G. Knepley .seealso: PetscQuadratureCreate(), PetscQuadratureGetData() 32340d8ff71SMatthew G. Knepley @*/ 324f9fd7fdbSMatthew G. Knepley PetscErrorCode PetscQuadratureView(PetscQuadrature quad, PetscViewer viewer) 325f9fd7fdbSMatthew G. Knepley { 326a6b92713SMatthew G. Knepley PetscInt q, d, c; 327f9fd7fdbSMatthew G. Knepley PetscErrorCode ierr; 328f9fd7fdbSMatthew G. Knepley 329f9fd7fdbSMatthew G. Knepley PetscFunctionBegin; 33098c3331eSBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)quad,viewer);CHKERRQ(ierr); 331a6b92713SMatthew G. Knepley if (quad->Nc > 1) {ierr = PetscViewerASCIIPrintf(viewer, "Quadrature on %D points with %D components\n (", quad->numPoints, quad->Nc);CHKERRQ(ierr);} 332a6b92713SMatthew G. Knepley else {ierr = PetscViewerASCIIPrintf(viewer, "Quadrature on %D points\n (", quad->numPoints);CHKERRQ(ierr);} 33321454ff5SMatthew G. Knepley for (q = 0; q < quad->numPoints; ++q) { 33421454ff5SMatthew G. Knepley for (d = 0; d < quad->dim; ++d) { 335f9fd7fdbSMatthew G. Knepley if (d) ierr = PetscViewerASCIIPrintf(viewer, ", ");CHKERRQ(ierr); 336ab15ae43SMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer, "%g\n", (double)quad->points[q*quad->dim+d]);CHKERRQ(ierr); 337f9fd7fdbSMatthew G. Knepley } 338a6b92713SMatthew G. Knepley if (quad->Nc > 1) { 339a6b92713SMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer, ") (");CHKERRQ(ierr); 340a6b92713SMatthew G. Knepley for (c = 0; c < quad->Nc; ++c) { 341a6b92713SMatthew G. Knepley if (c) ierr = PetscViewerASCIIPrintf(viewer, ", ");CHKERRQ(ierr); 342a6b92713SMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer, "%g", (double)quad->weights[q*quad->Nc+c]);CHKERRQ(ierr); 343a6b92713SMatthew G. Knepley } 344a6b92713SMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer, ")\n");CHKERRQ(ierr); 345a6b92713SMatthew G. Knepley } else { 346ab15ae43SMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer, ") %g\n", (double)quad->weights[q]);CHKERRQ(ierr); 347f9fd7fdbSMatthew G. Knepley } 348a6b92713SMatthew G. Knepley } 349bfa639d9SMatthew G. Knepley PetscFunctionReturn(0); 350bfa639d9SMatthew G. Knepley } 351bfa639d9SMatthew G. Knepley 35289710940SMatthew G. Knepley /*@C 35389710940SMatthew G. Knepley PetscQuadratureExpandComposite - Return a quadrature over the composite element, which has the original quadrature in each subelement 35489710940SMatthew G. Knepley 35589710940SMatthew G. Knepley Not collective 35689710940SMatthew G. Knepley 35789710940SMatthew G. Knepley Input Parameter: 35889710940SMatthew G. Knepley + q - The original PetscQuadrature 35989710940SMatthew G. Knepley . numSubelements - The number of subelements the original element is divided into 36089710940SMatthew G. Knepley . v0 - An array of the initial points for each subelement 36189710940SMatthew G. Knepley - jac - An array of the Jacobian mappings from the reference to each subelement 36289710940SMatthew G. Knepley 36389710940SMatthew G. Knepley Output Parameters: 36489710940SMatthew G. Knepley . dim - The dimension 36589710940SMatthew G. Knepley 36689710940SMatthew G. Knepley Note: Together v0 and jac define an affine mapping from the original reference element to each subelement 36789710940SMatthew G. Knepley 368*f5f57ec0SBarry Smith Not available from Fortran 369*f5f57ec0SBarry Smith 37089710940SMatthew G. Knepley Level: intermediate 37189710940SMatthew G. Knepley 37289710940SMatthew G. Knepley .seealso: PetscFECreate(), PetscSpaceGetDimension(), PetscDualSpaceGetDimension() 37389710940SMatthew G. Knepley @*/ 37489710940SMatthew G. Knepley PetscErrorCode PetscQuadratureExpandComposite(PetscQuadrature q, PetscInt numSubelements, const PetscReal v0[], const PetscReal jac[], PetscQuadrature *qref) 37589710940SMatthew G. Knepley { 37689710940SMatthew G. Knepley const PetscReal *points, *weights; 37789710940SMatthew G. Knepley PetscReal *pointsRef, *weightsRef; 378a6b92713SMatthew G. Knepley PetscInt dim, Nc, order, npoints, npointsRef, c, p, cp, d, e; 37989710940SMatthew G. Knepley PetscErrorCode ierr; 38089710940SMatthew G. Knepley 38189710940SMatthew G. Knepley PetscFunctionBegin; 38289710940SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSC_OBJECT_CLASSID, 1); 38389710940SMatthew G. Knepley PetscValidPointer(v0, 3); 38489710940SMatthew G. Knepley PetscValidPointer(jac, 4); 38589710940SMatthew G. Knepley PetscValidPointer(qref, 5); 38689710940SMatthew G. Knepley ierr = PetscQuadratureCreate(PETSC_COMM_SELF, qref);CHKERRQ(ierr); 38789710940SMatthew G. Knepley ierr = PetscQuadratureGetOrder(q, &order);CHKERRQ(ierr); 388a6b92713SMatthew G. Knepley ierr = PetscQuadratureGetData(q, &dim, &Nc, &npoints, &points, &weights);CHKERRQ(ierr); 38989710940SMatthew G. Knepley npointsRef = npoints*numSubelements; 39089710940SMatthew G. Knepley ierr = PetscMalloc1(npointsRef*dim,&pointsRef);CHKERRQ(ierr); 391a6b92713SMatthew G. Knepley ierr = PetscMalloc1(npointsRef*Nc, &weightsRef);CHKERRQ(ierr); 39289710940SMatthew G. Knepley for (c = 0; c < numSubelements; ++c) { 39389710940SMatthew G. Knepley for (p = 0; p < npoints; ++p) { 39489710940SMatthew G. Knepley for (d = 0; d < dim; ++d) { 39589710940SMatthew G. Knepley pointsRef[(c*npoints + p)*dim+d] = v0[c*dim+d]; 39689710940SMatthew G. Knepley for (e = 0; e < dim; ++e) { 39789710940SMatthew G. Knepley pointsRef[(c*npoints + p)*dim+d] += jac[(c*dim + d)*dim+e]*(points[p*dim+e] + 1.0); 39889710940SMatthew G. Knepley } 39989710940SMatthew G. Knepley } 40089710940SMatthew G. Knepley /* Could also use detJ here */ 401a6b92713SMatthew G. Knepley for (cp = 0; cp < Nc; ++cp) weightsRef[(c*npoints+p)*Nc+cp] = weights[p*Nc+cp]/numSubelements; 40289710940SMatthew G. Knepley } 40389710940SMatthew G. Knepley } 40489710940SMatthew G. Knepley ierr = PetscQuadratureSetOrder(*qref, order);CHKERRQ(ierr); 405a6b92713SMatthew G. Knepley ierr = PetscQuadratureSetData(*qref, dim, Nc, npointsRef, pointsRef, weightsRef);CHKERRQ(ierr); 40689710940SMatthew G. Knepley PetscFunctionReturn(0); 40789710940SMatthew G. Knepley } 40889710940SMatthew G. Knepley 40937045ce4SJed Brown /*@ 41037045ce4SJed Brown PetscDTLegendreEval - evaluate Legendre polynomial at points 41137045ce4SJed Brown 41237045ce4SJed Brown Not Collective 41337045ce4SJed Brown 41437045ce4SJed Brown Input Arguments: 41537045ce4SJed Brown + npoints - number of spatial points to evaluate at 41637045ce4SJed Brown . points - array of locations to evaluate at 41737045ce4SJed Brown . ndegree - number of basis degrees to evaluate 41837045ce4SJed Brown - degrees - sorted array of degrees to evaluate 41937045ce4SJed Brown 42037045ce4SJed Brown Output Arguments: 4210298fd71SBarry Smith + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or NULL) 4220298fd71SBarry Smith . D - row-oriented derivative evaluation matrix (or NULL) 4230298fd71SBarry Smith - D2 - row-oriented second derivative evaluation matrix (or NULL) 42437045ce4SJed Brown 42537045ce4SJed Brown Level: intermediate 42637045ce4SJed Brown 42737045ce4SJed Brown .seealso: PetscDTGaussQuadrature() 42837045ce4SJed Brown @*/ 42937045ce4SJed Brown PetscErrorCode PetscDTLegendreEval(PetscInt npoints,const PetscReal *points,PetscInt ndegree,const PetscInt *degrees,PetscReal *B,PetscReal *D,PetscReal *D2) 43037045ce4SJed Brown { 43137045ce4SJed Brown PetscInt i,maxdegree; 43237045ce4SJed Brown 43337045ce4SJed Brown PetscFunctionBegin; 43437045ce4SJed Brown if (!npoints || !ndegree) PetscFunctionReturn(0); 43537045ce4SJed Brown maxdegree = degrees[ndegree-1]; 43637045ce4SJed Brown for (i=0; i<npoints; i++) { 43737045ce4SJed Brown PetscReal pm1,pm2,pd1,pd2,pdd1,pdd2,x; 43837045ce4SJed Brown PetscInt j,k; 43937045ce4SJed Brown x = points[i]; 44037045ce4SJed Brown pm2 = 0; 44137045ce4SJed Brown pm1 = 1; 44237045ce4SJed Brown pd2 = 0; 44337045ce4SJed Brown pd1 = 0; 44437045ce4SJed Brown pdd2 = 0; 44537045ce4SJed Brown pdd1 = 0; 44637045ce4SJed Brown k = 0; 44737045ce4SJed Brown if (degrees[k] == 0) { 44837045ce4SJed Brown if (B) B[i*ndegree+k] = pm1; 44937045ce4SJed Brown if (D) D[i*ndegree+k] = pd1; 45037045ce4SJed Brown if (D2) D2[i*ndegree+k] = pdd1; 45137045ce4SJed Brown k++; 45237045ce4SJed Brown } 45337045ce4SJed Brown for (j=1; j<=maxdegree; j++,k++) { 45437045ce4SJed Brown PetscReal p,d,dd; 45537045ce4SJed Brown p = ((2*j-1)*x*pm1 - (j-1)*pm2)/j; 45637045ce4SJed Brown d = pd2 + (2*j-1)*pm1; 45737045ce4SJed Brown dd = pdd2 + (2*j-1)*pd1; 45837045ce4SJed Brown pm2 = pm1; 45937045ce4SJed Brown pm1 = p; 46037045ce4SJed Brown pd2 = pd1; 46137045ce4SJed Brown pd1 = d; 46237045ce4SJed Brown pdd2 = pdd1; 46337045ce4SJed Brown pdd1 = dd; 46437045ce4SJed Brown if (degrees[k] == j) { 46537045ce4SJed Brown if (B) B[i*ndegree+k] = p; 46637045ce4SJed Brown if (D) D[i*ndegree+k] = d; 46737045ce4SJed Brown if (D2) D2[i*ndegree+k] = dd; 46837045ce4SJed Brown } 46937045ce4SJed Brown } 47037045ce4SJed Brown } 47137045ce4SJed Brown PetscFunctionReturn(0); 47237045ce4SJed Brown } 47337045ce4SJed Brown 47437045ce4SJed Brown /*@ 47537045ce4SJed Brown PetscDTGaussQuadrature - create Gauss quadrature 47637045ce4SJed Brown 47737045ce4SJed Brown Not Collective 47837045ce4SJed Brown 47937045ce4SJed Brown Input Arguments: 48037045ce4SJed Brown + npoints - number of points 48137045ce4SJed Brown . a - left end of interval (often-1) 48237045ce4SJed Brown - b - right end of interval (often +1) 48337045ce4SJed Brown 48437045ce4SJed Brown Output Arguments: 48537045ce4SJed Brown + x - quadrature points 48637045ce4SJed Brown - w - quadrature weights 48737045ce4SJed Brown 48837045ce4SJed Brown Level: intermediate 48937045ce4SJed Brown 49037045ce4SJed Brown References: 49196a0c994SBarry Smith . 1. - Golub and Welsch, Calculation of Quadrature Rules, Math. Comp. 23(106), 1969. 49237045ce4SJed Brown 49337045ce4SJed Brown .seealso: PetscDTLegendreEval() 49437045ce4SJed Brown @*/ 49537045ce4SJed Brown PetscErrorCode PetscDTGaussQuadrature(PetscInt npoints,PetscReal a,PetscReal b,PetscReal *x,PetscReal *w) 49637045ce4SJed Brown { 49737045ce4SJed Brown PetscErrorCode ierr; 49837045ce4SJed Brown PetscInt i; 49937045ce4SJed Brown PetscReal *work; 50037045ce4SJed Brown PetscScalar *Z; 50137045ce4SJed Brown PetscBLASInt N,LDZ,info; 50237045ce4SJed Brown 50337045ce4SJed Brown PetscFunctionBegin; 5040bfcf5a5SMatthew G. Knepley ierr = PetscCitationsRegister(GaussCitation, &GaussCite);CHKERRQ(ierr); 50537045ce4SJed Brown /* Set up the Golub-Welsch system */ 50637045ce4SJed Brown for (i=0; i<npoints; i++) { 50737045ce4SJed Brown x[i] = 0; /* diagonal is 0 */ 50837045ce4SJed Brown if (i) w[i-1] = 0.5 / PetscSqrtReal(1 - 1./PetscSqr(2*i)); 50937045ce4SJed Brown } 510dcca6d9dSJed Brown ierr = PetscMalloc2(npoints*npoints,&Z,PetscMax(1,2*npoints-2),&work);CHKERRQ(ierr); 511c5df96a5SBarry Smith ierr = PetscBLASIntCast(npoints,&N);CHKERRQ(ierr); 51237045ce4SJed Brown LDZ = N; 51337045ce4SJed Brown ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 5148b83055fSJed Brown PetscStackCallBLAS("LAPACKsteqr",LAPACKsteqr_("I",&N,x,w,Z,&LDZ,work,&info)); 51537045ce4SJed Brown ierr = PetscFPTrapPop();CHKERRQ(ierr); 5161c3d6f74SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"xSTEQR error"); 51737045ce4SJed Brown 51837045ce4SJed Brown for (i=0; i<(npoints+1)/2; i++) { 51937045ce4SJed Brown PetscReal y = 0.5 * (-x[i] + x[npoints-i-1]); /* enforces symmetry */ 52037045ce4SJed Brown x[i] = (a+b)/2 - y*(b-a)/2; 52119a57d60SBarry Smith if (x[i] == -0.0) x[i] = 0.0; 52237045ce4SJed Brown x[npoints-i-1] = (a+b)/2 + y*(b-a)/2; 5230d644c17SKarl Rupp 52488393a60SJed Brown w[i] = w[npoints-1-i] = 0.5*(b-a)*(PetscSqr(PetscAbsScalar(Z[i*npoints])) + PetscSqr(PetscAbsScalar(Z[(npoints-i-1)*npoints]))); 52537045ce4SJed Brown } 52637045ce4SJed Brown ierr = PetscFree2(Z,work);CHKERRQ(ierr); 52737045ce4SJed Brown PetscFunctionReturn(0); 52837045ce4SJed Brown } 529194825f6SJed Brown 530744bafbcSMatthew G. Knepley /*@ 531744bafbcSMatthew G. Knepley PetscDTGaussTensorQuadrature - creates a tensor-product Gauss quadrature 532744bafbcSMatthew G. Knepley 533744bafbcSMatthew G. Knepley Not Collective 534744bafbcSMatthew G. Knepley 535744bafbcSMatthew G. Knepley Input Arguments: 536744bafbcSMatthew G. Knepley + dim - The spatial dimension 537a6b92713SMatthew G. Knepley . Nc - The number of components 538744bafbcSMatthew G. Knepley . npoints - number of points in one dimension 539744bafbcSMatthew G. Knepley . a - left end of interval (often-1) 540744bafbcSMatthew G. Knepley - b - right end of interval (often +1) 541744bafbcSMatthew G. Knepley 542744bafbcSMatthew G. Knepley Output Argument: 543744bafbcSMatthew G. Knepley . q - A PetscQuadrature object 544744bafbcSMatthew G. Knepley 545744bafbcSMatthew G. Knepley Level: intermediate 546744bafbcSMatthew G. Knepley 547744bafbcSMatthew G. Knepley .seealso: PetscDTGaussQuadrature(), PetscDTLegendreEval() 548744bafbcSMatthew G. Knepley @*/ 549a6b92713SMatthew G. Knepley PetscErrorCode PetscDTGaussTensorQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 550744bafbcSMatthew G. Knepley { 551a6b92713SMatthew G. Knepley PetscInt totpoints = dim > 1 ? dim > 2 ? npoints*PetscSqr(npoints) : PetscSqr(npoints) : npoints, i, j, k, c; 552744bafbcSMatthew G. Knepley PetscReal *x, *w, *xw, *ww; 553744bafbcSMatthew G. Knepley PetscErrorCode ierr; 554744bafbcSMatthew G. Knepley 555744bafbcSMatthew G. Knepley PetscFunctionBegin; 556744bafbcSMatthew G. Knepley ierr = PetscMalloc1(totpoints*dim,&x);CHKERRQ(ierr); 557a6b92713SMatthew G. Knepley ierr = PetscMalloc1(totpoints*Nc,&w);CHKERRQ(ierr); 558744bafbcSMatthew G. Knepley /* Set up the Golub-Welsch system */ 559744bafbcSMatthew G. Knepley switch (dim) { 560744bafbcSMatthew G. Knepley case 0: 561744bafbcSMatthew G. Knepley ierr = PetscFree(x);CHKERRQ(ierr); 562744bafbcSMatthew G. Knepley ierr = PetscFree(w);CHKERRQ(ierr); 563744bafbcSMatthew G. Knepley ierr = PetscMalloc1(1, &x);CHKERRQ(ierr); 564a6b92713SMatthew G. Knepley ierr = PetscMalloc1(Nc, &w);CHKERRQ(ierr); 565744bafbcSMatthew G. Knepley x[0] = 0.0; 566a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[c] = 1.0; 567744bafbcSMatthew G. Knepley break; 568744bafbcSMatthew G. Knepley case 1: 569a6b92713SMatthew G. Knepley ierr = PetscMalloc1(npoints,&ww);CHKERRQ(ierr); 570a6b92713SMatthew G. Knepley ierr = PetscDTGaussQuadrature(npoints, a, b, x, ww);CHKERRQ(ierr); 571a6b92713SMatthew G. Knepley for (i = 0; i < npoints; ++i) for (c = 0; c < Nc; ++c) w[i*Nc+c] = ww[i]; 572a6b92713SMatthew G. Knepley ierr = PetscFree(ww);CHKERRQ(ierr); 573744bafbcSMatthew G. Knepley break; 574744bafbcSMatthew G. Knepley case 2: 575744bafbcSMatthew G. Knepley ierr = PetscMalloc2(npoints,&xw,npoints,&ww);CHKERRQ(ierr); 576744bafbcSMatthew G. Knepley ierr = PetscDTGaussQuadrature(npoints, a, b, xw, ww);CHKERRQ(ierr); 577744bafbcSMatthew G. Knepley for (i = 0; i < npoints; ++i) { 578744bafbcSMatthew G. Knepley for (j = 0; j < npoints; ++j) { 579744bafbcSMatthew G. Knepley x[(i*npoints+j)*dim+0] = xw[i]; 580744bafbcSMatthew G. Knepley x[(i*npoints+j)*dim+1] = xw[j]; 581a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[(i*npoints+j)*Nc+c] = ww[i] * ww[j]; 582744bafbcSMatthew G. Knepley } 583744bafbcSMatthew G. Knepley } 584744bafbcSMatthew G. Knepley ierr = PetscFree2(xw,ww);CHKERRQ(ierr); 585744bafbcSMatthew G. Knepley break; 586744bafbcSMatthew G. Knepley case 3: 587744bafbcSMatthew G. Knepley ierr = PetscMalloc2(npoints,&xw,npoints,&ww);CHKERRQ(ierr); 588744bafbcSMatthew G. Knepley ierr = PetscDTGaussQuadrature(npoints, a, b, xw, ww);CHKERRQ(ierr); 589744bafbcSMatthew G. Knepley for (i = 0; i < npoints; ++i) { 590744bafbcSMatthew G. Knepley for (j = 0; j < npoints; ++j) { 591744bafbcSMatthew G. Knepley for (k = 0; k < npoints; ++k) { 592744bafbcSMatthew G. Knepley x[((i*npoints+j)*npoints+k)*dim+0] = xw[i]; 593744bafbcSMatthew G. Knepley x[((i*npoints+j)*npoints+k)*dim+1] = xw[j]; 594744bafbcSMatthew G. Knepley x[((i*npoints+j)*npoints+k)*dim+2] = xw[k]; 595a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[((i*npoints+j)*npoints+k)*Nc+c] = ww[i] * ww[j] * ww[k]; 596744bafbcSMatthew G. Knepley } 597744bafbcSMatthew G. Knepley } 598744bafbcSMatthew G. Knepley } 599744bafbcSMatthew G. Knepley ierr = PetscFree2(xw,ww);CHKERRQ(ierr); 600744bafbcSMatthew G. Knepley break; 601744bafbcSMatthew G. Knepley default: 602744bafbcSMatthew G. Knepley SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim); 603744bafbcSMatthew G. Knepley } 604744bafbcSMatthew G. Knepley ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr); 605dcce0ee2SMatthew G. Knepley ierr = PetscQuadratureSetOrder(*q, npoints-1);CHKERRQ(ierr); 606a6b92713SMatthew G. Knepley ierr = PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w);CHKERRQ(ierr); 607744bafbcSMatthew G. Knepley PetscFunctionReturn(0); 608744bafbcSMatthew G. Knepley } 609744bafbcSMatthew G. Knepley 610494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 611494e7359SMatthew G. Knepley Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 612494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTFactorial_Internal(PetscInt n, PetscReal *factorial) 613494e7359SMatthew G. Knepley { 614494e7359SMatthew G. Knepley PetscReal f = 1.0; 615494e7359SMatthew G. Knepley PetscInt i; 616494e7359SMatthew G. Knepley 617494e7359SMatthew G. Knepley PetscFunctionBegin; 618494e7359SMatthew G. Knepley for (i = 1; i < n+1; ++i) f *= i; 619494e7359SMatthew G. Knepley *factorial = f; 620494e7359SMatthew G. Knepley PetscFunctionReturn(0); 621494e7359SMatthew G. Knepley } 622494e7359SMatthew G. Knepley 623494e7359SMatthew G. Knepley /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 624494e7359SMatthew G. Knepley Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 625494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 626494e7359SMatthew G. Knepley { 627494e7359SMatthew G. Knepley PetscReal apb, pn1, pn2; 628494e7359SMatthew G. Knepley PetscInt k; 629494e7359SMatthew G. Knepley 630494e7359SMatthew G. Knepley PetscFunctionBegin; 631494e7359SMatthew G. Knepley if (!n) {*P = 1.0; PetscFunctionReturn(0);} 632494e7359SMatthew G. Knepley if (n == 1) {*P = 0.5 * (a - b + (a + b + 2.0) * x); PetscFunctionReturn(0);} 633494e7359SMatthew G. Knepley apb = a + b; 634494e7359SMatthew G. Knepley pn2 = 1.0; 635494e7359SMatthew G. Knepley pn1 = 0.5 * (a - b + (apb + 2.0) * x); 636494e7359SMatthew G. Knepley *P = 0.0; 637494e7359SMatthew G. Knepley for (k = 2; k < n+1; ++k) { 638494e7359SMatthew G. Knepley PetscReal a1 = 2.0 * k * (k + apb) * (2.0*k + apb - 2.0); 639494e7359SMatthew G. Knepley PetscReal a2 = (2.0 * k + apb - 1.0) * (a*a - b*b); 640494e7359SMatthew G. Knepley PetscReal a3 = (2.0 * k + apb - 2.0) * (2.0 * k + apb - 1.0) * (2.0 * k + apb); 641494e7359SMatthew G. Knepley PetscReal a4 = 2.0 * (k + a - 1.0) * (k + b - 1.0) * (2.0 * k + apb); 642494e7359SMatthew G. Knepley 643494e7359SMatthew G. Knepley a2 = a2 / a1; 644494e7359SMatthew G. Knepley a3 = a3 / a1; 645494e7359SMatthew G. Knepley a4 = a4 / a1; 646494e7359SMatthew G. Knepley *P = (a2 + a3 * x) * pn1 - a4 * pn2; 647494e7359SMatthew G. Knepley pn2 = pn1; 648494e7359SMatthew G. Knepley pn1 = *P; 649494e7359SMatthew G. Knepley } 650494e7359SMatthew G. Knepley PetscFunctionReturn(0); 651494e7359SMatthew G. Knepley } 652494e7359SMatthew G. Knepley 653494e7359SMatthew G. Knepley /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */ 654494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 655494e7359SMatthew G. Knepley { 656494e7359SMatthew G. Knepley PetscReal nP; 657494e7359SMatthew G. Knepley PetscErrorCode ierr; 658494e7359SMatthew G. Knepley 659494e7359SMatthew G. Knepley PetscFunctionBegin; 660494e7359SMatthew G. Knepley if (!n) {*P = 0.0; PetscFunctionReturn(0);} 661494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobi(a+1, b+1, n-1, x, &nP);CHKERRQ(ierr); 662494e7359SMatthew G. Knepley *P = 0.5 * (a + b + n + 1) * nP; 663494e7359SMatthew G. Knepley PetscFunctionReturn(0); 664494e7359SMatthew G. Knepley } 665494e7359SMatthew G. Knepley 666494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */ 667494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapSquareToTriangle_Internal(PetscReal x, PetscReal y, PetscReal *xi, PetscReal *eta) 668494e7359SMatthew G. Knepley { 669494e7359SMatthew G. Knepley PetscFunctionBegin; 670494e7359SMatthew G. Knepley *xi = 0.5 * (1.0 + x) * (1.0 - y) - 1.0; 671494e7359SMatthew G. Knepley *eta = y; 672494e7359SMatthew G. Knepley PetscFunctionReturn(0); 673494e7359SMatthew G. Knepley } 674494e7359SMatthew G. Knepley 675494e7359SMatthew G. Knepley /* Maps from [-1,1]^2 to the (-1,1) reference triangle */ 676494e7359SMatthew G. Knepley PETSC_STATIC_INLINE PetscErrorCode PetscDTMapCubeToTetrahedron_Internal(PetscReal x, PetscReal y, PetscReal z, PetscReal *xi, PetscReal *eta, PetscReal *zeta) 677494e7359SMatthew G. Knepley { 678494e7359SMatthew G. Knepley PetscFunctionBegin; 679494e7359SMatthew G. Knepley *xi = 0.25 * (1.0 + x) * (1.0 - y) * (1.0 - z) - 1.0; 680494e7359SMatthew G. Knepley *eta = 0.5 * (1.0 + y) * (1.0 - z) - 1.0; 681494e7359SMatthew G. Knepley *zeta = z; 682494e7359SMatthew G. Knepley PetscFunctionReturn(0); 683494e7359SMatthew G. Knepley } 684494e7359SMatthew G. Knepley 685494e7359SMatthew G. Knepley static PetscErrorCode PetscDTGaussJacobiQuadrature1D_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 686494e7359SMatthew G. Knepley { 687494e7359SMatthew G. Knepley PetscInt maxIter = 100; 688494e7359SMatthew G. Knepley PetscReal eps = 1.0e-8; 689a8291ba1SSatish Balay PetscReal a1, a2, a3, a4, a5, a6; 690494e7359SMatthew G. Knepley PetscInt k; 691494e7359SMatthew G. Knepley PetscErrorCode ierr; 692494e7359SMatthew G. Knepley 693494e7359SMatthew G. Knepley PetscFunctionBegin; 694a8291ba1SSatish Balay 6958b49ba18SBarry Smith a1 = PetscPowReal(2.0, a+b+1); 696a8291ba1SSatish Balay #if defined(PETSC_HAVE_TGAMMA) 6970646a658SBarry Smith a2 = PetscTGamma(a + npoints + 1); 6980646a658SBarry Smith a3 = PetscTGamma(b + npoints + 1); 6990646a658SBarry Smith a4 = PetscTGamma(a + b + npoints + 1); 700a8291ba1SSatish Balay #else 70129bcbfd0SToby Isaac { 702d24bbb91SToby Isaac PetscInt ia, ib; 70329bcbfd0SToby Isaac 704d24bbb91SToby Isaac ia = (PetscInt) a; 705d24bbb91SToby Isaac ib = (PetscInt) b; 706d24bbb91SToby 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 */ 707d24bbb91SToby Isaac ierr = PetscDTFactorial_Internal(ia + npoints, &a2);CHKERRQ(ierr); 708d24bbb91SToby Isaac ierr = PetscDTFactorial_Internal(ib + npoints, &a3);CHKERRQ(ierr); 709d24bbb91SToby Isaac ierr = PetscDTFactorial_Internal(ia + ib + npoints, &a4);CHKERRQ(ierr); 71029bcbfd0SToby Isaac } else { 711a8291ba1SSatish Balay SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"tgamma() - math routine is unavailable."); 71229bcbfd0SToby Isaac } 71329bcbfd0SToby Isaac } 714a8291ba1SSatish Balay #endif 715a8291ba1SSatish Balay 716494e7359SMatthew G. Knepley ierr = PetscDTFactorial_Internal(npoints, &a5);CHKERRQ(ierr); 717494e7359SMatthew G. Knepley a6 = a1 * a2 * a3 / a4 / a5; 718494e7359SMatthew G. Knepley /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses. 719494e7359SMatthew G. Knepley Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */ 720494e7359SMatthew G. Knepley for (k = 0; k < npoints; ++k) { 7218b49ba18SBarry Smith PetscReal r = -PetscCosReal((2.0*k + 1.0) * PETSC_PI / (2.0 * npoints)), dP; 722494e7359SMatthew G. Knepley PetscInt j; 723494e7359SMatthew G. Knepley 724494e7359SMatthew G. Knepley if (k > 0) r = 0.5 * (r + x[k-1]); 725494e7359SMatthew G. Knepley for (j = 0; j < maxIter; ++j) { 726494e7359SMatthew G. Knepley PetscReal s = 0.0, delta, f, fp; 727494e7359SMatthew G. Knepley PetscInt i; 728494e7359SMatthew G. Knepley 729494e7359SMatthew G. Knepley for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]); 730494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobi(a, b, npoints, r, &f);CHKERRQ(ierr); 731494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobiDerivative(a, b, npoints, r, &fp);CHKERRQ(ierr); 732494e7359SMatthew G. Knepley delta = f / (fp - f * s); 733494e7359SMatthew G. Knepley r = r - delta; 73477b4d14cSPeter Brune if (PetscAbsReal(delta) < eps) break; 735494e7359SMatthew G. Knepley } 736494e7359SMatthew G. Knepley x[k] = r; 737494e7359SMatthew G. Knepley ierr = PetscDTComputeJacobiDerivative(a, b, npoints, x[k], &dP);CHKERRQ(ierr); 738494e7359SMatthew G. Knepley w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP); 739494e7359SMatthew G. Knepley } 740494e7359SMatthew G. Knepley PetscFunctionReturn(0); 741494e7359SMatthew G. Knepley } 742494e7359SMatthew G. Knepley 743*f5f57ec0SBarry Smith /*@ 744494e7359SMatthew G. Knepley PetscDTGaussJacobiQuadrature - create Gauss-Jacobi quadrature for a simplex 745494e7359SMatthew G. Knepley 746494e7359SMatthew G. Knepley Not Collective 747494e7359SMatthew G. Knepley 748494e7359SMatthew G. Knepley Input Arguments: 749494e7359SMatthew G. Knepley + dim - The simplex dimension 750a6b92713SMatthew G. Knepley . Nc - The number of components 751dcce0ee2SMatthew G. Knepley . npoints - The number of points in one dimension 752494e7359SMatthew G. Knepley . a - left end of interval (often-1) 753494e7359SMatthew G. Knepley - b - right end of interval (often +1) 754494e7359SMatthew G. Knepley 755744bafbcSMatthew G. Knepley Output Argument: 756552aa4f7SMatthew G. Knepley . q - A PetscQuadrature object 757494e7359SMatthew G. Knepley 758494e7359SMatthew G. Knepley Level: intermediate 759494e7359SMatthew G. Knepley 760494e7359SMatthew G. Knepley References: 76196a0c994SBarry Smith . 1. - Karniadakis and Sherwin. FIAT 762494e7359SMatthew G. Knepley 763744bafbcSMatthew G. Knepley .seealso: PetscDTGaussTensorQuadrature(), PetscDTGaussQuadrature() 764494e7359SMatthew G. Knepley @*/ 765dcce0ee2SMatthew G. Knepley PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 766494e7359SMatthew G. Knepley { 767dcce0ee2SMatthew G. Knepley PetscInt totpoints = dim > 1 ? dim > 2 ? npoints*PetscSqr(npoints) : PetscSqr(npoints) : npoints; 768494e7359SMatthew G. Knepley PetscReal *px, *wx, *py, *wy, *pz, *wz, *x, *w; 769a6b92713SMatthew G. Knepley PetscInt i, j, k, c; 770494e7359SMatthew G. Knepley PetscErrorCode ierr; 771494e7359SMatthew G. Knepley 772494e7359SMatthew G. Knepley PetscFunctionBegin; 773494e7359SMatthew G. Knepley if ((a != -1.0) || (b != 1.0)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now"); 774dcce0ee2SMatthew G. Knepley ierr = PetscMalloc1(totpoints*dim, &x);CHKERRQ(ierr); 775dcce0ee2SMatthew G. Knepley ierr = PetscMalloc1(totpoints*Nc, &w);CHKERRQ(ierr); 776494e7359SMatthew G. Knepley switch (dim) { 777707aa5c5SMatthew G. Knepley case 0: 778707aa5c5SMatthew G. Knepley ierr = PetscFree(x);CHKERRQ(ierr); 779707aa5c5SMatthew G. Knepley ierr = PetscFree(w);CHKERRQ(ierr); 780785e854fSJed Brown ierr = PetscMalloc1(1, &x);CHKERRQ(ierr); 781a6b92713SMatthew G. Knepley ierr = PetscMalloc1(Nc, &w);CHKERRQ(ierr); 782707aa5c5SMatthew G. Knepley x[0] = 0.0; 783a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[c] = 1.0; 784707aa5c5SMatthew G. Knepley break; 785494e7359SMatthew G. Knepley case 1: 786dcce0ee2SMatthew G. Knepley ierr = PetscMalloc1(npoints,&wx);CHKERRQ(ierr); 787dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, x, wx);CHKERRQ(ierr); 788dcce0ee2SMatthew G. Knepley for (i = 0; i < npoints; ++i) for (c = 0; c < Nc; ++c) w[i*Nc+c] = wx[i]; 789a6b92713SMatthew G. Knepley ierr = PetscFree(wx);CHKERRQ(ierr); 790494e7359SMatthew G. Knepley break; 791494e7359SMatthew G. Knepley case 2: 792dcce0ee2SMatthew G. Knepley ierr = PetscMalloc4(npoints,&px,npoints,&wx,npoints,&py,npoints,&wy);CHKERRQ(ierr); 793dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr); 794dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr); 795dcce0ee2SMatthew G. Knepley for (i = 0; i < npoints; ++i) { 796dcce0ee2SMatthew G. Knepley for (j = 0; j < npoints; ++j) { 797dcce0ee2SMatthew G. Knepley ierr = PetscDTMapSquareToTriangle_Internal(px[i], py[j], &x[(i*npoints+j)*2+0], &x[(i*npoints+j)*2+1]);CHKERRQ(ierr); 798dcce0ee2SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[(i*npoints+j)*Nc+c] = 0.5 * wx[i] * wy[j]; 799494e7359SMatthew G. Knepley } 800494e7359SMatthew G. Knepley } 801494e7359SMatthew G. Knepley ierr = PetscFree4(px,wx,py,wy);CHKERRQ(ierr); 802494e7359SMatthew G. Knepley break; 803494e7359SMatthew G. Knepley case 3: 804dcce0ee2SMatthew G. Knepley ierr = PetscMalloc6(npoints,&px,npoints,&wx,npoints,&py,npoints,&wy,npoints,&pz,npoints,&wz);CHKERRQ(ierr); 805dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 0.0, 0.0, px, wx);CHKERRQ(ierr); 806dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 1.0, 0.0, py, wy);CHKERRQ(ierr); 807dcce0ee2SMatthew G. Knepley ierr = PetscDTGaussJacobiQuadrature1D_Internal(npoints, 2.0, 0.0, pz, wz);CHKERRQ(ierr); 808dcce0ee2SMatthew G. Knepley for (i = 0; i < npoints; ++i) { 809dcce0ee2SMatthew G. Knepley for (j = 0; j < npoints; ++j) { 810dcce0ee2SMatthew G. Knepley for (k = 0; k < npoints; ++k) { 811dcce0ee2SMatthew 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); 812dcce0ee2SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[((i*npoints+j)*npoints+k)*Nc+c] = 0.125 * wx[i] * wy[j] * wz[k]; 813494e7359SMatthew G. Knepley } 814494e7359SMatthew G. Knepley } 815494e7359SMatthew G. Knepley } 816494e7359SMatthew G. Knepley ierr = PetscFree6(px,wx,py,wy,pz,wz);CHKERRQ(ierr); 817494e7359SMatthew G. Knepley break; 818494e7359SMatthew G. Knepley default: 819494e7359SMatthew G. Knepley SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %d", dim); 820494e7359SMatthew G. Knepley } 82121454ff5SMatthew G. Knepley ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr); 822dcce0ee2SMatthew G. Knepley ierr = PetscQuadratureSetOrder(*q, npoints-1);CHKERRQ(ierr); 823dcce0ee2SMatthew G. Knepley ierr = PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w);CHKERRQ(ierr); 824494e7359SMatthew G. Knepley PetscFunctionReturn(0); 825494e7359SMatthew G. Knepley } 826494e7359SMatthew G. Knepley 827*f5f57ec0SBarry Smith /*@ 828b3c0f97bSTom Klotz PetscDTTanhSinhTensorQuadrature - create tanh-sinh quadrature for a tensor product cell 829b3c0f97bSTom Klotz 830b3c0f97bSTom Klotz Not Collective 831b3c0f97bSTom Klotz 832b3c0f97bSTom Klotz Input Arguments: 833b3c0f97bSTom Klotz + dim - The cell dimension 834b3c0f97bSTom Klotz . level - The number of points in one dimension, 2^l 835b3c0f97bSTom Klotz . a - left end of interval (often-1) 836b3c0f97bSTom Klotz - b - right end of interval (often +1) 837b3c0f97bSTom Klotz 838b3c0f97bSTom Klotz Output Argument: 839b3c0f97bSTom Klotz . q - A PetscQuadrature object 840b3c0f97bSTom Klotz 841b3c0f97bSTom Klotz Level: intermediate 842b3c0f97bSTom Klotz 843b3c0f97bSTom Klotz .seealso: PetscDTGaussTensorQuadrature() 844b3c0f97bSTom Klotz @*/ 845b3c0f97bSTom Klotz PetscErrorCode PetscDTTanhSinhTensorQuadrature(PetscInt dim, PetscInt level, PetscReal a, PetscReal b, PetscQuadrature *q) 846b3c0f97bSTom Klotz { 847b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 848b3c0f97bSTom Klotz const PetscReal alpha = (b-a)/2.; /* Half-width of the integration interval */ 849b3c0f97bSTom Klotz const PetscReal beta = (b+a)/2.; /* Center of the integration interval */ 850b3c0f97bSTom Klotz const PetscReal h = PetscPowReal(2.0, -level); /* Step size, length between x_k */ 851d84b4d08SMatthew G. Knepley PetscReal xk; /* Quadrature point x_k on reference domain [-1, 1] */ 852b3c0f97bSTom Klotz PetscReal wk = 0.5*PETSC_PI; /* Quadrature weight at x_k */ 853b3c0f97bSTom Klotz PetscReal *x, *w; 854b3c0f97bSTom Klotz PetscInt K, k, npoints; 855b3c0f97bSTom Klotz PetscErrorCode ierr; 856b3c0f97bSTom Klotz 857b3c0f97bSTom Klotz PetscFunctionBegin; 858b3c0f97bSTom Klotz if (dim > 1) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %d not yet implemented", dim); 859b3c0f97bSTom Klotz if (!level) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a number of significant digits"); 860b3c0f97bSTom Klotz /* Find K such that the weights are < 32 digits of precision */ 861b3c0f97bSTom Klotz for (K = 1; PetscAbsReal(PetscLog10Real(wk)) < 2*p; ++K) { 8629add2064SThomas Klotz wk = 0.5*h*PETSC_PI*PetscCoshReal(K*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(K*h))); 863b3c0f97bSTom Klotz } 864b3c0f97bSTom Klotz ierr = PetscQuadratureCreate(PETSC_COMM_SELF, q);CHKERRQ(ierr); 865b3c0f97bSTom Klotz ierr = PetscQuadratureSetOrder(*q, 2*K+1);CHKERRQ(ierr); 866b3c0f97bSTom Klotz npoints = 2*K-1; 867b3c0f97bSTom Klotz ierr = PetscMalloc1(npoints*dim, &x);CHKERRQ(ierr); 868b3c0f97bSTom Klotz ierr = PetscMalloc1(npoints, &w);CHKERRQ(ierr); 869b3c0f97bSTom Klotz /* Center term */ 870b3c0f97bSTom Klotz x[0] = beta; 871b3c0f97bSTom Klotz w[0] = 0.5*alpha*PETSC_PI; 872b3c0f97bSTom Klotz for (k = 1; k < K; ++k) { 8739add2064SThomas Klotz wk = 0.5*alpha*h*PETSC_PI*PetscCoshReal(k*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h))); 8741118d4bcSLisandro Dalcin xk = PetscTanhReal(0.5*PETSC_PI*PetscSinhReal(k*h)); 875b3c0f97bSTom Klotz x[2*k-1] = -alpha*xk+beta; 876b3c0f97bSTom Klotz w[2*k-1] = wk; 877b3c0f97bSTom Klotz x[2*k+0] = alpha*xk+beta; 878b3c0f97bSTom Klotz w[2*k+0] = wk; 879b3c0f97bSTom Klotz } 880a6b92713SMatthew G. Knepley ierr = PetscQuadratureSetData(*q, dim, 1, npoints, x, w);CHKERRQ(ierr); 881b3c0f97bSTom Klotz PetscFunctionReturn(0); 882b3c0f97bSTom Klotz } 883b3c0f97bSTom Klotz 884b3c0f97bSTom Klotz PetscErrorCode PetscDTTanhSinhIntegrate(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol) 885b3c0f97bSTom Klotz { 886b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 887b3c0f97bSTom Klotz const PetscReal alpha = (b-a)/2.; /* Half-width of the integration interval */ 888b3c0f97bSTom Klotz const PetscReal beta = (b+a)/2.; /* Center of the integration interval */ 889b3c0f97bSTom Klotz PetscReal h = 1.0; /* Step size, length between x_k */ 890b3c0f97bSTom Klotz PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 891b3c0f97bSTom Klotz PetscReal osum = 0.0; /* Integral on last level */ 892b3c0f97bSTom Klotz PetscReal psum = 0.0; /* Integral on the level before the last level */ 893b3c0f97bSTom Klotz PetscReal sum; /* Integral on current level */ 894446c295cSMatthew G. Knepley PetscReal yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 895b3c0f97bSTom Klotz PetscReal lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 896b3c0f97bSTom Klotz PetscReal wk; /* Quadrature weight at x_k */ 897b3c0f97bSTom Klotz PetscReal lval, rval; /* Terms in the quadature sum to the left and right of 0 */ 898b3c0f97bSTom Klotz PetscInt d; /* Digits of precision in the integral */ 899b3c0f97bSTom Klotz 900b3c0f97bSTom Klotz PetscFunctionBegin; 901b3c0f97bSTom Klotz if (digits <= 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 902b3c0f97bSTom Klotz /* Center term */ 903b3c0f97bSTom Klotz func(beta, &lval); 904b3c0f97bSTom Klotz sum = 0.5*alpha*PETSC_PI*lval; 905b3c0f97bSTom Klotz /* */ 906b3c0f97bSTom Klotz do { 907b3c0f97bSTom Klotz PetscReal lterm, rterm, maxTerm = 0.0, d1, d2, d3, d4; 908b3c0f97bSTom Klotz PetscInt k = 1; 909b3c0f97bSTom Klotz 910b3c0f97bSTom Klotz ++l; 911b3c0f97bSTom Klotz /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %D sum: %15.15f\n", l, sum); */ 912b3c0f97bSTom Klotz /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 913b3c0f97bSTom Klotz psum = osum; 914b3c0f97bSTom Klotz osum = sum; 915b3c0f97bSTom Klotz h *= 0.5; 916b3c0f97bSTom Klotz sum *= 0.5; 917b3c0f97bSTom Klotz do { 9189add2064SThomas Klotz wk = 0.5*h*PETSC_PI*PetscCoshReal(k*h)/PetscSqr(PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h))); 919446c295cSMatthew G. Knepley yk = 1.0/(PetscExpReal(0.5*PETSC_PI*PetscSinhReal(k*h)) * PetscCoshReal(0.5*PETSC_PI*PetscSinhReal(k*h))); 920446c295cSMatthew G. Knepley lx = -alpha*(1.0 - yk)+beta; 921446c295cSMatthew G. Knepley rx = alpha*(1.0 - yk)+beta; 922b3c0f97bSTom Klotz func(lx, &lval); 923b3c0f97bSTom Klotz func(rx, &rval); 924b3c0f97bSTom Klotz lterm = alpha*wk*lval; 925b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(lterm), maxTerm); 926b3c0f97bSTom Klotz sum += lterm; 927b3c0f97bSTom Klotz rterm = alpha*wk*rval; 928b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(rterm), maxTerm); 929b3c0f97bSTom Klotz sum += rterm; 930b3c0f97bSTom Klotz ++k; 931b3c0f97bSTom Klotz /* Only need to evaluate every other point on refined levels */ 932b3c0f97bSTom Klotz if (l != 1) ++k; 9339add2064SThomas Klotz } while (PetscAbsReal(PetscLog10Real(wk)) < p); /* Only need to evaluate sum until weights are < 32 digits of precision */ 934b3c0f97bSTom Klotz 935b3c0f97bSTom Klotz d1 = PetscLog10Real(PetscAbsReal(sum - osum)); 936b3c0f97bSTom Klotz d2 = PetscLog10Real(PetscAbsReal(sum - psum)); 937b3c0f97bSTom Klotz d3 = PetscLog10Real(maxTerm) - p; 93809d48545SBarry Smith if (PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm)) == 0.0) d4 = 0.0; 93909d48545SBarry Smith else d4 = PetscLog10Real(PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm))); 940b3c0f97bSTom Klotz d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1)/d2, 2*d1), d3), d4))); 9419add2064SThomas Klotz } while (d < digits && l < 12); 942b3c0f97bSTom Klotz *sol = sum; 943e510cb1fSThomas Klotz 944b3c0f97bSTom Klotz PetscFunctionReturn(0); 945b3c0f97bSTom Klotz } 946b3c0f97bSTom Klotz 94729f144ccSMatthew G. Knepley #ifdef PETSC_HAVE_MPFR 94829f144ccSMatthew G. Knepley PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol) 94929f144ccSMatthew G. Knepley { 950e510cb1fSThomas Klotz const PetscInt safetyFactor = 2; /* Calculate abcissa until 2*p digits */ 95129f144ccSMatthew G. Knepley PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 95229f144ccSMatthew G. Knepley mpfr_t alpha; /* Half-width of the integration interval */ 95329f144ccSMatthew G. Knepley mpfr_t beta; /* Center of the integration interval */ 95429f144ccSMatthew G. Knepley mpfr_t h; /* Step size, length between x_k */ 95529f144ccSMatthew G. Knepley mpfr_t osum; /* Integral on last level */ 95629f144ccSMatthew G. Knepley mpfr_t psum; /* Integral on the level before the last level */ 95729f144ccSMatthew G. Knepley mpfr_t sum; /* Integral on current level */ 95829f144ccSMatthew G. Knepley mpfr_t yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 95929f144ccSMatthew G. Knepley mpfr_t lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 96029f144ccSMatthew G. Knepley mpfr_t wk; /* Quadrature weight at x_k */ 96129f144ccSMatthew G. Knepley PetscReal lval, rval; /* Terms in the quadature sum to the left and right of 0 */ 96229f144ccSMatthew G. Knepley PetscInt d; /* Digits of precision in the integral */ 96329f144ccSMatthew G. Knepley mpfr_t pi2, kh, msinh, mcosh, maxTerm, curTerm, tmp; 96429f144ccSMatthew G. Knepley 96529f144ccSMatthew G. Knepley PetscFunctionBegin; 96629f144ccSMatthew G. Knepley if (digits <= 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 96729f144ccSMatthew G. Knepley /* Create high precision storage */ 968c9f744b5SMatthew 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); 96929f144ccSMatthew G. Knepley /* Initialization */ 97029f144ccSMatthew G. Knepley mpfr_set_d(alpha, 0.5*(b-a), MPFR_RNDN); 97129f144ccSMatthew G. Knepley mpfr_set_d(beta, 0.5*(b+a), MPFR_RNDN); 97229f144ccSMatthew G. Knepley mpfr_set_d(osum, 0.0, MPFR_RNDN); 97329f144ccSMatthew G. Knepley mpfr_set_d(psum, 0.0, MPFR_RNDN); 97429f144ccSMatthew G. Knepley mpfr_set_d(h, 1.0, MPFR_RNDN); 97529f144ccSMatthew G. Knepley mpfr_const_pi(pi2, MPFR_RNDN); 97629f144ccSMatthew G. Knepley mpfr_mul_d(pi2, pi2, 0.5, MPFR_RNDN); 97729f144ccSMatthew G. Knepley /* Center term */ 97829f144ccSMatthew G. Knepley func(0.5*(b+a), &lval); 97929f144ccSMatthew G. Knepley mpfr_set(sum, pi2, MPFR_RNDN); 98029f144ccSMatthew G. Knepley mpfr_mul(sum, sum, alpha, MPFR_RNDN); 98129f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, lval, MPFR_RNDN); 98229f144ccSMatthew G. Knepley /* */ 98329f144ccSMatthew G. Knepley do { 98429f144ccSMatthew G. Knepley PetscReal d1, d2, d3, d4; 98529f144ccSMatthew G. Knepley PetscInt k = 1; 98629f144ccSMatthew G. Knepley 98729f144ccSMatthew G. Knepley ++l; 98829f144ccSMatthew G. Knepley mpfr_set_d(maxTerm, 0.0, MPFR_RNDN); 98929f144ccSMatthew G. Knepley /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %D sum: %15.15f\n", l, sum); */ 99029f144ccSMatthew G. Knepley /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 99129f144ccSMatthew G. Knepley mpfr_set(psum, osum, MPFR_RNDN); 99229f144ccSMatthew G. Knepley mpfr_set(osum, sum, MPFR_RNDN); 99329f144ccSMatthew G. Knepley mpfr_mul_d(h, h, 0.5, MPFR_RNDN); 99429f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, 0.5, MPFR_RNDN); 99529f144ccSMatthew G. Knepley do { 99629f144ccSMatthew G. Knepley mpfr_set_si(kh, k, MPFR_RNDN); 99729f144ccSMatthew G. Knepley mpfr_mul(kh, kh, h, MPFR_RNDN); 99829f144ccSMatthew G. Knepley /* Weight */ 99929f144ccSMatthew G. Knepley mpfr_set(wk, h, MPFR_RNDN); 100029f144ccSMatthew G. Knepley mpfr_sinh_cosh(msinh, mcosh, kh, MPFR_RNDN); 100129f144ccSMatthew G. Knepley mpfr_mul(msinh, msinh, pi2, MPFR_RNDN); 100229f144ccSMatthew G. Knepley mpfr_mul(mcosh, mcosh, pi2, MPFR_RNDN); 100329f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 100429f144ccSMatthew G. Knepley mpfr_sqr(tmp, tmp, MPFR_RNDN); 100529f144ccSMatthew G. Knepley mpfr_mul(wk, wk, mcosh, MPFR_RNDN); 100629f144ccSMatthew G. Knepley mpfr_div(wk, wk, tmp, MPFR_RNDN); 100729f144ccSMatthew G. Knepley /* Abscissa */ 100829f144ccSMatthew G. Knepley mpfr_set_d(yk, 1.0, MPFR_RNDZ); 100929f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 101029f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 101129f144ccSMatthew G. Knepley mpfr_exp(tmp, msinh, MPFR_RNDN); 101229f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 101329f144ccSMatthew G. Knepley /* Quadrature points */ 101429f144ccSMatthew G. Knepley mpfr_sub_d(lx, yk, 1.0, MPFR_RNDZ); 101529f144ccSMatthew G. Knepley mpfr_mul(lx, lx, alpha, MPFR_RNDU); 101629f144ccSMatthew G. Knepley mpfr_add(lx, lx, beta, MPFR_RNDU); 101729f144ccSMatthew G. Knepley mpfr_d_sub(rx, 1.0, yk, MPFR_RNDZ); 101829f144ccSMatthew G. Knepley mpfr_mul(rx, rx, alpha, MPFR_RNDD); 101929f144ccSMatthew G. Knepley mpfr_add(rx, rx, beta, MPFR_RNDD); 102029f144ccSMatthew G. Knepley /* Evaluation */ 102129f144ccSMatthew G. Knepley func(mpfr_get_d(lx, MPFR_RNDU), &lval); 102229f144ccSMatthew G. Knepley func(mpfr_get_d(rx, MPFR_RNDD), &rval); 102329f144ccSMatthew G. Knepley /* Update */ 102429f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 102529f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, lval, 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_set(curTerm, tmp, MPFR_RNDN); 103029f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 103129f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, rval, MPFR_RNDN); 103229f144ccSMatthew G. Knepley mpfr_add(sum, sum, tmp, MPFR_RNDN); 103329f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 103429f144ccSMatthew G. Knepley mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN); 103529f144ccSMatthew G. Knepley mpfr_max(curTerm, curTerm, tmp, MPFR_RNDN); 103629f144ccSMatthew G. Knepley ++k; 103729f144ccSMatthew G. Knepley /* Only need to evaluate every other point on refined levels */ 103829f144ccSMatthew G. Knepley if (l != 1) ++k; 103929f144ccSMatthew G. Knepley mpfr_log10(tmp, wk, MPFR_RNDN); 104029f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 1041c9f744b5SMatthew G. Knepley } while (mpfr_get_d(tmp, MPFR_RNDN) < safetyFactor*digits); /* Only need to evaluate sum until weights are < 32 digits of precision */ 104229f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, osum, MPFR_RNDN); 104329f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 104429f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 104529f144ccSMatthew G. Knepley d1 = mpfr_get_d(tmp, MPFR_RNDN); 104629f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, psum, MPFR_RNDN); 104729f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 104829f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 104929f144ccSMatthew G. Knepley d2 = mpfr_get_d(tmp, MPFR_RNDN); 105029f144ccSMatthew G. Knepley mpfr_log10(tmp, maxTerm, MPFR_RNDN); 1051c9f744b5SMatthew G. Knepley d3 = mpfr_get_d(tmp, MPFR_RNDN) - digits; 105229f144ccSMatthew G. Knepley mpfr_log10(tmp, curTerm, MPFR_RNDN); 105329f144ccSMatthew G. Knepley d4 = mpfr_get_d(tmp, MPFR_RNDN); 105429f144ccSMatthew G. Knepley d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1)/d2, 2*d1), d3), d4))); 1055b0649871SThomas Klotz } while (d < digits && l < 8); 105629f144ccSMatthew G. Knepley *sol = mpfr_get_d(sum, MPFR_RNDN); 105729f144ccSMatthew G. Knepley /* Cleanup */ 105829f144ccSMatthew G. Knepley mpfr_clears(alpha, beta, h, sum, osum, psum, yk, wk, lx, rx, tmp, maxTerm, curTerm, pi2, kh, msinh, mcosh, NULL); 105929f144ccSMatthew G. Knepley PetscFunctionReturn(0); 106029f144ccSMatthew G. Knepley } 1061d525116cSMatthew G. Knepley #else 1062fbfcfee5SBarry Smith 1063d525116cSMatthew G. Knepley PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(PetscReal, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, PetscReal *sol) 1064d525116cSMatthew G. Knepley { 1065d525116cSMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "This method will not work without MPFR. Reconfigure using --download-mpfr --download-gmp"); 1066d525116cSMatthew G. Knepley } 106729f144ccSMatthew G. Knepley #endif 106829f144ccSMatthew G. Knepley 1069194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n 1070194825f6SJed Brown * A in column-major format 1071194825f6SJed Brown * Ainv in row-major format 1072194825f6SJed Brown * tau has length m 1073194825f6SJed Brown * worksize must be >= max(1,n) 1074194825f6SJed Brown */ 1075194825f6SJed Brown static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m,PetscInt mstride,PetscInt n,PetscReal *A_in,PetscReal *Ainv_out,PetscScalar *tau,PetscInt worksize,PetscScalar *work) 1076194825f6SJed Brown { 1077194825f6SJed Brown PetscErrorCode ierr; 1078194825f6SJed Brown PetscBLASInt M,N,K,lda,ldb,ldwork,info; 1079194825f6SJed Brown PetscScalar *A,*Ainv,*R,*Q,Alpha; 1080194825f6SJed Brown 1081194825f6SJed Brown PetscFunctionBegin; 1082194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 1083194825f6SJed Brown { 1084194825f6SJed Brown PetscInt i,j; 1085dcca6d9dSJed Brown ierr = PetscMalloc2(m*n,&A,m*n,&Ainv);CHKERRQ(ierr); 1086194825f6SJed Brown for (j=0; j<n; j++) { 1087194825f6SJed Brown for (i=0; i<m; i++) A[i+m*j] = A_in[i+mstride*j]; 1088194825f6SJed Brown } 1089194825f6SJed Brown mstride = m; 1090194825f6SJed Brown } 1091194825f6SJed Brown #else 1092194825f6SJed Brown A = A_in; 1093194825f6SJed Brown Ainv = Ainv_out; 1094194825f6SJed Brown #endif 1095194825f6SJed Brown 1096194825f6SJed Brown ierr = PetscBLASIntCast(m,&M);CHKERRQ(ierr); 1097194825f6SJed Brown ierr = PetscBLASIntCast(n,&N);CHKERRQ(ierr); 1098194825f6SJed Brown ierr = PetscBLASIntCast(mstride,&lda);CHKERRQ(ierr); 1099194825f6SJed Brown ierr = PetscBLASIntCast(worksize,&ldwork);CHKERRQ(ierr); 1100194825f6SJed Brown ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1101001a771dSBarry Smith PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&M,&N,A,&lda,tau,work,&ldwork,&info)); 1102194825f6SJed Brown ierr = PetscFPTrapPop();CHKERRQ(ierr); 1103194825f6SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xGEQRF error"); 1104194825f6SJed Brown R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */ 1105194825f6SJed Brown 1106194825f6SJed Brown /* Extract an explicit representation of Q */ 1107194825f6SJed Brown Q = Ainv; 1108194825f6SJed Brown ierr = PetscMemcpy(Q,A,mstride*n*sizeof(PetscScalar));CHKERRQ(ierr); 1109194825f6SJed Brown K = N; /* full rank */ 1110c964aadfSJose E. Roman PetscStackCallBLAS("LAPACKorgqr",LAPACKorgqr_(&M,&N,&K,Q,&lda,tau,work,&ldwork,&info)); 1111194825f6SJed Brown if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"xORGQR/xUNGQR error"); 1112194825f6SJed Brown 1113194825f6SJed Brown /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */ 1114194825f6SJed Brown Alpha = 1.0; 1115194825f6SJed Brown ldb = lda; 1116001a771dSBarry Smith PetscStackCallBLAS("BLAStrsm",BLAStrsm_("Right","Upper","ConjugateTranspose","NotUnitTriangular",&M,&N,&Alpha,R,&lda,Q,&ldb)); 1117194825f6SJed Brown /* Ainv is Q, overwritten with inverse */ 1118194825f6SJed Brown 1119194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 1120194825f6SJed Brown { 1121194825f6SJed Brown PetscInt i; 1122194825f6SJed Brown for (i=0; i<m*n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]); 1123194825f6SJed Brown ierr = PetscFree2(A,Ainv);CHKERRQ(ierr); 1124194825f6SJed Brown } 1125194825f6SJed Brown #endif 1126194825f6SJed Brown PetscFunctionReturn(0); 1127194825f6SJed Brown } 1128194825f6SJed Brown 1129194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */ 1130194825f6SJed Brown static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval,const PetscReal *x,PetscInt ndegree,const PetscInt *degrees,PetscBool Transpose,PetscReal *B) 1131194825f6SJed Brown { 1132194825f6SJed Brown PetscErrorCode ierr; 1133194825f6SJed Brown PetscReal *Bv; 1134194825f6SJed Brown PetscInt i,j; 1135194825f6SJed Brown 1136194825f6SJed Brown PetscFunctionBegin; 1137785e854fSJed Brown ierr = PetscMalloc1((ninterval+1)*ndegree,&Bv);CHKERRQ(ierr); 1138194825f6SJed Brown /* Point evaluation of L_p on all the source vertices */ 1139194825f6SJed Brown ierr = PetscDTLegendreEval(ninterval+1,x,ndegree,degrees,Bv,NULL,NULL);CHKERRQ(ierr); 1140194825f6SJed Brown /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */ 1141194825f6SJed Brown for (i=0; i<ninterval; i++) { 1142194825f6SJed Brown for (j=0; j<ndegree; j++) { 1143194825f6SJed Brown if (Transpose) B[i+ninterval*j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j]; 1144194825f6SJed Brown else B[i*ndegree+j] = Bv[(i+1)*ndegree+j] - Bv[i*ndegree+j]; 1145194825f6SJed Brown } 1146194825f6SJed Brown } 1147194825f6SJed Brown ierr = PetscFree(Bv);CHKERRQ(ierr); 1148194825f6SJed Brown PetscFunctionReturn(0); 1149194825f6SJed Brown } 1150194825f6SJed Brown 1151194825f6SJed Brown /*@ 1152194825f6SJed Brown PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals 1153194825f6SJed Brown 1154194825f6SJed Brown Not Collective 1155194825f6SJed Brown 1156194825f6SJed Brown Input Arguments: 1157194825f6SJed Brown + degree - degree of reconstruction polynomial 1158194825f6SJed Brown . nsource - number of source intervals 1159194825f6SJed Brown . sourcex - sorted coordinates of source cell boundaries (length nsource+1) 1160194825f6SJed Brown . ntarget - number of target intervals 1161194825f6SJed Brown - targetx - sorted coordinates of target cell boundaries (length ntarget+1) 1162194825f6SJed Brown 1163194825f6SJed Brown Output Arguments: 1164194825f6SJed Brown . R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s] 1165194825f6SJed Brown 1166194825f6SJed Brown Level: advanced 1167194825f6SJed Brown 1168194825f6SJed Brown .seealso: PetscDTLegendreEval() 1169194825f6SJed Brown @*/ 1170194825f6SJed Brown PetscErrorCode PetscDTReconstructPoly(PetscInt degree,PetscInt nsource,const PetscReal *sourcex,PetscInt ntarget,const PetscReal *targetx,PetscReal *R) 1171194825f6SJed Brown { 1172194825f6SJed Brown PetscErrorCode ierr; 1173194825f6SJed Brown PetscInt i,j,k,*bdegrees,worksize; 1174194825f6SJed Brown PetscReal xmin,xmax,center,hscale,*sourcey,*targety,*Bsource,*Bsinv,*Btarget; 1175194825f6SJed Brown PetscScalar *tau,*work; 1176194825f6SJed Brown 1177194825f6SJed Brown PetscFunctionBegin; 1178194825f6SJed Brown PetscValidRealPointer(sourcex,3); 1179194825f6SJed Brown PetscValidRealPointer(targetx,5); 1180194825f6SJed Brown PetscValidRealPointer(R,6); 1181194825f6SJed 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); 1182194825f6SJed Brown #if defined(PETSC_USE_DEBUG) 1183194825f6SJed Brown for (i=0; i<nsource; i++) { 118457622a8eSBarry 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]); 1185194825f6SJed Brown } 1186194825f6SJed Brown for (i=0; i<ntarget; i++) { 118757622a8eSBarry 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]); 1188194825f6SJed Brown } 1189194825f6SJed Brown #endif 1190194825f6SJed Brown xmin = PetscMin(sourcex[0],targetx[0]); 1191194825f6SJed Brown xmax = PetscMax(sourcex[nsource],targetx[ntarget]); 1192194825f6SJed Brown center = (xmin + xmax)/2; 1193194825f6SJed Brown hscale = (xmax - xmin)/2; 1194194825f6SJed Brown worksize = nsource; 1195dcca6d9dSJed Brown ierr = PetscMalloc4(degree+1,&bdegrees,nsource+1,&sourcey,nsource*(degree+1),&Bsource,worksize,&work);CHKERRQ(ierr); 1196dcca6d9dSJed Brown ierr = PetscMalloc4(nsource,&tau,nsource*(degree+1),&Bsinv,ntarget+1,&targety,ntarget*(degree+1),&Btarget);CHKERRQ(ierr); 1197194825f6SJed Brown for (i=0; i<=nsource; i++) sourcey[i] = (sourcex[i]-center)/hscale; 1198194825f6SJed Brown for (i=0; i<=degree; i++) bdegrees[i] = i+1; 1199194825f6SJed Brown ierr = PetscDTLegendreIntegrate(nsource,sourcey,degree+1,bdegrees,PETSC_TRUE,Bsource);CHKERRQ(ierr); 1200194825f6SJed Brown ierr = PetscDTPseudoInverseQR(nsource,nsource,degree+1,Bsource,Bsinv,tau,nsource,work);CHKERRQ(ierr); 1201194825f6SJed Brown for (i=0; i<=ntarget; i++) targety[i] = (targetx[i]-center)/hscale; 1202194825f6SJed Brown ierr = PetscDTLegendreIntegrate(ntarget,targety,degree+1,bdegrees,PETSC_FALSE,Btarget);CHKERRQ(ierr); 1203194825f6SJed Brown for (i=0; i<ntarget; i++) { 1204194825f6SJed Brown PetscReal rowsum = 0; 1205194825f6SJed Brown for (j=0; j<nsource; j++) { 1206194825f6SJed Brown PetscReal sum = 0; 1207194825f6SJed Brown for (k=0; k<degree+1; k++) { 1208194825f6SJed Brown sum += Btarget[i*(degree+1)+k] * Bsinv[k*nsource+j]; 1209194825f6SJed Brown } 1210194825f6SJed Brown R[i*nsource+j] = sum; 1211194825f6SJed Brown rowsum += sum; 1212194825f6SJed Brown } 1213194825f6SJed Brown for (j=0; j<nsource; j++) R[i*nsource+j] /= rowsum; /* normalize each row */ 1214194825f6SJed Brown } 1215194825f6SJed Brown ierr = PetscFree4(bdegrees,sourcey,Bsource,work);CHKERRQ(ierr); 1216194825f6SJed Brown ierr = PetscFree4(tau,Bsinv,targety,Btarget);CHKERRQ(ierr); 1217194825f6SJed Brown PetscFunctionReturn(0); 1218194825f6SJed Brown } 1219