import numpy as np
import matplotlib.pyplot as plt
import requests

#shot_no=39187 # standard
#shot_no=44914 #double
#shot_no=41642 #tripple
#shot_no=44625 #multi
#shot_no=39160 #no discharge

for shot_no in ['39187','44914','41642','44625','39160']:
    data = np.loadtxt("http://golem.fjfi.cvut.cz/discharges/" + str(shot_no) + "/Diagnostics/BasicDiagnostics/U_Loop.csv",delimiter=",")
    plasma_start=np.float64(requests.get("https://golem.fjfi.cvut.cz/getData.php?schema=diagnostics&table=plasmadetection&column=t_Plasma__start&ShotNo="+ str(shot_no)).text.strip())
    plasma_end=np.float64(requests.get("https://golem.fjfi.cvut.cz/getData.php?schema=diagnostics&table=plasmadetection&column=t_Plasma__end&ShotNo="+ str(shot_no)).text.strip())
    plt.plot(data[:,0], data[:,1]) #1. column vs 2. column
    plt.title('#'+str(shot_no))
    plt.xlabel('Time [ms]');plt.ylabel('$U_{Loop}$ [V]');
    plt.axvline(x=plasma_start, color='black', linestyle='--');
    plt.axvline(x=plasma_end, color='black', linestyle='--');
    plt.savefig('U_Loop-'+str(shot_no)+'.jpg')
    plt.show()