3 #include <msp/strings/format.h>
4 #include "environmentmap.h"
15 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, Renderable &r, Renderable &e):
16 EnvironmentMap(s, f, 1, r, e)
19 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, unsigned l, Renderable &r, Renderable &e):
23 irradiance_shprog(Resources::get_global().get<Program>("_envmap_irradiance.glsl.shader")),
24 specular_shprog(Resources::get_global().get<Program>("_envmap_specular.glsl.shader")),
25 fullscreen_mesh(Resources::get_global().get<Mesh>("_fullscreen_quad.mesh")),
26 sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp")),
27 mip_sampler(Resources::get_global().get<Sampler>("_mip_clamp.samp"))
29 if(!l || (1U<<(l-1))>=s)
30 throw invalid_argument("EnvironmentMap::EnvironmentMap");
32 env_tex.storage(f, size, l);
33 depth_buf.storage(DEPTH_COMPONENT32F, size, size, 1);
34 for(unsigned i=0; i<6; ++i)
36 TextureCubeFace face = static_cast<TextureCubeFace>(i);
37 faces[i].fbo.set_format((COLOR_ATTACHMENT,f, DEPTH_ATTACHMENT,DEPTH_COMPONENT32F));
38 faces[i].fbo.attach(COLOR_ATTACHMENT, env_tex, face, 0);
39 faces[i].fbo.attach(DEPTH_ATTACHMENT, depth_buf);
40 faces[i].camera.set_look_direction(TextureCube::get_face_direction(face));
41 faces[i].camera.set_up_direction(TextureCube::get_t_direction(face));
42 faces[i].camera.set_field_of_view(Geometry::Angle<float>::right());
43 faces[i].camera.set_aspect_ratio(1);
44 faces[i].camera.set_depth_clip(0.1, 100);
47 irradiance.storage(f, size/4, 1);
48 irradiance_fbo.set_format((COLOR_ATTACHMENT,f));
49 irradiance_fbo.attach_layered(COLOR_ATTACHMENT, irradiance);
53 specular_fbos.resize(l-1);
54 for(unsigned i=1; i<l; ++i)
56 specular_fbos[i-1].set_format((COLOR_ATTACHMENT,f));
57 specular_fbos[i-1].attach_layered(COLOR_ATTACHMENT, env_tex, i);
60 LinAl::Matrix<float, 3, 3> face_matrices[6];
61 for(unsigned i=0; i<6; ++i)
63 GL::TextureCubeFace face = static_cast<TextureCubeFace>(i);
64 GL::Vector3 columns[3];
65 columns[0] = GL::TextureCube::get_s_direction(face);
66 columns[1] = GL::TextureCube::get_t_direction(face);
67 columns[2] = GL::TextureCube::get_face_direction(face);
68 face_matrices[i] = LinAl::Matrix<float, 3, 3>::from_columns(columns);
71 prefilter_shdata.uniform_array("faces", 6, &face_matrices[0]);
72 prefilter_shdata.uniform("n_samples", 128);
73 prefilter_shdata.uniform("roughness", 1.0f);
76 shdata.uniform("env_world_matrix", LinAl::SquareMatrix<float, 3>::identity());
79 void EnvironmentMap::set_fixed_position(const Vector3 &p)
85 void EnvironmentMap::set_depth_clip(float n, float f)
87 for(unsigned i=0; i<6; ++i)
88 faces[i].camera.set_depth_clip(n, f);
91 void EnvironmentMap::set_update_interval(unsigned i)
94 update_delay = min(update_delay, update_interval-1);
97 void EnvironmentMap::queue_update()
102 void EnvironmentMap::setup_frame(Renderer &renderer)
108 renderable.setup_frame(renderer);
116 update_delay = update_interval-1;
117 environment.setup_frame(renderer);
121 center = fixed_position;
124 const Matrix *matrix = renderable.get_matrix();
128 center = matrix->column(3).slice<3>(0);
131 Renderer::Push push(renderer);
132 Renderer::Exclude exclude1(renderer, renderable);
133 Renderer::Exclude exclude2(renderer, *this);
135 for(unsigned i=0; i<6; ++i)
137 faces[i].camera.set_position(center);
138 renderer.set_framebuffer(&faces[i].fbo);
139 renderer.set_camera(faces[i].camera);
140 renderer.render(environment);
143 renderer.set_framebuffer(&irradiance_fbo);
144 renderer.set_shader_program(&irradiance_shprog, &prefilter_shdata);
145 renderer.set_texture("environment_map", &env_tex, &sampler);
146 fullscreen_mesh.draw(renderer);
148 renderer.set_shader_program(&specular_shprog);
149 for(unsigned i=0; i<specular_fbos.size(); ++i)
151 prefilter_shdata.uniform("roughness", 1.0f-sqrt(1.0f-static_cast<float>(i+1)/specular_fbos.size()));
152 renderer.set_framebuffer(&specular_fbos[i]);
153 fullscreen_mesh.draw(renderer);
157 void EnvironmentMap::finish_frame()
162 renderable.finish_frame();
163 environment.finish_frame();
167 void EnvironmentMap::render(Renderer &renderer, Tag tag) const
169 if(!enabled_methods.count(tag))
170 return renderer.render(renderable, tag);
172 Renderer::Push _push_rend(renderer);
174 renderer.set_texture("environment_map", &env_tex, &mip_sampler);
175 renderer.set_texture("irradiance_map", &irradiance, &sampler);
176 renderer.add_shader_data(shdata);
177 renderer.render(renderable, tag);
180 void EnvironmentMap::set_debug_name(const string &name)
183 env_tex.set_debug_name(name+"/environment.tex");
184 depth_buf.set_debug_name(name+"/environment_depth.rbuf");
185 static const char *const face_names[] = { "X+", "X-", "Y+", "Y-", "Z+", "Z-" };
186 for(unsigned i=0; i<6; ++i)
188 faces[i].fbo.set_debug_name(format("%s [FBO:%s]", name, face_names[i]));
189 faces[i].camera.set_debug_name(format("%s/%s.camera", name, face_names[i]));
192 irradiance.set_debug_name(name+"/irradiance.tex");
193 irradiance_fbo.set_debug_name(name+" [FBO:irradiance]");
194 for(unsigned i=0; i<specular_fbos.size(); ++i)
195 specular_fbos[i].set_debug_name(format("%s [FBO:specular_mip%d]", name, i+1));
196 prefilter_shdata.set_debug_name(name+" [UBO:prefilter]");
197 shdata.set_debug_name(name+" [UBO]");
204 EnvironmentMap *EnvironmentMap::Template::create(const map<string, Renderable *> &renderables) const
206 Renderable *content = get_item(renderables, content_name);
207 Renderable *environment = get_item(renderables, environment_name);
208 if(!content || !environment)
209 throw invalid_operation("EnvironmentMap::Template::create");
211 RefPtr<EnvironmentMap> env_map = new EnvironmentMap(size, format, roughness_levels, *content, *environment);
212 if(use_fixed_position)
213 env_map->set_fixed_position(fixed_position);
214 env_map->set_depth_clip(near_clip, far_clip);
216 return env_map.release();
220 DataFile::Loader::ActionMap EnvironmentMap::Template::Loader::shared_actions;
222 EnvironmentMap::Template::Loader::Loader(Template &t, Collection &c):
223 DerivedObjectLoader<Template, Effect::Template::Loader>(t, c)
225 set_actions(shared_actions);
228 void EnvironmentMap::Template::Loader::init_actions()
230 Effect::Template::Loader::init_actions();
231 add("depth_clip", &Template::near_clip, &Template::far_clip);
232 add("environment", &Template::environment_name);
233 add("fixed_position", &Loader::fixed_position);
234 add("format", &Template::format);
235 add("roughness_levels", &Template::roughness_levels);
236 add("size", &Template::size);
239 void EnvironmentMap::Template::Loader::fixed_position(float x, float y, float z)
241 obj.fixed_position = Vector3(x, y, z);
242 obj.use_fixed_position = true;