Source code for pyleecan.Classes.HoleM57

# -*- coding: utf-8 -*-
# File generated according to Generator/ClassesRef/Slot/HoleM57.csv
# WARNING! All changes made in this file will be lost!
"""Method code available at https://github.com/Eomys/pyleecan/tree/master/pyleecan/Methods/Slot/HoleM57
"""

from os import linesep
from sys import getsizeof
from logging import getLogger
from ._check import check_var, raise_
from ..Functions.get_logger import get_logger
from ..Functions.save import save
from ..Functions.load import load_init_dict
from ..Functions.Load.import_class import import_class
from copy import deepcopy
from .HoleMag import HoleMag

# Import all class method
# Try/catch to remove unnecessary dependencies in unused method
try:
    from ..Methods.Slot.HoleM57._comp_point_coordinate import _comp_point_coordinate
except ImportError as error:
    _comp_point_coordinate = error

try:
    from ..Methods.Slot.HoleM57.build_geometry import build_geometry
except ImportError as error:
    build_geometry = error

try:
    from ..Methods.Slot.HoleM57.check import check
except ImportError as error:
    check = error

try:
    from ..Methods.Slot.HoleM57.comp_magnetization_dict import comp_magnetization_dict
except ImportError as error:
    comp_magnetization_dict = error

try:
    from ..Methods.Slot.HoleM57.comp_radius import comp_radius
except ImportError as error:
    comp_radius = error

try:
    from ..Methods.Slot.HoleM57.comp_surface_magnet_id import comp_surface_magnet_id
except ImportError as error:
    comp_surface_magnet_id = error

try:
    from ..Methods.Slot.HoleM57.has_magnet import has_magnet
except ImportError as error:
    has_magnet = error

try:
    from ..Methods.Slot.HoleM57.plot_schematics import plot_schematics
except ImportError as error:
    plot_schematics = error

try:
    from ..Methods.Slot.HoleM57.remove_magnet import remove_magnet
except ImportError as error:
    remove_magnet = error


from numpy import isnan
from ._check import InitUnKnowClassError


