Add a new Record::getValueAsCode method to mirror the other getValueAs*
[oota-llvm.git] / 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 void Record::setName(const std::string &Name) {
593   if (Records.getDef(getName()) == this) {
594     Records.removeDef(getName());
595     this->Name = Name;
596     Records.addDef(this);
597   } else {
598     Records.removeClass(getName());
599     this->Name = Name;
600     Records.addClass(this);
601   }
602 }
603
604 /// resolveReferencesTo - If anything in this record refers to RV, replace the
605 /// reference to RV with the RHS of RV.  If RV is null, we resolve all possible
606 /// references.
607 void Record::resolveReferencesTo(const RecordVal *RV) {
608   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
609     if (Init *V = Values[i].getValue())
610       Values[i].setValue(V->resolveReferences(*this, RV));
611   }
612 }
613
614
615 void Record::dump() const { std::cerr << *this; }
616
617 std::ostream &llvm::operator<<(std::ostream &OS, const Record &R) {
618   OS << R.getName();
619
620   const std::vector<std::string> &TArgs = R.getTemplateArgs();
621   if (!TArgs.empty()) {
622     OS << "<";
623     for (unsigned i = 0, e = TArgs.size(); i != e; ++i) {
624       if (i) OS << ", ";
625       const RecordVal *RV = R.getValue(TArgs[i]);
626       assert(RV && "Template argument record not found??");
627       RV->print(OS, false);
628     }
629     OS << ">";
630   }
631
632   OS << " {";
633   const std::vector<Record*> &SC = R.getSuperClasses();
634   if (!SC.empty()) {
635     OS << "\t//";
636     for (unsigned i = 0, e = SC.size(); i != e; ++i)
637       OS << " " << SC[i]->getName();
638   }
639   OS << "\n";
640
641   const std::vector<RecordVal> &Vals = R.getValues();
642   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
643     if (Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
644       OS << Vals[i];
645   for (unsigned i = 0, e = Vals.size(); i != e; ++i)
646     if (!Vals[i].getPrefix() && !R.isTemplateArg(Vals[i].getName()))
647       OS << Vals[i];
648
649   return OS << "}\n";
650 }
651
652 /// getValueInit - Return the initializer for a value with the specified name,
653 /// or throw an exception if the field does not exist.
654 ///
655 Init *Record::getValueInit(const std::string &FieldName) const {
656   const RecordVal *R = getValue(FieldName);
657   if (R == 0 || R->getValue() == 0)
658     throw "Record `" + getName() + "' does not have a field named `" +
659       FieldName + "'!\n";
660   return R->getValue();
661 }
662
663
664 /// getValueAsString - This method looks up the specified field and returns its
665 /// value as a string, throwing an exception if the field does not exist or if
666 /// the value is not a string.
667 ///
668 std::string Record::getValueAsString(const std::string &FieldName) const {
669   const RecordVal *R = getValue(FieldName);
670   if (R == 0 || R->getValue() == 0)
671     throw "Record `" + getName() + "' does not have a field named `" +
672           FieldName + "'!\n";
673
674   if (const StringInit *SI = dynamic_cast<const StringInit*>(R->getValue()))
675     return SI->getValue();
676   throw "Record `" + getName() + "', field `" + FieldName +
677         "' does not have a string initializer!";
678 }
679
680 /// getValueAsBitsInit - This method looks up the specified field and returns
681 /// its value as a BitsInit, throwing an exception if the field does not exist
682 /// or if the value is not the right type.
683 ///
684 BitsInit *Record::getValueAsBitsInit(const std::string &FieldName) const {
685   const RecordVal *R = getValue(FieldName);
686   if (R == 0 || R->getValue() == 0)
687     throw "Record `" + getName() + "' does not have a field named `" +
688           FieldName + "'!\n";
689
690   if (BitsInit *BI = dynamic_cast<BitsInit*>(R->getValue()))
691     return BI;
692   throw "Record `" + getName() + "', field `" + FieldName +
693         "' does not have a BitsInit initializer!";
694 }
695
696 /// getValueAsListInit - This method looks up the specified field and returns
697 /// its value as a ListInit, throwing an exception if the field does not exist
698 /// or if the value is not the right type.
699 ///
700 ListInit *Record::getValueAsListInit(const std::string &FieldName) const {
701   const RecordVal *R = getValue(FieldName);
702   if (R == 0 || R->getValue() == 0)
703     throw "Record `" + getName() + "' does not have a field named `" +
704           FieldName + "'!\n";
705
706   if (ListInit *LI = dynamic_cast<ListInit*>(R->getValue()))
707     return LI;
708   throw "Record `" + getName() + "', field `" + FieldName +
709         "' does not have a list initializer!";
710 }
711
712 /// getValueAsInt - This method looks up the specified field and returns its
713 /// value as an int, throwing an exception if the field does not exist or if
714 /// the value is not the right type.
715 ///
716 int Record::getValueAsInt(const std::string &FieldName) const {
717   const RecordVal *R = getValue(FieldName);
718   if (R == 0 || R->getValue() == 0)
719     throw "Record `" + getName() + "' does not have a field named `" +
720           FieldName + "'!\n";
721
722   if (IntInit *II = dynamic_cast<IntInit*>(R->getValue()))
723     return II->getValue();
724   throw "Record `" + getName() + "', field `" + FieldName +
725         "' does not have a list initializer!";
726 }
727
728 /// getValueAsDef - This method looks up the specified field and returns its
729 /// value as a Record, throwing an exception if the field does not exist or if
730 /// the value is not the right type.
731 ///
732 Record *Record::getValueAsDef(const std::string &FieldName) const {
733   const RecordVal *R = getValue(FieldName);
734   if (R == 0 || R->getValue() == 0)
735     throw "Record `" + getName() + "' does not have a field named `" +
736       FieldName + "'!\n";
737
738   if (DefInit *DI = dynamic_cast<DefInit*>(R->getValue()))
739     return DI->getDef();
740   throw "Record `" + getName() + "', field `" + FieldName +
741         "' does not have a list initializer!";
742 }
743
744 /// getValueAsBit - This method looks up the specified field and returns its
745 /// value as a bit, throwing an exception if the field does not exist or if
746 /// the value is not the right type.
747 ///
748 bool Record::getValueAsBit(const std::string &FieldName) const {
749   const RecordVal *R = getValue(FieldName);
750   if (R == 0 || R->getValue() == 0)
751     throw "Record `" + getName() + "' does not have a field named `" +
752       FieldName + "'!\n";
753
754   if (BitInit *BI = dynamic_cast<BitInit*>(R->getValue()))
755     return BI->getValue();
756   throw "Record `" + getName() + "', field `" + FieldName +
757         "' does not have a bit initializer!";
758 }
759
760 /// getValueAsDag - This method looks up the specified field and returns its
761 /// value as an Dag, throwing an exception if the field does not exist or if
762 /// the value is not the right type.
763 ///
764 DagInit *Record::getValueAsDag(const std::string &FieldName) const {
765   const RecordVal *R = getValue(FieldName);
766   if (R == 0 || R->getValue() == 0)
767     throw "Record `" + getName() + "' does not have a field named `" +
768       FieldName + "'!\n";
769
770   if (DagInit *DI = dynamic_cast<DagInit*>(R->getValue()))
771     return DI;
772   throw "Record `" + getName() + "', field `" + FieldName +
773         "' does not have a dag initializer!";
774 }
775
776 std::string Record::getValueAsCode(const std::string &FieldName) const {
777   const RecordVal *R = getValue(FieldName);
778   if (R == 0 || R->getValue() == 0)
779     throw "Record `" + getName() + "' does not have a field named `" +
780       FieldName + "'!\n";
781   
782   if (const CodeInit *CI = dynamic_cast<const CodeInit*>(R->getValue()))
783     return CI->getValue();
784   throw "Record `" + getName() + "', field `" + FieldName +
785     "' does not have a code initializer!";
786 }
787
788
789 void RecordKeeper::dump() const { std::cerr << *this; }
790
791 std::ostream &llvm::operator<<(std::ostream &OS, const RecordKeeper &RK) {
792   OS << "------------- Classes -----------------\n";
793   const std::map<std::string, Record*> &Classes = RK.getClasses();
794   for (std::map<std::string, Record*>::const_iterator I = Classes.begin(),
795          E = Classes.end(); I != E; ++I)
796     OS << "class " << *I->second;
797
798   OS << "------------- Defs -----------------\n";
799   const std::map<std::string, Record*> &Defs = RK.getDefs();
800   for (std::map<std::string, Record*>::const_iterator I = Defs.begin(),
801          E = Defs.end(); I != E; ++I)
802     OS << "def " << *I->second;
803   return OS;
804 }
805
806
807 /// getAllDerivedDefinitions - This method returns all concrete definitions
808 /// that derive from the specified class name.  If a class with the specified
809 /// name does not exist, an error is printed and true is returned.
810 std::vector<Record*>
811 RecordKeeper::getAllDerivedDefinitions(const std::string &ClassName) const {
812   Record *Class = Records.getClass(ClassName);
813   if (!Class)
814     throw "ERROR: Couldn't find the `" + ClassName + "' class!\n";
815
816   std::vector<Record*> Defs;
817   for (std::map<std::string, Record*>::const_iterator I = getDefs().begin(),
818          E = getDefs().end(); I != E; ++I)
819     if (I->second->isSubClassOf(Class))
820       Defs.push_back(I->second);
821
822   return Defs;
823 }
824