#include "BaseTestCase.h"

#include "SpaceSegment.h"
#include "SpaceGridIterator.h"

class CollectGridIterator: public SpaceGridIterator
{
public:
    std::vector<Vector3> locations;
protected:
    virtual bool onCell(int x, int y, int z) override
    {
        locations.push_back(Vector3(x, y, z));
        return true;
    }
};

TEST(SpaceSegment, iterateOnGrid)
{
    CollectGridIterator it;
    SpaceSegment segment(Vector3(0.5, 1.5, 0.0), Vector3(2.5, 0.5, 0.0));
    segment.iterateOnGrid(it);

    ASSERT_EQ(4, it.locations.size());
    EXPECT_VECTOR3_COORDS(it.locations[0], 0.0, 1.0, 0.0);
    EXPECT_VECTOR3_COORDS(it.locations[1], 1.0, 1.0, 0.0);
    EXPECT_VECTOR3_COORDS(it.locations[2], 1.0, 0.0, 0.0);
    EXPECT_VECTOR3_COORDS(it.locations[3], 2.0, 0.0, 0.0);
}

TEST(SpaceSegment, iterateOnGrid_Corner)
{
    CollectGridIterator it;
    SpaceSegment segment(Vector3(0.5, 0.5, 0.5), Vector3(2.5, 2.5, 2.5));
    segment.iterateOnGrid(it);

    ASSERT_EQ(7, it.locations.size());
    EXPECT_VECTOR3_COORDS(it.locations[0], 0.0, 0.0, 0.0);
    EXPECT_VECTOR3_COORDS(it.locations[3], 1.0, 1.0, 1.0);
    EXPECT_VECTOR3_COORDS(it.locations[6], 2.0, 2.0, 2.0);
}

TEST(SpaceSegment, iterateOnGrid_OneCell)
{
    CollectGridIterator it;
    SpaceSegment segment(Vector3(8.1, 8.2, 8.9), Vector3(8.9, 8.3, 8.6));
    segment.iterateOnGrid(it);

    ASSERT_EQ(1, it.locations.size());
    EXPECT_VECTOR3_COORDS(it.locations[0], 8.0, 8.0, 8.0);
}

TEST(SpaceSegment, iterateOnGrid_Negative)
{
    CollectGridIterator it;
    SpaceSegment segment(Vector3(-8.1, -8.2, -8.9), Vector3(-8.9, -8.3, -8.6));
    segment.iterateOnGrid(it);

    ASSERT_EQ(1, it.locations.size());
    EXPECT_VECTOR3_COORDS(it.locations[0], -9.0, -9.0, -9.0);
}