new ekgplotter tool

ekgplotter/ekgplotter/ekg_plotter_unused.py

@@ -0,0 +1,169 @@
+import numpy as np
+import string,cgi,time, random, socket
+from os import curdir, sep
+from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
+from SocketServer import ThreadingMixIn, ForkingMixIn
+import select
+import re
+
+from collections import deque
+
+from PyQt4.QtCore import QBuffer, QByteArray, QIODevice
+from PyQt4 import QtGui
+import pyqtgraph as pg
+
+from pyqtgraph.widgets.PlotWidget import PlotWidget
+
+
+try:
+    from chaosc.c_osc_lib import decode_osc
+except ImportError as e:
+    print(e)
+    from chaosc.osc_lib import decode_osc
+
+QAPP = None
+
+def mkQApp():
+    if QtGui.QApplication.instance() is None:
+        global QAPP
+        QAPP = QtGui.QApplication([])
+
+
+class PlotWindow(PlotWidget):
+    def __init__(self, title=None, **kargs):
+        mkQApp()
+        self.win = QtGui.QMainWindow()
+        PlotWidget.__init__(self, **kargs)
+        self.win.setCentralWidget(self)
+        for m in ['resize']:
+            setattr(self, m, getattr(self.win, m))
+        if title is not None:
+            self.win.setWindowTitle(title)
+
+
+class MyHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        print "get"
+        global plotValues
+        try:
+            self.path=re.sub('[^.a-zA-Z0-9]', "",str(self.path))
+            if self.path=="" or self.path==None or self.path[:1]==".":
+                return
+            if self.path.endswith(".html"):
+                f = open(curdir + sep + self.path)
+                self.send_response(200)
+                self.send_header('Content-type',    'text/html')
+                self.end_headers()
+                self.wfile.write(f.read())
+                f.close()
+                return
+            if self.path.endswith(".mjpeg"):
+                self.send_response(200)
+
+                plot_data1 = deque([0] * 100)
+                plot_data2 = deque([254/3] * 100)
+                plot_data3 = deque([254/3*2] * 100)
+                plt = PlotWindow(title="EKG", name="Merle")
+                plt.addLegend()
+                #plt = pg.plot(pen=(0, 3*1.3))
+                plt.resize(1280, 720)
+                plotItem1 = pg.PlotCurveItem(pen=(0, 3*1.3), name="bjoern")
+                plotItem2 = pg.PlotCurveItem(pen=(1, 3*1.3), name="merle")
+                plotItem3 = pg.PlotCurveItem(pen=(2, 3*1.3), name="uwe")
+                plotItem1.setPos(0, 0*6)
+                plotItem2.setPos(0, 1*6)
+                plotItem3.setPos(0, 2*6)
+                plt.addItem(plotItem1)
+                plt.addItem(plotItem2)
+                plt.addItem(plotItem3)
+
+                plt.setLabel('left', "EKG")
+                plt.setLabel('bottom', "Time")
+                plt.showGrid(True, True)
+                ba = plt.getAxis("bottom")
+                bl = plt.getAxis("left")
+                ba.setTicks([])
+                bl.setTicks([])
+                plt.setYRange(0, 254)
+                print type(plt)
+                self.wfile.write("Content-Type: multipart/x-mixed-replace; boundary=--aaboundary")
+                self.wfile.write("\r\n\r\n")
+                osc_sock = socket.socket(2, 2, 17)
+                osc_sock.bind(("", 10000))
+                osc_sock.setblocking(0)
+                last = time.time()
+                now = last
+
+                while 1:
+                    for i in xrange(3):
+                        reads, writes, errs = select.select([osc_sock], [], [], 0.01)
+                        #print reads, writes, errs
+                        if reads:
+                            osc_input = reads[0].recv(4096)
+                            osc_address, typetags, args = decode_osc(osc_input, 0, len(osc_input))
+                            #print "osc", osc_address, typetags, args
+                            if osc_address.startswith("/bjoern"):
+                                plot_data1.appendleft(args[0] / 3)
+                                plot_data1.pop()
+                            elif osc_address.startswith("/merle"):
+                                plot_data2.appendleft(args[0] / 3 + 254/3)
+                                plot_data2.pop()
+                            elif osc_address.startswith("/uwe"):
+                                plot_data3.appendleft(args[0] / 3 + 254/3*2)
+                                plot_data3.pop()
+
+                    plotItem1.setData(y=np.array(plot_data1), clear=True)
+                    plotItem2.setData(y=np.array(plot_data2), clear=True)
+                    plotItem3.setData(y=np.array(plot_data3), clear=True)
+                    #item = plt.plot(plot_data1, pen=(0, 3*1.3), clear=True)
+
+                    exporter = pg.exporters.ImageExporter.ImageExporter(plt.plotItem)
+                    exporter.parameters()['width'] = 1280
+                    #exporter.parameters()['height'] = 720
+                    name = 'tmpfile'
+                    img = exporter.export(name, True)
+                    buffer = QBuffer()
+                    buffer.open(QIODevice.ReadWrite)
+                    img.save(buffer, "JPG", 100)
+                    JpegData = buffer.data()
+                    self.wfile.write("--aaboundary\r\n")
+                    self.wfile.write("Content-Type: image/jpeg\r\n")
+                    self.wfile.write("Content-length: %d\r\n\r\n" % len(JpegData))
+                    self.wfile.write(JpegData)
+                    self.wfile.write("\r\n\r\n\r\n")
+                    now = time.time()
+                    dur = now - last
+                    print dur
+                    wait = 0.04 - dur
+                    if wait > 0:
+                        time.sleep(wait)
+                    last = now
+                return
+            if self.path.endswith(".jpeg"):
+                f = open(curdir + sep + self.path)
+                self.send_response(200)
+                self.send_header('Content-type','image/jpeg')
+                self.end_headers()
+                self.wfile.write(f.read())
+                f.close()
+                return
+            return
+        except IOError:
+            self.send_error(404,'File Not Found: %s' % self.path)
+
+
+#class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
+class ThreadedHTTPServer(HTTPServer, ForkingMixIn):
+    """Handle requests in a separate thread."""
+
+def main():
+    try:
+        server = ThreadedHTTPServer(('0.0.0.0', 9000), MyHandler)
+        print 'started httpserver...'
+        server.serve_forever()
+    except KeyboardInterrupt:
+        print '^C received, shutting down server'
+        server.socket.close()
+
+if __name__ == '__main__':
+    main()

