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