Code/EnginePlugins/JoltPlugin/System/JoltDebugRenderer.cpp - dev - redgpu/ezEngine

Code/EnginePlugins/JoltPlugin/System/JoltDebugRenderer.cpp
156 строк5 KB
Jan Krassnigg
Updated Jolt to latest version (#1403)
09 окт 2024, 16:25
09 окт 2024, 16:25bb82f62
#include <JoltPlugin/JoltPluginPCH.h>

#include <JoltPlugin/System/JoltDebugRenderer.h>

#ifdef JPH_DEBUG_RENDERER

#  include <Jolt/Renderer/DebugRenderer.h>

void ezJoltDebugRenderer::TriangleBatch::AddRef()
{
  ++m_iRefCount;
}

void ezJoltDebugRenderer::TriangleBatch::Release()
{
  --m_iRefCount;

  if (m_iRefCount == 0)
  {
    auto* pThis = this;
    EZ_DEFAULT_DELETE(pThis);
  }
}

ezJoltDebugRenderer::ezJoltDebugRenderer()
{
  Initialize();
}

void ezJoltDebugRenderer::DrawLine(JPH::RVec3Arg inFrom, JPH::RVec3Arg inTo, JPH::ColorArg inColor)
{
  auto& l = m_Lines.ExpandAndGetRef();
  l.m_start = ezJoltConversionUtils::ToVec3(inFrom);
  l.m_end = ezJoltConversionUtils::ToVec3(inTo);
  l.m_startColor = ezJoltConversionUtils::ToColor(inColor);
  l.m_endColor = l.m_startColor;
}


void ezJoltDebugRenderer::DrawTriangle(JPH::Vec3Arg inV1, JPH::Vec3Arg inV2, JPH::Vec3Arg inV3, JPH::ColorArg inColor, ECastShadow inCastShadow)
{
  auto& t = m_Triangles.ExpandAndGetRef();
  t.m_position[0] = ezJoltConversionUtils::ToVec3(inV1);
  t.m_position[1] = ezJoltConversionUtils::ToVec3(inV2);
  t.m_position[2] = ezJoltConversionUtils::ToVec3(inV3);
  t.m_color = ezJoltConversionUtils::ToColor(inColor);
}


JPH::DebugRenderer::Batch ezJoltDebugRenderer::CreateTriangleBatch(const JPH::DebugRenderer::Triangle* pInTriangles, int iInTriangleCount)
{
  TriangleBatch* pBatch = EZ_DEFAULT_NEW(TriangleBatch);
  pBatch->m_Triangles.Reserve(iInTriangleCount);

  for (int i = 0; i < iInTriangleCount; ++i)
  {
    auto& t = pBatch->m_Triangles.ExpandAndGetRef();
    t.m_position[0] = ezJoltConversionUtils::ToVec3(pInTriangles[i].mV[0].mPosition);
    t.m_position[1] = ezJoltConversionUtils::ToVec3(pInTriangles[i].mV[1].mPosition);
    t.m_position[2] = ezJoltConversionUtils::ToVec3(pInTriangles[i].mV[2].mPosition);
    t.m_color = ezJoltConversionUtils::ToColor(pInTriangles[i].mV[0].mColor);
  }

  return pBatch;
}


JPH::DebugRenderer::Batch ezJoltDebugRenderer::CreateTriangleBatch(const JPH::DebugRenderer::Vertex* pInVertices, int iInVertexCount, const JPH::uint32* pInIndices, int iInIndexCount)
{
  const ezUInt32 numTris = iInIndexCount / 3;

  TriangleBatch* pBatch = EZ_DEFAULT_NEW(TriangleBatch);
  pBatch->m_Triangles.Reserve(numTris);

  ezUInt32 index = 0;

  for (ezUInt32 i = 0; i < numTris; ++i)
  {
    auto& t = pBatch->m_Triangles.ExpandAndGetRef();
    t.m_position[0] = ezJoltConversionUtils::ToVec3(pInVertices[pInIndices[index + 0]].mPosition);
    t.m_position[1] = ezJoltConversionUtils::ToVec3(pInVertices[pInIndices[index + 1]].mPosition);
    t.m_position[2] = ezJoltConversionUtils::ToVec3(pInVertices[pInIndices[index + 2]].mPosition);
    t.m_color = ezJoltConversionUtils::ToColor(pInVertices[pInIndices[index + 0]].mColor);

    index += 3;
  }

  return pBatch;
}


void ezJoltDebugRenderer::DrawGeometry(JPH::Mat44Arg modelMatrix, const JPH::AABox& worldSpaceBounds, float fInLODScaleSq, JPH::ColorArg inModelColor, const GeometryRef& geometry, ECullMode inCullMode /*= ECullMode::CullBackFace*/, ECastShadow inCastShadow /*= ECastShadow::On*/, EDrawMode inDrawMode /*= EDrawMode::Solid*/)
{
  if (geometry == nullptr)
    return;

  ezUInt32 uiLod = 0;
  if (geometry->mLODs.size() > 1)
    uiLod = 1;
  if (geometry->mLODs.size() > 2)
    uiLod = 2;

  const TriangleBatch* pBatch = static_cast<const TriangleBatch*>(geometry->mLODs[uiLod].mTriangleBatch.GetPtr());

  const ezMat4 trans = reinterpret_cast<const ezMat4&>(modelMatrix);
  const ezColor color = ezJoltConversionUtils::ToColor(inModelColor);

  if (inDrawMode == JPH::DebugRenderer::EDrawMode::Solid)
  {
    m_Triangles.Reserve(m_Triangles.GetCount() + pBatch->m_Triangles.GetCount() * ((inCullMode == JPH::DebugRenderer::ECullMode::Off) ? 2 : 1));

    if (inCullMode == JPH::DebugRenderer::ECullMode::CullBackFace || inCullMode == JPH::DebugRenderer::ECullMode::Off)
    {
      for (ezUInt32 t = 0; t < pBatch->m_Triangles.GetCount(); ++t)
      {
        auto& tri = m_Triangles.ExpandAndGetRef();
        tri.m_color = pBatch->m_Triangles[t].m_color * color;
        tri.m_position[0] = trans * pBatch->m_Triangles[t].m_position[0];
        tri.m_position[1] = trans * pBatch->m_Triangles[t].m_position[1];
        tri.m_position[2] = trans * pBatch->m_Triangles[t].m_position[2];
      }
    }

    if (inCullMode == JPH::DebugRenderer::ECullMode::CullFrontFace || inCullMode == JPH::DebugRenderer::ECullMode::Off)
    {
      for (ezUInt32 t = 0; t < pBatch->m_Triangles.GetCount(); ++t)
      {
        auto& tri = m_Triangles.ExpandAndGetRef();
        tri.m_color = pBatch->m_Triangles[t].m_color * color;
        tri.m_position[0] = trans * pBatch->m_Triangles[t].m_position[0];
        tri.m_position[1] = trans * pBatch->m_Triangles[t].m_position[2];
        tri.m_position[2] = trans * pBatch->m_Triangles[t].m_position[1];
      }
    }
  }
  else
  {
    m_Lines.Reserve(m_Lines.GetCount() + pBatch->m_Triangles.GetCount() * 3);

    for (ezUInt32 t = 0; t < pBatch->m_Triangles.GetCount(); ++t)
    {
      const auto& inTri = pBatch->m_Triangles[t];
      const ezColor col = pBatch->m_Triangles[t].m_color * color;

      const ezVec3 v0 = trans * inTri.m_position[0];
      const ezVec3 v1 = trans * inTri.m_position[1];
      const ezVec3 v2 = trans * inTri.m_position[2];

      m_Lines.PushBack({v0, v1, col});
      m_Lines.PushBack({v1, v2, col});
      m_Lines.PushBack({v2, v0, col});
    }
  }
}

#endif