1 | //==- DIAEnumSourceFiles.cpp - DIA Source File Enumerator impl ---*- 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 "llvm/DebugInfo/PDB/DIA/DIAEnumSourceFiles.h" |
10 | #include "llvm/DebugInfo/PDB/DIA/DIASourceFile.h" |
11 | #include "llvm/DebugInfo/PDB/PDBSymbol.h" |
12 | |
13 | using namespace llvm; |
14 | using namespace llvm::pdb; |
15 | |
16 | DIAEnumSourceFiles::DIAEnumSourceFiles( |
17 | const DIASession &PDBSession, CComPtr<IDiaEnumSourceFiles> DiaEnumerator) |
18 | : Session(PDBSession), Enumerator(DiaEnumerator) {} |
19 | |
20 | uint32_t DIAEnumSourceFiles::getChildCount() const { |
21 | LONG Count = 0; |
22 | return (S_OK == Enumerator->get_Count(&Count)) ? Count : 0; |
23 | } |
24 | |
25 | std::unique_ptr<IPDBSourceFile> |
26 | DIAEnumSourceFiles::getChildAtIndex(uint32_t Index) const { |
27 | CComPtr<IDiaSourceFile> Item; |
28 | if (S_OK != Enumerator->Item(Index, &Item)) |
29 | return nullptr; |
30 | |
31 | return std::unique_ptr<IPDBSourceFile>(new DIASourceFile(Session, Item)); |
32 | } |
33 | |
34 | std::unique_ptr<IPDBSourceFile> DIAEnumSourceFiles::getNext() { |
35 | CComPtr<IDiaSourceFile> Item; |
36 | ULONG NumFetched = 0; |
37 | if (S_OK != Enumerator->Next(1, &Item, &NumFetched)) |
38 | return nullptr; |
39 | |
40 | return std::unique_ptr<IPDBSourceFile>(new DIASourceFile(Session, Item)); |
41 | } |
42 | |
43 | void DIAEnumSourceFiles::reset() { Enumerator->Reset(); } |
44 |