Exemple #1
0
                outputGroup="EXOT12")

if globalflags.DataSource() == 'geant4':
    replaceBuggyAntiKt4TruthWZJets(exot12Seq)

#exot12Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT12Kernel_jet",
#                                                          SkimmingTools = [EXOT12SkimmingTool_jet])

exot12Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT12Kernel", ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT12ContentList import *
EXOT12SlimmingHelper = SlimmingHelper("EXOT12SlimmingHelper")
EXOT12SlimmingHelper.SmartCollections = EXOT12SmartContent
EXOT12SlimmingHelper.ExtraVariables = EXOT12ExtraVariablesContent
EXOT12SlimmingHelper.AllVariables = EXOT12AllVariablesContent
EXOT12SlimmingHelper.StaticContent = EXOT12Content
if globalflags.DataSource() == 'geant4':
    EXOT12SlimmingHelper.ExtraVariables += EXOT12ExtraVariablesContentTruth
    EXOT12SlimmingHelper.AllVariables += EXOT12AllVariablesContentTruth
    EXOT12SlimmingHelper.StaticContent += EXOT12ContentTruth

EXOT12SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT12SlimmingHelper.IncludeMuonTriggerContent = True

EXOT12SlimmingHelper.AppendContentToStream(EXOT12Stream)
Exemple #2
0
#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================

from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot7Seq
exot7Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT7Kernel_skim",
    AugmentationTools=[EXOT7DeltaRTool],
    SkimmingTools=[EXOT7StringSkimmingTool])
exot7Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT7Kernel", ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT7ContentList import *
EXOT7SlimmingHelper = SlimmingHelper("EXOT7SlimmingHelper")
EXOT7SlimmingHelper.StaticContent = EXOT7Content

EXOT7SlimmingHelper.SmartCollections = EXOT7SmartCollections
EXOT7SlimmingHelper.ExtraVariables = EXOT7ExtraVariables
EXOT7SlimmingHelper.AllVariables = EXOT7AllVariables
#EXOT7SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
EXOT7SlimmingHelper.IncludeJetTriggerContent = True
EXOT7SlimmingHelper.IncludeBJetTriggerContent = True

EXOT7SlimmingHelper.AppendContentToStream(EXOT7Stream)
def setup(TOPQname, stream):
    DFisMC = (globalflags.DataSource() == 'geant4')

    TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper")

    #=====================================================
    # ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY
    #=====================================================
    TOPQSlimmingHelper.AppendToDictionary = {
        "BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer",
        "BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer"
    }

    #================================
    # SMART SLIMMING
    #================================
    TOPQSlimmingHelper.SmartCollections = []
    TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections

    print "TOPQSlimmingHelper.SmartCollections: ", TOPQSlimmingHelper.SmartCollections

    #=================================================================
    # EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent
    #=================================================================
    TOPQSlimmingHelper.ExtraVariables = []
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus

    if DFisMC:
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth

    print "TOPQSlimmingHelper.ExtraVariables: ", TOPQSlimmingHelper.ExtraVariables

    #================================
    # EXTRA COLLECTIONS - user added
    #================================
    TOPQSlimmingHelper.AllVariables = []
    TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard
    TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger

    if DFisMC:
        TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth

    print "TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables

    #================================
    # CREATED ON-THE-FLY COLLECTIONS
    #================================
    TOPQSlimmingHelper.StaticContent = []
    TOPQSlimmingHelper.StaticContent += TOPQStaticContent

    if DFisMC:
        TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth

    print "TOPQSlimmingHelper.StaticContent: ", TOPQSlimmingHelper.StaticContent

    #================================
    # TRIGGER CONTENT
    #================================
    TOPQSlimmingHelper.IncludeMuonTriggerContent = True
    TOPQSlimmingHelper.IncludeEGammaTriggerContent = True
    TOPQSlimmingHelper.IncludeJetTriggerContent = True
    TOPQSlimmingHelper.IncludeTauTriggerContent = True
    TOPQSlimmingHelper.IncludeEtMissTriggerContent = True
    #TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True
    #TOPQSlimmingHelper.IncludeBJetTriggerContent         = True
    #TOPQSlimmingHelper.IncludeBPhysTriggerContent        = True
    #TOPQSlimmingHelper.IncludeMinBiasTriggerContent      = True

    #====================================
    # ADD AntiKT4EMPFlowJets MET OUTPUT
    #====================================
    # needs to be after where AllVaraibles[] and StaticContent[] are created
    # adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX
    # uses DerivationFrameworkJetEtMiss.METCommon.py
    addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"])

    #================================
    # THIS NEEDS TO BE THE LAST LINE
    #================================
    TOPQSlimmingHelper.AppendContentToStream(stream)
