13def PhysCommonAugmentationsCfg(flags,**kwargs):
14 """Configure the common augmentation"""
15 acc = ComponentAccumulator()
16
17 from TrkConfig.VertexFindingFlags import VertexSortingSetup
18 if flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.PVNNSorting:
19 from DerivationFrameworkPhys.DerivationPrimaryVertexConfig import PVNNVertexCfg
20 acc.merge(PVNNVertexCfg(flags))
21 elif flags.Tracking.PriVertex.sortingSetup is VertexSortingSetup.HGamNNSorting:
22 from DerivationFrameworkPhys.DerivationPrimaryVertexConfig import HGamNNVertexCfg
23 acc.merge(HGamNNVertexCfg(flags))
24
25 from DerivationFrameworkTau.TauCommonConfig import AddTauMassDecoratorCfg
26
27 acc.merge(AddTauMassDecoratorCfg(flags, TauContainerName="TauJets"))
28 if flags.Tau.TauEleRM_isAvailable:
29 acc.merge(AddTauMassDecoratorCfg(flags, TauContainerName="TauJets_EleRM"))
30
31
32 if flags.Input.isMC:
33 from DerivationFrameworkMCTruth.MCTruthCommonConfig import (
34 AddPhysTruthContentsCfg,
35 AddMiniTruthCollectionLinksCfg)
36 acc.merge(AddPhysTruthContentsCfg(flags))
37
38 acc.merge(AddMiniTruthCollectionLinksCfg(flags))
39
40
41 from DerivationFrameworkInDet.InDetCommonConfig import InDetCommonCfg
42 from DerivationFrameworkMuons.MuonsCommonConfig import MuonsCommonCfg
43 from DerivationFrameworkEGamma.EGammaCommonConfig import EGammaCommonCfg
44
45
46 acc.merge(InDetCommonCfg(flags,
47 DoVertexFinding = flags.Tracking.doVertexFinding,
48 AddPseudoTracks = flags.Tracking.doPseudoTracking,
49 DecoLRTTTVA = False,
50 DoR3LargeD0 = flags.Tracking.doLargeD0,
51 StoreSeparateLargeD0Container = flags.Tracking.storeSeparateLargeD0Container,
52 MergeLRT = False))
53 acc.merge(MuonsCommonCfg(flags))
54 acc.merge(EGammaCommonCfg(flags))
55
56 from DerivationFrameworkJetEtMiss.JetCommonConfig import JetCommonCfg
57 from DerivationFrameworkTau.TauCommonConfig import (AddMuonRemovalTauAODReRecoAlgCfg, AddTauIDDecorationCfg, AddTauTESCompatibilityDecorationCfg)
58 from DerivationFrameworkTau.DiTauCommonConfig import (AddDiTauLowPtCfg, AddDiTauChargeDecoratorCfg, AddDiTauIDDecorationCfg)
59 from DerivationFrameworkJetEtMiss.METCommonConfig import METCommonCfg
60 from DerivationFrameworkJetEtMiss.METCommonConfig import HadRecoilMETCfg
61 acc.merge(JetCommonCfg(flags))
62
63
64 from eflowRec.PFCfg import PFGlobalFlowElementLinkingCfg
65 acc.merge(PFGlobalFlowElementLinkingCfg(flags))
66 acc.merge(AddDiTauLowPtCfg(flags))
67 acc.merge(AddMuonRemovalTauAODReRecoAlgCfg(flags))
68
69 acc.merge(AddTauIDDecorationCfg(flags, TauContainerName="TauJets"))
70 acc.merge(AddTauIDDecorationCfg(flags, TauContainerName="TauJets_MuonRM"))
71
72 acc.merge(AddTauTESCompatibilityDecorationCfg(flags, TauContainerName="TauJets"))
73 acc.merge(AddTauTESCompatibilityDecorationCfg(flags, TauContainerName="TauJets_MuonRM"))
74
75 acc.merge(AddDiTauIDDecorationCfg(flags, DiTauContainerName="DiTauJets"))
76
77 if flags.Tau.TauEleRM_isAvailable:
78 acc.merge(AddTauIDDecorationCfg(flags, TauContainerName="TauJets_EleRM"))
79 acc.merge(AddTauTESCompatibilityDecorationCfg(flags, TauContainerName="TauJets_EleRM"))
80
81 acc.merge(AddDiTauChargeDecoratorCfg(flags, DiTauContainerName="DiTauJets"))
82 acc.merge(AddDiTauChargeDecoratorCfg(flags, DiTauContainerName="DiTauJetsLowPt"))
83 if flags.Reco.EnableBTagging:
84 from BTagging.FlavorTaggingConfig import FlavorTaggingCfg
85 acc.merge(FlavorTaggingCfg(flags, "AntiKt4EMPFlowJets"))
86 acc.merge(FlavorTaggingCfg(flags, "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets"))
87 if flags.BTagging.RunNewVrtSecInclusive:
88 from NewVrtSecInclusiveTool.NewVrtSecInclusiveAlgConfig import (
89 NewVrtSecInclusiveAlgTightCfg
90 )
91 acc.merge(NewVrtSecInclusiveAlgTightCfg(flags))
92 if flags.GeoModel.Run >= LHCPeriod.Run4:
93 acc.merge(FlavorTaggingCfg(flags, "AntiKt4EMTopoJets"))
94
95 acc.merge(METCommonCfg(flags))
96 acc.merge(HadRecoilMETCfg(flags))
97
98 from AssociationUtils.AssociationUtilsConfig import FEAssociationCfg
99 acc.merge(FEAssociationCfg(flags,
100 SmallRJetChargedFELinksDecorKey="",
101 SmallRJetNeutralFELinksDecorKey="",
102 LargeRJetChargedFELinksDecorKey="",
103 LargeRJetNeutralFELinksDecorKey=""))
104
105
106 if flags.Reco.EnableTrigger or flags.Trigger.triggerConfig == 'INFILE':
107 from JetTagDerivationUtils.TrigBTagCopierConfig import TrigBTagCopierAlgCfg
108 acc.merge(TrigBTagCopierAlgCfg(flags))
109
110 from DerivationFrameworkPhys.TriggerMatchingCommonConfig import TriggerMatchingCommonRun2Cfg
111 from DerivationFrameworkPhys.TriggerMatchingCommonConfig import TriggerMatchingCommonRun2ToRun3Cfg
112 from DerivationFrameworkPhys.TriggerMatchingCommonConfig import TriggerMatchingCommonRun3Cfg
113
114 triggerListsHelper = kwargs['TriggerListsHelper']
115 if flags.Trigger.EDMVersion == 2:
116
117
118
119
120
121 acc.merge(TriggerMatchingCommonRun2Cfg(
122 flags,
123 name = "PhysCommonTrigMatchNoTau",
124 OutputContainerPrefix = "TrigMatch_",
125 ChainNames = triggerListsHelper.Run2TriggerNamesNoTau))
126 acc.merge(TriggerMatchingCommonRun2Cfg(
127 flags,
128 name = "PhysCommonTrigMatchTau",
129 OutputContainerPrefix = "TrigMatch_",
130 ChainNames = triggerListsHelper.Run2TriggerNamesTau,
131 DRThreshold = 0.2))
132
133
134
135 acc.merge(TriggerMatchingCommonRun2ToRun3Cfg(
136 flags,
137 TriggerList = triggerListsHelper.Run2TriggerNamesNoTau +
138 triggerListsHelper.Run2TriggerNamesTau))
139 if flags.Trigger.EDMVersion == 3:
140
141 acc.merge(TriggerMatchingCommonRun3Cfg(
142 flags, TriggerList = triggerListsHelper.Run3TriggerNames))
143
144 return acc
145