o3de/Gems/AtomLyIntegration/CommonFeatures/Code/Source/SharedPreview/SharedPreviewContent.cpp

/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include <Atom/Feature/ImageBasedLights/ImageBasedLightFeatureProcessorInterface.h>
#include <Atom/Feature/PostProcess/PostProcessFeatureProcessorInterface.h>
#include <Atom/Feature/SkyBox/SkyBoxFeatureProcessorInterface.h>
#include <Atom/Feature/Utils/LightingPreset.h>
#include <Atom/RPI.Public/Scene.h>
#include <Atom/RPI.Reflect/Asset/AssetUtils.h>
#include <AtomLyIntegration/CommonFeatures/Material/MaterialComponentBus.h>
#include <AtomLyIntegration/CommonFeatures/Material/MaterialComponentConstants.h>
#include <AtomLyIntegration/CommonFeatures/Mesh/MeshComponentBus.h>
#include <AtomLyIntegration/CommonFeatures/Mesh/MeshComponentConstants.h>
#include <AzCore/Asset/AssetCommon.h>
#include <AzCore/Component/Entity.h>
#include <AzCore/Component/TransformBus.h>
#include <AzCore/Math/MatrixUtils.h>
#include <AzCore/Math/Transform.h>
#include <AzFramework/Components/TransformComponent.h>
#include <AzFramework/Entity/EntityContextBus.h>
#include <SharedPreview/SharedPreviewContent.h>

namespace AZ
{
    namespace LyIntegration
    {
        SharedPreviewContent::SharedPreviewContent(
            RPI::ScenePtr scene,
            RPI::ViewPtr view,
            AZ::Uuid entityContextId,
            const Data::AssetId& modelAssetId,
            const Data::AssetId& materialAssetId,
            const Data::AssetId& lightingPresetAssetId,
            const Render::MaterialPropertyOverrideMap& materialPropertyOverrides)
            : m_scene(scene)
            , m_view(view)
            , m_entityContextId(entityContextId)
            , m_materialPropertyOverrides(materialPropertyOverrides)
        {
            // Create preview model
            AzFramework::EntityContextRequestBus::EventResult(
                m_modelEntity, m_entityContextId, &AzFramework::EntityContextRequestBus::Events::CreateEntity, "ThumbnailPreviewModel");
            m_modelEntity->CreateComponent(Render::MeshComponentTypeId);
            m_modelEntity->CreateComponent(Render::MaterialComponentTypeId);
            m_modelEntity->CreateComponent(azrtti_typeid<AzFramework::TransformComponent>());
            m_modelEntity->Init();
            m_modelEntity->Activate();

            m_modelAsset.Create(modelAssetId);
            m_materialAsset.Create(materialAssetId);
            m_lightingPresetAsset.Create(lightingPresetAssetId);
        }

        SharedPreviewContent::~SharedPreviewContent()
        {
            if (m_modelEntity)
            {
                AzFramework::EntityContextRequestBus::Event(
                    m_entityContextId, &AzFramework::EntityContextRequestBus::Events::DestroyEntity, m_modelEntity);
                m_modelEntity = nullptr;
            }
        }

        void SharedPreviewContent::Load()
        {
            m_modelAsset.QueueLoad();
            m_materialAsset.QueueLoad();
            m_lightingPresetAsset.QueueLoad();
        }

        bool SharedPreviewContent::IsReady() const
        {
            return (!m_modelAsset.GetId().IsValid() || m_modelAsset.IsReady()) &&
                (!m_materialAsset.GetId().IsValid() || m_materialAsset.IsReady()) &&
                (!m_lightingPresetAsset.GetId().IsValid() || m_lightingPresetAsset.IsReady());
        }

        bool SharedPreviewContent::IsError() const
        {
            return m_modelAsset.IsError() || m_materialAsset.IsError() || m_lightingPresetAsset.IsError();
        }

