o3de/Gems/Atom/Feature/Common/Code/Source/SkinnedMesh/SkinnedMeshComputePass.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/SkinnedMesh/SkinnedMeshOutputStreamManagerInterface.h>
#include <Atom/RPI.Public/Scene.h>
#include <SkinnedMesh/SkinnedMeshComputePass.h>
#include <SkinnedMesh/SkinnedMeshFeatureProcessor.h>

#include <Atom/RPI.Public/Shader/Shader.h>

#include <Atom/RHI/CommandList.h>

namespace AZ
{
    namespace Render
    {
        SkinnedMeshComputePass::SkinnedMeshComputePass(const RPI::PassDescriptor& descriptor)
            : RPI::ComputePass(descriptor)
        {
        }

        RPI::Ptr<SkinnedMeshComputePass> SkinnedMeshComputePass::Create(const RPI::PassDescriptor& descriptor)
        {
            RPI::Ptr<SkinnedMeshComputePass> pass = aznew SkinnedMeshComputePass(descriptor);
            return pass;
        }

        Data::Instance<RPI::Shader> SkinnedMeshComputePass::GetShader() const
        {
            return m_shader;
        }

        void SkinnedMeshComputePass::SetupFrameGraphDependencies(RHI::FrameGraphInterface frameGraph)
        {
            if (auto featureProcessor = GetSkinnedMeshFeatureProcessor())
            {
                featureProcessor->SetupSkinningScope(frameGraph, Pass::GetDeviceIndex());
            }

            ComputePass::SetupFrameGraphDependencies(frameGraph);
        }

        void SkinnedMeshComputePass::BuildCommandListInternal(const RHI::FrameGraphExecuteContext& context)
        {
            if (auto featureProcessor = GetSkinnedMeshFeatureProcessor())
            {
                SetSrgsForDispatch(context);

                featureProcessor->SubmitSkinningDispatchItems(
                    context, context.GetSubmitRange().m_startIndex, context.GetSubmitRange().m_endIndex);
            }
        }

        void SkinnedMeshComputePass::OnShaderReinitialized(const RPI::Shader& shader)
        {
            ComputePass::OnShaderReinitialized(shader);
            if (auto featureProcessor = GetSkinnedMeshFeatureProcessor())
            {
                featureProcessor->OnSkinningShaderReinitialized(m_shader);
            }
        }

        void SkinnedMeshComputePass::OnShaderVariantReinitialized(const RPI::ShaderVariant& shaderVariant)
        {
            ComputePass::OnShaderVariantReinitialized(shaderVariant);
            if (auto featureProcessor = GetSkinnedMeshFeatureProcessor())
            {
                featureProcessor->OnSkinningShaderReinitialized(m_shader);
            }
        }

        SkinnedMeshFeatureProcessor* SkinnedMeshComputePass::GetSkinnedMeshFeatureProcessor()
        {
            AZ::RPI::Scene* scene{ GetScene() };
            if (!scene)
            {
                return nullptr;
            }
            auto result = scene->GetFeatureProcessor<SkinnedMeshFeatureProcessor>();
            return result;
        }
    }   // namespace Render
}   // namespace AZ