34 def makeAlgs(self, config):
35 config.setSourceName (self.containerName, self.containerName)
36
37
38 alg = config.createAlgorithm('CP::ParticleLevelChargeDecoratorAlg',
39 'ParticleLevelChargeDecoratorElectrons',
40 reentrant=True)
41 alg.particles = self.containerName
42
43
44 alg = config.createAlgorithm('CP::ParticleLevelIsolationAlg',
45 'ParticleLevelIsolationElectrons',
46 reentrant=True)
47 alg.particles = self.containerName
48 alg.isolation = 'isIsolated' + self.selectionName if self.isolated else 'isIsolatedButNotRequired' + self.selectionName
49 alg.notTauOrigin = 'notFromTau' + self.selectionName if self.notFromTau else 'notFromTauButNotRequired' + self.selectionName
50 alg.checkType = 'IsoElectron'
51
52 if self.isolated:
53 config.addSelection (self.containerName, self.selectionName, alg.isolation+',as_char')
54 if self.notFromTau:
55 config.addSelection (self.containerName, self.selectionName, alg.notTauOrigin+',as_char')
56
57
58 if 'pt' not in config.getOutputVars(self.containerName):
59 outputVars = [
60 ['pt_dressed', 'pt', 'float'],
61 ['eta_dressed', 'eta', 'float'],
62 ['phi_dressed', 'phi', 'float'],
63 ['e_dressed', 'e', 'float'],
64 ['charge', 'charge', 'float'],
65 ['classifierParticleType', 'type', 'unsigned'],
66 ['classifierParticleOrigin', 'origin', 'unsigned'],
67 ]
68 if self.saveUID:
69 outputVars += [['uid', 'uid', 'int']]
70 for decoration, branch, auxType in outputVars:
71 config.addOutputVar (self.containerName, decoration, branch, noSys=True, auxType=auxType)