def get_rvs(self, p):
rvs = 0.0
if self.nkep > 0:
kep_ps = self.kep_deparameterize(self.kep_pars(p))
kep_rvs = kepler.rv_model(self.t, kep_ps[:,0], kep_ps[:,2], kep_ps[:,3], kep_ps[:,4], 2.0*np.pi/kep_ps[:,1])
rvs += np.sum(kep_rvs, axis=0)
if self.Pfixed is not None:
kep_ps = self.kep_fixed_deparameterize(self.kep_fixed_pars(p))
kep_rvs = kepler.rv_model(self.t, kep_ps[:,0], kep_ps[:,2], kep_ps[:,3], kep_ps[:,4], 2.0*np.pi/kep_ps[:,1])
rvs += np.sum(kep_rvs, axis=0)
return rvs
def rv_model(ts, K, e, omega, chi, P):
"""Returns the radial velocity measurements associated with the
planets in parameters ps at times ts. The returned array has
shape (Nts,)."""
assert ts.ndim == 1, "ts must be one-dimensional"
return kp.rv_model(ts, K, e, omega, chi, P)
def rv_model(ts, K, e, omega, chi, P):
"""Returns the radial velocity measurements associated with the
planets in parameters ps at times ts. The returned array has
shape (Nts,)."""
assert ts.ndim == 1, 'ts must be one-dimensional'
return kp.rv_model(ts, K, e, omega, chi, P)
def rv_model(ts, ps):
"""Returns the radial velocity measurements associated with the
planets in parameters ps at times ts. The returned array has
shape (Npl, Nts)."""
assert ts.ndim == 1, 'ts must be one-dimensional'
return kp.rv_model(ts, ps.K, ps.e, ps.omega, ps.chi, ps.n)
def get_rvs(self, p):
rvs = 0.0
if self.nkep > 0:
kep_ps = self.kep_deparameterize(self.kep_pars(p))
kep_rvs = kepler.rv_model(self.t, kep_ps[:, 0], kep_ps[:, 2],
kep_ps[:, 3], kep_ps[:, 4],
2.0 * np.pi / kep_ps[:, 1])
rvs += np.sum(kep_rvs, axis=0)
if self.Pfixed is not None:
kep_ps = self.kep_fixed_deparameterize(self.kep_fixed_pars(p))
kep_rvs = kepler.rv_model(self.t, kep_ps[:, 0], kep_ps[:, 2],
kep_ps[:, 3], kep_ps[:, 4],
2.0 * np.pi / kep_ps[:, 1])
rvs += np.sum(kep_rvs, axis=0)
return rvs