#include <gtest/gtest.h>

#include "../VolumeIntegrationSceneFixture.h"

#include <XCEngine/Components/CameraComponent.h>
#include <XCEngine/Components/GameObject.h>
#include <XCEngine/Components/LightComponent.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";
constexpr const char* kOpenGLScreenshot = "volume_scene_opengl.ppm";
constexpr const char* kVulkanScreenshot = "volume_scene_vulkan.ppm";

const char* GetScreenshotFilename(RHIType backendType) {
    switch (backendType) {
        case RHIType::OpenGL:
            return kOpenGLScreenshot;
        case RHIType::Vulkan:
            return kVulkanScreenshot;
        case RHIType::D3D12:
        default:
            return kD3D12Screenshot;
    }
}

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",
        Vector4(1.0f, 1.0f, 1.0f, 1.0f),
        0.2f,
        1.0f,
        2000.0f,
        0.005f,
        Vector3(-0.72f, 0.16f, 0.67f),
        8.0f);
    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* lightObject = mScene->CreateGameObject("MainDirectionalLight");
    auto* light = lightObject->AddComponent<LightComponent>();
    light->SetLightType(LightType::Directional);
    light->SetColor(XCEngine::Math::Color(1.0f, 1.0f, 1.0f, 1.0f));
    light->SetIntensity(1.0f);
    lightObject->GetTransform()->LookAt(Vector3(-0.5f, -0.8f, -0.3f));

    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);
    ASSERT_TRUE(sceneData.lighting.HasMainDirectionalLight());
}

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

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

} // namespace

INSTANTIATE_TEST_SUITE_P(D3D12, VolumeSceneTest, ::testing::Values(RHIType::D3D12));
#if defined(XCENGINE_SUPPORT_OPENGL)
INSTANTIATE_TEST_SUITE_P(OpenGL, VolumeSceneTest, ::testing::Values(RHIType::OpenGL));
#endif
#if defined(XCENGINE_SUPPORT_VULKAN)
INSTANTIATE_TEST_SUITE_P(Vulkan, VolumeSceneTest, ::testing::Values(RHIType::Vulkan));
#endif

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