#include <catch2/catch_test_macros.hpp>
#include <memory>
#include "Engine/Operator/Attribute.h"
#include "Engine/Operator/AttributeHandle.h"
#include "Engine/Types.h"
#include "Engine/Operator/Geometry.h"
// #include "Engine/Operator/Primitive.h"


TEST_CASE("attrHandleInt")
{
    using namespace enzo;

    auto myAttrib = std::make_shared<ga::Attribute>("test", ga::AttrType::intT);
    ga::AttributeHandleInt myHandle(myAttrib);
    myHandle.addValue(5);
    myHandle.addValue(6);
    REQUIRE(myHandle.getValue(0)==5);
    REQUIRE(myHandle.getValue(1)==6);
}

TEST_CASE("geometry")
{
    using namespace enzo;
    geo::Geometry geo;
    // check add function
    ga::AttributeHandle<int> myHandle = geo.addIntAttribute(ga::AttrOwner::POINT, "index");
    myHandle.addValue(5);
    myHandle.addValue(6);
    REQUIRE(myHandle.getValue(0)==5);
    REQUIRE(myHandle.getValue(1)==6);

    // check getter
    std::shared_ptr<ga::Attribute> myAttribute = geo.getAttribByName(ga::AttrOwner::POINT, "index");
    ga::AttributeHandle<int> myHandle2(myAttribute);
    REQUIRE(myHandle2.getValue(0)==5);
    REQUIRE(myHandle2.getValue(1)==6);

    // check failed lookup
    std::shared_ptr<ga::Attribute> myAttribute2 = geo.getAttribByName(ga::AttrOwner::POINT, "nonExistant");
    REQUIRE(myAttribute2==nullptr);

}

TEST_CASE("attrHandleFloat")
{
    using namespace enzo;

    std::shared_ptr<ga::Attribute> myAttrib = std::make_shared<ga::Attribute>("test", ga::AttrType::floatT);
    ga::AttributeHandleFloat myHandle(myAttrib);
    myHandle.addValue(5.3f);
    myHandle.addValue(6.9f);
    REQUIRE(myHandle.getValue(0)==5.3f);
    REQUIRE(myHandle.getValue(1)==6.9f);
}

TEST_CASE("Attribute Type")
{
    using namespace enzo;
    REQUIRE(ga::AttributeType::intT ==      ga::AttributeType::intT);
    REQUIRE(ga::AttributeType::intT !=      ga::AttributeType::floatT);
    REQUIRE(ga::AttributeType::floatT ==    ga::AttributeType::floatT);
    REQUIRE(ga::AttributeType::listT ==     ga::AttributeType::listT);
    REQUIRE(ga::AttributeType::vectorT ==   ga::AttributeType::vectorT);

}