ekgplotter/ekgplotter/index.html

@@ -0,0 +1,7 @@
+<HTML>
+<BODY>
+
+<img src="/camera.mjpeg" alt="Smiley face">
+
+</BODY>
+</HTML>

ekgplotter/ekgplotter/main.py

@@ -1,3 +1,31 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# This file is part of sensors2osc package
+#
+# sensors2osc is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# sensors2osc is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with sensors2osc.  If not, see <http://www.gnu.org/licenses/>.
+#
+# found the mjpeg part here, thanks for the nice code :)
+# http://hardsoftlucid.wordpress.com/2013/04/11/mjpeg-server-for-webcam-in-python-with-opencv/
+# the osc integration stuff is implemented by me
+#
+# Copyright (C) 2014 Stefan Kögl
+
+from __future__ import absolute_import
+
+import threading
+import Queue
 import numpy as np
 import string,cgi,time, random, socket
 from os import curdir, sep
@@ -14,6 +42,10 @@ import pyqtgraph as pg
 
 from pyqtgraph.widgets.PlotWidget import PlotWidget
 
+from chaosc.argparser_groups import *
+
+QtGui.QApplication.setGraphicsSystem('raster')
+
 
 try:
     from chaosc.c_osc_lib import decode_osc
@@ -21,17 +53,11 @@ except ImportError as e:
     print(e)
     from chaosc.osc_lib import decode_osc
 
-QAPP = None
+QAPP = QtGui.QApplication([])
-
-def mkQApp():
-    if QtGui.QApplication.instance() is None:
-        global QAPP
-        QAPP = QtGui.QApplication([])
 
 
 class PlotWindow(PlotWidget):
     def __init__(self, title=None, **kargs):
