# -*- coding: utf-8 -*-
from numpy import zeros
[docs]def reverse_wind_mat(wind_mat):
"""reverse a Winding Matrix along the slot
Parameters
----------
wind_mat : numpy.ndarray
A Winding Matrix (Nlay_r, Nlay_theta, Zs, qs)
(created by comp_connection_mat)
Returns
-------
wind_mat: numpy.ndarray
The reverse matrix along the Slot
"""
assert len(wind_mat.shape) == 4, "wind_mat has a wrong shape (dim 4)"
wind_mat = wind_mat[:, :, ::-1, :]
if wind_mat.shape[1] == 2:
wind_mat = wind_mat[:, ::-1, :, :]
return wind_mat
[docs]def reverse_layer(wind_mat):
"""reverse the Winding Matrix along the layer
Parameters
----------
wind_mat : numpy.ndarray
A Winding Matrix (Nlay_r, Nlay_theta, Zs, qs)
(created by comp_connection_mat)
Returns
-------
wind_mat: numpy.ndarray
The reverse matrix along the layer
"""
assert len(wind_mat.shape) == 4, "wind_mat has a wrong shape (dim 4)"
wind_mat = wind_mat[:, ::-1, :, :]
wind_mat = wind_mat[::-1, :, :, :]
return wind_mat
[docs]def change_layer(wind_mat):
"""Change the Winding Matrix along the layer
Parameters
----------
wind_mat : numpy.ndarray
A Winding Matrix (Nlay_r, Nlay_theta, Zs, qs)
(created by comp_connection_mat)
Returns
-------
wind_mat: numpy.ndarray
The winding matrix with changed layers
"""
wind_shape = wind_mat.shape
assert len(wind_shape) == 4, "wind_mat has a wrong shape (dim 4)"
if wind_shape[0] == 1 and wind_shape[1] != 1:
wind_mat2 = zeros((wind_shape[1], wind_shape[0], wind_shape[2], wind_shape[3]))
wind_mat2[:, 0, :, :] = wind_mat[0, :, :, :]
return wind_mat2
elif wind_shape[0] != 1 and wind_shape[1] == 1:
wind_mat2 = zeros((wind_shape[1], wind_shape[0], wind_shape[2], wind_shape[3]))
wind_mat2[0, :, :, :] = wind_mat[:, 0, :, :]
return wind_mat2
else:
return wind_mat
[docs]def permute_B_C(wind_mat):
"""Permute B & C phases
Parameters
----------
wind_mat : numpy.ndarray
A Winding Matrix (Nlay_r, Nlay_theta, Zs, qs)
(created by comp_connection_mat)
Returns
-------
wind_mat: numpy.ndarray
The winding matrix with permuted phases
"""
wind_shape = wind_mat.shape
assert len(wind_shape) == 4, "wind_mat has a wrong shape (dim 4)"
assert wind_shape[3] > 2
wind_mat2 = wind_mat.copy()
wind_mat2[:, :, :, 1] = wind_mat[:, :, :, 2]
wind_mat2[:, :, :, 2] = wind_mat[:, :, :, 1]
return wind_mat2
