Moved conv_fun in Mode.plot_component after xy component splitting.

HelgeGehring · Oct 9, 2024 · 1ab3b06 · 1ab3b06
1 parent b5fd13d
commit 1ab3b06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/femwell/maxwell/waveguide.py b/femwell/maxwell/waveguide.py
@@ -329,8 +329,9 @@ def plot_component(
 
         if component == "x" or component == "y":
             plot_basis = mfield_t_basis.with_element(ElementVector(ElementDG(ElementTriP1())))
-            mfield_xy = plot_basis.project(mfield_t_basis.interpolate(conv_func(mfield_t)))
+            mfield_xy = plot_basis.project(mfield_t_basis.interpolate(mfield_t))
             (mfield_x, mfield_x_basis), (mfield_y, mfield_y_basis) = plot_basis.split(mfield_xy)
+            mfield_x, mfield_y = conv_func(mfield_x), conv_func(mfield_y)
             if component == "x":
                 mfield_x_basis.plot(mfield_x, ax=ax, shading="gouraud")
             else: