Support specialization constants in the GLSL compiler

[libs/gl.git] / source / glsl / compiler.cpp

diff --git a/source/glsl/compiler.cpp b/source/glsl/compiler.cpp
index ca6c58057022b8b3a9f69373433c0f6c54a2e35d..a18b8e8c2c3df57e0f26c2d126719edbaced8fb5 100644 (file)
--- a/source/glsl/compiler.cpp
+++ b/source/glsl/compiler.cpp
@@ -21,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()
@@ -63,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)
@@ -213,6 +221,7 @@ void Compiler::generate(Stage &stage, Mode mode)
        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);
 }