def test_shift_coords(self):
numPix = 10
deltaPix = 0.05
x_grid, y_grid, ra_at_xy_0, dec_at_xy_0, x_at_radec_0, y_at_radec_0, Mpix2coord, Mcoord2pix = util.make_grid_with_coordtransform(numPix=numPix, deltapix=deltaPix, subgrid_res=1, inverse=True)
# mask (1= model this pixel, 0= leave blanck)
kwargs_data = {'ra_at_xy_0': ra_at_xy_0, 'dec_at_xy_0': dec_at_xy_0,
'transform_pix2angle': Mpix2coord, 'image_data': np.ones((numPix, numPix))}
data = ImageData(**kwargs_data)
ra_shift = 0.05
dec_shift = 0.
kwargs_data['ra_shift'] = ra_shift
kwargs_data['dec_shift'] = dec_shift
data_shift = ImageData(**kwargs_data)
ra, dec = data.map_pix2coord(1, 1)
ra_new, dec_new = data_shift.map_pix2coord(1, 1)
npt.assert_almost_equal(ra_new - ra, ra_shift, decimal=10)
npt.assert_almost_equal(dec_new - dec, dec_shift, decimal=10)
ra_2, dec_2 = data_shift.map_pix2coord(2, 1)
npt.assert_almost_equal(ra, ra_2, decimal=10)
npt.assert_almost_equal(dec, dec_2, decimal=10)
x, y = data.map_coord2pix(0, 0)
x_new, y_new = data_shift.map_coord2pix(ra_shift, dec_shift)
npt.assert_almost_equal(x, x_new, decimal=10)
npt.assert_almost_equal(y, y_new, decimal=10)
def test_shift_coordinate_system(self):
x_shift = 0.05
y_shift = 0
numPix = 10
deltaPix = 0.05
x_grid, y_grid, ra_at_xy_0, dec_at_xy_0, x_at_radec_0, y_at_radec_0, Mpix2coord, Mcoord2pix = util.make_grid_with_coordtransform(
numPix=numPix, deltapix=deltaPix, subgrid_res=1, inverse=True)
kwargs_data = {'ra_at_xy_0': ra_at_xy_0, 'dec_at_xy_0': dec_at_xy_0,
'transform_pix2angle': Mpix2coord, 'image_data': np.ones((numPix, numPix))}
data = ImageData(**kwargs_data)
data_new = copy.deepcopy(data)
data_new.shift_coordinate_system(x_shift, y_shift, pixel_unit=False)
ra, dec = 0, 0
x, y = data.map_coord2pix(ra, dec)
x_new, y_new = data_new.map_coord2pix(ra + x_shift, dec + y_shift)
npt.assert_almost_equal(x, x_new, decimal=10)
npt.assert_almost_equal(y, y_new, decimal=10)
ra, dec = data.map_pix2coord(x, y)
ra_new, dec_new = data_new.map_pix2coord(x, y)
npt.assert_almost_equal(ra, ra_new-x_shift, decimal=10)
npt.assert_almost_equal(dec, dec_new-y_shift, decimal=10)
x_coords, y_coords = data.pixel_coordinates
x_coords_new, y_coords_new = data_new.pixel_coordinates
npt.assert_almost_equal(x_coords[0], x_coords_new[0]-x_shift, decimal=10)
npt.assert_almost_equal(y_coords[0], y_coords_new[0]-y_shift, decimal=10)
def setup(self):
# data specifics
sigma_bkg = .05 # background noise per pixel (Gaussian)
exp_time = 100. # exposure time (arbitrary units, flux per pixel is in units #photons/exp_time unit)
numPix = 100 # cutout pixel size
deltaPix = 0.05 # pixel size in arcsec (area per pixel = deltaPix**2)
fwhm = 0.1 # full width half max of PSF (only valid when psf_type='gaussian')
psf_type = 'GAUSSIAN' # 'GAUSSIAN', 'PIXEL', 'NONE'
# generate the coordinate grid and image properties
kwargs_data = sim_util.data_configure_simple(numPix, deltaPix, exp_time, sigma_bkg)
kwargs_data['exposure_time'] = exp_time * np.ones_like(kwargs_data['image_data'])
data_class = ImageData(**kwargs_data)
# generate the psf variables
kwargs_psf = {'psf_type': psf_type, 'pixel_size': deltaPix, 'fwhm': fwhm}
# kwargs_psf = sim_util.psf_configure_simple(psf_type=psf_type, fwhm=fwhm, kernelsize=kernel_size, deltaPix=deltaPix, kernel=kernel)
psf_class = PSF(**kwargs_psf)
# lensing quantities
kwargs_shear = {'gamma1': 0.02, 'gamma2': -0.04} # shear values to the source plane
kwargs_spemd = {'theta_E': 1.26, 'gamma': 2., 'center_x': 0.0, 'center_y': 0.0, 'e1': -0.1,
'e2': 0.05} # parameters of the deflector lens model
# the lens model is a supperposition of an elliptical lens model with external shear
lens_model_list = ['EPL', 'SHEAR']
kwargs_lens_true = [kwargs_spemd, kwargs_shear]
lens_model_class = LensModel(lens_model_list=lens_model_list)
# choice of source type
source_type = 'SERSIC' # 'SERSIC' or 'SHAPELETS'
source_x = 0.
source_y = 0.05
# Sersic parameters in the initial simulation
phi_G, q = 0.5, 0.8
e1, e2 = param_util.phi_q2_ellipticity(phi_G, q)
kwargs_sersic_source = {'amp': 1000, 'R_sersic': 0.05, 'n_sersic': 1, 'e1': e1, 'e2': e2, 'center_x': source_x,
'center_y': source_y}
# kwargs_else = {'sourcePos_x': source_x, 'sourcePos_y': source_y, 'quasar_amp': 400., 'gamma1_foreground': 0.0, 'gamma2_foreground':-0.0}
source_model_list = ['SERSIC_ELLIPSE']
kwargs_source_true = [kwargs_sersic_source]
source_model_class = LightModel(light_model_list=source_model_list)
lensEquationSolver = LensEquationSolver(lens_model_class)
x_image, y_image = lensEquationSolver.findBrightImage(source_x, source_y, kwargs_lens_true, numImages=4,
min_distance=deltaPix, search_window=numPix * deltaPix)
mag = lens_model_class.magnification(x_image, y_image, kwargs=kwargs_lens_true)
kwargs_numerics = {'supersampling_factor': 1}
imageModel = ImageModel(data_class, psf_class, lens_model_class, source_model_class,
kwargs_numerics=kwargs_numerics)
# generate image
model = imageModel.image(kwargs_lens_true, kwargs_source_true)
poisson = image_util.add_poisson(model, exp_time=exp_time)
bkg = image_util.add_background(model, sigma_bkd=sigma_bkg)
image_sim = model + bkg + poisson
data_class.update_data(image_sim)
kwargs_data['image_data'] = image_sim
kwargs_model = {'lens_model_list': lens_model_list,
'source_light_model_list': source_model_list,
}
# make cutous and data instances of them
x_pos, y_pos = data_class.map_coord2pix(x_image, y_image)
ra_grid, dec_grid = data_class.pixel_coordinates
multi_band_list = []
for i in range(len(x_pos)):
n_cut = 12
x_c = int(x_pos[i])
y_c = int(y_pos[i])
image_cut = image_sim[int(y_c - n_cut):int(y_c + n_cut), int(x_c - n_cut):int(x_c + n_cut)]
exposure_map_cut = data_class.exposure_map[int(y_c - n_cut):int(y_c + n_cut),
int(x_c - n_cut):int(x_c + n_cut)]
kwargs_data_i = {
'background_rms': data_class.background_rms,
'exposure_time': exposure_map_cut,
'ra_at_xy_0': ra_grid[y_c - n_cut, x_c - n_cut], 'dec_at_xy_0': dec_grid[y_c - n_cut, x_c - n_cut],
'transform_pix2angle': data_class.transform_pix2angle
, 'image_data': image_cut
}
multi_band_list.append([kwargs_data_i, kwargs_psf, kwargs_numerics])
kwargs_params = {'kwargs_lens': kwargs_lens_true, 'kwargs_source': kwargs_source_true}
self.multiPatch = MultiPatchPlot(multi_band_list, kwargs_model, kwargs_params, multi_band_type='joint-linear',
kwargs_likelihood=None, verbose=True, cmap_string="gist_heat")
self.data_class = data_class
self.model = model
self.lens_model_class = lens_model_class
self.kwargs_lens = kwargs_lens_true
