ChromaticAberration.cpp

Go to the documentation of this file.

 
 
#include <glad/glad.h>
 
#include "galaxy/core/Config.hpp"
#include "galaxy/core/ServiceLocator.hpp"
 
#include "ChromaticAberration.hpp"
 
constexpr const char* const chromaticaberration_vert = R"(
    #version 460 core
    layout (location = 0) in vec2 l_pos;
    layout (location = 1) in vec2 l_texels;
 
    void main()
    {
        gl_Position = vec4(l_pos, 0.0, 1.0);
    }
)";
 
constexpr const char* const chromaticaberration_frag = R"(
    #version 460 core
 
    out vec4 io_frag_colour;
 
    uniform vec2 u_direction;
    uniform vec3 u_rgb_offset;
    uniform sampler2D u_texture;
 
    void main()
    {
        vec2 tex_size = textureSize(u_texture, 0).xy;
        vec2 tex_coord = gl_FragCoord.xy / tex_size;
 
        vec2 direction = tex_coord - u_direction;
 
        io_frag_colour = texture(u_texture, tex_coord);
        io_frag_colour.r = texture(u_texture, tex_coord + (direction * vec2(u_rgb_offset.x))).r;
        io_frag_colour.g = texture(u_texture, tex_coord + (direction * vec2(u_rgb_offset.y))).g;
        io_frag_colour.b = texture(u_texture, tex_coord + (direction * vec2(u_rgb_offset.z))).b;
    }
)";
 
namespace galaxy
{
    namespace graphics
    {
        ChromaticAberration::ChromaticAberration(const int width, const int height)
            : m_r_offset {0.009f}
            , m_g_offset {0.006f}
            , m_b_offset {-0.006f}
        {
            m_fb.create(width, height);
 
            if (m_shader.parse(chromaticaberration_vert, chromaticaberration_frag))
            {
                m_shader.compile();
                m_shader.set_uniform("u_texture", 0);
                m_shader.set_uniform("u_rgb_offset", m_r_offset, m_g_offset, m_b_offset);
                m_shader.set_uniform("u_direction", 0.509167f, 0.598f); // may need tweaking, in general 0.5, 0.5 is default, adjust to change direction.
            }
        }
 
        void ChromaticAberration::resize(const int width, const int height)
        {
            m_fb.recreate(width, height);
        }
 
        unsigned int ChromaticAberration::render(const unsigned int input)
        {
            m_fb.bind(true);
            m_shader.bind();
 
            glActiveTexture(GL_TEXTURE0);
            glBindTexture(GL_TEXTURE_2D, input);
            glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
 
            return m_fb.texture();
        }
 
        void ChromaticAberration::set_r_offset(const float r)
        {
            m_r_offset = std::clamp(r, -1.0f, 1.0f);
            m_shader.set_uniform("u_rgb_offset", m_r_offset, m_g_offset, m_b_offset);
        }
 
        void ChromaticAberration::set_g_offset(const float g)
        {
            m_g_offset = std::clamp(g, -1.0f, 1.0f);
            m_shader.set_uniform("u_rgb_offset", m_r_offset, m_g_offset, m_b_offset);
        }
 
        void ChromaticAberration::set_b_offset(const float b)
        {
            m_b_offset = std::clamp(b, -1.0f, 1.0f);
            m_shader.set_uniform("u_rgb_offset", m_r_offset, m_g_offset, m_b_offset);
        }
 
        void ChromaticAberration::set_rgb_offset(const float r, const float g, const float b)
        {
            m_r_offset = std::clamp(r, -1.0f, 1.0f);
            m_g_offset = std::clamp(g, -1.0f, 1.0f);
            m_b_offset = std::clamp(b, -1.0f, 1.0f);
 
            m_shader.set_uniform("u_rgb_offset", m_r_offset, m_g_offset, m_b_offset);
        }
 
        bool ChromaticAberration::is_enabled()
        {
            return core::ServiceLocator<core::Config>::ref().get<bool>("chromatic_abberation", "graphics.effects");
        }
    } // namespace graphics
} // namespace galaxy