cubequest/questhelper/map/map_renderer.py
henriquenakashima d9916579b6 Render paths
2021-03-28 16:09:33 -04:00

121 lines
3.4 KiB
Python

import dataclasses
from map.quest_map import MapEncounter, QuestMap, EncounterType, QuestPath
from typing import List, Dict, Optional
_ENCOUNTER_ICONS = {
EncounterType.BATTLE_REGULAR: 'noun_Battle_3647629_cropped.png',
EncounterType.BATTLE_ELITE: 'noun_Shield_3806325_cropped.png',
EncounterType.BATTLE_BOSS: 'noun_dragon_3820621_cropped.png',
EncounterType.TOWN_TAVERN: 'noun_Beer_3054928_cropped.png',
EncounterType.TOWN_DOCKS: 'noun_Caravel_2421659_cropped.png',
EncounterType.TOWN_MARKET: 'noun_Store_786819_cropped.png',
EncounterType.TOWN_LIBRARY: 'noun_Library_3820590_cropped.png',
EncounterType.TOWN_WORKSHOP: 'noun_cogs_442166_cropped.png',
EncounterType.TOWN_TOWN_HALL: 'noun_brussels town hall_3284349_cropped.png',
EncounterType.TOWN_TEMPLE: 'noun_temple_2382601_cropped.png',
EncounterType.TOWN_REST: 'noun_Tent_3814999_cropped.png',
EncounterType.TOWN_ROAD: 'noun_Hiking Trail_2686687_cropped.png',
}
_PATH_ICONS = {
QuestPath.FORWARD: 'arrow_down.png',
QuestPath.FORWARD_LEFT: 'arrow_down_left.png',
QuestPath.FORWARD_RIGHT: 'arrow_down_right.png',
}
@dataclasses.dataclass(frozen=True)
class RenderedElement:
description: str
src: str
x: int
y: int
w: int
h: int
class RenderedEncounter(RenderedElement):
pass
class RenderedPath(RenderedElement):
pass
@dataclasses.dataclass(frozen=True)
class RenderedMap:
encounters: List[RenderedEncounter]
paths: List[RenderedPath]
def render_map(quest_map: QuestMap) -> RenderedMap:
rendered_encounters = []
row: List[MapEncounter]
for r, row in enumerate(quest_map.encounters):
for c, encounter in enumerate(row):
rendered_encounter = _render_encounter(r, c, encounter)
if rendered_encounter is not None:
rendered_encounters.append(rendered_encounter)
rendered_paths = []
row: List[QuestPath]
for r, row in enumerate(quest_map.paths):
for c, path in enumerate(row):
rendered_path = _render_path(r, c, path)
if rendered_path is not None:
rendered_paths.append(rendered_path)
return RenderedMap(rendered_encounters, rendered_paths)
_ENCOUNTER_W = 40
_ENCOUNTER_H = 40
_ARROW_W = 40
_ARROW_H = 20
_PADDING_W = 10
_PADDING_V = 5
_ENCOUNTER_COLUMN_W = _ENCOUNTER_W + _PADDING_W
_PATH_COLUMN_W = _ARROW_W + _PADDING_W
_COLUMN_W = _ENCOUNTER_COLUMN_W + _PATH_COLUMN_W
_ROW_H = _ENCOUNTER_H + _PADDING_V + _ARROW_H + _PADDING_V
def _render_encounter(r: int, c: int, encounter: MapEncounter) -> Optional[RenderedEncounter]:
if encounter.type == EncounterType.INACCESSIBLE:
return None
icon = f'map/{_ENCOUNTER_ICONS[encounter.type]}'
x = _COLUMN_W * c
y = _ROW_H * r
w = _ENCOUNTER_W
h = _ENCOUNTER_H
return RenderedEncounter(
description=encounter.description,
src=icon,
x=x,
y=y,
w=w,
h=h
)
def _render_path(r: int, c: int, path: QuestPath) -> Optional[RenderedPath]:
if path == QuestPath.NO_PATH:
return None
icon = f'map/{_PATH_ICONS[path]}'
x = _ENCOUNTER_COLUMN_W * ((c + 1) // 2) + _PATH_COLUMN_W * (c // 2)
y = (_ENCOUNTER_H + _PADDING_V) + _ROW_H * r
w = _ARROW_W
h = _ARROW_H
return RenderedEncounter(
description=path.description,
src=icon,
x=x,
y=y,
w=w,
h=h
)