114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: UTF-8 -*-
|
|
import _thread
|
|
import socket
|
|
import pygame
|
|
import math
|
|
import random
|
|
import numpy as np
|
|
from pygame.locals import *
|
|
import time
|
|
|
|
WIDTH=160
|
|
HEIGHT=24*6
|
|
PIXELSIZE=4 #For Simulator Display
|
|
UDPPORT=2323
|
|
DEBUG=True
|
|
BITSPERPIXEL=2
|
|
|
|
|
|
|
|
|
|
class FlipdotSim():
|
|
def __init__(self,
|
|
imageSize = (160,48), #80,16
|
|
pixelSize = 10,
|
|
udpPort = 2323,
|
|
bitsperpixel=1):
|
|
self.imageSize=imageSize
|
|
self.udpPort = udpPort
|
|
self.bitsperpixel=bitsperpixel
|
|
self.flipdotMatrixSimulatorWidget = FlipdotMatrixSimulatorWidget(imageSize, pixelSize, bitsperpixel) #comment out if no simulator needed
|
|
self.udpHostSocket = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM)
|
|
self.udpHostSocket.bind(("", self.udpPort))
|
|
|
|
self.timesincelastpacket=time.time()
|
|
self.timelastcalculation=0
|
|
|
|
self.bitsneeded=self.imageSize[0]*self.imageSize[1]*self.bitsperpixel
|
|
|
|
|
|
def run(self):
|
|
self.RunServer()
|
|
|
|
|
|
def RunServer(self):
|
|
try:
|
|
while True:
|
|
rawData = self.udpHostSocket.recv(4096*2)
|
|
blafoo=time.time()
|
|
|
|
self.flipdotMatrixSimulatorWidget.showFromRawData(rawData) #send to simulator display
|
|
calctime=time.time()-blafoo
|
|
print(str(round(calctime,4))+"s, FPS="+str(round(1/calctime,2))) #calculate time it took for calculation and drawing
|
|
|
|
finally:
|
|
self.udpHostSocket.close()
|
|
|
|
class FlipdotMatrixSimulatorWidget():
|
|
BLACKCOLOR = 0
|
|
WHITECOLOR = 1
|
|
|
|
def __init__(self,
|
|
imageSize = (160,48),
|
|
pixelSize = 4,
|
|
bitsperpixel = 1):
|
|
self.imageSize = imageSize
|
|
self.pixelSize = pixelSize
|
|
|
|
pygame.init()
|
|
self.screen = pygame.display.set_mode((imageSize[0]*pixelSize, imageSize[1]*pixelSize))
|
|
|
|
self.screen.fill((0,0,0))
|
|
self.pixelsurface=None
|
|
_thread.start_new_thread(self.watchCloseThread, ())
|
|
|
|
self.bitsperpixel=bitsperpixel
|
|
|
|
minimumbrightness=32
|
|
self.colorTable=[]
|
|
for i in range(pow(2,self.bitsperpixel)):
|
|
brightness=i/pow(2,self.bitsperpixel)
|
|
self.colorTable.append( ( int( (255-minimumbrightness)*brightness+minimumbrightness) , int(127*brightness+ minimumbrightness/2), int(minimumbrightness/4) ) )
|
|
|
|
|
|
|
|
def watchCloseThread(self):
|
|
while True:
|
|
for event in pygame.event.get():
|
|
if event.type in (QUIT, QUIT):
|
|
import os
|
|
os.kill(os.getpid(), 9)
|
|
pygame.time.delay(500)
|
|
|
|
def showFromRawData(self, rawData):
|
|
x=0 #pixel x position
|
|
y=0 #pixel y position
|
|
bitshifts=[x*self.bitsperpixel for x in reversed(range(int(8/self.bitsperpixel)))]
|
|
bitmask=int('00000011', 2)
|
|
|
|
for cbyte in rawData:
|
|
for numbershifts in bitshifts:
|
|
pixelbyte = ( cbyte & (bitmask<<numbershifts) ) >> numbershifts
|
|
|
|
pygame.draw.circle(self.screen, self.colorTable[pixelbyte], (int(x*self.pixelSize+self.pixelSize/2), int(y*self.pixelSize+self.pixelSize/2)), int(self.pixelSize/2)) #Draw LED as filled circles
|
|
|
|
x+=1 #next pixel
|
|
y+=int(x/self.imageSize[0]) #next column if end of row
|
|
x%=self.imageSize[0] #start at first column if end of row
|
|
|
|
pygame.display.update()
|
|
|
|
if __name__ == '__main__':
|
|
FlipdotSim(imageSize=(WIDTH,HEIGHT), pixelSize = PIXELSIZE, udpPort=UDPPORT, bitsperpixel=BITSPERPIXEL).run()
|