#include <catch2/catch_test_macros.hpp>
#include "../src/physics.h"
#include "../src/simulation.h"
#include "../src/config_loader.h"
#include <cmath>

TEST_CASE("Invalid parent: Earth should not become child of spacecraft",
          "[init][parent][bug]") {
    const double TIME_STEP = 60.0;

    SimulationState* sim = create_simulation(10, 0, 0, TIME_STEP);
    REQUIRE(load_system_config(sim, "tests/configs/earth_mars_simple.toml"));

    const int EARTH_IDX = 1;
    const int SPACECRAFT_IDX = 3;
    const int SUN_IDX = 0;

    REQUIRE(sim->bodies[EARTH_IDX].parent_index == SUN_IDX);

    for (int step = 0; step < 10; step++) {
        update_simulation(sim);

        int earth_parent = sim->bodies[EARTH_IDX].parent_index;

        INFO("Step " << step << ": Earth's parent = " << earth_parent
             << " (" << (earth_parent == SPACECRAFT_IDX ? "SPACECRAFT" :
                         earth_parent == SUN_IDX ? "Sun" : "Other") << ")");

        REQUIRE(earth_parent != SPACECRAFT_IDX);
    }

    destroy_simulation(sim);
}

TEST_CASE("Invalid parent: massive bodies never become children of small bodies",
          "[init][parent][hierarchy]") {
    const double TIME_STEP = 60.0;
    const double MASS_THRESHOLD_RATIO = 1000.0;

    SimulationState* sim = create_simulation(10, 0, 0, TIME_STEP);
    REQUIRE(load_system_config(sim, "tests/configs/earth_mars_simple.toml"));

    for (int i = 0; i < sim->body_count; i++) {
        if (sim->bodies[i].parent_index >= 0) {
            int parent_idx = sim->bodies[i].parent_index;
            double parent_mass = sim->bodies[parent_idx].mass;
            double child_mass = sim->bodies[i].mass;
            double mass_ratio = parent_mass / child_mass;

            INFO("Body " << i << " (" << sim->bodies[i].name
                 << ") parent " << parent_idx << " (" << sim->bodies[parent_idx].name
                 << "): mass ratio = " << mass_ratio);

            REQUIRE(mass_ratio >= 1.0);

            if (strcmp(sim->bodies[parent_idx].name, "Spacecraft") != 0 &&
                strcmp(sim->bodies[i].name, "Spacecraft") != 0) {
                REQUIRE(mass_ratio >= MASS_THRESHOLD_RATIO);
            }
        }
    }

    for (int step = 0; step < 100; step++) {
        update_simulation(sim);

        for (int i = 0; i < sim->body_count; i++) {
            if (sim->bodies[i].parent_index >= 0) {
                int parent_idx = sim->bodies[i].parent_index;
                double parent_mass = sim->bodies[parent_idx].mass;
                double child_mass = sim->bodies[i].mass;

                if (child_mass > 1e20) {
                    REQUIRE(parent_mass > child_mass);
                }
            }
        }
    }

    destroy_simulation(sim);
}

TEST_CASE("Invalid parent: detect placeholder config values",
          "[init][config][validation]") {
    const double TIME_STEP = 60.0;

    SimulationState* sim = create_simulation(10, 0, 0, TIME_STEP);
    REQUIRE(load_system_config(sim, "tests/configs/earth_mars_simple.toml"));

    const int EARTH_IDX = 1;
    const int SPACECRAFT_IDX = 3;

    Vec3 craft_pos = sim->bodies[SPACECRAFT_IDX].global_position;
    Vec3 earth_pos = sim->bodies[EARTH_IDX].global_position;

    double distance = vec3_distance(craft_pos, earth_pos);
    double min_distance = sim->bodies[EARTH_IDX].radius + sim->bodies[SPACECRAFT_IDX].radius;

    INFO("Distance Earth <-> Spacecraft: " << distance << " m");
    INFO("Earth position: (" << earth_pos.x << ", " << earth_pos.y << ", " << earth_pos.z << ")");
    INFO("Spacecraft position: (" << craft_pos.x << ", " << craft_pos.y << ", " << craft_pos.z << ")");
    INFO("Earth radius: " << sim->bodies[EARTH_IDX].radius << " m");
    INFO("Spacecraft radius: " << sim->bodies[SPACECRAFT_IDX].radius << " m");
    INFO("Minimum required: " << min_distance << " m (parent radius + body radius)");

    REQUIRE(distance >= min_distance);

    destroy_simulation(sim);
}

TEST_CASE("Mutual SOI: similar mass planets within SOI boundary",
          "[init][soi][mutual][edge_case]") {
    const double TIME_STEP = 60.0;

    SimulationState* sim = create_simulation(10, 0, 0, TIME_STEP);
    REQUIRE_FALSE(load_system_config(sim, "tests/configs/mutual_soi_close.toml"));

    destroy_simulation(sim);
}