# I/O stuff
#
ServiceMgr.EventSelector.InputCollections = [
    "root://eosatlas//eos/atlas/atlascerngroupdisk/proj-sit/digitization/RTT/mc10/mc10_7TeV.005008.CavernInput.merge.HITS.e4_e607_s951_s952_tid170551_00/HITS.170551._000011.pool.root.1"
]

# Set up for seeking.
ServiceMgr.EventSelector.CollectionType = "SeekableROOT"

include("AthenaPoolCnvSvc/WriteAthenaPool_jobOptions.py")
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream

Stream1 = AthenaPoolOutputStream("Stream1")
Stream1.OutputFile = "MultiplePassFile.root"
# ** mandatory parameter ** // The output file name
Stream1.ForceRead = TRUE  #force read of output data objs
Stream1.ItemList = ["EventInfo#*", "McEventCollection#*"]

#--------------------------------------------------------------
#--- Monitoring and Debug printouts
#--------------------------------------------------------------

MessageSvc.OutputLevel = INFO
Stream1.OutputLevel = DEBUG
#StoreGateSvc.Dump=1
#StoreGateSvc.OutputLevel=DEBUG

from AthenaCommon.AppMgr import theAuditorSvc
from GaudiAud.GaudiAudConf import ChronoAuditor

theAuditorSvc += ChronoAuditor()
#---------------------------------------------------------------------
# Perfmon
#--------------------------------------------------------------------

from PerfMonComps.PerfMonFlags import jobproperties
jobproperties.PerfMonFlags.doMonitoring = True
from PerfMonComps.JobOptCfg import PerfMonSvc
svcMgr += PerfMonSvc()
jobproperties.PerfMonFlags.doDetailedMonitoring = True

#--------------------------------------------------------------
# Event related parameters
#--------------------------------------------------------------
# Number of events to be processed (default is 10)
theApp.EvtMax = 10
theApp.EvtSel = "EventSelector"

#Create output StreamESD
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
StreamESD = AthenaPoolOutputStream("StreamESD",
                                   athenaCommonFlags.PoolESDOutput(), True)
StreamESD.ForceRead = True
# Put MetaData in ESD stream via StreamESD_FH.
StreamESD.MetadataItemList += ["LumiBlockCollection#*"]

include("EventAthenaPool/EventAthenaPoolItemList_joboptions.py")
StreamESD.ItemList += fullItemList

#David, instruction to get all the info in the DetectorStore
ServiceMgr.DetectorStore.Dump = True
Exemplo n.º 3
0
StreamDPD.ItemList += ['egDetailContainer#egDetailAOD']
StreamDPD.ItemList += ['MissingET#MET_RefFinal']
StreamDPD.ItemList += ['MissingET#MET_LocHadObj']
StreamDPD.ItemList += ['CaloClusterContainer#CaloCalTopoCluster']
StreamDPD.ItemList += ['Analysis::MuonContainer#StacoMuonCollection']
StreamDPD.ItemList += ['JetCollection#Kt4LCTopoJets']
StreamDPD.ItemList += ['JetKeyDescriptor#JetKeyMap']
StreamDPD.ItemList += ['JetCollection#Kt4TruthJets']
StreamDPD.ItemList += ['Rec::TrackParticleContainer#TrackParticleCandidate']
StreamDPD.ItemList += ['Rec::TrackParticleContainer#StacoTrackParticles']
StreamDPD.ItemList += ['TruthParticleContainer#SpclMC']
StreamDPD.ItemList += ['McEventCollection#GEN_AOD']

StreamDPD.OutputFile = "SkimmedThin.AOD.pool.root"
#StreamDPD.OutputFile = outFileName
StreamDPD.ForceRead=TRUE;  #force read of output data objs
StreamDPD.AcceptAlgs=["ttbarFilter"]


ChronoStatSvc = Service( "ChronoStatSvc" )
ChronoStatSvc.ChronoDestinationCout = True

ChronoStatSvc.PrintUserTime     = True
ChronoStatSvc.PrintSystemTime   = True
ChronoStatSvc.PrintEllapsedTime = True
AthenaPoolCnvSvc = Service( "AthenaPoolCnvSvc" )
AthenaPoolCnvSvc.UseDetailChronoStat = True

