# combined_spectra = ((transit - area[:, np.newaxis]) *
# spectrum_photo_flux + area[:, np.newaxis] *
# spectrum_spots_flux)
combined_spectra = transit * spectrum_photo_flux
#import ipdb; ipdb.set_trace()
# spectra = poisson(n_photons(wl, combined_spectra, mag,
# spectrum_photo.header, n_groups) *
# throughput(wl)[np.newaxis, :] * spitzer_var *
# (1 + flares) + background(wl, exptime)[np.newaxis, :])
spectra = poisson(n_photons(wl, combined_spectra, mag,
spectrum_photo.header, n_groups) *
throughput(wl)[np.newaxis, :] + background(wl, exptime)[np.newaxis, :])
# spectral_fluxes = np.sum(spectra, axis=1)
# plt.scatter(times, spectral_fluxes/spectral_fluxes.mean(),
# marker='.', s=4, label='spectrum model')
# plt.legend()
# plt.show()
subgroup.attrs['spot_radii'] = [s.r for s in star.spots]
subgroup.attrs['spot_contrast'] = star.spots[0].contrast
subgroup.attrs['t0'] = midtransit
subgroup.create_dataset('spectra', data=spectra, **dataset_kwargs)
subgroup.create_dataset('transit', data=transit, **dataset_kwargs)
subgroup.create_dataset('fluxes', data=fluxes, **dataset_kwargs)
subgroup.create_dataset('spotted_area', data=area, **dataset_kwargs)
# flares = inject_microflares(wl, times)
#spitzer_var = spitzer_variability(times)[:, np.newaxis]
granulation = k62_variability(times)[:, np.newaxis]
# oot = trappist_out_of_transit(times)
# planet_area = trappist1(planet).rp**2 * (1-trappist_out_of_transit(times).astype(int))
spectrum_photo_flux = spectrum_photo.interp_flux(wl)
#spectrum_spots_flux = spectrum_spots.interp_flux(wl)
combined_spectra = transit * spectrum_photo_flux
#import ipdb; ipdb.set_trace()
spectra = poisson(
n_photons(wl, combined_spectra, mag, spectrum_photo.header,
n_groups) * throughput(wl)[np.newaxis, :] * granulation *
(1 + flares) + background(wl, exptime)[np.newaxis, :])
# spectra = poisson(n_photons(wl, combined_spectra, mag,
# spectrum_photo.header, n_groups) *
# throughput(wl)[np.newaxis, :] + background(wl, exptime)[np.newaxis, :])
# spectral_fluxes = np.sum(spectra, axis=1)
# plt.scatter(times, spectral_fluxes/spectral_fluxes.mean(),
# marker='.', s=4, label='spectrum model')
# plt.legend()
# plt.show()
# subgroup.attrs['spot_radii'] = [s.r for s in star.spots]
# subgroup.attrs['spot_contrast'] = star.spots[0].contrast
subgroup.attrs['t0'] = midtransit
Time(midtransit, format='jd').isot))
star = Star.with_trappist1_spot_distribution()
area = star.spotted_area(times)
fluxes = star.fractional_flux(times)
#flares = inject_flares(wl, times)
flares = inject_microflares(wl, times)
spectra = np.zeros((len(times), len(wl)))
spitzer_var = spitzer_variability(times)
for i in range(len(times)):
combined_spectrum = ((1 - area[i]) * spectrum_photo +
area[i] * spectrum_spots)
spectra[i, :] = poisson(
combined_spectrum.n_photons(wl, exptime, mag) *
transit[i] * throughput(wl) * spitzer_var[i] *
(1 + flares[i, :]) + background(wl, exptime))
spectral_fluxes = np.sum(spectra, axis=1)
# plt.scatter(times, spectral_fluxes/spectral_fluxes.mean(),
# marker='.', s=4, label='spectrum model')
# plt.legend()
# plt.show()
subgroup.attrs['spot_radii'] = [s.r for s in star.spots]
subgroup.attrs['spot_contrast'] = star.spots[0].contrast
subgroup.attrs['t0'] = midtransit
subgroup.create_dataset('spectra', data=spectra, **dataset_kwargs)
subgroup.create_dataset('transit', data=transit, **dataset_kwargs)
subgroup.create_dataset('fluxes', data=fluxes, **dataset_kwargs)
subgroup.create_dataset('spotted_area',
#transit = transit_model(times, trappist1(planet))
transit = trappist1_all_transits(times)
subgroup = group.create_group("{0}".format(
Time(midtransit, format='jd').isot))
# star = Star.with_trappist1_spot_distribution()
# area = star.spotted_area(times)
# fluxes = star.fractional_flux(times)
# flares = inject_flares(wl, times)
# flares = inject_microflares(wl, times)
spitzer_var = spitzer_variability(times)[:, np.newaxis]
spectra = poisson(
spectrum_photo.n_photons(wl, exptime, mag) * transit *
throughput(wl)[np.newaxis, :] +
background(wl, exptime)[np.newaxis, :])
# spectral_fluxes = np.sum(spectra, axis=1)
# plt.scatter(times, spectral_fluxes/spectral_fluxes.mean(),
# marker='.', s=4, label='spectrum model')
# plt.legend()
# plt.show()
# subgroup.attrs['spot_radii'] = [s.r for s in star.spots]
# subgroup.attrs['spot_contrast'] = star.spots[0].contrast
subgroup.attrs['t0'] = midtransit
subgroup.create_dataset('spectra', data=spectra, **dataset_kwargs)
subgroup.create_dataset('transit', data=transit, **dataset_kwargs)
# subgroup.create_dataset('fluxes', data=fluxes, **dataset_kwargs)
# subgroup.create_dataset('spotted_area', data=area, **dataset_kwargs)
# oot = trappist_out_of_transit(times)
# planet_area = transit_params(planet).rp**2 * (1-trappist_out_of_transit(times).astype(int))
spectrum_photo_flux = spectrum_photo.interp_flux(wl)
spectrum_spots_flux = spectrum_spots.interp_flux(wl)
spectrum_secondary_flux = spectrum_secondary.interp_flux(wl)
combined_spectra = (dilution_factor * (
(transit - area[:, np.newaxis]) * spectrum_photo_flux +
area[:, np.newaxis] * spectrum_spots_flux) +
(1 - dilution_factor) * spectrum_secondary_flux)
# import ipdb; ipdb.set_trace()
spectra = poisson(
n_photons(wl, combined_spectra, exptime, mag, spectrum_photo.header
) * spitzer_var * throughput(wl)[np.newaxis, :] +
background(wl, exptime)[np.newaxis, :])
# spectral_fluxes = np.sum(spectra, axis=1)
# plt.scatter(times, spectral_fluxes/spectral_fluxes.mean(),
# marker='.', s=4, label='spectrum model')
# plt.legend()
# plt.show()
subgroup.attrs['spot_radii'] = [s.r for s in star.spots]
subgroup.attrs['spot_contrast'] = star.spots[0].contrast
subgroup.attrs['t0'] = midtransit
subgroup.create_dataset('spectra', data=spectra, **dataset_kwargs)
subgroup.create_dataset('transit', data=transit, **dataset_kwargs)
subgroup.create_dataset('fluxes', data=fluxes, **dataset_kwargs)
subgroup.create_dataset('spotted_area', data=area, **dataset_kwargs)