luxafor-python/luxafor-linux.py at master · aphex008/luxafor-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
import usb.core
import usb.util
import sys
import argparse

DEVICE = None
ACTION = None
LED = None

HEX = None
RED = None
GREEN = None
BLUE = None
SPEED = None
REPEAT = None
WAVE = None
PATTERN = None


def main():
    global RED
    global GREEN
    global BLUE
    setupDevice()
    setupArgs()

    if HEX:
        rgb = hex_to_rgb(HEX)
        RED = rgb[0]
        GREEN = rgb[1]
        BLUE = rgb[2]

    # Determine which action
    if ACTION == 'color':
        setColor()
    elif ACTION == 'fade':
        setFade()
    elif ACTION == 'strobe':
        setStrobe()
    elif ACTION == 'wave':
        setWave()
    elif ACTION == 'pattern':
        setPattern()

def setupArgs():
    global RED
    global GREEN
    global BLUE
    global SPEED
    global REPEAT
    global WAVE
    global PATTERN
    global ACTION
    global LED
    global HEX

    parser = initArgParser()
    args = parser.parse_args()

    ACTION  = args.action if args.action else 'color'
    RED     = args.r if args.r else 0
    GREEN   = args.g if args.g else 0
    BLUE    = args.b if args.b else 0
    SPEED   = args.s if args.s else 0
    REPEAT  = args.t if args.t else 0
    WAVE    = args.w if args.w else 0
    PATTERN = args.p if args.p else 0
    LED     = args.l if args.l else 255
    HEX     = args.x if args.x else 0


def hex_to_rgb(value): # http://stackoverflow.com/a/214657
    value = value.lstrip('#')
    lv = len(value)
    return tuple(int(value[i:i + lv // 3], 16) for i in range(0, lv, lv // 3))

def setupDevice():
    global DEVICE
    DEVICE = usb.core.find(idVendor=0x04d8, idProduct=0xf372)

    # Device found?
    if DEVICE is None:
        raise ValueError('Device not found')

    # Linux kernel sets up a device driver for USB device, which you have to detach.
    # Otherwise trying to interact with the device gives a 'Resource Busy' error.
    try:
      DEVICE.detach_kernel_driver(0)
    except Exception, e:
      pass

    DEVICE.set_configuration()

def writeValue( values ):
    DEVICE.write(1, values)
    DEVICE.write(1, values) # Run it again to ensure it works.

def setPattern():
    writeValue( [6,PATTERN,REPEAT,0,0,0,0] )

def setWave():
    writeValue( [4,WAVE,RED,GREEN,BLUE,0,REPEAT,SPEED] )

def setStrobe():
    writeValue( [3,LED,RED,GREEN,BLUE,SPEED,0,REPEAT] )

def setFade():
    writeValue( [2,LED,RED,GREEN,BLUE,SPEED,0] )

def setColor():
    writeValue( [1,LED,RED,GREEN,BLUE,0,0] )

def initArgParser():

    # Setup argument parser
    parser = argparse.ArgumentParser(description='Luxafor Arguments')
    parser.add_argument('action', help='Action', choices=["color", "fade", "wave", "strobe", "pattern"])
    parser.add_argument('-l', help='LED', type=int)
    parser.add_argument('-b', help='Blue Value', type=int)
    parser.add_argument('-r', help='Red Value', type=int)
    parser.add_argument('-g', help='Green Value', type=int)
    parser.add_argument('-s', help='Speed Value', type=int)
    parser.add_argument('-t', help='Repeat Value', type=int)
    parser.add_argument('-w', help='Wave Value', type=int)
    parser.add_argument('-p', help='Pattern Value', type=int)
    parser.add_argument('-x', help='Hex Color', type=str)

    return parser

if __name__ == '__main__':
    main()