Render paths

questhelper/map/map_renderer.py

@@ -1,8 +1,10 @@
-from map.quest_map import MapEncounter, QuestMap, EncounterType
+import dataclasses
+
+from map.quest_map import MapEncounter, QuestMap, EncounterType, QuestPath
 
 from typing import List, Dict, Optional
 
-_ICONS = {
+_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',
@@ -18,9 +20,39 @@ _ICONS = {
 }
 
 
-def render_map(quest_map: QuestMap) -> List[Dict]:
-    rendered_encounters = []
+_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):
@@ -28,23 +60,62 @@ def render_map(quest_map: QuestMap) -> List[Dict]:
             if rendered_encounter is not None:
                 rendered_encounters.append(rendered_encounter)
 
-    return rendered_encounters
+    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)
 
 
-def _render_encounter(r: int, c: int, encounter: MapEncounter) -> Optional[Dict]:
+_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/{_ICONS[encounter.type]}'
-    x = 120 * c
-    y = 100 * r
-    w = 40
-    h = 40
-    return {
-        'description': encounter.description,
-        'src': icon,
-        'x': x,
-        'y': y,
-        'w': w,
-        'h': h,
-    }
+    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
+    )