]> git.tdb.fi Git - libs/gl.git/blob - source/bloom.cpp
Initialize Texture::use_srgb_format
[libs/gl.git] / source / bloom.cpp
1 #include <cmath>
2 #include <msp/strings/format.h>
3 #include "blend.h"
4 #include "bloom.h"
5 #include "misc.h"
6 #include "renderer.h"
7 #include "shader.h"
8 #include "tests.h"
9 #include "texunit.h"
10
11 using namespace std;
12
13 namespace Msp {
14 namespace GL {
15
16 Bloom::Bloom(unsigned w, unsigned h):
17         blur_shader("bloom_blur.glsl"),
18         combine_shader("bloom_combine.glsl"),
19         quad(get_fullscreen_quad())
20 {
21         blur_shdata[0].uniform("delta", 1.0f/w, 0.0f);
22         blur_shdata[1].uniform("delta", 0.0f, 1.0f/h);
23
24         for(unsigned i=0; i<2; ++i)
25                 target[i] = new RenderTarget(w, h, (RENDER_COLOR,RGB16F));
26         target[1]->set_texture_filter(LINEAR);
27
28         common_shdata.uniform("source", 0);
29         common_shdata.uniform("blurred", 1);
30
31         combine_texturing.attach(1, target[1]->get_target_texture(RENDER_COLOR));
32
33         set_radius(2.0f);
34         set_strength(0.2f);
35 }
36
37 Bloom::~Bloom()
38 {
39         for(unsigned i=0; i<2; ++i)
40                 delete target[i];
41 }
42
43 void Bloom::set_radius(float r)
44 {
45         if(r<=0.0f)
46                 throw invalid_argument("Bloom::set_radius");
47
48         int size = min(static_cast<int>(r*3.0f), 9);
49         common_shdata.uniform("size", size);
50
51         vector<float> factors(size*2+1);
52         float sum = 0.0f;
53         r = 2*r*r;
54         for(int i=-size; i<=size; ++i)
55                 sum += (factors[size+i] = exp(-i*i/r));
56         for(int i=0; i<=size*2; ++i)
57                 factors[i] /= sum;
58
59         common_shdata.uniform1_array("factors", size*2+1, &factors.front());
60 }
61
62 void Bloom::set_strength(float s)
63 {
64         if(s<0.0f || s>1.0f)
65                 throw invalid_argument("Bloom::set_strength");
66         common_shdata.uniform("strength", s);
67 }
68
69 void Bloom::render(Renderer &renderer, const Texture2D &src, const Texture2D &)
70 {
71         Renderer::Push push(renderer);
72         renderer.set_shader_program(&blur_shader, &common_shdata);
73         for(unsigned i=0; i<2; ++i)
74         {
75                 BindRestore bind_fbo(target[i]->get_framebuffer());
76                 Renderer::Push push2(renderer);
77                 renderer.set_texture(i ? &target[0]->get_target_texture(RENDER_COLOR) : &src);
78                 renderer.add_shader_data(blur_shdata[i]);
79                 quad.draw(renderer);
80         }
81
82         combine_texturing.attach(0, src);
83         renderer.set_texturing(&combine_texturing);
84         renderer.set_shader_program(&combine_shader);
85         quad.draw(renderer);
86 }
87
88
89 Bloom::Template::Template():
90         radius(2.0f),
91         strength(0.2f)
92 { }
93
94 Bloom *Bloom::Template::create(unsigned width, unsigned height) const
95 {
96         RefPtr<Bloom> bloom = new Bloom(width/size_divisor, height/size_divisor);
97         bloom->set_radius(radius);
98         bloom->set_strength(strength);
99         return bloom.release();
100 }
101
102
103 Bloom::Template::Loader::Loader(Template &t):
104         DataFile::DerivedObjectLoader<Template, PostProcessor::Template::Loader>(t)
105 {
106         add("strength", &Template::strength);
107         add("radius", &Template::radius);
108 }
109
110 } // namespace GL
111 } // namespace Msp