#include <gtest/gtest.h>

#include "../VolumeIntegrationSceneFixture.h"

#include <XCEngine/Components/CameraComponent.h>
#include <XCEngine/Components/GameObject.h>
#include <XCEngine/Components/VolumeRendererComponent.h>
#include <XCEngine/Core/Math/Color.h>
#include <XCEngine/Core/Math/Vector3.h>
#include <XCEngine/Rendering/Extraction/RenderSceneExtractor.h>

using namespace VolumeIntegrationTestUtils;

namespace {

constexpr const char* kD3D12Screenshot = "volume_scene_d3d12.ppm";

class VolumeSceneTest : public VolumeIntegrationSceneFixture {
protected:
    const char* GetSceneName() const override { return "VolumeScene"; }
    void BuildScene() override;
    void ReleaseSceneResources() override;

private:
    Material* mVolumeMaterial = nullptr;
    VolumeField* mVolumeField = nullptr;
};

void VolumeSceneTest::BuildScene() {
    mVolumeMaterial = CreateVolumetricMaterial(
        "VolumeMaterial",
        "Tests/Rendering/VolumeScene/Volume.material");
    ASSERT_NE(mVolumeMaterial, nullptr);

    mVolumeField = LoadCloudVolumeField();
    ASSERT_NE(mVolumeField, nullptr);

    GameObject* cameraObject = mScene->CreateGameObject("MainCamera");
    auto* camera = cameraObject->AddComponent<CameraComponent>();
    camera->SetPrimary(true);
    camera->SetFieldOfView(45.0f);
    camera->SetNearClipPlane(0.1f);
    camera->SetFarClipPlane(5000.0f);
    camera->SetClearColor(XCEngine::Math::Color(0.03f, 0.04f, 0.06f, 1.0f));
    cameraObject->GetTransform()->SetLocalPosition(Vector3(-10.0f, 300.0f, -1200.0f));
    cameraObject->GetTransform()->LookAt(Vector3(-10.0f, 73.0f, 0.0f));

    GameObject* volumeObject = mScene->CreateGameObject("CloudVolume");
    auto* volumeRenderer = volumeObject->AddComponent<VolumeRendererComponent>();
    volumeRenderer->SetVolumeField(mVolumeField);
    volumeRenderer->SetMaterial(mVolumeMaterial);
    volumeRenderer->SetCastShadows(false);
    volumeRenderer->SetReceiveShadows(false);

    RenderSceneExtractor extractor;
    const RenderSceneData sceneData = extractor.Extract(*mScene, nullptr, kFrameWidth, kFrameHeight);
    ASSERT_EQ(sceneData.visibleItems.size(), 0u);
    ASSERT_EQ(sceneData.visibleVolumes.size(), 1u);
}

void VolumeSceneTest::ReleaseSceneResources() {
    delete mVolumeField;
    mVolumeField = nullptr;
    delete mVolumeMaterial;
    mVolumeMaterial = nullptr;
}

TEST_P(VolumeSceneTest, RenderNanoVdbVolumeScene) {
    RenderAndCompare(kD3D12Screenshot, 0.0f);
}

} // namespace

INSTANTIATE_TEST_SUITE_P(D3D12, VolumeSceneTest, ::testing::Values(RHIType::D3D12));

GTEST_API_ int main(int argc, char** argv) {
    return RunRenderingIntegrationTestMain(argc, argv);
}