mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
When the struct definition has no user-defined constructors, and there is at least one instance variable, add an implicit default constructor. This default constructor is currently empty; that's to be fixed next. Swift SVN r4868
579 lines
18 KiB
C++
579 lines
18 KiB
C++
//===--- Stmt.h - Swift Language Statement ASTs -----------------*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines the Stmt class and subclasses.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_AST_STMT_H
|
|
#define SWIFT_AST_STMT_H
|
|
|
|
#include "swift/Basic/LLVM.h"
|
|
#include "swift/Basic/SourceLoc.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ADT/NullablePtr.h"
|
|
#include "llvm/ADT/PointerUnion.h"
|
|
|
|
namespace swift {
|
|
class ASTContext;
|
|
class Decl;
|
|
class Expr;
|
|
class ASTWalker;
|
|
class Pattern;
|
|
class PatternBindingDecl;
|
|
class VarDecl;
|
|
|
|
enum class StmtKind {
|
|
#define STMT(ID, PARENT) ID,
|
|
#include "swift/AST/StmtNodes.def"
|
|
};
|
|
|
|
/// Stmt - Base class for all statements in swift.
|
|
class alignas(8) Stmt {
|
|
Stmt(const Stmt&) = delete;
|
|
void operator=(const Stmt&) = delete;
|
|
|
|
/// Kind - The subclass of Stmt that this is.
|
|
const StmtKind Kind;
|
|
|
|
public:
|
|
Stmt(StmtKind kind) : Kind(kind) {}
|
|
|
|
StmtKind getKind() const { return Kind; }
|
|
|
|
/// \brief Return the location of the start of the statement.
|
|
SourceLoc getStartLoc() const { return getSourceRange().Start; }
|
|
|
|
/// \brief Return the location of the end of the statement.
|
|
SourceLoc getEndLoc() const { return getSourceRange().End; }
|
|
|
|
SourceRange getSourceRange() const;
|
|
SourceLoc TrailingSemiLoc;
|
|
|
|
/// isImplicit - Determines whether this statement was implicitly-generated,
|
|
/// rather than explicitly written in the AST.
|
|
bool isImplicit() const;
|
|
|
|
/// walk - This recursively walks the AST rooted at this statement.
|
|
Stmt *walk(ASTWalker &walker);
|
|
Stmt *walk(ASTWalker &&walker) { return walk(walker); }
|
|
|
|
|
|
void dump() const;
|
|
void print(raw_ostream &OS, unsigned Indent = 0) const;
|
|
|
|
// Only allow allocation of Exprs using the allocator in ASTContext
|
|
// or by doing a placement new.
|
|
void *operator new(size_t Bytes, ASTContext &C,
|
|
unsigned Alignment = alignof(Stmt));
|
|
|
|
// Make vanilla new/delete illegal for Stmts.
|
|
void *operator new(size_t Bytes) throw() = delete;
|
|
void operator delete(void *Data) throw() = delete;
|
|
void *operator new(size_t Bytes, void *Mem) throw() = delete;
|
|
};
|
|
|
|
/// AssignStmt - A value assignment, like "x = y".
|
|
class AssignStmt : public Stmt {
|
|
Expr *Dest;
|
|
Expr *Src;
|
|
SourceLoc EqualLoc;
|
|
|
|
public:
|
|
AssignStmt(Expr *Dest, SourceLoc EqualLoc, Expr *Src)
|
|
: Stmt(StmtKind::Assign), Dest(Dest), Src(Src), EqualLoc(EqualLoc) {}
|
|
|
|
Expr *getDest() const { return Dest; }
|
|
void setDest(Expr *e) { Dest = e; }
|
|
Expr *getSrc() const { return Src; }
|
|
void setSrc(Expr *e) { Src = e; }
|
|
|
|
SourceLoc getEqualLoc() const { return EqualLoc; }
|
|
|
|
SourceRange getSourceRange() const;
|
|
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::Assign; }
|
|
};
|
|
|
|
|
|
/// BraceStmt - A brace enclosed sequence of expressions, stmts, or decls, like
|
|
/// { var x = 10; println(10) }.
|
|
class BraceStmt : public Stmt {
|
|
public:
|
|
typedef llvm::PointerUnion3<Expr*, Stmt*, Decl*> ExprStmtOrDecl;
|
|
|
|
private:
|
|
unsigned NumElements;
|
|
|
|
SourceLoc LBLoc;
|
|
SourceLoc RBLoc;
|
|
|
|
BraceStmt(SourceLoc lbloc, ArrayRef<ExprStmtOrDecl> elements,SourceLoc rbloc);
|
|
ExprStmtOrDecl *getElementsStorage() {
|
|
return reinterpret_cast<ExprStmtOrDecl*>(this + 1);
|
|
}
|
|
|
|
public:
|
|
static BraceStmt *create(ASTContext &ctx, SourceLoc lbloc,
|
|
ArrayRef<ExprStmtOrDecl> elements,
|
|
SourceLoc rbloc);
|
|
|
|
SourceLoc getLBraceLoc() const { return LBLoc; }
|
|
SourceLoc getRBraceLoc() const { return RBLoc; }
|
|
|
|
SourceRange getSourceRange() const { return SourceRange(LBLoc, RBLoc); }
|
|
|
|
/// The elements contained within the BraceStmt.
|
|
MutableArrayRef<ExprStmtOrDecl> getElements() {
|
|
return MutableArrayRef<ExprStmtOrDecl>(getElementsStorage(), NumElements);
|
|
}
|
|
|
|
/// The elements contained within the BraceStmt (const version).
|
|
ArrayRef<ExprStmtOrDecl> getElements() const {
|
|
return const_cast<BraceStmt*>(this)->getElements();
|
|
}
|
|
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::Brace; }
|
|
};
|
|
|
|
/// ReturnStmt - A return statement. The result is optional; "return" without
|
|
/// an expression is semantically equivalent to "return ()".
|
|
/// return 42
|
|
class ReturnStmt : public Stmt {
|
|
SourceLoc ReturnLoc;
|
|
Expr *Result;
|
|
|
|
public:
|
|
ReturnStmt(SourceLoc ReturnLoc, Expr *Result)
|
|
: Stmt(StmtKind::Return), ReturnLoc(ReturnLoc), Result(Result) {}
|
|
|
|
SourceRange getSourceRange() const;
|
|
SourceLoc getReturnLoc() const { return ReturnLoc; }
|
|
|
|
bool hasResult() { return Result != 0; }
|
|
Expr *getResult() const {
|
|
assert(Result && "ReturnStmt doesn't have a result");
|
|
return Result;
|
|
}
|
|
void setResult(Expr *e) { Result = e; }
|
|
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::Return;}
|
|
};
|
|
|
|
/// IfStmt - if/then/else statement. If no 'else' is specified, then the
|
|
/// ElseLoc location is not specified and the Else statement is null. After
|
|
/// type-checking, the condition is of type Builtin.Int1.
|
|
class IfStmt : public Stmt {
|
|
SourceLoc IfLoc;
|
|
SourceLoc ElseLoc;
|
|
Expr *Cond;
|
|
Stmt *Then;
|
|
Stmt *Else;
|
|
|
|
public:
|
|
IfStmt(SourceLoc IfLoc, Expr *Cond, Stmt *Then, SourceLoc ElseLoc,
|
|
Stmt *Else)
|
|
: Stmt(StmtKind::If),
|
|
IfLoc(IfLoc), ElseLoc(ElseLoc), Cond(Cond), Then(Then), Else(Else) {}
|
|
|
|
SourceLoc getIfLoc() const { return IfLoc; }
|
|
SourceLoc getElseLoc() const { return ElseLoc; }
|
|
|
|
SourceRange getSourceRange() const;
|
|
|
|
Expr *getCond() const { return Cond; }
|
|
void setCond(Expr *e) { Cond = e; }
|
|
|
|
Stmt *getThenStmt() const { return Then; }
|
|
void setThenStmt(Stmt *s) { Then = s; }
|
|
|
|
Stmt *getElseStmt() const { return Else; }
|
|
void setElseStmt(Stmt *s) { Else = s; }
|
|
|
|
// Implement isa/cast/dyncast/etc.
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::If; }
|
|
};
|
|
|
|
/// WhileStmt - while statement. After type-checking, the condition is of
|
|
/// type Builtin.Int1.
|
|
class WhileStmt : public Stmt {
|
|
SourceLoc WhileLoc;
|
|
Expr *Cond;
|
|
Stmt *Body;
|
|
|
|
public:
|
|
WhileStmt(SourceLoc WhileLoc, Expr *Cond, Stmt *Body)
|
|
: Stmt(StmtKind::While),
|
|
WhileLoc(WhileLoc), Cond(Cond), Body(Body) {}
|
|
|
|
SourceRange getSourceRange() const;
|
|
|
|
Expr *getCond() const { return Cond; }
|
|
void setCond(Expr *e) { Cond = e; }
|
|
|
|
Stmt *getBody() const { return Body; }
|
|
void setBody(Stmt *s) { Body = s; }
|
|
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::While; }
|
|
};
|
|
|
|
/// DoWhileStmt - do/while statement. After type-checking, the condition is of
|
|
/// type Builtin.Int1.
|
|
class DoWhileStmt : public Stmt {
|
|
SourceLoc DoLoc, WhileLoc;
|
|
Stmt *Body;
|
|
Expr *Cond;
|
|
|
|
public:
|
|
DoWhileStmt(SourceLoc DoLoc, Expr *Cond, SourceLoc WhileLoc, Stmt *Body)
|
|
: Stmt(StmtKind::DoWhile),
|
|
DoLoc(DoLoc), WhileLoc(WhileLoc), Body(Body), Cond(Cond) {}
|
|
|
|
SourceRange getSourceRange() const;
|
|
|
|
Stmt *getBody() const { return Body; }
|
|
void setBody(Stmt *s) { Body = s; }
|
|
|
|
Expr *getCond() const { return Cond; }
|
|
void setCond(Expr *e) { Cond = e; }
|
|
|
|
static bool classof(const Stmt *S) {return S->getKind() == StmtKind::DoWhile;}
|
|
};
|
|
|
|
/// ForStmt - for statement. After type-checking, the condition is of
|
|
/// type Builtin.Int1. Note that the condition is optional. If not present,
|
|
/// it always evaluates to true. The Initializer and Increment are also
|
|
/// optional.
|
|
class ForStmt : public Stmt {
|
|
SourceLoc ForLoc, Semi1Loc, Semi2Loc;
|
|
PointerUnion<Expr*, AssignStmt*> Initializer;
|
|
ArrayRef<Decl*> InitializerVarDecls;
|
|
NullablePtr<Expr> Cond;
|
|
PointerUnion<Expr*, AssignStmt*> Increment;
|
|
Stmt *Body;
|
|
|
|
public:
|
|
ForStmt(SourceLoc ForLoc,
|
|
PointerUnion<Expr*, AssignStmt*> Initializer,
|
|
ArrayRef<Decl*> InitializerVarDecls,
|
|
SourceLoc Semi1Loc, NullablePtr<Expr> Cond, SourceLoc Semi2Loc,
|
|
PointerUnion<Expr*, AssignStmt*> Increment,
|
|
Stmt *Body)
|
|
: Stmt(StmtKind::For), ForLoc(ForLoc), Semi1Loc(Semi1Loc),
|
|
Semi2Loc(Semi2Loc), Initializer(Initializer),
|
|
InitializerVarDecls(InitializerVarDecls),
|
|
Cond(Cond), Increment(Increment), Body(Body) {
|
|
}
|
|
|
|
SourceRange getSourceRange() const {
|
|
return SourceRange(ForLoc, Body->getEndLoc());
|
|
}
|
|
|
|
PointerUnion<Expr*, AssignStmt*> getInitializer() const { return Initializer;}
|
|
void setInitializer(PointerUnion<Expr*, AssignStmt*> V) { Initializer = V; }
|
|
|
|
ArrayRef<Decl*> getInitializerVarDecls() const { return InitializerVarDecls; }
|
|
void setInitializerVarDecls(ArrayRef<Decl*> D) { InitializerVarDecls = D; }
|
|
|
|
NullablePtr<Expr> getCond() const { return Cond; }
|
|
void setCond(NullablePtr<Expr> C) { Cond = C; }
|
|
PointerUnion<Expr*, AssignStmt*> getIncrement() const { return Increment; }
|
|
void setIncrement(PointerUnion<Expr*, AssignStmt*> V) { Increment = V; }
|
|
Stmt *getBody() const { return Body; }
|
|
void setBody(Stmt *s) { Body = s; }
|
|
|
|
static bool classof(const Stmt *S) { return S->getKind() == StmtKind::For; }
|
|
};
|
|
|
|
/// ForEachStmt - foreach statement that iterates over the elements in a
|
|
/// container.
|
|
///
|
|
/// Example:
|
|
/// \code
|
|
/// for i in 0..10 {
|
|
/// println(String(i))
|
|
/// }
|
|
/// \endcode
|
|
class ForEachStmt : public Stmt {
|
|
SourceLoc ForLoc;
|
|
SourceLoc InLoc;
|
|
llvm::PointerUnion<Pattern *, PatternBindingDecl *> Pat;
|
|
Expr *Container;
|
|
BraceStmt *Body;
|
|
|
|
/// Range - The range variable along with its initializer.
|
|
PatternBindingDecl *Range;
|
|
|
|
/// RangeEmpty - The expression that determines whether the range is empty.
|
|
Expr *RangeEmpty;
|
|
|
|
public:
|
|
ForEachStmt(SourceLoc ForLoc, Pattern *Pat, SourceLoc InLoc,
|
|
Expr *Container, BraceStmt *Body)
|
|
: Stmt(StmtKind::ForEach), ForLoc(ForLoc), InLoc(InLoc), Pat(Pat),
|
|
Container(Container), Body(Body), Range(), RangeEmpty() { }
|
|
|
|
/// getForLoc - Retrieve the location of the 'for' keyword.
|
|
SourceLoc getForLoc() const { return ForLoc; }
|
|
|
|
/// getInLoc - Retrieve the location of the 'in' keyword.
|
|
SourceLoc getInLoc() const { return InLoc; }
|
|
|
|
/// getPattern - Retrieve the pattern describing the iteration variables.
|
|
/// These variables will only be visible within the body of the loop.
|
|
Pattern *getPattern() const;
|
|
|
|
/// getContainer - Retrieve the container whose elements will be visited
|
|
/// by this foreach loop, as it was written in the source code and
|
|
/// subsequently type-checked. To determine the semantic behavior of this
|
|
/// expression to extract a range, use \c getRangeInit().
|
|
Expr *getContainer() const { return Container; }
|
|
void setContainer(Expr *C) { Container = C; }
|
|
|
|
/// getRange - Retrieve the pattern binding that contains the (implicit)
|
|
/// range variable and its initialization from the container.
|
|
PatternBindingDecl *getRange() const { return Range; }
|
|
void setRange(PatternBindingDecl *R) { Range = R; }
|
|
|
|
/// getRangeEmpty - Retrieve the expression that determines whether the
|
|
/// given range is empty.
|
|
Expr *getRangeEmpty() const { return RangeEmpty; }
|
|
void setRangeEmpty(Expr *E) { RangeEmpty = E; }
|
|
|
|
/// getElementInit - Retrieve the pattern binding that binds the pattern
|
|
/// (with the iteration variables) to the initialization of that pattern
|
|
/// from the result of getFirst().
|
|
PatternBindingDecl *getElementInit() const {
|
|
return Pat.dyn_cast<PatternBindingDecl *>();
|
|
}
|
|
void setElementInit(PatternBindingDecl *EI) { Pat = EI; }
|
|
|
|
/// getBody - Retrieve the body of the loop.
|
|
BraceStmt *getBody() const { return Body; }
|
|
void setBody(BraceStmt *B) { Body = B; }
|
|
|
|
SourceRange getSourceRange() const {
|
|
return SourceRange(ForLoc, Body->getEndLoc());
|
|
}
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::ForEach;
|
|
}
|
|
};
|
|
|
|
/// A 'case' or 'default' block of a switch statement. Only valid as the
|
|
/// substatement of a SwitchStmt. Type-checking synthesizes a condition expr
|
|
/// using the '=~' operator to compare the subject of the switch to the
|
|
/// value(s) of the case.
|
|
class CaseStmt : public Stmt {
|
|
SourceLoc CaseOrDefaultLoc;
|
|
SourceLoc ColonLoc;
|
|
Stmt *Body;
|
|
Expr *ConditionExpr;
|
|
unsigned ValueExprCount;
|
|
|
|
Expr * const *getValueExprBuffer() const {
|
|
return reinterpret_cast<Expr * const *>(this + 1);
|
|
}
|
|
Expr **getValueExprBuffer() {
|
|
return reinterpret_cast<Expr**>(this + 1);
|
|
}
|
|
|
|
CaseStmt(SourceLoc CaseOrDefaultLoc,
|
|
unsigned ValueExprCount,
|
|
SourceLoc ColonLoc,
|
|
BraceStmt *Body)
|
|
: Stmt(StmtKind::Case),
|
|
CaseOrDefaultLoc(CaseOrDefaultLoc),
|
|
ColonLoc(ColonLoc),
|
|
Body(Body),
|
|
ConditionExpr(nullptr),
|
|
ValueExprCount(ValueExprCount)
|
|
{}
|
|
|
|
public:
|
|
/// Allocate a new CaseStmt in the given ASTContext.
|
|
static CaseStmt *create(SourceLoc CaseOrDefaultLoc,
|
|
ArrayRef<Expr*> ValueExprs,
|
|
SourceLoc ColonLoc,
|
|
BraceStmt *Body,
|
|
ASTContext &C);
|
|
|
|
/// True if this is the 'default' block of a switch.
|
|
bool isDefault() const { return ValueExprCount == 0; }
|
|
|
|
/// Get the source location of the 'case' or 'default' keyword.
|
|
SourceLoc getCaseOrDefaultLoc() const { return CaseOrDefaultLoc; }
|
|
/// Get the source location of the label colon.
|
|
SourceLoc getColonLoc() const { return ColonLoc; }
|
|
|
|
SourceLoc getLoc() const { return CaseOrDefaultLoc; }
|
|
SourceRange getSourceRange() const {
|
|
return {CaseOrDefaultLoc, Body->getEndLoc()};
|
|
}
|
|
|
|
/// Get the list of value expressions for this case. Empty for a 'default'
|
|
/// block.
|
|
ArrayRef<Expr*> getValueExprs() const {
|
|
return {getValueExprBuffer(), ValueExprCount};
|
|
}
|
|
MutableArrayRef<Expr*> getMutableValueExprs() {
|
|
return {getValueExprBuffer(), ValueExprCount};
|
|
}
|
|
|
|
/// Get the body of the case or default block.
|
|
Stmt *getBody() const { return Body; }
|
|
void setBody(Stmt *S) { Body = S; }
|
|
|
|
/// Get the condition expr for this case. This expr is synthesized during
|
|
/// type-checking and returns an Int1 indicating whether this case branch
|
|
/// is taken. This returns null if the case has not been type-checked or if
|
|
/// this is the default case.
|
|
Expr *getConditionExpr() const { return ConditionExpr; }
|
|
void setConditionExpr(Expr *e) { ConditionExpr = e; }
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::Case;
|
|
}
|
|
};
|
|
|
|
/// Switch statement.
|
|
class SwitchStmt : public Stmt {
|
|
SourceLoc SwitchLoc, LBraceLoc, RBraceLoc;
|
|
Expr *SubjectExpr;
|
|
VarDecl *SubjectDecl;
|
|
unsigned CaseCount;
|
|
|
|
CaseStmt * const *getCaseBuffer() const {
|
|
return reinterpret_cast<CaseStmt * const *>(this + 1);
|
|
}
|
|
|
|
CaseStmt **getCaseBuffer() {
|
|
return reinterpret_cast<CaseStmt **>(this + 1);
|
|
}
|
|
|
|
SwitchStmt(SourceLoc SwitchLoc,
|
|
Expr *SubjectExpr,
|
|
VarDecl *SubjectDecl,
|
|
SourceLoc LBraceLoc,
|
|
unsigned CaseCount,
|
|
SourceLoc RBraceLoc)
|
|
: Stmt(StmtKind::Switch),
|
|
SwitchLoc(SwitchLoc), LBraceLoc(LBraceLoc), RBraceLoc(RBraceLoc),
|
|
SubjectExpr(SubjectExpr), SubjectDecl(SubjectDecl), CaseCount(CaseCount)
|
|
{}
|
|
|
|
public:
|
|
/// Allocate a new SwitchStmt in the given ASTContext.
|
|
static SwitchStmt *create(SourceLoc SwitchLoc,
|
|
Expr *SubjectExpr, VarDecl *SubjectDecl,
|
|
SourceLoc LBraceLoc,
|
|
ArrayRef<CaseStmt*> Cases,
|
|
SourceLoc RBraceLoc,
|
|
ASTContext &C);
|
|
|
|
/// Get the source location of the 'switch' keyword.
|
|
SourceLoc getSwitchLoc() const { return SwitchLoc; }
|
|
/// Get the source location of the opening brace.
|
|
SourceLoc getLBraceLoc() const { return LBraceLoc; }
|
|
/// Get the source location of the closing brace.
|
|
SourceLoc getRBraceLoc() const { return RBraceLoc; }
|
|
|
|
SourceLoc getLoc() const { return SwitchLoc; }
|
|
SourceRange getSourceRange() const { return {SwitchLoc, RBraceLoc}; }
|
|
|
|
/// Get the subject expression of the switch.
|
|
Expr *getSubjectExpr() const { return SubjectExpr; }
|
|
void setSubjectExpr(Expr *e) { SubjectExpr = e; }
|
|
|
|
/// Get the VarDecl bound to the result of the subject expression. This
|
|
/// decl cannot be referenced by name lookup; it is only used to synthesize
|
|
/// the condition exprs of CaseStmts.
|
|
VarDecl *getSubjectDecl() const { return SubjectDecl; }
|
|
void setSubjectDecl(VarDecl *d) { SubjectDecl = d; }
|
|
|
|
/// Get the list of case clauses.
|
|
ArrayRef<CaseStmt*> getCases() const {
|
|
return {getCaseBuffer(), CaseCount};
|
|
}
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::Switch;
|
|
}
|
|
};
|
|
|
|
/// BreakStmt - The keyword "break".
|
|
class BreakStmt : public Stmt {
|
|
SourceLoc Loc;
|
|
|
|
public:
|
|
BreakStmt(SourceLoc Loc) : Stmt(StmtKind::Break), Loc(Loc) {}
|
|
|
|
SourceLoc getLoc() const { return Loc; }
|
|
|
|
SourceRange getSourceRange() const { return Loc; }
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::Break;
|
|
}
|
|
};
|
|
|
|
/// ContinueStmt - The keyword "continue".
|
|
class ContinueStmt : public Stmt {
|
|
SourceLoc Loc;
|
|
|
|
public:
|
|
ContinueStmt(SourceLoc Loc) : Stmt(StmtKind::Continue), Loc(Loc) {}
|
|
|
|
SourceLoc getLoc() const { return Loc; }
|
|
|
|
SourceRange getSourceRange() const { return Loc; }
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::Continue;
|
|
}
|
|
};
|
|
|
|
/// FallthroughStmt - The keyword "fallthrough".
|
|
class FallthroughStmt : public Stmt {
|
|
SourceLoc Loc;
|
|
CaseStmt *FallthroughDest;
|
|
|
|
public:
|
|
FallthroughStmt(SourceLoc Loc)
|
|
: Stmt(StmtKind::Fallthrough), Loc(Loc), FallthroughDest(nullptr)
|
|
{}
|
|
|
|
SourceLoc getLoc() const { return Loc; }
|
|
|
|
SourceRange getSourceRange() const { return Loc; }
|
|
|
|
/// Get the CaseStmt block to which the fallthrough transfers control.
|
|
/// Set during Sema.
|
|
CaseStmt *getFallthroughDest() const {
|
|
assert(FallthroughDest && "fallthrough dest is not set until Sema");
|
|
return FallthroughDest;
|
|
}
|
|
void setFallthroughDest(CaseStmt *C) {
|
|
assert(!FallthroughDest && "fallthrough dest already set?!");
|
|
FallthroughDest = C;
|
|
}
|
|
|
|
static bool classof(const Stmt *S) {
|
|
return S->getKind() == StmtKind::Fallthrough;
|
|
}
|
|
};
|
|
|
|
} // end namespace swift
|
|
|
|
#endif
|