independent_merge_path_real.py

#!/usr/bin/env python3
 

 
import basf2
 
main = basf2.Path()
indep = basf2.Path()
 
# input (if you're doing signal embedding, this should be the ROE from data)
input1 = basf2.register_module('RootInput')
input1.param('inputFileName', basf2.find_file('embedding_jpsi_exp12_skim.root', 'examples'))
main.add_module(input1).set_name("--input_main--")
 
# and the other input (if you're doing signal embedding, this should be the signal from MC)
input2 = basf2.register_module('RootInput')
input2.param('inputFileName', basf2.find_file('embedding_kplusnunu_mcsig_skim.root', 'examples'))
input2.param('isSecondaryInput', True)  # <- set flag for secondary input module
indep.add_module(input2).set_name("input_indep")
 
# merge it!
# Use merge_back_event=['ALL'] to merge everything
# or specify explicitly merge_back_event=['EventMetaData', 'ECLClusters', 'Tracks', 'TracksToECLClusters']
# NOTE: StoreArrays have to be merged before their Relations (and 'EventMetaData' is strictly required!)
# NOTE: Also merge 'EventExtraInfo' if you want to do a consistency_check
main.add_independent_merge_path(
    indep,
    # <- skip event if charge of ROE/sig (or ROE/ROE) does not match [None or "" if you don't want to do this]
    consistency_check="charge",
    merge_back_event=['ALL'],  # <- list of content to be merged
    event_mixing=False,  # <- signal embedding or event mixing
    merge_same_file=False  # <- for event mixing, you can mix a file with itself
)
 
main.add_module('FixMergedObjects')
 
# output
output = basf2.register_module('RootOutput')
output.param('outputFileName', 'merged_10evts.root')
main.add_module(output)
 
# progress
main.add_module('Progress')
 
basf2.print_path(main)
basf2.process(main, calculateStatistics=True)
 
print(basf2.statistics)