metAlg.METName = 'MET_Reco_AntiKt4EMTopo'
metAlg_truth = getMETMakerAlg('Truth_AntiKt4EMTopo', 20e3)
metAlg_truth.METSoftClName = 'SoftTruthAll'
metAlg_truth.METSoftTrkName = 'SoftTruthCharged'
metAlg_truth.METName = 'MET_Truth_AntiKt4EMTopo'
from AthenaCommon.AlgSequence import AlgSequence
topSequence = AlgSequence()
topSequence += metAlg
topSequence += metAlg_truth
write_xAOD = True
if write_xAOD:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream("StreamXAOD",
"xAOD.METMaker.pool.root")
#
xaodStream.AddItem('xAOD::MissingETContainer#MET_Reco_AntiKt4EMTopo')
xaodStream.AddItem(
'xAOD::MissingETAuxContainer#MET_Reco_AntiKt4EMTopoAux.')
#
xaodStream.AddItem('xAOD::MissingETContainer#MET_Truth_AntiKt4EMTopo')
xaodStream.AddItem(
'xAOD::MissingETAuxContainer#MET_Truth_AntiKt4EMTopoAux.')
#
xaodStream.AddItem('xAOD::MissingETContainer#MET_Truth')
xaodStream.AddItem('xAOD::MissingETAuxContainer#MET_TruthAux.')
#
xaodStream.Print()
# from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
# ----------------------------------------------------------------------------
# add DiTauBuilder to Algorithm Sequence
# ----------------------------------------------------------------------------
from DiTauRec.DiTauRecConf import DiTauBuilder
DiTauBuilder = DiTauBuilder(
name="DiTauBuilder",
DiTauContainer="DiTau",
DiTauAuxContainer="DiTauAux.", # the dot is important
Tools=tools,
SeedJetName=_jet_container,
minPt=15000,
maxEta=2.5,
OutputLevel=2,
Rjet=_R_jet,
Rsubjet=_R_subjet,
Rcore=_R_core)
topSequence += DiTauBuilder
print topSequence
# ----------------------------------------------------------------------------
# xAOD output stream
# ----------------------------------------------------------------------------
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
outStream = MSMgr.NewPoolRootStream("DiTauXAODStream", "DiTauXAOD.pool.root")
# outStream.AddItem(['xAOD::JetContainer#GoodJets'])
# outStream.AddItem(['xAOD::JetAuxContainer#GoodJetsAux.'])
outStream.AddItem(['xAOD::DiTauJetContainer#DiTau'])
outStream.AddItem(['xAOD::DiTauJetAuxContainer#DiTauAux.'])
topSequence += DerivationFrameworkJob
if printSctDxAODConf:
msg.info(DerivationFrameworkJob)
msg.info(DerivationFrameworkJob.properties())
#################
### Steer output file content
#################
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
from D2PDMaker.D2PDHelpers import buildFileName
from PrimaryDPDMaker.PrimaryDPDFlags import primDPD
streamName = primDPD.WriteDAOD_SCTVALIDStream.StreamName
fileName = buildFileName(primDPD.WriteDAOD_SCTVALIDStream)
SCTVALIDStream = MSMgr.NewPoolRootStream(streamName, fileName)
SCTVALIDStream.AcceptAlgs(["SCTxAOD_DF_KERN"])
excludedAuxData = "-caloExtension.-cellAssociation.-clusterAssociation.-trackParameterCovarianceMatrices.-parameterX.-parameterY.-parameterZ.-parameterPX.-parameterPY.-parameterPZ.-parameterPosition"
# Add generic event information
SCTVALIDStream.AddItem("xAOD::EventInfo#*")
SCTVALIDStream.AddItem("xAOD::EventAuxInfo#*")
# Add track particles collection and traclets (if available)
SCTVALIDStream.AddItem("xAOD::TrackParticleContainer#InDetTrackParticles")
SCTVALIDStream.AddItem(
"xAOD::TrackParticleAuxContainer#InDetTrackParticlesAux." +
excludedAuxData)
# Add vertices
SCTVALIDStream.AddItem("xAOD::VertexContainer#PrimaryVertices")
print DerivationFrameworkJob.properties()
#################
### Steer output file content
#################
## Add service for metadata
ToolSvc += CfgMgr.xAODMaker__TriggerMenuMetaDataTool("TriggerMenuMetaDataTool")
svcMgr.MetaDataSvc.MetaDataTools += [ToolSvc.TriggerMenuMetaDataTool]
## Steer output file
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
from D2PDMaker.D2PDHelpers import buildFileName
from PrimaryDPDMaker.PrimaryDPDFlags import primDPD
streamName = primDPD.WriteDAOD_IDNCBStream.StreamName
fileName = buildFileName(primDPD.WriteDAOD_IDNCBStream)
IDNCBStream = MSMgr.NewPoolRootStream(streamName, fileName)
IDNCBStream.AcceptAlgs(["DFIDNCB_KERN"])
from AthenaServices.Configurables import ThinningSvc, createThinningSvc
augStream = MSMgr.GetStream(streamName)
evtStream = augStream.GetEventStream()
svcMgr += createThinningSvc(svcName="IDNCBThinningSvc", outStreams=[evtStream])
excludedAuxData = "-caloExtension.-cellAssociation.-clusterAssociation.-trackParameterCovarianceMatrices.-parameterX.-parameterY.-parameterZ.-parameterPX.-parameterPY.-parameterPZ.-parameterPosition"
excludedTRTData = "-T0.-TRTboard.-TRTchip.-bitPattern.-driftTimeToTCorrection.-driftTimeHTCorrection.-highThreshold.-strawnumber"
excludedSCTData = "-detectorElementID.-hitsInThirdTimeBin.-rdo_groupsize"
# Add generic event information
IDNCBStream.AddItem("xAOD::EventInfo#*")
IDNCBStream.AddItem("xAOD::EventAuxInfo#*")
#SKC Add jets!
# jetrec_trimm.JetModifiers += [charge]
# jetrec_trimm.JetModifiers += [subjetmaker]
# jetrec_trimm.JetModifiers += [subjetfinder]
ToolSvc += jetrec_trimm
# Add the algorithm. It runs the demo tools.
jetalg = JetAlgorithm("JetAlgTruth")
# jetalg.OutputLevel = INFO
jetalg.Tools += [copytruth, jetrec, jetrec_trimm]
topSequence += jetalg
###end jet stuff
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream("StreamAOD",
"XAOD_" + Name + ".pool.root")
# xaodStream.Stream.TakeItemsFromInput = True #this will only work for the event-by-event items. MetadataItems must still be specified
xaodStream.AddItem("xAOD::EventInfo#*")
xaodStream.AddItem("xAOD::EventAuxInfo#*")
xaodStream.AddItem("xAOD::EventShape#*")
xaodStream.AddItem("xAOD::EventShapeAuxInfo#*")
xaodStream.AddItem("xAOD::VertexContainer#*")
xaodStream.AddItem("xAOD::VertexAuxContainer#*")
# xaodStream.AddItem( "xAOD::TrackCaloClusterContainer#TrackCaloClusters")
# xaodStream.AddItem( "xAOD::TrackCaloClusterAuxContainer#TrackCaloClustersAux.")
# xaodStream.AddItem( "xAOD::TrackCaloClusterContainer#TrackCaloClustersCombined")
# xaodStream.AddItem( "xAOD::TrackCaloClusterAuxContainer#TrackCaloClustersCombinedAux.")
# xaodStream.AddItem( "xAOD::TrackCaloClusterContainer#TrackCaloClustersAll")
# xaodStream.AddItem( "xAOD::TrackCaloClusterAuxContainer#TrackCaloClustersAllAux.")
# xaodStream.AddItem( "xAOD::TrackCaloClusterContainer#TrackCaloClustersAllTrack")
# xaodStream.AddItem( "xAOD::TrackCaloClusterAuxContainer#TrackCaloClustersAllTrackAux.")
#################
### Steering options
#################
## Load common flags
from AthenaCommon.JobProperties import jobproperties as athCommonFlags
## Steer output file
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
from D2PDMaker.D2PDHelpers import buildFileName
from PrimaryDPDMaker.PrimaryDPDFlags import primDPD
streamName = primDPD.WriteDAOD_L1CALO3.StreamName
fileName = buildFileName(primDPD.WriteDAOD_L1CALO3)
L1CALO3Stream = MSMgr.NewPoolRootStream(streamName, fileName)
L1CALO3Stream.AcceptAlgs(["DFL1CALO3_KERN"])
augStream = MSMgr.GetStream(streamName)
evtStream = augStream.GetEventStream()
#################
### Setup Augmentation tools
#################
augmentationTools = []
from AthenaCommon import CfgMgr
# Set up stream auditor
from AthenaCommon.AppMgr import ServiceMgr as svcMgr
if not hasattr(svcMgr, "DecisionSvc"):
svcMgr += CfgMgr.DecisionSvc()
svcMgr.DecisionSvc.CalcStats = True
#====================================================================
]
#One of the existing master sequences where one should attach all algorithms
algSeq = CfgMgr.AthSequencer("AthAlgSeq")
# Add the test algorithm. the alg is a part of the tool, so it should be imported:
from PhotonEfficiencyCorrection.PhotonEfficiencyCorrectionConf import testAthenaPhotonAlg
alg = testAthenaPhotonAlg()
#define input files, for now it uses my own directory where the tool is, since it not a working part of the athena, please note that the current recomendation of PhotonID WG can be found here: https://twiki.cern.ch/twiki/bin/viewauth/AtlasProtected/PhotonEfficiencyCorrection
#Set inputs for ID efficiency
alg.PhotonEfficiencyCorrectionTool.CorrectionFileNameConv = "PhotonEfficiencyCorrection/v1/efficiencySF.offline.Tight.2015.13TeV.rel20.con.v01.root"
alg.PhotonEfficiencyCorrectionTool.CorrectionFileNameUnconv = "PhotonEfficiencyCorrection/v1/efficiencySF.offline.Tight.2015.13TeV.rel20.unc.v01.root"
#Or set inputs for track isolation efficiency (comment previous)
#alg.PhotonEfficiencyCorrectionTool.CorrectionFileNameConv = "PhotonEfficiencyCorrection/efficiencySF.offline.isolFixedCutLoose.2015.13TeV.rel20.con.v01.root";
#alg.PhotonEfficiencyCorrectionTool.CorrectionFileNameUnconv = "PhotonEfficiencyCorrection/efficiencySF.offline.isolFixedCutLoose.2015.13TeV.rel20.unc.v01.root";
#Set DataType: for data use 0 (or don't run the tool - faster), for FULLSIM use 1, and for FASTSIM use 3, please note that the input files are also should be different
alg.PhotonEfficiencyCorrectionTool.ForceDataType = 1
#comment this line
alg.OutputLevel = DEBUG
algSeq += alg
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
outStream = MSMgr.NewPoolRootStream("MyXAODStream", "myXAOD.pool.root")
outStream.AddItem(['xAOD::PhotonContainer_v1#MyPhotons'])
outStream.AddItem(['xAOD::PhotonAuxContainer_v1#MyPhotonsAux'])
pflow_ungroomed_modifiers.remove('truthassoc')
jtm.addJetFinder("AntiKt4EMPFlowPUPPIJets",
"AntiKt",
0.4,
myPUPPIgetters,
pflow_ungroomed_modifiers,
ghostArea=0.01,
ptmin=5000,
ptminFilter=10000,
calibOpt="")
############################################################################################
from JetRec.JetAlgorithm import addJetRecoToAlgSequence
addJetRecoToAlgSequence(job=topSequence, separateJetAlgs=True)
if hasattr(topSequence, "jetalgAntiKt4EMPFlowPUPPIJets"):
topSequence.jetalgAntiKt4EMPFlowPUPPIJets.Tools.insert(0, PUPPISequence)
############################################################################################
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xAODStream = MSMgr.NewPoolRootStream("StreamXAOD", outputName)
xAODStream.AddItem("xAOD::JetContainer#AntiKt4EMPFlowPUPPIJets")
xAODStream.AddItem("xAOD::JetAuxContainer#AntiKt4EMPFlowPUPPIJets")
xAODStream.AddItem("xAOD::PFOContainer#*ParticleFlowObjects*")
xAODStream.AddItem("xAOD::PFOAuxContainer#*ParticleFlowObjects*")
xAODStream.AddItem("xAOD::ShallowAuxContainer#*ParticleFlowObjects*")
import AthenaPoolCnvSvc.ReadAthenaPool
ServiceMgr.EventSelector.InputCollections = [FNAME]
# Access the algoritgm sequence:
from AthenaCommon.AlgSequence import AlgSequence
theJob = AlgSequence()
# Schedule the auxiliary store wrapper algorithm:
from xAODCoreCnv.xAODCoreCnvConf import xAODMaker__AuxStoreWrapper
alg = xAODMaker__AuxStoreWrapper()
alg.SGKeys = []
alg.OutputLevel = DEBUG
theJob += alg
# Create a derived xAOD file:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
derivedStream = MSMgr.NewPoolRootStream("StreamAOD", "DxAOD.pool.root")
# Set up its contents:
derivedStream.AddItem("xAOD::EventInfo#*")
derivedStream.AddItem("xAOD::CaloClusterContainer#*")
derivedStream.AddItem("xAOD::AuxInfoBase#*")
derivedStream.AddItem("xAOD::AuxContainerBase#*")
# Some final tweaking:
theApp.EvtMax = 10
ServiceMgr.MessageSvc.OutputLevel = INFO
ServiceMgr.MessageSvc.defaultLimit = 10000
ServiceMgr.StoreGateSvc.Dump = True
'xAOD::EventInfo#EventInfo', 'xAOD::EventAuxInfo#EventInfoAux.'
]
# --- add trigger to IDRE standalone ESD
StreamESD.ItemList += ["TrigDec::TrigDecision#TrigDecision"]
StreamESD.ItemList += ["HLT::HLTResult#HLTResult_L2"]
StreamESD.ItemList += ["HLT::HLTResult#HLTResult_EF"]
StreamESD.ForceRead = True # otherwise unread stuff is not copied
if InDetFlags.doDBMstandalone():
topSequence.StreamESD.ItemList += [
"TrackCollection#SiSPSeededTracks"
]
if doWriteAOD:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
# --- create *augmented* stream; this makes ROOT happy
StreamAOD_Augmented = MSMgr.NewPoolRootStream(
"StreamAOD", InDetKeys.OutputAODFileName())
# --- here is the old, backward-compatible event stream
StreamAOD = StreamAOD_Augmented.GetEventStream()
# --- save MC collections if truth turned on
if InDetFlags.doTruth():
StreamAOD.ItemList += truthList
# --- load list of objects
include("InDetRecExample/WriteInDetAOD.py")
StreamAOD.ItemList += [
'xAOD::EventInfo#EventInfo', 'xAOD::EventAuxInfo#EventInfoAux.'
]
# --- add trigger to IDRE standalone AOD
StreamAOD.ItemList += ["TrigDec::TrigDecision#TrigDecision"]
StreamAOD.ItemList += ["HLT::HLTResult#HLTResult_L2"]
"DFTSOS_KERN", AugmentationTools=augmentationTools, OutputLevel=INFO)
topSequence += DerivationFrameworkJob
if (printIdTrkDxAODConf):
print(DerivationFrameworkJob)
print(DerivationFrameworkJob.properties())
#################
### Steer output file content
#################
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
from D2PDMaker.D2PDHelpers import buildFileName
from PrimaryDPDMaker.PrimaryDPDFlags import primDPD
streamName = primDPD.WriteDAOD_IDTRKLUMIStream.StreamName
fileName = buildFileName(primDPD.WriteDAOD_IDTRKLUMIStream)
IDTRKLUMIStream = MSMgr.NewPoolRootStream(streamName, fileName)
excludedAuxData = "-caloExtension.-cellAssociation.-clusterAssociation.-trackParameterCovarianceMatrices.-parameterX.-parameterY.-parameterZ.-parameterPX.-parameterPY.-parameterPZ.-parameterPosition"
# Add generic event information
IDTRKLUMIStream.AddItem("xAOD::EventInfo#*")
IDTRKLUMIStream.AddItem("xAOD::EventAuxInfo#*")
# Add track particles collection and traclets (if available)
IDTRKLUMIStream.AddItem("xAOD::TrackParticleContainer#InDetTrackParticles")
IDTRKLUMIStream.AddItem(
"xAOD::TrackParticleAuxContainer#InDetTrackParticlesAux." +
excludedAuxData)
if InDetFlags.doTrackSegmentsPixel():
IDTRKLUMIStream.AddItem(
"xAOD::TrackParticleContainer#InDetPixelTrackParticles")
IDTRKLUMIStream.AddItem(
#
# Set up the reading of an input file:
INFILE = "AOD.pool.root"
from AthenaCommon.AthenaCommonFlags import athenaCommonFlags
athenaCommonFlags.FilesInput = [ INFILE ]
athenaCommonFlags.EvtMax = 10
# Turn off most of what RecExCommon does:
from RecExConfig.RecFlags import rec
rec.doCBNT.set_Value_and_Lock( False )
rec.doWriteAOD.set_Value_and_Lock( False )
rec.doWriteTAG.set_Value_and_Lock( False )
rec.AutoConfiguration = [ "everything" ]
# Read the file using RecExCommon:
include( "RecExCommon/RecExCommon_topOptions.py" )
# Set up the metadata tool:
ToolSvc += CfgMgr.xAODMaker__FileMetaDataCreatorTool( "FileMetaDataCreatorTool",
OutputLevel = 2 )
svcMgr.MetaDataSvc.MetaDataTools += [ ToolSvc.FileMetaDataCreatorTool ]
# Set up the writing of an output file:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
stream = MSMgr.NewPoolRootStream( "StreamDAOD", "DxAOD.pool.root" )
stream.AddItem( [ "xAOD::EventInfo#EventInfo",
"xAOD::EventAuxInfo#EventInfoAux." ] )
stream.AddMetaDataItem( [ "xAOD::FileMetaData#FileMetaData",
"xAOD::FileMetaDataAuxInfo#FileMetaDataAux." ] )
containerName= "SCTonlyVertex" AnAODList = [] AnAODList+=['xAOD::VertexContainer#'+containerName] AnAODList+=['xAOD::VertexAuxContainer#'+containerName+'Aux.-vxTrackAtVertex'] StreamAOD.ItemList += AnAODList if False: from OutputStreamAthenaPool.MultipleStreamManager import MSMgr SCTonlyVertexStream = MSMgr.NewPoolRootStream( "SCTonlyVertexStream", "SCTonlyVertex.pool.root" ) SCTonlyVertexStream.AddItem( "TrigDec::TrigDecision#TrigDecision" ) SCTonlyVertexStream.AddItem( 'xAOD::VertexContainer#'+containerName ) SCTonlyVertexStream.AddItem( 'xAOD::VertexAuxContainer#'+containerName+'Aux.-vxTrackAtVertex' ) SCTonlyVertexStream.AddItem( 'xAOD::EventInfo#EventInfo' ) SCTonlyVertexStream.AddItem( 'xAOD::EventAuxInfo#EventInfoAux.' )
from PyUtils.MetaReaderPeeker import convert_itemList
inputObjects = convert_itemList(layout=None)
def __addInput(stream, skip=[]):
for element in inputObjects:
type_key = "%s#%s" % element
if type_key not in skip:
stream.AddItem(type_key)
elStream = MSMgr.NewPoolStream('elXAODStream', 'elXAOD.pool.root')
__addInput(elStream)
phStream = MSMgr.NewPoolRootStream('phXAODStream', 'phXAOD.pool.root')
tauStream = MSMgr.NewPoolRootStream('tauXAODStream', 'tauXAOD.pool.root')
from TrigNavTools.TrigNavToolsConfig import navigationThinningSvc
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
dk = DerivationFramework__DerivationKernel()
tokeep = {
'electron': [
'HLT_TrigElectronContainer_L2ElectronFex',
'HLT_TrigElectronContainer_L2IDCaloFex',
'HLT_xAOD__TrigElectronContainer_L2ElectronFex',
'HLT_xAOD__TrigElectronContainer_L2IDCaloFex'
],
# HistToolList = histMgr.ToolList()
# )
#algseq += subSeq
# Make a cut: check that we have at least one Z-boson candidate
subSeq = CfgMgr.CutAlg("CutZExists",
Cut="count(ZmumuCands.pt > -100.0*GeV) >= 1")
# Make another cut: check the invariant mass of the di-muon system
subSeq += CfgMgr.CutAlg("CutZMass", Cut="count(ZmumuCands.m > 70.0*GeV) >= 1")
# ====================================================================
# Create a new xAOD:
# ====================================================================
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xAODStreamName = "MyFirstXAODStream"
xAODFileName = "myXAOD.pool.root"
MyFirstXAODStream = MSMgr.NewPoolRootStream(xAODStreamName, xAODFileName)
MyFirstXAODStream.AddItem(['xAOD::MuonContainer#Muons'])
MyFirstXAODStream.AddItem(['xAOD::MuonAuxContainer#MuonsAux.'])
MyFirstXAODStream.AddItem(['xAOD::CompositeParticleContainer#ZmumuCands'])
MyFirstXAODStream.AddItem(
['xAOD::CompositeParticleAuxContainer#ZmumuCandsAux.'])
# Only events that pass the filters listed below are written out
# AcceptAlgs = logical OR of filters
# RequireAlgs = logical AND of filters
# VetoAlgs = logical NOT of filters
MyFirstXAODStream.AddAcceptAlgs(["CutZMass"])
"mc14_13TeV.110401.PowhegPythia_P2012_ttbar_nonallhad.merge.AOD."
"e2928_s1982_s2008_r5787_r5853_tid01597980_00/"
"AOD.01597980._000420.pool.root.1"
]
athenaCommonFlags.EvtMax = -1
# Include RecExCommon:
from RecExConfig.RecFlags import rec
rec.doCBNT.set_Value_and_Lock(False)
rec.doWriteTAG.set_Value_and_Lock(False)
rec.doAOD.set_Value_and_Lock(False)
rec.doWriteAOD.set_Value_and_Lock(False)
include("RecExCommon/RecExCommon_topOptions.py")
# Configure the metadata tool:
ToolSvc += CfgMgr.xAODMaker__TriggerMenuMetaDataTool("TriggerMenuMetaDataTool",
OutputLevel=1)
svcMgr.MetaDataSvc.MetaDataTools += [ToolSvc.TriggerMenuMetaDataTool]
# Set up an output file:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
output = MSMgr.NewPoolRootStream("StreamAOD", "merged.pool.root")
# Add some event and metadata items to the stream:
output.AddItem(
["xAOD::EventInfo#EventInfo", "xAOD::EventAuxInfo#EventInfoAux."])
output.AddMetaDataItem([
"xAOD::TriggerMenuContainer#TriggerMenu",
"xAOD::TriggerMenuAuxContainer#TriggerMenuAux."
])
cfg_mrf.builders['Gamma'].OutputLevel = VERBOSE
cfg_mrf.builders['Tau'].OutputLevel = VERBOSE
cfg_mrf.builders['Jet'].OutputLevel = VERBOSE
cfg_mrf.builders['SoftTrk'].OutputLevel = VERBOSE
cfg_mrf.refiners['TrackFilter'].OutputLevel = VERBOSE
#from Valkyrie.JobOptCfg import ValgrindSvc
#svcMgr += ValgrindSvc( OutputLevel = DEBUG,
# ProfiledAlgs = ['METAssocAlg','METAssocAlg'],
# ProfiledIntervals = ['METAssocAlg.execute'])
from PerfMonComps.PerfMonFlags import jobproperties as pmon_properties
pmon_properties.PerfMonFlags.doSemiDetailedMonitoring = True
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream('StreamAOD',
'xAOD.METAssoc.testassoc.pool.root')
xaodStream.AddItem('xAOD::MissingETAssociationMap_v1#METAssoc_Akt4LC')
xaodStream.AddItem('xAOD::MissingETAuxAssociationMap_v1#METAssoc_Akt4LCAux.')
xaodStream.AddItem('xAOD::EventInfo_v1#EventInfo')
xaodStream.AddItem('xAOD::EventAuxInfo_v1#EventInfoAux.')
#xaodStream.AddItem('xAOD::ElectronContainer#ElectronCollection')
#xaodStream.AddItem('xAOD::ElectronAuxContainer#ElectronCollectionAux.')
#xaodStream.AddItem('xAOD::PhotonContainer#PhotonCollection')
#xaodStream.AddItem('xAOD::PhotonAuxContainer#PhotonCollectionAux.')
#xaodStream.AddItem('xAOD::TauJetContainer_v1#TauRecContainer')
#xaodStream.AddItem('xAOD::TauJetAuxContainer_v1#TauRecContainerAux.')
#xaodStream.AddItem('xAOD::JetContainer_v1#AntiKt4LCTopoJets')
#xaodStream.AddItem('xAOD::JetAuxContainer_v1#AntiKt4LCTopoJetsAux.')
#xaodStream.AddItem('xAOD::MuonContainer_v1#Muons')
#xaodStream.AddItem('xAOD::MuonAuxContainer_v1#MuonsAux.')
#xaodStream.AddItem('xAOD::MissingETComponentMap_v1#METMap_RefFinalFix')
if dataType == "data":
testFile = os.getenv('ASG_TEST_FILE_DATA')
elif dataType == "mc":
testFile = os.getenv('ASG_TEST_FILE_MC')
elif dataType == "afii":
testFile = os.getenv('ASG_TEST_FILE_MC_AFII')
jps.AthenaCommonFlags.FilesInput = [testFile]
from MuonAnalysisAlgorithms.MuonAnalysisAlgorithmsTest import makeSequence
algSeq = makeSequence(dataType)
print algSeq # For debugging
# Add all algorithms from the sequence to the job.
athAlgSeq += algSeq
# Write a mini-xAOD if requested:
if athArgs.write_xaod:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
minixAOD = MSMgr.NewPoolRootStream(
'AAOD_MUON', FileName='MuonAnalysisAlgorithmsTest.AAOD_MUON.pool.root')
minixAOD.AddItem([
'xAOD::EventInfo#EventInfo', 'xAOD::EventAuxInfo#EventInfoAux.-',
'xAOD::MuonContainer#AnalysisMuons_NOSYS',
'xAOD::AuxContainerBase#AnalysisMuons_NOSYSAux.eta.phi.pt'
])
# Reduce the printout from Athena:
include("AthAnalysisBaseComps/SuppressLogging.py")
except: print "WARNING: Release doesn't seem to have fix for missing version tags - will try to apply patch" needsOutputFilePatch = True else: needsOutputFilePatch = False InDetKeys.lockAllExceptAlias() if jobConfig['doPrintIndetConfig']: print "Printing InDetKeys" InDetKeys.print_JobProperties() if jobConfig['doPrimaryVertexing'] : jobConfig['VertexContainer'] = InDetKeys.PrimaryVerticesWithoutBeamConstraint() from OutputStreamAthenaPool.MultipleStreamManager import MSMgr RefittedVertexStream = MSMgr.NewPoolRootStream( "RefittedVertexStream", "RefittedVertex.pool.root" ) RefittedVertexStream.AddItem( 'xAOD::VertexContainer#'+InDetKeys.PrimaryVerticesWithoutBeamConstraint() ) RefittedVertexStream.AddItem( 'xAOD::VertexAuxContainer#'+InDetKeys.PrimaryVerticesWithoutBeamConstraint()+'Aux.-vxTrackAtVertex' ) RefittedVertexStream.AddItem( 'xAOD::EventInfo#EventInfo' ) RefittedVertexStream.AddItem( 'xAOD::EventAuxInfo#EventInfoAux.' ) #-------------------------------------------------------------- # enable statistics for reading ESD testing #-------------------------------------------------------------- InDetFlags.doStatistics = doInDetRecStatistics TrackCollectionKeys = [InDetKeys.Tracks()] TrackCollectionTruthKeys = [InDetKeys.TracksTruth()] #--------------------------------------------------------------
# Get a handle to the main athsequencer, for adding things to later!
AST99Job = CfgMgr.AthSequencer("AthAlgSeq")
AST99Job += CfgMgr.xAODMaker__AuxStoreWrapper("AST99AuxStoreWrapperAlg",
SGKeys=auxList,
OutputLevel=INFO)
###################################
# Create Output Stream and Thinning
###################################
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xAODStreamName = "AST99Stream"
AST99Stream = MSMgr.NewPoolRootStream(xAODStreamName, xAODFileName)
from AthenaServices.Configurables import ThinningSvc
svcMgr += ThinningSvc("AST99ThinningSvc",
Streams=[xAODStreamName],
OutputLevel=INFO)
thinningTools = []
skimmingTools = []
#########################################
# Configure SUSYTools and object thinning
#########################################
# Initial SUSYObjDef_xAOD
#====================================================================
IDTIDESequence += CfgMgr.DerivationFramework__DerivationKernel("IDTIDE1Kernel",
AugmentationTools = augmentationTools,
ThinningTools = thinningTools,
OutputLevel =INFO)
DerivationFrameworkJob += IDTIDESequence
#====================================================================
# SET UP STREAM
#====================================================================
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
from D2PDMaker.D2PDHelpers import buildFileName
from PrimaryDPDMaker.PrimaryDPDFlags import primDPD
streamName = primDPD.WriteDAOD_IDTIDEStream.StreamName
fileName = buildFileName( primDPD.WriteDAOD_IDTIDEStream )
IDTIDE1Stream = MSMgr.NewPoolRootStream( streamName, fileName )
IDTIDE1Stream.AcceptAlgs(["IDTIDE1Kernel"])
#idtideSeq = CfgMgr.AthSequencer("IDTIDE1Sequence")
#DerivationFrameworkJob += idtideSeq
#addTrackSumMoments("AntiKt4EMTopo")
#addDefaultTrimmedJets(idtideSeq,"IDTIDE1")
# SPECIAL LINES FOR THINNING
# Thinning service name must match the one passed to the thinning tools
from AthenaServices.Configurables import ThinningSvc, createThinningSvc
augStream = MSMgr.GetStream( streamName )
evtStream = augStream.GetEventStream()
svcMgr += createThinningSvc( svcName="IDTIDE1ThinningSvc", outStreams=[evtStream] )
#====================================================================
# $Id: xAODHIEventShape_test1_jobOptions.py 693667 2015-09-08 11:05:12Z krasznaa $
# Read in a data file holding the v1 EDM:
import AthenaPoolCnvSvc.ReadAthenaPool
svcMgr.EventSelector.InputCollections = [
"AOD.pool.root" ]
# Create an output file:
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
outputStream = MSMgr.NewPoolRootStream( "StreamAOD", "minixAOD1.pool.root" )
# Copy all xAOD::HIEventShape containers to the output:
outputStream.AddItem( [ "xAOD::HIEventShapeContainer#*",
"xAOD::HIEventShapeAuxContainer#*" ] )
# Some additional tweaks:
theApp.EvtMax = 10
svcMgr.MessageSvc.OutputLevel = INFO
svcMgr.MessageSvc.defaultLimit = 1000000
runNumber += i.partition("calib.")[2].partition(".calib")[0]
pass
#configuration of globalflags... do this FIRST so that anything else you do that might load conddb will be ready for it
from AthenaCommon.GlobalFlags import globalflags
globalflags.DataSource = 'data'
globalflags.DatabaseInstance = 'CONDBR2'
globalflags.InputFormat = 'bytestream'
globalflags.DetDescrVersion = "ATLAS-R2-2015-02-00-00"
from IOVDbSvc.CondDB import conddb
conddb.setGlobalTag("CONDBR2-ES1PA-2015-01")
# Output file
OutFile = "xAOD.L1Calo." + runNumber + ".pool.root"
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream("StreamXAOD", OutFile)
# Decorate trigger towers with Calo Cells?
# This should be setup better
decorateTT = True
# Number of events (-1 == all events)
nEvents = -1
# Tell Athena about the input file(s)
from ByteStreamCnvSvc import ReadByteStream
svcMgr.ByteStreamInputSvc.FullFileName = InFiles
topSequence = CfgMgr.AthSequencer("AthAlgSeq")
# Add L1Calo xAOD to output
#pflow_ungroomed_modifiers=filterout(['width'],jtm.modifiersMap["pflow_ungroomed"]) #width tool causes a problem for some reason
jtm.addJetFinder("MyAntiKt4EMPFlowJets2",
"AntiKt",
0.4,
myGetters,
"pflow_ungroomed",
ghostArea=0.01,
ptmin=5000,
ptminFilter=10000,
calibOpt="arj:pflow")
############################################################################################
from JetRec.JetAlgorithm import addJetRecoToAlgSequence
addJetRecoToAlgSequence(job=topSequence, separateJetAlgs=True)
#insert the JetConstituentModSequence before sequence-based PF jet reconstruction
if hasattr(topSequence, "jetalgMyAntiKt4EMPFlowJets2"):
topSequence.jetalgMyAntiKt4EMPFlowJets2.Tools.insert(0, PFSequence)
############################################################################################
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xAODStream = MSMgr.NewPoolRootStream("StreamXAOD", "test.Out.pool.root")
xAODStream.AddItem("xAOD::JetContainer#*AntiKt4EMPFlowJets*")
xAODStream.AddItem("xAOD::JetAuxContainer#*AntiKt4EMPFlowJets*")
xAODStream.AddItem("xAOD::PFOContainer#*ParticleFlowObjects*")
xAODStream.AddItem("xAOD::PFOAuxContainer#*ParticleFlowObjects*")
xAODStream.AddItem("xAOD::ShallowAuxContainer#*ParticleFlowObjects*")
# Create the xAOD::EventInfo object:
theJob += CfgMgr.xAODMaker__EventInfoCnvAlg("EventInfoCnvAlg")
# Run the xAOD truth builder
from xAODTruthCnv.xAODTruthCnvConf import xAODMaker__xAODTruthCnvAlg
alg = xAODMaker__xAODTruthCnvAlg()
# Pile up options... default is no pile-up
alg.WriteInTimePileUpTruth = False
alg.WriteAllPileUpTruth = False
alg.AODContainerName = "GEN_EVENT"
alg.OutputLevel = INFO
theJob += alg
# Output
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream("StreamAOD", "xAOD.pool.root")
# Additional type(s) created automatically by RecExCommon:
xaodStream.AddItem("xAOD::EventInfo#EventInfo")
xaodStream.AddItem("xAOD::EventAuxInfo#EventInfoAux.")
# The types created explicitly by this job:
xaodStream.AddItem("xAOD::TruthEventContainer#*")
xaodStream.AddItem("xAOD::TruthEventAuxContainer#*")
xaodStream.AddItem("xAOD::TruthPileupEventContainer#*")
xaodStream.AddItem("xAOD::TruthPileupEventAuxContainer#*")
xaodStream.AddItem("xAOD::TruthVertexContainer#*")
xaodStream.AddItem("xAOD::TruthVertexAuxContainer#*")
xaodStream.AddItem("xAOD::TruthParticleContainer#*")
xaodStream.AddItem("xAOD::TruthParticleAuxContainer#*")
import AthenaPoolCnvSvc.ReadAthenaPool
svcMgr.EventSelector.InputCollections = ["AOD.09897018._000001.pool.root.1"]
theApp.EvtMax = 100
algseq = CfgMgr.AthSequencer(
"AthAlgSeq"
) #gets a handle to the main athsequencer, for adding things to later!
algseq += CfgMgr.ThinGeantTruthAlg("ThinGeantTruthAlg",
StreamName='StreamXAOD')
from OutputStreamAthenaPool.MultipleStreamManager import MSMgr
xaodStream = MSMgr.NewPoolRootStream(
"StreamXAOD",
"xAOD.out.root") #suppose this is what your stream definition looked like
xaodStream.AddItem([
"xAOD::TruthParticleContainer#*", "xAOD::TruthParticleAuxContainer#*",
"xAOD::TruthVertexContainer#*", "xAOD::TruthVertexAuxContainer#*"
])
