import numpy as np
from matplotlib.pyplot import *
from golem_data import golem_data
from pylab import *
import matplotlib.image as mpimg


shotno=11722


figure(num=None, figsize=(8, 5), dpi=80, facecolor='w', edgecolor='k')
obj1 = golem_data(shotno , 'loop_voltage')
ylabel('$U_l$ [V]')
xlabel('Time [ms]')
ylim(0,26)
xlim(5,25)
yticks(arange(0, 26, 5))
xticks(arange(8, 30, 5))
title('#'+str(shotno))
plot(obj1.tvec*1000,obj1.data,label='Loop voltage') 
legend(loc=0)
savefig('Ul.pdf')
savefig('Ul.jpg')


figure(num=None, figsize=(8, 5), dpi=80, facecolor='w', edgecolor='k')
obj1 = golem_data(shotno , 'rogowski_current')
ylabel('$I_{p+ch}$ [kA]')
xlabel('Time [ms]')
ylim(0,3.5)
xlim(5,25)
yticks(arange(0, 4, 0.5))
xticks(arange(8, 30, 5))
title('#'+str(shotno))
plot(obj1.tvec*1000,obj1.data/1000,label='Rogowski coil current') 
legend(loc=0)
savefig('I_p+ch.pdf')
savefig('I_p+ch.jpg')


#exit()

fig=figure(1)
subplots_adjust(hspace=0.001)


sbp1=subplot(411)
obj1 = golem_data(11722 , 'loop_voltage')
ylabel('$U_l$ [V]')
ylim(0,26)
yticks(arange(0, 26, 5))
title("#11722")
plot(obj1.tvec*1000,obj1.data,label='Loop voltage') 
legend(loc=0)

sbp1=subplot(412, sharex=sbp1)
obj1 = golem_data(11722 , 'toroidal_field')
ylabel('$B_t$ [T]')
yticks(arange(0, 0.5 , 0.05))
ylim(0,0.28)
plot(obj1.tvec*1000,obj1.data,label='Toroidal mag. field') 
legend(loc=0)

sbp1=subplot(413, sharex=sbp1)
obj2 = golem_data(11722 , 'plasma_current')
ylabel('$I_p$ [kA]')
yticks(arange(0, 3, 0.5))
ylim(0,3.2)
plot(obj2.tvec*1000,obj2.data/1000,  label='Plasma current') 
legend(loc=0)


subplot(414, sharex=sbp1)
obj1 = golem_data(11722 , 'photodiode_alpha')
ylabel('Intensity [a.u.]')
yticks(arange(0, 0.11 , 0.02))
plot(obj1.tvec*1000,obj1.data,label='$H_{\\alpha}$ radiation') 
xticks(arange(8, 30, 5))
xlim(5,25)
ylim(0,0.12)
xlabel('Time [ms]')
legend(loc=0)


xticklabels = sbp1.get_xticklabels()
setp(xticklabels, visible=False)

#show()
savefig('basicgraph.pdf')