from __future__ import annotations

DEFAULT_TRAIN_RUN_NAME = "curriculum_consolidate_vertical_v2_stage5_no_leak"
ACTIVE_CHECKPOINT = "checkpoints/setattention_stage6plus_latest_no_leak.pt"

DEFAULT_CURRICULUM = "8:6:300:0.88,10:8:420:0.86,12:10:520:0.84,14:12:620:0.82"

TRAIN_DEFAULTS: dict[str, float | int | str] = {
    "envs": 196608,
    "eval_envs": 8192,
    "steps": 64,
    "epochs": 1,
    "minibatches": 48,
    "hidden": 256,
    "depth": 3,
    "policy_arch": "attention",
    "planner_mode": "direct",
    "sensor_mode": "topk",
    "sensor_topk": 16,
    "sensor_token_range": 9.5,
    "sensor_fov_degrees": 120.0,
    "sensor_token_sort": "distance",
    "observe_progress": 0,
    "lr": 4.0e-5,
    "entropy_coef": 0.008,
    "sil_coef": 0.030,
    "capture": 8,
    "log_every": 20,
    "min_stage_updates": 20,
}

PHYSICS_DEFAULTS: dict[str, float] = {
    "gravity": 30.0,
    "accel": 65.0,
    "ground_damping": 0.55,
    "air_damping": 0.91,
    "air_control": 0.10,
    "max_speed": 4.3,
    "jump_velocity": 7.2,
    "sprint_speed_mult": 1.35,
    "sprint_accel_mult": 1.25,
    "turn_speed": 3.2,
    "landing_pad_scale": 0.25,
    "landing_top_tolerance": 0.03,
    "landing_prev_slack": 0.12,
}

ENV_DEFAULTS: dict[str, float | int] = {
    "map_height_offset": 2.85,
    "map_vertical_scale": 1.70,
    "map_max_height": 7.0,
    "goal_progress_reward_scale": 1.35,
    "first_visit_reward": 0.12,
    "step_cost": 0.0005,
    "fall_penalty": 4.0,
    "success_time_bonus": 0.06,
}

MAPGEN_KWARGS: dict[str, float | int] = {
    "map_scenic_turn_scale": 0.62,
    "map_edge_gap_min": 0.95,
    "map_edge_gap_max": 1.58,
    "map_route_gap_scale": 1.22,
    "map_route_edge_gap_floor": 0.38,
    "map_goal_xy_min": 28.0,
    "map_goal_xy_per_jump_min": 1.90,
    "map_extra_span_min_frac": 0.04,
    "map_extra_span_max_frac": 0.88,
    "map_extra_span_end_margin": 1.25,
    "map_platform_min_scale": 0.52,
    "map_platform_max_scale": 1.42,
    "map_platform_tiny_fraction": 0.18,
    "map_platform_large_fraction": 0.18,
    "map_platform_aspect_jitter": 0.42,
    "map_route_tiny_fraction": 0.10,
    "map_redirect_period": 0,
    "map_route_detour_period": 4,
    "map_route_detour_depth": 0.58,
    "map_backtrack_period": 5,
    "map_backtrack_depth": 0.42,
    "map_hairpin_period": 7,
    "map_hairpin_angle": 1.24,
    "map_hairpin_depth": 0.48,
    "map_unintuitive_depth": 0.56,
    "map_valley_period": 5,
    "map_valley_depth": 0.58,
    "map_direct_decoy_fraction": 0.08,
    "map_cluster_extra_fraction": 0.10,
    "map_decoy_corridor_fraction": 0.06,
    "map_false_summit_fraction": 0.14,
    "map_false_branch_fraction": 0.20,
    "map_false_finish_fraction": 0.16,
    "map_greedy_trap_fraction": 0.22,
    "map_braid_bridge_fraction": 0.16,
    "map_braid_bridge_span": 4,
    "map_trap_jump_fraction": 0.40,
    "map_trap_jump_min_drop": 0.24,
    "map_trap_jump_drop": 0.70,
    "map_trap_jump_edge_margin": 0.04,
    "map_trap_jump_high_fraction": 0.20,
    "map_trap_jump_high_min_up": 0.24,
    "map_trap_jump_high_max_up": 0.94,
    "map_trap_jump_disguise": 0.82,
}