Rename Röntgen to Map Machine.

map_machine/mapcss.py

@@ -0,0 +1,205 @@
+"""
+MapCSS scheme creation.
+"""
+import argparse
+import logging
+from pathlib import Path
+from typing import Optional, TextIO
+
+from colour import Color
+
+from map_machine import __project__, __url__
+from map_machine.grid import IconCollection
+from map_machine.icon import ShapeExtractor
+from map_machine.osm_reader import STAGES_OF_DECAY
+from map_machine.scheme import Matcher, Scheme
+from map_machine.workspace import workspace
+
+__author__ = "Sergey Vartanov"
+__email__ = "me@enzet.ru"
+
+NODE_CONFIG: str = """
+node {
+    symbol-shape: circle;
+    symbol-size: 1;
+    text: auto;
+    text-color: black;
+    text-offset-y: -11;
+    text-anchor-horizontal: center;
+    font-size: 11;
+}"""
+
+WAY_CONFIG: str = """
+canvas {
+    fill-color: #FFFFFF;
+}
+way {
+    fill-opacity: 1;
+    text-color: black;
+    text-offset-y: -11;
+}
+relation {
+    fill-opacity: 1;
+    text-color: black;
+    text-offset-y: -11;
+}
+way[building] {
+    fill-color: #D8D0C8;
+    opacity: 1;
+}
+relation[building] {
+    fill-color: #D8D0C8;
+    opacity: 1;
+}"""
+
+HEADER: str = f"""
+/*
+Map paint style that adds icons from Röntgen icon set
+*/
+
+meta {{
+    title: "{__project__}";
+    description: "Map Machine map paint style for JOSM";
+    author: "{__author__}";
+    version: "0.1";
+    link: "{__url__}";
+}}"""
+
+
+class MapCSSWriter:
+    """
+    Writer that converts Map Machine scheme into MapCSS 0.2 format.
+    """
+
+    def __init__(
+        self,
+        scheme: Scheme,
+        icon_directory_name: str,
+        add_icons: bool = True,
+        add_ways: bool = True,
+        add_icons_for_lifecycle: bool = True,
+    ) -> None:
+        self.add_icons: bool = add_icons
+        self.add_ways: bool = add_ways
+        self.add_icons_for_lifecycle: bool = add_icons_for_lifecycle
+        self.icon_directory_name: str = icon_directory_name
+
+        self.point_matchers: list[Matcher] = scheme.node_matchers
+        self.line_matchers: list[Matcher] = scheme.way_matchers
+
+    def add_selector(
+        self,
+        target: str,
+        matcher: Matcher,
+        prefix: str = "",
+        opacity: Optional[float] = None,
+    ) -> str:
+        """
+        Add MapCSS 0.2 selector for node, way, relation, or area.
+
+        :param target: `node`, `way`, `relation`, or `area`
+        :param matcher: tag matcher of Map Machine scheme
+        :param prefix: tag prefix
+        :param opacity: icon opacity
+        :return: string representation of selector
+        """
+        elements: dict[str, str] = {}
+
+        clean_shapes = matcher.get_clean_shapes()
+        if clean_shapes:
+            elements["icon-image"] = (
+                f'"{self.icon_directory_name}/'
+                + "___".join(clean_shapes)
+                + '.svg"'
+            )
+
+            if opacity is not None:
+                elements["icon-opacity"] = f"{opacity:.2f}"
+
+        style: dict[str, str] = matcher.get_style()
+        if style:
+            if "fill" in style:
+                elements["fill-color"] = style["fill"]
+            if "stroke" in style:
+                elements["color"] = style["stroke"]
+            if "stroke-width" in style:
+                elements["width"] = style["stroke-width"]
+            if "stroke-dasharray" in style:
+                elements["dashes"] = style["stroke-dasharray"]
+            if "opacity" in style:
+                elements["fill-opacity"] = style["opacity"]
+                elements["opacity"] = style["opacity"]
+
+        if not elements:
+            return ""
+
+        selector: str = target + matcher.get_mapcss_selector(prefix) + " {\n"
+        for element in elements:
+            selector += f"    {element}: {elements[element]};\n"
+        selector += "}\n"
+
+        return selector
+
+    def write(self, output_file: TextIO) -> None:
+        """Construct icon selectors for MapCSS 0.2 scheme."""
+        output_file.write(HEADER + "\n\n")
+
+        if self.add_ways:
+            output_file.write(WAY_CONFIG + "\n\n")
+
+        if self.add_icons:
+            output_file.write(NODE_CONFIG + "\n\n")
+
+        if self.add_icons:
+            for matcher in self.point_matchers:
+                for target in ["node", "area"]:
+                    output_file.write(self.add_selector(target, matcher))
+
+        if self.add_ways:
+            for line_matcher in self.line_matchers:
+                for target in ["way", "relation"]:
+                    output_file.write(self.add_selector(target, line_matcher))
+
+        if not self.add_icons_for_lifecycle:
+            return
+
+        for index, stage_of_decay in enumerate(STAGES_OF_DECAY):
+            opacity: float = 0.6 - 0.4 * index / (len(STAGES_OF_DECAY) - 1)
+            for matcher in self.point_matchers:
+                if len(matcher.tags) > 1:
+                    continue
+                for target in ["node", "area"]:
+                    output_file.write(
+                        self.add_selector(
+                            target, matcher, stage_of_decay, opacity
+                        )
+                    )
+
+
+def ui(options: argparse.Namespace) -> None:
+    """Write MapCSS 0.2 scheme."""
+    directory: Path = workspace.get_mapcss_path()
+    icons_with_outline_path: Path = workspace.get_mapcss_icons_path()
+
+    scheme: Scheme = Scheme(workspace.DEFAULT_SCHEME_PATH)
+    extractor: ShapeExtractor = ShapeExtractor(
+        workspace.ICONS_PATH, workspace.ICONS_CONFIG_PATH
+    )
+    collection: IconCollection = IconCollection.from_scheme(scheme, extractor)
+    collection.draw_icons(
+        icons_with_outline_path,
+        color=Color("black"),
+        outline=True,
+        outline_opacity=0.5,
+    )
+    mapcss_writer: MapCSSWriter = MapCSSWriter(
+        scheme,
+        workspace.MAPCSS_ICONS_DIRECTORY_NAME,
+        options.icons,
+        options.ways,
+        options.lifecycle,
+    )
+    with workspace.get_mapcss_file_path().open("w+") as output_file:
+        mapcss_writer.write(output_file)
+
+    logging.info(f"MapCSS 0.2 scheme is written to {directory}.")