# geoelevation/map_viewer/map_manager.py """ Manages the retrieval, caching, and stitching of map tiles from a selected map service. This module interacts with a map service provider (implementing BaseMapService), handles local disk caching of tiles to support offline use and reduce network requests, and assembles individual tiles into a complete map image. """ # Standard library imports import logging import os import time import threading from pathlib import Path # For robust path manipulation from typing import Tuple, Optional, List, Dict # Ensure these are available import io # To handle byte stream from requests as an image import shutil # For cache clearing operations # Third-party imports try: import requests # For downloading map tiles REQUESTS_AVAILABLE = True except ImportError: requests = None # type: ignore REQUESTS_AVAILABLE = False logging.error("MapTileManager: 'requests' library not found. Online tile fetching will fail.") try: from PIL import Image, ImageDraw ImageType = Image.Image # type: ignore PIL_AVAILABLE_MANAGER = True except ImportError: Image = None # type: ignore ImageDraw = None # type: ignore ImageType = None # type: ignore PIL_AVAILABLE_MANAGER = False logging.error("MapTileManager: 'Pillow' library not found. Image operations will fail.") try: # MODIFIED: Import mercantile here. # WHY: map_manager uses mercantile directly in _get_bounds_for_tile_range. # HOW: Added import. import mercantile except ImportError: mercantile = None # type: ignore # No module-specific flag needed here, just check `if mercantile is None:` where used. logging.warning("MapTileManager: 'mercantile' library not found. Tile bounds calculation will fail.") # Local application/package imports from .map_services import BaseMapService try: from ..utils.logger import get_logger logger = get_logger(__name__) except ImportError: logger = logging.getLogger(__name__) logger.warning("MapTileManager: Could not import application logger. Using standard logging.") # Default values for the manager if not provided or configured DEFAULT_MAP_TILE_CACHE_ROOT_DIR = "map_tile_cache" DEFAULT_ENABLE_ONLINE_FETCHING = True DEFAULT_NETWORK_TIMEOUT_SECONDS = 10 DEFAULT_DOWNLOAD_RETRY_DELAY_SECONDS = 2 DEFAULT_MAX_DOWNLOAD_RETRIES = 2 DEFAULT_PLACEHOLDER_COLOR_RGB = (220, 220, 220) class MapTileManager: """ Manages fetching, caching, and assembling map tiles for a given map service. Requires 'requests' and 'Pillow' libraries to be installed. 'mercantile' is required for calculating geographic bounds of tile ranges. """ def __init__( self, map_service: BaseMapService, cache_root_directory: Optional[str] = None, enable_online_tile_fetching: Optional[bool] = None, tile_pixel_size: Optional[int] = None ) -> None: """ Initializes the MapTileManager. Args: map_service_instance: An instance of a map service provider (e.g., OpenStreetMapService). cache_root_directory: The root directory for caching tiles from all services. If None, uses DEFAULT_MAP_TILE_CACHE_ROOT_DIR. A subdirectory for the specific service will be created. enable_online_tile_fetching: Whether to download tiles if not found in cache. If None, uses DEFAULT_ENABLE_ONLINE_FETCHING. tile_pixel_size: The pixel dimension (width/height) of map tiles for this manager. If None, the size is taken from the map_service instance. Raises: TypeError: If map_service_instance is not a valid BaseMapService instance. ImportError: If 'requests' or 'Pillow' libraries are not installed. ValueError: If a tile_pixel_size is provided but invalid. """ logger.info("Initializing MapTileManager...") if not REQUESTS_AVAILABLE: raise ImportError("'requests' library is required by MapTileManager but not found.") if not (PIL_AVAILABLE_MANAGER and ImageDraw is not None): raise ImportError("'Pillow' library (or its drawing module ImageDraw) is required by MapTileManager but not fully available.") # MODIFIED: Add a check for mercantile availability here, as it's used later. # WHY: Inform the user early if a required dependency is missing. # HOW: Added if mercantile is None check. if mercantile is None: logger.error("'mercantile' library is required for tile bounds calculation but is not found or is None.") # Decide if this is a fatal error for the manager init. # The manager can still fetch/stitch if mercantile is missing, but cannot provide bounds. # Let's log an error but not raise, allowing partial functionality. if not isinstance(map_service, BaseMapService): logger.critical("Invalid map_service_instance provided. Must be an instance of BaseMapService.") raise TypeError("map_service_instance must be an instance of BaseMapService.") self.map_service: BaseMapService = map_service self.service_identifier_name: str = self.map_service.name if tile_pixel_size is not None: if not isinstance(tile_pixel_size, int) or tile_pixel_size <= 0: logger.error(f"Invalid provided tile_pixel_size: {tile_pixel_size}. Must be a positive integer.") raise ValueError(f"Invalid provided tile_pixel_size: {tile_pixel_size}. Must be a positive integer.") self.tile_size: int = tile_pixel_size logger.info(f"Map tile size explicitly set to {self.tile_size}px.") else: self.tile_size: int = self.map_service.tile_size logger.info(f"Map tile size inherited from service '{self.map_service.name}': {self.tile_size}px.") effective_cache_root_dir = cache_root_directory if cache_root_directory is not None \ else DEFAULT_MAP_TILE_CACHE_ROOT_DIR self.service_specific_cache_dir: Path = Path(effective_cache_root_dir) / self.service_identifier_name logger.info(f"Service-specific cache directory set to: {self.service_specific_cache_dir}") self.is_online_fetching_enabled: bool = enable_online_tile_fetching \ if enable_online_tile_fetching is not None else DEFAULT_ENABLE_ONLINE_FETCHING logger.info(f"Online tile fetching enabled: {self.is_online_fetching_enabled}") self.http_user_agent: str = "GeoElevationMapViewer/0.1 (Python Requests)" self.http_request_headers: Dict[str, str] = {"User-Agent": self.http_user_agent} self.http_request_timeout_seconds: int = DEFAULT_NETWORK_TIMEOUT_SECONDS self.download_max_retries: int = DEFAULT_MAX_DOWNLOAD_RETRIES self.download_retry_delay_seconds: int = DEFAULT_DOWNLOAD_RETRY_DELAY_SECONDS self._ensure_service_cache_directory_exists() self._cache_access_lock = threading.Lock() logger.info( f"MapTileManager initialized for service '{self.service_identifier_name}'. Online: {self.is_online_fetching_enabled}" ) def _ensure_service_cache_directory_exists(self) -> None: """Creates the service-specific cache directory if it doesn't exist.""" try: self.service_specific_cache_dir.mkdir(parents=True, exist_ok=True) logger.debug(f"Cache directory verified/created: {self.service_specific_cache_dir}") except OSError as e_mkdir: logger.error( f"Failed to create cache directory '{self.service_specific_cache_dir}': {e_mkdir}" ) except Exception as e_unexpected_mkdir: logger.exception( f"Unexpected error ensuring cache directory exists: {e_unexpected_mkdir}" ) def _get_tile_cache_file_path(self, z: int, x: int, y: int) -> Path: """ Constructs the full local file path for a specific map tile. The structure is: ///.png """ return self.service_specific_cache_dir / str(z) / str(x) / f"{y}.png" def _load_tile_from_cache( self, tile_cache_path: Path, tile_coordinates_log_str: str ) -> Optional[ImageType]: """Attempts to load a tile image from a cache file (thread-safe read).""" logger.debug(f"Checking cache for tile {tile_coordinates_log_str} at {tile_cache_path}") if not PIL_AVAILABLE_MANAGER: logger.error("Pillow not available, cannot load tile from cache.") return None try: with self._cache_access_lock: if tile_cache_path.is_file(): logger.info(f"Cache hit for tile {tile_coordinates_log_str}. Loading from disk.") if Image: pil_image = Image.open(tile_cache_path) pil_image.load() if pil_image.mode != "RGB": logger.debug( f"Converting cached image {tile_coordinates_log_str} from mode {pil_image.mode} to RGB." ) pil_image = pil_image.convert("RGB") return pil_image else: logger.error("Pillow's Image class is None, cannot open image from cache.") return None else: logger.debug(f"Cache miss for tile {tile_coordinates_log_str}.") return None except IOError as e_io_cache: logger.error( f"IOError reading cached tile {tile_cache_path}: {e_io_cache}. Treating as cache miss." ) return None except Exception as e_cache_unexpected: logger.exception( f"Unexpected error accessing cache file {tile_cache_path}: {e_cache_unexpected}" ) return None def _download_and_save_tile_to_cache( self, zoom_level: int, tile_x: int, tile_y: int, tile_cache_path: Path, tile_coordinates_log_str: str ) -> Optional[ImageType]: """Attempts to download a tile, process it, and save it to the cache.""" if not PIL_AVAILABLE_MANAGER: logger.error("Pillow not available, cannot download, process, or save tile.") return None if not self.is_online_fetching_enabled: logger.debug(f"Online fetching disabled. Cannot download tile {tile_coordinates_log_str}.") return None tile_download_url = self.map_service.get_tile_url(zoom_level, tile_x, tile_y) if not tile_download_url: logger.error(f"Failed to get URL for tile {tile_coordinates_log_str} from service.") return None logger.info(f"Downloading tile {tile_coordinates_log_str} from: {tile_download_url}") downloaded_pil_image: Optional[ImageType] = None for attempt_num in range(self.download_max_retries + 1): try: if requests: response = requests.get( tile_download_url, headers=self.http_request_headers, timeout=self.http_request_timeout_seconds, stream=True ) response.raise_for_status() else: logger.error("Requests library is None, cannot perform HTTP GET.") break image_binary_data = response.content if not image_binary_data: logger.warning(f"Downloaded empty content for tile {tile_coordinates_log_str}.") break try: if Image: pil_image = Image.open(io.BytesIO(image_binary_data)) pil_image.load() if pil_image.mode != "RGB": pil_image = pil_image.convert("RGB") logger.debug(f"Tile {tile_coordinates_log_str} downloaded (Attempt {attempt_num + 1}).") if tile_cache_path: self._save_image_to_cache_file(tile_cache_path, pil_image) else: logger.error(f"Invalid tile_cache_path for tile {tile_coordinates_log_str}. Cannot save.") downloaded_pil_image = pil_image break else: logger.error("Pillow's Image class is None, cannot process downloaded image data.") break except (IOError, Image.UnidentifiedImageError) as e_img_proc: logger.error( f"Failed to process image data for {tile_coordinates_log_str}: {e_img_proc}" ) break except Exception as e_proc_unexpected: logger.exception( f"Unexpected error processing downloaded image {tile_coordinates_log_str}: {e_proc_unexpected}" ) break except requests.exceptions.Timeout: logger.warning( f"Timeout downloading tile {tile_coordinates_log_str} (Attempt {attempt_num + 1})." ) except requests.exceptions.RequestException as e_req: status = getattr(e_req.response, "status_code", "N/A") logger.warning( f"Request error for tile {tile_coordinates_log_str} (Status: {status}, Attempt {attempt_num + 1}): {e_req}" ) if status == 404: break except Exception as e_dl_unexpected: logger.exception( f"Unexpected error downloading tile {tile_coordinates_log_str} (Attempt {attempt_num + 1}): {e_dl_unexpected}" ) break if attempt_num < self.download_max_retries: logger.debug(f"Waiting {self.download_retry_delay_seconds}s before retrying for {tile_coordinates_log_str}.") time.sleep(self.download_retry_delay_seconds) if downloaded_pil_image is None: logger.error(f"Failed to download tile {tile_coordinates_log_str} after all retries.") return downloaded_pil_image def _save_image_to_cache_file(self, tile_cache_path: Path, pil_image: ImageType) -> None: """Saves a PIL Image object to a file in the cache (thread-safe write).""" if not PIL_AVAILABLE_MANAGER or pil_image is None: logger.error("Pillow not available or image is None, cannot save tile to cache.") return with self._cache_access_lock: try: tile_cache_path.parent.mkdir(parents=True, exist_ok=True) if Image: pil_image.save(tile_cache_path, format='PNG') logger.debug(f"Saved tile to cache: {tile_cache_path}") else: logger.error("Pillow's Image class is None, cannot save image.") except IOError as e_io_save: logger.error(f"IOError saving tile to cache {tile_cache_path}: {e_io_save}") except Exception as e_save_unexpected: logger.exception( f"Unexpected error saving tile to cache {tile_cache_path}: {e_save_unexpected}" ) def get_tile_image( self, zoom_level: int, tile_x: int, tile_y: int, force_online_refresh: bool = False ) -> Optional[ImageType]: """ Retrieves a map tile image, using local cache first. If not cached or refresh is forced, attempts to download (if enabled). Returns a placeholder image if the tile cannot be retrieved. Args: zoom_level: The zoom level of the tile. tile_x: The X coordinate of the tile. tile_y: The Y coordinate of the tile. force_online_refresh: If True, bypasses cache and attempts download. Returns: A PIL Image object of the tile, or a placeholder image on failure. Returns None only if placeholder creation itself fails critically. """ tile_coords_log_str = f"({zoom_level},{tile_x},{tile_y})" logger.debug(f"Requesting tile image for {tile_coords_log_str}") if not self.map_service.is_zoom_level_valid(zoom_level): logger.error(f"Invalid zoom level {zoom_level} for map service '{self.service_identifier_name}'. Cannot get tile.") if PIL_AVAILABLE_MANAGER: return self._create_placeholder_tile_image(f"Invalid Zoom {zoom_level} Tile {tile_coords_log_str}") else: logger.error("Pillow not available, cannot create placeholder for invalid zoom tile.") return None tile_cache_file = self._get_tile_cache_file_path(zoom_level, tile_x, tile_y) retrieved_image: Optional[ImageType] = None if not force_online_refresh: retrieved_image = self._load_tile_from_cache(tile_cache_file, tile_coords_log_str) if retrieved_image is None: retrieved_image = self._download_and_save_tile_to_cache( zoom_level, tile_x, tile_y, tile_cache_file, tile_coords_log_str ) if retrieved_image is None: logger.warning(f"Failed to retrieve tile {tile_coords_log_str}. Using placeholder.") retrieved_image = self._create_placeholder_tile_image(tile_coords_log_str) if retrieved_image is None: logger.critical("Failed to create even a placeholder tile. Returning None.") return retrieved_image def stitch_map_image( self, zoom_level: int, x_tile_range: Tuple[int, int], y_tile_range: Tuple[int, int] ) -> Optional[ImageType]: """ Retrieves and stitches multiple map tiles to form a larger composite map image. Args: zoom_level: The zoom level for all tiles. x_tile_range: Inclusive start and end X tile coordinates (min_x, max_x). y_tile_range: Inclusive start and end Y tile coordinates (min_y, max_y). Returns: A PIL Image object of the stitched map, or None if a critical error occurs. Missing individual tiles are replaced by placeholders. """ logger.info( f"Request to stitch map: Zoom {zoom_level}, X-Range {x_tile_range}, Y-Range {y_tile_range}" ) min_tile_x, max_tile_x = x_tile_range min_tile_y, max_tile_y = y_tile_range if not (min_tile_x <= max_tile_x and min_tile_y <= max_tile_y): logger.error(f"Invalid tile ranges for stitching: X={x_tile_range}, Y={y_tile_range}") if PIL_AVAILABLE_MANAGER: try: placeholder_img = Image.new("RGB", (self.tile_size * 3, self.tile_size * 3), (255, 150, 150)) if ImageDraw: draw = ImageDraw.Draw(placeholder_img) error_text = f"Stitch Failed\nInvalid Tile Ranges:\nX={x_tile_range}, Y={y_tile_range}" try: try: from geoelevation.image_processor import DEFAULT_FONT font_to_use = DEFAULT_FONT except ImportError: font_to_use = None if font_to_use: if hasattr(draw, 'textbbox'): try: text_bbox = draw.textbbox((0,0), error_text, font=font_to_use, spacing=2) text_width = text_bbox[2] - text_bbox[0] text_height = text_bbox[3] - text_bbox[1] text_x = (placeholder_img.width - text_width) // 2 text_y = (placeholder_img.height - text_height) // 2 draw.text((text_x, text_y), error_text, fill="black", font=font_to_use, spacing=2, anchor="lt") except AttributeError: logger.warning("Pillow textbbox not available (Pillow < 8.0). Using textsize fallback for error text.") text_width, text_height = draw.textsize(error_text, font=font_to_use) text_x = (placeholder_img.width - text_width) // 2 text_y = (placeholder_img.height - text_height) // 2 draw.text((text_x, text_y), error_text, fill="black", font=font_to_use) except Exception as e_font_draw: logger.warning(f"Error drawing text with font on placeholder: {e_font_draw}. Falling back to simple draw.") draw.text((10, 10), error_text, fill="black") else: if font_to_use: text_width, text_height = draw.textsize(error_text, font=font_to_use) text_x = (placeholder_img.width - text_width) // 2 text_y = (placeholder_img.height - text_height) // 2 draw.text((text_x, text_y), error_text, fill="black", font=font_to_use) else: draw.text((10, 10), error_text, fill="black") except Exception as e_draw: logger.warning(f"Error drawing text on placeholder for invalid ranges: {e_draw}. Falling back to simple draw.") if ImageDraw: draw.text((10, 10), error_text, fill="black") return placeholder_img except Exception as e_placeholder_fail: logger.exception(f"Failed to create placeholder for invalid tile ranges: {e_placeholder_fail}") return None else: logger.error("Pillow not available, cannot create placeholder for invalid ranges.") return None single_tile_pixel_size = self.tile_size if single_tile_pixel_size <= 0: logger.error(f"Invalid tile size ({single_tile_pixel_size}) stored in manager. Cannot stitch.") if PIL_AVAILABLE_MANAGER: try: if ImageDraw is not None: placeholder_img = Image.new("RGB", (256 * 3, 256 * 3), DEFAULT_PLACEHOLDER_COLOR_RGB) draw = ImageDraw.Draw(placeholder_img) error_text = f"Stitch Failed\nInvalid Tile Size:\n{single_tile_pixel_size}" try: try: from geoelevation.image_processor import DEFAULT_FONT font_to_use = DEFAULT_FONT except ImportError: font_to_use = None if font_to_use: if hasattr(draw, 'textbbox'): try: text_bbox = draw.textbbox((0,0), error_text, font=font_to_use, spacing=2) text_width = text_bbox[2] - text_bbox[0] text_height = text_bbox[3] - text_bbox[1] text_pos = ((placeholder_img.width - text_width) // 2, (placeholder_img.height - text_height) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use, spacing=2, anchor="lt") except AttributeError: text_width, text_height = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_width) // 2, (placeholder_img.height - text_height) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) except Exception as e_font_draw: logger.warning(f"Error drawing text with font on placeholder: {e_font_draw}. Falling back to simple draw.") draw.text((10, 10), error_text, fill="black") else: if font_to_use: text_width, text_height = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_width) // 2, (placeholder_img.height - text_height) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) else: draw.text((10, 10), error_text, fill="black") else: draw.text((10, 10), error_text, fill="black") except Exception as e_draw: logger.warning(f"Error drawing text on placeholder: {e_draw}. Falling back to simple draw.") if ImageDraw: draw.text((10, 10), error_text, fill="black") return placeholder_img else: logger.error("Pillow's ImageDraw class is None, cannot create placeholder image.") return None except Exception as e_placeholder_fail: logger.exception(f"Failed to create large placeholder for stitching error: {e_placeholder_fail}") return None else: logger.error("Pillow not available, cannot create placeholder for invalid tile size.") return None num_tiles_wide = (max_tile_x - min_tile_x) + 1 num_tiles_high = (max_tile_y - min_tile_y) + 1 total_image_width = num_tiles_wide * single_tile_pixel_size total_image_height = num_tiles_high * single_tile_pixel_size logger.debug( f"Stitched image dimensions: {total_image_width}x{total_image_height} " f"({num_tiles_wide}x{num_tiles_high} tiles of {single_tile_pixel_size}px)" ) MAX_IMAGE_DIMENSION = 16384 if total_image_width > MAX_IMAGE_DIMENSION or total_image_height > MAX_IMAGE_DIMENSION: logger.error( f"Requested stitched image size ({total_image_width}x{total_image_height}) " f"exceeds maximum allowed dimension ({MAX_IMAGE_DIMENSION}). Aborting stitch." ) if PIL_AVAILABLE_MANAGER: try: if ImageDraw is not None: placeholder_img = Image.new("RGB", (256 * 3, 256 * 3), (255, 100, 100)) draw = ImageDraw.Draw(placeholder_img) error_text = f"Stitch Failed\nImage too large:\n{total_image_width}x{total_image_height}px" try: try: from geoelevation.image_processor import DEFAULT_FONT font_to_use = DEFAULT_FONT except ImportError: font_to_use = None if font_to_use: if hasattr(draw, 'textbbox'): try: text_bbox = draw.textbbox((0,0), error_text, font=font_to_use, spacing=2) text_w = text_bbox[2] - text_bbox[0] text_h = text_bbox[3] - text_bbox[1] text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use, spacing=2, anchor="lt") except AttributeError: text_w, text_h = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) except Exception as e_font_draw: logger.warning(f"Error drawing text with font on placeholder: {e_font_draw}. Falling back to simple draw.") draw.text((10, 10), error_text, fill="black") else: if font_to_use: text_w, text_h = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) else: draw.text((10, 10), error_text, fill="black") else: draw.text((10, 10), error_text, fill="black") except Exception as e_draw: logger.warning(f"Error drawing text on placeholder: {e_draw}. Falling back to simple draw.") if ImageDraw: draw.text((10, 10), error_text, fill="black") return placeholder_img else: logger.error("Pillow's ImageDraw class is None, cannot create placeholder image.") return None except Exception as e_placeholder_fail: logger.exception(f"Failed to create large placeholder for size error: {e_placeholder_fail}") return None else: logger.error("Pillow not available, cannot create placeholder for excessive size request.") return None try: if PIL_AVAILABLE_MANAGER: if Image: stitched_map_image = Image.new("RGB", (total_image_width, total_image_height)) else: raise ImportError("Pillow's Image class is None.") else: raise ImportError("Pillow not available to create new image.") except Exception as e_create_blank: logger.exception(f"Failed to create blank image for stitching: {e_create_blank}. Dimensions: {total_image_width}x{total_image_height}") if PIL_AVAILABLE_MANAGER: try: if ImageDraw is not None: placeholder_img = Image.new("RGB", (256 * 3, 256 * 3), (255, 100, 100)) draw = ImageDraw.Draw(placeholder_img) error_text = f"Stitch Failed\nCannot create image:\n{e_create_blank}" try: try: from geoelevation.image_processor import DEFAULT_FONT font_to_use = DEFAULT_FONT except ImportError: font_to_use = None if font_to_use: if hasattr(draw, 'textbbox'): try: text_bbox = draw.textbbox((0,0), error_text, font=font_to_use, spacing=2) text_w = text_bbox[2] - text_bbox[0] text_h = text_bbox[3] - text_bbox[1] text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use, spacing=2, anchor="lt") except AttributeError: text_w, text_h = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) except Exception as e_font_draw: logger.warning(f"Error drawing text with font on placeholder: {e_font_draw}. Falling back to simple draw.") draw.text((10, 10), error_text, fill="black") else: if font_to_use: text_w, text_h = draw.textsize(error_text, font=font_to_use) text_pos = ((placeholder_img.width - text_w) // 2, (placeholder_img.height - text_h) // 2) draw.text(text_pos, error_text, fill="black", font=font_to_use) else: draw.text((10, 10), error_text, fill="black") else: draw.text((10, 10), error_text, fill="black") except Exception as e_draw: logger.warning(f"Error drawing text on placeholder: {e_draw}. Falling back to simple draw.") if ImageDraw: draw.text((10, 10), error_text, fill="black") return placeholder_img else: logger.error("Pillow's ImageDraw class is None, cannot create placeholder image.") return None except Exception as e_placeholder_fail: logger.exception(f"Failed to create large placeholder for memory error: {e_placeholder_fail}") return None else: logger.error("Pillow not available, cannot create placeholder for image creation error.") return None for row_index, current_tile_y in enumerate(range(min_tile_y, max_tile_y + 1)): for col_index, current_tile_x in enumerate(range(min_tile_x, max_tile_x + 1)): tile_image_pil = self.get_tile_image(zoom_level, current_tile_x, current_tile_y) if tile_image_pil is None: logger.critical( f"Critical error: get_tile_image returned None for ({zoom_level},{current_tile_x},{current_tile_y}). Aborting stitch." ) return None paste_position_x = col_index * single_tile_pixel_size paste_position_y = row_index * single_tile_pixel_size logger.debug( f"Pasting tile ({zoom_level},{current_tile_x},{current_tile_y}) at " f"pixel position ({paste_position_x},{paste_position_y})" ) try: if tile_image_pil and tile_image_pil.size != (self.tile_size, self.tile_size): logger.warning(f"Tile image size {tile_image_pil.size} doesn't match expected {self.tile_size}. Resizing for stitch.") if PIL_AVAILABLE_MANAGER and Image: tile_image_pil = tile_image_pil.resize((self.tile_size, self.tile_size), Image.Resampling.LANCZOS) else: logger.error("Pillow not available, cannot resize tile for stitch.") continue if tile_image_pil: stitched_map_image.paste(tile_image_pil, (paste_position_x, paste_position_y)) except Exception as e_paste: logger.exception( f"Error pasting tile ({zoom_level},{current_tile_x},{current_tile_y}) " f"at ({paste_position_x},{paste_position_y}): {e_paste}" ) logger.info(f"Map stitching complete for zoom {zoom_level}, X={x_tile_range}, Y={y_tile_range}.") return stitched_map_image def _create_placeholder_tile_image(self, identifier: str = "N/A") -> Optional[ImageType]: """ Creates and returns a placeholder tile image (e.g., a grey square). Includes optional text identifier on the placeholder. """ if not (PIL_AVAILABLE_MANAGER and ImageDraw is not None): logger.warning("Cannot create placeholder tile: Pillow or ImageDraw library not available.") return None try: tile_pixel_size = self.tile_size placeholder_color = DEFAULT_PLACEHOLDER_COLOR_RGB if Image: placeholder_img = Image.new("RGB", (tile_pixel_size, tile_pixel_size), color=placeholder_color) else: logger.error("Pillow's Image class is None, cannot create placeholder image.") return None if ImageDraw: draw = ImageDraw.Draw(placeholder_img) else: logger.error("Pillow's ImageDraw class is None, cannot draw on placeholder.") return placeholder_img overlay_text = f"Tile Fail\n{identifier}" try: try: from geoelevation.image_processor import DEFAULT_FONT font_to_use = DEFAULT_FONT except ImportError: font_to_use = None if font_to_use is not None: if hasattr(draw, 'textbbox'): try: text_bbox = draw.textbbox((0,0), overlay_text, font=font_to_use, spacing=2) text_width = text_bbox[2] - text_bbox[0] text_height = text_bbox[3] - text_bbox[1] text_x = (tile_pixel_size - text_width) // 2 text_y = (tile_pixel_size - text_height) // 2 bg_padding = 1 bg_coords = [ text_x - bg_padding, text_y - bg_padding, text_x + text_width + bg_padding, text_y + text_height + bg_padding, ] draw.rectangle(bg_coords, fill="rgba(0, 0, 0, 150)") draw.text((text_x, text_y), overlay_text, fill="white", font=font_to_use, spacing=2, anchor="lt") except AttributeError: logger.warning("Pillow textbbox not available (Pillow < 8.0). Using textsize fallback for placeholder text.") if font_to_use: text_width, text_height = draw.textsize(overlay_text, font=font_to_use) if "\n" in overlay_text: line_count = overlay_text.count("\n") + 1 pass text_x = (tile_pixel_size - text_width) // 2 text_y = (tile_pixel_size - text_height) // 2 bg_padding = 1 bg_coords = [text_x - bg_padding, text_y - bg_padding, text_x + text_width + bg_padding, text_y + text_height + bg_padding] draw.rectangle(bg_coords, fill="rgba(0, 0, 0, 150)") draw.text((text_x, text_y), overlay_text, fill="white", font=font_to_use) else: draw.text((10, 10), overlay_text, fill="black") except Exception as e_font_draw: logger.warning(f"Error drawing text with font on placeholder: {e_font_draw}. Falling back to simple draw.") draw.text((10, 10), overlay_text, fill="black") else: if font_to_use: draw.text((10, 10), overlay_text, fill="black", font=font_to_use) else: draw.text((10, 10), overlay_text, fill="black") else: logger.debug("No font available for placeholder text drawing.") draw.text((10, 10), overlay_text, fill="black") except Exception as e_draw: logger.warning(f"Error drawing text on placeholder: {e_draw}. Falling back to simple draw.") if ImageDraw: draw.text((10, 10), overlay_text, fill="black") return placeholder_img except Exception as e_placeholder: logger.exception(f"Error creating placeholder tile image: {e_placeholder}") return None def _get_bounds_for_tile_range( self, zoom: int, tile_ranges: Tuple[Tuple[int, int], Tuple[int, int]] ) -> Optional[Tuple[float, float, float, float]]: """ Calculates the precise geographic bounds covered by a given range of tiles. Requires 'mercantile' library. """ # MODIFIED: Removed check for MERCANTILE_MODULE_LOCALLY_AVAILABLE. # WHY: This constant is defined in map_utils.py, not map_manager.py. # Instead, check if the locally imported `mercantile` is None. # HOW: Removed the check for MERCANTILE_MODULE_LOCALLY_AVAILABLE. if mercantile is None: # Check if mercantile is None from the local import attempt logger.error("mercantile library not found or is None, cannot calculate bounds for tile range.") return None try: min_x, max_x = tile_ranges[0] min_y, max_y = tile_ranges[1] # Get bounds of the top-left tile and bottom-right tile # mercantile.bounds(x, y, z) returns (west, south, east, north) # MODIFIED: Check for mercantile before using mercantile.bounds. # WHY: Prevent AttributeError. (Redundant with the check at the start of the method, but defensive). # HOW: Added `if mercantile:`. if mercantile: top_left_tile_bounds = mercantile.bounds(min_x, min_y, zoom) bottom_right_tile_bounds = mercantile.bounds(max_x, max_y, zoom) else: # This branch should ideally not be hit due to the check at the method start. logger.error("mercantile library is None during bounds calculation step.") return None overall_west_lon = top_left_tile_bounds.west overall_south_lat = bottom_right_tile_bounds.south overall_east_lon = bottom_right_tile_bounds.east overall_north_lat = top_left_tile_bounds.north return (overall_west_lon, overall_south_lat, overall_east_lon, overall_north_lat) except Exception as e_bounds_calc: logger.exception( f"Error calculating geographic bounds for tile range {tile_ranges} at zoom {zoom}: {e_bounds_calc}" ) return None def clear_entire_service_cache(self) -> None: """Deletes all cached tiles for the current map service.""" logger.info(f"Attempting to clear entire cache for service '{self.service_identifier_name}' at {self.service_specific_cache_dir}") if not self.service_specific_cache_dir.exists(): logger.warning(f"Cache directory '{self.service_specific_cache_dir}' does not exist. Nothing to clear.") return with self._cache_access_lock: try: if self.service_specific_cache_dir.is_dir(): shutil.rmtree(self.service_specific_cache_dir) logger.info(f"Successfully cleared cache at {self.service_specific_cache_dir}.") self._ensure_service_cache_directory_exists() else: logger.warning(f"Cache path '{self.service_specific_cache_dir}' is not a directory.") except OSError as e_os_clear: logger.error(f"OS Error clearing cache at '{self.service_specific_cache_dir}': {e_os_clear}") except Exception as e_clear_unexpected: logger.exception( f"Unexpected error clearing cache '{self.service_specific_cache_dir}': {e_clear_unexpected}" )