xref: /aosp_15_r20/external/skia/tests/SkSLWGSLTestbed.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1 /*
2  * Copyright 2023 Google LLC
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 "include/core/SkTypes.h"
9 #include "src/sksl/SkSLCompiler.h"
10 #include "src/sksl/SkSLProgramKind.h"
11 #include "src/sksl/SkSLProgramSettings.h"
12 #include "src/sksl/SkSLUtil.h"
13 #include "src/sksl/codegen/SkSLWGSLCodeGenerator.h"
14 #include "src/sksl/ir/SkSLProgram.h"
15 #include "tests/Test.h"
16 
17 #include <memory>
18 #include <string>
19 
test(skiatest::Reporter * r,const char * src,SkSL::ProgramKind kind=SkSL::ProgramKind::kFragment)20 static void test(skiatest::Reporter* r,
21                  const char* src,
22                  SkSL::ProgramKind kind = SkSL::ProgramKind::kFragment) {
23     SkSL::Compiler compiler;
24     SkSL::ProgramSettings settings;
25     std::unique_ptr<SkSL::Program> program = compiler.convertProgram(kind, std::string(src),
26                                                                      settings);
27     if (!program) {
28         SkDebugf("Unexpected error compiling %s\n%s", src, compiler.errorText().c_str());
29         REPORTER_ASSERT(r, program);
30     } else {
31         std::string output;
32         REPORTER_ASSERT(r, SkSL::ToWGSL(*program, SkSL::ShaderCapsFactory::Default(), &output));
33         REPORTER_ASSERT(r, output != "");
34         //SkDebugf("WGSL output:\n\n%s", output.c_str());
35     }
36 }
37 
DEF_TEST(SkSLWGSLTestbed,r)38 DEF_TEST(SkSLWGSLTestbed, r) {
39     // Add in your SkSL here.
40     test(r,
41          R"__SkSL__(
42              void main() {
43                  sk_FragColor = half4(0);
44              }
45          )__SkSL__");
46 }
47