Inheritance diagram for TrackingAnalysisConfig.InDetTrackWorkingPointConfig:

Collaboration diagram for TrackingAnalysisConfig.InDetTrackWorkingPointConfig:

Public Member Functions
	__init__ (self)
	instanceName (self)
	makeAlgs (self, config)

Public Attributes
	additionalCuts
	runTruthFilter
str	cutLevel = "Loose":
	fFakeTight
	calibFile
	fFakeLoose
	filterToolSeed
	trkEffSystScale
	containerName
	selectionName

Detailed Description

the ConfigBlock for the track working point

Definition at line 220 of file TrackingAnalysisConfig.py.

Constructor & Destructor Documentation

◆ init()

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.__init__ ( self )

Definition at line 223 of file TrackingAnalysisConfig.py.

    def __init__ (self) :
        super (InDetTrackWorkingPointConfig, self).__init__ ()
        self.addOption ('containerName', '', type=str,
            noneAction='error',
            info="the name of the input container.")
        self.addOption ('selectionName', '', type=str,
            noneAction='error',
            info="the name of the track selection to define (e.g. tightPrimary "
            "or loose).")
        self.addOption ('postfix', None, type=str,
            info="a postfix to apply to decorations and algorithm names. "
            "Typically not needed here as selectionName is used internally.")
        self.addOption ('cutLevel', None, type=str,
            noneAction='error',
            info="the selection WP (str) to use. Supported WPs for general "
            "use: `Loose` and `TightPrimary`. For expert studies, further "
            "WPs are available: `NoCut`, `LoosePrimary`, `LooseElectron`, "
            "`LooseMuon`, `LooseTau`, `MinBias`, `HILoose`, `HITight`, "
            "`HILooseOptimized`, `HITightOptimized`.",
            expertMode=["NoCut", "LoosePrimary", "LooseElectron",
            "LooseMuon", "LooseTau", "MinBias", "HILoose", "HITight",
            "HILooseOptimized", "HITightOptimized"])
        self.addOption ('additionalCuts', None, type=None,
            info="additional cuts to modify the selection WP. Only meant for "
            "expert studies of track selection. Passed as pairs of `cutName: value`. "
            "For an overview of available cuts, see twiki.cern.ch/twiki/bin/viewauth/"
            "AtlasProtected/InDetTrackSelectionTool#List_of_possible_cuts.",
            expertMode=True)
        self.addOption ('runTruthFilter', True, type=bool,
            info="whether to run the TruthFilterTool. This tool is only compatible "
            "with the cut levels 'Loose' and 'TightPrimary'.")
        self.addOption ('calibFile', None, type=str,
            info="name (str) of the calibration file to use for efficiencies "
            "in the TruthFilter tool. Expert option to override the "
            "recommendations based on the campaign. The default is None.",
            expertMode=True)
        self.addOption ('filterToolSeed', None, type=int,
            info="random seed (int) to be used by the InDetTrackTruthFilterTool. "
            "Expert option. The default is 0.",
            expertMode=True)
        self.addOption ('fFakeLoose', None, type=float,
            info="the fraction of fake tracks (float) in the Loose working point. "
            "Will be used by the InDetTrackTruthFilterTool. Expert option to "
            "override the recommendations.",
            expertMode=True)
        self.addOption ('fFakeTight', None, type=float,
            info="the fraction of fake tracks (float) in the TightPrimary working "
            "point. Will be used by the InDetTrackTruthFilterTool. Expert option "
            "to override the recommendations.",
            expertMode=True)
        self.addOption ('trkEffSystScale', None, type=float,
            info="the track efficiency systematic scale (float). Will be used "
            "by the InDetTrackTruthFilterTool. Expert option to override the "
            "recommendations. Default is 1.0",
            expertMode=True)
        self.addOption ('addSelectionToPreselection', True, type=bool,
            info="whether to retain only tracks satisfying the cutLevel "
            "requirements. The default is True.")
 

Member Function Documentation

◆ instanceName()

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.instanceName ( self )

Return the instance name for this block

Definition at line 282 of file TrackingAnalysisConfig.py.

    def instanceName (self) :
        """Return the instance name for this block"""
        if self.postfix is not None:
            return self.containerName + self.selectionName + self.postfix
        else:
            return self.containerName + self.selectionName
 

◆ makeAlgs()

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.makeAlgs	(		self,
			config )

