redgpu/ezEngine: An open source C++ game engine.

Projects/InspectorPlugin/CVars.cpp
190 строк4 KB
jankrassnigg
Milestone 4
27 мар 2014, 19:08
27 мар 2014, 19:08bfef15a
#include <PCH.h>
#include <Foundation/Communication/Telemetry.h>
#include <Foundation/Configuration/CVar.h>

static void TelemetryMessage(void* pPassThrough)
{
  ezTelemetryMessage Msg;

  while (ezTelemetry::RetrieveMessage('SVAR', Msg) == EZ_SUCCESS)
  {
    if (Msg.GetMessageID() == 'SET')
    {
      ezString sCVar;
      ezUInt8 uiType;

      float fValue;
      ezInt32 iValue;
      bool bValue;
      ezString sValue;

      Msg.GetReader() >> sCVar;
      Msg.GetReader() >> uiType;

      switch (uiType)
      {
      case ezCVarType::Float:
        Msg.GetReader() >> fValue;
        break;
      case ezCVarType::Int:
        Msg.GetReader() >> iValue;
        break;
      case ezCVarType::Bool:
        Msg.GetReader() >> bValue;
        break;
      case ezCVarType::String:
        Msg.GetReader() >> sValue;
        break;
      }

      ezCVar* pCVar = ezCVar::GetFirstInstance();

      while (pCVar)
      {
        if (((ezUInt8) pCVar->GetType() == uiType) && (pCVar->GetName() == sCVar))
        {
          switch (uiType)
          {
          case ezCVarType::Float:
            *((ezCVarFloat*) pCVar) = fValue;
            break;
          case ezCVarType::Int:
            *((ezCVarInt*) pCVar) = iValue;
            break;
          case ezCVarType::Bool:
            *((ezCVarBool*) pCVar) = bValue;
            break;
          case ezCVarType::String:
            *((ezCVarString*) pCVar) = sValue;
            break;
          }
        }

        pCVar = pCVar->GetNextInstance();
      }
    }
  }
}

static void SendCVarTelemetry(ezCVar* pCVar)
{
  ezTelemetryMessage msg;
  msg.SetMessageID('CVAR', 'DATA');
  msg.GetWriter() << pCVar->GetName();
  msg.GetWriter() << pCVar->GetPluginName();
  msg.GetWriter() << (ezUInt8) pCVar->GetFlags().GetValue();
  msg.GetWriter() << (ezUInt8) pCVar->GetType();
  msg.GetWriter() << pCVar->GetDescription();

  switch (pCVar->GetType())
  {
  case ezCVarType::Float:
    {
      const float val = ((ezCVarFloat*) pCVar)->GetValue();
      msg.GetWriter() << val;
    }
    break;
  case ezCVarType::Int:
    {
      const int val = ((ezCVarInt*) pCVar)->GetValue();
      msg.GetWriter() << val;
    }
    break;
  case ezCVarType::Bool:
    {
      const bool val = ((ezCVarBool*) pCVar)->GetValue();
      msg.GetWriter() << val;
    }
    break;
  case ezCVarType::String:
    {
      const char* val = ((ezCVarString*) pCVar)->GetValue().GetData();
      msg.GetWriter() << val;
    }
    break;
  }

  ezTelemetry::Broadcast(ezTelemetry::Reliable, msg);
}

static void SendAllCVarTelemetry()
{
  if (!ezTelemetry::IsConnectedToClient())
    return;

  // clear
  {
    ezTelemetryMessage msg;
    ezTelemetry::Broadcast(ezTelemetry::Reliable, 'CVAR', 'CLR', NULL, 0);
  }

  ezCVar* pCVar = ezCVar::GetFirstInstance();

  while (pCVar)
  {
    SendCVarTelemetry(pCVar);

    pCVar = pCVar->GetNextInstance();
  }

  {
    ezTelemetryMessage msg;
    ezTelemetry::Broadcast(ezTelemetry::Reliable, 'CVAR', 'SYNC', NULL, 0);
  }
}


static void TelemetryEventsHandler(const ezTelemetry::TelemetryEventData& e)
{
  switch (e.m_EventType)
  {
  case ezTelemetry::TelemetryEventData::ConnectedToClient:
    SendAllCVarTelemetry();
    break;
  }
}

static void CVarEventHandler(const ezCVar::CVarEvent& e)
{
  if (!ezTelemetry::IsConnectedToClient())
    return;

  switch (e.m_EventType)
  {
  case ezCVar::CVarEvent::ValueChanged:
    SendCVarTelemetry(e.m_pCVar);
    break;
  }
}

static void PluginEventHandler(const ezPlugin::PluginEvent& e)
{
  switch (e.m_EventType)
  {
  case ezPlugin::PluginEvent::AfterPluginChanges:
    SendAllCVarTelemetry();
    break;
  }
}

void AddCVarEventHandler()
{
  ezTelemetry::AddEventHandler(TelemetryEventsHandler);
  ezTelemetry::AcceptMessagesForSystem('SVAR', true, TelemetryMessage, NULL);

  ezCVar::s_AllCVarEvents.AddEventHandler(CVarEventHandler);
  ezPlugin::s_PluginEvents.AddEventHandler(PluginEventHandler);
}

void RemoveCVarEventHandler()
{
  ezPlugin::s_PluginEvents.RemoveEventHandler(PluginEventHandler);
  ezCVar::s_AllCVarEvents.RemoveEventHandler(CVarEventHandler);

  ezTelemetry::RemoveEventHandler(TelemetryEventsHandler);
  ezTelemetry::AcceptMessagesForSystem('SVAR', false);
}



EZ_STATICLINK_FILE(InspectorPlugin, InspectorPlugin_CVars);