tests/RasterPipelineCodeGeneratorTest.cpp - skia - Git at Google

 /*
  * Copyright 2022 Google LLC
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */

 #include "include/core/SkColor.h"
 #include "include/core/SkTypes.h"
 #include "include/private/SkSLProgramKind.h"
 #include "src/base/SkArenaAlloc.h"
 #include "src/core/SkRasterPipeline.h"
 #include "src/sksl/SkSLCompiler.h"
 #include "src/sksl/SkSLProgramSettings.h"
 #include "src/sksl/SkSLUtil.h"
 #include "src/sksl/codegen/SkSLRasterPipelineBuilder.h"
 #include "src/sksl/codegen/SkSLRasterPipelineCodeGenerator.h"
 #include "src/sksl/ir/SkSLFunctionDeclaration.h"
 #include "src/sksl/ir/SkSLProgram.h"
 #include "src/sksl/tracing/SkRPDebugTrace.h"
 #include "tests/Test.h"

 #include <memory>
 #include <optional>
 #include <string>

 //#define DUMP_PROGRAMS 1
 #if defined(DUMP_PROGRAMS)
 #include "src/core/SkStreamPriv.h"
 #endif

 static void test(skiatest::Reporter* r,
                  const char* src,
                  SkSpan<const float> uniforms,
                  SkColor4f startingColor,
                  std::optional<SkColor4f> expectedResult) {
     SkSL::Compiler compiler(SkSL::ShaderCapsFactory::Default());
     SkSL::ProgramSettings settings;
     settings.fMaxVersionAllowed = SkSL::Version::k300;
     std::unique_ptr<SkSL::Program> program = compiler.convertProgram(
             SkSL::ProgramKind::kRuntimeColorFilter, std::string(src), settings);
     if (!program) {
         ERRORF(r, "Unexpected error compiling %s\n%s", src, compiler.errorText().c_str());
         return;
     }
     const SkSL::FunctionDeclaration* main = program->getFunction("main");
     if (!main) {
         ERRORF(r, "Program must have a 'main' function");
         return;
     }
     SkArenaAlloc alloc(/*firstHeapAllocation=*/1000);
     SkRasterPipeline pipeline(&alloc);
     pipeline.append_constant_color(&alloc, startingColor);
     SkSL::SkRPDebugTrace debugTrace;
     std::unique_ptr<SkSL::RP::Program> rasterProg =
             SkSL::MakeRasterPipelineProgram(*program, *main->definition(), &debugTrace);
     if (!rasterProg && !expectedResult.has_value()) {
         // We didn't get a program, as expected. Test passes.
         return;
     }
     if (!rasterProg && expectedResult.has_value()) {
         ERRORF(r, "MakeRasterPipelineProgram failed");
         return;
     }
     if (rasterProg && !expectedResult.has_value()) {
         ERRORF(r, "MakeRasterPipelineProgram should have failed, but didn't");
         return;
     }

 #if defined(DUMP_PROGRAMS)
     // Dump the program instructions via SkDebugf.
     SkDebugf("-----\n\n");
     SkDebugfStream stream;
     rasterProg->dump(&stream);
     SkDebugf("\n-----\n\n");
 #endif

     // Append the SkSL program to the raster pipeline.
     rasterProg->appendStages(&pipeline, &alloc, uniforms);

     // Move the float values from RGBA into an 8888 memory buffer.
     uint32_t out[SkRasterPipeline_kMaxStride_highp] = {};
     SkRasterPipeline_MemoryCtx outCtx{/*pixels=*/out, /*stride=*/SkRasterPipeline_kMaxStride_highp};
     pipeline.append(SkRasterPipelineOp::store_8888, &outCtx);
     pipeline.run(0, 0, 1, 1);

     // Make sure the first pixel (exclusively) of `out` matches RGBA.
     uint32_t expected = expectedResult->toBytes_RGBA();
     REPORTER_ASSERT(r, out[0] == expected,
                     "Got:%02X%02X%02X%02X Expected:%02X%02X%02X%02X",
                     (out[0] >> 24) & 0xFF,
                     (out[0] >> 16) & 0xFF,
                     (out[0] >> 8) & 0xFF,
                     out[0] & 0xFF,
                     (expected >> 24) & 0xFF,
                     (expected >> 16) & 0xFF,
                     (expected >> 8) & 0xFF,
                     expected & 0xFF);

     // Make sure the rest of the pixels are untouched.
     for (size_t i = 1; i < std::size(out); ++i) {
         REPORTER_ASSERT(r, out[i] == 0);
     }
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorPassthroughTest, r) {
     test(r,
          R"__SkSL__(
              half4 main(half4 startingColor) {
                  return startingColor;
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{1.0f, 1.0f, 0.0f, 1.0f},
          /*expectedResult=*/SkColor4f{1.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorVariableGreenTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              half4 main(half4) {
                  half2 zeroOne = half2(0, 1);
                  half one = zeroOne.y, zero = zeroOne.x;
                  return half4(zero, zeroOne.yx, one);
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorAdditionTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              half4 main(half4 y) {
                  half4 x = half4(-1, 0, 1, 0);
                  half4 z = x.wzyx;
                  z += y.000w;
                  return y + z;
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 1.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorIfElseTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              const half4 colorWhite = half4(1);

              half4 ifElseTest(half4 colorBlue, half4 colorGreen, half4 colorRed) {
                  half4 result = half4(0);
                  if (colorWhite != colorBlue) {    // TRUE
                      if (colorGreen == colorRed) { // FALSE
                          result = colorRed;
                      } else {
                          result = colorGreen;
                      }
                  } else {
                      if (colorRed != colorGreen) { // TRUE, but in a false branch
                          result = colorBlue;
                      } else {                      // FALSE, and in a false branch
                          result = colorWhite;
                      }
                  }
                  if (colorRed == colorBlue) { // FALSE
                      return colorWhite;
                  }
                  if (colorRed != colorGreen) { // TRUE
                      return result;
                  }
                  if (colorRed == colorWhite) { // FALSE
                      return colorBlue;
                  }
                  return colorRed;
              }

              half4 main(half4) {
                  return ifElseTest(colorWhite.00b1, colorWhite.0g01, colorWhite.r001);
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorTernaryTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              half4 main(half4 colorWhite) {
                  half4 colorBlue  = colorWhite.00ba,
                        colorGreen = colorWhite.0g0a,
                        colorRed   = colorWhite.r00a;
                  // This ternary matches the initial if-else block inside IfElseTest.
                  half4 result;
                  result = (colorWhite != colorBlue)                              // TRUE
                             ? (colorGreen == colorRed ? colorRed : colorGreen)   // FALSE
                             : (colorRed != colorGreen ? colorBlue : colorWhite); // in false branch

                  // This ternary matches the second portion of IfElseTest.
                  return colorRed == colorBlue  ? colorWhite :
                         colorRed != colorGreen ? result :     // TRUE
                         colorRed == colorWhite ? colorBlue :
                                                  colorRed;
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{1.0, 1.0, 1.0, 1.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorTernarySideEffectTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              const half4 colorGreen = half4(0,1,0,1),
                          colorRed   = half4(1,0,0,1);
              half4 main(half4) {
                  half x = 1, y = 1;
                  (x == y) ? (x += 1) : (y += 1);  // TRUE,   x=2 y=1
                  (x == y) ? (x += 3) : (y += 3);  // FALSE,  x=2 y=4
                  (x <  y) ? (x += 5) : (y += 5);  // TRUE,   x=7 y=4
                  (y >= x) ? (x += 9) : (y += 9);  // FALSE,  x=7 y=13
                  (x != y) ? (x += 1) : (y     );  // TRUE,   x=8 y=13
                  (x == y) ? (x += 2) : (y     );  // FALSE,  x=8 y=13
                  (x != y) ? (x     ) : (y += 3);  // TRUE,   x=8 y=13
                  (x == y) ? (x     ) : (y += 4);  // FALSE,  x=8 y=17

                  bool b = true;
                  bool c = (b = false) ? false : b;

                  return c ? colorRed : (x == 8 && y == 17) ? colorGreen : colorRed;
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorNestedTernaryTest, r) {
     // Add in your SkSL here.
     test(r,
          R"__SkSL__(
              half4 main(half4) {
                  half three = 3, one = 1, two = 2;
                  half result = (three > (one > two ? 2.0 : 5.0)) ? 1.0 : 0.499;
                  return half4(result);
              }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.499f, 0.499f, 0.499f, 0.499f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorDoWhileTest, r) {
     // This is based on shared/DoWhileControlFlow.sksl (but avoids swizzles).
     test(r,
          R"__SkSL__(
             half r = 1.0, g = 1.0, b = 1.0;
             half4 main(half4) {
                 half a = 1.0;
                 // Verify that break is allowed in a do-while loop.
                 do {
                     r -= 0.25;
                     if (r <= 0) break;
                 } while (a == 1.0);

                 // Verify that continue is allowed in a do-while loop.
                 do {
                     b -= 0.25;
                     if (a == 1) continue; // should always happen
                     g = 0;
                 } while (b > 0.0);

                 return half4(r, g, b, a);
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorArithmeticTest, r) {
     test(r,
          R"__SkSL__(
             half4 main(half4) {
                 const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);

                 half a = 3.0, b = 4.0, c = a + b - 2.0;
                 if (a*a + b*b == c*c*c/5.0) {
                     int A = 3, B = 4, C = A + B - 2;
                     if (A*A + B*B == C*C*C/5) {
                         return colorGreen;
                     }
                 }

                 return colorRed;
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorShortCircuitLogicalOr, r) {
     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 1) || ((y += 1) == 2)) { // LHS true, RHS not executed but would be true
                     return (x == 1 && y == 1) ? colorGreen : colorRed;
                 } else {
                     return colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 1) || ((y += 1) == 3)) { // LHS true, RHS not executed but would be false
                     return (x == 1 && y == 1) ? colorGreen : colorRed;
                 } else {
                     return colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 2) || ((y += 1) == 2)) { // LHS false, RHS is executed and is true
                     return (x == 1 && y == 2) ? colorGreen : colorRed;
                 } else {
                     return colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 2) || ((y += 1) == 3)) { // LHS false, RHS is executed and is false
                     return colorRed;
                 } else {
                     return (x == 1 && y == 2) ? colorGreen : colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorShortCircuitLogicalAnd, r) {
     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 1) && ((y += 1) == 2)) { // LHS true, RHS is executed and is true
                     return (x == 1 && y == 2) ? colorGreen : colorRed;
                 } else {
                     return colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 1) && ((y += 1) == 3)) { // LHS true, RHS is executed and is false
                     return colorRed;
                 } else {
                     return (x == 1 && y == 2) ? colorGreen : colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 2) && ((y += 1) == 2)) { // LHS false, RHS not executed but would be true
                     return colorRed;
                 } else {
                     return (x == 1 && y == 1) ? colorGreen : colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

     test(r,
          R"__SkSL__(
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
             half4 main(half4) {
                 int x = 1, y = 1;
                 if ((x == 2) && ((y += 1) == 3)) { // LHS false, RHS not executed but would be false
                     return colorRed;
                 } else {
                     return (x == 1 && y == 1) ? colorGreen : colorRed;
                 }
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorCoercedTypeTest, r) {
     static constexpr float kUniforms[] = {0.0, 1.0, 0.0, 1.0,
                                           1.0, 0.0, 0.0, 1.0};
     test(r,
          R"__SkSL__(
              uniform half4 colorGreen;
              uniform float4 colorRed;
              half4 main(half4 color) {
                  return ((colorGreen + colorRed) == float4(1.0, 1.0, 0.0, 2.0)) ? colorGreen
                                                                                 : colorGreen.gr01;
              }
          )__SkSL__",
          kUniforms,
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0, 1.0, 0.0, 1.0});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorIdentitySwizzle, r) {
     static constexpr float kUniforms[] = {0.0, 1.0, 0.0, 1.0,
                                           1.0, 0.0, 0.0, 1.0};
     test(r,
          R"__SkSL__(
             uniform half4 colorGreen, colorRed;
             half4 main(vec4 color) {
                 return (color.r   == 0.5             &&
                         color.rg  == half2(0.5, 1.0) &&
                         color.rgb == half3(0.5, 1.0, 0.0)) ? colorGreen : colorRed;
             }
          )__SkSL__",
          kUniforms,
          /*startingColor=*/SkColor4f{0.5, 1.0, 0.0, 0.25},
          /*expectedResult=*/SkColor4f{0.0, 1.0, 0.0, 1.0});

 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaTernaryTest, r) {
     test(r,
          R"__SkSL__(
             half4 main(vec4 color) {
                 half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

                 half scale = luma < 0.33333 ? 0.5
                            : luma < 0.66666 ? (0.166666 + 2.0 * (luma - 0.33333)) / luma
                            :   /* else */     (0.833333 + 0.5 * (luma - 0.66666)) / luma;
                 return half4(color.rgb * scale, color.a);
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.25, 0.00, 0.75, 1.0},
          /*expectedResult=*/SkColor4f{0.125, 0.0, 0.375, 1.0});

 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaIfNoEarlyReturnTest, r) {
     test(r,
          R"__SkSL__(
             half4 main(vec4 color) {
                 half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

                 half scale = 0;
                 if (luma < 0.33333) {
                     scale = 0.5;
                 } else if (luma < 0.66666) {
                     scale = (0.166666 + 2.0 * (luma - 0.33333)) / luma;
                 } else {
                     scale = (0.833333 + 0.5 * (luma - 0.66666)) / luma;
                 }
                 return half4(color.rgb * scale, color.a);
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.25, 0.00, 0.75, 1.0},
          /*expectedResult=*/SkColor4f{0.125, 0.0, 0.375, 1.0});

 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaWithEarlyReturnTest, r) {
     test(r,
          R"__SkSL__(
             half4 main(half4 color) {
                 half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

                 half scale = 0;
                 if (luma < 0.33333) {
                     return half4(color.rgb * 0.5, color.a);
                 } else if (luma < 0.66666) {
                     scale = 0.166666 + 2.0 * (luma - 0.33333);
                 } else {
                     scale = 0.833333 + 0.5 * (luma - 0.66666);
                 }
                 return half4(color.rgb * (scale/luma), color.a);
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.25, 0.00, 0.75, 1.0},
          /*expectedResult=*/SkColor4f{0.125, 0.0, 0.375, 1.0});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorBitwiseNotTest, r) {
     static constexpr int32_t kUniforms[] = { 0,  12,  3456,  4567890,
                                             ~0, ~12, ~3456, ~4567890};
     test(r,
          R"__SkSL__(
             uniform int4 value, expected;
             const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);

             half4 main(vec4) {
                 return (~value.x    == expected.x     &&
                         ~value.xy   == expected.xy    &&
                         ~value.xyz  == expected.xyz   &&
                         ~value.xyzw == expected.xyzw) ? colorGreen : colorRed;
             }
          )__SkSL__",
          SkSpan((const float*)kUniforms, std::size(kUniforms)),
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0, 1.0, 0.0, 1.0});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorComparisonIntrinsicTest, r) {
     test(r,
          R"__SkSL__(
             half4 main(vec4) {
                 const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
                 half4 a = half4(1, 2, 0, 1),
                       b = half4(2, 2, 1, 0);
                 int3  c = int3(1111, 3333, 5555),
                       d = int3(1111, 5555, 3333);
                 uint2 e = uint2(1111111111u, 222),
                       f = uint2(3333333333u, 222);
                 return (lessThan(a, b)         == bool4(true, false, true, false)  &&
                         lessThan(c, d)         == bool3(false, true, false)        &&
                         lessThan(e, f)         == bool2(true, false)               &&
                         greaterThan(a, b)      == bool4(false, false, false, true) &&
                         greaterThan(c, d)      == bool3(false, false, true)        &&
                         greaterThan(e, f)      == bool2(false, false)              &&
                         lessThanEqual(a, b)    == bool4(true, true, true, false)   &&
                         lessThanEqual(c, d)    == bool3(true, true, false)         &&
                         lessThanEqual(e, f)    == bool2(true, true)                &&
                         greaterThanEqual(a, b) == bool4(false, true, false, true)  &&
                         greaterThanEqual(c, d) == bool3(true, false, true)         &&
                         greaterThanEqual(e, f) == bool2(false, true)               &&
                         equal(a, b)            == bool4(false, true, false, false) &&
                         equal(c, d)            == bool3(true, false, false)        &&
                         equal(e, f)            == bool2(false, true)               &&
                         notEqual(a, b)         == bool4(true, false, true, true)   &&
                         notEqual(c, d)         == bool3(false, true, true)         &&
                         notEqual(e, f)         == bool2(true, false)               &&
                         max(a, b)              == half4(2, 2, 1, 1)                &&
                         max(c, d)              == int3(1111, 5555, 5555)           &&
                         max(e, f)              == uint2(3333333333u, 222)          &&
                         max(a, 1)              == half4(1, 2, 1, 1)                &&
                         max(c, 3333)           == int3(3333, 3333, 5555)           &&
                         max(e, 7777)           == uint2(1111111111u, 7777)         &&
                         min(a, b)              == half4(1, 2, 0, 0)                &&
                         min(c, d)              == int3(1111, 3333, 3333)           &&
                         min(e, f)              == uint2(1111111111u, 222)          &&
                         min(a, 1)              == half4(1, 1, 0, 1)                &&
                         min(c, 3333)           == int3(1111, 3333, 3333)           &&
                         min(e, 7777)           == uint2(7777, 222)) ? colorGreen : colorRed;
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.0, 0.0, 0.0, 0.0},
          /*expectedResult=*/SkColor4f{0.0, 1.0, 0.0, 1.0});
 }

 DEF_TEST(SkSLRasterPipelineCodeGeneratorUnpremulTest, r) {
     test(r,
          R"__SkSL__(
             // This is `unpremul` verbatim from sksl_shared, but marked noinline.
             noinline half4 MyUnpremul(half4 color) {
                 return half4(color.rgb / max(color.a, 0.0001), color.a);
             }

             half4 main(vec4 color) {
                 return MyUnpremul(color);
             }
          )__SkSL__",
          /*uniforms=*/{},
          /*startingColor=*/SkColor4f{0.5, 0.25, 0.125, 0.5},
          /*expectedResult=*/SkColor4f{1.0, 0.5, 0.25, 0.5});
 }
	/*
	* Copyright 2022 Google LLC
	*
	* Use of this source code is governed by a BSD-style license that can be
	* found in the LICENSE file.
	*/

	#include "include/core/SkColor.h"
	#include "include/core/SkTypes.h"
	#include "include/private/SkSLProgramKind.h"
	#include "src/base/SkArenaAlloc.h"
	#include "src/core/SkRasterPipeline.h"
	#include "src/sksl/SkSLCompiler.h"
	#include "src/sksl/SkSLProgramSettings.h"
	#include "src/sksl/SkSLUtil.h"
	#include "src/sksl/codegen/SkSLRasterPipelineBuilder.h"
	#include "src/sksl/codegen/SkSLRasterPipelineCodeGenerator.h"
	#include "src/sksl/ir/SkSLFunctionDeclaration.h"
	#include "src/sksl/ir/SkSLProgram.h"
	#include "src/sksl/tracing/SkRPDebugTrace.h"
	#include "tests/Test.h"

	#include <memory>
	#include <optional>
	#include <string>

	//#define DUMP_PROGRAMS 1
	#if defined(DUMP_PROGRAMS)
	#include "src/core/SkStreamPriv.h"
	#endif

	static void test(skiatest::Reporter* r,
	const char* src,
	SkSpan<const float> uniforms,
	SkColor4f startingColor,
	std::optional<SkColor4f> expectedResult) {
	SkSL::Compiler compiler(SkSL::ShaderCapsFactory::Default());
	SkSL::ProgramSettings settings;
	settings.fMaxVersionAllowed = SkSL::Version::k300;
	std::unique_ptr<SkSL::Program> program = compiler.convertProgram(
	SkSL::ProgramKind::kRuntimeColorFilter, std::string(src), settings);
	if (!program) {
	ERRORF(r, "Unexpected error compiling %s\n%s", src, compiler.errorText().c_str());
	return;
	}
	const SkSL::FunctionDeclaration* main = program->getFunction("main");
	if (!main) {
	ERRORF(r, "Program must have a 'main' function");
	return;
	}
	SkArenaAlloc alloc(/firstHeapAllocation=/1000);
	SkRasterPipeline pipeline(&alloc);
	pipeline.append_constant_color(&alloc, startingColor);
	SkSL::SkRPDebugTrace debugTrace;
	std::unique_ptr<SkSL::RP::Program> rasterProg =
	SkSL::MakeRasterPipelineProgram(program, main->definition(), &debugTrace);
	if (!rasterProg && !expectedResult.has_value()) {
	// We didn't get a program, as expected. Test passes.
	return;
	}
	if (!rasterProg && expectedResult.has_value()) {
	ERRORF(r, "MakeRasterPipelineProgram failed");
	return;
	}
	if (rasterProg && !expectedResult.has_value()) {
	ERRORF(r, "MakeRasterPipelineProgram should have failed, but didn't");
	return;
	}

	#if defined(DUMP_PROGRAMS)
	// Dump the program instructions via SkDebugf.
	SkDebugf("-----\n\n");
	SkDebugfStream stream;
	rasterProg->dump(&stream);
	SkDebugf("\n-----\n\n");
	#endif

	// Append the SkSL program to the raster pipeline.
	rasterProg->appendStages(&pipeline, &alloc, uniforms);

	// Move the float values from RGBA into an 8888 memory buffer.
	uint32_t out[SkRasterPipeline_kMaxStride_highp] = {};
	SkRasterPipeline_MemoryCtx outCtx{/pixels=/out, /stride=/SkRasterPipeline_kMaxStride_highp};
	pipeline.append(SkRasterPipelineOp::store_8888, &outCtx);
	pipeline.run(0, 0, 1, 1);

	// Make sure the first pixel (exclusively) of `out` matches RGBA.
	uint32_t expected = expectedResult->toBytes_RGBA();
	REPORTER_ASSERT(r, out[0] == expected,
	"Got:%02X%02X%02X%02X Expected:%02X%02X%02X%02X",
	(out[0] >> 24) & 0xFF,
	(out[0] >> 16) & 0xFF,
	(out[0] >> 8) & 0xFF,
	out[0] & 0xFF,
	(expected >> 24) & 0xFF,
	(expected >> 16) & 0xFF,
	(expected >> 8) & 0xFF,
	expected & 0xFF);

	// Make sure the rest of the pixels are untouched.
	for (size_t i = 1; i < std::size(out); ++i) {
	REPORTER_ASSERT(r, out[i] == 0);
	}
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorPassthroughTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(half4 startingColor) {
	return startingColor;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{1.0f, 1.0f, 0.0f, 1.0f},
	/expectedResult=/SkColor4f{1.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorVariableGreenTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	half4 main(half4) {
	half2 zeroOne = half2(0, 1);
	half one = zeroOne.y, zero = zeroOne.x;
	return half4(zero, zeroOne.yx, one);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorAdditionTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	half4 main(half4 y) {
	half4 x = half4(-1, 0, 1, 0);
	half4 z = x.wzyx;
	z += y.000w;
	return y + z;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 1.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorIfElseTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	const half4 colorWhite = half4(1);

	half4 ifElseTest(half4 colorBlue, half4 colorGreen, half4 colorRed) {
	half4 result = half4(0);
	if (colorWhite != colorBlue) { // TRUE
	if (colorGreen == colorRed) { // FALSE
	result = colorRed;
	} else {
	result = colorGreen;
	}
	} else {
	if (colorRed != colorGreen) { // TRUE, but in a false branch
	result = colorBlue;
	} else { // FALSE, and in a false branch
	result = colorWhite;
	}
	}
	if (colorRed == colorBlue) { // FALSE
	return colorWhite;
	}
	if (colorRed != colorGreen) { // TRUE
	return result;
	}
	if (colorRed == colorWhite) { // FALSE
	return colorBlue;
	}
	return colorRed;
	}

	half4 main(half4) {
	return ifElseTest(colorWhite.00b1, colorWhite.0g01, colorWhite.r001);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorTernaryTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	half4 main(half4 colorWhite) {
	half4 colorBlue = colorWhite.00ba,
	colorGreen = colorWhite.0g0a,
	colorRed = colorWhite.r00a;
	// This ternary matches the initial if-else block inside IfElseTest.
	half4 result;
	result = (colorWhite != colorBlue) // TRUE
	? (colorGreen == colorRed ? colorRed : colorGreen) // FALSE
	: (colorRed != colorGreen ? colorBlue : colorWhite); // in false branch

	// This ternary matches the second portion of IfElseTest.
	return colorRed == colorBlue ? colorWhite :
	colorRed != colorGreen ? result : // TRUE
	colorRed == colorWhite ? colorBlue :
	colorRed;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{1.0, 1.0, 1.0, 1.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorTernarySideEffectTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1),
	colorRed = half4(1,0,0,1);
	half4 main(half4) {
	half x = 1, y = 1;
	(x == y) ? (x += 1) : (y += 1); // TRUE, x=2 y=1
	(x == y) ? (x += 3) : (y += 3); // FALSE, x=2 y=4
	(x < y) ? (x += 5) : (y += 5); // TRUE, x=7 y=4
	(y >= x) ? (x += 9) : (y += 9); // FALSE, x=7 y=13
	(x != y) ? (x += 1) : (y ); // TRUE, x=8 y=13
	(x == y) ? (x += 2) : (y ); // FALSE, x=8 y=13
	(x != y) ? (x ) : (y += 3); // TRUE, x=8 y=13
	(x == y) ? (x ) : (y += 4); // FALSE, x=8 y=17

	bool b = true;
	bool c = (b = false) ? false : b;

	return c ? colorRed : (x == 8 && y == 17) ? colorGreen : colorRed;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorNestedTernaryTest, r) {
	// Add in your SkSL here.
	test(r,
	R"__SkSL__(
	half4 main(half4) {
	half three = 3, one = 1, two = 2;
	half result = (three > (one > two ? 2.0 : 5.0)) ? 1.0 : 0.499;
	return half4(result);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.499f, 0.499f, 0.499f, 0.499f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorDoWhileTest, r) {
	// This is based on shared/DoWhileControlFlow.sksl (but avoids swizzles).
	test(r,
	R"__SkSL__(
	half r = 1.0, g = 1.0, b = 1.0;
	half4 main(half4) {
	half a = 1.0;
	// Verify that break is allowed in a do-while loop.
	do {
	r -= 0.25;
	if (r <= 0) break;
	} while (a == 1.0);

	// Verify that continue is allowed in a do-while loop.
	do {
	b -= 0.25;
	if (a == 1) continue; // should always happen
	g = 0;
	} while (b > 0.0);

	return half4(r, g, b, a);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorArithmeticTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(half4) {
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);

	half a = 3.0, b = 4.0, c = a + b - 2.0;
	if (aa + bb == ccc/5.0) {
	int A = 3, B = 4, C = A + B - 2;
	if (AA + BB == CCC/5) {
	return colorGreen;
	}
	}

	return colorRed;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorShortCircuitLogicalOr, r) {
	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 1) \|\| ((y += 1) == 2)) { // LHS true, RHS not executed but would be true
	return (x == 1 && y == 1) ? colorGreen : colorRed;
	} else {
	return colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 1) \|\| ((y += 1) == 3)) { // LHS true, RHS not executed but would be false
	return (x == 1 && y == 1) ? colorGreen : colorRed;
	} else {
	return colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 2) \|\| ((y += 1) == 2)) { // LHS false, RHS is executed and is true
	return (x == 1 && y == 2) ? colorGreen : colorRed;
	} else {
	return colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 2) \|\| ((y += 1) == 3)) { // LHS false, RHS is executed and is false
	return colorRed;
	} else {
	return (x == 1 && y == 2) ? colorGreen : colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorShortCircuitLogicalAnd, r) {
	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 1) && ((y += 1) == 2)) { // LHS true, RHS is executed and is true
	return (x == 1 && y == 2) ? colorGreen : colorRed;
	} else {
	return colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 1) && ((y += 1) == 3)) { // LHS true, RHS is executed and is false
	return colorRed;
	} else {
	return (x == 1 && y == 2) ? colorGreen : colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 2) && ((y += 1) == 2)) { // LHS false, RHS not executed but would be true
	return colorRed;
	} else {
	return (x == 1 && y == 1) ? colorGreen : colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});

	test(r,
	R"__SkSL__(
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 main(half4) {
	int x = 1, y = 1;
	if ((x == 2) && ((y += 1) == 3)) { // LHS false, RHS not executed but would be false
	return colorRed;
	} else {
	return (x == 1 && y == 1) ? colorGreen : colorRed;
	}
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0f, 1.0f, 0.0f, 1.0f});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorCoercedTypeTest, r) {
	static constexpr float kUniforms[] = {0.0, 1.0, 0.0, 1.0,
	1.0, 0.0, 0.0, 1.0};
	test(r,
	R"__SkSL__(
	uniform half4 colorGreen;
	uniform float4 colorRed;
	half4 main(half4 color) {
	return ((colorGreen + colorRed) == float4(1.0, 1.0, 0.0, 2.0)) ? colorGreen
	: colorGreen.gr01;
	}
	)__SkSL__",
	kUniforms,
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0, 1.0, 0.0, 1.0});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorIdentitySwizzle, r) {
	static constexpr float kUniforms[] = {0.0, 1.0, 0.0, 1.0,
	1.0, 0.0, 0.0, 1.0};
	test(r,
	R"__SkSL__(
	uniform half4 colorGreen, colorRed;
	half4 main(vec4 color) {
	return (color.r == 0.5 &&
	color.rg == half2(0.5, 1.0) &&
	color.rgb == half3(0.5, 1.0, 0.0)) ? colorGreen : colorRed;
	}
	)__SkSL__",
	kUniforms,
	/startingColor=/SkColor4f{0.5, 1.0, 0.0, 0.25},
	/expectedResult=/SkColor4f{0.0, 1.0, 0.0, 1.0});

	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaTernaryTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(vec4 color) {
	half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

	half scale = luma < 0.33333 ? 0.5
	: luma < 0.66666 ? (0.166666 + 2.0 * (luma - 0.33333)) / luma
	: /* else / (0.833333 + 0.5 (luma - 0.66666)) / luma;
	return half4(color.rgb * scale, color.a);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.25, 0.00, 0.75, 1.0},
	/expectedResult=/SkColor4f{0.125, 0.0, 0.375, 1.0});

	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaIfNoEarlyReturnTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(vec4 color) {
	half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

	half scale = 0;
	if (luma < 0.33333) {
	scale = 0.5;
	} else if (luma < 0.66666) {
	scale = (0.166666 + 2.0 * (luma - 0.33333)) / luma;
	} else {
	scale = (0.833333 + 0.5 * (luma - 0.66666)) / luma;
	}
	return half4(color.rgb * scale, color.a);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.25, 0.00, 0.75, 1.0},
	/expectedResult=/SkColor4f{0.125, 0.0, 0.375, 1.0});

	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorLumaWithEarlyReturnTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(half4 color) {
	half luma = dot(color.rgb, half3(0.3, 0.6, 0.1));

	half scale = 0;
	if (luma < 0.33333) {
	return half4(color.rgb * 0.5, color.a);
	} else if (luma < 0.66666) {
	scale = 0.166666 + 2.0 * (luma - 0.33333);
	} else {
	scale = 0.833333 + 0.5 * (luma - 0.66666);
	}
	return half4(color.rgb * (scale/luma), color.a);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.25, 0.00, 0.75, 1.0},
	/expectedResult=/SkColor4f{0.125, 0.0, 0.375, 1.0});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorBitwiseNotTest, r) {
	static constexpr int32_t kUniforms[] = { 0, 12, 3456, 4567890,
	~0, ~12, ~3456, ~4567890};
	test(r,
	R"__SkSL__(
	uniform int4 value, expected;
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);

	half4 main(vec4) {
	return (~value.x == expected.x &&
	~value.xy == expected.xy &&
	~value.xyz == expected.xyz &&
	~value.xyzw == expected.xyzw) ? colorGreen : colorRed;
	}
	)__SkSL__",
	SkSpan((const float*)kUniforms, std::size(kUniforms)),
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0, 1.0, 0.0, 1.0});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorComparisonIntrinsicTest, r) {
	test(r,
	R"__SkSL__(
	half4 main(vec4) {
	const half4 colorGreen = half4(0,1,0,1), colorRed = half4(1,0,0,1);
	half4 a = half4(1, 2, 0, 1),
	b = half4(2, 2, 1, 0);
	int3 c = int3(1111, 3333, 5555),
	d = int3(1111, 5555, 3333);
	uint2 e = uint2(1111111111u, 222),
	f = uint2(3333333333u, 222);
	return (lessThan(a, b) == bool4(true, false, true, false) &&
	lessThan(c, d) == bool3(false, true, false) &&
	lessThan(e, f) == bool2(true, false) &&
	greaterThan(a, b) == bool4(false, false, false, true) &&
	greaterThan(c, d) == bool3(false, false, true) &&
	greaterThan(e, f) == bool2(false, false) &&
	lessThanEqual(a, b) == bool4(true, true, true, false) &&
	lessThanEqual(c, d) == bool3(true, true, false) &&
	lessThanEqual(e, f) == bool2(true, true) &&
	greaterThanEqual(a, b) == bool4(false, true, false, true) &&
	greaterThanEqual(c, d) == bool3(true, false, true) &&
	greaterThanEqual(e, f) == bool2(false, true) &&
	equal(a, b) == bool4(false, true, false, false) &&
	equal(c, d) == bool3(true, false, false) &&
	equal(e, f) == bool2(false, true) &&
	notEqual(a, b) == bool4(true, false, true, true) &&
	notEqual(c, d) == bool3(false, true, true) &&
	notEqual(e, f) == bool2(true, false) &&
	max(a, b) == half4(2, 2, 1, 1) &&
	max(c, d) == int3(1111, 5555, 5555) &&
	max(e, f) == uint2(3333333333u, 222) &&
	max(a, 1) == half4(1, 2, 1, 1) &&
	max(c, 3333) == int3(3333, 3333, 5555) &&
	max(e, 7777) == uint2(1111111111u, 7777) &&
	min(a, b) == half4(1, 2, 0, 0) &&
	min(c, d) == int3(1111, 3333, 3333) &&
	min(e, f) == uint2(1111111111u, 222) &&
	min(a, 1) == half4(1, 1, 0, 1) &&
	min(c, 3333) == int3(1111, 3333, 3333) &&
	min(e, 7777) == uint2(7777, 222)) ? colorGreen : colorRed;
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.0, 0.0, 0.0, 0.0},
	/expectedResult=/SkColor4f{0.0, 1.0, 0.0, 1.0});
	}

	DEF_TEST(SkSLRasterPipelineCodeGeneratorUnpremulTest, r) {
	test(r,
	R"__SkSL__(
	// This is `unpremul` verbatim from sksl_shared, but marked noinline.
	noinline half4 MyUnpremul(half4 color) {
	return half4(color.rgb / max(color.a, 0.0001), color.a);
	}

	half4 main(vec4 color) {
	return MyUnpremul(color);
	}
	)__SkSL__",
	/uniforms=/{},
	/startingColor=/SkColor4f{0.5, 0.25, 0.125, 0.5},
	/expectedResult=/SkColor4f{1.0, 0.5, 0.25, 0.5});
	}