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/DiffuseGlobalIllumination/DiffuseProbeGridTextureRead...

61 lines
2.3 KiB
C++

/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#pragma once
#include <Atom/RHI/ScopeProducer.h>
#include <Atom/RPI.Public/Pass/AttachmentReadback.h>
#include <Atom/RPI.Public/Pass/Pass.h>
#include <Atom/Feature/DiffuseGlobalIllumination/DiffuseProbeGridFeatureProcessorInterface.h>
namespace AZ
{
namespace Render
{
class DiffuseProbeGrid;
enum class DiffuseProbeGridReadbackState
{
Idle,
Irradiance,
Distance,
Relocation,
Classification,
Complete
};
//! This class contains functionality necessary to read back the DiffuseProbeGrid textures, which
//! allows them to be saved as assets to run the DiffuseProbeGrid in non-realtime mode.
class DiffuseProbeGridTextureReadback final
{
public:
DiffuseProbeGridTextureReadback(DiffuseProbeGrid* diffuseProbeGrid);
~DiffuseProbeGridTextureReadback() = default;
void BeginTextureReadback(DiffuseProbeGridBakeTexturesCallback callback);
void Update(const AZ::Name& passName);
void FrameBegin(AZ::RPI::Pass::FramePrepareParams& params);
private:
DiffuseProbeGrid* m_diffuseProbeGrid = nullptr;
DiffuseProbeGridReadbackState m_readbackState = DiffuseProbeGridReadbackState::Idle;
AZStd::shared_ptr<AZ::RPI::AttachmentReadback> m_attachmentReadback;
DiffuseProbeGridBakeTexturesCallback m_callback;
AZ::RPI::AttachmentReadback::ReadbackResult m_irradianceReadbackResult;
AZ::RPI::AttachmentReadback::ReadbackResult m_distanceReadbackResult;
AZ::RPI::AttachmentReadback::ReadbackResult m_relocationReadbackResult;
AZ::RPI::AttachmentReadback::ReadbackResult m_classificationReadbackResult;
};
} // namespace Render
} // namespace AZ