contrib/llvm/lib/IR/DebugLoc.cpp

   1 //===-- DebugLoc.cpp - Implement DebugLoc class ---------------------------===//
   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
  10 #include "llvm/IR/DebugLoc.h"
  11 #include "LLVMContextImpl.h"
  12 #include "llvm/IR/DebugInfo.h"
  13 using namespace llvm;
  14
  15 //===----------------------------------------------------------------------===//
  16 // DebugLoc Implementation
  17 //===----------------------------------------------------------------------===//
  18 DebugLoc::DebugLoc(const DILocation *L) : Loc(const_cast<DILocation *>(L)) {}
  19 DebugLoc::DebugLoc(const MDNode *L) : Loc(const_cast<MDNode *>(L)) {}
  20
  21 DILocation *DebugLoc::get() const {
  22   return cast_or_null<DILocation>(Loc.get());
  23 }
  24
  25 unsigned DebugLoc::getLine() const {
  26   assert(get() && "Expected valid DebugLoc");
  27   return get()->getLine();
  28 }
  29
  30 unsigned DebugLoc::getCol() const {
  31   assert(get() && "Expected valid DebugLoc");
  32   return get()->getColumn();
  33 }
  34
  35 MDNode *DebugLoc::getScope() const {
  36   assert(get() && "Expected valid DebugLoc");
  37   return get()->getScope();
  38 }
  39
  40 DILocation *DebugLoc::getInlinedAt() const {
  41   assert(get() && "Expected valid DebugLoc");
  42   return get()->getInlinedAt();
  43 }
  44
  45 MDNode *DebugLoc::getInlinedAtScope() const {
  46   return cast<DILocation>(Loc)->getInlinedAtScope();
  47 }
  48
  49 DebugLoc DebugLoc::getFnDebugLoc() const {
  50   // FIXME: Add a method on \a DILocation that does this work.
  51   const MDNode *Scope = getInlinedAtScope();
  52   if (auto *SP = getDISubprogram(Scope))
  53     return DebugLoc::get(SP->getScopeLine(), 0, SP);
  54
  55   return DebugLoc();
  56 }
  57
  58 DebugLoc DebugLoc::get(unsigned Line, unsigned Col, const MDNode *Scope,
  59                        const MDNode *InlinedAt) {
  60   // If no scope is available, this is an unknown location.
  61   if (!Scope)
  62     return DebugLoc();
  63
  64   return DILocation::get(Scope->getContext(), Line, Col,
  65                          const_cast<MDNode *>(Scope),
  66                          const_cast<MDNode *>(InlinedAt));
  67 }
  68
  69 DebugLoc DebugLoc::appendInlinedAt(DebugLoc DL, DILocation *InlinedAt,
  70                                    LLVMContext &Ctx,
  71                                    DenseMap<const MDNode *, MDNode *> &Cache,
  72                                    bool ReplaceLast) {
  73   SmallVector<DILocation *, 3> InlinedAtLocations;
  74   DILocation *Last = InlinedAt;
  75   DILocation *CurInlinedAt = DL;
  76
  77   // Gather all the inlined-at nodes.
  78   while (DILocation *IA = CurInlinedAt->getInlinedAt()) {
  79     // Skip any we've already built nodes for.
  80     if (auto *Found = Cache[IA]) {
  81       Last = cast<DILocation>(Found);
  82       break;
  83     }
  84
  85     if (ReplaceLast && !IA->getInlinedAt())
  86       break;
  87     InlinedAtLocations.push_back(IA);
  88     CurInlinedAt = IA;
  89   }
  90
  91   // Starting from the top, rebuild the nodes to point to the new inlined-at
  92   // location (then rebuilding the rest of the chain behind it) and update the
  93   // map of already-constructed inlined-at nodes.
  94   for (const DILocation *MD : reverse(InlinedAtLocations))
  95     Cache[MD] = Last = DILocation::getDistinct(
  96         Ctx, MD->getLine(), MD->getColumn(), MD->getScope(), Last);
  97
  98   return Last;
  99 }
 100
 101 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
 102 LLVM_DUMP_METHOD void DebugLoc::dump() const {
 103   if (!Loc)
 104     return;
 105
 106   dbgs() << getLine();
 107   if (getCol() != 0)
 108     dbgs() << ',' << getCol();
 109   if (DebugLoc InlinedAtDL = DebugLoc(getInlinedAt())) {
 110     dbgs() << " @ ";
 111     InlinedAtDL.dump();
 112   } else
 113     dbgs() << "\n";
 114 }
 115 #endif
 116
 117 void DebugLoc::print(raw_ostream &OS) const {
 118   if (!Loc)
 119     return;
 120
 121   // Print source line info.
 122   auto *Scope = cast<DIScope>(getScope());
 123   OS << Scope->getFilename();
 124   OS << ':' << getLine();
 125   if (getCol() != 0)
 126     OS << ':' << getCol();
 127
 128   if (DebugLoc InlinedAtDL = getInlinedAt()) {
 129     OS << " @[ ";
 130     InlinedAtDL.print(OS);
 131     OS << " ]";
 132   }
 133 }