OBJMesh.h

// Copyright (c) 2024 Michael Carlie. All Rights Reserved.
//
// This software is released under the MIT License.
// https://opensource.org/licenses/MIT
 
#ifndef INCLUDE_VULKANENGINE_OBJMESH_H_
#define INCLUDE_VULKANENGINE_OBJMESH_H_
 
#include <VulkanEngine/BoundingBox.h>
#include <VulkanEngine/GraphicsPipeline.h>
#include <VulkanEngine/MeshBase.h>
#include <VulkanEngine/SceneObject.h>
#include <VulkanEngine/UniformBuffer.h>
 
#include <array>
#include <filesystem>  // NOLINT(build/c++17)
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "GraphicsPipeline.h"
 
namespace VulkanEngine {
 
class OBJMesh : public SceneObject {
 public:
  OBJMesh(std::filesystem::path obj_file, std::filesystem::path mtl_path = "",
          const std::shared_ptr<Shader> _shader = std::shared_ptr<Shader>());
 
  virtual ~OBJMesh();
 
  const BoundingBox<Eigen::Vector3f>& getBoundingBox() const;
 
 private:
#pragma pack(push, 1)
  struct MvpUbo {
    Eigen::Matrix4f model;
    Eigen::Matrix4f view;
    Eigen::Matrix4f projection;
  };
 
  struct Material {
    std::array<float, 4> ambient = {0.8, 0.8, 0.8, 0.0};
    std::array<float, 4> diffuse = {1.0, 1.0, 1.0, 0.0};
    std::array<float, 4> specular = {1.0, 1.0, 1.0, 0.0};
  };
#pragma pack(pop)
 
  void update(std::shared_ptr<SceneState> scene_state) override;
 
  void loadOBJ(const char* obj_path, const char* mtl_path);
 
  const std::string getVertexShaderString() const;
 
  const std::string getFragmentShaderString(bool has_texture) const;
 
  std::vector<std::shared_ptr<MeshBase>> meshes;
 
  std::vector<std::shared_ptr<Shader>> shaders;
 
  std::vector<std::shared_ptr<GraphicsPipeline>> graphics_pipelines;
 
  std::vector<std::shared_ptr<UniformBuffer<MvpUbo>>> mvp_buffers;
 
  std::vector<std::vector<std::shared_ptr<UniformBuffer<Material>>>>
      material_buffers;
 
  std::unordered_map<std::string, std::shared_ptr<Descriptor>> textures;
 
  bool graphics_pipeline_updated;
 
  BoundingBox<Eigen::Vector3f> bounding_box;
};
 
}  // namespace VulkanEngine
 
#endif  // INCLUDE_VULKANENGINE_OBJMESH_H_