produceSample_releaseValidation.py

#!/usr/bin/env python3
 
from ROOT import Belle2
 
import basf2
import logging
 
import reconstruction as re
 
 
def main():
    """Only execute the code if the script is run but not when it's imported."""
    basf2.set_random_seed(1234)
 
    path = basf2.create_path()
 
    environment = Belle2.Environment.Instance()
    environment.setNumberEventsOverride(10000)
 
    # Simulate events first (e.g. sample produced by validation/validation/EvtGenSim.py)
    path.add_module('RootInput', inputFileName='./EvtGenSim.root')
 
    # Some default modules
    path.add_module('Progress')
 
    path.add_module("Gearbox")
    path.add_module("Geometry")
 
    # Do not prune the tracks (usually all but first and last hit are removed)
    re.add_reconstruction(path, pruneTracks=False, reconstruct_cdst="rawFormat")
 
    # Set a few options of modules that we need for the studies
    for module in path.modules():
        if 'TrackFinderMCTruthRecoTracks' in module.name():
            module.param({"UseReassignedHits": True, 'UseNLoops': 1})
        if 'V0Finder' in module.name():
            module.param("Validation", True)
 
    # Add MC information for V0s
    path.add_module('MCV0Matcher', V0ColName='V0ValidationVertexs')
 
    # Add a bunch of branches that we need (on top of cdst)
    re.add_cdst_output(
        path,
        mc=True,
        filename='./validationSample.root',
        additionalBranches=[
            'SVDClusters',
            'PXDClusters',
            'CDCSimHits',
            'CDCSimHitsToCDCHits',
            'MCRecoTracks',
            'RecoHitInformations',
            'RecoTracksToRecoHitInformations',
            'MCRecoTracksToRecoHitInformations',
            'MCRecoTracksToMCParticles',
            'MCRecoTracksToRecoTracks',
            'RecoTracksToMCParticles',
            'RecoTracksToMCRecoTracks',
            'V0ValidationVertexs',
            'V0ValidationVertexsToMCParticles',
            'V0sToV0ValidationVertexs',
            'TracksToRecoTracks',
            'TracksToMCParticles'])
 
    basf2.print_path(path)
    basf2.process(path)
 
    logging.basicConfig(level=logging.INFO)
 
    print(basf2.statistics)
 
 
if __name__ == "__main__":
    main()