Add support for compound assignment and increment (#2526)

Add support for user-defined assignment, as well as compound assignment and increment, following the design direction in pending proposal #2511.

Some of this isn't fully testable yet: because explorer doesn't properly support `impl` specialization, the blanket `impl`s in the prelude prevent types from customizing assignment.
This commit is contained in:
Richard Smith
2023-01-20 18:48:13 -08:00
committed by GitHub
parent f1a0645551
commit 2fd7e2b65d
32 changed files with 975 additions and 215 deletions
+34 -1
View File
@@ -72,7 +72,13 @@ void Statement::PrintDepth(int depth, llvm::raw_ostream& out) const {
break;
case StatementKind::Assign: {
const auto& assign = cast<Assign>(*this);
out << assign.lhs() << " = " << assign.rhs() << ";";
out << assign.lhs() << " " << AssignOperatorToString(assign.op()) << " "
<< assign.rhs() << ";";
break;
}
case StatementKind::IncrementDecrement: {
const auto& inc_dec = cast<IncrementDecrement>(*this);
out << (inc_dec.is_increment() ? "++" : "--") << inc_dec.argument();
break;
}
case StatementKind::If: {
@@ -137,4 +143,31 @@ void Statement::PrintDepth(int depth, llvm::raw_ostream& out) const {
}
}
auto AssignOperatorToString(AssignOperator op) -> std::string_view {
switch (op) {
case AssignOperator::Plain:
return "=";
case AssignOperator::Add:
return "+=";
case AssignOperator::Div:
return "/=";
case AssignOperator::Mul:
return "*=";
case AssignOperator::Mod:
return "%=";
case AssignOperator::Sub:
return "-=";
case AssignOperator::And:
return "&=";
case AssignOperator::Or:
return "|=";
case AssignOperator::Xor:
return "^=";
case AssignOperator::ShiftLeft:
return "<<=";
case AssignOperator::ShiftRight:
return ">>=";
}
}
} // namespace Carbon