OM10-cosmoDC2 lensed AGN systems

README.rst

@@ -26,6 +26,7 @@ How to contribute to the repository?
 Check out our step-by-step instructions how you can contribute code to this project Contributing_ Guidelines.
 Are you new to coding and contributing to a software development project? No worries! We help you out! Write us on the LSST Slack channel #sl-pipeline.
 
+Note: `sim-pipeline` is written in object-oriented python. What fun!
 
 How can I participate in LSST strong lensing projects?
 ------------------------------------------------------

notebooks/galaxy_galaxy_lensing_tutorial.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -192,7 +192,7 @@
     "\n",
     "# generate montage indicating which bands are used for the rgb color image\n",
     "fig, axes = gg_plot.plot_montage(\n",
-    "    rgb_band_list=[\"i\", \"r\", \"g\"],\n",
+    "    rgb_band_list=[\"i\"],\n",
     "    add_noise=True,\n",
     "    n_horizont=5,\n",
     "    n_vertical=2,\n",
@@ -211,7 +211,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 5,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2023-07-10T22:03:19.932376Z",
@@ -307,13 +307,6 @@
     "corner.corner(np.array(lens_samples), labels=labels, **hist2dkwargs)\n",
     "plt.show()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
@@ -332,7 +325,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.7"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,

slsim/Deflectors/deflectors_base.py

@@ -10,7 +10,7 @@ class DeflectorsBase(ABC):
     def __init__(self, deflector_table, kwargs_cut, cosmo, sky_area):
         """
 
-        :param deflector_table: table with lens parameters
+        :param deflector_table: file or table with lens parameters
         :param kwargs_cut: cuts in parameters: band, band_mag, z_min, z_max
         :type kwargs_cut: dict
         :param cosmo: astropy.cosmology instance

slsim/Deflectors/om10_lens_galaxies.py

@@ -0,0 +1,31 @@
+from slsim.Deflectors.deflector_base import DeflectorBase
+
+
+class OM10Lens(DeflectorBase):
+    def __init__(self, deflector_table, kwargs_cut, cosmo, sky_area):
+        """
+        :param deflector_table: dict-like containing lens properties
+        :param kwargs_cut: cuts to impose on lens properties
+        :param cosmo: cosmology used
+        :param sky_area: area of sky used
+
+        """
+        super().__init__(deflector_table, kwargs_cut, cosmo, sky_area)
+        self.deflector_table = deflector_table
+
+    @property
+    def deflector_number(self):
+        """
+
+        :return: number of deflectors after applied cuts
+        """
+        return len(self.deflector_table)
+
+    def draw_deflector(self, index):
+        """
+        :param index: id number of lens and source
+
+        :return: dictionary of complete parameterization of deflector
+        """
+        self._chosen_deflector = dict(self.deflector_table.loc[index])
+        return self._chosen_deflector

slsim/Pipelines/catalog_pipeline.py

@@ -0,0 +1,38 @@
+import os
+import slsim
+import pandas as pd
+
+
+class catalogPipeline:
+    """Class for skypy configuration."""
+
+    def __init__(self, catalog_config=None):
+        """
+        :param catalog_config: path to configuration file.
+        :type catalog_config: string
+        """
+        path = os.path.dirname(slsim.__file__)
+        module_path, _ = os.path.split(path)
+        folder = os.path.join(module_path, catalog_config)
+        self._sources = pd.read_csv(os.path.join(folder, "sources.csv"), index_col=0)
+        self._deflectors = pd.read_csv(
+            os.path.join(folder, "deflectors.csv"), index_col=0
+        )
+
+    @property
+    def sources(self):
+        """Source properties.
+
+        :return: list of COSMODC2 Quasars
+        :rtype: list of dict
+        """
+        return self._sources
+
+    @property
+    def deflectors(self):
+        """Deflector properties.
+
+        :return: list of OM10 galaxies
+        :rtype: list of dict
+        """
+        return self._deflectors

slsim/Sources/cosmoDC2AGN.py

@@ -0,0 +1,29 @@
+from slsim.Sources.source_pop_base import SourcePopBase
+
+
+class cosmoDC2AGN(SourcePopBase):
+    def __init__(self, source_input, cosmo, sky_area):
+        """
+        :param source_input: dict-like containing source properties
+        :param cosmo: cosmology used
+        :param sky_area:
+        """
+        self.source_table = source_input
+        super().__init__(cosmo, sky_area)
+
+    @property
+    def source_number(self):
+        """Number of sources registered (within given area on the sky)
+
+        :return: number of sources
+        """
+        return len(self.source_table)
+
+    def draw_source(self, index):
+        """Choose source at random.
+
+        :param index: id number of lens and source
+        :return: dictionary of source
+        """
+        self._chosen_source = dict(self.source_table.loc[index])
+        return self._chosen_source

slsim/Util/param_util.py

@@ -3,14 +3,13 @@
 from scipy.signal import convolve2d
 from scipy.signal import fftconvolve
 from lenstronomy.Util.param_util import transform_e1e2_product_average
-from lenstronomy.Util.param_util import ellipticity2phi_q
 from astropy.io import fits
 from astropy import units as u
 import warnings
 
 
 def epsilon2e(epsilon):
-    """Translates ellipticity definitions from.
+    """Translates ellipticity definitions from epsilon to e.
 
     .. math::
         epsilon = \\equic \\frac{1 - q^2}{1 + q^2}
@@ -31,6 +30,21 @@ def epsilon2e(epsilon):
         raise ValueError('Value of "epsilon" is %s and needs to be in [0, 1]' % epsilon)
 
 
+def ellipticity2phi_q(e1, e2):
+    """Transforms complex ellipticity moduli in orientation angle and axis ratio.
+
+    :param e1: eccentricity in x-direction
+    :param e2: eccentricity in xy-direction
+    :return: angle in radian, axis ratio (minor/major)
+    """
+
+    phi = np.arctan2(e2, e1) / 2
+    c = np.sqrt(e1**2 + e2**2)
+    c = np.minimum(c, 0.9999)
+    q = (1 - c) / (1 + c)
+    return phi, q
+
+
 def e2epsilon(e):
     """Translates ellipticity definitions from.
 

