#!/usr/bin/python2
# -*- coding: utf-8 -*-

import matplotlib 
matplotlib.rcParams['backend'] = 'Agg'
matplotlib.rc('font',  size='10')
matplotlib.rc('text', usetex=True)  # FIXME !! nicer but slower !!!

from matplotlib.pyplot import *

from numpy import *
from pygolem_lite.config import *
from pygolem_lite.modules import *
from pygolem_lite import Shot

resonant_Bt = 0.0875
R = MajorRadius
a = MinorRadius

def where_first_or_nan(time, values, harmonic=1):
    possible = time[values >= resonant_Bt*harmonic]
    if len(possible) == 0:
        return nan
    else:
        return possible[0]


def make_plots():
    shot = Shot()
    preionization = shot["preionization"]
    # if preionization not in [4, 5, 6, 7]:
    #     saveconst("MWactive", 0)
    #     return
    # else:
    #     saveconst("MWactive", 1)
    t, mw_power = shot["nistandard", "mw_energy"]
    t, Iplasma = shot["plasma_current"]
    t, Uloop = shot["loop_voltage"]
    t, Bt = shot["toroidal_field"] 
    t, haphoto = shot["haphoto"]

    t *= 1e3

    Bt_LFS = R / (R + a) * Bt
    Bt_HFS = R / (R - a) * Bt

    loc = where(mw_power > mw_power.mean())[0]
    loc = where(mw_power > mw_power.mean())[0]
    mw_start = t[loc[0]]
    os.system(''.join(('echo ',str(mw_start),'>mw_start')))
    mw_end = t[loc[-1]]
    os.system(''.join(('echo ',str(mw_end),'>mw_end')))
                    

    t_resonnant_Bt_HFS = where_first_or_nan(t, Bt_HFS)
    os.system(''.join(('echo ',str(t_resonnant_Bt_HFS),'>t_resonnant_Bt_HFS')))
    t_resonnant_Bt = where_first_or_nan(t, Bt)
    os.system(''.join(('echo ',str(t_resonnant_Bt),'>t_resonnant_Bt')))
    t_resonnant_Bt_LFS = where_first_or_nan(t, Bt_LFS)
    os.system(''.join(('echo ',str(t_resonnant_Bt_LFS),'>t_resonnant_Bt_LFS')))
                        
    
    t_resonnant_2_Bt_HFS = where_first_or_nan(t, Bt_HFS, 2)
    t_resonnant_2_Bt = where_first_or_nan(t, Bt, 2)
    t_resonnant_2_Bt_LFS = where_first_or_nan(t, Bt_LFS, 2)

    def mark_regions():
        axvline(mw_start, c="g", ls="--")
        axvline(mw_end, c="g", ls="--")    
        axvline(t_resonnant_Bt, c="r", ls="--")
        axvline(t_resonnant_Bt_LFS, c="r", ls="--")
        axvline(t_resonnant_Bt_HFS, c="r", ls="--")

        axvline(t_resonnant_2_Bt, c="r", ls=":")
        axvline(t_resonnant_2_Bt_LFS, c="r", ls=":")
        axvline(t_resonnant_2_Bt_HFS, c="r", ls=":")

        
    figure(0)
    axes1 = subplot(511)
    axes1.plot(t, Bt_HFS, "m-", label=r"$B_t$ HFS")    
    axes1.plot(t, Bt, "b-", label=r"$B_t$ center")
    axes1.plot(t, Bt_LFS, "c-", label=r"$B_t$ LFS")
    ylabel(r"$B_t$ [T]")
    ylim(0, Bt_HFS.max()*1.1)
    axes1.axhline(resonant_Bt, c="k", label=("%.4f T" % resonant_Bt))
    axes1.axhline(resonant_Bt * 2, linestyle='--', c="k", label=("%.4f T" % (resonant_Bt * 2)))
    mark_regions()
    l = legend(loc=1, fancybox=True, prop={"size": 10})
    #l.get_frame().set_alpha(0.75)
    axes2 = subplot(512, sharex=axes1)
    axes2.plot(t, Uloop)
    ylabel(r"$U_{loop}$ [V]")
    u_max = shot["loop_voltage_max"]
    ylim(0, u_max+ 0.5)
    mark_regions()
    axes3 = subplot(513, sharex=axes1)
    axes3.plot(t, Iplasma)
    ylabel(r"$I_{plasma}$ [A]")
    ylim(ymin=0)
    mark_regions()
    axes4 = subplot(514, sharex=axes1)
    axes4.plot(t, mw_power)
    ylabel(r"$P_{MW}$ [a.u.]")
    mark_regions()
    axes5 = subplot(515, sharex=axes1)
    axes5.plot(t, haphoto)
    ylabel(r"$H_{\alpha}$ intensity [a.u.]")
    xlabel(r"$t$ [ms]")
    mark_regions()
    savefig("okenko.png")


if __name__ == "__main__":
    make_plots()
    saveconst('status', 0)