John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1 | // |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 2 | // Copyright (C) 2002-2005 3Dlabs Inc. Ltd. |
| 3 | // Copyright (C) 2013-2016 LunarG, Inc. |
John Kessenich | bd0747d | 2013-02-17 06:01:50 +0000 | [diff] [blame] | 4 | // |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 5 | // All rights reserved. |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 6 | // |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 7 | // Redistribution and use in source and binary forms, with or without |
| 8 | // modification, are permitted provided that the following conditions |
| 9 | // are met: |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 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 | // |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 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. |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 35 | // |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 36 | |
| 37 | // this only applies to the standalone wrapper, not the front end in general |
Aras Pranckevicius | 8e204b2 | 2017-05-10 16:52:50 +0300 | [diff] [blame] | 38 | #ifndef _CRT_SECURE_NO_WARNINGS |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 39 | #define _CRT_SECURE_NO_WARNINGS |
Aras Pranckevicius | 8e204b2 | 2017-05-10 16:52:50 +0300 | [diff] [blame] | 40 | #endif |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 41 | |
Lei Zhang | 8a9b1ee | 2016-05-19 13:31:43 -0400 | [diff] [blame] | 42 | #include "ResourceLimits.h" |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 43 | #include "Worklist.h" |
John Kessenich | 3494b4d | 2017-05-22 15:00:42 -0600 | [diff] [blame] | 44 | #include "DirStackFileIncluder.h" |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 45 | #include "./../glslang/Include/ShHandle.h" |
John Kessenich | 0da9eaa | 2015-08-01 17:10:02 -0600 | [diff] [blame] | 46 | #include "./../glslang/Include/revision.h" |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 47 | #include "./../glslang/Public/ShaderLang.h" |
John Kessenich | 0df0cde | 2015-03-03 17:09:43 +0000 | [diff] [blame] | 48 | #include "../SPIRV/GlslangToSpv.h" |
John Kessenich | 5e4b124 | 2015-08-06 22:53:06 -0600 | [diff] [blame] | 49 | #include "../SPIRV/GLSL.std.450.h" |
John Kessenich | acba772 | 2015-03-04 03:48:38 +0000 | [diff] [blame] | 50 | #include "../SPIRV/doc.h" |
| 51 | #include "../SPIRV/disassemble.h" |
John Kessenich | 3494b4d | 2017-05-22 15:00:42 -0600 | [diff] [blame] | 52 | |
John Kessenich | 66ec80e | 2016-08-05 14:04:23 -0600 | [diff] [blame] | 53 | #include <cstring> |
| 54 | #include <cstdlib> |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 55 | #include <cctype> |
John Kessenich | 66ec80e | 2016-08-05 14:04:23 -0600 | [diff] [blame] | 56 | #include <cmath> |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 57 | #include <array> |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 58 | #include <map> |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 59 | #include <memory> |
| 60 | #include <thread> |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 61 | |
baldurk | 876a0e3 | 2015-11-16 18:03:28 +0100 | [diff] [blame] | 62 | #include "../glslang/OSDependent/osinclude.h" |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 63 | |
| 64 | extern "C" { |
| 65 | SH_IMPORT_EXPORT void ShOutputHtml(); |
| 66 | } |
| 67 | |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 68 | // Command-line options |
| 69 | enum TOptions { |
John Kessenich | 906cc21 | 2016-12-09 19:22:20 -0700 | [diff] [blame] | 70 | 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-lunarg | e0b9deb | 2016-09-16 13:26:37 -0600 | [diff] [blame] | 91 | EOptionFlattenUniformArrays = (1 << 20), |
John Kessenich | 906cc21 | 2016-12-09 19:22:20 -0700 | [diff] [blame] | 92 | EOptionNoStorageFormat = (1 << 21), |
John Kessenich | 20f01e7 | 2016-12-12 11:41:43 -0700 | [diff] [blame] | 93 | EOptionKeepUncalled = (1 << 22), |
John Kessenich | 4f1403e | 2017-04-05 17:38:20 -0600 | [diff] [blame] | 94 | EOptionHlslOffsets = (1 << 23), |
steve-lunarg | be28355 | 2017-04-18 12:18:01 -0600 | [diff] [blame] | 95 | EOptionHlslIoMapping = (1 << 24), |
John Kessenich | 71facdf | 2017-05-17 18:28:19 -0600 | [diff] [blame] | 96 | EOptionAutoMapLocations = (1 << 25), |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 97 | EOptionDebug = (1 << 26), |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 98 | EOptionStdin = (1 << 27), |
GregF | cd1f169 | 2017-09-21 18:40:22 -0600 | [diff] [blame] | 99 | EOptionOptimizeDisable = (1 << 28), |
| 100 | EOptionOptimizeSize = (1 << 29), |
LoopDawg | b22c069 | 2017-12-06 16:52:03 -0700 | [diff] [blame] | 101 | EOptionInvertY = (1 << 30), |
John Kessenich | c6c80a6 | 2018-03-05 22:23:17 -0700 | [diff] [blame] | 102 | EOptionDumpBareVersion = (1 << 31), |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 103 | }; |
John Kessenich | 5d610ee | 2018-03-07 18:05:55 -0700 | [diff] [blame] | 104 | bool targetHlslFunctionality1 = false; |
John Kessenich | e215622 | 2018-06-11 18:12:15 -0600 | [diff] [blame] | 105 | bool SpvToolsDisassembler = false; |
John Kessenich | c340425 | 2018-08-23 15:29:08 -0600 | [diff] [blame] | 106 | bool SpvToolsValidate = false; |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 107 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 108 | // |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 109 | // Return codes from main/exit(). |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 110 | // |
| 111 | enum TFailCode { |
| 112 | ESuccess = 0, |
| 113 | EFailUsage, |
| 114 | EFailCompile, |
| 115 | EFailLink, |
| 116 | EFailCompilerCreate, |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 117 | EFailThreadCreate, |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 118 | EFailLinkerCreate |
| 119 | }; |
| 120 | |
| 121 | // |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 122 | // Forward declarations. |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 123 | // |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 124 | EShLanguage FindLanguage(const std::string& name, bool parseSuffix=true); |
John Kessenich | 51cdd90 | 2014-02-18 23:37:57 +0000 | [diff] [blame] | 125 | void CompileFile(const char* fileName, ShHandle); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 126 | void usage(); |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 127 | char* ReadFileData(const char* fileName); |
| 128 | void FreeFileData(char* data); |
John Kessenich | 54d8cda | 2013-02-11 22:36:01 +0000 | [diff] [blame] | 129 | void InfoLogMsg(const char* msg, const char* name, const int num); |
John Kessenich | 41cf6b5 | 2013-06-25 18:10:05 +0000 | [diff] [blame] | 130 | |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 131 | // Globally track if any compile or link failure. |
| 132 | bool CompileFailed = false; |
| 133 | bool LinkFailed = false; |
| 134 | |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 135 | // array of unique places to leave the shader names and infologs for the asynchronous compiles |
| 136 | std::vector<std::unique_ptr<glslang::TWorkItem>> WorkItems; |
| 137 | |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 138 | TBuiltInResource Resources; |
| 139 | std::string ConfigFile; |
| 140 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 141 | // |
John Kessenich | f0bcb0a | 2016-04-02 13:09:14 -0600 | [diff] [blame] | 142 | // Parse either a .conf file provided by the user or the default from glslang::DefaultTBuiltInResource |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 143 | // |
| 144 | void ProcessConfigFile() |
John Kessenich | b51f62c | 2013-04-11 16:31:09 +0000 | [diff] [blame] | 145 | { |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 146 | if (ConfigFile.size() == 0) |
Lei Zhang | 414eb60 | 2016-03-04 16:22:34 -0500 | [diff] [blame] | 147 | Resources = glslang::DefaultTBuiltInResource; |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 148 | else { |
| 149 | char* configString = ReadFileData(ConfigFile.c_str()); |
| 150 | glslang::DecodeResourceLimits(&Resources, configString); |
| 151 | FreeFileData(configString); |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 152 | } |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 153 | } |
| 154 | |
baldurk | 6d47785 | 2019-01-29 15:45:56 +0000 | [diff] [blame] | 155 | int ReflectOptions = EShReflectionDefault; |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 156 | int Options = 0; |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 157 | const char* ExecutableName = nullptr; |
| 158 | const char* binaryFileName = nullptr; |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 159 | const char* entryPointName = nullptr; |
steve-lunarg | f1e0c87 | 2016-10-31 15:13:43 -0600 | [diff] [blame] | 160 | const char* sourceEntryPointName = nullptr; |
Dan Baker | c6ede89 | 2016-08-11 14:06:06 -0400 | [diff] [blame] | 161 | const char* shaderStageName = nullptr; |
Flavio | aea3c89 | 2017-02-06 11:46:35 -0800 | [diff] [blame] | 162 | const char* variableName = nullptr; |
Rex Xu | cb61eec | 2018-03-07 13:10:01 +0800 | [diff] [blame] | 163 | bool HlslEnable16BitTypes = false; |
John Kessenich | bd1c183 | 2018-12-07 18:38:26 -0700 | [diff] [blame] | 164 | bool HlslDX9compatible = false; |
John Kessenich | 971a0a8 | 2017-06-07 15:06:58 -0600 | [diff] [blame] | 165 | std::vector<std::string> IncludeDirectoryList; |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 166 | |
| 167 | // Source environment |
| 168 | // (source 'Client' is currently the same as target 'Client') |
| 169 | int ClientInputSemanticsVersion = 100; |
| 170 | |
| 171 | // Target environment |
| 172 | glslang::EShClient Client = glslang::EShClientNone; // will stay EShClientNone if only validating |
| 173 | glslang::EShTargetClientVersion ClientVersion; // not valid until Client is set |
| 174 | glslang::EShTargetLanguage TargetLanguage = glslang::EShTargetNone; |
| 175 | glslang::EShTargetLanguageVersion TargetVersion; // not valid until TargetLanguage is set |
| 176 | |
John Kessenich | 66011cb | 2018-03-06 16:12:04 -0700 | [diff] [blame] | 177 | std::vector<std::string> Processes; // what should be recorded by OpModuleProcessed, or equivalent |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 178 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 179 | // Per descriptor-set binding base data |
| 180 | typedef std::map<unsigned int, unsigned int> TPerSetBaseBinding; |
| 181 | |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 182 | std::vector<std::pair<std::string, int>> uniformLocationOverrides; |
Neil Roberts | b0f3d79 | 2018-03-20 17:41:05 +0100 | [diff] [blame] | 183 | int uniformBase = 0; |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 184 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 185 | std::array<std::array<unsigned int, EShLangCount>, glslang::EResCount> baseBinding; |
| 186 | std::array<std::array<TPerSetBaseBinding, EShLangCount>, glslang::EResCount> baseBindingForSet; |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 187 | std::array<std::vector<std::string>, EShLangCount> baseResourceSetBinding; |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 188 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 189 | // Add things like "#define ..." to a preamble to use in the beginning of the shader. |
| 190 | class TPreamble { |
| 191 | public: |
| 192 | TPreamble() { } |
| 193 | |
| 194 | bool isSet() const { return text.size() > 0; } |
| 195 | const char* get() const { return text.c_str(); } |
| 196 | |
| 197 | // #define... |
| 198 | void addDef(std::string def) |
| 199 | { |
| 200 | text.append("#define "); |
| 201 | fixLine(def); |
| 202 | |
John Kessenich | 2a27116 | 2017-07-20 20:00:36 -0600 | [diff] [blame] | 203 | Processes.push_back("D"); |
| 204 | Processes.back().append(def); |
| 205 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 206 | // The first "=" needs to turn into a space |
LoopDawg | b97b25e | 2017-07-12 09:04:39 -0600 | [diff] [blame] | 207 | const size_t equal = def.find_first_of("="); |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 208 | if (equal != def.npos) |
| 209 | def[equal] = ' '; |
| 210 | |
| 211 | text.append(def); |
| 212 | text.append("\n"); |
| 213 | } |
| 214 | |
| 215 | // #undef... |
| 216 | void addUndef(std::string undef) |
| 217 | { |
| 218 | text.append("#undef "); |
| 219 | fixLine(undef); |
John Kessenich | 2a27116 | 2017-07-20 20:00:36 -0600 | [diff] [blame] | 220 | |
| 221 | Processes.push_back("U"); |
| 222 | Processes.back().append(undef); |
| 223 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 224 | text.append(undef); |
| 225 | text.append("\n"); |
| 226 | } |
| 227 | |
| 228 | protected: |
| 229 | void fixLine(std::string& line) |
| 230 | { |
| 231 | // Can't go past a newline in the line |
LoopDawg | b97b25e | 2017-07-12 09:04:39 -0600 | [diff] [blame] | 232 | const size_t end = line.find_first_of("\n"); |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 233 | if (end != line.npos) |
| 234 | line = line.substr(0, end); |
| 235 | } |
| 236 | |
| 237 | std::string text; // contents of preamble |
| 238 | }; |
| 239 | |
| 240 | TPreamble UserPreamble; |
| 241 | |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 242 | // |
| 243 | // Create the default name for saving a binary if -o is not provided. |
| 244 | // |
| 245 | const char* GetBinaryName(EShLanguage stage) |
| 246 | { |
| 247 | const char* name; |
| 248 | if (binaryFileName == nullptr) { |
| 249 | switch (stage) { |
| 250 | case EShLangVertex: name = "vert.spv"; break; |
| 251 | case EShLangTessControl: name = "tesc.spv"; break; |
| 252 | case EShLangTessEvaluation: name = "tese.spv"; break; |
| 253 | case EShLangGeometry: name = "geom.spv"; break; |
| 254 | case EShLangFragment: name = "frag.spv"; break; |
| 255 | case EShLangCompute: name = "comp.spv"; break; |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 256 | #ifdef NV_EXTENSIONS |
Chao Chen | b50c02e | 2018-09-19 11:42:24 -0700 | [diff] [blame] | 257 | case EShLangRayGenNV: name = "rgen.spv"; break; |
| 258 | case EShLangIntersectNV: name = "rint.spv"; break; |
| 259 | case EShLangAnyHitNV: name = "rahit.spv"; break; |
| 260 | case EShLangClosestHitNV: name = "rchit.spv"; break; |
| 261 | case EShLangMissNV: name = "rmiss.spv"; break; |
| 262 | case EShLangCallableNV: name = "rcall.spv"; break; |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 263 | case EShLangMeshNV: name = "mesh.spv"; break; |
| 264 | case EShLangTaskNV: name = "task.spv"; break; |
| 265 | #endif |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 266 | default: name = "unknown"; break; |
| 267 | } |
| 268 | } else |
| 269 | name = binaryFileName; |
| 270 | |
| 271 | return name; |
| 272 | } |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 273 | |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 274 | // |
| 275 | // *.conf => this is a config file that can set limits/resources |
| 276 | // |
| 277 | bool SetConfigFile(const std::string& name) |
| 278 | { |
| 279 | if (name.size() < 5) |
| 280 | return false; |
| 281 | |
John Kessenich | 4c70685 | 2013-10-11 16:28:43 +0000 | [diff] [blame] | 282 | if (name.compare(name.size() - 5, 5, ".conf") == 0) { |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 283 | ConfigFile = name; |
| 284 | return true; |
| 285 | } |
| 286 | |
| 287 | return false; |
| 288 | } |
| 289 | |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 290 | // |
| 291 | // Give error and exit with failure code. |
| 292 | // |
| 293 | void Error(const char* message) |
| 294 | { |
John Kessenich | af52799 | 2017-11-02 22:48:15 -0600 | [diff] [blame] | 295 | fprintf(stderr, "%s: Error %s (use -h for usage)\n", ExecutableName, message); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 296 | exit(EFailUsage); |
| 297 | } |
| 298 | |
| 299 | // |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 300 | // Process an optional binding base of one the forms: |
| 301 | // --argname [stage] base // base for stage (if given) or all stages (if not) |
LoopDawg | e570955 | 2017-10-21 10:46:39 -0600 | [diff] [blame] | 302 | // --argname [stage] [base set]... // set/base pairs: set the base for given binding set. |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 303 | |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 304 | // Where stage is one of the forms accepted by FindLanguage, and base is an integer |
| 305 | // |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 306 | void ProcessBindingBase(int& argc, char**& argv, glslang::TResourceType res) |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 307 | { |
| 308 | if (argc < 2) |
| 309 | usage(); |
| 310 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 311 | EShLanguage lang = EShLangCount; |
| 312 | int singleBase = 0; |
| 313 | TPerSetBaseBinding perSetBase; |
| 314 | int arg = 1; |
| 315 | |
| 316 | // Parse stage, if given |
| 317 | if (!isdigit(argv[arg][0])) { |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 318 | if (argc < 3) // this form needs one more argument |
| 319 | usage(); |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 320 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 321 | lang = FindLanguage(argv[arg++], false); |
| 322 | } |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 323 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 324 | if ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) { |
| 325 | // Parse a per-set binding base |
| 326 | while ((argc - arg) > 2 && isdigit(argv[arg+0][0]) && isdigit(argv[arg+1][0])) { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 327 | const int baseNum = atoi(argv[arg++]); |
LoopDawg | e570955 | 2017-10-21 10:46:39 -0600 | [diff] [blame] | 328 | const int setNum = atoi(argv[arg++]); |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 329 | perSetBase[setNum] = baseNum; |
| 330 | } |
| 331 | } else { |
| 332 | // Parse single binding base |
| 333 | singleBase = atoi(argv[arg++]); |
| 334 | } |
| 335 | |
| 336 | argc -= (arg-1); |
| 337 | argv += (arg-1); |
| 338 | |
| 339 | // Set one or all languages |
| 340 | const int langMin = (lang < EShLangCount) ? lang+0 : 0; |
| 341 | const int langMax = (lang < EShLangCount) ? lang+1 : EShLangCount; |
| 342 | |
| 343 | for (int lang = langMin; lang < langMax; ++lang) { |
| 344 | if (!perSetBase.empty()) |
John Kessenich | 251901a | 2018-08-12 22:05:59 -0600 | [diff] [blame] | 345 | baseBindingForSet[res][lang].insert(perSetBase.begin(), perSetBase.end()); |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 346 | else |
| 347 | baseBinding[res][lang] = singleBase; |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 348 | } |
| 349 | } |
| 350 | |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 351 | void ProcessResourceSetBindingBase(int& argc, char**& argv, std::array<std::vector<std::string>, EShLangCount>& base) |
| 352 | { |
| 353 | if (argc < 2) |
| 354 | usage(); |
| 355 | |
| 356 | if (!isdigit(argv[1][0])) { |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 357 | if (argc < 3) // this form needs one more argument |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 358 | usage(); |
| 359 | |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 360 | // Parse form: --argname stage [regname set base...], or: |
| 361 | // --argname stage set |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 362 | const EShLanguage lang = FindLanguage(argv[1], false); |
| 363 | |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 364 | argc--; |
| 365 | argv++; |
| 366 | |
| 367 | while (argc > 1 && argv[1] != nullptr && argv[1][0] != '-') { |
| 368 | base[lang].push_back(argv[1]); |
| 369 | |
| 370 | argc--; |
| 371 | argv++; |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 372 | } |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 373 | |
| 374 | // Must have one arg, or a multiple of three (for [regname set binding] triples) |
| 375 | if (base[lang].size() != 1 && (base[lang].size() % 3) != 0) |
| 376 | usage(); |
| 377 | |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 378 | } else { |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 379 | // Parse form: --argname set |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 380 | for (int lang=0; lang<EShLangCount; ++lang) |
| 381 | base[lang].push_back(argv[1]); |
| 382 | |
| 383 | argc--; |
| 384 | argv++; |
| 385 | } |
| 386 | } |
| 387 | |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 388 | // |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 389 | // Do all command-line argument parsing. This includes building up the work-items |
| 390 | // to be processed later, and saving all the command-line options. |
| 391 | // |
| 392 | // Does not return (it exits) if command-line is fatally flawed. |
| 393 | // |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 394 | void ProcessArguments(std::vector<std::unique_ptr<glslang::TWorkItem>>& workItems, int argc, char* argv[]) |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 395 | { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 396 | for (int res = 0; res < glslang::EResCount; ++res) |
| 397 | baseBinding[res].fill(0); |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 398 | |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 399 | ExecutableName = argv[0]; |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 400 | workItems.reserve(argc); |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 401 | |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 402 | const auto bumpArg = [&]() { |
| 403 | if (argc > 0) { |
| 404 | argc--; |
| 405 | argv++; |
| 406 | } |
| 407 | }; |
| 408 | |
| 409 | // read a string directly attached to a single-letter option |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 410 | const auto getStringOperand = [&](const char* desc) { |
| 411 | if (argv[0][2] == 0) { |
| 412 | printf("%s must immediately follow option (no spaces)\n", desc); |
| 413 | exit(EFailUsage); |
| 414 | } |
| 415 | return argv[0] + 2; |
| 416 | }; |
| 417 | |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 418 | // read a number attached to a single-letter option |
| 419 | const auto getAttachedNumber = [&](const char* desc) { |
| 420 | int num = atoi(argv[0] + 2); |
| 421 | if (num == 0) { |
| 422 | printf("%s: expected attached non-0 number\n", desc); |
| 423 | exit(EFailUsage); |
| 424 | } |
| 425 | return num; |
| 426 | }; |
| 427 | |
| 428 | // minimum needed (without overriding something else) to target Vulkan SPIR-V |
| 429 | const auto setVulkanSpv = []() { |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 430 | if (Client == glslang::EShClientNone) |
| 431 | ClientVersion = glslang::EShTargetVulkan_1_0; |
| 432 | Client = glslang::EShClientVulkan; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 433 | Options |= EOptionSpv; |
| 434 | Options |= EOptionVulkanRules; |
| 435 | Options |= EOptionLinkProgram; |
| 436 | }; |
| 437 | |
| 438 | // minimum needed (without overriding something else) to target OpenGL SPIR-V |
| 439 | const auto setOpenGlSpv = []() { |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 440 | if (Client == glslang::EShClientNone) |
| 441 | ClientVersion = glslang::EShTargetOpenGL_450; |
| 442 | Client = glslang::EShClientOpenGL; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 443 | Options |= EOptionSpv; |
| 444 | Options |= EOptionLinkProgram; |
| 445 | // undo a -H default to Vulkan |
| 446 | Options &= ~EOptionVulkanRules; |
| 447 | }; |
| 448 | |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 449 | const auto getUniformOverride = [getStringOperand]() { |
| 450 | const char *arg = getStringOperand("-u<name>:<location>"); |
| 451 | const char *split = strchr(arg, ':'); |
| 452 | if (split == NULL) { |
| 453 | printf("%s: missing location\n", arg); |
| 454 | exit(EFailUsage); |
| 455 | } |
| 456 | errno = 0; |
| 457 | int location = ::strtol(split + 1, NULL, 10); |
| 458 | if (errno) { |
| 459 | printf("%s: invalid location\n", arg); |
| 460 | exit(EFailUsage); |
| 461 | } |
| 462 | return std::make_pair(std::string(arg, split - arg), location); |
| 463 | }; |
| 464 | |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 465 | for (bumpArg(); argc >= 1; bumpArg()) { |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 466 | if (argv[0][0] == '-') { |
John Kessenich | 2aa7f3a | 2015-05-15 16:02:07 +0000 | [diff] [blame] | 467 | switch (argv[0][1]) { |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 468 | case '-': |
| 469 | { |
| 470 | std::string lowerword(argv[0]+2); |
| 471 | std::transform(lowerword.begin(), lowerword.end(), lowerword.begin(), ::tolower); |
| 472 | |
| 473 | // handle --word style options |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 474 | if (lowerword == "auto-map-bindings" || // synonyms |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 475 | lowerword == "auto-map-binding" || |
| 476 | lowerword == "amb") { |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 477 | Options |= EOptionAutoMapBindings; |
| 478 | } else if (lowerword == "auto-map-locations" || // synonyms |
| 479 | lowerword == "aml") { |
| 480 | Options |= EOptionAutoMapLocations; |
Neil Roberts | b0f3d79 | 2018-03-20 17:41:05 +0100 | [diff] [blame] | 481 | } else if (lowerword == "uniform-base") { |
| 482 | if (argc <= 1) |
| 483 | Error("no <base> provided for --uniform-base"); |
| 484 | uniformBase = ::strtol(argv[1], NULL, 10); |
| 485 | bumpArg(); |
| 486 | break; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 487 | } else if (lowerword == "client") { |
| 488 | if (argc > 1) { |
| 489 | if (strcmp(argv[1], "vulkan100") == 0) |
| 490 | setVulkanSpv(); |
| 491 | else if (strcmp(argv[1], "opengl100") == 0) |
| 492 | setOpenGlSpv(); |
| 493 | else |
| 494 | Error("--client expects vulkan100 or opengl100"); |
| 495 | } |
| 496 | bumpArg(); |
John Kessenich | 640bd09 | 2018-08-09 14:15:00 -0600 | [diff] [blame] | 497 | } else if (lowerword == "entry-point") { |
| 498 | entryPointName = argv[1]; |
| 499 | if (argc <= 1) |
| 500 | Error("no <name> provided for --entry-point"); |
| 501 | bumpArg(); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 502 | } else if (lowerword == "flatten-uniform-arrays" || // synonyms |
| 503 | lowerword == "flatten-uniform-array" || |
| 504 | lowerword == "fua") { |
| 505 | Options |= EOptionFlattenUniformArrays; |
| 506 | } else if (lowerword == "hlsl-offsets") { |
| 507 | Options |= EOptionHlslOffsets; |
| 508 | } else if (lowerword == "hlsl-iomap" || |
| 509 | lowerword == "hlsl-iomapper" || |
| 510 | lowerword == "hlsl-iomapping") { |
| 511 | Options |= EOptionHlslIoMapping; |
Rex Xu | cb61eec | 2018-03-07 13:10:01 +0800 | [diff] [blame] | 512 | } else if (lowerword == "hlsl-enable-16bit-types") { |
| 513 | HlslEnable16BitTypes = true; |
John Kessenich | bd1c183 | 2018-12-07 18:38:26 -0700 | [diff] [blame] | 514 | } else if (lowerword == "hlsl-dx9-compatible") { |
| 515 | HlslDX9compatible = true; |
John Kessenich | c6c80a6 | 2018-03-05 22:23:17 -0700 | [diff] [blame] | 516 | } else if (lowerword == "invert-y" || // synonyms |
| 517 | lowerword == "iy") { |
| 518 | Options |= EOptionInvertY; |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 519 | } else if (lowerword == "keep-uncalled" || // synonyms |
| 520 | lowerword == "ku") { |
| 521 | Options |= EOptionKeepUncalled; |
| 522 | } else if (lowerword == "no-storage-format" || // synonyms |
| 523 | lowerword == "nsf") { |
| 524 | Options |= EOptionNoStorageFormat; |
John Kessenich | 2a27116 | 2017-07-20 20:00:36 -0600 | [diff] [blame] | 525 | } else if (lowerword == "relaxed-errors") { |
| 526 | Options |= EOptionRelaxedErrors; |
baldurk | 15c37f7 | 2019-01-29 12:12:59 +0000 | [diff] [blame] | 527 | } else if (lowerword == "reflect-strict-array-suffix") { |
| 528 | ReflectOptions |= EShReflectionStrictArraySuffix; |
baldurk | edf8212 | 2019-01-29 15:49:00 +0000 | [diff] [blame] | 529 | } else if (lowerword == "reflect-basic-array-suffix") { |
| 530 | ReflectOptions |= EShReflectionBasicArraySuffix; |
baldurk | 0af5e3e | 2019-01-29 16:04:44 +0000 | [diff] [blame] | 531 | } else if (lowerword == "reflect-intermediate-io") { |
| 532 | ReflectOptions |= EShReflectionIntermediateIO; |
baldurk | 657acc0 | 2019-01-30 14:18:43 +0000 | [diff] [blame^] | 533 | } else if (lowerword == "reflect-separate-buffers") { |
| 534 | ReflectOptions |= EShReflectionSeparateBuffers; |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 535 | } else if (lowerword == "resource-set-bindings" || // synonyms |
| 536 | lowerword == "resource-set-binding" || |
| 537 | lowerword == "rsb") { |
| 538 | ProcessResourceSetBindingBase(argc, argv, baseResourceSetBinding); |
steve-lunarg | 9088be4 | 2016-11-01 10:31:42 -0600 | [diff] [blame] | 539 | } else if (lowerword == "shift-image-bindings" || // synonyms |
| 540 | lowerword == "shift-image-binding" || |
| 541 | lowerword == "sib") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 542 | ProcessBindingBase(argc, argv, glslang::EResImage); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 543 | } else if (lowerword == "shift-sampler-bindings" || // synonyms |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 544 | lowerword == "shift-sampler-binding" || |
| 545 | lowerword == "ssb") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 546 | ProcessBindingBase(argc, argv, glslang::EResSampler); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 547 | } else if (lowerword == "shift-uav-bindings" || // synonyms |
| 548 | lowerword == "shift-uav-binding" || |
| 549 | lowerword == "suavb") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 550 | ProcessBindingBase(argc, argv, glslang::EResUav); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 551 | } else if (lowerword == "shift-texture-bindings" || // synonyms |
| 552 | lowerword == "shift-texture-binding" || |
| 553 | lowerword == "stb") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 554 | ProcessBindingBase(argc, argv, glslang::EResTexture); |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 555 | } else if (lowerword == "shift-ubo-bindings" || // synonyms |
| 556 | lowerword == "shift-ubo-binding" || |
steve-lunarg | be28355 | 2017-04-18 12:18:01 -0600 | [diff] [blame] | 557 | lowerword == "shift-cbuffer-bindings" || |
| 558 | lowerword == "shift-cbuffer-binding" || |
| 559 | lowerword == "sub" || |
| 560 | lowerword == "scb") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 561 | ProcessBindingBase(argc, argv, glslang::EResUbo); |
steve-lunarg | 932bb5c | 2017-02-21 17:19:08 -0700 | [diff] [blame] | 562 | } else if (lowerword == "shift-ssbo-bindings" || // synonyms |
| 563 | lowerword == "shift-ssbo-binding" || |
| 564 | lowerword == "sbb") { |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 565 | ProcessBindingBase(argc, argv, glslang::EResSsbo); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 566 | } else if (lowerword == "source-entrypoint" || // synonyms |
| 567 | lowerword == "sep") { |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 568 | if (argc <= 1) |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 569 | Error("no <entry-point> provided for --source-entrypoint"); |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 570 | sourceEntryPointName = argv[1]; |
| 571 | bumpArg(); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 572 | break; |
John Kessenich | e215622 | 2018-06-11 18:12:15 -0600 | [diff] [blame] | 573 | } else if (lowerword == "spirv-dis") { |
| 574 | SpvToolsDisassembler = true; |
John Kessenich | c340425 | 2018-08-23 15:29:08 -0600 | [diff] [blame] | 575 | } else if (lowerword == "spirv-val") { |
| 576 | SpvToolsValidate = true; |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 577 | } else if (lowerword == "stdin") { |
| 578 | Options |= EOptionStdin; |
| 579 | shaderStageName = argv[1]; |
John Kessenich | 2a27116 | 2017-07-20 20:00:36 -0600 | [diff] [blame] | 580 | } else if (lowerword == "suppress-warnings") { |
| 581 | Options |= EOptionSuppressWarnings; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 582 | } else if (lowerword == "target-env") { |
| 583 | if (argc > 1) { |
| 584 | if (strcmp(argv[1], "vulkan1.0") == 0) { |
| 585 | setVulkanSpv(); |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 586 | ClientVersion = glslang::EShTargetVulkan_1_0; |
John Kessenich | 66011cb | 2018-03-06 16:12:04 -0700 | [diff] [blame] | 587 | } else if (strcmp(argv[1], "vulkan1.1") == 0) { |
| 588 | setVulkanSpv(); |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 589 | ClientVersion = glslang::EShTargetVulkan_1_1; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 590 | } else if (strcmp(argv[1], "opengl") == 0) { |
| 591 | setOpenGlSpv(); |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 592 | ClientVersion = glslang::EShTargetOpenGL_450; |
| 593 | } else if (strcmp(argv[1], "spirv1.0") == 0) { |
| 594 | TargetLanguage = glslang::EShTargetSpv; |
| 595 | TargetVersion = glslang::EShTargetSpv_1_0; |
| 596 | } else if (strcmp(argv[1], "spirv1.1") == 0) { |
| 597 | TargetLanguage = glslang::EShTargetSpv; |
| 598 | TargetVersion = glslang::EShTargetSpv_1_1; |
| 599 | } else if (strcmp(argv[1], "spirv1.2") == 0) { |
| 600 | TargetLanguage = glslang::EShTargetSpv; |
| 601 | TargetVersion = glslang::EShTargetSpv_1_2; |
| 602 | } else if (strcmp(argv[1], "spirv1.3") == 0) { |
| 603 | TargetLanguage = glslang::EShTargetSpv; |
| 604 | TargetVersion = glslang::EShTargetSpv_1_3; |
| 605 | } else if (strcmp(argv[1], "spirv1.4") == 0) { |
| 606 | TargetLanguage = glslang::EShTargetSpv; |
| 607 | TargetVersion = glslang::EShTargetSpv_1_4; |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 608 | } else |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 609 | Error("--target-env expected one of: vulkan1.0, vulkan1.1, opengl, spirv1.0, spirv1.1, spirv1.2, or spirv1.3"); |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 610 | } |
| 611 | bumpArg(); |
Flavio | 15017db | 2017-02-15 14:29:33 -0800 | [diff] [blame] | 612 | } else if (lowerword == "variable-name" || // synonyms |
John Kessenich | 66011cb | 2018-03-06 16:12:04 -0700 | [diff] [blame] | 613 | lowerword == "vn") { |
Flavio | 15017db | 2017-02-15 14:29:33 -0800 | [diff] [blame] | 614 | Options |= EOptionOutputHexadecimal; |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 615 | if (argc <= 1) |
Flavio | 15017db | 2017-02-15 14:29:33 -0800 | [diff] [blame] | 616 | Error("no <C-variable-name> provided for --variable-name"); |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 617 | variableName = argv[1]; |
| 618 | bumpArg(); |
Flavio | 15017db | 2017-02-15 14:29:33 -0800 | [diff] [blame] | 619 | break; |
John Kessenich | c6c80a6 | 2018-03-05 22:23:17 -0700 | [diff] [blame] | 620 | } else if (lowerword == "version") { |
| 621 | Options |= EOptionDumpVersions; |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 622 | } else { |
| 623 | usage(); |
| 624 | } |
| 625 | } |
| 626 | break; |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 627 | case 'C': |
| 628 | Options |= EOptionCascadingErrors; |
| 629 | break; |
| 630 | case 'D': |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 631 | if (argv[0][2] == 0) |
| 632 | Options |= EOptionReadHlsl; |
| 633 | else |
| 634 | UserPreamble.addDef(getStringOperand("-D<macro> macro name")); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 635 | break; |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 636 | case 'u': |
| 637 | uniformLocationOverrides.push_back(getUniformOverride()); |
| 638 | break; |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 639 | case 'E': |
| 640 | Options |= EOptionOutputPreprocessed; |
| 641 | break; |
| 642 | case 'G': |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 643 | // OpenGL client |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 644 | setOpenGlSpv(); |
| 645 | if (argv[0][2] != 0) |
| 646 | ClientInputSemanticsVersion = getAttachedNumber("-G<num> client input semantics"); |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 647 | break; |
John Kessenich | 2aa7f3a | 2015-05-15 16:02:07 +0000 | [diff] [blame] | 648 | case 'H': |
| 649 | Options |= EOptionHumanReadableSpv; |
John Kessenich | 91e4aa5 | 2016-07-07 17:46:42 -0600 | [diff] [blame] | 650 | if ((Options & EOptionSpv) == 0) { |
| 651 | // default to Vulkan |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 652 | setVulkanSpv(); |
John Kessenich | 91e4aa5 | 2016-07-07 17:46:42 -0600 | [diff] [blame] | 653 | } |
| 654 | break; |
John Kessenich | 971a0a8 | 2017-06-07 15:06:58 -0600 | [diff] [blame] | 655 | case 'I': |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 656 | IncludeDirectoryList.push_back(getStringOperand("-I<dir> include path")); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 657 | break; |
GregF | cd1f169 | 2017-09-21 18:40:22 -0600 | [diff] [blame] | 658 | case 'O': |
| 659 | if (argv[0][2] == 'd') |
| 660 | Options |= EOptionOptimizeDisable; |
| 661 | else if (argv[0][2] == 's') |
GregF | fb03a55 | 2018-03-29 11:49:14 -0600 | [diff] [blame] | 662 | #if ENABLE_OPT |
GregF | cd1f169 | 2017-09-21 18:40:22 -0600 | [diff] [blame] | 663 | Options |= EOptionOptimizeSize; |
| 664 | #else |
| 665 | Error("-Os not available; optimizer not linked"); |
| 666 | #endif |
| 667 | else |
| 668 | Error("unknown -O option"); |
| 669 | break; |
Dan Baker | 5afdd78 | 2016-08-11 17:53:57 -0400 | [diff] [blame] | 670 | case 'S': |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 671 | if (argc <= 1) |
Dan Baker | 5afdd78 | 2016-08-11 17:53:57 -0400 | [diff] [blame] | 672 | Error("no <stage> specified for -S"); |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 673 | shaderStageName = argv[1]; |
| 674 | bumpArg(); |
dankbaker | 45d49bc | 2016-08-08 21:43:07 -0400 | [diff] [blame] | 675 | break; |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 676 | case 'U': |
| 677 | UserPreamble.addUndef(getStringOperand("-U<macro>: macro name")); |
| 678 | break; |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 679 | case 'V': |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 680 | setVulkanSpv(); |
| 681 | if (argv[0][2] != 0) |
David Neto | 506d2c2 | 2018-02-27 21:55:23 -0500 | [diff] [blame] | 682 | ClientInputSemanticsVersion = getAttachedNumber("-V<num> client input semantics"); |
John Kessenich | c555ddd | 2015-06-17 02:38:44 +0000 | [diff] [blame] | 683 | break; |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 684 | case 'c': |
| 685 | Options |= EOptionDumpConfig; |
| 686 | break; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 687 | case 'd': |
John Kessenich | c6c80a6 | 2018-03-05 22:23:17 -0700 | [diff] [blame] | 688 | if (strncmp(&argv[0][1], "dumpversion", strlen(&argv[0][1]) + 1) == 0 || |
| 689 | strncmp(&argv[0][1], "dumpfullversion", strlen(&argv[0][1]) + 1) == 0) |
| 690 | Options |= EOptionDumpBareVersion; |
| 691 | else |
| 692 | Options |= EOptionDefaultDesktop; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 693 | break; |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 694 | case 'e': |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 695 | entryPointName = argv[1]; |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 696 | if (argc <= 1) |
John Kessenich | a25530c | 2017-09-11 20:13:49 -0600 | [diff] [blame] | 697 | Error("no <name> provided for -e"); |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 698 | bumpArg(); |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 699 | break; |
John Kessenich | 5d610ee | 2018-03-07 18:05:55 -0700 | [diff] [blame] | 700 | case 'f': |
| 701 | if (strcmp(&argv[0][2], "hlsl_functionality1") == 0) |
| 702 | targetHlslFunctionality1 = true; |
| 703 | else |
| 704 | Error("-f: expected hlsl_functionality1"); |
| 705 | break; |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 706 | case 'g': |
| 707 | Options |= EOptionDebug; |
| 708 | break; |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 709 | case 'h': |
| 710 | usage(); |
| 711 | break; |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 712 | case 'i': |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 713 | Options |= EOptionIntermediate; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 714 | break; |
| 715 | case 'l': |
John Kessenich | d78e351 | 2014-08-25 20:07:55 +0000 | [diff] [blame] | 716 | Options |= EOptionLinkProgram; |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 717 | break; |
| 718 | case 'm': |
| 719 | Options |= EOptionMemoryLeakMode; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 720 | break; |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 721 | case 'o': |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 722 | if (argc <= 1) |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 723 | Error("no <file> provided for -o"); |
John Kessenich | c178f0a | 2017-06-23 10:58:31 -0600 | [diff] [blame] | 724 | binaryFileName = argv[1]; |
| 725 | bumpArg(); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 726 | break; |
John Kessenich | 11f9fc7 | 2013-11-07 01:06:34 +0000 | [diff] [blame] | 727 | case 'q': |
| 728 | Options |= EOptionDumpReflection; |
| 729 | break; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 730 | case 'r': |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 731 | Options |= EOptionRelaxedErrors; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 732 | break; |
| 733 | case 's': |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 734 | Options |= EOptionSuppressInfolog; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 735 | break; |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 736 | case 't': |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 737 | Options |= EOptionMultiThreaded; |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 738 | break; |
John Kessenich | 319de23 | 2013-12-04 04:43:40 +0000 | [diff] [blame] | 739 | case 'v': |
| 740 | Options |= EOptionDumpVersions; |
| 741 | break; |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 742 | case 'w': |
| 743 | Options |= EOptionSuppressWarnings; |
| 744 | break; |
Johannes van Waveren | ecb0f3b | 2016-05-27 12:55:53 -0500 | [diff] [blame] | 745 | case 'x': |
| 746 | Options |= EOptionOutputHexadecimal; |
Johannes van Waveren | ecb0f3b | 2016-05-27 12:55:53 -0500 | [diff] [blame] | 747 | break; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 748 | default: |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 749 | usage(); |
| 750 | break; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 751 | } |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 752 | } else { |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 753 | std::string name(argv[0]); |
| 754 | if (! SetConfigFile(name)) { |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 755 | workItems.push_back(std::unique_ptr<glslang::TWorkItem>(new glslang::TWorkItem(name))); |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 756 | } |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 757 | } |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 758 | } |
| 759 | |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 760 | // Make sure that -S is always specified if --stdin is specified |
| 761 | if ((Options & EOptionStdin) && shaderStageName == nullptr) |
| 762 | Error("must provide -S when --stdin is given"); |
| 763 | |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 764 | // Make sure that -E is not specified alongside linking (which includes SPV generation) |
| 765 | if ((Options & EOptionOutputPreprocessed) && (Options & EOptionLinkProgram)) |
| 766 | Error("can't use -E when linking is selected"); |
John Kessenich | c555ddd | 2015-06-17 02:38:44 +0000 | [diff] [blame] | 767 | |
Johannes van Waveren | ecb0f3b | 2016-05-27 12:55:53 -0500 | [diff] [blame] | 768 | // -o or -x makes no sense if there is no target binary |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 769 | if (binaryFileName && (Options & EOptionSpv) == 0) |
| 770 | Error("no binary generation requested (e.g., -V)"); |
steve-lunarg | e0b9deb | 2016-09-16 13:26:37 -0600 | [diff] [blame] | 771 | |
| 772 | if ((Options & EOptionFlattenUniformArrays) != 0 && |
| 773 | (Options & EOptionReadHlsl) == 0) |
| 774 | Error("uniform array flattening only valid when compiling HLSL source."); |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 775 | |
| 776 | // rationalize client and target language |
| 777 | if (TargetLanguage == glslang::EShTargetNone) { |
| 778 | switch (ClientVersion) { |
| 779 | case glslang::EShTargetVulkan_1_0: |
| 780 | TargetLanguage = glslang::EShTargetSpv; |
| 781 | TargetVersion = glslang::EShTargetSpv_1_0; |
| 782 | break; |
| 783 | case glslang::EShTargetVulkan_1_1: |
| 784 | TargetLanguage = glslang::EShTargetSpv; |
| 785 | TargetVersion = glslang::EShTargetSpv_1_3; |
| 786 | break; |
| 787 | case glslang::EShTargetOpenGL_450: |
| 788 | TargetLanguage = glslang::EShTargetSpv; |
| 789 | TargetVersion = glslang::EShTargetSpv_1_0; |
| 790 | break; |
| 791 | default: |
| 792 | break; |
| 793 | } |
| 794 | } |
| 795 | if (TargetLanguage != glslang::EShTargetNone && Client == glslang::EShClientNone) |
| 796 | Error("To generate SPIR-V, also specify client semantics. See -G and -V."); |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 797 | } |
| 798 | |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 799 | // |
| 800 | // Translate the meaningful subset of command-line options to parser-behavior options. |
| 801 | // |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 802 | void SetMessageOptions(EShMessages& messages) |
| 803 | { |
| 804 | if (Options & EOptionRelaxedErrors) |
| 805 | messages = (EShMessages)(messages | EShMsgRelaxedErrors); |
| 806 | if (Options & EOptionIntermediate) |
| 807 | messages = (EShMessages)(messages | EShMsgAST); |
| 808 | if (Options & EOptionSuppressWarnings) |
| 809 | messages = (EShMessages)(messages | EShMsgSuppressWarnings); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 810 | if (Options & EOptionSpv) |
| 811 | messages = (EShMessages)(messages | EShMsgSpvRules); |
| 812 | if (Options & EOptionVulkanRules) |
| 813 | messages = (EShMessages)(messages | EShMsgVulkanRules); |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 814 | if (Options & EOptionOutputPreprocessed) |
| 815 | messages = (EShMessages)(messages | EShMsgOnlyPreprocessor); |
John Kessenich | 66e2faf | 2016-03-12 18:34:36 -0700 | [diff] [blame] | 816 | if (Options & EOptionReadHlsl) |
| 817 | messages = (EShMessages)(messages | EShMsgReadHlsl); |
John Kessenich | a86836e | 2016-07-09 14:50:57 -0600 | [diff] [blame] | 818 | if (Options & EOptionCascadingErrors) |
| 819 | messages = (EShMessages)(messages | EShMsgCascadingErrors); |
John Kessenich | 906cc21 | 2016-12-09 19:22:20 -0700 | [diff] [blame] | 820 | if (Options & EOptionKeepUncalled) |
| 821 | messages = (EShMessages)(messages | EShMsgKeepUncalled); |
John Kessenich | 4f1403e | 2017-04-05 17:38:20 -0600 | [diff] [blame] | 822 | if (Options & EOptionHlslOffsets) |
| 823 | messages = (EShMessages)(messages | EShMsgHlslOffsets); |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 824 | if (Options & EOptionDebug) |
| 825 | messages = (EShMessages)(messages | EShMsgDebugInfo); |
Rex Xu | cb61eec | 2018-03-07 13:10:01 +0800 | [diff] [blame] | 826 | if (HlslEnable16BitTypes) |
| 827 | messages = (EShMessages)(messages | EShMsgHlslEnable16BitTypes); |
GregF | fb03a55 | 2018-03-29 11:49:14 -0600 | [diff] [blame] | 828 | if ((Options & EOptionOptimizeDisable) || !ENABLE_OPT) |
| 829 | messages = (EShMessages)(messages | EShMsgHlslLegalization); |
John Kessenich | bd1c183 | 2018-12-07 18:38:26 -0700 | [diff] [blame] | 830 | if (HlslDX9compatible) |
| 831 | messages = (EShMessages)(messages | EShMsgHlslDX9Compatible); |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 832 | } |
| 833 | |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 834 | // |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 835 | // Thread entry point, for non-linking asynchronous mode. |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 836 | // |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 837 | void CompileShaders(glslang::TWorklist& worklist) |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 838 | { |
John Kessenich | 7f0bcfd | 2018-04-05 19:00:01 -0600 | [diff] [blame] | 839 | if (Options & EOptionDebug) |
| 840 | Error("cannot generate debug information unless linking to generate code"); |
| 841 | |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 842 | glslang::TWorkItem* workItem; |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 843 | if (Options & EOptionStdin) { |
John Kessenich | e7f9cae | 2018-07-12 15:11:07 -0600 | [diff] [blame] | 844 | if (worklist.remove(workItem)) { |
| 845 | ShHandle compiler = ShConstructCompiler(FindLanguage("stdin"), Options); |
| 846 | if (compiler == nullptr) |
| 847 | return; |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 848 | |
John Kessenich | e7f9cae | 2018-07-12 15:11:07 -0600 | [diff] [blame] | 849 | CompileFile("stdin", compiler); |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 850 | |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 851 | if (! (Options & EOptionSuppressInfolog)) |
| 852 | workItem->results = ShGetInfoLog(compiler); |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 853 | |
John Kessenich | e7f9cae | 2018-07-12 15:11:07 -0600 | [diff] [blame] | 854 | ShDestruct(compiler); |
| 855 | } |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 856 | } else { |
| 857 | while (worklist.remove(workItem)) { |
| 858 | ShHandle compiler = ShConstructCompiler(FindLanguage(workItem->name), Options); |
| 859 | if (compiler == 0) |
| 860 | return; |
| 861 | |
| 862 | CompileFile(workItem->name.c_str(), compiler); |
| 863 | |
| 864 | if (! (Options & EOptionSuppressInfolog)) |
| 865 | workItem->results = ShGetInfoLog(compiler); |
| 866 | |
| 867 | ShDestruct(compiler); |
| 868 | } |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 869 | } |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 870 | } |
| 871 | |
John Kessenich | 6626cad | 2015-06-19 05:14:19 +0000 | [diff] [blame] | 872 | // Outputs the given string, but only if it is non-null and non-empty. |
| 873 | // This prevents erroneous newlines from appearing. |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 874 | void PutsIfNonEmpty(const char* str) |
John Kessenich | 6626cad | 2015-06-19 05:14:19 +0000 | [diff] [blame] | 875 | { |
| 876 | if (str && str[0]) { |
| 877 | puts(str); |
| 878 | } |
| 879 | } |
| 880 | |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 881 | // Outputs the given string to stderr, but only if it is non-null and non-empty. |
| 882 | // This prevents erroneous newlines from appearing. |
| 883 | void StderrIfNonEmpty(const char* str) |
| 884 | { |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 885 | if (str && str[0]) |
| 886 | fprintf(stderr, "%s\n", str); |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 887 | } |
| 888 | |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 889 | // Simple bundling of what makes a compilation unit for ease in passing around, |
| 890 | // and separation of handling file IO versus API (programmatic) compilation. |
| 891 | struct ShaderCompUnit { |
| 892 | EShLanguage stage; |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 893 | static const int maxCount = 1; |
| 894 | int count; // live number of strings/names |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 895 | const char* text[maxCount]; // memory owned/managed externally |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 896 | std::string fileName[maxCount]; // hold's the memory, but... |
| 897 | const char* fileNameList[maxCount]; // downstream interface wants pointers |
dankbaker | afe6e9c | 2016-08-21 12:29:08 -0400 | [diff] [blame] | 898 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 899 | ShaderCompUnit(EShLanguage stage) : stage(stage), count(0) { } |
dankbaker | afe6e9c | 2016-08-21 12:29:08 -0400 | [diff] [blame] | 900 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 901 | ShaderCompUnit(const ShaderCompUnit& rhs) |
dankbaker | afe6e9c | 2016-08-21 12:29:08 -0400 | [diff] [blame] | 902 | { |
| 903 | stage = rhs.stage; |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 904 | count = rhs.count; |
| 905 | for (int i = 0; i < count; ++i) { |
| 906 | fileName[i] = rhs.fileName[i]; |
| 907 | text[i] = rhs.text[i]; |
| 908 | fileNameList[i] = rhs.fileName[i].c_str(); |
| 909 | } |
dankbaker | afe6e9c | 2016-08-21 12:29:08 -0400 | [diff] [blame] | 910 | } |
| 911 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 912 | void addString(std::string& ifileName, const char* itext) |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 913 | { |
| 914 | assert(count < maxCount); |
| 915 | fileName[count] = ifileName; |
| 916 | text[count] = itext; |
| 917 | fileNameList[count] = fileName[count].c_str(); |
| 918 | ++count; |
| 919 | } |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 920 | }; |
| 921 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 922 | // |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 923 | // For linking mode: Will independently parse each compilation unit, but then put them |
| 924 | // in the same program and link them together, making at most one linked module per |
| 925 | // pipeline stage. |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 926 | // |
| 927 | // Uses the new C++ interface instead of the old handle-based interface. |
| 928 | // |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 929 | |
| 930 | void CompileAndLinkShaderUnits(std::vector<ShaderCompUnit> compUnits) |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 931 | { |
| 932 | // keep track of what to free |
| 933 | std::list<glslang::TShader*> shaders; |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 934 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 935 | EShMessages messages = EShMsgDefault; |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 936 | SetMessageOptions(messages); |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 937 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 938 | // |
| 939 | // Per-shader processing... |
| 940 | // |
| 941 | |
John Kessenich | 5b0f13a | 2013-11-01 03:08:40 +0000 | [diff] [blame] | 942 | glslang::TProgram& program = *new glslang::TProgram; |
rdb | 32084e8 | 2016-02-23 22:17:38 +0100 | [diff] [blame] | 943 | for (auto it = compUnits.cbegin(); it != compUnits.cend(); ++it) { |
| 944 | const auto &compUnit = *it; |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 945 | glslang::TShader* shader = new glslang::TShader(compUnit.stage); |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 946 | shader->setStringsWithLengthsAndNames(compUnit.text, NULL, compUnit.fileNameList, compUnit.count); |
John Kessenich | 640bd09 | 2018-08-09 14:15:00 -0600 | [diff] [blame] | 947 | if (entryPointName) |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 948 | shader->setEntryPoint(entryPointName); |
John Kessenich | 3693e63 | 2017-09-29 17:51:52 -0600 | [diff] [blame] | 949 | if (sourceEntryPointName) { |
| 950 | if (entryPointName == nullptr) |
| 951 | printf("Warning: Changing source entry point name without setting an entry-point name.\n" |
| 952 | "Use '-e <name>'.\n"); |
steve-lunarg | f1e0c87 | 2016-10-31 15:13:43 -0600 | [diff] [blame] | 953 | shader->setSourceEntryPoint(sourceEntryPointName); |
John Kessenich | 3693e63 | 2017-09-29 17:51:52 -0600 | [diff] [blame] | 954 | } |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 955 | if (UserPreamble.isSet()) |
| 956 | shader->setPreamble(UserPreamble.get()); |
John Kessenich | 2a27116 | 2017-07-20 20:00:36 -0600 | [diff] [blame] | 957 | shader->addProcesses(Processes); |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 958 | |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 959 | // Set IO mapper binding shift values |
| 960 | for (int r = 0; r < glslang::EResCount; ++r) { |
| 961 | const glslang::TResourceType res = glslang::TResourceType(r); |
| 962 | |
| 963 | // Set base bindings |
| 964 | shader->setShiftBinding(res, baseBinding[res][compUnit.stage]); |
| 965 | |
| 966 | // Set bindings for particular resource sets |
| 967 | // TODO: use a range based for loop here, when available in all environments. |
| 968 | for (auto i = baseBindingForSet[res][compUnit.stage].begin(); |
| 969 | i != baseBindingForSet[res][compUnit.stage].end(); ++i) |
LoopDawg | e570955 | 2017-10-21 10:46:39 -0600 | [diff] [blame] | 970 | shader->setShiftBindingForSet(res, i->second, i->first); |
LoopDawg | 08a1442 | 2017-10-17 19:27:14 -0600 | [diff] [blame] | 971 | } |
| 972 | |
steve-lunarg | e0b9deb | 2016-09-16 13:26:37 -0600 | [diff] [blame] | 973 | shader->setFlattenUniformArrays((Options & EOptionFlattenUniformArrays) != 0); |
steve-lunarg | cce8d48 | 2016-10-14 18:36:42 -0600 | [diff] [blame] | 974 | shader->setNoStorageFormat((Options & EOptionNoStorageFormat) != 0); |
Hyangran Park | 36dc829 | 2017-05-02 16:27:29 +0900 | [diff] [blame] | 975 | shader->setResourceSetBinding(baseResourceSetBinding[compUnit.stage]); |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 976 | |
steve-lunarg | be28355 | 2017-04-18 12:18:01 -0600 | [diff] [blame] | 977 | if (Options & EOptionHlslIoMapping) |
| 978 | shader->setHlslIoMapping(true); |
| 979 | |
steve-lunarg | 7f7c2ed | 2016-09-07 15:20:19 -0600 | [diff] [blame] | 980 | if (Options & EOptionAutoMapBindings) |
| 981 | shader->setAutoMapBindings(true); |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 982 | |
John Kessenich | 71facdf | 2017-05-17 18:28:19 -0600 | [diff] [blame] | 983 | if (Options & EOptionAutoMapLocations) |
| 984 | shader->setAutoMapLocations(true); |
| 985 | |
LoopDawg | b22c069 | 2017-12-06 16:52:03 -0700 | [diff] [blame] | 986 | if (Options & EOptionInvertY) |
| 987 | shader->setInvertY(true); |
| 988 | |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 989 | for (auto& uniOverride : uniformLocationOverrides) { |
| 990 | shader->addUniformLocationOverride(uniOverride.first.c_str(), |
| 991 | uniOverride.second); |
| 992 | } |
| 993 | |
Neil Roberts | b0f3d79 | 2018-03-20 17:41:05 +0100 | [diff] [blame] | 994 | shader->setUniformLocationBase(uniformBase); |
| 995 | |
John Kessenich | 4be4aeb | 2017-06-23 10:50:22 -0600 | [diff] [blame] | 996 | // Set up the environment, some subsettings take precedence over earlier |
| 997 | // ways of setting things. |
| 998 | if (Options & EOptionSpv) { |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 999 | shader->setEnvInput((Options & EOptionReadHlsl) ? glslang::EShSourceHlsl |
| 1000 | : glslang::EShSourceGlsl, |
| 1001 | compUnit.stage, Client, ClientInputSemanticsVersion); |
| 1002 | shader->setEnvClient(Client, ClientVersion); |
| 1003 | shader->setEnvTarget(TargetLanguage, TargetVersion); |
John Kessenich | 5d610ee | 2018-03-07 18:05:55 -0700 | [diff] [blame] | 1004 | if (targetHlslFunctionality1) |
| 1005 | shader->setEnvTargetHlslFunctionality1(); |
John Kessenich | 4be4aeb | 2017-06-23 10:50:22 -0600 | [diff] [blame] | 1006 | } |
| 1007 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1008 | shaders.push_back(shader); |
John Kessenich | b329715 | 2015-07-11 18:01:03 -0600 | [diff] [blame] | 1009 | |
John Kessenich | 4be4aeb | 2017-06-23 10:50:22 -0600 | [diff] [blame] | 1010 | const int defaultVersion = Options & EOptionDefaultDesktop ? 110 : 100; |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1011 | |
John Kessenich | 3494b4d | 2017-05-22 15:00:42 -0600 | [diff] [blame] | 1012 | DirStackFileIncluder includer; |
John Kessenich | 971a0a8 | 2017-06-07 15:06:58 -0600 | [diff] [blame] | 1013 | std::for_each(IncludeDirectoryList.rbegin(), IncludeDirectoryList.rend(), [&includer](const std::string& dir) { |
| 1014 | includer.pushExternalLocalDirectory(dir); }); |
John Kessenich | c555ddd | 2015-06-17 02:38:44 +0000 | [diff] [blame] | 1015 | if (Options & EOptionOutputPreprocessed) { |
| 1016 | std::string str; |
John Kessenich | 086febc | 2018-10-25 12:43:02 -0600 | [diff] [blame] | 1017 | if (shader->preprocess(&Resources, defaultVersion, ENoProfile, false, false, messages, &str, includer)) { |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 1018 | PutsIfNonEmpty(str.c_str()); |
| 1019 | } else { |
| 1020 | CompileFailed = true; |
| 1021 | } |
| 1022 | StderrIfNonEmpty(shader->getInfoLog()); |
| 1023 | StderrIfNonEmpty(shader->getInfoDebugLog()); |
John Kessenich | c555ddd | 2015-06-17 02:38:44 +0000 | [diff] [blame] | 1024 | continue; |
| 1025 | } |
John Kessenich | 086febc | 2018-10-25 12:43:02 -0600 | [diff] [blame] | 1026 | |
John Kessenich | 3494b4d | 2017-05-22 15:00:42 -0600 | [diff] [blame] | 1027 | if (! shader->parse(&Resources, defaultVersion, false, messages, includer)) |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1028 | CompileFailed = true; |
John Kessenich | c555ddd | 2015-06-17 02:38:44 +0000 | [diff] [blame] | 1029 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1030 | program.addShader(shader); |
| 1031 | |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1032 | if (! (Options & EOptionSuppressInfolog) && |
| 1033 | ! (Options & EOptionMemoryLeakMode)) { |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1034 | PutsIfNonEmpty(compUnit.fileName[0].c_str()); |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 1035 | PutsIfNonEmpty(shader->getInfoLog()); |
| 1036 | PutsIfNonEmpty(shader->getInfoDebugLog()); |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1037 | } |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1038 | } |
| 1039 | |
| 1040 | // |
| 1041 | // Program-level processing... |
| 1042 | // |
| 1043 | |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 1044 | // Link |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1045 | if (! (Options & EOptionOutputPreprocessed) && ! program.link(messages)) |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1046 | LinkFailed = true; |
| 1047 | |
steve-lunarg | 9ae3474 | 2016-10-05 13:40:13 -0600 | [diff] [blame] | 1048 | // Map IO |
| 1049 | if (Options & EOptionSpv) { |
| 1050 | if (!program.mapIO()) |
| 1051 | LinkFailed = true; |
| 1052 | } |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 1053 | |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 1054 | // Report |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1055 | if (! (Options & EOptionSuppressInfolog) && |
| 1056 | ! (Options & EOptionMemoryLeakMode)) { |
Andrew Woloszyn | aae1ad8 | 2015-06-24 17:00:46 -0400 | [diff] [blame] | 1057 | PutsIfNonEmpty(program.getInfoLog()); |
| 1058 | PutsIfNonEmpty(program.getInfoDebugLog()); |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1059 | } |
| 1060 | |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 1061 | // Reflect |
John Kessenich | 11f9fc7 | 2013-11-07 01:06:34 +0000 | [diff] [blame] | 1062 | if (Options & EOptionDumpReflection) { |
baldurk | 6d47785 | 2019-01-29 15:45:56 +0000 | [diff] [blame] | 1063 | program.buildReflection(ReflectOptions); |
John Kessenich | 11f9fc7 | 2013-11-07 01:06:34 +0000 | [diff] [blame] | 1064 | program.dumpReflection(); |
| 1065 | } |
| 1066 | |
John Kessenich | 4d65ee3 | 2016-03-12 18:17:47 -0700 | [diff] [blame] | 1067 | // Dump SPIR-V |
John Kessenich | 0df0cde | 2015-03-03 17:09:43 +0000 | [diff] [blame] | 1068 | if (Options & EOptionSpv) { |
John Kessenich | 92f9038 | 2014-07-28 04:21:04 +0000 | [diff] [blame] | 1069 | if (CompileFailed || LinkFailed) |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1070 | printf("SPIR-V is not generated for failed compile or link\n"); |
John Kessenich | 92f9038 | 2014-07-28 04:21:04 +0000 | [diff] [blame] | 1071 | else { |
| 1072 | for (int stage = 0; stage < EShLangCount; ++stage) { |
John Kessenich | a7a68a9 | 2014-08-24 18:21:00 +0000 | [diff] [blame] | 1073 | if (program.getIntermediate((EShLanguage)stage)) { |
John Kessenich | 0df0cde | 2015-03-03 17:09:43 +0000 | [diff] [blame] | 1074 | std::vector<unsigned int> spirv; |
Lei Zhang | 09caf12 | 2016-05-02 18:11:54 -0400 | [diff] [blame] | 1075 | std::string warningsErrors; |
Lei Zhang | 17535f7 | 2016-05-04 15:55:59 -0400 | [diff] [blame] | 1076 | spv::SpvBuildLogger logger; |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 1077 | glslang::SpvOptions spvOptions; |
| 1078 | if (Options & EOptionDebug) |
| 1079 | spvOptions.generateDebugInfo = true; |
GregF | 52fe3d5 | 2017-09-28 10:08:32 -0600 | [diff] [blame] | 1080 | spvOptions.disableOptimizer = (Options & EOptionOptimizeDisable) != 0; |
| 1081 | spvOptions.optimizeSize = (Options & EOptionOptimizeSize) != 0; |
John Kessenich | 717c80a | 2018-08-23 15:17:10 -0600 | [diff] [blame] | 1082 | spvOptions.disassemble = SpvToolsDisassembler; |
John Kessenich | c340425 | 2018-08-23 15:29:08 -0600 | [diff] [blame] | 1083 | spvOptions.validate = SpvToolsValidate; |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 1084 | glslang::GlslangToSpv(*program.getIntermediate((EShLanguage)stage), spirv, &logger, &spvOptions); |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1085 | |
| 1086 | // Dump the spv to a file or stdout, etc., but only if not doing |
| 1087 | // memory/perf testing, as it's not internal to programmatic use. |
| 1088 | if (! (Options & EOptionMemoryLeakMode)) { |
Johannes van Waveren | ecb0f3b | 2016-05-27 12:55:53 -0500 | [diff] [blame] | 1089 | printf("%s", logger.getAllMessages().c_str()); |
| 1090 | if (Options & EOptionOutputHexadecimal) { |
John Kessenich | 8f674e8 | 2017-02-18 09:45:40 -0700 | [diff] [blame] | 1091 | glslang::OutputSpvHex(spirv, GetBinaryName((EShLanguage)stage), variableName); |
Johannes van Waveren | ecb0f3b | 2016-05-27 12:55:53 -0500 | [diff] [blame] | 1092 | } else { |
| 1093 | glslang::OutputSpvBin(spirv, GetBinaryName((EShLanguage)stage)); |
| 1094 | } |
John Kessenich | e215622 | 2018-06-11 18:12:15 -0600 | [diff] [blame] | 1095 | if (!SpvToolsDisassembler && (Options & EOptionHumanReadableSpv)) |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1096 | spv::Disassemble(std::cout, spirv); |
John Kessenich | acba772 | 2015-03-04 03:48:38 +0000 | [diff] [blame] | 1097 | } |
John Kessenich | a7a68a9 | 2014-08-24 18:21:00 +0000 | [diff] [blame] | 1098 | } |
John Kessenich | 92f9038 | 2014-07-28 04:21:04 +0000 | [diff] [blame] | 1099 | } |
| 1100 | } |
| 1101 | } |
| 1102 | |
John Kessenich | 5b0f13a | 2013-11-01 03:08:40 +0000 | [diff] [blame] | 1103 | // Free everything up, program has to go before the shaders |
| 1104 | // because it might have merged stuff from the shaders, and |
| 1105 | // the stuff from the shaders has to have its destructors called |
| 1106 | // before the pools holding the memory in the shaders is freed. |
| 1107 | delete &program; |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1108 | while (shaders.size() > 0) { |
| 1109 | delete shaders.back(); |
| 1110 | shaders.pop_back(); |
| 1111 | } |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1112 | } |
| 1113 | |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1114 | // |
| 1115 | // Do file IO part of compile and link, handing off the pure |
| 1116 | // API/programmatic mode to CompileAndLinkShaderUnits(), which can |
| 1117 | // be put in a loop for testing memory footprint and performance. |
| 1118 | // |
| 1119 | // This is just for linking mode: meaning all the shaders will be put into the |
| 1120 | // the same program linked together. |
| 1121 | // |
| 1122 | // This means there are a limited number of work items (not multi-threading mode) |
| 1123 | // and that the point is testing at the linking level. Hence, to enable |
| 1124 | // performance and memory testing, the actual compile/link can be put in |
| 1125 | // a loop, independent of processing the work items and file IO. |
| 1126 | // |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1127 | void CompileAndLinkShaderFiles(glslang::TWorklist& Worklist) |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1128 | { |
| 1129 | std::vector<ShaderCompUnit> compUnits; |
| 1130 | |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1131 | // If this is using stdin, we can't really detect multiple different file |
| 1132 | // units by input type. We need to assume that we're just being given one |
| 1133 | // file of a certain type. |
| 1134 | if ((Options & EOptionStdin) != 0) { |
| 1135 | ShaderCompUnit compUnit(FindLanguage("stdin")); |
| 1136 | std::istreambuf_iterator<char> begin(std::cin), end; |
| 1137 | std::string tempString(begin, end); |
| 1138 | char* fileText = strdup(tempString.c_str()); |
| 1139 | std::string fileName = "stdin"; |
| 1140 | compUnit.addString(fileName, fileText); |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1141 | compUnits.push_back(compUnit); |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1142 | } else { |
| 1143 | // Transfer all the work items from to a simple list of |
| 1144 | // of compilation units. (We don't care about the thread |
| 1145 | // work-item distribution properties in this path, which |
| 1146 | // is okay due to the limited number of shaders, know since |
| 1147 | // they are all getting linked together.) |
| 1148 | glslang::TWorkItem* workItem; |
| 1149 | while (Worklist.remove(workItem)) { |
| 1150 | ShaderCompUnit compUnit(FindLanguage(workItem->name)); |
| 1151 | char* fileText = ReadFileData(workItem->name.c_str()); |
| 1152 | if (fileText == nullptr) |
| 1153 | usage(); |
| 1154 | compUnit.addString(workItem->name, fileText); |
| 1155 | compUnits.push_back(compUnit); |
| 1156 | } |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1157 | } |
| 1158 | |
| 1159 | // Actual call to programmatic processing of compile and link, |
| 1160 | // in a loop for testing memory and performance. This part contains |
| 1161 | // all the perf/memory that a programmatic consumer will care about. |
| 1162 | for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) { |
| 1163 | for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) |
| 1164 | CompileAndLinkShaderUnits(compUnits); |
| 1165 | |
| 1166 | if (Options & EOptionMemoryLeakMode) |
| 1167 | glslang::OS_DumpMemoryCounters(); |
| 1168 | } |
| 1169 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 1170 | // free memory from ReadFileData, which got stored in a const char* |
| 1171 | // as the first string above |
rdb | 32084e8 | 2016-02-23 22:17:38 +0100 | [diff] [blame] | 1172 | for (auto it = compUnits.begin(); it != compUnits.end(); ++it) |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 1173 | FreeFileData(const_cast<char*>(it->text[0])); |
John Kessenich | c57b2a9 | 2016-01-16 15:30:03 -0700 | [diff] [blame] | 1174 | } |
| 1175 | |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 1176 | int singleMain() |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1177 | { |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1178 | glslang::TWorklist workList; |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 1179 | std::for_each(WorkItems.begin(), WorkItems.end(), [&workList](std::unique_ptr<glslang::TWorkItem>& item) { |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1180 | assert(item); |
| 1181 | workList.add(item.get()); |
| 1182 | }); |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1183 | |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 1184 | if (Options & EOptionDumpConfig) { |
Lei Zhang | 414eb60 | 2016-03-04 16:22:34 -0500 | [diff] [blame] | 1185 | printf("%s", glslang::GetDefaultTBuiltInResourceString().c_str()); |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1186 | if (workList.empty()) |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 1187 | return ESuccess; |
| 1188 | } |
| 1189 | |
John Kessenich | c6c80a6 | 2018-03-05 22:23:17 -0700 | [diff] [blame] | 1190 | if (Options & EOptionDumpBareVersion) { |
| 1191 | printf("%d.%d.%d\n", |
| 1192 | glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL); |
| 1193 | if (workList.empty()) |
| 1194 | return ESuccess; |
| 1195 | } else if (Options & EOptionDumpVersions) { |
| 1196 | printf("Glslang Version: %d.%d.%d\n", |
| 1197 | glslang::GetSpirvGeneratorVersion(), GLSLANG_MINOR_VERSION, GLSLANG_PATCH_LEVEL); |
John Kessenich | 319de23 | 2013-12-04 04:43:40 +0000 | [diff] [blame] | 1198 | printf("ESSL Version: %s\n", glslang::GetEsslVersionString()); |
| 1199 | printf("GLSL Version: %s\n", glslang::GetGlslVersionString()); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1200 | std::string spirvVersion; |
| 1201 | glslang::GetSpirvVersion(spirvVersion); |
John Kessenich | 0da9eaa | 2015-08-01 17:10:02 -0600 | [diff] [blame] | 1202 | printf("SPIR-V Version %s\n", spirvVersion.c_str()); |
John Kessenich | 5e4b124 | 2015-08-06 22:53:06 -0600 | [diff] [blame] | 1203 | printf("GLSL.std.450 Version %d, Revision %d\n", GLSLstd450Version, GLSLstd450Revision); |
John Kessenich | 55e7d11 | 2015-11-15 21:33:39 -0700 | [diff] [blame] | 1204 | printf("Khronos Tool ID %d\n", glslang::GetKhronosToolId()); |
John Kessenich | a372a3e | 2017-11-02 22:32:14 -0600 | [diff] [blame] | 1205 | printf("SPIR-V Generator Version %d\n", glslang::GetSpirvGeneratorVersion()); |
John Kessenich | b84313d | 2016-07-20 16:03:29 -0600 | [diff] [blame] | 1206 | printf("GL_KHR_vulkan_glsl version %d\n", 100); |
| 1207 | printf("ARB_GL_gl_spirv version %d\n", 100); |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1208 | if (workList.empty()) |
John Kessenich | 319de23 | 2013-12-04 04:43:40 +0000 | [diff] [blame] | 1209 | return ESuccess; |
| 1210 | } |
| 1211 | |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1212 | if (workList.empty() && ((Options & EOptionStdin) == 0)) { |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 1213 | usage(); |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 1214 | } |
| 1215 | |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1216 | if (Options & EOptionStdin) { |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 1217 | WorkItems.push_back(std::unique_ptr<glslang::TWorkItem>{new glslang::TWorkItem("stdin")}); |
| 1218 | workList.add(WorkItems.back().get()); |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1219 | } |
| 1220 | |
John Kessenich | 05a7063 | 2013-09-17 19:26:08 +0000 | [diff] [blame] | 1221 | ProcessConfigFile(); |
| 1222 | |
John Kessenich | 086febc | 2018-10-25 12:43:02 -0600 | [diff] [blame] | 1223 | if ((Options & EOptionReadHlsl) && !((Options & EOptionOutputPreprocessed) || (Options & EOptionSpv))) |
| 1224 | Error("ERROR: HLSL requires SPIR-V code generation (or preprocessing only)"); |
| 1225 | |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1226 | // |
| 1227 | // Two modes: |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 1228 | // 1) linking all arguments together, single-threaded, new C++ interface |
| 1229 | // 2) independent arguments, can be tackled by multiple asynchronous threads, for testing thread safety, using the old handle interface |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1230 | // |
John Kessenich | 086febc | 2018-10-25 12:43:02 -0600 | [diff] [blame] | 1231 | if (Options & (EOptionLinkProgram | EOptionOutputPreprocessed)) { |
John Kessenich | c36e1d8 | 2013-11-01 17:41:52 +0000 | [diff] [blame] | 1232 | glslang::InitializeProcess(); |
John Kessenich | e2c15b4 | 2017-11-16 22:48:41 -0700 | [diff] [blame] | 1233 | glslang::InitializeProcess(); // also test reference counting of users |
| 1234 | glslang::InitializeProcess(); // also test reference counting of users |
| 1235 | glslang::FinalizeProcess(); // also test reference counting of users |
| 1236 | glslang::FinalizeProcess(); // also test reference counting of users |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1237 | CompileAndLinkShaderFiles(workList); |
John Kessenich | c36e1d8 | 2013-11-01 17:41:52 +0000 | [diff] [blame] | 1238 | glslang::FinalizeProcess(); |
| 1239 | } else { |
| 1240 | ShInitialize(); |
John Kessenich | e2c15b4 | 2017-11-16 22:48:41 -0700 | [diff] [blame] | 1241 | ShInitialize(); // also test reference counting of users |
| 1242 | ShFinalize(); // also test reference counting of users |
John Kessenich | c36e1d8 | 2013-11-01 17:41:52 +0000 | [diff] [blame] | 1243 | |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1244 | bool printShaderNames = workList.size() > 1; |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1245 | |
John Kessenich | e2c15b4 | 2017-11-16 22:48:41 -0700 | [diff] [blame] | 1246 | if (Options & EOptionMultiThreaded) { |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1247 | std::array<std::thread, 16> threads; |
John Kessenich | e2c15b4 | 2017-11-16 22:48:41 -0700 | [diff] [blame] | 1248 | for (unsigned int t = 0; t < threads.size(); ++t) { |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1249 | threads[t] = std::thread(CompileShaders, std::ref(workList)); |
John Kessenich | e2c15b4 | 2017-11-16 22:48:41 -0700 | [diff] [blame] | 1250 | if (threads[t].get_id() == std::thread::id()) { |
John Kessenich | af52799 | 2017-11-02 22:48:15 -0600 | [diff] [blame] | 1251 | fprintf(stderr, "Failed to create thread\n"); |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 1252 | return EFailThreadCreate; |
| 1253 | } |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1254 | } |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1255 | |
| 1256 | std::for_each(threads.begin(), threads.end(), [](std::thread& t) { t.join(); }); |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1257 | } else |
Juan Lopez | a558b26 | 2017-04-02 23:04:00 +0200 | [diff] [blame] | 1258 | CompileShaders(workList); |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 1259 | |
| 1260 | // Print out all the resulting infologs |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 1261 | for (size_t w = 0; w < WorkItems.size(); ++w) { |
| 1262 | if (WorkItems[w]) { |
| 1263 | if (printShaderNames || WorkItems[w]->results.size() > 0) |
| 1264 | PutsIfNonEmpty(WorkItems[w]->name.c_str()); |
| 1265 | PutsIfNonEmpty(WorkItems[w]->results.c_str()); |
John Kessenich | 38f3b89 | 2013-09-06 19:52:57 +0000 | [diff] [blame] | 1266 | } |
| 1267 | } |
John Kessenich | c36e1d8 | 2013-11-01 17:41:52 +0000 | [diff] [blame] | 1268 | |
| 1269 | ShFinalize(); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1270 | } |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1271 | |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1272 | if (CompileFailed) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1273 | return EFailCompile; |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1274 | if (LinkFailed) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1275 | return EFailLink; |
| 1276 | |
| 1277 | return 0; |
| 1278 | } |
| 1279 | |
John Kessenich | 94f28eb | 2017-11-13 01:32:06 -0700 | [diff] [blame] | 1280 | int C_DECL main(int argc, char* argv[]) |
| 1281 | { |
| 1282 | ProcessArguments(WorkItems, argc, argv); |
| 1283 | |
| 1284 | int ret = 0; |
| 1285 | |
| 1286 | // Loop over the entire init/finalize cycle to watch memory changes |
| 1287 | const int iterations = 1; |
| 1288 | if (iterations > 1) |
| 1289 | glslang::OS_DumpMemoryCounters(); |
| 1290 | for (int i = 0; i < iterations; ++i) { |
| 1291 | ret = singleMain(); |
| 1292 | if (iterations > 1) |
| 1293 | glslang::OS_DumpMemoryCounters(); |
| 1294 | } |
| 1295 | |
| 1296 | return ret; |
| 1297 | } |
| 1298 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1299 | // |
| 1300 | // Deduce the language from the filename. Files must end in one of the |
| 1301 | // following extensions: |
| 1302 | // |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1303 | // .vert = vertex |
| 1304 | // .tesc = tessellation control |
| 1305 | // .tese = tessellation evaluation |
| 1306 | // .geom = geometry |
| 1307 | // .frag = fragment |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 1308 | // .comp = compute |
Chao Chen | b50c02e | 2018-09-19 11:42:24 -0700 | [diff] [blame] | 1309 | // .rgen = ray generation |
| 1310 | // .rint = ray intersection |
| 1311 | // .rahit = ray any hit |
| 1312 | // .rchit = ray closest hit |
| 1313 | // .rmiss = ray miss |
| 1314 | // .rcall = ray callable |
Sahil Parmar | 035cbbe | 2018-10-04 16:39:18 -0700 | [diff] [blame] | 1315 | // .mesh = mesh |
| 1316 | // .task = task |
Grigory Dzhavadyan | 3350741 | 2018-04-12 14:35:24 -0700 | [diff] [blame] | 1317 | // Additionally, the file names may end in .<stage>.glsl and .<stage>.hlsl |
| 1318 | // where <stage> is one of the stages listed above. |
| 1319 | // |
| 1320 | EShLanguage FindLanguage(const std::string& name, bool parseStageName) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1321 | { |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1322 | std::string stageName; |
Dan Baker | c6ede89 | 2016-08-11 14:06:06 -0400 | [diff] [blame] | 1323 | if (shaderStageName) |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1324 | stageName = shaderStageName; |
| 1325 | else if (parseStageName) { |
Grigory Dzhavadyan | 3350741 | 2018-04-12 14:35:24 -0700 | [diff] [blame] | 1326 | // Note: "first" extension means "first from the end", i.e. |
| 1327 | // if the file is named foo.vert.glsl, then "glsl" is first, |
| 1328 | // "vert" is second. |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1329 | size_t firstExtStart = name.find_last_of("."); |
| 1330 | bool hasFirstExt = firstExtStart != std::string::npos; |
| 1331 | size_t secondExtStart = hasFirstExt ? name.find_last_of(".", firstExtStart - 1) : std::string::npos; |
| 1332 | bool hasSecondExt = secondExtStart != std::string::npos; |
| 1333 | std::string firstExt = name.substr(firstExtStart + 1, std::string::npos); |
| 1334 | bool usesUnifiedExt = hasFirstExt && (firstExt == "glsl" || firstExt == "hlsl"); |
John Kessenich | 3beac94 | 2018-04-17 21:02:19 -0600 | [diff] [blame] | 1335 | if (usesUnifiedExt && firstExt == "hlsl") |
| 1336 | Options |= EOptionReadHlsl; |
| 1337 | if (hasFirstExt && !usesUnifiedExt) |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1338 | stageName = firstExt; |
John Kessenich | 3beac94 | 2018-04-17 21:02:19 -0600 | [diff] [blame] | 1339 | else if (usesUnifiedExt && hasSecondExt) |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1340 | stageName = name.substr(secondExtStart + 1, firstExtStart - secondExtStart - 1); |
John Kessenich | 3beac94 | 2018-04-17 21:02:19 -0600 | [diff] [blame] | 1341 | else { |
Grigory Dzhavadyan | 3350741 | 2018-04-12 14:35:24 -0700 | [diff] [blame] | 1342 | usage(); |
| 1343 | return EShLangVertex; |
John Kessenich | e751bca | 2017-03-16 11:20:38 -0600 | [diff] [blame] | 1344 | } |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1345 | } else |
| 1346 | stageName = name; |
dankbaker | 45d49bc | 2016-08-08 21:43:07 -0400 | [diff] [blame] | 1347 | |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1348 | if (stageName == "vert") |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1349 | return EShLangVertex; |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1350 | else if (stageName == "tesc") |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1351 | return EShLangTessControl; |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1352 | else if (stageName == "tese") |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1353 | return EShLangTessEvaluation; |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1354 | else if (stageName == "geom") |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1355 | return EShLangGeometry; |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1356 | else if (stageName == "frag") |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1357 | return EShLangFragment; |
John Kessenich | fccbb8b | 2018-04-17 17:24:03 -0600 | [diff] [blame] | 1358 | else if (stageName == "comp") |
John Kessenich | c027579 | 2013-08-09 17:14:49 +0000 | [diff] [blame] | 1359 | return EShLangCompute; |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 1360 | #ifdef NV_EXTENSIONS |
Chao Chen | b50c02e | 2018-09-19 11:42:24 -0700 | [diff] [blame] | 1361 | else if (stageName == "rgen") |
| 1362 | return EShLangRayGenNV; |
| 1363 | else if (stageName == "rint") |
| 1364 | return EShLangIntersectNV; |
| 1365 | else if (stageName == "rahit") |
| 1366 | return EShLangAnyHitNV; |
| 1367 | else if (stageName == "rchit") |
| 1368 | return EShLangClosestHitNV; |
| 1369 | else if (stageName == "rmiss") |
| 1370 | return EShLangMissNV; |
| 1371 | else if (stageName == "rcall") |
| 1372 | return EShLangCallableNV; |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 1373 | else if (stageName == "mesh") |
| 1374 | return EShLangMeshNV; |
| 1375 | else if (stageName == "task") |
| 1376 | return EShLangTaskNV; |
| 1377 | #endif |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1378 | |
| 1379 | usage(); |
John Kessenich | e95ecc5 | 2012-12-12 21:34:14 +0000 | [diff] [blame] | 1380 | return EShLangVertex; |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1381 | } |
| 1382 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1383 | // |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 1384 | // Read a file's data into a string, and compile it using the old interface ShCompile, |
John Kessenich | 69f4b51 | 2013-09-04 21:19:27 +0000 | [diff] [blame] | 1385 | // for non-linkable results. |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1386 | // |
John Kessenich | 51cdd90 | 2014-02-18 23:37:57 +0000 | [diff] [blame] | 1387 | void CompileFile(const char* fileName, ShHandle compiler) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1388 | { |
John Kessenich | ca3457f | 2015-05-18 01:59:45 +0000 | [diff] [blame] | 1389 | int ret = 0; |
Sven-Hendrik Haase | 0dd1285 | 2017-09-02 19:34:54 +0200 | [diff] [blame] | 1390 | char* shaderString; |
| 1391 | if ((Options & EOptionStdin) != 0) { |
| 1392 | std::istreambuf_iterator<char> begin(std::cin), end; |
| 1393 | std::string tempString(begin, end); |
| 1394 | shaderString = strdup(tempString.c_str()); |
| 1395 | } else { |
| 1396 | shaderString = ReadFileData(fileName); |
| 1397 | } |
John Kessenich | 41cf6b5 | 2013-06-25 18:10:05 +0000 | [diff] [blame] | 1398 | |
| 1399 | // move to length-based strings, rather than null-terminated strings |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1400 | int* lengths = new int[1]; |
| 1401 | lengths[0] = (int)strlen(shaderString); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1402 | |
John Kessenich | 52ac67e | 2013-05-05 23:46:22 +0000 | [diff] [blame] | 1403 | EShMessages messages = EShMsgDefault; |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 1404 | SetMessageOptions(messages); |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 1405 | |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 1406 | if (UserPreamble.isSet()) |
| 1407 | Error("-D and -U options require -l (linking)\n"); |
| 1408 | |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 1409 | for (int i = 0; i < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++i) { |
| 1410 | for (int j = 0; j < ((Options & EOptionMemoryLeakMode) ? 100 : 1); ++j) { |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 1411 | // ret = ShCompile(compiler, shaderStrings, NumShaderStrings, lengths, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages); |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1412 | ret = ShCompile(compiler, &shaderString, 1, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages); |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 1413 | // const char* multi[12] = { "# ve", "rsion", " 300 e", "s", "\n#err", |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 1414 | // "or should be l", "ine 1", "string 5\n", "float glo", "bal", |
John Kessenich | ea869fb | 2013-10-28 18:12:06 +0000 | [diff] [blame] | 1415 | // ";\n#error should be line 2\n void main() {", "global = 2.3;}" }; |
John Kessenich | 927608b | 2017-01-06 12:34:14 -0700 | [diff] [blame] | 1416 | // const char* multi[7] = { "/", "/", "\\", "\n", "\n", "#", "version 300 es" }; |
| 1417 | // ret = ShCompile(compiler, multi, 7, nullptr, EShOptNone, &Resources, Options, (Options & EOptionDefaultDesktop) ? 110 : 100, false, messages); |
John Kessenich | 41cf6b5 | 2013-06-25 18:10:05 +0000 | [diff] [blame] | 1418 | } |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1419 | |
John Kessenich | 94a81fb | 2013-08-31 02:41:30 +0000 | [diff] [blame] | 1420 | if (Options & EOptionMemoryLeakMode) |
John Kessenich | 2b07c7e | 2013-07-31 18:44:13 +0000 | [diff] [blame] | 1421 | glslang::OS_DumpMemoryCounters(); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1422 | } |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1423 | |
John Kessenich | 41cf6b5 | 2013-06-25 18:10:05 +0000 | [diff] [blame] | 1424 | delete [] lengths; |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1425 | FreeFileData(shaderString); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1426 | |
John Kessenich | c999ba2 | 2013-11-07 23:33:24 +0000 | [diff] [blame] | 1427 | if (ret == 0) |
| 1428 | CompileFailed = true; |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1429 | } |
| 1430 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1431 | // |
| 1432 | // print usage to stdout |
| 1433 | // |
| 1434 | void usage() |
| 1435 | { |
John Kessenich | 319de23 | 2013-12-04 04:43:40 +0000 | [diff] [blame] | 1436 | printf("Usage: glslangValidator [option]... [file]...\n" |
| 1437 | "\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1438 | "'file' can end in .<stage> for auto-stage classification, where <stage> is:\n" |
| 1439 | " .conf to provide a config file that replaces the default configuration\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1440 | " (see -c option below for generating a template)\n" |
| 1441 | " .vert for a vertex shader\n" |
| 1442 | " .tesc for a tessellation control shader\n" |
| 1443 | " .tese for a tessellation evaluation shader\n" |
| 1444 | " .geom for a geometry shader\n" |
| 1445 | " .frag for a fragment shader\n" |
| 1446 | " .comp for a compute shader\n" |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 1447 | #ifdef NV_EXTENSIONS |
| 1448 | " .mesh for a mesh shader\n" |
| 1449 | " .task for a task shader\n" |
Chao Chen | b50c02e | 2018-09-19 11:42:24 -0700 | [diff] [blame] | 1450 | " .rgen for a ray generation shader\n" |
| 1451 | " .rint for a ray intersection shader\n" |
| 1452 | " .rahit for a ray any hit shader\n" |
| 1453 | " .rchit for a ray closest hit shader\n" |
| 1454 | " .rmiss for a ray miss shader\n" |
Sahil Parmar | 035cbbe | 2018-10-04 16:39:18 -0700 | [diff] [blame] | 1455 | " .rcall for a ray callable shader\n" |
Chao Chen | 3c36699 | 2018-09-19 11:41:59 -0700 | [diff] [blame] | 1456 | #endif |
John Kessenich | 2ead40f | 2018-04-17 17:44:11 -0600 | [diff] [blame] | 1457 | " .glsl for .vert.glsl, .tesc.glsl, ..., .comp.glsl compound suffixes\n" |
| 1458 | " .hlsl for .vert.hlsl, .tesc.hlsl, ..., .comp.hlsl compound suffixes\n" |
John Kessenich | 319de23 | 2013-12-04 04:43:40 +0000 | [diff] [blame] | 1459 | "\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1460 | "Options:\n" |
| 1461 | " -C cascading errors; risk crash from accumulation of error recoveries\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1462 | " -D input is HLSL (this is the default when any suffix is .hlsl)\n" |
John Kessenich | a931366 | 2017-06-15 10:40:49 -0600 | [diff] [blame] | 1463 | " -D<macro=def>\n" |
| 1464 | " -D<macro> define a pre-processor macro\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1465 | " -E print pre-processed GLSL; cannot be used with -l;\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1466 | " errors will appear on stderr\n" |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 1467 | " -G[ver] create SPIR-V binary, under OpenGL semantics; turns on -l;\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1468 | " default file name is <stage>.spv (-o overrides this);\n" |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 1469 | " 'ver', when present, is the version of the input semantics,\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1470 | " which will appear in #define GL_SPIRV ver;\n" |
| 1471 | " '--client opengl100' is the same as -G100;\n" |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 1472 | " a '--target-env' for OpenGL will also imply '-G'\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1473 | " -H print human readable form of SPIR-V; turns on -V\n" |
John Kessenich | 971a0a8 | 2017-06-07 15:06:58 -0600 | [diff] [blame] | 1474 | " -I<dir> add dir to the include search path; includer's directory\n" |
| 1475 | " is searched first, followed by left-to-right order of -I\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1476 | " -Od disables optimization; may cause illegal SPIR-V for HLSL\n" |
| 1477 | " -Os optimizes SPIR-V to minimize size\n" |
John Kessenich | e751bca | 2017-03-16 11:20:38 -0600 | [diff] [blame] | 1478 | " -S <stage> uses specified stage rather than parsing the file extension\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1479 | " choices for <stage> are vert, tesc, tese, geom, frag, or comp\n" |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 1480 | " -U<macro> undefine a pre-processor macro\n" |
| 1481 | " -V[ver] create SPIR-V binary, under Vulkan semantics; turns on -l;\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1482 | " default file name is <stage>.spv (-o overrides this)\n" |
John Kessenich | 6353d55 | 2017-06-23 11:11:09 -0600 | [diff] [blame] | 1483 | " 'ver', when present, is the version of the input semantics,\n" |
| 1484 | " which will appear in #define VULKAN ver\n" |
| 1485 | " '--client vulkan100' is the same as -V100\n" |
| 1486 | " a '--target-env' for Vulkan will also imply '-V'\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1487 | " -c configuration dump;\n" |
| 1488 | " creates the default configuration file (redirect to a .conf file)\n" |
| 1489 | " -d default to desktop (#version 110) when there is no shader #version\n" |
| 1490 | " (default is ES version 100)\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1491 | " -e <name> | --entry-point <name>\n" |
| 1492 | " specify <name> as the entry-point function name\n" |
John Kessenich | 5d610ee | 2018-03-07 18:05:55 -0700 | [diff] [blame] | 1493 | " -f{hlsl_functionality1}\n" |
| 1494 | " 'hlsl_functionality1' enables use of the\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1495 | " SPV_GOOGLE_hlsl_functionality1 extension\n" |
John Kessenich | 121853f | 2017-05-31 17:11:16 -0600 | [diff] [blame] | 1496 | " -g generate debug information\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1497 | " -h print this usage message\n" |
| 1498 | " -i intermediate tree (glslang AST) is printed out\n" |
| 1499 | " -l link all input files together to form a single module\n" |
| 1500 | " -m memory leak mode\n" |
John Kessenich | a25530c | 2017-09-11 20:13:49 -0600 | [diff] [blame] | 1501 | " -o <file> save binary to <file>, requires a binary option (e.g., -V)\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1502 | " -q dump reflection query database\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1503 | " -r | --relaxed-errors" |
| 1504 | " relaxed GLSL semantic error-checking mode\n" |
John Kessenich | b63f4a3 | 2017-11-16 22:32:20 -0700 | [diff] [blame] | 1505 | " -s silence syntax and semantic error reporting\n" |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1506 | " -t multi-threaded mode\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1507 | " -v | --version\n" |
| 1508 | " print version strings\n" |
| 1509 | " -w | --suppress-warnings\n" |
| 1510 | " suppress GLSL warnings, except as required by \"#extension : warn\"\n" |
John Kessenich | 6263fb1 | 2017-06-14 15:52:44 -0600 | [diff] [blame] | 1511 | " -x save binary output as text-based 32-bit hexadecimal numbers\n" |
Neil Roberts | 16f5347 | 2018-03-20 17:30:53 +0100 | [diff] [blame] | 1512 | " -u<name>:<loc> specify a uniform location override for --aml\n" |
Neil Roberts | b0f3d79 | 2018-03-20 17:41:05 +0100 | [diff] [blame] | 1513 | " --uniform-base <base> set a base to use for generated uniform locations\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1514 | " --auto-map-bindings | --amb automatically bind uniform variables\n" |
| 1515 | " without explicit bindings\n" |
| 1516 | " --auto-map-locations | --aml automatically locate input/output lacking\n" |
| 1517 | " 'location' (fragile, not cross stage)\n" |
| 1518 | " --client {vulkan<ver>|opengl<ver>} see -V and -G\n" |
| 1519 | " -dumpfullversion | -dumpversion print bare major.minor.patchlevel\n" |
| 1520 | " --flatten-uniform-arrays | --fua flatten uniform texture/sampler arrays to\n" |
| 1521 | " scalars\n" |
| 1522 | " --hlsl-offsets allow block offsets to follow HLSL rules\n" |
| 1523 | " works independently of source language\n" |
| 1524 | " --hlsl-iomap perform IO mapping in HLSL register space\n" |
| 1525 | " --hlsl-enable-16bit-types allow 16-bit types in SPIR-V for HLSL\n" |
John Kessenich | bd1c183 | 2018-12-07 18:38:26 -0700 | [diff] [blame] | 1526 | " --hlsl-dx9-compatible interprets sampler declarations as a texture/sampler combo like DirectX9 would." |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1527 | " --invert-y | --iy invert position.Y output in vertex shader\n" |
| 1528 | " --keep-uncalled | --ku don't eliminate uncalled functions\n" |
| 1529 | " --no-storage-format | --nsf use Unknown image format\n" |
baldurk | 15c37f7 | 2019-01-29 12:12:59 +0000 | [diff] [blame] | 1530 | " --reflect-strict-array-suffix use strict array suffix rules when reflecting\n" |
baldurk | edf8212 | 2019-01-29 15:49:00 +0000 | [diff] [blame] | 1531 | " --reflect-basic-array-suffix arrays of basic types will have trailing [0]\n" |
baldurk | 0af5e3e | 2019-01-29 16:04:44 +0000 | [diff] [blame] | 1532 | " --reflect-intermediate-io reflection includes inputs/outputs of linked shaders\n" |
| 1533 | " rather than just vertex/fragment\n" |
baldurk | 657acc0 | 2019-01-30 14:18:43 +0000 | [diff] [blame^] | 1534 | " --reflect-separate-buffers reflect buffer variables and blocks separately to uniforms\n" |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 1535 | " --resource-set-binding [stage] name set binding\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1536 | " set descriptor set and binding for\n" |
| 1537 | " individual resources\n" |
LoopDawg | 5201719 | 2017-07-14 15:15:47 -0600 | [diff] [blame] | 1538 | " --resource-set-binding [stage] set\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1539 | " set descriptor set for all resources\n" |
| 1540 | " --rsb synonym for --resource-set-binding\n" |
| 1541 | " --shift-image-binding [stage] num\n" |
| 1542 | " base binding number for images (uav)\n" |
| 1543 | " --shift-image-binding [stage] [num set]...\n" |
| 1544 | " per-descriptor-set shift values\n" |
| 1545 | " --sib synonym for --shift-image-binding\n" |
| 1546 | " --shift-sampler-binding [stage] num\n" |
| 1547 | " base binding number for samplers\n" |
| 1548 | " --shift-sampler-binding [stage] [num set]...\n" |
| 1549 | " per-descriptor-set shift values\n" |
| 1550 | " --ssb synonym for --shift-sampler-binding\n" |
| 1551 | " --shift-ssbo-binding [stage] num base binding number for SSBOs\n" |
| 1552 | " --shift-ssbo-binding [stage] [num set]...\n" |
| 1553 | " per-descriptor-set shift values\n" |
| 1554 | " --sbb synonym for --shift-ssbo-binding\n" |
| 1555 | " --shift-texture-binding [stage] num\n" |
| 1556 | " base binding number for textures\n" |
| 1557 | " --shift-texture-binding [stage] [num set]...\n" |
| 1558 | " per-descriptor-set shift values\n" |
| 1559 | " --stb synonym for --shift-texture-binding\n" |
| 1560 | " --shift-uav-binding [stage] num base binding number for UAVs\n" |
| 1561 | " --shift-uav-binding [stage] [num set]...\n" |
| 1562 | " per-descriptor-set shift values\n" |
| 1563 | " --suavb synonym for --shift-uav-binding\n" |
| 1564 | " --shift-UBO-binding [stage] num base binding number for UBOs\n" |
| 1565 | " --shift-UBO-binding [stage] [num set]...\n" |
| 1566 | " per-descriptor-set shift values\n" |
| 1567 | " --sub synonym for --shift-UBO-binding\n" |
| 1568 | " --shift-cbuffer-binding | --scb synonyms for --shift-UBO-binding\n" |
| 1569 | " --spirv-dis output standard-form disassembly; works only\n" |
| 1570 | " when a SPIR-V generation option is also used\n" |
John Kessenich | c340425 | 2018-08-23 15:29:08 -0600 | [diff] [blame] | 1571 | " --spirv-val execute the SPIRV-Tools validator\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1572 | " --source-entrypoint <name> the given shader source function is\n" |
| 1573 | " renamed to be the <name> given in -e\n" |
| 1574 | " --sep synonym for --source-entrypoint\n" |
| 1575 | " --stdin read from stdin instead of from a file;\n" |
| 1576 | " requires providing the shader stage using -S\n" |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 1577 | " --target-env {vulkan1.0 | vulkan1.1 | opengl | \n" |
| 1578 | " spirv1.0 | spirv1.1 | spirv1.2 | spirv1.3}\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1579 | " set execution environment that emitted code\n" |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 1580 | " will execute in (versus source language\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1581 | " semantics selected by --client) defaults:\n" |
| 1582 | " * 'vulkan1.0' under '--client vulkan<ver>'\n" |
| 1583 | " * 'opengl' under '--client opengl<ver>'\n" |
John Kessenich | 8717a5d | 2018-10-26 10:12:32 -0600 | [diff] [blame] | 1584 | " * 'spirv1.0' under --target-env vulkan1.0\n" |
| 1585 | " * 'spirv1.3' under --target-env vulkan1.1\n" |
| 1586 | " multiple --targen-env can be specified.\n" |
John Kessenich | ade8bbb | 2018-08-12 21:24:55 -0600 | [diff] [blame] | 1587 | " --variable-name <name>\n" |
| 1588 | " --vn <name> creates a C header file that contains a\n" |
| 1589 | " uint32_t array named <name>\n" |
| 1590 | " initialized with the shader binary code\n" |
John Kessenich | b0a7eb5 | 2013-11-07 17:44:20 +0000 | [diff] [blame] | 1591 | ); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1592 | |
| 1593 | exit(EFailUsage); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1594 | } |
| 1595 | |
John Kessenich | 3ce4e59 | 2014-10-06 19:57:34 +0000 | [diff] [blame] | 1596 | #if !defined _MSC_VER && !defined MINGW_HAS_SECURE_API |
John Kessenich | cfd643e | 2013-03-08 23:14:42 +0000 | [diff] [blame] | 1597 | |
| 1598 | #include <errno.h> |
| 1599 | |
| 1600 | int fopen_s( |
| 1601 | FILE** pFile, |
John Kessenich | 51cdd90 | 2014-02-18 23:37:57 +0000 | [diff] [blame] | 1602 | const char* filename, |
| 1603 | const char* mode |
John Kessenich | cfd643e | 2013-03-08 23:14:42 +0000 | [diff] [blame] | 1604 | ) |
| 1605 | { |
| 1606 | if (!pFile || !filename || !mode) { |
| 1607 | return EINVAL; |
| 1608 | } |
| 1609 | |
| 1610 | FILE* f = fopen(filename, mode); |
| 1611 | if (! f) { |
| 1612 | if (errno != 0) { |
| 1613 | return errno; |
| 1614 | } else { |
| 1615 | return ENOENT; |
| 1616 | } |
| 1617 | } |
| 1618 | *pFile = f; |
| 1619 | |
| 1620 | return 0; |
| 1621 | } |
| 1622 | |
| 1623 | #endif |
| 1624 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1625 | // |
| 1626 | // Malloc a string of sufficient size and read a string into it. |
| 1627 | // |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1628 | char* ReadFileData(const char* fileName) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1629 | { |
John Kessenich | b329715 | 2015-07-11 18:01:03 -0600 | [diff] [blame] | 1630 | FILE *in = nullptr; |
John Kessenich | 3ce4e59 | 2014-10-06 19:57:34 +0000 | [diff] [blame] | 1631 | int errorCode = fopen_s(&in, fileName, "r"); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1632 | if (errorCode || in == nullptr) |
| 1633 | Error("unable to open input file"); |
John Kessenich | ecba76f | 2017-01-06 00:34:48 -0700 | [diff] [blame] | 1634 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1635 | int count = 0; |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1636 | while (fgetc(in) != EOF) |
| 1637 | count++; |
| 1638 | |
John Kessenich | d6c72a4 | 2014-08-18 19:42:35 +0000 | [diff] [blame] | 1639 | fseek(in, 0, SEEK_SET); |
John Kessenich | ca3457f | 2015-05-18 01:59:45 +0000 | [diff] [blame] | 1640 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1641 | char* return_data = (char*)malloc(count + 1); // freed in FreeFileData() |
| 1642 | if ((int)fread(return_data, 1, count, in) != count) { |
| 1643 | free(return_data); |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1644 | Error("can't read input file"); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1645 | } |
John Kessenich | 68d78fd | 2015-07-12 19:28:10 -0600 | [diff] [blame] | 1646 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1647 | return_data[count] = '\0'; |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1648 | fclose(in); |
John Kessenich | b329715 | 2015-07-11 18:01:03 -0600 | [diff] [blame] | 1649 | |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1650 | return return_data; |
| 1651 | } |
| 1652 | |
John Kessenich | 04acb1b | 2017-06-14 17:36:50 -0600 | [diff] [blame] | 1653 | void FreeFileData(char* data) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1654 | { |
John Kessenich | b329715 | 2015-07-11 18:01:03 -0600 | [diff] [blame] | 1655 | free(data); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1656 | } |
| 1657 | |
John Kessenich | 54d8cda | 2013-02-11 22:36:01 +0000 | [diff] [blame] | 1658 | void InfoLogMsg(const char* msg, const char* name, const int num) |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1659 | { |
John Kessenich | fae38ee | 2015-06-10 23:23:12 +0000 | [diff] [blame] | 1660 | if (num >= 0 ) |
| 1661 | printf("#### %s %s %d INFO LOG ####\n", msg, name, num); |
| 1662 | else |
| 1663 | printf("#### %s %s INFO LOG ####\n", msg, name); |
John Kessenich | a0af473 | 2012-12-12 21:15:54 +0000 | [diff] [blame] | 1664 | } |