Exemple #4
0
DerivationFrameworkJob += exot0Seq
exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel_skim", SkimmingTools = [EXOT0SkimmingTool])
exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel", ThinningTools = thinningTools)

#====================================================================
# SET UP STREAM   
#====================================================================
streamName = derivationFlags.WriteDAOD_EXOT0Stream.StreamName
fileName   = buildFileName( derivationFlags.WriteDAOD_EXOT0Stream )
EXOT0Stream = MSMgr.NewPoolRootStream( streamName, fileName )
EXOT0Stream.AcceptAlgs(["EXOT0Kernel"])
# Thinning 
from AthenaServices.Configurables import ThinningSvc, createThinningSvc
augStream = MSMgr.GetStream( streamName )
evtStream = augStream.GetEventStream()
svcMgr += createThinningSvc( svcName="EXOT0ThinningSvc", outStreams=[evtStream] )

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT0ContentList import *
EXOT0SlimmingHelper = SlimmingHelper("EXOT0SlimmingHelper")
EXOT0SlimmingHelper.StaticContent = EXOT0Content
EXOT0SlimmingHelper.AllVariables = EXOT0AllVariables
EXOT0SlimmingHelper.SmartCollections = EXOT0SmartCollections
EXOT0SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT0SlimmingHelper.IncludeMuonTriggerContent = True
EXOT0SlimmingHelper.AppendContentToStream(EXOT0Stream)

Exemple #5
0
from AthenaCommon.GlobalFlags import globalflags
if globalflags.DataSource() == 'geant4':
    ToolSvc += EXOT13MCThinningTool
    thinningTools.append(EXOT13MCThinningTool)

#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================

from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot13Seq
exot13Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT13Kernel_skim", SkimmingTools=[EXOT13StringSkimmingTool])
exot13Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT13Kernel", ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
# Currently using same list as EXOT13
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT13ContentList import *
EXOT13SlimmingHelper = SlimmingHelper("EXOT13SlimmingHelper")
EXOT13SlimmingHelper.SmartCollections = EXOT13SmartContent
EXOT13SlimmingHelper.AllVariables = EXOT13AllVariablesContent
EXOT13SlimmingHelper.StaticContent = EXOT13UnslimmedContent
EXOT13SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT13SlimmingHelper.IncludeMuonTriggerContent = True
EXOT13SlimmingHelper.IncludeJetTriggerContent = True
EXOT13SlimmingHelper.AppendContentToStream(EXOT13Stream)
Exemple #6
0
    "xAOD::VertexAuxContainer#SoftBVrtClusterTool_Medium_VerticesAux." +
    excludedVertexAuxData
]
StaticContent += ["xAOD::VertexContainer#SoftBVrtClusterTool_Loose_Vertices"]
StaticContent += [
    "xAOD::VertexAuxContainer#SoftBVrtClusterTool_Loose_VerticesAux." +
    excludedVertexAuxData
]
StaticContent += [
    "xAOD::VertexContainer#VrtSecInclusive_SoftBtagCandidateVertices"
]
StaticContent += [
    "xAOD::VertexAuxContainer#VrtSecInclusive_SoftBtagCandidateVerticesAux."
]

FTAG4SlimmingHelper.StaticContent = StaticContent
#----------------------------------------------------------------------

addJetOutputs(FTAG4SlimmingHelper, ["FTAG4"])

FTAG4SlimmingHelper.IncludeMuonTriggerContent = True
FTAG4SlimmingHelper.IncludeEGammaTriggerContent = True
FTAG4SlimmingHelper.IncludeJetTriggerContent = True
FTAG4SlimmingHelper.IncludeEtMissTriggerContent = False
FTAG4SlimmingHelper.IncludeBJetTriggerContent = True

