utils/TableGen/Record.cpp

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