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   return new UnOpInit(opc, lhs, Type);
 718 }
 719
 720 const Init *UnOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
 721   switch (getOpcode()) {
 722   default: assert(0 && "Unknown unop");
 723   case CAST: {
 724     if (getType()->getAsString() == "string") {
 725       const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
 726       if (LHSs) {
 727         return LHSs;
 728       }
 729
 730       const DefInit *LHSd = dynamic_cast<const DefInit*>(LHS);
 731       if (LHSd) {
 732         return StringInit::get(LHSd->getDef()->getName());
 733       }
 734     } else {
 735       const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
 736       if (LHSs) {
 737         std::string Name = LHSs->getValue();
 738
 739         // From TGParser::ParseIDValue
 740         if (CurRec) {
 741           if (const RecordVal *RV = CurRec->getValue(Name)) {
 742             if (RV->getType() != getType())
 743               throw "type mismatch in cast";
 744             return VarInit::get(Name, RV->getType());
 745           }
 746
 747           std::string TemplateArgName = CurRec->getName()+":"+Name;
 748           if (CurRec->isTemplateArg(TemplateArgName)) {
 749             const RecordVal *RV = CurRec->getValue(TemplateArgName);
 750             assert(RV && "Template arg doesn't exist??");
 751
 752             if (RV->getType() != getType())
 753               throw "type mismatch in cast";
 754
 755             return VarInit::get(TemplateArgName, RV->getType());
 756           }
 757         }
 758
 759         if (CurMultiClass) {
 760           std::string MCName = CurMultiClass->Rec.getName()+"::"+Name;
 761           if (CurMultiClass->Rec.isTemplateArg(MCName)) {
 762             const RecordVal *RV = CurMultiClass->Rec.getValue(MCName);
 763             assert(RV && "Template arg doesn't exist??");
 764
 765             if (RV->getType() != getType())
 766               throw "type mismatch in cast";
 767
 768             return VarInit::get(MCName, RV->getType());
 769           }
 770         }
 771
 772         if (Record *D = (CurRec->getRecords()).getDef(Name))
 773           return DefInit::get(D);
 774
 775         throw TGError(CurRec->getLoc(), "Undefined reference:'" + Name + "'\n");
 776       }
 777     }
 778     break;
 779   }
 780   case HEAD: {
 781     const ListInit *LHSl = dynamic_cast<const ListInit*>(LHS);
 782     if (LHSl) {
 783       if (LHSl->getSize() == 0) {
 784         assert(0 && "Empty list in car");
 785         return 0;
 786       }
 787       return LHSl->getElement(0);
 788     }
 789     break;
 790   }
 791   case TAIL: {
 792     const ListInit *LHSl = dynamic_cast<const ListInit*>(LHS);
 793     if (LHSl) {
 794       if (LHSl->getSize() == 0) {
 795         assert(0 && "Empty list in cdr");
 796         return 0;
 797       }
 798       // Note the +1.  We can't just pass the result of getValues()
 799       // directly.
 800       ArrayRef<const Init *>::iterator begin = LHSl->getValues().begin()+1;
 801       ArrayRef<const Init *>::iterator end   = LHSl->getValues().end();
 802       const ListInit *Result =
 803         ListInit::get(ArrayRef<const Init *>(begin, end - begin),
 804                       LHSl->getType());
 805       return Result;
 806     }
 807     break;
 808   }
 809   case EMPTY: {
 810     const ListInit *LHSl = dynamic_cast<const ListInit*>(LHS);
 811     if (LHSl) {
 812       if (LHSl->getSize() == 0) {
 813         return IntInit::get(1);
 814       } else {
 815         return IntInit::get(0);
 816       }
 817     }
 818     const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
 819     if (LHSs) {
 820       if (LHSs->getValue().empty()) {
 821         return IntInit::get(1);
 822       } else {
 823         return IntInit::get(0);
 824       }
 825     }
 826
 827     break;
 828   }
 829   }
 830   return this;
 831 }
 832
 833 const Init *UnOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
 834   const Init *lhs = LHS->resolveReferences(R, RV);
 835
 836   if (LHS != lhs)
 837     return (UnOpInit::get(getOpcode(), lhs, getType()))->Fold(&R, 0);
 838   return Fold(&R, 0);
 839 }
 840
 841 std::string UnOpInit::getAsString() const {
 842   std::string Result;
 843   switch (Opc) {
 844   case CAST: Result = "!cast<" + getType()->getAsString() + ">"; break;
 845   case HEAD: Result = "!head"; break;
 846   case TAIL: Result = "!tail"; break;
 847   case EMPTY: Result = "!empty"; break;
 848   }
 849   return Result + "(" + LHS->getAsString() + ")";
 850 }
 851
 852 const BinOpInit *BinOpInit::get(BinaryOp opc, const Init *lhs,
 853                                 const Init *rhs, RecTy *Type) {
 854   return new BinOpInit(opc, lhs, rhs, Type);
 855 }
 856
 857 const Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
 858   switch (getOpcode()) {
 859   default: assert(0 && "Unknown binop");
 860   case CONCAT: {
 861     const DagInit *LHSs = dynamic_cast<const DagInit*>(LHS);
 862     const DagInit *RHSs = dynamic_cast<const DagInit*>(RHS);
 863     if (LHSs && RHSs) {
 864       const DefInit *LOp = dynamic_cast<const DefInit*>(LHSs->getOperator());
 865       const DefInit *ROp = dynamic_cast<const DefInit*>(RHSs->getOperator());
 866       if (LOp == 0 || ROp == 0 || LOp->getDef() != ROp->getDef())
 867         throw "Concated Dag operators do not match!";
 868       std::vector<const Init*> Args;
 869       std::vector<std::string> ArgNames;
 870       for (unsigned i = 0, e = LHSs->getNumArgs(); i != e; ++i) {
 871         Args.push_back(LHSs->getArg(i));
 872         ArgNames.push_back(LHSs->getArgName(i));
 873       }
 874       for (unsigned i = 0, e = RHSs->getNumArgs(); i != e; ++i) {
 875         Args.push_back(RHSs->getArg(i));
 876         ArgNames.push_back(RHSs->getArgName(i));
 877       }
 878       return DagInit::get(LHSs->getOperator(), "", Args, ArgNames);
 879     }
 880     break;
 881   }
 882   case STRCONCAT: {
 883     const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
 884     const StringInit *RHSs = dynamic_cast<const StringInit*>(RHS);
 885     if (LHSs && RHSs)
 886       return StringInit::get(LHSs->getValue() + RHSs->getValue());
 887     break;
 888   }
 889   case EQ: {
 890     // try to fold eq comparison for 'bit' and 'int', otherwise fallback
 891     // to string objects.
 892     const IntInit* L =
 893       dynamic_cast<const IntInit*>(LHS->convertInitializerTo(IntRecTy::get()));
 894     const IntInit* R =
 895       dynamic_cast<const IntInit*>(RHS->convertInitializerTo(IntRecTy::get()));
 896
 897     if (L && R)
 898       return IntInit::get(L->getValue() == R->getValue());
 899
 900     const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
 901     const StringInit *RHSs = dynamic_cast<const StringInit*>(RHS);
 902
 903     // Make sure we've resolved
 904     if (LHSs && RHSs)
 905       return IntInit::get(LHSs->getValue() == RHSs->getValue());
 906
 907     break;
 908   }
 909   case SHL:
 910   case SRA:
 911   case SRL: {
 912     const IntInit *LHSi = dynamic_cast<const IntInit*>(LHS);
 913     const IntInit *RHSi = dynamic_cast<const IntInit*>(RHS);
 914     if (LHSi && RHSi) {
 915       int64_t LHSv = LHSi->getValue(), RHSv = RHSi->getValue();
 916       int64_t Result;
 917       switch (getOpcode()) {
 918       default: assert(0 && "Bad opcode!");
 919       case SHL: Result = LHSv << RHSv; break;
 920       case SRA: Result = LHSv >> RHSv; break;
 921       case SRL: Result = (uint64_t)LHSv >> (uint64_t)RHSv; break;
 922       }
 923       return IntInit::get(Result);
 924     }
 925     break;
 926   }
 927   }
 928   return this;
 929 }
 930
 931 const Init *BinOpInit::resolveReferences(Record &R, const RecordVal *RV) const {
 932   const Init *lhs = LHS->resolveReferences(R, RV);
 933   const Init *rhs = RHS->resolveReferences(R, RV);
 934
 935   if (LHS != lhs || RHS != rhs)
 936     return (BinOpInit::get(getOpcode(), lhs, rhs, getType()))->Fold(&R, 0);
 937   return Fold(&R, 0);
 938 }
 939
 940 std::string BinOpInit::getAsString() const {
 941   std::string Result;
 942   switch (Opc) {
 943   case CONCAT: Result = "!con"; break;
 944   case SHL: Result = "!shl"; break;
 945   case SRA: Result = "!sra"; break;
 946   case SRL: Result = "!srl"; break;
 947   case EQ: Result = "!eq"; break;
 948   case STRCONCAT: Result = "!strconcat"; break;
 949   }
 950   return Result + "(" + LHS->getAsString() + ", " + RHS->getAsString() + ")";
 951 }
 952
 953 const TernOpInit *TernOpInit::get(TernaryOp opc, const Init *lhs,
 954                                   const Init *mhs, const Init *rhs,
 955                                   RecTy *Type) {
 956   return new TernOpInit(opc, lhs, mhs, rhs, Type);
 957 }
 958
 959 static const Init *ForeachHelper(const Init *LHS, const Init *MHS,
 960                                  const Init *RHS, RecTy *Type,
 961                                  Record *CurRec, MultiClass *CurMultiClass);
 962
 963 static const Init *EvaluateOperation(const OpInit *RHSo, const Init *LHS,
 964                                      const Init *Arg,
 965                                      RecTy *Type, Record *CurRec,
 966                                      MultiClass *CurMultiClass) {
 967   std::vector<const Init *> NewOperands;
 968
 969   const TypedInit *TArg = dynamic_cast<const TypedInit*>(Arg);
 970
 971   // If this is a dag, recurse
 972   if (TArg && TArg->getType()->getAsString() == "dag") {
 973     const Init *Result = ForeachHelper(LHS, Arg, RHSo, Type,
 974                                  CurRec, CurMultiClass);
 975     if (Result != 0) {
 976       return Result;
 977     } else {
 978       return 0;
 979     }
 980   }
 981
 982   for (int i = 0; i < RHSo->getNumOperands(); ++i) {
 983     const OpInit *RHSoo = dynamic_cast<const OpInit*>(RHSo->getOperand(i));
 984
 985     if (RHSoo) {
 986       const Init *Result = EvaluateOperation(RHSoo, LHS, Arg,
 987                                        Type, CurRec, CurMultiClass);
 988       if (Result != 0) {
 989         NewOperands.push_back(Result);
 990       } else {
 991         NewOperands.push_back(Arg);
 992       }
 993     } else if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
 994       NewOperands.push_back(Arg);
 995     } else {
 996       NewOperands.push_back(RHSo->getOperand(i));
 997     }
 998   }
 999
