334def IDTIDECfg(flags):
335 """Main config fragment for IDTIDE"""
336 acc = ComponentAccumulator()
337
338
339 if flags.Detector.GeometryID:
340 acc.merge(IDTIDEKernelCfg(flags, StreamName='StreamDAOD_IDTIDE'))
341 if flags.Detector.GeometryITk:
342 acc.merge(ITkTIDEKernelCfg(flags, StreamName='StreamDAOD_IDTIDE'))
343
344
345
346
347 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
348 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
349 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
351 "IDTIDESlimmingHelper",
352 NamesAndTypes=flags.Input.TypedCollections,
353 flags=flags)
354
355 AllVariables = []
356 StaticContent = []
357 SmartCollections = []
358 ExtraVariables = []
359
360 IDTIDESlimmingHelper.AppendToDictionary.update({
361 "EventInfo": "xAOD::EventInfo", "EventInfoAux": "xAOD::EventAuxInfo",
362 "CombinedMuonTrackParticles": "xAOD::TrackParticleContainer", "CombinedMuonTrackParticlesAux": "xAOD::TrackParticleAuxContainer",
363 "Muons": "xAOD::MuonContainer", "MuonsAux": "xAOD::MuonAuxContainer",
364 "Electrons": "xAOD::ElectronContainer",
365 "ElectronsAux": "xAOD::ElectronAuxContainer",
366 "Photons": "xAOD::PhotonContainer",
367 "PhotonsAux": "xAOD::PhotonAuxContainer",
368 "JetETMissNeutralParticleFlowObjects": "xAOD::FlowElementContainer",
369 "JetETMissNeutralParticleFlowObjectsAux": "xAOD::FlowElementAuxContainer",
370 "JetETMissChargedParticleFlowObjects": "xAOD::FlowElementContainer",
371 "JetETMissChargedParticleFlowObjectsAux": "xAOD::FlowElementAuxContainer",
372 "TauJets": "xAOD::TauJetContainer",
373 "TauJetsAux": "xAOD::TauJetAuxContainer",
374 "InDetTrackParticles": "xAOD::TrackParticleContainer",
375 "InDetTrackParticlesAux": "xAOD::TrackParticleAuxContainer",
376 "InDetLargeD0TrackParticles": "xAOD::TrackParticleContainer",
377 "InDetLargeD0TrackParticlesAux": "xAOD::TrackParticleAuxContainer",
378 "Kt4EMTopoOriginEventShape": "xAOD::EventShape",
379 "Kt4EMTopoOriginEventShapeAux": "xAOD::EventShapeAuxInfo",
380 "Kt4LCTopoOriginEventShape": "xAOD::EventShape",
381 "Kt4LCTopoOriginEventShapeAux": "xAOD::EventShapeAuxInfo",
382 "NeutralParticleFlowIsoCentralEventShape": "xAOD::EventShape",
383 "NeutralParticleFlowIsoCentralEventShapeAux": "xAOD::EventShapeAuxInfo",
384 "NeutralParticleFlowIsoForwardEventShape": "xAOD::EventShape",
385 "NeutralParticleFlowIsoForwardEventShapeAux": "xAOD::EventShapeAuxInfo",
386 "TopoClusterIsoCentralEventShape": "xAOD::EventShape",
387 "TopoClusterIsoCentralEventShapeAux": "xAOD::EventShapeAuxInfo",
388 "TopoClusterIsoForwardEventShape": "xAOD::EventShape",
389 "TopoClusterIsoForwardEventShapeAux": "xAOD::EventShapeAuxInfo"}
390 )
391 if flags.Detector.GeometryID:
392 IDTIDESlimmingHelper.AppendToDictionary.update({
393 "PixelClusters": "xAOD::TrackMeasurementValidationContainer",
394 "PixelClustersAux": "xAOD::TrackMeasurementValidationAuxContainer",
395 "SCT_Clusters": "xAOD::TrackMeasurementValidationContainer",
396 "SCT_ClustersAux": "xAOD::TrackMeasurementValidationAuxContainer"
397 })
398 if flags.Detector.GeometryITk:
399 IDTIDESlimmingHelper.AppendToDictionary.update({
400 "ITkPixelMeasurements": "xAOD::TrackMeasurementValidationContainer",
401 "ITkPixelMeasurementsAux": "xAOD::TrackMeasurementValidationAuxContainer",
402 "ITkStripMeasurements": "xAOD::TrackMeasurementValidationContainer",
403 "ITkStripMeasurementsAux": "xAOD::TrackMeasurementValidationAuxContainer"
404 })
405
406 SmartCollections += ["Muons", "Electrons", "Photons"]
407
408 AllVariables += ["EventInfo",
409 "JetETMissNeutralParticleFlowObjects",
410 "JetETMissChargedParticleFlowObjects",
411 "InDetTrackParticles",
412 "InDetLargeD0TrackParticles",
413 "PixelClusters",
414 "SCT_Clusters",
415 "Kt4EMTopoOriginEventShape",
416 "Kt4LCTopoOriginEventShape",
417 "NeutralParticleFlowIsoCentralEventShape",
418 "NeutralParticleFlowIsoForwardEventShape",
419 "TopoClusterIsoCentralEventShape",
420 "TopoClusterIsoForwardEventShape",
421 ]
422 if flags.Detector.GeometryID:
423 AllVariables += ["PixelClusters", "SCT_Clusters"]
424 if flags.Detector.GeometryITk:
425 AllVariables += ["ITkPixelMeasurements", "ITkStripMeasurements"]
426
427 IDTIDESlimmingHelper.AppendToDictionary.update({
428 "Kt4EMPFlowEventShape": "xAOD::EventShape",
429 "Kt4EMPFlowEventShapeAux": "xAOD::EventShapeAuxInfo",
430 "Kt4EMPFlowNeutEventShape": "xAOD::EventShape",
431 "Kt4EMPFlowNeutEventShapeAux": "xAOD::EventShapeAuxInfo",
432 "PrimaryVertices": "xAOD::VertexContainer",
433 "PrimaryVerticesAux": "xAOD::VertexAuxContainer",
434 "InDetTrackParticlesClusterAssociations": "xAOD::TrackParticleClusterAssociationContainer",
435 "InDetTrackParticlesClusterAssociationsAux": "xAOD::TrackParticleClusterAssociationAuxContainer",
436 "AntiKt4EMTopoJets": "xAOD::JetContainer",
437 "AntiKt4EMTopoJetsAux": "xAOD::JetAuxContainer",
438 "AntiKt4EMPFlowJets": "xAOD::JetContainer",
439 "AntiKt4EMPFlowJetsAux": "xAOD::JetAuxContainer"})
440
441 ExtraVariables += ["TauJets.ABS_ETA_LEAD_TRACK.ClusterTotalEnergy.ClustersMeanCenterLambda.ClustersMeanEMProbability.ClustersMeanFirstEngDens.ClustersMeanPresamplerFrac.ClustersMeanSecondLambda.EMFracFixed.LeadClusterFrac.NNDecayMode.NNDecayModeProb_1p0n.NNDecayModeProb_1p1n.NNDecayModeProb_1pXn.NNDecayModeProb_3p0n.NNDecayModeProb_3pXn.PFOEngRelDiff.PanTau_DecayModeExtended.UpsilonCluster.absipSigLeadTrk.chargedFELinks.etHotShotWin.etHotShotWinOverPtLeadTrk.etaCombined.hadLeakFracFixed.leadTrackProbHT.mCombined.mu.nConversionTracks.nFakeTracks.nModifiedIsolationTracks.nVtxPU.neutralFELinks.passThinning.phiCombined.ptCombined.ptIntermediateAxisEM.rho"]
442 ExtraVariables += ["PrimaryVertices.sumPt2.x.y.z"]
443
444 AllVariables += ["Kt4EMPFlowEventShape", "Kt4EMPFlowNeutEventShape",
445 "InDetTrackParticlesClusterAssociations",
446 "AntiKt4EMTopoJets", "AntiKt4EMPFlowJets"]
447
448 if flags.Detector.EnablePixel:
449 IDTIDESlimmingHelper.AppendToDictionary.update(
450 {'PixelMSOSs': 'xAOD::TrackStateValidationContainer',
451 'PixelMSOSsAux': 'xAOD::TrackStateValidationAuxContainer'})
452 AllVariables += ["PixelMSOSs"]
453
454 if flags.Detector.EnableSCT:
455 IDTIDESlimmingHelper.AppendToDictionary.update(
456 {'SCT_MSOSs': 'xAOD::TrackStateValidationContainer',
457 'SCT_MSOSsAux': 'xAOD::TrackStateValidationAuxContainer'})
458 AllVariables += ["SCT_MSOSs"]
459
460 if flags.Detector.EnableTRT:
461 IDTIDESlimmingHelper.AppendToDictionary.update(
462 {'TRT_MSOSs': 'xAOD::TrackStateValidationContainer',
463 'TRT_MSOSsAux': 'xAOD::TrackStateValidationAuxContainer'})
464 AllVariables += ["TRT_MSOSs"]
465
466 if flags.Detector.EnableITkPixel:
467 IDTIDESlimmingHelper.AppendToDictionary.update(
468 {'ITkPixelMSOSs': 'xAOD::TrackStateValidationContainer',
469 'ITkPixelMSOSsAux': 'xAOD::TrackStateValidationAuxContainer'})
470 AllVariables += ["ITkPixelMSOSs"]
471
472 if flags.Detector.EnableITkStrip:
473 IDTIDESlimmingHelper.AppendToDictionary.update(
474 {'ITkStripMSOSs': 'xAOD::TrackStateValidationContainer',
475 'ITkStripMSOSsAux': 'xAOD::TrackStateValidationAuxContainer'})
476 AllVariables += ["ITkStripMSOSs"]
477
478 if flags.Input.isMC:
479
480 IDTIDESlimmingHelper.AppendToDictionary.update({
481 "AntiKt4TruthJets": "xAOD::JetContainer",
482 "AntiKt4TruthJetsAux": "xAOD::JetAuxContainer",
483 "InTimeAntiKt4TruthJets": "xAOD::JetContainer",
484 "InTimeAntiKt4TruthJetsAux": "xAOD::JetAuxContainer",
485 "JetInputTruthParticles": "xAOD::TruthParticleContainer",
486 "JetInputTruthParticlesNoWZ": "xAOD::TruthParticleContainer",
487 "TruthEvents": "xAOD::TruthEventContainer",
488 "TruthEventsAux": "xAOD::TruthEventAuxContainer",
489 "TruthParticles": "xAOD::TruthParticleContainer",
490 "TruthParticlesAux": "xAOD::TruthParticleAuxContainer",
491 "egammaTruthParticles": "xAOD::TruthParticleContainer",
492 "egammaTruthParticlesAux": "xAOD::TruthParticleAuxContainer",
493 "MuonTruthParticles": "xAOD::TruthParticleContainer",
494 "MuonTruthParticlesAux": "xAOD::TruthParticleAuxContainer",
495 "LRTegammaTruthParticles": "xAOD::TruthParticleContainer",
496 "LRTegammaTruthParticlesAux": "xAOD::TruthParticleAuxContainer",
497 "TruthVertices": "xAOD::TruthVertexContainer",
498 "TruthVerticesAux": "xAOD::TruthVertexAuxContainer"})
499
500 AllVariables += ["AntiKt4TruthJets",
501 "InTimeAntiKt4TruthJets",
502 "JetInputTruthParticles",
503 "JetInputTruthParticlesNoWZ",
504 "TruthEvents",
505 "TruthParticles",
506 "egammaTruthParticles",
507 "MuonTruthParticles",
508 "LRTegammaTruthParticles",
509 "TruthVertices"]
510
511 list_aux = ["BHadronsFinal", "BHadronsInitial", "BQuarksFinal",
512 "CHadronsFinal", "CHadronsInitial", "CQuarksFinal",
513 "HBosons", "Partons", "TQuarksFinal", "TausFinal",
514 "WBosons", "ZBosons"]
515 for item in list_aux:
516 label = "TruthLabel"+item
517 labelAux = label+"Aux"
518 IDTIDESlimmingHelper.AppendToDictionary.update(
519 {label: "xAOD::TruthParticleContainer",
520 labelAux: "xAOD::TruthParticleAuxContainer"})
521 AllVariables += [label]
522
523
524
525 IDTIDESlimmingHelper.IncludeTriggerNavigation = True
526 IDTIDESlimmingHelper.IncludeAdditionalTriggerContent = True
527
528 IDTIDESlimmingHelper.AllVariables = AllVariables
529 IDTIDESlimmingHelper.StaticContent = StaticContent
530 IDTIDESlimmingHelper.SmartCollections = SmartCollections
531 IDTIDESlimmingHelper.ExtraVariables = ExtraVariables
532
533
534 IDTIDEItemList = IDTIDESlimmingHelper.GetItemList()
535 acc.merge(OutputStreamCfg(flags, "DAOD_IDTIDE",
536 ItemList=IDTIDEItemList, AcceptAlgs=["IDTIDEKernel"]))
537 acc.merge(SetupMetaDataForStreamCfg(
538 flags, "DAOD_IDTIDE", AcceptAlgs=["IDTIDEKernel"],
539 createMetadata=[MetadataCategory.CutFlowMetaData, MetadataCategory.TriggerMenuMetaData]))
540
541 return acc