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> 707218a29SMatthew G. Knepley #include <petsc/private/petscfeimpl.h> /* For CoordinatesRefToReal() */ 8665c2dedSJed Brown #include <petscviewer.h> 959804f93SMatthew G. Knepley #include <petscdmplex.h> 1059804f93SMatthew G. Knepley #include <petscdmshell.h> 1137045ce4SJed Brown 1298c04793SMatthew G. Knepley #if defined(PETSC_HAVE_MPFR) 1398c04793SMatthew G. Knepley #include <mpfr.h> 1498c04793SMatthew G. Knepley #endif 1598c04793SMatthew G. Knepley 16d3c69ad0SToby Isaac const char *const PetscDTNodeTypes_shifted[] = {"default", "gaussjacobi", "equispaced", "tanhsinh", "PETSCDTNODES_", NULL}; 17d3c69ad0SToby Isaac const char *const *const PetscDTNodeTypes = PetscDTNodeTypes_shifted + 1; 18d3c69ad0SToby Isaac 19d3c69ad0SToby Isaac const char *const PetscDTSimplexQuadratureTypes_shifted[] = {"default", "conic", "minsym", "PETSCDTSIMPLEXQUAD_", NULL}; 20d3c69ad0SToby Isaac const char *const *const PetscDTSimplexQuadratureTypes = PetscDTSimplexQuadratureTypes_shifted + 1; 21d4afb720SToby Isaac 22e6a796c3SToby Isaac static PetscBool GolubWelschCite = PETSC_FALSE; 23e6a796c3SToby Isaac const char GolubWelschCitation[] = "@article{GolubWelsch1969,\n" 240bfcf5a5SMatthew G. Knepley " author = {Golub and Welsch},\n" 250bfcf5a5SMatthew G. Knepley " title = {Calculation of Quadrature Rules},\n" 260bfcf5a5SMatthew G. Knepley " journal = {Math. Comp.},\n" 270bfcf5a5SMatthew G. Knepley " volume = {23},\n" 280bfcf5a5SMatthew G. Knepley " number = {106},\n" 290bfcf5a5SMatthew G. Knepley " pages = {221--230},\n" 300bfcf5a5SMatthew G. Knepley " year = {1969}\n}\n"; 310bfcf5a5SMatthew G. Knepley 32c4762a1bSJed Brown /* Numerical tests in src/dm/dt/tests/ex1.c show that when computing the nodes and weights of Gauss-Jacobi 3394e21283SToby Isaac quadrature rules: 34e6a796c3SToby Isaac 3594e21283SToby Isaac - in double precision, Newton's method and Golub & Welsch both work for moderate degrees (< 100), 3694e21283SToby Isaac - in single precision, Newton's method starts producing incorrect roots around n = 15, but 3794e21283SToby Isaac the weights from Golub & Welsch become a problem before then: they produces errors 3894e21283SToby Isaac in computing the Jacobi-polynomial Gram matrix around n = 6. 3994e21283SToby Isaac 4094e21283SToby Isaac So we default to Newton's method (required fewer dependencies) */ 4194e21283SToby Isaac PetscBool PetscDTGaussQuadratureNewton_Internal = PETSC_TRUE; 422cd22861SMatthew G. Knepley 432cd22861SMatthew G. Knepley PetscClassId PETSCQUADRATURE_CLASSID = 0; 442cd22861SMatthew G. Knepley 4540d8ff71SMatthew G. Knepley /*@ 46dce8aebaSBarry Smith PetscQuadratureCreate - Create a `PetscQuadrature` object 4740d8ff71SMatthew G. Knepley 48d083f849SBarry Smith Collective 4940d8ff71SMatthew G. Knepley 5040d8ff71SMatthew G. Knepley Input Parameter: 51dce8aebaSBarry Smith . comm - The communicator for the `PetscQuadrature` object 5240d8ff71SMatthew G. Knepley 5340d8ff71SMatthew G. Knepley Output Parameter: 5420f4b53cSBarry Smith . q - The `PetscQuadrature` object 5540d8ff71SMatthew G. Knepley 5640d8ff71SMatthew G. Knepley Level: beginner 5740d8ff71SMatthew G. Knepley 58dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `Petscquadraturedestroy()`, `PetscQuadratureGetData()` 5940d8ff71SMatthew G. Knepley @*/ 60d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureCreate(MPI_Comm comm, PetscQuadrature *q) 61d71ae5a4SJacob Faibussowitsch { 6221454ff5SMatthew G. Knepley PetscFunctionBegin; 634f572ea9SToby Isaac PetscAssertPointer(q, 2); 649566063dSJacob Faibussowitsch PetscCall(DMInitializePackage()); 659566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(*q, PETSCQUADRATURE_CLASSID, "PetscQuadrature", "Quadrature", "DT", comm, PetscQuadratureDestroy, PetscQuadratureView)); 664366bac7SMatthew G. Knepley (*q)->ct = DM_POLYTOPE_UNKNOWN; 6721454ff5SMatthew G. Knepley (*q)->dim = -1; 68a6b92713SMatthew G. Knepley (*q)->Nc = 1; 69bcede257SMatthew G. Knepley (*q)->order = -1; 7021454ff5SMatthew G. Knepley (*q)->numPoints = 0; 7121454ff5SMatthew G. Knepley (*q)->points = NULL; 7221454ff5SMatthew G. Knepley (*q)->weights = NULL; 733ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 7421454ff5SMatthew G. Knepley } 7521454ff5SMatthew G. Knepley 76c9638911SMatthew G. Knepley /*@ 77dce8aebaSBarry Smith PetscQuadratureDuplicate - Create a deep copy of the `PetscQuadrature` object 78c9638911SMatthew G. Knepley 7920f4b53cSBarry Smith Collective 80c9638911SMatthew G. Knepley 81c9638911SMatthew G. Knepley Input Parameter: 82dce8aebaSBarry Smith . q - The `PetscQuadrature` object 83c9638911SMatthew G. Knepley 84c9638911SMatthew G. Knepley Output Parameter: 85dce8aebaSBarry Smith . r - The new `PetscQuadrature` object 86c9638911SMatthew G. Knepley 87c9638911SMatthew G. Knepley Level: beginner 88c9638911SMatthew G. Knepley 89dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureDestroy()`, `PetscQuadratureGetData()` 90c9638911SMatthew G. Knepley @*/ 91d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureDuplicate(PetscQuadrature q, PetscQuadrature *r) 92d71ae5a4SJacob Faibussowitsch { 934366bac7SMatthew G. Knepley DMPolytopeType ct; 94a6b92713SMatthew G. Knepley PetscInt order, dim, Nc, Nq; 95c9638911SMatthew G. Knepley const PetscReal *points, *weights; 96c9638911SMatthew G. Knepley PetscReal *p, *w; 97c9638911SMatthew G. Knepley 98c9638911SMatthew G. Knepley PetscFunctionBegin; 994f572ea9SToby Isaac PetscAssertPointer(q, 1); 1009566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PetscObjectComm((PetscObject)q), r)); 1014366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetCellType(q, &ct)); 1024366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*r, ct)); 1039566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q, &order)); 1049566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*r, order)); 1059566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &Nq, &points, &weights)); 1069566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nq * dim, &p)); 1079566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nq * Nc, &w)); 1089566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(p, points, Nq * dim)); 1099566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(w, weights, Nc * Nq)); 1109566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*r, dim, Nc, Nq, p, w)); 1113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 112c9638911SMatthew G. Knepley } 113c9638911SMatthew G. Knepley 11440d8ff71SMatthew G. Knepley /*@ 115dce8aebaSBarry Smith PetscQuadratureDestroy - Destroys a `PetscQuadrature` object 11640d8ff71SMatthew G. Knepley 11720f4b53cSBarry Smith Collective 11840d8ff71SMatthew G. Knepley 11940d8ff71SMatthew G. Knepley Input Parameter: 120dce8aebaSBarry Smith . q - The `PetscQuadrature` object 12140d8ff71SMatthew G. Knepley 12240d8ff71SMatthew G. Knepley Level: beginner 12340d8ff71SMatthew G. Knepley 124dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 12540d8ff71SMatthew G. Knepley @*/ 126d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureDestroy(PetscQuadrature *q) 127d71ae5a4SJacob Faibussowitsch { 128bfa639d9SMatthew G. Knepley PetscFunctionBegin; 1293ba16761SJacob Faibussowitsch if (!*q) PetscFunctionReturn(PETSC_SUCCESS); 1302cd22861SMatthew G. Knepley PetscValidHeaderSpecific((*q), PETSCQUADRATURE_CLASSID, 1); 13121454ff5SMatthew G. Knepley if (--((PetscObject)(*q))->refct > 0) { 13221454ff5SMatthew G. Knepley *q = NULL; 1333ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 13421454ff5SMatthew G. Knepley } 1359566063dSJacob Faibussowitsch PetscCall(PetscFree((*q)->points)); 1369566063dSJacob Faibussowitsch PetscCall(PetscFree((*q)->weights)); 1379566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(q)); 1383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 13921454ff5SMatthew G. Knepley } 14021454ff5SMatthew G. Knepley 141bcede257SMatthew G. Knepley /*@ 1424366bac7SMatthew G. Knepley PetscQuadratureGetCellType - Return the cell type of the integration domain 1434366bac7SMatthew G. Knepley 1444366bac7SMatthew G. Knepley Not Collective 1454366bac7SMatthew G. Knepley 1464366bac7SMatthew G. Knepley Input Parameter: 1474366bac7SMatthew G. Knepley . q - The `PetscQuadrature` object 1484366bac7SMatthew G. Knepley 1494366bac7SMatthew G. Knepley Output Parameter: 1504366bac7SMatthew G. Knepley . ct - The cell type of the integration domain 1514366bac7SMatthew G. Knepley 1524366bac7SMatthew G. Knepley Level: intermediate 1534366bac7SMatthew G. Knepley 1544366bac7SMatthew G. Knepley .seealso: `PetscQuadrature`, `PetscQuadratureSetCellType()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 1554366bac7SMatthew G. Knepley @*/ 1564366bac7SMatthew G. Knepley PetscErrorCode PetscQuadratureGetCellType(PetscQuadrature q, DMPolytopeType *ct) 1574366bac7SMatthew G. Knepley { 1584366bac7SMatthew G. Knepley PetscFunctionBegin; 1594366bac7SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 1604f572ea9SToby Isaac PetscAssertPointer(ct, 2); 1614366bac7SMatthew G. Knepley *ct = q->ct; 1624366bac7SMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 1634366bac7SMatthew G. Knepley } 1644366bac7SMatthew G. Knepley 1654366bac7SMatthew G. Knepley /*@ 1664366bac7SMatthew G. Knepley PetscQuadratureSetCellType - Set the cell type of the integration domain 1674366bac7SMatthew G. Knepley 1684366bac7SMatthew G. Knepley Not Collective 1694366bac7SMatthew G. Knepley 1704366bac7SMatthew G. Knepley Input Parameters: 1714366bac7SMatthew G. Knepley + q - The `PetscQuadrature` object 1724366bac7SMatthew G. Knepley - ct - The cell type of the integration domain 1734366bac7SMatthew G. Knepley 1744366bac7SMatthew G. Knepley Level: intermediate 1754366bac7SMatthew G. Knepley 1764366bac7SMatthew G. Knepley .seealso: `PetscQuadrature`, `PetscQuadratureGetCellType()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 1774366bac7SMatthew G. Knepley @*/ 1784366bac7SMatthew G. Knepley PetscErrorCode PetscQuadratureSetCellType(PetscQuadrature q, DMPolytopeType ct) 1794366bac7SMatthew G. Knepley { 1804366bac7SMatthew G. Knepley PetscFunctionBegin; 1814366bac7SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 1824366bac7SMatthew G. Knepley q->ct = ct; 1834366bac7SMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 1844366bac7SMatthew G. Knepley } 1854366bac7SMatthew G. Knepley 1864366bac7SMatthew G. Knepley /*@ 187dce8aebaSBarry Smith PetscQuadratureGetOrder - Return the order of the method in the `PetscQuadrature` 188bcede257SMatthew G. Knepley 18920f4b53cSBarry Smith Not Collective 190bcede257SMatthew G. Knepley 191bcede257SMatthew G. Knepley Input Parameter: 192dce8aebaSBarry Smith . q - The `PetscQuadrature` object 193bcede257SMatthew G. Knepley 194bcede257SMatthew G. Knepley Output Parameter: 195bcede257SMatthew G. Knepley . order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated 196bcede257SMatthew G. Knepley 197bcede257SMatthew G. Knepley Level: intermediate 198bcede257SMatthew G. Knepley 199dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureSetOrder()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 200bcede257SMatthew G. Knepley @*/ 201d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetOrder(PetscQuadrature q, PetscInt *order) 202d71ae5a4SJacob Faibussowitsch { 203bcede257SMatthew G. Knepley PetscFunctionBegin; 2042cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 2054f572ea9SToby Isaac PetscAssertPointer(order, 2); 206bcede257SMatthew G. Knepley *order = q->order; 2073ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 208bcede257SMatthew G. Knepley } 209bcede257SMatthew G. Knepley 210bcede257SMatthew G. Knepley /*@ 211dce8aebaSBarry Smith PetscQuadratureSetOrder - Set the order of the method in the `PetscQuadrature` 212bcede257SMatthew G. Knepley 21320f4b53cSBarry Smith Not Collective 214bcede257SMatthew G. Knepley 215bcede257SMatthew G. Knepley Input Parameters: 216dce8aebaSBarry Smith + q - The `PetscQuadrature` object 217bcede257SMatthew G. Knepley - order - The order of the quadrature, i.e. the highest degree polynomial that is exactly integrated 218bcede257SMatthew G. Knepley 219bcede257SMatthew G. Knepley Level: intermediate 220bcede257SMatthew G. Knepley 221dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureGetOrder()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 222bcede257SMatthew G. Knepley @*/ 223d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetOrder(PetscQuadrature q, PetscInt order) 224d71ae5a4SJacob Faibussowitsch { 225bcede257SMatthew G. Knepley PetscFunctionBegin; 2262cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 227bcede257SMatthew G. Knepley q->order = order; 2283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 229bcede257SMatthew G. Knepley } 230bcede257SMatthew G. Knepley 231a6b92713SMatthew G. Knepley /*@ 232a6b92713SMatthew G. Knepley PetscQuadratureGetNumComponents - Return the number of components for functions to be integrated 233a6b92713SMatthew G. Knepley 23420f4b53cSBarry Smith Not Collective 235a6b92713SMatthew G. Knepley 236a6b92713SMatthew G. Knepley Input Parameter: 237dce8aebaSBarry Smith . q - The `PetscQuadrature` object 238a6b92713SMatthew G. Knepley 239a6b92713SMatthew G. Knepley Output Parameter: 240a6b92713SMatthew G. Knepley . Nc - The number of components 241a6b92713SMatthew G. Knepley 24220f4b53cSBarry Smith Level: intermediate 24320f4b53cSBarry Smith 244dce8aebaSBarry Smith Note: 245*1d27aa22SBarry Smith We are performing an integral $\int f(x) w(x) dx$, where both $f$ and $w$ (the weight) have `Nc` components. 246a6b92713SMatthew G. Knepley 247dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureSetNumComponents()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 248a6b92713SMatthew G. Knepley @*/ 249d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetNumComponents(PetscQuadrature q, PetscInt *Nc) 250d71ae5a4SJacob Faibussowitsch { 251a6b92713SMatthew G. Knepley PetscFunctionBegin; 2522cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 2534f572ea9SToby Isaac PetscAssertPointer(Nc, 2); 254a6b92713SMatthew G. Knepley *Nc = q->Nc; 2553ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 256a6b92713SMatthew G. Knepley } 257a6b92713SMatthew G. Knepley 258a6b92713SMatthew G. Knepley /*@ 259*1d27aa22SBarry Smith PetscQuadratureSetNumComponents - Sets the number of components for functions to be integrated 260a6b92713SMatthew G. Knepley 26120f4b53cSBarry Smith Not Collective 262a6b92713SMatthew G. Knepley 263a6b92713SMatthew G. Knepley Input Parameters: 2642fe279fdSBarry Smith + q - The `PetscQuadrature` object 265a6b92713SMatthew G. Knepley - Nc - The number of components 266a6b92713SMatthew G. Knepley 26720f4b53cSBarry Smith Level: intermediate 26820f4b53cSBarry Smith 269dce8aebaSBarry Smith Note: 270*1d27aa22SBarry Smith We are performing an integral $\int f(x) w(x) dx$, where both $f$ and $w$ (the weight) have `Nc` components. 271a6b92713SMatthew G. Knepley 272dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureGetNumComponents()`, `PetscQuadratureGetData()`, `PetscQuadratureSetData()` 273a6b92713SMatthew G. Knepley @*/ 274d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetNumComponents(PetscQuadrature q, PetscInt Nc) 275d71ae5a4SJacob Faibussowitsch { 276a6b92713SMatthew G. Knepley PetscFunctionBegin; 2772cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 278a6b92713SMatthew G. Knepley q->Nc = Nc; 2793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 280a6b92713SMatthew G. Knepley } 281a6b92713SMatthew G. Knepley 28240d8ff71SMatthew G. Knepley /*@C 283dce8aebaSBarry Smith PetscQuadratureGetData - Returns the data defining the `PetscQuadrature` 28440d8ff71SMatthew G. Knepley 28520f4b53cSBarry Smith Not Collective 28640d8ff71SMatthew G. Knepley 28740d8ff71SMatthew G. Knepley Input Parameter: 288dce8aebaSBarry Smith . q - The `PetscQuadrature` object 28940d8ff71SMatthew G. Knepley 29040d8ff71SMatthew G. Knepley Output Parameters: 29140d8ff71SMatthew G. Knepley + dim - The spatial dimension 292805e7170SToby Isaac . Nc - The number of components 29340d8ff71SMatthew G. Knepley . npoints - The number of quadrature points 29440d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point 29540d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point 29640d8ff71SMatthew G. Knepley 29740d8ff71SMatthew G. Knepley Level: intermediate 29840d8ff71SMatthew G. Knepley 299*1d27aa22SBarry Smith Fortran Note: 300*1d27aa22SBarry Smith Call `PetscQuadratureRestoreData()` when you are done with the data 3011fd49c25SBarry Smith 302dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureSetData()` 30340d8ff71SMatthew G. Knepley @*/ 304d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureGetData(PetscQuadrature q, PetscInt *dim, PetscInt *Nc, PetscInt *npoints, const PetscReal *points[], const PetscReal *weights[]) 305d71ae5a4SJacob Faibussowitsch { 30621454ff5SMatthew G. Knepley PetscFunctionBegin; 3072cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 30821454ff5SMatthew G. Knepley if (dim) { 3094f572ea9SToby Isaac PetscAssertPointer(dim, 2); 31021454ff5SMatthew G. Knepley *dim = q->dim; 31121454ff5SMatthew G. Knepley } 312a6b92713SMatthew G. Knepley if (Nc) { 3134f572ea9SToby Isaac PetscAssertPointer(Nc, 3); 314a6b92713SMatthew G. Knepley *Nc = q->Nc; 315a6b92713SMatthew G. Knepley } 31621454ff5SMatthew G. Knepley if (npoints) { 3174f572ea9SToby Isaac PetscAssertPointer(npoints, 4); 31821454ff5SMatthew G. Knepley *npoints = q->numPoints; 31921454ff5SMatthew G. Knepley } 32021454ff5SMatthew G. Knepley if (points) { 3214f572ea9SToby Isaac PetscAssertPointer(points, 5); 32221454ff5SMatthew G. Knepley *points = q->points; 32321454ff5SMatthew G. Knepley } 32421454ff5SMatthew G. Knepley if (weights) { 3254f572ea9SToby Isaac PetscAssertPointer(weights, 6); 32621454ff5SMatthew G. Knepley *weights = q->weights; 32721454ff5SMatthew G. Knepley } 3283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 32921454ff5SMatthew G. Knepley } 33021454ff5SMatthew G. Knepley 3314f9ab2b4SJed Brown /*@ 3324f9ab2b4SJed Brown PetscQuadratureEqual - determine whether two quadratures are equivalent 3334f9ab2b4SJed Brown 3344f9ab2b4SJed Brown Input Parameters: 335dce8aebaSBarry Smith + A - A `PetscQuadrature` object 336dce8aebaSBarry Smith - B - Another `PetscQuadrature` object 3374f9ab2b4SJed Brown 3382fe279fdSBarry Smith Output Parameter: 339dce8aebaSBarry Smith . equal - `PETSC_TRUE` if the quadratures are the same 3404f9ab2b4SJed Brown 3414f9ab2b4SJed Brown Level: intermediate 3424f9ab2b4SJed Brown 343dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()` 3444f9ab2b4SJed Brown @*/ 345d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureEqual(PetscQuadrature A, PetscQuadrature B, PetscBool *equal) 346d71ae5a4SJacob Faibussowitsch { 3474f9ab2b4SJed Brown PetscFunctionBegin; 3484f9ab2b4SJed Brown PetscValidHeaderSpecific(A, PETSCQUADRATURE_CLASSID, 1); 3494f9ab2b4SJed Brown PetscValidHeaderSpecific(B, PETSCQUADRATURE_CLASSID, 2); 3504f572ea9SToby Isaac PetscAssertPointer(equal, 3); 3514f9ab2b4SJed Brown *equal = PETSC_FALSE; 3524366bac7SMatthew G. Knepley if (A->ct != B->ct || A->dim != B->dim || A->Nc != B->Nc || A->order != B->order || A->numPoints != B->numPoints) PetscFunctionReturn(PETSC_SUCCESS); 3534f9ab2b4SJed Brown for (PetscInt i = 0; i < A->numPoints * A->dim; i++) { 3543ba16761SJacob Faibussowitsch if (PetscAbsReal(A->points[i] - B->points[i]) > PETSC_SMALL) PetscFunctionReturn(PETSC_SUCCESS); 3554f9ab2b4SJed Brown } 3564f9ab2b4SJed Brown if (!A->weights && !B->weights) { 3574f9ab2b4SJed Brown *equal = PETSC_TRUE; 3583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3594f9ab2b4SJed Brown } 3604f9ab2b4SJed Brown if (A->weights && B->weights) { 3614f9ab2b4SJed Brown for (PetscInt i = 0; i < A->numPoints; i++) { 3623ba16761SJacob Faibussowitsch if (PetscAbsReal(A->weights[i] - B->weights[i]) > PETSC_SMALL) PetscFunctionReturn(PETSC_SUCCESS); 3634f9ab2b4SJed Brown } 3644f9ab2b4SJed Brown *equal = PETSC_TRUE; 3654f9ab2b4SJed Brown } 3663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3674f9ab2b4SJed Brown } 3684f9ab2b4SJed Brown 369d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobianInverse_Internal(PetscInt m, PetscInt n, const PetscReal J[], PetscReal Jinv[]) 370d71ae5a4SJacob Faibussowitsch { 371907761f8SToby Isaac PetscScalar *Js, *Jinvs; 372907761f8SToby Isaac PetscInt i, j, k; 373907761f8SToby Isaac PetscBLASInt bm, bn, info; 374907761f8SToby Isaac 375907761f8SToby Isaac PetscFunctionBegin; 3763ba16761SJacob Faibussowitsch if (!m || !n) PetscFunctionReturn(PETSC_SUCCESS); 3779566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(m, &bm)); 3789566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &bn)); 379907761f8SToby Isaac #if defined(PETSC_USE_COMPLEX) 3809566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m * n, &Js, m * n, &Jinvs)); 38128222859SToby Isaac for (i = 0; i < m * n; i++) Js[i] = J[i]; 382907761f8SToby Isaac #else 383907761f8SToby Isaac Js = (PetscReal *)J; 384907761f8SToby Isaac Jinvs = Jinv; 385907761f8SToby Isaac #endif 386907761f8SToby Isaac if (m == n) { 387907761f8SToby Isaac PetscBLASInt *pivots; 388907761f8SToby Isaac PetscScalar *W; 389907761f8SToby Isaac 3909566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m, &pivots, m, &W)); 391907761f8SToby Isaac 3929566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(Jinvs, Js, m * m)); 393792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bm, &bm, Jinvs, &bm, pivots, &info)); 39463a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 395792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bm, Jinvs, &bm, pivots, W, &bm, &info)); 39663a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 3979566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 398907761f8SToby Isaac } else if (m < n) { 399907761f8SToby Isaac PetscScalar *JJT; 400907761f8SToby Isaac PetscBLASInt *pivots; 401907761f8SToby Isaac PetscScalar *W; 402907761f8SToby Isaac 4039566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m * m, &JJT)); 4049566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m, &pivots, m, &W)); 405907761f8SToby Isaac for (i = 0; i < m; i++) { 406907761f8SToby Isaac for (j = 0; j < m; j++) { 407907761f8SToby Isaac PetscScalar val = 0.; 408907761f8SToby Isaac 409907761f8SToby Isaac for (k = 0; k < n; k++) val += Js[i * n + k] * Js[j * n + k]; 410907761f8SToby Isaac JJT[i * m + j] = val; 411907761f8SToby Isaac } 412907761f8SToby Isaac } 413907761f8SToby Isaac 414792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bm, &bm, JJT, &bm, pivots, &info)); 41563a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 416792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bm, JJT, &bm, pivots, W, &bm, &info)); 41763a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 418907761f8SToby Isaac for (i = 0; i < n; i++) { 419907761f8SToby Isaac for (j = 0; j < m; j++) { 420907761f8SToby Isaac PetscScalar val = 0.; 421907761f8SToby Isaac 422907761f8SToby Isaac for (k = 0; k < m; k++) val += Js[k * n + i] * JJT[k * m + j]; 423907761f8SToby Isaac Jinvs[i * m + j] = val; 424907761f8SToby Isaac } 425907761f8SToby Isaac } 4269566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 4279566063dSJacob Faibussowitsch PetscCall(PetscFree(JJT)); 428907761f8SToby Isaac } else { 429907761f8SToby Isaac PetscScalar *JTJ; 430907761f8SToby Isaac PetscBLASInt *pivots; 431907761f8SToby Isaac PetscScalar *W; 432907761f8SToby Isaac 4339566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &JTJ)); 4349566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(n, &pivots, n, &W)); 435907761f8SToby Isaac for (i = 0; i < n; i++) { 436907761f8SToby Isaac for (j = 0; j < n; j++) { 437907761f8SToby Isaac PetscScalar val = 0.; 438907761f8SToby Isaac 439907761f8SToby Isaac for (k = 0; k < m; k++) val += Js[k * n + i] * Js[k * n + j]; 440907761f8SToby Isaac JTJ[i * n + j] = val; 441907761f8SToby Isaac } 442907761f8SToby Isaac } 443907761f8SToby Isaac 444792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&bn, &bn, JTJ, &bn, pivots, &info)); 44563a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetrf %" PetscInt_FMT, (PetscInt)info); 446792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&bn, JTJ, &bn, pivots, W, &bn, &info)); 44763a3b9bcSJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Error returned from LAPACKgetri %" PetscInt_FMT, (PetscInt)info); 448907761f8SToby Isaac for (i = 0; i < n; i++) { 449907761f8SToby Isaac for (j = 0; j < m; j++) { 450907761f8SToby Isaac PetscScalar val = 0.; 451907761f8SToby Isaac 452907761f8SToby Isaac for (k = 0; k < n; k++) val += JTJ[i * n + k] * Js[j * n + k]; 453907761f8SToby Isaac Jinvs[i * m + j] = val; 454907761f8SToby Isaac } 455907761f8SToby Isaac } 4569566063dSJacob Faibussowitsch PetscCall(PetscFree2(pivots, W)); 4579566063dSJacob Faibussowitsch PetscCall(PetscFree(JTJ)); 458907761f8SToby Isaac } 459907761f8SToby Isaac #if defined(PETSC_USE_COMPLEX) 46028222859SToby Isaac for (i = 0; i < m * n; i++) Jinv[i] = PetscRealPart(Jinvs[i]); 4619566063dSJacob Faibussowitsch PetscCall(PetscFree2(Js, Jinvs)); 462907761f8SToby Isaac #endif 4633ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 464907761f8SToby Isaac } 465907761f8SToby Isaac 466907761f8SToby Isaac /*@ 467907761f8SToby Isaac PetscQuadraturePushForward - Push forward a quadrature functional under an affine transformation. 468907761f8SToby Isaac 46920f4b53cSBarry Smith Collective 470907761f8SToby Isaac 4714165533cSJose E. Roman Input Parameters: 472907761f8SToby Isaac + q - the quadrature functional 473907761f8SToby Isaac . imageDim - the dimension of the image of the transformation 474907761f8SToby Isaac . origin - a point in the original space 475907761f8SToby Isaac . originImage - the image of the origin under the transformation 476907761f8SToby Isaac . J - the Jacobian of the image: an [imageDim x dim] matrix in row major order 477*1d27aa22SBarry 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`), 478*1d27aa22SBarry Smith it is assumed that they represent multiple k-forms) [see `PetscDTAltVPullback()` for interpretation of `formDegree`] 479907761f8SToby Isaac 4802fe279fdSBarry Smith Output Parameter: 481*1d27aa22SBarry Smith . 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 482*1d27aa22SBarry Smith been pulled-back by the pseudoinverse of `J` to the k-form weights in the image space. 483907761f8SToby Isaac 4846c877ef6SSatish Balay Level: intermediate 4856c877ef6SSatish Balay 486dce8aebaSBarry Smith Note: 487*1d27aa22SBarry Smith The new quadrature rule will have a different number of components if spaces have different dimensions. 488*1d27aa22SBarry Smith 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. 489dce8aebaSBarry Smith 490dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTAltVPullback()`, `PetscDTAltVPullbackMatrix()` 491907761f8SToby Isaac @*/ 492d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadraturePushForward(PetscQuadrature q, PetscInt imageDim, const PetscReal origin[], const PetscReal originImage[], const PetscReal J[], PetscInt formDegree, PetscQuadrature *Jinvstarq) 493d71ae5a4SJacob Faibussowitsch { 494907761f8SToby Isaac PetscInt dim, Nc, imageNc, formSize, Ncopies, imageFormSize, Npoints, pt, i, j, c; 495907761f8SToby Isaac const PetscReal *points; 496907761f8SToby Isaac const PetscReal *weights; 497907761f8SToby Isaac PetscReal *imagePoints, *imageWeights; 498907761f8SToby Isaac PetscReal *Jinv; 499907761f8SToby Isaac PetscReal *Jinvstar; 500907761f8SToby Isaac 501907761f8SToby Isaac PetscFunctionBegin; 502d4afb720SToby Isaac PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 50363a3b9bcSJacob 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); 5049566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &Npoints, &points, &weights)); 5059566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, PetscAbsInt(formDegree), &formSize)); 50663a3b9bcSJacob 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); 507907761f8SToby Isaac Ncopies = Nc / formSize; 5089566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(imageDim, PetscAbsInt(formDegree), &imageFormSize)); 509907761f8SToby Isaac imageNc = Ncopies * imageFormSize; 5109566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Npoints * imageDim, &imagePoints)); 5119566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Npoints * imageNc, &imageWeights)); 5129566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(imageDim * dim, &Jinv, formSize * imageFormSize, &Jinvstar)); 5139566063dSJacob Faibussowitsch PetscCall(PetscDTJacobianInverse_Internal(imageDim, dim, J, Jinv)); 5149566063dSJacob Faibussowitsch PetscCall(PetscDTAltVPullbackMatrix(imageDim, dim, Jinv, formDegree, Jinvstar)); 515907761f8SToby Isaac for (pt = 0; pt < Npoints; pt++) { 516907761f8SToby Isaac const PetscReal *point = &points[pt * dim]; 517907761f8SToby Isaac PetscReal *imagePoint = &imagePoints[pt * imageDim]; 518907761f8SToby Isaac 519907761f8SToby Isaac for (i = 0; i < imageDim; i++) { 520907761f8SToby Isaac PetscReal val = originImage[i]; 521907761f8SToby Isaac 522907761f8SToby Isaac for (j = 0; j < dim; j++) val += J[i * dim + j] * (point[j] - origin[j]); 523907761f8SToby Isaac imagePoint[i] = val; 524907761f8SToby Isaac } 525907761f8SToby Isaac for (c = 0; c < Ncopies; c++) { 526907761f8SToby Isaac const PetscReal *form = &weights[pt * Nc + c * formSize]; 527907761f8SToby Isaac PetscReal *imageForm = &imageWeights[pt * imageNc + c * imageFormSize]; 528907761f8SToby Isaac 529907761f8SToby Isaac for (i = 0; i < imageFormSize; i++) { 530907761f8SToby Isaac PetscReal val = 0.; 531907761f8SToby Isaac 532907761f8SToby Isaac for (j = 0; j < formSize; j++) val += Jinvstar[i * formSize + j] * form[j]; 533907761f8SToby Isaac imageForm[i] = val; 534907761f8SToby Isaac } 535907761f8SToby Isaac } 536907761f8SToby Isaac } 5379566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PetscObjectComm((PetscObject)q), Jinvstarq)); 5389566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*Jinvstarq, imageDim, imageNc, Npoints, imagePoints, imageWeights)); 5399566063dSJacob Faibussowitsch PetscCall(PetscFree2(Jinv, Jinvstar)); 5403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 541907761f8SToby Isaac } 542907761f8SToby Isaac 54340d8ff71SMatthew G. Knepley /*@C 54440d8ff71SMatthew G. Knepley PetscQuadratureSetData - Sets the data defining the quadrature 54540d8ff71SMatthew G. Knepley 54620f4b53cSBarry Smith Not Collective 54740d8ff71SMatthew G. Knepley 54840d8ff71SMatthew G. Knepley Input Parameters: 549dce8aebaSBarry Smith + q - The `PetscQuadrature` object 55040d8ff71SMatthew G. Knepley . dim - The spatial dimension 551e2b35d93SBarry Smith . Nc - The number of components 55240d8ff71SMatthew G. Knepley . npoints - The number of quadrature points 55340d8ff71SMatthew G. Knepley . points - The coordinates of each quadrature point 55440d8ff71SMatthew G. Knepley - weights - The weight of each quadrature point 55540d8ff71SMatthew G. Knepley 55640d8ff71SMatthew G. Knepley Level: intermediate 55740d8ff71SMatthew G. Knepley 558dce8aebaSBarry Smith Note: 559dce8aebaSBarry Smith This routine owns the references to points and weights, so they must be allocated using `PetscMalloc()` and the user should not free them. 560dce8aebaSBarry Smith 561dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 56240d8ff71SMatthew G. Knepley @*/ 563d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureSetData(PetscQuadrature q, PetscInt dim, PetscInt Nc, PetscInt npoints, const PetscReal points[], const PetscReal weights[]) 564d71ae5a4SJacob Faibussowitsch { 56521454ff5SMatthew G. Knepley PetscFunctionBegin; 5662cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 56721454ff5SMatthew G. Knepley if (dim >= 0) q->dim = dim; 568a6b92713SMatthew G. Knepley if (Nc >= 0) q->Nc = Nc; 56921454ff5SMatthew G. Knepley if (npoints >= 0) q->numPoints = npoints; 57021454ff5SMatthew G. Knepley if (points) { 5714f572ea9SToby Isaac PetscAssertPointer(points, 5); 57221454ff5SMatthew G. Knepley q->points = points; 57321454ff5SMatthew G. Knepley } 57421454ff5SMatthew G. Knepley if (weights) { 5754f572ea9SToby Isaac PetscAssertPointer(weights, 6); 57621454ff5SMatthew G. Knepley q->weights = weights; 57721454ff5SMatthew G. Knepley } 5783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 579f9fd7fdbSMatthew G. Knepley } 580f9fd7fdbSMatthew G. Knepley 581d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscQuadratureView_Ascii(PetscQuadrature quad, PetscViewer v) 582d71ae5a4SJacob Faibussowitsch { 583d9bac1caSLisandro Dalcin PetscInt q, d, c; 584d9bac1caSLisandro Dalcin PetscViewerFormat format; 585d9bac1caSLisandro Dalcin 586d9bac1caSLisandro Dalcin PetscFunctionBegin; 5874366bac7SMatthew G. Knepley if (quad->Nc > 1) 5884366bac7SMatthew G. Knepley PetscCall(PetscViewerASCIIPrintf(v, "Quadrature on a %s of order %" PetscInt_FMT " on %" PetscInt_FMT " points (dim %" PetscInt_FMT ") with %" PetscInt_FMT " components\n", DMPolytopeTypes[quad->ct], quad->order, quad->numPoints, quad->dim, quad->Nc)); 5894366bac7SMatthew G. Knepley else PetscCall(PetscViewerASCIIPrintf(v, "Quadrature on a %s of order %" PetscInt_FMT " on %" PetscInt_FMT " points (dim %" PetscInt_FMT ")\n", DMPolytopeTypes[quad->ct], quad->order, quad->numPoints, quad->dim)); 5909566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(v, &format)); 5913ba16761SJacob Faibussowitsch if (format != PETSC_VIEWER_ASCII_INFO_DETAIL) PetscFunctionReturn(PETSC_SUCCESS); 592d9bac1caSLisandro Dalcin for (q = 0; q < quad->numPoints; ++q) { 59363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "p%" PetscInt_FMT " (", q)); 5949566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(v, PETSC_FALSE)); 595d9bac1caSLisandro Dalcin for (d = 0; d < quad->dim; ++d) { 5969566063dSJacob Faibussowitsch if (d) PetscCall(PetscViewerASCIIPrintf(v, ", ")); 5979566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "%+g", (double)quad->points[q * quad->dim + d])); 598d9bac1caSLisandro Dalcin } 5999566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, ") ")); 60063a3b9bcSJacob Faibussowitsch if (quad->Nc > 1) PetscCall(PetscViewerASCIIPrintf(v, "w%" PetscInt_FMT " (", q)); 601d9bac1caSLisandro Dalcin for (c = 0; c < quad->Nc; ++c) { 6029566063dSJacob Faibussowitsch if (c) PetscCall(PetscViewerASCIIPrintf(v, ", ")); 6039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "%+g", (double)quad->weights[q * quad->Nc + c])); 604d9bac1caSLisandro Dalcin } 6059566063dSJacob Faibussowitsch if (quad->Nc > 1) PetscCall(PetscViewerASCIIPrintf(v, ")")); 6069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "\n")); 6079566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(v, PETSC_TRUE)); 608d9bac1caSLisandro Dalcin } 6093ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 610d9bac1caSLisandro Dalcin } 611d9bac1caSLisandro Dalcin 61240d8ff71SMatthew G. Knepley /*@C 613dce8aebaSBarry Smith PetscQuadratureView - View a `PetscQuadrature` object 61440d8ff71SMatthew G. Knepley 61520f4b53cSBarry Smith Collective 61640d8ff71SMatthew G. Knepley 61740d8ff71SMatthew G. Knepley Input Parameters: 618dce8aebaSBarry Smith + quad - The `PetscQuadrature` object 619dce8aebaSBarry Smith - viewer - The `PetscViewer` object 62040d8ff71SMatthew G. Knepley 62140d8ff71SMatthew G. Knepley Level: beginner 62240d8ff71SMatthew G. Knepley 623dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscViewer`, `PetscQuadratureCreate()`, `PetscQuadratureGetData()` 62440d8ff71SMatthew G. Knepley @*/ 625d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureView(PetscQuadrature quad, PetscViewer viewer) 626d71ae5a4SJacob Faibussowitsch { 627d9bac1caSLisandro Dalcin PetscBool iascii; 628f9fd7fdbSMatthew G. Knepley 629f9fd7fdbSMatthew G. Knepley PetscFunctionBegin; 630d9bac1caSLisandro Dalcin PetscValidHeader(quad, 1); 631d9bac1caSLisandro Dalcin if (viewer) PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 6329566063dSJacob Faibussowitsch if (!viewer) PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)quad), &viewer)); 6339566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii)); 6349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6359566063dSJacob Faibussowitsch if (iascii) PetscCall(PetscQuadratureView_Ascii(quad, viewer)); 6369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 6373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 638bfa639d9SMatthew G. Knepley } 639bfa639d9SMatthew G. Knepley 64089710940SMatthew G. Knepley /*@C 64189710940SMatthew G. Knepley PetscQuadratureExpandComposite - Return a quadrature over the composite element, which has the original quadrature in each subelement 64289710940SMatthew G. Knepley 64320f4b53cSBarry Smith Not Collective; No Fortran Support 64489710940SMatthew G. Knepley 645d8d19677SJose E. Roman Input Parameters: 646dce8aebaSBarry Smith + q - The original `PetscQuadrature` 64789710940SMatthew G. Knepley . numSubelements - The number of subelements the original element is divided into 64889710940SMatthew G. Knepley . v0 - An array of the initial points for each subelement 64989710940SMatthew G. Knepley - jac - An array of the Jacobian mappings from the reference to each subelement 65089710940SMatthew G. Knepley 6512fe279fdSBarry Smith Output Parameter: 65260225df5SJacob Faibussowitsch . qref - The dimension 65389710940SMatthew G. Knepley 65420f4b53cSBarry Smith Level: intermediate 65520f4b53cSBarry Smith 656dce8aebaSBarry Smith Note: 657*1d27aa22SBarry Smith Together `v0` and `jac` define an affine mapping from the original reference element to each subelement 65889710940SMatthew G. Knepley 659dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscFECreate()`, `PetscSpaceGetDimension()`, `PetscDualSpaceGetDimension()` 66089710940SMatthew G. Knepley @*/ 661d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscQuadratureExpandComposite(PetscQuadrature q, PetscInt numSubelements, const PetscReal v0[], const PetscReal jac[], PetscQuadrature *qref) 662d71ae5a4SJacob Faibussowitsch { 6634366bac7SMatthew G. Knepley DMPolytopeType ct; 66489710940SMatthew G. Knepley const PetscReal *points, *weights; 66589710940SMatthew G. Knepley PetscReal *pointsRef, *weightsRef; 666a6b92713SMatthew G. Knepley PetscInt dim, Nc, order, npoints, npointsRef, c, p, cp, d, e; 66789710940SMatthew G. Knepley 66889710940SMatthew G. Knepley PetscFunctionBegin; 6692cd22861SMatthew G. Knepley PetscValidHeaderSpecific(q, PETSCQUADRATURE_CLASSID, 1); 6704f572ea9SToby Isaac PetscAssertPointer(v0, 3); 6714f572ea9SToby Isaac PetscAssertPointer(jac, 4); 6724f572ea9SToby Isaac PetscAssertPointer(qref, 5); 6739566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, qref)); 6744366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetCellType(q, &ct)); 6759566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q, &order)); 6769566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q, &dim, &Nc, &npoints, &points, &weights)); 67789710940SMatthew G. Knepley npointsRef = npoints * numSubelements; 6789566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npointsRef * dim, &pointsRef)); 6799566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npointsRef * Nc, &weightsRef)); 68089710940SMatthew G. Knepley for (c = 0; c < numSubelements; ++c) { 68189710940SMatthew G. Knepley for (p = 0; p < npoints; ++p) { 68289710940SMatthew G. Knepley for (d = 0; d < dim; ++d) { 68389710940SMatthew G. Knepley pointsRef[(c * npoints + p) * dim + d] = v0[c * dim + d]; 684ad540459SPierre 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); 68589710940SMatthew G. Knepley } 68689710940SMatthew G. Knepley /* Could also use detJ here */ 687a6b92713SMatthew G. Knepley for (cp = 0; cp < Nc; ++cp) weightsRef[(c * npoints + p) * Nc + cp] = weights[p * Nc + cp] / numSubelements; 68889710940SMatthew G. Knepley } 68989710940SMatthew G. Knepley } 6904366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*qref, ct)); 6919566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*qref, order)); 6929566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*qref, dim, Nc, npointsRef, pointsRef, weightsRef)); 6933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 69489710940SMatthew G. Knepley } 69589710940SMatthew G. Knepley 69694e21283SToby Isaac /* Compute the coefficients for the Jacobi polynomial recurrence, 697*1d27aa22SBarry Smith 698*1d27aa22SBarry Smith J^{a,b}_n(x) = (cnm1 + cnm1x * x) * J^{a,b}_{n-1}(x) - cnm2 * J^{a,b}_{n-2}(x). 69994e21283SToby Isaac */ 70094e21283SToby Isaac #define PetscDTJacobiRecurrence_Internal(n, a, b, cnm1, cnm1x, cnm2) \ 70194e21283SToby Isaac do { \ 70294e21283SToby Isaac PetscReal _a = (a); \ 70394e21283SToby Isaac PetscReal _b = (b); \ 70494e21283SToby Isaac PetscReal _n = (n); \ 70594e21283SToby Isaac if (n == 1) { \ 70694e21283SToby Isaac (cnm1) = (_a - _b) * 0.5; \ 70794e21283SToby Isaac (cnm1x) = (_a + _b + 2.) * 0.5; \ 70894e21283SToby Isaac (cnm2) = 0.; \ 70994e21283SToby Isaac } else { \ 71094e21283SToby Isaac PetscReal _2n = _n + _n; \ 71194e21283SToby Isaac PetscReal _d = (_2n * (_n + _a + _b) * (_2n + _a + _b - 2)); \ 71294e21283SToby Isaac PetscReal _n1 = (_2n + _a + _b - 1.) * (_a * _a - _b * _b); \ 71394e21283SToby Isaac PetscReal _n1x = (_2n + _a + _b - 1.) * (_2n + _a + _b) * (_2n + _a + _b - 2); \ 71494e21283SToby Isaac PetscReal _n2 = 2. * ((_n + _a - 1.) * (_n + _b - 1.) * (_2n + _a + _b)); \ 71594e21283SToby Isaac (cnm1) = _n1 / _d; \ 71694e21283SToby Isaac (cnm1x) = _n1x / _d; \ 71794e21283SToby Isaac (cnm2) = _n2 / _d; \ 71894e21283SToby Isaac } \ 71994e21283SToby Isaac } while (0) 72094e21283SToby Isaac 721fbdc3dfeSToby Isaac /*@ 722fbdc3dfeSToby Isaac PetscDTJacobiNorm - Compute the weighted L2 norm of a Jacobi polynomial. 723fbdc3dfeSToby Isaac 724fbdc3dfeSToby Isaac $\| P^{\alpha,\beta}_n \|_{\alpha,\beta}^2 = \int_{-1}^1 (1 + x)^{\alpha} (1 - x)^{\beta} P^{\alpha,\beta}_n (x)^2 dx.$ 725fbdc3dfeSToby Isaac 7264165533cSJose E. Roman Input Parameters: 72760225df5SJacob Faibussowitsch + alpha - the left exponent > -1 728fbdc3dfeSToby Isaac . beta - the right exponent > -1 72960225df5SJacob Faibussowitsch - n - the polynomial degree 730fbdc3dfeSToby Isaac 7314165533cSJose E. Roman Output Parameter: 732fbdc3dfeSToby Isaac . norm - the weighted L2 norm 733fbdc3dfeSToby Isaac 734fbdc3dfeSToby Isaac Level: beginner 735fbdc3dfeSToby Isaac 736dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTJacobiEval()` 737fbdc3dfeSToby Isaac @*/ 738d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiNorm(PetscReal alpha, PetscReal beta, PetscInt n, PetscReal *norm) 739d71ae5a4SJacob Faibussowitsch { 740fbdc3dfeSToby Isaac PetscReal twoab1; 741fbdc3dfeSToby Isaac PetscReal gr; 742fbdc3dfeSToby Isaac 743fbdc3dfeSToby Isaac PetscFunctionBegin; 74408401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Exponent alpha %g <= -1. invalid", (double)alpha); 74508401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Exponent beta %g <= -1. invalid", (double)beta); 74663a3b9bcSJacob Faibussowitsch PetscCheck(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "n %" PetscInt_FMT " < 0 invalid", n); 747fbdc3dfeSToby Isaac twoab1 = PetscPowReal(2., alpha + beta + 1.); 748fbdc3dfeSToby Isaac #if defined(PETSC_HAVE_LGAMMA) 749fbdc3dfeSToby Isaac if (!n) { 750fbdc3dfeSToby Isaac gr = PetscExpReal(PetscLGamma(alpha + 1.) + PetscLGamma(beta + 1.) - PetscLGamma(alpha + beta + 2.)); 751fbdc3dfeSToby Isaac } else { 752fbdc3dfeSToby Isaac gr = PetscExpReal(PetscLGamma(n + alpha + 1.) + PetscLGamma(n + beta + 1.) - (PetscLGamma(n + 1.) + PetscLGamma(n + alpha + beta + 1.))) / (n + n + alpha + beta + 1.); 753fbdc3dfeSToby Isaac } 754fbdc3dfeSToby Isaac #else 755fbdc3dfeSToby Isaac { 756fbdc3dfeSToby Isaac PetscInt alphai = (PetscInt)alpha; 757fbdc3dfeSToby Isaac PetscInt betai = (PetscInt)beta; 758fbdc3dfeSToby Isaac PetscInt i; 759fbdc3dfeSToby Isaac 760fbdc3dfeSToby Isaac gr = n ? (1. / (n + n + alpha + beta + 1.)) : 1.; 761fbdc3dfeSToby Isaac if ((PetscReal)alphai == alpha) { 762fbdc3dfeSToby Isaac if (!n) { 763fbdc3dfeSToby Isaac for (i = 0; i < alphai; i++) gr *= (i + 1.) / (beta + i + 1.); 764fbdc3dfeSToby Isaac gr /= (alpha + beta + 1.); 765fbdc3dfeSToby Isaac } else { 766fbdc3dfeSToby Isaac for (i = 0; i < alphai; i++) gr *= (n + i + 1.) / (n + beta + i + 1.); 767fbdc3dfeSToby Isaac } 768fbdc3dfeSToby Isaac } else if ((PetscReal)betai == beta) { 769fbdc3dfeSToby Isaac if (!n) { 770fbdc3dfeSToby Isaac for (i = 0; i < betai; i++) gr *= (i + 1.) / (alpha + i + 2.); 771fbdc3dfeSToby Isaac gr /= (alpha + beta + 1.); 772fbdc3dfeSToby Isaac } else { 773fbdc3dfeSToby Isaac for (i = 0; i < betai; i++) gr *= (n + i + 1.) / (n + alpha + i + 1.); 774fbdc3dfeSToby Isaac } 775fbdc3dfeSToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 776fbdc3dfeSToby Isaac } 777fbdc3dfeSToby Isaac #endif 778fbdc3dfeSToby Isaac *norm = PetscSqrtReal(twoab1 * gr); 7793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 780fbdc3dfeSToby Isaac } 781fbdc3dfeSToby Isaac 782d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobiEval_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscInt k, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *p) 783d71ae5a4SJacob Faibussowitsch { 78494e21283SToby Isaac PetscReal ak, bk; 78594e21283SToby Isaac PetscReal abk1; 78694e21283SToby Isaac PetscInt i, l, maxdegree; 78794e21283SToby Isaac 78894e21283SToby Isaac PetscFunctionBegin; 78994e21283SToby Isaac maxdegree = degrees[ndegree - 1] - k; 79094e21283SToby Isaac ak = a + k; 79194e21283SToby Isaac bk = b + k; 79294e21283SToby Isaac abk1 = a + b + k + 1.; 79394e21283SToby Isaac if (maxdegree < 0) { 7949371c9d4SSatish Balay for (i = 0; i < npoints; i++) 7959371c9d4SSatish Balay for (l = 0; l < ndegree; l++) p[i * ndegree + l] = 0.; 7963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 79794e21283SToby Isaac } 79894e21283SToby Isaac for (i = 0; i < npoints; i++) { 79994e21283SToby Isaac PetscReal pm1, pm2, x; 80094e21283SToby Isaac PetscReal cnm1, cnm1x, cnm2; 80194e21283SToby Isaac PetscInt j, m; 80294e21283SToby Isaac 80394e21283SToby Isaac x = points[i]; 80494e21283SToby Isaac pm2 = 1.; 80594e21283SToby Isaac PetscDTJacobiRecurrence_Internal(1, ak, bk, cnm1, cnm1x, cnm2); 80694e21283SToby Isaac pm1 = (cnm1 + cnm1x * x); 80794e21283SToby Isaac l = 0; 808ad540459SPierre Jolivet while (l < ndegree && degrees[l] - k < 0) p[l++] = 0.; 80994e21283SToby Isaac while (l < ndegree && degrees[l] - k == 0) { 81094e21283SToby Isaac p[l] = pm2; 81194e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + m) * 0.5; 81294e21283SToby Isaac l++; 81394e21283SToby Isaac } 81494e21283SToby Isaac while (l < ndegree && degrees[l] - k == 1) { 81594e21283SToby Isaac p[l] = pm1; 81694e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + 1 + m) * 0.5; 81794e21283SToby Isaac l++; 81894e21283SToby Isaac } 81994e21283SToby Isaac for (j = 2; j <= maxdegree; j++) { 82094e21283SToby Isaac PetscReal pp; 82194e21283SToby Isaac 82294e21283SToby Isaac PetscDTJacobiRecurrence_Internal(j, ak, bk, cnm1, cnm1x, cnm2); 82394e21283SToby Isaac pp = (cnm1 + cnm1x * x) * pm1 - cnm2 * pm2; 82494e21283SToby Isaac pm2 = pm1; 82594e21283SToby Isaac pm1 = pp; 82694e21283SToby Isaac while (l < ndegree && degrees[l] - k == j) { 82794e21283SToby Isaac p[l] = pp; 82894e21283SToby Isaac for (m = 0; m < k; m++) p[l] *= (abk1 + j + m) * 0.5; 82994e21283SToby Isaac l++; 83094e21283SToby Isaac } 83194e21283SToby Isaac } 83294e21283SToby Isaac p += ndegree; 83394e21283SToby Isaac } 8343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 83594e21283SToby Isaac } 83694e21283SToby Isaac 83737045ce4SJed Brown /*@ 838dce8aebaSBarry Smith PetscDTJacobiEvalJet - Evaluate the jet (function and derivatives) of the Jacobi polynomials basis up to a given degree. 839fbdc3dfeSToby Isaac 8404165533cSJose E. Roman Input Parameters: 841fbdc3dfeSToby Isaac + alpha - the left exponent of the weight 842fbdc3dfeSToby Isaac . beta - the right exponetn of the weight 843fbdc3dfeSToby Isaac . npoints - the number of points to evaluate the polynomials at 844fbdc3dfeSToby Isaac . points - [npoints] array of point coordinates 845fbdc3dfeSToby Isaac . degree - the maximm degree polynomial space to evaluate, (degree + 1) will be evaluated total. 846fbdc3dfeSToby Isaac - k - the maximum derivative to evaluate in the jet, (k + 1) will be evaluated total. 847fbdc3dfeSToby Isaac 8482fe279fdSBarry Smith Output Parameter: 8492fe279fdSBarry Smith . p - an array containing the evaluations of the Jacobi polynomials's jets on the points. the size is (degree + 1) x 850fbdc3dfeSToby Isaac (k + 1) x npoints, which also describes the order of the dimensions of this three-dimensional array: the first 851fbdc3dfeSToby Isaac (slowest varying) dimension is polynomial degree; the second dimension is derivative order; the third (fastest 852fbdc3dfeSToby Isaac varying) dimension is the index of the evaluation point. 853fbdc3dfeSToby Isaac 854fbdc3dfeSToby Isaac Level: advanced 855fbdc3dfeSToby Isaac 856a4e35b19SJacob Faibussowitsch Notes: 857a4e35b19SJacob Faibussowitsch The Jacobi polynomials with indices $\alpha$ and $\beta$ are orthogonal with respect to the 858a4e35b19SJacob Faibussowitsch weighted inner product $\langle f, g \rangle = \int_{-1}^1 (1+x)^{\alpha} (1-x)^{\beta} f(x) 859a4e35b19SJacob Faibussowitsch g(x) dx$. 860a4e35b19SJacob Faibussowitsch 861db781477SPatrick Sanan .seealso: `PetscDTJacobiEval()`, `PetscDTPKDEvalJet()` 862fbdc3dfeSToby Isaac @*/ 863d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiEvalJet(PetscReal alpha, PetscReal beta, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt k, PetscReal p[]) 864d71ae5a4SJacob Faibussowitsch { 865fbdc3dfeSToby Isaac PetscInt i, j, l; 866fbdc3dfeSToby Isaac PetscInt *degrees; 867fbdc3dfeSToby Isaac PetscReal *psingle; 868fbdc3dfeSToby Isaac 869fbdc3dfeSToby Isaac PetscFunctionBegin; 870fbdc3dfeSToby Isaac if (degree == 0) { 871fbdc3dfeSToby Isaac PetscInt zero = 0; 872fbdc3dfeSToby Isaac 87348a46eb9SPierre Jolivet for (i = 0; i <= k; i++) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, i, points, 1, &zero, &p[i * npoints])); 8743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 875fbdc3dfeSToby Isaac } 8769566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(degree + 1, °rees)); 8779566063dSJacob Faibussowitsch PetscCall(PetscMalloc1((degree + 1) * npoints, &psingle)); 878fbdc3dfeSToby Isaac for (i = 0; i <= degree; i++) degrees[i] = i; 879fbdc3dfeSToby Isaac for (i = 0; i <= k; i++) { 8809566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, i, points, degree + 1, degrees, psingle)); 881fbdc3dfeSToby Isaac for (j = 0; j <= degree; j++) { 882ad540459SPierre Jolivet for (l = 0; l < npoints; l++) p[(j * (k + 1) + i) * npoints + l] = psingle[l * (degree + 1) + j]; 883fbdc3dfeSToby Isaac } 884fbdc3dfeSToby Isaac } 8859566063dSJacob Faibussowitsch PetscCall(PetscFree(psingle)); 8869566063dSJacob Faibussowitsch PetscCall(PetscFree(degrees)); 8873ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 888fbdc3dfeSToby Isaac } 889fbdc3dfeSToby Isaac 890fbdc3dfeSToby Isaac /*@ 891dce8aebaSBarry Smith PetscDTJacobiEval - evaluate Jacobi polynomials for the weight function $(1.+x)^{\alpha} (1.-x)^{\beta}$ at a set of points 89294e21283SToby Isaac at points 89394e21283SToby Isaac 89494e21283SToby Isaac Not Collective 89594e21283SToby Isaac 8964165533cSJose E. Roman Input Parameters: 89794e21283SToby Isaac + npoints - number of spatial points to evaluate at 89894e21283SToby Isaac . alpha - the left exponent > -1 89994e21283SToby Isaac . beta - the right exponent > -1 90094e21283SToby Isaac . points - array of locations to evaluate at 90194e21283SToby Isaac . ndegree - number of basis degrees to evaluate 90294e21283SToby Isaac - degrees - sorted array of degrees to evaluate 90394e21283SToby Isaac 9044165533cSJose E. Roman Output Parameters: 905*1d27aa22SBarry Smith + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension npoints*ndegrees, allocated by caller) (or `NULL`) 906*1d27aa22SBarry Smith . D - row-oriented derivative evaluation matrix (or `NULL`) 907*1d27aa22SBarry Smith - D2 - row-oriented second derivative evaluation matrix (or `NULL`) 90894e21283SToby Isaac 90994e21283SToby Isaac Level: intermediate 91094e21283SToby Isaac 911dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()`, `PetscDTLegendreEval()` 91294e21283SToby Isaac @*/ 913d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTJacobiEval(PetscInt npoints, PetscReal alpha, PetscReal beta, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *B, PetscReal *D, PetscReal *D2) 914d71ae5a4SJacob Faibussowitsch { 91594e21283SToby Isaac PetscFunctionBegin; 91608401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 91708401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 9183ba16761SJacob Faibussowitsch if (!npoints || !ndegree) PetscFunctionReturn(PETSC_SUCCESS); 9199566063dSJacob Faibussowitsch if (B) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 0, points, ndegree, degrees, B)); 9209566063dSJacob Faibussowitsch if (D) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 1, points, ndegree, degrees, D)); 9219566063dSJacob Faibussowitsch if (D2) PetscCall(PetscDTJacobiEval_Internal(npoints, alpha, beta, 2, points, ndegree, degrees, D2)); 9223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 92394e21283SToby Isaac } 92494e21283SToby Isaac 92594e21283SToby Isaac /*@ 92694e21283SToby Isaac PetscDTLegendreEval - evaluate Legendre polynomials at points 92737045ce4SJed Brown 92837045ce4SJed Brown Not Collective 92937045ce4SJed Brown 9304165533cSJose E. Roman Input Parameters: 93137045ce4SJed Brown + npoints - number of spatial points to evaluate at 93237045ce4SJed Brown . points - array of locations to evaluate at 93337045ce4SJed Brown . ndegree - number of basis degrees to evaluate 93437045ce4SJed Brown - degrees - sorted array of degrees to evaluate 93537045ce4SJed Brown 9364165533cSJose E. Roman Output Parameters: 937*1d27aa22SBarry Smith + B - row-oriented basis evaluation matrix B[point*ndegree + degree] (dimension `npoints`*`ndegrees`, allocated by caller) (or `NULL`) 938*1d27aa22SBarry Smith . D - row-oriented derivative evaluation matrix (or `NULL`) 939*1d27aa22SBarry Smith - D2 - row-oriented second derivative evaluation matrix (or `NULL`) 94037045ce4SJed Brown 94137045ce4SJed Brown Level: intermediate 94237045ce4SJed Brown 943db781477SPatrick Sanan .seealso: `PetscDTGaussQuadrature()` 94437045ce4SJed Brown @*/ 945d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTLegendreEval(PetscInt npoints, const PetscReal *points, PetscInt ndegree, const PetscInt *degrees, PetscReal *B, PetscReal *D, PetscReal *D2) 946d71ae5a4SJacob Faibussowitsch { 94737045ce4SJed Brown PetscFunctionBegin; 9489566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiEval(npoints, 0., 0., points, ndegree, degrees, B, D, D2)); 9493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 95037045ce4SJed Brown } 95137045ce4SJed Brown 952fbdc3dfeSToby Isaac /*@ 953*1d27aa22SBarry Smith 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, 954*1d27aa22SBarry Smith then the index of x is smaller than the index of y) 955fbdc3dfeSToby Isaac 956fbdc3dfeSToby Isaac Input Parameters: 957fbdc3dfeSToby Isaac + len - the desired length of the degree tuple 958fbdc3dfeSToby Isaac - index - the index to convert: should be >= 0 959fbdc3dfeSToby Isaac 960fbdc3dfeSToby Isaac Output Parameter: 961fbdc3dfeSToby Isaac . degtup - will be filled with a tuple of degrees 962fbdc3dfeSToby Isaac 963fbdc3dfeSToby Isaac Level: beginner 964fbdc3dfeSToby Isaac 965dce8aebaSBarry Smith Note: 966dce8aebaSBarry Smith For two tuples x and y with the same degree sum, partial degree sums over the final elements of the tuples 967fbdc3dfeSToby 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 968fbdc3dfeSToby Isaac last two elements is smaller for the former, so (2, 1, 1) < (1, 2, 1). 969fbdc3dfeSToby Isaac 970db781477SPatrick Sanan .seealso: `PetscDTGradedOrderToIndex()` 971fbdc3dfeSToby Isaac @*/ 972d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTIndexToGradedOrder(PetscInt len, PetscInt index, PetscInt degtup[]) 973d71ae5a4SJacob Faibussowitsch { 974fbdc3dfeSToby Isaac PetscInt i, total; 975fbdc3dfeSToby Isaac PetscInt sum; 976fbdc3dfeSToby Isaac 977fbdc3dfeSToby Isaac PetscFunctionBeginHot; 97808401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 97908401ef6SPierre Jolivet PetscCheck(index >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index must be non-negative"); 980fbdc3dfeSToby Isaac total = 1; 981fbdc3dfeSToby Isaac sum = 0; 982fbdc3dfeSToby Isaac while (index >= total) { 983fbdc3dfeSToby Isaac index -= total; 984fbdc3dfeSToby Isaac total = (total * (len + sum)) / (sum + 1); 985fbdc3dfeSToby Isaac sum++; 986fbdc3dfeSToby Isaac } 987fbdc3dfeSToby Isaac for (i = 0; i < len; i++) { 988fbdc3dfeSToby Isaac PetscInt c; 989fbdc3dfeSToby Isaac 990fbdc3dfeSToby Isaac degtup[i] = sum; 991fbdc3dfeSToby Isaac for (c = 0, total = 1; c < sum; c++) { 992fbdc3dfeSToby Isaac /* going into the loop, total is the number of way to have a tuple of sum exactly c with length len - 1 - i */ 993fbdc3dfeSToby Isaac if (index < total) break; 994fbdc3dfeSToby Isaac index -= total; 995fbdc3dfeSToby Isaac total = (total * (len - 1 - i + c)) / (c + 1); 996fbdc3dfeSToby Isaac degtup[i]--; 997fbdc3dfeSToby Isaac } 998fbdc3dfeSToby Isaac sum -= degtup[i]; 999fbdc3dfeSToby Isaac } 10003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1001fbdc3dfeSToby Isaac } 1002fbdc3dfeSToby Isaac 1003fbdc3dfeSToby Isaac /*@ 1004dce8aebaSBarry Smith PetscDTGradedOrderToIndex - convert a tuple into an index in a graded order, the inverse of `PetscDTIndexToGradedOrder()`. 1005fbdc3dfeSToby Isaac 1006fbdc3dfeSToby Isaac Input Parameters: 1007fbdc3dfeSToby Isaac + len - the length of the degree tuple 1008fbdc3dfeSToby Isaac - degtup - tuple with this length 1009fbdc3dfeSToby Isaac 1010fbdc3dfeSToby Isaac Output Parameter: 1011fbdc3dfeSToby Isaac . index - index in graded order: >= 0 1012fbdc3dfeSToby Isaac 101360225df5SJacob Faibussowitsch Level: beginner 1014fbdc3dfeSToby Isaac 1015dce8aebaSBarry Smith Note: 1016dce8aebaSBarry Smith For two tuples x and y with the same degree sum, partial degree sums over the final elements of the tuples 1017fbdc3dfeSToby 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 1018fbdc3dfeSToby Isaac last two elements is smaller for the former, so (2, 1, 1) < (1, 2, 1). 1019fbdc3dfeSToby Isaac 1020db781477SPatrick Sanan .seealso: `PetscDTIndexToGradedOrder()` 1021fbdc3dfeSToby Isaac @*/ 1022d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGradedOrderToIndex(PetscInt len, const PetscInt degtup[], PetscInt *index) 1023d71ae5a4SJacob Faibussowitsch { 1024fbdc3dfeSToby Isaac PetscInt i, idx, sum, total; 1025fbdc3dfeSToby Isaac 1026fbdc3dfeSToby Isaac PetscFunctionBeginHot; 102708401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 1028fbdc3dfeSToby Isaac for (i = 0, sum = 0; i < len; i++) sum += degtup[i]; 1029fbdc3dfeSToby Isaac idx = 0; 1030fbdc3dfeSToby Isaac total = 1; 1031fbdc3dfeSToby Isaac for (i = 0; i < sum; i++) { 1032fbdc3dfeSToby Isaac idx += total; 1033fbdc3dfeSToby Isaac total = (total * (len + i)) / (i + 1); 1034fbdc3dfeSToby Isaac } 1035fbdc3dfeSToby Isaac for (i = 0; i < len - 1; i++) { 1036fbdc3dfeSToby Isaac PetscInt c; 1037fbdc3dfeSToby Isaac 1038fbdc3dfeSToby Isaac total = 1; 1039fbdc3dfeSToby Isaac sum -= degtup[i]; 1040fbdc3dfeSToby Isaac for (c = 0; c < sum; c++) { 1041fbdc3dfeSToby Isaac idx += total; 1042fbdc3dfeSToby Isaac total = (total * (len - 1 - i + c)) / (c + 1); 1043fbdc3dfeSToby Isaac } 1044fbdc3dfeSToby Isaac } 1045fbdc3dfeSToby Isaac *index = idx; 10463ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1047fbdc3dfeSToby Isaac } 1048fbdc3dfeSToby Isaac 1049e3aa2e09SToby Isaac static PetscBool PKDCite = PETSC_FALSE; 1050e3aa2e09SToby Isaac const char PKDCitation[] = "@article{Kirby2010,\n" 1051e3aa2e09SToby Isaac " title={Singularity-free evaluation of collapsed-coordinate orthogonal polynomials},\n" 1052e3aa2e09SToby Isaac " author={Kirby, Robert C},\n" 1053e3aa2e09SToby Isaac " journal={ACM Transactions on Mathematical Software (TOMS)},\n" 1054e3aa2e09SToby Isaac " volume={37},\n" 1055e3aa2e09SToby Isaac " number={1},\n" 1056e3aa2e09SToby Isaac " pages={1--16},\n" 1057e3aa2e09SToby Isaac " year={2010},\n" 1058e3aa2e09SToby Isaac " publisher={ACM New York, NY, USA}\n}\n"; 1059e3aa2e09SToby Isaac 1060fbdc3dfeSToby Isaac /*@ 1061d8f25ad8SToby Isaac PetscDTPKDEvalJet - Evaluate the jet (function and derivatives) of the Proriol-Koornwinder-Dubiner (PKD) basis for 1062a4e35b19SJacob Faibussowitsch the space of polynomials up to a given degree. 1063fbdc3dfeSToby Isaac 10644165533cSJose E. Roman Input Parameters: 1065fbdc3dfeSToby Isaac + dim - the number of variables in the multivariate polynomials 1066fbdc3dfeSToby Isaac . npoints - the number of points to evaluate the polynomials at 1067fbdc3dfeSToby Isaac . points - [npoints x dim] array of point coordinates 1068fbdc3dfeSToby 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. 1069fbdc3dfeSToby Isaac - k - the maximum order partial derivative to evaluate in the jet. There are (dim + k choose dim) partial derivatives 1070fbdc3dfeSToby Isaac in the jet. Choosing k = 0 means to evaluate just the function and no derivatives 1071fbdc3dfeSToby Isaac 10722fe279fdSBarry Smith Output Parameter: 10732fe279fdSBarry Smith . p - an array containing the evaluations of the PKD polynomials' jets on the points. The size is ((dim + degree) 1074fbdc3dfeSToby Isaac choose dim) x ((dim + k) choose dim) x npoints, which also describes the order of the dimensions of this 1075fbdc3dfeSToby Isaac three-dimensional array: the first (slowest varying) dimension is basis function index; the second dimension is jet 1076fbdc3dfeSToby Isaac index; the third (fastest varying) dimension is the index of the evaluation point. 1077fbdc3dfeSToby Isaac 1078fbdc3dfeSToby Isaac Level: advanced 1079fbdc3dfeSToby Isaac 1080dce8aebaSBarry Smith Notes: 1081a4e35b19SJacob Faibussowitsch The PKD basis is L2-orthonormal on the biunit simplex (which is used as the reference element 1082a4e35b19SJacob Faibussowitsch for finite elements in PETSc), which makes it a stable basis to use for evaluating 1083a4e35b19SJacob Faibussowitsch polynomials in that domain. 1084a4e35b19SJacob Faibussowitsch 1085dce8aebaSBarry Smith The ordering of the basis functions, and the ordering of the derivatives in the jet, both follow the graded 1086dce8aebaSBarry Smith ordering of `PetscDTIndexToGradedOrder()` and `PetscDTGradedOrderToIndex()`. For example, in 3D, the polynomial with 1087dce8aebaSBarry 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); 1088fbdc3dfeSToby 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). 1089fbdc3dfeSToby Isaac 1090*1d27aa22SBarry Smith The implementation uses Kirby's singularity-free evaluation algorithm, <https://doi.org/10.1145/1644001.1644006>. 1091e3aa2e09SToby Isaac 1092db781477SPatrick Sanan .seealso: `PetscDTGradedOrderToIndex()`, `PetscDTIndexToGradedOrder()`, `PetscDTJacobiEvalJet()` 1093fbdc3dfeSToby Isaac @*/ 1094d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPKDEvalJet(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt k, PetscReal p[]) 1095d71ae5a4SJacob Faibussowitsch { 1096fbdc3dfeSToby Isaac PetscInt degidx, kidx, d, pt; 1097fbdc3dfeSToby Isaac PetscInt Nk, Ndeg; 1098fbdc3dfeSToby Isaac PetscInt *ktup, *degtup; 1099fbdc3dfeSToby Isaac PetscReal *scales, initscale, scaleexp; 1100fbdc3dfeSToby Isaac 1101fbdc3dfeSToby Isaac PetscFunctionBegin; 11029566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(PKDCitation, &PKDCite)); 11039566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + k, k, &Nk)); 11049566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + dim, degree, &Ndeg)); 11059566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(dim, °tup, dim, &ktup)); 11069566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Ndeg, &scales)); 1107fbdc3dfeSToby Isaac initscale = 1.; 1108fbdc3dfeSToby Isaac if (dim > 1) { 11099566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(dim, 2, &scaleexp)); 11102f613bf5SBarry Smith initscale = PetscPowReal(2., scaleexp * 0.5); 1111fbdc3dfeSToby Isaac } 1112fbdc3dfeSToby Isaac for (degidx = 0; degidx < Ndeg; degidx++) { 1113fbdc3dfeSToby Isaac PetscInt e, i; 1114fbdc3dfeSToby Isaac PetscInt m1idx = -1, m2idx = -1; 1115fbdc3dfeSToby Isaac PetscInt n; 1116fbdc3dfeSToby Isaac PetscInt degsum; 1117fbdc3dfeSToby Isaac PetscReal alpha; 1118fbdc3dfeSToby Isaac PetscReal cnm1, cnm1x, cnm2; 1119fbdc3dfeSToby Isaac PetscReal norm; 1120fbdc3dfeSToby Isaac 11219566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, degidx, degtup)); 11229371c9d4SSatish Balay for (d = dim - 1; d >= 0; d--) 11239371c9d4SSatish Balay if (degtup[d]) break; 1124fbdc3dfeSToby Isaac if (d < 0) { /* constant is 1 everywhere, all derivatives are zero */ 1125fbdc3dfeSToby Isaac scales[degidx] = initscale; 1126fbdc3dfeSToby Isaac for (e = 0; e < dim; e++) { 11279566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiNorm(e, 0., 0, &norm)); 1128fbdc3dfeSToby Isaac scales[degidx] /= norm; 1129fbdc3dfeSToby Isaac } 1130fbdc3dfeSToby Isaac for (i = 0; i < npoints; i++) p[degidx * Nk * npoints + i] = 1.; 1131fbdc3dfeSToby Isaac for (i = 0; i < (Nk - 1) * npoints; i++) p[(degidx * Nk + 1) * npoints + i] = 0.; 1132fbdc3dfeSToby Isaac continue; 1133fbdc3dfeSToby Isaac } 1134fbdc3dfeSToby Isaac n = degtup[d]; 1135fbdc3dfeSToby Isaac degtup[d]--; 11369566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, degtup, &m1idx)); 1137fbdc3dfeSToby Isaac if (degtup[d] > 0) { 1138fbdc3dfeSToby Isaac degtup[d]--; 11399566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, degtup, &m2idx)); 1140fbdc3dfeSToby Isaac degtup[d]++; 1141fbdc3dfeSToby Isaac } 1142fbdc3dfeSToby Isaac degtup[d]++; 1143fbdc3dfeSToby Isaac for (e = 0, degsum = 0; e < d; e++) degsum += degtup[e]; 1144fbdc3dfeSToby Isaac alpha = 2 * degsum + d; 1145fbdc3dfeSToby Isaac PetscDTJacobiRecurrence_Internal(n, alpha, 0., cnm1, cnm1x, cnm2); 1146fbdc3dfeSToby Isaac 1147fbdc3dfeSToby Isaac scales[degidx] = initscale; 1148fbdc3dfeSToby Isaac for (e = 0, degsum = 0; e < dim; e++) { 1149fbdc3dfeSToby Isaac PetscInt f; 1150fbdc3dfeSToby Isaac PetscReal ealpha; 1151fbdc3dfeSToby Isaac PetscReal enorm; 1152fbdc3dfeSToby Isaac 1153fbdc3dfeSToby Isaac ealpha = 2 * degsum + e; 1154fbdc3dfeSToby Isaac for (f = 0; f < degsum; f++) scales[degidx] *= 2.; 11559566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiNorm(ealpha, 0., degtup[e], &enorm)); 1156fbdc3dfeSToby Isaac scales[degidx] /= enorm; 1157fbdc3dfeSToby Isaac degsum += degtup[e]; 1158fbdc3dfeSToby Isaac } 1159fbdc3dfeSToby Isaac 1160fbdc3dfeSToby Isaac for (pt = 0; pt < npoints; pt++) { 1161fbdc3dfeSToby Isaac /* compute the multipliers */ 1162fbdc3dfeSToby Isaac PetscReal thetanm1, thetanm1x, thetanm2; 1163fbdc3dfeSToby Isaac 1164fbdc3dfeSToby Isaac thetanm1x = dim - (d + 1) + 2. * points[pt * dim + d]; 1165fbdc3dfeSToby Isaac for (e = d + 1; e < dim; e++) thetanm1x += points[pt * dim + e]; 1166fbdc3dfeSToby Isaac thetanm1x *= 0.5; 1167fbdc3dfeSToby Isaac thetanm1 = (2. - (dim - (d + 1))); 1168fbdc3dfeSToby Isaac for (e = d + 1; e < dim; e++) thetanm1 -= points[pt * dim + e]; 1169fbdc3dfeSToby Isaac thetanm1 *= 0.5; 1170fbdc3dfeSToby Isaac thetanm2 = thetanm1 * thetanm1; 1171fbdc3dfeSToby Isaac 1172fbdc3dfeSToby Isaac for (kidx = 0; kidx < Nk; kidx++) { 1173fbdc3dfeSToby Isaac PetscInt f; 1174fbdc3dfeSToby Isaac 11759566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, kidx, ktup)); 1176fbdc3dfeSToby Isaac /* first sum in the same derivative terms */ 1177fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] = (cnm1 * thetanm1 + cnm1x * thetanm1x) * p[(m1idx * Nk + kidx) * npoints + pt]; 1178ad540459SPierre Jolivet if (m2idx >= 0) p[(degidx * Nk + kidx) * npoints + pt] -= cnm2 * thetanm2 * p[(m2idx * Nk + kidx) * npoints + pt]; 1179fbdc3dfeSToby Isaac 1180fbdc3dfeSToby Isaac for (f = d; f < dim; f++) { 1181fbdc3dfeSToby Isaac PetscInt km1idx, mplty = ktup[f]; 1182fbdc3dfeSToby Isaac 1183fbdc3dfeSToby Isaac if (!mplty) continue; 1184fbdc3dfeSToby Isaac ktup[f]--; 11859566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, ktup, &km1idx)); 1186fbdc3dfeSToby Isaac 1187fbdc3dfeSToby Isaac /* the derivative of cnm1x * thetanm1x wrt x variable f is 0.5 * cnm1x if f > d otherwise it is cnm1x */ 1188fbdc3dfeSToby Isaac /* the derivative of cnm1 * thetanm1 wrt x variable f is 0 if f == d, otherwise it is -0.5 * cnm1 */ 1189fbdc3dfeSToby Isaac /* the derivative of -cnm2 * thetanm2 wrt x variable f is 0 if f == d, otherwise it is cnm2 * thetanm1 */ 1190fbdc3dfeSToby Isaac if (f > d) { 1191fbdc3dfeSToby Isaac PetscInt f2; 1192fbdc3dfeSToby Isaac 1193fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * 0.5 * (cnm1x - cnm1) * p[(m1idx * Nk + km1idx) * npoints + pt]; 1194fbdc3dfeSToby Isaac if (m2idx >= 0) { 1195fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * cnm2 * thetanm1 * p[(m2idx * Nk + km1idx) * npoints + pt]; 1196fbdc3dfeSToby Isaac /* second derivatives of -cnm2 * thetanm2 wrt x variable f,f2 is like - 0.5 * cnm2 */ 1197fbdc3dfeSToby Isaac for (f2 = f; f2 < dim; f2++) { 1198fbdc3dfeSToby Isaac PetscInt km2idx, mplty2 = ktup[f2]; 1199fbdc3dfeSToby Isaac PetscInt factor; 1200fbdc3dfeSToby Isaac 1201fbdc3dfeSToby Isaac if (!mplty2) continue; 1202fbdc3dfeSToby Isaac ktup[f2]--; 12039566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, ktup, &km2idx)); 1204fbdc3dfeSToby Isaac 1205fbdc3dfeSToby Isaac factor = mplty * mplty2; 1206fbdc3dfeSToby Isaac if (f == f2) factor /= 2; 1207fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] -= 0.5 * factor * cnm2 * p[(m2idx * Nk + km2idx) * npoints + pt]; 1208fbdc3dfeSToby Isaac ktup[f2]++; 1209fbdc3dfeSToby Isaac } 12103034baaeSToby Isaac } 1211fbdc3dfeSToby Isaac } else { 1212fbdc3dfeSToby Isaac p[(degidx * Nk + kidx) * npoints + pt] += mplty * cnm1x * p[(m1idx * Nk + km1idx) * npoints + pt]; 1213fbdc3dfeSToby Isaac } 1214fbdc3dfeSToby Isaac ktup[f]++; 1215fbdc3dfeSToby Isaac } 1216fbdc3dfeSToby Isaac } 1217fbdc3dfeSToby Isaac } 1218fbdc3dfeSToby Isaac } 1219fbdc3dfeSToby Isaac for (degidx = 0; degidx < Ndeg; degidx++) { 1220fbdc3dfeSToby Isaac PetscReal scale = scales[degidx]; 1221fbdc3dfeSToby Isaac PetscInt i; 1222fbdc3dfeSToby Isaac 1223fbdc3dfeSToby Isaac for (i = 0; i < Nk * npoints; i++) p[degidx * Nk * npoints + i] *= scale; 1224fbdc3dfeSToby Isaac } 12259566063dSJacob Faibussowitsch PetscCall(PetscFree(scales)); 12269566063dSJacob Faibussowitsch PetscCall(PetscFree2(degtup, ktup)); 12273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1228fbdc3dfeSToby Isaac } 1229fbdc3dfeSToby Isaac 1230d8f25ad8SToby Isaac /*@ 1231d8f25ad8SToby Isaac PetscDTPTrimmedSize - The size of the trimmed polynomial space of k-forms with a given degree and form degree, 1232dce8aebaSBarry Smith which can be evaluated in `PetscDTPTrimmedEvalJet()`. 1233d8f25ad8SToby Isaac 1234d8f25ad8SToby Isaac Input Parameters: 1235d8f25ad8SToby Isaac + dim - the number of variables in the multivariate polynomials 1236d8f25ad8SToby Isaac . degree - the degree (sum of degrees on the variables in a monomial) of the trimmed polynomial space. 1237d8f25ad8SToby Isaac - formDegree - the degree of the form 1238d8f25ad8SToby Isaac 12392fe279fdSBarry Smith Output Parameter: 124060225df5SJacob Faibussowitsch . size - The number ((`dim` + `degree`) choose (`dim` + `formDegree`)) x ((`degree` + `formDegree` - 1) choose (`formDegree`)) 1241d8f25ad8SToby Isaac 1242d8f25ad8SToby Isaac Level: advanced 1243d8f25ad8SToby Isaac 1244db781477SPatrick Sanan .seealso: `PetscDTPTrimmedEvalJet()` 1245d8f25ad8SToby Isaac @*/ 1246d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPTrimmedSize(PetscInt dim, PetscInt degree, PetscInt formDegree, PetscInt *size) 1247d71ae5a4SJacob Faibussowitsch { 1248d8f25ad8SToby Isaac PetscInt Nrk, Nbpt; // number of trimmed polynomials 1249d8f25ad8SToby Isaac 1250d8f25ad8SToby Isaac PetscFunctionBegin; 1251d8f25ad8SToby Isaac formDegree = PetscAbsInt(formDegree); 12529566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + dim, degree + formDegree, &Nbpt)); 12539566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(degree + formDegree - 1, formDegree, &Nrk)); 1254d8f25ad8SToby Isaac Nbpt *= Nrk; 1255d8f25ad8SToby Isaac *size = Nbpt; 12563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1257d8f25ad8SToby Isaac } 1258d8f25ad8SToby Isaac 1259d8f25ad8SToby Isaac /* there was a reference implementation based on section 4.4 of Arnold, Falk & Winther (acta numerica, 2006), but it 1260d8f25ad8SToby Isaac * was inferior to this implementation */ 1261d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPTrimmedEvalJet_Internal(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt formDegree, PetscInt jetDegree, PetscReal p[]) 1262d71ae5a4SJacob Faibussowitsch { 1263d8f25ad8SToby Isaac PetscInt formDegreeOrig = formDegree; 1264d8f25ad8SToby Isaac PetscBool formNegative = (formDegreeOrig < 0) ? PETSC_TRUE : PETSC_FALSE; 1265d8f25ad8SToby Isaac 1266d8f25ad8SToby Isaac PetscFunctionBegin; 1267d8f25ad8SToby Isaac formDegree = PetscAbsInt(formDegreeOrig); 1268d8f25ad8SToby Isaac if (formDegree == 0) { 12699566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree, jetDegree, p)); 12703ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1271d8f25ad8SToby Isaac } 1272d8f25ad8SToby Isaac if (formDegree == dim) { 12739566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree - 1, jetDegree, p)); 12743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1275d8f25ad8SToby Isaac } 1276d8f25ad8SToby Isaac PetscInt Nbpt; 12779566063dSJacob Faibussowitsch PetscCall(PetscDTPTrimmedSize(dim, degree, formDegree, &Nbpt)); 1278d8f25ad8SToby Isaac PetscInt Nf; 12799566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, formDegree, &Nf)); 1280d8f25ad8SToby Isaac PetscInt Nk; 12819566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + jetDegree, dim, &Nk)); 12829566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(p, Nbpt * Nf * Nk * npoints)); 1283d8f25ad8SToby Isaac 1284d8f25ad8SToby Isaac PetscInt Nbpm1; // number of scalar polynomials up to degree - 1; 12859566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim + degree - 1, dim, &Nbpm1)); 1286d8f25ad8SToby Isaac PetscReal *p_scalar; 12879566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nbpm1 * Nk * npoints, &p_scalar)); 12889566063dSJacob Faibussowitsch PetscCall(PetscDTPKDEvalJet(dim, npoints, points, degree - 1, jetDegree, p_scalar)); 1289d8f25ad8SToby Isaac PetscInt total = 0; 1290d8f25ad8SToby Isaac // First add the full polynomials up to degree - 1 into the basis: take the scalar 1291d8f25ad8SToby Isaac // and copy one for each form component 1292d8f25ad8SToby Isaac for (PetscInt i = 0; i < Nbpm1; i++) { 1293d8f25ad8SToby Isaac const PetscReal *src = &p_scalar[i * Nk * npoints]; 1294d8f25ad8SToby Isaac for (PetscInt f = 0; f < Nf; f++) { 1295d8f25ad8SToby Isaac PetscReal *dest = &p[(total++ * Nf + f) * Nk * npoints]; 12969566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(dest, src, Nk * npoints)); 1297d8f25ad8SToby Isaac } 1298d8f25ad8SToby Isaac } 1299d8f25ad8SToby Isaac PetscInt *form_atoms; 13009566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(formDegree + 1, &form_atoms)); 1301d8f25ad8SToby Isaac // construct the interior product pattern 1302d8f25ad8SToby Isaac PetscInt(*pattern)[3]; 1303d8f25ad8SToby Isaac PetscInt Nf1; // number of formDegree + 1 forms 13049566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, formDegree + 1, &Nf1)); 1305d8f25ad8SToby Isaac PetscInt nnz = Nf1 * (formDegree + 1); 13069566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nf1 * (formDegree + 1), &pattern)); 13079566063dSJacob Faibussowitsch PetscCall(PetscDTAltVInteriorPattern(dim, formDegree + 1, pattern)); 1308d8f25ad8SToby Isaac PetscReal centroid = (1. - dim) / (dim + 1.); 1309d8f25ad8SToby Isaac PetscInt *deriv; 13109566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim, &deriv)); 1311d8f25ad8SToby Isaac for (PetscInt d = dim; d >= formDegree + 1; d--) { 1312d8f25ad8SToby Isaac PetscInt Nfd1; // number of formDegree + 1 forms in dimension d that include dx_0 1313d8f25ad8SToby Isaac // (equal to the number of formDegree forms in dimension d-1) 13149566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(d - 1, formDegree, &Nfd1)); 1315d8f25ad8SToby Isaac // The number of homogeneous (degree-1) scalar polynomials in d variables 1316d8f25ad8SToby Isaac PetscInt Nh; 13179566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(d - 1 + degree - 1, d - 1, &Nh)); 1318d8f25ad8SToby Isaac const PetscReal *h_scalar = &p_scalar[(Nbpm1 - Nh) * Nk * npoints]; 1319d8f25ad8SToby Isaac for (PetscInt b = 0; b < Nh; b++) { 1320d8f25ad8SToby Isaac const PetscReal *h_s = &h_scalar[b * Nk * npoints]; 1321d8f25ad8SToby Isaac for (PetscInt f = 0; f < Nfd1; f++) { 1322d8f25ad8SToby Isaac // construct all formDegree+1 forms that start with dx_(dim - d) /\ ... 1323d8f25ad8SToby Isaac form_atoms[0] = dim - d; 13249566063dSJacob Faibussowitsch PetscCall(PetscDTEnumSubset(d - 1, formDegree, f, &form_atoms[1])); 1325ad540459SPierre Jolivet for (PetscInt i = 0; i < formDegree; i++) form_atoms[1 + i] += form_atoms[0] + 1; 1326d8f25ad8SToby Isaac PetscInt f_ind; // index of the resulting form 13279566063dSJacob Faibussowitsch PetscCall(PetscDTSubsetIndex(dim, formDegree + 1, form_atoms, &f_ind)); 1328d8f25ad8SToby Isaac PetscReal *p_f = &p[total++ * Nf * Nk * npoints]; 1329d8f25ad8SToby Isaac for (PetscInt nz = 0; nz < nnz; nz++) { 1330d8f25ad8SToby Isaac PetscInt i = pattern[nz][0]; // formDegree component 1331d8f25ad8SToby Isaac PetscInt j = pattern[nz][1]; // (formDegree + 1) component 1332d8f25ad8SToby Isaac PetscInt v = pattern[nz][2]; // coordinate component 1333d8f25ad8SToby Isaac PetscReal scale = v < 0 ? -1. : 1.; 1334d8f25ad8SToby Isaac 1335d8f25ad8SToby Isaac i = formNegative ? (Nf - 1 - i) : i; 1336d8f25ad8SToby Isaac scale = (formNegative && (i & 1)) ? -scale : scale; 1337d8f25ad8SToby Isaac v = v < 0 ? -(v + 1) : v; 1338ad540459SPierre Jolivet if (j != f_ind) continue; 1339d8f25ad8SToby Isaac PetscReal *p_i = &p_f[i * Nk * npoints]; 1340d8f25ad8SToby Isaac for (PetscInt jet = 0; jet < Nk; jet++) { 1341d8f25ad8SToby Isaac const PetscReal *h_jet = &h_s[jet * npoints]; 1342d8f25ad8SToby Isaac PetscReal *p_jet = &p_i[jet * npoints]; 1343d8f25ad8SToby Isaac 1344ad540459SPierre Jolivet for (PetscInt pt = 0; pt < npoints; pt++) p_jet[pt] += scale * h_jet[pt] * (points[pt * dim + v] - centroid); 13459566063dSJacob Faibussowitsch PetscCall(PetscDTIndexToGradedOrder(dim, jet, deriv)); 1346d8f25ad8SToby Isaac deriv[v]++; 1347d8f25ad8SToby Isaac PetscReal mult = deriv[v]; 1348d8f25ad8SToby Isaac PetscInt l; 13499566063dSJacob Faibussowitsch PetscCall(PetscDTGradedOrderToIndex(dim, deriv, &l)); 1350ad540459SPierre Jolivet if (l >= Nk) continue; 1351d8f25ad8SToby Isaac p_jet = &p_i[l * npoints]; 1352ad540459SPierre Jolivet for (PetscInt pt = 0; pt < npoints; pt++) p_jet[pt] += scale * mult * h_jet[pt]; 1353d8f25ad8SToby Isaac deriv[v]--; 1354d8f25ad8SToby Isaac } 1355d8f25ad8SToby Isaac } 1356d8f25ad8SToby Isaac } 1357d8f25ad8SToby Isaac } 1358d8f25ad8SToby Isaac } 135908401ef6SPierre Jolivet PetscCheck(total == Nbpt, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Incorrectly counted P trimmed polynomials"); 13609566063dSJacob Faibussowitsch PetscCall(PetscFree(deriv)); 13619566063dSJacob Faibussowitsch PetscCall(PetscFree(pattern)); 13629566063dSJacob Faibussowitsch PetscCall(PetscFree(form_atoms)); 13639566063dSJacob Faibussowitsch PetscCall(PetscFree(p_scalar)); 13643ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1365d8f25ad8SToby Isaac } 1366d8f25ad8SToby Isaac 1367d8f25ad8SToby Isaac /*@ 1368d8f25ad8SToby Isaac PetscDTPTrimmedEvalJet - Evaluate the jet (function and derivatives) of a basis of the trimmed polynomial k-forms up to 1369d8f25ad8SToby Isaac a given degree. 1370d8f25ad8SToby Isaac 1371d8f25ad8SToby Isaac Input Parameters: 1372d8f25ad8SToby Isaac + dim - the number of variables in the multivariate polynomials 1373d8f25ad8SToby Isaac . npoints - the number of points to evaluate the polynomials at 1374d8f25ad8SToby Isaac . points - [npoints x dim] array of point coordinates 1375d8f25ad8SToby Isaac . degree - the degree (sum of degrees on the variables in a monomial) of the trimmed polynomial space to evaluate. 1376d8f25ad8SToby Isaac There are ((dim + degree) choose (dim + formDegree)) x ((degree + formDegree - 1) choose (formDegree)) polynomials in this space. 1377dce8aebaSBarry Smith (You can use `PetscDTPTrimmedSize()` to compute this size.) 1378d8f25ad8SToby Isaac . formDegree - the degree of the form 1379d8f25ad8SToby Isaac - jetDegree - the maximum order partial derivative to evaluate in the jet. There are ((dim + jetDegree) choose dim) partial derivatives 1380d8f25ad8SToby Isaac in the jet. Choosing jetDegree = 0 means to evaluate just the function and no derivatives 1381d8f25ad8SToby Isaac 138220f4b53cSBarry Smith Output Parameter: 1383a4e35b19SJacob Faibussowitsch . p - an array containing the evaluations of the PKD polynomials' jets on the points. 138460225df5SJacob Faibussowitsch 1385a4e35b19SJacob Faibussowitsch Level: advanced 1386a4e35b19SJacob Faibussowitsch 1387a4e35b19SJacob Faibussowitsch Notes: 1388a4e35b19SJacob Faibussowitsch The size of `p` is `PetscDTPTrimmedSize()` x ((dim + formDegree) choose dim) x ((dim + k) 1389a4e35b19SJacob Faibussowitsch choose dim) x npoints,which also describes the order of the dimensions of this 1390a4e35b19SJacob Faibussowitsch four-dimensional array\: 1391a4e35b19SJacob Faibussowitsch 1392d8f25ad8SToby Isaac the first (slowest varying) dimension is basis function index; 1393d8f25ad8SToby Isaac the second dimension is component of the form; 1394d8f25ad8SToby Isaac the third dimension is jet index; 1395d8f25ad8SToby Isaac the fourth (fastest varying) dimension is the index of the evaluation point. 1396d8f25ad8SToby Isaac 1397dce8aebaSBarry Smith The ordering of the basis functions is not graded, so the basis functions are not nested by degree like `PetscDTPKDEvalJet()`. 1398d8f25ad8SToby Isaac The basis functions are not an L2-orthonormal basis on any particular domain. 1399d8f25ad8SToby Isaac 1400d8f25ad8SToby Isaac The implementation is based on the description of the trimmed polynomials up to degree r as 1401d8f25ad8SToby Isaac the direct sum of polynomials up to degree (r-1) and the Koszul differential applied to 1402d8f25ad8SToby Isaac homogeneous polynomials of degree (r-1). 1403d8f25ad8SToby Isaac 1404db781477SPatrick Sanan .seealso: `PetscDTPKDEvalJet()`, `PetscDTPTrimmedSize()` 1405d8f25ad8SToby Isaac @*/ 1406d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTPTrimmedEvalJet(PetscInt dim, PetscInt npoints, const PetscReal points[], PetscInt degree, PetscInt formDegree, PetscInt jetDegree, PetscReal p[]) 1407d71ae5a4SJacob Faibussowitsch { 1408d8f25ad8SToby Isaac PetscFunctionBegin; 14099566063dSJacob Faibussowitsch PetscCall(PetscDTPTrimmedEvalJet_Internal(dim, npoints, points, degree, formDegree, jetDegree, p)); 14103ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1411d8f25ad8SToby Isaac } 1412d8f25ad8SToby Isaac 1413e6a796c3SToby Isaac /* solve the symmetric tridiagonal eigenvalue system, writing the eigenvalues into eigs and the eigenvectors into V 1414e6a796c3SToby Isaac * with lds n; diag and subdiag are overwritten */ 1415d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTSymmetricTridiagonalEigensolve(PetscInt n, PetscReal diag[], PetscReal subdiag[], PetscReal eigs[], PetscScalar V[]) 1416d71ae5a4SJacob Faibussowitsch { 1417e6a796c3SToby Isaac char jobz = 'V'; /* eigenvalues and eigenvectors */ 1418e6a796c3SToby Isaac char range = 'A'; /* all eigenvalues will be found */ 1419e6a796c3SToby Isaac PetscReal VL = 0.; /* ignored because range is 'A' */ 1420e6a796c3SToby Isaac PetscReal VU = 0.; /* ignored because range is 'A' */ 1421e6a796c3SToby Isaac PetscBLASInt IL = 0; /* ignored because range is 'A' */ 1422e6a796c3SToby Isaac PetscBLASInt IU = 0; /* ignored because range is 'A' */ 1423e6a796c3SToby Isaac PetscReal abstol = 0.; /* unused */ 1424e6a796c3SToby Isaac PetscBLASInt bn, bm, ldz; /* bm will equal bn on exit */ 1425e6a796c3SToby Isaac PetscBLASInt *isuppz; 1426e6a796c3SToby Isaac PetscBLASInt lwork, liwork; 1427e6a796c3SToby Isaac PetscReal workquery; 1428e6a796c3SToby Isaac PetscBLASInt iworkquery; 1429e6a796c3SToby Isaac PetscBLASInt *iwork; 1430e6a796c3SToby Isaac PetscBLASInt info; 1431e6a796c3SToby Isaac PetscReal *work = NULL; 1432e6a796c3SToby Isaac 1433e6a796c3SToby Isaac PetscFunctionBegin; 1434e6a796c3SToby Isaac #if !defined(PETSCDTGAUSSIANQUADRATURE_EIG) 1435e6a796c3SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "A LAPACK symmetric tridiagonal eigensolver could not be found"); 1436e6a796c3SToby Isaac #endif 14379566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &bn)); 14389566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &ldz)); 1439e6a796c3SToby Isaac #if !defined(PETSC_MISSING_LAPACK_STEGR) 14409566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(2 * n, &isuppz)); 1441e6a796c3SToby Isaac lwork = -1; 1442e6a796c3SToby Isaac liwork = -1; 1443792fecdfSBarry Smith PetscCallBLAS("LAPACKstegr", LAPACKstegr_(&jobz, &range, &bn, diag, subdiag, &VL, &VU, &IL, &IU, &abstol, &bm, eigs, V, &ldz, isuppz, &workquery, &lwork, &iworkquery, &liwork, &info)); 144428b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEGR error"); 1445e6a796c3SToby Isaac lwork = (PetscBLASInt)workquery; 1446e6a796c3SToby Isaac liwork = (PetscBLASInt)iworkquery; 14479566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(lwork, &work, liwork, &iwork)); 14489566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 1449792fecdfSBarry Smith PetscCallBLAS("LAPACKstegr", LAPACKstegr_(&jobz, &range, &bn, diag, subdiag, &VL, &VU, &IL, &IU, &abstol, &bm, eigs, V, &ldz, isuppz, work, &lwork, iwork, &liwork, &info)); 14509566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 145128b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEGR error"); 14529566063dSJacob Faibussowitsch PetscCall(PetscFree2(work, iwork)); 14539566063dSJacob Faibussowitsch PetscCall(PetscFree(isuppz)); 1454e6a796c3SToby Isaac #elif !defined(PETSC_MISSING_LAPACK_STEQR) 1455e6a796c3SToby Isaac jobz = 'I'; /* Compute eigenvalues and eigenvectors of the 1456e6a796c3SToby Isaac tridiagonal matrix. Z is initialized to the identity 1457e6a796c3SToby Isaac matrix. */ 14589566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(PetscMax(1, 2 * n - 2), &work)); 1459792fecdfSBarry Smith PetscCallBLAS("LAPACKsteqr", LAPACKsteqr_("I", &bn, diag, subdiag, V, &ldz, work, &info)); 14609566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 146128b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_PLIB, "xSTEQR error"); 14629566063dSJacob Faibussowitsch PetscCall(PetscFree(work)); 14639566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(eigs, diag, n)); 1464e6a796c3SToby Isaac #endif 14653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1466e6a796c3SToby Isaac } 1467e6a796c3SToby Isaac 1468e6a796c3SToby Isaac /* Formula for the weights at the endpoints (-1 and 1) of Gauss-Lobatto-Jacobi 1469e6a796c3SToby Isaac * quadrature rules on the interval [-1, 1] */ 1470d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussLobattoJacobiEndweights_Internal(PetscInt n, PetscReal alpha, PetscReal beta, PetscReal *leftw, PetscReal *rightw) 1471d71ae5a4SJacob Faibussowitsch { 1472e6a796c3SToby Isaac PetscReal twoab1; 1473e6a796c3SToby Isaac PetscInt m = n - 2; 1474e6a796c3SToby Isaac PetscReal a = alpha + 1.; 1475e6a796c3SToby Isaac PetscReal b = beta + 1.; 1476e6a796c3SToby Isaac PetscReal gra, grb; 1477e6a796c3SToby Isaac 1478e6a796c3SToby Isaac PetscFunctionBegin; 1479e6a796c3SToby Isaac twoab1 = PetscPowReal(2., a + b - 1.); 1480e6a796c3SToby Isaac #if defined(PETSC_HAVE_LGAMMA) 14819371c9d4SSatish Balay grb = PetscExpReal(2. * PetscLGamma(b + 1.) + PetscLGamma(m + 1.) + PetscLGamma(m + a + 1.) - (PetscLGamma(m + b + 1) + PetscLGamma(m + a + b + 1.))); 14829371c9d4SSatish Balay gra = PetscExpReal(2. * PetscLGamma(a + 1.) + PetscLGamma(m + 1.) + PetscLGamma(m + b + 1.) - (PetscLGamma(m + a + 1) + PetscLGamma(m + a + b + 1.))); 1483e6a796c3SToby Isaac #else 1484e6a796c3SToby Isaac { 1485e6a796c3SToby Isaac PetscInt alphai = (PetscInt)alpha; 1486e6a796c3SToby Isaac PetscInt betai = (PetscInt)beta; 1487e6a796c3SToby Isaac 1488e6a796c3SToby Isaac if ((PetscReal)alphai == alpha && (PetscReal)betai == beta) { 1489e6a796c3SToby Isaac PetscReal binom1, binom2; 1490e6a796c3SToby Isaac 14919566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + b, b, &binom1)); 14929566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a + b, b, &binom2)); 1493e6a796c3SToby Isaac grb = 1. / (binom1 * binom2); 14949566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a, a, &binom1)); 14959566063dSJacob Faibussowitsch PetscCall(PetscDTBinomial(m + a + b, a, &binom2)); 1496e6a796c3SToby Isaac gra = 1. / (binom1 * binom2); 1497e6a796c3SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 1498e6a796c3SToby Isaac } 1499e6a796c3SToby Isaac #endif 1500e6a796c3SToby Isaac *leftw = twoab1 * grb / b; 1501e6a796c3SToby Isaac *rightw = twoab1 * gra / a; 15023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1503e6a796c3SToby Isaac } 1504e6a796c3SToby Isaac 1505e6a796c3SToby Isaac /* Evaluates the nth jacobi polynomial with weight parameters a,b at a point x. 1506e6a796c3SToby Isaac Recurrence relations implemented from the pseudocode given in Karniadakis and Sherwin, Appendix B */ 1507d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscDTComputeJacobi(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscReal *P) 1508d71ae5a4SJacob Faibussowitsch { 150994e21283SToby Isaac PetscReal pn1, pn2; 151094e21283SToby Isaac PetscReal cnm1, cnm1x, cnm2; 1511e6a796c3SToby Isaac PetscInt k; 1512e6a796c3SToby Isaac 1513e6a796c3SToby Isaac PetscFunctionBegin; 15149371c9d4SSatish Balay if (!n) { 15159371c9d4SSatish Balay *P = 1.0; 15163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 15179371c9d4SSatish Balay } 151894e21283SToby Isaac PetscDTJacobiRecurrence_Internal(1, a, b, cnm1, cnm1x, cnm2); 151994e21283SToby Isaac pn2 = 1.; 152094e21283SToby Isaac pn1 = cnm1 + cnm1x * x; 15219371c9d4SSatish Balay if (n == 1) { 15229371c9d4SSatish Balay *P = pn1; 15233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 15249371c9d4SSatish Balay } 1525e6a796c3SToby Isaac *P = 0.0; 1526e6a796c3SToby Isaac for (k = 2; k < n + 1; ++k) { 152794e21283SToby Isaac PetscDTJacobiRecurrence_Internal(k, a, b, cnm1, cnm1x, cnm2); 1528e6a796c3SToby Isaac 152994e21283SToby Isaac *P = (cnm1 + cnm1x * x) * pn1 - cnm2 * pn2; 1530e6a796c3SToby Isaac pn2 = pn1; 1531e6a796c3SToby Isaac pn1 = *P; 1532e6a796c3SToby Isaac } 15333ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1534e6a796c3SToby Isaac } 1535e6a796c3SToby Isaac 1536e6a796c3SToby Isaac /* Evaluates the first derivative of P_{n}^{a,b} at a point x. */ 1537d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscDTComputeJacobiDerivative(PetscReal a, PetscReal b, PetscInt n, PetscReal x, PetscInt k, PetscReal *P) 1538d71ae5a4SJacob Faibussowitsch { 1539e6a796c3SToby Isaac PetscReal nP; 1540e6a796c3SToby Isaac PetscInt i; 1541e6a796c3SToby Isaac 1542e6a796c3SToby Isaac PetscFunctionBegin; 154317a42bb7SSatish Balay *P = 0.0; 15443ba16761SJacob Faibussowitsch if (k > n) PetscFunctionReturn(PETSC_SUCCESS); 15459566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(a + k, b + k, n - k, x, &nP)); 1546e6a796c3SToby Isaac for (i = 0; i < k; i++) nP *= (a + b + n + 1. + i) * 0.5; 1547e6a796c3SToby Isaac *P = nP; 15483ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1549e6a796c3SToby Isaac } 1550e6a796c3SToby Isaac 1551d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_Newton_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal x[], PetscReal w[]) 1552d71ae5a4SJacob Faibussowitsch { 1553e6a796c3SToby Isaac PetscInt maxIter = 100; 155494e21283SToby Isaac PetscReal eps = PetscExpReal(0.75 * PetscLogReal(PETSC_MACHINE_EPSILON)); 1555200b5abcSJed Brown PetscReal a1, a6, gf; 1556e6a796c3SToby Isaac PetscInt k; 1557e6a796c3SToby Isaac 1558e6a796c3SToby Isaac PetscFunctionBegin; 1559e6a796c3SToby Isaac 1560e6a796c3SToby Isaac a1 = PetscPowReal(2.0, a + b + 1); 156194e21283SToby Isaac #if defined(PETSC_HAVE_LGAMMA) 1562200b5abcSJed Brown { 1563200b5abcSJed Brown PetscReal a2, a3, a4, a5; 156494e21283SToby Isaac a2 = PetscLGamma(a + npoints + 1); 156594e21283SToby Isaac a3 = PetscLGamma(b + npoints + 1); 156694e21283SToby Isaac a4 = PetscLGamma(a + b + npoints + 1); 156794e21283SToby Isaac a5 = PetscLGamma(npoints + 1); 156894e21283SToby Isaac gf = PetscExpReal(a2 + a3 - (a4 + a5)); 1569200b5abcSJed Brown } 1570e6a796c3SToby Isaac #else 1571e6a796c3SToby Isaac { 1572e6a796c3SToby Isaac PetscInt ia, ib; 1573e6a796c3SToby Isaac 1574e6a796c3SToby Isaac ia = (PetscInt)a; 1575e6a796c3SToby Isaac ib = (PetscInt)b; 157694e21283SToby Isaac gf = 1.; 157794e21283SToby Isaac if (ia == a && ia >= 0) { /* compute ratio of rising factorals wrt a */ 157894e21283SToby Isaac for (k = 0; k < ia; k++) gf *= (npoints + 1. + k) / (npoints + b + 1. + k); 157994e21283SToby Isaac } else if (b == b && ib >= 0) { /* compute ratio of rising factorials wrt b */ 158094e21283SToby Isaac for (k = 0; k < ib; k++) gf *= (npoints + 1. + k) / (npoints + a + 1. + k); 158194e21283SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "lgamma() - math routine is unavailable."); 1582e6a796c3SToby Isaac } 1583e6a796c3SToby Isaac #endif 1584e6a796c3SToby Isaac 158594e21283SToby Isaac a6 = a1 * gf; 1586e6a796c3SToby Isaac /* Computes the m roots of P_{m}^{a,b} on [-1,1] by Newton's method with Chebyshev points as initial guesses. 1587e6a796c3SToby Isaac Algorithm implemented from the pseudocode given by Karniadakis and Sherwin and Python in FIAT */ 1588e6a796c3SToby Isaac for (k = 0; k < npoints; ++k) { 158994e21283SToby Isaac PetscReal r = PetscCosReal(PETSC_PI * (1. - (4. * k + 3. + 2. * b) / (4. * npoints + 2. * (a + b + 1.)))), dP; 1590e6a796c3SToby Isaac PetscInt j; 1591e6a796c3SToby Isaac 1592e6a796c3SToby Isaac if (k > 0) r = 0.5 * (r + x[k - 1]); 1593e6a796c3SToby Isaac for (j = 0; j < maxIter; ++j) { 1594e6a796c3SToby Isaac PetscReal s = 0.0, delta, f, fp; 1595e6a796c3SToby Isaac PetscInt i; 1596e6a796c3SToby Isaac 1597e6a796c3SToby Isaac for (i = 0; i < k; ++i) s = s + 1.0 / (r - x[i]); 15989566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(a, b, npoints, r, &f)); 15999566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobiDerivative(a, b, npoints, r, 1, &fp)); 1600e6a796c3SToby Isaac delta = f / (fp - f * s); 1601e6a796c3SToby Isaac r = r - delta; 1602e6a796c3SToby Isaac if (PetscAbsReal(delta) < eps) break; 1603e6a796c3SToby Isaac } 1604e6a796c3SToby Isaac x[k] = r; 16059566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobiDerivative(a, b, npoints, x[k], 1, &dP)); 1606e6a796c3SToby Isaac w[k] = a6 / (1.0 - PetscSqr(x[k])) / PetscSqr(dP); 1607e6a796c3SToby Isaac } 16083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1609e6a796c3SToby Isaac } 1610e6a796c3SToby Isaac 161194e21283SToby Isaac /* Compute the diagonals of the Jacobi matrix used in Golub & Welsch algorithms for Gauss-Jacobi 1612e6a796c3SToby Isaac * quadrature weight calculations on [-1,1] for exponents (1. + x)^a (1.-x)^b */ 1613d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTJacobiMatrix_Internal(PetscInt nPoints, PetscReal a, PetscReal b, PetscReal *d, PetscReal *s) 1614d71ae5a4SJacob Faibussowitsch { 1615e6a796c3SToby Isaac PetscInt i; 1616e6a796c3SToby Isaac 1617e6a796c3SToby Isaac PetscFunctionBegin; 1618e6a796c3SToby Isaac for (i = 0; i < nPoints; i++) { 161994e21283SToby Isaac PetscReal A, B, C; 1620e6a796c3SToby Isaac 162194e21283SToby Isaac PetscDTJacobiRecurrence_Internal(i + 1, a, b, A, B, C); 162294e21283SToby Isaac d[i] = -A / B; 162394e21283SToby Isaac if (i) s[i - 1] *= C / B; 162494e21283SToby Isaac if (i < nPoints - 1) s[i] = 1. / B; 1625e6a796c3SToby Isaac } 16263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1627e6a796c3SToby Isaac } 1628e6a796c3SToby Isaac 1629d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_GolubWelsch_Internal(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 1630d71ae5a4SJacob Faibussowitsch { 1631e6a796c3SToby Isaac PetscReal mu0; 1632e6a796c3SToby Isaac PetscReal ga, gb, gab; 1633e6a796c3SToby Isaac PetscInt i; 1634e6a796c3SToby Isaac 1635e6a796c3SToby Isaac PetscFunctionBegin; 16369566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(GolubWelschCitation, &GolubWelschCite)); 1637e6a796c3SToby Isaac 1638e6a796c3SToby Isaac #if defined(PETSC_HAVE_TGAMMA) 1639e6a796c3SToby Isaac ga = PetscTGamma(a + 1); 1640e6a796c3SToby Isaac gb = PetscTGamma(b + 1); 1641e6a796c3SToby Isaac gab = PetscTGamma(a + b + 2); 1642e6a796c3SToby Isaac #else 1643e6a796c3SToby Isaac { 1644e6a796c3SToby Isaac PetscInt ia, ib; 1645e6a796c3SToby Isaac 1646e6a796c3SToby Isaac ia = (PetscInt)a; 1647e6a796c3SToby Isaac ib = (PetscInt)b; 1648e6a796c3SToby Isaac if (ia == a && ib == b && ia + 1 > 0 && ib + 1 > 0 && ia + ib + 2 > 0) { /* All gamma(x) terms are (x-1)! terms */ 16499566063dSJacob Faibussowitsch PetscCall(PetscDTFactorial(ia, &ga)); 16509566063dSJacob Faibussowitsch PetscCall(PetscDTFactorial(ib, &gb)); 16516205a86aSPierre Jolivet PetscCall(PetscDTFactorial(ia + ib + 1, &gab)); 1652e6a796c3SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "tgamma() - math routine is unavailable."); 1653e6a796c3SToby Isaac } 1654e6a796c3SToby Isaac #endif 1655e6a796c3SToby Isaac mu0 = PetscPowReal(2., a + b + 1.) * ga * gb / gab; 1656e6a796c3SToby Isaac 1657e6a796c3SToby Isaac #if defined(PETSCDTGAUSSIANQUADRATURE_EIG) 1658e6a796c3SToby Isaac { 1659e6a796c3SToby Isaac PetscReal *diag, *subdiag; 1660e6a796c3SToby Isaac PetscScalar *V; 1661e6a796c3SToby Isaac 16629566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &diag, npoints, &subdiag)); 16639566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints * npoints, &V)); 16649566063dSJacob Faibussowitsch PetscCall(PetscDTJacobiMatrix_Internal(npoints, a, b, diag, subdiag)); 1665e6a796c3SToby Isaac for (i = 0; i < npoints - 1; i++) subdiag[i] = PetscSqrtReal(subdiag[i]); 16669566063dSJacob Faibussowitsch PetscCall(PetscDTSymmetricTridiagonalEigensolve(npoints, diag, subdiag, x, V)); 166794e21283SToby Isaac for (i = 0; i < npoints; i++) w[i] = PetscSqr(PetscRealPart(V[i * npoints])) * mu0; 16689566063dSJacob Faibussowitsch PetscCall(PetscFree(V)); 16699566063dSJacob Faibussowitsch PetscCall(PetscFree2(diag, subdiag)); 1670e6a796c3SToby Isaac } 1671e6a796c3SToby Isaac #else 1672e6a796c3SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "A LAPACK symmetric tridiagonal eigensolver could not be found"); 1673e6a796c3SToby Isaac #endif 167494e21283SToby Isaac { /* As of March 2, 2020, The Sun Performance Library breaks the LAPACK contract for xstegr and xsteqr: the 167594e21283SToby Isaac eigenvalues are not guaranteed to be in ascending order. So we heave a passive aggressive sigh and check that 167694e21283SToby Isaac the eigenvalues are sorted */ 167794e21283SToby Isaac PetscBool sorted; 167894e21283SToby Isaac 16799566063dSJacob Faibussowitsch PetscCall(PetscSortedReal(npoints, x, &sorted)); 168094e21283SToby Isaac if (!sorted) { 168194e21283SToby Isaac PetscInt *order, i; 168294e21283SToby Isaac PetscReal *tmp; 168394e21283SToby Isaac 16849566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &order, npoints, &tmp)); 168594e21283SToby Isaac for (i = 0; i < npoints; i++) order[i] = i; 16869566063dSJacob Faibussowitsch PetscCall(PetscSortRealWithPermutation(npoints, x, order)); 16879566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, x, npoints)); 168894e21283SToby Isaac for (i = 0; i < npoints; i++) x[i] = tmp[order[i]]; 16899566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, w, npoints)); 169094e21283SToby Isaac for (i = 0; i < npoints; i++) w[i] = tmp[order[i]]; 16919566063dSJacob Faibussowitsch PetscCall(PetscFree2(order, tmp)); 169294e21283SToby Isaac } 169394e21283SToby Isaac } 16943ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1695e6a796c3SToby Isaac } 1696e6a796c3SToby Isaac 1697d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussJacobiQuadrature_Internal(PetscInt npoints, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[], PetscBool newton) 1698d71ae5a4SJacob Faibussowitsch { 1699e6a796c3SToby Isaac PetscFunctionBegin; 170008401ef6SPierre Jolivet PetscCheck(npoints >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be positive"); 1701e6a796c3SToby Isaac /* If asking for a 1D Lobatto point, just return the non-Lobatto 1D point */ 170208401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 170308401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 1704e6a796c3SToby Isaac 17051baa6e33SBarry Smith if (newton) PetscCall(PetscDTGaussJacobiQuadrature_Newton_Internal(npoints, alpha, beta, x, w)); 17061baa6e33SBarry Smith else PetscCall(PetscDTGaussJacobiQuadrature_GolubWelsch_Internal(npoints, alpha, beta, x, w)); 1707e6a796c3SToby Isaac if (alpha == beta) { /* symmetrize */ 1708e6a796c3SToby Isaac PetscInt i; 1709e6a796c3SToby Isaac for (i = 0; i < (npoints + 1) / 2; i++) { 1710e6a796c3SToby Isaac PetscInt j = npoints - 1 - i; 1711e6a796c3SToby Isaac PetscReal xi = x[i]; 1712e6a796c3SToby Isaac PetscReal xj = x[j]; 1713e6a796c3SToby Isaac PetscReal wi = w[i]; 1714e6a796c3SToby Isaac PetscReal wj = w[j]; 1715e6a796c3SToby Isaac 1716e6a796c3SToby Isaac x[i] = (xi - xj) / 2.; 1717e6a796c3SToby Isaac x[j] = (xj - xi) / 2.; 1718e6a796c3SToby Isaac w[i] = w[j] = (wi + wj) / 2.; 1719e6a796c3SToby Isaac } 1720e6a796c3SToby Isaac } 17213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1722e6a796c3SToby Isaac } 1723e6a796c3SToby Isaac 172494e21283SToby Isaac /*@ 1725*1d27aa22SBarry Smith PetscDTGaussJacobiQuadrature - quadrature for the interval $[a, b]$ with the weight function 172694e21283SToby Isaac $(x-a)^\alpha (x-b)^\beta$. 172794e21283SToby Isaac 172820f4b53cSBarry Smith Not Collective 172994e21283SToby Isaac 173094e21283SToby Isaac Input Parameters: 173194e21283SToby Isaac + npoints - the number of points in the quadrature rule 173294e21283SToby Isaac . a - the left endpoint of the interval 173394e21283SToby Isaac . b - the right endpoint of the interval 173494e21283SToby Isaac . alpha - the left exponent 173594e21283SToby Isaac - beta - the right exponent 173694e21283SToby Isaac 173794e21283SToby Isaac Output Parameters: 173820f4b53cSBarry Smith + x - array of length `npoints`, the locations of the quadrature points 173920f4b53cSBarry Smith - w - array of length `npoints`, the weights of the quadrature points 174094e21283SToby Isaac 174194e21283SToby Isaac Level: intermediate 174294e21283SToby Isaac 1743dce8aebaSBarry Smith Note: 1744*1d27aa22SBarry Smith This quadrature rule is exact for polynomials up to degree 2*`npoints` - 1. 1745dce8aebaSBarry Smith 1746dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()` 174794e21283SToby Isaac @*/ 1748d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussJacobiQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[]) 1749d71ae5a4SJacob Faibussowitsch { 175094e21283SToby Isaac PetscInt i; 1751e6a796c3SToby Isaac 1752e6a796c3SToby Isaac PetscFunctionBegin; 17539566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints, alpha, beta, x, w, PetscDTGaussQuadratureNewton_Internal)); 175494e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 175594e21283SToby Isaac for (i = 0; i < npoints; i++) { 175694e21283SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 175794e21283SToby Isaac w[i] *= (b - a) / 2.; 175894e21283SToby Isaac } 175994e21283SToby Isaac } 17603ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1761e6a796c3SToby Isaac } 1762e6a796c3SToby Isaac 1763d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTGaussLobattoJacobiQuadrature_Internal(PetscInt npoints, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[], PetscBool newton) 1764d71ae5a4SJacob Faibussowitsch { 1765e6a796c3SToby Isaac PetscInt i; 1766e6a796c3SToby Isaac 1767e6a796c3SToby Isaac PetscFunctionBegin; 176808401ef6SPierre Jolivet PetscCheck(npoints >= 2, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be positive"); 1769e6a796c3SToby Isaac /* If asking for a 1D Lobatto point, just return the non-Lobatto 1D point */ 177008401ef6SPierre Jolivet PetscCheck(alpha > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "alpha must be > -1."); 177108401ef6SPierre Jolivet PetscCheck(beta > -1., PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "beta must be > -1."); 1772e6a796c3SToby Isaac 1773e6a796c3SToby Isaac x[0] = -1.; 1774e6a796c3SToby Isaac x[npoints - 1] = 1.; 177548a46eb9SPierre Jolivet if (npoints > 2) PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints - 2, alpha + 1., beta + 1., &x[1], &w[1], newton)); 1776ad540459SPierre Jolivet for (i = 1; i < npoints - 1; i++) w[i] /= (1. - x[i] * x[i]); 17779566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiEndweights_Internal(npoints, alpha, beta, &w[0], &w[npoints - 1])); 17783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1779e6a796c3SToby Isaac } 1780e6a796c3SToby Isaac 178137045ce4SJed Brown /*@ 1782*1d27aa22SBarry Smith PetscDTGaussLobattoJacobiQuadrature - quadrature for the interval $[a, b]$ with the weight function 1783*1d27aa22SBarry Smith $(x-a)^\alpha (x-b)^\beta$, with endpoints `a` and `b` included as quadrature points. 178494e21283SToby Isaac 178520f4b53cSBarry Smith Not Collective 178694e21283SToby Isaac 178794e21283SToby Isaac Input Parameters: 178894e21283SToby Isaac + npoints - the number of points in the quadrature rule 178994e21283SToby Isaac . a - the left endpoint of the interval 179094e21283SToby Isaac . b - the right endpoint of the interval 179194e21283SToby Isaac . alpha - the left exponent 179294e21283SToby Isaac - beta - the right exponent 179394e21283SToby Isaac 179494e21283SToby Isaac Output Parameters: 179520f4b53cSBarry Smith + x - array of length `npoints`, the locations of the quadrature points 179620f4b53cSBarry Smith - w - array of length `npoints`, the weights of the quadrature points 179794e21283SToby Isaac 179894e21283SToby Isaac Level: intermediate 179994e21283SToby Isaac 1800dce8aebaSBarry Smith Note: 1801*1d27aa22SBarry Smith This quadrature rule is exact for polynomials up to degree 2*`npoints` - 3. 1802dce8aebaSBarry Smith 1803dce8aebaSBarry Smith .seealso: `PetscDTGaussJacobiQuadrature()` 180494e21283SToby Isaac @*/ 1805d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussLobattoJacobiQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal alpha, PetscReal beta, PetscReal x[], PetscReal w[]) 1806d71ae5a4SJacob Faibussowitsch { 180794e21283SToby Isaac PetscInt i; 180894e21283SToby Isaac 180994e21283SToby Isaac PetscFunctionBegin; 18109566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiQuadrature_Internal(npoints, alpha, beta, x, w, PetscDTGaussQuadratureNewton_Internal)); 181194e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 181294e21283SToby Isaac for (i = 0; i < npoints; i++) { 181394e21283SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 181494e21283SToby Isaac w[i] *= (b - a) / 2.; 181594e21283SToby Isaac } 181694e21283SToby Isaac } 18173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 181894e21283SToby Isaac } 181994e21283SToby Isaac 182094e21283SToby Isaac /*@ 1821e6a796c3SToby Isaac PetscDTGaussQuadrature - create Gauss-Legendre quadrature 182237045ce4SJed Brown 182337045ce4SJed Brown Not Collective 182437045ce4SJed Brown 18254165533cSJose E. Roman Input Parameters: 182637045ce4SJed Brown + npoints - number of points 182737045ce4SJed Brown . a - left end of interval (often-1) 182837045ce4SJed Brown - b - right end of interval (often +1) 182937045ce4SJed Brown 18304165533cSJose E. Roman Output Parameters: 183137045ce4SJed Brown + x - quadrature points 183237045ce4SJed Brown - w - quadrature weights 183337045ce4SJed Brown 183437045ce4SJed Brown Level: intermediate 183537045ce4SJed Brown 1836*1d27aa22SBarry Smith Note: 1837*1d27aa22SBarry Smith See {cite}`golub1969calculation` 183837045ce4SJed Brown 1839dce8aebaSBarry Smith .seealso: `PetscDTLegendreEval()`, `PetscDTGaussJacobiQuadrature()` 184037045ce4SJed Brown @*/ 1841d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussQuadrature(PetscInt npoints, PetscReal a, PetscReal b, PetscReal *x, PetscReal *w) 1842d71ae5a4SJacob Faibussowitsch { 184337045ce4SJed Brown PetscInt i; 184437045ce4SJed Brown 184537045ce4SJed Brown PetscFunctionBegin; 18469566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature_Internal(npoints, 0., 0., x, w, PetscDTGaussQuadratureNewton_Internal)); 184794e21283SToby Isaac if (a != -1. || b != 1.) { /* shift */ 184837045ce4SJed Brown for (i = 0; i < npoints; i++) { 1849e6a796c3SToby Isaac x[i] = (x[i] + 1.) * ((b - a) / 2.) + a; 1850e6a796c3SToby Isaac w[i] *= (b - a) / 2.; 185137045ce4SJed Brown } 185237045ce4SJed Brown } 18533ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 185437045ce4SJed Brown } 1855194825f6SJed Brown 18568272889dSSatish Balay /*@C 18578272889dSSatish Balay PetscDTGaussLobattoLegendreQuadrature - creates a set of the locations and weights of the Gauss-Lobatto-Legendre 1858*1d27aa22SBarry Smith nodes of a given size on the domain $[-1,1]$ 18598272889dSSatish Balay 18608272889dSSatish Balay Not Collective 18618272889dSSatish Balay 1862d8d19677SJose E. Roman Input Parameters: 186360225df5SJacob Faibussowitsch + npoints - number of grid nodes 1864dce8aebaSBarry Smith - type - `PETSCGAUSSLOBATTOLEGENDRE_VIA_LINEAR_ALGEBRA` or `PETSCGAUSSLOBATTOLEGENDRE_VIA_NEWTON` 18658272889dSSatish Balay 18664165533cSJose E. Roman Output Parameters: 18678272889dSSatish Balay + x - quadrature points 18688272889dSSatish Balay - w - quadrature weights 18698272889dSSatish Balay 1870dce8aebaSBarry Smith Level: intermediate 1871dce8aebaSBarry Smith 18728272889dSSatish Balay Notes: 18738272889dSSatish Balay For n > 30 the Newton approach computes duplicate (incorrect) values for some nodes because the initial guess is apparently not 18748272889dSSatish Balay close enough to the desired solution 18758272889dSSatish Balay 18768272889dSSatish Balay These are useful for implementing spectral methods based on Gauss-Lobatto-Legendre (GLL) nodes 18778272889dSSatish Balay 1878*1d27aa22SBarry 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 18798272889dSSatish Balay 1880dce8aebaSBarry Smith .seealso: `PetscDTGaussQuadrature()`, `PetscGaussLobattoLegendreCreateType` 18818272889dSSatish Balay 18828272889dSSatish Balay @*/ 1883d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussLobattoLegendreQuadrature(PetscInt npoints, PetscGaussLobattoLegendreCreateType type, PetscReal *x, PetscReal *w) 1884d71ae5a4SJacob Faibussowitsch { 1885e6a796c3SToby Isaac PetscBool newton; 18868272889dSSatish Balay 18878272889dSSatish Balay PetscFunctionBegin; 188808401ef6SPierre Jolivet PetscCheck(npoints >= 2, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must provide at least 2 grid points per element"); 188994e21283SToby Isaac newton = (PetscBool)(type == PETSCGAUSSLOBATTOLEGENDRE_VIA_NEWTON); 18909566063dSJacob Faibussowitsch PetscCall(PetscDTGaussLobattoJacobiQuadrature_Internal(npoints, 0., 0., x, w, newton)); 18913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 18928272889dSSatish Balay } 18938272889dSSatish Balay 1894744bafbcSMatthew G. Knepley /*@ 1895744bafbcSMatthew G. Knepley PetscDTGaussTensorQuadrature - creates a tensor-product Gauss quadrature 1896744bafbcSMatthew G. Knepley 1897744bafbcSMatthew G. Knepley Not Collective 1898744bafbcSMatthew G. Knepley 18994165533cSJose E. Roman Input Parameters: 1900744bafbcSMatthew G. Knepley + dim - The spatial dimension 1901a6b92713SMatthew G. Knepley . Nc - The number of components 1902744bafbcSMatthew G. Knepley . npoints - number of points in one dimension 1903744bafbcSMatthew G. Knepley . a - left end of interval (often-1) 1904744bafbcSMatthew G. Knepley - b - right end of interval (often +1) 1905744bafbcSMatthew G. Knepley 19064165533cSJose E. Roman Output Parameter: 1907dce8aebaSBarry Smith . q - A `PetscQuadrature` object 1908744bafbcSMatthew G. Knepley 1909744bafbcSMatthew G. Knepley Level: intermediate 1910744bafbcSMatthew G. Knepley 1911db781477SPatrick Sanan .seealso: `PetscDTGaussQuadrature()`, `PetscDTLegendreEval()` 1912744bafbcSMatthew G. Knepley @*/ 1913d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTGaussTensorQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 1914d71ae5a4SJacob Faibussowitsch { 19154366bac7SMatthew G. Knepley DMPolytopeType ct; 19164366bac7SMatthew G. Knepley PetscInt totpoints = dim > 1 ? dim > 2 ? npoints * PetscSqr(npoints) : PetscSqr(npoints) : npoints; 1917744bafbcSMatthew G. Knepley PetscReal *x, *w, *xw, *ww; 1918744bafbcSMatthew G. Knepley 1919744bafbcSMatthew G. Knepley PetscFunctionBegin; 19209566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * dim, &x)); 19219566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * Nc, &w)); 1922744bafbcSMatthew G. Knepley /* Set up the Golub-Welsch system */ 1923744bafbcSMatthew G. Knepley switch (dim) { 1924744bafbcSMatthew G. Knepley case 0: 19254366bac7SMatthew G. Knepley ct = DM_POLYTOPE_POINT; 19269566063dSJacob Faibussowitsch PetscCall(PetscFree(x)); 19279566063dSJacob Faibussowitsch PetscCall(PetscFree(w)); 19289566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(1, &x)); 19299566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Nc, &w)); 19303c1919fdSMatthew G. Knepley totpoints = 1; 1931744bafbcSMatthew G. Knepley x[0] = 0.0; 19324366bac7SMatthew G. Knepley for (PetscInt c = 0; c < Nc; ++c) w[c] = 1.0; 1933744bafbcSMatthew G. Knepley break; 1934744bafbcSMatthew G. Knepley case 1: 19354366bac7SMatthew G. Knepley ct = DM_POLYTOPE_SEGMENT; 19369566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints, &ww)); 19379566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, x, ww)); 19384366bac7SMatthew G. Knepley for (PetscInt i = 0; i < npoints; ++i) 19394366bac7SMatthew G. Knepley for (PetscInt c = 0; c < Nc; ++c) w[i * Nc + c] = ww[i]; 19409566063dSJacob Faibussowitsch PetscCall(PetscFree(ww)); 1941744bafbcSMatthew G. Knepley break; 1942744bafbcSMatthew G. Knepley case 2: 19434366bac7SMatthew G. Knepley ct = DM_POLYTOPE_QUADRILATERAL; 19449566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &xw, npoints, &ww)); 19459566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, xw, ww)); 19464366bac7SMatthew G. Knepley for (PetscInt i = 0; i < npoints; ++i) { 19474366bac7SMatthew G. Knepley for (PetscInt j = 0; j < npoints; ++j) { 1948744bafbcSMatthew G. Knepley x[(i * npoints + j) * dim + 0] = xw[i]; 1949744bafbcSMatthew G. Knepley x[(i * npoints + j) * dim + 1] = xw[j]; 19504366bac7SMatthew G. Knepley for (PetscInt c = 0; c < Nc; ++c) w[(i * npoints + j) * Nc + c] = ww[i] * ww[j]; 1951744bafbcSMatthew G. Knepley } 1952744bafbcSMatthew G. Knepley } 19539566063dSJacob Faibussowitsch PetscCall(PetscFree2(xw, ww)); 1954744bafbcSMatthew G. Knepley break; 1955744bafbcSMatthew G. Knepley case 3: 19564366bac7SMatthew G. Knepley ct = DM_POLYTOPE_HEXAHEDRON; 19579566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &xw, npoints, &ww)); 19589566063dSJacob Faibussowitsch PetscCall(PetscDTGaussQuadrature(npoints, a, b, xw, ww)); 19594366bac7SMatthew G. Knepley for (PetscInt i = 0; i < npoints; ++i) { 19604366bac7SMatthew G. Knepley for (PetscInt j = 0; j < npoints; ++j) { 19614366bac7SMatthew G. Knepley for (PetscInt k = 0; k < npoints; ++k) { 1962744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 0] = xw[i]; 1963744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 1] = xw[j]; 1964744bafbcSMatthew G. Knepley x[((i * npoints + j) * npoints + k) * dim + 2] = xw[k]; 19654366bac7SMatthew G. Knepley for (PetscInt c = 0; c < Nc; ++c) w[((i * npoints + j) * npoints + k) * Nc + c] = ww[i] * ww[j] * ww[k]; 1966744bafbcSMatthew G. Knepley } 1967744bafbcSMatthew G. Knepley } 1968744bafbcSMatthew G. Knepley } 19699566063dSJacob Faibussowitsch PetscCall(PetscFree2(xw, ww)); 1970744bafbcSMatthew G. Knepley break; 1971d71ae5a4SJacob Faibussowitsch default: 1972d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot construct quadrature rule for dimension %" PetscInt_FMT, dim); 1973744bafbcSMatthew G. Knepley } 19749566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 19754366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*q, ct)); 19769566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * npoints - 1)); 19779566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w)); 19789566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)*q, "GaussTensor")); 19793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1980744bafbcSMatthew G. Knepley } 1981744bafbcSMatthew G. Knepley 1982f5f57ec0SBarry Smith /*@ 1983*1d27aa22SBarry Smith PetscDTStroudConicalQuadrature - create Stroud conical quadrature for a simplex {cite}`karniadakis2005spectral` 1984494e7359SMatthew G. Knepley 1985494e7359SMatthew G. Knepley Not Collective 1986494e7359SMatthew G. Knepley 19874165533cSJose E. Roman Input Parameters: 1988494e7359SMatthew G. Knepley + dim - The simplex dimension 1989a6b92713SMatthew G. Knepley . Nc - The number of components 1990dcce0ee2SMatthew G. Knepley . npoints - The number of points in one dimension 1991494e7359SMatthew G. Knepley . a - left end of interval (often-1) 1992494e7359SMatthew G. Knepley - b - right end of interval (often +1) 1993494e7359SMatthew G. Knepley 19944165533cSJose E. Roman Output Parameter: 199520f4b53cSBarry Smith . q - A `PetscQuadrature` object 1996494e7359SMatthew G. Knepley 1997494e7359SMatthew G. Knepley Level: intermediate 1998494e7359SMatthew G. Knepley 1999dce8aebaSBarry Smith Note: 200020f4b53cSBarry Smith For `dim` == 1, this is Gauss-Legendre quadrature 2001dce8aebaSBarry Smith 2002db781477SPatrick Sanan .seealso: `PetscDTGaussTensorQuadrature()`, `PetscDTGaussQuadrature()` 2003494e7359SMatthew G. Knepley @*/ 2004d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTStroudConicalQuadrature(PetscInt dim, PetscInt Nc, PetscInt npoints, PetscReal a, PetscReal b, PetscQuadrature *q) 2005d71ae5a4SJacob Faibussowitsch { 20064366bac7SMatthew G. Knepley DMPolytopeType ct; 2007fbdc3dfeSToby Isaac PetscInt totpoints; 2008fbdc3dfeSToby Isaac PetscReal *p1, *w1; 2009fbdc3dfeSToby Isaac PetscReal *x, *w; 2010494e7359SMatthew G. Knepley 2011494e7359SMatthew G. Knepley PetscFunctionBegin; 201208401ef6SPierre Jolivet PetscCheck(!(a != -1.0) && !(b != 1.0), PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must use default internal right now"); 20134366bac7SMatthew G. Knepley switch (dim) { 20144366bac7SMatthew G. Knepley case 0: 20154366bac7SMatthew G. Knepley ct = DM_POLYTOPE_POINT; 20164366bac7SMatthew G. Knepley break; 20174366bac7SMatthew G. Knepley case 1: 20184366bac7SMatthew G. Knepley ct = DM_POLYTOPE_SEGMENT; 20194366bac7SMatthew G. Knepley break; 20204366bac7SMatthew G. Knepley case 2: 20214366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TRIANGLE; 20224366bac7SMatthew G. Knepley break; 20234366bac7SMatthew G. Knepley case 3: 20244366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TETRAHEDRON; 20254366bac7SMatthew G. Knepley break; 20264366bac7SMatthew G. Knepley default: 20274366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 20284366bac7SMatthew G. Knepley } 2029fbdc3dfeSToby Isaac totpoints = 1; 20304366bac7SMatthew G. Knepley for (PetscInt i = 0; i < dim; ++i) totpoints *= npoints; 20319566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * dim, &x)); 20329566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(totpoints * Nc, &w)); 20339566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(npoints, &p1, npoints, &w1)); 20344366bac7SMatthew G. Knepley for (PetscInt i = 0; i < totpoints * Nc; ++i) w[i] = 1.; 20354366bac7SMatthew G. Knepley for (PetscInt i = 0, totprev = 1, totrem = totpoints / npoints; i < dim; ++i) { 2036fbdc3dfeSToby Isaac PetscReal mul; 2037fbdc3dfeSToby Isaac 2038fbdc3dfeSToby Isaac mul = PetscPowReal(2., -i); 20399566063dSJacob Faibussowitsch PetscCall(PetscDTGaussJacobiQuadrature(npoints, -1., 1., i, 0.0, p1, w1)); 20404366bac7SMatthew G. Knepley for (PetscInt pt = 0, l = 0; l < totprev; l++) { 20414366bac7SMatthew G. Knepley for (PetscInt j = 0; j < npoints; j++) { 20424366bac7SMatthew G. Knepley for (PetscInt m = 0; m < totrem; m++, pt++) { 20434366bac7SMatthew G. Knepley for (PetscInt k = 0; k < i; k++) x[pt * dim + k] = (x[pt * dim + k] + 1.) * (1. - p1[j]) * 0.5 - 1.; 2044fbdc3dfeSToby Isaac x[pt * dim + i] = p1[j]; 20454366bac7SMatthew G. Knepley for (PetscInt c = 0; c < Nc; c++) w[pt * Nc + c] *= mul * w1[j]; 2046494e7359SMatthew G. Knepley } 2047494e7359SMatthew G. Knepley } 2048494e7359SMatthew G. Knepley } 2049fbdc3dfeSToby Isaac totprev *= npoints; 2050fbdc3dfeSToby Isaac totrem /= npoints; 2051494e7359SMatthew G. Knepley } 20529566063dSJacob Faibussowitsch PetscCall(PetscFree2(p1, w1)); 20539566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 20544366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*q, ct)); 20559566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * npoints - 1)); 20569566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, totpoints, x, w)); 20579566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)*q, "StroudConical")); 20583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2059494e7359SMatthew G. Knepley } 2060494e7359SMatthew G. Knepley 2061d3c69ad0SToby Isaac static PetscBool MinSymTriQuadCite = PETSC_FALSE; 20629371c9d4SSatish Balay const char MinSymTriQuadCitation[] = "@article{WitherdenVincent2015,\n" 2063d3c69ad0SToby Isaac " title = {On the identification of symmetric quadrature rules for finite element methods},\n" 2064d3c69ad0SToby Isaac " journal = {Computers & Mathematics with Applications},\n" 2065d3c69ad0SToby Isaac " volume = {69},\n" 2066d3c69ad0SToby Isaac " number = {10},\n" 2067d3c69ad0SToby Isaac " pages = {1232-1241},\n" 2068d3c69ad0SToby Isaac " year = {2015},\n" 2069d3c69ad0SToby Isaac " issn = {0898-1221},\n" 2070d3c69ad0SToby Isaac " doi = {10.1016/j.camwa.2015.03.017},\n" 2071d3c69ad0SToby Isaac " url = {https://www.sciencedirect.com/science/article/pii/S0898122115001224},\n" 2072d3c69ad0SToby Isaac " author = {F.D. Witherden and P.E. Vincent},\n" 2073d3c69ad0SToby Isaac "}\n"; 2074d3c69ad0SToby Isaac 2075d3c69ad0SToby Isaac #include "petscdttriquadrules.h" 2076d3c69ad0SToby Isaac 2077d3c69ad0SToby Isaac static PetscBool MinSymTetQuadCite = PETSC_FALSE; 20789371c9d4SSatish Balay const char MinSymTetQuadCitation[] = "@article{JaskowiecSukumar2021\n" 2079d3c69ad0SToby Isaac " author = {Jaskowiec, Jan and Sukumar, N.},\n" 2080d3c69ad0SToby Isaac " title = {High-order symmetric cubature rules for tetrahedra and pyramids},\n" 2081d3c69ad0SToby Isaac " journal = {International Journal for Numerical Methods in Engineering},\n" 2082d3c69ad0SToby Isaac " volume = {122},\n" 2083d3c69ad0SToby Isaac " number = {1},\n" 2084d3c69ad0SToby Isaac " pages = {148-171},\n" 2085d3c69ad0SToby Isaac " doi = {10.1002/nme.6528},\n" 2086d3c69ad0SToby Isaac " url = {https://onlinelibrary.wiley.com/doi/abs/10.1002/nme.6528},\n" 2087d3c69ad0SToby Isaac " eprint = {https://onlinelibrary.wiley.com/doi/pdf/10.1002/nme.6528},\n" 2088d3c69ad0SToby Isaac " year = {2021}\n" 2089d3c69ad0SToby Isaac "}\n"; 2090d3c69ad0SToby Isaac 2091d3c69ad0SToby Isaac #include "petscdttetquadrules.h" 2092d3c69ad0SToby Isaac 2093d3c69ad0SToby Isaac // https://en.wikipedia.org/wiki/Partition_(number_theory) 2094d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPartitionNumber(PetscInt n, PetscInt *p) 2095d71ae5a4SJacob Faibussowitsch { 2096d3c69ad0SToby Isaac // sequence A000041 in the OEIS 2097d3c69ad0SToby 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}; 2098d3c69ad0SToby Isaac PetscInt tabulated_max = PETSC_STATIC_ARRAY_LENGTH(partition) - 1; 2099d3c69ad0SToby Isaac 2100d3c69ad0SToby Isaac PetscFunctionBegin; 2101d3c69ad0SToby Isaac PetscCheck(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Partition number not defined for negative number %" PetscInt_FMT, n); 2102d3c69ad0SToby Isaac // not implementing the pentagonal number recurrence, we don't need partition numbers for n that high 2103d3c69ad0SToby 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); 2104d3c69ad0SToby Isaac *p = partition[n]; 21053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2106d3c69ad0SToby Isaac } 2107d3c69ad0SToby Isaac 2108d3c69ad0SToby Isaac /*@ 2109d3c69ad0SToby Isaac PetscDTSimplexQuadrature - Create a quadrature rule for a simplex that exactly integrates polynomials up to a given degree. 2110d3c69ad0SToby Isaac 2111d3c69ad0SToby Isaac Not Collective 2112d3c69ad0SToby Isaac 2113d3c69ad0SToby Isaac Input Parameters: 2114d3c69ad0SToby Isaac + dim - The spatial dimension of the simplex (1 = segment, 2 = triangle, 3 = tetrahedron) 2115d3c69ad0SToby Isaac . degree - The largest polynomial degree that is required to be integrated exactly 2116d3c69ad0SToby Isaac - type - left end of interval (often-1) 2117d3c69ad0SToby Isaac 2118d3c69ad0SToby Isaac Output Parameter: 2119dce8aebaSBarry Smith . quad - A `PetscQuadrature` object for integration over the biunit simplex 2120d3c69ad0SToby Isaac (defined by the bounds $x_i >= -1$ and $\sum_i x_i <= 2 - d$) that is exact for 2121d3c69ad0SToby Isaac polynomials up to the given degree 2122d3c69ad0SToby Isaac 2123d3c69ad0SToby Isaac Level: intermediate 2124d3c69ad0SToby Isaac 2125dce8aebaSBarry Smith .seealso: `PetscDTSimplexQuadratureType`, `PetscDTGaussQuadrature()`, `PetscDTStroudCononicalQuadrature()`, `PetscQuadrature` 2126d3c69ad0SToby Isaac @*/ 2127d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTSimplexQuadrature(PetscInt dim, PetscInt degree, PetscDTSimplexQuadratureType type, PetscQuadrature *quad) 2128d71ae5a4SJacob Faibussowitsch { 2129d3c69ad0SToby Isaac PetscDTSimplexQuadratureType orig_type = type; 2130d3c69ad0SToby Isaac 2131d3c69ad0SToby Isaac PetscFunctionBegin; 2132d3c69ad0SToby Isaac PetscCheck(dim >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative dimension %" PetscInt_FMT, dim); 2133d3c69ad0SToby Isaac PetscCheck(degree >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative degree %" PetscInt_FMT, degree); 2134ad540459SPierre Jolivet if (type == PETSCDTSIMPLEXQUAD_DEFAULT) type = PETSCDTSIMPLEXQUAD_MINSYM; 2135d3c69ad0SToby Isaac if (type == PETSCDTSIMPLEXQUAD_CONIC || dim < 2) { 2136d3c69ad0SToby Isaac PetscInt points_per_dim = (degree + 2) / 2; // ceil((degree + 1) / 2); 2137d3c69ad0SToby Isaac PetscCall(PetscDTStroudConicalQuadrature(dim, 1, points_per_dim, -1, 1, quad)); 2138d3c69ad0SToby Isaac } else { 21394366bac7SMatthew G. Knepley DMPolytopeType ct; 2140d3c69ad0SToby Isaac PetscInt n = dim + 1; 2141d3c69ad0SToby Isaac PetscInt fact = 1; 2142d3c69ad0SToby Isaac PetscInt *part, *perm; 2143d3c69ad0SToby Isaac PetscInt p = 0; 2144d3c69ad0SToby Isaac PetscInt max_degree; 2145d3c69ad0SToby Isaac const PetscInt *nodes_per_type = NULL; 2146d3c69ad0SToby Isaac const PetscInt *all_num_full_nodes = NULL; 2147d3c69ad0SToby Isaac const PetscReal **weights_list = NULL; 2148d3c69ad0SToby Isaac const PetscReal **compact_nodes_list = NULL; 2149d3c69ad0SToby Isaac const char *citation = NULL; 2150d3c69ad0SToby Isaac PetscBool *cited = NULL; 2151d3c69ad0SToby Isaac 2152d3c69ad0SToby Isaac switch (dim) { 21534366bac7SMatthew G. Knepley case 0: 21544366bac7SMatthew G. Knepley ct = DM_POLYTOPE_POINT; 21554366bac7SMatthew G. Knepley break; 21564366bac7SMatthew G. Knepley case 1: 21574366bac7SMatthew G. Knepley ct = DM_POLYTOPE_SEGMENT; 21584366bac7SMatthew G. Knepley break; 21594366bac7SMatthew G. Knepley case 2: 21604366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TRIANGLE; 21614366bac7SMatthew G. Knepley break; 21624366bac7SMatthew G. Knepley case 3: 21634366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TETRAHEDRON; 21644366bac7SMatthew G. Knepley break; 21654366bac7SMatthew G. Knepley default: 21664366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 21674366bac7SMatthew G. Knepley } 21684366bac7SMatthew G. Knepley switch (dim) { 2169d3c69ad0SToby Isaac case 2: 2170d3c69ad0SToby Isaac cited = &MinSymTriQuadCite; 2171d3c69ad0SToby Isaac citation = MinSymTriQuadCitation; 2172d3c69ad0SToby Isaac max_degree = PetscDTWVTriQuad_max_degree; 2173d3c69ad0SToby Isaac nodes_per_type = PetscDTWVTriQuad_num_orbits; 2174d3c69ad0SToby Isaac all_num_full_nodes = PetscDTWVTriQuad_num_nodes; 2175d3c69ad0SToby Isaac weights_list = PetscDTWVTriQuad_weights; 2176d3c69ad0SToby Isaac compact_nodes_list = PetscDTWVTriQuad_orbits; 2177d3c69ad0SToby Isaac break; 2178d3c69ad0SToby Isaac case 3: 2179d3c69ad0SToby Isaac cited = &MinSymTetQuadCite; 2180d3c69ad0SToby Isaac citation = MinSymTetQuadCitation; 2181d3c69ad0SToby Isaac max_degree = PetscDTJSTetQuad_max_degree; 2182d3c69ad0SToby Isaac nodes_per_type = PetscDTJSTetQuad_num_orbits; 2183d3c69ad0SToby Isaac all_num_full_nodes = PetscDTJSTetQuad_num_nodes; 2184d3c69ad0SToby Isaac weights_list = PetscDTJSTetQuad_weights; 2185d3c69ad0SToby Isaac compact_nodes_list = PetscDTJSTetQuad_orbits; 2186d3c69ad0SToby Isaac break; 2187d71ae5a4SJacob Faibussowitsch default: 2188d71ae5a4SJacob Faibussowitsch max_degree = -1; 2189d71ae5a4SJacob Faibussowitsch break; 2190d3c69ad0SToby Isaac } 2191d3c69ad0SToby Isaac 2192d3c69ad0SToby Isaac if (degree > max_degree) { 2193d3c69ad0SToby Isaac if (orig_type == PETSCDTSIMPLEXQUAD_DEFAULT) { 2194d3c69ad0SToby Isaac // fall back to conic 2195d3c69ad0SToby Isaac PetscCall(PetscDTSimplexQuadrature(dim, degree, PETSCDTSIMPLEXQUAD_CONIC, quad)); 21963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2197d3c69ad0SToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Minimal symmetric quadrature for dim %" PetscInt_FMT ", degree %" PetscInt_FMT " unsupported", dim, degree); 2198d3c69ad0SToby Isaac } 2199d3c69ad0SToby Isaac 2200d3c69ad0SToby Isaac PetscCall(PetscCitationsRegister(citation, cited)); 2201d3c69ad0SToby Isaac 2202d3c69ad0SToby Isaac PetscCall(PetscDTPartitionNumber(n, &p)); 2203d3c69ad0SToby Isaac for (PetscInt d = 2; d <= n; d++) fact *= d; 2204d3c69ad0SToby Isaac 2205d3c69ad0SToby Isaac PetscInt num_full_nodes = all_num_full_nodes[degree]; 2206d3c69ad0SToby Isaac const PetscReal *all_compact_nodes = compact_nodes_list[degree]; 2207d3c69ad0SToby Isaac const PetscReal *all_compact_weights = weights_list[degree]; 2208d3c69ad0SToby Isaac nodes_per_type = &nodes_per_type[p * degree]; 2209d3c69ad0SToby Isaac 2210d3c69ad0SToby Isaac PetscReal *points; 2211d3c69ad0SToby Isaac PetscReal *counts; 2212d3c69ad0SToby Isaac PetscReal *weights; 2213d3c69ad0SToby Isaac PetscReal *bary_to_biunit; // row-major transformation of barycentric coordinate to biunit 2214d3c69ad0SToby Isaac PetscQuadrature q; 2215d3c69ad0SToby Isaac 2216d3c69ad0SToby Isaac // compute the transformation 2217d3c69ad0SToby Isaac PetscCall(PetscMalloc1(n * dim, &bary_to_biunit)); 2218d3c69ad0SToby Isaac for (PetscInt d = 0; d < dim; d++) { 2219ad540459SPierre Jolivet for (PetscInt b = 0; b < n; b++) bary_to_biunit[d * n + b] = (d == b) ? 1.0 : -1.0; 2220d3c69ad0SToby Isaac } 2221d3c69ad0SToby Isaac 2222d3c69ad0SToby Isaac PetscCall(PetscMalloc3(n, &part, n, &perm, n, &counts)); 2223d3c69ad0SToby Isaac PetscCall(PetscCalloc1(num_full_nodes * dim, &points)); 2224d3c69ad0SToby Isaac PetscCall(PetscMalloc1(num_full_nodes, &weights)); 2225d3c69ad0SToby Isaac 2226d3c69ad0SToby Isaac // (0, 0, ...) is the first partition lexicographically 2227d3c69ad0SToby Isaac PetscCall(PetscArrayzero(part, n)); 2228d3c69ad0SToby Isaac PetscCall(PetscArrayzero(counts, n)); 2229d3c69ad0SToby Isaac counts[0] = n; 2230d3c69ad0SToby Isaac 2231d3c69ad0SToby Isaac // for each partition 2232d3c69ad0SToby Isaac for (PetscInt s = 0, node_offset = 0; s < p; s++) { 2233d3c69ad0SToby Isaac PetscInt num_compact_coords = part[n - 1] + 1; 2234d3c69ad0SToby Isaac 2235d3c69ad0SToby Isaac const PetscReal *compact_nodes = all_compact_nodes; 2236d3c69ad0SToby Isaac const PetscReal *compact_weights = all_compact_weights; 2237d3c69ad0SToby Isaac all_compact_nodes += num_compact_coords * nodes_per_type[s]; 2238d3c69ad0SToby Isaac all_compact_weights += nodes_per_type[s]; 2239d3c69ad0SToby Isaac 2240d3c69ad0SToby Isaac // for every permutation of the vertices 2241d3c69ad0SToby Isaac for (PetscInt f = 0; f < fact; f++) { 2242d3c69ad0SToby Isaac PetscCall(PetscDTEnumPerm(n, f, perm, NULL)); 2243d3c69ad0SToby Isaac 2244d3c69ad0SToby Isaac // check if it is a valid permutation 2245d3c69ad0SToby Isaac PetscInt digit; 2246d3c69ad0SToby Isaac for (digit = 1; digit < n; digit++) { 2247d3c69ad0SToby Isaac // skip permutations that would duplicate a node because it has a smaller symmetry group 2248d3c69ad0SToby Isaac if (part[digit - 1] == part[digit] && perm[digit - 1] > perm[digit]) break; 2249d3c69ad0SToby Isaac } 2250d3c69ad0SToby Isaac if (digit < n) continue; 2251d3c69ad0SToby Isaac 2252d3c69ad0SToby Isaac // create full nodes from this permutation of the compact nodes 2253d3c69ad0SToby Isaac PetscReal *full_nodes = &points[node_offset * dim]; 2254d3c69ad0SToby Isaac PetscReal *full_weights = &weights[node_offset]; 2255d3c69ad0SToby Isaac 2256d3c69ad0SToby Isaac PetscCall(PetscArraycpy(full_weights, compact_weights, nodes_per_type[s])); 2257d3c69ad0SToby Isaac for (PetscInt b = 0; b < n; b++) { 2258d3c69ad0SToby Isaac for (PetscInt d = 0; d < dim; d++) { 2259ad540459SPierre 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]]; 2260d3c69ad0SToby Isaac } 2261d3c69ad0SToby Isaac } 2262d3c69ad0SToby Isaac node_offset += nodes_per_type[s]; 2263d3c69ad0SToby Isaac } 2264d3c69ad0SToby Isaac 2265d3c69ad0SToby Isaac if (s < p - 1) { // Generate the next partition 2266d3c69ad0SToby Isaac /* A partition is described by the number of coordinates that are in 2267d3c69ad0SToby Isaac * each set of duplicates (counts) and redundantly by mapping each 2268d3c69ad0SToby Isaac * index to its set of duplicates (part) 2269d3c69ad0SToby Isaac * 2270d3c69ad0SToby Isaac * Counts should always be in nonincreasing order 2271d3c69ad0SToby Isaac * 2272d3c69ad0SToby Isaac * We want to generate the partitions lexically by part, which means 2273d3c69ad0SToby Isaac * finding the last index where count > 1 and reducing by 1. 2274d3c69ad0SToby Isaac * 2275d3c69ad0SToby Isaac * For the new counts beyond that index, we eagerly assign the remaining 2276d3c69ad0SToby Isaac * capacity of the partition to smaller indices (ensures lexical ordering), 2277d3c69ad0SToby Isaac * while respecting the nonincreasing invariant of the counts 2278d3c69ad0SToby Isaac */ 2279d3c69ad0SToby Isaac PetscInt last_digit = part[n - 1]; 2280d3c69ad0SToby Isaac PetscInt last_digit_with_extra = last_digit; 2281d3c69ad0SToby Isaac while (counts[last_digit_with_extra] == 1) last_digit_with_extra--; 2282d3c69ad0SToby Isaac PetscInt limit = --counts[last_digit_with_extra]; 2283d3c69ad0SToby Isaac PetscInt total_to_distribute = last_digit - last_digit_with_extra + 1; 2284d3c69ad0SToby Isaac for (PetscInt digit = last_digit_with_extra + 1; digit < n; digit++) { 2285d3c69ad0SToby Isaac counts[digit] = PetscMin(limit, total_to_distribute); 2286d3c69ad0SToby Isaac total_to_distribute -= PetscMin(limit, total_to_distribute); 2287d3c69ad0SToby Isaac } 2288d3c69ad0SToby Isaac for (PetscInt digit = 0, offset = 0; digit < n; digit++) { 2289d3c69ad0SToby Isaac PetscInt count = counts[digit]; 2290ad540459SPierre Jolivet for (PetscInt c = 0; c < count; c++) part[offset++] = digit; 2291d3c69ad0SToby Isaac } 2292d3c69ad0SToby Isaac } 2293d3c69ad0SToby Isaac } 2294d3c69ad0SToby Isaac PetscCall(PetscFree3(part, perm, counts)); 2295d3c69ad0SToby Isaac PetscCall(PetscFree(bary_to_biunit)); 2296d3c69ad0SToby Isaac PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, &q)); 22974366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(q, ct)); 2298b414c505SJed Brown PetscCall(PetscQuadratureSetOrder(q, degree)); 2299d3c69ad0SToby Isaac PetscCall(PetscQuadratureSetData(q, dim, 1, num_full_nodes, points, weights)); 2300d3c69ad0SToby Isaac *quad = q; 2301d3c69ad0SToby Isaac } 23023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2303d3c69ad0SToby Isaac } 2304d3c69ad0SToby Isaac 2305f5f57ec0SBarry Smith /*@ 2306b3c0f97bSTom Klotz PetscDTTanhSinhTensorQuadrature - create tanh-sinh quadrature for a tensor product cell 2307b3c0f97bSTom Klotz 2308b3c0f97bSTom Klotz Not Collective 2309b3c0f97bSTom Klotz 23104165533cSJose E. Roman Input Parameters: 2311b3c0f97bSTom Klotz + dim - The cell dimension 2312*1d27aa22SBarry Smith . level - The number of points in one dimension, $2^l$ 2313b3c0f97bSTom Klotz . a - left end of interval (often-1) 2314b3c0f97bSTom Klotz - b - right end of interval (often +1) 2315b3c0f97bSTom Klotz 23164165533cSJose E. Roman Output Parameter: 2317dce8aebaSBarry Smith . q - A `PetscQuadrature` object 2318b3c0f97bSTom Klotz 2319b3c0f97bSTom Klotz Level: intermediate 2320b3c0f97bSTom Klotz 2321dce8aebaSBarry Smith .seealso: `PetscDTGaussTensorQuadrature()`, `PetscQuadrature` 2322b3c0f97bSTom Klotz @*/ 2323d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhTensorQuadrature(PetscInt dim, PetscInt level, PetscReal a, PetscReal b, PetscQuadrature *q) 2324d71ae5a4SJacob Faibussowitsch { 23254366bac7SMatthew G. Knepley DMPolytopeType ct; 2326b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 2327b3c0f97bSTom Klotz const PetscReal alpha = (b - a) / 2.; /* Half-width of the integration interval */ 2328b3c0f97bSTom Klotz const PetscReal beta = (b + a) / 2.; /* Center of the integration interval */ 2329b3c0f97bSTom Klotz const PetscReal h = PetscPowReal(2.0, -level); /* Step size, length between x_k */ 2330d84b4d08SMatthew G. Knepley PetscReal xk; /* Quadrature point x_k on reference domain [-1, 1] */ 2331b3c0f97bSTom Klotz PetscReal wk = 0.5 * PETSC_PI; /* Quadrature weight at x_k */ 2332b3c0f97bSTom Klotz PetscReal *x, *w; 2333b3c0f97bSTom Klotz PetscInt K, k, npoints; 2334b3c0f97bSTom Klotz 2335b3c0f97bSTom Klotz PetscFunctionBegin; 233663a3b9bcSJacob Faibussowitsch PetscCheck(dim <= 1, PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not yet implemented", dim); 233728b400f6SJacob Faibussowitsch PetscCheck(level, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a number of significant digits"); 23384366bac7SMatthew G. Knepley switch (dim) { 23394366bac7SMatthew G. Knepley case 0: 23404366bac7SMatthew G. Knepley ct = DM_POLYTOPE_POINT; 23414366bac7SMatthew G. Knepley break; 23424366bac7SMatthew G. Knepley case 1: 23434366bac7SMatthew G. Knepley ct = DM_POLYTOPE_SEGMENT; 23444366bac7SMatthew G. Knepley break; 23454366bac7SMatthew G. Knepley case 2: 23464366bac7SMatthew G. Knepley ct = DM_POLYTOPE_QUADRILATERAL; 23474366bac7SMatthew G. Knepley break; 23484366bac7SMatthew G. Knepley case 3: 23494366bac7SMatthew G. Knepley ct = DM_POLYTOPE_HEXAHEDRON; 23504366bac7SMatthew G. Knepley break; 23514366bac7SMatthew G. Knepley default: 23524366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 23534366bac7SMatthew G. Knepley } 2354b3c0f97bSTom Klotz /* Find K such that the weights are < 32 digits of precision */ 2355ad540459SPierre 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))); 23569566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 23574366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*q, ct)); 23589566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, 2 * K + 1)); 2359b3c0f97bSTom Klotz npoints = 2 * K - 1; 23609566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints * dim, &x)); 23619566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(npoints, &w)); 2362b3c0f97bSTom Klotz /* Center term */ 2363b3c0f97bSTom Klotz x[0] = beta; 2364b3c0f97bSTom Klotz w[0] = 0.5 * alpha * PETSC_PI; 2365b3c0f97bSTom Klotz for (k = 1; k < K; ++k) { 23669add2064SThomas Klotz wk = 0.5 * alpha * h * PETSC_PI * PetscCoshReal(k * h) / PetscSqr(PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 23671118d4bcSLisandro Dalcin xk = PetscTanhReal(0.5 * PETSC_PI * PetscSinhReal(k * h)); 2368b3c0f97bSTom Klotz x[2 * k - 1] = -alpha * xk + beta; 2369b3c0f97bSTom Klotz w[2 * k - 1] = wk; 2370b3c0f97bSTom Klotz x[2 * k + 0] = alpha * xk + beta; 2371b3c0f97bSTom Klotz w[2 * k + 0] = wk; 2372b3c0f97bSTom Klotz } 23739566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, 1, npoints, x, w)); 23743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2375b3c0f97bSTom Klotz } 2376b3c0f97bSTom Klotz 2377d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrate(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2378d71ae5a4SJacob Faibussowitsch { 2379b3c0f97bSTom Klotz const PetscInt p = 16; /* Digits of precision in the evaluation */ 2380b3c0f97bSTom Klotz const PetscReal alpha = (b - a) / 2.; /* Half-width of the integration interval */ 2381b3c0f97bSTom Klotz const PetscReal beta = (b + a) / 2.; /* Center of the integration interval */ 2382b3c0f97bSTom Klotz PetscReal h = 1.0; /* Step size, length between x_k */ 2383b3c0f97bSTom Klotz PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 2384b3c0f97bSTom Klotz PetscReal osum = 0.0; /* Integral on last level */ 2385b3c0f97bSTom Klotz PetscReal psum = 0.0; /* Integral on the level before the last level */ 2386b3c0f97bSTom Klotz PetscReal sum; /* Integral on current level */ 2387446c295cSMatthew G. Knepley PetscReal yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 2388b3c0f97bSTom Klotz PetscReal lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 2389b3c0f97bSTom Klotz PetscReal wk; /* Quadrature weight at x_k */ 2390b3c0f97bSTom Klotz PetscReal lval, rval; /* Terms in the quadature sum to the left and right of 0 */ 2391b3c0f97bSTom Klotz PetscInt d; /* Digits of precision in the integral */ 2392b3c0f97bSTom Klotz 2393b3c0f97bSTom Klotz PetscFunctionBegin; 239408401ef6SPierre Jolivet PetscCheck(digits > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 23952b6f951bSStefano Zampini PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 2396b3c0f97bSTom Klotz /* Center term */ 2397d6685f55SMatthew G. Knepley func(&beta, ctx, &lval); 2398b3c0f97bSTom Klotz sum = 0.5 * alpha * PETSC_PI * lval; 2399b3c0f97bSTom Klotz /* */ 2400b3c0f97bSTom Klotz do { 2401b3c0f97bSTom Klotz PetscReal lterm, rterm, maxTerm = 0.0, d1, d2, d3, d4; 2402b3c0f97bSTom Klotz PetscInt k = 1; 2403b3c0f97bSTom Klotz 2404b3c0f97bSTom Klotz ++l; 240563a3b9bcSJacob Faibussowitsch /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %" PetscInt_FMT " sum: %15.15f\n", l, sum); */ 2406b3c0f97bSTom Klotz /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 2407b3c0f97bSTom Klotz psum = osum; 2408b3c0f97bSTom Klotz osum = sum; 2409b3c0f97bSTom Klotz h *= 0.5; 2410b3c0f97bSTom Klotz sum *= 0.5; 2411b3c0f97bSTom Klotz do { 24129add2064SThomas Klotz wk = 0.5 * h * PETSC_PI * PetscCoshReal(k * h) / PetscSqr(PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 2413446c295cSMatthew G. Knepley yk = 1.0 / (PetscExpReal(0.5 * PETSC_PI * PetscSinhReal(k * h)) * PetscCoshReal(0.5 * PETSC_PI * PetscSinhReal(k * h))); 2414446c295cSMatthew G. Knepley lx = -alpha * (1.0 - yk) + beta; 2415446c295cSMatthew G. Knepley rx = alpha * (1.0 - yk) + beta; 2416d6685f55SMatthew G. Knepley func(&lx, ctx, &lval); 2417d6685f55SMatthew G. Knepley func(&rx, ctx, &rval); 2418b3c0f97bSTom Klotz lterm = alpha * wk * lval; 2419b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(lterm), maxTerm); 2420b3c0f97bSTom Klotz sum += lterm; 2421b3c0f97bSTom Klotz rterm = alpha * wk * rval; 2422b3c0f97bSTom Klotz maxTerm = PetscMax(PetscAbsReal(rterm), maxTerm); 2423b3c0f97bSTom Klotz sum += rterm; 2424b3c0f97bSTom Klotz ++k; 2425b3c0f97bSTom Klotz /* Only need to evaluate every other point on refined levels */ 2426b3c0f97bSTom Klotz if (l != 1) ++k; 24279add2064SThomas Klotz } while (PetscAbsReal(PetscLog10Real(wk)) < p); /* Only need to evaluate sum until weights are < 32 digits of precision */ 2428b3c0f97bSTom Klotz 2429b3c0f97bSTom Klotz d1 = PetscLog10Real(PetscAbsReal(sum - osum)); 2430b3c0f97bSTom Klotz d2 = PetscLog10Real(PetscAbsReal(sum - psum)); 2431b3c0f97bSTom Klotz d3 = PetscLog10Real(maxTerm) - p; 243209d48545SBarry Smith if (PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm)) == 0.0) d4 = 0.0; 243309d48545SBarry Smith else d4 = PetscLog10Real(PetscMax(PetscAbsReal(lterm), PetscAbsReal(rterm))); 2434b3c0f97bSTom Klotz d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1) / d2, 2 * d1), d3), d4))); 24359add2064SThomas Klotz } while (d < digits && l < 12); 2436b3c0f97bSTom Klotz *sol = sum; 24372b6f951bSStefano Zampini PetscCall(PetscFPTrapPop()); 24383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2439b3c0f97bSTom Klotz } 2440b3c0f97bSTom Klotz 2441497880caSRichard Tran Mills #if defined(PETSC_HAVE_MPFR) 2442d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2443d71ae5a4SJacob Faibussowitsch { 2444e510cb1fSThomas Klotz const PetscInt safetyFactor = 2; /* Calculate abcissa until 2*p digits */ 244529f144ccSMatthew G. Knepley PetscInt l = 0; /* Level of refinement, h = 2^{-l} */ 244629f144ccSMatthew G. Knepley mpfr_t alpha; /* Half-width of the integration interval */ 244729f144ccSMatthew G. Knepley mpfr_t beta; /* Center of the integration interval */ 244829f144ccSMatthew G. Knepley mpfr_t h; /* Step size, length between x_k */ 244929f144ccSMatthew G. Knepley mpfr_t osum; /* Integral on last level */ 245029f144ccSMatthew G. Knepley mpfr_t psum; /* Integral on the level before the last level */ 245129f144ccSMatthew G. Knepley mpfr_t sum; /* Integral on current level */ 245229f144ccSMatthew G. Knepley mpfr_t yk; /* Quadrature point 1 - x_k on reference domain [-1, 1] */ 245329f144ccSMatthew G. Knepley mpfr_t lx, rx; /* Quadrature points to the left and right of 0 on the real domain [a, b] */ 245429f144ccSMatthew G. Knepley mpfr_t wk; /* Quadrature weight at x_k */ 24551fbc92bbSMatthew G. Knepley PetscReal lval, rval, rtmp; /* Terms in the quadature sum to the left and right of 0 */ 245629f144ccSMatthew G. Knepley PetscInt d; /* Digits of precision in the integral */ 245729f144ccSMatthew G. Knepley mpfr_t pi2, kh, msinh, mcosh, maxTerm, curTerm, tmp; 245829f144ccSMatthew G. Knepley 245929f144ccSMatthew G. Knepley PetscFunctionBegin; 246008401ef6SPierre Jolivet PetscCheck(digits > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must give a positive number of significant digits"); 246129f144ccSMatthew G. Knepley /* Create high precision storage */ 2462c9f744b5SMatthew 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); 246329f144ccSMatthew G. Knepley /* Initialization */ 246429f144ccSMatthew G. Knepley mpfr_set_d(alpha, 0.5 * (b - a), MPFR_RNDN); 246529f144ccSMatthew G. Knepley mpfr_set_d(beta, 0.5 * (b + a), MPFR_RNDN); 246629f144ccSMatthew G. Knepley mpfr_set_d(osum, 0.0, MPFR_RNDN); 246729f144ccSMatthew G. Knepley mpfr_set_d(psum, 0.0, MPFR_RNDN); 246829f144ccSMatthew G. Knepley mpfr_set_d(h, 1.0, MPFR_RNDN); 246929f144ccSMatthew G. Knepley mpfr_const_pi(pi2, MPFR_RNDN); 247029f144ccSMatthew G. Knepley mpfr_mul_d(pi2, pi2, 0.5, MPFR_RNDN); 247129f144ccSMatthew G. Knepley /* Center term */ 24721fbc92bbSMatthew G. Knepley rtmp = 0.5 * (b + a); 24731fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &lval); 247429f144ccSMatthew G. Knepley mpfr_set(sum, pi2, MPFR_RNDN); 247529f144ccSMatthew G. Knepley mpfr_mul(sum, sum, alpha, MPFR_RNDN); 247629f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, lval, MPFR_RNDN); 247729f144ccSMatthew G. Knepley /* */ 247829f144ccSMatthew G. Knepley do { 247929f144ccSMatthew G. Knepley PetscReal d1, d2, d3, d4; 248029f144ccSMatthew G. Knepley PetscInt k = 1; 248129f144ccSMatthew G. Knepley 248229f144ccSMatthew G. Knepley ++l; 248329f144ccSMatthew G. Knepley mpfr_set_d(maxTerm, 0.0, MPFR_RNDN); 248463a3b9bcSJacob Faibussowitsch /* PetscPrintf(PETSC_COMM_SELF, "LEVEL %" PetscInt_FMT " sum: %15.15f\n", l, sum); */ 248529f144ccSMatthew G. Knepley /* At each level of refinement, h --> h/2 and sum --> sum/2 */ 248629f144ccSMatthew G. Knepley mpfr_set(psum, osum, MPFR_RNDN); 248729f144ccSMatthew G. Knepley mpfr_set(osum, sum, MPFR_RNDN); 248829f144ccSMatthew G. Knepley mpfr_mul_d(h, h, 0.5, MPFR_RNDN); 248929f144ccSMatthew G. Knepley mpfr_mul_d(sum, sum, 0.5, MPFR_RNDN); 249029f144ccSMatthew G. Knepley do { 249129f144ccSMatthew G. Knepley mpfr_set_si(kh, k, MPFR_RNDN); 249229f144ccSMatthew G. Knepley mpfr_mul(kh, kh, h, MPFR_RNDN); 249329f144ccSMatthew G. Knepley /* Weight */ 249429f144ccSMatthew G. Knepley mpfr_set(wk, h, MPFR_RNDN); 249529f144ccSMatthew G. Knepley mpfr_sinh_cosh(msinh, mcosh, kh, MPFR_RNDN); 249629f144ccSMatthew G. Knepley mpfr_mul(msinh, msinh, pi2, MPFR_RNDN); 249729f144ccSMatthew G. Knepley mpfr_mul(mcosh, mcosh, pi2, MPFR_RNDN); 249829f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 249929f144ccSMatthew G. Knepley mpfr_sqr(tmp, tmp, MPFR_RNDN); 250029f144ccSMatthew G. Knepley mpfr_mul(wk, wk, mcosh, MPFR_RNDN); 250129f144ccSMatthew G. Knepley mpfr_div(wk, wk, tmp, MPFR_RNDN); 250229f144ccSMatthew G. Knepley /* Abscissa */ 250329f144ccSMatthew G. Knepley mpfr_set_d(yk, 1.0, MPFR_RNDZ); 250429f144ccSMatthew G. Knepley mpfr_cosh(tmp, msinh, MPFR_RNDN); 250529f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 250629f144ccSMatthew G. Knepley mpfr_exp(tmp, msinh, MPFR_RNDN); 250729f144ccSMatthew G. Knepley mpfr_div(yk, yk, tmp, MPFR_RNDZ); 250829f144ccSMatthew G. Knepley /* Quadrature points */ 250929f144ccSMatthew G. Knepley mpfr_sub_d(lx, yk, 1.0, MPFR_RNDZ); 251029f144ccSMatthew G. Knepley mpfr_mul(lx, lx, alpha, MPFR_RNDU); 251129f144ccSMatthew G. Knepley mpfr_add(lx, lx, beta, MPFR_RNDU); 251229f144ccSMatthew G. Knepley mpfr_d_sub(rx, 1.0, yk, MPFR_RNDZ); 251329f144ccSMatthew G. Knepley mpfr_mul(rx, rx, alpha, MPFR_RNDD); 251429f144ccSMatthew G. Knepley mpfr_add(rx, rx, beta, MPFR_RNDD); 251529f144ccSMatthew G. Knepley /* Evaluation */ 25161fbc92bbSMatthew G. Knepley rtmp = mpfr_get_d(lx, MPFR_RNDU); 25171fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &lval); 25181fbc92bbSMatthew G. Knepley rtmp = mpfr_get_d(rx, MPFR_RNDD); 25191fbc92bbSMatthew G. Knepley func(&rtmp, ctx, &rval); 252029f144ccSMatthew G. Knepley /* Update */ 252129f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 252229f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, lval, MPFR_RNDN); 252329f144ccSMatthew G. Knepley mpfr_add(sum, sum, tmp, MPFR_RNDN); 252429f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 252529f144ccSMatthew G. Knepley mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN); 252629f144ccSMatthew G. Knepley mpfr_set(curTerm, tmp, MPFR_RNDN); 252729f144ccSMatthew G. Knepley mpfr_mul(tmp, wk, alpha, MPFR_RNDN); 252829f144ccSMatthew G. Knepley mpfr_mul_d(tmp, tmp, rval, MPFR_RNDN); 252929f144ccSMatthew G. Knepley mpfr_add(sum, sum, tmp, MPFR_RNDN); 253029f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 253129f144ccSMatthew G. Knepley mpfr_max(maxTerm, maxTerm, tmp, MPFR_RNDN); 253229f144ccSMatthew G. Knepley mpfr_max(curTerm, curTerm, tmp, MPFR_RNDN); 253329f144ccSMatthew G. Knepley ++k; 253429f144ccSMatthew G. Knepley /* Only need to evaluate every other point on refined levels */ 253529f144ccSMatthew G. Knepley if (l != 1) ++k; 253629f144ccSMatthew G. Knepley mpfr_log10(tmp, wk, MPFR_RNDN); 253729f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 2538c9f744b5SMatthew G. Knepley } while (mpfr_get_d(tmp, MPFR_RNDN) < safetyFactor * digits); /* Only need to evaluate sum until weights are < 32 digits of precision */ 253929f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, osum, MPFR_RNDN); 254029f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 254129f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 254229f144ccSMatthew G. Knepley d1 = mpfr_get_d(tmp, MPFR_RNDN); 254329f144ccSMatthew G. Knepley mpfr_sub(tmp, sum, psum, MPFR_RNDN); 254429f144ccSMatthew G. Knepley mpfr_abs(tmp, tmp, MPFR_RNDN); 254529f144ccSMatthew G. Knepley mpfr_log10(tmp, tmp, MPFR_RNDN); 254629f144ccSMatthew G. Knepley d2 = mpfr_get_d(tmp, MPFR_RNDN); 254729f144ccSMatthew G. Knepley mpfr_log10(tmp, maxTerm, MPFR_RNDN); 2548c9f744b5SMatthew G. Knepley d3 = mpfr_get_d(tmp, MPFR_RNDN) - digits; 254929f144ccSMatthew G. Knepley mpfr_log10(tmp, curTerm, MPFR_RNDN); 255029f144ccSMatthew G. Knepley d4 = mpfr_get_d(tmp, MPFR_RNDN); 255129f144ccSMatthew G. Knepley d = PetscAbsInt(PetscMin(0, PetscMax(PetscMax(PetscMax(PetscSqr(d1) / d2, 2 * d1), d3), d4))); 2552b0649871SThomas Klotz } while (d < digits && l < 8); 255329f144ccSMatthew G. Knepley *sol = mpfr_get_d(sum, MPFR_RNDN); 255429f144ccSMatthew G. Knepley /* Cleanup */ 255529f144ccSMatthew G. Knepley mpfr_clears(alpha, beta, h, sum, osum, psum, yk, wk, lx, rx, tmp, maxTerm, curTerm, pi2, kh, msinh, mcosh, NULL); 25563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 255729f144ccSMatthew G. Knepley } 2558d525116cSMatthew G. Knepley #else 2559fbfcfee5SBarry Smith 2560d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTanhSinhIntegrateMPFR(void (*func)(const PetscReal[], void *, PetscReal *), PetscReal a, PetscReal b, PetscInt digits, void *ctx, PetscReal *sol) 2561d71ae5a4SJacob Faibussowitsch { 2562d525116cSMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "This method will not work without MPFR. Reconfigure using --download-mpfr --download-gmp"); 2563d525116cSMatthew G. Knepley } 256429f144ccSMatthew G. Knepley #endif 256529f144ccSMatthew G. Knepley 25662df84da0SMatthew G. Knepley /*@ 25672df84da0SMatthew G. Knepley PetscDTTensorQuadratureCreate - create the tensor product quadrature from two lower-dimensional quadratures 25682df84da0SMatthew G. Knepley 25692df84da0SMatthew G. Knepley Not Collective 25702df84da0SMatthew G. Knepley 25712df84da0SMatthew G. Knepley Input Parameters: 25722df84da0SMatthew G. Knepley + q1 - The first quadrature 25732df84da0SMatthew G. Knepley - q2 - The second quadrature 25742df84da0SMatthew G. Knepley 25752df84da0SMatthew G. Knepley Output Parameter: 2576dce8aebaSBarry Smith . q - A `PetscQuadrature` object 25772df84da0SMatthew G. Knepley 25782df84da0SMatthew G. Knepley Level: intermediate 25792df84da0SMatthew G. Knepley 2580dce8aebaSBarry Smith .seealso: `PetscQuadrature`, `PetscDTGaussTensorQuadrature()` 25812df84da0SMatthew G. Knepley @*/ 2582d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTTensorQuadratureCreate(PetscQuadrature q1, PetscQuadrature q2, PetscQuadrature *q) 2583d71ae5a4SJacob Faibussowitsch { 25844366bac7SMatthew G. Knepley DMPolytopeType ct1, ct2, ct; 25852df84da0SMatthew G. Knepley const PetscReal *x1, *w1, *x2, *w2; 25862df84da0SMatthew G. Knepley PetscReal *x, *w; 25872df84da0SMatthew G. Knepley PetscInt dim1, Nc1, Np1, order1, qa, d1; 25882df84da0SMatthew G. Knepley PetscInt dim2, Nc2, Np2, order2, qb, d2; 25892df84da0SMatthew G. Knepley PetscInt dim, Nc, Np, order, qc, d; 25902df84da0SMatthew G. Knepley 25912df84da0SMatthew G. Knepley PetscFunctionBegin; 25922df84da0SMatthew G. Knepley PetscValidHeaderSpecific(q1, PETSCQUADRATURE_CLASSID, 1); 25932df84da0SMatthew G. Knepley PetscValidHeaderSpecific(q2, PETSCQUADRATURE_CLASSID, 2); 25944f572ea9SToby Isaac PetscAssertPointer(q, 3); 25959566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q1, &order1)); 25969566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetOrder(q2, &order2)); 25972df84da0SMatthew G. Knepley PetscCheck(order1 == order2, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Order1 %" PetscInt_FMT " != %" PetscInt_FMT " Order2", order1, order2); 25989566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q1, &dim1, &Nc1, &Np1, &x1, &w1)); 25994366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetCellType(q1, &ct1)); 26009566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q2, &dim2, &Nc2, &Np2, &x2, &w2)); 26014366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetCellType(q2, &ct2)); 26022df84da0SMatthew G. Knepley PetscCheck(Nc1 == Nc2, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "NumComp1 %" PetscInt_FMT " != %" PetscInt_FMT " NumComp2", Nc1, Nc2); 26032df84da0SMatthew G. Knepley 26044366bac7SMatthew G. Knepley switch (ct1) { 26054366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 26064366bac7SMatthew G. Knepley ct = ct2; 26074366bac7SMatthew G. Knepley break; 26084366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 26094366bac7SMatthew G. Knepley switch (ct2) { 26104366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 26114366bac7SMatthew G. Knepley ct = ct1; 26124366bac7SMatthew G. Knepley break; 26134366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 26144366bac7SMatthew G. Knepley ct = DM_POLYTOPE_QUADRILATERAL; 26154366bac7SMatthew G. Knepley break; 26164366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 26174366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TRI_PRISM; 26184366bac7SMatthew G. Knepley break; 26194366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 26204366bac7SMatthew G. Knepley ct = DM_POLYTOPE_HEXAHEDRON; 26214366bac7SMatthew G. Knepley break; 26224366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 26234366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26244366bac7SMatthew G. Knepley break; 26254366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 26264366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26274366bac7SMatthew G. Knepley break; 26284366bac7SMatthew G. Knepley default: 26294366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26304366bac7SMatthew G. Knepley } 26314366bac7SMatthew G. Knepley break; 26324366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 26334366bac7SMatthew G. Knepley switch (ct2) { 26344366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 26354366bac7SMatthew G. Knepley ct = ct1; 26364366bac7SMatthew G. Knepley break; 26374366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 26384366bac7SMatthew G. Knepley ct = DM_POLYTOPE_TRI_PRISM; 26394366bac7SMatthew G. Knepley break; 26404366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 26414366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26424366bac7SMatthew G. Knepley break; 26434366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 26444366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26454366bac7SMatthew G. Knepley break; 26464366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 26474366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26484366bac7SMatthew G. Knepley break; 26494366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 26504366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26514366bac7SMatthew G. Knepley break; 26524366bac7SMatthew G. Knepley default: 26534366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26544366bac7SMatthew G. Knepley } 26554366bac7SMatthew G. Knepley break; 26564366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 26574366bac7SMatthew G. Knepley switch (ct2) { 26584366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 26594366bac7SMatthew G. Knepley ct = ct1; 26604366bac7SMatthew G. Knepley break; 26614366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 26624366bac7SMatthew G. Knepley ct = DM_POLYTOPE_HEXAHEDRON; 26634366bac7SMatthew G. Knepley break; 26644366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 26654366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26664366bac7SMatthew G. Knepley break; 26674366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 26684366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26694366bac7SMatthew G. Knepley break; 26704366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 26714366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26724366bac7SMatthew G. Knepley break; 26734366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 26744366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26754366bac7SMatthew G. Knepley break; 26764366bac7SMatthew G. Knepley default: 26774366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26784366bac7SMatthew G. Knepley } 26794366bac7SMatthew G. Knepley break; 26804366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 26814366bac7SMatthew G. Knepley switch (ct2) { 26824366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 26834366bac7SMatthew G. Knepley ct = ct1; 26844366bac7SMatthew G. Knepley break; 26854366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 26864366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26874366bac7SMatthew G. Knepley break; 26884366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 26894366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26904366bac7SMatthew G. Knepley break; 26914366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 26924366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26934366bac7SMatthew G. Knepley break; 26944366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 26954366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26964366bac7SMatthew G. Knepley break; 26974366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 26984366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 26994366bac7SMatthew G. Knepley break; 27004366bac7SMatthew G. Knepley default: 27014366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27024366bac7SMatthew G. Knepley } 27034366bac7SMatthew G. Knepley break; 27044366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 27054366bac7SMatthew G. Knepley switch (ct2) { 27064366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT: 27074366bac7SMatthew G. Knepley ct = ct1; 27084366bac7SMatthew G. Knepley break; 27094366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 27104366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27114366bac7SMatthew G. Knepley break; 27124366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 27134366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27144366bac7SMatthew G. Knepley break; 27154366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 27164366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27174366bac7SMatthew G. Knepley break; 27184366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 27194366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27204366bac7SMatthew G. Knepley break; 27214366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 27224366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27234366bac7SMatthew G. Knepley break; 27244366bac7SMatthew G. Knepley default: 27254366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27264366bac7SMatthew G. Knepley } 27274366bac7SMatthew G. Knepley break; 27284366bac7SMatthew G. Knepley default: 27294366bac7SMatthew G. Knepley ct = DM_POLYTOPE_UNKNOWN; 27304366bac7SMatthew G. Knepley } 27312df84da0SMatthew G. Knepley dim = dim1 + dim2; 27322df84da0SMatthew G. Knepley Nc = Nc1; 27332df84da0SMatthew G. Knepley Np = Np1 * Np2; 27342df84da0SMatthew G. Knepley order = order1; 27359566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF, q)); 27364366bac7SMatthew G. Knepley PetscCall(PetscQuadratureSetCellType(*q, ct)); 27379566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetOrder(*q, order)); 27389566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Np * dim, &x)); 27399566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(Np, &w)); 27402df84da0SMatthew G. Knepley for (qa = 0, qc = 0; qa < Np1; ++qa) { 27412df84da0SMatthew G. Knepley for (qb = 0; qb < Np2; ++qb, ++qc) { 2742ad540459SPierre Jolivet for (d1 = 0, d = 0; d1 < dim1; ++d1, ++d) x[qc * dim + d] = x1[qa * dim1 + d1]; 2743ad540459SPierre Jolivet for (d2 = 0; d2 < dim2; ++d2, ++d) x[qc * dim + d] = x2[qb * dim2 + d2]; 27442df84da0SMatthew G. Knepley w[qc] = w1[qa] * w2[qb]; 27452df84da0SMatthew G. Knepley } 27462df84da0SMatthew G. Knepley } 27479566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*q, dim, Nc, Np, x, w)); 27483ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 27492df84da0SMatthew G. Knepley } 27502df84da0SMatthew G. Knepley 2751194825f6SJed Brown /* Overwrites A. Can only handle full-rank problems with m>=n 2752dce8aebaSBarry Smith A in column-major format 2753dce8aebaSBarry Smith Ainv in row-major format 2754dce8aebaSBarry Smith tau has length m 2755dce8aebaSBarry Smith worksize must be >= max(1,n) 2756194825f6SJed Brown */ 2757d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTPseudoInverseQR(PetscInt m, PetscInt mstride, PetscInt n, PetscReal *A_in, PetscReal *Ainv_out, PetscScalar *tau, PetscInt worksize, PetscScalar *work) 2758d71ae5a4SJacob Faibussowitsch { 2759194825f6SJed Brown PetscBLASInt M, N, K, lda, ldb, ldwork, info; 2760194825f6SJed Brown PetscScalar *A, *Ainv, *R, *Q, Alpha; 2761194825f6SJed Brown 2762194825f6SJed Brown PetscFunctionBegin; 2763194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 2764194825f6SJed Brown { 2765194825f6SJed Brown PetscInt i, j; 27669566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(m * n, &A, m * n, &Ainv)); 2767194825f6SJed Brown for (j = 0; j < n; j++) { 2768194825f6SJed Brown for (i = 0; i < m; i++) A[i + m * j] = A_in[i + mstride * j]; 2769194825f6SJed Brown } 2770194825f6SJed Brown mstride = m; 2771194825f6SJed Brown } 2772194825f6SJed Brown #else 2773194825f6SJed Brown A = A_in; 2774194825f6SJed Brown Ainv = Ainv_out; 2775194825f6SJed Brown #endif 2776194825f6SJed Brown 27779566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(m, &M)); 27789566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &N)); 27799566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(mstride, &lda)); 27809566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(worksize, &ldwork)); 27819566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 2782792fecdfSBarry Smith PetscCallBLAS("LAPACKgeqrf", LAPACKgeqrf_(&M, &N, A, &lda, tau, work, &ldwork, &info)); 27839566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 278428b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "xGEQRF error"); 2785194825f6SJed Brown R = A; /* Upper triangular part of A now contains R, the rest contains the elementary reflectors */ 2786194825f6SJed Brown 2787194825f6SJed Brown /* Extract an explicit representation of Q */ 2788194825f6SJed Brown Q = Ainv; 27899566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(Q, A, mstride * n)); 2790194825f6SJed Brown K = N; /* full rank */ 2791792fecdfSBarry Smith PetscCallBLAS("LAPACKorgqr", LAPACKorgqr_(&M, &N, &K, Q, &lda, tau, work, &ldwork, &info)); 279228b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "xORGQR/xUNGQR error"); 2793194825f6SJed Brown 2794194825f6SJed Brown /* Compute A^{-T} = (R^{-1} Q^T)^T = Q R^{-T} */ 2795194825f6SJed Brown Alpha = 1.0; 2796194825f6SJed Brown ldb = lda; 2797792fecdfSBarry Smith PetscCallBLAS("BLAStrsm", BLAStrsm_("Right", "Upper", "ConjugateTranspose", "NotUnitTriangular", &M, &N, &Alpha, R, &lda, Q, &ldb)); 2798194825f6SJed Brown /* Ainv is Q, overwritten with inverse */ 2799194825f6SJed Brown 2800194825f6SJed Brown #if defined(PETSC_USE_COMPLEX) 2801194825f6SJed Brown { 2802194825f6SJed Brown PetscInt i; 2803194825f6SJed Brown for (i = 0; i < m * n; i++) Ainv_out[i] = PetscRealPart(Ainv[i]); 28049566063dSJacob Faibussowitsch PetscCall(PetscFree2(A, Ainv)); 2805194825f6SJed Brown } 2806194825f6SJed Brown #endif 28073ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2808194825f6SJed Brown } 2809194825f6SJed Brown 2810194825f6SJed Brown /* Computes integral of L_p' over intervals {(x0,x1),(x1,x2),...} */ 2811d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscDTLegendreIntegrate(PetscInt ninterval, const PetscReal *x, PetscInt ndegree, const PetscInt *degrees, PetscBool Transpose, PetscReal *B) 2812d71ae5a4SJacob Faibussowitsch { 2813194825f6SJed Brown PetscReal *Bv; 2814194825f6SJed Brown PetscInt i, j; 2815194825f6SJed Brown 2816194825f6SJed Brown PetscFunctionBegin; 28179566063dSJacob Faibussowitsch PetscCall(PetscMalloc1((ninterval + 1) * ndegree, &Bv)); 2818194825f6SJed Brown /* Point evaluation of L_p on all the source vertices */ 28199566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreEval(ninterval + 1, x, ndegree, degrees, Bv, NULL, NULL)); 2820194825f6SJed Brown /* Integral over each interval: \int_a^b L_p' = L_p(b)-L_p(a) */ 2821194825f6SJed Brown for (i = 0; i < ninterval; i++) { 2822194825f6SJed Brown for (j = 0; j < ndegree; j++) { 2823194825f6SJed Brown if (Transpose) B[i + ninterval * j] = Bv[(i + 1) * ndegree + j] - Bv[i * ndegree + j]; 2824194825f6SJed Brown else B[i * ndegree + j] = Bv[(i + 1) * ndegree + j] - Bv[i * ndegree + j]; 2825194825f6SJed Brown } 2826194825f6SJed Brown } 28279566063dSJacob Faibussowitsch PetscCall(PetscFree(Bv)); 28283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2829194825f6SJed Brown } 2830194825f6SJed Brown 2831194825f6SJed Brown /*@ 2832194825f6SJed Brown PetscDTReconstructPoly - create matrix representing polynomial reconstruction using cell intervals and evaluation at target intervals 2833194825f6SJed Brown 2834194825f6SJed Brown Not Collective 2835194825f6SJed Brown 28364165533cSJose E. Roman Input Parameters: 2837194825f6SJed Brown + degree - degree of reconstruction polynomial 2838194825f6SJed Brown . nsource - number of source intervals 2839*1d27aa22SBarry Smith . sourcex - sorted coordinates of source cell boundaries (length `nsource`+1) 2840194825f6SJed Brown . ntarget - number of target intervals 2841*1d27aa22SBarry Smith - targetx - sorted coordinates of target cell boundaries (length `ntarget`+1) 2842194825f6SJed Brown 28434165533cSJose E. Roman Output Parameter: 2844194825f6SJed Brown . R - reconstruction matrix, utarget = sum_s R[t*nsource+s] * usource[s] 2845194825f6SJed Brown 2846194825f6SJed Brown Level: advanced 2847194825f6SJed Brown 2848db781477SPatrick Sanan .seealso: `PetscDTLegendreEval()` 2849194825f6SJed Brown @*/ 2850d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTReconstructPoly(PetscInt degree, PetscInt nsource, const PetscReal *sourcex, PetscInt ntarget, const PetscReal *targetx, PetscReal *R) 2851d71ae5a4SJacob Faibussowitsch { 2852194825f6SJed Brown PetscInt i, j, k, *bdegrees, worksize; 2853194825f6SJed Brown PetscReal xmin, xmax, center, hscale, *sourcey, *targety, *Bsource, *Bsinv, *Btarget; 2854194825f6SJed Brown PetscScalar *tau, *work; 2855194825f6SJed Brown 2856194825f6SJed Brown PetscFunctionBegin; 28574f572ea9SToby Isaac PetscAssertPointer(sourcex, 3); 28584f572ea9SToby Isaac PetscAssertPointer(targetx, 5); 28594f572ea9SToby Isaac PetscAssertPointer(R, 6); 286063a3b9bcSJacob 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); 286176bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2862ad540459SPierre 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]); 2863ad540459SPierre 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]); 286476bd3646SJed Brown } 2865194825f6SJed Brown xmin = PetscMin(sourcex[0], targetx[0]); 2866194825f6SJed Brown xmax = PetscMax(sourcex[nsource], targetx[ntarget]); 2867194825f6SJed Brown center = (xmin + xmax) / 2; 2868194825f6SJed Brown hscale = (xmax - xmin) / 2; 2869194825f6SJed Brown worksize = nsource; 28709566063dSJacob Faibussowitsch PetscCall(PetscMalloc4(degree + 1, &bdegrees, nsource + 1, &sourcey, nsource * (degree + 1), &Bsource, worksize, &work)); 28719566063dSJacob Faibussowitsch PetscCall(PetscMalloc4(nsource, &tau, nsource * (degree + 1), &Bsinv, ntarget + 1, &targety, ntarget * (degree + 1), &Btarget)); 2872194825f6SJed Brown for (i = 0; i <= nsource; i++) sourcey[i] = (sourcex[i] - center) / hscale; 2873194825f6SJed Brown for (i = 0; i <= degree; i++) bdegrees[i] = i + 1; 28749566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreIntegrate(nsource, sourcey, degree + 1, bdegrees, PETSC_TRUE, Bsource)); 28759566063dSJacob Faibussowitsch PetscCall(PetscDTPseudoInverseQR(nsource, nsource, degree + 1, Bsource, Bsinv, tau, nsource, work)); 2876194825f6SJed Brown for (i = 0; i <= ntarget; i++) targety[i] = (targetx[i] - center) / hscale; 28779566063dSJacob Faibussowitsch PetscCall(PetscDTLegendreIntegrate(ntarget, targety, degree + 1, bdegrees, PETSC_FALSE, Btarget)); 2878194825f6SJed Brown for (i = 0; i < ntarget; i++) { 2879194825f6SJed Brown PetscReal rowsum = 0; 2880194825f6SJed Brown for (j = 0; j < nsource; j++) { 2881194825f6SJed Brown PetscReal sum = 0; 2882ad540459SPierre Jolivet for (k = 0; k < degree + 1; k++) sum += Btarget[i * (degree + 1) + k] * Bsinv[k * nsource + j]; 2883194825f6SJed Brown R[i * nsource + j] = sum; 2884194825f6SJed Brown rowsum += sum; 2885194825f6SJed Brown } 2886194825f6SJed Brown for (j = 0; j < nsource; j++) R[i * nsource + j] /= rowsum; /* normalize each row */ 2887194825f6SJed Brown } 28889566063dSJacob Faibussowitsch PetscCall(PetscFree4(bdegrees, sourcey, Bsource, work)); 28899566063dSJacob Faibussowitsch PetscCall(PetscFree4(tau, Bsinv, targety, Btarget)); 28903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2891194825f6SJed Brown } 2892916e780bShannah_mairs 2893916e780bShannah_mairs /*@C 2894916e780bShannah_mairs PetscGaussLobattoLegendreIntegrate - Compute the L2 integral of a function on the GLL points 2895916e780bShannah_mairs 2896916e780bShannah_mairs Not Collective 2897916e780bShannah_mairs 2898d8d19677SJose E. Roman Input Parameters: 2899916e780bShannah_mairs + n - the number of GLL nodes 2900916e780bShannah_mairs . nodes - the GLL nodes 2901916e780bShannah_mairs . weights - the GLL weights 2902f0fc11ceSJed Brown - f - the function values at the nodes 2903916e780bShannah_mairs 2904916e780bShannah_mairs Output Parameter: 2905916e780bShannah_mairs . in - the value of the integral 2906916e780bShannah_mairs 2907916e780bShannah_mairs Level: beginner 2908916e780bShannah_mairs 2909db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()` 2910916e780bShannah_mairs @*/ 2911d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreIntegrate(PetscInt n, PetscReal *nodes, PetscReal *weights, const PetscReal *f, PetscReal *in) 2912d71ae5a4SJacob Faibussowitsch { 2913916e780bShannah_mairs PetscInt i; 2914916e780bShannah_mairs 2915916e780bShannah_mairs PetscFunctionBegin; 2916916e780bShannah_mairs *in = 0.; 2917ad540459SPierre Jolivet for (i = 0; i < n; i++) *in += f[i] * f[i] * weights[i]; 29183ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2919916e780bShannah_mairs } 2920916e780bShannah_mairs 2921916e780bShannah_mairs /*@C 2922916e780bShannah_mairs PetscGaussLobattoLegendreElementLaplacianCreate - computes the Laplacian for a single 1d GLL element 2923916e780bShannah_mairs 2924916e780bShannah_mairs Not Collective 2925916e780bShannah_mairs 2926d8d19677SJose E. Roman Input Parameters: 2927916e780bShannah_mairs + n - the number of GLL nodes 2928916e780bShannah_mairs . nodes - the GLL nodes 2929f0fc11ceSJed Brown - weights - the GLL weights 2930916e780bShannah_mairs 2931916e780bShannah_mairs Output Parameter: 293260225df5SJacob Faibussowitsch . AA - the stiffness element 2933916e780bShannah_mairs 2934916e780bShannah_mairs Level: beginner 2935916e780bShannah_mairs 2936916e780bShannah_mairs Notes: 2937dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementLaplacianDestroy()` 2938916e780bShannah_mairs 2939916e780bShannah_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) 2940916e780bShannah_mairs 2941db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianDestroy()` 2942916e780bShannah_mairs @*/ 2943d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementLaplacianCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 2944d71ae5a4SJacob Faibussowitsch { 2945916e780bShannah_mairs PetscReal **A; 2946916e780bShannah_mairs const PetscReal *gllnodes = nodes; 2947916e780bShannah_mairs const PetscInt p = n - 1; 2948916e780bShannah_mairs PetscReal z0, z1, z2 = -1, x, Lpj, Lpr; 2949916e780bShannah_mairs PetscInt i, j, nn, r; 2950916e780bShannah_mairs 2951916e780bShannah_mairs PetscFunctionBegin; 29529566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &A)); 29539566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &A[0])); 2954916e780bShannah_mairs for (i = 1; i < n; i++) A[i] = A[i - 1] + n; 2955916e780bShannah_mairs 2956916e780bShannah_mairs for (j = 1; j < p; j++) { 2957916e780bShannah_mairs x = gllnodes[j]; 2958916e780bShannah_mairs z0 = 1.; 2959916e780bShannah_mairs z1 = x; 2960916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2961916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2962916e780bShannah_mairs z0 = z1; 2963916e780bShannah_mairs z1 = z2; 2964916e780bShannah_mairs } 2965916e780bShannah_mairs Lpj = z2; 2966916e780bShannah_mairs for (r = 1; r < p; r++) { 2967916e780bShannah_mairs if (r == j) { 2968916e780bShannah_mairs A[j][j] = 2. / (3. * (1. - gllnodes[j] * gllnodes[j]) * Lpj * Lpj); 2969916e780bShannah_mairs } else { 2970916e780bShannah_mairs x = gllnodes[r]; 2971916e780bShannah_mairs z0 = 1.; 2972916e780bShannah_mairs z1 = x; 2973916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2974916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2975916e780bShannah_mairs z0 = z1; 2976916e780bShannah_mairs z1 = z2; 2977916e780bShannah_mairs } 2978916e780bShannah_mairs Lpr = z2; 2979916e780bShannah_mairs A[r][j] = 4. / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * Lpr * (gllnodes[j] - gllnodes[r]) * (gllnodes[j] - gllnodes[r])); 2980916e780bShannah_mairs } 2981916e780bShannah_mairs } 2982916e780bShannah_mairs } 2983916e780bShannah_mairs for (j = 1; j < p + 1; j++) { 2984916e780bShannah_mairs x = gllnodes[j]; 2985916e780bShannah_mairs z0 = 1.; 2986916e780bShannah_mairs z1 = x; 2987916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 2988916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 2989916e780bShannah_mairs z0 = z1; 2990916e780bShannah_mairs z1 = z2; 2991916e780bShannah_mairs } 2992916e780bShannah_mairs Lpj = z2; 2993916e780bShannah_mairs A[j][0] = 4. * PetscPowRealInt(-1., p) / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * (1. + gllnodes[j]) * (1. + gllnodes[j])); 2994916e780bShannah_mairs A[0][j] = A[j][0]; 2995916e780bShannah_mairs } 2996916e780bShannah_mairs for (j = 0; j < p; j++) { 2997916e780bShannah_mairs x = gllnodes[j]; 2998916e780bShannah_mairs z0 = 1.; 2999916e780bShannah_mairs z1 = x; 3000916e780bShannah_mairs for (nn = 1; nn < p; nn++) { 3001916e780bShannah_mairs z2 = x * z1 * (2. * ((PetscReal)nn) + 1.) / (((PetscReal)nn) + 1.) - z0 * (((PetscReal)nn) / (((PetscReal)nn) + 1.)); 3002916e780bShannah_mairs z0 = z1; 3003916e780bShannah_mairs z1 = z2; 3004916e780bShannah_mairs } 3005916e780bShannah_mairs Lpj = z2; 3006916e780bShannah_mairs 3007916e780bShannah_mairs A[p][j] = 4. / (((PetscReal)p) * (((PetscReal)p) + 1.) * Lpj * (1. - gllnodes[j]) * (1. - gllnodes[j])); 3008916e780bShannah_mairs A[j][p] = A[p][j]; 3009916e780bShannah_mairs } 3010916e780bShannah_mairs A[0][0] = 0.5 + (((PetscReal)p) * (((PetscReal)p) + 1.) - 2.) / 6.; 3011916e780bShannah_mairs A[p][p] = A[0][0]; 3012916e780bShannah_mairs *AA = A; 30133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3014916e780bShannah_mairs } 3015916e780bShannah_mairs 3016916e780bShannah_mairs /*@C 3017dce8aebaSBarry Smith PetscGaussLobattoLegendreElementLaplacianDestroy - frees the Laplacian for a single 1d GLL element created with `PetscGaussLobattoLegendreElementLaplacianCreate()` 3018916e780bShannah_mairs 3019916e780bShannah_mairs Not Collective 3020916e780bShannah_mairs 3021d8d19677SJose E. Roman Input Parameters: 3022916e780bShannah_mairs + n - the number of GLL nodes 3023916e780bShannah_mairs . nodes - the GLL nodes 3024916e780bShannah_mairs . weights - the GLL weightss 302560225df5SJacob Faibussowitsch - AA - the stiffness element 3026916e780bShannah_mairs 3027916e780bShannah_mairs Level: beginner 3028916e780bShannah_mairs 3029db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()` 3030916e780bShannah_mairs @*/ 3031d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementLaplacianDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 3032d71ae5a4SJacob Faibussowitsch { 3033916e780bShannah_mairs PetscFunctionBegin; 30349566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 30359566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 3036916e780bShannah_mairs *AA = NULL; 30373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3038916e780bShannah_mairs } 3039916e780bShannah_mairs 3040916e780bShannah_mairs /*@C 3041916e780bShannah_mairs PetscGaussLobattoLegendreElementGradientCreate - computes the gradient for a single 1d GLL element 3042916e780bShannah_mairs 3043916e780bShannah_mairs Not Collective 3044916e780bShannah_mairs 304560225df5SJacob Faibussowitsch Input Parameters: 3046916e780bShannah_mairs + n - the number of GLL nodes 3047916e780bShannah_mairs . nodes - the GLL nodes 304860225df5SJacob Faibussowitsch - weights - the GLL weights 3049916e780bShannah_mairs 3050d8d19677SJose E. Roman Output Parameters: 305160225df5SJacob Faibussowitsch + AA - the stiffness element 305220f4b53cSBarry Smith - AAT - the transpose of AA (pass in `NULL` if you do not need this array) 3053916e780bShannah_mairs 3054916e780bShannah_mairs Level: beginner 3055916e780bShannah_mairs 3056916e780bShannah_mairs Notes: 3057dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementGradientDestroy()` 3058916e780bShannah_mairs 3059916e780bShannah_mairs You can access entries in these arrays with AA[i][j] but in memory it is stored in contiguous memory, row oriented 3060916e780bShannah_mairs 3061dce8aebaSBarry Smith .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianDestroy()`, `PetscGaussLobattoLegendreElementGradientDestroy()` 3062916e780bShannah_mairs @*/ 3063d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementGradientCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA, PetscReal ***AAT) 3064d71ae5a4SJacob Faibussowitsch { 3065916e780bShannah_mairs PetscReal **A, **AT = NULL; 3066916e780bShannah_mairs const PetscReal *gllnodes = nodes; 3067916e780bShannah_mairs const PetscInt p = n - 1; 3068e6a796c3SToby Isaac PetscReal Li, Lj, d0; 3069916e780bShannah_mairs PetscInt i, j; 3070916e780bShannah_mairs 3071916e780bShannah_mairs PetscFunctionBegin; 30729566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &A)); 30739566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &A[0])); 3074916e780bShannah_mairs for (i = 1; i < n; i++) A[i] = A[i - 1] + n; 3075916e780bShannah_mairs 3076916e780bShannah_mairs if (AAT) { 30779566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &AT)); 30789566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n * n, &AT[0])); 3079916e780bShannah_mairs for (i = 1; i < n; i++) AT[i] = AT[i - 1] + n; 3080916e780bShannah_mairs } 3081916e780bShannah_mairs 3082ad540459SPierre Jolivet if (n == 1) A[0][0] = 0.; 3083916e780bShannah_mairs d0 = (PetscReal)p * ((PetscReal)p + 1.) / 4.; 3084916e780bShannah_mairs for (i = 0; i < n; i++) { 3085916e780bShannah_mairs for (j = 0; j < n; j++) { 3086916e780bShannah_mairs A[i][j] = 0.; 30879566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(0., 0., p, gllnodes[i], &Li)); 30889566063dSJacob Faibussowitsch PetscCall(PetscDTComputeJacobi(0., 0., p, gllnodes[j], &Lj)); 3089916e780bShannah_mairs if (i != j) A[i][j] = Li / (Lj * (gllnodes[i] - gllnodes[j])); 3090916e780bShannah_mairs if ((j == i) && (i == 0)) A[i][j] = -d0; 3091916e780bShannah_mairs if (j == i && i == p) A[i][j] = d0; 3092916e780bShannah_mairs if (AT) AT[j][i] = A[i][j]; 3093916e780bShannah_mairs } 3094916e780bShannah_mairs } 3095916e780bShannah_mairs if (AAT) *AAT = AT; 3096916e780bShannah_mairs *AA = A; 30973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3098916e780bShannah_mairs } 3099916e780bShannah_mairs 3100916e780bShannah_mairs /*@C 3101dce8aebaSBarry Smith PetscGaussLobattoLegendreElementGradientDestroy - frees the gradient for a single 1d GLL element obtained with `PetscGaussLobattoLegendreElementGradientCreate()` 3102916e780bShannah_mairs 3103916e780bShannah_mairs Not Collective 3104916e780bShannah_mairs 3105d8d19677SJose E. Roman Input Parameters: 3106916e780bShannah_mairs + n - the number of GLL nodes 3107916e780bShannah_mairs . nodes - the GLL nodes 3108916e780bShannah_mairs . weights - the GLL weights 3109916e780bShannah_mairs . AA - the stiffness element 3110916e780bShannah_mairs - AAT - the transpose of the element 3111916e780bShannah_mairs 3112916e780bShannah_mairs Level: beginner 3113916e780bShannah_mairs 3114db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()`, `PetscGaussLobattoLegendreElementAdvectionCreate()` 3115916e780bShannah_mairs @*/ 3116d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementGradientDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA, PetscReal ***AAT) 3117d71ae5a4SJacob Faibussowitsch { 3118916e780bShannah_mairs PetscFunctionBegin; 31199566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 31209566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 3121916e780bShannah_mairs *AA = NULL; 31229ea709c2SMatthew G. Knepley if (AAT) { 31239566063dSJacob Faibussowitsch PetscCall(PetscFree((*AAT)[0])); 31249566063dSJacob Faibussowitsch PetscCall(PetscFree(*AAT)); 3125916e780bShannah_mairs *AAT = NULL; 3126916e780bShannah_mairs } 31273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3128916e780bShannah_mairs } 3129916e780bShannah_mairs 3130916e780bShannah_mairs /*@C 3131916e780bShannah_mairs PetscGaussLobattoLegendreElementAdvectionCreate - computes the advection operator for a single 1d GLL element 3132916e780bShannah_mairs 3133916e780bShannah_mairs Not Collective 3134916e780bShannah_mairs 3135d8d19677SJose E. Roman Input Parameters: 3136916e780bShannah_mairs + n - the number of GLL nodes 3137916e780bShannah_mairs . nodes - the GLL nodes 3138f0fc11ceSJed Brown - weights - the GLL weightss 3139916e780bShannah_mairs 3140916e780bShannah_mairs Output Parameter: 3141916e780bShannah_mairs . AA - the stiffness element 3142916e780bShannah_mairs 3143916e780bShannah_mairs Level: beginner 3144916e780bShannah_mairs 3145916e780bShannah_mairs Notes: 3146dce8aebaSBarry Smith Destroy this with `PetscGaussLobattoLegendreElementAdvectionDestroy()` 3147916e780bShannah_mairs 3148916e780bShannah_mairs This is the same as the Gradient operator multiplied by the diagonal mass matrix 3149916e780bShannah_mairs 3150916e780bShannah_mairs You can access entries in this array with AA[i][j] but in memory it is stored in contiguous memory, row oriented 3151916e780bShannah_mairs 3152db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementLaplacianCreate()`, `PetscGaussLobattoLegendreElementAdvectionDestroy()` 3153916e780bShannah_mairs @*/ 3154d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementAdvectionCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 3155d71ae5a4SJacob Faibussowitsch { 3156916e780bShannah_mairs PetscReal **D; 3157916e780bShannah_mairs const PetscReal *gllweights = weights; 3158916e780bShannah_mairs const PetscInt glln = n; 3159916e780bShannah_mairs PetscInt i, j; 3160916e780bShannah_mairs 3161916e780bShannah_mairs PetscFunctionBegin; 31629566063dSJacob Faibussowitsch PetscCall(PetscGaussLobattoLegendreElementGradientCreate(n, nodes, weights, &D, NULL)); 3163916e780bShannah_mairs for (i = 0; i < glln; i++) { 3164ad540459SPierre Jolivet for (j = 0; j < glln; j++) D[i][j] = gllweights[i] * D[i][j]; 3165916e780bShannah_mairs } 3166916e780bShannah_mairs *AA = D; 31673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3168916e780bShannah_mairs } 3169916e780bShannah_mairs 3170916e780bShannah_mairs /*@C 3171dce8aebaSBarry Smith PetscGaussLobattoLegendreElementAdvectionDestroy - frees the advection stiffness for a single 1d GLL element created with `PetscGaussLobattoLegendreElementAdvectionCreate()` 3172916e780bShannah_mairs 3173916e780bShannah_mairs Not Collective 3174916e780bShannah_mairs 3175d8d19677SJose E. Roman Input Parameters: 3176916e780bShannah_mairs + n - the number of GLL nodes 3177916e780bShannah_mairs . nodes - the GLL nodes 3178916e780bShannah_mairs . weights - the GLL weights 317960225df5SJacob Faibussowitsch - AA - advection 3180916e780bShannah_mairs 3181916e780bShannah_mairs Level: beginner 3182916e780bShannah_mairs 3183db781477SPatrick Sanan .seealso: `PetscDTGaussLobattoLegendreQuadrature()`, `PetscGaussLobattoLegendreElementAdvectionCreate()` 3184916e780bShannah_mairs @*/ 3185d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementAdvectionDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 3186d71ae5a4SJacob Faibussowitsch { 3187916e780bShannah_mairs PetscFunctionBegin; 31889566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 31899566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 3190916e780bShannah_mairs *AA = NULL; 31913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3192916e780bShannah_mairs } 3193916e780bShannah_mairs 3194d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementMassCreate(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 3195d71ae5a4SJacob Faibussowitsch { 3196916e780bShannah_mairs PetscReal **A; 3197916e780bShannah_mairs const PetscReal *gllweights = weights; 3198916e780bShannah_mairs const PetscInt glln = n; 3199916e780bShannah_mairs PetscInt i, j; 3200916e780bShannah_mairs 3201916e780bShannah_mairs PetscFunctionBegin; 32029566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(glln, &A)); 32039566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(glln * glln, &A[0])); 3204916e780bShannah_mairs for (i = 1; i < glln; i++) A[i] = A[i - 1] + glln; 3205ad540459SPierre Jolivet if (glln == 1) A[0][0] = 0.; 3206916e780bShannah_mairs for (i = 0; i < glln; i++) { 3207916e780bShannah_mairs for (j = 0; j < glln; j++) { 3208916e780bShannah_mairs A[i][j] = 0.; 3209916e780bShannah_mairs if (j == i) A[i][j] = gllweights[i]; 3210916e780bShannah_mairs } 3211916e780bShannah_mairs } 3212916e780bShannah_mairs *AA = A; 32133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3214916e780bShannah_mairs } 3215916e780bShannah_mairs 3216d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscGaussLobattoLegendreElementMassDestroy(PetscInt n, PetscReal *nodes, PetscReal *weights, PetscReal ***AA) 3217d71ae5a4SJacob Faibussowitsch { 3218916e780bShannah_mairs PetscFunctionBegin; 32199566063dSJacob Faibussowitsch PetscCall(PetscFree((*AA)[0])); 32209566063dSJacob Faibussowitsch PetscCall(PetscFree(*AA)); 3221916e780bShannah_mairs *AA = NULL; 32223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3223916e780bShannah_mairs } 3224d4afb720SToby Isaac 3225d4afb720SToby Isaac /*@ 3226d4afb720SToby Isaac PetscDTIndexToBary - convert an index into a barycentric coordinate. 3227d4afb720SToby Isaac 3228d4afb720SToby Isaac Input Parameters: 3229d4afb720SToby 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) 3230d4afb720SToby Isaac . sum - the value that the sum of the barycentric coordinates (which will be non-negative integers) should sum to 3231d4afb720SToby Isaac - index - the index to convert: should be >= 0 and < Binomial(len - 1 + sum, sum) 3232d4afb720SToby Isaac 3233d4afb720SToby Isaac Output Parameter: 3234d4afb720SToby Isaac . coord - will be filled with the barycentric coordinate 3235d4afb720SToby Isaac 3236d4afb720SToby Isaac Level: beginner 3237d4afb720SToby Isaac 3238dce8aebaSBarry Smith Note: 3239dce8aebaSBarry Smith The indices map to barycentric coordinates in lexicographic order, where the first index is the 3240d4afb720SToby Isaac least significant and the last index is the most significant. 3241d4afb720SToby Isaac 3242db781477SPatrick Sanan .seealso: `PetscDTBaryToIndex()` 3243d4afb720SToby Isaac @*/ 3244d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTIndexToBary(PetscInt len, PetscInt sum, PetscInt index, PetscInt coord[]) 3245d71ae5a4SJacob Faibussowitsch { 3246d4afb720SToby Isaac PetscInt c, d, s, total, subtotal, nexttotal; 3247d4afb720SToby Isaac 3248d4afb720SToby Isaac PetscFunctionBeginHot; 324908401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 325008401ef6SPierre Jolivet PetscCheck(index >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index must be non-negative"); 3251d4afb720SToby Isaac if (!len) { 32523ba16761SJacob Faibussowitsch if (!sum && !index) PetscFunctionReturn(PETSC_SUCCESS); 3253d4afb720SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid index or sum for length 0 barycentric coordinate"); 3254d4afb720SToby Isaac } 3255d4afb720SToby Isaac for (c = 1, total = 1; c <= len; c++) { 3256d4afb720SToby Isaac /* total is the number of ways to have a tuple of length c with sum */ 3257d4afb720SToby Isaac if (index < total) break; 3258d4afb720SToby Isaac total = (total * (sum + c)) / c; 3259d4afb720SToby Isaac } 326008401ef6SPierre Jolivet PetscCheck(c <= len, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index out of range"); 3261d4afb720SToby Isaac for (d = c; d < len; d++) coord[d] = 0; 3262d4afb720SToby Isaac for (s = 0, subtotal = 1, nexttotal = 1; c > 0;) { 3263d4afb720SToby Isaac /* subtotal is the number of ways to have a tuple of length c with sum s */ 3264d4afb720SToby Isaac /* nexttotal is the number of ways to have a tuple of length c-1 with sum s */ 3265d4afb720SToby Isaac if ((index + subtotal) >= total) { 3266d4afb720SToby Isaac coord[--c] = sum - s; 3267d4afb720SToby Isaac index -= (total - subtotal); 3268d4afb720SToby Isaac sum = s; 3269d4afb720SToby Isaac total = nexttotal; 3270d4afb720SToby Isaac subtotal = 1; 3271d4afb720SToby Isaac nexttotal = 1; 3272d4afb720SToby Isaac s = 0; 3273d4afb720SToby Isaac } else { 3274d4afb720SToby Isaac subtotal = (subtotal * (c + s)) / (s + 1); 3275d4afb720SToby Isaac nexttotal = (nexttotal * (c - 1 + s)) / (s + 1); 3276d4afb720SToby Isaac s++; 3277d4afb720SToby Isaac } 3278d4afb720SToby Isaac } 32793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3280d4afb720SToby Isaac } 3281d4afb720SToby Isaac 3282d4afb720SToby Isaac /*@ 3283d4afb720SToby Isaac PetscDTBaryToIndex - convert a barycentric coordinate to an index 3284d4afb720SToby Isaac 3285d4afb720SToby Isaac Input Parameters: 3286d4afb720SToby 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) 3287d4afb720SToby Isaac . sum - the value that the sum of the barycentric coordinates (which will be non-negative integers) should sum to 3288d4afb720SToby Isaac - coord - a barycentric coordinate with the given length and sum 3289d4afb720SToby Isaac 3290d4afb720SToby Isaac Output Parameter: 3291d4afb720SToby Isaac . index - the unique index for the coordinate, >= 0 and < Binomial(len - 1 + sum, sum) 3292d4afb720SToby Isaac 3293d4afb720SToby Isaac Level: beginner 3294d4afb720SToby Isaac 3295dce8aebaSBarry Smith Note: 3296dce8aebaSBarry Smith The indices map to barycentric coordinates in lexicographic order, where the first index is the 3297d4afb720SToby Isaac least significant and the last index is the most significant. 3298d4afb720SToby Isaac 3299db781477SPatrick Sanan .seealso: `PetscDTIndexToBary` 3300d4afb720SToby Isaac @*/ 3301d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDTBaryToIndex(PetscInt len, PetscInt sum, const PetscInt coord[], PetscInt *index) 3302d71ae5a4SJacob Faibussowitsch { 3303d4afb720SToby Isaac PetscInt c; 3304d4afb720SToby Isaac PetscInt i; 3305d4afb720SToby Isaac PetscInt total; 3306d4afb720SToby Isaac 3307d4afb720SToby Isaac PetscFunctionBeginHot; 330808401ef6SPierre Jolivet PetscCheck(len >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "length must be non-negative"); 3309d4afb720SToby Isaac if (!len) { 3310d4afb720SToby Isaac if (!sum) { 3311d4afb720SToby Isaac *index = 0; 33123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3313d4afb720SToby Isaac } 3314d4afb720SToby Isaac SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid index or sum for length 0 barycentric coordinate"); 3315d4afb720SToby Isaac } 3316d4afb720SToby Isaac for (c = 1, total = 1; c < len; c++) total = (total * (sum + c)) / c; 3317d4afb720SToby Isaac i = total - 1; 3318d4afb720SToby Isaac c = len - 1; 3319d4afb720SToby Isaac sum -= coord[c]; 3320d4afb720SToby Isaac while (sum > 0) { 3321d4afb720SToby Isaac PetscInt subtotal; 3322d4afb720SToby Isaac PetscInt s; 3323d4afb720SToby Isaac 3324d4afb720SToby Isaac for (s = 1, subtotal = 1; s < sum; s++) subtotal = (subtotal * (c + s)) / s; 3325d4afb720SToby Isaac i -= subtotal; 3326d4afb720SToby Isaac sum -= coord[--c]; 3327d4afb720SToby Isaac } 3328d4afb720SToby Isaac *index = i; 33293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3330d4afb720SToby Isaac } 333107218a29SMatthew G. Knepley 33324366bac7SMatthew G. Knepley /*@ 33334366bac7SMatthew G. Knepley PetscQuadratureComputePermutations - Compute permutations of quadrature points corresponding to domain orientations 33344366bac7SMatthew G. Knepley 33354366bac7SMatthew G. Knepley Input Parameter: 33364366bac7SMatthew G. Knepley . quad - The `PetscQuadrature` 33374366bac7SMatthew G. Knepley 33384366bac7SMatthew G. Knepley Output Parameters: 33394366bac7SMatthew G. Knepley + Np - The number of domain orientations 33404366bac7SMatthew G. Knepley - perm - An array of `IS` permutations, one for ech orientation, 33414366bac7SMatthew G. Knepley 334260820804SBarry Smith Level: developer 33434366bac7SMatthew G. Knepley 33444366bac7SMatthew G. Knepley .seealso: `PetscQuadratureSetCellType()`, `PetscQuadrature` 33454366bac7SMatthew G. Knepley @*/ 33464366bac7SMatthew G. Knepley PetscErrorCode PetscQuadratureComputePermutations(PetscQuadrature quad, PetscInt *Np, IS *perm[]) 334707218a29SMatthew G. Knepley { 33484366bac7SMatthew G. Knepley DMPolytopeType ct; 334907218a29SMatthew G. Knepley const PetscReal *xq, *wq; 335007218a29SMatthew G. Knepley PetscInt dim, qdim, d, Na, o, Nq, q, qp; 335107218a29SMatthew G. Knepley 335207218a29SMatthew G. Knepley PetscFunctionBegin; 33534366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetData(quad, &qdim, NULL, &Nq, &xq, &wq)); 33544366bac7SMatthew G. Knepley PetscCall(PetscQuadratureGetCellType(quad, &ct)); 335507218a29SMatthew G. Knepley dim = DMPolytopeTypeGetDim(ct); 335607218a29SMatthew G. Knepley Na = DMPolytopeTypeGetNumArrangments(ct); 335707218a29SMatthew G. Knepley PetscCall(PetscMalloc1(Na, perm)); 33584366bac7SMatthew G. Knepley if (Np) *Np = Na; 33594366bac7SMatthew G. Knepley Na /= 2; 33604366bac7SMatthew G. Knepley for (o = -Na; o < Na; ++o) { 336107218a29SMatthew G. Knepley DM refdm; 336207218a29SMatthew G. Knepley PetscInt *idx; 336307218a29SMatthew G. Knepley PetscReal xi0[3] = {-1., -1., -1.}, v0[3], J[9], detJ, txq[3]; 336407218a29SMatthew G. Knepley PetscBool flg; 336507218a29SMatthew G. Knepley 336607218a29SMatthew G. Knepley PetscCall(DMPlexCreateReferenceCell(PETSC_COMM_SELF, ct, &refdm)); 336707218a29SMatthew G. Knepley PetscCall(DMPlexOrientPoint(refdm, 0, o)); 336807218a29SMatthew G. Knepley PetscCall(DMPlexComputeCellGeometryFEM(refdm, 0, NULL, v0, J, NULL, &detJ)); 336907218a29SMatthew G. Knepley PetscCall(PetscMalloc1(Nq, &idx)); 337007218a29SMatthew G. Knepley for (q = 0; q < Nq; ++q) { 337107218a29SMatthew G. Knepley CoordinatesRefToReal(dim, dim, xi0, v0, J, &xq[q * dim], txq); 337207218a29SMatthew G. Knepley for (qp = 0; qp < Nq; ++qp) { 337307218a29SMatthew G. Knepley PetscReal diff = 0.; 337407218a29SMatthew G. Knepley 337507218a29SMatthew G. Knepley for (d = 0; d < dim; ++d) diff += PetscAbsReal(txq[d] - xq[qp * dim + d]); 337607218a29SMatthew G. Knepley if (diff < PETSC_SMALL) break; 337707218a29SMatthew G. Knepley } 337807218a29SMatthew G. Knepley PetscCheck(qp < Nq, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Transformed quad point %" PetscInt_FMT " does not match another quad point", q); 337907218a29SMatthew G. Knepley idx[q] = qp; 338007218a29SMatthew G. Knepley } 338107218a29SMatthew G. Knepley PetscCall(DMDestroy(&refdm)); 33824366bac7SMatthew G. Knepley PetscCall(ISCreateGeneral(PETSC_COMM_SELF, Nq, idx, PETSC_OWN_POINTER, &(*perm)[o + Na])); 33834366bac7SMatthew G. Knepley PetscCall(ISGetInfo((*perm)[o + Na], IS_PERMUTATION, IS_LOCAL, PETSC_TRUE, &flg)); 338407218a29SMatthew G. Knepley PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Ordering for orientation %" PetscInt_FMT " was not a permutation", o); 33854366bac7SMatthew G. Knepley PetscCall(ISSetPermutation((*perm)[o + Na])); 33864366bac7SMatthew G. Knepley } 33874366bac7SMatthew G. Knepley if (!Na) (*perm)[0] = NULL; 33884366bac7SMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 33894366bac7SMatthew G. Knepley } 33904366bac7SMatthew G. Knepley 33914366bac7SMatthew G. Knepley /*@ 33924366bac7SMatthew G. Knepley PetscDTCreateDefaultQuadrature - Create default quadrature for a given cell 33934366bac7SMatthew G. Knepley 33944366bac7SMatthew G. Knepley Not collective 33954366bac7SMatthew G. Knepley 33964366bac7SMatthew G. Knepley Input Parameters: 33974366bac7SMatthew G. Knepley + ct - The integration domain 33984366bac7SMatthew G. Knepley - qorder - The desired quadrature order 33994366bac7SMatthew G. Knepley 34004366bac7SMatthew G. Knepley Output Parameters: 34014366bac7SMatthew G. Knepley + q - The cell quadrature 34024366bac7SMatthew G. Knepley - fq - The face quadrature 34034366bac7SMatthew G. Knepley 34044366bac7SMatthew G. Knepley Level: developer 34054366bac7SMatthew G. Knepley 34064366bac7SMatthew G. Knepley .seealso: `PetscFECreateDefault()`, `PetscDTGaussTensorQuadrature()`, `PetscDTSimplexQuadrature()`, `PetscDTTensorQuadratureCreate()` 34074366bac7SMatthew G. Knepley @*/ 34084366bac7SMatthew G. Knepley PetscErrorCode PetscDTCreateDefaultQuadrature(DMPolytopeType ct, PetscInt qorder, PetscQuadrature *q, PetscQuadrature *fq) 34094366bac7SMatthew G. Knepley { 34104366bac7SMatthew G. Knepley const PetscInt quadPointsPerEdge = PetscMax(qorder + 1, 1); 34114366bac7SMatthew G. Knepley const PetscInt dim = DMPolytopeTypeGetDim(ct); 34124366bac7SMatthew G. Knepley 34134366bac7SMatthew G. Knepley PetscFunctionBegin; 34144366bac7SMatthew G. Knepley switch (ct) { 34154366bac7SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 34164366bac7SMatthew G. Knepley case DM_POLYTOPE_POINT_PRISM_TENSOR: 34174366bac7SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 34184366bac7SMatthew G. Knepley case DM_POLYTOPE_SEG_PRISM_TENSOR: 34194366bac7SMatthew G. Knepley case DM_POLYTOPE_HEXAHEDRON: 34204366bac7SMatthew G. Knepley case DM_POLYTOPE_QUAD_PRISM_TENSOR: 34214366bac7SMatthew G. Knepley PetscCall(PetscDTGaussTensorQuadrature(dim, 1, quadPointsPerEdge, -1.0, 1.0, q)); 34224366bac7SMatthew G. Knepley PetscCall(PetscDTGaussTensorQuadrature(dim - 1, 1, quadPointsPerEdge, -1.0, 1.0, fq)); 34234366bac7SMatthew G. Knepley break; 34244366bac7SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 34254366bac7SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 34264366bac7SMatthew G. Knepley PetscCall(PetscDTSimplexQuadrature(dim, 2 * qorder, PETSCDTSIMPLEXQUAD_DEFAULT, q)); 34274366bac7SMatthew G. Knepley PetscCall(PetscDTSimplexQuadrature(dim - 1, 2 * qorder, PETSCDTSIMPLEXQUAD_DEFAULT, fq)); 34284366bac7SMatthew G. Knepley break; 34294366bac7SMatthew G. Knepley case DM_POLYTOPE_TRI_PRISM: 34304366bac7SMatthew G. Knepley case DM_POLYTOPE_TRI_PRISM_TENSOR: { 34314366bac7SMatthew G. Knepley PetscQuadrature q1, q2; 34324366bac7SMatthew G. Knepley 34334366bac7SMatthew G. Knepley // TODO: this should be able to use symmetric rules, but doing so causes tests to fail 34344366bac7SMatthew G. Knepley PetscCall(PetscDTSimplexQuadrature(2, 2 * qorder, PETSCDTSIMPLEXQUAD_CONIC, &q1)); 34354366bac7SMatthew G. Knepley PetscCall(PetscDTGaussTensorQuadrature(1, 1, quadPointsPerEdge, -1.0, 1.0, &q2)); 34364366bac7SMatthew G. Knepley PetscCall(PetscDTTensorQuadratureCreate(q1, q2, q)); 34374366bac7SMatthew G. Knepley PetscCall(PetscQuadratureDestroy(&q2)); 34384366bac7SMatthew G. Knepley *fq = q1; 34394366bac7SMatthew G. Knepley /* TODO Need separate quadratures for each face */ 34404366bac7SMatthew G. Knepley } break; 34414366bac7SMatthew G. Knepley default: 34424366bac7SMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No quadrature for celltype %s", DMPolytopeTypes[PetscMin(ct, DM_POLYTOPE_UNKNOWN)]); 344307218a29SMatthew G. Knepley } 344407218a29SMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 344507218a29SMatthew G. Knepley } 3446