utils/TableGen/Record.cpp

   1 //===- Record.cpp - Record implementation ---------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "Record.h"
  14 #include "llvm/Support/DataTypes.h"
  15 using namespace llvm;
  16
  17 //===----------------------------------------------------------------------===//
  18 //    Type implementations
  19 //===----------------------------------------------------------------------===//
  20
  21 void RecTy::dump() const { print(std::cerr); }
  22
  23 Init *BitRecTy::convertValue(BitsInit *BI) {
  24   if (BI->getNumBits() != 1) return 0; // Only accept if just one bit!
  25   return BI->getBit(0);
  26 }
  27
  28 bool BitRecTy::baseClassOf(const BitsRecTy *RHS) const {
  29   return RHS->getNumBits() == 1;
  30 }
  31
  32 Init *BitRecTy::convertValue(IntInit *II) {
  33   int Val = II->getValue();
  34   if (Val != 0 && Val != 1) return 0;  // Only accept 0 or 1 for a bit!
  35
  36   return new BitInit(Val != 0);
  37 }
  38
  39 Init *BitRecTy::convertValue(TypedInit *VI) {
  40   if (dynamic_cast<BitRecTy*>(VI->getType()))
  41     return VI;  // Accept variable if it is already of bit type!
  42   return 0;
  43 }
  44
  45 Init *BitsRecTy::convertValue(UnsetInit *UI) {
  46   BitsInit *Ret = new BitsInit(Size);
  47
  48   for (unsigned i = 0; i != Size; ++i)
  49     Ret->setBit(i, new UnsetInit());
  50   return Ret;
  51 }
  52
  53 Init *BitsRecTy::convertValue(BitInit *UI) {
  54   if (Size != 1) return 0;  // Can only convert single bit...
  55   BitsInit *Ret = new BitsInit(1);
  56   Ret->setBit(0, UI);
  57   return Ret;
  58 }
  59
  60 // convertValue from Int initializer to bits type: Split the integer up into the
  61 // appropriate bits...
  62 //
  63 Init *BitsRecTy::convertValue(IntInit *II) {
  64   int64_t Value = II->getValue();
  65   // Make sure this bitfield is large enough to hold the integer value...
  66   if (Value >= 0) {
  67     if (Value & ~((1LL << Size)-1))
  68       return 0;
  69   } else {
  70     if ((Value >> Size) != -1 || ((Value & (1 << (Size-1))) == 0))
  71       return 0;
  72   }
  73
  74   BitsInit *Ret = new BitsInit(Size);
  75   for (unsigned i = 0; i != Size; ++i)
  76     Ret->setBit(i, new BitInit(Value & (1 << i)));
  77
  78   return Ret;
  79 }
  80
  81 Init *BitsRecTy::convertValue(BitsInit *BI) {
  82   // If the number of bits is right, return it.  Otherwise we need to expand or
  83   // truncate...
  84   if (BI->getNumBits() == Size) return BI;
  85   return 0;
  86 }
  87
  88 Init *BitsRecTy::convertValue(TypedInit *VI) {
  89   if (BitsRecTy *BRT = dynamic_cast<BitsRecTy*>(VI->getType()))
  90     if (BRT->Size == Size) {
  91       BitsInit *Ret = new BitsInit(Size);
  92       for (unsigned i = 0; i != Size; ++i)
  93         Ret->setBit(i, new VarBitInit(VI, i));
  94       return Ret;
  95     }
  96   if (Size == 1 && dynamic_cast<BitRecTy*>(VI->getType())) {
  97     BitsInit *Ret = new BitsInit(1);
  98     Ret->setBit(0, VI);
  99     return Ret;
 100   }
 101
 102   return 0;
 103 }
 104
 105 Init *IntRecTy::convertValue(BitInit *BI) {
 106   return new IntInit(BI->getValue());
 107 }
 108
 109 Init *IntRecTy::convertValue(BitsInit *BI) {
 110   int Result = 0;
 111   for (unsigned i = 0, e = BI->getNumBits(); i != e; ++i)
 112     if (BitInit *Bit = dynamic_cast<BitInit*>(BI->getBit(i))) {
 113       Result |= Bit->getValue() << i;
 114     } else {
 115       return 0;
 116     }
 117   return new IntInit(Result);
 118 }
 119
 120 Init *IntRecTy::convertValue(TypedInit *TI) {
 121   if (TI->getType()->typeIsConvertibleTo(this))
 122     return TI;  // Accept variable if already of the right type!
 123   return 0;
 124 }
 125
 126 Init *StringRecTy::convertValue(TypedInit *TI) {
 127   if (dynamic_cast<StringRecTy*>(TI->getType()))
 128     return TI;  // Accept variable if already of the right type!
 129   return 0;
 130 }
 131
 132 void ListRecTy::print(std::ostream &OS) const {
 133   OS << "list<" << *Ty << ">";
 134 }
 135
 136 Init *ListRecTy::convertValue(ListInit *LI) {
 137   std::vector<Init*> Elements;
 138
 139   // Verify that all of the elements of the list are subclasses of the
 140   // appropriate class!
 141   for (unsigned i = 0, e = LI->getSize(); i != e; ++i)
 142     if (Init *CI = LI->getElement(i)->convertInitializerTo(Ty))
 143       Elements.push_back(CI);
 144     else
 145       return 0;
 146
 147   return new ListInit(Elements);
 148 }
 149
 150 Init *ListRecTy::convertValue(TypedInit *TI) {
 151   // Ensure that TI is compatible with our class.
 152   if (ListRecTy *LRT = dynamic_cast<ListRecTy*>(TI->getType()))
 153     if (LRT->getElementType()->typeIsConvertibleTo(getElementType()))
 154       return TI;
 155   return 0;
 156 }
 157
 158 Init *CodeRecTy::convertValue(TypedInit *TI) {
 159   if (TI->getType()->typeIsConvertibleTo(this))
 160     return TI;
 161   return 0;
 162 }
 163
 164 Init *DagRecTy::convertValue(TypedInit *TI) {
 165   if (TI->getType()->typeIsConvertibleTo(this))
 166     return TI;
 167   return 0;
 168 }
 169
 170
 171 void RecordRecTy::print(std::ostream &OS) const {
 172   OS << Rec->getName();
 173 }
 174
 175 Init *RecordRecTy::convertValue(DefInit *DI) {
 176   // Ensure that DI is a subclass of Rec.
 177   if (!DI->getDef()->isSubClassOf(Rec))
 178     return 0;
 179   return DI;
 180 }
 181
 182 Init *RecordRecTy::convertValue(TypedInit *TI) {
 183   // Ensure that TI is compatible with Rec.
 184   if (RecordRecTy *RRT = dynamic_cast<RecordRecTy*>(TI->getType()))
 185     if (RRT->getRecord()->isSubClassOf(getRecord()) ||
 186         RRT->getRecord() == getRecord())
 187       return TI;
 188   return 0;
 189 }
 190
 191 bool RecordRecTy::baseClassOf(const RecordRecTy *RHS) const {
 192   return Rec == RHS->getRecord() || RHS->getRecord()->isSubClassOf(Rec);
 193 }
 194
 195
 196 //===----------------------------------------------------------------------===//
 197 //    Initializer implementations
 198 //===----------------------------------------------------------------------===//
 199
 200 void Init::dump() const { return print(std::cerr); }
 201
 202 Init *BitsInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
 203   BitsInit *BI = new BitsInit(Bits.size());
 204   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 205     if (Bits[i] >= getNumBits()) {
 206       delete BI;
 207       return 0;
 208     }
 209     BI->setBit(i, getBit(Bits[i]));
 210   }
 211   return BI;
 212 }
 213
 214 void BitsInit::print(std::ostream &OS) const {
 215   //if (!printInHex(OS)) return;
 216   //if (!printAsVariable(OS)) return;
 217   //if (!printAsUnset(OS)) return;
 218
 219   OS << "{ ";
 220   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 221     if (i) OS << ", ";
 222     if (Init *Bit = getBit(e-i-1))
 223       Bit->print(OS);
 224     else
 225       OS << "*";
 226   }
 227   OS << " }";
 228 }
 229
 230 bool BitsInit::printInHex(std::ostream &OS) const {
 231   // First, attempt to convert the value into an integer value...
 232   int Result = 0;
 233   for (unsigned i = 0, e = getNumBits(); i != e; ++i)
 234     if (BitInit *Bit = dynamic_cast<BitInit*>(getBit(i))) {
 235       Result |= Bit->getValue() << i;
 236     } else {
 237       return true;
 238     }
 239
 240   OS << "0x" << std::hex << Result << std::dec;
 241   return false;
 242 }
 243
 244 bool BitsInit::printAsVariable(std::ostream &OS) const {
 245   // Get the variable that we may be set equal to...
 246   assert(getNumBits() != 0);
 247   VarBitInit *FirstBit = dynamic_cast<VarBitInit*>(getBit(0));
 248   if (FirstBit == 0) return true;
 249   TypedInit *Var = FirstBit->getVariable();
 250
 251   // Check to make sure the types are compatible.
 252   BitsRecTy *Ty = dynamic_cast<BitsRecTy*>(FirstBit->getVariable()->getType());
 253   if (Ty == 0) return true;
 254   if (Ty->getNumBits() != getNumBits()) return true; // Incompatible types!
 255
 256   // Check to make sure all bits are referring to the right bits in the variable
 257   for (unsigned i = 0, e = getNumBits(); i != e; ++i) {
 258     VarBitInit *Bit = dynamic_cast<VarBitInit*>(getBit(i));
 259     if (Bit == 0 || Bit->getVariable() != Var || Bit->getBitNum() != i)
 260       return true;
 261   }
 262
 263   Var->print(OS);
 264   return false;
 265 }
 266
 267 bool BitsInit::printAsUnset(std::ostream &OS) const {
 268   for (unsigned i = 0, e = getNumBits(); i != e; ++i)
 269     if (!dynamic_cast<UnsetInit*>(getBit(i)))
 270       return true;
 271   OS << "?";
 272   return false;
 273 }
 274
 275 // resolveReferences - If there are any field references that refer to fields
 276 // that have been filled in, we can propagate the values now.
 277 //
 278 Init *BitsInit::resolveReferences(Record &R, const RecordVal *RV) {
 279   bool Changed = false;
 280   BitsInit *New = new BitsInit(getNumBits());
 281
 282   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 283     Init *B;
 284     Init *CurBit = getBit(i);
 285
 286     do {
 287       B = CurBit;
 288       CurBit = CurBit->resolveReferences(R, RV);
 289       Changed |= B != CurBit;
 290     } while (B != CurBit);
 291     New->setBit(i, CurBit);
 292   }
 293
 294   if (Changed)
 295     return New;
 296   delete New;
 297   return this;
 298 }
 299
 300 Init *IntInit::getBinaryOp(BinaryOp Op, Init *RHS) {
 301   IntInit *RHSi = dynamic_cast<IntInit*>(RHS);
 302   if (RHSi == 0) return 0;
 303
 304   int NewValue;
 305   switch (Op) {
 306   default: assert(0 && "Unknown binop");
 307   case SHL: NewValue = Value << RHSi->getValue(); break;
 308   case SRA: NewValue = Value >> RHSi->getValue(); break;
 309   case SRL: NewValue = (unsigned)Value >> (unsigned)RHSi->getValue(); break;
 310   }
 311   return new IntInit(NewValue);
 312 }
 313
 314
 315 Init *IntInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
 316   BitsInit *BI = new BitsInit(Bits.size());
 317
 318   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 319     if (Bits[i] >= 32) {
 320       delete BI;
 321       return 0;
 322     }
 323     BI->setBit(i, new BitInit(Value & (1 << Bits[i])));
 324   }
 325   return BI;
 326 }
 327
 328 Init *ListInit::convertInitListSlice(const std::vector<unsigned> &Elements) {
 329   std::vector<Init*> Vals;
 330   for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
 331     if (Elements[i] >= getSize())
 332       return 0;
 333     Vals.push_back(getElement(Elements[i]));
 334   }
 335   return new ListInit(Vals);
 336 }
 337
 338 Init *ListInit::resolveReferences(Record &R, const RecordVal *RV) {
 339   std::vector<Init*> Resolved;
 340   Resolved.reserve(getSize());
 341   bool Changed = false;
 342
 343   for (unsigned i = 0, e = getSize(); i != e; ++i) {
 344     Init *E;
 345     Init *CurElt = getElement(i);
 346
 347     do {
 348       E = CurElt;
 349       CurElt = CurElt->resolveReferences(R, RV);
 350       Changed |= E != CurElt;
 351     } while (E != CurElt);
 352     Resolved.push_back(E);
 353   }
 354
 355   if (Changed)
 356     return new ListInit(Resolved);
 357   return this;
 358 }
 359
 360 void ListInit::print(std::ostream &OS) const {
 361   OS << "[";
 362   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
 363     if (i) OS << ", ";
 364     OS << *Values[i];
 365   }
 366   OS << "]";
 367 }
 368
 369 Init *TypedInit::convertInitializerBitRange(const std::vector<unsigned> &Bits) {
 370   BitsRecTy *T = dynamic_cast<BitsRecTy*>(getType());
 371   if (T == 0) return 0;  // Cannot subscript a non-bits variable...
 372   unsigned NumBits = T->getNumBits();
 373
 374   BitsInit *BI = new BitsInit(Bits.size());
 375   for (unsigned i = 0, e = Bits.size(); i != e; ++i) {
 376     if (Bits[i] >= NumBits) {
 377       delete BI;
 378       return 0;
 379     }
 380     BI->setBit(i, new VarBitInit(this, Bits[i]));
 381   }
 382   return BI;
 383 }
 384
 385 Init *TypedInit::convertInitListSlice(const std::vector<unsigned> &Elements) {
 386   ListRecTy *T = dynamic_cast<ListRecTy*>(getType());
 387   if (T == 0) return 0;  // Cannot subscript a non-list variable...
 388
 389   if (Elements.size() == 1)
 390     return new VarListElementInit(this, Elements[0]);
 391
 392   std::vector<Init*> ListInits;
 393   ListInits.reserve(Elements.size());
 394   for (unsigned i = 0, e = Elements.size(); i != e; ++i)
 395     ListInits.push_back(new VarListElementInit(this, Elements[i]));
 396   return new ListInit(ListInits);
 397 }
 398
 399
 400 Init *VarInit::resolveBitReference(Record &R, const RecordVal *IRV,
 401                                    unsigned Bit) {
 402   if (R.isTemplateArg(getName())) return 0;
 403   if (IRV && IRV->getName() != getName()) return 0;
 404
 405   RecordVal *RV = R.getValue(getName());
 406   assert(RV && "Reference to a non-existant variable?");
 407   assert(dynamic_cast<BitsInit*>(RV->getValue()));
 408   BitsInit *BI = (BitsInit*)RV->getValue();
 409
 410   assert(Bit < BI->getNumBits() && "Bit reference out of range!");
 411   Init *B = BI->getBit(Bit);
 412
 413   if (!dynamic_cast<UnsetInit*>(B))  // If the bit is not set...
 414     return B;                        // Replace the VarBitInit with it.
 415   return 0;
 416 }
 417
 418 Init *VarInit::resolveListElementReference(Record &R, const RecordVal *IRV,
 419                                            unsigned Elt) {
 420   if (R.isTemplateArg(getName())) return 0;
 421   if (IRV && IRV->getName() != getName()) return 0;
 422
 423   RecordVal *RV = R.getValue(getName());
 424   assert(RV && "Reference to a non-existant variable?");
 425   ListInit *LI = dynamic_cast<ListInit*>(RV->getValue());
 426   assert(LI && "Invalid list element!");
 427
 428   if (Elt >= LI->getSize())
 429     return 0;  // Out of range reference.
 430   Init *E = LI->getElement(Elt);
 431   if (!dynamic_cast<UnsetInit*>(E))  // If the element is set
 432     return E;                        // Replace the VarListElementInit with it.
 433   return 0;
 434 }
 435
 436
 437 RecTy *VarInit::getFieldType(const std::string &FieldName) const {
 438   if (RecordRecTy *RTy = dynamic_cast<RecordRecTy*>(getType()))
 439     if (const RecordVal *RV = RTy->getRecord()->getValue(FieldName))
 440       return RV->getType();
 441   return 0;
 442 }
 443
 444 Init *VarInit::getFieldInit(Record &R, const std::string &FieldName) const {
 445   if (RecordRecTy *RTy = dynamic_cast<RecordRecTy*>(getType()))
 446     if (const RecordVal *RV = R.getValue(VarName)) {
 447       Init *TheInit = RV->getValue();
 448       assert(TheInit != this && "Infinite loop detected!");
 449       if (Init *I = TheInit->getFieldInit(R, FieldName))
 450         return I;
 451       else
 452         return 0;
 453     }
 454   return 0;
 455 }
 456
 457 /// resolveReferences - This method is used by classes that refer to other
 458 /// variables which may not be defined at the time they expression is formed.
 459 /// If a value is set for the variable later, this method will be called on
 460 /// users of the value to allow the value to propagate out.
 461 ///
 462 Init *VarInit::resolveReferences(Record &R, const RecordVal *RV) {
 463   if (RecordVal *Val = R.getValue(VarName))
 464     if (RV == Val || (RV == 0 && !dynamic_cast<UnsetInit*>(Val->getValue())))
 465       return Val->getValue();
 466   return this;
 467 }
 468
 469
 470 Init *VarBitInit::resolveReferences(Record &R, const RecordVal *RV) {
 471   if (Init *I = getVariable()->resolveBitReference(R, RV, getBitNum()))
 472     return I;
 473   return this;
 474 }
 475
 476 Init *VarListElementInit::resolveReferences(Record &R, const RecordVal *RV) {
 477   if (Init *I = getVariable()->resolveListElementReference(R, RV,
 478                                                            getElementNum()))
 479     return I;
 480   return this;
 481 }
 482
 483 Init *VarListElementInit::resolveBitReference(Record &R, const RecordVal *RV,
 484                                               unsigned Bit) {
 485   // FIXME: This should be implemented, to support references like:
 486   // bit B = AA[0]{1};
 487   return 0;
 488 }
 489
 490 Init *VarListElementInit::
 491 resolveListElementReference(Record &R, const RecordVal *RV, unsigned Elt) {
 492   // FIXME: This should be implemented, to support references like:
 493   // int B = AA[0][1];
 494   return 0;
 495 }
 496
 497 RecTy *DefInit::getFieldType(const std::string &FieldName) const {
 498   if (const RecordVal *RV = Def->getValue(FieldName))
 499     return RV->getType();
 500   return 0;
 501 }
 502
 503 Init *DefInit::getFieldInit(Record &R, const std::string &FieldName) const {
 504   return Def->getValue(FieldName)->getValue();
 505 }
 506
 507
 508 void DefInit::print(std::ostream &OS) const {
 509   OS << Def->getName();
 510 }
 511
 512 Init *FieldInit::resolveBitReference(Record &R, const RecordVal *RV,
 513                                      unsigned Bit) {
 514   if (Init *BitsVal = Rec->getFieldInit(R, FieldName))
 515     if (BitsInit *BI = dynamic_cast<BitsInit*>(BitsVal)) {
 516       assert(Bit < BI->getNumBits() && "Bit reference out of range!");
 517       Init *B = BI->getBit(Bit);
 518
 519       if (dynamic_cast<BitInit*>(B))  // If the bit is set...
 520         return B;                     // Replace the VarBitInit with it.
 521     }
 522   return 0;
 523 }
 524
 525 Init *FieldInit::resolveListElementReference(Record &R, const RecordVal *RV,
 526                                              unsigned Elt) {
 527   if (Init *ListVal = Rec->getFieldInit(R, FieldName))
 528     if (ListInit *LI = dynamic_cast<ListInit*>(ListVal)) {
 529       if (Elt >= LI->getSize()) return 0;
 530       Init *E = LI->getElement(Elt);
 531
 532       if (!dynamic_cast<UnsetInit*>(E))  // If the bit is set...
 533         return E;                  // Replace the VarListElementInit with it.
 534     }
 535   return 0;
 536 }
 537
 538 Init *FieldInit::resolveReferences(Record &R, const RecordVal *RV) {
 539   Init *NewRec = RV ? Rec->resolveReferences(R, RV) : Rec;
 540
 541   Init *BitsVal = NewRec->getFieldInit(R, FieldName);
 542   if (BitsVal) {
 543     Init *BVR = BitsVal->resolveReferences(R, RV);
 544     return BVR->isComplete() ? BVR : this;
 545   }
 546
 547   if (NewRec != Rec) {
 548     dump();
 549     NewRec->dump(); std::cerr << "\n";
 550     return new FieldInit(NewRec, FieldName);
 551   }
 552   return this;
 553 }
 554
 555
 556 void DagInit::print(std::ostream &OS) const {
 557   OS << "(" << NodeTypeDef->getName();
 558   if (Args.size()) {
 559     OS << " " << *Args[0];
 560     if (!ArgNames[0].empty()) OS << ":$" << ArgNames[0];
 561     for (unsigned i = 1, e = Args.size(); i != e; ++i) {
 562       OS << ", " << *Args[i];
 563       if (!ArgNames[i].empty()) OS << ":$" << ArgNames[i];
 564     }
 565   }
 566   OS << ")";
 567 }
 568
 569
 570 //===----------------------------------------------------------------------===//
 571 //    Other implementations
 572 //===----------------------------------------------------------------------===//
 573
 574 RecordVal::RecordVal(const std::string &N, RecTy *T, unsigned P)
 575   : Name(N), Ty(T), Prefix(P) {
 576   Value = Ty->convertValue(new UnsetInit());
 577   assert(Value && "Cannot create unset value for current type!");
 578 }
 579
 580 void RecordVal::dump() const { std::cerr << *this; }
 581
 582 void RecordVal::print(std::ostream &OS, bool PrintSem) const {
 583   if (getPrefix()) OS << "field ";
 584   OS << *getType() << " " << getName();
 585
 586   if (getValue())
 587     OS << " = " << *getValue();
 588
 589   if (PrintSem) OS << ";\n";
 590 }
 591
 592 /// resolveReferencesTo - If anything in this record refers to RV, replace the
 593 /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
 594 /// references.
 595 void Record::resolveReferencesTo(const RecordVal *RV) {
 596   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
 597     if (Init *V = Values[i].getValue())
 598       Values[i].setValue(V->resolveReferences(*this, RV));
 599   }
 600 }
 601
 602
 603 void Record::dump() const { std::cerr << *this; }
 604
 605 std::ostream &llvm::operator<<(std::ostream &OS, const Record &R) {
 606   OS << R.getName();
 607
 608   const std::vector<std::string> &TArgs = R.getTemplateArgs();
 609   if (!TArgs.empty()) {
 610     OS << "<";
 611     for (unsigned i = 0, e = TArgs.size(); i != e; ++i) {
 612       if (i) OS << ", ";
 613       const RecordVal *RV = R.getValue(TArgs[i]);
 614       assert(RV && "Template argument record not found??");
 615       RV->print(OS, false);
 616     }
 617     OS << ">";
 618   }
 619
 620   OS << " {";
 621   const std::vector<Record*> &SC = R.getSuperClasses();
 622   if (!SC.empty()) {
 623     OS << "\t//";
 624     for (unsigned i = 0, e = SC.size(); i != e; ++i)
 625       OS << " " << SC[i]->getName();
 626   }
 627   OS << "\n";
 628
 629   const std::vector<RecordVal> &Vals = R.getValues();
 630   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
 631     if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
 632       OS << Vals[i];
 633   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
 634     if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
 635       OS << Vals[i];
 636
 637   return OS << "}\n";
 638 }
 639
 640 /// getValueInit - Return the initializer for a value with the specified name,
 641 /// or throw an exception if the field does not exist.
 642 ///
 643 Init *Record::getValueInit(const std::string &FieldName) const {
 644   const RecordVal *R = getValue(FieldName);
 645   if (R == 0 || R->getValue() == 0)
 646     throw "Record `" + getName() + "' does not have a field named `" +
 647       FieldName + "'!\n";
 648   return R->getValue();
 649 }
 650
 651
 652 /// getValueAsString - This method looks up the specified field and returns its
 653 /// value as a string, throwing an exception if the field does not exist or if
 654 /// the value is not a string.
 655 ///
 656 std::string Record::getValueAsString(const std::string &FieldName) const {
 657   const RecordVal *R = getValue(FieldName);
 658   if (R == 0 || R->getValue() == 0)
 659     throw "Record `" + getName() + "' does not have a field named `" +
 660           FieldName + "'!\n";
 661
 662   if (const StringInit *SI = dynamic_cast<const StringInit*>(R->getValue()))
 663     return SI->getValue();
 664   throw "Record `" + getName() + "', field `" + FieldName +
 665         "' does not have a string initializer!";
 666 }
 667
 668 /// getValueAsBitsInit - This method looks up the specified field and returns
 669 /// its value as a BitsInit, throwing an exception if the field does not exist
 670 /// or if the value is not the right type.
 671 ///
 672 BitsInit *Record::getValueAsBitsInit(const std::string &FieldName) const {
 673   const RecordVal *R = getValue(FieldName);
 674   if (R == 0 || R->getValue() == 0)
 675     throw "Record `" + getName() + "' does not have a field named `" +
 676           FieldName + "'!\n";
 677
 678   if (BitsInit *BI = dynamic_cast<BitsInit*>(R->getValue()))
 679     return BI;
 680   throw "Record `" + getName() + "', field `" + FieldName +
 681         "' does not have a BitsInit initializer!";
 682 }
 683
 684 /// getValueAsListInit - This method looks up the specified field and returns
 685 /// its value as a ListInit, throwing an exception if the field does not exist
 686 /// or if the value is not the right type.
 687 ///
 688 ListInit *Record::getValueAsListInit(const std::string &FieldName) const {
 689   const RecordVal *R = getValue(FieldName);
 690   if (R == 0 || R->getValue() == 0)
 691     throw "Record `" + getName() + "' does not have a field named `" +
 692           FieldName + "'!\n";
 693
 694   if (ListInit *LI = dynamic_cast<ListInit*>(R->getValue()))
 695     return LI;
 696   throw "Record `" + getName() + "', field `" + FieldName +
 697         "' does not have a list initializer!";
 698 }
 699
 700 /// getValueAsInt - This method looks up the specified field and returns its
 701 /// value as an int, throwing an exception if the field does not exist or if
 702 /// the value is not the right type.
 703 ///
 704 int Record::getValueAsInt(const std::string &FieldName) const {
 705   const RecordVal *R = getValue(FieldName);
 706   if (R == 0 || R->getValue() == 0)
 707     throw "Record `" + getName() + "' does not have a field named `" +
 708           FieldName + "'!\n";
 709
 710   if (IntInit *II = dynamic_cast<IntInit*>(R->getValue()))
 711     return II->getValue();
 712   throw "Record `" + getName() + "', field `" + FieldName +
 713         "' does not have a list initializer!";
 714 }
 715
 716 /// getValueAsDef - This method looks up the specified field and returns its
 717 /// value as a Record, throwing an exception if the field does not exist or if
 718 /// the value is not the right type.
 719 ///
 720 Record *Record::getValueAsDef(const std::string &FieldName) const {
 721   const RecordVal *R = getValue(FieldName);
 722   if (R == 0 || R->getValue() == 0)
 723     throw "Record `" + getName() + "' does not have a field named `" +
 724       FieldName + "'!\n";
 725
 726   if (DefInit *DI = dynamic_cast<DefInit*>(R->getValue()))
 727     return DI->getDef();
 728   throw "Record `" + getName() + "', field `" + FieldName +
 729         "' does not have a list initializer!";
 730 }
 731
 732 /// getValueAsBit - This method looks up the specified field and returns its
 733 /// value as a bit, throwing an exception if the field does not exist or if
 734 /// the value is not the right type.
 735 ///
 736 bool Record::getValueAsBit(const std::string &FieldName) const {
 737   const RecordVal *R = getValue(FieldName);
 738   if (R == 0 || R->getValue() == 0)
 739     throw "Record `" + getName() + "' does not have a field named `" +
 740       FieldName + "'!\n";
 741
 742   if (BitInit *BI = dynamic_cast<BitInit*>(R->getValue()))
 743     return BI->getValue();
 744   throw "Record `" + getName() + "', field `" + FieldName +
 745         "' does not have a bit initializer!";
 746 }
 747
 748 /// getValueAsDag - This method looks up the specified field and returns its
 749 /// value as an Dag, throwing an exception if the field does not exist or if
 750 /// the value is not the right type.
 751 ///
 752 DagInit *Record::getValueAsDag(const std::string &FieldName) const {
 753   const RecordVal *R = getValue(FieldName);
 754   if (R == 0 || R->getValue() == 0)
 755     throw "Record `" + getName() + "' does not have a field named `" +
 756       FieldName + "'!\n";
 757
 758   if (DagInit *DI = dynamic_cast<DagInit*>(R->getValue()))
 759     return DI;
 760   throw "Record `" + getName() + "', field `" + FieldName +
 761         "' does not have a dag initializer!";
 762 }
 763
 764
 765 void RecordKeeper::dump() const { std::cerr << *this; }
 766
 767 std::ostream &llvm::operator<<(std::ostream &OS, const RecordKeeper &RK) {
 768   OS << "------------- Classes -----------------\n";
 769   const std::map<std::string, Record*> &Classes = RK.getClasses();
 770   for (std::map<std::string, Record*>::const_iterator I = Classes.begin(),
 771          E = Classes.end(); I != E; ++I)
 772     OS << "class " << *I->second;
 773
 774   OS << "------------- Defs -----------------\n";
 775   const std::map<std::string, Record*> &Defs = RK.getDefs();
 776   for (std::map<std::string, Record*>::const_iterator I = Defs.begin(),
 777          E = Defs.end(); I != E; ++I)
 778     OS << "def " << *I->second;
 779   return OS;
 780 }
 781
 782
 783 /// getAllDerivedDefinitions - This method returns all concrete definitions
 784 /// that derive from the specified class name.  If a class with the specified
 785 /// name does not exist, an error is printed and true is returned.
 786 std::vector<Record*>
 787 RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
 788   Record *Class = Records.getClass(ClassName);
 789   if (!Class)
 790     throw "ERROR: Couldn't find the `" + ClassName + "' class!\n";
 791
 792   std::vector<Record*> Defs;
 793   for (std::map<std::string, Record*>::const_iterator I = getDefs().begin(),
 794          E = getDefs().end(); I != E; ++I)
 795     if (I->second->isSubClassOf(Class))
 796       Defs.push_back(I->second);
 797
 798   return Defs;
 799 }
 800