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 
8984c287e3SPeter Collingbourne void RecTy::dump() const { print(errs()); }
9084c287e3SPeter Collingbourne 
9184c287e3SPeter Collingbourne ListRecTy *RecTy::getListTy() {
9284c287e3SPeter Collingbourne   if (!ListTy)
9317b3a378SCraig Topper     ListTy.reset(new ListRecTy(this));
9417b3a378SCraig Topper   return ListTy.get();
9584c287e3SPeter Collingbourne }
9684c287e3SPeter Collingbourne 
9715864f15SCraig Topper bool RecTy::typeIsConvertibleTo(const RecTy *RHS) const {
9893be7d59SSean Silva   assert(RHS && "NULL pointer");
9993be7d59SSean Silva   return Kind == RHS->getRecTyKind();
10093be7d59SSean Silva }
10193be7d59SSean Silva 
10215864f15SCraig Topper bool BitRecTy::typeIsConvertibleTo(const RecTy *RHS) const{
10315864f15SCraig Topper   if(RecTy::typeIsConvertibleTo(RHS) || RHS->getRecTyKind() == IntRecTyKind)
10493be7d59SSean Silva     return true;
10593be7d59SSean Silva   if(const BitsRecTy *BitsTy = dyn_cast<BitsRecTy>(RHS))
10693be7d59SSean Silva     return BitsTy->getNumBits() == 1;
10793be7d59SSean Silva   return false;
10893be7d59SSean Silva }
10993be7d59SSean Silva 
11084c287e3SPeter Collingbourne BitsRecTy *BitsRecTy::get(unsigned Sz) {
11117b3a378SCraig Topper   static std::vector<std::unique_ptr<BitsRecTy>> Shared;
11284c287e3SPeter Collingbourne   if (Sz >= Shared.size())
11384c287e3SPeter Collingbourne     Shared.resize(Sz + 1);
11417b3a378SCraig Topper   std::unique_ptr<BitsRecTy> &Ty = Shared[Sz];
11584c287e3SPeter Collingbourne   if (!Ty)
11617b3a378SCraig Topper     Ty.reset(new BitsRecTy(Sz));
11717b3a378SCraig Topper   return Ty.get();
11884c287e3SPeter Collingbourne }
11984c287e3SPeter Collingbourne 
12084c287e3SPeter Collingbourne std::string BitsRecTy::getAsString() const {
12184c287e3SPeter Collingbourne   return "bits<" + utostr(Size) + ">";
12284c287e3SPeter Collingbourne }
12384c287e3SPeter Collingbourne 
12415864f15SCraig Topper bool BitsRecTy::typeIsConvertibleTo(const RecTy *RHS) const {
12515864f15SCraig Topper   if (RecTy::typeIsConvertibleTo(RHS)) //argument and the sender are same type
12693be7d59SSean Silva     return cast<BitsRecTy>(RHS)->Size == Size;
12793be7d59SSean Silva   RecTyKind kind = RHS->getRecTyKind();
12893be7d59SSean Silva   return (kind == BitRecTyKind && Size == 1) || (kind == IntRecTyKind);
12993be7d59SSean Silva }
13093be7d59SSean Silva 
13115864f15SCraig Topper bool IntRecTy::typeIsConvertibleTo(const RecTy *RHS) const {
13293be7d59SSean Silva   RecTyKind kind = RHS->getRecTyKind();
13393be7d59SSean Silva   return kind==BitRecTyKind || kind==BitsRecTyKind || kind==IntRecTyKind;
13493be7d59SSean Silva }
13593be7d59SSean Silva 
1366548196cSCraig Topper std::string StringRecTy::getAsString() const {
1376548196cSCraig Topper   return "string";
1386548196cSCraig Topper }
13984c287e3SPeter Collingbourne 
14084c287e3SPeter Collingbourne std::string ListRecTy::getAsString() const {
14184c287e3SPeter Collingbourne   return "list<" + Ty->getAsString() + ">";
14284c287e3SPeter Collingbourne }
14384c287e3SPeter Collingbourne 
14415864f15SCraig Topper bool ListRecTy::typeIsConvertibleTo(const RecTy *RHS) const {
14515864f15SCraig Topper   if (const auto *ListTy = dyn_cast<ListRecTy>(RHS))
14615864f15SCraig Topper     return Ty->typeIsConvertibleTo(ListTy->getElementType());
14793be7d59SSean Silva   return false;
14893be7d59SSean Silva }
14993be7d59SSean Silva 
1506548196cSCraig Topper std::string DagRecTy::getAsString() const {
1516548196cSCraig Topper   return "dag";
1526548196cSCraig Topper }
1536548196cSCraig Topper 
15484c287e3SPeter Collingbourne RecordRecTy *RecordRecTy::get(Record *R) {
155b534eabfSCraig Topper   return dyn_cast<RecordRecTy>(R->getDefInit()->getType());
15684c287e3SPeter Collingbourne }
15784c287e3SPeter Collingbourne 
15884c287e3SPeter Collingbourne std::string RecordRecTy::getAsString() const {
15984c287e3SPeter Collingbourne   return Rec->getName();
16084c287e3SPeter Collingbourne }
16184c287e3SPeter Collingbourne 
16215864f15SCraig Topper bool RecordRecTy::typeIsConvertibleTo(const RecTy *RHS) const {
16393be7d59SSean Silva   const RecordRecTy *RTy = dyn_cast<RecordRecTy>(RHS);
16493be7d59SSean Silva   if (!RTy)
16593be7d59SSean Silva     return false;
16693be7d59SSean Silva 
16715864f15SCraig Topper   if (RTy->getRecord() == Rec || Rec->isSubClassOf(RTy->getRecord()))
16884c287e3SPeter Collingbourne     return true;
16984c287e3SPeter Collingbourne 
17015864f15SCraig Topper   for (Record *SC : RTy->getRecord()->getSuperClasses())
17115864f15SCraig Topper     if (Rec->isSubClassOf(SC))
17284c287e3SPeter Collingbourne       return true;
17384c287e3SPeter Collingbourne 
17484c287e3SPeter Collingbourne   return false;
17584c287e3SPeter Collingbourne }
17684c287e3SPeter Collingbourne 
17784c287e3SPeter Collingbourne /// resolveTypes - Find a common type that T1 and T2 convert to.
178e8005f90SCraig Topper /// Return null if no such type exists.
17984c287e3SPeter Collingbourne ///
18084c287e3SPeter Collingbourne RecTy *llvm::resolveTypes(RecTy *T1, RecTy *T2) {
1818f43c6f0SSean Silva   if (T1->typeIsConvertibleTo(T2))
1828f43c6f0SSean Silva     return T2;
1838f43c6f0SSean Silva   if (T2->typeIsConvertibleTo(T1))
1848f43c6f0SSean Silva     return T1;
1858f43c6f0SSean Silva 
18684c287e3SPeter Collingbourne   // If one is a Record type, check superclasses
18798c61711SSean Silva   if (RecordRecTy *RecTy1 = dyn_cast<RecordRecTy>(T1)) {
18884c287e3SPeter Collingbourne     // See if T2 inherits from a type T1 also inherits from
1891f429e49SCraig Topper     for (Record *SuperRec1 : RecTy1->getRecord()->getSuperClasses()) {
1901f429e49SCraig Topper       RecordRecTy *SuperRecTy1 = RecordRecTy::get(SuperRec1);
19184c287e3SPeter Collingbourne       RecTy *NewType1 = resolveTypes(SuperRecTy1, T2);
192d0599130SCraig Topper       if (NewType1)
19384c287e3SPeter Collingbourne         return NewType1;
19484c287e3SPeter Collingbourne     }
19584c287e3SPeter Collingbourne   }
19698c61711SSean Silva   if (RecordRecTy *RecTy2 = dyn_cast<RecordRecTy>(T2)) {
19784c287e3SPeter Collingbourne     // See if T1 inherits from a type T2 also inherits from
1981f429e49SCraig Topper     for (Record *SuperRec2 : RecTy2->getRecord()->getSuperClasses()) {
1991f429e49SCraig Topper       RecordRecTy *SuperRecTy2 = RecordRecTy::get(SuperRec2);
20084c287e3SPeter Collingbourne       RecTy *NewType2 = resolveTypes(T1, SuperRecTy2);
201d0599130SCraig Topper       if (NewType2)
20284c287e3SPeter Collingbourne         return NewType2;
20384c287e3SPeter Collingbourne     }
20484c287e3SPeter Collingbourne   }
205011817a0SCraig Topper   return nullptr;
20684c287e3SPeter Collingbourne }
20784c287e3SPeter Collingbourne 
20884c287e3SPeter Collingbourne 
20984c287e3SPeter Collingbourne //===----------------------------------------------------------------------===//
21084c287e3SPeter Collingbourne //    Initializer implementations
21184c287e3SPeter Collingbourne //===----------------------------------------------------------------------===//
21284c287e3SPeter Collingbourne 
213a379b181SDavid Blaikie void Init::anchor() { }
21484c287e3SPeter Collingbourne void Init::dump() const { return print(errs()); }
21584c287e3SPeter Collingbourne 
21684c287e3SPeter Collingbourne UnsetInit *UnsetInit::get() {
21784c287e3SPeter Collingbourne   static UnsetInit TheInit;
21884c287e3SPeter Collingbourne   return &TheInit;
21984c287e3SPeter Collingbourne }
22084c287e3SPeter Collingbourne 
22195819069SCraig Topper Init *UnsetInit::convertInitializerTo(RecTy *Ty) const {
22295819069SCraig Topper   if (auto *BRT = dyn_cast<BitsRecTy>(Ty)) {
22395819069SCraig Topper     SmallVector<Init *, 16> NewBits(BRT->getNumBits());
22495819069SCraig Topper 
22595819069SCraig Topper     for (unsigned i = 0; i != BRT->getNumBits(); ++i)
22695819069SCraig Topper       NewBits[i] = UnsetInit::get();
22795819069SCraig Topper 
22895819069SCraig Topper     return BitsInit::get(NewBits);
22995819069SCraig Topper   }
23095819069SCraig Topper 
23195819069SCraig Topper   // All other types can just be returned.
23295819069SCraig Topper   return const_cast<UnsetInit *>(this);
23395819069SCraig Topper }
234a379b181SDavid Blaikie 
23584c287e3SPeter Collingbourne BitInit *BitInit::get(bool V) {
23684c287e3SPeter Collingbourne   static BitInit True(true);
23784c287e3SPeter Collingbourne   static BitInit False(false);
23884c287e3SPeter Collingbourne 
23984c287e3SPeter Collingbourne   return V ? &True : &False;
24084c287e3SPeter Collingbourne }
24184c287e3SPeter Collingbourne 
24295819069SCraig Topper Init *BitInit::convertInitializerTo(RecTy *Ty) const {
24395819069SCraig Topper   if (isa<BitRecTy>(Ty))
24495819069SCraig Topper     return const_cast<BitInit *>(this);
24595819069SCraig Topper 
24695819069SCraig Topper   if (isa<IntRecTy>(Ty))
24795819069SCraig Topper     return IntInit::get(getValue());
24895819069SCraig Topper 
24995819069SCraig Topper   if (auto *BRT = dyn_cast<BitsRecTy>(Ty)) {
25095819069SCraig Topper     // Can only convert single bit.
25195819069SCraig Topper     if (BRT->getNumBits() == 1)
25295819069SCraig Topper       return BitsInit::get(const_cast<BitInit *>(this));
25395819069SCraig Topper   }
25495819069SCraig Topper 
25595819069SCraig Topper   return nullptr;
25695819069SCraig Topper }
25795819069SCraig Topper 
25884c287e3SPeter Collingbourne static void
25984c287e3SPeter Collingbourne ProfileBitsInit(FoldingSetNodeID &ID, ArrayRef<Init *> Range) {
26084c287e3SPeter Collingbourne   ID.AddInteger(Range.size());
26184c287e3SPeter Collingbourne 
262119998dbSCraig Topper   for (Init *I : Range)
263119998dbSCraig Topper     ID.AddPointer(I);
26484c287e3SPeter Collingbourne }
26584c287e3SPeter Collingbourne 
26684c287e3SPeter Collingbourne BitsInit *BitsInit::get(ArrayRef<Init *> Range) {
26717b3a378SCraig Topper   static FoldingSet<BitsInit> ThePool;
26817b3a378SCraig Topper   static std::vector<std::unique_ptr<BitsInit>> TheActualPool;
26984c287e3SPeter Collingbourne 
27084c287e3SPeter Collingbourne   FoldingSetNodeID ID;
27184c287e3SPeter Collingbourne   ProfileBitsInit(ID, Range);
27284c287e3SPeter Collingbourne 
273011817a0SCraig Topper   void *IP = nullptr;
27484c287e3SPeter Collingbourne   if (BitsInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
27584c287e3SPeter Collingbourne     return I;
27684c287e3SPeter Collingbourne 
27784c287e3SPeter Collingbourne   BitsInit *I = new BitsInit(Range);
27884c287e3SPeter Collingbourne   ThePool.InsertNode(I, IP);
27917b3a378SCraig Topper   TheActualPool.push_back(std::unique_ptr<BitsInit>(I));
28084c287e3SPeter Collingbourne   return I;
28184c287e3SPeter Collingbourne }
28284c287e3SPeter Collingbourne 
28384c287e3SPeter Collingbourne void BitsInit::Profile(FoldingSetNodeID &ID) const {
28484c287e3SPeter Collingbourne   ProfileBitsInit(ID, Bits);
28584c287e3SPeter Collingbourne }
28684c287e3SPeter Collingbourne 
28795819069SCraig Topper Init *BitsInit::convertInitializerTo(RecTy *Ty) const {
28895819069SCraig Topper   if (isa<BitRecTy>(Ty)) {
28995819069SCraig Topper     if (getNumBits() != 1) return nullptr; // Only accept if just one bit!
29095819069SCraig Topper     return getBit(0);
29195819069SCraig Topper   }
29295819069SCraig Topper 
29395819069SCraig Topper   if (auto *BRT = dyn_cast<BitsRecTy>(Ty)) {
29495819069SCraig Topper     // If the number of bits is right, return it.  Otherwise we need to expand
29595819069SCraig Topper     // or truncate.
29695819069SCraig Topper     if (getNumBits() != BRT->getNumBits()) return nullptr;
29795819069SCraig Topper     return const_cast<BitsInit *>(this);
29895819069SCraig Topper   }
29995819069SCraig Topper 
30095819069SCraig Topper   if (isa<IntRecTy>(Ty)) {
30195819069SCraig Topper     int64_t Result = 0;
30295819069SCraig Topper     for (unsigned i = 0, e = getNumBits(); i != e; ++i)
30395819069SCraig Topper       if (auto *Bit = dyn_cast<BitInit>(getBit(i)))
30495819069SCraig Topper         Result |= static_cast<int64_t>(Bit->getValue()) << i;
30595819069SCraig Topper       else
30695819069SCraig Topper         return nullptr;
30795819069SCraig Topper     return IntInit::get(Result);
30895819069SCraig Topper   }
30995819069SCraig Topper 
31095819069SCraig Topper   return nullptr;
31195819069SCraig Topper }
31295819069SCraig Topper 
31384c287e3SPeter Collingbourne Init *
31484c287e3SPeter Collingbourne BitsInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
31584c287e3SPeter Collingbourne   SmallVector<Init *, 16> NewBits(Bits.size());
31684c287e3SPeter Collingbourne 
31784c287e3SPeter Collingbourne   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
31884c287e3SPeter Collingbourne     if (Bits[i] >= getNumBits())
319011817a0SCraig Topper       return nullptr;
32084c287e3SPeter Collingbourne     NewBits[i] = getBit(Bits[i]);
32184c287e3SPeter Collingbourne   }
32284c287e3SPeter Collingbourne   return BitsInit::get(NewBits);
32384c287e3SPeter Collingbourne }
32484c287e3SPeter Collingbourne 
32584c287e3SPeter Collingbourne std::string BitsInit::getAsString() const {
32684c287e3SPeter Collingbourne   std::string Result = "{ ";
32784c287e3SPeter Collingbourne   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
32884c287e3SPeter Collingbourne     if (i) Result += ", ";
32984c287e3SPeter Collingbourne     if (Init *Bit = getBit(e-i-1))
33084c287e3SPeter Collingbourne       Result += Bit->getAsString();
33184c287e3SPeter Collingbourne     else
33284c287e3SPeter Collingbourne       Result += "*";
33384c287e3SPeter Collingbourne   }
33484c287e3SPeter Collingbourne   return Result + " }";
33584c287e3SPeter Collingbourne }
33684c287e3SPeter Collingbourne 
337026f8333SMichael Liao // Fix bit initializer to preserve the behavior that bit reference from a unset
338026f8333SMichael Liao // bits initializer will resolve into VarBitInit to keep the field name and bit
339026f8333SMichael Liao // number used in targets with fixed insn length.
340026f8333SMichael Liao static Init *fixBitInit(const RecordVal *RV, Init *Before, Init *After) {
3411bf3d1f5SCraig Topper   if (RV || !isa<UnsetInit>(After))
342026f8333SMichael Liao     return After;
343026f8333SMichael Liao   return Before;
344026f8333SMichael Liao }
345026f8333SMichael Liao 
34684c287e3SPeter Collingbourne // resolveReferences - If there are any field references that refer to fields
34784c287e3SPeter Collingbourne // that have been filled in, we can propagate the values now.
34884c287e3SPeter Collingbourne //
34984c287e3SPeter Collingbourne Init *BitsInit::resolveReferences(Record &R, const RecordVal *RV) const {
35084c287e3SPeter Collingbourne   bool Changed = false;
35184c287e3SPeter Collingbourne   SmallVector<Init *, 16> NewBits(getNumBits());
35284c287e3SPeter Collingbourne 
353011817a0SCraig Topper   Init *CachedInit = nullptr;
354011817a0SCraig Topper   Init *CachedBitVar = nullptr;
355026f8333SMichael Liao   bool CachedBitVarChanged = false;
35684c287e3SPeter Collingbourne 
357026f8333SMichael Liao   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
358026f8333SMichael Liao     Init *CurBit = Bits[i];
359026f8333SMichael Liao     Init *CurBitVar = CurBit->getBitVar();
360026f8333SMichael Liao 
36184c287e3SPeter Collingbourne     NewBits[i] = CurBit;
362026f8333SMichael Liao 
363026f8333SMichael Liao     if (CurBitVar == CachedBitVar) {
364026f8333SMichael Liao       if (CachedBitVarChanged) {
365026f8333SMichael Liao         Init *Bit = CachedInit->getBit(CurBit->getBitNum());
366026f8333SMichael Liao         NewBits[i] = fixBitInit(RV, CurBit, Bit);
367026f8333SMichael Liao       }
368026f8333SMichael Liao       continue;
369026f8333SMichael Liao     }
370026f8333SMichael Liao     CachedBitVar = CurBitVar;
371026f8333SMichael Liao     CachedBitVarChanged = false;
372026f8333SMichael Liao 
373026f8333SMichael Liao     Init *B;
374026f8333SMichael Liao     do {
375026f8333SMichael Liao       B = CurBitVar;
376026f8333SMichael Liao       CurBitVar = CurBitVar->resolveReferences(R, RV);
377026f8333SMichael Liao       CachedBitVarChanged |= B != CurBitVar;
378026f8333SMichael Liao       Changed |= B != CurBitVar;
379026f8333SMichael Liao     } while (B != CurBitVar);
380026f8333SMichael Liao     CachedInit = CurBitVar;
381026f8333SMichael Liao 
382026f8333SMichael Liao     if (CachedBitVarChanged) {
383026f8333SMichael Liao       Init *Bit = CurBitVar->getBit(CurBit->getBitNum());
384026f8333SMichael Liao       NewBits[i] = fixBitInit(RV, CurBit, Bit);
385026f8333SMichael Liao     }
38684c287e3SPeter Collingbourne   }
38784c287e3SPeter Collingbourne 
38884c287e3SPeter Collingbourne   if (Changed)
38984c287e3SPeter Collingbourne     return BitsInit::get(NewBits);
39084c287e3SPeter Collingbourne 
39184c287e3SPeter Collingbourne   return const_cast<BitsInit *>(this);
39284c287e3SPeter Collingbourne }
39384c287e3SPeter Collingbourne 
39484c287e3SPeter Collingbourne IntInit *IntInit::get(int64_t V) {
3950e04bee8SCraig Topper   static DenseMap<int64_t, std::unique_ptr<IntInit>> ThePool;
39684c287e3SPeter Collingbourne 
3970e04bee8SCraig Topper   std::unique_ptr<IntInit> &I = ThePool[V];
3980e04bee8SCraig Topper   if (!I) I.reset(new IntInit(V));
3990e04bee8SCraig Topper   return I.get();
40084c287e3SPeter Collingbourne }
40184c287e3SPeter Collingbourne 
40284c287e3SPeter Collingbourne std::string IntInit::getAsString() const {
40384c287e3SPeter Collingbourne   return itostr(Value);
40484c287e3SPeter Collingbourne }
40584c287e3SPeter Collingbourne 
40695819069SCraig Topper /// canFitInBitfield - Return true if the number of bits is large enough to hold
40795819069SCraig Topper /// the integer value.
40895819069SCraig Topper static bool canFitInBitfield(int64_t Value, unsigned NumBits) {
40995819069SCraig Topper   // For example, with NumBits == 4, we permit Values from [-7 .. 15].
41095819069SCraig Topper   return (NumBits >= sizeof(Value) * 8) ||
41195819069SCraig Topper          (Value >> NumBits == 0) || (Value >> (NumBits-1) == -1);
41295819069SCraig Topper }
41395819069SCraig Topper 
41495819069SCraig Topper Init *IntInit::convertInitializerTo(RecTy *Ty) const {
41595819069SCraig Topper   if (isa<IntRecTy>(Ty))
41695819069SCraig Topper     return const_cast<IntInit *>(this);
41795819069SCraig Topper 
41895819069SCraig Topper   if (isa<BitRecTy>(Ty)) {
41995819069SCraig Topper     int64_t Val = getValue();
42095819069SCraig Topper     if (Val != 0 && Val != 1) return nullptr;  // Only accept 0 or 1 for a bit!
42195819069SCraig Topper     return BitInit::get(Val != 0);
42295819069SCraig Topper   }
42395819069SCraig Topper 
42495819069SCraig Topper   if (auto *BRT = dyn_cast<BitsRecTy>(Ty)) {
42595819069SCraig Topper     int64_t Value = getValue();
42695819069SCraig Topper     // Make sure this bitfield is large enough to hold the integer value.
42795819069SCraig Topper     if (!canFitInBitfield(Value, BRT->getNumBits()))
42895819069SCraig Topper       return nullptr;
42995819069SCraig Topper 
43095819069SCraig Topper     SmallVector<Init *, 16> NewBits(BRT->getNumBits());
43195819069SCraig Topper     for (unsigned i = 0; i != BRT->getNumBits(); ++i)
43295819069SCraig Topper       NewBits[i] = BitInit::get(Value & (1LL << i));
43395819069SCraig Topper 
43495819069SCraig Topper     return BitsInit::get(NewBits);
43595819069SCraig Topper   }
43695819069SCraig Topper 
43795819069SCraig Topper   return nullptr;
43895819069SCraig Topper }
43995819069SCraig Topper 
44084c287e3SPeter Collingbourne Init *
44184c287e3SPeter Collingbourne IntInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
44284c287e3SPeter Collingbourne   SmallVector<Init *, 16> NewBits(Bits.size());
44384c287e3SPeter Collingbourne 
44484c287e3SPeter Collingbourne   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
44584c287e3SPeter Collingbourne     if (Bits[i] >= 64)
446011817a0SCraig Topper       return nullptr;
44784c287e3SPeter Collingbourne 
44884c287e3SPeter Collingbourne     NewBits[i] = BitInit::get(Value & (INT64_C(1) << Bits[i]));
44984c287e3SPeter Collingbourne   }
45084c287e3SPeter Collingbourne   return BitsInit::get(NewBits);
45184c287e3SPeter Collingbourne }
45284c287e3SPeter Collingbourne 
4539d1c5eebSJakob Stoklund Olesen StringInit *StringInit::get(StringRef V) {
4540e04bee8SCraig Topper   static StringMap<std::unique_ptr<StringInit>> ThePool;
45584c287e3SPeter Collingbourne 
4560e04bee8SCraig Topper   std::unique_ptr<StringInit> &I = ThePool[V];
4570e04bee8SCraig Topper   if (!I) I.reset(new StringInit(V));
4580e04bee8SCraig Topper   return I.get();
45984c287e3SPeter Collingbourne }
46084c287e3SPeter Collingbourne 
46195819069SCraig Topper Init *StringInit::convertInitializerTo(RecTy *Ty) const {
46295819069SCraig Topper   if (isa<StringRecTy>(Ty))
46395819069SCraig Topper     return const_cast<StringInit *>(this);
46495819069SCraig Topper 
46595819069SCraig Topper   return nullptr;
46695819069SCraig Topper }
46795819069SCraig Topper 
46884c287e3SPeter Collingbourne static void ProfileListInit(FoldingSetNodeID &ID,
46984c287e3SPeter Collingbourne                             ArrayRef<Init *> Range,
47084c287e3SPeter Collingbourne                             RecTy *EltTy) {
47184c287e3SPeter Collingbourne   ID.AddInteger(Range.size());
47284c287e3SPeter Collingbourne   ID.AddPointer(EltTy);
47384c287e3SPeter Collingbourne 
474119998dbSCraig Topper   for (Init *I : Range)
475119998dbSCraig Topper     ID.AddPointer(I);
47684c287e3SPeter Collingbourne }
47784c287e3SPeter Collingbourne 
47884c287e3SPeter Collingbourne ListInit *ListInit::get(ArrayRef<Init *> Range, RecTy *EltTy) {
47917b3a378SCraig Topper   static FoldingSet<ListInit> ThePool;
4802b6b3d1cSDaniel Sanders   static std::vector<std::unique_ptr<ListInit>> TheActualPool;
48184c287e3SPeter Collingbourne 
48284c287e3SPeter Collingbourne   FoldingSetNodeID ID;
48384c287e3SPeter Collingbourne   ProfileListInit(ID, Range, EltTy);
48484c287e3SPeter Collingbourne 
485011817a0SCraig Topper   void *IP = nullptr;
48684c287e3SPeter Collingbourne   if (ListInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
48784c287e3SPeter Collingbourne     return I;
48884c287e3SPeter Collingbourne 
48984c287e3SPeter Collingbourne   ListInit *I = new ListInit(Range, EltTy);
49084c287e3SPeter Collingbourne   ThePool.InsertNode(I, IP);
4912b6b3d1cSDaniel Sanders   TheActualPool.push_back(std::unique_ptr<ListInit>(I));
49284c287e3SPeter Collingbourne   return I;
49384c287e3SPeter Collingbourne }
49484c287e3SPeter Collingbourne 
49584c287e3SPeter Collingbourne void ListInit::Profile(FoldingSetNodeID &ID) const {
496ed5a9508SCraig Topper   RecTy *EltTy = cast<ListRecTy>(getType())->getElementType();
49784c287e3SPeter Collingbourne 
49884c287e3SPeter Collingbourne   ProfileListInit(ID, Values, EltTy);
49984c287e3SPeter Collingbourne }
50084c287e3SPeter Collingbourne 
50195819069SCraig Topper Init *ListInit::convertInitializerTo(RecTy *Ty) const {
50295819069SCraig Topper   if (auto *LRT = dyn_cast<ListRecTy>(Ty)) {
50395819069SCraig Topper     std::vector<Init*> Elements;
50495819069SCraig Topper 
50595819069SCraig Topper     // Verify that all of the elements of the list are subclasses of the
50695819069SCraig Topper     // appropriate class!
507ef0578a8SCraig Topper     for (Init *I : getValues())
508ef0578a8SCraig Topper       if (Init *CI = I->convertInitializerTo(LRT->getElementType()))
50995819069SCraig Topper         Elements.push_back(CI);
51095819069SCraig Topper       else
51195819069SCraig Topper         return nullptr;
51295819069SCraig Topper 
51395819069SCraig Topper     if (isa<ListRecTy>(getType()))
51495819069SCraig Topper       return ListInit::get(Elements, Ty);
51595819069SCraig Topper   }
51695819069SCraig Topper 
51795819069SCraig Topper   return nullptr;
51895819069SCraig Topper }
51995819069SCraig Topper 
52084c287e3SPeter Collingbourne Init *
52184c287e3SPeter Collingbourne ListInit::convertInitListSlice(const std::vector<unsigned> &Elements) const {
52284c287e3SPeter Collingbourne   std::vector<Init*> Vals;
52384c287e3SPeter Collingbourne   for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
524664f6a04SCraig Topper     if (Elements[i] >= size())
525011817a0SCraig Topper       return nullptr;
52684c287e3SPeter Collingbourne     Vals.push_back(getElement(Elements[i]));
52784c287e3SPeter Collingbourne   }
52884c287e3SPeter Collingbourne   return ListInit::get(Vals, getType());
52984c287e3SPeter Collingbourne }
53084c287e3SPeter Collingbourne 
53184c287e3SPeter Collingbourne Record *ListInit::getElementAsRecord(unsigned i) const {
53284c287e3SPeter Collingbourne   assert(i < Values.size() && "List element index out of range!");
533fb509ed1SSean Silva   DefInit *DI = dyn_cast<DefInit>(Values[i]);
534011817a0SCraig Topper   if (!DI)
535635debe8SJoerg Sonnenberger     PrintFatalError("Expected record in list!");
53684c287e3SPeter Collingbourne   return DI->getDef();
53784c287e3SPeter Collingbourne }
53884c287e3SPeter Collingbourne 
53984c287e3SPeter Collingbourne Init *ListInit::resolveReferences(Record &R, const RecordVal *RV) const {
54084c287e3SPeter Collingbourne   std::vector<Init*> Resolved;
541664f6a04SCraig Topper   Resolved.reserve(size());
54284c287e3SPeter Collingbourne   bool Changed = false;
54384c287e3SPeter Collingbourne 
544ef0578a8SCraig Topper   for (Init *CurElt : getValues()) {
54584c287e3SPeter Collingbourne     Init *E;
54684c287e3SPeter Collingbourne 
54784c287e3SPeter Collingbourne     do {
54884c287e3SPeter Collingbourne       E = CurElt;
54984c287e3SPeter Collingbourne       CurElt = CurElt->resolveReferences(R, RV);
55084c287e3SPeter Collingbourne       Changed |= E != CurElt;
55184c287e3SPeter Collingbourne     } while (E != CurElt);
55284c287e3SPeter Collingbourne     Resolved.push_back(E);
55384c287e3SPeter Collingbourne   }
55484c287e3SPeter Collingbourne 
55584c287e3SPeter Collingbourne   if (Changed)
55684c287e3SPeter Collingbourne     return ListInit::get(Resolved, getType());
55784c287e3SPeter Collingbourne   return const_cast<ListInit *>(this);
55884c287e3SPeter Collingbourne }
55984c287e3SPeter Collingbourne 
56084c287e3SPeter Collingbourne Init *ListInit::resolveListElementReference(Record &R, const RecordVal *IRV,
56184c287e3SPeter Collingbourne                                             unsigned Elt) const {
562664f6a04SCraig Topper   if (Elt >= size())
563011817a0SCraig Topper     return nullptr;  // Out of range reference.
56484c287e3SPeter Collingbourne   Init *E = getElement(Elt);
56584c287e3SPeter Collingbourne   // If the element is set to some value, or if we are resolving a reference
56684c287e3SPeter Collingbourne   // to a specific variable and that variable is explicitly unset, then
56784c287e3SPeter Collingbourne   // replace the VarListElementInit with it.
56888eb8dd4SSean Silva   if (IRV || !isa<UnsetInit>(E))
56984c287e3SPeter Collingbourne     return E;
570011817a0SCraig Topper   return nullptr;
57184c287e3SPeter Collingbourne }
57284c287e3SPeter Collingbourne 
57384c287e3SPeter Collingbourne std::string ListInit::getAsString() const {
57484c287e3SPeter Collingbourne   std::string Result = "[";
57584c287e3SPeter Collingbourne   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
57684c287e3SPeter Collingbourne     if (i) Result += ", ";
57784c287e3SPeter Collingbourne     Result += Values[i]->getAsString();
57884c287e3SPeter Collingbourne   }
57984c287e3SPeter Collingbourne   return Result + "]";
58084c287e3SPeter Collingbourne }
58184c287e3SPeter Collingbourne 
58284c287e3SPeter Collingbourne Init *OpInit::resolveListElementReference(Record &R, const RecordVal *IRV,
58384c287e3SPeter Collingbourne                                           unsigned Elt) const {
584a8b8ab6bSDavid Greene   Init *Resolved = resolveReferences(R, IRV);
585fb509ed1SSean Silva   OpInit *OResolved = dyn_cast<OpInit>(Resolved);
586a8b8ab6bSDavid Greene   if (OResolved) {
587011817a0SCraig Topper     Resolved = OResolved->Fold(&R, nullptr);
588a8b8ab6bSDavid Greene   }
58984c287e3SPeter Collingbourne 
590a8b8ab6bSDavid Greene   if (Resolved != this) {
591ed5a9508SCraig Topper     TypedInit *Typed = cast<TypedInit>(Resolved);
592ed5a9508SCraig Topper     if (Init *New = Typed->resolveListElementReference(R, IRV, Elt))
593a8b8ab6bSDavid Greene       return New;
594a8b8ab6bSDavid Greene     return VarListElementInit::get(Typed, Elt);
59584c287e3SPeter Collingbourne   }
59684c287e3SPeter Collingbourne 
597011817a0SCraig Topper   return nullptr;
59884c287e3SPeter Collingbourne }
59984c287e3SPeter Collingbourne 
600026f8333SMichael Liao Init *OpInit::getBit(unsigned Bit) const {
601026f8333SMichael Liao   if (getType() == BitRecTy::get())
602026f8333SMichael Liao     return const_cast<OpInit*>(this);
603026f8333SMichael Liao   return VarBitInit::get(const_cast<OpInit*>(this), Bit);
604026f8333SMichael Liao }
605026f8333SMichael Liao 
60684c287e3SPeter Collingbourne UnOpInit *UnOpInit::get(UnaryOp opc, Init *lhs, RecTy *Type) {
60784c287e3SPeter Collingbourne   typedef std::pair<std::pair<unsigned, Init *>, RecTy *> Key;
6080e04bee8SCraig Topper   static DenseMap<Key, std::unique_ptr<UnOpInit>> ThePool;
60984c287e3SPeter Collingbourne 
61084c287e3SPeter Collingbourne   Key TheKey(std::make_pair(std::make_pair(opc, lhs), Type));
61184c287e3SPeter Collingbourne 
6120e04bee8SCraig Topper   std::unique_ptr<UnOpInit> &I = ThePool[TheKey];
6130e04bee8SCraig Topper   if (!I) I.reset(new UnOpInit(opc, lhs, Type));
6140e04bee8SCraig Topper   return I.get();
61584c287e3SPeter Collingbourne }
61684c287e3SPeter Collingbourne 
61784c287e3SPeter Collingbourne Init *UnOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
61884c287e3SPeter Collingbourne   switch (getOpcode()) {
61984c287e3SPeter Collingbourne   case CAST: {
6202a8d4b29SCraig Topper     if (isa<StringRecTy>(getType())) {
62188eb8dd4SSean Silva       if (StringInit *LHSs = dyn_cast<StringInit>(LHS))
62284c287e3SPeter Collingbourne         return LHSs;
62384c287e3SPeter Collingbourne 
62488eb8dd4SSean Silva       if (DefInit *LHSd = dyn_cast<DefInit>(LHS))
62585693aebSCraig Topper         return StringInit::get(LHSd->getAsString());
6266291f3a4SDavid Greene 
62788eb8dd4SSean Silva       if (IntInit *LHSi = dyn_cast<IntInit>(LHS))
6286291f3a4SDavid Greene         return StringInit::get(LHSi->getAsString());
62984c287e3SPeter Collingbourne     } else {
63088eb8dd4SSean Silva       if (StringInit *LHSs = dyn_cast<StringInit>(LHS)) {
63184c287e3SPeter Collingbourne         std::string Name = LHSs->getValue();
63284c287e3SPeter Collingbourne 
63384c287e3SPeter Collingbourne         // From TGParser::ParseIDValue
63484c287e3SPeter Collingbourne         if (CurRec) {
63584c287e3SPeter Collingbourne           if (const RecordVal *RV = CurRec->getValue(Name)) {
63684c287e3SPeter Collingbourne             if (RV->getType() != getType())
637635debe8SJoerg Sonnenberger               PrintFatalError("type mismatch in cast");
63884c287e3SPeter Collingbourne             return VarInit::get(Name, RV->getType());
63984c287e3SPeter Collingbourne           }
64084c287e3SPeter Collingbourne 
641db10e691SDavid Greene           Init *TemplateArgName = QualifyName(*CurRec, CurMultiClass, Name,
642db10e691SDavid Greene                                               ":");
643db10e691SDavid Greene 
64484c287e3SPeter Collingbourne           if (CurRec->isTemplateArg(TemplateArgName)) {
64584c287e3SPeter Collingbourne             const RecordVal *RV = CurRec->getValue(TemplateArgName);
64684c287e3SPeter Collingbourne             assert(RV && "Template arg doesn't exist??");
64784c287e3SPeter Collingbourne 
64884c287e3SPeter Collingbourne             if (RV->getType() != getType())
649635debe8SJoerg Sonnenberger               PrintFatalError("type mismatch in cast");
65084c287e3SPeter Collingbourne 
65184c287e3SPeter Collingbourne             return VarInit::get(TemplateArgName, RV->getType());
65284c287e3SPeter Collingbourne           }
65384c287e3SPeter Collingbourne         }
65484c287e3SPeter Collingbourne 
65584c287e3SPeter Collingbourne         if (CurMultiClass) {
656a9642b4eSCraig Topper           Init *MCName = QualifyName(CurMultiClass->Rec, CurMultiClass, Name,
657a9642b4eSCraig Topper                                      "::");
658db10e691SDavid Greene 
65984c287e3SPeter Collingbourne           if (CurMultiClass->Rec.isTemplateArg(MCName)) {
66084c287e3SPeter Collingbourne             const RecordVal *RV = CurMultiClass->Rec.getValue(MCName);
66184c287e3SPeter Collingbourne             assert(RV && "Template arg doesn't exist??");
66284c287e3SPeter Collingbourne 
66384c287e3SPeter Collingbourne             if (RV->getType() != getType())
664635debe8SJoerg Sonnenberger               PrintFatalError("type mismatch in cast");
66584c287e3SPeter Collingbourne 
66684c287e3SPeter Collingbourne             return VarInit::get(MCName, RV->getType());
66784c287e3SPeter Collingbourne           }
66884c287e3SPeter Collingbourne         }
6695be22a12SMichael Ilseman         assert(CurRec && "NULL pointer");
67084c287e3SPeter Collingbourne         if (Record *D = (CurRec->getRecords()).getDef(Name))
67184c287e3SPeter Collingbourne           return DefInit::get(D);
67284c287e3SPeter Collingbourne 
673635debe8SJoerg Sonnenberger         PrintFatalError(CurRec->getLoc(),
674635debe8SJoerg Sonnenberger                         "Undefined reference:'" + Name + "'\n");
67584c287e3SPeter Collingbourne       }
67684c287e3SPeter Collingbourne     }
67784c287e3SPeter Collingbourne     break;
67884c287e3SPeter Collingbourne   }
67984c287e3SPeter Collingbourne   case HEAD: {
68088eb8dd4SSean Silva     if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) {
68141ce4de0SCraig Topper       assert(!LHSl->empty() && "Empty list in head");
68284c287e3SPeter Collingbourne       return LHSl->getElement(0);
68384c287e3SPeter Collingbourne     }
68484c287e3SPeter Collingbourne     break;
68584c287e3SPeter Collingbourne   }
68684c287e3SPeter Collingbourne   case TAIL: {
68788eb8dd4SSean Silva     if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) {
68841ce4de0SCraig Topper       assert(!LHSl->empty() && "Empty list in tail");
68984c287e3SPeter Collingbourne       // Note the +1.  We can't just pass the result of getValues()
69084c287e3SPeter Collingbourne       // directly.
691607ac92dSCraig Topper       return ListInit::get(LHSl->getValues().slice(1), LHSl->getType());
69284c287e3SPeter Collingbourne     }
69384c287e3SPeter Collingbourne     break;
69484c287e3SPeter Collingbourne   }
69584c287e3SPeter Collingbourne   case EMPTY: {
6968ddb0d82SCraig Topper     if (ListInit *LHSl = dyn_cast<ListInit>(LHS))
6978ddb0d82SCraig Topper       return IntInit::get(LHSl->empty());
6988ddb0d82SCraig Topper     if (StringInit *LHSs = dyn_cast<StringInit>(LHS))
6998ddb0d82SCraig Topper       return IntInit::get(LHSs->getValue().empty());
70084c287e3SPeter Collingbourne 
70184c287e3SPeter Collingbourne     break;
70284c287e3SPeter Collingbourne   }
70384c287e3SPeter Collingbourne   }
70484c287e3SPeter Collingbourne   return const_cast<UnOpInit *>(this);
70584c287e3SPeter Collingbourne }
70684c287e3SPeter Collingbourne 
70784c287e3SPeter Collingbourne Init *UnOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
70884c287e3SPeter Collingbourne   Init *lhs = LHS->resolveReferences(R, RV);
70984c287e3SPeter Collingbourne 
71084c287e3SPeter Collingbourne   if (LHS != lhs)
711011817a0SCraig Topper     return (UnOpInit::get(getOpcode(), lhs, getType()))->Fold(&R, nullptr);
712011817a0SCraig Topper   return Fold(&R, nullptr);
71384c287e3SPeter Collingbourne }
71484c287e3SPeter Collingbourne 
71584c287e3SPeter Collingbourne std::string UnOpInit::getAsString() const {
71684c287e3SPeter Collingbourne   std::string Result;
71784c287e3SPeter Collingbourne   switch (Opc) {
71884c287e3SPeter Collingbourne   case CAST: Result = "!cast<" + getType()->getAsString() + ">"; break;
71984c287e3SPeter Collingbourne   case HEAD: Result = "!head"; break;
72084c287e3SPeter Collingbourne   case TAIL: Result = "!tail"; break;
72184c287e3SPeter Collingbourne   case EMPTY: Result = "!empty"; break;
72284c287e3SPeter Collingbourne   }
72384c287e3SPeter Collingbourne   return Result + "(" + LHS->getAsString() + ")";
72484c287e3SPeter Collingbourne }
72584c287e3SPeter Collingbourne 
72684c287e3SPeter Collingbourne BinOpInit *BinOpInit::get(BinaryOp opc, Init *lhs,
72784c287e3SPeter Collingbourne                           Init *rhs, RecTy *Type) {
72884c287e3SPeter Collingbourne   typedef std::pair<
72984c287e3SPeter Collingbourne     std::pair<std::pair<unsigned, Init *>, Init *>,
73084c287e3SPeter Collingbourne     RecTy *
73184c287e3SPeter Collingbourne     > Key;
73284c287e3SPeter Collingbourne 
7330e04bee8SCraig Topper   static DenseMap<Key, std::unique_ptr<BinOpInit>> ThePool;
73484c287e3SPeter Collingbourne 
73584c287e3SPeter Collingbourne   Key TheKey(std::make_pair(std::make_pair(std::make_pair(opc, lhs), rhs),
73684c287e3SPeter Collingbourne                             Type));
73784c287e3SPeter Collingbourne 
7380e04bee8SCraig Topper   std::unique_ptr<BinOpInit> &I = ThePool[TheKey];
7390e04bee8SCraig Topper   if (!I) I.reset(new BinOpInit(opc, lhs, rhs, Type));
7400e04bee8SCraig Topper   return I.get();
74184c287e3SPeter Collingbourne }
74284c287e3SPeter Collingbourne 
74384c287e3SPeter Collingbourne Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
74484c287e3SPeter Collingbourne   switch (getOpcode()) {
74584c287e3SPeter Collingbourne   case CONCAT: {
746fb509ed1SSean Silva     DagInit *LHSs = dyn_cast<DagInit>(LHS);
747fb509ed1SSean Silva     DagInit *RHSs = dyn_cast<DagInit>(RHS);
74884c287e3SPeter Collingbourne     if (LHSs && RHSs) {
749fb509ed1SSean Silva       DefInit *LOp = dyn_cast<DefInit>(LHSs->getOperator());
750fb509ed1SSean Silva       DefInit *ROp = dyn_cast<DefInit>(RHSs->getOperator());
751011817a0SCraig Topper       if (!LOp || !ROp || LOp->getDef() != ROp->getDef())
752635debe8SJoerg Sonnenberger         PrintFatalError("Concated Dag operators do not match!");
75384c287e3SPeter Collingbourne       std::vector<Init*> Args;
75484c287e3SPeter Collingbourne       std::vector<std::string> ArgNames;
75584c287e3SPeter Collingbourne       for (unsigned i = 0, e = LHSs->getNumArgs(); i != e; ++i) {
75684c287e3SPeter Collingbourne         Args.push_back(LHSs->getArg(i));
75784c287e3SPeter Collingbourne         ArgNames.push_back(LHSs->getArgName(i));
75884c287e3SPeter Collingbourne       }
75984c287e3SPeter Collingbourne       for (unsigned i = 0, e = RHSs->getNumArgs(); i != e; ++i) {
76084c287e3SPeter Collingbourne         Args.push_back(RHSs->getArg(i));
76184c287e3SPeter Collingbourne         ArgNames.push_back(RHSs->getArgName(i));
76284c287e3SPeter Collingbourne       }
76384c287e3SPeter Collingbourne       return DagInit::get(LHSs->getOperator(), "", Args, ArgNames);
76484c287e3SPeter Collingbourne     }
76584c287e3SPeter Collingbourne     break;
76684c287e3SPeter Collingbourne   }
767314e80e5SDaniel Sanders   case LISTCONCAT: {
768314e80e5SDaniel Sanders     ListInit *LHSs = dyn_cast<ListInit>(LHS);
769314e80e5SDaniel Sanders     ListInit *RHSs = dyn_cast<ListInit>(RHS);
770314e80e5SDaniel Sanders     if (LHSs && RHSs) {
771314e80e5SDaniel Sanders       std::vector<Init *> Args;
772314e80e5SDaniel Sanders       Args.insert(Args.end(), LHSs->begin(), LHSs->end());
773314e80e5SDaniel Sanders       Args.insert(Args.end(), RHSs->begin(), RHSs->end());
774314e80e5SDaniel Sanders       return ListInit::get(
77523fdafe6SCraig Topper           Args, cast<ListRecTy>(LHSs->getType())->getElementType());
776314e80e5SDaniel Sanders     }
777314e80e5SDaniel Sanders     break;
778314e80e5SDaniel Sanders   }
77984c287e3SPeter Collingbourne   case STRCONCAT: {
780fb509ed1SSean Silva     StringInit *LHSs = dyn_cast<StringInit>(LHS);
781fb509ed1SSean Silva     StringInit *RHSs = dyn_cast<StringInit>(RHS);
78284c287e3SPeter Collingbourne     if (LHSs && RHSs)
78384c287e3SPeter Collingbourne       return StringInit::get(LHSs->getValue() + RHSs->getValue());
78484c287e3SPeter Collingbourne     break;
78584c287e3SPeter Collingbourne   }
78684c287e3SPeter Collingbourne   case EQ: {
78784c287e3SPeter Collingbourne     // try to fold eq comparison for 'bit' and 'int', otherwise fallback
78884c287e3SPeter Collingbourne     // to string objects.
78984c287e3SPeter Collingbourne     IntInit *L =
790fb509ed1SSean Silva       dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get()));
79184c287e3SPeter Collingbourne     IntInit *R =
792fb509ed1SSean Silva       dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get()));
79384c287e3SPeter Collingbourne 
79484c287e3SPeter Collingbourne     if (L && R)
79584c287e3SPeter Collingbourne       return IntInit::get(L->getValue() == R->getValue());
79684c287e3SPeter Collingbourne 
797fb509ed1SSean Silva     StringInit *LHSs = dyn_cast<StringInit>(LHS);
798fb509ed1SSean Silva     StringInit *RHSs = dyn_cast<StringInit>(RHS);
79984c287e3SPeter Collingbourne 
80084c287e3SPeter Collingbourne     // Make sure we've resolved
80184c287e3SPeter Collingbourne     if (LHSs && RHSs)
80284c287e3SPeter Collingbourne       return IntInit::get(LHSs->getValue() == RHSs->getValue());
80384c287e3SPeter Collingbourne 
80484c287e3SPeter Collingbourne     break;
80584c287e3SPeter Collingbourne   }
806c7d4dc13SHal Finkel   case ADD:
8076b41a990SJoerg Sonnenberger   case AND:
80884c287e3SPeter Collingbourne   case SHL:
80984c287e3SPeter Collingbourne   case SRA:
81084c287e3SPeter Collingbourne   case SRL: {
811017fca02SAdam Nemet     IntInit *LHSi =
812017fca02SAdam Nemet       dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get()));
813017fca02SAdam Nemet     IntInit *RHSi =
814017fca02SAdam Nemet       dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get()));
81584c287e3SPeter Collingbourne     if (LHSi && RHSi) {
81684c287e3SPeter Collingbourne       int64_t LHSv = LHSi->getValue(), RHSv = RHSi->getValue();
81784c287e3SPeter Collingbourne       int64_t Result;
81884c287e3SPeter Collingbourne       switch (getOpcode()) {
819a2886c21SCraig Topper       default: llvm_unreachable("Bad opcode!");
820c7d4dc13SHal Finkel       case ADD: Result = LHSv +  RHSv; break;
8216b41a990SJoerg Sonnenberger       case AND: Result = LHSv &  RHSv; break;
82284c287e3SPeter Collingbourne       case SHL: Result = LHSv << RHSv; break;
82384c287e3SPeter Collingbourne       case SRA: Result = LHSv >> RHSv; break;
82484c287e3SPeter Collingbourne       case SRL: Result = (uint64_t)LHSv >> (uint64_t)RHSv; break;
82584c287e3SPeter Collingbourne       }
82684c287e3SPeter Collingbourne       return IntInit::get(Result);
82784c287e3SPeter Collingbourne     }
82884c287e3SPeter Collingbourne     break;
82984c287e3SPeter Collingbourne   }
83084c287e3SPeter Collingbourne   }
83184c287e3SPeter Collingbourne   return const_cast<BinOpInit *>(this);
83284c287e3SPeter Collingbourne }
83384c287e3SPeter Collingbourne 
83484c287e3SPeter Collingbourne Init *BinOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
83584c287e3SPeter Collingbourne   Init *lhs = LHS->resolveReferences(R, RV);
83684c287e3SPeter Collingbourne   Init *rhs = RHS->resolveReferences(R, RV);
83784c287e3SPeter Collingbourne 
83884c287e3SPeter Collingbourne   if (LHS != lhs || RHS != rhs)
839011817a0SCraig Topper     return (BinOpInit::get(getOpcode(), lhs, rhs, getType()))->Fold(&R,nullptr);
840011817a0SCraig Topper   return Fold(&R, nullptr);
84184c287e3SPeter Collingbourne }
84284c287e3SPeter Collingbourne 
84384c287e3SPeter Collingbourne std::string BinOpInit::getAsString() const {
84484c287e3SPeter Collingbourne   std::string Result;
84584c287e3SPeter Collingbourne   switch (Opc) {
84684c287e3SPeter Collingbourne   case CONCAT: Result = "!con"; break;
847c7d4dc13SHal Finkel   case ADD: Result = "!add"; break;
8486b41a990SJoerg Sonnenberger   case AND: Result = "!and"; break;
84984c287e3SPeter Collingbourne   case SHL: Result = "!shl"; break;
85084c287e3SPeter Collingbourne   case SRA: Result = "!sra"; break;
85184c287e3SPeter Collingbourne   case SRL: Result = "!srl"; break;
85284c287e3SPeter Collingbourne   case EQ: Result = "!eq"; break;
853314e80e5SDaniel Sanders   case LISTCONCAT: Result = "!listconcat"; break;
85484c287e3SPeter Collingbourne   case STRCONCAT: Result = "!strconcat"; break;
85584c287e3SPeter Collingbourne   }
85684c287e3SPeter Collingbourne   return Result + "(" + LHS->getAsString() + ", " + RHS->getAsString() + ")";
85784c287e3SPeter Collingbourne }
85884c287e3SPeter Collingbourne 
859a9642b4eSCraig Topper TernOpInit *TernOpInit::get(TernaryOp opc, Init *lhs, Init *mhs, Init *rhs,
86084c287e3SPeter Collingbourne                             RecTy *Type) {
86184c287e3SPeter Collingbourne   typedef std::pair<
86284c287e3SPeter Collingbourne     std::pair<
86384c287e3SPeter Collingbourne       std::pair<std::pair<unsigned, RecTy *>, Init *>,
86484c287e3SPeter Collingbourne       Init *
86584c287e3SPeter Collingbourne       >,
86684c287e3SPeter Collingbourne     Init *
86784c287e3SPeter Collingbourne     > Key;
86884c287e3SPeter Collingbourne 
86917b3a378SCraig Topper   static DenseMap<Key, std::unique_ptr<TernOpInit>> ThePool;
87084c287e3SPeter Collingbourne 
87184c287e3SPeter Collingbourne   Key TheKey(std::make_pair(std::make_pair(std::make_pair(std::make_pair(opc,
87284c287e3SPeter Collingbourne                                                                          Type),
87384c287e3SPeter Collingbourne                                                           lhs),
87484c287e3SPeter Collingbourne                                            mhs),
87584c287e3SPeter Collingbourne                             rhs));
87684c287e3SPeter Collingbourne 
87717b3a378SCraig Topper   std::unique_ptr<TernOpInit> &I = ThePool[TheKey];
87817b3a378SCraig Topper   if (!I) I.reset(new TernOpInit(opc, lhs, mhs, rhs, Type));
87917b3a378SCraig Topper   return I.get();
88084c287e3SPeter Collingbourne }
88184c287e3SPeter Collingbourne 
88284c287e3SPeter Collingbourne static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
88384c287e3SPeter Collingbourne                            Record *CurRec, MultiClass *CurMultiClass);
88484c287e3SPeter Collingbourne 
88584c287e3SPeter Collingbourne static Init *EvaluateOperation(OpInit *RHSo, Init *LHS, Init *Arg,
88684c287e3SPeter Collingbourne                                RecTy *Type, Record *CurRec,
88784c287e3SPeter Collingbourne                                MultiClass *CurMultiClass) {
88884c287e3SPeter Collingbourne   // If this is a dag, recurse
8899ef76499SCraig Topper   if (auto *TArg = dyn_cast<TypedInit>(Arg))
8902a8d4b29SCraig Topper     if (isa<DagRecTy>(TArg->getType()))
8919ef76499SCraig Topper       return ForeachHelper(LHS, Arg, RHSo, Type, CurRec, CurMultiClass);
89284c287e3SPeter Collingbourne 
89339ba33b8SCraig Topper   std::vector<Init *> NewOperands;
89484c287e3SPeter Collingbourne   for (int i = 0; i < RHSo->getNumOperands(); ++i) {
8959ef76499SCraig Topper     if (auto *RHSoo = dyn_cast<OpInit>(RHSo->getOperand(i))) {
8969ef76499SCraig Topper       if (Init *Result = EvaluateOperation(RHSoo, LHS, Arg,
8979ef76499SCraig Topper                                            Type, CurRec, CurMultiClass))
89884c287e3SPeter Collingbourne         NewOperands.push_back(Result);
8999ef76499SCraig Topper       else
90084c287e3SPeter Collingbourne         NewOperands.push_back(Arg);
90184c287e3SPeter Collingbourne     } else if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
90284c287e3SPeter Collingbourne       NewOperands.push_back(Arg);
90384c287e3SPeter Collingbourne     } else {
90484c287e3SPeter Collingbourne       NewOperands.push_back(RHSo->getOperand(i));
90584c287e3SPeter Collingbourne     }
90684c287e3SPeter Collingbourne   }
90784c287e3SPeter Collingbourne 
90884c287e3SPeter Collingbourne   // Now run the operator and use its result as the new leaf
90984c287e3SPeter Collingbourne   const OpInit *NewOp = RHSo->clone(NewOperands);
91084c287e3SPeter Collingbourne   Init *NewVal = NewOp->Fold(CurRec, CurMultiClass);
911011817a0SCraig Topper   return (NewVal != NewOp) ? NewVal : nullptr;
91284c287e3SPeter Collingbourne }
91384c287e3SPeter Collingbourne 
91484c287e3SPeter Collingbourne static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
91584c287e3SPeter Collingbourne                            Record *CurRec, MultiClass *CurMultiClass) {
91684c287e3SPeter Collingbourne 
917fb509ed1SSean Silva   OpInit *RHSo = dyn_cast<OpInit>(RHS);
91884c287e3SPeter Collingbourne 
919a9642b4eSCraig Topper   if (!RHSo)
920635debe8SJoerg Sonnenberger     PrintFatalError(CurRec->getLoc(), "!foreach requires an operator\n");
92184c287e3SPeter Collingbourne 
922fb509ed1SSean Silva   TypedInit *LHSt = dyn_cast<TypedInit>(LHS);
92384c287e3SPeter Collingbourne 
924635debe8SJoerg Sonnenberger   if (!LHSt)
925635debe8SJoerg Sonnenberger     PrintFatalError(CurRec->getLoc(), "!foreach requires typed variable\n");
92684c287e3SPeter Collingbourne 
927daf263deSCraig Topper   DagInit *MHSd = dyn_cast<DagInit>(MHS);
928daf263deSCraig Topper   if (MHSd && isa<DagRecTy>(Type)) {
92984c287e3SPeter Collingbourne     Init *Val = MHSd->getOperator();
930*5904beb6SCraig Topper     if (Init *Result = EvaluateOperation(RHSo, LHS, Val,
931*5904beb6SCraig Topper                                          Type, CurRec, CurMultiClass))
93284c287e3SPeter Collingbourne       Val = Result;
93384c287e3SPeter Collingbourne 
93484c287e3SPeter Collingbourne     std::vector<std::pair<Init *, std::string> > args;
93584c287e3SPeter Collingbourne     for (unsigned int i = 0; i < MHSd->getNumArgs(); ++i) {
93625a849caSCraig Topper       Init *Arg = MHSd->getArg(i);
93725a849caSCraig Topper       std::string ArgName = MHSd->getArgName(i);
93884c287e3SPeter Collingbourne 
93984c287e3SPeter Collingbourne       // Process args
940*5904beb6SCraig Topper       if (Init *Result = EvaluateOperation(RHSo, LHS, Arg, Type,
941*5904beb6SCraig Topper                                            CurRec, CurMultiClass))
94284c287e3SPeter Collingbourne         Arg = Result;
94384c287e3SPeter Collingbourne 
94484c287e3SPeter Collingbourne       // TODO: Process arg names
94584c287e3SPeter Collingbourne       args.push_back(std::make_pair(Arg, ArgName));
94684c287e3SPeter Collingbourne     }
94784c287e3SPeter Collingbourne 
94884c287e3SPeter Collingbourne     return DagInit::get(Val, "", args);
94984c287e3SPeter Collingbourne   }
950daf263deSCraig Topper 
951daf263deSCraig Topper   ListInit *MHSl = dyn_cast<ListInit>(MHS);
952daf263deSCraig Topper   if (MHSl && isa<ListRecTy>(Type)) {
95384c287e3SPeter Collingbourne     std::vector<Init *> NewOperands;
95484c287e3SPeter Collingbourne     std::vector<Init *> NewList(MHSl->begin(), MHSl->end());
95584c287e3SPeter Collingbourne 
9569ed595e4SCraig Topper     for (Init *&Item : NewList) {
95784c287e3SPeter Collingbourne       NewOperands.clear();
95884c287e3SPeter Collingbourne       for(int i = 0; i < RHSo->getNumOperands(); ++i) {
95984c287e3SPeter Collingbourne         // First, replace the foreach variable with the list item
960a9642b4eSCraig Topper         if (LHS->getAsString() == RHSo->getOperand(i)->getAsString())
96184c287e3SPeter Collingbourne           NewOperands.push_back(Item);
962a9642b4eSCraig Topper         else
96384c287e3SPeter Collingbourne           NewOperands.push_back(RHSo->getOperand(i));
96484c287e3SPeter Collingbourne       }
96584c287e3SPeter Collingbourne 
96684c287e3SPeter Collingbourne       // Now run the operator and use its result as the new list item
96784c287e3SPeter Collingbourne       const OpInit *NewOp = RHSo->clone(NewOperands);
96884c287e3SPeter Collingbourne       Init *NewItem = NewOp->Fold(CurRec, CurMultiClass);
96984c287e3SPeter Collingbourne       if (NewItem != NewOp)
9709ed595e4SCraig Topper         Item = NewItem;
97184c287e3SPeter Collingbourne     }
97284c287e3SPeter Collingbourne     return ListInit::get(NewList, MHSl->getType());
97384c287e3SPeter Collingbourne   }
974011817a0SCraig Topper   return nullptr;
97584c287e3SPeter Collingbourne }
97684c287e3SPeter Collingbourne 
97784c287e3SPeter Collingbourne Init *TernOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
97884c287e3SPeter Collingbourne   switch (getOpcode()) {
97984c287e3SPeter Collingbourne   case SUBST: {
980fb509ed1SSean Silva     DefInit *LHSd = dyn_cast<DefInit>(LHS);
981fb509ed1SSean Silva     VarInit *LHSv = dyn_cast<VarInit>(LHS);
982fb509ed1SSean Silva     StringInit *LHSs = dyn_cast<StringInit>(LHS);
98384c287e3SPeter Collingbourne 
984fb509ed1SSean Silva     DefInit *MHSd = dyn_cast<DefInit>(MHS);
985fb509ed1SSean Silva     VarInit *MHSv = dyn_cast<VarInit>(MHS);
986fb509ed1SSean Silva     StringInit *MHSs = dyn_cast<StringInit>(MHS);
98784c287e3SPeter Collingbourne 
988fb509ed1SSean Silva     DefInit *RHSd = dyn_cast<DefInit>(RHS);
989fb509ed1SSean Silva     VarInit *RHSv = dyn_cast<VarInit>(RHS);
990fb509ed1SSean Silva     StringInit *RHSs = dyn_cast<StringInit>(RHS);
99184c287e3SPeter Collingbourne 
992b1846a35SCraig Topper     if (LHSd && MHSd && RHSd) {
99384c287e3SPeter Collingbourne       Record *Val = RHSd->getDef();
994a9642b4eSCraig Topper       if (LHSd->getAsString() == RHSd->getAsString())
99584c287e3SPeter Collingbourne         Val = MHSd->getDef();
99684c287e3SPeter Collingbourne       return DefInit::get(Val);
99784c287e3SPeter Collingbourne     }
998b1846a35SCraig Topper     if (LHSv && MHSv && RHSv) {
99984c287e3SPeter Collingbourne       std::string Val = RHSv->getName();
1000a9642b4eSCraig Topper       if (LHSv->getAsString() == RHSv->getAsString())
100184c287e3SPeter Collingbourne         Val = MHSv->getName();
100284c287e3SPeter Collingbourne       return VarInit::get(Val, getType());
100384c287e3SPeter Collingbourne     }
1004b1846a35SCraig Topper     if (LHSs && MHSs && RHSs) {
100584c287e3SPeter Collingbourne       std::string Val = RHSs->getValue();
100684c287e3SPeter Collingbourne 
100784c287e3SPeter Collingbourne       std::string::size_type found;
100884c287e3SPeter Collingbourne       std::string::size_type idx = 0;
10096ee494b6SCraig Topper       while (true) {
101084c287e3SPeter Collingbourne         found = Val.find(LHSs->getValue(), idx);
10116ee494b6SCraig Topper         if (found == std::string::npos)
10126ee494b6SCraig Topper           break;
101384c287e3SPeter Collingbourne         Val.replace(found, LHSs->getValue().size(), MHSs->getValue());
101484c287e3SPeter Collingbourne         idx = found + MHSs->getValue().size();
10156ee494b6SCraig Topper       }
101684c287e3SPeter Collingbourne 
101784c287e3SPeter Collingbourne       return StringInit::get(Val);
101884c287e3SPeter Collingbourne     }
101984c287e3SPeter Collingbourne     break;
102084c287e3SPeter Collingbourne   }
102184c287e3SPeter Collingbourne 
102284c287e3SPeter Collingbourne   case FOREACH: {
102342467f25SCraig Topper     if (Init *Result = ForeachHelper(LHS, MHS, RHS, getType(),
102442467f25SCraig Topper                                      CurRec, CurMultiClass))
102584c287e3SPeter Collingbourne       return Result;
102684c287e3SPeter Collingbourne     break;
102784c287e3SPeter Collingbourne   }
102884c287e3SPeter Collingbourne 
102984c287e3SPeter Collingbourne   case IF: {
1030fb509ed1SSean Silva     IntInit *LHSi = dyn_cast<IntInit>(LHS);
103184c287e3SPeter Collingbourne     if (Init *I = LHS->convertInitializerTo(IntRecTy::get()))
1032fb509ed1SSean Silva       LHSi = dyn_cast<IntInit>(I);
103384c287e3SPeter Collingbourne     if (LHSi) {
1034a9642b4eSCraig Topper       if (LHSi->getValue())
103584c287e3SPeter Collingbourne         return MHS;
103684c287e3SPeter Collingbourne       return RHS;
103784c287e3SPeter Collingbourne     }
103884c287e3SPeter Collingbourne     break;
103984c287e3SPeter Collingbourne   }
104084c287e3SPeter Collingbourne   }
104184c287e3SPeter Collingbourne 
104284c287e3SPeter Collingbourne   return const_cast<TernOpInit *>(this);
104384c287e3SPeter Collingbourne }
104484c287e3SPeter Collingbourne 
104584c287e3SPeter Collingbourne Init *TernOpInit::resolveReferences(Record &R,
104684c287e3SPeter Collingbourne                                     const RecordVal *RV) const {
104784c287e3SPeter Collingbourne   Init *lhs = LHS->resolveReferences(R, RV);
104884c287e3SPeter Collingbourne 
104984c287e3SPeter Collingbourne   if (Opc == IF && lhs != LHS) {
1050fb509ed1SSean Silva     IntInit *Value = dyn_cast<IntInit>(lhs);
105184c287e3SPeter Collingbourne     if (Init *I = lhs->convertInitializerTo(IntRecTy::get()))
1052fb509ed1SSean Silva       Value = dyn_cast<IntInit>(I);
1053011817a0SCraig Topper     if (Value) {
105484c287e3SPeter Collingbourne       // Short-circuit
105584c287e3SPeter Collingbourne       if (Value->getValue()) {
105684c287e3SPeter Collingbourne         Init *mhs = MHS->resolveReferences(R, RV);
105784c287e3SPeter Collingbourne         return (TernOpInit::get(getOpcode(), lhs, mhs,
1058011817a0SCraig Topper                                 RHS, getType()))->Fold(&R, nullptr);
1059a9642b4eSCraig Topper       }
106084c287e3SPeter Collingbourne       Init *rhs = RHS->resolveReferences(R, RV);
106184c287e3SPeter Collingbourne       return (TernOpInit::get(getOpcode(), lhs, MHS,
1062011817a0SCraig Topper                               rhs, getType()))->Fold(&R, nullptr);
106384c287e3SPeter Collingbourne     }
106484c287e3SPeter Collingbourne   }
106584c287e3SPeter Collingbourne 
106684c287e3SPeter Collingbourne   Init *mhs = MHS->resolveReferences(R, RV);
106784c287e3SPeter Collingbourne   Init *rhs = RHS->resolveReferences(R, RV);
106884c287e3SPeter Collingbourne 
106984c287e3SPeter Collingbourne   if (LHS != lhs || MHS != mhs || RHS != rhs)
107084c287e3SPeter Collingbourne     return (TernOpInit::get(getOpcode(), lhs, mhs, rhs,
1071011817a0SCraig Topper                             getType()))->Fold(&R, nullptr);
1072011817a0SCraig Topper   return Fold(&R, nullptr);
107384c287e3SPeter Collingbourne }
107484c287e3SPeter Collingbourne 
107584c287e3SPeter Collingbourne std::string TernOpInit::getAsString() const {
107684c287e3SPeter Collingbourne   std::string Result;
107784c287e3SPeter Collingbourne   switch (Opc) {
107884c287e3SPeter Collingbourne   case SUBST: Result = "!subst"; break;
107984c287e3SPeter Collingbourne   case FOREACH: Result = "!foreach"; break;
108084c287e3SPeter Collingbourne   case IF: Result = "!if"; break;
108184c287e3SPeter Collingbourne   }
108285c07007SCraig Topper   return Result + "(" + LHS->getAsString() + ", " + MHS->getAsString() + ", " +
108385c07007SCraig Topper          RHS->getAsString() + ")";
108484c287e3SPeter Collingbourne }
108584c287e3SPeter Collingbourne 
108684c287e3SPeter Collingbourne RecTy *TypedInit::getFieldType(const std::string &FieldName) const {
1087a475e433SSean Silva   if (RecordRecTy *RecordType = dyn_cast<RecordRecTy>(getType()))
1088a475e433SSean Silva     if (RecordVal *Field = RecordType->getRecord()->getValue(FieldName))
108984c287e3SPeter Collingbourne       return Field->getType();
1090011817a0SCraig Topper   return nullptr;
109184c287e3SPeter Collingbourne }
109284c287e3SPeter Collingbourne 
109384c287e3SPeter Collingbourne Init *
109495819069SCraig Topper TypedInit::convertInitializerTo(RecTy *Ty) const {
109595819069SCraig Topper   if (isa<IntRecTy>(Ty)) {
109695819069SCraig Topper     if (getType()->typeIsConvertibleTo(Ty))
109795819069SCraig Topper       return const_cast<TypedInit *>(this);
109895819069SCraig Topper     return nullptr;
109995819069SCraig Topper   }
110095819069SCraig Topper 
110195819069SCraig Topper   if (isa<StringRecTy>(Ty)) {
110295819069SCraig Topper     if (isa<StringRecTy>(getType()))
110395819069SCraig Topper       return const_cast<TypedInit *>(this);
110495819069SCraig Topper     return nullptr;
110595819069SCraig Topper   }
110695819069SCraig Topper 
110795819069SCraig Topper   if (isa<BitRecTy>(Ty)) {
110895819069SCraig Topper     // Accept variable if it is already of bit type!
110995819069SCraig Topper     if (isa<BitRecTy>(getType()))
111095819069SCraig Topper       return const_cast<TypedInit *>(this);
111195819069SCraig Topper     if (auto *BitsTy = dyn_cast<BitsRecTy>(getType())) {
111295819069SCraig Topper       // Accept only bits<1> expression.
111395819069SCraig Topper       if (BitsTy->getNumBits() == 1)
111495819069SCraig Topper         return const_cast<TypedInit *>(this);
111595819069SCraig Topper       return nullptr;
111695819069SCraig Topper     }
111795819069SCraig Topper     // Ternary !if can be converted to bit, but only if both sides are
111895819069SCraig Topper     // convertible to a bit.
111995819069SCraig Topper     if (const auto *TOI = dyn_cast<TernOpInit>(this)) {
112095819069SCraig Topper       if (TOI->getOpcode() == TernOpInit::TernaryOp::IF &&
112195819069SCraig Topper           TOI->getMHS()->convertInitializerTo(BitRecTy::get()) &&
112295819069SCraig Topper           TOI->getRHS()->convertInitializerTo(BitRecTy::get()))
112395819069SCraig Topper         return const_cast<TypedInit *>(this);
112495819069SCraig Topper       return nullptr;
112595819069SCraig Topper     }
112695819069SCraig Topper     return nullptr;
112795819069SCraig Topper   }
112895819069SCraig Topper 
112995819069SCraig Topper   if (auto *BRT = dyn_cast<BitsRecTy>(Ty)) {
113095819069SCraig Topper     if (BRT->getNumBits() == 1 && isa<BitRecTy>(getType()))
113195819069SCraig Topper       return BitsInit::get(const_cast<TypedInit *>(this));
113295819069SCraig Topper 
113395819069SCraig Topper     if (getType()->typeIsConvertibleTo(BRT)) {
113495819069SCraig Topper       SmallVector<Init *, 16> NewBits(BRT->getNumBits());
113595819069SCraig Topper 
113695819069SCraig Topper       for (unsigned i = 0; i != BRT->getNumBits(); ++i)
113795819069SCraig Topper         NewBits[i] = VarBitInit::get(const_cast<TypedInit *>(this), i);
113895819069SCraig Topper       return BitsInit::get(NewBits);
113995819069SCraig Topper     }
114095819069SCraig Topper 
114195819069SCraig Topper     return nullptr;
114295819069SCraig Topper   }
114395819069SCraig Topper 
114495819069SCraig Topper   if (auto *DLRT = dyn_cast<ListRecTy>(Ty)) {
114595819069SCraig Topper     if (auto *SLRT = dyn_cast<ListRecTy>(getType()))
114695819069SCraig Topper       if (SLRT->getElementType()->typeIsConvertibleTo(DLRT->getElementType()))
114795819069SCraig Topper         return const_cast<TypedInit *>(this);
114895819069SCraig Topper     return nullptr;
114995819069SCraig Topper   }
115095819069SCraig Topper 
115195819069SCraig Topper   if (auto *DRT = dyn_cast<DagRecTy>(Ty)) {
115295819069SCraig Topper     if (getType()->typeIsConvertibleTo(DRT))
115395819069SCraig Topper       return const_cast<TypedInit *>(this);
115495819069SCraig Topper     return nullptr;
115595819069SCraig Topper   }
115695819069SCraig Topper 
115795819069SCraig Topper   if (auto *SRRT = dyn_cast<RecordRecTy>(Ty)) {
115895819069SCraig Topper     // Ensure that this is compatible with Rec.
115995819069SCraig Topper     if (RecordRecTy *DRRT = dyn_cast<RecordRecTy>(getType()))
116095819069SCraig Topper       if (DRRT->getRecord()->isSubClassOf(SRRT->getRecord()) ||
116195819069SCraig Topper           DRRT->getRecord() == SRRT->getRecord())
116295819069SCraig Topper         return const_cast<TypedInit *>(this);
116395819069SCraig Topper     return nullptr;
116495819069SCraig Topper   }
116595819069SCraig Topper 
116695819069SCraig Topper   return nullptr;
116795819069SCraig Topper }
116895819069SCraig Topper 
116995819069SCraig Topper Init *
117084c287e3SPeter Collingbourne TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
117198c61711SSean Silva   BitsRecTy *T = dyn_cast<BitsRecTy>(getType());
1172011817a0SCraig Topper   if (!T) return nullptr;  // Cannot subscript a non-bits variable.
117384c287e3SPeter Collingbourne   unsigned NumBits = T->getNumBits();
117484c287e3SPeter Collingbourne 
117584c287e3SPeter Collingbourne   SmallVector<Init *, 16> NewBits(Bits.size());
117684c287e3SPeter Collingbourne   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
117784c287e3SPeter Collingbourne     if (Bits[i] >= NumBits)
1178011817a0SCraig Topper       return nullptr;
117984c287e3SPeter Collingbourne 
118084c287e3SPeter Collingbourne     NewBits[i] = VarBitInit::get(const_cast<TypedInit *>(this), Bits[i]);
118184c287e3SPeter Collingbourne   }
118284c287e3SPeter Collingbourne   return BitsInit::get(NewBits);
118384c287e3SPeter Collingbourne }
118484c287e3SPeter Collingbourne 
118584c287e3SPeter Collingbourne Init *
118684c287e3SPeter Collingbourne TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) const {
118798c61711SSean Silva   ListRecTy *T = dyn_cast<ListRecTy>(getType());
1188011817a0SCraig Topper   if (!T) return nullptr;  // Cannot subscript a non-list variable.
118984c287e3SPeter Collingbourne 
119084c287e3SPeter Collingbourne   if (Elements.size() == 1)
119184c287e3SPeter Collingbourne     return VarListElementInit::get(const_cast<TypedInit *>(this), Elements[0]);
119284c287e3SPeter Collingbourne 
119384c287e3SPeter Collingbourne   std::vector<Init*> ListInits;
119484c287e3SPeter Collingbourne   ListInits.reserve(Elements.size());
119584c287e3SPeter Collingbourne   for (unsigned i = 0, e = Elements.size(); i != e; ++i)
119684c287e3SPeter Collingbourne     ListInits.push_back(VarListElementInit::get(const_cast<TypedInit *>(this),
119784c287e3SPeter Collingbourne                                                 Elements[i]));
119884c287e3SPeter Collingbourne   return ListInit::get(ListInits, T);
119984c287e3SPeter Collingbourne }
120084c287e3SPeter Collingbourne 
120184c287e3SPeter Collingbourne 
120284c287e3SPeter Collingbourne VarInit *VarInit::get(const std::string &VN, RecTy *T) {
1203914adf0eSDavid Greene   Init *Value = StringInit::get(VN);
1204914adf0eSDavid Greene   return VarInit::get(Value, T);
1205914adf0eSDavid Greene }
1206914adf0eSDavid Greene 
1207914adf0eSDavid Greene VarInit *VarInit::get(Init *VN, RecTy *T) {
1208914adf0eSDavid Greene   typedef std::pair<RecTy *, Init *> Key;
12090e04bee8SCraig Topper   static DenseMap<Key, std::unique_ptr<VarInit>> ThePool;
121084c287e3SPeter Collingbourne 
121184c287e3SPeter Collingbourne   Key TheKey(std::make_pair(T, VN));
121284c287e3SPeter Collingbourne 
12130e04bee8SCraig Topper   std::unique_ptr<VarInit> &I = ThePool[TheKey];
12140e04bee8SCraig Topper   if (!I) I.reset(new VarInit(VN, T));
12150e04bee8SCraig Topper   return I.get();
121684c287e3SPeter Collingbourne }
121784c287e3SPeter Collingbourne 
1218914adf0eSDavid Greene const std::string &VarInit::getName() const {
1219ed5a9508SCraig Topper   StringInit *NameString = cast<StringInit>(getNameInit());
1220914adf0eSDavid Greene   return NameString->getValue();
1221914adf0eSDavid Greene }
1222914adf0eSDavid Greene 
1223026f8333SMichael Liao Init *VarInit::getBit(unsigned Bit) const {
1224026f8333SMichael Liao   if (getType() == BitRecTy::get())
1225026f8333SMichael Liao     return const_cast<VarInit*>(this);
1226026f8333SMichael Liao   return VarBitInit::get(const_cast<VarInit*>(this), Bit);
122784c287e3SPeter Collingbourne }
122884c287e3SPeter Collingbourne 
122984c287e3SPeter Collingbourne Init *VarInit::resolveListElementReference(Record &R,
123084c287e3SPeter Collingbourne                                            const RecordVal *IRV,
123184c287e3SPeter Collingbourne                                            unsigned Elt) const {
1232011817a0SCraig Topper   if (R.isTemplateArg(getNameInit())) return nullptr;
1233011817a0SCraig Topper   if (IRV && IRV->getNameInit() != getNameInit()) return nullptr;
123484c287e3SPeter Collingbourne 
12359d1c5eebSJakob Stoklund Olesen   RecordVal *RV = R.getValue(getNameInit());
123684c287e3SPeter Collingbourne   assert(RV && "Reference to a non-existent variable?");
1237fb509ed1SSean Silva   ListInit *LI = dyn_cast<ListInit>(RV->getValue());
1238a9642b4eSCraig Topper   if (!LI)
1239a9642b4eSCraig Topper     return VarListElementInit::get(cast<TypedInit>(RV->getValue()), Elt);
124084c287e3SPeter Collingbourne 
1241664f6a04SCraig Topper   if (Elt >= LI->size())
1242011817a0SCraig Topper     return nullptr;  // Out of range reference.
124384c287e3SPeter Collingbourne   Init *E = LI->getElement(Elt);
124484c287e3SPeter Collingbourne   // If the element is set to some value, or if we are resolving a reference
124584c287e3SPeter Collingbourne   // to a specific variable and that variable is explicitly unset, then
124684c287e3SPeter Collingbourne   // replace the VarListElementInit with it.
124788eb8dd4SSean Silva   if (IRV || !isa<UnsetInit>(E))
124884c287e3SPeter Collingbourne     return E;
1249011817a0SCraig Topper   return nullptr;
125084c287e3SPeter Collingbourne }
125184c287e3SPeter Collingbourne 
125284c287e3SPeter Collingbourne 
125384c287e3SPeter Collingbourne RecTy *VarInit::getFieldType(const std::string &FieldName) const {
125498c61711SSean Silva   if (RecordRecTy *RTy = dyn_cast<RecordRecTy>(getType()))
125584c287e3SPeter Collingbourne     if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName))
125684c287e3SPeter Collingbourne       return RV->getType();
1257011817a0SCraig Topper   return nullptr;
125884c287e3SPeter Collingbourne }
125984c287e3SPeter Collingbourne 
126084c287e3SPeter Collingbourne Init *VarInit::getFieldInit(Record &R, const RecordVal *RV,
126184c287e3SPeter Collingbourne                             const std::string &FieldName) const {
1262a475e433SSean Silva   if (isa<RecordRecTy>(getType()))
126384c287e3SPeter Collingbourne     if (const RecordVal *Val = R.getValue(VarName)) {
126488eb8dd4SSean Silva       if (RV != Val && (RV || isa<UnsetInit>(Val->getValue())))
1265011817a0SCraig Topper         return nullptr;
126684c287e3SPeter Collingbourne       Init *TheInit = Val->getValue();
126784c287e3SPeter Collingbourne       assert(TheInit != this && "Infinite loop detected!");
126884c287e3SPeter Collingbourne       if (Init *I = TheInit->getFieldInit(R, RV, FieldName))
126984c287e3SPeter Collingbourne         return I;
1270011817a0SCraig Topper       return nullptr;
127184c287e3SPeter Collingbourne     }
1272011817a0SCraig Topper   return nullptr;
127384c287e3SPeter Collingbourne }
127484c287e3SPeter Collingbourne 
127584c287e3SPeter Collingbourne /// resolveReferences - This method is used by classes that refer to other
127684c287e3SPeter Collingbourne /// variables which may not be defined at the time the expression is formed.
127784c287e3SPeter Collingbourne /// If a value is set for the variable later, this method will be called on
127884c287e3SPeter Collingbourne /// users of the value to allow the value to propagate out.
127984c287e3SPeter Collingbourne ///
128084c287e3SPeter Collingbourne Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) const {
128184c287e3SPeter Collingbourne   if (RecordVal *Val = R.getValue(VarName))
1282011817a0SCraig Topper     if (RV == Val || (!RV && !isa<UnsetInit>(Val->getValue())))
128384c287e3SPeter Collingbourne       return Val->getValue();
128484c287e3SPeter Collingbourne   return const_cast<VarInit *>(this);
128584c287e3SPeter Collingbourne }
128684c287e3SPeter Collingbourne 
128784c287e3SPeter Collingbourne VarBitInit *VarBitInit::get(TypedInit *T, unsigned B) {
128884c287e3SPeter Collingbourne   typedef std::pair<TypedInit *, unsigned> Key;
12890e04bee8SCraig Topper   static DenseMap<Key, std::unique_ptr<VarBitInit>> ThePool;
129084c287e3SPeter Collingbourne 
129184c287e3SPeter Collingbourne   Key TheKey(std::make_pair(T, B));
129284c287e3SPeter Collingbourne 
12930e04bee8SCraig Topper   std::unique_ptr<VarBitInit> &I = ThePool[TheKey];
12940e04bee8SCraig Topper   if (!I) I.reset(new VarBitInit(T, B));
12950e04bee8SCraig Topper   return I.get();
129684c287e3SPeter Collingbourne }
129784c287e3SPeter Collingbourne 
129895819069SCraig Topper Init *VarBitInit::convertInitializerTo(RecTy *Ty) const {
129995819069SCraig Topper   if (isa<BitRecTy>(Ty))
130095819069SCraig Topper     return const_cast<VarBitInit *>(this);
130195819069SCraig Topper 
130295819069SCraig Topper   return nullptr;
130395819069SCraig Topper }
130495819069SCraig Topper 
130584c287e3SPeter Collingbourne std::string VarBitInit::getAsString() const {
130684c287e3SPeter Collingbourne   return TI->getAsString() + "{" + utostr(Bit) + "}";
130784c287e3SPeter Collingbourne }
130884c287e3SPeter Collingbourne 
130984c287e3SPeter Collingbourne Init *VarBitInit::resolveReferences(Record &R, const RecordVal *RV) const {
1310026f8333SMichael Liao   Init *I = TI->resolveReferences(R, RV);
1311026f8333SMichael Liao   if (TI != I)
1312026f8333SMichael Liao     return I->getBit(getBitNum());
1313026f8333SMichael Liao 
131484c287e3SPeter Collingbourne   return const_cast<VarBitInit*>(this);
131584c287e3SPeter Collingbourne }
131684c287e3SPeter Collingbourne 
131784c287e3SPeter Collingbourne VarListElementInit *VarListElementInit::get(TypedInit *T,
131884c287e3SPeter Collingbourne                                             unsigned E) {
131984c287e3SPeter Collingbourne   typedef std::pair<TypedInit *, unsigned> Key;
132017b3a378SCraig Topper   static DenseMap<Key, std::unique_ptr<VarListElementInit>> ThePool;
132184c287e3SPeter Collingbourne 
132284c287e3SPeter Collingbourne   Key TheKey(std::make_pair(T, E));
132384c287e3SPeter Collingbourne 
132417b3a378SCraig Topper   std::unique_ptr<VarListElementInit> &I = ThePool[TheKey];
132517b3a378SCraig Topper   if (!I) I.reset(new VarListElementInit(T, E));
132617b3a378SCraig Topper   return I.get();
132784c287e3SPeter Collingbourne }
132884c287e3SPeter Collingbourne 
132984c287e3SPeter Collingbourne std::string VarListElementInit::getAsString() const {
133084c287e3SPeter Collingbourne   return TI->getAsString() + "[" + utostr(Element) + "]";
133184c287e3SPeter Collingbourne }
133284c287e3SPeter Collingbourne 
133384c287e3SPeter Collingbourne Init *
133484c287e3SPeter Collingbourne VarListElementInit::resolveReferences(Record &R, const RecordVal *RV) const {
133584c287e3SPeter Collingbourne   if (Init *I = getVariable()->resolveListElementReference(R, RV,
133684c287e3SPeter Collingbourne                                                            getElementNum()))
133784c287e3SPeter Collingbourne     return I;
133884c287e3SPeter Collingbourne   return const_cast<VarListElementInit *>(this);
133984c287e3SPeter Collingbourne }
134084c287e3SPeter Collingbourne 
1341026f8333SMichael Liao Init *VarListElementInit::getBit(unsigned Bit) const {
1342026f8333SMichael Liao   if (getType() == BitRecTy::get())
1343026f8333SMichael Liao     return const_cast<VarListElementInit*>(this);
1344026f8333SMichael Liao   return VarBitInit::get(const_cast<VarListElementInit*>(this), Bit);
134584c287e3SPeter Collingbourne }
134684c287e3SPeter Collingbourne 
134784c287e3SPeter Collingbourne Init *VarListElementInit:: resolveListElementReference(Record &R,
134884c287e3SPeter Collingbourne                                                        const RecordVal *RV,
134984c287e3SPeter Collingbourne                                                        unsigned Elt) const {
1350a9642b4eSCraig Topper   if (Init *Result = TI->resolveListElementReference(R, RV, Element)) {
135188eb8dd4SSean Silva     if (TypedInit *TInit = dyn_cast<TypedInit>(Result)) {
1352*5904beb6SCraig Topper       if (Init *Result2 = TInit->resolveListElementReference(R, RV, Elt))
1353*5904beb6SCraig Topper         return Result2;
135417b3a378SCraig Topper       return VarListElementInit::get(TInit, Elt);
135584c287e3SPeter Collingbourne     }
135684c287e3SPeter Collingbourne     return Result;
135784c287e3SPeter Collingbourne   }
135884c287e3SPeter Collingbourne 
1359011817a0SCraig Topper   return nullptr;
136084c287e3SPeter Collingbourne }
136184c287e3SPeter Collingbourne 
136284c287e3SPeter Collingbourne DefInit *DefInit::get(Record *R) {
136384c287e3SPeter Collingbourne   return R->getDefInit();
136484c287e3SPeter Collingbourne }
136584c287e3SPeter Collingbourne 
136695819069SCraig Topper Init *DefInit::convertInitializerTo(RecTy *Ty) const {
136795819069SCraig Topper   if (auto *RRT = dyn_cast<RecordRecTy>(Ty))
136895819069SCraig Topper     if (getDef()->isSubClassOf(RRT->getRecord()))
136995819069SCraig Topper       return const_cast<DefInit *>(this);
137095819069SCraig Topper   return nullptr;
137195819069SCraig Topper }
137295819069SCraig Topper 
137384c287e3SPeter Collingbourne RecTy *DefInit::getFieldType(const std::string &FieldName) const {
137484c287e3SPeter Collingbourne   if (const RecordVal *RV = Def->getValue(FieldName))
137584c287e3SPeter Collingbourne     return RV->getType();
1376011817a0SCraig Topper   return nullptr;
137784c287e3SPeter Collingbourne }
137884c287e3SPeter Collingbourne 
137984c287e3SPeter Collingbourne Init *DefInit::getFieldInit(Record &R, const RecordVal *RV,
138084c287e3SPeter Collingbourne                             const std::string &FieldName) const {
138184c287e3SPeter Collingbourne   return Def->getValue(FieldName)->getValue();
138284c287e3SPeter Collingbourne }
138384c287e3SPeter Collingbourne 
138484c287e3SPeter Collingbourne 
138584c287e3SPeter Collingbourne std::string DefInit::getAsString() const {
138684c287e3SPeter Collingbourne   return Def->getName();
138784c287e3SPeter Collingbourne }
138884c287e3SPeter Collingbourne 
138984c287e3SPeter Collingbourne FieldInit *FieldInit::get(Init *R, const std::string &FN) {
139084c287e3SPeter Collingbourne   typedef std::pair<Init *, TableGenStringKey> Key;
139117b3a378SCraig Topper   static DenseMap<Key, std::unique_ptr<FieldInit>> ThePool;
139284c287e3SPeter Collingbourne 
139384c287e3SPeter Collingbourne   Key TheKey(std::make_pair(R, FN));
139484c287e3SPeter Collingbourne 
139517b3a378SCraig Topper   std::unique_ptr<FieldInit> &I = ThePool[TheKey];
139617b3a378SCraig Topper   if (!I) I.reset(new FieldInit(R, FN));
139717b3a378SCraig Topper   return I.get();
139884c287e3SPeter Collingbourne }
139984c287e3SPeter Collingbourne 
1400026f8333SMichael Liao Init *FieldInit::getBit(unsigned Bit) const {
1401026f8333SMichael Liao   if (getType() == BitRecTy::get())
1402026f8333SMichael Liao     return const_cast<FieldInit*>(this);
1403026f8333SMichael Liao   return VarBitInit::get(const_cast<FieldInit*>(this), Bit);
140484c287e3SPeter Collingbourne }
140584c287e3SPeter Collingbourne 
140684c287e3SPeter Collingbourne Init *FieldInit::resolveListElementReference(Record &R, const RecordVal *RV,
140784c287e3SPeter Collingbourne                                              unsigned Elt) const {
140884c287e3SPeter Collingbourne   if (Init *ListVal = Rec->getFieldInit(R, RV, FieldName))
1409fb509ed1SSean Silva     if (ListInit *LI = dyn_cast<ListInit>(ListVal)) {
1410664f6a04SCraig Topper       if (Elt >= LI->size()) return nullptr;
141184c287e3SPeter Collingbourne       Init *E = LI->getElement(Elt);
141284c287e3SPeter Collingbourne 
141384c287e3SPeter Collingbourne       // If the element is set to some value, or if we are resolving a
141484c287e3SPeter Collingbourne       // reference to a specific variable and that variable is explicitly
141584c287e3SPeter Collingbourne       // unset, then replace the VarListElementInit with it.
141688eb8dd4SSean Silva       if (RV || !isa<UnsetInit>(E))
141784c287e3SPeter Collingbourne         return E;
141884c287e3SPeter Collingbourne     }
1419011817a0SCraig Topper   return nullptr;
142084c287e3SPeter Collingbourne }
142184c287e3SPeter Collingbourne 
142284c287e3SPeter Collingbourne Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) const {
142384c287e3SPeter Collingbourne   Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec;
142484c287e3SPeter Collingbourne 
1425a9642b4eSCraig Topper   if (Init *BitsVal = NewRec->getFieldInit(R, RV, FieldName)) {
142684c287e3SPeter Collingbourne     Init *BVR = BitsVal->resolveReferences(R, RV);
142784c287e3SPeter Collingbourne     return BVR->isComplete() ? BVR : const_cast<FieldInit *>(this);
142884c287e3SPeter Collingbourne   }
142984c287e3SPeter Collingbourne 
1430a9642b4eSCraig Topper   if (NewRec != Rec)
143184c287e3SPeter Collingbourne     return FieldInit::get(NewRec, FieldName);
143284c287e3SPeter Collingbourne   return const_cast<FieldInit *>(this);
143384c287e3SPeter Collingbourne }
143484c287e3SPeter Collingbourne 
14356ecb1e78SBenjamin Kramer static void ProfileDagInit(FoldingSetNodeID &ID, Init *V, const std::string &VN,
143684c287e3SPeter Collingbourne                            ArrayRef<Init *> ArgRange,
143784c287e3SPeter Collingbourne                            ArrayRef<std::string> NameRange) {
143884c287e3SPeter Collingbourne   ID.AddPointer(V);
143984c287e3SPeter Collingbourne   ID.AddString(VN);
144084c287e3SPeter Collingbourne 
144184c287e3SPeter Collingbourne   ArrayRef<Init *>::iterator Arg  = ArgRange.begin();
144284c287e3SPeter Collingbourne   ArrayRef<std::string>::iterator  Name = NameRange.begin();
144384c287e3SPeter Collingbourne   while (Arg != ArgRange.end()) {
144484c287e3SPeter Collingbourne     assert(Name != NameRange.end() && "Arg name underflow!");
144584c287e3SPeter Collingbourne     ID.AddPointer(*Arg++);
144684c287e3SPeter Collingbourne     ID.AddString(*Name++);
144784c287e3SPeter Collingbourne   }
144884c287e3SPeter Collingbourne   assert(Name == NameRange.end() && "Arg name overflow!");
144984c287e3SPeter Collingbourne }
145084c287e3SPeter Collingbourne 
145184c287e3SPeter Collingbourne DagInit *
145284c287e3SPeter Collingbourne DagInit::get(Init *V, const std::string &VN,
145384c287e3SPeter Collingbourne              ArrayRef<Init *> ArgRange,
145484c287e3SPeter Collingbourne              ArrayRef<std::string> NameRange) {
145517b3a378SCraig Topper   static FoldingSet<DagInit> ThePool;
145617b3a378SCraig Topper   static std::vector<std::unique_ptr<DagInit>> TheActualPool;
145784c287e3SPeter Collingbourne 
145884c287e3SPeter Collingbourne   FoldingSetNodeID ID;
145984c287e3SPeter Collingbourne   ProfileDagInit(ID, V, VN, ArgRange, NameRange);
146084c287e3SPeter Collingbourne 
1461011817a0SCraig Topper   void *IP = nullptr;
146284c287e3SPeter Collingbourne   if (DagInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
146384c287e3SPeter Collingbourne     return I;
146484c287e3SPeter Collingbourne 
146584c287e3SPeter Collingbourne   DagInit *I = new DagInit(V, VN, ArgRange, NameRange);
146684c287e3SPeter Collingbourne   ThePool.InsertNode(I, IP);
146717b3a378SCraig Topper   TheActualPool.push_back(std::unique_ptr<DagInit>(I));
146884c287e3SPeter Collingbourne   return I;
146984c287e3SPeter Collingbourne }
147084c287e3SPeter Collingbourne 
147184c287e3SPeter Collingbourne DagInit *
147284c287e3SPeter Collingbourne DagInit::get(Init *V, const std::string &VN,
147384c287e3SPeter Collingbourne              const std::vector<std::pair<Init*, std::string> > &args) {
147484c287e3SPeter Collingbourne   std::vector<Init *> Args;
147584c287e3SPeter Collingbourne   std::vector<std::string> Names;
147684c287e3SPeter Collingbourne 
1477119998dbSCraig Topper   for (const auto &Arg : args) {
1478119998dbSCraig Topper     Args.push_back(Arg.first);
1479119998dbSCraig Topper     Names.push_back(Arg.second);
148084c287e3SPeter Collingbourne   }
148184c287e3SPeter Collingbourne 
148284c287e3SPeter Collingbourne   return DagInit::get(V, VN, Args, Names);
148384c287e3SPeter Collingbourne }
148484c287e3SPeter Collingbourne 
148584c287e3SPeter Collingbourne void DagInit::Profile(FoldingSetNodeID &ID) const {
148684c287e3SPeter Collingbourne   ProfileDagInit(ID, Val, ValName, Args, ArgNames);
148784c287e3SPeter Collingbourne }
148884c287e3SPeter Collingbourne 
148995819069SCraig Topper Init *DagInit::convertInitializerTo(RecTy *Ty) const {
149095819069SCraig Topper   if (isa<DagRecTy>(Ty))
149195819069SCraig Topper     return const_cast<DagInit *>(this);
149295819069SCraig Topper 
149395819069SCraig Topper   return nullptr;
149495819069SCraig Topper }
149595819069SCraig Topper 
149684c287e3SPeter Collingbourne Init *DagInit::resolveReferences(Record &R, const RecordVal *RV) const {
149784c287e3SPeter Collingbourne   std::vector<Init*> NewArgs;
149884c287e3SPeter Collingbourne   for (unsigned i = 0, e = Args.size(); i != e; ++i)
149984c287e3SPeter Collingbourne     NewArgs.push_back(Args[i]->resolveReferences(R, RV));
150084c287e3SPeter Collingbourne 
150184c287e3SPeter Collingbourne   Init *Op = Val->resolveReferences(R, RV);
150284c287e3SPeter Collingbourne 
150384c287e3SPeter Collingbourne   if (Args != NewArgs || Op != Val)
150484c287e3SPeter Collingbourne     return DagInit::get(Op, ValName, NewArgs, ArgNames);
150584c287e3SPeter Collingbourne 
150684c287e3SPeter Collingbourne   return const_cast<DagInit *>(this);
150784c287e3SPeter Collingbourne }
150884c287e3SPeter Collingbourne 
150984c287e3SPeter Collingbourne 
151084c287e3SPeter Collingbourne std::string DagInit::getAsString() const {
151184c287e3SPeter Collingbourne   std::string Result = "(" + Val->getAsString();
151284c287e3SPeter Collingbourne   if (!ValName.empty())
151384c287e3SPeter Collingbourne     Result += ":" + ValName;
15148c0809c7SAlexander Kornienko   if (!Args.empty()) {
151584c287e3SPeter Collingbourne     Result += " " + Args[0]->getAsString();
151684c287e3SPeter Collingbourne     if (!ArgNames[0].empty()) Result += ":$" + ArgNames[0];
151784c287e3SPeter Collingbourne     for (unsigned i = 1, e = Args.size(); i != e; ++i) {
151884c287e3SPeter Collingbourne       Result += ", " + Args[i]->getAsString();
151984c287e3SPeter Collingbourne       if (!ArgNames[i].empty()) Result += ":$" + ArgNames[i];
152084c287e3SPeter Collingbourne     }
152184c287e3SPeter Collingbourne   }
152284c287e3SPeter Collingbourne   return Result + ")";
152384c287e3SPeter Collingbourne }
152484c287e3SPeter Collingbourne 
152584c287e3SPeter Collingbourne 
152684c287e3SPeter Collingbourne //===----------------------------------------------------------------------===//
152784c287e3SPeter Collingbourne //    Other implementations
152884c287e3SPeter Collingbourne //===----------------------------------------------------------------------===//
152984c287e3SPeter Collingbourne 
1530b849203cSCraig Topper RecordVal::RecordVal(Init *N, RecTy *T, bool P)
1531b849203cSCraig Topper   : NameAndPrefix(N, P), Ty(T) {
153295819069SCraig Topper   Value = UnsetInit::get()->convertInitializerTo(Ty);
153384c287e3SPeter Collingbourne   assert(Value && "Cannot create unset value for current type!");
153484c287e3SPeter Collingbourne }
153584c287e3SPeter Collingbourne 
1536b849203cSCraig Topper RecordVal::RecordVal(const std::string &N, RecTy *T, bool P)
1537b849203cSCraig Topper   : NameAndPrefix(StringInit::get(N), P), Ty(T) {
153895819069SCraig Topper   Value = UnsetInit::get()->convertInitializerTo(Ty);
153984c287e3SPeter Collingbourne   assert(Value && "Cannot create unset value for current type!");
154084c287e3SPeter Collingbourne }
154184c287e3SPeter Collingbourne 
154284c287e3SPeter Collingbourne const std::string &RecordVal::getName() const {
1543b849203cSCraig Topper   return cast<StringInit>(getNameInit())->getValue();
154484c287e3SPeter Collingbourne }
154584c287e3SPeter Collingbourne 
154684c287e3SPeter Collingbourne void RecordVal::dump() const { errs() << *this; }
154784c287e3SPeter Collingbourne 
154884c287e3SPeter Collingbourne void RecordVal::print(raw_ostream &OS, bool PrintSem) const {
154984c287e3SPeter Collingbourne   if (getPrefix()) OS << "field ";
1550658a4b71SDavid Greene   OS << *getType() << " " << getNameInitAsString();
155184c287e3SPeter Collingbourne 
155284c287e3SPeter Collingbourne   if (getValue())
155384c287e3SPeter Collingbourne     OS << " = " << *getValue();
155484c287e3SPeter Collingbourne 
155584c287e3SPeter Collingbourne   if (PrintSem) OS << ";\n";
155684c287e3SPeter Collingbourne }
155784c287e3SPeter Collingbourne 
155884c287e3SPeter Collingbourne unsigned Record::LastID = 0;
155984c287e3SPeter Collingbourne 
1560b77fc0d9SDavid Greene void Record::init() {
1561b77fc0d9SDavid Greene   checkName();
1562d699161aSDavid Greene 
1563d699161aSDavid Greene   // Every record potentially has a def at the top.  This value is
1564d699161aSDavid Greene   // replaced with the top-level def name at instantiation time.
1565d699161aSDavid Greene   RecordVal DN("NAME", StringRecTy::get(), 0);
1566d699161aSDavid Greene   addValue(DN);
1567b77fc0d9SDavid Greene }
1568b77fc0d9SDavid Greene 
156984c287e3SPeter Collingbourne void Record::checkName() {
157084c287e3SPeter Collingbourne   // Ensure the record name has string type.
1571ed5a9508SCraig Topper   const TypedInit *TypedName = cast<const TypedInit>(Name);
157284c287e3SPeter Collingbourne   RecTy *Type = TypedName->getType();
1573a475e433SSean Silva   if (!isa<StringRecTy>(Type))
1574635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record name is not a string!");
157584c287e3SPeter Collingbourne }
157684c287e3SPeter Collingbourne 
157784c287e3SPeter Collingbourne DefInit *Record::getDefInit() {
157817b3a378SCraig Topper   static DenseMap<Record *, std::unique_ptr<DefInit>> ThePool;
157917b3a378SCraig Topper   if (TheInit)
158084c287e3SPeter Collingbourne     return TheInit;
158117b3a378SCraig Topper 
158217b3a378SCraig Topper   std::unique_ptr<DefInit> &I = ThePool[this];
158317b3a378SCraig Topper   if (!I) I.reset(new DefInit(this, new RecordRecTy(this)));
158417b3a378SCraig Topper   return I.get();
158584c287e3SPeter Collingbourne }
158684c287e3SPeter Collingbourne 
158784c287e3SPeter Collingbourne const std::string &Record::getName() const {
1588ed5a9508SCraig Topper   return cast<StringInit>(Name)->getValue();
158984c287e3SPeter Collingbourne }
159084c287e3SPeter Collingbourne 
159184c287e3SPeter Collingbourne void Record::setName(Init *NewName) {
1592f651a2a5SDavid Greene   Name = NewName;
159384c287e3SPeter Collingbourne   checkName();
159484c287e3SPeter Collingbourne   // DO NOT resolve record values to the name at this point because
159584c287e3SPeter Collingbourne   // there might be default values for arguments of this def.  Those
159684c287e3SPeter Collingbourne   // arguments might not have been resolved yet so we don't want to
159784c287e3SPeter Collingbourne   // prematurely assume values for those arguments were not passed to
159884c287e3SPeter Collingbourne   // this def.
159984c287e3SPeter Collingbourne   //
160084c287e3SPeter Collingbourne   // Nonetheless, it may be that some of this Record's values
160184c287e3SPeter Collingbourne   // reference the record name.  Indeed, the reason for having the
160284c287e3SPeter Collingbourne   // record name be an Init is to provide this flexibility.  The extra
160384c287e3SPeter Collingbourne   // resolve steps after completely instantiating defs takes care of
160484c287e3SPeter Collingbourne   // this.  See TGParser::ParseDef and TGParser::ParseDefm.
160584c287e3SPeter Collingbourne }
160684c287e3SPeter Collingbourne 
160784c287e3SPeter Collingbourne void Record::setName(const std::string &Name) {
160884c287e3SPeter Collingbourne   setName(StringInit::get(Name));
160984c287e3SPeter Collingbourne }
161084c287e3SPeter Collingbourne 
161184c287e3SPeter Collingbourne /// resolveReferencesTo - If anything in this record refers to RV, replace the
161284c287e3SPeter Collingbourne /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
161384c287e3SPeter Collingbourne /// references.
161484c287e3SPeter Collingbourne void Record::resolveReferencesTo(const RecordVal *RV) {
161584c287e3SPeter Collingbourne   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
1616aa0f752fSJakob Stoklund Olesen     if (RV == &Values[i]) // Skip resolve the same field as the given one
1617aa0f752fSJakob Stoklund Olesen       continue;
161884c287e3SPeter Collingbourne     if (Init *V = Values[i].getValue())
1619026f8333SMichael Liao       if (Values[i].setValue(V->resolveReferences(*this, RV)))
162085c07007SCraig Topper         PrintFatalError(getLoc(), "Invalid value is found when setting '" +
162185c07007SCraig Topper                         Values[i].getNameInitAsString() +
162285c07007SCraig Topper                         "' after resolving references" +
162385c07007SCraig Topper                         (RV ? " against '" + RV->getNameInitAsString() +
162485c07007SCraig Topper                               "' of (" + RV->getValue()->getAsUnquotedString() +
162585c07007SCraig Topper                               ")"
162685c07007SCraig Topper                             : "") + "\n");
162784c287e3SPeter Collingbourne   }
16289effff28SDavid Greene   Init *OldName = getNameInit();
16299effff28SDavid Greene   Init *NewName = Name->resolveReferences(*this, RV);
16309effff28SDavid Greene   if (NewName != OldName) {
16319effff28SDavid Greene     // Re-register with RecordKeeper.
16329effff28SDavid Greene     setName(NewName);
16339effff28SDavid Greene   }
163484c287e3SPeter Collingbourne }
163584c287e3SPeter Collingbourne 
163684c287e3SPeter Collingbourne void Record::dump() const { errs() << *this; }
163784c287e3SPeter Collingbourne 
163884c287e3SPeter Collingbourne raw_ostream &llvm::operator<<(raw_ostream &OS, const Record &R) {
163998be78aeSDavid Greene   OS << R.getNameInitAsString();
164084c287e3SPeter Collingbourne 
1641db10e691SDavid Greene   const std::vector<Init *> &TArgs = R.getTemplateArgs();
164284c287e3SPeter Collingbourne   if (!TArgs.empty()) {
164384c287e3SPeter Collingbourne     OS << "<";
16445ec17246SCraig Topper     bool NeedComma = false;
16455ec17246SCraig Topper     for (const Init *TA : TArgs) {
16465ec17246SCraig Topper       if (NeedComma) OS << ", ";
16475ec17246SCraig Topper       NeedComma = true;
16485ec17246SCraig Topper       const RecordVal *RV = R.getValue(TA);
164984c287e3SPeter Collingbourne       assert(RV && "Template argument record not found??");
165084c287e3SPeter Collingbourne       RV->print(OS, false);
165184c287e3SPeter Collingbourne     }
165284c287e3SPeter Collingbourne     OS << ">";
165384c287e3SPeter Collingbourne   }
165484c287e3SPeter Collingbourne 
165584c287e3SPeter Collingbourne   OS << " {";
165684c287e3SPeter Collingbourne   const std::vector<Record*> &SC = R.getSuperClasses();
165784c287e3SPeter Collingbourne   if (!SC.empty()) {
165884c287e3SPeter Collingbourne     OS << "\t//";
16595ec17246SCraig Topper     for (const Record *Super : SC)
16605ec17246SCraig Topper       OS << " " << Super->getNameInitAsString();
166184c287e3SPeter Collingbourne   }
166284c287e3SPeter Collingbourne   OS << "\n";
166384c287e3SPeter Collingbourne 
16645ec17246SCraig Topper   for (const RecordVal &Val : R.getValues())
16655ec17246SCraig Topper     if (Val.getPrefix() && !R.isTemplateArg(Val.getName()))
16665ec17246SCraig Topper       OS << Val;
16675ec17246SCraig Topper   for (const RecordVal &Val : R.getValues())
16685ec17246SCraig Topper     if (!Val.getPrefix() && !R.isTemplateArg(Val.getName()))
16695ec17246SCraig Topper       OS << Val;
167084c287e3SPeter Collingbourne 
167184c287e3SPeter Collingbourne   return OS << "}\n";
167284c287e3SPeter Collingbourne }
167384c287e3SPeter Collingbourne 
167484c287e3SPeter Collingbourne /// getValueInit - Return the initializer for a value with the specified name,
1675635debe8SJoerg Sonnenberger /// or abort if the field does not exist.
167684c287e3SPeter Collingbourne ///
167784c287e3SPeter Collingbourne Init *Record::getValueInit(StringRef FieldName) const {
167884c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1679011817a0SCraig Topper   if (!R || !R->getValue())
1680635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
168148e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
168284c287e3SPeter Collingbourne   return R->getValue();
168384c287e3SPeter Collingbourne }
168484c287e3SPeter Collingbourne 
168584c287e3SPeter Collingbourne 
168684c287e3SPeter Collingbourne /// getValueAsString - This method looks up the specified field and returns its
1687635debe8SJoerg Sonnenberger /// value as a string, aborts if the field does not exist or if
168884c287e3SPeter Collingbourne /// the value is not a string.
168984c287e3SPeter Collingbourne ///
169084c287e3SPeter Collingbourne std::string Record::getValueAsString(StringRef FieldName) const {
169184c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1692011817a0SCraig Topper   if (!R || !R->getValue())
1693635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
169448e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
169584c287e3SPeter Collingbourne 
1696fb509ed1SSean Silva   if (StringInit *SI = dyn_cast<StringInit>(R->getValue()))
169784c287e3SPeter Collingbourne     return SI->getValue();
1698635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
169948e7e85dSBenjamin Kramer     FieldName + "' does not have a string initializer!");
170084c287e3SPeter Collingbourne }
170184c287e3SPeter Collingbourne 
170284c287e3SPeter Collingbourne /// getValueAsBitsInit - This method looks up the specified field and returns
1703635debe8SJoerg Sonnenberger /// its value as a BitsInit, aborts if the field does not exist or if
1704635debe8SJoerg Sonnenberger /// the value is not the right type.
170584c287e3SPeter Collingbourne ///
170684c287e3SPeter Collingbourne BitsInit *Record::getValueAsBitsInit(StringRef FieldName) const {
170784c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1708011817a0SCraig Topper   if (!R || !R->getValue())
1709635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
171048e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
171184c287e3SPeter Collingbourne 
1712fb509ed1SSean Silva   if (BitsInit *BI = dyn_cast<BitsInit>(R->getValue()))
171384c287e3SPeter Collingbourne     return BI;
1714635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
171548e7e85dSBenjamin Kramer     FieldName + "' does not have a BitsInit initializer!");
171684c287e3SPeter Collingbourne }
171784c287e3SPeter Collingbourne 
171884c287e3SPeter Collingbourne /// getValueAsListInit - This method looks up the specified field and returns
1719635debe8SJoerg Sonnenberger /// its value as a ListInit, aborting if the field does not exist or if
1720635debe8SJoerg Sonnenberger /// the value is not the right type.
172184c287e3SPeter Collingbourne ///
172284c287e3SPeter Collingbourne ListInit *Record::getValueAsListInit(StringRef FieldName) const {
172384c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1724011817a0SCraig Topper   if (!R || !R->getValue())
1725635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
172648e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
172784c287e3SPeter Collingbourne 
1728fb509ed1SSean Silva   if (ListInit *LI = dyn_cast<ListInit>(R->getValue()))
172984c287e3SPeter Collingbourne     return LI;
1730635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
173148e7e85dSBenjamin Kramer     FieldName + "' does not have a list initializer!");
173284c287e3SPeter Collingbourne }
173384c287e3SPeter Collingbourne 
173484c287e3SPeter Collingbourne /// getValueAsListOfDefs - This method looks up the specified field and returns
1735635debe8SJoerg Sonnenberger /// its value as a vector of records, aborting if the field does not exist
1736635debe8SJoerg Sonnenberger /// or if the value is not the right type.
173784c287e3SPeter Collingbourne ///
173884c287e3SPeter Collingbourne std::vector<Record*>
173984c287e3SPeter Collingbourne Record::getValueAsListOfDefs(StringRef FieldName) const {
174084c287e3SPeter Collingbourne   ListInit *List = getValueAsListInit(FieldName);
174184c287e3SPeter Collingbourne   std::vector<Record*> Defs;
1742ef0578a8SCraig Topper   for (Init *I : List->getValues()) {
1743ef0578a8SCraig Topper     if (DefInit *DI = dyn_cast<DefInit>(I))
174484c287e3SPeter Collingbourne       Defs.push_back(DI->getDef());
1745a9642b4eSCraig Topper     else
1746635debe8SJoerg Sonnenberger       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
174748e7e85dSBenjamin Kramer         FieldName + "' list is not entirely DefInit!");
174884c287e3SPeter Collingbourne   }
174984c287e3SPeter Collingbourne   return Defs;
175084c287e3SPeter Collingbourne }
175184c287e3SPeter Collingbourne 
175284c287e3SPeter Collingbourne /// getValueAsInt - This method looks up the specified field and returns its
1753635debe8SJoerg Sonnenberger /// value as an int64_t, aborting if the field does not exist or if the value
1754635debe8SJoerg Sonnenberger /// is not the right type.
175584c287e3SPeter Collingbourne ///
175684c287e3SPeter Collingbourne int64_t Record::getValueAsInt(StringRef FieldName) const {
175784c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1758011817a0SCraig Topper   if (!R || !R->getValue())
1759635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
176048e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
176184c287e3SPeter Collingbourne 
1762fb509ed1SSean Silva   if (IntInit *II = dyn_cast<IntInit>(R->getValue()))
176384c287e3SPeter Collingbourne     return II->getValue();
1764635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
176548e7e85dSBenjamin Kramer     FieldName + "' does not have an int initializer!");
176684c287e3SPeter Collingbourne }
176784c287e3SPeter Collingbourne 
176884c287e3SPeter Collingbourne /// getValueAsListOfInts - This method looks up the specified field and returns
1769635debe8SJoerg Sonnenberger /// its value as a vector of integers, aborting if the field does not exist or
1770635debe8SJoerg Sonnenberger /// if the value is not the right type.
177184c287e3SPeter Collingbourne ///
177284c287e3SPeter Collingbourne std::vector<int64_t>
177384c287e3SPeter Collingbourne Record::getValueAsListOfInts(StringRef FieldName) const {
177484c287e3SPeter Collingbourne   ListInit *List = getValueAsListInit(FieldName);
177584c287e3SPeter Collingbourne   std::vector<int64_t> Ints;
1776ef0578a8SCraig Topper   for (Init *I : List->getValues()) {
1777ef0578a8SCraig Topper     if (IntInit *II = dyn_cast<IntInit>(I))
177884c287e3SPeter Collingbourne       Ints.push_back(II->getValue());
1779a9642b4eSCraig Topper     else
1780635debe8SJoerg Sonnenberger       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
178148e7e85dSBenjamin Kramer         FieldName + "' does not have a list of ints initializer!");
178284c287e3SPeter Collingbourne   }
178384c287e3SPeter Collingbourne   return Ints;
178484c287e3SPeter Collingbourne }
178584c287e3SPeter Collingbourne 
178684c287e3SPeter Collingbourne /// getValueAsListOfStrings - This method looks up the specified field and
1787635debe8SJoerg Sonnenberger /// returns its value as a vector of strings, aborting if the field does not
1788635debe8SJoerg Sonnenberger /// exist or if the value is not the right type.
178984c287e3SPeter Collingbourne ///
179084c287e3SPeter Collingbourne std::vector<std::string>
179184c287e3SPeter Collingbourne Record::getValueAsListOfStrings(StringRef FieldName) const {
179284c287e3SPeter Collingbourne   ListInit *List = getValueAsListInit(FieldName);
179384c287e3SPeter Collingbourne   std::vector<std::string> Strings;
1794ef0578a8SCraig Topper   for (Init *I : List->getValues()) {
1795ef0578a8SCraig Topper     if (StringInit *SI = dyn_cast<StringInit>(I))
1796ef0578a8SCraig Topper       Strings.push_back(SI->getValue());
1797a9642b4eSCraig Topper     else
1798635debe8SJoerg Sonnenberger       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
179948e7e85dSBenjamin Kramer         FieldName + "' does not have a list of strings initializer!");
180084c287e3SPeter Collingbourne   }
180184c287e3SPeter Collingbourne   return Strings;
180284c287e3SPeter Collingbourne }
180384c287e3SPeter Collingbourne 
180484c287e3SPeter Collingbourne /// getValueAsDef - This method looks up the specified field and returns its
1805635debe8SJoerg Sonnenberger /// value as a Record, aborting if the field does not exist or if the value
1806635debe8SJoerg Sonnenberger /// is not the right type.
180784c287e3SPeter Collingbourne ///
180884c287e3SPeter Collingbourne Record *Record::getValueAsDef(StringRef FieldName) const {
180984c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1810011817a0SCraig Topper   if (!R || !R->getValue())
1811635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
181248e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
181384c287e3SPeter Collingbourne 
1814fb509ed1SSean Silva   if (DefInit *DI = dyn_cast<DefInit>(R->getValue()))
181584c287e3SPeter Collingbourne     return DI->getDef();
1816635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
181748e7e85dSBenjamin Kramer     FieldName + "' does not have a def initializer!");
181884c287e3SPeter Collingbourne }
181984c287e3SPeter Collingbourne 
182084c287e3SPeter Collingbourne /// getValueAsBit - This method looks up the specified field and returns its
1821635debe8SJoerg Sonnenberger /// value as a bit, aborting if the field does not exist or if the value is
1822635debe8SJoerg Sonnenberger /// not the right type.
182384c287e3SPeter Collingbourne ///
182484c287e3SPeter Collingbourne bool Record::getValueAsBit(StringRef FieldName) const {
182584c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1826011817a0SCraig Topper   if (!R || !R->getValue())
1827635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
182848e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
182984c287e3SPeter Collingbourne 
1830fb509ed1SSean Silva   if (BitInit *BI = dyn_cast<BitInit>(R->getValue()))
183184c287e3SPeter Collingbourne     return BI->getValue();
1832635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
183348e7e85dSBenjamin Kramer     FieldName + "' does not have a bit initializer!");
183484c287e3SPeter Collingbourne }
183584c287e3SPeter Collingbourne 
1836af507bf4SJakob Stoklund Olesen bool Record::getValueAsBitOrUnset(StringRef FieldName, bool &Unset) const {
1837af507bf4SJakob Stoklund Olesen   const RecordVal *R = getValue(FieldName);
1838011817a0SCraig Topper   if (!R || !R->getValue())
1839635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
1840635debe8SJoerg Sonnenberger       "' does not have a field named `" + FieldName.str() + "'!\n");
1841af507bf4SJakob Stoklund Olesen 
18421bf3d1f5SCraig Topper   if (isa<UnsetInit>(R->getValue())) {
1843af507bf4SJakob Stoklund Olesen     Unset = true;
1844af507bf4SJakob Stoklund Olesen     return false;
1845af507bf4SJakob Stoklund Olesen   }
1846af507bf4SJakob Stoklund Olesen   Unset = false;
1847fb509ed1SSean Silva   if (BitInit *BI = dyn_cast<BitInit>(R->getValue()))
1848af507bf4SJakob Stoklund Olesen     return BI->getValue();
1849635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
185048e7e85dSBenjamin Kramer     FieldName + "' does not have a bit initializer!");
1851af507bf4SJakob Stoklund Olesen }
1852af507bf4SJakob Stoklund Olesen 
185384c287e3SPeter Collingbourne /// getValueAsDag - This method looks up the specified field and returns its
1854635debe8SJoerg Sonnenberger /// value as an Dag, aborting if the field does not exist or if the value is
1855635debe8SJoerg Sonnenberger /// not the right type.
185684c287e3SPeter Collingbourne ///
185784c287e3SPeter Collingbourne DagInit *Record::getValueAsDag(StringRef FieldName) const {
185884c287e3SPeter Collingbourne   const RecordVal *R = getValue(FieldName);
1859011817a0SCraig Topper   if (!R || !R->getValue())
1860635debe8SJoerg Sonnenberger     PrintFatalError(getLoc(), "Record `" + getName() +
186148e7e85dSBenjamin Kramer       "' does not have a field named `" + FieldName + "'!\n");
186284c287e3SPeter Collingbourne 
1863fb509ed1SSean Silva   if (DagInit *DI = dyn_cast<DagInit>(R->getValue()))
186484c287e3SPeter Collingbourne     return DI;
1865635debe8SJoerg Sonnenberger   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
186648e7e85dSBenjamin Kramer     FieldName + "' does not have a dag initializer!");
186784c287e3SPeter Collingbourne }
186884c287e3SPeter Collingbourne 
186984c287e3SPeter Collingbourne 
187084c287e3SPeter Collingbourne void MultiClass::dump() const {
187184c287e3SPeter Collingbourne   errs() << "Record:\n";
187284c287e3SPeter Collingbourne   Rec.dump();
187384c287e3SPeter Collingbourne 
187484c287e3SPeter Collingbourne   errs() << "Defs:\n";
1875a9642b4eSCraig Topper   for (const auto &Proto : DefPrototypes)
1876119998dbSCraig Topper     Proto->dump();
187784c287e3SPeter Collingbourne }
187884c287e3SPeter Collingbourne 
187984c287e3SPeter Collingbourne 
188084c287e3SPeter Collingbourne void RecordKeeper::dump() const { errs() << *this; }
188184c287e3SPeter Collingbourne 
188284c287e3SPeter Collingbourne raw_ostream &llvm::operator<<(raw_ostream &OS, const RecordKeeper &RK) {
188384c287e3SPeter Collingbourne   OS << "------------- Classes -----------------\n";
1884a0ff540bSCraig Topper   for (const auto &C : RK.getClasses())
188580f0e432SDylan Noblesmith     OS << "class " << *C.second;
188684c287e3SPeter Collingbourne 
188784c287e3SPeter Collingbourne   OS << "------------- Defs -----------------\n";
1888a0ff540bSCraig Topper   for (const auto &D : RK.getDefs())
188980f0e432SDylan Noblesmith     OS << "def " << *D.second;
189084c287e3SPeter Collingbourne   return OS;
189184c287e3SPeter Collingbourne }
189284c287e3SPeter Collingbourne 
189384c287e3SPeter Collingbourne 
189484c287e3SPeter Collingbourne /// getAllDerivedDefinitions - This method returns all concrete definitions
189584c287e3SPeter Collingbourne /// that derive from the specified class name.  If a class with the specified
189684c287e3SPeter Collingbourne /// name does not exist, an error is printed and true is returned.
189784c287e3SPeter Collingbourne std::vector<Record*>
189884c287e3SPeter Collingbourne RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
189984c287e3SPeter Collingbourne   Record *Class = getClass(ClassName);
190084c287e3SPeter Collingbourne   if (!Class)
1901635debe8SJoerg Sonnenberger     PrintFatalError("ERROR: Couldn't find the `" + ClassName + "' class!\n");
190284c287e3SPeter Collingbourne 
190384c287e3SPeter Collingbourne   std::vector<Record*> Defs;
190480f0e432SDylan Noblesmith   for (const auto &D : getDefs())
190580f0e432SDylan Noblesmith     if (D.second->isSubClassOf(Class))
1906085fc4d6SDylan Noblesmith       Defs.push_back(D.second.get());
190784c287e3SPeter Collingbourne 
190884c287e3SPeter Collingbourne   return Defs;
190984c287e3SPeter Collingbourne }
191084c287e3SPeter Collingbourne 
1911e714512cSDavid Greene /// QualifyName - Return an Init with a qualifier prefix referring
1912e714512cSDavid Greene /// to CurRec's name.
1913e714512cSDavid Greene Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass,
1914e714512cSDavid Greene                         Init *Name, const std::string &Scoper) {
1915619c4e57SBenjamin Kramer   RecTy *Type = cast<TypedInit>(Name)->getType();
1916e714512cSDavid Greene 
1917e714512cSDavid Greene   BinOpInit *NewName =
1918e714512cSDavid Greene     BinOpInit::get(BinOpInit::STRCONCAT,
1919e714512cSDavid Greene                       BinOpInit::get(BinOpInit::STRCONCAT,
1920e714512cSDavid Greene                                         CurRec.getNameInit(),
1921e714512cSDavid Greene                                         StringInit::get(Scoper),
1922e714512cSDavid Greene                                         Type)->Fold(&CurRec, CurMultiClass),
1923e714512cSDavid Greene                       Name,
1924e714512cSDavid Greene                       Type);
1925e714512cSDavid Greene 
1926e714512cSDavid Greene   if (CurMultiClass && Scoper != "::") {
1927e714512cSDavid Greene     NewName =
1928e714512cSDavid Greene       BinOpInit::get(BinOpInit::STRCONCAT,
1929e714512cSDavid Greene                         BinOpInit::get(BinOpInit::STRCONCAT,
1930e714512cSDavid Greene                                           CurMultiClass->Rec.getNameInit(),
1931e714512cSDavid Greene                                           StringInit::get("::"),
1932e714512cSDavid Greene                                           Type)->Fold(&CurRec, CurMultiClass),
1933e714512cSDavid Greene                         NewName->Fold(&CurRec, CurMultiClass),
1934e714512cSDavid Greene                         Type);
1935e714512cSDavid Greene   }
1936e714512cSDavid Greene 
1937e714512cSDavid Greene   return NewName->Fold(&CurRec, CurMultiClass);
1938e714512cSDavid Greene }
1939e714512cSDavid Greene 
1940e714512cSDavid Greene /// QualifyName - Return an Init with a qualifier prefix referring
1941e714512cSDavid Greene /// to CurRec's name.
1942e714512cSDavid Greene Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass,
1943e714512cSDavid Greene                         const std::string &Name,
1944e714512cSDavid Greene                         const std::string &Scoper) {
1945e714512cSDavid Greene   return QualifyName(CurRec, CurMultiClass, StringInit::get(Name), Scoper);
1946e714512cSDavid Greene }
1947