Data/Samples/RTS/CppSource/RTSPlugin/Components/TorpedoComponent.cpp - dev - redgpu/ezEngine

Data/Samples/RTS/CppSource/RTSPlugin/Components/TorpedoComponent.cpp
119 строк3 KB
Jan Krassnigg
C++ Project Generation now also builds stand-alone game exe (#1465)
30 дек 2024, 12:59
30 дек 2024, 12:59d369e9b
#include <RTSPlugin/RTSPluginPCH.h>

#include <RTSPlugin/Components/ComponentMessages.h>
#include <RTSPlugin/Components/TorpedoComponent.h>
#include <RTSPlugin/GameState/RTSGameState.h>

// clang-format off
EZ_BEGIN_COMPONENT_TYPE(RtsTorpedoComponent, 1, ezComponentMode::Dynamic)
{
  EZ_BEGIN_PROPERTIES
  {
    EZ_MEMBER_PROPERTY("Speed", m_fSpeed)->AddAttributes(new ezDefaultValueAttribute(10.0f)),
    EZ_MEMBER_PROPERTY("Damage", m_iDamage)->AddAttributes(new ezDefaultValueAttribute(10)),
  }
  EZ_END_PROPERTIES;

  EZ_BEGIN_MESSAGEHANDLERS
  {
    EZ_MESSAGE_HANDLER(RtsMsgSetTarget, OnMsgSetTarget),
  }
  EZ_END_MESSAGEHANDLERS;

  EZ_BEGIN_ATTRIBUTES
  {
    new ezCategoryAttribute("RTS Sample"),
  }
  EZ_END_ATTRIBUTES;
}
EZ_END_COMPONENT_TYPE;
// clang-format on

RtsTorpedoComponent::RtsTorpedoComponent()
{
  m_vTargetPosition.SetZero();
}

RtsTorpedoComponent::~RtsTorpedoComponent() = default;

void RtsTorpedoComponent::SerializeComponent(ezWorldWriter& inout_stream) const
{
  SUPER::SerializeComponent(inout_stream);

  auto& s = inout_stream.GetStream();

  s << m_fSpeed;
  s << m_iDamage;
}

void RtsTorpedoComponent::DeserializeComponent(ezWorldReader& inout_stream)
{
  SUPER::DeserializeComponent(inout_stream);
  const ezUInt32 uiVersion = inout_stream.GetComponentTypeVersion(GetStaticRTTI());

  auto& s = inout_stream.GetStream();

  s >> m_fSpeed;
  s >> m_iDamage;
}

void RtsTorpedoComponent::OnMsgSetTarget(RtsMsgSetTarget& ref_msg)
{
  m_vTargetPosition.SetZero();
  m_hTargetObject = ref_msg.m_hObject;
}

void RtsTorpedoComponent::Update()
{
  // TODO: do this check in the component manager
  if (RTSGameState::GetSingleton() == nullptr || RTSGameState::GetSingleton()->GetActiveGameMode() != RtsActiveGameMode::BattleMode)
    return;

  ezGameObject* pObject = nullptr;
  if (GetWorld()->TryGetObject(m_hTargetObject, pObject))
  {
    m_vTargetPosition = pObject->GetGlobalTransform().m_vPosition.GetAsVec2();
  }

  const ezTransform transform = GetOwner()->GetGlobalTransform();
  const ezVec2 vCurPos = transform.m_vPosition.GetAsVec2();
  const ezVec2 vDir = m_vTargetPosition - vCurPos;
  const float fDist = vDir.GetLength();

  const float tDiff = (float)GetWorld()->GetClock().GetTimeDiff().GetSeconds();

  bool bExplode = false;

  if (fDist > 0)
  {
    const ezVec2 vDirNorm = vDir / fDist;
    float fTravelDist = m_fSpeed * tDiff;

    if (fTravelDist >= fDist)
    {
      fTravelDist = fDist;
      bExplode = true;
    }

    const ezVec3 vNewPos = transform.m_vPosition + vDirNorm.GetAsVec3(0) * fTravelDist;

    GetOwner()->SetGlobalPosition(vNewPos);
  }
  else
  {
    bExplode = true;
  }

  if (bExplode)
  {
    RtsMsgApplyDamage msg;
    msg.m_iDamage = m_iDamage;

    if (!m_hTargetObject.IsInvalidated())
    {
      GetWorld()->SendMessage(m_hTargetObject, msg);
    }

    GetWorld()->DeleteObjectDelayed(GetOwner()->GetHandle());
  }
}