#!/usr/bin/env python3

"""
Problem: For keplerian, the c. o. m. is NOT in the centre of (u, v) coordinates! :-(

Problem: Ditto for RVs!

"""

import sys
import numpy as np
import phoebe
from astropy import units

from phoebe.parameters import hierarchy as _hierarchy

#logger = phoebe.logger(clevel='INFO')
#logger = phoebe.logger(clevel='DEBUG')

phoebe.conf.devel_on()

b = phoebe.Bundle()

b.add_star(component='starA', color='red')
b.add_star(component='starB', color='green')

b.add_orbit(component='orbit1', period=1.0)

hier1 = _hierarchy.binaryorbit(b['orbit1'], b['starA'], b['starB'])

b.set_hierarchy(hier1)

print("hier = ", b.get_hierarchy())

f = open('twigs.txt', 'w')
for twig in b.twigs:
  f.write("%s\n" % (twig))
f.close()

print("DONE!")

b.add_compute(dynamics_method='rebound')

b.add_dataset('mesh', times=[0.25])

b.set_value('q@orbit', value=0.1)

b.run_compute()

print("b['mass@starA@component'] = ", b['mass@starA@component'])
print("b['mass@starB@component'] = ", b['mass@starB@component'])

fig, plt = b.plot(show=True, x='us', y='vs')
plt.savefig('test_phoebe24_rebound.png')

sys.exit(1)