#include "systems/building_collision_registry.h"
#include <gtest/gtest.h>

using namespace Game::Systems;

namespace {
constexpr float kDefaultGridCellSize = 1.0F;
} // namespace

class BuildingCollisionRegistryTest : public ::testing::Test {
protected:
  void SetUp() override { BuildingCollisionRegistry::instance().clear(); }

  void TearDown() override { BuildingCollisionRegistry::instance().clear(); }
};

TEST_F(BuildingCollisionRegistryTest, PointInsideBuilding) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  EXPECT_TRUE(registry.is_point_in_building(0.0F, 0.0F));
  EXPECT_TRUE(registry.is_point_in_building(1.0F, 1.0F));
  EXPECT_FALSE(registry.is_point_in_building(10.0F, 10.0F));
}

TEST_F(BuildingCollisionRegistryTest, PointOutsideBuilding) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  EXPECT_FALSE(registry.is_point_in_building(5.0F, 0.0F));
  EXPECT_FALSE(registry.is_point_in_building(0.0F, 5.0F));
  EXPECT_FALSE(registry.is_point_in_building(-5.0F, 0.0F));
  EXPECT_FALSE(registry.is_point_in_building(0.0F, -5.0F));
}

TEST_F(BuildingCollisionRegistryTest, CircleOverlappingBuilding) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  EXPECT_TRUE(registry.is_circle_overlapping_building(0.0F, 0.0F, 0.5F));

  EXPECT_TRUE(registry.is_circle_overlapping_building(1.0F, 1.0F, 0.5F));

  EXPECT_TRUE(registry.is_circle_overlapping_building(2.5F, 0.0F, 1.0F));
}

TEST_F(BuildingCollisionRegistryTest, CircleNotOverlappingBuilding) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  EXPECT_FALSE(registry.is_circle_overlapping_building(10.0F, 0.0F, 0.5F));
  EXPECT_FALSE(registry.is_circle_overlapping_building(0.0F, 10.0F, 0.5F));

  EXPECT_FALSE(registry.is_circle_overlapping_building(5.0F, 0.0F, 0.5F));
}

TEST_F(BuildingCollisionRegistryTest, CircleTouchingBuildingEdge) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  EXPECT_TRUE(registry.is_circle_overlapping_building(2.0F, 0.0F, 0.5F));

  EXPECT_TRUE(registry.is_circle_overlapping_building(3.0F, 0.0F, 1.0F));
}

TEST_F(BuildingCollisionRegistryTest, LargeUnitRadiusPreventedFromClipping) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  float const large_radius = 2.0F;

  EXPECT_TRUE(
      registry.is_circle_overlapping_building(3.5F, 0.0F, large_radius));

  EXPECT_TRUE(
      registry.is_circle_overlapping_building(0.0F, 3.5F, large_radius));

  EXPECT_FALSE(
      registry.is_circle_overlapping_building(5.0F, 0.0F, large_radius));
}

TEST_F(BuildingCollisionRegistryTest, IgnoreEntityId) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);
  registry.register_building(2, "barracks", 10.0F, 10.0F, 0);

  EXPECT_TRUE(registry.is_circle_overlapping_building(0.0F, 0.0F, 0.5F, 0));

  EXPECT_FALSE(registry.is_circle_overlapping_building(0.0F, 0.0F, 0.5F, 1));

  EXPECT_TRUE(registry.is_circle_overlapping_building(10.0F, 10.0F, 0.5F, 1));
}

TEST_F(BuildingCollisionRegistryTest, MultipleBuildings) {
  auto &registry = BuildingCollisionRegistry::instance();

  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);
  registry.register_building(2, "barracks", 10.0F, 0.0F, 0);

  EXPECT_TRUE(registry.is_circle_overlapping_building(0.0F, 0.0F, 0.5F));
  EXPECT_TRUE(registry.is_circle_overlapping_building(10.0F, 0.0F, 0.5F));

  EXPECT_FALSE(registry.is_circle_overlapping_building(5.0F, 0.0F, 0.5F));
}

TEST_F(BuildingCollisionRegistryTest, GridPaddingAccountsForUnitRadius) {
  auto &registry = BuildingCollisionRegistry::instance();

  // Barracks is 4x4, centered at (0,0), so bounds are [-2, 2] x [-2, 2]
  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  // Default grid padding should be at least 1.0 to account for largest unit
  // sizes
  EXPECT_GE(BuildingCollisionRegistry::get_grid_padding(), 1.0F);

  // Get occupied grid cells and verify they include padding
  const auto &buildings = registry.get_all_buildings();
  ASSERT_EQ(buildings.size(), 1);

  auto cells = BuildingCollisionRegistry::get_occupied_grid_cells(
      buildings[0], kDefaultGridCellSize);

  // With padding of 1.0 and building bounds [-2, 2]:
  // min_grid should be floor(-2 - 1.0) = -3
  // max_grid should be ceil(2 + 1.0) = 3
  // So cells should span from -3 to 2 (since loop uses < max_grid)
  bool has_min_x = false;
  bool has_max_x = false;
  for (const auto &cell : cells) {
    if (cell.first <= -2)
      has_min_x = true;
    if (cell.first >= 2)
      has_max_x = true;
  }
  EXPECT_TRUE(has_min_x);
  EXPECT_TRUE(has_max_x);
}

TEST_F(BuildingCollisionRegistryTest, UnitWithLargeRadiusCloseToBuilding) {
  auto &registry = BuildingCollisionRegistry::instance();

  // Barracks is 4x4, centered at (0,0), so bounds are [-2, 2] x [-2, 2]
  registry.register_building(1, "barracks", 0.0F, 0.0F, 0);

  // Unit with radius 1.0 (mounted knight-sized)
  float const unit_radius = 1.0F;

  // Building edge is at x=2. Unit center at x=2.5 with radius 1.0
  // would extend from x=1.5 to x=3.5, overlapping the building
  EXPECT_TRUE(registry.is_circle_overlapping_building(2.5F, 0.0F, unit_radius));

  // Unit center at x=3.0 with radius 1.0 would extend from x=2.0 to x=4.0
  // x=2.0 is exactly at the building edge, so should still overlap
  EXPECT_TRUE(registry.is_circle_overlapping_building(3.0F, 0.0F, unit_radius));

  // Unit center at x=3.1 with radius 1.0 would extend from x=2.1 to x=4.1
  // This should not overlap since building ends at x=2.0
  EXPECT_FALSE(
      registry.is_circle_overlapping_building(3.1F, 0.0F, unit_radius));
}