137045ce4SJed Brown /* Discretization tools */ 237045ce4SJed Brown 30c35b76eSJed Brown #include <petscdt.h> /*I "petscdt.h" I*/ 437045ce4SJed Brown #include <petscblaslapack.h> 5af0996ceSBarry Smith #include <petsc/private/petscimpl.h> 6af0996ceSBarry Smith #include <petsc/private/dtimpl.h> 7665c2dedSJed Brown #include <petscviewer.h> 859804f93SMatthew G. Knepley #include <petscdmplex.h> 959804f93SMatthew G. Knepley #include <petscdmshell.h> 1037045ce4SJed Brown 1198c04793SMatthew G. Knepley #if defined(PETSC_HAVE_MPFR) 1298c04793SMatthew G. Knepley #include <mpfr.h> 1398c04793SMatthew G. Knepley #endif 1498c04793SMatthew G. Knepley 15d3c69ad0SToby Isaac const char *const PetscDTNodeTypes_shifted[] = {"default", "gaussjacobi", "equispaced", "tanhsinh", "PETSCDTNODES_", NULL}; 16d3c69ad0SToby Isaac const char *const *const PetscDTNodeTypes = PetscDTNodeTypes_shifted + 1; 17d3c69ad0SToby Isaac 18d3c69ad0SToby Isaac const char *const PetscDTSimplexQuadratureTypes_shifted[] = {"default", "conic", "minsym", "PETSCDTSIMPLEXQUAD_", NULL}; 19d3c69ad0SToby Isaac const char *const *const PetscDTSimplexQuadratureTypes = PetscDTSimplexQuadratureTypes_shifted + 1; 20d4afb720SToby Isaac 21e6a796c3SToby Isaac static PetscBool GolubWelschCite = PETSC_FALSE; 22e6a796c3SToby Isaac const char GolubWelschCitation[] = "@article{GolubWelsch1969,\n" 230bfcf5a5SMatthew G. Knepley " author = {Golub and Welsch},\n" 240bfcf5a5SMatthew G. Knepley " title = {Calculation of Quadrature Rules},\n" 250bfcf5a5SMatthew G. Knepley " journal = {Math. Comp.},\n" 260bfcf5a5SMatthew G. Knepley " volume = {23},\n" 270bfcf5a5SMatthew G. Knepley " number = {106},\n" 280bfcf5a5SMatthew G. Knepley " pages = {221--230},\n" 290bfcf5a5SMatthew G. Knepley " year = {1969}\n}\n"; 300bfcf5a5SMatthew G. Knepley 31c4762a1bSJed Brown /* Numerical tests in src/dm/dt/tests/ex1.c show that when computing the nodes and weights of Gauss-Jacobi 3294e21283SToby Isaac quadrature rules: 33e6a796c3SToby Isaac 3494e21283SToby Isaac - in double precision, Newton's method and Golub & Welsch both work for moderate degrees (< 100), 3594e21283SToby Isaac - in single precision, Newton's method starts producing incorrect roots around n = 15, but 3694e21283SToby Isaac the weights from Golub & Welsch become a problem before then: they produces errors 3794e21283SToby Isaac in computing the Jacobi-polynomial Gram matrix around n = 6. 3894e21283SToby Isaac 3994e21283SToby Isaac So we default to Newton's method (required fewer dependencies) */ 4094e21283SToby Isaac PetscBool PetscDTGaussQuadratureNewton_Internal = PETSC_TRUE; 412cd22861SMatthew G. Knepley 422cd22861SMatthew G. Knepley PetscClassId PETSCQUADRATURE_CLASSID = 0; 432cd22861SMatthew G. Knepley 4440d8ff71SMatthew G. Knepley /*@ 45dce8aebaSBarry Smith PetscQuadratureCreate - Create a `PetscQuadrature` object 4640d8ff71SMatthew G. Knepley 47d083f849SBarry Smith Collective 4840d8ff71SMatthew G. Knepley 4940d8ff71SMatthew G. Knepley Input Parameter: 50dce8aebaSBarry Smith . comm - The communicator for the `PetscQuadrature` object 5140d8ff71SMatthew G. Knepley 5240d8ff71SMatthew G. Knepley Output Parameter: 53*20f4b53cSBarry Smith . q - The `PetscQuadrature` object 5440d8ff71SMatthew G. Knepley 5540d8ff71SMatthew G. Knepley Level: beginner 5640d8ff71SMatthew G. Knepley 57dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `Petscquadraturedestroy()`, `PetscQuadratureGetData()` 5840d8ff71SMatthew G. Knepley @*/ 59d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureCreate(MPI_Comm comm, PetscQuadrature *q) 60d71ae5a4SJacob Faibussowitsch { 6121454ff5SMatthew G. Knepley PetscFunctionBegin; 6221454ff5SMatthew G. Knepley PetscValidPointer(q, 2); 639566063dSJacob Faibussowitsch PetscCall(DMInitializePackage()); 649566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(*q, PETSCQUADRATURE_CLASSID, "PetscQuadrature", "Quadrature", "DT", comm, PetscQuadratureDestroy, PetscQuadratureView)); 6521454ff5SMatthew G. Knepley (*q)->dim = -1; 66a6b92713SMatthew G. Knepley (*q)->Nc = 1; 67bcede257SMatthew G. Knepley (*q)->order = -1; 6821454ff5SMatthew G. Knepley (*q)->numPoints = 0; 6921454ff5SMatthew G. Knepley (*q)->points = NULL; 7021454ff5SMatthew G. Knepley (*q)->weights = NULL; 713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 7221454ff5SMatthew G. Knepley } 7321454ff5SMatthew G. Knepley 74c9638911SMatthew G. Knepley /*@ 75dce8aebaSBarry Smith PetscQuadratureDuplicate - Create a deep copy of the `PetscQuadrature` object 76c9638911SMatthew G. Knepley 77*20f4b53cSBarry Smith Collective 78c9638911SMatthew G. Knepley 79c9638911SMatthew G. Knepley Input Parameter: 80dce8aebaSBarry Smith . q - The `PetscQuadrature` object 81c9638911SMatthew G. Knepley 82c9638911SMatthew G. Knepley Output Parameter: 83dce8aebaSBarry Smith . r - The new `PetscQuadrature` object 84c9638911SMatthew G. Knepley 85c9638911SMatthew G. Knepley Level: beginner 86c9638911SMatthew G. Knepley 87dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureDestroy()`, `PetscQuadratureGetData()` 88c9638911SMatthew G. Knepley @*/ 89d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureDuplicate(PetscQuadrature q, PetscQuadrature *r) 90d71ae5a4SJacob Faibussowitsch { 91a6b92713SMatthew G. Knepley PetscInt order, dim, Nc, Nq; 92c9638911SMatthew G. Knepley const PetscReal *points, *weights; 93c9638911SMatthew G. Knepley PetscReal *p, *w; 94c9638911SMatthew G. Knepley 95c9638911SMatthew G. Knepley PetscFunctionBegin; 96064a246eSJacob Faibussowitsch PetscValidPointer(q, 1); 979566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PetscObjectComm((PetscObject)q), r)); 989566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q, &order)); 999566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*r, order)); 1009566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &Nq, &points, &weights)); 1019566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nq * dim, &p)); 1029566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nq * Nc, &w)); 1039566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(p, points, Nq * dim)); 1049566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(w, weights, Nc * Nq)); 1059566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*r, dim, Nc, Nq, p, w)); 1063ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 107c9638911SMatthew G. Knepley } 108c9638911SMatthew G. Knepley 10940d8ff71SMatthew G. Knepley /*@ 110dce8aebaSBarry Smith PetscQuadratureDestroy - Destroys a `PetscQuadrature` object 11140d8ff71SMatthew G. Knepley 112*20f4b53cSBarry Smith Collective 11340d8ff71SMatthew G. Knepley 11440d8ff71SMatthew G. Knepley Input Parameter: 115dce8aebaSBarry Smith . q - The `PetscQuadrature` object 11640d8ff71SMatthew G. Knepley 11740d8ff71SMatthew G. Knepley Level: beginner 11840d8ff71SMatthew G. Knepley 119dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 12040d8ff71SMatthew G. Knepley @*/ 121d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureDestroy(PetscQuadrature *q) 122d71ae5a4SJacob Faibussowitsch { 123bfa639d9SMatthew G. Knepley PetscFunctionBegin; 1243ba16761SJacob Faibussowitsch if (!*q) PetscFunctionReturn(PETSC_SUCCESS); 1252cd22861SMatthew G. Knepley PetscValidHeaderSpecific((*q), PETSCQUADRATURE_CLASSID, 1); 12621454ff5SMatthew G. Knepley if (--((PetscObject)(*q))->refct > 0) { 12721454ff5SMatthew G. Knepley *q = NULL; 1283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 12921454ff5SMatthew G. Knepley } 1309566063dSJacob Faibussowitsch PetscCall(PetscFree((*q)->points)); 1319566063dSJacob Faibussowitsch PetscCall(PetscFree((*q)->weights)); 1329566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(q)); 1333ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 13421454ff5SMatthew G. Knepley } 13521454ff5SMatthew G. Knepley 136bcede257SMatthew G. Knepley /*@ 137dce8aebaSBarry Smith PetscQuadratureGetOrder - Return the order of the method in the `PetscQuadrature` 138bcede257SMatthew G. Knepley 139*20f4b53cSBarry Smith Not Collective 140bcede257SMatthew G. Knepley 141bcede257SMatthew G. Knepley Input Parameter: 142dce8aebaSBarry Smith . q - The `PetscQuadrature` object 143bcede257SMatthew G. Knepley 144bcede257SMatthew G. Knepley Output Parameter: 145bcede257SMatthew G. Knepley . order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated 146bcede257SMatthew G. Knepley 147bcede257SMatthew G. Knepley Level: intermediate 148bcede257SMatthew G. Knepley 149dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureSetOrder()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 150bcede257SMatthew G. Knepley @*/ 151d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetOrder(PetscQuadrature q, PetscInt *order) 152d71ae5a4SJacob Faibussowitsch { 153bcede257SMatthew G. Knepley PetscFunctionBegin; 1542cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 155dadcf809SJacob Faibussowitsch PetscValidIntPointer(order, 2); 156bcede257SMatthew G. Knepley *order = q->order; 1573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 158bcede257SMatthew G. Knepley } 159bcede257SMatthew G. Knepley 160bcede257SMatthew G. Knepley /*@ 161dce8aebaSBarry Smith PetscQuadratureSetOrder - Set the order of the method in the `PetscQuadrature` 162bcede257SMatthew G. Knepley 163*20f4b53cSBarry Smith Not Collective 164bcede257SMatthew G. Knepley 165bcede257SMatthew G. Knepley Input Parameters: 166dce8aebaSBarry Smith + q - The `PetscQuadrature` object 167bcede257SMatthew G. Knepley - order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated 168bcede257SMatthew G. Knepley 169bcede257SMatthew G. Knepley Level: intermediate 170bcede257SMatthew G. Knepley 171dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureGetOrder()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 172bcede257SMatthew G. Knepley @*/ 173d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetOrder(PetscQuadrature q, PetscInt order) 174d71ae5a4SJacob Faibussowitsch { 175bcede257SMatthew G. Knepley PetscFunctionBegin; 1762cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 177bcede257SMatthew G. Knepley q->order = order; 1783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 179bcede257SMatthew G. Knepley } 180bcede257SMatthew G. Knepley 181a6b92713SMatthew G. Knepley /*@ 182a6b92713SMatthew G. Knepley PetscQuadratureGetNumComponents - Return the number of components for functions to be integrated 183a6b92713SMatthew G. Knepley 184*20f4b53cSBarry Smith Not Collective 185a6b92713SMatthew G. Knepley 186a6b92713SMatthew G. Knepley Input Parameter: 187dce8aebaSBarry Smith . q - The `PetscQuadrature` object 188a6b92713SMatthew G. Knepley 189a6b92713SMatthew G. Knepley Output Parameter: 190a6b92713SMatthew G. Knepley . Nc - The number of components 191a6b92713SMatthew G. Knepley 192*20f4b53cSBarry Smith Level: intermediate 193*20f4b53cSBarry Smith 194dce8aebaSBarry Smith Note: 195dce8aebaSBarry Smith We are performing an integral int f(x) . w(x) dx, where both f and w (the weight) have Nc components. 196a6b92713SMatthew G. Knepley 197dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureSetNumComponents()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 198a6b92713SMatthew G. Knepley @*/ 199d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetNumComponents(PetscQuadrature q, PetscInt *Nc) 200d71ae5a4SJacob Faibussowitsch { 201a6b92713SMatthew G. Knepley PetscFunctionBegin; 2022cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 203dadcf809SJacob Faibussowitsch PetscValidIntPointer(Nc, 2); 204a6b92713SMatthew G. Knepley *Nc = q->Nc; 2053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 206a6b92713SMatthew G. Knepley } 207a6b92713SMatthew G. Knepley 208a6b92713SMatthew G. Knepley /*@ 209a6b92713SMatthew G. Knepley PetscQuadratureSetNumComponents - Return the number of components for functions to be integrated 210a6b92713SMatthew G. Knepley 211*20f4b53cSBarry Smith Not Collective 212a6b92713SMatthew G. Knepley 213a6b92713SMatthew G. Knepley Input Parameters: 214a6b92713SMatthew G. Knepley + q - The PetscQuadrature object 215a6b92713SMatthew G. Knepley - Nc - The number of components 216a6b92713SMatthew G. Knepley 217*20f4b53cSBarry Smith Level: intermediate 218*20f4b53cSBarry Smith 219dce8aebaSBarry Smith Note: 220dce8aebaSBarry Smith We are performing an integral int f(x) . w(x) dx, where both f and w (the weight) have Nc components. 221a6b92713SMatthew G. Knepley 222dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureGetNumComponents()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 223a6b92713SMatthew G. Knepley @*/ 224d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetNumComponents(PetscQuadrature q, PetscInt Nc) 225d71ae5a4SJacob Faibussowitsch { 226a6b92713SMatthew G. Knepley PetscFunctionBegin; 2272cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 228a6b92713SMatthew G. Knepley q->Nc = Nc; 2293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 230a6b92713SMatthew G. Knepley } 231a6b92713SMatthew G. Knepley 23240d8ff71SMatthew G. Knepley /*@C 233dce8aebaSBarry Smith PetscQuadratureGetData - Returns the data defining the `PetscQuadrature` 23440d8ff71SMatthew G. Knepley 235*20f4b53cSBarry Smith Not Collective 23640d8ff71SMatthew G. Knepley 23740d8ff71SMatthew G. Knepley Input Parameter: 238dce8aebaSBarry Smith . q - The `PetscQuadrature` object 23940d8ff71SMatthew G. Knepley 24040d8ff71SMatthew G. Knepley Output Parameters: 24140d8ff71SMatthew G. Knepley + dim - The spatial dimension 242805e7170SToby Isaac . Nc - The number of components 24340d8ff71SMatthew G. Knepley . npoints - The number of quadrature points 24440d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point 24540d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point 24640d8ff71SMatthew G. Knepley 24740d8ff71SMatthew G. Knepley Level: intermediate 24840d8ff71SMatthew G. Knepley 249dce8aebaSBarry Smith Fortran Note: 250dce8aebaSBarry Smith From Fortran you must call `PetscQuadratureRestoreData()` when you are done with the data 2511fd49c25SBarry Smith 252dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureSetData()` 25340d8ff71SMatthew G. Knepley @*/ 254d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetData(PetscQuadrature q, PetscInt *dim, PetscInt *Nc, PetscInt *npoints, const PetscReal *points[], const PetscReal *weights[]) 255d71ae5a4SJacob Faibussowitsch { 25621454ff5SMatthew G. Knepley PetscFunctionBegin; 2572cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 25821454ff5SMatthew G. Knepley if (dim) { 259dadcf809SJacob Faibussowitsch PetscValidIntPointer(dim, 2); 26021454ff5SMatthew G. Knepley *dim = q->dim; 26121454ff5SMatthew G. Knepley } 262a6b92713SMatthew G. Knepley if (Nc) { 263dadcf809SJacob Faibussowitsch PetscValidIntPointer(Nc, 3); 264a6b92713SMatthew G. Knepley *Nc = q->Nc; 265a6b92713SMatthew G. Knepley } 26621454ff5SMatthew G. Knepley if (npoints) { 267dadcf809SJacob Faibussowitsch PetscValidIntPointer(npoints, 4); 26821454ff5SMatthew G. Knepley *npoints = q->numPoints; 26921454ff5SMatthew G. Knepley } 27021454ff5SMatthew G. Knepley if (points) { 271a6b92713SMatthew G. Knepley PetscValidPointer(points, 5); 27221454ff5SMatthew G. Knepley *points = q->points; 27321454ff5SMatthew G. Knepley } 27421454ff5SMatthew G. Knepley if (weights) { 275a6b92713SMatthew G. Knepley PetscValidPointer(weights, 6); 27621454ff5SMatthew G. Knepley *weights = q->weights; 27721454ff5SMatthew G. Knepley } 2783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 27921454ff5SMatthew G. Knepley } 28021454ff5SMatthew G. Knepley 2814f9ab2b4SJed Brown /*@ 2824f9ab2b4SJed Brown PetscQuadratureEqual - determine whether two quadratures are equivalent 2834f9ab2b4SJed Brown 2844f9ab2b4SJed Brown Input Parameters: 285dce8aebaSBarry Smith + A - A `PetscQuadrature` object 286dce8aebaSBarry Smith - B - Another `PetscQuadrature` object 2874f9ab2b4SJed Brown 2884f9ab2b4SJed Brown Output Parameters: 289dce8aebaSBarry Smith . equal - `PETSC_TRUE` if the quadratures are the same 2904f9ab2b4SJed Brown 2914f9ab2b4SJed Brown Level: intermediate 2924f9ab2b4SJed Brown 293dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()` 2944f9ab2b4SJed Brown @*/ 295d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureEqual(PetscQuadrature A, PetscQuadrature B, PetscBool *equal) 296d71ae5a4SJacob Faibussowitsch { 2974f9ab2b4SJed Brown PetscFunctionBegin; 2984f9ab2b4SJed Brown PetscValidHeaderSpecific(A, PETSCQUADRATURE_CLASSID, 1); 2994f9ab2b4SJed Brown PetscValidHeaderSpecific(B, PETSCQUADRATURE_CLASSID, 2); 3004f9ab2b4SJed Brown PetscValidBoolPointer(equal, 3); 3014f9ab2b4SJed Brown *equal = PETSC_FALSE; 3023ba16761SJacob Faibussowitsch if (A->dim != B->dim || A->Nc != B->Nc || A->order != B->order || A->numPoints != B->numPoints) PetscFunctionReturn(PETSC_SUCCESS); 3034f9ab2b4SJed Brown for (PetscInt i = 0; i < A->numPoints * A->dim; i++) { 3043ba16761SJacob Faibussowitsch if (PetscAbsReal(A->points[i] - B->points[i]) > PETSC_SMALL) PetscFunctionReturn(PETSC_SUCCESS); 3054f9ab2b4SJed Brown } 3064f9ab2b4SJed Brown if (!A->weights && !B->weights) { 3074f9ab2b4SJed Brown *equal = PETSC_TRUE; 3083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3094f9ab2b4SJed Brown } 3104f9ab2b4SJed Brown if (A->weights && B->weights) { 3114f9ab2b4SJed Brown for (PetscInt i = 0; i < A->numPoints; i++) { 3123ba16761SJacob Faibussowitsch if (PetscAbsReal(A->weights[i] - B->weights[i]) > PETSC_SMALL) PetscFunctionReturn(PETSC_SUCCESS); 3134f9ab2b4SJed Brown } 3144f9ab2b4SJed Brown *equal = PETSC_TRUE; 3154f9ab2b4SJed Brown } 3163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3174f9ab2b4SJed Brown } 3184f9ab2b4SJed Brown 319d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobianInverse_Internal(PetscInt m, PetscInt n, const PetscReal J[], PetscReal Jinv[]) 320d71ae5a4SJacob Faibussowitsch { 321907761f8SToby Isaac PetscScalar *Js, *Jinvs; 322907761f8SToby Isaac PetscInt i, j, k; 323907761f8SToby Isaac PetscBLASInt bm, bn, info; 324907761f8SToby Isaac 325907761f8SToby Isaac PetscFunctionBegin; 3263ba16761SJacob Faibussowitsch if (!m || !n) PetscFunctionReturn(PETSC_SUCCESS); 3279566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(m, &bm)); 3289566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &bn)); 329907761f8SToby Isaac #if defined(PETSC_USE_COMPLEX) 3309566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m * n, &Js, m * n, &Jinvs)); 33128222859SToby Isaac for (i = 0; i < m * n; i++) Js[i] = J[i]; 332907761f8SToby Isaac #else 333907761f8SToby Isaac Js = (PetscReal *)J; 334907761f8SToby Isaac Jinvs = Jinv; 335907761f8SToby Isaac #endif 336907761f8SToby Isaac if (m == n) { 337907761f8SToby Isaac PetscBLASInt *pivots; 338907761f8SToby Isaac PetscScalar *W; 339907761f8SToby Isaac 3409566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m, &pivots, m, &W)); 341907761f8SToby Isaac 3429566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(Jinvs, Js, m * m)); 343792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bm, &bm, Jinvs, &bm, pivots, &info)); 34463a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 345792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bm, Jinvs, &bm, pivots, W, &bm, &info)); 34663a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 3479566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 348907761f8SToby Isaac } else if (m < n) { 349907761f8SToby Isaac PetscScalar *JJT; 350907761f8SToby Isaac PetscBLASInt *pivots; 351907761f8SToby Isaac PetscScalar *W; 352907761f8SToby Isaac 3539566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m * m, &JJT)); 3549566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m, &pivots, m, &W)); 355907761f8SToby Isaac for (i = 0; i < m; i++) { 356907761f8SToby Isaac for (j = 0; j < m; j++) { 357907761f8SToby Isaac PetscScalar val = 0.; 358907761f8SToby Isaac 359907761f8SToby Isaac for (k = 0; k < n; k++) val += Js[i * n + k] * Js[j * n + k]; 360907761f8SToby Isaac JJT[i * m + j] = val; 361907761f8SToby Isaac } 362907761f8SToby Isaac } 363907761f8SToby Isaac 364792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bm, &bm, JJT, &bm, pivots, &info)); 36563a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 366792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bm, JJT, &bm, pivots, W, &bm, &info)); 36763a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 368907761f8SToby Isaac for (i = 0; i < n; i++) { 369907761f8SToby Isaac for (j = 0; j < m; j++) { 370907761f8SToby Isaac PetscScalar val = 0.; 371907761f8SToby Isaac 372907761f8SToby Isaac for (k = 0; k < m; k++) val += Js[k * n + i] * JJT[k * m + j]; 373907761f8SToby Isaac Jinvs[i * m + j] = val; 374907761f8SToby Isaac } 375907761f8SToby Isaac } 3769566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 3779566063dSJacob Faibussowitsch PetscCall(PetscFree(JJT)); 378907761f8SToby Isaac } else { 379907761f8SToby Isaac PetscScalar *JTJ; 380907761f8SToby Isaac PetscBLASInt *pivots; 381907761f8SToby Isaac PetscScalar *W; 382907761f8SToby Isaac 3839566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &JTJ)); 3849566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(n, &pivots, n, &W)); 385907761f8SToby Isaac for (i = 0; i < n; i++) { 386907761f8SToby Isaac for (j = 0; j < n; j++) { 387907761f8SToby Isaac PetscScalar val = 0.; 388907761f8SToby Isaac 389907761f8SToby Isaac for (k = 0; k < m; k++) val += Js[k * n + i] * Js[k * n + j]; 390907761f8SToby Isaac JTJ[i * n + j] = val; 391907761f8SToby Isaac } 392907761f8SToby Isaac } 393907761f8SToby Isaac 394792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bn, &bn, JTJ, &bn, pivots, &info)); 39563a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 396792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bn, JTJ, &bn, pivots, W, &bn, &info)); 39763a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 398907761f8SToby Isaac for (i = 0; i < n; i++) { 399907761f8SToby Isaac for (j = 0; j < m; j++) { 400907761f8SToby Isaac PetscScalar val = 0.; 401907761f8SToby Isaac 402907761f8SToby Isaac for (k = 0; k < n; k++) val += JTJ[i * n + k] * Js[j * n + k]; 403907761f8SToby Isaac Jinvs[i * m + j] = val; 404907761f8SToby Isaac } 405907761f8SToby Isaac } 4069566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 4079566063dSJacob Faibussowitsch PetscCall(PetscFree(JTJ)); 408907761f8SToby Isaac } 409907761f8SToby Isaac #if defined(PETSC_USE_COMPLEX) 41028222859SToby Isaac for (i = 0; i < m * n; i++) Jinv[i] = PetscRealPart(Jinvs[i]); 4119566063dSJacob Faibussowitsch PetscCall(PetscFree2(Js, Jinvs)); 412907761f8SToby Isaac #endif 4133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 414907761f8SToby Isaac } 415907761f8SToby Isaac 416907761f8SToby Isaac /*@ 417907761f8SToby Isaac PetscQuadraturePushForward - Push forward a quadrature functional under an affine transformation. 418907761f8SToby Isaac 419*20f4b53cSBarry Smith Collective 420907761f8SToby Isaac 4214165533cSJose E. Roman Input Parameters: 422907761f8SToby Isaac + q - the quadrature functional 423907761f8SToby Isaac . imageDim - the dimension of the image of the transformation 424907761f8SToby Isaac . origin - a point in the original space 425907761f8SToby Isaac . originImage - the image of the origin under the transformation 426907761f8SToby Isaac . J - the Jacobian of the image: an [imageDim x dim] matrix in row major order 427dce8aebaSBarry Smith - formDegree - transform the quadrature weights as k-forms of this form degree (if the number of components is a multiple of (dim choose formDegree), it is assumed that they represent multiple k-forms) [see `PetscDTAltVPullback()` for interpretation of formDegree] 428907761f8SToby Isaac 4294165533cSJose E. Roman Output Parameters: 430907761f8SToby Isaac . Jinvstarq - a quadrature rule where each point is the image of a point in the original quadrature rule, and where the k-form weights have been pulled-back by the pseudoinverse of J to the k-form weights in the image space. 431907761f8SToby Isaac 4326c877ef6SSatish Balay Level: intermediate 4336c877ef6SSatish Balay 434dce8aebaSBarry Smith Note: 435dce8aebaSBarry Smith The new quadrature rule will have a different number of components if spaces have different dimensions. For example, pushing a 2-form forward from a two dimensional space to a three dimensional space changes the number of components from 1 to 3. 436dce8aebaSBarry Smith 437dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTAltVPullback()`, `PetscDTAltVPullbackMatrix()` 438907761f8SToby Isaac @*/ 439d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadraturePushForward(PetscQuadrature q, PetscInt imageDim, const PetscReal origin[], const PetscReal originImage[], const PetscReal J[], PetscInt formDegree, PetscQuadrature *Jinvstarq) 440d71ae5a4SJacob Faibussowitsch { 441907761f8SToby Isaac PetscInt dim, Nc, imageNc, formSize, Ncopies, imageFormSize, Npoints, pt, i, j, c; 442907761f8SToby Isaac const PetscReal *points; 443907761f8SToby Isaac const PetscReal *weights; 444907761f8SToby Isaac PetscReal *imagePoints, *imageWeights; 445907761f8SToby Isaac PetscReal *Jinv; 446907761f8SToby Isaac PetscReal *Jinvstar; 447907761f8SToby Isaac 448907761f8SToby Isaac PetscFunctionBegin; 449d4afb720SToby Isaac PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 45063a3b9bcSJacob Faibussowitsch PetscCheck(imageDim >= PetscAbsInt(formDegree), PetscObjectComm((PetscObject)q), PETSC_ERR_ARG_INCOMP, "Cannot represent a %" PetscInt_FMT "-form in %" PetscInt_FMT " dimensions", PetscAbsInt(formDegree), imageDim); 4519566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &Npoints, &points, &weights)); 4529566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, PetscAbsInt(formDegree), &formSize)); 45363a3b9bcSJacob Faibussowitsch PetscCheck(Nc % formSize == 0, PetscObjectComm((PetscObject)q), PETSC_ERR_ARG_INCOMP, "Number of components %" PetscInt_FMT " is not a multiple of formSize %" PetscInt_FMT, Nc, formSize); 454907761f8SToby Isaac Ncopies = Nc / formSize; 4559566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(imageDim, PetscAbsInt(formDegree), &imageFormSize)); 456907761f8SToby Isaac imageNc = Ncopies * imageFormSize; 4579566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Npoints * imageDim, &imagePoints)); 4589566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Npoints * imageNc, &imageWeights)); 4599566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(imageDim * dim, &Jinv, formSize * imageFormSize, &Jinvstar)); 4609566063dSJacob Faibussowitsch PetscCall(PetscDTJacobianInverse_Internal(imageDim, dim, J, Jinv)); 4619566063dSJacob Faibussowitsch PetscCall(PetscDTAltVPullbackMatrix(imageDim, dim, Jinv, formDegree, Jinvstar)); 462907761f8SToby Isaac for (pt = 0; pt < Npoints; pt++) { 463907761f8SToby Isaac const PetscReal *point = &points[pt * dim]; 464907761f8SToby Isaac PetscReal *imagePoint = &imagePoints[pt * imageDim]; 465907761f8SToby Isaac 466907761f8SToby Isaac for (i = 0; i < imageDim; i++) { 467907761f8SToby Isaac PetscReal val = originImage[i]; 468907761f8SToby Isaac 469907761f8SToby Isaac for (j = 0; j < dim; j++) val += J[i * dim + j] * (point[j] - origin[j]); 470907761f8SToby Isaac imagePoint[i] = val; 471907761f8SToby Isaac } 472907761f8SToby Isaac for (c = 0; c < Ncopies; c++) { 473907761f8SToby Isaac const PetscReal *form = &weights[pt * Nc + c * formSize]; 474907761f8SToby Isaac PetscReal *imageForm = &imageWeights[pt * imageNc + c * imageFormSize]; 475907761f8SToby Isaac 476907761f8SToby Isaac for (i = 0; i < imageFormSize; i++) { 477907761f8SToby Isaac PetscReal val = 0.; 478907761f8SToby Isaac 479907761f8SToby Isaac for (j = 0; j < formSize; j++) val += Jinvstar[i * formSize + j] * form[j]; 480907761f8SToby Isaac imageForm[i] = val; 481907761f8SToby Isaac } 482907761f8SToby Isaac } 483907761f8SToby Isaac } 4849566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PetscObjectComm((PetscObject)q), Jinvstarq)); 4859566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*Jinvstarq, imageDim, imageNc, Npoints, imagePoints, imageWeights)); 4869566063dSJacob Faibussowitsch PetscCall(PetscFree2(Jinv, Jinvstar)); 4873ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 488907761f8SToby Isaac } 489907761f8SToby Isaac 49040d8ff71SMatthew G. Knepley /*@C 49140d8ff71SMatthew G. Knepley PetscQuadratureSetData - Sets the data defining the quadrature 49240d8ff71SMatthew G. Knepley 493*20f4b53cSBarry Smith Not Collective 49440d8ff71SMatthew G. Knepley 49540d8ff71SMatthew G. Knepley Input Parameters: 496dce8aebaSBarry Smith + q - The `PetscQuadrature` object 49740d8ff71SMatthew G. Knepley . dim - The spatial dimension 498e2b35d93SBarry Smith . Nc - The number of components 49940d8ff71SMatthew G. Knepley . npoints - The number of quadrature points 50040d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point 50140d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point 50240d8ff71SMatthew G. Knepley 50340d8ff71SMatthew G. Knepley Level: intermediate 50440d8ff71SMatthew G. Knepley 505dce8aebaSBarry Smith Note: 506dce8aebaSBarry Smith This routine owns the references to points and weights, so they must be allocated using `PetscMalloc()` and the user should not free them. 507dce8aebaSBarry Smith 508dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 50940d8ff71SMatthew G. Knepley @*/ 510d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetData(PetscQuadrature q, PetscInt dim, PetscInt Nc, PetscInt npoints, const PetscReal points[], const PetscReal weights[]) 511d71ae5a4SJacob Faibussowitsch { 51221454ff5SMatthew G. Knepley PetscFunctionBegin; 5132cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 51421454ff5SMatthew G. Knepley if (dim >= 0) q->dim = dim; 515a6b92713SMatthew G. Knepley if (Nc >= 0) q->Nc = Nc; 51621454ff5SMatthew G. Knepley if (npoints >= 0) q->numPoints = npoints; 51721454ff5SMatthew G. Knepley if (points) { 518dadcf809SJacob Faibussowitsch PetscValidRealPointer(points, 5); 51921454ff5SMatthew G. Knepley q->points = points; 52021454ff5SMatthew G. Knepley } 52121454ff5SMatthew G. Knepley if (weights) { 522dadcf809SJacob Faibussowitsch PetscValidRealPointer(weights, 6); 52321454ff5SMatthew G. Knepley q->weights = weights; 52421454ff5SMatthew G. Knepley } 5253ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 526f9fd7fdbSMatthew G. Knepley } 527f9fd7fdbSMatthew G. Knepley 528d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscQuadratureView_Ascii(PetscQuadrature quad, PetscViewer v) 529d71ae5a4SJacob Faibussowitsch { 530d9bac1caSLisandro Dalcin PetscInt q, d, c; 531d9bac1caSLisandro Dalcin PetscViewerFormat format; 532d9bac1caSLisandro Dalcin 533d9bac1caSLisandro Dalcin PetscFunctionBegin; 53463a3b9bcSJacob Faibussowitsch if (quad->Nc > 1) PetscCall(PetscViewerASCIIPrintf(v, "Quadrature of order %" PetscInt_FMT " on %" PetscInt_FMT " points (dim %" PetscInt_FMT ") with %" PetscInt_FMT " components\n", quad->order, quad->numPoints, quad->dim, quad->Nc)); 53563a3b9bcSJacob Faibussowitsch else PetscCall(PetscViewerASCIIPrintf(v, "Quadrature of order %" PetscInt_FMT " on %" PetscInt_FMT " points (dim %" PetscInt_FMT ")\n", quad->order, quad->numPoints, quad->dim)); 5369566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(v, &format)); 5373ba16761SJacob Faibussowitsch if (format != PETSC_VIEWER_ASCII_INFO_DETAIL) PetscFunctionReturn(PETSC_SUCCESS); 538d9bac1caSLisandro Dalcin for (q = 0; q < quad->numPoints; ++q) { 53963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "p%" PetscInt_FMT " (", q)); 5409566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(v, PETSC_FALSE)); 541d9bac1caSLisandro Dalcin for (d = 0; d < quad->dim; ++d) { 5429566063dSJacob Faibussowitsch if (d) PetscCall(PetscViewerASCIIPrintf(v, ", ")); 5439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "%+g", (double)quad->points[q * quad->dim + d])); 544d9bac1caSLisandro Dalcin } 5459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, ") ")); 54663a3b9bcSJacob Faibussowitsch if (quad->Nc > 1) PetscCall(PetscViewerASCIIPrintf(v, "w%" PetscInt_FMT " (", q)); 547d9bac1caSLisandro Dalcin for (c = 0; c < quad->Nc; ++c) { 5489566063dSJacob Faibussowitsch if (c) PetscCall(PetscViewerASCIIPrintf(v, ", ")); 5499566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "%+g", (double)quad->weights[q * quad->Nc + c])); 550d9bac1caSLisandro Dalcin } 5519566063dSJacob Faibussowitsch if (quad->Nc > 1) PetscCall(PetscViewerASCIIPrintf(v, ")")); 5529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "\n")); 5539566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(v, PETSC_TRUE)); 554d9bac1caSLisandro Dalcin } 5553ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 556d9bac1caSLisandro Dalcin } 557d9bac1caSLisandro Dalcin 55840d8ff71SMatthew G. Knepley /*@C 559dce8aebaSBarry Smith PetscQuadratureView - View a `PetscQuadrature` object 56040d8ff71SMatthew G. Knepley 561*20f4b53cSBarry Smith Collective 56240d8ff71SMatthew G. Knepley 56340d8ff71SMatthew G. Knepley Input Parameters: 564dce8aebaSBarry Smith + quad - The `PetscQuadrature` object 565dce8aebaSBarry Smith - viewer - The `PetscViewer` object 56640d8ff71SMatthew G. Knepley 56740d8ff71SMatthew G. Knepley Level: beginner 56840d8ff71SMatthew G. Knepley 569dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscViewer`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 57040d8ff71SMatthew G. Knepley @*/ 571d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureView(PetscQuadrature quad, PetscViewer viewer) 572d71ae5a4SJacob Faibussowitsch { 573d9bac1caSLisandro Dalcin PetscBool iascii; 574f9fd7fdbSMatthew G. Knepley 575f9fd7fdbSMatthew G. Knepley PetscFunctionBegin; 576d9bac1caSLisandro Dalcin PetscValidHeader(quad, 1); 577d9bac1caSLisandro Dalcin if (viewer) PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 5789566063dSJacob Faibussowitsch if (!viewer) PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)quad), &viewer)); 5799566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii)); 5809566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5819566063dSJacob Faibussowitsch if (iascii) PetscCall(PetscQuadratureView_Ascii(quad, viewer)); 5829566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 5833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 584bfa639d9SMatthew G. Knepley } 585bfa639d9SMatthew G. Knepley 58689710940SMatthew G. Knepley /*@C 58789710940SMatthew G. Knepley PetscQuadratureExpandComposite - Return a quadrature over the composite element, which has the original quadrature in each subelement 58889710940SMatthew G. Knepley 589*20f4b53cSBarry Smith Not Collective; No Fortran Support 59089710940SMatthew G. Knepley 591d8d19677SJose E. Roman Input Parameters: 592dce8aebaSBarry Smith + q - The original `PetscQuadrature` 59389710940SMatthew G. Knepley . numSubelements - The number of subelements the original element is divided into 59489710940SMatthew G. Knepley . v0 - An array of the initial points for each subelement 59589710940SMatthew G. Knepley - jac - An array of the Jacobian mappings from the reference to each subelement 59689710940SMatthew G. Knepley 59789710940SMatthew G. Knepley Output Parameters: 59889710940SMatthew G. Knepley . dim - The dimension 59989710940SMatthew G. Knepley 600*20f4b53cSBarry Smith Level: intermediate 601*20f4b53cSBarry Smith 602dce8aebaSBarry Smith Note: 603dce8aebaSBarry Smith Together v0 and jac define an affine mapping from the original reference element to each subelement 60489710940SMatthew G. Knepley 605dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscFECreate()`, `PetscSpaceGetDimension()`, `PetscDualSpaceGetDimension()` 60689710940SMatthew G. Knepley @*/ 607d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureExpandComposite(PetscQuadrature q, PetscInt numSubelements, const PetscReal v0[], const PetscReal jac[], PetscQuadrature *qref) 608d71ae5a4SJacob Faibussowitsch { 60989710940SMatthew G. Knepley const PetscReal *points, *weights; 61089710940SMatthew G. Knepley PetscReal *pointsRef, *weightsRef; 611a6b92713SMatthew G. Knepley PetscInt dim, Nc, order, npoints, npointsRef, c, p, cp, d, e; 61289710940SMatthew G. Knepley 61389710940SMatthew G. Knepley PetscFunctionBegin; 6142cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 615dadcf809SJacob Faibussowitsch PetscValidRealPointer(v0, 3); 616dadcf809SJacob Faibussowitsch PetscValidRealPointer(jac, 4); 61789710940SMatthew G. Knepley PetscValidPointer(qref, 5); 6189566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, qref)); 6199566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q, &order)); 6209566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &npoints, &points, &weights)); 62189710940SMatthew G. Knepley npointsRef = npoints * numSubelements; 6229566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npointsRef * dim, &pointsRef)); 6239566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npointsRef * Nc, &weightsRef)); 62489710940SMatthew G. Knepley for (c = 0; c < numSubelements; ++c) { 62589710940SMatthew G. Knepley for (p = 0; p < npoints; ++p) { 62689710940SMatthew G. Knepley for (d = 0; d < dim; ++d) { 62789710940SMatthew G. Knepley pointsRef[(c * npoints + p) * dim + d] = v0[c * dim + d]; 628ad540459SPierre Jolivet for (e = 0; e < dim; ++e) pointsRef[(c * npoints + p) * dim + d] += jac[(c * dim + d) * dim + e] * (points[p * dim + e] + 1.0); 62989710940SMatthew G. Knepley } 63089710940SMatthew G. Knepley /* Could also use detJ here */ 631a6b92713SMatthew G. Knepley for (cp = 0; cp < Nc; ++cp) weightsRef[(c * npoints + p) * Nc + cp] = weights[p * Nc + cp] / numSubelements; 63289710940SMatthew G. Knepley } 63389710940SMatthew G. Knepley } 6349566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*qref, order)); 6359566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*qref, dim, Nc, npointsRef, pointsRef, weightsRef)); 6363ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 63789710940SMatthew G. Knepley } 63889710940SMatthew G. Knepley 63994e21283SToby Isaac /* Compute the coefficients for the Jacobi polynomial recurrence, 64094e21283SToby Isaac * 64194e21283SToby Isaac * J^{a,b}_n(x) = (cnm1 + cnm1x * x) * J^{a,b}_{n-1}(x) - cnm2 * J^{a,b}_{n-2}(x). 64294e21283SToby Isaac */ 64394e21283SToby Isaac #define PetscDTJacobiRecurrence_Internal(n, a, b, cnm1, cnm1x, cnm2) \ 64494e21283SToby Isaac do { \ 64594e21283SToby Isaac PetscReal _a = (a); \ 64694e21283SToby Isaac PetscReal _b = (b); \ 64794e21283SToby Isaac PetscReal _n = (n); \ 64894e21283SToby Isaac if (n == 1) { \ 64994e21283SToby Isaac (cnm1) = (_a - _b) * 0.5; \ 65094e21283SToby Isaac (cnm1x) = (_a + _b + 2.) * 0.5; \ 65194e21283SToby Isaac (cnm2) = 0.; \ 65294e21283SToby Isaac } else { \ 65394e21283SToby Isaac PetscReal _2n = _n + _n; \ 65494e21283SToby Isaac PetscReal _d = (_2n * (_n + _a + _b) * (_2n + _a + _b - 2)); \ 65594e21283SToby Isaac PetscReal _n1 = (_2n + _a + _b - 1.) * (_a * _a - _b * _b); \ 65694e21283SToby Isaac PetscReal _n1x = (_2n + _a + _b - 1.) * (_2n + _a + _b) * (_2n + _a + _b - 2); \ 65794e21283SToby Isaac PetscReal _n2 = 2. * ((_n + _a - 1.) * (_n + _b - 1.) * (_2n + _a + _b)); \ 65894e21283SToby Isaac (cnm1) = _n1 / _d; \ 65994e21283SToby Isaac (cnm1x) = _n1x / _d; \ 66094e21283SToby Isaac (cnm2) = _n2 / _d; \ 66194e21283SToby Isaac } \ 66294e21283SToby Isaac } while (0) 66394e21283SToby Isaac 664fbdc3dfeSToby Isaac /*@ 665fbdc3dfeSToby Isaac PetscDTJacobiNorm - Compute the weighted L2 norm of a Jacobi polynomial. 666fbdc3dfeSToby Isaac 667fbdc3dfeSToby Isaac $\| P^{\alpha,\beta}_n \|_{\alpha,\beta}^2 = \int_{-1}^1 (1 + x)^{\alpha} (1 - x)^{\beta} P^{\alpha,\beta}_n (x)^2 dx.$ 668fbdc3dfeSToby Isaac 6694165533cSJose E. Roman Input Parameters: 670fbdc3dfeSToby Isaac - alpha - the left exponent > -1 671fbdc3dfeSToby Isaac . beta - the right exponent > -1 672fbdc3dfeSToby Isaac + n - the polynomial degree 673fbdc3dfeSToby Isaac 6744165533cSJose E. Roman Output Parameter: 675fbdc3dfeSToby Isaac . norm - the weighted L2 norm 676fbdc3dfeSToby Isaac 677fbdc3dfeSToby Isaac Level: beginner 678fbdc3dfeSToby Isaac 679dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTJacobiEval()` 680fbdc3dfeSToby Isaac @*/ 681d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiNorm(PetscReal alpha, PetscReal beta, PetscInt n, PetscReal *norm) 682d71ae5a4SJacob Faibussowitsch { 683fbdc3dfeSToby Isaac PetscReal twoab1; 684fbdc3dfeSToby Isaac PetscReal gr; 685fbdc3dfeSToby Isaac 686fbdc3dfeSToby Isaac PetscFunctionBegin; 68708401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Exponent alpha %g <= -1. invalid", (double)alpha); 68808401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Exponent beta %g <= -1. invalid", (double)beta); 68963a3b9bcSJacob Faibussowitsch PetscCheck(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "n %" PetscInt_FMT " < 0 invalid", n); 690fbdc3dfeSToby Isaac twoab1 = PetscPowReal(2., alpha + beta + 1.); 691fbdc3dfeSToby Isaac #if defined(PETSC_HAVE_LGAMMA) 692fbdc3dfeSToby Isaac if (!n) { 693fbdc3dfeSToby Isaac gr = PetscExpReal(PetscLGamma(alpha + 1.) + PetscLGamma(beta + 1.) - PetscLGamma(alpha + beta + 2.)); 694fbdc3dfeSToby Isaac } else { 695fbdc3dfeSToby Isaac gr = PetscExpReal(PetscLGamma(n + alpha + 1.) + PetscLGamma(n + beta + 1.) - (PetscLGamma(n + 1.) + PetscLGamma(n + alpha + beta + 1.))) / (n + n + alpha + beta + 1.); 696fbdc3dfeSToby Isaac } 697fbdc3dfeSToby Isaac #else 698fbdc3dfeSToby Isaac { 699fbdc3dfeSToby Isaac PetscInt alphai = (PetscInt)alpha; 700fbdc3dfeSToby Isaac PetscInt betai = (PetscInt)beta; 701fbdc3dfeSToby Isaac PetscInt i; 702fbdc3dfeSToby Isaac 703fbdc3dfeSToby Isaac gr = n ? (1. / (n + n + alpha + beta + 1.)) : 1.; 704fbdc3dfeSToby Isaac if ((PetscReal)alphai == alpha) { 705fbdc3dfeSToby Isaac if (!n) { 706fbdc3dfeSToby Isaac for (i = 0; i < alphai; i++) gr *= (i + 1.) / (beta + i + 1.); 707fbdc3dfeSToby Isaac gr /= (alpha + beta + 1.); 708fbdc3dfeSToby Isaac } else { 709fbdc3dfeSToby Isaac for (i = 0; i < alphai; i++) gr *= (n + i + 1.) / (n + beta + i + 1.); 710fbdc3dfeSToby Isaac } 711fbdc3dfeSToby Isaac } else if ((PetscReal)betai == beta) { 712fbdc3dfeSToby Isaac if (!n) { 713fbdc3dfeSToby Isaac for (i = 0; i < betai; i++) gr *= (i + 1.) / (alpha + i + 2.); 714fbdc3dfeSToby Isaac gr /= (alpha + beta + 1.); 715fbdc3dfeSToby Isaac } else { 716fbdc3dfeSToby Isaac for (i = 0; i < betai; i++) gr *= (n + i + 1.) / (n + alpha + i + 1.); 717fbdc3dfeSToby Isaac } 718fbdc3dfeSToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 719fbdc3dfeSToby Isaac } 720fbdc3dfeSToby Isaac #endif 721fbdc3dfeSToby Isaac *norm = PetscSqrtReal(twoab1 * gr); 7223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 723fbdc3dfeSToby Isaac } 724fbdc3dfeSToby Isaac 725d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobiEval_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscInt k, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *p) 726d71ae5a4SJacob Faibussowitsch { 72794e21283SToby Isaac PetscReal ak, bk; 72894e21283SToby Isaac PetscReal abk1; 72994e21283SToby Isaac PetscInt i, l, maxdegree; 73094e21283SToby Isaac 73194e21283SToby Isaac PetscFunctionBegin; 73294e21283SToby Isaac maxdegree = degrees[ndegree - 1] - k; 73394e21283SToby Isaac ak = a + k; 73494e21283SToby Isaac bk = b + k; 73594e21283SToby Isaac abk1 = a + b + k + 1.; 73694e21283SToby Isaac if (maxdegree < 0) { 7379371c9d4SSatish Balay for (i = 0; i < npoints; i++) 7389371c9d4SSatish Balay for (l = 0; l < ndegree; l++) p[i * ndegree + l] = 0.; 7393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 74094e21283SToby Isaac } 74194e21283SToby Isaac for (i = 0; i < npoints; i++) { 74294e21283SToby Isaac PetscReal pm1, pm2, x; 74394e21283SToby Isaac PetscReal cnm1, cnm1x, cnm2; 74494e21283SToby Isaac PetscInt j, m; 74594e21283SToby Isaac 74694e21283SToby Isaac x = points[i]; 74794e21283SToby Isaac pm2 = 1.; 74894e21283SToby Isaac PetscDTJacobiRecurrence_Internal(1, ak, bk, cnm1, cnm1x, cnm2); 74994e21283SToby Isaac pm1 = (cnm1 + cnm1x * x); 75094e21283SToby Isaac l = 0; 751ad540459SPierre Jolivet while (l < ndegree && degrees[l] - k < 0) p[l++] = 0.; 75294e21283SToby Isaac while (l < ndegree && degrees[l] - k == 0) { 75394e21283SToby Isaac p[l] = pm2; 75494e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + m) * 0.5; 75594e21283SToby Isaac l++; 75694e21283SToby Isaac } 75794e21283SToby Isaac while (l < ndegree && degrees[l] - k == 1) { 75894e21283SToby Isaac p[l] = pm1; 75994e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + 1 + m) * 0.5; 76094e21283SToby Isaac l++; 76194e21283SToby Isaac } 76294e21283SToby Isaac for (j = 2; j <= maxdegree; j++) { 76394e21283SToby Isaac PetscReal pp; 76494e21283SToby Isaac 76594e21283SToby Isaac PetscDTJacobiRecurrence_Internal(j, ak, bk, cnm1, cnm1x, cnm2); 76694e21283SToby Isaac pp = (cnm1 + cnm1x * x) * pm1 - cnm2 * pm2; 76794e21283SToby Isaac pm2 = pm1; 76894e21283SToby Isaac pm1 = pp; 76994e21283SToby Isaac while (l < ndegree && degrees[l] - k == j) { 77094e21283SToby Isaac p[l] = pp; 77194e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + j + m) * 0.5; 77294e21283SToby Isaac l++; 77394e21283SToby Isaac } 77494e21283SToby Isaac } 77594e21283SToby Isaac p += ndegree; 77694e21283SToby Isaac } 7773ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 77894e21283SToby Isaac } 77994e21283SToby Isaac 78037045ce4SJed Brown /*@ 781dce8aebaSBarry Smith PetscDTJacobiEvalJet - Evaluate the jet (function and derivatives) of the Jacobi polynomials basis up to a given degree. 782dce8aebaSBarry Smith The Jacobi polynomials with indices $\alpha$ and $\beta$ are orthogonal with respect to the weighted inner product 783dce8aebaSBarry Smith $\langle f, g \rangle = \int_{-1}^1 (1+x)^{\alpha} (1-x)^{\beta} f(x) g(x) dx$. 784fbdc3dfeSToby Isaac 7854165533cSJose E. Roman Input Parameters: 786fbdc3dfeSToby Isaac + alpha - the left exponent of the weight 787fbdc3dfeSToby Isaac . beta - the right exponetn of the weight 788fbdc3dfeSToby Isaac . npoints - the number of points to evaluate the polynomials at 789fbdc3dfeSToby Isaac . points - [npoints] array of point coordinates 790fbdc3dfeSToby Isaac . degree - the maximm degree polynomial space to evaluate, (degree + 1) will be evaluated total. 791fbdc3dfeSToby Isaac - k - the maximum derivative to evaluate in the jet, (k + 1) will be evaluated total. 792fbdc3dfeSToby Isaac 7936aad120cSJose E. Roman Output Parameters: 794fbdc3dfeSToby Isaac - p - an array containing the evaluations of the Jacobi polynomials's jets on the points. the size is (degree + 1) x 795fbdc3dfeSToby Isaac (k + 1) x npoints, which also describes the order of the dimensions of this three-dimensional array: the first 796fbdc3dfeSToby Isaac (slowest varying) dimension is polynomial degree; the second dimension is derivative order; the third (fastest 797fbdc3dfeSToby Isaac varying) dimension is the index of the evaluation point. 798fbdc3dfeSToby Isaac 799fbdc3dfeSToby Isaac Level: advanced 800fbdc3dfeSToby Isaac 801db781477SPatrick Sanan .seealso: `PetscDTJacobiEval()`, `PetscDTPKDEvalJet()` 802fbdc3dfeSToby Isaac @*/ 803d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiEvalJet(PetscReal alpha, PetscReal beta, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt k, PetscReal p[]) 804d71ae5a4SJacob Faibussowitsch { 805fbdc3dfeSToby Isaac PetscInt i, j, l; 806fbdc3dfeSToby Isaac PetscInt *degrees; 807fbdc3dfeSToby Isaac PetscReal *psingle; 808fbdc3dfeSToby Isaac 809fbdc3dfeSToby Isaac PetscFunctionBegin; 810fbdc3dfeSToby Isaac if (degree == 0) { 811fbdc3dfeSToby Isaac PetscInt zero = 0; 812fbdc3dfeSToby Isaac 81348a46eb9SPierre Jolivet for (i = 0; i <= k; i++) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, i, points, 1, &zero, &p[i * npoints])); 8143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 815fbdc3dfeSToby Isaac } 8169566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(degree + 1, °rees)); 8179566063dSJacob Faibussowitsch PetscCall(PetscMalloc1((degree + 1) * npoints, &psingle)); 818fbdc3dfeSToby Isaac for (i = 0; i <= degree; i++) degrees[i] = i; 819fbdc3dfeSToby Isaac for (i = 0; i <= k; i++) { 8209566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, i, points, degree + 1, degrees, psingle)); 821fbdc3dfeSToby Isaac for (j = 0; j <= degree; j++) { 822ad540459SPierre Jolivet for (l = 0; l < npoints; l++) p[(j * (k + 1) + i) * npoints + l] = psingle[l * (degree + 1) + j]; 823fbdc3dfeSToby Isaac } 824fbdc3dfeSToby Isaac } 8259566063dSJacob Faibussowitsch PetscCall(PetscFree(psingle)); 8269566063dSJacob Faibussowitsch PetscCall(PetscFree(degrees)); 8273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 828fbdc3dfeSToby Isaac } 829fbdc3dfeSToby Isaac 830fbdc3dfeSToby Isaac /*@ 831dce8aebaSBarry Smith PetscDTJacobiEval - evaluate Jacobi polynomials for the weight function $(1.+x)^{\alpha} (1.-x)^{\beta}$ at a set of points 83294e21283SToby Isaac at points 83394e21283SToby Isaac 83494e21283SToby Isaac Not Collective 83594e21283SToby Isaac 8364165533cSJose E. Roman Input Parameters: 83794e21283SToby Isaac + npoints - number of spatial points to evaluate at 83894e21283SToby Isaac . alpha - the left exponent > -1 83994e21283SToby Isaac . beta - the right exponent > -1 84094e21283SToby Isaac . points - array of locations to evaluate at 84194e21283SToby Isaac . ndegree - number of basis degrees to evaluate 84294e21283SToby Isaac - degrees - sorted array of degrees to evaluate 84394e21283SToby Isaac 8444165533cSJose E. Roman Output Parameters: 84594e21283SToby Isaac + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or NULL) 84694e21283SToby Isaac . D - row-oriented derivative evaluation matrix (or NULL) 84794e21283SToby Isaac - D2 - row-oriented second derivative evaluation matrix (or NULL) 84894e21283SToby Isaac 84994e21283SToby Isaac Level: intermediate 85094e21283SToby Isaac 851dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()`, `PetscDTLegendreEval()` 85294e21283SToby Isaac @*/ 853d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiEval(PetscInt npoints, PetscReal alpha, PetscReal beta, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *B, PetscReal *D, PetscReal *D2) 854d71ae5a4SJacob Faibussowitsch { 85594e21283SToby Isaac PetscFunctionBegin; 85608401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 85708401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 8583ba16761SJacob Faibussowitsch if (!npoints || !ndegree) PetscFunctionReturn(PETSC_SUCCESS); 8599566063dSJacob Faibussowitsch if (B) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 0, points, ndegree, degrees, B)); 8609566063dSJacob Faibussowitsch if (D) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 1, points, ndegree, degrees, D)); 8619566063dSJacob Faibussowitsch if (D2) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 2, points, ndegree, degrees, D2)); 8623ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 86394e21283SToby Isaac } 86494e21283SToby Isaac 86594e21283SToby Isaac /*@ 86694e21283SToby Isaac PetscDTLegendreEval - evaluate Legendre polynomials at points 86737045ce4SJed Brown 86837045ce4SJed Brown Not Collective 86937045ce4SJed Brown 8704165533cSJose E. Roman Input Parameters: 87137045ce4SJed Brown + npoints - number of spatial points to evaluate at 87237045ce4SJed Brown . points - array of locations to evaluate at 87337045ce4SJed Brown . ndegree - number of basis degrees to evaluate 87437045ce4SJed Brown - degrees - sorted array of degrees to evaluate 87537045ce4SJed Brown 8764165533cSJose E. Roman Output Parameters: 8770298fd71SBarry Smith + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or NULL) 8780298fd71SBarry Smith . D - row-oriented derivative evaluation matrix (or NULL) 8790298fd71SBarry Smith - D2 - row-oriented second derivative evaluation matrix (or NULL) 88037045ce4SJed Brown 88137045ce4SJed Brown Level: intermediate 88237045ce4SJed Brown 883db781477SPatrick Sanan .seealso: `PetscDTGaussQuadrature()` 88437045ce4SJed Brown @*/ 885d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTLegendreEval(PetscInt npoints, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *B, PetscReal *D, PetscReal *D2) 886d71ae5a4SJacob Faibussowitsch { 88737045ce4SJed Brown PetscFunctionBegin; 8889566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiEval(npoints, 0., 0., points, ndegree, degrees, B, D, D2)); 8893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 89037045ce4SJed Brown } 89137045ce4SJed Brown 892fbdc3dfeSToby Isaac /*@ 893fbdc3dfeSToby Isaac PetscDTIndexToGradedOrder - convert an index into a tuple of monomial degrees in a graded order (that is, if the degree sum of tuple x is less than the degree sum of tuple y, then the index of x is smaller than the index of y) 894fbdc3dfeSToby Isaac 895fbdc3dfeSToby Isaac Input Parameters: 896fbdc3dfeSToby Isaac + len - the desired length of the degree tuple 897fbdc3dfeSToby Isaac - index - the index to convert: should be >= 0 898fbdc3dfeSToby Isaac 899fbdc3dfeSToby Isaac Output Parameter: 900fbdc3dfeSToby Isaac . degtup - will be filled with a tuple of degrees 901fbdc3dfeSToby Isaac 902fbdc3dfeSToby Isaac Level: beginner 903fbdc3dfeSToby Isaac 904dce8aebaSBarry Smith Note: 905dce8aebaSBarry Smith For two tuples x and y with the same degree sum, partial degree sums over the final elements of the tuples 906fbdc3dfeSToby Isaac acts as a tiebreaker. For example, (2, 1, 1) and (1, 2, 1) have the same degree sum, but the degree sum over the 907fbdc3dfeSToby Isaac last two elements is smaller for the former, so (2, 1, 1) < (1, 2, 1). 908fbdc3dfeSToby Isaac 909db781477SPatrick Sanan .seealso: `PetscDTGradedOrderToIndex()` 910fbdc3dfeSToby Isaac @*/ 911d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTIndexToGradedOrder(PetscInt len, PetscInt index, PetscInt degtup[]) 912d71ae5a4SJacob Faibussowitsch { 913fbdc3dfeSToby Isaac PetscInt i, total; 914fbdc3dfeSToby Isaac PetscInt sum; 915fbdc3dfeSToby Isaac 916fbdc3dfeSToby Isaac PetscFunctionBeginHot; 91708401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 91808401ef6SPierre Jolivet PetscCheck(index >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index must be non-negative"); 919fbdc3dfeSToby Isaac total = 1; 920fbdc3dfeSToby Isaac sum = 0; 921fbdc3dfeSToby Isaac while (index >= total) { 922fbdc3dfeSToby Isaac index -= total; 923fbdc3dfeSToby Isaac total = (total * (len + sum)) / (sum + 1); 924fbdc3dfeSToby Isaac sum++; 925fbdc3dfeSToby Isaac } 926fbdc3dfeSToby Isaac for (i = 0; i < len; i++) { 927fbdc3dfeSToby Isaac PetscInt c; 928fbdc3dfeSToby Isaac 929fbdc3dfeSToby Isaac degtup[i] = sum; 930fbdc3dfeSToby Isaac for (c = 0, total = 1; c < sum; c++) { 931fbdc3dfeSToby Isaac /* going into the loop, total is the number of way to have a tuple of sum exactly c with length len - 1 - i */ 932fbdc3dfeSToby Isaac if (index < total) break; 933fbdc3dfeSToby Isaac index -= total; 934fbdc3dfeSToby Isaac total = (total * (len - 1 - i + c)) / (c + 1); 935fbdc3dfeSToby Isaac degtup[i]--; 936fbdc3dfeSToby Isaac } 937fbdc3dfeSToby Isaac sum -= degtup[i]; 938fbdc3dfeSToby Isaac } 9393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 940fbdc3dfeSToby Isaac } 941fbdc3dfeSToby Isaac 942fbdc3dfeSToby Isaac /*@ 943dce8aebaSBarry Smith PetscDTGradedOrderToIndex - convert a tuple into an index in a graded order, the inverse of `PetscDTIndexToGradedOrder()`. 944fbdc3dfeSToby Isaac 945fbdc3dfeSToby Isaac Input Parameters: 946fbdc3dfeSToby Isaac + len - the length of the degree tuple 947fbdc3dfeSToby Isaac - degtup - tuple with this length 948fbdc3dfeSToby Isaac 949fbdc3dfeSToby Isaac Output Parameter: 950fbdc3dfeSToby Isaac . index - index in graded order: >= 0 951fbdc3dfeSToby Isaac 952fbdc3dfeSToby Isaac Level: Beginner 953fbdc3dfeSToby Isaac 954dce8aebaSBarry Smith Note: 955dce8aebaSBarry Smith For two tuples x and y with the same degree sum, partial degree sums over the final elements of the tuples 956fbdc3dfeSToby Isaac acts as a tiebreaker. For example, (2, 1, 1) and (1, 2, 1) have the same degree sum, but the degree sum over the 957fbdc3dfeSToby Isaac last two elements is smaller for the former, so (2, 1, 1) < (1, 2, 1). 958fbdc3dfeSToby Isaac 959db781477SPatrick Sanan .seealso: `PetscDTIndexToGradedOrder()` 960fbdc3dfeSToby Isaac @*/ 961d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGradedOrderToIndex(PetscInt len, const PetscInt degtup[], PetscInt *index) 962d71ae5a4SJacob Faibussowitsch { 963fbdc3dfeSToby Isaac PetscInt i, idx, sum, total; 964fbdc3dfeSToby Isaac 965fbdc3dfeSToby Isaac PetscFunctionBeginHot; 96608401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 967fbdc3dfeSToby Isaac for (i = 0, sum = 0; i < len; i++) sum += degtup[i]; 968fbdc3dfeSToby Isaac idx = 0; 969fbdc3dfeSToby Isaac total = 1; 970fbdc3dfeSToby Isaac for (i = 0; i < sum; i++) { 971fbdc3dfeSToby Isaac idx += total; 972fbdc3dfeSToby Isaac total = (total * (len + i)) / (i + 1); 973fbdc3dfeSToby Isaac } 974fbdc3dfeSToby Isaac for (i = 0; i < len - 1; i++) { 975fbdc3dfeSToby Isaac PetscInt c; 976fbdc3dfeSToby Isaac 977fbdc3dfeSToby Isaac total = 1; 978fbdc3dfeSToby Isaac sum -= degtup[i]; 979fbdc3dfeSToby Isaac for (c = 0; c < sum; c++) { 980fbdc3dfeSToby Isaac idx += total; 981fbdc3dfeSToby Isaac total = (total * (len - 1 - i + c)) / (c + 1); 982fbdc3dfeSToby Isaac } 983fbdc3dfeSToby Isaac } 984fbdc3dfeSToby Isaac *index = idx; 9853ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 986fbdc3dfeSToby Isaac } 987fbdc3dfeSToby Isaac 988e3aa2e09SToby Isaac static PetscBool PKDCite = PETSC_FALSE; 989e3aa2e09SToby Isaac const char PKDCitation[] = "@article{Kirby2010,\n" 990e3aa2e09SToby Isaac " title={Singularity-free evaluation of collapsed-coordinate orthogonal polynomials},\n" 991e3aa2e09SToby Isaac " author={Kirby, Robert C},\n" 992e3aa2e09SToby Isaac " journal={ACM Transactions on Mathematical Software (TOMS)},\n" 993e3aa2e09SToby Isaac " volume={37},\n" 994e3aa2e09SToby Isaac " number={1},\n" 995e3aa2e09SToby Isaac " pages={1--16},\n" 996e3aa2e09SToby Isaac " year={2010},\n" 997e3aa2e09SToby Isaac " publisher={ACM New York, NY, USA}\n}\n"; 998e3aa2e09SToby Isaac 999fbdc3dfeSToby Isaac /*@ 1000d8f25ad8SToby Isaac PetscDTPKDEvalJet - Evaluate the jet (function and derivatives) of the Proriol-Koornwinder-Dubiner (PKD) basis for 1001fbdc3dfeSToby Isaac the space of polynomials up to a given degree. The PKD basis is L2-orthonormal on the biunit simplex (which is used 1002fbdc3dfeSToby Isaac as the reference element for finite elements in PETSc), which makes it a stable basis to use for evaluating 1003fbdc3dfeSToby Isaac polynomials in that domain. 1004fbdc3dfeSToby Isaac 10054165533cSJose E. Roman Input Parameters: 1006fbdc3dfeSToby Isaac + dim - the number of variables in the multivariate polynomials 1007fbdc3dfeSToby Isaac . npoints - the number of points to evaluate the polynomials at 1008fbdc3dfeSToby Isaac . points - [npoints x dim] array of point coordinates 1009fbdc3dfeSToby Isaac . degree - the degree (sum of degrees on the variables in a monomial) of the polynomial space to evaluate. There are ((dim + degree) choose dim) polynomials in this space. 1010fbdc3dfeSToby Isaac - k - the maximum order partial derivative to evaluate in the jet. There are (dim + k choose dim) partial derivatives 1011fbdc3dfeSToby Isaac in the jet. Choosing k = 0 means to evaluate just the function and no derivatives 1012fbdc3dfeSToby Isaac 10136aad120cSJose E. Roman Output Parameters: 1014fbdc3dfeSToby Isaac - p - an array containing the evaluations of the PKD polynomials' jets on the points. The size is ((dim + degree) 1015fbdc3dfeSToby Isaac choose dim) x ((dim + k) choose dim) x npoints, which also describes the order of the dimensions of this 1016fbdc3dfeSToby Isaac three-dimensional array: the first (slowest varying) dimension is basis function index; the second dimension is jet 1017fbdc3dfeSToby Isaac index; the third (fastest varying) dimension is the index of the evaluation point. 1018fbdc3dfeSToby Isaac 1019fbdc3dfeSToby Isaac Level: advanced 1020fbdc3dfeSToby Isaac 1021dce8aebaSBarry Smith Notes: 1022dce8aebaSBarry Smith The ordering of the basis functions, and the ordering of the derivatives in the jet, both follow the graded 1023dce8aebaSBarry Smith ordering of `PetscDTIndexToGradedOrder()` and `PetscDTGradedOrderToIndex()`. For example, in 3D, the polynomial with 1024dce8aebaSBarry Smith leading monomial x^2,y^0,z^1, which has degree tuple (2,0,1), which by `PetscDTGradedOrderToIndex()` has index 12 (it is the 13th basis function in the space); 1025fbdc3dfeSToby Isaac the partial derivative $\partial_x \partial_z$ has order tuple (1,0,1), appears at index 6 in the jet (it is the 7th partial derivative in the jet). 1026fbdc3dfeSToby Isaac 1027e3aa2e09SToby Isaac The implementation uses Kirby's singularity-free evaluation algorithm, https://doi.org/10.1145/1644001.1644006. 1028e3aa2e09SToby Isaac 1029db781477SPatrick Sanan .seealso: `PetscDTGradedOrderToIndex()`, `PetscDTIndexToGradedOrder()`, `PetscDTJacobiEvalJet()` 1030fbdc3dfeSToby Isaac @*/ 1031d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPKDEvalJet(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt k, PetscReal p[]) 1032d71ae5a4SJacob Faibussowitsch { 1033fbdc3dfeSToby Isaac PetscInt degidx, kidx, d, pt; 1034fbdc3dfeSToby Isaac PetscInt Nk, Ndeg; 1035fbdc3dfeSToby Isaac PetscInt *ktup, *degtup; 1036fbdc3dfeSToby Isaac PetscReal *scales, initscale, scaleexp; 1037fbdc3dfeSToby Isaac 1038fbdc3dfeSToby Isaac PetscFunctionBegin; 10399566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(PKDCitation, &PKDCite)); 10409566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + k, k, &Nk)); 10419566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + dim, degree, &Ndeg)); 10429566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(dim, °tup, dim, &ktup)); 10439566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Ndeg, &scales)); 1044fbdc3dfeSToby Isaac initscale = 1.; 1045fbdc3dfeSToby Isaac if (dim > 1) { 10469566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(dim, 2, &scaleexp)); 10472f613bf5SBarry Smith initscale = PetscPowReal(2., scaleexp * 0.5); 1048fbdc3dfeSToby Isaac } 1049fbdc3dfeSToby Isaac for (degidx = 0; degidx < Ndeg; degidx++) { 1050fbdc3dfeSToby Isaac PetscInt e, i; 1051fbdc3dfeSToby Isaac PetscInt m1idx = -1, m2idx = -1; 1052fbdc3dfeSToby Isaac PetscInt n; 1053fbdc3dfeSToby Isaac PetscInt degsum; 1054fbdc3dfeSToby Isaac PetscReal alpha; 1055fbdc3dfeSToby Isaac PetscReal cnm1, cnm1x, cnm2; 1056fbdc3dfeSToby Isaac PetscReal norm; 1057fbdc3dfeSToby Isaac 10589566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, degidx, degtup)); 10599371c9d4SSatish Balay for (d = dim - 1; d >= 0; d--) 10609371c9d4SSatish Balay if (degtup[d]) break; 1061fbdc3dfeSToby Isaac if (d < 0) { /* constant is 1 everywhere, all derivatives are zero */ 1062fbdc3dfeSToby Isaac scales[degidx] = initscale; 1063fbdc3dfeSToby Isaac for (e = 0; e < dim; e++) { 10649566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiNorm(e, 0., 0, &norm)); 1065fbdc3dfeSToby Isaac scales[degidx] /= norm; 1066fbdc3dfeSToby Isaac } 1067fbdc3dfeSToby Isaac for (i = 0; i < npoints; i++) p[degidx * Nk * npoints + i] = 1.; 1068fbdc3dfeSToby Isaac for (i = 0; i < (Nk - 1) * npoints; i++) p[(degidx * Nk + 1) * npoints + i] = 0.; 1069fbdc3dfeSToby Isaac continue; 1070fbdc3dfeSToby Isaac } 1071fbdc3dfeSToby Isaac n = degtup[d]; 1072fbdc3dfeSToby Isaac degtup[d]--; 10739566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, degtup, &m1idx)); 1074fbdc3dfeSToby Isaac if (degtup[d] > 0) { 1075fbdc3dfeSToby Isaac degtup[d]--; 10769566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, degtup, &m2idx)); 1077fbdc3dfeSToby Isaac degtup[d]++; 1078fbdc3dfeSToby Isaac } 1079fbdc3dfeSToby Isaac degtup[d]++; 1080fbdc3dfeSToby Isaac for (e = 0, degsum = 0; e < d; e++) degsum += degtup[e]; 1081fbdc3dfeSToby Isaac alpha = 2 * degsum + d; 1082fbdc3dfeSToby Isaac PetscDTJacobiRecurrence_Internal(n, alpha, 0., cnm1, cnm1x, cnm2); 1083fbdc3dfeSToby Isaac 1084fbdc3dfeSToby Isaac scales[degidx] = initscale; 1085fbdc3dfeSToby Isaac for (e = 0, degsum = 0; e < dim; e++) { 1086fbdc3dfeSToby Isaac PetscInt f; 1087fbdc3dfeSToby Isaac PetscReal ealpha; 1088fbdc3dfeSToby Isaac PetscReal enorm; 1089fbdc3dfeSToby Isaac 1090fbdc3dfeSToby Isaac ealpha = 2 * degsum + e; 1091fbdc3dfeSToby Isaac for (f = 0; f < degsum; f++) scales[degidx] *= 2.; 10929566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiNorm(ealpha, 0., degtup[e], &enorm)); 1093fbdc3dfeSToby Isaac scales[degidx] /= enorm; 1094fbdc3dfeSToby Isaac degsum += degtup[e]; 1095fbdc3dfeSToby Isaac } 1096fbdc3dfeSToby Isaac 1097fbdc3dfeSToby Isaac for (pt = 0; pt < npoints; pt++) { 1098fbdc3dfeSToby Isaac /* compute the multipliers */ 1099fbdc3dfeSToby Isaac PetscReal thetanm1, thetanm1x, thetanm2; 1100fbdc3dfeSToby Isaac 1101fbdc3dfeSToby Isaac thetanm1x = dim - (d + 1) + 2. * points[pt * dim + d]; 1102fbdc3dfeSToby Isaac for (e = d + 1; e < dim; e++) thetanm1x += points[pt * dim + e]; 1103fbdc3dfeSToby Isaac thetanm1x *= 0.5; 1104fbdc3dfeSToby Isaac thetanm1 = (2. - (dim - (d + 1))); 1105fbdc3dfeSToby Isaac for (e = d + 1; e < dim; e++) thetanm1 -= points[pt * dim + e]; 1106fbdc3dfeSToby Isaac thetanm1 *= 0.5; 1107fbdc3dfeSToby Isaac thetanm2 = thetanm1 * thetanm1; 1108fbdc3dfeSToby Isaac 1109fbdc3dfeSToby Isaac for (kidx = 0; kidx < Nk; kidx++) { 1110fbdc3dfeSToby Isaac PetscInt f; 1111fbdc3dfeSToby Isaac 11129566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, kidx, ktup)); 1113fbdc3dfeSToby Isaac /* first sum in the same derivative terms */ 1114fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] = (cnm1 * thetanm1 + cnm1x * thetanm1x) * p[(m1idx * Nk + kidx) * npoints + pt]; 1115ad540459SPierre Jolivet if (m2idx >= 0) p[(degidx * Nk + kidx) * npoints + pt] -= cnm2 * thetanm2 * p[(m2idx * Nk + kidx) * npoints + pt]; 1116fbdc3dfeSToby Isaac 1117fbdc3dfeSToby Isaac for (f = d; f < dim; f++) { 1118fbdc3dfeSToby Isaac PetscInt km1idx, mplty = ktup[f]; 1119fbdc3dfeSToby Isaac 1120fbdc3dfeSToby Isaac if (!mplty) continue; 1121fbdc3dfeSToby Isaac ktup[f]--; 11229566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, ktup, &km1idx)); 1123fbdc3dfeSToby Isaac 1124fbdc3dfeSToby Isaac /* the derivative of cnm1x * thetanm1x wrt x variable f is 0.5 * cnm1x if f > d otherwise it is cnm1x */ 1125fbdc3dfeSToby Isaac /* the derivative of cnm1 * thetanm1 wrt x variable f is 0 if f == d, otherwise it is -0.5 * cnm1 */ 1126fbdc3dfeSToby Isaac /* the derivative of -cnm2 * thetanm2 wrt x variable f is 0 if f == d, otherwise it is cnm2 * thetanm1 */ 1127fbdc3dfeSToby Isaac if (f > d) { 1128fbdc3dfeSToby Isaac PetscInt f2; 1129fbdc3dfeSToby Isaac 1130fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * 0.5 * (cnm1x - cnm1) * p[(m1idx * Nk + km1idx) * npoints + pt]; 1131fbdc3dfeSToby Isaac if (m2idx >= 0) { 1132fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * cnm2 * thetanm1 * p[(m2idx * Nk + km1idx) * npoints + pt]; 1133fbdc3dfeSToby Isaac /* second derivatives of -cnm2 * thetanm2 wrt x variable f,f2 is like - 0.5 * cnm2 */ 1134fbdc3dfeSToby Isaac for (f2 = f; f2 < dim; f2++) { 1135fbdc3dfeSToby Isaac PetscInt km2idx, mplty2 = ktup[f2]; 1136fbdc3dfeSToby Isaac PetscInt factor; 1137fbdc3dfeSToby Isaac 1138fbdc3dfeSToby Isaac if (!mplty2) continue; 1139fbdc3dfeSToby Isaac ktup[f2]--; 11409566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, ktup, &km2idx)); 1141fbdc3dfeSToby Isaac 1142fbdc3dfeSToby Isaac factor = mplty * mplty2; 1143fbdc3dfeSToby Isaac if (f == f2) factor /= 2; 1144fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] -= 0.5 * factor * cnm2 * p[(m2idx * Nk + km2idx) * npoints + pt]; 1145fbdc3dfeSToby Isaac ktup[f2]++; 1146fbdc3dfeSToby Isaac } 11473034baaeSToby Isaac } 1148fbdc3dfeSToby Isaac } else { 1149fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * cnm1x * p[(m1idx * Nk + km1idx) * npoints + pt]; 1150fbdc3dfeSToby Isaac } 1151fbdc3dfeSToby Isaac ktup[f]++; 1152fbdc3dfeSToby Isaac } 1153fbdc3dfeSToby Isaac } 1154fbdc3dfeSToby Isaac } 1155fbdc3dfeSToby Isaac } 1156fbdc3dfeSToby Isaac for (degidx = 0; degidx < Ndeg; degidx++) { 1157fbdc3dfeSToby Isaac PetscReal scale = scales[degidx]; 1158fbdc3dfeSToby Isaac PetscInt i; 1159fbdc3dfeSToby Isaac 1160fbdc3dfeSToby Isaac for (i = 0; i < Nk * npoints; i++) p[degidx * Nk * npoints + i] *= scale; 1161fbdc3dfeSToby Isaac } 11629566063dSJacob Faibussowitsch PetscCall(PetscFree(scales)); 11639566063dSJacob Faibussowitsch PetscCall(PetscFree2(degtup, ktup)); 11643ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1165fbdc3dfeSToby Isaac } 1166fbdc3dfeSToby Isaac 1167d8f25ad8SToby Isaac /*@ 1168d8f25ad8SToby Isaac PetscDTPTrimmedSize - The size of the trimmed polynomial space of k-forms with a given degree and form degree, 1169dce8aebaSBarry Smith which can be evaluated in `PetscDTPTrimmedEvalJet()`. 1170d8f25ad8SToby Isaac 1171d8f25ad8SToby Isaac Input Parameters: 1172d8f25ad8SToby Isaac + dim - the number of variables in the multivariate polynomials 1173d8f25ad8SToby Isaac . degree - the degree (sum of degrees on the variables in a monomial) of the trimmed polynomial space. 1174d8f25ad8SToby Isaac - formDegree - the degree of the form 1175d8f25ad8SToby Isaac 11766aad120cSJose E. Roman Output Parameters: 1177*20f4b53cSBarry Smith - size - The number ((`dim` + `degree`) choose (`dim` + `formDegree`)) x ((`degree` + `formDegree` - 1) choose (`formDegree`)) 1178d8f25ad8SToby Isaac 1179d8f25ad8SToby Isaac Level: advanced 1180d8f25ad8SToby Isaac 1181db781477SPatrick Sanan .seealso: `PetscDTPTrimmedEvalJet()` 1182d8f25ad8SToby Isaac @*/ 1183d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPTrimmedSize(PetscInt dim, PetscInt degree, PetscInt formDegree, PetscInt *size) 1184d71ae5a4SJacob Faibussowitsch { 1185d8f25ad8SToby Isaac PetscInt Nrk, Nbpt; // number of trimmed polynomials 1186d8f25ad8SToby Isaac 1187d8f25ad8SToby Isaac PetscFunctionBegin; 1188d8f25ad8SToby Isaac formDegree = PetscAbsInt(formDegree); 11899566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + dim, degree + formDegree, &Nbpt)); 11909566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + formDegree - 1, formDegree, &Nrk)); 1191d8f25ad8SToby Isaac Nbpt *= Nrk; 1192d8f25ad8SToby Isaac *size = Nbpt; 11933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1194d8f25ad8SToby Isaac } 1195d8f25ad8SToby Isaac 1196d8f25ad8SToby Isaac /* there was a reference implementation based on section 4.4 of Arnold, Falk & Winther (acta numerica, 2006), but it 1197d8f25ad8SToby Isaac * was inferior to this implementation */ 1198d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPTrimmedEvalJet_Internal(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt formDegree, PetscInt jetDegree, PetscReal p[]) 1199d71ae5a4SJacob Faibussowitsch { 1200d8f25ad8SToby Isaac PetscInt formDegreeOrig = formDegree; 1201d8f25ad8SToby Isaac PetscBool formNegative = (formDegreeOrig < 0) ? PETSC_TRUE : PETSC_FALSE; 1202d8f25ad8SToby Isaac 1203d8f25ad8SToby Isaac PetscFunctionBegin; 1204d8f25ad8SToby Isaac formDegree = PetscAbsInt(formDegreeOrig); 1205d8f25ad8SToby Isaac if (formDegree == 0) { 12069566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree, jetDegree, p)); 12073ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1208d8f25ad8SToby Isaac } 1209d8f25ad8SToby Isaac if (formDegree == dim) { 12109566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree - 1, jetDegree, p)); 12113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1212d8f25ad8SToby Isaac } 1213d8f25ad8SToby Isaac PetscInt Nbpt; 12149566063dSJacob Faibussowitsch PetscCall(PetscDTPTrimmedSize(dim, degree, formDegree, &Nbpt)); 1215d8f25ad8SToby Isaac PetscInt Nf; 12169566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, formDegree, &Nf)); 1217d8f25ad8SToby Isaac PetscInt Nk; 12189566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + jetDegree, dim, &Nk)); 12199566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(p, Nbpt * Nf * Nk * npoints)); 1220d8f25ad8SToby Isaac 1221d8f25ad8SToby Isaac PetscInt Nbpm1; // number of scalar polynomials up to degree - 1; 12229566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + degree - 1, dim, &Nbpm1)); 1223d8f25ad8SToby Isaac PetscReal *p_scalar; 12249566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nbpm1 * Nk * npoints, &p_scalar)); 12259566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree - 1, jetDegree, p_scalar)); 1226d8f25ad8SToby Isaac PetscInt total = 0; 1227d8f25ad8SToby Isaac // First add the full polynomials up to degree - 1 into the basis: take the scalar 1228d8f25ad8SToby Isaac // and copy one for each form component 1229d8f25ad8SToby Isaac for (PetscInt i = 0; i < Nbpm1; i++) { 1230d8f25ad8SToby Isaac const PetscReal *src = &p_scalar[i * Nk * npoints]; 1231d8f25ad8SToby Isaac for (PetscInt f = 0; f < Nf; f++) { 1232d8f25ad8SToby Isaac PetscReal *dest = &p[(total++ * Nf + f) * Nk * npoints]; 12339566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(dest, src, Nk * npoints)); 1234d8f25ad8SToby Isaac } 1235d8f25ad8SToby Isaac } 1236d8f25ad8SToby Isaac PetscInt *form_atoms; 12379566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(formDegree + 1, &form_atoms)); 1238d8f25ad8SToby Isaac // construct the interior product pattern 1239d8f25ad8SToby Isaac PetscInt(*pattern)[3]; 1240d8f25ad8SToby Isaac PetscInt Nf1; // number of formDegree + 1 forms 12419566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, formDegree + 1, &Nf1)); 1242d8f25ad8SToby Isaac PetscInt nnz = Nf1 * (formDegree + 1); 12439566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nf1 * (formDegree + 1), &pattern)); 12449566063dSJacob Faibussowitsch PetscCall(PetscDTAltVInteriorPattern(dim, formDegree + 1, pattern)); 1245d8f25ad8SToby Isaac PetscReal centroid = (1. - dim) / (dim + 1.); 1246d8f25ad8SToby Isaac PetscInt *deriv; 12479566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim, &deriv)); 1248d8f25ad8SToby Isaac for (PetscInt d = dim; d >= formDegree + 1; d--) { 1249d8f25ad8SToby Isaac PetscInt Nfd1; // number of formDegree + 1 forms in dimension d that include dx_0 1250d8f25ad8SToby Isaac // (equal to the number of formDegree forms in dimension d-1) 12519566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(d - 1, formDegree, &Nfd1)); 1252d8f25ad8SToby Isaac // The number of homogeneous (degree-1) scalar polynomials in d variables 1253d8f25ad8SToby Isaac PetscInt Nh; 12549566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(d - 1 + degree - 1, d - 1, &Nh)); 1255d8f25ad8SToby Isaac const PetscReal *h_scalar = &p_scalar[(Nbpm1 - Nh) * Nk * npoints]; 1256d8f25ad8SToby Isaac for (PetscInt b = 0; b < Nh; b++) { 1257d8f25ad8SToby Isaac const PetscReal *h_s = &h_scalar[b * Nk * npoints]; 1258d8f25ad8SToby Isaac for (PetscInt f = 0; f < Nfd1; f++) { 1259d8f25ad8SToby Isaac // construct all formDegree+1 forms that start with dx_(dim - d) /\ ... 1260d8f25ad8SToby Isaac form_atoms[0] = dim - d; 12619566063dSJacob Faibussowitsch PetscCall(PetscDTEnumSubset(d - 1, formDegree, f, &form_atoms[1])); 1262ad540459SPierre Jolivet for (PetscInt i = 0; i < formDegree; i++) form_atoms[1 + i] += form_atoms[0] + 1; 1263d8f25ad8SToby Isaac PetscInt f_ind; // index of the resulting form 12649566063dSJacob Faibussowitsch PetscCall(PetscDTSubsetIndex(dim, formDegree + 1, form_atoms, &f_ind)); 1265d8f25ad8SToby Isaac PetscReal *p_f = &p[total++ * Nf * Nk * npoints]; 1266d8f25ad8SToby Isaac for (PetscInt nz = 0; nz < nnz; nz++) { 1267d8f25ad8SToby Isaac PetscInt i = pattern[nz][0]; // formDegree component 1268d8f25ad8SToby Isaac PetscInt j = pattern[nz][1]; // (formDegree + 1) component 1269d8f25ad8SToby Isaac PetscInt v = pattern[nz][2]; // coordinate component 1270d8f25ad8SToby Isaac PetscReal scale = v < 0 ? -1. : 1.; 1271d8f25ad8SToby Isaac 1272d8f25ad8SToby Isaac i = formNegative ? (Nf - 1 - i) : i; 1273d8f25ad8SToby Isaac scale = (formNegative && (i & 1)) ? -scale : scale; 1274d8f25ad8SToby Isaac v = v < 0 ? -(v + 1) : v; 1275ad540459SPierre Jolivet if (j != f_ind) continue; 1276d8f25ad8SToby Isaac PetscReal *p_i = &p_f[i * Nk * npoints]; 1277d8f25ad8SToby Isaac for (PetscInt jet = 0; jet < Nk; jet++) { 1278d8f25ad8SToby Isaac const PetscReal *h_jet = &h_s[jet * npoints]; 1279d8f25ad8SToby Isaac PetscReal *p_jet = &p_i[jet * npoints]; 1280d8f25ad8SToby Isaac 1281ad540459SPierre Jolivet for (PetscInt pt = 0; pt < npoints; pt++) p_jet[pt] += scale * h_jet[pt] * (points[pt * dim + v] - centroid); 12829566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, jet, deriv)); 1283d8f25ad8SToby Isaac deriv[v]++; 1284d8f25ad8SToby Isaac PetscReal mult = deriv[v]; 1285d8f25ad8SToby Isaac PetscInt l; 12869566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, deriv, &l)); 1287ad540459SPierre Jolivet if (l >= Nk) continue; 1288d8f25ad8SToby Isaac p_jet = &p_i[l * npoints]; 1289ad540459SPierre Jolivet for (PetscInt pt = 0; pt < npoints; pt++) p_jet[pt] += scale * mult * h_jet[pt]; 1290d8f25ad8SToby Isaac deriv[v]--; 1291d8f25ad8SToby Isaac } 1292d8f25ad8SToby Isaac } 1293d8f25ad8SToby Isaac } 1294d8f25ad8SToby Isaac } 1295d8f25ad8SToby Isaac } 129608401ef6SPierre Jolivet PetscCheck(total == Nbpt, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Incorrectly counted P trimmed polynomials"); 12979566063dSJacob Faibussowitsch PetscCall(PetscFree(deriv)); 12989566063dSJacob Faibussowitsch PetscCall(PetscFree(pattern)); 12999566063dSJacob Faibussowitsch PetscCall(PetscFree(form_atoms)); 13009566063dSJacob Faibussowitsch PetscCall(PetscFree(p_scalar)); 13013ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1302d8f25ad8SToby Isaac } 1303d8f25ad8SToby Isaac 1304d8f25ad8SToby Isaac /*@ 1305d8f25ad8SToby Isaac PetscDTPTrimmedEvalJet - Evaluate the jet (function and derivatives) of a basis of the trimmed polynomial k-forms up to 1306d8f25ad8SToby Isaac a given degree. 1307d8f25ad8SToby Isaac 1308d8f25ad8SToby Isaac Input Parameters: 1309d8f25ad8SToby Isaac + dim - the number of variables in the multivariate polynomials 1310d8f25ad8SToby Isaac . npoints - the number of points to evaluate the polynomials at 1311d8f25ad8SToby Isaac . points - [npoints x dim] array of point coordinates 1312d8f25ad8SToby Isaac . degree - the degree (sum of degrees on the variables in a monomial) of the trimmed polynomial space to evaluate. 1313d8f25ad8SToby Isaac There are ((dim + degree) choose (dim + formDegree)) x ((degree + formDegree - 1) choose (formDegree)) polynomials in this space. 1314dce8aebaSBarry Smith (You can use `PetscDTPTrimmedSize()` to compute this size.) 1315d8f25ad8SToby Isaac . formDegree - the degree of the form 1316d8f25ad8SToby Isaac - jetDegree - the maximum order partial derivative to evaluate in the jet. There are ((dim + jetDegree) choose dim) partial derivatives 1317d8f25ad8SToby Isaac in the jet. Choosing jetDegree = 0 means to evaluate just the function and no derivatives 1318d8f25ad8SToby Isaac 1319*20f4b53cSBarry Smith Output Parameter: 1320*20f4b53cSBarry Smith . p - an array containing the evaluations of the PKD polynomials' jets on the points. The size is 1321dce8aebaSBarry Smith `PetscDTPTrimmedSize()` x ((dim + formDegree) choose dim) x ((dim + k) choose dim) x npoints, 1322d8f25ad8SToby Isaac which also describes the order of the dimensions of this 1323d8f25ad8SToby Isaac four-dimensional array: 1324d8f25ad8SToby Isaac the first (slowest varying) dimension is basis function index; 1325d8f25ad8SToby Isaac the second dimension is component of the form; 1326d8f25ad8SToby Isaac the third dimension is jet index; 1327d8f25ad8SToby Isaac the fourth (fastest varying) dimension is the index of the evaluation point. 1328d8f25ad8SToby Isaac 1329d8f25ad8SToby Isaac Level: advanced 1330d8f25ad8SToby Isaac 1331dce8aebaSBarry Smith Notes: 1332dce8aebaSBarry Smith The ordering of the basis functions is not graded, so the basis functions are not nested by degree like `PetscDTPKDEvalJet()`. 1333d8f25ad8SToby Isaac The basis functions are not an L2-orthonormal basis on any particular domain. 1334d8f25ad8SToby Isaac 1335d8f25ad8SToby Isaac The implementation is based on the description of the trimmed polynomials up to degree r as 1336d8f25ad8SToby Isaac the direct sum of polynomials up to degree (r-1) and the Koszul differential applied to 1337d8f25ad8SToby Isaac homogeneous polynomials of degree (r-1). 1338d8f25ad8SToby Isaac 1339db781477SPatrick Sanan .seealso: `PetscDTPKDEvalJet()`, `PetscDTPTrimmedSize()` 1340d8f25ad8SToby Isaac @*/ 1341d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPTrimmedEvalJet(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt formDegree, PetscInt jetDegree, PetscReal p[]) 1342d71ae5a4SJacob Faibussowitsch { 1343d8f25ad8SToby Isaac PetscFunctionBegin; 13449566063dSJacob Faibussowitsch PetscCall(PetscDTPTrimmedEvalJet_Internal(dim, npoints, points, degree, formDegree, jetDegree, p)); 13453ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1346d8f25ad8SToby Isaac } 1347d8f25ad8SToby Isaac 1348e6a796c3SToby Isaac /* solve the symmetric tridiagonal eigenvalue system, writing the eigenvalues into eigs and the eigenvectors into V 1349e6a796c3SToby Isaac * with lds n; diag and subdiag are overwritten */ 1350d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTSymmetricTridiagonalEigensolve(PetscInt n, PetscReal diag[], PetscReal subdiag[], PetscReal eigs[], PetscScalar V[]) 1351d71ae5a4SJacob Faibussowitsch { 1352e6a796c3SToby Isaac char jobz = 'V'; /* eigenvalues and eigenvectors */ 1353e6a796c3SToby Isaac char range = 'A'; /* all eigenvalues will be found */ 1354e6a796c3SToby Isaac PetscReal VL = 0.; /* ignored because range is 'A' */ 1355e6a796c3SToby Isaac PetscReal VU = 0.; /* ignored because range is 'A' */ 1356e6a796c3SToby Isaac PetscBLASInt IL = 0; /* ignored because range is 'A' */ 1357e6a796c3SToby Isaac PetscBLASInt IU = 0; /* ignored because range is 'A' */ 1358e6a796c3SToby Isaac PetscReal abstol = 0.; /* unused */ 1359e6a796c3SToby Isaac PetscBLASInt bn, bm, ldz; /* bm will equal bn on exit */ 1360e6a796c3SToby Isaac PetscBLASInt *isuppz; 1361e6a796c3SToby Isaac PetscBLASInt lwork, liwork; 1362e6a796c3SToby Isaac PetscReal workquery; 1363e6a796c3SToby Isaac PetscBLASInt iworkquery; 1364e6a796c3SToby Isaac PetscBLASInt *iwork; 1365e6a796c3SToby Isaac PetscBLASInt info; 1366e6a796c3SToby Isaac PetscReal *work = NULL; 1367e6a796c3SToby Isaac 1368e6a796c3SToby Isaac PetscFunctionBegin; 1369e6a796c3SToby Isaac #if !defined(PETSCDTGAUSSIANQUADRATURE_EIG) 1370e6a796c3SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "A LAPACK symmetric tridiagonal eigensolver could not be found"); 1371e6a796c3SToby Isaac #endif 13729566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &bn)); 13739566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &ldz)); 1374e6a796c3SToby Isaac #if !defined(PETSC_MISSING_LAPACK_STEGR) 13759566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(2 * n, &isuppz)); 1376e6a796c3SToby Isaac lwork = -1; 1377e6a796c3SToby Isaac liwork = -1; 1378792fecdfSBarry Smith PetscCallBLAS("LAPACKstegr", LAPACKstegr_(&jobz, &range, &bn, diag, subdiag, &VL, &VU, &IL, &IU, &abstol, &bm, eigs, V, &ldz, isuppz, &workquery, &lwork, &iworkquery, &liwork, &info)); 137928b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEGR error"); 1380e6a796c3SToby Isaac lwork = (PetscBLASInt)workquery; 1381e6a796c3SToby Isaac liwork = (PetscBLASInt)iworkquery; 13829566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(lwork, &work, liwork, &iwork)); 13839566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 1384792fecdfSBarry Smith PetscCallBLAS("LAPACKstegr", LAPACKstegr_(&jobz, &range, &bn, diag, subdiag, &VL, &VU, &IL, &IU, &abstol, &bm, eigs, V, &ldz, isuppz, work, &lwork, iwork, &liwork, &info)); 13859566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 138628b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEGR error"); 13879566063dSJacob Faibussowitsch PetscCall(PetscFree2(work, iwork)); 13889566063dSJacob Faibussowitsch PetscCall(PetscFree(isuppz)); 1389e6a796c3SToby Isaac #elif !defined(PETSC_MISSING_LAPACK_STEQR) 1390e6a796c3SToby Isaac jobz = 'I'; /* Compute eigenvalues and eigenvectors of the 1391e6a796c3SToby Isaac tridiagonal matrix. Z is initialized to the identity 1392e6a796c3SToby Isaac matrix. */ 13939566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(PetscMax(1, 2 * n - 2), &work)); 1394792fecdfSBarry Smith PetscCallBLAS("LAPACKsteqr", LAPACKsteqr_("I", &bn, diag, subdiag, V, &ldz, work, &info)); 13959566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 139628b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEQR error"); 13979566063dSJacob Faibussowitsch PetscCall(PetscFree(work)); 13989566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(eigs, diag, n)); 1399e6a796c3SToby Isaac #endif 14003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1401e6a796c3SToby Isaac } 1402e6a796c3SToby Isaac 1403e6a796c3SToby Isaac /* Formula for the weights at the endpoints (-1 and 1) of Gauss-Lobatto-Jacobi 1404e6a796c3SToby Isaac * quadrature rules on the interval [-1, 1] */ 1405d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussLobattoJacobiEndweights_Internal(PetscInt n, PetscReal alpha, PetscReal beta, PetscReal *leftw, PetscReal *rightw) 1406d71ae5a4SJacob Faibussowitsch { 1407e6a796c3SToby Isaac PetscReal twoab1; 1408e6a796c3SToby Isaac PetscInt m = n - 2; 1409e6a796c3SToby Isaac PetscReal a = alpha + 1.; 1410e6a796c3SToby Isaac PetscReal b = beta + 1.; 1411e6a796c3SToby Isaac PetscReal gra, grb; 1412e6a796c3SToby Isaac 1413e6a796c3SToby Isaac PetscFunctionBegin; 1414e6a796c3SToby Isaac twoab1 = PetscPowReal(2., a + b - 1.); 1415e6a796c3SToby Isaac #if defined(PETSC_HAVE_LGAMMA) 14169371c9d4SSatish Balay grb = PetscExpReal(2. * PetscLGamma(b + 1.) + PetscLGamma(m + 1.) + PetscLGamma(m + a + 1.) - (PetscLGamma(m + b + 1) + PetscLGamma(m + a + b + 1.))); 14179371c9d4SSatish Balay gra = PetscExpReal(2. * PetscLGamma(a + 1.) + PetscLGamma(m + 1.) + PetscLGamma(m + b + 1.) - (PetscLGamma(m + a + 1) + PetscLGamma(m + a + b + 1.))); 1418e6a796c3SToby Isaac #else 1419e6a796c3SToby Isaac { 1420e6a796c3SToby Isaac PetscInt alphai = (PetscInt)alpha; 1421e6a796c3SToby Isaac PetscInt betai = (PetscInt)beta; 1422e6a796c3SToby Isaac 1423e6a796c3SToby Isaac if ((PetscReal)alphai == alpha && (PetscReal)betai == beta) { 1424e6a796c3SToby Isaac PetscReal binom1, binom2; 1425e6a796c3SToby Isaac 14269566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + b, b, &binom1)); 14279566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a + b, b, &binom2)); 1428e6a796c3SToby Isaac grb = 1. / (binom1 * binom2); 14299566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a, a, &binom1)); 14309566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a + b, a, &binom2)); 1431e6a796c3SToby Isaac gra = 1. / (binom1 * binom2); 1432e6a796c3SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 1433e6a796c3SToby Isaac } 1434e6a796c3SToby Isaac #endif 1435e6a796c3SToby Isaac *leftw = twoab1 * grb / b; 1436e6a796c3SToby Isaac *rightw = twoab1 * gra / a; 14373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1438e6a796c3SToby Isaac } 1439e6a796c3SToby Isaac 1440e6a796c3SToby Isaac /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 1441e6a796c3SToby Isaac Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 1442d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 1443d71ae5a4SJacob Faibussowitsch { 144494e21283SToby Isaac PetscReal pn1, pn2; 144594e21283SToby Isaac PetscReal cnm1, cnm1x, cnm2; 1446e6a796c3SToby Isaac PetscInt k; 1447e6a796c3SToby Isaac 1448e6a796c3SToby Isaac PetscFunctionBegin; 14499371c9d4SSatish Balay if (!n) { 14509371c9d4SSatish Balay *P = 1.0; 14513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14529371c9d4SSatish Balay } 145394e21283SToby Isaac PetscDTJacobiRecurrence_Internal(1, a, b, cnm1, cnm1x, cnm2); 145494e21283SToby Isaac pn2 = 1.; 145594e21283SToby Isaac pn1 = cnm1 + cnm1x * x; 14569371c9d4SSatish Balay if (n == 1) { 14579371c9d4SSatish Balay *P = pn1; 14583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 14599371c9d4SSatish Balay } 1460e6a796c3SToby Isaac *P = 0.0; 1461e6a796c3SToby Isaac for (k = 2; k < n + 1; ++k) { 146294e21283SToby Isaac PetscDTJacobiRecurrence_Internal(k, a, b, cnm1, cnm1x, cnm2); 1463e6a796c3SToby Isaac 146494e21283SToby Isaac *P = (cnm1 + cnm1x * x) * pn1 - cnm2 * pn2; 1465e6a796c3SToby Isaac pn2 = pn1; 1466e6a796c3SToby Isaac pn1 = *P; 1467e6a796c3SToby Isaac } 14683ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1469e6a796c3SToby Isaac } 1470e6a796c3SToby Isaac 1471e6a796c3SToby Isaac /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */ 1472d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscInt k, PetscReal *P) 1473d71ae5a4SJacob Faibussowitsch { 1474e6a796c3SToby Isaac PetscReal nP; 1475e6a796c3SToby Isaac PetscInt i; 1476e6a796c3SToby Isaac 1477e6a796c3SToby Isaac PetscFunctionBegin; 147817a42bb7SSatish Balay *P = 0.0; 14793ba16761SJacob Faibussowitsch if (k > n) PetscFunctionReturn(PETSC_SUCCESS); 14809566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(a + k, b + k, n - k, x, &nP)); 1481e6a796c3SToby Isaac for (i = 0; i < k; i++) nP *= (a + b + n + 1. + i) * 0.5; 1482e6a796c3SToby Isaac *P = nP; 14833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1484e6a796c3SToby Isaac } 1485e6a796c3SToby Isaac 1486d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_Newton_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal x[], PetscReal w[]) 1487d71ae5a4SJacob Faibussowitsch { 1488e6a796c3SToby Isaac PetscInt maxIter = 100; 148994e21283SToby Isaac PetscReal eps = PetscExpReal(0.75 * PetscLogReal(PETSC_MACHINE_EPSILON)); 1490200b5abcSJed Brown PetscReal a1, a6, gf; 1491e6a796c3SToby Isaac PetscInt k; 1492e6a796c3SToby Isaac 1493e6a796c3SToby Isaac PetscFunctionBegin; 1494e6a796c3SToby Isaac 1495e6a796c3SToby Isaac a1 = PetscPowReal(2.0, a + b + 1); 149694e21283SToby Isaac #if defined(PETSC_HAVE_LGAMMA) 1497200b5abcSJed Brown { 1498200b5abcSJed Brown PetscReal a2, a3, a4, a5; 149994e21283SToby Isaac a2 = PetscLGamma(a + npoints + 1); 150094e21283SToby Isaac a3 = PetscLGamma(b + npoints + 1); 150194e21283SToby Isaac a4 = PetscLGamma(a + b + npoints + 1); 150294e21283SToby Isaac a5 = PetscLGamma(npoints + 1); 150394e21283SToby Isaac gf = PetscExpReal(a2 + a3 - (a4 + a5)); 1504200b5abcSJed Brown } 1505e6a796c3SToby Isaac #else 1506e6a796c3SToby Isaac { 1507e6a796c3SToby Isaac PetscInt ia, ib; 1508e6a796c3SToby Isaac 1509e6a796c3SToby Isaac ia = (PetscInt)a; 1510e6a796c3SToby Isaac ib = (PetscInt)b; 151194e21283SToby Isaac gf = 1.; 151294e21283SToby Isaac if (ia == a && ia >= 0) { /* compute ratio of rising factorals wrt a */ 151394e21283SToby Isaac for (k = 0; k < ia; k++) gf *= (npoints + 1. + k) / (npoints + b + 1. + k); 151494e21283SToby Isaac } else if (b == b && ib >= 0) { /* compute ratio of rising factorials wrt b */ 151594e21283SToby Isaac for (k = 0; k < ib; k++) gf *= (npoints + 1. + k) / (npoints + a + 1. + k); 151694e21283SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 1517e6a796c3SToby Isaac } 1518e6a796c3SToby Isaac #endif 1519e6a796c3SToby Isaac 152094e21283SToby Isaac a6 = a1 * gf; 1521e6a796c3SToby Isaac /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses. 1522e6a796c3SToby Isaac Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */ 1523e6a796c3SToby Isaac for (k = 0; k < npoints; ++k) { 152494e21283SToby Isaac PetscReal r = PetscCosReal(PETSC_PI * (1. - (4. * k + 3. + 2. * b) / (4. * npoints + 2. * (a + b + 1.)))), dP; 1525e6a796c3SToby Isaac PetscInt j; 1526e6a796c3SToby Isaac 1527e6a796c3SToby Isaac if (k > 0) r = 0.5 * (r + x[k - 1]); 1528e6a796c3SToby Isaac for (j = 0; j < maxIter; ++j) { 1529e6a796c3SToby Isaac PetscReal s = 0.0, delta, f, fp; 1530e6a796c3SToby Isaac PetscInt i; 1531e6a796c3SToby Isaac 1532e6a796c3SToby Isaac for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]); 15339566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(a, b, npoints, r, &f)); 15349566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobiDerivative(a, b, npoints, r, 1, &fp)); 1535e6a796c3SToby Isaac delta = f / (fp - f * s); 1536e6a796c3SToby Isaac r = r - delta; 1537e6a796c3SToby Isaac if (PetscAbsReal(delta) < eps) break; 1538e6a796c3SToby Isaac } 1539e6a796c3SToby Isaac x[k] = r; 15409566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobiDerivative(a, b, npoints, x[k], 1, &dP)); 1541e6a796c3SToby Isaac w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP); 1542e6a796c3SToby Isaac } 15433ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1544e6a796c3SToby Isaac } 1545e6a796c3SToby Isaac 154694e21283SToby Isaac /* Compute the diagonals of the Jacobi matrix used in Golub & Welsch algorithms for Gauss-Jacobi 1547e6a796c3SToby Isaac * quadrature weight calculations on [-1,1] for exponents (1. + x)^a (1.-x)^b */ 1548d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobiMatrix_Internal(PetscInt nPoints, PetscReal a, PetscReal b, PetscReal *d, PetscReal *s) 1549d71ae5a4SJacob Faibussowitsch { 1550e6a796c3SToby Isaac PetscInt i; 1551e6a796c3SToby Isaac 1552e6a796c3SToby Isaac PetscFunctionBegin; 1553e6a796c3SToby Isaac for (i = 0; i < nPoints; i++) { 155494e21283SToby Isaac PetscReal A, B, C; 1555e6a796c3SToby Isaac 155694e21283SToby Isaac PetscDTJacobiRecurrence_Internal(i + 1, a, b, A, B, C); 155794e21283SToby Isaac d[i] = -A / B; 155894e21283SToby Isaac if (i) s[i - 1] *= C / B; 155994e21283SToby Isaac if (i < nPoints - 1) s[i] = 1. / B; 1560e6a796c3SToby Isaac } 15613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1562e6a796c3SToby Isaac } 1563e6a796c3SToby Isaac 1564d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_GolubWelsch_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 1565d71ae5a4SJacob Faibussowitsch { 1566e6a796c3SToby Isaac PetscReal mu0; 1567e6a796c3SToby Isaac PetscReal ga, gb, gab; 1568e6a796c3SToby Isaac PetscInt i; 1569e6a796c3SToby Isaac 1570e6a796c3SToby Isaac PetscFunctionBegin; 15719566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(GolubWelschCitation, &GolubWelschCite)); 1572e6a796c3SToby Isaac 1573e6a796c3SToby Isaac #if defined(PETSC_HAVE_TGAMMA) 1574e6a796c3SToby Isaac ga = PetscTGamma(a + 1); 1575e6a796c3SToby Isaac gb = PetscTGamma(b + 1); 1576e6a796c3SToby Isaac gab = PetscTGamma(a + b + 2); 1577e6a796c3SToby Isaac #else 1578e6a796c3SToby Isaac { 1579e6a796c3SToby Isaac PetscInt ia, ib; 1580e6a796c3SToby Isaac 1581e6a796c3SToby Isaac ia = (PetscInt)a; 1582e6a796c3SToby Isaac ib = (PetscInt)b; 1583e6a796c3SToby Isaac if (ia == a && ib == b && ia + 1 > 0 && ib + 1 > 0 && ia + ib + 2 > 0) { /* All gamma(x) terms are (x-1)! terms */ 15849566063dSJacob Faibussowitsch PetscCall(PetscDTFactorial(ia, &ga)); 15859566063dSJacob Faibussowitsch PetscCall(PetscDTFactorial(ib, &gb)); 15869566063dSJacob Faibussowitsch PetscCall(PetscDTFactorial(ia + ib + 1, &gb)); 1587e6a796c3SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "tgamma() - math routine is unavailable."); 1588e6a796c3SToby Isaac } 1589e6a796c3SToby Isaac #endif 1590e6a796c3SToby Isaac mu0 = PetscPowReal(2., a + b + 1.) * ga * gb / gab; 1591e6a796c3SToby Isaac 1592e6a796c3SToby Isaac #if defined(PETSCDTGAUSSIANQUADRATURE_EIG) 1593e6a796c3SToby Isaac { 1594e6a796c3SToby Isaac PetscReal *diag, *subdiag; 1595e6a796c3SToby Isaac PetscScalar *V; 1596e6a796c3SToby Isaac 15979566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &diag, npoints, &subdiag)); 15989566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints * npoints, &V)); 15999566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiMatrix_Internal(npoints, a, b, diag, subdiag)); 1600e6a796c3SToby Isaac for (i = 0; i < npoints - 1; i++) subdiag[i] = PetscSqrtReal(subdiag[i]); 16019566063dSJacob Faibussowitsch PetscCall(PetscDTSymmetricTridiagonalEigensolve(npoints, diag, subdiag, x, V)); 160294e21283SToby Isaac for (i = 0; i < npoints; i++) w[i] = PetscSqr(PetscRealPart(V[i * npoints])) * mu0; 16039566063dSJacob Faibussowitsch PetscCall(PetscFree(V)); 16049566063dSJacob Faibussowitsch PetscCall(PetscFree2(diag, subdiag)); 1605e6a796c3SToby Isaac } 1606e6a796c3SToby Isaac #else 1607e6a796c3SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "A LAPACK symmetric tridiagonal eigensolver could not be found"); 1608e6a796c3SToby Isaac #endif 160994e21283SToby Isaac { /* As of March 2, 2020, The Sun Performance Library breaks the LAPACK contract for xstegr and xsteqr: the 161094e21283SToby Isaac eigenvalues are not guaranteed to be in ascending order. So we heave a passive aggressive sigh and check that 161194e21283SToby Isaac the eigenvalues are sorted */ 161294e21283SToby Isaac PetscBool sorted; 161394e21283SToby Isaac 16149566063dSJacob Faibussowitsch PetscCall(PetscSortedReal(npoints, x, &sorted)); 161594e21283SToby Isaac if (!sorted) { 161694e21283SToby Isaac PetscInt *order, i; 161794e21283SToby Isaac PetscReal *tmp; 161894e21283SToby Isaac 16199566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &order, npoints, &tmp)); 162094e21283SToby Isaac for (i = 0; i < npoints; i++) order[i] = i; 16219566063dSJacob Faibussowitsch PetscCall(PetscSortRealWithPermutation(npoints, x, order)); 16229566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, x, npoints)); 162394e21283SToby Isaac for (i = 0; i < npoints; i++) x[i] = tmp[order[i]]; 16249566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, w, npoints)); 162594e21283SToby Isaac for (i = 0; i < npoints; i++) w[i] = tmp[order[i]]; 16269566063dSJacob Faibussowitsch PetscCall(PetscFree2(order, tmp)); 162794e21283SToby Isaac } 162894e21283SToby Isaac } 16293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1630e6a796c3SToby Isaac } 1631e6a796c3SToby Isaac 1632d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_Internal(PetscInt npoints, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[], PetscBool newton) 1633d71ae5a4SJacob Faibussowitsch { 1634e6a796c3SToby Isaac PetscFunctionBegin; 163508401ef6SPierre Jolivet PetscCheck(npoints >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be positive"); 1636e6a796c3SToby Isaac /* If asking for a 1D Lobatto point, just return the non-Lobatto 1D point */ 163708401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 163808401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 1639e6a796c3SToby Isaac 16401baa6e33SBarry Smith if (newton) PetscCall(PetscDTGaussJacobiQuadrature_Newton_Internal(npoints, alpha, beta, x, w)); 16411baa6e33SBarry Smith else PetscCall(PetscDTGaussJacobiQuadrature_GolubWelsch_Internal(npoints, alpha, beta, x, w)); 1642e6a796c3SToby Isaac if (alpha == beta) { /* symmetrize */ 1643e6a796c3SToby Isaac PetscInt i; 1644e6a796c3SToby Isaac for (i = 0; i < (npoints + 1) / 2; i++) { 1645e6a796c3SToby Isaac PetscInt j = npoints - 1 - i; 1646e6a796c3SToby Isaac PetscReal xi = x[i]; 1647e6a796c3SToby Isaac PetscReal xj = x[j]; 1648e6a796c3SToby Isaac PetscReal wi = w[i]; 1649e6a796c3SToby Isaac PetscReal wj = w[j]; 1650e6a796c3SToby Isaac 1651e6a796c3SToby Isaac x[i] = (xi - xj) / 2.; 1652e6a796c3SToby Isaac x[j] = (xj - xi) / 2.; 1653e6a796c3SToby Isaac w[i] = w[j] = (wi + wj) / 2.; 1654e6a796c3SToby Isaac } 1655e6a796c3SToby Isaac } 16563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1657e6a796c3SToby Isaac } 1658e6a796c3SToby Isaac 165994e21283SToby Isaac /*@ 166094e21283SToby Isaac PetscDTGaussJacobiQuadrature - quadrature for the interval [a, b] with the weight function 166194e21283SToby Isaac $(x-a)^\alpha (x-b)^\beta$. 166294e21283SToby Isaac 1663*20f4b53cSBarry Smith Not Collective 166494e21283SToby Isaac 166594e21283SToby Isaac Input Parameters: 166694e21283SToby Isaac + npoints - the number of points in the quadrature rule 166794e21283SToby Isaac . a - the left endpoint of the interval 166894e21283SToby Isaac . b - the right endpoint of the interval 166994e21283SToby Isaac . alpha - the left exponent 167094e21283SToby Isaac - beta - the right exponent 167194e21283SToby Isaac 167294e21283SToby Isaac Output Parameters: 1673*20f4b53cSBarry Smith + x - array of length `npoints`, the locations of the quadrature points 1674*20f4b53cSBarry Smith - w - array of length `npoints`, the weights of the quadrature points 167594e21283SToby Isaac 167694e21283SToby Isaac Level: intermediate 167794e21283SToby Isaac 1678dce8aebaSBarry Smith Note: 1679dce8aebaSBarry Smith This quadrature rule is exact for polynomials up to degree 2*npoints - 1. 1680dce8aebaSBarry Smith 1681dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()` 168294e21283SToby Isaac @*/ 1683d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[]) 1684d71ae5a4SJacob Faibussowitsch { 168594e21283SToby Isaac PetscInt i; 1686e6a796c3SToby Isaac 1687e6a796c3SToby Isaac PetscFunctionBegin; 16889566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints, alpha, beta, x, w, PetscDTGaussQuadratureNewton_Internal)); 168994e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 169094e21283SToby Isaac for (i = 0; i < npoints; i++) { 169194e21283SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 169294e21283SToby Isaac w[i] *= (b - a) / 2.; 169394e21283SToby Isaac } 169494e21283SToby Isaac } 16953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1696e6a796c3SToby Isaac } 1697e6a796c3SToby Isaac 1698d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussLobattoJacobiQuadrature_Internal(PetscInt npoints, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[], PetscBool newton) 1699d71ae5a4SJacob Faibussowitsch { 1700e6a796c3SToby Isaac PetscInt i; 1701e6a796c3SToby Isaac 1702e6a796c3SToby Isaac PetscFunctionBegin; 170308401ef6SPierre Jolivet PetscCheck(npoints >= 2, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be positive"); 1704e6a796c3SToby Isaac /* If asking for a 1D Lobatto point, just return the non-Lobatto 1D point */ 170508401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 170608401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 1707e6a796c3SToby Isaac 1708e6a796c3SToby Isaac x[0] = -1.; 1709e6a796c3SToby Isaac x[npoints - 1] = 1.; 171048a46eb9SPierre Jolivet if (npoints > 2) PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints - 2, alpha + 1., beta + 1., &x[1], &w[1], newton)); 1711ad540459SPierre Jolivet for (i = 1; i < npoints - 1; i++) w[i] /= (1. - x[i] * x[i]); 17129566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiEndweights_Internal(npoints, alpha, beta, &w[0], &w[npoints - 1])); 17133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1714e6a796c3SToby Isaac } 1715e6a796c3SToby Isaac 171637045ce4SJed Brown /*@ 171794e21283SToby Isaac PetscDTGaussLobattoJacobiQuadrature - quadrature for the interval [a, b] with the weight function 171894e21283SToby Isaac $(x-a)^\alpha (x-b)^\beta$, with endpoints a and b included as quadrature points. 171994e21283SToby Isaac 1720*20f4b53cSBarry Smith Not Collective 172194e21283SToby Isaac 172294e21283SToby Isaac Input Parameters: 172394e21283SToby Isaac + npoints - the number of points in the quadrature rule 172494e21283SToby Isaac . a - the left endpoint of the interval 172594e21283SToby Isaac . b - the right endpoint of the interval 172694e21283SToby Isaac . alpha - the left exponent 172794e21283SToby Isaac - beta - the right exponent 172894e21283SToby Isaac 172994e21283SToby Isaac Output Parameters: 1730*20f4b53cSBarry Smith + x - array of length `npoints`, the locations of the quadrature points 1731*20f4b53cSBarry Smith - w - array of length `npoints`, the weights of the quadrature points 173294e21283SToby Isaac 173394e21283SToby Isaac Level: intermediate 173494e21283SToby Isaac 1735dce8aebaSBarry Smith Note: 1736dce8aebaSBarry Smith This quadrature rule is exact for polynomials up to degree 2*npoints - 3. 1737dce8aebaSBarry Smith 1738dce8aebaSBarry Smith .seealso: `PetscDTGaussJacobiQuadrature()` 173994e21283SToby Isaac @*/ 1740d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussLobattoJacobiQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[]) 1741d71ae5a4SJacob Faibussowitsch { 174294e21283SToby Isaac PetscInt i; 174394e21283SToby Isaac 174494e21283SToby Isaac PetscFunctionBegin; 17459566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiQuadrature_Internal(npoints, alpha, beta, x, w, PetscDTGaussQuadratureNewton_Internal)); 174694e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 174794e21283SToby Isaac for (i = 0; i < npoints; i++) { 174894e21283SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 174994e21283SToby Isaac w[i] *= (b - a) / 2.; 175094e21283SToby Isaac } 175194e21283SToby Isaac } 17523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 175394e21283SToby Isaac } 175494e21283SToby Isaac 175594e21283SToby Isaac /*@ 1756e6a796c3SToby Isaac PetscDTGaussQuadrature - create Gauss-Legendre quadrature 175737045ce4SJed Brown 175837045ce4SJed Brown Not Collective 175937045ce4SJed Brown 17604165533cSJose E. Roman Input Parameters: 176137045ce4SJed Brown + npoints - number of points 176237045ce4SJed Brown . a - left end of interval (often-1) 176337045ce4SJed Brown - b - right end of interval (often +1) 176437045ce4SJed Brown 17654165533cSJose E. Roman Output Parameters: 176637045ce4SJed Brown + x - quadrature points 176737045ce4SJed Brown - w - quadrature weights 176837045ce4SJed Brown 176937045ce4SJed Brown Level: intermediate 177037045ce4SJed Brown 177137045ce4SJed Brown References: 1772606c0280SSatish Balay . * - Golub and Welsch, Calculation of Quadrature Rules, Math. Comp. 23(106), 1969. 177337045ce4SJed Brown 1774dce8aebaSBarry Smith .seealso: `PetscDTLegendreEval()`, `PetscDTGaussJacobiQuadrature()` 177537045ce4SJed Brown @*/ 1776d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 1777d71ae5a4SJacob Faibussowitsch { 177837045ce4SJed Brown PetscInt i; 177937045ce4SJed Brown 178037045ce4SJed Brown PetscFunctionBegin; 17819566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints, 0., 0., x, w, PetscDTGaussQuadratureNewton_Internal)); 178294e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 178337045ce4SJed Brown for (i = 0; i < npoints; i++) { 1784e6a796c3SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 1785e6a796c3SToby Isaac w[i] *= (b - a) / 2.; 178637045ce4SJed Brown } 178737045ce4SJed Brown } 17883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 178937045ce4SJed Brown } 1790194825f6SJed Brown 17918272889dSSatish Balay /*@C 17928272889dSSatish Balay PetscDTGaussLobattoLegendreQuadrature - creates a set of the locations and weights of the Gauss-Lobatto-Legendre 17938272889dSSatish Balay nodes of a given size on the domain [-1,1] 17948272889dSSatish Balay 17958272889dSSatish Balay Not Collective 17968272889dSSatish Balay 1797d8d19677SJose E. Roman Input Parameters: 17988272889dSSatish Balay + n - number of grid nodes 1799dce8aebaSBarry Smith - type - `PETSCGAUSSLOBATTOLEGENDRE_VIA_LINEAR_ALGEBRA` or `PETSCGAUSSLOBATTOLEGENDRE_VIA_NEWTON` 18008272889dSSatish Balay 18014165533cSJose E. Roman Output Parameters: 18028272889dSSatish Balay + x - quadrature points 18038272889dSSatish Balay - w - quadrature weights 18048272889dSSatish Balay 1805dce8aebaSBarry Smith Level: intermediate 1806dce8aebaSBarry Smith 18078272889dSSatish Balay Notes: 18088272889dSSatish Balay For n > 30 the Newton approach computes duplicate (incorrect) values for some nodes because the initial guess is apparently not 18098272889dSSatish Balay close enough to the desired solution 18108272889dSSatish Balay 18118272889dSSatish Balay These are useful for implementing spectral methods based on Gauss-Lobatto-Legendre (GLL) nodes 18128272889dSSatish Balay 1813a8d69d7bSBarry Smith See https://epubs.siam.org/doi/abs/10.1137/110855442 https://epubs.siam.org/doi/abs/10.1137/120889873 for better ways to compute GLL nodes 18148272889dSSatish Balay 1815dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()`, `PetscGaussLobattoLegendreCreateType` 18168272889dSSatish Balay 18178272889dSSatish Balay @*/ 1818d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussLobattoLegendreQuadrature(PetscInt npoints, PetscGaussLobattoLegendreCreateType type, PetscReal *x, PetscReal *w) 1819d71ae5a4SJacob Faibussowitsch { 1820e6a796c3SToby Isaac PetscBool newton; 18218272889dSSatish Balay 18228272889dSSatish Balay PetscFunctionBegin; 182308401ef6SPierre Jolivet PetscCheck(npoints >= 2, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must provide at least 2 grid points per element"); 182494e21283SToby Isaac newton = (PetscBool)(type == PETSCGAUSSLOBATTOLEGENDRE_VIA_NEWTON); 18259566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiQuadrature_Internal(npoints, 0., 0., x, w, newton)); 18263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 18278272889dSSatish Balay } 18288272889dSSatish Balay 1829744bafbcSMatthew G. Knepley /*@ 1830744bafbcSMatthew G. Knepley PetscDTGaussTensorQuadrature - creates a tensor-product Gauss quadrature 1831744bafbcSMatthew G. Knepley 1832744bafbcSMatthew G. Knepley Not Collective 1833744bafbcSMatthew G. Knepley 18344165533cSJose E. Roman Input Parameters: 1835744bafbcSMatthew G. Knepley + dim - The spatial dimension 1836a6b92713SMatthew G. Knepley . Nc - The number of components 1837744bafbcSMatthew G. Knepley . npoints - number of points in one dimension 1838744bafbcSMatthew G. Knepley . a - left end of interval (often-1) 1839744bafbcSMatthew G. Knepley - b - right end of interval (often +1) 1840744bafbcSMatthew G. Knepley 18414165533cSJose E. Roman Output Parameter: 1842dce8aebaSBarry Smith . q - A `PetscQuadrature` object 1843744bafbcSMatthew G. Knepley 1844744bafbcSMatthew G. Knepley Level: intermediate 1845744bafbcSMatthew G. Knepley 1846db781477SPatrick Sanan .seealso: `PetscDTGaussQuadrature()`, `PetscDTLegendreEval()` 1847744bafbcSMatthew G. Knepley @*/ 1848d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussTensorQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 1849d71ae5a4SJacob Faibussowitsch { 1850a6b92713SMatthew G. Knepley PetscInt totpoints = dim > 1 ? dim > 2 ? npoints * PetscSqr(npoints) : PetscSqr(npoints) : npoints, i, j, k, c; 1851744bafbcSMatthew G. Knepley PetscReal *x, *w, *xw, *ww; 1852744bafbcSMatthew G. Knepley 1853744bafbcSMatthew G. Knepley PetscFunctionBegin; 18549566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * dim, &x)); 18559566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * Nc, &w)); 1856744bafbcSMatthew G. Knepley /* Set up the Golub-Welsch system */ 1857744bafbcSMatthew G. Knepley switch (dim) { 1858744bafbcSMatthew G. Knepley case 0: 18599566063dSJacob Faibussowitsch PetscCall(PetscFree(x)); 18609566063dSJacob Faibussowitsch PetscCall(PetscFree(w)); 18619566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(1, &x)); 18629566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nc, &w)); 1863744bafbcSMatthew G. Knepley x[0] = 0.0; 1864a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[c] = 1.0; 1865744bafbcSMatthew G. Knepley break; 1866744bafbcSMatthew G. Knepley case 1: 18679566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints, &ww)); 18689566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, x, ww)); 18699371c9d4SSatish Balay for (i = 0; i < npoints; ++i) 18709371c9d4SSatish Balay for (c = 0; c < Nc; ++c) w[i * Nc + c] = ww[i]; 18719566063dSJacob Faibussowitsch PetscCall(PetscFree(ww)); 1872744bafbcSMatthew G. Knepley break; 1873744bafbcSMatthew G. Knepley case 2: 18749566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &xw, npoints, &ww)); 18759566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, xw, ww)); 1876744bafbcSMatthew G. Knepley for (i = 0; i < npoints; ++i) { 1877744bafbcSMatthew G. Knepley for (j = 0; j < npoints; ++j) { 1878744bafbcSMatthew G. Knepley x[(i * npoints + j) * dim + 0] = xw[i]; 1879744bafbcSMatthew G. Knepley x[(i * npoints + j) * dim + 1] = xw[j]; 1880a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[(i * npoints + j) * Nc + c] = ww[i] * ww[j]; 1881744bafbcSMatthew G. Knepley } 1882744bafbcSMatthew G. Knepley } 18839566063dSJacob Faibussowitsch PetscCall(PetscFree2(xw, ww)); 1884744bafbcSMatthew G. Knepley break; 1885744bafbcSMatthew G. Knepley case 3: 18869566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &xw, npoints, &ww)); 18879566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, xw, ww)); 1888744bafbcSMatthew G. Knepley for (i = 0; i < npoints; ++i) { 1889744bafbcSMatthew G. Knepley for (j = 0; j < npoints; ++j) { 1890744bafbcSMatthew G. Knepley for (k = 0; k < npoints; ++k) { 1891744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 0] = xw[i]; 1892744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 1] = xw[j]; 1893744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 2] = xw[k]; 1894a6b92713SMatthew G. Knepley for (c = 0; c < Nc; ++c) w[((i * npoints + j) * npoints + k) * Nc + c] = ww[i] * ww[j] * ww[k]; 1895744bafbcSMatthew G. Knepley } 1896744bafbcSMatthew G. Knepley } 1897744bafbcSMatthew G. Knepley } 18989566063dSJacob Faibussowitsch PetscCall(PetscFree2(xw, ww)); 1899744bafbcSMatthew G. Knepley break; 1900d71ae5a4SJacob Faibussowitsch default: 1901d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %" PetscInt_FMT, dim); 1902744bafbcSMatthew G. Knepley } 19039566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 19049566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * npoints - 1)); 19059566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w)); 19069566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)*q, "GaussTensor")); 19073ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1908744bafbcSMatthew G. Knepley } 1909744bafbcSMatthew G. Knepley 1910f5f57ec0SBarry Smith /*@ 1911e6a796c3SToby Isaac PetscDTStroudConicalQuadrature - create Stroud conical quadrature for a simplex 1912494e7359SMatthew G. Knepley 1913494e7359SMatthew G. Knepley Not Collective 1914494e7359SMatthew G. Knepley 19154165533cSJose E. Roman Input Parameters: 1916494e7359SMatthew G. Knepley + dim - The simplex dimension 1917a6b92713SMatthew G. Knepley . Nc - The number of components 1918dcce0ee2SMatthew G. Knepley . npoints - The number of points in one dimension 1919494e7359SMatthew G. Knepley . a - left end of interval (often-1) 1920494e7359SMatthew G. Knepley - b - right end of interval (often +1) 1921494e7359SMatthew G. Knepley 19224165533cSJose E. Roman Output Parameter: 1923*20f4b53cSBarry Smith . q - A `PetscQuadrature` object 1924494e7359SMatthew G. Knepley 1925494e7359SMatthew G. Knepley Level: intermediate 1926494e7359SMatthew G. Knepley 1927dce8aebaSBarry Smith Note: 1928*20f4b53cSBarry Smith For `dim` == 1, this is Gauss-Legendre quadrature 1929dce8aebaSBarry Smith 1930494e7359SMatthew G. Knepley References: 1931606c0280SSatish Balay . * - Karniadakis and Sherwin. FIAT 1932494e7359SMatthew G. Knepley 1933db781477SPatrick Sanan .seealso: `PetscDTGaussTensorQuadrature()`, `PetscDTGaussQuadrature()` 1934494e7359SMatthew G. Knepley @*/ 1935d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTStroudConicalQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 1936d71ae5a4SJacob Faibussowitsch { 1937fbdc3dfeSToby Isaac PetscInt totprev, totrem; 1938fbdc3dfeSToby Isaac PetscInt totpoints; 1939fbdc3dfeSToby Isaac PetscReal *p1, *w1; 1940fbdc3dfeSToby Isaac PetscReal *x, *w; 1941fbdc3dfeSToby Isaac PetscInt i, j, k, l, m, pt, c; 1942494e7359SMatthew G. Knepley 1943494e7359SMatthew G. Knepley PetscFunctionBegin; 194408401ef6SPierre Jolivet PetscCheck(!(a != -1.0) && !(b != 1.0), PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now"); 1945fbdc3dfeSToby Isaac totpoints = 1; 1946fbdc3dfeSToby Isaac for (i = 0, totpoints = 1; i < dim; i++) totpoints *= npoints; 19479566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * dim, &x)); 19489566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * Nc, &w)); 19499566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &p1, npoints, &w1)); 1950fbdc3dfeSToby Isaac for (i = 0; i < totpoints * Nc; i++) w[i] = 1.; 1951fbdc3dfeSToby Isaac for (i = 0, totprev = 1, totrem = totpoints / npoints; i < dim; i++) { 1952fbdc3dfeSToby Isaac PetscReal mul; 1953fbdc3dfeSToby Isaac 1954fbdc3dfeSToby Isaac mul = PetscPowReal(2., -i); 19559566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature(npoints, -1., 1., i, 0.0, p1, w1)); 1956fbdc3dfeSToby Isaac for (pt = 0, l = 0; l < totprev; l++) { 1957fbdc3dfeSToby Isaac for (j = 0; j < npoints; j++) { 1958fbdc3dfeSToby Isaac for (m = 0; m < totrem; m++, pt++) { 1959fbdc3dfeSToby Isaac for (k = 0; k < i; k++) x[pt * dim + k] = (x[pt * dim + k] + 1.) * (1. - p1[j]) * 0.5 - 1.; 1960fbdc3dfeSToby Isaac x[pt * dim + i] = p1[j]; 1961fbdc3dfeSToby Isaac for (c = 0; c < Nc; c++) w[pt * Nc + c] *= mul * w1[j]; 1962494e7359SMatthew G. Knepley } 1963494e7359SMatthew G. Knepley } 1964494e7359SMatthew G. Knepley } 1965fbdc3dfeSToby Isaac totprev *= npoints; 1966fbdc3dfeSToby Isaac totrem /= npoints; 1967494e7359SMatthew G. Knepley } 19689566063dSJacob Faibussowitsch PetscCall(PetscFree2(p1, w1)); 19699566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 19709566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * npoints - 1)); 19719566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w)); 19729566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)*q, "StroudConical")); 19733ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1974494e7359SMatthew G. Knepley } 1975494e7359SMatthew G. Knepley 1976d3c69ad0SToby Isaac static PetscBool MinSymTriQuadCite = PETSC_FALSE; 19779371c9d4SSatish Balay const char MinSymTriQuadCitation[] = "@article{WitherdenVincent2015,\n" 1978d3c69ad0SToby Isaac " title = {On the identification of symmetric quadrature rules for finite element methods},\n" 1979d3c69ad0SToby Isaac " journal = {Computers & Mathematics with Applications},\n" 1980d3c69ad0SToby Isaac " volume = {69},\n" 1981d3c69ad0SToby Isaac " number = {10},\n" 1982d3c69ad0SToby Isaac " pages = {1232-1241},\n" 1983d3c69ad0SToby Isaac " year = {2015},\n" 1984d3c69ad0SToby Isaac " issn = {0898-1221},\n" 1985d3c69ad0SToby Isaac " doi = {10.1016/j.camwa.2015.03.017},\n" 1986d3c69ad0SToby Isaac " url = {https://www.sciencedirect.com/science/article/pii/S0898122115001224},\n" 1987d3c69ad0SToby Isaac " author = {F.D. Witherden and P.E. Vincent},\n" 1988d3c69ad0SToby Isaac "}\n"; 1989d3c69ad0SToby Isaac 1990d3c69ad0SToby Isaac #include "petscdttriquadrules.h" 1991d3c69ad0SToby Isaac 1992d3c69ad0SToby Isaac static PetscBool MinSymTetQuadCite = PETSC_FALSE; 19939371c9d4SSatish Balay const char MinSymTetQuadCitation[] = "@article{JaskowiecSukumar2021\n" 1994d3c69ad0SToby Isaac " author = {Jaskowiec, Jan and Sukumar, N.},\n" 1995d3c69ad0SToby Isaac " title = {High-order symmetric cubature rules for tetrahedra and pyramids},\n" 1996d3c69ad0SToby Isaac " journal = {International Journal for Numerical Methods in Engineering},\n" 1997d3c69ad0SToby Isaac " volume = {122},\n" 1998d3c69ad0SToby Isaac " number = {1},\n" 1999d3c69ad0SToby Isaac " pages = {148-171},\n" 2000d3c69ad0SToby Isaac " doi = {10.1002/nme.6528},\n" 2001d3c69ad0SToby Isaac " url = {https://onlinelibrary.wiley.com/doi/abs/10.1002/nme.6528},\n" 2002d3c69ad0SToby Isaac " eprint = {https://onlinelibrary.wiley.com/doi/pdf/10.1002/nme.6528},\n" 2003d3c69ad0SToby Isaac " year = {2021}\n" 2004d3c69ad0SToby Isaac "}\n"; 2005d3c69ad0SToby Isaac 2006d3c69ad0SToby Isaac #include "petscdttetquadrules.h" 2007d3c69ad0SToby Isaac 2008d3c69ad0SToby Isaac // https://en.wikipedia.org/wiki/Partition_(number_theory) 2009d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPartitionNumber(PetscInt n, PetscInt *p) 2010d71ae5a4SJacob Faibussowitsch { 2011d3c69ad0SToby Isaac // sequence A000041 in the OEIS 2012d3c69ad0SToby Isaac const PetscInt partition[] = {1, 1, 2, 3, 5, 7, 11, 15, 22, 30, 42, 56, 77, 101, 135, 176, 231, 297, 385, 490, 627, 792, 1002, 1255, 1575, 1958, 2436, 3010, 3718, 4565, 5604}; 2013d3c69ad0SToby Isaac PetscInt tabulated_max = PETSC_STATIC_ARRAY_LENGTH(partition) - 1; 2014d3c69ad0SToby Isaac 2015d3c69ad0SToby Isaac PetscFunctionBegin; 2016d3c69ad0SToby Isaac PetscCheck(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Partition number not defined for negative number %" PetscInt_FMT, n); 2017d3c69ad0SToby Isaac // not implementing the pentagonal number recurrence, we don't need partition numbers for n that high 2018d3c69ad0SToby Isaac PetscCheck(n <= tabulated_max, PETSC_COMM_SELF, PETSC_ERR_SUP, "Partition numbers only tabulated up to %" PetscInt_FMT ", not computed for %" PetscInt_FMT, tabulated_max, n); 2019d3c69ad0SToby Isaac *p = partition[n]; 20203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2021d3c69ad0SToby Isaac } 2022d3c69ad0SToby Isaac 2023d3c69ad0SToby Isaac /*@ 2024d3c69ad0SToby Isaac PetscDTSimplexQuadrature - Create a quadrature rule for a simplex that exactly integrates polynomials up to a given degree. 2025d3c69ad0SToby Isaac 2026d3c69ad0SToby Isaac Not Collective 2027d3c69ad0SToby Isaac 2028d3c69ad0SToby Isaac Input Parameters: 2029d3c69ad0SToby Isaac + dim - The spatial dimension of the simplex (1 = segment, 2 = triangle, 3 = tetrahedron) 2030d3c69ad0SToby Isaac . degree - The largest polynomial degree that is required to be integrated exactly 2031d3c69ad0SToby Isaac - type - left end of interval (often-1) 2032d3c69ad0SToby Isaac 2033d3c69ad0SToby Isaac Output Parameter: 2034dce8aebaSBarry Smith . quad - A `PetscQuadrature` object for integration over the biunit simplex 2035d3c69ad0SToby Isaac (defined by the bounds $x_i >= -1$ and $\sum_i x_i <= 2 - d$) that is exact for 2036d3c69ad0SToby Isaac polynomials up to the given degree 2037d3c69ad0SToby Isaac 2038d3c69ad0SToby Isaac Level: intermediate 2039d3c69ad0SToby Isaac 2040dce8aebaSBarry Smith .seealso: `PetscDTSimplexQuadratureType`, `PetscDTGaussQuadrature()`, `PetscDTStroudCononicalQuadrature()`, `PetscQuadrature` 2041d3c69ad0SToby Isaac @*/ 2042d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTSimplexQuadrature(PetscInt dim, PetscInt degree, PetscDTSimplexQuadratureType type, PetscQuadrature *quad) 2043d71ae5a4SJacob Faibussowitsch { 2044d3c69ad0SToby Isaac PetscDTSimplexQuadratureType orig_type = type; 2045d3c69ad0SToby Isaac 2046d3c69ad0SToby Isaac PetscFunctionBegin; 2047d3c69ad0SToby Isaac PetscCheck(dim >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative dimension %" PetscInt_FMT, dim); 2048d3c69ad0SToby Isaac PetscCheck(degree >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative degree %" PetscInt_FMT, degree); 2049ad540459SPierre Jolivet if (type == PETSCDTSIMPLEXQUAD_DEFAULT) type = PETSCDTSIMPLEXQUAD_MINSYM; 2050d3c69ad0SToby Isaac if (type == PETSCDTSIMPLEXQUAD_CONIC || dim < 2) { 2051d3c69ad0SToby Isaac PetscInt points_per_dim = (degree + 2) / 2; // ceil((degree + 1) / 2); 2052d3c69ad0SToby Isaac PetscCall(PetscDTStroudConicalQuadrature(dim, 1, points_per_dim, -1, 1, quad)); 2053d3c69ad0SToby Isaac } else { 2054d3c69ad0SToby Isaac PetscInt n = dim + 1; 2055d3c69ad0SToby Isaac PetscInt fact = 1; 2056d3c69ad0SToby Isaac PetscInt *part, *perm; 2057d3c69ad0SToby Isaac PetscInt p = 0; 2058d3c69ad0SToby Isaac PetscInt max_degree; 2059d3c69ad0SToby Isaac const PetscInt *nodes_per_type = NULL; 2060d3c69ad0SToby Isaac const PetscInt *all_num_full_nodes = NULL; 2061d3c69ad0SToby Isaac const PetscReal **weights_list = NULL; 2062d3c69ad0SToby Isaac const PetscReal **compact_nodes_list = NULL; 2063d3c69ad0SToby Isaac const char *citation = NULL; 2064d3c69ad0SToby Isaac PetscBool *cited = NULL; 2065d3c69ad0SToby Isaac 2066d3c69ad0SToby Isaac switch (dim) { 2067d3c69ad0SToby Isaac case 2: 2068d3c69ad0SToby Isaac cited = &MinSymTriQuadCite; 2069d3c69ad0SToby Isaac citation = MinSymTriQuadCitation; 2070d3c69ad0SToby Isaac max_degree = PetscDTWVTriQuad_max_degree; 2071d3c69ad0SToby Isaac nodes_per_type = PetscDTWVTriQuad_num_orbits; 2072d3c69ad0SToby Isaac all_num_full_nodes = PetscDTWVTriQuad_num_nodes; 2073d3c69ad0SToby Isaac weights_list = PetscDTWVTriQuad_weights; 2074d3c69ad0SToby Isaac compact_nodes_list = PetscDTWVTriQuad_orbits; 2075d3c69ad0SToby Isaac break; 2076d3c69ad0SToby Isaac case 3: 2077d3c69ad0SToby Isaac cited = &MinSymTetQuadCite; 2078d3c69ad0SToby Isaac citation = MinSymTetQuadCitation; 2079d3c69ad0SToby Isaac max_degree = PetscDTJSTetQuad_max_degree; 2080d3c69ad0SToby Isaac nodes_per_type = PetscDTJSTetQuad_num_orbits; 2081d3c69ad0SToby Isaac all_num_full_nodes = PetscDTJSTetQuad_num_nodes; 2082d3c69ad0SToby Isaac weights_list = PetscDTJSTetQuad_weights; 2083d3c69ad0SToby Isaac compact_nodes_list = PetscDTJSTetQuad_orbits; 2084d3c69ad0SToby Isaac break; 2085d71ae5a4SJacob Faibussowitsch default: 2086d71ae5a4SJacob Faibussowitsch max_degree = -1; 2087d71ae5a4SJacob Faibussowitsch break; 2088d3c69ad0SToby Isaac } 2089d3c69ad0SToby Isaac 2090d3c69ad0SToby Isaac if (degree > max_degree) { 2091d3c69ad0SToby Isaac if (orig_type == PETSCDTSIMPLEXQUAD_DEFAULT) { 2092d3c69ad0SToby Isaac // fall back to conic 2093d3c69ad0SToby Isaac PetscCall(PetscDTSimplexQuadrature(dim, degree, PETSCDTSIMPLEXQUAD_CONIC, quad)); 20943ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2095d3c69ad0SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Minimal symmetric quadrature for dim %" PetscInt_FMT ", degree %" PetscInt_FMT " unsupported", dim, degree); 2096d3c69ad0SToby Isaac } 2097d3c69ad0SToby Isaac 2098d3c69ad0SToby Isaac PetscCall(PetscCitationsRegister(citation, cited)); 2099d3c69ad0SToby Isaac 2100d3c69ad0SToby Isaac PetscCall(PetscDTPartitionNumber(n, &p)); 2101d3c69ad0SToby Isaac for (PetscInt d = 2; d <= n; d++) fact *= d; 2102d3c69ad0SToby Isaac 2103d3c69ad0SToby Isaac PetscInt num_full_nodes = all_num_full_nodes[degree]; 2104d3c69ad0SToby Isaac const PetscReal *all_compact_nodes = compact_nodes_list[degree]; 2105d3c69ad0SToby Isaac const PetscReal *all_compact_weights = weights_list[degree]; 2106d3c69ad0SToby Isaac nodes_per_type = &nodes_per_type[p * degree]; 2107d3c69ad0SToby Isaac 2108d3c69ad0SToby Isaac PetscReal *points; 2109d3c69ad0SToby Isaac PetscReal *counts; 2110d3c69ad0SToby Isaac PetscReal *weights; 2111d3c69ad0SToby Isaac PetscReal *bary_to_biunit; // row-major transformation of barycentric coordinate to biunit 2112d3c69ad0SToby Isaac PetscQuadrature q; 2113d3c69ad0SToby Isaac 2114d3c69ad0SToby Isaac // compute the transformation 2115d3c69ad0SToby Isaac PetscCall(PetscMalloc1(n * dim, &bary_to_biunit)); 2116d3c69ad0SToby Isaac for (PetscInt d = 0; d < dim; d++) { 2117ad540459SPierre Jolivet for (PetscInt b = 0; b < n; b++) bary_to_biunit[d * n + b] = (d == b) ? 1.0 : -1.0; 2118d3c69ad0SToby Isaac } 2119d3c69ad0SToby Isaac 2120d3c69ad0SToby Isaac PetscCall(PetscMalloc3(n, &part, n, &perm, n, &counts)); 2121d3c69ad0SToby Isaac PetscCall(PetscCalloc1(num_full_nodes * dim, &points)); 2122d3c69ad0SToby Isaac PetscCall(PetscMalloc1(num_full_nodes, &weights)); 2123d3c69ad0SToby Isaac 2124d3c69ad0SToby Isaac // (0, 0, ...) is the first partition lexicographically 2125d3c69ad0SToby Isaac PetscCall(PetscArrayzero(part, n)); 2126d3c69ad0SToby Isaac PetscCall(PetscArrayzero(counts, n)); 2127d3c69ad0SToby Isaac counts[0] = n; 2128d3c69ad0SToby Isaac 2129d3c69ad0SToby Isaac // for each partition 2130d3c69ad0SToby Isaac for (PetscInt s = 0, node_offset = 0; s < p; s++) { 2131d3c69ad0SToby Isaac PetscInt num_compact_coords = part[n - 1] + 1; 2132d3c69ad0SToby Isaac 2133d3c69ad0SToby Isaac const PetscReal *compact_nodes = all_compact_nodes; 2134d3c69ad0SToby Isaac const PetscReal *compact_weights = all_compact_weights; 2135d3c69ad0SToby Isaac all_compact_nodes += num_compact_coords * nodes_per_type[s]; 2136d3c69ad0SToby Isaac all_compact_weights += nodes_per_type[s]; 2137d3c69ad0SToby Isaac 2138d3c69ad0SToby Isaac // for every permutation of the vertices 2139d3c69ad0SToby Isaac for (PetscInt f = 0; f < fact; f++) { 2140d3c69ad0SToby Isaac PetscCall(PetscDTEnumPerm(n, f, perm, NULL)); 2141d3c69ad0SToby Isaac 2142d3c69ad0SToby Isaac // check if it is a valid permutation 2143d3c69ad0SToby Isaac PetscInt digit; 2144d3c69ad0SToby Isaac for (digit = 1; digit < n; digit++) { 2145d3c69ad0SToby Isaac // skip permutations that would duplicate a node because it has a smaller symmetry group 2146d3c69ad0SToby Isaac if (part[digit - 1] == part[digit] && perm[digit - 1] > perm[digit]) break; 2147d3c69ad0SToby Isaac } 2148d3c69ad0SToby Isaac if (digit < n) continue; 2149d3c69ad0SToby Isaac 2150d3c69ad0SToby Isaac // create full nodes from this permutation of the compact nodes 2151d3c69ad0SToby Isaac PetscReal *full_nodes = &points[node_offset * dim]; 2152d3c69ad0SToby Isaac PetscReal *full_weights = &weights[node_offset]; 2153d3c69ad0SToby Isaac 2154d3c69ad0SToby Isaac PetscCall(PetscArraycpy(full_weights, compact_weights, nodes_per_type[s])); 2155d3c69ad0SToby Isaac for (PetscInt b = 0; b < n; b++) { 2156d3c69ad0SToby Isaac for (PetscInt d = 0; d < dim; d++) { 2157ad540459SPierre Jolivet for (PetscInt node = 0; node < nodes_per_type[s]; node++) full_nodes[node * dim + d] += bary_to_biunit[d * n + perm[b]] * compact_nodes[node * num_compact_coords + part[b]]; 2158d3c69ad0SToby Isaac } 2159d3c69ad0SToby Isaac } 2160d3c69ad0SToby Isaac node_offset += nodes_per_type[s]; 2161d3c69ad0SToby Isaac } 2162d3c69ad0SToby Isaac 2163d3c69ad0SToby Isaac if (s < p - 1) { // Generate the next partition 2164d3c69ad0SToby Isaac /* A partition is described by the number of coordinates that are in 2165d3c69ad0SToby Isaac * each set of duplicates (counts) and redundantly by mapping each 2166d3c69ad0SToby Isaac * index to its set of duplicates (part) 2167d3c69ad0SToby Isaac * 2168d3c69ad0SToby Isaac * Counts should always be in nonincreasing order 2169d3c69ad0SToby Isaac * 2170d3c69ad0SToby Isaac * We want to generate the partitions lexically by part, which means 2171d3c69ad0SToby Isaac * finding the last index where count > 1 and reducing by 1. 2172d3c69ad0SToby Isaac * 2173d3c69ad0SToby Isaac * For the new counts beyond that index, we eagerly assign the remaining 2174d3c69ad0SToby Isaac * capacity of the partition to smaller indices (ensures lexical ordering), 2175d3c69ad0SToby Isaac * while respecting the nonincreasing invariant of the counts 2176d3c69ad0SToby Isaac */ 2177d3c69ad0SToby Isaac PetscInt last_digit = part[n - 1]; 2178d3c69ad0SToby Isaac PetscInt last_digit_with_extra = last_digit; 2179d3c69ad0SToby Isaac while (counts[last_digit_with_extra] == 1) last_digit_with_extra--; 2180d3c69ad0SToby Isaac PetscInt limit = --counts[last_digit_with_extra]; 2181d3c69ad0SToby Isaac PetscInt total_to_distribute = last_digit - last_digit_with_extra + 1; 2182d3c69ad0SToby Isaac for (PetscInt digit = last_digit_with_extra + 1; digit < n; digit++) { 2183d3c69ad0SToby Isaac counts[digit] = PetscMin(limit, total_to_distribute); 2184d3c69ad0SToby Isaac total_to_distribute -= PetscMin(limit, total_to_distribute); 2185d3c69ad0SToby Isaac } 2186d3c69ad0SToby Isaac for (PetscInt digit = 0, offset = 0; digit < n; digit++) { 2187d3c69ad0SToby Isaac PetscInt count = counts[digit]; 2188ad540459SPierre Jolivet for (PetscInt c = 0; c < count; c++) part[offset++] = digit; 2189d3c69ad0SToby Isaac } 2190d3c69ad0SToby Isaac } 2191d3c69ad0SToby Isaac } 2192d3c69ad0SToby Isaac PetscCall(PetscFree3(part, perm, counts)); 2193d3c69ad0SToby Isaac PetscCall(PetscFree(bary_to_biunit)); 2194d3c69ad0SToby Isaac PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, &q)); 2195b414c505SJed Brown PetscCall(PetscQuadratureSetOrder(q, degree)); 2196d3c69ad0SToby Isaac PetscCall(PetscQuadratureSetData(q, dim, 1, num_full_nodes, points, weights)); 2197d3c69ad0SToby Isaac *quad = q; 2198d3c69ad0SToby Isaac } 21993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2200d3c69ad0SToby Isaac } 2201d3c69ad0SToby Isaac 2202f5f57ec0SBarry Smith /*@ 2203b3c0f97bSTom Klotz PetscDTTanhSinhTensorQuadrature - create tanh-sinh quadrature for a tensor product cell 2204b3c0f97bSTom Klotz 2205b3c0f97bSTom Klotz Not Collective 2206b3c0f97bSTom Klotz 22074165533cSJose E. Roman Input Parameters: 2208b3c0f97bSTom Klotz + dim - The cell dimension 2209b3c0f97bSTom Klotz . level - The number of points in one dimension, 2^l 2210b3c0f97bSTom Klotz . a - left end of interval (often-1) 2211b3c0f97bSTom Klotz - b - right end of interval (often +1) 2212b3c0f97bSTom Klotz 22134165533cSJose E. Roman Output Parameter: 2214dce8aebaSBarry Smith . q - A `PetscQuadrature` object 2215b3c0f97bSTom Klotz 2216b3c0f97bSTom Klotz Level: intermediate 2217b3c0f97bSTom Klotz 2218dce8aebaSBarry Smith .seealso: `PetscDTGaussTensorQuadrature()`, `PetscQuadrature` 2219b3c0f97bSTom Klotz @*/ 2220d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhTensorQuadrature(PetscInt dim, PetscInt level, PetscReal a, PetscReal b, PetscQuadrature *q) 2221d71ae5a4SJacob Faibussowitsch { 2222b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 2223b3c0f97bSTom Klotz const PetscReal alpha = (b - a) / 2.; /* Half-width of the integration interval */ 2224b3c0f97bSTom Klotz const PetscReal beta = (b + a) / 2.; /* Center of the integration interval */ 2225b3c0f97bSTom Klotz const PetscReal h = PetscPowReal(2.0, -level); /* Step size, length between x_k */ 2226d84b4d08SMatthew G. Knepley PetscReal xk; /* Quadrature point x_k on reference domain [-1, 1] */ 2227b3c0f97bSTom Klotz PetscReal wk = 0.5 * PETSC_PI; /* Quadrature weight at x_k */ 2228b3c0f97bSTom Klotz PetscReal *x, *w; 2229b3c0f97bSTom Klotz PetscInt K, k, npoints; 2230b3c0f97bSTom Klotz 2231b3c0f97bSTom Klotz PetscFunctionBegin; 223263a3b9bcSJacob Faibussowitsch PetscCheck(dim <= 1, PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not yet implemented", dim); 223328b400f6SJacob Faibussowitsch PetscCheck(level, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a number of significant digits"); 2234b3c0f97bSTom Klotz /* Find K such that the weights are < 32 digits of precision */ 2235ad540459SPierre Jolivet for (K = 1; PetscAbsReal(PetscLog10Real(wk)) < 2 * p; ++K) wk = 0.5 * h * PETSC_PI * PetscCoshReal(K * h) / PetscSqr(PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(K * h))); 22369566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 22379566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * K + 1)); 2238b3c0f97bSTom Klotz npoints = 2 * K - 1; 22399566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints * dim, &x)); 22409566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints, &w)); 2241b3c0f97bSTom Klotz /* Center term */ 2242b3c0f97bSTom Klotz x[0] = beta; 2243b3c0f97bSTom Klotz w[0] = 0.5 * alpha * PETSC_PI; 2244b3c0f97bSTom Klotz for (k = 1; k < K; ++k) { 22459add2064SThomas Klotz wk = 0.5 * alpha * h * PETSC_PI * PetscCoshReal(k * h) / PetscSqr(PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 22461118d4bcSLisandro Dalcin xk = PetscTanhReal(0.5 * PETSC_PI * PetscSinhReal(k * h)); 2247b3c0f97bSTom Klotz x[2 * k - 1] = -alpha * xk + beta; 2248b3c0f97bSTom Klotz w[2 * k - 1] = wk; 2249b3c0f97bSTom Klotz x[2 * k + 0] = alpha * xk + beta; 2250b3c0f97bSTom Klotz w[2 * k + 0] = wk; 2251b3c0f97bSTom Klotz } 22529566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, 1, npoints, x, w)); 22533ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2254b3c0f97bSTom Klotz } 2255b3c0f97bSTom Klotz 2256d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrate(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2257d71ae5a4SJacob Faibussowitsch { 2258b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 2259b3c0f97bSTom Klotz const PetscReal alpha = (b - a) / 2.; /* Half-width of the integration interval */ 2260b3c0f97bSTom Klotz const PetscReal beta = (b + a) / 2.; /* Center of the integration interval */ 2261b3c0f97bSTom Klotz PetscReal h = 1.0; /* Step size, length between x_k */ 2262b3c0f97bSTom Klotz PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 2263b3c0f97bSTom Klotz PetscReal osum = 0.0; /* Integral on last level */ 2264b3c0f97bSTom Klotz PetscReal psum = 0.0; /* Integral on the level before the last level */ 2265b3c0f97bSTom Klotz PetscReal sum; /* Integral on current level */ 2266446c295cSMatthew G. Knepley PetscReal yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 2267b3c0f97bSTom Klotz PetscReal lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 2268b3c0f97bSTom Klotz PetscReal wk; /* Quadrature weight at x_k */ 2269b3c0f97bSTom Klotz PetscReal lval, rval; /* Terms in the quadature sum to the left and right of 0 */ 2270b3c0f97bSTom Klotz PetscInt d; /* Digits of precision in the integral */ 2271b3c0f97bSTom Klotz 2272b3c0f97bSTom Klotz PetscFunctionBegin; 227308401ef6SPierre Jolivet PetscCheck(digits > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 2274b3c0f97bSTom Klotz /* Center term */ 2275d6685f55SMatthew G. Knepley func(&beta, ctx, &lval); 2276b3c0f97bSTom Klotz sum = 0.5 * alpha * PETSC_PI * lval; 2277b3c0f97bSTom Klotz /* */ 2278b3c0f97bSTom Klotz do { 2279b3c0f97bSTom Klotz PetscReal lterm, rterm, maxTerm = 0.0, d1, d2, d3, d4; 2280b3c0f97bSTom Klotz PetscInt k = 1; 2281b3c0f97bSTom Klotz 2282b3c0f97bSTom Klotz ++l; 228363a3b9bcSJacob Faibussowitsch /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %" PetscInt_FMT " sum: %15.15f\n", l, sum); */ 2284b3c0f97bSTom Klotz /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 2285b3c0f97bSTom Klotz psum = osum; 2286b3c0f97bSTom Klotz osum = sum; 2287b3c0f97bSTom Klotz h *= 0.5; 2288b3c0f97bSTom Klotz sum *= 0.5; 2289b3c0f97bSTom Klotz do { 22909add2064SThomas Klotz wk = 0.5 * h * PETSC_PI * PetscCoshReal(k * h) / PetscSqr(PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 2291446c295cSMatthew G. Knepley yk = 1.0 / (PetscExpReal(0.5 * PETSC_PI * PetscSinhReal(k * h)) * PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 2292446c295cSMatthew G. Knepley lx = -alpha * (1.0 - yk) + beta; 2293446c295cSMatthew G. Knepley rx = alpha * (1.0 - yk) + beta; 2294d6685f55SMatthew G. Knepley func(&lx, ctx, &lval); 2295d6685f55SMatthew G. Knepley func(&rx, ctx, &rval); 2296b3c0f97bSTom Klotz lterm = alpha * wk * lval; 2297b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(lterm), maxTerm); 2298b3c0f97bSTom Klotz sum += lterm; 2299b3c0f97bSTom Klotz rterm = alpha * wk * rval; 2300b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(rterm), maxTerm); 2301b3c0f97bSTom Klotz sum += rterm; 2302b3c0f97bSTom Klotz ++k; 2303b3c0f97bSTom Klotz /* Only need to evaluate every other point on refined levels */ 2304b3c0f97bSTom Klotz if (l != 1) ++k; 23059add2064SThomas Klotz } while (PetscAbsReal(PetscLog10Real(wk)) < p); /* Only need to evaluate sum until weights are < 32 digits of precision */ 2306b3c0f97bSTom Klotz 2307b3c0f97bSTom Klotz d1 = PetscLog10Real(PetscAbsReal(sum - osum)); 2308b3c0f97bSTom Klotz d2 = PetscLog10Real(PetscAbsReal(sum - psum)); 2309b3c0f97bSTom Klotz d3 = PetscLog10Real(maxTerm) - p; 231009d48545SBarry Smith if (PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm)) == 0.0) d4 = 0.0; 231109d48545SBarry Smith else d4 = PetscLog10Real(PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm))); 2312b3c0f97bSTom Klotz d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1) / d2, 2 * d1), d3), d4))); 23139add2064SThomas Klotz } while (d < digits && l < 12); 2314b3c0f97bSTom Klotz *sol = sum; 2315e510cb1fSThomas Klotz 23163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2317b3c0f97bSTom Klotz } 2318b3c0f97bSTom Klotz 2319497880caSRichard Tran Mills #if defined(PETSC_HAVE_MPFR) 2320d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2321d71ae5a4SJacob Faibussowitsch { 2322e510cb1fSThomas Klotz const PetscInt safetyFactor = 2; /* Calculate abcissa until 2*p digits */ 232329f144ccSMatthew G. Knepley PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 232429f144ccSMatthew G. Knepley mpfr_t alpha; /* Half-width of the integration interval */ 232529f144ccSMatthew G. Knepley mpfr_t beta; /* Center of the integration interval */ 232629f144ccSMatthew G. Knepley mpfr_t h; /* Step size, length between x_k */ 232729f144ccSMatthew G. Knepley mpfr_t osum; /* Integral on last level */ 232829f144ccSMatthew G. Knepley mpfr_t psum; /* Integral on the level before the last level */ 232929f144ccSMatthew G. Knepley mpfr_t sum; /* Integral on current level */ 233029f144ccSMatthew G. Knepley mpfr_t yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 233129f144ccSMatthew G. Knepley mpfr_t lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 233229f144ccSMatthew G. Knepley mpfr_t wk; /* Quadrature weight at x_k */ 23331fbc92bbSMatthew G. Knepley PetscReal lval, rval, rtmp; /* Terms in the quadature sum to the left and right of 0 */ 233429f144ccSMatthew G. Knepley PetscInt d; /* Digits of precision in the integral */ 233529f144ccSMatthew G. Knepley mpfr_t pi2, kh, msinh, mcosh, maxTerm, curTerm, tmp; 233629f144ccSMatthew G. Knepley 233729f144ccSMatthew G. Knepley PetscFunctionBegin; 233808401ef6SPierre Jolivet PetscCheck(digits > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 233929f144ccSMatthew G. Knepley /* Create high precision storage */ 2340c9f744b5SMatthew 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); 234129f144ccSMatthew G. Knepley /* Initialization */ 234229f144ccSMatthew G. Knepley mpfr_set_d(alpha, 0.5 * (b - a), MPFR_RNDN); 234329f144ccSMatthew G. Knepley mpfr_set_d(beta, 0.5 * (b + a), MPFR_RNDN); 234429f144ccSMatthew G. Knepley mpfr_set_d(osum, 0.0, MPFR_RNDN); 234529f144ccSMatthew G. Knepley mpfr_set_d(psum, 0.0, MPFR_RNDN); 234629f144ccSMatthew G. Knepley mpfr_set_d(h, 1.0, MPFR_RNDN); 234729f144ccSMatthew G. Knepley mpfr_const_pi(pi2, MPFR_RNDN); 234829f144ccSMatthew G. Knepley mpfr_mul_d(pi2, pi2, 0.5, MPFR_RNDN); 234929f144ccSMatthew G. Knepley /* Center term */ 23501fbc92bbSMatthew G. Knepley rtmp = 0.5 * (b + a); 23511fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &lval); 235229f144ccSMatthew G. Knepley mpfr_set(sum, pi2, MPFR_RNDN); 235329f144ccSMatthew G. Knepley mpfr_mul(sum, sum, alpha, MPFR_RNDN); 235429f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, lval, MPFR_RNDN); 235529f144ccSMatthew G. Knepley /* */ 235629f144ccSMatthew G. Knepley do { 235729f144ccSMatthew G. Knepley PetscReal d1, d2, d3, d4; 235829f144ccSMatthew G. Knepley PetscInt k = 1; 235929f144ccSMatthew G. Knepley 236029f144ccSMatthew G. Knepley ++l; 236129f144ccSMatthew G. Knepley mpfr_set_d(maxTerm, 0.0, MPFR_RNDN); 236263a3b9bcSJacob Faibussowitsch /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %" PetscInt_FMT " sum: %15.15f\n", l, sum); */ 236329f144ccSMatthew G. Knepley /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 236429f144ccSMatthew G. Knepley mpfr_set(psum, osum, MPFR_RNDN); 236529f144ccSMatthew G. Knepley mpfr_set(osum, sum, MPFR_RNDN); 236629f144ccSMatthew G. Knepley mpfr_mul_d(h, h, 0.5, MPFR_RNDN); 236729f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, 0.5, MPFR_RNDN); 236829f144ccSMatthew G. Knepley do { 236929f144ccSMatthew G. Knepley mpfr_set_si(kh, k, MPFR_RNDN); 237029f144ccSMatthew G. Knepley mpfr_mul(kh, kh, h, MPFR_RNDN); 237129f144ccSMatthew G. Knepley /* Weight */ 237229f144ccSMatthew G. Knepley mpfr_set(wk, h, MPFR_RNDN); 237329f144ccSMatthew G. Knepley mpfr_sinh_cosh(msinh, mcosh, kh, MPFR_RNDN); 237429f144ccSMatthew G. Knepley mpfr_mul(msinh, msinh, pi2, MPFR_RNDN); 237529f144ccSMatthew G. Knepley mpfr_mul(mcosh, mcosh, pi2, MPFR_RNDN); 237629f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 237729f144ccSMatthew G. Knepley mpfr_sqr(tmp, tmp, MPFR_RNDN); 237829f144ccSMatthew G. Knepley mpfr_mul(wk, wk, mcosh, MPFR_RNDN); 237929f144ccSMatthew G. Knepley mpfr_div(wk, wk, tmp, MPFR_RNDN); 238029f144ccSMatthew G. Knepley /* Abscissa */ 238129f144ccSMatthew G. Knepley mpfr_set_d(yk, 1.0, MPFR_RNDZ); 238229f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 238329f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 238429f144ccSMatthew G. Knepley mpfr_exp(tmp, msinh, MPFR_RNDN); 238529f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 238629f144ccSMatthew G. Knepley /* Quadrature points */ 238729f144ccSMatthew G. Knepley mpfr_sub_d(lx, yk, 1.0, MPFR_RNDZ); 238829f144ccSMatthew G. Knepley mpfr_mul(lx, lx, alpha, MPFR_RNDU); 238929f144ccSMatthew G. Knepley mpfr_add(lx, lx, beta, MPFR_RNDU); 239029f144ccSMatthew G. Knepley mpfr_d_sub(rx, 1.0, yk, MPFR_RNDZ); 239129f144ccSMatthew G. Knepley mpfr_mul(rx, rx, alpha, MPFR_RNDD); 239229f144ccSMatthew G. Knepley mpfr_add(rx, rx, beta, MPFR_RNDD); 239329f144ccSMatthew G. Knepley /* Evaluation */ 23941fbc92bbSMatthew G. Knepley rtmp = mpfr_get_d(lx, MPFR_RNDU); 23951fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &lval); 23961fbc92bbSMatthew G. Knepley rtmp = mpfr_get_d(rx, MPFR_RNDD); 23971fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &rval); 239829f144ccSMatthew G. Knepley /* Update */ 239929f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 240029f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, lval, MPFR_RNDN); 240129f144ccSMatthew G. Knepley mpfr_add(sum, sum, tmp, MPFR_RNDN); 240229f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 240329f144ccSMatthew G. Knepley mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN); 240429f144ccSMatthew G. Knepley mpfr_set(curTerm, tmp, MPFR_RNDN); 240529f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 240629f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, rval, MPFR_RNDN); 240729f144ccSMatthew G. Knepley mpfr_add(sum, sum, tmp, MPFR_RNDN); 240829f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 240929f144ccSMatthew G. Knepley mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN); 241029f144ccSMatthew G. Knepley mpfr_max(curTerm, curTerm, tmp, MPFR_RNDN); 241129f144ccSMatthew G. Knepley ++k; 241229f144ccSMatthew G. Knepley /* Only need to evaluate every other point on refined levels */ 241329f144ccSMatthew G. Knepley if (l != 1) ++k; 241429f144ccSMatthew G. Knepley mpfr_log10(tmp, wk, MPFR_RNDN); 241529f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 2416c9f744b5SMatthew G. Knepley } while (mpfr_get_d(tmp, MPFR_RNDN) < safetyFactor * digits); /* Only need to evaluate sum until weights are < 32 digits of precision */ 241729f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, osum, MPFR_RNDN); 241829f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 241929f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 242029f144ccSMatthew G. Knepley d1 = mpfr_get_d(tmp, MPFR_RNDN); 242129f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, psum, MPFR_RNDN); 242229f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 242329f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 242429f144ccSMatthew G. Knepley d2 = mpfr_get_d(tmp, MPFR_RNDN); 242529f144ccSMatthew G. Knepley mpfr_log10(tmp, maxTerm, MPFR_RNDN); 2426c9f744b5SMatthew G. Knepley d3 = mpfr_get_d(tmp, MPFR_RNDN) - digits; 242729f144ccSMatthew G. Knepley mpfr_log10(tmp, curTerm, MPFR_RNDN); 242829f144ccSMatthew G. Knepley d4 = mpfr_get_d(tmp, MPFR_RNDN); 242929f144ccSMatthew G. Knepley d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1) / d2, 2 * d1), d3), d4))); 2430b0649871SThomas Klotz } while (d < digits && l < 8); 243129f144ccSMatthew G. Knepley *sol = mpfr_get_d(sum, MPFR_RNDN); 243229f144ccSMatthew G. Knepley /* Cleanup */ 243329f144ccSMatthew G. Knepley mpfr_clears(alpha, beta, h, sum, osum, psum, yk, wk, lx, rx, tmp, maxTerm, curTerm, pi2, kh, msinh, mcosh, NULL); 24343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 243529f144ccSMatthew G. Knepley } 2436d525116cSMatthew G. Knepley #else 2437fbfcfee5SBarry Smith 2438d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2439d71ae5a4SJacob Faibussowitsch { 2440d525116cSMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "This method will not work without MPFR. Reconfigure using --download-mpfr --download-gmp"); 2441d525116cSMatthew G. Knepley } 244229f144ccSMatthew G. Knepley #endif 244329f144ccSMatthew G. Knepley 24442df84da0SMatthew G. Knepley /*@ 24452df84da0SMatthew G. Knepley PetscDTTensorQuadratureCreate - create the tensor product quadrature from two lower-dimensional quadratures 24462df84da0SMatthew G. Knepley 24472df84da0SMatthew G. Knepley Not Collective 24482df84da0SMatthew G. Knepley 24492df84da0SMatthew G. Knepley Input Parameters: 24502df84da0SMatthew G. Knepley + q1 - The first quadrature 24512df84da0SMatthew G. Knepley - q2 - The second quadrature 24522df84da0SMatthew G. Knepley 24532df84da0SMatthew G. Knepley Output Parameter: 2454dce8aebaSBarry Smith . q - A `PetscQuadrature` object 24552df84da0SMatthew G. Knepley 24562df84da0SMatthew G. Knepley Level: intermediate 24572df84da0SMatthew G. Knepley 2458dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTGaussTensorQuadrature()` 24592df84da0SMatthew G. Knepley @*/ 2460d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTensorQuadratureCreate(PetscQuadrature q1, PetscQuadrature q2, PetscQuadrature *q) 2461d71ae5a4SJacob Faibussowitsch { 24622df84da0SMatthew G. Knepley const PetscReal *x1, *w1, *x2, *w2; 24632df84da0SMatthew G. Knepley PetscReal *x, *w; 24642df84da0SMatthew G. Knepley PetscInt dim1, Nc1, Np1, order1, qa, d1; 24652df84da0SMatthew G. Knepley PetscInt dim2, Nc2, Np2, order2, qb, d2; 24662df84da0SMatthew G. Knepley PetscInt dim, Nc, Np, order, qc, d; 24672df84da0SMatthew G. Knepley 24682df84da0SMatthew G. Knepley PetscFunctionBegin; 24692df84da0SMatthew G. Knepley PetscValidHeaderSpecific(q1, PETSCQUADRATURE_CLASSID, 1); 24702df84da0SMatthew G. Knepley PetscValidHeaderSpecific(q2, PETSCQUADRATURE_CLASSID, 2); 24712df84da0SMatthew G. Knepley PetscValidPointer(q, 3); 24729566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q1, &order1)); 24739566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q2, &order2)); 24742df84da0SMatthew G. Knepley PetscCheck(order1 == order2, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Order1 %" PetscInt_FMT " != %" PetscInt_FMT " Order2", order1, order2); 24759566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q1, &dim1, &Nc1, &Np1, &x1, &w1)); 24769566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q2, &dim2, &Nc2, &Np2, &x2, &w2)); 24772df84da0SMatthew G. Knepley PetscCheck(Nc1 == Nc2, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "NumComp1 %" PetscInt_FMT " != %" PetscInt_FMT " NumComp2", Nc1, Nc2); 24782df84da0SMatthew G. Knepley 24792df84da0SMatthew G. Knepley dim = dim1 + dim2; 24802df84da0SMatthew G. Knepley Nc = Nc1; 24812df84da0SMatthew G. Knepley Np = Np1 * Np2; 24822df84da0SMatthew G. Knepley order = order1; 24839566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 24849566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, order)); 24859566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Np * dim, &x)); 24869566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Np, &w)); 24872df84da0SMatthew G. Knepley for (qa = 0, qc = 0; qa < Np1; ++qa) { 24882df84da0SMatthew G. Knepley for (qb = 0; qb < Np2; ++qb, ++qc) { 2489ad540459SPierre Jolivet for (d1 = 0, d = 0; d1 < dim1; ++d1, ++d) x[qc * dim + d] = x1[qa * dim1 + d1]; 2490ad540459SPierre Jolivet for (d2 = 0; d2 < dim2; ++d2, ++d) x[qc * dim + d] = x2[qb * dim2 + d2]; 24912df84da0SMatthew G. Knepley w[qc] = w1[qa] * w2[qb]; 24922df84da0SMatthew G. Knepley } 24932df84da0SMatthew G. Knepley } 24949566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, Np, x, w)); 24953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 24962df84da0SMatthew G. Knepley } 24972df84da0SMatthew G. Knepley 2498194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n 2499dce8aebaSBarry Smith A in column-major format 2500dce8aebaSBarry Smith Ainv in row-major format 2501dce8aebaSBarry Smith tau has length m 2502dce8aebaSBarry Smith worksize must be >= max(1,n) 2503194825f6SJed Brown */ 2504d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m, PetscInt mstride, PetscInt n, PetscReal *A_in, PetscReal *Ainv_out, PetscScalar *tau, PetscInt worksize, PetscScalar *work) 2505d71ae5a4SJacob Faibussowitsch { 2506194825f6SJed Brown PetscBLASInt M, N, K, lda, ldb, ldwork, info; 2507194825f6SJed Brown PetscScalar *A, *Ainv, *R, *Q, Alpha; 2508194825f6SJed Brown 2509194825f6SJed Brown PetscFunctionBegin; 2510194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 2511194825f6SJed Brown { 2512194825f6SJed Brown PetscInt i, j; 25139566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m * n, &A, m * n, &Ainv)); 2514194825f6SJed Brown for (j = 0; j < n; j++) { 2515194825f6SJed Brown for (i = 0; i < m; i++) A[i + m * j] = A_in[i + mstride * j]; 2516194825f6SJed Brown } 2517194825f6SJed Brown mstride = m; 2518194825f6SJed Brown } 2519194825f6SJed Brown #else 2520194825f6SJed Brown A = A_in; 2521194825f6SJed Brown Ainv = Ainv_out; 2522194825f6SJed Brown #endif 2523194825f6SJed Brown 25249566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(m, &M)); 25259566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &N)); 25269566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(mstride, &lda)); 25279566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(worksize, &ldwork)); 25289566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 2529792fecdfSBarry Smith PetscCallBLAS("LAPACKgeqrf", LAPACKgeqrf_(&M, &N, A, &lda, tau, work, &ldwork, &info)); 25309566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 253128b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "xGEQRF error"); 2532194825f6SJed Brown R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */ 2533194825f6SJed Brown 2534194825f6SJed Brown /* Extract an explicit representation of Q */ 2535194825f6SJed Brown Q = Ainv; 25369566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(Q, A, mstride * n)); 2537194825f6SJed Brown K = N; /* full rank */ 2538792fecdfSBarry Smith PetscCallBLAS("LAPACKorgqr", LAPACKorgqr_(&M, &N, &K, Q, &lda, tau, work, &ldwork, &info)); 253928b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "xORGQR/xUNGQR error"); 2540194825f6SJed Brown 2541194825f6SJed Brown /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */ 2542194825f6SJed Brown Alpha = 1.0; 2543194825f6SJed Brown ldb = lda; 2544792fecdfSBarry Smith PetscCallBLAS("BLAStrsm", BLAStrsm_("Right", "Upper", "ConjugateTranspose", "NotUnitTriangular", &M, &N, &Alpha, R, &lda, Q, &ldb)); 2545194825f6SJed Brown /* Ainv is Q, overwritten with inverse */ 2546194825f6SJed Brown 2547194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 2548194825f6SJed Brown { 2549194825f6SJed Brown PetscInt i; 2550194825f6SJed Brown for (i = 0; i < m * n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]); 25519566063dSJacob Faibussowitsch PetscCall(PetscFree2(A, Ainv)); 2552194825f6SJed Brown } 2553194825f6SJed Brown #endif 25543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2555194825f6SJed Brown } 2556194825f6SJed Brown 2557194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */ 2558d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval, const PetscReal *x, PetscInt ndegree, const PetscInt *degrees, PetscBool Transpose, PetscReal *B) 2559d71ae5a4SJacob Faibussowitsch { 2560194825f6SJed Brown PetscReal *Bv; 2561194825f6SJed Brown PetscInt i, j; 2562194825f6SJed Brown 2563194825f6SJed Brown PetscFunctionBegin; 25649566063dSJacob Faibussowitsch PetscCall(PetscMalloc1((ninterval + 1) * ndegree, &Bv)); 2565194825f6SJed Brown /* Point evaluation of L_p on all the source vertices */ 25669566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreEval(ninterval + 1, x, ndegree, degrees, Bv, NULL, NULL)); 2567194825f6SJed Brown /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */ 2568194825f6SJed Brown for (i = 0; i < ninterval; i++) { 2569194825f6SJed Brown for (j = 0; j < ndegree; j++) { 2570194825f6SJed Brown if (Transpose) B[i + ninterval * j] = Bv[(i + 1) * ndegree + j] - Bv[i * ndegree + j]; 2571194825f6SJed Brown else B[i * ndegree + j] = Bv[(i + 1) * ndegree + j] - Bv[i * ndegree + j]; 2572194825f6SJed Brown } 2573194825f6SJed Brown } 25749566063dSJacob Faibussowitsch PetscCall(PetscFree(Bv)); 25753ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2576194825f6SJed Brown } 2577194825f6SJed Brown 2578194825f6SJed Brown /*@ 2579194825f6SJed Brown PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals 2580194825f6SJed Brown 2581194825f6SJed Brown Not Collective 2582194825f6SJed Brown 25834165533cSJose E. Roman Input Parameters: 2584194825f6SJed Brown + degree - degree of reconstruction polynomial 2585194825f6SJed Brown . nsource - number of source intervals 2586194825f6SJed Brown . sourcex - sorted coordinates of source cell boundaries (length nsource+1) 2587194825f6SJed Brown . ntarget - number of target intervals 2588194825f6SJed Brown - targetx - sorted coordinates of target cell boundaries (length ntarget+1) 2589194825f6SJed Brown 25904165533cSJose E. Roman Output Parameter: 2591194825f6SJed Brown . R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s] 2592194825f6SJed Brown 2593194825f6SJed Brown Level: advanced 2594194825f6SJed Brown 2595db781477SPatrick Sanan .seealso: `PetscDTLegendreEval()` 2596194825f6SJed Brown @*/ 2597d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTReconstructPoly(PetscInt degree, PetscInt nsource, const PetscReal *sourcex, PetscInt ntarget, const PetscReal *targetx, PetscReal *R) 2598d71ae5a4SJacob Faibussowitsch { 2599194825f6SJed Brown PetscInt i, j, k, *bdegrees, worksize; 2600194825f6SJed Brown PetscReal xmin, xmax, center, hscale, *sourcey, *targety, *Bsource, *Bsinv, *Btarget; 2601194825f6SJed Brown PetscScalar *tau, *work; 2602194825f6SJed Brown 2603194825f6SJed Brown PetscFunctionBegin; 2604194825f6SJed Brown PetscValidRealPointer(sourcex, 3); 2605194825f6SJed Brown PetscValidRealPointer(targetx, 5); 2606194825f6SJed Brown PetscValidRealPointer(R, 6); 260763a3b9bcSJacob Faibussowitsch PetscCheck(degree < nsource, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Reconstruction degree %" PetscInt_FMT " must be less than number of source intervals %" PetscInt_FMT, degree, nsource); 260876bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2609ad540459SPierre Jolivet for (i = 0; i < nsource; i++) PetscCheck(sourcex[i] < sourcex[i + 1], PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT, "Source interval %" PetscInt_FMT " has negative orientation (%g,%g)", i, (double)sourcex[i], (double)sourcex[i + 1]); 2610ad540459SPierre Jolivet for (i = 0; i < ntarget; i++) PetscCheck(targetx[i] < targetx[i + 1], PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT, "Target interval %" PetscInt_FMT " has negative orientation (%g,%g)", i, (double)targetx[i], (double)targetx[i + 1]); 261176bd3646SJed Brown } 2612194825f6SJed Brown xmin = PetscMin(sourcex[0], targetx[0]); 2613194825f6SJed Brown xmax = PetscMax(sourcex[nsource], targetx[ntarget]); 2614194825f6SJed Brown center = (xmin + xmax) / 2; 2615194825f6SJed Brown hscale = (xmax - xmin) / 2; 2616194825f6SJed Brown worksize = nsource; 26179566063dSJacob Faibussowitsch PetscCall(PetscMalloc4(degree + 1, &bdegrees, nsource + 1, &sourcey, nsource * (degree + 1), &Bsource, worksize, &work)); 26189566063dSJacob Faibussowitsch PetscCall(PetscMalloc4(nsource, &tau, nsource * (degree + 1), &Bsinv, ntarget + 1, &targety, ntarget * (degree + 1), &Btarget)); 2619194825f6SJed Brown for (i = 0; i <= nsource; i++) sourcey[i] = (sourcex[i] - center) / hscale; 2620194825f6SJed Brown for (i = 0; i <= degree; i++) bdegrees[i] = i + 1; 26219566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreIntegrate(nsource, sourcey, degree + 1, bdegrees, PETSC_TRUE, Bsource)); 26229566063dSJacob Faibussowitsch PetscCall(PetscDTPseudoInverseQR(nsource, nsource, degree + 1, Bsource, Bsinv, tau, nsource, work)); 2623194825f6SJed Brown for (i = 0; i <= ntarget; i++) targety[i] = (targetx[i] - center) / hscale; 26249566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreIntegrate(ntarget, targety, degree + 1, bdegrees, PETSC_FALSE, Btarget)); 2625194825f6SJed Brown for (i = 0; i < ntarget; i++) { 2626194825f6SJed Brown PetscReal rowsum = 0; 2627194825f6SJed Brown for (j = 0; j < nsource; j++) { 2628194825f6SJed Brown PetscReal sum = 0; 2629ad540459SPierre Jolivet for (k = 0; k < degree + 1; k++) sum += Btarget[i * (degree + 1) + k] * Bsinv[k * nsource + j]; 2630194825f6SJed Brown R[i * nsource + j] = sum; 2631194825f6SJed Brown rowsum += sum; 2632194825f6SJed Brown } 2633194825f6SJed Brown for (j = 0; j < nsource; j++) R[i * nsource + j] /= rowsum; /* normalize each row */ 2634194825f6SJed Brown } 26359566063dSJacob Faibussowitsch PetscCall(PetscFree4(bdegrees, sourcey, Bsource, work)); 26369566063dSJacob Faibussowitsch PetscCall(PetscFree4(tau, Bsinv, targety, Btarget)); 26373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2638194825f6SJed Brown } 2639916e780bShannah_mairs 2640916e780bShannah_mairs /*@C 2641916e780bShannah_mairs PetscGaussLobattoLegendreIntegrate - Compute the L2 integral of a function on the GLL points 2642916e780bShannah_mairs 2643916e780bShannah_mairs Not Collective 2644916e780bShannah_mairs 2645d8d19677SJose E. Roman Input Parameters: 2646916e780bShannah_mairs + n - the number of GLL nodes 2647916e780bShannah_mairs . nodes - the GLL nodes 2648916e780bShannah_mairs . weights - the GLL weights 2649f0fc11ceSJed Brown - f - the function values at the nodes 2650916e780bShannah_mairs 2651916e780bShannah_mairs Output Parameter: 2652916e780bShannah_mairs . in - the value of the integral 2653916e780bShannah_mairs 2654916e780bShannah_mairs Level: beginner 2655916e780bShannah_mairs 2656db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()` 2657916e780bShannah_mairs @*/ 2658d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreIntegrate(PetscInt n, PetscReal *nodes, PetscReal *weights, const PetscReal *f, PetscReal *in) 2659d71ae5a4SJacob Faibussowitsch { 2660916e780bShannah_mairs PetscInt i; 2661916e780bShannah_mairs 2662916e780bShannah_mairs PetscFunctionBegin; 2663916e780bShannah_mairs *in = 0.; 2664ad540459SPierre Jolivet for (i = 0; i < n; i++) *in += f[i] * f[i] * weights[i]; 26653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2666916e780bShannah_mairs } 2667916e780bShannah_mairs 2668916e780bShannah_mairs /*@C 2669916e780bShannah_mairs PetscGaussLobattoLegendreElementLaplacianCreate - computes the Laplacian for a single 1d GLL element 2670916e780bShannah_mairs 2671916e780bShannah_mairs Not Collective 2672916e780bShannah_mairs 2673d8d19677SJose E. Roman Input Parameters: 2674916e780bShannah_mairs + n - the number of GLL nodes 2675916e780bShannah_mairs . nodes - the GLL nodes 2676f0fc11ceSJed Brown - weights - the GLL weights 2677916e780bShannah_mairs 2678916e780bShannah_mairs Output Parameter: 2679916e780bShannah_mairs . A - the stiffness element 2680916e780bShannah_mairs 2681916e780bShannah_mairs Level: beginner 2682916e780bShannah_mairs 2683916e780bShannah_mairs Notes: 2684dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementLaplacianDestroy()` 2685916e780bShannah_mairs 2686916e780bShannah_mairs You can access entries in this array with AA[i][j] but in memory it is stored in contiguous memory, row oriented (the array is symmetric) 2687916e780bShannah_mairs 2688db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianDestroy()` 2689916e780bShannah_mairs @*/ 2690d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementLaplacianCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2691d71ae5a4SJacob Faibussowitsch { 2692916e780bShannah_mairs PetscReal **A; 2693916e780bShannah_mairs const PetscReal *gllnodes = nodes; 2694916e780bShannah_mairs const PetscInt p = n - 1; 2695916e780bShannah_mairs PetscReal z0, z1, z2 = -1, x, Lpj, Lpr; 2696916e780bShannah_mairs PetscInt i, j, nn, r; 2697916e780bShannah_mairs 2698916e780bShannah_mairs PetscFunctionBegin; 26999566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &A)); 27009566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &A[0])); 2701916e780bShannah_mairs for (i = 1; i < n; i++) A[i] = A[i - 1] + n; 2702916e780bShannah_mairs 2703916e780bShannah_mairs for (j = 1; j < p; j++) { 2704916e780bShannah_mairs x = gllnodes[j]; 2705916e780bShannah_mairs z0 = 1.; 2706916e780bShannah_mairs z1 = x; 2707916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2708916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2709916e780bShannah_mairs z0 = z1; 2710916e780bShannah_mairs z1 = z2; 2711916e780bShannah_mairs } 2712916e780bShannah_mairs Lpj = z2; 2713916e780bShannah_mairs for (r = 1; r < p; r++) { 2714916e780bShannah_mairs if (r == j) { 2715916e780bShannah_mairs A[j][j] = 2. / (3. * (1. - gllnodes[j] * gllnodes[j]) * Lpj * Lpj); 2716916e780bShannah_mairs } else { 2717916e780bShannah_mairs x = gllnodes[r]; 2718916e780bShannah_mairs z0 = 1.; 2719916e780bShannah_mairs z1 = x; 2720916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2721916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2722916e780bShannah_mairs z0 = z1; 2723916e780bShannah_mairs z1 = z2; 2724916e780bShannah_mairs } 2725916e780bShannah_mairs Lpr = z2; 2726916e780bShannah_mairs A[r][j] = 4. / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * Lpr * (gllnodes[j] - gllnodes[r]) * (gllnodes[j] - gllnodes[r])); 2727916e780bShannah_mairs } 2728916e780bShannah_mairs } 2729916e780bShannah_mairs } 2730916e780bShannah_mairs for (j = 1; j < p + 1; j++) { 2731916e780bShannah_mairs x = gllnodes[j]; 2732916e780bShannah_mairs z0 = 1.; 2733916e780bShannah_mairs z1 = x; 2734916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2735916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2736916e780bShannah_mairs z0 = z1; 2737916e780bShannah_mairs z1 = z2; 2738916e780bShannah_mairs } 2739916e780bShannah_mairs Lpj = z2; 2740916e780bShannah_mairs A[j][0] = 4. * PetscPowRealInt(-1., p) / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * (1. + gllnodes[j]) * (1. + gllnodes[j])); 2741916e780bShannah_mairs A[0][j] = A[j][0]; 2742916e780bShannah_mairs } 2743916e780bShannah_mairs for (j = 0; j < p; j++) { 2744916e780bShannah_mairs x = gllnodes[j]; 2745916e780bShannah_mairs z0 = 1.; 2746916e780bShannah_mairs z1 = x; 2747916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2748916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2749916e780bShannah_mairs z0 = z1; 2750916e780bShannah_mairs z1 = z2; 2751916e780bShannah_mairs } 2752916e780bShannah_mairs Lpj = z2; 2753916e780bShannah_mairs 2754916e780bShannah_mairs A[p][j] = 4. / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * (1. - gllnodes[j]) * (1. - gllnodes[j])); 2755916e780bShannah_mairs A[j][p] = A[p][j]; 2756916e780bShannah_mairs } 2757916e780bShannah_mairs A[0][0] = 0.5 + (((PetscReal)p) * (((PetscReal)p) + 1.) - 2.) / 6.; 2758916e780bShannah_mairs A[p][p] = A[0][0]; 2759916e780bShannah_mairs *AA = A; 27603ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2761916e780bShannah_mairs } 2762916e780bShannah_mairs 2763916e780bShannah_mairs /*@C 2764dce8aebaSBarry Smith PetscGaussLobattoLegendreElementLaplacianDestroy - frees the Laplacian for a single 1d GLL element created with `PetscGaussLobattoLegendreElementLaplacianCreate()` 2765916e780bShannah_mairs 2766916e780bShannah_mairs Not Collective 2767916e780bShannah_mairs 2768d8d19677SJose E. Roman Input Parameters: 2769916e780bShannah_mairs + n - the number of GLL nodes 2770916e780bShannah_mairs . nodes - the GLL nodes 2771916e780bShannah_mairs . weights - the GLL weightss 2772916e780bShannah_mairs - A - the stiffness element 2773916e780bShannah_mairs 2774916e780bShannah_mairs Level: beginner 2775916e780bShannah_mairs 2776db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()` 2777916e780bShannah_mairs @*/ 2778d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementLaplacianDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2779d71ae5a4SJacob Faibussowitsch { 2780916e780bShannah_mairs PetscFunctionBegin; 27819566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 27829566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 2783916e780bShannah_mairs *AA = NULL; 27843ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2785916e780bShannah_mairs } 2786916e780bShannah_mairs 2787916e780bShannah_mairs /*@C 2788916e780bShannah_mairs PetscGaussLobattoLegendreElementGradientCreate - computes the gradient for a single 1d GLL element 2789916e780bShannah_mairs 2790916e780bShannah_mairs Not Collective 2791916e780bShannah_mairs 2792916e780bShannah_mairs Input Parameter: 2793916e780bShannah_mairs + n - the number of GLL nodes 2794916e780bShannah_mairs . nodes - the GLL nodes 2795916e780bShannah_mairs . weights - the GLL weights 2796916e780bShannah_mairs 2797d8d19677SJose E. Roman Output Parameters: 2798916e780bShannah_mairs . AA - the stiffness element 2799*20f4b53cSBarry Smith - AAT - the transpose of AA (pass in `NULL` if you do not need this array) 2800916e780bShannah_mairs 2801916e780bShannah_mairs Level: beginner 2802916e780bShannah_mairs 2803916e780bShannah_mairs Notes: 2804dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementGradientDestroy()` 2805916e780bShannah_mairs 2806916e780bShannah_mairs You can access entries in these arrays with AA[i][j] but in memory it is stored in contiguous memory, row oriented 2807916e780bShannah_mairs 2808dce8aebaSBarry Smith .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianDestroy()`, `PetscGaussLobattoLegendreElementGradientDestroy()` 2809916e780bShannah_mairs @*/ 2810d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementGradientCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA, PetscReal ***AAT) 2811d71ae5a4SJacob Faibussowitsch { 2812916e780bShannah_mairs PetscReal **A, **AT = NULL; 2813916e780bShannah_mairs const PetscReal *gllnodes = nodes; 2814916e780bShannah_mairs const PetscInt p = n - 1; 2815e6a796c3SToby Isaac PetscReal Li, Lj, d0; 2816916e780bShannah_mairs PetscInt i, j; 2817916e780bShannah_mairs 2818916e780bShannah_mairs PetscFunctionBegin; 28199566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &A)); 28209566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &A[0])); 2821916e780bShannah_mairs for (i = 1; i < n; i++) A[i] = A[i - 1] + n; 2822916e780bShannah_mairs 2823916e780bShannah_mairs if (AAT) { 28249566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &AT)); 28259566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &AT[0])); 2826916e780bShannah_mairs for (i = 1; i < n; i++) AT[i] = AT[i - 1] + n; 2827916e780bShannah_mairs } 2828916e780bShannah_mairs 2829ad540459SPierre Jolivet if (n == 1) A[0][0] = 0.; 2830916e780bShannah_mairs d0 = (PetscReal)p * ((PetscReal)p + 1.) / 4.; 2831916e780bShannah_mairs for (i = 0; i < n; i++) { 2832916e780bShannah_mairs for (j = 0; j < n; j++) { 2833916e780bShannah_mairs A[i][j] = 0.; 28349566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(0., 0., p, gllnodes[i], &Li)); 28359566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(0., 0., p, gllnodes[j], &Lj)); 2836916e780bShannah_mairs if (i != j) A[i][j] = Li / (Lj * (gllnodes[i] - gllnodes[j])); 2837916e780bShannah_mairs if ((j == i) && (i == 0)) A[i][j] = -d0; 2838916e780bShannah_mairs if (j == i && i == p) A[i][j] = d0; 2839916e780bShannah_mairs if (AT) AT[j][i] = A[i][j]; 2840916e780bShannah_mairs } 2841916e780bShannah_mairs } 2842916e780bShannah_mairs if (AAT) *AAT = AT; 2843916e780bShannah_mairs *AA = A; 28443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2845916e780bShannah_mairs } 2846916e780bShannah_mairs 2847916e780bShannah_mairs /*@C 2848dce8aebaSBarry Smith PetscGaussLobattoLegendreElementGradientDestroy - frees the gradient for a single 1d GLL element obtained with `PetscGaussLobattoLegendreElementGradientCreate()` 2849916e780bShannah_mairs 2850916e780bShannah_mairs Not Collective 2851916e780bShannah_mairs 2852d8d19677SJose E. Roman Input Parameters: 2853916e780bShannah_mairs + n - the number of GLL nodes 2854916e780bShannah_mairs . nodes - the GLL nodes 2855916e780bShannah_mairs . weights - the GLL weights 2856916e780bShannah_mairs . AA - the stiffness element 2857916e780bShannah_mairs - AAT - the transpose of the element 2858916e780bShannah_mairs 2859916e780bShannah_mairs Level: beginner 2860916e780bShannah_mairs 2861db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()`, `PetscGaussLobattoLegendreElementAdvectionCreate()` 2862916e780bShannah_mairs @*/ 2863d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementGradientDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA, PetscReal ***AAT) 2864d71ae5a4SJacob Faibussowitsch { 2865916e780bShannah_mairs PetscFunctionBegin; 28669566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 28679566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 2868916e780bShannah_mairs *AA = NULL; 2869916e780bShannah_mairs if (*AAT) { 28709566063dSJacob Faibussowitsch PetscCall(PetscFree((*AAT)[0])); 28719566063dSJacob Faibussowitsch PetscCall(PetscFree(*AAT)); 2872916e780bShannah_mairs *AAT = NULL; 2873916e780bShannah_mairs } 28743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2875916e780bShannah_mairs } 2876916e780bShannah_mairs 2877916e780bShannah_mairs /*@C 2878916e780bShannah_mairs PetscGaussLobattoLegendreElementAdvectionCreate - computes the advection operator for a single 1d GLL element 2879916e780bShannah_mairs 2880916e780bShannah_mairs Not Collective 2881916e780bShannah_mairs 2882d8d19677SJose E. Roman Input Parameters: 2883916e780bShannah_mairs + n - the number of GLL nodes 2884916e780bShannah_mairs . nodes - the GLL nodes 2885f0fc11ceSJed Brown - weights - the GLL weightss 2886916e780bShannah_mairs 2887916e780bShannah_mairs Output Parameter: 2888916e780bShannah_mairs . AA - the stiffness element 2889916e780bShannah_mairs 2890916e780bShannah_mairs Level: beginner 2891916e780bShannah_mairs 2892916e780bShannah_mairs Notes: 2893dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementAdvectionDestroy()` 2894916e780bShannah_mairs 2895916e780bShannah_mairs This is the same as the Gradient operator multiplied by the diagonal mass matrix 2896916e780bShannah_mairs 2897916e780bShannah_mairs You can access entries in this array with AA[i][j] but in memory it is stored in contiguous memory, row oriented 2898916e780bShannah_mairs 2899db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()`, `PetscGaussLobattoLegendreElementAdvectionDestroy()` 2900916e780bShannah_mairs @*/ 2901d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementAdvectionCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2902d71ae5a4SJacob Faibussowitsch { 2903916e780bShannah_mairs PetscReal **D; 2904916e780bShannah_mairs const PetscReal *gllweights = weights; 2905916e780bShannah_mairs const PetscInt glln = n; 2906916e780bShannah_mairs PetscInt i, j; 2907916e780bShannah_mairs 2908916e780bShannah_mairs PetscFunctionBegin; 29099566063dSJacob Faibussowitsch PetscCall(PetscGaussLobattoLegendreElementGradientCreate(n, nodes, weights, &D, NULL)); 2910916e780bShannah_mairs for (i = 0; i < glln; i++) { 2911ad540459SPierre Jolivet for (j = 0; j < glln; j++) D[i][j] = gllweights[i] * D[i][j]; 2912916e780bShannah_mairs } 2913916e780bShannah_mairs *AA = D; 29143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2915916e780bShannah_mairs } 2916916e780bShannah_mairs 2917916e780bShannah_mairs /*@C 2918dce8aebaSBarry Smith PetscGaussLobattoLegendreElementAdvectionDestroy - frees the advection stiffness for a single 1d GLL element created with `PetscGaussLobattoLegendreElementAdvectionCreate()` 2919916e780bShannah_mairs 2920916e780bShannah_mairs Not Collective 2921916e780bShannah_mairs 2922d8d19677SJose E. Roman Input Parameters: 2923916e780bShannah_mairs + n - the number of GLL nodes 2924916e780bShannah_mairs . nodes - the GLL nodes 2925916e780bShannah_mairs . weights - the GLL weights 2926916e780bShannah_mairs - A - advection 2927916e780bShannah_mairs 2928916e780bShannah_mairs Level: beginner 2929916e780bShannah_mairs 2930db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementAdvectionCreate()` 2931916e780bShannah_mairs @*/ 2932d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementAdvectionDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2933d71ae5a4SJacob Faibussowitsch { 2934916e780bShannah_mairs PetscFunctionBegin; 29359566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 29369566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 2937916e780bShannah_mairs *AA = NULL; 29383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2939916e780bShannah_mairs } 2940916e780bShannah_mairs 2941d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementMassCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2942d71ae5a4SJacob Faibussowitsch { 2943916e780bShannah_mairs PetscReal **A; 2944916e780bShannah_mairs const PetscReal *gllweights = weights; 2945916e780bShannah_mairs const PetscInt glln = n; 2946916e780bShannah_mairs PetscInt i, j; 2947916e780bShannah_mairs 2948916e780bShannah_mairs PetscFunctionBegin; 29499566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(glln, &A)); 29509566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(glln * glln, &A[0])); 2951916e780bShannah_mairs for (i = 1; i < glln; i++) A[i] = A[i - 1] + glln; 2952ad540459SPierre Jolivet if (glln == 1) A[0][0] = 0.; 2953916e780bShannah_mairs for (i = 0; i < glln; i++) { 2954916e780bShannah_mairs for (j = 0; j < glln; j++) { 2955916e780bShannah_mairs A[i][j] = 0.; 2956916e780bShannah_mairs if (j == i) A[i][j] = gllweights[i]; 2957916e780bShannah_mairs } 2958916e780bShannah_mairs } 2959916e780bShannah_mairs *AA = A; 29603ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2961916e780bShannah_mairs } 2962916e780bShannah_mairs 2963d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementMassDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2964d71ae5a4SJacob Faibussowitsch { 2965916e780bShannah_mairs PetscFunctionBegin; 29669566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 29679566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 2968916e780bShannah_mairs *AA = NULL; 29693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2970916e780bShannah_mairs } 2971d4afb720SToby Isaac 2972d4afb720SToby Isaac /*@ 2973d4afb720SToby Isaac PetscDTIndexToBary - convert an index into a barycentric coordinate. 2974d4afb720SToby Isaac 2975d4afb720SToby Isaac Input Parameters: 2976d4afb720SToby Isaac + len - the desired length of the barycentric tuple (usually 1 more than the dimension it represents, so a barycentric coordinate in a triangle has length 3) 2977d4afb720SToby Isaac . sum - the value that the sum of the barycentric coordinates (which will be non-negative integers) should sum to 2978d4afb720SToby Isaac - index - the index to convert: should be >= 0 and < Binomial(len - 1 + sum, sum) 2979d4afb720SToby Isaac 2980d4afb720SToby Isaac Output Parameter: 2981d4afb720SToby Isaac . coord - will be filled with the barycentric coordinate 2982d4afb720SToby Isaac 2983d4afb720SToby Isaac Level: beginner 2984d4afb720SToby Isaac 2985dce8aebaSBarry Smith Note: 2986dce8aebaSBarry Smith The indices map to barycentric coordinates in lexicographic order, where the first index is the 2987d4afb720SToby Isaac least significant and the last index is the most significant. 2988d4afb720SToby Isaac 2989db781477SPatrick Sanan .seealso: `PetscDTBaryToIndex()` 2990d4afb720SToby Isaac @*/ 2991d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTIndexToBary(PetscInt len, PetscInt sum, PetscInt index, PetscInt coord[]) 2992d71ae5a4SJacob Faibussowitsch { 2993d4afb720SToby Isaac PetscInt c, d, s, total, subtotal, nexttotal; 2994d4afb720SToby Isaac 2995d4afb720SToby Isaac PetscFunctionBeginHot; 299608401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 299708401ef6SPierre Jolivet PetscCheck(index >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index must be non-negative"); 2998d4afb720SToby Isaac if (!len) { 29993ba16761SJacob Faibussowitsch if (!sum && !index) PetscFunctionReturn(PETSC_SUCCESS); 3000d4afb720SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid index or sum for length 0 barycentric coordinate"); 3001d4afb720SToby Isaac } 3002d4afb720SToby Isaac for (c = 1, total = 1; c <= len; c++) { 3003d4afb720SToby Isaac /* total is the number of ways to have a tuple of length c with sum */ 3004d4afb720SToby Isaac if (index < total) break; 3005d4afb720SToby Isaac total = (total * (sum + c)) / c; 3006d4afb720SToby Isaac } 300708401ef6SPierre Jolivet PetscCheck(c <= len, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index out of range"); 3008d4afb720SToby Isaac for (d = c; d < len; d++) coord[d] = 0; 3009d4afb720SToby Isaac for (s = 0, subtotal = 1, nexttotal = 1; c > 0;) { 3010d4afb720SToby Isaac /* subtotal is the number of ways to have a tuple of length c with sum s */ 3011d4afb720SToby Isaac /* nexttotal is the number of ways to have a tuple of length c-1 with sum s */ 3012d4afb720SToby Isaac if ((index + subtotal) >= total) { 3013d4afb720SToby Isaac coord[--c] = sum - s; 3014d4afb720SToby Isaac index -= (total - subtotal); 3015d4afb720SToby Isaac sum = s; 3016d4afb720SToby Isaac total = nexttotal; 3017d4afb720SToby Isaac subtotal = 1; 3018d4afb720SToby Isaac nexttotal = 1; 3019d4afb720SToby Isaac s = 0; 3020d4afb720SToby Isaac } else { 3021d4afb720SToby Isaac subtotal = (subtotal * (c + s)) / (s + 1); 3022d4afb720SToby Isaac nexttotal = (nexttotal * (c - 1 + s)) / (s + 1); 3023d4afb720SToby Isaac s++; 3024d4afb720SToby Isaac } 3025d4afb720SToby Isaac } 30263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3027d4afb720SToby Isaac } 3028d4afb720SToby Isaac 3029d4afb720SToby Isaac /*@ 3030d4afb720SToby Isaac PetscDTBaryToIndex - convert a barycentric coordinate to an index 3031d4afb720SToby Isaac 3032d4afb720SToby Isaac Input Parameters: 3033d4afb720SToby Isaac + len - the desired length of the barycentric tuple (usually 1 more than the dimension it represents, so a barycentric coordinate in a triangle has length 3) 3034d4afb720SToby Isaac . sum - the value that the sum of the barycentric coordinates (which will be non-negative integers) should sum to 3035d4afb720SToby Isaac - coord - a barycentric coordinate with the given length and sum 3036d4afb720SToby Isaac 3037d4afb720SToby Isaac Output Parameter: 3038d4afb720SToby Isaac . index - the unique index for the coordinate, >= 0 and < Binomial(len - 1 + sum, sum) 3039d4afb720SToby Isaac 3040d4afb720SToby Isaac Level: beginner 3041d4afb720SToby Isaac 3042dce8aebaSBarry Smith Note: 3043dce8aebaSBarry Smith The indices map to barycentric coordinates in lexicographic order, where the first index is the 3044d4afb720SToby Isaac least significant and the last index is the most significant. 3045d4afb720SToby Isaac 3046db781477SPatrick Sanan .seealso: `PetscDTIndexToBary` 3047d4afb720SToby Isaac @*/ 3048d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTBaryToIndex(PetscInt len, PetscInt sum, const PetscInt coord[], PetscInt *index) 3049d71ae5a4SJacob Faibussowitsch { 3050d4afb720SToby Isaac PetscInt c; 3051d4afb720SToby Isaac PetscInt i; 3052d4afb720SToby Isaac PetscInt total; 3053d4afb720SToby Isaac 3054d4afb720SToby Isaac PetscFunctionBeginHot; 305508401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 3056d4afb720SToby Isaac if (!len) { 3057d4afb720SToby Isaac if (!sum) { 3058d4afb720SToby Isaac *index = 0; 30593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3060d4afb720SToby Isaac } 3061d4afb720SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid index or sum for length 0 barycentric coordinate"); 3062d4afb720SToby Isaac } 3063d4afb720SToby Isaac for (c = 1, total = 1; c < len; c++) total = (total * (sum + c)) / c; 3064d4afb720SToby Isaac i = total - 1; 3065d4afb720SToby Isaac c = len - 1; 3066d4afb720SToby Isaac sum -= coord[c]; 3067d4afb720SToby Isaac while (sum > 0) { 3068d4afb720SToby Isaac PetscInt subtotal; 3069d4afb720SToby Isaac PetscInt s; 3070d4afb720SToby Isaac 3071d4afb720SToby Isaac for (s = 1, subtotal = 1; s < sum; s++) subtotal = (subtotal * (c + s)) / s; 3072d4afb720SToby Isaac i -= subtotal; 3073d4afb720SToby Isaac sum -= coord[--c]; 3074d4afb720SToby Isaac } 3075d4afb720SToby Isaac *index = i; 30763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3077d4afb720SToby Isaac } 3078