#FTAG4 TrigNav Thinning
FTAG4ThinningHelper = ThinningHelper("FTAG4ThinningHelper")
FTAG4ThinningHelper.TriggerChains = 'HLT_e[2-9][0-9]_.*|HLT_e[1-9][0-9][0-9]_.*|HLT_mu[2-9][0-9]_.*|HLT_mu[1-9][0-9][0-9]_.*|HLT_mu[2-9][0-9]'
FTAG4ThinningHelper.AppendToStream(FTAG4Stream)
Exemple #7
0
                                       "DiTauJetsLowPt",
                                       "AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
                                       "AntiKtVR30Rmax4Rmin02TrackJets_BTagging201903",
                                       "BTagging_AntiKtVR30Rmax4Rmin02Track_201903"
                                      ]

excludedVertexAuxData = "-vxTrackAtVertex.-MvfFitInfo.-isInitialized.-VTAV"
StaticContent = []
StaticContent += ["xAOD::VertexContainer#SoftBVrtClusterTool_Tight_Vertices"]
StaticContent += ["xAOD::VertexAuxContainer#SoftBVrtClusterTool_Tight_VerticesAux." + excludedVertexAuxData]
StaticContent += ["xAOD::VertexContainer#SoftBVrtClusterTool_Medium_Vertices"]
StaticContent += ["xAOD::VertexAuxContainer#SoftBVrtClusterTool_Medium_VerticesAux." + excludedVertexAuxData]
StaticContent += ["xAOD::VertexContainer#SoftBVrtClusterTool_Loose_Vertices"]
StaticContent += ["xAOD::VertexAuxContainer#SoftBVrtClusterTool_Loose_VerticesAux." + excludedVertexAuxData]

PHYS2SlimmingHelper.StaticContent = StaticContent

# Trigger content
PHYS2SlimmingHelper.IncludeTriggerNavigation = False
PHYS2SlimmingHelper.IncludeJetTriggerContent = False
PHYS2SlimmingHelper.IncludeMuonTriggerContent = False
PHYS2SlimmingHelper.IncludeEGammaTriggerContent = False
PHYS2SlimmingHelper.IncludeJetTauEtMissTriggerContent = False
PHYS2SlimmingHelper.IncludeTauTriggerContent = False
PHYS2SlimmingHelper.IncludeEtMissTriggerContent = False
PHYS2SlimmingHelper.IncludeBJetTriggerContent = False
PHYS2SlimmingHelper.IncludeBPhysTriggerContent = False
PHYS2SlimmingHelper.IncludeMinBiasTriggerContent = False

# Add the jet containers to the stream (defined in JetCommon if import needed)
#addJetOutputs(PHYS2SlimmingHelper,["PHYS2"])
HIGG6D1SlimmingHelper.ExtraVariables += [
    "Electrons.DFCommonElectronsLHLoose.DFCommonElectronsLHMedium.DFCommonElectronsLHTight.DFCommonElectronsML.author.OQ.charge.LHLoose.LHMedium.LHTight.LHValue"
]
HIGG6D1SlimmingHelper.ExtraVariables += [
    "Muons.DFCommonGoodMuons", "CombinedMuonTrackParticles.d0.z0.vz",
    "InDetTrackParticles.numberOfTRTHits.numberOfTRTOutliers"
]
HIGG6D1SlimmingHelper.ExtraVariables += ["PrimaryVertices.x.y.z.vertexType"]

if is_MC:
    HIGG6D1SlimmingHelper.StaticContent = [
        "xAOD::TruthParticleContainer#TruthMuons",
        "xAOD::TruthParticleAuxContainer#TruthMuonsAux.",
        "xAOD::TruthParticleContainer#TruthElectrons",
        "xAOD::TruthParticleAuxContainer#TruthElectronsAux.",
        "xAOD::TruthParticleContainer#TruthPhotons",
        "xAOD::TruthParticleAuxContainer#TruthPhotonsAux.",
        "xAOD::TruthParticleContainer#TruthNeutrinos",
        "xAOD::TruthParticleAuxContainer#TruthNeutrinosAux.",
        "xAOD::TruthParticleContainer#TruthTaus",
        "xAOD::TruthParticleAuxContainer#TruthTausAux."
    ]
    HIGG6D1SlimmingHelper.AllVariables = [
        "TruthParticles", "TruthEvents", "MET_Truth", "METMap_Truth",
        "TruthVertices", "AntiKt4TruthJets"
    ]
    HIGG6D1SlimmingHelper.ExtraVariables += [
        "AntiKt4LCTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T"
    ]
    HIGG6D1SlimmingHelper.ExtraVariables += [
        "AntiKt4EMTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T"
    ]
