"""Simulation configuration classes for different modes."""

from dataclasses import dataclass, field
from typing import List, Optional, Dict, Any


@dataclass
class EntityConfig:
    """Configuration for entity hyperparameters."""

    # Global entity settings (apply to all entity types unless overridden)
    max_acceleration: float = 0.125
    max_angular_acceleration: float = 0.25
    max_velocity: float = 1.0
    max_rotational_velocity: float = 3.0

    # Entity type specific configs (all entity parameters should be defined here)
    entity_types: Dict[str, Dict[str, Any]] = field(default_factory=lambda: {
        "default_cell": {
            "reproduction_energy": 1700,
            "starting_energy": 1000,
            "interaction_radius": 50,
            "drag_coefficient": 0.02,
            "energy_cost_base": 1.5,
            "neural_network_complexity_cost": 0.08,
            "movement_cost": 0.25,
            "food_energy_value": 140,
            "max_visual_width": 10,
            "reproduction_count": 2,
            "mutation_rate": 0.05,
            "offspring_offset_range": 10
        }
    })


@dataclass
class SimulationConfig:
    """Configuration for simulation setup."""
    grid_width: int = 50
    grid_height: int = 50
    cell_size: int = 20
    initial_cells: int = 50
    initial_food: int = 500
    food_spawning: bool = True
    random_seed: int = 0
    default_tps: float = 40.0
    entities: EntityConfig = field(default_factory=EntityConfig)


@dataclass
class OutputConfig:
    """Configuration for data output."""
    enabled: bool = True
    directory: str = "simulation_output"
    formats: List[str] = field(default_factory=lambda: ['json'])
    collect_metrics: bool = False
    collect_entities: bool = False
    collect_evolution: bool = False
    metrics_interval: int = 100
    entities_interval: int = 1000
    evolution_interval: int = 1000
    real_time: bool = False


@dataclass
class HeadlessConfig:
    """Configuration for headless simulation mode."""
    # Simulation settings
    max_ticks: Optional[int] = None
    max_duration: Optional[float] = None  # seconds
    early_stop: bool = False  # Stop when 0 cells remaining

    # Output settings
    output: OutputConfig = field(default_factory=OutputConfig)

    # Simulation core config
    simulation: SimulationConfig = field(default_factory=SimulationConfig)


@dataclass
class InteractiveConfig:
    """Configuration for interactive simulation mode with UI."""
    # Window settings
    window_width: int = 0  # 0 = auto-detect
    window_height: int = 0  # 0 = auto-detect
    vsync: bool = True
    resizable: bool = True

    # UI settings
    show_grid: bool = True
    show_interaction_radius: bool = False
    show_legend: bool = True
    control_bar_height: int = 48
    inspector_width: int = 260
    properties_width: int = 320
    console_height: int = 120

    # Simulation core config
    simulation: SimulationConfig = field(default_factory=lambda: SimulationConfig(initial_cells=350))


@dataclass
class ExperimentConfig:
    """Configuration for automated experiments."""
    name: str = "experiment"
    description: str = ""

    # Multiple runs
    runs: int = 1
    run_duration: Optional[float] = None  # seconds per run
    run_ticks: Optional[int] = None  # ticks per run

    # Variables to test (for parameter sweeps)
    variables: dict = field(default_factory=dict)

    # Base configuration
    base_config: HeadlessConfig = field(default_factory=HeadlessConfig)

    # Output aggregation
    aggregate_results: bool = True
    aggregate_format: str = "csv"