|
11 | 11 | import numpy.typing as _npt
|
12 | 12 | import ufl as _ufl
|
13 | 13 | # TODO: remove gdim arguments once UFL handles cells better
|
14 |
| -# TODO: remove IrreducibleInt once UFL handles element degrees better |
15 |
| -from ufl.algorithms.estimate_degrees import IrreducibleInt as _IrreducibleInt |
16 | 14 | from ufl.finiteelement import FiniteElementBase as _FiniteElementBase
|
17 | 15 |
|
18 | 16 | import basix as _basix
|
@@ -51,7 +49,7 @@ class _ElementBase(_FiniteElementBase):
|
51 | 49 | """
|
52 | 50 |
|
53 | 51 | def __init__(self, repr: str, name: str, cellname: str, value_shape: _typing.Tuple[int, ...],
|
54 |
| - degree: _typing.Union[int, _IrreducibleInt] = -1, mapname: _typing.Optional[str] = None, |
| 52 | + degree: int = -1, mapname: _typing.Optional[str] = None, |
55 | 53 | gdim: _typing.Optional[int] = None):
|
56 | 54 | """Initialise the element."""
|
57 | 55 | super().__init__(name, _ufl.cell.Cell(cellname, gdim), degree, None, value_shape, value_shape)
|
@@ -342,15 +340,9 @@ def __init__(self, element: _basix.finite_element.FiniteElement, gdim: _typing.O
|
342 | 340 | repr = (f"Basix element ({element.family.name}, {element.cell_type.name}, {element.degree}, "
|
343 | 341 | f"{element.lagrange_variant.name}, {element.dpc_variant.name}, {element.discontinuous})")
|
344 | 342 |
|
345 |
| - if element.cell_type.name in ["interval", "triangle", "tetrahedron"]: |
346 |
| - super().__init__( |
347 |
| - repr, element.family.name, element.cell_type.name, tuple(element.value_shape), element.degree, |
348 |
| - _map_type_to_string(element.map_type), gdim=gdim) |
349 |
| - else: |
350 |
| - # TODO: remove IrreducibleInt once UFL handles element degrees better |
351 |
| - super().__init__( |
352 |
| - repr, element.family.name, element.cell_type.name, tuple(element.value_shape), |
353 |
| - _IrreducibleInt(element.degree), _map_type_to_string(element.map_type), gdim=gdim) |
| 343 | + super().__init__( |
| 344 | + repr, element.family.name, element.cell_type.name, tuple(element.value_shape), element.degree, |
| 345 | + _map_type_to_string(element.map_type), gdim=gdim) |
354 | 346 |
|
355 | 347 | self.element = element
|
356 | 348 |
|
|
0 commit comments