Fixing FIAT element used.

Correction by Pablo
NGSolve · Oct 30, 2023 · 0e30cbc · 0e30cbc
1 parent 671f530
commit 0e30cbc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ngsPETSc/utils/firedrake.py b/ngsPETSc/utils/firedrake.py
@@ -5,9 +5,11 @@
 '''
 try:
     import firedrake as fd
+    import FIAT
     import ufl
 except ImportError:
     fd = None
+    FIAT = None
     ufl = None
 
 import warnings
@@ -71,6 +73,8 @@ def curveField(self, order):
 
     #Computing reference points using fiat
     fiat_element = newFunctionCoordinates.function_space().finat_element.fiat_equivalent
+    if isinstance(fiat_element, FIAT.DiscontinuousElement):
+        fiat_element = fiat_element._element
     entity_ids = fiat_element.entity_dofs()
     nodes = fiat_element.dual_basis()
     refPts = []