[docs]class HoleM57(HoleMag): """V shape slot for buried magnet""" VERSION = 1 IS_SYMMETRICAL = 1 # Check ImportError to remove unnecessary dependencies in unused method # cf Methods.Slot.HoleM57._comp_point_coordinate if isinstance(_comp_point_coordinate, ImportError): _comp_point_coordinate = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method _comp_point_coordinate: " + str(_comp_point_coordinate) ) ) ) else: _comp_point_coordinate = _comp_point_coordinate # cf Methods.Slot.HoleM57.build_geometry if isinstance(build_geometry, ImportError): build_geometry = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method build_geometry: " + str(build_geometry) ) ) ) else: build_geometry = build_geometry # cf Methods.Slot.HoleM57.check if isinstance(check, ImportError): check = property( fget=lambda x: raise_( ImportError("Can't use HoleM57 method check: " + str(check)) ) ) else: check = check # cf Methods.Slot.HoleM57.comp_magnetization_dict if isinstance(comp_magnetization_dict, ImportError): comp_magnetization_dict = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method comp_magnetization_dict: " + str(comp_magnetization_dict) ) ) ) else: comp_magnetization_dict = comp_magnetization_dict # cf Methods.Slot.HoleM57.comp_radius if isinstance(comp_radius, ImportError): comp_radius = property( fget=lambda x: raise_( ImportError("Can't use HoleM57 method comp_radius: " + str(comp_radius)) ) ) else: comp_radius = comp_radius # cf Methods.Slot.HoleM57.comp_surface_magnet_id if isinstance(comp_surface_magnet_id, ImportError): comp_surface_magnet_id = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method comp_surface_magnet_id: " + str(comp_surface_magnet_id) ) ) ) else: comp_surface_magnet_id = comp_surface_magnet_id # cf Methods.Slot.HoleM57.has_magnet if isinstance(has_magnet, ImportError): has_magnet = property( fget=lambda x: raise_( ImportError("Can't use HoleM57 method has_magnet: " + str(has_magnet)) ) ) else: has_magnet = has_magnet # cf Methods.Slot.HoleM57.plot_schematics if isinstance(plot_schematics, ImportError): plot_schematics = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method plot_schematics: " + str(plot_schematics) ) ) ) else: plot_schematics = plot_schematics # cf Methods.Slot.HoleM57.remove_magnet if isinstance(remove_magnet, ImportError): remove_magnet = property( fget=lambda x: raise_( ImportError( "Can't use HoleM57 method remove_magnet: " + str(remove_magnet) ) ) ) else: remove_magnet = remove_magnet # generic save method is available in all object save = save # get_logger method is available in all object get_logger = get_logger def __init__( self, W0=0.2, H1=0, W1=0.013, H2=0.02, W2=0.01, W3=0.01, W4=0.01, magnet_0=-1, magnet_1=-1, Zh=36, mat_void=-1, magnetization_dict_offset=None, Alpha0=0, init_dict=None, init_str=None, ): """Constructor of the class. Can be use in three ways : - __init__ (arg1 = 1, arg3 = 5) every parameters have name and default values for pyleecan type, -1 will call the default constructor - __init__ (init_dict = d) d must be a dictionary with property names as keys - __init__ (init_str = s) s must be a string s is the file path to load ndarray or list can be given for Vector and Matrix object or dict can be given for pyleecan Object""" if init_str is not None: # Load from a file init_dict = load_init_dict(init_str)[1] if init_dict is not None: # Initialisation by dict assert type(init_dict) is dict # Overwrite default value with init_dict content if "W0" in list(init_dict.keys()): W0 = init_dict["W0"] if "H1" in list(init_dict.keys()): H1 = init_dict["H1"] if "W1" in list(init_dict.keys()): W1 = init_dict["W1"] if "H2" in list(init_dict.keys()): H2 = init_dict["H2"] if "W2" in list(init_dict.keys()): W2 = init_dict["W2"] if "W3" in list(init_dict.keys()): W3 = init_dict["W3"] if "W4" in list(init_dict.keys()): W4 = init_dict["W4"] if "magnet_0" in list(init_dict.keys()): magnet_0 = init_dict["magnet_0"] if "magnet_1" in list(init_dict.keys()): magnet_1 = init_dict["magnet_1"] if "Zh" in list(init_dict.keys()): Zh = init_dict["Zh"] if "mat_void" in list(init_dict.keys()): mat_void = init_dict["mat_void"] if "magnetization_dict_offset" in list(init_dict.keys()): magnetization_dict_offset = init_dict["magnetization_dict_offset"] if "Alpha0" in list(init_dict.keys()): Alpha0 = init_dict["Alpha0"] # Set the properties (value check and convertion are done in setter) self.W0 = W0 self.H1 = H1 self.W1 = W1 self.H2 = H2 self.W2 = W2 self.W3 = W3 self.W4 = W4 self.magnet_0 = magnet_0 self.magnet_1 = magnet_1 # Call HoleMag init super(HoleM57, self).__init__( Zh=Zh, mat_void=mat_void, magnetization_dict_offset=magnetization_dict_offset, Alpha0=Alpha0, ) # The class is frozen (in HoleMag init), for now it's impossible to # add new properties def __str__(self): """Convert this object in a readeable string (for print)""" HoleM57_str = "" # Get the properties inherited from HoleMag HoleM57_str += super(HoleM57, self).__str__() HoleM57_str += "W0 = " + str(self.W0) + linesep HoleM57_str += "H1 = " + str(self.H1) + linesep HoleM57_str += "W1 = " + str(self.W1) + linesep HoleM57_str += "H2 = " + str(self.H2) + linesep HoleM57_str += "W2 = " + str(self.W2) + linesep HoleM57_str += "W3 = " + str(self.W3) + linesep HoleM57_str += "W4 = " + str(self.W4) + linesep if self.magnet_0 is not None: tmp = self.magnet_0.__str__().replace(linesep, linesep + "\t").rstrip("\t") HoleM57_str += "magnet_0 = " + tmp else: HoleM57_str += "magnet_0 = None" + linesep + linesep if self.magnet_1 is not None: tmp = self.magnet_1.__str__().replace(linesep, linesep + "\t").rstrip("\t") HoleM57_str += "magnet_1 = " + tmp else: HoleM57_str += "magnet_1 = None" + linesep + linesep return HoleM57_str def __eq__(self, other): """Compare two objects (skip parent)""" if type(other) != type(self): return False # Check the properties inherited from HoleMag if not super(HoleM57, self).__eq__(other): return False if other.W0 != self.W0: return False if other.H1 != self.H1: return False if other.W1 != self.W1: return False if other.H2 != self.H2: return False if other.W2 != self.W2: return False if other.W3 != self.W3: return False if other.W4 != self.W4: return False if other.magnet_0 != self.magnet_0: return False if other.magnet_1 != self.magnet_1: return False return True
[docs] def compare(self, other, name="self", ignore_list=None, is_add_value=False): """Compare two objects and return list of differences""" if ignore_list is None: ignore_list = list() if type(other) != type(self): return ["type(" + name + ")"] diff_list = list() # Check the properties inherited from HoleMag diff_list.extend( super(HoleM57, self).compare( other, name=name, ignore_list=ignore_list, is_add_value=is_add_value ) ) if ( other._W0 is not None and self._W0 is not None and isnan(other._W0) and isnan(self._W0) ): pass elif other._W0 != self._W0: if is_add_value: val_str = " (self=" + str(self._W0) + ", other=" + str(other._W0) + ")" diff_list.append(name + ".W0" + val_str) else: diff_list.append(name + ".W0") if ( other._H1 is not None and self._H1 is not None and isnan(other._H1) and isnan(self._H1) ): pass elif other._H1 != self._H1: if is_add_value: val_str = " (self=" + str(self._H1) + ", other=" + str(other._H1) + ")" diff_list.append(name + ".H1" + val_str) else: diff_list.append(name + ".H1") if ( other._W1 is not None and self._W1 is not None and isnan(other._W1) and isnan(self._W1) ): pass elif other._W1 != self._W1: if is_add_value: val_str = " (self=" + str(self._W1) + ", other=" + str(other._W1) + ")" diff_list.append(name + ".W1" + val_str) else: diff_list.append(name + ".W1") if ( other._H2 is not None and self._H2 is not None and isnan(other._H2) and isnan(self._H2) ): pass elif other._H2 != self._H2: if is_add_value: val_str = " (self=" + str(self._H2) + ", other=" + str(other._H2) + ")" diff_list.append(name + ".H2" + val_str) else: diff_list.append(name + ".H2") if ( other._W2 is not None and self._W2 is not None and isnan(other._W2) and isnan(self._W2) ): pass elif other._W2 != self._W2: if is_add_value: val_str = " (self=" + str(self._W2) + ", other=" + str(other._W2) + ")" diff_list.append(name + ".W2" + val_str) else: diff_list.append(name + ".W2") if ( other._W3 is not None and self._W3 is not None and isnan(other._W3) and isnan(self._W3) ): pass elif other._W3 != self._W3: if is_add_value: val_str = " (self=" + str(self._W3) + ", other=" + str(other._W3) + ")" diff_list.append(name + ".W3" + val_str) else: diff_list.append(name + ".W3") if ( other._W4 is not None and self._W4 is not None and isnan(other._W4) and isnan(self._W4) ): pass elif other._W4 != self._W4: if is_add_value: val_str = " (self=" + str(self._W4) + ", other=" + str(other._W4) + ")" diff_list.append(name + ".W4" + val_str) else: diff_list.append(name + ".W4") if (other.magnet_0 is None and self.magnet_0 is not None) or ( other.magnet_0 is not None and self.magnet_0 is None ): diff_list.append(name + ".magnet_0 None mismatch") elif self.magnet_0 is not None: diff_list.extend( self.magnet_0.compare( other.magnet_0, name=name + ".magnet_0", ignore_list=ignore_list, is_add_value=is_add_value, ) ) if (other.magnet_1 is None and self.magnet_1 is not None) or ( other.magnet_1 is not None and self.magnet_1 is None ): diff_list.append(name + ".magnet_1 None mismatch") elif self.magnet_1 is not None: diff_list.extend( self.magnet_1.compare( other.magnet_1, name=name + ".magnet_1", ignore_list=ignore_list, is_add_value=is_add_value, ) ) # Filter ignore differences diff_list = list(filter(lambda x: x not in ignore_list, diff_list)) return diff_list
def __sizeof__(self): """Return the size in memory of the object (including all subobject)""" S = 0 # Full size of the object # Get size of the properties inherited from HoleMag S += super(HoleM57, self).__sizeof__() S += getsizeof(self.W0) S += getsizeof(self.H1) S += getsizeof(self.W1) S += getsizeof(self.H2) S += getsizeof(self.W2) S += getsizeof(self.W3) S += getsizeof(self.W4) S += getsizeof(self.magnet_0) S += getsizeof(self.magnet_1) return S
[docs] def as_dict(self, type_handle_ndarray=0, keep_function=False, **kwargs): """ Convert this object in a json serializable dict (can be use in __init__). type_handle_ndarray: int How to handle ndarray (0: tolist, 1: copy, 2: nothing) keep_function : bool True to keep the function object, else return str Optional keyword input parameter is for internal use only and may prevent json serializability. """ # Get the properties inherited from HoleMag HoleM57_dict = super(HoleM57, self).as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) HoleM57_dict["W0"] = self.W0 HoleM57_dict["H1"] = self.H1 HoleM57_dict["W1"] = self.W1 HoleM57_dict["H2"] = self.H2 HoleM57_dict["W2"] = self.W2 HoleM57_dict["W3"] = self.W3 HoleM57_dict["W4"] = self.W4 if self.magnet_0 is None: HoleM57_dict["magnet_0"] = None else: HoleM57_dict["magnet_0"] = self.magnet_0.as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) if self.magnet_1 is None: HoleM57_dict["magnet_1"] = None else: HoleM57_dict["magnet_1"] = self.magnet_1.as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) # The class name is added to the dict for deserialisation purpose # Overwrite the mother class name HoleM57_dict["__class__"] = "HoleM57" return HoleM57_dict
[docs] def copy(self): """Creates a deepcopy of the object""" # Handle deepcopy of all the properties W0_val = self.W0 H1_val = self.H1 W1_val = self.W1 H2_val = self.H2 W2_val = self.W2 W3_val = self.W3 W4_val = self.W4 if self.magnet_0 is None: magnet_0_val = None else: magnet_0_val = self.magnet_0.copy() if self.magnet_1 is None: magnet_1_val = None else: magnet_1_val = self.magnet_1.copy() Zh_val = self.Zh if self.mat_void is None: mat_void_val = None else: mat_void_val = self.mat_void.copy() if self.magnetization_dict_offset is None: magnetization_dict_offset_val = None else: magnetization_dict_offset_val = self.magnetization_dict_offset.copy() Alpha0_val = self.Alpha0 # Creates new object of the same type with the copied properties obj_copy = type(self)( W0=W0_val, H1=H1_val, W1=W1_val, H2=H2_val, W2=W2_val, W3=W3_val, W4=W4_val, magnet_0=magnet_0_val, magnet_1=magnet_1_val, Zh=Zh_val, mat_void=mat_void_val, magnetization_dict_offset=magnetization_dict_offset_val, Alpha0=Alpha0_val, ) return obj_copy
def _set_None(self): """Set all the properties to None (except pyleecan object)""" self.W0 = None self.H1 = None self.W1 = None self.H2 = None self.W2 = None self.W3 = None self.W4 = None if self.magnet_0 is not None: self.magnet_0._set_None() if self.magnet_1 is not None: self.magnet_1._set_None() # Set to None the properties inherited from HoleMag super(HoleM57, self)._set_None() def _get_W0(self): """getter of W0""" return self._W0 def _set_W0(self, value): """setter of W0""" check_var("W0", value, "float", Vmin=0, Vmax=3.15) self._W0 = value W0 = property( fget=_get_W0, fset=_set_W0, doc=u"""V angle :Type: float :min: 0 :max: 3.15 """, ) def _get_H1(self): """getter of H1""" return self._H1 def _set_H1(self, value): """setter of H1""" check_var("H1", value, "float", Vmin=0) self._H1 = value H1 = property( fget=_get_H1, fset=_set_H1, doc=u"""Distance from the lamination Bore :Type: float :min: 0 """, ) def _get_W1(self): """getter of W1""" return self._W1 def _set_W1(self, value): """setter of W1""" check_var("W1", value, "float", Vmin=0) self._W1 = value W1 = property( fget=_get_W1, fset=_set_W1, doc=u"""Tooth width (at V bottom) :Type: float :min: 0 """, ) def _get_H2(self): """getter of H2""" return self._H2 def _set_H2(self, value): """setter of H2""" check_var("H2", value, "float", Vmin=0) self._H2 = value H2 = property( fget=_get_H2, fset=_set_H2, doc=u"""Magnet height :Type: float :min: 0 """, ) def _get_W2(self): """getter of W2""" return self._W2 def _set_W2(self, value): """setter of W2""" check_var("W2", value, "float", Vmin=0) self._W2 = value W2 = property( fget=_get_W2, fset=_set_W2, doc=u"""Distance Magnet to top of the V :Type: float :min: 0 """, ) def _get_W3(self): """getter of W3""" return self._W3 def _set_W3(self, value): """setter of W3""" check_var("W3", value, "float", Vmin=0) self._W3 = value W3 = property( fget=_get_W3, fset=_set_W3, doc=u"""Tooth width (at V top) :Type: float :min: 0 """, ) def _get_W4(self): """getter of W4""" return self._W4 def _set_W4(self, value): """setter of W4""" check_var("W4", value, "float", Vmin=0) self._W4 = value W4 = property( fget=_get_W4, fset=_set_W4, doc=u"""Magnet Width :Type: float :min: 0 """, ) def _get_magnet_0(self): """getter of magnet_0""" return self._magnet_0 def _set_magnet_0(self, value): """setter of magnet_0""" if isinstance(value, str): # Load from file try: value = load_init_dict(value)[1] except Exception as e: self.get_logger().error( "Error while loading " + value + ", setting None instead" ) value = None if isinstance(value, dict) and "__class__" in value: class_obj = import_class( "pyleecan.Classes", value.get("__class__"), "magnet_0" ) value = class_obj(init_dict=value) elif type(value) is int and value == -1: # Default constructor Magnet = import_class("pyleecan.Classes", "Magnet", "magnet_0") value = Magnet() check_var("magnet_0", value, "Magnet") self._magnet_0 = value if self._magnet_0 is not None: self._magnet_0.parent = self magnet_0 = property( fget=_get_magnet_0, fset=_set_magnet_0, doc=u"""First Magnet :Type: Magnet """, ) def _get_magnet_1(self): """getter of magnet_1""" return self._magnet_1 def _set_magnet_1(self, value): """setter of magnet_1""" if isinstance(value, str): # Load from file try: value = load_init_dict(value)[1] except Exception as e: self.get_logger().error( "Error while loading " + value + ", setting None instead" ) value = None if isinstance(value, dict) and "__class__" in value: class_obj = import_class( "pyleecan.Classes", value.get("__class__"), "magnet_1" ) value = class_obj(init_dict=value) elif type(value) is int and value == -1: # Default constructor Magnet = import_class("pyleecan.Classes", "Magnet", "magnet_1") value = Magnet() check_var("magnet_1", value, "Magnet") self._magnet_1 = value if self._magnet_1 is not None: self._magnet_1.parent = self magnet_1 = property( fget=_get_magnet_1, fset=_set_magnet_1, doc=u"""Second Magnet :Type: Magnet """, )