ATLAS Offline Software
Loading...
Searching...
No Matches
python.DiTauAnalysisConfig.DiTauCalibrationConfig Class Reference
Inheritance diagram for python.DiTauAnalysisConfig.DiTauCalibrationConfig:
Collaboration diagram for python.DiTauAnalysisConfig.DiTauCalibrationConfig:

Public Member Functions

 __init__ (self)
 instanceName (self)
 makeAlgs (self, config)

Public Attributes

 inputContainer
 containerName
 decorateExtraVariables

Detailed Description

the ConfigBlock for the tau four-momentum correction

Definition at line 9 of file DiTauAnalysisConfig.py.

Constructor & Destructor Documentation

◆ __init__()

python.DiTauAnalysisConfig.DiTauCalibrationConfig.__init__ ( self)

Definition at line 12 of file DiTauAnalysisConfig.py.

12 def __init__ (self) :
13 super().__init__()
14 self.setBlockName('DiTaus')
15 self.addOption ('inputContainer', '', type=str,
16 info="the name of the input ditau-jet container.")
17 self.addOption ('containerName', '', type=str,
18 noneAction='error',
19 info="the name of the output container after calibration.",
20 meta={'role':'container'})
21 self.addOption ('postfix', '', type=str,
22 info="a postfix to apply to decorations and algorithm names. "
23 "Typically not needed here since the calibration is common to "
24 "all ditau-jets.")
25 self.addOption ('quality', None, type=str,
26 info="the ID WP to use. Supported ID WPs: `Tight`, `Medium`,`Loose`,`NoID`.",
27 meta={'choices':(['Tight','Medium','Loose','NoID'],1)})
28 self.addOption ('rerunTruthMatching', True, type=bool,
29 info="whether to rerun truth matching (sets up an instance of "
30 "`CP::DiTauTruthMatchingAlg`).")
31 self.addOption ('decorateTruth', False, type=bool,
32 info="decorate the truth particle information on the reconstructed one.")
33 self.addOption ('decorateExtraVariables', True, type=bool,
34 info="decorate extra variables for the reconstructed ditau-jet.")
35

Member Function Documentation

◆ instanceName()

python.DiTauAnalysisConfig.DiTauCalibrationConfig.instanceName ( self)
Return the instance name for this block

Definition at line 36 of file DiTauAnalysisConfig.py.

36 def instanceName (self) :
37 """Return the instance name for this block"""
38 return self.containerName + self.postfix
39

◆ makeAlgs()

python.DiTauAnalysisConfig.DiTauCalibrationConfig.makeAlgs ( self,
config )

Definition at line 40 of file DiTauAnalysisConfig.py.

