2 #include "ambientocclusion.h"
14 AmbientOcclusion::AmbientOcclusion(unsigned w, unsigned h, float):
15 occlude_target(w, h, (RENDER_COLOR,RGB)),
16 occlude_shader("ambientocclusion_occlude.glsl"),
17 combine_shader("ambientocclusion_combine.glsl"),
18 quad(get_fullscreen_quad())
20 texturing.attach(2, occlude_target.get_target_texture(RENDER_COLOR));
23 rotate_lookup.storage(RGBA, 4, 4);
24 rotate_lookup.set_filter(NEAREST);
25 unsigned char data[64];
26 for(unsigned i=0; i<16; ++i)
28 Geometry::Angle<float> a = Geometry::Angle<float>::from_turns(random(seed));
29 unsigned char c = (cos(a)*0.5f+0.5f)*255;
30 unsigned char s = (sin(a)*0.5f+0.5f)*255;
34 data[i*3+4] = ((i+i/4)%2)*255;
36 rotate_lookup.image(0, RGBA, UNSIGNED_BYTE, data);
38 texturing.attach(3, rotate_lookup);
40 shdata.uniform("source", 0);
41 shdata.uniform("depth", 1);
42 shdata.uniform("occlusion", 2);
43 shdata.uniform("rotate", 3);
44 shdata.uniform("inverse_projection", Matrix());
47 set_occlusion_radius(0.5f);
51 float AmbientOcclusion::random(unsigned &seed)
53 static const unsigned modulus = (1U<<31)-1;
54 seed = (seed*48271)%modulus; // minstd
55 return static_cast<float>(seed)/(modulus-1);
58 void AmbientOcclusion::set_n_samples(unsigned n)
61 throw out_of_range("AmbientOcclusion::set_n_samples");
64 float radius_divisor = (n-1)*(n-1);
65 Vector3 sample_points[32];
66 for(unsigned i=0; i<n; ++i)
68 Vector3 v(random(seed)-0.5f, random(seed)-0.5f, random(seed)-0.5f);
69 sample_points[i] = normalize(v)*(0.1f+0.9f*i*i/radius_divisor);
71 shdata.uniform3_array("sample_points", n, &sample_points[0].x);
72 shdata.uniform("n_samples", static_cast<int>(n));
75 void AmbientOcclusion::set_occlusion_radius(float r)
77 shdata.uniform("occlusion_radius", r);
80 void AmbientOcclusion::set_depth_ratio(float)
84 void AmbientOcclusion::set_darkness(float darkness)
86 shdata.uniform("darkness", darkness);
89 void AmbientOcclusion::render(Renderer &renderer, const Texture2D &color, const Texture2D &depth)
91 texturing.attach(0, color);
92 texturing.attach(1, depth);
94 if(renderer.get_camera())
95 shdata.uniform("inverse_projection", invert(renderer.get_camera()->get_projection_matrix()));
97 Renderer::Push push(renderer);
98 renderer.set_texturing(&texturing);
99 renderer.set_shader_program(&occlude_shader, &shdata);
102 BindRestore bind_fbo(occlude_target.get_framebuffer());
106 renderer.set_shader_program(&combine_shader);