def create_class_instances(lens_model_list=[],
z_lens=None,
z_source=None,
lens_redshift_list=None,
multi_plane=False,
observed_convention_index=None,
source_light_model_list=[],
lens_light_model_list=[],
point_source_model_list=[],
fixed_magnification_list=None,
additional_images_list=None,
min_distance=0.01,
search_window=5,
precision_limit=10**(-10),
num_iter_max=100,
source_deflection_scaling_list=None,
source_redshift_list=None,
cosmo=None,
index_lens_model_list=None,
index_source_light_model_list=None,
index_lens_light_model_list=None,
index_point_source_model_list=None,
optical_depth_model_list=[],
index_optical_depth_model_list=None,
band_index=0,
tau0_index_list=None,
all_models=False,
point_source_magnification_limit=None):
"""
:param lens_model_list: list of strings indicating the type of lens models
:param z_lens: redshift of the deflector (for single lens plane mode, but only relevant when computing physical quantities)
:param z_source: redshift of source (for single source plane mode, or for multiple source planes the redshift of the point source). In regard to this redshift the reduced deflection angles are defined in the lens model.
:param lens_redshift_list:
:param multi_plane:
:param observed_convention_index:
:param source_light_model_list:
:param lens_light_model_list:
:param point_source_model_list:
:param fixed_magnification_list:
:param additional_images_list:
:param min_distance:
:param search_window:
:param precision_limit:
:param num_iter_max:
:param source_deflection_scaling_list:
:param source_redshift_list:
:param cosmo: astropy.cosmology instance
:param index_lens_model_list:
:param index_source_light_model_list:
:param index_lens_light_model_list:
:param index_point_source_model_list:
:param optical_depth_model_list: list of strings indicating the optical depth model to compute (differential) extinctions from the source
:param band_index: int, index of band to consider. Has an effect if only partial models are considered for a specific band
:param tau0_index_list: list of integers of the specific extinction scaling parameter tau0 for each band
:param all_models: bool, if True, will make class instances of all models ignoring potential keywords that are excluding specific models as indicated.
:param point_source_magnification_limit: float >0 or None, if set and additional images are computed, then it will cut the point sources computed to the limiting (absolute) magnification
:return:
"""
if index_lens_model_list is None or all_models is True:
lens_model_list_i = lens_model_list
lens_redshift_list_i = lens_redshift_list
observed_convention_index_i = observed_convention_index
else:
lens_model_list_i = [
lens_model_list[k] for k in index_lens_model_list[band_index]
]
if lens_redshift_list is not None:
lens_redshift_list_i = [
lens_redshift_list[k]
for k in index_lens_model_list[band_index]
]
else:
lens_redshift_list_i = lens_redshift_list
if observed_convention_index is not None:
counter = 0
observed_convention_index_i = []
for k in index_lens_model_list[band_index]:
if k in observed_convention_index:
observed_convention_index_i.append(counter)
counter += 1
else:
observed_convention_index_i = observed_convention_index
lens_model_class = LensModel(
lens_model_list=lens_model_list_i,
z_lens=z_lens,
z_source=z_source,
lens_redshift_list=lens_redshift_list_i,
multi_plane=multi_plane,
cosmo=cosmo,
observed_convention_index=observed_convention_index_i)
if index_source_light_model_list is None or all_models is True:
source_light_model_list_i = source_light_model_list
source_deflection_scaling_list_i = source_deflection_scaling_list
source_redshift_list_i = source_redshift_list
else:
source_light_model_list_i = [
source_light_model_list[k]
for k in index_source_light_model_list[band_index]
]
if source_deflection_scaling_list is None:
source_deflection_scaling_list_i = source_deflection_scaling_list
else:
source_deflection_scaling_list_i = [
source_deflection_scaling_list[k]
for k in index_source_light_model_list[band_index]
]
if source_redshift_list is None:
source_redshift_list_i = source_redshift_list
else:
source_redshift_list_i = [
source_redshift_list[k]
for k in index_source_light_model_list[band_index]
]
source_model_class = LightModel(
light_model_list=source_light_model_list_i,
deflection_scaling_list=source_deflection_scaling_list_i,
source_redshift_list=source_redshift_list_i)
if index_lens_light_model_list is None or all_models is True:
lens_light_model_list_i = lens_light_model_list
else:
lens_light_model_list_i = [
lens_light_model_list[k]
for k in index_lens_light_model_list[band_index]
]
lens_light_model_class = LightModel(
light_model_list=lens_light_model_list_i)
point_source_model_list_i = point_source_model_list
fixed_magnification_list_i = fixed_magnification_list
additional_images_list_i = additional_images_list
if index_point_source_model_list is not None and not all_models:
point_source_model_list_i = [
point_source_model_list[k]
for k in index_point_source_model_list[band_index]
]
if fixed_magnification_list is not None:
fixed_magnification_list_i = [
fixed_magnification_list[k]
for k in index_point_source_model_list[band_index]
]
if additional_images_list is not None:
additional_images_list_i = [
additional_images_list[k]
for k in index_point_source_model_list[band_index]
]
point_source_class = PointSource(
point_source_type_list=point_source_model_list_i,
lensModel=lens_model_class,
fixed_magnification_list=fixed_magnification_list_i,
additional_images_list=additional_images_list_i,
min_distance=min_distance,
search_window=search_window,
precision_limit=precision_limit,
num_iter_max=num_iter_max,
magnification_limit=point_source_magnification_limit)
if tau0_index_list is None:
tau0_index = 0
else:
tau0_index = tau0_index_list[band_index]
if index_optical_depth_model_list is not None:
optical_depth_model_list_i = [
optical_depth_model_list[k]
for k in index_optical_depth_model_list[band_index]
]
else:
optical_depth_model_list_i = optical_depth_model_list
extinction_class = DifferentialExtinction(
optical_depth_model=optical_depth_model_list_i, tau0_index=tau0_index)
return lens_model_class, source_model_class, lens_light_model_class, point_source_class, extinction_class