-        mkQApp()
         self.win = QtGui.QMainWindow()
         PlotWidget.__init__(self, **kargs)
         self.win.setCentralWidget(self)
@@ -41,14 +67,61 @@ class PlotWindow(PlotWidget):
             self.win.setWindowTitle(title)
 
 
+class OSCThread(threading.Thread):
+    def __init__(self, args):
+        super(OSCThread, self).__init__()
+        self.running = True
+        self.osc_sock = socket.socket(2, 2, 17)
+        self.osc_sock.bind((args.own_host, args.own_port))
+        self.osc_sock.setblocking(0)
+
+    def run(self):
+
+        while self.running:
+            reads, writes, errs = select.select([self.osc_sock], [], [], 0.05)
+            if reads:
+                osc_input = reads[0].recv(4096)
+                osc_address, typetags, messages = decode_osc(osc_input, 0, len(osc_input))
+                if osc_address.find("ekg") > -1 or osc_address.find("plot") != -1:
+                    queue.put_nowait((osc_address, messages))
+            else:
+                queue.put_nowait(("/bjoern/ekg", [0]))
+                queue.put_nowait(("/merle/ekg", [0]))
+                queue.put_nowait(("/uwe/ekg", [0]))
+        print "OSCThread is going down"
+
+
+queue = Queue.Queue()
+
 class MyHandler(BaseHTTPRequestHandler):
+
     def do_GET(self):
-        print "get"
+
-        global plotValues
+        self.thread = thread = OSCThread()
+        thread.daemon = True
+        thread.start()
+
+        actors = list()
+        is_item1 = True
+        is_item2 = True
+        is_item3 = True
+
+        def setPositions():
+            for ix, item in enumerate(actors):
+                item.setPos(0, ix*6)
+
+
+        def scale_data(data, ix, max_items):
+            scale = 254 / max_items * ix
+            return [value / max_items + scale for value in data]
+
+
         try:
+
             self.path=re.sub('[^.a-zA-Z0-9]', "",str(self.path))
             if self.path=="" or self.path==None or self.path[:1]==".":
                 return
+
             if self.path.endswith(".html"):
                 f = open(curdir + sep + self.path)
                 self.send_response(200)
@@ -56,20 +129,26 @@ class MyHandler(BaseHTTPRequestHandler):
                 self.end_headers()
                 self.wfile.write(f.read())
                 f.close()
-                return
+            elif self.path.endswith(".mjpeg"):
-            if self.path.endswith(".mjpeg"):
+                data_points = 1000
-                self.send_response(200)
 
-                plot_data1 = deque([0] * 100)
+                self.send_response(200)
-                plot_data2 = deque([254/3] * 100)
+                plot_data1 = data = deque([0] * data_points)
-                plot_data3 = deque([254/3*2] * 100)
+                plot_data2 = data = deque([0] * data_points)
-                plt = PlotWindow(title="EKG", name="Merle")
+                plot_data3 = data = deque([0] * data_points)
-                plt.addLegend()
+                plt = PlotWidget(title="<h1>EKG</h1>", name="Merle")
-                #plt = pg.plot(pen=(0, 3*1.3))
+                plt.hide()
-                plt.resize(1280, 720)
+                plotItem1 = pg.PlotCurveItem(pen=pg.mkPen('r', width=2), name="bjoern")
-                plotItem1 = pg.PlotCurveItem(pen=(0, 3*1.3), name="bjoern")
+                plotItem2 = pg.PlotCurveItem(pen=pg.mkPen('g', width=2), name="merle")
-                plotItem2 = pg.PlotCurveItem(pen=(1, 3*1.3), name="merle")
+                plotItem3 = pg.PlotCurveItem(pen=pg.mkPen('b', width=2), name="uwe")
-                plotItem3 = pg.PlotCurveItem(pen=(2, 3*1.3), name="uwe")
+                print type(plotItem1)
+                pen = pg.mkPen(254, 254, 254)
+                plotItem1.setShadowPen(pen=pen, width=6, cosmetic=True)
+                plotItem2.setShadowPen(pen=pen, width=6, cosmetic=True)
+                plotItem3.setShadowPen(pen=pen, width=6, cosmetic=True)
+                actors.append(plotItem1)
+                actors.append(plotItem2)
+                actors.append(plotItem3)
                 plotItem1.setPos(0, 0*6)
                 plotItem2.setPos(0, 1*6)
                 plotItem3.setPos(0, 2*6)
