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
13using namespace llvm;
14using namespace llvm::pdb;
15
16DIAEnumSourceFiles::DIAEnumSourceFiles(
17    const DIASession &PDBSession, CComPtr<IDiaEnumSourceFiles> DiaEnumerator)
18    : Session(PDBSession), Enumerator(DiaEnumerator) {}
19
20uint32_t DIAEnumSourceFiles::getChildCount() const {
21  LONG Count = 0;
22  return (S_OK == Enumerator->get_Count(&Count)) ? Count : 0;
23}
24
25std::unique_ptr<IPDBSourceFile>
26DIAEnumSourceFiles::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
34std::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
43void DIAEnumSourceFiles::reset() { Enumerator->Reset(); }
44