%matplotlib inline

import numpy as np
import matplotlib.pyplot as plt

from isochrones.dartmouth import Dartmouth_Isochrone
dar = Dartmouth_Isochrone()

from isochrones.starmodel import StarModel

mod = StarModel(dar, Teff=(5642,50), logg=(4.44,0.06), feh=(-0.27,0.08))

mod.fit_mcmc()

mod.plot_samples('radius')
plt.axvline(0.979, color='r', ls='--')
mod.plot_samples('mass')
plt.axvline(0.970, color='r', ls='--')
mod.plot_samples('age')

from isochrones.dartmouth import MASTERDF
subdf = MASTERDF.query('feh > -0.31 and feh < -0.19 and M > 0.95 and M < 1 and logg > 4.42 and logg < 4.46')
plt.hist(np.array(10**subdf['logTeff']));

mod.save_hdf('kep22.h5')

mod2 = StarModel.load_hdf('kep22.h5')

mod2.plot_samples('radius')