valetudo-map-parser 0.1.9b27__py3-none-any.whl → 0.1.9b29__py3-none-any.whl

valetudo_map_parser/config/auto_crop.py

@@ -25,15 +25,20 @@ class TrimError(Exception):
 class AutoCrop:
     """Auto Crop Class for trimming and zooming images."""
 
-    def __init__(self, image_handler):
+    def __init__(self, image_handler, shared):
         self.imh = image_handler
-        self.file_name = self.imh.file_name
+        self.file_name = shared.file_name
+        self.shared = shared
 
-    def validate_crop_dimensions(self, shared):
+    @staticmethod
+    def validate_crop_dimensions(shared):
         """Ensure width and height are valid before processing cropping."""
         if shared.image_ref_width <= 0 or shared.image_ref_height <= 0:
-            _LOGGER.warning("Auto-crop failed: Invalid dimensions (width=%s, height=%s). Using original image.",
-                            shared.image_ref_width, shared.image_ref_height)
+            _LOGGER.warning(
+                "Auto-crop failed: Invalid dimensions (width=%s, height=%s). Using original image.",
+                shared.image_ref_width,
+                shared.image_ref_height,
+            )
             return False
         return True
 
@@ -65,18 +70,18 @@ class AutoCrop:
             ),
         )
         # Ensure shared reference dimensions are updated
