1//===--- Function.h - Bytecode function for the VM --------------*- 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 "Function.h"
10#include "Program.h"
11#include "Opcode.h"
12#include "clang/AST/Decl.h"
13#include "clang/AST/DeclCXX.h"
14
15using namespace clang;
16using namespace clang::interp;
17
18Function::Function(Program &P, const FunctionDecl *F, unsigned ArgSize,
19                   llvm::SmallVector<PrimType, 8> &&ParamTypes,
20                   llvm::DenseMap<unsigned, ParamDescriptor> &&Params)
21    : P(P), Loc(F->getBeginLoc()), F(F), ArgSize(ArgSize),
22      ParamTypes(std::move(ParamTypes)), Params(std::move(Params)) {}
23
24CodePtr Function::getCodeBegin() const { return Code.data(); }
25
26CodePtr Function::getCodeEnd() const { return Code.data() + Code.size(); }
27
28Function::ParamDescriptor Function::getParamDescriptor(unsigned Offset) const {
29  auto It = Params.find(Offset);
30  assert(It != Params.end() && "Invalid parameter offset");
31  return It->second;
32}
33
34SourceInfo Function::getSource(CodePtr PC) const {
35  unsigned Offset = PC - getCodeBegin();
36  using Elem = std::pair<unsigned, SourceInfo>;
37  auto It = std::lower_bound(SrcMap.begin(), SrcMap.end(), Elem{Offset, {}},
38                             [](Elem A, Elem B) { return A.first < B.first; });
39  if (It == SrcMap.end() || It->first != Offset)
40    llvm::report_fatal_error("missing source location");
41  return It->second;
42}
43
44bool Function::isVirtual() const {
45  if (auto *M = dyn_cast<CXXMethodDecl>(F))
46    return M->isVirtual();
47  return false;
48}
49