blob: c0fdeb99322d3f9a6ba68c6c318edd90483700d7 [file] [log] [blame]
John Kessenicha0af4732012-12-12 21:15:54 +00001//
2//Copyright (C) 2002-2005 3Dlabs Inc. Ltd.
John Kessenichbd0747d2013-02-17 06:01:50 +00003//Copyright (C) 2013 LunarG, Inc.
4//
John Kessenicha0af4732012-12-12 21:15:54 +00005//All rights reserved.
6//
7//Redistribution and use in source and binary forms, with or without
8//modification, are permitted provided that the following conditions
9//are met:
10//
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//
23//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.
35//
John Kessenich05a70632013-09-17 19:26:08 +000036
37// this only applies to the standalone wrapper, not the front end in general
38#define _CRT_SECURE_NO_WARNINGS
39
Lei Zhang8a9b1ee2016-05-19 13:31:43 -040040#include "ResourceLimits.h"
John Kessenich2b07c7e2013-07-31 18:44:13 +000041#include "Worklist.h"
John Kessenicha0af4732012-12-12 21:15:54 +000042#include "./../glslang/Include/ShHandle.h"
John Kessenich0da9eaa2015-08-01 17:10:02 -060043#include "./../glslang/Include/revision.h"
John Kessenicha0af4732012-12-12 21:15:54 +000044#include "./../glslang/Public/ShaderLang.h"
John Kessenich0df0cde2015-03-03 17:09:43 +000045#include "../SPIRV/GlslangToSpv.h"
John Kessenich5e4b1242015-08-06 22:53:06 -060046#include "../SPIRV/GLSL.std.450.h"
John Kessenichacba7722015-03-04 03:48:38 +000047#include "../SPIRV/doc.h"
48#include "../SPIRV/disassemble.h"
John Kessenich66ec80e2016-08-05 14:04:23 -060049#include <cstring>
50#include <cstdlib>
51#include <cmath>
John Kessenicha0af4732012-12-12 21:15:54 +000052
baldurk876a0e32015-11-16 18:03:28 +010053#include "../glslang/OSDependent/osinclude.h"
John Kessenicha0af4732012-12-12 21:15:54 +000054
55extern "C" {
56 SH_IMPORT_EXPORT void ShOutputHtml();
57}
58
John Kessenich94a81fb2013-08-31 02:41:30 +000059// Command-line options
60enum TOptions {
John Kessenicha86836e2016-07-09 14:50:57 -060061 EOptionNone = 0,
62 EOptionIntermediate = (1 << 0),
63 EOptionSuppressInfolog = (1 << 1),
64 EOptionMemoryLeakMode = (1 << 2),
65 EOptionRelaxedErrors = (1 << 3),
66 EOptionGiveWarnings = (1 << 4),
67 EOptionLinkProgram = (1 << 5),
68 EOptionMultiThreaded = (1 << 6),
69 EOptionDumpConfig = (1 << 7),
70 EOptionDumpReflection = (1 << 8),
71 EOptionSuppressWarnings = (1 << 9),
72 EOptionDumpVersions = (1 << 10),
73 EOptionSpv = (1 << 11),
74 EOptionHumanReadableSpv = (1 << 12),
75 EOptionVulkanRules = (1 << 13),
76 EOptionDefaultDesktop = (1 << 14),
77 EOptionOutputPreprocessed = (1 << 15),
78 EOptionOutputHexadecimal = (1 << 16),
79 EOptionReadHlsl = (1 << 17),
80 EOptionCascadingErrors = (1 << 18),
John Kessenich94a81fb2013-08-31 02:41:30 +000081};
82
John Kessenicha0af4732012-12-12 21:15:54 +000083//
John Kessenich68d78fd2015-07-12 19:28:10 -060084// Return codes from main/exit().
John Kessenicha0af4732012-12-12 21:15:54 +000085//
86enum TFailCode {
87 ESuccess = 0,
88 EFailUsage,
89 EFailCompile,
90 EFailLink,
91 EFailCompilerCreate,
John Kessenich2b07c7e2013-07-31 18:44:13 +000092 EFailThreadCreate,
John Kessenicha0af4732012-12-12 21:15:54 +000093 EFailLinkerCreate
94};
95
96//
John Kessenich68d78fd2015-07-12 19:28:10 -060097// Forward declarations.
John Kessenicha0af4732012-12-12 21:15:54 +000098//
John Kessenichb603f912013-08-29 00:39:25 +000099EShLanguage FindLanguage(const std::string& name);
John Kessenich51cdd902014-02-18 23:37:57 +0000100void CompileFile(const char* fileName, ShHandle);
John Kessenicha0af4732012-12-12 21:15:54 +0000101void usage();
John Kessenichea869fb2013-10-28 18:12:06 +0000102void FreeFileData(char** data);
103char** ReadFileData(const char* fileName);
John Kessenich54d8cda2013-02-11 22:36:01 +0000104void InfoLogMsg(const char* msg, const char* name, const int num);
John Kessenich41cf6b52013-06-25 18:10:05 +0000105
John Kessenichc999ba22013-11-07 23:33:24 +0000106// Globally track if any compile or link failure.
107bool CompileFailed = false;
108bool LinkFailed = false;
109
John Kessenich05a70632013-09-17 19:26:08 +0000110// Use to test breaking up a single shader file into multiple strings.
John Kessenich68d78fd2015-07-12 19:28:10 -0600111// Set in ReadFileData().
John Kessenichea869fb2013-10-28 18:12:06 +0000112int NumShaderStrings;
John Kessenicha0af4732012-12-12 21:15:54 +0000113
John Kessenich05a70632013-09-17 19:26:08 +0000114TBuiltInResource Resources;
115std::string ConfigFile;
116
John Kessenicha0af4732012-12-12 21:15:54 +0000117//
John Kessenichf0bcb0a2016-04-02 13:09:14 -0600118// Parse either a .conf file provided by the user or the default from glslang::DefaultTBuiltInResource
John Kessenich05a70632013-09-17 19:26:08 +0000119//
120void ProcessConfigFile()
John Kessenichb51f62c2013-04-11 16:31:09 +0000121{
John Kessenich05a70632013-09-17 19:26:08 +0000122 char** configStrings = 0;
John Kessenich51cdd902014-02-18 23:37:57 +0000123 char* config = 0;
John Kessenich05a70632013-09-17 19:26:08 +0000124 if (ConfigFile.size() > 0) {
John Kessenichea869fb2013-10-28 18:12:06 +0000125 configStrings = ReadFileData(ConfigFile.c_str());
John Kessenich05a70632013-09-17 19:26:08 +0000126 if (configStrings)
127 config = *configStrings;
128 else {
129 printf("Error opening configuration file; will instead use the default configuration\n");
130 usage();
131 }
132 }
133
134 if (config == 0) {
Lei Zhang414eb602016-03-04 16:22:34 -0500135 Resources = glslang::DefaultTBuiltInResource;
136 return;
John Kessenich05a70632013-09-17 19:26:08 +0000137 }
138
Lei Zhang1b141722016-05-19 13:50:49 -0400139 glslang::DecodeResourceLimits(&Resources, config);
John Kessenich05a70632013-09-17 19:26:08 +0000140
John Kessenich05a70632013-09-17 19:26:08 +0000141 if (configStrings)
142 FreeFileData(configStrings);
Andrew Woloszynb891c2b2016-01-18 09:26:25 -0500143 else
144 delete[] config;
John Kessenicha0af4732012-12-12 21:15:54 +0000145}
146
John Kessenich38f3b892013-09-06 19:52:57 +0000147// thread-safe list of shaders to asynchronously grab and compile
John Kessenich2b07c7e2013-07-31 18:44:13 +0000148glslang::TWorklist Worklist;
John Kessenich38f3b892013-09-06 19:52:57 +0000149
150// array of unique places to leave the shader names and infologs for the asynchronous compiles
John Kessenichfd305422014-06-05 16:30:53 +0000151glslang::TWorkItem** Work = 0;
John Kessenich38f3b892013-09-06 19:52:57 +0000152int NumWorkItems = 0;
153
John Kessenich94a81fb2013-08-31 02:41:30 +0000154int Options = 0;
John Kessenich68d78fd2015-07-12 19:28:10 -0600155const char* ExecutableName = nullptr;
156const char* binaryFileName = nullptr;
John Kessenich4d65ee32016-03-12 18:17:47 -0700157const char* entryPointName = nullptr;
John Kessenich68d78fd2015-07-12 19:28:10 -0600158
159//
160// Create the default name for saving a binary if -o is not provided.
161//
162const char* GetBinaryName(EShLanguage stage)
163{
164 const char* name;
165 if (binaryFileName == nullptr) {
166 switch (stage) {
167 case EShLangVertex: name = "vert.spv"; break;
168 case EShLangTessControl: name = "tesc.spv"; break;
169 case EShLangTessEvaluation: name = "tese.spv"; break;
170 case EShLangGeometry: name = "geom.spv"; break;
171 case EShLangFragment: name = "frag.spv"; break;
172 case EShLangCompute: name = "comp.spv"; break;
173 default: name = "unknown"; break;
174 }
175 } else
176 name = binaryFileName;
177
178 return name;
179}
John Kessenich2b07c7e2013-07-31 18:44:13 +0000180
John Kessenich05a70632013-09-17 19:26:08 +0000181//
182// *.conf => this is a config file that can set limits/resources
183//
184bool SetConfigFile(const std::string& name)
185{
186 if (name.size() < 5)
187 return false;
188
John Kessenich4c706852013-10-11 16:28:43 +0000189 if (name.compare(name.size() - 5, 5, ".conf") == 0) {
John Kessenich05a70632013-09-17 19:26:08 +0000190 ConfigFile = name;
191 return true;
192 }
193
194 return false;
195}
196
John Kessenich68d78fd2015-07-12 19:28:10 -0600197//
198// Give error and exit with failure code.
199//
200void Error(const char* message)
201{
202 printf("%s: Error %s (use -h for usage)\n", ExecutableName, message);
203 exit(EFailUsage);
204}
205
206//
207// Do all command-line argument parsing. This includes building up the work-items
208// to be processed later, and saving all the command-line options.
209//
210// Does not return (it exits) if command-line is fatally flawed.
211//
212void ProcessArguments(int argc, char* argv[])
John Kessenich2b07c7e2013-07-31 18:44:13 +0000213{
John Kessenich38f3b892013-09-06 19:52:57 +0000214 ExecutableName = argv[0];
215 NumWorkItems = argc; // will include some empties where the '-' options were, but it doesn't matter, they'll be 0
John Kessenichfd305422014-06-05 16:30:53 +0000216 Work = new glslang::TWorkItem*[NumWorkItems];
John Kessenich68d78fd2015-07-12 19:28:10 -0600217 for (int w = 0; w < NumWorkItems; ++w)
218 Work[w] = 0;
John Kessenich38f3b892013-09-06 19:52:57 +0000219
John Kessenich2b07c7e2013-07-31 18:44:13 +0000220 argc--;
221 argv++;
222 for (; argc >= 1; argc--, argv++) {
223 if (argv[0][0] == '-') {
John Kessenich2aa7f3a2015-05-15 16:02:07 +0000224 switch (argv[0][1]) {
225 case 'H':
226 Options |= EOptionHumanReadableSpv;
John Kessenich91e4aa52016-07-07 17:46:42 -0600227 if ((Options & EOptionSpv) == 0) {
228 // default to Vulkan
229 Options |= EOptionSpv;
230 Options |= EOptionVulkanRules;
231 Options |= EOptionLinkProgram;
232 }
233 break;
John Kessenich0df0cde2015-03-03 17:09:43 +0000234 case 'V':
235 Options |= EOptionSpv;
John Kessenich68d78fd2015-07-12 19:28:10 -0600236 Options |= EOptionVulkanRules;
237 Options |= EOptionLinkProgram;
238 break;
239 case 'G':
240 Options |= EOptionSpv;
John Kessenichd78e3512014-08-25 20:07:55 +0000241 Options |= EOptionLinkProgram;
John Kessenich91e4aa52016-07-07 17:46:42 -0600242 // undo a -H default to Vulkan
243 Options &= ~EOptionVulkanRules;
John Kessenich92f90382014-07-28 04:21:04 +0000244 break;
John Kessenichc555ddd2015-06-17 02:38:44 +0000245 case 'E':
246 Options |= EOptionOutputPreprocessed;
247 break;
John Kessenich05a70632013-09-17 19:26:08 +0000248 case 'c':
249 Options |= EOptionDumpConfig;
250 break;
John Kessenicha86836e2016-07-09 14:50:57 -0600251 case 'C':
252 Options |= EOptionCascadingErrors;
253 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000254 case 'd':
John Kessenich26ad2682014-08-13 20:17:19 +0000255 Options |= EOptionDefaultDesktop;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000256 break;
John Kessenich66e2faf2016-03-12 18:34:36 -0700257 case 'D':
258 Options |= EOptionReadHlsl;
259 break;
John Kessenich4d65ee32016-03-12 18:17:47 -0700260 case 'e':
261 // HLSL todo: entry point handle needs much more sophistication.
262 // This is okay for one compilation unit with one entry point.
263 entryPointName = argv[1];
264 if (argc > 0) {
265 argc--;
266 argv++;
267 } else
268 Error("no <entry-point> provided for -e");
269 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600270 case 'h':
271 usage();
272 break;
John Kessenich05a70632013-09-17 19:26:08 +0000273 case 'i':
John Kessenich94a81fb2013-08-31 02:41:30 +0000274 Options |= EOptionIntermediate;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000275 break;
276 case 'l':
John Kessenichd78e3512014-08-25 20:07:55 +0000277 Options |= EOptionLinkProgram;
John Kessenich94a81fb2013-08-31 02:41:30 +0000278 break;
279 case 'm':
280 Options |= EOptionMemoryLeakMode;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000281 break;
John Kessenich68d78fd2015-07-12 19:28:10 -0600282 case 'o':
283 binaryFileName = argv[1];
284 if (argc > 0) {
285 argc--;
286 argv++;
287 } else
288 Error("no <file> provided for -o");
289 break;
John Kessenich11f9fc72013-11-07 01:06:34 +0000290 case 'q':
291 Options |= EOptionDumpReflection;
292 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000293 case 'r':
John Kessenich94a81fb2013-08-31 02:41:30 +0000294 Options |= EOptionRelaxedErrors;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000295 break;
296 case 's':
John Kessenich94a81fb2013-08-31 02:41:30 +0000297 Options |= EOptionSuppressInfolog;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000298 break;
John Kessenich38f3b892013-09-06 19:52:57 +0000299 case 't':
300 #ifdef _WIN32
John Kessenichb0a7eb52013-11-07 17:44:20 +0000301 Options |= EOptionMultiThreaded;
John Kessenich38f3b892013-09-06 19:52:57 +0000302 #endif
303 break;
John Kessenich319de232013-12-04 04:43:40 +0000304 case 'v':
305 Options |= EOptionDumpVersions;
306 break;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000307 case 'w':
308 Options |= EOptionSuppressWarnings;
309 break;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500310 case 'x':
311 Options |= EOptionOutputHexadecimal;
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500312 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000313 default:
John Kessenich68d78fd2015-07-12 19:28:10 -0600314 usage();
315 break;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000316 }
John Kessenich38f3b892013-09-06 19:52:57 +0000317 } else {
John Kessenich05a70632013-09-17 19:26:08 +0000318 std::string name(argv[0]);
319 if (! SetConfigFile(name)) {
320 Work[argc] = new glslang::TWorkItem(name);
321 Worklist.add(Work[argc]);
322 }
John Kessenich38f3b892013-09-06 19:52:57 +0000323 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000324 }
325
John Kessenich68d78fd2015-07-12 19:28:10 -0600326 // Make sure that -E is not specified alongside linking (which includes SPV generation)
327 if ((Options & EOptionOutputPreprocessed) && (Options & EOptionLinkProgram))
328 Error("can't use -E when linking is selected");
John Kessenichc555ddd2015-06-17 02:38:44 +0000329
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500330 // -o or -x makes no sense if there is no target binary
John Kessenich68d78fd2015-07-12 19:28:10 -0600331 if (binaryFileName && (Options & EOptionSpv) == 0)
332 Error("no binary generation requested (e.g., -V)");
John Kessenich2b07c7e2013-07-31 18:44:13 +0000333}
334
John Kessenich68d78fd2015-07-12 19:28:10 -0600335//
336// Translate the meaningful subset of command-line options to parser-behavior options.
337//
John Kessenichb0a7eb52013-11-07 17:44:20 +0000338void SetMessageOptions(EShMessages& messages)
339{
340 if (Options & EOptionRelaxedErrors)
341 messages = (EShMessages)(messages | EShMsgRelaxedErrors);
342 if (Options & EOptionIntermediate)
343 messages = (EShMessages)(messages | EShMsgAST);
344 if (Options & EOptionSuppressWarnings)
345 messages = (EShMessages)(messages | EShMsgSuppressWarnings);
John Kessenich68d78fd2015-07-12 19:28:10 -0600346 if (Options & EOptionSpv)
347 messages = (EShMessages)(messages | EShMsgSpvRules);
348 if (Options & EOptionVulkanRules)
349 messages = (EShMessages)(messages | EShMsgVulkanRules);
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400350 if (Options & EOptionOutputPreprocessed)
351 messages = (EShMessages)(messages | EShMsgOnlyPreprocessor);
John Kessenich66e2faf2016-03-12 18:34:36 -0700352 if (Options & EOptionReadHlsl)
353 messages = (EShMessages)(messages | EShMsgReadHlsl);
John Kessenicha86836e2016-07-09 14:50:57 -0600354 if (Options & EOptionCascadingErrors)
355 messages = (EShMessages)(messages | EShMsgCascadingErrors);
John Kessenichb0a7eb52013-11-07 17:44:20 +0000356}
357
John Kessenich68d78fd2015-07-12 19:28:10 -0600358//
John Kessenich69f4b512013-09-04 21:19:27 +0000359// Thread entry point, for non-linking asynchronous mode.
John Kessenichc999ba22013-11-07 23:33:24 +0000360//
361// Return 0 for failure, 1 for success.
362//
Pyry Haulos5f6892e2015-12-01 12:59:53 -0800363unsigned int CompileShaders(void*)
John Kessenich2b07c7e2013-07-31 18:44:13 +0000364{
John Kessenich38f3b892013-09-06 19:52:57 +0000365 glslang::TWorkItem* workItem;
366 while (Worklist.remove(workItem)) {
367 ShHandle compiler = ShConstructCompiler(FindLanguage(workItem->name), Options);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000368 if (compiler == 0)
John Kessenichc999ba22013-11-07 23:33:24 +0000369 return 0;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000370
John Kessenichb0a7eb52013-11-07 17:44:20 +0000371 CompileFile(workItem->name.c_str(), compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000372
John Kessenich94a81fb2013-08-31 02:41:30 +0000373 if (! (Options & EOptionSuppressInfolog))
John Kessenich38f3b892013-09-06 19:52:57 +0000374 workItem->results = ShGetInfoLog(compiler);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000375
376 ShDestruct(compiler);
377 }
378
379 return 0;
380}
381
John Kessenich6626cad2015-06-19 05:14:19 +0000382// Outputs the given string, but only if it is non-null and non-empty.
383// This prevents erroneous newlines from appearing.
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400384void PutsIfNonEmpty(const char* str)
John Kessenich6626cad2015-06-19 05:14:19 +0000385{
386 if (str && str[0]) {
387 puts(str);
388 }
389}
390
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400391// Outputs the given string to stderr, but only if it is non-null and non-empty.
392// This prevents erroneous newlines from appearing.
393void StderrIfNonEmpty(const char* str)
394{
395 if (str && str[0]) {
396 fprintf(stderr, "%s\n", str);
397 }
398}
399
John Kessenichc57b2a92016-01-16 15:30:03 -0700400// Simple bundling of what makes a compilation unit for ease in passing around,
401// and separation of handling file IO versus API (programmatic) compilation.
402struct ShaderCompUnit {
403 EShLanguage stage;
404 std::string fileName;
405 char** text; // memory owned/managed externally
406};
407
John Kessenich69f4b512013-09-04 21:19:27 +0000408//
John Kessenichc57b2a92016-01-16 15:30:03 -0700409// For linking mode: Will independently parse each compilation unit, but then put them
410// in the same program and link them together, making at most one linked module per
411// pipeline stage.
John Kessenich69f4b512013-09-04 21:19:27 +0000412//
413// Uses the new C++ interface instead of the old handle-based interface.
414//
John Kessenichc57b2a92016-01-16 15:30:03 -0700415
416void CompileAndLinkShaderUnits(std::vector<ShaderCompUnit> compUnits)
John Kessenich69f4b512013-09-04 21:19:27 +0000417{
418 // keep track of what to free
419 std::list<glslang::TShader*> shaders;
John Kessenichc57b2a92016-01-16 15:30:03 -0700420
John Kessenich69f4b512013-09-04 21:19:27 +0000421 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000422 SetMessageOptions(messages);
John Kessenich69f4b512013-09-04 21:19:27 +0000423
John Kessenich69f4b512013-09-04 21:19:27 +0000424 //
425 // Per-shader processing...
426 //
427
John Kessenich5b0f13a2013-11-01 03:08:40 +0000428 glslang::TProgram& program = *new glslang::TProgram;
rdb32084e82016-02-23 22:17:38 +0100429 for (auto it = compUnits.cbegin(); it != compUnits.cend(); ++it) {
430 const auto &compUnit = *it;
John Kessenichc57b2a92016-01-16 15:30:03 -0700431 glslang::TShader* shader = new glslang::TShader(compUnit.stage);
432 shader->setStrings(compUnit.text, 1);
John Kessenich4d65ee32016-03-12 18:17:47 -0700433 if (entryPointName) // HLSL todo: this needs to be tracked per compUnits
434 shader->setEntryPoint(entryPointName);
John Kessenich69f4b512013-09-04 21:19:27 +0000435 shaders.push_back(shader);
John Kessenichb3297152015-07-11 18:01:03 -0600436
John Kessenichc555ddd2015-06-17 02:38:44 +0000437 const int defaultVersion = Options & EOptionDefaultDesktop? 110: 100;
John Kessenich69f4b512013-09-04 21:19:27 +0000438
John Kessenichc555ddd2015-06-17 02:38:44 +0000439 if (Options & EOptionOutputPreprocessed) {
440 std::string str;
Andrew Woloszyna132af52016-03-07 13:23:09 -0500441 glslang::TShader::ForbidInclude includer;
Dejan Mircevski7be4b822015-06-17 11:40:33 -0400442 if (shader->preprocess(&Resources, defaultVersion, ENoProfile, false, false,
Andrew Woloszyna132af52016-03-07 13:23:09 -0500443 messages, &str, includer)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400444 PutsIfNonEmpty(str.c_str());
445 } else {
446 CompileFailed = true;
447 }
448 StderrIfNonEmpty(shader->getInfoLog());
449 StderrIfNonEmpty(shader->getInfoDebugLog());
John Kessenichc555ddd2015-06-17 02:38:44 +0000450 continue;
451 }
452 if (! shader->parse(&Resources, defaultVersion, false, messages))
John Kessenichc999ba22013-11-07 23:33:24 +0000453 CompileFailed = true;
John Kessenichc555ddd2015-06-17 02:38:44 +0000454
John Kessenich69f4b512013-09-04 21:19:27 +0000455 program.addShader(shader);
456
John Kessenichc57b2a92016-01-16 15:30:03 -0700457 if (! (Options & EOptionSuppressInfolog) &&
458 ! (Options & EOptionMemoryLeakMode)) {
459 PutsIfNonEmpty(compUnit.fileName.c_str());
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400460 PutsIfNonEmpty(shader->getInfoLog());
461 PutsIfNonEmpty(shader->getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000462 }
John Kessenich69f4b512013-09-04 21:19:27 +0000463 }
464
465 //
466 // Program-level processing...
467 //
468
John Kessenich4d65ee32016-03-12 18:17:47 -0700469 // Link
John Kessenich68d78fd2015-07-12 19:28:10 -0600470 if (! (Options & EOptionOutputPreprocessed) && ! program.link(messages))
John Kessenichc999ba22013-11-07 23:33:24 +0000471 LinkFailed = true;
472
John Kessenich4d65ee32016-03-12 18:17:47 -0700473 // Report
John Kessenichc57b2a92016-01-16 15:30:03 -0700474 if (! (Options & EOptionSuppressInfolog) &&
475 ! (Options & EOptionMemoryLeakMode)) {
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400476 PutsIfNonEmpty(program.getInfoLog());
477 PutsIfNonEmpty(program.getInfoDebugLog());
John Kessenich69f4b512013-09-04 21:19:27 +0000478 }
479
John Kessenich4d65ee32016-03-12 18:17:47 -0700480 // Reflect
John Kessenich11f9fc72013-11-07 01:06:34 +0000481 if (Options & EOptionDumpReflection) {
482 program.buildReflection();
483 program.dumpReflection();
484 }
485
John Kessenich4d65ee32016-03-12 18:17:47 -0700486 // Dump SPIR-V
John Kessenich0df0cde2015-03-03 17:09:43 +0000487 if (Options & EOptionSpv) {
John Kessenich92f90382014-07-28 04:21:04 +0000488 if (CompileFailed || LinkFailed)
John Kessenich68d78fd2015-07-12 19:28:10 -0600489 printf("SPIR-V is not generated for failed compile or link\n");
John Kessenich92f90382014-07-28 04:21:04 +0000490 else {
491 for (int stage = 0; stage < EShLangCount; ++stage) {
John Kessenicha7a68a92014-08-24 18:21:00 +0000492 if (program.getIntermediate((EShLanguage)stage)) {
John Kessenich0df0cde2015-03-03 17:09:43 +0000493 std::vector<unsigned int> spirv;
Lei Zhang09caf122016-05-02 18:11:54 -0400494 std::string warningsErrors;
Lei Zhang17535f72016-05-04 15:55:59 -0400495 spv::SpvBuildLogger logger;
496 glslang::GlslangToSpv(*program.getIntermediate((EShLanguage)stage), spirv, &logger);
John Kessenichc57b2a92016-01-16 15:30:03 -0700497
498 // Dump the spv to a file or stdout, etc., but only if not doing
499 // memory/perf testing, as it's not internal to programmatic use.
500 if (! (Options & EOptionMemoryLeakMode)) {
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500501 printf("%s", logger.getAllMessages().c_str());
502 if (Options & EOptionOutputHexadecimal) {
503 glslang::OutputSpvHex(spirv, GetBinaryName((EShLanguage)stage));
504 } else {
505 glslang::OutputSpvBin(spirv, GetBinaryName((EShLanguage)stage));
506 }
John Kessenichc57b2a92016-01-16 15:30:03 -0700507 if (Options & EOptionHumanReadableSpv) {
John Kessenichc57b2a92016-01-16 15:30:03 -0700508 spv::Disassemble(std::cout, spirv);
509 }
John Kessenichacba7722015-03-04 03:48:38 +0000510 }
John Kessenicha7a68a92014-08-24 18:21:00 +0000511 }
John Kessenich92f90382014-07-28 04:21:04 +0000512 }
513 }
514 }
515
John Kessenich5b0f13a2013-11-01 03:08:40 +0000516 // Free everything up, program has to go before the shaders
517 // because it might have merged stuff from the shaders, and
518 // the stuff from the shaders has to have its destructors called
519 // before the pools holding the memory in the shaders is freed.
520 delete &program;
John Kessenich69f4b512013-09-04 21:19:27 +0000521 while (shaders.size() > 0) {
522 delete shaders.back();
523 shaders.pop_back();
524 }
John Kessenich69f4b512013-09-04 21:19:27 +0000525}
526
John Kessenichc57b2a92016-01-16 15:30:03 -0700527//
528// Do file IO part of compile and link, handing off the pure
529// API/programmatic mode to CompileAndLinkShaderUnits(), which can
530// be put in a loop for testing memory footprint and performance.
531//
532// This is just for linking mode: meaning all the shaders will be put into the
533// the same program linked together.
534//
535// This means there are a limited number of work items (not multi-threading mode)
536// and that the point is testing at the linking level. Hence, to enable
537// performance and memory testing, the actual compile/link can be put in
538// a loop, independent of processing the work items and file IO.
539//
540void CompileAndLinkShaderFiles()
541{
542 std::vector<ShaderCompUnit> compUnits;
543
544 // Transfer all the work items from to a simple list of
545 // of compilation units. (We don't care about the thread
546 // work-item distribution properties in this path, which
547 // is okay due to the limited number of shaders, know since
548 // they are all getting linked together.)
549 glslang::TWorkItem* workItem;
550 while (Worklist.remove(workItem)) {
551 ShaderCompUnit compUnit = {
552 FindLanguage(workItem->name),
553 workItem->name,
554 ReadFileData(workItem->name.c_str())
555 };
556
557 if (! compUnit.text) {
558 usage();
559 return;
560 }
561
562 compUnits.push_back(compUnit);
563 }
564
565 // Actual call to programmatic processing of compile and link,
566 // in a loop for testing memory and performance. This part contains
567 // all the perf/memory that a programmatic consumer will care about.
568 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
569 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j)
570 CompileAndLinkShaderUnits(compUnits);
571
572 if (Options & EOptionMemoryLeakMode)
573 glslang::OS_DumpMemoryCounters();
574 }
575
rdb32084e82016-02-23 22:17:38 +0100576 for (auto it = compUnits.begin(); it != compUnits.end(); ++it)
577 FreeFileData(it->text);
John Kessenichc57b2a92016-01-16 15:30:03 -0700578}
579
John Kessenicha0af4732012-12-12 21:15:54 +0000580int C_DECL main(int argc, char* argv[])
581{
John Kessenich68d78fd2015-07-12 19:28:10 -0600582 ProcessArguments(argc, argv);
John Kessenich2b07c7e2013-07-31 18:44:13 +0000583
John Kessenich05a70632013-09-17 19:26:08 +0000584 if (Options & EOptionDumpConfig) {
Lei Zhang414eb602016-03-04 16:22:34 -0500585 printf("%s", glslang::GetDefaultTBuiltInResourceString().c_str());
John Kessenich05a70632013-09-17 19:26:08 +0000586 if (Worklist.empty())
587 return ESuccess;
588 }
589
John Kessenich0da9eaa2015-08-01 17:10:02 -0600590 if (Options & EOptionDumpVersions) {
591 printf("Glslang Version: %s %s\n", GLSLANG_REVISION, GLSLANG_DATE);
John Kessenich319de232013-12-04 04:43:40 +0000592 printf("ESSL Version: %s\n", glslang::GetEsslVersionString());
593 printf("GLSL Version: %s\n", glslang::GetGlslVersionString());
John Kessenich68d78fd2015-07-12 19:28:10 -0600594 std::string spirvVersion;
595 glslang::GetSpirvVersion(spirvVersion);
John Kessenich0da9eaa2015-08-01 17:10:02 -0600596 printf("SPIR-V Version %s\n", spirvVersion.c_str());
John Kessenich5e4b1242015-08-06 22:53:06 -0600597 printf("GLSL.std.450 Version %d, Revision %d\n", GLSLstd450Version, GLSLstd450Revision);
John Kessenich55e7d112015-11-15 21:33:39 -0700598 printf("Khronos Tool ID %d\n", glslang::GetKhronosToolId());
John Kessenichb84313d2016-07-20 16:03:29 -0600599 printf("GL_KHR_vulkan_glsl version %d\n", 100);
600 printf("ARB_GL_gl_spirv version %d\n", 100);
John Kessenich319de232013-12-04 04:43:40 +0000601 if (Worklist.empty())
602 return ESuccess;
603 }
604
John Kessenich05a70632013-09-17 19:26:08 +0000605 if (Worklist.empty()) {
606 usage();
John Kessenich05a70632013-09-17 19:26:08 +0000607 }
608
609 ProcessConfigFile();
610
John Kessenich69f4b512013-09-04 21:19:27 +0000611 //
612 // Two modes:
John Kessenich38f3b892013-09-06 19:52:57 +0000613 // 1) linking all arguments together, single-threaded, new C++ interface
614 // 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 +0000615 //
John Kessenichc555ddd2015-06-17 02:38:44 +0000616 if (Options & EOptionLinkProgram ||
617 Options & EOptionOutputPreprocessed) {
John Kessenichc36e1d82013-11-01 17:41:52 +0000618 glslang::InitializeProcess();
John Kessenichc57b2a92016-01-16 15:30:03 -0700619 CompileAndLinkShaderFiles();
John Kessenichc36e1d82013-11-01 17:41:52 +0000620 glslang::FinalizeProcess();
Andrew Woloszynb891c2b2016-01-18 09:26:25 -0500621 for (int w = 0; w < NumWorkItems; ++w) {
622 if (Work[w]) {
623 delete Work[w];
624 }
625 }
John Kessenichc36e1d82013-11-01 17:41:52 +0000626 } else {
627 ShInitialize();
628
John Kessenich38f3b892013-09-06 19:52:57 +0000629 bool printShaderNames = Worklist.size() > 1;
John Kessenich69f4b512013-09-04 21:19:27 +0000630
John Kessenich38f3b892013-09-06 19:52:57 +0000631 if (Options & EOptionMultiThreaded) {
632 const int NumThreads = 16;
633 void* threads[NumThreads];
634 for (int t = 0; t < NumThreads; ++t) {
635 threads[t] = glslang::OS_CreateThread(&CompileShaders);
636 if (! threads[t]) {
637 printf("Failed to create thread\n");
638 return EFailThreadCreate;
639 }
John Kessenicha0af4732012-12-12 21:15:54 +0000640 }
John Kessenich38f3b892013-09-06 19:52:57 +0000641 glslang::OS_WaitForAllThreads(threads, NumThreads);
John Kessenichc999ba22013-11-07 23:33:24 +0000642 } else
643 CompileShaders(0);
John Kessenich38f3b892013-09-06 19:52:57 +0000644
645 // Print out all the resulting infologs
646 for (int w = 0; w < NumWorkItems; ++w) {
647 if (Work[w]) {
Kenneth Perryb07e6be2015-12-15 10:52:34 -0600648 if (printShaderNames || Work[w]->results.size() > 0)
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400649 PutsIfNonEmpty(Work[w]->name.c_str());
650 PutsIfNonEmpty(Work[w]->results.c_str());
John Kessenich38f3b892013-09-06 19:52:57 +0000651 delete Work[w];
652 }
653 }
John Kessenichc36e1d82013-11-01 17:41:52 +0000654
655 ShFinalize();
John Kessenicha0af4732012-12-12 21:15:54 +0000656 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000657
Andrew Woloszynb891c2b2016-01-18 09:26:25 -0500658 delete[] Work;
659
John Kessenichc999ba22013-11-07 23:33:24 +0000660 if (CompileFailed)
John Kessenicha0af4732012-12-12 21:15:54 +0000661 return EFailCompile;
John Kessenichc999ba22013-11-07 23:33:24 +0000662 if (LinkFailed)
John Kessenicha0af4732012-12-12 21:15:54 +0000663 return EFailLink;
664
665 return 0;
666}
667
668//
669// Deduce the language from the filename. Files must end in one of the
670// following extensions:
671//
John Kessenich2b07c7e2013-07-31 18:44:13 +0000672// .vert = vertex
673// .tesc = tessellation control
674// .tese = tessellation evaluation
675// .geom = geometry
676// .frag = fragment
John Kessenich94a81fb2013-08-31 02:41:30 +0000677// .comp = compute
John Kessenicha0af4732012-12-12 21:15:54 +0000678//
John Kessenichb603f912013-08-29 00:39:25 +0000679EShLanguage FindLanguage(const std::string& name)
John Kessenicha0af4732012-12-12 21:15:54 +0000680{
John Kessenich2b07c7e2013-07-31 18:44:13 +0000681 size_t ext = name.rfind('.');
682 if (ext == std::string::npos) {
683 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +0000684 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +0000685 }
686
John Kessenich2b07c7e2013-07-31 18:44:13 +0000687 std::string suffix = name.substr(ext + 1, std::string::npos);
688 if (suffix == "vert")
689 return EShLangVertex;
690 else if (suffix == "tesc")
691 return EShLangTessControl;
692 else if (suffix == "tese")
693 return EShLangTessEvaluation;
694 else if (suffix == "geom")
695 return EShLangGeometry;
696 else if (suffix == "frag")
697 return EShLangFragment;
John Kessenichc0275792013-08-09 17:14:49 +0000698 else if (suffix == "comp")
699 return EShLangCompute;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000700
701 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +0000702 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +0000703}
704
John Kessenicha0af4732012-12-12 21:15:54 +0000705//
John Kessenich69f4b512013-09-04 21:19:27 +0000706// Read a file's data into a string, and compile it using the old interface ShCompile,
707// for non-linkable results.
John Kessenicha0af4732012-12-12 21:15:54 +0000708//
John Kessenich51cdd902014-02-18 23:37:57 +0000709void CompileFile(const char* fileName, ShHandle compiler)
John Kessenicha0af4732012-12-12 21:15:54 +0000710{
John Kessenichca3457f2015-05-18 01:59:45 +0000711 int ret = 0;
John Kessenich41cf6b52013-06-25 18:10:05 +0000712 char** shaderStrings = ReadFileData(fileName);
John Kessenichdb4cd542013-06-26 22:42:55 +0000713 if (! shaderStrings) {
714 usage();
John Kessenichdb4cd542013-06-26 22:42:55 +0000715 }
716
John Kessenich41cf6b52013-06-25 18:10:05 +0000717 int* lengths = new int[NumShaderStrings];
718
719 // move to length-based strings, rather than null-terminated strings
720 for (int s = 0; s < NumShaderStrings; ++s)
John Kessenich35f04bd2014-02-19 02:47:20 +0000721 lengths[s] = (int)strlen(shaderStrings[s]);
John Kessenich09da79e2013-04-17 19:34:23 +0000722
John Kessenichc999ba22013-11-07 23:33:24 +0000723 if (! shaderStrings) {
724 CompileFailed = true;
725 return;
726 }
John Kessenicha0af4732012-12-12 21:15:54 +0000727
John Kessenich52ac67e2013-05-05 23:46:22 +0000728 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000729 SetMessageOptions(messages);
John Kessenich69f4b512013-09-04 21:19:27 +0000730
John Kessenich94a81fb2013-08-31 02:41:30 +0000731 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
732 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) {
John Kessenich26ad2682014-08-13 20:17:19 +0000733 //ret = ShCompile(compiler, shaderStrings, NumShaderStrings, lengths, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenichca3457f2015-05-18 01:59:45 +0000734 ret = ShCompile(compiler, shaderStrings, NumShaderStrings, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenichea869fb2013-10-28 18:12:06 +0000735 //const char* multi[12] = { "# ve", "rsion", " 300 e", "s", "\n#err",
736 // "or should be l", "ine 1", "string 5\n", "float glo", "bal",
737 // ";\n#error should be line 2\n void main() {", "global = 2.3;}" };
John Kessenich41cf6b52013-06-25 18:10:05 +0000738 //const char* multi[7] = { "/", "/", "\\", "\n", "\n", "#", "version 300 es" };
John Kessenichca3457f2015-05-18 01:59:45 +0000739 //ret = ShCompile(compiler, multi, 7, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich41cf6b52013-06-25 18:10:05 +0000740 }
John Kessenicha0af4732012-12-12 21:15:54 +0000741
John Kessenich94a81fb2013-08-31 02:41:30 +0000742 if (Options & EOptionMemoryLeakMode)
John Kessenich2b07c7e2013-07-31 18:44:13 +0000743 glslang::OS_DumpMemoryCounters();
John Kessenicha0af4732012-12-12 21:15:54 +0000744 }
John Kessenicha0af4732012-12-12 21:15:54 +0000745
John Kessenich41cf6b52013-06-25 18:10:05 +0000746 delete [] lengths;
747 FreeFileData(shaderStrings);
John Kessenicha0af4732012-12-12 21:15:54 +0000748
John Kessenichc999ba22013-11-07 23:33:24 +0000749 if (ret == 0)
750 CompileFailed = true;
John Kessenicha0af4732012-12-12 21:15:54 +0000751}
752
John Kessenicha0af4732012-12-12 21:15:54 +0000753//
754// print usage to stdout
755//
756void usage()
757{
John Kessenich319de232013-12-04 04:43:40 +0000758 printf("Usage: glslangValidator [option]... [file]...\n"
759 "\n"
John Kessenich0df0cde2015-03-03 17:09:43 +0000760 "Where: each 'file' ends in .<stage>, where <stage> is one of\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600761 " .conf to provide an optional config file that replaces the default configuration\n"
762 " (see -c option below for generating a template)\n"
763 " .vert for a vertex shader\n"
764 " .tesc for a tessellation control shader\n"
765 " .tese for a tessellation evaluation shader\n"
766 " .geom for a geometry shader\n"
767 " .frag for a fragment shader\n"
768 " .comp for a compute shader\n"
John Kessenich319de232013-12-04 04:43:40 +0000769 "\n"
John Kessenich4586dbd2013-08-05 15:52:03 +0000770 "Compilation warnings and errors will be printed to stdout.\n"
John Kessenich319de232013-12-04 04:43:40 +0000771 "\n"
John Kessenich4586dbd2013-08-05 15:52:03 +0000772 "To get other information, use one of the following options:\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600773 "Each option must be specified separately.\n"
774 " -V create SPIR-V binary, under Vulkan semantics; turns on -l;\n"
775 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600776 " -G create SPIR-V binary, under OpenGL semantics; turns on -l;\n"
777 " default file name is <stage>.spv (-o overrides this)\n"
778 " -H print human readable form of SPIR-V; turns on -V\n"
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400779 " -E print pre-processed GLSL; cannot be used with -l;\n"
780 " errors will appear on stderr.\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600781 " -c configuration dump;\n"
782 " creates the default configuration file (redirect to a .conf file)\n"
John Kessenicha86836e2016-07-09 14:50:57 -0600783 " -C cascading errors; risks crashes from accumulation of error recoveries\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600784 " -d default to desktop (#version 110) when there is no shader #version\n"
785 " (default is ES version 100)\n"
John Kessenich66e2faf2016-03-12 18:34:36 -0700786 " -D input is HLSL\n"
John Kessenich4d65ee32016-03-12 18:17:47 -0700787 " -e specify entry-point name\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600788 " -h print this usage message\n"
789 " -i intermediate tree (glslang AST) is printed out\n"
790 " -l link all input files together to form a single module\n"
791 " -m memory leak mode\n"
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500792 " -o <file> save binary to <file>, requires a binary option (e.g., -V)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600793 " -q dump reflection query database\n"
794 " -r relaxed semantic error-checking mode\n"
795 " -s silent mode\n"
796 " -t multi-threaded mode\n"
797 " -v print version strings\n"
798 " -w suppress warnings (except as required by #extension : warn)\n"
Johannes van Waveren1fd01752016-05-31 08:39:41 -0500799 " -x save 32-bit hexadecimal numbers as text, requires a binary option (e.g., -V)\n"
John Kessenichb0a7eb52013-11-07 17:44:20 +0000800 );
John Kessenich68d78fd2015-07-12 19:28:10 -0600801
802 exit(EFailUsage);
John Kessenicha0af4732012-12-12 21:15:54 +0000803}
804
John Kessenich3ce4e592014-10-06 19:57:34 +0000805#if !defined _MSC_VER && !defined MINGW_HAS_SECURE_API
John Kessenichcfd643e2013-03-08 23:14:42 +0000806
807#include <errno.h>
808
809int fopen_s(
810 FILE** pFile,
John Kessenich51cdd902014-02-18 23:37:57 +0000811 const char* filename,
812 const char* mode
John Kessenichcfd643e2013-03-08 23:14:42 +0000813)
814{
815 if (!pFile || !filename || !mode) {
816 return EINVAL;
817 }
818
819 FILE* f = fopen(filename, mode);
820 if (! f) {
821 if (errno != 0) {
822 return errno;
823 } else {
824 return ENOENT;
825 }
826 }
827 *pFile = f;
828
829 return 0;
830}
831
832#endif
833
John Kessenicha0af4732012-12-12 21:15:54 +0000834//
835// Malloc a string of sufficient size and read a string into it.
836//
John Kessenich51cdd902014-02-18 23:37:57 +0000837char** ReadFileData(const char* fileName)
John Kessenicha0af4732012-12-12 21:15:54 +0000838{
John Kessenichb3297152015-07-11 18:01:03 -0600839 FILE *in = nullptr;
John Kessenich3ce4e592014-10-06 19:57:34 +0000840 int errorCode = fopen_s(&in, fileName, "r");
John Kessenichd6c72a42014-08-18 19:42:35 +0000841
John Kessenicha0af4732012-12-12 21:15:54 +0000842 int count = 0;
John Kessenichb3297152015-07-11 18:01:03 -0600843 const int maxSourceStrings = 5; // for testing splitting shader/tokens across multiple strings
844 char** return_data = (char**)malloc(sizeof(char *) * (maxSourceStrings+1)); // freed in FreeFileData()
John Kessenicha0af4732012-12-12 21:15:54 +0000845
John Kessenich68d78fd2015-07-12 19:28:10 -0600846 if (errorCode || in == nullptr)
847 Error("unable to open input file");
John Kessenicha0af4732012-12-12 21:15:54 +0000848
849 while (fgetc(in) != EOF)
850 count++;
851
John Kessenichd6c72a42014-08-18 19:42:35 +0000852 fseek(in, 0, SEEK_SET);
John Kessenichca3457f2015-05-18 01:59:45 +0000853
John Kessenichb3297152015-07-11 18:01:03 -0600854 char *fdata = (char*)malloc(count+2); // freed before return of this function
John Kessenich68d78fd2015-07-12 19:28:10 -0600855 if (! fdata)
856 Error("can't allocate memory");
857
John Kessenichb3297152015-07-11 18:01:03 -0600858 if ((int)fread(fdata, 1, count, in) != count) {
John Kessenichb3297152015-07-11 18:01:03 -0600859 free(fdata);
John Kessenich68d78fd2015-07-12 19:28:10 -0600860 Error("can't read input file");
John Kessenicha0af4732012-12-12 21:15:54 +0000861 }
John Kessenich68d78fd2015-07-12 19:28:10 -0600862
John Kessenicha0af4732012-12-12 21:15:54 +0000863 fdata[count] = '\0';
864 fclose(in);
John Kessenichb3297152015-07-11 18:01:03 -0600865
John Kessenichea869fb2013-10-28 18:12:06 +0000866 if (count == 0) {
John Kessenichb3297152015-07-11 18:01:03 -0600867 // recover from empty file
868 return_data[0] = (char*)malloc(count+2); // freed in FreeFileData()
John Kessenicha0af4732012-12-12 21:15:54 +0000869 return_data[0][0]='\0';
John Kessenichea869fb2013-10-28 18:12:06 +0000870 NumShaderStrings = 0;
John Kessenichb3297152015-07-11 18:01:03 -0600871 free(fdata);
John Kessenicha0af4732012-12-12 21:15:54 +0000872
John Kessenichb3297152015-07-11 18:01:03 -0600873 return return_data;
874 } else
875 NumShaderStrings = 1; // Set to larger than 1 for testing multiple strings
876
877 // compute how to split up the file into multiple strings, for testing multiple strings
John Kessenichd6c72a42014-08-18 19:42:35 +0000878 int len = (int)(ceil)((float)count/(float)NumShaderStrings);
John Kessenichb3297152015-07-11 18:01:03 -0600879 int ptr_len = 0;
880 int i = 0;
881 while (count > 0) {
882 return_data[i] = (char*)malloc(len + 2); // freed in FreeFileData()
883 memcpy(return_data[i], fdata + ptr_len, len);
884 return_data[i][len] = '\0';
885 count -= len;
886 ptr_len += len;
887 if (count < len) {
888 if (count == 0) {
889 NumShaderStrings = i + 1;
John Kessenicha0af4732012-12-12 21:15:54 +0000890 break;
891 }
John Kessenichb3297152015-07-11 18:01:03 -0600892 len = count;
John Kessenichd6c72a42014-08-18 19:42:35 +0000893 }
894 ++i;
895 }
John Kessenichb3297152015-07-11 18:01:03 -0600896
897 free(fdata);
898
John Kessenicha0af4732012-12-12 21:15:54 +0000899 return return_data;
900}
901
John Kessenich51cdd902014-02-18 23:37:57 +0000902void FreeFileData(char** data)
John Kessenicha0af4732012-12-12 21:15:54 +0000903{
John Kessenichb3297152015-07-11 18:01:03 -0600904 for(int i = 0; i < NumShaderStrings; i++)
John Kessenicha0af4732012-12-12 21:15:54 +0000905 free(data[i]);
John Kessenichb3297152015-07-11 18:01:03 -0600906
907 free(data);
John Kessenicha0af4732012-12-12 21:15:54 +0000908}
909
John Kessenich54d8cda2013-02-11 22:36:01 +0000910void InfoLogMsg(const char* msg, const char* name, const int num)
John Kessenicha0af4732012-12-12 21:15:54 +0000911{
John Kessenichfae38ee2015-06-10 23:23:12 +0000912 if (num >= 0 )
913 printf("#### %s %s %d INFO LOG ####\n", msg, name, num);
914 else
915 printf("#### %s %s INFO LOG ####\n", msg, name);
John Kessenicha0af4732012-12-12 21:15:54 +0000916}