-        if hasattr(self.imh.shared, "image_ref_height") and hasattr(
-            self.imh.shared, "image_ref_width"
+        if hasattr(self.shared, "image_ref_height") and hasattr(
+            self.shared, "image_ref_width"
         ):
-            self.imh.shared.image_ref_height = trimmed_height
-            self.imh.shared.image_ref_width = trimmed_width
+            self.shared.image_ref_height = trimmed_height
+            self.shared.image_ref_width = trimmed_width
         else:
             _LOGGER.warning(
                 "Shared attributes for image dimensions are not initialized."
             )
         return trimmed_width, trimmed_height
 
-    async def _async_auto_crop_data(self, tdata: TrimsData = None):  # , tdata=None
+    async def _async_auto_crop_data(self, tdata: TrimsData):  # , tdata=None
         """Load the auto crop data from the Camera config."""
         if not self.imh.auto_crop:
             trims_data = TrimCropData.from_dict(dict(tdata.to_dict())).to_list()
@@ -100,8 +105,8 @@ class AutoCrop:
 
     async def _init_auto_crop(self):
         """Initialize the auto crop data."""
-        if not self.imh.auto_crop and self.imh.shared.vacuum_state == "docked":
-            self.imh.auto_crop = await self._async_auto_crop_data(self.imh.shared.trims)
+        if not self.imh.auto_crop and self.shared.vacuum_state == "docked":
+            self.imh.auto_crop = await self._async_auto_crop_data(self.shared.trims)
             if self.imh.auto_crop:
                 self.auto_crop_offset()
         else:
@@ -138,8 +143,8 @@ class AutoCrop:
 
         if (
             zoom
-            and self.imh.shared.vacuum_state == "cleaning"
-            and self.imh.shared.image_auto_zoom
+            and self.shared.vacuum_state == "cleaning"
+            and self.shared.image_auto_zoom
         ):
             # Zoom the image based on the robot's position.
             _LOGGER.debug(
@@ -209,10 +214,7 @@ class AutoCrop:
         Automatically crops and trims a numpy array and returns the processed image.
         """
         try:
-            if self.validate_crop_dimensions(self.imh.shared):
-                await self._init_auto_crop()
-            else:
-                return image_array
+            await self._init_auto_crop()
             if self.imh.auto_crop is None:
                 _LOGGER.debug("%s: Calculating auto trim box", self.file_name)
                 # Find the coordinates of the first occurrence of a non-background color
@@ -249,7 +251,7 @@ class AutoCrop:
                     self.imh.trim_right,
                     self.imh.trim_down,
                 ).to_list()
-                # if self.imh.shared.vacuum_state == "docked":
+                # if self.shared.vacuum_state == "docked":
                 #     await (
                 #         self._async_save_auto_crop_data()
                 #     )  # Save the crop data to the disk

valetudo_map_parser/config/colors_man.py

@@ -0,0 +1,248 @@
+"""
+Colors RGBA
+Version: v2025.02.0
+"""
+
+import logging
+
+from .types import (
+    ALPHA_BACKGROUND,
+    ALPHA_CHARGER,
+    ALPHA_GO_TO,
+    ALPHA_MOVE,
+    ALPHA_NO_GO,
+    ALPHA_ROBOT,
+    ALPHA_ROOM_0,
+    ALPHA_ROOM_1,
+    ALPHA_ROOM_2,
+    ALPHA_ROOM_3,
+    ALPHA_ROOM_4,
+    ALPHA_ROOM_5,
+    ALPHA_ROOM_6,
+    ALPHA_ROOM_7,
+    ALPHA_ROOM_8,
+    ALPHA_ROOM_9,
+    ALPHA_ROOM_10,
+    ALPHA_ROOM_11,
+    ALPHA_ROOM_12,
+    ALPHA_ROOM_13,
+    ALPHA_ROOM_14,
+    ALPHA_ROOM_15,
+    ALPHA_TEXT,
+    ALPHA_WALL,
+    ALPHA_ZONE_CLEAN,
+    COLOR_BACKGROUND,
+    COLOR_CHARGER,
+    COLOR_GO_TO,
+    COLOR_MOVE,
+    COLOR_NO_GO,
+    COLOR_ROBOT,
+    COLOR_ROOM_0,
+    COLOR_ROOM_1,
+    COLOR_ROOM_2,
+    COLOR_ROOM_3,
+    COLOR_ROOM_4,
+    COLOR_ROOM_5,
+    COLOR_ROOM_6,
+    COLOR_ROOM_7,
+    COLOR_ROOM_8,
+    COLOR_ROOM_9,
+    COLOR_ROOM_10,
+    COLOR_ROOM_11,
+    COLOR_ROOM_12,
+    COLOR_ROOM_13,
+    COLOR_ROOM_14,
+    COLOR_ROOM_15,
+    COLOR_TEXT,
+    COLOR_WALL,
+    COLOR_ZONE_CLEAN,
+)
+
+color_transparent = (0, 0, 0, 0)
+color_charger = (0, 128, 0, 255)
+color_move = (238, 247, 255, 255)
+color_robot = (255, 255, 204, 255)
+color_no_go = (255, 0, 0, 255)
+color_go_to = (0, 255, 0, 255)
+color_background = (0, 125, 255, 255)
+color_zone_clean = (255, 255, 255, 25)
+color_wall = (255, 255, 0, 255)
+color_text = (255, 255, 255, 255)
+color_grey = (125, 125, 125, 255)
+color_black = (0, 0, 0, 255)
+color_room_0 = (135, 206, 250, 255)
+color_room_1 = (176, 226, 255, 255)
+color_room_2 = (164, 211, 238, 255)
+color_room_3 = (141, 182, 205, 255)
+color_room_4 = (96, 123, 139, 255)
+color_room_5 = (224, 255, 255, 255)
+color_room_6 = (209, 238, 238, 255)
+color_room_7 = (180, 205, 205, 255)
+color_room_8 = (122, 139, 139, 255)
+color_room_9 = (175, 238, 238, 255)
+color_room_10 = (84, 153, 199, 255)
+color_room_11 = (133, 193, 233, 255)
+color_room_12 = (245, 176, 65, 255)
+color_room_13 = (82, 190, 128, 255)
+color_room_14 = (72, 201, 176, 255)
+color_room_15 = (165, 105, 18, 255)
+
+rooms_color = [
+    color_room_0,
+    color_room_1,
+    color_room_2,
+    color_room_3,
+    color_room_4,
+    color_room_5,
+    color_room_6,
+    color_room_7,
+    color_room_8,
+    color_room_9,
+    color_room_10,
+    color_room_11,
+    color_room_12,
+    color_room_13,
+    color_room_14,
+    color_room_15,
+]
+
+base_colors_array = [
+    color_wall,
+    color_zone_clean,
+    color_robot,
+    color_background,
+    color_move,
+    color_charger,
+    color_no_go,
+    color_go_to,
+    color_text,
+]
+
+color_array = [
+    base_colors_array[0],  # color_wall
+    base_colors_array[6],  # color_no_go
+    base_colors_array[7],  # color_go_to
+    color_black,
+    base_colors_array[2],  # color_robot
+    base_colors_array[5],  # color_charger
+    color_text,
+    base_colors_array[4],  # color_move
+    base_colors_array[3],  # color_background
+    base_colors_array[1],  # color_zone_clean
+    color_transparent,
+    rooms_color,
+]
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class ColorsManagment:
+    """Class to manage the colors.
+    Imports and updates the colors from the user configuration."""
+
+    def __init__(self, shared_var):
+        self.shared_var = shared_var
+
+    @staticmethod
+    def add_alpha_to_rgb(alpha_channels, rgb_colors):
+        """
+        Add alpha channel to RGB colors using corresponding alpha channels.
+
+        Args:
+            alpha_channels (List[Optional[float]]): List of alpha channel values (0.0-255.0).
+            rgb_colors (List[Tuple[int, int, int]]): List of RGB colors.
+
+        Returns:
+            List[Tuple[int, int, int, int]]: List of RGBA colors with alpha channel added.
+        """
+        if len(alpha_channels) != len(rgb_colors):
+            _LOGGER.error("Input lists must have the same length.")
+            return []
+
+        result = []
+        for alpha, rgb in zip(alpha_channels, rgb_colors):
+            try:
+                alpha_int = int(alpha)
+                if alpha_int < 0:
+                    alpha_int = 0
+                elif alpha_int > 255:
+                    alpha_int = 255
+
+                if rgb is None:
+                    result.append((0, 0, 0, alpha_int))
+                else:
+                    result.append((rgb[0], rgb[1], rgb[2], alpha_int))
+            except (ValueError, TypeError):
+                result.append(None)
+
+        return result
+
+    def set_initial_colours(self, device_info: dict) -> None:
+        """Set the initial colours for the map."""
+        try:
+            user_colors = [
+                device_info.get(COLOR_WALL, color_wall),
+                device_info.get(COLOR_ZONE_CLEAN, color_zone_clean),
+                device_info.get(COLOR_ROBOT, color_robot),
+                device_info.get(COLOR_BACKGROUND, color_background),
+                device_info.get(COLOR_MOVE, color_move),
+                device_info.get(COLOR_CHARGER, color_charger),
+                device_info.get(COLOR_NO_GO, color_no_go),
+                device_info.get(COLOR_GO_TO, color_go_to),
+                device_info.get(COLOR_TEXT, color_text),
+            ]
+            user_alpha = [
+                device_info.get(ALPHA_WALL, 255),
+                device_info.get(ALPHA_ZONE_CLEAN, 255),
+                device_info.get(ALPHA_ROBOT, 255),
+                device_info.get(ALPHA_BACKGROUND, 255),
+                device_info.get(ALPHA_MOVE, 255),
+                device_info.get(ALPHA_CHARGER, 255),
+                device_info.get(ALPHA_NO_GO, 255),
+                device_info.get(ALPHA_GO_TO, 255),
+                device_info.get(ALPHA_TEXT, 255),
+            ]
+            rooms_colors = [
+                device_info.get(COLOR_ROOM_0, color_room_0),
+                device_info.get(COLOR_ROOM_1, color_room_1),
+                device_info.get(COLOR_ROOM_2, color_room_2),
+                device_info.get(COLOR_ROOM_3, color_room_3),
+                device_info.get(COLOR_ROOM_4, color_room_4),
+                device_info.get(COLOR_ROOM_5, color_room_5),
+                device_info.get(COLOR_ROOM_6, color_room_6),
+                device_info.get(COLOR_ROOM_7, color_room_7),
+                device_info.get(COLOR_ROOM_8, color_room_8),
+                device_info.get(COLOR_ROOM_9, color_room_9),
+                device_info.get(COLOR_ROOM_10, color_room_10),
+                device_info.get(COLOR_ROOM_11, color_room_11),
+                device_info.get(COLOR_ROOM_12, color_room_12),
+                device_info.get(COLOR_ROOM_13, color_room_13),
+                device_info.get(COLOR_ROOM_14, color_room_14),
+                device_info.get(COLOR_ROOM_15, color_room_15),
+            ]
+            rooms_alpha = [
+                device_info.get(ALPHA_ROOM_0, 255),
+                device_info.get(ALPHA_ROOM_1, 255),
+                device_info.get(ALPHA_ROOM_2, 255),
+                device_info.get(ALPHA_ROOM_3, 255),
+                device_info.get(ALPHA_ROOM_4, 255),
+                device_info.get(ALPHA_ROOM_5, 255),
+                device_info.get(ALPHA_ROOM_6, 255),
+                device_info.get(ALPHA_ROOM_7, 255),
+                device_info.get(ALPHA_ROOM_8, 255),
+                device_info.get(ALPHA_ROOM_9, 255),
+                device_info.get(ALPHA_ROOM_10, 255),
+                device_info.get(ALPHA_ROOM_11, 255),
+                device_info.get(ALPHA_ROOM_12, 255),
+                device_info.get(ALPHA_ROOM_13, 255),
+                device_info.get(ALPHA_ROOM_14, 255),
+                device_info.get(ALPHA_ROOM_15, 255),
+            ]
+            self.shared_var.update_user_colors(
+                self.add_alpha_to_rgb(user_alpha, user_colors)
+            )
+            self.shared_var.update_rooms_colors(
+                self.add_alpha_to_rgb(rooms_alpha, rooms_colors)
+            )
+        except (ValueError, IndexError, UnboundLocalError) as e:
+            _LOGGER.error("Error while populating colors: %s", e)

valetudo_map_parser/config/utils.py

@@ -489,7 +489,11 @@ async def async_resize_image(params: ResizeParams):
         if wsf == 0 or hsf == 0 or params.width <= 0 or params.height <= 0:
             _LOGGER.warning(
                 "Invalid aspect ratio parameters: width=%s, height=%s, wsf=%s, hsf=%s. Returning original image.",
-                params.width, params.height, wsf, hsf)
+                params.width,
+                params.height,
+                wsf,
+                hsf,
+            )
             return params.pil_img  # Return original image if invalid
 
         new_aspect_ratio = wsf / hsf

valetudo_map_parser/hypfer_handler.py

@@ -53,7 +53,7 @@ class HypferMapImageHandler(BaseHandler):
         self.offset_left = self.shared.offset_left  # offset left
         self.offset_right = self.shared.offset_right  # offset right
         self.imd = ImDraw(self)
-        self.ac = AutoCrop(self)
+        self.ac = AutoCrop(self, self.shared)
         self.color_grey = (128, 128, 128, 255)
         self.file_name = self.shared.file_name  # file name of the vacuum.
 

valetudo_map_parser/rand25_handler.py

@@ -64,7 +64,7 @@ class ReImageHandler(BaseHandler):
         self.offset_left = self.shared.offset_left  # offset left
         self.offset_right = self.shared.offset_right  # offset right
         self.imd = ImageDraw(self)  # Image Draw
-        self.crop = AutoCrop(self)
+        self.crop = AutoCrop(self, self.shared)
 
     async def extract_room_properties(
         self, json_data: JsonType, destinations: JsonType
@@ -260,7 +260,9 @@ class ReImageHandler(BaseHandler):
 
     async def _finalize_image(self, pil_img):
         if not self.shared.image_ref_width or not self.shared.image_ref_height:
-            _LOGGER.warning("Image finalization failed: Invalid image dimensions. Returning original image.")
+            _LOGGER.warning(
+                "Image finalization failed: Invalid image dimensions. Returning original image."
+            )
             return pil_img
         if self.check_zoom_and_aspect_ratio():
             resize_params = prepare_resize_params(self, pil_img, True)

{valetudo_map_parser-0.1.9b27.dist-info → valetudo_map_parser-0.1.9b29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: valetudo-map-parser
-Version: 0.1.9b27
+Version: 0.1.9b29
 Summary: A Python library to parse Valetudo map data returning a PIL Image object.
 License: Apache-2.0
 Author: Sandro Cantarella

{valetudo_map_parser-0.1.9b27.dist-info → valetudo_map_parser-0.1.9b29.dist-info}/RECORD

@@ -1,20 +1,21 @@
 valetudo_map_parser/__init__.py,sha256=Wmd20bdI1btzMq-0x8NxGYWskTjdUmD-Fem9MTfziwU,810
 valetudo_map_parser/config/__init__.py,sha256=DQ9plV3ZF_K25Dp5ZQHPDoG-40dQoJNdNi-dfNeR3Zc,48
-valetudo_map_parser/config/auto_crop.py,sha256=J_1f4W_V1_kbALkWNd2VJosnjjSk5YR3IvHtDGmVGx0,10894
+valetudo_map_parser/config/auto_crop.py,sha256=qtP_qqovnsC0442FNyizferkZinyXeSrBvRtuBznYz0,10813
 valetudo_map_parser/config/colors.py,sha256=IzTT9JvF12YGGJxaTiEJRuwUdCCsFCLzsR9seCDfYWs,6515
+valetudo_map_parser/config/colors_man.py,sha256=-n2_UVvN3Swm9vxCuHe6ajDGA9XUDEPcp3UubdUY5-4,7832
 valetudo_map_parser/config/drawable.py,sha256=hsrEJCMVOrjs5sJfr26SeqJD0VNlYWwxcVkkHeaxx7U,20356
 valetudo_map_parser/config/rand25_parser.py,sha256=kIayyqVZBfQfAMkiArzqrrj9vqZB3pkgT0Y5ufrQmGA,16448
 valetudo_map_parser/config/shared.py,sha256=jk7x8xCiE0UnE1oXcZ4iIBGz1Mv0CTHQOeZN2K94eXA,9743
 valetudo_map_parser/config/types.py,sha256=wcWtYAc5sc9CWYzRJ4aOJRmuvM2rMuCfcDgAhpV8yEM,17144
-valetudo_map_parser/config/utils.py,sha256=5P33Ren99EUwF7vfQ-GhSeCiR6jI1ZtiKQaofbgIKp8,18914
+valetudo_map_parser/config/utils.py,sha256=LExxpOesfaAeskusEfWU4nrUR068_1yM_U_1JVHYQW0,18976
 valetudo_map_parser/hypfer_draw.py,sha256=1trtil-CQcDSiAMBWPBmuP5L9MWHGTp5OlY7MX8FgDg,14932
-valetudo_map_parser/hypfer_handler.py,sha256=ibW2DQSWxCV2PHA6FwfIOIBDu6hryCs7ae6HD-Nhm6A,13635
+valetudo_map_parser/hypfer_handler.py,sha256=g8y_hMuR1t4ILsWC5a6afGz-DTPyWbDUYP8BksX8-NQ,13648
 valetudo_map_parser/map_data.py,sha256=6FbQfgxFB6E4kcOWokReJOVSekVaE1kStyhTQhAhiOg,19469
 valetudo_map_parser/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-valetudo_map_parser/rand25_handler.py,sha256=nFOX1yVvZrAYxfY-aij7oIz0nKcUUL28y9_gsNJWWLw,15746
+valetudo_map_parser/rand25_handler.py,sha256=rYcXJX3wAwReTvvE3EzTGlbqAdEWvVRDnJgh47398W4,15789
 valetudo_map_parser/reimg_draw.py,sha256=V0JUASavKVnEtAhv7nOV4pjsRxZrNsjIUtctbKO8wvk,12507
-valetudo_map_parser-0.1.9b27.dist-info/LICENSE,sha256=Lh-qBbuRV0-jiCIBhfV7NgdwFxQFOXH3BKOzK865hRs,10480
-valetudo_map_parser-0.1.9b27.dist-info/METADATA,sha256=SzlOLmV3VGgFRnaCjyyqrt2Cn7CIMin6rGGuwt9huyA,1029
-valetudo_map_parser-0.1.9b27.dist-info/NOTICE.txt,sha256=5lTOuWiU9aiEnJ2go8sc7lTJ7ntMBx0g0GFnNrswCY4,2533
-valetudo_map_parser-0.1.9b27.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-valetudo_map_parser-0.1.9b27.dist-info/RECORD,,
+valetudo_map_parser-0.1.9b29.dist-info/LICENSE,sha256=Lh-qBbuRV0-jiCIBhfV7NgdwFxQFOXH3BKOzK865hRs,10480
+valetudo_map_parser-0.1.9b29.dist-info/METADATA,sha256=vScCR5abIYaNxqRenIkS5irR70yNEkJd72BcPuxjuTY,1029
+valetudo_map_parser-0.1.9b29.dist-info/NOTICE.txt,sha256=5lTOuWiU9aiEnJ2go8sc7lTJ7ntMBx0g0GFnNrswCY4,2533
+valetudo_map_parser-0.1.9b29.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+valetudo_map_parser-0.1.9b29.dist-info/RECORD,,