From: Mikko Rasa <tdb@tdb.fi>
Date: Fri, 9 Apr 2021 15:22:00 +0000 (+0300)
Subject: Add an option to compile SPIR-V in the command-line GLSL compiler
X-Git-Url: http://git.tdb.fi/?a=commitdiff_plain;h=cbe2ec1f080b117faf9bf0ad2d92d5d4f2379584;p=libs%2Fgl.git

Add an option to compile SPIR-V in the command-line GLSL compiler
---

diff --git a/tools/glslcompiler.cpp b/tools/glslcompiler.cpp
index a1eb1a1f..4fe6bc8b 100644
--- a/tools/glslcompiler.cpp
+++ b/tools/glslcompiler.cpp
@@ -37,7 +37,8 @@ GlslCompiler::GlslCompiler(int argc, char **argv):
 {
 	string stage_str;
 	vector<string> spec_values_in;
-	bool as_module = false;
+	unsigned as_module = 0;
+	string module_type = "glsl";
 	unsigned target_version = 0;
 
 	GetOpt getopt;
@@ -46,7 +47,7 @@ GlslCompiler::GlslCompiler(int argc, char **argv):
 	getopt.add_option('p', "parse_only", parse_only, GetOpt::NO_ARG).set_help("Only parse the loaded source (implies -a)");
 	getopt.add_option('e', "specialize", spec_values_in, GetOpt::REQUIRED_ARG).set_help("Set specialization constant", "NAME:VALUE");
 	getopt.add_option('s', "stage", stage_str, GetOpt::REQUIRED_ARG).set_help("Output GLSL for STAGE", "STAGE");
-	getopt.add_option('m', "module", as_module, GetOpt::NO_ARG).set_help("Compile as unspecialized module");
+	getopt.add_option('m', "module", module_type, GetOpt::OPTIONAL_ARG).bind_seen_count(as_module).set_help("Compile as unspecialized module");
 	getopt.add_option('t', "target-version", target_version, GetOpt::REQUIRED_ARG).set_help("Specify target GLSL version", "VER");
 	getopt.add_option('o', "out-file", out_filename, GetOpt::REQUIRED_ARG).set_help("Write output to file instead of stdout", "FILE");
 	getopt.add_argument("source", source_fn, GetOpt::REQUIRED_ARG).set_help("GLSL file to compile");
@@ -56,7 +57,17 @@ GlslCompiler::GlslCompiler(int argc, char **argv):
 		features = GL::SL::Features::from_version(GL::Version(target_version/100, target_version%100));
 
 	if(as_module)
-		compile_mode = GL::SL::Compiler::MODULE;
+	{
+		if(module_type=="glsl" || module_type=="GLSL")
+			compile_mode = GL::SL::Compiler::MODULE;
+		else if(module_type=="spirv" || module_type=="spir-v" || module_type=="SPIRV" || module_type=="SPIR-V")
+			compile_mode = GL::SL::Compiler::SPIRV;
+		else
+			throw usage_error("Invalid module type");
+	}
+
+	if(compile_mode==GL::SL::Compiler::SPIRV && out_filename.empty())
+		throw usage_error("-o is required for SPIR-V");
 
 	if(parse_only)
 	{
@@ -65,7 +76,9 @@ GlslCompiler::GlslCompiler(int argc, char **argv):
 		dump_ast = true;
 	}
 
-	if(stage_str=="vertex")
+	if(!stage_str.empty() && compile_mode==GL::SL::Compiler::SPIRV)
+		throw usage_error("-s can't be used with SPIR-V");
+	else if(stage_str=="vertex")
 		stage = GL::SL::Stage::VERTEX;
 	else if(stage_str=="geometry")
 		stage = GL::SL::Stage::GEOMETRY;
@@ -134,7 +147,12 @@ int GlslCompiler::main()
 		out = out_file.get();
 	}
 
-	if(combined)
+	if(compile_mode==GL::SL::Compiler::SPIRV)
+	{
+		vector<UInt32> code = compiler.get_combined_spirv();
+		out->write(reinterpret_cast<char *>(&code.front()), code.size()*4);
+	}
+	else if(combined)
 		IO::print(*out, "%s\n", compiler.get_combined_glsl());
 	else if(stage!=GL::SL::Stage::SHARED)
 		IO::print(*out, "%s\n", compiler.get_stage_glsl(stage));