# From Will Buttinger to suppress the event loop heartbeat as it is somewhat I/O hungry for
# no real gain in analysis scenarii
if not hasattr(svcMgr, theApp.EventLoop):
svcMgr += getattr(CfgMgr, theApp.EventLoop)()
evtloop = getattr(svcMgr, theApp.EventLoop)
try:
evtloop.EventPrintoutInterval = 10000
except Exception, err:
msg.info(
'failed suppressing event loop heartbeat. performances might be sub-par... sorry.'
)
pass
# Add in AthenaPoolAddressProviderSvc
if not hasattr(svcMgr, 'AthenaPoolAddressProviderSvc'):
svcMgr += CfgMgr.AthenaPoolAddressProviderSvc(
"AthenaPoolAddressProviderSvc")
svcMgr.ProxyProviderSvc.ProviderNames += ["AthenaPoolAddressProviderSvc"]
# Set up DataVector/DataProxyStorage backwards compatibility.
#from DataModelAthenaPool import DataModelCompatSvc
svcMgr += CfgMgr.DataModelCompatSvc()
msg.debug("Configuring Athena for reading POOL files... [DONE]")
return
## configuration at module import
_configureReadAthenaPool()
## clean-up: avoid running multiple times this method
del _configureReadAthenaPool
def _configureReadAthenaPoolDouble():
""" Helper method to configure Athena to read back POOL files """
from AthenaCommon.Logging import logging
msg = logging.getLogger( 'configureReadAthenaPool' )
msg.debug( "Configuring Athena for reading POOL files..." )
from AthenaCommon import CfgMgr
from AthenaCommon.AppMgr import ServiceMgr as svcMgr
from AthenaCommon.AppMgr import theApp
# Load the basic services
import AthenaPoolCnvSvc.AthenaPool
# Switch on TTreeCache for CollectionTree
svcMgr.AthenaPoolCnvSvc.InputPoolAttributes += [ "DatabaseName = '*'; ContainerName = 'CollectionTree'; TREE_CACHE = '-1'" ]
# Load ProxyProviderSvc
if not hasattr (svcMgr, 'ProxyProviderSvc'):
svcMgr += CfgMgr.ProxyProviderSvc()
# Add in MetaDataSvc
if not hasattr (svcMgr, 'MetaDataSvc'):
svcMgr += CfgMgr.MetaDataSvc ("MetaDataSvc")
svcMgr.ProxyProviderSvc.ProviderNames += [ "MetaDataSvc" ]
# Add in MetaData Stores
if not hasattr (svcMgr, 'InputMetaDataStore'):
svcMgr += CfgMgr.StoreGateSvc ("InputMetaDataStore")
if not hasattr (svcMgr, 'MetaDataStore'):
svcMgr += CfgMgr.StoreGateSvc ("MetaDataStore")
# Enable IOVDbSvc to read MetaData
svcMgr.MetaDataSvc.MetaDataContainer = "MetaDataHdr"
svcMgr.MetaDataSvc.MetaDataTools += [ "IOVDbMetaDataTool" ]
if not hasattr (svcMgr.ToolSvc, 'IOVDbMetaDataTool'):
svcMgr.ToolSvc += CfgMgr.IOVDbMetaDataTool()
# BS specifics
from OverlayCommonAlgs.OverlayFlags import overlayFlags
if overlayFlags.isDataOverlay():
# Load ByteStreamEventStorageInputSvc
if not hasattr (svcMgr, 'ByteStreamInputSvc'):
svcMgr += CfgMgr.ByteStreamEventStorageInputSvc ("ByteStreamInputSvc",
EventInfoKey=overlayFlags.bkgPrefix() + "EventInfo")
# Load ROBDataProviderSvc
if not hasattr (svcMgr, 'ROBDataProviderSvc'):
svcMgr += CfgMgr.ROBDataProviderSvc ("ROBDataProviderSvc")
# Load ByteStreamCnvSvc
if not hasattr (svcMgr, 'ByteStreamCnvSvc'):
svcMgr += CfgMgr.ByteStreamCnvSvc ("ByteStreamCnvSvc")
svcMgr.EventPersistencySvc.CnvServices += [ "ByteStreamCnvSvc" ]
# Add in ByteStreamAddressProviderSvc
if not hasattr (svcMgr, 'ByteStreamAddressProviderSvc'):
svcMgr += CfgMgr.ByteStreamAddressProviderSvc ("ByteStreamAddressProviderSvc")
svcMgr.ProxyProviderSvc.ProviderNames += [ "ByteStreamAddressProviderSvc" ]
# Enable ByteStream to read MetaData
svcMgr.MetaDataSvc.MetaDataTools += [ "ByteStreamMetadataTool" ]
if not hasattr (svcMgr.ToolSvc, 'ByteStreamMetadataTool'):
svcMgr.ToolSvc += CfgMgr.ByteStreamMetadataTool()
# Add in EventSelector
if overlayFlags.isDataOverlay():
svcMgr += CfgMgr.EventSelectorByteStream("SecondaryEventSelector", IsSecondary=True, ByteStreamInputSvc="ByteStreamInputSvc")
else:
svcMgr += CfgMgr.EventSelectorAthenaPool("SecondaryEventSelector", IsSecondary=True)
svcMgr += CfgMgr.DoubleEventSelectorAthenaPool("DoubleEventSelector")
theApp.EvtSel = "DoubleEventSelectorAthenaPool/DoubleEventSelector"
# Add in AthenaPoolAddressProviderSvc
if not hasattr (svcMgr, 'AthenaPoolAddressProviderSvcPrimary'):
svcMgr += CfgMgr.AthenaPoolAddressProviderSvc ("AthenaPoolAddressProviderSvcPrimary")
svcMgr.ProxyProviderSvc.ProviderNames += [ "AthenaPoolAddressProviderSvc/AthenaPoolAddressProviderSvcPrimary" ]
svcMgr.AthenaPoolAddressProviderSvcPrimary.DataHeaderKey = "EventSelector"
if not overlayFlags.isDataOverlay():
svcMgr.AthenaPoolAddressProviderSvcPrimary.AttributeListKey = "Input"
if not hasattr (svcMgr, 'AthenaPoolAddressProviderSvcSecondary'):
svcMgr += CfgMgr.AthenaPoolAddressProviderSvc ("AthenaPoolAddressProviderSvcSecondary")
svcMgr.ProxyProviderSvc.ProviderNames += [ "AthenaPoolAddressProviderSvc/AthenaPoolAddressProviderSvcSecondary" ]
svcMgr.AthenaPoolAddressProviderSvcSecondary.DataHeaderKey = "SecondaryEventSelector"
# Set up DataVector/DataProxyStorage backwards compatibility.
#from DataModelAthenaPool import DataModelCompatSvc
svcMgr += CfgMgr.DataModelCompatSvc()
# Always want AddressRemappingSvc, since that's responsible for suppressing
# the read of objects that are produced by a WriteHandle.
from SGComps.AddressRemappingSvc import getAddressRemappingSvc
getAddressRemappingSvc()
msg.debug( "Configuring Athena for reading two POOL files concurrently... [DONE]" )
return
def _configureReadAthenaPool():
""" Helper method to configure Athena to read back POOL files """
from AthenaCommon.Logging import logging
msg = logging.getLogger('configureReadAthenaPool')
msg.debug("Configuring Athena for reading POOL files...")
from AthenaCommon import CfgMgr
from AthenaCommon.AppMgr import ServiceMgr as svcMgr
from AthenaCommon.AppMgr import theApp
# Load the basic services
import AthenaPoolCnvSvc.AthenaPool
# Switch on TTreeCache for CollectionTree
svcMgr.AthenaPoolCnvSvc.InputPoolAttributes += [
"DatabaseName = '*'; ContainerName = 'CollectionTree'; TREE_CACHE = '-1'"
]
# Load ProxyProviderSvc
if not hasattr(svcMgr, 'ProxyProviderSvc'):
svcMgr += CfgMgr.ProxyProviderSvc()
# Add in MetaDataSvc
if not hasattr(svcMgr, 'MetaDataSvc'):
svcMgr += CfgMgr.MetaDataSvc("MetaDataSvc")
svcMgr.ProxyProviderSvc.ProviderNames += ["MetaDataSvc"]
# Add in MetaData Stores
if not hasattr(svcMgr, 'InputMetaDataStore'):
svcMgr += CfgMgr.StoreGateSvc("InputMetaDataStore")
if not hasattr(svcMgr, 'MetaDataStore'):
svcMgr += CfgMgr.StoreGateSvc("MetaDataStore")
# Enable IOVDbSvc to read MetaData
svcMgr.MetaDataSvc.MetaDataContainer = "MetaDataHdr"
svcMgr.MetaDataSvc.MetaDataTools += ["IOVDbMetaDataTool"]
if not hasattr(svcMgr.ToolSvc, 'IOVDbMetaDataTool'):
svcMgr.ToolSvc += CfgMgr.IOVDbMetaDataTool()
# Add in EventSelector
svcMgr += CfgMgr.EventSelectorAthenaPool("EventSelector")
#default InputCollections to FilesInput value of AthenaCommonFlags
from AthenaCommon.JobProperties import jobproperties as jps
svcMgr.EventSelector.InputCollections = jps.AthenaCommonFlags.FilesInput()
_n = svcMgr.EventSelector.getFullJobOptName()
theApp.EvtSel = _n
del _n
# For Analysis release use lower heartbeat
import os
if "AthAnalysisBase" in os.environ.get('CMTEXTRATAGS', ""):
# From Will Buttinger to suppress the event loop heartbeat as it is somewhat I/O hungry for
# no real gain in analysis scenarii
if not hasattr(svcMgr, theApp.EventLoop):
svcMgr += getattr(CfgMgr, theApp.EventLoop)()
evtloop = getattr(svcMgr, theApp.EventLoop)
try:
evtloop.EventPrintoutInterval = 10000
except Exception:
msg.info(
'failed suppressing event loop heartbeat. performances might be sub-par... sorry.'
)
pass
# Add in AthenaPoolAddressProviderSvc
if not hasattr(svcMgr, 'AthenaPoolAddressProviderSvc'):
svcMgr += CfgMgr.AthenaPoolAddressProviderSvc(
"AthenaPoolAddressProviderSvc")
svcMgr.ProxyProviderSvc.ProviderNames += ["AthenaPoolAddressProviderSvc"]
# Set up DataVector/DataProxyStorage backwards compatibility.
#from DataModelAthenaPool import DataModelCompatSvc
svcMgr += CfgMgr.DataModelCompatSvc()
# Always want AddressRemappingSvc, since that's responsible for suppressing
# the read of objects that are produced by a WriteHandle.
from SGComps.AddressRemappingSvc import getAddressRemappingSvc
getAddressRemappingSvc()
msg.debug("Configuring Athena for reading POOL files... [DONE]")
return