@@ -77,69 +156,100 @@ class MyHandler(BaseHTTPRequestHandler):
                 plt.addItem(plotItem2)
                 plt.addItem(plotItem3)
 
-                plt.setLabel('left', "EKG")
+                plt.setLabel('left', "<h2>Amplitude</h2>")
-                plt.setLabel('bottom', "Time")
+                plt.setLabel('bottom', "<h2>Time</h2>")
                 plt.showGrid(True, True)
                 ba = plt.getAxis("bottom")
                 bl = plt.getAxis("left")
                 ba.setTicks([])
                 bl.setTicks([])
                 plt.setYRange(0, 254)
-                print type(plt)
+
                 self.wfile.write("Content-Type: multipart/x-mixed-replace; boundary=--aaboundary")
                 self.wfile.write("\r\n\r\n")
-                osc_sock = socket.socket(2, 2, 17)
+
-                osc_sock.bind(("", 10000))
+
-                osc_sock.setblocking(0)
+                plt.resize(1280, 720)
-                last = time.time()
-                now = last
 
                 while 1:
-                    for i in xrange(3):
+                    while 1:
-                        reads, writes, errs = select.select([osc_sock], [], [], 0.01)
+                        try:
-                        #print reads, writes, errs
+                            osc_address, args = queue.get_nowait()
-                        if reads:
+                        except Queue.Empty:
-                            osc_input = reads[0].recv(4096)
+                            break
-                            osc_address, typetags, args = decode_osc(osc_input, 0, len(osc_input))
-                            #print "osc", osc_address, typetags, args
-                            if osc_address.startswith("/bjoern"):
-                                plot_data1.appendleft(args[0] / 3)
-                                plot_data1.pop()
-                            elif osc_address.startswith("/merle"):
-                                plot_data2.appendleft(args[0] / 3 + 254/3)
-                                plot_data2.pop()
-                            elif osc_address.startswith("/uwe"):
-                                plot_data3.appendleft(args[0] / 3 + 254/3*2)
-                                plot_data3.pop()
 
-                    plotItem1.setData(y=np.array(plot_data1), clear=True)
+                        value = args[0]
-                    plotItem2.setData(y=np.array(plot_data2), clear=True)
+                        if osc_address == "/bjoern/ekg":
-                    plotItem3.setData(y=np.array(plot_data3), clear=True)
+                            plot_data1.append(value)
-                    #item = plt.plot(plot_data1, pen=(0, 3*1.3), clear=True)
+                            plot_data1.popleft()
+                            try:
+                                plotItem1.setData(y=np.array(scale_data(plot_data1, actors.index(plotItem1), len(actors))), clear=True)
+                            except ValueError:
+                                pass
+                        elif osc_address == "/merle/ekg":
+                            plot_data2.append(value)
+                            plot_data2.popleft()
+                            try:
+                                plotItem2.setData(y=np.array(scale_data(plot_data2, actors.index(plotItem2), len(actors))), clear=True)
+                            except ValueError:
+                                pass
+                        elif osc_address == "/uwe/ekg":
+                            plot_data3.append(value)
+                            plot_data3.popleft()
+                            try:
+                                plotItem3.setData(y=np.array(scale_data(plot_data3, actors.index(plotItem3), len(actors))), clear=True)
+                            except ValueError:
+                                pass
+                        elif osc_address == "/plot/uwe":
+                            if value == 1 and is_item3 == False:
+                                print "uwe on"
+                                plt.addItem(plotItem3)
+                                is_item3 = True
+                                actors.append(plotItem3)
+                                setPositions()
+                            elif value == 0 and is_item3 == True:
+                                print "uwe off"
+                                plt.removeItem(plotItem3)
+                                is_item3 = False
+                                actors.remove(plotItem3)
+                                setPositions()
+                        elif osc_address == "/plot/merle":
+                            if value == 1 and is_item2 == False:
+                                print "merle on"
+                                plt.addItem(plotItem2)
+                                is_item2 = True
+                                actors.append(plotItem2)
+                                setPositions()
+                            elif value == 0 and is_item2 == True:
+                                print "merle off"
+                                plt.removeItem(plotItem2)
+                                is_item2 = False
+                                actors.remove(plotItem2)
+                                setPositions()
+                        elif osc_address == "/plot/bjoern":
+                            if value == 1 and is_item1 == False:
+                                print "bjoern on"
+                                plt.addItem(plotItem1)
+                                is_item1 = True
+                                actors.append(plotItem1)
+                                setPositions()
+                            elif value == 0 and is_item1 == True:
+                                print "bjoern off"
+                                plt.removeItem(plotItem1)
+                                is_item1 = False
+                                actors.remove(plotItem1)
+                                setPositions()
 
                     exporter = pg.exporters.ImageExporter.ImageExporter(plt.plotItem)
