]> git.tdb.fi Git - libs/gl.git/blob - tools/glslcompiler.cpp
Refactor resolution of declarations for MemberAccess
[libs/gl.git] / tools / glslcompiler.cpp
1 #include <msp/core/application.h>
2 #include <msp/core/getopt.h>
3 #include <msp/gl/glsl/compiler.h>
4 #include <msp/gl/glsl/glsl_error.h>
5 #include <msp/io/print.h>
6 #include <msp/strings/utils.h>
7
8 class GlslCompiler: public Msp::RegisteredApplication<GlslCompiler>
9 {
10 private:
11         std::string source_fn;
12         std::map<std::string, int> spec_values;
13         bool parse_only;
14         bool combined;
15         Msp::GL::SL::Stage::Type stage;
16         bool dump_ast;
17
18 public:
19         GlslCompiler(int, char **);
20
21         virtual int main();
22 };
23
24 using namespace std;
25 using namespace Msp;
26
27 GlslCompiler::GlslCompiler(int argc, char **argv):
28         parse_only(false),
29         combined(false),
30         stage(GL::SL::Stage::SHARED),
31         dump_ast(false)
32 {
33         string stage_str;
34         vector<string> spec_values_in;
35
36         GetOpt getopt;
37         getopt.add_option('c', "combined", combined, GetOpt::NO_ARG).set_help("Output combined GLSL");
38         getopt.add_option('a', "dump-ast", dump_ast, GetOpt::NO_ARG).set_help("Dump AST for debugging");
39         getopt.add_option('p', "parse_only", parse_only, GetOpt::NO_ARG).set_help("Only parse the loaded source, don't compile");
40         getopt.add_option('e', "specialize", spec_values_in, GetOpt::REQUIRED_ARG).set_help("Set specialization constant", "NAME:VALUE");
41         getopt.add_option('s', "stage", stage_str, GetOpt::REQUIRED_ARG).set_help("Output GLSL for STAGE", "STAGE");
42         getopt.add_argument("source", source_fn, GetOpt::REQUIRED_ARG).set_help("GLSL file to compile");
43         getopt(argc, argv);
44
45         if(stage_str=="vertex")
46                 stage = GL::SL::Stage::VERTEX;
47         else if(stage_str=="geometry")
48                 stage = GL::SL::Stage::GEOMETRY;
49         else if(stage_str=="fragment")
50                 stage = GL::SL::Stage::FRAGMENT;
51         else if(!dump_ast)
52                 combined = true;
53
54         for(vector<string>::const_iterator i=spec_values_in.begin(); i!=spec_values_in.end(); ++i)
55         {
56                 unsigned colon = i->find(':');
57                 if(colon==string::npos || colon==0 || colon+1>=i->size())
58                         throw usage_error("Invalid specialization value");
59
60                 string value_str = i->substr(colon+1);
61                 int value;
62                 if(isnumrc(value_str))
63                         value = lexical_cast<int>(value_str);
64                 else
65                         value = lexical_cast<bool>(value_str);
66                 spec_values[i->substr(0, colon)] = value;
67         }
68 }
69
70 int GlslCompiler::main()
71 {
72         GL::SL::Compiler compiler(GL::SL::Features::all());
73         IO::File file(source_fn);
74         compiler.load_source(file, source_fn);
75         compiler.specialize(spec_values);
76         if(!parse_only)
77         {
78                 try
79                 {
80                         compiler.compile(GL::SL::Compiler::PROGRAM);
81                 }
82                 catch(const GL::SL::invalid_shader_source &exc)
83                 {
84                         if(!dump_ast)
85                                 throw;
86
87                         IO::print("Compilation resulted in errors:\n%s\n", exc.what());
88                         combined = false;
89                         stage = GL::SL::Stage::SHARED;
90                 }
91         }
92
93         if(dump_ast)
94         {
95                 vector<GL::SL::Stage::Type> stages = compiler.get_stages();
96                 for(vector<GL::SL::Stage::Type>::const_iterator i=stages.begin(); i!=stages.end(); ++i)
97                         IO::print("%s\n", compiler.get_stage_debug(*i));
98         }
99
100         if(combined)
101                 IO::print("%s\n", compiler.get_combined_glsl());
102         else if(stage!=GL::SL::Stage::SHARED)
103                 IO::print("%s\n", compiler.get_stage_glsl(stage));
104
105         return 0;
106 }