1 //===--- StmtObjC.cpp - Classes for representing ObjC statements ---------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file implements the subclesses of Stmt class declared in StmtObjC.h 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