# 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)
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) # subgroup.create_dataset('flares', data=1 + flares, **dataset_kwargs)
#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 subgroup.create_dataset('spectra', data=spectra, **dataset_kwargs)
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', data=area,