56 lines
1.6 KiB
Python
Executable file
56 lines
1.6 KiB
Python
Executable file
#!/usr/bin/python
|
|
"""This is a 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()
|