lib/Sema/CoroutineStmtBuilder.h

   1 //===- CoroutineStmtBuilder.h - Implicit coroutine stmt builder -*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //===----------------------------------------------------------------------===//
   8 //
   9 //  This file defines CoroutineStmtBuilder, a class for building the implicit
  10 //  statements required for building a coroutine body.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_CLANG_LIB_SEMA_COROUTINESTMTBUILDER_H
  15 #define LLVM_CLANG_LIB_SEMA_COROUTINESTMTBUILDER_H
  16
  17 #include "clang/AST/Decl.h"
  18 #include "clang/AST/ExprCXX.h"
  19 #include "clang/AST/StmtCXX.h"
  20 #include "clang/Lex/Preprocessor.h"
  21 #include "clang/Sema/SemaInternal.h"
  22
  23 namespace clang {
  24
  25 class CoroutineStmtBuilder : public CoroutineBodyStmt::CtorArgs {
  26   Sema &S;
  27   FunctionDecl &FD;
  28   sema::FunctionScopeInfo &Fn;
  29   bool IsValid = true;
  30   SourceLocation Loc;
  31   SmallVector<Stmt *, 4> ParamMovesVector;
  32   const bool IsPromiseDependentType;
  33   CXXRecordDecl *PromiseRecordDecl = nullptr;
  34
  35 public:
  36   /// Construct a CoroutineStmtBuilder and initialize the promise
  37   /// statement and initial/final suspends from the FunctionScopeInfo.
  38   CoroutineStmtBuilder(Sema &S, FunctionDecl &FD, sema::FunctionScopeInfo &Fn,
  39                        Stmt *Body);
  40
  41   /// Build the coroutine body statements, including the
  42   /// "promise dependent" statements when the promise type is not dependent.
  43   bool buildStatements();
  44
  45   /// Build the coroutine body statements that require a non-dependent
  46   /// promise type in order to construct.
  47   ///
  48   /// For example different new/delete overloads are selected depending on
  49   /// if the promise type provides `unhandled_exception()`, and therefore they
  50   /// cannot be built until the promise type is complete so that we can perform
  51   /// name lookup.
  52   bool buildDependentStatements();
  53
  54   bool isInvalid() const { return !this->IsValid; }
  55
  56 private:
  57   bool makePromiseStmt();
  58   bool makeInitialAndFinalSuspend();
  59   bool makeNewAndDeleteExpr();
  60   bool makeOnFallthrough();
  61   bool makeOnException();
  62   bool makeReturnObject();
  63   bool makeGroDeclAndReturnStmt();
  64   bool makeReturnOnAllocFailure();
  65 };
  66
  67 } // end namespace clang
  68
  69 #endif // LLVM_CLANG_LIB_SEMA_COROUTINESTMTBUILDER_H