/*
* 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 <PostProcessing/LookModificationTransformPass.h>
#include <PostProcess/PostProcessFeatureProcessor.h>
#include <PostProcessing/BlendColorGradingLutsPass.h>
#include <PostProcessing/LookModificationCompositePass.h>

namespace AZ
{
    namespace Render
    {
        RPI::Ptr<LookModificationPass> LookModificationPass::Create(const RPI::PassDescriptor& descriptor)
        {
            RPI::Ptr<LookModificationPass> pass = aznew LookModificationPass(descriptor);
            return pass;
        }

        LookModificationPass::LookModificationPass(const RPI::PassDescriptor& descriptor)
            : RPI::ParentPass(descriptor)
        {
            AzFramework::NativeWindowHandle windowHandle = nullptr;
            AzFramework::WindowSystemRequestBus::BroadcastResult(
                windowHandle,
                &AzFramework::WindowSystemRequestBus::Events::GetDefaultWindowHandle);
        }

        LookModificationPass::~LookModificationPass()
        {
        }

        void LookModificationPass::BuildAttachmentsInternal()
        {
            RPI::PassAttachmentRef passAttachmentRef;
            passAttachmentRef.m_pass = "Parent";
            passAttachmentRef.m_attachment = "SwapChainOutput";
            m_swapChainAttachmentBinding = FindAdjacentBinding(passAttachmentRef);
            AZ_Assert(m_swapChainAttachmentBinding->m_attachment->GetAttachmentType() == RHI::AttachmentType::Image, "Invalid attachment");
            ParentPass::BuildAttachmentsInternal();
        }

        void LookModificationPass::FrameBeginInternal(FramePrepareParams params)
        {
            AZ_UNUSED(params);
            const RHI::TransientImageDescriptor imageDesc = m_swapChainAttachmentBinding->m_attachment->GetTransientImageDescriptor();
            if (imageDesc.m_imageDescriptor.m_format != m_displayBufferFormat)
            {
                m_displayBufferFormat = imageDesc.m_imageDescriptor.m_format;
                m_outputDeviceTransformType = AcesDisplayMapperFeatureProcessor::GetOutputDeviceTransformType(m_displayBufferFormat);
                m_shaperParams = GetAcesShaperParameters(m_outputDeviceTransformType);

                // Update the children passes
                for (const AZ::RPI::Ptr<Pass>& child : m_children)
                {
                    BlendColorGradingLutsPass* blendPass = azrtti_cast<BlendColorGradingLutsPass*>(child.get());
                    if (blendPass)
                    {
                        blendPass->SetShaperParameters(m_shaperParams);
                        continue;
                    }
                    LookModificationCompositePass* compositePass = azrtti_cast<LookModificationCompositePass*>(child.get());
                    if (compositePass)
                    {
                        compositePass->SetShaperParameters(m_shaperParams);
                        continue;
                    }
                }
            }
            ParentPass::FrameBeginInternal(params);
        }
    }   // namespace Render
}   // namespace AZ