123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155 |
- // Copyright (c) 2015-2016 The Khronos Group Inc.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- #include <cstdio>
- #include <cstring>
- #include <vector>
- #include "source/spirv_target_env.h"
- #include "spirv-tools/libspirv.h"
- #include "tools/io.h"
- void print_usage(char* argv0) {
- printf(
- R"(%s - Create a SPIR-V binary module from SPIR-V assembly text
- Usage: %s [options] [<filename>]
- The SPIR-V assembly text is read from <filename>. If no file is specified,
- or if the filename is "-", then the assembly text is read from standard input.
- The SPIR-V binary module is written to file "out.spv", unless the -o option
- is used.
- Options:
- -h, --help Print this help.
- -o <filename> Set the output filename. Use '-' to mean stdout.
- --version Display assembler version information.
- --preserve-numeric-ids
- Numeric IDs in the binary will have the same values as in the
- source. Non-numeric IDs are allocated by filling in the gaps,
- starting with 1 and going up.
- --target-env {vulkan1.0|vulkan1.1|spv1.0|spv1.1|spv1.2|spv1.3}
- Use Vulkan 1.0, Vulkan 1.1, SPIR-V 1.0, SPIR-V 1.1,
- SPIR-V 1.2, or SPIR-V 1.3
- )",
- argv0, argv0);
- }
- static const auto kDefaultEnvironment = SPV_ENV_UNIVERSAL_1_3;
- int main(int argc, char** argv) {
- const char* inFile = nullptr;
- const char* outFile = nullptr;
- uint32_t options = 0;
- spv_target_env target_env = kDefaultEnvironment;
- for (int argi = 1; argi < argc; ++argi) {
- if ('-' == argv[argi][0]) {
- switch (argv[argi][1]) {
- case 'h': {
- print_usage(argv[0]);
- return 0;
- }
- case 'o': {
- if (!outFile && argi + 1 < argc) {
- outFile = argv[++argi];
- } else {
- print_usage(argv[0]);
- return 1;
- }
- } break;
- case 0: {
- // Setting a filename of "-" to indicate stdin.
- if (!inFile) {
- inFile = argv[argi];
- } else {
- fprintf(stderr, "error: More than one input file specified\n");
- return 1;
- }
- } break;
- case '-': {
- // Long options
- if (0 == strcmp(argv[argi], "--version")) {
- printf("%s\n", spvSoftwareVersionDetailsString());
- printf("Target: %s\n",
- spvTargetEnvDescription(kDefaultEnvironment));
- return 0;
- } else if (0 == strcmp(argv[argi], "--help")) {
- print_usage(argv[0]);
- return 0;
- } else if (0 == strcmp(argv[argi], "--preserve-numeric-ids")) {
- options |= SPV_TEXT_TO_BINARY_OPTION_PRESERVE_NUMERIC_IDS;
- } else if (0 == strcmp(argv[argi], "--target-env")) {
- if (argi + 1 < argc) {
- const auto env_str = argv[++argi];
- if (!spvParseTargetEnv(env_str, &target_env)) {
- fprintf(stderr, "error: Unrecognized target env: %s\n",
- env_str);
- return 1;
- }
- } else {
- fprintf(stderr, "error: Missing argument to --target-env\n");
- return 1;
- }
- } else {
- fprintf(stderr, "error: Unrecognized option: %s\n\n", argv[argi]);
- print_usage(argv[0]);
- return 1;
- }
- } break;
- default:
- fprintf(stderr, "error: Unrecognized option: %s\n\n", argv[argi]);
- print_usage(argv[0]);
- return 1;
- }
- } else {
- if (!inFile) {
- inFile = argv[argi];
- } else {
- fprintf(stderr, "error: More than one input file specified\n");
- return 1;
- }
- }
- }
- if (!outFile) {
- outFile = "out.spv";
- }
- std::vector<char> contents;
- if (!ReadFile<char>(inFile, "r", &contents)) return 1;
- spv_binary binary;
- spv_diagnostic diagnostic = nullptr;
- spv_context context = spvContextCreate(target_env);
- spv_result_t error = spvTextToBinaryWithOptions(
- context, contents.data(), contents.size(), options, &binary, &diagnostic);
- spvContextDestroy(context);
- if (error) {
- spvDiagnosticPrint(diagnostic);
- spvDiagnosticDestroy(diagnostic);
- return error;
- }
- if (!WriteFile<uint32_t>(outFile, "wb", binary->code, binary->wordCount)) {
- spvBinaryDestroy(binary);
- return 1;
- }
- spvBinaryDestroy(binary);
- return 0;
- }
|