llvm/lib/DWARFLinkerParallel/IndexedValuesMap.h

   1 //===- IndexedValuesMap.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_LIB_DWARFLINKERPARALLEL_INDEXEDVALUESMAP_H
  10 #define LLVM_LIB_DWARFLINKERPARALLEL_INDEXEDVALUESMAP_H
  11
  12 #include "llvm/ADT/DenseMap.h"
  13 #include "llvm/ADT/SmallVector.h"
  14 #include <cstdint>
  15 #include <utility>
  16
  17 namespace llvm {
  18 namespace dwarflinker_parallel {
  19
  20 template <typename T> class IndexedValuesMap {
  21 public:
  22   uint64_t getValueIndex(T Value) {
  23     typename ValueToIndexMapTy::iterator It = ValueToIndexMap.find(Value);
  24     if (It == ValueToIndexMap.end()) {
  25       It = ValueToIndexMap.insert(std::make_pair(Value, Values.size())).first;
  26       Values.push_back(Value);
  27     }
  28     return It->second;
  29   }
  30
  31   const SmallVector<T> &getValues() { return Values; }
  32
  33   void clear() {
  34     ValueToIndexMap.clear();
  35     Values.clear();
  36   }
  37
  38   bool empty() { return Values.empty(); }
  39
  40 protected:
  41   using ValueToIndexMapTy = DenseMap<T, uint64_t>;
  42   ValueToIndexMapTy ValueToIndexMap;
  43   SmallVector<T> Values;
  44 };
  45
  46 } // end of namespace dwarflinker_parallel
  47 } // end namespace llvm
  48
  49 #endif // LLVM_LIB_DWARFLINKERPARALLEL_INDEXEDVALUESMAP_H