MessageSvc = Service("MessageSvc")
#MessageSvc.OutputLevel = DEBUG
#MessageSvc.OutputLevel = ERROR
Exemplo n.º 4
0
  
  if doWriteESD:
    # --- create stream
    StreamESD            = AthenaPoolOutputStream ( "StreamESD" )
    StreamESD.OutputFile = "InDetRecESD.root"
    # --- save MC collection if truth turned on
    if InDetFlags.doTruth():
      StreamESD.ItemList += ["McEventCollection#*"]
    # ---- load list of objects
    include ( "InDetRecExample/WriteInDetESD.py" )
      
  if doWriteAOD:
    # --- create stream
    StreamAOD            = AthenaPoolOutputStream ( "StreamAOD" )
    StreamAOD.OutputFile = "InDetRecAOD.root"
    # --- save MC collection if truth turned on
    if InDetFlags.doTruth():
      StreamAOD.ItemList += ["McEventCollection#*"]
    # --- load list of objects
    include ( "InDetRecExample/WriteInDetAOD.py" )
  
  if 'doCopyRDO' in dir() and doCopyRDO:
    # --- create stream
    StreamRDO            = AthenaPoolOutputStream ( "StreamRDO" )
    StreamRDO.OutputFile = "InDetRecRDO.root"
    # --- add output to list
    StreamRDO.ItemList   +=  ['TRT_RDO_Container#*','SCT_RDO_Container#*','PixelRDO_Container#*']
    StreamRDO.ItemList   +=  ['InDetSimDataCollection#*','McEventCollection#*']
    # --- Force read
    StreamRDO.ForceRead  = TRUE;  #force read of output data objs
  theApp.TopAlg = [ e for e in theApp.TopAlg if e not in ['JetTagNtupleBuilder/JTNtBuilder'] ]
  theApp.TopAlg += ['JetTagNtupleBuilder/JTNtBuilder']
if BTaggingFlags.doJetTagSlimNtuple:
  theApp.TopAlg = [ e for e in theApp.TopAlg if e not in ['JetTagNtupleBuilder/JTNtSlimBuilder'] ]
  theApp.TopAlg += ['JetTagNtupleBuilder/JTNtSlimBuilder']

# override DB: to have the worst case for pixel dead modules
if OverridePixMapOverlay != "":
  print '#BTAG# Overriding PixMapOverlay with ',OverridePixMapOverlay
  conddb.addOverride('/PIXEL/PixMapOverlay', OverridePixMapOverlay)

# --- output a new AOD file if requested (not fully validated ! strange behavior for re-tagged collections !)
if WriteNewAODFile:
  from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
  StreamAOD = AthenaPoolOutputStream( "StreamAOD", "newAOD.pool.root", True )
  StreamAOD.ForceRead = True 
  StreamAOD.ExtendProvenanceRecord = False 
  StreamAOD.TakeItemsFromInput=True
  taggedJetList = []
  for key in BTaggingFlags.Jets:
    if key in BTaggingFlags.RetagJets:
#      newJetColl = "JetCollection#"+key+"JetsAOD"
      newJetColl = "JetCollection#"+key+"JetsReTagged"
    else:
      newJetColl = "JetCollection#"+key+"Jets"
    print '#BTAG# jet collection: ',newJetColl
    taggedJetList +=  [ newJetColl ]
  StreamAOD.ItemList += taggedJetList
  print '#BTAG# writing out :',StreamAOD.ItemList

