saber/run.py at develop · dash-dash/saber · 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
import argparse
from threading import Thread
from queue import Queue, Empty
import time

from saber import Saber, rgb

PATTERNS = ['r', 'g', 'b', 'rand', 's']


def run(thread_name, queue, brightness, freq):
    saber = Saber(brightness)
    print('Running')
    current_pattern = 'g'
    while True:
        try:
            item = queue.get(False)
            current_pattern = item
            # if item in PATTERNS:
            #     current_pattern = item
            if item is None:
                break
        except Empty:
            pass

        if current_pattern == 'r':
            saber.swipe(rgb(255, 0, 0), freq / 1000.0)
            saber.reverse_swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'g':
            saber.swipe(rgb(0, 255, 0), freq / 1000.0)
            saber.reverse_swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'b':
            saber.swipe(rgb(0, 0, 255), freq / 1000.0)
            saber.reverse_swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'rf':
            saber.swipe(rgb(255, 0, 0), freq / 1000.0)
            saber.swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'gf':
            saber.swipe(rgb(0, 255, 0), freq / 1000.0)
            saber.swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'bf':
            saber.swipe(rgb(0, 0, 255), freq / 1000.0)
            saber.swipe(rgb(0, 0, 0), freq / 1000.0)

        elif current_pattern == 'rand':
            saber.random(freq / 1000.0)

        elif current_pattern.startswith('s'):
            # s,30,40,233,1,5
            vals = current_pattern.split(',')
            r, g, b, s, size = int(vals[1]), int(vals[2]), int(vals[3]), int(vals[4]), int(vals[5])
            saber.up_down(rgb(r, g, b), s / 1000.0, size)

    saber.swipe(rgb(0, 0, 0), freq / 1000.0)


def key_input(queue):
    choice = input("")
    if choice == 'q':
        queue.put(None)
        return 0
    queue.put(choice)
    return 1


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-c', '--clear', action='store_true', help='clear the display on exit')
    parser.add_argument('-b', '--brightness', default=255, help='the max brightness of the strip', type=int)
    parser.add_argument('-f', '--frequency', default=50, help='frequency at witch it runs', type=float)
    args = parser.parse_args()

    print('Press Ctrl-C to quit.')
    if not args.clear:
        print('Use "-c" argument to clear LEDs on exit')
    print('brightness: {}, frequency: {}'.format(args.brightness, args.frequency))
    q = Queue()
    thread = Thread(target=run, args=('run', q, args.brightness, args.frequency))
    thread.start()

    while key_input(queue=q):
        pass