184c287e3SPeter Collingbourne //===- Record.cpp - Record implementation ---------------------------------===// 284c287e3SPeter Collingbourne // 384c287e3SPeter Collingbourne // The LLVM Compiler Infrastructure 484c287e3SPeter Collingbourne // 584c287e3SPeter Collingbourne // This file is distributed under the University of Illinois Open Source 684c287e3SPeter Collingbourne // License. See LICENSE.TXT for details. 784c287e3SPeter Collingbourne // 884c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 984c287e3SPeter Collingbourne // 1084c287e3SPeter Collingbourne // Implement the tablegen record classes. 1184c287e3SPeter Collingbourne // 1284c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 1384c287e3SPeter Collingbourne 1484c287e3SPeter Collingbourne #include "llvm/TableGen/Record.h" 1584c287e3SPeter Collingbourne #include "llvm/ADT/DenseMap.h" 1684c287e3SPeter Collingbourne #include "llvm/ADT/FoldingSet.h" 1708a47fd7SChandler Carruth #include "llvm/ADT/Hashing.h" 1884c287e3SPeter Collingbourne #include "llvm/ADT/STLExtras.h" 19ed0881b2SChandler Carruth #include "llvm/ADT/SmallVector.h" 2084c287e3SPeter Collingbourne #include "llvm/ADT/StringExtras.h" 2184c287e3SPeter Collingbourne #include "llvm/ADT/StringMap.h" 22ed0881b2SChandler Carruth #include "llvm/Support/DataTypes.h" 23ed0881b2SChandler Carruth #include "llvm/Support/ErrorHandling.h" 24ed0881b2SChandler Carruth #include "llvm/Support/Format.h" 25ed0881b2SChandler Carruth #include "llvm/TableGen/Error.h" 2684c287e3SPeter Collingbourne 2784c287e3SPeter Collingbourne using namespace llvm; 2884c287e3SPeter Collingbourne 2984c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 3084c287e3SPeter Collingbourne // std::string wrapper for DenseMap purposes 3184c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 3284c287e3SPeter Collingbourne 3308a47fd7SChandler Carruth namespace llvm { 3408a47fd7SChandler Carruth 3584c287e3SPeter Collingbourne /// TableGenStringKey - This is a wrapper for std::string suitable for 3684c287e3SPeter Collingbourne /// using as a key to a DenseMap. Because there isn't a particularly 3784c287e3SPeter Collingbourne /// good way to indicate tombstone or empty keys for strings, we want 3884c287e3SPeter Collingbourne /// to wrap std::string to indicate that this is a "special" string 3984c287e3SPeter Collingbourne /// not expected to take on certain values (those of the tombstone and 4084c287e3SPeter Collingbourne /// empty keys). This makes things a little safer as it clarifies 4184c287e3SPeter Collingbourne /// that DenseMap is really not appropriate for general strings. 4284c287e3SPeter Collingbourne 4384c287e3SPeter Collingbourne class TableGenStringKey { 4484c287e3SPeter Collingbourne public: 4584c287e3SPeter Collingbourne TableGenStringKey(const std::string &str) : data(str) {} 4684c287e3SPeter Collingbourne TableGenStringKey(const char *str) : data(str) {} 4784c287e3SPeter Collingbourne 4884c287e3SPeter Collingbourne const std::string &str() const { return data; } 4984c287e3SPeter Collingbourne 5008a47fd7SChandler Carruth friend hash_code hash_value(const TableGenStringKey &Value) { 5108a47fd7SChandler Carruth using llvm::hash_value; 5208a47fd7SChandler Carruth return hash_value(Value.str()); 5308a47fd7SChandler Carruth } 5484c287e3SPeter Collingbourne private: 5584c287e3SPeter Collingbourne std::string data; 5684c287e3SPeter Collingbourne }; 5784c287e3SPeter Collingbourne 5884c287e3SPeter Collingbourne /// Specialize DenseMapInfo for TableGenStringKey. 5984c287e3SPeter Collingbourne template<> struct DenseMapInfo<TableGenStringKey> { 6084c287e3SPeter Collingbourne static inline TableGenStringKey getEmptyKey() { 6184c287e3SPeter Collingbourne TableGenStringKey Empty("<<<EMPTY KEY>>>"); 6284c287e3SPeter Collingbourne return Empty; 6384c287e3SPeter Collingbourne } 6484c287e3SPeter Collingbourne static inline TableGenStringKey getTombstoneKey() { 6584c287e3SPeter Collingbourne TableGenStringKey Tombstone("<<<TOMBSTONE KEY>>>"); 6684c287e3SPeter Collingbourne return Tombstone; 6784c287e3SPeter Collingbourne } 6884c287e3SPeter Collingbourne static unsigned getHashValue(const TableGenStringKey& Val) { 6908a47fd7SChandler Carruth using llvm::hash_value; 7008a47fd7SChandler Carruth return hash_value(Val); 7184c287e3SPeter Collingbourne } 7284c287e3SPeter Collingbourne static bool isEqual(const TableGenStringKey& LHS, 7384c287e3SPeter Collingbourne const TableGenStringKey& RHS) { 7484c287e3SPeter Collingbourne return LHS.str() == RHS.str(); 7584c287e3SPeter Collingbourne } 7684c287e3SPeter Collingbourne }; 7784c287e3SPeter Collingbourne 7808a47fd7SChandler Carruth } // namespace llvm 7984c287e3SPeter Collingbourne 8084c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 8184c287e3SPeter Collingbourne // Type implementations 8284c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 8384c287e3SPeter Collingbourne 8484c287e3SPeter Collingbourne BitRecTy BitRecTy::Shared; 8584c287e3SPeter Collingbourne IntRecTy IntRecTy::Shared; 8684c287e3SPeter Collingbourne StringRecTy StringRecTy::Shared; 8784c287e3SPeter Collingbourne DagRecTy DagRecTy::Shared; 8884c287e3SPeter Collingbourne 89a379b181SDavid Blaikie void RecTy::anchor() { } 9084c287e3SPeter Collingbourne void RecTy::dump() const { print(errs()); } 9184c287e3SPeter Collingbourne 9284c287e3SPeter Collingbourne ListRecTy *RecTy::getListTy() { 9384c287e3SPeter Collingbourne if (!ListTy) 9417b3a378SCraig Topper ListTy.reset(new ListRecTy(this)); 9517b3a378SCraig Topper return ListTy.get(); 9684c287e3SPeter Collingbourne } 9784c287e3SPeter Collingbourne 9893be7d59SSean Silva bool RecTy::baseClassOf(const RecTy *RHS) const { 9993be7d59SSean Silva assert (RHS && "NULL pointer"); 10093be7d59SSean Silva return Kind == RHS->getRecTyKind(); 10193be7d59SSean Silva } 10293be7d59SSean Silva 10384c287e3SPeter Collingbourne Init *BitRecTy::convertValue(BitsInit *BI) { 104011817a0SCraig Topper if (BI->getNumBits() != 1) return nullptr; // Only accept if just one bit! 10584c287e3SPeter Collingbourne return BI->getBit(0); 10684c287e3SPeter Collingbourne } 10784c287e3SPeter Collingbourne 10884c287e3SPeter Collingbourne Init *BitRecTy::convertValue(IntInit *II) { 10984c287e3SPeter Collingbourne int64_t Val = II->getValue(); 110011817a0SCraig Topper if (Val != 0 && Val != 1) return nullptr; // Only accept 0 or 1 for a bit! 11184c287e3SPeter Collingbourne 11284c287e3SPeter Collingbourne return BitInit::get(Val != 0); 11384c287e3SPeter Collingbourne } 11484c287e3SPeter Collingbourne 11584c287e3SPeter Collingbourne Init *BitRecTy::convertValue(TypedInit *VI) { 116026f8333SMichael Liao RecTy *Ty = VI->getType(); 11799ad2a3bSPete Cooper if (isa<BitRecTy>(Ty)) 11884c287e3SPeter Collingbourne return VI; // Accept variable if it is already of bit type! 11999ad2a3bSPete Cooper if (auto *BitsTy = dyn_cast<BitsRecTy>(Ty)) 12099ad2a3bSPete Cooper // Accept only bits<1> expression. 12199ad2a3bSPete Cooper return BitsTy->getNumBits() == 1 ? VI : nullptr; 12294891ddfSPete Cooper // Ternary !if can be converted to bit, but only if both sides are 12394891ddfSPete Cooper // convertible to a bit. 12494891ddfSPete Cooper if (TernOpInit *TOI = dyn_cast<TernOpInit>(VI)) { 12594891ddfSPete Cooper if (TOI->getOpcode() != TernOpInit::TernaryOp::IF) 12694891ddfSPete Cooper return nullptr; 12794891ddfSPete Cooper if (!TOI->getMHS()->convertInitializerTo(BitRecTy::get()) || 12894891ddfSPete Cooper !TOI->getRHS()->convertInitializerTo(BitRecTy::get())) 12994891ddfSPete Cooper return nullptr; 13094891ddfSPete Cooper return TOI; 13194891ddfSPete Cooper } 132011817a0SCraig Topper return nullptr; 13384c287e3SPeter Collingbourne } 13484c287e3SPeter Collingbourne 13593be7d59SSean Silva bool BitRecTy::baseClassOf(const RecTy *RHS) const{ 136fe0cdf98SCraig Topper if(RecTy::baseClassOf(RHS) || RHS->getRecTyKind() == IntRecTyKind) 13793be7d59SSean Silva return true; 13893be7d59SSean Silva if(const BitsRecTy *BitsTy = dyn_cast<BitsRecTy>(RHS)) 13993be7d59SSean Silva return BitsTy->getNumBits() == 1; 14093be7d59SSean Silva return false; 14193be7d59SSean Silva } 14293be7d59SSean Silva 14384c287e3SPeter Collingbourne BitsRecTy *BitsRecTy::get(unsigned Sz) { 14417b3a378SCraig Topper static std::vector<std::unique_ptr<BitsRecTy>> Shared; 14584c287e3SPeter Collingbourne if (Sz >= Shared.size()) 14684c287e3SPeter Collingbourne Shared.resize(Sz + 1); 14717b3a378SCraig Topper std::unique_ptr<BitsRecTy> &Ty = Shared[Sz]; 14884c287e3SPeter Collingbourne if (!Ty) 14917b3a378SCraig Topper Ty.reset(new BitsRecTy(Sz)); 15017b3a378SCraig Topper return Ty.get(); 15184c287e3SPeter Collingbourne } 15284c287e3SPeter Collingbourne 15384c287e3SPeter Collingbourne std::string BitsRecTy::getAsString() const { 15484c287e3SPeter Collingbourne return "bits<" + utostr(Size) + ">"; 15584c287e3SPeter Collingbourne } 15684c287e3SPeter Collingbourne 15784c287e3SPeter Collingbourne Init *BitsRecTy::convertValue(UnsetInit *UI) { 15884c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Size); 15984c287e3SPeter Collingbourne 16084c287e3SPeter Collingbourne for (unsigned i = 0; i != Size; ++i) 16184c287e3SPeter Collingbourne NewBits[i] = UnsetInit::get(); 16284c287e3SPeter Collingbourne 16384c287e3SPeter Collingbourne return BitsInit::get(NewBits); 16484c287e3SPeter Collingbourne } 16584c287e3SPeter Collingbourne 16684c287e3SPeter Collingbourne Init *BitsRecTy::convertValue(BitInit *UI) { 167011817a0SCraig Topper if (Size != 1) return nullptr; // Can only convert single bit. 16884c287e3SPeter Collingbourne return BitsInit::get(UI); 16984c287e3SPeter Collingbourne } 17084c287e3SPeter Collingbourne 17184c287e3SPeter Collingbourne /// canFitInBitfield - Return true if the number of bits is large enough to hold 17284c287e3SPeter Collingbourne /// the integer value. 17384c287e3SPeter Collingbourne static bool canFitInBitfield(int64_t Value, unsigned NumBits) { 17484c287e3SPeter Collingbourne // For example, with NumBits == 4, we permit Values from [-7 .. 15]. 17584c287e3SPeter Collingbourne return (NumBits >= sizeof(Value) * 8) || 17684c287e3SPeter Collingbourne (Value >> NumBits == 0) || (Value >> (NumBits-1) == -1); 17784c287e3SPeter Collingbourne } 17884c287e3SPeter Collingbourne 17984c287e3SPeter Collingbourne /// convertValue from Int initializer to bits type: Split the integer up into the 18084c287e3SPeter Collingbourne /// appropriate bits. 18184c287e3SPeter Collingbourne /// 18284c287e3SPeter Collingbourne Init *BitsRecTy::convertValue(IntInit *II) { 18384c287e3SPeter Collingbourne int64_t Value = II->getValue(); 18484c287e3SPeter Collingbourne // Make sure this bitfield is large enough to hold the integer value. 18584c287e3SPeter Collingbourne if (!canFitInBitfield(Value, Size)) 186011817a0SCraig Topper return nullptr; 18784c287e3SPeter Collingbourne 18884c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Size); 18984c287e3SPeter Collingbourne 19084c287e3SPeter Collingbourne for (unsigned i = 0; i != Size; ++i) 19184c287e3SPeter Collingbourne NewBits[i] = BitInit::get(Value & (1LL << i)); 19284c287e3SPeter Collingbourne 19384c287e3SPeter Collingbourne return BitsInit::get(NewBits); 19484c287e3SPeter Collingbourne } 19584c287e3SPeter Collingbourne 19684c287e3SPeter Collingbourne Init *BitsRecTy::convertValue(BitsInit *BI) { 19784c287e3SPeter Collingbourne // If the number of bits is right, return it. Otherwise we need to expand or 19884c287e3SPeter Collingbourne // truncate. 19984c287e3SPeter Collingbourne if (BI->getNumBits() == Size) return BI; 200011817a0SCraig Topper return nullptr; 20184c287e3SPeter Collingbourne } 20284c287e3SPeter Collingbourne 20384c287e3SPeter Collingbourne Init *BitsRecTy::convertValue(TypedInit *VI) { 204a475e433SSean Silva if (Size == 1 && isa<BitRecTy>(VI->getType())) 205026f8333SMichael Liao return BitsInit::get(VI); 206026f8333SMichael Liao 207026f8333SMichael Liao if (VI->getType()->typeIsConvertibleTo(this)) { 20884c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Size); 20984c287e3SPeter Collingbourne 21084c287e3SPeter Collingbourne for (unsigned i = 0; i != Size; ++i) 21184c287e3SPeter Collingbourne NewBits[i] = VarBitInit::get(VI, i); 21284c287e3SPeter Collingbourne return BitsInit::get(NewBits); 21384c287e3SPeter Collingbourne } 21484c287e3SPeter Collingbourne 215011817a0SCraig Topper return nullptr; 21684c287e3SPeter Collingbourne } 21784c287e3SPeter Collingbourne 21893be7d59SSean Silva bool BitsRecTy::baseClassOf(const RecTy *RHS) const{ 21993be7d59SSean Silva if (RecTy::baseClassOf(RHS)) //argument and the receiver are the same type 22093be7d59SSean Silva return cast<BitsRecTy>(RHS)->Size == Size; 22193be7d59SSean Silva RecTyKind kind = RHS->getRecTyKind(); 22293be7d59SSean Silva return (kind == BitRecTyKind && Size == 1) || (kind == IntRecTyKind); 22393be7d59SSean Silva } 22493be7d59SSean Silva 22584c287e3SPeter Collingbourne Init *IntRecTy::convertValue(BitInit *BI) { 22684c287e3SPeter Collingbourne return IntInit::get(BI->getValue()); 22784c287e3SPeter Collingbourne } 22884c287e3SPeter Collingbourne 22984c287e3SPeter Collingbourne Init *IntRecTy::convertValue(BitsInit *BI) { 23084c287e3SPeter Collingbourne int64_t Result = 0; 23184c287e3SPeter Collingbourne for (unsigned i = 0, e = BI->getNumBits(); i != e; ++i) 232a9642b4eSCraig Topper if (BitInit *Bit = dyn_cast<BitInit>(BI->getBit(i))) 23384c287e3SPeter Collingbourne Result |= Bit->getValue() << i; 234a9642b4eSCraig Topper else 235011817a0SCraig Topper return nullptr; 23684c287e3SPeter Collingbourne return IntInit::get(Result); 23784c287e3SPeter Collingbourne } 23884c287e3SPeter Collingbourne 23984c287e3SPeter Collingbourne Init *IntRecTy::convertValue(TypedInit *TI) { 24084c287e3SPeter Collingbourne if (TI->getType()->typeIsConvertibleTo(this)) 24184c287e3SPeter Collingbourne return TI; // Accept variable if already of the right type! 242011817a0SCraig Topper return nullptr; 24384c287e3SPeter Collingbourne } 24484c287e3SPeter Collingbourne 24593be7d59SSean Silva bool IntRecTy::baseClassOf(const RecTy *RHS) const{ 24693be7d59SSean Silva RecTyKind kind = RHS->getRecTyKind(); 24793be7d59SSean Silva return kind==BitRecTyKind || kind==BitsRecTyKind || kind==IntRecTyKind; 24893be7d59SSean Silva } 24993be7d59SSean Silva 25084c287e3SPeter Collingbourne Init *StringRecTy::convertValue(UnOpInit *BO) { 25184c287e3SPeter Collingbourne if (BO->getOpcode() == UnOpInit::CAST) { 25284c287e3SPeter Collingbourne Init *L = BO->getOperand()->convertInitializerTo(this); 253011817a0SCraig Topper if (!L) return nullptr; 25484c287e3SPeter Collingbourne if (L != BO->getOperand()) 255b15012bcSCraig Topper return UnOpInit::get(UnOpInit::CAST, L, StringRecTy::get()); 25684c287e3SPeter Collingbourne return BO; 25784c287e3SPeter Collingbourne } 25884c287e3SPeter Collingbourne 25984c287e3SPeter Collingbourne return convertValue((TypedInit*)BO); 26084c287e3SPeter Collingbourne } 26184c287e3SPeter Collingbourne 26284c287e3SPeter Collingbourne Init *StringRecTy::convertValue(BinOpInit *BO) { 26384c287e3SPeter Collingbourne if (BO->getOpcode() == BinOpInit::STRCONCAT) { 26484c287e3SPeter Collingbourne Init *L = BO->getLHS()->convertInitializerTo(this); 26584c287e3SPeter Collingbourne Init *R = BO->getRHS()->convertInitializerTo(this); 266011817a0SCraig Topper if (!L || !R) return nullptr; 26784c287e3SPeter Collingbourne if (L != BO->getLHS() || R != BO->getRHS()) 268b15012bcSCraig Topper return BinOpInit::get(BinOpInit::STRCONCAT, L, R, StringRecTy::get()); 26984c287e3SPeter Collingbourne return BO; 27084c287e3SPeter Collingbourne } 27184c287e3SPeter Collingbourne 27284c287e3SPeter Collingbourne return convertValue((TypedInit*)BO); 27384c287e3SPeter Collingbourne } 27484c287e3SPeter Collingbourne 27584c287e3SPeter Collingbourne 27684c287e3SPeter Collingbourne Init *StringRecTy::convertValue(TypedInit *TI) { 277a475e433SSean Silva if (isa<StringRecTy>(TI->getType())) 27884c287e3SPeter Collingbourne return TI; // Accept variable if already of the right type! 279011817a0SCraig Topper return nullptr; 28084c287e3SPeter Collingbourne } 28184c287e3SPeter Collingbourne 28284c287e3SPeter Collingbourne std::string ListRecTy::getAsString() const { 28384c287e3SPeter Collingbourne return "list<" + Ty->getAsString() + ">"; 28484c287e3SPeter Collingbourne } 28584c287e3SPeter Collingbourne 28684c287e3SPeter Collingbourne Init *ListRecTy::convertValue(ListInit *LI) { 28784c287e3SPeter Collingbourne std::vector<Init*> Elements; 28884c287e3SPeter Collingbourne 28984c287e3SPeter Collingbourne // Verify that all of the elements of the list are subclasses of the 29084c287e3SPeter Collingbourne // appropriate class! 29184c287e3SPeter Collingbourne for (unsigned i = 0, e = LI->getSize(); i != e; ++i) 29284c287e3SPeter Collingbourne if (Init *CI = LI->getElement(i)->convertInitializerTo(Ty)) 29384c287e3SPeter Collingbourne Elements.push_back(CI); 29484c287e3SPeter Collingbourne else 295011817a0SCraig Topper return nullptr; 29684c287e3SPeter Collingbourne 297a475e433SSean Silva if (!isa<ListRecTy>(LI->getType())) 298011817a0SCraig Topper return nullptr; 29984c287e3SPeter Collingbourne 30084c287e3SPeter Collingbourne return ListInit::get(Elements, this); 30184c287e3SPeter Collingbourne } 30284c287e3SPeter Collingbourne 30384c287e3SPeter Collingbourne Init *ListRecTy::convertValue(TypedInit *TI) { 30484c287e3SPeter Collingbourne // Ensure that TI is compatible with our class. 30598c61711SSean Silva if (ListRecTy *LRT = dyn_cast<ListRecTy>(TI->getType())) 30684c287e3SPeter Collingbourne if (LRT->getElementType()->typeIsConvertibleTo(getElementType())) 30784c287e3SPeter Collingbourne return TI; 308011817a0SCraig Topper return nullptr; 30984c287e3SPeter Collingbourne } 31084c287e3SPeter Collingbourne 31193be7d59SSean Silva bool ListRecTy::baseClassOf(const RecTy *RHS) const{ 31293be7d59SSean Silva if(const ListRecTy* ListTy = dyn_cast<ListRecTy>(RHS)) 31393be7d59SSean Silva return ListTy->getElementType()->typeIsConvertibleTo(Ty); 31493be7d59SSean Silva return false; 31593be7d59SSean Silva } 31693be7d59SSean Silva 31784c287e3SPeter Collingbourne Init *DagRecTy::convertValue(TypedInit *TI) { 31884c287e3SPeter Collingbourne if (TI->getType()->typeIsConvertibleTo(this)) 31984c287e3SPeter Collingbourne return TI; 320011817a0SCraig Topper return nullptr; 32184c287e3SPeter Collingbourne } 32284c287e3SPeter Collingbourne 32384c287e3SPeter Collingbourne Init *DagRecTy::convertValue(UnOpInit *BO) { 32484c287e3SPeter Collingbourne if (BO->getOpcode() == UnOpInit::CAST) { 32584c287e3SPeter Collingbourne Init *L = BO->getOperand()->convertInitializerTo(this); 326011817a0SCraig Topper if (!L) return nullptr; 32784c287e3SPeter Collingbourne if (L != BO->getOperand()) 32884c287e3SPeter Collingbourne return UnOpInit::get(UnOpInit::CAST, L, new DagRecTy); 32984c287e3SPeter Collingbourne return BO; 33084c287e3SPeter Collingbourne } 331011817a0SCraig Topper return nullptr; 33284c287e3SPeter Collingbourne } 33384c287e3SPeter Collingbourne 33484c287e3SPeter Collingbourne Init *DagRecTy::convertValue(BinOpInit *BO) { 33584c287e3SPeter Collingbourne if (BO->getOpcode() == BinOpInit::CONCAT) { 33684c287e3SPeter Collingbourne Init *L = BO->getLHS()->convertInitializerTo(this); 33784c287e3SPeter Collingbourne Init *R = BO->getRHS()->convertInitializerTo(this); 338011817a0SCraig Topper if (!L || !R) return nullptr; 33984c287e3SPeter Collingbourne if (L != BO->getLHS() || R != BO->getRHS()) 34084c287e3SPeter Collingbourne return BinOpInit::get(BinOpInit::CONCAT, L, R, new DagRecTy); 34184c287e3SPeter Collingbourne return BO; 34284c287e3SPeter Collingbourne } 343011817a0SCraig Topper return nullptr; 34484c287e3SPeter Collingbourne } 34584c287e3SPeter Collingbourne 34684c287e3SPeter Collingbourne RecordRecTy *RecordRecTy::get(Record *R) { 347b534eabfSCraig Topper return dyn_cast<RecordRecTy>(R->getDefInit()->getType()); 34884c287e3SPeter Collingbourne } 34984c287e3SPeter Collingbourne 35084c287e3SPeter Collingbourne std::string RecordRecTy::getAsString() const { 35184c287e3SPeter Collingbourne return Rec->getName(); 35284c287e3SPeter Collingbourne } 35384c287e3SPeter Collingbourne 35484c287e3SPeter Collingbourne Init *RecordRecTy::convertValue(DefInit *DI) { 35584c287e3SPeter Collingbourne // Ensure that DI is a subclass of Rec. 35684c287e3SPeter Collingbourne if (!DI->getDef()->isSubClassOf(Rec)) 357011817a0SCraig Topper return nullptr; 35884c287e3SPeter Collingbourne return DI; 35984c287e3SPeter Collingbourne } 36084c287e3SPeter Collingbourne 36184c287e3SPeter Collingbourne Init *RecordRecTy::convertValue(TypedInit *TI) { 36284c287e3SPeter Collingbourne // Ensure that TI is compatible with Rec. 36398c61711SSean Silva if (RecordRecTy *RRT = dyn_cast<RecordRecTy>(TI->getType())) 36484c287e3SPeter Collingbourne if (RRT->getRecord()->isSubClassOf(getRecord()) || 36584c287e3SPeter Collingbourne RRT->getRecord() == getRecord()) 36684c287e3SPeter Collingbourne return TI; 367011817a0SCraig Topper return nullptr; 36884c287e3SPeter Collingbourne } 36984c287e3SPeter Collingbourne 37093be7d59SSean Silva bool RecordRecTy::baseClassOf(const RecTy *RHS) const{ 37193be7d59SSean Silva const RecordRecTy *RTy = dyn_cast<RecordRecTy>(RHS); 37293be7d59SSean Silva if (!RTy) 37393be7d59SSean Silva return false; 37493be7d59SSean Silva 37593be7d59SSean Silva if (Rec == RTy->getRecord() || RTy->getRecord()->isSubClassOf(Rec)) 37684c287e3SPeter Collingbourne return true; 37784c287e3SPeter Collingbourne 37884c287e3SPeter Collingbourne const std::vector<Record*> &SC = Rec->getSuperClasses(); 37984c287e3SPeter Collingbourne for (unsigned i = 0, e = SC.size(); i != e; ++i) 38093be7d59SSean Silva if (RTy->getRecord()->isSubClassOf(SC[i])) 38184c287e3SPeter Collingbourne return true; 38284c287e3SPeter Collingbourne 38384c287e3SPeter Collingbourne return false; 38484c287e3SPeter Collingbourne } 38584c287e3SPeter Collingbourne 38684c287e3SPeter Collingbourne /// resolveTypes - Find a common type that T1 and T2 convert to. 387e8005f90SCraig Topper /// Return null if no such type exists. 38884c287e3SPeter Collingbourne /// 38984c287e3SPeter Collingbourne RecTy *llvm::resolveTypes(RecTy *T1, RecTy *T2) { 3908f43c6f0SSean Silva if (T1->typeIsConvertibleTo(T2)) 3918f43c6f0SSean Silva return T2; 3928f43c6f0SSean Silva if (T2->typeIsConvertibleTo(T1)) 3938f43c6f0SSean Silva return T1; 3948f43c6f0SSean Silva 39584c287e3SPeter Collingbourne // If one is a Record type, check superclasses 39698c61711SSean Silva if (RecordRecTy *RecTy1 = dyn_cast<RecordRecTy>(T1)) { 39784c287e3SPeter Collingbourne // See if T2 inherits from a type T1 also inherits from 3981f429e49SCraig Topper for (Record *SuperRec1 : RecTy1->getRecord()->getSuperClasses()) { 3991f429e49SCraig Topper RecordRecTy *SuperRecTy1 = RecordRecTy::get(SuperRec1); 40084c287e3SPeter Collingbourne RecTy *NewType1 = resolveTypes(SuperRecTy1, T2); 401d0599130SCraig Topper if (NewType1) 40284c287e3SPeter Collingbourne return NewType1; 40384c287e3SPeter Collingbourne } 40484c287e3SPeter Collingbourne } 40598c61711SSean Silva if (RecordRecTy *RecTy2 = dyn_cast<RecordRecTy>(T2)) { 40684c287e3SPeter Collingbourne // See if T1 inherits from a type T2 also inherits from 4071f429e49SCraig Topper for (Record *SuperRec2 : RecTy2->getRecord()->getSuperClasses()) { 4081f429e49SCraig Topper RecordRecTy *SuperRecTy2 = RecordRecTy::get(SuperRec2); 40984c287e3SPeter Collingbourne RecTy *NewType2 = resolveTypes(T1, SuperRecTy2); 410d0599130SCraig Topper if (NewType2) 41184c287e3SPeter Collingbourne return NewType2; 41284c287e3SPeter Collingbourne } 41384c287e3SPeter Collingbourne } 414011817a0SCraig Topper return nullptr; 41584c287e3SPeter Collingbourne } 41684c287e3SPeter Collingbourne 41784c287e3SPeter Collingbourne 41884c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 41984c287e3SPeter Collingbourne // Initializer implementations 42084c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 42184c287e3SPeter Collingbourne 422a379b181SDavid Blaikie void Init::anchor() { } 42384c287e3SPeter Collingbourne void Init::dump() const { return print(errs()); } 42484c287e3SPeter Collingbourne 425a379b181SDavid Blaikie void UnsetInit::anchor() { } 426a379b181SDavid Blaikie 42784c287e3SPeter Collingbourne UnsetInit *UnsetInit::get() { 42884c287e3SPeter Collingbourne static UnsetInit TheInit; 42984c287e3SPeter Collingbourne return &TheInit; 43084c287e3SPeter Collingbourne } 43184c287e3SPeter Collingbourne 432a379b181SDavid Blaikie void BitInit::anchor() { } 433a379b181SDavid Blaikie 43484c287e3SPeter Collingbourne BitInit *BitInit::get(bool V) { 43584c287e3SPeter Collingbourne static BitInit True(true); 43684c287e3SPeter Collingbourne static BitInit False(false); 43784c287e3SPeter Collingbourne 43884c287e3SPeter Collingbourne return V ? &True : &False; 43984c287e3SPeter Collingbourne } 44084c287e3SPeter Collingbourne 44184c287e3SPeter Collingbourne static void 44284c287e3SPeter Collingbourne ProfileBitsInit(FoldingSetNodeID &ID, ArrayRef<Init *> Range) { 44384c287e3SPeter Collingbourne ID.AddInteger(Range.size()); 44484c287e3SPeter Collingbourne 445119998dbSCraig Topper for (Init *I : Range) 446119998dbSCraig Topper ID.AddPointer(I); 44784c287e3SPeter Collingbourne } 44884c287e3SPeter Collingbourne 44984c287e3SPeter Collingbourne BitsInit *BitsInit::get(ArrayRef<Init *> Range) { 45017b3a378SCraig Topper static FoldingSet<BitsInit> ThePool; 45117b3a378SCraig Topper static std::vector<std::unique_ptr<BitsInit>> TheActualPool; 45284c287e3SPeter Collingbourne 45384c287e3SPeter Collingbourne FoldingSetNodeID ID; 45484c287e3SPeter Collingbourne ProfileBitsInit(ID, Range); 45584c287e3SPeter Collingbourne 456011817a0SCraig Topper void *IP = nullptr; 45784c287e3SPeter Collingbourne if (BitsInit *I = ThePool.FindNodeOrInsertPos(ID, IP)) 45884c287e3SPeter Collingbourne return I; 45984c287e3SPeter Collingbourne 46084c287e3SPeter Collingbourne BitsInit *I = new BitsInit(Range); 46184c287e3SPeter Collingbourne ThePool.InsertNode(I, IP); 46217b3a378SCraig Topper TheActualPool.push_back(std::unique_ptr<BitsInit>(I)); 46384c287e3SPeter Collingbourne return I; 46484c287e3SPeter Collingbourne } 46584c287e3SPeter Collingbourne 46684c287e3SPeter Collingbourne void BitsInit::Profile(FoldingSetNodeID &ID) const { 46784c287e3SPeter Collingbourne ProfileBitsInit(ID, Bits); 46884c287e3SPeter Collingbourne } 46984c287e3SPeter Collingbourne 47084c287e3SPeter Collingbourne Init * 47184c287e3SPeter Collingbourne BitsInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const { 47284c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Bits.size()); 47384c287e3SPeter Collingbourne 47484c287e3SPeter Collingbourne for (unsigned i = 0, e = Bits.size(); i != e; ++i) { 47584c287e3SPeter Collingbourne if (Bits[i] >= getNumBits()) 476011817a0SCraig Topper return nullptr; 47784c287e3SPeter Collingbourne NewBits[i] = getBit(Bits[i]); 47884c287e3SPeter Collingbourne } 47984c287e3SPeter Collingbourne return BitsInit::get(NewBits); 48084c287e3SPeter Collingbourne } 48184c287e3SPeter Collingbourne 48284c287e3SPeter Collingbourne std::string BitsInit::getAsString() const { 48384c287e3SPeter Collingbourne std::string Result = "{ "; 48484c287e3SPeter Collingbourne for (unsigned i = 0, e = getNumBits(); i != e; ++i) { 48584c287e3SPeter Collingbourne if (i) Result += ", "; 48684c287e3SPeter Collingbourne if (Init *Bit = getBit(e-i-1)) 48784c287e3SPeter Collingbourne Result += Bit->getAsString(); 48884c287e3SPeter Collingbourne else 48984c287e3SPeter Collingbourne Result += "*"; 49084c287e3SPeter Collingbourne } 49184c287e3SPeter Collingbourne return Result + " }"; 49284c287e3SPeter Collingbourne } 49384c287e3SPeter Collingbourne 494026f8333SMichael Liao // Fix bit initializer to preserve the behavior that bit reference from a unset 495026f8333SMichael Liao // bits initializer will resolve into VarBitInit to keep the field name and bit 496026f8333SMichael Liao // number used in targets with fixed insn length. 497026f8333SMichael Liao static Init *fixBitInit(const RecordVal *RV, Init *Before, Init *After) { 4981bf3d1f5SCraig Topper if (RV || !isa<UnsetInit>(After)) 499026f8333SMichael Liao return After; 500026f8333SMichael Liao return Before; 501026f8333SMichael Liao } 502026f8333SMichael Liao 50384c287e3SPeter Collingbourne // resolveReferences - If there are any field references that refer to fields 50484c287e3SPeter Collingbourne // that have been filled in, we can propagate the values now. 50584c287e3SPeter Collingbourne // 50684c287e3SPeter Collingbourne Init *BitsInit::resolveReferences(Record &R, const RecordVal *RV) const { 50784c287e3SPeter Collingbourne bool Changed = false; 50884c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(getNumBits()); 50984c287e3SPeter Collingbourne 510011817a0SCraig Topper Init *CachedInit = nullptr; 511011817a0SCraig Topper Init *CachedBitVar = nullptr; 512026f8333SMichael Liao bool CachedBitVarChanged = false; 51384c287e3SPeter Collingbourne 514026f8333SMichael Liao for (unsigned i = 0, e = getNumBits(); i != e; ++i) { 515026f8333SMichael Liao Init *CurBit = Bits[i]; 516026f8333SMichael Liao Init *CurBitVar = CurBit->getBitVar(); 517026f8333SMichael Liao 51884c287e3SPeter Collingbourne NewBits[i] = CurBit; 519026f8333SMichael Liao 520026f8333SMichael Liao if (CurBitVar == CachedBitVar) { 521026f8333SMichael Liao if (CachedBitVarChanged) { 522026f8333SMichael Liao Init *Bit = CachedInit->getBit(CurBit->getBitNum()); 523026f8333SMichael Liao NewBits[i] = fixBitInit(RV, CurBit, Bit); 524026f8333SMichael Liao } 525026f8333SMichael Liao continue; 526026f8333SMichael Liao } 527026f8333SMichael Liao CachedBitVar = CurBitVar; 528026f8333SMichael Liao CachedBitVarChanged = false; 529026f8333SMichael Liao 530026f8333SMichael Liao Init *B; 531026f8333SMichael Liao do { 532026f8333SMichael Liao B = CurBitVar; 533026f8333SMichael Liao CurBitVar = CurBitVar->resolveReferences(R, RV); 534026f8333SMichael Liao CachedBitVarChanged |= B != CurBitVar; 535026f8333SMichael Liao Changed |= B != CurBitVar; 536026f8333SMichael Liao } while (B != CurBitVar); 537026f8333SMichael Liao CachedInit = CurBitVar; 538026f8333SMichael Liao 539026f8333SMichael Liao if (CachedBitVarChanged) { 540026f8333SMichael Liao Init *Bit = CurBitVar->getBit(CurBit->getBitNum()); 541026f8333SMichael Liao NewBits[i] = fixBitInit(RV, CurBit, Bit); 542026f8333SMichael Liao } 54384c287e3SPeter Collingbourne } 54484c287e3SPeter Collingbourne 54584c287e3SPeter Collingbourne if (Changed) 54684c287e3SPeter Collingbourne return BitsInit::get(NewBits); 54784c287e3SPeter Collingbourne 54884c287e3SPeter Collingbourne return const_cast<BitsInit *>(this); 54984c287e3SPeter Collingbourne } 55084c287e3SPeter Collingbourne 55184c287e3SPeter Collingbourne IntInit *IntInit::get(int64_t V) { 5520e04bee8SCraig Topper static DenseMap<int64_t, std::unique_ptr<IntInit>> ThePool; 55384c287e3SPeter Collingbourne 5540e04bee8SCraig Topper std::unique_ptr<IntInit> &I = ThePool[V]; 5550e04bee8SCraig Topper if (!I) I.reset(new IntInit(V)); 5560e04bee8SCraig Topper return I.get(); 55784c287e3SPeter Collingbourne } 55884c287e3SPeter Collingbourne 55984c287e3SPeter Collingbourne std::string IntInit::getAsString() const { 56084c287e3SPeter Collingbourne return itostr(Value); 56184c287e3SPeter Collingbourne } 56284c287e3SPeter Collingbourne 56384c287e3SPeter Collingbourne Init * 56484c287e3SPeter Collingbourne IntInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const { 56584c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Bits.size()); 56684c287e3SPeter Collingbourne 56784c287e3SPeter Collingbourne for (unsigned i = 0, e = Bits.size(); i != e; ++i) { 56884c287e3SPeter Collingbourne if (Bits[i] >= 64) 569011817a0SCraig Topper return nullptr; 57084c287e3SPeter Collingbourne 57184c287e3SPeter Collingbourne NewBits[i] = BitInit::get(Value & (INT64_C(1) << Bits[i])); 57284c287e3SPeter Collingbourne } 57384c287e3SPeter Collingbourne return BitsInit::get(NewBits); 57484c287e3SPeter Collingbourne } 57584c287e3SPeter Collingbourne 576a379b181SDavid Blaikie void StringInit::anchor() { } 577a379b181SDavid Blaikie 5789d1c5eebSJakob Stoklund Olesen StringInit *StringInit::get(StringRef V) { 5790e04bee8SCraig Topper static StringMap<std::unique_ptr<StringInit>> ThePool; 58084c287e3SPeter Collingbourne 5810e04bee8SCraig Topper std::unique_ptr<StringInit> &I = ThePool[V]; 5820e04bee8SCraig Topper if (!I) I.reset(new StringInit(V)); 5830e04bee8SCraig Topper return I.get(); 58484c287e3SPeter Collingbourne } 58584c287e3SPeter Collingbourne 58684c287e3SPeter Collingbourne static void ProfileListInit(FoldingSetNodeID &ID, 58784c287e3SPeter Collingbourne ArrayRef<Init *> Range, 58884c287e3SPeter Collingbourne RecTy *EltTy) { 58984c287e3SPeter Collingbourne ID.AddInteger(Range.size()); 59084c287e3SPeter Collingbourne ID.AddPointer(EltTy); 59184c287e3SPeter Collingbourne 592119998dbSCraig Topper for (Init *I : Range) 593119998dbSCraig Topper ID.AddPointer(I); 59484c287e3SPeter Collingbourne } 59584c287e3SPeter Collingbourne 59684c287e3SPeter Collingbourne ListInit *ListInit::get(ArrayRef<Init *> Range, RecTy *EltTy) { 59717b3a378SCraig Topper static FoldingSet<ListInit> ThePool; 5982b6b3d1cSDaniel Sanders static std::vector<std::unique_ptr<ListInit>> TheActualPool; 59984c287e3SPeter Collingbourne 60084c287e3SPeter Collingbourne FoldingSetNodeID ID; 60184c287e3SPeter Collingbourne ProfileListInit(ID, Range, EltTy); 60284c287e3SPeter Collingbourne 603011817a0SCraig Topper void *IP = nullptr; 60484c287e3SPeter Collingbourne if (ListInit *I = ThePool.FindNodeOrInsertPos(ID, IP)) 60584c287e3SPeter Collingbourne return I; 60684c287e3SPeter Collingbourne 60784c287e3SPeter Collingbourne ListInit *I = new ListInit(Range, EltTy); 60884c287e3SPeter Collingbourne ThePool.InsertNode(I, IP); 6092b6b3d1cSDaniel Sanders TheActualPool.push_back(std::unique_ptr<ListInit>(I)); 61084c287e3SPeter Collingbourne return I; 61184c287e3SPeter Collingbourne } 61284c287e3SPeter Collingbourne 61384c287e3SPeter Collingbourne void ListInit::Profile(FoldingSetNodeID &ID) const { 614ed5a9508SCraig Topper RecTy *EltTy = cast<ListRecTy>(getType())->getElementType(); 61584c287e3SPeter Collingbourne 61684c287e3SPeter Collingbourne ProfileListInit(ID, Values, EltTy); 61784c287e3SPeter Collingbourne } 61884c287e3SPeter Collingbourne 61984c287e3SPeter Collingbourne Init * 62084c287e3SPeter Collingbourne ListInit::convertInitListSlice(const std::vector<unsigned> &Elements) const { 62184c287e3SPeter Collingbourne std::vector<Init*> Vals; 62284c287e3SPeter Collingbourne for (unsigned i = 0, e = Elements.size(); i != e; ++i) { 62384c287e3SPeter Collingbourne if (Elements[i] >= getSize()) 624011817a0SCraig Topper return nullptr; 62584c287e3SPeter Collingbourne Vals.push_back(getElement(Elements[i])); 62684c287e3SPeter Collingbourne } 62784c287e3SPeter Collingbourne return ListInit::get(Vals, getType()); 62884c287e3SPeter Collingbourne } 62984c287e3SPeter Collingbourne 63084c287e3SPeter Collingbourne Record *ListInit::getElementAsRecord(unsigned i) const { 63184c287e3SPeter Collingbourne assert(i < Values.size() && "List element index out of range!"); 632fb509ed1SSean Silva DefInit *DI = dyn_cast<DefInit>(Values[i]); 633011817a0SCraig Topper if (!DI) 634635debe8SJoerg Sonnenberger PrintFatalError("Expected record in list!"); 63584c287e3SPeter Collingbourne return DI->getDef(); 63684c287e3SPeter Collingbourne } 63784c287e3SPeter Collingbourne 63884c287e3SPeter Collingbourne Init *ListInit::resolveReferences(Record &R, const RecordVal *RV) const { 63984c287e3SPeter Collingbourne std::vector<Init*> Resolved; 64084c287e3SPeter Collingbourne Resolved.reserve(getSize()); 64184c287e3SPeter Collingbourne bool Changed = false; 64284c287e3SPeter Collingbourne 64384c287e3SPeter Collingbourne for (unsigned i = 0, e = getSize(); i != e; ++i) { 64484c287e3SPeter Collingbourne Init *E; 64584c287e3SPeter Collingbourne Init *CurElt = getElement(i); 64684c287e3SPeter Collingbourne 64784c287e3SPeter Collingbourne do { 64884c287e3SPeter Collingbourne E = CurElt; 64984c287e3SPeter Collingbourne CurElt = CurElt->resolveReferences(R, RV); 65084c287e3SPeter Collingbourne Changed |= E != CurElt; 65184c287e3SPeter Collingbourne } while (E != CurElt); 65284c287e3SPeter Collingbourne Resolved.push_back(E); 65384c287e3SPeter Collingbourne } 65484c287e3SPeter Collingbourne 65584c287e3SPeter Collingbourne if (Changed) 65684c287e3SPeter Collingbourne return ListInit::get(Resolved, getType()); 65784c287e3SPeter Collingbourne return const_cast<ListInit *>(this); 65884c287e3SPeter Collingbourne } 65984c287e3SPeter Collingbourne 66084c287e3SPeter Collingbourne Init *ListInit::resolveListElementReference(Record &R, const RecordVal *IRV, 66184c287e3SPeter Collingbourne unsigned Elt) const { 66284c287e3SPeter Collingbourne if (Elt >= getSize()) 663011817a0SCraig Topper return nullptr; // Out of range reference. 66484c287e3SPeter Collingbourne Init *E = getElement(Elt); 66584c287e3SPeter Collingbourne // If the element is set to some value, or if we are resolving a reference 66684c287e3SPeter Collingbourne // to a specific variable and that variable is explicitly unset, then 66784c287e3SPeter Collingbourne // replace the VarListElementInit with it. 66888eb8dd4SSean Silva if (IRV || !isa<UnsetInit>(E)) 66984c287e3SPeter Collingbourne return E; 670011817a0SCraig Topper return nullptr; 67184c287e3SPeter Collingbourne } 67284c287e3SPeter Collingbourne 67384c287e3SPeter Collingbourne std::string ListInit::getAsString() const { 67484c287e3SPeter Collingbourne std::string Result = "["; 67584c287e3SPeter Collingbourne for (unsigned i = 0, e = Values.size(); i != e; ++i) { 67684c287e3SPeter Collingbourne if (i) Result += ", "; 67784c287e3SPeter Collingbourne Result += Values[i]->getAsString(); 67884c287e3SPeter Collingbourne } 67984c287e3SPeter Collingbourne return Result + "]"; 68084c287e3SPeter Collingbourne } 68184c287e3SPeter Collingbourne 68284c287e3SPeter Collingbourne Init *OpInit::resolveListElementReference(Record &R, const RecordVal *IRV, 68384c287e3SPeter Collingbourne unsigned Elt) const { 684a8b8ab6bSDavid Greene Init *Resolved = resolveReferences(R, IRV); 685fb509ed1SSean Silva OpInit *OResolved = dyn_cast<OpInit>(Resolved); 686a8b8ab6bSDavid Greene if (OResolved) { 687011817a0SCraig Topper Resolved = OResolved->Fold(&R, nullptr); 688a8b8ab6bSDavid Greene } 68984c287e3SPeter Collingbourne 690a8b8ab6bSDavid Greene if (Resolved != this) { 691ed5a9508SCraig Topper TypedInit *Typed = cast<TypedInit>(Resolved); 692ed5a9508SCraig Topper if (Init *New = Typed->resolveListElementReference(R, IRV, Elt)) 693a8b8ab6bSDavid Greene return New; 694a8b8ab6bSDavid Greene return VarListElementInit::get(Typed, Elt); 69584c287e3SPeter Collingbourne } 69684c287e3SPeter Collingbourne 697011817a0SCraig Topper return nullptr; 69884c287e3SPeter Collingbourne } 69984c287e3SPeter Collingbourne 700026f8333SMichael Liao Init *OpInit::getBit(unsigned Bit) const { 701026f8333SMichael Liao if (getType() == BitRecTy::get()) 702026f8333SMichael Liao return const_cast<OpInit*>(this); 703026f8333SMichael Liao return VarBitInit::get(const_cast<OpInit*>(this), Bit); 704026f8333SMichael Liao } 705026f8333SMichael Liao 70684c287e3SPeter Collingbourne UnOpInit *UnOpInit::get(UnaryOp opc, Init *lhs, RecTy *Type) { 70784c287e3SPeter Collingbourne typedef std::pair<std::pair<unsigned, Init *>, RecTy *> Key; 7080e04bee8SCraig Topper static DenseMap<Key, std::unique_ptr<UnOpInit>> ThePool; 70984c287e3SPeter Collingbourne 71084c287e3SPeter Collingbourne Key TheKey(std::make_pair(std::make_pair(opc, lhs), Type)); 71184c287e3SPeter Collingbourne 7120e04bee8SCraig Topper std::unique_ptr<UnOpInit> &I = ThePool[TheKey]; 7130e04bee8SCraig Topper if (!I) I.reset(new UnOpInit(opc, lhs, Type)); 7140e04bee8SCraig Topper return I.get(); 71584c287e3SPeter Collingbourne } 71684c287e3SPeter Collingbourne 71784c287e3SPeter Collingbourne Init *UnOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const { 71884c287e3SPeter Collingbourne switch (getOpcode()) { 71984c287e3SPeter Collingbourne case CAST: { 72084c287e3SPeter Collingbourne if (getType()->getAsString() == "string") { 72188eb8dd4SSean Silva if (StringInit *LHSs = dyn_cast<StringInit>(LHS)) 72284c287e3SPeter Collingbourne return LHSs; 72384c287e3SPeter Collingbourne 72488eb8dd4SSean Silva if (DefInit *LHSd = dyn_cast<DefInit>(LHS)) 72584c287e3SPeter Collingbourne return StringInit::get(LHSd->getDef()->getName()); 7266291f3a4SDavid Greene 72788eb8dd4SSean Silva if (IntInit *LHSi = dyn_cast<IntInit>(LHS)) 7286291f3a4SDavid Greene return StringInit::get(LHSi->getAsString()); 72984c287e3SPeter Collingbourne } else { 73088eb8dd4SSean Silva if (StringInit *LHSs = dyn_cast<StringInit>(LHS)) { 73184c287e3SPeter Collingbourne std::string Name = LHSs->getValue(); 73284c287e3SPeter Collingbourne 73384c287e3SPeter Collingbourne // From TGParser::ParseIDValue 73484c287e3SPeter Collingbourne if (CurRec) { 73584c287e3SPeter Collingbourne if (const RecordVal *RV = CurRec->getValue(Name)) { 73684c287e3SPeter Collingbourne if (RV->getType() != getType()) 737635debe8SJoerg Sonnenberger PrintFatalError("type mismatch in cast"); 73884c287e3SPeter Collingbourne return VarInit::get(Name, RV->getType()); 73984c287e3SPeter Collingbourne } 74084c287e3SPeter Collingbourne 741db10e691SDavid Greene Init *TemplateArgName = QualifyName(*CurRec, CurMultiClass, Name, 742db10e691SDavid Greene ":"); 743db10e691SDavid Greene 74484c287e3SPeter Collingbourne if (CurRec->isTemplateArg(TemplateArgName)) { 74584c287e3SPeter Collingbourne const RecordVal *RV = CurRec->getValue(TemplateArgName); 74684c287e3SPeter Collingbourne assert(RV && "Template arg doesn't exist??"); 74784c287e3SPeter Collingbourne 74884c287e3SPeter Collingbourne if (RV->getType() != getType()) 749635debe8SJoerg Sonnenberger PrintFatalError("type mismatch in cast"); 75084c287e3SPeter Collingbourne 75184c287e3SPeter Collingbourne return VarInit::get(TemplateArgName, RV->getType()); 75284c287e3SPeter Collingbourne } 75384c287e3SPeter Collingbourne } 75484c287e3SPeter Collingbourne 75584c287e3SPeter Collingbourne if (CurMultiClass) { 756a9642b4eSCraig Topper Init *MCName = QualifyName(CurMultiClass->Rec, CurMultiClass, Name, 757a9642b4eSCraig Topper "::"); 758db10e691SDavid Greene 75984c287e3SPeter Collingbourne if (CurMultiClass->Rec.isTemplateArg(MCName)) { 76084c287e3SPeter Collingbourne const RecordVal *RV = CurMultiClass->Rec.getValue(MCName); 76184c287e3SPeter Collingbourne assert(RV && "Template arg doesn't exist??"); 76284c287e3SPeter Collingbourne 76384c287e3SPeter Collingbourne if (RV->getType() != getType()) 764635debe8SJoerg Sonnenberger PrintFatalError("type mismatch in cast"); 76584c287e3SPeter Collingbourne 76684c287e3SPeter Collingbourne return VarInit::get(MCName, RV->getType()); 76784c287e3SPeter Collingbourne } 76884c287e3SPeter Collingbourne } 7695be22a12SMichael Ilseman assert(CurRec && "NULL pointer"); 77084c287e3SPeter Collingbourne if (Record *D = (CurRec->getRecords()).getDef(Name)) 77184c287e3SPeter Collingbourne return DefInit::get(D); 77284c287e3SPeter Collingbourne 773635debe8SJoerg Sonnenberger PrintFatalError(CurRec->getLoc(), 774635debe8SJoerg Sonnenberger "Undefined reference:'" + Name + "'\n"); 77584c287e3SPeter Collingbourne } 77684c287e3SPeter Collingbourne } 77784c287e3SPeter Collingbourne break; 77884c287e3SPeter Collingbourne } 77984c287e3SPeter Collingbourne case HEAD: { 78088eb8dd4SSean Silva if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) { 78135b2f757SCraig Topper assert(LHSl->getSize() != 0 && "Empty list in car"); 78284c287e3SPeter Collingbourne return LHSl->getElement(0); 78384c287e3SPeter Collingbourne } 78484c287e3SPeter Collingbourne break; 78584c287e3SPeter Collingbourne } 78684c287e3SPeter Collingbourne case TAIL: { 78788eb8dd4SSean Silva if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) { 78835b2f757SCraig Topper assert(LHSl->getSize() != 0 && "Empty list in cdr"); 78984c287e3SPeter Collingbourne // Note the +1. We can't just pass the result of getValues() 79084c287e3SPeter Collingbourne // directly. 791*607ac92dSCraig Topper return ListInit::get(LHSl->getValues().slice(1), LHSl->getType()); 79284c287e3SPeter Collingbourne } 79384c287e3SPeter Collingbourne break; 79484c287e3SPeter Collingbourne } 79584c287e3SPeter Collingbourne case EMPTY: { 79688eb8dd4SSean Silva if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) { 797a9642b4eSCraig Topper if (LHSl->getSize() == 0) 79884c287e3SPeter Collingbourne return IntInit::get(1); 79984c287e3SPeter Collingbourne return IntInit::get(0); 80084c287e3SPeter Collingbourne } 80188eb8dd4SSean Silva if (StringInit *LHSs = dyn_cast<StringInit>(LHS)) { 802a9642b4eSCraig Topper if (LHSs->getValue().empty()) 80384c287e3SPeter Collingbourne return IntInit::get(1); 80484c287e3SPeter Collingbourne return IntInit::get(0); 80584c287e3SPeter Collingbourne } 80684c287e3SPeter Collingbourne 80784c287e3SPeter Collingbourne break; 80884c287e3SPeter Collingbourne } 80984c287e3SPeter Collingbourne } 81084c287e3SPeter Collingbourne return const_cast<UnOpInit *>(this); 81184c287e3SPeter Collingbourne } 81284c287e3SPeter Collingbourne 81384c287e3SPeter Collingbourne Init *UnOpInit::resolveReferences(Record &R, const RecordVal *RV) const { 81484c287e3SPeter Collingbourne Init *lhs = LHS->resolveReferences(R, RV); 81584c287e3SPeter Collingbourne 81684c287e3SPeter Collingbourne if (LHS != lhs) 817011817a0SCraig Topper return (UnOpInit::get(getOpcode(), lhs, getType()))->Fold(&R, nullptr); 818011817a0SCraig Topper return Fold(&R, nullptr); 81984c287e3SPeter Collingbourne } 82084c287e3SPeter Collingbourne 82184c287e3SPeter Collingbourne std::string UnOpInit::getAsString() const { 82284c287e3SPeter Collingbourne std::string Result; 82384c287e3SPeter Collingbourne switch (Opc) { 82484c287e3SPeter Collingbourne case CAST: Result = "!cast<" + getType()->getAsString() + ">"; break; 82584c287e3SPeter Collingbourne case HEAD: Result = "!head"; break; 82684c287e3SPeter Collingbourne case TAIL: Result = "!tail"; break; 82784c287e3SPeter Collingbourne case EMPTY: Result = "!empty"; break; 82884c287e3SPeter Collingbourne } 82984c287e3SPeter Collingbourne return Result + "(" + LHS->getAsString() + ")"; 83084c287e3SPeter Collingbourne } 83184c287e3SPeter Collingbourne 83284c287e3SPeter Collingbourne BinOpInit *BinOpInit::get(BinaryOp opc, Init *lhs, 83384c287e3SPeter Collingbourne Init *rhs, RecTy *Type) { 83484c287e3SPeter Collingbourne typedef std::pair< 83584c287e3SPeter Collingbourne std::pair<std::pair<unsigned, Init *>, Init *>, 83684c287e3SPeter Collingbourne RecTy * 83784c287e3SPeter Collingbourne > Key; 83884c287e3SPeter Collingbourne 8390e04bee8SCraig Topper static DenseMap<Key, std::unique_ptr<BinOpInit>> ThePool; 84084c287e3SPeter Collingbourne 84184c287e3SPeter Collingbourne Key TheKey(std::make_pair(std::make_pair(std::make_pair(opc, lhs), rhs), 84284c287e3SPeter Collingbourne Type)); 84384c287e3SPeter Collingbourne 8440e04bee8SCraig Topper std::unique_ptr<BinOpInit> &I = ThePool[TheKey]; 8450e04bee8SCraig Topper if (!I) I.reset(new BinOpInit(opc, lhs, rhs, Type)); 8460e04bee8SCraig Topper return I.get(); 84784c287e3SPeter Collingbourne } 84884c287e3SPeter Collingbourne 84984c287e3SPeter Collingbourne Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const { 85084c287e3SPeter Collingbourne switch (getOpcode()) { 85184c287e3SPeter Collingbourne case CONCAT: { 852fb509ed1SSean Silva DagInit *LHSs = dyn_cast<DagInit>(LHS); 853fb509ed1SSean Silva DagInit *RHSs = dyn_cast<DagInit>(RHS); 85484c287e3SPeter Collingbourne if (LHSs && RHSs) { 855fb509ed1SSean Silva DefInit *LOp = dyn_cast<DefInit>(LHSs->getOperator()); 856fb509ed1SSean Silva DefInit *ROp = dyn_cast<DefInit>(RHSs->getOperator()); 857011817a0SCraig Topper if (!LOp || !ROp || LOp->getDef() != ROp->getDef()) 858635debe8SJoerg Sonnenberger PrintFatalError("Concated Dag operators do not match!"); 85984c287e3SPeter Collingbourne std::vector<Init*> Args; 86084c287e3SPeter Collingbourne std::vector<std::string> ArgNames; 86184c287e3SPeter Collingbourne for (unsigned i = 0, e = LHSs->getNumArgs(); i != e; ++i) { 86284c287e3SPeter Collingbourne Args.push_back(LHSs->getArg(i)); 86384c287e3SPeter Collingbourne ArgNames.push_back(LHSs->getArgName(i)); 86484c287e3SPeter Collingbourne } 86584c287e3SPeter Collingbourne for (unsigned i = 0, e = RHSs->getNumArgs(); i != e; ++i) { 86684c287e3SPeter Collingbourne Args.push_back(RHSs->getArg(i)); 86784c287e3SPeter Collingbourne ArgNames.push_back(RHSs->getArgName(i)); 86884c287e3SPeter Collingbourne } 86984c287e3SPeter Collingbourne return DagInit::get(LHSs->getOperator(), "", Args, ArgNames); 87084c287e3SPeter Collingbourne } 87184c287e3SPeter Collingbourne break; 87284c287e3SPeter Collingbourne } 873314e80e5SDaniel Sanders case LISTCONCAT: { 874314e80e5SDaniel Sanders ListInit *LHSs = dyn_cast<ListInit>(LHS); 875314e80e5SDaniel Sanders ListInit *RHSs = dyn_cast<ListInit>(RHS); 876314e80e5SDaniel Sanders if (LHSs && RHSs) { 877314e80e5SDaniel Sanders std::vector<Init *> Args; 878314e80e5SDaniel Sanders Args.insert(Args.end(), LHSs->begin(), LHSs->end()); 879314e80e5SDaniel Sanders Args.insert(Args.end(), RHSs->begin(), RHSs->end()); 880314e80e5SDaniel Sanders return ListInit::get( 88123fdafe6SCraig Topper Args, cast<ListRecTy>(LHSs->getType())->getElementType()); 882314e80e5SDaniel Sanders } 883314e80e5SDaniel Sanders break; 884314e80e5SDaniel Sanders } 88584c287e3SPeter Collingbourne case STRCONCAT: { 886fb509ed1SSean Silva StringInit *LHSs = dyn_cast<StringInit>(LHS); 887fb509ed1SSean Silva StringInit *RHSs = dyn_cast<StringInit>(RHS); 88884c287e3SPeter Collingbourne if (LHSs && RHSs) 88984c287e3SPeter Collingbourne return StringInit::get(LHSs->getValue() + RHSs->getValue()); 89084c287e3SPeter Collingbourne break; 89184c287e3SPeter Collingbourne } 89284c287e3SPeter Collingbourne case EQ: { 89384c287e3SPeter Collingbourne // try to fold eq comparison for 'bit' and 'int', otherwise fallback 89484c287e3SPeter Collingbourne // to string objects. 89584c287e3SPeter Collingbourne IntInit *L = 896fb509ed1SSean Silva dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get())); 89784c287e3SPeter Collingbourne IntInit *R = 898fb509ed1SSean Silva dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get())); 89984c287e3SPeter Collingbourne 90084c287e3SPeter Collingbourne if (L && R) 90184c287e3SPeter Collingbourne return IntInit::get(L->getValue() == R->getValue()); 90284c287e3SPeter Collingbourne 903fb509ed1SSean Silva StringInit *LHSs = dyn_cast<StringInit>(LHS); 904fb509ed1SSean Silva StringInit *RHSs = dyn_cast<StringInit>(RHS); 90584c287e3SPeter Collingbourne 90684c287e3SPeter Collingbourne // Make sure we've resolved 90784c287e3SPeter Collingbourne if (LHSs && RHSs) 90884c287e3SPeter Collingbourne return IntInit::get(LHSs->getValue() == RHSs->getValue()); 90984c287e3SPeter Collingbourne 91084c287e3SPeter Collingbourne break; 91184c287e3SPeter Collingbourne } 912c7d4dc13SHal Finkel case ADD: 9136b41a990SJoerg Sonnenberger case AND: 91484c287e3SPeter Collingbourne case SHL: 91584c287e3SPeter Collingbourne case SRA: 91684c287e3SPeter Collingbourne case SRL: { 917017fca02SAdam Nemet IntInit *LHSi = 918017fca02SAdam Nemet dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get())); 919017fca02SAdam Nemet IntInit *RHSi = 920017fca02SAdam Nemet dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get())); 92184c287e3SPeter Collingbourne if (LHSi && RHSi) { 92284c287e3SPeter Collingbourne int64_t LHSv = LHSi->getValue(), RHSv = RHSi->getValue(); 92384c287e3SPeter Collingbourne int64_t Result; 92484c287e3SPeter Collingbourne switch (getOpcode()) { 925a2886c21SCraig Topper default: llvm_unreachable("Bad opcode!"); 926c7d4dc13SHal Finkel case ADD: Result = LHSv + RHSv; break; 9276b41a990SJoerg Sonnenberger case AND: Result = LHSv & RHSv; break; 92884c287e3SPeter Collingbourne case SHL: Result = LHSv << RHSv; break; 92984c287e3SPeter Collingbourne case SRA: Result = LHSv >> RHSv; break; 93084c287e3SPeter Collingbourne case SRL: Result = (uint64_t)LHSv >> (uint64_t)RHSv; break; 93184c287e3SPeter Collingbourne } 93284c287e3SPeter Collingbourne return IntInit::get(Result); 93384c287e3SPeter Collingbourne } 93484c287e3SPeter Collingbourne break; 93584c287e3SPeter Collingbourne } 93684c287e3SPeter Collingbourne } 93784c287e3SPeter Collingbourne return const_cast<BinOpInit *>(this); 93884c287e3SPeter Collingbourne } 93984c287e3SPeter Collingbourne 94084c287e3SPeter Collingbourne Init *BinOpInit::resolveReferences(Record &R, const RecordVal *RV) const { 94184c287e3SPeter Collingbourne Init *lhs = LHS->resolveReferences(R, RV); 94284c287e3SPeter Collingbourne Init *rhs = RHS->resolveReferences(R, RV); 94384c287e3SPeter Collingbourne 94484c287e3SPeter Collingbourne if (LHS != lhs || RHS != rhs) 945011817a0SCraig Topper return (BinOpInit::get(getOpcode(), lhs, rhs, getType()))->Fold(&R,nullptr); 946011817a0SCraig Topper return Fold(&R, nullptr); 94784c287e3SPeter Collingbourne } 94884c287e3SPeter Collingbourne 94984c287e3SPeter Collingbourne std::string BinOpInit::getAsString() const { 95084c287e3SPeter Collingbourne std::string Result; 95184c287e3SPeter Collingbourne switch (Opc) { 95284c287e3SPeter Collingbourne case CONCAT: Result = "!con"; break; 953c7d4dc13SHal Finkel case ADD: Result = "!add"; break; 9546b41a990SJoerg Sonnenberger case AND: Result = "!and"; break; 95584c287e3SPeter Collingbourne case SHL: Result = "!shl"; break; 95684c287e3SPeter Collingbourne case SRA: Result = "!sra"; break; 95784c287e3SPeter Collingbourne case SRL: Result = "!srl"; break; 95884c287e3SPeter Collingbourne case EQ: Result = "!eq"; break; 959314e80e5SDaniel Sanders case LISTCONCAT: Result = "!listconcat"; break; 96084c287e3SPeter Collingbourne case STRCONCAT: Result = "!strconcat"; break; 96184c287e3SPeter Collingbourne } 96284c287e3SPeter Collingbourne return Result + "(" + LHS->getAsString() + ", " + RHS->getAsString() + ")"; 96384c287e3SPeter Collingbourne } 96484c287e3SPeter Collingbourne 965a9642b4eSCraig Topper TernOpInit *TernOpInit::get(TernaryOp opc, Init *lhs, Init *mhs, Init *rhs, 96684c287e3SPeter Collingbourne RecTy *Type) { 96784c287e3SPeter Collingbourne typedef std::pair< 96884c287e3SPeter Collingbourne std::pair< 96984c287e3SPeter Collingbourne std::pair<std::pair<unsigned, RecTy *>, Init *>, 97084c287e3SPeter Collingbourne Init * 97184c287e3SPeter Collingbourne >, 97284c287e3SPeter Collingbourne Init * 97384c287e3SPeter Collingbourne > Key; 97484c287e3SPeter Collingbourne 97517b3a378SCraig Topper static DenseMap<Key, std::unique_ptr<TernOpInit>> ThePool; 97684c287e3SPeter Collingbourne 97784c287e3SPeter Collingbourne Key TheKey(std::make_pair(std::make_pair(std::make_pair(std::make_pair(opc, 97884c287e3SPeter Collingbourne Type), 97984c287e3SPeter Collingbourne lhs), 98084c287e3SPeter Collingbourne mhs), 98184c287e3SPeter Collingbourne rhs)); 98284c287e3SPeter Collingbourne 98317b3a378SCraig Topper std::unique_ptr<TernOpInit> &I = ThePool[TheKey]; 98417b3a378SCraig Topper if (!I) I.reset(new TernOpInit(opc, lhs, mhs, rhs, Type)); 98517b3a378SCraig Topper return I.get(); 98684c287e3SPeter Collingbourne } 98784c287e3SPeter Collingbourne 98884c287e3SPeter Collingbourne static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type, 98984c287e3SPeter Collingbourne Record *CurRec, MultiClass *CurMultiClass); 99084c287e3SPeter Collingbourne 99184c287e3SPeter Collingbourne static Init *EvaluateOperation(OpInit *RHSo, Init *LHS, Init *Arg, 99284c287e3SPeter Collingbourne RecTy *Type, Record *CurRec, 99384c287e3SPeter Collingbourne MultiClass *CurMultiClass) { 99484c287e3SPeter Collingbourne // If this is a dag, recurse 9959ef76499SCraig Topper if (auto *TArg = dyn_cast<TypedInit>(Arg)) 9969ef76499SCraig Topper if (TArg->getType()->getAsString() == "dag") 9979ef76499SCraig Topper return ForeachHelper(LHS, Arg, RHSo, Type, CurRec, CurMultiClass); 99884c287e3SPeter Collingbourne 99939ba33b8SCraig Topper std::vector<Init *> NewOperands; 100084c287e3SPeter Collingbourne for (int i = 0; i < RHSo->getNumOperands(); ++i) { 10019ef76499SCraig Topper if (auto *RHSoo = dyn_cast<OpInit>(RHSo->getOperand(i))) { 10029ef76499SCraig Topper if (Init *Result = EvaluateOperation(RHSoo, LHS, Arg, 10039ef76499SCraig Topper Type, CurRec, CurMultiClass)) 100484c287e3SPeter Collingbourne NewOperands.push_back(Result); 10059ef76499SCraig Topper else 100684c287e3SPeter Collingbourne NewOperands.push_back(Arg); 100784c287e3SPeter Collingbourne } else if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) { 100884c287e3SPeter Collingbourne NewOperands.push_back(Arg); 100984c287e3SPeter Collingbourne } else { 101084c287e3SPeter Collingbourne NewOperands.push_back(RHSo->getOperand(i)); 101184c287e3SPeter Collingbourne } 101284c287e3SPeter Collingbourne } 101384c287e3SPeter Collingbourne 101484c287e3SPeter Collingbourne // Now run the operator and use its result as the new leaf 101584c287e3SPeter Collingbourne const OpInit *NewOp = RHSo->clone(NewOperands); 101684c287e3SPeter Collingbourne Init *NewVal = NewOp->Fold(CurRec, CurMultiClass); 1017011817a0SCraig Topper return (NewVal != NewOp) ? NewVal : nullptr; 101884c287e3SPeter Collingbourne } 101984c287e3SPeter Collingbourne 102084c287e3SPeter Collingbourne static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type, 102184c287e3SPeter Collingbourne Record *CurRec, MultiClass *CurMultiClass) { 1022fb509ed1SSean Silva DagInit *MHSd = dyn_cast<DagInit>(MHS); 1023fb509ed1SSean Silva ListInit *MHSl = dyn_cast<ListInit>(MHS); 102484c287e3SPeter Collingbourne 1025fb509ed1SSean Silva OpInit *RHSo = dyn_cast<OpInit>(RHS); 102684c287e3SPeter Collingbourne 1027a9642b4eSCraig Topper if (!RHSo) 1028635debe8SJoerg Sonnenberger PrintFatalError(CurRec->getLoc(), "!foreach requires an operator\n"); 102984c287e3SPeter Collingbourne 1030fb509ed1SSean Silva TypedInit *LHSt = dyn_cast<TypedInit>(LHS); 103184c287e3SPeter Collingbourne 1032635debe8SJoerg Sonnenberger if (!LHSt) 1033635debe8SJoerg Sonnenberger PrintFatalError(CurRec->getLoc(), "!foreach requires typed variable\n"); 103484c287e3SPeter Collingbourne 1035a475e433SSean Silva if ((MHSd && isa<DagRecTy>(Type)) || (MHSl && isa<ListRecTy>(Type))) { 103684c287e3SPeter Collingbourne if (MHSd) { 103784c287e3SPeter Collingbourne Init *Val = MHSd->getOperator(); 103884c287e3SPeter Collingbourne Init *Result = EvaluateOperation(RHSo, LHS, Val, 103984c287e3SPeter Collingbourne Type, CurRec, CurMultiClass); 1040a9642b4eSCraig Topper if (Result) 104184c287e3SPeter Collingbourne Val = Result; 104284c287e3SPeter Collingbourne 104384c287e3SPeter Collingbourne std::vector<std::pair<Init *, std::string> > args; 104484c287e3SPeter Collingbourne for (unsigned int i = 0; i < MHSd->getNumArgs(); ++i) { 104584c287e3SPeter Collingbourne Init *Arg; 104684c287e3SPeter Collingbourne std::string ArgName; 104784c287e3SPeter Collingbourne Arg = MHSd->getArg(i); 104884c287e3SPeter Collingbourne ArgName = MHSd->getArgName(i); 104984c287e3SPeter Collingbourne 105084c287e3SPeter Collingbourne // Process args 105184c287e3SPeter Collingbourne Init *Result = EvaluateOperation(RHSo, LHS, Arg, Type, 105284c287e3SPeter Collingbourne CurRec, CurMultiClass); 1053a9642b4eSCraig Topper if (Result) 105484c287e3SPeter Collingbourne Arg = Result; 105584c287e3SPeter Collingbourne 105684c287e3SPeter Collingbourne // TODO: Process arg names 105784c287e3SPeter Collingbourne args.push_back(std::make_pair(Arg, ArgName)); 105884c287e3SPeter Collingbourne } 105984c287e3SPeter Collingbourne 106084c287e3SPeter Collingbourne return DagInit::get(Val, "", args); 106184c287e3SPeter Collingbourne } 106284c287e3SPeter Collingbourne if (MHSl) { 106384c287e3SPeter Collingbourne std::vector<Init *> NewOperands; 106484c287e3SPeter Collingbourne std::vector<Init *> NewList(MHSl->begin(), MHSl->end()); 106584c287e3SPeter Collingbourne 10669ed595e4SCraig Topper for (Init *&Item : NewList) { 106784c287e3SPeter Collingbourne NewOperands.clear(); 106884c287e3SPeter Collingbourne for(int i = 0; i < RHSo->getNumOperands(); ++i) { 106984c287e3SPeter Collingbourne // First, replace the foreach variable with the list item 1070a9642b4eSCraig Topper if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) 107184c287e3SPeter Collingbourne NewOperands.push_back(Item); 1072a9642b4eSCraig Topper else 107384c287e3SPeter Collingbourne NewOperands.push_back(RHSo->getOperand(i)); 107484c287e3SPeter Collingbourne } 107584c287e3SPeter Collingbourne 107684c287e3SPeter Collingbourne // Now run the operator and use its result as the new list item 107784c287e3SPeter Collingbourne const OpInit *NewOp = RHSo->clone(NewOperands); 107884c287e3SPeter Collingbourne Init *NewItem = NewOp->Fold(CurRec, CurMultiClass); 107984c287e3SPeter Collingbourne if (NewItem != NewOp) 10809ed595e4SCraig Topper Item = NewItem; 108184c287e3SPeter Collingbourne } 108284c287e3SPeter Collingbourne return ListInit::get(NewList, MHSl->getType()); 108384c287e3SPeter Collingbourne } 108484c287e3SPeter Collingbourne } 1085011817a0SCraig Topper return nullptr; 108684c287e3SPeter Collingbourne } 108784c287e3SPeter Collingbourne 108884c287e3SPeter Collingbourne Init *TernOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const { 108984c287e3SPeter Collingbourne switch (getOpcode()) { 109084c287e3SPeter Collingbourne case SUBST: { 1091fb509ed1SSean Silva DefInit *LHSd = dyn_cast<DefInit>(LHS); 1092fb509ed1SSean Silva VarInit *LHSv = dyn_cast<VarInit>(LHS); 1093fb509ed1SSean Silva StringInit *LHSs = dyn_cast<StringInit>(LHS); 109484c287e3SPeter Collingbourne 1095fb509ed1SSean Silva DefInit *MHSd = dyn_cast<DefInit>(MHS); 1096fb509ed1SSean Silva VarInit *MHSv = dyn_cast<VarInit>(MHS); 1097fb509ed1SSean Silva StringInit *MHSs = dyn_cast<StringInit>(MHS); 109884c287e3SPeter Collingbourne 1099fb509ed1SSean Silva DefInit *RHSd = dyn_cast<DefInit>(RHS); 1100fb509ed1SSean Silva VarInit *RHSv = dyn_cast<VarInit>(RHS); 1101fb509ed1SSean Silva StringInit *RHSs = dyn_cast<StringInit>(RHS); 110284c287e3SPeter Collingbourne 110385c07007SCraig Topper if ((LHSd && MHSd && RHSd) || 110485c07007SCraig Topper (LHSv && MHSv && RHSv) || 110585c07007SCraig Topper (LHSs && MHSs && RHSs)) { 110684c287e3SPeter Collingbourne if (RHSd) { 110784c287e3SPeter Collingbourne Record *Val = RHSd->getDef(); 1108a9642b4eSCraig Topper if (LHSd->getAsString() == RHSd->getAsString()) 110984c287e3SPeter Collingbourne Val = MHSd->getDef(); 111084c287e3SPeter Collingbourne return DefInit::get(Val); 111184c287e3SPeter Collingbourne } 111284c287e3SPeter Collingbourne if (RHSv) { 111384c287e3SPeter Collingbourne std::string Val = RHSv->getName(); 1114a9642b4eSCraig Topper if (LHSv->getAsString() == RHSv->getAsString()) 111584c287e3SPeter Collingbourne Val = MHSv->getName(); 111684c287e3SPeter Collingbourne return VarInit::get(Val, getType()); 111784c287e3SPeter Collingbourne } 111884c287e3SPeter Collingbourne if (RHSs) { 111984c287e3SPeter Collingbourne std::string Val = RHSs->getValue(); 112084c287e3SPeter Collingbourne 112184c287e3SPeter Collingbourne std::string::size_type found; 112284c287e3SPeter Collingbourne std::string::size_type idx = 0; 112384c287e3SPeter Collingbourne do { 112484c287e3SPeter Collingbourne found = Val.find(LHSs->getValue(), idx); 1125a9642b4eSCraig Topper if (found != std::string::npos) 112684c287e3SPeter Collingbourne Val.replace(found, LHSs->getValue().size(), MHSs->getValue()); 112784c287e3SPeter Collingbourne idx = found + MHSs->getValue().size(); 112884c287e3SPeter Collingbourne } while (found != std::string::npos); 112984c287e3SPeter Collingbourne 113084c287e3SPeter Collingbourne return StringInit::get(Val); 113184c287e3SPeter Collingbourne } 113284c287e3SPeter Collingbourne } 113384c287e3SPeter Collingbourne break; 113484c287e3SPeter Collingbourne } 113584c287e3SPeter Collingbourne 113684c287e3SPeter Collingbourne case FOREACH: { 113784c287e3SPeter Collingbourne Init *Result = ForeachHelper(LHS, MHS, RHS, getType(), 113884c287e3SPeter Collingbourne CurRec, CurMultiClass); 1139a9642b4eSCraig Topper if (Result) 114084c287e3SPeter Collingbourne return Result; 114184c287e3SPeter Collingbourne break; 114284c287e3SPeter Collingbourne } 114384c287e3SPeter Collingbourne 114484c287e3SPeter Collingbourne case IF: { 1145fb509ed1SSean Silva IntInit *LHSi = dyn_cast<IntInit>(LHS); 114684c287e3SPeter Collingbourne if (Init *I = LHS->convertInitializerTo(IntRecTy::get())) 1147fb509ed1SSean Silva LHSi = dyn_cast<IntInit>(I); 114884c287e3SPeter Collingbourne if (LHSi) { 1149a9642b4eSCraig Topper if (LHSi->getValue()) 115084c287e3SPeter Collingbourne return MHS; 115184c287e3SPeter Collingbourne return RHS; 115284c287e3SPeter Collingbourne } 115384c287e3SPeter Collingbourne break; 115484c287e3SPeter Collingbourne } 115584c287e3SPeter Collingbourne } 115684c287e3SPeter Collingbourne 115784c287e3SPeter Collingbourne return const_cast<TernOpInit *>(this); 115884c287e3SPeter Collingbourne } 115984c287e3SPeter Collingbourne 116084c287e3SPeter Collingbourne Init *TernOpInit::resolveReferences(Record &R, 116184c287e3SPeter Collingbourne const RecordVal *RV) const { 116284c287e3SPeter Collingbourne Init *lhs = LHS->resolveReferences(R, RV); 116384c287e3SPeter Collingbourne 116484c287e3SPeter Collingbourne if (Opc == IF && lhs != LHS) { 1165fb509ed1SSean Silva IntInit *Value = dyn_cast<IntInit>(lhs); 116684c287e3SPeter Collingbourne if (Init *I = lhs->convertInitializerTo(IntRecTy::get())) 1167fb509ed1SSean Silva Value = dyn_cast<IntInit>(I); 1168011817a0SCraig Topper if (Value) { 116984c287e3SPeter Collingbourne // Short-circuit 117084c287e3SPeter Collingbourne if (Value->getValue()) { 117184c287e3SPeter Collingbourne Init *mhs = MHS->resolveReferences(R, RV); 117284c287e3SPeter Collingbourne return (TernOpInit::get(getOpcode(), lhs, mhs, 1173011817a0SCraig Topper RHS, getType()))->Fold(&R, nullptr); 1174a9642b4eSCraig Topper } 117584c287e3SPeter Collingbourne Init *rhs = RHS->resolveReferences(R, RV); 117684c287e3SPeter Collingbourne return (TernOpInit::get(getOpcode(), lhs, MHS, 1177011817a0SCraig Topper rhs, getType()))->Fold(&R, nullptr); 117884c287e3SPeter Collingbourne } 117984c287e3SPeter Collingbourne } 118084c287e3SPeter Collingbourne 118184c287e3SPeter Collingbourne Init *mhs = MHS->resolveReferences(R, RV); 118284c287e3SPeter Collingbourne Init *rhs = RHS->resolveReferences(R, RV); 118384c287e3SPeter Collingbourne 118484c287e3SPeter Collingbourne if (LHS != lhs || MHS != mhs || RHS != rhs) 118584c287e3SPeter Collingbourne return (TernOpInit::get(getOpcode(), lhs, mhs, rhs, 1186011817a0SCraig Topper getType()))->Fold(&R, nullptr); 1187011817a0SCraig Topper return Fold(&R, nullptr); 118884c287e3SPeter Collingbourne } 118984c287e3SPeter Collingbourne 119084c287e3SPeter Collingbourne std::string TernOpInit::getAsString() const { 119184c287e3SPeter Collingbourne std::string Result; 119284c287e3SPeter Collingbourne switch (Opc) { 119384c287e3SPeter Collingbourne case SUBST: Result = "!subst"; break; 119484c287e3SPeter Collingbourne case FOREACH: Result = "!foreach"; break; 119584c287e3SPeter Collingbourne case IF: Result = "!if"; break; 119684c287e3SPeter Collingbourne } 119785c07007SCraig Topper return Result + "(" + LHS->getAsString() + ", " + MHS->getAsString() + ", " + 119885c07007SCraig Topper RHS->getAsString() + ")"; 119984c287e3SPeter Collingbourne } 120084c287e3SPeter Collingbourne 120184c287e3SPeter Collingbourne RecTy *TypedInit::getFieldType(const std::string &FieldName) const { 1202a475e433SSean Silva if (RecordRecTy *RecordType = dyn_cast<RecordRecTy>(getType())) 1203a475e433SSean Silva if (RecordVal *Field = RecordType->getRecord()->getValue(FieldName)) 120484c287e3SPeter Collingbourne return Field->getType(); 1205011817a0SCraig Topper return nullptr; 120684c287e3SPeter Collingbourne } 120784c287e3SPeter Collingbourne 120884c287e3SPeter Collingbourne Init * 120984c287e3SPeter Collingbourne TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const { 121098c61711SSean Silva BitsRecTy *T = dyn_cast<BitsRecTy>(getType()); 1211011817a0SCraig Topper if (!T) return nullptr; // Cannot subscript a non-bits variable. 121284c287e3SPeter Collingbourne unsigned NumBits = T->getNumBits(); 121384c287e3SPeter Collingbourne 121484c287e3SPeter Collingbourne SmallVector<Init *, 16> NewBits(Bits.size()); 121584c287e3SPeter Collingbourne for (unsigned i = 0, e = Bits.size(); i != e; ++i) { 121684c287e3SPeter Collingbourne if (Bits[i] >= NumBits) 1217011817a0SCraig Topper return nullptr; 121884c287e3SPeter Collingbourne 121984c287e3SPeter Collingbourne NewBits[i] = VarBitInit::get(const_cast<TypedInit *>(this), Bits[i]); 122084c287e3SPeter Collingbourne } 122184c287e3SPeter Collingbourne return BitsInit::get(NewBits); 122284c287e3SPeter Collingbourne } 122384c287e3SPeter Collingbourne 122484c287e3SPeter Collingbourne Init * 122584c287e3SPeter Collingbourne TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) const { 122698c61711SSean Silva ListRecTy *T = dyn_cast<ListRecTy>(getType()); 1227011817a0SCraig Topper if (!T) return nullptr; // Cannot subscript a non-list variable. 122884c287e3SPeter Collingbourne 122984c287e3SPeter Collingbourne if (Elements.size() == 1) 123084c287e3SPeter Collingbourne return VarListElementInit::get(const_cast<TypedInit *>(this), Elements[0]); 123184c287e3SPeter Collingbourne 123284c287e3SPeter Collingbourne std::vector<Init*> ListInits; 123384c287e3SPeter Collingbourne ListInits.reserve(Elements.size()); 123484c287e3SPeter Collingbourne for (unsigned i = 0, e = Elements.size(); i != e; ++i) 123584c287e3SPeter Collingbourne ListInits.push_back(VarListElementInit::get(const_cast<TypedInit *>(this), 123684c287e3SPeter Collingbourne Elements[i])); 123784c287e3SPeter Collingbourne return ListInit::get(ListInits, T); 123884c287e3SPeter Collingbourne } 123984c287e3SPeter Collingbourne 124084c287e3SPeter Collingbourne 124184c287e3SPeter Collingbourne VarInit *VarInit::get(const std::string &VN, RecTy *T) { 1242914adf0eSDavid Greene Init *Value = StringInit::get(VN); 1243914adf0eSDavid Greene return VarInit::get(Value, T); 1244914adf0eSDavid Greene } 1245914adf0eSDavid Greene 1246914adf0eSDavid Greene VarInit *VarInit::get(Init *VN, RecTy *T) { 1247914adf0eSDavid Greene typedef std::pair<RecTy *, Init *> Key; 12480e04bee8SCraig Topper static DenseMap<Key, std::unique_ptr<VarInit>> ThePool; 124984c287e3SPeter Collingbourne 125084c287e3SPeter Collingbourne Key TheKey(std::make_pair(T, VN)); 125184c287e3SPeter Collingbourne 12520e04bee8SCraig Topper std::unique_ptr<VarInit> &I = ThePool[TheKey]; 12530e04bee8SCraig Topper if (!I) I.reset(new VarInit(VN, T)); 12540e04bee8SCraig Topper return I.get(); 125584c287e3SPeter Collingbourne } 125684c287e3SPeter Collingbourne 1257914adf0eSDavid Greene const std::string &VarInit::getName() const { 1258ed5a9508SCraig Topper StringInit *NameString = cast<StringInit>(getNameInit()); 1259914adf0eSDavid Greene return NameString->getValue(); 1260914adf0eSDavid Greene } 1261914adf0eSDavid Greene 1262026f8333SMichael Liao Init *VarInit::getBit(unsigned Bit) const { 1263026f8333SMichael Liao if (getType() == BitRecTy::get()) 1264026f8333SMichael Liao return const_cast<VarInit*>(this); 1265026f8333SMichael Liao return VarBitInit::get(const_cast<VarInit*>(this), Bit); 126684c287e3SPeter Collingbourne } 126784c287e3SPeter Collingbourne 126884c287e3SPeter Collingbourne Init *VarInit::resolveListElementReference(Record &R, 126984c287e3SPeter Collingbourne const RecordVal *IRV, 127084c287e3SPeter Collingbourne unsigned Elt) const { 1271011817a0SCraig Topper if (R.isTemplateArg(getNameInit())) return nullptr; 1272011817a0SCraig Topper if (IRV && IRV->getNameInit() != getNameInit()) return nullptr; 127384c287e3SPeter Collingbourne 12749d1c5eebSJakob Stoklund Olesen RecordVal *RV = R.getValue(getNameInit()); 127584c287e3SPeter Collingbourne assert(RV && "Reference to a non-existent variable?"); 1276fb509ed1SSean Silva ListInit *LI = dyn_cast<ListInit>(RV->getValue()); 1277a9642b4eSCraig Topper if (!LI) 1278a9642b4eSCraig Topper return VarListElementInit::get(cast<TypedInit>(RV->getValue()), Elt); 127984c287e3SPeter Collingbourne 128084c287e3SPeter Collingbourne if (Elt >= LI->getSize()) 1281011817a0SCraig Topper return nullptr; // Out of range reference. 128284c287e3SPeter Collingbourne Init *E = LI->getElement(Elt); 128384c287e3SPeter Collingbourne // If the element is set to some value, or if we are resolving a reference 128484c287e3SPeter Collingbourne // to a specific variable and that variable is explicitly unset, then 128584c287e3SPeter Collingbourne // replace the VarListElementInit with it. 128688eb8dd4SSean Silva if (IRV || !isa<UnsetInit>(E)) 128784c287e3SPeter Collingbourne return E; 1288011817a0SCraig Topper return nullptr; 128984c287e3SPeter Collingbourne } 129084c287e3SPeter Collingbourne 129184c287e3SPeter Collingbourne 129284c287e3SPeter Collingbourne RecTy *VarInit::getFieldType(const std::string &FieldName) const { 129398c61711SSean Silva if (RecordRecTy *RTy = dyn_cast<RecordRecTy>(getType())) 129484c287e3SPeter Collingbourne if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName)) 129584c287e3SPeter Collingbourne return RV->getType(); 1296011817a0SCraig Topper return nullptr; 129784c287e3SPeter Collingbourne } 129884c287e3SPeter Collingbourne 129984c287e3SPeter Collingbourne Init *VarInit::getFieldInit(Record &R, const RecordVal *RV, 130084c287e3SPeter Collingbourne const std::string &FieldName) const { 1301a475e433SSean Silva if (isa<RecordRecTy>(getType())) 130284c287e3SPeter Collingbourne if (const RecordVal *Val = R.getValue(VarName)) { 130388eb8dd4SSean Silva if (RV != Val && (RV || isa<UnsetInit>(Val->getValue()))) 1304011817a0SCraig Topper return nullptr; 130584c287e3SPeter Collingbourne Init *TheInit = Val->getValue(); 130684c287e3SPeter Collingbourne assert(TheInit != this && "Infinite loop detected!"); 130784c287e3SPeter Collingbourne if (Init *I = TheInit->getFieldInit(R, RV, FieldName)) 130884c287e3SPeter Collingbourne return I; 1309011817a0SCraig Topper return nullptr; 131084c287e3SPeter Collingbourne } 1311011817a0SCraig Topper return nullptr; 131284c287e3SPeter Collingbourne } 131384c287e3SPeter Collingbourne 131484c287e3SPeter Collingbourne /// resolveReferences - This method is used by classes that refer to other 131584c287e3SPeter Collingbourne /// variables which may not be defined at the time the expression is formed. 131684c287e3SPeter Collingbourne /// If a value is set for the variable later, this method will be called on 131784c287e3SPeter Collingbourne /// users of the value to allow the value to propagate out. 131884c287e3SPeter Collingbourne /// 131984c287e3SPeter Collingbourne Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) const { 132084c287e3SPeter Collingbourne if (RecordVal *Val = R.getValue(VarName)) 1321011817a0SCraig Topper if (RV == Val || (!RV && !isa<UnsetInit>(Val->getValue()))) 132284c287e3SPeter Collingbourne return Val->getValue(); 132384c287e3SPeter Collingbourne return const_cast<VarInit *>(this); 132484c287e3SPeter Collingbourne } 132584c287e3SPeter Collingbourne 132684c287e3SPeter Collingbourne VarBitInit *VarBitInit::get(TypedInit *T, unsigned B) { 132784c287e3SPeter Collingbourne typedef std::pair<TypedInit *, unsigned> Key; 13280e04bee8SCraig Topper static DenseMap<Key, std::unique_ptr<VarBitInit>> ThePool; 132984c287e3SPeter Collingbourne 133084c287e3SPeter Collingbourne Key TheKey(std::make_pair(T, B)); 133184c287e3SPeter Collingbourne 13320e04bee8SCraig Topper std::unique_ptr<VarBitInit> &I = ThePool[TheKey]; 13330e04bee8SCraig Topper if (!I) I.reset(new VarBitInit(T, B)); 13340e04bee8SCraig Topper return I.get(); 133584c287e3SPeter Collingbourne } 133684c287e3SPeter Collingbourne 133784c287e3SPeter Collingbourne std::string VarBitInit::getAsString() const { 133884c287e3SPeter Collingbourne return TI->getAsString() + "{" + utostr(Bit) + "}"; 133984c287e3SPeter Collingbourne } 134084c287e3SPeter Collingbourne 134184c287e3SPeter Collingbourne Init *VarBitInit::resolveReferences(Record &R, const RecordVal *RV) const { 1342026f8333SMichael Liao Init *I = TI->resolveReferences(R, RV); 1343026f8333SMichael Liao if (TI != I) 1344026f8333SMichael Liao return I->getBit(getBitNum()); 1345026f8333SMichael Liao 134684c287e3SPeter Collingbourne return const_cast<VarBitInit*>(this); 134784c287e3SPeter Collingbourne } 134884c287e3SPeter Collingbourne 134984c287e3SPeter Collingbourne VarListElementInit *VarListElementInit::get(TypedInit *T, 135084c287e3SPeter Collingbourne unsigned E) { 135184c287e3SPeter Collingbourne typedef std::pair<TypedInit *, unsigned> Key; 135217b3a378SCraig Topper static DenseMap<Key, std::unique_ptr<VarListElementInit>> ThePool; 135384c287e3SPeter Collingbourne 135484c287e3SPeter Collingbourne Key TheKey(std::make_pair(T, E)); 135584c287e3SPeter Collingbourne 135617b3a378SCraig Topper std::unique_ptr<VarListElementInit> &I = ThePool[TheKey]; 135717b3a378SCraig Topper if (!I) I.reset(new VarListElementInit(T, E)); 135817b3a378SCraig Topper return I.get(); 135984c287e3SPeter Collingbourne } 136084c287e3SPeter Collingbourne 136184c287e3SPeter Collingbourne std::string VarListElementInit::getAsString() const { 136284c287e3SPeter Collingbourne return TI->getAsString() + "[" + utostr(Element) + "]"; 136384c287e3SPeter Collingbourne } 136484c287e3SPeter Collingbourne 136584c287e3SPeter Collingbourne Init * 136684c287e3SPeter Collingbourne VarListElementInit::resolveReferences(Record &R, const RecordVal *RV) const { 136784c287e3SPeter Collingbourne if (Init *I = getVariable()->resolveListElementReference(R, RV, 136884c287e3SPeter Collingbourne getElementNum())) 136984c287e3SPeter Collingbourne return I; 137084c287e3SPeter Collingbourne return const_cast<VarListElementInit *>(this); 137184c287e3SPeter Collingbourne } 137284c287e3SPeter Collingbourne 1373026f8333SMichael Liao Init *VarListElementInit::getBit(unsigned Bit) const { 1374026f8333SMichael Liao if (getType() == BitRecTy::get()) 1375026f8333SMichael Liao return const_cast<VarListElementInit*>(this); 1376026f8333SMichael Liao return VarBitInit::get(const_cast<VarListElementInit*>(this), Bit); 137784c287e3SPeter Collingbourne } 137884c287e3SPeter Collingbourne 137984c287e3SPeter Collingbourne Init *VarListElementInit:: resolveListElementReference(Record &R, 138084c287e3SPeter Collingbourne const RecordVal *RV, 138184c287e3SPeter Collingbourne unsigned Elt) const { 1382a9642b4eSCraig Topper if (Init *Result = TI->resolveListElementReference(R, RV, Element)) { 138388eb8dd4SSean Silva if (TypedInit *TInit = dyn_cast<TypedInit>(Result)) { 13847475ad05SDavid Greene Init *Result2 = TInit->resolveListElementReference(R, RV, Elt); 13857475ad05SDavid Greene if (Result2) return Result2; 138617b3a378SCraig Topper return VarListElementInit::get(TInit, Elt); 138784c287e3SPeter Collingbourne } 138884c287e3SPeter Collingbourne return Result; 138984c287e3SPeter Collingbourne } 139084c287e3SPeter Collingbourne 1391011817a0SCraig Topper return nullptr; 139284c287e3SPeter Collingbourne } 139384c287e3SPeter Collingbourne 139484c287e3SPeter Collingbourne DefInit *DefInit::get(Record *R) { 139584c287e3SPeter Collingbourne return R->getDefInit(); 139684c287e3SPeter Collingbourne } 139784c287e3SPeter Collingbourne 139884c287e3SPeter Collingbourne RecTy *DefInit::getFieldType(const std::string &FieldName) const { 139984c287e3SPeter Collingbourne if (const RecordVal *RV = Def->getValue(FieldName)) 140084c287e3SPeter Collingbourne return RV->getType(); 1401011817a0SCraig Topper return nullptr; 140284c287e3SPeter Collingbourne } 140384c287e3SPeter Collingbourne 140484c287e3SPeter Collingbourne Init *DefInit::getFieldInit(Record &R, const RecordVal *RV, 140584c287e3SPeter Collingbourne const std::string &FieldName) const { 140684c287e3SPeter Collingbourne return Def->getValue(FieldName)->getValue(); 140784c287e3SPeter Collingbourne } 140884c287e3SPeter Collingbourne 140984c287e3SPeter Collingbourne 141084c287e3SPeter Collingbourne std::string DefInit::getAsString() const { 141184c287e3SPeter Collingbourne return Def->getName(); 141284c287e3SPeter Collingbourne } 141384c287e3SPeter Collingbourne 141484c287e3SPeter Collingbourne FieldInit *FieldInit::get(Init *R, const std::string &FN) { 141584c287e3SPeter Collingbourne typedef std::pair<Init *, TableGenStringKey> Key; 141617b3a378SCraig Topper static DenseMap<Key, std::unique_ptr<FieldInit>> ThePool; 141784c287e3SPeter Collingbourne 141884c287e3SPeter Collingbourne Key TheKey(std::make_pair(R, FN)); 141984c287e3SPeter Collingbourne 142017b3a378SCraig Topper std::unique_ptr<FieldInit> &I = ThePool[TheKey]; 142117b3a378SCraig Topper if (!I) I.reset(new FieldInit(R, FN)); 142217b3a378SCraig Topper return I.get(); 142384c287e3SPeter Collingbourne } 142484c287e3SPeter Collingbourne 1425026f8333SMichael Liao Init *FieldInit::getBit(unsigned Bit) const { 1426026f8333SMichael Liao if (getType() == BitRecTy::get()) 1427026f8333SMichael Liao return const_cast<FieldInit*>(this); 1428026f8333SMichael Liao return VarBitInit::get(const_cast<FieldInit*>(this), Bit); 142984c287e3SPeter Collingbourne } 143084c287e3SPeter Collingbourne 143184c287e3SPeter Collingbourne Init *FieldInit::resolveListElementReference(Record &R, const RecordVal *RV, 143284c287e3SPeter Collingbourne unsigned Elt) const { 143384c287e3SPeter Collingbourne if (Init *ListVal = Rec->getFieldInit(R, RV, FieldName)) 1434fb509ed1SSean Silva if (ListInit *LI = dyn_cast<ListInit>(ListVal)) { 1435011817a0SCraig Topper if (Elt >= LI->getSize()) return nullptr; 143684c287e3SPeter Collingbourne Init *E = LI->getElement(Elt); 143784c287e3SPeter Collingbourne 143884c287e3SPeter Collingbourne // If the element is set to some value, or if we are resolving a 143984c287e3SPeter Collingbourne // reference to a specific variable and that variable is explicitly 144084c287e3SPeter Collingbourne // unset, then replace the VarListElementInit with it. 144188eb8dd4SSean Silva if (RV || !isa<UnsetInit>(E)) 144284c287e3SPeter Collingbourne return E; 144384c287e3SPeter Collingbourne } 1444011817a0SCraig Topper return nullptr; 144584c287e3SPeter Collingbourne } 144684c287e3SPeter Collingbourne 144784c287e3SPeter Collingbourne Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) const { 144884c287e3SPeter Collingbourne Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec; 144984c287e3SPeter Collingbourne 1450a9642b4eSCraig Topper if (Init *BitsVal = NewRec->getFieldInit(R, RV, FieldName)) { 145184c287e3SPeter Collingbourne Init *BVR = BitsVal->resolveReferences(R, RV); 145284c287e3SPeter Collingbourne return BVR->isComplete() ? BVR : const_cast<FieldInit *>(this); 145384c287e3SPeter Collingbourne } 145484c287e3SPeter Collingbourne 1455a9642b4eSCraig Topper if (NewRec != Rec) 145684c287e3SPeter Collingbourne return FieldInit::get(NewRec, FieldName); 145784c287e3SPeter Collingbourne return const_cast<FieldInit *>(this); 145884c287e3SPeter Collingbourne } 145984c287e3SPeter Collingbourne 14606ecb1e78SBenjamin Kramer static void ProfileDagInit(FoldingSetNodeID &ID, Init *V, const std::string &VN, 146184c287e3SPeter Collingbourne ArrayRef<Init *> ArgRange, 146284c287e3SPeter Collingbourne ArrayRef<std::string> NameRange) { 146384c287e3SPeter Collingbourne ID.AddPointer(V); 146484c287e3SPeter Collingbourne ID.AddString(VN); 146584c287e3SPeter Collingbourne 146684c287e3SPeter Collingbourne ArrayRef<Init *>::iterator Arg = ArgRange.begin(); 146784c287e3SPeter Collingbourne ArrayRef<std::string>::iterator Name = NameRange.begin(); 146884c287e3SPeter Collingbourne while (Arg != ArgRange.end()) { 146984c287e3SPeter Collingbourne assert(Name != NameRange.end() && "Arg name underflow!"); 147084c287e3SPeter Collingbourne ID.AddPointer(*Arg++); 147184c287e3SPeter Collingbourne ID.AddString(*Name++); 147284c287e3SPeter Collingbourne } 147384c287e3SPeter Collingbourne assert(Name == NameRange.end() && "Arg name overflow!"); 147484c287e3SPeter Collingbourne } 147584c287e3SPeter Collingbourne 147684c287e3SPeter Collingbourne DagInit * 147784c287e3SPeter Collingbourne DagInit::get(Init *V, const std::string &VN, 147884c287e3SPeter Collingbourne ArrayRef<Init *> ArgRange, 147984c287e3SPeter Collingbourne ArrayRef<std::string> NameRange) { 148017b3a378SCraig Topper static FoldingSet<DagInit> ThePool; 148117b3a378SCraig Topper static std::vector<std::unique_ptr<DagInit>> TheActualPool; 148284c287e3SPeter Collingbourne 148384c287e3SPeter Collingbourne FoldingSetNodeID ID; 148484c287e3SPeter Collingbourne ProfileDagInit(ID, V, VN, ArgRange, NameRange); 148584c287e3SPeter Collingbourne 1486011817a0SCraig Topper void *IP = nullptr; 148784c287e3SPeter Collingbourne if (DagInit *I = ThePool.FindNodeOrInsertPos(ID, IP)) 148884c287e3SPeter Collingbourne return I; 148984c287e3SPeter Collingbourne 149084c287e3SPeter Collingbourne DagInit *I = new DagInit(V, VN, ArgRange, NameRange); 149184c287e3SPeter Collingbourne ThePool.InsertNode(I, IP); 149217b3a378SCraig Topper TheActualPool.push_back(std::unique_ptr<DagInit>(I)); 149384c287e3SPeter Collingbourne return I; 149484c287e3SPeter Collingbourne } 149584c287e3SPeter Collingbourne 149684c287e3SPeter Collingbourne DagInit * 149784c287e3SPeter Collingbourne DagInit::get(Init *V, const std::string &VN, 149884c287e3SPeter Collingbourne const std::vector<std::pair<Init*, std::string> > &args) { 149984c287e3SPeter Collingbourne std::vector<Init *> Args; 150084c287e3SPeter Collingbourne std::vector<std::string> Names; 150184c287e3SPeter Collingbourne 1502119998dbSCraig Topper for (const auto &Arg : args) { 1503119998dbSCraig Topper Args.push_back(Arg.first); 1504119998dbSCraig Topper Names.push_back(Arg.second); 150584c287e3SPeter Collingbourne } 150684c287e3SPeter Collingbourne 150784c287e3SPeter Collingbourne return DagInit::get(V, VN, Args, Names); 150884c287e3SPeter Collingbourne } 150984c287e3SPeter Collingbourne 151084c287e3SPeter Collingbourne void DagInit::Profile(FoldingSetNodeID &ID) const { 151184c287e3SPeter Collingbourne ProfileDagInit(ID, Val, ValName, Args, ArgNames); 151284c287e3SPeter Collingbourne } 151384c287e3SPeter Collingbourne 151484c287e3SPeter Collingbourne Init *DagInit::resolveReferences(Record &R, const RecordVal *RV) const { 151584c287e3SPeter Collingbourne std::vector<Init*> NewArgs; 151684c287e3SPeter Collingbourne for (unsigned i = 0, e = Args.size(); i != e; ++i) 151784c287e3SPeter Collingbourne NewArgs.push_back(Args[i]->resolveReferences(R, RV)); 151884c287e3SPeter Collingbourne 151984c287e3SPeter Collingbourne Init *Op = Val->resolveReferences(R, RV); 152084c287e3SPeter Collingbourne 152184c287e3SPeter Collingbourne if (Args != NewArgs || Op != Val) 152284c287e3SPeter Collingbourne return DagInit::get(Op, ValName, NewArgs, ArgNames); 152384c287e3SPeter Collingbourne 152484c287e3SPeter Collingbourne return const_cast<DagInit *>(this); 152584c287e3SPeter Collingbourne } 152684c287e3SPeter Collingbourne 152784c287e3SPeter Collingbourne 152884c287e3SPeter Collingbourne std::string DagInit::getAsString() const { 152984c287e3SPeter Collingbourne std::string Result = "(" + Val->getAsString(); 153084c287e3SPeter Collingbourne if (!ValName.empty()) 153184c287e3SPeter Collingbourne Result += ":" + ValName; 15328c0809c7SAlexander Kornienko if (!Args.empty()) { 153384c287e3SPeter Collingbourne Result += " " + Args[0]->getAsString(); 153484c287e3SPeter Collingbourne if (!ArgNames[0].empty()) Result += ":$" + ArgNames[0]; 153584c287e3SPeter Collingbourne for (unsigned i = 1, e = Args.size(); i != e; ++i) { 153684c287e3SPeter Collingbourne Result += ", " + Args[i]->getAsString(); 153784c287e3SPeter Collingbourne if (!ArgNames[i].empty()) Result += ":$" + ArgNames[i]; 153884c287e3SPeter Collingbourne } 153984c287e3SPeter Collingbourne } 154084c287e3SPeter Collingbourne return Result + ")"; 154184c287e3SPeter Collingbourne } 154284c287e3SPeter Collingbourne 154384c287e3SPeter Collingbourne 154484c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 154584c287e3SPeter Collingbourne // Other implementations 154684c287e3SPeter Collingbourne //===----------------------------------------------------------------------===// 154784c287e3SPeter Collingbourne 154884c287e3SPeter Collingbourne RecordVal::RecordVal(Init *N, RecTy *T, unsigned P) 154984c287e3SPeter Collingbourne : Name(N), Ty(T), Prefix(P) { 155084c287e3SPeter Collingbourne Value = Ty->convertValue(UnsetInit::get()); 155184c287e3SPeter Collingbourne assert(Value && "Cannot create unset value for current type!"); 155284c287e3SPeter Collingbourne } 155384c287e3SPeter Collingbourne 155484c287e3SPeter Collingbourne RecordVal::RecordVal(const std::string &N, RecTy *T, unsigned P) 155584c287e3SPeter Collingbourne : Name(StringInit::get(N)), Ty(T), Prefix(P) { 155684c287e3SPeter Collingbourne Value = Ty->convertValue(UnsetInit::get()); 155784c287e3SPeter Collingbourne assert(Value && "Cannot create unset value for current type!"); 155884c287e3SPeter Collingbourne } 155984c287e3SPeter Collingbourne 156084c287e3SPeter Collingbourne const std::string &RecordVal::getName() const { 1561ed5a9508SCraig Topper return cast<StringInit>(Name)->getValue(); 156284c287e3SPeter Collingbourne } 156384c287e3SPeter Collingbourne 156484c287e3SPeter Collingbourne void RecordVal::dump() const { errs() << *this; } 156584c287e3SPeter Collingbourne 156684c287e3SPeter Collingbourne void RecordVal::print(raw_ostream &OS, bool PrintSem) const { 156784c287e3SPeter Collingbourne if (getPrefix()) OS << "field "; 1568658a4b71SDavid Greene OS << *getType() << " " << getNameInitAsString(); 156984c287e3SPeter Collingbourne 157084c287e3SPeter Collingbourne if (getValue()) 157184c287e3SPeter Collingbourne OS << " = " << *getValue(); 157284c287e3SPeter Collingbourne 157384c287e3SPeter Collingbourne if (PrintSem) OS << ";\n"; 157484c287e3SPeter Collingbourne } 157584c287e3SPeter Collingbourne 157684c287e3SPeter Collingbourne unsigned Record::LastID = 0; 157784c287e3SPeter Collingbourne 1578b77fc0d9SDavid Greene void Record::init() { 1579b77fc0d9SDavid Greene checkName(); 1580d699161aSDavid Greene 1581d699161aSDavid Greene // Every record potentially has a def at the top. This value is 1582d699161aSDavid Greene // replaced with the top-level def name at instantiation time. 1583d699161aSDavid Greene RecordVal DN("NAME", StringRecTy::get(), 0); 1584d699161aSDavid Greene addValue(DN); 1585b77fc0d9SDavid Greene } 1586b77fc0d9SDavid Greene 158784c287e3SPeter Collingbourne void Record::checkName() { 158884c287e3SPeter Collingbourne // Ensure the record name has string type. 1589ed5a9508SCraig Topper const TypedInit *TypedName = cast<const TypedInit>(Name); 159084c287e3SPeter Collingbourne RecTy *Type = TypedName->getType(); 1591a475e433SSean Silva if (!isa<StringRecTy>(Type)) 1592635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record name is not a string!"); 159384c287e3SPeter Collingbourne } 159484c287e3SPeter Collingbourne 159584c287e3SPeter Collingbourne DefInit *Record::getDefInit() { 159617b3a378SCraig Topper static DenseMap<Record *, std::unique_ptr<DefInit>> ThePool; 159717b3a378SCraig Topper if (TheInit) 159884c287e3SPeter Collingbourne return TheInit; 159917b3a378SCraig Topper 160017b3a378SCraig Topper std::unique_ptr<DefInit> &I = ThePool[this]; 160117b3a378SCraig Topper if (!I) I.reset(new DefInit(this, new RecordRecTy(this))); 160217b3a378SCraig Topper return I.get(); 160384c287e3SPeter Collingbourne } 160484c287e3SPeter Collingbourne 160584c287e3SPeter Collingbourne const std::string &Record::getName() const { 1606ed5a9508SCraig Topper return cast<StringInit>(Name)->getValue(); 160784c287e3SPeter Collingbourne } 160884c287e3SPeter Collingbourne 160984c287e3SPeter Collingbourne void Record::setName(Init *NewName) { 1610f651a2a5SDavid Greene Name = NewName; 161184c287e3SPeter Collingbourne checkName(); 161284c287e3SPeter Collingbourne // DO NOT resolve record values to the name at this point because 161384c287e3SPeter Collingbourne // there might be default values for arguments of this def. Those 161484c287e3SPeter Collingbourne // arguments might not have been resolved yet so we don't want to 161584c287e3SPeter Collingbourne // prematurely assume values for those arguments were not passed to 161684c287e3SPeter Collingbourne // this def. 161784c287e3SPeter Collingbourne // 161884c287e3SPeter Collingbourne // Nonetheless, it may be that some of this Record's values 161984c287e3SPeter Collingbourne // reference the record name. Indeed, the reason for having the 162084c287e3SPeter Collingbourne // record name be an Init is to provide this flexibility. The extra 162184c287e3SPeter Collingbourne // resolve steps after completely instantiating defs takes care of 162284c287e3SPeter Collingbourne // this. See TGParser::ParseDef and TGParser::ParseDefm. 162384c287e3SPeter Collingbourne } 162484c287e3SPeter Collingbourne 162584c287e3SPeter Collingbourne void Record::setName(const std::string &Name) { 162684c287e3SPeter Collingbourne setName(StringInit::get(Name)); 162784c287e3SPeter Collingbourne } 162884c287e3SPeter Collingbourne 162984c287e3SPeter Collingbourne /// resolveReferencesTo - If anything in this record refers to RV, replace the 163084c287e3SPeter Collingbourne /// reference to RV with the RHS of RV. If RV is null, we resolve all possible 163184c287e3SPeter Collingbourne /// references. 163284c287e3SPeter Collingbourne void Record::resolveReferencesTo(const RecordVal *RV) { 163384c287e3SPeter Collingbourne for (unsigned i = 0, e = Values.size(); i != e; ++i) { 1634aa0f752fSJakob Stoklund Olesen if (RV == &Values[i]) // Skip resolve the same field as the given one 1635aa0f752fSJakob Stoklund Olesen continue; 163684c287e3SPeter Collingbourne if (Init *V = Values[i].getValue()) 1637026f8333SMichael Liao if (Values[i].setValue(V->resolveReferences(*this, RV))) 163885c07007SCraig Topper PrintFatalError(getLoc(), "Invalid value is found when setting '" + 163985c07007SCraig Topper Values[i].getNameInitAsString() + 164085c07007SCraig Topper "' after resolving references" + 164185c07007SCraig Topper (RV ? " against '" + RV->getNameInitAsString() + 164285c07007SCraig Topper "' of (" + RV->getValue()->getAsUnquotedString() + 164385c07007SCraig Topper ")" 164485c07007SCraig Topper : "") + "\n"); 164584c287e3SPeter Collingbourne } 16469effff28SDavid Greene Init *OldName = getNameInit(); 16479effff28SDavid Greene Init *NewName = Name->resolveReferences(*this, RV); 16489effff28SDavid Greene if (NewName != OldName) { 16499effff28SDavid Greene // Re-register with RecordKeeper. 16509effff28SDavid Greene setName(NewName); 16519effff28SDavid Greene } 165284c287e3SPeter Collingbourne } 165384c287e3SPeter Collingbourne 165484c287e3SPeter Collingbourne void Record::dump() const { errs() << *this; } 165584c287e3SPeter Collingbourne 165684c287e3SPeter Collingbourne raw_ostream &llvm::operator<<(raw_ostream &OS, const Record &R) { 165798be78aeSDavid Greene OS << R.getNameInitAsString(); 165884c287e3SPeter Collingbourne 1659db10e691SDavid Greene const std::vector<Init *> &TArgs = R.getTemplateArgs(); 166084c287e3SPeter Collingbourne if (!TArgs.empty()) { 166184c287e3SPeter Collingbourne OS << "<"; 166284c287e3SPeter Collingbourne for (unsigned i = 0, e = TArgs.size(); i != e; ++i) { 166384c287e3SPeter Collingbourne if (i) OS << ", "; 166484c287e3SPeter Collingbourne const RecordVal *RV = R.getValue(TArgs[i]); 166584c287e3SPeter Collingbourne assert(RV && "Template argument record not found??"); 166684c287e3SPeter Collingbourne RV->print(OS, false); 166784c287e3SPeter Collingbourne } 166884c287e3SPeter Collingbourne OS << ">"; 166984c287e3SPeter Collingbourne } 167084c287e3SPeter Collingbourne 167184c287e3SPeter Collingbourne OS << " {"; 167284c287e3SPeter Collingbourne const std::vector<Record*> &SC = R.getSuperClasses(); 167384c287e3SPeter Collingbourne if (!SC.empty()) { 167484c287e3SPeter Collingbourne OS << "\t//"; 167584c287e3SPeter Collingbourne for (unsigned i = 0, e = SC.size(); i != e; ++i) 1676077d84e6SDavid Greene OS << " " << SC[i]->getNameInitAsString(); 167784c287e3SPeter Collingbourne } 167884c287e3SPeter Collingbourne OS << "\n"; 167984c287e3SPeter Collingbourne 168084c287e3SPeter Collingbourne const std::vector<RecordVal> &Vals = R.getValues(); 168184c287e3SPeter Collingbourne for (unsigned i = 0, e = Vals.size(); i != e; ++i) 168284c287e3SPeter Collingbourne if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName())) 168384c287e3SPeter Collingbourne OS << Vals[i]; 168484c287e3SPeter Collingbourne for (unsigned i = 0, e = Vals.size(); i != e; ++i) 168584c287e3SPeter Collingbourne if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName())) 168684c287e3SPeter Collingbourne OS << Vals[i]; 168784c287e3SPeter Collingbourne 168884c287e3SPeter Collingbourne return OS << "}\n"; 168984c287e3SPeter Collingbourne } 169084c287e3SPeter Collingbourne 169184c287e3SPeter Collingbourne /// getValueInit - Return the initializer for a value with the specified name, 1692635debe8SJoerg Sonnenberger /// or abort if the field does not exist. 169384c287e3SPeter Collingbourne /// 169484c287e3SPeter Collingbourne Init *Record::getValueInit(StringRef FieldName) const { 169584c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1696011817a0SCraig Topper if (!R || !R->getValue()) 1697635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 169848e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 169984c287e3SPeter Collingbourne return R->getValue(); 170084c287e3SPeter Collingbourne } 170184c287e3SPeter Collingbourne 170284c287e3SPeter Collingbourne 170384c287e3SPeter Collingbourne /// getValueAsString - This method looks up the specified field and returns its 1704635debe8SJoerg Sonnenberger /// value as a string, aborts if the field does not exist or if 170584c287e3SPeter Collingbourne /// the value is not a string. 170684c287e3SPeter Collingbourne /// 170784c287e3SPeter Collingbourne std::string Record::getValueAsString(StringRef FieldName) const { 170884c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1709011817a0SCraig Topper if (!R || !R->getValue()) 1710635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 171148e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 171284c287e3SPeter Collingbourne 1713fb509ed1SSean Silva if (StringInit *SI = dyn_cast<StringInit>(R->getValue())) 171484c287e3SPeter Collingbourne return SI->getValue(); 1715635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 171648e7e85dSBenjamin Kramer FieldName + "' does not have a string initializer!"); 171784c287e3SPeter Collingbourne } 171884c287e3SPeter Collingbourne 171984c287e3SPeter Collingbourne /// getValueAsBitsInit - This method looks up the specified field and returns 1720635debe8SJoerg Sonnenberger /// its value as a BitsInit, aborts if the field does not exist or if 1721635debe8SJoerg Sonnenberger /// the value is not the right type. 172284c287e3SPeter Collingbourne /// 172384c287e3SPeter Collingbourne BitsInit *Record::getValueAsBitsInit(StringRef FieldName) const { 172484c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1725011817a0SCraig Topper if (!R || !R->getValue()) 1726635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 172748e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 172884c287e3SPeter Collingbourne 1729fb509ed1SSean Silva if (BitsInit *BI = dyn_cast<BitsInit>(R->getValue())) 173084c287e3SPeter Collingbourne return BI; 1731635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 173248e7e85dSBenjamin Kramer FieldName + "' does not have a BitsInit initializer!"); 173384c287e3SPeter Collingbourne } 173484c287e3SPeter Collingbourne 173584c287e3SPeter Collingbourne /// getValueAsListInit - This method looks up the specified field and returns 1736635debe8SJoerg Sonnenberger /// its value as a ListInit, aborting if the field does not exist or if 1737635debe8SJoerg Sonnenberger /// the value is not the right type. 173884c287e3SPeter Collingbourne /// 173984c287e3SPeter Collingbourne ListInit *Record::getValueAsListInit(StringRef FieldName) const { 174084c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1741011817a0SCraig Topper if (!R || !R->getValue()) 1742635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 174348e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 174484c287e3SPeter Collingbourne 1745fb509ed1SSean Silva if (ListInit *LI = dyn_cast<ListInit>(R->getValue())) 174684c287e3SPeter Collingbourne return LI; 1747635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 174848e7e85dSBenjamin Kramer FieldName + "' does not have a list initializer!"); 174984c287e3SPeter Collingbourne } 175084c287e3SPeter Collingbourne 175184c287e3SPeter Collingbourne /// getValueAsListOfDefs - This method looks up the specified field and returns 1752635debe8SJoerg Sonnenberger /// its value as a vector of records, aborting if the field does not exist 1753635debe8SJoerg Sonnenberger /// or if the value is not the right type. 175484c287e3SPeter Collingbourne /// 175584c287e3SPeter Collingbourne std::vector<Record*> 175684c287e3SPeter Collingbourne Record::getValueAsListOfDefs(StringRef FieldName) const { 175784c287e3SPeter Collingbourne ListInit *List = getValueAsListInit(FieldName); 175884c287e3SPeter Collingbourne std::vector<Record*> Defs; 175984c287e3SPeter Collingbourne for (unsigned i = 0; i < List->getSize(); i++) { 1760a9642b4eSCraig Topper if (DefInit *DI = dyn_cast<DefInit>(List->getElement(i))) 176184c287e3SPeter Collingbourne Defs.push_back(DI->getDef()); 1762a9642b4eSCraig Topper else 1763635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 176448e7e85dSBenjamin Kramer FieldName + "' list is not entirely DefInit!"); 176584c287e3SPeter Collingbourne } 176684c287e3SPeter Collingbourne return Defs; 176784c287e3SPeter Collingbourne } 176884c287e3SPeter Collingbourne 176984c287e3SPeter Collingbourne /// getValueAsInt - This method looks up the specified field and returns its 1770635debe8SJoerg Sonnenberger /// value as an int64_t, aborting if the field does not exist or if the value 1771635debe8SJoerg Sonnenberger /// is not the right type. 177284c287e3SPeter Collingbourne /// 177384c287e3SPeter Collingbourne int64_t Record::getValueAsInt(StringRef FieldName) const { 177484c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1775011817a0SCraig Topper if (!R || !R->getValue()) 1776635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 177748e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 177884c287e3SPeter Collingbourne 1779fb509ed1SSean Silva if (IntInit *II = dyn_cast<IntInit>(R->getValue())) 178084c287e3SPeter Collingbourne return II->getValue(); 1781635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 178248e7e85dSBenjamin Kramer FieldName + "' does not have an int initializer!"); 178384c287e3SPeter Collingbourne } 178484c287e3SPeter Collingbourne 178584c287e3SPeter Collingbourne /// getValueAsListOfInts - This method looks up the specified field and returns 1786635debe8SJoerg Sonnenberger /// its value as a vector of integers, aborting if the field does not exist or 1787635debe8SJoerg Sonnenberger /// if the value is not the right type. 178884c287e3SPeter Collingbourne /// 178984c287e3SPeter Collingbourne std::vector<int64_t> 179084c287e3SPeter Collingbourne Record::getValueAsListOfInts(StringRef FieldName) const { 179184c287e3SPeter Collingbourne ListInit *List = getValueAsListInit(FieldName); 179284c287e3SPeter Collingbourne std::vector<int64_t> Ints; 179384c287e3SPeter Collingbourne for (unsigned i = 0; i < List->getSize(); i++) { 1794a9642b4eSCraig Topper if (IntInit *II = dyn_cast<IntInit>(List->getElement(i))) 179584c287e3SPeter Collingbourne Ints.push_back(II->getValue()); 1796a9642b4eSCraig Topper else 1797635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 179848e7e85dSBenjamin Kramer FieldName + "' does not have a list of ints initializer!"); 179984c287e3SPeter Collingbourne } 180084c287e3SPeter Collingbourne return Ints; 180184c287e3SPeter Collingbourne } 180284c287e3SPeter Collingbourne 180384c287e3SPeter Collingbourne /// getValueAsListOfStrings - This method looks up the specified field and 1804635debe8SJoerg Sonnenberger /// returns its value as a vector of strings, aborting if the field does not 1805635debe8SJoerg Sonnenberger /// exist or if the value is not the right type. 180684c287e3SPeter Collingbourne /// 180784c287e3SPeter Collingbourne std::vector<std::string> 180884c287e3SPeter Collingbourne Record::getValueAsListOfStrings(StringRef FieldName) const { 180984c287e3SPeter Collingbourne ListInit *List = getValueAsListInit(FieldName); 181084c287e3SPeter Collingbourne std::vector<std::string> Strings; 181184c287e3SPeter Collingbourne for (unsigned i = 0; i < List->getSize(); i++) { 1812a9642b4eSCraig Topper if (StringInit *II = dyn_cast<StringInit>(List->getElement(i))) 181384c287e3SPeter Collingbourne Strings.push_back(II->getValue()); 1814a9642b4eSCraig Topper else 1815635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 181648e7e85dSBenjamin Kramer FieldName + "' does not have a list of strings initializer!"); 181784c287e3SPeter Collingbourne } 181884c287e3SPeter Collingbourne return Strings; 181984c287e3SPeter Collingbourne } 182084c287e3SPeter Collingbourne 182184c287e3SPeter Collingbourne /// getValueAsDef - This method looks up the specified field and returns its 1822635debe8SJoerg Sonnenberger /// value as a Record, aborting if the field does not exist or if the value 1823635debe8SJoerg Sonnenberger /// is not the right type. 182484c287e3SPeter Collingbourne /// 182584c287e3SPeter Collingbourne Record *Record::getValueAsDef(StringRef FieldName) const { 182684c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1827011817a0SCraig Topper if (!R || !R->getValue()) 1828635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 182948e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 183084c287e3SPeter Collingbourne 1831fb509ed1SSean Silva if (DefInit *DI = dyn_cast<DefInit>(R->getValue())) 183284c287e3SPeter Collingbourne return DI->getDef(); 1833635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 183448e7e85dSBenjamin Kramer FieldName + "' does not have a def initializer!"); 183584c287e3SPeter Collingbourne } 183684c287e3SPeter Collingbourne 183784c287e3SPeter Collingbourne /// getValueAsBit - This method looks up the specified field and returns its 1838635debe8SJoerg Sonnenberger /// value as a bit, aborting if the field does not exist or if the value is 1839635debe8SJoerg Sonnenberger /// not the right type. 184084c287e3SPeter Collingbourne /// 184184c287e3SPeter Collingbourne bool Record::getValueAsBit(StringRef FieldName) const { 184284c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1843011817a0SCraig Topper if (!R || !R->getValue()) 1844635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 184548e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 184684c287e3SPeter Collingbourne 1847fb509ed1SSean Silva if (BitInit *BI = dyn_cast<BitInit>(R->getValue())) 184884c287e3SPeter Collingbourne return BI->getValue(); 1849635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 185048e7e85dSBenjamin Kramer FieldName + "' does not have a bit initializer!"); 185184c287e3SPeter Collingbourne } 185284c287e3SPeter Collingbourne 1853af507bf4SJakob Stoklund Olesen bool Record::getValueAsBitOrUnset(StringRef FieldName, bool &Unset) const { 1854af507bf4SJakob Stoklund Olesen const RecordVal *R = getValue(FieldName); 1855011817a0SCraig Topper if (!R || !R->getValue()) 1856635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 1857635debe8SJoerg Sonnenberger "' does not have a field named `" + FieldName.str() + "'!\n"); 1858af507bf4SJakob Stoklund Olesen 18591bf3d1f5SCraig Topper if (isa<UnsetInit>(R->getValue())) { 1860af507bf4SJakob Stoklund Olesen Unset = true; 1861af507bf4SJakob Stoklund Olesen return false; 1862af507bf4SJakob Stoklund Olesen } 1863af507bf4SJakob Stoklund Olesen Unset = false; 1864fb509ed1SSean Silva if (BitInit *BI = dyn_cast<BitInit>(R->getValue())) 1865af507bf4SJakob Stoklund Olesen return BI->getValue(); 1866635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 186748e7e85dSBenjamin Kramer FieldName + "' does not have a bit initializer!"); 1868af507bf4SJakob Stoklund Olesen } 1869af507bf4SJakob Stoklund Olesen 187084c287e3SPeter Collingbourne /// getValueAsDag - This method looks up the specified field and returns its 1871635debe8SJoerg Sonnenberger /// value as an Dag, aborting if the field does not exist or if the value is 1872635debe8SJoerg Sonnenberger /// not the right type. 187384c287e3SPeter Collingbourne /// 187484c287e3SPeter Collingbourne DagInit *Record::getValueAsDag(StringRef FieldName) const { 187584c287e3SPeter Collingbourne const RecordVal *R = getValue(FieldName); 1876011817a0SCraig Topper if (!R || !R->getValue()) 1877635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + 187848e7e85dSBenjamin Kramer "' does not have a field named `" + FieldName + "'!\n"); 187984c287e3SPeter Collingbourne 1880fb509ed1SSean Silva if (DagInit *DI = dyn_cast<DagInit>(R->getValue())) 188184c287e3SPeter Collingbourne return DI; 1882635debe8SJoerg Sonnenberger PrintFatalError(getLoc(), "Record `" + getName() + "', field `" + 188348e7e85dSBenjamin Kramer FieldName + "' does not have a dag initializer!"); 188484c287e3SPeter Collingbourne } 188584c287e3SPeter Collingbourne 188684c287e3SPeter Collingbourne 188784c287e3SPeter Collingbourne void MultiClass::dump() const { 188884c287e3SPeter Collingbourne errs() << "Record:\n"; 188984c287e3SPeter Collingbourne Rec.dump(); 189084c287e3SPeter Collingbourne 189184c287e3SPeter Collingbourne errs() << "Defs:\n"; 1892a9642b4eSCraig Topper for (const auto &Proto : DefPrototypes) 1893119998dbSCraig Topper Proto->dump(); 189484c287e3SPeter Collingbourne } 189584c287e3SPeter Collingbourne 189684c287e3SPeter Collingbourne 189784c287e3SPeter Collingbourne void RecordKeeper::dump() const { errs() << *this; } 189884c287e3SPeter Collingbourne 189984c287e3SPeter Collingbourne raw_ostream &llvm::operator<<(raw_ostream &OS, const RecordKeeper &RK) { 190084c287e3SPeter Collingbourne OS << "------------- Classes -----------------\n"; 1901a0ff540bSCraig Topper for (const auto &C : RK.getClasses()) 190280f0e432SDylan Noblesmith OS << "class " << *C.second; 190384c287e3SPeter Collingbourne 190484c287e3SPeter Collingbourne OS << "------------- Defs -----------------\n"; 1905a0ff540bSCraig Topper for (const auto &D : RK.getDefs()) 190680f0e432SDylan Noblesmith OS << "def " << *D.second; 190784c287e3SPeter Collingbourne return OS; 190884c287e3SPeter Collingbourne } 190984c287e3SPeter Collingbourne 191084c287e3SPeter Collingbourne 191184c287e3SPeter Collingbourne /// getAllDerivedDefinitions - This method returns all concrete definitions 191284c287e3SPeter Collingbourne /// that derive from the specified class name. If a class with the specified 191384c287e3SPeter Collingbourne /// name does not exist, an error is printed and true is returned. 191484c287e3SPeter Collingbourne std::vector<Record*> 191584c287e3SPeter Collingbourne RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const { 191684c287e3SPeter Collingbourne Record *Class = getClass(ClassName); 191784c287e3SPeter Collingbourne if (!Class) 1918635debe8SJoerg Sonnenberger PrintFatalError("ERROR: Couldn't find the `" + ClassName + "' class!\n"); 191984c287e3SPeter Collingbourne 192084c287e3SPeter Collingbourne std::vector<Record*> Defs; 192180f0e432SDylan Noblesmith for (const auto &D : getDefs()) 192280f0e432SDylan Noblesmith if (D.second->isSubClassOf(Class)) 1923085fc4d6SDylan Noblesmith Defs.push_back(D.second.get()); 192484c287e3SPeter Collingbourne 192584c287e3SPeter Collingbourne return Defs; 192684c287e3SPeter Collingbourne } 192784c287e3SPeter Collingbourne 1928e714512cSDavid Greene /// QualifyName - Return an Init with a qualifier prefix referring 1929e714512cSDavid Greene /// to CurRec's name. 1930e714512cSDavid Greene Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass, 1931e714512cSDavid Greene Init *Name, const std::string &Scoper) { 1932619c4e57SBenjamin Kramer RecTy *Type = cast<TypedInit>(Name)->getType(); 1933e714512cSDavid Greene 1934e714512cSDavid Greene BinOpInit *NewName = 1935e714512cSDavid Greene BinOpInit::get(BinOpInit::STRCONCAT, 1936e714512cSDavid Greene BinOpInit::get(BinOpInit::STRCONCAT, 1937e714512cSDavid Greene CurRec.getNameInit(), 1938e714512cSDavid Greene StringInit::get(Scoper), 1939e714512cSDavid Greene Type)->Fold(&CurRec, CurMultiClass), 1940e714512cSDavid Greene Name, 1941e714512cSDavid Greene Type); 1942e714512cSDavid Greene 1943e714512cSDavid Greene if (CurMultiClass && Scoper != "::") { 1944e714512cSDavid Greene NewName = 1945e714512cSDavid Greene BinOpInit::get(BinOpInit::STRCONCAT, 1946e714512cSDavid Greene BinOpInit::get(BinOpInit::STRCONCAT, 1947e714512cSDavid Greene CurMultiClass->Rec.getNameInit(), 1948e714512cSDavid Greene StringInit::get("::"), 1949e714512cSDavid Greene Type)->Fold(&CurRec, CurMultiClass), 1950e714512cSDavid Greene NewName->Fold(&CurRec, CurMultiClass), 1951e714512cSDavid Greene Type); 1952e714512cSDavid Greene } 1953e714512cSDavid Greene 1954e714512cSDavid Greene return NewName->Fold(&CurRec, CurMultiClass); 1955e714512cSDavid Greene } 1956e714512cSDavid Greene 1957e714512cSDavid Greene /// QualifyName - Return an Init with a qualifier prefix referring 1958e714512cSDavid Greene /// to CurRec's name. 1959e714512cSDavid Greene Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass, 1960e714512cSDavid Greene const std::string &Name, 1961e714512cSDavid Greene const std::string &Scoper) { 1962e714512cSDavid Greene return QualifyName(CurRec, CurMultiClass, StringInit::get(Name), Scoper); 1963e714512cSDavid Greene } 1964