Code/Tools/Libs/GuiFoundation/PropertyGrid/Implementation/VisualizerManager.cpp - dev - redgpu/ezEngine

Code/Tools/Libs/GuiFoundation/PropertyGrid/Implementation/VisualizerManager.cpp
114 строк3 KB
Jan Krassnigg
Fixed #762: Toggling visualizers didn't work for scene layers
26 окт 2022, 22:08
26 окт 2022, 22:0882c21f5
#include <GuiFoundation/GuiFoundationPCH.h>

#include <GuiFoundation/PropertyGrid/VisualizerManager.h>
#include <ToolsFoundation/Object/DocumentObjectManager.h>

EZ_IMPLEMENT_SINGLETON(ezVisualizerManager);

// clang-format off
EZ_BEGIN_SUBSYSTEM_DECLARATION(GuiFoundation, VisualizerManager)

  ON_CORESYSTEMS_STARTUP
  {
    EZ_DEFAULT_NEW(ezVisualizerManager);
  }

  ON_CORESYSTEMS_SHUTDOWN
  {
    if (ezVisualizerManager::GetSingleton())
    {
      auto ptr = ezVisualizerManager::GetSingleton();
      EZ_DEFAULT_DELETE(ptr);
    }
  }


EZ_END_SUBSYSTEM_DECLARATION;
// clang-format on

ezVisualizerManager::ezVisualizerManager()
  : m_SingletonRegistrar(this)
{
  ezDocumentManager::s_Events.AddEventHandler(ezMakeDelegate(&ezVisualizerManager::DocumentManagerEventHandler, this));
}

ezVisualizerManager::~ezVisualizerManager()
{
  ezDocumentManager::s_Events.RemoveEventHandler(ezMakeDelegate(&ezVisualizerManager::DocumentManagerEventHandler, this));
}

void ezVisualizerManager::SetVisualizersActive(const ezDocument* pDoc, bool bActive)
{
  if (m_DocsSubscribed[pDoc].m_bActivated == bActive)
    return;

  m_DocsSubscribed[pDoc].m_bActivated = bActive;

  SendEventToRecreateVisualizers(pDoc);
}

bool ezVisualizerManager::GetVisualizersActive(const ezDocument* pDoc)
{
  return m_DocsSubscribed[pDoc].m_bActivated;
}

void ezVisualizerManager::SelectionEventHandler(const ezSelectionManagerEvent& event)
{
  if (!m_DocsSubscribed[event.m_pDocument].m_bActivated)
    return;

  SendEventToRecreateVisualizers(event.m_pDocument);
}

void ezVisualizerManager::SendEventToRecreateVisualizers(const ezDocument* pDoc)
{
  if (m_DocsSubscribed[pDoc].m_bActivated)
  {
    const auto& sel = pDoc->GetSelectionManager()->GetSelection();

    ezVisualizerManagerEvent e;
    e.m_pSelection = &sel;
    e.m_pDocument = pDoc;
    m_Events.Broadcast(e);
  }
  else
  {
    ezDeque<const ezDocumentObject*> sel;

    ezVisualizerManagerEvent e;
    e.m_pSelection = &sel;
    e.m_pDocument = pDoc;

    m_Events.Broadcast(e);
  }
}

void ezVisualizerManager::DocumentManagerEventHandler(const ezDocumentManager::Event& e)
{
  if (e.m_Type == ezDocumentManager::Event::Type::DocumentOpened)
  {
    e.m_pDocument->GetSelectionManager()->m_Events.AddEventHandler(ezMakeDelegate(&ezVisualizerManager::SelectionEventHandler, this));
    e.m_pDocument->GetObjectManager()->m_StructureEvents.AddEventHandler(ezMakeDelegate(&ezVisualizerManager::StructureEventHandler, this));
  }

  if (e.m_Type == ezDocumentManager::Event::Type::DocumentClosing)
  {
    e.m_pDocument->GetSelectionManager()->m_Events.RemoveEventHandler(ezMakeDelegate(&ezVisualizerManager::SelectionEventHandler, this));
    e.m_pDocument->GetObjectManager()->m_StructureEvents.RemoveEventHandler(ezMakeDelegate(&ezVisualizerManager::StructureEventHandler, this));

    SetVisualizersActive(e.m_pDocument, false);
  }
}

void ezVisualizerManager::StructureEventHandler(const ezDocumentObjectStructureEvent& event)
{
  if (!m_DocsSubscribed[event.m_pDocument].m_bActivated)
    return;

  if (!event.m_pDocument->GetSelectionManager()->IsSelectionEmpty() &&
      (event.m_EventType == ezDocumentObjectStructureEvent::Type::AfterObjectAdded ||
        event.m_EventType == ezDocumentObjectStructureEvent::Type::AfterObjectRemoved))
  {
    SendEventToRecreateVisualizers(event.m_pDocument);
  }
}