ATLAS Offline Software
Functions
python.JETM2 Namespace Reference

Functions

def JETM2KernelCfg (flags, name='JETM2Kernel', **kwargs)
 
def JETM2ExtraContentCfg (flags)
 
def JETM2Cfg (flags)
 

Function Documentation

◆ JETM2Cfg()

def python.JETM2.JETM2Cfg (   flags)

Definition at line 99 of file JETM2.py.

99 def JETM2Cfg(flags):
100 
101  acc = ComponentAccumulator()
102 
103  # Get the lists of triggers needed for trigger matching.
104  # This is needed at this scope (for the slimming) and further down in the config chain
105  # for actually configuring the matching, so we create it here and pass it down
106  # TODO: this should ideally be called higher up to avoid it being run multiple times in a train
107  from DerivationFrameworkPhys.TriggerListsHelper import TriggerListsHelper
108  JETM2TriggerListsHelper = TriggerListsHelper(flags)
109 
110  # Skimming, thinning, augmentation, extra content
111  acc.merge(JETM2KernelCfg(flags, name="JETM2Kernel", StreamName = 'StreamDAOD_JETM2', TriggerListsHelper = JETM2TriggerListsHelper))
112 
113  # ============================
114  # Define contents of the format
115  # =============================
116  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
117  from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
118  from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
119 
120  JETM2SlimmingHelper = SlimmingHelper("JETM2SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
121 
122  JETM2SlimmingHelper.SmartCollections = ["EventInfo","InDetTrackParticles", "PrimaryVertices",
123  "Electrons", "Photons", "Muons", "TauJets",
124  "MET_Baseline_AntiKt4EMPFlow",
125  "AntiKt4EMTopoNoPtCutJets","AntiKt4EMPFlowJets",
126  "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets",
127  "AntiKt4EMPFlowJets_FTAG",
128  "BTagging_AntiKt4EMPFlow",
129  "BTagging_AntiKtVR30Rmax4Rmin02Track"]
130 
131  JETM2SlimmingHelper.AllVariables = ["CaloCalFwdTopoTowers",
132  "CHSGChargedParticleFlowObjects","CHSGNeutralParticleFlowObjects",
133  "CSSKGChargedParticleFlowObjects","CSSKGNeutralParticleFlowObjects",
134  "Kt4EMTopoOriginEventShape","Kt4EMPFlowEventShape","Kt4EMPFlowPUSBEventShape",
135  "Kt4EMPFlowNeutEventShape","Kt4UFOCSSKEventShape","Kt4UFOCSSKNeutEventShape"]
136 
137  # Low-level inputs
138  from DerivationFrameworkJetEtMiss.CommonJETMXContent import ClusterVariables, FlowElementVariables, UFOVariables, TrackingVariables, TrackingVariablesHGTD, FELinks
139  JETM2SlimmingHelper.ExtraVariables += [".".join(["CaloCalTopoClusters"] + ClusterVariables)]
140  JETM2SlimmingHelper.ExtraVariables += [".".join(["GlobalChargedParticleFlowObjects"] + FlowElementVariables)]
141  JETM2SlimmingHelper.ExtraVariables += [".".join(["GlobalNeutralParticleFlowObjects"] + FlowElementVariables)]
142  JETM2SlimmingHelper.ExtraVariables += [".".join(["UFO"] + UFOVariables)]
143  JETM2SlimmingHelper.ExtraVariables += [".".join(["UFOCSSK"] + UFOVariables)]
144  JETM2SlimmingHelper.ExtraVariables += [".".join(["InDetTrackParticles"] + TrackingVariables)]
145  if flags.GeoModel.Run >= LHCPeriod.Run4:
146  JETM2SlimmingHelper.ExtraVariables += [".".join(["InDetTrackParticles"] + TrackingVariablesHGTD)]
147 
148  # Links of physics objects to FlowElements
149  JETM2SlimmingHelper.ExtraVariables += FELinks
150 
151  # Detailed substructure information
152  from DerivationFrameworkJetEtMiss.CommonJETMXContent import ExtraJSSVariables
153  JETM2SlimmingHelper.ExtraVariables += [".".join(["AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets"] + ExtraJSSVariables)]
154 
155  # sub-structure taggers score
156  from DerivationFrameworkJetEtMiss.CommonJETMXContent import JSSTaggersVariables
157  JETM2SlimmingHelper.ExtraVariables += JSSTaggersVariables
158 
159  JETM2SlimmingHelper.ExtraVariables += ["AntiKt4EMPFlowJets.GhostTower.IsoFixedCone5Pt.IsoFixedCone5PtPUsub",
160  "AntiKt4EMTopoNoPtCutJets.IsoFixedCone5Pt.IsoFixedCone5PtPUsub",
161  "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets.SizeParameter.GhostTrack",
162  "GSFTrackParticles.particleHypothesis.vx.vy.vz",
163  "PrimaryVertices.x.y.z.covariance.trackWeights",
164  "TauJets.clusterLinks",
165  "Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus.clusterLinks.FSR_CandidateEnergy",
166  "MuonSegments.x.y.z.px.py.pz",
167  "BTagging_AntiKt4EMPFlow.jetLink",
168  "BTagging_AntiKtVR30Rmax4Rmin02Track.jetLink"]
169 
170  JETM2SlimmingHelper.AppendToDictionary.update({'CSSKGNeutralParticleFlowObjects': 'xAOD::FlowElementContainer',
171  'CSSKGNeutralParticleFlowObjectsAux': 'xAOD::ShallowAuxContainer',
172  'CSSKGChargedParticleFlowObjects': 'xAOD::FlowElementContainer',
173  'CSSKGChargedParticleFlowObjectsAux': 'xAOD::ShallowAuxContainer',
174  'UFO': 'xAOD::FlowElementContainer',
175  'UFOAux': 'xAOD::FlowElementAuxContainer',
176  'Kt4UFOCSSKEventShape': 'xAOD::EventShape',
177  'Kt4UFOCSSKEventShapeAux': 'xAOD::EventShapeAuxInfo',
178  'Kt4UFOCSSKNeutEventShape': 'xAOD::EventShape',
179  'Kt4UFOCSSKNeutEventShapeAux': 'xAOD::EventShapeAuxInfo',
180  'AntiKt4EMTopoNoPtCutJets': 'xAOD::JetContainer',
181  'AntiKt4EMTopoNoPtCutJetsAux': 'xAOD::JetAuxContainer'})
182 
183  from DerivationFrameworkJetEtMiss.JetCommonConfig import addOriginCorrectedClustersToSlimmingTool
184  addOriginCorrectedClustersToSlimmingTool(JETM2SlimmingHelper,writeLC=True,writeEM=True)
185 
186  # Truth containers
187  if flags.Input.isMC:
188 
189  from DerivationFrameworkMCTruth.MCTruthCommonConfig import addTruth3ContentToSlimmerTool
190  addTruth3ContentToSlimmerTool(JETM2SlimmingHelper)
191 
192  JETM2SlimmingHelper.AppendToDictionary.update({'TruthParticles': 'xAOD::TruthParticleContainer',
193  'TruthParticlesAux': 'xAOD::TruthParticleAuxContainer'})
194 
195  JETM2SlimmingHelper.AllVariables += ["TruthTopQuarkWithDecayParticles","TruthTopQuarkWithDecayVertices","TruthHFWithDecayParticles",
196  "AntiKt4TruthJets", "InTimeAntiKt4TruthJets", "OutOfTimeAntiKt4TruthJets",
197  "TruthParticles", "TruthVertices","TruthEvents"]
198  JETM2SlimmingHelper.ExtraVariables += ["AntiKt10TruthSoftDropBeta100Zcut10Jets.SizeParameter"]
199  JETM2SlimmingHelper.SmartCollections += ["AntiKt4TruthJets","AntiKt10TruthJets","AntiKt4TruthWZJets","AntiKt10TruthDressedWZSoftDropBeta100Zcut10Jets"]
200 
201  JETM2SlimmingHelper.ExtraVariables += ["AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets.GhostTQuarksFinalCount.GhostHBosonsCount.GhostZBosonsCount.GhostWBosonsCount",
202  "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets.GhostTQuarksFinalPt.GhostHBosonsPt.GhostZBosonsPt.GhostWBosonsPt",
203  "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets.GhostBHadronsFinalPt.GhostCHadronsFinalPt",
204  "AntiKt4TruthWZJets.IsoFixedCone5Pt",
205  "AntiKt4TruthDressedWZJets.IsoFixedCone5Pt"]
206 
207  # Trigger content
208  JETM2SlimmingHelper.IncludeTriggerNavigation = False
209  JETM2SlimmingHelper.IncludeJetTriggerContent = False
210  JETM2SlimmingHelper.IncludeMuonTriggerContent = False
211  JETM2SlimmingHelper.IncludeEGammaTriggerContent = False
212  JETM2SlimmingHelper.IncludeTauTriggerContent = False
213  JETM2SlimmingHelper.IncludeEtMissTriggerContent = False
214  JETM2SlimmingHelper.IncludeBJetTriggerContent = False
215  JETM2SlimmingHelper.IncludeBPhysTriggerContent = False
216  JETM2SlimmingHelper.IncludeMinBiasTriggerContent = False
217 
218  jetOutputList = ["AntiKt4UFOCSSKNoPtCutJets","AntiKt4EMPFlowNoPtCutJets"]
219  from DerivationFrameworkJetEtMiss.JetCommonConfig import addJetsToSlimmingTool
220  addJetsToSlimmingTool(JETM2SlimmingHelper, jetOutputList, JETM2SlimmingHelper.SmartCollections)
221 
222  # Output stream
223  JETM2ItemList = JETM2SlimmingHelper.GetItemList()
224  acc.merge(OutputStreamCfg(flags, "DAOD_JETM2", ItemList=JETM2ItemList, AcceptAlgs=["JETM2Kernel"]))
225  acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_JETM2", AcceptAlgs=["JETM2Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
226 
227  return acc
228 

◆ JETM2ExtraContentCfg()

def python.JETM2.JETM2ExtraContentCfg (   flags)

Definition at line 42 of file JETM2.py.

42 def JETM2ExtraContentCfg(flags):
43 
44  acc = ComponentAccumulator()
45 
46  from JetRecConfig.JetRecConfig import JetRecCfg, getInputAlgs, getConstitPJGAlg
47  from JetRecConfig.JetInputConfig import buildEventShapeAlg
48  from JetRecConfig.StandardJetConstits import stdConstitDic as cst
49  from JetRecConfig.StandardSmallRJets import AntiKt4UFOCSSKNoPtCut, AntiKt4EMPFlowNoPtCut, AntiKt4EMTopoNoPtCut
50  from JetRecConfig.StandardLargeRJets import AntiKt10TruthDressedWZSoftDrop
51 
52  #=======================================
53  # CHS R = 0.4 UFO jets
54  #=======================================
55 
56  algs = getInputAlgs(cst.UFO, flags=flags)
57  for alg in algs:
58  if isinstance(alg, ComponentAccumulator):
59  acc.merge(alg)
60  else:
61  acc.addEventAlgo(alg)
62 
63  #=======================================
64  # NoPtCut containers
65  #=======================================
66  jetList = [AntiKt4UFOCSSKNoPtCut,AntiKt4EMPFlowNoPtCut,AntiKt4EMTopoNoPtCut]
67 
68  #=======================================
69  # R = 1.0 truth WZ Dressed jets
70  #=======================================
71  if flags.Input.isMC:
72  jetList += [AntiKt10TruthDressedWZSoftDrop]
73 
74 
75  for jd in jetList:
76  acc.merge(JetRecCfg(flags,jd))
77 
78  #=======================================
79  # UFO CSSK event shape
80  #=======================================
81 
82  acc.addEventAlgo(buildEventShapeAlg(cst.UFOCSSK,'', suffix=None))
83  acc.addEventAlgo(getConstitPJGAlg(cst.UFOCSSK, suffix='Neut'))
84  acc.addEventAlgo(buildEventShapeAlg(cst.UFOCSSK,'', suffix='Neut'))
85 
86  #=======================================
87  # More detailed truth information
88  #=======================================
89 
90  if flags.Input.isMC:
91  from DerivationFrameworkMCTruth.MCTruthCommonConfig import AddTopQuarkAndDownstreamParticlesCfg, AddTruthCollectionNavigationDecorationsCfg
92  acc.merge(AddTopQuarkAndDownstreamParticlesCfg(flags))
93  acc.merge(AddTruthCollectionNavigationDecorationsCfg(flags, TruthCollections=["TruthTopQuarkWithDecayParticles","TruthBosonsWithDecayParticles"],prefix='Top'))
94 
95 
96  return acc
97 
98 

◆ JETM2KernelCfg()

def python.JETM2.JETM2KernelCfg (   flags,
  name = 'JETM2Kernel',
**  kwargs 
)
Configure the derivation framework driving algorithm (kernel) for JETM2

Definition at line 11 of file JETM2.py.

11 def JETM2KernelCfg(flags, name='JETM2Kernel', **kwargs):
12  """Configure the derivation framework driving algorithm (kernel) for JETM2"""
13  acc = ComponentAccumulator()
14 
15  # Common augmentations
16  from DerivationFrameworkPhys.PhysCommonConfig import PhysCommonAugmentationsCfg
17  acc.merge(PhysCommonAugmentationsCfg(flags, TriggerListsHelper = kwargs['TriggerListsHelper']))
18 
19  if flags.Input.isMC:
20  # thinning tools:
21  truthThinningTool = CompFactory.DerivationFramework.MenuTruthThinning(name = "JETM2TruthThinning",
22  StreamName = kwargs['StreamName'],
23  WriteAllStable = True,
24  WritePartons = False,
25  WriteHadrons = False,
26  WriteBHadrons = True,
27  WriteCHadrons = False,
28  WriteGeant = False,
29  WriteFirstN = 10)
30 
31  acc.addPublicTool(truthThinningTool)
32 
33  DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
34  acc.addEventAlgo(DerivationKernel(name, ThinningTools = [truthThinningTool] if flags.Input.isMC else []))
35 
36  # Extra jet content:
37  acc.merge(JETM2ExtraContentCfg(flags))
38 
39  return acc
40 
41 
python.OutputStreamConfig.OutputStreamCfg
def OutputStreamCfg(flags, streamName, ItemList=None, MetadataItemList=None, disableEventTag=False, trigNavThinningSvc=None, takeItemsFromInput=False, extendProvenanceRecord=True, keepProvenanceTagsRegEx=None, AcceptAlgs=None, HelperTools=None)
Definition: OutputStreamConfig.py:13
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.JetRecConfig.getInputAlgs
def getInputAlgs(jetOrConstitdef, flags, context="default", monTool=None)
Definition: JetRecConfig.py:279
python.JetCommonConfig.addOriginCorrectedClustersToSlimmingTool
def addOriginCorrectedClustersToSlimmingTool(slimhelper, writeLC=False, writeEM=False)
Helper to add origin corrected clusters to output.
Definition: JetCommonConfig.py:251
python.JetInputConfig.buildEventShapeAlg
def buildEventShapeAlg(jetOrConstitdef, inputspec, voronoiRf=0.9, radius=0.4, suffix=None)
Definition: JetInputConfig.py:152
python.MCTruthCommonConfig.AddTopQuarkAndDownstreamParticlesCfg
def AddTopQuarkAndDownstreamParticlesCfg(flags, generations=1, rejectHadronChildren=False)
Definition: MCTruthCommonConfig.py:258
python.JetRecConfig.JetRecCfg
def JetRecCfg(flags, jetdef, returnConfiguredDef=False)
Top level functions returning ComponentAccumulator out of JetDefinition.
Definition: JetRecConfig.py:36
python.JETM2.JETM2ExtraContentCfg
def JETM2ExtraContentCfg(flags)
Definition: JETM2.py:42
python.JetCommonConfig.addJetsToSlimmingTool
def addJetsToSlimmingTool(slimhelper, contentlist, smartlist=[])
Definition: JetCommonConfig.py:237
python.JETM2.JETM2Cfg
def JETM2Cfg(flags)
Definition: JETM2.py:99
python.JETM2.JETM2KernelCfg
def JETM2KernelCfg(flags, name='JETM2Kernel', **kwargs)
Definition: JETM2.py:11
TCS::join
std::string join(const std::vector< std::string > &v, const char c=',')
Definition: Trigger/TrigT1/L1Topo/L1TopoCommon/Root/StringUtils.cxx:10
python.JetRecConfig.getConstitPJGAlg
def getConstitPJGAlg(constitdef, suffix=None, flags=None, parent_jetdef=None)
Definition: JetRecConfig.py:350
python.MCTruthCommonConfig.addTruth3ContentToSlimmerTool
def addTruth3ContentToSlimmerTool(slimmer)
Definition: MCTruthCommonConfig.py:469
python.MCTruthCommonConfig.AddTruthCollectionNavigationDecorationsCfg
def AddTruthCollectionNavigationDecorationsCfg(flags, TruthCollections=[], prefix='')
Definition: MCTruthCommonConfig.py:337
python.PhysCommonConfig.PhysCommonAugmentationsCfg
def PhysCommonAugmentationsCfg(flags, **kwargs)
Definition: PhysCommonConfig.py:13
InfileMetaDataConfig.SetupMetaDataForStreamCfg
def SetupMetaDataForStreamCfg(flags, streamName="", AcceptAlgs=None, createMetadata=None, propagateMetadataFromInput=True, *args, **kwargs)
Definition: InfileMetaDataConfig.py:222
SlimmingHelper
Definition: SlimmingHelper.py:1
python.HION12.DerivationKernel
DerivationKernel
Definition: HION12.py:66