Optimize (boolExpr == true) and (boolExpr != false) into boolExpr.

Additionally, restructure the unit test to return a color (green for
pass, red for fail).

Change-Id: Ib1bb6bd8771c72cc751d8d2c65cc14a693166d4c
Bug: skia:11112
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/356301
Commit-Queue: John Stiles <johnstiles@google.com>
Commit-Queue: Brian Osman <brianosman@google.com>
Auto-Submit: John Stiles <johnstiles@google.com>
Reviewed-by: Brian Osman <brianosman@google.com>

src/sksl/SkSLConstantFolder.cpp

@@ -39,6 +39,14 @@ static std::unique_ptr<Expression> short_circuit_boolean(const Expression& left,
         return leftVal ? std::make_unique<BoolLiteral>(left.fOffset, /*value=*/true, &left.type())
                        : right.clone();
     }
+    if (op == Token::Kind::TK_EQEQ && leftVal) {
+        // (true == expr) -> (expr)
+        return right.clone();
+    }
+    if (op == Token::Kind::TK_NEQ && !leftVal) {
+        // (false != expr) -> (expr)
+        return right.clone();
+    }
     if (op == Token::Kind::TK_LOGICALXOR && !leftVal) {
         // (false ^^ expr) -> (expr)
         return right.clone();

tests/sksl/folding/ShortCircuitBoolFolding.sksl

@@ -1,37 +1,35 @@
-void main() {
+bool test() {
-    bool expr1 = sk_FragCoord.x > 0;
+    bool expr = sqrt(1) > 0;
-    bool expr2 = sk_FragCoord.y > 0;
 
-    if (true && expr1) {           // -> if (expr1)
+    int ok = 0, bad = 0;
-        sk_FragColor.r = 1;
-    } else if (false && expr1) {   // -> if (false) -> block removed
-        sk_FragColor.r = -2;
-    } else if (true ^^ expr1) {    // -> if (!expr1)
-        sk_FragColor.r = 3;
-    } else if (false ^^ expr2) {   // -> if (expr2)
-        sk_FragColor.r = 4;
-    } else if (false || expr2) {   // -> if (expr2)
-        sk_FragColor.r = 5;
-    } else if (true || expr2) {    // -> if (true) -> replaces unreachable else
-        sk_FragColor.r = 6;
-    } else {                       // removed
-        sk_FragColor.r = -7;
-    }
 
-    // Test short-circuiting of right hand side boolean literals
+    // Test boolean short-circuiting with constants on the left side.
-    if (expr1 && true) {           // -> if (expr1)
+    if (true  && expr) { ++ok; } else { ++bad; } // -> (expr)
-        sk_FragColor.r = 1;
+    if (false && expr) { ++bad; } else { ++ok; } // -> (false) -> block removed
-    } else if (expr1 && false) {   // -> if (false) -> block removed
+    if (true  ^^ expr) { ++bad; } else { ++ok; } // -> unchanged
-        sk_FragColor.r = -2;
+    if (false ^^ expr) { ++ok; } else { ++bad; } // -> (expr)
-    } else if (expr1 ^^ true) {    // -> if (!expr1)
+    if (true  || expr) { ++ok; } else { ++bad; } // -> (true)
-        sk_FragColor.r = 3;
+    if (false || expr) { ++ok; } else { ++bad; } // -> (expr)
-    } else if (expr2 ^^ false) {   // -> if (expr2)
+    if (true  == expr) { ++ok; } else { ++bad; } // -> (expr)
-        sk_FragColor.r = 4;
+    if (false == expr) { ++bad; } else { ++ok; } // -> unchanged
-    } else if (expr2 || false) {   // -> if (expr2)
+    if (true  != expr) { ++bad; } else { ++ok; } // -> unchanged
-        sk_FragColor.r = 5;
+    if (false != expr) { ++ok; } else { ++bad; } // -> (expr)
-    } else if (expr2 || true) {    // -> if (true) -> replaces unreachable else
+
-        sk_FragColor.r = 6;
+    // Test boolean short-circuiting with constants on the right side.
-    } else {                       // removed
+    if (expr && true ) { ++ok; } else { ++bad; } // -> (expr)
-        sk_FragColor.r = -7;
+    if (expr && false) { ++bad; } else { ++ok; } // -> (false) -> block removed
-    }
+    if (expr ^^ true ) { ++bad; } else { ++ok; } // -> unchanged
+    if (expr ^^ false) { ++ok; } else { ++bad; } // -> (expr)
+    if (expr || true ) { ++ok; } else { ++bad; } // -> (true)
+    if (expr || false) { ++ok; } else { ++bad; } // -> (expr)
+    if (expr == true ) { ++ok; } else { ++bad; } // -> (expr)
+    if (expr == false) { ++bad; } else { ++ok; } // -> unchanged
+    if (expr != true ) { ++bad; } else { ++ok; } // -> unchanged
+    if (expr != false) { ++ok; } else { ++bad; } // -> (expr)
+
+    return ok == 20 && bad == 0;
+}
+
+half4 main() {
+    return test() ? half4(0,1,0,1) : half4(1,0,0,1);
 }

tests/sksl/folding/golden/ShortCircuitBoolFolding.glsl

@@ -1,28 +1,101 @@
 
-out vec4 sk_FragColor;
+vec4 main() {
-void main() {
+    bool _1_expr = sqrt(1.0) > 0.0;
-    bool expr1 = gl_FragCoord.x > 0.0;
+    int _2_ok = 0;
-    bool expr2 = gl_FragCoord.y > 0.0;
+    int _3_bad = 0;
-    if (expr1) {
+
-        sk_FragColor.x = 1.0;
+    if (_1_expr) {
-    } else if (true ^^ expr1) {
+        ++_2_ok;
-        sk_FragColor.x = 3.0;
-    } else if (expr2) {
-        sk_FragColor.x = 4.0;
-    } else if (expr2) {
-        sk_FragColor.x = 5.0;
     } else {
-        sk_FragColor.x = 6.0;
+        ++_3_bad;
     }
-    if (expr1) {
+    {
-        sk_FragColor.x = 1.0;
+        ++_2_ok;
-    } else if (expr1 ^^ true) {
+    }
-        sk_FragColor.x = 3.0;
+    if (true ^^ _1_expr) {
-    } else if (expr2) {
+        ++_3_bad;
-        sk_FragColor.x = 4.0;
-    } else if (expr2) {
-        sk_FragColor.x = 5.0;
     } else {
-        sk_FragColor.x = 6.0;
+        ++_2_ok;
     }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    {
+        ++_2_ok;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    if (false == _1_expr) {
+        ++_3_bad;
+    } else {
+        ++_2_ok;
+    }
+    if (true != _1_expr) {
+        ++_3_bad;
+    } else {
+        ++_2_ok;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    {
+        ++_2_ok;
+    }
+    if (_1_expr ^^ true) {
+        ++_3_bad;
+    } else {
+        ++_2_ok;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    {
+        ++_2_ok;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    if (_1_expr == false) {
+        ++_3_bad;
+    } else {
+        ++_2_ok;
+    }
+    if (_1_expr != true) {
+        ++_3_bad;
+    } else {
+        ++_2_ok;
+    }
+    if (_1_expr) {
+        ++_2_ok;
+    } else {
+        ++_3_bad;
+    }
+    return _2_ok == 20 && _3_bad == 0 ? vec4(0.0, 1.0, 0.0, 1.0) : vec4(1.0, 0.0, 0.0, 1.0);
+
 }