stream-sndh-fast.py

import Queue
import serial # using pyserial package - https://pypi.org/project/pyserial/
import sys
import threading
import time

class UART_manager(threading.Thread):

    def __init__(self, fd, q):
        super(UART_manager, self).__init__()
        self.__fd = fd
        self.__q = q

    def run(self):
        time.sleep(1) # Wait for buffer to fill up
        while not self.__q.empty():
            # Read number of bytes to send
            available = ord(self.__fd.read())
            available = (available<<8) + ord(self.__fd.read())
            sys.stdout.write("\x1b[2K\rYM_empty: {}\tPC_buffer: {}".format(available, self.__q.qsize()))
            sys.stdout.flush()
            # Assume we fill the PC buffer faster than the ATmega buffer get consumed
            self.__fd.write(''.join([self.__q.get() for _ in xrange(available)]))


def main():
    fd = serial.Serial("/dev/ttyUSB0", 500000, timeout=2)
    q = Queue.Queue()
    um = UART_manager(fd, q)

    um.start()
    l = sys.stdin.readline()
    prev_ts = 0
    while l:
        if q.qsize() < 3000:
            _, ts, regs = l.split()
            ts = int(ts, 16)
            delta = ts - prev_ts
            prev_ts = ts
            q.put(chr(delta >> 8))
            q.put(chr(delta & 0xff))
            en_regs = filter(lambda x:x[1] != '..', enumerate(regs.split('-')))
            for n,r in en_regs:
                q.put(chr(n))
                q.put(chr(int(r,16)))
            q.put('\xff') # End of sample
            l = sys.stdin.readline()
        else:
            time.sleep(0.01)
    q.put('\xfe') # End of song

    um.join()
    fd.close()
    print ""


main()