Implemented threads and a forced framerate.

This commit is contained in:
CodePanter 2016-03-03 21:55:17 +01:00
parent b5d7f5c066
commit c5272c6439

View file

@ -11,10 +11,13 @@ __author__ = "Jan Klopper <jan@underdark.nl>"
import pygame import pygame
import socket import socket
import struct import struct
import Queue
import threading
import time
def main(): def main():
"""Runs a pixelvloed server""" """Runs a pixelvloed server"""
PixelVloed(debug=True) PixelVloed()
class PixelVloed(object): class PixelVloed(object):
"""PixelVloed server class""" """PixelVloed server class"""
@ -23,10 +26,15 @@ class PixelVloed(object):
"""Init the pixelVloed server""" """Init the pixelVloed server"""
self.debug = debug self.debug = debug
self.pixeloffset = 2 self.pixeloffset = 2
self.fps = 30
self.screen = None self.screen = None
self.canvas() self.canvas()
self.set_title() self.set_title()
self.Socket(udp_ip, udp_port) self.queue = Queue.Queue()
self.receivethread = threading.Thread(target=self.Socket, args=(udp_ip, udp_port), name="ReceiveThread")
self.drawthread = threading.Thread(target=self.CanvasUpdate, name="DrawThread")
self.receivethread.start()
self.drawthread.start()
@staticmethod @staticmethod
def set_title(text=None): def set_title(text=None):
@ -36,7 +44,7 @@ class PixelVloed(object):
title += ' ' + text title += ' ' + text
pygame.display.set_caption(title) pygame.display.set_caption(title)
def canvas(self, width=640, height=480): def canvas(self, width=768, height=1366):
"""Init the pygame canvas""" """Init the pygame canvas"""
pygame.init() pygame.init()
self.screen = pygame.display.set_mode((width, height)) self.screen = pygame.display.set_mode((width, height))
@ -56,7 +64,21 @@ class PixelVloed(object):
sock.bind((ipaddress, port)) sock.bind((ipaddress, port))
while True: while True:
data, addr = sock.recvfrom(1024) self.queue.put(sock.recvfrom(1024))
self.queue.join()
def CanvasUpdate(self):
lasttime = time.time()
while True:
currenttime = time.time()
self.Draw()
if currenttime - lasttime >= 1 / self.fps:
pygame.display.flip()
lasttime = time.time()
def Draw(self):
data, addr = self.queue.get()
self.queue.task_done()
preamble = struct.unpack_from("<?", data)[0] preamble = struct.unpack_from("<?", data)[0]
protocol = struct.unpack_from("<B", data, 1)[0] protocol = struct.unpack_from("<B", data, 1)[0]
pixellength = 7 #xx,yy,r,g,b pixellength = 7 #xx,yy,r,g,b
@ -75,7 +97,6 @@ class PixelVloed(object):
if self.debug: if self.debug:
print pixel print pixel
self.Pixel(*pixel) self.Pixel(*pixel)
pygame.display.flip()
if __name__ == '__main__': if __name__ == '__main__':
main() main()