Restore Editor viewport icon rendering (#879)

This introduces an EditorViewportIconDisplayInterface that will eventually be used to outright remove CIconManager, it provides a simple interface for loading 2D image assets and rendering them on-screen. It also introduces AtomBridge::PerViewportDynamicDraw for getting a dynamic draw instance on a per-viewport basis
2021-05-26 16:36:54 -07:00
parent dc0f1ff0b1
commit d1863c6c5b
25 changed files with 1050 additions and 24 deletions
@@ -0,0 +1,119 @@
+/*
+* 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 "PerViewportDynamicDrawManager.h"
+
+#include <Atom/RPI.Public/ViewportContextBus.h>
+#include <Atom/RPI.Public/ViewportContext.h>
+
+namespace AZ::AtomBridge
+{
+    PerViewportDynamicDrawManager::PerViewportDynamicDrawManager()
+    {
+        PerViewportDynamicDraw::Register(this);
+    }
+
+    PerViewportDynamicDrawManager::~PerViewportDynamicDrawManager()
+    {
+        PerViewportDynamicDraw::Unregister(this);
+    }
+
+    void PerViewportDynamicDrawManager::RegisterDynamicDrawContext(AZ::Name name, DrawContextFactory contextInitializer)
+    {
+        AZStd::lock_guard lock(m_mutexDrawContexts);
+
+        const bool alreadyRegistered = m_registeredDrawContexts.find(name) != m_registeredDrawContexts.end();
+        AZ_Error("AtomBridge", !alreadyRegistered, "Attempted to call RegisterDynamicDrawContext for already registered name: \"%s\"", name.GetCStr());
+        if (alreadyRegistered)
+        {
+            return;
+        }
+        m_registeredDrawContexts[name] = contextInitializer;
+    }
+
+    void PerViewportDynamicDrawManager::UnregisterDynamicDrawContext(AZ::Name name)
+    {
+        AZStd::lock_guard lock(m_mutexDrawContexts);
+
+        auto drawContextFactoryIt = m_registeredDrawContexts.find(name);
+        const bool registered = drawContextFactoryIt != m_registeredDrawContexts.end();
+        AZ_Error("AtomBridge", registered, "Attempted to call UnregisterDynamicDrawContext for unregistered name: \"%s\"", name.GetCStr());
+        if (!registered)
+        {
+            return;
+        }
+        m_registeredDrawContexts.erase(drawContextFactoryIt);
+
+        for (auto& viewportData : m_viewportData)
+        {
+            viewportData.second.m_dynamicDrawContexts.erase(name);
+        }
+    }
+
+    RHI::Ptr<RPI::DynamicDrawContext> PerViewportDynamicDrawManager::GetDynamicDrawContextForViewport(
+        AZ::Name name, AzFramework::ViewportId viewportId)
+    {
+        AZStd::lock_guard lock(m_mutexDrawContexts);
+
+        auto contextFactoryIt = m_registeredDrawContexts.find(name);
+        if (contextFactoryIt == m_registeredDrawContexts.end())
+        {
+            return nullptr;
+        }
+
+        auto viewportContextManager = RPI::ViewportContextRequests::Get();
+        RPI::ViewportContextPtr viewportContext = viewportContextManager->GetViewportContextById(viewportId);
+        if (viewportContext == nullptr)
+        {
+            return nullptr;
+        }
+
+        // Get or create a ViewportData if one doesn't already exist
+        ViewportData& viewportData = m_viewportData[viewportId];
+        if (!viewportData.m_initialized)
+        {
+            viewportData.m_pipelineChangedHandler = AZ::Event<RPI::RenderPipelinePtr>::Handler([this, viewportId](RPI::RenderPipelinePtr pipeline)
+            {
+                AZStd::lock_guard lock(m_mutexDrawContexts);
+                ViewportData& viewportData = m_viewportData[viewportId];
+                for (auto& context : viewportData.m_dynamicDrawContexts)
+                {
+                    context.second->SetRenderPipeline(pipeline.get());
+                }
+            });
+            viewportData.m_viewportDestroyedHandler = AZ::Event<AzFramework::ViewportId>::Handler([this, viewportId](AzFramework::ViewportId id)
+            {
+                AZStd::lock_guard lock(m_mutexDrawContexts);
+                m_viewportData.erase(id);
+            });
+
+            viewportContext->ConnectCurrentPipelineChangedHandler(viewportData.m_pipelineChangedHandler);
+            viewportContext->ConnectAboutToBeDestroyedHandler(viewportData.m_viewportDestroyedHandler);
+
+            viewportData.m_initialized = true;
+        }
+
+        RHI::Ptr<RPI::DynamicDrawContext>& context = viewportData.m_dynamicDrawContexts[name];
+        if (context == nullptr)
+        {
+            auto pipeline = viewportContext->GetCurrentPipeline().get();
+            if (pipeline == nullptr)
+            {
+                return nullptr;
+            }
+            context = RPI::DynamicDrawInterface::Get()->CreateDynamicDrawContext(pipeline);
+            contextFactoryIt->second(context);
+        }
+
+        return context;
+    }
+} //namespace AZ::AtomBridge