Exemple #9
0
    "TauJets"
]
#                                                    "MET_Reference_AntiKt4EMTopo",
#                                                    "MET_Reference_AntiKt4LCTopo",
#                                                    "AntiKt4EMTopoJets",
#                                                    "AntiKt4LCTopoJets",
#                                                    "BTagging_AntiKt4EMTopo",
#                                                    "BTagging_AntiKt4LCTopo",
#                                                    "InDetTrackParticles",
#                                                    "PrimaryVertices"]

if DerivationFrameworkIsMonteCarlo:
    TAUP2SlimmingHelper.StaticContent = [
        "xAOD::TruthParticleContainer#TruthElectrons",
        "xAOD::TruthParticleAuxContainer#TruthElectronsAux.",
        "xAOD::TruthParticleContainer#TruthMuons",
        "xAOD::TruthParticleAuxContainer#TruthMuonsAux.",
        "xAOD::TruthParticleContainer#TruthTaus",
        "xAOD::TruthParticleAuxContainer#TruthTausAux."
    ]
#                                                    "xAOD::TruthParticleContainer#TruthPhotons",
#                                                    "xAOD::TruthParticleAuxContainer#TruthPhotonsAux.",
#                                                    "xAOD::TruthParticleContainer#TruthNeutrinos",
#                                                    "xAOD::TruthParticleAuxContainer#TruthNeutrinosAux."]

# TAUP2SlimmingHelper.IncludeMuonTriggerContent    = True
# TAUP2SlimmingHelper.IncludeTauTriggerContent     = True
# TAUP2SlimmingHelper.IncludeEGammaTriggerContent  = True
# TAUP2SlimmingHelper.IncludeEtMissTriggerContent  = True
# TAUP2SlimmingHelper.IncludeJetTriggerContent     = True
# TAUP2SlimmingHelper.IncludeBJetTriggerContent    = True
Exemple #10
0
HIGG2D5SlimmingHelper.SmartCollections = ["Electrons",
                                          "Photons",
                                          "Muons",
                                          "TauJets",
                                          "MET_Reference_AntiKt4EMTopo",
                                          "MET_Reference_AntiKt4LCTopo",
                                          "AntiKt4EMTopoJets",
                                          "AntiKt4LCTopoJets",
                                          "BTagging_AntiKt4EMTopo",
                                          "BTagging_AntiKt4LCTopo",
                                          "InDetTrackParticles",
                                          "PrimaryVertices"]

HIGG2D5SlimmingHelper.ExtraVariables = HIGG2D5ExtraContent
HIGG2D5SlimmingHelper.AllVariables = HIGG2D5ExtraContainers
if globalflags.DataSource()=='geant4':
    HIGG2D5SlimmingHelper.ExtraVariables += HIGG2D5ExtraContentTruth
    HIGG2D5SlimmingHelper.AllVariables += HIGG2D5ExtraContainersTruth

# For J/psi vertex augmentation
HIGG2D5SlimmingHelper.StaticContent = HIGG2D5StaticContent

# Add MET_RefFinalFix
addMETOutputs(HIGG2D5SlimmingHelper,["AntiKt4LCTopo","Track"])

HIGG2D5SlimmingHelper.IncludeMuonTriggerContent = True
HIGG2D5SlimmingHelper.IncludeBPhysTriggerContent = True

HIGG2D5SlimmingHelper.AppendContentToStream(HIGG2D5Stream)
Exemple #11
0
    'TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink',
]
EXOT5SlimmingHelper.AllVariables = [
    'AntiKt4EMTopoJets',
    'AntiKt4TruthJets',
    'MET_Truth',
    'MET_Track',
    'MET_LocHadTopo',
    'TruthEvents',
    'TruthParticles',
    'MuonSegments',
]
if DerivationFrameworkIsMonteCarlo:
    EXOT5SlimmingHelper.StaticContent = [
        'xAOD::TruthParticleContainer#EXOT5TruthMuons',
        'xAOD::TruthParticleAuxContainer#EXOT5TruthMuonsAux.',
        'xAOD::TruthParticleContainer#EXOT5TruthElectrons',
        'xAOD::TruthParticleAuxContainer#EXOT5TruthElectronsAux.',
        'xAOD::TruthParticleContainer#EXOT5TruthNeutrinos',
        'xAOD::TruthParticleAuxContainer#EXOT5TruthNeutrinosAux.',
        'xAOD::TruthParticleContainer#TruthTaus',
        'xAOD::TruthParticleAuxContainer#TruthTausAux.',
    ]
