#!/usr/bin/env python
# -*- coding: utf-8 -*-

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

from numpy import *
import numpy as np
import os
import sys
import re
from pygolem_lite.modules import list2array, multiplot, get_data
from pygolem_lite import Shot, saveconst, save_adv,load_adv


channels = [1,2]


  
def LoadData():
    
    Data = Shot()
    plasma_start = Data['plasma_start']
    plasma_end = Data['plasma_end']
    plasma = Data['plasma']
    return plasma_end,plasma_start, plasma


    
def Calc_Mean_RakeProbe():
    shot = Shot()
    t, data_st = shot['papouch_st']

    plasma_start = shot['plasma_start']
    plasma_end = shot['plasma_end']

    where_plasma = (plasma_start <= t) & (t <= plasma_end)

    rpm = [mean(data_st[where_plasma,i]) for i in range(data_st.shape[1])]
    
    positions = np.arange(1, len(rpm)+1)
    try:
        probe_position = float(re.findall('position=(\\d*)', str(shot['wwwcomment']))[0])
        positions *= 5
        positions += probe_position
        plt.xlabel('Radial probe position [mm]')
    except IndexError:
        plt.xlabel('Rake probe \#')
    

    plt.plot(positions, rpm,'ro')
    plt.ylabel('$U_{fl}$ mean')

    plt.savefig("graf.png")


def main():
    if sys.argv[1] ==  "analysis":
	Calc_Mean_RakeProbe()

	saveconst('status', 0)



if __name__ == "__main__":
    main()