map-machine/map_machine/map_configuration.py
2021-09-26 10:00:47 +03:00

82 lines
1.8 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: str = "normal"
AUTHOR: str = "author"
TIME: str = "time"
class LabelMode(Enum):
"""
Label drawing mode.
"""
NO: str = "no"
MAIN: str = "main"
ALL: str = "all"
class BuildingMode(Enum):
"""
Building drawing mode.
"""
NO: str = "no"
FLAT: str = "flat"
ISOMETRIC: str = "isometric"
ISOMETRIC_NO_PARTS: str = "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