EXOT5SlimmingHelper.UserContent = []
EXOT5SlimmingHelper.IncludeMuonTriggerContent = True
EXOT5SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT5SlimmingHelper.IncludeJetTriggerContent = True
EXOT5SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT5SlimmingHelper.IncludeEtMissTriggerContent = True
EXOT5SlimmingHelper.AppendContentToStream(EXOT5Stream)
Exemple #12
0
def setup(TOPQname, stream):
    DFisMC = (globalflags.DataSource() == 'geant4')

    TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper")

    #=====================================================
    # ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY
    #=====================================================
    TOPQSlimmingHelper.AppendToDictionary = {
        "BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer",
        "BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer"
    }

    #================================
    # SMART SLIMMING
    #================================
    TOPQSlimmingHelper.SmartCollections = []
    TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections

    print("TOPQSlimmingHelper.SmartCollections: ",
          TOPQSlimmingHelper.SmartCollections)

    #=================================================================
    # EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent
    #=================================================================
    TOPQSlimmingHelper.ExtraVariables = []
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus
    TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTrackJets
    TOPQSlimmingHelper.ExtraVariables += Config.GetExtraPromptVariablesForDxAOD(
    )
    #  TOPQSlimmingHelper.ExtraVariables += ["CaloCalTopoClusters.calE.calEta.calPhi.calM.rawM.rawE.rawEta.rawPhi.e_sampl.eta_sampl.etaCalo.phiCalo"]
    TOPQSlimmingHelper.ExtraVariables += [
        "CaloCalTopoClusters.calPt.calEta.calPhi.calM.calE.CENTER_MAG"
    ]

    if DFisMC:
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsBTag_HLT_Container
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_EF_Container
        TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_Split_Container

    print("TOPQSlimmingHelper.ExtraVariables: ",
          TOPQSlimmingHelper.ExtraVariables)

    #================================
    # EXTRA COLLECTIONS - user added
    #================================
    TOPQSlimmingHelper.AllVariables = []
    TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard
    TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger

    if DFisMC:
        TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth

    print("TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables)

    #================================
    # CREATED ON-THE-FLY COLLECTIONS
    #================================
    TOPQSlimmingHelper.StaticContent = []
    TOPQSlimmingHelper.StaticContent += TOPQStaticContent

    if DFisMC:
        TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth

    print("TOPQSlimmingHelper.StaticContent: ",
          TOPQSlimmingHelper.StaticContent)

    #================================
    # TRIGGER CONTENT
    #================================
    TOPQSlimmingHelper.IncludeMuonTriggerContent = True
    TOPQSlimmingHelper.IncludeEGammaTriggerContent = True
    TOPQSlimmingHelper.IncludeJetTriggerContent = True
    TOPQSlimmingHelper.IncludeTauTriggerContent = True
    TOPQSlimmingHelper.IncludeEtMissTriggerContent = True
    #TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True
    #TOPQSlimmingHelper.IncludeBJetTriggerContent         = True
    #TOPQSlimmingHelper.IncludeBPhysTriggerContent        = True
    #TOPQSlimmingHelper.IncludeMinBiasTriggerContent      = True

    #====================================
    # ADD AntiKT4EMPFlowJets MET OUTPUT
    #====================================
    # needs to be after where AllVaraibles[] and StaticContent[] are created
    # adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX
    # uses DerivationFrameworkJetEtMiss.METCommon.py
    addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"])

    #====================================
    # ADD CUSTOM JET OUTPUT (new rel 21 method)
    #====================================
    addJetOutputs(
        TOPQSlimmingHelper,
        ["SmallR", "LargeR", TOPQname],
        [],  # smart list
        [
            "AntiKt3PV0TrackJets",
            "AntiKt4PV0TrackJets",
            "AntiKt4LCTopoJets",
            "AntiKt10LCTopoJets",
            "AntiKt10TruthWZJets",
        ]  # veto list
    )

    #================================
    # THIS NEEDS TO BE THE LAST LINE
    #================================
    TOPQSlimmingHelper.AppendContentToStream(stream)
Exemple #13
0
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__xAODStringSkimmingTool
EXOT9SkimmingTool = DerivationFramework__xAODStringSkimmingTool(
    name="EXOT9SkimmingTool1", expression=triggerStrategy)
ToolSvc += EXOT9SkimmingTool
printfunc(EXOT9SkimmingTool)

#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================

from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot9Seq
exot9Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT9Kernel_skim", SkimmingTools=[EXOT9SkimmingTool])
exot9Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT9Kernel", ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT9ContentList import *
EXOT9SlimmingHelper = SlimmingHelper("EXOT9SlimmingHelper")
EXOT9SlimmingHelper.StaticContent = EXOT9Content
EXOT9SlimmingHelper.AllVariables = EXOT9AllVariables
EXOT9SlimmingHelper.SmartCollections = EXOT9SmartCollections
EXOT9SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT9SlimmingHelper.IncludeMuonTriggerContent = True
EXOT9SlimmingHelper.AppendContentToStream(EXOT9Stream)
Exemple #14
0
skimmingTools.append(EXOT1SkimmingTool)

from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__FilterCombinationOR
EXOT1ORSkimmingTool = DerivationFramework__FilterCombinationOR(
    name="EXOT1ORSkimmingTool", FilterList=skimmingTools)
ToolSvc += EXOT1ORSkimmingTool
printfunc(EXOT1SkimmingTool)

#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================

from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot1Seq
exot1Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT1Kernel_skim", SkimmingTools=[EXOT1ORSkimmingTool])
exot1Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT1Kernel", ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
# Currently using same list as EXOT1
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT1ContentList import *
EXOT1SlimmingHelper = SlimmingHelper("EXOT1SlimmingHelper")
EXOT1SlimmingHelper.StaticContent = EXOT1Content
EXOT1SlimmingHelper.SmartCollections = EXOT1SmartContent
EXOT1SlimmingHelper.AllVariables = EXOT1AllVariablesContent
EXOT1SlimmingHelper.AppendContentToStream(EXOT1Stream)
Exemple #15
0
    EXOT14PhotonTPThinningTool, EXOT14PhotonCCThinningTool,
    EXOT14ElectronCCThinningTool
]
if globalflags.DataSource() == 'geant4':
    EXOT14ThinningTools.append(EXOT14MCThinningTool)

