|
4
|
1 import asyncio
|
|
|
2 import struct
|
|
|
3
|
|
5
|
4 import aiomqtt
|
|
|
5 import pygame
|
|
4
|
6
|
|
5
|
7 screen = pygame.display.set_mode((320, 320))
|
|
4
|
8 clock = pygame.time.Clock()
|
|
|
9
|
|
|
10
|
|
5
|
11 async def on_message(client):
|
|
|
12 await client.subscribe('display/squib/updates')
|
|
|
13 async for message in client.messages:
|
|
|
14 payload = bytes(message.payload)
|
|
|
15 head, x, y, w, h = struct.unpack("HHHHH", payload[:10])
|
|
|
16 buf = payload[10:]
|
|
|
17 pygame.draw.rect(screen, (255, 255, 0), (x, y, w, h))
|
|
|
18 pygame.display.flip()
|
|
|
19 await asyncio.sleep(.02)
|
|
|
20 for dy in range(h):
|
|
|
21 for dx in range(w):
|
|
|
22 off = w * dy + dx
|
|
|
23 r, g, b = buf[off * 3:off * 3 + 3]
|
|
|
24 screen.set_at((x + dx, y + dy), (r, g, b))
|
|
|
25 pygame.display.flip()
|
|
4
|
26
|
|
|
27
|
|
|
28 async def main():
|
|
|
29 async with aiomqtt.Client("mqtt2.bigasterisk.com") as client:
|
|
5
|
30 asyncio.create_task(on_message(client))
|
|
4
|
31
|
|
|
32 while True:
|
|
|
33 for event in pygame.event.get():
|
|
5
|
34 pass
|
|
|
35 await asyncio.sleep(1 / 30)
|
|
4
|
36
|
|
|
37
|
|
|
38 asyncio.run(main())
|
