1 //===--- StmtCXX.cpp - Classes for representing C++ 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 StmtCXX.h 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "clang/AST/StmtCXX.h" 14 15 #include "clang/AST/ASTContext.h" 16 17 using namespace clang; 18 19 QualType CXXCatchStmt::getCaughtType() const { 20 if (ExceptionDecl) 21 return ExceptionDecl->getType(); 22 return QualType(); 23 } 24 25 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, SourceLocation tryLoc, 26 Stmt *tryBlock, ArrayRef<Stmt *> handlers) { 27 const size_t Size = totalSizeToAlloc<Stmt *>(handlers.size() + 1); 28 void *Mem = C.Allocate(Size, alignof(CXXTryStmt)); 29 return new (Mem) CXXTryStmt(tryLoc, tryBlock, handlers); 30 } 31 32 CXXTryStmt *CXXTryStmt::Create(const ASTContext &C, EmptyShell Empty, 33 unsigned numHandlers) { 34 const size_t Size = totalSizeToAlloc<Stmt *>(numHandlers + 1); 35 void *Mem = C.Allocate(Size, alignof(CXXTryStmt)); 36 return new (Mem) CXXTryStmt(Empty, numHandlers); 37 } 38 39 CXXTryStmt::CXXTryStmt(SourceLocation tryLoc, Stmt *tryBlock, 40 ArrayRef<Stmt *> handlers) 41 : Stmt(CXXTryStmtClass), TryLoc(tryLoc), NumHandlers(handlers.size()) { 42 Stmt **Stmts = getStmts(); 43 Stmts[0] = tryBlock; 44 std::copy(handlers.begin(), handlers.end(), Stmts + 1); 45 } 46 47 CXXForRangeStmt::CXXForRangeStmt(Stmt *Init, DeclStmt *Range, 48 DeclStmt *BeginStmt, DeclStmt *EndStmt, 49 Expr *Cond, Expr *Inc, DeclStmt *LoopVar, 50 Stmt *Body, SourceLocation FL, 51 SourceLocation CAL, SourceLocation CL, 52 SourceLocation RPL) 53 : Stmt(CXXForRangeStmtClass), ForLoc(FL), CoawaitLoc(CAL), ColonLoc(CL), 54 RParenLoc(RPL) { 55 SubExprs[INIT] = Init; 56 SubExprs[RANGE] = Range; 57 SubExprs[BEGINSTMT] = BeginStmt; 58 SubExprs[ENDSTMT] = EndStmt; 59 SubExprs[COND] = Cond; 60 SubExprs[INC] = Inc; 61 SubExprs[LOOPVAR] = LoopVar; 62 SubExprs[BODY] = Body; 63 } 64 65 Expr *CXXForRangeStmt::getRangeInit() { 66 DeclStmt *RangeStmt = getRangeStmt(); 67 VarDecl *RangeDecl = dyn_cast_or_null<VarDecl>(RangeStmt->getSingleDecl()); 68 assert(RangeDecl && "for-range should have a single var decl"); 69 return RangeDecl->getInit(); 70 } 71 72 const Expr *CXXForRangeStmt::getRangeInit() const { 73 return const_cast<CXXForRangeStmt *>(this)->getRangeInit(); 74 } 75 76 VarDecl *CXXForRangeStmt::getLoopVariable() { 77 Decl *LV = cast<DeclStmt>(getLoopVarStmt())->getSingleDecl(); 78 assert(LV && "No loop variable in CXXForRangeStmt"); 79 return cast<VarDecl>(LV); 80 } 81 82 const VarDecl *CXXForRangeStmt::getLoopVariable() const { 83 return const_cast<CXXForRangeStmt *>(this)->getLoopVariable(); 84 } 85 86 CoroutineBodyStmt *CoroutineBodyStmt::Create( 87 const ASTContext &C, CoroutineBodyStmt::CtorArgs const &Args) { 88 std::size_t Size = totalSizeToAlloc<Stmt *>( 89 CoroutineBodyStmt::FirstParamMove + Args.ParamMoves.size()); 90 91 void *Mem = C.Allocate(Size, alignof(CoroutineBodyStmt)); 92 return new (Mem) CoroutineBodyStmt(Args); 93 } 94 95 CoroutineBodyStmt *CoroutineBodyStmt::Create(const ASTContext &C, EmptyShell, 96 unsigned NumParams) { 97 std::size_t Size = totalSizeToAlloc<Stmt *>( 98 CoroutineBodyStmt::FirstParamMove + NumParams); 99 100 void *Mem = C.Allocate(Size, alignof(CoroutineBodyStmt)); 101 auto *Result = new (Mem) CoroutineBodyStmt(CtorArgs()); 102 Result->NumParams = NumParams; 103 auto *ParamBegin = Result->getStoredStmts() + SubStmt::FirstParamMove; 104 std::uninitialized_fill(ParamBegin, ParamBegin + NumParams, 105 static_cast<Stmt *>(nullptr)); 106 return Result; 107 } 108 109 CoroutineBodyStmt::CoroutineBodyStmt(CoroutineBodyStmt::CtorArgs const &Args) 110 : Stmt(CoroutineBodyStmtClass), NumParams(Args.ParamMoves.size()) { 111 Stmt **SubStmts = getStoredStmts(); 112 SubStmts[CoroutineBodyStmt::Body] = Args.Body; 113 SubStmts[CoroutineBodyStmt::Promise] = Args.Promise; 114 SubStmts[CoroutineBodyStmt::InitSuspend] = Args.InitialSuspend; 115 SubStmts[CoroutineBodyStmt::FinalSuspend] = Args.FinalSuspend; 116 SubStmts[CoroutineBodyStmt::OnException] = Args.OnException; 117 SubStmts[CoroutineBodyStmt::OnFallthrough] = Args.OnFallthrough; 118 SubStmts[CoroutineBodyStmt::Allocate] = Args.Allocate; 119 SubStmts[CoroutineBodyStmt::Deallocate] = Args.Deallocate; 120 SubStmts[CoroutineBodyStmt::ReturnValue] = Args.ReturnValue; 121 SubStmts[CoroutineBodyStmt::ResultDecl] = Args.ResultDecl; 122 SubStmts[CoroutineBodyStmt::ReturnStmt] = Args.ReturnStmt; 123 SubStmts[CoroutineBodyStmt::ReturnStmtOnAllocFailure] = 124 Args.ReturnStmtOnAllocFailure; 125 std::copy(Args.ParamMoves.begin(), Args.ParamMoves.end(), 126 const_cast<Stmt **>(getParamMoves().data())); 127 } 128