-
Notifications
You must be signed in to change notification settings - Fork 0
/
controller.py
95 lines (81 loc) · 2.65 KB
/
controller.py
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
from view import UI
from model import BucketToolField as Field
def _normalize(i):
return int( (65025 - i*i)**0.5 )
def heatmap(timestamp):
timestamp %= 256*3
epoch = timestamp / 256
era = timestamp % 256
if epoch == 0:
return _normalize(era),era,0
elif epoch == 1:
return 0,_normalize(era),era
else:
return era,0,_normalize(era)
class Console:
capture_path = "capture-{:0>3}.png"
colors = {
"white": (255, 255, 255),
"red": (255, 0, 0),
"yellow": (255, 255, 0),
"green": ( 0, 255, 0),
"cyan": ( 0, 255, 255),
"blue": ( 0, 0, 255),
"magenta": (255, 0, 255),
"black": ( 0, 0, 0)
}
def __init__(self):
self.ui = UI()
self.field = Field()
self.heatmap = False
self.color = Console.colors["blue"]
def load_image(self, filename = "image.png"):
self.ui.load(filename)
def setFPS(self, fps):
self.ui.setFPS(20)
def useHeatMap(self):
self.heatmap = True
def useColor(self, color):
self.heatmap = False
self.color = color
def run(self):
exit_loop = False
saves = 0
ticks = 0
while not exit_loop:
for e in self.ui.getEvents():
if e == UI.events.exit:
print e
exit_loop = True
elif e == UI.events.click:
pos = self.ui.getMouse()
print e, pos
self.field.setGrid( self.ui.select(pos) )
self.field.seed(pos)
elif e == UI.events.save:
print e
saves += 1
self.ui.save( Console.capture_path.format(saves) )
elif e == UI.events.toggle:
print e
if self.heatmap:
self.useColor(Console.colors["blue"])
else:
self.useHeatMap()
elif e == UI.events.reset:
print e
ticks = 0
self.ui.reset()
self.field.reset()
else:
print "unknown event type:", e
self.field.update()
updates = self.field.getUpdates()
if len(updates) > 0:
if self.heatmap:
color = heatmap(ticks)
ticks += 1
else:
color = self.color
self.ui.draw(updates,color)
self.ui.tick()