Ejemplo n.º 1
0
    def _set_event_PSPL(self):
        """
        Sets internal variable to MulensModel.Event instance
        that uses PSPL model.
        """
        datasets = []
        for band in self.bands:
            mask = (self._filter == band)
            if not np.any(mask):
                continue
            times = self._JD[mask]
            flux = self._simulated_flux[band]
            sigma_flux = self._sigma_flux[band]
            plot = {}
            data = MM.MulensData([times, flux, sigma_flux],
                                 phot_fmt='flux',
                                 bandpass=band,
                                 plot_properties={'label': 'LSST ' + band})
            datasets.append(data)

        if self._follow_up_Chilean is not None:
            if self._follow_up_Chilean.n_epochs > 0:
                datasets.append(self._follow_up_Chilean)

        if self._follow_up_nonChilean is not None:
            if self._follow_up_nonChilean.n_epochs > 0:
                datasets.append(self._follow_up_nonChilean)

        model = MM.Model(
            {p: self._parameters[p]
             for p in ['t_0', 'u_0', 't_E']})
        self._event_PSPL = MM.Event(datasets, model)