# --- Filter on GRL first:
Exemplo n.º 6
0
    def hits_persistency():
        """ HITS POOL file persistency
        """
        from G4AtlasApps.SimFlags import simFlags
        from AthenaCommon.DetFlags import DetFlags
        from AthenaCommon.AthenaCommonFlags import athenaCommonFlags
        from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream

        ## Not yet understood, but need to treat StreamHITS as alg in Hive.
        ## Seems to also work fine outside of Hive, but to be extra safe I'm
        ## only changing it in Hive.
        from AthenaCommon.ConcurrencyFlags import jobproperties as concurrencyProps
        if concurrencyProps.ConcurrencyFlags.NumThreads() > 0:
            as_alg = True
        else:
            as_alg = False
        ## NB. Two-arg constructor is needed, since otherwise metadata writing fails!
        stream1 = AthenaPoolOutputStream("StreamHITS",
                                         athenaCommonFlags.PoolHitsOutput(),
                                         asAlg=as_alg)

        ## Write geometry tag info - move to main method
        #import EventInfoMgt.EventInfoMgtInit

        ## EventInfo & TruthEvent always written by default
        stream1.ForceRead = True
        stream1.ItemList = [
            "EventInfo#*", "McEventCollection#TruthEvent", "JetCollection#*"
        ]

        ## If we are running quasi-stable particle simulation, include the original event record
        if hasattr(
                simFlags, 'IncludeParentsInG4Event'
        ) and simFlags.IncludeParentsInG4Event.statusOn and simFlags.IncludeParentsInG4Event(
        ):
            stream1.ItemList += ["McEventCollection#GEN_EVENT"]

        stream1.ItemList += ["xAOD::JetContainer#*", "xAOD::JetAuxContainer#*"]

        ## Make stream aware of aborted events
        stream1.AcceptAlgs = ["G4AtlasAlg"]

        ## Detectors

        ## Inner Detector
        if DetFlags.ID_on():
            stream1.ItemList += [
                "SiHitCollection#*", "TRTUncompressedHitCollection#*",
                "TrackRecordCollection#CaloEntryLayer"
            ]
        ## Calo
        if DetFlags.Calo_on():
            stream1.ItemList += [
                "CaloCalibrationHitContainer#*",
                "LArHitContainer#*",
                "TileHitVector#*",
                #"SimpleScintillatorHitCollection#*",
                "TrackRecordCollection#MuonEntryLayer"
            ]
        ## Muon
        if DetFlags.Muon_on():
            stream1.ItemList += [
                "RPCSimHitCollection#*", "TGCSimHitCollection#*",
                "CSCSimHitCollection#*", "MDTSimHitCollection#*",
                "TrackRecordCollection#MuonExitLayer"
            ]
            if hasattr(simFlags, 'SimulateNewSmallWheel'):
                if simFlags.SimulateNewSmallWheel():
                    stream1.ItemList += ["GenericMuonSimHitCollection#*"]
        ## Lucid
        if DetFlags.Lucid_on():
            stream1.ItemList += ["LUCID_SimHitCollection#*"]

        ## FwdRegion
        if DetFlags.FwdRegion_on():
            stream1.ItemList += ["SimulationHitCollection#*"]

        ## ZDC
        if DetFlags.ZDC_on():
            stream1.ItemList += [
                "ZDC_SimPixelHit_Collection#*", "ZDC_SimStripHit_Collection#*"
            ]
        ## ALFA
        if DetFlags.ALFA_on():
            stream1.ItemList += [
                "ALFA_HitCollection#*", "ALFA_ODHitCollection#*"
            ]

        ## AFP
        if DetFlags.AFP_on():
            stream1.ItemList += [
                "AFP_TDSimHitCollection#*", "AFP_SIDSimHitCollection#*"
            ]

        ### Ancillary scintillators
        #stream1.ItemList += ["ScintillatorHitCollection#*"]

        ## TimingAlg
        stream1.ItemList += ["RecoTimingObj#EVNTtoHITS_timings"]

        ## Add cosmics and test beam configuration hit persistency if required cf. geom tag
        layout = simFlags.SimLayout.get_Value()
        if "tb" not in layout:
            from AthenaCommon.BeamFlags import jobproperties
            if jobproperties.Beam.beamType() == 'cosmics' or \
                    (hasattr(simFlags, "WriteTR") and simFlags.WriteTR.statusOn) or \
                    (hasattr(simFlags, "ReadTR") and simFlags.ReadTR.statusOn):
                stream1.ItemList += [
                    "TrackRecordCollection#CosmicRecord",
                    "TrackRecordCollection#CosmicPerigee"
                ]
        else:
            ## CTB-specific
            if layout.startswith("ctb"):
                if simFlags.LArFarUpstreamMaterial.statusOn and simFlags.LArFarUpstreamMaterial.get_Value(
                ):
                    stream1.ItemList.append(
                        "TrackRecordCollection#LArFarUpstreamMaterialExitLayer"
                    )
            ## Persistency of test-beam layout
            if layout.startswith('ctb') or layout.startswith('tb_Tile2000_'):
                stream1.ItemList += ["TBElementContainer#*"]
