hiz_pyramid.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "gpu_sync.hpp"
#include "shader_layout.hpp"
#include "vulkan_pipeline.hpp"
 
#include <glm/glm.hpp>
#include <optional>
#include <vector>
 
namespace renderer {
class VulkanCore;

class HiZPyramid {
public:
  void create(const VulkanCore& core, vk::ImageView depth, vk::Extent2D depth_extent);

  void record(vk::CommandBuffer cmd, vk::Extent2D scene_extent, const glm::mat4& view_projection);

  void beforeRead(sync::BarrierBatch& barriers);

  [[nodiscard]]
  auto isReady() const -> bool {
    return m_view.has_value() && m_pipeline.isReady();
  }

  [[nodiscard]]
  auto view() const -> vk::ImageView {
    return m_view.has_value() ? **m_view : vk::ImageView {};
  }
 
  [[nodiscard]]
  auto sampler() const -> vk::Sampler {
    return *m_sampler;
  }

  [[nodiscard]]
  auto builtExtent() const -> vk::Extent2D {
    return m_built_extent;
  }
 
  [[nodiscard]]
  auto builtLevels() const -> uint32_t {
    return m_built_levels;
  }
 
  [[nodiscard]]
  auto builtViewProjection() const -> const glm::mat4& {
    return m_built_view_projection;
  }

  [[nodiscard]]
  static auto levelCount(vk::Extent2D extent) -> uint32_t;

  static constexpr uint32_t k_max_levels = 13;
 
private:
  void buildPipeline(const VulkanCore& core);

  struct ReducePush {
    glm::uvec2 source_size {0};
    uint32_t level_count = 0;
    uint32_t group_count = 0;
  };
 
  ShaderLayout m_layout;
  VulkanPipeline m_pipeline;
  vk::raii::Sampler m_sampler = nullptr;
 
  std::optional<vma::raii::Image> m_image;
  std::optional<vk::raii::ImageView> m_view;
  std::vector<vk::raii::ImageView> m_level_views;
  std::optional<vk::raii::DescriptorSet> m_set;
  std::optional<vma::raii::Buffer> m_counter;
  bool m_counter_cleared = false;
  sync::ImageState m_state;
  uint32_t m_levels = 0;
 
  vk::Extent2D m_built_extent {0, 0};
  uint32_t m_built_levels = 0;
  glm::mat4 m_built_view_projection {1.0f};
};
 
}