Exemple #1
0
 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
Exemple #2
0
    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