source/effects/environmentmap.cpp

   1 #include <algorithm>
   2 #include <cmath>
   3 #include <msp/core/raii.h>
   4 #include <msp/strings/format.h>
   5 #include "environmentmap.h"
   6 #include "error.h"
   7 #include "mesh.h"
   8 #include "renderer.h"
   9 #include "resources.h"
  10
  11 using namespace std;
  12
  13 namespace Msp {
  14 namespace GL {
  15
  16 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, Renderable &c, Renderable &e):
  17         EnvironmentMap(s, f, 1, c, e)
  18 { }
  19
  20 EnvironmentMap::EnvironmentMap(unsigned s, PixelFormat f, unsigned l, Renderable &c, Renderable &e):
  21         Effect(c),
  22         size(s),
  23         environment(e),
  24         irradiance_shprog(Resources::get_global().get<Program>("_envmap_irradiance.glsl.shader")),
  25         specular_shprog(Resources::get_global().get<Program>("_envmap_specular.glsl.shader")),
  26         fullscreen_mesh(Resources::get_global().get<Mesh>("_fullscreen_quad.mesh")),
  27         sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp")),
  28         mip_sampler(Resources::get_global().get<Sampler>("_mip_clamp.samp"))
  29 {
  30         if(!l || (1U<<(l-1))>=s)
  31                 throw invalid_argument("EnvironmentMap::EnvironmentMap");
  32
  33         env_tex.storage(f, size, l);
  34         depth_buf.storage(DEPTH_COMPONENT32F, size, size, 1);
  35         for(unsigned i=0; i<6; ++i)
  36         {
  37                 TextureCubeFace face = static_cast<TextureCubeFace>(i);
  38                 faces[i].fbo.set_format((COLOR_ATTACHMENT,f, DEPTH_ATTACHMENT,DEPTH_COMPONENT32F));
  39                 faces[i].fbo.attach(COLOR_ATTACHMENT, env_tex, face, 0);
  40                 faces[i].fbo.attach(DEPTH_ATTACHMENT, depth_buf);
  41                 faces[i].camera.set_look_direction(TextureCube::get_face_direction(face));
  42                 faces[i].camera.set_up_direction(TextureCube::get_t_direction(face));
  43                 faces[i].camera.set_field_of_view(Geometry::Angle<float>::right());
  44                 faces[i].camera.set_aspect_ratio(1);
  45                 faces[i].camera.set_depth_clip(0.1, 100);
  46         }
  47
  48         irradiance.storage(f, size/4, 1);
  49         irradiance_fbo.set_format((COLOR_ATTACHMENT,f));
  50         irradiance_fbo.attach_layered(COLOR_ATTACHMENT, irradiance);
  51
  52         if(l>1)
  53         {
  54                 specular_fbos.resize(l-1);
  55                 for(unsigned i=0; i+1<l; ++i)
  56                 {
  57                         specular_fbos[i].set_format((COLOR_ATTACHMENT,f));
  58                         specular_fbos[i].attach_layered(COLOR_ATTACHMENT, env_tex, i+1);
  59                 }
  60
  61                 LinAl::Matrix<float, 3, 3> face_matrices[6];
  62                 for(unsigned i=0; i<6; ++i)
  63                 {
  64                         GL::TextureCubeFace face = static_cast<TextureCubeFace>(i);
  65                         GL::Vector3 columns[3];
  66                         columns[0] = GL::TextureCube::get_s_direction(face);
  67                         columns[1] = GL::TextureCube::get_t_direction(face);
  68                         columns[2] = GL::TextureCube::get_face_direction(face);
  69                         face_matrices[i] = LinAl::Matrix<float, 3, 3>::from_columns(columns);
  70                 }
  71
  72                 prefilter_shdata.uniform_array("faces", 6, &face_matrices[0]);
  73                 prefilter_shdata.uniform("n_samples", 128);
  74                 prefilter_shdata.uniform("roughness", 1.0f);
  75         }
  76
  77         shdata.uniform("env_world_matrix", LinAl::Matrix<float, 3, 3>::identity());
  78 }
  79
  80 void EnvironmentMap::set_fixed_position(const Vector3 &p)
  81 {
  82         fixed_position = p;
  83         use_fixed_pos = true;
  84 }
  85
  86 void EnvironmentMap::set_depth_clip(float n, float f)
  87 {
  88         for(unsigned i=0; i<6; ++i)
  89                 faces[i].camera.set_depth_clip(n, f);
  90 }
  91
  92 void EnvironmentMap::set_update_interval(unsigned i)
  93 {
  94         update_interval = i;
  95         update_delay = min(update_delay, update_interval-1);
  96 }
  97
  98 void EnvironmentMap::queue_update()
  99 {
 100         update_delay = 0;
 101 }
 102
 103 void EnvironmentMap::setup_frame(Renderer &renderer)
 104 {
 105         if(rendered)
 106                 return;
 107
 108         rendered = true;
 109         content.setup_frame(renderer);
 110
 111         if(update_delay)
 112         {
 113                 if(update_interval)
 114                         --update_delay;
 115                 return;
 116         }
 117         update_delay = update_interval-1;
 118         environment.setup_frame(renderer);
 119
 120         SetFlag set_in(in_setup_frame);
 121
 122         Vector3 center;
 123         if(use_fixed_pos)
 124                 center = fixed_position;
 125         else
 126         {
 127                 const Matrix *matrix = content.get_matrix();
 128                 if(!matrix)
 129                         return;
 130
 131                 center = matrix->column(3).slice<3>(0);
 132         }
 133
 134         Renderer::Push push(renderer);
 135
 136         for(unsigned i=0; i<6; ++i)
 137         {
 138                 faces[i].camera.set_position(center);
 139                 renderer.set_framebuffer(&faces[i].fbo);
 140                 renderer.set_camera(faces[i].camera);
 141                 environment.render(renderer);
 142         }
 143
 144         renderer.set_pipeline_key(this);
 145         renderer.set_framebuffer(&irradiance_fbo);
 146         renderer.set_shader_program(&irradiance_shprog, &prefilter_shdata);
 147         renderer.set_texture("environment_map", &env_tex, 0, &sampler);
 148         fullscreen_mesh.draw(renderer);
 149
 150         renderer.set_shader_program(&specular_shprog);
 151         for(unsigned i=0; i<specular_fbos.size(); ++i)
 152         {
 153                 prefilter_shdata.uniform("roughness", 1.0f-sqrt(1.0f-static_cast<float>(i+1)/specular_fbos.size()));
 154                 renderer.set_pipeline_key(this, 1+i);
 155                 renderer.set_framebuffer(&specular_fbos[i]);
 156                 fullscreen_mesh.draw(renderer);
 157         }
 158 }
 159
 160 void EnvironmentMap::finish_frame()
 161 {
 162         if(rendered)
 163         {
 164                 rendered = false;
 165                 content.finish_frame();
 166                 environment.finish_frame();
 167         }
 168 }
 169
 170 void EnvironmentMap::render(Renderer &renderer, Tag tag) const
 171 {
 172         if(in_setup_frame)
 173                 return;
 174         if(!is_enabled_for_method(tag))
 175                 return content.render(renderer, tag);
 176
 177         Renderer::Push _push_rend(renderer);
 178
 179         renderer.set_texture("environment_map", &env_tex, &mip_sampler);
 180         renderer.set_texture("irradiance_map", &irradiance, &sampler);
 181         renderer.add_shader_data(shdata);
 182         content.render(renderer, tag);
 183 }
 184
 185 void EnvironmentMap::set_debug_name(const string &name)
 186 {
 187 #ifdef DEBUG
 188         env_tex.set_debug_name(name+"/environment.tex");
 189         depth_buf.set_debug_name(name+"/environment_depth.rbuf");
 190         static const char *const face_names[] = { "X+", "X-", "Y+", "Y-", "Z+", "Z-" };
 191         for(unsigned i=0; i<6; ++i)
 192         {
 193                 faces[i].fbo.set_debug_name(format("%s [FBO:%s]", name, face_names[i]));
 194                 faces[i].camera.set_debug_name(format("%s/%s.camera", name, face_names[i]));
 195         }
 196
 197         irradiance.set_debug_name(name+"/irradiance.tex");
 198         irradiance_fbo.set_debug_name(name+" [FBO:irradiance]");
 199         for(unsigned i=0; i<specular_fbos.size(); ++i)
 200                 specular_fbos[i].set_debug_name(format("%s [FBO:specular_mip%d]", name, i+1));
 201         prefilter_shdata.set_debug_name(name+" [UBO:prefilter]");
 202         shdata.set_debug_name(name+" [UBO]");
 203 #else
 204         (void)name;
 205 #endif
 206 }
 207
 208
 209 EnvironmentMap *EnvironmentMap::Template::create(const map<string, Renderable *> &renderables) const
 210 {
 211         Renderable *content = get_item(renderables, content_name);
 212         Renderable *environment = get_item(renderables, environment_name);
 213         if(!content || !environment)
 214                 throw invalid_operation("EnvironmentMap::Template::create");
 215
 216         RefPtr<EnvironmentMap> env_map = new EnvironmentMap(size, format, roughness_levels, *content, *environment);
 217         if(use_fixed_position)
 218                 env_map->set_fixed_position(fixed_position);
 219         env_map->set_depth_clip(near_clip, far_clip);
 220
 221         create_base(*env_map);
 222
 223         return env_map.release();
 224 }
 225
 226
 227 DataFile::Loader::ActionMap EnvironmentMap::Template::Loader::shared_actions;
 228
 229 EnvironmentMap::Template::Loader::Loader(Template &t, Collection &c):
 230         DerivedObjectLoader<Template, Effect::Template::Loader>(t, c)
 231 {
 232         set_actions(shared_actions);
 233 }
 234
 235 void EnvironmentMap::Template::Loader::init_actions()
 236 {
 237         Effect::Template::Loader::init_actions();
 238         add("depth_clip", &Template::near_clip, &Template::far_clip);
 239         add("environment", &Template::environment_name);
 240         add("fixed_position", &Loader::fixed_position);
 241         add("format", &Template::format);
 242         add("roughness_levels", &Template::roughness_levels);
 243         add("size", &Template::size);
 244 }
 245
 246 void EnvironmentMap::Template::Loader::fixed_position(float x, float y, float z)
 247 {
 248         obj.fixed_position = Vector3(x, y, z);
 249         obj.use_fixed_position = true;
 250 }
 251
 252 } // namespace GL
 253 } // namespace Msp