1 /*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "stdio.h"
9 #include <fstream>
10 #include "SkSLCompiler.h"
11 #include "GrContextOptions.h"
12
13 /**
14 * Very simple standalone executable to facilitate testing.
15 */
main(int argc,const char ** argv)16 int main(int argc, const char** argv) {
17 if (argc != 3) {
18 printf("usage: skslc <input> <output>\n");
19 exit(1);
20 }
21 SkSL::Program::Kind kind;
22 size_t len = strlen(argv[1]);
23 if (len > 5 && !strcmp(argv[1] + strlen(argv[1]) - 5, ".vert")) {
24 kind = SkSL::Program::kVertex_Kind;
25 } else if (len > 5 && !strcmp(argv[1] + strlen(argv[1]) - 5, ".frag")) {
26 kind = SkSL::Program::kFragment_Kind;
27 } else if (len > 5 && !strcmp(argv[1] + strlen(argv[1]) - 5, ".geom")) {
28 kind = SkSL::Program::kGeometry_Kind;
29 } else {
30 printf("input filename must end in '.vert', '.frag', or '.geom'\n");
31 exit(1);
32 }
33
34 std::ifstream in(argv[1]);
35 std::string stdText((std::istreambuf_iterator<char>(in)),
36 std::istreambuf_iterator<char>());
37 SkString text(stdText.c_str());
38 if (in.rdstate()) {
39 printf("error reading '%s'\n", argv[1]);
40 exit(2);
41 }
42 SkSL::Program::Settings settings;
43 sk_sp<GrShaderCaps> caps = SkSL::ShaderCapsFactory::Default();
44 settings.fCaps = caps.get();
45 SkString name(argv[2]);
46 if (name.endsWith(".spirv")) {
47 SkFILEWStream out(argv[2]);
48 SkSL::Compiler compiler;
49 if (!out.isValid()) {
50 printf("error writing '%s'\n", argv[2]);
51 exit(4);
52 }
53 std::unique_ptr<SkSL::Program> program = compiler.convertProgram(kind, text, settings);
54 if (!program || !compiler.toSPIRV(*program, out)) {
55 printf("%s", compiler.errorText().c_str());
56 exit(3);
57 }
58 } else if (name.endsWith(".glsl")) {
59 SkFILEWStream out(argv[2]);
60 SkSL::Compiler compiler;
61 if (!out.isValid()) {
62 printf("error writing '%s'\n", argv[2]);
63 exit(4);
64 }
65 std::unique_ptr<SkSL::Program> program = compiler.convertProgram(kind, text, settings);
66 if (!program || !compiler.toGLSL(*program, out)) {
67 printf("%s", compiler.errorText().c_str());
68 exit(3);
69 }
70 } else {
71 printf("expected output filename to end with '.spirv' or '.glsl'");
72 }
73 }
74