13def BPHY4Cfg(flags):
14 from DerivationFrameworkBPhys.commonBPHYMethodsCfg import (BPHY_V0ToolCfg, BPHY_InDetDetailedTrackSelectorToolCfg, BPHY_TrkVKalVrtFitterCfg)
15 from JpsiUpsilonTools.JpsiUpsilonToolsConfig import PrimaryVertexRefittingToolCfg
16 acc = ComponentAccumulator()
17 V0Tools = acc.popToolsAndMerge(BPHY_V0ToolCfg(flags, BPHYDerivationName))
18 vkalvrt = acc.popToolsAndMerge(BPHY_TrkVKalVrtFitterCfg(flags, BPHYDerivationName))
19 acc.addPublicTool(vkalvrt)
20 acc.addPublicTool(V0Tools)
21 trackselect = acc.popToolsAndMerge(BPHY_InDetDetailedTrackSelectorToolCfg(flags, BPHYDerivationName))
22 acc.addPublicTool(trackselect)
23
24 BPHY4FourMuonTool = CompFactory.DerivationFramework.FourMuonTool(
25 name = "BPHY4FourMuonTool",
26 ptCut = 2500.0,
27 etaCut = 2.5,
28 muonCollectionKey = "Muons",
29 TrackParticleCollection = "InDetTrackParticles",
30 V0VertexFitterTool = None,
31 useV0Fitter = False,
32 TrkVertexFitterTool = vkalvrt,
33 TrackSelectorTool = trackselect)
34 BPHY4_Reco_4mu = CompFactory.DerivationFramework.Reco_4mu(
35 name = "BPHY4_Reco_4mu",
36 FourMuonTool = BPHY4FourMuonTool,
37 V0Tools = V0Tools,
38 PVRefitter = acc.popToolsAndMerge(PrimaryVertexRefittingToolCfg(flags)),
39 PairContainerName = "BPHY4Pairs",
40 QuadrupletContainerName = "BPHY4Quads",
41 PVContainerName = "PrimaryVertices",
42 RefPVContainerName = "BPHY4RefittedPrimaryVertices",
43 RefitPV = True,
44 MaxPVrefit = 100000,
45 DoVertexType = 7)
46
47 BPHY4MuonTPThinningTool = CompFactory.DerivationFramework.MuonTrackParticleThinning(name = "BPHY4MuonTPThinningTool",
48 MuonKey = "Muons",
49 StreamName = streamName,
50 InDetTrackParticlesKey = "InDetTrackParticles")
51 BPHY4ElectronTPThinningTool = CompFactory.DerivationFramework.EgammaTrackParticleThinning(name = "BPHY4ElectronTPThinningTool",
52 SGKey = "Electrons",
53 GSFTrackParticlesKey = "GSFTrackParticles",
54 StreamName = streamName,
55 InDetTrackParticlesKey = "InDetTrackParticles")
56
57
58 from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
59 xAODStringSkimmingToolCfg)
60 BPHY4_SelectEvent = acc.getPrimaryAndMerge(xAODStringSkimmingToolCfg(
61 flags, name = 'BPHY4_SelectEvent',
62 expression = '(count(Muons.BPHY4MuonIndex>=0)>0)'))
63
64 BPHY4ThinningTools = [BPHY4MuonTPThinningTool, BPHY4ElectronTPThinningTool]
65 BPHY4SlimTools = [BPHY4_SelectEvent]
66 BPHY4AugTools = [BPHY4_Reco_4mu]
67 for t in BPHY4ThinningTools + BPHY4SlimTools + BPHY4AugTools: acc.addPublicTool(t)
68 acc.addEventAlgo(CompFactory.DerivationFramework.DerivationKernel("BPHY4Kernel",
69 AugmentationTools= BPHY4AugTools, SkimmingTools = BPHY4SlimTools, ThinningTools = BPHY4ThinningTools ))
70
71 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
72 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
73 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
74 BPHY4SlimmingHelper =
SlimmingHelper(
"BPHY4SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
75 from DerivationFrameworkBPhys.commonBPHYMethodsCfg import getDefaultAllVariables
76 BPHY4AllVariables = getDefaultAllVariables()
77 BPHY4StaticContent = []
78 BPHY4SmartVariables =[]
79 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
80
81
82 BPHY4SlimmingHelper.IncludeMuonTriggerContent = True
83 BPHY4SlimmingHelper.IncludeBPhysTriggerContent = True
84
85
86 BPHY4AllVariables += ["PrimaryVertices"]
87 BPHY4StaticContent += ["xAOD::VertexContainer#BPHY4RefittedPrimaryVertices"]
88 BPHY4StaticContent += ["xAOD::VertexAuxContainer#BPHY4RefittedPrimaryVerticesAux."]
89
90
91 BPHY4AllVariables += ["InDetTrackParticles"]
92 BPHY4SmartVariables += ["InDetTrackParticles"]
93
94
97 BPHY4AllVariables += ["CombinedMuonTrackParticles"]
98 BPHY4AllVariables += ["ExtrapolatedMuonTrackParticles"]
99
100
101 BPHY4SmartVariables += ["Muons"]
102
103
104 BPHY4SmartVariables += ["Electrons"]
105
106
107 BPHY4StaticContent += ["xAOD::VertexContainer#%s" % BPHY4_Reco_4mu.PairContainerName]
108 BPHY4StaticContent += ["xAOD::VertexContainer#%s" % BPHY4_Reco_4mu.QuadrupletContainerName]
109
110
111 BPHY4StaticContent += ["xAOD::VertexAuxContainer#%sAux.-vxTrackAtVertex" % BPHY4_Reco_4mu.PairContainerName]
112 BPHY4StaticContent += ["xAOD::VertexAuxContainer#%sAux.-vxTrackAtVertex" % BPHY4_Reco_4mu.QuadrupletContainerName]
113
114 BPHY4SlimmingHelper.AllVariables = BPHY4AllVariables
115 BPHY4SlimmingHelper.StaticContent = BPHY4StaticContent
116 BPHY4SlimmingHelper.SmartCollections = BPHY4SmartVariables
117 BPHY4ItemList = BPHY4SlimmingHelper.GetItemList()
118 acc.merge(OutputStreamCfg(flags, "DAOD_BPHY4", ItemList=BPHY4ItemList, AcceptAlgs=["BPHY4Kernel"]))
119 acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_BPHY4", AcceptAlgs=["BPHY4Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
120 acc.printConfig(withDetails=True, summariseProps=True, onlyComponents = [], printDefaults=True)
121 return acc