import sys, struct, time, serial

'connects to a fnordlicht'
""" http://github.com/fd0/fnordlicht/blob/master/doc/PROTOCOL """

class Fnordlicht:

    tty = ''
    verbose = False

    def __init__(self, tty, verbose = False, addr = 255):
        self.tty = tty
        self.addr = addr
        if verbose:
            self.verbose = True
            print 'using tty "%s", address "%s"' % (self.tty, self.addr)

    def __send(self, data):
        if self.verbose:
            print 'sending package:',data
        # pad data package with 0-characters
        for i in range(15 - len(data)):
            data.append(0)
        try:
            # open serial port at '19200,8N1', 1s timeout
            ser = serial.Serial(self.tty, 19200, timeout=1)
            for i in range(15):
                ser.write(chr(data[i]))
            ser.flush()
            ser.close()
        except:
            print 'unable to send string on %s: %s' % (self.tty, sys.exc_info()[0])

    def sync(self):
        if self.verbose:
            print 'sync()'
        data = []
        for i in range(14):
            data.append(0x1b)
        data.append(self.addr)
        self.__send(data)

    def fade_rgb(self, r, g, b, step, delay):
        if self.verbose:
            print 'fade_rgb(%d, %d, %d, %d, %d)' % (r, g, b, step, delay)
        data = [self.addr, 0x01]
        data.append(step)
        data.append(delay)
        data.append(r)
        data.append(g)
        data.append(b)
        self.__send(data)

    def fade_hsv(self, h, s, v, step, delay):
        if self.verbose:
            print 'fade_hsv(%d, %d, %d, %d, %d)' % (h, s, v, step, delay)
        data = [self.addr, 0x02]
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<h', h)[0]))
        data.append(ord(struct.pack('<h', h)[1]))
        data.append(s)
        data.append(v)
        self.__send(data)

    def save_rgb(self, slot, r, g, b, step, delay, pause):
        # TODO untested
        if self.verbose:
            print 'save_rgb(%d, %d, %d, %d, %d, %d, %d)' % (slot, r, g, b, step, delay, pause)
        data = [self.addr, 0x03]
        data.append(slot)
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<H', pause)[0]))
        data.append(ord(struct.pack('<H', pause)[1]))
        data.append(r)
        data.append(g)
        data.append(b)
        self.__send(data)

    def save_hsv(self, slot, h, s, v, step, delay, pause):
        # TODO untested
        if self.verbose:
            print 'save_hsv(%d, %d, %d, %d, %d, %d, %d)' % (slot, h, s, v, step, delay, pause)
        data = [self.addr, 0x04]
        data.append(slot)
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<H', pause)[0]))
        data.append(ord(struct.pack('<H', pause)[1]))
        data.append(ord(struct.pack('<h', h)[0]))
        data.append(ord(struct.pack('<h', h)[1]))
        data.append(s)
        data.append(v)
        self.__send(data)

    def save_current(self, slot, step, delay, pause):
        # TODO untested
        if self.verbose:
            print 'save_current(%d, %d, %d, %d)' % (slot, step, delay, pause)
        data = [self.addr, 0x05]
        data.append(slot)
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<H', pause)[0]))
        data.append(ord(struct.pack('<H', pause)[1]))
        self.__send(data)

    def config_offsets(self, step, delay, h, s, v):
        # TODO untested
        if self.verbose:
            print 'config_offsets(%d, %d, %d, %d, %d)' % (step, delay, h, s, v)
        data = [self.addr, 0x06]
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<h', h)[0]))
        data.append(ord(struct.pack('<h', h)[1]))
        data.append(s)
        data.append(v)
        self.__send(data)

    def __start_program(self, program, params):
        if self.verbose:
            print 'start_program(%d, %s)' % (program, params)
        data = [self.addr, 0x07]
        data.append(program)
        data.extend(params)
        self.__send(data)

    def program_colorwheel(self, fade_step, fade_delay, fade_sleep, hue_start, hue_step, add_addr, saturation, value):
        if self.verbose:
            print 'program_colorwheel(%d, %d, %d, %d, %d, %d, %d, %d)' % (fade_step, fade_delay, fade_sleep, hue_start, hue_step, add_addr, saturation, value)
        data = [fade_step, fade_delay, fade_sleep]
        data.append(ord(struct.pack('<H', hue_start)[0]))
        data.append(ord(struct.pack('<H', hue_start)[1]))
        data.append(ord(struct.pack('<h', hue_step)[0]))
        data.append(ord(struct.pack('<h', hue_step)[1]))
        data.append(add_addr)
        data.append(saturation)
        data.append(value)
        self.__start_program(0, data)

    def program_random(self, seed, flags, fade_step, fade_delay, fade_sleep, saturation, value, min_distance):
        if self.verbose:
            print 'program_random(%d, %d, %d, %d, %d, %d, %d, %d)' % (seed, flags, fade_step, fade_delay, fade_sleep, saturation, value, min_distance)
        data = []
        data.append(ord(struct.pack('<H', seed)[0]))
        data.append(ord(struct.pack('<H', seed)[1]))
        data.append(flags)
        data.append(fade_step)
        data.append(fade_delay)
        data.append(ord(struct.pack('<H', fade_sleep)[0]))
        data.append(ord(struct.pack('<H', fade_sleep)[1]))
        data.append(saturation)
        data.append(value)
        data.append(min_distance)
        self.__start_program(1, data)

    def stop(self, fading = 1):
        if self.verbose:
            print 'stop(%d)' % (fading)
        data = [self.addr, 0x08]
        data.append(fading)
        self.__send(data)

    def modify_current(self, step, delay, r, g, b, h, s, v):
        if self.verbose:
            print 'modify_current(%d, %d, %d, %d, %d, %d, %d, %d)' % (step, delay, r, g, b, h, s, v)
        data = [self.addr, 0x09]
        data.append(step)
        data.append(delay)
        data.append(ord(struct.pack('<b', r)))
        data.append(ord(struct.pack('<b', g)))
        data.append(ord(struct.pack('<b', b)))
        data.append(ord(struct.pack('<h', h)[0]))
        data.append(ord(struct.pack('<h', h)[1]))
        data.append(ord(struct.pack('<b', s)))
        data.append(ord(struct.pack('<b', v)))
        self.__send(data)

    def pull_int(self, delay):
        # TODO untested
        if self.verbose:
            print 'pull_int(%d)' % (delay)
        data = [self.addr, 0x0a]
        data.append(delay)
        self.__send(data)

    def config_startup_nothing(self):
        # TODO untested
        if self.verbose:
            print 'config_startup_nothing()'
        data = [self.addr, 0x0b]
        data.append(0)
        self.__send(data)

    def config_startup_program(self, program, params):
        # TODO untested
        if self.verbose:
            print 'config_startup_program(%d, %s)' % (program, params)
        data = [self.addr, 0x0b]
        data.append(1)
        data.append(program)
        data.extend(params)
        self.__send(data)

    def powerdown(self):
        # TODO untested
        if self.verbose:
            print 'powerdown()'
        data = [self.addr, 0x0c]
        self.__send(data)

    # secondary functions

    def fade_updown_rgb(self, r, g, b, step, delay, sleep_time):
        self.fade_rgb(r, g, b, step, delay)
        time.sleep(sleep_time)
        self.fade_rgb(0, 0, 0, step, delay)
        time.sleep(sleep_time)

    def fade_updown_hsv(self, h, s, v, step, delay, sleep_time):
        self.fade_hsv(h, s, v, step, delay)
        time.sleep(sleep_time)
        self.fade_hsv(h, s, 0, step, delay)
        time.sleep(sleep_time)


#fnordlicht = Fnordlicht('/dev/ttyUSB0', verbose = True)
#fnordlicht.sync()
#fnordlicht.fade_rgb(255, 0, 0, 100, 1)
#fnordlicht.fade_hsv(300, 255, 255, 100, 1)
#fnordlicht.start_program(0, [1, 1, 1, 0, 0, 60, 0, 255, 255, 255]) # colorwheel
#fnordlicht.stop(fading = 1)
#fnordlicht.modify_current(1, 1,   0, 0, 0,   100, 100, 100)
#fnordlicht.fade_updown_rgb(255, 0, 0, 10, 1, 1)
#fnordlicht.program_random(23, 3, 1, 5, 1, 255, 255, 90)