dataset_kwargs = dict(compression='gzip') import sys planet = sys.argv[1] with ObservationArchive(run_name+'_'+planet, 'w') as obs: if 'wavelengths' not in obs.archive: obs.archive.create_dataset('wavelengths', data=wl) if planet in obs.archive: print('deleting old copy') del obs.archive[planet] group = obs.archive.create_group(planet) u1, u2 = trappist1(planet).u duration = transit_duration(trappist1(planet)) spectrum_photo = IRTFTemplate(sptype_phot) spectrum_spots = IRTFTemplate(sptype_spot)#spectrum_photo.scale_temperature(delta_teff) for midtransit in trappist_transits["{0} {1}".format(name, planet)]: print('midtransit', midtransit) times = np.arange(midtransit - 1.5*duration, midtransit + 1.5*duration, exptime.to(u.day).value) # times = np.arange(midtransit - 1*duration, midtransit, exptime.to(u.day).value) #transit = transit_model(times, trappist1(planet)) transit = trappist1_all_transits(times) subgroup = group.create_group("{0}".format(Time(midtransit, format='jd').isot))
def get_value(self, t): params = deepcopy(trappist1('b')) params.rp = self.depth**0.5 params.t0 = self.t0 + original_params.t0 return self.amp * transit_model(t, params)
def get_value(self, t): params = deepcopy(original_params) params.rp = self.depth**0.5 params.t0 = self.t0 return self.amp * transit_model(t, params) run_name = 'trappist1_bright' output_dir = '/gscratch/stf/bmmorris/libra/' from mpi4py import MPI #with ObservationArchive(run_name, 'a', outputs_dir=output_dir) as obs: for planet in list('h'): original_params = trappist1(planet) with ObservationArchive(run_name, 'a', comm=MPI.COMM_WORLD, driver='mpio') as obs: for obs_planet in getattr(obs, planet): print(planet, obs_planet.path) mask = mask_simultaneous_transits(obs_planet.times, planet) obs_time = obs_planet.times[mask] obs_flux = np.sum(obs_planet.spectra[mask], axis=1) obs_err = np.sqrt(obs_flux) params = trappist1(planet) # plt.plot(obs_time, obs_flux, '.') mid_transit_answer = obs_planet.attrs['t0']
import matplotlib.pyplot as plt import numpy as np import astropy.units as u import celerite from celerite import terms from scipy.optimize import minimize from celerite.modeling import Model from copy import deepcopy from astropy.utils.console import ProgressBar from libra import (trappist1, transit_model, ObservationArchive, trappist_out_of_transit, nirspec_pixel_wavelengths, transit_duration) t0 = trappist1('b').t0 original_params = trappist1('b') duration = transit_duration(original_params) times = np.arange(t0 - 1.5 * duration, t0 + 1.5 * duration, 1 / 60 / 60 / 24) import sys j = int(sys.argv[1]) outputs_dir = '/astro/store/scratch/tmp/bmmorris/libra/' with ObservationArchive('trappist1_bright2_b', 'r', outputs_dir=outputs_dir) as obs: # with ObservationArchive('trappist1_bright2_b', 'r') as obs: sim = obs.b[j] times = sim.times[:] spectra = sim.spectra[:] plt.plot(sim.times, np.sum(sim.spectra, axis=1))