NeoPixel/DancingTeethSlowDemo.py at main · alextoft/NeoPixel · 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
import time
import board
import neopixel
import random

# GPIO control settings
pixel_pin = board.D18
num_pixels = 256
order = neopixel.GRB

# Individual pixel characteristics
pixels = neopixel.NeoPixel(
    pixel_pin, num_pixels, brightness=0.1, auto_write=False, pixel_order=order
)

# Define teeth resting state
tooth1 = [ 0, 1, 14, 15, 17, 16 ]
tooth2 = [ 32, 33, 34, 45, 46, 47, 50, 49, 48 ]
tooth3 = [ 64, 65, 66, 67, 76, 77, 78, 79, 83, 82, 81, 80 ]
tooth4 = [ 96, 97, 98, 99, 108, 109, 110, 111, 115, 114, 113, 112 ]
tooth5 = [ 128, 129, 130, 141, 142, 143, 146, 145, 144 ]
tooth6 = [ 160, 161, 174, 177, 175, 176 ]
teeth = [ tooth1, tooth2, tooth3, tooth4, tooth5, tooth6 ]

# Teeth limits (lo/hi pixels)
tlow1 = [ 1, 14, 17 ]
tlow2 = [ 34, 45, 50 ]
tlow3 = [ 67, 76, 83 ]
tlow4 = [ 99, 108, 115 ]
tlow5 = [ 130, 141, 146 ]
tlow6 = [ 161, 174, 177 ]
tlow = [ tlow1, tlow2, tlow3, tlow4, tlow5, tlow6 ]

thigh1 = [ 7, 8, 23 ]
thigh2 = [ 39, 40, 55 ]
thigh3 = [ 71, 72, 87 ]
thigh4 = [ 103, 104, 119 ]
thigh5 = [ 135, 136, 151 ]
thigh6 = [ 167, 168, 183 ]
thigh = [ thigh1, thigh2, thigh3, thigh4, thigh5, thigh6 ]

# Define RGB colours we are going to use
on = [ 0, 255, 0 ]
off = [ 0, 0, 0 ]
dim = [ 0, 80, 0 ]
red = [ 255, 0, 0 ]
yellow = [ 255, 255, 0 ]

# Reset pixels
for x in range(0,180):
  pixels[x] = off
pixels.show()

# Draw teeth with low brightness initially
for tooth in teeth:
  for x in tooth:
    pixels[x] = dim
pixels.show()

# Dance, Baby!
while True:

  for x in teeth:

    index = random.randint(0, 5)
    state = tlow[index]
    low = tlow[index]
    high = thigh[index]

    # Set brightness to full for active tooth
    for y in teeth[index]:
      pixels[y] = on
    pixels.show()

    # If bottom 2 rows, red. If 2 rows above, yellow.
    while state[0] < high[0]:

      # Descend Line
      if state[0] == high[0]-1 or state[0] == high[0]-2:
        pixels[state[0]+1] = red
        pixels[state[1]-1] = red
        pixels[state[2]+1] = red
      elif state[0] == high[0]-3 or state[0] == high[0]-4:
        pixels[state[0]+1] = yellow
        pixels[state[1]-1] = yellow
        pixels[state[2]+1] = yellow
      else:
        pixels[state[0]+1] = on
        pixels[state[1]-1] = on
        pixels[state[2]+1] = on
      pixels.show()
      state = [ state[0]+1, state[1]-1, state[2]+1 ]

      # Sleep for demo purposes
      time.sleep(0.2)

    # Bounce up again
    while state[0] > low[0]:
      # Remove lowest line until we hit tlow for tooth
      pixels[state[0]] = off
      pixels[state[1]] = off
      pixels[state[2]] = off
      pixels.show()
      time.sleep(0.2)
      state = [ state[0]-1, state[1]+1, state[2]-1 ]

    # Set active tooth back to dim at end of bounce
    for y in teeth[index]:
      pixels[y] = dim
    pixels.show()