1//==- DIAEnumDebugStreams.cpp - DIA Debug Stream 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/DIAEnumDebugStreams.h"
10#include "llvm/DebugInfo/PDB/DIA/DIADataStream.h"
11#include "llvm/DebugInfo/PDB/PDBSymbol.h"
12
13using namespace llvm;
14using namespace llvm::pdb;
15
16DIAEnumDebugStreams::DIAEnumDebugStreams(
17    CComPtr<IDiaEnumDebugStreams> DiaEnumerator)
18    : Enumerator(DiaEnumerator) {}
19
20uint32_t DIAEnumDebugStreams::getChildCount() const {
21  LONG Count = 0;
22  return (S_OK == Enumerator->get_Count(&Count)) ? Count : 0;
23}
24
25std::unique_ptr<IPDBDataStream>
26DIAEnumDebugStreams::getChildAtIndex(uint32_t Index) const {
27  CComPtr<IDiaEnumDebugStreamData> Item;
28  VARIANT VarIndex;
29  VarIndex.vt = VT_I4;
30  VarIndex.lVal = Index;
31  if (S_OK != Enumerator->Item(VarIndex, &Item))
32    return nullptr;
33
34  return std::unique_ptr<IPDBDataStream>(new DIADataStream(Item));
35}
36
37std::unique_ptr<IPDBDataStream> DIAEnumDebugStreams::getNext() {
38  CComPtr<IDiaEnumDebugStreamData> Item;
39  ULONG NumFetched = 0;
40  if (S_OK != Enumerator->Next(1, &Item, &NumFetched))
41    return nullptr;
42
43  return std::unique_ptr<IPDBDataStream>(new DIADataStream(Item));
44}
45
46void DIAEnumDebugStreams::reset() { Enumerator->Reset(); }
47