psychose/ekgplotter/ekgplotter/main.py

462 lines
17 KiB
Python
Raw Normal View History

2014-03-15 19:57:14 +00:00
#!/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
2014-03-24 05:47:02 +00:00
2014-03-15 19:57:14 +00:00
#import objgraph
2014-03-24 05:47:02 +00:00
#import gc
#gc.set_debug(gc.DEBUG_LEAK)
2014-03-16 10:35:23 +00:00
from datetime import datetime
from queue import Empty
2014-03-13 03:22:06 +00:00
import numpy as np
import string,cgi,time, random, socket
from os import curdir, sep
2014-03-24 05:47:02 +00:00
import os.path
from http.server import BaseHTTPRequestHandler, HTTPServer
from socketserver import ThreadingMixIn, ForkingMixIn
2014-03-11 17:35:45 +00:00
import select
2014-03-13 03:22:06 +00:00
import re
2014-03-11 17:35:45 +00:00
from collections import deque
2014-03-13 03:22:06 +00:00
from PyQt4.QtCore import QBuffer, QByteArray, QIODevice
2014-03-24 05:47:02 +00:00
from PyQt4 import QtGui, QtCore
2014-03-13 03:22:06 +00:00
import pyqtgraph as pg
2014-03-11 17:35:45 +00:00
2014-03-13 03:22:06 +00:00
from pyqtgraph.widgets.PlotWidget import PlotWidget
2014-03-15 19:57:14 +00:00
from chaosc.argparser_groups import *
from chaosc.lib import resolve_host
2014-03-15 19:57:14 +00:00
from multiprocessing import Process, Queue
2014-03-25 21:26:19 +00:00
try:
from chaosc.c_osc_lib import *
except ImportError:
from chaosc.osc_lib import *
2014-03-16 10:35:23 +00:00
2014-03-13 03:22:06 +00:00
try:
from chaosc.c_osc_lib import decode_osc
except ImportError as e:
print(e)
from chaosc.osc_lib import decode_osc
class OSCThread(Process):
def __init__(self, args, queue):
2014-03-15 19:57:14 +00:00
super(OSCThread, self).__init__()
self.msg_queue = queue
2014-03-16 10:35:23 +00:00
self.args = args
2014-03-15 19:57:14 +00:00
self.running = True
2014-03-16 10:35:23 +00:00
2014-03-25 21:26:19 +00:00
self.own_address = resolve_host(args.own_host, args.own_port, args.address_family)
2014-03-16 10:35:23 +00:00
2014-03-25 21:26:19 +00:00
self.chaosc_address = chaosc_host, chaosc_port = resolve_host(args.chaosc_host, args.chaosc_port, args.address_family)
self.osc_sock = socket.socket(args.address_family, 2, 17)
2014-03-16 10:35:23 +00:00
self.osc_sock.bind(self.own_address)
2014-03-15 19:57:14 +00:00
self.osc_sock.setblocking(0)
2014-03-24 05:47:02 +00:00
print("%s: starting up osc receiver on '%s:%d'" % (
datetime.now().strftime("%x %X"), self.own_address[0], self.own_address[1]))
2014-03-16 10:35:23 +00:00
self.subscribe_me()
def subscribe_me(self):
"""Use this procedure for a quick'n dirty subscription to your chaosc instance.
:param chaosc_address: (chaosc_host, chaosc_port)
:type chaosc_address: tuple
:param receiver_address: (host, port)
:type receiver_address: tuple
:param token: token to get authorized for subscription
:type token: str
"""
2014-03-24 05:47:02 +00:00
print("%s: subscribing to '%s:%d' with label %r" % (datetime.now().strftime("%x %X"), self.chaosc_address[0], self.chaosc_address[1], self.args.subscriber_label))
msg = OSCMessage(b"/subscribe")
msg.appendTypedArg(bytes(self.own_address[0], "ascii"), b"s")
msg.appendTypedArg(self.own_address[1], b"i")
msg.appendTypedArg(bytes(self.args.authenticate, "ascii"), b"s")
2014-03-16 10:35:23 +00:00
if self.args.subscriber_label is not None:
2014-03-24 05:47:02 +00:00
msg.appendTypedArg(bytes(self.args.subscriber_label, "ascii"), b"s")
2014-03-16 10:35:23 +00:00
self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address)
def unsubscribe_me(self):
if self.args.keep_subscribed:
return
2014-03-24 05:47:02 +00:00
print("%s: unsubscribing from '%s:%d'" % (datetime.now().strftime("%x %X"), self.chaosc_address[0], self.chaosc_address[1]))
msg = OSCMessage(b"/unsubscribe")
msg.appendTypedArg(bytes(self.own_address[0], "ascii"), b"s")
msg.appendTypedArg(self.own_address[1], b"i")
msg.appendTypedArg(bytes(self.args.authenticate, "ascii"), b"s")
2014-03-16 10:35:23 +00:00
self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address)
2014-03-15 19:57:14 +00:00
def run(self):
while self.running:
reads, writes, errs = select.select([self.osc_sock], [], [], 0.05)
if reads:
2014-03-25 21:26:19 +00:00
osc_input = reads[0].recv(128)
2014-03-15 19:57:14 +00:00
osc_address, typetags, messages = decode_osc(osc_input, 0, len(osc_input))
print("thread: osc_address", osc_address)
2014-03-24 05:47:02 +00:00
if osc_address.find(b"ekg") > -1 or osc_address.find(b"plot") != -1:
2014-03-25 21:26:19 +00:00
#print("send", osc_address)
self.msg_queue.put_nowait((osc_address, messages))
2014-03-15 19:57:14 +00:00
else:
print("thread: dummy values")
self.msg_queue.put_nowait((b"/bjoern/ekg", [0]))
self.msg_queue.put_nowait((b"/merle/ekg", [0]))
self.msg_queue.put_nowait((b"/uwe/ekg", [0]))
time.sleep(0.01)
2014-03-16 10:35:23 +00:00
self.unsubscribe_me()
2014-03-24 05:47:02 +00:00
print("OSCThread is going down")
2014-03-15 19:57:14 +00:00
class Actor(object):
shadowPen = pg.mkPen(255, 255, 255)
brush = pg.mkBrush("w")
def __init__(self, name, num_data, color):
self.data = [0] * num_data
self.data_pointer = 0
self.name = name
self.active = True
self.plotItem = pg.PlotCurveItem(pen=pg.mkPen(color, width=3), name=name)
self.num_data = num_data
#self.plotItem.setShadowPen(pen=Actor.shadowPen, width=3, cosmetic=True)
self.plotPoint = pg.ScatterPlotItem(pen=Actor.shadowPen, brush=self.brush, size=5)
def scale_data(self, ix, max_items):
scale = 255 / max_items * ix
return [value / max_items + scale for value in self.data]
def set_point(self, value, ix, max_items):
scale = 255 / max_items * ix
self.plotPoint.setData(x = [self.data_pointer], y = [value / max_items + scale])
#def find_max_value(self, item_data):
#max_index = -1
#for ix, i in enumerate(item_data):
#if i > 250:
#return ix, i
#return None, None
#def rearrange(self, item_data, actual_pos, max_items):
#max_value_index, max_value = find_max_value(item_data)
#if max_value_index is None:
#return actual_pos
#mean = int(max_items / 2.)
#start = mean - max_value_index
#if start != 0:
#item_data.rotate(start)
#pos = (actual_pos + start) % max_items
#else:
#pos = actual_pos
#print "rearrange", mean, start, actual_pos, pos, item_data
#return pos
def set_value(self, value):
self.data[self.data_pointer] = value
self.data_pointer = (self.data_pointer + 1) % self.num_data
#def resize(item_data, max_length, new_max_length, pos):
#print "resize", max_length, new_max_length
#if new_max_length < 15:
#return max_length, pos
#if new_max_length > max_length:
#pad = (new_max_length - max_length)
#print "pad", pad
#for i in range(pad):
#if i % 2 == 0:
#item_data.append(0)
#else:
#item_data.appendleft(0)
#pos += 1
#return new_max_length, pos
#elif new_max_length < max_length:
#pad = (max_length - new_max_length)
#for i in range(pad):
#if i % 2 == 0:
#item_data.pop()
#if pos >= new_max_length:
#pos = 0
#else:
#item_data.popleft()
#if pos > 0:
#pos -= 1
#return new_max_length, pos
#return max_length, pos
class EkgPlot(object):
def __init__(self, actor_names, num_data, colors):
self.plot = pg.PlotWidget(title="<h1>EKG</h1>")
2014-03-24 05:47:02 +00:00
self.plot.setAttribute(QtCore.Qt.WA_DeleteOnClose)
self.plot.hide()
self.plot.setLabel('left', "<h2>Amplitude</h2>")
self.plot.setLabel('bottom', "<h2><sup>Time</sup></h2>")
self.plot.showGrid(True, True)
self.plot.setYRange(0, 255)
self.plot.setXRange(0, num_data)
self.plot.resize(1280, 720)
ba = self.plot.getAxis("bottom")
bl = self.plot.getAxis("left")
ba.setTicks([])
bl.setTicks([])
self.active_actors = list()
self.actors = dict()
self.lengths1 = [0]
self.num_data = num_data
for actor_name, color in zip(actor_names, colors):
self.add_actor(actor_name, num_data, color)
self.set_positions()
2014-03-24 05:47:02 +00:00
self.ekg_regex = re.compile(b"^/(.*?)/ekg$")
self.ctl_regex = re.compile(b"^/plot/(.*?)$")
self.updated_actors = set()
def add_actor(self, actor_name, num_data, color):
actor_obj = Actor(actor_name, num_data, color)
self.actors[actor_name] = actor_obj
self.plot.addItem(actor_obj.plotItem)
self.plot.addItem(actor_obj.plotPoint)
self.active_actors.append(actor_obj)
def set_positions(self):
for ix, actor_obj in enumerate(self.active_actors):
actor_obj.plotItem.setPos(0, ix * 6)
actor_obj.plotPoint.setPos(0, ix * 6)
def active_actor_count(self):
return len(self.active_actors)
def update(self, osc_address, value):
res = self.ekg_regex.match(osc_address)
if res:
2014-03-25 21:26:19 +00:00
#print("matched data")
actor_name = res.group(1)
actor_obj = self.actors[actor_name]
max_actors = len(self.active_actors)
actor_data = actor_obj.data
data_pointer = actor_obj.data_pointer
actor_data[data_pointer] = value
2014-03-25 21:26:19 +00:00
try:
ix = self.active_actors.index(actor_obj)
actor_obj.set_point(value, ix, max_actors)
actor_obj.plotItem.setData(y=np.array(actor_obj.scale_data(ix, max_actors)), clear=True)
except ValueError as e:
#print("data", e)
pass
actor_obj.data_pointer = (data_pointer + 1) % self.num_data
return
res = self.ctl_regex.match(osc_address)
if res:
actor_name = res.group(1)
actor_obj = self.actors[actor_name]
if value == 1 and not actor_obj.active:
2014-03-25 21:26:19 +00:00
#print("actor on", actor_name, actor_obj, self.active_actors)
actor_obj.active = True
2014-03-25 21:26:19 +00:00
if actor_obj not in self.active_actors:
self.plot.addItem(actor_obj.plotItem)
self.plot.addItem(actor_obj.plotPoint)
self.active_actors.append(actor_obj)
elif value == 0 and actor_obj.active:
#print "actor off", actor_name
2014-03-25 21:26:19 +00:00
actor_obj.active = False
self.plot.removeItem(actor_obj.plotItem)
self.plot.removeItem(actor_obj.plotPoint)
try:
self.active_actors.remove(actor_obj)
except ValueError as e:
pass
assert actor_obj not in self.active_actors
self.set_positions()
2014-03-13 03:22:06 +00:00
class MyHandler(BaseHTTPRequestHandler):
2014-03-15 19:57:14 +00:00
2014-03-20 23:19:09 +00:00
2014-03-13 03:22:06 +00:00
def do_GET(self):
2014-03-13 03:22:06 +00:00
try:
self.path=re.sub('[^.a-zA-Z0-9]', "",str(self.path))
if self.path=="" or self.path==None or self.path[:1]==".":
2014-03-24 05:47:02 +00:00
self.send_error(403)
2014-03-15 19:57:14 +00:00
2014-03-13 03:22:06 +00:00
if self.path.endswith(".html"):
2014-03-24 05:47:02 +00:00
directory = os.path.dirname(os.path.abspath(__file__))
data = open(os.path.join(directory, self.path), "rb").read()
2014-03-13 03:22:06 +00:00
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
2014-03-24 05:47:02 +00:00
self.wfile.write(data)
2014-03-15 19:57:14 +00:00
elif self.path.endswith(".mjpeg"):
msg_queue = Queue()
self.thread = thread = OSCThread(self.server.args, msg_queue)
thread.start()
2014-03-13 03:22:06 +00:00
2014-03-15 19:57:14 +00:00
self.send_response(200)
2014-03-24 05:47:02 +00:00
actor_names = [b"bjoern", b"merle", b"uwe"]
num_data = 100
colors = ["r", "g", "b"]
qtapp = QtGui.QApplication([])
self.plotter = plotter = EkgPlot(actor_names, num_data, colors)
2014-03-13 03:22:06 +00:00
2014-03-24 05:47:02 +00:00
self.send_header("Content-Type", "multipart/x-mixed-replace; boundary=--aaboundary\r\n\r\n")
self.end_headers()
#lastTime = time.time()
#fps = None
event_loop = QtCore.QEventLoop()
2014-03-13 03:22:06 +00:00
while 1:
event_loop.processEvents()
qtapp.sendPostedEvents(None, 0)
2014-03-15 19:57:14 +00:00
while 1:
try:
2014-03-24 05:47:02 +00:00
osc_address, args = msg_queue.get_nowait()
except Empty as e:
print("queue error", e, type(e))
2014-03-15 19:57:14 +00:00
break
print("main: got value", osc_address, args)
plotter.update(osc_address, args[0])
exporter = pg.exporters.ImageExporter.ImageExporter(plotter.plot.plotItem)
2014-03-15 19:57:14 +00:00
img = exporter.export("tmpfile", True)
2014-03-13 03:22:06 +00:00
buffer = QBuffer()
2014-03-15 19:57:14 +00:00
buffer.open(QIODevice.WriteOnly)
2014-03-13 03:22:06 +00:00
img.save(buffer, "JPG", 100)
2014-03-24 05:47:02 +00:00
JpegData = buffer.data().data()
try:
self.send_header("--aaboundary", None)
self.send_header("Content-Type", "image/jpeg")
self.send_header("Content-length", len(JpegData))
self.end_headers()
self.wfile.write(JpegData + b"\r\n\r\n\r\n")
except (BrokenPipeError, ConnectionResetError) as e:
print("Error:", e)
if hasattr(self, "plotter"):
print("children", plotter.plot.children())
plotter.plot.deleteLater()
plotter.plot.close()
del self.plotter.plot
del self.plotter
del plotter
thread.running = False
thread.join()
return
del JpegData
del buffer
del img
del exporter
#now = time.time()
#dt = now - lastTime
#lastTime = now
#if fps is None:
#fps = 1.0/dt
#else:
#s = np.clip(dt*3., 0, 1)
#fps = fps * (1-s) + (1.0/dt) * s
#print '%0.2f fps' % fps
2014-03-15 19:57:14 +00:00
elif self.path.endswith(".jpeg"):
2014-03-24 05:47:02 +00:00
directory = os.path.dirname(os.path.abspath(__file__))
data = open(os.path.join(directory, self.path), "rb").read()
2014-03-13 03:22:06 +00:00
self.send_response(200)
self.send_header('Content-type','image/jpeg')
self.end_headers()
2014-03-24 05:47:02 +00:00
self.wfile.write(data)
2014-03-13 03:22:06 +00:00
return
2014-03-15 19:57:14 +00:00
except (KeyboardInterrupt, SystemError):
2014-03-25 21:26:19 +00:00
print("queue size", msg_queue.qsize())
2014-03-15 19:57:14 +00:00
thread.running = False
thread.join()
2014-03-24 05:47:02 +00:00
if hasattr(self, "plotter"):
plotter.plot.deleteLater()
2014-03-25 21:26:19 +00:00
except IOError as e:
print("ioerror", e)
print('-'*40)
print('Exception happened during processing of request from')
import traceback
traceback.print_exc() # XXX But this goes to stderr!
print( '-'*40)
self.send_error(404, 'File Not Found: %s' % self.path)
2014-03-13 03:22:06 +00:00
2014-03-15 19:57:14 +00:00
class JustAHTTPServer(HTTPServer):
pass
2014-03-13 03:22:06 +00:00
2014-03-11 17:35:45 +00:00
def main():
2014-03-25 21:26:19 +00:00
arg_parser = create_arg_parser("ekgplotter")
own_group = add_main_group(arg_parser)
own_group.add_argument('-x', "--http_host", default="::",
2014-03-16 10:35:23 +00:00
help='my host, defaults to "socket.gethostname()"')
own_group.add_argument('-X', "--http_port", default=9000,
type=int, help='my port, defaults to 9000')
2014-03-25 21:26:19 +00:00
add_chaosc_group(arg_parser)
add_subscriber_group(arg_parser, "ekgplotter")
args = finalize_arg_parser(arg_parser)
2014-03-15 19:57:14 +00:00
2014-03-25 21:26:19 +00:00
http_host, http_port = resolve_host(args.http_host, args.http_port, args.address_family)
server = JustAHTTPServer((http_host, http_port), MyHandler)
2014-03-25 21:26:19 +00:00
server.address_family = args.address_family
server.args = args
2014-03-24 05:47:02 +00:00
print("%s: starting up http server on '%s:%d'" % (
datetime.now().strftime("%x %X"), http_host, http_port))
try:
2014-03-13 03:22:06 +00:00
server.serve_forever()
except KeyboardInterrupt:
server.socket.close()
2014-03-15 19:57:14 +00:00
2014-03-13 03:22:06 +00:00
if __name__ == '__main__':
main()
2014-03-15 19:57:14 +00:00