blob: 63e9878a5e8e0805283abda8e333cc500b2f57ea [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 Kessenicha0af4732012-12-12 21:15:54 +000049#include <string.h>
John Kessenichcfd643e2013-03-08 23:14:42 +000050#include <stdlib.h>
John Kessenicha0af4732012-12-12 21:15:54 +000051#include <math.h>
52
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 Kessenich319de232013-12-04 04:43:40 +0000599 if (Worklist.empty())
600 return ESuccess;
601 }
602
John Kessenich05a70632013-09-17 19:26:08 +0000603 if (Worklist.empty()) {
604 usage();
John Kessenich05a70632013-09-17 19:26:08 +0000605 }
606
607 ProcessConfigFile();
608
John Kessenich69f4b512013-09-04 21:19:27 +0000609 //
610 // Two modes:
John Kessenich38f3b892013-09-06 19:52:57 +0000611 // 1) linking all arguments together, single-threaded, new C++ interface
612 // 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 +0000613 //
John Kessenichc555ddd2015-06-17 02:38:44 +0000614 if (Options & EOptionLinkProgram ||
615 Options & EOptionOutputPreprocessed) {
John Kessenichc36e1d82013-11-01 17:41:52 +0000616 glslang::InitializeProcess();
John Kessenichc57b2a92016-01-16 15:30:03 -0700617 CompileAndLinkShaderFiles();
John Kessenichc36e1d82013-11-01 17:41:52 +0000618 glslang::FinalizeProcess();
Andrew Woloszynb891c2b2016-01-18 09:26:25 -0500619 for (int w = 0; w < NumWorkItems; ++w) {
620 if (Work[w]) {
621 delete Work[w];
622 }
623 }
John Kessenichc36e1d82013-11-01 17:41:52 +0000624 } else {
625 ShInitialize();
626
John Kessenich38f3b892013-09-06 19:52:57 +0000627 bool printShaderNames = Worklist.size() > 1;
John Kessenich69f4b512013-09-04 21:19:27 +0000628
John Kessenich38f3b892013-09-06 19:52:57 +0000629 if (Options & EOptionMultiThreaded) {
630 const int NumThreads = 16;
631 void* threads[NumThreads];
632 for (int t = 0; t < NumThreads; ++t) {
633 threads[t] = glslang::OS_CreateThread(&CompileShaders);
634 if (! threads[t]) {
635 printf("Failed to create thread\n");
636 return EFailThreadCreate;
637 }
John Kessenicha0af4732012-12-12 21:15:54 +0000638 }
John Kessenich38f3b892013-09-06 19:52:57 +0000639 glslang::OS_WaitForAllThreads(threads, NumThreads);
John Kessenichc999ba22013-11-07 23:33:24 +0000640 } else
641 CompileShaders(0);
John Kessenich38f3b892013-09-06 19:52:57 +0000642
643 // Print out all the resulting infologs
644 for (int w = 0; w < NumWorkItems; ++w) {
645 if (Work[w]) {
Kenneth Perryb07e6be2015-12-15 10:52:34 -0600646 if (printShaderNames || Work[w]->results.size() > 0)
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400647 PutsIfNonEmpty(Work[w]->name.c_str());
648 PutsIfNonEmpty(Work[w]->results.c_str());
John Kessenich38f3b892013-09-06 19:52:57 +0000649 delete Work[w];
650 }
651 }
John Kessenichc36e1d82013-11-01 17:41:52 +0000652
653 ShFinalize();
John Kessenicha0af4732012-12-12 21:15:54 +0000654 }
John Kessenich2b07c7e2013-07-31 18:44:13 +0000655
Andrew Woloszynb891c2b2016-01-18 09:26:25 -0500656 delete[] Work;
657
John Kessenichc999ba22013-11-07 23:33:24 +0000658 if (CompileFailed)
John Kessenicha0af4732012-12-12 21:15:54 +0000659 return EFailCompile;
John Kessenichc999ba22013-11-07 23:33:24 +0000660 if (LinkFailed)
John Kessenicha0af4732012-12-12 21:15:54 +0000661 return EFailLink;
662
663 return 0;
664}
665
666//
667// Deduce the language from the filename. Files must end in one of the
668// following extensions:
669//
John Kessenich2b07c7e2013-07-31 18:44:13 +0000670// .vert = vertex
671// .tesc = tessellation control
672// .tese = tessellation evaluation
673// .geom = geometry
674// .frag = fragment
John Kessenich94a81fb2013-08-31 02:41:30 +0000675// .comp = compute
John Kessenicha0af4732012-12-12 21:15:54 +0000676//
John Kessenichb603f912013-08-29 00:39:25 +0000677EShLanguage FindLanguage(const std::string& name)
John Kessenicha0af4732012-12-12 21:15:54 +0000678{
John Kessenich2b07c7e2013-07-31 18:44:13 +0000679 size_t ext = name.rfind('.');
680 if (ext == std::string::npos) {
681 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +0000682 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +0000683 }
684
John Kessenich2b07c7e2013-07-31 18:44:13 +0000685 std::string suffix = name.substr(ext + 1, std::string::npos);
686 if (suffix == "vert")
687 return EShLangVertex;
688 else if (suffix == "tesc")
689 return EShLangTessControl;
690 else if (suffix == "tese")
691 return EShLangTessEvaluation;
692 else if (suffix == "geom")
693 return EShLangGeometry;
694 else if (suffix == "frag")
695 return EShLangFragment;
John Kessenichc0275792013-08-09 17:14:49 +0000696 else if (suffix == "comp")
697 return EShLangCompute;
John Kessenich2b07c7e2013-07-31 18:44:13 +0000698
699 usage();
John Kesseniche95ecc52012-12-12 21:34:14 +0000700 return EShLangVertex;
John Kessenicha0af4732012-12-12 21:15:54 +0000701}
702
John Kessenicha0af4732012-12-12 21:15:54 +0000703//
John Kessenich69f4b512013-09-04 21:19:27 +0000704// Read a file's data into a string, and compile it using the old interface ShCompile,
705// for non-linkable results.
John Kessenicha0af4732012-12-12 21:15:54 +0000706//
John Kessenich51cdd902014-02-18 23:37:57 +0000707void CompileFile(const char* fileName, ShHandle compiler)
John Kessenicha0af4732012-12-12 21:15:54 +0000708{
John Kessenichca3457f2015-05-18 01:59:45 +0000709 int ret = 0;
John Kessenich41cf6b52013-06-25 18:10:05 +0000710 char** shaderStrings = ReadFileData(fileName);
John Kessenichdb4cd542013-06-26 22:42:55 +0000711 if (! shaderStrings) {
712 usage();
John Kessenichdb4cd542013-06-26 22:42:55 +0000713 }
714
John Kessenich41cf6b52013-06-25 18:10:05 +0000715 int* lengths = new int[NumShaderStrings];
716
717 // move to length-based strings, rather than null-terminated strings
718 for (int s = 0; s < NumShaderStrings; ++s)
John Kessenich35f04bd2014-02-19 02:47:20 +0000719 lengths[s] = (int)strlen(shaderStrings[s]);
John Kessenich09da79e2013-04-17 19:34:23 +0000720
John Kessenichc999ba22013-11-07 23:33:24 +0000721 if (! shaderStrings) {
722 CompileFailed = true;
723 return;
724 }
John Kessenicha0af4732012-12-12 21:15:54 +0000725
John Kessenich52ac67e2013-05-05 23:46:22 +0000726 EShMessages messages = EShMsgDefault;
John Kessenichb0a7eb52013-11-07 17:44:20 +0000727 SetMessageOptions(messages);
John Kessenich69f4b512013-09-04 21:19:27 +0000728
John Kessenich94a81fb2013-08-31 02:41:30 +0000729 for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) {
730 for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) {
John Kessenich26ad2682014-08-13 20:17:19 +0000731 //ret = ShCompile(compiler, shaderStrings, NumShaderStrings, lengths, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenichca3457f2015-05-18 01:59:45 +0000732 ret = ShCompile(compiler, shaderStrings, NumShaderStrings, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenichea869fb2013-10-28 18:12:06 +0000733 //const char* multi[12] = { "# ve", "rsion", " 300 e", "s", "\n#err",
734 // "or should be l", "ine 1", "string 5\n", "float glo", "bal",
735 // ";\n#error should be line 2\n void main() {", "global = 2.3;}" };
John Kessenich41cf6b52013-06-25 18:10:05 +0000736 //const char* multi[7] = { "/", "/", "\\", "\n", "\n", "#", "version 300 es" };
John Kessenichca3457f2015-05-18 01:59:45 +0000737 //ret = ShCompile(compiler, multi, 7, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages);
John Kessenich41cf6b52013-06-25 18:10:05 +0000738 }
John Kessenicha0af4732012-12-12 21:15:54 +0000739
John Kessenich94a81fb2013-08-31 02:41:30 +0000740 if (Options & EOptionMemoryLeakMode)
John Kessenich2b07c7e2013-07-31 18:44:13 +0000741 glslang::OS_DumpMemoryCounters();
John Kessenicha0af4732012-12-12 21:15:54 +0000742 }
John Kessenicha0af4732012-12-12 21:15:54 +0000743
John Kessenich41cf6b52013-06-25 18:10:05 +0000744 delete [] lengths;
745 FreeFileData(shaderStrings);
John Kessenicha0af4732012-12-12 21:15:54 +0000746
John Kessenichc999ba22013-11-07 23:33:24 +0000747 if (ret == 0)
748 CompileFailed = true;
John Kessenicha0af4732012-12-12 21:15:54 +0000749}
750
John Kessenicha0af4732012-12-12 21:15:54 +0000751//
752// print usage to stdout
753//
754void usage()
755{
John Kessenich319de232013-12-04 04:43:40 +0000756 printf("Usage: glslangValidator [option]... [file]...\n"
757 "\n"
John Kessenich0df0cde2015-03-03 17:09:43 +0000758 "Where: each 'file' ends in .<stage>, where <stage> is one of\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600759 " .conf to provide an optional config file that replaces the default configuration\n"
760 " (see -c option below for generating a template)\n"
761 " .vert for a vertex shader\n"
762 " .tesc for a tessellation control shader\n"
763 " .tese for a tessellation evaluation shader\n"
764 " .geom for a geometry shader\n"
765 " .frag for a fragment shader\n"
766 " .comp for a compute shader\n"
John Kessenich319de232013-12-04 04:43:40 +0000767 "\n"
John Kessenich4586dbd2013-08-05 15:52:03 +0000768 "Compilation warnings and errors will be printed to stdout.\n"
John Kessenich319de232013-12-04 04:43:40 +0000769 "\n"
John Kessenich4586dbd2013-08-05 15:52:03 +0000770 "To get other information, use one of the following options:\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600771 "Each option must be specified separately.\n"
772 " -V create SPIR-V binary, under Vulkan semantics; turns on -l;\n"
773 " default file name is <stage>.spv (-o overrides this)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600774 " -G create SPIR-V binary, under OpenGL semantics; turns on -l;\n"
775 " default file name is <stage>.spv (-o overrides this)\n"
776 " -H print human readable form of SPIR-V; turns on -V\n"
Andrew Woloszynaae1ad82015-06-24 17:00:46 -0400777 " -E print pre-processed GLSL; cannot be used with -l;\n"
778 " errors will appear on stderr.\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600779 " -c configuration dump;\n"
780 " creates the default configuration file (redirect to a .conf file)\n"
John Kessenicha86836e2016-07-09 14:50:57 -0600781 " -C cascading errors; risks crashes from accumulation of error recoveries\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600782 " -d default to desktop (#version 110) when there is no shader #version\n"
783 " (default is ES version 100)\n"
John Kessenich66e2faf2016-03-12 18:34:36 -0700784 " -D input is HLSL\n"
John Kessenich4d65ee32016-03-12 18:17:47 -0700785 " -e specify entry-point name\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600786 " -h print this usage message\n"
787 " -i intermediate tree (glslang AST) is printed out\n"
788 " -l link all input files together to form a single module\n"
789 " -m memory leak mode\n"
Johannes van Waverenecb0f3b2016-05-27 12:55:53 -0500790 " -o <file> save binary to <file>, requires a binary option (e.g., -V)\n"
John Kessenich68d78fd2015-07-12 19:28:10 -0600791 " -q dump reflection query database\n"
792 " -r relaxed semantic error-checking mode\n"
793 " -s silent mode\n"
794 " -t multi-threaded mode\n"
795 " -v print version strings\n"
796 " -w suppress warnings (except as required by #extension : warn)\n"
Johannes van Waveren1fd01752016-05-31 08:39:41 -0500797 " -x save 32-bit hexadecimal numbers as text, requires a binary option (e.g., -V)\n"
John Kessenichb0a7eb52013-11-07 17:44:20 +0000798 );
John Kessenich68d78fd2015-07-12 19:28:10 -0600799
800 exit(EFailUsage);
John Kessenicha0af4732012-12-12 21:15:54 +0000801}
802
John Kessenich3ce4e592014-10-06 19:57:34 +0000803#if !defined _MSC_VER && !defined MINGW_HAS_SECURE_API
John Kessenichcfd643e2013-03-08 23:14:42 +0000804
805#include <errno.h>
806
807int fopen_s(
808 FILE** pFile,
John Kessenich51cdd902014-02-18 23:37:57 +0000809 const char* filename,
810 const char* mode
John Kessenichcfd643e2013-03-08 23:14:42 +0000811)
812{
813 if (!pFile || !filename || !mode) {
814 return EINVAL;
815 }
816
817 FILE* f = fopen(filename, mode);
818 if (! f) {
819 if (errno != 0) {
820 return errno;
821 } else {
822 return ENOENT;
823 }
824 }
825 *pFile = f;
826
827 return 0;
828}
829
830#endif
831
John Kessenicha0af4732012-12-12 21:15:54 +0000832//
833// Malloc a string of sufficient size and read a string into it.
834//
John Kessenich51cdd902014-02-18 23:37:57 +0000835char** ReadFileData(const char* fileName)
John Kessenicha0af4732012-12-12 21:15:54 +0000836{
John Kessenichb3297152015-07-11 18:01:03 -0600837 FILE *in = nullptr;
John Kessenich3ce4e592014-10-06 19:57:34 +0000838 int errorCode = fopen_s(&in, fileName, "r");
John Kessenichd6c72a42014-08-18 19:42:35 +0000839
John Kessenicha0af4732012-12-12 21:15:54 +0000840 int count = 0;
John Kessenichb3297152015-07-11 18:01:03 -0600841 const int maxSourceStrings = 5; // for testing splitting shader/tokens across multiple strings
842 char** return_data = (char**)malloc(sizeof(char *) * (maxSourceStrings+1)); // freed in FreeFileData()
John Kessenicha0af4732012-12-12 21:15:54 +0000843
John Kessenich68d78fd2015-07-12 19:28:10 -0600844 if (errorCode || in == nullptr)
845 Error("unable to open input file");
John Kessenicha0af4732012-12-12 21:15:54 +0000846
847 while (fgetc(in) != EOF)
848 count++;
849
John Kessenichd6c72a42014-08-18 19:42:35 +0000850 fseek(in, 0, SEEK_SET);
John Kessenichca3457f2015-05-18 01:59:45 +0000851
John Kessenichb3297152015-07-11 18:01:03 -0600852 char *fdata = (char*)malloc(count+2); // freed before return of this function
John Kessenich68d78fd2015-07-12 19:28:10 -0600853 if (! fdata)
854 Error("can't allocate memory");
855
John Kessenichb3297152015-07-11 18:01:03 -0600856 if ((int)fread(fdata, 1, count, in) != count) {
John Kessenichb3297152015-07-11 18:01:03 -0600857 free(fdata);
John Kessenich68d78fd2015-07-12 19:28:10 -0600858 Error("can't read input file");
John Kessenicha0af4732012-12-12 21:15:54 +0000859 }
John Kessenich68d78fd2015-07-12 19:28:10 -0600860
John Kessenicha0af4732012-12-12 21:15:54 +0000861 fdata[count] = '\0';
862 fclose(in);
John Kessenichb3297152015-07-11 18:01:03 -0600863
John Kessenichea869fb2013-10-28 18:12:06 +0000864 if (count == 0) {
John Kessenichb3297152015-07-11 18:01:03 -0600865 // recover from empty file
866 return_data[0] = (char*)malloc(count+2); // freed in FreeFileData()
John Kessenicha0af4732012-12-12 21:15:54 +0000867 return_data[0][0]='\0';
John Kessenichea869fb2013-10-28 18:12:06 +0000868 NumShaderStrings = 0;
John Kessenichb3297152015-07-11 18:01:03 -0600869 free(fdata);
John Kessenicha0af4732012-12-12 21:15:54 +0000870
John Kessenichb3297152015-07-11 18:01:03 -0600871 return return_data;
872 } else
873 NumShaderStrings = 1; // Set to larger than 1 for testing multiple strings
874
875 // compute how to split up the file into multiple strings, for testing multiple strings
John Kessenichd6c72a42014-08-18 19:42:35 +0000876 int len = (int)(ceil)((float)count/(float)NumShaderStrings);
John Kessenichb3297152015-07-11 18:01:03 -0600877 int ptr_len = 0;
878 int i = 0;
879 while (count > 0) {
880 return_data[i] = (char*)malloc(len + 2); // freed in FreeFileData()
881 memcpy(return_data[i], fdata + ptr_len, len);
882 return_data[i][len] = '\0';
883 count -= len;
884 ptr_len += len;
885 if (count < len) {
886 if (count == 0) {
887 NumShaderStrings = i + 1;
John Kessenicha0af4732012-12-12 21:15:54 +0000888 break;
889 }
John Kessenichb3297152015-07-11 18:01:03 -0600890 len = count;
John Kessenichd6c72a42014-08-18 19:42:35 +0000891 }
892 ++i;
893 }
John Kessenichb3297152015-07-11 18:01:03 -0600894
895 free(fdata);
896
John Kessenicha0af4732012-12-12 21:15:54 +0000897 return return_data;
898}
899
John Kessenich51cdd902014-02-18 23:37:57 +0000900void FreeFileData(char** data)
John Kessenicha0af4732012-12-12 21:15:54 +0000901{
John Kessenichb3297152015-07-11 18:01:03 -0600902 for(int i = 0; i < NumShaderStrings; i++)
John Kessenicha0af4732012-12-12 21:15:54 +0000903 free(data[i]);
John Kessenichb3297152015-07-11 18:01:03 -0600904
905 free(data);
John Kessenicha0af4732012-12-12 21:15:54 +0000906}
907
John Kessenich54d8cda2013-02-11 22:36:01 +0000908void InfoLogMsg(const char* msg, const char* name, const int num)
John Kessenicha0af4732012-12-12 21:15:54 +0000909{
John Kessenichfae38ee2015-06-10 23:23:12 +0000910 if (num >= 0 )
911 printf("#### %s %s %d INFO LOG ####\n", msg, name, num);
912 else
913 printf("#### %s %s INFO LOG ####\n", msg, name);
John Kessenicha0af4732012-12-12 21:15:54 +0000914}