Home | History | Annotate | Line # | Download | only in AST 
      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