lighting.h

#ifndef OE_SCENE_COMPONENT_LIGHTING_H
#define OE_SCENE_COMPONENT_LIGHTING_H
 
#include <map>
 
#include "../manager.h"
#include "../node.h"
 
#include "../../render/pbr.h"
 
namespace oe::render
{
    class Shader;
    class Texture;
    class Cubemap;
    class TextureManager;
}
 
namespace oe::component
{
    class PointLight;
    class AreaLight;
 
    struct ReflectionProbe
    {
        std::shared_ptr<oe::render::Cubemap> irradiance = nullptr;
        std::shared_ptr<oe::render::Cubemap> specular = nullptr;
    };
 
    class Lighting : public oe::scene::SceneComponent
    {
        public:
            Lighting(oe::scene::Manager&, oe::render::TextureManager& texture_manager);
 
            virtual void onUpdate(const std::chrono::nanoseconds /*delta*/, const int8_t /*flags = 0*/) override;
 
            void fillShader(oe::render::ShaderBase& shader) const noexcept;
 
            void fillReflectionsShader(oe::render::ShaderBase& shader) const noexcept;
 
            void fillTransparencyShader(oe::render::ShaderBase& shader) const noexcept;
 
            void setRefractionTexture(const std::shared_ptr<oe::render::Texture>& refraction_texture)
            {
                _refraction_texture = refraction_texture;
            }
 
            const std::vector<glm::mat4> getAreaLights() const
            {
                return _area_light_matrices;
            }
 
            void setDefaultReflectionProbe(std::shared_ptr<oe::render::Cubemap> cubemap);
 
            int8_t match_flags = 0;
 
        protected:
            std::vector<std::pair<PointLight*, oe::scene::Node*>> _point_lights;
            std::vector<std::pair<AreaLight*, oe::scene::Node*>> _area_lights;
 
            std::vector<glm::vec3> _point_light_positions;
            std::vector<glm::vec3> _point_light_colors;
 
            std::vector<glm::mat4> _area_light_matrices;
 
            ReflectionProbe _default_reflection_probe;
 
            oe::render::TextureManager& _texture_manager;
 
            oe::render::Pbr _pbr;
 
            std::shared_ptr<oe::render::Texture> _brdf_texture;
            std::shared_ptr<oe::render::Texture> _refraction_texture;
 
            friend PointLight;
            friend AreaLight;
    };
 
    class PointLight : public oe::scene::NodeComponent
    {
        public:
            PointLight(oe::scene::Node&);
 
            glm::vec3 color = glm::vec3(1.0);
    };
 
    class AreaLight : public oe::scene::NodeComponent
    {
        public:
 
            glm::vec3 color = glm::vec3(1.0);
    };
}
 
#endif