def addMuons(streamName=""): if not streamName == "": stream = MSMgr.GetStream(streamName) ##----------------------------------------------------------------------------- ## Write Muon related containers ##----------------------------------------------------------------------------- stream.AddItem(["Analysis::MuonContainer#*"]) stream.AddItem(["TrackRecordCollection#MuonEntryRecordFilter"]) stream.AddItem(["Rec::TrackParticleContainer#MuidCombTrackParticles"]) stream.AddItem( ["Rec::TrackParticleContainer#MuidCombTrackParticlesLowPt"]) stream.AddItem(["Rec::TrackParticleContainer#MuidExtrTrackParticles"]) stream.AddItem(["Rec::TrackParticleContainer#MuTagTrackParticles"]) stream.AddItem([ "Rec::TrackParticleContainer#MuonboyMuonSpectroOnlyTrackParticles" ]) stream.AddItem(["Rec::TrackParticleContainer#MuonboyTrackParticles"]) stream.AddItem(["Rec::TrackParticleContainer#StacoTrackParticles"]) stream.AddItem(["Rec::TrackParticleContainer#MooreTrackParticles"]) stream.AddItem(["Rec::MuonSpShowerContainer#MuonSpShowers"]) stream.AddItem(["CaloCompactCellContainer#AODCellContainer" ]) ## Requires detector description stream.AddItem(["CaloCellContainer#AODCellContainer" ]) ## Requires detector description stream.AddItem(["CaloClusterContainer#CaloCalTopoCluster"]) if rec.readESD(): stream.AddItem(["TileMuContainer#TileMuObj"]) stream.AddItem(["Trk::SegmentCollection#*"]) stream.AddItem(["Muon::CscPrepDataContainer#CSC_Clusters"]) stream.AddItem(["Muon::MdtPrepDataContainer#MDT_DriftCircles"]) stream.AddItem(["Muon::RpcPrepDataContainer#RPC_Measurements"]) stream.AddItem(["Muon::TgcPrepDataContainer#TGC_Measurements"]) stream.AddItem( ["MuonCaloEnergyContainer#MuonCaloEnergyCollection"]) if primDPD.UseMCTruth(): stream.AddItem(["PRD_MultiTruthCollection#*"]) pass else: primaryDPD_BasicOutput_msg.warning( 'Can not add the basic output collections to the stream %s', streamName) pass return
def addTruth(streamName=""): if not streamName == "": stream = MSMgr.GetStream(streamName) if primDPD.UseMCTruth(): from ParticleBuilderOptions.AODFlags import AODFlags if rec.readAOD: AODFlags.McEventKey = "GEN_AOD" else: AODFlags.McEventKey = "GEN_EVENT" if primDPD.ApplyTruthSlimming(): #include("PrimaryDPDMaker/SlimTruth.py") stream.AddItem(["McEventCollection#GEN_DPD"]) stream.AddItem(["TruthEtIsolationContainer#*"]) stream.AddItem(["TruthParticleContainer#SpclMC"]) pass else: stream.AddItem(["McEventCollection#GEN_AOD"]) stream.AddItem(["TruthParticleContainer#SpclMC"]) pass stream.AddItem(["JetCollection#Cone4TruthJets"]) stream.AddItem(["JetCollection#Cone7TruthJets"]) #stream.AddItem( ["JetCollection#AntiKt4TruthJets"] ) stream.AddItem(["MissingEtTruth#MET_Truth"]) stream.AddItem(["MissingEtTruth#MET_Truth_PileUp"]) stream.AddItem( ["TrackParticleTruthCollection#TrackParticleTruthCollection"]) pass pass else: primaryDPD_BasicOutput_msg.warning( 'Can not add the basic output collections to the stream %s', streamName) pass return
#================================== from RecExConfig.RecFlags import rec # Turn off most of RecExCommon... (optional) rec.doWriteRDO = False rec.doESD = False rec.doWriteESD = False rec.doAOD = False rec.doWriteAOD = False rec.doWriteTAG = False rec.doHist = False rec.doCBNT = False #DR leave perfmon on by default, needed also for DPD making #rec.doPerfMon = False #varies with the settings rec.doTruth = primDPD.UseMCTruth() rec.doAODCaloCells = False #================================== #Setting AOD flags. Not much needs to be done if you read an ESD 'cause everything's already built. #See https://twiki.cern.ch/twiki/bin/view/Atlas/UserAnalysisTest#The_AOD_Production_Flags #================================== from ParticleBuilderOptions.AODFlags import AODFlags AODFlags.Photon = False AODFlags.Electron = False AODFlags.egammaTrackSlimmer = False AODFlags.JetTag = False AODFlags.Streaming = False AODFlags.FastSimulation = False AODFlags.MissingEtTruth = False AODFlags.FastTrackParticle = False