fix: interpolator map support map for all 2d supports

Loop3D · Jun 4, 2024 · 547fea0 · 547fea0
1 parent 4be3370
commit 547fea0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/LoopStructural/interpolators/__init__.py b/LoopStructural/interpolators/__init__.py
@@ -102,9 +102,12 @@ class InterpolatorType(IntEnum):
         2: SupportType.StructuredGrid2D,
         3: SupportType.StructuredGrid,
     },
-    InterpolatorType.DISCRETE_FOLD: SupportType.TetMesh,
-    InterpolatorType.PIECEWISE_LINEAR: SupportType.TetMesh,
-    InterpolatorType.PIECEWISE_QUADRATIC: SupportType.P2UnstructuredTetMesh,
+    InterpolatorType.DISCRETE_FOLD: {3: SupportType.TetMesh, 2: SupportType.P1Unstructured2d},
+    InterpolatorType.PIECEWISE_LINEAR: {3: SupportType.TetMesh, 2: SupportType.P1Unstructured2d},
+    InterpolatorType.PIECEWISE_QUADRATIC: {
+        3: SupportType.P2UnstructuredTetMesh,
+        2: SupportType.P2Unstructured2d,
+    },
 }
 
 from ._interpolator_factory import InterpolatorFactory
diff --git a/LoopStructural/interpolators/_interpolator_factory.py b/LoopStructural/interpolators/_interpolator_factory.py
@@ -30,12 +30,8 @@ def create_interpolator(
             interpolatortype = interpolator_string_map[interpolatortype]
         if support is None:
             # raise Exception("Support must be specified")
-            if interpolatortype is InterpolatorType.FINITE_DIFFERENCE:
 
-                supporttype = support_interpolator_map[interpolatortype][boundingbox.dimensions]
-
-            else:
-                supporttype = support_interpolator_map[interpolatortype]
+            supporttype = support_interpolator_map[interpolatortype][boundingbox.dimensions]
 
             support = SupportFactory.create_support_from_bbox(
                 supporttype,