trgcdct3dDQM.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
# -----------------------------------------------------------------------------------
#
#                        CDCTRGT3D Unpacker and Converter
#
#    usage : %> basf2 trgcdct3dUnpackerConverter -i [input sroot file name] -o [output root file name]
#
# -----------------------------------------------------------------------------------
 
import basf2
import ROOT.Belle2
 
import sys  # get argv
argvs = sys.argv  # get arg
argc = len(argvs)  # of arg
 
if argc == 3:
    inname = argvs[1]
    histname = argvs[2]
    # outname=argvs[2]
    # histname=argvs[3]
else:
    sys.exit("trgcdct3dDQM.py> # of arg is strange. Exit.")
 
 
# class is2DSkim(basf2.Module):
#    def event(self):
#        Tracks = ROOT.Belle2.PyStoreArray("Tracks")
#        TRGGRLT3DUnpackerStores = ROOT.Belle2.PyStoreArray("TRGCDCT3DUnpackerStores")
#
#        isTwoTrack = False
#        if len(Tracks) == 2:
#            isTwoTrack = True
#
#        areTracksClean = True
#        for track in Tracks:
#            trackResult = track.getTrackFitResultWithClosestMass(ROOT.Belle2.Const.pion)
#            if trackResult.getPValue() > 0.1:
#                areTracksClean = False
#
#        enoughHits = True
#        for track in Tracks:
#            trackResult = track.getTrackFitResultWithClosestMass(ROOT.Belle2.Const.pion)
#            nHitStSl = 0
#            nHitAxSl = 0
#            for iAx in range(5):
#                if (trackResult.getHitPatternCDC().getSLayerNHits(2 * iAx) > 3):
#                    nHitAxSl += 1
#            for iSt in range(4):
#                if (trackResult.getHitPatternCDC().getSLayerNHits(2 * iSt + 1) > 3):
#                    nHitStSl += 1
#            if nHitStSl < 3:
#                enoughHits = False
#            if nHitAxSl < 4:
#                enoughHits = False
#
#        is2D = False
#        for iClk, data in enumerate(TRGGRLT3DUnpackerStores):
#            if (data.m_t2d_fnf != 0):
#                is2D = True
#
#        self.return_value(isTwoTrack and is2D and areTracksClean and enoughHits)
 
 
if __name__ == '__main__':
 
    basf2.use_central_database("development")
#    Hot fix for CDCFudgeFactorFromSigma
    basf2.use_local_database(ROOT.Belle2.FileSystem.findFile("data/trg/cdc/db_CDCFudgeFactorsForSigma.txt"), "localdb")
 
    empty_path = basf2.create_path()
#   skim = basf2.register_module(is2DSkim())
#   skim.if_value('=0', empty_path, basf2.AfterConditionPath.END)
 
    # Create main path
    main = basf2.create_path()
    # Add modules to main path
    main.add_module('RootInput', inputFileName=inname)
#   main.add_module('SeqRootInput', inputFileName=inname)
 
    nmod = [0, 1, 2, 3]
    for mod in nmod:
        main.add_module('TRGCDCT3DUnpacker', T3DMOD=mod)
 
    main.add_module('Gearbox')
 
    for mod in nmod:
        # main.add_module(skim)
        main.add_module('TRGCDCT3DConverter',
                        hitCollectionName='FirmCDCTriggerSegmentHits' + str(mod),
                        addTSToDatastore=True,
                        EventTimeName='FirmBinnedEventT0' + str(mod),
                        addEventTimeToDatastore=True,
                        inputCollectionName='FirmTRGCDC2DFinderTracks' + str(mod),
                        add2DFinderToDatastore=True,
                        outputCollectionName='FirmTRGCDC3DFitterTracks' + str(mod),
                        add3DToDatastore=True,
                        fit3DWithTSIM=0,
                        firmwareResultCollectionName='TRGCDCT3DUnpackerStore' + str(mod),
                        isVerbose=0)
 
    for mod in nmod:
        main.add_module('TRGCDCT3DDQM', postScriptName=histname, T3DMOD=mod)
 
    main.add_module('HistoManager', histoFileName=histname)
 
    # Process all events
    basf2.process(main)
    print(basf2.statistics)