lib/DebugInfo/PDB/DIA/DIAEnumFrameData.cpp

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