1 //==- NativeEnumModules.cpp - Native Symbol Enumerator impl ------*- 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 #include "llvm/DebugInfo/PDB/Native/NativeEnumModules.h"
11 
12 #include "llvm/DebugInfo/PDB/IPDBEnumChildren.h"
13 #include "llvm/DebugInfo/PDB/Native/NativeCompilandSymbol.h"
14 #include "llvm/DebugInfo/PDB/Native/NativeExeSymbol.h"
15 #include "llvm/DebugInfo/PDB/Native/NativeSession.h"
16 #include "llvm/DebugInfo/PDB/PDBSymbol.h"
17 #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
18 #include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
19 
20 namespace llvm {
21 namespace pdb {
22 
23 NativeEnumModules::NativeEnumModules(NativeSession &PDBSession, uint32_t Index)
24     : Session(PDBSession), Index(Index) {}
25 
26 uint32_t NativeEnumModules::getChildCount() const {
27   return Session.getSymbolCache().getNumCompilands();
28 }
29 
30 std::unique_ptr<PDBSymbol>
31 NativeEnumModules::getChildAtIndex(uint32_t N) const {
32   return Session.getSymbolCache().getOrCreateCompiland(N);
33 }
34 
35 std::unique_ptr<PDBSymbol> NativeEnumModules::getNext() {
36   if (Index >= getChildCount())
37     return nullptr;
38   return getChildAtIndex(Index++);
39 }
40 
41 void NativeEnumModules::reset() { Index = 0; }
42 
43 }
44 }
45