45 def makeAlgs (self, config) :
46
47
48 if 'EleRM' in self.inputContainer:
49 if config.dataType() is DataType.Data and config.dataYear() <= 2023:
50 raise RuntimeError("EleRM taus are only available from 2024 dataset onward")
51 elif config.dataType() is not DataType.Data and config.campaign() <= Campaign.MC23d:
52 raise RuntimeError("EleRM taus are only available from 2024 dataset onward")
53
54 postfix = self.postfix
55 if postfix != '' and postfix[0] != '_' :
56 postfix = '_' + postfix
57
58 inputContainer = "AnalysisTauJets" if config.isPhyslite() else "TauJets"
59 if self.inputContainer:
60 inputContainer = self.inputContainer
61 config.setSourceName (self.containerName, inputContainer)
62
63
64 if config.wantCopy (self.containerName) :
65 alg = config.createAlgorithm( 'CP::AsgShallowCopyAlg', 'TauShallowCopyAlg' )
66 alg.input = config.readName (self.containerName)
67 alg.output = config.copyName (self.containerName)
68 alg.outputType = 'xAOD::TauJetContainer'
69 decorations = []
70 if self.addGlobalFELinksDep:
71 decorations += ['neutralGlobalFELinks', 'chargedGlobalFELinks']
72 if config.dataType() is not DataType.Data:
73 decorations += ['IsTruthMatched', 'truthJetLink', 'truthParticleLink']
74 if decorations:
75 alg.declareDecorations = decorations
76
77
78 if self.rerunTruthMatching and config.dataType() is not DataType.Data:
79 alg = config.createAlgorithm( 'CP::TauTruthMatchingAlg',
80 'TauTruthMatchingAlg' )
81 config.addPrivateTool( 'matchingTool',
82 'TauAnalysisTools::TauTruthMatchingTool' )
83 alg.matchingTool.TruthJetContainerName = 'AntiKt4TruthDressedWZJets'
84 alg.taus = config.readName (self.containerName)
85 alg.preselection = config.getPreselection (self.containerName, '')
86
87
88 if self.decorateTruth and config.dataType() is not DataType.Data:
89 alg = config.createAlgorithm( 'CP::TauTruthDecorationsAlg',
90 'TauTruthDecorationsAlg',
91 reentrant=True )
92 alg.taus = config.readName (self.containerName, nominal=True)
93 alg.doubleDecorations = ['pt_vis', 'pt_invis', 'eta_vis', 'eta_invis', 'phi_vis', 'phi_invis', 'm_vis', 'm_invis']
94 alg.floatDecorations = []
95 alg.intDecorations = ['pdgId']
96 alg.unsignedIntDecorations = ['classifierParticleOrigin', 'classifierParticleType']
97 alg.charDecorations = ['IsHadronicTau']
98 alg.prefix = 'truth_'
99
100
101 for var in ['DecayMode', 'ParticleType', 'PartonTruthLabelID'] + alg.doubleDecorations[:] + alg.floatDecorations[:] + alg.intDecorations[:] + alg.unsignedIntDecorations[:] + alg.charDecorations[:]:
102 branchName = alg.prefix + var
103 if 'classifierParticle' in var:
104 branchOutput = alg.prefix + var.replace('classifierParticle', '').lower()
105 else:
106 branchOutput = branchName
107 config.addOutputVar (self.containerName, branchName, branchOutput, noSys=True)
108
109
110 if self.decorateExtraVariables:
111 alg = config.createAlgorithm( 'CP::TauExtraVariablesAlg',
112 'TauExtraVariablesAlg',
113 reentrant=True )
114 alg.taus = config.readName (self.containerName, nominal=True)
115
116
117 alg = config.createAlgorithm( 'CP::TauSmearingAlg', 'TauSmearingAlg' )
118 config.addPrivateTool( 'smearingTool', 'TauAnalysisTools::TauSmearingTool' )
119 alg.smearingTool.useFastSim = config.dataType() is DataType.FastSim
120 alg.smearingTool.Campaign = "mc23" if config.geometry() is LHCPeriod.Run3 else "mc20"
121 alg.taus = config.readName (self.containerName)
122 alg.tausOut = config.copyName (self.containerName)
123 config.setExtraOutputs ({('xAOD::IParticleContainer' , 'StoreGateSvc+' + config.readName(self.containerName, nominal=True) + '.RNNEleScoreSigTrans_v1')})
124 alg.preselection = config.getPreselection (self.containerName, '')
125
126
127 alg = config.createAlgorithm( 'CP::AsgEnergyDecoratorAlg', 'EnergyDecorator' )
128 alg.particles = config.readName (self.containerName)
129
130 config.addOutputVar (self.containerName, 'pt', 'pt')
131 config.addOutputVar (self.containerName, 'eta', 'eta', noSys=True)
132 config.addOutputVar (self.containerName, 'phi', 'phi', noSys=True)
133 config.addOutputVar (self.containerName, 'e_%SYS%', 'e')
134 config.addOutputVar (self.containerName, 'charge', 'charge', noSys=True)
135 config.addOutputVar (self.containerName, 'NNDecayMode', 'NNDecayMode', noSys=True, auxType='int')
136 config.addOutputVar (self.containerName, 'passTATTauMuonOLR', 'passTATTauMuonOLR', noSys=True, auxType='char')
137 config.addOutputVar (self.containerName, 'TESCompatibility', 'TESCompatibility')
138 if self.decorateExtraVariables:
139 config.addOutputVar (self.containerName, 'nTracksCharged', 'nTracksCharged', noSys=True)
140
141
1.14.0