rcArchiver.py

#!/usr/local/bin/python
#
#                               Copyright 2004
#                                     by
#                        The Board of Trustees of the
#                     Leland Stanford Junior University.
#                            All rights reserved.
#

__facility__ = "Online"
__abstract__ = "Event archiver class"
__author__   = "S. Tuvi <stuvi@SLAC.Stanford.edu> SLAC - GLAST LAT I&T/Online"
__date__     = "02/13/2003"
__version__  = "$Revision: 2.3 $"
__credits__  = "SLAC"

from time import gmtime, strftime
import os
from struct import unpack
from struct import pack

from LATTE.client.gEvtCli import EvtCli

MODE_CREATE = 0
MODE_APPEND = 1
MODE_READONLY = 2

class rcArchiver(object):
  """\brief Event Archiver

  Default event archiver class to store event data to a file.
  This class can be inherited to create a custom event class and
  using it from a RunControl class by calling the setArchiverClass
  method.
  """

  def __init__(self, filePath=None, fileName=None, mode=MODE_CREATE, userId=123):
    if fileName is None:
      ts   = strftime('%y%m%d%H%M%S', gmtime())
      self.__fileName = rcArchiver.getFileNameFromTS(ts)
    else:
      self.__fileName = fileName
    if filePath is None:
      self.__filePath = ""
    else:
      self.__filePath = filePath + os.sep
    self.__mode = mode
    if mode == MODE_CREATE:
      fileMode = 'w+b'
      self.__archFile = file(os.path.join(self.__filePath, self.__fileName), fileMode)
    elif mode == MODE_APPEND:
      fileMode = 'a+b'
      self.__archFile = file(os.path.join(self.__filePath, self.__fileName), fileMode)
    elif mode == MODE_READONLY:
      fileMode = 'rb'
      self.__archFile = None
      self.__evtCli = EvtCli()
      self.__evtCli.open(os.path.join(self.__filePath, self.__fileName), fileMode)
    else:
      raise RuntimeError, "Invalid archive file mode"

  def write(self, rawEventData):
    if not self.__archFile.closed:
      self.__archFile.write(rawEventData)

  def writeTuple(self, tuple):
    for item in tuple:
      self.__archFile.write(item)

  def read(self):
    try:
      (header, dat) = self.__evtCli.read()
      return dat
    except:
      return None

  def close(self):
    if self.__mode == MODE_READONLY:
      self.__evtCli.close()
    else:
      self.__archFile.close()

  def isClosed(self):
    return self.__archFile.closed

  def getFileName(self):
      return self.__fileName

  def getFilePath(self):
    return self.__filePath

  def getFileNameFromTS(ts):
    return ts + ".ldf"
  getFileNameFromTS = staticmethod(getFileNameFromTS)

---