ShaderProgram.hpp

#pragma once
#ifdef StevEngine_RENDERER_GL
#include "Shader.hpp"
#include "utilities/Vector3.hpp"
#include "utilities/Vector2.hpp"
#include "utilities/Color.hpp"
#include "utilities/Matrix4.hpp"
#include "utilities/Stream.hpp"
#include "visuals/Texture.hpp"
 
#include <map>
#include <cstdint>
#include <glad/gl.h>
 
namespace StevEngine::Renderer {
    class ShaderProgram {
        public:
            ShaderProgram(ShaderType shaderType, bool useDefaultShaders = true);

            ShaderProgram(const std::vector<Shader>& shaders, bool useDefaultShaders = true);

            ShaderProgram(Utilities::Stream& stream);

            ShaderProgram() {};

            uint32_t AddShader(Shader shader);

            void RemoveShader(uint32_t location);

            void RelinkProgram();

            void DeleteProgram();
 
            // Uniform setters
            void SetShaderUniform(const char* name, Utilities::Matrix4 value) const;

            void SetShaderUniform(const char* name, Utilities::Color value) const;

            void SetShaderUniform(const char* name, Utilities::Vector3 value) const;

            void SetShaderUniform(const char* name, Utilities::Vector2 value) const;

            void SetShaderUniform(const char* name, bool value) const;

            void SetShaderUniform(const char* name, int32_t value) const;

            void SetShaderUniform(const char* name, uint32_t value) const;

            void SetShaderUniform(const char* name, float value) const;

            void SetShaderUniform(const char* name, double value) const;

            uint32_t GetLocation() const { return location; }

            bool IsModified() const { return modified; }

            ShaderType GetType() const { return shaderType; };

            Utilities::Stream Export(Utilities::StreamType type) const;
 
        protected:
            uint32_t location;    
            ShaderType shaderType;  
            bool modified;        
            std::map<uint32_t, Shader> shaders;  
    };
 
    class ComputeShader : public ShaderProgram {
        public:
            ComputeShader(const char* source);

            ComputeShader(const std::vector<Shader>& shaders) : ShaderProgram(shaders) {};

            ComputeShader(Utilities::Stream& stream) : ShaderProgram(stream) {};

            void Run(const Visuals::ComputeTexture& output) const { return Run(output, output.width, output.height, 1); }
            void Run(const Visuals::ComputeTexture& output, uint32_t groupsX, uint32_t groupsY, uint32_t groupsZ = 1) const;
            void Run(uint32_t groupsX, uint32_t groupsY, uint32_t groupsZ = 1) const;
 
    };
}
#endif