blob: 280ca7b0d8ffaf6cf6b4a993a9a97ca41c87c62e [file] [log] [blame]
John Kessenicha0af4732012-12-12 21:15:54 +00001//
John Kessenich927608b2017-01-06 12:34:14 -07002// Copyright (C) 2002-2005 3Dlabs Inc. Ltd.
3// Copyright (C) 2013-2016 LunarG, Inc.
John Kessenichbd0747d2013-02-17 06:01:50 +00004//
John Kessenich927608b2017-01-06 12:34:14 -07005// All rights reserved.
John Kessenicha0af4732012-12-12 21:15:54 +00006//
John Kessenich927608b2017-01-06 12:34:14 -07007// Redistribution and use in source and binary forms, with or without
8// modification, are permitted provided that the following conditions
9// are met:
John Kessenicha0af4732012-12-12 21:15:54 +000010//
11// Redistributions of source code must retain the above copyright
12// notice, this list of conditions and the following disclaimer.
13//
14// Redistributions in binary form must reproduce the above
15// copyright notice, this list of conditions and the following
16// disclaimer in the documentation and/or other materials provided
17// with the distribution.
18//
19// Neither the name of 3Dlabs Inc. Ltd. nor the names of its
20// contributors may be used to endorse or promote products derived
21// from this software without specific prior written permission.
22//
John Kessenich927608b2017-01-06 12:34:14 -070023// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
26// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
27// COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
28// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
29// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
30// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
31// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
33// ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34// POSSIBILITY OF SUCH DAMAGE.
John Kessenicha0af4732012-12-12 21:15:54 +000035//
John Kessenich05a70632013-09-17 19:26:08 +000036
37// this only applies to the standalone wrapper, not the front end in general
Aras Pranckevicius8e204b22017-05-10 16:52:50 +030038#ifndef _CRT_SECURE_NO_WARNINGS
John Kessenich05a70632013-09-17 19:26:08 +000039#define _CRT_SECURE_NO_WARNINGS
Aras Pranckevicius8e204b22017-05-10 16:52:50 +030040#endif
John Kessenich05a70632013-09-17 19:26:08 +000041
Lei Zhang8a9b1ee2016-05-19 13:31:43 -040042#include "ResourceLimits.h"
John Kessenich2b07c7e2013-07-31 18:44:13 +000043#include "Worklist.h"
John Kessenich3494b4d2017-05-22 15:00:42 -060044#include "DirStackFileIncluder.h"
John Kessenicha0af4732012-12-12 21:15:54 +000045#include "./../glslang/Include/ShHandle.h"
John Kessenich0da9eaa2015-08-01 17:10:02 -060046#include "./../glslang/Include/revision.h"
John Kessenicha0af4732012-12-12 21:15:54 +000047#include "./../glslang/Public/ShaderLang.h"
John Kessenich0df0cde2015-03-03 17:09:43 +000048#include "../SPIRV/GlslangToSpv.h"
John Kessenich5e4b1242015-08-06 22:53:06 -060049#include "../SPIRV/GLSL.std.450.h"
John Kessenichacba7722015-03-04 03:48:38 +000050#include "../SPIRV/doc.h"
51#include "../SPIRV/disassemble.h"
John Kessenich3494b4d2017-05-22 15:00:42 -060052
John Kessenich66ec80e2016-08-05 14:04:23 -060053#include <cstring>
54#include <cstdlib>
steve-lunarg7f7c2ed2016-09-07 15:20:19 -060055#include <cctype>
John Kessenich66ec80e2016-08-05 14:04:23 -060056#include <cmath>
steve-lunarg7f7c2ed2016-09-07 15:20:19 -060057#include <array>
LoopDawg08a14422017-10-17 19:27:14 -060058#include <map>
Juan Lopeza558b262017-04-02 23:04:00 +020059#include <memory>
60#include <thread>
John Kessenicha0af4732012-12-12 21:15:54 +000061
baldurk876a0e32015-11-16 18:03:28 +010062#include "../glslang/OSDependent/osinclude.h"
John Kessenicha0af4732012-12-12 21:15:54 +000063
64extern "C" {
65 SH_IMPORT_EXPORT void ShOutputHtml();
66}
67
John Kessenich94a81fb2013-08-31 02:41:30 +000068// Command-line options
69enum TOptions {
John Kessenich906cc212016-12-09 19:22:20 -070070 EOptionNone = 0,
71 EOptionIntermediate = (1 << 0),
72 EOptionSuppressInfolog = (1 << 1),
73 EOptionMemoryLeakMode = (1 << 2),
74 EOptionRelaxedErrors = (1 << 3),
75 EOptionGiveWarnings = (1 << 4),
76 EOptionLinkProgram = (1 << 5),
77 EOptionMultiThreaded = (1 << 6),
78 EOptionDumpConfig = (1 << 7),
79 EOptionDumpReflection = (1 << 8),
80 EOptionSuppressWarnings = (1 << 9),
81 EOptionDumpVersions = (1 << 10),
82 EOptionSpv = (1 << 11),
83 EOptionHumanReadableSpv = (1 << 12),
84 EOptionVulkanRules = (1 << 13),
85 EOptionDefaultDesktop = (1 << 14),
86 EOptionOutputPreprocessed = (1 << 15),
87 EOptionOutputHexadecimal = (1 << 16),
88 EOptionReadHlsl = (1 << 17),
89 EOptionCascadingErrors = (1 << 18),
90 EOptionAutoMapBindings = (1 << 19),
steve-lunarge0b9deb2016-09-16 13:26:37 -060091 EOptionFlattenUniformArrays = (1 << 20),
John Kessenich906cc212016-12-09 19:22:20 -070092 EOptionNoStorageFormat = (1 << 21),
John Kessenich20f01e72016-12-12 11:41:43 -070093 EOptionKeepUncalled = (1 << 22),
John Kessenich4f1403e2017-04-05 17:38:20 -060094 EOptionHlslOffsets = (1 << 23),
steve-lunargbe283552017-04-18 12:18:01 -060095 EOptionHlslIoMapping = (1 << 24),
John Kessenich71facdf2017-05-17 18:28:19 -060096 EOptionAutoMapLocations = (1 << 25),
John Kessenich121853f2017-05-31 17:11:16 -060097 EOptionDebug = (1 << 26),
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +020098 EOptionStdin = (1 << 27),
GregFcd1f1692017-09-21 18:40:22 -060099 EOptionOptimizeDisable = (1 << 28),
100 EOptionOptimizeSize = (1 << 29),
LoopDawgb22c0692017-12-06 16:52:03 -0700101 EOptionInvertY = (1 << 30),
John Kessenichc6c80a62018-03-05 22:23:17 -0700102 EOptionDumpBareVersion = (1 << 31),
John Kessenich94a81fb2013-08-31 02:41:30 +0000103};
John Kessenich5d610ee2018-03-07 18:05:55 -0700104bool targetHlslFunctionality1 = false;
John Kessenich94a81fb2013-08-31 02:41:30 +0000105
John Kessenicha0af4732012-12-12 21:15:54 +0000106//
John Kessenich68d78fd2015-07-12 19:28:10 -0600107// Return codes from main/exit().
John Kessenicha0af4732012-12-12 21:15:54 +0000108//
109enum TFailCode {
110 ESuccess = 0,
111 EFailUsage,
112 EFailCompile,
113 EFailLink,
114 EFailCompilerCreate,
John Kessenich2b07c7e2013-07-31 18:44:13 +0000115 EFailThreadCreate,
John Kessenicha0af4732012-12-12 21:15:54 +0000116 EFailLinkerCreate
117};
118
119//
John Kessenich68d78fd2015-07-12 19:28:10 -0600120// Forward declarations.
John Kessenicha0af4732012-12-12 21:15:54 +0000121//
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600122EShLanguage FindLanguage(const std::string& name, bool parseSuffix=true);
John Kessenich51cdd902014-02-18 23:37:57 +0000123void CompileFile(const char* fileName, ShHandle);
John Kessenicha0af4732012-12-12 21:15:54 +0000124void usage();
John Kessenich04acb1b2017-06-14 17:36:50 -0600125char* ReadFileData(const char* fileName);
126void FreeFileData(char* data);
John Kessenich54d8cda2013-02-11 22:36:01 +0000127void InfoLogMsg(const char* msg, const char* name, const int num);
John Kessenich41cf6b52013-06-25 18:10:05 +0000128
John Kessenichc999ba22013-11-07 23:33:24 +0000129// Globally track if any compile or link failure.
130bool CompileFailed = false;
131bool LinkFailed = false;
132
John Kessenich94f28eb2017-11-13 01:32:06 -0700133// array of unique places to leave the shader names and infologs for the asynchronous compiles
134std::vector<std::unique_ptr<glslang::TWorkItem>> WorkItems;
135
John Kessenich05a70632013-09-17 19:26:08 +0000136TBuiltInResource Resources;
137std::string ConfigFile;
138
John Kessenicha0af4732012-12-12 21:15:54 +0000139//
John Kessenichf0bcb0a2016-04-02 13:09:14 -0600140// Parse either a .conf file provided by the user or the default from glslang::DefaultTBuiltInResource
John Kessenich05a70632013-09-17 19:26:08 +0000141//
142void ProcessConfigFile()
John Kessenichb51f62c2013-04-11 16:31:09 +0000143{
John Kessenich04acb1b2017-06-14 17:36:50 -0600144 if (ConfigFile.size() == 0)
Lei Zhang414eb602016-03-04 16:22:34 -0500145 Resources = glslang::DefaultTBuiltInResource;
John Kessenich04acb1b2017-06-14 17:36:50 -0600146 else {
147 char* configString = ReadFileData(ConfigFile.c_str());
148 glslang::DecodeResourceLimits(&Resources, configString);
149 FreeFileData(configString);
John Kessenich05a70632013-09-17 19:26:08 +0000150 }
John Kessenicha0af4732012-12-12 21:15:54 +0000151}
152
John Kessenich94a81fb2013-08-31 02:41:30 +0000153int Options = 0;
John Kessenich68d78fd2015-07-12 19:28:10 -0600154const char* ExecutableName = nullptr;
155const char* binaryFileName = nullptr;
John Kessenich4d65ee32016-03-12 18:17:47 -0700156const char* entryPointName = nullptr;
steve-lunargf1e0c872016-10-31 15:13:43 -0600157const char* sourceEntryPointName = nullptr;
Dan Bakerc6ede892016-08-11 14:06:06 -0400158const char* shaderStageName = nullptr;
Flavioaea3c892017-02-06 11:46:35 -0800159const char* variableName = nullptr;
Rex Xucb61eec2018-03-07 13:10:01 +0800160bool HlslEnable16BitTypes = false;
John Kessenich971a0a82017-06-07 15:06:58 -0600161std::vector<std::string> IncludeDirectoryList;
John Kessenich66011cb2018-03-06 16:12:04 -0700162int ClientInputSemanticsVersion = 100; // maps to, say, #define VULKAN 100
163glslang::EshTargetClientVersion VulkanClientVersion =
164 glslang::EShTargetVulkan_1_0; // would map to, say, Vulkan 1.0
165glslang::EshTargetClientVersion OpenGLClientVersion =
166 glslang::EShTargetOpenGL_450; // doesn't influence anything yet, but maps to OpenGL 4.50
167glslang::EShTargetLanguageVersion TargetVersion =
168 glslang::EShTargetSpv_1_0; // maps to, say, SPIR-V 1.0
169std::vector<std::string> Processes; // what should be recorded by OpModuleProcessed, or equivalent
John Kessenich68d78fd2015-07-12 19:28:10 -0600170
LoopDawg08a14422017-10-17 19:27:14 -0600171// Per descriptor-set binding base data
172typedef std::map<unsigned int, unsigned int> TPerSetBaseBinding;
173
174std::array<std::array<unsigned int, EShLangCount>, glslang::EResCount> baseBinding;
175std::array<std::array<TPerSetBaseBinding, EShLangCount>, glslang::EResCount> baseBindingForSet;
Hyangran Park36dc8292017-05-02 16:27:29 +0900176std::array<std::vector<std::string>, EShLangCount> baseResourceSetBinding;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600177
John Kessenicha9313662017-06-15 10:40:49 -0600178// Add things like "#define ..." to a preamble to use in the beginning of the shader.
179class TPreamble {
180public:
181 TPreamble() { }
182
183 bool isSet() const { return text.size() > 0; }
184 const char* get() const { return text.c_str(); }
185
186 // #define...
187 void addDef(std::string def)
188 {
189 text.append("#define ");
190 fixLine(def);
191
John Kessenich2a271162017-07-20 20:00:36 -0600192 Processes.push_back("D");
193 Processes.back().append(def);
194
John Kessenicha9313662017-06-15 10:40:49 -0600195 // The first "=" needs to turn into a space
LoopDawgb97b25e2017-07-12 09:04:39 -0600196 const size_t equal = def.find_first_of("=");
John Kessenicha9313662017-06-15 10:40:49 -0600197 if (equal != def.npos)
198 def[equal] = ' ';
199
200 text.append(def);
201 text.append("\n");
202 }
203
204 // #undef...
205 void addUndef(std::string undef)
206 {
207 text.append("#undef ");
208 fixLine(undef);
John Kessenich2a271162017-07-20 20:00:36 -0600209
210 Processes.push_back("U");
211 Processes.back().append(undef);
212
John Kessenicha9313662017-06-15 10:40:49 -0600213 text.append(undef);
214 text.append("\n");
215 }
216
217protected:
218 void fixLine(std::string& line)
219 {
220 // Can't go past a newline in the line
LoopDawgb97b25e2017-07-12 09:04:39 -0600221 const size_t end = line.find_first_of("\n");
John Kessenicha9313662017-06-15 10:40:49 -0600222 if (end != line.npos)
223 line = line.substr(0, end);
224 }
225
226 std::string text; // contents of preamble
227};
228
229TPreamble UserPreamble;
230
John Kessenich68d78fd2015-07-12 19:28:10 -0600231//
232// Create the default name for saving a binary if -o is not provided.
233//
234const char* GetBinaryName(EShLanguage stage)
235{
236 const char* name;
237 if (binaryFileName == nullptr) {
238 switch (stage) {
239 case EShLangVertex: name = "vert.spv"; break;
240 case EShLangTessControl: name = "tesc.spv"; break;
241 case EShLangTessEvaluation: name = "tese.spv"; break;
242 case EShLangGeometry: name = "geom.spv"; break;
243 case EShLangFragment: name = "frag.spv"; break;
244 case EShLangCompute: name = "comp.spv"; break;
245 default: name = "unknown"; break;
246 }
247 } else
248 name = binaryFileName;
249
250 return name;
251}
John Kessenich2b07c7e2013-07-31 18:44:13 +0000252
John Kessenich05a70632013-09-17 19:26:08 +0000253//
254// *.conf => this is a config file that can set limits/resources
255//
256bool SetConfigFile(const std::string& name)
257{
258 if (name.size() < 5)
259 return false;
260
John Kessenich4c706852013-10-11 16:28:43 +0000261 if (name.compare(name.size() - 5, 5, ".conf") == 0) {
John Kessenich05a70632013-09-17 19:26:08 +0000262 ConfigFile = name;
263 return true;
264 }
265
266 return false;
267}
268
John Kessenich68d78fd2015-07-12 19:28:10 -0600269//
270// Give error and exit with failure code.
271//
272void Error(const char* message)
273{
John Kessenichaf527992017-11-02 22:48:15 -0600274 fprintf(stderr, "%s: Error %s (use -h for usage)\n", ExecutableName, message);
John Kessenich68d78fd2015-07-12 19:28:10 -0600275 exit(EFailUsage);
276}
277
278//
LoopDawg08a14422017-10-17 19:27:14 -0600279// Process an optional binding base of one the forms:
280// --argname [stage] base // base for stage (if given) or all stages (if not)
LoopDawge5709552017-10-21 10:46:39 -0600281// --argname [stage] [base set]... // set/base pairs: set the base for given binding set.
LoopDawg08a14422017-10-17 19:27:14 -0600282
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600283// Where stage is one of the forms accepted by FindLanguage, and base is an integer
284//
LoopDawg08a14422017-10-17 19:27:14 -0600285void ProcessBindingBase(int& argc, char**& argv, glslang::TResourceType res)
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600286{
287 if (argc < 2)
288 usage();
289
LoopDawg08a14422017-10-17 19:27:14 -0600290 EShLanguage lang = EShLangCount;
291 int singleBase = 0;
292 TPerSetBaseBinding perSetBase;
293 int arg = 1;
294
295 // Parse stage, if given
296 if (!isdigit(argv[arg][0])) {
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600297 if (argc < 3) // this form needs one more argument
298 usage();
John Kessenichecba76f2017-01-06 00:34:48 -0700299
LoopDawg08a14422017-10-17 19:27:14 -0600300 lang = FindLanguage(argv[arg++], false);
301 }
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600302
LoopDawg08a14422017-10-17 19:27:14 -0600303 if ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) {
304 // Parse a per-set binding base
305 while ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) {
LoopDawg08a14422017-10-17 19:27:14 -0600306 const int baseNum = atoi(argv[arg++]);
LoopDawge5709552017-10-21 10:46:39 -0600307 const int setNum = atoi(argv[arg++]);
LoopDawg08a14422017-10-17 19:27:14 -0600308 perSetBase[setNum] = baseNum;
309 }
310 } else {
311 // Parse single binding base
312 singleBase = atoi(argv[arg++]);
313 }
314
315 argc -= (arg-1);
316 argv += (arg-1);
317
318 // Set one or all languages
319 const int langMin = (lang < EShLangCount) ? lang+0 : 0;
320 const int langMax = (lang < EShLangCount) ? lang+1 : EShLangCount;
321
322 for (int lang = langMin; lang < langMax; ++lang) {
323 if (!perSetBase.empty())
324 baseBindingForSet[res][lang] = perSetBase;
325 else
326 baseBinding[res][lang] = singleBase;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600327 }
328}
329
Hyangran Park36dc8292017-05-02 16:27:29 +0900330void ProcessResourceSetBindingBase(int& argc, char**& argv, std::array<std::vector<std::string>, EShLangCount>& base)
331{
332 if (argc < 2)
333 usage();
334
335 if (!isdigit(argv[1][0])) {
LoopDawg52017192017-07-14 15:15:47 -0600336 if (argc < 3) // this form needs one more argument
Hyangran Park36dc8292017-05-02 16:27:29 +0900337 usage();
338
LoopDawg52017192017-07-14 15:15:47 -0600339 // Parse form: --argname stage [regname set base...], or:
340 // --argname stage set
Hyangran Park36dc8292017-05-02 16:27:29 +0900341 const EShLanguage lang = FindLanguage(argv[1], false);
342
LoopDawg52017192017-07-14 15:15:47 -0600343 argc--;
344 argv++;
345
346 while (argc > 1 && argv[1] != nullptr && argv[1][0] != '-') {
347 base[lang].push_back(argv[1]);
348
349 argc--;
350 argv++;
Hyangran Park36dc8292017-05-02 16:27:29 +0900351 }
LoopDawg52017192017-07-14 15:15:47 -0600352
353 // Must have one arg, or a multiple of three (for [regname set binding] triples)
354 if (base[lang].size() != 1 && (base[lang].size() % 3) != 0)
355 usage();
356
Hyangran Park36dc8292017-05-02 16:27:29 +0900357 } else {
LoopDawg52017192017-07-14 15:15:47 -0600358 // Parse form: --argname set
Hyangran Park36dc8292017-05-02 16:27:29 +0900359 for (int lang=0; lang<EShLangCount; ++lang)
360 base[lang].push_back(argv[1]);
361
362 argc--;
363 argv++;
364 }
365}
366
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600367//
John Kessenich68d78fd2015-07-12 19:28:10 -0600368// Do all command-line argument parsing. This includes building up the work-items
369// to be processed later, and saving all the command-line options.
370//
371// Does not return (it exits) if command-line is fatally flawed.
372//
Juan Lopeza558b262017-04-02 23:04:00 +0200373void ProcessArguments(std::vector<std::unique_ptr<glslang::TWorkItem>>& workItems, int argc, char* argv[])
John Kessenich2b07c7e2013-07-31 18:44:13 +0000374{
LoopDawg08a14422017-10-17 19:27:14 -0600375 for (int res = 0; res < glslang::EResCount; ++res)
376 baseBinding[res].fill(0);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600377
John Kessenich38f3b892013-09-06 19:52:57 +0000378 ExecutableName = argv[0];
Juan Lopeza558b262017-04-02 23:04:00 +0200379 workItems.reserve(argc);
John Kessenich38f3b892013-09-06 19:52:57 +0000380
John Kessenichc178f0a2017-06-23 10:58:31 -0600381 const auto bumpArg = [&]() {
382 if (argc > 0) {
383 argc--;
384 argv++;
385 }
386 };
387
388 // read a string directly attached to a single-letter option
John Kessenich6263fb12017-06-14 15:52:44 -0600389 const auto getStringOperand = [&](const char* desc) {
390 if (argv[0][2] == 0) {
391 printf("%s must immediately follow option (no spaces)\n", desc);
392 exit(EFailUsage);
393 }
394 return argv[0] + 2;
395 };
396
John Kessenich6353d552017-06-23 11:11:09 -0600397 // read a number attached to a single-letter option
398 const auto getAttachedNumber = [&](const char* desc) {
399 int num = atoi(argv[0] + 2);
400 if (num == 0) {
401 printf("%s: expected attached non-0 number\n", desc);
402 exit(EFailUsage);
403 }
404 return num;
405 };
406
407 // minimum needed (without overriding something else) to target Vulkan SPIR-V
408 const auto setVulkanSpv = []() {
409 Options |= EOptionSpv;
410 Options |= EOptionVulkanRules;
411 Options |= EOptionLinkProgram;
412 };
413
414 // minimum needed (without overriding something else) to target OpenGL SPIR-V
415 const auto setOpenGlSpv = []() {
416 Options |= EOptionSpv;
417 Options |= EOptionLinkProgram;
418 // undo a -H default to Vulkan
419 Options &= ~EOptionVulkanRules;
420 };
421
John Kessenichc178f0a2017-06-23 10:58:31 -0600422 for (bumpArg(); argc >= 1; bumpArg()) {
John Kessenich2b07c7e2013-07-31 18:44:13 +0000423 if (argv[0][0] == '-') {
John Kessenich2aa7f3a2015-05-15 16:02:07 +0000424 switch (argv[0][1]) {
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600425 case '-':
426 {
427 std::string lowerword(argv[0]+2);
428 std::transform(lowerword.begin(), lowerword.end(), lowerword.begin(), ::tolower);
429
430 // handle --word style options
John Kessenich6263fb12017-06-14 15:52:44 -0600431 if (lowerword == "auto-map-bindings" || // synonyms
John Kessenich6353d552017-06-23 11:11:09 -0600432 lowerword == "auto-map-binding" ||
433 lowerword == "amb") {
John Kessenich6263fb12017-06-14 15:52:44 -0600434 Options |= EOptionAutoMapBindings;
435 } else if (lowerword == "auto-map-locations" || // synonyms
436 lowerword == "aml") {
437 Options |= EOptionAutoMapLocations;
John Kessenich6353d552017-06-23 11:11:09 -0600438 } else if (lowerword == "client") {
439 if (argc > 1) {
440 if (strcmp(argv[1], "vulkan100") == 0)
441 setVulkanSpv();
442 else if (strcmp(argv[1], "opengl100") == 0)
443 setOpenGlSpv();
444 else
445 Error("--client expects vulkan100 or opengl100");
446 }
447 bumpArg();
John Kessenich6263fb12017-06-14 15:52:44 -0600448 } else if (lowerword == "flatten-uniform-arrays" || // synonyms
449 lowerword == "flatten-uniform-array" ||
450 lowerword == "fua") {
451 Options |= EOptionFlattenUniformArrays;
452 } else if (lowerword == "hlsl-offsets") {
453 Options |= EOptionHlslOffsets;
454 } else if (lowerword == "hlsl-iomap" ||
455 lowerword == "hlsl-iomapper" ||
456 lowerword == "hlsl-iomapping") {
457 Options |= EOptionHlslIoMapping;
Rex Xucb61eec2018-03-07 13:10:01 +0800458 } else if (lowerword == "hlsl-enable-16bit-types") {
459 HlslEnable16BitTypes = true;
John Kessenichc6c80a62018-03-05 22:23:17 -0700460 } else if (lowerword == "invert-y" || // synonyms
461 lowerword == "iy") {
462 Options |= EOptionInvertY;
John Kessenich6263fb12017-06-14 15:52:44 -0600463 } else if (lowerword == "keep-uncalled" || // synonyms
464 lowerword == "ku") {
465 Options |= EOptionKeepUncalled;
466 } else if (lowerword == "no-storage-format" || // synonyms
467 lowerword == "nsf") {
468 Options |= EOptionNoStorageFormat;
John Kessenich2a271162017-07-20 20:00:36 -0600469 } else if (lowerword == "relaxed-errors") {
470 Options |= EOptionRelaxedErrors;
John Kessenich6263fb12017-06-14 15:52:44 -0600471 } else if (lowerword == "resource-set-bindings" || // synonyms
472 lowerword == "resource-set-binding" ||
473 lowerword == "rsb") {
474 ProcessResourceSetBindingBase(argc, argv, baseResourceSetBinding);
steve-lunarg9088be42016-11-01 10:31:42 -0600475 } else if (lowerword == "shift-image-bindings" || // synonyms
476 lowerword == "shift-image-binding" ||
477 lowerword == "sib") {
LoopDawg08a14422017-10-17 19:27:14 -0600478 ProcessBindingBase(argc, argv, glslang::EResImage);
John Kessenich6263fb12017-06-14 15:52:44 -0600479 } else if (lowerword == "shift-sampler-bindings" || // synonyms
480 lowerword == "shift-sampler-binding" ||
481 lowerword == "ssb") {
LoopDawg08a14422017-10-17 19:27:14 -0600482 ProcessBindingBase(argc, argv, glslang::EResSampler);
John Kessenich6263fb12017-06-14 15:52:44 -0600483 } else if (lowerword == "shift-uav-bindings" || // synonyms
484 lowerword == "shift-uav-binding" ||
485 lowerword == "suavb") {
LoopDawg08a14422017-10-17 19:27:14 -0600486 ProcessBindingBase(argc, argv, glslang::EResUav);
John Kessenich6263fb12017-06-14 15:52:44 -0600487 } else if (lowerword == "shift-texture-bindings" || // synonyms
488 lowerword == "shift-texture-binding" ||
489 lowerword == "stb") {
LoopDawg08a14422017-10-17 19:27:14 -0600490 ProcessBindingBase(argc, argv, glslang::EResTexture);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600491 } else if (lowerword == "shift-ubo-bindings" || // synonyms
492 lowerword == "shift-ubo-binding" ||
steve-lunargbe283552017-04-18 12:18:01 -0600493 lowerword == "shift-cbuffer-bindings" ||
494 lowerword == "shift-cbuffer-binding" ||
495 lowerword == "sub" ||
496 lowerword == "scb") {
LoopDawg08a14422017-10-17 19:27:14 -0600497 ProcessBindingBase(argc, argv, glslang::EResUbo);
steve-lunarg932bb5c2017-02-21 17:19:08 -0700498 } else if (lowerword == "shift-ssbo-bindings" || // synonyms
499 lowerword == "shift-ssbo-binding" ||
500 lowerword == "sbb") {
LoopDawg08a14422017-10-17 19:27:14 -0600501 ProcessBindingBase(argc, argv, glslang::EResSsbo);
John Kessenich6263fb12017-06-14 15:52:44 -0600502 } else if (lowerword == "source-entrypoint" || // synonyms
503 lowerword == "sep") {
John Kessenichc178f0a2017-06-23 10:58:31 -0600504 if (argc <= 1)
John Kessenich6263fb12017-06-14 15:52:44 -0600505 Error("no <entry-point> provided for --source-entrypoint");
John Kessenichc178f0a2017-06-23 10:58:31 -0600506 sourceEntryPointName = argv[1];
507 bumpArg();
John Kessenich6263fb12017-06-14 15:52:44 -0600508 break;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200509 } else if (lowerword == "stdin") {
510 Options |= EOptionStdin;
511 shaderStageName = argv[1];
John Kessenich2a271162017-07-20 20:00:36 -0600512 } else if (lowerword == "suppress-warnings") {
513 Options |= EOptionSuppressWarnings;
John Kessenich6353d552017-06-23 11:11:09 -0600514 } else if (lowerword == "target-env") {
515 if (argc > 1) {
516 if (strcmp(argv[1], "vulkan1.0") == 0) {
517 setVulkanSpv();
John Kessenich66011cb2018-03-06 16:12:04 -0700518 VulkanClientVersion = glslang::EShTargetVulkan_1_0;
519 } else if (strcmp(argv[1], "vulkan1.1") == 0) {
520 setVulkanSpv();
521 TargetVersion = glslang::EShTargetSpv_1_3;
522 VulkanClientVersion = glslang::EShTargetVulkan_1_1;
John Kessenich6353d552017-06-23 11:11:09 -0600523 } else if (strcmp(argv[1], "opengl") == 0) {
524 setOpenGlSpv();
John Kessenich66011cb2018-03-06 16:12:04 -0700525 OpenGLClientVersion = glslang::EShTargetOpenGL_450;
John Kessenich6353d552017-06-23 11:11:09 -0600526 } else
John Kessenich5d610ee2018-03-07 18:05:55 -0700527 Error("--target-env expected vulkan1.0, vulkan1.1, or opengl");
John Kessenich6353d552017-06-23 11:11:09 -0600528 }
529 bumpArg();
Flavio15017db2017-02-15 14:29:33 -0800530 } else if (lowerword == "variable-name" || // synonyms
John Kessenich66011cb2018-03-06 16:12:04 -0700531 lowerword == "vn") {
Flavio15017db2017-02-15 14:29:33 -0800532 Options |= EOptionOutputHexadecimal;
John Kessenichc178f0a2017-06-23 10:58:31 -0600533 if (argc <= 1)
Flavio15017db2017-02-15 14:29:33 -0800534 Error("no <C-variable-name> provided for --variable-name");
John Kessenichc178f0a2017-06-23 10:58:31 -0600535 variableName = argv[1];
536 bumpArg();
Flavio15017db2017-02-15 14:29:33 -0800537 break;
John Kessenichc6c80a62018-03-05 22:23:17 -0700538 } else if (lowerword == "version") {
539 Options |= EOptionDumpVersions;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600540 } else {
541 usage();
542 }
543 }
544 break;
John Kessenich6263fb12017-06-14 15:52:44 -0600545 case 'C':
546 Options |= EOptionCascadingErrors;
547 break;
548 case 'D':
John Kessenicha9313662017-06-15 10:40:49 -0600549 if (argv[0][2] == 0)
550 Options |= EOptionReadHlsl;
551 else
552 UserPreamble.addDef(getStringOperand("-D<macro> macro name"));
John Kessenich6263fb12017-06-14 15:52:44 -0600553 break;
554 case 'E':
555 Options |= EOptionOutputPreprocessed;
556 break;
557 case 'G':
John Kessenich6353d552017-06-23 11:11:09 -0600558 // OpenGL Client
559 setOpenGlSpv();
560 if (argv[0][2] != 0)
561 ClientInputSemanticsVersion = getAttachedNumber("-G<num> client input semantics");
John Kessenich6263fb12017-06-14 15:52:44 -0600562 break;
John Kessenich2aa7f3a2015-05-15 16:02:07 +0000563 case 'H':
564 Options |= EOptionHumanReadableSpv;
John Kessenich91e4aa52016-07-07 17:46:42 -0600565 if ((Options & EOptionSpv) == 0) {
566 // default to Vulkan
John Kessenich6353d552017-06-23 11:11:09 -0600567 setVulkanSpv();
John Kessenich91e4aa52016-07-07 17:46:42 -0600568 }
569 break;
John Kessenich971a0a82017-06-07 15:06:58 -0600570 case 'I':
John Kessenicha9313662017-06-15 10:40:49 -0600571 IncludeDirectoryList.push_back(getStringOperand("-I<dir> include path"));
John Kessenich68d78fd2015-07-12 19:28:10 -0600572 break;
GregFcd1f1692017-09-21 18:40:22 -0600573 case 'O':
574 if (argv[0][2] == 'd')
575 Options |= EOptionOptimizeDisable;
576 else if (argv[0][2] == 's')
577#ifdef ENABLE_OPT
578 Options |= EOptionOptimizeSize;
579#else
580 Error("-Os not available; optimizer not linked");
581#endif
582 else
583 Error("unknown -O option");
584 break;
Dan Baker5afdd782016-08-11 17:53:57 -0400585 case 'S':
John Kessenichc178f0a2017-06-23 10:58:31 -0600586 if (argc <= 1)
Dan Baker5afdd782016-08-11 17:53:57 -0400587 Error("no <stage> specified for -S");
John Kessenichc178f0a2017-06-23 10:58:31 -0600588 shaderStageName = argv[1];
589 bumpArg();
dankbaker45d49bc2016-08-08 21:43:07 -0400590 break;
John Kessenicha9313662017-06-15 10:40:49 -0600591 case 'U':
592 UserPreamble.addUndef(getStringOperand("-U<macro>: macro name"));
593 break;
John Kessenich6263fb12017-06-14 15:52:44 -0600594 case 'V':
John Kessenich6353d552017-06-23 11:11:09 -0600595 setVulkanSpv();
596 if (argv[0][2] != 0)
David Neto506d2c22018-02-27 21:55:23 -0500597 ClientInputSemanticsVersion = getAttachedNumber("-V<num> client input semantics");
John Kessenichc555ddd2015-06-17 02:38:44 +0000598 break;
John Kessenich05a70632013-09-17 19:26:08 +0000599 case 'c':
600 Options |= EOptionDumpConfig;
601 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000602 case 'd':
John Kessenichc6c80a62018-03-05 22:23:17 -0700603 if (strncmp(&argv[0][1], "dumpversion", strlen(&argv[0][1]) + 1) == 0 ||
604 strncmp(&argv[0][1], "dumpfullversion", strlen(&argv[0][1]) + 1) == 0)
605 Options |= EOptionDumpBareVersion;
606 else
607 Options |= EOptionDefaultDesktop;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000608 break;
John Kessenich4d65ee32016-03-12 18:17:47 -0700609 case 'e':
610 // HLSL todo: entry point handle needs much more sophistication.
611 // This is okay for one compilation unit with one entry point.
612 entryPointName = argv[1];
John Kessenichc178f0a2017-06-23 10:58:31 -0600613 if (argc <= 1)
John Kessenicha25530c2017-09-11 20:13:49 -0600614 Error("no <name> provided for -e");
John Kessenichc178f0a2017-06-23 10:58:31 -0600615 bumpArg();
John Kessenich4d65ee32016-03-12 18:17:47 -0700616 break;
John Kessenich5d610ee2018-03-07 18:05:55 -0700617 case 'f':
618 if (strcmp(&argv[0][2], "hlsl_functionality1") == 0)
619 targetHlslFunctionality1 = true;
620 else
621 Error("-f: expected hlsl_functionality1");
622 break;
John Kessenich121853f2017-05-31 17:11:16 -0600623 case 'g':
624 Options |= EOptionDebug;
625 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600626 case 'h':
627 usage();
628 break;
John Kessenich05a70632013-09-17 19:26:08 +0000629 case 'i':
John Kessenich94a81fb2013-08-31 02:41:30 +0000630 Options |= EOptionIntermediate;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000631 break;
632 case 'l':
John Kessenichd78e3512014-08-25 20:07:55 +0000633 Options |= EOptionLinkProgram;
John Kessenich94a81fb2013-08-31 02:41:30 +0000634 break;
635 case 'm':
636 Options |= EOptionMemoryLeakMode;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000637 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600638 case 'o':
John Kessenichc178f0a2017-06-23 10:58:31 -0600639 if (argc <= 1)
John Kessenich68d78fd2015-07-12 19:28:10 -0600640 Error("no <file> provided for -o");
John Kessenichc178f0a2017-06-23 10:58:31 -0600641 binaryFileName = argv[1];
642 bumpArg();
John Kessenich68d78fd2015-07-12 19:28:10 -0600643 break;
John Kessenich11f9fc72013-11-07 01:06:34 +0000644 case 'q':
645 Options |= EOptionDumpReflection;
646 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000647 case 'r':
John Kessenich94a81fb2013-08-31 02:41:30 +0000648 Options |= EOptionRelaxedErrors;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000649 break;
650 case 's':
John Kessenich94a81fb2013-08-31 02:41:30 +0000651 Options |= EOptionSuppressInfolog;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000652 break;
John Kessenich38f3b892013-09-06 19:52:57 +0000653 case 't':
Juan Lopeza558b262017-04-02 23:04:00 +0200654 Options |= EOptionMultiThreaded;
John Kessenich38f3b892013-09-06 19:52:57 +0000655 break;
John Kessenich319de232013-12-04 04:43:40 +0000656 case 'v':
657 Options |= EOptionDumpVersions;
658 break;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000659 case 'w':
660 Options |= EOptionSuppressWarnings;
661 break;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500662 case 'x':
663 Options |= EOptionOutputHexadecimal;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500664 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000665 default:
John Kessenich68d78fd2015-07-12 19:28:10 -0600666 usage();
667 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000668 }
John Kessenich38f3b892013-09-06 19:52:57 +0000669 } else {
John Kessenich05a70632013-09-17 19:26:08 +0000670 std::string name(argv[0]);
671 if (! SetConfigFile(name)) {
Juan Lopeza558b262017-04-02 23:04:00 +0200672 workItems.push_back(std::unique_ptr<glslang::TWorkItem>(new glslang::TWorkItem(name)));
John Kessenich05a70632013-09-17 19:26:08 +0000673 }
John Kessenich38f3b892013-09-06 19:52:57 +0000674 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000675 }
676
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200677 // Make sure that -S is always specified if --stdin is specified
678 if ((Options & EOptionStdin) && shaderStageName == nullptr)
679 Error("must provide -S when --stdin is given");
680
John Kessenich68d78fd2015-07-12 19:28:10 -0600681 // Make sure that -E is not specified alongside linking (which includes SPV generation)
682 if ((Options & EOptionOutputPreprocessed) && (Options & EOptionLinkProgram))
683 Error("can't use -E when linking is selected");
John Kessenichc555ddd2015-06-17 02:38:44 +0000684
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500685 // -o or -x makes no sense if there is no target binary
John Kessenich68d78fd2015-07-12 19:28:10 -0600686 if (binaryFileName && (Options & EOptionSpv) == 0)
687 Error("no binary generation requested (e.g., -V)");
steve-lunarge0b9deb2016-09-16 13:26:37 -0600688
689 if ((Options & EOptionFlattenUniformArrays) != 0 &&
690 (Options & EOptionReadHlsl) == 0)
691 Error("uniform array flattening only valid when compiling HLSL source.");
John Kessenich2b07c7e2013-07-31 18:44:13 +0000692}
693
John Kessenich68d78fd2015-07-12 19:28:10 -0600694//
695// Translate the meaningful subset of command-line options to parser-behavior options.
696//
John Kessenichb0a7eb52013-11-07 17:44:20 +0000697void SetMessageOptions(EShMessages& messages)
698{
699 if (Options & EOptionRelaxedErrors)
700 messages = (EShMessages)(messages | EShMsgRelaxedErrors);
701 if (Options & EOptionIntermediate)
702 messages = (EShMessages)(messages | EShMsgAST);
703 if (Options & EOptionSuppressWarnings)
704 messages = (EShMessages)(messages | EShMsgSuppressWarnings);
John Kessenich68d78fd2015-07-12 19:28:10 -0600705 if (Options & EOptionSpv)
706 messages = (EShMessages)(messages | EShMsgSpvRules);
707 if (Options & EOptionVulkanRules)
708 messages = (EShMessages)(messages | EShMsgVulkanRules);
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400709 if (Options & EOptionOutputPreprocessed)
710 messages = (EShMessages)(messages | EShMsgOnlyPreprocessor);
John Kessenich66e2faf2016-03-12 18:34:36 -0700711 if (Options & EOptionReadHlsl)
712 messages = (EShMessages)(messages | EShMsgReadHlsl);
John Kessenicha86836e2016-07-09 14:50:57 -0600713 if (Options & EOptionCascadingErrors)
714 messages = (EShMessages)(messages | EShMsgCascadingErrors);
John Kessenich906cc212016-12-09 19:22:20 -0700715 if (Options & EOptionKeepUncalled)
716 messages = (EShMessages)(messages | EShMsgKeepUncalled);
John Kessenich4f1403e2017-04-05 17:38:20 -0600717 if (Options & EOptionHlslOffsets)
718 messages = (EShMessages)(messages | EShMsgHlslOffsets);
John Kessenich121853f2017-05-31 17:11:16 -0600719 if (Options & EOptionDebug)
720 messages = (EShMessages)(messages | EShMsgDebugInfo);
Rex Xucb61eec2018-03-07 13:10:01 +0800721 if (HlslEnable16BitTypes)
722 messages = (EShMessages)(messages | EShMsgHlslEnable16BitTypes);
John Kessenichb0a7eb52013-11-07 17:44:20 +0000723}
724
John Kessenich68d78fd2015-07-12 19:28:10 -0600725//
John Kessenich69f4b512013-09-04 21:19:27 +0000726// Thread entry point, for non-linking asynchronous mode.
John Kessenichc999ba22013-11-07 23:33:24 +0000727//
Juan Lopeza558b262017-04-02 23:04:00 +0200728void CompileShaders(glslang::TWorklist& worklist)
John Kessenich2b07c7e2013-07-31 18:44:13 +0000729{
John Kessenich38f3b892013-09-06 19:52:57 +0000730 glslang::TWorkItem* workItem;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200731 if (Options & EOptionStdin) {
732 worklist.remove(workItem);
733 ShHandle compiler = ShConstructCompiler(FindLanguage("stdin"), Options);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000734 if (compiler == 0)
Juan Lopeza558b262017-04-02 23:04:00 +0200735 return;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000736
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200737 CompileFile("stdin", compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000738
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200739 if (! (Options & EOptionSuppressInfolog))
740 workItem->results = ShGetInfoLog(compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000741
742 ShDestruct(compiler);
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200743 } else {
744 while (worklist.remove(workItem)) {
745 ShHandle compiler = ShConstructCompiler(FindLanguage(workItem->name), Options);
746 if (compiler == 0)
747 return;
748
749 CompileFile(workItem->name.c_str(), compiler);
750
751 if (! (Options & EOptionSuppressInfolog))
752 workItem->results = ShGetInfoLog(compiler);
753
754 ShDestruct(compiler);
755 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000756 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000757}
758
John Kessenich6626cad2015-06-19 05:14:19 +0000759// Outputs the given string, but only if it is non-null and non-empty.
760// This prevents erroneous newlines from appearing.
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400761void PutsIfNonEmpty(const char* str)
John Kessenich6626cad2015-06-19 05:14:19 +0000762{
763 if (str && str[0]) {
764 puts(str);
765 }
766}
767
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400768// Outputs the given string to stderr, but only if it is non-null and non-empty.
769// This prevents erroneous newlines from appearing.
770void StderrIfNonEmpty(const char* str)
771{
John Kessenich04acb1b2017-06-14 17:36:50 -0600772 if (str && str[0])
773 fprintf(stderr, "%s\n", str);
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400774}
775
John Kessenichc57b2a92016-01-16 15:30:03 -0700776// Simple bundling of what makes a compilation unit for ease in passing around,
777// and separation of handling file IO versus API (programmatic) compilation.
778struct ShaderCompUnit {
779 EShLanguage stage;
John Kessenich04acb1b2017-06-14 17:36:50 -0600780 static const int maxCount = 1;
781 int count; // live number of strings/names
John Kessenicha9313662017-06-15 10:40:49 -0600782 const char* text[maxCount]; // memory owned/managed externally
John Kessenich04acb1b2017-06-14 17:36:50 -0600783 std::string fileName[maxCount]; // hold's the memory, but...
784 const char* fileNameList[maxCount]; // downstream interface wants pointers
dankbakerafe6e9c2016-08-21 12:29:08 -0400785
John Kessenich04acb1b2017-06-14 17:36:50 -0600786 ShaderCompUnit(EShLanguage stage) : stage(stage), count(0) { }
dankbakerafe6e9c2016-08-21 12:29:08 -0400787
John Kessenich04acb1b2017-06-14 17:36:50 -0600788 ShaderCompUnit(const ShaderCompUnit& rhs)
dankbakerafe6e9c2016-08-21 12:29:08 -0400789 {
790 stage = rhs.stage;
John Kessenich04acb1b2017-06-14 17:36:50 -0600791 count = rhs.count;
792 for (int i = 0; i < count; ++i) {
793 fileName[i] = rhs.fileName[i];
794 text[i] = rhs.text[i];
795 fileNameList[i] = rhs.fileName[i].c_str();
796 }
dankbakerafe6e9c2016-08-21 12:29:08 -0400797 }
798
John Kessenicha9313662017-06-15 10:40:49 -0600799 void addString(std::string& ifileName, const char* itext)
John Kessenich04acb1b2017-06-14 17:36:50 -0600800 {
801 assert(count < maxCount);
802 fileName[count] = ifileName;
803 text[count] = itext;
804 fileNameList[count] = fileName[count].c_str();
805 ++count;
806 }
John Kessenichc57b2a92016-01-16 15:30:03 -0700807};
808
John Kessenich69f4b512013-09-04 21:19:27 +0000809//
John Kessenichc57b2a92016-01-16 15:30:03 -0700810// For linking mode: Will independently parse each compilation unit, but then put them
811// in the same program and link them together, making at most one linked module per
812// pipeline stage.
John Kessenich69f4b512013-09-04 21:19:27 +0000813//
814// Uses the new C++ interface instead of the old handle-based interface.
815//
John Kessenichc57b2a92016-01-16 15:30:03 -0700816
817void CompileAndLinkShaderUnits(std::vector<ShaderCompUnit> compUnits)
John Kessenich69f4b512013-09-04 21:19:27 +0000818{
819 // keep track of what to free
820 std::list<glslang::TShader*> shaders;
John Kessenichc57b2a92016-01-16 15:30:03 -0700821
John Kessenich69f4b512013-09-04 21:19:27 +0000822 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000823 SetMessageOptions(messages);
John Kessenich69f4b512013-09-04 21:19:27 +0000824
John Kessenich69f4b512013-09-04 21:19:27 +0000825 //
826 // Per-shader processing...
827 //
828
John Kessenich5b0f13a2013-11-01 03:08:40 +0000829 glslang::TProgram& program = *new glslang::TProgram;
rdb32084e82016-02-23 22:17:38 +0100830 for (auto it = compUnits.cbegin(); it != compUnits.cend(); ++it) {
831 const auto &compUnit = *it;
John Kessenichc57b2a92016-01-16 15:30:03 -0700832 glslang::TShader* shader = new glslang::TShader(compUnit.stage);
John Kessenich04acb1b2017-06-14 17:36:50 -0600833 shader->setStringsWithLengthsAndNames(compUnit.text, NULL, compUnit.fileNameList, compUnit.count);
John Kessenich4d65ee32016-03-12 18:17:47 -0700834 if (entryPointName) // HLSL todo: this needs to be tracked per compUnits
835 shader->setEntryPoint(entryPointName);
John Kessenich3693e632017-09-29 17:51:52 -0600836 if (sourceEntryPointName) {
837 if (entryPointName == nullptr)
838 printf("Warning: Changing source entry point name without setting an entry-point name.\n"
839 "Use '-e <name>'.\n");
steve-lunargf1e0c872016-10-31 15:13:43 -0600840 shader->setSourceEntryPoint(sourceEntryPointName);
John Kessenich3693e632017-09-29 17:51:52 -0600841 }
John Kessenicha9313662017-06-15 10:40:49 -0600842 if (UserPreamble.isSet())
843 shader->setPreamble(UserPreamble.get());
John Kessenich2a271162017-07-20 20:00:36 -0600844 shader->addProcesses(Processes);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600845
LoopDawg08a14422017-10-17 19:27:14 -0600846 // Set IO mapper binding shift values
847 for (int r = 0; r < glslang::EResCount; ++r) {
848 const glslang::TResourceType res = glslang::TResourceType(r);
849
850 // Set base bindings
851 shader->setShiftBinding(res, baseBinding[res][compUnit.stage]);
852
853 // Set bindings for particular resource sets
854 // TODO: use a range based for loop here, when available in all environments.
855 for (auto i = baseBindingForSet[res][compUnit.stage].begin();
856 i != baseBindingForSet[res][compUnit.stage].end(); ++i)
LoopDawge5709552017-10-21 10:46:39 -0600857 shader->setShiftBindingForSet(res, i->second, i->first);
LoopDawg08a14422017-10-17 19:27:14 -0600858 }
859
steve-lunarge0b9deb2016-09-16 13:26:37 -0600860 shader->setFlattenUniformArrays((Options & EOptionFlattenUniformArrays) != 0);
steve-lunargcce8d482016-10-14 18:36:42 -0600861 shader->setNoStorageFormat((Options & EOptionNoStorageFormat) != 0);
Hyangran Park36dc8292017-05-02 16:27:29 +0900862 shader->setResourceSetBinding(baseResourceSetBinding[compUnit.stage]);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600863
steve-lunargbe283552017-04-18 12:18:01 -0600864 if (Options & EOptionHlslIoMapping)
865 shader->setHlslIoMapping(true);
866
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600867 if (Options & EOptionAutoMapBindings)
868 shader->setAutoMapBindings(true);
John Kessenichecba76f2017-01-06 00:34:48 -0700869
John Kessenich71facdf2017-05-17 18:28:19 -0600870 if (Options & EOptionAutoMapLocations)
871 shader->setAutoMapLocations(true);
872
LoopDawgb22c0692017-12-06 16:52:03 -0700873 if (Options & EOptionInvertY)
874 shader->setInvertY(true);
875
John Kessenich4be4aeb2017-06-23 10:50:22 -0600876 // Set up the environment, some subsettings take precedence over earlier
877 // ways of setting things.
878 if (Options & EOptionSpv) {
879 if (Options & EOptionVulkanRules) {
880 shader->setEnvInput((Options & EOptionReadHlsl) ? glslang::EShSourceHlsl
881 : glslang::EShSourceGlsl,
John Kessenich6353d552017-06-23 11:11:09 -0600882 compUnit.stage, glslang::EShClientVulkan, ClientInputSemanticsVersion);
883 shader->setEnvClient(glslang::EShClientVulkan, VulkanClientVersion);
John Kessenich4be4aeb2017-06-23 10:50:22 -0600884 } else {
885 shader->setEnvInput((Options & EOptionReadHlsl) ? glslang::EShSourceHlsl
886 : glslang::EShSourceGlsl,
John Kessenich6353d552017-06-23 11:11:09 -0600887 compUnit.stage, glslang::EShClientOpenGL, ClientInputSemanticsVersion);
888 shader->setEnvClient(glslang::EShClientOpenGL, OpenGLClientVersion);
John Kessenich4be4aeb2017-06-23 10:50:22 -0600889 }
John Kessenich5d610ee2018-03-07 18:05:55 -0700890 shader->setEnvTarget(glslang::EShTargetSpv, TargetVersion);
891 if (targetHlslFunctionality1)
892 shader->setEnvTargetHlslFunctionality1();
John Kessenich4be4aeb2017-06-23 10:50:22 -0600893 }
894
John Kessenich69f4b512013-09-04 21:19:27 +0000895 shaders.push_back(shader);
John Kessenichb3297152015-07-11 18:01:03 -0600896
John Kessenich4be4aeb2017-06-23 10:50:22 -0600897 const int defaultVersion = Options & EOptionDefaultDesktop ? 110 : 100;
John Kessenich69f4b512013-09-04 21:19:27 +0000898
John Kessenich3494b4d2017-05-22 15:00:42 -0600899 DirStackFileIncluder includer;
John Kessenich971a0a82017-06-07 15:06:58 -0600900 std::for_each(IncludeDirectoryList.rbegin(), IncludeDirectoryList.rend(), [&includer](const std::string& dir) {
901 includer.pushExternalLocalDirectory(dir); });
John Kessenichc555ddd2015-06-17 02:38:44 +0000902 if (Options & EOptionOutputPreprocessed) {
903 std::string str;
Dejan Mircevski7be4b822015-06-17 11:40:33 -0400904 if (shader->preprocess(&Resources, defaultVersion, ENoProfile, false, false,
Andrew Woloszyna132af52016-03-07 13:23:09 -0500905 messages, &str, includer)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400906 PutsIfNonEmpty(str.c_str());
907 } else {
908 CompileFailed = true;
909 }
910 StderrIfNonEmpty(shader->getInfoLog());
911 StderrIfNonEmpty(shader->getInfoDebugLog());
John Kessenichc555ddd2015-06-17 02:38:44 +0000912 continue;
913 }
John Kessenich3494b4d2017-05-22 15:00:42 -0600914 if (! shader->parse(&Resources, defaultVersion, false, messages, includer))
John Kessenichc999ba22013-11-07 23:33:24 +0000915 CompileFailed = true;
John Kessenichc555ddd2015-06-17 02:38:44 +0000916
John Kessenich69f4b512013-09-04 21:19:27 +0000917 program.addShader(shader);
918
John Kessenichc57b2a92016-01-16 15:30:03 -0700919 if (! (Options & EOptionSuppressInfolog) &&
920 ! (Options & EOptionMemoryLeakMode)) {
John Kessenich04acb1b2017-06-14 17:36:50 -0600921 PutsIfNonEmpty(compUnit.fileName[0].c_str());
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400922 PutsIfNonEmpty(shader->getInfoLog());
923 PutsIfNonEmpty(shader->getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000924 }
John Kessenich69f4b512013-09-04 21:19:27 +0000925 }
926
927 //
928 // Program-level processing...
929 //
930
John Kessenich4d65ee32016-03-12 18:17:47 -0700931 // Link
John Kessenich68d78fd2015-07-12 19:28:10 -0600932 if (! (Options & EOptionOutputPreprocessed) && ! program.link(messages))
John Kessenichc999ba22013-11-07 23:33:24 +0000933 LinkFailed = true;
934
steve-lunarg9ae34742016-10-05 13:40:13 -0600935 // Map IO
936 if (Options & EOptionSpv) {
937 if (!program.mapIO())
938 LinkFailed = true;
939 }
John Kessenichecba76f2017-01-06 00:34:48 -0700940
John Kessenich4d65ee32016-03-12 18:17:47 -0700941 // Report
John Kessenichc57b2a92016-01-16 15:30:03 -0700942 if (! (Options & EOptionSuppressInfolog) &&
943 ! (Options & EOptionMemoryLeakMode)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400944 PutsIfNonEmpty(program.getInfoLog());
945 PutsIfNonEmpty(program.getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000946 }
947
John Kessenich4d65ee32016-03-12 18:17:47 -0700948 // Reflect
John Kessenich11f9fc72013-11-07 01:06:34 +0000949 if (Options & EOptionDumpReflection) {
950 program.buildReflection();
951 program.dumpReflection();
952 }
953
John Kessenich4d65ee32016-03-12 18:17:47 -0700954 // Dump SPIR-V
John Kessenich0df0cde2015-03-03 17:09:43 +0000955 if (Options & EOptionSpv) {
John Kessenich92f90382014-07-28 04:21:04 +0000956 if (CompileFailed || LinkFailed)
John Kessenich68d78fd2015-07-12 19:28:10 -0600957 printf("SPIR-V is not generated for failed compile or link\n");
John Kessenich92f90382014-07-28 04:21:04 +0000958 else {
959 for (int stage = 0; stage < EShLangCount; ++stage) {
John Kessenicha7a68a92014-08-24 18:21:00 +0000960 if (program.getIntermediate((EShLanguage)stage)) {
John Kessenich0df0cde2015-03-03 17:09:43 +0000961 std::vector<unsigned int> spirv;
Lei Zhang09caf122016-05-02 18:11:54 -0400962 std::string warningsErrors;
Lei Zhang17535f72016-05-04 15:55:59 -0400963 spv::SpvBuildLogger logger;
John Kessenich121853f2017-05-31 17:11:16 -0600964 glslang::SpvOptions spvOptions;
965 if (Options & EOptionDebug)
966 spvOptions.generateDebugInfo = true;
GregF52fe3d52017-09-28 10:08:32 -0600967 spvOptions.disableOptimizer = (Options & EOptionOptimizeDisable) != 0;
968 spvOptions.optimizeSize = (Options & EOptionOptimizeSize) != 0;
John Kessenich121853f2017-05-31 17:11:16 -0600969 glslang::GlslangToSpv(*program.getIntermediate((EShLanguage)stage), spirv, &logger, &spvOptions);
John Kessenichc57b2a92016-01-16 15:30:03 -0700970
971 // Dump the spv to a file or stdout, etc., but only if not doing
972 // memory/perf testing, as it's not internal to programmatic use.
973 if (! (Options & EOptionMemoryLeakMode)) {
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500974 printf("%s", logger.getAllMessages().c_str());
975 if (Options & EOptionOutputHexadecimal) {
John Kessenich8f674e82017-02-18 09:45:40 -0700976 glslang::OutputSpvHex(spirv, GetBinaryName((EShLanguage)stage), variableName);
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500977 } else {
978 glslang::OutputSpvBin(spirv, GetBinaryName((EShLanguage)stage));
979 }
John Kessenichc57b2a92016-01-16 15:30:03 -0700980 if (Options & EOptionHumanReadableSpv) {
John Kessenichc57b2a92016-01-16 15:30:03 -0700981 spv::Disassemble(std::cout, spirv);
982 }
John Kessenichacba7722015-03-04 03:48:38 +0000983 }
John Kessenicha7a68a92014-08-24 18:21:00 +0000984 }
John Kessenich92f90382014-07-28 04:21:04 +0000985 }
986 }
987 }
988
John Kessenich5b0f13a2013-11-01 03:08:40 +0000989 // Free everything up, program has to go before the shaders
990 // because it might have merged stuff from the shaders, and
991 // the stuff from the shaders has to have its destructors called
992 // before the pools holding the memory in the shaders is freed.
993 delete &program;
John Kessenich69f4b512013-09-04 21:19:27 +0000994 while (shaders.size() > 0) {
995 delete shaders.back();
996 shaders.pop_back();
997 }
John Kessenich69f4b512013-09-04 21:19:27 +0000998}
999
John Kessenichc57b2a92016-01-16 15:30:03 -07001000//
1001// Do file IO part of compile and link, handing off the pure
1002// API/programmatic mode to CompileAndLinkShaderUnits(), which can
1003// be put in a loop for testing memory footprint and performance.
1004//
1005// This is just for linking mode: meaning all the shaders will be put into the
1006// the same program linked together.
1007//
1008// This means there are a limited number of work items (not multi-threading mode)
1009// and that the point is testing at the linking level. Hence, to enable
1010// performance and memory testing, the actual compile/link can be put in
1011// a loop, independent of processing the work items and file IO.
1012//
Juan Lopeza558b262017-04-02 23:04:00 +02001013void CompileAndLinkShaderFiles(glslang::TWorklist& Worklist)
John Kessenichc57b2a92016-01-16 15:30:03 -07001014{
1015 std::vector<ShaderCompUnit> compUnits;
1016
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001017 // If this is using stdin, we can't really detect multiple different file
1018 // units by input type. We need to assume that we're just being given one
1019 // file of a certain type.
1020 if ((Options & EOptionStdin) != 0) {
1021 ShaderCompUnit compUnit(FindLanguage("stdin"));
1022 std::istreambuf_iterator<char> begin(std::cin), end;
1023 std::string tempString(begin, end);
1024 char* fileText = strdup(tempString.c_str());
1025 std::string fileName = "stdin";
1026 compUnit.addString(fileName, fileText);
John Kessenichc57b2a92016-01-16 15:30:03 -07001027 compUnits.push_back(compUnit);
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001028 } else {
1029 // Transfer all the work items from to a simple list of
1030 // of compilation units. (We don't care about the thread
1031 // work-item distribution properties in this path, which
1032 // is okay due to the limited number of shaders, know since
1033 // they are all getting linked together.)
1034 glslang::TWorkItem* workItem;
1035 while (Worklist.remove(workItem)) {
1036 ShaderCompUnit compUnit(FindLanguage(workItem->name));
1037 char* fileText = ReadFileData(workItem->name.c_str());
1038 if (fileText == nullptr)
1039 usage();
1040 compUnit.addString(workItem->name, fileText);
1041 compUnits.push_back(compUnit);
1042 }
John Kessenichc57b2a92016-01-16 15:30:03 -07001043 }
1044
1045 // Actual call to programmatic processing of compile and link,
1046 // in a loop for testing memory and performance. This part contains
1047 // all the perf/memory that a programmatic consumer will care about.
1048 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
1049 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j)
1050 CompileAndLinkShaderUnits(compUnits);
1051
1052 if (Options & EOptionMemoryLeakMode)
1053 glslang::OS_DumpMemoryCounters();
1054 }
1055
John Kessenicha9313662017-06-15 10:40:49 -06001056 // free memory from ReadFileData, which got stored in a const char*
1057 // as the first string above
rdb32084e82016-02-23 22:17:38 +01001058 for (auto it = compUnits.begin(); it != compUnits.end(); ++it)
John Kessenicha9313662017-06-15 10:40:49 -06001059 FreeFileData(const_cast<char*>(it->text[0]));
John Kessenichc57b2a92016-01-16 15:30:03 -07001060}
1061
John Kessenich94f28eb2017-11-13 01:32:06 -07001062int singleMain()
John Kessenicha0af4732012-12-12 21:15:54 +00001063{
Juan Lopeza558b262017-04-02 23:04:00 +02001064 glslang::TWorklist workList;
John Kessenich94f28eb2017-11-13 01:32:06 -07001065 std::for_each(WorkItems.begin(), WorkItems.end(), [&workList](std::unique_ptr<glslang::TWorkItem>& item) {
Juan Lopeza558b262017-04-02 23:04:00 +02001066 assert(item);
1067 workList.add(item.get());
1068 });
John Kessenich2b07c7e2013-07-31 18:44:13 +00001069
John Kessenich05a70632013-09-17 19:26:08 +00001070 if (Options & EOptionDumpConfig) {
Lei Zhang414eb602016-03-04 16:22:34 -05001071 printf("%s", glslang::GetDefaultTBuiltInResourceString().c_str());
Juan Lopeza558b262017-04-02 23:04:00 +02001072 if (workList.empty())
John Kessenich05a70632013-09-17 19:26:08 +00001073 return ESuccess;
1074 }
1075
John Kessenichc6c80a62018-03-05 22:23:17 -07001076 if (Options & EOptionDumpBareVersion) {
1077 printf("%d.%d.%d\n",
1078 glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL);
1079 if (workList.empty())
1080 return ESuccess;
1081 } else if (Options & EOptionDumpVersions) {
1082 printf("Glslang Version: %d.%d.%d\n",
1083 glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL);
John Kessenich319de232013-12-04 04:43:40 +00001084 printf("ESSL Version: %s\n", glslang::GetEsslVersionString());
1085 printf("GLSL Version: %s\n", glslang::GetGlslVersionString());
John Kessenich68d78fd2015-07-12 19:28:10 -06001086 std::string spirvVersion;
1087 glslang::GetSpirvVersion(spirvVersion);
John Kessenich0da9eaa2015-08-01 17:10:02 -06001088 printf("SPIR-V Version %s\n", spirvVersion.c_str());
John Kessenich5e4b1242015-08-06 22:53:06 -06001089 printf("GLSL.std.450 Version %d, Revision %d\n", GLSLstd450Version, GLSLstd450Revision);
John Kessenich55e7d112015-11-15 21:33:39 -07001090 printf("Khronos Tool ID %d\n", glslang::GetKhronosToolId());
John Kessenicha372a3e2017-11-02 22:32:14 -06001091 printf("SPIR-V Generator Version %d\n", glslang::GetSpirvGeneratorVersion());
John Kessenichb84313d2016-07-20 16:03:29 -06001092 printf("GL_KHR_vulkan_glsl version %d\n", 100);
1093 printf("ARB_GL_gl_spirv version %d\n", 100);
Juan Lopeza558b262017-04-02 23:04:00 +02001094 if (workList.empty())
John Kessenich319de232013-12-04 04:43:40 +00001095 return ESuccess;
1096 }
1097
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001098 if (workList.empty() && ((Options & EOptionStdin) == 0)) {
John Kessenich05a70632013-09-17 19:26:08 +00001099 usage();
John Kessenich05a70632013-09-17 19:26:08 +00001100 }
1101
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001102 if (Options & EOptionStdin) {
John Kessenich94f28eb2017-11-13 01:32:06 -07001103 WorkItems.push_back(std::unique_ptr<glslang::TWorkItem>{new glslang::TWorkItem("stdin")});
1104 workList.add(WorkItems.back().get());
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001105 }
1106
John Kessenich05a70632013-09-17 19:26:08 +00001107 ProcessConfigFile();
1108
John Kessenich69f4b512013-09-04 21:19:27 +00001109 //
1110 // Two modes:
John Kessenich38f3b892013-09-06 19:52:57 +00001111 // 1) linking all arguments together, single-threaded, new C++ interface
1112 // 2) independent arguments, can be tackled by multiple asynchronous threads, for testing thread safety, using the old handle interface
John Kessenich69f4b512013-09-04 21:19:27 +00001113 //
John Kessenichc555ddd2015-06-17 02:38:44 +00001114 if (Options & EOptionLinkProgram ||
1115 Options & EOptionOutputPreprocessed) {
John Kessenichc36e1d82013-11-01 17:41:52 +00001116 glslang::InitializeProcess();
John Kesseniche2c15b42017-11-16 22:48:41 -07001117 glslang::InitializeProcess(); // also test reference counting of users
1118 glslang::InitializeProcess(); // also test reference counting of users
1119 glslang::FinalizeProcess(); // also test reference counting of users
1120 glslang::FinalizeProcess(); // also test reference counting of users
Juan Lopeza558b262017-04-02 23:04:00 +02001121 CompileAndLinkShaderFiles(workList);
John Kessenichc36e1d82013-11-01 17:41:52 +00001122 glslang::FinalizeProcess();
1123 } else {
1124 ShInitialize();
John Kesseniche2c15b42017-11-16 22:48:41 -07001125 ShInitialize(); // also test reference counting of users
1126 ShFinalize(); // also test reference counting of users
John Kessenichc36e1d82013-11-01 17:41:52 +00001127
Juan Lopeza558b262017-04-02 23:04:00 +02001128 bool printShaderNames = workList.size() > 1;
John Kessenich69f4b512013-09-04 21:19:27 +00001129
John Kesseniche2c15b42017-11-16 22:48:41 -07001130 if (Options & EOptionMultiThreaded) {
Juan Lopeza558b262017-04-02 23:04:00 +02001131 std::array<std::thread, 16> threads;
John Kesseniche2c15b42017-11-16 22:48:41 -07001132 for (unsigned int t = 0; t < threads.size(); ++t) {
Juan Lopeza558b262017-04-02 23:04:00 +02001133 threads[t] = std::thread(CompileShaders, std::ref(workList));
John Kesseniche2c15b42017-11-16 22:48:41 -07001134 if (threads[t].get_id() == std::thread::id()) {
John Kessenichaf527992017-11-02 22:48:15 -06001135 fprintf(stderr, "Failed to create thread\n");
John Kessenich38f3b892013-09-06 19:52:57 +00001136 return EFailThreadCreate;
1137 }
John Kessenicha0af4732012-12-12 21:15:54 +00001138 }
Juan Lopeza558b262017-04-02 23:04:00 +02001139
1140 std::for_each(threads.begin(), threads.end(), [](std::thread& t) { t.join(); });
John Kessenichc999ba22013-11-07 23:33:24 +00001141 } else
Juan Lopeza558b262017-04-02 23:04:00 +02001142 CompileShaders(workList);
John Kessenich38f3b892013-09-06 19:52:57 +00001143
1144 // Print out all the resulting infologs
John Kessenich94f28eb2017-11-13 01:32:06 -07001145 for (size_t w = 0; w < WorkItems.size(); ++w) {
1146 if (WorkItems[w]) {
1147 if (printShaderNames || WorkItems[w]->results.size() > 0)
1148 PutsIfNonEmpty(WorkItems[w]->name.c_str());
1149 PutsIfNonEmpty(WorkItems[w]->results.c_str());
John Kessenich38f3b892013-09-06 19:52:57 +00001150 }
1151 }
John Kessenichc36e1d82013-11-01 17:41:52 +00001152
1153 ShFinalize();
John Kessenicha0af4732012-12-12 21:15:54 +00001154 }
John Kessenich2b07c7e2013-07-31 18:44:13 +00001155
John Kessenichc999ba22013-11-07 23:33:24 +00001156 if (CompileFailed)
John Kessenicha0af4732012-12-12 21:15:54 +00001157 return EFailCompile;
John Kessenichc999ba22013-11-07 23:33:24 +00001158 if (LinkFailed)
John Kessenicha0af4732012-12-12 21:15:54 +00001159 return EFailLink;
1160
1161 return 0;
1162}
1163
John Kessenich94f28eb2017-11-13 01:32:06 -07001164int C_DECL main(int argc, char* argv[])
1165{
1166 ProcessArguments(WorkItems, argc, argv);
1167
1168 int ret = 0;
1169
1170 // Loop over the entire init/finalize cycle to watch memory changes
1171 const int iterations = 1;
1172 if (iterations > 1)
1173 glslang::OS_DumpMemoryCounters();
1174 for (int i = 0; i < iterations; ++i) {
1175 ret = singleMain();
1176 if (iterations > 1)
1177 glslang::OS_DumpMemoryCounters();
1178 }
1179
1180 return ret;
1181}
1182
John Kessenicha0af4732012-12-12 21:15:54 +00001183//
1184// Deduce the language from the filename. Files must end in one of the
1185// following extensions:
1186//
John Kessenich2b07c7e2013-07-31 18:44:13 +00001187// .vert = vertex
1188// .tesc = tessellation control
1189// .tese = tessellation evaluation
1190// .geom = geometry
1191// .frag = fragment
John Kessenich94a81fb2013-08-31 02:41:30 +00001192// .comp = compute
John Kessenicha0af4732012-12-12 21:15:54 +00001193//
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001194EShLanguage FindLanguage(const std::string& name, bool parseSuffix)
John Kessenicha0af4732012-12-12 21:15:54 +00001195{
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001196 size_t ext = 0;
John Kesseniche751bca2017-03-16 11:20:38 -06001197 std::string suffix;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001198
Dan Bakerc6ede892016-08-11 14:06:06 -04001199 if (shaderStageName)
1200 suffix = shaderStageName;
John Kesseniche751bca2017-03-16 11:20:38 -06001201 else {
1202 // Search for a suffix on a filename: e.g, "myfile.frag". If given
1203 // the suffix directly, we skip looking for the '.'
1204 if (parseSuffix) {
1205 ext = name.rfind('.');
1206 if (ext == std::string::npos) {
1207 usage();
1208 return EShLangVertex;
1209 }
1210 ++ext;
1211 }
1212 suffix = name.substr(ext, std::string::npos);
1213 }
dankbaker45d49bc2016-08-08 21:43:07 -04001214
John Kessenich2b07c7e2013-07-31 18:44:13 +00001215 if (suffix == "vert")
1216 return EShLangVertex;
1217 else if (suffix == "tesc")
1218 return EShLangTessControl;
1219 else if (suffix == "tese")
1220 return EShLangTessEvaluation;
1221 else if (suffix == "geom")
1222 return EShLangGeometry;
1223 else if (suffix == "frag")
1224 return EShLangFragment;
John Kessenichc0275792013-08-09 17:14:49 +00001225 else if (suffix == "comp")
1226 return EShLangCompute;
John Kessenich2b07c7e2013-07-31 18:44:13 +00001227
1228 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +00001229 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +00001230}
1231
John Kessenicha0af4732012-12-12 21:15:54 +00001232//
John Kessenichecba76f2017-01-06 00:34:48 -07001233// Read a file's data into a string, and compile it using the old interface ShCompile,
John Kessenich69f4b512013-09-04 21:19:27 +00001234// for non-linkable results.
John Kessenicha0af4732012-12-12 21:15:54 +00001235//
John Kessenich51cdd902014-02-18 23:37:57 +00001236void CompileFile(const char* fileName, ShHandle compiler)
John Kessenicha0af4732012-12-12 21:15:54 +00001237{
John Kessenichca3457f2015-05-18 01:59:45 +00001238 int ret = 0;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001239 char* shaderString;
1240 if ((Options & EOptionStdin) != 0) {
1241 std::istreambuf_iterator<char> begin(std::cin), end;
1242 std::string tempString(begin, end);
1243 shaderString = strdup(tempString.c_str());
1244 } else {
1245 shaderString = ReadFileData(fileName);
1246 }
John Kessenich41cf6b52013-06-25 18:10:05 +00001247
1248 // move to length-based strings, rather than null-terminated strings
John Kessenich04acb1b2017-06-14 17:36:50 -06001249 int* lengths = new int[1];
1250 lengths[0] = (int)strlen(shaderString);
John Kessenicha0af4732012-12-12 21:15:54 +00001251
John Kessenich52ac67e2013-05-05 23:46:22 +00001252 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +00001253 SetMessageOptions(messages);
John Kessenichecba76f2017-01-06 00:34:48 -07001254
John Kessenicha9313662017-06-15 10:40:49 -06001255 if (UserPreamble.isSet())
1256 Error("-D and -U options require -l (linking)\n");
1257
John Kessenich94a81fb2013-08-31 02:41:30 +00001258 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
1259 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) {
John Kessenich927608b2017-01-06 12:34:14 -07001260 // ret = ShCompile(compiler, shaderStrings, NumShaderStrings, lengths, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich04acb1b2017-06-14 17:36:50 -06001261 ret = ShCompile(compiler, &shaderString, 1, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich927608b2017-01-06 12:34:14 -07001262 // const char* multi[12] = { "# ve", "rsion", " 300 e", "s", "\n#err",
John Kessenichecba76f2017-01-06 00:34:48 -07001263 // "or should be l", "ine 1", "string 5\n", "float glo", "bal",
John Kessenichea869fb2013-10-28 18:12:06 +00001264 // ";\n#error should be line 2\n void main() {", "global = 2.3;}" };
John Kessenich927608b2017-01-06 12:34:14 -07001265 // const char* multi[7] = { "/", "/", "\\", "\n", "\n", "#", "version 300 es" };
1266 // ret = ShCompile(compiler, multi, 7, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich41cf6b52013-06-25 18:10:05 +00001267 }
John Kessenicha0af4732012-12-12 21:15:54 +00001268
John Kessenich94a81fb2013-08-31 02:41:30 +00001269 if (Options & EOptionMemoryLeakMode)
John Kessenich2b07c7e2013-07-31 18:44:13 +00001270 glslang::OS_DumpMemoryCounters();
John Kessenicha0af4732012-12-12 21:15:54 +00001271 }
John Kessenicha0af4732012-12-12 21:15:54 +00001272
John Kessenich41cf6b52013-06-25 18:10:05 +00001273 delete [] lengths;
John Kessenich04acb1b2017-06-14 17:36:50 -06001274 FreeFileData(shaderString);
John Kessenicha0af4732012-12-12 21:15:54 +00001275
John Kessenichc999ba22013-11-07 23:33:24 +00001276 if (ret == 0)
1277 CompileFailed = true;
John Kessenicha0af4732012-12-12 21:15:54 +00001278}
1279
John Kessenicha0af4732012-12-12 21:15:54 +00001280//
1281// print usage to stdout
1282//
1283void usage()
1284{
John Kessenich319de232013-12-04 04:43:40 +00001285 printf("Usage: glslangValidator [option]... [file]...\n"
1286 "\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001287 "'file' can end in .<stage> for auto-stage classification, where <stage> is:\n"
1288 " .conf to provide a config file that replaces the default configuration\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001289 " (see -c option below for generating a template)\n"
1290 " .vert for a vertex shader\n"
1291 " .tesc for a tessellation control shader\n"
1292 " .tese for a tessellation evaluation shader\n"
1293 " .geom for a geometry shader\n"
1294 " .frag for a fragment shader\n"
1295 " .comp for a compute shader\n"
John Kessenich319de232013-12-04 04:43:40 +00001296 "\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001297 "Options:\n"
1298 " -C cascading errors; risk crash from accumulation of error recoveries\n"
1299 " -D input is HLSL\n"
John Kessenicha9313662017-06-15 10:40:49 -06001300 " -D<macro=def>\n"
1301 " -D<macro> define a pre-processor macro\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001302 " -E print pre-processed GLSL; cannot be used with -l;\n"
1303 " errors will appear on stderr.\n"
John Kessenich6353d552017-06-23 11:11:09 -06001304 " -G[ver] create SPIR-V binary, under OpenGL semantics; turns on -l;\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001305 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich6353d552017-06-23 11:11:09 -06001306 " 'ver', when present, is the version of the input semantics,\n"
1307 " which will appear in #define GL_SPIRV ver\n"
1308 " '--client opengl100' is the same as -G100\n"
1309 " a '--target-env' for OpenGL will also imply '-G'\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001310 " -H print human readable form of SPIR-V; turns on -V\n"
John Kessenich971a0a82017-06-07 15:06:58 -06001311 " -I<dir> add dir to the include search path; includer's directory\n"
1312 " is searched first, followed by left-to-right order of -I\n"
GregFcd1f1692017-09-21 18:40:22 -06001313 " -Od disables optimization. May cause illegal SPIR-V for HLSL.\n"
1314 " -Os optimizes SPIR-V to minimize size.\n"
John Kesseniche751bca2017-03-16 11:20:38 -06001315 " -S <stage> uses specified stage rather than parsing the file extension\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001316 " choices for <stage> are vert, tesc, tese, geom, frag, or comp\n"
John Kessenich6353d552017-06-23 11:11:09 -06001317 " -U<macro> undefine a pre-processor macro\n"
1318 " -V[ver] create SPIR-V binary, under Vulkan semantics; turns on -l;\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001319 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich6353d552017-06-23 11:11:09 -06001320 " 'ver', when present, is the version of the input semantics,\n"
1321 " which will appear in #define VULKAN ver\n"
1322 " '--client vulkan100' is the same as -V100\n"
1323 " a '--target-env' for Vulkan will also imply '-V'\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001324 " -c configuration dump;\n"
1325 " creates the default configuration file (redirect to a .conf file)\n"
1326 " -d default to desktop (#version 110) when there is no shader #version\n"
1327 " (default is ES version 100)\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001328 " -e <name> specify <name> as the entry-point name\n"
John Kessenich5d610ee2018-03-07 18:05:55 -07001329 " -f{hlsl_functionality1}\n"
1330 " 'hlsl_functionality1' enables use of the\n"
1331 " SPV_GOOGLE_hlsl_functionality1 extension\n"
John Kessenich121853f2017-05-31 17:11:16 -06001332 " -g generate debug information\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001333 " -h print this usage message\n"
1334 " -i intermediate tree (glslang AST) is printed out\n"
1335 " -l link all input files together to form a single module\n"
1336 " -m memory leak mode\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001337 " -o <file> save binary to <file>, requires a binary option (e.g., -V)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001338 " -q dump reflection query database\n"
John Kessenich2a271162017-07-20 20:00:36 -06001339 " -r synonym for --relaxed-errors\n"
John Kessenichb63f4a32017-11-16 22:32:20 -07001340 " -s silence syntax and semantic error reporting\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001341 " -t multi-threaded mode\n"
1342 " -v print version strings\n"
John Kessenich2a271162017-07-20 20:00:36 -06001343 " -w synonym for --suppress-warnings\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001344 " -x save binary output as text-based 32-bit hexadecimal numbers\n"
1345 " --auto-map-bindings automatically bind uniform variables\n"
1346 " without explicit bindings.\n"
1347 " --amb synonym for --auto-map-bindings\n"
1348 " --auto-map-locations automatically locate input/output lacking\n"
John Kessenichc178f0a2017-06-23 10:58:31 -06001349 " 'location' (fragile, not cross stage)\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001350 " --aml synonym for --auto-map-locations\n"
John Kessenich6353d552017-06-23 11:11:09 -06001351 " --client {vulkan<ver>|opengl<ver>} see -V and -G\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001352 " -dumpfullversion print bare major.minor.patchlevel\n"
1353 " -dumpversion same as -dumpfullversion\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001354 " --flatten-uniform-arrays flatten uniform texture/sampler arrays to\n"
1355 " scalars\n"
1356 " --fua synonym for --flatten-uniform-arrays\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001357 " --hlsl-offsets Allow block offsets to follow HLSL rules\n"
1358 " Works independently of source language\n"
1359 " --hlsl-iomap Perform IO mapping in HLSL register space\n"
Rex Xucb61eec2018-03-07 13:10:01 +08001360 " --hlsl-enable-16bit-types Allow use of 16-bit types in SPIR-V for HLSL\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001361 " --invert-y | --iy invert position.Y output in vertex shader\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001362 " --keep-uncalled don't eliminate uncalled functions\n"
1363 " --ku synonym for --keep-uncalled\n"
1364 " --no-storage-format use Unknown image format\n"
1365 " --nsf synonym for --no-storage-format\n"
John Kessenich2a271162017-07-20 20:00:36 -06001366 " --relaxed-errors relaxed GLSL semantic error-checking mode\n"
LoopDawg52017192017-07-14 15:15:47 -06001367 " --resource-set-binding [stage] name set binding\n"
1368 " Set descriptor set and binding for individual resources\n"
1369 " --resource-set-binding [stage] set\n"
1370 " Set descriptor set for all resources\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001371 " --rsb [stage] type set binding synonym for --resource-set-binding\n"
1372 " --shift-image-binding [stage] num base binding number for images (uav)\n"
LoopDawge5709552017-10-21 10:46:39 -06001373 " --shift-image-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001374 " --sib [stage] num synonym for --shift-image-binding\n"
1375 " --shift-sampler-binding [stage] num base binding number for samplers\n"
LoopDawge5709552017-10-21 10:46:39 -06001376 " --shift-sampler-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001377 " --ssb [stage] num synonym for --shift-sampler-binding\n"
1378 " --shift-ssbo-binding [stage] num base binding number for SSBOs\n"
LoopDawge5709552017-10-21 10:46:39 -06001379 " --shift-ssbo-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001380 " --sbb [stage] num synonym for --shift-ssbo-binding\n"
1381 " --shift-texture-binding [stage] num base binding number for textures\n"
LoopDawge5709552017-10-21 10:46:39 -06001382 " --shift-texture-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001383 " --stb [stage] num synonym for --shift-texture-binding\n"
1384 " --shift-uav-binding [stage] num base binding number for UAVs\n"
LoopDawge5709552017-10-21 10:46:39 -06001385 " --shift-uav-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001386 " --suavb [stage] num synonym for --shift-uav-binding\n"
1387 " --shift-UBO-binding [stage] num base binding number for UBOs\n"
LoopDawge5709552017-10-21 10:46:39 -06001388 " --shift-UBO-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001389 " --shift-cbuffer-binding [stage] num synonym for --shift-UBO-binding\n"
LoopDawge5709552017-10-21 10:46:39 -06001390 " --shift-cbuffer-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001391 " --sub [stage] num synonym for --shift-UBO-binding\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001392 " --source-entrypoint <name> the given shader source function is\n"
1393 " renamed to be the <name> given in -e\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001394 " --sep synonym for --source-entrypoint\n"
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001395 " --stdin Read from stdin instead of from a file.\n"
1396 " You'll have to provide the shader stage\n"
1397 " using -S.\n"
John Kessenich2a271162017-07-20 20:00:36 -06001398 " --suppress-warnings suppress GLSL warnings\n"
1399 " (except as required by #extension : warn)\n"
John Kessenich66011cb2018-03-06 16:12:04 -07001400 " --target-env {vulkan1.0 | vulkan1.1 | opengl} \n"
1401 " set execution environment that emitted code\n"
1402 " will execute in (as opposed to the language\n"
John Kessenich2a271162017-07-20 20:00:36 -06001403 " semantics selected by --client) defaults:\n"
John Kessenich5d610ee2018-03-07 18:05:55 -07001404 " 'vulkan1.0' under '--client vulkan<ver>'\n"
1405 " 'opengl' under '--client opengl<ver>'\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001406 " --variable-name <name> Creates a C header file that contains a\n"
1407 " uint32_t array named <name>\n"
1408 " initialized with the shader binary code.\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001409 " --version synonym for -v\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001410 " --vn <name> synonym for --variable-name <name>\n"
John Kessenichb0a7eb52013-11-07 17:44:20 +00001411 );
John Kessenich68d78fd2015-07-12 19:28:10 -06001412
1413 exit(EFailUsage);
John Kessenicha0af4732012-12-12 21:15:54 +00001414}
1415
John Kessenich3ce4e592014-10-06 19:57:34 +00001416#if !defined _MSC_VER && !defined MINGW_HAS_SECURE_API
John Kessenichcfd643e2013-03-08 23:14:42 +00001417
1418#include <errno.h>
1419
1420int fopen_s(
1421 FILE** pFile,
John Kessenich51cdd902014-02-18 23:37:57 +00001422 const char* filename,
1423 const char* mode
John Kessenichcfd643e2013-03-08 23:14:42 +00001424)
1425{
1426 if (!pFile || !filename || !mode) {
1427 return EINVAL;
1428 }
1429
1430 FILE* f = fopen(filename, mode);
1431 if (! f) {
1432 if (errno != 0) {
1433 return errno;
1434 } else {
1435 return ENOENT;
1436 }
1437 }
1438 *pFile = f;
1439
1440 return 0;
1441}
1442
1443#endif
1444
John Kessenicha0af4732012-12-12 21:15:54 +00001445//
1446// Malloc a string of sufficient size and read a string into it.
1447//
John Kessenich04acb1b2017-06-14 17:36:50 -06001448char* ReadFileData(const char* fileName)
John Kessenicha0af4732012-12-12 21:15:54 +00001449{
John Kessenichb3297152015-07-11 18:01:03 -06001450 FILE *in = nullptr;
John Kessenich3ce4e592014-10-06 19:57:34 +00001451 int errorCode = fopen_s(&in, fileName, "r");
John Kessenich68d78fd2015-07-12 19:28:10 -06001452 if (errorCode || in == nullptr)
1453 Error("unable to open input file");
John Kessenichecba76f2017-01-06 00:34:48 -07001454
John Kessenich04acb1b2017-06-14 17:36:50 -06001455 int count = 0;
John Kessenicha0af4732012-12-12 21:15:54 +00001456 while (fgetc(in) != EOF)
1457 count++;
1458
John Kessenichd6c72a42014-08-18 19:42:35 +00001459 fseek(in, 0, SEEK_SET);
John Kessenichca3457f2015-05-18 01:59:45 +00001460
John Kessenich04acb1b2017-06-14 17:36:50 -06001461 char* return_data = (char*)malloc(count + 1); // freed in FreeFileData()
1462 if ((int)fread(return_data, 1, count, in) != count) {
1463 free(return_data);
John Kessenich68d78fd2015-07-12 19:28:10 -06001464 Error("can't read input file");
John Kessenicha0af4732012-12-12 21:15:54 +00001465 }
John Kessenich68d78fd2015-07-12 19:28:10 -06001466
John Kessenich04acb1b2017-06-14 17:36:50 -06001467 return_data[count] = '\0';
John Kessenicha0af4732012-12-12 21:15:54 +00001468 fclose(in);
John Kessenichb3297152015-07-11 18:01:03 -06001469
John Kessenicha0af4732012-12-12 21:15:54 +00001470 return return_data;
1471}
1472
John Kessenich04acb1b2017-06-14 17:36:50 -06001473void FreeFileData(char* data)
John Kessenicha0af4732012-12-12 21:15:54 +00001474{
John Kessenichb3297152015-07-11 18:01:03 -06001475 free(data);
John Kessenicha0af4732012-12-12 21:15:54 +00001476}
1477
John Kessenich54d8cda2013-02-11 22:36:01 +00001478void InfoLogMsg(const char* msg, const char* name, const int num)
John Kessenicha0af4732012-12-12 21:15:54 +00001479{
John Kessenichfae38ee2015-06-10 23:23:12 +00001480 if (num >= 0 )
1481 printf("#### %s %s %d INFO LOG ####\n", msg, name, num);
1482 else
1483 printf("#### %s %s INFO LOG ####\n", msg, name);
John Kessenicha0af4732012-12-12 21:15:54 +00001484}