Check the flat qualifier from the correct member

[libs/gl.git] / source / materials / pbrmaterial.cpp

diff --git a/source/materials/pbrmaterial.cpp b/source/materials/pbrmaterial.cpp
index 16c287b4e40f6ad97987ac517ba57315646dc711..4a5f0a419d370ef1bc935bdf15fc3ff0cd0929d5 100644 (file)
--- a/source/materials/pbrmaterial.cpp
+++ b/source/materials/pbrmaterial.cpp
@@ -1,22 +1,73 @@
+#include "framebuffer.h"
+#include "mesh.h"
 #include "pbrmaterial.h"
+#include "renderer.h"
+#include "resources.h"
+#include "texture2d.h"
 
 using namespace std;
 
 namespace Msp {
 namespace GL {
 
+const Tag PbrMaterial::texture_tags[] =
+{
+       Tag("base_color_map"),
+       Tag("normal_map"),
+       Tag("metalness_map"),
+       Tag("roughness_map"),
+       Tag("occlusion_map"),
+       Tag("emission_map"),
+       Tag("fresnel_lookup"),
+       Tag()
+};
+
 PbrMaterial::PbrMaterial():
-       receive_shadows(false)
+       fresnel_lookup(get_or_create_fresnel_lookup()),
+       fresnel_sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp"))
 {
        set_base_color(0.8f);
+       set_tint(1.0f);
        set_metalness(0.0f);
        set_roughness(0.5f);
        set_emission(0.0f);
 }
 
+const Texture2D &PbrMaterial::get_or_create_fresnel_lookup()
+{
+       Resources &resources = Resources::get_global();
+
+       static const string name = "_pbr_fresnel_lookup.tex";
+       Texture2D *fresnel_lookup = resources.find<Texture2D>(name);
+       if(fresnel_lookup)
+               return *fresnel_lookup;
+
+       fresnel_lookup = new Texture2D;
+       fresnel_lookup->storage(RG8, 128, 128, 1);
+       resources.add(name, fresnel_lookup);
+
+       const Program &shprog = resources.get<Program>("_pbr_fresnel_lookup.glsl.shader");
+       ProgramData shdata;
+       shdata.uniform("n_samples", 1024);
+       // Not actually used here, but put it in to satisfy the interface
+       shdata.uniform("roughness", 0.0f);
+
+       const Mesh &mesh = resources.get<Mesh>("_fullscreen_quad.mesh");
+       Framebuffer fresnel_lookup_fbo((COLOR_ATTACHMENT,RG8));
+       fresnel_lookup_fbo.attach(COLOR_ATTACHMENT, *fresnel_lookup);
+       Renderer renderer;
+       renderer.begin();
+       renderer.set_framebuffer(&fresnel_lookup_fbo);
+       renderer.set_shader_program(&shprog, &shdata);
+       mesh.draw(renderer);
+       renderer.end();
+
+       return *fresnel_lookup;
+}
+
 void PbrMaterial::fill_program_info(string &module_name, map<string, int> &spec_values) const
 {
-       module_name = "cooktorrance.glsl";
+       module_name = "pbr_material.glsl";
        spec_values["use_base_color_map"] = (base_color.texture!=0);
        spec_values["use_normal_map"] = (normal.texture!=0);
        spec_values["use_metalness_map"] = (metalness.texture!=0);
@@ -25,17 +76,34 @@ void PbrMaterial::fill_program_info(string &module_name, map<string, int> &spec_
        bool use_emission = (emission.texture || emission.value.r || emission.value.g || emission.value.b);
        spec_values["use_emission"] = use_emission;
        spec_values["use_emission_map"] = (emission.texture!=0);
-       spec_values["use_shadow_map"] = receive_shadows;
 }
 
-void PbrMaterial::attach_textures_to(Texturing &texturing, ProgramData &tex_shdata) const
+const Texture *PbrMaterial::get_texture(Tag tag) const
 {
-       attach_texture_to(base_color.texture, texturing, tex_shdata, "base_color_map");
-       attach_texture_to(metalness.texture, texturing, tex_shdata, "metalness_map");
-       attach_texture_to(roughness.texture, texturing, tex_shdata, "roughness_map");
-       attach_texture_to(normal.texture, texturing, tex_shdata, "normal_map");
-       attach_texture_to(occlusion.texture, texturing, tex_shdata, "occlusion_map");
-       attach_texture_to(emission.texture, texturing, tex_shdata, "emission_map");
+       if(tag==texture_tags[0])
+               return base_color.texture;
+       else if(tag==texture_tags[1])
+               return normal.texture;
+       else if(tag==texture_tags[2])
+               return metalness.texture;
+       else if(tag==texture_tags[3])
+               return roughness.texture;
+       else if(tag==texture_tags[4])
+               return occlusion.texture;
+       else if(tag==texture_tags[5])
+               return emission.texture;
+       else if(tag==texture_tags[6])
+               return &fresnel_lookup;
+       else
+               return 0;
+}
+
+const Sampler *PbrMaterial::get_sampler(Tag tag) const
+{
+       if(tag==texture_tags[6])
+               return &fresnel_sampler;
+       else
+               return sampler;
 }
 
 void PbrMaterial::set_base_color(const Color &color)
@@ -49,6 +117,12 @@ void PbrMaterial::set_base_color_map(const Texture *tex)
        base_color.texture = tex;
 }
 
+void PbrMaterial::set_tint(const Color &color)
+{
+       tint.value = color;
+       shdata.uniform("pbr_material.tint", color);
+}
+
 void PbrMaterial::set_normal_map(const Texture *tex)
 {
        normal.texture = tex;
@@ -92,20 +166,9 @@ void PbrMaterial::set_emission_map(const Texture *tex)
        emission.texture = tex;
 }
 
-void PbrMaterial::set_receive_shadows(bool s)
-{
-       receive_shadows = s;
-}
-
 
 DataFile::Loader::ActionMap PbrMaterial::Loader::shared_actions;
 
-PbrMaterial::Loader::Loader(PbrMaterial &m):
-       DerivedObjectLoader<PbrMaterial, Material::PropertyLoader<PbrMaterial> >(m)
-{
-       set_actions(shared_actions);
-}
-
 PbrMaterial::Loader::Loader(PbrMaterial &m, Collection &c):
        DerivedObjectLoader<PbrMaterial, Material::PropertyLoader<PbrMaterial> >(m, c)
 {
@@ -121,7 +184,7 @@ void PbrMaterial::Loader::init_actions()
        add_property("roughness", &PbrMaterial::set_roughness, &PbrMaterial::set_roughness_map);
        add_property("occlusion", &PbrMaterial::set_occlusion_map);
        add_property("emission", &PbrMaterial::set_emission, &PbrMaterial::set_emission_map, false);
-       add("receive_shadows", &PbrMaterial::receive_shadows);
+       add_property("tint", &PbrMaterial::set_tint, 0, true);
 }
 
 } // namespace GL