source/effects/sky.cpp

   1 #include <msp/geometry/hypersphere.h>
   2 #include <msp/geometry/ray.h>
   3 #include "directionallight.h"
   4 #include "error.h"
   5 #include "mesh.h"
   6 #include "renderer.h"
   7 #include "resources.h"
   8 #include "texture2d.h"
   9 #include "sky.h"
  10
  11 using namespace std;
  12
  13 namespace Msp {
  14 namespace GL {
  15
  16 Sky::Sky(Renderable &c, DirectionalLight &s):
  17         Effect(c),
  18         sun(s),
  19         transmittance_lookup(128, 64, (COLOR_ATTACHMENT,RGBA16F)),
  20         transmittance_shprog(Resources::get_global().get<Program>("_sky_transmittance.glsl.shader")),
  21         transmittance_lookup_dirty(true),
  22         distant(256, 128, (COLOR_ATTACHMENT,RGBA16F)),
  23         distant_shprog(Resources::get_global().get<Program>("_sky_distant.glsl.shader")),
  24         fullscreen_mesh(Resources::get_global().get<Mesh>("_fullscreen_quad.mesh")),
  25         backdrop_shprog(Resources::get_global().get<Program>("_sky_backdrop.glsl.shader")),
  26         sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp")),
  27         wrap_sampler(Resources::get_global().get<Sampler>("_linear_clamp_v.samp")),
  28         rendered(false)
  29 {
  30         shdata.uniform("n_steps", 50);
  31
  32         set_planet(Planet::earth());
  33         set_view_height(5.0f);
  34 }
  35
  36 void Sky::set_planet(const Planet &p)
  37 {
  38         planet = p;
  39         shdata.uniform("events.rayleigh_scatter", planet.rayleigh_scatter.r, planet.rayleigh_scatter.g, planet.rayleigh_scatter.b);
  40         shdata.uniform("events.mie_scatter", planet.mie_scatter.r, planet.mie_scatter.g, planet.mie_scatter.b);
  41         shdata.uniform("events.mie_absorb", planet.mie_absorb.r, planet.mie_absorb.g, planet.mie_absorb.b);
  42         shdata.uniform("events.ozone_absorb", planet.ozone_absorb.r, planet.ozone_absorb.g, planet.ozone_absorb.b);
  43         shdata.uniform("rayleigh_density_decay", planet.rayleigh_density_decay);
  44         shdata.uniform("mie_density_decay", planet.mie_density_decay);
  45         shdata.uniform("ozone_band_center", planet.ozone_band_center);
  46         shdata.uniform("ozone_band_extent", planet.ozone_band_extent);
  47         shdata.uniform("atmosphere_thickness", planet.atmosphere_thickness);
  48         shdata.uniform("planet_radius", planet.planet_radius);
  49         shdata.uniform("ground_albedo", planet.ground_albedo.r, planet.ground_albedo.g, planet.ground_albedo.b);
  50 }
  51
  52 void Sky::set_view_height(float h)
  53 {
  54         view_height = h;
  55         shdata.uniform("view_height", h);
  56 }
  57
  58 Color Sky::get_transmittance(const Vector3 &look_dir)
  59 {
  60         Vector3 pos(0.0f, 0.0f, view_height);
  61         Vector3 planet_center(0.0f, 0.0f, -planet.planet_radius);
  62
  63         Geometry::Ray<float, 3> ray(pos-planet_center, look_dir);
  64         Geometry::HyperSphere<float, 3> surface(planet.planet_radius);
  65         Geometry::SurfacePoint<float, 3> intersection;
  66         if(surface.get_intersections(ray, &intersection, 1))
  67                 return Color(0.0f);
  68
  69         Geometry::HyperSphere<float, 3> space(planet.planet_radius+planet.atmosphere_thickness);
  70         if(!space.get_intersections(ray, &intersection, 1))
  71                 return Color(1.0f);
  72
  73         float step_size = intersection.distance/50;
  74
  75         Color path_extinction(0.0f);
  76         for(unsigned i=0; i<50; ++i)
  77         {
  78                 Vector3 from_center = pos-planet_center;
  79                 float height = from_center.norm()-planet.planet_radius;
  80
  81                 Color rayleigh_scatter = planet.rayleigh_scatter*exp(height/planet.rayleigh_density_decay);
  82                 Color mie_scatter = planet.mie_scatter*exp(height/planet.mie_density_decay);
  83                 Color mie_absorb = planet.mie_absorb*exp(height/planet.mie_density_decay);
  84                 Color ozone_absorb = planet.ozone_absorb*max(1.0f-abs(height-planet.ozone_band_center)/planet.ozone_band_extent, 0.0f);
  85
  86                 path_extinction = path_extinction+(rayleigh_scatter+mie_scatter+mie_absorb+ozone_absorb)*step_size;
  87                 pos += look_dir*step_size;
  88         }
  89
  90         return Color(exp(-path_extinction.r), exp(-path_extinction.g), exp(-path_extinction.b));
  91 }
  92
  93 void Sky::setup_frame(Renderer &renderer)
  94 {
  95         if(rendered)
  96                 return;
  97
  98         rendered = true;
  99
 100         shdata.uniform("light_color", sun.get_color());
 101         shdata.uniform("light_dir", -sun.get_direction());
 102         sun.set_transmittance(get_transmittance(-sun.get_direction()));
 103
 104         {
 105                 Renderer::Push push(renderer);
 106
 107                 renderer.add_shader_data(shdata);
 108
 109                 if(transmittance_lookup_dirty)
 110                 {
 111                         transmittance_lookup_dirty = false;
 112                         renderer.set_pipeline_key(this);
 113                         renderer.set_framebuffer(&transmittance_lookup.get_framebuffer());
 114                         renderer.set_shader_program(&transmittance_shprog);
 115                         fullscreen_mesh.draw(renderer);
 116                 }
 117
 118                 renderer.set_pipeline_key(this, 1);
 119                 renderer.set_framebuffer(&distant.get_framebuffer());
 120                 renderer.set_shader_program(&distant_shprog);
 121                 renderer.set_texture("transmittance_lookup", &transmittance_lookup.get_target_texture(0), &sampler);
 122                 fullscreen_mesh.draw(renderer);
 123         }
 124
 125         content.setup_frame(renderer);
 126 }
 127
 128 void Sky::finish_frame()
 129 {
 130         if(rendered)
 131         {
 132                 rendered = false;
 133                 content.finish_frame();
 134         }
 135 }
 136
 137 void Sky::render(Renderer &renderer, Tag tag) const
 138 {
 139         content.render(renderer, tag);
 140
 141         if(!is_enabled_for_method(tag))
 142                 return;
 143
 144         Renderer::Push push(renderer);
 145
 146         renderer.set_pipeline_key(this, 2);
 147         renderer.set_shader_program(&backdrop_shprog, &shdata);
 148         renderer.set_texture("distant", &distant.get_target_texture(0), &wrap_sampler);
 149         fullscreen_mesh.draw(renderer);
 150 }
 151
 152 void Sky::set_debug_name(const string &name)
 153 {
 154 #ifdef DEBUG
 155         transmittance_lookup.set_debug_name(name+" [RT:transmittance]");
 156         distant.set_debug_name(name+" [RT:distant]");
 157         shdata.set_debug_name(name+" [UBO]");
 158 #else
 159         (void)name;
 160 #endif
 161 }
 162
 163
 164 Sky::Planet::Planet():
 165         rayleigh_scatter(0.0f),
 166         mie_scatter(0.0f),
 167         mie_absorb(0.0f),
 168         ozone_absorb(0.0f),
 169         rayleigh_density_decay(1e3f),
 170         mie_density_decay(1e3f),
 171         ozone_band_center(1e4f),
 172         ozone_band_extent(1e2f),
 173         atmosphere_thickness(2e4f),
 174         planet_radius(1e6f),
 175         ground_albedo(0.2f)
 176 { }
 177
 178 Sky::Planet Sky::Planet::earth()
 179 {
 180         Planet planet;
 181         planet.rayleigh_scatter = Color(5.802e-6f, 13.558e-6f, 33.1e-6f);
 182         planet.mie_scatter = Color(3.996e-6f, 3.996e-6f, 3.996e-6f);
 183         planet.mie_absorb = Color(4.4e-6f, 4.4e-6f, 4.4e-6f);
 184         planet.ozone_absorb = Color(0.65e-6f, 1.881e-6f, 0.085e-6f);
 185         planet.rayleigh_density_decay = -8e3f;
 186         planet.mie_density_decay = -1.2e3f;
 187         planet.ozone_band_center = 25e3f;
 188         planet.ozone_band_extent = 15e3f;
 189         planet.atmosphere_thickness = 1e5f;
 190         planet.planet_radius = 6.36e6f;
 191         return planet;
 192 }
 193
 194
 195 Sky *Sky::Template::create(const map<string, Renderable *> &renderables) const
 196 {
 197         Renderable *content = get_item(renderables, content_name);
 198         if(!content || !sun)
 199                 throw invalid_operation("Sky::Template::create");
 200         return new Sky(*content, *sun);
 201 }
 202
 203
 204 DataFile::Loader::ActionMap Sky::Template::Loader::shared_actions;
 205
 206 Sky::Template::Loader::Loader(Template &t, Collection &c):
 207         DerivedObjectLoader<Template, Effect::Template::Loader>(t, c)
 208 {
 209         set_actions(shared_actions);
 210 }
 211
 212 void Sky::Template::Loader::init_actions()
 213 {
 214         Effect::Template::Loader::init_actions();
 215         add("sun", &Template::sun);
 216 }
 217
 218 } // namespace GL
 219 } // namespace Msp