clang/lib/AST/Interp/ByteCodeStmtGen.h

   1 //===--- ByteCodeStmtGen.h - Code generator for expressions -----*- C++ -*-===//
   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 // Defines the constexpr bytecode compiler.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_CLANG_AST_INTERP_BYTECODESTMTGEN_H
  14 #define LLVM_CLANG_AST_INTERP_BYTECODESTMTGEN_H
  15
  16 #include "ByteCodeEmitter.h"
  17 #include "ByteCodeExprGen.h"
  18 #include "EvalEmitter.h"
  19 #include "PrimType.h"
  20 #include "clang/AST/StmtVisitor.h"
  21
  22 namespace clang {
  23 namespace interp {
  24
  25 template <class Emitter> class LoopScope;
  26 template <class Emitter> class SwitchScope;
  27 template <class Emitter> class LabelScope;
  28
  29 /// Compilation context for statements.
  30 template <class Emitter>
  31 class ByteCodeStmtGen final : public ByteCodeExprGen<Emitter> {
  32   using LabelTy = typename Emitter::LabelTy;
  33   using AddrTy = typename Emitter::AddrTy;
  34   using OptLabelTy = std::optional<LabelTy>;
  35   using CaseMap = llvm::DenseMap<const SwitchCase *, LabelTy>;
  36
  37 public:
  38   template<typename... Tys>
  39   ByteCodeStmtGen(Tys&&... Args)
  40       : ByteCodeExprGen<Emitter>(std::forward<Tys>(Args)...) {}
  41
  42 protected:
  43   bool visitFunc(const FunctionDecl *F) override;
  44
  45 private:
  46   friend class LabelScope<Emitter>;
  47   friend class LoopScope<Emitter>;
  48   friend class SwitchScope<Emitter>;
  49
  50   // Statement visitors.
  51   bool visitStmt(const Stmt *S);
  52   bool visitCompoundStmt(const CompoundStmt *S);
  53   bool visitLoopBody(const Stmt *S);
  54   bool visitDeclStmt(const DeclStmt *DS);
  55   bool visitReturnStmt(const ReturnStmt *RS);
  56   bool visitIfStmt(const IfStmt *IS);
  57   bool visitWhileStmt(const WhileStmt *S);
  58   bool visitDoStmt(const DoStmt *S);
  59   bool visitForStmt(const ForStmt *S);
  60   bool visitCXXForRangeStmt(const CXXForRangeStmt *S);
  61   bool visitBreakStmt(const BreakStmt *S);
  62   bool visitContinueStmt(const ContinueStmt *S);
  63   bool visitSwitchStmt(const SwitchStmt *S);
  64   bool visitCaseStmt(const CaseStmt *S);
  65   bool visitDefaultStmt(const DefaultStmt *S);
  66   bool visitAsmStmt(const AsmStmt *S);
  67   bool visitAttributedStmt(const AttributedStmt *S);
  68   bool visitCXXTryStmt(const CXXTryStmt *S);
  69
  70   bool emitLambdaStaticInvokerBody(const CXXMethodDecl *MD);
  71
  72   /// Type of the expression returned by the function.
  73   std::optional<PrimType> ReturnType;
  74
  75   /// Switch case mapping.
  76   CaseMap CaseLabels;
  77
  78   /// Point to break to.
  79   OptLabelTy BreakLabel;
  80   /// Point to continue to.
  81   OptLabelTy ContinueLabel;
  82   /// Default case label.
  83   OptLabelTy DefaultLabel;
  84 };
  85
  86 extern template class ByteCodeExprGen<EvalEmitter>;
  87
  88 } // namespace interp
  89 } // namespace clang
  90
  91 #endif