Source code for pyleecan.Classes.Interpolation

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

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 ._frozen import FrozenClass

from numpy import isnan
from ._check import InitUnKnowClassError


[docs]class Interpolation(FrozenClass): """Store shape functions""" VERSION = 1 # 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, ref_cell=None, gauss_point=None, scalar_product=None, 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 "ref_cell" in list(init_dict.keys()): ref_cell = init_dict["ref_cell"] if "gauss_point" in list(init_dict.keys()): gauss_point = init_dict["gauss_point"] if "scalar_product" in list(init_dict.keys()): scalar_product = init_dict["scalar_product"] # Set the properties (value check and convertion are done in setter) self.parent = None self.ref_cell = ref_cell self.gauss_point = gauss_point self.scalar_product = scalar_product # The class is frozen, for now it's impossible to add new properties self._freeze() def __str__(self): """Convert this object in a readeable string (for print)""" Interpolation_str = "" if self.parent is None: Interpolation_str += "parent = None " + linesep else: Interpolation_str += ( "parent = " + str(type(self.parent)) + " object" + linesep ) if self.ref_cell is not None: tmp = self.ref_cell.__str__().replace(linesep, linesep + "\t").rstrip("\t") Interpolation_str += "ref_cell = " + tmp else: Interpolation_str += "ref_cell = None" + linesep + linesep if self.gauss_point is not None: tmp = ( self.gauss_point.__str__().replace(linesep, linesep + "\t").rstrip("\t") ) Interpolation_str += "gauss_point = " + tmp else: Interpolation_str += "gauss_point = None" + linesep + linesep if self.scalar_product is not None: tmp = ( self.scalar_product.__str__() .replace(linesep, linesep + "\t") .rstrip("\t") ) Interpolation_str += "scalar_product = " + tmp else: Interpolation_str += "scalar_product = None" + linesep + linesep return Interpolation_str def __eq__(self, other): """Compare two objects (skip parent)""" if type(other) != type(self): return False if other.ref_cell != self.ref_cell: return False if other.gauss_point != self.gauss_point: return False if other.scalar_product != self.scalar_product: 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() if (other.ref_cell is None and self.ref_cell is not None) or ( other.ref_cell is not None and self.ref_cell is None ): diff_list.append(name + ".ref_cell None mismatch") elif self.ref_cell is not None: diff_list.extend( self.ref_cell.compare( other.ref_cell, name=name + ".ref_cell", ignore_list=ignore_list, is_add_value=is_add_value, ) ) if (other.gauss_point is None and self.gauss_point is not None) or ( other.gauss_point is not None and self.gauss_point is None ): diff_list.append(name + ".gauss_point None mismatch") elif self.gauss_point is not None: diff_list.extend( self.gauss_point.compare( other.gauss_point, name=name + ".gauss_point", ignore_list=ignore_list, is_add_value=is_add_value, ) ) if (other.scalar_product is None and self.scalar_product is not None) or ( other.scalar_product is not None and self.scalar_product is None ): diff_list.append(name + ".scalar_product None mismatch") elif self.scalar_product is not None: diff_list.extend( self.scalar_product.compare( other.scalar_product, name=name + ".scalar_product", 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 S += getsizeof(self.ref_cell) S += getsizeof(self.gauss_point) S += getsizeof(self.scalar_product) 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. """ Interpolation_dict = dict() if self.ref_cell is None: Interpolation_dict["ref_cell"] = None else: Interpolation_dict["ref_cell"] = self.ref_cell.as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) if self.gauss_point is None: Interpolation_dict["gauss_point"] = None else: Interpolation_dict["gauss_point"] = self.gauss_point.as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) if self.scalar_product is None: Interpolation_dict["scalar_product"] = None else: Interpolation_dict["scalar_product"] = self.scalar_product.as_dict( type_handle_ndarray=type_handle_ndarray, keep_function=keep_function, **kwargs ) # The class name is added to the dict for deserialisation purpose Interpolation_dict["__class__"] = "Interpolation" return Interpolation_dict
[docs] def copy(self): """Creates a deepcopy of the object""" # Handle deepcopy of all the properties if self.ref_cell is None: ref_cell_val = None else: ref_cell_val = self.ref_cell.copy() if self.gauss_point is None: gauss_point_val = None else: gauss_point_val = self.gauss_point.copy() if self.scalar_product is None: scalar_product_val = None else: scalar_product_val = self.scalar_product.copy() # Creates new object of the same type with the copied properties obj_copy = type(self)( ref_cell=ref_cell_val, gauss_point=gauss_point_val, scalar_product=scalar_product_val, ) return obj_copy
def _set_None(self): """Set all the properties to None (except pyleecan object)""" if self.ref_cell is not None: self.ref_cell._set_None() if self.gauss_point is not None: self.gauss_point._set_None() if self.scalar_product is not None: self.scalar_product._set_None() def _get_ref_cell(self): """getter of ref_cell""" return self._ref_cell def _set_ref_cell(self, value): """setter of ref_cell""" 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__"), "ref_cell" ) value = class_obj(init_dict=value) elif type(value) is int and value == -1: # Default constructor RefCell = import_class("pyleecan.Classes", "RefCell", "ref_cell") value = RefCell() check_var("ref_cell", value, "RefCell") self._ref_cell = value if self._ref_cell is not None: self._ref_cell.parent = self ref_cell = property( fget=_get_ref_cell, fset=_set_ref_cell, doc=u""" :Type: RefCell """, ) def _get_gauss_point(self): """getter of gauss_point""" return self._gauss_point def _set_gauss_point(self, value): """setter of gauss_point""" 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__"), "gauss_point" ) value = class_obj(init_dict=value) elif type(value) is int and value == -1: # Default constructor GaussPoint = import_class("pyleecan.Classes", "GaussPoint", "gauss_point") value = GaussPoint() check_var("gauss_point", value, "GaussPoint") self._gauss_point = value if self._gauss_point is not None: self._gauss_point.parent = self gauss_point = property( fget=_get_gauss_point, fset=_set_gauss_point, doc=u""" :Type: GaussPoint """, ) def _get_scalar_product(self): """getter of scalar_product""" return self._scalar_product def _set_scalar_product(self, value): """setter of scalar_product""" 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__"), "scalar_product" ) value = class_obj(init_dict=value) elif type(value) is int and value == -1: # Default constructor ScalarProduct = import_class( "pyleecan.Classes", "ScalarProduct", "scalar_product" ) value = ScalarProduct() check_var("scalar_product", value, "ScalarProduct") self._scalar_product = value if self._scalar_product is not None: self._scalar_product.parent = self scalar_product = property( fget=_get_scalar_product, fset=_set_scalar_product, doc=u""" :Type: ScalarProduct """, )