clang/lib/AST/Interp/Source.cpp

   1 //===--- Source.cpp - Source expression tracking ----------------*- 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 #include "Source.h"
  10 #include "clang/AST/Expr.h"
  11
  12 using namespace clang;
  13 using namespace clang::interp;
  14
  15 SourceLocation SourceInfo::getLoc() const {
  16   if (const Expr *E = asExpr())
  17     return E->getExprLoc();
  18   if (const Stmt *S = asStmt())
  19     return S->getBeginLoc();
  20   if (const Decl *D = asDecl())
  21     return D->getBeginLoc();
  22   return SourceLocation();
  23 }
  24
  25 SourceRange SourceInfo::getRange() const {
  26   if (const Expr *E = asExpr())
  27     return E->getSourceRange();
  28   if (const Stmt *S = asStmt())
  29     return S->getSourceRange();
  30   if (const Decl *D = asDecl())
  31     return D->getSourceRange();
  32   return SourceRange();
  33 }
  34
  35 const Expr *SourceInfo::asExpr() const {
  36   if (auto *S = Source.dyn_cast<const Stmt *>())
  37     return dyn_cast<Expr>(S);
  38   return nullptr;
  39 }
  40
  41 const Expr *SourceMapper::getExpr(const Function *F, CodePtr PC) const {
  42   if (const Expr *E = getSource(F, PC).asExpr())
  43     return E;
  44   llvm::report_fatal_error("missing source expression");
  45 }
  46
  47 SourceLocation SourceMapper::getLocation(const Function *F, CodePtr PC) const {
  48   return getSource(F, PC).getLoc();
  49 }
  50
  51 SourceRange SourceMapper::getRange(const Function *F, CodePtr PC) const {
  52   return getSource(F, PC).getRange();
  53 }