import numpy as np
import BBHobject as BBH5

stas = BBH5.BBHfile("stas_test1.hdf5")

pars = {"m1" : 5.0e6, "m2" : 1.0e6, "a1" : 0.4, "a2" : 0.78, "model" : "SEOBNRv2"}
units = {'m1': "soalr mass", 'm2': "solar mass", 'a1': "mass^2", 'a2': "mass^2", 'model': "approximant"]

stas.addSource("Stas_BBH1", pars, units, overwrite=True)


pars = {"m1" : 3.2e6, "m2" : 1.5e6, "a1" : 0.83, "a2" : 0.78, "model" : "SEOBNRv2"}
units = {'m1': "soalr mass", 'm2': "solar mass", 'a1': "mass^2", 'a2': "mass^2", 'model': "approximant"]


stas.addSource("Stas_BBH2", pars, units, overwrite=True)

print stas.getNumSources()

srcs =  stas.readSources("stas_test1.hdf5")

[source0, attr]  = srcs[0]

print 10*"---"
print source0['m1']