def make_image_fixed_source(self, param, kwargs_options, kwargs_data, kwargs_lens, kwargs_source, kwargs_psf, kwargs_lens_light, kwargs_else, add_noise=False): """ make an image with fixed source parameters :param kwargs_lens: :param kwargs_source: :param kwargs_psf: :param kwargs_lens_light: :param kwargs_else: :return: """ subgrid_res = kwargs_options['subgrid_res'] num_order = kwargs_options.get('shapelet_order', 0) image = [] residuals = [] for i in range(self.num_bands(kwargs_data)): kwargs_data_i = kwargs_data[i] kwargs_psf_i = kwargs_psf["image" + str(i + 1)] param_i = param[i] deltaPix = kwargs_data_i['deltaPix'] x_grid, y_grid = kwargs_data_i['x_coords'], kwargs_data_i[ 'y_coords'] x_grid_sub, y_grid_sub = util.make_subgrid(x_grid, y_grid, subgrid_res) numPix = len(kwargs_data_i['image_data']) makeImage = ImageModel(kwargs_options, kwargs_data_i) image_i, error_map = makeImage.make_image_with_params( x_grid_sub, y_grid_sub, kwargs_lens, kwargs_source, kwargs_psf_i, kwargs_lens_light, kwargs_else, numPix, deltaPix, subgrid_res, param_i, num_order) residuals_i = util.image2array( makeImage.reduced_residuals(image_i, error_map)) residuals.append(residuals_i) if add_noise: image_i = makeImage.add_noise2image(image_i) image.append(image_i) return image, residuals
def make_mock_image(self, kwargs_options, kwargs_data, kwargs_lens, kwargs_source, kwargs_psf, kwargs_lens_light, kwargs_else, add_noise=False, noMask=True): """ :return: mock image, without substructure """ subgrid_res = kwargs_options['subgrid_res'] image = [] param = [] residuals = [] for i in range(self.num_bands(kwargs_data)): kwargs_data_i = kwargs_data[i] kwargs_psf_i = kwargs_psf["image" + str(i + 1)] deltaPix = kwargs_data_i['deltaPix'] x_grid, y_grid = kwargs_data_i['x_coords'], kwargs_data_i[ 'y_coords'] x_grid_sub, y_grid_sub = util.make_subgrid(x_grid, y_grid, subgrid_res) numPix = len(kwargs_data_i['image_data']) makeImage = ImageModel(kwargs_options, kwargs_data_i) if noMask: image_i, param_i, error_map = makeImage.make_image_ideal_noMask( x_grid_sub, y_grid_sub, kwargs_lens, kwargs_source, kwargs_psf_i, kwargs_lens_light, kwargs_else, numPix, deltaPix, subgrid_res, inv_bool=False) else: image_i, error_map, cov_param, param_i = makeImage.make_image_ideal( x_grid_sub, y_grid_sub, kwargs_lens, kwargs_source, kwargs_psf_i, kwargs_lens_light, kwargs_else, numPix, deltaPix, subgrid_res, inv_bool=False) residuals_i = util.image2array( makeImage.reduced_residuals(image_i, error_map)) residuals.append(residuals_i) if add_noise: image_i = makeImage.add_noise2image(image_i) image.append(image_i) param.append(param_i) return image, param, residuals