update tensor product interface

Author: mscroggs

.github/workflows/dolfinx-tests.yml

@@ -36,7 +36,7 @@ jobs:
         if: github.event_name != 'workflow_dispatch'
         run: |
           python3 -m pip install git+https://github.com/FEniCS/ufl.git
-          python3 -m pip install git+https://github.com/FEniCS/basix.git
+          python3 -m pip install git+https://github.com/FEniCS/basix.git@chris/tp-element
       - name: Install UFL and Basix (specified branches/tags)
         if: github.event_name == 'workflow_dispatch'
         run: |

.github/workflows/pythonapp.yml

@@ -39,7 +39,7 @@ jobs:
       - name: Install FEniCS dependencies (Python)
         run: |
           pip install git+https://github.com/FEniCS/ufl.git
-          pip install git+https://github.com/FEniCS/basix.git
+          pip install git+https://github.com/FEniCS/basix.git@chris/tp-element
       - name: Install FFCx
         run: pip install .[ci]
       - name: Lint with flake8

ffcx/ir/elementtables.py

@@ -409,7 +409,7 @@ def build_optimized_tables(quadrature_rule, cell, integral_type, entitytype,
             factors = element.get_tensor_product_representation()
 
             tensor_factors = []
-            for i, j in enumerate(factors[0][0]):
+            for i, j in enumerate(factors[0]):
                 pts = quadrature_rule.tensor_factors[i][0]
                 d = local_derivatives[i]
                 sub_tbl = j.tabulate(d, pts)[d]

test/test_tensor_product.py

@@ -27,13 +27,9 @@ def create_tensor_product_element(cell_type, degree, variant, shape=None):
     """Create tensor product element."""
     gdim = cell_to_gdim(cell_type)
     family = basix.ElementFamily.P
-    ref = basix.create_element(family, cell_type, degree, variant)
-    factors = ref.get_tensor_product_representation()[0]
-    perm = factors[1]
-    dof_ordering = np.argsort(perm)
-    element = basix.create_element(family, cell_type, degree, variant,
-                                   dof_ordering=dof_ordering)
-    uflelement = basix.ufl._BasixElement(element, gdim=gdim)
+    element = basix.create_tp_element(family, cell_type, degree, variant)
+    factors = element.get_tensor_product_representation()[0]
+    uflelement = basix.ufl.wrap_element(element, gdim=gdim)
     if shape is None:
         return uflelement
     else: