python.BeamSpotUpdate Namespace Reference

Classes
class	BeamSpotWriteAlg
class	Config

Functions
	setBeamSpot (run, lb, x, y, z, status=7, dbname='sqlite://;schema=beampos.db;dbname=CONDBR2', tag='IndetBeamposOnl-HLT-UPD1-001-00')
	addFolderUpdate (event)
	modify (event)
	setup ()
	run (flags)

Variables
list	folderList = []
	log = logging.getLogger(__name__)
	parser = argparse.ArgumentParser(description=__doc__)
	metavar
	nargs
	help
	type
	int
	default
	str
	args = parser.parse_args()
	dr = EventStorage.pickDataReader(args.file[0])
	output
int	i = 0
	newevt = modify(event)

Detailed Description

Module to test beamspot updates using a local sqlite file. Two steps are necessary:

1) Create a modified bytestream file that contains the correct COOL update information
   in the CTP fragment. To do this simply run this script with a RAW file and it will
   write a modified file into the current directory with beamspot updates on LBs as
   specified in Config.lb_updateBeamspot.

2) Configure a job with the BeamSpotWriteAlg in the topSequence. This algorithm will
   write a new beamspot into the sqlite file one LB before the actual update is
   triggered via the CTP fragment. See TrigP1Test/python/BeamSpotReader.py
   for an example.

Function Documentation

addFolderUpdate()

python.BeamSpotUpdate.addFolderUpdate

(

event

)

Add COOL folder update to event

Definition at line 78 of file BeamSpotUpdate.py.

def addFolderUpdate(event):
   """Add COOL folder update to event"""
   
   ctp_robs = [rob for rob in event if rob.source_id().subdetector_id()==eformat.helper.SubDetector.TDAQ_CTP]
 
   if len(ctp_robs)==0:
      log.error("Cannot find CTP ROB in event")
      return event
 
   # Copy event except CTP ROBs
   new_event = eformat.write.FullEventFragment()
   new_event.copy_header(event)
   for r in event:
      if r.source_id().subdetector_id() != eformat.helper.SubDetector.TDAQ_CTP:
         new_event.append(eformat.write.ROBFragment(r))
 
   # Set the CTP extra payload      
   for rob in ctp_robs:
      x = CTPfragment.ExtraPayload()  # start from empty payload object
      for f in folderList: x.updateFolder(f)
      new_ctp_rob = CTPfragment.setHltExtraPayloadWords(rob, [d for d in x.serialize()])
      new_event.append(eformat.write.ROBFragment(new_ctp_rob))
      
   return new_event
 
   

modify()

python.BeamSpotUpdate.modify

(

event

)

Definition at line 104 of file BeamSpotUpdate.py.

def modify(event):
   event = eformat.write.FullEventFragment(event)
 
   # Modify LB and HLT counter in CTP fragment
   newevt = trigbs_modifyEvent.modify(event)
   lb = newevt.lumi_block()
 
   # Write conditions update into CTP fragment
   if lb in Config.lb_updateBeamspot:
      lb_for_update,status = Config.lb_updateBeamspot.pop(lb)
 
      fe = CTPfragment.FolderEntry()
      fe.folderIndex = 0  # /Indet/Onl/Beampos
      fe.lumiBlock = lb_for_update
      folderList.append(fe)
      log.info('Added COOL folder update to event: folderIndex=%d, LB=%d', fe.folderIndex,fe.lumiBlock)
      
   return addFolderUpdate(newevt).readonly()
 
 

run()

python.BeamSpotUpdate.run

(

flags

)

CA cfg function to be used from athenaHLT

Definition at line 139 of file BeamSpotUpdate.py.

def run(flags):
   """CA cfg function to be used from athenaHLT"""
 
   from AthenaCommon.Constants import DEBUG
   from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
   from AthenaConfiguration.ComponentFactory import CompFactory
   from IOVDbSvc.IOVDbSvcConfig import addFolders
 
   from pathlib import Path
 
   flags.lock()
   cfg = ComponentAccumulator()
 
   # addFolders requires at least an empty sqlite file
   Path('beampos.db').touch()
 
   # These folders are filled in Testing/condStopStart.trans
   cfg.merge( addFolders(flags, '/Indet/Onl/Beampos <key>/Indet/Beampos</key>',
                         detDb='beampos.db',
                         tag='IndetBeamposOnl-HLT-UPD1-001-00',
                         className='AthenaAttributeList',
                         extensible=True) )
 
   cfg.addEventAlgo( BeamSpotWriteAlg() )
   cfg.addEventAlgo( CompFactory.InDet.InDetBeamSpotReader(VxContainer = "") )
   cfg.addCondAlgo( CompFactory.BeamSpotCondAlg(OutputLevel = DEBUG) )
 
   return cfg
 
 
# For standalone running and writing a new output file

setBeamSpot()

python.BeamSpotUpdate.setBeamSpot	(		run,
			lb,
			x,
			y,
			z,
			status = 7,
			dbname = 'sqlite://;schema=beampos.db;dbname=CONDBR2',
			tag = 'IndetBeamposOnl-HLT-UPD1-001-00' )

Definition at line 62 of file BeamSpotUpdate.py.

                tag='IndetBeamposOnl-HLT-UPD1-001-00'):
 
   log.info('============================= Creating new beamspot in COOL ===================================')
   log.info('run=%d, lb=%d, x=%f, y=%f, z=%f', run, lb, x, y, z)
   sys.stdout.flush()
   os.system("beamSpotOnl_set.py --output=sqlite:beampos.db --run=%d --lbn=%d %d %f %f %f" % (run,lb,status,x,y,z))
   if log.level<=logging.DEBUG:
      log.info('Current content of beampos.db sqlite file:')
      os.system("AtlCoolConsole.py 'sqlite://;schema=beampos.db;dbname=CONDBR2' <<< 'more Indet/Onl/Beampos'")
   log.info('===============================================================================================')
   sys.stdout.flush()   
 
 

setup()

python.BeamSpotUpdate.setup

(

)

Initial setup

Definition at line 124 of file BeamSpotUpdate.py.

def setup():
   """Initial setup"""
 
   log.info('Will perform beamspot udpate on these LBs (LB,status): %s', sorted(Config.lb_updateBeamspot.values()))
 
   # Delete any previous sqlite file
   try:
      os.remove('beampos.db')
   except OSError:
      pass
 
   # Create an open-ended IOV with a default beamspot
   setBeamSpot(1,0,0.06,1.06,-4.6,4)
 
 

Variable Documentation

args

python.BeamSpotUpdate.args = parser.parse_args()

Definition at line 179 of file BeamSpotUpdate.py.

default

python.BeamSpotUpdate.default

Definition at line 176 of file BeamSpotUpdate.py.

dr

python.BeamSpotUpdate.dr = EventStorage.pickDataReader(args.file[0])

Definition at line 180 of file BeamSpotUpdate.py.

folderList

list python.BeamSpotUpdate.folderList = []

Definition at line 35 of file BeamSpotUpdate.py.

help

python.BeamSpotUpdate.help

Definition at line 175 of file BeamSpotUpdate.py.

i

int python.BeamSpotUpdate.i = 0

Definition at line 190 of file BeamSpotUpdate.py.

int

python.BeamSpotUpdate.int

Definition at line 176 of file BeamSpotUpdate.py.

log

python.BeamSpotUpdate.log = logging.getLogger(__name__)

Definition at line 36 of file BeamSpotUpdate.py.

metavar

python.BeamSpotUpdate.metavar

Definition at line 175 of file BeamSpotUpdate.py.

nargs

python.BeamSpotUpdate.nargs

Definition at line 175 of file BeamSpotUpdate.py.

newevt

python.BeamSpotUpdate.newevt = modify(event)

Definition at line 195 of file BeamSpotUpdate.py.

output

python.BeamSpotUpdate.output

Initial value:

=  eformat.ostream(core_name = args.output or dr.fileNameCore(),
                            run_number = dr.runNumber(),
                            trigger_type = dr.triggerType(),
                            detector_mask = dr.detectorMask(),
                            beam_type = dr.beamType(),
                            beam_energy = dr.beamEnergy(),
                            meta_data_strings=dr.freeMetaDataStrings(),
                            compression=dr.compression())

Definition at line 181 of file BeamSpotUpdate.py.

parser

python.BeamSpotUpdate.parser = argparse.ArgumentParser(description=__doc__)

Definition at line 174 of file BeamSpotUpdate.py.

str

python.BeamSpotUpdate.str

Definition at line 177 of file BeamSpotUpdate.py.

type

python.BeamSpotUpdate.type

Definition at line 176 of file BeamSpotUpdate.py.