-
Notifications
You must be signed in to change notification settings - Fork 0
/
planetdisplay.py
226 lines (192 loc) · 5.29 KB
/
planetdisplay.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
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
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
from math import sin, cos, pi
from PySide.QtCore import Qt
from PySide.QtGui import QColor, QGridLayout, QImage, QPainter, QWidget, QSizePolicy
from sphereview import SphereView
def redbluescale(v):
r = 255 - 255 * v
b = 255 * v
return r, 0, b
def colorscale(v):
m = 1275
r = (255 - m * v if v < 0.2 else
0 if v < 0.6 else
m * (v - 0.6) if v < 0.8 else
255)
g = (0 if v < 0.2 else
m * (v - 0.2) if v < 0.4 else
255 if v < 0.8 else
255 - m * (v - 0.8))
b = (255 if v < 0.4 else
255 - m * (v - 0.4) if v < 0.6 else
0)
return r, g, b
def warmscale(v):
m = 510
r = 255
g = v * m if v < 0.5 else 255
b = 0 if v < 0.5 else m * (v - 0.5)
return r, g, b
def coolscale(v):
m = 1020
r = 255 - m * v if v < 0.25 else 0
g = (255 if v < 0.75 else
255 - m * (v - 0.75))
b = (255 - m * v if v < 0.25 else
m * (v - 0.25) if v < 0.5 else
255)
return r, g, b
def color(tile):
h, c = tile.elevation, tile.climate
l = c.life if c else 0
k = c.koeppen if c else None
colors = {
u'A' : {
u'f' : (0,96,48),
u'm' : (0,168,84),
u'w' : (0,255,128) },
u'B' : {
u'S' : (208,208,208),
u'W' : None },
u'C' : {
u'f' : (0,96,0),
u's' : (0,168,0),
u'w' : (0,255,0) },
u'D' : {
u'f' : (48,96,48),
u's' : (84,168,84),
u'w' : (128,255,128) },
u'E' : {
u'F' : (255, 255, 255),
u'T' : (255, 255, 255) }
}
if h > 0:
try:
f = tile.layers[-1].rock['felsity']
except KeyError:
f = 0
rockcolor = (128, 64 + 64 * f, 128 * f)
if k:
if k[0] == u'E':
color = (255,255,255)
else:
lifecolor = (0,l*255,0)
color = tuple([(r+l)/2 for r,l in zip(rockcolor, lifecolor)])
else:
color = rockcolor
else:
color = (0, 0, 128)
return color
def climatecolor(tile):
h, c = tile.elevation, tile.climate
if c == None:
return (255,255,255)
k = c.koeppen
colors = {
u'A' : {
u'f' : (0,0,255),
u'm' : (0,63,255),
u'w' : (0,127,255) },
u'B' : {
u'S' : (255,127,0),
u'W' : (255,0,0) },
u'C' : {
u'f' : (0,255,0),
u's' : (255,255,0),
u'w' : (127,255,0) },
u'D' : {
u'f' : (0,255,255),
u's' : (255,0,255),
u'w' : (127,127,255) },
u'E' : {
u'F' : (127,127,127),
u'T' : (191,191,191) }
}
if h > 0:
color = colors[k[0]][k[1]]
else:
color = 0, 0, 0
return color
def elevationcolor(tile):
h = tile.elevation
if h > 0:
value = int(128 + 12.5 * h)
color = (value, value, value)
else:
color = (0, 0, 0)
return color
def thicknesscolor(tile):
h = tile.thickness
if h > 0:
value = int(128 + 1.69 * h)
color = (value, value, value)
else:
color = (0, 0, 0)
return color
def subductioncolor(tile):
h = tile.elevation
if h > 0:
value = int(128 + 12.5 * h)
s = tile.subduction
if s > 0:
r = value + (s/10.0) * (255 - value)
color = (255, value, value)
else:
color = (value, value, value)
else:
color = (0, 0, 0)
return color
def rockcolor(tile):
h = tile.elevation
if h > 0:
value = int(128 + 12.5 * h)
try:
rocktype = tile.layers[-1].rock['type']
except TypeError:
rocktype = tile.layers[-1].rock
if rocktype == 'I':
color = (255, value, value)
elif rocktype == 'S':
color = (value, value, 255)
else:
color = (value, 255, value)
else:
color = (0, 0, 0)
return color
class PlanetDisplay(QWidget):
dt = 0.01
_colorfunctions = [climatecolor, color, elevationcolor, rockcolor, subductioncolor, thicknesscolor]
def __init__(self, sim, selecthandler):
QWidget.__init__(self)
self._sim = sim
self._screen = None
self._rotate = 0
self._aspect = self._colorfunctions.index(elevationcolor)
self._select = selecthandler
self.selected = None
self.setLayout(QGridLayout())
self.invalidate()
@property
def rotate(self):
return self._rotate
@rotate.setter
def rotate(self, value):
self._rotate = value
self._screen.rotate(self._rotate)
@property
def aspect(self):
return self._aspect
@aspect.setter
def aspect(self, value):
self._aspect = value
self.invalidate()
def tilecolor(self, tile):
return self._colorfunctions[self._aspect](tile)
def invalidate(self):
if self._screen is not None:
self._screen.deleteLater()
self._screen = SphereView(
self._sim.grid,
{ v: self.tilecolor(t) for (v, t) in self._sim.tiles.iteritems() },
self)
self._screen.rotate(self._rotate)
self.layout().addWidget(self._screen)