X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fambientocclusion.cpp;h=ed7d1636d711d34ee6539fd4ddb11adbb4677f44;hp=53b51e127d0993ce8e15093d1d1b1dc25c59c192;hb=99d25b5ef615a23ef63645fea87596b3384b5ede;hpb=d1e3975c163694d7bca6417463462be950019e5e diff --git a/source/ambientocclusion.cpp b/source/ambientocclusion.cpp index 53b51e12..ed7d1636 100644 --- a/source/ambientocclusion.cpp +++ b/source/ambientocclusion.cpp @@ -1,92 +1,146 @@ -#define _USE_MATH_DEFINES -#include +#include #include "ambientocclusion.h" #include "blend.h" +#include "camera.h" #include "renderer.h" #include "shader.h" #include "tests.h" +using namespace std; + namespace Msp { namespace GL { -AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float depth_ratio): +AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float): + occlude_target(w, h, (RENDER_COLOR,RED)), occlude_shader("ambientocclusion_occlude.glsl"), combine_shader("ambientocclusion_combine.glsl"), quad(get_fullscreen_quad()) { - occlusion.storage(RGB, w, h); - occlusion.set_min_filter(NEAREST); - occlusion.set_mag_filter(NEAREST); - occlusion.set_wrap(CLAMP_TO_EDGE); - fbo.attach(COLOR_ATTACHMENT0, occlusion, 0); - fbo.require_complete(); - - combine_texturing.attach(2, occlusion); - - rotate_lookup.storage(RGBA, 4, 4); - rotate_lookup.set_min_filter(NEAREST); - rotate_lookup.set_mag_filter(NEAREST); + occlude_target.set_texture_filter(LINEAR); + texturing.attach(2, occlude_target.get_target_texture(RENDER_COLOR)); + + unsigned seed = 1; + rotate_lookup.storage(RGBA, 4, 4, 1); + rotate_lookup.set_filter(NEAREST); unsigned char data[64]; for(unsigned i=0; i<16; ++i) { - float a = ((i*541)%16)*M_PI/32; - float c = cos(a); - float s = sin(a); - data[i*3 ] = static_cast(127+c*127); - data[i*3+1] = static_cast(127+s*127); - data[i*3+2] = static_cast(127-s*127); - data[i*3+4] = static_cast(127+c*127); + Geometry::Angle a = Geometry::Angle::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*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); - occlude_texturing.attach(1, rotate_lookup); + texturing.attach(3, rotate_lookup); - occlude_shdata.uniform("depth", 0); - occlude_shdata.uniform("rotate", 1); - occlude_shdata.uniform("screen_size", static_cast(w), static_cast(h)); + shdata.uniform("source", 0); + shdata.uniform("depth", 1); + shdata.uniform("occlusion", 2); + shdata.uniform("rotate", 3); + shdata.uniform("inverse_projection", Matrix()); + + set_n_samples(16); + set_occlusion_radius(0.5f); + set_darkness(1.0f); + set_edge_depth_threshold(0.1f); +} + +float AmbientOcclusion::random(unsigned &seed) +{ + static const unsigned modulus = (1U<<31)-1; + seed = (static_cast(seed)*48271)%modulus; // minstd + return static_cast(seed)/(modulus-1); +} - combine_shdata.uniform("source", 1); - combine_shdata.uniform("depth", 0); - combine_shdata.uniform("occlusion", 2); - combine_shdata.uniform("screen_size", static_cast(w), static_cast(h)); +void AmbientOcclusion::set_n_samples(unsigned n) +{ + if(n<1 || n>32) + throw out_of_range("AmbientOcclusion::set_n_samples"); - set_depth_ratio(depth_ratio); - set_darkness(1.5); + unsigned seed = 1; + float radius_divisor = (n-1)*(n-1); + Vector3 sample_points[32]; + for(unsigned i=0; i(n)); } -void AmbientOcclusion::set_depth_ratio(float depth_ratio) +void AmbientOcclusion::set_occlusion_radius(float r) { - depth_ratio = 1/depth_ratio; + shdata.uniform("occlusion_radius", r); +} - occlude_shdata.uniform("depth_ratio", depth_ratio, 1+depth_ratio); - combine_shdata.uniform("depth_ratio", depth_ratio, 1+depth_ratio); +void AmbientOcclusion::set_depth_ratio(float) +{ } void AmbientOcclusion::set_darkness(float darkness) { - occlude_shdata.uniform("darkness", 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) { - occlude_texturing.attach(0, depth); - combine_texturing.attach(0, depth); - combine_texturing.attach(1, color); + texturing.attach(0, color); + texturing.attach(1, depth); + + if(renderer.get_camera()) + shdata.uniform("inverse_projection", invert(renderer.get_camera()->get_projection_matrix())); + Renderer::Push push(renderer); + renderer.set_texturing(&texturing); + renderer.set_shader_program(&occlude_shader, &shdata); { - Renderer::Push push(renderer); - BindRestore bind_fbo(fbo); - renderer.set_texturing(&occlude_texturing); - renderer.set_shader_program(&occlude_shader, &occlude_shdata); + BindRestore bind_fbo(occlude_target.get_framebuffer()); quad.draw(renderer); } - Renderer::Push push(renderer); - renderer.set_texturing(&combine_texturing); - renderer.set_shader_program(&combine_shader, &combine_shdata); + renderer.set_shader_program(&combine_shader); quad.draw(renderer); } + +AmbientOcclusion::Template::Template(): + n_samples(16), + occlusion_radius(0.5f), + darkness(1.0f), + edge_depth_threshold(0.1f) +{ } + +AmbientOcclusion *AmbientOcclusion::Template::create(unsigned width, unsigned height) const +{ + RefPtr ao = new AmbientOcclusion(width/size_divisor, height/size_divisor); + 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(); +} + + +AmbientOcclusion::Template::Loader::Loader(Template &t): + DataFile::DerivedObjectLoader(t) +{ + add("darkness", &Template::darkness); + add("edge_depth_threshold", &Template::edge_depth_threshold); + add("occlusion_radius", &Template::occlusion_radius); + add("samples", &Template::n_samples); +} + } // namespace GL } // namespace Msp