| import numpy as np
|
| from PIL import Image
|
| import os
|
| from pathlib import Path
|
| import config
|
|
|
|
|
| def load_tiles():
|
| """
|
| Load all tiles from the tile directory.
|
|
|
| NOTE: This loads tiles every time - inefficient!
|
| """
|
| tile_path = Path(config.tile_directory)
|
|
|
| if not tile_path.exists():
|
| print(f"Warning: Tile directory {config.tile_directory} not found!")
|
| print("Creating sample tiles...")
|
| create_sample_tiles()
|
| tile_path = Path(config.tile_directory)
|
|
|
|
|
| valid_extensions = {'.jpg', '.jpeg', '.png', '.bmp'}
|
| tile_files = [f for f in tile_path.iterdir()
|
| if f.suffix.lower() in valid_extensions]
|
|
|
| if not tile_files:
|
| print("No tile images found! Creating sample tiles...")
|
| config.create_sample_tiles()
|
| tile_files = [f for f in tile_path.iterdir()
|
| if f.suffix.lower() in valid_extensions]
|
|
|
| print(f"Loading {len(tile_files)} tiles...")
|
|
|
| for tile_file in tile_files:
|
| try:
|
| tile = Image.open(tile_file)
|
| tile = tile.convert('RGB')
|
| tile = tile.resize((32, 32))
|
| tile_array = np.array(tile)
|
| config.tiles.append(tile_array)
|
|
|
| avg_color = np.mean(tile_array, axis=(0, 1))
|
| config.tile_colors.append(avg_color)
|
| except Exception as e:
|
| print(f"Error loading {tile_file}: {e}")
|
|
|
| print(f"Loaded {len(config.tiles)} tiles successfully")
|
|
|
| def create_sample_tiles():
|
| """
|
| Create sample colored tiles if no tiles are provided.
|
| This creates a basic set of tiles with different colors.
|
| """
|
| os.makedirs(config.tile_directory, exist_ok=True)
|
|
|
|
|
| colors = [
|
| (255, 0, 0),
|
| (0, 255, 0),
|
| (0, 0, 255),
|
| (255, 255, 0),
|
| (255, 0, 255),
|
| (0, 255, 255),
|
| (255, 255, 255),
|
| (128, 128, 128),
|
| (0, 0, 0),
|
| (255, 128, 0),
|
| (128, 0, 255),
|
| (0, 128, 128),
|
| ]
|
|
|
| for i, color in enumerate(colors):
|
| tile = np.full((32, 32, 3), color, dtype=np.uint8)
|
| img = Image.fromarray(tile)
|
| img.save(os.path.join(config.tile_directory, f'tile_{i:02d}.png'))
|
|
|
| print(f"Created {len(colors)} sample tiles in {config.tile_directory}")
|
|
|
| def find_best_tile(target_color):
|
| """
|
| Find the tile that best matches the target color.
|
|
|
| Parameters:
|
| -----------
|
| target_color : np.ndarray
|
| Target RGB color
|
|
|
| Returns:
|
| --------
|
| int : Index of best matching tile
|
| """
|
| target_color = np.array(target_color)
|
| differences = config.tile_colors - target_color
|
| distances = np.linalg.norm(differences, axis=1)
|
| distances = np.abs(distances)
|
| return np.argmin(distances)
|
|
|
|
|
