ai-lab-one/write_out.py

93 lines
2.3 KiB
Python
Raw Normal View History

from emis_funky_funktions import *
from typing import Tuple
from world import Point, Terrain, World
2023-02-12 04:58:01 +00:00
from PIL import Image
# A function is just a dictionary with class
def terrain_to_color(t: Terrain) -> Tuple[int, int, int]:
"""
Maps a terrain to the color used to denote it in the spec
Where the spec is
https://cs.rit.edu/~jro/courses/intelSys/labs/orienteering/
Return format is a 3-tuple of integer values between 0 and 255, representing RGB
components of a pixel.
>>> terrain_to_color(Terrain.MEDIUM_FOREST)
(2, 208, 60)
"""
match t:
case Terrain.OPEN_LAND:
return (248,148,18)
case Terrain.ROUGH_MEADOW:
return (255,192,0)
case Terrain.EASY_FOREST:
return (255,255,255)
case Terrain.MEDIUM_FOREST:
return (2,208,60)
case Terrain.WALK_FOREST:
return (2,136,40)
case Terrain.BRUSH:
return (5,73,24)
case Terrain.WET:
return (0,0,255)
case Terrain.ROAD:
return (71,51,3)
case Terrain.FOOTPATH:
return (0,0,0)
case Terrain.OOB:
return (205,0,101)
2023-02-12 04:58:01 +00:00
def render_map_with_path(world: World, path: Iterable[Point]) -> bytes:
"""
Compute a series of RGB bytes depicting a path through the world
Uses the colors specified in `terrain_to_color()` for depicting the terrain, with the
path itself drawn in pure, aggressive red (#f00).
>>> tiles = [
... (Terrain.OPEN_LAND, 0), (Terrain.ROUGH_MEADOW, 0), (Terrain.WET, 0),
... (Terrain.OPEN_LAND, 0), (Terrain.ROUGH_MEADOW, 0), (Terrain.WALK_FOREST, 0),
... (Terrain.BRUSH, 0), (Terrain.BRUSH, 0), (Terrain.MEDIUM_FOREST, 0)
... ]
>>> path = [(0, 0), (1, 0), (1, 1), (2, 1)]
>>> world = World(tiles, width=3)
>>> render_map_with_path(world, path) #doctest: +NORMALIZE_WHITESPACE
(255, 0, 0,
255, 0, 0,
0, 0, 255,
248, 148, 18,
255, 0, 0,
255, 0, 0,
5, 73, 24,
5, 73, 24,
2, 208, 60)
"""
path_linear = {
x + y * world.width
for (x, y) in path
}
2023-02-12 04:58:01 +00:00
return bytes(
byte
for (pos, (terrain, _)) in enumerate(world.tiles)
for byte in (
(255, 0, 0)
if pos in path_linear else
terrain_to_color(terrain)
)
)
2023-02-12 04:58:01 +00:00
def save_map(file_path: str, world: World, world_path: Iterable[Point]):
Image.frombytes(
'RGB',
(world.width, len(world.tiles) // world.width),
render_map_with_path(world, world_path)
).save(file_path)
if __name__ == '__main__':
import doctest
doctest.testmod()