Supports switch expressions by AST translators.

PiperOrigin-RevId: 729284458

Author: tomball

translator/src/main/java/com/google/devtools/j2objc/translate/Autoboxer.java

@@ -40,6 +40,7 @@
 import com.google.devtools.j2objc.ast.SimpleName;
 import com.google.devtools.j2objc.ast.SuperConstructorInvocation;
 import com.google.devtools.j2objc.ast.SuperMethodInvocation;
+import com.google.devtools.j2objc.ast.SwitchExpression;
 import com.google.devtools.j2objc.ast.SwitchStatement;
 import com.google.devtools.j2objc.ast.TreeNode;
 import com.google.devtools.j2objc.ast.TreeUtil;
@@ -463,6 +464,15 @@ public void endVisit(WhileStatement node) {
     }
   }
 
+  @Override
+  @SuppressWarnings("UngroupedOverloads")
+  public void endVisit(SwitchExpression node) {
+    Expression expression = node.getExpression();
+    if (!expression.getTypeMirror().getKind().isPrimitive()) {
+      unbox(expression);
+    }
+  }
+
   @Override
   public void endVisit(SwitchStatement node) {
     Expression expression = node.getExpression();

translator/src/main/java/com/google/devtools/j2objc/translate/NilCheckResolver.java

@@ -50,6 +50,7 @@
 import com.google.devtools.j2objc.ast.SuperConstructorInvocation;
 import com.google.devtools.j2objc.ast.SuperMethodInvocation;
 import com.google.devtools.j2objc.ast.SwitchCase;
+import com.google.devtools.j2objc.ast.SwitchExpression;
 import com.google.devtools.j2objc.ast.SwitchStatement;
 import com.google.devtools.j2objc.ast.ThrowStatement;
 import com.google.devtools.j2objc.ast.TreeNode;
@@ -714,6 +715,20 @@ public boolean visit(ForStatement node) {
     return false;
   }
 
+  @Override
+  @SuppressWarnings("UngroupedOverloads")
+  public boolean visit(SwitchExpression node) {
+    // No need to test for no matched value like SwitchStatement,
+    // as switch expressions are exhaustive.
+    node.getExpression().accept(this);
+    pushLoopOrSwitchScope(null);
+    for (Statement stmt : node.getStatements()) {
+      stmt.accept(this);
+    }
+    popAndMerge();
+    return false;
+  }
+
   @Override
   public boolean visit(SwitchStatement node) {
     node.getExpression().accept(this);

translator/src/main/java/com/google/devtools/j2objc/translate/UnsequencedExpressionRewriter.java

@@ -43,6 +43,7 @@
 import com.google.devtools.j2objc.ast.SimpleName;
 import com.google.devtools.j2objc.ast.Statement;
 import com.google.devtools.j2objc.ast.SuperConstructorInvocation;
+import com.google.devtools.j2objc.ast.SwitchExpression;
 import com.google.devtools.j2objc.ast.SwitchStatement;
 import com.google.devtools.j2objc.ast.SynchronizedStatement;
 import com.google.devtools.j2objc.ast.ThrowStatement;
@@ -531,6 +532,18 @@ public boolean visit(IfStatement node) {
     return false;
   }
 
+  @Override
+  @SuppressWarnings("UngroupedOverloads")
+  public boolean visit(SwitchExpression node) {
+    Expression expr = node.getExpression();
+    newExpression(expr);
+    expr.accept(this);
+    for (Statement stmt : node.getStatements()) {
+      stmt.accept(this);
+    }
+    return false;
+  }
+
   @Override
   public boolean visit(SwitchStatement node) {
     visitAndExtract(node.getExpression(), node);