Exemplo n.º 7
0
SkipEvents = 0
ServiceMgr.EventSelector.SkipEvents = SkipEvents

#--------------------------------------------------------------
# Setup Output
#--------------------------------------------------------------
if hasattr(runArgs, "outputRDO_MRGFile"): Out = runArgs.outputRDO_MRGFile

from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
try:
    StreamRDO = AthenaPoolOutputStream("StreamRDO", Out, True)
except:
    StreamRDO = AthenaPoolOutputStream("StreamRDO", "DidNotSetOutputName.root",
                                       True)
StreamRDO.TakeItemsFromInput = TRUE
StreamRDO.ForceRead = TRUE
#force read of output data objs
# The next line is an example on how to exclude clid's if they are causing a  problem
#StreamRDO.ExcludeList = ['6421#*']

# Look for lists of filter algorithms
try:
    StreamRDO.AcceptAlgs = AcceptList
except:
    print "No accept algs indicated in AcceptList"
try:
    StreamRDO.RequireAlgs = RequireList
except:
    print "No accept algs indicated in RequireList"
try:
    StreamRDO.VetoAlgs = VetoList
#services that need to be reinit
ServiceMgr.MultipleEventLoopMgr.ToBeReinitialized = ["StoreGateSvc", "DetectorStore"]

#
# I/O stuff
#
ServiceMgr.EventSelector.InputCollections = [
    "root://eosatlas//eos/atlas/atlascerngroupdisk/proj-sit/digitization/RTT/mc10/mc10_7TeV.005008.CavernInput.merge.HITS.e4_e607_s951_s952_tid170551_00/HITS.170551._000011.pool.root.1" ]
# Set up for seeking.
ServiceMgr.EventSelector.CollectionType = "SeekableROOT"

include( "AthenaPoolCnvSvc/WriteAthenaPool_jobOptions.py" )
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
Stream1 = AthenaPoolOutputStream( "Stream1" )
Stream1.OutputFile = "MultiplePassFile.root"; # ** mandatory parameter ** // The output file name
Stream1.ForceRead=TRUE  #force read of output data objs
Stream1.ItemList=["EventInfo#*", "McEventCollection#*"]

#--------------------------------------------------------------
#--- Monitoring and Debug printouts
#--------------------------------------------------------------

MessageSvc.OutputLevel      = INFO
Stream1.OutputLevel = DEBUG
#StoreGateSvc.Dump=1
#StoreGateSvc.OutputLevel=DEBUG

from AthenaCommon.AppMgr import theAuditorSvc
from GaudiAud.GaudiAudConf import ChronoAuditor
theAuditorSvc += ChronoAuditor()
Exemplo n.º 9
0
#--------------------------------------------------------------
# Write Digitization MetaData
#--------------------------------------------------------------
from Digitization.DigitizationWriteMetaData import writeDigitizationMetadata

writeDigitizationMetadata()

