Skip to content

Commit

Permalink
port animations to main class
Browse files Browse the repository at this point in the history
  • Loading branch information
@petabite
petabite committed Jul 1, 2019
1 parent 328875a commit 05e7e78
Show file tree
Hide file tree
Showing 2 changed files with 157 additions and 161 deletions.
157 changes: 3 additions & 154 deletions uPixels/main.py
View file
Original file line number Diff line number Diff line change
@@ -1,155 +1,4 @@
import machine, neopixel, time, urandom
from uPixels import uPixels

pin = machine.Pin(4, machine.Pin.OUT)
np = neopixel.NeoPixel(pin, 30)

def randInt(lower, upper):
randomNum = urandom.getrandbits(len(bin(upper)[2:])) + lower
if randomNum < upper:
return randomNum
else:
return upper - 1

def randColor():
return (randInt(0,256),randInt(0,256),randInt(0,256))

def clear():
for i in range(np.n):
np[i] = (0,0,0)
np.write()

def chase(ms=20, color=randColor(), direction='right'):
if direction == 'right':
led_iter = range(np.n)
else:
led_iter = range(np.n-1, -1, -1)
for i in led_iter:
np[i] = color
np.write()
time.sleep_ms(ms)
np[i] = (0,0,0)

def fillStrip(ms=25, color=randColor()):
count = np.n
while count > 0:
for i in range(count):
np[i] = color
np.write()
time.sleep_ms(ms)
if i != count-1:
np[i] = (0,0,0)
count -= 1

def fillFromMiddle(ms=40, color=randColor()):
midpoint = int(np.n / 2)
counter = 0
while counter != midpoint:
np[midpoint + counter] = color
if np.n % 2 == 0:
np[midpoint - 1 - counter] = color
else:
np[midpoint - counter] = color
np.write()
time.sleep_ms(ms)
counter += 1

def fillFromSides(ms=40, color=randColor()):
midpoint = int(np.n / 2)
counter = 0
while counter != midpoint:
np[0 + counter] = color
np[np.n - 1 - counter] = color
np.write()
time.sleep_ms(ms)
counter += 1

def randomFill(ms=150, color=True):
random_positions = []
while len(random_positions) < np.n:
random_pos = randInt(0, np.n)
if random_pos not in random_positions:
random_positions.append(random_pos)
for position in random_positions:
if color == True:
np[position] = randColor()
else:
np[position] = color
np.write()
time.sleep_ms(ms)

def setSegment(segment_of_leds, color):
for led in segment_of_leds:
np[led] = color
np.write()

def altColors(ms=125, firstColor=randColor(), secondColor=randColor()):
while True:
for i in range(np.n):
if i % 2 == 0:
np[i] = firstColor
else:
np[i] = secondColor
np.write()
time.sleep_ms(ms)
for i in range(np.n):
if i % 2 == 0:
np[i] = secondColor
else:
np[i] = firstColor
np.write()
time.sleep_ms(ms)

def bounce(ms=20, color=False):
while True:
if color == False:
chase(ms, randColor(), 'right')
chase(ms, randColor(), 'left')
else:
chase(ms, color, 'right')
chase(ms, color, 'left')

def rgbFade(ms=20):
for channel in range(3):
for v in range(256):
if channel == 0:
setSegment(list(range(10)), (v,0,0))
if channel == 1:
setSegment(list(range(10)), (0,v,0))
if channel == 2:
setSegment(list(range(10)), (0,0,v))
time.sleep_ms(ms)
for v in range(255,-1,-1):
if channel == 0:
setSegment(list(range(10)), (v,0,0))
if channel == 1:
setSegment(list(range(10)), (0,v,0))
if channel == 2:
setSegment(list(range(10)), (0,0,v))
time.sleep_ms(ms)

def wheel(pos):
if pos < 85:
return (pos * 3, 255 - pos * 3, 0)
elif pos < 170:
pos -= 85
return (255 - pos * 3, 0, pos * 3)
else:
pos -= 170
return (0, pos * 3, 255 - pos * 3)

