1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | |
9 | |
10 | |
11 | |
12 | |
13 | #include "clang/AST/StmtObjC.h" |
14 | |
15 | #include "clang/AST/Expr.h" |
16 | #include "clang/AST/ASTContext.h" |
17 | |
18 | using namespace clang; |
19 | |
20 | ObjCForCollectionStmt::ObjCForCollectionStmt(Stmt *Elem, Expr *Collect, |
21 | Stmt *Body, SourceLocation FCL, |
22 | SourceLocation RPL) |
23 | : Stmt(ObjCForCollectionStmtClass) { |
24 | SubExprs[ELEM] = Elem; |
25 | SubExprs[COLLECTION] = Collect; |
26 | SubExprs[BODY] = Body; |
27 | ForLoc = FCL; |
28 | RParenLoc = RPL; |
29 | } |
30 | |
31 | ObjCAtTryStmt::ObjCAtTryStmt(SourceLocation atTryLoc, Stmt *atTryStmt, |
32 | Stmt **CatchStmts, unsigned NumCatchStmts, |
33 | Stmt *atFinallyStmt) |
34 | : Stmt(ObjCAtTryStmtClass), AtTryLoc(atTryLoc), |
35 | NumCatchStmts(NumCatchStmts), HasFinally(atFinallyStmt != nullptr) { |
36 | Stmt **Stmts = getStmts(); |
37 | Stmts[0] = atTryStmt; |
38 | for (unsigned I = 0; I != NumCatchStmts; ++I) |
39 | Stmts[I + 1] = CatchStmts[I]; |
40 | |
41 | if (HasFinally) |
42 | Stmts[NumCatchStmts + 1] = atFinallyStmt; |
43 | } |
44 | |
45 | ObjCAtTryStmt *ObjCAtTryStmt::Create(const ASTContext &Context, |
46 | SourceLocation atTryLoc, Stmt *atTryStmt, |
47 | Stmt **CatchStmts, unsigned NumCatchStmts, |
48 | Stmt *atFinallyStmt) { |
49 | unsigned Size = |
50 | sizeof(ObjCAtTryStmt) + |
51 | (1 + NumCatchStmts + (atFinallyStmt != nullptr)) * sizeof(Stmt *); |
52 | void *Mem = Context.Allocate(Size, alignof(ObjCAtTryStmt)); |
53 | return new (Mem) ObjCAtTryStmt(atTryLoc, atTryStmt, CatchStmts, NumCatchStmts, |
54 | atFinallyStmt); |
55 | } |
56 | |
57 | ObjCAtTryStmt *ObjCAtTryStmt::CreateEmpty(const ASTContext &Context, |
58 | unsigned NumCatchStmts, |
59 | bool HasFinally) { |
60 | unsigned Size = |
61 | sizeof(ObjCAtTryStmt) + (1 + NumCatchStmts + HasFinally) * sizeof(Stmt *); |
62 | void *Mem = Context.Allocate(Size, alignof(ObjCAtTryStmt)); |
63 | return new (Mem) ObjCAtTryStmt(EmptyShell(), NumCatchStmts, HasFinally); |
64 | } |
65 | |
66 | SourceLocation ObjCAtTryStmt::getEndLoc() const { |
67 | if (HasFinally) |
68 | return getFinallyStmt()->getEndLoc(); |
69 | if (NumCatchStmts) |
70 | return getCatchStmt(NumCatchStmts - 1)->getEndLoc(); |
71 | return getTryBody()->getEndLoc(); |
72 | } |
73 | |