]> git.tdb.fi Git - libs/gl.git/blob - source/environmentmap.cpp
Add an EnvironmentMap effect
[libs/gl.git] / source / environmentmap.cpp
1 #include <cmath>
2 #include "environmentmap.h"
3 #include "renderer.h"
4 #include "texunit.h"
5
6 namespace Msp {
7 namespace GL {
8
9 EnvironmentMap::EnvironmentMap(unsigned s, Renderable &r, Renderable &e):
10         Effect(r),
11         size(s),
12         environment(e),
13         rendered(false)
14 {
15         env_tex.storage(RGB, size);
16         env_tex.set_wrap(CLAMP_TO_EDGE);
17         env_tex.set_min_filter(LINEAR);
18         depth_buf.storage(DEPTH_COMPONENT, size, size);
19         for(unsigned i=0; i<6; ++i)
20         {
21                 fbo[i].attach(COLOR_ATTACHMENT0, env_tex, TextureCube::enumerate_faces(i), 0);
22                 fbo[i].attach(DEPTH_ATTACHMENT, depth_buf);
23         }
24
25         // XXX Make the depth range configurable
26         camera.set_field_of_view(M_PI/2);
27         camera.set_aspect(1);
28         camera.set_depth_clip(0.1, 100);
29
30         shdata.uniform("environment", 4);
31 }
32
33 void EnvironmentMap::setup_frame() const
34 {
35         if(rendered)
36                 return;
37
38         rendered = true;
39         renderable.setup_frame();
40         environment.setup_frame();
41
42         const Matrix *matrix = renderable.get_matrix();
43         Vector3 position = (*matrix)*Vector3();
44         camera.set_position(position);
45         for(unsigned i=0; i<6; ++i)
46         {
47                 TextureCubeFace face = TextureCube::enumerate_faces(i);
48                 Bind _bind_fbo(fbo[i]);
49                 fbo[i].clear(COLOR_BUFFER_BIT|DEPTH_BUFFER_BIT);
50                 camera.set_look_direction(env_tex.get_face_direction(face));
51                 camera.set_up_direction(env_tex.get_t_direction(face));
52                 Renderer env_renderer(&camera);
53                 env_renderer.exclude(renderable);
54                 env_renderer.render(environment);
55         }
56 }
57
58 void EnvironmentMap::finish_frame() const
59 {
60         if(rendered)
61         {
62                 rendered = false;
63                 renderable.finish_frame();
64                 environment.finish_frame();
65         }
66 }
67
68 void EnvironmentMap::render(Renderer &renderer, const Tag &tag) const
69 {
70         if(!enabled_passes.count(tag))
71                 return renderer.render(renderable, tag);
72
73         const Matrix &view_matrix = renderer.get_camera()->get_matrix();
74         // XXX The camera should maybe have store its own object matrix
75         float env_mdata[9];
76         env_mdata[0] = view_matrix[0];
77         env_mdata[1] = view_matrix[4];
78         env_mdata[2] = view_matrix[8];
79         env_mdata[3] = view_matrix[1];
80         env_mdata[4] = view_matrix[5];
81         env_mdata[5] = view_matrix[9];
82         env_mdata[6] = view_matrix[2];
83         env_mdata[7] = view_matrix[6];
84         env_mdata[8] = view_matrix[10];
85         shdata.uniform_matrix3("env_eye_matrix", env_mdata);
86
87         env_tex.bind_to(4);
88         TexUnit::activate(0);
89
90         Renderer::Push _push_rend(renderer);
91         renderer.add_shader_data(shdata);
92         renderer.render(renderable, tag);
93
94         env_tex.unbind_from(4);
95         TexUnit::activate(0);
96 }
97
98 } // namespace GL
99 } // namespace Msp