import numpy as np
import matplotlib.pyplot as plt
plt.style.use("dark_background")
"""IMPORT DATA"""
Pshotno=24940
Vshotno=24942
ds = np.DataSource(None)
PlasmaData = ds.open("http://golem.fjfi.cvut.cz/shots/"+str(Pshotno)+"//DAS/0416RigolDS1074a.ON/data_all")
VacuumData = ds.open("http://golem.fjfi.cvut.cz/shots/"+str(Vshotno)+"//DAS/0416RigolDS1074a.ON/data_all")
Readings=np.loadtxt(PlasmaData)
Vacuum=np.loadtxt(VacuumData)
"""INTERPRET DATA"""
Current = (-np.cumsum(Readings[:,2])) - (-np.cumsum(Vacuum[:,2]))
Uloop = -Readings[:,1]
Diode = -Readings[:,3]
Coil = np.cumsum(Readings[:,0])
"""FIGURE SETUP"""
with plt.rc_context({'axes.edgecolor':'white', 'xtick.color':'black', 'ytick.color':'black', 'figure.facecolor':'white'}):
fig = plt.figure(figsize=(10, 8))
fig.suptitle("Task 13", fontsize=16, color="black")
ax1 = plt.subplot2grid((4,1),(0,0))
ax1.tick_params(labelsize=6)
ax1.text(-105,9,"Toroidal\nMagnetic\nfield\n(integrated)", fontsize=9, style="italic", ha="center", color="black", bbox=dict(facecolor='#d998cb', boxstyle="round,pad=0.5"))
ax1.axvline(50, ls='-.', color='white')
ax1.axvline(950, ls='-.', color='white')
ax2 = plt.subplot2grid((4,1),(1,0))
ax2.tick_params(labelsize=6)
ax2.text(-105,9,"Loop\nVoltage", fontsize=9, style="italic", ha="center", color="black", bbox=dict(facecolor='#52bacc', boxstyle="round,pad=0.5"))
ax2.axvline(50, ls='-.', color='white')
ax2.axvline(950, ls='-.', color='white')
ax3 = plt.subplot2grid((4,1),(2,0))
ax3.tick_params(labelsize=6)
ax3.text(-105,20,"Plasma\ncurrent\n(integrated)", fontsize=9, style="italic", ha="center", color="black", bbox=dict(facecolor='#b6d957', boxstyle="round,pad=0.5"))
ax3.axvline(50, ls='-.', color='white')
ax3.axvline(950, ls='-.', color='white')
ax4 = plt.subplot2grid((4,1),(3,0))
ax4.tick_params(labelsize=6)
ax4.text(-105,0.1,"Diode\nsignal", fontsize=9, style="italic", ha="center", color="black", bbox=dict(facecolor='#fac364', boxstyle="round,pad=0.5"))
ax4.axvline(50, ls='-.', color='white')
ax4.axvline(950, ls='-.', color='white')
"""PLOT"""
ax1.plot(Coil, color="#d998cb")
ax2.plot(Uloop, color="#52bacc")
ax3.plot(Current, color="#b6d957")
ax4.plot(Diode, color="#fac364")
#print(plt.style.available)
plt.show()
