VXDTF2TrackingValidation.py

#!/usr/bin/env python3
 
 
 
"""
<header>
  <contact>software-tracking@belle2.org</contact>
  <input>EvtGenSimNoBkg.root</input>
  <output>VXDTF2TrackingValidation.root</output>
  <description>
  This module validates that the VXDTF2 SVD-only track finding is capable of reconstructing tracks in Y(4S) runs.
  </description>
</header>
"""
 
import tracking
from tracking.validation.run import TrackingValidationRun
import logging
import basf2
import os
 
VALIDATION_OUTPUT_FILE = 'VXDTF2TrackingValidation.root'
N_EVENTS = 1000
ACTIVE = False
 
 
def setupFinderModule(path):
    tracking.add_hit_preparation_modules(path, components=["SVD"])
    tracking.add_vxd_track_finding_vxdtf2(path, components=["SVD"])
 
 
class VXDTF2TrackingValidation(TrackingValidationRun):
    """
    Validation class for the four 4-SVD Layer tracking
    """
    
    n_events = N_EVENTS
    
    generator_module = 'generic'
    
    root_input_file = '../EvtGenSimNoBkg.root'
    
    components = None
 
    
    finder_module = staticmethod(setupFinderModule)
 
    
    tracking_coverage = {
        'WhichParticles': ['SVD'],  # Include all particles seen in the SVD detector, also secondaries
        'UsePXDHits': False,
        'UseSVDHits': True,
        'UseCDCHits': False,
    }
 
    
    fit_tracks = True
    
    pulls = True
    
    use_expert_folder = False
    
    output_file_name = VALIDATION_OUTPUT_FILE
 
    # ugly way to set a local SectorMap if corresponding environment var is set
    # TODO: Find more elegant way!
    def adjust_path(self, path):
        # if environment variable is set, use the SectorMap this variable is pointing to!
        testing_secmap_name = os.getenv("BELLE2_TESTING_VXDTF2_SECMAP")
        if testing_secmap_name is not None:
            basf2.B2WARNING("Using non-default (the one in the db) SectorMap: " + testing_secmap_name)
            basf2.set_module_parameters(path, "SectorMapBootstrap", ReadSecMapFromDB=False)
            basf2.set_module_parameters(path, "SectorMapBootstrap", ReadSectorMap=True)
            basf2.set_module_parameters(path, "SectorMapBootstrap", SectorMapsInputFile=testing_secmap_name)
 
 
def main():
    """
    create SVD validation class and execute
    """
    basf2.set_random_seed(1337)
    validation_run = VXDTF2TrackingValidation()
    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.")