SNRLightProfiles work

Author: Jammy2211

autolens/__init__.py

@@ -51,6 +51,7 @@
     light_and_mass_profiles as lmp,
     scaling_relations as sr,
 )
+from autogalaxy.profiles.light_profiles import snr_light_profiles as lp_snr
 from autogalaxy import convert
 
 from . import plot

autolens/imaging/imaging.py

@@ -76,6 +76,12 @@ def via_tracer_from(self, tracer, grid, name=None):
             A seed for random noise_maps generation
         """
 
+        tracer.set_snr_of_snr_light_profiles(
+            grid=grid,
+            exposure_time=self.exposure_time,
+            background_sky_level=self.background_sky_level,
+        )
+
         image = tracer.padded_image_2d_from(
             grid=grid, psf_shape_2d=self.psf.shape_native
         )

autolens/lens/ray_tracing.py

@@ -10,7 +10,9 @@
 
 from autoarray.inversion.inversion.factory import inversion_imaging_unpacked_from
 from autoarray.inversion.inversion.factory import inversion_interferometer_unpacked_from
+
 from autogalaxy.lensing import LensingObject
+from autogalaxy.profiles.light_profiles.snr_light_profiles import LightProfileSNR
 
 from autolens.lens.model.preloads import Preloads
 
@@ -982,6 +984,25 @@ def galaxy_profile_visibilities_dict_via_transformer_from(
 
         return galaxy_profile_visibilities_image_dict
 
+    def set_snr_of_snr_light_profiles(
+        self,
+        grid: aa.type.Grid2DLike,
+        exposure_time: float,
+        background_sky_level: float = 0.0,
+    ):
+
+        traced_grids_of_planes = self.traced_grids_of_planes_from(grid=grid)
+
+        for plane_index, plane in enumerate(self.planes):
+            for galaxy in plane.galaxies:
+                for light_profile in galaxy.light_profiles:
+                    if isinstance(light_profile, LightProfileSNR):
+                        light_profile.set_intensity_from(
+                            grid=traced_grids_of_planes[plane_index],
+                            exposure_time=exposure_time,
+                            background_sky_level=background_sky_level,
+                        )
+
 
 class Tracer(AbstractTracerData):
     @property

test_autolens/lens/test_ray_tracing.py

@@ -3141,6 +3141,41 @@ def test__extract_plane_index_of_profile(self):
         assert plane_index == 2
 
 
+class TestSNRLightProfiles:
+    def test__signal_to_noise_via_simulator_correct(self):
+
+        background_sky_level = 10.0
+        exposure_time = 300.0
+
+        grid = al.Grid2D.uniform(shape_native=(3, 3), pixel_scales=1.0)
+
+        mass = al.mp.SphIsothermal(einstein_radius=1.0)
+
+        sersic = al.lp_snr.EllSersic(signal_to_noise_ratio=10.0, effective_radius=0.01)
+
+        tracer = al.Tracer.from_galaxies(
+            galaxies=[
+                al.Galaxy(redshift=0.5, mass=mass),
+                al.Galaxy(redshift=1.0, light=sersic),
+            ]
+        )
+
+        simulator = al.SimulatorImaging(
+            exposure_time=exposure_time,
+            noise_seed=1,
+            background_sky_level=background_sky_level,
+        )
+
+        imaging = simulator.via_tracer_from(tracer=tracer, grid=grid)
+
+        print(imaging.signal_to_noise_map)
+
+        assert 8.0 < imaging.signal_to_noise_map.native[0, 1] < 12.0
+        assert 8.0 < imaging.signal_to_noise_map.native[1, 0] < 12.0
+        assert 8.0 < imaging.signal_to_noise_map.native[1, 2] < 12.0
+        assert 8.0 < imaging.signal_to_noise_map.native[2, 1] < 12.0
+
+
 class TestRegression:
     def test__centre_of_profile_in_right_place(self):