include/llvm/DebugInfo/PDB/ConcreteSymbolEnumerator.h

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