/*
* 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 <AtomBridge/PerViewportDynamicDrawInterface.h>

namespace AZ::AtomBridge
{
    class PerViewportDynamicDrawManager final : public PerViewportDynamicDrawInterface
    {
    public:
        AZ_TYPE_INFO(PerViewportDynamicDrawManager, "{BED66185-00A7-43F7-BD28-C56BC8E4C535}");

        PerViewportDynamicDrawManager();
        ~PerViewportDynamicDrawManager();

        // PerViewportDynamicDrawInterface overrides...
        void RegisterDynamicDrawContext(AZ::Name name, DrawContextFactory contextInitializer) override;
        void UnregisterDynamicDrawContext(AZ::Name name) override;
        RHI::Ptr<RPI::DynamicDrawContext> GetDynamicDrawContextForViewport(AZ::Name name, AzFramework::ViewportId viewportId) override;

    private:
        struct ViewportData
        {
            AZStd::unordered_map<AZ::Name, RHI::Ptr<RPI::DynamicDrawContext>> m_dynamicDrawContexts;

            // Event handlers
            AZ::Event<RPI::RenderPipelinePtr>::Handler m_pipelineChangedHandler;
            AZ::Event<AzFramework::ViewportId>::Handler m_viewportDestroyedHandler;

            // Cached state
            bool m_initialized = false;
        };
        AZStd::map<AzFramework::ViewportId, ViewportData> m_viewportData;
        AZStd::unordered_map<AZ::Name, DrawContextFactory> m_registeredDrawContexts;
        AZStd::mutex m_mutexDrawContexts;
    };
} //namespace AZ::AtomBridge