valetudo-map-parser 0.1.9b22__py3-none-any.whl → 0.1.9b23__py3-none-any.whl

valetudo_map_parser/__init__.py

@@ -16,6 +16,7 @@ from .config.types import (
 from .hypfer_handler import HypferMapImageHandler
 from .rand25_handler import ReImageHandler
 
+
 __all__ = [
     "HypferMapImageHandler",
     "ReImageHandler",

valetudo_map_parser/config/auto_crop.py

@@ -10,6 +10,7 @@ from numpy import rot90
 
 from .types import Color, NumpyArray, TrimCropData
 
+
 _LOGGER = logging.getLogger(__name__)
 
 

valetudo_map_parser/config/colors.py

@@ -6,6 +6,7 @@ import logging
 from enum import StrEnum
 from typing import Dict, List, Tuple
 
+
 _LOGGER = logging.getLogger(__name__)
 
 Color = Tuple[int, int, int, int]  # RGBA color definition

valetudo_map_parser/config/rand25_parser.py

@@ -10,6 +10,7 @@ import struct
 from enum import Enum
 from typing import Any, Callable, Dict, List, Optional, TypeVar
 
+
 _CallableT = TypeVar("_CallableT", bound=Callable[..., Any])
 
 

valetudo_map_parser/config/shared.py

@@ -6,6 +6,7 @@ Version: v2024.12.0
 
 import asyncio
 import logging
+from typing import Dict, List
 
 from .types import (
     ATTR_CALIBRATION_POINTS,
@@ -36,8 +37,10 @@ from .types import (
     DEFAULT_VALUES,
     CameraModes,
     Colors,
+    TrimsData,
 )
 
+
 _LOGGER = logging.getLogger(__name__)
 
 
@@ -99,8 +102,15 @@ class CameraShared:
         self.map_pred_points = None  # Predefined points data
         self.map_new_path = None  # New path data
         self.map_old_path = None  # Old path data
-        self.trim_crop_data = None
         self.user_language = None  # User language
+        self.trim_crop_data = None
+        self.trims: Dict[TrimsData, int] = {
+            TrimsData.TRIM_LEFT: 0,
+            TrimsData.TRIM_UP: 0,
+            TrimsData.TRIM_RIGHT: 0,
+            TrimsData.TRIM_DOWN: 0,
+        }
+        self.skip_room_ids: List[str] = []
 
     def update_user_colors(self, user_colors):
         """Update the user colors."""

valetudo_map_parser/config/types.py

@@ -7,11 +7,13 @@ import asyncio
 import json
 import logging
 from dataclasses import dataclass
+from enum import Enum
 from typing import Any, Dict, Tuple, Union
 
 import numpy as np
 from PIL import Image
 
+
 DEFAULT_ROOMS = 1
 
 MY_LOGGER = logging.getLogger(__name__)
@@ -587,3 +589,12 @@ class CameraModes:
     CAMERA_STANDBY = "camera_standby"
     CAMERA_OFF = False
     CAMERA_ON = True
+
+
+class TrimsData(Enum):
+    """Constants for the trims data."""
+
+    TRIM_LEFT = "trim_left"
+    TRIM_UP = "trim_up"
+    TRIM_RIGHT = "trim_right"
+    TRIM_DOWN = "trim_down"

valetudo_map_parser/config/utils.py

@@ -4,10 +4,12 @@ import hashlib
 import json
 from dataclasses import dataclass
 from logging import getLogger
+from typing import Callable, List, Optional
 
-from PIL import Image, ImageOps
+from PIL import ImageOps
+
+from .types import ChargerPosition, ImageSize, NumpyArray, PilPNG, RobotPosition
 
-from .types import ChargerPosition, ImageSize, NumpyArray, RobotPosition
 
 _LOGGER = getLogger(__name__)
 
@@ -16,13 +18,13 @@ _LOGGER = getLogger(__name__)
 class ResizeParams:
     """Resize the image to the given dimensions and aspect ratio."""
 
-    pil_img: Image  # PIL image object
+    pil_img: PilPNG
     width: int
     height: int
-    aspect_ratio: str = None
-    crop_size: list = None
-    is_rand: bool = False
-    offset_func: callable = None  # Function reference for offset calculation
+    aspect_ratio: str
+    crop_size: List[int]
+    is_rand: Optional[bool] = False
+    offset_func: Optional[Callable] = None
 
 
 @dataclass
@@ -33,7 +35,7 @@ class OffsetParams:
     hsf: int
     width: int
     height: int
-    rand256: bool = False
+    rand256: Optional[bool] = False
 
 
 class BaseHandler:
@@ -91,7 +93,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_1_1(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 1:1."""
 
@@ -118,7 +120,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_2_1(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 2:1."""
 
@@ -146,7 +148,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_3_2(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 3:2."""
 
@@ -177,7 +179,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_5_4(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 5:4."""
 
@@ -209,7 +211,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_9_16(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 9:16."""
 
@@ -237,7 +239,7 @@ class BaseHandler:
         )
 
     def _set_image_offset_ratio_16_9(
-        self, width: int, height: int, rand256: bool = False
+        self, width: int, height: int, rand256: Optional[bool] = False
     ) -> None:
         """Set the image offset ratio to 16:9."""
 
@@ -298,8 +300,8 @@ class BaseHandler:
 
     @staticmethod
     async def calculate_array_hash(
-        layers: dict, active: list[int] = None
-    ) -> str or None:
+        layers: dict, active: Optional[List[int]]
+    ) -> str | None:
         """Calculate the hash of the image based on layers and active zones."""
         if layers and active:
             data_to_hash = {
@@ -499,7 +501,7 @@ async def async_resize_image(params: ResizeParams):
             str(hsf),
         )
 
-        if params.crop_size is not None:
+        if (params.crop_size is not None) and (params.offset_func is not None):
             offset = OffsetParams(wsf, hsf, new_width, new_height, params.is_rand)
             params.crop_size[0], params.crop_size[1] = await params.offset_func(offset)
 
@@ -507,6 +509,7 @@ async def async_resize_image(params: ResizeParams):
 
     return ImageOps.pad(params.pil_img, (params.width, params.height))
 
+
 def prepare_resize_params(handler, pil_img, rand):
     """Prepare resize parameters for image resizing."""
     return ResizeParams(
@@ -516,5 +519,5 @@ def prepare_resize_params(handler, pil_img, rand):
         aspect_ratio=handler.shared.image_aspect_ratio,
         crop_size=handler.crop_img_size,
         offset_func=handler.async_map_coordinates_offset,
-        is_rand=rand
+        is_rand=rand,
     )

valetudo_map_parser/hypfer_draw.py

@@ -10,6 +10,7 @@ import logging
 
 from .config.types import Color, JsonType, NumpyArray, RobotPosition
 
+
 _LOGGER = logging.getLogger(__name__)
 
 

valetudo_map_parser/hypfer_handler.py

@@ -20,6 +20,7 @@ from .config.utils import BaseHandler, prepare_resize_params
 from .hypfer_draw import ImageDraw as ImDraw
 from .map_data import ImageData
 
+
 _LOGGER = logging.getLogger(__name__)
 
 

valetudo_map_parser/rand25_handler.py

@@ -28,6 +28,7 @@ from .config.utils import BaseHandler, prepare_resize_params
 from .map_data import RandImageData
 from .reimg_draw import ImageDraw
 
+
 _LOGGER = logging.getLogger(__name__)
 
 

valetudo_map_parser/reimg_draw.py

@@ -12,6 +12,7 @@ from .config.drawable import Drawable
 from .config.types import Color, JsonType, NumpyArray
 from .map_data import ImageData, RandImageData
 
+
 _LOGGER = logging.getLogger(__name__)
 
 

{valetudo_map_parser-0.1.9b22.dist-info → valetudo_map_parser-0.1.9b23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: valetudo-map-parser
-Version: 0.1.9b22
+Version: 0.1.9b23
 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.9b23.dist-info/RECORD

@@ -0,0 +1,20 @@
+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=OCEUxmqaCy4jbuIkXFuutJ2W3usarfowEJBXKak-1nw,11019
+valetudo_map_parser/config/colors.py,sha256=IzTT9JvF12YGGJxaTiEJRuwUdCCsFCLzsR9seCDfYWs,6515
+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=BPfkcsfZnWqiwCfTECyXw-8LX9NFc6SuwiYU99cbWtE,9735
+valetudo_map_parser/config/types.py,sha256=VWzrR0xzlcTBdL9Eu6hJLpVjO0YusQomR1vOIj1g2a0,16306
+valetudo_map_parser/config/utils.py,sha256=8TBqH5jKr7wMM5dtWqnk5pPwZa_D0ApOpmdZUaqiRl8,18679
+valetudo_map_parser/hypfer_draw.py,sha256=1trtil-CQcDSiAMBWPBmuP5L9MWHGTp5OlY7MX8FgDg,14932
+valetudo_map_parser/hypfer_handler.py,sha256=dW3EKF-Hc8OT74IcUXk8faPx4CvhkL3WH79tq_bLCfw,13516
+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=N0i4JBnX5UY-TMpRa11oXE6DrjIU3fqSM89aaKVP0Eo,15318
+valetudo_map_parser/reimg_draw.py,sha256=V0JUASavKVnEtAhv7nOV4pjsRxZrNsjIUtctbKO8wvk,12507
+valetudo_map_parser-0.1.9b23.dist-info/LICENSE,sha256=Lh-qBbuRV0-jiCIBhfV7NgdwFxQFOXH3BKOzK865hRs,10480
+valetudo_map_parser-0.1.9b23.dist-info/METADATA,sha256=xTp6WcfWrNFVtRMW58-WAZZqAd2_VSbH-sJfG4Rjyhc,1029
+valetudo_map_parser-0.1.9b23.dist-info/NOTICE.txt,sha256=5lTOuWiU9aiEnJ2go8sc7lTJ7ntMBx0g0GFnNrswCY4,2533
+valetudo_map_parser-0.1.9b23.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+valetudo_map_parser-0.1.9b23.dist-info/RECORD,,

valetudo_map_parser-0.1.9b22.dist-info/RECORD

@@ -1,20 +0,0 @@
-valetudo_map_parser/__init__.py,sha256=pWRSVGM2OtPKTzdpPTEgLrNYjZ9P2duwaDLI_KN-iLY,809
-valetudo_map_parser/config/__init__.py,sha256=DQ9plV3ZF_K25Dp5ZQHPDoG-40dQoJNdNi-dfNeR3Zc,48
-valetudo_map_parser/config/auto_crop.py,sha256=z97Yqoe96ZiWmgaKTFv7sODb9UD6doRduTeQXMYWXVA,11018
-valetudo_map_parser/config/colors.py,sha256=0sTI_x0ZMchSDPTsEivdTXDn58nEbyE4REXIp0IpWdg,6514
-valetudo_map_parser/config/drawable.py,sha256=hsrEJCMVOrjs5sJfr26SeqJD0VNlYWwxcVkkHeaxx7U,20356
-valetudo_map_parser/config/rand25_parser.py,sha256=fehyF18hRWRWbXbojocQCIaIch21Lbh1wtl2XdKRSl0,16447
-valetudo_map_parser/config/shared.py,sha256=LQV5K8tbVhEKUkby9ssjEmh_T4Ai-Euzsbag_HWYVRc,9448
-valetudo_map_parser/config/types.py,sha256=sdjqhcxpRSxNFLVHIyCg-RDL9JIjtQHoR2C2WpocWBc,16107
-valetudo_map_parser/config/utils.py,sha256=QHF0rx72TS0Kpt2iTqqFwr6MlV0TJXK6-_008wDg8nU,18573
-valetudo_map_parser/hypfer_draw.py,sha256=s58ak9IBYLjJyoddfDC99PfQ12HTjkSfJYXqCi4vZKs,14931
-valetudo_map_parser/hypfer_handler.py,sha256=8uAPum9JK_JDDSEoD_gLGxMRvdOg4oAABqc3PvW-GTs,13515
-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=iGWxNgLzl-4btakBGKEbLqKOp7j2gDD6bmSAp4YGjkY,15317
-valetudo_map_parser/reimg_draw.py,sha256=dtdbYKKxmQnbOaHBHayWEF07OdSnTKo2CPSOW0qpgH0,12506
-valetudo_map_parser-0.1.9b22.dist-info/LICENSE,sha256=Lh-qBbuRV0-jiCIBhfV7NgdwFxQFOXH3BKOzK865hRs,10480
-valetudo_map_parser-0.1.9b22.dist-info/METADATA,sha256=B6EqNlL0I8Pmt-hy_nvT8HA9EdUTOb-_-X6dDAmowpo,1029
-valetudo_map_parser-0.1.9b22.dist-info/NOTICE.txt,sha256=5lTOuWiU9aiEnJ2go8sc7lTJ7ntMBx0g0GFnNrswCY4,2533
-valetudo_map_parser-0.1.9b22.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-valetudo_map_parser-0.1.9b22.dist-info/RECORD,,