#include <gtest/gtest.h>

#include <XCEngine/Resources/Volume/VolumeField.h>

#include <utility>

using namespace XCEngine::Resources;

namespace {

TEST(VolumeField, CreatePreservesPayloadAndMetadata) {
    const unsigned char payload[] = { 1, 2, 3, 4, 5, 6 };

    VolumeField volumeField;
    IResource::ConstructParams params;
    params.name = "cloud.nvdb";
    params.path = "Assets/cloud.nvdb";
    params.guid = ResourceGUID::Generate(params.path);
    volumeField.Initialize(params);

    XCEngine::Math::Bounds bounds;
    bounds.SetMinMax(
        XCEngine::Math::Vector3(-1.0f, -2.0f, -3.0f),
        XCEngine::Math::Vector3(4.0f, 5.0f, 6.0f));

    ASSERT_TRUE(volumeField.Create(
        VolumeStorageKind::NanoVDB,
        payload,
        sizeof(payload),
        bounds,
        XCEngine::Math::Vector3(0.5f, 0.25f, 0.125f),
        VolumeIndexBounds{ -4, -5, -6, 7, 8, 9 },
        1u,
        2u));

    EXPECT_TRUE(volumeField.IsValid());
    EXPECT_EQ(volumeField.GetType(), ResourceType::VolumeField);
    EXPECT_EQ(volumeField.GetStorageKind(), VolumeStorageKind::NanoVDB);
    EXPECT_EQ(volumeField.GetPayloadSize(), sizeof(payload));
    EXPECT_EQ(volumeField.GetBounds().GetMin(), XCEngine::Math::Vector3(-1.0f, -2.0f, -3.0f));
    EXPECT_EQ(volumeField.GetBounds().GetMax(), XCEngine::Math::Vector3(4.0f, 5.0f, 6.0f));
    EXPECT_EQ(volumeField.GetVoxelSize(), XCEngine::Math::Vector3(0.5f, 0.25f, 0.125f));
    EXPECT_EQ(volumeField.GetIndexBounds(), (VolumeIndexBounds{ -4, -5, -6, 7, 8, 9 }));
    EXPECT_EQ(volumeField.GetGridType(), 1u);
    EXPECT_EQ(volumeField.GetGridClass(), 2u);
    EXPECT_EQ(static_cast<const unsigned char*>(volumeField.GetPayloadData())[0], 1u);
    EXPECT_GT(volumeField.GetMemorySize(), sizeof(VolumeField));
}

TEST(VolumeField, CreateOwnedPreservesPayloadAndMetadata) {
    XCEngine::Containers::Array<unsigned char> payload;
    payload.Resize(4);
    payload[0] = 9u;
    payload[1] = 8u;
    payload[2] = 7u;
    payload[3] = 6u;

    VolumeField volumeField;
    IResource::ConstructParams params;
    params.name = "cloud.xcvol";
    params.path = "Library/Artifacts/ab/main.xcvol";
    params.guid = ResourceGUID::Generate(params.path);
    volumeField.Initialize(params);

    ASSERT_TRUE(volumeField.CreateOwned(
        VolumeStorageKind::NanoVDB,
        std::move(payload),
        XCEngine::Math::Bounds(),
        XCEngine::Math::Vector3(1.0f, 2.0f, 3.0f),
        VolumeIndexBounds{ 1, 2, 3, 4, 5, 6 },
        3u,
        4u));

    EXPECT_TRUE(volumeField.IsValid());
    EXPECT_TRUE(payload.Empty());
    EXPECT_EQ(volumeField.GetPayloadSize(), 4u);
    EXPECT_EQ(static_cast<const unsigned char*>(volumeField.GetPayloadData())[0], 9u);
    EXPECT_EQ(volumeField.GetVoxelSize(), XCEngine::Math::Vector3(1.0f, 2.0f, 3.0f));
    EXPECT_EQ(volumeField.GetIndexBounds(), (VolumeIndexBounds{ 1, 2, 3, 4, 5, 6 }));
    EXPECT_EQ(volumeField.GetGridType(), 3u);
    EXPECT_EQ(volumeField.GetGridClass(), 4u);
}

} // namespace