In This Package:

SpadeAlgo.py

Go to the documentation of this file.

from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE
from DybPython.DybPythonAlg import DybPythonAlg

class SpadeAlgo(DybPythonAlg):

    SITE_SAB = 32
    
    def __init__(self, name="SpadeAlg"):
        DybPythonAlg.__init__(self, name)
        self.fss = None
        self.opened = []
        self.tomark = []
        self.marked = []
        import DybPython.Control
        DybPython.Control.nuwa.spadeSvc = self
        return


    def initialize(self):
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status

        self.site = None
        self.startTime = None
        self.endTime = None
        self.fss = self.svc('IRootIOFileStateSvc','RootIOFileStateSvc')

        self.debug('Initialized "%s"' % self.name())
        return status


    def execute(self):
        # Get this event's time from the TES
        #eventTime = self.get('/Event/...')
        header = self.get('/Event/Readout/ReadoutHeader')
        if None == header:
            self.debug("Got no ReadoutHeader")
            return SUCCESS
        from time import gmtime, strftime, timezone
        data = header.readout()
        if not data:
            data = header.daqCrate()
        if not data:
            self.warning("got no readout nor daqcrate from ReadoutHeader")
            return SUCCESS

        eventTime = data.triggerTime().GetSeconds()

        #self.debug('Site Value: %s'% header.context().GetSite())
        #self.debug( "event time %s thish is %s" % \
        #     ( eventTime strftime('%Y-%m-%dT%H:%M:%S', gmtime(eventTime))))
        if None == self.startTime:
            self.site = header.context().GetSite()
            self.startTime = eventTime
        self.endTime = eventTime

        self._update_output_files()        
        self._make_sems(self.tomark)
        return SUCCESS


    def finalize(self):
        # NB: this is not a race condition as we delay the .meta.xml
        # to .sem conversion until after the running stops.  This is
        # done in DybPython.Control.nuwa
        self.debug("final cleanup")
        self._update_output_files()        
        self._make_sems(self.tomark)
        self._make_sems(self.opened)
        return DybPythonAlg.finalize(self)


    def _make_sems(self,filenames):
        'Make sem files corresponding to all given filenames'
        self.debug('marking: %s'%filenames)

        # make copy so the .remove() later doesn't mess up what we are
        # iterating on
        filenames = list(filenames) 

        for fn in filenames:
            self._make_one_sem(fn)
            self.marked.append(fn)
            self.debug('marked= %s' % self.marked)
            self.debug('tomark= %s' % self.tomark)
            try:
                self.tomark.remove(fn)
            except ValueError:
                pass
            continue
        return
        

    def _listify(self,vector):
        'Turn an STL vector into a Python list'
        l = []
        siz = vector.size()
        for ind in range(siz):
            l.append(vector[ind])
        return l


    def _update_output_files(self):
        'Update list of opened and tomark files'
        self.opened = self._listify(self.fss.openOutput())

        closed = self._listify(self.fss.closedOutput())
        for fn in closed:
            if fn not in self.marked and fn not in self.tomark:
                self.debug('new file to mark: "%s"'%fn)
                self.tomark.append(fn)
                pass
            continue
        return

    def _make_one_sem(self,filename):
        'Make a sem file based on given filename'

        self.debug('making sem file for "%s"'%filename)

        from Metadata import Metadata
        from os.path import basename

        basePath = self.fileBase(filename)
        metadata = Metadata(basename(basePath))

        try:
            metadata.subcategory = self.subcategory
        except AttributeError:
            metadata.subcategory = 'unclassified'
            
        # default to SAB for MDC code
        if None != self.site:
            metadata.sensor = metadata.SENSOR_SAB

        from time import gmtime, strftime
        if None == self.startTime:
            start = gmtime()
        else:
            start = gmtime(int(self.startTime) - 1)
            pass
        metadata.startTime = start
        if None == self.endTime:
            end = gmtime()
        else:
            end = gmtime(int(self.endTime) + 1)
            pass
        metadata.endTime = end

        metadata.commandLine = self.getCommandLine()
        outFile = open(basePath + '.meta.xml', 'w')
        outFile.write(metadata.createXML().toxml())
        outFile.close()
        return SUCCESS

    def fileBase(self,
                 filename):
        cut = filename.rfind('.')
        if -1 == cut:
            return filename
        return  filename[:cut]

    def getCommandLine(self):
        import sys
        result = ''
        for arg in sys.argv:
            if 0 != len(result):
                result += ' '
            quote = ''
            if -1 != arg.find(' '):
                quote = '"'
                result += quote
            result += arg.replace('"',
                                  '\\"') + quote
        return result

---