import matplotlib.pyplot as plt
from numpy import pi, exp
from ....Classes.Arc1 import Arc1
from ....Classes.LamSlot import LamSlot
from ....Classes.Segment import Segment
from ....definitions import config_dict
from ....Functions.Plot import (
ARROW_COLOR,
ARROW_WIDTH,
MAIN_LINE_COLOR,
MAIN_LINE_STYLE,
MAIN_LINE_WIDTH,
P_FONT_SIZE,
SC_FONT_SIZE,
SC_LINE_COLOR,
SC_LINE_STYLE,
SC_LINE_WIDTH,
TEXT_BOX,
)
from ....Methods import ParentMissingError
MAGNET_COLOR = config_dict["PLOT"]["COLOR_DICT"]["MAGNET_COLOR"]
[docs]def plot_schematics(
self,
is_default=False,
is_enforce_default_H0_bore=None,
is_add_point_label=False,
is_add_schematics=True,
is_add_main_line=True,
type_add_active=0,
save_path=None,
is_show_fig=True,
fig=None,
ax=None,
):
"""Plot the schematics of the slot
Parameters
----------
self : SlotCirc
A SlotCirc object
is_default : int
0: current slot values, 1: default internal rotor schematics, 2: default external stator schematics
is_enforce_default_H0_bore : bool
To enforce is_H0_bore on default slot (is_default=True only), None use default value
is_add_point_label : bool
True to display the name of the points (Z1, Z2....)
is_add_schematics : bool
True to display the schematics information (W0, H0...)
is_add_main_line : bool
True to display "main lines" (slot opening and 0x axis)
type_add_active : int
0: No active surface, 1: active surface as winding, 2: active surface as magnet
save_path : str
full path including folder, name and extension of the file to save if save_path is not None
is_show_fig : bool
To call show at the end of the method
fig : Matplotlib.figure.Figure
existing figure to use if None create a new one
ax : Matplotlib.axes.Axes object
Axis on which to plot the data
Returns
-------
fig : Matplotlib.figure.Figure
Figure containing the schematics
ax : Matplotlib.axes.Axes object
Axis containing the schematics
"""
# Use some default parameter
if is_default:
slot = type(self)(Zs=8, H0=10e-3, W0=45e-3, is_H0_bore=True)
if is_enforce_default_H0_bore is not None:
slot.is_H0_bore = is_enforce_default_H0_bore
if is_default == 1: # Internal rotor schematics
lam = LamSlot(
Rint=0.1, Rext=0.135, is_internal=True, is_stator=False, slot=slot
)
else: # External stator schematics
lam = LamSlot(
Rint=0.1, Rext=0.135, is_internal=False, is_stator=True, slot=slot
)
return slot.plot_schematics(
is_default=False,
is_add_point_label=is_add_point_label,
is_add_schematics=is_add_schematics,
is_add_main_line=is_add_main_line,
type_add_active=type_add_active,
save_path=save_path,
is_show_fig=is_show_fig,
fig=fig,
ax=ax,
)
else:
# Getting the main plot
if self.parent is None:
raise ParentMissingError("Error: The slot is not inside a Lamination")
lam = self.parent
fig, ax = lam.plot(
alpha=pi / self.Zs, is_show_fig=False, fig=fig, ax=ax
) # center slot on Ox axis
point_dict = self._comp_point_coordinate()
# Adding point label
if is_add_point_label:
for name, Z in point_dict.items():
ax.text(
Z.real,
Z.imag,
name,
fontsize=P_FONT_SIZE,
bbox=TEXT_BOX,
)
# Adding schematics
if is_add_schematics:
# W0
line = Segment(point_dict["Z1"], point_dict["Z2"])
if self.is_outwards():
offset_label = -1 * self.H0 * 0.9 + 1j * 0.3 * self.W0
else:
offset_label = self.H0 * 0.3 + 1j * 0.3 * self.W0
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W0",
offset_label=offset_label,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# H0
line = Segment(point_dict["ZH"], point_dict["ZM"])
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="H0",
offset_label=1j * -0.15 * self.W0,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
if is_add_main_line:
# Top arc
line = Arc1(
begin=point_dict["Z1"],
end=point_dict["Z2"],
radius=self.get_Rbo(),
is_trigo_direction=True,
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
# Zooming and cleaning
W = self.W0 / 2 * 1.4
Rint, Rext = self.comp_radius()
ax.axis("equal")
ax.set_xlim(Rint, Rext)
ax.set_ylim(-W, W)
manager = plt.get_current_fig_manager()
if manager is not None:
manager.set_window_title(type(self).__name__ + " Schematics")
ax.set_title("")
ax.get_legend().remove()
ax.set_axis_off()
# Save / Show
if save_path is not None:
fig.savefig(save_path)
plt.close(fig=fig)
if is_show_fig:
fig.show()
return fig, ax
