Pixelflut/pixelflut/brain.py
2014-05-07 15:56:04 +02:00

143 lines
3.2 KiB
Python

import pixelflut
import os
import time
pixelcount = 0
def guess_IP():
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
s.connect(("google.com", 80))
return s.getsockname()[0]
finally:
s.close()
IP = guess_IP()
port = 1234
text = 'P1XELFLUT! v%s (%d)\n' % (
pixelflut.__version__,
os.stat(__file__).st_mtime)
text += '$ echo "HELP" | netcat %s %d\n' % (IP, port)
text += 'https://github.com/defnull/pixelflut\n'
help = 'Commands:\n'
help += '>>> HELP\n'
help += '>>> SIZE\n'
help += '>>> QUIT\n'
help += '>>> TEXT x y text (currently disabled)\n'
help += '>>> PX x y [RRGGBB[AA]]\n'
@on('LOAD')
def callback(c):
c.load_font('./font.png')
c.set_title('@ %s:%d' % (IP, c.port))
@on('UNLOAD')
def callback(canvas):
canvas.text(5, 5, 'Reloading ...')
@on('RESIZE')
def on_resize(c):
c.text(5, 5, 'Screen Size: %dx%d' % c.get_size())
@on('CONNECT')
def on_connect(c, client):
pass
@on('KEYDOWN-c')
def on_key_c(c):
c.clear()
pixelflut.async(c.text, 5, 5, text, delay=0.1)
@on('KEYDOWN-s')
def on_key_s(c):
import os
i, mask = 0, 'screen%05d.png'
while os.path.exists(mask%i): i += 1
c.save_as(mask%i)
c.text(5,5, 'Saved as %s' % mask % i)
@on('KEYDOWN-k')
def on_key_k(c):
for ip, client in c.clients.iteritems():
client.disconnect()
@on('COMMAND-HELP')
def on_help(canvas, client):
client.send(help)
@on('COMMAND-TEXT')
def on_text(canvas, client, x, y, *words):
x, y = int(x), int(y)
text = ' '.join(words)[:200]
canvas.text(x, y, text, delay=1)
@on('COMMAND-SIZE')
def on_size(canvas, client):
client.send('SIZE %d %d' % canvas.get_size())
@on('COMMAND-QUIT')
def on_quit(canvas, client):
client.disconnect()
@on('COMMAND-GODMODE')
def on_quit(canvas, client, mode):
if mode == 'on':
client.pps = 100000
else:
client.pps = 1000
@on('COMMAND-PX')
def on_px(canvas, client, x, y, color=None):
global pixelcount
pixelcount += 1
x, y = int(x), int(y)
if color:
c = int(color, 16)
if len(color) == 6:
r = (c & 0xff0000) >> 16
g = (c & 0x00ff00) >> 8
b = c & 0x0000ff
a = 0xff
elif len(color) == 8:
r = (c & 0xff000000) >> 24
g = (c & 0x00ff0000) >> 16
b = (c & 0x0000ff00) >> 8
a = c & 0x000000ff
else:
return
canvas.set_pixel(x, y, r, g, b, a)
else:
r,g,b,a = canvas.get_pixel(x,y)
client.send('PX %d %d %02x%02x%02x%02x' % (x,y,r,g,b,a))
last_save = 0
status_text = ''
import math
@on('TICK')
def on_tick(canvas):
global last_save, pixelcount, status_text
canvas.text(5, 5, status_text, delay=0)
ccount = len([c for c in canvas.clients.values() if c.socket])
if ccount > 300:
for ip, client in canvas.clients.iteritems():
client.disconnect()
if time.time() > last_save:
status_text = text
status_text += 'px/s: %d \n' % (pixelcount / 5)
status_text += 'Connections: %d' % ccount
pixelcount = 0
last_save = time.time() + 5
canvas.save_as('save/mov_%d.png' % last_save)