def __init__(self, light_model_list, smoothing=0.001):
"""
:param light_model_list: list of light models
:param smoothing: smoothing factor for certain models (deprecated)
"""
self.profile_type_list = light_model_list
self.func_list = []
for profile_type in light_model_list:
if profile_type == 'GAUSSIAN':
from lenstronomy.LightModel.Profiles.gaussian import Gaussian
self.func_list.append(Gaussian())
elif profile_type == 'GAUSSIAN_ELLIPSE':
from lenstronomy.LightModel.Profiles.gaussian import GaussianEllipse
self.func_list.append(GaussianEllipse())
elif profile_type == 'ELLIPSOID':
from lenstronomy.LightModel.Profiles.ellipsoid import Ellipsoid
self.func_list.append(Ellipsoid())
elif profile_type == 'MULTI_GAUSSIAN':
from lenstronomy.LightModel.Profiles.gaussian import MultiGaussian
self.func_list.append(MultiGaussian())
elif profile_type == 'MULTI_GAUSSIAN_ELLIPSE':
from lenstronomy.LightModel.Profiles.gaussian import MultiGaussianEllipse
self.func_list.append(MultiGaussianEllipse())
elif profile_type == 'SERSIC':
from lenstronomy.LightModel.Profiles.sersic import Sersic
self.func_list.append(Sersic(smoothing=smoothing))
elif profile_type == 'SERSIC_ELLIPSE':
from lenstronomy.LightModel.Profiles.sersic import SersicElliptic
self.func_list.append(
SersicElliptic(smoothing=smoothing,
sersic_major_axis=sersic_major_axis_conf))
elif profile_type == 'CORE_SERSIC':
from lenstronomy.LightModel.Profiles.sersic import CoreSersic
self.func_list.append(CoreSersic(smoothing=smoothing))
elif profile_type == 'SHAPELETS':
from lenstronomy.LightModel.Profiles.shapelets import ShapeletSet
self.func_list.append(ShapeletSet())
elif profile_type == 'SHAPELETS_POLAR':
from lenstronomy.LightModel.Profiles.shapelets_polar import ShapeletSetPolar
self.func_list.append(ShapeletSetPolar(exponential=False))
elif profile_type == 'SHAPELETS_POLAR_EXP':
from lenstronomy.LightModel.Profiles.shapelets_polar import ShapeletSetPolar
self.func_list.append(ShapeletSetPolar(exponential=True))
elif profile_type == 'HERNQUIST':
from lenstronomy.LightModel.Profiles.hernquist import Hernquist
self.func_list.append(Hernquist())
elif profile_type == 'HERNQUIST_ELLIPSE':
from lenstronomy.LightModel.Profiles.hernquist import HernquistEllipse
self.func_list.append(HernquistEllipse())
elif profile_type == 'PJAFFE':
from lenstronomy.LightModel.Profiles.p_jaffe import PJaffe
self.func_list.append(PJaffe())
elif profile_type == 'PJAFFE_ELLIPSE':
from lenstronomy.LightModel.Profiles.p_jaffe import PJaffe_Ellipse
self.func_list.append(PJaffe_Ellipse())
elif profile_type == 'UNIFORM':
from lenstronomy.LightModel.Profiles.uniform import Uniform
self.func_list.append(Uniform())
elif profile_type == 'POWER_LAW':
from lenstronomy.LightModel.Profiles.power_law import PowerLaw
self.func_list.append(PowerLaw())
elif profile_type == 'NIE':
from lenstronomy.LightModel.Profiles.nie import NIE
self.func_list.append(NIE())
elif profile_type == 'CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import Chameleon
self.func_list.append(Chameleon())
elif profile_type == 'DOUBLE_CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import DoubleChameleon
self.func_list.append(DoubleChameleon())
elif profile_type == 'TRIPLE_CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import TripleChameleon
self.func_list.append(TripleChameleon())
elif profile_type == 'INTERPOL':
from lenstronomy.LightModel.Profiles.interpolation import Interpol
self.func_list.append(Interpol())
elif profile_type == 'SLIT_STARLETS':
from lenstronomy.LightModel.Profiles.starlets import SLIT_Starlets
self.func_list.append(
SLIT_Starlets(fast_inverse=True, second_gen=False))
elif profile_type == 'SLIT_STARLETS_GEN2':
from lenstronomy.LightModel.Profiles.starlets import SLIT_Starlets
self.func_list.append(SLIT_Starlets(second_gen=True))
else:
raise ValueError(
'No light model of type %s found! Supported are the following models: %s'
% (profile_type, _MODELS_SUPPORTED))
self._num_func = len(self.func_list)
def __init__(self, light_model_list, smoothing=0.0000001):
"""
:param light_model_list: list of light models
:param deflection_scaling_list: list of floats, rescales the original reduced deflection angles from the lens model
to enable different models to be placed at different optical (redshift) distances. None means they are all
:param source_redshift_list: list of redshifts of the model components
:param smoothing: smoothing factor for certain models (deprecated)
"""
self.profile_type_list = light_model_list
self.func_list = []
for profile_type in light_model_list:
if profile_type == 'GAUSSIAN':
from lenstronomy.LightModel.Profiles.gaussian import Gaussian
self.func_list.append(Gaussian())
elif profile_type == 'GAUSSIAN_ELLIPSE':
from lenstronomy.LightModel.Profiles.gaussian import GaussianEllipse
self.func_list.append(GaussianEllipse())
elif profile_type == 'MULTI_GAUSSIAN':
from lenstronomy.LightModel.Profiles.gaussian import MultiGaussian
self.func_list.append(MultiGaussian())
elif profile_type == 'MULTI_GAUSSIAN_ELLIPSE':
from lenstronomy.LightModel.Profiles.gaussian import MultiGaussianEllipse
self.func_list.append(MultiGaussianEllipse())
elif profile_type == 'SERSIC':
from lenstronomy.LightModel.Profiles.sersic import Sersic
self.func_list.append(Sersic(smoothing=smoothing))
elif profile_type == 'SERSIC_ELLIPSE':
from lenstronomy.LightModel.Profiles.sersic import SersicElliptic
self.func_list.append(SersicElliptic(smoothing=smoothing))
elif profile_type == 'CORE_SERSIC':
from lenstronomy.LightModel.Profiles.sersic import CoreSersic
self.func_list.append(CoreSersic(smoothing=smoothing))
elif profile_type == 'SHAPELETS':
from lenstronomy.LightModel.Profiles.shapelets import ShapeletSet
self.func_list.append(ShapeletSet())
elif profile_type == 'SHAPELETS_POLAR':
from lenstronomy.LightModel.Profiles.shapelets_polar import ShapeletSetPolar
self.func_list.append(ShapeletSetPolar(exponential=False))
elif profile_type == 'SHAPELETS_POLAR_EXP':
from lenstronomy.LightModel.Profiles.shapelets_polar import ShapeletSetPolar
self.func_list.append(ShapeletSetPolar(exponential=True))
elif profile_type == 'HERNQUIST':
from lenstronomy.LightModel.Profiles.hernquist import Hernquist
self.func_list.append(Hernquist())
elif profile_type == 'HERNQUIST_ELLIPSE':
from lenstronomy.LightModel.Profiles.hernquist import HernquistEllipse
self.func_list.append(HernquistEllipse())
elif profile_type == 'PJAFFE':
from lenstronomy.LightModel.Profiles.p_jaffe import PJaffe
self.func_list.append(PJaffe())
elif profile_type == 'PJAFFE_ELLIPSE':
from lenstronomy.LightModel.Profiles.p_jaffe import PJaffe_Ellipse
self.func_list.append(PJaffe_Ellipse())
elif profile_type == 'UNIFORM':
from lenstronomy.LightModel.Profiles.uniform import Uniform
self.func_list.append(Uniform())
elif profile_type == 'POWER_LAW':
from lenstronomy.LightModel.Profiles.power_law import PowerLaw
self.func_list.append(PowerLaw())
elif profile_type == 'NIE':
from lenstronomy.LightModel.Profiles.nie import NIE
self.func_list.append(NIE())
elif profile_type == 'CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import Chameleon
self.func_list.append(Chameleon())
elif profile_type == 'DOUBLE_CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import DoubleChameleon
self.func_list.append(DoubleChameleon())
elif profile_type == 'TRIPLE_CHAMELEON':
from lenstronomy.LightModel.Profiles.chameleon import TripleChameleon
self.func_list.append(TripleChameleon())
elif profile_type == 'INTERPOL':
from lenstronomy.LightModel.Profiles.interpolation import Interpol
self.func_list.append(Interpol())
else:
raise ValueError('Warning! No light model of type',
profile_type, ' found!')
def setup(self):
self.shapeletSet = ShapeletSetPolar(exponential=True)
self.shapelets = ShapeletsPolarExp()
self.x, self.y = util.make_grid(10, 0.1, 1)
class TestShapeletSetPolarExp(object):
"""
class to test Shapelets
"""
def setup(self):
self.shapeletSet = ShapeletSetPolar(exponential=True)
self.shapelets = ShapeletsPolarExp()
self.x, self.y = util.make_grid(10, 0.1, 1)
def test_shapelet_set(self):
"""
#:return:
"""
n_max = 2
beta = 1.
amp = [1, 0, 0, 0, 0, 0, 0, 0, 0]
output = self.shapeletSet.function(np.array(1),
np.array(1),
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output, 0.19397908887786985, decimal=8)
input = np.array(0.)
input += output
output = self.shapeletSet.function(self.x,
self.y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output[10], 0.4511844400064266, decimal=8)
output = self.shapeletSet.function(1,
1,
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output, 0.19397908887786985, decimal=8)
n_max = -1
beta = 1.
amp = [1, 0, 0]
output = self.shapeletSet.function(np.array(1),
np.array(1),
amp,
n_max,
beta,
center_x=0,
center_y=0)
assert output == 0
def test_decomposition(self):
"""
#:return:
"""
scale = 10
n_max = 2
beta = 1. * scale
deltaPix = 0.5 * scale
amp = np.array([1, 1, -1, 1, 1, 1, 1, 1, 1])
x, y = util.make_grid(1000, deltaPix, 1)
input = self.shapeletSet.function(x,
y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
amp_out = self.shapeletSet.decomposition(input,
x,
y,
n_max,
beta,
deltaPix,
center_x=0,
center_y=0)
print(amp_out, 'amp_out')
for i in range(len(amp)):
print(self.shapeletSet.shapelets.index2poly(i))
for i in range(len(amp)):
print(i, 'i test')
npt.assert_almost_equal(amp_out[i], amp[i], decimal=2)
def test_function_split(self):
n_max = 2
beta = 10.
deltaPix = 0.1
amp = np.array([1, 1, 1, 1, 1, 1, 1, 1, 1])
x, y = util.make_grid(10, deltaPix, 1)
function_set = self.shapeletSet.function_split(x,
y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
test_flux = self.shapelets.function(x,
y,
amp=1.,
n=0,
m=0,
complex_bool=False,
beta=beta,
center_x=0,
center_y=0)
print(np.shape(function_set))
print(np.shape(test_flux))
assert function_set[0][10] == test_flux[10]
def test_index2poly(self):
index = 0
n, m, complex_bool = self.shapeletSet.index2poly(index)
assert n == 0
assert m == 0
assert complex_bool is False
def setup(self):
self.shapeletSet = ShapeletSetPolar()
self.shapelets = ShapeletsPolar()
self.x, self.y = util.make_grid(10, 0.1, 1)
class TestShapeletSetPolar(object):
"""
class to test Shapelets
"""
def setup(self):
self.shapeletSet = ShapeletSetPolar()
self.shapelets = ShapeletsPolar()
self.x, self.y = util.make_grid(10, 0.1, 1)
def test_shapelet_set(self):
"""
:return:
"""
n_max = 2
beta = 1.
amp = [1, 0, 0, 0, 0, 0]
output = self.shapeletSet.function(np.array(1),
np.array(1),
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output, 0.20755374871029739, decimal=8)
input = np.array(0.)
input += output
output = self.shapeletSet.function(self.x,
self.y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output[10], 0.47957022395315946, decimal=8)
output = self.shapeletSet.function(1,
1,
amp,
n_max,
beta,
center_x=0,
center_y=0)
npt.assert_almost_equal(output, 0.20755374871029739, decimal=8)
n_max = -1
beta = 1.
amp = [1, 0, 0, 0, 0, 0]
output = self.shapeletSet.function(np.array(1),
np.array(1),
amp,
n_max,
beta,
center_x=0,
center_y=0)
assert output == 0
def test_decomposition(self):
"""
:return:
"""
n_max = 2
beta = 10.
deltaPix = 2
amp = np.array([1, 1, -1, 1, 1, 1])
x, y = util.make_grid(100, deltaPix, 1)
input = self.shapeletSet.function(x,
y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
amp_out = self.shapeletSet.decomposition(input,
x,
y,
n_max,
beta,
deltaPix,
center_x=0,
center_y=0)
print(amp_out, 'amp_out')
for i in range(len(amp)):
print(i, 'i test')
npt.assert_almost_equal(amp_out[i], amp[i], decimal=4)
def test_function_split(self):
n_max = 2
beta = 10.
deltaPix = 0.1
amp = np.array([1, 1, 1, 1, 1, 1])
x, y = util.make_grid(10, deltaPix, 1)
function_set = self.shapeletSet.function_split(x,
y,
amp,
n_max,
beta,
center_x=0,
center_y=0)
test_flux = self.shapelets.function(x,
y,
amp=1.,
n=0,
m=0,
complex_bool=False,
beta=beta,
center_x=0,
center_y=0)
print(np.shape(function_set))
print(np.shape(test_flux))
assert function_set[0][10] == test_flux[10]