This repository has been archived on 2024-01-26. You can view files and clone it, but cannot push or open issues or pull requests.
rgb_led_matrix/matrix-simulator/matrix-simulator.py
2015-03-09 22:15:08 +01:00

37 lines
1.1 KiB
Python

import sys
import socket
import pygame
import numpy as np
import binascii
pygame.init()
screen = pygame.display.set_mode((320, 320))
UDP_IP = "127.0.0.1"
UDP_PORT = 8888
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.bind((UDP_IP, UDP_PORT))
while True:
data, addr = sock.recvfrom(768)
# strip newline
print type(data)
data = data.strip()
print "received {} bytes from {}:{}".format(len(data), addr[0], addr[1])
print "received data:"
r = bytearray(data[2::3])
g = bytearray(data[1::3])
b = bytearray(data[0::3])
for j in range(16):
for i in range(16):
if (j*16 + i) < len(r) and (j*16 + i) < len(g) and (j*16 + i) < len(b):
pygame.draw.circle(screen, pygame.Color(r[16*j+i], g[16*j+i], b[16*j+i]), (i*20+10, 320-j*20-10), 5)
print "{:s} ".format(binascii.hexlify(bytearray((r[16*j+i], g[16*j+i], b[16*j+i])))),
else:
pygame.draw.circle(screen, pygame.Color(30, 30, 30), (i*20+10, 320-j*20-10), 5)
print ""
pygame.display.update()