Make ambient occlusion edge detection threshold adjustable

[libs/gl.git] / source / ambientocclusion.cpp

diff --git a/source/ambientocclusion.cpp b/source/ambientocclusion.cpp
index 2378bcc78cb323e21439b01c601c8f5590a4ab3c..ed7d1636d711d34ee6539fd4ddb11adbb4677f44 100644 (file)
--- a/source/ambientocclusion.cpp
+++ b/source/ambientocclusion.cpp
@@ -12,7 +12,7 @@ namespace Msp {
 namespace GL {
 
 AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float):
-       occlude_target(w, h, (RENDER_COLOR,RGB)),
+       occlude_target(w, h, (RENDER_COLOR,RED)),
        occlude_shader("ambientocclusion_occlude.glsl"),
        combine_shader("ambientocclusion_combine.glsl"),
        quad(get_fullscreen_quad())
@@ -29,10 +29,10 @@ AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float):
                Geometry::Angle<float> a = Geometry::Angle<float>::from_turns(random(seed));
                unsigned char c = (cos(a)*0.5f+0.5f)*255;
                unsigned char s = (sin(a)*0.5f+0.5f)*255;
-               data[i*3  ] = c;
-               data[i*3+1] = s;
-               data[i*3+2] = 255-s;
-               data[i*3+4] = ((i+i/4)%2)*255;
+               data[i*4  ] = c;
+               data[i*4+1] = s;
+               data[i*4+2] = 255-s;
+               data[i*4+3] = ((i+i/4)%2)*255;
        }
        rotate_lookup.image(0, RGBA, UNSIGNED_BYTE, data);
 
@@ -47,6 +47,7 @@ AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float):
        set_n_samples(16);
        set_occlusion_radius(0.5f);
        set_darkness(1.0f);
+       set_edge_depth_threshold(0.1f);
 }
 
 float AmbientOcclusion::random(unsigned &seed)
@@ -87,6 +88,11 @@ void AmbientOcclusion::set_darkness(float darkness)
        shdata.uniform("darkness", darkness);
 }
 
+void AmbientOcclusion::set_edge_depth_threshold(float edt)
+{
+       shdata.uniform("edge_depth_threshold", edt);
+}
+
 void AmbientOcclusion::render(Renderer &renderer, const Texture2D &color, const Texture2D &depth)
 {
        texturing.attach(0, color);
@@ -112,7 +118,8 @@ void AmbientOcclusion::render(Renderer &renderer, const Texture2D &color, const
 AmbientOcclusion::Template::Template():
        n_samples(16),
        occlusion_radius(0.5f),
-       darkness(1.0f)
+       darkness(1.0f),
+       edge_depth_threshold(0.1f)
 { }
 
 AmbientOcclusion *AmbientOcclusion::Template::create(unsigned width, unsigned height) const
@@ -121,6 +128,7 @@ AmbientOcclusion *AmbientOcclusion::Template::create(unsigned width, unsigned he
        ao->set_n_samples(n_samples);
        ao->set_occlusion_radius(occlusion_radius);
        ao->set_darkness(darkness);
+       ao->set_edge_depth_threshold(edge_depth_threshold);
        return ao.release();
 }
 
@@ -129,6 +137,7 @@ AmbientOcclusion::Template::Loader::Loader(Template &t):
        DataFile::DerivedObjectLoader<Template, PostProcessor::Template::Loader>(t)
 {
        add("darkness", &Template::darkness);
+       add("edge_depth_threshold", &Template::edge_depth_threshold);
        add("occlusion_radius", &Template::occlusion_radius);
        add("samples", &Template::n_samples);
 }