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 // Duplicates are not inserted. The expansion refers to different 28 // MachineOperands using the same virtual register. 29 Expansions.try_emplace(Name, Expansion); 30 } 31 redeclare(StringRef Name,StringRef Expansion)32 void redeclare(StringRef Name, StringRef Expansion) { 33 Expansions[Name] = Expansion; 34 } 35 lookup(StringRef Variable)36 std::string lookup(StringRef Variable) const { 37 return Expansions.lookup(Variable); 38 } 39 begin()40 const_iterator begin() const { return Expansions.begin(); } end()41 const_iterator end() const { return Expansions.end(); } find(StringRef Variable)42 const_iterator find(StringRef Variable) const { 43 return Expansions.find(Variable); 44 } 45 }; 46 } // end namespace llvm 47 #endif // ifndef LLVM_UTILS_TABLEGEN_CODEEXPANSIONS_H 48