mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user