forked from pvx/litsimaja
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.5 KiB
79 lines
2.5 KiB
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
|
|
|