map-machine/map_machine/map_configuration.py
2021-11-02 23:51:49 +03:00

75 lines
1.7 KiB
Python

"""
Map drawing configuration.
"""
import argparse
from dataclasses import dataclass
from enum import Enum
__author__ = "Sergey Vartanov"
__email__ = "me@enzet.ru"
class DrawingMode(Enum):
"""Map drawing mode."""
NORMAL = "normal"
AUTHOR = "author"
TIME = "time"
class LabelMode(Enum):
"""Label drawing mode."""
NO = "no"
MAIN = "main"
ALL = "all"
ADDRESS = "address"
class BuildingMode(Enum):
"""Building drawing mode."""
NO = "no"
FLAT = "flat"
ISOMETRIC = "isometric"
ISOMETRIC_NO_PARTS = "isometric-no-parts"
@dataclass
class MapConfiguration:
"""Map drawing configuration."""
drawing_mode: DrawingMode = DrawingMode.NORMAL
building_mode: BuildingMode = BuildingMode.FLAT
label_mode: LabelMode = LabelMode.MAIN
zoom_level: float = 18.0
overlap: int = 12
level: str = "overground"
seed: str = ""
show_tooltips: bool = False
country: str = "world"
ignore_level_matching: bool = False
draw_roofs: bool = True
@classmethod
def from_options(
cls, options: argparse.Namespace, zoom_level: float
) -> "MapConfiguration":
"""Initialize from command-line options."""
return cls(
DrawingMode(options.mode),
BuildingMode(options.buildings),
LabelMode(options.label_mode),
zoom_level,
options.overlap,
options.level,
options.seed,
options.tooltips,
options.country,
options.ignore_level_matching,
options.roofs,
)
def is_wireframe(self) -> bool:
"""Whether drawing mode is special."""
return self.drawing_mode != DrawingMode.NORMAL