You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Gems/Atom/Feature/Common/Code/Source/PostProcessing/DepthOfFieldWriteFocusDepth...

54 lines
2.1 KiB
C++

/*
* 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
*
*/
#pragma once
#include <AzCore/Memory/SystemAllocator.h>
#include <Atom/RPI.Public/Pass/ComputePass.h>
#include <Atom/RPI.Public/Buffer/Buffer.h>
namespace AZ
{
namespace Render
{
//! This pass is used to write the depth value of the specified screen coordinates to the buffer.
class DepthOfFieldWriteFocusDepthFromGpuPass final
: public RPI::ComputePass
{
AZ_RPI_PASS(DepthOfFieldWriteFocusDepthFromGpuPass);
public:
AZ_RTTI(DepthOfFieldWriteFocusDepthFromGpuPass, "{60DF04D2-A9FE-4B21-8050-96AFFC46BB87}", RPI::ComputePass);
AZ_CLASS_ALLOCATOR(DepthOfFieldWriteFocusDepthFromGpuPass, SystemAllocator, 0);
/// Creates a DepthOfFieldWriteFocusDepthFromGpuPass
static RPI::Ptr<DepthOfFieldWriteFocusDepthFromGpuPass> Create(const RPI::PassDescriptor& descriptor);
// Set pass parameter interfaces...
void SetScreenPosition(const AZ::Vector2& screenPosition);
void SetBufferRef(RPI::Ptr<RPI::Buffer> bufferRef);
private:
DepthOfFieldWriteFocusDepthFromGpuPass(const RPI::PassDescriptor& descriptor);
// SRG binding indices...
RHI::ShaderInputNameIndex m_autoFocusDataBufferIndex = "m_outputFocusDepth";
RHI::ShaderInputNameIndex m_autoFocusScreenPositionIndex = "m_autoFocusScreenPosition";
RPI::Ptr<RPI::Buffer> m_bufferRef = nullptr;
AZ::Vector2 m_autoFocusScreenPosition{ 0.0f, 0.0f };
// Scope producer functions...
void SetupFrameGraphDependencies(RHI::FrameGraphInterface frameGraph) override;
void CompileResources(const RHI::FrameGraphCompileContext& context) override;
// Pass overrides
void BuildInternal() override;
};
} // namespace Render
} // namespace AZ