creatumlibre.graphics.boolean_operations.image_boolean

 1# pylint: disable=no-member
 2
 3
 4import cv2
 5import numpy as np
 6
 7from creatumlibre.ui.manager.image_handler import ImageHandler
 8
 9
10def merge(
11    from_obj: ImageHandler, to_obj: ImageHandler
12):  # pylint: disable=too-many-locals
13    """Composites the 'from' image into the 'to' image using its mask and position."""
14    overlay = from_obj.get_image()
15    mask = from_obj.get_mask()
16    act_postion = XY.from_tuple(from_obj.get_position())
17
18    base = to_obj.get_image()
19    h, w = overlay.shape[:2]
20    if h < 1 or w < 1:
21        return
22
23    base_h, base_w = base.shape[:2]
24
25    # Begrenzung berechnen
26    posTopLeft = act_postion.max_XY(XY(0, 0))
27    posRightBottom = posTopLeft.add(XY(w, h)).min_XY(XY(base_w, base_h))
28
29    # Falls komplett außerhalb: abbrechen
30    if posTopLeft.x >= posRightBottom.x or posTopLeft.y >= posRightBottom.y:
31        return
32
33    # Offset im Overlay berechnen
34    overlay_1 = posTopLeft.sub(act_postion)
35    overlay_2 = overlay_1.add(posRightBottom.sub(posTopLeft))
36
37    roi = base[posTopLeft.y : posRightBottom.y, posTopLeft.x : posRightBottom.x].astype(
38        np.float32
39    )
40    overlay_crop = overlay[overlay_1.y : overlay_2.y, overlay_1.x : overlay_2.x].astype(
41        np.float32
42    )
43
44    if mask is not None:
45        alpha = mask[overlay_1.y : overlay_2.y, overlay_1.x : overlay_2.x].astype(
46            np.float32
47        )
48        if len(alpha.shape) == 2:
49            alpha = cv2.merge([alpha] * 3)
50    else:
51        alpha = np.ones_like(overlay_crop, dtype=np.float32)
52
53    blended = overlay_crop * alpha + roi * (1 - alpha)
54    base[posTopLeft.y : posRightBottom.y, posTopLeft.x : posRightBottom.x] = (
55        blended.astype(np.uint8)
56    )
57
58    to_obj.set_image(base)
59
60
61class XY:
62    """2D vector class with basic operations."""
63
64    def __init__(self, x: int, y: int | None = None):
65        self.x = x
66        self.y = y
67
68    @classmethod
69    def from_tuple(cls, vector: tuple[int, int]) -> "XY":
70        return cls(vector[0], vector[1])
71
72    def to_tuple(self) -> tuple[int, int]:
73        return (self.x, self.y)
74
75    def add(self, other: "XY") -> "XY":
76        return XY(self.x + other.x, self.y + other.y)
77
78    def sub(self, other: "XY") -> "XY":
79        return XY(self.x - other.x, self.y - other.y)
80
81    def max_XY(self, ref: "XY") -> "XY":
82        return XY(max(self.x, ref.x), max(self.y, ref.y))
83
84    def min_XY(self, ref: "XY") -> "XY":
85        return XY(min(self.x, ref.x), min(self.y, ref.y))
86
87    def __repr__(self):
88        return f"XY(x={self.x}, y={self.y})"