def create_testcase04():
bjd0 = bjd_obs - Tref
y_pla = instrument['RV_offset1'] + \
kp.kepler_RV_T0P(bjd0,
planet_b['f'],
planet_b['P'],
planet_b['K'],
planet_b['e'],
planet_b['o']) + \
kp.kepler_RV_T0P(bjd0,
planet_c['f'],
planet_c['P'],
planet_c['K'],
planet_c['e'],
planet_c['o'])
mod_pl = np.random.normal(y_pla, instrument['RV_precision'])
#Tcent_b = kp.kepler_Tcent_T0P(planet_b['P'], planet_b['f'], planet_b['e'], planet_b['o']) + Tref
Tcent_b = np.random.normal(
np.arange(0, 10) * planet_b['P'] + kp.kepler_Tcent_T0P(
planet_b['P'], planet_b['f'], planet_b['e'], planet_b['o']) + Tref,
instrument['T0_precision'])
Tcent_c = np.random.normal(
np.arange(0, 10) * planet_c['P'] + kp.kepler_Tcent_T0P(
planet_c['P'], planet_c['f'], planet_c['e'], planet_c['o']) + Tref,
instrument['T0_precision'])
fileout = open('TestCase04_RV.dat', 'w')
for b, r in zip(bjd_obs, mod_pl):
fileout.write('{0:f} {1:.2f} {2:.2f} {3:d} {4:d} {5:d}\n'.format(
b, r, instrument['RV_precision'], 0, 0, -1))
fileout.close()
fileout = open('TestCase04_Tcent_b.dat', 'w')
for i_Tc, v_Tc in enumerate(Tcent_b):
fileout.write('{0:d} {1:.4f} {2:.4f} {3:d}\n'.format(
i_Tc, v_Tc, instrument['T0_precision'], 0))
fileout.close()
fileout = open('TestCase04_Tcent_c.dat', 'w')
for i_Tc, v_Tc in enumerate(Tcent_c):
fileout.write('{0:d} {1:.4f} {2:.4f} {3:d}\n'.format(
i_Tc, v_Tc, instrument['T0_precision'], 0))
fileout.close()
def create_testcase08():
bjd0 = bjd_obs - Tref
y_pla = kp.kepler_RV_T0P(bjd0,
planet_b['f'],
planet_b['P'],
planet_b['K'],
planet_b['e'],
planet_b['o']) \
+ instrument['RV_offset1'] \
+ polynomial_trend['c1']*bjd0 + polynomial_trend['c2']*bjd0*bjd0
mod_pl = np.random.normal(y_pla, instrument['RV_precision'])
Tcent_b = np.random.normal(
np.arange(0, 5)*planet_b['P'] + kp.kepler_Tcent_T0P(planet_b['P'], planet_b['f'], planet_b['e'], planet_b['o']) + Tref,
instrument['T0_precision'])
fileout = open('TestCase08_RV.dat', 'w')
for b, r in zip(bjd_obs, mod_pl):
fileout.write('{0:f} {1:.2f} {2:.2f} {3:d} {4:d} {5:d}\n'.format(b, r, instrument['RV_precision'], 0, 0, -1))
fileout.close()
fileout = open('TestCase08_Tcent_b.dat', 'w')
for i_Tc, v_Tc in enumerate(Tcent_b):
fileout.write('{0:d} {1:.4f} {2:.4f} {3:d}\n'.format(i_Tc, v_Tc, instrument['T0_precision'], 0))
fileout.close()
def create_test2():
x = np.arange(6000, 6100, 1, dtype=np.double)
x = np.random.normal(x, 0.4)
Tref = np.mean(x, dtype=np.double)
x0 = x - Tref
print Tref
P = 23.4237346
K = 43.47672
phase = 0.34658203
e = 0.13
omega = 0.673434
offset = 45605
y_pla = kp.kepler_RV_T0P(x0, phase, P, K, e, omega) + offset
mod_pl = np.random.normal(y_pla, 2)
trip = np.arange(10, 3, -1)
Transit_Time = kp.kepler_Tcent_T0P(P, phase, e, omega) + Tref - trip * P
Transit_Time = kp.kepler_Tcent_T0P(P, phase, e, omega) + Tref
print 'Transit Time:', Transit_Time
print 'transit Time - Tref', Transit_Time - Tref
plt.scatter(x, mod_pl)
plt.axvline(Transit_Time)
plt.show()
fileout = open('test2_RV.dat', 'w')
for ii in xrange(0, np.size(x)):
fileout.write('{0:14f} {1:14f} {2:14f} {3:5d} {4:5d} {5:5d} \n'.format(
x[ii], mod_pl[ii], 2., 0, 0, -1))
fileout.close()
fileout = open('test2_Tcent_0.dat', 'w')
for ii in xrange(0, np.size(Transit_Time)):
fileout.write('{0:14f} {1:14f} \n'.format(Transit_Time, 0.01))
fileout.close()
def phase_calc(p, t_obs, period, e, omega):
t_mod = kp.kepler_Tcent_T0P(period, p, e, omega)
return t_obs-t_mod
def create_testcase07():
import george
bjd0 = bjd_obs - Tref
err = bjd0 * 0 + instrument['RV_precision']
gp_pams = np.zeros(4)
gp_pams[0] = np.log(activity['Hamp_RV1']) * 2
gp_pams[1] = np.log(activity['Pdec'])*2
gp_pams[2] = 1. / (2*activity['Oamp'] ** 2)
gp_pams[3] = np.log(activity['Prot'])
kernel = np.exp(gp_pams[0]) * \
george.kernels.ExpSquaredKernel(metric=np.exp(gp_pams[1])) * \
george.kernels.ExpSine2Kernel(gamma=gp_pams[1], log_period=gp_pams[2])
gp = george.GP(kernel)
gp.compute(bjd0, err)
prediction1 = gp.sample(bjd0)
gp_pams[0] = np.log(activity['Hamp_RV2']) * 2
kernel = np.exp(gp_pams[0]) * \
george.kernels.ExpSquaredKernel(metric=np.exp(gp_pams[1])) * \
george.kernels.ExpSine2Kernel(gamma=gp_pams[1], log_period=gp_pams[2])
gp = george.GP(kernel)
gp.compute(bjd0, err)
prediction2 = gp.sample(bjd0)
y_pla = kp.kepler_RV_T0P(bjd0,
planet_b['f'],
planet_b['P'],
planet_b['K'],
planet_b['e'],
planet_b['o'])
mod_pl1 = np.random.normal(y_pla + prediction1 + instrument['RV_offset1'], instrument['RV_precision'])
mod_pl2 = np.random.normal(y_pla + prediction2 + instrument['RV_offset2'], instrument['RV_precision'])
Tcent_b = np.random.normal(
np.arange(0,1)*planet_b['P'] + kp.kepler_Tcent_T0P(planet_b['P'], planet_b['f'], planet_b['e'], planet_b['o']) + Tref,
instrument['T0_precision'])
fileout = open('TestCase07_RV_dataset1.dat', 'w')
for b, r in zip(bjd_obs, mod_pl1):
fileout.write('{0:f} {1:.2f} {2:.2f} {3:d} {4:d} {5:d}\n'.format(b, r, instrument['RV_precision'], 0, 0, -1))
fileout.close()
fileout = open('TestCase07_RV_dataset2.dat', 'w')
for b, r in zip(bjd_obs, mod_pl2):
fileout.write('{0:f} {1:.2f} {2:.2f} {3:d} {4:d} {5:d}\n'.format(b, r, instrument['RV_precision'], 0, 0, -1))
fileout.close()
fileout = open('TestCase07_Tcent_b.dat', 'w')
for i_Tc, v_Tc in enumerate(Tcent_b):
fileout.write('{0:d} {1:.4f} {2:.4f} {3:d}\n'.format(i_Tc, v_Tc, instrument['T0_precision'], 0))
fileout.close()
def create_testcase06():
import george
bjd0 = photometry['phot_bjd'] - Tref
err = bjd0 * 0 + photometry['phot_precision']
""" Conversion of the physical parameter to the internally defined parameter
to be passed to george
"""
gp_pams = np.zeros(4)
gp_pams[0] = np.log(activity['Hamp_PH']) * 2
gp_pams[1] = np.log(activity['Pdec']) * 2
gp_pams[2] = 1. / (2 * activity['Oamp'] ** 2)
gp_pams[3] = np.log(activity['Prot'])
kernel = np.exp(gp_pams[0]) * \
george.kernels.ExpSquaredKernel(metric=np.exp(gp_pams[1])) * \
george.kernels.ExpSine2Kernel(gamma=gp_pams[1], log_period=gp_pams[2])
gp = george.GP(kernel)
gp.compute(bjd0, err)
prediction = gp.sample(bjd0)
obs_photometry = np.random.normal(prediction, instrument['RV_precision'])
fileout = open('TestCase06_photometry.dat', 'w')
for b, p in zip(photometry['phot_bjd'], obs_photometry):
fileout.write('{0:14f} {1:14f} {2:14f} {3:5d} {4:5d} {5:5d} \n'.format(
b, p, photometry['phot_precision'], 0, 0, -1))
fileout.close()
bjd0 = bjd_obs - Tref
err = bjd0 * 0 + instrument['RV_precision']
gp_pams[0] = np.log(activity['Hamp_RV1']) * 2
kernel = np.exp(gp_pams[0]) * \
george.kernels.ExpSquaredKernel(metric=np.exp(gp_pams[1])) * \
george.kernels.ExpSine2Kernel(gamma=gp_pams[1], log_period=gp_pams[2])
gp = george.GP(kernel)
gp.compute(bjd0, err)
prediction = gp.sample(bjd0)
y_pla = kp.kepler_RV_T0P(bjd0,
planet_b['f'],
planet_b['P'],
planet_b['K'],
planet_b['e'],
planet_b['o']) + instrument['RV_offset1']
mod_pl = np.random.normal(y_pla + prediction, instrument['RV_precision'])
Tcent_b = np.random.normal(
np.arange(0,1)*planet_b['P'] + kp.kepler_Tcent_T0P(planet_b['P'], planet_b['f'], planet_b['e'], planet_b['o']) + Tref,
instrument['T0_precision'])
fileout = open('TestCase06_RV.dat', 'w')
for b, r in zip(bjd_obs, mod_pl):
fileout.write('{0:f} {1:.2f} {2:.2f} {3:d} {4:d} {5:d}\n'.format(b, r, instrument['RV_precision'], 0, 0, -1))
fileout.close()