#!/usr/bin/env python3
"""
NovaFarma - Mass TSX Generator
================================

Generates individual .tsx tileset files for all 3877 separated object images.
This enables each object to be used independently in Tiled Map Editor.

Author: Antigravity AI
Date: 2025-12-22
"""

import os
from pathlib import Path
from PIL import Image

# Configuration
ASSETS_DIR = r"c:\novafarma\assets"
SOURCE_DIRS = [
    "narezano_loceno/topdown_objects",
    "narezano_loceno/krvava_zetev_sprites", 
    "narezano_loceno/tiled_sprites"
]
OUTPUT_DIR = r"c:\novafarma\assets\tilesets\individual_objects"

# TSX template
TSX_TEMPLATE = '''<?xml version="1.0" encoding="UTF-8"?>
<tileset version="1.10" tiledversion="1.11.0" name="{name}" tilewidth="{width}" tileheight="{height}" tilecount="1" columns="1">
 <image source="{relative_path}" width="{width}" height="{height}"/>
</tileset>
'''

def get_image_dimensions(image_path):
    """Get width and height of an image."""
    with Image.open(image_path) as img:
        return img.size

def create_tsx_file(png_path, output_dir):
    """
    Create a .tsx file for a given PNG image.
    
    Args:
        png_path: Path to PNG file
        output_dir: Directory to save .tsx file
    """
    # Get image info
    width, height = get_image_dimensions(png_path)
    
    # Create name from filename (without extension)
    filename = os.path.basename(png_path)
    name = os.path.splitext(filename)[0]
    
    # Calculate relative path from tilesets/individual_objects/ to the PNG
    # tilesets/individual_objects/ -> ../../narezano_loceno/[subfolder]/[file].png
    png_relative = os.path.relpath(png_path, output_dir).replace('\\', '/')
    
    # Generate TSX content
    tsx_content = TSX_TEMPLATE.format(
        name=name,
        width=width,
        height=height,
        relative_path=png_relative
    )
    
    # Create .tsx filename
    tsx_filename = f"{name}.tsx"
    tsx_path = os.path.join(output_dir, tsx_filename)
    
    # Write TSX file using UTF-8 BOM encoding for XML stability
    with open(tsx_path, 'w', encoding='utf-8-sig') as f:
        f.write(tsx_content)
    
    return tsx_path

def main():
    """
    Main function to generate .tsx files for all separated objects.
    """
    print("=" * 70)
    print("NovaFarma - Mass TSX Generator")
    print("=" * 70)
    print()
    
    # Create output directory
    os.makedirs(OUTPUT_DIR, exist_ok=True)
    print(f"[OK] Output directory: {OUTPUT_DIR}")
    print()
    
    # Process each source directory
    total_created = 0
    
    for source_dir in SOURCE_DIRS:
        full_source_path = os.path.join(ASSETS_DIR, source_dir)
        
        if not os.path.exists(full_source_path):
            print(f"[WARN] Directory not found: {source_dir}")
            continue
        
        # Find all PNG files
        png_files = list(Path(full_source_path).glob("*.png"))
        
        if len(png_files) == 0:
            print(f"[WARN] No PNG files in: {source_dir}")
            continue
        
        print(f"[INFO] Processing: {source_dir}")
        print(f"       Found {len(png_files)} PNG files")
        
        # Create TSX for each PNG
        created_count = 0
        error_count = 0
        for png_file in png_files:
            try:
                tsx_path = create_tsx_file(str(png_file), OUTPUT_DIR)
                created_count += 1
                
                # Show progress every 100 files
                if created_count % 100 == 0:
                    print(f"       ... {created_count}/{len(png_files)} processed")
                    
            except Exception as e:
                error_count += 1
                if error_count <= 5:  # Show first 5 errors only
                    print(f"[ERROR] processing {png_file.name}: {e}")
        
        print(f"[OK] Created {created_count} .tsx files")
        if error_count > 0:
            print(f"[WARN] Failed to process {error_count} files")
        print()
        total_created += created_count
    
    # Summary
    print("=" * 70)
    print("SUMMARY")
    print("=" * 70)
    print(f"[OK] Total .tsx files created: {total_created}")
    print(f"[OK] Output location: {OUTPUT_DIR}")
    print()
    print(">>> All individual objects now ready for Tiled Map Editor!")
    print("=" * 70)
    print()
    print("===============================================================")
    print("NEXT STEPS:")
    print("===============================================================")
    print("1. Open Tiled Map Editor")
    print("2. Open your map (e.g., micro_farm_128x128.tmx)")
    print("3. Map -> Add External Tileset...")
    print("4. Browse to: assets/tilesets/individual_objects/")
    print("5. Select the .tsx files you want to use")
    print("6. Start placing objects on your map!")
    print()

if __name__ == "__main__":
    main()