Code/Engine/Foundation/Memory/Implementation/FrameAllocator.cpp - dev - redgpu/ezEngine

Code/Engine/Foundation/Memory/Implementation/FrameAllocator.cpp

90 строк2 KB

C-Core

Temp allocator (#1860)

10 мар 2026, 18:19

10 мар 2026, 18:198b61066

#include <Foundation/FoundationPCH.h>

#include <Foundation/Configuration/Startup.h>
#include <Foundation/Memory/FrameAllocator.h>
#include <Foundation/Profiling/Profiling.h>
#include <Foundation/Strings/StringBuilder.h>

ezDoubleBufferedLinearAllocator::ezDoubleBufferedLinearAllocator(ezStringView sName0, ezAllocator* pParent)
{
  constexpr ezUInt32 uiInitialSize = 1024 * 1024; // 1 MB

  ezStringBuilder sName = sName0;
  sName.Append("0");

  m_pCurrentAllocator = EZ_DEFAULT_NEW(LinearAllocatorType, sName, pParent, uiInitialSize);

  sName = sName0;
  sName.Append("1");

  m_pOtherAllocator = EZ_DEFAULT_NEW(LinearAllocatorType, sName, pParent, uiInitialSize);
}

ezDoubleBufferedLinearAllocator::~ezDoubleBufferedLinearAllocator()
{
  EZ_DEFAULT_DELETE(m_pCurrentAllocator);
  EZ_DEFAULT_DELETE(m_pOtherAllocator);
}

void ezDoubleBufferedLinearAllocator::Swap()
{
  ezMath::Swap(m_pCurrentAllocator, m_pOtherAllocator);

  m_pCurrentAllocator->Reset();
}

void ezDoubleBufferedLinearAllocator::Reset()
{
  m_pCurrentAllocator->Reset();
  m_pOtherAllocator->Reset();
}


// clang-format off
EZ_BEGIN_SUBSYSTEM_DECLARATION(Foundation, FrameAllocator)

  ON_CORESYSTEMS_STARTUP
  {
    ezFrameAllocator::Startup();
  }

  ON_CORESYSTEMS_SHUTDOWN
  {
    ezFrameAllocator::Shutdown();
  }

EZ_END_SUBSYSTEM_DECLARATION;
// clang-format on

ezDoubleBufferedLinearAllocator* ezFrameAllocator::s_pAllocator;

// static
void ezFrameAllocator::Swap()
{
  EZ_PROFILE_SCOPE("FrameAllocator.Swap");

  s_pAllocator->Swap();
}

// static
void ezFrameAllocator::Reset()
{
  if (s_pAllocator)
  {
    s_pAllocator->Reset();
  }
}

// static
void ezFrameAllocator::Startup()
{
  s_pAllocator = EZ_DEFAULT_NEW(ezDoubleBufferedLinearAllocator, "FrameAllocator", ezFoundation::GetAlignedAllocator());
}

// static
void ezFrameAllocator::Shutdown()
{
  EZ_DEFAULT_DELETE(s_pAllocator);
}

EZ_STATICLINK_FILE(Foundation, Foundation_Memory_Implementation_FrameAllocator);