1 //===-- DebugLoc.cpp - Implement DebugLoc class ---------------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/Support/DebugLoc.h"
11 #include "llvm/ADT/DenseMapInfo.h"
12 #include "LLVMContextImpl.h"
15 //===----------------------------------------------------------------------===//
16 // DebugLoc Implementation
17 //===----------------------------------------------------------------------===//
19 MDNode *DebugLoc::getScope(const LLVMContext &Ctx) const {
20 if (ScopeIdx == 0) return 0;
23 // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
24 // position specified.
25 assert(unsigned(ScopeIdx) <= Ctx.pImpl->ScopeRecords.size() &&
27 return Ctx.pImpl->ScopeRecords[ScopeIdx-1].get();
30 // Otherwise, the index is in the ScopeInlinedAtRecords array.
31 assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
33 return Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].first.get();
36 MDNode *DebugLoc::getInlinedAt(const LLVMContext &Ctx) const {
37 // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
38 // position specified. Zero is invalid.
39 if (ScopeIdx >= 0) return 0;
41 // Otherwise, the index is in the ScopeInlinedAtRecords array.
42 assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
44 return Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].second.get();
47 /// Return both the Scope and the InlinedAt values.
48 void DebugLoc::getScopeAndInlinedAt(MDNode *&Scope, MDNode *&IA,
49 const LLVMContext &Ctx) const {
56 // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
57 // position specified.
58 assert(unsigned(ScopeIdx) <= Ctx.pImpl->ScopeRecords.size() &&
60 Scope = Ctx.pImpl->ScopeRecords[ScopeIdx-1].get();
65 // Otherwise, the index is in the ScopeInlinedAtRecords array.
66 assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
68 Scope = Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].first.get();
69 IA = Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].second.get();
73 DebugLoc DebugLoc::get(unsigned Line, unsigned Col,
74 MDNode *Scope, MDNode *InlinedAt) {
77 // If no scope is available, this is an unknown location.
78 if (Scope == 0) return Result;
80 // Saturate line and col to "unknown".
81 if (Col > 255) Col = 0;
82 if (Line >= (1 << 24)) Line = 0;
83 Result.LineCol = Line | (Col << 24);
85 LLVMContext &Ctx = Scope->getContext();
87 // If there is no inlined-at location, use the ScopeRecords array.
89 Result.ScopeIdx = Ctx.pImpl->getOrAddScopeRecordIdxEntry(Scope, 0);
91 Result.ScopeIdx = Ctx.pImpl->getOrAddScopeInlinedAtIdxEntry(Scope,
97 /// getAsMDNode - This method converts the compressed DebugLoc node into a
98 /// DILocation compatible MDNode.
99 MDNode *DebugLoc::getAsMDNode(const LLVMContext &Ctx) const {
100 if (isUnknown()) return 0;
103 getScopeAndInlinedAt(Scope, IA, Ctx);
104 assert(Scope && "If scope is null, this should be isUnknown()");
106 LLVMContext &Ctx2 = Scope->getContext();
107 Type *Int32 = Type::getInt32Ty(Ctx2);
109 ConstantInt::get(Int32, getLine()), ConstantInt::get(Int32, getCol()),
112 return MDNode::get(Ctx2, Elts);
115 /// getFromDILocation - Translate the DILocation quad into a DebugLoc.
116 DebugLoc DebugLoc::getFromDILocation(MDNode *N) {
117 if (N == 0 || N->getNumOperands() != 4) return DebugLoc();
119 MDNode *Scope = dyn_cast_or_null<MDNode>(N->getOperand(2));
120 if (Scope == 0) return DebugLoc();
122 unsigned LineNo = 0, ColNo = 0;
123 if (ConstantInt *Line = dyn_cast_or_null<ConstantInt>(N->getOperand(0)))
124 LineNo = Line->getZExtValue();
125 if (ConstantInt *Col = dyn_cast_or_null<ConstantInt>(N->getOperand(1)))
126 ColNo = Col->getZExtValue();
128 return get(LineNo, ColNo, Scope, dyn_cast_or_null<MDNode>(N->getOperand(3)));
131 /// getFromDILexicalBlock - Translate the DILexicalBlock into a DebugLoc.
132 DebugLoc DebugLoc::getFromDILexicalBlock(MDNode *N) {
133 if (N == 0 || N->getNumOperands() < 3) return DebugLoc();
135 MDNode *Scope = dyn_cast_or_null<MDNode>(N->getOperand(1));
136 if (Scope == 0) return DebugLoc();
138 unsigned LineNo = 0, ColNo = 0;
139 if (ConstantInt *Line = dyn_cast_or_null<ConstantInt>(N->getOperand(2)))
140 LineNo = Line->getZExtValue();
141 if (ConstantInt *Col = dyn_cast_or_null<ConstantInt>(N->getOperand(3)))
142 ColNo = Col->getZExtValue();
144 return get(LineNo, ColNo, Scope, NULL);
147 void DebugLoc::dump(const LLVMContext &Ctx) const {
152 dbgs() << ',' << getCol();
153 DebugLoc InlinedAtDL = DebugLoc::getFromDILocation(getInlinedAt(Ctx));
154 if (!InlinedAtDL.isUnknown()) {
156 InlinedAtDL.dump(Ctx);
163 //===----------------------------------------------------------------------===//
164 // DenseMap specialization
165 //===----------------------------------------------------------------------===//
167 DebugLoc DenseMapInfo<DebugLoc>::getEmptyKey() {
168 return DebugLoc::getEmptyKey();
171 DebugLoc DenseMapInfo<DebugLoc>::getTombstoneKey() {
172 return DebugLoc::getTombstoneKey();
175 unsigned DenseMapInfo<DebugLoc>::getHashValue(const DebugLoc &Key) {
177 ID.AddInteger(Key.LineCol);
178 ID.AddInteger(Key.ScopeIdx);
179 return ID.ComputeHash();
182 bool DenseMapInfo<DebugLoc>::isEqual(const DebugLoc &LHS, const DebugLoc &RHS) {
186 //===----------------------------------------------------------------------===//
187 // LLVMContextImpl Implementation
188 //===----------------------------------------------------------------------===//
190 int LLVMContextImpl::getOrAddScopeRecordIdxEntry(MDNode *Scope,
192 // If we already have an entry for this scope, return it.
193 int &Idx = ScopeRecordIdx[Scope];
196 // If we don't have an entry, but ExistingIdx is specified, use it.
198 return Idx = ExistingIdx;
200 // Otherwise add a new entry.
202 // Start out ScopeRecords with a minimal reasonable size to avoid
203 // excessive reallocation starting out.
204 if (ScopeRecords.empty())
205 ScopeRecords.reserve(128);
207 // Index is biased by 1 for index.
208 Idx = ScopeRecords.size()+1;
209 ScopeRecords.push_back(DebugRecVH(Scope, this, Idx));
213 int LLVMContextImpl::getOrAddScopeInlinedAtIdxEntry(MDNode *Scope, MDNode *IA,
215 // If we already have an entry, return it.
216 int &Idx = ScopeInlinedAtIdx[std::make_pair(Scope, IA)];
219 // If we don't have an entry, but ExistingIdx is specified, use it.
221 return Idx = ExistingIdx;
223 // Start out ScopeInlinedAtRecords with a minimal reasonable size to avoid
224 // excessive reallocation starting out.
225 if (ScopeInlinedAtRecords.empty())
226 ScopeInlinedAtRecords.reserve(128);
228 // Index is biased by 1 and negated.
229 Idx = -ScopeInlinedAtRecords.size()-1;
230 ScopeInlinedAtRecords.push_back(std::make_pair(DebugRecVH(Scope, this, Idx),
231 DebugRecVH(IA, this, Idx)));
236 //===----------------------------------------------------------------------===//
237 // DebugRecVH Implementation
238 //===----------------------------------------------------------------------===//
240 /// deleted - The MDNode this is pointing to got deleted, so this pointer needs
241 /// to drop to null and we need remove our entry from the DenseMap.
242 void DebugRecVH::deleted() {
243 // If this is a non-canonical reference, just drop the value to null, we know
244 // it doesn't have a map entry.
252 // If the index is positive, it is an entry in ScopeRecords.
254 assert(Ctx->ScopeRecordIdx[Cur] == Idx && "Mapping out of date!");
255 Ctx->ScopeRecordIdx.erase(Cur);
256 // Reset this VH to null and we're done.
262 // Otherwise, it is an entry in ScopeInlinedAtRecords, we don't know if it
263 // is the scope or the inlined-at record entry.
264 assert(unsigned(-Idx-1) < Ctx->ScopeInlinedAtRecords.size());
265 std::pair<DebugRecVH, DebugRecVH> &Entry = Ctx->ScopeInlinedAtRecords[-Idx-1];
266 assert((this == &Entry.first || this == &Entry.second) &&
267 "Mapping out of date!");
269 MDNode *OldScope = Entry.first.get();
270 MDNode *OldInlinedAt = Entry.second.get();
271 assert(OldScope != 0 && OldInlinedAt != 0 &&
272 "Entry should be non-canonical if either val dropped to null");
274 // Otherwise, we do have an entry in it, nuke it and we're done.
275 assert(Ctx->ScopeInlinedAtIdx[std::make_pair(OldScope, OldInlinedAt)] == Idx&&
276 "Mapping out of date");
277 Ctx->ScopeInlinedAtIdx.erase(std::make_pair(OldScope, OldInlinedAt));
279 // Reset this VH to null. Drop both 'Idx' values to null to indicate that
280 // we're in non-canonical form now.
282 Entry.first.Idx = Entry.second.Idx = 0;
285 void DebugRecVH::allUsesReplacedWith(Value *NewVa) {
286 // If being replaced with a non-mdnode value (e.g. undef) handle this as if
287 // the mdnode got deleted.
288 MDNode *NewVal = dyn_cast<MDNode>(NewVa);
289 if (NewVal == 0) return deleted();
291 // If this is a non-canonical reference, just change it, we know it already
292 // doesn't have a map entry.
298 MDNode *OldVal = get();
299 assert(OldVal != NewVa && "Node replaced with self?");
301 // If the index is positive, it is an entry in ScopeRecords.
303 assert(Ctx->ScopeRecordIdx[OldVal] == Idx && "Mapping out of date!");
304 Ctx->ScopeRecordIdx.erase(OldVal);
307 int NewEntry = Ctx->getOrAddScopeRecordIdxEntry(NewVal, Idx);
309 // If NewVal already has an entry, this becomes a non-canonical reference,
310 // just drop Idx to 0 to signify this.
316 // Otherwise, it is an entry in ScopeInlinedAtRecords, we don't know if it
317 // is the scope or the inlined-at record entry.
318 assert(unsigned(-Idx-1) < Ctx->ScopeInlinedAtRecords.size());
319 std::pair<DebugRecVH, DebugRecVH> &Entry = Ctx->ScopeInlinedAtRecords[-Idx-1];
320 assert((this == &Entry.first || this == &Entry.second) &&
321 "Mapping out of date!");
323 MDNode *OldScope = Entry.first.get();
324 MDNode *OldInlinedAt = Entry.second.get();
325 assert(OldScope != 0 && OldInlinedAt != 0 &&
326 "Entry should be non-canonical if either val dropped to null");
328 // Otherwise, we do have an entry in it, nuke it and we're done.
329 assert(Ctx->ScopeInlinedAtIdx[std::make_pair(OldScope, OldInlinedAt)] == Idx&&
330 "Mapping out of date");
331 Ctx->ScopeInlinedAtIdx.erase(std::make_pair(OldScope, OldInlinedAt));
333 // Reset this VH to the new value.
336 int NewIdx = Ctx->getOrAddScopeInlinedAtIdxEntry(Entry.first.get(),
337 Entry.second.get(), Idx);
338 // If NewVal already has an entry, this becomes a non-canonical reference,
339 // just drop Idx to 0 to signify this.
341 std::pair<DebugRecVH, DebugRecVH> &Entry=Ctx->ScopeInlinedAtRecords[-Idx-1];
342 Entry.first.Idx = Entry.second.Idx = 0;