3_bunchFinderPlots.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
<header>
  <input>EvtGenSimRec.root, EvtGenSimRec_B2Kpi.root</input>
  <output>bunchFinder.root</output>
  <contact>marko.staric@ijs.si</contact>
  <description>makes validation plots for TOPBunchFinder</description>
</header>
"""
 
from basf2 import create_path, process, statistics, Module
from ROOT import Belle2
from ROOT import TH1F, TFile, TNamed
import math
 
 
class MakePlots(Module):
    '''
    Makes validation histograms for bunch finder
    '''
 
    def set_descr(self, h, D):
        '''
        Sets description, check and contact to validation histogram.
        :param h validation histogram
        :param D description text
        '''
        descr = TNamed('Description', D)
        h.GetListOfFunctions().Add(descr)
        check = TNamed('Check', 'Must agree with reference')
        h.GetListOfFunctions().Add(check)
        contact = TNamed('Contact', 'marko.staric@ijs.si')
        h.GetListOfFunctions().Add(contact)
 
    def initialize(self):
        ''' Initializes the Module: book histograms and set descriptions and checks'''
 
        
        self.tfiletfile = TFile.Open("bunchFinder.root", "recreate")
 
        
        self.recBunchNorecBunchNo = TH1F("recBunchNo",
                               "Reconstructed relative bunch number; relative bunch number; events per bunch",
                               100, -50.0 - 0.5, 50.0 - 0.5)
        self.set_descrset_descr(self.recBunchNorecBunchNo, 'Reconstructed bunch number relative to L1 trigger.')
 
        
        self.offsetoffset = TH1F("offset",
                           "Offset to reconstructed bunch; offset [ns]; events per bin",
                           100, -1.0, 1.0)
        self.set_descrset_descr(self.offsetoffset, 'Offset to reconstructed bunch.')
 
        
        self.numTracksnumTracks = TH1F("numTracks",
                              "Number of tracks in TOP acceptance; number of tracks; entries per bin",
                              10, 0.5, 10.5)
        self.set_descrset_descr(self.numTracksnumTracks, 'Number of tracks per event in the acceptance of TOP.')
 
        
        self.usedTracksusedTracks = TH1F("usedTracks",
                               "Number of tracks used for bunch finding; number of used track; entries per bins",
                               10, 0.5, 10.5)
        self.set_descrset_descr(self.usedTracksusedTracks, 'Number of tracks in the event actually used for finding the bunch crossing.')
 
        
        self.effi_numTrackseffi_numTracks = TH1F("effi_numTracks",
                                   "Efficiency vs. number of tracks; number of tracks; efficiency",
                                   10, 0.5, 10.5)
        self.set_descrset_descr(self.effi_numTrackseffi_numTracks, 'Efficieny of finding the correct bunch crossing versus '
                       'the number of tracks per event in the acceptance of TOP.')
 
        
        self.effi_usedTrackseffi_usedTracks = TH1F("effi_usedTracks",
                                    "Efficiency vs. number of used tracks; number of used tracks; efficiency",
                                    10, 0.5, 10.5)
        self.set_descrset_descr(self.effi_usedTrackseffi_usedTracks, 'Efficieny of finding the correct bunch crossing versus '
                       'the number of tracks per event actually used.')
 
        
        self.effieffi = TH1F("effi", "Acceptance and efficiency", 2, 0.5, 2.5)
        self.effieffi.GetXaxis().SetBinLabel(1, "acceptance")
        self.effieffi.GetXaxis().SetBinLabel(2, "efficiency")
        self.effieffi.GetXaxis().SetLabelSize(0.08)
        self.effieffi.GetXaxis().SetAlphanumeric()
        self.set_descrset_descr(self.effieffi, 'Acceptance is defined as fraction of events with at least one track crossing TOP. '
                       'Efficiency is defined as fraction of correctly reconstructed bunch crossings in accepted events.')
 
        
        self.nevnev = 0
 
    def event(self):
 
        ''' Event processor: fill histograms '''
 
        recBunch = Belle2.PyStoreObj('TOPRecBunch')
        if not recBunch.isValid():
            return
        self.nevnev += 1
        if recBunch.isReconstructed():
            self.recBunchNorecBunchNo.Fill(recBunch.getBunchNo())
            self.offsetoffset.Fill(recBunch.getCurrentOffset())
            self.numTracksnumTracks.Fill(recBunch.getNumTracks())
            self.usedTracksusedTracks.Fill(recBunch.getUsedTracks())
            if recBunch.getBunchNo() == recBunch.getMCBunchNo():
                self.effi_numTrackseffi_numTracks.Fill(recBunch.getNumTracks())
                self.effi_usedTrackseffi_usedTracks.Fill(recBunch.getUsedTracks())
 
    def terminate(self):
        ''' Processes and writes histograms to file'''
 
        eff = self.numTracksnumTracks.GetEntries() / self.nevnev
        err = math.sqrt(eff * (1 - eff) / self.nevnev)
        self.effieffi.SetBinContent(1, eff)
        self.effieffi.SetBinError(1, err)
 
        eff = self.effi_numTrackseffi_numTracks.GetEntries() / self.numTracksnumTracks.GetEntries()
        err = math.sqrt(eff * (1 - eff) / self.numTracksnumTracks.GetEntries())
        self.effieffi.SetBinContent(2, eff)
        self.effieffi.SetBinError(2, err)
 
        self.numTracksnumTracks.Sumw2()
        self.effi_numTrackseffi_numTracks.Sumw2()
        self.effi_numTrackseffi_numTracks.Divide(self.effi_numTrackseffi_numTracks, self.numTracksnumTracks, 1, 1, "B")
 
        self.usedTracksusedTracks.Sumw2()
        self.effi_usedTrackseffi_usedTracks.Sumw2()
        self.effi_usedTrackseffi_usedTracks.Divide(self.effi_usedTrackseffi_usedTracks, self.usedTracksusedTracks, 1, 1, "B")
 
        self.tfiletfile.Write()
        self.tfiletfile.Close()
 
 
# Create path
main = create_path()
 
# Input
main.add_module('RootInput', inputFileNames=['../EvtGenSimRec_B2Kpi.root', '../EvtGenSimRec.root'])
 
# Make plots
main.add_module(MakePlots())
 
main.add_module('Progress')
# Process events
process(main)
 
# Print call statistics
print(statistics)