shader_manager.hpp

#pragma once
#include <unordered_map>
#include <array>
#include "globals.hpp"
 
struct ObjectData;
struct LightData;
enum class ShaderType {
  kVertex,
  kFragment,
  kGeometry
};
 
enum FragmentNames {
  kFragmentEmpty = 0,
  kFragmentUnlit,
  kFragmentSpotLight,
  kFragmentScreenQuad,
  kFragmentDirectional,
  kFragmentPointLight,
  kFragmentPointDepth
};
enum VertexNames {
  kVertexLightProjection,
  kVertexUnlit,
  kVertexSpotLight,
  kVertexScreenQuad,
  kVertexDirectional,
  kVertexPointLight,
  kVertexPointDepth
};
enum GeometryNames {
  kGeometryPointDepth
};
 
//light uniforms
static constexpr std::string_view  kULightColor = "u_light.light_color";
static constexpr std::string_view  kULightPosition = "u_light.light_position";
static constexpr std::string_view  kULightConstant = "u_light.constant_falloff";
static constexpr std::string_view  kULightLinear = "u_light.linear_falloff";
static constexpr std::string_view  kULightQuadratic = "u_light.quadratic_falloff";
static constexpr std::string_view  kULightDirection = "u_light.light_direction";
static constexpr std::string_view  kULightCutoff = "u_light.cut_off";
static constexpr std::string_view  kULightOuterCutoff = "u_light.outer_cut_off";
static constexpr std::string_view  kULightIntensity = "u_light.intensity";
 
static constexpr std::string_view  kUObjectColor = "u_color";
static constexpr std::string_view  kUObjectTransform = "u_m_matrix";
static constexpr std::string_view  kUObjectRoughness = "u_roughness";
static constexpr std::string_view kUObjectShininess = "u_shininess";
 
static constexpr std::string_view  u_view_matrix = "u_view_matrix";
static constexpr std::string_view  u_projection_matrix = "u_projection_matrix";
static constexpr std::string_view  u_light_space_matrix = "u_light_space_matrix";
 
static constexpr std::string_view  u_ambient_strength = "u_ambient_strength";
static constexpr std::string_view  u_camera_position = "u_camera_position";
 
 
 
struct LightCmp;
struct TransformCmp;
class ShaderManager {
public:
  ShaderManager();
  ~ShaderManager();
  ShaderManager(const ShaderManager& other) = delete;
  ShaderManager(const ShaderManager&& other) = delete;
  ShaderManager& operator=(const ShaderManager& other) = delete;
  ShaderManager& operator=(const ShaderManager&& other) = delete;
 
  bool CreateDefaultShaders();
 
  bool CreateShader(unsigned int shader, ShaderType type,
                    const std::string& source);
 
  void attachShaders(unsigned int program, unsigned int fragment,
                     unsigned int vertex, unsigned int geometry = -1) const;
 
  void useProgram(unsigned int program) const;
 
  void setAllUniform(DataType dt, const float* number,
                     const std::string_view& uniform_name) const;
 
  void uploadLightData(const LightData& data,
                           unsigned int program) const;
  void uploadObjectData(const ObjectData& data,
                         unsigned int program) const;
 
  void setUniformValue(DataType dt, const float* number, unsigned int program,
                       const std::string_view& uniform_name, int quantity = 1) const;
 
private:
  std::array<unsigned int, 10> fragments_;
  std::array<unsigned int, 10> vertex_;
  std::array<unsigned int, 5> geometry_;
  std::array<unsigned int, 12> programs_;
};