diff --git a/scripts/plot_noise_and_signal.py b/scripts/plot_noise_and_signal.py new file mode 100644 index 0000000..a8852e4 --- /dev/null +++ b/scripts/plot_noise_and_signal.py @@ -0,0 +1,58 @@ +# -*- coding: utf-8 -*- +# @Author: Elena Graverini +# @Date: 2017-04-28 15:34:06 +# @Last Modified by: Elena Graverini +# @Last Modified time: 2017-04-28 15:56:59 +from __future__ import print_function +import sys +import os +import ROOT as r + + +def draw_cali_step(i, sig, noi, layer, sector, ns): + cuts_noi = "(odinStep==%s && sector==%s)" % (i, sector) + cuts_sig = "(TrChi2/TrNDoF<3.0) &&"\ + "(GhostP<0.01+0.1/1.6*TrChi2/TrNDoF) &&"\ + "(GhostP<0.01-0.1/3.4*(TrChi2/TrNDoF-5.0))" + if 'IT' in layer: + cuts_sig = "(GhostP<0.1*TrChi2/TrNDoF) &&"\ + "(GhostP<0.2-0.1*(TrChi2/TrNDoF-5.0))" + cuts_sig = cuts_noi + '&&' + cuts_sig + c = r.TCanvas() + h_sig = r.TH1F("h_sig", "Signal histo", 208, -32.0, 176.0) + h_sig.SetLineWidth(3) + h_sig.SetLineColor(r.kBlue) + h_bkg = r.TH1F("h_bkg", "Background histo", 208, -32.0, 176.0) + h_bkg.SetLineWidth(3) + h_bkg.SetLineColor(r.kRed) + noi.Draw("val%s>>h_bkg" % ns, cuts_noi) + sig.Draw("val%s>>h_sig" % ns, cuts_sig, "same") + c.Modified() + c.Update() + os.system("mkdir -p plot_noise_and_signal") + c.SaveAs("%s_%s_val%s_step%s.pdf") + h_sig.Delete() + h_bkg.Delete() + + +if __name__ == '__main__': + + detector, layer = "TT", "TTaU" + ns = 5 + + location = os.path.expandvars('$DISK/data/ST/Aging_Tuples/') + macros = os.path.expandvars('$CCEHOME/macros/CCEScan/') + r.gROOT.SetBatch(True) + + fill = int(sys.argv[1]) + sector = int(sys.argv[2]) + + filename = location + '/' + detector + '/' + layer + '/%s.root' % fill + file = r.TFile(filename, 'read') + sig = file.Get('STADCTrackMonitor/HitInfo/%s' % layer) + noi = file.Get('STADCTrackMonitor/NoiseInfo/%s' % layer) + + for i in range(66): + draw_cali_step(i, sig, noi, layer, sector, ns) + + file.Close()