From: Mikko Rasa <tdb@tdb.fi>
Date: Thu, 11 Jun 2020 13:23:19 +0000 (+0300)
Subject: Give materials the capability to automatically create a suitable shader
X-Git-Url: http://git.tdb.fi/?a=commitdiff_plain;h=fd62e55d37716787fe909883a1b18e5b8128ec80;p=libs%2Fgl.git

Give materials the capability to automatically create a suitable shader
---

diff --git a/source/basicmaterial.cpp b/source/basicmaterial.cpp
index 7ba8380c..858c65da 100644
--- a/source/basicmaterial.cpp
+++ b/source/basicmaterial.cpp
@@ -14,6 +14,36 @@ BasicMaterial::BasicMaterial()
 	set_reflectivity(0.0f);
 }
 
+string BasicMaterial::create_program_source() const
+{
+	string source = "import phong;\n";
+	if(diffuse.texture)
+		source += "const bool use_diffuse_map = true;\n";
+	if(specular.texture || specular.value.r || specular.value.g || specular.value.b)
+	{
+		source += "const bool use_specular = true;\n";
+		if(specular.texture)
+			source += "const bool use_specular_map = true;\n";
+		if(shininess.texture)
+			source += "const bool use_shininess_map = true;\n";
+	}
+	if(normal.texture)
+		source += "const bool use_normal_map = true;\n";
+	if(emission.texture || emission.value.r || emission.value.g || emission.value.b)
+	{
+		source += "const bool use_emission = true;\n";
+		if(emission.texture)
+			source += "const bool use_emission_map = true;\n";
+	}
+	if(reflectivity.value || reflectivity.texture)
+	{
+		source += "const bool use_reflectivity = true;\n";
+		if (reflectivity.texture)
+			source += "const bool use_reflectivity_map = true;\n";
+	}
+	return source;
+}
+
 void BasicMaterial::attach_textures_to(Texturing &texturing, ProgramData &tex_shdata) const
 {
 	attach_texture_to(diffuse.texture, texturing, tex_shdata, "diffuse_map");
diff --git a/source/basicmaterial.h b/source/basicmaterial.h
index 9b7eab07..5fca56a0 100644
--- a/source/basicmaterial.h
+++ b/source/basicmaterial.h
@@ -33,6 +33,10 @@ private:
 public:
 	BasicMaterial();
 
+protected:
+	virtual std::string create_program_source() const;
+
+public:
 	void attach_textures_to(Texturing &, ProgramData &) const;
 
 	void set_diffuse(const Color &);
diff --git a/source/material.cpp b/source/material.cpp
index b0f62bbc..ef3662e9 100644
--- a/source/material.cpp
+++ b/source/material.cpp
@@ -1,3 +1,5 @@
+#include <msp/core/hash.h>
+#include <msp/strings/format.h>
 #include "basicmaterial.h"
 #include "gl.h"
 #include "resources.h"
@@ -9,6 +11,33 @@ using namespace std;
 namespace Msp {
 namespace GL {
 
+Program *Material::create_compatible_shader() const
+{
+	return new Program(create_program_source());
+}
+
+const Program *Material::create_compatible_shader(DataFile::Collection &coll) const
+{
+	string source = create_program_source();
+	string name = format("_material_%016x.glsl", hash64(source));
+	Program *shprog = coll.find<Program>(name);
+	if(shprog)
+		return shprog;
+
+	shprog = new Program(create_program_source());
+	try
+	{
+		coll.add(name, shprog);
+	}
+	catch(...)
+	{
+		delete shprog;
+		throw;
+	}
+
+	return shprog;
+}
+
 void Material::attach_texture_to(const Texture *tex, Texturing &texturing, ProgramData &tex_shdata, const string &name) const
 {
 	if(!tex)
diff --git a/source/material.h b/source/material.h
index df55a567..3e9765d2 100644
--- a/source/material.h
+++ b/source/material.h
@@ -82,6 +82,11 @@ protected:
 public:
 	virtual ~Material() { }
 
+	virtual Program *create_compatible_shader() const;
+	virtual const Program *create_compatible_shader(DataFile::Collection &) const;
+protected:
+	virtual std::string create_program_source() const = 0;
+
 public:
 	/** Returns the uniforms for the material. */
 	const ProgramData &get_shader_data() const { return shdata; }
diff --git a/source/renderpass.cpp b/source/renderpass.cpp
index 7c7cf2c4..b6eb1d95 100644
--- a/source/renderpass.cpp
+++ b/source/renderpass.cpp
@@ -17,6 +17,7 @@ namespace GL {
 
 RenderPass::RenderPass():
 	shprog(0),
+	shprog_from_material(false),
 	shdata(0),
 	material(0),
 	texturing(0),
@@ -25,6 +26,7 @@ RenderPass::RenderPass():
 
 RenderPass::RenderPass(const RenderPass &other):
 	shprog(other.shprog),
+	shprog_from_material(other.shprog_from_material),
 	shdata(other.shdata),
 	uniform_slots(other.uniform_slots),
 	material(other.material),
@@ -37,6 +39,7 @@ RenderPass::RenderPass(const RenderPass &other):
 RenderPass &RenderPass::operator=(const RenderPass &other)
 {
 	shprog = other.shprog;
+	shprog_from_material = other.shprog_from_material;
 	shdata = other.shdata;
 	uniform_slots = other.uniform_slots;
 	material = other.material;
@@ -52,8 +55,9 @@ RenderPass::~RenderPass()
 	delete texturing;
 }
 
-void RenderPass::finalize_material()
+void RenderPass::finalize_material(DataFile::Collection *coll)
 {
+	maybe_create_material_shader(coll);
 	ensure_private_shader_data();
 
 	if(!texturing)
@@ -61,21 +65,44 @@ void RenderPass::finalize_material()
 	material->attach_textures_to(*texturing, *shdata);
 }
 
+void RenderPass::maybe_create_material_shader(DataFile::Collection *coll)
+{
+	if(shprog && !shprog_from_material)
+		return;
+
+	if(coll)
+	{
+		shprog = material->create_compatible_shader(*coll);
+		shprog.keep();
+	}
+	else
+		shprog = material->create_compatible_shader();
+
+	if(shdata)
+		shdata = new ProgramData(*shdata, shprog.get());
+
+	shprog_from_material = true;
+}
+
 void RenderPass::ensure_private_shader_data()
 {
 	if(!shprog)
 		throw invalid_operation("RenderPass::ensure_private_shader_data");
 
 	if(!shdata)
-		shdata = new ProgramData(shprog);
-	else if(shdata.refcount() > 1)
+		shdata = new ProgramData(shprog.get());
+	else if(shdata.refcount()>1)
 		shdata = new ProgramData(*shdata);
 }
 
 void RenderPass::set_shader_program(const Program *prog, const ProgramData *data)
 {
 	shprog = prog;
+	shprog.keep();
+	shprog_from_material = false;
 	shdata = (data ? new ProgramData(*data) : 0);
+	if(material)
+		finalize_material(0);
 }
 
 const string &RenderPass::get_slotted_uniform_name(const string &slot) const
@@ -93,7 +120,7 @@ void RenderPass::set_material(const Material *mat)
 {
 	material = mat;
 	material.keep();
-	finalize_material();
+	finalize_material(0);
 }
 
 void RenderPass::set_texture(unsigned index, const Texture *tex)
@@ -116,7 +143,7 @@ void RenderPass::apply(Renderer &renderer) const
 {
 	renderer.set_texturing(texturing);
 	renderer.set_material(material.get());
-	renderer.set_shader_program(shprog, shdata.get());
+	renderer.set_shader_program(shprog.get(), shdata.get());
 	renderer.set_reverse_winding(back_faces);
 }
 
@@ -152,23 +179,25 @@ void RenderPass::Loader::material_inline()
 	Material::GenericLoader ldr(coll);
 	load_sub_with(ldr);
 	obj.material = ldr.get_material();
-	obj.finalize_material();
+	obj.finalize_material(coll);
 }
 
 void RenderPass::Loader::material(const string &name)
 {
 	obj.material = &get_collection().get<Material>(name);
 	obj.material.keep();
-	obj.finalize_material();
+	obj.finalize_material(coll);
 }
 
 void RenderPass::Loader::shader(const string &n)
 {
 	obj.shprog = &get_collection().get<Program>(n);
+	obj.shprog.keep();
+	obj.shprog_from_material = false;
 	if(obj.shdata)
-		obj.shdata = new ProgramData(*obj.shdata, obj.shprog);
+		obj.shdata = new ProgramData(*obj.shdata, obj.shprog.get());
 	if(obj.material)
-		obj.finalize_material();
+		obj.finalize_material(coll);
 }
 
 void RenderPass::Loader::texunit(unsigned i)
diff --git a/source/renderpass.h b/source/renderpass.h
index 26166914..428c74ce 100644
--- a/source/renderpass.h
+++ b/source/renderpass.h
@@ -59,7 +59,8 @@ private:
 		void texture2d();
 	};
 
-	const Program *shprog;
+	RefPtr<const Program> shprog;
+	bool shprog_from_material;
 	RefPtr<ProgramData> shdata;
 	std::map<std::string, std::string> uniform_slots;
 	RefPtr<const Material> material;
@@ -75,12 +76,13 @@ public:
 	~RenderPass();
 
 private:
-	void finalize_material();
+	void finalize_material(DataFile::Collection *);
+	void maybe_create_material_shader(DataFile::Collection *);
 	void ensure_private_shader_data();
 
 public:
 	void set_shader_program(const Program *, const ProgramData *);
-	const Program *get_shader_program() const { return shprog; }
+	const Program *get_shader_program() const { return shprog.get(); }
 	const ProgramData *get_shader_data() const { return shdata.get(); }
 	const std::string &get_slotted_uniform_name(const std::string &) const;
 	void set_material(const Material *);