Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
if tracer.has_mass_profile:
sub_plotter.setup_subplot(number_subplots=number_subplots, subplot_index=2)
convergence(tracer=tracer, grid=grid, include=include, plotter=sub_plotter)
sub_plotter.setup_subplot(number_subplots=number_subplots, subplot_index=3)
potential(tracer=tracer, grid=grid, include=include, plotter=sub_plotter)
sub_plotter.setup_subplot(number_subplots=number_subplots, subplot_index=4)
source_plane_grid = tracer.traced_grids_of_planes_from_grid(grid=grid)[-1]
ag.plot.Plane.plane_image(
plane=tracer.source_plane,
grid=source_plane_grid,
caustics=include.caustics_from_obj(obj=tracer),
include=include,
plotter=sub_plotter,
)
if tracer.has_mass_profile:
sub_plotter.setup_subplot(number_subplots=number_subplots, subplot_index=5)
deflections_y(tracer=tracer, grid=grid, include=include, plotter=sub_plotter)
sub_plotter.setup_subplot(number_subplots=number_subplots, subplot_index=6)
deflections_x(tracer=tracer, grid=grid, include=include, plotter=sub_plotter)
plotter=plotter,
)
if plot_convergence:
convergence(tracer=tracer, grid=grid, include=include, plotter=plotter)
if plot_potential:
potential(tracer=tracer, grid=grid, include=include, plotter=plotter)
if plot_source_plane:
source_plane_grid = tracer.traced_grids_of_planes_from_grid(grid=grid)[-1]
ag.plot.Plane.plane_image(
plane=tracer.source_plane,
grid=source_plane_grid,
caustics=include.caustics_from_obj(obj=tracer),
positions=None,
include=include,
plotter=plotter.plotter_with_new_output(filename="source_plane"),
)
if plot_deflections:
deflections_y(tracer=tracer, grid=grid, include=include, plotter=plotter)
deflections_x(tracer=tracer, grid=grid, include=include, plotter=plotter)
if plot_magnification: