remove duplicated imports; add file paths in marantares2; and other changes for plotting

REINFORCE_Timeslices.py

@@ -5,40 +5,49 @@
 import matplotlib as mpl
 import pandas as pd               # the main HDF5 reader
 import numpy as np                # must have
-import km3pipe as kp              # some KM3NeT related helper functions
+#import km3pipe as kp              # some KM3NeT related helper functions
 import seaborn as sns              # beautiful statistical plots!
 import sys, itertools, aa, collections
 from ROOT import EventFile, Det, Timer
-import matplotlib.pylab as pylab
-pylab.rcParams['figure.figsize'] = 10, 6
 import glob
 import os, sys, glob, numpy, matplotlib, ROOT
 from matplotlib import pyplot
-import pylab as P
 import csv, math
-import matplotlib as plt
-get_ipython().magic(u'pylab inline')
-import matplotlib.pylab as pylab
-pylab.rcParams['figure.figsize'] = 14, 10
-import numpy as np
-import pylab as P
+import matplotlib.pylab as plt
 from numpy import sin, cos, pi, linspace
 from numpy.random import randn
 from scipy.signal import lfilter, lfilter_zi, filtfilt, butter
 
 from matplotlib.pyplot import plot, legend, show, grid, figure, savefig
-pylab.rcParams['figure.figsize'] = 14, 10
+plt.rcParams['figure.figsize'] = 14, 10
 from datetime import datetime, timedelta, time
 
 
-from ROOT import EventFile, Det
-import aa
 EventFile.read_timeslices = True
 EventFile.timeslice_treename = "KM3NET_TIMESLICE_SN"
 
 
-det = Det("/pbs/throng/km3net/detectors/KM3NeT_00000029_20170920.detx")
-f = EventFile("/sps/km3net/users/gdewasse/L0-files/KM3NeT_00000029_00002984.root") #3094 - 2984 #Feel free to add more runs!
+
+# files on cc-lyon:
+#filename="/sps/km3net/users/gdewasse/L0-files/KM3NeT_00000029_00002984.root"
+#detxname="/pbs/throng/km3net/detectors/KM3NeT_00000029_20170920.detx"
+
+# files on marantares2:
+filename="/baie/huang/data/KM3NeT_00000049_00009050.root"
+detxname="/baie/huang/data/KM3NeT_00000049_20200210.detx"
+#filename="/baie/huang/data/L0-files/KM3NeT_00000029_00002984.root"
+#detxname="/baie/huang/data/L0-files/KM3NeT_00000029_20170920.detx"
+
+det = Det(detxname)
+f = EventFile(filename) #3094 - 2984 #Feel free to add more runs!
+
+df_pmt_channel_id = pd.DataFrame( [['A', 'A1', 22],
+         ['B', 'B1', 14], ['B', 'B2', 19], ['B', 'B3', 25], ['B', 'B4', 24], ['B', 'B5', 26], ['B', 'B6', 18],
+         ['C', 'C1', 13], ['C', 'C2', 21], ['C', 'C3', 29], ['C', 'C4', 28], ['C', 'C5', 20], ['C', 'C6', 17],
+         ['D', 'D1', 12], ['D', 'D2', 15], ['D', 'D3', 23], ['D', 'D4', 30], ['D', 'D5', 27], ['D', 'D6', 16],
+         ['E', 'E1', 10], ['E', 'E2', 6 ], ['E', 'E3', 3 ], ['E', 'E4', 2 ], ['E', 'E5', 1 ], ['E', 'E6', 11],
+         ['F', 'F1', 9 ], ['F', 'F2', 8 ], ['F', 'F3', 4 ], ['F', 'F4', 0 ], ['F', 'F5', 5 ], ['F', 'F6', 7 ]],
+         columns=('ring', 'PMT_position', 'channel_id'))
 
 
 R = []
@@ -53,6 +62,9 @@ RSUM = []
 T = [] 
 DATE = []
 i = 0 
+select_dom_id = 808982077   #if dom_id == 808982077 : #808432835:809544061:808982077:808997793  #some DOM IDs
+print("events in file" , f.size())
+print("events in file" , f.index)
 for evt in f:
     if i == 0 :
         #print('time', evt.t.AsString())
@@ -63,17 +75,19 @@ for evt in f:
     f.get_rates( det ) 
     event_time = evt.t.GetSec() + 1e-9 * evt.t.GetNanoSec()
     # now, det knows the rate on each PMT
