Source code for pyleecan.GUI.Dialog.DMachineSetup.SMagnet.SMagnet

# -*- coding: utf-8 -*-

from numpy import pi
from PySide2.QtCore import Signal
from PySide2.QtGui import QPixmap
from PySide2.QtWidgets import QDialog, QMessageBox

from .....Classes.Magnet import Magnet
from .....Classes.MagnetType13 import MagnetType13
from .....Classes.Slot import Slot
from .....GUI.Dialog.DMachineSetup.SMagnet.PMagnet10.PMagnet10 import PMagnet10
from .....GUI.Dialog.DMachineSetup.SMagnet.PMagnet11.PMagnet11 import PMagnet11
from .....GUI.Dialog.DMachineSetup.SMagnet.PMagnet12.PMagnet12 import PMagnet12
from .....GUI.Dialog.DMachineSetup.SMagnet.PMagnet13.PMagnet13 import PMagnet13
from .....GUI.Dialog.DMachineSetup.SMagnet.PMagnet14.PMagnet14 import PMagnet14
from .....GUI.Dialog.DMachineSetup.SMagnet.Ui_SMagnet import Ui_SMagnet

[docs]class SMagnet(Ui_SMagnet, QDialog): """Step to set the magnet (and slot) for SPMSM/SIPMSM""" # Signal to DMachineSetup to know that the save popup is needed saveNeeded = Signal() # Information for DMachineType nav step_name = "Magnet" def __init__(self, machine, matlib, is_stator=False): """Initialize the widget according to machine Parameters ---------- self : SMagnet A SMagnet widget machine : Machine current machine to edit matlib : MatLib Material Library is_stator : bool To adapt the GUI to set either the stator or the rotor """ # Build the interface according to the .ui file QDialog.__init__(self) self.setupUi(self) # Saving arguments self.machine = machine self.matlib = matlib self.is_stator = is_stator # Get the correct available widgets according to the machine if self.machine.rotor.is_internal: self.wid_list = [PMagnet10, PMagnet11, PMagnet12, PMagnet13, PMagnet14] elif self.machine.type_machine == 6: # For SPMSM only polar magnet for external rotor self.wid_list = [PMagnet11] else: # Magnet 14 not available for SIPMSM self.wid_list = [PMagnet10, PMagnet11, PMagnet12, PMagnet13] self.mag_type_index = [wid.mag_type for wid in self.wid_list] self.slot_type_index = [wid.slot_type for wid in self.wid_list] self.mag_name_index = [wid.mag_name for wid in self.wid_list] # Help URL self.b_help.url = "" self.b_help.url += "how-to-set-up-inset-magnet" # Fill the combobox with the available magnet self.c_type.clear() for mag in self.mag_name_index: self.c_type.addItem(mag) # Avoid erase all the parameters when navigating though the magnets self.previous_mag = dict() for mag_type in self.mag_type_index: self.previous_mag[mag_type] = None # Set the GUI with the current values if provided if ( type(self.machine.rotor.slot) is Slot or self.machine.rotor.slot.magnet is None or type(self.machine.rotor.slot.magnet[0]) is Magnet ): # Magnet or slot not set # Type 11 is default self.set_type(1) self.c_type.setCurrentIndex(1) else: # The Magnet is set => load the parameter # Set the type index = self.mag_type_index.index(type(self.machine.rotor.slot.magnet[0])) self.set_type_gui(index) self.c_type.setCurrentIndex(index) # Only one magnet in the GUI (for now) self.machine.rotor.slot.W3 = 0 # Set magnetization if self.machine.rotor.slot.magnet[0].type_magnetization is None: self.machine.rotor.slot.magnet[0].type_magnetization = 0 self.c_type_magnetization.setCurrentIndex( self.machine.rotor.slot.magnet[0].type_magnetization ) self.set_slot_pitch() # Set material self.w_mat.setText("mat_mag:")) self.w_mat.def_mat = "Magnet1" self.w_mat.update(self.machine.rotor.slot.magnet[0], "mat_type", self.matlib) # Connect signals self.c_type.currentIndexChanged.connect(self.set_type) self.c_type_magnetization.currentIndexChanged.connect( self.s_set_type_magnetization ) self.b_plot.clicked.connect(self.s_plot) self.w_mat.saveNeeded.connect(self.emit_save)
[docs] def set_slot_pitch(self): """Update the slot pitch text Parameters ---------- self : SMagnet A SMagnet object """ Nmag_txt ="Number of magnets = 2p = ") if self.machine.rotor.slot.Zs is not None: Zs = self.machine.rotor.slot.Zs out = Nmag_txt + str(Zs) + " => " Slot_pitch = 360.0 / Zs Slot_pitch_rad = Slot_pitch * pi / 180 pitch_txt ="Slot pitch = ") out += ( pitch_txt + "%.4g" % (Slot_pitch) + u" ° (" + "%.4g" % (Slot_pitch_rad) + " rad)" ) self.out_Nmag.setText(out) else: self.out_Nmag.setText(Nmag_txt + "?")
[docs] def emit_save(self): """Emit the saveNeeded signal for the DMachineSetup""" self.saveNeeded.emit()
[docs] def set_type_gui(self, index): """Change the GUI to the correct Magnet widget Parameters ---------- self : SMagnet A SMagnet object index : int Index of the Magnet type to use """ # Regenerate the pages with the new values self.w_mag.setParent(None) self.w_mag = self.wid_list[index](self.machine) self.w_mag.saveNeeded.connect(self.emit_save) # Refresh the GUI self.main_layout.removeWidget(self.w_mag) self.main_layout.insertWidget(2, self.w_mag)
[docs] def set_type_obj(self, index): """Set the type of magnet (update only the object) Parameters ---------- self : SMagnet A SMagnet object index : int Index of the Magnet type to use """ # Save the mag slot = self.machine.rotor.slot if slot.magnet is not None: self.previous_mag[type(slot.magnet[0])] = slot if self.previous_mag[self.mag_type_index[index]] is None: # Set the slot Zs = self.machine.rotor.slot.Zs self.machine.rotor.slot = self.slot_type_index[index]() self.machine.rotor.slot._set_None() self.machine.rotor.slot.Zs = Zs self.machine.rotor.slot.W3 = 0 # Set the magnet self.machine.rotor.slot.magnet = list() mag = self.mag_type_index[index]() self.machine.rotor.slot.magnet.append(mag) self.machine.rotor.slot.magnet[0]._set_None() else: self.machine.rotor.slot = self.previous_mag[self.mag_type_index[index]]
[docs] def set_type(self, index): """Signal to set the type of magnet (update both the object and the gui) Parameters ---------- self : SMagnet A SMagnet object index : int Index of the Magnet type to use """ self.set_type_obj(index) self.set_type_gui(index) # Notify the machine GUI that the machine has changed self.saveNeeded.emit()
[docs] def s_plot(self): """Plot the current machine Parameters ---------- self : SMagnet A SMagnet object """ # We have to make sure the slot is right before truing to plot it error = self.check(self.machine.rotor) if error: # Error => Display it QMessageBox().critical(self,"Error"), error) else: # No error => Plot the machine self.machine.plot()
[docs] @staticmethod def check(lamination): """Check that the current machine have all the needed field set Parameters ---------- lamination : Lamination Lamination to check Returns ------- error: str Error message (return None if no error) """ try: # Check that everything is set if lamination.slot.magnet[0].Wmag is None: return "You must set Wmag !" if lamination.slot.magnet[0].Hmag is None: return "You must set Hmag !" if ( hasattr(lamination.slot.magnet[0], "Rtop") and lamination.slot.magnet[0].Rtop is None ): return "You must set Rtopm !" if lamination.slot.H0 is None: return "You must set H0 !" except Exception as e: return str(e) # Check that everything is set right try: mec_gap = lamination.parent.comp_width_airgap_mec() except: return "Unable to draw the magnet, " "please check your geometry !" if mec_gap <= 0: return "You must have gap_min > 0 (reduce Hmag) !" if hasattr(lamination.slot.magnet[0], "Rtop"): if ( type(lamination.slot.magnet[0]) is MagnetType13 and lamination.slot.magnet[0].Rtop < lamination.slot.magnet[0].Wmag / 2.0 ): return "You must have Rtopm >= Wmag/2 !"
[docs] def s_set_type_magnetization(self, index): """Signal to update the value of type_magnetization according to the combobox Parameters ---------- self : SMagnet A SMagnet object index : int Current index of the combobox """ self.machine.rotor.slot.magnet[0].type_magnetization = index # Notify the machine GUI that the machine has changed self.saveNeeded.emit()