%matplotlib inline

import numpy as np

from isochrones.dartmouth import Dartmouth_Isochrone
from isochrones.starmodel import BinaryStarModel, StarModel
from isochrones.starmodel import TripleStarModel

dar = Dartmouth_Isochrone()

mass = np.ones(100)
age, feh = (9.5,0.0)
AV = np.ones(100)*0.1
dist = np.arange(100)+1

dar(mass,age, feh, AV=AV, distance=dist)

mags = dict( W1=(11.808,0.022), W2=(11.834,0.021), W3=(11.675,0.170), 
            J=(12.119,0.021), H=(11.913,0.017), K=(11.873,0.019))

mod1 = StarModel(dar, maxAV=0.5, **mags)
mod2 = BinaryStarModel(dar, maxAV=0.5, **mags)
mod3 = TripleStarModel(dar, maxAV=0.5, **mags)

#mod1.fit_mcmc()
mod2.fit_mcmc()
#mod3.fit_mcmc()

mod2._make_samples(lnprob_thresh=0.01)

mod2.triangle_plots(extent=0.99);

mod1.triangle_plots();



mod3.triangle_plots();