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)