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
|
"""Generate example of Twinkly movie file"""
import argparse
import random
from ttls.client import TwinklyFrame
RED = (0xFF, 0x00, 0x00)
GREEN = (0x00, 0xFF, 0x00)
BLUE = (0x00, 0x00, 0xFF)
def generate_xmas_frame(n: int) -> TwinklyFrame:
"""Generate a very merry frame"""
res = []
for _ in range(n):
if random.random() > 0.5:
res.append(RED)
else:
res.append(GREEN)
return res
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument(
"--leds",
dest="leds",
metavar="n",
type=int,
default=105,
required=False,
help="Number of LEDs",
)
parser.add_argument(
"--count",
dest="count",
metavar="n",
type=int,
default=10,
required=False,
help="Number of iterations",
)
parser.add_argument(
"--output",
dest="output",
metavar="filename",
type=str,
default="movie.bin",
help="Output file",
)
args = parser.parse_args()
movie = []
for _ in range(args.count):
frame = [v for sublist in generate_xmas_frame(args.leds) for v in sublist]
movie.extend(frame)
with open(args.output, "wb") as f:
f.write(bytes(movie))
if __name__ == "__main__":
main()
|
