3pp-mirror/map-machine
1
0
Fork 0
mirror of https://github.com/enzet/map-machine.git synced 2025-05-11 16:16:51 +02:00
Code Issues Projects Releases Packages Wiki Activity

Rename Röntgen to Map Machine.

Browse source
This commit is contained in:
Sergey Vartanov 2021-09-08 03:55:16 +03:00
parent 38c4e00de3
commit 2bd89a6539
57 changed files with 252 additions and 236 deletions
Download patch file Download diff file Expand all files Collapse all files

232
roentgen/grid.py
View file

@ -1,232 +0,0 @@
"""
Icon grid drawing.
"""
import logging
from dataclasses import dataclass
from pathlib import Path
from typing import Optional, Union
import numpy as np
from colour import Color
from svgwrite import Drawing
from svgwrite.shapes import Rect
from roentgen.icon import Icon, Shape, ShapeExtractor, ShapeSpecification
from roentgen.scheme import NodeMatcher, Scheme
from roentgen.workspace import workspace
__author__ = "Sergey Vartanov"
__email__ = "me@enzet.ru"
@dataclass
class IconCollection:
"""
Collection of icons.
"""
icons: list[Icon]
@classmethod
def from_scheme(
cls,
scheme: Scheme,
extractor: ShapeExtractor,
background_color: Color = Color("white"),
color: Color = Color("black"),
add_unused: bool = False,
add_all: bool = False,
) -> "IconCollection":
"""
Collect all possible icon combinations in grid.
:param scheme: tag specification
:param extractor: shape extractor for icon creation
:param background_color: background color
:param color: icon color
:param add_unused: create icons from shapes that have no corresponding
tags
:param add_all: create icons from all possible shapes including parts
"""
icons: list[Icon] = []
def add() -> Icon:
"""Construct icon and add it to the list."""
specifications: list[ShapeSpecification] = [
scheme.get_shape_specification(x, extractor)
for x in current_set
]
constructed_icon: Icon = Icon(specifications)
constructed_icon.recolor(color, white=background_color)
if constructed_icon not in icons:
icons.append(constructed_icon)
return constructed_icon
current_set: list[Union[str, dict[str, str]]]
for matcher in scheme.node_matchers:
matcher: NodeMatcher
if matcher.shapes:
current_set = matcher.shapes
add()
if matcher.add_shapes:
current_set = matcher.add_shapes
add()
if not matcher.over_icon:
continue
if matcher.under_icon:
for icon_id in matcher.under_icon:
current_set = [icon_id] + matcher.over_icon
add()
if not (matcher.under_icon and matcher.with_icon):
continue
for icon_id in matcher.under_icon:
for icon_2_id in matcher.with_icon:
current_set: list[str] = (
[icon_id] + [icon_2_id] + matcher.over_icon
)
add()
for icon_2_id in matcher.with_icon:
for icon_3_id in matcher.with_icon:
current_set = (
[icon_id]
+ [icon_2_id]
+ [icon_3_id]
+ matcher.over_icon
)
if (
icon_2_id != icon_3_id
and icon_2_id != icon_id
and icon_3_id != icon_id
):
add()
specified_ids: set[str] = set()
for icon in icons:
specified_ids |= set(icon.get_shape_ids())
if add_unused:
for shape_id in extractor.shapes.keys() - specified_ids:
shape: Shape = extractor.get_shape(shape_id)
if shape.is_part:
continue
icon: Icon = Icon([ShapeSpecification(shape)])
icon.recolor(color)
icons.append(icon)
if add_all:
for shape_id in extractor.shapes.keys():
shape: Shape = extractor.get_shape(shape_id)
icon: Icon = Icon([ShapeSpecification(shape)])
icon.recolor(color)
icons.append(icon)
return cls(icons)
def draw_icons(
self,
output_directory: Path,
by_name: bool = False,
color: Optional[Color] = None,
outline: bool = False,
outline_opacity: float = 1.0,
) -> None:
"""
:param output_directory: path to the directory to store individual SVG
files for icons
:param by_name: use names instead of identifiers
:param color: fill color
:param outline: if true, draw outline beneath the icon
:param outline_opacity: opacity of the outline
"""
if by_name:
def get_file_name(x: Icon) -> str:
"""Generate human-readable file name."""
return f"Röntgen {' + '.join(x.get_names())}.svg"
else:
def get_file_name(x: Icon) -> str:
"""Generate file name with unique identifier."""
return f"{'___'.join(x.get_shape_ids())}.svg"
for icon in self.icons:
icon.draw_to_file(
output_directory / get_file_name(icon),
color=color,
outline=outline,
outline_opacity=outline_opacity,
)
def draw_grid(
self,
file_name: Path,
columns: int = 16,
step: float = 24,
background_color: Color = Color("white"),
) -> None:
"""
Draw icons in the form of table.
:param file_name: output SVG file name
:param columns: number of columns in grid
:param step: horizontal and vertical distance between icons in grid
:param background_color: background color
"""
point: np.ndarray = np.array((step / 2, step / 2))
width: float = step * columns
height: int = int(int(len(self.icons) / (width / step) + 1) * step)
svg: Drawing = Drawing(str(file_name), (width, height))
svg.add(svg.rect((0, 0), (width, height), fill=background_color.hex))
for icon in self.icons:
icon: Icon
rectangle: Rect = svg.rect(
point - np.array((10, 10)), (20, 20), fill=background_color.hex
)
svg.add(rectangle)
icon.draw(svg, point)
point += np.array((step, 0))
if point[0] > width - 8:
point[0] = step / 2
point += np.array((0, step))
height += step
with file_name.open("w") as output_file:
svg.write(output_file)
def __len__(self) -> int:
return len(self.icons)
def sort(self) -> None:
"""Sort icon list."""
self.icons = sorted(self.icons)
def draw_icons() -> None:
"""
Draw all possible icon shapes combinations as grid in one SVG file and as
individual SVG files.
"""
scheme: Scheme = Scheme(workspace.DEFAULT_SCHEME_PATH)
extractor: ShapeExtractor = ShapeExtractor(
workspace.ICONS_PATH, workspace.ICONS_CONFIG_PATH
)
collection: IconCollection = IconCollection.from_scheme(
scheme, extractor, add_all=True
)
icon_grid_path: Path = workspace.get_icon_grid_path()
collection.draw_grid(icon_grid_path)
logging.info(f"Icon grid is written to {icon_grid_path}.")
icons_by_id_path: Path = workspace.get_icons_by_id_path()
icons_by_name_path: Path = workspace.get_icons_by_name_path()
collection.draw_icons(icons_by_id_path)
collection.draw_icons(icons_by_name_path, by_name=True)
logging.info(
f"Icons are written to {icons_by_name_path} and {icons_by_id_path}."
)