import CactusTool

name = 'RIT0377_res0'
sim = CactusTool.load(name, '/Volumes/simulations/HUST_Catalog', 'RIT0377')

# name = 'BBH0000_D25'
# sim = CactusTool.load(name, '/Volumes/simulations/HUST_Catalog')

trajectory = sim.ThornOutput('puncturetracker')
bbh = sim.ThornOutput('twopunctures')

trajectory.r.norm.preview()

trajectory.e_Omega(bbh.ADMMass, bbh.eta, bbh.chi1, bbh.chi2)
print("lambda_R:", trajectory.lambda_R)
print("lambda_T:", trajectory.lambda_T)

lambda_R: -0.35780127397455164
lambda_T: 0.9868898552161286

trajectory.eccentricity(p0=[0, 0, 0.01, 0.02, 0])

import matplotlib.pyplot as plt

tstart = 0
tend = -1
x1 = trajectory.x1[tstart:tend]
y1 = trajectory.y1[tstart:tend]
x2 = trajectory.x2[tstart:tend]
y2 = trajectory.y2[tstart:tend]
print(trajectory.t[tend])
plt.plot(x1, y1, label='BH 1')
plt.plot(x2, y2, label='BH 2')
plt.plot((x1+x2)/2, (y1+y2)/2)
plt.legend()
plt.title(name)
plt.xlabel('X [M]')
plt.ylabel('Y [M]')
plt.axis('equal')

print(bbh)

m1 = 0.5000000001
m2 = 0.5
ADMMass = 0.9915292239612099
Mass Ratio = 1.0000000002
eta = 0.25
separation = 12.55
spin1 = [0.2 0.  0. ]
spin2 = [0. 0. 0.]
chi1 = [0.8 0.  0. ]
chi2 = [0. 0. 0.]
chi_eff = 0.0
chi_p = 0.79999999968
Omega_orb = 0.02036642521583944
CutoffFrequency = 0.030549637823759163