ctdo
/
psychose
16
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ported to python3.3

This commit is contained in:
Stefan Kögl 2014-03-24 06:47:02 +01:00
parent 7c0f56bf20
commit 9bcdc6b377
2 changed files with 83 additions and 69 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

151
ekgplotter/ekgplotter/main.py
View File

@ -22,19 +22,23 @@
# #
# Copyright (C) 2014 Stefan Kögl # Copyright (C) 2014 Stefan Kögl
from __future__ import absolute_import
#import objgraph #import objgraph
#import gc
#gc.set_debug(gc.DEBUG_LEAK)
from datetime import datetime from datetime import datetime
import threading import threading
import Queue import queue
import numpy as np import numpy as np
import string,cgi,time, random, socket import string,cgi,time, random, socket
from os import curdir, sep from os import curdir, sep
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer import os.path
from SocketServer import ThreadingMixIn, ForkingMixIn from http.server import BaseHTTPRequestHandler, HTTPServer
from socketserver import ThreadingMixIn, ForkingMixIn
import select import select
import re import re
@ -42,7 +46,7 @@ from collections import deque
from PyQt4.QtCore import QBuffer, QByteArray, QIODevice from PyQt4.QtCore import QBuffer, QByteArray, QIODevice
from PyQt4 import QtGui from PyQt4 import QtGui, QtCore
import pyqtgraph as pg import pyqtgraph as pg
@ -56,9 +60,6 @@ from chaosc.lib import resolve_host
#except ImportError: #except ImportError:
from chaosc.osc_lib import * from chaosc.osc_lib import *
QtGui.QApplication.setGraphicsSystem('opengl')
print "systemInfo", pg.systemInfo()
try: try:
from chaosc.c_osc_lib import decode_osc from chaosc.c_osc_lib import decode_osc
@ -66,18 +67,6 @@ except ImportError as e:
print(e) print(e)
from chaosc.osc_lib import decode_osc from chaosc.osc_lib import decode_osc
QAPP = QtGui.QApplication([])
class PlotWindow(PlotWidget):
def __init__(self, title=None, **kargs):
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 OSCThread(threading.Thread): class OSCThread(threading.Thread):
@ -93,8 +82,8 @@ class OSCThread(threading.Thread):
self.osc_sock.bind(self.own_address) self.osc_sock.bind(self.own_address)
self.osc_sock.setblocking(0) self.osc_sock.setblocking(0)
print "%s: starting up osc receiver on '%s:%d'" % ( print("%s: starting up osc receiver on '%s:%d'" % (
datetime.now().strftime("%x %X"), self.own_address[0], self.own_address[1]) datetime.now().strftime("%x %X"), self.own_address[0], self.own_address[1]))
self.subscribe_me() self.subscribe_me()
@ -110,13 +99,13 @@ class OSCThread(threading.Thread):
:param token: token to get authorized for subscription :param token: token to get authorized for subscription
:type token: str :type token: str
""" """
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) 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("/subscribe") msg = OSCMessage(b"/subscribe")
msg.appendTypedArg(self.own_address[0], "s") msg.appendTypedArg(bytes(self.own_address[0], "ascii"), b"s")
msg.appendTypedArg(self.own_address[1], "i") msg.appendTypedArg(self.own_address[1], b"i")
msg.appendTypedArg(self.args.authenticate, "s") msg.appendTypedArg(bytes(self.args.authenticate, "ascii"), b"s")
if self.args.subscriber_label is not None: if self.args.subscriber_label is not None:
msg.appendTypedArg(self.args.subscriber_label, "s") msg.appendTypedArg(bytes(self.args.subscriber_label, "ascii"), b"s")
self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address) self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address)
@ -124,11 +113,11 @@ class OSCThread(threading.Thread):
if self.args.keep_subscribed: if self.args.keep_subscribed:
return return
print "%s: unsubscribing from '%s:%d'" % (datetime.now().strftime("%x %X"), self.chaosc_address[0], self.chaosc_address[1]) print("%s: unsubscribing from '%s:%d'" % (datetime.now().strftime("%x %X"), self.chaosc_address[0], self.chaosc_address[1]))
msg = OSCMessage("/unsubscribe") msg = OSCMessage(b"/unsubscribe")
msg.appendTypedArg(self.own_address[0], "s") msg.appendTypedArg(bytes(self.own_address[0], "ascii"), b"s")
msg.appendTypedArg(self.own_address[1], "i") msg.appendTypedArg(self.own_address[1], b"i")
msg.appendTypedArg(self.args.authenticate, "s") msg.appendTypedArg(bytes(self.args.authenticate, "ascii"), b"s")
self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address) self.osc_sock.sendto(msg.encode_osc(), self.chaosc_address)
def run(self): def run(self):
@ -138,18 +127,18 @@ class OSCThread(threading.Thread):
if reads: if reads:
osc_input = reads[0].recv(4096) osc_input = reads[0].recv(4096)
osc_address, typetags, messages = decode_osc(osc_input, 0, len(osc_input)) osc_address, typetags, messages = decode_osc(osc_input, 0, len(osc_input))
#print "thread osc_address", osc_address #print("thread osc_address", osc_address)
if osc_address.find("ekg") > -1 or osc_address.find("plot") != -1: if osc_address.find(b"ekg") > -1 or osc_address.find(b"plot") != -1:
queue.put_nowait((osc_address, messages)) msg_queue.put_nowait((osc_address, messages))
else: else:
queue.put_nowait(("/bjoern/ekg", [0])) msg_queue.put_nowait((b"/bjoern/ekg", [0]))
queue.put_nowait(("/merle/ekg", [0])) msg_queue.put_nowait((b"/merle/ekg", [0]))
queue.put_nowait(("/uwe/ekg", [0])) msg_queue.put_nowait((b"/uwe/ekg", [0]))
self.unsubscribe_me() self.unsubscribe_me()
print "OSCThread is going down" print("OSCThread is going down")
queue = Queue.Queue() msg_queue = queue.Queue()
class Actor(object): class Actor(object):
shadowPen = pg.mkPen(255, 255, 255) shadowPen = pg.mkPen(255, 255, 255)
@ -231,8 +220,9 @@ class Actor(object):
class EkgPlot(object): class EkgPlot(object):
def __init__(self, actor_names, num_data, colors): def __init__(self, actor_names, num_data, colors):
self.qtapp = QtGui.QApplication([])
self.plot = pg.PlotWidget(title="<h1>EKG</h1>") self.plot = pg.PlotWidget(title="<h1>EKG</h1>")
self.plot.setAttribute(QtCore.Qt.WA_DeleteOnClose)
self.plot.hide() self.plot.hide()
self.plot.setLabel('left', "<h2>Amplitude</h2>") self.plot.setLabel('left', "<h2>Amplitude</h2>")
self.plot.setLabel('bottom', "<h2><sup>Time</sup></h2>") self.plot.setLabel('bottom', "<h2><sup>Time</sup></h2>")
@ -256,8 +246,8 @@ class EkgPlot(object):
self.set_positions() self.set_positions()
self.ekg_regex = re.compile("^/(.*?)/ekg$") self.ekg_regex = re.compile(b"^/(.*?)/ekg$")
self.ctl_regex = re.compile("^/plot/(.*?)$") self.ctl_regex = re.compile(b"^/plot/(.*?)$")
self.updated_actors = set() self.updated_actors = set()
@ -298,12 +288,12 @@ class EkgPlot(object):
actor_name = res.group(1) actor_name = res.group(1)
actor_obj = self.actors[actor_name] actor_obj = self.actors[actor_name]
if value == 1 and not actor_obj.active: if value == 1 and not actor_obj.active:
print "actor on", actor_name print("actor on", actor_name)
self.plot.addItem(actor_obj) self.plot.addItem(actor_obj)
actor_obj.active = True actor_obj.active = True
self.active_actors.append(actor_obj) self.active_actors.append(actor_obj)
elif value == 0 and not actor_obj.active: elif value == 0 and not actor_obj.active:
print "actor off", actor_name print("actor off", actor_name)
self.plot.removeItem(actor_obj) self.plot.removeItem(actor_obj)
actor_obj.active = True actor_obj.active = True
self.active_actors.remove(actor_obj) self.active_actors.remove(actor_obj)
@ -313,45 +303,43 @@ class EkgPlot(object):
class MyHandler(BaseHTTPRequestHandler): class MyHandler(BaseHTTPRequestHandler):
def __del__(self):
self.thread.running = False
self.thread.join()
def do_GET(self): def do_GET(self):
try: try:
self.path=re.sub('[^.a-zA-Z0-9]', "",str(self.path)) self.path=re.sub('[^.a-zA-Z0-9]', "",str(self.path))
if self.path=="" or self.path==None or self.path[:1]==".": if self.path=="" or self.path==None or self.path[:1]==".":
self.send_error(403,'Forbidden') self.send_error(403)
if self.path.endswith(".html"): if self.path.endswith(".html"):
f = open(curdir + sep + self.path) directory = os.path.dirname(os.path.abspath(__file__))
data = open(os.path.join(directory, self.path), "rb").read()
self.send_response(200) self.send_response(200)
self.send_header('Content-type', 'text/html') self.send_header('Content-type', 'text/html')
self.end_headers() self.end_headers()
self.wfile.write(f.read()) self.wfile.write(data)
f.close()
elif self.path.endswith(".mjpeg"): elif self.path.endswith(".mjpeg"):
self.thread = thread = OSCThread(self.server.args) self.thread = thread = OSCThread(self.server.args)
thread.daemon = True thread.daemon = True
thread.start() thread.start()
self.send_response(200) self.send_response(200)
actor_names = ["bjoern", "merle", "uwe"] actor_names = [b"bjoern", b"merle", b"uwe"]
num_data = 100 num_data = 100
colors = ["r", "g", "b"] colors = ["r", "g", "b"]
plotter = EkgPlot(actor_names, num_data, colors) self.plotter = plotter = EkgPlot(actor_names, num_data, colors)
self.wfile.write("Content-Type: multipart/x-mixed-replace; boundary=--aaboundary\r\n\r\n") self.send_header("Content-Type", "multipart/x-mixed-replace; boundary=--aaboundary\r\n\r\n")
self.end_headers()
#lastTime = time.time() #lastTime = time.time()
#fps = None #fps = None
while 1: while 1:
while 1: while 1:
try: try:
osc_address, args = queue.get_nowait() osc_address, args = msg_queue.get_nowait()
except Queue.Empty: except queue.Empty:
break break
plotter.update(osc_address, args[0]) plotter.update(osc_address, args[0])
@ -361,8 +349,28 @@ class MyHandler(BaseHTTPRequestHandler):
buffer = QBuffer() buffer = QBuffer()
buffer.open(QIODevice.WriteOnly) buffer.open(QIODevice.WriteOnly)
img.save(buffer, "JPG", 100) img.save(buffer, "JPG", 100)
JpegData = buffer.data() JpegData = buffer.data().data()
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)) 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()
self.plotter.qtapp.processEvents()
del self.plotter.plot
del self.plotter.qtapp
del self.plotter
del plotter
thread.running = False
thread.join()
return
del JpegData del JpegData
del buffer del buffer
del img del img
@ -378,17 +386,19 @@ class MyHandler(BaseHTTPRequestHandler):
#print '%0.2f fps' % fps #print '%0.2f fps' % fps
elif self.path.endswith(".jpeg"): elif self.path.endswith(".jpeg"):
f = open(curdir + sep + self.path) directory = os.path.dirname(os.path.abspath(__file__))
data = open(os.path.join(directory, self.path), "rb").read()
self.send_response(200) self.send_response(200)
self.send_header('Content-type','image/jpeg') self.send_header('Content-type','image/jpeg')
self.end_headers() self.end_headers()
self.wfile.write(f.read()) self.wfile.write(data)
f.close()
return return
except (KeyboardInterrupt, SystemError): except (KeyboardInterrupt, SystemError):
print "queue size", queue.qsize() print("queue size", queue.qsize())
thread.running = False thread.running = False
thread.join() thread.join()
if hasattr(self, "plotter"):
plotter.plot.deleteLater()
except IOError: except IOError:
self.send_error(404,'File Not Found: %s' % self.path) self.send_error(404,'File Not Found: %s' % self.path)
@ -411,19 +421,22 @@ def main():
http_host, http_port = resolve_host(args.http_host, args.http_port) http_host, http_port = resolve_host(args.http_host, args.http_port)
print http_host, http_port print(http_host, http_port)
server = JustAHTTPServer((http_host, http_port), MyHandler) server = JustAHTTPServer((http_host, http_port), MyHandler)
server.args = args server.args = args
print "%s: starting up http server on '%s:%d'" % ( print("%s: starting up http server on '%s:%d'" % (
datetime.now().strftime("%x %X"), http_host, http_port) datetime.now().strftime("%x %X"), http_host, http_port))
print "before start:" print("before start:")
#objgraph.show_growth() #objgraph.show_growth()
try: try:
server.serve_forever() server.serve_forever()
except KeyboardInterrupt: except KeyboardInterrupt:
print '^C received, shutting down server' print('^C received, shutting down server')
print()
#print(gc.garbage)
#print()
#print "queue size", queue.qsize() #print "queue size", queue.qsize()
#print "show growth", objgraph.show_growth() #print "show growth", objgraph.show_growth()
#import random #import random

1
ekgplotter/setup.py
View File

@ -7,6 +7,7 @@ use_setuptools()
import sys import sys
from setuptools import find_packages, setup from setuptools import find_packages, setup
extras = dict()
if sys.version_info >= (3,): if sys.version_info >= (3,):
extras['use_2to3'] = True extras['use_2to3'] = True