o3de-atom-sampleviewer/Gem/Code/Lib/MaterialFunctors/StacksShaderCollectionFunctor.cpp

/*
* 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.
*
*/

#include <MaterialFunctors/StacksShaderCollectionFunctor.h>
#include <Atom/RPI.Public/Material/Material.h>
#include <Atom/RPI.Reflect/Shader/ShaderOptionGroup.h>

namespace AtomSampleViewer
{
    void StacksShaderCollectionFunctor::Reflect(AZ::ReflectContext* context)
    {
        if (auto* serializeContext = azrtti_cast<AZ::SerializeContext*>(context))
        {
            serializeContext->Class<StacksShaderCollectionFunctor, AZ::RPI::MaterialFunctor>()
                ->Version(1)
                ->Field("m_stackCountProperty", &StacksShaderCollectionFunctor::m_stackCountProperty)
                ->Field("m_highlightLastStackProperty", &StacksShaderCollectionFunctor::m_highlightLastStackProperty)
                ->Field("m_highlightLastStackOption", &StacksShaderCollectionFunctor::m_highlightLastStackOption)
                ;
        }
    }

    void StacksShaderCollectionFunctor::Process(RuntimeContext& context)
    {
        using namespace AZ::RPI;

        const uint32_t stackCount = context.GetMaterialPropertyValue<uint32_t>(m_stackCountProperty);
        const bool highlightLastStack = context.GetMaterialPropertyValue<bool>(m_highlightLastStackProperty);

        static const int AvailableStackCount = 4;
        for (uint32_t i = 0; i < AvailableStackCount; ++i)
        {
            const bool isLastStack = (i == stackCount - 1);
            const bool shouldHighlight = highlightLastStack && isLastStack;
            context.SetShaderOptionValue(i, m_highlightLastStackOption, ShaderOptionValue{ shouldHighlight });
            context.SetShaderEnabled(i, i < stackCount);
        }
    }
} // namespace AtomSampleViewer