onemillioncheckboxes/live-viewer.py

133 lines
4.6 KiB
Python

from base64 import b64decode
import tkinter as tk
from PIL import Image, ImageTk
from socketio import SimpleClient, exceptions
from threading import Thread
from requests import get
from datetime import datetime
COLORS = [(0x33, 0x33, 0x66), (0x96, 0x96, 0xe0)]
COLORS_UNPACKED = COLORS[0] + ((0, 0, 0) * 254) + COLORS[1]
SCREENSHOTS_TMPL = "./screenshots/%Y%m%d_%H%M%S.png"
class App(tk.Tk):
def __init__(self, url: str = "https://onemillioncheckboxes.com") -> None:
self.url = url
super().__init__()
self.title("1M pixels")
self.sio = SimpleClient()
self._canvas = tk.Canvas(self)
self._canvas.config(width=1000, height=1000, borderwidth=0, highlightthickness=0)
self._canvas.pack()
self.config(width=1000, height=1000, borderwidth=0, highlightthickness=0)
self._last_update = 0
self._running = False
self._image = Image.new("RGB", (1000, 1000), COLORS[0])
self._canvas.bind("<Motion>", self._on_mouse_move)
self._canvas.bind("<Button>", self._on_mouse_click)
self.bind("<KeyPress>", self._on_key_down)
def _on_mouse_move(self, event: tk.Event):
self.title(f"1M pixels: {event.x}, {event.y}")
def _on_mouse_click(self, event: tk.Event):
x, y = event.x, event.y
self.sio.emit("toggle_bit", { "index": x + y * 1000 })
def _save_image(self):
ts = datetime.fromtimestamp(self._last_update / 1000)
path = ts.strftime(SCREENSHOTS_TMPL)
print("SAVED", path)
self._image.save(path)
self.after(1000, self._save_image)
def _on_key_down(self, event: tk.Event):
# <KeyPress event state=Control keysym=r keycode=27 char='\x12' x=538 y=556>
if event.keysym == "r":
print("FULL REFRESH")
with get(f"{self.url}/api/initial-state") as req:
data = req.json()
buffer = b64decode(data["full_state"].encode() + b"=")
img = Image.frombytes("1", (1000, 1000), buffer).convert("P")
img.putpalette(COLORS_UNPACKED)
self._image.paste(img.convert("RGB"))
self._last_update = data["timestamp"]
print("FULL REFRESH DONE")
def _reader(self):
while self._running:
try:
name, data = self.sio.receive()
except exceptions.TimeoutError:
print("Timeout")
continue
if name == "batched_bit_toggles":
bits_on, bits_off, timestamp = data
if timestamp < self._last_update:
print("SKIP partial updates: too old")
continue
self._last_update = timestamp
for ndx in bits_on:
y, x = divmod(ndx, 1000)
self._image.putpixel((x, y), COLORS[1])
for ndx in bits_off:
y, x = divmod(ndx, 1000)
self._image.putpixel((x, y), COLORS[0])
self._tk_image.paste(self._image)
print("partial update", len(bits_on), len(bits_off))
elif name == "full_state":
print("full update")
buffer = b64decode(data["full_state"].encode() + b"=")
img = Image.frombytes("1", (1000, 1000), buffer).convert("P")
img.putpalette(COLORS_UNPACKED)
self._image.paste(img.convert("RGB"))
self._tk_image.paste(self._image)
else:
print(name, data)
self.sio.disconnect()
print("_reader exited")
def _close(self):
self._running = False
print("waiting for reader to close")
self._reader_thr.join()
self.destroy()
def start(self):
self.sio.connect(f"{self.url}/socket.io")
with get(f"{self.url}/api/initial-state") as req:
data = req.json()
buffer = b64decode(data["full_state"].encode() + b"=")
self._last_update = data["timestamp"]
img = Image.frombytes("1", (1000, 1000), buffer).convert("P")
img.putpalette(COLORS_UNPACKED)
self._image.paste(img.convert("RGB"))
self._tk_image = ImageTk.PhotoImage(self._image)
self._tk_image_id = self._canvas.create_image(0, 0, anchor="nw", image=self._tk_image)
self._running = True
self._reader_thr = Thread(target=self._reader, name="Reader thread")
self._reader_thr.start()
self.protocol("WM_DELETE_WINDOW", self._close)
self.after(1000, self._save_image)
self.mainloop()
if __name__ == "__main__":
app = App()
app.start()