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 |
54 |