noCKFValidationBkg.py

#!/usr/bin/env python3
 
 
 
"""
<header>
  <contact>Nils.Braun@kit.edu</contact>
  <input>EvtGenSim.root</input>
  <output>NoCKFValidationBkg.root</output>
  <description>This script validates the tracking chain without the CKF in Y(4S) runs with background.</description>
</header>
"""
 
from tracking.validation.run import TrackingValidationRun
from tracking import add_track_finding
from tracking.path_utils import add_hit_preparation_modules
import logging
import basf2
VALIDATION_OUTPUT_FILE = 'NoCKFValidationBkg.root'
N_EVENTS = 1000
ACTIVE = True
 
 
def setupFinderModule(path):
    add_hit_preparation_modules(path, components=["SVD", "PXD"])
    add_track_finding(path, svd_ckf_mode="SVD_alone")
 
 
class CKFBkg(TrackingValidationRun):
    """Validate the track-finding chain, excluding the Kalman fitter/filter, with Y(4S) events and background overlay"""
    
    n_events = N_EVENTS
    
    generator_module = 'generic'
    
    root_input_file = '../EvtGenSim.root'
    
    finder_module = staticmethod(setupFinderModule)
    
    tracking_coverage = {
        'WhichParticles': [],  # Include all particles, also secondaries
        'UsePXDHits': True,
        'UseSVDHits': True,
        'UseCDCHits': True,
        "UseReassignedHits": True,
        'UseOnlyBeforeTOP': True,
        'UseNLoops': 1
    }
    
    fit_tracks = True
    
    use_fit_information = True
    
    pulls = True
    
    resolution = True
    
    output_file_name = VALIDATION_OUTPUT_FILE
 
 
def main():
    basf2.set_random_seed(1337)
    validation_run = CKFBkg()
    validation_run.configure_and_execute_from_commandline()
 
 
if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO)
    if ACTIVE:
        main()
    else:
        print("This validation deactivated and thus basf2 is not executed.\n"
              "If you want to run this validation, please set the 'ACTIVE' flag above to 'True'.\n"
              "Exiting.")