        void SharedPreviewContent::ReportErrors()
        {
            AZ_Warning(
                "SharedPreviewContent", !m_modelAsset.GetId().IsValid() || m_modelAsset.IsReady(), "Asset failed to load in time: %s",
                m_modelAsset.ToString<AZStd::string>().c_str());
            AZ_Warning(
                "SharedPreviewContent", !m_materialAsset.GetId().IsValid() || m_materialAsset.IsReady(), "Asset failed to load in time: %s",
                m_materialAsset.ToString<AZStd::string>().c_str());
            AZ_Warning(
                "SharedPreviewContent", !m_lightingPresetAsset.GetId().IsValid() || m_lightingPresetAsset.IsReady(),
                "Asset failed to load in time: %s", m_lightingPresetAsset.ToString<AZStd::string>().c_str());
        }

        void SharedPreviewContent::Update()
        {
            UpdateModel();
            UpdateLighting();
            UpdateCamera();
        }

        void SharedPreviewContent::UpdateModel()
        {
            Render::MeshComponentRequestBus::Event(
                m_modelEntity->GetId(), &Render::MeshComponentRequestBus::Events::SetModelAsset, m_modelAsset);

            Render::MaterialComponentRequestBus::Event(
                m_modelEntity->GetId(), &Render::MaterialComponentRequestBus::Events::SetMaterialOverride,
                Render::DefaultMaterialAssignmentId, m_materialAsset.GetId());

            Render::MaterialComponentRequestBus::Event(
                m_modelEntity->GetId(), &Render::MaterialComponentRequestBus::Events::SetPropertyOverrides,
                Render::DefaultMaterialAssignmentId, m_materialPropertyOverrides);
        }

        void SharedPreviewContent::UpdateLighting()
        {
            if (m_lightingPresetAsset.IsReady())
            {
                auto preset = m_lightingPresetAsset->GetDataAs<Render::LightingPreset>();
                if (preset)
                {
                    auto iblFeatureProcessor = m_scene->GetFeatureProcessor<Render::ImageBasedLightFeatureProcessorInterface>();
                    auto postProcessFeatureProcessor = m_scene->GetFeatureProcessor<Render::PostProcessFeatureProcessorInterface>();
                    auto postProcessSettingInterface = postProcessFeatureProcessor->GetOrCreateSettingsInterface(EntityId());
                    auto exposureControlSettingInterface = postProcessSettingInterface->GetOrCreateExposureControlSettingsInterface();
                    auto directionalLightFeatureProcessor =
                        m_scene->GetFeatureProcessor<Render::DirectionalLightFeatureProcessorInterface>();
                    auto skyboxFeatureProcessor = m_scene->GetFeatureProcessor<Render::SkyBoxFeatureProcessorInterface>();
                    skyboxFeatureProcessor->Enable(true);
                    skyboxFeatureProcessor->SetSkyboxMode(Render::SkyBoxMode::Cubemap);

                    Camera::Configuration cameraConfig;
                    cameraConfig.m_fovRadians = FieldOfView;
                    cameraConfig.m_nearClipDistance = NearDist;
                    cameraConfig.m_farClipDistance = FarDist;
                    cameraConfig.m_frustumWidth = 100.0f;
                    cameraConfig.m_frustumHeight = 100.0f;

                    AZStd::vector<Render::DirectionalLightFeatureProcessorInterface::LightHandle> lightHandles;

                    preset->ApplyLightingPreset(
                        iblFeatureProcessor, skyboxFeatureProcessor, exposureControlSettingInterface, directionalLightFeatureProcessor,
                        cameraConfig, lightHandles);
                }
            }
        }

        void SharedPreviewContent::UpdateCamera()
        {
            // Get bounding sphere of the model asset and estimate how far the camera needs to be see all of it
            Vector3 center = {};
            float radius = {};
            if (m_modelAsset.IsReady())
            {
                m_modelAsset->GetAabb().GetAsSphere(center, radius);
            }

            const auto distance = radius + NearDist;
            const auto cameraRotation = Quaternion::CreateFromAxisAngle(Vector3::CreateAxisZ(), CameraRotationAngle);
            const auto cameraPosition = center + cameraRotation.TransformVector(Vector3(0.0f, distance, 0.0f));
            const auto cameraTransform = Transform::CreateLookAt(cameraPosition, center);
            m_view->SetCameraTransform(Matrix3x4::CreateFromTransform(cameraTransform));
        }
    } // namespace LyIntegration
} // namespace AZ