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