40 def makeAlgs (self, config) :
41
42 postfix = self.postfix
43 if postfix != '' and postfix[0] != '_' :
44 postfix = '_' + postfix
45
46 inputContainer = "DiTauJets"
47 if self.inputContainer:
48 inputContainer = self.inputContainer
49 config.setSourceName (self.containerName, inputContainer)
50
51 # Set up the tau truth matching algorithm:
52 if self.rerunTruthMatching and config.dataType() is not DataType.Data:
53 alg = config.createAlgorithm( 'CP::DiTauTruthMatchingAlg',
54 'DiTauTruthMatchingAlg' )
55 config.addPrivateTool( 'matchingTool',
56 'TauAnalysisTools::DiTauTruthMatchingTool' )
57 alg.taus = config.readName (self.containerName)
58 alg.preselection = config.getPreselection (self.containerName, '')
59
60
61 # decorate truth tau information on the reconstructed object:
62 if self.decorateTruth and self.rerunTruthMatching and config.dataType() is not DataType.Data:
63 # in the case of the ditau, the DiTauTruthMatchingTool decorates directly the reco ditau with truth information.
64 # So information can be written directly out without any additional algorithm
65 config.addOutputVar (self.containerName, 'TruthVisLeadPt', 'TruthVisLeadPt', noSys=True)
66 config.addOutputVar (self.containerName, 'TruthVisLeadEta', 'TruthVisLeadEta', noSys=True)
67 config.addOutputVar (self.containerName, 'TruthVisLeadPhi', 'TruthVisLeadPhi', noSys=True)
68 config.addOutputVar (self.containerName, 'TruthVisLeadM', 'TruthVisLeadM', noSys=True)
69 config.addOutputVar (self.containerName, 'TruthLeadPdgID', 'TruthLeadPdgID', noSys=True)
70 config.addOutputVar (self.containerName, 'TruthVisSubleadPt', 'TruthVisSubleadPt', noSys=True)
71 config.addOutputVar (self.containerName, 'TruthVisSubleadEta', 'TruthVisSubleadEta', noSys=True)
72 config.addOutputVar (self.containerName, 'TruthVisSubleadPhi', 'TruthVisSubleadPhi', noSys=True)
73 config.addOutputVar (self.containerName, 'TruthVisSubleadM', 'TruthVisSubleadM', noSys=True)
74 config.addOutputVar (self.containerName, 'TruthSubleadPdgID', 'TruthSubleadPdgID', noSys=True)
75 config.addOutputVar (self.containerName, 'TruthVisDeltaR', 'TruthVisDeltaR', noSys=True)
76 config.addOutputVar (self.containerName, 'TruthVisMass', 'TruthVisMass', noSys=True)
77 config.addOutputVar (self.containerName, 'IsTruthMatched', 'IsTruthMatched', noSys=True)
78 config.addOutputVar (self.containerName, 'IsTruthHadronic', 'IsTruthHadronic', noSys=True)
79
80 # Decorate extra variables
81 if self.decorateExtraVariables:
82 alg = config.createAlgorithm( 'CP::DiTauExtraVariablesAlg',
83 'DiTauExtraVariablesAlg',
84 reentrant=True )
85 alg.ditaus = config.readName (self.containerName)
86 config.addOutputVar (self.containerName, 'omniScore', 'omniScore', noSys=True)
87 config.addOutputVar (self.containerName, 'nSubjets', 'nSubjets', noSys=True)
88 config.addOutputVar (self.containerName, 'leadSubjetPt', 'leadSubjetPt', noSys=True)
89 config.addOutputVar (self.containerName, 'leadSubjetEta', 'leadSubjetEta', noSys=True)
90 config.addOutputVar (self.containerName, 'leadSubjetPhi', 'leadSubjetPhi', noSys=True)
91 config.addOutputVar (self.containerName, 'leadSubjetE', 'leadSubjetE', noSys=True)
92 config.addOutputVar (self.containerName, 'leadSubjetNTracks', 'leadSubjetNTracks', noSys=True)
93 config.addOutputVar (self.containerName, 'leadSubjetCharge', 'leadSubjetCharge', noSys=True)
94 config.addOutputVar (self.containerName, 'subleadSubjetPt', 'subleadSubjetPt', noSys=True)
95 config.addOutputVar (self.containerName, 'subleadSubjetEta', 'subleadSubjetEta', noSys=True)
96 config.addOutputVar (self.containerName, 'subleadSubjetPhi', 'subleadSubjetPhi', noSys=True)
97 config.addOutputVar (self.containerName, 'subleadSubjetE', 'subleadSubjetE', noSys=True)
98 config.addOutputVar (self.containerName, 'subleadSubjetNTracks', 'subleadSubjetNTracks', noSys=True)
99 config.addOutputVar (self.containerName, 'subleadSubjetCharge', 'subleadSubjetCharge', noSys=True)
100
101 # Set up the tau 4-momentum smearing algorithm:
102 alg = config.createAlgorithm( 'CP::DiTauSmearingAlg', 'DiTauSmearingAlg' )
103 config.addPrivateTool( 'smearingTool', 'TauAnalysisTools::DiTauSmearingTool' )
104 alg.taus = config.readName (self.containerName)
105 alg.tausOut = config.copyName (self.containerName)
106 alg.preselection = config.getPreselection (self.containerName, '')
107
108 # Save base kinematic ditau variables in output
109 config.addOutputVar (self.containerName, 'pt', 'pt')
110 config.addOutputVar (self.containerName, 'eta', 'eta', noSys=True)
111 config.addOutputVar (self.containerName, 'phi', 'phi', noSys=True)
112 config.addOutputVar (self.containerName, 'm', 'm', noSys=True)
113
114
115

Member Data Documentation

◆ containerName

python.DiTauAnalysisConfig.DiTauCalibrationConfig.containerName

Definition at line 49 of file DiTauAnalysisConfig.py.

◆ decorateExtraVariables

python.DiTauAnalysisConfig.DiTauCalibrationConfig.decorateExtraVariables

Definition at line 81 of file DiTauAnalysisConfig.py.

◆ inputContainer

python.DiTauAnalysisConfig.DiTauCalibrationConfig.inputContainer

Definition at line 47 of file DiTauAnalysisConfig.py.


The documentation for this class was generated from the following file: