Source code for pyleecan.Methods.Machine.MagnetType10.build_geometry

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

from numpy import abs, angle, array, exp

from ....Classes.SurfLine import SurfLine
from ....Classes.Segment import Segment
from ....Methods import ParentMissingError


def build_geometry(self, alpha=0, delta=0, is_simplified=False):
    """Compute the curve (Line) needed to plot the Magnet.
    The list represents a closed surface.
    The ending point of a curve is always the starting point of the next curve
    in the list

    Parameters
    ----------
    self : MagnetType10
        A MagnetType10 object
    alpha : float
        Angle for rotation [rad]
    delta : complex
        Complex value for translation
    is_simplified: bool
        True to avoid line superposition

    Returns
    -------
    surf_list : list
        list of surfaces needed to draw the magnet

    """
    # defining label for type_magnetization
    if self.type_magnetization == 0:
        t_p = "Radial"
    else:
        t_p = "Parallel"

    if self.parent is not None:
        (Z1, Z2) = self.parent.get_point_bottom()
        H0 = self.parent.H0
        W0 = self.parent.comp_W0m()
    else:
        raise ParentMissingError(
            "Error: The magnet object is not inside a " + "slot object"
        )

    # comp point coordinate (in complex)
    if W0 > self.Wmag:  # The magnet is smaller than the slot => center the mag
        Z1 = Z1 + 1j * (W0 - self.Wmag) / 2
        Z2 = Z2 - 1j * (W0 - self.Wmag) / 2

    if self.is_outwards():
        Z3 = Z1 - self.Hmag
        Zs3 = Z1 - H0
        Z4 = Z2 - self.Hmag
        Zs4 = Z2 - H0
    else:
        Z3 = Z1 + self.Hmag
        Zs3 = Z1 + H0
        Z4 = Z2 + self.Hmag
        Zs4 = Z2 + H0

    # Creation of curve
    curve_list = list()
    if is_simplified and W0 > self.Wmag:
        curve_list.append(Segment(Z1, Z3))
    elif is_simplified and H0 < self.Hmag:
        curve_list.append(Segment(Zs3, Z3))
    elif not is_simplified:
        curve_list.append(Segment(Z1, Z3))

    curve_list.append(Segment(Z3, Z4))

    if is_simplified and W0 > self.Wmag:
        curve_list.append(Segment(Z4, Z2))
    elif is_simplified and H0 < self.Hmag:
        curve_list.append(Segment(Z4, Zs4))
    elif not is_simplified:
        curve_list.append(Segment(Z4, Z2))

    if not is_simplified:
        curve_list.append(Segment(Z2, Z1))

    # Create the surface
    surf_list = list()
    surf_list.append(
        SurfLine(
            line_list=curve_list,
            label="MagnetRotor" + t_p + "_N_R0_T0_S0",
            point_ref=(Z1 + Z2 + Z3 + Z4) / 4,
        )
    )

    # Apply the transformation
    for surf in surf_list:
        surf.rotate(alpha)
        surf.translate(delta)

    return surf_list