utils/TableGen/Record.cpp

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