from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot14Seq
exot14Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT14Kernel_skim", SkimmingTools=[EXOT14SkimmingTool])
exot14Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT14Kernel", ThinningTools=EXOT14ThinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT14ContentList import *
EXOT14SlimmingHelper = SlimmingHelper("EXOT14SlimmingHelper")
EXOT14SlimmingHelper.StaticContent = EXOT14Content

EXOT14SlimmingHelper.SmartCollections = EXOT14SmartContent
EXOT14SlimmingHelper.AllVariables = EXOT14AllVariablesContent
EXOT14SlimmingHelper.UserContent = EXOT14Content
EXOT14SlimmingHelper.ExtraVariables = EXOT14ExtraVariablesContent

EXOT14SlimmingHelper.IncludeJetTriggerContent = True
EXOT14SlimmingHelper.IncludeEtMissTriggerContent = True

EXOT14SlimmingHelper.AppendContentToStream(EXOT14Stream)
Exemple #16
0
exot6Seq += CfgMgr.DerivationFramework__DerivationKernel(
    "EXOT6Kernel",
    ThinningTools=[
        EXOT6TPThinningTool, EXOT6MuonTPThinningTool,
        EXOT6ElectronTPThinningTool, EXOT6PhotonTPThinningTool,
        EXOT6PhotonCCThinningTool
    ])

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT6ContentList import EXOT6SmartContent, EXOT6AllVariablesContent, EXOT6ExtraVariables, EXOT6UnslimmedContent

EXOT6SlimmingHelper = SlimmingHelper("EXOT6SlimmingHelper")

# Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis
# /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38
EXOT6SlimmingHelper.SmartCollections = EXOT6SmartContent

# Keep all variables for containers which we don't want to smart slim and were
# not created in the derivation
EXOT6SlimmingHelper.AllVariables = EXOT6AllVariablesContent

# Adding extra missing variables from the smart slimming
EXOT6SlimmingHelper.ExtraVariables = EXOT6ExtraVariables

EXOT6SlimmingHelper.StaticContent = EXOT6UnslimmedContent

EXOT6SlimmingHelper.AppendContentToStream(EXOT6Stream)
Exemple #17
0
    "EXOT3Kernel",
    #									ThinningTools = [EXOT3TPThinningTool,EXOT3MuonTPThinningTool,EXOT3ElectronTPThinningTool, EXOT3PhotonTPThinningTool]
    ThinningTools=thinningTools)

