counters.py

#!/usr/bin/python
import os
import math
import array
import numpy as np
import ROOT

import config
from config import *


COUNTERS      = ["Pixels/chip", "Clusters/chip", "Track Seeds", "Good Tracks", "Selected Tracks"]
counters_cols = [ROOT.kBlack,   ROOT.kBlue,      ROOT.kRed,     ROOT.kOrange+1, ROOT.kGreen+2 ]
counters_mrks = [23,            22,              26,            24,             20 ]

counters_x_trg = array.array('d')
counters_y_val = {}

def init_global_counters():
    for counter in COUNTERS: counters_y_val.update({counter:array.array('d')})

def append_global_counters():
    for counter in COUNTERS:
        counters_y_val[counter].append(0)

def set_global_counter(counter,idx,val):
    counters_y_val[counter][idx] = val


def plot_counters():
    gmax = -1e10
    gmin = +1e10
    for i,counter in enumerate(COUNTERS):
        mx = max(counters_y_val[counter])
        mn = min(counters_y_val[counter])
        gmax = mx if(mx>gmax) else gmax
        gmin = mn if(mn<gmin) else gmin
    # gmin = gmin if(gmin>0) else 0.5
    gmin = 0.5
    gmax = gmax*5
    print(f"gmin={gmin}, gmax={gmax}")
    
    graphs = {}
    mg = ROOT.TMultiGraph()
    # leg = ROOT.TLegend(0.0,0.1,1.0,0.9)
    # leg = ROOT.TLegend(0.14,0.13,0.88,0.3)
    leg = ROOT.TLegend(0.125,0.85,0.86,1)
    leg.SetNColumns(len(COUNTERS))
    leg.SetFillStyle(4000) # will be transparent
    leg.SetFillColor(0)
    leg.SetTextFont(42)
    leg.SetTextSize(0.037)
    leg.SetBorderSize(0)
    for i,counter in enumerate(COUNTERS):
        counter_name = counter.replace("/","_per_")
        gname = f"{counter}_vs_trg"
        graphs.update( {gname:ROOT.TGraph( len(counters_x_trg), counters_x_trg, counters_y_val[counter] )} )
        graphs[gname].SetBit(ROOT.TGraph.kIsSortedX)
        graphs[gname].GetXaxis().SetLimits(counters_x_trg[0]-2,counters_x_trg[-1]+2)
        graphs[gname].SetLineColor(counters_cols[i])
        graphs[gname].SetMarkerColor(counters_cols[i])
        graphs[gname].SetMarkerStyle(counters_mrks[i])
        graphs[gname].SetMarkerSize(0.8)
        graphs[gname].SetMaximum(gmax)
        graphs[gname].SetMinimum(gmin)
        leg.AddEntry(graphs[gname],f"{counter}","lp")
        mg.Add(graphs[gname])


    # cnv = ROOT.TCanvas("cnv_hits_vs_trg_all","",1200,500)
    # pL = ROOT.TPad("left_pad", "", 0.0, 0.0, 0.8, 1.0)
    # pR = ROOT.TPad("right_pad", "", 0.8, 0.0, 1.0, 1.0)
    # pR.SetMargin(0.0, 0.0, 0.0, 0.0)
    # pL.SetRightMargin(0.05)
    # pL.Draw()
    # pR.Draw()
    #
    # pL.cd()
    # pL.SetTicks(1,1)
    # pL.SetGridx()
    # pL.SetGridy()
    # pL.SetLogy()
    # mg.Draw("ap")
    # mg.SetTitle(f";Trigger number;Multiplicity")
    # mg.SetMaximum(gmax)
    # mg.SetMinimum(gmin)
    # mg.GetXaxis().SetLimits(counters_x_trg[0],counters_x_trg[-1])
    # pL.RedrawAxis()
    #
    # pR.cd()
    # leg.Draw()
    
    
    cnv = ROOT.TCanvas("cnv_hits_vs_trg_all","",1000,500)
    cnv.SetTicks(1,1)
    cnv.SetGridx()
    cnv.SetGridy()
    cnv.SetLogy()
    mg.Draw("alp")
    mg.SetTitle(f";Trigger number;Multiplicity")
    mg.SetMaximum(gmax)
    mg.SetMinimum(gmin)
    mg.GetXaxis().SetLimits(counters_x_trg[0]-2,counters_x_trg[-1]+2)
    cnv.RedrawAxis()
    leg.Draw()
    
    
    cnv.Update()
    cnv.SaveAs("multiplicities_vs_triggers.pdf")
    
    ctr = COUNTERS[-1]
    print(f"Avg+/-Std for {ctr}: {np.mean(counters_y_val[ctr])} +/- {np.std(counters_y_val[ctr])}")