1000   // Now run the operator and use its result as the new leaf
1001   const OpInit *NewOp = RHSo->clone(NewOperands);
1002   const Init *NewVal = NewOp->Fold(CurRec, CurMultiClass);
1003   if (NewVal != NewOp)
1004     return NewVal;
1005
1006   return 0;
1007 }
1008
1009 static const Init *ForeachHelper(const Init *LHS, const Init *MHS,
1010                                  const Init *RHS, RecTy *Type,
1011                                  Record *CurRec,
1012                                  MultiClass *CurMultiClass) {
1013   const DagInit *MHSd = dynamic_cast<const DagInit*>(MHS);
1014   const ListInit *MHSl = dynamic_cast<const ListInit*>(MHS);
1015
1016   DagRecTy *DagType = dynamic_cast<DagRecTy*>(Type);
1017   ListRecTy *ListType = dynamic_cast<ListRecTy*>(Type);
1018
1019   const OpInit *RHSo = dynamic_cast<const OpInit*>(RHS);
1020
1021   if (!RHSo) {
1022     throw TGError(CurRec->getLoc(), "!foreach requires an operator\n");
1023   }
1024
1025   const TypedInit *LHSt = dynamic_cast<const TypedInit*>(LHS);
1026
1027   if (!LHSt) {
1028     throw TGError(CurRec->getLoc(), "!foreach requires typed variable\n");
1029   }
1030
1031   if ((MHSd && DagType) || (MHSl && ListType)) {
1032     if (MHSd) {
1033       const Init *Val = MHSd->getOperator();
1034       const Init *Result = EvaluateOperation(RHSo, LHS, Val,
1035                                        Type, CurRec, CurMultiClass);
1036       if (Result != 0) {
1037         Val = Result;
1038       }
1039
1040       std::vector<std::pair<const Init *, std::string> > args;
1041       for (unsigned int i = 0; i < MHSd->getNumArgs(); ++i) {
1042         const Init *Arg;
1043         std::string ArgName;
1044         Arg = MHSd->getArg(i);
1045         ArgName = MHSd->getArgName(i);
1046
1047         // Process args
1048         const Init *Result = EvaluateOperation(RHSo, LHS, Arg, Type,
1049                                          CurRec, CurMultiClass);
1050         if (Result != 0) {
1051           Arg = Result;
1052         }
1053
1054         // TODO: Process arg names
1055         args.push_back(std::make_pair(Arg, ArgName));
1056       }
1057
1058       return DagInit::get(Val, "", args);
1059     }
1060     if (MHSl) {
1061       std::vector<const Init *> NewOperands;
1062       std::vector<const Init *> NewList(MHSl->begin(), MHSl->end());
1063
1064       for (std::vector<const Init *>::iterator li = NewList.begin(),
1065              liend = NewList.end();
1066            li != liend;
1067            ++li) {
1068         const Init *Item = *li;
1069         NewOperands.clear();
1070         for(int i = 0; i < RHSo->getNumOperands(); ++i) {
1071           // First, replace the foreach variable with the list item
1072           if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
1073             NewOperands.push_back(Item);
1074           } else {
1075             NewOperands.push_back(RHSo->getOperand(i));
1076           }
1077         }
1078
1079         // Now run the operator and use its result as the new list item
1080         const OpInit *NewOp = RHSo->clone(NewOperands);
1081         const Init *NewItem = NewOp->Fold(CurRec, CurMultiClass);
1082         if (NewItem != NewOp)
1083           *li = NewItem;
1084       }
1085       return ListInit::get(NewList, MHSl->getType());
1086     }
1087   }
1088   return 0;
1089 }
1090
1091 const Init *TernOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) const {
1092   switch (getOpcode()) {
1093   default: assert(0 && "Unknown binop");
1094   case SUBST: {
1095     const DefInit *LHSd = dynamic_cast<const DefInit*>(LHS);
1096     const VarInit *LHSv = dynamic_cast<const VarInit*>(LHS);
1097     const StringInit *LHSs = dynamic_cast<const StringInit*>(LHS);
1098
1099     const DefInit *MHSd = dynamic_cast<const DefInit*>(MHS);
1100     const VarInit *MHSv = dynamic_cast<const VarInit*>(MHS);
1101     const StringInit *MHSs = dynamic_cast<const StringInit*>(MHS);
1102
1103     const DefInit *RHSd = dynamic_cast<const DefInit*>(RHS);
1104     const VarInit *RHSv = dynamic_cast<const VarInit*>(RHS);
1105     const StringInit *RHSs = dynamic_cast<const StringInit*>(RHS);
1106
1107     if ((LHSd && MHSd && RHSd)
1108         || (LHSv && MHSv && RHSv)
1109         || (LHSs && MHSs && RHSs)) {
1110       if (RHSd) {
1111         Record *Val = RHSd->getDef();
1112         if (LHSd->getAsString() == RHSd->getAsString()) {
1113           Val = MHSd->getDef();
1114         }
1115         return DefInit::get(Val);
1116       }
1117       if (RHSv) {
1118         std::string Val = RHSv->getName();
1119         if (LHSv->getAsString() == RHSv->getAsString()) {
1120           Val = MHSv->getName();
1121         }
1122         return VarInit::get(Val, getType());
1123       }
1124       if (RHSs) {
1125         std::string Val = RHSs->getValue();
1126
1127         std::string::size_type found;
1128         std::string::size_type idx = 0;
1129         do {
1130           found = Val.find(LHSs->getValue(), idx);
1131           if (found != std::string::npos) {
1132             Val.replace(found, LHSs->getValue().size(), MHSs->getValue());
1133           }
1134           idx = found +  MHSs->getValue().size();
1135         } while (found != std::string::npos);
1136
1137         return StringInit::get(Val);
1138       }
1139     }
1140     break;
1141   }
1142
1143   case FOREACH: {
1144     const Init *Result = ForeachHelper(LHS, MHS, RHS, getType(),
1145                                  CurRec, CurMultiClass);
1146     if (Result != 0) {
1147       return Result;
1148     }
1149     break;
1150   }
1151
1152   case IF: {
1153     const IntInit *LHSi = dynamic_cast<const IntInit*>(LHS);
1154     if (const Init *I = LHS->convertInitializerTo(IntRecTy::get()))
1155       LHSi = dynamic_cast<const IntInit*>(I);
1156     if (LHSi) {
1157       if (LHSi->getValue()) {
1158         return MHS;
1159       } else {
1160         return RHS;
1161       }
1162     }
1163     break;
1164   }
1165   }
1166
1167   return this;
1168 }
1169
1170 const Init *TernOpInit::resolveReferences(Record &R,
1171                                           const RecordVal *RV) const {
1172   const Init *lhs = LHS->resolveReferences(R, RV);
1173
1174   if (Opc == IF && lhs != LHS) {
1175     const IntInit *Value = dynamic_cast<const IntInit*>(lhs);
1176     if (const Init *I = lhs->convertInitializerTo(IntRecTy::get()))
1177       Value = dynamic_cast<const IntInit*>(I);
1178     if (Value != 0) {
1179       // Short-circuit
1180       if (Value->getValue()) {
1181         const Init *mhs = MHS->resolveReferences(R, RV);
1182         return (TernOpInit::get(getOpcode(), lhs, mhs,
1183                                 RHS, getType()))->Fold(&R, 0);
1184       } else {
1185         const Init *rhs = RHS->resolveReferences(R, RV);
1186         return (TernOpInit::get(getOpcode(), lhs, MHS,
1187                                 rhs, getType()))->Fold(&R, 0);
1188       }
1189     }
1190   }
1191
1192   const Init *mhs = MHS->resolveReferences(R, RV);
1193   const Init *rhs = RHS->resolveReferences(R, RV);
1194
1195   if (LHS != lhs || MHS != mhs || RHS != rhs)
1196     return (TernOpInit::get(getOpcode(), lhs, mhs, rhs,
1197                             getType()))->Fold(&R, 0);
1198   return Fold(&R, 0);
1199 }
1200
1201 std::string TernOpInit::getAsString() const {
1202   std::string Result;
1203   switch (Opc) {
1204   case SUBST: Result = "!subst"; break;
1205   case FOREACH: Result = "!foreach"; break;
1206   case IF: Result = "!if"; break;
1207  }
1208   return Result + "(" + LHS->getAsString() + ", " + MHS->getAsString() + ", "
1209     + RHS->getAsString() + ")";
1210 }
1211
1212 RecTy *TypedInit::getFieldType(const std::string &FieldName) const {
1213   RecordRecTy *RecordType = dynamic_cast<RecordRecTy *>(getType());
1214   if (RecordType) {
1215     RecordVal *Field = RecordType->getRecord()->getValue(FieldName);
1216     if (Field) {
1217       return Field->getType();
1218     }
1219   }
1220   return 0;
1221 }
1222
1223 const Init *
1224 TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) const {
1225   BitsRecTy *T = dynamic_cast<BitsRecTy*>(getType());
1226   if (T == 0) return 0;  // Cannot subscript a non-bits variable.
1227   unsigned NumBits = T->getNumBits();
1228
1229   SmallVector<const Init *, 16> NewBits(Bits.size());
1230   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
1231     if (Bits[i] >= NumBits)
1232       return 0;
1233
1234     NewBits[i] = VarBitInit::get(this, Bits[i]);
1235   }
1236   return BitsInit::get(NewBits);
1237 }
1238
1239 const Init *
1240 TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) const {
1241   ListRecTy *T = dynamic_cast<ListRecTy*>(getType());
1242   if (T == 0) return 0;  // Cannot subscript a non-list variable.
1243
1244   if (Elements.size() == 1)
1245     return VarListElementInit::get(this, Elements[0]);
1246
1247   std::vector<const Init*> ListInits;
1248   ListInits.reserve(Elements.size());
1249   for (unsigned i = 0, e = Elements.size(); i != e; ++i)
1250     ListInits.push_back(VarListElementInit::get(this, Elements[i]));
1251   return ListInit::get(ListInits, T);
1252 }
1253
1254
1255 const VarInit *VarInit::get(const std::string &VN, RecTy *T) {
1256   return new VarInit(VN, T);
1257 }
1258
1259 const Init *VarInit::resolveBitReference(Record &R, const RecordVal *IRV,
1260                                          unsigned Bit) const {
1261   if (R.isTemplateArg(getName())) return 0;
1262   if (IRV && IRV->getName() != getName()) return 0;
1263
1264   RecordVal *RV = R.getValue(getName());
1265   assert(RV && "Reference to a non-existent variable?");
1266   assert(dynamic_cast<const BitsInit*>(RV->getValue()));
1267   const BitsInit *BI = (const BitsInit*)RV->getValue();
1268
1269   assert(Bit < BI->getNumBits() && "Bit reference out of range!");
1270   const Init *B = BI->getBit(Bit);
1271
1272   // If the bit is set to some value, or if we are resolving a reference to a
1273   // specific variable and that variable is explicitly unset, then replace the
1274   // VarBitInit with it.
1275   if (IRV || !dynamic_cast<const UnsetInit*>(B))
1276     return B;
1277   return 0;
1278 }
1279
1280 const Init *VarInit::resolveListElementReference(Record &R,
1281                                                  const RecordVal *IRV,
1282                                                  unsigned Elt) const {
1283   if (R.isTemplateArg(getName())) return 0;
1284   if (IRV && IRV->getName() != getName()) return 0;
1285
1286   RecordVal *RV = R.getValue(getName());
1287   assert(RV && "Reference to a non-existent variable?");
1288   const ListInit *LI = dynamic_cast<const ListInit*>(RV->getValue());
1289   if (!LI) {
1290     const VarInit *VI = dynamic_cast<const VarInit*>(RV->getValue());
1291     assert(VI && "Invalid list element!");
1292     return VarListElementInit::get(VI, Elt);
1293   }
1294
1295   if (Elt >= LI->getSize())
1296     return 0;  // Out of range reference.
1297   const Init *E = LI->getElement(Elt);
1298   // If the element is set to some value, or if we are resolving a reference
1299   // to a specific variable and that variable is explicitly unset, then
1300   // replace the VarListElementInit with it.
1301   if (IRV || !dynamic_cast<const UnsetInit*>(E))
1302     return E;
1303   return 0;
1304 }
1305
1306
1307 RecTy *VarInit::getFieldType(const std::string &FieldName) const {
1308   if (RecordRecTy *RTy = dynamic_cast<RecordRecTy*>(getType()))
1309     if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName))
1310       return RV->getType();
1311   return 0;
1312 }
1313
1314 const Init *VarInit::getFieldInit(Record &R, const RecordVal *RV,
1315                                   const std::string &FieldName) const {
1316   if (dynamic_cast<RecordRecTy*>(getType()))
1317     if (const RecordVal *Val = R.getValue(VarName)) {
1318       if (RV != Val && (RV || dynamic_cast<const UnsetInit*>(Val->getValue())))
1319         return 0;
1320       const Init *TheInit = Val->getValue();
1321       assert(TheInit != this && "Infinite loop detected!");
1322       if (const Init *I = TheInit->getFieldInit(R, RV, FieldName))
1323         return I;
1324       else
1325         return 0;
1326     }
1327   return 0;
1328 }
1329
1330 /// resolveReferences - This method is used by classes that refer to other
1331 /// variables which may not be defined at the time the expression is formed.
1332 /// If a value is set for the variable later, this method will be called on
1333 /// users of the value to allow the value to propagate out.
1334 ///
1335 const Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) const {
1336   if (RecordVal *Val = R.getValue(VarName))
1337     if (RV == Val || (RV == 0 && !dynamic_cast<const UnsetInit*>(Val->getValue())))
1338       return Val->getValue();
1339   return this;
1340 }
1341
1342 const VarBitInit *VarBitInit::get(const TypedInit *T, unsigned B) {
1343   return new VarBitInit(T, B);
1344 }
1345
1346 std::string VarBitInit::getAsString() const {
1347    return TI->getAsString() + "{" + utostr(Bit) + "}";
1348 }
1349
1350 const Init *VarBitInit::resolveReferences(Record &R,
1351                                           const RecordVal *RV) const {
1352   if (const Init *I = getVariable()->resolveBitReference(R, RV, getBitNum()))
1353     return I;
1354   return this;
1355 }
1356
1357 const VarListElementInit *VarListElementInit::get(const TypedInit *T,
1358                                                   unsigned E) {
1359   return new VarListElementInit(T, E);
1360 }
1361
1362 std::string VarListElementInit::getAsString() const {
1363   return TI->getAsString() + "[" + utostr(Element) + "]";
1364 }
1365
1366 const Init *VarListElementInit::resolveReferences(Record &R,
1367                                                   const RecordVal *RV) const {
1368   if (const Init *I = getVariable()->resolveListElementReference(R, RV,
1369                                                            getElementNum()))
1370     return I;
1371   return this;
1372 }
1373
1374 const Init *VarListElementInit::resolveBitReference(Record &R,
1375                                                     const RecordVal *RV,
1376                                                     unsigned Bit) const {
1377   // FIXME: This should be implemented, to support references like:
1378   // bit B = AA[0]{1};
1379   return 0;
1380 }
1381
1382 const Init *VarListElementInit::
1383 resolveListElementReference(Record &R, const RecordVal *RV,
1384                             unsigned Elt) const {
1385   // FIXME: This should be implemented, to support references like:
1386   // int B = AA[0][1];
1387   return 0;
1388 }
1389
1390 const DefInit *DefInit::get(Record *R) {
1391   return R->getDefInit();
1392 }
1393
1394 RecTy *DefInit::getFieldType(const std::string &FieldName) const {
1395   if (const RecordVal *RV = Def->getValue(FieldName))
1396     return RV->getType();
1397   return 0;
1398 }
1399
1400 const Init *DefInit::getFieldInit(Record &R, const RecordVal *RV,
1401                                   const std::string &FieldName) const {
1402   return Def->getValue(FieldName)->getValue();
1403 }
1404
1405
1406 std::string DefInit::getAsString() const {
1407   return Def->getName();
1408 }
1409
1410 const FieldInit *FieldInit::get(const Init *R, const std::string &FN) {
1411   return new FieldInit(R, FN);
1412 }
1413
1414 const Init *FieldInit::resolveBitReference(Record &R, const RecordVal *RV,
1415                                            unsigned Bit) const {
1416   if (const Init *BitsVal = Rec->getFieldInit(R, RV, FieldName))
1417     if (const BitsInit *BI = dynamic_cast<const BitsInit*>(BitsVal)) {
1418       assert(Bit < BI->getNumBits() && "Bit reference out of range!");
1419       const Init *B = BI->getBit(Bit);
1420
1421       if (dynamic_cast<const BitInit*>(B))  // If the bit is set.
1422         return B;                     // Replace the VarBitInit with it.
1423     }
1424   return 0;
1425 }
1426
1427 const Init *FieldInit::resolveListElementReference(Record &R,
1428                                                    const RecordVal *RV,
1429                                                    unsigned Elt) const {
1430   if (const Init *ListVal = Rec->getFieldInit(R, RV, FieldName))
1431     if (const ListInit *LI = dynamic_cast<const ListInit*>(ListVal)) {
1432       if (Elt >= LI->getSize()) return 0;
1433       const Init *E = LI->getElement(Elt);
1434
1435       // If the element is set to some value, or if we are resolving a
1436       // reference to a specific variable and that variable is explicitly
1437       // unset, then replace the VarListElementInit with it.
1438       if (RV || !dynamic_cast<const UnsetInit*>(E))
1439         return E;
1440     }
1441   return 0;
1442 }
1443
1444 const Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) const {
1445   const Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec;
1446
1447   const Init *BitsVal = NewRec->getFieldInit(R, RV, FieldName);
1448   if (BitsVal) {
1449     const Init *BVR = BitsVal->resolveReferences(R, RV);
1450     return BVR->isComplete() ? BVR : this;
1451   }
1452
1453   if (NewRec != Rec) {
1454     return FieldInit::get(NewRec, FieldName);
1455   }
1456   return this;
1457 }
1458
1459 const DagInit *
1460 DagInit::get(const Init *V, const std::string &VN,
1461              const std::vector<std::pair<const Init*, std::string> > &args) {
1462   typedef std::pair<const Init*, std::string> PairType;
1463
1464   std::vector<const Init *> Args;
1465   std::vector<std::string> Names;
1466
1467   for (std::vector<PairType>::const_iterator i = args.begin(),
1468          iend = args.end();
1469        i != iend;
1470        ++i) {
1471     Args.push_back(i->first);
1472     Names.push_back(i->second);
1473   }
1474
1475   return DagInit::get(V, VN, Args, Names);
1476 }
1477
1478 const DagInit *
1479 DagInit::get(const Init *V, const std::string &VN,
1480              const std::vector<const Init*> &args,
1481              const std::vector<std::string> &argNames) {
1482   return new DagInit(V, VN, args, argNames);
1483 }
1484
1485 const Init *DagInit::resolveReferences(Record &R, const RecordVal *RV) const {
1486   std::vector<const Init*> NewArgs;
1487   for (unsigned i = 0, e = Args.size(); i != e; ++i)
1488     NewArgs.push_back(Args[i]->resolveReferences(R, RV));
1489
1490   const Init *Op = Val->resolveReferences(R, RV);
1491
1492   if (Args != NewArgs || Op != Val)
1493     return DagInit::get(Op, ValName, NewArgs, ArgNames);
1494
1495   return this;
1496 }
1497
1498
1499 std::string DagInit::getAsString() const {
1500   std::string Result = "(" + Val->getAsString();
1501   if (!ValName.empty())
1502     Result += ":" + ValName;
1503   if (Args.size()) {
1504     Result += " " + Args[0]->getAsString();
1505     if (!ArgNames[0].empty()) Result += ":$" + ArgNames[0];
1506     for (unsigned i = 1, e = Args.size(); i != e; ++i) {
1507       Result += ", " + Args[i]->getAsString();
1508       if (!ArgNames[i].empty()) Result += ":$" + ArgNames[i];
1509     }
1510   }
1511   return Result + ")";
1512 }
1513
1514
1515 //===----------------------------------------------------------------------===//
1516 //    Other implementations
1517 //===----------------------------------------------------------------------===//
1518
1519 RecordVal::RecordVal(const std::string &N, RecTy *T, unsigned P)
1520   : Name(N), Ty(T), Prefix(P) {
1521   Value = Ty->convertValue(UnsetInit::get());
1522   assert(Value && "Cannot create unset value for current type!");
1523 }
1524
1525 void RecordVal::dump() const { errs() << *this; }
1526
1527 void RecordVal::print(raw_ostream &OS, bool PrintSem) const {
1528   if (getPrefix()) OS << "field ";
1529   OS << *getType() << " " << getName();
1530
1531   if (getValue())
1532     OS << " = " << *getValue();
1533
1534   if (PrintSem) OS << ";\n";
1535 }
1536
1537 unsigned Record::LastID = 0;
1538
1539 DefInit *Record::getDefInit() {
1540   if (!TheInit)
1541     TheInit = new DefInit(this, new RecordRecTy(this));
1542   return TheInit;
1543 }
1544
1545 void Record::setName(const std::string &Name) {
1546   if (TrackedRecords.getDef(getName()) == this) {
1547     TrackedRecords.removeDef(getName());
1548     this->Name = Name;
1549     TrackedRecords.addDef(this);
1550   } else {
1551     TrackedRecords.removeClass(getName());
1552     this->Name = Name;
1553     TrackedRecords.addClass(this);
1554   }
1555 }
1556
1557 /// resolveReferencesTo - If anything in this record refers to RV, replace the
1558 /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
1559 /// references.
1560 void Record::resolveReferencesTo(const RecordVal *RV) {
1561   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
1562     if (const Init *V = Values[i].getValue())
1563       Values[i].setValue(V->resolveReferences(*this, RV));
1564   }
1565 }
1566
1567 void Record::dump() const { errs() << *this; }
1568
1569 raw_ostream &llvm::operator<<(raw_ostream &OS, const Record &R) {
1570   OS << R.getName();
1571
1572   const std::vector<std::string> &TArgs = R.getTemplateArgs();
1573   if (!TArgs.empty()) {
1574     OS << "<";
1575     for (unsigned i = 0, e = TArgs.size(); i != e; ++i) {
1576       if (i) OS << ", ";
1577       const RecordVal *RV = R.getValue(TArgs[i]);
1578       assert(RV && "Template argument record not found??");
1579       RV->print(OS, false);
1580     }
1581     OS << ">";
1582   }
1583
1584   OS << " {";
1585   const std::vector<Record*> &SC = R.getSuperClasses();
1586   if (!SC.empty()) {
1587     OS << "\t//";
1588     for (unsigned i = 0, e = SC.size(); i != e; ++i)
1589       OS << " " << SC[i]->getName();
1590   }
1591   OS << "\n";
1592
1593   const std::vector<RecordVal> &Vals = R.getValues();
1594   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1595     if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1596       OS << Vals[i];
1597   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1598     if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1599       OS << Vals[i];
1600
1601   return OS << "}\n";
1602 }
1603
1604 /// getValueInit - Return the initializer for a value with the specified name,
1605 /// or throw an exception if the field does not exist.
1606 ///
1607 const Init *Record::getValueInit(StringRef FieldName) const {
1608   const RecordVal *R = getValue(FieldName);
1609   if (R == 0 || R->getValue() == 0)
1610     throw "Record `" + getName() + "' does not have a field named `" +
1611       FieldName.str() + "'!\n";
1612   return R->getValue();
1613 }
1614
1615
1616 /// getValueAsString - This method looks up the specified field and returns its
1617 /// value as a string, throwing an exception if the field does not exist or if
1618 /// the value is not a string.
1619 ///
1620 std::string Record::getValueAsString(StringRef FieldName) const {
1621   const RecordVal *R = getValue(FieldName);
1622   if (R == 0 || R->getValue() == 0)
1623     throw "Record `" + getName() + "' does not have a field named `" +
1624           FieldName.str() + "'!\n";
1625
1626   if (const StringInit *SI = dynamic_cast<const StringInit*>(R->getValue()))
1627     return SI->getValue();
1628   throw "Record `" + getName() + "', field `" + FieldName.str() +
1629         "' does not have a string initializer!";
1630 }
1631
1632 /// getValueAsBitsInit - This method looks up the specified field and returns
1633 /// its value as a BitsInit, throwing an exception if the field does not exist
1634 /// or if the value is not the right type.
1635 ///
1636 const BitsInit *Record::getValueAsBitsInit(StringRef FieldName) const {
1637   const RecordVal *R = getValue(FieldName);
1638   if (R == 0 || R->getValue() == 0)
1639     throw "Record `" + getName() + "' does not have a field named `" +
1640           FieldName.str() + "'!\n";
1641
1642   if (const BitsInit *BI = dynamic_cast<const BitsInit*>(R->getValue()))
1643     return BI;
1644   throw "Record `" + getName() + "', field `" + FieldName.str() +
1645         "' does not have a BitsInit initializer!";
1646 }
1647
1648 /// getValueAsListInit - This method looks up the specified field and returns
1649 /// its value as a ListInit, throwing an exception if the field does not exist
1650 /// or if the value is not the right type.
1651 ///
1652 const ListInit *Record::getValueAsListInit(StringRef FieldName) const {
1653   const RecordVal *R = getValue(FieldName);
1654   if (R == 0 || R->getValue() == 0)
1655     throw "Record `" + getName() + "' does not have a field named `" +
1656           FieldName.str() + "'!\n";
1657
1658   if (const ListInit *LI = dynamic_cast<const ListInit*>(R->getValue()))
1659     return LI;
1660   throw "Record `" + getName() + "', field `" + FieldName.str() +
1661         "' does not have a list initializer!";
1662 }
1663
1664 /// getValueAsListOfDefs - This method looks up the specified field and returns
1665 /// its value as a vector of records, throwing an exception if the field does
1666 /// not exist or if the value is not the right type.
1667 ///
1668 std::vector<Record*>
1669 Record::getValueAsListOfDefs(StringRef FieldName) const {
1670   const ListInit *List = getValueAsListInit(FieldName);
1671   std::vector<Record*> Defs;
1672   for (unsigned i = 0; i < List->getSize(); i++) {
1673     if (const DefInit *DI = dynamic_cast<const DefInit*>(List->getElement(i))) {
1674       Defs.push_back(DI->getDef());
1675     } else {
1676       throw "Record `" + getName() + "', field `" + FieldName.str() +
1677             "' list is not entirely DefInit!";
1678     }
1679   }
1680   return Defs;
1681 }
1682
1683 /// getValueAsInt - This method looks up the specified field and returns its
1684 /// value as an int64_t, throwing an exception if the field does not exist or if
1685 /// the value is not the right type.
1686 ///
1687 int64_t Record::getValueAsInt(StringRef FieldName) const {
1688   const RecordVal *R = getValue(FieldName);
1689   if (R == 0 || R->getValue() == 0)
1690     throw "Record `" + getName() + "' does not have a field named `" +
1691           FieldName.str() + "'!\n";
1692
1693   if (const IntInit *II = dynamic_cast<const IntInit*>(R->getValue()))
1694     return II->getValue();
1695   throw "Record `" + getName() + "', field `" + FieldName.str() +
1696         "' does not have an int initializer!";
1697 }
1698
1699 /// getValueAsListOfInts - This method looks up the specified field and returns
1700 /// its value as a vector of integers, throwing an exception if the field does
1701 /// not exist or if the value is not the right type.
1702 ///
1703 std::vector<int64_t>
1704 Record::getValueAsListOfInts(StringRef FieldName) const {
1705   const ListInit *List = getValueAsListInit(FieldName);
1706   std::vector<int64_t> Ints;
1707   for (unsigned i = 0; i < List->getSize(); i++) {
1708     if (const IntInit *II = dynamic_cast<const IntInit*>(List->getElement(i))) {
1709       Ints.push_back(II->getValue());
1710     } else {
1711       throw "Record `" + getName() + "', field `" + FieldName.str() +
1712             "' does not have a list of ints initializer!";
1713     }
1714   }
1715   return Ints;
1716 }
1717
1718 /// getValueAsListOfStrings - This method looks up the specified field and
1719 /// returns its value as a vector of strings, throwing an exception if the
1720 /// field does not exist or if the value is not the right type.
1721 ///
1722 std::vector<std::string>
1723 Record::getValueAsListOfStrings(StringRef FieldName) const {
1724   const ListInit *List = getValueAsListInit(FieldName);
1725   std::vector<std::string> Strings;
1726   for (unsigned i = 0; i < List->getSize(); i++) {
1727     if (const StringInit *II = dynamic_cast<const StringInit*>(List->getElement(i))) {
1728       Strings.push_back(II->getValue());
1729     } else {
1730       throw "Record `" + getName() + "', field `" + FieldName.str() +
1731             "' does not have a list of strings initializer!";
1732     }
1733   }
1734   return Strings;
1735 }
1736
1737 /// getValueAsDef - This method looks up the specified field and returns its
1738 /// value as a Record, throwing an exception if the field does not exist or if
1739 /// the value is not the right type.
1740 ///
1741 Record *Record::getValueAsDef(StringRef FieldName) const {
1742   const RecordVal *R = getValue(FieldName);
1743   if (R == 0 || R->getValue() == 0)
1744     throw "Record `" + getName() + "' does not have a field named `" +
1745       FieldName.str() + "'!\n";
1746
1747   if (const DefInit *DI = dynamic_cast<const DefInit*>(R->getValue()))
1748     return DI->getDef();
1749   throw "Record `" + getName() + "', field `" + FieldName.str() +
1750         "' does not have a def initializer!";
1751 }
1752
1753 /// getValueAsBit - This method looks up the specified field and returns its
1754 /// value as a bit, throwing an exception if the field does not exist or if
1755 /// the value is not the right type.
1756 ///
1757 bool Record::getValueAsBit(StringRef FieldName) const {
1758   const RecordVal *R = getValue(FieldName);
1759   if (R == 0 || R->getValue() == 0)
1760     throw "Record `" + getName() + "' does not have a field named `" +
1761       FieldName.str() + "'!\n";
1762
1763   if (const BitInit *BI = dynamic_cast<const BitInit*>(R->getValue()))
1764     return BI->getValue();
1765   throw "Record `" + getName() + "', field `" + FieldName.str() +
1766         "' does not have a bit initializer!";
1767 }
1768
1769 /// getValueAsDag - This method looks up the specified field and returns its
1770 /// value as an Dag, throwing an exception if the field does not exist or if
1771 /// the value is not the right type.
1772 ///
1773 const DagInit *Record::getValueAsDag(StringRef FieldName) const {
1774   const RecordVal *R = getValue(FieldName);
1775   if (R == 0 || R->getValue() == 0)
1776     throw "Record `" + getName() + "' does not have a field named `" +
1777       FieldName.str() + "'!\n";
1778
1779   if (const DagInit *DI = dynamic_cast<const DagInit*>(R->getValue()))
1780     return DI;
1781   throw "Record `" + getName() + "', field `" + FieldName.str() +
1782         "' does not have a dag initializer!";
1783 }
1784
1785 std::string Record::getValueAsCode(StringRef FieldName) const {
1786   const RecordVal *R = getValue(FieldName);
1787   if (R == 0 || R->getValue() == 0)
1788     throw "Record `" + getName() + "' does not have a field named `" +
1789       FieldName.str() + "'!\n";
1790
1791   if (const CodeInit *CI = dynamic_cast<const CodeInit*>(R->getValue()))
1792     return CI->getValue();
1793   throw "Record `" + getName() + "', field `" + FieldName.str() +
1794     "' does not have a code initializer!";
1795 }
1796
1797
1798 void MultiClass::dump() const {
1799   errs() << "Record:\n";
1800   Rec.dump();
1801
1802   errs() << "Defs:\n";
1803   for (RecordVector::const_iterator r = DefPrototypes.begin(),
1804          rend = DefPrototypes.end();
1805        r != rend;
1806        ++r) {
1807     (*r)->dump();
1808   }
1809 }
1810
1811
1812 void RecordKeeper::dump() const { errs() << *this; }
1813
1814 raw_ostream &llvm::operator<<(raw_ostream &OS, const RecordKeeper &RK) {
1815   OS << "------------- Classes -----------------\n";
1816   const std::map<std::string, Record*> &Classes = RK.getClasses();
1817   for (std::map<std::string, Record*>::const_iterator I = Classes.begin(),
1818          E = Classes.end(); I != E; ++I)
1819     OS << "class " << *I->second;
1820
1821   OS << "------------- Defs -----------------\n";
1822   const std::map<std::string, Record*> &Defs = RK.getDefs();
1823   for (std::map<std::string, Record*>::const_iterator I = Defs.begin(),
1824          E = Defs.end(); I != E; ++I)
1825     OS << "def " << *I->second;
1826   return OS;
1827 }
1828
1829
1830 /// getAllDerivedDefinitions - This method returns all concrete definitions
1831 /// that derive from the specified class name.  If a class with the specified
1832 /// name does not exist, an error is printed and true is returned.
1833 std::vector<Record*>
1834 RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
1835   Record *Class = getClass(ClassName);
1836   if (!Class)
1837     throw "ERROR: Couldn't find the `" + ClassName + "' class!\n";
1838
1839   std::vector<Record*> Defs;
1840   for (std::map<std::string, Record*>::const_iterator I = getDefs().begin(),
1841          E = getDefs().end(); I != E; ++I)
1842     if (I->second->isSubClassOf(Class))
1843       Defs.push_back(I->second);
1844
1845   return Defs;
1846 }
1847