Pass more variants around (#141)

* add variant for MixedElement (#129) * Allow VectorElement to take a variant argument. (#17) Co-authored-by: Patrick Farrell <[email protected]> Co-authored-by: Pablo Brubeck <[email protected]> Co-authored-by: Patrick E. Farrell <[email protected]> Co-authored-by: Patrick Farrell <[email protected]>
FEniCS · Jan 6, 2023 · 8382ed0 · 8382ed0
1 parent f762039
commit 8382ed0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ufl/finiteelement/mixedelement.py b/ufl/finiteelement/mixedelement.py
@@ -234,6 +234,13 @@ def degree(self, component=None):
     def reconstruct(self, **kwargs):
         return MixedElement(*[e.reconstruct(**kwargs) for e in self.sub_elements()])
 
+    def variant(self):
+        try:
+            variant, = {e.variant() for e in self.sub_elements()}
+            return variant
+        except ValueError:
+            return None
+
     def __str__(self):
         "Format as string for pretty printing."
         tmp = ", ".join(str(element) for element in self._sub_elements)