source/pbrmaterial.cpp

   1 #include "pbrmaterial.h"
   2
   3 using namespace std;
   4
   5 namespace Msp {
   6 namespace GL {
   7
   8 PbrMaterial::PbrMaterial():
   9         receive_shadows(false)
  10 {
  11         set_base_color(0.8f);
  12         set_metalness(0.0f);
  13         set_roughness(0.5f);
  14         set_emission(0.0f);
  15 }
  16
  17 string PbrMaterial::create_program_source() const
  18 {
  19         string source = "import cooktorrance;\n";
  20         if(base_color.texture)
  21                 source += "const bool use_base_color_map = true;\n";
  22         if(normal.texture)
  23                 source += "const bool use_normal_map = true;\n";
  24         if(metalness.texture)
  25                 source += "const bool use_metalness_map = true;\n";
  26         if(roughness.texture)
  27                 source += "const bool use_roughness_map = true;\n";
  28         if(occlusion.texture)
  29                 source += "const bool use_occlusion_map = true;\n";
  30         if(emission.texture || emission.value.r || emission.value.g || emission.value.b)
  31         {
  32                 source += "const bool use_emission = true;\n";
  33                 if(emission.texture)
  34                         source += "const bool use_emission_map = true;\n";
  35         }
  36         if(receive_shadows)
  37                 source += "const bool use_shadow_map = true;\n";
  38         return source;
  39 }
  40
  41 void PbrMaterial::attach_textures_to(Texturing &texturing, ProgramData &tex_shdata) const
  42 {
  43         attach_texture_to(base_color.texture, texturing, tex_shdata, "base_color_map");
  44         attach_texture_to(metalness.texture, texturing, tex_shdata, "metalness_map");
  45         attach_texture_to(roughness.texture, texturing, tex_shdata, "roughness_map");
  46         attach_texture_to(normal.texture, texturing, tex_shdata, "normal_map");
  47         attach_texture_to(occlusion.texture, texturing, tex_shdata, "occlusion_map");
  48         attach_texture_to(emission.texture, texturing, tex_shdata, "emission_map");
  49 }
  50
  51 void PbrMaterial::set_base_color(const Color &color)
  52 {
  53         base_color.value = color;
  54         shdata.uniform("pbr_material.base_color", color);
  55 }
  56
  57 void PbrMaterial::set_base_color_map(const Texture *tex)
  58 {
  59         base_color.texture = tex;
  60 }
  61
  62 void PbrMaterial::set_normal_map(const Texture *tex)
  63 {
  64         normal.texture = tex;
  65 }
  66
  67 void PbrMaterial::set_metalness(float value)
  68 {
  69         metalness.value = value;
  70         shdata.uniform("pbr_material.metalness", value);
  71 }
  72
  73 void PbrMaterial::set_metalness_map(const Texture *tex)
  74 {
  75         metalness.texture = tex;
  76 }
  77
  78 void PbrMaterial::set_roughness(float value)
  79 {
  80         roughness.value = value;
  81         shdata.uniform("pbr_material.roughness", value);
  82 }
  83
  84 void PbrMaterial::set_roughness_map(const Texture *tex)
  85 {
  86         roughness.texture = tex;
  87 }
  88
  89 void PbrMaterial::set_occlusion_map(const Texture *tex)
  90 {
  91         occlusion.texture = tex;
  92 }
  93
  94 void PbrMaterial::set_emission(const Color &color)
  95 {
  96         emission.value = color;
  97         shdata.uniform("pbr_material.emission", color);
  98 }
  99
 100 void PbrMaterial::set_emission_map(const Texture *tex)
 101 {
 102         emission.texture = tex;
 103 }
 104
 105 void PbrMaterial::set_receive_shadows(bool s)
 106 {
 107         receive_shadows = s;
 108 }
 109
 110
 111 DataFile::Loader::ActionMap PbrMaterial::Loader::shared_actions;
 112
 113 PbrMaterial::Loader::Loader(PbrMaterial &m):
 114         DerivedObjectLoader<PbrMaterial, Material::LoaderBase<PbrMaterial> >(m)
 115 {
 116         set_actions(shared_actions);
 117 }
 118
 119 PbrMaterial::Loader::Loader(PbrMaterial &m, Collection &c):
 120         DerivedObjectLoader<PbrMaterial, Material::LoaderBase<PbrMaterial> >(m, c)
 121 {
 122         set_actions(shared_actions);
 123 }
 124
 125 void PbrMaterial::Loader::init_actions()
 126 {
 127         add_property("base_color", &PbrMaterial::set_base_color, &PbrMaterial::set_base_color_map, true);
 128         add_property("normal", &PbrMaterial::set_normal_map);
 129         add_property("metalness", &PbrMaterial::set_metalness, &PbrMaterial::set_metalness_map);
 130         add_property("roughness", &PbrMaterial::set_roughness, &PbrMaterial::set_roughness_map);
 131         add_property("occlusion", &PbrMaterial::set_occlusion_map);
 132         add_property("emission", &PbrMaterial::set_emission, &PbrMaterial::set_emission_map, false);
 133         add("receive_shadows", &PbrMaterial::receive_shadows);
 134 }
 135
 136 } // namespace GL
 137 } // namespace Msp