AsicBackgroundLibraryGenerator.py

import basf2
import sys
import rawdata
import tracking
 
 
def main():
    """
       Sample script to generte ASIC cross-talk library.  Usage:
 
        basf2 AsicBackgroundLibraryGenerator.py <cosmic_raw_data_file> <output_asicbg_root_file>
    """
    global_tag = "data_reprocessing_prompt_rel4_patchb"
    basf2.conditions.override_globaltags()
    basf2.conditions.expert_settings(usable_globaltag_states={"TESTING", "VALIDATED",
                                                              "PUBLISHED", "RUNNING", "OPEN"})
    basf2.conditions.append_globaltag(global_tag)
    path = basf2.create_path()
    path.add_module("Progress")
 
    # Input raw data file
    inputFilename = sys.argv[1]
    # Output file
    file_name = sys.argv[2]
 
    branches = ['EventMetaData',  'RawCDCs']
    path.add_module("RootInput", inputFileNames=inputFilename, branchNames=branches)
    path.add_module("Gearbox")
    path.add_module("Geometry", useDB=True)
 
    unpackers = ['CDC']
    rawdata.add_unpackers(path, components=unpackers)
 
    tracking.add_track_finding(path, components=['CDC'])
 
    path.add_module("AsicBackgroundLibraryCreator",
                    inputTracks="CDCTrackVector",
                    inputWireHits="CDCWireHitVector",
                    AsicLibraryFileName=file_name,
                    #                  maximalDistanceSignal=100.2,
                    #                  minimalDistanceBackground=0.0,
                    #                  useAxialHitsOnly=False,
                    #                  minimalHitsOnTrack=0,
                    writeExtraVars=True
                    )
 
    basf2.print_path(path)
 
    # Run the execution path
    basf2.process(path)
 
    print(basf2.statistics)
 
if __name__ == "__main__":
    main()