1 //==- DIAEnumSourceFiles.cpp - DIA Source File 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/PDBSymbol.h"
11 #include "llvm/DebugInfo/PDB/DIA/DIAEnumSourceFiles.h"
12 #include "llvm/DebugInfo/PDB/DIA/DIASourceFile.h"
13 
14 using namespace llvm;
15 
DIAEnumSourceFiles(const DIASession & PDBSession,CComPtr<IDiaEnumSourceFiles> DiaEnumerator)16 DIAEnumSourceFiles::DIAEnumSourceFiles(
17     const DIASession &PDBSession, CComPtr<IDiaEnumSourceFiles> DiaEnumerator)
18     : Session(PDBSession), Enumerator(DiaEnumerator) {}
19 
getChildCount() const20 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>
getChildAtIndex(uint32_t Index) const26 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 
getNext()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 
reset()43 void DIAEnumSourceFiles::reset() { Enumerator->Reset(); }
44 
clone() const45 DIAEnumSourceFiles *DIAEnumSourceFiles::clone() const {
46   CComPtr<IDiaEnumSourceFiles> EnumeratorClone;
47   if (S_OK != Enumerator->Clone(&EnumeratorClone))
48     return nullptr;
49   return new DIAEnumSourceFiles(Session, EnumeratorClone);
50 }
51