Reapply "DebugInfo: Loosen DILexicalBlock constructor"
[oota-llvm.git] / include / llvm / IR / DebugInfo.h
1 //===- DebugInfo.h - Debug Information Helpers ------------------*- C++ -*-===//
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 // This file defines a bunch of datatypes that are useful for creating and
11 // walking debug info in LLVM IR form. They essentially provide wrappers around
12 // the information in the global variables that's needed when constructing the
13 // DWARF information.
14 //
15 //===----------------------------------------------------------------------===//
16
17 #ifndef LLVM_IR_DEBUGINFO_H
18 #define LLVM_IR_DEBUGINFO_H
19
20 #include "llvm/ADT/DenseMap.h"
21 #include "llvm/ADT/SmallPtrSet.h"
22 #include "llvm/ADT/SmallVector.h"
23 #include "llvm/ADT/StringRef.h"
24 #include "llvm/ADT/iterator_range.h"
25 #include "llvm/IR/DebugInfoMetadata.h"
26 #include "llvm/Support/Casting.h"
27 #include "llvm/Support/Dwarf.h"
28 #include "llvm/Support/ErrorHandling.h"
29 #include <iterator>
30
31 namespace llvm {
32 class BasicBlock;
33 class Constant;
34 class Function;
35 class GlobalVariable;
36 class Module;
37 class Type;
38 class Value;
39 class DbgDeclareInst;
40 class DbgValueInst;
41 class Instruction;
42 class Metadata;
43 class MDNode;
44 class MDString;
45 class NamedMDNode;
46 class LLVMContext;
47 class raw_ostream;
48
49 class DIFile;
50 class DISubprogram;
51 class DILexicalBlock;
52 class DILexicalBlockFile;
53 class DIVariable;
54 class DIType;
55 class DIScope;
56 class DIObjCProperty;
57
58 /// \brief Maps from type identifier to the actual MDNode.
59 typedef DenseMap<const MDString *, MDNode *> DITypeIdentifierMap;
60
61 /// \brief A thin wraper around MDNode to access encoded debug info.
62 ///
63 /// This should not be stored in a container, because the underlying MDNode may
64 /// change in certain situations.
65 class DIDescriptor {
66   // Befriends DIRef so DIRef can befriend the protected member
67   // function: getFieldAs<DIRef>.
68   template <typename T> friend class DIRef;
69
70 public:
71   /// \brief Duplicated debug info flags.
72   ///
73   /// \see DebugNode::DIFlags.
74   enum {
75 #define HANDLE_DI_FLAG(ID, NAME) Flag##NAME = DebugNode::Flag##NAME,
76 #include "llvm/IR/DebugInfoFlags.def"
77     FlagAccessibility = DebugNode::FlagAccessibility
78   };
79
80   static unsigned getFlag(StringRef Flag);
81   static const char *getFlagString(unsigned Flag);
82
83   /// \brief Split up a flags bitfield.
84   ///
85   /// Split \c Flags into \c SplitFlags, a vector of its components.  Returns
86   /// any remaining (unrecognized) bits.
87   static unsigned splitFlags(unsigned Flags,
88                              SmallVectorImpl<unsigned> &SplitFlags);
89
90 protected:
91   const MDNode *DbgNode;
92
93   DIDescriptor getDescriptorField(unsigned Elt) const;
94   template <typename DescTy> DescTy getFieldAs(unsigned Elt) const {
95     return DescTy(getDescriptorField(Elt));
96   }
97
98 public:
99   explicit DIDescriptor(const MDNode *N = nullptr) : DbgNode(N) {}
100
101   MDNode *get() const { return const_cast<MDNode *>(DbgNode); }
102   operator MDNode *() const { return get(); }
103   MDNode *operator->() const { return get(); }
104   MDNode &operator*() const {
105     assert(get() && "Expected valid pointer");
106     return *get();
107   }
108
109   // An explicit operator bool so that we can do testing of DI values
110   // easily.
111   // FIXME: This operator bool isn't actually protecting anything at the
112   // moment due to the conversion operator above making DIDescriptor nodes
113   // implicitly convertable to bool.
114   explicit operator bool() const { return DbgNode != nullptr; }
115
116   bool operator==(DIDescriptor Other) const { return DbgNode == Other.DbgNode; }
117   bool operator!=(DIDescriptor Other) const { return !operator==(Other); }
118
119   uint16_t getTag() const {
120     if (auto *N = dyn_cast_or_null<DebugNode>(get()))
121       return N->getTag();
122     return 0;
123   }
124
125   bool isDerivedType() const { return get() && isa<MDDerivedTypeBase>(get()); }
126   bool isCompositeType() const {
127     return get() && isa<MDCompositeTypeBase>(get());
128   }
129   bool isSubroutineType() const {
130     return get() && isa<MDSubroutineType>(get());
131   }
132   bool isBasicType() const { return get() && isa<MDBasicType>(get()); }
133   bool isVariable() const { return get() && isa<MDLocalVariable>(get()); }
134   bool isSubprogram() const { return get() && isa<MDSubprogram>(get()); }
135   bool isGlobalVariable() const {
136     return get() && isa<MDGlobalVariable>(get());
137   }
138   bool isScope() const { return get() && isa<MDScope>(get()); }
139   bool isFile() const { return get() && isa<MDFile>(get()); }
140   bool isCompileUnit() const { return get() && isa<MDCompileUnit>(get()); }
141   bool isNameSpace() const{ return get() && isa<MDNamespace>(get()); }
142   bool isLexicalBlockFile() const {
143     return get() && isa<MDLexicalBlockFile>(get());
144   }
145   bool isLexicalBlock() const {
146     return get() && isa<MDLexicalBlockBase>(get());
147   }
148   bool isSubrange() const { return get() && isa<MDSubrange>(get()); }
149   bool isEnumerator() const { return get() && isa<MDEnumerator>(get()); }
150   bool isType() const { return get() && isa<MDType>(get()); }
151   bool isTemplateTypeParameter() const {
152     return get() && isa<MDTemplateTypeParameter>(get());
153   }
154   bool isTemplateValueParameter() const {
155     return get() && isa<MDTemplateValueParameter>(get());
156   }
157   bool isObjCProperty() const { return get() && isa<MDObjCProperty>(get()); }
158   bool isImportedEntity() const {
159     return get() && isa<MDImportedEntity>(get());
160   }
161   bool isExpression() const { return get() && isa<MDExpression>(get()); }
162
163   void print(raw_ostream &OS) const;
164   void dump() const;
165
166   /// \brief Replace all uses of debug info referenced by this descriptor.
167   void replaceAllUsesWith(LLVMContext &VMContext, DIDescriptor D);
168   void replaceAllUsesWith(MDNode *D);
169 };
170
171 /// \brief This is used to represent ranges, for array bounds.
172 class DISubrange : public DIDescriptor {
173 public:
174   explicit DISubrange(const MDNode *N = nullptr) : DIDescriptor(N) {}
175   DISubrange(const MDSubrange *N) : DIDescriptor(N) {}
176
177   MDSubrange *get() const {
178     return cast_or_null<MDSubrange>(DIDescriptor::get());
179   }
180   operator MDSubrange *() const { return get(); }
181   MDSubrange *operator->() const { return get(); }
182   MDSubrange &operator*() const {
183     assert(get() && "Expected valid pointer");
184     return *get();
185   }
186
187   int64_t getLo() const { return get()->getLo(); }
188   int64_t getCount() const { return get()->getCount(); }
189 };
190
191 /// \brief This descriptor holds an array of nodes with type T.
192 template <typename T> class DITypedArray : public DIDescriptor {
193 public:
194   explicit DITypedArray(const MDNode *N = nullptr) : DIDescriptor(N) {}
195   operator MDTuple *() const {
196     return const_cast<MDTuple *>(cast_or_null<MDTuple>(DbgNode));
197   }
198   unsigned getNumElements() const {
199     return DbgNode ? DbgNode->getNumOperands() : 0;
200   }
201   T getElement(unsigned Idx) const { return getFieldAs<T>(Idx); }
202 };
203
204 typedef DITypedArray<DIDescriptor> DIArray;
205
206 /// \brief A wrapper for an enumerator (e.g. X and Y in 'enum {X,Y}').
207 ///
208 /// FIXME: it seems strange that this doesn't have either a reference to the
209 /// type/precision or a file/line pair for location info.
210 class DIEnumerator : public DIDescriptor {
211 public:
212   explicit DIEnumerator(const MDNode *N = nullptr) : DIDescriptor(N) {}
213   DIEnumerator(const MDEnumerator *N) : DIDescriptor(N) {}
214
215   MDEnumerator *get() const {
216     return cast_or_null<MDEnumerator>(DIDescriptor::get());
217   }
218   operator MDEnumerator *() const { return get(); }
219   MDEnumerator *operator->() const { return get(); }
220   MDEnumerator &operator*() const {
221     assert(get() && "Expected valid pointer");
222     return *get();
223   }
224
225   StringRef getName() const { return get()->getName(); }
226   int64_t getEnumValue() const { return get()->getValue(); }
227 };
228
229 template <typename T> class DIRef;
230 typedef DIRef<DIDescriptor> DIDescriptorRef;
231 typedef DIRef<DIScope> DIScopeRef;
232 typedef DIRef<DIType> DITypeRef;
233 typedef DITypedArray<DITypeRef> DITypeArray;
234
235 /// \brief A base class for various scopes.
236 ///
237 /// Although, implementation-wise, DIScope is the parent class of most
238 /// other DIxxx classes, including DIType and its descendants, most of
239 /// DIScope's descendants are not a substitutable subtype of
240 /// DIScope. The DIDescriptor::isScope() method only is true for
241 /// DIScopes that are scopes in the strict lexical scope sense
242 /// (DICompileUnit, DISubprogram, etc.), but not for, e.g., a DIType.
243 class DIScope : public DIDescriptor {
244 public:
245   explicit DIScope(const MDNode *N = nullptr) : DIDescriptor(N) {}
246   DIScope(const MDScope *N) : DIDescriptor(N) {}
247
248   MDScope *get() const { return cast_or_null<MDScope>(DIDescriptor::get()); }
249   operator MDScope *() const { return get(); }
250   MDScope *operator->() const { return get(); }
251   MDScope &operator*() const {
252     assert(get() && "Expected valid pointer");
253     return *get();
254   }
255
256   /// \brief Get the parent scope.
257   ///
258   /// Gets the parent scope for this scope node or returns a default
259   /// constructed scope.
260   DIScopeRef getContext() const;
261   /// \brief Get the scope name.
262   ///
263   /// If the scope node has a name, return that, else return an empty string.
264   StringRef getName() const;
265   StringRef getFilename() const;
266   StringRef getDirectory() const;
267
268   /// \brief Generate a reference to this DIScope.
269   ///
270   /// Uses the type identifier instead of the actual MDNode if possible, to
271   /// help type uniquing.
272   DIScopeRef getRef() const;
273 };
274
275 /// \brief Represents reference to a DIDescriptor.
276 ///
277 /// Abstracts over direct and identifier-based metadata references.
278 template <typename T> class DIRef {
279   template <typename DescTy>
280   friend DescTy DIDescriptor::getFieldAs(unsigned Elt) const;
281   friend DIScopeRef DIScope::getContext() const;
282   friend DIScopeRef DIScope::getRef() const;
283   friend class DIType;
284
285   /// \brief Val can be either a MDNode or a MDString.
286   ///
287   /// In the latter, MDString specifies the type identifier.
288   const Metadata *Val;
289   explicit DIRef(const Metadata *V);
290
291 public:
292   template <class U>
293   DIRef(const TypedDebugNodeRef<U> &Ref,
294         typename std::enable_if<std::is_convertible<U *, T>::value>::type * =
295             nullptr)
296       : Val(Ref) {}
297
298   T resolve(const DITypeIdentifierMap &Map) const;
299   operator Metadata *() const { return const_cast<Metadata *>(Val); }
300
301   static DIRef get(const Metadata *MD) { return DIRef(MD); }
302 };
303
304 template <typename T>
305 T DIRef<T>::resolve(const DITypeIdentifierMap &Map) const {
306   if (!Val)
307     return T();
308
309   if (const MDNode *MD = dyn_cast<MDNode>(Val))
310     return T(MD);
311
312   const MDString *MS = cast<MDString>(Val);
313   // Find the corresponding MDNode.
314   DITypeIdentifierMap::const_iterator Iter = Map.find(MS);
315   assert(Iter != Map.end() && "Identifier not in the type map?");
316   assert(DIDescriptor(Iter->second).isType() &&
317          "MDNode in DITypeIdentifierMap should be a DIType.");
318   return T(Iter->second);
319 }
320
321 /// \brief Handle fields that are references to DIDescriptors.
322 template <>
323 DIDescriptorRef DIDescriptor::getFieldAs<DIDescriptorRef>(unsigned Elt) const;
324 /// \brief Specialize DIRef constructor for DIDescriptorRef.
325 template <> DIRef<DIDescriptor>::DIRef(const Metadata *V);
326
327 /// \brief Handle fields that are references to DIScopes.
328 template <> DIScopeRef DIDescriptor::getFieldAs<DIScopeRef>(unsigned Elt) const;
329 /// \brief Specialize DIRef constructor for DIScopeRef.
330 template <> DIRef<DIScope>::DIRef(const Metadata *V);
331
332 /// \brief Handle fields that are references to DITypes.
333 template <> DITypeRef DIDescriptor::getFieldAs<DITypeRef>(unsigned Elt) const;
334 /// \brief Specialize DIRef constructor for DITypeRef.
335 template <> DIRef<DIType>::DIRef(const Metadata *V);
336
337 /// \brief This is a wrapper for a type.
338 ///
339 /// FIXME: Types should be factored much better so that CV qualifiers and
340 /// others do not require a huge and empty descriptor full of zeros.
341 class DIType : public DIScope {
342 public:
343   explicit DIType(const MDNode *N = nullptr) : DIScope(N) {}
344   DIType(const MDType *N) : DIScope(N) {}
345
346   MDType *get() const { return cast_or_null<MDType>(DIDescriptor::get()); }
347   operator MDType *() const { return get(); }
348   MDType *operator->() const { return get(); }
349   MDType &operator*() const {
350     assert(get() && "Expected valid pointer");
351     return *get();
352   }
353
354   DIScopeRef getContext() const { return DIScopeRef::get(get()->getScope()); }
355   StringRef getName() const { return get()->getName(); }
356   unsigned getLineNumber() const { return get()->getLine(); }
357   uint64_t getSizeInBits() const { return get()->getSizeInBits(); }
358   uint64_t getAlignInBits() const { return get()->getAlignInBits(); }
359   // FIXME: Offset is only used for DW_TAG_member nodes.  Making every type
360   // carry this is just plain insane.
361   uint64_t getOffsetInBits() const { return get()->getOffsetInBits(); }
362   unsigned getFlags() const { return get()->getFlags(); }
363   bool isPrivate() const {
364     return (getFlags() & FlagAccessibility) == FlagPrivate;
365   }
366   bool isProtected() const {
367     return (getFlags() & FlagAccessibility) == FlagProtected;
368   }
369   bool isPublic() const {
370     return (getFlags() & FlagAccessibility) == FlagPublic;
371   }
372   bool isForwardDecl() const { return (getFlags() & FlagFwdDecl) != 0; }
373   bool isAppleBlockExtension() const {
374     return (getFlags() & FlagAppleBlock) != 0;
375   }
376   bool isBlockByrefStruct() const {
377     return (getFlags() & FlagBlockByrefStruct) != 0;
378   }
379   bool isVirtual() const { return (getFlags() & FlagVirtual) != 0; }
380   bool isArtificial() const { return (getFlags() & FlagArtificial) != 0; }
381   bool isObjectPointer() const { return (getFlags() & FlagObjectPointer) != 0; }
382   bool isObjcClassComplete() const {
383     return (getFlags() & FlagObjcClassComplete) != 0;
384   }
385   bool isVector() const { return (getFlags() & FlagVector) != 0; }
386   bool isStaticMember() const { return (getFlags() & FlagStaticMember) != 0; }
387   bool isLValueReference() const {
388     return (getFlags() & FlagLValueReference) != 0;
389   }
390   bool isRValueReference() const {
391     return (getFlags() & FlagRValueReference) != 0;
392   }
393   bool isValid() const { return DbgNode && isType(); }
394 };
395
396 /// \brief A basic type, like 'int' or 'float'.
397 class DIBasicType : public DIType {
398 public:
399   explicit DIBasicType(const MDNode *N = nullptr) : DIType(N) {}
400   DIBasicType(const MDBasicType *N) : DIType(N) {}
401
402   MDBasicType *get() const {
403     return cast_or_null<MDBasicType>(DIDescriptor::get());
404   }
405   operator MDBasicType *() const { return get(); }
406   MDBasicType *operator->() const { return get(); }
407   MDBasicType &operator*() const {
408     assert(get() && "Expected valid pointer");
409     return *get();
410   }
411
412   unsigned getEncoding() const { return get()->getEncoding(); }
413 };
414
415 /// \brief A simple derived type
416 ///
417 /// Like a const qualified type, a typedef, a pointer or reference, et cetera.
418 /// Or, a data member of a class/struct/union.
419 class DIDerivedType : public DIType {
420 public:
421   explicit DIDerivedType(const MDNode *N = nullptr) : DIType(N) {}
422   DIDerivedType(const MDDerivedTypeBase *N) : DIType(N) {}
423
424   MDDerivedTypeBase *get() const {
425     return cast_or_null<MDDerivedTypeBase>(DIDescriptor::get());
426   }
427   operator MDDerivedTypeBase *() const { return get(); }
428   MDDerivedTypeBase *operator->() const { return get(); }
429   MDDerivedTypeBase &operator*() const {
430     assert(get() && "Expected valid pointer");
431     return *get();
432   }
433
434   DITypeRef getTypeDerivedFrom() const {
435     return DITypeRef::get(get()->getBaseType());
436   }
437
438   /// \brief Return property node, if this ivar is associated with one.
439   MDNode *getObjCProperty() const {
440     if (auto *N = dyn_cast<MDDerivedType>(get()))
441       return dyn_cast_or_null<MDNode>(N->getExtraData());
442     return nullptr;
443   }
444
445   DITypeRef getClassType() const {
446     assert(getTag() == dwarf::DW_TAG_ptr_to_member_type);
447     if (auto *N = dyn_cast<MDDerivedType>(get()))
448       return DITypeRef::get(N->getExtraData());
449     return DITypeRef::get(nullptr);
450   }
451
452   Constant *getConstant() const {
453     assert((getTag() == dwarf::DW_TAG_member) && isStaticMember());
454     if (auto *N = dyn_cast<MDDerivedType>(get()))
455       if (auto *C = dyn_cast_or_null<ConstantAsMetadata>(N->getExtraData()))
456         return C->getValue();
457
458     return nullptr;
459   }
460 };
461
462 /// \brief Types that refer to multiple other types.
463 ///
464 /// This descriptor holds a type that can refer to multiple other types, like a
465 /// function or struct.
466 ///
467 /// DICompositeType is derived from DIDerivedType because some
468 /// composite types (such as enums) can be derived from basic types
469 // FIXME: Make this derive from DIType directly & just store the
470 // base type in a single DIType field.
471 class DICompositeType : public DIDerivedType {
472   friend class DIBuilder;
473
474   /// \brief Set the array of member DITypes.
475   void setArraysHelper(MDNode *Elements, MDNode *TParams);
476
477 public:
478   explicit DICompositeType(const MDNode *N = nullptr) : DIDerivedType(N) {}
479   DICompositeType(const MDCompositeTypeBase *N) : DIDerivedType(N) {}
480
481   MDCompositeTypeBase *get() const {
482     return cast_or_null<MDCompositeTypeBase>(DIDescriptor::get());
483   }
484   operator MDCompositeTypeBase *() const { return get(); }
485   MDCompositeTypeBase *operator->() const { return get(); }
486   MDCompositeTypeBase &operator*() const {
487     assert(get() && "Expected valid pointer");
488     return *get();
489   }
490
491   DIArray getElements() const {
492     assert(!isSubroutineType() && "no elements for DISubroutineType");
493     return DIArray(get()->getElements());
494   }
495
496 private:
497   template <typename T>
498   void setArrays(DITypedArray<T> Elements, DIArray TParams = DIArray()) {
499     assert(
500         (!TParams || DbgNode->getNumOperands() == 8) &&
501         "If you're setting the template parameters this should include a slot "
502         "for that!");
503     setArraysHelper(Elements, TParams);
504   }
505
506 public:
507   unsigned getRunTimeLang() const { return get()->getRuntimeLang(); }
508   DITypeRef getContainingType() const {
509     return DITypeRef::get(get()->getVTableHolder());
510   }
511
512 private:
513   /// \brief Set the containing type.
514   void setContainingType(DICompositeType ContainingType);
515
516 public:
517   DIArray getTemplateParams() const {
518     return DIArray(get()->getTemplateParams());
519   }
520   MDString *getIdentifier() const { return get()->getRawIdentifier(); }
521 };
522
523 class DISubroutineType : public DICompositeType {
524 public:
525   explicit DISubroutineType(const MDNode *N = nullptr) : DICompositeType(N) {}
526   DISubroutineType(const MDSubroutineType *N) : DICompositeType(N) {}
527
528   MDSubroutineType *get() const {
529     return cast_or_null<MDSubroutineType>(DIDescriptor::get());
530   }
531   operator MDSubroutineType *() const { return get(); }
532   MDSubroutineType *operator->() const { return get(); }
533   MDSubroutineType &operator*() const {
534     assert(get() && "Expected valid pointer");
535     return *get();
536   }
537
538   DITypedArray<DITypeRef> getTypeArray() const {
539     return DITypedArray<DITypeRef>(get()->getTypeArray());
540   }
541 };
542
543 /// \brief This is a wrapper for a file.
544 class DIFile : public DIScope {
545 public:
546   explicit DIFile(const MDNode *N = nullptr) : DIScope(N) {}
547   DIFile(const MDFile *N) : DIScope(N) {}
548
549   MDFile *get() const { return cast_or_null<MDFile>(DIDescriptor::get()); }
550   operator MDFile *() const { return get(); }
551   MDFile *operator->() const { return get(); }
552   MDFile &operator*() const {
553     assert(get() && "Expected valid pointer");
554     return *get();
555   }
556
557   /// \brief Retrieve the MDNode for the directory/file pair.
558   MDNode *getFileNode() const { return get(); }
559 };
560
561 /// \brief A wrapper for a compile unit.
562 class DICompileUnit : public DIScope {
563 public:
564   explicit DICompileUnit(const MDNode *N = nullptr) : DIScope(N) {}
565   DICompileUnit(const MDCompileUnit *N) : DIScope(N) {}
566
567   MDCompileUnit *get() const {
568     return cast_or_null<MDCompileUnit>(DIDescriptor::get());
569   }
570   operator MDCompileUnit *() const { return get(); }
571   MDCompileUnit *operator->() const { return get(); }
572   MDCompileUnit &operator*() const {
573     assert(get() && "Expected valid pointer");
574     return *get();
575   }
576
577   dwarf::SourceLanguage getLanguage() const {
578     return static_cast<dwarf::SourceLanguage>(get()->getSourceLanguage());
579   }
580   StringRef getProducer() const { return get()->getProducer(); }
581   bool isOptimized() const { return get()->isOptimized(); }
582   StringRef getFlags() const { return get()->getFlags(); }
583   unsigned getRunTimeVersion() const { return get()->getRuntimeVersion(); }
584
585   DIArray getEnumTypes() const { return DIArray(get()->getEnumTypes()); }
586   DIArray getRetainedTypes() const {
587     return DIArray(get()->getRetainedTypes());
588   }
589   DIArray getSubprograms() const { return DIArray(get()->getSubprograms()); }
590   DIArray getGlobalVariables() const {
591     return DIArray(get()->getGlobalVariables());
592   }
593   DIArray getImportedEntities() const {
594     return DIArray(get()->getImportedEntities());
595   }
596
597   void replaceSubprograms(DIArray Subprograms);
598   void replaceGlobalVariables(DIArray GlobalVariables);
599
600   StringRef getSplitDebugFilename() const {
601     return get()->getSplitDebugFilename();
602   }
603   unsigned getEmissionKind() const { return get()->getEmissionKind(); }
604 };
605
606 /// \brief This is a wrapper for a subprogram (e.g. a function).
607 class DISubprogram : public DIScope {
608 public:
609   explicit DISubprogram(const MDNode *N = nullptr) : DIScope(N) {}
610   DISubprogram(const MDSubprogram *N) : DIScope(N) {}
611
612   MDSubprogram *get() const {
613     return cast_or_null<MDSubprogram>(DIDescriptor::get());
614   }
615   operator MDSubprogram *() const { return get(); }
616   MDSubprogram *operator->() const { return get(); }
617   MDSubprogram &operator*() const {
618     assert(get() && "Expected valid pointer");
619     return *get();
620   }
621
622   StringRef getName() const { return get()->getName(); }
623   StringRef getDisplayName() const { return get()->getDisplayName(); }
624   StringRef getLinkageName() const { return get()->getLinkageName(); }
625   unsigned getLineNumber() const { return get()->getLine(); }
626
627   /// \brief Check if this is local (like 'static' in C).
628   unsigned isLocalToUnit() const { return get()->isLocalToUnit(); }
629   unsigned isDefinition() const { return get()->isDefinition(); }
630
631   unsigned getVirtuality() const { return get()->getVirtuality(); }
632   unsigned getVirtualIndex() const { return get()->getVirtualIndex(); }
633
634   unsigned getFlags() const { return get()->getFlags(); }
635
636   unsigned isOptimized() const { return get()->isOptimized(); }
637
638   /// \brief Get the beginning of the scope of the function (not the name).
639   unsigned getScopeLineNumber() const { return get()->getScopeLine(); }
640
641   DIScopeRef getContext() const { return DIScopeRef::get(get()->getScope()); }
642   DISubroutineType getType() const {
643     return DISubroutineType(get()->getType());
644   }
645
646   DITypeRef getContainingType() const {
647     return DITypeRef::get(get()->getContainingType());
648   }
649
650   /// \brief Check if this provides debugging information for the function F.
651   bool describes(const Function *F);
652
653   Function *getFunction() const;
654
655   void replaceFunction(Function *F) {
656     if (auto *N = get())
657       N->replaceFunction(F);
658   }
659   DIArray getTemplateParams() const {
660     return DIArray(get()->getTemplateParams());
661   }
662   DISubprogram getFunctionDeclaration() const {
663     return DISubprogram(get()->getDeclaration());
664   }
665   MDNode *getVariablesNodes() const { return getVariables(); }
666   DIArray getVariables() const { return DIArray(get()->getVariables()); }
667
668   unsigned isArtificial() const { return (getFlags() & FlagArtificial) != 0; }
669   /// \brief Check for the "private" access specifier.
670   bool isPrivate() const {
671     return (getFlags() & FlagAccessibility) == FlagPrivate;
672   }
673   /// \brief Check for the "protected" access specifier.
674   bool isProtected() const {
675     return (getFlags() & FlagAccessibility) == FlagProtected;
676   }
677   /// \brief Check for the "public" access specifier.
678   bool isPublic() const {
679     return (getFlags() & FlagAccessibility) == FlagPublic;
680   }
681   /// \brief Check for "explicit".
682   bool isExplicit() const { return (getFlags() & FlagExplicit) != 0; }
683   /// \brief Check if this is prototyped.
684   bool isPrototyped() const { return (getFlags() & FlagPrototyped) != 0; }
685
686   /// \brief Check if this is reference-qualified.
687   ///
688   /// Return true if this subprogram is a C++11 reference-qualified non-static
689   /// member function (void foo() &).
690   unsigned isLValueReference() const {
691     return (getFlags() & FlagLValueReference) != 0;
692   }
693
694   /// \brief Check if this is rvalue-reference-qualified.
695   ///
696   /// Return true if this subprogram is a C++11 rvalue-reference-qualified
697   /// non-static member function (void foo() &&).
698   unsigned isRValueReference() const {
699     return (getFlags() & FlagRValueReference) != 0;
700   }
701 };
702
703 /// \brief This is a wrapper for a lexical block.
704 class DILexicalBlock : public DIScope {
705 public:
706   explicit DILexicalBlock(const MDNode *N = nullptr) : DIScope(N) {}
707   DILexicalBlock(const MDLexicalBlockBase *N) : DIScope(N) {}
708
709   MDLexicalBlockBase *get() const {
710     return cast_or_null<MDLexicalBlockBase>(DIDescriptor::get());
711   }
712   operator MDLexicalBlockBase *() const { return get(); }
713   MDLexicalBlockBase *operator->() const { return get(); }
714   MDLexicalBlockBase &operator*() const {
715     assert(get() && "Expected valid pointer");
716     return *get();
717   }
718
719   DIScope getContext() const { return DIScope(get()->getScope()); }
720   unsigned getLineNumber() const {
721     if (auto *N = dyn_cast<MDLexicalBlock>(get()))
722       return N->getLine();
723     return 0;
724   }
725   unsigned getColumnNumber() const {
726     if (auto *N = dyn_cast<MDLexicalBlock>(get()))
727       return N->getColumn();
728     return 0;
729   }
730 };
731
732 /// \brief This is a wrapper for a lexical block with a filename change.
733 class DILexicalBlockFile : public DIScope {
734 public:
735   explicit DILexicalBlockFile(const MDNode *N = nullptr) : DIScope(N) {}
736   DILexicalBlockFile(const MDLexicalBlockFile *N) : DIScope(N) {}
737
738   MDLexicalBlockFile *get() const {
739     return cast_or_null<MDLexicalBlockFile>(DIDescriptor::get());
740   }
741   operator MDLexicalBlockFile *() const { return get(); }
742   MDLexicalBlockFile *operator->() const { return get(); }
743   MDLexicalBlockFile &operator*() const {
744     assert(get() && "Expected valid pointer");
745     return *get();
746   }
747
748   DIScope getContext() const { return getScope(); }
749   unsigned getLineNumber() const { return getScope().getLineNumber(); }
750   unsigned getColumnNumber() const { return getScope().getColumnNumber(); }
751   DILexicalBlock getScope() const { return DILexicalBlock(get()->getScope()); }
752   unsigned getDiscriminator() const { return get()->getDiscriminator(); }
753 };
754
755 /// \brief A wrapper for a C++ style name space.
756 class DINameSpace : public DIScope {
757 public:
758   explicit DINameSpace(const MDNode *N = nullptr) : DIScope(N) {}
759   DINameSpace(const MDNamespace *N) : DIScope(N) {}
760
761   MDNamespace *get() const {
762     return cast_or_null<MDNamespace>(DIDescriptor::get());
763   }
764   operator MDNamespace *() const { return get(); }
765   MDNamespace *operator->() const { return get(); }
766   MDNamespace &operator*() const {
767     assert(get() && "Expected valid pointer");
768     return *get();
769   }
770
771   StringRef getName() const { return get()->getName(); }
772   unsigned getLineNumber() const { return get()->getLine(); }
773   DIScope getContext() const { return DIScope(get()->getScope()); }
774 };
775
776 /// \brief This is a wrapper for template type parameter.
777 class DITemplateTypeParameter : public DIDescriptor {
778 public:
779   explicit DITemplateTypeParameter(const MDNode *N = nullptr)
780       : DIDescriptor(N) {}
781   DITemplateTypeParameter(const MDTemplateTypeParameter *N) : DIDescriptor(N) {}
782
783   MDTemplateTypeParameter *get() const {
784     return cast_or_null<MDTemplateTypeParameter>(DIDescriptor::get());
785   }
786   operator MDTemplateTypeParameter *() const { return get(); }
787   MDTemplateTypeParameter *operator->() const { return get(); }
788   MDTemplateTypeParameter &operator*() const {
789     assert(get() && "Expected valid pointer");
790     return *get();
791   }
792
793   StringRef getName() const { return get()->getName(); }
794
795   DITypeRef getType() const { return DITypeRef::get(get()->getType()); }
796 };
797
798 /// \brief This is a wrapper for template value parameter.
799 class DITemplateValueParameter : public DIDescriptor {
800 public:
801   explicit DITemplateValueParameter(const MDNode *N = nullptr)
802       : DIDescriptor(N) {}
803   DITemplateValueParameter(const MDTemplateValueParameter *N)
804       : DIDescriptor(N) {}
805
806   MDTemplateValueParameter *get() const {
807     return cast_or_null<MDTemplateValueParameter>(DIDescriptor::get());
808   }
809   operator MDTemplateValueParameter *() const { return get(); }
810   MDTemplateValueParameter *operator->() const { return get(); }
811   MDTemplateValueParameter &operator*() const {
812     assert(get() && "Expected valid pointer");
813     return *get();
814   }
815
816   StringRef getName() const { return get()->getName(); }
817   DITypeRef getType() const { return DITypeRef::get(get()->getType()); }
818   Metadata *getValue() const { return get()->getValue(); }
819 };
820
821 /// \brief This is a wrapper for a global variable.
822 class DIGlobalVariable : public DIDescriptor {
823   DIFile getFile() const { return DIFile(get()->getFile()); }
824
825 public:
826   explicit DIGlobalVariable(const MDNode *N = nullptr) : DIDescriptor(N) {}
827   DIGlobalVariable(const MDGlobalVariable *N) : DIDescriptor(N) {}
828
829   MDGlobalVariable *get() const {
830     return cast_or_null<MDGlobalVariable>(DIDescriptor::get());
831   }
832   operator MDGlobalVariable *() const { return get(); }
833   MDGlobalVariable *operator->() const { return get(); }
834   MDGlobalVariable &operator*() const {
835     assert(get() && "Expected valid pointer");
836     return *get();
837   }
838
839   StringRef getName() const { return get()->getName(); }
840   StringRef getDisplayName() const { return get()->getDisplayName(); }
841   StringRef getLinkageName() const { return get()->getLinkageName(); }
842   unsigned getLineNumber() const { return get()->getLine(); }
843   unsigned isLocalToUnit() const { return get()->isLocalToUnit(); }
844   unsigned isDefinition() const { return get()->isDefinition(); }
845
846   DIScope getContext() const { return DIScope(get()->getScope()); }
847   StringRef getFilename() const { return getFile().getFilename(); }
848   StringRef getDirectory() const { return getFile().getDirectory(); }
849   DITypeRef getType() const { return DITypeRef::get(get()->getType()); }
850
851   GlobalVariable *getGlobal() const;
852   Constant *getConstant() const {
853     if (auto *N = get())
854       if (auto *C = dyn_cast_or_null<ConstantAsMetadata>(N->getVariable()))
855         return C->getValue();
856     return nullptr;
857   }
858   DIDerivedType getStaticDataMemberDeclaration() const {
859     return DIDerivedType(get()->getStaticDataMemberDeclaration());
860   }
861 };
862
863 /// \brief This is a wrapper for a variable (e.g. parameter, local, global etc).
864 class DIVariable : public DIDescriptor {
865   unsigned getFlags() const { return get()->getFlags(); }
866
867 public:
868   explicit DIVariable(const MDNode *N = nullptr) : DIDescriptor(N) {}
869   DIVariable(const MDLocalVariable *N) : DIDescriptor(N) {}
870
871   MDLocalVariable *get() const {
872     return cast_or_null<MDLocalVariable>(DIDescriptor::get());
873   }
874   operator MDLocalVariable *() const { return get(); }
875   MDLocalVariable *operator->() const { return get(); }
876   MDLocalVariable &operator*() const {
877     assert(get() && "Expected valid pointer");
878     return *get();
879   }
880
881   StringRef getName() const { return get()->getName(); }
882   unsigned getLineNumber() const { return get()->getLine(); }
883   unsigned getArgNumber() const { return get()->getArg(); }
884
885   DIScope getContext() const { return DIScope(get()->getScope()); }
886   DIFile getFile() const { return DIFile(get()->getFile()); }
887   DITypeRef getType() const { return DITypeRef::get(get()->getType()); }
888
889   /// \brief Return true if this variable is marked as "artificial".
890   bool isArtificial() const {
891     return (getFlags() & FlagArtificial) != 0;
892   }
893
894   bool isObjectPointer() const {
895     return (getFlags() & FlagObjectPointer) != 0;
896   }
897
898   /// \brief If this variable is inlined then return inline location.
899   MDNode *getInlinedAt() const { return DIDescriptor(get()->getInlinedAt()); }
900
901   /// \brief Check if this is a "__block" variable (Apple Blocks).
902   bool isBlockByrefVariable(const DITypeIdentifierMap &Map) const {
903     return (getType().resolve(Map)).isBlockByrefStruct();
904   }
905
906   /// \brief Check if this is an inlined function argument.
907   bool isInlinedFnArgument(const Function *CurFn);
908
909   /// \brief Return the size reported by the variable's type.
910   unsigned getSizeInBits(const DITypeIdentifierMap &Map);
911
912   void printExtendedName(raw_ostream &OS) const;
913 };
914
915 /// \brief A complex location expression in postfix notation.
916 ///
917 /// This is (almost) a DWARF expression that modifies the location of a
918 /// variable or (or the location of a single piece of a variable).
919 ///
920 /// FIXME: Instead of DW_OP_plus taking an argument, this should use DW_OP_const
921 /// and have DW_OP_plus consume the topmost elements on the stack.
922 class DIExpression : public DIDescriptor {
923 public:
924   explicit DIExpression(const MDNode *N = nullptr) : DIDescriptor(N) {}
925   DIExpression(const MDExpression *N) : DIDescriptor(N) {}
926
927   MDExpression *get() const {
928     return cast_or_null<MDExpression>(DIDescriptor::get());
929   }
930   operator MDExpression *() const { return get(); }
931   MDExpression *operator->() const { return get(); }
932   MDExpression &operator*() const {
933     assert(get() && "Expected valid pointer");
934     return *get();
935   }
936
937   /// \brief Return the number of elements in the complex expression.
938   unsigned getNumElements() const { return get()->getNumElements(); }
939
940   /// \brief return the Idx'th complex address element.
941   uint64_t getElement(unsigned I) const { return get()->getElement(I); }
942
943   /// \brief Return whether this is a piece of an aggregate variable.
944   bool isBitPiece() const;
945   /// \brief Return the offset of this piece in bits.
946   uint64_t getBitPieceOffset() const;
947   /// \brief Return the size of this piece in bits.
948   uint64_t getBitPieceSize() const;
949
950   class iterator;
951   /// \brief A lightweight wrapper around an element of a DIExpression.
952   class Operand {
953     friend class iterator;
954     MDExpression::element_iterator I;
955     Operand() {}
956     Operand(MDExpression::element_iterator I) : I(I) {}
957   public:
958     /// \brief Operands such as DW_OP_piece have explicit (non-stack) arguments.
959     /// Argument 0 is the operand itself.
960     uint64_t getArg(unsigned N) const {
961       MDExpression::element_iterator In = I;
962       std::advance(In, N);
963       return *In;
964     }
965     operator uint64_t () const { return *I; }
966     /// \brief Returns underlying MDExpression::element_iterator.
967     const MDExpression::element_iterator &getBase() const { return I; }
968     /// \brief Returns the next operand.
969     iterator getNext() const;
970   };
971
972   /// \brief An iterator for DIExpression elements.
973   class iterator : public std::iterator<std::input_iterator_tag, StringRef,
974                                         unsigned, const Operand*, Operand> {
975     friend class Operand;
976     MDExpression::element_iterator I;
977     Operand Tmp;
978
979   public:
980     iterator(MDExpression::element_iterator I) : I(I) {}
981     const Operand &operator*() { return Tmp = Operand(I); }
982     const Operand *operator->() { return &(Tmp = Operand(I)); }
983     iterator &operator++() {
984       increment();
985       return *this;
986     }
987     iterator operator++(int) {
988       iterator X(*this);
989       increment();
990       return X;
991     }
992     bool operator==(const iterator &X) const { return I == X.I; }
993     bool operator!=(const iterator &X) const { return !(*this == X); }
994
995   private:
996     void increment() {
997       switch (**this) {
998       case dwarf::DW_OP_bit_piece: std::advance(I, 3); break;
999       case dwarf::DW_OP_plus:      std::advance(I, 2); break;
1000       case dwarf::DW_OP_deref:     std::advance(I, 1); break;
1001       default:
1002         llvm_unreachable("unsupported operand");
1003       }
1004     }
1005   };
1006
1007   iterator begin() const { return get()->elements_begin(); }
1008   iterator end() const { return get()->elements_end(); }
1009 };
1010
1011 /// \brief This object holds location information.
1012 ///
1013 /// This object is not associated with any DWARF tag.
1014 class DILocation : public DIDescriptor {
1015 public:
1016   explicit DILocation(const MDNode *N) : DIDescriptor(N) {}
1017   DILocation(const MDLocation *N) : DIDescriptor(N) {}
1018
1019   MDLocation *get() const {
1020     return cast_or_null<MDLocation>(DIDescriptor::get());
1021   }
1022   operator MDLocation *() const { return get(); }
1023   MDLocation *operator->() const { return get(); }
1024   MDLocation &operator*() const {
1025     assert(get() && "Expected valid pointer");
1026     return *get();
1027   }
1028
1029   unsigned getLineNumber() const { return get()->getLine(); }
1030   unsigned getColumnNumber() const { return get()->getColumn(); }
1031   DIScope getScope() const { return DIScope(get()->getScope()); }
1032   DILocation getOrigLocation() const {
1033     return DILocation(get()->getInlinedAt());
1034   }
1035   StringRef getFilename() const { return getScope().getFilename(); }
1036   StringRef getDirectory() const { return getScope().getDirectory(); }
1037   bool atSameLineAs(const DILocation &Other) const {
1038     return (getLineNumber() == Other.getLineNumber() &&
1039             getFilename() == Other.getFilename());
1040   }
1041   /// \brief Get the DWAF discriminator.
1042   ///
1043   /// DWARF discriminators are used to distinguish identical file locations for
1044   /// instructions that are on different basic blocks. If two instructions are
1045   /// inside the same lexical block and are in different basic blocks, we
1046   /// create a new lexical block with identical location as the original but
1047   /// with a different discriminator value
1048   /// (lib/Transforms/Util/AddDiscriminators.cpp for details).
1049   unsigned getDiscriminator() const {
1050     // Since discriminators are associated with lexical blocks, make
1051     // sure this location is a lexical block before retrieving its
1052     // value.
1053     return getScope().isLexicalBlockFile()
1054                ? DILexicalBlockFile(
1055                      cast<MDNode>(cast<MDLocation>(DbgNode)->getScope()))
1056                      .getDiscriminator()
1057                : 0;
1058   }
1059
1060   /// \brief Generate a new discriminator value for this location.
1061   unsigned computeNewDiscriminator(LLVMContext &Ctx);
1062
1063   /// \brief Return a copy of this location with a different scope.
1064   DILocation copyWithNewScope(LLVMContext &Ctx, DILexicalBlockFile NewScope);
1065 };
1066
1067 class DIObjCProperty : public DIDescriptor {
1068 public:
1069   explicit DIObjCProperty(const MDNode *N) : DIDescriptor(N) {}
1070   DIObjCProperty(const MDObjCProperty *N) : DIDescriptor(N) {}
1071
1072   MDObjCProperty *get() const {
1073     return cast_or_null<MDObjCProperty>(DIDescriptor::get());
1074   }
1075   operator MDObjCProperty *() const { return get(); }
1076   MDObjCProperty *operator->() const { return get(); }
1077   MDObjCProperty &operator*() const {
1078     assert(get() && "Expected valid pointer");
1079     return *get();
1080   }
1081
1082   StringRef getObjCPropertyName() const { return get()->getName(); }
1083   DIFile getFile() const { return DIFile(get()->getFile()); }
1084   unsigned getLineNumber() const { return get()->getLine(); }
1085
1086   StringRef getObjCPropertyGetterName() const { return get()->getGetterName(); }
1087   StringRef getObjCPropertySetterName() const { return get()->getSetterName(); }
1088   unsigned getAttributes() const { return get()->getAttributes(); }
1089   bool isReadOnlyObjCProperty() const {
1090     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_readonly) != 0;
1091   }
1092   bool isReadWriteObjCProperty() const {
1093     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_readwrite) != 0;
1094   }
1095   bool isAssignObjCProperty() const {
1096     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_assign) != 0;
1097   }
1098   bool isRetainObjCProperty() const {
1099     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_retain) != 0;
1100   }
1101   bool isCopyObjCProperty() const {
1102     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_copy) != 0;
1103   }
1104   bool isNonAtomicObjCProperty() const {
1105     return (getAttributes() & dwarf::DW_APPLE_PROPERTY_nonatomic) != 0;
1106   }
1107
1108   /// \brief Get the type.
1109   ///
1110   /// \note Objective-C doesn't have an ODR, so there is no benefit in storing
1111   /// the type as a DITypeRef here.
1112   DIType getType() const { return DIType(get()->getType()); }
1113 };
1114
1115 /// \brief An imported module (C++ using directive or similar).
1116 class DIImportedEntity : public DIDescriptor {
1117 public:
1118   DIImportedEntity() = default;
1119   explicit DIImportedEntity(const MDNode *N) : DIDescriptor(N) {}
1120   DIImportedEntity(const MDImportedEntity *N) : DIDescriptor(N) {}
1121
1122   MDImportedEntity *get() const {
1123     return cast_or_null<MDImportedEntity>(DIDescriptor::get());
1124   }
1125   operator MDImportedEntity *() const { return get(); }
1126   MDImportedEntity *operator->() const { return get(); }
1127   MDImportedEntity &operator*() const {
1128     assert(get() && "Expected valid pointer");
1129     return *get();
1130   }
1131
1132   DIScope getContext() const { return DIScope(get()->getScope()); }
1133   DIDescriptorRef getEntity() const {
1134     return DIDescriptorRef::get(get()->getEntity());
1135   }
1136   unsigned getLineNumber() const { return get()->getLine(); }
1137   StringRef getName() const { return get()->getName(); }
1138 };
1139
1140 /// \brief Find subprogram that is enclosing this scope.
1141 DISubprogram getDISubprogram(const MDNode *Scope);
1142
1143 /// \brief Find debug info for a given function.
1144 /// \returns a valid DISubprogram, if found. Otherwise, it returns an empty
1145 /// DISubprogram.
1146 DISubprogram getDISubprogram(const Function *F);
1147
1148 /// \brief Find underlying composite type.
1149 DICompositeType getDICompositeType(DIType T);
1150
1151 /// \brief Create a new inlined variable based on current variable.
1152 ///
1153 /// @param DV            Current Variable.
1154 /// @param InlinedScope  Location at current variable is inlined.
1155 DIVariable createInlinedVariable(MDNode *DV, MDNode *InlinedScope,
1156                                  LLVMContext &VMContext);
1157
1158 /// \brief Remove inlined scope from the variable.
1159 DIVariable cleanseInlinedVariable(MDNode *DV, LLVMContext &VMContext);
1160
1161 /// \brief Generate map by visiting all retained types.
1162 DITypeIdentifierMap generateDITypeIdentifierMap(const NamedMDNode *CU_Nodes);
1163
1164 /// \brief Strip debug info in the module if it exists.
1165 ///
1166 /// To do this, we remove all calls to the debugger intrinsics and any named
1167 /// metadata for debugging. We also remove debug locations for instructions.
1168 /// Return true if module is modified.
1169 bool StripDebugInfo(Module &M);
1170 bool stripDebugInfo(Function &F);
1171
1172 /// \brief Return Debug Info Metadata Version by checking module flags.
1173 unsigned getDebugMetadataVersionFromModule(const Module &M);
1174
1175 /// \brief Utility to find all debug info in a module.
1176 ///
1177 /// DebugInfoFinder tries to list all debug info MDNodes used in a module. To
1178 /// list debug info MDNodes used by an instruction, DebugInfoFinder uses
1179 /// processDeclare, processValue and processLocation to handle DbgDeclareInst,
1180 /// DbgValueInst and DbgLoc attached to instructions. processModule will go
1181 /// through all DICompileUnits in llvm.dbg.cu and list debug info MDNodes
1182 /// used by the CUs.
1183 class DebugInfoFinder {
1184 public:
1185   DebugInfoFinder() : TypeMapInitialized(false) {}
1186
1187   /// \brief Process entire module and collect debug info anchors.
1188   void processModule(const Module &M);
1189
1190   /// \brief Process DbgDeclareInst.
1191   void processDeclare(const Module &M, const DbgDeclareInst *DDI);
1192   /// \brief Process DbgValueInst.
1193   void processValue(const Module &M, const DbgValueInst *DVI);
1194   /// \brief Process DILocation.
1195   void processLocation(const Module &M, DILocation Loc);
1196
1197   /// \brief Clear all lists.
1198   void reset();
1199
1200 private:
1201   void InitializeTypeMap(const Module &M);
1202
1203   void processType(DIType DT);
1204   void processSubprogram(DISubprogram SP);
1205   void processScope(DIScope Scope);
1206   bool addCompileUnit(DICompileUnit CU);
1207   bool addGlobalVariable(DIGlobalVariable DIG);
1208   bool addSubprogram(DISubprogram SP);
1209   bool addType(DIType DT);
1210   bool addScope(DIScope Scope);
1211
1212 public:
1213   typedef SmallVectorImpl<DICompileUnit>::const_iterator compile_unit_iterator;
1214   typedef SmallVectorImpl<DISubprogram>::const_iterator subprogram_iterator;
1215   typedef SmallVectorImpl<DIGlobalVariable>::const_iterator
1216       global_variable_iterator;
1217   typedef SmallVectorImpl<DIType>::const_iterator type_iterator;
1218   typedef SmallVectorImpl<DIScope>::const_iterator scope_iterator;
1219
1220   iterator_range<compile_unit_iterator> compile_units() const {
1221     return iterator_range<compile_unit_iterator>(CUs.begin(), CUs.end());
1222   }
1223
1224   iterator_range<subprogram_iterator> subprograms() const {
1225     return iterator_range<subprogram_iterator>(SPs.begin(), SPs.end());
1226   }
1227
1228   iterator_range<global_variable_iterator> global_variables() const {
1229     return iterator_range<global_variable_iterator>(GVs.begin(), GVs.end());
1230   }
1231
1232   iterator_range<type_iterator> types() const {
1233     return iterator_range<type_iterator>(TYs.begin(), TYs.end());
1234   }
1235
1236   iterator_range<scope_iterator> scopes() const {
1237     return iterator_range<scope_iterator>(Scopes.begin(), Scopes.end());
1238   }
1239
1240   unsigned compile_unit_count() const { return CUs.size(); }
1241   unsigned global_variable_count() const { return GVs.size(); }
1242   unsigned subprogram_count() const { return SPs.size(); }
1243   unsigned type_count() const { return TYs.size(); }
1244   unsigned scope_count() const { return Scopes.size(); }
1245
1246 private:
1247   SmallVector<DICompileUnit, 8> CUs;
1248   SmallVector<DISubprogram, 8> SPs;
1249   SmallVector<DIGlobalVariable, 8> GVs;
1250   SmallVector<DIType, 8> TYs;
1251   SmallVector<DIScope, 8> Scopes;
1252   SmallPtrSet<MDNode *, 64> NodesSeen;
1253   DITypeIdentifierMap TypeIdentifierMap;
1254
1255   /// \brief Specify if TypeIdentifierMap is initialized.
1256   bool TypeMapInitialized;
1257 };
1258
1259 DenseMap<const Function *, DISubprogram> makeSubprogramMap(const Module &M);
1260
1261 } // end namespace llvm
1262
1263 #endif