From d55f22d046f5b54ec6d77af040aa1805e12ea9d1 Mon Sep 17 00:00:00 2001
From: David Faust <david.faust@oracle.com>
Date: Wed, 5 Oct 2022 10:10:18 -0700
Subject: ast: dump assignment and compound assignment expr

---
 gcc/rust/ast/rust-ast-dump.cc | 56 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 54 insertions(+), 2 deletions(-)

(limited to 'gcc')

diff --git a/gcc/rust/ast/rust-ast-dump.cc b/gcc/rust/ast/rust-ast-dump.cc
index 8ad00b5..f137f4e 100644
--- a/gcc/rust/ast/rust-ast-dump.cc
+++ b/gcc/rust/ast/rust-ast-dump.cc
@@ -281,11 +281,63 @@ Dump::visit (TypeCastExpr &expr)
 
 void
 Dump::visit (AssignmentExpr &expr)
-{}
+{
+  expr.visit_lhs (*this);
+  stream << " = ";
+  expr.visit_rhs (*this);
+}
 
 void
 Dump::visit (CompoundAssignmentExpr &expr)
-{}
+{
+  auto op = "";
+  switch (expr.get_expr_type ())
+    {
+    case CompoundAssignmentOperator::ADD:
+      op = "+";
+      break;
+
+    case CompoundAssignmentOperator::SUBTRACT:
+      op = "-";
+      break;
+
+    case CompoundAssignmentOperator::MULTIPLY:
+      op = "*";
+      break;
+
+    case CompoundAssignmentOperator::DIVIDE:
+      op = "/";
+      break;
+
+    case CompoundAssignmentOperator::MODULUS:
+      op = "%";
+      break;
+
+    case CompoundAssignmentOperator::BITWISE_AND:
+      op = "&";
+      break;
+
+    case CompoundAssignmentOperator::BITWISE_OR:
+      op = "|";
+      break;
+
+    case CompoundAssignmentOperator::BITWISE_XOR:
+      op = "^";
+      break;
+
+    case CompoundAssignmentOperator::LEFT_SHIFT:
+      op = "<<";
+      break;
+
+    case CompoundAssignmentOperator::RIGHT_SHIFT:
+      op = ">>";
+      break;
+    }
+
+  expr.get_left_expr ()->accept_vis (*this);
+  stream << " " << op << "= ";
+  expr.get_right_expr ()->accept_vis (*this);
+}
 
 void
 Dump::visit (GroupedExpr &expr)
-- 
cgit v1.1