Millefeuille / cubEngine

import math from kandinsky import * def shade_color(intensity, dist, way, color): if color >= 255 - 20: color = 255 - 20 if way: ret = ((color / 255.0) + 0.1) * (intensity * (dist / 255.0)) else: ret = ((color / 255.0) + 0.1) * (intensity * (255.0 / dist)) ret = color + 20 if ret > color + 20 else ret ret = 0 if ret < 0 else ret return int(ret) class CubColor: def __init__(self, r, g, b): self.r = r self.g = g self.b = b self.shadedR = 0 self.shadedG = 0 self.shadedB = 0 def shade(self, intensity, dist, way): self.shadedR = shade_color(intensity, dist, way, self.r) self.shadedG = shade_color(intensity, dist, way, self.g) self.shadedB = shade_color(intensity, dist, way, self.b) return [self.shadedR, self.shadedG, self.shadedB] def get_color(self): return [self.r, self.g, self.b] class CubVector: def __init__(self, x, y): self.x = x self.y = y def translate(self, x, y): self.x += x self.y += y def rotate(self, angle): old = CubVector(self.x, self.y) self.x = old.x * math.cos(math.radians(angle)) - old.y * math.sin(math.radians(angle)) self.y = old.x * math.sin(math.radians(angle)) + old.y * math.cos(math.radians(angle)) class CubPlayer: def __init__(self, pos, prov_dir): self.pos = pos self.dir = prov_dir self.plan = CubVector(0, 0.66) self.view_offset = 0 def move(self, angle, speed, prov_map): dire = CubVector(self.dir.x, self.dir.y) dire.rotate(angle) n_pos = CubVector( self.pos.x + dire.x / speed, self.pos.y + dire.y / speed ) if prov_map[int(n_pos.y + dire.y / (speed / 2))][int(n_pos.x + dire.x / (speed / 2))] == 1: return self.pos.x = n_pos.x self.pos.y = n_pos.y def look_sides(self, angle): self.dir.rotate(angle) self.plan.rotate(angle) def look_up(self, speed): if self.view_offset < 100: self.view_offset += speed def look_down(self, speed): if self.view_offset > -100: self.view_offset -= speed class CubRay: def __init__(self, cam): self.origin = CubVector(player.pos.x, player.pos.y) self.dir = CubVector( player.dir.x + player.plan.x * cam, player.dir.y + player.plan.y * cam ) self.mpos = CubVector(int(player.pos.x), int(player.pos.y)) self.dist = CubVector( 1 if self.dir.x == 0 else 0, 1 if self.dir.y == 0 else 0 ) if self.dir.x != 0 and self.dir.y != 0: self.dist.x = abs(1 / self.dir.x) self.dist.y = abs(1 / self.dir.y) self.step = CubVector(1, 1) self.sdist = CubVector( (self.mpos.x + 1.0 - player.pos.x) * self.dist.x, (self.mpos.y + 1.0 - player.pos.y) * self.dist.y ) if self.dir.x < 0: self.step.x = -1 self.sdist.x = (player.pos.x - self.mpos.x) * self.dist.x if self.dir.y < 0: self.step.y = -1 self.sdist.y = (player.pos.y - self.mpos.y) * self.dist.y self.hit = 0 self.side = 0 def cast(self): while self.hit == 0: if self.sdist.x < self.sdist.y: self.side = 0 self.sdist.x += self.dist.x self.mpos.x += self.step.x else: self.side = 1 self.sdist.y += self.dist.y self.mpos.y += self.step.y if prov_map[int(self.mpos.y)][int(self.mpos.x)] == 1: self.hit = 1 def calc(self, x): wall = CubColor(255, 55, 0) if self.side == 0: dist = (self.mpos.x - player.pos.x + (1 - self.step.x) / 2) / self.dir.x else: dist = (self.mpos.y - player.pos.y + (1 - self.step.y) / 2) / self.dir.y if dist == 0: line_height = 0 else: line_height = int(height / dist) start = int(-line_height / 2 + height / 2 + player.view_offset) end = int(line_height / 2 + height / 2 + player.view_offset) return [wall.shade(5, dist, False), x, start, end] prov_map = [ [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1] ] width = 320 height = 222 player = CubPlayer(CubVector(12.5, 10.5), CubVector(1, 0)) ceil = CubColor(50, 50, 50) floor = CubColor(20, 20, 20) running = True def draw_line(sc_x, sc_y, color): for i in range(sc_y[0], sc_y[1]): set_pixel(sc_x, i, color) def render(): ceil = CubColor(50, 50, 50) floor = CubColor(20, 20, 20) #for x in range(width): # draw_line(x, (0, height), (0,0,0)) for x in range(width): cam = 2 * x / width - 1 ray = CubRay(cam) ray.cast() lst = ray.calc(x) draw_line(lst[1], (0, lst[2]), ceil.get_color()) draw_line(lst[1], (lst[2], lst[3]), lst[0]) draw_line(lst[1], (lst[3], height), floor.get_color()) def update(keys): global running tm = 4 if "right" in keys: player.move(90, int(10 / tm), prov_map) elif "left" in keys: player.move(-90, int(10 / tm), prov_map) if "down" in keys: player.move(180, int(10 / tm), prov_map) elif "up" in keys: player.move(0, int(10 / tm), prov_map) if "8" in keys: player.look_up(int((height / 100)) * tm) elif "5" in keys: player.look_down(int((height / 100)) * tm) if "4" in keys: player.look_sides(int(width / -200) * tm) elif "6" in keys: player.look_sides(int(width / 200) * tm) if "0" in keys: running = False def game(): global running #render() running = True while running: keys = get_keys() if keys != set(): update(keys) render()