57 def makeAlgs (self, config) :
58
59 log = logging.getLogger('MuonCalibrationConfig')
60
61
62
63 if self.calibMode == 'correctData_CB':
64 calibMode = 0
65 elif self.calibMode == 'correctData_IDMS':
66 calibMode = 1
67 elif self.calibMode == 'notCorrectData_IDMS':
68 calibMode = 2
69 elif self.calibMode == 'notCorrectData_CB':
70 calibMode = 3
71 else :
72 raise ValueError ("invalid calibMode: \"" + self.calibMode + "\". Allowed values are correctData_CB, correctData_IDMS, notCorrectData_IDMS, notCorrectData_CB")
73
74 inputContainer = "AnalysisMuons" if config.isPhyslite() else "Muons"
75 if self.inputContainer:
76 inputContainer = self.inputContainer
77 config.setSourceName (self.containerName, inputContainer)
78 config.setContainerMeta (self.containerName, 'calibMode', calibMode)
79
80
81 if config.wantCopy (self.containerName) :
82 alg = config.createAlgorithm( 'CP::AsgShallowCopyAlg', 'MuonShallowCopyAlg' )
83 alg.input = config.readName (self.containerName)
84 alg.output = config.copyName (self.containerName)
85 alg.outputType = 'xAOD::MuonContainer'
86 decorationList = ['DFCommonJetDr','neflowisol20_CloseByCorr',
87 'ptvarcone30_Nonprompt_All_MaxWeightTTVA_pt1000_CloseByCorr',
88 'ptvarcone30_Nonprompt_All_MaxWeightTTVA_pt500_CloseByCorr',
89 'topoetcone20_CloseByCorr']
90 if self.addGlobalFELinksDep:
91 decorationList += ['neutralGlobalFELinks', 'chargedGlobalFELinks']
92 if config.dataType() is not DataType.Data:
93 decorationList += ['TruthLink']
94 if self.addGlobalFELinksDep or config.dataType() is not DataType.Data:
95 alg.declareDecorations = decorationList
96
97
98 alg = config.createAlgorithm( 'CP::AsgSelectionAlg',
99 'MuonEtaCutAlg' )
100 config.addPrivateTool( 'selectionTool', 'CP::AsgPtEtaSelectionTool' )
101 alg.selectionTool.maxEta = self.maxEta
102 alg.selectionDecoration = 'selectEta' + self.postfix + ',as_bits'
103 alg.particles = config.readName (self.containerName)
104 alg.preselection = config.getPreselection (self.containerName, '')
105 config.addSelection (self.containerName, '', alg.selectionDecoration)
106
107
108 alg = config.createAlgorithm( 'CP::MuonCalibrationAndSmearingAlg',
109 'MuonCalibrationAndSmearingAlg' )
110 config.addPrivateTool( 'calibrationAndSmearingTool',
111 'CP::MuonCalibTool' )
112
113 alg.calibrationAndSmearingTool.IsRun3Geo = config.geometry() >= LHCPeriod.Run3
114 alg.calibrationAndSmearingTool.calibMode = calibMode
115 if config.geometry() is LHCPeriod.Run4:
116 log.warning("Disabling NSW hits for Run4 geometry")
117 alg.calibrationAndSmearingTool.ExcludeNSWFromPrecisionLayers = True
118 else:
119 alg.calibrationAndSmearingTool.ExcludeNSWFromPrecisionLayers = self.excludeNSWFromPrecisionLayers and (config.geometry() >= LHCPeriod.Run3)
120 alg.muons = config.readName (self.containerName)
121 alg.muonsOut = config.copyName (self.containerName)
122 alg.preselection = config.getPreselection (self.containerName, '')
123 if config.isPhyslite() and not self.recalibratePhyslite :
124 alg.skipNominal = True
125
126
127 if self.minPt > 0:
128 alg = config.createAlgorithm( 'CP::AsgSelectionAlg', 'MuonPtCutAlg' )
129 alg.selectionDecoration = 'selectPt' + self.postfix + ',as_bits'
130 config.addPrivateTool( 'selectionTool', 'CP::AsgPtEtaSelectionTool' )
131 alg.particles = config.readName (self.containerName)
132 alg.selectionTool.minPt = self.minPt
133 alg.preselection = config.getPreselection (self.containerName, '')
134 config.addSelection (self.containerName, '', alg.selectionDecoration,
135 preselection = True)
136
137 alg = config.createAlgorithm( 'CP::AsgEnergyDecoratorAlg', 'EnergyDecorator' )
138 alg.particles = config.readName (self.containerName)
139
140 config.addOutputVar (self.containerName, 'pt', 'pt')
141 config.addOutputVar (self.containerName, 'eta', 'eta', noSys=True)
142 config.addOutputVar (self.containerName, 'phi', 'phi', noSys=True)
143 config.addOutputVar (self.containerName, 'e_%SYS%', 'e')
144 config.addOutputVar (self.containerName, 'charge', 'charge', noSys=True)
145
146
147 if self.decorateTruth and config.dataType() is not DataType.Data:
148 config.addOutputVar (self.containerName, "truthType", "truth_type", noSys=True)
149 config.addOutputVar (self.containerName, "truthOrigin", "truth_origin", noSys=True)
150
151 config.addOutputVar (self.containerName, 'muonType', 'muonType', noSys=True, enabled=self.writeColumnarToolVariables)
152
153
1.14.0