Add TKinter Visualizer Emulator(Slow but works)

pyleds/lib/strip/WindowStrip.py

@@ -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