-    for dom_id, dom in det.doms :
-        rates = [ pmt.rate for pmt in dom.pmts ]
-        #print( dom_id , rates[:5] )
-        if dom_id == 808982077 : #808432835:809544061:808982077:808997793  #some DOM IDs
-                R.append(rates[0])
-                R2.append(rates[3])
-                R3.append(rates[6])
-                #print(event_time)
-                TIME.append(event_time - time_first_event)
-                Rsum.append(sum(rates))
-    #print(max(TIME) - min(TIME)) 
+    #for dom_id, dom in det.doms :
+    #    rates = [ pmt.rate for pmt in dom.pmts ]
+    #    #print( dom_id , rates[:5] )
+    dom = det.doms[select_dom_id]
+    rates = [ pmt.rate for pmt in dom.pmts ]
+    dict_pmt_rates = [pmt.id : pmt.rate for pmt in dom.pmts ]
+    print("dict_pmt_rates", dict_pmt_rates)
+    R.append(rates[0])
+    R2.append(rates[3])
+    R3.append(rates[6])
+    #print(event_time)
+    TIME.append(event_time - time_first_event)
+    Rsum.append(sum(rates))
     if max(TIME) - min(TIME) > 120: 
         DATE.append(evt.t.AsString())
         T.append(TIME)
@@ -90,22 +104,26 @@ for evt in f:
     #print("\n\n\n")
 
 
+print("TIME", TIME)
+print("T", T)
+print("R", R)
 i = 0 
 while i < len(RR):
     
-    fig = pylab.figure()
+    fig = plt.figure()
     
     plt.plot(T[i], RSUM[i], linestyle = 'None', marker = 'o')
-    v = [min(TIME), max(TIME), 190e3, 350e3]#220000, 310000]
-    axis(v)
-    pylab.xlabel(r'Time [s]', size = 20)
-    pylab.ylabel(r'Rate [kHZ]', size = 20)
-    pylab.xticks([0,10,20,30, 40,50,60, 70,80, 90,100,110, 120], size=20)
-    pylab.yticks([190e3,200e3,210e3, 220e3,230e3,240e3,250e3,260e3,270e3,280e3,290e3,300e3,310e3,320e3,330e3,340e3,350e3],['190', '200', '210','220','230','240','250','260','270','280','290','300','310', '320', '330', '340','350'],size=20)
+    #v = [min(TIME), max(TIME), 190e3, 350e3]#220000, 310000]
+    #plt.axis(v)
+    plt.xlabel(r'Time [s]', size = 20)
+    plt.ylabel(r'Rate [kHZ]', size = 20)
+    plt.xticks([0,10,20,30, 40,50,60, 70,80, 90,100,110, 120], size=20)
+    #plt.yticks([190e3,200e3,210e3, 220e3,230e3,240e3,250e3,260e3,270e3,280e3,290e3,300e3,310e3,320e3,330e3,340e3,350e3],['190', '200', '210','220','230','240','250','260','270','280','290','300','310', '320', '330', '340','350'],size=20)
     plt.grid()
-    plt.savefig((str(DATE[i])))
+    plt.savefig(("plots/"+str(DATE[i])+".png"))
+    plt.savefig(("plots/"+str(DATE[i])+".pdf"))
     #fig.save()
-    plt.show()
+    #plt.show()
     
     i+=1
     #plt.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3,
@@ -123,7 +141,6 @@ while i < len(T):
     INFO_T.append(info_T)
     i+=1
 
-
 i = 0 
 info_sorted = []
 while i < len(INFO_T):
@@ -134,12 +151,13 @@ while i < len(INFO_T):
 i = 0
 f = open('REINFORCE_DATA.txt', 'w')
 
-while i < 2:#len(info_sorted):
+#while i < 2:#len(info_sorted):
+while i < len(info_sorted):
     j = 0
     f.write('New event \n')
     f.write(info_sorted[i][0][2])
     while j < len(info_sorted[i]):
-        print(info_sorted[i][j][0],info_sorted[i][j][1])
+        #print(info_sorted[i][j][0],info_sorted[i][j][1])
         L = [str(info_sorted[i][j][0]), ' ', str(info_sorted[i][j][1]), '\n']
         f.writelines(L) 
         j+=1