#!/usr/bin/env python3

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

from phoebe.parameters import hierarchy as _hierarchy

def test_spectroscopy6(verbose=True):

    phoebe.conf.devel_on()

    b = phoebe.default_star()

    b.set_value('distance', context='system', value=100*u_.pc)
    b.set_value('requiv@starA@component', value=1.0*u_.solRad)
    b.set_value('ntriangles@starA', context='compute', value=2000)

    f = open('twigs.txt', 'w')
    for twig in b.twigs:
      f.write("%s\n" % (twig))
    f.close()
 
    dir_ = os.path.dirname(os.path.realpath(__file__))

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

    b.set_value('columns', value=['abs_intensities@lc01'])

#    for val in [0.16, 0.17, 0.18]:
    for val in [0.16]:

        b.set_value('period@starA@component', value=val*u_.d)

        b.run_compute()

        fig, plt = b['mesh'].plot(show=True, fc='abs_intensities', ec=None, draw_sidebars=True, fcmap='gray', fclim=[0.0, 5.0e13])
        plt.savefig("mesh_%0.2f.png" % val)

if __name__ == "__main__":

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

    test_spectroscopy6()