pixelvloed/client.py

#!/usr/bin/python
"""This is an udp / binary client

Inspired by the PixelFlut beamer on eth0:winter 2016 and
code from https://github.com/defnull/pixelflut/
"""

__version__ = 0.1
__author__ = "Jan Klopper <jan@underdark.nl>"

import socket
import struct
import random

def RGBPixel(x, y, r, g, b, a=None): # pylint: disable=C0103
  """Generates the packed data for a pixel"""
  if a:
    return struct.pack("<2H4B", x, y, r, g, b, a)
  return struct.pack("<2H3B", x, y, r, g, b)

def SetRGBAMode(mode):
  """Generate the rgb/rgba bit"""
  return struct.pack("<?", mode)

def SetVersionBit(protocol=1):
  """Generate the Version bit"""
  return struct.pack("<B", protocol)

def RandomFill(width=640, height=480):
  """Generates a random number of pixels with a random color"""
  message = SetRGBAMode(False)
  message += SetVersionBit()
  for pixel in xrange(0, random.randint(10, 100)):
    message += RGBPixel(random.randint(0, width),
                        random.randint(0, height),
                        random.randint(0, 255),
                        random.randint(0, 255),
                        random.randint(0, 255))
  return message

def SendPacket(ipaddress, port, message):
  """Sends the message to the udp server"""
  sock = socket.socket(socket.AF_INET, # Internet
                       socket.SOCK_DGRAM) # UDP
  sock.sendto(message, (ipaddress, port))

def main():
  """Sends some random pixels untill we break with ctrl+c"""
  ipaddress = "127.0.0.1"
  port = 5005

  while True:
    SendPacket(ipaddress, port, RandomFill())

if __name__ == '__main__':
  main()