Add the initial implementation

nyan/__main__.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python3
+
+from collections import namedtuple
+import math
+import os.path as path
+import random
+import sys
+import time
+from PIL import Image # Requires the Pillow library
+
+DISP_WIDTH  = 150
+DISP_HEIGHT = 16
+
+
+tail_colors = [
+	(0xff, 0x00, 0x00),
+	(0xff, 0x99, 0x00),
+	(0xff, 0xff, 0x00),
+	(0x00, 0xff, 0x00),
+	(0x00, 0x99, 0xff),
+	(0x66, 0x33, 0xff),
+]
+
+def read_image(filename):
+	img = Image.open(filename)
+	return img
+
+anim_cat = []
+for i in range(0, 6):
+	frame = read_image('%s/cat/%d.png' % (path.dirname(__file__), i))
+	assert frame.size[0] == DISP_WIDTH
+	assert frame.size[1] == DISP_HEIGHT
+	anim_cat.append(frame)
+anim_sparkle = []
+for i in range(0, 5):
+	frame = read_image('%s/sparkle/%d.png' % (path.dirname(__file__), i))
+	anim_sparkle.append(frame)
+
+class Sparkle(object):
+	def __init__(self):
+		half_h = anim_sparkle[0].size[1] // 2
+		self.x = random.randint(0, DISP_WIDTH)
+		self.y = random.randint(half_h, DISP_HEIGHT - half_h)
+		self.frame_index = 0
+
+sparkles = []
+while True:
+	for anim_frame in anim_cat:
+		if len(sparkles) < 32:
+			sparkles.append(Sparkle())
+
+		frame = bytearray(DISP_WIDTH * DISP_HEIGHT * 3)
+
+		# Render the tail
+		for x in range(DISP_WIDTH):
+			for y in range(DISP_HEIGHT):
+				i = y * DISP_WIDTH + x
+				t = time.time()
+				col_y = y - (DISP_HEIGHT // 2 - len(tail_colors) // 2) + int(math.sin(x / 6 + t * math.pi) * 4 * math.sin(t * 8))
+				if x < 130 and 0 <= col_y < len(tail_colors):
+					color = tail_colors[col_y]
+				else:
+					color = (0x0f, 0x4d, 0x8f)
+				frame[i*3:i*3+3] = color
+
+		# Paste animated frame
+		for i in range(DISP_WIDTH * DISP_HEIGHT):
+			pix = anim_frame.getpixel((i % DISP_WIDTH, i / DISP_WIDTH))
+			if pix[3] != 0:
+				frame[i*3:i*3+3] = pix[:3]
+
+		# Render and update the sparkles
+		for sp in sparkles:
+			for x in range(anim_sparkle[0].size[0]):
+				for y in range(anim_sparkle[0].size[1]):
+					pix_x = sp.x + x - anim_sparkle[0].size[0] // 2
+					pix_y = sp.y + y - anim_sparkle[0].size[1] // 2
+					if 0 <= pix_x < DISP_WIDTH and 0 <= pix_y < DISP_HEIGHT:
+						i = pix_y * DISP_WIDTH + pix_x
+						pix = anim_sparkle[sp.frame_index].getpixel((x, y))
+						if pix[3] != 0:
+							frame[i*3:i*3+3] = pix[:3]
+			sp.x -= 3
+			sp.frame_index += 1
+		# Remove expired sparkles
+		sparkles = list(filter(lambda sp: sp.frame_index < len(anim_sparkle), sparkles))
+
+		sys.stdout.buffer.write(frame)
+		time.sleep(1 / 30)