1 #include "framebuffer.h"
3 #include "pbrmaterial.h"
13 const Tag PbrMaterial::texture_tags[] =
15 Tag("base_color_map"),
21 Tag("fresnel_lookup"),
25 PbrMaterial::PbrMaterial():
26 fresnel_lookup(get_or_create_fresnel_lookup()),
27 fresnel_sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp"))
35 const Texture2D &PbrMaterial::get_or_create_fresnel_lookup()
37 Resources &resources = Resources::get_global();
39 static const string name = "_pbr_fresnel_lookup.tex2d";
40 Texture2D *fresnel_lookup = resources.find<Texture2D>(name);
42 return *fresnel_lookup;
44 fresnel_lookup = new Texture2D;
45 fresnel_lookup->storage(RG8, 128, 128, 1);
46 resources.add(name, fresnel_lookup);
48 const Program &shprog = resources.get<Program>("_pbr_fresnel_lookup.glsl.shader");
50 shdata.uniform("n_samples", 1024);
51 // Not actually used here, but put it in to satisfy the interface
52 shdata.uniform("roughness", 0.0f);
54 const Mesh &mesh = resources.get<Mesh>("_fullscreen_quad.mesh");
55 Framebuffer fresnel_lookup_fbo;
56 fresnel_lookup_fbo.attach(COLOR_ATTACHMENT0, *fresnel_lookup);
57 Bind bind_fbo(fresnel_lookup_fbo);
59 renderer.set_shader_program(&shprog, &shdata);
62 return *fresnel_lookup;
65 void PbrMaterial::fill_program_info(string &module_name, map<string, int> &spec_values) const
67 module_name = "cooktorrance.glsl";
68 spec_values["use_base_color_map"] = (base_color.texture!=0);
69 spec_values["use_normal_map"] = (normal.texture!=0);
70 spec_values["use_metalness_map"] = (metalness.texture!=0);
71 spec_values["use_roughness_map"] = (roughness.texture!=0);
72 spec_values["use_occlusion_map"] = (occlusion.texture!=0);
73 bool use_emission = (emission.texture || emission.value.r || emission.value.g || emission.value.b);
74 spec_values["use_emission"] = use_emission;
75 spec_values["use_emission_map"] = (emission.texture!=0);
78 const Texture *PbrMaterial::get_texture(Tag tag) const
80 if(tag==texture_tags[0])
81 return base_color.texture;
82 else if(tag==texture_tags[1])
83 return normal.texture;
84 else if(tag==texture_tags[2])
85 return metalness.texture;
86 else if(tag==texture_tags[3])
87 return roughness.texture;
88 else if(tag==texture_tags[4])
89 return occlusion.texture;
90 else if(tag==texture_tags[5])
91 return emission.texture;
92 else if(tag==texture_tags[6])
93 return &fresnel_lookup;
98 const Sampler *PbrMaterial::get_sampler(Tag tag) const
100 if(tag==texture_tags[6])
101 return &fresnel_sampler;
106 void PbrMaterial::set_base_color(const Color &color)
108 base_color.value = color;
109 shdata.uniform("pbr_material.base_color", color);
112 void PbrMaterial::set_base_color_map(const Texture *tex)
114 base_color.texture = tex;
117 void PbrMaterial::set_normal_map(const Texture *tex)
119 normal.texture = tex;
122 void PbrMaterial::set_metalness(float value)
124 metalness.value = value;
125 shdata.uniform("pbr_material.metalness", value);
128 void PbrMaterial::set_metalness_map(const Texture *tex)
130 metalness.texture = tex;
133 void PbrMaterial::set_roughness(float value)
135 roughness.value = value;
136 shdata.uniform("pbr_material.roughness", value);
139 void PbrMaterial::set_roughness_map(const Texture *tex)
141 roughness.texture = tex;
144 void PbrMaterial::set_occlusion_map(const Texture *tex)
146 occlusion.texture = tex;
149 void PbrMaterial::set_emission(const Color &color)
151 emission.value = color;
152 shdata.uniform("pbr_material.emission", color);
155 void PbrMaterial::set_emission_map(const Texture *tex)
157 emission.texture = tex;
161 DataFile::Loader::ActionMap PbrMaterial::Loader::shared_actions;
163 PbrMaterial::Loader::Loader(PbrMaterial &m):
164 DerivedObjectLoader<PbrMaterial, Material::PropertyLoader<PbrMaterial> >(m)
166 set_actions(shared_actions);
169 PbrMaterial::Loader::Loader(PbrMaterial &m, Collection &c):
170 DerivedObjectLoader<PbrMaterial, Material::PropertyLoader<PbrMaterial> >(m, c)
172 set_actions(shared_actions);
175 void PbrMaterial::Loader::init_actions()
177 Material::PropertyLoader<PbrMaterial>::init_actions();
178 add_property("base_color", &PbrMaterial::set_base_color, &PbrMaterial::set_base_color_map, true);
179 add_property("normal", &PbrMaterial::set_normal_map);
180 add_property("metalness", &PbrMaterial::set_metalness, &PbrMaterial::set_metalness_map);
181 add_property("roughness", &PbrMaterial::set_roughness, &PbrMaterial::set_roughness_map);
182 add_property("occlusion", &PbrMaterial::set_occlusion_map);
183 add_property("emission", &PbrMaterial::set_emission, &PbrMaterial::set_emission_map, false);