blob: 9f6a62ff37c29624a6e9df975ad582bf659fa3ca (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright (C) 2025 Free Software Foundation, Inc.
// This file is part of GCC.
// GCC is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 3, or (at your option) any later
// version.
// GCC is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
// for more details.
// You should have received a copy of the GNU General Public License
// along with GCC; see the file COPYING3. If not see
// <http://www.gnu.org/licenses/>.
#include "rust-expression-yeast.h"
#include "rust-ast-visitor.h"
#include "rust-desugar-question-mark.h"
#include "rust-desugar-try-block.h"
#include "rust-desugar-for-loops.h"
#include "rust-ast-full.h"
#include "rust-desugar-while-let.h"
#include "rust-expr.h"
#include "rust-stmt.h"
namespace Rust {
namespace AST {
void
ExpressionYeast::go (AST::Crate &crate)
{
DefaultASTVisitor::visit (crate);
}
void
ExpressionYeast::dispatch_loops (std::unique_ptr<Expr> &loop_expr)
{
auto &loop = static_cast<BaseLoopExpr &> (*loop_expr.get ());
switch (loop.get_loop_kind ())
{
case BaseLoopExpr::Kind::For:
DesugarForLoops::go (loop_expr);
break;
case BaseLoopExpr::Kind::WhileLet:
DesugarWhileLet::go (loop_expr);
break;
default:
break;
}
}
void
ExpressionYeast::dispatch (std::unique_ptr<Expr> &expr)
{
switch (expr->get_expr_kind ())
{
case Expr::Kind::ErrorPropagation:
DesugarQuestionMark::go (expr);
break;
case Expr::Kind::Try:
DesugarTryBlock::go (expr);
break;
case Expr::Kind::Loop:
dispatch_loops (expr);
break;
default:
break;
}
}
void
ExpressionYeast::visit (ExprStmt &stmt)
{
dispatch (stmt.get_expr_ptr ());
DefaultASTVisitor::visit (stmt);
}
void
ExpressionYeast::visit (CallExpr &call)
{
dispatch (call.get_function_expr_ptr ());
for (auto &arg : call.get_params ())
dispatch (arg);
DefaultASTVisitor::visit (call);
}
void
ExpressionYeast::visit (BlockExpr &block)
{
for (auto &stmt : block.get_statements ())
if (stmt->get_stmt_kind () == Stmt::Kind::Expr)
dispatch (static_cast<ExprStmt &> (*stmt).get_expr_ptr ());
if (block.has_tail_expr ())
dispatch (block.get_tail_expr_ptr ());
DefaultASTVisitor::visit (block);
}
void
ExpressionYeast::visit (LetStmt &stmt)
{
if (stmt.has_init_expr ())
dispatch (stmt.get_init_expr_ptr ());
DefaultASTVisitor::visit (stmt);
}
} // namespace AST
} // namespace Rust
|
