lumberyard-employee-dm
b9824ed172
* Updated all array_view uses with the C++20 span. The updates were done in the following order 1. `AZStd::array_view<([^>].+)\* ?>` -> `AZStd::span<\1 const>` 2. `AZStd::array_view<(?:const )(.+)>` -> `AZStd::span<const \1>` 3. `AZStd::array_view` -> `AZStd::span` Removed the implementation of array_view. Signed-off-by: lumberyard-employee-dm <56135373+lumberyard-employee-dm@users.noreply.github.com> * Added missing whitespace between `const` and the typename for spans. Updated the ShaderTest comparison of the ShaderResourceGroupLayout span to compare the sizes as well Updated comments on some of the methods that stated that they return "an array" to mention they return "a span". Signed-off-by: lumberyard-employee-dm <56135373+lumberyard-employee-dm@users.noreply.github.com>
102 lines
3.4 KiB
C++
102 lines
3.4 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
|
|
*
|
|
*/
|
|
|
|
#include <TerrainRenderer/BindlessImageArrayHandler.h>
|
|
#include <Atom/RPI.Reflect/Image/Image.h>
|
|
#include <Atom/RPI.Public/Image/ImageSystemInterface.h>
|
|
|
|
namespace AZ::Render
|
|
{
|
|
namespace
|
|
{
|
|
[[maybe_unused]] const char* BindlessImageArrayHandlerName = "TerrainFeatureProcessor";
|
|
}
|
|
|
|
void BindlessImageArrayHandler::Initialize(AZ::Data::Instance<AZ::RPI::ShaderResourceGroup>& srg, const AZ::Name& propertyName)
|
|
{
|
|
if (!m_isInitialized)
|
|
{
|
|
m_isInitialized = UpdateSrgIndices(srg, propertyName);
|
|
}
|
|
else
|
|
{
|
|
AZ_Error(BindlessImageArrayHandlerName, false, "Already initialized.");
|
|
}
|
|
}
|
|
|
|
void BindlessImageArrayHandler::Reset()
|
|
{
|
|
m_texturesIndex = {};
|
|
m_isInitialized = false;
|
|
}
|
|
|
|
bool BindlessImageArrayHandler::IsInitialized() const
|
|
{
|
|
return m_isInitialized;
|
|
}
|
|
|
|
bool BindlessImageArrayHandler::UpdateSrgIndices(AZ::Data::Instance<AZ::RPI::ShaderResourceGroup>& srg, const AZ::Name& propertyName)
|
|
{
|
|
if (srg)
|
|
{
|
|
m_texturesIndex = srg->GetLayout()->FindShaderInputImageUnboundedArrayIndex(propertyName);
|
|
AZ_Error(BindlessImageArrayHandlerName, m_texturesIndex.IsValid(), "Failed to find srg input constant %s.", propertyName.GetCStr());
|
|
}
|
|
else
|
|
{
|
|
AZ_Error(BindlessImageArrayHandlerName, false, "Cannot initialize using a null shader resource group.");
|
|
}
|
|
return m_texturesIndex.IsValid();
|
|
}
|
|
|
|
uint16_t BindlessImageArrayHandler::AppendBindlessImage(const AZ::RHI::ImageView* imageView)
|
|
{
|
|
uint16_t imageIndex = 0xFFFF;
|
|
|
|
AZStd::unique_lock<AZStd::shared_mutex> lock(m_updateMutex);
|
|
if (m_bindlessImageViewFreeList.size() > 0)
|
|
{
|
|
imageIndex = m_bindlessImageViewFreeList.back();
|
|
m_bindlessImageViewFreeList.pop_back();
|
|
m_bindlessImageViews.at(imageIndex) = imageView;
|
|
}
|
|
else
|
|
{
|
|
imageIndex = aznumeric_cast<uint16_t>(m_bindlessImageViews.size());
|
|
m_bindlessImageViews.push_back(imageView);
|
|
}
|
|
return imageIndex;
|
|
}
|
|
|
|
void BindlessImageArrayHandler::UpdateBindlessImage(uint16_t index, const AZ::RHI::ImageView* imageView)
|
|
{
|
|
AZStd::shared_lock<AZStd::shared_mutex> lock(m_updateMutex);
|
|
m_bindlessImageViews.at(index) = imageView;
|
|
}
|
|
|
|
void BindlessImageArrayHandler::RemoveBindlessImage(uint16_t index)
|
|
{
|
|
AZStd::unique_lock<AZStd::shared_mutex> lock(m_updateMutex);
|
|
m_bindlessImageViews.at(index) = AZ::RPI::ImageSystemInterface::Get()->GetSystemImage(AZ::RPI::SystemImage::Magenta)->GetImageView();
|
|
m_bindlessImageViewFreeList.push_back(index);
|
|
}
|
|
|
|
bool BindlessImageArrayHandler::UpdateSrg(AZ::Data::Instance<AZ::RPI::ShaderResourceGroup>& srg) const
|
|
{
|
|
if (!m_isInitialized)
|
|
{
|
|
AZ_Error("BindlessImageArrayHandler", false, "BindlessImageArrayHandler not initialized")
|
|
return false;
|
|
}
|
|
|
|
AZStd::span<const AZ::RHI::ImageView* const> imageViews(m_bindlessImageViews.data(), m_bindlessImageViews.size());
|
|
return srg->SetImageViewUnboundedArray(m_texturesIndex, imageViews);
|
|
}
|
|
|
|
}
|