Code/Tools/Inspector/ReflectionWidget.cpp - dev - redgpu/ezEngine

Code/Tools/Inspector/ReflectionWidget.cpp
191 строка5 KB
Jan Krassnigg
Updated ADS and fixed document layout restoration (#1497)
28 фев 2025, 12:35
28 фев 2025, 12:352e065c5
#include <Inspector/InspectorPCH.h>

#include <Foundation/Communication/Telemetry.h>
#include <Inspector/MainWindow.moc.h>
#include <Inspector/ReflectionWidget.moc.h>
#include <qlistwidget.h>

ezQtReflectionWidget* ezQtReflectionWidget::s_pWidget = nullptr;

ezQtReflectionWidget::ezQtReflectionWidget(ads::CDockManager* pDockManager, QWidget* pParent)
  : ads::CDockWidget(pDockManager, "Reflection Widget", pParent)
{
  s_pWidget = this;

  setupUi(this);
  setWidget(ReflectionWidgetFrame);

  setIcon(QIcon(":/Icons/Icons/Type.svg"));

  ResetStats();
}

void ezQtReflectionWidget::ResetStats()
{
  TypeTree->clear();

  {
    QStringList Headers;
    Headers.append(" Type ");
    Headers.append(" Property ");
    Headers.append(" Size (Bytes) ");
    Headers.append(" Plugin ");

    TypeTree->setColumnCount(static_cast<int>(Headers.size()));
    TypeTree->setHeaderLabels(Headers);
    TypeTree->header()->show();
  }
}

void ezQtReflectionWidget::ProcessTelemetry(void* pUnuseed)
{
  if (!s_pWidget)
    return;

  ezTelemetryMessage msg;

  bool bUpdate = false;

  while (ezTelemetry::RetrieveMessage('RFLC', msg) == EZ_SUCCESS)
  {
    if (msg.GetMessageID() == ' CLR')
    {
      s_pWidget->m_Types.Clear();
      s_pWidget->TypeTree->clear();
    }

    if (msg.GetMessageID() == 'DATA')
    {
      bUpdate = true;

      ezString sName;
      msg.GetReader() >> sName;

      TypeData& sd = s_pWidget->m_Types[sName];

      msg.GetReader() >> sd.m_sParentType;
      msg.GetReader() >> sd.m_uiSize;
      msg.GetReader() >> sd.m_sPlugin;

      {
        ezUInt32 num = 0;
        msg.GetReader() >> num;

        for (ezUInt32 i = 0; i < num; ++i)
        {
          PropertyData pd;
          msg.GetReader() >> pd.m_sPropertyName;
          msg.GetReader() >> pd.m_iCategory;
          msg.GetReader() >> pd.m_sType;

          for (ezUInt32 j = 0; j < sd.m_Properties.GetCount(); ++j)
          {
            if (sd.m_Properties[j].m_sPropertyName == pd.m_sPropertyName)
              goto found;
          }

          sd.m_Properties.PushBack(pd);

        found:
          continue;
        }
      }

      /// \todo should read the message handlers here
    }
  }

  if (bUpdate)
  {
    while (s_pWidget->UpdateTree())
    {
    }

    s_pWidget->TypeTree->resizeColumnToContents(3);
    s_pWidget->TypeTree->resizeColumnToContents(2);
    s_pWidget->TypeTree->resizeColumnToContents(1);
    s_pWidget->TypeTree->resizeColumnToContents(0);
  }
}

bool ezQtReflectionWidget::UpdateTree()
{
  bool bAddedAny = false;

  for (auto it = m_Types.GetIterator(); it.IsValid(); ++it)
  {
    if (it.Value().m_pTreeItem == nullptr)
    {
      bAddedAny = true;

      // has parent type, but parent type has not been inserted into tree yet
      if (!it.Value().m_sParentType.IsEmpty() && m_Types[it.Value().m_sParentType].m_pTreeItem == nullptr)
        continue;

      QTreeWidgetItem* pParent = (it.Value().m_sParentType.IsEmpty()) ? nullptr : m_Types[it.Value().m_sParentType].m_pTreeItem;

      QTreeWidgetItem* pItem = new QTreeWidgetItem();
      it.Value().m_pTreeItem = pItem;

      ezStringBuilder sText;
      sText.SetFormat("{0}", it.Value().m_uiSize);

      pItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
      pItem->setText(0, it.Key().GetData());
      pItem->setText(2, sText.GetData());
      pItem->setText(3, it.Value().m_sPlugin.GetData());

      pItem->setIcon(0, QIcon(":/Icons/Icons/Type.svg"));

      if (pParent)
      {
        pParent->addChild(pItem);
        pParent->setExpanded(false);
      }
      else
      {
        TypeTree->addTopLevelItem(pItem);
      }

      for (ezUInt32 i = 0; i < it.Value().m_Properties.GetCount(); ++i)
      {
        QTreeWidgetItem* pProperty = new QTreeWidgetItem();
        pProperty->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);

        switch (it.Value().m_Properties[i].m_iCategory)
        {
          case -1:
            pProperty->setText(0, "Message");
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Message.svg"));
            break;
          case ezPropertyCategory::Member:
            pProperty->setText(0, it.Value().m_Properties[i].m_sType.GetData());
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Member.svg"));
            break;
          case ezPropertyCategory::Function:
            pProperty->setText(0, "Function");
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Function.svg"));
            break;
          case ezPropertyCategory::Array:
            pProperty->setText(0, it.Value().m_Properties[i].m_sType.GetData());
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Array.svg"));
            break;
          case ezPropertyCategory::Set:
            pProperty->setText(0, it.Value().m_Properties[i].m_sType.GetData());
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Set.svg"));
            break;
          case ezPropertyCategory::Map:
            pProperty->setText(0, it.Value().m_Properties[i].m_sType.GetData());
            pProperty->setIcon(0, QIcon(":/Icons/Icons/Map.svg"));
            break;
        }

        pProperty->setText(1, it.Value().m_Properties[i].m_sPropertyName.GetData());

        pItem->addChild(pProperty);
      }
    }
  }

  return bAddedAny;
}