1 //===- CodeExpansions.h - Record expansions for CodeExpander --------------===//
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 /// \file Record the expansions to use in a CodeExpander.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "llvm/ADT/StringMap.h"
14 
15 #ifndef LLVM_UTILS_TABLEGEN_CODEEXPANSIONS_H
16 #define LLVM_UTILS_TABLEGEN_CODEEXPANSIONS_H
17 namespace llvm {
18 class CodeExpansions {
19 public:
20   using const_iterator = StringMap<std::string>::const_iterator;
21 
22 protected:
23   StringMap<std::string> Expansions;
24 
25 public:
declare(StringRef Name,StringRef Expansion)26   void declare(StringRef Name, StringRef Expansion) {
27     bool Inserted = Expansions.try_emplace(Name, Expansion).second;
28     assert(Inserted && "Declared variable twice");
29     (void)Inserted;
30   }
31 
lookup(StringRef Variable)32   std::string lookup(StringRef Variable) const {
33     return Expansions.lookup(Variable);
34   }
35 
begin()36   const_iterator begin() const { return Expansions.begin(); }
end()37   const_iterator end() const { return Expansions.end(); }
find(StringRef Variable)38   const_iterator find(StringRef Variable) const {
39     return Expansions.find(Variable);
40   }
41 };
42 } // end namespace llvm
43 #endif // ifndef LLVM_UTILS_TABLEGEN_CODEEXPANSIONS_H
44