testTrending.py

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

__facility__ = "Online"
__abstract__ = "Housekeeping trending tester module"
__author__   = "J. Panetta <panetta@SLAC.Stanford.edu> SLAC - GLAST LAT I&T/Online"
__date__     = "7/20/2004"
__version__  = "$Revision: 1.3 $"
__credits__  = "SLAC"


import LATTE.copyright_SLAC

from LATTE.monitoring.Receiver import *
import logging
import qt


class QtReceiver(qt.QObject,UDPReceiver):
    class TrendUpdateEvent(qt.QCustomEvent):
        def __init__(self,trendables):
            qt.QCustomEvent.__init__(self,qt.QEvent.MaxUser-20)
            self.trendables=trendables
    def __init__(self,*args):
        qt.QObject.__init__(self,*args)
        Receiver.__init__(self)
    def process(self,trendables):
        qt.QApplication.postEvent(self,QtReceiver.TrendUpdateEvent(trendables))
    def customEvent(self,event):
        self.emit(qt.PYSIGNAL('TrendUpdate(trendables)'),(event.trendables,))

class TrendTestGUI(qt.QTextEdit):
    def __init__(self,*args):
        qt.QTextEdit.__init__(self,*args)
        self.setTextFormat(qt.Qt.LogText)
        self.setMaxLogLines(1000)
    def update(self,trendables):
        for t in trendables:
            self.append(str(t))

class TrendTree(qt.QListView):
    def __init__(self,*args):
        qt.QListView.__init__(self,*args)
        self.columns=['Name','Time']
        self.addColumn('Name')
        self.addColumn('Time')
        self.setRootIsDecorated(1)
        self.setResizeMode(qt.QListView.AllColumns)
        self.setAllColumnsShowFocus(1)
        self.setShowSortIndicator(1)
        self.setSelectionMode(qt.QListView.Extended)
        self.__dynamic=1

    def isDynamic(self):
        return self.__dynamic

    def setDynamic(self,bool):
        self.__dynamic=bool

    def addLeaf(self,name):
        if not self.findItem(name,0):
            bname=self.getBranchName(name)
            branch=self.findItem(bname,0)
            if not branch:
                branch=qt.QListViewItem(self,bname)
            leaf=qt.QListViewItem(branch,name)
            return leaf

    def update(self,trendables):
        for t in trendables:
            leaf=self.findItem(t.name,0)
            if not leaf and self.__dynamic:
                leaf=self.addLeaf(t.name)
            if leaf:
                from time import ctime
                leaf.setText(1,ctime(t.time))
                for k in t.content.keys():
                    try:
                        c=self.columns.index(k)
                    except ValueError, e:
                        self.addColumn(k)
                        self.columns.append(k)
                        c=self.columns.index(k)
                    leaf.setText(c,str(t.content[k]))

    def getBranchName(self,name):
        return name.split('_')[0]


class TrendPlotter(qt.QWidget):
    def __init__(self,*args):
        qt.QWidget.__init__(self,*args)

        self.__ntuples={}
        self.__plots={}
        self.__views={}
        self.__defaultDepth=150
        self.__defaultLabels=['Time','Counts']

        self.__dynamic=1

        self.grid=qt.QGridLayout(self,1,4)
        self.grid.setAutoAdd(1)

    def isDynamic(self):
        return self.__dynamic
    def setDynamic(self,bool):
        self.__dynamic=bool

    def addPlot(self,trendable):
        name=trendable.name
        labels=self.__defaultLabels
        depth=self.__defaultDepth
        import sihippo
        # Create the Circular buffer to store the data
        ntc=sihippo.NTupleController.instance()
        nt=ntc.createCircularBuffer(labels)
        nt.reserve(depth)
        nt.setTitle(name)
        nt.setName(name)
        # Create the Plot
        dc=sihippo.DisplayController.instance()
        # plot=dc.createDisplay('XY Plot',nt,labels)
        plot=dc.createDisplay('Strip Chart',nt,labels)
        view=sihippo.QtViewWidget(self)
        view.setPlotter(plot)
        view.setDoubleBuffering(1)
        view.resize(200,100)
        view.setMinimumSize(qt.QSize(100,50))
        #self.grid.addWidget(view)
        view.show()

        self.__ntuples[name]=nt
        self.__plots[name]=plot
        self.__views[name]=view

    def update(self,trendables):
        for t in trendables:
            if t.name not in self.__views.keys() and self.__dynamic:
                self.addPlot(t)
            if t.name not in self.__views.keys():
                return
            x=t.time
            y=t.content['Raw Value']
            self.__ntuples[t.name].addRow([x,y])


# This is just a dumb way to control the TrendTree
class FancyTrendTree(qt.QVBox):
    def __init__(self,*args):
        qt.QVBox.__init__(self,*args)

        self.tree=TrendTree(self)

        self.menu=qt.QPopupMenu(self)
        self.menu.insertItem('Remove Selected',self.removeSelected)
        self.menu.insertItem('New Leaf',self.newTrend)
        self.menu.insertItem('Toggle Dynamic',self.toggleDynamic)

        self.connect(self.tree,
                     qt.SIGNAL('contextMenuRequested'\
                               +'(QListViewItem*,const QPoint&,int)')
                     ,self.contextMenu)

    def toggleDynamic(self):
        self.tree.setDynamic(not self.tree.isDynamic())

    def update(self,trendables):
        self.tree.update(trendables)

    def newTrend(self):
        name,ok=qt.QInputDialog.getText('New Leaf',
                                        'Name of new leaf',
                                        qt.QLineEdit.Normal,
                                        '',self)
        if ok:
            self.tree.addLeaf(str(name))


    def removeSelected(self):
        it=qt.QListViewItemIterator(self.tree,
                                    qt.QListViewItemIterator.Selected)
        while it.current():
            item=it.current()
            if item.parent():
                item.parent().takeItem(item)
            else:
                self.tree.takeItem(item)
            it+=1

    def contextMenu(self,item,point,pos):
        self.menu.popup(point)

###############################################################
def testQtReceiver(args):
    app=qt.QApplication(args)

    logging.basicConfig()
    logging.root.setLevel(logging.DEBUG)

    r=QtReceiver()
    udpServerName='testTrendingServer'
    udpServerAddr='54320@239.255.1.14'
    r.setUDPClient(udpServerName,udpServerAddr)

    # gui=TrendTestGUI()
    # gui=TrendTree()
    # gui=FancyTrendTree()
    gui=TrendPlotter()
    gui.connect(r,qt.PYSIGNAL('TrendUpdate(trendables)'),gui.update)

    app.connect(app,qt.SIGNAL("lastWindowClosed()"),
                app,qt.SLOT("quit()"))

    app.setMainWidget(gui)
    gui.show()

    r.setReceiving(1)
    app.exec_loop()
    r.shutdown()

def testReceiver():
    logging.basicConfig()
    logging.root.setLevel(logging.DEBUG)

    udpServerName='testTrendingServer'
    udpServerAddr='54320@239.255.1.14'

    r=UDPReceiver()
    r.setUDPClient(udpServerName,udpServerAddr)
    r.setReceiving(1)
    from time import sleep
    try:
        while 1:
            sleep(.1)
    except Exception, e:
        r.shutdown()

if __name__ == '__main__':
    import sys
    testQtReceiver(sys.argv)
    #testReceiver()

---