def ThinTau(Seq=None, streamName=None, selectionString=None):
if not Seq or not streamName or hasattr(Seq,
"TauThinningKernel" + Seq.name()):
print("Tau thinning will not be scheduled")
return
if not selectionString:
selectionString = "(TauJets.pt >= 15.*GeV) && (TauJets.nTracks>0 && TauJets.nTracks<4)"
print("Adding tau thinning:", selectionString)
from AthenaCommon.AppMgr import ToolSvc
# TauJets thinning
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
TauJetsThinningTool = DerivationFramework__GenericObjectThinning(
name="TauJetsThinningTool" + Seq.name(),
StreamName=streamName,
ContainerName="TauJets",
SelectionString=selectionString)
ToolSvc += TauJetsThinningTool
print(TauJetsThinningTool)
# Only keep tau tracks (and associated ID tracks) classified as charged tracks
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="TauTPThinningTool" + Seq.name(),
StreamName=streamName,
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles",
SelectionString=selectionString,
DoTauTracksThinning=True,
TauTracksKey="TauTracks")
ToolSvc += TauTPThinningTool
print(TauTPThinningTool)
Seq += CfgMgr.DerivationFramework__DerivationKernel(
"TauThinningKernel" + Seq.name(),
ThinningTools=[TauJetsThinningTool, TauTPThinningTool])
def _make_thinning_tools(self, stream):
stream_name = self.stream_name
self.thinning_helper = ThinningHelper(stream_name+"ThinningHelper")
self.thinning_helper.AppendToStream(stream)
self.thinning_tools = []
tool = DerivationFramework__TrackParticleThinning(
stream_name + "TPThinningTool",
StreamName = stream_name,
SelectionString = "InDetTrackParticles.pt > {0}*GeV".format(
self.track_threshold) )
if self.track_sel is not None:
tool.SelectionString += " || InDetTrackParticles.{0.common_prefix}{0.track_sel}".format(self)
self.thinning_tools.append(tool)
self.thinning_tools += [
DerivationFramework__MuonTrackParticleThinning(
stream_name + "MuonTPThinningTool",
StreamName = stream_name,
MuonKey = "Muons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__EgammaTrackParticleThinning(
stream_name + "ElectronTPThinningTool",
StreamName = stream_name,
SGKey = "Electrons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__EgammaTrackParticleThinning(
stream_name + "PhotonTPThinningTool",
StreamName = stream_name,
SGKey = "Photons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__TauTrackParticleThinning(
stream_name + "TauTPThinningTool",
StreamName = stream_name,
TauKey = "TauJets",
InDetTrackParticlesKey = "InDetTrackParticles")
]
global ToolSvc
ToolSvc += self.thinning_tools
thinningTools.append(JETM7ElectronTPThinningTool)
# TrackParticles associated with electrons
JETM7PhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="JETM7PhotonTPThinningTool",
StreamName=streamName,
SGKey="Photons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += JETM7PhotonTPThinningTool
thinningTools.append(JETM7PhotonTPThinningTool)
# TrackParticles associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
JETM7TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="JETM7TauTPThinningTool",
StreamName=streamName,
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += JETM7TauTPThinningTool
thinningTools.append(JETM7TauTPThinningTool)
# Truth particle thinning
doTruthThinning = True
preserveAllDescendants = False
from AthenaCommon.GlobalFlags import globalflags
if doTruthThinning and DerivationFrameworkIsMonteCarlo:
truth_cond_WZH = "((abs(TruthParticles.pdgId) >= 23) && (abs(TruthParticles.pdgId) <= 25))" # W, Z and Higgs
truth_cond_Lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 16))" # Leptons
truth_cond_Quark = "((abs(TruthParticles.pdgId) <= 5 && (TruthParticles.pt > 10000.)) || (abs(TruthParticles.pdgId) == 6))" # Quarks
truth_cond_Gluon = "((abs(TruthParticles.pdgId) == 21) && (TruthParticles.pt > 10000.))" # Gluons
truth_cond_Photon = "((abs(TruthParticles.pdgId) == 22) && (TruthParticles.pt > 10000.) && (TruthParticles.barcode < 200000))" # Photon
# TauJets thinning
tau_thinning_expression = "(AnalysisTauJets.ptFinalCalib >= 13.*GeV) && (AnalysisTauJets.nTracks>=1) && (AnalysisTauJets.nTracks<=3) && (AnalysisTauJets.RNNJetScoreSigTrans>0.01)"
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
PHYSLITETauJetsThinningTool = DerivationFramework__GenericObjectThinning(name = "PHYSLITETauJetsThinningTool",
ThinningService = PHYSLITEThinningHelper.ThinningSvc(),
ContainerName = "AnalysisTauJets",
SelectionString = tau_thinning_expression)
ToolSvc += PHYSLITETauJetsThinningTool
thinningTools.append(PHYSLITETauJetsThinningTool)
# Only keep tau tracks (and associated ID tracks) classified as charged tracks
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
PHYSLITETauTPThinningTool = DerivationFramework__TauTrackParticleThinning(name = "PHYSLITETauTPThinningTool",
ThinningService = PHYSLITEThinningHelper.ThinningSvc(),
TauKey = "AnalysisTauJets",
InDetTrackParticlesKey = "InDetTrackParticles",
SelectionString = tau_thinning_expression,
ApplyAnd = False,
DoTauTracksThinning = True,
TauTracksKey = "TauTracks")
ToolSvc += PHYSLITETauTPThinningTool
thinningTools.append(PHYSLITETauTPThinningTool)
# Only keep the highest sum pT2 primary vertex
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__VertexThinning
PHYSLITEVertexThinningTool = DerivationFramework__VertexThinning(name = "PHYSLITEVertexThinningTool",
ThinningService = PHYSLITEThinningHelper.ThinningSvc(),
VertexKey = "PrimaryVertices")
ToolSvc += PHYSLITEVertexThinningTool
thinningTools.append(PHYSLITEVertexThinningTool)
#==============================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
SUSY5ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="SUSY5ElectronTPThinningTool",
ThinningService="SUSY5ThinningSvc",
SGKey="ElectronCollection",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += SUSY5ElectronTPThinningTool
thinningTools.append(SUSY5ElectronTPThinningTool)
# TrackParticles associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
SUSY5TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="SUSY5TauTPThinningTool",
ThinningService="SUSY5ThinningSvc",
TauKey="TauRecContainer",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += SUSY5TauTPThinningTool
thinningTools.append(SUSY5TauTPThinningTool)
#====================================================================
# TRUTH THINNING
#====================================================================
truth_cond_WZH = "((abs(TruthParticle.pdgId) >= 23) && (abs(TruthParticle.pdgId) <= 25))" # W, Z and Higgs
truth_cond_Lepton = "((abs(TruthParticle.pdgId) >= 11) && (abs(TruthParticle.pdgId) <= 16))" # Leptons
truth_cond_Quark = "((abs(TruthParticle.pdgId) == 6))" # Top quark
truth_cond_Photon = "((abs(TruthParticle.pdgId) == 22) && (TruthParticle.pt > 10000.))" # Photon
truth_cond_SUSY = "((abs(TruthParticle.pdgId) >= 1000001) && (abs(TruthParticle.pdgId) <= 1000039))" # SUSY particles
truth_expression = '(' + truth_cond_WZH + ' || ' + truth_cond_Lepton + ' || ' + truth_cond_Quark + ' || ' + truth_cond_Photon + ' || ' + truth_cond_SUSY + ')'
# Tracks associated with Photons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
EGAM8PhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning( name = "EGAM8PhotonTPThinningTool",
ThinningService = "EGAM8ThinningSvc",
SGKey = "Photons",
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += EGAM8PhotonTPThinningTool
print EGAM8PhotonTPThinningTool
#thinningTools.append(EGAM8PhotonTPThinningTool)
# Tracks associated with Taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
EGAM8TauTPThinningTool = DerivationFramework__TauTrackParticleThinning( name = "EGAM8TauTPThinningTool",
ThinningService = "EGAM8ThinningSvc",
TauKey = "TauJets",
ConeSize = 0.6,
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += EGAM8TauTPThinningTool
print EGAM8TauTPThinningTool
#thinningTools.append(EGAM8TauTPThinningTool)
# Tracks from primary vertex
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
EGAM8TPThinningTool = DerivationFramework__TrackParticleThinning( name = "EGAM8TPThinningTool",
ThinningService = "EGAM8ThinningSvc",
SelectionString = "abs( DFCommonInDetTrackZ0AtPV * sin(InDetTrackParticles.theta)) < 3.0",
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += EGAM8TPThinningTool
print EGAM8TPThinningTool
#thinningTools.append(EGAM8TPThinningTool)
def setup(TOPQname, TOPQThinningSvc, ToolSvc):
thinningTools = []
#========================
# Track Particle Thinning !!! BUGGY (yes, that's a technical term)...CURRENTLY NOT USING !!!
#========================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/TrackParticleThinning.cxx
thinning_expression = "(InDetTrackParticles.pt > 0.5*GeV) && (InDetTrackParticles.numberOfPixelHits > 0) && (InDetTrackParticles.numberOfSCTHits > 5) && (abs(DFCommonInDetTrackZ0AtPV) < 1.5)"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
TOPQTPThinningTool = DerivationFramework__TrackParticleThinning(
name=TOPQname + "TPThinningTool",
ThinningService=TOPQThinningSvc,
SelectionString=thinning_expression,
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=False)
#ToolSvc += TOPQTPThinningTool
#thinningTools.append(TOPQTPThinningTool)
#print TOPQname+".py", TOPQname+"TPThinningTool: ", TOPQTPThinningTool
#============================
# Jet Track Particle Thinning
#============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/JetTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
TOPQJetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name=TOPQname + "JetTPThinningTool",
ThinningService=TOPQThinningSvc,
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=False)
ToolSvc += TOPQJetTPThinningTool
thinningTools.append(TOPQJetTPThinningTool)
print TOPQname + ".py", TOPQname + "JetTPThinningTool: ", TOPQJetTPThinningTool
#=================================
# Electron Track Particle Thinning
#=================================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/EgammaTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
TOPQElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name=TOPQname + "ElectronTPThinningTool",
ThinningService=TOPQThinningSvc,
SGKey="Electrons",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=False)
ToolSvc += TOPQElectronTPThinningTool
thinningTools.append(TOPQElectronTPThinningTool)
print TOPQname + ".py", TOPQname + "ElectronTPThinningTool: ", TOPQElectronTPThinningTool
#===============================
# Electron Calo Cluster Thinning
#===============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkCalo/trunk/src/CaloClusterThinning.cxx
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
TOPQElectronCCThinningTool = DerivationFramework__CaloClusterThinning(
name=TOPQname + "ElectronCCThinningTool",
ThinningService=TOPQThinningSvc,
SGKey="Electrons",
CaloClCollectionSGKey="egammaClusters",
SelectionString=
"(Electrons.Loose || Electrons.DFCommonElectronsLHLoose)",
ConeSize=0.4,
ApplyAnd=False)
ToolSvc += TOPQElectronCCThinningTool
thinningTools.append(TOPQElectronCCThinningTool)
print TOPQname + ".py", TOPQname + "ElectronCCThinningTool: ", TOPQElectronCCThinningTool
#===============================
# Photon Calo Cluster Thinning
#===============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkCalo/trunk/src/CaloClusterThinning.cxx
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
TOPQPhotonCCThinningTool = DerivationFramework__CaloClusterThinning(
name=TOPQname + "PhotonCCThinningTool",
ThinningService=TOPQThinningSvc,
SGKey="Photons",
CaloClCollectionSGKey="egammaClusters",
SelectionString="(Photons.Loose || Photons.DFCommonPhotonsIsEMLoose)",
ConeSize=0.4,
ApplyAnd=False)
ToolSvc += TOPQPhotonCCThinningTool
thinningTools.append(TOPQPhotonCCThinningTool)
print TOPQname + ".py", TOPQname + "PhotonCCThinningTool: ", TOPQPhotonCCThinningTool
#=============================
# Muon Track Particle Thinning
#=============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/MuonTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
TOPQMuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name=TOPQname + "MuonTPThinningTool",
ThinningService=TOPQThinningSvc,
MuonKey="Muons",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=False)
ToolSvc += TOPQMuonTPThinningTool
thinningTools.append(TOPQMuonTPThinningTool)
print TOPQname + ".py", TOPQname + "MuonTPThinningTool: ", TOPQMuonTPThinningTool
#============================
# Tau Track Particle Thinning
#============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/TauTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
TOPQTauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name=TOPQname + "TauTPThinningTool",
ThinningService=TOPQThinningSvc,
TauKey="TauJets",
ConeSize=0.6,
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=False)
ToolSvc += TOPQTauTPThinningTool
thinningTools.append(TOPQTauTPThinningTool)
print TOPQname + ".py", TOPQname + "TauTPThinningTool: ", TOPQTauTPThinningTool
#==============================================
# TRUTH THINNING
#==============================================
from AthenaCommon.GlobalFlags import globalflags
if globalflags.DataSource() == 'geant4':
#==========================
# MENU TRUTH THINNING
#==========================
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__MenuTruthThinning
TOPQTruthThinningTool = DerivationFramework__MenuTruthThinning(
name=TOPQname + "TruthThinningTool",
ThinningService=TOPQThinningSvc,
ParticlesKey="TruthParticles",
VerticesKey="TruthVertices",
EventsKey="TruthEvents",
WritePartons=True, # keep partons? ###maybe set to FALSE?
WriteHadrons=False, # keep hadrons?
WriteBHadrons=False, # keep b-hadrons?
WriteGeant=False, # keep Geant particles?
GeantPhotonPtThresh=
-1.0, # Set to <0 to not write any Geant photons; otherwise write with a pT above threshold
WriteTauHad=True, # keep hadronic taus?
PartonPtThresh=-1.0, # write partons with pT aboe this threhold
WriteBSM=True, # keep BSM particles?
WriteBSMProducts=True, # keep BSM particle decay products?
WriteBosons=True, # keep bosons?
WriteBosonProducts=True, # keep boson decay products?
WriteTopAndDecays=
True, # keep top partons and immediate decay products?
WriteEverything=False, # keep everything?; overrides all other flags
WriteAllLeptons=True, # keep absolutely all leptons?
WriteLeptonsNotFromHadrons=
False, # keep leptons not from hadron decays
WriteAllStable=
True, # keep all stable particles? ###maybe set to FALSE?
WriteStatus3=False, # keep all particles with status code 3?
WriteFirstN=10, # keep first N particles in record
PreserveDescendants=False, # keep descendants of retained particles?
PreserveAncestors=True, # keep ancestors of retained particles?
SimBarcodeOffset=200000, # barcode offset for simulation particles
WritettHFHadrons=True) # keep tt+HF hadrons
ToolSvc += TOPQTruthThinningTool
thinningTools.append(TOPQTruthThinningTool)
print TOPQname + ".py", TOPQname + "TruthThinningTool: ", TOPQTruthThinningTool
#==============================================================================
# Thinning the photon truth collection : no photons from pi0 (origin=42)
#==============================================================================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkMCTruth/trunk/src/GenericTruthThinning.cxx
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
TOPQPhotonThinning = DerivationFramework__GenericTruthThinning(
name=TOPQname + "PhotonThinning",
ThinningService=TOPQThinningSvc,
ParticlesKey="TruthPhotons",
ParticleSelectionString=
"(TruthPhotons.classifierParticleOrigin != 42) || (TruthPhotons.pt > 20.0*GeV)"
)
ToolSvc += TOPQPhotonThinning
thinningTools.append(TOPQPhotonThinning)
print TOPQname + ".py", TOPQname + "PhotonThinning: ", TOPQPhotonThinning
#==============================================================================
# Thinning TruthParticles : no gluons
#==============================================================================
## PhysicsAnalysis/DerivationFramework/DerivationFrameworkMCTruth/trunk/src/GenericTruthThinning.cxx
#from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
#TOPQGluonThinning = DerivationFramework__GenericTruthThinning(
# name = TOPQname + "GluonThinning",
# ThinningService = TOPQThinningSvc,
# ParticlesKey = "TruthParticles",
# ParticleSelectionString = "(TruthParticles.pdgId != 21)")
#ToolSvc += TOPQGluonThinning
#thinningTools.append(TOPQGluonThinning)
#print TOPQname+".py", TOPQname+"GluonThinning: ", TOPQGluonThinning
return thinningTools
ThinningService="HIGG6D1ThinningSvc",
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
thinningTools.append(HIGG6D1JetTPThinningTool)
ToolSvc += HIGG6D1JetTPThinningTool
#====================================================================
# TauTrackParticleThinning
#====================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
HIGG6D1TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="HIGG6D1TauTPThinningTool",
ThinningService="HIGG6D1ThinningSvc",
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles",
# SelectionString = tauSel,
ConeSize=0.6,
ApplyAnd=False)
thinningTools.append(HIGG6D1TauTPThinningTool)
ToolSvc += HIGG6D1TauTPThinningTool
#====================================================================
# MuonTrackParticleThinning
#====================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
HIGG6D1MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name="HIGG6D1MuonTPThinningTool",
ThinningService="HIGG6D1ThinningSvc",
MuonKey="Muons",
thinningTools.append(EXOT5MuonTPThinningTool)
# Keep tracks associated with electrons
EXOT5ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name='EXOT5ElectronTPThinningTool',
ThinningService=EXOT5ThinningHelper.ThinningSvc(),
SGKey='Electrons',
GSFTrackParticlesKey='GSFTrackParticles',
InDetTrackParticlesKey='InDetTrackParticles')
ToolSvc += EXOT5ElectronTPThinningTool
thinningTools.append(EXOT5ElectronTPThinningTool)
# Keep tracks associated with taus
EXOT5TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name='EXOT5TauTPThinningTool',
ThinningService=EXOT5ThinningHelper.ThinningSvc(),
TauKey='TauJets',
InDetTrackParticlesKey='InDetTrackParticles')
ToolSvc += EXOT5TauTPThinningTool
thinningTools.append(EXOT5TauTPThinningTool)
# Keep tracks associated with jets
#EXOT5JetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
#name = 'EXOT5JetTPThinningTool',
#ThinningService = EXOT5ThinningHelper.ThinningSvc(),
#JetKey = 'AntiKt4EMTopoJets',
#InDetTrackParticlesKey = 'InDetTrackParticles',
#SelectionString = 'AntiKt4EMTopoJets.pt > 20*GeV')
#ToolSvc += EXOT5JetTPThinningTool
#thinningTools.append(EXOT5JetTPThinningTool)
def setup(HIGG4DxName, streamName, ToolSvc):
thinningTools = []
#jets and tracks
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
HIGG4DxJetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name=HIGG4DxName + "JetTPThinningTool",
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
SelectionString="AntiKt4EMTopoJets.pt > 18*GeV",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
ToolSvc += HIGG4DxJetTPThinningTool
thinningTools.append(HIGG4DxJetTPThinningTool)
HIGG4DxJetLCTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name=HIGG4DxName + "JetLCTPThinningTool",
StreamName=streamName,
JetKey="AntiKt4LCTopoJets",
SelectionString="AntiKt4LCTopoJets.pt > 18*GeV",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
ToolSvc += HIGG4DxJetLCTPThinningTool
thinningTools.append(HIGG4DxJetLCTPThinningTool)
#fat jets and track thinning
if HIGG4DxName in ['HIGG4D2', 'HIGG4D3']:
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
HIGG4DxJetTrackThinningTool1 = DerivationFramework__JetTrackParticleThinning(
name=HIGG4DxName + "JetTrackThinningTool1",
StreamName=streamName,
JetKey="AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
SelectionString=
"AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets.pt > 150*GeV",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
ToolSvc += HIGG4DxJetTrackThinningTool1
thinningTools.append(HIGG4DxJetTrackThinningTool1)
if HIGG4DxName in ['HIGG4D2', 'HIGG4D3']:
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
HIGG4DxJetTrackThinningTool2 = DerivationFramework__JetTrackParticleThinning(
name=HIGG4DxName + "JetTrackThinningTool2",
StreamName=streamName,
JetKey="AntiKt2PV0TrackJets",
SelectionString="AntiKt2PV0TrackJets.pt > 6*GeV",
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
#ToolSvc += HIGG4DxJetTrackThinningTool2
#thinningTools.append(HIGG4DxJetTrackThinningTool2)
# Tracks associated with Muons
HIGG4DxMuonSelectionString = ""
if HIGG4DxName in ['HIGG4D3', 'HIGG4D4']:
HIGG4DxMuonSelectionString = "Muons.pt > 1*GeV"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
HIGG4DxMuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name=HIGG4DxName + "MuonTPThinningTool",
StreamName=streamName,
MuonKey="Muons",
SelectionString=HIGG4DxMuonSelectionString,
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += HIGG4DxMuonTPThinningTool
thinningTools.append(HIGG4DxMuonTPThinningTool)
# Tracks associated with Electrons
HIGG4DxElectronSelectionString = ""
if HIGG4DxName in ['HIGG4D3', 'HIGG4D4']:
HIGG4DxElectronSelectionString = "Electrons.pt > 1*GeV"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
HIGG4DxElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name=HIGG4DxName + "ElectronTPThinningTool",
StreamName=streamName,
SGKey="Electrons",
SelectionString=HIGG4DxElectronSelectionString,
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += HIGG4DxElectronTPThinningTool
thinningTools.append(HIGG4DxElectronTPThinningTool)
# Tracks associated with taus
HIGG4DxTauSelectionString = "TauJets.pt > 18*GeV"
if HIGG4DxName == 'HIGG4D4':
HIGG4DxTauSelectionString = "TauJets.pt > 40*GeV"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
HIGG4DxTauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name=HIGG4DxName + "TauTPThinningTool",
StreamName=streamName,
TauKey="TauJets",
SelectionString=HIGG4DxTauSelectionString,
ConeSize=0.6,
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += HIGG4DxTauTPThinningTool
thinningTools.append(HIGG4DxTauTPThinningTool)
#Truth tau/nutau and their ancestors and descendants
truth_cond_tau = "((abs(TruthParticles.pdgId) == 15 || abs(TruthParticles.pdgId) == 16) && (TruthParticles.pt > 0.01*GeV) && (TruthParticles.barcode<200000))"
truth_cond_lep = "((abs(TruthParticles.pdgId) >= 11 && abs(TruthParticles.pdgId) <= 14) && (TruthParticles.pt > 4.0*GeV) && (TruthParticles.status == 1) && (TruthParticles.barcode<200000))"
truth_cond_comb = "(" + truth_cond_lep + "||" + truth_cond_tau + ")"
#truth_photon_thinning = "(TruthPhotons.pt > 1.0*GeV) && (abs(TruthPhotons.eta)<3.0)"
# PreserveGeneratorDescendants only keeps particles that came directly from the event generator
# PreserveDescendants keeps all particles including those that come from Geant processes
if DFisMC:
#thin special containers
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
#thin taus
HIGG4DxTruthTool_TAU = DerivationFramework__GenericTruthThinning(
name=HIGG4DxName + "TruthTool_TAU",
StreamName=streamName,
ParticleSelectionString=truth_cond_tau,
PreserveDescendants=False,
PreserveGeneratorDescendants=True,
PreserveAncestors=False,
TauHandling=False)
ToolSvc += HIGG4DxTruthTool_TAU
thinningTools.append(HIGG4DxTruthTool_TAU)
#thin leptons and taus
HIGG4DxTruthTool_COMB = DerivationFramework__GenericTruthThinning(
name=HIGG4DxName + "TruthTool_COMB",
StreamName=streamName,
ParticleSelectionString=truth_cond_comb,
PreserveDescendants=False,
PreserveGeneratorDescendants=False,
PreserveAncestors=True,
TauHandling=False)
ToolSvc += HIGG4DxTruthTool_COMB
thinningTools.append(HIGG4DxTruthTool_COMB)
#thin photons
"""HIGG4DxTruthTool_PHO = DerivationFramework__GenericTruthThinning(name = "HIGG4DxTruthTool_PHO",
StreamName = streamName,
ParticlesKey = "TruthPhotons",
ParticleSelectionString = truth_photon_thinning)
ToolSvc += HIGG4DxTruthTool_PHO
thinningTools.append(HIGG4DxTruthTool_PHO)"""
#thinning the master truth collection
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__MenuTruthThinning
HIGG4DxTruthTool_MENU = DerivationFramework__MenuTruthThinning(
name=HIGG4DxName + "TruthTool_MENU",
StreamName=streamName,
WritePartons=False,
WriteHadrons=False,
WriteBHadrons=False,
WriteGeant=False,
GeantPhotonPtThresh=-1.0,
WriteTauHad=False,
PartonPtThresh=-1.0,
WriteBSM=False,
WriteBosons=False,
WriteBSMProducts=False,
WriteBosonProducts=False,
WriteTopAndDecays=False,
WriteEverything=False,
WriteAllLeptons=False,
WriteStatus3=False,
PreserveDescendants=False,
PreserveGeneratorDescendants=False,
PreserveAncestors=True,
WriteFirstN=15)
ToolSvc += HIGG4DxTruthTool_MENU
thinningTools.append(HIGG4DxTruthTool_MENU)
#end of MC
# return thinning tools
return thinningTools
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles",
TrackSelectionString=thinning_expression)
thinningTools.append(HIGG6D2JetTPThinningTool)
ToolSvc += HIGG6D2JetTPThinningTool
#====================================================================
# TauTrackParticleThinning
#====================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
HIGG6D2TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="HIGG6D2TauTPThinningTool",
StreamName=streamName,
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles",
# SelectionString = tauSel,
ConeSize=0.6)
thinningTools.append(HIGG6D2TauTPThinningTool)
ToolSvc += HIGG6D2TauTPThinningTool
#====================================================================
# MuonTrackParticleThinning
#====================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
HIGG6D2MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name="HIGG6D2MuonTPThinningTool",
StreamName=streamName,
MuonKey="Muons",
tau_thinning_expression = "(TauJets.ptFinalCalib >= 13.*GeV) && (TauJets.nTracks>=1) && (TauJets.nTracks<=3) && (TauJets.RNNJetScoreSigTrans>0.01)"
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
PHYSTauJetsThinningTool = DerivationFramework__GenericObjectThinning(
name="PHYSTauJetsThinningTool",
StreamName=PHYSStream.Name,
ContainerName="TauJets",
SelectionString=tau_thinning_expression)
ToolSvc += PHYSTauJetsThinningTool
thinningTools.append(PHYSTauJetsThinningTool)
# Only keep tau tracks (and associated ID tracks) classified as charged tracks
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
PHYSTauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="PHYSTauTPThinningTool",
StreamName=PHYSStream.Name,
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles",
SelectionString=tau_thinning_expression,
DoTauTracksThinning=True,
TauTracksKey="TauTracks")
ToolSvc += PHYSTauTPThinningTool
thinningTools.append(PHYSTauTPThinningTool)
# ID tracks associated with high-pt di-tau
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__DiTauTrackParticleThinning
PHYSDiTauTPThinningTool = DerivationFramework__DiTauTrackParticleThinning(
name="PHYSDiTauTPThinningTool",
StreamName=PHYSStream.Name,
DiTauKey="DiTauJets",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += PHYSDiTauTPThinningTool
thinningTools.append(PHYSDiTauTPThinningTool)
thinningTools.append(JETM6ElectronTPThinningTool)
# TrackParticles associated with photons
JETM6PhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="JETM6PhotonTPThinningTool",
ThinningService="JETM6ThinningSvc",
SGKey="Photons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += JETM6PhotonTPThinningTool
thinningTools.append(JETM6PhotonTPThinningTool)
# TrackParticles associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
JETM6TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="JETM6TauTPThinningTool",
ThinningService="JETM6ThinningSvc",
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += JETM6TauTPThinningTool
thinningTools.append(JETM6TauTPThinningTool)
# Truth particle thinning
doTruthThinning = True
preserveAllDescendants = False
from AthenaCommon.GlobalFlags import globalflags
if doTruthThinning and globalflags.DataSource() == 'geant4':
truth_cond_WZH = "((abs(TruthParticles.pdgId) >= 23) && (abs(TruthParticles.pdgId) <= 25))" # W, Z and Higgs
truth_cond_Lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 16) && (TruthParticles.barcode < 200000))" # Leptons
truth_cond_Quark = "((abs(TruthParticles.pdgId) <= 5 && (TruthParticles.pt > 10000.)) || (abs(TruthParticles.pdgId) == 6))" # Quarks
truth_cond_Gluon = "((abs(TruthParticles.pdgId) == 21) && (TruthParticles.pt > 10000.))" # Gluons
truth_cond_Photon = "((abs(TruthParticles.pdgId) == 22) && (TruthParticles.pt > 10000.) && (TruthParticles.barcode < 200000))" # Photon
def setup(TOPQname, streamName, ToolSvc):
thinningTools = []
#========================
# Track Particle Thinning !!! BUGGY (yes, that's a technical term)...CURRENTLY NOT USING !!!
#========================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/TrackParticleThinning.cxx
thinning_expression = "(InDetTrackParticles.pt > 0.5*GeV) && (InDetTrackParticles.numberOfPixelHits > 0) && (InDetTrackParticles.numberOfSCTHits > 5) && (abs(DFCommonInDetTrackZ0AtPV) < 1.5)"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
TOPQTPThinningTool = DerivationFramework__TrackParticleThinning(
name=TOPQname + "TPThinningTool",
StreamName=streamName,
SelectionString=thinning_expression,
InDetTrackParticlesKey="InDetTrackParticles")
#ToolSvc += TOPQTPThinningTool
#thinningTools.append(TOPQTPThinningTool)
#print (TOPQname+".py", TOPQname+"TPThinningTool: ", TOPQTPThinningTool)
#============================
# Jet Track Particle Thinning
#============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/JetTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
TOPQJetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name=TOPQname + "JetTPThinningTool",
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles",
SelectionString="(AntiKt4EMTopoJets.DFCommonJets_Calib_pt > 20*GeV)")
ToolSvc += TOPQJetTPThinningTool
thinningTools.append(TOPQJetTPThinningTool)
print(TOPQname + ".py", TOPQname + "JetTPThinningTool: ",
TOPQJetTPThinningTool)
from DerivationFrameworkTop.DerivationFrameworkTopConf import DerivationFramework__SV1TrackThinning
TOPQSV1ThinningTool = DerivationFramework__SV1TrackThinning(
name=TOPQname + "SV1TrackThinning",
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles",
SelectionString="(AntiKt4EMTopoJets.DFCommonJets_Calib_pt > 20*GeV)")
ToolSvc += TOPQSV1ThinningTool
thinningTools.append(TOPQSV1ThinningTool)
print(TOPQname + ".py", TOPQname + "SV1ThinningTool: ",
TOPQSV1ThinningTool)
TOPQLargeJetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name=TOPQname + "LargeJetTPThinningTool",
StreamName=streamName,
JetKey="AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
InDetTrackParticlesKey="InDetTrackParticles",
SelectionString="AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets.pt>25*GeV")
ToolSvc += TOPQLargeJetTPThinningTool
thinningTools.append(TOPQLargeJetTPThinningTool)
print(TOPQname + ".py", TOPQname + "LargeJetTPThinningTool: ",
TOPQLargeJetTPThinningTool)
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__JetCaloClusterThinning
TOPQAK4CCThinningTool = DerivationFramework__JetCaloClusterThinning(
name=TOPQname + "AK4CCThinningTool",
StreamName=streamName,
SGKey="AntiKt4EMTopoJets",
TopoClCollectionSGKey="CaloCalTopoClusters",
SelectionString="AntiKt4EMTopoJets.DFCommonJets_Calib_pt > 7*GeV",
ConeSize=0.5)
ToolSvc += TOPQAK4CCThinningTool
thinningTools.append(TOPQAK4CCThinningTool)
print(TOPQname + ".py", TOPQname + "AK4CCThinningTool: ",
TOPQAK4CCThinningTool)
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__JetCaloClusterThinning
TOPQLargeJetCCThinningTool = DerivationFramework__JetCaloClusterThinning(
name=TOPQname + "LargeJetCCThinningTool",
StreamName=streamName,
SGKey="AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
TopoClCollectionSGKey="CaloCalTopoClusters",
SelectionString=
"AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets.DFCommonJets_Calib_pt > 7*GeV",
ConeSize=1.1)
ToolSvc += TOPQLargeJetCCThinningTool
thinningTools.append(TOPQLargeJetCCThinningTool)
print(TOPQname + ".py", TOPQname + "LargeJetCCThinningTool: ",
TOPQLargeJetCCThinningTool)
#=================================
# Electron Track Particle Thinning
#=================================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/EgammaTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
TOPQElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name=TOPQname + "ElectronTPThinningTool",
StreamName=streamName,
SGKey="Electrons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += TOPQElectronTPThinningTool
thinningTools.append(TOPQElectronTPThinningTool)
print(TOPQname + ".py", TOPQname + "ElectronTPThinningTool: ",
TOPQElectronTPThinningTool)
#===============================
# Electron Calo Cluster Thinning
#===============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkCalo/trunk/src/CaloClusterThinning.cxx
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
TOPQElectronCCThinningTool = DerivationFramework__CaloClusterThinning(
name=TOPQname + "ElectronCCThinningTool",
StreamName=streamName,
SGKey="Electrons",
CaloClCollectionSGKey="egammaClusters",
SelectionString=
"(Electrons.Loose || Electrons.DFCommonElectronsLHLoose)",
ConeSize=0.4)
ToolSvc += TOPQElectronCCThinningTool
thinningTools.append(TOPQElectronCCThinningTool)
print(TOPQname + ".py", TOPQname + "ElectronCCThinningTool: ",
TOPQElectronCCThinningTool)
#========================
# Photon Track Particle Thinning
#========================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
TOPQPhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name=TOPQname + "PhotonTPThinningTool",
StreamName=streamName,
SGKey="Photons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += TOPQPhotonTPThinningTool
thinningTools.append(TOPQPhotonTPThinningTool)
print(TOPQname + ".py", TOPQname + "PhotonTPThinningTool: ",
TOPQPhotonTPThinningTool)
#===============================
# Photon Calo Cluster Thinning
#===============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkCalo/trunk/src/CaloClusterThinning.cxx
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
TOPQPhotonCCThinningTool = DerivationFramework__CaloClusterThinning(
name=TOPQname + "PhotonCCThinningTool",
StreamName=streamName,
SGKey="Photons",
CaloClCollectionSGKey="egammaClusters",
SelectionString="(Photons.Loose || Photons.DFCommonPhotonsIsEMLoose)",
ConeSize=0.4)
ToolSvc += TOPQPhotonCCThinningTool
thinningTools.append(TOPQPhotonCCThinningTool)
print(TOPQname + ".py", TOPQname + "PhotonCCThinningTool: ",
TOPQPhotonCCThinningTool)
#=============================
# Muon Track Particle Thinning
#=============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/MuonTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
TOPQMuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name=TOPQname + "MuonTPThinningTool",
StreamName=streamName,
MuonKey="Muons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += TOPQMuonTPThinningTool
thinningTools.append(TOPQMuonTPThinningTool)
print(TOPQname + ".py", TOPQname + "MuonTPThinningTool: ",
TOPQMuonTPThinningTool)
#============================
# Tau Track Particle Thinning
#============================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkInDet/trunk/src/TauTrackParticleThinning.cxx
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
TOPQTauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name=TOPQname + "TauTPThinningTool",
StreamName=streamName,
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += TOPQTauTPThinningTool
thinningTools.append(TOPQTauTPThinningTool)
print(TOPQname + ".py", TOPQname + "TauTPThinningTool: ",
TOPQTauTPThinningTool)
# #==============================
# # Large-R jet thinning
# #==============================
#
# # we remove low pT large-R jets from the various collections
# from AthenaCommon.GlobalFlags import globalflags
# largeRjetCollsToThin = ["AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets"]
# if globalflags.DataSource()=='geant4':
# largeRjetCollsToThin.append("AntiKt10TruthTrimmedPtFrac5SmallR20Jets")
# largeRjetCollsToThin.append("AntiKt10TruthJets")
# from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
# for largeRjetColl in largeRjetCollsToThin:
# #pTvariableToCutOn = ".DFCommonJets_Calib_pt" # use calibrated pT for reco jets
# pTvariableToCutOn = ".pt" # well for now the uncalibrated pT seems a more reasonable option
# # use normal pT for truth jets
# if largeRjetColl.find('Truth') > -1:
# pTvariableToCutOn = ".pt"
# TOPQLargeRJetThinning = DerivationFramework__GenericObjectThinning(
# name = TOPQname + largeRjetColl + "Thinning_LargeR",
# StreamName = streamName,
# ContainerName = largeRjetColl,
# SelectionString = largeRjetColl + pTvariableToCutOn + " > 150.0*GeV")
# ToolSvc += TOPQLargeRJetThinning
# thinningTools.append(TOPQLargeRJetThinning)
# print (TOPQname+".py", TOPQname+largeRjetColl+"Thinning: ", TOPQLargeRJetThinning)
#==============================
# Photon thinning
#==============================
# we remove photons below 9GeV
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
pTPhotonVariableToCutOn = ".pt"
photonColl = "Photons"
TOPQPhotonThinning = DerivationFramework__GenericObjectThinning(
name=TOPQname + photonColl + "Thinning_lowpTphotons",
StreamName=streamName,
ContainerName=photonColl,
SelectionString=photonColl + ".pt > 9.0*GeV")
ToolSvc += TOPQPhotonThinning
thinningTools.append(TOPQPhotonThinning)
print(TOPQname + ".py", TOPQname + photonColl + "Thinning: ",
TOPQPhotonThinning)
#==============================================
# TRUTH THINNING
#==============================================
from AthenaCommon.GlobalFlags import globalflags
if globalflags.DataSource() == 'geant4':
#==========================
# MENU TRUTH THINNING
#==========================
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__MenuTruthThinning
TOPQTruthThinningTool = DerivationFramework__MenuTruthThinning(
name=TOPQname + "TruthThinningTool",
StreamName=streamName,
ParticlesKey="TruthParticles",
VerticesKey="TruthVertices",
EventsKey="TruthEvents",
WritePartons=True, # keep partons? ###maybe set to FALSE?
WriteHadrons=False, # keep hadrons?
WriteBHadrons=False, # keep b-hadrons?
WriteGeant=False, # keep Geant particles?
GeantPhotonPtThresh=
-1.0, # Set to <0 to not write any Geant photons; otherwise write with a pT above threshold
WriteTauHad=True, # keep hadronic taus?
PartonPtThresh=-1.0, # write partons with pT aboe this threhold
WriteBSM=True, # keep BSM particles?
WriteBSMProducts=True, # keep BSM particle decay products?
WriteBosons=True, # keep bosons?
WriteBosonProducts=True, # keep boson decay products?
WriteTopAndDecays=
True, # keep top partons and immediate decay products?
WriteEverything=False, # keep everything?; overrides all other flags
WriteAllLeptons=True, # keep absolutely all leptons?
WriteLeptonsNotFromHadrons=
False, # keep leptons not from hadron decays
WriteAllStable=
True, # keep all stable particles? ###maybe set to FALSE?
WriteStatus3=False, # keep all particles with status code 3?
WriteFirstN=10, # keep first N particles in record
PreserveDescendants=False, # keep descendants of retained particles?
PreserveAncestors=True, # keep ancestors of retained particles?
SimBarcodeOffset=200000, # barcode offset for simulation particles
WritettHFHadrons=True) # keep tt+HF hadrons
ToolSvc += TOPQTruthThinningTool
thinningTools.append(TOPQTruthThinningTool)
print(TOPQname + ".py", TOPQname + "TruthThinningTool: ",
TOPQTruthThinningTool)
#==============================================================================
# Thinning the photon truth collection : no photons from pi0 (origin=42)
#==============================================================================
# PhysicsAnalysis/DerivationFramework/DerivationFrameworkMCTruth/trunk/src/GenericTruthThinning.cxx
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
TOPQPhotonThinning = DerivationFramework__GenericTruthThinning(
name=TOPQname + "PhotonThinning",
StreamName=streamName,
ParticlesKey="TruthPhotons",
ParticleSelectionString=
"(TruthPhotons.classifierParticleOrigin != 42) || (TruthPhotons.pt > 20.0*GeV)"
)
ToolSvc += TOPQPhotonThinning
thinningTools.append(TOPQPhotonThinning)
print(TOPQname + ".py", TOPQname + "PhotonThinning: ",
TOPQPhotonThinning)
#==============================================================================
# Thinning TruthParticles : no gluons
#==============================================================================
## PhysicsAnalysis/DerivationFramework/DerivationFrameworkMCTruth/trunk/src/GenericTruthThinning.cxx
#from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
#TOPQGluonThinning = DerivationFramework__GenericTruthThinning(
# name = TOPQname + "GluonThinning",
# StreamName = streamName,
# ParticlesKey = "TruthParticles",
# ParticleSelectionString = "(TruthParticles.pdgId != 21)")
#ToolSvc += TOPQGluonThinning
#thinningTools.append(TOPQGluonThinning)
#print (TOPQname+".py", TOPQname+"GluonThinning: ", TOPQGluonThinning)
return thinningTools
thinningTools.append(SUSY2ElectronTPThinningTool)
# TrackParticles associated with photons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
SUSY2PhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="SUSY2PhotonTPThinningTool",
ThinningService=SUSY2ThinningHelper.ThinningSvc(),
SGKey="Photons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += SUSY2PhotonTPThinningTool
thinningTools.append(SUSY2PhotonTPThinningTool)
# TrackParticles associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
SUSY2TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="SUSY2TauTPThinningTool",
ThinningService=SUSY2ThinningHelper.ThinningSvc(),
TauKey="TauJets",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += SUSY2TauTPThinningTool
thinningTools.append(SUSY2TauTPThinningTool)
#====================================================================
# TRUTH THINNING
#====================================================================
if DerivationFrameworkIsMonteCarlo:
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__MenuTruthThinning
SUSY2TruthThinningTool = DerivationFramework__MenuTruthThinning(
name="SUSY2TruthThinningTool",
ThinningService=SUSY2ThinningHelper.ThinningSvc(),
WritePartons=False,
WriteHadrons=False,
# Tracks associated with Electrons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
STDM3ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="STDM3ElectronTPThinningTool",
StreamName=streamName,
SGKey="Electrons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += STDM3ElectronTPThinningTool
thinningTools.append(STDM3ElectronTPThinningTool)
# Tracks associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
STDM3TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="STDM3TauTPThinningTool",
StreamName=streamName,
TauKey="TauJets",
SelectionString="TauJets.pt > 15*GeV",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += STDM3TauTPThinningTool
thinningTools.append(STDM3TauTPThinningTool)
# Truth leptons and their ancestors and descendants
truth_cond_boson = "((abs(TruthParticles.pdgId) == 23) || (abs(TruthParticles.pdgId) == 24))"
truth_cond_lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 14) &&(TruthParticles.pt > 1*GeV) && (TruthParticles.status ==1) && (TruthParticles.barcode<200000))"
photonthinningexpr = "(TruthPhotons.classifierParticleOrigin != 42) && !(TruthPhotons.classifierParticleOrigin >= 23 && TruthPhotons.classifierParticleOrigin <= 35)"
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
if globalflags.DataSource() == 'geant4':
from DerivationFrameworkSM.STDMCommonTruthTools import *
# Tracks associated with Photons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
HIGG5D2PhotonTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name="HIGG5D2PhotonTPThinningTool",
ThinningService=HIGG5D2ThinningHelper.ThinningSvc(),
SGKey="Photons",
InDetTrackParticlesKey="InDetTrackParticles",
BestMatchOnly=True)
ToolSvc += HIGG5D2PhotonTPThinningTool
thinningTools.append(HIGG5D2PhotonTPThinningTool)
# Tracks associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
HIGG5D2TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name="HIGG5D2TauTPThinningTool",
ThinningService=HIGG5D2ThinningHelper.ThinningSvc(),
TauKey="TauJets",
ConeSize=0.6,
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += HIGG5D2TauTPThinningTool
thinningTools.append(HIGG5D2TauTPThinningTool)
#========================================================================
# lepton selection (keep isolation failed leptons for QCD-MJ estimation)
#========================================================================
lepSel = '( count( (Muons.pt > 20.0*GeV) && (abs(Muons.eta) < 2.6) && (Muons.DFCommonGoodMuon) ) + count( (Electrons.pt > 20.0*GeV) && (abs(Electrons.eta) < 2.6) && (Electrons.DFCommonElectronsLHLoose) ) ) > 0'
#====================================================================
# jet selection
#====================================================================
#jetSel = '(( count( (AntiKt4EMTopoJets.pt > 0.0*GeV) && (abs(AntiKt4EMTopoJets.eta) < 2.6) ) ) > 1)'
jetSel = '(( count( (AntiKt4EMTopoJets.DFCommonJets_Calib_pt > 15.*GeV) && (abs(AntiKt4EMTopoJets.DFCommonJets_Calib_eta) < 2.6) ) ) > 0)'
# Tracks associated with Electrons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
HIGG2D2ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(name = "HIGG2D2ElectronTPThinningTool",
StreamName = streamName,
SGKey = "Electrons",
InDetTrackParticlesKey = "InDetTrackParticles",
BestMatchOnly = False)
ToolSvc += HIGG2D2ElectronTPThinningTool
thinningTools.append(HIGG2D2ElectronTPThinningTool)
# Tracks associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
HIGG2D2TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(name = "HIGG2D2TauTPThinningTool",
StreamName = streamName,
TauKey = "TauJets",
ConeSize = 0.6,
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += HIGG2D2TauTPThinningTool
thinningTools.append(HIGG2D2TauTPThinningTool)
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
HIGG2D2MuonCCThinningTool = DerivationFramework__CaloClusterThinning(name = "HIGG2D2MuonCCThinningTool",
StreamName = streamName,
SGKey = "Muons",
TopoClCollectionSGKey = "CaloCalTopoClusters",
SelectionString = "Muons.pt>0.*GeV",
ConeSize = 0.6)
ToolSvc += HIGG2D2MuonCCThinningTool
thinningTools.append(HIGG2D2MuonCCThinningTool)
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += STDM5MuonTPThinningTool
# Tracks associated with Electrons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
STDM5ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning( name = "STDM5ElectronTPThinningTool",
ThinningService = STDM5ThinningHelper.ThinningSvc(),
SGKey = "Electrons",
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += STDM5ElectronTPThinningTool
# Tracks associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
STDM5TauTPThinningTool = DerivationFramework__TauTrackParticleThinning( name = "STDM5TauTPThinningTool",
ThinningService = STDM5ThinningHelper.ThinningSvc(),
TauKey = "TauJets",
SelectionString = "TauJets.pt > 15*GeV",
ConeSize = 0.6,
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += STDM5TauTPThinningTool
thinningTools= [STDM5JetTPThinningTool,STDM5MuonTPThinningTool,STDM5ElectronTPThinningTool,STDM5TauTPThinningTool]
if globalflags.DataSource()=='geant4':
from DerivationFrameworkSM.STDMCommonTruthTools import *
#TruthThinning
# Truth leptons and their ancestors and descendants
truth_cond_boson = "((abs(TruthParticles.pdgId) == 23) || (abs(TruthParticles.pdgId) == 24))"
thinningTools.append(EXOT5MuonTPThinningTool)
# Keep tracks associated with electrons
EXOT5ElectronTPThinningTool = DerivationFramework__EgammaTrackParticleThinning(
name='EXOT5ElectronTPThinningTool',
StreamName=stream_name,
SGKey='Electrons',
GSFTrackParticlesKey='GSFTrackParticles',
InDetTrackParticlesKey='InDetTrackParticles')
ToolSvc += EXOT5ElectronTPThinningTool
thinningTools.append(EXOT5ElectronTPThinningTool)
# Keep tracks associated with taus
EXOT5TauTPThinningTool = DerivationFramework__TauTrackParticleThinning(
name='EXOT5TauTPThinningTool',
StreamName=stream_name,
TauKey='TauJets',
InDetTrackParticlesKey='InDetTrackParticles')
ToolSvc += EXOT5TauTPThinningTool
thinningTools.append(EXOT5TauTPThinningTool)
# Keep tracks associated with jets
#EXOT5JetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
#name = 'EXOT5JetTPThinningTool',
#StreamName = stream_name,
#JetKey = 'AntiKt4EMTopoJets',
#InDetTrackParticlesKey = 'InDetTrackParticles',
#SelectionString = 'AntiKt4EMTopoJets.pt > 20*GeV')
#ToolSvc += EXOT5JetTPThinningTool
#thinningTools.append(EXOT5JetTPThinningTool)