#--------------------------------------------------------------
# Pool Output (Change this to use a different file)
#--------------------------------------------------------------
if DetFlags.writeRDOPool.any_on():
    from AthenaCommon.AthenaCommonFlags import athenaCommonFlags
    from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
    streamRDO = AthenaPoolOutputStream(
        "StreamRDO", athenaCommonFlags.PoolRDOOutput.get_Value(), asAlg=True)
    streamRDO.ForceRead = True
    from Digitization.DigiOutput import getStreamRDO_ItemList
    streamRDO.ItemList = getStreamRDO_ItemList(logDigitization_flags)
    streamRDO.AcceptAlgs += [digitizationFlags.digiSteeringConf.get_Value()]
    streamRDO.OutputFile = athenaCommonFlags.PoolRDOOutput()
    if athenaCommonFlags.UseLZMA():
        ServiceMgr.AthenaPoolCnvSvc.PoolAttributes += [
            "DatabaseName = '" + athenaCommonFlags.PoolRDOOutput() +
            "'; COMPRESSION_ALGORITHM = '2'"
        ]
        ServiceMgr.AthenaPoolCnvSvc.PoolAttributes += [
            "DatabaseName = '" + athenaCommonFlags.PoolRDOOutput() +
            "'; COMPRESSION_LEVEL = '1'"
        ]
    else:
        ServiceMgr.AthenaPoolCnvSvc.PoolAttributes += [
Exemplo n.º 10
0
ServiceMgr.EventSelector.SkipEvents = SkipEvents

#--------------------------------------------------------------
# Setup Output
#--------------------------------------------------------------
if not hasattr(runArgs, "outputHitsFile"):
    raise RuntimeError("No outputHitsFile provided.")
Out = runArgs.outputHitsFile
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
try:
    StreamHITS = AthenaPoolOutputStream("StreamHITS", Out, True)
except:
    StreamHITS = AthenaPoolOutputStream("StreamHITS",
                                        "DidNotSetOutputName.root", True)
StreamHITS.TakeItemsFromInput = TRUE
StreamHITS.ForceRead = TRUE
#force read of output data objs
# The next line is an example on how to exclude clid's if they are causing a  problem
#StreamHITS.ExcludeList = ['6421#*']

# Look for lists of filter algorithms
try:
    StreamHITS.AcceptAlgs = AcceptList
except:
    print "No accept algs indicated in AcceptList"
try:
    StreamHITS.RequireAlgs = RequireList
except:
    print "No accept algs indicated in RequireList"
try:
    StreamHITS.VetoAlgs = VetoList
Exemplo n.º 11
0
                                           InDetKeys.OutputESDFileName(),
                                           asAlg=True)
        # --- save MC collections if truth turned on
        if InDetFlags.doTruth():
            StreamESD.ItemList += truthList
        # ---- load list of objects
        include("InDetRecExample/WriteInDetESD.py")
        StreamESD.ItemList += InDetESDList
        StreamESD.ItemList += [
            '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
StreamDPD = AthenaPoolOutputStream("StreamDPD")

## configure the thinning service
from AthenaServices.Configurables import ThinningSvc

svcMgr += ThinningSvc(Streams=['StreamDPD'])

######################Write Item List##########################################
StreamDPD.ItemList = ['EventInfo#*', 'TrackRecordCollection#*']
StreamDPD.ItemList += ['ElectronContainer#ElectronAODCollection']
StreamDPD.ItemList += ['egDetailContainer#egDetailAOD']
StreamDPD.ItemList += ['Analysis::MuonContainer#StacoMuonCollection']
StreamDPD.ItemList += ['JetKeyDescriptor#JetKeyMap']
StreamDPD.ItemList += ['JetCollection#Cone4H1TowerJets']
StreamDPD.ItemList += ['JetCollection#Cone4TruthJets']
StreamDPD.ItemList += ['Rec::TrackParticleContainer#TrackParticleCandidate']
StreamDPD.ItemList += ['Rec::TrackParticleContainer#StacoTrackParticles']
StreamDPD.ItemList += ['VxContainer#*']

StreamDPD.ForceRead = TRUE
#force read of output data objs

#electron Filter
StreamDPD.AcceptAlgs = ["ElectronFilter"]

#=============================================================
#
# End of job options file
#
###############################################################
Exemplo n.º 13
0
    from McParticleAlgs.McParticleAlgsConf import McAodTupleWriter
    topSequence += McAodTupleWriter(McWriter=McAodTupleWriterTool(
        TruthParticles="SpclMC", Output=TUPLEFILENAME),
                                    OutputLevel=DEBUG)
    pass

#--------------------------------------------------------------
# Output options
#--------------------------------------------------------------
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
outStream = AthenaPoolOutputStream("StreamAOD")
import os
outStream.OutputFile = os.path.join( \
    os.path.dirname(INPUT[0]),
    "reaccessed.%s" % os.path.basename(INPUT[0]) )
outStream.ForceRead = True

outStream.ItemList += ["EventInfo#McEventInfo"]
outStream.ItemList += ["McEventCollection#GEN_EVENT"]
outStream.ItemList += ["McEventCollection#GEN_AOD"]
outStream.ItemList += ["TruthParticleContainer#SpclMC"]
outStream.ItemList += ["TruthEtIsolationsContainer#TruthEtIsol_GEN_EVENT"]
outStream.ItemList += ["TruthEtIsolationsContainer#TruthEtIsol_GEN_AOD"]

## tweak the default commit interval
svcMgr.AthenaPoolCnvSvc.CommitInterval = 100

svcMgr.MessageSvc.defaultLimit = 4000000
svcMgr.MessageSvc.OutputLevel = ERROR

#==============================================================
Exemplo n.º 14
0
    pass
if ALGMODE == 'py':
    from McParticleTests.Lib import PyMcAodSymLinkTests as McAodSymLinkTests
else:
    from McParticleTests.McParticleTestsConf import McAodSymLinkTests
topSequence += McAodSymLinkTests(OutputLevel=DEBUG)

#---------------------------------------------------------------
# Pool Persistency
#---------------------------------------------------------------
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
outStream = AthenaPoolOutputStream("OutStream")
outStream.ItemList = ["EventInfo#*"]
outStream.ItemList += ["McEventCollection#GEN_EVENT"]
outStream.ItemList += ["TruthParticleContainer#SpclMC"]

if 'OUTPUT' not in dir():
    OUTPUT = "mc.aod.symlinktests.pool"
    pass
outStream.OutputFile = OUTPUT
outStream.EvtConversionSvc = "AthenaPoolCnvSvc"
outStream.ForceRead = True  #force read of output data objs

svcMgr.MessageSvc.defaultLimit = 4000000
svcMgr.MessageSvc.OutputLevel = ERROR
#==============================================================
#
# End of job options file
#
###############################################################
Exemplo n.º 15
0
#Stream1.ItemList += ["CaloCalibrationHitContainer#LArCalibrationHitActive",
#                     "CaloCalibrationHitContainer#LArCalibrationHitDeadMaterial",
#                     "CaloCalibrationHitContainer#LArCalibrationHitInactive",
#                     "CaloCalibrationHitContainer#TileCalibHitActiveCell",
#                     "CaloCalibrationHitContainer#TileCalibHitInactiveCell",
#                     "CaloCalibrationHitContainer#TileCalibHitDeadMaterial" ]
#CSC
Stream1.ItemList+=["CSCSimHitCollection#CSC_Hits"]
#MDT
Stream1.ItemList+=["MDTSimHitCollection#MDT_Hits"]
#RPC
Stream1.ItemList+=["RPCSimHitCollection#RPC_Hits"]
#TGC
Stream1.ItemList+=["TGCSimHitCollection#TGC_Hits"]

Stream1.ForceRead = True

#--------------------------------------------------------------
# the Tile, LAr and Calo detector description package
#--------------------------------------------------------------
from AtlasGeoModel import SetGeometryVersion
from AtlasGeoModel import GeoModelInit

include( "CaloIdCnv/CaloIdCnv_joboptions.py" )
include( "TileIdCnv/TileIdCnv_jobOptions.py" )
include( "LArDetDescr/LArDetDescr_joboptions.py" )
try:
    import MagFieldServices.SetupField
except:
    filterHitLog.warning("C++ magnetic field not available in this release.")
    include( "BFieldAth/BFieldAth_jobOptions.py" )
Exemplo n.º 16
0
#--------------------------------------------------------------
#RootHistSvc = Service( "RootHistSvc" )
#HbookHistSvc.NPAWC = 1500000
#HistogramPersistencySvc = Service( "HistogramPersistencySvc" )
#HistogramPersistencySvc.OutputFile  = "histo.root"
svcMgr.NTupleSvc.Output = [
    "FILE1  DATAFILE='pythiaB.root'  OPT='NEW'  TYP='ROOT'"
]

###############################################################
# Add POOL persistency

from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
include("GeneratorObjectsAthenaPool/GeneratorObjectsAthenaPool_joboptions.py")

Stream1 = AthenaPoolOutputStream("Stream1")
Stream1.ForceRead = TRUE
Stream1.OutputFile = "pythiaB.pool.root"
# 2101 = EventInfo
# 133273 = MCTruth (HepMC)
Stream1.ItemList += ["2101#*", "133273#*"]

include("AthenaSealSvc/AthenaSealSvc_joboptions.py")
AthenaSealSvc.CheckDictionary = TRUE

#==============================================================
#
# End of job options file
#
###############################################################
elif hasattr(runArgs, "outputEVNT_PreFile"):
    poolFile = runArgs.outputEVNT_PreFile
else:
    raise RuntimeError(
        "Output pool file, either EVNT or EVNT_Pre, is not known.")

#StreamEVGEN = AthenaPoolOutputStream("StreamEVGEN", runArgs.outputEVNTFile)
StreamEVGEN = AthenaPoolOutputStream("StreamEVGEN", poolFile)
if hasattr(runArgs, "inputEVNT_PreFile"):
    #  import AthenaPoolCnvSvc.ReadAthenaPool
    #  from AthenaCommon.AthenaCommonFlags import athenaCommonFlags
    #  athenaCommonFlags.PoolEvgenInput.set_Value_and_Lock( runArgs.inputEVNT_PreFile )
    svcMgr.EventSelector.InputCollections = runArgs.inputEVNT_PreFile
    StreamEVGEN.TakeItemsFromInput = True

StreamEVGEN.ForceRead = True
StreamEVGEN.ItemList += ["EventInfo#*", "McEventCollection#*"]
StreamEVGEN.RequireAlgs += ["EvgenFilterSeq"]
## Used for pile-up (remove dynamic variables except flavour labels)
if evgenConfig.saveJets:
    StreamEVGEN.ItemList += ["xAOD::JetContainer_v1#*"]
    StreamEVGEN.ItemList += [
        "xAOD::JetAuxContainer_v1#*.TruthLabelID.PartonTruthLabelID"
    ]

## Set the run numbers
svcMgr.EventSelector.RunNumber = runArgs.runNumber
# TODO: set EventType::mc_channel_number = runArgs.runNumber

## Handle beam info
import EventInfoMgt.EventInfoMgtInit
    SkipEvents = runArgs.skipEvents
ServiceMgr.EventSelector.SkipEvents = SkipEvents

#--------------------------------------------------------------
# Setup Output
#--------------------------------------------------------------
if not hasattr(runArgs,"outputHitsFile"):
    raise RuntimeError("No outputHitsFile provided.")
Out = runArgs.outputHitsFile 
from AthenaPoolCnvSvc.WriteAthenaPool import AthenaPoolOutputStream
try: 
  StreamHITS = AthenaPoolOutputStream( "StreamHITS", Out, True )
except:
  StreamHITS = AthenaPoolOutputStream( "StreamHITS", "DidNotSetOutputName.root", True )
StreamHITS.TakeItemsFromInput=TRUE;
StreamHITS.ForceRead=TRUE;  #force read of output data objs
# The next line is an example on how to exclude clid's if they are causing a  problem
#StreamHITS.ExcludeList = ['6421#*']

# Look for lists of filter algorithms
try:
  StreamHITS.AcceptAlgs = AcceptList
except:
  print "No accept algs indicated in AcceptList"
try:
  StreamHITS.RequireAlgs = RequireList
except:
  print "No accept algs indicated in RequireList"
try:
  StreamHITS.VetoAlgs = VetoList
except:
Exemplo n.º 19
0
  if doWriteESD:
    # --- create stream
    StreamESD            = AthenaPoolOutputStream ( "StreamESD", InDetKeys.OutputESDFileName(),asAlg=True) 
    # --- save MC collections if truth turned on
    if InDetFlags.doTruth():
      StreamESD.ItemList += truthList
    # ---- load list of objects
    include ( "InDetRecExample/WriteInDetESD.py" )
    StreamESD.ItemList  += InDetESDList
    StreamESD.ItemList  += [ '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" )