Source code for pyleecan.Methods.Slot.SlotM16.build_geometry

# -*- coding: utf-8 -*-
from ....Classes.Segment import Segment


def build_geometry(self):
    """Compute the curve (Line) needed to plot the Slot.
    The ending point of a curve is the starting point of the next curve in
    the list

    Parameters
    ----------
    self : SlotM16
        A SlotM16 object

    Returns
    -------
    curve_list: list
        A list of 11 Segment

    """

    point_dict = self._comp_point_coordinate()

    # Creation of curve
    curve_list = list()
    if self.H0 > 0:
        curve_list.append(Segment(point_dict["Z1"], point_dict["Z2"]))
    curve_list.append(Segment(point_dict["Z2"], point_dict["Z3"]))
    if self.H1 > 0:
        curve_list.append(Segment(point_dict["Z3"], point_dict["Z4"]))
    curve_list.append(Segment(point_dict["Z4"], point_dict["Z5"]))
    if self.H1 > 0:
        curve_list.append(Segment(point_dict["Z5"], point_dict["Z6"]))
    curve_list.append(Segment(point_dict["Z6"], point_dict["Z7"]))
    if self.H0 > 0:
        curve_list.append(Segment(point_dict["Z7"], point_dict["Z8"]))

    return curve_list