| 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 (const auto *S = Source.dyn_cast<const Stmt *>()) |
| 37 | return dyn_cast<Expr>(Val: 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(reason: "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 | } |
| 54 |