forked from Blinkenbunt/blup
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
158 lines
3.8 KiB
158 lines
3.8 KiB
#!/usr/bin/python |
|
|
|
# inspired by: |
|
# http://lodev.org/cgtutor/plasma.html |
|
|
|
import sys |
|
import time |
|
import math |
|
#import pygame |
|
import colorsys |
|
import blup.frame |
|
import blup.output |
|
import blup.animation |
|
import blup.writebml |
|
|
|
|
|
class Plasma(object): |
|
def __init__(self, width, height): |
|
self.width = width |
|
self.height = height |
|
|
|
self.pixelvalues = [[0] * height for i in range(width)] |
|
self.pixelsReady = False |
|
self.offset = 0 |
|
self.update() |
|
|
|
@property |
|
def minValue(self): |
|
return min(list(map(min, self.pixelvalues))) |
|
|
|
@property |
|
def maxValue(self): |
|
return max(list(map(max, self.pixelvalues))) |
|
|
|
def applyPalette(self, palette): |
|
min = self.minValue |
|
max = self.maxValue |
|
norm = lambda x: ((x - min) / max) |
|
|
|
pixeldata = [ [None] * self.height for i in range(self.width) ] |
|
|
|
for x in range(self.width): |
|
rowvalues = self.pixelvalues[x] |
|
for y in range(self.height): |
|
pixeldata[x][y] = palette.getColorValue(norm(rowvalues[y])) |
|
|
|
return pixeldata |
|
|
|
def update(self): |
|
#if self.pixelsReady: |
|
# return |
|
self.offset += 1 |
|
|
|
for x in range(self.width): |
|
for y in range(self.height): |
|
xx = x+self.offset |
|
yy = y+self.offset |
|
p = 128 + 128 * math.sin(xx / 9.0) |
|
p += 128 + 128 * math.sin(yy / 16.0) |
|
p += 128 + 128 * math.sin((yy + xx) / 16.0) |
|
p += 128 + 128 * math.sin(math.sqrt(yy*yy + xx*xx) / 16.0) |
|
p += 128 + 8 * math.sin(math.sqrt(yy* xx) / 16.0) |
|
self.pixelvalues[x][y] = p |
|
self.pixelsReady = True |
|
|
|
class Palette(object): |
|
def __init__(self): |
|
self.offset = 0 |
|
self.cache = {} |
|
|
|
def getColorValue(self, x): |
|
x = round(x, 3) |
|
if x in self.cache: |
|
return self.cache[x] |
|
c = colorsys.hsv_to_rgb((x + self.offset) % 1, 1, 1) |
|
c = list(map(lambda x: int(x * 255), c)) |
|
self.cache[x] = c |
|
return c |
|
|
|
def update(self): |
|
self.cache = {} |
|
self.offset += 0.01 |
|
if self.offset > 1: |
|
self.offset -= 1 |
|
|
|
############################################## |
|
|
|
w = 22 |
|
h = 16 |
|
|
|
#w = 80 |
|
#h = 18 |
|
|
|
scalex = 30 |
|
scaley = 60 |
|
|
|
plasma = Plasma(w,h) |
|
palette= Palette() |
|
|
|
#screen = pygame.display.set_mode((w*scalex, h*scaley)) |
|
#pygame.display.update() |
|
|
|
#dim = blup.frame.FrameDimension(w, h, 8, 3) |
|
dim = blup.frame.FrameDimension(w, h, 256, 3) |
|
#out = blup.output.getOutput('serialblup:/dev/ttyUSB0:115200') |
|
#out = blup.output.getOutput('colorfulshell') |
|
#out = blup.output.getOutput('e3blp:bastel0:4242') |
|
|
|
if len(sys.argv) > 1: |
|
outfile = sys.argv[1] |
|
anim = blup.animation.Animation(dim) |
|
else: |
|
out = blup.output.getOutput('e3blp') |
|
anim = None |
|
|
|
num_frames = 200 |
|
delay = 50 |
|
|
|
i = 0 |
|
while i < num_frames: |
|
#screen.fill((0,0,0)) |
|
|
|
#print 'updating palette...' |
|
palette.update() |
|
#print 'updating plasma...' |
|
plasma.update() |
|
#print 'applying palette...' |
|
pixeldata = plasma.applyPalette(palette) |
|
|
|
#print 'drawing...' |
|
if anim is not None: |
|
frame = blup.animation.AnimationFrame(dim, delay) |
|
else: |
|
frame = blup.frame.Frame(dim) |
|
|
|
for x in range(w): |
|
#rowdata = pixeldata[x] |
|
for y in range(h): |
|
#pygame.draw.rect(screen, pixeldata[x][y], (x*scalex, y*scaley, scalex, scaley), 0) |
|
#pygame.draw.circle(screen, rowdata[y], (x, y), 1, 1) |
|
frame.setPixel(x, y, pixeldata[x][y]) |
|
#print frame.channels, frame.depth |
|
#print 'done drawing' |
|
|
|
if anim is not None: |
|
anim.addFrame(frame) |
|
i += 1 |
|
else: |
|
out.sendFrame(frame) |
|
time.sleep(delay / 1000) |
|
|
|
#pygame.display.update() |
|
#print 'done updating' |
|
#time.sleep(0.05) |
|
|
|
if anim is not None: |
|
writebml.writeBml(anim, outfile) |
|
|
|
|