#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT3ContentList import *

EXOT3SlimmingHelper = SlimmingHelper("EXOT3SlimmingHelper")

# Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis
# /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38
EXOT3SlimmingHelper.SmartCollections = EXOT3SmartContent
EXOT3SlimmingHelper.ExtraVariables = EXOT3ExtraVariables

# Keep all variables for containers which we don't want to smart slim and were
# not created in the derivation
EXOT3SlimmingHelper.AllVariables = EXOT3AllVariablesContent

# Add jet collections created by derivation job
EXOT3SlimmingHelper.StaticContent = EXOT3StaticContent

# (Dont) Add jet triger content
#EXOT3SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
EXOT3SlimmingHelper.IncludeJetTriggerContent = True
EXOT3SlimmingHelper.IncludeEGammaTriggerContent = True

EXOT3SlimmingHelper.AppendContentToStream(EXOT3Stream)
Exemple #18
0
EXOT8SlimmingHelper.AllVariables = [
    "TruthParticles",
    "TruthEvents",
    "TruthVertices",
    "AntiKt10LCTopoJets",
    "AntiKt2PV0TrackJets",
    "AntiKt3PV0TrackJets",
    "AntiKt4PV0TrackJets",
    #"BTagging_AntiKt4TruthWZ",    #removed in new tags of MC15
    "AntiKt4TruthJets",
    "AntiKt10TruthJets",
    "CombinedMuonTrackParticles",
    "ExtrapolatedMuonTrackParticles",
]

EXOT8SlimmingHelper.StaticContent = [
    "xAOD::JetContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
    "xAOD::JetAuxContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20JetsAux.",
]
if globalflags.DataSource() == "geant4":
    EXOT8SlimmingHelper.StaticContent += [
        "xAOD::JetContainer#AntiKt10TruthTrimmedPtFrac5SmallR20Jets",
        "xAOD::JetAuxContainer#AntiKt10TruthTrimmedPtFrac5SmallR20JetsAux.",
    ]

EXOT8SlimmingHelper.IncludeJetTriggerContent = True
EXOT8SlimmingHelper.IncludeBJetTriggerContent = True
EXOT8SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT8SlimmingHelper.IncludeMuonTriggerContent = True
EXOT8SlimmingHelper.AppendContentToStream(EXOT8Stream)
Exemple #19
0
#addStandardJets("AntiKt", 0.4, "PV0Track", 2000, mods="track_ungroomed", algseq=bphy5Seq, outputGroup="MUON2")

# Added by ASC
# Truth information for MC only
if isSimulation:
    AllVariables += [
        "TruthEvents", "TruthParticles", "TruthVertices", "MuonTruthParticles",
        "egammaTruthParticles"
    ]
    AllVariables += ["AntiKt4TruthJets", "AntiKt4TruthWZJets"]
    #    addStandardJets("AntiKt", 0.4, "Truth", 5000, mods="truth_ungroomed", algseq=bphy5Seq, outputGroup="MUON2")
    #    addStandardJets("AntiKt", 0.4, "TruthWZ", 5000, mods="truth_ungroomed", algseq=bphy5Seq, outputGroup="MUON2")
    tagJetCollections += ["AntiKt4TruthJets", "AntiKt4TruthWZJets"]

from DerivationFrameworkJetEtMiss.ExtendedJetCommon import replaceAODReducedJets
replaceAODReducedJets(tagJetCollections, bphy5Seq, "MUON2")

AllVariables = list(set(AllVariables))  # remove duplicates

ExtraVariables = [
    "InDetTrackParticles.numberOfTRTHits.numberOfTRTOutliers.numberOfTRTHoles.numberOfTRTHighThresholdHits.numberOfTRTHighThresholdHitsTotal.numberOfTRTHighThresholdOutliers.numberOfTRTDeadStraws.numberOfTRTTubeHits.numberOfTRTXenonHits.TRTTrackOccupancy.numberOfTRTSharedHits.vx.vy.vz"
]

MUON2SlimmingHelper.AllVariables = AllVariables
MUON2SlimmingHelper.ExtraVariables = ExtraVariables
MUON2SlimmingHelper.StaticContent = StaticContent
MUON2SlimmingHelper.SmartCollections = SmartVar

MUON2SlimmingHelper.AppendContentToStream(MUON2Stream)