forked from pvx/litsimaja
Compare commits
3 Commits
feature/vi
...
ba02a41c13
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ba02a41c13 | ||
|
|
dd0fd60379 | ||
|
|
84e2837f4b |
5
.gitignore
vendored
Normal file
5
.gitignore
vendored
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
.idea/
|
||||||
|
.git/
|
||||||
|
__pycache__/
|
||||||
|
*.pyc
|
||||||
|
*.swp
|
||||||
52
pyleds/lib/Color.py
Normal file
52
pyleds/lib/Color.py
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
Color = lambda r, g, b: (r << 16) | (g << 8) | b
|
||||||
|
|
||||||
|
|
||||||
|
def Color_to_list(color):
|
||||||
|
rgb = [(color >> 16) & 255, (color >> 8) & 255, color & 255]
|
||||||
|
w = (color >> 24) & 255
|
||||||
|
if w:
|
||||||
|
rgb.append(w)
|
||||||
|
return rgb
|
||||||
|
|
||||||
|
|
||||||
|
def Color_to_hex(color):
|
||||||
|
s = [(color >> 16) & 255, (color >> 8) & 255, color & 255]
|
||||||
|
return "#"+''.join(map(lambda y: ("0" + hex(int(y))[2:])[-2:], s))
|
||||||
|
|
||||||
|
|
||||||
|
def Color_to_rgb(color):
|
||||||
|
return (color >> 16) & 255, (color >> 8) & 255, color & 255
|
||||||
|
|
||||||
|
|
||||||
|
def hex_to_rgb(value):
|
||||||
|
value = value.lstrip('#') if value[0] == "#" else value
|
||||||
|
lv = len(value)
|
||||||
|
return [int(value[i:i + lv // 3], 16) for i in range(0, lv, lv // 3)]
|
||||||
|
|
||||||
|
|
||||||
|
def to_Color(value):
|
||||||
|
if type(value) == str:
|
||||||
|
r, g, b = hex_to_rgb(value)
|
||||||
|
return Color(r, g, b)
|
||||||
|
if type(value) == int:
|
||||||
|
if 0 <= value <= 16777215:
|
||||||
|
return value
|
||||||
|
else:
|
||||||
|
raise ValueError("RGB Int exceeded the lower and upper limits")
|
||||||
|
if type(value) == list or type(value) == tuple:
|
||||||
|
out = []
|
||||||
|
for n, code in enumerate(value):
|
||||||
|
if not str(code).isnumeric():
|
||||||
|
raise ValueError("RGB color must be numeric")
|
||||||
|
if n > 4:
|
||||||
|
raise ValueError("RGB contains more than 4 values")
|
||||||
|
_code = int(code)
|
||||||
|
if _code < 0:
|
||||||
|
print(f"RGB value under 0")
|
||||||
|
out.append(0)
|
||||||
|
elif _code > 255:
|
||||||
|
print(f"RGB value over 255")
|
||||||
|
out.append(255)
|
||||||
|
else:
|
||||||
|
out.append(_code)
|
||||||
|
return Color(out[0], out[1], out[2])
|
||||||
@@ -1,21 +1,21 @@
|
|||||||
from rpi_ws281x import PixelStrip
|
from rpi_ws281x import PixelStrip
|
||||||
from lib.FakeStrip import FakeStrip
|
# from lib.strip.WindowStrip import WindowStrip
|
||||||
|
# from lib.strip.TkinterStrip import TkinterStrip
|
||||||
from lib.LoopSwitch import LoopSwitch
|
from lib.LoopSwitch import LoopSwitch
|
||||||
|
|
||||||
|
|
||||||
class Litsimaja(object):
|
class Litsimaja(object):
|
||||||
_strip: PixelStrip
|
|
||||||
_loops: []
|
_loops: []
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._strip = FakeStrip(290, 18, 800000, 10, False, 255, 0, 4104)
|
self._strip = PixelStrip(290, 18, 800000, 10, False, 255, 0, 4104)
|
||||||
self._loops = []
|
self._loops = []
|
||||||
self._strip.begin()
|
self._strip.begin()
|
||||||
|
|
||||||
def count_pixels(self) -> int:
|
def count_pixels(self) -> int:
|
||||||
return self._strip.numPixels()
|
return self._strip.numPixels()
|
||||||
|
|
||||||
def get_strip(self) -> PixelStrip:
|
def get_strip(self):
|
||||||
return self._strip
|
return self._strip
|
||||||
|
|
||||||
def set_pixel_color(self, n: int, color: int) -> None:
|
def set_pixel_color(self, n: int, color: int) -> None:
|
||||||
|
|||||||
@@ -1,13 +1,15 @@
|
|||||||
from rpi_ws281x import PixelStrip
|
|
||||||
import atexit
|
import atexit
|
||||||
|
from lib.Color import Color
|
||||||
|
|
||||||
|
|
||||||
class _LedData(object):
|
class _LedData(object):
|
||||||
_led: []
|
|
||||||
|
|
||||||
def __init__(self, channel, size):
|
def __init__(self, channel, size):
|
||||||
self.size = size
|
self.size = size
|
||||||
self.channel = channel
|
self.channel = channel
|
||||||
|
self._led = {}
|
||||||
|
for i in range(size):
|
||||||
|
self._led[i] = 0
|
||||||
|
|
||||||
def __getitem__(self, pos):
|
def __getitem__(self, pos):
|
||||||
"""Return the 24-bit RGB color value at the provided position or slice
|
"""Return the 24-bit RGB color value at the provided position or slice
|
||||||
@@ -17,7 +19,6 @@ class _LedData(object):
|
|||||||
# and returning them in a list.
|
# and returning them in a list.
|
||||||
if isinstance(pos, slice):
|
if isinstance(pos, slice):
|
||||||
return [self._led[n] for n in range(*pos.indices(self.size))]
|
return [self._led[n] for n in range(*pos.indices(self.size))]
|
||||||
# Else assume the passed in value is a number to the position.
|
|
||||||
else:
|
else:
|
||||||
return self._led[pos]
|
return self._led[pos]
|
||||||
|
|
||||||
@@ -33,11 +34,11 @@ class _LedData(object):
|
|||||||
self._led[n] = value[index]
|
self._led[n] = value[index]
|
||||||
index += 1
|
index += 1
|
||||||
# Else assume the passed in value is a number to the position.
|
# Else assume the passed in value is a number to the position.
|
||||||
# else:
|
else:
|
||||||
self._led[pos] = value
|
self._led[pos] = value
|
||||||
|
|
||||||
|
|
||||||
class FakeStrip(PixelStrip):
|
class FakeStrip(object):
|
||||||
_brightness: int
|
_brightness: int
|
||||||
|
|
||||||
def __init__(self, num, pin, freq_hz=800000, dma=10, invert=False,
|
def __init__(self, num, pin, freq_hz=800000, dma=10, invert=False,
|
||||||
@@ -78,7 +79,7 @@ class FakeStrip(PixelStrip):
|
|||||||
|
|
||||||
def show(self):
|
def show(self):
|
||||||
"""Update the display with the data from the LED buffer."""
|
"""Update the display with the data from the LED buffer."""
|
||||||
return # render
|
return
|
||||||
|
|
||||||
def getBrightness(self) -> int:
|
def getBrightness(self) -> int:
|
||||||
return self._brightness
|
return self._brightness
|
||||||
@@ -92,3 +93,32 @@ class FakeStrip(PixelStrip):
|
|||||||
def numPixels(self):
|
def numPixels(self):
|
||||||
"""Return the number of pixels in the display."""
|
"""Return the number of pixels in the display."""
|
||||||
return self._led_data.size
|
return self._led_data.size
|
||||||
|
|
||||||
|
def setPixelColor(self, n, color):
|
||||||
|
"""Set LED at position n to the provided 24-bit color value (in RGB order).
|
||||||
|
"""
|
||||||
|
self._led_data[n] = color
|
||||||
|
|
||||||
|
def setPixelColorRGB(self, n, red, green, blue):
|
||||||
|
"""Set LED at position n to the provided red, green, and blue color.
|
||||||
|
Each color component should be a value from 0 to 255 (where 0 is the
|
||||||
|
lowest intensity and 255 is the highest intensity).
|
||||||
|
"""
|
||||||
|
self.setPixelColor(n, Color(red, green, blue))
|
||||||
|
|
||||||
|
def getPixels(self):
|
||||||
|
"""Return an object which allows access to the LED display data as if
|
||||||
|
it were a sequence of 24-bit RGB values.
|
||||||
|
"""
|
||||||
|
return self._led_data
|
||||||
|
|
||||||
|
def getPixelColor(self, n):
|
||||||
|
"""Get the 24-bit RGB color value for the LED at position n."""
|
||||||
|
return self._led_data[n]
|
||||||
|
|
||||||
|
def getPixelColorRGB(self, n):
|
||||||
|
c = lambda: None
|
||||||
|
setattr(c, 'r', self._led_data[n] >> 16 & 0xff)
|
||||||
|
setattr(c, 'g', self._led_data[n] >> 8 & 0xff)
|
||||||
|
setattr(c, 'b', self._led_data[n] & 0xff)
|
||||||
|
return c
|
||||||
73
pyleds/lib/strip/TkinterStrip.py
Normal file
73
pyleds/lib/strip/TkinterStrip.py
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
from threading import Thread
|
||||||
|
from tkinter import Tk, Label
|
||||||
|
from .FakeStrip import FakeStrip
|
||||||
|
from lib.Color import Color_to_list
|
||||||
|
from time import sleep
|
||||||
|
import numpy as np
|
||||||
|
from PIL import Image, ImageTk
|
||||||
|
|
||||||
|
|
||||||
|
class Visualizer(Thread):
|
||||||
|
|
||||||
|
def __init__(self, x, y, multi):
|
||||||
|
Thread.__init__(self)
|
||||||
|
self.x, self.y = x, y
|
||||||
|
self.multi = multi
|
||||||
|
self.root = None
|
||||||
|
self.panel = None
|
||||||
|
self.first_loop = True
|
||||||
|
self.start()
|
||||||
|
sleep(0.04)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.root = Tk()
|
||||||
|
self.root.title("RGB Visualizer")
|
||||||
|
self.panel = Label(self.root)
|
||||||
|
self.root.geometry(f'{int(self.x * self.multi)}x{int(self.y * self.multi)}')
|
||||||
|
self.panel.pack()
|
||||||
|
self.root.mainloop()
|
||||||
|
|
||||||
|
def print_image(self, frame):
|
||||||
|
self.panel.configure(image=frame)
|
||||||
|
self.panel.frame = frame
|
||||||
|
|
||||||
|
|
||||||
|
class TkinterStrip(FakeStrip):
|
||||||
|
|
||||||
|
def __init__(self, num, pin, freq_hz=800000, dma=10, invert=False,
|
||||||
|
brightness=255, channel=0, strip_type=None, gamma=None):
|
||||||
|
super().__init__(num, pin, freq_hz, dma, invert, brightness, channel, strip_type, gamma)
|
||||||
|
self.viz = Visualizer(95, 50, 6)
|
||||||
|
# self.ms = time() * 1000.0
|
||||||
|
self.show()
|
||||||
|
|
||||||
|
def show(self):
|
||||||
|
"""Update the display with the data from the LED buffer."""
|
||||||
|
bp = [49, 99, 194, 244, 290]
|
||||||
|
x = self.viz.x
|
||||||
|
y = self.viz.y
|
||||||
|
|
||||||
|
data = np.zeros((y, x, 3), dtype=np.uint8)
|
||||||
|
|
||||||
|
for s in range(self.numPixels()):
|
||||||
|
x1 = y1 = 0
|
||||||
|
if s < 49:
|
||||||
|
x1 = x / 2 + s - 1
|
||||||
|
elif s < 99:
|
||||||
|
x1 = x - 1
|
||||||
|
y1 = (s - 49)
|
||||||
|
elif s < 194:
|
||||||
|
x1 = x + (99 - s) - 1
|
||||||
|
y1 = y - 1
|
||||||
|
elif s < 244:
|
||||||
|
y1 = y + (194 - s) - 1
|
||||||
|
elif s < 290:
|
||||||
|
x1 = s - 244
|
||||||
|
x1 = int(x1)
|
||||||
|
y1 = int(y1)
|
||||||
|
data[y1, x1] = Color_to_list(self.getPixelColor(s))
|
||||||
|
img = Image.fromarray(data, 'RGB').resize((int(self.viz.x * self.viz.multi), int(self.viz.y * self.viz.multi)))
|
||||||
|
self.viz.print_image(ImageTk.PhotoImage(img))
|
||||||
|
# ms = time() * 1000.0
|
||||||
|
# print(ms - self.ms)
|
||||||
|
# self.ms = ms
|
||||||
79
pyleds/lib/strip/WindowStrip.py
Normal file
79
pyleds/lib/strip/WindowStrip.py
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
from threading import Thread
|
||||||
|
from tkinter import Tk, Canvas
|
||||||
|
from lib.strip.FakeStrip import FakeStrip
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
class Visualizer(Thread):
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
Thread.__init__(self)
|
||||||
|
self.x, self.y = 1000, 500
|
||||||
|
self.first_loop = True
|
||||||
|
self.start()
|
||||||
|
time.sleep(0.01)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
self.root = Tk()
|
||||||
|
self.canvas = Canvas(self.root, width=self.x, height=self.y)
|
||||||
|
self.root.geometry(f'{self.x}x{self.y}')
|
||||||
|
self.canvas.grid()
|
||||||
|
self.root.mainloop()
|
||||||
|
|
||||||
|
|
||||||
|
class WindowStrip(FakeStrip):
|
||||||
|
|
||||||
|
def __init__(self, num, pin, freq_hz=800000, dma=10, invert=False,
|
||||||
|
brightness=255, channel=0, strip_type=None, gamma=None):
|
||||||
|
super().__init__(num, pin, freq_hz, dma, invert, brightness, channel, strip_type, gamma)
|
||||||
|
self.viz = Visualizer()
|
||||||
|
self.show()
|
||||||
|
|
||||||
|
def show(self):
|
||||||
|
"""Update the display with the data from the LED buffer."""
|
||||||
|
size = 10.5
|
||||||
|
bp = [47, 97, 191, 242, 289]
|
||||||
|
x = self.viz.x
|
||||||
|
y = self.viz.y
|
||||||
|
first_loop = self.viz.first_loop
|
||||||
|
|
||||||
|
for s in range(self.numPixels()):
|
||||||
|
rgb_int = self.getPixelColor(s)
|
||||||
|
r = rgb_int & 255
|
||||||
|
g = (rgb_int >> 8) & 255
|
||||||
|
b = (rgb_int >> 16) & 255
|
||||||
|
|
||||||
|
if not first_loop:
|
||||||
|
self.viz.canvas.itemconfig(s + 1, fill="#%02x%02x%02x" % (r, g, b))
|
||||||
|
else:
|
||||||
|
x1 = 0
|
||||||
|
y1 = 0
|
||||||
|
x2 = 0
|
||||||
|
y2 = 0
|
||||||
|
if s < bp[0]:
|
||||||
|
x1 = (x / 2) + s * size
|
||||||
|
y1 = 0
|
||||||
|
x2 = x1 + size
|
||||||
|
y2 = size
|
||||||
|
elif s < bp[1]:
|
||||||
|
x1 = x - size
|
||||||
|
y1 = -(bp[0] - s) * size
|
||||||
|
x2 = x1 + size
|
||||||
|
y2 = y1 + size
|
||||||
|
elif s < bp[2]:
|
||||||
|
x1 = x + (bp[1] - s) * size
|
||||||
|
y1 = y
|
||||||
|
x2 = x1 - size
|
||||||
|
y2 = y1 - size
|
||||||
|
elif s < bp[3]:
|
||||||
|
x1 = 0
|
||||||
|
y1 = y + (bp[2] - s) * size
|
||||||
|
x2 = x1 + size
|
||||||
|
y2 = y1 - size
|
||||||
|
elif s < bp[4]:
|
||||||
|
x1 = size * - (bp[3] - s)
|
||||||
|
y1 = 0
|
||||||
|
x2 = x1 + size
|
||||||
|
y2 = y1 + size
|
||||||
|
self.viz.canvas.create_rectangle(x1, y1, x2, y2, fill="#%02x%02x%02x" % (r, g, b), outline="")
|
||||||
|
self.viz.first_loop = False
|
||||||
36
pyleds/program/peter/DiskoPidu.py
Normal file
36
pyleds/program/peter/DiskoPidu.py
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# Ported by Peter
|
||||||
|
# Palun!
|
||||||
|
|
||||||
|
from lib.Program import Program
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
|
||||||
|
|
||||||
|
def name():
|
||||||
|
return 'DiskoPidu'
|
||||||
|
|
||||||
|
|
||||||
|
class DiskoPidu(Program):
|
||||||
|
|
||||||
|
def disco(self, segmentLength, wait_ms):
|
||||||
|
color = random.randint(0, 0xffffff)
|
||||||
|
totalLength = self._lm.count_pixels()
|
||||||
|
for p in range(totalLength):
|
||||||
|
if p % segmentLength == 0:
|
||||||
|
color = random.randint(0, 0xffffff)
|
||||||
|
self._lm.set_pixel_color(p, color)
|
||||||
|
self._lm.show()
|
||||||
|
time.sleep(wait_ms / 1000.0)
|
||||||
|
|
||||||
|
# Main program logic follows:
|
||||||
|
def run(self, args=None):
|
||||||
|
loop = False
|
||||||
|
if 'loop' in args and args['loop']:
|
||||||
|
loop = args['loop']
|
||||||
|
|
||||||
|
while self.get_loop().status():
|
||||||
|
self.disco(10, 500)
|
||||||
|
|
||||||
|
if not loop:
|
||||||
|
break
|
||||||
@@ -1,6 +1,13 @@
|
|||||||
from lib.Program import Program
|
from lib.Program import Program
|
||||||
from lib.Litsimaja import Litsimaja
|
from lib.Litsimaja import Litsimaja
|
||||||
from rpi_ws281x import Color
|
|
||||||
|
|
||||||
|
def Color(red, green, blue):
|
||||||
|
"""Convert the provided red, green, blue color to a 24-bit color value.
|
||||||
|
Each color component should be a value 0-255 where 0 is the lowest intensity
|
||||||
|
and 255 is the highest intensity.
|
||||||
|
"""
|
||||||
|
return (red << 16) | (green << 8) | blue
|
||||||
|
|
||||||
|
|
||||||
def name():
|
def name():
|
||||||
|
|||||||
Reference in New Issue
Block a user