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
18using namespace clang;
19
20ObjCForCollectionStmt::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
31ObjCAtTryStmt::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
45ObjCAtTryStmt *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
57ObjCAtTryStmt *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
66SourceLocation 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