def model_plot(self,
ax,
v_min=None,
v_max=None,
image_names=False,
colorbar_label=r'log$_{10}$ flux',
font_size=15,
text='Reconstructed',
**kwargs):
"""
:param ax:
:param model:
:param v_min:
:param v_max:
:return:
"""
if v_min is None:
v_min = self._v_min_default
if v_max is None:
v_max = self._v_max_default
im = ax.matshow(np.log10(self._model),
origin='lower',
vmin=v_min,
vmax=v_max,
extent=[0, self._frame_size, 0, self._frame_size],
cmap=self._cmap)
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.autoscale(False)
plot_util.scale_bar(ax,
self._frame_size,
dist=1,
text='1"',
font_size=font_size)
plot_util.text_description(ax,
self._frame_size,
text=text,
color="w",
backgroundcolor='k',
font_size=font_size)
if 'no_arrow' not in kwargs or not kwargs['no_arrow']:
plot_util.coordinate_arrows(ax,
self._frame_size,
self._coords,
color='w',
arrow_size=self._arrow_size,
font_size=font_size)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)
cb = plt.colorbar(im, cax=cax)
cb.set_label(colorbar_label, fontsize=font_size)
#plot_line_set(ax, self._coords, self._ra_caustic_list, self._dec_caustic_list, color='b')
#plot_line_set(ax, self._coords, self._ra_crit_list, self._dec_crit_list, color='r')
if image_names is True:
ra_image, dec_image = self.bandmodel.PointSource.image_position(
self._kwargs_ps_partial, self._kwargs_lens_partial)
plot_util.image_position_plot(ax, self._coords, ra_image,
dec_image)
def test_image_position_plot(self):
coords = Coordinates(transform_pix2angle=[[1, 0], [0, 1]],
ra_at_xy_0=0,
dec_at_xy_0=0)
f, ax = plt.subplots(1, 1, figsize=(4, 4))
ra_image, dec_image = np.array([1, 2]), np.array([1, 2])
ax = plot_util.image_position_plot(ax,
coords,
ra_image,
dec_image,
color='w',
image_name_list=None,
origin=None,
flipped_x=False)
plt.close()
ax = plot_util.image_position_plot(ax,
coords,
ra_image,
dec_image,
color='w',
image_name_list=['A', 'B'],
origin=[1, 1],
flipped_x=True)
plt.close()
def deflection_plot(self, ax, v_min=None, v_max=None, axis=0,
with_caustics=False, image_name_list=None,
text="Deflection model", font_size=15,
colorbar_label=r'arcsec'):
"""
:param kwargs_lens:
:param kwargs_else:
:return:
"""
alpha1, alpha2 = self._lensModel.alpha(self._x_grid, self._y_grid, self._kwargs_lens_partial)
alpha1 = util.array2image(alpha1)
alpha2 = util.array2image(alpha2)
if axis == 0:
alpha = alpha1
else:
alpha = alpha2
im = ax.matshow(alpha, origin='lower', extent=[0, self._frame_size, 0, self._frame_size],
vmin=v_min, vmax=v_max, cmap=self._cmap, alpha=0.5)
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.autoscale(False)
plot_util.scale_bar(ax, self._frame_size, dist=1, text='1"', color='k', font_size=font_size)
plot_util.coordinate_arrows(ax, self._frame_size, self._coords, color='k',
arrow_size=self._arrow_size, font_size=font_size)
plot_util.text_description(ax, self._frame_size, text=text, color="k",
backgroundcolor='w', font_size=font_size)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)
cb = plt.colorbar(im, cax=cax)
cb.set_label(colorbar_label, fontsize=font_size)
if with_caustics is True:
ra_crit_list, dec_crit_list = self._critical_curves()
ra_caustic_list, dec_caustic_list = self._caustics()
plot_util.plot_line_set(ax, self._coords, ra_caustic_list, dec_caustic_list, color='b')
plot_util.plot_line_set(ax, self._coords, ra_crit_list, dec_crit_list, color='r')
ra_image, dec_image = self.bandmodel.PointSource.image_position(self._kwargs_ps_partial, self._kwargs_lens_partial)
plot_util.image_position_plot(ax, self._coords, ra_image, dec_image, image_name_list=image_name_list)
return ax
def magnification_plot(self, ax, v_min=-10, v_max=10,
image_name_list=None, font_size=15, no_arrow=False,
text="Magnification model",
colorbar_label=r"$\det\ (\mathsf{A}^{-1})$",
**kwargs):
"""
:param ax: matplotib axis instance
:param v_min: minimum range of plotting
:param v_max: maximum range of plotting
:param kwargs: kwargs to send to matplotlib.pyplot.matshow()
:return:
"""
if not 'cmap' in kwargs:
kwargs['cmap'] = self._cmap
if not 'alpha' in kwargs:
kwargs['alpha'] = 0.5
mag_result = util.array2image(self._lensModel.magnification(self._x_grid, self._y_grid, self._kwargs_lens_partial))
im = ax.matshow(mag_result, origin='lower', extent=[0, self._frame_size, 0, self._frame_size],
vmin=v_min, vmax=v_max, **kwargs)
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.autoscale(False)
plot_util.scale_bar(ax, self._frame_size, dist=1, text='1"', color='k', font_size=font_size)
if not no_arrow:
plot_util.coordinate_arrows(ax, self._frame_size, self._coords, color='k', arrow_size=self._arrow_size,
font_size=font_size)
plot_util.text_description(ax, self._frame_size, text=text, color="k",
backgroundcolor='w', font_size=font_size)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)
cb = plt.colorbar(im, cax=cax)
cb.set_label(colorbar_label, fontsize=font_size)
ra_image, dec_image = self.bandmodel.PointSource.image_position(self._kwargs_ps_partial, self._kwargs_lens_partial)
plot_util.image_position_plot(ax, self._coords, ra_image, dec_image, color='k', image_name_list=image_name_list)
return ax
