from __future__ import division
import logging
import usb.core
from usb.util import CTRL_TYPE_VENDOR, CTRL_RECIPIENT_DEVICE, CTRL_OUT

log = logging.getLogger('udmx')

"""
Send dmx to one of these:
http://www.amazon.com/Interface-Adapter-Controller-Lighting-Freestyler/dp/B00W52VIOS

[4520784.059479] usb 1-2.3: new low-speed USB device number 6 using xhci_hcd
[4520784.157410] usb 1-2.3: New USB device found, idVendor=16c0, idProduct=05dc
[4520784.157416] usb 1-2.3: New USB device strings: Mfr=1, Product=2, SerialNumber=3
[4520784.157419] usb 1-2.3: Product: uDMX
[4520784.157422] usb 1-2.3: Manufacturer: www.anyma.ch
[4520784.157424] usb 1-2.3: SerialNumber: ilLUTZminator001

See https://www.illutzmination.de/udmxfirmware.html?&L=1
    sources/commandline/uDMX.c
or https://github.com/markusb/uDMX-linux/blob/master/uDMX.c
"""

cmd_SetChannelRange = 0x0002

class Udmx(object):
    def __init__(self, bus):
        self.dev = None
        for dev in usb.core.find(idVendor=0x16c0, idProduct=0x05dc, find_all=True):
            print "udmx device at %r" % dev.bus
            if bus is None or bus == dev.bus:
                self.dev = dev
        if not self.dev:
            raise IOError('no matching udmx device found for requested bus %r' % bus)
        log.info('found udmx at %r', self.dev)
        
    def SendDMX(self, buf):
        ret = self.dev.ctrl_transfer(
           bmRequestType=CTRL_TYPE_VENDOR | CTRL_RECIPIENT_DEVICE | CTRL_OUT,
           bRequest=cmd_SetChannelRange,
           wValue=len(buf),
           wIndex=0,
           data_or_wLength=buf)
        if ret < 0:
            raise ValueError("ctrl_transfer returned %r" % ret)


def demo(chan, fps=44):
    import time, math
    u = Udmx()
    while True:
        nsin = math.sin(time.time() * 6.28) / 2.0 + 0.5
        nsin8 = int(255 * nsin)
        try:
            u.SendDMX('\x00' * (chan - 1) +
                      chr(210) +
                      chr(nsin8) + chr(nsin8) + chr(nsin8))
        except usb.core.USBError as e:
            print "err", time.time(), repr(e)
        time.sleep(1 / fps)