def rainbow(ms=20, iterations = 2):
for j in range(256*iterations):
for i in range(np.n):
np[i] = wheel(((i * 256 // np.n) + j) & 255)
np.write()
time.sleep_ms(ms)

def rainbowChase(ms=50):
for j in range(256):
for q in range(3):
for i in range(0, np.n, 3):
np[i+q] = wheel((i+j) % 255)
np.write()
time.sleep_ms(ms)
for i in range(0, np.n, 3):
np[i+q] = (0, 0, 0)
pixels = uPixels(4, 30)
pixels.startServer()
161 changes: 154 additions & 7 deletions uPixels/uPixels.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,22 +3,169 @@

class uPixels:
def __init__(self, pin, num_leds, address="0.0.0.0", port=8000):
self.pin = pin
self.num_leds = num_leds
self.pin = machine.Pin(pin, machine.Pin.OUT) # configure pin for leds
self.np = neopixel.NeoPixel(self.pin, num_leds) # configure neopixel library
self.address = address
self.port = port
self.server = uWeb(address, port)
server.routes({
(uWeb.GET, "/"): self.app
})

# web server methods
def startServer(self):
self.server = uWeb(self.address, self.port)
self.server.routes({
(uWeb.GET, "/"): self.app
})
self.server.start()

def app(self):
server.render('uPixels.html')
self.server.render('uPixels.html')

# animation methods
def chase(self, ms=20, color=randColor(), direction='right'):
if direction == 'right':
led_iter = range(self.np.n)
else:
led_iter = range(self.np.n - 1, -1, -1)
for i in led_iter:
self.np[i] = color
self.np.write()
time.sleep_ms(ms)
self.np[i] = (0,0,0)

def fillStrip(self, ms=25, color=randColor()):
count = self.np.n
while count > 0:
for i in range(count):
self.np[i] = color
self.np.write()
time.sleep_ms(ms)
if i != count-1:
self.np[i] = (0,0,0)
count -= 1

def fillFromMiddle(self, ms=40, color=randColor()):
midpoint = int(self.np.n / 2)
counter = 0
while counter != midpoint:
self.np[midpoint + counter] = color
if self.np.n % 2 == 0:
self.np[midpoint - 1 - counter] = color
else:
self.np[midpoint - counter] = color
self.np.write()
time.sleep_ms(ms)
counter += 1

def fillFromSides(self, ms=40, color=randColor()):
midpoint = int(self.np.n / 2)
counter = 0
while counter != midpoint:
self.np[0 + counter] = color
self.np[self.np.n - 1 - counter] = color
self.np.write()
time.sleep_ms(ms)
counter += 1

def randomFill(self, ms=150, color=True):
random_positions = []
while len(random_positions) < self.np.n:
random_pos = randInt(0, self.np.n)
if random_pos not in random_positions:
random_positions.append(random_pos)
for position in random_positions:
if color == True:
self.np[position] = randColor()
else:
self.np[position] = color
self.np.write()
time.sleep_ms(ms)

def altColors(self, ms=125, firstColor=randColor(), secondColor=randColor()):
while True:
for i in range(self.np.n):
if i % 2 == 0:
self.np[i] = firstColor
else:
self.np[i] = secondColor
self.np.write()
time.sleep_ms(ms)
for i in range(self.np.n):
if i % 2 == 0:
self.np[i] = secondColor
else:
self.np[i] = firstColor
self.np.write()
time.sleep_ms(ms)

def bounce(self, ms=20, color=False):
while True:
if color == False:
self.chase(ms, randColor(), 'right')
self.chase(ms, randColor(), 'left')
else:
self.chase(ms, color, 'right')
self.chase(ms, color, 'left')

def rgbFade(self, ms=20):
for channel in range(3):
for v in range(256):
if channel == 0:
self.setSegment(list(range(10)), (v,0,0))
if channel == 1:
self.setSegment(list(range(10)), (0,v,0))
if channel == 2:
self.setSegment(list(range(10)), (0,0,v))
time.sleep_ms(ms)
for v in range(255,-1,-1):
if channel == 0:
self.setSegment(list(range(10)), (v,0,0))
if channel == 1:
self.setSegment(list(range(10)), (0,v,0))
if channel == 2:
self.setSegment(list(range(10)), (0,0,v))
time.sleep_ms(ms)

def rainbow(self, ms=20, iterations = 2):
for j in range(256*iterations):
for i in range(self.np.n):
self.np[i] = wheel(((i * 256 // self.np.n) + j) & 255)
self.np.write()
time.sleep_ms(ms)

def rainbowChase(self, ms=50):
for j in range(256):
for q in range(3):
for i in range(0, self.np.n, 3):
self.np[i+q] = wheel((i+j) % 255)
self.np.write()
time.sleep_ms(ms)
for i in range(0, self.np.n, 3):
self.np[i+q] = (0, 0, 0)

def clear(self):
self.setSegment(list(range(self.np.n)), (0,0,0))

# helper methods
def setSegment(self, segment_of_leds, color):
for led in segment_of_leds:
self.np[led] = color
self.np.write()

def randInt(self, lower, upper):
randomNum = urandom.getrandbits(len(bin(upper)[2:])) + lower
if randomNum < upper:
return randomNum
else:
return upper - 1

def randColor(self):
return (randInt(0,256),randInt(0,256),randInt(0,256))

def wheel(self, pos):
if pos < 85:
return (pos * 3, 255 - pos * 3, 0)
elif pos < 170:
pos -= 85
return (255 - pos * 3, 0, pos * 3)
else:
pos -= 170
return (0, pos * 3, 255 - pos * 3)

0 comments on commit 05e7e78

Please sign in to comment.