import numpy as np
import BBHobject as BBH5
import sys


BBH1 = {'f_ini': 0.00046212835056309973, 'polarization': 4.272859502926378, 'mass1': 280533.87320000003, 'mass2': 39644.299799999993, 'longitude': 1.7468329951063812, 'spin2x': -0.089107513540094066, 'spin2y': -0.036672450110586896, 'spin2z': 0.57637078285907972, 'spin1y': 0.13218738769457944, 'spin1x': 0.18194033053713282, 'spin1z': 0.7627158337536476, 'latitude': -1.1835431460598578, 'phi_ref': 1.0210176124166828, 'model': 'SEOBNRv3', 'Lum_Distance': 19299.88338632301, 'cadence': 3.0, 'inclination': 1.5333000000000001}
units_BBH1 = {'f_ini': 'Hz', 'polarization': 'rad', 'mass1': 'solar mass',  'mass2': 'solar mass', 'longitude': 'rad',  'spin1x': 'dimensionless', 'spin1y': 'dimensionless', 'spin1z': 'dimensionless', 'spin2x': 'dimensionless', 'spin2y': 'dimensionless', 'spin2z': 'dimensionless', 'latitude': 'rad', 'phi_ref': 'rad', 'model': 'approximant', 'Lum_Distance': 'Mpc', 'cadence': 'sec', 'inclination': 'rad'}

print BBH1['spin2x']

for ky in BBH1.keys():
    print ky, units_BBH1[ky]


BBH2 = {'f_ini': 5.4149167944182696e-05, 'polarization': 2.513841157676472, 'mass1': 1017008.01, 'mass2': 855692.75199999986, 'longitude': 2.8528769795557283, 'spin2x': -0.007399579040544671, 'spin2y': 0.11838681079578713, 'spin2z': 0.6632972196983983, 'spin1y': 0.052815631062973156, 'spin1x': 0.07269447971335197, 'spin1z': 0.5338913576137236, 'latitude': 0.56960904449273952, 'phi_ref': 1.0210176124166828, 'model': 'SEOBNRv3', 'Lum_Distance': 33579.197472579268, 'cadence': 3.0, 'inclination': 1.1641}

units_BBH2 = {'f_ini': 'Hz', 'polarization': 'rad', 'mass1': 'solar mass',  'mass2': 'solar mass', 'longitude': 'rad',  'spin1x': 'dimensionless', 'spin1y': 'dimensionless', 'spin1z': 'dimensionless', 'spin2x': 'dimensionless', 'spin2y': 'dimensionless', 'spin2z': 'dimensionless', 'latitude': 'rad', 'phi_ref': 'rad', 'model': 'approximant', 'Lum_Distance': 'Mpc', 'cadence': 'sec', 'inclination': 'rad'}

tdiFile_BBH1 = "BBH_0-TDI.txt"

tdiFile_BBH2 = "BBH_1-TDI.txt"

#print BBH2['mass1']

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

stas.addSource("BBH1", BBH1, units_BBH1, overwrite=True, TDIfile=tdiFile_BBH1)
stas.addSource("BBH2", BBH2, units_BBH2, overwrite=True, TDIfile=tdiFile_BBH2)