map-machine/map_machine/mapcss.py

"""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.osm.osm_reader import STAGES_OF_DECAY
from map_machine.pictogram.icon import ShapeExtractor
from map_machine.pictogram.icon_collection import IconCollection
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] = {}

        for value in matcher.tags.values():
            if value.startswith("^"):
                return ""

        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 key, value in elements.items():
            selector += f"    {key}: {value};\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.0)
            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 generate_mapcss(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.from_file(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,
        workspace.ICONS_LICENSE_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+", encoding="utf-8"
    ) as output_file:
        mapcss_writer.write(output_file)

    logging.info(f"MapCSS 0.2 scheme is written to {directory}.")