source/effects/environmentmap.cpp

   1 #include <algorithm>
   2 #include <cmath>
   3 #include <msp/strings/format.h>
   4 #include "environmentmap.h"
   5 #include "error.h"
   6 #include "mesh.h"
   7 #include "renderer.h"
   8 #include "resources.h"
   9
  10 using namespace std;
  11
  12 namespace Msp {
  13 namespace GL {
  14
  15 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, Renderable &r, Renderable &e):
  16         EnvironmentMap(s, f, 1, r, e)
  17 { }
  18
  19 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, unsigned l, Renderable &r, Renderable &e):
  20         Effect(r),
  21         size(s),
  22         environment(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"))
  28 {
  29         if(!l || (1U<<(l-1))>=s)
  30                 throw invalid_argument("EnvironmentMap::EnvironmentMap");
  31
  32         env_tex.storage(f, size, l);
  33         depth_buf.storage(DEPTH_COMPONENT32F, size, size, 1);
  34         for(unsigned i=0; i<6; ++i)
  35         {
  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);
  45         }
  46
  47         irradiance.storage(f, size/4, 1);
  48         irradiance_fbo.set_format((COLOR_ATTACHMENT,f));
  49         irradiance_fbo.attach_layered(COLOR_ATTACHMENT, irradiance);
  50
  51         if(l>1)
  52         {
  53                 specular_fbos.resize(l-1);
  54                 for(unsigned i=1; i<l; ++i)
  55                 {
  56                         specular_fbos[i-1].set_format((COLOR_ATTACHMENT,f));
  57                         specular_fbos[i-1].attach_layered(COLOR_ATTACHMENT, env_tex, i);
  58                 }
  59
  60                 LinAl::Matrix<float, 3, 3> face_matrices[6];
  61                 for(unsigned i=0; i<6; ++i)
  62                 {
  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);
  69                 }
  70
  71                 prefilter_shdata.uniform_array("faces", 6, &face_matrices[0]);
  72                 prefilter_shdata.uniform("n_samples", 128);
  73                 prefilter_shdata.uniform("roughness", 1.0f);
  74         }
  75
  76         shdata.uniform("env_world_matrix", LinAl::SquareMatrix<float, 3>::identity());
  77 }
  78
  79 void EnvironmentMap::set_fixed_position(const Vector3 &p)
  80 {
  81         fixed_position = p;
  82         use_fixed_pos = true;
  83 }
  84
  85 void EnvironmentMap::set_depth_clip(float n, float f)
  86 {
  87         for(unsigned i=0; i<6; ++i)
  88                 faces[i].camera.set_depth_clip(n, f);
  89 }
  90
  91 void EnvironmentMap::set_update_interval(unsigned i)
  92 {
  93         update_interval = i;
  94         update_delay = min(update_delay, update_interval-1);
  95 }
  96
  97 void EnvironmentMap::queue_update()
  98 {
  99         update_delay = 0;
 100 }
 101
 102 void EnvironmentMap::setup_frame(Renderer &renderer)
 103 {
 104         if(rendered)
 105                 return;
 106
 107         rendered = true;
 108         renderable.setup_frame(renderer);
 109
 110         if(update_delay)
 111         {
 112                 if(update_interval)
 113                         --update_delay;
 114                 return;
 115         }
 116         update_delay = update_interval-1;
 117         environment.setup_frame(renderer);
 118
 119         Vector3 center;
 120         if(use_fixed_pos)
 121                 center = fixed_position;
 122         else
 123         {
 124                 const Matrix *matrix = renderable.get_matrix();
 125                 if(!matrix)
 126                         return;
 127
 128                 center = matrix->column(3).slice<3>(0);
 129         }
 130
 131         Renderer::Push push(renderer);
 132         Renderer::Exclude exclude1(renderer, renderable);
 133         Renderer::Exclude exclude2(renderer, *this);
 134
 135         for(unsigned i=0; i<6; ++i)
 136         {
 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);
 141         }
 142
 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);
 147
 148         renderer.set_shader_program(&specular_shprog);
 149         for(unsigned i=0; i<specular_fbos.size(); ++i)
 150         {
 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);
 154         }
 155 }
 156
 157 void EnvironmentMap::finish_frame()
 158 {
 159         if(rendered)
 160         {
 161                 rendered = false;
 162                 renderable.finish_frame();
 163                 environment.finish_frame();
 164         }
 165 }
 166
 167 void EnvironmentMap::render(Renderer &renderer, Tag tag) const
 168 {
 169         if(!enabled_methods.count(tag))
 170                 return renderer.render(renderable, tag);
 171
 172         Renderer::Push _push_rend(renderer);
 173
 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);
 178 }
 179
 180 void EnvironmentMap::set_debug_name(const string &name)
 181 {
 182 #ifdef DEBUG
 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)
 187         {
 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]));
 190         }
 191
 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]");
 198 #else
 199         (void)name;
 200 #endif
 201 }
 202
 203
 204 EnvironmentMap *EnvironmentMap::Template::create(const map<string, Renderable *> &renderables) const
 205 {
 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");
 210
 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);
 215
 216         return env_map.release();
 217 }
 218
 219
 220 DataFile::Loader::ActionMap EnvironmentMap::Template::Loader::shared_actions;
 221
 222 EnvironmentMap::Template::Loader::Loader(Template &t, Collection &c):
 223         DerivedObjectLoader<Template, Effect::Template::Loader>(t, c)
 224 {
 225         set_actions(shared_actions);
 226 }
 227
 228 void EnvironmentMap::Template::Loader::init_actions()
 229 {
 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);
 237 }
 238
 239 void EnvironmentMap::Template::Loader::fixed_position(float x, float y, float z)
 240 {
 241         obj.fixed_position = Vector3(x, y, z);
 242         obj.use_fixed_position = true;
 243 }
 244
 245 } // namespace GL
 246 } // namespace Msp