mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-10-06 09:24:46 +01:00
Switch Statement to std::variant (#639)
Aligns Assign/Assignment, If/IfStatement naming to favor the name in the enum.
This commit is contained in:
@@ -30,68 +30,96 @@ enum class StatementKind {
|
||||
|
||||
struct Statement;
|
||||
|
||||
struct Assignment {
|
||||
struct ExpressionStatement {
|
||||
static constexpr StatementKind Kind = StatementKind::ExpressionStatement;
|
||||
const Expression* exp;
|
||||
};
|
||||
|
||||
struct Assign {
|
||||
static constexpr StatementKind Kind = StatementKind::Assign;
|
||||
const Expression* lhs;
|
||||
const Expression* rhs;
|
||||
};
|
||||
|
||||
struct VariableDefinition {
|
||||
static constexpr StatementKind Kind = StatementKind::VariableDefinition;
|
||||
const Expression* pat;
|
||||
const Expression* init;
|
||||
};
|
||||
|
||||
struct IfStatement {
|
||||
struct If {
|
||||
static constexpr StatementKind Kind = StatementKind::If;
|
||||
const Expression* cond;
|
||||
const Statement* then_stmt;
|
||||
const Statement* else_stmt;
|
||||
};
|
||||
|
||||
struct Return {
|
||||
static constexpr StatementKind Kind = StatementKind::Return;
|
||||
const Expression* exp;
|
||||
};
|
||||
|
||||
struct Sequence {
|
||||
static constexpr StatementKind Kind = StatementKind::Sequence;
|
||||
const Statement* stmt;
|
||||
const Statement* next;
|
||||
};
|
||||
|
||||
struct Block {
|
||||
static constexpr StatementKind Kind = StatementKind::Block;
|
||||
const Statement* stmt;
|
||||
};
|
||||
|
||||
struct While {
|
||||
static constexpr StatementKind Kind = StatementKind::While;
|
||||
const Expression* cond;
|
||||
const Statement* body;
|
||||
};
|
||||
|
||||
struct Break {
|
||||
static constexpr StatementKind Kind = StatementKind::Break;
|
||||
};
|
||||
|
||||
struct Continue {
|
||||
static constexpr StatementKind Kind = StatementKind::Continue;
|
||||
};
|
||||
|
||||
struct Match {
|
||||
static constexpr StatementKind Kind = StatementKind::Match;
|
||||
const Expression* exp;
|
||||
std::list<std::pair<const Expression*, const Statement*>>* clauses;
|
||||
};
|
||||
|
||||
struct Continuation {
|
||||
std::string* continuation_variable;
|
||||
static constexpr StatementKind Kind = StatementKind::Continuation;
|
||||
std::string continuation_variable;
|
||||
const Statement* body;
|
||||
};
|
||||
|
||||
struct Run {
|
||||
static constexpr StatementKind Kind = StatementKind::Run;
|
||||
const Expression* argument;
|
||||
};
|
||||
|
||||
struct Statement {
|
||||
// TODO: change Statement to a class and make all members private
|
||||
int line_num;
|
||||
StatementKind tag;
|
||||
struct Await {
|
||||
static constexpr StatementKind Kind = StatementKind::Await;
|
||||
};
|
||||
|
||||
struct Statement {
|
||||
// Constructors
|
||||
static auto MakeExpStmt(int line_num, const Expression* exp)
|
||||
static auto MakeExpressionStatement(int line_num, const Expression* exp)
|
||||
-> const Statement*;
|
||||
static auto MakeAssign(int line_num, const Expression* lhs,
|
||||
const Expression* rhs) -> const Statement*;
|
||||
static auto MakeVarDef(int line_num, const Expression* pat,
|
||||
const Expression* init) -> const Statement*;
|
||||
static auto MakeVariableDefinition(int line_num, const Expression* pat,
|
||||
const Expression* init)
|
||||
-> const Statement*;
|
||||
static auto MakeIf(int line_num, const Expression* cond,
|
||||
const Statement* then_stmt, const Statement* else_stmt)
|
||||
-> const Statement*;
|
||||
static auto MakeReturn(int line_num, const Expression* e) -> const Statement*;
|
||||
static auto MakeSeq(int line_num, const Statement* s1, const Statement* s2)
|
||||
-> const Statement*;
|
||||
static auto MakeSequence(int line_num, const Statement* s1,
|
||||
const Statement* s2) -> const Statement*;
|
||||
static auto MakeBlock(int line_num, const Statement* s) -> const Statement*;
|
||||
static auto MakeWhile(int line_num, const Expression* cond,
|
||||
const Statement* body) -> const Statement*;
|
||||
@@ -119,33 +147,32 @@ struct Statement {
|
||||
// __await;
|
||||
static auto MakeAwait(int line_num) -> const Statement*;
|
||||
|
||||
// Access to the alternatives
|
||||
const Expression* GetExpression() const;
|
||||
Assignment GetAssign() const;
|
||||
VariableDefinition GetVariableDefinition() const;
|
||||
IfStatement GetIf() const;
|
||||
const Expression* GetReturn() const;
|
||||
Sequence GetSequence() const;
|
||||
Block GetBlock() const;
|
||||
While GetWhile() const;
|
||||
Match GetMatch() const;
|
||||
Continuation GetContinuation() const;
|
||||
Run GetRun() const;
|
||||
auto GetExpressionStatement() const -> const ExpressionStatement&;
|
||||
auto GetAssign() const -> const Assign&;
|
||||
auto GetVariableDefinition() const -> const VariableDefinition&;
|
||||
auto GetIf() const -> const If&;
|
||||
auto GetReturn() const -> const Return&;
|
||||
auto GetSequence() const -> const Sequence&;
|
||||
auto GetBlock() const -> const Block&;
|
||||
auto GetWhile() const -> const While&;
|
||||
auto GetBreak() const -> const Break&;
|
||||
auto GetContinue() const -> const Continue&;
|
||||
auto GetMatch() const -> const Match&;
|
||||
auto GetContinuation() const -> const Continuation&;
|
||||
auto GetRun() const -> const Run&;
|
||||
auto GetAwait() const -> const Await&;
|
||||
|
||||
inline auto tag() const -> StatementKind {
|
||||
return std::visit([](const auto& t) { return t.Kind; }, value);
|
||||
}
|
||||
|
||||
int line_num;
|
||||
|
||||
private:
|
||||
union {
|
||||
const Expression* exp;
|
||||
Assignment assign;
|
||||
VariableDefinition variable_definition;
|
||||
IfStatement if_stmt;
|
||||
const Expression* return_stmt;
|
||||
Sequence sequence;
|
||||
Block block;
|
||||
While while_stmt;
|
||||
Match match_stmt;
|
||||
Continuation continuation;
|
||||
Run run;
|
||||
} u;
|
||||
std::variant<ExpressionStatement, Assign, VariableDefinition, If, Return,
|
||||
Sequence, Block, While, Break, Continue, Match, Continuation,
|
||||
Run, Await>
|
||||
value;
|
||||
};
|
||||
|
||||
void PrintStatement(const Statement*, int);
|
||||
|
||||
Reference in New Issue
Block a user