1 //===- TGParser.cpp - Parser for TableGen Files ---------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // Implement the Parser for TableGen. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "TGParser.h" 14 #include "llvm/ADT/None.h" 15 #include "llvm/ADT/STLExtras.h" 16 #include "llvm/ADT/SmallVector.h" 17 #include "llvm/ADT/StringExtras.h" 18 #include "llvm/ADT/Twine.h" 19 #include "llvm/Config/llvm-config.h" 20 #include "llvm/Support/Casting.h" 21 #include "llvm/Support/Compiler.h" 22 #include "llvm/Support/ErrorHandling.h" 23 #include "llvm/Support/raw_ostream.h" 24 #include "llvm/Support/SourceMgr.h" 25 #include <algorithm> 26 #include <cassert> 27 #include <cstdint> 28 29 using namespace llvm; 30 31 //===----------------------------------------------------------------------===// 32 // Support Code for the Semantic Actions. 33 //===----------------------------------------------------------------------===// 34 35 namespace llvm { 36 37 struct SubClassReference { 38 SMRange RefRange; 39 Record *Rec; 40 SmallVector<Init*, 4> TemplateArgs; 41 42 SubClassReference() : Rec(nullptr) {} 43 44 bool isInvalid() const { return Rec == nullptr; } 45 }; 46 47 struct SubMultiClassReference { 48 SMRange RefRange; 49 MultiClass *MC; 50 SmallVector<Init*, 4> TemplateArgs; 51 52 SubMultiClassReference() : MC(nullptr) {} 53 54 bool isInvalid() const { return MC == nullptr; } 55 void dump() const; 56 }; 57 58 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 59 LLVM_DUMP_METHOD void SubMultiClassReference::dump() const { 60 errs() << "Multiclass:\n"; 61 62 MC->dump(); 63 64 errs() << "Template args:\n"; 65 for (Init *TA : TemplateArgs) 66 TA->dump(); 67 } 68 #endif 69 70 } // end namespace llvm 71 72 static bool checkBitsConcrete(Record &R, const RecordVal &RV) { 73 BitsInit *BV = cast<BitsInit>(RV.getValue()); 74 for (unsigned i = 0, e = BV->getNumBits(); i != e; ++i) { 75 Init *Bit = BV->getBit(i); 76 bool IsReference = false; 77 if (auto VBI = dyn_cast<VarBitInit>(Bit)) { 78 if (auto VI = dyn_cast<VarInit>(VBI->getBitVar())) { 79 if (R.getValue(VI->getName())) 80 IsReference = true; 81 } 82 } else if (isa<VarInit>(Bit)) { 83 IsReference = true; 84 } 85 if (!(IsReference || Bit->isConcrete())) 86 return false; 87 } 88 return true; 89 } 90 91 static void checkConcrete(Record &R) { 92 for (const RecordVal &RV : R.getValues()) { 93 // HACK: Disable this check for variables declared with 'field'. This is 94 // done merely because existing targets have legitimate cases of 95 // non-concrete variables in helper defs. Ideally, we'd introduce a 96 // 'maybe' or 'optional' modifier instead of this. 97 if (RV.getPrefix()) 98 continue; 99 100 if (Init *V = RV.getValue()) { 101 bool Ok = isa<BitsInit>(V) ? checkBitsConcrete(R, RV) : V->isConcrete(); 102 if (!Ok) { 103 PrintError(R.getLoc(), 104 Twine("Initializer of '") + RV.getNameInitAsString() + 105 "' in '" + R.getNameInitAsString() + 106 "' could not be fully resolved: " + 107 RV.getValue()->getAsString()); 108 } 109 } 110 } 111 } 112 113 /// Return an Init with a qualifier prefix referring 114 /// to CurRec's name. 115 static Init *QualifyName(Record &CurRec, MultiClass *CurMultiClass, 116 Init *Name, StringRef Scoper) { 117 Init *NewName = 118 BinOpInit::getStrConcat(CurRec.getNameInit(), StringInit::get(Scoper)); 119 NewName = BinOpInit::getStrConcat(NewName, Name); 120 if (CurMultiClass && Scoper != "::") { 121 Init *Prefix = BinOpInit::getStrConcat(CurMultiClass->Rec.getNameInit(), 122 StringInit::get("::")); 123 NewName = BinOpInit::getStrConcat(Prefix, NewName); 124 } 125 126 if (BinOpInit *BinOp = dyn_cast<BinOpInit>(NewName)) 127 NewName = BinOp->Fold(&CurRec); 128 return NewName; 129 } 130 131 /// Return the qualified version of the implicit 'NAME' template argument. 132 static Init *QualifiedNameOfImplicitName(Record &Rec, 133 MultiClass *MC = nullptr) { 134 return QualifyName(Rec, MC, StringInit::get("NAME"), MC ? "::" : ":"); 135 } 136 137 static Init *QualifiedNameOfImplicitName(MultiClass *MC) { 138 return QualifiedNameOfImplicitName(MC->Rec, MC); 139 } 140 141 bool TGParser::AddValue(Record *CurRec, SMLoc Loc, const RecordVal &RV) { 142 if (!CurRec) 143 CurRec = &CurMultiClass->Rec; 144 145 if (RecordVal *ERV = CurRec->getValue(RV.getNameInit())) { 146 // The value already exists in the class, treat this as a set. 147 if (ERV->setValue(RV.getValue())) 148 return Error(Loc, "New definition of '" + RV.getName() + "' of type '" + 149 RV.getType()->getAsString() + "' is incompatible with " + 150 "previous definition of type '" + 151 ERV->getType()->getAsString() + "'"); 152 } else { 153 CurRec->addValue(RV); 154 } 155 return false; 156 } 157 158 /// SetValue - 159 /// Return true on error, false on success. 160 bool TGParser::SetValue(Record *CurRec, SMLoc Loc, Init *ValName, 161 ArrayRef<unsigned> BitList, Init *V, 162 bool AllowSelfAssignment) { 163 if (!V) return false; 164 165 if (!CurRec) CurRec = &CurMultiClass->Rec; 166 167 RecordVal *RV = CurRec->getValue(ValName); 168 if (!RV) 169 return Error(Loc, "Value '" + ValName->getAsUnquotedString() + 170 "' unknown!"); 171 172 // Do not allow assignments like 'X = X'. This will just cause infinite loops 173 // in the resolution machinery. 174 if (BitList.empty()) 175 if (VarInit *VI = dyn_cast<VarInit>(V)) 176 if (VI->getNameInit() == ValName && !AllowSelfAssignment) 177 return Error(Loc, "Recursion / self-assignment forbidden"); 178 179 // If we are assigning to a subset of the bits in the value... then we must be 180 // assigning to a field of BitsRecTy, which must have a BitsInit 181 // initializer. 182 // 183 if (!BitList.empty()) { 184 BitsInit *CurVal = dyn_cast<BitsInit>(RV->getValue()); 185 if (!CurVal) 186 return Error(Loc, "Value '" + ValName->getAsUnquotedString() + 187 "' is not a bits type"); 188 189 // Convert the incoming value to a bits type of the appropriate size... 190 Init *BI = V->getCastTo(BitsRecTy::get(BitList.size())); 191 if (!BI) 192 return Error(Loc, "Initializer is not compatible with bit range"); 193 194 SmallVector<Init *, 16> NewBits(CurVal->getNumBits()); 195 196 // Loop over bits, assigning values as appropriate. 197 for (unsigned i = 0, e = BitList.size(); i != e; ++i) { 198 unsigned Bit = BitList[i]; 199 if (NewBits[Bit]) 200 return Error(Loc, "Cannot set bit #" + Twine(Bit) + " of value '" + 201 ValName->getAsUnquotedString() + "' more than once"); 202 NewBits[Bit] = BI->getBit(i); 203 } 204 205 for (unsigned i = 0, e = CurVal->getNumBits(); i != e; ++i) 206 if (!NewBits[i]) 207 NewBits[i] = CurVal->getBit(i); 208 209 V = BitsInit::get(NewBits); 210 } 211 212 if (RV->setValue(V, Loc)) { 213 std::string InitType; 214 if (BitsInit *BI = dyn_cast<BitsInit>(V)) 215 InitType = (Twine("' of type bit initializer with length ") + 216 Twine(BI->getNumBits())).str(); 217 else if (TypedInit *TI = dyn_cast<TypedInit>(V)) 218 InitType = (Twine("' of type '") + TI->getType()->getAsString()).str(); 219 return Error(Loc, "Field '" + ValName->getAsUnquotedString() + 220 "' of type '" + RV->getType()->getAsString() + 221 "' is incompatible with value '" + 222 V->getAsString() + InitType + "'"); 223 } 224 return false; 225 } 226 227 /// AddSubClass - Add SubClass as a subclass to CurRec, resolving its template 228 /// args as SubClass's template arguments. 229 bool TGParser::AddSubClass(Record *CurRec, SubClassReference &SubClass) { 230 Record *SC = SubClass.Rec; 231 // Add all of the values in the subclass into the current class. 232 for (const RecordVal &Val : SC->getValues()) 233 if (AddValue(CurRec, SubClass.RefRange.Start, Val)) 234 return true; 235 236 ArrayRef<Init *> TArgs = SC->getTemplateArgs(); 237 238 // Ensure that an appropriate number of template arguments are specified. 239 if (TArgs.size() < SubClass.TemplateArgs.size()) 240 return Error(SubClass.RefRange.Start, 241 "More template args specified than expected"); 242 243 // Loop over all of the template arguments, setting them to the specified 244 // value or leaving them as the default if necessary. 245 MapResolver R(CurRec); 246 247 for (unsigned i = 0, e = TArgs.size(); i != e; ++i) { 248 if (i < SubClass.TemplateArgs.size()) { 249 // If a value is specified for this template arg, set it now. 250 if (SetValue(CurRec, SubClass.RefRange.Start, TArgs[i], 251 None, SubClass.TemplateArgs[i])) 252 return true; 253 } else if (!CurRec->getValue(TArgs[i])->getValue()->isComplete()) { 254 return Error(SubClass.RefRange.Start, 255 "Value not specified for template argument #" + 256 Twine(i) + " (" + TArgs[i]->getAsUnquotedString() + 257 ") of subclass '" + SC->getNameInitAsString() + "'!"); 258 } 259 260 R.set(TArgs[i], CurRec->getValue(TArgs[i])->getValue()); 261 262 CurRec->removeValue(TArgs[i]); 263 } 264 265 Init *Name; 266 if (CurRec->isClass()) 267 Name = 268 VarInit::get(QualifiedNameOfImplicitName(*CurRec), StringRecTy::get()); 269 else 270 Name = CurRec->getNameInit(); 271 R.set(QualifiedNameOfImplicitName(*SC), Name); 272 273 CurRec->resolveReferences(R); 274 275 // Since everything went well, we can now set the "superclass" list for the 276 // current record. 277 ArrayRef<std::pair<Record *, SMRange>> SCs = SC->getSuperClasses(); 278 for (const auto &SCPair : SCs) { 279 if (CurRec->isSubClassOf(SCPair.first)) 280 return Error(SubClass.RefRange.Start, 281 "Already subclass of '" + SCPair.first->getName() + "'!\n"); 282 CurRec->addSuperClass(SCPair.first, SCPair.second); 283 } 284 285 if (CurRec->isSubClassOf(SC)) 286 return Error(SubClass.RefRange.Start, 287 "Already subclass of '" + SC->getName() + "'!\n"); 288 CurRec->addSuperClass(SC, SubClass.RefRange); 289 return false; 290 } 291 292 bool TGParser::AddSubClass(RecordsEntry &Entry, SubClassReference &SubClass) { 293 if (Entry.Rec) 294 return AddSubClass(Entry.Rec.get(), SubClass); 295 296 for (auto &E : Entry.Loop->Entries) { 297 if (AddSubClass(E, SubClass)) 298 return true; 299 } 300 301 return false; 302 } 303 304 /// AddSubMultiClass - Add SubMultiClass as a subclass to 305 /// CurMC, resolving its template args as SubMultiClass's 306 /// template arguments. 307 bool TGParser::AddSubMultiClass(MultiClass *CurMC, 308 SubMultiClassReference &SubMultiClass) { 309 MultiClass *SMC = SubMultiClass.MC; 310 311 ArrayRef<Init *> SMCTArgs = SMC->Rec.getTemplateArgs(); 312 if (SMCTArgs.size() < SubMultiClass.TemplateArgs.size()) 313 return Error(SubMultiClass.RefRange.Start, 314 "More template args specified than expected"); 315 316 // Prepare the mapping of template argument name to value, filling in default 317 // values if necessary. 318 SubstStack TemplateArgs; 319 for (unsigned i = 0, e = SMCTArgs.size(); i != e; ++i) { 320 if (i < SubMultiClass.TemplateArgs.size()) { 321 TemplateArgs.emplace_back(SMCTArgs[i], SubMultiClass.TemplateArgs[i]); 322 } else { 323 Init *Default = SMC->Rec.getValue(SMCTArgs[i])->getValue(); 324 if (!Default->isComplete()) { 325 return Error(SubMultiClass.RefRange.Start, 326 "value not specified for template argument #" + Twine(i) + 327 " (" + SMCTArgs[i]->getAsUnquotedString() + 328 ") of multiclass '" + SMC->Rec.getNameInitAsString() + 329 "'"); 330 } 331 TemplateArgs.emplace_back(SMCTArgs[i], Default); 332 } 333 } 334 335 TemplateArgs.emplace_back( 336 QualifiedNameOfImplicitName(SMC), 337 VarInit::get(QualifiedNameOfImplicitName(CurMC), StringRecTy::get())); 338 339 // Add all of the defs in the subclass into the current multiclass. 340 return resolve(SMC->Entries, TemplateArgs, false, &CurMC->Entries); 341 } 342 343 /// Add a record or foreach loop to the current context (global record keeper, 344 /// current inner-most foreach loop, or multiclass). 345 bool TGParser::addEntry(RecordsEntry E) { 346 assert(!E.Rec || !E.Loop); 347 348 if (!Loops.empty()) { 349 Loops.back()->Entries.push_back(std::move(E)); 350 return false; 351 } 352 353 if (E.Loop) { 354 SubstStack Stack; 355 return resolve(*E.Loop, Stack, CurMultiClass == nullptr, 356 CurMultiClass ? &CurMultiClass->Entries : nullptr); 357 } 358 359 if (CurMultiClass) { 360 CurMultiClass->Entries.push_back(std::move(E)); 361 return false; 362 } 363 364 return addDefOne(std::move(E.Rec)); 365 } 366 367 /// Resolve the entries in \p Loop, going over inner loops recursively 368 /// and making the given subsitutions of (name, value) pairs. 369 /// 370 /// The resulting records are stored in \p Dest if non-null. Otherwise, they 371 /// are added to the global record keeper. 372 bool TGParser::resolve(const ForeachLoop &Loop, SubstStack &Substs, 373 bool Final, std::vector<RecordsEntry> *Dest, 374 SMLoc *Loc) { 375 MapResolver R; 376 for (const auto &S : Substs) 377 R.set(S.first, S.second); 378 Init *List = Loop.ListValue->resolveReferences(R); 379 auto LI = dyn_cast<ListInit>(List); 380 if (!LI) { 381 if (!Final) { 382 Dest->emplace_back(std::make_unique<ForeachLoop>(Loop.Loc, Loop.IterVar, 383 List)); 384 return resolve(Loop.Entries, Substs, Final, &Dest->back().Loop->Entries, 385 Loc); 386 } 387 388 PrintError(Loop.Loc, Twine("attempting to loop over '") + 389 List->getAsString() + "', expected a list"); 390 return true; 391 } 392 393 bool Error = false; 394 for (auto Elt : *LI) { 395 if (Loop.IterVar) 396 Substs.emplace_back(Loop.IterVar->getNameInit(), Elt); 397 Error = resolve(Loop.Entries, Substs, Final, Dest); 398 if (Loop.IterVar) 399 Substs.pop_back(); 400 if (Error) 401 break; 402 } 403 return Error; 404 } 405 406 /// Resolve the entries in \p Source, going over loops recursively and 407 /// making the given substitutions of (name, value) pairs. 408 /// 409 /// The resulting records are stored in \p Dest if non-null. Otherwise, they 410 /// are added to the global record keeper. 411 bool TGParser::resolve(const std::vector<RecordsEntry> &Source, 412 SubstStack &Substs, bool Final, 413 std::vector<RecordsEntry> *Dest, SMLoc *Loc) { 414 bool Error = false; 415 for (auto &E : Source) { 416 if (E.Loop) { 417 Error = resolve(*E.Loop, Substs, Final, Dest); 418 } else { 419 auto Rec = std::make_unique<Record>(*E.Rec); 420 if (Loc) 421 Rec->appendLoc(*Loc); 422 423 MapResolver R(Rec.get()); 424 for (const auto &S : Substs) 425 R.set(S.first, S.second); 426 Rec->resolveReferences(R); 427 428 if (Dest) 429 Dest->push_back(std::move(Rec)); 430 else 431 Error = addDefOne(std::move(Rec)); 432 } 433 if (Error) 434 break; 435 } 436 return Error; 437 } 438 439 /// Resolve the record fully and add it to the record keeper. 440 bool TGParser::addDefOne(std::unique_ptr<Record> Rec) { 441 if (Record *Prev = Records.getDef(Rec->getNameInitAsString())) { 442 if (!Rec->isAnonymous()) { 443 PrintError(Rec->getLoc(), 444 "def already exists: " + Rec->getNameInitAsString()); 445 PrintNote(Prev->getLoc(), "location of previous definition"); 446 return true; 447 } 448 Rec->setName(Records.getNewAnonymousName()); 449 } 450 451 Rec->resolveReferences(); 452 checkConcrete(*Rec); 453 454 if (!isa<StringInit>(Rec->getNameInit())) { 455 PrintError(Rec->getLoc(), Twine("record name '") + 456 Rec->getNameInit()->getAsString() + 457 "' could not be fully resolved"); 458 return true; 459 } 460 461 // If ObjectBody has template arguments, it's an error. 462 assert(Rec->getTemplateArgs().empty() && "How'd this get template args?"); 463 464 for (DefsetRecord *Defset : Defsets) { 465 DefInit *I = Rec->getDefInit(); 466 if (!I->getType()->typeIsA(Defset->EltTy)) { 467 PrintError(Rec->getLoc(), Twine("adding record of incompatible type '") + 468 I->getType()->getAsString() + 469 "' to defset"); 470 PrintNote(Defset->Loc, "location of defset declaration"); 471 return true; 472 } 473 Defset->Elements.push_back(I); 474 } 475 476 Records.addDef(std::move(Rec)); 477 return false; 478 } 479 480 //===----------------------------------------------------------------------===// 481 // Parser Code 482 //===----------------------------------------------------------------------===// 483 484 /// isObjectStart - Return true if this is a valid first token for an Object. 485 static bool isObjectStart(tgtok::TokKind K) { 486 return K == tgtok::Class || K == tgtok::Def || K == tgtok::Defm || 487 K == tgtok::Let || K == tgtok::MultiClass || K == tgtok::Foreach || 488 K == tgtok::Defset || K == tgtok::Defvar || K == tgtok::If; 489 } 490 491 bool TGParser::consume(tgtok::TokKind K) { 492 if (Lex.getCode() == K) { 493 Lex.Lex(); 494 return true; 495 } 496 return false; 497 } 498 499 /// ParseObjectName - If a valid object name is specified, return it. If no 500 /// name is specified, return the unset initializer. Return nullptr on parse 501 /// error. 502 /// ObjectName ::= Value [ '#' Value ]* 503 /// ObjectName ::= /*empty*/ 504 /// 505 Init *TGParser::ParseObjectName(MultiClass *CurMultiClass) { 506 switch (Lex.getCode()) { 507 case tgtok::colon: 508 case tgtok::semi: 509 case tgtok::l_brace: 510 // These are all of the tokens that can begin an object body. 511 // Some of these can also begin values but we disallow those cases 512 // because they are unlikely to be useful. 513 return UnsetInit::get(); 514 default: 515 break; 516 } 517 518 Record *CurRec = nullptr; 519 if (CurMultiClass) 520 CurRec = &CurMultiClass->Rec; 521 522 Init *Name = ParseValue(CurRec, StringRecTy::get(), ParseNameMode); 523 if (!Name) 524 return nullptr; 525 526 if (CurMultiClass) { 527 Init *NameStr = QualifiedNameOfImplicitName(CurMultiClass); 528 HasReferenceResolver R(NameStr); 529 Name->resolveReferences(R); 530 if (!R.found()) 531 Name = BinOpInit::getStrConcat(VarInit::get(NameStr, StringRecTy::get()), 532 Name); 533 } 534 535 return Name; 536 } 537 538 /// ParseClassID - Parse and resolve a reference to a class name. This returns 539 /// null on error. 540 /// 541 /// ClassID ::= ID 542 /// 543 Record *TGParser::ParseClassID() { 544 if (Lex.getCode() != tgtok::Id) { 545 TokError("expected name for ClassID"); 546 return nullptr; 547 } 548 549 Record *Result = Records.getClass(Lex.getCurStrVal()); 550 if (!Result) { 551 std::string Msg("Couldn't find class '" + Lex.getCurStrVal() + "'"); 552 if (MultiClasses[Lex.getCurStrVal()].get()) 553 TokError(Msg + ". Use 'defm' if you meant to use multiclass '" + 554 Lex.getCurStrVal() + "'"); 555 else 556 TokError(Msg); 557 } 558 559 Lex.Lex(); 560 return Result; 561 } 562 563 /// ParseMultiClassID - Parse and resolve a reference to a multiclass name. 564 /// This returns null on error. 565 /// 566 /// MultiClassID ::= ID 567 /// 568 MultiClass *TGParser::ParseMultiClassID() { 569 if (Lex.getCode() != tgtok::Id) { 570 TokError("expected name for MultiClassID"); 571 return nullptr; 572 } 573 574 MultiClass *Result = MultiClasses[Lex.getCurStrVal()].get(); 575 if (!Result) 576 TokError("Couldn't find multiclass '" + Lex.getCurStrVal() + "'"); 577 578 Lex.Lex(); 579 return Result; 580 } 581 582 /// ParseSubClassReference - Parse a reference to a subclass or to a templated 583 /// subclass. This returns a SubClassRefTy with a null Record* on error. 584 /// 585 /// SubClassRef ::= ClassID 586 /// SubClassRef ::= ClassID '<' ValueList '>' 587 /// 588 SubClassReference TGParser:: 589 ParseSubClassReference(Record *CurRec, bool isDefm) { 590 SubClassReference Result; 591 Result.RefRange.Start = Lex.getLoc(); 592 593 if (isDefm) { 594 if (MultiClass *MC = ParseMultiClassID()) 595 Result.Rec = &MC->Rec; 596 } else { 597 Result.Rec = ParseClassID(); 598 } 599 if (!Result.Rec) return Result; 600 601 // If there is no template arg list, we're done. 602 if (!consume(tgtok::less)) { 603 Result.RefRange.End = Lex.getLoc(); 604 return Result; 605 } 606 607 if (Lex.getCode() == tgtok::greater) { 608 TokError("subclass reference requires a non-empty list of template values"); 609 Result.Rec = nullptr; 610 return Result; 611 } 612 613 ParseValueList(Result.TemplateArgs, CurRec, Result.Rec); 614 if (Result.TemplateArgs.empty()) { 615 Result.Rec = nullptr; // Error parsing value list. 616 return Result; 617 } 618 619 if (!consume(tgtok::greater)) { 620 TokError("expected '>' in template value list"); 621 Result.Rec = nullptr; 622 return Result; 623 } 624 Result.RefRange.End = Lex.getLoc(); 625 626 return Result; 627 } 628 629 /// ParseSubMultiClassReference - Parse a reference to a subclass or to a 630 /// templated submulticlass. This returns a SubMultiClassRefTy with a null 631 /// Record* on error. 632 /// 633 /// SubMultiClassRef ::= MultiClassID 634 /// SubMultiClassRef ::= MultiClassID '<' ValueList '>' 635 /// 636 SubMultiClassReference TGParser:: 637 ParseSubMultiClassReference(MultiClass *CurMC) { 638 SubMultiClassReference Result; 639 Result.RefRange.Start = Lex.getLoc(); 640 641 Result.MC = ParseMultiClassID(); 642 if (!Result.MC) return Result; 643 644 // If there is no template arg list, we're done. 645 if (!consume(tgtok::less)) { 646 Result.RefRange.End = Lex.getLoc(); 647 return Result; 648 } 649 650 if (Lex.getCode() == tgtok::greater) { 651 TokError("subclass reference requires a non-empty list of template values"); 652 Result.MC = nullptr; 653 return Result; 654 } 655 656 ParseValueList(Result.TemplateArgs, &CurMC->Rec, &Result.MC->Rec); 657 if (Result.TemplateArgs.empty()) { 658 Result.MC = nullptr; // Error parsing value list. 659 return Result; 660 } 661 662 if (!consume(tgtok::greater)) { 663 TokError("expected '>' in template value list"); 664 Result.MC = nullptr; 665 return Result; 666 } 667 Result.RefRange.End = Lex.getLoc(); 668 669 return Result; 670 } 671 672 /// ParseRangePiece - Parse a bit/value range. 673 /// RangePiece ::= INTVAL 674 /// RangePiece ::= INTVAL '...' INTVAL 675 /// RangePiece ::= INTVAL '-' INTVAL 676 /// RangePiece ::= INTVAL INTVAL 677 // The last two forms are deprecated. 678 bool TGParser::ParseRangePiece(SmallVectorImpl<unsigned> &Ranges, 679 TypedInit *FirstItem) { 680 Init *CurVal = FirstItem; 681 if (!CurVal) 682 CurVal = ParseValue(nullptr); 683 684 IntInit *II = dyn_cast_or_null<IntInit>(CurVal); 685 if (!II) 686 return TokError("expected integer or bitrange"); 687 688 int64_t Start = II->getValue(); 689 int64_t End; 690 691 if (Start < 0) 692 return TokError("invalid range, cannot be negative"); 693 694 switch (Lex.getCode()) { 695 default: 696 Ranges.push_back(Start); 697 return false; 698 699 case tgtok::dotdotdot: 700 case tgtok::minus: { 701 Lex.Lex(); // eat 702 703 Init *I_End = ParseValue(nullptr); 704 IntInit *II_End = dyn_cast_or_null<IntInit>(I_End); 705 if (!II_End) { 706 TokError("expected integer value as end of range"); 707 return true; 708 } 709 710 End = II_End->getValue(); 711 break; 712 } 713 case tgtok::IntVal: { 714 End = -Lex.getCurIntVal(); 715 Lex.Lex(); 716 break; 717 } 718 } 719 if (End < 0) 720 return TokError("invalid range, cannot be negative"); 721 722 // Add to the range. 723 if (Start < End) 724 for (; Start <= End; ++Start) 725 Ranges.push_back(Start); 726 else 727 for (; Start >= End; --Start) 728 Ranges.push_back(Start); 729 return false; 730 } 731 732 /// ParseRangeList - Parse a list of scalars and ranges into scalar values. 733 /// 734 /// RangeList ::= RangePiece (',' RangePiece)* 735 /// 736 void TGParser::ParseRangeList(SmallVectorImpl<unsigned> &Result) { 737 // Parse the first piece. 738 if (ParseRangePiece(Result)) { 739 Result.clear(); 740 return; 741 } 742 while (consume(tgtok::comma)) 743 // Parse the next range piece. 744 if (ParseRangePiece(Result)) { 745 Result.clear(); 746 return; 747 } 748 } 749 750 /// ParseOptionalRangeList - Parse either a range list in <>'s or nothing. 751 /// OptionalRangeList ::= '<' RangeList '>' 752 /// OptionalRangeList ::= /*empty*/ 753 bool TGParser::ParseOptionalRangeList(SmallVectorImpl<unsigned> &Ranges) { 754 SMLoc StartLoc = Lex.getLoc(); 755 if (!consume(tgtok::less)) 756 return false; 757 758 // Parse the range list. 759 ParseRangeList(Ranges); 760 if (Ranges.empty()) return true; 761 762 if (!consume(tgtok::greater)) { 763 TokError("expected '>' at end of range list"); 764 return Error(StartLoc, "to match this '<'"); 765 } 766 return false; 767 } 768 769 /// ParseOptionalBitList - Parse either a bit list in {}'s or nothing. 770 /// OptionalBitList ::= '{' RangeList '}' 771 /// OptionalBitList ::= /*empty*/ 772 bool TGParser::ParseOptionalBitList(SmallVectorImpl<unsigned> &Ranges) { 773 SMLoc StartLoc = Lex.getLoc(); 774 if (!consume(tgtok::l_brace)) 775 return false; 776 777 // Parse the range list. 778 ParseRangeList(Ranges); 779 if (Ranges.empty()) return true; 780 781 if (!consume(tgtok::r_brace)) { 782 TokError("expected '}' at end of bit list"); 783 return Error(StartLoc, "to match this '{'"); 784 } 785 return false; 786 } 787 788 /// ParseType - Parse and return a tblgen type. This returns null on error. 789 /// 790 /// Type ::= STRING // string type 791 /// Type ::= CODE // code type 792 /// Type ::= BIT // bit type 793 /// Type ::= BITS '<' INTVAL '>' // bits<x> type 794 /// Type ::= INT // int type 795 /// Type ::= LIST '<' Type '>' // list<x> type 796 /// Type ::= DAG // dag type 797 /// Type ::= ClassID // Record Type 798 /// 799 RecTy *TGParser::ParseType() { 800 switch (Lex.getCode()) { 801 default: TokError("Unknown token when expecting a type"); return nullptr; 802 case tgtok::String: Lex.Lex(); return StringRecTy::get(); 803 case tgtok::Code: Lex.Lex(); return CodeRecTy::get(); 804 case tgtok::Bit: Lex.Lex(); return BitRecTy::get(); 805 case tgtok::Int: Lex.Lex(); return IntRecTy::get(); 806 case tgtok::Dag: Lex.Lex(); return DagRecTy::get(); 807 case tgtok::Id: 808 if (Record *R = ParseClassID()) return RecordRecTy::get(R); 809 TokError("unknown class name"); 810 return nullptr; 811 case tgtok::Bits: { 812 if (Lex.Lex() != tgtok::less) { // Eat 'bits' 813 TokError("expected '<' after bits type"); 814 return nullptr; 815 } 816 if (Lex.Lex() != tgtok::IntVal) { // Eat '<' 817 TokError("expected integer in bits<n> type"); 818 return nullptr; 819 } 820 uint64_t Val = Lex.getCurIntVal(); 821 if (Lex.Lex() != tgtok::greater) { // Eat count. 822 TokError("expected '>' at end of bits<n> type"); 823 return nullptr; 824 } 825 Lex.Lex(); // Eat '>' 826 return BitsRecTy::get(Val); 827 } 828 case tgtok::List: { 829 if (Lex.Lex() != tgtok::less) { // Eat 'bits' 830 TokError("expected '<' after list type"); 831 return nullptr; 832 } 833 Lex.Lex(); // Eat '<' 834 RecTy *SubType = ParseType(); 835 if (!SubType) return nullptr; 836 837 if (!consume(tgtok::greater)) { 838 TokError("expected '>' at end of list<ty> type"); 839 return nullptr; 840 } 841 return ListRecTy::get(SubType); 842 } 843 } 844 } 845 846 /// ParseIDValue - This is just like ParseIDValue above, but it assumes the ID 847 /// has already been read. 848 Init *TGParser::ParseIDValue(Record *CurRec, StringInit *Name, SMLoc NameLoc, 849 IDParseMode Mode) { 850 if (CurRec) { 851 if (const RecordVal *RV = CurRec->getValue(Name)) 852 return VarInit::get(Name, RV->getType()); 853 } 854 855 if ((CurRec && CurRec->isClass()) || CurMultiClass) { 856 Init *TemplateArgName; 857 if (CurMultiClass) { 858 TemplateArgName = 859 QualifyName(CurMultiClass->Rec, CurMultiClass, Name, "::"); 860 } else 861 TemplateArgName = QualifyName(*CurRec, CurMultiClass, Name, ":"); 862 863 Record *TemplateRec = CurMultiClass ? &CurMultiClass->Rec : CurRec; 864 if (TemplateRec->isTemplateArg(TemplateArgName)) { 865 const RecordVal *RV = TemplateRec->getValue(TemplateArgName); 866 assert(RV && "Template arg doesn't exist??"); 867 return VarInit::get(TemplateArgName, RV->getType()); 868 } else if (Name->getValue() == "NAME") { 869 return VarInit::get(TemplateArgName, StringRecTy::get()); 870 } 871 } 872 873 if (CurLocalScope) 874 if (Init *I = CurLocalScope->getVar(Name->getValue())) 875 return I; 876 877 // If this is in a foreach loop, make sure it's not a loop iterator 878 for (const auto &L : Loops) { 879 if (L->IterVar) { 880 VarInit *IterVar = dyn_cast<VarInit>(L->IterVar); 881 if (IterVar && IterVar->getNameInit() == Name) 882 return IterVar; 883 } 884 } 885 886 if (Mode == ParseNameMode) 887 return Name; 888 889 if (Init *I = Records.getGlobal(Name->getValue())) 890 return I; 891 892 // Allow self-references of concrete defs, but delay the lookup so that we 893 // get the correct type. 894 if (CurRec && !CurRec->isClass() && !CurMultiClass && 895 CurRec->getNameInit() == Name) 896 return UnOpInit::get(UnOpInit::CAST, Name, CurRec->getType()); 897 898 Error(NameLoc, "Variable not defined: '" + Name->getValue() + "'"); 899 return nullptr; 900 } 901 902 /// ParseOperation - Parse an operator. This returns null on error. 903 /// 904 /// Operation ::= XOperator ['<' Type '>'] '(' Args ')' 905 /// 906 Init *TGParser::ParseOperation(Record *CurRec, RecTy *ItemType) { 907 switch (Lex.getCode()) { 908 default: 909 TokError("unknown bang operator"); 910 return nullptr; 911 case tgtok::XNOT: 912 case tgtok::XHead: 913 case tgtok::XTail: 914 case tgtok::XSize: 915 case tgtok::XEmpty: 916 case tgtok::XCast: 917 case tgtok::XGetDagOp: { // Value ::= !unop '(' Value ')' 918 UnOpInit::UnaryOp Code; 919 RecTy *Type = nullptr; 920 921 switch (Lex.getCode()) { 922 default: llvm_unreachable("Unhandled code!"); 923 case tgtok::XCast: 924 Lex.Lex(); // eat the operation 925 Code = UnOpInit::CAST; 926 927 Type = ParseOperatorType(); 928 929 if (!Type) { 930 TokError("did not get type for unary operator"); 931 return nullptr; 932 } 933 934 break; 935 case tgtok::XNOT: 936 Lex.Lex(); // eat the operation 937 Code = UnOpInit::NOT; 938 Type = IntRecTy::get(); 939 break; 940 case tgtok::XHead: 941 Lex.Lex(); // eat the operation 942 Code = UnOpInit::HEAD; 943 break; 944 case tgtok::XTail: 945 Lex.Lex(); // eat the operation 946 Code = UnOpInit::TAIL; 947 break; 948 case tgtok::XSize: 949 Lex.Lex(); 950 Code = UnOpInit::SIZE; 951 Type = IntRecTy::get(); 952 break; 953 case tgtok::XEmpty: 954 Lex.Lex(); // eat the operation 955 Code = UnOpInit::EMPTY; 956 Type = IntRecTy::get(); 957 break; 958 case tgtok::XGetDagOp: 959 Lex.Lex(); // eat the operation 960 if (Lex.getCode() == tgtok::less) { 961 // Parse an optional type suffix, so that you can say 962 // !getdagop<BaseClass>(someDag) as a shorthand for 963 // !cast<BaseClass>(!getdagop(someDag)). 964 Type = ParseOperatorType(); 965 966 if (!Type) { 967 TokError("did not get type for unary operator"); 968 return nullptr; 969 } 970 971 if (!isa<RecordRecTy>(Type)) { 972 TokError("type for !getdagop must be a record type"); 973 // but keep parsing, to consume the operand 974 } 975 } else { 976 Type = RecordRecTy::get({}); 977 } 978 Code = UnOpInit::GETDAGOP; 979 break; 980 } 981 if (!consume(tgtok::l_paren)) { 982 TokError("expected '(' after unary operator"); 983 return nullptr; 984 } 985 986 Init *LHS = ParseValue(CurRec); 987 if (!LHS) return nullptr; 988 989 if (Code == UnOpInit::EMPTY || Code == UnOpInit::SIZE) { 990 ListInit *LHSl = dyn_cast<ListInit>(LHS); 991 StringInit *LHSs = dyn_cast<StringInit>(LHS); 992 DagInit *LHSd = dyn_cast<DagInit>(LHS); 993 TypedInit *LHSt = dyn_cast<TypedInit>(LHS); 994 if (!LHSl && !LHSs && !LHSd && !LHSt) { 995 TokError("expected string, list, or dag type argument in unary operator"); 996 return nullptr; 997 } 998 if (LHSt) { 999 ListRecTy *LType = dyn_cast<ListRecTy>(LHSt->getType()); 1000 StringRecTy *SType = dyn_cast<StringRecTy>(LHSt->getType()); 1001 DagRecTy *DType = dyn_cast<DagRecTy>(LHSt->getType()); 1002 if (!LType && !SType && !DType) { 1003 TokError("expected string, list, or dag type argument in unary operator"); 1004 return nullptr; 1005 } 1006 } 1007 } 1008 1009 if (Code == UnOpInit::HEAD || Code == UnOpInit::TAIL) { 1010 ListInit *LHSl = dyn_cast<ListInit>(LHS); 1011 TypedInit *LHSt = dyn_cast<TypedInit>(LHS); 1012 if (!LHSl && !LHSt) { 1013 TokError("expected list type argument in unary operator"); 1014 return nullptr; 1015 } 1016 if (LHSt) { 1017 ListRecTy *LType = dyn_cast<ListRecTy>(LHSt->getType()); 1018 if (!LType) { 1019 TokError("expected list type argument in unary operator"); 1020 return nullptr; 1021 } 1022 } 1023 1024 if (LHSl && LHSl->empty()) { 1025 TokError("empty list argument in unary operator"); 1026 return nullptr; 1027 } 1028 if (LHSl) { 1029 Init *Item = LHSl->getElement(0); 1030 TypedInit *Itemt = dyn_cast<TypedInit>(Item); 1031 if (!Itemt) { 1032 TokError("untyped list element in unary operator"); 1033 return nullptr; 1034 } 1035 Type = (Code == UnOpInit::HEAD) ? Itemt->getType() 1036 : ListRecTy::get(Itemt->getType()); 1037 } else { 1038 assert(LHSt && "expected list type argument in unary operator"); 1039 ListRecTy *LType = dyn_cast<ListRecTy>(LHSt->getType()); 1040 Type = (Code == UnOpInit::HEAD) ? LType->getElementType() : LType; 1041 } 1042 } 1043 1044 if (!consume(tgtok::r_paren)) { 1045 TokError("expected ')' in unary operator"); 1046 return nullptr; 1047 } 1048 return (UnOpInit::get(Code, LHS, Type))->Fold(CurRec); 1049 } 1050 1051 case tgtok::XIsA: { 1052 // Value ::= !isa '<' Type '>' '(' Value ')' 1053 Lex.Lex(); // eat the operation 1054 1055 RecTy *Type = ParseOperatorType(); 1056 if (!Type) 1057 return nullptr; 1058 1059 if (!consume(tgtok::l_paren)) { 1060 TokError("expected '(' after type of !isa"); 1061 return nullptr; 1062 } 1063 1064 Init *LHS = ParseValue(CurRec); 1065 if (!LHS) 1066 return nullptr; 1067 1068 if (!consume(tgtok::r_paren)) { 1069 TokError("expected ')' in !isa"); 1070 return nullptr; 1071 } 1072 1073 return (IsAOpInit::get(Type, LHS))->Fold(); 1074 } 1075 1076 case tgtok::XConcat: 1077 case tgtok::XADD: 1078 case tgtok::XSUB: 1079 case tgtok::XMUL: 1080 case tgtok::XAND: 1081 case tgtok::XOR: 1082 case tgtok::XXOR: 1083 case tgtok::XSRA: 1084 case tgtok::XSRL: 1085 case tgtok::XSHL: 1086 case tgtok::XEq: 1087 case tgtok::XNe: 1088 case tgtok::XLe: 1089 case tgtok::XLt: 1090 case tgtok::XGe: 1091 case tgtok::XGt: 1092 case tgtok::XListConcat: 1093 case tgtok::XListSplat: 1094 case tgtok::XStrConcat: 1095 case tgtok::XInterleave: 1096 case tgtok::XSetDagOp: { // Value ::= !binop '(' Value ',' Value ')' 1097 tgtok::TokKind OpTok = Lex.getCode(); 1098 SMLoc OpLoc = Lex.getLoc(); 1099 Lex.Lex(); // eat the operation 1100 1101 BinOpInit::BinaryOp Code; 1102 switch (OpTok) { 1103 default: llvm_unreachable("Unhandled code!"); 1104 case tgtok::XConcat: Code = BinOpInit::CONCAT; break; 1105 case tgtok::XADD: Code = BinOpInit::ADD; break; 1106 case tgtok::XSUB: Code = BinOpInit::SUB; break; 1107 case tgtok::XMUL: Code = BinOpInit::MUL; break; 1108 case tgtok::XAND: Code = BinOpInit::AND; break; 1109 case tgtok::XOR: Code = BinOpInit::OR; break; 1110 case tgtok::XXOR: Code = BinOpInit::XOR; break; 1111 case tgtok::XSRA: Code = BinOpInit::SRA; break; 1112 case tgtok::XSRL: Code = BinOpInit::SRL; break; 1113 case tgtok::XSHL: Code = BinOpInit::SHL; break; 1114 case tgtok::XEq: Code = BinOpInit::EQ; break; 1115 case tgtok::XNe: Code = BinOpInit::NE; break; 1116 case tgtok::XLe: Code = BinOpInit::LE; break; 1117 case tgtok::XLt: Code = BinOpInit::LT; break; 1118 case tgtok::XGe: Code = BinOpInit::GE; break; 1119 case tgtok::XGt: Code = BinOpInit::GT; break; 1120 case tgtok::XListConcat: Code = BinOpInit::LISTCONCAT; break; 1121 case tgtok::XListSplat: Code = BinOpInit::LISTSPLAT; break; 1122 case tgtok::XStrConcat: Code = BinOpInit::STRCONCAT; break; 1123 case tgtok::XInterleave: Code = BinOpInit::INTERLEAVE; break; 1124 case tgtok::XSetDagOp: Code = BinOpInit::SETDAGOP; break; 1125 } 1126 1127 RecTy *Type = nullptr; 1128 RecTy *ArgType = nullptr; 1129 switch (OpTok) { 1130 default: 1131 llvm_unreachable("Unhandled code!"); 1132 case tgtok::XConcat: 1133 case tgtok::XSetDagOp: 1134 Type = DagRecTy::get(); 1135 ArgType = DagRecTy::get(); 1136 break; 1137 case tgtok::XAND: 1138 case tgtok::XOR: 1139 case tgtok::XXOR: 1140 case tgtok::XSRA: 1141 case tgtok::XSRL: 1142 case tgtok::XSHL: 1143 case tgtok::XADD: 1144 case tgtok::XSUB: 1145 case tgtok::XMUL: 1146 Type = IntRecTy::get(); 1147 ArgType = IntRecTy::get(); 1148 break; 1149 case tgtok::XEq: 1150 case tgtok::XNe: 1151 Type = BitRecTy::get(); 1152 // ArgType for Eq / Ne is not known at this point 1153 break; 1154 case tgtok::XLe: 1155 case tgtok::XLt: 1156 case tgtok::XGe: 1157 case tgtok::XGt: 1158 Type = BitRecTy::get(); 1159 ArgType = IntRecTy::get(); 1160 break; 1161 case tgtok::XListConcat: 1162 // We don't know the list type until we parse the first argument 1163 ArgType = ItemType; 1164 break; 1165 case tgtok::XListSplat: 1166 // Can't do any typechecking until we parse the first argument. 1167 break; 1168 case tgtok::XStrConcat: 1169 Type = StringRecTy::get(); 1170 ArgType = StringRecTy::get(); 1171 break; 1172 case tgtok::XInterleave: 1173 Type = StringRecTy::get(); 1174 // The first argument type is not yet known. 1175 } 1176 1177 if (Type && ItemType && !Type->typeIsConvertibleTo(ItemType)) { 1178 Error(OpLoc, Twine("expected value of type '") + 1179 ItemType->getAsString() + "', got '" + 1180 Type->getAsString() + "'"); 1181 return nullptr; 1182 } 1183 1184 if (!consume(tgtok::l_paren)) { 1185 TokError("expected '(' after binary operator"); 1186 return nullptr; 1187 } 1188 1189 SmallVector<Init*, 2> InitList; 1190 1191 // Note that this loop consumes an arbitrary number of arguments. 1192 // The actual count is checked later. 1193 for (;;) { 1194 SMLoc InitLoc = Lex.getLoc(); 1195 InitList.push_back(ParseValue(CurRec, ArgType)); 1196 if (!InitList.back()) return nullptr; 1197 1198 TypedInit *InitListBack = dyn_cast<TypedInit>(InitList.back()); 1199 if (!InitListBack) { 1200 Error(OpLoc, Twine("expected value to be a typed value, got '" + 1201 InitList.back()->getAsString() + "'")); 1202 return nullptr; 1203 } 1204 RecTy *ListType = InitListBack->getType(); 1205 1206 if (!ArgType) { 1207 // Argument type must be determined from the argument itself. 1208 ArgType = ListType; 1209 1210 switch (Code) { 1211 case BinOpInit::LISTCONCAT: 1212 if (!isa<ListRecTy>(ArgType)) { 1213 Error(InitLoc, Twine("expected a list, got value of type '") + 1214 ArgType->getAsString() + "'"); 1215 return nullptr; 1216 } 1217 break; 1218 case BinOpInit::LISTSPLAT: 1219 if (ItemType && InitList.size() == 1) { 1220 if (!isa<ListRecTy>(ItemType)) { 1221 Error(OpLoc, 1222 Twine("expected output type to be a list, got type '") + 1223 ItemType->getAsString() + "'"); 1224 return nullptr; 1225 } 1226 if (!ArgType->getListTy()->typeIsConvertibleTo(ItemType)) { 1227 Error(OpLoc, Twine("expected first arg type to be '") + 1228 ArgType->getAsString() + 1229 "', got value of type '" + 1230 cast<ListRecTy>(ItemType) 1231 ->getElementType() 1232 ->getAsString() + 1233 "'"); 1234 return nullptr; 1235 } 1236 } 1237 if (InitList.size() == 2 && !isa<IntRecTy>(ArgType)) { 1238 Error(InitLoc, Twine("expected second parameter to be an int, got " 1239 "value of type '") + 1240 ArgType->getAsString() + "'"); 1241 return nullptr; 1242 } 1243 ArgType = nullptr; // Broken invariant: types not identical. 1244 break; 1245 case BinOpInit::EQ: 1246 case BinOpInit::NE: 1247 if (!ArgType->typeIsConvertibleTo(IntRecTy::get()) && 1248 !ArgType->typeIsConvertibleTo(StringRecTy::get())) { 1249 Error(InitLoc, Twine("expected int, bits, or string; got value of " 1250 "type '") + ArgType->getAsString() + "'"); 1251 return nullptr; 1252 } 1253 break; 1254 case BinOpInit::INTERLEAVE: 1255 switch (InitList.size()) { 1256 case 1: // First argument must be a list of strings or integers. 1257 if (ArgType != StringRecTy::get()->getListTy() && 1258 !ArgType->typeIsConvertibleTo(IntRecTy::get()->getListTy())) { 1259 Error(InitLoc, Twine("expected list of string, int, bits, or bit; " 1260 "got value of type '") + 1261 ArgType->getAsString() + "'"); 1262 return nullptr; 1263 } 1264 break; 1265 case 2: // Second argument must be a string. 1266 if (!isa<StringRecTy>(ArgType)) { 1267 Error(InitLoc, Twine("expected second argument to be a string, " 1268 "got value of type '") + 1269 ArgType->getAsString() + "'"); 1270 return nullptr; 1271 } 1272 break; 1273 default: ; 1274 } 1275 ArgType = nullptr; // Broken invariant: types not identical. 1276 break; 1277 default: llvm_unreachable("other ops have fixed argument types"); 1278 } 1279 1280 } else { 1281 // Desired argument type is a known and in ArgType. 1282 RecTy *Resolved = resolveTypes(ArgType, ListType); 1283 if (!Resolved) { 1284 Error(InitLoc, Twine("expected value of type '") + 1285 ArgType->getAsString() + "', got '" + 1286 ListType->getAsString() + "'"); 1287 return nullptr; 1288 } 1289 if (Code != BinOpInit::ADD && Code != BinOpInit::SUB && 1290 Code != BinOpInit::AND && Code != BinOpInit::OR && 1291 Code != BinOpInit::XOR && Code != BinOpInit::SRA && 1292 Code != BinOpInit::SRL && Code != BinOpInit::SHL && 1293 Code != BinOpInit::MUL) 1294 ArgType = Resolved; 1295 } 1296 1297 // Deal with BinOps whose arguments have different types, by 1298 // rewriting ArgType in between them. 1299 switch (Code) { 1300 case BinOpInit::SETDAGOP: 1301 // After parsing the first dag argument, switch to expecting 1302 // a record, with no restriction on its superclasses. 1303 ArgType = RecordRecTy::get({}); 1304 break; 1305 default: 1306 break; 1307 } 1308 1309 if (!consume(tgtok::comma)) 1310 break; 1311 } 1312 1313 if (!consume(tgtok::r_paren)) { 1314 TokError("expected ')' in operator"); 1315 return nullptr; 1316 } 1317 1318 // listconcat returns a list with type of the argument. 1319 if (Code == BinOpInit::LISTCONCAT) 1320 Type = ArgType; 1321 // listsplat returns a list of type of the *first* argument. 1322 if (Code == BinOpInit::LISTSPLAT) 1323 Type = cast<TypedInit>(InitList.front())->getType()->getListTy(); 1324 1325 // We allow multiple operands to associative operators like !strconcat as 1326 // shorthand for nesting them. 1327 if (Code == BinOpInit::STRCONCAT || Code == BinOpInit::LISTCONCAT || 1328 Code == BinOpInit::CONCAT || Code == BinOpInit::ADD || 1329 Code == BinOpInit::AND || Code == BinOpInit::OR || 1330 Code == BinOpInit::XOR || Code == BinOpInit::MUL) { 1331 while (InitList.size() > 2) { 1332 Init *RHS = InitList.pop_back_val(); 1333 RHS = (BinOpInit::get(Code, InitList.back(), RHS, Type))->Fold(CurRec); 1334 InitList.back() = RHS; 1335 } 1336 } 1337 1338 if (InitList.size() == 2) 1339 return (BinOpInit::get(Code, InitList[0], InitList[1], Type)) 1340 ->Fold(CurRec); 1341 1342 Error(OpLoc, "expected two operands to operator"); 1343 return nullptr; 1344 } 1345 1346 case tgtok::XForEach: { 1347 // Value ::= !foreach '(' Id ',' Value ',' Value ')' 1348 SMLoc OpLoc = Lex.getLoc(); 1349 Lex.Lex(); // eat the operation 1350 if (Lex.getCode() != tgtok::l_paren) { 1351 TokError("expected '(' after !foreach"); 1352 return nullptr; 1353 } 1354 1355 if (Lex.Lex() != tgtok::Id) { // eat the '(' 1356 TokError("first argument of !foreach must be an identifier"); 1357 return nullptr; 1358 } 1359 1360 Init *LHS = StringInit::get(Lex.getCurStrVal()); 1361 Lex.Lex(); 1362 1363 if (CurRec && CurRec->getValue(LHS)) { 1364 TokError((Twine("iteration variable '") + LHS->getAsString() + 1365 "' already defined") 1366 .str()); 1367 return nullptr; 1368 } 1369 1370 if (!consume(tgtok::comma)) { // eat the id 1371 TokError("expected ',' in ternary operator"); 1372 return nullptr; 1373 } 1374 1375 Init *MHS = ParseValue(CurRec); 1376 if (!MHS) 1377 return nullptr; 1378 1379 if (!consume(tgtok::comma)) { 1380 TokError("expected ',' in ternary operator"); 1381 return nullptr; 1382 } 1383 1384 TypedInit *MHSt = dyn_cast<TypedInit>(MHS); 1385 if (!MHSt) { 1386 TokError("could not get type of !foreach input"); 1387 return nullptr; 1388 } 1389 1390 RecTy *InEltType = nullptr; 1391 RecTy *OutEltType = nullptr; 1392 bool IsDAG = false; 1393 1394 if (ListRecTy *InListTy = dyn_cast<ListRecTy>(MHSt->getType())) { 1395 InEltType = InListTy->getElementType(); 1396 if (ItemType) { 1397 if (ListRecTy *OutListTy = dyn_cast<ListRecTy>(ItemType)) { 1398 OutEltType = OutListTy->getElementType(); 1399 } else { 1400 Error(OpLoc, 1401 "expected value of type '" + Twine(ItemType->getAsString()) + 1402 "', but got !foreach of list type"); 1403 return nullptr; 1404 } 1405 } 1406 } else if (DagRecTy *InDagTy = dyn_cast<DagRecTy>(MHSt->getType())) { 1407 InEltType = InDagTy; 1408 if (ItemType && !isa<DagRecTy>(ItemType)) { 1409 Error(OpLoc, 1410 "expected value of type '" + Twine(ItemType->getAsString()) + 1411 "', but got !foreach of dag type"); 1412 return nullptr; 1413 } 1414 IsDAG = true; 1415 } else { 1416 TokError("!foreach must have list or dag input"); 1417 return nullptr; 1418 } 1419 1420 // We need to create a temporary record to provide a scope for the 1421 // iteration variable. 1422 std::unique_ptr<Record> ParseRecTmp; 1423 Record *ParseRec = CurRec; 1424 if (!ParseRec) { 1425 ParseRecTmp = std::make_unique<Record>(".parse", ArrayRef<SMLoc>{}, Records); 1426 ParseRec = ParseRecTmp.get(); 1427 } 1428 1429 ParseRec->addValue(RecordVal(LHS, InEltType, false)); 1430 Init *RHS = ParseValue(ParseRec, OutEltType); 1431 ParseRec->removeValue(LHS); 1432 if (!RHS) 1433 return nullptr; 1434 1435 if (!consume(tgtok::r_paren)) { 1436 TokError("expected ')' in binary operator"); 1437 return nullptr; 1438 } 1439 1440 RecTy *OutType; 1441 if (IsDAG) { 1442 OutType = InEltType; 1443 } else { 1444 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1445 if (!RHSt) { 1446 TokError("could not get type of !foreach result"); 1447 return nullptr; 1448 } 1449 OutType = RHSt->getType()->getListTy(); 1450 } 1451 1452 return (TernOpInit::get(TernOpInit::FOREACH, LHS, MHS, RHS, OutType)) 1453 ->Fold(CurRec); 1454 } 1455 1456 case tgtok::XDag: 1457 case tgtok::XIf: 1458 case tgtok::XSubst: { // Value ::= !ternop '(' Value ',' Value ',' Value ')' 1459 TernOpInit::TernaryOp Code; 1460 RecTy *Type = nullptr; 1461 1462 tgtok::TokKind LexCode = Lex.getCode(); 1463 Lex.Lex(); // eat the operation 1464 switch (LexCode) { 1465 default: llvm_unreachable("Unhandled code!"); 1466 case tgtok::XDag: 1467 Code = TernOpInit::DAG; 1468 Type = DagRecTy::get(); 1469 ItemType = nullptr; 1470 break; 1471 case tgtok::XIf: 1472 Code = TernOpInit::IF; 1473 break; 1474 case tgtok::XSubst: 1475 Code = TernOpInit::SUBST; 1476 break; 1477 } 1478 if (!consume(tgtok::l_paren)) { 1479 TokError("expected '(' after ternary operator"); 1480 return nullptr; 1481 } 1482 1483 Init *LHS = ParseValue(CurRec); 1484 if (!LHS) return nullptr; 1485 1486 if (!consume(tgtok::comma)) { 1487 TokError("expected ',' in ternary operator"); 1488 return nullptr; 1489 } 1490 1491 SMLoc MHSLoc = Lex.getLoc(); 1492 Init *MHS = ParseValue(CurRec, ItemType); 1493 if (!MHS) 1494 return nullptr; 1495 1496 if (!consume(tgtok::comma)) { 1497 TokError("expected ',' in ternary operator"); 1498 return nullptr; 1499 } 1500 1501 SMLoc RHSLoc = Lex.getLoc(); 1502 Init *RHS = ParseValue(CurRec, ItemType); 1503 if (!RHS) 1504 return nullptr; 1505 1506 if (!consume(tgtok::r_paren)) { 1507 TokError("expected ')' in binary operator"); 1508 return nullptr; 1509 } 1510 1511 switch (LexCode) { 1512 default: llvm_unreachable("Unhandled code!"); 1513 case tgtok::XDag: { 1514 TypedInit *MHSt = dyn_cast<TypedInit>(MHS); 1515 if (!MHSt && !isa<UnsetInit>(MHS)) { 1516 Error(MHSLoc, "could not determine type of the child list in !dag"); 1517 return nullptr; 1518 } 1519 if (MHSt && !isa<ListRecTy>(MHSt->getType())) { 1520 Error(MHSLoc, Twine("expected list of children, got type '") + 1521 MHSt->getType()->getAsString() + "'"); 1522 return nullptr; 1523 } 1524 1525 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1526 if (!RHSt && !isa<UnsetInit>(RHS)) { 1527 Error(RHSLoc, "could not determine type of the name list in !dag"); 1528 return nullptr; 1529 } 1530 if (RHSt && StringRecTy::get()->getListTy() != RHSt->getType()) { 1531 Error(RHSLoc, Twine("expected list<string>, got type '") + 1532 RHSt->getType()->getAsString() + "'"); 1533 return nullptr; 1534 } 1535 1536 if (!MHSt && !RHSt) { 1537 Error(MHSLoc, 1538 "cannot have both unset children and unset names in !dag"); 1539 return nullptr; 1540 } 1541 break; 1542 } 1543 case tgtok::XIf: { 1544 RecTy *MHSTy = nullptr; 1545 RecTy *RHSTy = nullptr; 1546 1547 if (TypedInit *MHSt = dyn_cast<TypedInit>(MHS)) 1548 MHSTy = MHSt->getType(); 1549 if (BitsInit *MHSbits = dyn_cast<BitsInit>(MHS)) 1550 MHSTy = BitsRecTy::get(MHSbits->getNumBits()); 1551 if (isa<BitInit>(MHS)) 1552 MHSTy = BitRecTy::get(); 1553 1554 if (TypedInit *RHSt = dyn_cast<TypedInit>(RHS)) 1555 RHSTy = RHSt->getType(); 1556 if (BitsInit *RHSbits = dyn_cast<BitsInit>(RHS)) 1557 RHSTy = BitsRecTy::get(RHSbits->getNumBits()); 1558 if (isa<BitInit>(RHS)) 1559 RHSTy = BitRecTy::get(); 1560 1561 // For UnsetInit, it's typed from the other hand. 1562 if (isa<UnsetInit>(MHS)) 1563 MHSTy = RHSTy; 1564 if (isa<UnsetInit>(RHS)) 1565 RHSTy = MHSTy; 1566 1567 if (!MHSTy || !RHSTy) { 1568 TokError("could not get type for !if"); 1569 return nullptr; 1570 } 1571 1572 Type = resolveTypes(MHSTy, RHSTy); 1573 if (!Type) { 1574 TokError(Twine("inconsistent types '") + MHSTy->getAsString() + 1575 "' and '" + RHSTy->getAsString() + "' for !if"); 1576 return nullptr; 1577 } 1578 break; 1579 } 1580 case tgtok::XSubst: { 1581 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1582 if (!RHSt) { 1583 TokError("could not get type for !subst"); 1584 return nullptr; 1585 } 1586 Type = RHSt->getType(); 1587 break; 1588 } 1589 } 1590 return (TernOpInit::get(Code, LHS, MHS, RHS, Type))->Fold(CurRec); 1591 } 1592 1593 case tgtok::XCond: 1594 return ParseOperationCond(CurRec, ItemType); 1595 1596 case tgtok::XFoldl: { 1597 // Value ::= !foldl '(' Value ',' Value ',' Id ',' Id ',' Expr ')' 1598 Lex.Lex(); // eat the operation 1599 if (!consume(tgtok::l_paren)) { 1600 TokError("expected '(' after !foldl"); 1601 return nullptr; 1602 } 1603 1604 Init *StartUntyped = ParseValue(CurRec); 1605 if (!StartUntyped) 1606 return nullptr; 1607 1608 TypedInit *Start = dyn_cast<TypedInit>(StartUntyped); 1609 if (!Start) { 1610 TokError(Twine("could not get type of !foldl start: '") + 1611 StartUntyped->getAsString() + "'"); 1612 return nullptr; 1613 } 1614 1615 if (!consume(tgtok::comma)) { 1616 TokError("expected ',' in !foldl"); 1617 return nullptr; 1618 } 1619 1620 Init *ListUntyped = ParseValue(CurRec); 1621 if (!ListUntyped) 1622 return nullptr; 1623 1624 TypedInit *List = dyn_cast<TypedInit>(ListUntyped); 1625 if (!List) { 1626 TokError(Twine("could not get type of !foldl list: '") + 1627 ListUntyped->getAsString() + "'"); 1628 return nullptr; 1629 } 1630 1631 ListRecTy *ListType = dyn_cast<ListRecTy>(List->getType()); 1632 if (!ListType) { 1633 TokError(Twine("!foldl list must be a list, but is of type '") + 1634 List->getType()->getAsString()); 1635 return nullptr; 1636 } 1637 1638 if (Lex.getCode() != tgtok::comma) { 1639 TokError("expected ',' in !foldl"); 1640 return nullptr; 1641 } 1642 1643 if (Lex.Lex() != tgtok::Id) { // eat the ',' 1644 TokError("third argument of !foldl must be an identifier"); 1645 return nullptr; 1646 } 1647 1648 Init *A = StringInit::get(Lex.getCurStrVal()); 1649 if (CurRec && CurRec->getValue(A)) { 1650 TokError((Twine("left !foldl variable '") + A->getAsString() + 1651 "' already defined") 1652 .str()); 1653 return nullptr; 1654 } 1655 1656 if (Lex.Lex() != tgtok::comma) { // eat the id 1657 TokError("expected ',' in !foldl"); 1658 return nullptr; 1659 } 1660 1661 if (Lex.Lex() != tgtok::Id) { // eat the ',' 1662 TokError("fourth argument of !foldl must be an identifier"); 1663 return nullptr; 1664 } 1665 1666 Init *B = StringInit::get(Lex.getCurStrVal()); 1667 if (CurRec && CurRec->getValue(B)) { 1668 TokError((Twine("right !foldl variable '") + B->getAsString() + 1669 "' already defined") 1670 .str()); 1671 return nullptr; 1672 } 1673 1674 if (Lex.Lex() != tgtok::comma) { // eat the id 1675 TokError("expected ',' in !foldl"); 1676 return nullptr; 1677 } 1678 Lex.Lex(); // eat the ',' 1679 1680 // We need to create a temporary record to provide a scope for the 1681 // two variables. 1682 std::unique_ptr<Record> ParseRecTmp; 1683 Record *ParseRec = CurRec; 1684 if (!ParseRec) { 1685 ParseRecTmp = std::make_unique<Record>(".parse", ArrayRef<SMLoc>{}, Records); 1686 ParseRec = ParseRecTmp.get(); 1687 } 1688 1689 ParseRec->addValue(RecordVal(A, Start->getType(), false)); 1690 ParseRec->addValue(RecordVal(B, ListType->getElementType(), false)); 1691 Init *ExprUntyped = ParseValue(ParseRec); 1692 ParseRec->removeValue(A); 1693 ParseRec->removeValue(B); 1694 if (!ExprUntyped) 1695 return nullptr; 1696 1697 TypedInit *Expr = dyn_cast<TypedInit>(ExprUntyped); 1698 if (!Expr) { 1699 TokError("could not get type of !foldl expression"); 1700 return nullptr; 1701 } 1702 1703 if (Expr->getType() != Start->getType()) { 1704 TokError(Twine("!foldl expression must be of same type as start (") + 1705 Start->getType()->getAsString() + "), but is of type " + 1706 Expr->getType()->getAsString()); 1707 return nullptr; 1708 } 1709 1710 if (!consume(tgtok::r_paren)) { 1711 TokError("expected ')' in fold operator"); 1712 return nullptr; 1713 } 1714 1715 return FoldOpInit::get(Start, List, A, B, Expr, Start->getType()) 1716 ->Fold(CurRec); 1717 } 1718 } 1719 } 1720 1721 /// ParseOperatorType - Parse a type for an operator. This returns 1722 /// null on error. 1723 /// 1724 /// OperatorType ::= '<' Type '>' 1725 /// 1726 RecTy *TGParser::ParseOperatorType() { 1727 RecTy *Type = nullptr; 1728 1729 if (!consume(tgtok::less)) { 1730 TokError("expected type name for operator"); 1731 return nullptr; 1732 } 1733 1734 Type = ParseType(); 1735 1736 if (!Type) { 1737 TokError("expected type name for operator"); 1738 return nullptr; 1739 } 1740 1741 if (!consume(tgtok::greater)) { 1742 TokError("expected type name for operator"); 1743 return nullptr; 1744 } 1745 1746 return Type; 1747 } 1748 1749 Init *TGParser::ParseOperationCond(Record *CurRec, RecTy *ItemType) { 1750 Lex.Lex(); // eat the operation 'cond' 1751 1752 if (!consume(tgtok::l_paren)) { 1753 TokError("expected '(' after !cond operator"); 1754 return nullptr; 1755 } 1756 1757 // Parse through '[Case: Val,]+' 1758 SmallVector<Init *, 4> Case; 1759 SmallVector<Init *, 4> Val; 1760 while (true) { 1761 if (consume(tgtok::r_paren)) 1762 break; 1763 1764 Init *V = ParseValue(CurRec); 1765 if (!V) 1766 return nullptr; 1767 Case.push_back(V); 1768 1769 if (!consume(tgtok::colon)) { 1770 TokError("expected ':' following a condition in !cond operator"); 1771 return nullptr; 1772 } 1773 1774 V = ParseValue(CurRec, ItemType); 1775 if (!V) 1776 return nullptr; 1777 Val.push_back(V); 1778 1779 if (consume(tgtok::r_paren)) 1780 break; 1781 1782 if (!consume(tgtok::comma)) { 1783 TokError("expected ',' or ')' following a value in !cond operator"); 1784 return nullptr; 1785 } 1786 } 1787 1788 if (Case.size() < 1) { 1789 TokError("there should be at least 1 'condition : value' in the !cond operator"); 1790 return nullptr; 1791 } 1792 1793 // resolve type 1794 RecTy *Type = nullptr; 1795 for (Init *V : Val) { 1796 RecTy *VTy = nullptr; 1797 if (TypedInit *Vt = dyn_cast<TypedInit>(V)) 1798 VTy = Vt->getType(); 1799 if (BitsInit *Vbits = dyn_cast<BitsInit>(V)) 1800 VTy = BitsRecTy::get(Vbits->getNumBits()); 1801 if (isa<BitInit>(V)) 1802 VTy = BitRecTy::get(); 1803 1804 if (Type == nullptr) { 1805 if (!isa<UnsetInit>(V)) 1806 Type = VTy; 1807 } else { 1808 if (!isa<UnsetInit>(V)) { 1809 RecTy *RType = resolveTypes(Type, VTy); 1810 if (!RType) { 1811 TokError(Twine("inconsistent types '") + Type->getAsString() + 1812 "' and '" + VTy->getAsString() + "' for !cond"); 1813 return nullptr; 1814 } 1815 Type = RType; 1816 } 1817 } 1818 } 1819 1820 if (!Type) { 1821 TokError("could not determine type for !cond from its arguments"); 1822 return nullptr; 1823 } 1824 return CondOpInit::get(Case, Val, Type)->Fold(CurRec); 1825 } 1826 1827 /// ParseSimpleValue - Parse a tblgen value. This returns null on error. 1828 /// 1829 /// SimpleValue ::= IDValue 1830 /// SimpleValue ::= INTVAL 1831 /// SimpleValue ::= STRVAL+ 1832 /// SimpleValue ::= CODEFRAGMENT 1833 /// SimpleValue ::= '?' 1834 /// SimpleValue ::= '{' ValueList '}' 1835 /// SimpleValue ::= ID '<' ValueListNE '>' 1836 /// SimpleValue ::= '[' ValueList ']' 1837 /// SimpleValue ::= '(' IDValue DagArgList ')' 1838 /// SimpleValue ::= CONCATTOK '(' Value ',' Value ')' 1839 /// SimpleValue ::= ADDTOK '(' Value ',' Value ')' 1840 /// SimpleValue ::= SUBTOK '(' Value ',' Value ')' 1841 /// SimpleValue ::= SHLTOK '(' Value ',' Value ')' 1842 /// SimpleValue ::= SRATOK '(' Value ',' Value ')' 1843 /// SimpleValue ::= SRLTOK '(' Value ',' Value ')' 1844 /// SimpleValue ::= LISTCONCATTOK '(' Value ',' Value ')' 1845 /// SimpleValue ::= LISTSPLATTOK '(' Value ',' Value ')' 1846 /// SimpleValue ::= STRCONCATTOK '(' Value ',' Value ')' 1847 /// SimpleValue ::= COND '(' [Value ':' Value,]+ ')' 1848 /// 1849 Init *TGParser::ParseSimpleValue(Record *CurRec, RecTy *ItemType, 1850 IDParseMode Mode) { 1851 Init *R = nullptr; 1852 switch (Lex.getCode()) { 1853 default: TokError("Unknown token when parsing a value"); break; 1854 case tgtok::IntVal: R = IntInit::get(Lex.getCurIntVal()); Lex.Lex(); break; 1855 case tgtok::BinaryIntVal: { 1856 auto BinaryVal = Lex.getCurBinaryIntVal(); 1857 SmallVector<Init*, 16> Bits(BinaryVal.second); 1858 for (unsigned i = 0, e = BinaryVal.second; i != e; ++i) 1859 Bits[i] = BitInit::get(BinaryVal.first & (1LL << i)); 1860 R = BitsInit::get(Bits); 1861 Lex.Lex(); 1862 break; 1863 } 1864 case tgtok::StrVal: { 1865 std::string Val = Lex.getCurStrVal(); 1866 Lex.Lex(); 1867 1868 // Handle multiple consecutive concatenated strings. 1869 while (Lex.getCode() == tgtok::StrVal) { 1870 Val += Lex.getCurStrVal(); 1871 Lex.Lex(); 1872 } 1873 1874 R = StringInit::get(Val); 1875 break; 1876 } 1877 case tgtok::CodeFragment: 1878 R = CodeInit::get(Lex.getCurStrVal(), Lex.getLoc()); 1879 Lex.Lex(); 1880 break; 1881 case tgtok::question: 1882 R = UnsetInit::get(); 1883 Lex.Lex(); 1884 break; 1885 case tgtok::Id: { 1886 SMLoc NameLoc = Lex.getLoc(); 1887 StringInit *Name = StringInit::get(Lex.getCurStrVal()); 1888 if (Lex.Lex() != tgtok::less) // consume the Id. 1889 return ParseIDValue(CurRec, Name, NameLoc, Mode); // Value ::= IDValue 1890 1891 // Value ::= ID '<' ValueListNE '>' 1892 if (Lex.Lex() == tgtok::greater) { 1893 TokError("expected non-empty value list"); 1894 return nullptr; 1895 } 1896 1897 // This is a CLASS<initvalslist> expression. This is supposed to synthesize 1898 // a new anonymous definition, deriving from CLASS<initvalslist> with no 1899 // body. 1900 Record *Class = Records.getClass(Name->getValue()); 1901 if (!Class) { 1902 Error(NameLoc, "Expected a class name, got '" + Name->getValue() + "'"); 1903 return nullptr; 1904 } 1905 1906 SmallVector<Init *, 8> Args; 1907 ParseValueList(Args, CurRec, Class); 1908 if (Args.empty()) return nullptr; 1909 1910 if (!consume(tgtok::greater)) { 1911 TokError("expected '>' at end of value list"); 1912 return nullptr; 1913 } 1914 1915 // Typecheck the template arguments list 1916 ArrayRef<Init *> ExpectedArgs = Class->getTemplateArgs(); 1917 if (ExpectedArgs.size() < Args.size()) { 1918 Error(NameLoc, 1919 "More template args specified than expected"); 1920 return nullptr; 1921 } 1922 1923 for (unsigned i = 0, e = ExpectedArgs.size(); i != e; ++i) { 1924 RecordVal *ExpectedArg = Class->getValue(ExpectedArgs[i]); 1925 if (i < Args.size()) { 1926 if (TypedInit *TI = dyn_cast<TypedInit>(Args[i])) { 1927 RecTy *ExpectedType = ExpectedArg->getType(); 1928 if (!TI->getType()->typeIsConvertibleTo(ExpectedType)) { 1929 Error(NameLoc, 1930 "Value specified for template argument #" + Twine(i) + " (" + 1931 ExpectedArg->getNameInitAsString() + ") is of type '" + 1932 TI->getType()->getAsString() + "', expected '" + 1933 ExpectedType->getAsString() + "': " + TI->getAsString()); 1934 return nullptr; 1935 } 1936 continue; 1937 } 1938 } else if (ExpectedArg->getValue()->isComplete()) 1939 continue; 1940 1941 Error(NameLoc, 1942 "Value not specified for template argument #" + Twine(i) + " (" + 1943 ExpectedArgs[i]->getAsUnquotedString() + ")"); 1944 return nullptr; 1945 } 1946 1947 return VarDefInit::get(Class, Args)->Fold(); 1948 } 1949 case tgtok::l_brace: { // Value ::= '{' ValueList '}' 1950 SMLoc BraceLoc = Lex.getLoc(); 1951 Lex.Lex(); // eat the '{' 1952 SmallVector<Init*, 16> Vals; 1953 1954 if (Lex.getCode() != tgtok::r_brace) { 1955 ParseValueList(Vals, CurRec); 1956 if (Vals.empty()) return nullptr; 1957 } 1958 if (!consume(tgtok::r_brace)) { 1959 TokError("expected '}' at end of bit list value"); 1960 return nullptr; 1961 } 1962 1963 SmallVector<Init *, 16> NewBits; 1964 1965 // As we parse { a, b, ... }, 'a' is the highest bit, but we parse it 1966 // first. We'll first read everything in to a vector, then we can reverse 1967 // it to get the bits in the correct order for the BitsInit value. 1968 for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 1969 // FIXME: The following two loops would not be duplicated 1970 // if the API was a little more orthogonal. 1971 1972 // bits<n> values are allowed to initialize n bits. 1973 if (BitsInit *BI = dyn_cast<BitsInit>(Vals[i])) { 1974 for (unsigned i = 0, e = BI->getNumBits(); i != e; ++i) 1975 NewBits.push_back(BI->getBit((e - i) - 1)); 1976 continue; 1977 } 1978 // bits<n> can also come from variable initializers. 1979 if (VarInit *VI = dyn_cast<VarInit>(Vals[i])) { 1980 if (BitsRecTy *BitsRec = dyn_cast<BitsRecTy>(VI->getType())) { 1981 for (unsigned i = 0, e = BitsRec->getNumBits(); i != e; ++i) 1982 NewBits.push_back(VI->getBit((e - i) - 1)); 1983 continue; 1984 } 1985 // Fallthrough to try convert this to a bit. 1986 } 1987 // All other values must be convertible to just a single bit. 1988 Init *Bit = Vals[i]->getCastTo(BitRecTy::get()); 1989 if (!Bit) { 1990 Error(BraceLoc, "Element #" + Twine(i) + " (" + Vals[i]->getAsString() + 1991 ") is not convertable to a bit"); 1992 return nullptr; 1993 } 1994 NewBits.push_back(Bit); 1995 } 1996 std::reverse(NewBits.begin(), NewBits.end()); 1997 return BitsInit::get(NewBits); 1998 } 1999 case tgtok::l_square: { // Value ::= '[' ValueList ']' 2000 Lex.Lex(); // eat the '[' 2001 SmallVector<Init*, 16> Vals; 2002 2003 RecTy *DeducedEltTy = nullptr; 2004 ListRecTy *GivenListTy = nullptr; 2005 2006 if (ItemType) { 2007 ListRecTy *ListType = dyn_cast<ListRecTy>(ItemType); 2008 if (!ListType) { 2009 TokError(Twine("Encountered a list when expecting a ") + 2010 ItemType->getAsString()); 2011 return nullptr; 2012 } 2013 GivenListTy = ListType; 2014 } 2015 2016 if (Lex.getCode() != tgtok::r_square) { 2017 ParseValueList(Vals, CurRec, nullptr, 2018 GivenListTy ? GivenListTy->getElementType() : nullptr); 2019 if (Vals.empty()) return nullptr; 2020 } 2021 if (!consume(tgtok::r_square)) { 2022 TokError("expected ']' at end of list value"); 2023 return nullptr; 2024 } 2025 2026 RecTy *GivenEltTy = nullptr; 2027 if (consume(tgtok::less)) { 2028 // Optional list element type 2029 GivenEltTy = ParseType(); 2030 if (!GivenEltTy) { 2031 // Couldn't parse element type 2032 return nullptr; 2033 } 2034 2035 if (!consume(tgtok::greater)) { 2036 TokError("expected '>' at end of list element type"); 2037 return nullptr; 2038 } 2039 } 2040 2041 // Check elements 2042 RecTy *EltTy = nullptr; 2043 for (Init *V : Vals) { 2044 TypedInit *TArg = dyn_cast<TypedInit>(V); 2045 if (TArg) { 2046 if (EltTy) { 2047 EltTy = resolveTypes(EltTy, TArg->getType()); 2048 if (!EltTy) { 2049 TokError("Incompatible types in list elements"); 2050 return nullptr; 2051 } 2052 } else { 2053 EltTy = TArg->getType(); 2054 } 2055 } 2056 } 2057 2058 if (GivenEltTy) { 2059 if (EltTy) { 2060 // Verify consistency 2061 if (!EltTy->typeIsConvertibleTo(GivenEltTy)) { 2062 TokError("Incompatible types in list elements"); 2063 return nullptr; 2064 } 2065 } 2066 EltTy = GivenEltTy; 2067 } 2068 2069 if (!EltTy) { 2070 if (!ItemType) { 2071 TokError("No type for list"); 2072 return nullptr; 2073 } 2074 DeducedEltTy = GivenListTy->getElementType(); 2075 } else { 2076 // Make sure the deduced type is compatible with the given type 2077 if (GivenListTy) { 2078 if (!EltTy->typeIsConvertibleTo(GivenListTy->getElementType())) { 2079 TokError(Twine("Element type mismatch for list: element type '") + 2080 EltTy->getAsString() + "' not convertible to '" + 2081 GivenListTy->getElementType()->getAsString()); 2082 return nullptr; 2083 } 2084 } 2085 DeducedEltTy = EltTy; 2086 } 2087 2088 return ListInit::get(Vals, DeducedEltTy); 2089 } 2090 case tgtok::l_paren: { // Value ::= '(' IDValue DagArgList ')' 2091 Lex.Lex(); // eat the '(' 2092 if (Lex.getCode() != tgtok::Id && Lex.getCode() != tgtok::XCast && 2093 Lex.getCode() != tgtok::question && Lex.getCode() != tgtok::XGetDagOp) { 2094 TokError("expected identifier in dag init"); 2095 return nullptr; 2096 } 2097 2098 Init *Operator = ParseValue(CurRec); 2099 if (!Operator) return nullptr; 2100 2101 // If the operator name is present, parse it. 2102 StringInit *OperatorName = nullptr; 2103 if (consume(tgtok::colon)) { 2104 if (Lex.getCode() != tgtok::VarName) { // eat the ':' 2105 TokError("expected variable name in dag operator"); 2106 return nullptr; 2107 } 2108 OperatorName = StringInit::get(Lex.getCurStrVal()); 2109 Lex.Lex(); // eat the VarName. 2110 } 2111 2112 SmallVector<std::pair<llvm::Init*, StringInit*>, 8> DagArgs; 2113 if (Lex.getCode() != tgtok::r_paren) { 2114 ParseDagArgList(DagArgs, CurRec); 2115 if (DagArgs.empty()) return nullptr; 2116 } 2117 2118 if (!consume(tgtok::r_paren)) { 2119 TokError("expected ')' in dag init"); 2120 return nullptr; 2121 } 2122 2123 return DagInit::get(Operator, OperatorName, DagArgs); 2124 } 2125 2126 case tgtok::XHead: 2127 case tgtok::XTail: 2128 case tgtok::XSize: 2129 case tgtok::XEmpty: 2130 case tgtok::XCast: 2131 case tgtok::XGetDagOp: // Value ::= !unop '(' Value ')' 2132 case tgtok::XIsA: 2133 case tgtok::XConcat: 2134 case tgtok::XDag: 2135 case tgtok::XADD: 2136 case tgtok::XSUB: 2137 case tgtok::XMUL: 2138 case tgtok::XNOT: 2139 case tgtok::XAND: 2140 case tgtok::XOR: 2141 case tgtok::XXOR: 2142 case tgtok::XSRA: 2143 case tgtok::XSRL: 2144 case tgtok::XSHL: 2145 case tgtok::XEq: 2146 case tgtok::XNe: 2147 case tgtok::XLe: 2148 case tgtok::XLt: 2149 case tgtok::XGe: 2150 case tgtok::XGt: 2151 case tgtok::XListConcat: 2152 case tgtok::XListSplat: 2153 case tgtok::XStrConcat: 2154 case tgtok::XInterleave: 2155 case tgtok::XSetDagOp: // Value ::= !binop '(' Value ',' Value ')' 2156 case tgtok::XIf: 2157 case tgtok::XCond: 2158 case tgtok::XFoldl: 2159 case tgtok::XForEach: 2160 case tgtok::XSubst: { // Value ::= !ternop '(' Value ',' Value ',' Value ')' 2161 return ParseOperation(CurRec, ItemType); 2162 } 2163 } 2164 2165 return R; 2166 } 2167 2168 /// ParseValue - Parse a tblgen value. This returns null on error. 2169 /// 2170 /// Value ::= SimpleValue ValueSuffix* 2171 /// ValueSuffix ::= '{' BitList '}' 2172 /// ValueSuffix ::= '[' BitList ']' 2173 /// ValueSuffix ::= '.' ID 2174 /// 2175 Init *TGParser::ParseValue(Record *CurRec, RecTy *ItemType, IDParseMode Mode) { 2176 Init *Result = ParseSimpleValue(CurRec, ItemType, Mode); 2177 if (!Result) return nullptr; 2178 2179 // Parse the suffixes now if present. 2180 while (true) { 2181 switch (Lex.getCode()) { 2182 default: return Result; 2183 case tgtok::l_brace: { 2184 if (Mode == ParseNameMode) 2185 // This is the beginning of the object body. 2186 return Result; 2187 2188 SMLoc CurlyLoc = Lex.getLoc(); 2189 Lex.Lex(); // eat the '{' 2190 SmallVector<unsigned, 16> Ranges; 2191 ParseRangeList(Ranges); 2192 if (Ranges.empty()) return nullptr; 2193 2194 // Reverse the bitlist. 2195 std::reverse(Ranges.begin(), Ranges.end()); 2196 Result = Result->convertInitializerBitRange(Ranges); 2197 if (!Result) { 2198 Error(CurlyLoc, "Invalid bit range for value"); 2199 return nullptr; 2200 } 2201 2202 // Eat the '}'. 2203 if (!consume(tgtok::r_brace)) { 2204 TokError("expected '}' at end of bit range list"); 2205 return nullptr; 2206 } 2207 break; 2208 } 2209 case tgtok::l_square: { 2210 SMLoc SquareLoc = Lex.getLoc(); 2211 Lex.Lex(); // eat the '[' 2212 SmallVector<unsigned, 16> Ranges; 2213 ParseRangeList(Ranges); 2214 if (Ranges.empty()) return nullptr; 2215 2216 Result = Result->convertInitListSlice(Ranges); 2217 if (!Result) { 2218 Error(SquareLoc, "Invalid range for list slice"); 2219 return nullptr; 2220 } 2221 2222 // Eat the ']'. 2223 if (!consume(tgtok::r_square)) { 2224 TokError("expected ']' at end of list slice"); 2225 return nullptr; 2226 } 2227 break; 2228 } 2229 case tgtok::dot: { 2230 if (Lex.Lex() != tgtok::Id) { // eat the . 2231 TokError("expected field identifier after '.'"); 2232 return nullptr; 2233 } 2234 StringInit *FieldName = StringInit::get(Lex.getCurStrVal()); 2235 if (!Result->getFieldType(FieldName)) { 2236 TokError("Cannot access field '" + Lex.getCurStrVal() + "' of value '" + 2237 Result->getAsString() + "'"); 2238 return nullptr; 2239 } 2240 Result = FieldInit::get(Result, FieldName)->Fold(CurRec); 2241 Lex.Lex(); // eat field name 2242 break; 2243 } 2244 2245 case tgtok::paste: 2246 SMLoc PasteLoc = Lex.getLoc(); 2247 TypedInit *LHS = dyn_cast<TypedInit>(Result); 2248 if (!LHS) { 2249 Error(PasteLoc, "LHS of paste is not typed!"); 2250 return nullptr; 2251 } 2252 2253 // Check if it's a 'listA # listB' 2254 if (isa<ListRecTy>(LHS->getType())) { 2255 Lex.Lex(); // Eat the '#'. 2256 2257 assert(Mode == ParseValueMode && "encountered paste of lists in name"); 2258 2259 switch (Lex.getCode()) { 2260 case tgtok::colon: 2261 case tgtok::semi: 2262 case tgtok::l_brace: 2263 Result = LHS; // trailing paste, ignore. 2264 break; 2265 default: 2266 Init *RHSResult = ParseValue(CurRec, ItemType, ParseValueMode); 2267 if (!RHSResult) 2268 return nullptr; 2269 Result = BinOpInit::getListConcat(LHS, RHSResult); 2270 } 2271 break; 2272 } 2273 2274 // Create a !strconcat() operation, first casting each operand to 2275 // a string if necessary. 2276 if (LHS->getType() != StringRecTy::get()) { 2277 auto CastLHS = dyn_cast<TypedInit>( 2278 UnOpInit::get(UnOpInit::CAST, LHS, StringRecTy::get()) 2279 ->Fold(CurRec)); 2280 if (!CastLHS) { 2281 Error(PasteLoc, 2282 Twine("can't cast '") + LHS->getAsString() + "' to string"); 2283 return nullptr; 2284 } 2285 LHS = CastLHS; 2286 } 2287 2288 TypedInit *RHS = nullptr; 2289 2290 Lex.Lex(); // Eat the '#'. 2291 switch (Lex.getCode()) { 2292 case tgtok::colon: 2293 case tgtok::semi: 2294 case tgtok::l_brace: 2295 // These are all of the tokens that can begin an object body. 2296 // Some of these can also begin values but we disallow those cases 2297 // because they are unlikely to be useful. 2298 2299 // Trailing paste, concat with an empty string. 2300 RHS = StringInit::get(""); 2301 break; 2302 2303 default: 2304 Init *RHSResult = ParseValue(CurRec, nullptr, ParseNameMode); 2305 if (!RHSResult) 2306 return nullptr; 2307 RHS = dyn_cast<TypedInit>(RHSResult); 2308 if (!RHS) { 2309 Error(PasteLoc, "RHS of paste is not typed!"); 2310 return nullptr; 2311 } 2312 2313 if (RHS->getType() != StringRecTy::get()) { 2314 auto CastRHS = dyn_cast<TypedInit>( 2315 UnOpInit::get(UnOpInit::CAST, RHS, StringRecTy::get()) 2316 ->Fold(CurRec)); 2317 if (!CastRHS) { 2318 Error(PasteLoc, 2319 Twine("can't cast '") + RHS->getAsString() + "' to string"); 2320 return nullptr; 2321 } 2322 RHS = CastRHS; 2323 } 2324 2325 break; 2326 } 2327 2328 Result = BinOpInit::getStrConcat(LHS, RHS); 2329 break; 2330 } 2331 } 2332 } 2333 2334 /// ParseDagArgList - Parse the argument list for a dag literal expression. 2335 /// 2336 /// DagArg ::= Value (':' VARNAME)? 2337 /// DagArg ::= VARNAME 2338 /// DagArgList ::= DagArg 2339 /// DagArgList ::= DagArgList ',' DagArg 2340 void TGParser::ParseDagArgList( 2341 SmallVectorImpl<std::pair<llvm::Init*, StringInit*>> &Result, 2342 Record *CurRec) { 2343 2344 while (true) { 2345 // DagArg ::= VARNAME 2346 if (Lex.getCode() == tgtok::VarName) { 2347 // A missing value is treated like '?'. 2348 StringInit *VarName = StringInit::get(Lex.getCurStrVal()); 2349 Result.emplace_back(UnsetInit::get(), VarName); 2350 Lex.Lex(); 2351 } else { 2352 // DagArg ::= Value (':' VARNAME)? 2353 Init *Val = ParseValue(CurRec); 2354 if (!Val) { 2355 Result.clear(); 2356 return; 2357 } 2358 2359 // If the variable name is present, add it. 2360 StringInit *VarName = nullptr; 2361 if (Lex.getCode() == tgtok::colon) { 2362 if (Lex.Lex() != tgtok::VarName) { // eat the ':' 2363 TokError("expected variable name in dag literal"); 2364 Result.clear(); 2365 return; 2366 } 2367 VarName = StringInit::get(Lex.getCurStrVal()); 2368 Lex.Lex(); // eat the VarName. 2369 } 2370 2371 Result.push_back(std::make_pair(Val, VarName)); 2372 } 2373 if (!consume(tgtok::comma)) 2374 break; 2375 } 2376 } 2377 2378 /// ParseValueList - Parse a comma separated list of values, returning them as a 2379 /// vector. Note that this always expects to be able to parse at least one 2380 /// value. It returns an empty list if this is not possible. 2381 /// 2382 /// ValueList ::= Value (',' Value) 2383 /// 2384 void TGParser::ParseValueList(SmallVectorImpl<Init*> &Result, Record *CurRec, 2385 Record *ArgsRec, RecTy *EltTy) { 2386 RecTy *ItemType = EltTy; 2387 unsigned int ArgN = 0; 2388 if (ArgsRec && !EltTy) { 2389 ArrayRef<Init *> TArgs = ArgsRec->getTemplateArgs(); 2390 if (TArgs.empty()) { 2391 TokError("template argument provided to non-template class"); 2392 Result.clear(); 2393 return; 2394 } 2395 const RecordVal *RV = ArgsRec->getValue(TArgs[ArgN]); 2396 if (!RV) { 2397 errs() << "Cannot find template arg " << ArgN << " (" << TArgs[ArgN] 2398 << ")\n"; 2399 } 2400 assert(RV && "Template argument record not found??"); 2401 ItemType = RV->getType(); 2402 ++ArgN; 2403 } 2404 Result.push_back(ParseValue(CurRec, ItemType)); 2405 if (!Result.back()) { 2406 Result.clear(); 2407 return; 2408 } 2409 2410 while (consume(tgtok::comma)) { 2411 // ignore trailing comma for lists 2412 if (Lex.getCode() == tgtok::r_square) 2413 return; 2414 2415 if (ArgsRec && !EltTy) { 2416 ArrayRef<Init *> TArgs = ArgsRec->getTemplateArgs(); 2417 if (ArgN >= TArgs.size()) { 2418 TokError("too many template arguments"); 2419 Result.clear(); 2420 return; 2421 } 2422 const RecordVal *RV = ArgsRec->getValue(TArgs[ArgN]); 2423 assert(RV && "Template argument record not found??"); 2424 ItemType = RV->getType(); 2425 ++ArgN; 2426 } 2427 Result.push_back(ParseValue(CurRec, ItemType)); 2428 if (!Result.back()) { 2429 Result.clear(); 2430 return; 2431 } 2432 } 2433 } 2434 2435 /// ParseDeclaration - Read a declaration, returning the name of field ID, or an 2436 /// empty string on error. This can happen in a number of different context's, 2437 /// including within a def or in the template args for a def (which which case 2438 /// CurRec will be non-null) and within the template args for a multiclass (in 2439 /// which case CurRec will be null, but CurMultiClass will be set). This can 2440 /// also happen within a def that is within a multiclass, which will set both 2441 /// CurRec and CurMultiClass. 2442 /// 2443 /// Declaration ::= FIELD? Type ID ('=' Value)? 2444 /// 2445 Init *TGParser::ParseDeclaration(Record *CurRec, 2446 bool ParsingTemplateArgs) { 2447 // Read the field prefix if present. 2448 bool HasField = consume(tgtok::Field); 2449 2450 RecTy *Type = ParseType(); 2451 if (!Type) return nullptr; 2452 2453 if (Lex.getCode() != tgtok::Id) { 2454 TokError("Expected identifier in declaration"); 2455 return nullptr; 2456 } 2457 2458 std::string Str = Lex.getCurStrVal(); 2459 if (Str == "NAME") { 2460 TokError("'" + Str + "' is a reserved variable name"); 2461 return nullptr; 2462 } 2463 2464 SMLoc IdLoc = Lex.getLoc(); 2465 Init *DeclName = StringInit::get(Str); 2466 Lex.Lex(); 2467 2468 if (ParsingTemplateArgs) { 2469 if (CurRec) 2470 DeclName = QualifyName(*CurRec, CurMultiClass, DeclName, ":"); 2471 else 2472 assert(CurMultiClass); 2473 if (CurMultiClass) 2474 DeclName = QualifyName(CurMultiClass->Rec, CurMultiClass, DeclName, 2475 "::"); 2476 } 2477 2478 // Add the value. 2479 if (AddValue(CurRec, IdLoc, RecordVal(DeclName, IdLoc, Type, HasField))) 2480 return nullptr; 2481 2482 // If a value is present, parse it. 2483 if (consume(tgtok::equal)) { 2484 SMLoc ValLoc = Lex.getLoc(); 2485 Init *Val = ParseValue(CurRec, Type); 2486 if (!Val || 2487 SetValue(CurRec, ValLoc, DeclName, None, Val)) 2488 // Return the name, even if an error is thrown. This is so that we can 2489 // continue to make some progress, even without the value having been 2490 // initialized. 2491 return DeclName; 2492 } 2493 2494 return DeclName; 2495 } 2496 2497 /// ParseForeachDeclaration - Read a foreach declaration, returning 2498 /// the name of the declared object or a NULL Init on error. Return 2499 /// the name of the parsed initializer list through ForeachListName. 2500 /// 2501 /// ForeachDeclaration ::= ID '=' '{' RangeList '}' 2502 /// ForeachDeclaration ::= ID '=' RangePiece 2503 /// ForeachDeclaration ::= ID '=' Value 2504 /// 2505 VarInit *TGParser::ParseForeachDeclaration(Init *&ForeachListValue) { 2506 if (Lex.getCode() != tgtok::Id) { 2507 TokError("Expected identifier in foreach declaration"); 2508 return nullptr; 2509 } 2510 2511 Init *DeclName = StringInit::get(Lex.getCurStrVal()); 2512 Lex.Lex(); 2513 2514 // If a value is present, parse it. 2515 if (!consume(tgtok::equal)) { 2516 TokError("Expected '=' in foreach declaration"); 2517 return nullptr; 2518 } 2519 2520 RecTy *IterType = nullptr; 2521 SmallVector<unsigned, 16> Ranges; 2522 2523 switch (Lex.getCode()) { 2524 case tgtok::l_brace: { // '{' RangeList '}' 2525 Lex.Lex(); // eat the '{' 2526 ParseRangeList(Ranges); 2527 if (!consume(tgtok::r_brace)) { 2528 TokError("expected '}' at end of bit range list"); 2529 return nullptr; 2530 } 2531 break; 2532 } 2533 2534 default: { 2535 SMLoc ValueLoc = Lex.getLoc(); 2536 Init *I = ParseValue(nullptr); 2537 if (!I) 2538 return nullptr; 2539 2540 TypedInit *TI = dyn_cast<TypedInit>(I); 2541 if (TI && isa<ListRecTy>(TI->getType())) { 2542 ForeachListValue = I; 2543 IterType = cast<ListRecTy>(TI->getType())->getElementType(); 2544 break; 2545 } 2546 2547 if (TI) { 2548 if (ParseRangePiece(Ranges, TI)) 2549 return nullptr; 2550 break; 2551 } 2552 2553 std::string Type; 2554 if (TI) 2555 Type = (Twine("' of type '") + TI->getType()->getAsString()).str(); 2556 Error(ValueLoc, "expected a list, got '" + I->getAsString() + Type + "'"); 2557 if (CurMultiClass) { 2558 PrintNote({}, "references to multiclass template arguments cannot be " 2559 "resolved at this time"); 2560 } 2561 return nullptr; 2562 } 2563 } 2564 2565 2566 if (!Ranges.empty()) { 2567 assert(!IterType && "Type already initialized?"); 2568 IterType = IntRecTy::get(); 2569 std::vector<Init*> Values; 2570 for (unsigned R : Ranges) 2571 Values.push_back(IntInit::get(R)); 2572 ForeachListValue = ListInit::get(Values, IterType); 2573 } 2574 2575 if (!IterType) 2576 return nullptr; 2577 2578 return VarInit::get(DeclName, IterType); 2579 } 2580 2581 /// ParseTemplateArgList - Read a template argument list, which is a non-empty 2582 /// sequence of template-declarations in <>'s. If CurRec is non-null, these are 2583 /// template args for a def, which may or may not be in a multiclass. If null, 2584 /// these are the template args for a multiclass. 2585 /// 2586 /// TemplateArgList ::= '<' Declaration (',' Declaration)* '>' 2587 /// 2588 bool TGParser::ParseTemplateArgList(Record *CurRec) { 2589 assert(Lex.getCode() == tgtok::less && "Not a template arg list!"); 2590 Lex.Lex(); // eat the '<' 2591 2592 Record *TheRecToAddTo = CurRec ? CurRec : &CurMultiClass->Rec; 2593 2594 // Read the first declaration. 2595 Init *TemplArg = ParseDeclaration(CurRec, true/*templateargs*/); 2596 if (!TemplArg) 2597 return true; 2598 2599 TheRecToAddTo->addTemplateArg(TemplArg); 2600 2601 while (consume(tgtok::comma)) { 2602 // Read the following declarations. 2603 SMLoc Loc = Lex.getLoc(); 2604 TemplArg = ParseDeclaration(CurRec, true/*templateargs*/); 2605 if (!TemplArg) 2606 return true; 2607 2608 if (TheRecToAddTo->isTemplateArg(TemplArg)) 2609 return Error(Loc, "template argument with the same name has already been " 2610 "defined"); 2611 2612 TheRecToAddTo->addTemplateArg(TemplArg); 2613 } 2614 2615 if (!consume(tgtok::greater)) 2616 return TokError("expected '>' at end of template argument list"); 2617 return false; 2618 } 2619 2620 /// ParseBodyItem - Parse a single item at within the body of a def or class. 2621 /// 2622 /// BodyItem ::= Declaration ';' 2623 /// BodyItem ::= LET ID OptionalBitList '=' Value ';' 2624 /// BodyItem ::= Defvar 2625 bool TGParser::ParseBodyItem(Record *CurRec) { 2626 if (Lex.getCode() == tgtok::Defvar) 2627 return ParseDefvar(); 2628 2629 if (Lex.getCode() != tgtok::Let) { 2630 if (!ParseDeclaration(CurRec, false)) 2631 return true; 2632 2633 if (!consume(tgtok::semi)) 2634 return TokError("expected ';' after declaration"); 2635 return false; 2636 } 2637 2638 // LET ID OptionalRangeList '=' Value ';' 2639 if (Lex.Lex() != tgtok::Id) 2640 return TokError("expected field identifier after let"); 2641 2642 SMLoc IdLoc = Lex.getLoc(); 2643 StringInit *FieldName = StringInit::get(Lex.getCurStrVal()); 2644 Lex.Lex(); // eat the field name. 2645 2646 SmallVector<unsigned, 16> BitList; 2647 if (ParseOptionalBitList(BitList)) 2648 return true; 2649 std::reverse(BitList.begin(), BitList.end()); 2650 2651 if (!consume(tgtok::equal)) 2652 return TokError("expected '=' in let expression"); 2653 2654 RecordVal *Field = CurRec->getValue(FieldName); 2655 if (!Field) 2656 return TokError("Value '" + FieldName->getValue() + "' unknown!"); 2657 2658 RecTy *Type = Field->getType(); 2659 if (!BitList.empty() && isa<BitsRecTy>(Type)) { 2660 // When assigning to a subset of a 'bits' object, expect the RHS to have 2661 // the type of that subset instead of the type of the whole object. 2662 Type = BitsRecTy::get(BitList.size()); 2663 } 2664 2665 Init *Val = ParseValue(CurRec, Type); 2666 if (!Val) return true; 2667 2668 if (!consume(tgtok::semi)) 2669 return TokError("expected ';' after let expression"); 2670 2671 return SetValue(CurRec, IdLoc, FieldName, BitList, Val); 2672 } 2673 2674 /// ParseBody - Read the body of a class or def. Return true on error, false on 2675 /// success. 2676 /// 2677 /// Body ::= ';' 2678 /// Body ::= '{' BodyList '}' 2679 /// BodyList BodyItem* 2680 /// 2681 bool TGParser::ParseBody(Record *CurRec) { 2682 // If this is a null definition, just eat the semi and return. 2683 if (consume(tgtok::semi)) 2684 return false; 2685 2686 if (!consume(tgtok::l_brace)) 2687 return TokError("Expected ';' or '{' to start body"); 2688 2689 // An object body introduces a new scope for local variables. 2690 TGLocalVarScope *BodyScope = PushLocalScope(); 2691 2692 while (Lex.getCode() != tgtok::r_brace) 2693 if (ParseBodyItem(CurRec)) 2694 return true; 2695 2696 PopLocalScope(BodyScope); 2697 2698 // Eat the '}'. 2699 Lex.Lex(); 2700 return false; 2701 } 2702 2703 /// Apply the current let bindings to \a CurRec. 2704 /// \returns true on error, false otherwise. 2705 bool TGParser::ApplyLetStack(Record *CurRec) { 2706 for (SmallVectorImpl<LetRecord> &LetInfo : LetStack) 2707 for (LetRecord &LR : LetInfo) 2708 if (SetValue(CurRec, LR.Loc, LR.Name, LR.Bits, LR.Value)) 2709 return true; 2710 return false; 2711 } 2712 2713 bool TGParser::ApplyLetStack(RecordsEntry &Entry) { 2714 if (Entry.Rec) 2715 return ApplyLetStack(Entry.Rec.get()); 2716 2717 for (auto &E : Entry.Loop->Entries) { 2718 if (ApplyLetStack(E)) 2719 return true; 2720 } 2721 2722 return false; 2723 } 2724 2725 /// ParseObjectBody - Parse the body of a def or class. This consists of an 2726 /// optional ClassList followed by a Body. CurRec is the current def or class 2727 /// that is being parsed. 2728 /// 2729 /// ObjectBody ::= BaseClassList Body 2730 /// BaseClassList ::= /*empty*/ 2731 /// BaseClassList ::= ':' BaseClassListNE 2732 /// BaseClassListNE ::= SubClassRef (',' SubClassRef)* 2733 /// 2734 bool TGParser::ParseObjectBody(Record *CurRec) { 2735 // If there is a baseclass list, read it. 2736 if (consume(tgtok::colon)) { 2737 2738 // Read all of the subclasses. 2739 SubClassReference SubClass = ParseSubClassReference(CurRec, false); 2740 while (true) { 2741 // Check for error. 2742 if (!SubClass.Rec) return true; 2743 2744 // Add it. 2745 if (AddSubClass(CurRec, SubClass)) 2746 return true; 2747 2748 if (!consume(tgtok::comma)) 2749 break; 2750 SubClass = ParseSubClassReference(CurRec, false); 2751 } 2752 } 2753 2754 if (ApplyLetStack(CurRec)) 2755 return true; 2756 2757 return ParseBody(CurRec); 2758 } 2759 2760 /// ParseDef - Parse and return a top level or multiclass def, return the record 2761 /// corresponding to it. This returns null on error. 2762 /// 2763 /// DefInst ::= DEF ObjectName ObjectBody 2764 /// 2765 bool TGParser::ParseDef(MultiClass *CurMultiClass) { 2766 SMLoc DefLoc = Lex.getLoc(); 2767 assert(Lex.getCode() == tgtok::Def && "Unknown tok"); 2768 Lex.Lex(); // Eat the 'def' token. 2769 2770 // Parse ObjectName and make a record for it. 2771 std::unique_ptr<Record> CurRec; 2772 Init *Name = ParseObjectName(CurMultiClass); 2773 if (!Name) 2774 return true; 2775 2776 if (isa<UnsetInit>(Name)) 2777 CurRec = std::make_unique<Record>(Records.getNewAnonymousName(), DefLoc, Records, 2778 /*Anonymous=*/true); 2779 else 2780 CurRec = std::make_unique<Record>(Name, DefLoc, Records); 2781 2782 if (ParseObjectBody(CurRec.get())) 2783 return true; 2784 2785 return addEntry(std::move(CurRec)); 2786 } 2787 2788 /// ParseDefset - Parse a defset statement. 2789 /// 2790 /// Defset ::= DEFSET Type Id '=' '{' ObjectList '}' 2791 /// 2792 bool TGParser::ParseDefset() { 2793 assert(Lex.getCode() == tgtok::Defset); 2794 Lex.Lex(); // Eat the 'defset' token 2795 2796 DefsetRecord Defset; 2797 Defset.Loc = Lex.getLoc(); 2798 RecTy *Type = ParseType(); 2799 if (!Type) 2800 return true; 2801 if (!isa<ListRecTy>(Type)) 2802 return Error(Defset.Loc, "expected list type"); 2803 Defset.EltTy = cast<ListRecTy>(Type)->getElementType(); 2804 2805 if (Lex.getCode() != tgtok::Id) 2806 return TokError("expected identifier"); 2807 StringInit *DeclName = StringInit::get(Lex.getCurStrVal()); 2808 if (Records.getGlobal(DeclName->getValue())) 2809 return TokError("def or global variable of this name already exists"); 2810 2811 if (Lex.Lex() != tgtok::equal) // Eat the identifier 2812 return TokError("expected '='"); 2813 if (Lex.Lex() != tgtok::l_brace) // Eat the '=' 2814 return TokError("expected '{'"); 2815 SMLoc BraceLoc = Lex.getLoc(); 2816 Lex.Lex(); // Eat the '{' 2817 2818 Defsets.push_back(&Defset); 2819 bool Err = ParseObjectList(nullptr); 2820 Defsets.pop_back(); 2821 if (Err) 2822 return true; 2823 2824 if (!consume(tgtok::r_brace)) { 2825 TokError("expected '}' at end of defset"); 2826 return Error(BraceLoc, "to match this '{'"); 2827 } 2828 2829 Records.addExtraGlobal(DeclName->getValue(), 2830 ListInit::get(Defset.Elements, Defset.EltTy)); 2831 return false; 2832 } 2833 2834 /// ParseDefvar - Parse a defvar statement. 2835 /// 2836 /// Defvar ::= DEFVAR Id '=' Value ';' 2837 /// 2838 bool TGParser::ParseDefvar() { 2839 assert(Lex.getCode() == tgtok::Defvar); 2840 Lex.Lex(); // Eat the 'defvar' token 2841 2842 if (Lex.getCode() != tgtok::Id) 2843 return TokError("expected identifier"); 2844 StringInit *DeclName = StringInit::get(Lex.getCurStrVal()); 2845 if (CurLocalScope) { 2846 if (CurLocalScope->varAlreadyDefined(DeclName->getValue())) 2847 return TokError("local variable of this name already exists"); 2848 } else { 2849 if (Records.getGlobal(DeclName->getValue())) 2850 return TokError("def or global variable of this name already exists"); 2851 } 2852 2853 Lex.Lex(); 2854 if (!consume(tgtok::equal)) 2855 return TokError("expected '='"); 2856 2857 Init *Value = ParseValue(nullptr); 2858 if (!Value) 2859 return true; 2860 2861 if (!consume(tgtok::semi)) 2862 return TokError("expected ';'"); 2863 2864 if (CurLocalScope) 2865 CurLocalScope->addVar(DeclName->getValue(), Value); 2866 else 2867 Records.addExtraGlobal(DeclName->getValue(), Value); 2868 2869 return false; 2870 } 2871 2872 /// ParseForeach - Parse a for statement. Return the record corresponding 2873 /// to it. This returns true on error. 2874 /// 2875 /// Foreach ::= FOREACH Declaration IN '{ ObjectList '}' 2876 /// Foreach ::= FOREACH Declaration IN Object 2877 /// 2878 bool TGParser::ParseForeach(MultiClass *CurMultiClass) { 2879 SMLoc Loc = Lex.getLoc(); 2880 assert(Lex.getCode() == tgtok::Foreach && "Unknown tok"); 2881 Lex.Lex(); // Eat the 'for' token. 2882 2883 // Make a temporary object to record items associated with the for 2884 // loop. 2885 Init *ListValue = nullptr; 2886 VarInit *IterName = ParseForeachDeclaration(ListValue); 2887 if (!IterName) 2888 return TokError("expected declaration in for"); 2889 2890 if (!consume(tgtok::In)) 2891 return TokError("Unknown tok"); 2892 2893 // Create a loop object and remember it. 2894 Loops.push_back(std::make_unique<ForeachLoop>(Loc, IterName, ListValue)); 2895 2896 // A foreach loop introduces a new scope for local variables. 2897 TGLocalVarScope *ForeachScope = PushLocalScope(); 2898 2899 if (Lex.getCode() != tgtok::l_brace) { 2900 // FOREACH Declaration IN Object 2901 if (ParseObject(CurMultiClass)) 2902 return true; 2903 } else { 2904 SMLoc BraceLoc = Lex.getLoc(); 2905 // Otherwise, this is a group foreach. 2906 Lex.Lex(); // eat the '{'. 2907 2908 // Parse the object list. 2909 if (ParseObjectList(CurMultiClass)) 2910 return true; 2911 2912 if (!consume(tgtok::r_brace)) { 2913 TokError("expected '}' at end of foreach command"); 2914 return Error(BraceLoc, "to match this '{'"); 2915 } 2916 } 2917 2918 PopLocalScope(ForeachScope); 2919 2920 // Resolve the loop or store it for later resolution. 2921 std::unique_ptr<ForeachLoop> Loop = std::move(Loops.back()); 2922 Loops.pop_back(); 2923 2924 return addEntry(std::move(Loop)); 2925 } 2926 2927 /// ParseIf - Parse an if statement. 2928 /// 2929 /// If ::= IF Value THEN IfBody 2930 /// If ::= IF Value THEN IfBody ELSE IfBody 2931 /// 2932 bool TGParser::ParseIf(MultiClass *CurMultiClass) { 2933 SMLoc Loc = Lex.getLoc(); 2934 assert(Lex.getCode() == tgtok::If && "Unknown tok"); 2935 Lex.Lex(); // Eat the 'if' token. 2936 2937 // Make a temporary object to record items associated with the for 2938 // loop. 2939 Init *Condition = ParseValue(nullptr); 2940 if (!Condition) 2941 return true; 2942 2943 if (!consume(tgtok::Then)) 2944 return TokError("Unknown tok"); 2945 2946 // We have to be able to save if statements to execute later, and they have 2947 // to live on the same stack as foreach loops. The simplest implementation 2948 // technique is to convert each 'then' or 'else' clause *into* a foreach 2949 // loop, over a list of length 0 or 1 depending on the condition, and with no 2950 // iteration variable being assigned. 2951 2952 ListInit *EmptyList = ListInit::get({}, BitRecTy::get()); 2953 ListInit *SingletonList = ListInit::get({BitInit::get(1)}, BitRecTy::get()); 2954 RecTy *BitListTy = ListRecTy::get(BitRecTy::get()); 2955 2956 // The foreach containing the then-clause selects SingletonList if 2957 // the condition is true. 2958 Init *ThenClauseList = 2959 TernOpInit::get(TernOpInit::IF, Condition, SingletonList, EmptyList, 2960 BitListTy) 2961 ->Fold(nullptr); 2962 Loops.push_back(std::make_unique<ForeachLoop>(Loc, nullptr, ThenClauseList)); 2963 2964 if (ParseIfBody(CurMultiClass, "then")) 2965 return true; 2966 2967 std::unique_ptr<ForeachLoop> Loop = std::move(Loops.back()); 2968 Loops.pop_back(); 2969 2970 if (addEntry(std::move(Loop))) 2971 return true; 2972 2973 // Now look for an optional else clause. The if-else syntax has the usual 2974 // dangling-else ambiguity, and by greedily matching an else here if we can, 2975 // we implement the usual resolution of pairing with the innermost unmatched 2976 // if. 2977 if (consume(tgtok::ElseKW)) { 2978 // The foreach containing the else-clause uses the same pair of lists as 2979 // above, but this time, selects SingletonList if the condition is *false*. 2980 Init *ElseClauseList = 2981 TernOpInit::get(TernOpInit::IF, Condition, EmptyList, SingletonList, 2982 BitListTy) 2983 ->Fold(nullptr); 2984 Loops.push_back( 2985 std::make_unique<ForeachLoop>(Loc, nullptr, ElseClauseList)); 2986 2987 if (ParseIfBody(CurMultiClass, "else")) 2988 return true; 2989 2990 Loop = std::move(Loops.back()); 2991 Loops.pop_back(); 2992 2993 if (addEntry(std::move(Loop))) 2994 return true; 2995 } 2996 2997 return false; 2998 } 2999 3000 /// ParseIfBody - Parse the then-clause or else-clause of an if statement. 3001 /// 3002 /// IfBody ::= Object 3003 /// IfBody ::= '{' ObjectList '}' 3004 /// 3005 bool TGParser::ParseIfBody(MultiClass *CurMultiClass, StringRef Kind) { 3006 TGLocalVarScope *BodyScope = PushLocalScope(); 3007 3008 if (Lex.getCode() != tgtok::l_brace) { 3009 // A single object. 3010 if (ParseObject(CurMultiClass)) 3011 return true; 3012 } else { 3013 SMLoc BraceLoc = Lex.getLoc(); 3014 // A braced block. 3015 Lex.Lex(); // eat the '{'. 3016 3017 // Parse the object list. 3018 if (ParseObjectList(CurMultiClass)) 3019 return true; 3020 3021 if (!consume(tgtok::r_brace)) { 3022 TokError("expected '}' at end of '" + Kind + "' clause"); 3023 return Error(BraceLoc, "to match this '{'"); 3024 } 3025 } 3026 3027 PopLocalScope(BodyScope); 3028 return false; 3029 } 3030 3031 /// ParseClass - Parse a tblgen class definition. 3032 /// 3033 /// ClassInst ::= CLASS ID TemplateArgList? ObjectBody 3034 /// 3035 bool TGParser::ParseClass() { 3036 assert(Lex.getCode() == tgtok::Class && "Unexpected token!"); 3037 Lex.Lex(); 3038 3039 if (Lex.getCode() != tgtok::Id) 3040 return TokError("expected class name after 'class' keyword"); 3041 3042 Record *CurRec = Records.getClass(Lex.getCurStrVal()); 3043 if (CurRec) { 3044 // If the body was previously defined, this is an error. 3045 if (!CurRec->getValues().empty() || 3046 !CurRec->getSuperClasses().empty() || 3047 !CurRec->getTemplateArgs().empty()) 3048 return TokError("Class '" + CurRec->getNameInitAsString() + 3049 "' already defined"); 3050 } else { 3051 // If this is the first reference to this class, create and add it. 3052 auto NewRec = 3053 std::make_unique<Record>(Lex.getCurStrVal(), Lex.getLoc(), Records, 3054 /*Class=*/true); 3055 CurRec = NewRec.get(); 3056 Records.addClass(std::move(NewRec)); 3057 } 3058 Lex.Lex(); // eat the name. 3059 3060 // If there are template args, parse them. 3061 if (Lex.getCode() == tgtok::less) 3062 if (ParseTemplateArgList(CurRec)) 3063 return true; 3064 3065 return ParseObjectBody(CurRec); 3066 } 3067 3068 /// ParseLetList - Parse a non-empty list of assignment expressions into a list 3069 /// of LetRecords. 3070 /// 3071 /// LetList ::= LetItem (',' LetItem)* 3072 /// LetItem ::= ID OptionalRangeList '=' Value 3073 /// 3074 void TGParser::ParseLetList(SmallVectorImpl<LetRecord> &Result) { 3075 do { 3076 if (Lex.getCode() != tgtok::Id) { 3077 TokError("expected identifier in let definition"); 3078 Result.clear(); 3079 return; 3080 } 3081 3082 StringInit *Name = StringInit::get(Lex.getCurStrVal()); 3083 SMLoc NameLoc = Lex.getLoc(); 3084 Lex.Lex(); // Eat the identifier. 3085 3086 // Check for an optional RangeList. 3087 SmallVector<unsigned, 16> Bits; 3088 if (ParseOptionalRangeList(Bits)) { 3089 Result.clear(); 3090 return; 3091 } 3092 std::reverse(Bits.begin(), Bits.end()); 3093 3094 if (!consume(tgtok::equal)) { 3095 TokError("expected '=' in let expression"); 3096 Result.clear(); 3097 return; 3098 } 3099 3100 Init *Val = ParseValue(nullptr); 3101 if (!Val) { 3102 Result.clear(); 3103 return; 3104 } 3105 3106 // Now that we have everything, add the record. 3107 Result.emplace_back(Name, Bits, Val, NameLoc); 3108 } while (consume(tgtok::comma)); 3109 } 3110 3111 /// ParseTopLevelLet - Parse a 'let' at top level. This can be a couple of 3112 /// different related productions. This works inside multiclasses too. 3113 /// 3114 /// Object ::= LET LetList IN '{' ObjectList '}' 3115 /// Object ::= LET LetList IN Object 3116 /// 3117 bool TGParser::ParseTopLevelLet(MultiClass *CurMultiClass) { 3118 assert(Lex.getCode() == tgtok::Let && "Unexpected token"); 3119 Lex.Lex(); 3120 3121 // Add this entry to the let stack. 3122 SmallVector<LetRecord, 8> LetInfo; 3123 ParseLetList(LetInfo); 3124 if (LetInfo.empty()) return true; 3125 LetStack.push_back(std::move(LetInfo)); 3126 3127 if (!consume(tgtok::In)) 3128 return TokError("expected 'in' at end of top-level 'let'"); 3129 3130 TGLocalVarScope *LetScope = PushLocalScope(); 3131 3132 // If this is a scalar let, just handle it now 3133 if (Lex.getCode() != tgtok::l_brace) { 3134 // LET LetList IN Object 3135 if (ParseObject(CurMultiClass)) 3136 return true; 3137 } else { // Object ::= LETCommand '{' ObjectList '}' 3138 SMLoc BraceLoc = Lex.getLoc(); 3139 // Otherwise, this is a group let. 3140 Lex.Lex(); // eat the '{'. 3141 3142 // Parse the object list. 3143 if (ParseObjectList(CurMultiClass)) 3144 return true; 3145 3146 if (!consume(tgtok::r_brace)) { 3147 TokError("expected '}' at end of top level let command"); 3148 return Error(BraceLoc, "to match this '{'"); 3149 } 3150 } 3151 3152 PopLocalScope(LetScope); 3153 3154 // Outside this let scope, this let block is not active. 3155 LetStack.pop_back(); 3156 return false; 3157 } 3158 3159 /// ParseMultiClass - Parse a multiclass definition. 3160 /// 3161 /// MultiClassInst ::= MULTICLASS ID TemplateArgList? 3162 /// ':' BaseMultiClassList '{' MultiClassObject+ '}' 3163 /// MultiClassObject ::= DefInst 3164 /// MultiClassObject ::= MultiClassInst 3165 /// MultiClassObject ::= DefMInst 3166 /// MultiClassObject ::= LETCommand '{' ObjectList '}' 3167 /// MultiClassObject ::= LETCommand Object 3168 /// 3169 bool TGParser::ParseMultiClass() { 3170 assert(Lex.getCode() == tgtok::MultiClass && "Unexpected token"); 3171 Lex.Lex(); // Eat the multiclass token. 3172 3173 if (Lex.getCode() != tgtok::Id) 3174 return TokError("expected identifier after multiclass for name"); 3175 std::string Name = Lex.getCurStrVal(); 3176 3177 auto Result = 3178 MultiClasses.insert(std::make_pair(Name, 3179 std::make_unique<MultiClass>(Name, Lex.getLoc(),Records))); 3180 3181 if (!Result.second) 3182 return TokError("multiclass '" + Name + "' already defined"); 3183 3184 CurMultiClass = Result.first->second.get(); 3185 Lex.Lex(); // Eat the identifier. 3186 3187 // If there are template args, parse them. 3188 if (Lex.getCode() == tgtok::less) 3189 if (ParseTemplateArgList(nullptr)) 3190 return true; 3191 3192 bool inherits = false; 3193 3194 // If there are submulticlasses, parse them. 3195 if (consume(tgtok::colon)) { 3196 inherits = true; 3197 3198 // Read all of the submulticlasses. 3199 SubMultiClassReference SubMultiClass = 3200 ParseSubMultiClassReference(CurMultiClass); 3201 while (true) { 3202 // Check for error. 3203 if (!SubMultiClass.MC) return true; 3204 3205 // Add it. 3206 if (AddSubMultiClass(CurMultiClass, SubMultiClass)) 3207 return true; 3208 3209 if (!consume(tgtok::comma)) 3210 break; 3211 SubMultiClass = ParseSubMultiClassReference(CurMultiClass); 3212 } 3213 } 3214 3215 if (Lex.getCode() != tgtok::l_brace) { 3216 if (!inherits) 3217 return TokError("expected '{' in multiclass definition"); 3218 if (!consume(tgtok::semi)) 3219 return TokError("expected ';' in multiclass definition"); 3220 } else { 3221 if (Lex.Lex() == tgtok::r_brace) // eat the '{'. 3222 return TokError("multiclass must contain at least one def"); 3223 3224 // A multiclass body introduces a new scope for local variables. 3225 TGLocalVarScope *MulticlassScope = PushLocalScope(); 3226 3227 while (Lex.getCode() != tgtok::r_brace) { 3228 switch (Lex.getCode()) { 3229 default: 3230 return TokError("expected 'let', 'def', 'defm', 'defvar', 'foreach' " 3231 "or 'if' in multiclass body"); 3232 case tgtok::Let: 3233 case tgtok::Def: 3234 case tgtok::Defm: 3235 case tgtok::Defvar: 3236 case tgtok::Foreach: 3237 case tgtok::If: 3238 if (ParseObject(CurMultiClass)) 3239 return true; 3240 break; 3241 } 3242 } 3243 Lex.Lex(); // eat the '}'. 3244 3245 PopLocalScope(MulticlassScope); 3246 } 3247 3248 CurMultiClass = nullptr; 3249 return false; 3250 } 3251 3252 /// ParseDefm - Parse the instantiation of a multiclass. 3253 /// 3254 /// DefMInst ::= DEFM ID ':' DefmSubClassRef ';' 3255 /// 3256 bool TGParser::ParseDefm(MultiClass *CurMultiClass) { 3257 assert(Lex.getCode() == tgtok::Defm && "Unexpected token!"); 3258 Lex.Lex(); // eat the defm 3259 3260 Init *DefmName = ParseObjectName(CurMultiClass); 3261 if (!DefmName) 3262 return true; 3263 if (isa<UnsetInit>(DefmName)) { 3264 DefmName = Records.getNewAnonymousName(); 3265 if (CurMultiClass) 3266 DefmName = BinOpInit::getStrConcat( 3267 VarInit::get(QualifiedNameOfImplicitName(CurMultiClass), 3268 StringRecTy::get()), 3269 DefmName); 3270 } 3271 3272 if (Lex.getCode() != tgtok::colon) 3273 return TokError("expected ':' after defm identifier"); 3274 3275 // Keep track of the new generated record definitions. 3276 std::vector<RecordsEntry> NewEntries; 3277 3278 // This record also inherits from a regular class (non-multiclass)? 3279 bool InheritFromClass = false; 3280 3281 // eat the colon. 3282 Lex.Lex(); 3283 3284 SMLoc SubClassLoc = Lex.getLoc(); 3285 SubClassReference Ref = ParseSubClassReference(nullptr, true); 3286 3287 while (true) { 3288 if (!Ref.Rec) return true; 3289 3290 // To instantiate a multiclass, we need to first get the multiclass, then 3291 // instantiate each def contained in the multiclass with the SubClassRef 3292 // template parameters. 3293 MultiClass *MC = MultiClasses[std::string(Ref.Rec->getName())].get(); 3294 assert(MC && "Didn't lookup multiclass correctly?"); 3295 ArrayRef<Init*> TemplateVals = Ref.TemplateArgs; 3296 3297 // Verify that the correct number of template arguments were specified. 3298 ArrayRef<Init *> TArgs = MC->Rec.getTemplateArgs(); 3299 if (TArgs.size() < TemplateVals.size()) 3300 return Error(SubClassLoc, 3301 "more template args specified than multiclass expects"); 3302 3303 SubstStack Substs; 3304 for (unsigned i = 0, e = TArgs.size(); i != e; ++i) { 3305 if (i < TemplateVals.size()) { 3306 Substs.emplace_back(TArgs[i], TemplateVals[i]); 3307 } else { 3308 Init *Default = MC->Rec.getValue(TArgs[i])->getValue(); 3309 if (!Default->isComplete()) { 3310 return Error(SubClassLoc, 3311 "value not specified for template argument #" + 3312 Twine(i) + " (" + TArgs[i]->getAsUnquotedString() + 3313 ") of multiclass '" + MC->Rec.getNameInitAsString() + 3314 "'"); 3315 } 3316 Substs.emplace_back(TArgs[i], Default); 3317 } 3318 } 3319 3320 Substs.emplace_back(QualifiedNameOfImplicitName(MC), DefmName); 3321 3322 if (resolve(MC->Entries, Substs, CurMultiClass == nullptr, &NewEntries, 3323 &SubClassLoc)) 3324 return true; 3325 3326 if (!consume(tgtok::comma)) 3327 break; 3328 3329 if (Lex.getCode() != tgtok::Id) 3330 return TokError("expected identifier"); 3331 3332 SubClassLoc = Lex.getLoc(); 3333 3334 // A defm can inherit from regular classes (non-multiclass) as 3335 // long as they come in the end of the inheritance list. 3336 InheritFromClass = (Records.getClass(Lex.getCurStrVal()) != nullptr); 3337 3338 if (InheritFromClass) 3339 break; 3340 3341 Ref = ParseSubClassReference(nullptr, true); 3342 } 3343 3344 if (InheritFromClass) { 3345 // Process all the classes to inherit as if they were part of a 3346 // regular 'def' and inherit all record values. 3347 SubClassReference SubClass = ParseSubClassReference(nullptr, false); 3348 while (true) { 3349 // Check for error. 3350 if (!SubClass.Rec) return true; 3351 3352 // Get the expanded definition prototypes and teach them about 3353 // the record values the current class to inherit has 3354 for (auto &E : NewEntries) { 3355 // Add it. 3356 if (AddSubClass(E, SubClass)) 3357 return true; 3358 } 3359 3360 if (!consume(tgtok::comma)) 3361 break; 3362 SubClass = ParseSubClassReference(nullptr, false); 3363 } 3364 } 3365 3366 for (auto &E : NewEntries) { 3367 if (ApplyLetStack(E)) 3368 return true; 3369 3370 addEntry(std::move(E)); 3371 } 3372 3373 if (!consume(tgtok::semi)) 3374 return TokError("expected ';' at end of defm"); 3375 3376 return false; 3377 } 3378 3379 /// ParseObject 3380 /// Object ::= ClassInst 3381 /// Object ::= DefInst 3382 /// Object ::= MultiClassInst 3383 /// Object ::= DefMInst 3384 /// Object ::= LETCommand '{' ObjectList '}' 3385 /// Object ::= LETCommand Object 3386 /// Object ::= Defset 3387 /// Object ::= Defvar 3388 bool TGParser::ParseObject(MultiClass *MC) { 3389 switch (Lex.getCode()) { 3390 default: 3391 return TokError("Expected class, def, defm, defset, multiclass, let, " 3392 "foreach or if"); 3393 case tgtok::Let: return ParseTopLevelLet(MC); 3394 case tgtok::Def: return ParseDef(MC); 3395 case tgtok::Foreach: return ParseForeach(MC); 3396 case tgtok::If: return ParseIf(MC); 3397 case tgtok::Defm: return ParseDefm(MC); 3398 case tgtok::Defset: 3399 if (MC) 3400 return TokError("defset is not allowed inside multiclass"); 3401 return ParseDefset(); 3402 case tgtok::Defvar: 3403 return ParseDefvar(); 3404 case tgtok::Class: 3405 if (MC) 3406 return TokError("class is not allowed inside multiclass"); 3407 if (!Loops.empty()) 3408 return TokError("class is not allowed inside foreach loop"); 3409 return ParseClass(); 3410 case tgtok::MultiClass: 3411 if (!Loops.empty()) 3412 return TokError("multiclass is not allowed inside foreach loop"); 3413 return ParseMultiClass(); 3414 } 3415 } 3416 3417 /// ParseObjectList 3418 /// ObjectList :== Object* 3419 bool TGParser::ParseObjectList(MultiClass *MC) { 3420 while (isObjectStart(Lex.getCode())) { 3421 if (ParseObject(MC)) 3422 return true; 3423 } 3424 return false; 3425 } 3426 3427 bool TGParser::ParseFile() { 3428 Lex.Lex(); // Prime the lexer. 3429 if (ParseObjectList()) return true; 3430 3431 // If we have unread input at the end of the file, report it. 3432 if (Lex.getCode() == tgtok::Eof) 3433 return false; 3434 3435 return TokError("Unexpected input at top level"); 3436 } 3437 3438 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 3439 LLVM_DUMP_METHOD void RecordsEntry::dump() const { 3440 if (Loop) 3441 Loop->dump(); 3442 if (Rec) 3443 Rec->dump(); 3444 } 3445 3446 LLVM_DUMP_METHOD void ForeachLoop::dump() const { 3447 errs() << "foreach " << IterVar->getAsString() << " = " 3448 << ListValue->getAsString() << " in {\n"; 3449 3450 for (const auto &E : Entries) 3451 E.dump(); 3452 3453 errs() << "}\n"; 3454 } 3455 3456 LLVM_DUMP_METHOD void MultiClass::dump() const { 3457 errs() << "Record:\n"; 3458 Rec.dump(); 3459 3460 errs() << "Defs:\n"; 3461 for (const auto &E : Entries) 3462 E.dump(); 3463 } 3464 #endif 3465