studyLaserLight.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
# ------------------------------------------------------------------------
# Module to produce the laser time histograms that serve as input to the laser
# resolution fitter
#
# usage: basf2 studyLaserResolution.py dbaddress (path|none) type (local|pocket|root)
#                                      output_name.root path_to_sroot run1 run2 ... runN
#        The run number accepts wildcards
# ------------------------------------------------------------------------
 
 
import basf2 as b2
import sys
import glob
from ROOT import Belle2
from ROOT import TFile, TH2F
from laserResolutionTools import fitLaserResolution, plotLaserResolution
 
 
class TOPLaserHistogrammerModule(b2.Module):
 
    ''' Module to study resolution and performance of the top laser calibration.'''
 
    
    h_LaserTimingVSChannel = TH2F(
        'LaserTimingVSChannel',
        'Laser timing in as function of the channel number',
        512 * 16,
        0,
        512 * 16,
        10000,
        0.,
        100)  # 10 ps binning
    
    h_LaserTimingVSChannelOneSlot = TH2F(
        'LaserTimingVSChannelOneSlot',
        'Laser timing in as function of the channel number',
        512,
        0,
        512,
        10000,
        0.,
        100)  # 10 ps binning
 
    
    h_crossOccupancy = [[TH2F(
        'crossOccupancy_' + str(slotA) + '_' + str(slotB),
        ' ',
        200,
        0,
        200,
        200,
        0.,
        200) for slotA in range(16)] for slotB in range(16)]
 
    
    outname = 'outStudyLaserResolution.root'
 
    
    m_ignoreNotCalibrated = True
    
    m_runOnData = True
 
    
    m_maxWidth = 3.
    
    m_minWidth = 0.5
    
    m_maxAmp = 700.
    
    m_minAmp = 250.
    
    m_mcCorrectionsFile = '/group/belle2/group/detector/TOP/calibration/MCreferences/t0MC.root'
    
    m_MCPeaks = [[]]
 
    def setOutputName(self, outputname):
        ''' Sets the output file name '''
        
        self.outnameoutnameoutname = outputname
 
    def setMaxWidth(self, maxWidth):
        ''' Sets the maximum calpulse width '''
        
        self.m_maxWidthm_maxWidthm_maxWidth = maxWidth
 
    def setMinWidth(self, minWidth):
        ''' Sets the minimum calpulse width '''
        
        self.m_minWidthm_minWidthm_minWidth = minWidth
 
    def setMaxAmp(self, maxAmp):
        ''' Sets the maximum calpulse amplitude '''
        
        self.m_maxAmpm_maxAmpm_maxAmp = maxAmp
 
    def setMinAmp(self, minAmp):
        ''' Sets the minimum calpulse amplitude '''
        
        self.m_minAmpm_minAmpm_minAmp = minAmp
 
    def setMCCorrectionsFile(self, MCfile):
        ''' Sets the file containing the MC correction'''
        
        self.m_mcCorrectionsFilem_mcCorrectionsFilem_mcCorrectionsFile = MCfile
 
    def ignoreNotCalibrated(self, ignoreNotCal):
        ''' Sets the flag to ingore the hits without calibration '''
        
        self.m_ignoreNotCalibratedm_ignoreNotCalibratedm_ignoreNotCalibrated = ignoreNotCal
 
    def event(self):
        ''' Event processor: fill histograms '''
 
        digits = Belle2.PyStoreArray('TOPDigits')
        nhits = [0 for i in range(16)]
        for digit in digits:
            if(not self.ignoreNotCalibratedignoreNotCalibrated and not digit.isTimeBaseCalibrated()):
                continue
            if (digit.getHitQuality() == 1 and
                digit.getPulseWidth() > self.m_minWidthm_minWidthm_minWidth and digit.getPulseWidth() < self.m_maxWidthm_maxWidthm_maxWidth and
                    digit.getPulseHeight() > self.m_minAmpm_minAmpm_minAmp and digit.getPulseHeight() < self.m_maxAmpm_maxAmpm_maxAmp):
                slotID = digit.getModuleID()
                hwchan = digit.getChannel()
                self.h_LaserTimingVSChannelh_LaserTimingVSChannel.Fill(512 * (slotID - 1) + hwchan, digit.getTime())
                simhits = digit.getRelationsWith('TOPSimHits')
                nhits[slotID - 1] = nhits[slotID - 1] + 1
                for simhit in simhits:
                    self.h_LaserTimingVSChannelOneSloth_LaserTimingVSChannelOneSlot.Fill(hwchan, simhit.getTime())
        for slotA in range(16):
            for slotB in range(16):
                self.h_crossOccupancyh_crossOccupancy[slotA][slotB].Fill(nhits[slotA], nhits[slotB])
 
    def terminate(self):
        ''' Write histograms to file, fills and fits the resolution plots'''
        tfile = TFile(self.outnameoutnameoutname, 'recreate')
        self.h_LaserTimingVSChannelh_LaserTimingVSChannel.Write()
        self.h_LaserTimingVSChannelOneSloth_LaserTimingVSChannelOneSlot.Write()
        for slotA in range(16):
            for slotB in range(16):
                self.h_crossOccupancyh_crossOccupancy[slotA][slotB].Write()
        tfile.Close()
 
 
