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


#main script from the vagnetic profile animation generator. 
#Autor :Tomas Odstrcil


import time
cas = time.time()

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 os
import sys
from pygolem_lite.modules import saveconst
from scipy.signal import  fftconvolve,gaussian
from scipy.interpolate import interpolate
import MagFieldProfileGen

from pygolem_lite import Shot
print 'including time: ',  time.time()-cas




#constants
a = 0.085		#[m]
R_0 = 0.4		#[m]





  
def LoadData():
    Data = Shot()
    
    
    tvec, R_position =  Data['plasma_position_r']
    tvec, Z_position =  Data['plasma_position_z']

    Ip = array(Data['plasma_current'], copy = False).T


    return tvec,R_position,Z_position,Ip


def Resample(tvec,vec,t_min,t_max,n):
    
    std = (t_max-t_min)/(tvec[1]-tvec[0])/n/2
    gauss_win = gaussian(std*3,std)
    gauss_win/= sum(gauss_win)
    vec_smmoth = fftconvolve(vec,gauss_win, mode = 'same')
    tvec_new = linspace(t_min,t_max,n)
    f = interpolate.interp1d(tvec,vec_smmoth,bounds_error = False, fill_value = 0, copy = False)
    vec_new = f(tvec_new)
    

    return tvec_new,vec_new
    
    

def graphs():
    tvec,R_position,Z_position,Ip = LoadData()    
    tvec,Ip = Resample(Ip[:,0],Ip[:,1],tvec[0],tvec[-1],len(tvec)) 
    position = vstack((R_position,Z_position)).T
    
    
    #plot animation -- very slow!!!
    print "plotting animation" 
    MagFieldProfileGen.PlotProfile(tvec,Ip,position, 'animation')
    

def main():
    
    if not os.path.exists('video'):
	os.mkdir('video')

    if sys.argv[1] ==  "plots":
	graphs()
	saveconst('status', 0)



if __name__ == "__main__":
    main()