from matplotlib.patches import Patch, Polygon
from numpy import array, exp, pi
from ....definitions import config_dict
from ....Classes.Material import Material
from ....Functions.init_fig import init_fig
from ....Functions.Plot import dict_2D
from ....Functions.Winding.gen_phase_list import gen_name
PHASE_COLORS = config_dict["PLOT"]["COLOR_DICT"]["PHASE_COLORS"]
if "WEDGE_COLOR" not in config_dict["PLOT"]["COLOR_DICT"]:
config_dict["PLOT"]["COLOR_DICT"]["WEDGE_COLOR"] = "y"
WEDGE_COLOR = config_dict["PLOT"]["COLOR_DICT"]["WEDGE_COLOR"]
[docs]def plot_active(
self,
wind_mat=None,
fig=None,
ax=None,
is_bar=False,
is_show_fig=True,
enforced_default_color=None,
alpha=0,
delta=0,
is_add_wedge=False,
):
"""Plot the active area of the lamination according to the wind_mat
Parameters
----------
self : Slot
A Slot object
wind_mat : numpy.ndarray
A matrix [Nrad,Ntan,Zs,qs] representing the active (Default value = None)
fig :
if None, open a new fig and plot, else add to the current
one (Default value = None)
is_bar : bool
To adapt the legend text for squirrel cage bar (Default value = False)
is_show_fig : bool
To call show at the end of the method
enforced_default_color : str
If not None enforce the active color (when wind_mat is None)
alpha : float
Angle for rotation (Default value = 0) [rad]
delta : Complex
complex for translation (Default value = 0)
is_add_wedge : bool
True to add the wedges surfaces
Returns
-------
None
"""
if enforced_default_color is None:
enforced_default_color = PHASE_COLORS[0]
if wind_mat is None: # Default : Only one zone monocolor
Nrad, Ntan, qs = 1, 1, 1
Zs = self.Zs
else:
(Nrad, Ntan, Zs, qs) = wind_mat.shape
qs_name = gen_name(qs)
surf_list = self.build_geometry_active(Nrad, Ntan, alpha=alpha, delta=delta)
patches = list()
for ii in range(len(surf_list)):
# Compute the coordinate for one zone
point_list = list()
for curve in surf_list[ii].get_lines():
point_list.extend(curve.discretize().tolist())
point_list = array(point_list)
for jj in range(Zs):
if wind_mat is None or len(surf_list) != Ntan * Nrad:
x, y = point_list.real, point_list.imag
patches.append(Polygon(list(zip(x, y)), color=enforced_default_color))
else:
# print "Nrad, Ntan, Zs : "+str((ii%Nrad,ii/Nrad,jj))
color = get_color(wind_mat, ii % Nrad, ii // Nrad, jj)
x, y = point_list.real, point_list.imag
patches.append(Polygon(list(zip(x, y)), color=color))
point_list = point_list * exp(1j * (2 * pi) / self.Zs)
# Display the result
(fig, ax, patch_leg, label_leg) = init_fig(fig, ax=ax)
ax.set_xlabel("[m]")
ax.set_ylabel("[m]")
ax.set_title("Winding Pattern")
# Add wedges
if is_add_wedge:
# Add mp blanck wedge wedge for plot
to_clean = False
if self.wedge_mat is None:
self.wedge_mat = Material()
to_clean = True
wedge_surf_list = self.get_surface_wedges()
for surf in wedge_surf_list:
for ii in range(Zs):
surf2 = surf.copy()
surf2.rotate(angle=ii * (2 * pi) / self.Zs)
surf2.plot(fig=fig, ax=ax, color=WEDGE_COLOR, is_show_fig=False)
# Remove tmp blanck wedge
if to_clean:
self.wedge_mat = None
# Add the magnet to the fig
for patch in patches:
ax.add_patch(patch)
# Axis Setup
ax.axis("equal")
Rbo = self.get_Rbo()
Lim = Rbo * 1.2
ax.set_xlim(-Lim, Lim)
ax.set_ylim(-Lim, Lim)
# Legend setup
if wind_mat is None or len(surf_list) != Ntan * Nrad:
# No winding matrix => Only one zone
if not is_bar and not ("Winding" in label_leg):
# Avoid adding twice the same label
patch_leg.append(Patch(color=PHASE_COLORS[0]))
label_leg.append("Winding")
elif is_bar and not ("Rotor bar" in label_leg):
# Avoid adding twice the same label
patch_leg.append(Patch(color=PHASE_COLORS[0]))
label_leg.append("Rotor bar")
else: # Add every phase to the legend
for ii in range(qs):
if not ("Phase " + qs_name[ii] in label_leg):
# Avoid adding twice the same label
index = ii % len(PHASE_COLORS)
patch_leg.append(Patch(color=PHASE_COLORS[index]))
label_leg.append("Phase " + qs_name[ii])
ax.legend(
patch_leg,
label_leg,
prop={"family": dict_2D["font_name"], "size": dict_2D["font_size_legend"]},
)
for item in (
[ax.xaxis.label, ax.yaxis.label] + ax.get_xticklabels() + ax.get_yticklabels()
):
item.set_fontname(dict_2D["font_name"])
item.set_fontsize(dict_2D["font_size_label"])
ax.title.set_fontname(dict_2D["font_name"])
ax.title.set_fontsize(dict_2D["font_size_title"])
if is_show_fig:
fig.show()
return fig, ax
[docs]def get_color(wind_mat, Nrad, Ntan, Zs):
"""Return the color (corresponding phase) for the zone (Nrad,Ntan,Zs)
Parameters
----------
wind_mat :
A matrix [Nrad,Ntan,Zs,qs] representing the winding
Nrad :
Zone radial coordinate
Ntan :
Zone tagential coordinate
Zs :
Zone slot number coordinate
Returns
-------
str
color: Color of the zone
"""
A = wind_mat[Nrad, Ntan, Zs, :]
for zz in range(len(A)):
if A[zz] != 0:
return PHASE_COLORS[zz]
return "w" # If all the phase are at 0 : the zone is empty => white
