added copyright notice and improved readability and improved the ringbuffer

This commit is contained in:
Stefan Kögl 2014-03-09 21:24:39 +01:00
parent 9dcddddde0
commit 1802548cc6
1 changed files with 33 additions and 32 deletions

View File

@ -1,6 +1,23 @@
#!/usr/bin/python #!/usr/bin/python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# This file is part of chaosc
#
# chaosc 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.
#
# chaosc 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 chaosc. If not, see <http://www.gnu.org/licenses/>.
#
# Copyright (C) 2014 Stefan Kögl
import argparse import argparse
import os.path import os.path
import select import select
@ -9,7 +26,6 @@ import socket
import sys import sys
import datetime import datetime
try: try:
from chaosc.c_osc_lib import OSCMessage from chaosc.c_osc_lib import OSCMessage
except ImportError as e: except ImportError as e:
@ -64,10 +80,6 @@ class EHealth2OSC(Forwarder):
osc_sock.sendall(osc_message.encode_osc()) osc_sock.sendall(osc_message.encode_osc())
class EKG2OSC(Forwarder): class EKG2OSC(Forwarder):
def __init__(self, actor, platform, device): def __init__(self, actor, platform, device):
super(EKG2OSC, self).__init__(actor, platform, device) super(EKG2OSC, self).__init__(actor, platform, device)
@ -79,7 +91,6 @@ class EKG2OSC(Forwarder):
osc_sock.sendall(osc_message.encode_osc()) osc_sock.sendall(osc_message.encode_osc())
class RingBuffer(object): class RingBuffer(object):
def __init__(self, length): def __init__(self, length):
self.length = length self.length = length
@ -98,7 +109,7 @@ class RingBuffer(object):
if value == -1: if value == -1:
raise ValueError("not complete") raise ValueError("not complete")
data.append(value) data.append(value)
if data[0] != 0x0 and data[1] != 0xff: if data[0] != 0x0 or data[1] != 0xff:
raise ValueError("not synced") raise ValueError("not synced")
return data[2:] return data[2:]
@ -108,7 +119,7 @@ class Pulse2OSC(Forwarder):
def __init__(self, actor, platform, device): def __init__(self, actor, platform, device):
super(Pulse2OSC, self).__init__(actor, platform, device) super(Pulse2OSC, self).__init__(actor, platform, device)
self.buf = RingBuffer(6) self.buf = RingBuffer(6)
self.heartbeat_send = False self.heartbeat_on = False
def handle_read(self, osc_sock): def handle_read(self, osc_sock):
t = ord(self.serial.read(1)) t = ord(self.serial.read(1))
@ -118,17 +129,17 @@ class Pulse2OSC(Forwarder):
try: try:
heart_signal, heart_rate, o2, pulse = self.buf.getData() heart_signal, heart_rate, o2, pulse = self.buf.getData()
if pulse == 245 and not self.heartbeat_send: if pulse == 245 and not self.heartbeat_on:
osc_message = OSCMessage("/%s/heartbeat" % self.actor) osc_message = OSCMessage("/%s/heartbeat" % self.actor)
osc_message.appendTypedArg(1, "i") osc_message.appendTypedArg(1, "i")
osc_message.appendTypedArg(heart_rate, "i") osc_message.appendTypedArg(heart_rate, "i")
osc_message.appendTypedArg(o2, "i") osc_message.appendTypedArg(o2, "i")
osc_sock.sendall(osc_message.encode_osc()) osc_sock.sendall(osc_message.encode_osc())
print "heartbeat", datetime.datetime.now(), heart_signal print "heartbeat", datetime.datetime.now(), heart_signal
self.heartbeat_send = True self.heartbeat_on = True
elif pulse == 1 and self.heartbeat_send: elif pulse == 1 and self.heartbeat_on:
#print "off heartbeat", datetime.datetime.now(), heart_signal #print "off heartbeat", datetime.datetime.now(), heart_signal
self.heartbeat_send = False self.heartbeat_on = False
osc_message = OSCMessage("/%s/heartbeat" % self.actor) osc_message = OSCMessage("/%s/heartbeat" % self.actor)
osc_message.appendTypedArg(0, "i") osc_message.appendTypedArg(0, "i")
osc_message.appendTypedArg(heart_rate, "i") osc_message.appendTypedArg(heart_rate, "i")
@ -138,7 +149,6 @@ class Pulse2OSC(Forwarder):
print e print e
def main(): def main():
parser = argparse.ArgumentParser(prog='psychose_actor') parser = argparse.ArgumentParser(prog='psychose_actor')
parser.add_argument("-H", '--chaosc_host', required=True, parser.add_argument("-H", '--chaosc_host', required=True,
@ -146,29 +156,21 @@ def main():
parser.add_argument("-p", '--chaosc_port', required=True, parser.add_argument("-p", '--chaosc_port', required=True,
type=int, help='port of chaosc instance to control') type=int, help='port of chaosc instance to control')
args = parser.parse_args(sys.argv[1:]) args = parser.parse_args(sys.argv[1:])
osc_sock = socket.socket(2, 2, 17) osc_sock = socket.socket(2, 2, 17)
osc_sock.connect((args.chaosc_host, args.chaosc_port)) osc_sock.connect((args.chaosc_host, args.chaosc_port))
naming = { naming = {
"/dev/ttyUSB0" : ["merle", "ehealth"], "/dev/ttyUSB0" : ["bjoern", "ehealth"],
"/dev/ttyUSB1" : ["merle", "ekg"], "/dev/ttyACM0" : ["bjoern", "ekg"],
"/dev/ttyUSB2" : ["merle", "pulse"], "/dev/ttyACM1" : ["bjoern", "pulse"],
"/dev/ttyUSB3" : ["bjoern", "ehealth"], "/dev/ttyUSB1" : ["merle", "ehealth"],
"/dev/ttyUSB4" : ["bjoern", "ekg"], "/dev/ttyACM2" : ["merle", "ekg"],
"/dev/ttyUSB5" : ["bjoern", "pulse"], "/dev/ttyACM3" : ["merle", "pulse"],
"/dev/ttyUSB6" : ["uwe", "ehealth"], "/dev/ttyUSB2" : ["uwe", "ehealth"],
"/dev/ttyUSB7" : ["uwe", "ekg"], "/dev/ttyACM4" : ["uwe", "ekg"],
"/dev/ttyUSB8" : ["uwe", "pulse"] "/dev/ttyACM5" : ["uwe", "pulse"]
}
naming = {
#"/dev/ttyACM0" : ["merle", "pulse"],
"/dev/ttyUSB0" : ["merle", "ehealth"],
#"/dev/ttyACM1" : ["merle", "pulse"]
} }
used_devices = dict() used_devices = dict()
@ -201,6 +203,5 @@ def main():
read_map[forwarder.serial] = forwarder.handle_read read_map[forwarder.serial] = forwarder.handle_read
readers, writers, errors = select.select(read_map, [], [], 0.1) readers, writers, errors = select.select(read_map, [], [], 0.1)
#print "readers", readers
for reader in readers: for reader in readers:
read_map[reader](osc_sock) read_map[reader](osc_sock)