argvs = sys.argv
print('usage: basf2', argvs[0], 'runNumber outfileName')
 
lookBack = argvs[1]         # lookbackWindows setting
dbaddress = argvs[2]        # path to the calibration DB  (absolute path  or 'none')
datatype = argvs[3]         # data type  ('pocket' or 'local' or 'root', if root files have to be analyzed)
outfile = argvs[4]          # output name
folder = argvs[5]           # folder containing the sroot files
runnumbers = sys.argv[6:]   # run numbers
 
files = []
for runnumber in runnumbers:
    if datatype == 'root':
        files += [f for f in glob.glob(str(folder) + '/*' + str(runnumber) + '*.root')]
    else:
        #        print('running on sroot files in folder ' + glob.glob(str(folder))
        files += [f for f in glob.glob(str(folder) + '/*' + str(runnumber) + '*.sroot')]
for fname in files:
    print("file: " + fname)
 
if dbaddress != 'none':
    print("using local DB " + dbaddress)
    b2.reset_database()
    b2.use_local_database(dbaddress + "/localDB.txt", dbaddress)
else:
    print("database not set. Continuing without calibrations")
 
# Suppress messages and warnings during processing
b2.set_log_level(b2.LogLevel.ERROR)
 
# Define a global tag (note: the one given bellow can be out-dated!)
b2.use_central_database('data_reprocessing_proc8')
 
# Create path
main = b2.create_path()
 
# input
if datatype == 'root':
    roinput = b2.register_module('RootInput')
    roinput.param('inputFileNames', files)
    main.add_module(roinput)
else:
    roinput = b2.register_module('SeqRootInput')
    roinput.param('inputFileNames', files)
    main.add_module(roinput)
 
# conversion from RawCOPPER or RawDataBlock to RawDetector objects
if datatype == 'pocket':
    print('pocket DAQ data assumed')
    converter = b2.register_module('Convert2RawDet')
    main.add_module(converter)
 
# Initialize TOP geometry parameters (creation of Geant geometry is not needed)
main.add_module('TOPGeometryParInitializer')
 
if datatype != 'root':
    # Unpacking (format auto detection works now)
    unpack = b2.register_module('TOPUnpacker')
    main.add_module(unpack)
 
    # Add multiple hits by running feature extraction offline
    featureExtractor = b2.register_module('TOPWaveformFeatureExtractor')
    main.add_module(featureExtractor)
 
    # Convert to TOPDigits
    converter = b2.register_module('TOPRawDigitConverter')
    if dbaddress == 'none':
        print("Not using Calibrations")
        converter.param('useSampleTimeCalibration', False)
        converter.param('useAsicShiftCalibration', False)
        converter.param('useChannelT0Calibration', False)
        converter.param('useModuleT0Calibration', False)
    else:
        print("Using Calibrations")
        converter.param('useSampleTimeCalibration', True)
        converter.param('useAsicShiftCalibration', True)
        converter.param('useChannelT0Calibration', True)
        converter.param('useModuleT0Calibration', True)
    converter.param('useCommonT0Calibration', False)
    converter.param('calibrationChannel', -1)  # do not specify the calpulse channel
    converter.param('lookBackWindows', int(lookBack))  # in number of windows
    main.add_module(converter)
 
# resolution plotter
resolutionModule = TOPLaserHistogrammerModule()
resolutionModule.setOutputName(outfile)
resolutionModule.setMinWidth(0.1)  # good TOPDigit selection
resolutionModule.setMaxWidth(999)  # good TOPDigit selection
resolutionModule.setMinAmp(100)  # good TOPDigit selection
resolutionModule.setMaxAmp(999)  # good TOPDigit selection
if dbaddress == 'none':
    resolutionModule.ignoreNotCalibrated(True)
else:
    resolutionModule.ignoreNotCalibrated(False)
main.add_module(resolutionModule)
 
# Show progress of processing
progress = b2.register_module('Progress')
main.add_module(progress)
 
# Process events
b2.process(main)
 
# Print call statistics
print(b2.statistics)
print(b2.statistics(b2.statistics.TERM))
 
 
fitLaserResolution(dataFile=outfile, outputFile='laserResolutionResults.root', pdfType='cb', saveFits=True)
plotLaserResolution()