def visualize_ray_tracing(self, paths: af.Paths, tracer, during_analysis):
plotter = self.plotter_from_paths(paths=paths, subfolders="ray_tracing")
sub_plotter = self.sub_plotter_from_paths(paths=paths, subfolders="ray_tracing")
if self.plot_subplot_ray_tracing:
ray_tracing_plots.subplot_tracer(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset
),
include=self.include,
sub_plotter=sub_plotter,
)
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset
),
plot_image=self.plot_ray_tracing_image,
plot_source_plane=self.plot_ray_tracing_source_plane,
plot_convergence=self.plot_ray_tracing_convergence,
plot_potential=self.plot_ray_tracing_potential,
plot_deflections=self.plot_ray_tracing_deflections,
plot_magnification=self.plot_ray_tracing_magnification,
include=self.include,
plotter=plotter,
)
if not during_analysis:
if self.plot_ray_tracing_all_at_end_png:
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset
),
plot_image=True,
plot_source_plane=True,
plot_convergence=True,
plot_potential=True,
plot_deflections=True,
include=self.include,
plotter=plotter,
)
if self.plot_ray_tracing_all_at_end_fits:
self.visualize_ray_tracing_in_fits(paths=paths, tracer=tracer)
def visualize_ray_tracing(self, tracer, during_analysis):
plotter = self.plotter.plotter_with_new_output(
path=self.plotter.output.path + "ray_tracing/")
if self.plot_subplot_ray_tracing:
ray_tracing_plots.subplot_tracer(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset),
include=self.include,
sub_plotter=self.sub_plotter,
)
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset),
plot_profile_image=self.plot_ray_tracing_profile_image,
plot_source_plane=self.plot_ray_tracing_source_plane,
plot_convergence=self.plot_ray_tracing_convergence,
plot_potential=self.plot_ray_tracing_potential,
plot_deflections=self.plot_ray_tracing_deflections,
plot_magnification=self.plot_ray_tracing_magnification,
include=self.include,
plotter=plotter,
)
if not during_analysis:
if self.plot_ray_tracing_all_at_end_png:
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
positions=self.include.positions_from_masked_dataset(
masked_dataset=self.masked_dataset),
plot_profile_image=True,
plot_source_plane=True,
plot_convergence=True,
plot_potential=True,
plot_deflections=True,
include=self.include,
plotter=plotter,
)
if self.plot_ray_tracing_all_at_end_fits:
self.visualize_ray_tracing_in_fits(tracer=tracer)
def visualize_ray_tracing_in_fits(self, tracer):
fits_plotter = self.plotter.plotter_with_new_output(
path=self.plotter.output.path + "/ray_tracing/fits/",
format="fits")
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
plot_profile_image=True,
plot_source_plane=True,
plot_convergence=True,
plot_potential=True,
plot_deflections=True,
include=self.include,
plotter=fits_plotter,
)
def visualize_ray_tracing_in_fits(self, paths: af.Paths, tracer):
fits_plotter = self.plotter_from_paths(
paths=paths, subfolders=path.join("ray_tracing", "fits"), format="fits"
)
ray_tracing_plots.individual(
tracer=tracer,
grid=self.masked_dataset.grid,
plot_image=True,
plot_source_plane=True,
plot_convergence=True,
plot_potential=True,
plot_deflections=True,
include=self.include,
plotter=fits_plotter,
)