# coding: utf-8

# In[2]:

import numpy as np
from matplotlib.pyplot import *
from pylab import *
import matplotlib.image as mpimg
#from urllib  import urlopen #python 2.7 
from urllib.request  import urlopen #python 3.0 

ShotNo = 22471


#Create a path to data
#baseURL = "http://golem.fjfi.cvut.cz/utils/data/" #remote access
baseURL = "file:///golem/database/operation/shots/" #local access
#dataURL = urlopen(baseURL+ str(ShotNo) + '/' + diagnSPEC + '.npz')


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

sbp1=subplot(511)
dataURL = urlopen(baseURL+ str(ShotNo) + '/loop_voltage')
obj1=np.loadtxt(dataURL, delimiter='\t')
ylim(0,26);yticks(arange(0, 30, 5))
title('#'+str(ShotNo));ylabel('$U_l$ [V]')
plt.plot(obj1[:,0]*1000, obj1[:,1], 'k-',label='Loop voltage $U_l$' );legend(loc=0)

sbp1=subplot(512, sharex=sbp1)
dataURL = urlopen(baseURL+ str(ShotNo) + '/toroidal_field')
obj1=np.loadtxt(dataURL, delimiter='\t')
yticks(arange(0, 0.5 , 0.1));ylim(0,0.35)
ylabel('$B_t$ [T]')
plt.plot(obj1[:,0]*1000, obj1[:,1], 'k-',label='Toroidal mag. field $B_t$');legend(loc=0)

sbp1=subplot(513, sharex=sbp1)
dataURL = urlopen(baseURL+ str(ShotNo) + '/plasma_current')
obj1=np.loadtxt(dataURL, delimiter='\t')
yticks(arange(0, 4.5, 1));ylim(0,4.5)
ylabel('$I_p$ [kA]')
plt.plot(obj1[:,0]*1000, obj1[:,1]/1000, 'k-',label='Plasma current $I_p$');legend(loc=0)

sbp1=subplot(514, sharex=sbp1)
dataURL = urlopen(baseURL+ str(ShotNo) + '/photodiode_alpha')
obj1=np.loadtxt(dataURL, delimiter='\t')
yticks(arange(0, 0.09 , 0.02));ylim(0,0.09)
ylabel('Intensity [a.u.]')
plt.plot(obj1[:,0]*1000, obj1[:,1], 'k-',label='$H_\\alpha$ radiation');legend(loc=0)

sbp1=subplot(515, sharex=sbp1)
dataURL = urlopen(baseURL+ str(ShotNo) + '/electron_density')
obj2=np.loadtxt(dataURL, delimiter=' ')
yticks(arange(0, 0.8 , 0.2));ylim(0,0.8)
ylabel('$n_e$')
xticks(arange(8, 30, 5));xlim(5,25)
xlabel('Time [ms]')
plt.plot(obj2[:,0]*1000, obj2[:,1]*1e-19, 'k-',label='electron density $n_e$');legend(loc=0)

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

savefig('basicgraph.pdf')
savefig('basicgraph.jpg')

show()


# In[ ]:




# In[ ]:




# In[ ]: