Code/EnginePlugins/GameComponentsPlugin/Effects/Implementation/MeshDecalComponent.cpp - dev - redgpu/ezEngine

Code/EnginePlugins/GameComponentsPlugin/Effects/Implementation/MeshDecalComponent.cpp
210 строк6 KB
C-Core
Exposed more to Visual Script (#1879)
27 мар 2026, 11:31
27 мар 2026, 11:312a6c682
#include <GameComponentsPlugin/GameComponentsPCH.h>

#include <GameComponentsPlugin/Effects/MeshDecalComponent.h>

#include <Core/Messages/SetColorMessage.h>
#include <Core/WorldSerializer/WorldReader.h>
#include <Core/WorldSerializer/WorldWriter.h>
#include <Foundation/Containers/ArrayMap.h>
#include <Foundation/SimdMath/SimdRandom.h>
#include <RendererCore/Decals/Implementation/DecalManager.h>
#include <RendererCore/Lights/LightComponent.h>
#include <RendererCore/Pipeline/View.h>
#include <RendererCore/Textures/Texture2DResource.h>

//////////////////////////////////////////////////////////////////////////

// clang-format off
EZ_BEGIN_STATIC_REFLECTED_TYPE(ezMeshDecalDescription, ezNoBase, 1, ezRTTIDefaultAllocator<ezMeshDecalDescription>)
{
  EZ_BEGIN_PROPERTIES
  {
    EZ_MEMBER_PROPERTY("Index", m_uiIndex)->AddAttributes(new ezClampValueAttribute(0, 7)),
    EZ_RESOURCE_MEMBER_PROPERTY("BaseColorTexture", m_hBaseColorTexture)->AddAttributes(new ezAssetBrowserAttribute("CompatibleAsset_Texture_2D")),
  }
  EZ_END_PROPERTIES;
}
EZ_END_STATIC_REFLECTED_TYPE;
// clang-format on

ezResult ezMeshDecalDescription::Serialize(ezStreamWriter& inout_stream) const
{
  inout_stream << m_uiIndex;
  inout_stream << m_hBaseColorTexture;

  return EZ_SUCCESS;
}

ezResult ezMeshDecalDescription::Deserialize(ezStreamReader& inout_stream)
{
  inout_stream >> m_uiIndex;
  inout_stream >> m_hBaseColorTexture;

  return EZ_SUCCESS;
}

//////////////////////////////////////////////////////////////////////////////

// clang-format off
EZ_BEGIN_COMPONENT_TYPE(ezMeshDecalComponent, 1, ezComponentMode::Static)
{
  EZ_BEGIN_PROPERTIES
  {
    EZ_ARRAY_ACCESSOR_PROPERTY("Decals", Decals_GetCount, Decals_Get, Decals_Set, Decals_Insert, Decals_Remove),
  }
  EZ_END_PROPERTIES;
  EZ_BEGIN_ATTRIBUTES
  {
    new ezCategoryAttribute("Effects"),
  }
  EZ_END_ATTRIBUTES;
  EZ_BEGIN_MESSAGEHANDLERS
  {
    EZ_MESSAGE_HANDLER(ezMsgExtractRenderData, OnMsgExtractRenderData),
  }
  EZ_END_MESSAGEHANDLERS;
}
EZ_END_COMPONENT_TYPE
// clang-format on

void ezMeshDecalComponent::SerializeComponent(ezWorldWriter& inout_stream) const
{
  SUPER::SerializeComponent(inout_stream);
  ezStreamWriter& s = inout_stream.GetStream();

  s.WriteArray(m_DecalDescs).AssertSuccess();
}

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

  ezStreamReader& s = inout_stream.GetStream();

  s.ReadArray(m_DecalDescs).AssertSuccess();
}

void ezMeshDecalComponent::OnActivated()
{
  SUPER::OnActivated();

  UpdateDecals();
}

void ezMeshDecalComponent::OnDeactivated()
{
  SUPER::OnDeactivated();

  DeleteDecals();
}

void ezMeshDecalComponent::OnMsgExtractRenderData(ezMsgExtractRenderData& msg) const
{
  if (msg.m_pView->GetCameraUsageHint() == ezCameraUsageHint::Shadow)
    return;

  const float fScreenSpaceSize = ezLightComponent::CalculateScreenSpaceSize(GetOwner()->GetGlobalBounds().GetSphere(), *msg.m_pView->GetCullingCamera());

  for (ezDecalId decalId : m_DecalIds)
  {
    ezDecalManager::MarkRuntimeDecalAsUsed(decalId, fScreenSpaceSize, msg.m_pView);
  }
}

ezUInt32 ezMeshDecalComponent::Decals_GetCount() const
{
  return m_DecalDescs.GetCount();
}

const ezMeshDecalDescription& ezMeshDecalComponent::Decals_Get(ezUInt32 uiIndex) const
{
  return m_DecalDescs[uiIndex];
}

void ezMeshDecalComponent::Decals_Set(ezUInt32 uiIndex, const ezMeshDecalDescription& desc)
{
  m_DecalDescs[uiIndex] = desc;

  UpdateDecals();
}

void ezMeshDecalComponent::Decals_Insert(ezUInt32 uiIndex, const ezMeshDecalDescription& desc)
{
  m_DecalDescs.InsertAt(uiIndex, desc);

  UpdateDecals();
}

void ezMeshDecalComponent::Decals_Remove(ezUInt32 uiIndex)
{
  m_DecalDescs.RemoveAtAndCopy(uiIndex);

  UpdateDecals();
}

void ezMeshDecalComponent::UpdateDecals()
{
  if (!IsActiveAndInitialized())
    return;

  DeleteDecals();

  ezArrayMap<ezUInt32, ezTexture2DResourceHandle, ezTempAllocatorWrapper> indexToTexture;
  indexToTexture.Reserve(m_DecalDescs.GetCount());

  for (auto& desc : m_DecalDescs)
  {
    indexToTexture.Insert(desc.m_uiIndex, desc.m_hBaseColorTexture);
  }

  ezUInt32 uiRandomSeed = GetOwner()->GetStableRandomSeed();
  int iRandomPos = 0;
  auto RandomIndex = [uiRandomSeed, &iRandomPos](ezUInt32 uiMin, ezUInt32 uiMax)
  {
    ezUInt32 uiIndex = static_cast<ezUInt32>(ezSimdRandom::FloatMinMax(ezSimdVec4i(iRandomPos), ezSimdVec4f(float(uiMin)), ezSimdVec4f(float(uiMax)), ezSimdVec4u(uiRandomSeed)).x());
    iRandomPos++;
    return uiIndex;
  };

  ezUInt16 decalIndices[8] = {};
  for (ezUInt32 i = 0; i < 8; ++i)
  {
    decalIndices[i] = ezSmallInvalidIndex;

    const ezUInt32 uiLowerBound = indexToTexture.LowerBound(i);
    const ezUInt32 uiUpperBound = ezMath::Min(indexToTexture.UpperBound(i), indexToTexture.GetCount());

    if (uiLowerBound != ezInvalidIndex && uiUpperBound > uiLowerBound)
    {
      const ezUInt32 uiIndex = RandomIndex(uiLowerBound, uiUpperBound);
      auto& hTexture = indexToTexture.GetValue(uiIndex);
      if (hTexture.IsValid())
      {
        ezDecalId decalId = ezDecalManager::GetOrCreateRuntimeDecal(hTexture);
        decalIndices[i] = decalId.m_InstanceIndex;

        if (!m_DecalIds.Contains(decalId))
          m_DecalIds.PushBack(decalId);
      }
    }
  }

  ezMsgSetCustomData msg;
  msg.m_vData = *reinterpret_cast<const ezVec4*>(decalIndices);

  GetOwner()->PostMessage(msg, ezTime::MakeZero(), ezObjectMsgQueueType::AfterInitialized);
}

void ezMeshDecalComponent::DeleteDecals()
{
  for (ezDecalId decalId : m_DecalIds)
  {
    ezDecalManager::DeleteRuntimeDecal(decalId);
  }

  m_DecalIds.Clear();
}


EZ_STATICLINK_FILE(GameComponentsPlugin, GameComponentsPlugin_Effects_Implementation_MeshDecalComponent);