Code/Samples/MeshRenderSample/MeshRenderSample.cpp - dev - redgpu/ezEngine

Code/Samples/MeshRenderSample/MeshRenderSample.cpp
331 строка12 KB
Sanakan8472
GAL Polishing (#1833)
02 мар 2026, 00:04
02 мар 2026, 00:04f93769d
#include <Core/Graphics/Geometry.h>
#include <Core/Input/DeviceTypes/MouseKeyboard.h>
#include <Core/Input/InputManager.h>
#include <Core/ResourceManager/ResourceManager.h>
#include <Core/System/Window.h>
#include <Foundation/Application/Application.h>
#include <Foundation/IO/FileSystem/DataDirTypeFolder.h>
#include <Foundation/IO/FileSystem/FileSystem.h>
#include <Foundation/Logging/ConsoleWriter.h>
#include <Foundation/Threading/TaskSystem.h>
#include <Foundation/Time/Clock.h>
#include <Foundation/Utilities/GraphicsUtils.h>
#include <RendererCore/Material/MaterialResource.h>
#include <RendererCore/Meshes/MeshBufferResource.h>
#include <RendererCore/RenderContext/RenderContext.h>
#include <RendererCore/ShaderCompiler/ShaderManager.h>
#include <RendererFoundation/Device/DeviceFactory.h>

// Constant buffer definition is shared between shader code and C++
#include <RendererCore/../../../Data/Samples/MeshRenderSample/Shaders/SampleConstantBuffer.h>


#ifdef BUILDSYSTEM_ENABLE_VULKAN_SUPPORT
constexpr const char* szDefaultRenderer = "Vulkan";
#else
constexpr const char* szDefaultRenderer = "DX11";
#endif

/// \brief The application class.
/// Instantiated and run through the EZ_APPLICATION_ENTRY_POINT macro at the end of this file.
class MeshRenderSample : public ezApplication
{
public:
  MeshRenderSample()
    : ezApplication("MeshRenderSample")
  {
  }

  ezResult BeforeCoreSystemsStartup() override
  {
    ezGlobalLog::AddLogWriter(ezLogWriter::Console::LogMessageHandler);

    return ezApplication::BeforeCoreSystemsStartup();
  }

  void AfterCoreSystemsStartup() override
  {
    ezDataDirectory::FolderType::s_sRedirectionFile = "AssetCache/Default.ezAidlt";
    ezDataDirectory::FolderType::s_sRedirectionPrefix = "AssetCache/";
    ezStringBuilder sProjectDir = ">sdk/Data/Samples/MeshRenderSample";
    ezStringBuilder sProjectDirResolved;
    ezFileSystem::ResolveSpecialDirectory(sProjectDir, sProjectDirResolved).AssertSuccess();

    ezFileSystem::SetSpecialDirectory("project", sProjectDirResolved);

    if (ezFileSystem::AddDataDirectory(">sdk/Output", "ShaderCache", "shadercache", ezDataDirUsage::AllowWrites).Failed())
    {
      ezFileSystem::AddDataDirectory(">sdk/Output", "ShaderCache", "shadercache", ezDataDirUsage::ReadOnly).AssertSuccess();
    }

    ezFileSystem::AddDataDirectory(">sdk/Data/Base", "Base", "base").AssertSuccess();
    ezFileSystem::AddDataDirectory(">project/", "Project", "project").AssertSuccess();

    ezStringView sRendererName = ezCommandLineUtils::GetGlobalInstance()->GetStringOption("-renderer", 0, szDefaultRenderer);

    const char* szShaderModel = "";
    const char* szShaderCompiler = "";
    ezGALDeviceFactory::GetShaderModelAndCompiler(sRendererName, szShaderModel, szShaderCompiler);

    ezShaderManager::Configure(szShaderModel, true);
    ezPlugin::LoadPlugin(szShaderCompiler).IgnoreResult();

    // Create a window for rendering (this also sets up the mouse/keyboard input device)
    {
      ezWindowCreationDesc WindowCreationDesc;
      WindowCreationDesc.m_Resolution.width = 1024;
      WindowCreationDesc.m_Resolution.height = 768;

      m_pWindow = EZ_DEFAULT_NEW(ezWindow);
      m_pWindow->Initialize(WindowCreationDesc).AssertSuccess();

      m_pWindow->WindowEvents().AddEventHandler([this](const ezWindowEvent& e)
        {
          if (e.m_Type == ezWindowEvent::Type::CloseButtonClicked)
          {
            this->QuitApplication();
          }
          //
        });

      if (auto pInput = ezDynamicCast<ezInputDeviceMouseKeyboard*>(m_pWindow->GetInputDevice()))
      {
        pInput->SetShowMouseCursor(true);
        pInput->SetClipMouseCursor(ezMouseCursorClipMode::NoClip);
      }
    }

    // Register Input
    {
      ezInputActionConfig cfg;

      cfg = ezInputManager::GetInputActionConfig("Main", "CloseApp");
      cfg.m_sInputSlotTrigger[0] = ezInputSlot_KeyEscape;
      ezInputManager::SetInputActionConfig("Main", "CloseApp", cfg, true);
    }

    // Create a rendering device
    {
      ezGALDeviceCreationDescription DeviceInit;
      DeviceInit.m_bDebugDevice = true;

      m_pDevice = ezGALDeviceFactory::CreateDevice(sRendererName, ezFoundation::GetDefaultAllocator(), DeviceInit);
      EZ_ASSERT_DEV(m_pDevice != nullptr, "Device implemention for '{}' not found", sRendererName);
      EZ_VERIFY(m_pDevice->Init() == EZ_SUCCESS, "Device init failed!");

      ezGALDevice::SetDefaultDevice(m_pDevice);
    }

    // now that we have a window and device, tell the engine to initialize the rendering infrastructure
    ezStartup::StartupHighLevelSystems();

    // Create a Swapchain
    {
      ezGALWindowSwapChainCreationDescription swapChainDesc;
      swapChainDesc.m_pWindow = m_pWindow;
      swapChainDesc.m_SampleCount = ezGALMSAASampleCount::None;
      m_hSwapChain = ezGALWindowSwapChain::Create(swapChainDesc);

      const ezGALSwapChain* pPrimarySwapChain = m_pDevice->GetSwapChain(m_hSwapChain);
      const ezSizeU32 windowSize = m_pWindow->GetClientAreaSize();

      ezGALTextureCreationDescription texDesc;
      texDesc.m_uiWidth = windowSize.width;
      texDesc.m_uiHeight = windowSize.height;
      texDesc.m_Format = ezGALResourceFormat::D24S8;
      texDesc.m_TextureFlags.Add(ezGALTextureUsageFlags::RenderTarget);

      m_hDepthStencilTexture = m_pDevice->CreateTexture(texDesc);

      m_hBBRTV = m_pDevice->GetDefaultRenderTargetView(pPrimarySwapChain->GetBackBufferTexture());
      m_hBBDSV = m_pDevice->GetDefaultRenderTargetView(m_hDepthStencilTexture);
    }

    // Setup Shaders and Materials
    {
      // the shader (referenced by the material) also defines the render pipeline state
      // such as backface-culling and depth-testing
      m_hMaterial1 = ezResourceManager::LoadResource<ezMaterialResource>("Materials/Sample1.ezMaterial");
      m_hMaterial2 = ezResourceManager::LoadResource<ezMaterialResource>("Materials/Sample2.ezMaterial");

      // Create the mesh that we use for rendering
      m_hMeshBuffer1 = CreateTorusMesh();
      m_hMeshBuffer2 = CreateSphereMesh();
    }

    // Setup constant buffer that this sample uses
    {
      m_hSampleConstants = ezRenderContext::CreateConstantBufferStorage(m_pSampleConstantBuffer);
    }
  }


  void Run() override
  {
    EZ_LOG_BLOCK("Frame");

    // make sure time goes on
    ezClock::GetGlobalClock()->Update();

    // tick the input system
    ezInputManager::Update(ezClock::GetGlobalClock()->GetTimeDiff());

    // close if the Escape key was pressed
    if (ezInputManager::GetInputActionState("Main", "CloseApp") == ezKeyState::Pressed)
    {
      QuitApplication();
      return;
    }

    m_pWindow->ProcessWindowMessages();

    // do the rendering
    {
      // Before starting to render in a frame call this function
      m_pDevice->EnqueueFrameSwapChain(m_hSwapChain);
      m_pDevice->BeginFrame();

      ezGALCommandEncoder* pCommandEncoder = m_pDevice->BeginCommands("ezMeshRenderSampleMainPass");

      const ezGALSwapChain* pPrimarySwapChain = m_pDevice->GetSwapChain(m_hSwapChain);

      m_hBBRTV = m_pDevice->GetDefaultRenderTargetView(pPrimarySwapChain->GetBackBufferTexture());
      m_hBBDSV = m_pDevice->GetDefaultRenderTargetView(m_hDepthStencilTexture);

      ezGALRenderingSetup renderingSetup;
      renderingSetup.SetColorTarget(0, m_hBBRTV).SetDepthStencilTarget(m_hBBDSV);
      renderingSetup.SetClearColor(0).SetClearDepth();

      const float fWindowWidth = (float)m_pWindow->GetClientAreaSize().width;
      const float fWindowHeight = (float)m_pWindow->GetClientAreaSize().height;

      ezRenderContext::GetDefaultInstance()->BeginRendering(renderingSetup, ezRectFloat(0.0f, 0.0f, fWindowWidth, fWindowHeight));

      const ezMat4 mProj = ezGraphicsUtils::CreatePerspectiveProjectionMatrixFromFovY(ezAngle::MakeFromDegree(70), fWindowWidth / fWindowHeight, 0.1f, 1000.0f);

      ezBindGroupBuilder& bindGroupSample = ezRenderContext::GetDefaultInstance()->GetBindGroup();
      bindGroupSample.BindBuffer("ezMeshRenderSampleConstants", m_hSampleConstants);

      m_CameraRotation += ezAngle::MakeFromDegree(ezClock::GetGlobalClock()->GetTimeDiff().AsFloatInSeconds() * 90.0f);
      m_CameraRotation.NormalizeRange();

      ezVec3 vCamPos(0);
      vCamPos.x = ezMath::Sin(m_CameraRotation) * 5.0f;
      vCamPos.y = ezMath::Cos(m_CameraRotation) * 5.0f;
      vCamPos.z = 3.0f;

      const ezMat4 mView = ezGraphicsUtils::CreateLookAtViewMatrix(vCamPos, ezVec3::MakeZero(), ezVec3(0, 0, 1));

      // Update the constant buffer
      {
        ezMeshRenderSampleConstants& cb = m_pSampleConstantBuffer->GetDataForWriting();
        cb.ModelMatrix = mView;
        cb.ViewProjectionMatrix = mProj;
        cb.TintColor = ezColor::CornflowerBlue;
      }

      ezRenderContext::GetDefaultInstance()->BindMaterial(m_hMaterial1);
      ezRenderContext::GetDefaultInstance()->BindMeshBuffer(m_hMeshBuffer1);
      ezRenderContext::GetDefaultInstance()->DrawMeshBuffer().IgnoreResult();

      ezRenderContext::GetDefaultInstance()->BindMaterial(m_hMaterial2);
      ezRenderContext::GetDefaultInstance()->BindMeshBuffer(m_hMeshBuffer2);
      ezRenderContext::GetDefaultInstance()->DrawMeshBuffer().IgnoreResult();

      ezRenderContext::GetDefaultInstance()->EndRendering();
      m_pDevice->EndCommands(pCommandEncoder);

      m_pDevice->EndFrame();
    }

    // needs to be called once per frame to finish resource loading
    ezResourceManager::PerFrameUpdate();

    // tell the task system to finish its work for this frame
    // this has to be done at the very end, so that the task system will only use up the time that is left in this frame for
    // uploading GPU data etc.
    ezTaskSystem::FinishFrameTasks();
  }

  void BeforeCoreSystemsShutdown() override
  {
    // make sure that no textures are continue to be streamed in while the engine shuts down
    ezResourceManager::EngineAboutToShutdown();

    ezRenderContext::DeleteConstantBufferStorage(m_hSampleConstants);

    m_pDevice->DestroyTexture(m_hDepthStencilTexture);

    m_hMaterial1.Invalidate();
    m_hMaterial2.Invalidate();
    m_hMeshBuffer1.Invalidate();
    m_hMeshBuffer2.Invalidate();

    // tell the engine that we are about to destroy window and graphics device,
    // and that it therefore needs to cleanup anything that depends on that
    ezStartup::ShutdownHighLevelSystems();

    ezResourceManager::FreeAllUnusedResources();

    m_pDevice->DestroySwapChain(m_hSwapChain);

    // now we can destroy the graphics device
    m_pDevice->Shutdown().AssertSuccess();
    EZ_DEFAULT_DELETE(m_pDevice);

    // finally destroy the window
    m_pWindow->DestroyWindow();
    EZ_DEFAULT_DELETE(m_pWindow);
  }

  ezMeshBufferResourceHandle CreateTorusMesh()
  {
    ezGeometry geom;
    ezGeometry::GeoOptions opt;
    opt.m_Color = ezColor::Black;
    geom.AddTorus(2.0f, 3.5f, 24, 12, true, opt);
    geom.TriangulatePolygons();

    ezMeshBufferResourceDescriptor desc;
    desc.AddCommonStreams();

    desc.AllocateStreamsFromGeometry(geom);

    return ezResourceManager::GetOrCreateResource<ezMeshBufferResource>("TorusMesh", std::move(desc));
  }

  ezMeshBufferResourceHandle CreateSphereMesh()
  {
    ezGeometry geom;
    ezGeometry::GeoOptions opt;
    opt.m_Color = ezColor::Black;
    geom.AddGeodesicSphere(1.5f, 1);
    geom.TriangulatePolygons();

    ezMeshBufferResourceDescriptor desc;
    desc.AddCommonStreams();

    desc.AllocateStreamsFromGeometry(geom);

    return ezResourceManager::GetOrCreateResource<ezMeshBufferResource>("SphereMesh", std::move(desc));
  }

private:
  ezWindow* m_pWindow = nullptr;
  ezGALDevice* m_pDevice = nullptr;

  ezGALSwapChainHandle m_hSwapChain;
  ezGALRenderTargetViewHandle m_hBBRTV;
  ezGALRenderTargetViewHandle m_hBBDSV;
  ezGALTextureHandle m_hDepthStencilTexture;

  ezConstantBufferStorageHandle m_hSampleConstants;
  ezConstantBufferStorage<ezMeshRenderSampleConstants>* m_pSampleConstantBuffer;
  ezMaterialResourceHandle m_hMaterial1;
  ezMaterialResourceHandle m_hMaterial2;
  ezMeshBufferResourceHandle m_hMeshBuffer1;
  ezMeshBufferResourceHandle m_hMeshBuffer2;
  ezAngle m_CameraRotation = ezAngle::MakeZero();
};

EZ_APPLICATION_ENTRY_POINT(MeshRenderSample);