valetudo-map-parser 0.1.9b41__py3-none-any.whl → 0.1.9b43__py3-none-any.whl

valetudo_map_parser/config/utils.py

@@ -2,9 +2,11 @@
 
 import hashlib
 import json
+import logging
+import numpy as np
 from dataclasses import dataclass
 from logging import getLogger
-from typing import Callable, List, Optional
+from typing import Callable, Dict, List, Optional, Tuple, Union
 
 from PIL import ImageOps
 
@@ -530,3 +532,410 @@ def prepare_resize_params(handler, pil_img, rand):
         offset_func=handler.async_map_coordinates_offset,
         is_rand=rand,
     )
+
+
+def initialize_drawing_config(handler):
+    """
+    Initialize drawing configuration from device_info.
+
+    Args:
+        handler: The handler instance with shared data and file_name attributes
+
+    Returns:
+        Tuple of (DrawingConfig, Drawable, EnhancedDrawable)
+    """
+    from .drawable import Drawable
+    from .drawable_elements import DrawableElement, DrawingConfig
+    from .enhanced_drawable import EnhancedDrawable
+
+    # Initialize drawing configuration
+    drawing_config = DrawingConfig()
+
+    # Get logger from the handler
+    _LOGGER = logging.getLogger(handler.__class__.__module__)
+
+    if hasattr(handler.shared, "device_info") and handler.shared.device_info is not None:
+        _LOGGER.info(
+            "%s: Initializing drawing config from device_info", handler.file_name
+        )
+        _LOGGER.info(
+            "%s: device_info contains disable_obstacles: %s",
+            handler.file_name,
+            "disable_obstacles" in handler.shared.device_info,
+        )
+        _LOGGER.info(
+            "%s: device_info contains disable_path: %s",
+            handler.file_name,
+            "disable_path" in handler.shared.device_info,
+        )
+        _LOGGER.info(
+            "%s: device_info contains disable_elements: %s",
+            handler.file_name,
+            "disable_elements" in handler.shared.device_info,
+        )
+
+        if "disable_obstacles" in handler.shared.device_info:
+            _LOGGER.info(
+                "%s: disable_obstacles value: %s",
+                handler.file_name,
+                handler.shared.device_info["disable_obstacles"],
+            )
+        if "disable_path" in handler.shared.device_info:
+            _LOGGER.info(
+                "%s: disable_path value: %s",
+                handler.file_name,
+                handler.shared.device_info["disable_path"],
+            )
+        if "disable_elements" in handler.shared.device_info:
+            _LOGGER.info(
+                "%s: disable_elements value: %s",
+                handler.file_name,
+                handler.shared.device_info["disable_elements"],
+            )
+
+        drawing_config.update_from_device_info(handler.shared.device_info)
+
+        # Verify elements are disabled
+        _LOGGER.info(
+            "%s: After initialization, PATH enabled: %s",
+            handler.file_name,
+            drawing_config.is_enabled(DrawableElement.PATH),
+        )
+        _LOGGER.info(
+            "%s: After initialization, OBSTACLE enabled: %s",
+            handler.file_name,
+            drawing_config.is_enabled(DrawableElement.OBSTACLE),
+        )
+
+    # Initialize both drawable systems for backward compatibility
+    draw = Drawable()  # Legacy drawing utilities
+    enhanced_draw = EnhancedDrawable(drawing_config)  # New enhanced drawing system
+
+    return drawing_config, draw, enhanced_draw
+
+
+def blend_colors(base_color, overlay_color):
+    """
+    Blend two RGBA colors using alpha compositing.
+
+    Args:
+        base_color: Base RGBA color tuple (r, g, b, a)
+        overlay_color: Overlay RGBA color tuple (r, g, b, a)
+
+    Returns:
+        Blended RGBA color tuple (r, g, b, a)
+    """
+    r1, g1, b1, a1 = base_color
+    r2, g2, b2, a2 = overlay_color
+
+    # Convert alpha to 0-1 range
+    a1 = a1 / 255.0
+    a2 = a2 / 255.0
+
+    # Calculate resulting alpha
+    a_out = a1 + a2 * (1 - a1)
+
+    # Avoid division by zero
+    if a_out < 0.0001:
+        return (0, 0, 0, 0)
+
+    # Calculate blended RGB components
+    r_out = (r1 * a1 + r2 * a2 * (1 - a1)) / a_out
+    g_out = (g1 * a1 + g2 * a2 * (1 - a1)) / a_out
+    b_out = (b1 * a1 + b2 * a2 * (1 - a1)) / a_out
+
+    # Convert back to 0-255 range and return as tuple
+    return (
+        int(max(0, min(255, r_out))),
+        int(max(0, min(255, g_out))),
+        int(max(0, min(255, b_out))),
+        int(max(0, min(255, a_out * 255))),
+    )
+
+
+def blend_pixel(array, x, y, color, element, element_map=None, drawing_config=None):
+    """
+    Blend a pixel color with the existing color at the specified position.
+    Also updates the element map if the new element has higher z-index.
+
+    Args:
+        array: The image array to modify
+        x: X coordinate
+        y: Y coordinate
+        color: RGBA color tuple to blend
+        element: Element code for the pixel
+        element_map: Optional element map to update
+        drawing_config: Optional drawing configuration for z-index lookup
+
+    Returns:
+        None
+    """
+    # Check bounds
+    if not (0 <= y < array.shape[0] and 0 <= x < array.shape[1]):
+        return
+
+    # Get current element at this position
+    current_element = None
+    if element_map is not None:
+        current_element = element_map[y, x]
+
+    # Get z-index values for comparison
+    current_z = 0
+    new_z = 0
+
+    if drawing_config is not None:
+        current_z = (
+            drawing_config.get_property(current_element, "z_index", 0)
+            if current_element
+            else 0
+        )
+        new_z = drawing_config.get_property(element, "z_index", 0)
+
+    # Update element map if new element has higher z-index
+    if element_map is not None and new_z >= current_z:
+        element_map[y, x] = element
+
+    # Blend colors
+    base_color = array[y, x]
+    blended_color = blend_colors(base_color, color)
+    array[y, x] = blended_color
+
+
+def get_element_at_position(element_map, x, y):
+    """
+    Get the element code at a specific position in the element map.
+
+    Args:
+        element_map: The element map array
+        x: X coordinate
+        y: Y coordinate
+
+    Returns:
+        Element code or None if out of bounds
+    """
+    if element_map is not None:
+        if 0 <= y < element_map.shape[0] and 0 <= x < element_map.shape[1]:
+            return element_map[y, x]
+    return None
+
+
+def get_room_at_position(element_map, x, y, room_base=101):
+    """
+    Get the room ID at a specific position, or None if not a room.
+
+    Args:
+        element_map: The element map array
+        x: X coordinate
+        y: Y coordinate
+        room_base: Base value for room elements (default: 101 for DrawableElement.ROOM_1)
+
+    Returns:
+        Room ID (1-15) or None if not a room
+    """
+    element = get_element_at_position(element_map, x, y)
+    if element is not None and room_base <= element <= room_base + 14:  # 15 rooms max
+        return element - room_base + 1
+    return None
+
+
+def update_element_map_with_robot(element_map, robot_position, robot_element=3, robot_radius=25):
+    """
+    Update the element map with the robot position.
+
+    Args:
+        element_map: The element map to update
+        robot_position: Tuple of (x, y) coordinates for the robot
+        robot_element: Element code for the robot (default: 3 for DrawableElement.ROBOT)
+        robot_radius: Radius of the robot in pixels
+
+    Returns:
+        None
+    """
+    if element_map is None or robot_position is None:
+        return
+
+    # Update element map for robot position
+    for dy in range(-robot_radius, robot_radius + 1):
+        for dx in range(-robot_radius, robot_radius + 1):
+            if dx * dx + dy * dy <= robot_radius * robot_radius:
+                rx, ry = (
+                    int(robot_position[0] + dx),
+                    int(robot_position[1] + dy),
+                )
+                if (
+                    0 <= ry < element_map.shape[0]
+                    and 0 <= rx < element_map.shape[1]
+                ):
+                    element_map[ry, rx] = robot_element
+
+
+def manage_drawable_elements(handler, action, element_code=None, element_codes=None, property_name=None, value=None):
+    """
+    Manage drawable elements (enable, disable, set elements, set properties).
+
+    Args:
+        handler: The handler instance with drawing_config attribute
+        action: Action to perform ('enable', 'disable', 'set_elements', 'set_property')
+        element_code: Element code for enable/disable/set_property actions
+        element_codes: List of element codes for set_elements action
+        property_name: Property name for set_property action
+        value: Property value for set_property action
+
+    Returns:
+        None
+    """
+    if not hasattr(handler, "drawing_config") or handler.drawing_config is None:
+        return
+
+    if action == "enable" and element_code is not None:
+        handler.drawing_config.enable_element(element_code)
+    elif action == "disable" and element_code is not None:
+        handler.drawing_config.disable_element(element_code)
+    elif action == "set_elements" and element_codes is not None:
+        handler.drawing_config.set_elements(element_codes)
+    elif action == "set_property" and element_code is not None and property_name is not None:
+        handler.drawing_config.set_property(element_code, property_name, value)
+
+
+def handle_room_outline_error(file_name, room_id, error, logger=None):
+    """
+    Handle errors during room outline extraction.
+
+    Args:
+        file_name: Name of the file for logging
+        room_id: Room ID for logging
+        error: The error that occurred
+        logger: Logger instance (optional)
+
+    Returns:
+        None
+    """
+    _LOGGER = logger or logging.getLogger(__name__)
+
+    _LOGGER.warning(
+        "%s: Failed to trace outline for room %s: %s",
+        file_name, str(room_id), str(error)
+    )
+
+
+async def async_extract_room_outline(room_mask, min_x, min_y, max_x, max_y, file_name, room_id_int, logger=None):
+    """
+    Extract the outline of a room from a binary mask.
+
+    Args:
+        room_mask: Binary mask where room pixels are 1 and non-room pixels are 0
+        min_x: Minimum x coordinate of the room
+        min_y: Minimum y coordinate of the room
+        max_x: Maximum x coordinate of the room
+        max_y: Maximum y coordinate of the room
+        file_name: Name of the file for logging
+        room_id_int: Room ID for logging
+        logger: Logger instance (optional)
+
+    Returns:
+        List of (x, y) points forming the room outline
+    """
+    # Use the provided logger or create a new one
+    _LOGGER = logger or logging.getLogger(__name__)
+
+    # Get the dimensions of the mask
+    height, width = room_mask.shape
+
+    # Find the coordinates of all room pixels
+    room_y, room_x = np.where(room_mask > 0)
+    if len(room_y) == 0:
+        return [(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)]
+
+    # Get the bounding box of the room
+    min_y, max_y = np.min(room_y), np.max(room_y)
+    min_x, max_x = np.min(room_x), np.max(room_x)
+
+    # For simple rooms, just use the rectangular outline
+    rect_outline = [
+        (min_x, min_y),  # Top-left
+        (max_x, min_y),  # Top-right
+        (max_x, max_y),  # Bottom-right
+        (min_x, max_y),  # Bottom-left
+    ]
+
+    # For more complex room shapes, trace the boundary
+    # This is a custom boundary tracing algorithm that works without OpenCV
+    try:
+        # Create a padded mask to handle edge cases
+        padded_mask = np.zeros((height + 2, width + 2), dtype=np.uint8)
+        padded_mask[1:-1, 1:-1] = room_mask
+
+        # Find boundary pixels (pixels that have at least one non-room neighbor)
+        boundary_points = []
+
+        # More efficient boundary detection - only check pixels that are part of the room
+        for y, x in zip(room_y, room_x):
+            # Check if this is a boundary pixel (at least one neighbor is 0)
+            is_boundary = False
+            for dy, dx in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
+                ny, nx = y + dy, x + dx
+                if (ny < 0 or ny >= height or nx < 0 or nx >= width or
+                    room_mask[ny, nx] == 0):
+                    is_boundary = True
+                    break
+            if is_boundary:
+                boundary_points.append((x, y))
+
+        # Log the number of boundary points found
+        _LOGGER.debug(
+            "%s: Room %s has %d boundary points",
+            file_name, str(room_id_int), len(boundary_points)
+        )
+
+        # If we found too few boundary points, use the rectangular outline
+        if len(boundary_points) < 8:  # Need at least 8 points for a meaningful shape
+            _LOGGER.debug(
+                "%s: Room %s has too few boundary points (%d), using rectangular outline",
+                file_name, str(room_id_int), len(boundary_points)
+            )
+            return rect_outline
+
+        # Use a more sophisticated algorithm to create a coherent outline
+        # We'll use a convex hull approach to get the main shape
+        # Sort points by angle from centroid
+        centroid_x = np.mean([p[0] for p in boundary_points])
+        centroid_y = np.mean([p[1] for p in boundary_points])
+
+        # Calculate angles from centroid
+        def calculate_angle(point):
+            return np.arctan2(point[1] - centroid_y, point[0] - centroid_x)
+
+        # Sort boundary points by angle
+        boundary_points.sort(key=calculate_angle)
+
+        # Simplify the outline if it has too many points
+        if len(boundary_points) > 20:
+            # Take every Nth point to simplify
+            step = len(boundary_points) // 20
+            simplified_outline = [boundary_points[i] for i in range(0, len(boundary_points), step)]
+            # Make sure we have at least 8 points
+            if len(simplified_outline) < 8:
+                simplified_outline = boundary_points[::len(boundary_points)//8]
+        else:
+            simplified_outline = boundary_points
+
+        # Make sure to close the loop
+        if simplified_outline[0] != simplified_outline[-1]:
+            simplified_outline.append(simplified_outline[0])
+
+        # Convert NumPy int64 values to regular Python integers
+        simplified_outline = [(int(x), int(y)) for x, y in simplified_outline]
+
+        _LOGGER.debug(
+            "%s: Room %s outline has %d points",
+            file_name, str(room_id_int), len(simplified_outline)
+        )
+
+        return simplified_outline
+
+    except (ValueError, IndexError, TypeError, ArithmeticError) as e:
+        _LOGGER.warning(
+            "%s: Error tracing room outline: %s. Using rectangular outline instead.",
+            file_name, str(e)
+        )
+        return rect_outline