def make_wrp_phase_x_slice_ax(ax, **kwargs):
wrp_phase = kwargs.get('wrp_phase')
unwrp_phase = kwargs.get('unwrp_phase')
geometry_center = kwargs.get('geometry_center')
wrapped_ymin, wrapped_ymax = kwargs.get('wrapped_ylims', [-np.pi, np.pi])
linewidth = kwargs.get('linewidth', 1.5)
grid_centering = kwargs.get('grid_centering', False) # mm
y_max, x_max = __find_max_coordinates(geometry_center, wrp_phase, unwrp_phase)
wrp_phase_xslice_x, wrp_phase_xslice_y = get_slice(wrp_phase, x_max, xslice=True)
# Центрирование координатной сетки
if grid_centering:
wrp_phase_xslice_x -= wrp_phase_xslice_x[wrp_phase_xslice_x.size // 2] # plot()
ax.title.set_text('x slice')
ax.plot(wrp_phase_xslice_x, wrp_phase_xslice_y, linewidth=linewidth, label=f'x: {y_max}')
ax.set_ylim([wrapped_ymin, wrapped_ymax])
ax.legend()
def make_intensity_y_slice_ax(ax, **kwargs):
intensity = kwargs.get('intensity')
geometry_center = kwargs.get('geometry_center')
linewidth = kwargs.get('linewidth', 1.5)
ymin, ymax = kwargs.get('ylims', [None, None])
xmin, xmax = kwargs.get('xlims', [None, None])
if geometry_center:
max_indexes = [intensity.shape[0] // 2, intensity.shape[1] // 2]
else:
max_indexes = np.unravel_index(np.argmax(intensity, axis=None), intensity.shape)
intensity_yslice_x, intensity_yslice_y = get_slice(intensity, max_indexes[1], xslice=False)
intensity_yslice_x -= intensity_yslice_x[intensity_yslice_x.size // 2]
ax.plot(intensity_yslice_x, intensity_yslice_y, linewidth=linewidth, label=f'y: {max_indexes[0]}')
ax.title.set_text('y slice')
ax.set_ylim([ymin, ymax])
ax.set_xlim([xmin, xmax])
ax.legend()
def make_unwrp_phase_y_slice_ax(ax, **kwargs):
wrp_phase = kwargs.get('wrp_phase')
unwrp_phase = kwargs.get('unwrp_phase')
geometry_center = kwargs.get('geometry_center')
unwrapped_ymin, unwrapped_ymax = kwargs.get('unwrapped_ylims', [None, None])
linewidth = kwargs.get('linewidth', 1.5)
grid_centering = kwargs.get('grid_centering', False) # mm
xmin, xmax = kwargs.get('xlims', [None, None])
y_max, x_max = __find_max_coordinates(geometry_center, wrp_phase, unwrp_phase)
unwrp_phase_yslice_x, unwrp_phase_yslice_y = get_slice(unwrp_phase, y_max, xslice=False)
# Центрирование координатной сетки
if grid_centering:
unwrp_phase_yslice_x -= unwrp_phase_yslice_x[unwrp_phase_yslice_x.size // 2] # plot()
ax.title.set_text('y slice')
ax.plot(unwrp_phase_yslice_x, unwrp_phase_yslice_y, linewidth=linewidth, label=f'y: {x_max}')
ax.set_ylim([unwrapped_ymin, unwrapped_ymax])
ax.set_xlim([xmin, xmax])
ax.legend()