lib/TableGen/Record.cpp

   1 //===- Record.cpp - Record implementation ---------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // Implement the tablegen record classes.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/TableGen/Record.h"
  15 #include "llvm/ADT/DenseMap.h"
  16 #include "llvm/ADT/FoldingSet.h"
  17 #include "llvm/ADT/Hashing.h"
  18 #include "llvm/ADT/STLExtras.h"
  19 #include "llvm/ADT/SmallVector.h"
  20 #include "llvm/ADT/StringExtras.h"
  21 #include "llvm/ADT/StringMap.h"
  22 #include "llvm/Support/DataTypes.h"
  23 #include "llvm/Support/ErrorHandling.h"
  24 #include "llvm/Support/Format.h"
  25 #include "llvm/TableGen/Error.h"
  26
  27 using namespace llvm;
  28
  29 //===----------------------------------------------------------------------===//
  30 //    std::string wrapper for DenseMap purposes
  31 //===----------------------------------------------------------------------===//
  32
  33 namespace llvm {
  34
  35 /// TableGenStringKey - This is a wrapper for std::string suitable for
  36 /// using as a key to a DenseMap.  Because there isn't a particularly
  37 /// good way to indicate tombstone or empty keys for strings, we want
  38 /// to wrap std::string to indicate that this is a "special" string
  39 /// not expected to take on certain values (those of the tombstone and
  40 /// empty keys).  This makes things a little safer as it clarifies
  41 /// that DenseMap is really not appropriate for general strings.
  42
  43 class TableGenStringKey {
  44 public:
  45   TableGenStringKey(const std::string &str) : data(str) {}
  46   TableGenStringKey(const char *str) : data(str) {}
  47
  48   const std::string &str() const { return data; }
  49
  50   friend hash_code hash_value(const TableGenStringKey &Value) {
  51     using llvm::hash_value;
  52     return hash_value(Value.str());
  53   }
  54 private:
  55   std::string data;
  56 };
  57
  58 /// Specialize DenseMapInfo for TableGenStringKey.
  59 template<> struct DenseMapInfo<TableGenStringKey> {
  60   static inline TableGenStringKey getEmptyKey() {
  61     TableGenStringKey Empty("<<<EMPTY KEY>>>");
  62     return Empty;
  63   }
  64   static inline TableGenStringKey getTombstoneKey() {
  65     TableGenStringKey Tombstone("<<<TOMBSTONE KEY>>>");
  66     return Tombstone;
  67   }
  68   static unsigned getHashValue(const TableGenStringKey& Val) {
  69     using llvm::hash_value;
  70     return hash_value(Val);
  71   }
  72   static bool isEqual(const TableGenStringKey& LHS,
  73                       const TableGenStringKey& RHS) {
  74     return LHS.str() == RHS.str();
  75   }
  76 };
  77
  78 } // namespace llvm
  79
  80 //===----------------------------------------------------------------------===//
  81 //    Type implementations
  82 //===----------------------------------------------------------------------===//
  83
  84 BitRecTy BitRecTy::Shared;
  85 IntRecTy IntRecTy::Shared;
  86 StringRecTy StringRecTy::Shared;
  87 DagRecTy DagRecTy::Shared;
  88
  89 void RecTy::anchor() { }
  90 void RecTy::dump() const { print(errs()); }
  91
  92 ListRecTy *RecTy::getListTy() {
  93   if (!ListTy)
  94     ListTy.reset(new ListRecTy(this));
  95   return ListTy.get();
  96 }
  97
  98 bool RecTy::baseClassOf(const RecTy *RHS) const {
  99   assert (RHS && "NULL pointer");
 100   return Kind == RHS->getRecTyKind();
 101 }
 102
 103 Init *BitRecTy::convertValue(BitsInit *BI) {
 104   if (BI->getNumBits() != 1) return nullptr; // Only accept if just one bit!
 105   return BI->getBit(0);
 106 }
 107
 108 Init *BitRecTy::convertValue(IntInit *II) {
 109   int64_t Val = II->getValue();
 110   if (Val != 0 && Val != 1) return nullptr;  // Only accept 0 or 1 for a bit!
 111
 112   return BitInit::get(Val != 0);
 113 }
 114
 115 Init *BitRecTy::convertValue(TypedInit *TI) {
 116   RecTy *Ty = TI->getType();
 117   if (isa<BitRecTy>(Ty))
 118     return TI;  // Accept variable if it is already of bit type!
 119   if (auto *BitsTy = dyn_cast<BitsRecTy>(Ty))
 120     // Accept only bits<1> expression.
 121     return BitsTy->getNumBits() == 1 ? TI : nullptr;
 122   // Ternary !if can be converted to bit, but only if both sides are
 123   // convertible to a bit.
 124   if (TernOpInit *TOI = dyn_cast<TernOpInit>(TI)) {
 125     if (TOI->getOpcode() != TernOpInit::TernaryOp::IF)
 126       return nullptr;
 127     if (!TOI->getMHS()->convertInitializerTo(BitRecTy::get()) ||
 128         !TOI->getRHS()->convertInitializerTo(BitRecTy::get()))
 129       return nullptr;
 130     return TOI;
 131   }
 132   return nullptr;
 133 }
 134
 135 bool BitRecTy::baseClassOf(const RecTy *RHS) const{
 136   if(RecTy::baseClassOf(RHS) || RHS->getRecTyKind() == IntRecTyKind)
 137     return true;
 138   if(const BitsRecTy *BitsTy = dyn_cast<BitsRecTy>(RHS))
 139     return BitsTy->getNumBits() == 1;
 140   return false;
 141 }
 142
 143 BitsRecTy *BitsRecTy::get(unsigned Sz) {
 144   static std::vector<std::unique_ptr<BitsRecTy>> Shared;
 145   if (Sz >= Shared.size())
 146     Shared.resize(Sz + 1);
 147   std::unique_ptr<BitsRecTy> &Ty = Shared[Sz];
 148   if (!Ty)
 149     Ty.reset(new BitsRecTy(Sz));
 150   return Ty.get();
 151 }
 152
 153 std::string BitsRecTy::getAsString() const {
 154   return "bits<" + utostr(Size) + ">";
 155 }
 156
 157 Init *BitsRecTy::convertValue(UnsetInit *UI) {
 158   SmallVector<Init *, 16> NewBits(Size);
 159
 160   for (unsigned i = 0; i != Size; ++i)
 161     NewBits[i] = UnsetInit::get();
 162
 163   return BitsInit::get(NewBits);
 164 }
 165
 166 Init *BitsRecTy::convertValue(BitInit *BI) {
 167   if (Size != 1) return nullptr;  // Can only convert single bit.
 168   return BitsInit::get(BI);
 169 }
 170
 171 /// canFitInBitfield - Return true if the number of bits is large enough to hold
 172 /// the integer value.
 173 static bool canFitInBitfield(int64_t Value, unsigned NumBits) {
 174   // For example, with NumBits == 4, we permit Values from [-7 .. 15].
 175   return (NumBits >= sizeof(Value) * 8) ||
 176          (Value >> NumBits == 0) || (Value >> (NumBits-1) == -1);
 177 }
 178
 179 /// convertValue from Int initializer to bits type: Split the integer up into the
 180 /// appropriate bits.
 181 ///
 182 Init *BitsRecTy::convertValue(IntInit *II) {
 183   int64_t Value = II->getValue();
 184   // Make sure this bitfield is large enough to hold the integer value.
 185   if (!canFitInBitfield(Value, Size))
 186     return nullptr;
 187
 188   SmallVector<Init *, 16> NewBits(Size);
 189
 190   for (unsigned i = 0; i != Size; ++i)
 191     NewBits[i] = BitInit::get(Value & (1LL << i));
 192
 193   return BitsInit::get(NewBits);
 194 }
 195
 196 Init *BitsRecTy::convertValue(BitsInit *BI) {
 197   // If the number of bits is right, return it.  Otherwise we need to expand or
 198   // truncate.
 199   if (BI->getNumBits() == Size) return BI;
 200   return nullptr;
 201 }
 202
 203 Init *BitsRecTy::convertValue(TypedInit *TI) {
 204   if (Size == 1 && isa<BitRecTy>(TI->getType()))
 205     return BitsInit::get(TI);
 206
 207   if (TI->getType()->typeIsConvertibleTo(this)) {
 208     SmallVector<Init *, 16> NewBits(Size);
 209
 210     for (unsigned i = 0; i != Size; ++i)
 211       NewBits[i] = VarBitInit::get(TI, i);
 212     return BitsInit::get(NewBits);
 213   }
 214
 215   return nullptr;
 216 }
 217
 218 bool BitsRecTy::baseClassOf(const RecTy *RHS) const{
 219   if (RecTy::baseClassOf(RHS)) //argument and the receiver are the same type
 220     return cast<BitsRecTy>(RHS)->Size == Size;
 221   RecTyKind kind = RHS->getRecTyKind();
 222   return (kind == BitRecTyKind && Size == 1) || (kind == IntRecTyKind);
 223 }
 224
 225 Init *IntRecTy::convertValue(BitInit *BI) {
 226   return IntInit::get(BI->getValue());
 227 }
 228
 229 Init *IntRecTy::convertValue(BitsInit *BI) {
 230   int64_t Result = 0;
 231   for (unsigned i = 0, e = BI->getNumBits(); i != e; ++i)
 232     if (BitInit *Bit = dyn_cast<BitInit>(BI->getBit(i)))
 233       Result |= static_cast<int64_t>(Bit->getValue()) << i;
 234     else
 235       return nullptr;
 236   return IntInit::get(Result);
 237 }
 238
 239 Init *IntRecTy::convertValue(TypedInit *TI) {
 240   if (TI->getType()->typeIsConvertibleTo(this))
 241     return TI;  // Accept variable if already of the right type!
 242   return nullptr;
 243 }
 244
 245 bool IntRecTy::baseClassOf(const RecTy *RHS) const{
 246   RecTyKind kind = RHS->getRecTyKind();
 247   return kind==BitRecTyKind || kind==BitsRecTyKind || kind==IntRecTyKind;
 248 }
 249
 250 Init *StringRecTy::convertValue(UnOpInit *UO) {
 251   if (UO->getOpcode() == UnOpInit::CAST) {
 252     Init *L = UO->getOperand()->convertInitializerTo(this);
 253     if (!L) return nullptr;
 254     if (L != UO->getOperand())
 255       return UnOpInit::get(UnOpInit::CAST, L, StringRecTy::get());
 256     return UO;
 257   }
 258
 259   return convertValue((TypedInit*)UO);
 260 }
 261
 262 Init *StringRecTy::convertValue(BinOpInit *BO) {
 263   if (BO->getOpcode() == BinOpInit::STRCONCAT) {
 264     Init *L = BO->getLHS()->convertInitializerTo(this);
 265     Init *R = BO->getRHS()->convertInitializerTo(this);
 266     if (!L || !R) return nullptr;
 267     if (L != BO->getLHS() || R != BO->getRHS())
 268       return BinOpInit::get(BinOpInit::STRCONCAT, L, R, StringRecTy::get());
 269     return BO;
 270   }
 271
 272   return convertValue((TypedInit*)BO);
 273 }
 274
 275
 276 Init *StringRecTy::convertValue(TypedInit *TI) {
 277   if (isa<StringRecTy>(TI->getType()))
 278     return TI;  // Accept variable if already of the right type!
 279   return nullptr;
 280 }
 281
 282 std::string ListRecTy::getAsString() const {
 283   return "list<" + Ty->getAsString() + ">";
 284 }
 285
 286 Init *ListRecTy::convertValue(ListInit *LI) {
 287   std::vector<Init*> Elements;
 288
 289   // Verify that all of the elements of the list are subclasses of the
 290   // appropriate class!
 291   for (unsigned i = 0, e = LI->getSize(); i != e; ++i)
 292     if (Init *CI = LI->getElement(i)->convertInitializerTo(Ty))
 293       Elements.push_back(CI);
 294     else
 295       return nullptr;
 296
 297   if (!isa<ListRecTy>(LI->getType()))
 298     return nullptr;
 299
 300   return ListInit::get(Elements, this);
 301 }
 302
 303 Init *ListRecTy::convertValue(TypedInit *TI) {
 304   // Ensure that TI is compatible with our class.
 305   if (ListRecTy *LRT = dyn_cast<ListRecTy>(TI->getType()))
 306     if (LRT->getElementType()->typeIsConvertibleTo(getElementType()))
 307       return TI;
 308   return nullptr;
 309 }
 310
 311 bool ListRecTy::baseClassOf(const RecTy *RHS) const{
 312   if(const ListRecTy* ListTy = dyn_cast<ListRecTy>(RHS))
 313     return ListTy->getElementType()->typeIsConvertibleTo(Ty);
 314   return false;
 315 }
 316
 317 Init *DagRecTy::convertValue(TypedInit *TI) {
 318   if (TI->getType()->typeIsConvertibleTo(this))
 319     return TI;
 320   return nullptr;
 321 }
 322
 323 Init *DagRecTy::convertValue(UnOpInit *UO) {
 324   if (UO->getOpcode() == UnOpInit::CAST) {
 325     Init *L = UO->getOperand()->convertInitializerTo(this);
 326     if (!L) return nullptr;
 327     if (L != UO->getOperand())
 328       return UnOpInit::get(UnOpInit::CAST, L, DagRecTy::get());
 329     return UO;
 330   }
 331   return nullptr;
 332 }
 333
 334 Init *DagRecTy::convertValue(BinOpInit *BO) {
 335   if (BO->getOpcode() == BinOpInit::CONCAT) {
 336     Init *L = BO->getLHS()->convertInitializerTo(this);
 337     Init *R = BO->getRHS()->convertInitializerTo(this);
 338     if (!L || !R) return nullptr;
 339     if (L != BO->getLHS() || R != BO->getRHS())
 340       return BinOpInit::get(BinOpInit::CONCAT, L, R, DagRecTy::get());
 341     return BO;
 342   }
 343   return nullptr;
 344 }
 345
 346 RecordRecTy *RecordRecTy::get(Record *R) {
 347   return dyn_cast<RecordRecTy>(R->getDefInit()->getType());
 348 }
 349
 350 std::string RecordRecTy::getAsString() const {
 351   return Rec->getName();
 352 }
 353
 354 Init *RecordRecTy::convertValue(DefInit *DI) {
 355   // Ensure that DI is a subclass of Rec.
 356   if (!DI->getDef()->isSubClassOf(Rec))
 357     return nullptr;
 358   return DI;
 359 }
 360
 361 Init *RecordRecTy::convertValue(TypedInit *TI) {
 362   // Ensure that TI is compatible with Rec.
 363   if (RecordRecTy *RRT = dyn_cast<RecordRecTy>(TI->getType()))
 364     if (RRT->getRecord()->isSubClassOf(getRecord()) ||
 365         RRT->getRecord() == getRecord())
 366       return TI;
 367   return nullptr;
 368 }
 369
 370 bool RecordRecTy::baseClassOf(const RecTy *RHS) const{
 371   const RecordRecTy *RTy = dyn_cast<RecordRecTy>(RHS);
 372   if (!RTy)
 373     return false;
 374
 375   if (Rec == RTy->getRecord() || RTy->getRecord()->isSubClassOf(Rec))
 376     return true;
 377
 378   const std::vector<Record*> &SC = Rec->getSuperClasses();
 379   for (unsigned i = 0, e = SC.size(); i != e; ++i)
 380     if (RTy->getRecord()->isSubClassOf(SC[i]))
 381       return true;
 382
 383   return false;
 384 }
 385
 386 /// resolveTypes - Find a common type that T1 and T2 convert to.
 387 /// Return null if no such type exists.
 388 ///
 389 RecTy *llvm::resolveTypes(RecTy *T1, RecTy *T2) {
 390   if (T1->typeIsConvertibleTo(T2))
 391     return T2;
 392   if (T2->typeIsConvertibleTo(T1))
 393     return T1;
 394
 395   // If one is a Record type, check superclasses
 396   if (RecordRecTy *RecTy1 = dyn_cast<RecordRecTy>(T1)) {
 397     // See if T2 inherits from a type T1 also inherits from
 398     for (Record *SuperRec1 : RecTy1->getRecord()->getSuperClasses()) {
 399       RecordRecTy *SuperRecTy1 = RecordRecTy::get(SuperRec1);
 400       RecTy *NewType1 = resolveTypes(SuperRecTy1, T2);
 401       if (NewType1)
 402         return NewType1;
 403     }
 404   }
 405   if (RecordRecTy *RecTy2 = dyn_cast<RecordRecTy>(T2)) {
 406     // See if T1 inherits from a type T2 also inherits from
 407     for (Record *SuperRec2 : RecTy2->getRecord()->getSuperClasses()) {
 408       RecordRecTy *SuperRecTy2 = RecordRecTy::get(SuperRec2);
 409       RecTy *NewType2 = resolveTypes(T1, SuperRecTy2);
 410       if (NewType2)
 411         return NewType2;
 412     }
 413   }
 414   return nullptr;
 415 }
 416
 417
 418 //===----------------------------------------------------------------------===//
 419 //    Initializer implementations
 420 //===----------------------------------------------------------------------===//
 421
 422 void Init::anchor() { }
 423 void Init::dump() const { return print(errs()); }
 424
 425 void UnsetInit::anchor() { }
 426
 427 UnsetInit *UnsetInit::get() {
 428   static UnsetInit TheInit;
 429   return &TheInit;
 430 }
 431
 432 void BitInit::anchor() { }
 433
 434 BitInit *BitInit::get(bool V) {
 435   static BitInit True(true);
 436   static BitInit False(false);
 437
 438   return V ? &True : &False;
 439 }
 440
 441 static void
 442 ProfileBitsInit(FoldingSetNodeID &ID, ArrayRef<Init *> Range) {
 443   ID.AddInteger(Range.size());
 444
 445   for (Init *I : Range)
 446     ID.AddPointer(I);
 447 }
 448
 449 BitsInit *BitsInit::get(ArrayRef<Init *> Range) {
 450   static FoldingSet<BitsInit> ThePool;
 451   static std::vector<std::unique_ptr<BitsInit>> TheActualPool;
 452
 453   FoldingSetNodeID ID;
 454   ProfileBitsInit(ID, Range);
 455
 456   void *IP = nullptr;
 457   if (BitsInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
 458     return I;
 459
 460   BitsInit *I = new BitsInit(Range);
 461   ThePool.InsertNode(I, IP);
 462   TheActualPool.push_back(std::unique_ptr<BitsInit>(I));
 463   return I;
 464 }
 465
 466 void BitsInit::Profile(FoldingSetNodeID &ID) const {
 467   ProfileBitsInit(ID, Bits);
 468 }
 469
 470 Init *
 471 BitsInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
 472   SmallVector<Init *, 16> NewBits(Bits.size());
 473
 474   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 475     if (Bits[i] >= getNumBits())
 476       return nullptr;
 477     NewBits[i] = getBit(Bits[i]);
 478   }
 479   return BitsInit::get(NewBits);
 480 }
 481
 482 std::string BitsInit::getAsString() const {
 483   std::string Result = "{ ";
 484   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 485     if (i) Result += ", ";
 486     if (Init *Bit = getBit(e-i-1))
 487       Result += Bit->getAsString();
 488     else
 489       Result += "*";
 490   }
 491   return Result + " }";
 492 }
 493
 494 // Fix bit initializer to preserve the behavior that bit reference from a unset
 495 // bits initializer will resolve into VarBitInit to keep the field name and bit
 496 // number used in targets with fixed insn length.
 497 static Init *fixBitInit(const RecordVal *RV, Init *Before, Init *After) {
 498   if (RV || !isa<UnsetInit>(After))
 499     return After;
 500   return Before;
 501 }
 502
 503 // resolveReferences - If there are any field references that refer to fields
 504 // that have been filled in, we can propagate the values now.
 505 //
 506 Init *BitsInit::resolveReferences(Record &R, const RecordVal *RV) const {
 507   bool Changed = false;
 508   SmallVector<Init *, 16> NewBits(getNumBits());
 509
 510   Init *CachedInit = nullptr;
 511   Init *CachedBitVar = nullptr;
 512   bool CachedBitVarChanged = false;
 513
 514   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 515     Init *CurBit = Bits[i];
 516     Init *CurBitVar = CurBit->getBitVar();
 517
 518     NewBits[i] = CurBit;
 519
 520     if (CurBitVar == CachedBitVar) {
 521       if (CachedBitVarChanged) {
 522         Init *Bit = CachedInit->getBit(CurBit->getBitNum());
 523         NewBits[i] = fixBitInit(RV, CurBit, Bit);
 524       }
 525       continue;
 526     }
 527     CachedBitVar = CurBitVar;
 528     CachedBitVarChanged = false;
 529
 530     Init *B;
 531     do {
 532       B = CurBitVar;
 533       CurBitVar = CurBitVar->resolveReferences(R, RV);
 534       CachedBitVarChanged |= B != CurBitVar;
 535       Changed |= B != CurBitVar;
 536     } while (B != CurBitVar);
 537     CachedInit = CurBitVar;
 538
 539     if (CachedBitVarChanged) {
 540       Init *Bit = CurBitVar->getBit(CurBit->getBitNum());
 541       NewBits[i] = fixBitInit(RV, CurBit, Bit);
 542     }
 543   }
 544
 545   if (Changed)
 546     return BitsInit::get(NewBits);
 547
 548   return const_cast<BitsInit *>(this);
 549 }
 550
 551 IntInit *IntInit::get(int64_t V) {
 552   static DenseMap<int64_t, std::unique_ptr<IntInit>> ThePool;
 553
 554   std::unique_ptr<IntInit> &I = ThePool[V];
 555   if (!I) I.reset(new IntInit(V));
 556   return I.get();
 557 }
 558
 559 std::string IntInit::getAsString() const {
 560   return itostr(Value);
 561 }
 562
 563 Init *
 564 IntInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
 565   SmallVector<Init *, 16> NewBits(Bits.size());
 566
 567   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 568     if (Bits[i] >= 64)
 569       return nullptr;
 570
 571     NewBits[i] = BitInit::get(Value & (INT64_C(1) << Bits[i]));
 572   }
 573   return BitsInit::get(NewBits);
 574 }
 575
 576 void StringInit::anchor() { }
 577
 578 StringInit *StringInit::get(StringRef V) {
 579   static StringMap<std::unique_ptr<StringInit>> ThePool;
 580
 581   std::unique_ptr<StringInit> &I = ThePool[V];
 582   if (!I) I.reset(new StringInit(V));
 583   return I.get();
 584 }
 585
 586 static void ProfileListInit(FoldingSetNodeID &ID,
 587                             ArrayRef<Init *> Range,
 588                             RecTy *EltTy) {
 589   ID.AddInteger(Range.size());
 590   ID.AddPointer(EltTy);
 591
 592   for (Init *I : Range)
 593     ID.AddPointer(I);
 594 }
 595
 596 ListInit *ListInit::get(ArrayRef<Init *> Range, RecTy *EltTy) {
 597   static FoldingSet<ListInit> ThePool;
 598   static std::vector<std::unique_ptr<ListInit>> TheActualPool;
 599
 600   FoldingSetNodeID ID;
 601   ProfileListInit(ID, Range, EltTy);
 602
 603   void *IP = nullptr;
 604   if (ListInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
 605     return I;
 606
 607   ListInit *I = new ListInit(Range, EltTy);
 608   ThePool.InsertNode(I, IP);
 609   TheActualPool.push_back(std::unique_ptr<ListInit>(I));
 610   return I;
 611 }
 612
 613 void ListInit::Profile(FoldingSetNodeID &ID) const {
 614   RecTy *EltTy = cast<ListRecTy>(getType())->getElementType();
 615
 616   ProfileListInit(ID, Values, EltTy);
 617 }
 618
 619 Init *
 620 ListInit::convertInitListSlice(const std::vector<unsigned> &Elements) const {
 621   std::vector<Init*> Vals;
 622   for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
 623     if (Elements[i] >= getSize())
 624       return nullptr;
 625     Vals.push_back(getElement(Elements[i]));
 626   }
 627   return ListInit::get(Vals, getType());
 628 }
 629
 630 Record *ListInit::getElementAsRecord(unsigned i) const {
 631   assert(i < Values.size() && "List element index out of range!");
 632   DefInit *DI = dyn_cast<DefInit>(Values[i]);
 633   if (!DI)
 634     PrintFatalError("Expected record in list!");
 635   return DI->getDef();
 636 }
 637
 638 Init *ListInit::resolveReferences(Record &R, const RecordVal *RV) const {
 639   std::vector<Init*> Resolved;
 640   Resolved.reserve(getSize());
 641   bool Changed = false;
 642
 643   for (unsigned i = 0, e = getSize(); i != e; ++i) {
 644     Init *E;
 645     Init *CurElt = getElement(i);
 646
 647     do {
 648       E = CurElt;
 649       CurElt = CurElt->resolveReferences(R, RV);
 650       Changed |= E != CurElt;
 651     } while (E != CurElt);
 652     Resolved.push_back(E);
 653   }
 654
 655   if (Changed)
 656     return ListInit::get(Resolved, getType());
 657   return const_cast<ListInit *>(this);
 658 }
 659
 660 Init *ListInit::resolveListElementReference(Record &R, const RecordVal *IRV,
 661                                             unsigned Elt) const {
 662   if (Elt >= getSize())
 663     return nullptr;  // Out of range reference.
 664   Init *E = getElement(Elt);
 665   // If the element is set to some value, or if we are resolving a reference
 666   // to a specific variable and that variable is explicitly unset, then
 667   // replace the VarListElementInit with it.
 668   if (IRV || !isa<UnsetInit>(E))
 669     return E;
 670   return nullptr;
 671 }
 672
 673 std::string ListInit::getAsString() const {
 674   std::string Result = "[";
 675   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
 676     if (i) Result += ", ";
 677     Result += Values[i]->getAsString();
 678   }
 679   return Result + "]";
 680 }
 681
 682 Init *OpInit::resolveListElementReference(Record &R, const RecordVal *IRV,
 683                                           unsigned Elt) const {
 684   Init *Resolved = resolveReferences(R, IRV);
 685   OpInit *OResolved = dyn_cast<OpInit>(Resolved);
 686   if (OResolved) {
 687     Resolved = OResolved->Fold(&R, nullptr);
 688   }
 689
 690   if (Resolved != this) {
 691     TypedInit *Typed = cast<TypedInit>(Resolved);
 692     if (Init *New = Typed->resolveListElementReference(R, IRV, Elt))
 693       return New;
 694     return VarListElementInit::get(Typed, Elt);
 695   }
 696
 697   return nullptr;
 698 }
 699
 700 Init *OpInit::getBit(unsigned Bit) const {
 701   if (getType() == BitRecTy::get())
 702     return const_cast<OpInit*>(this);
 703   return VarBitInit::get(const_cast<OpInit*>(this), Bit);
 704 }
 705
 706 UnOpInit *UnOpInit::get(UnaryOp opc, Init *lhs, RecTy *Type) {
 707   typedef std::pair<std::pair<unsigned, Init *>, RecTy *> Key;
 708   static DenseMap<Key, std::unique_ptr<UnOpInit>> ThePool;
 709
 710   Key TheKey(std::make_pair(std::make_pair(opc, lhs), Type));
 711
 712   std::unique_ptr<UnOpInit> &I = ThePool[TheKey];
 713   if (!I) I.reset(new UnOpInit(opc, lhs, Type));
 714   return I.get();
 715 }
 716
 717 Init *UnOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
 718   switch (getOpcode()) {
 719   case CAST: {
 720     if (isa<StringRecTy>(getType())) {
 721       if (StringInit *LHSs = dyn_cast<StringInit>(LHS))
 722         return LHSs;
 723
 724       if (DefInit *LHSd = dyn_cast<DefInit>(LHS))
 725         return StringInit::get(LHSd->getDef()->getName());
 726
 727       if (IntInit *LHSi = dyn_cast<IntInit>(LHS))
 728         return StringInit::get(LHSi->getAsString());
 729     } else {
 730       if (StringInit *LHSs = dyn_cast<StringInit>(LHS)) {
 731         std::string Name = LHSs->getValue();
 732
 733         // From TGParser::ParseIDValue
 734         if (CurRec) {
 735           if (const RecordVal *RV = CurRec->getValue(Name)) {
 736             if (RV->getType() != getType())
 737               PrintFatalError("type mismatch in cast");
 738             return VarInit::get(Name, RV->getType());
 739           }
 740
 741           Init *TemplateArgName = QualifyName(*CurRec, CurMultiClass, Name,
 742                                               ":");
 743
 744           if (CurRec->isTemplateArg(TemplateArgName)) {
 745             const RecordVal *RV = CurRec->getValue(TemplateArgName);
 746             assert(RV && "Template arg doesn't exist??");
 747
 748             if (RV->getType() != getType())
 749               PrintFatalError("type mismatch in cast");
 750
 751             return VarInit::get(TemplateArgName, RV->getType());
 752           }
 753         }
 754
 755         if (CurMultiClass) {
 756           Init *MCName = QualifyName(CurMultiClass->Rec, CurMultiClass, Name,
 757                                      "::");
 758
 759           if (CurMultiClass->Rec.isTemplateArg(MCName)) {
 760             const RecordVal *RV = CurMultiClass->Rec.getValue(MCName);
 761             assert(RV && "Template arg doesn't exist??");
 762
 763             if (RV->getType() != getType())
 764               PrintFatalError("type mismatch in cast");
 765
 766             return VarInit::get(MCName, RV->getType());
 767           }
 768         }
 769         assert(CurRec && "NULL pointer");
 770         if (Record *D = (CurRec->getRecords()).getDef(Name))
 771           return DefInit::get(D);
 772
 773         PrintFatalError(CurRec->getLoc(),
 774                         "Undefined reference:'" + Name + "'\n");
 775       }
 776     }
 777     break;
 778   }
 779   case HEAD: {
 780     if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) {
 781       assert(!LHSl->empty() && "Empty list in head");
 782       return LHSl->getElement(0);
 783     }
 784     break;
 785   }
 786   case TAIL: {
 787     if (ListInit *LHSl = dyn_cast<ListInit>(LHS)) {
 788       assert(!LHSl->empty() && "Empty list in tail");
 789       // Note the +1.  We can't just pass the result of getValues()
 790       // directly.
 791       return ListInit::get(LHSl->getValues().slice(1), LHSl->getType());
 792     }
 793     break;
 794   }
 795   case EMPTY: {
 796     if (ListInit *LHSl = dyn_cast<ListInit>(LHS))
 797       return IntInit::get(LHSl->empty());
 798     if (StringInit *LHSs = dyn_cast<StringInit>(LHS))
 799       return IntInit::get(LHSs->getValue().empty());
 800
 801     break;
 802   }
 803   }
 804   return const_cast<UnOpInit *>(this);
 805 }
 806
 807 Init *UnOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
 808   Init *lhs = LHS->resolveReferences(R, RV);
 809
 810   if (LHS != lhs)
 811     return (UnOpInit::get(getOpcode(), lhs, getType()))->Fold(&R, nullptr);
 812   return Fold(&R, nullptr);
 813 }
 814
 815 std::string UnOpInit::getAsString() const {
 816   std::string Result;
 817   switch (Opc) {
 818   case CAST: Result = "!cast<" + getType()->getAsString() + ">"; break;
 819   case HEAD: Result = "!head"; break;
 820   case TAIL: Result = "!tail"; break;
 821   case EMPTY: Result = "!empty"; break;
 822   }
 823   return Result + "(" + LHS->getAsString() + ")";
 824 }
 825
 826 BinOpInit *BinOpInit::get(BinaryOp opc, Init *lhs,
 827                           Init *rhs, RecTy *Type) {
 828   typedef std::pair<
 829     std::pair<std::pair<unsigned, Init *>, Init *>,
 830     RecTy *
 831     > Key;
 832
 833   static DenseMap<Key, std::unique_ptr<BinOpInit>> ThePool;
 834
 835   Key TheKey(std::make_pair(std::make_pair(std::make_pair(opc, lhs), rhs),
 836                             Type));
 837
 838   std::unique_ptr<BinOpInit> &I = ThePool[TheKey];
 839   if (!I) I.reset(new BinOpInit(opc, lhs, rhs, Type));
 840   return I.get();
 841 }
 842
 843 Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
 844   switch (getOpcode()) {
 845   case CONCAT: {
 846     DagInit *LHSs = dyn_cast<DagInit>(LHS);
 847     DagInit *RHSs = dyn_cast<DagInit>(RHS);
 848     if (LHSs && RHSs) {
 849       DefInit *LOp = dyn_cast<DefInit>(LHSs->getOperator());
 850       DefInit *ROp = dyn_cast<DefInit>(RHSs->getOperator());
 851       if (!LOp || !ROp || LOp->getDef() != ROp->getDef())
 852         PrintFatalError("Concated Dag operators do not match!");
 853       std::vector<Init*> Args;
 854       std::vector<std::string> ArgNames;
 855       for (unsigned i = 0, e = LHSs->getNumArgs(); i != e; ++i) {
 856         Args.push_back(LHSs->getArg(i));
 857         ArgNames.push_back(LHSs->getArgName(i));
 858       }
 859       for (unsigned i = 0, e = RHSs->getNumArgs(); i != e; ++i) {
 860         Args.push_back(RHSs->getArg(i));
 861         ArgNames.push_back(RHSs->getArgName(i));
 862       }
 863       return DagInit::get(LHSs->getOperator(), "", Args, ArgNames);
 864     }
 865     break;
 866   }
 867   case LISTCONCAT: {
 868     ListInit *LHSs = dyn_cast<ListInit>(LHS);
 869     ListInit *RHSs = dyn_cast<ListInit>(RHS);
 870     if (LHSs && RHSs) {
 871       std::vector<Init *> Args;
 872       Args.insert(Args.end(), LHSs->begin(), LHSs->end());
 873       Args.insert(Args.end(), RHSs->begin(), RHSs->end());
 874       return ListInit::get(
 875           Args, cast<ListRecTy>(LHSs->getType())->getElementType());
 876     }
 877     break;
 878   }
 879   case STRCONCAT: {
 880     StringInit *LHSs = dyn_cast<StringInit>(LHS);
 881     StringInit *RHSs = dyn_cast<StringInit>(RHS);
 882     if (LHSs && RHSs)
 883       return StringInit::get(LHSs->getValue() + RHSs->getValue());
 884     break;
 885   }
 886   case EQ: {
 887     // try to fold eq comparison for 'bit' and 'int', otherwise fallback
 888     // to string objects.
 889     IntInit *L =
 890       dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get()));
 891     IntInit *R =
 892       dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get()));
 893
 894     if (L && R)
 895       return IntInit::get(L->getValue() == R->getValue());
 896
 897     StringInit *LHSs = dyn_cast<StringInit>(LHS);
 898     StringInit *RHSs = dyn_cast<StringInit>(RHS);
 899
 900     // Make sure we've resolved
 901     if (LHSs && RHSs)
 902       return IntInit::get(LHSs->getValue() == RHSs->getValue());
 903
 904     break;
 905   }
 906   case ADD:
 907   case AND:
 908   case SHL:
 909   case SRA:
 910   case SRL: {
 911     IntInit *LHSi =
 912       dyn_cast_or_null<IntInit>(LHS->convertInitializerTo(IntRecTy::get()));
 913     IntInit *RHSi =
 914       dyn_cast_or_null<IntInit>(RHS->convertInitializerTo(IntRecTy::get()));
 915     if (LHSi && RHSi) {
 916       int64_t LHSv = LHSi->getValue(), RHSv = RHSi->getValue();
 917       int64_t Result;
 918       switch (getOpcode()) {
 919       default: llvm_unreachable("Bad opcode!");
 920       case ADD: Result = LHSv +  RHSv; break;
 921       case AND: Result = LHSv &  RHSv; break;
 922       case SHL: Result = LHSv << RHSv; break;
 923       case SRA: Result = LHSv >> RHSv; break;
 924       case SRL: Result = (uint64_t)LHSv >> (uint64_t)RHSv; break;
 925       }
 926       return IntInit::get(Result);
 927     }
 928     break;
 929   }
 930   }
 931   return const_cast<BinOpInit *>(this);
 932 }
 933
 934 Init *BinOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
 935   Init *lhs = LHS->resolveReferences(R, RV);
 936   Init *rhs = RHS->resolveReferences(R, RV);
 937
 938   if (LHS != lhs || RHS != rhs)
 939     return (BinOpInit::get(getOpcode(), lhs, rhs, getType()))->Fold(&R,nullptr);
 940   return Fold(&R, nullptr);
 941 }
 942
 943 std::string BinOpInit::getAsString() const {
 944   std::string Result;
 945   switch (Opc) {
 946   case CONCAT: Result = "!con"; break;
 947   case ADD: Result = "!add"; break;
 948   case AND: Result = "!and"; break;
 949   case SHL: Result = "!shl"; break;
 950   case SRA: Result = "!sra"; break;
 951   case SRL: Result = "!srl"; break;
 952   case EQ: Result = "!eq"; break;
 953   case LISTCONCAT: Result = "!listconcat"; break;
 954   case STRCONCAT: Result = "!strconcat"; break;
 955   }
 956   return Result + "(" + LHS->getAsString() + ", " + RHS->getAsString() + ")";
 957 }
 958
 959 TernOpInit *TernOpInit::get(TernaryOp opc, Init *lhs, Init *mhs, Init *rhs,
 960                             RecTy *Type) {
 961   typedef std::pair<
 962     std::pair<
 963       std::pair<std::pair<unsigned, RecTy *>, Init *>,
 964       Init *
 965       >,
 966     Init *
 967     > Key;
 968
 969   static DenseMap<Key, std::unique_ptr<TernOpInit>> ThePool;
 970
 971   Key TheKey(std::make_pair(std::make_pair(std::make_pair(std::make_pair(opc,
 972                                                                          Type),
 973                                                           lhs),
 974                                            mhs),
 975                             rhs));
 976
 977   std::unique_ptr<TernOpInit> &I = ThePool[TheKey];
 978   if (!I) I.reset(new TernOpInit(opc, lhs, mhs, rhs, Type));
 979   return I.get();
 980 }
 981
 982 static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
 983                            Record *CurRec, MultiClass *CurMultiClass);
 984
 985 static Init *EvaluateOperation(OpInit *RHSo, Init *LHS, Init *Arg,
 986                                RecTy *Type, Record *CurRec,
 987                                MultiClass *CurMultiClass) {
 988   // If this is a dag, recurse
 989   if (auto *TArg = dyn_cast<TypedInit>(Arg))
 990     if (isa<DagRecTy>(TArg->getType()))
 991       return ForeachHelper(LHS, Arg, RHSo, Type, CurRec, CurMultiClass);
 992
 993   std::vector<Init *> NewOperands;
 994   for (int i = 0; i < RHSo->getNumOperands(); ++i) {
 995     if (auto *RHSoo = dyn_cast<OpInit>(RHSo->getOperand(i))) {
 996       if (Init *Result = EvaluateOperation(RHSoo, LHS, Arg,
 997                                            Type, CurRec, CurMultiClass))
 998         NewOperands.push_back(Result);
 999       else
1000         NewOperands.push_back(Arg);
1001     } else if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
1002       NewOperands.push_back(Arg);
1003     } else {
1004       NewOperands.push_back(RHSo->getOperand(i));
1005     }
1006   }
1007
1008   // Now run the operator and use its result as the new leaf
1009   const OpInit *NewOp = RHSo->clone(NewOperands);
1010   Init *NewVal = NewOp->Fold(CurRec, CurMultiClass);
1011   return (NewVal != NewOp) ? NewVal : nullptr;
1012 }
1013
1014 static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
1015                            Record *CurRec, MultiClass *CurMultiClass) {
1016   DagInit *MHSd = dyn_cast<DagInit>(MHS);
1017   ListInit *MHSl = dyn_cast<ListInit>(MHS);
1018
1019   OpInit *RHSo = dyn_cast<OpInit>(RHS);
1020
1021   if (!RHSo)
1022     PrintFatalError(CurRec->getLoc(), "!foreach requires an operator\n");
1023
1024   TypedInit *LHSt = dyn_cast<TypedInit>(LHS);
1025
1026   if (!LHSt)
1027     PrintFatalError(CurRec->getLoc(), "!foreach requires typed variable\n");
1028
1029   if ((MHSd && isa<DagRecTy>(Type)) || (MHSl && isa<ListRecTy>(Type))) {
1030     if (MHSd) {
1031       Init *Val = MHSd->getOperator();
1032       Init *Result = EvaluateOperation(RHSo, LHS, Val,
1033                                        Type, CurRec, CurMultiClass);
1034       if (Result)
1035         Val = Result;
1036
1037       std::vector<std::pair<Init *, std::string> > args;
1038       for (unsigned int i = 0; i < MHSd->getNumArgs(); ++i) {
1039         Init *Arg;
1040         std::string ArgName;
1041         Arg = MHSd->getArg(i);
1042         ArgName = MHSd->getArgName(i);
1043
1044         // Process args
1045         Init *Result = EvaluateOperation(RHSo, LHS, Arg, Type,
1046                                          CurRec, CurMultiClass);
1047         if (Result)
1048           Arg = Result;
1049
1050         // TODO: Process arg names
1051         args.push_back(std::make_pair(Arg, ArgName));
1052       }
1053
1054       return DagInit::get(Val, "", args);
1055     }
1056     if (MHSl) {
1057       std::vector<Init *> NewOperands;
1058       std::vector<Init *> NewList(MHSl->begin(), MHSl->end());
1059
1060       for (Init *&Item : NewList) {
1061         NewOperands.clear();
1062         for(int i = 0; i < RHSo->getNumOperands(); ++i) {
1063           // First, replace the foreach variable with the list item
1064           if (LHS->getAsString() == RHSo->getOperand(i)->getAsString())
1065             NewOperands.push_back(Item);
1066           else
1067             NewOperands.push_back(RHSo->getOperand(i));
1068         }
1069
1070         // Now run the operator and use its result as the new list item
1071         const OpInit *NewOp = RHSo->clone(NewOperands);
1072         Init *NewItem = NewOp->Fold(CurRec, CurMultiClass);
1073         if (NewItem != NewOp)
1074           Item = NewItem;
1075       }
1076       return ListInit::get(NewList, MHSl->getType());
1077     }
1078   }
1079   return nullptr;
1080 }
1081
1082 Init *TernOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
1083   switch (getOpcode()) {
1084   case SUBST: {
1085     DefInit *LHSd = dyn_cast<DefInit>(LHS);
1086     VarInit *LHSv = dyn_cast<VarInit>(LHS);
1087     StringInit *LHSs = dyn_cast<StringInit>(LHS);
1088
1089     DefInit *MHSd = dyn_cast<DefInit>(MHS);
1090     VarInit *MHSv = dyn_cast<VarInit>(MHS);
1091     StringInit *MHSs = dyn_cast<StringInit>(MHS);
1092
1093     DefInit *RHSd = dyn_cast<DefInit>(RHS);
1094     VarInit *RHSv = dyn_cast<VarInit>(RHS);
1095     StringInit *RHSs = dyn_cast<StringInit>(RHS);
1096
1097     if (LHSd && MHSd && RHSd) {
1098       Record *Val = RHSd->getDef();
1099       if (LHSd->getAsString() == RHSd->getAsString())
1100         Val = MHSd->getDef();
1101       return DefInit::get(Val);
1102     }
1103     if (LHSv && MHSv && RHSv) {
1104       std::string Val = RHSv->getName();
1105       if (LHSv->getAsString() == RHSv->getAsString())
1106         Val = MHSv->getName();
1107       return VarInit::get(Val, getType());
1108     }
1109     if (LHSs && MHSs && RHSs) {
1110       std::string Val = RHSs->getValue();
1111
1112       std::string::size_type found;
1113       std::string::size_type idx = 0;
1114       while (true) {
1115         found = Val.find(LHSs->getValue(), idx);
1116         if (found == std::string::npos)
1117           break;
1118         Val.replace(found, LHSs->getValue().size(), MHSs->getValue());
1119         idx = found + MHSs->getValue().size();
1120       }
1121
1122       return StringInit::get(Val);
1123     }
1124     break;
1125   }
1126
1127   case FOREACH: {
1128     if (Init *Result = ForeachHelper(LHS, MHS, RHS, getType(),
1129                                      CurRec, CurMultiClass))
1130       return Result;
1131     break;
1132   }
1133
1134   case IF: {
1135     IntInit *LHSi = dyn_cast<IntInit>(LHS);
1136     if (Init *I = LHS->convertInitializerTo(IntRecTy::get()))
1137       LHSi = dyn_cast<IntInit>(I);
1138     if (LHSi) {
1139       if (LHSi->getValue())
1140         return MHS;
1141       return RHS;
1142     }
1143     break;
1144   }
1145   }
1146
1147   return const_cast<TernOpInit *>(this);
1148 }
1149
1150 Init *TernOpInit::resolveReferences(Record &R,
1151                                     const RecordVal *RV) const {
1152   Init *lhs = LHS->resolveReferences(R, RV);
1153
1154   if (Opc == IF && lhs != LHS) {
1155     IntInit *Value = dyn_cast<IntInit>(lhs);
1156     if (Init *I = lhs->convertInitializerTo(IntRecTy::get()))
1157       Value = dyn_cast<IntInit>(I);
1158     if (Value) {
1159       // Short-circuit
1160       if (Value->getValue()) {
1161         Init *mhs = MHS->resolveReferences(R, RV);
1162         return (TernOpInit::get(getOpcode(), lhs, mhs,
1163                                 RHS, getType()))->Fold(&R, nullptr);
1164       }
1165       Init *rhs = RHS->resolveReferences(R, RV);
1166       return (TernOpInit::get(getOpcode(), lhs, MHS,
1167                               rhs, getType()))->Fold(&R, nullptr);
1168     }
1169   }
1170
1171   Init *mhs = MHS->resolveReferences(R, RV);
1172   Init *rhs = RHS->resolveReferences(R, RV);
1173
1174   if (LHS != lhs || MHS != mhs || RHS != rhs)
1175     return (TernOpInit::get(getOpcode(), lhs, mhs, rhs,
1176                             getType()))->Fold(&R, nullptr);
1177   return Fold(&R, nullptr);
1178 }
1179
1180 std::string TernOpInit::getAsString() const {
1181   std::string Result;
1182   switch (Opc) {
1183   case SUBST: Result = "!subst"; break;
1184   case FOREACH: Result = "!foreach"; break;
1185   case IF: Result = "!if"; break;
1186   }
1187   return Result + "(" + LHS->getAsString() + ", " + MHS->getAsString() + ", " +
1188          RHS->getAsString() + ")";
1189 }
1190
1191 RecTy *TypedInit::getFieldType(const std::string &FieldName) const {
1192   if (RecordRecTy *RecordType = dyn_cast<RecordRecTy>(getType()))
1193     if (RecordVal *Field = RecordType->getRecord()->getValue(FieldName))
1194       return Field->getType();
1195   return nullptr;
1196 }
1197
1198 Init *
1199 TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
1200   BitsRecTy *T = dyn_cast<BitsRecTy>(getType());
1201   if (!T) return nullptr;  // Cannot subscript a non-bits variable.
1202   unsigned NumBits = T->getNumBits();
1203
1204   SmallVector<Init *, 16> NewBits(Bits.size());
1205   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
1206     if (Bits[i] >= NumBits)
1207       return nullptr;
1208
1209     NewBits[i] = VarBitInit::get(const_cast<TypedInit *>(this), Bits[i]);
1210   }
1211   return BitsInit::get(NewBits);
1212 }
1213
1214 Init *
1215 TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) const {
1216   ListRecTy *T = dyn_cast<ListRecTy>(getType());
1217   if (!T) return nullptr;  // Cannot subscript a non-list variable.
1218
1219   if (Elements.size() == 1)
1220     return VarListElementInit::get(const_cast<TypedInit *>(this), Elements[0]);
1221
1222   std::vector<Init*> ListInits;
1223   ListInits.reserve(Elements.size());
1224   for (unsigned i = 0, e = Elements.size(); i != e; ++i)
1225     ListInits.push_back(VarListElementInit::get(const_cast<TypedInit *>(this),
1226                                                 Elements[i]));
1227   return ListInit::get(ListInits, T);
1228 }
1229
1230
1231 VarInit *VarInit::get(const std::string &VN, RecTy *T) {
1232   Init *Value = StringInit::get(VN);
1233   return VarInit::get(Value, T);
1234 }
1235
1236 VarInit *VarInit::get(Init *VN, RecTy *T) {
1237   typedef std::pair<RecTy *, Init *> Key;
1238   static DenseMap<Key, std::unique_ptr<VarInit>> ThePool;
1239
1240   Key TheKey(std::make_pair(T, VN));
1241
1242   std::unique_ptr<VarInit> &I = ThePool[TheKey];
1243   if (!I) I.reset(new VarInit(VN, T));
1244   return I.get();
1245 }
1246
1247 const std::string &VarInit::getName() const {
1248   StringInit *NameString = cast<StringInit>(getNameInit());
1249   return NameString->getValue();
1250 }
1251
1252 Init *VarInit::getBit(unsigned Bit) const {
1253   if (getType() == BitRecTy::get())
1254     return const_cast<VarInit*>(this);
1255   return VarBitInit::get(const_cast<VarInit*>(this), Bit);
1256 }
1257
1258 Init *VarInit::resolveListElementReference(Record &R,
1259                                            const RecordVal *IRV,
1260                                            unsigned Elt) const {
1261   if (R.isTemplateArg(getNameInit())) return nullptr;
1262   if (IRV && IRV->getNameInit() != getNameInit()) return nullptr;
1263
1264   RecordVal *RV = R.getValue(getNameInit());
1265   assert(RV && "Reference to a non-existent variable?");
1266   ListInit *LI = dyn_cast<ListInit>(RV->getValue());
1267   if (!LI)
1268     return VarListElementInit::get(cast<TypedInit>(RV->getValue()), Elt);
1269
1270   if (Elt >= LI->getSize())
1271     return nullptr;  // Out of range reference.
1272   Init *E = LI->getElement(Elt);
1273   // If the element is set to some value, or if we are resolving a reference
1274   // to a specific variable and that variable is explicitly unset, then
1275   // replace the VarListElementInit with it.
1276   if (IRV || !isa<UnsetInit>(E))
1277     return E;
1278   return nullptr;
1279 }
1280
1281
1282 RecTy *VarInit::getFieldType(const std::string &FieldName) const {
1283   if (RecordRecTy *RTy = dyn_cast<RecordRecTy>(getType()))
1284     if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName))
1285       return RV->getType();
1286   return nullptr;
1287 }
1288
1289 Init *VarInit::getFieldInit(Record &R, const RecordVal *RV,
1290                             const std::string &FieldName) const {
1291   if (isa<RecordRecTy>(getType()))
1292     if (const RecordVal *Val = R.getValue(VarName)) {
1293       if (RV != Val && (RV || isa<UnsetInit>(Val->getValue())))
1294         return nullptr;
1295       Init *TheInit = Val->getValue();
1296       assert(TheInit != this && "Infinite loop detected!");
1297       if (Init *I = TheInit->getFieldInit(R, RV, FieldName))
1298         return I;
1299       return nullptr;
1300     }
1301   return nullptr;
1302 }
1303
1304 /// resolveReferences - This method is used by classes that refer to other
1305 /// variables which may not be defined at the time the expression is formed.
1306 /// If a value is set for the variable later, this method will be called on
1307 /// users of the value to allow the value to propagate out.
1308 ///
1309 Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) const {
1310   if (RecordVal *Val = R.getValue(VarName))
1311     if (RV == Val || (!RV && !isa<UnsetInit>(Val->getValue())))
1312       return Val->getValue();
1313   return const_cast<VarInit *>(this);
1314 }
1315
1316 VarBitInit *VarBitInit::get(TypedInit *T, unsigned B) {
1317   typedef std::pair<TypedInit *, unsigned> Key;
1318   static DenseMap<Key, std::unique_ptr<VarBitInit>> ThePool;
1319
1320   Key TheKey(std::make_pair(T, B));
1321
1322   std::unique_ptr<VarBitInit> &I = ThePool[TheKey];
1323   if (!I) I.reset(new VarBitInit(T, B));
1324   return I.get();
1325 }
1326
1327 std::string VarBitInit::getAsString() const {
1328   return TI->getAsString() + "{" + utostr(Bit) + "}";
1329 }
1330
1331 Init *VarBitInit::resolveReferences(Record &R, const RecordVal *RV) const {
1332   Init *I = TI->resolveReferences(R, RV);
1333   if (TI != I)
1334     return I->getBit(getBitNum());
1335
1336   return const_cast<VarBitInit*>(this);
1337 }
1338
1339 VarListElementInit *VarListElementInit::get(TypedInit *T,
1340                                             unsigned E) {
1341   typedef std::pair<TypedInit *, unsigned> Key;
1342   static DenseMap<Key, std::unique_ptr<VarListElementInit>> ThePool;
1343
1344   Key TheKey(std::make_pair(T, E));
1345
1346   std::unique_ptr<VarListElementInit> &I = ThePool[TheKey];
1347   if (!I) I.reset(new VarListElementInit(T, E));
1348   return I.get();
1349 }
1350
1351 std::string VarListElementInit::getAsString() const {
1352   return TI->getAsString() + "[" + utostr(Element) + "]";
1353 }
1354
1355 Init *
1356 VarListElementInit::resolveReferences(Record &R, const RecordVal *RV) const {
1357   if (Init *I = getVariable()->resolveListElementReference(R, RV,
1358                                                            getElementNum()))
1359     return I;
1360   return const_cast<VarListElementInit *>(this);
1361 }
1362
1363 Init *VarListElementInit::getBit(unsigned Bit) const {
1364   if (getType() == BitRecTy::get())
1365     return const_cast<VarListElementInit*>(this);
1366   return VarBitInit::get(const_cast<VarListElementInit*>(this), Bit);
1367 }
1368
1369 Init *VarListElementInit:: resolveListElementReference(Record &R,
1370                                                        const RecordVal *RV,
1371                                                        unsigned Elt) const {
1372   if (Init *Result = TI->resolveListElementReference(R, RV, Element)) {
1373     if (TypedInit *TInit = dyn_cast<TypedInit>(Result)) {
1374       Init *Result2 = TInit->resolveListElementReference(R, RV, Elt);
1375       if (Result2) return Result2;
1376       return VarListElementInit::get(TInit, Elt);
1377     }
1378     return Result;
1379   }
1380
1381   return nullptr;
1382 }
1383
1384 DefInit *DefInit::get(Record *R) {
1385   return R->getDefInit();
1386 }
1387
1388 RecTy *DefInit::getFieldType(const std::string &FieldName) const {
1389   if (const RecordVal *RV = Def->getValue(FieldName))
1390     return RV->getType();
1391   return nullptr;
1392 }
1393
1394 Init *DefInit::getFieldInit(Record &R, const RecordVal *RV,
1395                             const std::string &FieldName) const {
1396   return Def->getValue(FieldName)->getValue();
1397 }
1398
1399
1400 std::string DefInit::getAsString() const {
1401   return Def->getName();
1402 }
1403
1404 FieldInit *FieldInit::get(Init *R, const std::string &FN) {
1405   typedef std::pair<Init *, TableGenStringKey> Key;
1406   static DenseMap<Key, std::unique_ptr<FieldInit>> ThePool;
1407
1408   Key TheKey(std::make_pair(R, FN));
1409
1410   std::unique_ptr<FieldInit> &I = ThePool[TheKey];
1411   if (!I) I.reset(new FieldInit(R, FN));
1412   return I.get();
1413 }
1414
1415 Init *FieldInit::getBit(unsigned Bit) const {
1416   if (getType() == BitRecTy::get())
1417     return const_cast<FieldInit*>(this);
1418   return VarBitInit::get(const_cast<FieldInit*>(this), Bit);
1419 }
1420
1421 Init *FieldInit::resolveListElementReference(Record &R, const RecordVal *RV,
1422                                              unsigned Elt) const {
1423   if (Init *ListVal = Rec->getFieldInit(R, RV, FieldName))
1424     if (ListInit *LI = dyn_cast<ListInit>(ListVal)) {
1425       if (Elt >= LI->getSize()) return nullptr;
1426       Init *E = LI->getElement(Elt);
1427
1428       // If the element is set to some value, or if we are resolving a
1429       // reference to a specific variable and that variable is explicitly
1430       // unset, then replace the VarListElementInit with it.
1431       if (RV || !isa<UnsetInit>(E))
1432         return E;
1433     }
1434   return nullptr;
1435 }
1436
1437 Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) const {
1438   Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec;
1439
1440   if (Init *BitsVal = NewRec->getFieldInit(R, RV, FieldName)) {
1441     Init *BVR = BitsVal->resolveReferences(R, RV);
1442     return BVR->isComplete() ? BVR : const_cast<FieldInit *>(this);
1443   }
1444
1445   if (NewRec != Rec)
1446     return FieldInit::get(NewRec, FieldName);
1447   return const_cast<FieldInit *>(this);
1448 }
1449
1450 static void ProfileDagInit(FoldingSetNodeID &ID, Init *V, const std::string &VN,
1451                            ArrayRef<Init *> ArgRange,
1452                            ArrayRef<std::string> NameRange) {
1453   ID.AddPointer(V);
1454   ID.AddString(VN);
1455
1456   ArrayRef<Init *>::iterator Arg  = ArgRange.begin();
1457   ArrayRef<std::string>::iterator  Name = NameRange.begin();
1458   while (Arg != ArgRange.end()) {
1459     assert(Name != NameRange.end() && "Arg name underflow!");
1460     ID.AddPointer(*Arg++);
1461     ID.AddString(*Name++);
1462   }
1463   assert(Name == NameRange.end() && "Arg name overflow!");
1464 }
1465
1466 DagInit *
1467 DagInit::get(Init *V, const std::string &VN,
1468              ArrayRef<Init *> ArgRange,
1469              ArrayRef<std::string> NameRange) {
1470   static FoldingSet<DagInit> ThePool;
1471   static std::vector<std::unique_ptr<DagInit>> TheActualPool;
1472
1473   FoldingSetNodeID ID;
1474   ProfileDagInit(ID, V, VN, ArgRange, NameRange);
1475
1476   void *IP = nullptr;
1477   if (DagInit *I = ThePool.FindNodeOrInsertPos(ID, IP))
1478     return I;
1479
1480   DagInit *I = new DagInit(V, VN, ArgRange, NameRange);
1481   ThePool.InsertNode(I, IP);
1482   TheActualPool.push_back(std::unique_ptr<DagInit>(I));
1483   return I;
1484 }
1485
1486 DagInit *
1487 DagInit::get(Init *V, const std::string &VN,
1488              const std::vector<std::pair<Init*, std::string> > &args) {
1489   std::vector<Init *> Args;
1490   std::vector<std::string> Names;
1491
1492   for (const auto &Arg : args) {
1493     Args.push_back(Arg.first);
1494     Names.push_back(Arg.second);
1495   }
1496
1497   return DagInit::get(V, VN, Args, Names);
1498 }
1499
1500 void DagInit::Profile(FoldingSetNodeID &ID) const {
1501   ProfileDagInit(ID, Val, ValName, Args, ArgNames);
1502 }
1503
1504 Init *DagInit::resolveReferences(Record &R, const RecordVal *RV) const {
1505   std::vector<Init*> NewArgs;
1506   for (unsigned i = 0, e = Args.size(); i != e; ++i)
1507     NewArgs.push_back(Args[i]->resolveReferences(R, RV));
1508
1509   Init *Op = Val->resolveReferences(R, RV);
1510
1511   if (Args != NewArgs || Op != Val)
1512     return DagInit::get(Op, ValName, NewArgs, ArgNames);
1513
1514   return const_cast<DagInit *>(this);
1515 }
1516
1517
1518 std::string DagInit::getAsString() const {
1519   std::string Result = "(" + Val->getAsString();
1520   if (!ValName.empty())
1521     Result += ":" + ValName;
1522   if (!Args.empty()) {
1523     Result += " " + Args[0]->getAsString();
1524     if (!ArgNames[0].empty()) Result += ":$" + ArgNames[0];
1525     for (unsigned i = 1, e = Args.size(); i != e; ++i) {
1526       Result += ", " + Args[i]->getAsString();
1527       if (!ArgNames[i].empty()) Result += ":$" + ArgNames[i];
1528     }
1529   }
1530   return Result + ")";
1531 }
1532
1533
1534 //===----------------------------------------------------------------------===//
1535 //    Other implementations
1536 //===----------------------------------------------------------------------===//
1537
1538 RecordVal::RecordVal(Init *N, RecTy *T, unsigned P)
1539   : Name(N), Ty(T), Prefix(P) {
1540   Value = Ty->convertValue(UnsetInit::get());
1541   assert(Value && "Cannot create unset value for current type!");
1542 }
1543
1544 RecordVal::RecordVal(const std::string &N, RecTy *T, unsigned P)
1545   : Name(StringInit::get(N)), Ty(T), Prefix(P) {
1546   Value = Ty->convertValue(UnsetInit::get());
1547   assert(Value && "Cannot create unset value for current type!");
1548 }
1549
1550 const std::string &RecordVal::getName() const {
1551   return cast<StringInit>(Name)->getValue();
1552 }
1553
1554 void RecordVal::dump() const { errs() << *this; }
1555
1556 void RecordVal::print(raw_ostream &OS, bool PrintSem) const {
1557   if (getPrefix()) OS << "field ";
1558   OS << *getType() << " " << getNameInitAsString();
1559
1560   if (getValue())
1561     OS << " = " << *getValue();
1562
1563   if (PrintSem) OS << ";\n";
1564 }
1565
1566 unsigned Record::LastID = 0;
1567
1568 void Record::init() {
1569   checkName();
1570
1571   // Every record potentially has a def at the top.  This value is
1572   // replaced with the top-level def name at instantiation time.
1573   RecordVal DN("NAME", StringRecTy::get(), 0);
1574   addValue(DN);
1575 }
1576
1577 void Record::checkName() {
1578   // Ensure the record name has string type.
1579   const TypedInit *TypedName = cast<const TypedInit>(Name);
1580   RecTy *Type = TypedName->getType();
1581   if (!isa<StringRecTy>(Type))
1582     PrintFatalError(getLoc(), "Record name is not a string!");
1583 }
1584
1585 DefInit *Record::getDefInit() {
1586   static DenseMap<Record *, std::unique_ptr<DefInit>> ThePool;
1587   if (TheInit)
1588     return TheInit;
1589
1590   std::unique_ptr<DefInit> &I = ThePool[this];
1591   if (!I) I.reset(new DefInit(this, new RecordRecTy(this)));
1592   return I.get();
1593 }
1594
1595 const std::string &Record::getName() const {
1596   return cast<StringInit>(Name)->getValue();
1597 }
1598
1599 void Record::setName(Init *NewName) {
1600   Name = NewName;
1601   checkName();
1602   // DO NOT resolve record values to the name at this point because
1603   // there might be default values for arguments of this def.  Those
1604   // arguments might not have been resolved yet so we don't want to
1605   // prematurely assume values for those arguments were not passed to
1606   // this def.
1607   //
1608   // Nonetheless, it may be that some of this Record's values
1609   // reference the record name.  Indeed, the reason for having the
1610   // record name be an Init is to provide this flexibility.  The extra
1611   // resolve steps after completely instantiating defs takes care of
1612   // this.  See TGParser::ParseDef and TGParser::ParseDefm.
1613 }
1614
1615 void Record::setName(const std::string &Name) {
1616   setName(StringInit::get(Name));
1617 }
1618
1619 /// resolveReferencesTo - If anything in this record refers to RV, replace the
1620 /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
1621 /// references.
1622 void Record::resolveReferencesTo(const RecordVal *RV) {
1623   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
1624     if (RV == &Values[i]) // Skip resolve the same field as the given one
1625       continue;
1626     if (Init *V = Values[i].getValue())
1627       if (Values[i].setValue(V->resolveReferences(*this, RV)))
1628         PrintFatalError(getLoc(), "Invalid value is found when setting '" +
1629                         Values[i].getNameInitAsString() +
1630                         "' after resolving references" +
1631                         (RV ? " against '" + RV->getNameInitAsString() +
1632                               "' of (" + RV->getValue()->getAsUnquotedString() +
1633                               ")"
1634                             : "") + "\n");
1635   }
1636   Init *OldName = getNameInit();
1637   Init *NewName = Name->resolveReferences(*this, RV);
1638   if (NewName != OldName) {
1639     // Re-register with RecordKeeper.
1640     setName(NewName);
1641   }
1642 }
1643
1644 void Record::dump() const { errs() << *this; }
1645
1646 raw_ostream &llvm::operator<<(raw_ostream &OS, const Record &R) {
1647   OS << R.getNameInitAsString();
1648
1649   const std::vector<Init *> &TArgs = R.getTemplateArgs();
1650   if (!TArgs.empty()) {
1651     OS << "<";
1652     for (unsigned i = 0, e = TArgs.size(); i != e; ++i) {
1653       if (i) OS << ", ";
1654       const RecordVal *RV = R.getValue(TArgs[i]);
1655       assert(RV && "Template argument record not found??");
1656       RV->print(OS, false);
1657     }
1658     OS << ">";
1659   }
1660
1661   OS << " {";
1662   const std::vector<Record*> &SC = R.getSuperClasses();
1663   if (!SC.empty()) {
1664     OS << "\t//";
1665     for (unsigned i = 0, e = SC.size(); i != e; ++i)
1666       OS << " " << SC[i]->getNameInitAsString();
1667   }
1668   OS << "\n";
1669
1670   const std::vector<RecordVal> &Vals = R.getValues();
1671   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1672     if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1673       OS << Vals[i];
1674   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1675     if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1676       OS << Vals[i];
1677
1678   return OS << "}\n";
1679 }
1680
1681 /// getValueInit - Return the initializer for a value with the specified name,
1682 /// or abort if the field does not exist.
1683 ///
1684 Init *Record::getValueInit(StringRef FieldName) const {
1685   const RecordVal *R = getValue(FieldName);
1686   if (!R || !R->getValue())
1687     PrintFatalError(getLoc(), "Record `" + getName() +
1688       "' does not have a field named `" + FieldName + "'!\n");
1689   return R->getValue();
1690 }
1691
1692
1693 /// getValueAsString - This method looks up the specified field and returns its
1694 /// value as a string, aborts if the field does not exist or if
1695 /// the value is not a string.
1696 ///
1697 std::string Record::getValueAsString(StringRef FieldName) const {
1698   const RecordVal *R = getValue(FieldName);
1699   if (!R || !R->getValue())
1700     PrintFatalError(getLoc(), "Record `" + getName() +
1701       "' does not have a field named `" + FieldName + "'!\n");
1702
1703   if (StringInit *SI = dyn_cast<StringInit>(R->getValue()))
1704     return SI->getValue();
1705   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1706     FieldName + "' does not have a string initializer!");
1707 }
1708
1709 /// getValueAsBitsInit - This method looks up the specified field and returns
1710 /// its value as a BitsInit, aborts if the field does not exist or if
1711 /// the value is not the right type.
1712 ///
1713 BitsInit *Record::getValueAsBitsInit(StringRef FieldName) const {
1714   const RecordVal *R = getValue(FieldName);
1715   if (!R || !R->getValue())
1716     PrintFatalError(getLoc(), "Record `" + getName() +
1717       "' does not have a field named `" + FieldName + "'!\n");
1718
1719   if (BitsInit *BI = dyn_cast<BitsInit>(R->getValue()))
1720     return BI;
1721   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1722     FieldName + "' does not have a BitsInit initializer!");
1723 }
1724
1725 /// getValueAsListInit - This method looks up the specified field and returns
1726 /// its value as a ListInit, aborting if the field does not exist or if
1727 /// the value is not the right type.
1728 ///
1729 ListInit *Record::getValueAsListInit(StringRef FieldName) const {
1730   const RecordVal *R = getValue(FieldName);
1731   if (!R || !R->getValue())
1732     PrintFatalError(getLoc(), "Record `" + getName() +
1733       "' does not have a field named `" + FieldName + "'!\n");
1734
1735   if (ListInit *LI = dyn_cast<ListInit>(R->getValue()))
1736     return LI;
1737   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1738     FieldName + "' does not have a list initializer!");
1739 }
1740
1741 /// getValueAsListOfDefs - This method looks up the specified field and returns
1742 /// its value as a vector of records, aborting if the field does not exist
1743 /// or if the value is not the right type.
1744 ///
1745 std::vector<Record*>
1746 Record::getValueAsListOfDefs(StringRef FieldName) const {
1747   ListInit *List = getValueAsListInit(FieldName);
1748   std::vector<Record*> Defs;
1749   for (unsigned i = 0; i < List->getSize(); i++) {
1750     if (DefInit *DI = dyn_cast<DefInit>(List->getElement(i)))
1751       Defs.push_back(DI->getDef());
1752     else
1753       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1754         FieldName + "' list is not entirely DefInit!");
1755   }
1756   return Defs;
1757 }
1758
1759 /// getValueAsInt - This method looks up the specified field and returns its
1760 /// value as an int64_t, aborting if the field does not exist or if the value
1761 /// is not the right type.
1762 ///
1763 int64_t Record::getValueAsInt(StringRef FieldName) const {
1764   const RecordVal *R = getValue(FieldName);
1765   if (!R || !R->getValue())
1766     PrintFatalError(getLoc(), "Record `" + getName() +
1767       "' does not have a field named `" + FieldName + "'!\n");
1768
1769   if (IntInit *II = dyn_cast<IntInit>(R->getValue()))
1770     return II->getValue();
1771   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1772     FieldName + "' does not have an int initializer!");
1773 }
1774
1775 /// getValueAsListOfInts - This method looks up the specified field and returns
1776 /// its value as a vector of integers, aborting if the field does not exist or
1777 /// if the value is not the right type.
1778 ///
1779 std::vector<int64_t>
1780 Record::getValueAsListOfInts(StringRef FieldName) const {
1781   ListInit *List = getValueAsListInit(FieldName);
1782   std::vector<int64_t> Ints;
1783   for (unsigned i = 0; i < List->getSize(); i++) {
1784     if (IntInit *II = dyn_cast<IntInit>(List->getElement(i)))
1785       Ints.push_back(II->getValue());
1786     else
1787       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1788         FieldName + "' does not have a list of ints initializer!");
1789   }
1790   return Ints;
1791 }
1792
1793 /// getValueAsListOfStrings - This method looks up the specified field and
1794 /// returns its value as a vector of strings, aborting if the field does not
1795 /// exist or if the value is not the right type.
1796 ///
1797 std::vector<std::string>
1798 Record::getValueAsListOfStrings(StringRef FieldName) const {
1799   ListInit *List = getValueAsListInit(FieldName);
1800   std::vector<std::string> Strings;
1801   for (unsigned i = 0; i < List->getSize(); i++) {
1802     if (StringInit *II = dyn_cast<StringInit>(List->getElement(i)))
1803       Strings.push_back(II->getValue());
1804     else
1805       PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1806         FieldName + "' does not have a list of strings initializer!");
1807   }
1808   return Strings;
1809 }
1810
1811 /// getValueAsDef - This method looks up the specified field and returns its
1812 /// value as a Record, aborting if the field does not exist or if the value
1813 /// is not the right type.
1814 ///
1815 Record *Record::getValueAsDef(StringRef FieldName) const {
1816   const RecordVal *R = getValue(FieldName);
1817   if (!R || !R->getValue())
1818     PrintFatalError(getLoc(), "Record `" + getName() +
1819       "' does not have a field named `" + FieldName + "'!\n");
1820
1821   if (DefInit *DI = dyn_cast<DefInit>(R->getValue()))
1822     return DI->getDef();
1823   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1824     FieldName + "' does not have a def initializer!");
1825 }
1826
1827 /// getValueAsBit - This method looks up the specified field and returns its
1828 /// value as a bit, aborting if the field does not exist or if the value is
1829 /// not the right type.
1830 ///
1831 bool Record::getValueAsBit(StringRef FieldName) const {
1832   const RecordVal *R = getValue(FieldName);
1833   if (!R || !R->getValue())
1834     PrintFatalError(getLoc(), "Record `" + getName() +
1835       "' does not have a field named `" + FieldName + "'!\n");
1836
1837   if (BitInit *BI = dyn_cast<BitInit>(R->getValue()))
1838     return BI->getValue();
1839   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1840     FieldName + "' does not have a bit initializer!");
1841 }
1842
1843 bool Record::getValueAsBitOrUnset(StringRef FieldName, bool &Unset) const {
1844   const RecordVal *R = getValue(FieldName);
1845   if (!R || !R->getValue())
1846     PrintFatalError(getLoc(), "Record `" + getName() +
1847       "' does not have a field named `" + FieldName.str() + "'!\n");
1848
1849   if (isa<UnsetInit>(R->getValue())) {
1850     Unset = true;
1851     return false;
1852   }
1853   Unset = false;
1854   if (BitInit *BI = dyn_cast<BitInit>(R->getValue()))
1855     return BI->getValue();
1856   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1857     FieldName + "' does not have a bit initializer!");
1858 }
1859
1860 /// getValueAsDag - This method looks up the specified field and returns its
1861 /// value as an Dag, aborting if the field does not exist or if the value is
1862 /// not the right type.
1863 ///
1864 DagInit *Record::getValueAsDag(StringRef FieldName) const {
1865   const RecordVal *R = getValue(FieldName);
1866   if (!R || !R->getValue())
1867     PrintFatalError(getLoc(), "Record `" + getName() +
1868       "' does not have a field named `" + FieldName + "'!\n");
1869
1870   if (DagInit *DI = dyn_cast<DagInit>(R->getValue()))
1871     return DI;
1872   PrintFatalError(getLoc(), "Record `" + getName() + "', field `" +
1873     FieldName + "' does not have a dag initializer!");
1874 }
1875
1876
1877 void MultiClass::dump() const {
1878   errs() << "Record:\n";
1879   Rec.dump();
1880
1881   errs() << "Defs:\n";
1882   for (const auto &Proto : DefPrototypes)
1883     Proto->dump();
1884 }
1885
1886
1887 void RecordKeeper::dump() const { errs() << *this; }
1888
1889 raw_ostream &llvm::operator<<(raw_ostream &OS, const RecordKeeper &RK) {
1890   OS << "------------- Classes -----------------\n";
1891   for (const auto &C : RK.getClasses())
1892     OS << "class " << *C.second;
1893
1894   OS << "------------- Defs -----------------\n";
1895   for (const auto &D : RK.getDefs())
1896     OS << "def " << *D.second;
1897   return OS;
1898 }
1899
1900
1901 /// getAllDerivedDefinitions - This method returns all concrete definitions
1902 /// that derive from the specified class name.  If a class with the specified
1903 /// name does not exist, an error is printed and true is returned.
1904 std::vector<Record*>
1905 RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
1906   Record *Class = getClass(ClassName);
1907   if (!Class)
1908     PrintFatalError("ERROR: Couldn't find the `" + ClassName + "' class!\n");
1909
1910   std::vector<Record*> Defs;
1911   for (const auto &D : getDefs())
1912     if (D.second->isSubClassOf(Class))
1913       Defs.push_back(D.second.get());
1914
1915   return Defs;
1916 }
1917
1918 /// QualifyName - Return an Init with a qualifier prefix referring
1919 /// to CurRec's name.
1920 Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass,
1921                         Init *Name, const std::string &Scoper) {
1922   RecTy *Type = cast<TypedInit>(Name)->getType();
1923
1924   BinOpInit *NewName =
1925     BinOpInit::get(BinOpInit::STRCONCAT,
1926                       BinOpInit::get(BinOpInit::STRCONCAT,
1927                                         CurRec.getNameInit(),
1928                                         StringInit::get(Scoper),
1929                                         Type)->Fold(&CurRec, CurMultiClass),
1930                       Name,
1931                       Type);
1932
1933   if (CurMultiClass && Scoper != "::") {
1934     NewName =
1935       BinOpInit::get(BinOpInit::STRCONCAT,
1936                         BinOpInit::get(BinOpInit::STRCONCAT,
1937                                           CurMultiClass->Rec.getNameInit(),
1938                                           StringInit::get("::"),
1939                                           Type)->Fold(&CurRec, CurMultiClass),
1940                         NewName->Fold(&CurRec, CurMultiClass),
1941                         Type);
1942   }
1943
1944   return NewName->Fold(&CurRec, CurMultiClass);
1945 }
1946
1947 /// QualifyName - Return an Init with a qualifier prefix referring
1948 /// to CurRec's name.
1949 Init *llvm::QualifyName(Record &CurRec, MultiClass *CurMultiClass,
1950                         const std::string &Name,
1951                         const std::string &Scoper) {
1952   return QualifyName(CurRec, CurMultiClass, StringInit::get(Name), Scoper);
1953 }