Bemis-100/run.py at master · mboyd/Bemis-100 · 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
#!/usr/bin/env python2.6
from __future__ import division

from led import bemis100, pattern, ledctl, ge, wave, new_wave
# from led import beat, spectrogram
import optparse, sys, os, time

if __name__ == '__main__':
    p = optparse.OptionParser(
        "Usage: python %prog [pattern | pattern_dir | wave | new_wave] [options]")

    p.add_option('-n', type="int", action='store', dest='num_lights',
                    default='50', help='Number of RGB pixels')

    p.add_option('-d', action="store", dest="device", default="COM5",
                    help="Device path")

    p.add_option('-f', type='int', action='store', dest='framerate',
                    default='30', help='Framerate (Hz, default 30)')

    p.add_option('-r', action='store_true', dest='repeat', default=False,
                    help="Repeat pattern[s] forever")

    p.add_option('-c', type='int', action='store', dest='count',
                    default='-1',
            help='Repetition count (exit when done, default is to loop forever)')

    p.add_option('-s','--sim', action='store_true', dest='sim',default=False,
                    help='Simulate the Bemis100 only')

    p.add_option('-b','--beat', action='store_true', dest='beat',default=False,
            help='Beat pattern')

    p.add_option('-w','--wave', action='store_true',dest='wave',default=False,
                 help='Wave beat pattern')

    p.add_option('-g', '--gelights', action='store_true', dest='ge', default=False,
            help="Use GE ColorEffects lights instead of Bemis100")

    (options, args) = p.parse_args()


    # if len(args) < 1:
    #     p.print_help()
    #     sys.exit(1)

    print "Opening port...",

    if not options.sim:
        if options.device == '':
            devices = filter(os.path.exists, ['/dev/tty.usbserial',
                                              '/dev/ttyUSB0',
                                              '/dev/tty.usbmodemfd141'])
            if len(devices) > 0:
                options.device = devices[0]
        if options.ge:
            print "GE Lights selected"
            b = ge.GEController(options.device,
                                num_lights=options.num_lights,
                                framerate=options.framerate)
        else:
            b = bemis100.Bemis100(options.device,
                                  num_boards=options.num_lights//2,
                                  framerate=options.framerate)
    else:
        b = ledctl.LEDController(options.device, options.framerate)

    print "done\nLoading patterns...",


    patterns = []

    if not args:
        args = ['new_wave']
    for fn in args:
        if os.path.isfile(fn):
            if options.beat:
                new_pattern = beat.BeatPattern(pattern.Bemis100Pattern(fn,
                    options.num_lights))
            else:
                new_pattern = pattern.Bemis100Pattern(fn,options.num_lights)
            patterns.append(new_pattern)
        elif os.path.isdir(fn):
            names = [os.path.join(fn, f) for f in os.listdir(fn)]
            patterns.extend([
                pattern.Bemis100Pattern(f, options.num_lights) for f in names])
        elif fn == 'wave':
            patterns.append(wave.WavePattern(num_lights=options.num_lights))
        elif fn == 'new_wave':
            patterns.append(new_wave.NewWavePattern(num_lights=options.num_lights))
        else:
            print "Not a pattern file or directory: %s\n\n" % fn

    print "done\nPlaying...",
    sys.stdout.flush()

    b.play()

    while True:
        print patterns
        for p in patterns:
            print p.image_data
            try:
                b.add_pattern(p, num_times=options.count, async=False)
                print "added pattern"
            except (KeyboardInterrupt, SystemExit):
                options.repeat = False
                break

        if not options.repeat:
            break

    b.quit()

    print 'done'