1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
"""Rendering classes and helper functions."""
from __future__ import division
import color
import random
import constants
from constants import *
import math
from show import Show
class ShootingStar(object):
"""Object displayed on the bridge, probably a fish."""
def __init__(self):
"""Create fish with default parameters."""
self.x = random.random() * BRIDGE_WIDTH
self.y = random.random() * 1.5
self.width = 5
self.height = 0.5
self.color = color.RGB(1, 1, 1)
if random.random() >= 0.5:
self.speed = 8
else:
self.speed = -8
def update(self):
self.x += self.speed
self.speed *= 0.9
return abs(self.speed) > 1 and self.x < BRIDGE_WIDTH
class Light(object):
"""Represent a bridge light and its state."""
def __init__(self, x, y, light):
"""Store all facts about a light."""
self.light = light
self.color = color.black
self.result_color = self.color
self.transitioning_to = self.color
self.step = 0
self.on = False
self.increasing = False
self.x = x
self.y = y
self.fade_step = 0
def update_base(self):
"""Change color depending on where the fish is."""
if not self.on and random.random() < 0.0005:
self.transitioning_to = color.RGB(1, 1, 1)
self.step = abs(self.transitioning_to - self.color) / (FRAME_RATE / 2)
self.increasing = True
self.on = True
self.color = transition(self.color, self.transitioning_to, self.step)
elif self.on:
if abs(self.transitioning_to - self.color) <= self.step:
if self.increasing == True:
self.transitioning_to = color.RGB(0, 0, 0)
self.step = abs(self.transitioning_to - self.color) / (FRAME_RATE / 2)
self.increasing = False
else:
self.increasing = False
self.on = False
self.color = transition(self.color, self.transitioning_to, self.step)
self.result_color = self.color
def update_star(self, star):
coverage = compute_coverage(star, self.x, self.y)
self.result_color = coverage * star.color + (1 - coverage) * self.result_color
def blit(self):
self.light.setRGBRaw(
*map(lambda x: clamp(0, x, 1), self.result_color.components)
)
def fade_to_black(self, steps):
hue = self.color.to_HSV().h()
sat = self.color.to_HSV().s()
value = self.color.to_HSV().v()
new_color = color.HSV(hue, sat, value - value / steps * self.fade_step).to_RGB()
self.fade_step += 1
self.light.setRGBRaw(*map(lambda x: clamp(0, x, 1), new_color.components))
class TwinkleShow(Show):
def init(self):
self.lights = [[Light(x, y, self.bridge.get_light(x, y))
for y in xrange(self.bridge.HEIGHT)]
for x in xrange(self.bridge.WIDTH)]
#self.sun = Sun()
self.stars = set()
self.tick_count = 0
def update(self):
if self.tick_count > 1000:
fade_time = 3 # sec
if self.tick_count > 1000 + fade_time * FRAME_RATE:
self.stop()
return
for col in self.lights:
for light in col:
light.fade_to_black(fade_time * FRAME_RATE)
else:
if random.random() < 0.01:
print "Adding star"
self.stars.add(ShootingStar())
for col in self.lights:
for light in col:
light.update_base()
for star in list(self.stars):
if not star.update():
self.stars.remove(star)
for x in xrange(int(star.x), int(star.x+star.width+1)):
if x < len(self.lights):
for light in self.lights[x]:
light.update_star(star)
for col in self.lights:
for light in col:
light.blit()
self.tick_count += 1
print self.tick_count
def compute_coverage(fish, x, y):
"""Calculate the coverage of a fish over a box."""
middle_of_fish = fish.x + fish.width / 2
middle_of_panel = x + 0.5
diff = abs(middle_of_fish - middle_of_panel) / (fish.width / 2)
scale = 1 - diff
min_x = min(max(x, fish.x), x + 1)
max_x = max(min(x+1, fish.x + fish.width), x)
min_y = min(max(y, fish.y), y + 1)
max_y = max(min(y+1, fish.y + fish.height), y)
coverage = (max_x - min_x) * (max_y - min_y)
return coverage * scale
def transition(start, end, step):
"""Return a vector equal to (end - start) * step + start."""
if start == end:
return start
diff_vector = end - start
step_vector = diff_vector / abs(diff_vector) * step
result = step_vector + start
return result
def clamp(low, x, high):
"""Return v such that low <= v <= high and |x - v| is minimal."""
return max(low, min(x, high))
|
