Line data    Source code

       1             : //===- ConcreteSymbolEnumerator.h -------------------------------*- 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             : #ifndef LLVM_DEBUGINFO_PDB_CONCRETESYMBOLENUMERATOR_H
      11             : #define LLVM_DEBUGINFO_PDB_CONCRETESYMBOLENUMERATOR_H
      12             : 
      13             : #include "llvm/DebugInfo/PDB/IPDBEnumChildren.h"
      14             : #include "llvm/DebugInfo/PDB/PDBTypes.h"
      15             : #include "llvm/Support/Casting.h"
      16             : #include <algorithm>
      17             : #include <cstdint>
      18             : #include <memory>
      19             : 
      20             : namespace llvm {
      21             : namespace pdb {
      22             : 
      23             : template <typename ChildType>
      24             : class ConcreteSymbolEnumerator : public IPDBEnumChildren<ChildType> {
      25             : public:
      26         116 :   ConcreteSymbolEnumerator(std::unique_ptr<IPDBEnumSymbols> SymbolEnumerator)
      27         116 :       : Enumerator(std::move(SymbolEnumerator)) {}
      28             : 
      29         232 :   ~ConcreteSymbolEnumerator() override = default;
      30             : 
      31          82 :   uint32_t getChildCount() const override {
      32          82 :     return Enumerator->getChildCount();
      33             :   }
      34           0 : 
      35           2 :   std::unique_ptr<ChildType> getChildAtIndex(uint32_t Index) const override {
      36           2 :     std::unique_ptr<PDBSymbol> Child = Enumerator->getChildAtIndex(Index);
      37           3 :     return unique_dyn_cast_or_null<ChildType>(Child);
      38           1 :   }
      39             : 
      40         234 :   std::unique_ptr<ChildType> getNext() override {
      41         234 :     return unique_dyn_cast_or_null<ChildType>(Enumerator->getNext());
      42           0 :   }
      43           1 : 
      44           3 :   void reset() override { Enumerator->reset(); }
      45           0 : 
      46           1 : private:
      47           1 : 
      48           0 :   std::unique_ptr<IPDBEnumSymbols> Enumerator;
      49           1 : };
      50           1 : 
      51           0 : } // end namespace pdb
      52           1 : } // end namespace llvm
      53           1 : 
      54           0 : #endif // LLVM_DEBUGINFO_PDB_CONCRETESYMBOLENUMERATOR_H