import matplotlib.pyplot as plt
from numpy import pi, exp, angle
from ....Classes.Arc1 import Arc1
from ....Classes.LamHole import LamHole
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,
plot_quote,
)
from ....Methods import ParentMissingError
MAGNET_COLOR = config_dict["PLOT"]["COLOR_DICT"]["MAGNET_COLOR"]
[docs]def plot_schematics(
self,
is_default=False,
is_add_point_label=False,
is_add_schematics=True,
is_add_main_line=True,
type_add_active=True,
save_path=None,
is_show_fig=True,
fig=None,
ax=None,
):
"""Plot the schematics of the slot
Parameters
----------
self : HoleM51
A HoleM51 object
is_default : bool
True: plot default schematics, else use current slot values
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:
hole = type(self)(
Zh=8,
W0=0.016,
W1=pi / 6,
W2=0.004,
W3=0.01,
W4=0.004,
W5=0.008,
W6=0.004,
W7=0.008,
H0=0.01196,
H1=0.0025,
H2=0.0055,
)
lam = LamHole(
Rint=45e-3 / 2, Rext=81.5e-3, is_internal=True, is_stator=False, hole=[hole]
)
return hole.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,
)
elif type_add_active == 0:
# Remove magnets
lam = self.parent.copy()
lam.hole[0].remove_magnet()
return lam.hole[0].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=2,
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 hole is not inside a Lamination")
lam = self.parent
alpha = pi / 2 # To rotate the schematics
fig, ax = lam.plot(
alpha=pi / self.Zh + alpha,
is_show_fig=False,
is_lam_only=type_add_active == 0,
fig=fig,
ax=ax,
) # center hole on Ox axis
sp = 2 * pi / self.Zh
Rbo = self.get_Rbo()
point_dict = self._comp_point_coordinate()
# Adding point label
if is_add_point_label:
for name, Z in point_dict.items():
Z = Z * exp(1j * alpha)
ax.text(
Z.real,
Z.imag,
name,
fontsize=P_FONT_SIZE,
bbox=TEXT_BOX,
)
# Adding schematics
if is_add_schematics:
# W0
Zlim1 = point_dict["Z2"] + self.H0 * 0.2
Zlim2 = point_dict["Z25"] + self.H0 * 0.2
plot_quote(
point_dict["Z2"] * exp(1j * alpha),
Zlim1 * exp(1j * alpha),
Zlim2 * exp(1j * alpha),
point_dict["Z25"] * exp(1j * alpha),
offset_label=(1 + 1j) * self.H2 * 0.3,
fig=fig,
ax=ax,
label="W0",
)
# W1
R = Rbo + self.H0 * 0.1
line = Arc1(
begin=R * exp(-1j * self.W1 / 2) * exp(1j * alpha),
end=R * exp(1j * self.W1 / 2) * exp(1j * alpha),
radius=R,
is_trigo_direction=True,
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W1",
offset_label=1j * self.H2 * 0.3,
fontsize=SC_FONT_SIZE,
)
# W2
line = Segment(
(point_dict["Z2"] + point_dict["Z12"]) / 2 * exp(1j * alpha),
(point_dict["Z1"] + point_dict["Z13"]) / 2 * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W2",
offset_label=-self.H2 * 0.2 - 1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# W3
line = Segment(
point_dict["Z13"] * exp(1j * alpha), point_dict["Z14"] * exp(1j * alpha)
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W3",
offset_label=-1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# W4
line = Segment(
(point_dict["Z3"] + point_dict["Z9"]) / 2 * exp(1j * alpha),
(point_dict["Z10"] + point_dict["Z2"]) / 2 * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W4",
offset_label=-self.H2 * 0.2 - 1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# W5
line = Segment(
point_dict["Z9"] * exp(1j * alpha), point_dict["Z8"] * exp(1j * alpha)
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W5",
offset_label=-self.H2 * 0.2 - 1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# W6
line = Segment(
(point_dict["Z25"] + point_dict["Z17"]) / 2 * exp(1j * alpha),
(point_dict["Z24"] + point_dict["Z18"]) / 2 * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W6",
offset_label=-1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# W7
line = Segment(
point_dict["Z18"] * exp(1j * alpha), point_dict["Z19"] * exp(1j * alpha)
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="W7",
offset_label=-1j * self.H2 * 0.5,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# H0
line = Segment(
point_dict["Z1"].real * exp(1j * alpha), Rbo * exp(1j * alpha)
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="H0",
offset_label=self.H2 * 0.2 + 1j * self.H0 * 0.2,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# H1
A = angle(point_dict["Z22"])
line = Segment(
Rbo * exp(1j * A) * exp(1j * alpha),
(Rbo - self.H1) * exp(1j * A) * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="H1",
offset_label=self.H2 * 0.2,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
# H2
line = Segment(
point_dict["Z5"] * exp(1j * alpha), point_dict["Z7"] * exp(1j * alpha)
)
line.plot(
fig=fig,
ax=ax,
color=ARROW_COLOR,
linewidth=ARROW_WIDTH,
label="H2",
offset_label=-self.H2 * 0.6,
is_arrow=True,
fontsize=SC_FONT_SIZE,
)
if is_add_main_line:
# Ox axis
line = Segment(0, lam.Rext * 1.5 * exp(1j * alpha))
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
# W1 lines
line = Segment(
0,
abs(point_dict["Z5"])
* 1.5
* exp(1j * angle(point_dict["Z5"]))
* exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
0,
abs(point_dict["Z22"])
* 1.5
* exp(1j * angle(point_dict["Z22"]))
* exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
# H1 radius
line = Arc1(
begin=(Rbo - self.H1) * exp(-1j * pi / 2 * 0.9) * exp(1j * alpha),
end=(Rbo - self.H1) * exp(1j * pi / 2 * 0.9) * exp(1j * alpha),
radius=Rbo - self.H1,
is_trigo_direction=True,
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
# Triangle lines
line = Segment(
point_dict["Z7"] * exp(1j * alpha),
point_dict["Z5"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
point_dict["Z2"] * exp(1j * alpha),
point_dict["Z10"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
point_dict["Z2"] * exp(1j * alpha),
point_dict["Z12"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
point_dict["Z25"] * exp(1j * alpha),
point_dict["Z15"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
point_dict["Z25"] * exp(1j * alpha),
point_dict["Z17"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
line = Segment(
point_dict["Z22"] * exp(1j * alpha),
point_dict["Z20"] * exp(1j * alpha),
)
line.plot(
fig=fig,
ax=ax,
color=MAIN_LINE_COLOR,
linestyle=MAIN_LINE_STYLE,
linewidth=MAIN_LINE_WIDTH,
)
# Zooming and cleaning
W = (point_dict["Z22"] * exp(1j * alpha)).real * 1.2
Rint = (point_dict["Z13"] * exp(1j * alpha)).imag * 0.9
Rext = self.parent.Rext * 1.1
ax.axis("equal")
ax.set_ylim(Rint, Rext)
ax.set_xlim(-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
