redgpu/ezEngine: An open source C++ game engine.

Code/Samples/ComputeShaderHistogram/ComputeShaderHistogram.h

53 строки2 KB

Sanakan8472

Render Buffer Refactor and SRV / UAV split (#1288)

13 май 2024, 21:34

13 май 2024, 21:346b98b48

#pragma once

#include <Foundation/Basics.h>

#include <Foundation/IO/DirectoryWatcher.h>
#include <Foundation/Types/UniquePtr.h>
#include <GameEngine/GameApplication/GameApplication.h>
#include <RendererCore/Material/MaterialResource.h>
#include <RendererCore/Meshes/MeshResource.h>

class ezWindow;
class ezDirectoryWatcher;

/// Uses shader reloading mechanism of the ShaderExplorer sample for quick prototyping.
class ezComputeShaderHistogramApp : public ezGameApplication
{
public:
  using SUPER = ezGameApplication;

  ezComputeShaderHistogramApp();
  ~ezComputeShaderHistogramApp();

  virtual ezApplication::Execution Run() override;

  virtual void AfterCoreSystemsStartup() override;
  virtual void BeforeHighLevelSystemsShutdown() override;

private:
  void CreateHistogramQuad();
  void OnFileChanged(ezStringView sFilename, ezDirectoryWatcherAction action, ezDirectoryWatcherType type);

  ezGALTextureHandle m_hScreenTexture;
  ezGALRenderTargetViewHandle m_hScreenRTV;
  ezGALTextureResourceViewHandle m_hScreenSRV;

  // Could use buffer, but access and organisation with texture is more straight forward.
  ezGALTextureHandle m_hHistogramTexture;
  ezGALTextureUnorderedAccessViewHandle m_hHistogramUAV;
  ezGALTextureResourceViewHandle m_hHistogramSRV;

  ezWindowBase* m_pWindow = nullptr;
  ezGALSwapChainHandle m_hSwapChain;

  ezShaderResourceHandle m_hScreenShader;
  ezShaderResourceHandle m_hHistogramDisplayShader;
  ezShaderResourceHandle m_hHistogramComputeShader;

  ezMeshBufferResourceHandle m_hHistogramQuadMeshBuffer;

  ezUniquePtr<ezDirectoryWatcher> m_pDirectoryWatcher;

  bool m_bStuffChanged = false;
};