-                    exporter.parameters()['width'] = 1280
+                    img = exporter.export("tmpfile", True)
-                    #exporter.parameters()['height'] = 720
-                    name = 'tmpfile'
-                    img = exporter.export(name, True)
                     buffer = QBuffer()
-                    buffer.open(QIODevice.ReadWrite)
+                    buffer.open(QIODevice.WriteOnly)
                     img.save(buffer, "JPG", 100)
                     JpegData = buffer.data()
-                    self.wfile.write("--aaboundary\r\n")
+                    del buffer
-                    self.wfile.write("Content-Type: image/jpeg\r\n")
+                    self.wfile.write("--aaboundary\r\nContent-Type: image/jpeg\r\nContent-length: %d\r\n\r\n%s\r\n\r\n\r\n" % (len(JpegData), JpegData))
-                    self.wfile.write("Content-length: %d\r\n\r\n" % len(JpegData))
+
-                    self.wfile.write(JpegData)
+            elif self.path.endswith(".jpeg"):
-                    self.wfile.write("\r\n\r\n\r\n")
-                    now = time.time()
-                    dur = now - last
-                    print dur
-                    wait = 0.04 - dur
-                    if wait > 0:
-                        time.sleep(wait)
-                    last = now
-                return
-            if self.path.endswith(".jpeg"):
                 f = open(curdir + sep + self.path)
                 self.send_response(200)
                 self.send_header('Content-type','image/jpeg')
@@ -147,23 +257,37 @@ class MyHandler(BaseHTTPRequestHandler):
                 self.wfile.write(f.read())
                 f.close()
             return
-            return
+        except (KeyboardInterrupt, SystemError):
+            thread.running = False
+            thread.join()
         except IOError:
             self.send_error(404,'File Not Found: %s' % self.path)
 
+    def __del__(self):
+        self.thread.running = False
+        self.thread.join()
+
+
+class JustAHTTPServer(HTTPServer):
+    pass
 
-#class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
-class ThreadedHTTPServer(HTTPServer, ForkingMixIn):
-    """Handle requests in a separate thread."""
 
 def main():
+    a = create_arg_parser("ekgplotter")
+    add_main_group(a)
+    add_chaosc_group(a)
+    add_subscriber_group(a, "ekgplotter")
+    args = finalize_arg_parser(a)
+
     try:
-        server = ThreadedHTTPServer(('0.0.0.0', 9000), MyHandler)
+        server = JustAHTTPServer(('0.0.0.0', 9000), MyHandler)
         print 'started httpserver...'
         server.serve_forever()
     except KeyboardInterrupt:
         print '^C received, shutting down server'
         server.socket.close()
 
+
 if __name__ == '__main__':
     main()
+

ekgplotter/setup.py

@@ -22,7 +22,7 @@ setup(
 
     exclude_package_data = {'': ['.gitignore']},
 
-    install_requires=[],
+    install_requires=["pyqtgraph"],
 
     # installing unzipped
     zip_safe = False,