source: qtprobe/traits_probe.py @ 814

#!/usr/bin/env python

########### SVN repository information ###################
# $Date: 2012-04-22 23:49:05 +0000 (Sun, 22 Apr 2012) $
# $Author: jemian $
# $Revision: 814 $
# $URL$
# $Id: traits_probe.py 814 2012-04-22 23:49:05Z jemian $
########### SVN repository information ###################

import argparse
import epics
import sys                  #@UnusedImport

choose_Qt4 = True
#choose_Qt4 = False
from traits.etsconfig.api import ETSConfig
if choose_Qt4:
    ETSConfig.toolkit = 'qt4'
else:
    ETSConfig.toolkit = 'wx'

from traits.api import *    #@UnusedWildImport
from traitsui.api import *  #@UnusedWildImport

from traitsui.key_bindings \
    import KeyBinding, KeyBindings


key_bindings = KeyBindings(
    KeyBinding( binding1    = 'Enter',
                description = 'Connect to new PV',
                method_name = 'connect' ),
)


# TraitsUI Handler class for bound methods
class CodeHandler ( Handler ):
    
    def connect ( self, info ):
        if len(info.object.name) > 0:
            try:
                if info.object._chid is not None:
                    info.object._chid.cancel_callback()
                    info.object._chid.disconnect()
            except:
                pass
            info.object._chid = epics.PV(str(info.object.name), 
                                         callback=info.object.do_callback)


class Probe( HasTraits ):
    name = Str
    value = Str
    _chid = Instance( epics.PV, value = None )

    def do_callback(self, value=None, **kwds):
        "simple monitor callback"
        self.value = str(value)

    traits_view = View(
        Item('name'),
        Readonly('value'),
        resizable=True,
        width = 250,
        title='Traits-based EPICS Probe',
        key_bindings = key_bindings,
        handler   = CodeHandler()
    )

if __name__ == '__main__':
    #pv = ''
    #if len(sys.argv) == 2:
    #    pv = sys.argv[1]
    parser = argparse.ArgumentParser(description="traits_probe")
    parser.add_argument('pv', action='store', nargs='?', help="EPICS PV name", default="EpicsDemo1")
    results = parser.parse_args()

    probe = Probe(name=results.pv)
    probe.configure_traits()