def test_elevation_color(self): for i in range(0, 20): v = i / 2.0 c = ra, ga, ba = elevation_color(v) delta = 0.0000001 c_low = rb, gb, bb = elevation_color(v - delta) c_high = rc, gc, bc = elevation_color(v + delta) # we want values to be in range self._assert_is_valid_color(c, "color for %f" % v) self._assert_is_valid_color(c_low, "color for %f (low)" % (v - delta)) self._assert_is_valid_color(c_high, "color for %f (high)" % (v + delta))
def draw_simple_elevation_on_screen(world, canvas): width = world.width height = world.height for y in range(0, height): for x in range(0, width): e = world.elevation['data'][y][x] r, g, b = elevation_color(e) col = QtGui.QColor(int(r * 255), int(g * 255), int(b * 255)) canvas.setPixel(x, y, col.rgb())