#!/usr/bin/python # -*- 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 . # # Copyright (C) 2014 Stefan Kögl import argparse import os.path import select import serial import socket import sys import datetime try: from chaosc.c_osc_lib import OSCMessage except ImportError as e: from chaosc.osc_lib import OSCMessage class Forwarder(object): def __init__(self, actor, platform, device): self.actor = actor self.platform = platform self.device = device self.serial = serial.Serial() self.serial.port = device self.serial.baudrate = 115200 self.serial.timeout = 0 self.buf_ser2osc = "" self.serial.open() def close(self): """Close all resources and unpublish service""" logger.info("%s: closing...", self.device) self.serial.close() class EHealth2OSC(Forwarder): def __init__(self, actor, platform, device): super(EHealth2OSC, self).__init__(actor, platform, device) def handle_read(self, osc_sock): data = self.serial.readline()[:-2] try: airFlow, emg, temp = data.split(";") except ValueError: return try: airFlow = int(airFlow) emg = int(emg) temp = int(temp); except ValueError: return osc_message = OSCMessage("/%s/airFlow" % self.actor) osc_message.appendTypedArg(airFlow, "i") osc_sock.sendall(osc_message.encode_osc()) osc_message = OSCMessage("/%s/emg" % self.actor) osc_message.appendTypedArg(emg, "i") osc_sock.sendall(osc_message.encode_osc()) osc_message = OSCMessage("/%s/temperatur" % self.actor) osc_message.appendTypedArg(temp, "i") osc_sock.sendall(osc_message.encode_osc()) class EKG2OSC(Forwarder): def __init__(self, actor, platform, device): super(EKG2OSC, self).__init__(actor, platform, device) def handle_read(self, osc_sock): t = ord(self.serial.read(1)) osc_message = OSCMessage("/%s/ekg" % self.actor) osc_message.appendTypedArg(t, "i") osc_sock.sendall(osc_message.encode_osc()) class RingBuffer(object): def __init__(self, length): self.length = length self.ring_buf = list() self.reset() def reset(self): self.ring_buf = [-1] * self.length self.head = 0 def append(self, value): self.ring_buf[self.head] = value self.head = (self.head + 1) % self.length def getData(self): #print "getData", self.ring_buf, self.head data = list() for i in range(7, 1, -1): value = self.ring_buf[(self.head - i) % self.length] if value == -1: self.reset() self.ring_buf[0] = 0 self.head = 1 raise ValueError("not complete - ringbuffer resettet") data.append(value) if data[0] != 0x0 or data[1] != 0xff: print "issue", data self.reset() self.ring_buf[0] = 0 self.head = 1 raise ValueError("not synced - ringbuffer resettet") return data[2:] class Pulse2OSC(Forwarder): def __init__(self, actor, platform, device): super(Pulse2OSC, self).__init__(actor, platform, device) self.buf = RingBuffer(6) self.heartbeat_on = False def handle_read(self, osc_sock): t = ord(self.serial.read(1)) self.buf.append(t) if t == 0: try: heart_signal, heart_rate, o2, pulse = self.buf.getData() if pulse == 245 and not self.heartbeat_on: osc_message = OSCMessage("/%s/heartbeat" % self.actor) osc_message.appendTypedArg(1, "i") osc_message.appendTypedArg(heart_rate, "i") osc_message.appendTypedArg(o2, "i") osc_sock.sendall(osc_message.encode_osc()) #print "heartbeat", datetime.datetime.now(), heart_signal self.heartbeat_on = True elif pulse == 1 and self.heartbeat_on: #print "off heartbeat", datetime.datetime.now(), heart_signal self.heartbeat_on = False osc_message = OSCMessage("/%s/heartbeat" % self.actor) osc_message.appendTypedArg(0, "i") osc_message.appendTypedArg(heart_rate, "i") osc_message.appendTypedArg(o2, "i") osc_sock.sendall(osc_message.encode_osc()) except ValueError, e: print e def main(): parser = argparse.ArgumentParser(prog='psychose_actor') parser.add_argument("-H", '--chaosc_host', required=True, type=str, help='host of chaosc instance to control') parser.add_argument("-p", '--chaosc_port', required=True, type=int, help='port of chaosc instance to control') args = parser.parse_args(sys.argv[1:]) osc_sock = socket.socket(2, 2, 17) osc_sock.connect((args.chaosc_host, args.chaosc_port)) naming = { "/dev/ttyUSB0" : ["bjoern", "ehealth"], "/dev/ttyACM0" : ["bjoern", "ekg"], "/dev/ttyACM1" : ["bjoern", "pulse"], "/dev/ttyUSB1" : ["merle", "ehealth"], "/dev/ttyACM2" : ["merle", "ekg"], "/dev/ttyACM3" : ["merle", "pulse"], "/dev/ttyUSB2" : ["uwe", "ehealth"], "/dev/ttyACM4" : ["uwe", "ekg"], "/dev/ttyACM5" : ["uwe", "pulse"] } used_devices = dict() while 1: for device, description in naming.iteritems(): if os.path.exists(device): if device not in used_devices: actor, platform = naming[device] if description[1] == "ehealth": print device, actor, platform used_devices[device] = EHealth2OSC(actor, platform, device) elif description[1] == "ekg": print device, actor, platform used_devices[device] = EKG2OSC(actor, platform, device) elif description[1] == "pulse": print device, actor, platform used_devices[device] = Pulse2OSC(actor, platform, device) else: raise ValueError("unknown description %r for device %r" % (description, device)) else: print "device missing", device message = OSCMessage("/DeviceMissing") message.appendTypedArg(description[0], "s") message.appendTypedArg(description[1], "s") osc_sock.sendall(message.encode_osc()) read_map = {} for forwarder in used_devices.values(): read_map[forwarder.serial] = forwarder.handle_read readers, writers, errors = select.select(read_map, [], [], 0.1) for reader in readers: read_map[reader](osc_sock)