[BC] plot mmf

Author: BonneelP

pyleecan/GUI/Dialog/DMachineSetup/SPreview/WMachineTable/WMachineTable.py

@@ -101,7 +101,7 @@ def plot_mmf(self):
         """Plot the unit mmf of the stator"""
         try:
             if self.machine is not None:
-                self.machine.stator.plot_mmf_unit(is_create_appli=False)
+                self.machine.stator.plot_mmf_unit()
                 set_plot_gui_icon()
         except Exception as e:
             err_msg = "Error while plotting Stator mmf unit:\n" + str(e)

pyleecan/GUI/Dialog/DMachineSetup/SWinding/SWinding.py

@@ -80,6 +80,8 @@ def __init__(self, machine, material_dict, is_stator=False):
             # Enforce star of slot
             self.c_wind_type.setEnabled(False)
             self.c_wind_type.setCurrentIndex(0)
+            # No plot_mmf
+            self.b_plot_mmf.hide()
 
         # Pattern Group setup
         if self.obj.winding is None:
@@ -561,7 +563,7 @@ def s_plot_mmf(self):
         """Plot the unit mmf of the stator"""
         if self.machine is not None:
             try:
-                self.plot_widget = self.obj.plot_mmf_unit(is_create_appli=False)
+                self.obj.plot_mmf_unit()
             except Exception as e:
                 if self.obj.is_stator:  # Adapt the text to the current lamination
                     err_msg = "Error while plotting Stator mmf unit:\n" + str(e)

pyleecan/Methods/Machine/LamSlotWind/plot_mmf_unit.py

@@ -1,24 +1,21 @@
 import matplotlib.pyplot as plt
-from numpy import min as np_min, max as np_max
 
-from PySide2.QtCore import *
-from PySide2.QtGui import *
-from PySide2.QtWidgets import *
-
-from ....Functions.Plot.set_plot_gui_icon import set_plot_gui_icon
 from ....Functions.Plot import dict_2D
 from ....definitions import config_dict
 
 
-def plot_mmf_unit(self, is_create_appli=True, save_path=None):
+def plot_mmf_unit(self, r_max=100, fig=None, is_show_fig=True):
     """Plot the winding unit mmf as a function of space
     Parameters
     ----------
-
     self : LamSlotWind
         an LamSlotWind object
-    is_create_appli : bool
-        True to create an QApplication (required if not already created by another GUI)
+    Na : int
+        Space discretization
+    fig : Matplotlib.figure.Figure
+        existing figure to use if None create a new one
+    is_show_fig : bool
+        To call show at the end of the method
     """
 
     name = ""
@@ -30,55 +27,36 @@ def plot_mmf_unit(self, is_create_appli=True, save_path=None):
         name += "Rotor "
 
     # Compute the winding function and mmf
-    wf = self.comp_wind_function(per_a=1)
     qs = self.winding.qs
-    MMF_U, WF = self.comp_mmf_unit(Nt=1, Na=wf.shape[1])
+    p = self.get_pole_pair_number()
+    MMF_U, WF = self.comp_mmf_unit(Nt=1, Na=400 * p)
 
     color_list = config_dict["PLOT"]["COLOR_DICT"]["COLOR_LIST"][:qs]
 
-    plot_arg_dict = dict_2D.copy()
-    plot_arg_dict["color_list"] = color_list + ["k"]
-    plot_arg_dict["data_list"] = [MMF_U]
-
-    if save_path is not None:
-        WF.plot_2D_Data(
-            "angle{°}",
-            "phase[]",
-            unit="A",
-            y_min=np_min(MMF_U.values) * 1.1,
-            y_max=np_max(MMF_U.values) * 1.1,
-            is_show_fig=False,
-            save_path=save_path,
-            **plot_arg_dict,
-        )
-
-    elif is_create_appli:
-        WF.plot(
-            "angle{°}",
-            "phase[]",
-            unit="A",
-            z_min=np_min(MMF_U.values) * 1.1,
-            z_max=np_max(MMF_U.values) * 1.1,
-            plot_arg_dict=plot_arg_dict,
-            is_create_appli=is_create_appli,
-            frozen_type=2,
-        )
-    else:
-        wid = WF.plot(
-            "angle{°}",
-            "phase[]",
-            unit="A",
-            z_min=np_min(MMF_U.values) * 1.1,
-            z_max=np_max(MMF_U.values) * 1.1,
-            plot_arg_dict=plot_arg_dict,
-            is_create_appli=is_create_appli,
-            frozen_type=2,
-        )
-
-        wid.setWindowTitle(name + "Phase MMF plot")
-        set_plot_gui_icon()
-        # Change default file name
-        wid.canvas.get_default_filename = (
-            lambda: wid.windowTitle().replace(" ", "_").replace(":", "") + ".png"
-        )
-        return wid
+    fig, axs = plt.subplots(2, 1, tight_layout=True, figsize=(8, 8))
+
+    dict_2D_0 = dict_2D.copy()
+    dict_2D_0["color_list"] = color_list + ["k"]
+
+    WF.plot_2D_Data(
+        "angle{°}",
+        "phase[]",
+        data_list=[MMF_U],
+        fig=fig,
+        ax=axs[0],
+        is_show_fig=is_show_fig,
+        win_title=name + "phase MMF",
+        **dict_2D_0,
+    )
+
+    dict_2D_0["color_list"] = [color_list[0], "k"]
+
+    WF.plot_2D_Data(
+        "wavenumber=[0," + str(r_max) + "]",
+        data_list=[MMF_U],
+        fig=fig,
+        ax=axs[1],
+        is_show_fig=is_show_fig,
+        win_title=name + "phase MMF FFT",
+        **dict_2D_0,
+    )