Class_InformationHandler.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
import string
 
class InformationHandler:
    
    def __init__(self, flags):
        
        #setup the member variables
        # they are filled in the helper functions setupInfo_*
        self.m_Infos_Int        = {}
        self.m_Infos_Double     = {}
        self.m_Infos_String     = {}
        self.m_Infos_VecDouble  = {}
        self.m_Infos_VecString  = {}
        
        #fill member variables
        self.setupInfo_Int(flags)
        self.setupInfo_Double(flags)
        self.setupInfo_String(flags)
        self.setupInfo_VecDouble(flags)
        self.setupInfo_VecString(flags)
        
        #fill the feature types and default values
        # -> a bit more involved, have extra function for this
        self.setupInfo_Features(flags)
    
    
    def dump(self):
        print("PanTau::InformationHandler: Printing stored information")
        
        print("PanTau::InformationHandler: \tIntegers:")
        for iInfo in sorted(self.m_Infos_Int.iterkeys()):
            print("\t\t" + string.ljust(iInfo, 50) + ": " + string.rjust(str(self.m_Infos_Int[iInfo]), 10) )
 
        print("PanTau::InformationHandler: \tDoubles:")
        for iInfo in sorted(self.m_Infos_Double.iterkeys()):
            print("\t\t" + string.ljust(iInfo, 50) + ": " + string.rjust(str(self.m_Infos_Double[iInfo]), 10) )
 
        print("PanTau::InformationHandler: \tStrings:")
        for iInfo in sorted(self.m_Infos_String.iterkeys()):
            print("\t\t" + string.ljust(iInfo, 50) + ": " + string.ljust(self.m_Infos_String[iInfo], 80) )
 
        print("PanTau::InformationHandler: \tVecDoubles:")
        for iInfo in sorted(self.m_Infos_VecDouble.iterkeys()):
            print("InformationHandler: \t\t" + string.ljust(iInfo, 50) + ":")
            for iVal in self.m_Infos_VecDouble[iInfo]:
                print("InformationHandler: \t\t\t" + string.rjust(str(iVal), 10) )
 
        print("PanTau::InformationHandler: \tVecStrings:")
        for iInfo in sorted(self.m_Infos_VecString.iterkeys()):
            print("InformationHandler: \t\t" + string.ljust(iInfo, 50) + ":")
            for iVal in self.m_Infos_VecString[iInfo]:
                print("InformationHandler: \t\t\t" + string.ljust(str(iVal), 80) )
 
        print("PanTau::InformationHandler: Done printing stored information")
 
    
    
    def setupInfo_Int(self, flags):
        self.m_Infos_Int["TauConstituents_UsePionMass"]     = int(flags.Tau.PanTau.TauConstituents_UsePionMass)
        self.m_Infos_Int["FeatureExtractor_UseEmptySeeds"]  = int(flags.Tau.PanTau.FeatureExtractor_UseEmptySeeds)
        #nothing to do here yet (no integer infos...)
 
 
    def setupInfo_Double(self, flags):
        self.m_Infos_Double["TauConstituents_Types_DeltaRCore"]                 = flags.Tau.PanTau.TauConstituents_Types_DeltaRCore
        self.m_Infos_Double["TauConstituents_MaxEta"]                           = flags.Tau.PanTau.TauConstituents_MaxEta
        self.m_Infos_Double["TauConstituents_PreselectionMinEnergy"]            = flags.Tau.PanTau.TauConstituents_PreselectionMinEnergy
        
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R10X_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R10X_CellBased
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R11X_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R11X_CellBased
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R110_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R110_CellBased
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R1XX_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R1XX_CellBased
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R30X_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R30X_CellBased
        self.m_Infos_Double["DecayModeDeterminator_BDTCutValue_R3XX_CellBased"] = flags.Tau.PanTau.DecayModeDeterminator_BDTCutValue_R3XX_CellBased
 
    def setupInfo_String(self, flags):
        self.m_Infos_String["ModeDiscriminator_TMVAMethod"]     = flags.Tau.PanTau.ModeDiscriminator_TMVAMethod
 
 
    def setupInfo_VecDouble(self, flags):
        self.m_Infos_VecDouble["TauConstituents_BinEdges_Eta"]                       = flags.Tau.PanTau.TauConstituents_BinEdges_Eta
        self.m_Infos_VecDouble["TauConstituents_Selection_Neutral_EtaBinned_EtCut"]  = flags.Tau.PanTau.TauConstituents_Selection_Neutral_EtaBinned_EtCut
        self.m_Infos_VecDouble["CellBased_BinEdges_Eta"]                             = flags.Tau.PanTau.CellBased_BinEdges_Eta
        self.m_Infos_VecDouble["CellBased_EtaBinned_Pi0MVACut_1prong"]               = flags.Tau.PanTau.CellBased_EtaBinned_Pi0MVACut_1prong
        self.m_Infos_VecDouble["CellBased_EtaBinned_Pi0MVACut_3prong"]               = flags.Tau.PanTau.CellBased_EtaBinned_Pi0MVACut_3prong
        self.m_Infos_VecDouble["ModeDiscriminator_BinEdges_Pt"]                      = flags.Tau.PanTau.ModeDiscriminator_BinEdges_Pt
 
 
    def setupInfo_VecString(self, flags):
        self.m_Infos_VecString["ModeDiscriminator_BDTVariableNames_CellBased_1p0n_vs_1p1n"] = flags.Tau.PanTau.ModeDiscriminator_BDTVariableNames_CellBased_1p0n_vs_1p1n
        self.m_Infos_VecString["ModeDiscriminator_BDTVariableNames_CellBased_1p1n_vs_1pXn"] = flags.Tau.PanTau.ModeDiscriminator_BDTVariableNames_CellBased_1p1n_vs_1pXn
        self.m_Infos_VecString["ModeDiscriminator_BDTVariableNames_CellBased_3p0n_vs_3pXn"] = flags.Tau.PanTau.ModeDiscriminator_BDTVariableNames_CellBased_3p0n_vs_3pXn
 
 
    def setupInfo_Features(self, flags):
        
        #get list of signal modes from config
        List_ModeCases      = flags.Tau.PanTau.Names_ModeCases
        
        #get the feature handler
        from PanTauAlgs.Class_FeatureHandler import FeatureHandler
        theFeatureHandler = FeatureHandler()
        
        #for each algorithm and mode, loop over variables to be used in BDT
        # and fetch the type of the variable and the name
        # these are needed for the PanTauFillerTool in TauD3PDMaker
        for iModeCase in List_ModeCases:
                
            infoKey_Names = "ModeDiscriminator_BDTVariableNames_CellBased_" + iModeCase
            curVarList = self.m_Infos_VecString[infoKey_Names]
                
            BDTVariable_TypeList        = []
            BDTVariable_DefaultValList  = []
                
            for iVar in curVarList:
                theType     = theFeatureHandler.m_Feature_Types[iVar]
                theDefVal   = theFeatureHandler.m_Feature_Defaults[iVar]
                BDTVariable_TypeList        += [theType]
                BDTVariable_DefaultValList  += [theDefVal]
            #end loop over variables
                
            infoKey_Types       = "ModeDiscriminator_BDTVariableTypes_CellBased_" + iModeCase
            self.m_Infos_VecString[infoKey_Types] = BDTVariable_TypeList
                
            infoKey_Defaults    = "ModeDiscriminator_BDTVariableDefaults_CellBased_" + iModeCase
            self.m_Infos_VecDouble[infoKey_Defaults] = BDTVariable_DefaultValList    
        #end loop over signal modes
        
        #add the prefixes for the variables
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Sum"]         = theFeatureHandler.m_VarTypeName_Sum
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Ratio"]       = theFeatureHandler.m_VarTypeName_Ratio
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Isolation"]   = theFeatureHandler.m_VarTypeName_Isolation
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Num"]         = theFeatureHandler.m_VarTypeName_Num
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Mean"]        = theFeatureHandler.m_VarTypeName_Mean
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_StdDev"]      = theFeatureHandler.m_VarTypeName_StdDev
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_HLV"]         = theFeatureHandler.m_VarTypeName_HLV
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Angle"]       = theFeatureHandler.m_VarTypeName_Angle
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_DeltaR"]      = theFeatureHandler.m_VarTypeName_DeltaR
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_JetMoment"]   = theFeatureHandler.m_VarTypeName_JetMoment
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Combined"]    = theFeatureHandler.m_VarTypeName_Combined
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Basic"]       = theFeatureHandler.m_VarTypeName_Basic
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_PID"]         = theFeatureHandler.m_VarTypeName_PID
        self.m_Infos_String["FeatureExtractor_VarTypeName_varTypeName_Shots"]       = theFeatureHandler.m_VarTypeName_Shots
    #end def setupInfo_Features
    
#end class config_FeatureCalculator