hkc
/
onemillioncheckboxes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added negation support to `text` element

This commit is contained in:
Casey 2024-07-08 22:09:25 +03:00
parent 9979897bd0
commit 9c4c9774ca
Signed by: hkc
GPG Key ID: F0F6CFE11CDB0960
1 changed files with 92 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
async-bot.py
View File

@ -4,7 +4,14 @@ from socketio import AsyncClient, AsyncSimpleClient
import socketio import socketio
from aiohttp import ClientSession, ClientTimeout from aiohttp import ClientSession, ClientTimeout
from aiohttp_socks import ProxyConnector from aiohttp_socks import ProxyConnector
from PIL import Image, ImageFont, ImageDraw, ImageFilter, ImageSequence, ImageChops from PIL import (
Image,
ImageFont,
ImageDraw,
ImageFilter,
ImageSequence,
ImageChops,
)
from base64 import b64decode from base64 import b64decode
from random import choice, randint from random import choice, randint
from json import load from json import load
@ -19,6 +26,7 @@ Font = ImageFont.FreeTypeFont | ImageFont.ImageFont
TIMEOUT = ClientTimeout(120) TIMEOUT = ClientTimeout(120)
class AsyncBotManager: class AsyncBotManager:
def __init__(self, base: str = "https://onemillioncheckboxes.com"): def __init__(self, base: str = "https://onemillioncheckboxes.com"):
self.base = base self.base = base
@ -46,20 +54,31 @@ class AsyncBotManager:
self.ready_event = asyncio.Event() self.ready_event = asyncio.Event()
@staticmethod @staticmethod
def get_text_image(text: str, font: ImageFont.ImageFont | ImageFont.FreeTypeFont) -> Image.Image: def get_text_image(
text: str,
font: ImageFont.ImageFont | ImageFont.FreeTypeFont,
negative: bool = False,
) -> Image.Image:
left, top, right, bottom = font.getbbox(text) left, top, right, bottom = font.getbbox(text)
with Image.new("LA", (int(right - left) + 4, int(bottom - top) + 16), 0) as im: with Image.new(
"LA", (int(right - left) + 4, int(bottom - top) + 16), 0
) as im:
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.rectangle((0, 0, im.width, im.height), (0, 0)) draw.rectangle((0, 0, im.width, im.height), (0, 0))
draw.text((left + 2, top + 2), text, font=font, fill=(255, 0), draw.text(
anchor="lt") (left + 2, top + 2),
text,
font=font,
fill=(0 if negative else 255, 0),
anchor="lt",
)
alpha = im.convert("L").filter(ImageFilter.MaxFilter(5)) alpha = im.convert("L").filter(ImageFilter.MaxFilter(5))
im.putalpha(alpha) im.putalpha(alpha)
return im.copy() return im.copy()
def get_font(self, font_name: str, size: int = 8): def get_font(self, font_name: str, size: int = 8):
if (font := self.fonts.get((font_name, size))): if font := self.fonts.get((font_name, size)):
return font return font
if font_name == "default": if font_name == "default":
font = ImageFont.load_default(size) font = ImageFont.load_default(size)
@ -68,9 +87,20 @@ class AsyncBotManager:
self.fonts[font_name, size] = font self.fonts[font_name, size] = font
return font return font
def put_text(self, x: int, y: int, text: str, font: str, size: int = 8): def put_text(
self.put_image(x, y, self.get_text_image(text, self.get_font(font, size))) self,
x: int,
y: int,
text: str,
font: str,
size: int = 8,
negative: bool = False,
):
self.put_image(
x,
y,
self.get_text_image(text, self.get_font(font, size), negative),
)
def get_image_diff(self, ox: int, oy: int, im: Image.Image) -> PixelMap: def get_image_diff(self, ox: int, oy: int, im: Image.Image) -> PixelMap:
pixmap = PixelMap({}) pixmap = PixelMap({})
@ -154,7 +184,9 @@ class AsyncBotManager:
async with http.get(f"{self.base}/api/initial-state") as req: async with http.get(f"{self.base}/api/initial-state") as req:
data = await req.json() data = await req.json()
buffer = b64decode(data["full_state"].encode() + b"=") buffer = b64decode(data["full_state"].encode() + b"=")
self.canvas.paste(Image.frombytes("1", (1000, 1000), buffer)) self.canvas.paste(
Image.frombytes("1", (1000, 1000), buffer)
)
self._last_update = data["timestamp"] self._last_update = data["timestamp"]
print("Initial state received") print("Initial state received")
async with AsyncSimpleClient(http_session=http) as sio: async with AsyncSimpleClient(http_session=http) as sio:
@ -172,7 +204,9 @@ class AsyncBotManager:
self.reader_event.set() self.reader_event.set()
self.ready_event.set() self.ready_event.set()
if event == "full_state": if event == "full_state":
buffer = b64decode(data["full_state"].encode() + b"=") buffer = b64decode(
data["full_state"].encode() + b"="
)
image = Image.frombytes("1", (1000, 1000), buffer) image = Image.frombytes("1", (1000, 1000), buffer)
self.canvas.paste(image) self.canvas.paste(image)
image.close() image.close()
@ -183,7 +217,9 @@ class AsyncBotManager:
print("SKIPPING UPDATES: TOO OLD") print("SKIPPING UPDATES: TOO OLD")
else: else:
self._last_update = timestamp self._last_update = timestamp
self._read_boxes += len(bits_on) + len(bits_off) self._read_boxes += len(bits_on) + len(
bits_off
)
for ndx in bits_on: for ndx in bits_on:
y, x = divmod(ndx, 1000) y, x = divmod(ndx, 1000)
self.canvas.putpixel((x, y), 255) self.canvas.putpixel((x, y), 255)
@ -194,10 +230,16 @@ class AsyncBotManager:
print("unknown event", event, data) print("unknown event", event, data)
now = time_now() now = time_now()
if (now - self._last_printout) > 5: if (now - self._last_printout) > 5:
outgoing = self._written_boxes / (now - self._last_printout) outgoing = self._written_boxes / (
incoming = self._read_boxes / (now - self._last_printout) now - self._last_printout
)
incoming = self._read_boxes / (
now - self._last_printout
)
print() print()
print(f"I/O: {incoming:7.2f}/s | {outgoing:7.2f}/s") print(
f"I/O: {incoming:7.2f}/s | {outgoing:7.2f}/s"
)
print(f"Alive workers: {len(self._active)}") print(f"Alive workers: {len(self._active)}")
if len(self._active) < 2 and self._n_printouts > 3: if len(self._active) < 2 and self._n_printouts > 3:
@ -216,7 +258,9 @@ class AsyncBotManager:
else: else:
n_correct += 1 n_correct += 1
print(f"Invalid: {n_wrong:4d} Valid: {n_correct:4d}") print(
f"Invalid: {n_wrong:4d} Valid: {n_correct:4d}"
)
self._written_boxes = 0 self._written_boxes = 0
self._read_boxes = 0 self._read_boxes = 0
self._active.clear() self._active.clear()
@ -307,12 +351,27 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
for elem in settings["elements"]: for elem in settings["elements"]:
if elem["type"] == "text": if elem["type"] == "text":
mgr.put_text(elem["x"], elem["y"], elem["text"], elem.get("font", "default"), elem.get("size", 8)) mgr.put_text(
elem["x"],
elem["y"],
elem["text"],
elem.get("font", "default"),
elem.get("size", 8),
elem.get("negative", False),
)
print("ADD text", elem) print("ADD text", elem)
elif elem["type"] == "text_anim": elif elem["type"] == "text_anim":
frames: list[Image.Image] = [] frames: list[Image.Image] = []
for text in elem["lines"]: for text in elem["lines"]:
frames.append(mgr.get_text_image(text, mgr.get_font(elem.get("font", "default"), elem.get("size", 8)))) frames.append(
mgr.get_text_image(
text,
mgr.get_font(
elem.get("font", "default"),
elem.get("size", 8),
),
)
)
mgr.add_animation(elem["x"], elem["y"], frames, elem["spf"]) mgr.add_animation(elem["x"], elem["y"], frames, elem["spf"])
for frame in frames: for frame in frames:
frame.close() frame.close()
@ -325,7 +384,10 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
elif elem["type"] == "time": elif elem["type"] == "time":
time_format = elem["format"] time_format = elem["format"]
pos_x, pos_y = elem["x"], elem["y"] pos_x, pos_y = elem["x"], elem["y"]
font = mgr.get_font(elem.get("font", "default"), elem.get("size", 8)) font = mgr.get_font(
elem.get("font", "default"), elem.get("size", 8)
)
def update() -> PixelMap: def update() -> PixelMap:
now = datetime.datetime.now(datetime.timezone.utc) now = datetime.datetime.now(datetime.timezone.utc)
txt = now.strftime(time_format) txt = now.strftime(time_format)
@ -333,6 +395,7 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
pixmap = mgr.get_image_diff(pos_x, pos_y, img) pixmap = mgr.get_image_diff(pos_x, pos_y, img)
img.close() img.close()
return pixmap return pixmap
mgr.animation_functions.append(update) mgr.animation_functions.append(update)
elif elem["type"] == "tile": elif elem["type"] == "tile":
with Image.open(elem["path"]).convert("LA") as im: with Image.open(elem["path"]).convert("LA") as im:
@ -349,7 +412,9 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
frames: list[Image.Image] = [] frames: list[Image.Image] = []
for frame in ImageSequence.Iterator(anim): for frame in ImageSequence.Iterator(anim):
frames.append(frame.copy()) frames.append(frame.copy())
mgr.add_animation(elem["x"], elem["y"], frames, elem["spf"]) mgr.add_animation(
elem["x"], elem["y"], frames, elem["spf"]
)
for frame in frames: for frame in frames:
frame.close() frame.close()
print("ADD animation", elem) print("ADD animation", elem)
@ -380,7 +445,7 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
# pack rgb888 to rgb565 # pack rgb888 to rgb565
rgb: int = (r >> 3) << 11 rgb: int = (r >> 3) << 11
rgb |= (g >> 2) << 5 rgb |= (g >> 2) << 5
rgb |= (b >> 3) rgb |= b >> 3
# write to a bitstream # write to a bitstream
for i in range(16): for i in range(16):
@ -396,7 +461,7 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
offset = elem["offset"] offset = elem["offset"]
length = elem.get("length", 1000000) length = elem.get("length", 1000000)
written = 0 written = 0
while (char := fp.read(1)): while char := fp.read(1):
byte = char[0] byte = char[0]
if written > length: if written > length:
break break
@ -424,7 +489,10 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
) )
else: else:
res = await asyncio.gather( res = await asyncio.gather(
*[mgr.writer(i, None, settings["delay"]) for i in range(settings["n_bots"])], *[
mgr.writer(i, None, settings["delay"])
for i in range(settings["n_bots"])
],
) )
for ret in res: for ret in res:
@ -436,4 +504,5 @@ async def amain(conf_path: str = "settings.json", *_) -> None:
if __name__ == "__main__": if __name__ == "__main__":
from sys import argv from sys import argv
asyncio.run(amain(*argv[1:]), debug=True) asyncio.run(amain(*argv[1:]), debug=True)