creatumlibre.graphics.selection.region_manager

 1import numpy as np
 2
 3
 4class RegionManager:
 5    """Handles selection modifications, mask updates, and dynamic resizing."""
 6
 7    def __init__(self):
 8        self.mask = None  # Stores selection mask (0-255)
 9        self.bounding_rect = None  # Stores current selection bounds
10
11    def copy(self):
12        new = RegionManager()
13        new.mask = self.mask.copy()
14        new.bounding_rect = self.bounding_rect
15        return new
16
17    def set_bounding_rect(self, x: int, y: int, width: int, height: int):
18        self.bounding_rect = (x, y, width, height)
19
20    def get_bounding_rect(self):
21        return self.bounding_rect
22
23    def get_mask(self):
24        return self.mask
25
26    def initialize_mask(self, image_shape):
27        """Creates an empty mask matching the image dimensions.
28        0: transparent, 1: opaque (visible)
29        """
30        self.mask = np.ones((image_shape[0], image_shape[1]), dtype=np.float32)
31
32    def update_mask(self, selection):
33        """Updates the mask based on selection coordinates."""
34        x, y, width, height = selection.get_rect()
35        self.mask[y : y + height, x : x + width] = (
36            1  # Mark selection area as fully opaque
37        )
38
39        # Update bounding rect
40        self.bounding_rect = (x, y, width, height)
41
42    def apply_mask(self, image, mask):
43        """Applies the mask onto the image, modifying selection visibility."""
44        masked_image = image.copy()
45        masked_image[mask == 0] = (0, 0, 0)  # Hide unselected regions
46
47        return masked_image