ATLAS Offline Software
Loading...
Searching...
No Matches
python.IDTR2 Namespace Reference

Functions

 IDTR2Cfg (flags)

Variables

str streamName = "StreamDAOD_IDTR2"

Function Documentation

◆ IDTR2Cfg()

IDTR2Cfg ( flags)

Definition at line 14 of file IDTR2.py.

14def IDTR2Cfg(flags):
15
16 acc = ComponentAccumulator()
17
18 # LRT merge
19 TrackParticleContainer = "InDetTrackParticles"
20 if flags.Tracking.doLargeD0:
21 from DerivationFrameworkInDet.InDetToolsConfig import InDetLRTMergeCfg
22 acc.merge(InDetLRTMergeCfg(flags))
23 TrackParticleContainer = "InDetWithLRTTrackParticles"
24
25 # VrtSecInclusive
26 from VrtSecInclusive.VrtSecInclusiveConfig import VrtSecInclusiveCfg
27 acc.merge(VrtSecInclusiveCfg(
28 flags,
29 name="VrtSecInclusive",
30 AugmentingVersionString="",
31 FillIntermediateVertices=False,
32 TrackLocation=TrackParticleContainer))
33
34 # NewVrtSecInclusive
35 from NewVrtSecInclusiveTool.NewVrtSecInclusiveConfig import (
36 MaterialSVFinderToolCfg, DVFinderToolCfg, KsFinderToolCfg)
37 MaterialSVFinderTool = acc.popToolsAndMerge(
38 MaterialSVFinderToolCfg(flags, AugmentingVersionString="_Material"))
39 acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
40 name="NewVrtSecInclusive_Material",
41 TrackParticleContainer=TrackParticleContainer,
42 PrimaryVertexContainer="PrimaryVertices",
43 BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_Material",
44 BVertexTool=MaterialSVFinderTool))
45
46 DVFinderTool = acc.popToolsAndMerge(
47 DVFinderToolCfg(flags, AugmentingVersionString="_DV"))
48 acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
49 name="NewVrtSecInclusive_DV",
50 TrackParticleContainer=TrackParticleContainer,
51 PrimaryVertexContainer="PrimaryVertices",
52 BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_DV",
53 BVertexTool=DVFinderTool))
54
55 KsFinderTool = acc.popToolsAndMerge(
56 KsFinderToolCfg(flags, AugmentingVersionString="_Ks"))
57 acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
58 name="NewVrtSecInclusive_Ks",
59 TrackParticleContainer=TrackParticleContainer,
60 PrimaryVertexContainer="PrimaryVertices",
61 BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_Ks",
62 BVertexTool=KsFinderTool))
63
64 # V0Finder
65 IDTR2V0ContainerName = "IDTR2RecoV0Candidates"
66 IDTR2KshortContainerName = "IDTR2RecoKshortCandidates"
67 IDTR2LambdaContainerName = "IDTR2RecoLambdaCandidates"
68 IDTR2LambdabarContainerName = "IDTR2RecoLambdabarCandidates"
69
70 from InDetConfig.InDetV0FinderConfig import IDTR2_V0FinderToolCfg
71 V0FinderTool = acc.popToolsAndMerge(IDTR2_V0FinderToolCfg(
72 flags,
73 name="IDTR2_V0FinderTool",
74 TrackParticleCollection=TrackParticleContainer,
75 V0ContainerName=IDTR2V0ContainerName,
76 KshortContainerName=IDTR2KshortContainerName,
77 LambdaContainerName=IDTR2LambdaContainerName,
78 LambdabarContainerName=IDTR2LambdabarContainerName))
79
80 from DerivationFrameworkBPhys.V0ToolConfig import BPHY_Reco_V0FinderCfg
81 IDTR2_Reco_V0Finder = acc.popToolsAndMerge(BPHY_Reco_V0FinderCfg(
82 flags, derivation = "IDTR2",
83 V0ContainerName = IDTR2V0ContainerName,
84 KshortContainerName = IDTR2KshortContainerName,
85 LambdaContainerName = IDTR2LambdaContainerName,
86 LambdabarContainerName = IDTR2LambdabarContainerName,
87 CheckVertexContainers = ['PrimaryVertices'],
88 V0FinderTool = V0FinderTool))
89
90 skimmingTools = []
91 augmentationTools = [IDTR2_Reco_V0Finder]
92
93 if flags.Derivation.InDet.doTrackSystematics:
94 from InDetTrackSystematicsTools.InDetTrackSystematicsToolsConfig import TrackSystematicsAlgCfg
95 # Standard and LRT track efficiency variations applied together
96 SystTrackParticleContainer = (
97 TrackParticleContainer
98 + "_TRK_EFF_LOOSE_COMBINED_TRK_EFF_LARGED0_GLOBAL__1down")
99 acc.merge(TrackSystematicsAlgCfg(
100 flags,
101 InputTrackContainer=TrackParticleContainer,
102 OutputTrackContainer=SystTrackParticleContainer,
103 SystematicVariationSTD="TRK_EFF_LOOSE_COMBINED",
104 SystematicVariationLRT="TRK_EFF_LARGED0_GLOBAL"))
105
106 V0FinderToolSyst = acc.popToolsAndMerge(IDTR2_V0FinderToolCfg(
107 flags,
108 name="IDTR2_V0FinderTool_Syst",
109 TrackParticleCollection=SystTrackParticleContainer,
110 V0ContainerName=IDTR2V0ContainerName + "Syst",
111 KshortContainerName=IDTR2KshortContainerName + "Syst",
112 LambdaContainerName=IDTR2LambdaContainerName + "Syst",
113 LambdabarContainerName=IDTR2LambdabarContainerName + "Syst"))
114
115 IDTR2_Reco_V0FinderSyst = acc.popToolsAndMerge(BPHY_Reco_V0FinderCfg(
116 flags, derivation = "IDTR2", suffix = "_Syst",
117 V0ContainerName = IDTR2V0ContainerName + "Syst",
118 KshortContainerName = IDTR2KshortContainerName + "Syst",
119 LambdaContainerName = IDTR2LambdaContainerName + "Syst",
120 LambdabarContainerName = IDTR2LambdabarContainerName + "Syst",
121 CheckVertexContainers = ['PrimaryVertices'],
122 V0FinderTool = V0FinderToolSyst))
123
124 augmentationTools += [IDTR2_Reco_V0FinderSyst]
125
126 for t in augmentationTools:
127 acc.addPublicTool(t)
128
129 # Define the main kernel
130 DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
131 acc.addEventAlgo(DerivationKernel("IDTR2Kernel",
132 AugmentationTools=augmentationTools,
133 SkimmingTools=skimmingTools))
134
135 # ============================
136 # Define contents of the format
137 # =============================
138 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
139 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
140 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
141
142 IDTR2SlimmingHelper = SlimmingHelper(
143 "IDTR2SlimmingHelper",
144 NamesAndTypes=flags.Input.TypedCollections,
145 flags=flags)
146
147 IDTR2SlimmingHelper.AllVariables = [
148 "EventInfo",
149 "PrimaryVertices",
150 "InDetTrackParticles",
151 "TruthParticles",
152 "TruthVertices"
153 ]
154 if flags.Tracking.doLargeD0:
155 IDTR2SlimmingHelper.AllVariables += ["InDetLargeD0TrackParticles"]
156
157 StaticContent = []
158 StaticContent += ["xAOD::VertexContainer#VrtSecInclusive_SecondaryVertices"]
159 StaticContent += ["xAOD::VertexAuxContainer#VrtSecInclusive_SecondaryVerticesAux."]
160
161 for vertexContainer in ["NewVrtSecInclusive_SecondaryVertices_DV",
162 "NewVrtSecInclusive_SecondaryVertices_Material",
163 "NewVrtSecInclusive_SecondaryVertices_Ks",
164 "IDTR2RecoV0Candidates",
165 "IDTR2RecoKshortCandidates",
166 "IDTR2RecoLambdaCandidates",
167 "IDTR2RecoLambdabarCandidates",
168 "IDTR2RecoV0CandidatesSyst",
169 "IDTR2RecoKshortCandidatesSyst",
170 "IDTR2RecoLambdaCandidatesSyst",
171 "IDTR2RecoLambdabarCandidatesSyst"]:
172 StaticContent += ["xAOD::VertexContainer#%s" % vertexContainer]
173 StaticContent += ["xAOD::VertexAuxContainer#%sAux.-vxTrackAtVertex" %
174 vertexContainer]
175
176 IDTR2SlimmingHelper.StaticContent = StaticContent
177
178 IDTR2ItemList = IDTR2SlimmingHelper.GetItemList()
179 acc.merge(OutputStreamCfg(flags, "DAOD_IDTR2",
180 ItemList=IDTR2ItemList,
181 AcceptAlgs=["IDTR2Kernel"]))
182 acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_IDTR2",
183 AcceptAlgs=["IDTR2Kernel"],
184 createMetadata=[MetadataCategory.CutFlowMetaData]))
185
186 return acc

Variable Documentation

◆ streamName

str python.IDTR2.streamName = "StreamDAOD_IDTR2"

Definition at line 11 of file IDTR2.py.