Merge pull request firedrakeproject#117 from FInAT/pbrubeck/variants

Support Lagrange/IntegratedLegendre variants
angus-g · Feb 7, 2024 · ecf345a · ecf345a
2 parents a7080b1 + 3de9323
commit ecf345a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/finat/fiat_elements.py b/finat/fiat_elements.py
@@ -346,8 +346,8 @@ def __init__(self, cell, degree):
 
 
 class Lagrange(ScalarFiatElement):
-    def __init__(self, cell, degree):
-        super(Lagrange, self).__init__(FIAT.Lagrange(cell, degree))
+    def __init__(self, cell, degree, variant=None):
+        super(Lagrange, self).__init__(FIAT.Lagrange(cell, degree, variant=variant))
 
 
 class KongMulderVeldhuizen(ScalarFiatElement):

diff --git a/finat/spectral.py b/finat/spectral.py
@@ -65,18 +65,18 @@ def basis_evaluation(self, order, ps, entity=None, coordinate_mapping=None):
 
 
 class Legendre(ScalarFiatElement):
-    """1D DG element with Legendre polynomials."""
+    """DG element with Legendre polynomials."""
 
     def __init__(self, cell, degree):
         fiat_element = FIAT.Legendre(cell, degree)
         super(Legendre, self).__init__(fiat_element)
 
 
 class IntegratedLegendre(ScalarFiatElement):
-    """1D CG element with integrated Legendre polynomials."""
+    """CG element with integrated Legendre polynomials."""
 
-    def __init__(self, cell, degree):
-        fiat_element = FIAT.IntegratedLegendre(cell, degree)
+    def __init__(self, cell, degree, variant=None):
+        fiat_element = FIAT.IntegratedLegendre(cell, degree, variant=variant)
         super(IntegratedLegendre, self).__init__(fiat_element)