slsim/image_simulation.py

@@ -96,6 +96,7 @@
     :return: 2d array unblurred image
     """
     kwargs_model, kwargs_params = lens_class.lenstronomy_kwargs(band)
+
     kwargs_band = {
         "pixel_scale": delta_pix,
         "magnitude_zero_point": mag_zero_point,
@@ -166,6 +167,69 @@
     return image_rgb
 
 
+def quasar_image(
+    lens_class,
+    band,
+    kwargs_band,
+    num_pix,
+    with_source=True,
+    with_ps=True,
+    with_deflector=True,
+    unconvolved=False,
+    add_noise=True,
+    bg_noise=True,
+):
+    """
+    :param lens_class: Lens() object
+    :param band: imaging band
+    :param kwargs_band: key word arguments including: read_noise, pixel_scale, ccd_gain,
+    exposure_time, sky_brightness, magnitude_zero_point, num_exposures, seeing, psf_type,
+    kernel_point_source
+    :param num_pix: number of pixels per axis
+    :param with_source: bool, if True includes extended source light
+    :param with_ps: bool, if True includes point source light
+    :param with_deflector: bool, if True includes deflector light
+    :param unconvolved: bool, if True PSF is not convolved with extended source light or lens light
+    :param add_noise: bool, if True add noise
+    :return: 2d array of image
+    """
+    kwargs_model, kwargs_params = lens_class.lenstronomy_kwargs(band)
+    kwargs_band = kwargs_band
+    sim_api = SimAPI(
+        numpix=num_pix, kwargs_single_band=kwargs_band, kwargs_model=kwargs_model
+    )
+    kwargs_lens_light, kwargs_source, kwargs_ps = sim_api.magnitude2amplitude(
+        kwargs_lens_light_mag=kwargs_params.get("kwargs_lens_light", None),
+        kwargs_source_mag=kwargs_params.get("kwargs_source", None),
+        kwargs_ps_mag=kwargs_params.get("kwargs_ps", None),
+    )
+    kwargs_numerics = {"supersampling_factor": 1}
+    image_model = sim_api.image_model_class(kwargs_numerics)
+    kwargs_lens = kwargs_params.get("kwargs_lens", None)
+    image = image_model.image(
+        kwargs_lens=kwargs_lens,
+        kwargs_source=kwargs_source,
+        kwargs_lens_light=kwargs_lens_light,
+        kwargs_ps=kwargs_ps,
+        unconvolved=unconvolved,
+        source_add=with_source,
+        lens_light_add=with_deflector,
+        point_source_add=with_ps,
+    )
+    if add_noise:
+        image += sim_api.noise_for_model(model=image, background_noise=bg_noise)
+
+    return (
+        image,
+        kwargs_model,
+        kwargs_params,
+        kwargs_lens,
+        kwargs_lens_light,
+        kwargs_source,
+        kwargs_ps,
+    )
+
+
 def rgb_image_from_image_list(image_list, stretch):
     """Creates a rgb image using list of images in r, g, and b.