blob: e96680d531cab75940796321b9d23c43177f3510 [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};
104
John Kessenicha0af4732012-12-12 21:15:54 +0000105//
John Kessenich68d78fd2015-07-12 19:28:10 -0600106// Return codes from main/exit().
John Kessenicha0af4732012-12-12 21:15:54 +0000107//
108enum TFailCode {
109 ESuccess = 0,
110 EFailUsage,
111 EFailCompile,
112 EFailLink,
113 EFailCompilerCreate,
John Kessenich2b07c7e2013-07-31 18:44:13 +0000114 EFailThreadCreate,
John Kessenicha0af4732012-12-12 21:15:54 +0000115 EFailLinkerCreate
116};
117
118//
John Kessenich68d78fd2015-07-12 19:28:10 -0600119// Forward declarations.
John Kessenicha0af4732012-12-12 21:15:54 +0000120//
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600121EShLanguage FindLanguage(const std::string& name, bool parseSuffix=true);
John Kessenich51cdd902014-02-18 23:37:57 +0000122void CompileFile(const char* fileName, ShHandle);
John Kessenicha0af4732012-12-12 21:15:54 +0000123void usage();
John Kessenich04acb1b2017-06-14 17:36:50 -0600124char* ReadFileData(const char* fileName);
125void FreeFileData(char* data);
John Kessenich54d8cda2013-02-11 22:36:01 +0000126void InfoLogMsg(const char* msg, const char* name, const int num);
John Kessenich41cf6b52013-06-25 18:10:05 +0000127
John Kessenichc999ba22013-11-07 23:33:24 +0000128// Globally track if any compile or link failure.
129bool CompileFailed = false;
130bool LinkFailed = false;
131
John Kessenich94f28eb2017-11-13 01:32:06 -0700132// array of unique places to leave the shader names and infologs for the asynchronous compiles
133std::vector<std::unique_ptr<glslang::TWorkItem>> WorkItems;
134
John Kessenich05a70632013-09-17 19:26:08 +0000135TBuiltInResource Resources;
136std::string ConfigFile;
137
John Kessenicha0af4732012-12-12 21:15:54 +0000138//
John Kessenichf0bcb0a2016-04-02 13:09:14 -0600139// Parse either a .conf file provided by the user or the default from glslang::DefaultTBuiltInResource
John Kessenich05a70632013-09-17 19:26:08 +0000140//
141void ProcessConfigFile()
John Kessenichb51f62c2013-04-11 16:31:09 +0000142{
John Kessenich04acb1b2017-06-14 17:36:50 -0600143 if (ConfigFile.size() == 0)
Lei Zhang414eb602016-03-04 16:22:34 -0500144 Resources = glslang::DefaultTBuiltInResource;
John Kessenich04acb1b2017-06-14 17:36:50 -0600145 else {
146 char* configString = ReadFileData(ConfigFile.c_str());
147 glslang::DecodeResourceLimits(&Resources, configString);
148 FreeFileData(configString);
John Kessenich05a70632013-09-17 19:26:08 +0000149 }
John Kessenicha0af4732012-12-12 21:15:54 +0000150}
151
John Kessenich94a81fb2013-08-31 02:41:30 +0000152int Options = 0;
John Kessenich68d78fd2015-07-12 19:28:10 -0600153const char* ExecutableName = nullptr;
154const char* binaryFileName = nullptr;
John Kessenich4d65ee32016-03-12 18:17:47 -0700155const char* entryPointName = nullptr;
steve-lunargf1e0c872016-10-31 15:13:43 -0600156const char* sourceEntryPointName = nullptr;
Dan Bakerc6ede892016-08-11 14:06:06 -0400157const char* shaderStageName = nullptr;
Flavioaea3c892017-02-06 11:46:35 -0800158const char* variableName = nullptr;
John Kessenich971a0a82017-06-07 15:06:58 -0600159std::vector<std::string> IncludeDirectoryList;
John Kessenich66011cb2018-03-06 16:12:04 -0700160int ClientInputSemanticsVersion = 100; // maps to, say, #define VULKAN 100
161glslang::EshTargetClientVersion VulkanClientVersion =
162 glslang::EShTargetVulkan_1_0; // would map to, say, Vulkan 1.0
163glslang::EshTargetClientVersion OpenGLClientVersion =
164 glslang::EShTargetOpenGL_450; // doesn't influence anything yet, but maps to OpenGL 4.50
165glslang::EShTargetLanguageVersion TargetVersion =
166 glslang::EShTargetSpv_1_0; // maps to, say, SPIR-V 1.0
167std::vector<std::string> Processes; // what should be recorded by OpModuleProcessed, or equivalent
John Kessenich68d78fd2015-07-12 19:28:10 -0600168
LoopDawg08a14422017-10-17 19:27:14 -0600169// Per descriptor-set binding base data
170typedef std::map<unsigned int, unsigned int> TPerSetBaseBinding;
171
172std::array<std::array<unsigned int, EShLangCount>, glslang::EResCount> baseBinding;
173std::array<std::array<TPerSetBaseBinding, EShLangCount>, glslang::EResCount> baseBindingForSet;
Hyangran Park36dc8292017-05-02 16:27:29 +0900174std::array<std::vector<std::string>, EShLangCount> baseResourceSetBinding;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600175
John Kessenicha9313662017-06-15 10:40:49 -0600176// Add things like "#define ..." to a preamble to use in the beginning of the shader.
177class TPreamble {
178public:
179 TPreamble() { }
180
181 bool isSet() const { return text.size() > 0; }
182 const char* get() const { return text.c_str(); }
183
184 // #define...
185 void addDef(std::string def)
186 {
187 text.append("#define ");
188 fixLine(def);
189
John Kessenich2a271162017-07-20 20:00:36 -0600190 Processes.push_back("D");
191 Processes.back().append(def);
192
John Kessenicha9313662017-06-15 10:40:49 -0600193 // The first "=" needs to turn into a space
LoopDawgb97b25e2017-07-12 09:04:39 -0600194 const size_t equal = def.find_first_of("=");
John Kessenicha9313662017-06-15 10:40:49 -0600195 if (equal != def.npos)
196 def[equal] = ' ';
197
198 text.append(def);
199 text.append("\n");
200 }
201
202 // #undef...
203 void addUndef(std::string undef)
204 {
205 text.append("#undef ");
206 fixLine(undef);
John Kessenich2a271162017-07-20 20:00:36 -0600207
208 Processes.push_back("U");
209 Processes.back().append(undef);
210
John Kessenicha9313662017-06-15 10:40:49 -0600211 text.append(undef);
212 text.append("\n");
213 }
214
215protected:
216 void fixLine(std::string& line)
217 {
218 // Can't go past a newline in the line
LoopDawgb97b25e2017-07-12 09:04:39 -0600219 const size_t end = line.find_first_of("\n");
John Kessenicha9313662017-06-15 10:40:49 -0600220 if (end != line.npos)
221 line = line.substr(0, end);
222 }
223
224 std::string text; // contents of preamble
225};
226
227TPreamble UserPreamble;
228
John Kessenich68d78fd2015-07-12 19:28:10 -0600229//
230// Create the default name for saving a binary if -o is not provided.
231//
232const char* GetBinaryName(EShLanguage stage)
233{
234 const char* name;
235 if (binaryFileName == nullptr) {
236 switch (stage) {
237 case EShLangVertex: name = "vert.spv"; break;
238 case EShLangTessControl: name = "tesc.spv"; break;
239 case EShLangTessEvaluation: name = "tese.spv"; break;
240 case EShLangGeometry: name = "geom.spv"; break;
241 case EShLangFragment: name = "frag.spv"; break;
242 case EShLangCompute: name = "comp.spv"; break;
243 default: name = "unknown"; break;
244 }
245 } else
246 name = binaryFileName;
247
248 return name;
249}
John Kessenich2b07c7e2013-07-31 18:44:13 +0000250
John Kessenich05a70632013-09-17 19:26:08 +0000251//
252// *.conf => this is a config file that can set limits/resources
253//
254bool SetConfigFile(const std::string& name)
255{
256 if (name.size() < 5)
257 return false;
258
John Kessenich4c706852013-10-11 16:28:43 +0000259 if (name.compare(name.size() - 5, 5, ".conf") == 0) {
John Kessenich05a70632013-09-17 19:26:08 +0000260 ConfigFile = name;
261 return true;
262 }
263
264 return false;
265}
266
John Kessenich68d78fd2015-07-12 19:28:10 -0600267//
268// Give error and exit with failure code.
269//
270void Error(const char* message)
271{
John Kessenichaf527992017-11-02 22:48:15 -0600272 fprintf(stderr, "%s: Error %s (use -h for usage)\n", ExecutableName, message);
John Kessenich68d78fd2015-07-12 19:28:10 -0600273 exit(EFailUsage);
274}
275
276//
LoopDawg08a14422017-10-17 19:27:14 -0600277// Process an optional binding base of one the forms:
278// --argname [stage] base // base for stage (if given) or all stages (if not)
LoopDawge5709552017-10-21 10:46:39 -0600279// --argname [stage] [base set]... // set/base pairs: set the base for given binding set.
LoopDawg08a14422017-10-17 19:27:14 -0600280
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600281// Where stage is one of the forms accepted by FindLanguage, and base is an integer
282//
LoopDawg08a14422017-10-17 19:27:14 -0600283void ProcessBindingBase(int& argc, char**& argv, glslang::TResourceType res)
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600284{
285 if (argc < 2)
286 usage();
287
LoopDawg08a14422017-10-17 19:27:14 -0600288 EShLanguage lang = EShLangCount;
289 int singleBase = 0;
290 TPerSetBaseBinding perSetBase;
291 int arg = 1;
292
293 // Parse stage, if given
294 if (!isdigit(argv[arg][0])) {
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600295 if (argc < 3) // this form needs one more argument
296 usage();
John Kessenichecba76f2017-01-06 00:34:48 -0700297
LoopDawg08a14422017-10-17 19:27:14 -0600298 lang = FindLanguage(argv[arg++], false);
299 }
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600300
LoopDawg08a14422017-10-17 19:27:14 -0600301 if ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) {
302 // Parse a per-set binding base
303 while ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) {
LoopDawg08a14422017-10-17 19:27:14 -0600304 const int baseNum = atoi(argv[arg++]);
LoopDawge5709552017-10-21 10:46:39 -0600305 const int setNum = atoi(argv[arg++]);
LoopDawg08a14422017-10-17 19:27:14 -0600306 perSetBase[setNum] = baseNum;
307 }
308 } else {
309 // Parse single binding base
310 singleBase = atoi(argv[arg++]);
311 }
312
313 argc -= (arg-1);
314 argv += (arg-1);
315
316 // Set one or all languages
317 const int langMin = (lang < EShLangCount) ? lang+0 : 0;
318 const int langMax = (lang < EShLangCount) ? lang+1 : EShLangCount;
319
320 for (int lang = langMin; lang < langMax; ++lang) {
321 if (!perSetBase.empty())
322 baseBindingForSet[res][lang] = perSetBase;
323 else
324 baseBinding[res][lang] = singleBase;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600325 }
326}
327
Hyangran Park36dc8292017-05-02 16:27:29 +0900328void ProcessResourceSetBindingBase(int& argc, char**& argv, std::array<std::vector<std::string>, EShLangCount>& base)
329{
330 if (argc < 2)
331 usage();
332
333 if (!isdigit(argv[1][0])) {
LoopDawg52017192017-07-14 15:15:47 -0600334 if (argc < 3) // this form needs one more argument
Hyangran Park36dc8292017-05-02 16:27:29 +0900335 usage();
336
LoopDawg52017192017-07-14 15:15:47 -0600337 // Parse form: --argname stage [regname set base...], or:
338 // --argname stage set
Hyangran Park36dc8292017-05-02 16:27:29 +0900339 const EShLanguage lang = FindLanguage(argv[1], false);
340
LoopDawg52017192017-07-14 15:15:47 -0600341 argc--;
342 argv++;
343
344 while (argc > 1 && argv[1] != nullptr && argv[1][0] != '-') {
345 base[lang].push_back(argv[1]);
346
347 argc--;
348 argv++;
Hyangran Park36dc8292017-05-02 16:27:29 +0900349 }
LoopDawg52017192017-07-14 15:15:47 -0600350
351 // Must have one arg, or a multiple of three (for [regname set binding] triples)
352 if (base[lang].size() != 1 && (base[lang].size() % 3) != 0)
353 usage();
354
Hyangran Park36dc8292017-05-02 16:27:29 +0900355 } else {
LoopDawg52017192017-07-14 15:15:47 -0600356 // Parse form: --argname set
Hyangran Park36dc8292017-05-02 16:27:29 +0900357 for (int lang=0; lang<EShLangCount; ++lang)
358 base[lang].push_back(argv[1]);
359
360 argc--;
361 argv++;
362 }
363}
364
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600365//
John Kessenich68d78fd2015-07-12 19:28:10 -0600366// Do all command-line argument parsing. This includes building up the work-items
367// to be processed later, and saving all the command-line options.
368//
369// Does not return (it exits) if command-line is fatally flawed.
370//
Juan Lopeza558b262017-04-02 23:04:00 +0200371void ProcessArguments(std::vector<std::unique_ptr<glslang::TWorkItem>>& workItems, int argc, char* argv[])
John Kessenich2b07c7e2013-07-31 18:44:13 +0000372{
LoopDawg08a14422017-10-17 19:27:14 -0600373 for (int res = 0; res < glslang::EResCount; ++res)
374 baseBinding[res].fill(0);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600375
John Kessenich38f3b892013-09-06 19:52:57 +0000376 ExecutableName = argv[0];
Juan Lopeza558b262017-04-02 23:04:00 +0200377 workItems.reserve(argc);
John Kessenich38f3b892013-09-06 19:52:57 +0000378
John Kessenichc178f0a2017-06-23 10:58:31 -0600379 const auto bumpArg = [&]() {
380 if (argc > 0) {
381 argc--;
382 argv++;
383 }
384 };
385
386 // read a string directly attached to a single-letter option
John Kessenich6263fb12017-06-14 15:52:44 -0600387 const auto getStringOperand = [&](const char* desc) {
388 if (argv[0][2] == 0) {
389 printf("%s must immediately follow option (no spaces)\n", desc);
390 exit(EFailUsage);
391 }
392 return argv[0] + 2;
393 };
394
John Kessenich6353d552017-06-23 11:11:09 -0600395 // read a number attached to a single-letter option
396 const auto getAttachedNumber = [&](const char* desc) {
397 int num = atoi(argv[0] + 2);
398 if (num == 0) {
399 printf("%s: expected attached non-0 number\n", desc);
400 exit(EFailUsage);
401 }
402 return num;
403 };
404
405 // minimum needed (without overriding something else) to target Vulkan SPIR-V
406 const auto setVulkanSpv = []() {
407 Options |= EOptionSpv;
408 Options |= EOptionVulkanRules;
409 Options |= EOptionLinkProgram;
410 };
411
412 // minimum needed (without overriding something else) to target OpenGL SPIR-V
413 const auto setOpenGlSpv = []() {
414 Options |= EOptionSpv;
415 Options |= EOptionLinkProgram;
416 // undo a -H default to Vulkan
417 Options &= ~EOptionVulkanRules;
418 };
419
John Kessenichc178f0a2017-06-23 10:58:31 -0600420 for (bumpArg(); argc >= 1; bumpArg()) {
John Kessenich2b07c7e2013-07-31 18:44:13 +0000421 if (argv[0][0] == '-') {
John Kessenich2aa7f3a2015-05-15 16:02:07 +0000422 switch (argv[0][1]) {
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600423 case '-':
424 {
425 std::string lowerword(argv[0]+2);
426 std::transform(lowerword.begin(), lowerword.end(), lowerword.begin(), ::tolower);
427
428 // handle --word style options
John Kessenich6263fb12017-06-14 15:52:44 -0600429 if (lowerword == "auto-map-bindings" || // synonyms
John Kessenich6353d552017-06-23 11:11:09 -0600430 lowerword == "auto-map-binding" ||
431 lowerword == "amb") {
John Kessenich6263fb12017-06-14 15:52:44 -0600432 Options |= EOptionAutoMapBindings;
433 } else if (lowerword == "auto-map-locations" || // synonyms
434 lowerword == "aml") {
435 Options |= EOptionAutoMapLocations;
John Kessenich6353d552017-06-23 11:11:09 -0600436 } else if (lowerword == "client") {
437 if (argc > 1) {
438 if (strcmp(argv[1], "vulkan100") == 0)
439 setVulkanSpv();
440 else if (strcmp(argv[1], "opengl100") == 0)
441 setOpenGlSpv();
442 else
443 Error("--client expects vulkan100 or opengl100");
444 }
445 bumpArg();
John Kessenich6263fb12017-06-14 15:52:44 -0600446 } else if (lowerword == "flatten-uniform-arrays" || // synonyms
447 lowerword == "flatten-uniform-array" ||
448 lowerword == "fua") {
449 Options |= EOptionFlattenUniformArrays;
450 } else if (lowerword == "hlsl-offsets") {
451 Options |= EOptionHlslOffsets;
452 } else if (lowerword == "hlsl-iomap" ||
453 lowerword == "hlsl-iomapper" ||
454 lowerword == "hlsl-iomapping") {
455 Options |= EOptionHlslIoMapping;
John Kessenichc6c80a62018-03-05 22:23:17 -0700456 } else if (lowerword == "invert-y" || // synonyms
457 lowerword == "iy") {
458 Options |= EOptionInvertY;
John Kessenich6263fb12017-06-14 15:52:44 -0600459 } else if (lowerword == "keep-uncalled" || // synonyms
460 lowerword == "ku") {
461 Options |= EOptionKeepUncalled;
462 } else if (lowerword == "no-storage-format" || // synonyms
463 lowerword == "nsf") {
464 Options |= EOptionNoStorageFormat;
John Kessenich2a271162017-07-20 20:00:36 -0600465 } else if (lowerword == "relaxed-errors") {
466 Options |= EOptionRelaxedErrors;
John Kessenich6263fb12017-06-14 15:52:44 -0600467 } else if (lowerword == "resource-set-bindings" || // synonyms
468 lowerword == "resource-set-binding" ||
469 lowerword == "rsb") {
470 ProcessResourceSetBindingBase(argc, argv, baseResourceSetBinding);
steve-lunarg9088be42016-11-01 10:31:42 -0600471 } else if (lowerword == "shift-image-bindings" || // synonyms
472 lowerword == "shift-image-binding" ||
473 lowerword == "sib") {
LoopDawg08a14422017-10-17 19:27:14 -0600474 ProcessBindingBase(argc, argv, glslang::EResImage);
John Kessenich6263fb12017-06-14 15:52:44 -0600475 } else if (lowerword == "shift-sampler-bindings" || // synonyms
476 lowerword == "shift-sampler-binding" ||
477 lowerword == "ssb") {
LoopDawg08a14422017-10-17 19:27:14 -0600478 ProcessBindingBase(argc, argv, glslang::EResSampler);
John Kessenich6263fb12017-06-14 15:52:44 -0600479 } else if (lowerword == "shift-uav-bindings" || // synonyms
480 lowerword == "shift-uav-binding" ||
481 lowerword == "suavb") {
LoopDawg08a14422017-10-17 19:27:14 -0600482 ProcessBindingBase(argc, argv, glslang::EResUav);
John Kessenich6263fb12017-06-14 15:52:44 -0600483 } else if (lowerword == "shift-texture-bindings" || // synonyms
484 lowerword == "shift-texture-binding" ||
485 lowerword == "stb") {
LoopDawg08a14422017-10-17 19:27:14 -0600486 ProcessBindingBase(argc, argv, glslang::EResTexture);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600487 } else if (lowerword == "shift-ubo-bindings" || // synonyms
488 lowerword == "shift-ubo-binding" ||
steve-lunargbe283552017-04-18 12:18:01 -0600489 lowerword == "shift-cbuffer-bindings" ||
490 lowerword == "shift-cbuffer-binding" ||
491 lowerword == "sub" ||
492 lowerword == "scb") {
LoopDawg08a14422017-10-17 19:27:14 -0600493 ProcessBindingBase(argc, argv, glslang::EResUbo);
steve-lunarg932bb5c2017-02-21 17:19:08 -0700494 } else if (lowerword == "shift-ssbo-bindings" || // synonyms
495 lowerword == "shift-ssbo-binding" ||
496 lowerword == "sbb") {
LoopDawg08a14422017-10-17 19:27:14 -0600497 ProcessBindingBase(argc, argv, glslang::EResSsbo);
John Kessenich6263fb12017-06-14 15:52:44 -0600498 } else if (lowerword == "source-entrypoint" || // synonyms
499 lowerword == "sep") {
John Kessenichc178f0a2017-06-23 10:58:31 -0600500 if (argc <= 1)
John Kessenich6263fb12017-06-14 15:52:44 -0600501 Error("no <entry-point> provided for --source-entrypoint");
John Kessenichc178f0a2017-06-23 10:58:31 -0600502 sourceEntryPointName = argv[1];
503 bumpArg();
John Kessenich6263fb12017-06-14 15:52:44 -0600504 break;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200505 } else if (lowerword == "stdin") {
506 Options |= EOptionStdin;
507 shaderStageName = argv[1];
John Kessenich2a271162017-07-20 20:00:36 -0600508 } else if (lowerword == "suppress-warnings") {
509 Options |= EOptionSuppressWarnings;
John Kessenich6353d552017-06-23 11:11:09 -0600510 } else if (lowerword == "target-env") {
511 if (argc > 1) {
512 if (strcmp(argv[1], "vulkan1.0") == 0) {
513 setVulkanSpv();
John Kessenich66011cb2018-03-06 16:12:04 -0700514 VulkanClientVersion = glslang::EShTargetVulkan_1_0;
515 } else if (strcmp(argv[1], "vulkan1.1") == 0) {
516 setVulkanSpv();
517 TargetVersion = glslang::EShTargetSpv_1_3;
518 VulkanClientVersion = glslang::EShTargetVulkan_1_1;
John Kessenich6353d552017-06-23 11:11:09 -0600519 } else if (strcmp(argv[1], "opengl") == 0) {
520 setOpenGlSpv();
John Kessenich66011cb2018-03-06 16:12:04 -0700521 OpenGLClientVersion = glslang::EShTargetOpenGL_450;
John Kessenich6353d552017-06-23 11:11:09 -0600522 } else
523 Error("--target-env expected vulkan1.0 or opengl");
524 }
525 bumpArg();
Flavio15017db2017-02-15 14:29:33 -0800526 } else if (lowerword == "variable-name" || // synonyms
John Kessenich66011cb2018-03-06 16:12:04 -0700527 lowerword == "vn") {
Flavio15017db2017-02-15 14:29:33 -0800528 Options |= EOptionOutputHexadecimal;
John Kessenichc178f0a2017-06-23 10:58:31 -0600529 if (argc <= 1)
Flavio15017db2017-02-15 14:29:33 -0800530 Error("no <C-variable-name> provided for --variable-name");
John Kessenichc178f0a2017-06-23 10:58:31 -0600531 variableName = argv[1];
532 bumpArg();
Flavio15017db2017-02-15 14:29:33 -0800533 break;
John Kessenichc6c80a62018-03-05 22:23:17 -0700534 } else if (lowerword == "version") {
535 Options |= EOptionDumpVersions;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600536 } else {
537 usage();
538 }
539 }
540 break;
John Kessenich6263fb12017-06-14 15:52:44 -0600541 case 'C':
542 Options |= EOptionCascadingErrors;
543 break;
544 case 'D':
John Kessenicha9313662017-06-15 10:40:49 -0600545 if (argv[0][2] == 0)
546 Options |= EOptionReadHlsl;
547 else
548 UserPreamble.addDef(getStringOperand("-D<macro> macro name"));
John Kessenich6263fb12017-06-14 15:52:44 -0600549 break;
550 case 'E':
551 Options |= EOptionOutputPreprocessed;
552 break;
553 case 'G':
John Kessenich6353d552017-06-23 11:11:09 -0600554 // OpenGL Client
555 setOpenGlSpv();
556 if (argv[0][2] != 0)
557 ClientInputSemanticsVersion = getAttachedNumber("-G<num> client input semantics");
John Kessenich6263fb12017-06-14 15:52:44 -0600558 break;
John Kessenich2aa7f3a2015-05-15 16:02:07 +0000559 case 'H':
560 Options |= EOptionHumanReadableSpv;
John Kessenich91e4aa52016-07-07 17:46:42 -0600561 if ((Options & EOptionSpv) == 0) {
562 // default to Vulkan
John Kessenich6353d552017-06-23 11:11:09 -0600563 setVulkanSpv();
John Kessenich91e4aa52016-07-07 17:46:42 -0600564 }
565 break;
John Kessenich971a0a82017-06-07 15:06:58 -0600566 case 'I':
John Kessenicha9313662017-06-15 10:40:49 -0600567 IncludeDirectoryList.push_back(getStringOperand("-I<dir> include path"));
John Kessenich68d78fd2015-07-12 19:28:10 -0600568 break;
GregFcd1f1692017-09-21 18:40:22 -0600569 case 'O':
570 if (argv[0][2] == 'd')
571 Options |= EOptionOptimizeDisable;
572 else if (argv[0][2] == 's')
573#ifdef ENABLE_OPT
574 Options |= EOptionOptimizeSize;
575#else
576 Error("-Os not available; optimizer not linked");
577#endif
578 else
579 Error("unknown -O option");
580 break;
Dan Baker5afdd782016-08-11 17:53:57 -0400581 case 'S':
John Kessenichc178f0a2017-06-23 10:58:31 -0600582 if (argc <= 1)
Dan Baker5afdd782016-08-11 17:53:57 -0400583 Error("no <stage> specified for -S");
John Kessenichc178f0a2017-06-23 10:58:31 -0600584 shaderStageName = argv[1];
585 bumpArg();
dankbaker45d49bc2016-08-08 21:43:07 -0400586 break;
John Kessenicha9313662017-06-15 10:40:49 -0600587 case 'U':
588 UserPreamble.addUndef(getStringOperand("-U<macro>: macro name"));
589 break;
John Kessenich6263fb12017-06-14 15:52:44 -0600590 case 'V':
John Kessenich6353d552017-06-23 11:11:09 -0600591 setVulkanSpv();
592 if (argv[0][2] != 0)
David Neto506d2c22018-02-27 21:55:23 -0500593 ClientInputSemanticsVersion = getAttachedNumber("-V<num> client input semantics");
John Kessenichc555ddd2015-06-17 02:38:44 +0000594 break;
John Kessenich05a70632013-09-17 19:26:08 +0000595 case 'c':
596 Options |= EOptionDumpConfig;
597 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000598 case 'd':
John Kessenichc6c80a62018-03-05 22:23:17 -0700599 if (strncmp(&argv[0][1], "dumpversion", strlen(&argv[0][1]) + 1) == 0 ||
600 strncmp(&argv[0][1], "dumpfullversion", strlen(&argv[0][1]) + 1) == 0)
601 Options |= EOptionDumpBareVersion;
602 else
603 Options |= EOptionDefaultDesktop;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000604 break;
John Kessenich4d65ee32016-03-12 18:17:47 -0700605 case 'e':
606 // HLSL todo: entry point handle needs much more sophistication.
607 // This is okay for one compilation unit with one entry point.
608 entryPointName = argv[1];
John Kessenichc178f0a2017-06-23 10:58:31 -0600609 if (argc <= 1)
John Kessenicha25530c2017-09-11 20:13:49 -0600610 Error("no <name> provided for -e");
John Kessenichc178f0a2017-06-23 10:58:31 -0600611 bumpArg();
John Kessenich4d65ee32016-03-12 18:17:47 -0700612 break;
John Kessenich121853f2017-05-31 17:11:16 -0600613 case 'g':
614 Options |= EOptionDebug;
615 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600616 case 'h':
617 usage();
618 break;
John Kessenich05a70632013-09-17 19:26:08 +0000619 case 'i':
John Kessenich94a81fb2013-08-31 02:41:30 +0000620 Options |= EOptionIntermediate;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000621 break;
622 case 'l':
John Kessenichd78e3512014-08-25 20:07:55 +0000623 Options |= EOptionLinkProgram;
John Kessenich94a81fb2013-08-31 02:41:30 +0000624 break;
625 case 'm':
626 Options |= EOptionMemoryLeakMode;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000627 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600628 case 'o':
John Kessenichc178f0a2017-06-23 10:58:31 -0600629 if (argc <= 1)
John Kessenich68d78fd2015-07-12 19:28:10 -0600630 Error("no <file> provided for -o");
John Kessenichc178f0a2017-06-23 10:58:31 -0600631 binaryFileName = argv[1];
632 bumpArg();
John Kessenich68d78fd2015-07-12 19:28:10 -0600633 break;
John Kessenich11f9fc72013-11-07 01:06:34 +0000634 case 'q':
635 Options |= EOptionDumpReflection;
636 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000637 case 'r':
John Kessenich94a81fb2013-08-31 02:41:30 +0000638 Options |= EOptionRelaxedErrors;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000639 break;
640 case 's':
John Kessenich94a81fb2013-08-31 02:41:30 +0000641 Options |= EOptionSuppressInfolog;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000642 break;
John Kessenich38f3b892013-09-06 19:52:57 +0000643 case 't':
Juan Lopeza558b262017-04-02 23:04:00 +0200644 Options |= EOptionMultiThreaded;
John Kessenich38f3b892013-09-06 19:52:57 +0000645 break;
John Kessenich319de232013-12-04 04:43:40 +0000646 case 'v':
647 Options |= EOptionDumpVersions;
648 break;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000649 case 'w':
650 Options |= EOptionSuppressWarnings;
651 break;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500652 case 'x':
653 Options |= EOptionOutputHexadecimal;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500654 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000655 default:
John Kessenich68d78fd2015-07-12 19:28:10 -0600656 usage();
657 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000658 }
John Kessenich38f3b892013-09-06 19:52:57 +0000659 } else {
John Kessenich05a70632013-09-17 19:26:08 +0000660 std::string name(argv[0]);
661 if (! SetConfigFile(name)) {
Juan Lopeza558b262017-04-02 23:04:00 +0200662 workItems.push_back(std::unique_ptr<glslang::TWorkItem>(new glslang::TWorkItem(name)));
John Kessenich05a70632013-09-17 19:26:08 +0000663 }
John Kessenich38f3b892013-09-06 19:52:57 +0000664 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000665 }
666
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200667 // Make sure that -S is always specified if --stdin is specified
668 if ((Options & EOptionStdin) && shaderStageName == nullptr)
669 Error("must provide -S when --stdin is given");
670
John Kessenich68d78fd2015-07-12 19:28:10 -0600671 // Make sure that -E is not specified alongside linking (which includes SPV generation)
672 if ((Options & EOptionOutputPreprocessed) && (Options & EOptionLinkProgram))
673 Error("can't use -E when linking is selected");
John Kessenichc555ddd2015-06-17 02:38:44 +0000674
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500675 // -o or -x makes no sense if there is no target binary
John Kessenich68d78fd2015-07-12 19:28:10 -0600676 if (binaryFileName && (Options & EOptionSpv) == 0)
677 Error("no binary generation requested (e.g., -V)");
steve-lunarge0b9deb2016-09-16 13:26:37 -0600678
679 if ((Options & EOptionFlattenUniformArrays) != 0 &&
680 (Options & EOptionReadHlsl) == 0)
681 Error("uniform array flattening only valid when compiling HLSL source.");
John Kessenich2b07c7e2013-07-31 18:44:13 +0000682}
683
John Kessenich68d78fd2015-07-12 19:28:10 -0600684//
685// Translate the meaningful subset of command-line options to parser-behavior options.
686//
John Kessenichb0a7eb52013-11-07 17:44:20 +0000687void SetMessageOptions(EShMessages& messages)
688{
689 if (Options & EOptionRelaxedErrors)
690 messages = (EShMessages)(messages | EShMsgRelaxedErrors);
691 if (Options & EOptionIntermediate)
692 messages = (EShMessages)(messages | EShMsgAST);
693 if (Options & EOptionSuppressWarnings)
694 messages = (EShMessages)(messages | EShMsgSuppressWarnings);
John Kessenich68d78fd2015-07-12 19:28:10 -0600695 if (Options & EOptionSpv)
696 messages = (EShMessages)(messages | EShMsgSpvRules);
697 if (Options & EOptionVulkanRules)
698 messages = (EShMessages)(messages | EShMsgVulkanRules);
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400699 if (Options & EOptionOutputPreprocessed)
700 messages = (EShMessages)(messages | EShMsgOnlyPreprocessor);
John Kessenich66e2faf2016-03-12 18:34:36 -0700701 if (Options & EOptionReadHlsl)
702 messages = (EShMessages)(messages | EShMsgReadHlsl);
John Kessenicha86836e2016-07-09 14:50:57 -0600703 if (Options & EOptionCascadingErrors)
704 messages = (EShMessages)(messages | EShMsgCascadingErrors);
John Kessenich906cc212016-12-09 19:22:20 -0700705 if (Options & EOptionKeepUncalled)
706 messages = (EShMessages)(messages | EShMsgKeepUncalled);
John Kessenich4f1403e2017-04-05 17:38:20 -0600707 if (Options & EOptionHlslOffsets)
708 messages = (EShMessages)(messages | EShMsgHlslOffsets);
John Kessenich121853f2017-05-31 17:11:16 -0600709 if (Options & EOptionDebug)
710 messages = (EShMessages)(messages | EShMsgDebugInfo);
John Kessenichb0a7eb52013-11-07 17:44:20 +0000711}
712
John Kessenich68d78fd2015-07-12 19:28:10 -0600713//
John Kessenich69f4b512013-09-04 21:19:27 +0000714// Thread entry point, for non-linking asynchronous mode.
John Kessenichc999ba22013-11-07 23:33:24 +0000715//
Juan Lopeza558b262017-04-02 23:04:00 +0200716void CompileShaders(glslang::TWorklist& worklist)
John Kessenich2b07c7e2013-07-31 18:44:13 +0000717{
John Kessenich38f3b892013-09-06 19:52:57 +0000718 glslang::TWorkItem* workItem;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200719 if (Options & EOptionStdin) {
720 worklist.remove(workItem);
721 ShHandle compiler = ShConstructCompiler(FindLanguage("stdin"), Options);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000722 if (compiler == 0)
Juan Lopeza558b262017-04-02 23:04:00 +0200723 return;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000724
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200725 CompileFile("stdin", compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000726
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200727 if (! (Options & EOptionSuppressInfolog))
728 workItem->results = ShGetInfoLog(compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000729
730 ShDestruct(compiler);
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +0200731 } else {
732 while (worklist.remove(workItem)) {
733 ShHandle compiler = ShConstructCompiler(FindLanguage(workItem->name), Options);
734 if (compiler == 0)
735 return;
736
737 CompileFile(workItem->name.c_str(), compiler);
738
739 if (! (Options & EOptionSuppressInfolog))
740 workItem->results = ShGetInfoLog(compiler);
741
742 ShDestruct(compiler);
743 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000744 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000745}
746
John Kessenich6626cad2015-06-19 05:14:19 +0000747// Outputs the given string, but only if it is non-null and non-empty.
748// This prevents erroneous newlines from appearing.
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400749void PutsIfNonEmpty(const char* str)
John Kessenich6626cad2015-06-19 05:14:19 +0000750{
751 if (str && str[0]) {
752 puts(str);
753 }
754}
755
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400756// Outputs the given string to stderr, but only if it is non-null and non-empty.
757// This prevents erroneous newlines from appearing.
758void StderrIfNonEmpty(const char* str)
759{
John Kessenich04acb1b2017-06-14 17:36:50 -0600760 if (str && str[0])
761 fprintf(stderr, "%s\n", str);
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400762}
763
John Kessenichc57b2a92016-01-16 15:30:03 -0700764// Simple bundling of what makes a compilation unit for ease in passing around,
765// and separation of handling file IO versus API (programmatic) compilation.
766struct ShaderCompUnit {
767 EShLanguage stage;
John Kessenich04acb1b2017-06-14 17:36:50 -0600768 static const int maxCount = 1;
769 int count; // live number of strings/names
John Kessenicha9313662017-06-15 10:40:49 -0600770 const char* text[maxCount]; // memory owned/managed externally
John Kessenich04acb1b2017-06-14 17:36:50 -0600771 std::string fileName[maxCount]; // hold's the memory, but...
772 const char* fileNameList[maxCount]; // downstream interface wants pointers
dankbakerafe6e9c2016-08-21 12:29:08 -0400773
John Kessenich04acb1b2017-06-14 17:36:50 -0600774 ShaderCompUnit(EShLanguage stage) : stage(stage), count(0) { }
dankbakerafe6e9c2016-08-21 12:29:08 -0400775
John Kessenich04acb1b2017-06-14 17:36:50 -0600776 ShaderCompUnit(const ShaderCompUnit& rhs)
dankbakerafe6e9c2016-08-21 12:29:08 -0400777 {
778 stage = rhs.stage;
John Kessenich04acb1b2017-06-14 17:36:50 -0600779 count = rhs.count;
780 for (int i = 0; i < count; ++i) {
781 fileName[i] = rhs.fileName[i];
782 text[i] = rhs.text[i];
783 fileNameList[i] = rhs.fileName[i].c_str();
784 }
dankbakerafe6e9c2016-08-21 12:29:08 -0400785 }
786
John Kessenicha9313662017-06-15 10:40:49 -0600787 void addString(std::string& ifileName, const char* itext)
John Kessenich04acb1b2017-06-14 17:36:50 -0600788 {
789 assert(count < maxCount);
790 fileName[count] = ifileName;
791 text[count] = itext;
792 fileNameList[count] = fileName[count].c_str();
793 ++count;
794 }
John Kessenichc57b2a92016-01-16 15:30:03 -0700795};
796
John Kessenich69f4b512013-09-04 21:19:27 +0000797//
John Kessenichc57b2a92016-01-16 15:30:03 -0700798// For linking mode: Will independently parse each compilation unit, but then put them
799// in the same program and link them together, making at most one linked module per
800// pipeline stage.
John Kessenich69f4b512013-09-04 21:19:27 +0000801//
802// Uses the new C++ interface instead of the old handle-based interface.
803//
John Kessenichc57b2a92016-01-16 15:30:03 -0700804
805void CompileAndLinkShaderUnits(std::vector<ShaderCompUnit> compUnits)
John Kessenich69f4b512013-09-04 21:19:27 +0000806{
807 // keep track of what to free
808 std::list<glslang::TShader*> shaders;
John Kessenichc57b2a92016-01-16 15:30:03 -0700809
John Kessenich69f4b512013-09-04 21:19:27 +0000810 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000811 SetMessageOptions(messages);
John Kessenich69f4b512013-09-04 21:19:27 +0000812
John Kessenich69f4b512013-09-04 21:19:27 +0000813 //
814 // Per-shader processing...
815 //
816
John Kessenich5b0f13a2013-11-01 03:08:40 +0000817 glslang::TProgram& program = *new glslang::TProgram;
rdb32084e82016-02-23 22:17:38 +0100818 for (auto it = compUnits.cbegin(); it != compUnits.cend(); ++it) {
819 const auto &compUnit = *it;
John Kessenichc57b2a92016-01-16 15:30:03 -0700820 glslang::TShader* shader = new glslang::TShader(compUnit.stage);
John Kessenich04acb1b2017-06-14 17:36:50 -0600821 shader->setStringsWithLengthsAndNames(compUnit.text, NULL, compUnit.fileNameList, compUnit.count);
John Kessenich4d65ee32016-03-12 18:17:47 -0700822 if (entryPointName) // HLSL todo: this needs to be tracked per compUnits
823 shader->setEntryPoint(entryPointName);
John Kessenich3693e632017-09-29 17:51:52 -0600824 if (sourceEntryPointName) {
825 if (entryPointName == nullptr)
826 printf("Warning: Changing source entry point name without setting an entry-point name.\n"
827 "Use '-e <name>'.\n");
steve-lunargf1e0c872016-10-31 15:13:43 -0600828 shader->setSourceEntryPoint(sourceEntryPointName);
John Kessenich3693e632017-09-29 17:51:52 -0600829 }
John Kessenicha9313662017-06-15 10:40:49 -0600830 if (UserPreamble.isSet())
831 shader->setPreamble(UserPreamble.get());
John Kessenich2a271162017-07-20 20:00:36 -0600832 shader->addProcesses(Processes);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600833
LoopDawg08a14422017-10-17 19:27:14 -0600834 // Set IO mapper binding shift values
835 for (int r = 0; r < glslang::EResCount; ++r) {
836 const glslang::TResourceType res = glslang::TResourceType(r);
837
838 // Set base bindings
839 shader->setShiftBinding(res, baseBinding[res][compUnit.stage]);
840
841 // Set bindings for particular resource sets
842 // TODO: use a range based for loop here, when available in all environments.
843 for (auto i = baseBindingForSet[res][compUnit.stage].begin();
844 i != baseBindingForSet[res][compUnit.stage].end(); ++i)
LoopDawge5709552017-10-21 10:46:39 -0600845 shader->setShiftBindingForSet(res, i->second, i->first);
LoopDawg08a14422017-10-17 19:27:14 -0600846 }
847
steve-lunarge0b9deb2016-09-16 13:26:37 -0600848 shader->setFlattenUniformArrays((Options & EOptionFlattenUniformArrays) != 0);
steve-lunargcce8d482016-10-14 18:36:42 -0600849 shader->setNoStorageFormat((Options & EOptionNoStorageFormat) != 0);
Hyangran Park36dc8292017-05-02 16:27:29 +0900850 shader->setResourceSetBinding(baseResourceSetBinding[compUnit.stage]);
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600851
steve-lunargbe283552017-04-18 12:18:01 -0600852 if (Options & EOptionHlslIoMapping)
853 shader->setHlslIoMapping(true);
854
steve-lunarg7f7c2ed2016-09-07 15:20:19 -0600855 if (Options & EOptionAutoMapBindings)
856 shader->setAutoMapBindings(true);
John Kessenichecba76f2017-01-06 00:34:48 -0700857
John Kessenich71facdf2017-05-17 18:28:19 -0600858 if (Options & EOptionAutoMapLocations)
859 shader->setAutoMapLocations(true);
860
LoopDawgb22c0692017-12-06 16:52:03 -0700861 if (Options & EOptionInvertY)
862 shader->setInvertY(true);
863
John Kessenich4be4aeb2017-06-23 10:50:22 -0600864 // Set up the environment, some subsettings take precedence over earlier
865 // ways of setting things.
866 if (Options & EOptionSpv) {
867 if (Options & EOptionVulkanRules) {
868 shader->setEnvInput((Options & EOptionReadHlsl) ? glslang::EShSourceHlsl
869 : glslang::EShSourceGlsl,
John Kessenich6353d552017-06-23 11:11:09 -0600870 compUnit.stage, glslang::EShClientVulkan, ClientInputSemanticsVersion);
871 shader->setEnvClient(glslang::EShClientVulkan, VulkanClientVersion);
John Kessenich22f02d02018-01-31 17:53:24 -0700872 shader->setEnvTarget(glslang::EShTargetSpv, TargetVersion);
John Kessenich4be4aeb2017-06-23 10:50:22 -0600873 } else {
874 shader->setEnvInput((Options & EOptionReadHlsl) ? glslang::EShSourceHlsl
875 : glslang::EShSourceGlsl,
John Kessenich6353d552017-06-23 11:11:09 -0600876 compUnit.stage, glslang::EShClientOpenGL, ClientInputSemanticsVersion);
877 shader->setEnvClient(glslang::EShClientOpenGL, OpenGLClientVersion);
878 shader->setEnvTarget(glslang::EshTargetSpv, TargetVersion);
John Kessenich4be4aeb2017-06-23 10:50:22 -0600879 }
880 }
881
John Kessenich69f4b512013-09-04 21:19:27 +0000882 shaders.push_back(shader);
John Kessenichb3297152015-07-11 18:01:03 -0600883
John Kessenich4be4aeb2017-06-23 10:50:22 -0600884 const int defaultVersion = Options & EOptionDefaultDesktop ? 110 : 100;
John Kessenich69f4b512013-09-04 21:19:27 +0000885
John Kessenich3494b4d2017-05-22 15:00:42 -0600886 DirStackFileIncluder includer;
John Kessenich971a0a82017-06-07 15:06:58 -0600887 std::for_each(IncludeDirectoryList.rbegin(), IncludeDirectoryList.rend(), [&includer](const std::string& dir) {
888 includer.pushExternalLocalDirectory(dir); });
John Kessenichc555ddd2015-06-17 02:38:44 +0000889 if (Options & EOptionOutputPreprocessed) {
890 std::string str;
Dejan Mircevski7be4b822015-06-17 11:40:33 -0400891 if (shader->preprocess(&Resources, defaultVersion, ENoProfile, false, false,
Andrew Woloszyna132af52016-03-07 13:23:09 -0500892 messages, &str, includer)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400893 PutsIfNonEmpty(str.c_str());
894 } else {
895 CompileFailed = true;
896 }
897 StderrIfNonEmpty(shader->getInfoLog());
898 StderrIfNonEmpty(shader->getInfoDebugLog());
John Kessenichc555ddd2015-06-17 02:38:44 +0000899 continue;
900 }
John Kessenich3494b4d2017-05-22 15:00:42 -0600901 if (! shader->parse(&Resources, defaultVersion, false, messages, includer))
John Kessenichc999ba22013-11-07 23:33:24 +0000902 CompileFailed = true;
John Kessenichc555ddd2015-06-17 02:38:44 +0000903
John Kessenich69f4b512013-09-04 21:19:27 +0000904 program.addShader(shader);
905
John Kessenichc57b2a92016-01-16 15:30:03 -0700906 if (! (Options & EOptionSuppressInfolog) &&
907 ! (Options & EOptionMemoryLeakMode)) {
John Kessenich04acb1b2017-06-14 17:36:50 -0600908 PutsIfNonEmpty(compUnit.fileName[0].c_str());
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400909 PutsIfNonEmpty(shader->getInfoLog());
910 PutsIfNonEmpty(shader->getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000911 }
John Kessenich69f4b512013-09-04 21:19:27 +0000912 }
913
914 //
915 // Program-level processing...
916 //
917
John Kessenich4d65ee32016-03-12 18:17:47 -0700918 // Link
John Kessenich68d78fd2015-07-12 19:28:10 -0600919 if (! (Options & EOptionOutputPreprocessed) && ! program.link(messages))
John Kessenichc999ba22013-11-07 23:33:24 +0000920 LinkFailed = true;
921
steve-lunarg9ae34742016-10-05 13:40:13 -0600922 // Map IO
923 if (Options & EOptionSpv) {
924 if (!program.mapIO())
925 LinkFailed = true;
926 }
John Kessenichecba76f2017-01-06 00:34:48 -0700927
John Kessenich4d65ee32016-03-12 18:17:47 -0700928 // Report
John Kessenichc57b2a92016-01-16 15:30:03 -0700929 if (! (Options & EOptionSuppressInfolog) &&
930 ! (Options & EOptionMemoryLeakMode)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400931 PutsIfNonEmpty(program.getInfoLog());
932 PutsIfNonEmpty(program.getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000933 }
934
John Kessenich4d65ee32016-03-12 18:17:47 -0700935 // Reflect
John Kessenich11f9fc72013-11-07 01:06:34 +0000936 if (Options & EOptionDumpReflection) {
937 program.buildReflection();
938 program.dumpReflection();
939 }
940
John Kessenich4d65ee32016-03-12 18:17:47 -0700941 // Dump SPIR-V
John Kessenich0df0cde2015-03-03 17:09:43 +0000942 if (Options & EOptionSpv) {
John Kessenich92f90382014-07-28 04:21:04 +0000943 if (CompileFailed || LinkFailed)
John Kessenich68d78fd2015-07-12 19:28:10 -0600944 printf("SPIR-V is not generated for failed compile or link\n");
John Kessenich92f90382014-07-28 04:21:04 +0000945 else {
946 for (int stage = 0; stage < EShLangCount; ++stage) {
John Kessenicha7a68a92014-08-24 18:21:00 +0000947 if (program.getIntermediate((EShLanguage)stage)) {
John Kessenich0df0cde2015-03-03 17:09:43 +0000948 std::vector<unsigned int> spirv;
Lei Zhang09caf122016-05-02 18:11:54 -0400949 std::string warningsErrors;
Lei Zhang17535f72016-05-04 15:55:59 -0400950 spv::SpvBuildLogger logger;
John Kessenich121853f2017-05-31 17:11:16 -0600951 glslang::SpvOptions spvOptions;
952 if (Options & EOptionDebug)
953 spvOptions.generateDebugInfo = true;
GregF52fe3d52017-09-28 10:08:32 -0600954 spvOptions.disableOptimizer = (Options & EOptionOptimizeDisable) != 0;
955 spvOptions.optimizeSize = (Options & EOptionOptimizeSize) != 0;
John Kessenich121853f2017-05-31 17:11:16 -0600956 glslang::GlslangToSpv(*program.getIntermediate((EShLanguage)stage), spirv, &logger, &spvOptions);
John Kessenichc57b2a92016-01-16 15:30:03 -0700957
958 // Dump the spv to a file or stdout, etc., but only if not doing
959 // memory/perf testing, as it's not internal to programmatic use.
960 if (! (Options & EOptionMemoryLeakMode)) {
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500961 printf("%s", logger.getAllMessages().c_str());
962 if (Options & EOptionOutputHexadecimal) {
John Kessenich8f674e82017-02-18 09:45:40 -0700963 glslang::OutputSpvHex(spirv, GetBinaryName((EShLanguage)stage), variableName);
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500964 } else {
965 glslang::OutputSpvBin(spirv, GetBinaryName((EShLanguage)stage));
966 }
John Kessenichc57b2a92016-01-16 15:30:03 -0700967 if (Options & EOptionHumanReadableSpv) {
John Kessenichc57b2a92016-01-16 15:30:03 -0700968 spv::Disassemble(std::cout, spirv);
969 }
John Kessenichacba7722015-03-04 03:48:38 +0000970 }
John Kessenicha7a68a92014-08-24 18:21:00 +0000971 }
John Kessenich92f90382014-07-28 04:21:04 +0000972 }
973 }
974 }
975
John Kessenich5b0f13a2013-11-01 03:08:40 +0000976 // Free everything up, program has to go before the shaders
977 // because it might have merged stuff from the shaders, and
978 // the stuff from the shaders has to have its destructors called
979 // before the pools holding the memory in the shaders is freed.
980 delete &program;
John Kessenich69f4b512013-09-04 21:19:27 +0000981 while (shaders.size() > 0) {
982 delete shaders.back();
983 shaders.pop_back();
984 }
John Kessenich69f4b512013-09-04 21:19:27 +0000985}
986
John Kessenichc57b2a92016-01-16 15:30:03 -0700987//
988// Do file IO part of compile and link, handing off the pure
989// API/programmatic mode to CompileAndLinkShaderUnits(), which can
990// be put in a loop for testing memory footprint and performance.
991//
992// This is just for linking mode: meaning all the shaders will be put into the
993// the same program linked together.
994//
995// This means there are a limited number of work items (not multi-threading mode)
996// and that the point is testing at the linking level. Hence, to enable
997// performance and memory testing, the actual compile/link can be put in
998// a loop, independent of processing the work items and file IO.
999//
Juan Lopeza558b262017-04-02 23:04:00 +02001000void CompileAndLinkShaderFiles(glslang::TWorklist& Worklist)
John Kessenichc57b2a92016-01-16 15:30:03 -07001001{
1002 std::vector<ShaderCompUnit> compUnits;
1003
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001004 // If this is using stdin, we can't really detect multiple different file
1005 // units by input type. We need to assume that we're just being given one
1006 // file of a certain type.
1007 if ((Options & EOptionStdin) != 0) {
1008 ShaderCompUnit compUnit(FindLanguage("stdin"));
1009 std::istreambuf_iterator<char> begin(std::cin), end;
1010 std::string tempString(begin, end);
1011 char* fileText = strdup(tempString.c_str());
1012 std::string fileName = "stdin";
1013 compUnit.addString(fileName, fileText);
John Kessenichc57b2a92016-01-16 15:30:03 -07001014 compUnits.push_back(compUnit);
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001015 } else {
1016 // Transfer all the work items from to a simple list of
1017 // of compilation units. (We don't care about the thread
1018 // work-item distribution properties in this path, which
1019 // is okay due to the limited number of shaders, know since
1020 // they are all getting linked together.)
1021 glslang::TWorkItem* workItem;
1022 while (Worklist.remove(workItem)) {
1023 ShaderCompUnit compUnit(FindLanguage(workItem->name));
1024 char* fileText = ReadFileData(workItem->name.c_str());
1025 if (fileText == nullptr)
1026 usage();
1027 compUnit.addString(workItem->name, fileText);
1028 compUnits.push_back(compUnit);
1029 }
John Kessenichc57b2a92016-01-16 15:30:03 -07001030 }
1031
1032 // Actual call to programmatic processing of compile and link,
1033 // in a loop for testing memory and performance. This part contains
1034 // all the perf/memory that a programmatic consumer will care about.
1035 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
1036 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j)
1037 CompileAndLinkShaderUnits(compUnits);
1038
1039 if (Options & EOptionMemoryLeakMode)
1040 glslang::OS_DumpMemoryCounters();
1041 }
1042
John Kessenicha9313662017-06-15 10:40:49 -06001043 // free memory from ReadFileData, which got stored in a const char*
1044 // as the first string above
rdb32084e82016-02-23 22:17:38 +01001045 for (auto it = compUnits.begin(); it != compUnits.end(); ++it)
John Kessenicha9313662017-06-15 10:40:49 -06001046 FreeFileData(const_cast<char*>(it->text[0]));
John Kessenichc57b2a92016-01-16 15:30:03 -07001047}
1048
John Kessenich94f28eb2017-11-13 01:32:06 -07001049int singleMain()
John Kessenicha0af4732012-12-12 21:15:54 +00001050{
Juan Lopeza558b262017-04-02 23:04:00 +02001051 glslang::TWorklist workList;
John Kessenich94f28eb2017-11-13 01:32:06 -07001052 std::for_each(WorkItems.begin(), WorkItems.end(), [&workList](std::unique_ptr<glslang::TWorkItem>& item) {
Juan Lopeza558b262017-04-02 23:04:00 +02001053 assert(item);
1054 workList.add(item.get());
1055 });
John Kessenich2b07c7e2013-07-31 18:44:13 +00001056
John Kessenich05a70632013-09-17 19:26:08 +00001057 if (Options & EOptionDumpConfig) {
Lei Zhang414eb602016-03-04 16:22:34 -05001058 printf("%s", glslang::GetDefaultTBuiltInResourceString().c_str());
Juan Lopeza558b262017-04-02 23:04:00 +02001059 if (workList.empty())
John Kessenich05a70632013-09-17 19:26:08 +00001060 return ESuccess;
1061 }
1062
John Kessenichc6c80a62018-03-05 22:23:17 -07001063 if (Options & EOptionDumpBareVersion) {
1064 printf("%d.%d.%d\n",
1065 glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL);
1066 if (workList.empty())
1067 return ESuccess;
1068 } else if (Options & EOptionDumpVersions) {
1069 printf("Glslang Version: %d.%d.%d\n",
1070 glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL);
John Kessenich319de232013-12-04 04:43:40 +00001071 printf("ESSL Version: %s\n", glslang::GetEsslVersionString());
1072 printf("GLSL Version: %s\n", glslang::GetGlslVersionString());
John Kessenich68d78fd2015-07-12 19:28:10 -06001073 std::string spirvVersion;
1074 glslang::GetSpirvVersion(spirvVersion);
John Kessenich0da9eaa2015-08-01 17:10:02 -06001075 printf("SPIR-V Version %s\n", spirvVersion.c_str());
John Kessenich5e4b1242015-08-06 22:53:06 -06001076 printf("GLSL.std.450 Version %d, Revision %d\n", GLSLstd450Version, GLSLstd450Revision);
John Kessenich55e7d112015-11-15 21:33:39 -07001077 printf("Khronos Tool ID %d\n", glslang::GetKhronosToolId());
John Kessenicha372a3e2017-11-02 22:32:14 -06001078 printf("SPIR-V Generator Version %d\n", glslang::GetSpirvGeneratorVersion());
John Kessenichb84313d2016-07-20 16:03:29 -06001079 printf("GL_KHR_vulkan_glsl version %d\n", 100);
1080 printf("ARB_GL_gl_spirv version %d\n", 100);
Juan Lopeza558b262017-04-02 23:04:00 +02001081 if (workList.empty())
John Kessenich319de232013-12-04 04:43:40 +00001082 return ESuccess;
1083 }
1084
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001085 if (workList.empty() && ((Options & EOptionStdin) == 0)) {
John Kessenich05a70632013-09-17 19:26:08 +00001086 usage();
John Kessenich05a70632013-09-17 19:26:08 +00001087 }
1088
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001089 if (Options & EOptionStdin) {
John Kessenich94f28eb2017-11-13 01:32:06 -07001090 WorkItems.push_back(std::unique_ptr<glslang::TWorkItem>{new glslang::TWorkItem("stdin")});
1091 workList.add(WorkItems.back().get());
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001092 }
1093
John Kessenich05a70632013-09-17 19:26:08 +00001094 ProcessConfigFile();
1095
John Kessenich69f4b512013-09-04 21:19:27 +00001096 //
1097 // Two modes:
John Kessenich38f3b892013-09-06 19:52:57 +00001098 // 1) linking all arguments together, single-threaded, new C++ interface
1099 // 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 +00001100 //
John Kessenichc555ddd2015-06-17 02:38:44 +00001101 if (Options & EOptionLinkProgram ||
1102 Options & EOptionOutputPreprocessed) {
John Kessenichc36e1d82013-11-01 17:41:52 +00001103 glslang::InitializeProcess();
John Kesseniche2c15b42017-11-16 22:48:41 -07001104 glslang::InitializeProcess(); // also test reference counting of users
1105 glslang::InitializeProcess(); // also test reference counting of users
1106 glslang::FinalizeProcess(); // also test reference counting of users
1107 glslang::FinalizeProcess(); // also test reference counting of users
Juan Lopeza558b262017-04-02 23:04:00 +02001108 CompileAndLinkShaderFiles(workList);
John Kessenichc36e1d82013-11-01 17:41:52 +00001109 glslang::FinalizeProcess();
1110 } else {
1111 ShInitialize();
John Kesseniche2c15b42017-11-16 22:48:41 -07001112 ShInitialize(); // also test reference counting of users
1113 ShFinalize(); // also test reference counting of users
John Kessenichc36e1d82013-11-01 17:41:52 +00001114
Juan Lopeza558b262017-04-02 23:04:00 +02001115 bool printShaderNames = workList.size() > 1;
John Kessenich69f4b512013-09-04 21:19:27 +00001116
John Kesseniche2c15b42017-11-16 22:48:41 -07001117 if (Options & EOptionMultiThreaded) {
Juan Lopeza558b262017-04-02 23:04:00 +02001118 std::array<std::thread, 16> threads;
John Kesseniche2c15b42017-11-16 22:48:41 -07001119 for (unsigned int t = 0; t < threads.size(); ++t) {
Juan Lopeza558b262017-04-02 23:04:00 +02001120 threads[t] = std::thread(CompileShaders, std::ref(workList));
John Kesseniche2c15b42017-11-16 22:48:41 -07001121 if (threads[t].get_id() == std::thread::id()) {
John Kessenichaf527992017-11-02 22:48:15 -06001122 fprintf(stderr, "Failed to create thread\n");
John Kessenich38f3b892013-09-06 19:52:57 +00001123 return EFailThreadCreate;
1124 }
John Kessenicha0af4732012-12-12 21:15:54 +00001125 }
Juan Lopeza558b262017-04-02 23:04:00 +02001126
1127 std::for_each(threads.begin(), threads.end(), [](std::thread& t) { t.join(); });
John Kessenichc999ba22013-11-07 23:33:24 +00001128 } else
Juan Lopeza558b262017-04-02 23:04:00 +02001129 CompileShaders(workList);
John Kessenich38f3b892013-09-06 19:52:57 +00001130
1131 // Print out all the resulting infologs
John Kessenich94f28eb2017-11-13 01:32:06 -07001132 for (size_t w = 0; w < WorkItems.size(); ++w) {
1133 if (WorkItems[w]) {
1134 if (printShaderNames || WorkItems[w]->results.size() > 0)
1135 PutsIfNonEmpty(WorkItems[w]->name.c_str());
1136 PutsIfNonEmpty(WorkItems[w]->results.c_str());
John Kessenich38f3b892013-09-06 19:52:57 +00001137 }
1138 }
John Kessenichc36e1d82013-11-01 17:41:52 +00001139
1140 ShFinalize();
John Kessenicha0af4732012-12-12 21:15:54 +00001141 }
John Kessenich2b07c7e2013-07-31 18:44:13 +00001142
John Kessenichc999ba22013-11-07 23:33:24 +00001143 if (CompileFailed)
John Kessenicha0af4732012-12-12 21:15:54 +00001144 return EFailCompile;
John Kessenichc999ba22013-11-07 23:33:24 +00001145 if (LinkFailed)
John Kessenicha0af4732012-12-12 21:15:54 +00001146 return EFailLink;
1147
1148 return 0;
1149}
1150
John Kessenich94f28eb2017-11-13 01:32:06 -07001151int C_DECL main(int argc, char* argv[])
1152{
1153 ProcessArguments(WorkItems, argc, argv);
1154
1155 int ret = 0;
1156
1157 // Loop over the entire init/finalize cycle to watch memory changes
1158 const int iterations = 1;
1159 if (iterations > 1)
1160 glslang::OS_DumpMemoryCounters();
1161 for (int i = 0; i < iterations; ++i) {
1162 ret = singleMain();
1163 if (iterations > 1)
1164 glslang::OS_DumpMemoryCounters();
1165 }
1166
1167 return ret;
1168}
1169
John Kessenicha0af4732012-12-12 21:15:54 +00001170//
1171// Deduce the language from the filename. Files must end in one of the
1172// following extensions:
1173//
John Kessenich2b07c7e2013-07-31 18:44:13 +00001174// .vert = vertex
1175// .tesc = tessellation control
1176// .tese = tessellation evaluation
1177// .geom = geometry
1178// .frag = fragment
John Kessenich94a81fb2013-08-31 02:41:30 +00001179// .comp = compute
John Kessenicha0af4732012-12-12 21:15:54 +00001180//
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001181EShLanguage FindLanguage(const std::string& name, bool parseSuffix)
John Kessenicha0af4732012-12-12 21:15:54 +00001182{
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001183 size_t ext = 0;
John Kesseniche751bca2017-03-16 11:20:38 -06001184 std::string suffix;
steve-lunarg7f7c2ed2016-09-07 15:20:19 -06001185
Dan Bakerc6ede892016-08-11 14:06:06 -04001186 if (shaderStageName)
1187 suffix = shaderStageName;
John Kesseniche751bca2017-03-16 11:20:38 -06001188 else {
1189 // Search for a suffix on a filename: e.g, "myfile.frag". If given
1190 // the suffix directly, we skip looking for the '.'
1191 if (parseSuffix) {
1192 ext = name.rfind('.');
1193 if (ext == std::string::npos) {
1194 usage();
1195 return EShLangVertex;
1196 }
1197 ++ext;
1198 }
1199 suffix = name.substr(ext, std::string::npos);
1200 }
dankbaker45d49bc2016-08-08 21:43:07 -04001201
John Kessenich2b07c7e2013-07-31 18:44:13 +00001202 if (suffix == "vert")
1203 return EShLangVertex;
1204 else if (suffix == "tesc")
1205 return EShLangTessControl;
1206 else if (suffix == "tese")
1207 return EShLangTessEvaluation;
1208 else if (suffix == "geom")
1209 return EShLangGeometry;
1210 else if (suffix == "frag")
1211 return EShLangFragment;
John Kessenichc0275792013-08-09 17:14:49 +00001212 else if (suffix == "comp")
1213 return EShLangCompute;
John Kessenich2b07c7e2013-07-31 18:44:13 +00001214
1215 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +00001216 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +00001217}
1218
John Kessenicha0af4732012-12-12 21:15:54 +00001219//
John Kessenichecba76f2017-01-06 00:34:48 -07001220// Read a file's data into a string, and compile it using the old interface ShCompile,
John Kessenich69f4b512013-09-04 21:19:27 +00001221// for non-linkable results.
John Kessenicha0af4732012-12-12 21:15:54 +00001222//
John Kessenich51cdd902014-02-18 23:37:57 +00001223void CompileFile(const char* fileName, ShHandle compiler)
John Kessenicha0af4732012-12-12 21:15:54 +00001224{
John Kessenichca3457f2015-05-18 01:59:45 +00001225 int ret = 0;
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001226 char* shaderString;
1227 if ((Options & EOptionStdin) != 0) {
1228 std::istreambuf_iterator<char> begin(std::cin), end;
1229 std::string tempString(begin, end);
1230 shaderString = strdup(tempString.c_str());
1231 } else {
1232 shaderString = ReadFileData(fileName);
1233 }
John Kessenich41cf6b52013-06-25 18:10:05 +00001234
1235 // move to length-based strings, rather than null-terminated strings
John Kessenich04acb1b2017-06-14 17:36:50 -06001236 int* lengths = new int[1];
1237 lengths[0] = (int)strlen(shaderString);
John Kessenicha0af4732012-12-12 21:15:54 +00001238
John Kessenich52ac67e2013-05-05 23:46:22 +00001239 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +00001240 SetMessageOptions(messages);
John Kessenichecba76f2017-01-06 00:34:48 -07001241
John Kessenicha9313662017-06-15 10:40:49 -06001242 if (UserPreamble.isSet())
1243 Error("-D and -U options require -l (linking)\n");
1244
John Kessenich94a81fb2013-08-31 02:41:30 +00001245 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
1246 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) {
John Kessenich927608b2017-01-06 12:34:14 -07001247 // ret = ShCompile(compiler, shaderStrings, NumShaderStrings, lengths, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich04acb1b2017-06-14 17:36:50 -06001248 ret = ShCompile(compiler, &shaderString, 1, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich927608b2017-01-06 12:34:14 -07001249 // const char* multi[12] = { "# ve", "rsion", " 300 e", "s", "\n#err",
John Kessenichecba76f2017-01-06 00:34:48 -07001250 // "or should be l", "ine 1", "string 5\n", "float glo", "bal",
John Kessenichea869fb2013-10-28 18:12:06 +00001251 // ";\n#error should be line 2\n void main() {", "global = 2.3;}" };
John Kessenich927608b2017-01-06 12:34:14 -07001252 // const char* multi[7] = { "/", "/", "\\", "\n", "\n", "#", "version 300 es" };
1253 // ret = ShCompile(compiler, multi, 7, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich41cf6b52013-06-25 18:10:05 +00001254 }
John Kessenicha0af4732012-12-12 21:15:54 +00001255
John Kessenich94a81fb2013-08-31 02:41:30 +00001256 if (Options & EOptionMemoryLeakMode)
John Kessenich2b07c7e2013-07-31 18:44:13 +00001257 glslang::OS_DumpMemoryCounters();
John Kessenicha0af4732012-12-12 21:15:54 +00001258 }
John Kessenicha0af4732012-12-12 21:15:54 +00001259
John Kessenich41cf6b52013-06-25 18:10:05 +00001260 delete [] lengths;
John Kessenich04acb1b2017-06-14 17:36:50 -06001261 FreeFileData(shaderString);
John Kessenicha0af4732012-12-12 21:15:54 +00001262
John Kessenichc999ba22013-11-07 23:33:24 +00001263 if (ret == 0)
1264 CompileFailed = true;
John Kessenicha0af4732012-12-12 21:15:54 +00001265}
1266
John Kessenicha0af4732012-12-12 21:15:54 +00001267//
1268// print usage to stdout
1269//
1270void usage()
1271{
John Kessenich319de232013-12-04 04:43:40 +00001272 printf("Usage: glslangValidator [option]... [file]...\n"
1273 "\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001274 "'file' can end in .<stage> for auto-stage classification, where <stage> is:\n"
1275 " .conf to provide a config file that replaces the default configuration\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001276 " (see -c option below for generating a template)\n"
1277 " .vert for a vertex shader\n"
1278 " .tesc for a tessellation control shader\n"
1279 " .tese for a tessellation evaluation shader\n"
1280 " .geom for a geometry shader\n"
1281 " .frag for a fragment shader\n"
1282 " .comp for a compute shader\n"
John Kessenich319de232013-12-04 04:43:40 +00001283 "\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001284 "Options:\n"
1285 " -C cascading errors; risk crash from accumulation of error recoveries\n"
1286 " -D input is HLSL\n"
John Kessenicha9313662017-06-15 10:40:49 -06001287 " -D<macro=def>\n"
1288 " -D<macro> define a pre-processor macro\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001289 " -E print pre-processed GLSL; cannot be used with -l;\n"
1290 " errors will appear on stderr.\n"
John Kessenich6353d552017-06-23 11:11:09 -06001291 " -G[ver] create SPIR-V binary, under OpenGL semantics; turns on -l;\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001292 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich6353d552017-06-23 11:11:09 -06001293 " 'ver', when present, is the version of the input semantics,\n"
1294 " which will appear in #define GL_SPIRV ver\n"
1295 " '--client opengl100' is the same as -G100\n"
1296 " a '--target-env' for OpenGL will also imply '-G'\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001297 " -H print human readable form of SPIR-V; turns on -V\n"
John Kessenich971a0a82017-06-07 15:06:58 -06001298 " -I<dir> add dir to the include search path; includer's directory\n"
1299 " is searched first, followed by left-to-right order of -I\n"
GregFcd1f1692017-09-21 18:40:22 -06001300 " -Od disables optimization. May cause illegal SPIR-V for HLSL.\n"
1301 " -Os optimizes SPIR-V to minimize size.\n"
John Kesseniche751bca2017-03-16 11:20:38 -06001302 " -S <stage> uses specified stage rather than parsing the file extension\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001303 " choices for <stage> are vert, tesc, tese, geom, frag, or comp\n"
John Kessenich6353d552017-06-23 11:11:09 -06001304 " -U<macro> undefine a pre-processor macro\n"
1305 " -V[ver] create SPIR-V binary, under Vulkan semantics; turns on -l;\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001306 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich6353d552017-06-23 11:11:09 -06001307 " 'ver', when present, is the version of the input semantics,\n"
1308 " which will appear in #define VULKAN ver\n"
1309 " '--client vulkan100' is the same as -V100\n"
1310 " a '--target-env' for Vulkan will also imply '-V'\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001311 " -c configuration dump;\n"
1312 " creates the default configuration file (redirect to a .conf file)\n"
1313 " -d default to desktop (#version 110) when there is no shader #version\n"
1314 " (default is ES version 100)\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001315 " -e <name> specify <name> as the entry-point name\n"
John Kessenich121853f2017-05-31 17:11:16 -06001316 " -g generate debug information\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001317 " -h print this usage message\n"
1318 " -i intermediate tree (glslang AST) is printed out\n"
1319 " -l link all input files together to form a single module\n"
1320 " -m memory leak mode\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001321 " -o <file> save binary to <file>, requires a binary option (e.g., -V)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001322 " -q dump reflection query database\n"
John Kessenich2a271162017-07-20 20:00:36 -06001323 " -r synonym for --relaxed-errors\n"
John Kessenichb63f4a32017-11-16 22:32:20 -07001324 " -s silence syntax and semantic error reporting\n"
John Kessenich68d78fd2015-07-12 19:28:10 -06001325 " -t multi-threaded mode\n"
1326 " -v print version strings\n"
John Kessenich2a271162017-07-20 20:00:36 -06001327 " -w synonym for --suppress-warnings\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001328 " -x save binary output as text-based 32-bit hexadecimal numbers\n"
1329 " --auto-map-bindings automatically bind uniform variables\n"
1330 " without explicit bindings.\n"
1331 " --amb synonym for --auto-map-bindings\n"
1332 " --auto-map-locations automatically locate input/output lacking\n"
John Kessenichc178f0a2017-06-23 10:58:31 -06001333 " 'location' (fragile, not cross stage)\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001334 " --aml synonym for --auto-map-locations\n"
John Kessenich6353d552017-06-23 11:11:09 -06001335 " --client {vulkan<ver>|opengl<ver>} see -V and -G\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001336 " -dumpfullversion print bare major.minor.patchlevel\n"
1337 " -dumpversion same as -dumpfullversion\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001338 " --flatten-uniform-arrays flatten uniform texture/sampler arrays to\n"
1339 " scalars\n"
1340 " --fua synonym for --flatten-uniform-arrays\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001341 " --hlsl-offsets Allow block offsets to follow HLSL rules\n"
1342 " Works independently of source language\n"
1343 " --hlsl-iomap Perform IO mapping in HLSL register space\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001344 " --invert-y | --iy invert position.Y output in vertex shader\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001345 " --keep-uncalled don't eliminate uncalled functions\n"
1346 " --ku synonym for --keep-uncalled\n"
1347 " --no-storage-format use Unknown image format\n"
1348 " --nsf synonym for --no-storage-format\n"
John Kessenich2a271162017-07-20 20:00:36 -06001349 " --relaxed-errors relaxed GLSL semantic error-checking mode\n"
LoopDawg52017192017-07-14 15:15:47 -06001350 " --resource-set-binding [stage] name set binding\n"
1351 " Set descriptor set and binding for individual resources\n"
1352 " --resource-set-binding [stage] set\n"
1353 " Set descriptor set for all resources\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001354 " --rsb [stage] type set binding synonym for --resource-set-binding\n"
1355 " --shift-image-binding [stage] num base binding number for images (uav)\n"
LoopDawge5709552017-10-21 10:46:39 -06001356 " --shift-image-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001357 " --sib [stage] num synonym for --shift-image-binding\n"
1358 " --shift-sampler-binding [stage] num base binding number for samplers\n"
LoopDawge5709552017-10-21 10:46:39 -06001359 " --shift-sampler-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001360 " --ssb [stage] num synonym for --shift-sampler-binding\n"
1361 " --shift-ssbo-binding [stage] num base binding number for SSBOs\n"
LoopDawge5709552017-10-21 10:46:39 -06001362 " --shift-ssbo-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001363 " --sbb [stage] num synonym for --shift-ssbo-binding\n"
1364 " --shift-texture-binding [stage] num base binding number for textures\n"
LoopDawge5709552017-10-21 10:46:39 -06001365 " --shift-texture-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001366 " --stb [stage] num synonym for --shift-texture-binding\n"
1367 " --shift-uav-binding [stage] num base binding number for UAVs\n"
LoopDawge5709552017-10-21 10:46:39 -06001368 " --shift-uav-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001369 " --suavb [stage] num synonym for --shift-uav-binding\n"
1370 " --shift-UBO-binding [stage] num base binding number for UBOs\n"
LoopDawge5709552017-10-21 10:46:39 -06001371 " --shift-UBO-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001372 " --shift-cbuffer-binding [stage] num synonym for --shift-UBO-binding\n"
LoopDawge5709552017-10-21 10:46:39 -06001373 " --shift-cbuffer-binding [stage] [num set]... per-descriptor-set shift values\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001374 " --sub [stage] num synonym for --shift-UBO-binding\n"
John Kessenicha25530c2017-09-11 20:13:49 -06001375 " --source-entrypoint <name> the given shader source function is\n"
1376 " renamed to be the <name> given in -e\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001377 " --sep synonym for --source-entrypoint\n"
Sven-Hendrik Haase0dd12852017-09-02 19:34:54 +02001378 " --stdin Read from stdin instead of from a file.\n"
1379 " You'll have to provide the shader stage\n"
1380 " using -S.\n"
John Kessenich2a271162017-07-20 20:00:36 -06001381 " --suppress-warnings suppress GLSL warnings\n"
1382 " (except as required by #extension : warn)\n"
John Kessenich66011cb2018-03-06 16:12:04 -07001383 " --target-env {vulkan1.0 | vulkan1.1 | opengl} \n"
1384 " set execution environment that emitted code\n"
1385 " will execute in (as opposed to the language\n"
John Kessenich2a271162017-07-20 20:00:36 -06001386 " semantics selected by --client) defaults:\n"
1387 " 'vulkan1.0' under '--client vulkan<ver>'\n"
1388 " 'opengl' under '--client opengl<ver>'\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001389 " --variable-name <name> Creates a C header file that contains a\n"
1390 " uint32_t array named <name>\n"
1391 " initialized with the shader binary code.\n"
John Kessenichc6c80a62018-03-05 22:23:17 -07001392 " --version synonym for -v\n"
John Kessenich6263fb12017-06-14 15:52:44 -06001393 " --vn <name> synonym for --variable-name <name>\n"
John Kessenichb0a7eb52013-11-07 17:44:20 +00001394 );
John Kessenich68d78fd2015-07-12 19:28:10 -06001395
1396 exit(EFailUsage);
John Kessenicha0af4732012-12-12 21:15:54 +00001397}
1398
John Kessenich3ce4e592014-10-06 19:57:34 +00001399#if !defined _MSC_VER && !defined MINGW_HAS_SECURE_API
John Kessenichcfd643e2013-03-08 23:14:42 +00001400
1401#include <errno.h>
1402
1403int fopen_s(
1404 FILE** pFile,
John Kessenich51cdd902014-02-18 23:37:57 +00001405 const char* filename,
1406 const char* mode
John Kessenichcfd643e2013-03-08 23:14:42 +00001407)
1408{
1409 if (!pFile || !filename || !mode) {
1410 return EINVAL;
1411 }
1412
1413 FILE* f = fopen(filename, mode);
1414 if (! f) {
1415 if (errno != 0) {
1416 return errno;
1417 } else {
1418 return ENOENT;
1419 }
1420 }
1421 *pFile = f;
1422
1423 return 0;
1424}
1425
1426#endif
1427
John Kessenicha0af4732012-12-12 21:15:54 +00001428//
1429// Malloc a string of sufficient size and read a string into it.
1430//
John Kessenich04acb1b2017-06-14 17:36:50 -06001431char* ReadFileData(const char* fileName)
John Kessenicha0af4732012-12-12 21:15:54 +00001432{
John Kessenichb3297152015-07-11 18:01:03 -06001433 FILE *in = nullptr;
John Kessenich3ce4e592014-10-06 19:57:34 +00001434 int errorCode = fopen_s(&in, fileName, "r");
John Kessenich68d78fd2015-07-12 19:28:10 -06001435 if (errorCode || in == nullptr)
1436 Error("unable to open input file");
John Kessenichecba76f2017-01-06 00:34:48 -07001437
John Kessenich04acb1b2017-06-14 17:36:50 -06001438 int count = 0;
John Kessenicha0af4732012-12-12 21:15:54 +00001439 while (fgetc(in) != EOF)
1440 count++;
1441
John Kessenichd6c72a42014-08-18 19:42:35 +00001442 fseek(in, 0, SEEK_SET);
John Kessenichca3457f2015-05-18 01:59:45 +00001443
John Kessenich04acb1b2017-06-14 17:36:50 -06001444 char* return_data = (char*)malloc(count + 1); // freed in FreeFileData()
1445 if ((int)fread(return_data, 1, count, in) != count) {
1446 free(return_data);
John Kessenich68d78fd2015-07-12 19:28:10 -06001447 Error("can't read input file");
John Kessenicha0af4732012-12-12 21:15:54 +00001448 }
John Kessenich68d78fd2015-07-12 19:28:10 -06001449
John Kessenich04acb1b2017-06-14 17:36:50 -06001450 return_data[count] = '\0';
John Kessenicha0af4732012-12-12 21:15:54 +00001451 fclose(in);
John Kessenichb3297152015-07-11 18:01:03 -06001452
John Kessenicha0af4732012-12-12 21:15:54 +00001453 return return_data;
1454}
1455
John Kessenich04acb1b2017-06-14 17:36:50 -06001456void FreeFileData(char* data)
John Kessenicha0af4732012-12-12 21:15:54 +00001457{
John Kessenichb3297152015-07-11 18:01:03 -06001458 free(data);
John Kessenicha0af4732012-12-12 21:15:54 +00001459}
1460
John Kessenich54d8cda2013-02-11 22:36:01 +00001461void InfoLogMsg(const char* msg, const char* name, const int num)
John Kessenicha0af4732012-12-12 21:15:54 +00001462{
John Kessenichfae38ee2015-06-10 23:23:12 +00001463 if (num >= 0 )
1464 printf("#### %s %s %d INFO LOG ####\n", msg, name, num);
1465 else
1466 printf("#### %s %s INFO LOG ####\n", msg, name);
John Kessenicha0af4732012-12-12 21:15:54 +00001467}