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 8 of file DiTauAnalysisConfig.py.

Constructor & Destructor Documentation

◆ __init__()

python.DiTauAnalysisConfig.DiTauCalibrationConfig.__init__ ( self)

Definition at line 11 of file DiTauAnalysisConfig.py.

11 def __init__ (self) :
12 super (DiTauCalibrationConfig, self).__init__ ()
13 self.setBlockName('DiTaus')
14 self.addOption ('inputContainer', '', type=str,
15 info="select ditau input container, by default set to DiTauJets")
16 self.addOption ('containerName', '', type=str,
17 noneAction='error',
18 info="the name of the output container after calibration.")
19 self.addOption ('postfix', '', type=str,
20 info="a postfix to apply to decorations and algorithm names. "
21 "Typically not needed here since the calibration is common to "
22 "all ditaus.")
23 self.addOption ('rerunTruthMatching', True, type=bool,
24 info="whether to rerun truth matching (sets up an instance of "
25 "CP::DiTauTruthMatchingAlg). The default is True.")
26 self.addOption ('decorateTruth', False, type=bool,
27 info="decorate truth particle information on the reconstructed one")
28 self.addOption ('decorateExtraVariables', True, type=bool,
29 info="decorate extra variables for the reconstructed ditau")
30

Member Function Documentation

◆ instanceName()

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

Definition at line 31 of file DiTauAnalysisConfig.py.

31 def instanceName (self) :
32 """Return the instance name for this block"""
33 return self.containerName + self.postfix
34

◆ makeAlgs()

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

Definition at line 35 of file DiTauAnalysisConfig.py.

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

Member Data Documentation

◆ containerName

python.DiTauAnalysisConfig.DiTauCalibrationConfig.containerName

Definition at line 44 of file DiTauAnalysisConfig.py.

◆ decorateExtraVariables

python.DiTauAnalysisConfig.DiTauCalibrationConfig.decorateExtraVariables

Definition at line 76 of file DiTauAnalysisConfig.py.

◆ inputContainer

python.DiTauAnalysisConfig.DiTauCalibrationConfig.inputContainer

Definition at line 42 of file DiTauAnalysisConfig.py.


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