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.
108 lines
2.9 KiB
108 lines
2.9 KiB
7 years ago
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import random
|
||
|
|
import math
|
||
|
|
import argparse
|
||
|
|
import collections
|
||
|
|
import blup.frame
|
||
|
|
import blup.output
|
||
|
|
import blup.animation
|
||
|
|
import blup.writebml
|
||
|
|
import colorsys
|
||
|
|
import functools
|
||
|
|
|
||
|
|
|
||
|
|
WIDTH = 22
|
||
|
|
HEIGHT = 16
|
||
|
|
DEPTH = 256
|
||
|
|
|
||
|
|
|
||
|
|
class Point(collections.namedtuple('Point', ['x', 'y'])):
|
||
|
|
def __add__(self, other):
|
||
|
|
return Point(self.x + other.x, self.y + other.y)
|
||
|
|
def __sub__(self, other):
|
||
|
|
return Point(self.x - other.x, self.y - other.y)
|
||
|
|
def __mul__(self, other):
|
||
|
|
if isinstance(other, Point):
|
||
|
|
raise Exception('not implemented')
|
||
|
|
else:
|
||
|
|
return Point(self.x * other, self.y * other)
|
||
|
|
def __truediv__(self, other):
|
||
|
|
if isinstance(other, Point):
|
||
|
|
raise ValueError()
|
||
|
|
else:
|
||
|
|
return self * (1/other)
|
||
|
|
@property
|
||
|
|
def len(self):
|
||
|
|
return math.sqrt(self.x**2 + self.y**2)
|
||
|
|
@property
|
||
|
|
def n(self):
|
||
|
|
n = Point(self.y, -self.x)
|
||
|
|
return n / n.len
|
||
|
|
@property
|
||
|
|
def intp(self):
|
||
|
|
return Point(int(round(self.x)), int(round(self.y)))
|
||
|
|
|
||
|
|
|
||
|
|
def convert_color(c):
|
||
|
|
return list(map(lambda x: int(round(x*(DEPTH-1))), c))
|
||
|
|
|
||
|
|
|
||
|
|
def get_random_color():
|
||
|
|
return colorsys.hsv_to_rgb(random.random(), 0.5 + random.random() / 2, 1)
|
||
|
|
|
||
|
|
|
||
|
|
class CurveFunction():
|
||
|
|
def __init__(self, p1, p2):
|
||
|
|
self.p1, self.p2 = p1, p2
|
||
|
|
d = p2 - p1
|
||
|
|
self.pp = p1 + d/2 + d.n * (d.len * 0.8)
|
||
|
|
|
||
|
|
def __call__(self, t):
|
||
|
|
ret = (self.p1 - self.pp*2 + self.p2) * t**2
|
||
|
|
ret += (self.p1 * -2 + self.pp * 2) * t
|
||
|
|
ret += self.p1
|
||
|
|
return ret
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser(description='Generate animations')
|
||
|
|
parser.add_argument('-d', '--delay', type=int, default=50)
|
||
|
|
parser.add_argument('-t', '--time', type=int, default=15)
|
||
|
|
parser.add_argument('output_file')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
dim = blup.frame.FrameDimension(WIDTH, HEIGHT, DEPTH, 3)
|
||
|
|
anim = blup.animation.Animation(dim)
|
||
|
|
anim.tags['description'] = ' '.join(sys.argv)
|
||
|
|
|
||
|
|
t = 0
|
||
|
|
dt = args.delay / 1000
|
||
|
|
while t < args.time:
|
||
|
|
t += dt
|
||
|
|
|
||
|
|
p1 = p2 = None
|
||
|
|
while p1 == p2 or (p1 - p2).len < max(WIDTH, HEIGHT) / 2:
|
||
|
|
p1 = Point(random.randint(0, WIDTH-1), random.randint(0, HEIGHT-1))
|
||
|
|
p2 = Point(random.randint(0, WIDTH-1), random.randint(0, HEIGHT-1))
|
||
|
|
cf = CurveFunction(p1, p2)
|
||
|
|
color = convert_color(get_random_color())
|
||
|
|
tt = 0
|
||
|
|
points = []
|
||
|
|
while tt <= 1:
|
||
|
|
t += dt
|
||
|
|
tt += 0.025
|
||
|
|
p = cf(tt).intp
|
||
|
|
points.append(p)
|
||
|
|
frame = blup.animation.AnimationFrame(dim, args.delay)
|
||
|
|
for p in points:
|
||
|
|
try:
|
||
|
|
frame.setPixel(p.x, p.y, color)
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
anim.addFrame(frame)
|
||
|
|
|
||
|
|
blup.writebml.writeBml(anim, args.output_file)
|