Changeset 171 for trunk/synapse

Timestamp:

08/26/10 13:57:22 (11 years ago)

Author:

sc

Message:

synapse/Puzzlebox/Synapse/Interface.py:

• matplotlib calls moved to separate module

synapse/Puzzlebox/Synapse/Interface_Plot.py:

• matplotlib calls moved to separate module

Location:

trunk/synapse/Puzzlebox/Synapse

Files:

• Interface.py (modified) (6 diffs)
• Interface_Plot.py (added)

trunk/synapse/Puzzlebox/Synapse/Interface.py

@@ -10 +10 @@
 
 __changelog__ = """\
-Last Update: 2010.08.25
+Last Update: 2010.08.26
 
 """
@@ -23 +23 @@
 import simplejson as json
 
-import random
-from numpy import arange, sin, pi
-from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
-from matplotlib.figure import Figure
+try:
+        from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
+        from matplotlib.figure import Figure
+        from numpy import arange, sin, pi
+        from Interface_Plot import rawEEGMatplotlibCanvas
+        MATPLOTLIB_AVAILABLE = True
+except:
+        MATPLOTLIB_AVAILABLE = False
 
 if (sys.platform != 'win32'):
@@ -164 +168 @@
                 
                 
-                self.matplot = rawEEGMatplotlibCanvas(self.tabEEGSignals, \
-                                                       width=8, \
-                                                       height=4, \
-                                                       dpi=100, \
-                                                       title='Raw EEG Waves')
+                if MATPLOTLIB_AVAILABLE:
+                        self.matplot = rawEEGMatplotlibCanvas(self.tabEEGSignals, \
+                                                              width=8, \
+                                                              height=4, \
+                                                              dpi=100, \
+                                                              title='Raw EEG Waves')
         
         
@@ -363 +368 @@
         ##################################################################
         
-        def updateRawEegInterface(self):
-                
-                pass
-        
-        
-        ##################################################################
-        
         def processPacketThinkGear(self, packet):
                 
@@ -378 +376 @@
                 if ('rawEeg' in packet.keys()):
                         value = packet['rawEeg']
-                        self.matplot.update_figure(value)
+                        if MATPLOTLIB_AVAILABLE:
+                                self.matplot.update_figure(value)
                 
                 
@@ -598 +597 @@
                         event.ignore()
 
-
-#####################################################################
-#####################################################################
-
-class matplotlibCanvas(FigureCanvas):
-        
-        """Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.)."""
-        
-        def __init__(self, parent=None, width=8, height=4, dpi=100, title=None):
-                
-                fig = Figure(figsize=(width, height), dpi=dpi)
-                self.axes_top = fig.add_subplot(211)
-                self.axes_bottom = fig.add_subplot(212)
-                # We want the axes cleared every time plot() is called
-                self.axes_top.hold(False)
-                self.axes_bottom.hold(False)
-                
-                if title != None:
-                        fig.suptitle(title, fontsize=12)
-                
-                FigureCanvas.__init__(self, fig)
-                self.setParent(parent)
-                
-                FigureCanvas.setSizePolicy(self,
-                                                                                        QtGui.QSizePolicy.Expanding,
-                                                                                        QtGui.QSizePolicy.Expanding)
-                FigureCanvas.updateGeometry(self)
-
-
-#####################################################################
-#####################################################################
-
-class rawEEGMatplotlibCanvas(matplotlibCanvas):
-        
-        def __init__(self,  *args, **kwargs):
-                
-                matplotlibCanvas.__init__(self, *args, **kwargs)
-                
-                self.DEBUG=DEBUG
-                
-                #timer = QtCore.QTimer(self)
-                #QtCore.QObject.connect(timer, QtCore.SIGNAL("timeout()"), self.update_figure)
-                #timer.start(256)
-                
-                self.update_top_frequency = \
-                   INTERFACE_RAW_EEG_UPDATE_FREQUENCY
-                self.update_bottom_frequency = \
-                   INTERFACE_RAW_EEG_UPDATE_FREQUENCY / 2
-                
-                self.values_top = []
-                self.values_bottom = []
-                
-                self.axes_top.set_xbound(0, self.update_top_frequency)
-                self.axes_top.set_ybound(-2048, 2047)
-                
-                self.axes_bottom.set_xbound(0, self.update_bottom_frequency)
-                self.axes_bottom.set_ybound(-512, 512)
-                
-                self.axes_top.grid(True)
-                self.axes_bottom.grid(True)
-                
-                self.axes_top.text(self.update_top_frequency + 24, \
-                                   0, \
-                                   '%i Hz' % self.update_top_frequency, \
-                                   rotation='vertical', \
-                                   verticalalignment='center')
-                
-                self.axes_bottom.text(self.update_bottom_frequency + 12, \
-                                   0, \
-                                   '%i Hz' % self.update_bottom_frequency, \
-                                   rotation='vertical', \
-                                   verticalalignment='center')
-                
-                self.axes_top.set_autoscale_on(False)
-                self.axes_bottom.set_autoscale_on(False)
-        
-        
-        ##################################################################
-        
-        def update_figure(self, value):
-                
-                self.values_top.append(value)
-                self.values_bottom.append(value)
-                
-                if len(self.values_top) == self.update_top_frequency:
-                        
-                        self.axes_top.plot(range(self.update_top_frequency), \
-                                       self.values_top, \
-                                       'b-', \
-                                       scalex=False, \
-                                       scaley=False)
-                        
-                        #self.axes_top.set_ylabel('%i Hz' % self.update_top_frequency)
-                        self.axes_top.grid(True)
-                        
-                        self.axes_top.text(self.update_top_frequency + 24, \
-                                           0, \
-                                           '%i Hz' % self.update_top_frequency, \
-                                           rotation='vertical', \
-                                           verticalalignment='center')
-                        
-                        self.draw()
-                        
-                        self.values_top = []
-                
-                
-                if len(self.values_bottom) == self.update_bottom_frequency:
-                        
-                        self.axes_bottom.plot(range(self.update_bottom_frequency), \
-                                       self.values_bottom, \
-                                       'r-', \
-                                       scalex=False, \
-                                       scaley=False)
-                        
-                        #self.axes_bottom.set_ylabel('%i Hz' % self.update_bottom_frequency)
-                        self.axes_bottom.grid(True)
-                        
-                        self.axes_bottom.text(self.update_bottom_frequency + 12, \
-                                           0, \
-                                           '%i Hz' % self.update_bottom_frequency, \
-                                           rotation='vertical', \
-                                           verticalalignment='center')
-                        
-                        self.draw()
-                        
-                        self.values_bottom = []
-        
-