mesh.hpp

Go to the documentation of this file.

#pragma once
 
#include <vk_mem_alloc.h>
#include <vulkan/vulkan_core.h>
 
#include <vulkan/vulkan.hpp>
#include <wren/math/matrix.hpp>
#include <wren/math/vector.hpp>
#include <wren/vk/buffer.hpp>
#include <wren/vk/shader.hpp>
 
#include "utils/device.hpp"
 
namespace wren {
 
struct UBO {
  wren::math::Mat4f model;
};
 
struct Vertex {
  wren::math::Vec3f pos;
  wren::math::Vec3f normal;
  wren::math::Vec4f colour;
};
 
const std::array kTriangleVertices = {
    Vertex{.pos = {0.0f, -0.5f, 0.0f}, .normal = {1.0f, 0.0f, 0.0f}},
    Vertex{.pos = {0.5f, 0.5f, 0.0f}, .normal = {0.0f, 1.0f, 0.0f}},
    Vertex{.pos = {-0.5f, 0.5f, 0.0f}, .normal = {0.0f, 0.0f, 1.0f}},
};
 
const std::array kQuadVertices = {
    Vertex{
        .pos = {-0.5f, -0.5f, 0.0f},
        .normal = {1.0f, 0.0f, 0.0f},
        .colour = {1, 1, 1, 1},
    },
    Vertex{
        .pos = {0.5f, -0.5f, 0.0f},
        .normal = {0.0f, 1.0f, 0.0f},
        .colour = {1, 1, 1, 1},
    },
    Vertex{
        .pos = {0.5f, 0.5f, 0.0f},
        .normal = {0.0f, 0.0f, 1.0f},
        .colour = {1, 1, 1, 1},
    },
    Vertex{
        .pos = {-0.5f, 0.5f, 0.0f},
        .normal = {1.0f, 1.0f, 1.0f},
        .colour = {1, 1, 1, 1},
    },
};
 
const std::vector<uint16_t> kQuadIndices = {0, 2, 1, 0, 3, 2};
 
class Mesh {
 public:
  static auto create_quad() -> Mesh {
    return {std::vector<Vertex>{kQuadVertices.begin(), kQuadVertices.end()},
            std::vector<uint16_t>{kQuadIndices.begin(), kQuadIndices.end()}};
  }
 
  Mesh() = default;
 
  Mesh(const vulkan::Device& device, VmaAllocator allocator);
  Mesh(const std::vector<Vertex>& vertices,
       const std::vector<uint16_t>& indices);
 
  void load(const vulkan::Device& device, VmaAllocator allocator);
 
  void shader(const std::shared_ptr<vk::Shader>& shader) { shader_ = shader; }
  void draw(const ::vk::CommandBuffer& cmd) const;
  void bind(const ::vk::CommandBuffer& cmd) const;
 
  [[nodiscard]] auto loaded() const { return loaded_; }
 
 private:
  bool loaded_ = false;
 
  std::shared_ptr<vk::Shader> shader_;
  std::vector<Vertex> vertices_;
  std::vector<uint16_t> indices_;
  std::shared_ptr<vk::Buffer> index_buffer_;
  std::shared_ptr<vk::Buffer> vertex_buffer_;
  std::shared_ptr<vk::Buffer> uniform_buffer_;
};
 
}  // namespace wren