mirror of
https://github.com/enzet/map-machine.git
synced 2025-05-23 14:06:23 +02:00
Rename Röntgen to Map Machine.
This commit is contained in:
Sergey Vartanov
parent
38c4e00de3
commit
2bd89a6539
57 changed files with 252 additions and 236 deletions
265
map_machine/point.py
Normal file
265
map_machine/point.py
Normal file
|
|
@ -0,0 +1,265 @@
|
|||
"""
|
||||
Point: node representation on the map.
|
||||
"""
|
||||
from typing import Optional
|
||||
|
||||
import numpy as np
|
||||
import svgwrite
|
||||
from colour import Color
|
||||
|
||||
from map_machine.icon import Icon, IconSet
|
||||
from map_machine.map_configuration import LabelMode
|
||||
from map_machine.osm_reader import Tagged
|
||||
from map_machine.text import Label
|
||||
|
||||
__author__ = "Sergey Vartanov"
|
||||
__email__ = "me@enzet.ru"
|
||||
|
||||
DEFAULT_FONT: str = "Roboto"
|
||||
|
||||
|
||||
class Occupied:
|
||||
"""
|
||||
Structure that remembers places of the canvas occupied by elements (icons,
|
||||
texts, shapes).
|
||||
"""
|
||||
|
||||
def __init__(self, width: int, height: int, overlap: int) -> None:
|
||||
self.matrix = np.full((int(width), int(height)), False, dtype=bool)
|
||||
self.width: float = width
|
||||
self.height: float = height
|
||||
self.overlap: int = overlap
|
||||
|
||||
def check(self, point: np.ndarray) -> bool:
|
||||
"""Check whether point is already occupied by other elements."""
|
||||
if 0 <= point[0] < self.width and 0 <= point[1] < self.height:
|
||||
return self.matrix[point[0], point[1]]
|
||||
return True
|
||||
|
||||
def register(self, point: np.ndarray) -> None:
|
||||
"""Register that point is occupied by an element."""
|
||||
if 0 <= point[0] < self.width and 0 <= point[1] < self.height:
|
||||
self.matrix[point[0], point[1]] = True
|
||||
assert self.matrix[point[0], point[1]]
|
||||
|
||||
|
||||
class Point(Tagged):
|
||||
"""
|
||||
Object on the map with no dimensional attributes.
|
||||
|
||||
It may have icons and labels.
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
icon_set: IconSet,
|
||||
labels: list[Label],
|
||||
tags: dict[str, str],
|
||||
processed: set[str],
|
||||
point: np.ndarray,
|
||||
priority: float = 0,
|
||||
is_for_node: bool = True,
|
||||
draw_outline: bool = True,
|
||||
) -> None:
|
||||
super().__init__(tags)
|
||||
|
||||
assert point is not None
|
||||
|
||||
self.icon_set: IconSet = icon_set
|
||||
self.labels: list[Label] = labels
|
||||
self.processed: set[str] = processed
|
||||
self.point: np.ndarray = point
|
||||
self.priority: float = priority
|
||||
self.layer: float = 0
|
||||
self.is_for_node: bool = is_for_node
|
||||
self.draw_outline: bool = draw_outline
|
||||
|
||||
self.y = 0
|
||||
self.main_icon_painted: bool = False
|
||||
|
||||
def draw_main_shapes(
|
||||
self, svg: svgwrite.Drawing, occupied: Optional[Occupied] = None
|
||||
) -> None:
|
||||
"""Draw main shape for one node."""
|
||||
keys_left = [x for x in self.tags.keys() if x not in self.processed]
|
||||
if (
|
||||
self.icon_set.main_icon.is_default()
|
||||
and not self.icon_set.extra_icons
|
||||
and (not keys_left or not self.is_for_node)
|
||||
):
|
||||
return
|
||||
|
||||
position = self.point + np.array((0, self.y))
|
||||
self.main_icon_painted: bool = self.draw_point_shape(
|
||||
svg, self.icon_set.main_icon, position, occupied, tags=self.tags
|
||||
)
|
||||
if self.main_icon_painted:
|
||||
self.y += 16
|
||||
|
||||
def draw_extra_shapes(
|
||||
self, svg: svgwrite.Drawing, occupied: Optional[Occupied] = None
|
||||
) -> None:
|
||||
"""Draw secondary shapes."""
|
||||
if not self.icon_set.extra_icons or not self.main_icon_painted:
|
||||
return
|
||||
|
||||
is_place_for_extra: bool = True
|
||||
if occupied:
|
||||
left: float = -(len(self.icon_set.extra_icons) - 1) * 8
|
||||
for _ in self.icon_set.extra_icons:
|
||||
point: np.ndarray = np.array(
|
||||
(int(self.point[0] + left), int(self.point[1] + self.y))
|
||||
)
|
||||
if occupied.check(point):
|
||||
is_place_for_extra = False
|
||||
break
|
||||
left += 16
|
||||
|
||||
if is_place_for_extra:
|
||||
left: float = -(len(self.icon_set.extra_icons) - 1) * 8
|
||||
for icon in self.icon_set.extra_icons:
|
||||
point: np.ndarray = self.point + np.array((left, self.y))
|
||||
self.draw_point_shape(svg, icon, point, occupied=occupied)
|
||||
left += 16
|
||||
if self.icon_set.extra_icons:
|
||||
self.y += 16
|
||||
|
||||
def draw_point_shape(
|
||||
self,
|
||||
svg: svgwrite.Drawing,
|
||||
icon: Icon,
|
||||
position: np.ndarray,
|
||||
occupied: Occupied,
|
||||
tags: Optional[dict[str, str]] = None,
|
||||
) -> bool:
|
||||
"""Draw one combined icon and its outline."""
|
||||
# Down-cast floats to integers to make icons pixel-perfect.
|
||||
position: np.ndarray = np.array((int(position[0]), int(position[1])))
|
||||
|
||||
if occupied and occupied.check(position):
|
||||
return False
|
||||
|
||||
if self.draw_outline:
|
||||
icon.draw(svg, position, outline=True)
|
||||
|
||||
icon.draw(svg, position, tags=tags)
|
||||
|
||||
if occupied:
|
||||
overlap: int = occupied.overlap
|
||||
for i in range(-overlap, overlap):
|
||||
for j in range(-overlap, overlap):
|
||||
occupied.register(
|
||||
np.array((position[0] + i, position[1] + j))
|
||||
)
|
||||
|
||||
return True
|
||||
|
||||
def draw_texts(
|
||||
self,
|
||||
svg: svgwrite.Drawing,
|
||||
occupied: Optional[Occupied] = None,
|
||||
label_mode: str = LabelMode.MAIN,
|
||||
) -> None:
|
||||
"""Draw all labels."""
|
||||
labels: list[Label]
|
||||
|
||||
if label_mode == LabelMode.MAIN:
|
||||
labels = self.labels[:1]
|
||||
elif label_mode == LabelMode.ALL:
|
||||
labels = self.labels
|
||||
else:
|
||||
return
|
||||
|
||||
for label in labels:
|
||||
text = label.text
|
||||
text = text.replace(""", '"')
|
||||
text = text.replace("&", "&")
|
||||
text = text[:26] + ("..." if len(text) > 26 else "")
|
||||
point = self.point + np.array((0, self.y + 2))
|
||||
self.draw_text(
|
||||
svg, text, point, occupied, label.fill, size=label.size
|
||||
)
|
||||
|
||||
def draw_text(
|
||||
self,
|
||||
svg: svgwrite.Drawing,
|
||||
text: str,
|
||||
point: np.ndarray,
|
||||
occupied: Optional[Occupied],
|
||||
fill: Color,
|
||||
size: float = 10.0,
|
||||
out_fill: Color = Color("white"),
|
||||
out_opacity: float = 0.5,
|
||||
out_fill_2: Optional[Color] = None,
|
||||
out_opacity_2: float = 1.0,
|
||||
is_debug: bool = False,
|
||||
) -> None:
|
||||
"""
|
||||
Drawing text.
|
||||
|
||||
###### ### outline 2
|
||||
#------# --- outline 1
|
||||
#| Text |#
|
||||
#------#
|
||||
######
|
||||
"""
|
||||
length: int = len(text) * 6 # FIXME
|
||||
|
||||
if occupied:
|
||||
is_occupied: bool = False
|
||||
for i in range(-int(length / 2), int(length / 2)):
|
||||
text_position: np.ndarray = np.array(
|
||||
(int(point[0] + i), int(point[1] - 4))
|
||||
)
|
||||
if occupied.check(text_position):
|
||||
is_occupied = True
|
||||
break
|
||||
|
||||
if is_occupied:
|
||||
return
|
||||
|
||||
for i in range(-int(length / 2), int(length / 2)):
|
||||
for j in range(-12, 5):
|
||||
occupied.register(
|
||||
np.array((int(point[0] + i), int(point[1] + j)))
|
||||
)
|
||||
if is_debug:
|
||||
svg.add(svg.rect((point[0] + i, point[1] + j), (1, 1)))
|
||||
|
||||
if out_fill_2:
|
||||
text_element = svg.text(
|
||||
text, point, font_size=size, text_anchor="middle",
|
||||
font_family=DEFAULT_FONT, fill=out_fill_2.hex,
|
||||
stroke_linejoin="round", stroke_width=5, stroke=out_fill_2.hex,
|
||||
opacity=out_opacity_2
|
||||
) # fmt: skip
|
||||
svg.add(text_element)
|
||||
if out_fill:
|
||||
text_element = svg.text(
|
||||
text, point, font_size=size, text_anchor="middle",
|
||||
font_family=DEFAULT_FONT, fill=out_fill.hex,
|
||||
stroke_linejoin="round", stroke_width=3, stroke=out_fill.hex,
|
||||
opacity=out_opacity,
|
||||
) # fmt: skip
|
||||
svg.add(text_element)
|
||||
text_element = svg.text(
|
||||
text, point, font_size=size, text_anchor="middle",
|
||||
font_family=DEFAULT_FONT, fill=fill.hex,
|
||||
) # fmt: skip
|
||||
svg.add(text_element)
|
||||
|
||||
self.y += 11
|
||||
|
||||
def get_size(self) -> np.ndarray:
|
||||
"""
|
||||
Get width and height of the point visual representation if there is
|
||||
space for all elements.
|
||||
"""
|
||||
icon_size: int = 16
|
||||
width: int = icon_size * (
|
||||
1 + max(2, len(self.icon_set.extra_icons) - 1)
|
||||
)
|
||||
height: int = icon_size * (1 + int(len(self.icon_set.extra_icons) / 3))
|
||||
if len(self.labels):
|
||||
height += 4 + 11 * len(self.labels)
|
||||
return np.array((width, height))
|
||||
Loading…
Add table
Add a link
Reference in a new issue