max/pixelvloed
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1 from CodePanter/master

Browse source 
Implemented threads and a forced framerate.
This commit is contained in:
JanKlopper 2016-03-03 21:59:18 +01:00
commit 67a9085262
1 changed files with 43 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

65
vloed.py
View file

@ -11,10 +11,13 @@ __author__ = "Jan Klopper <jan@underdark.nl>"
import pygame
import socket
import struct
import Queue
import threading
import time
def main():
"""Runs a pixelvloed server"""
PixelVloed(debug=True)
PixelVloed()
class PixelVloed(object):
"""PixelVloed server class"""
@ -23,10 +26,15 @@ class PixelVloed(object):
"""Init the pixelVloed server"""
self.debug = debug
self.pixeloffset = 2
self.fps = 30
self.screen = None
self.canvas()
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
def set_title(text=None):
@ -36,7 +44,7 @@ class PixelVloed(object):
title += ' ' + text
pygame.display.set_caption(title)
def canvas(self, width=640, height=480):
def canvas(self, width=768, height=1366):
"""Init the pygame canvas"""
pygame.init()
self.screen = pygame.display.set_mode((width, height))
@ -56,26 +64,39 @@ class PixelVloed(object):
sock.bind((ipaddress, port))
while True:
data, addr = sock.recvfrom(1024)
preamble = struct.unpack_from("<?", data)[0]
protocol = struct.unpack_from("<B", data, 1)[0]
pixellength = 7 #xx,yy,r,g,b
if preamble:
if self.debug:
print 'rgba mode for %s:%d' % (addr[0], addr[1])
pixellength = 8 #xx,yy,r,g,b,a
pixelcount = (len(data)-1) / pixellength
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]
protocol = struct.unpack_from("<B", data, 1)[0]
pixellength = 7 #xx,yy,r,g,b
if preamble:
if self.debug:
print '%d pixels received, protocol V %d ' % (pixelcount, protocol)
for i in xrange(0, pixelcount):
pixel = struct.unpack_from(
("<2H4B" if preamble else "<2H3B"),
data,
self.pixeloffset + (i*pixellength))
if self.debug:
print pixel
self.Pixel(*pixel)
pygame.display.flip()
print 'rgba mode for %s:%d' % (addr[0], addr[1])
pixellength = 8 #xx,yy,r,g,b,a
pixelcount = (len(data)-1) / pixellength
if self.debug:
print '%d pixels received, protocol V %d ' % (pixelcount, protocol)
for i in xrange(0, pixelcount):
pixel = struct.unpack_from(
("<2H4B" if preamble else "<2H3B"),
data,
self.pixeloffset + (i*pixellength))
if self.debug:
print pixel
self.Pixel(*pixel)
if __name__ == '__main__':
main()