X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;ds=sidebyside;f=source%2Fglsl%2Fcompiler.cpp;h=5bec9c8b1b8f475163b50a35b57b0a92e5bec84b;hb=3f899b1fc2e04f4fe74c99ad3e8ebb900c257214;hp=4a7a19e29759770966db8ae105c0967bdd5c4f9b;hpb=c0be4e4ae1a8b6ac31ff6b7080e2242c13d947ff;p=libs%2Fgl.git

diff --git a/source/glsl/compiler.cpp b/source/glsl/compiler.cpp
index 4a7a19e2..5bec9c8b 100644
--- a/source/glsl/compiler.cpp
+++ b/source/glsl/compiler.cpp
@@ -3,6 +3,7 @@
 #include <msp/strings/format.h>
 #include "compatibility.h"
 #include "compiler.h"
+#include "debug.h"
 #include "error.h"
 #include "generate.h"
 #include "optimize.h"
@@ -20,12 +21,14 @@ namespace SL {
 
 Compiler::Compiler():
 	features(Features::from_context()),
-	module(0)
+	module(0),
+	specialized(false)
 { }
 
 Compiler::Compiler(const Features &f):
 	features(f),
-	module(0)
+	module(0),
+	specialized(false)
 { }
 
 Compiler::~Compiler()
@@ -62,6 +65,12 @@ void Compiler::load_source(IO::Base &io, const string &src_name)
 	load_source(io, 0, src_name);
 }
 
+void Compiler::specialize(const map<string, int> &sv)
+{
+	specialized = true;
+	spec_values = sv;
+}
+
 void Compiler::compile(Mode mode)
 {
 	for(list<Stage>::iterator i=module->stages.begin(); i!=module->stages.end(); ++i)
@@ -132,6 +141,14 @@ const SourceMap &Compiler::get_source_map() const
 	return module->source_map;
 }
 
+string Compiler::get_stage_debug(Stage::Type stage_type) const
+{
+	for(list<Stage>::iterator i=module->stages.begin(); i!=module->stages.end(); ++i)
+		if(i->type==stage_type)
+			return DumpTree().apply(*i);
+	throw key_error(Stage::get_stage_name(stage_type));
+}
+
 void Compiler::append_module(Module &mod, DataFile::Collection *res)
 {
 	module->source_map.merge_from(mod.source_map);
@@ -197,13 +214,20 @@ void Compiler::generate(Stage &stage, Mode mode)
 	inject_block(stage.content, module->shared.content);
 
 	DeclarationReorderer().apply(stage);
-	BlockResolver().apply(stage);
+
+	// Initial resolving pass
+	BlockHierarchyResolver().apply(stage);
 	FunctionResolver().apply(stage);
 	VariableResolver().apply(stage);
+
+	/* All variables local to a stage have been resolved.  Resolve non-local
+	variables through interfaces. */
 	InterfaceGenerator().apply(stage);
 	VariableResolver().apply(stage);
+
 	DeclarationReorderer().apply(stage);
 	FunctionResolver().apply(stage);
+	ConstantSpecializer().apply(stage, (mode==PROGRAM && specialized ? &spec_values : 0));
 	if(mode==PROGRAM)
 		LegacyConverter().apply(stage, features);
 }
@@ -213,9 +237,11 @@ bool Compiler::optimize(Stage &stage)
 	ConstantConditionEliminator().apply(stage);
 
 	FunctionInliner().apply(stage);
-	BlockResolver().apply(stage);
+	BlockHierarchyResolver().apply(stage);
 	VariableResolver().apply(stage);
 
+	/* Removing variables or functions may cause things from the previous stage
+	to become unused. */
 	bool result = UnusedVariableRemover().apply(stage);
 	result |= UnusedFunctionRemover().apply(stage);