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 = RecTy1->getRecord()->getSuperClasses();
 288         for(std::vector<Record *>::const_iterator i = T1SuperClasses.begin(),
 289               iend = T1SuperClasses.end();
 290             i != iend;
 291             ++i) {
 292           RecordRecTy *SuperRecTy1 = new RecordRecTy(*i);
 293           RecTy *NewType1 = resolveTypes(SuperRecTy1, T2);
 294           if (NewType1 != 0) {
 295             if (NewType1 != SuperRecTy1) {
 296               delete SuperRecTy1;
 297             }
 298             return NewType1;
 299           }
 300         }
 301       }
 302       RecordRecTy *RecTy2 = dynamic_cast<RecordRecTy*>(T2);
 303       if (RecTy2) {
 304         // See if T1 inherits from a type T2 also inherits from
 305         const std::vector<Record *> &T2SuperClasses = RecTy2->getRecord()->getSuperClasses();
 306         for(std::vector<Record *>::const_iterator i = T2SuperClasses.begin(),
 307               iend = T2SuperClasses.end();
 308             i != iend;
 309             ++i) {
 310           RecordRecTy *SuperRecTy2 = new RecordRecTy(*i);
 311           RecTy *NewType2 = resolveTypes(T1, SuperRecTy2);
 312           if (NewType2 != 0) {
 313             if (NewType2 != SuperRecTy2) {
 314               delete SuperRecTy2;
 315             }
 316             return NewType2;
 317           }
 318         }
 319       }
 320       return 0;
 321     }
 322     return T2;
 323   }
 324   return T1;
 325 }
 326
 327
 328 //===----------------------------------------------------------------------===//
 329 //    Initializer implementations
 330 //===----------------------------------------------------------------------===//
 331
 332 void Init::dump() const { return print(errs()); }
 333
 334 Init *BitsInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
 335   BitsInit *BI = new BitsInit(Bits.size());
 336   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 337     if (Bits[i] >= getNumBits()) {
 338       delete BI;
 339       return 0;
 340     }
 341     BI->setBit(i, getBit(Bits[i]));
 342   }
 343   return BI;
 344 }
 345
 346 std::string BitsInit::getAsString() const {
 347   //if (!printInHex(OS)) return;
 348   //if (!printAsVariable(OS)) return;
 349   //if (!printAsUnset(OS)) return;
 350
 351   std::string Result = "{ ";
 352   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 353     if (i) Result += ", ";
 354     if (Init *Bit = getBit(e-i-1))
 355       Result += Bit->getAsString();
 356     else
 357       Result += "*";
 358   }
 359   return Result + " }";
 360 }
 361
 362 bool BitsInit::printInHex(raw_ostream &OS) const {
 363   // First, attempt to convert the value into an integer value...
 364   int64_t Result = 0;
 365   for (unsigned i = 0, e = getNumBits(); i != e; ++i)
 366     if (BitInit *Bit = dynamic_cast<BitInit*>(getBit(i))) {
 367       Result |= Bit->getValue() << i;
 368     } else {
 369       return true;
 370     }
 371
 372   OS << format("0x%x", Result);
 373   return false;
 374 }
 375
 376 bool BitsInit::printAsVariable(raw_ostream &OS) const {
 377   // Get the variable that we may be set equal to...
 378   assert(getNumBits() != 0);
 379   VarBitInit *FirstBit = dynamic_cast<VarBitInit*>(getBit(0));
 380   if (FirstBit == 0) return true;
 381   TypedInit *Var = FirstBit->getVariable();
 382
 383   // Check to make sure the types are compatible.
 384   BitsRecTy *Ty = dynamic_cast<BitsRecTy*>(FirstBit->getVariable()->getType());
 385   if (Ty == 0) return true;
 386   if (Ty->getNumBits() != getNumBits()) return true; // Incompatible types!
 387
 388   // Check to make sure all bits are referring to the right bits in the variable
 389   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 390     VarBitInit *Bit = dynamic_cast<VarBitInit*>(getBit(i));
 391     if (Bit == 0 || Bit->getVariable() != Var || Bit->getBitNum() != i)
 392       return true;
 393   }
 394
 395   Var->print(OS);
 396   return false;
 397 }
 398
 399 bool BitsInit::printAsUnset(raw_ostream &OS) const {
 400   for (unsigned i = 0, e = getNumBits(); i != e; ++i)
 401     if (!dynamic_cast<UnsetInit*>(getBit(i)))
 402       return true;
 403   OS << "?";
 404   return false;
 405 }
 406
 407 // resolveReferences - If there are any field references that refer to fields
 408 // that have been filled in, we can propagate the values now.
 409 //
 410 Init *BitsInit::resolveReferences(Record &R, const RecordVal *RV) {
 411   bool Changed = false;
 412   BitsInit *New = new BitsInit(getNumBits());
 413
 414   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 415     Init *B;
 416     Init *CurBit = getBit(i);
 417
 418     do {
 419       B = CurBit;
 420       CurBit = CurBit->resolveReferences(R, RV);
 421       Changed |= B != CurBit;
 422     } while (B != CurBit);
 423     New->setBit(i, CurBit);
 424   }
 425
 426   if (Changed)
 427     return New;
 428   delete New;
 429   return this;
 430 }
 431
 432 std::string IntInit::getAsString() const {
 433   return itostr(Value);
 434 }
 435
 436 Init *IntInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
 437   BitsInit *BI = new BitsInit(Bits.size());
 438
 439   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 440     if (Bits[i] >= 64) {
 441       delete BI;
 442       return 0;
 443     }
 444     BI->setBit(i, new BitInit(Value & (INT64_C(1) << Bits[i])));
 445   }
 446   return BI;
 447 }
 448
 449 Init *ListInit::convertInitListSlice(const std::vector<unsigned> &Elements) {
 450   std::vector<Init*> Vals;
 451   for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
 452     if (Elements[i] >= getSize())
 453       return 0;
 454     Vals.push_back(getElement(Elements[i]));
 455   }
 456   return new ListInit(Vals, getType());
 457 }
 458
 459 Record *ListInit::getElementAsRecord(unsigned i) const {
 460   assert(i < Values.size() && "List element index out of range!");
 461   DefInit *DI = dynamic_cast<DefInit*>(Values[i]);
 462   if (DI == 0) throw "Expected record in list!";
 463   return DI->getDef();
 464 }
 465
 466 Init *ListInit::resolveReferences(Record &R, const RecordVal *RV) {
 467   std::vector<Init*> Resolved;
 468   Resolved.reserve(getSize());
 469   bool Changed = false;
 470
 471   for (unsigned i = 0, e = getSize(); i != e; ++i) {
 472     Init *E;
 473     Init *CurElt = getElement(i);
 474
 475     do {
 476       E = CurElt;
 477       CurElt = CurElt->resolveReferences(R, RV);
 478       Changed |= E != CurElt;
 479     } while (E != CurElt);
 480     Resolved.push_back(E);
 481   }
 482
 483   if (Changed)
 484     return new ListInit(Resolved, getType());
 485   return this;
 486 }
 487
 488 Init *ListInit::resolveListElementReference(Record &R, const RecordVal *IRV,
 489                                            unsigned Elt) {
 490   if (Elt >= getSize())
 491     return 0;  // Out of range reference.
 492   Init *E = getElement(Elt);
 493   // If the element is set to some value, or if we are resolving a reference
 494   // to a specific variable and that variable is explicitly unset, then
 495   // replace the VarListElementInit with it.
 496   if (IRV || !dynamic_cast<UnsetInit*>(E))
 497     return E;
 498   return 0;
 499 }
 500
 501 std::string ListInit::getAsString() const {
 502   std::string Result = "[";
 503   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
 504     if (i) Result += ", ";
 505     Result += Values[i]->getAsString();
 506   }
 507   return Result + "]";
 508 }
 509
 510 Init *OpInit::resolveBitReference(Record &R, const RecordVal *IRV,
 511                                    unsigned Bit) {
 512   Init *Folded = Fold(&R, 0);
 513
 514   if (Folded != this) {
 515     TypedInit *Typed = dynamic_cast<TypedInit *>(Folded);
 516     if (Typed) {
 517       return Typed->resolveBitReference(R, IRV, Bit);
 518     }
 519   }
 520
 521   return 0;
 522 }
 523
 524 Init *OpInit::resolveListElementReference(Record &R, const RecordVal *IRV,
 525                                            unsigned Elt) {
 526   Init *Folded = Fold(&R, 0);
 527
 528   if (Folded != this) {
 529     TypedInit *Typed = dynamic_cast<TypedInit *>(Folded);
 530     if (Typed) {
 531       return Typed->resolveListElementReference(R, IRV, Elt);
 532     }
 533   }
 534
 535   return 0;
 536 }
 537
 538 Init *UnOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) {
 539   switch (getOpcode()) {
 540   default: assert(0 && "Unknown unop");
 541   case CAST: {
 542     if (getType()->getAsString() == "string") {
 543       StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 544       if (LHSs) {
 545         return LHSs;
 546       }
 547
 548       DefInit *LHSd = dynamic_cast<DefInit*>(LHS);
 549       if (LHSd) {
 550         return new StringInit(LHSd->getDef()->getName());
 551       }
 552     }
 553     else {
 554       StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 555       if (LHSs) {
 556         std::string Name = LHSs->getValue();
 557
 558         // From TGParser::ParseIDValue
 559         if (CurRec) {
 560           if (const RecordVal *RV = CurRec->getValue(Name)) {
 561             if (RV->getType() != getType()) {
 562               throw "type mismatch in nameconcat";
 563             }
 564             return new VarInit(Name, RV->getType());
 565           }
 566
 567           std::string TemplateArgName = CurRec->getName()+":"+Name;
 568           if (CurRec->isTemplateArg(TemplateArgName)) {
 569             const RecordVal *RV = CurRec->getValue(TemplateArgName);
 570             assert(RV && "Template arg doesn't exist??");
 571
 572             if (RV->getType() != getType()) {
 573               throw "type mismatch in nameconcat";
 574             }
 575
 576             return new VarInit(TemplateArgName, RV->getType());
 577           }
 578         }
 579
 580         if (CurMultiClass) {
 581           std::string MCName = CurMultiClass->Rec.getName()+"::"+Name;
 582           if (CurMultiClass->Rec.isTemplateArg(MCName)) {
 583             const RecordVal *RV = CurMultiClass->Rec.getValue(MCName);
 584             assert(RV && "Template arg doesn't exist??");
 585
 586             if (RV->getType() != getType()) {
 587               throw "type mismatch in nameconcat";
 588             }
 589
 590             return new VarInit(MCName, RV->getType());
 591           }
 592         }
 593
 594         if (Record *D = Records.getDef(Name))
 595           return new DefInit(D);
 596
 597         errs() << "Variable not defined: '" + Name + "'\n";
 598         assert(0 && "Variable not found");
 599         return 0;
 600       }
 601     }
 602     break;
 603   }
 604   case CAR: {
 605     ListInit *LHSl = dynamic_cast<ListInit*>(LHS);
 606     if (LHSl) {
 607       if (LHSl->getSize() == 0) {
 608         assert(0 && "Empty list in car");
 609         return 0;
 610       }
 611       return LHSl->getElement(0);
 612     }
 613     break;
 614   }
 615   case CDR: {
 616     ListInit *LHSl = dynamic_cast<ListInit*>(LHS);
 617     if (LHSl) {
 618       if (LHSl->getSize() == 0) {
 619         assert(0 && "Empty list in cdr");
 620         return 0;
 621       }
 622       ListInit *Result = new ListInit(LHSl->begin()+1, LHSl->end(), LHSl->getType());
 623       return Result;
 624     }
 625     break;
 626   }
 627   case LNULL: {
 628     ListInit *LHSl = dynamic_cast<ListInit*>(LHS);
 629     if (LHSl) {
 630       if (LHSl->getSize() == 0) {
 631         return new IntInit(1);
 632       }
 633       else {
 634         return new IntInit(0);
 635       }
 636     }
 637     StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 638     if (LHSs) {
 639       if (LHSs->getValue().empty()) {
 640         return new IntInit(1);
 641       }
 642       else {
 643         return new IntInit(0);
 644       }
 645     }
 646
 647     break;
 648   }
 649   }
 650   return this;
 651 }
 652
 653 Init *UnOpInit::resolveReferences(Record &R, const RecordVal *RV) {
 654   Init *lhs = LHS->resolveReferences(R, RV);
 655
 656   if (LHS != lhs)
 657     return (new UnOpInit(getOpcode(), lhs, getType()))->Fold(&R, 0);
 658   return Fold(&R, 0);
 659 }
 660
 661 std::string UnOpInit::getAsString() const {
 662   std::string Result;
 663   switch (Opc) {
 664   case CAST: Result = "!cast<" + getType()->getAsString() + ">"; break;
 665   case CAR: Result = "!car"; break;
 666   case CDR: Result = "!cdr"; break;
 667   case LNULL: Result = "!null"; break;
 668   }
 669   return Result + "(" + LHS->getAsString() + ")";
 670 }
 671
 672 RecTy *UnOpInit::getFieldType(const std::string &FieldName) const {
 673   switch (getOpcode()) {
 674   default: assert(0 && "Unknown unop");
 675   case CAST: {
 676     RecordRecTy *RecordType = dynamic_cast<RecordRecTy *>(getType());
 677     if (RecordType) {
 678       RecordVal *Field = RecordType->getRecord()->getValue(FieldName);
 679       if (Field) {
 680         return Field->getType();
 681       }
 682     }
 683     break;
 684   }
 685   }
 686   return 0;
 687 }
 688
 689 Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) {
 690   switch (getOpcode()) {
 691   default: assert(0 && "Unknown binop");
 692   case CONCAT: {
 693     DagInit *LHSs = dynamic_cast<DagInit*>(LHS);
 694     DagInit *RHSs = dynamic_cast<DagInit*>(RHS);
 695     if (LHSs && RHSs) {
 696       DefInit *LOp = dynamic_cast<DefInit*>(LHSs->getOperator());
 697       DefInit *ROp = dynamic_cast<DefInit*>(RHSs->getOperator());
 698       if (LOp->getDef() != ROp->getDef()) {
 699         bool LIsOps =
 700           LOp->getDef()->getName() == "outs" ||
 701           LOp->getDef()->getName() != "ins" ||
 702           LOp->getDef()->getName() != "defs";
 703         bool RIsOps =
 704           ROp->getDef()->getName() == "outs" ||
 705           ROp->getDef()->getName() != "ins" ||
 706           ROp->getDef()->getName() != "defs";
 707         if (!LIsOps || !RIsOps)
 708           throw "Concated Dag operators do not match!";
 709       }
 710       std::vector<Init*> Args;
 711       std::vector<std::string> ArgNames;
 712       for (unsigned i = 0, e = LHSs->getNumArgs(); i != e; ++i) {
 713         Args.push_back(LHSs->getArg(i));
 714         ArgNames.push_back(LHSs->getArgName(i));
 715       }
 716       for (unsigned i = 0, e = RHSs->getNumArgs(); i != e; ++i) {
 717         Args.push_back(RHSs->getArg(i));
 718         ArgNames.push_back(RHSs->getArgName(i));
 719       }
 720       return new DagInit(LHSs->getOperator(), "", Args, ArgNames);
 721     }
 722     break;
 723   }
 724   case STRCONCAT: {
 725     StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 726     StringInit *RHSs = dynamic_cast<StringInit*>(RHS);
 727     if (LHSs && RHSs)
 728       return new StringInit(LHSs->getValue() + RHSs->getValue());
 729     break;
 730   }
 731   case NAMECONCAT: {
 732     StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 733     StringInit *RHSs = dynamic_cast<StringInit*>(RHS);
 734     if (LHSs && RHSs) {
 735       std::string Name(LHSs->getValue() + RHSs->getValue());
 736
 737       // From TGParser::ParseIDValue
 738       if (CurRec) {
 739         if (const RecordVal *RV = CurRec->getValue(Name)) {
 740           if (RV->getType() != getType()) {
 741             throw "type mismatch in nameconcat";
 742           }
 743           return new VarInit(Name, RV->getType());
 744         }
 745
 746         std::string TemplateArgName = CurRec->getName()+":"+Name;
 747         if (CurRec->isTemplateArg(TemplateArgName)) {
 748           const RecordVal *RV = CurRec->getValue(TemplateArgName);
 749           assert(RV && "Template arg doesn't exist??");
 750
 751           if (RV->getType() != getType()) {
 752             throw "type mismatch in nameconcat";
 753           }
 754
 755           return new VarInit(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 nameconcat";
 767           }
 768
 769           return new VarInit(MCName, RV->getType());
 770         }
 771       }
 772
 773       if (Record *D = Records.getDef(Name))
 774         return new DefInit(D);
 775
 776       errs() << "Variable not defined in !nameconcat: '" + Name + "'\n";
 777       assert(0 && "Variable not found in !nameconcat");
 778       return 0;
 779     }
 780     break;
 781   }
 782   case SHL:
 783   case SRA:
 784   case SRL: {
 785     IntInit *LHSi = dynamic_cast<IntInit*>(LHS);
 786     IntInit *RHSi = dynamic_cast<IntInit*>(RHS);
 787     if (LHSi && RHSi) {
 788       int64_t LHSv = LHSi->getValue(), RHSv = RHSi->getValue();
 789       int64_t Result;
 790       switch (getOpcode()) {
 791       default: assert(0 && "Bad opcode!");
 792       case SHL: Result = LHSv << RHSv; break;
 793       case SRA: Result = LHSv >> RHSv; break;
 794       case SRL: Result = (uint64_t)LHSv >> (uint64_t)RHSv; break;
 795       }
 796       return new IntInit(Result);
 797     }
 798     break;
 799   }
 800   }
 801   return this;
 802 }
 803
 804 Init *BinOpInit::resolveReferences(Record &R, const RecordVal *RV) {
 805   Init *lhs = LHS->resolveReferences(R, RV);
 806   Init *rhs = RHS->resolveReferences(R, RV);
 807
 808   if (LHS != lhs || RHS != rhs)
 809     return (new BinOpInit(getOpcode(), lhs, rhs, getType()))->Fold(&R, 0);
 810   return Fold(&R, 0);
 811 }
 812
 813 std::string BinOpInit::getAsString() const {
 814   std::string Result;
 815   switch (Opc) {
 816   case CONCAT: Result = "!con"; break;
 817   case SHL: Result = "!shl"; break;
 818   case SRA: Result = "!sra"; break;
 819   case SRL: Result = "!srl"; break;
 820   case STRCONCAT: Result = "!strconcat"; break;
 821   case NAMECONCAT:
 822     Result = "!nameconcat<" + getType()->getAsString() + ">"; break;
 823   }
 824   return Result + "(" + LHS->getAsString() + ", " + RHS->getAsString() + ")";
 825 }
 826
 827 static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
 828                            Record *CurRec, MultiClass *CurMultiClass);
 829
 830 static Init *EvaluateOperation(OpInit *RHSo, Init *LHS, Init *Arg,
 831                                RecTy *Type, Record *CurRec,
 832                                MultiClass *CurMultiClass) {
 833   std::vector<Init *> NewOperands;
 834
 835   TypedInit *TArg = dynamic_cast<TypedInit*>(Arg);
 836
 837   // If this is a dag, recurse
 838   if (TArg && TArg->getType()->getAsString() == "dag") {
 839     Init *Result = ForeachHelper(LHS, Arg, RHSo, Type,
 840                                  CurRec, CurMultiClass);
 841     if (Result != 0) {
 842       return Result;
 843     }
 844     else {
 845       return 0;
 846     }
 847   }
 848
 849   for (int i = 0; i < RHSo->getNumOperands(); ++i) {
 850     OpInit *RHSoo = dynamic_cast<OpInit*>(RHSo->getOperand(i));
 851
 852     if (RHSoo) {
 853       Init *Result = EvaluateOperation(RHSoo, LHS, Arg,
 854                                        Type, CurRec, CurMultiClass);
 855       if (Result != 0) {
 856         NewOperands.push_back(Result);
 857       }
 858       else {
 859         NewOperands.push_back(Arg);
 860       }
 861     }
 862     else if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
 863       NewOperands.push_back(Arg);
 864     }
 865     else {
 866       NewOperands.push_back(RHSo->getOperand(i));
 867     }
 868   }
 869
 870   // Now run the operator and use its result as the new leaf
 871   OpInit *NewOp = RHSo->clone(NewOperands);
 872   Init *NewVal = NewOp->Fold(CurRec, CurMultiClass);
 873   if (NewVal != NewOp) {
 874     delete NewOp;
 875     return NewVal;
 876   }
 877   return 0;
 878 }
 879
 880 static Init *ForeachHelper(Init *LHS, Init *MHS, Init *RHS, RecTy *Type,
 881                            Record *CurRec, MultiClass *CurMultiClass) {
 882   DagInit *MHSd = dynamic_cast<DagInit*>(MHS);
 883   ListInit *MHSl = dynamic_cast<ListInit*>(MHS);
 884
 885   DagRecTy *DagType = dynamic_cast<DagRecTy*>(Type);
 886   ListRecTy *ListType = dynamic_cast<ListRecTy*>(Type);
 887
 888   OpInit *RHSo = dynamic_cast<OpInit*>(RHS);
 889
 890   if (!RHSo) {
 891     errs() << "!foreach requires an operator\n";
 892     assert(0 && "No operator for !foreach");
 893   }
 894
 895   TypedInit *LHSt = dynamic_cast<TypedInit*>(LHS);
 896
 897   if (!LHSt) {
 898     errs() << "!foreach requires typed variable\n";
 899     assert(0 && "No typed variable for !foreach");
 900   }
 901
 902   if ((MHSd && DagType) || (MHSl && ListType)) {
 903     if (MHSd) {
 904       Init *Val = MHSd->getOperator();
 905       Init *Result = EvaluateOperation(RHSo, LHS, Val,
 906                                        Type, CurRec, CurMultiClass);
 907       if (Result != 0) {
 908         Val = Result;
 909       }
 910
 911       std::vector<std::pair<Init *, std::string> > args;
 912       for (unsigned int i = 0; i < MHSd->getNumArgs(); ++i) {
 913         Init *Arg;
 914         std::string ArgName;
 915         Arg = MHSd->getArg(i);
 916         ArgName = MHSd->getArgName(i);
 917
 918         // Process args
 919         Init *Result = EvaluateOperation(RHSo, LHS, Arg, Type,
 920                                          CurRec, CurMultiClass);
 921         if (Result != 0) {
 922           Arg = Result;
 923         }
 924
 925         // TODO: Process arg names
 926         args.push_back(std::make_pair(Arg, ArgName));
 927       }
 928
 929       return new DagInit(Val, "", args);
 930     }
 931     if (MHSl) {
 932       std::vector<Init *> NewOperands;
 933       std::vector<Init *> NewList(MHSl->begin(), MHSl->end());
 934
 935       for (ListInit::iterator li = NewList.begin(),
 936              liend = NewList.end();
 937            li != liend;
 938            ++li) {
 939         Init *Item = *li;
 940         NewOperands.clear();
 941         for(int i = 0; i < RHSo->getNumOperands(); ++i) {
 942           // First, replace the foreach variable with the list item
 943           if (LHS->getAsString() == RHSo->getOperand(i)->getAsString()) {
 944             NewOperands.push_back(Item);
 945           }
 946           else {
 947             NewOperands.push_back(RHSo->getOperand(i));
 948           }
 949         }
 950
 951         // Now run the operator and use its result as the new list item
 952         OpInit *NewOp = RHSo->clone(NewOperands);
 953         Init *NewItem = NewOp->Fold(CurRec, CurMultiClass);
 954         if (NewItem != NewOp) {
 955           *li = NewItem;
 956           delete NewOp;
 957         }
 958       }
 959       return new ListInit(NewList, MHSl->getType());
 960     }
 961   }
 962   return 0;
 963 }
 964
 965 Init *TernOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) {
 966   switch (getOpcode()) {
 967   default: assert(0 && "Unknown binop");
 968   case SUBST: {
 969     DefInit *LHSd = dynamic_cast<DefInit*>(LHS);
 970     VarInit *LHSv = dynamic_cast<VarInit*>(LHS);
 971     StringInit *LHSs = dynamic_cast<StringInit*>(LHS);
 972
 973     DefInit *MHSd = dynamic_cast<DefInit*>(MHS);
 974     VarInit *MHSv = dynamic_cast<VarInit*>(MHS);
 975     StringInit *MHSs = dynamic_cast<StringInit*>(MHS);
 976
 977     DefInit *RHSd = dynamic_cast<DefInit*>(RHS);
 978     VarInit *RHSv = dynamic_cast<VarInit*>(RHS);
 979     StringInit *RHSs = dynamic_cast<StringInit*>(RHS);
 980
 981     if ((LHSd && MHSd && RHSd)
 982         || (LHSv && MHSv && RHSv)
 983         || (LHSs && MHSs && RHSs)) {
 984       if (RHSd) {
 985         Record *Val = RHSd->getDef();
 986         if (LHSd->getAsString() == RHSd->getAsString()) {
 987           Val = MHSd->getDef();
 988         }
 989         return new DefInit(Val);
 990       }
 991       if (RHSv) {
 992         std::string Val = RHSv->getName();
 993         if (LHSv->getAsString() == RHSv->getAsString()) {
 994           Val = MHSv->getName();
 995         }
 996         return new VarInit(Val, getType());
 997       }
 998       if (RHSs) {
 999         std::string Val = RHSs->getValue();
1000
1001         std::string::size_type found;
1002         do {
1003           found = Val.find(LHSs->getValue());
1004           if (found != std::string::npos) {
1005             Val.replace(found, LHSs->getValue().size(), MHSs->getValue());
1006           }
1007         } while (found != std::string::npos);
1008
1009         return new StringInit(Val);
1010       }
1011     }
1012     break;
1013   }
1014
1015   case FOREACH: {
1016     Init *Result = ForeachHelper(LHS, MHS, RHS, getType(),
1017                                  CurRec, CurMultiClass);
1018     if (Result != 0) {
1019       return Result;
1020     }
1021     break;
1022   }
1023
1024   case IF: {
1025     IntInit *LHSi = dynamic_cast<IntInit*>(LHS);
1026     if (LHSi) {
1027       if (LHSi->getValue()) {
1028         return MHS;
1029       }
1030       else {
1031         return RHS;
1032       }
1033     }
1034     break;
1035   }
1036   }
1037
1038   return this;
1039 }
1040
1041 Init *TernOpInit::resolveReferences(Record &R, const RecordVal *RV) {
1042   Init *lhs = LHS->resolveReferences(R, RV);
1043
1044   if (Opc == IF && lhs != LHS) {
1045     IntInit *Value = dynamic_cast<IntInit*>(lhs);
1046     if (Value != 0) {
1047       // Short-circuit
1048       if (Value->getValue()) {
1049         Init *mhs = MHS->resolveReferences(R, RV);
1050         return (new TernOpInit(getOpcode(), lhs, mhs, RHS, getType()))->Fold(&R, 0);
1051       }
1052       else {
1053         Init *rhs = RHS->resolveReferences(R, RV);
1054         return (new TernOpInit(getOpcode(), lhs, MHS, rhs, getType()))->Fold(&R, 0);
1055       }
1056     }
1057   }
1058
1059   Init *mhs = MHS->resolveReferences(R, RV);
1060   Init *rhs = RHS->resolveReferences(R, RV);
1061
1062   if (LHS != lhs || MHS != mhs || RHS != rhs)
1063     return (new TernOpInit(getOpcode(), lhs, mhs, rhs, getType()))->Fold(&R, 0);
1064   return Fold(&R, 0);
1065 }
1066
1067 std::string TernOpInit::getAsString() const {
1068   std::string Result;
1069   switch (Opc) {
1070   case SUBST: Result = "!subst"; break;
1071   case FOREACH: Result = "!foreach"; break;
1072   case IF: Result = "!if"; break;
1073  }
1074   return Result + "(" + LHS->getAsString() + ", " + MHS->getAsString() + ", "
1075     + RHS->getAsString() + ")";
1076 }
1077
1078 Init *TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
1079   BitsRecTy *T = dynamic_cast<BitsRecTy*>(getType());
1080   if (T == 0) return 0;  // Cannot subscript a non-bits variable...
1081   unsigned NumBits = T->getNumBits();
1082
1083   BitsInit *BI = new BitsInit(Bits.size());
1084   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
1085     if (Bits[i] >= NumBits) {
1086       delete BI;
1087       return 0;
1088     }
1089     BI->setBit(i, new VarBitInit(this, Bits[i]));
1090   }
1091   return BI;
1092 }
1093
1094 Init *TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) {
1095   ListRecTy *T = dynamic_cast<ListRecTy*>(getType());
1096   if (T == 0) return 0;  // Cannot subscript a non-list variable...
1097
1098   if (Elements.size() == 1)
1099     return new VarListElementInit(this, Elements[0]);
1100
1101   std::vector<Init*> ListInits;
1102   ListInits.reserve(Elements.size());
1103   for (unsigned i = 0, e = Elements.size(); i != e; ++i)
1104     ListInits.push_back(new VarListElementInit(this, Elements[i]));
1105   return new ListInit(ListInits, T);
1106 }
1107
1108
1109 Init *VarInit::resolveBitReference(Record &R, const RecordVal *IRV,
1110                                    unsigned Bit) {
1111   if (R.isTemplateArg(getName())) return 0;
1112   if (IRV && IRV->getName() != getName()) return 0;
1113
1114   RecordVal *RV = R.getValue(getName());
1115   assert(RV && "Reference to a non-existent variable?");
1116   assert(dynamic_cast<BitsInit*>(RV->getValue()));
1117   BitsInit *BI = (BitsInit*)RV->getValue();
1118
1119   assert(Bit < BI->getNumBits() && "Bit reference out of range!");
1120   Init *B = BI->getBit(Bit);
1121
1122   // If the bit is set to some value, or if we are resolving a reference to a
1123   // specific variable and that variable is explicitly unset, then replace the
1124   // VarBitInit with it.
1125   if (IRV || !dynamic_cast<UnsetInit*>(B))
1126     return B;
1127   return 0;
1128 }
1129
1130 Init *VarInit::resolveListElementReference(Record &R, const RecordVal *IRV,
1131                                            unsigned Elt) {
1132   if (R.isTemplateArg(getName())) return 0;
1133   if (IRV && IRV->getName() != getName()) return 0;
1134
1135   RecordVal *RV = R.getValue(getName());
1136   assert(RV && "Reference to a non-existent variable?");
1137   ListInit *LI = dynamic_cast<ListInit*>(RV->getValue());
1138   if (!LI) {
1139     VarInit *VI = dynamic_cast<VarInit*>(RV->getValue());
1140     assert(VI && "Invalid list element!");
1141     return new VarListElementInit(VI, Elt);
1142   }
1143
1144   if (Elt >= LI->getSize())
1145     return 0;  // Out of range reference.
1146   Init *E = LI->getElement(Elt);
1147   // If the element is set to some value, or if we are resolving a reference
1148   // to a specific variable and that variable is explicitly unset, then
1149   // replace the VarListElementInit with it.
1150   if (IRV || !dynamic_cast<UnsetInit*>(E))
1151     return E;
1152   return 0;
1153 }
1154
1155
1156 RecTy *VarInit::getFieldType(const std::string &FieldName) const {
1157   if (RecordRecTy *RTy = dynamic_cast<RecordRecTy*>(getType()))
1158     if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName))
1159       return RV->getType();
1160   return 0;
1161 }
1162
1163 Init *VarInit::getFieldInit(Record &R, const std::string &FieldName) const {
1164   if (dynamic_cast<RecordRecTy*>(getType()))
1165     if (const RecordVal *RV = R.getValue(VarName)) {
1166       Init *TheInit = RV->getValue();
1167       assert(TheInit != this && "Infinite loop detected!");
1168       if (Init *I = TheInit->getFieldInit(R, FieldName))
1169         return I;
1170       else
1171         return 0;
1172     }
1173   return 0;
1174 }
1175
1176 /// resolveReferences - This method is used by classes that refer to other
1177 /// variables which may not be defined at the time the expression is formed.
1178 /// If a value is set for the variable later, this method will be called on
1179 /// users of the value to allow the value to propagate out.
1180 ///
1181 Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) {
1182   if (RecordVal *Val = R.getValue(VarName))
1183     if (RV == Val || (RV == 0 && !dynamic_cast<UnsetInit*>(Val->getValue())))
1184       return Val->getValue();
1185   return this;
1186 }
1187
1188 std::string VarBitInit::getAsString() const {
1189    return TI->getAsString() + "{" + utostr(Bit) + "}";
1190 }
1191
1192 Init *VarBitInit::resolveReferences(Record &R, const RecordVal *RV) {
1193   if (Init *I = getVariable()->resolveBitReference(R, RV, getBitNum()))
1194     return I;
1195   return this;
1196 }
1197
1198 std::string VarListElementInit::getAsString() const {
1199   return TI->getAsString() + "[" + utostr(Element) + "]";
1200 }
1201
1202 Init *VarListElementInit::resolveReferences(Record &R, const RecordVal *RV) {
1203   if (Init *I = getVariable()->resolveListElementReference(R, RV,
1204                                                            getElementNum()))
1205     return I;
1206   return this;
1207 }
1208
1209 Init *VarListElementInit::resolveBitReference(Record &R, const RecordVal *RV,
1210                                               unsigned Bit) {
1211   // FIXME: This should be implemented, to support references like:
1212   // bit B = AA[0]{1};
1213   return 0;
1214 }
1215
1216 Init *VarListElementInit::
1217 resolveListElementReference(Record &R, const RecordVal *RV, unsigned Elt) {
1218   // FIXME: This should be implemented, to support references like:
1219   // int B = AA[0][1];
1220   return 0;
1221 }
1222
1223 RecTy *DefInit::getFieldType(const std::string &FieldName) const {
1224   if (const RecordVal *RV = Def->getValue(FieldName))
1225     return RV->getType();
1226   return 0;
1227 }
1228
1229 Init *DefInit::getFieldInit(Record &R, const std::string &FieldName) const {
1230   return Def->getValue(FieldName)->getValue();
1231 }
1232
1233
1234 std::string DefInit::getAsString() const {
1235   return Def->getName();
1236 }
1237
1238 Init *FieldInit::resolveBitReference(Record &R, const RecordVal *RV,
1239                                      unsigned Bit) {
1240   if (Init *BitsVal = Rec->getFieldInit(R, FieldName))
1241     if (BitsInit *BI = dynamic_cast<BitsInit*>(BitsVal)) {
1242       assert(Bit < BI->getNumBits() && "Bit reference out of range!");
1243       Init *B = BI->getBit(Bit);
1244
1245       if (dynamic_cast<BitInit*>(B))  // If the bit is set...
1246         return B;                     // Replace the VarBitInit with it.
1247     }
1248   return 0;
1249 }
1250
1251 Init *FieldInit::resolveListElementReference(Record &R, const RecordVal *RV,
1252                                              unsigned Elt) {
1253   if (Init *ListVal = Rec->getFieldInit(R, FieldName))
1254     if (ListInit *LI = dynamic_cast<ListInit*>(ListVal)) {
1255       if (Elt >= LI->getSize()) return 0;
1256       Init *E = LI->getElement(Elt);
1257
1258       // If the element is set to some value, or if we are resolving a
1259       // reference to a specific variable and that variable is explicitly
1260       // unset, then replace the VarListElementInit with it.
1261       if (RV || !dynamic_cast<UnsetInit*>(E))
1262         return E;
1263     }
1264   return 0;
1265 }
1266
1267 Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) {
1268   Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec;
1269
1270   Init *BitsVal = NewRec->getFieldInit(R, FieldName);
1271   if (BitsVal) {
1272     Init *BVR = BitsVal->resolveReferences(R, RV);
1273     return BVR->isComplete() ? BVR : this;
1274   }
1275
1276   if (NewRec != Rec) {
1277     return new FieldInit(NewRec, FieldName);
1278   }
1279   return this;
1280 }
1281
1282 Init *DagInit::resolveReferences(Record &R, const RecordVal *RV) {
1283   std::vector<Init*> NewArgs;
1284   for (unsigned i = 0, e = Args.size(); i != e; ++i)
1285     NewArgs.push_back(Args[i]->resolveReferences(R, RV));
1286
1287   Init *Op = Val->resolveReferences(R, RV);
1288
1289   if (Args != NewArgs || Op != Val)
1290     return new DagInit(Op, "", NewArgs, ArgNames);
1291
1292   return this;
1293 }
1294
1295
1296 std::string DagInit::getAsString() const {
1297   std::string Result = "(" + Val->getAsString();
1298   if (!ValName.empty())
1299     Result += ":" + ValName;
1300   if (Args.size()) {
1301     Result += " " + Args[0]->getAsString();
1302     if (!ArgNames[0].empty()) Result += ":$" + ArgNames[0];
1303     for (unsigned i = 1, e = Args.size(); i != e; ++i) {
1304       Result += ", " + Args[i]->getAsString();
1305       if (!ArgNames[i].empty()) Result += ":$" + ArgNames[i];
1306     }
1307   }
1308   return Result + ")";
1309 }
1310
1311
1312 //===----------------------------------------------------------------------===//
1313 //    Other implementations
1314 //===----------------------------------------------------------------------===//
1315
1316 RecordVal::RecordVal(const std::string &N, RecTy *T, unsigned P)
1317   : Name(N), Ty(T), Prefix(P) {
1318   Value = Ty->convertValue(new UnsetInit());
1319   assert(Value && "Cannot create unset value for current type!");
1320 }
1321
1322 void RecordVal::dump() const { errs() << *this; }
1323
1324 void RecordVal::print(raw_ostream &OS, bool PrintSem) const {
1325   if (getPrefix()) OS << "field ";
1326   OS << *getType() << " " << getName();
1327
1328   if (getValue())
1329     OS << " = " << *getValue();
1330
1331   if (PrintSem) OS << ";\n";
1332 }
1333
1334 unsigned Record::LastID = 0;
1335
1336 void Record::setName(const std::string &Name) {
1337   if (Records.getDef(getName()) == this) {
1338     Records.removeDef(getName());
1339     this->Name = Name;
1340     Records.addDef(this);
1341   } else {
1342     Records.removeClass(getName());
1343     this->Name = Name;
1344     Records.addClass(this);
1345   }
1346 }
1347
1348 /// resolveReferencesTo - If anything in this record refers to RV, replace the
1349 /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
1350 /// references.
1351 void Record::resolveReferencesTo(const RecordVal *RV) {
1352   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
1353     if (Init *V = Values[i].getValue())
1354       Values[i].setValue(V->resolveReferences(*this, RV));
1355   }
1356 }
1357
1358
1359 void Record::dump() const { errs() << *this; }
1360
1361 raw_ostream &llvm::operator<<(raw_ostream &OS, const Record &R) {
1362   OS << R.getName();
1363
1364   const std::vector<std::string> &TArgs = R.getTemplateArgs();
1365   if (!TArgs.empty()) {
1366     OS << "<";
1367     for (unsigned i = 0, e = TArgs.size(); i != e; ++i) {
1368       if (i) OS << ", ";
1369       const RecordVal *RV = R.getValue(TArgs[i]);
1370       assert(RV && "Template argument record not found??");
1371       RV->print(OS, false);
1372     }
1373     OS << ">";
1374   }
1375
1376   OS << " {";
1377   const std::vector<Record*> &SC = R.getSuperClasses();
1378   if (!SC.empty()) {
1379     OS << "\t//";
1380     for (unsigned i = 0, e = SC.size(); i != e; ++i)
1381       OS << " " << SC[i]->getName();
1382   }
1383   OS << "\n";
1384
1385   const std::vector<RecordVal> &Vals = R.getValues();
1386   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1387     if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1388       OS << Vals[i];
1389   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
1390     if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
1391       OS << Vals[i];
1392
1393   return OS << "}\n";
1394 }
1395
1396 /// getValueInit - Return the initializer for a value with the specified name,
1397 /// or throw an exception if the field does not exist.
1398 ///
1399 Init *Record::getValueInit(StringRef FieldName) const {
1400   const RecordVal *R = getValue(FieldName);
1401   if (R == 0 || R->getValue() == 0)
1402     throw "Record `" + getName() + "' does not have a field named `" +
1403       FieldName.str() + "'!\n";
1404   return R->getValue();
1405 }
1406
1407
1408 /// getValueAsString - This method looks up the specified field and returns its
1409 /// value as a string, throwing an exception if the field does not exist or if
1410 /// the value is not a string.
1411 ///
1412 std::string Record::getValueAsString(StringRef FieldName) const {
1413   const RecordVal *R = getValue(FieldName);
1414   if (R == 0 || R->getValue() == 0)
1415     throw "Record `" + getName() + "' does not have a field named `" +
1416           FieldName.str() + "'!\n";
1417
1418   if (const StringInit *SI = dynamic_cast<const StringInit*>(R->getValue()))
1419     return SI->getValue();
1420   throw "Record `" + getName() + "', field `" + FieldName.str() +
1421         "' does not have a string initializer!";
1422 }
1423
1424 /// getValueAsBitsInit - This method looks up the specified field and returns
1425 /// its value as a BitsInit, throwing an exception if the field does not exist
1426 /// or if the value is not the right type.
1427 ///
1428 BitsInit *Record::getValueAsBitsInit(StringRef FieldName) const {
1429   const RecordVal *R = getValue(FieldName);
1430   if (R == 0 || R->getValue() == 0)
1431     throw "Record `" + getName() + "' does not have a field named `" +
1432           FieldName.str() + "'!\n";
1433
1434   if (BitsInit *BI = dynamic_cast<BitsInit*>(R->getValue()))
1435     return BI;
1436   throw "Record `" + getName() + "', field `" + FieldName.str() +
1437         "' does not have a BitsInit initializer!";
1438 }
1439
1440 /// getValueAsListInit - This method looks up the specified field and returns
1441 /// its value as a ListInit, throwing an exception if the field does not exist
1442 /// or if the value is not the right type.
1443 ///
1444 ListInit *Record::getValueAsListInit(StringRef FieldName) const {
1445   const RecordVal *R = getValue(FieldName);
1446   if (R == 0 || R->getValue() == 0)
1447     throw "Record `" + getName() + "' does not have a field named `" +
1448           FieldName.str() + "'!\n";
1449
1450   if (ListInit *LI = dynamic_cast<ListInit*>(R->getValue()))
1451     return LI;
1452   throw "Record `" + getName() + "', field `" + FieldName.str() +
1453         "' does not have a list initializer!";
1454 }
1455
1456 /// getValueAsListOfDefs - This method looks up the specified field and returns
1457 /// its value as a vector of records, throwing an exception if the field does
1458 /// not exist or if the value is not the right type.
1459 ///
1460 std::vector<Record*>
1461 Record::getValueAsListOfDefs(StringRef FieldName) const {
1462   ListInit *List = getValueAsListInit(FieldName);
1463   std::vector<Record*> Defs;
1464   for (unsigned i = 0; i < List->getSize(); i++) {
1465     if (DefInit *DI = dynamic_cast<DefInit*>(List->getElement(i))) {
1466       Defs.push_back(DI->getDef());
1467     } else {
1468       throw "Record `" + getName() + "', field `" + FieldName.str() +
1469             "' list is not entirely DefInit!";
1470     }
1471   }
1472   return Defs;
1473 }
1474
1475 /// getValueAsInt - This method looks up the specified field and returns its
1476 /// value as an int64_t, throwing an exception if the field does not exist or if
1477 /// the value is not the right type.
1478 ///
1479 int64_t Record::getValueAsInt(StringRef FieldName) const {
1480   const RecordVal *R = getValue(FieldName);
1481   if (R == 0 || R->getValue() == 0)
1482     throw "Record `" + getName() + "' does not have a field named `" +
1483           FieldName.str() + "'!\n";
1484
1485   if (IntInit *II = dynamic_cast<IntInit*>(R->getValue()))
1486     return II->getValue();
1487   throw "Record `" + getName() + "', field `" + FieldName.str() +
1488         "' does not have an int initializer!";
1489 }
1490
1491 /// getValueAsListOfInts - This method looks up the specified field and returns
1492 /// its value as a vector of integers, throwing an exception if the field does
1493 /// not exist or if the value is not the right type.
1494 ///
1495 std::vector<int64_t>
1496 Record::getValueAsListOfInts(StringRef FieldName) const {
1497   ListInit *List = getValueAsListInit(FieldName);
1498   std::vector<int64_t> Ints;
1499   for (unsigned i = 0; i < List->getSize(); i++) {
1500     if (IntInit *II = dynamic_cast<IntInit*>(List->getElement(i))) {
1501       Ints.push_back(II->getValue());
1502     } else {
1503       throw "Record `" + getName() + "', field `" + FieldName.str() +
1504             "' does not have a list of ints initializer!";
1505     }
1506   }
1507   return Ints;
1508 }
1509
1510 /// getValueAsDef - This method looks up the specified field and returns its
1511 /// value as a Record, throwing an exception if the field does not exist or if
1512 /// the value is not the right type.
1513 ///
1514 Record *Record::getValueAsDef(StringRef FieldName) const {
1515   const RecordVal *R = getValue(FieldName);
1516   if (R == 0 || R->getValue() == 0)
1517     throw "Record `" + getName() + "' does not have a field named `" +
1518       FieldName.str() + "'!\n";
1519
1520   if (DefInit *DI = dynamic_cast<DefInit*>(R->getValue()))
1521     return DI->getDef();
1522   throw "Record `" + getName() + "', field `" + FieldName.str() +
1523         "' does not have a def initializer!";
1524 }
1525
1526 /// getValueAsBit - This method looks up the specified field and returns its
1527 /// value as a bit, throwing an exception if the field does not exist or if
1528 /// the value is not the right type.
1529 ///
1530 bool Record::getValueAsBit(StringRef FieldName) const {
1531   const RecordVal *R = getValue(FieldName);
1532   if (R == 0 || R->getValue() == 0)
1533     throw "Record `" + getName() + "' does not have a field named `" +
1534       FieldName.str() + "'!\n";
1535
1536   if (BitInit *BI = dynamic_cast<BitInit*>(R->getValue()))
1537     return BI->getValue();
1538   throw "Record `" + getName() + "', field `" + FieldName.str() +
1539         "' does not have a bit initializer!";
1540 }
1541
1542 /// getValueAsDag - This method looks up the specified field and returns its
1543 /// value as an Dag, throwing an exception if the field does not exist or if
1544 /// the value is not the right type.
1545 ///
1546 DagInit *Record::getValueAsDag(StringRef FieldName) const {
1547   const RecordVal *R = getValue(FieldName);
1548   if (R == 0 || R->getValue() == 0)
1549     throw "Record `" + getName() + "' does not have a field named `" +
1550       FieldName.str() + "'!\n";
1551
1552   if (DagInit *DI = dynamic_cast<DagInit*>(R->getValue()))
1553     return DI;
1554   throw "Record `" + getName() + "', field `" + FieldName.str() +
1555         "' does not have a dag initializer!";
1556 }
1557
1558 std::string Record::getValueAsCode(StringRef FieldName) const {
1559   const RecordVal *R = getValue(FieldName);
1560   if (R == 0 || R->getValue() == 0)
1561     throw "Record `" + getName() + "' does not have a field named `" +
1562       FieldName.str() + "'!\n";
1563
1564   if (const CodeInit *CI = dynamic_cast<const CodeInit*>(R->getValue()))
1565     return CI->getValue();
1566   throw "Record `" + getName() + "', field `" + FieldName.str() +
1567     "' does not have a code initializer!";
1568 }
1569
1570
1571 void MultiClass::dump() const {
1572   errs() << "Record:\n";
1573   Rec.dump();
1574
1575   errs() << "Defs:\n";
1576   for (RecordVector::const_iterator r = DefPrototypes.begin(),
1577          rend = DefPrototypes.end();
1578        r != rend;
1579        ++r) {
1580     (*r)->dump();
1581   }
1582 }
1583
1584
1585 void RecordKeeper::dump() const { errs() << *this; }
1586
1587 raw_ostream &llvm::operator<<(raw_ostream &OS, const RecordKeeper &RK) {
1588   OS << "------------- Classes -----------------\n";
1589   const std::map<std::string, Record*> &Classes = RK.getClasses();
1590   for (std::map<std::string, Record*>::const_iterator I = Classes.begin(),
1591          E = Classes.end(); I != E; ++I)
1592     OS << "class " << *I->second;
1593
1594   OS << "------------- Defs -----------------\n";
1595   const std::map<std::string, Record*> &Defs = RK.getDefs();
1596   for (std::map<std::string, Record*>::const_iterator I = Defs.begin(),
1597          E = Defs.end(); I != E; ++I)
1598     OS << "def " << *I->second;
1599   return OS;
1600 }
1601
1602
1603 /// getAllDerivedDefinitions - This method returns all concrete definitions
1604 /// that derive from the specified class name.  If a class with the specified
1605 /// name does not exist, an error is printed and true is returned.
1606 std::vector<Record*>
1607 RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
1608   Record *Class = Records.getClass(ClassName);
1609   if (!Class)
1610     throw "ERROR: Couldn't find the `" + ClassName + "' class!\n";
1611
1612   std::vector<Record*> Defs;
1613   for (std::map<std::string, Record*>::const_iterator I = getDefs().begin(),
1614          E = getDefs().end(); I != E; ++I)
1615     if (I->second->isSubClassOf(Class))
1616       Defs.push_back(I->second);
1617
1618   return Defs;
1619 }
1620