cosmicsTrackingValidation.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
"""
<header>
  <contact>software-tracking@belle2.org</contact>
  <input>CosmicsSimNoBkgTrackingVolume.root</input>
  <output>CosmicsTrackingValidation.root</output>
  <description>Validation of the default cosmic track finding (VXD+CDC).</description>
</header>
"""
 
VALIDATION_OUTPUT_FILE = 'CosmicsTrackingValidation.root'
CONTACT = 'software-tracking@belle2.org'
N_EVENTS = 10000
ACTIVE = True
 
import basf2
basf2.set_random_seed(1337)
 
import logging
 
 
from tracking import add_cr_tracking_reconstruction
from tracking.validation.run import TrackingValidationRun
 
 
class Cosmics(TrackingValidationRun):
    """
    derived class that sets up the cosmics validation
    """
    
    n_events = N_EVENTS
    
    generator_module = 'Cosmics'
    
    root_input_file = '../CosmicsSimNoBkgTrackingVolume.root'
    
    components = None
    
    finder_module = staticmethod(add_cr_tracking_reconstruction)
    
    tracking_coverage = {
        'WhichParticles': ['CDC'],  # Include all particles seen in CDC, also secondaries (dont put SVD, PXD here)
        'UsePXDHits': True,
        'UseSVDHits': True,
        'UseCDCHits': True,
        'UseOnlyAxialCDCHits': False,
        "UseReassignedHits": True,
        'UseNLoops': 1
    }
    
    pulls = True
    
    contact = CONTACT
    
    output_file_name = VALIDATION_OUTPUT_FILE
 
 
def main():
    validation_run = Cosmics()
    validation_run.configure_and_execute_from_commandline()
 
 
if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO)
    if ACTIVE:
        main()