Definition at line 289 of file TrackingAnalysisConfig.py.

    def makeAlgs (self, config) :
        log = logging.getLogger('InDetTrackWorkingPointConfig')
 
        selectionPostfix = self.selectionName
        if selectionPostfix != '' and selectionPostfix[0] != '_' :
            selectionPostfix = '_' + selectionPostfix
 
        postfix = self.postfix
        if postfix is None :
            postfix = self.selectionName
        if postfix != '' and postfix[0] != '_' :
            postfix = '_' + postfix
 
        cutLevels = ["NoCut", "Loose", "LoosePrimary", "TightPrimary", "LooseMuon",
                     "LooseElectron", "LooseTau", "MinBias", "HILoose", "HITight",
                     "HILooseOptimized", "HITightOptimized"]
        alg = config.createAlgorithm( 'CP::InDetTrackSelectionAlg', 'InDetTrackSelectionAlg' )
        alg.selectionDecoration = 'selectTrack' + postfix + '_%SYS%,as_bits'
        config.addPrivateTool( 'selectionTool', 'InDet::InDetTrackSelectionTool')
        if self.cutLevel is None:
            log.warning("No selection WP chosen, not setting up InDetTrackSelectionTool.")
        elif self.cutLevel not in cutLevels:
            raise ValueError ('Invalid cut level: \"' + self.cutLevel + '\", has '
                              'to be one of: ' + ', '.join(cutLevels))
        elif self.cutLevel in ["Loose", "TightPrimary"]:
            alg.selectionTool.CutLevel = self.cutLevel
        else:
            log.warning('Using cut level: \"' + self.cutLevel + '\" that is not '
                        'meant for general use, but only expert studies.')
            alg.selectionTool.CutLevel = self.cutLevel
        if self.additionalCuts:
            for cutName, value in self.additionalCuts.items():
                setattr(alg.selectionTool, cutName, value)
        # Set up the truth filtering algorithm:
        if config.dataType() is not DataType.Data:
            if not self.runTruthFilter:
                log.warning('Disabling the TruthFilterTool.')
            else:
                config.addPrivateTool( 'filterTool', 'InDet::InDetTrackTruthFilterTool' )
                config.addPrivateTool( 'filterTool.trackOriginTool', 'InDet::InDetTrackTruthOriginTool' )
                # Set working point based on cut level
                if self.cutLevel == "Loose":
                    alg.filterWP = "LOOSE"
                elif self.cutLevel == "TightPrimary":
                    alg.filterWP = "TIGHT"
                else:
                    raise ValueError ('Attempting to set TruthFilter WP based on cut level: \"'
                                      + self.efficiencyWP + '\" that is not supported.')
                # Set calibFile and fake rates based on campaign
                if config.geometry() is LHCPeriod.Run2:
                    # Run 2 recommendations (MC20)
                    alg.filterTool.calibFileNomEff = "InDetTrackSystematicsTools/CalibData_22.0_2022-v00/TrackingRecommendations_prelim_rel22.root"
                    alg.filterTool.fFakeLoose = 0.10
                    alg.filterTool.fFakeTight = 1.00
                elif config.geometry() is LHCPeriod.Run3:
                    if config.campaign() in [Campaign.MC23a, Campaign.MC23d, Campaign.MC23e]:
                        # 2022/23/24 recommendations (MC23a/d/e)
                        alg.filterTool.calibFileNomEff = "InDetTrackSystematicsTools/CalibData_22.0_2022-v00/TrackingRecommendations_prelim_rel22.root"
                        alg.filterTool.fFakeLoose = 0.40
                        alg.filterTool.fFakeTight = 1.00
                    elif not (self.calibFile and self.fFakeLoose and self.fFakeTight):
                        raise ValueError ('No efficiency recommendations found for campaign \"'
                                          + config.campaign().value + '\" in Run 3. '
                                          'Please check that the recommendations exist.')
                elif not (self.calibFile and self.fFakeLoose and self.fFakeTight):
                    raise ValueError ('No efficiency recommendations found for geometry \"'
                                      + config.geometry().value + '\". Please check '
                                      'the configuration.')
                # Set custom calibFile, fake rates, or random seed
                if self.calibFile:
                    alg.filterTool.calibFileNomEff = self.calibFile
                if self.fFakeLoose:
                    alg.filterTool.fFakeLoose = self.fFakeLoose
                if self.fFakeTight:
                    alg.filterTool.fFakeTight = self.fFakeTight
                if self.filterToolSeed:
                    alg.filterTool.Seed = self.filterToolSeed
                if self.trkEffSystScale:
                    alg.filterTool.trkEffSystScale = self.trkEffSystScale
        alg.inDetTracks = config.readName (self.containerName)
        alg.preselection = config.getPreselection (self.containerName, '')
        config.addSelection (self.containerName, self.selectionName, alg.selectionDecoration,
                             preselection=self.addSelectionToPreselection)

Member Data Documentation

◆ additionalCuts

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.additionalCuts

Definition at line 319 of file TrackingAnalysisConfig.py.

◆ calibFile

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.calibFile

Definition at line 358 of file TrackingAnalysisConfig.py.

◆ containerName

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.containerName

Definition at line 370 of file TrackingAnalysisConfig.py.

◆ cutLevel

str TrackingAnalysisConfig.InDetTrackWorkingPointConfig.cutLevel = "Loose":

Definition at line 330 of file TrackingAnalysisConfig.py.

◆ fFakeLoose

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.fFakeLoose

Definition at line 360 of file TrackingAnalysisConfig.py.

◆ fFakeTight

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.fFakeTight

Definition at line 349 of file TrackingAnalysisConfig.py.

◆ filterToolSeed

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.filterToolSeed

Definition at line 364 of file TrackingAnalysisConfig.py.

◆ runTruthFilter

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.runTruthFilter

Definition at line 324 of file TrackingAnalysisConfig.py.

◆ selectionName

TrackingAnalysisConfig.InDetTrackWorkingPointConfig.selectionName