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: 803 case tgtok::Code: Lex.Lex(); return StringRecTy::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 case tgtok::XLe: 1152 case tgtok::XLt: 1153 case tgtok::XGe: 1154 case tgtok::XGt: 1155 Type = BitRecTy::get(); 1156 // ArgType for the comparison operators is not yet known. 1157 break; 1158 case tgtok::XListConcat: 1159 // We don't know the list type until we parse the first argument 1160 ArgType = ItemType; 1161 break; 1162 case tgtok::XListSplat: 1163 // Can't do any typechecking until we parse the first argument. 1164 break; 1165 case tgtok::XStrConcat: 1166 Type = StringRecTy::get(); 1167 ArgType = StringRecTy::get(); 1168 break; 1169 case tgtok::XInterleave: 1170 Type = StringRecTy::get(); 1171 // The first argument type is not yet known. 1172 } 1173 1174 if (Type && ItemType && !Type->typeIsConvertibleTo(ItemType)) { 1175 Error(OpLoc, Twine("expected value of type '") + 1176 ItemType->getAsString() + "', got '" + 1177 Type->getAsString() + "'"); 1178 return nullptr; 1179 } 1180 1181 if (!consume(tgtok::l_paren)) { 1182 TokError("expected '(' after binary operator"); 1183 return nullptr; 1184 } 1185 1186 SmallVector<Init*, 2> InitList; 1187 1188 // Note that this loop consumes an arbitrary number of arguments. 1189 // The actual count is checked later. 1190 for (;;) { 1191 SMLoc InitLoc = Lex.getLoc(); 1192 InitList.push_back(ParseValue(CurRec, ArgType)); 1193 if (!InitList.back()) return nullptr; 1194 1195 TypedInit *InitListBack = dyn_cast<TypedInit>(InitList.back()); 1196 if (!InitListBack) { 1197 Error(OpLoc, Twine("expected value to be a typed value, got '" + 1198 InitList.back()->getAsString() + "'")); 1199 return nullptr; 1200 } 1201 RecTy *ListType = InitListBack->getType(); 1202 1203 if (!ArgType) { 1204 // Argument type must be determined from the argument itself. 1205 ArgType = ListType; 1206 1207 switch (Code) { 1208 case BinOpInit::LISTCONCAT: 1209 if (!isa<ListRecTy>(ArgType)) { 1210 Error(InitLoc, Twine("expected a list, got value of type '") + 1211 ArgType->getAsString() + "'"); 1212 return nullptr; 1213 } 1214 break; 1215 case BinOpInit::LISTSPLAT: 1216 if (ItemType && InitList.size() == 1) { 1217 if (!isa<ListRecTy>(ItemType)) { 1218 Error(OpLoc, 1219 Twine("expected output type to be a list, got type '") + 1220 ItemType->getAsString() + "'"); 1221 return nullptr; 1222 } 1223 if (!ArgType->getListTy()->typeIsConvertibleTo(ItemType)) { 1224 Error(OpLoc, Twine("expected first arg type to be '") + 1225 ArgType->getAsString() + 1226 "', got value of type '" + 1227 cast<ListRecTy>(ItemType) 1228 ->getElementType() 1229 ->getAsString() + 1230 "'"); 1231 return nullptr; 1232 } 1233 } 1234 if (InitList.size() == 2 && !isa<IntRecTy>(ArgType)) { 1235 Error(InitLoc, Twine("expected second parameter to be an int, got " 1236 "value of type '") + 1237 ArgType->getAsString() + "'"); 1238 return nullptr; 1239 } 1240 ArgType = nullptr; // Broken invariant: types not identical. 1241 break; 1242 case BinOpInit::EQ: 1243 case BinOpInit::NE: 1244 if (!ArgType->typeIsConvertibleTo(IntRecTy::get()) && 1245 !ArgType->typeIsConvertibleTo(StringRecTy::get()) && 1246 !ArgType->typeIsConvertibleTo(RecordRecTy::get({}))) { 1247 Error(InitLoc, Twine("expected bit, bits, int, string, or record; " 1248 "got value of type '") + ArgType->getAsString() + 1249 "'"); 1250 return nullptr; 1251 } 1252 break; 1253 case BinOpInit::LE: 1254 case BinOpInit::LT: 1255 case BinOpInit::GE: 1256 case BinOpInit::GT: 1257 if (!ArgType->typeIsConvertibleTo(IntRecTy::get()) && 1258 !ArgType->typeIsConvertibleTo(StringRecTy::get())) { 1259 Error(InitLoc, Twine("expected bit, bits, int, or string; " 1260 "got value of type '") + ArgType->getAsString() + 1261 "'"); 1262 return nullptr; 1263 } 1264 break; 1265 case BinOpInit::INTERLEAVE: 1266 switch (InitList.size()) { 1267 case 1: // First argument must be a list of strings or integers. 1268 if (ArgType != StringRecTy::get()->getListTy() && 1269 !ArgType->typeIsConvertibleTo(IntRecTy::get()->getListTy())) { 1270 Error(InitLoc, Twine("expected list of string, int, bits, or bit; " 1271 "got value of type '") + 1272 ArgType->getAsString() + "'"); 1273 return nullptr; 1274 } 1275 break; 1276 case 2: // Second argument must be a string. 1277 if (!isa<StringRecTy>(ArgType)) { 1278 Error(InitLoc, Twine("expected second argument to be a string, " 1279 "got value of type '") + 1280 ArgType->getAsString() + "'"); 1281 return nullptr; 1282 } 1283 break; 1284 default: ; 1285 } 1286 ArgType = nullptr; // Broken invariant: types not identical. 1287 break; 1288 default: llvm_unreachable("other ops have fixed argument types"); 1289 } 1290 1291 } else { 1292 // Desired argument type is a known and in ArgType. 1293 RecTy *Resolved = resolveTypes(ArgType, ListType); 1294 if (!Resolved) { 1295 Error(InitLoc, Twine("expected value of type '") + 1296 ArgType->getAsString() + "', got '" + 1297 ListType->getAsString() + "'"); 1298 return nullptr; 1299 } 1300 if (Code != BinOpInit::ADD && Code != BinOpInit::SUB && 1301 Code != BinOpInit::AND && Code != BinOpInit::OR && 1302 Code != BinOpInit::XOR && Code != BinOpInit::SRA && 1303 Code != BinOpInit::SRL && Code != BinOpInit::SHL && 1304 Code != BinOpInit::MUL) 1305 ArgType = Resolved; 1306 } 1307 1308 // Deal with BinOps whose arguments have different types, by 1309 // rewriting ArgType in between them. 1310 switch (Code) { 1311 case BinOpInit::SETDAGOP: 1312 // After parsing the first dag argument, switch to expecting 1313 // a record, with no restriction on its superclasses. 1314 ArgType = RecordRecTy::get({}); 1315 break; 1316 default: 1317 break; 1318 } 1319 1320 if (!consume(tgtok::comma)) 1321 break; 1322 } 1323 1324 if (!consume(tgtok::r_paren)) { 1325 TokError("expected ')' in operator"); 1326 return nullptr; 1327 } 1328 1329 // listconcat returns a list with type of the argument. 1330 if (Code == BinOpInit::LISTCONCAT) 1331 Type = ArgType; 1332 // listsplat returns a list of type of the *first* argument. 1333 if (Code == BinOpInit::LISTSPLAT) 1334 Type = cast<TypedInit>(InitList.front())->getType()->getListTy(); 1335 1336 // We allow multiple operands to associative operators like !strconcat as 1337 // shorthand for nesting them. 1338 if (Code == BinOpInit::STRCONCAT || Code == BinOpInit::LISTCONCAT || 1339 Code == BinOpInit::CONCAT || Code == BinOpInit::ADD || 1340 Code == BinOpInit::AND || Code == BinOpInit::OR || 1341 Code == BinOpInit::XOR || Code == BinOpInit::MUL) { 1342 while (InitList.size() > 2) { 1343 Init *RHS = InitList.pop_back_val(); 1344 RHS = (BinOpInit::get(Code, InitList.back(), RHS, Type))->Fold(CurRec); 1345 InitList.back() = RHS; 1346 } 1347 } 1348 1349 if (InitList.size() == 2) 1350 return (BinOpInit::get(Code, InitList[0], InitList[1], Type)) 1351 ->Fold(CurRec); 1352 1353 Error(OpLoc, "expected two operands to operator"); 1354 return nullptr; 1355 } 1356 1357 case tgtok::XForEach: 1358 case tgtok::XFilter: { 1359 return ParseOperationForEachFilter(CurRec, ItemType); 1360 } 1361 1362 case tgtok::XDag: 1363 case tgtok::XIf: 1364 case tgtok::XSubst: { // Value ::= !ternop '(' Value ',' Value ',' Value ')' 1365 TernOpInit::TernaryOp Code; 1366 RecTy *Type = nullptr; 1367 1368 tgtok::TokKind LexCode = Lex.getCode(); 1369 Lex.Lex(); // eat the operation 1370 switch (LexCode) { 1371 default: llvm_unreachable("Unhandled code!"); 1372 case tgtok::XDag: 1373 Code = TernOpInit::DAG; 1374 Type = DagRecTy::get(); 1375 ItemType = nullptr; 1376 break; 1377 case tgtok::XIf: 1378 Code = TernOpInit::IF; 1379 break; 1380 case tgtok::XSubst: 1381 Code = TernOpInit::SUBST; 1382 break; 1383 } 1384 if (!consume(tgtok::l_paren)) { 1385 TokError("expected '(' after ternary operator"); 1386 return nullptr; 1387 } 1388 1389 Init *LHS = ParseValue(CurRec); 1390 if (!LHS) return nullptr; 1391 1392 if (!consume(tgtok::comma)) { 1393 TokError("expected ',' in ternary operator"); 1394 return nullptr; 1395 } 1396 1397 SMLoc MHSLoc = Lex.getLoc(); 1398 Init *MHS = ParseValue(CurRec, ItemType); 1399 if (!MHS) 1400 return nullptr; 1401 1402 if (!consume(tgtok::comma)) { 1403 TokError("expected ',' in ternary operator"); 1404 return nullptr; 1405 } 1406 1407 SMLoc RHSLoc = Lex.getLoc(); 1408 Init *RHS = ParseValue(CurRec, ItemType); 1409 if (!RHS) 1410 return nullptr; 1411 1412 if (!consume(tgtok::r_paren)) { 1413 TokError("expected ')' in binary operator"); 1414 return nullptr; 1415 } 1416 1417 switch (LexCode) { 1418 default: llvm_unreachable("Unhandled code!"); 1419 case tgtok::XDag: { 1420 TypedInit *MHSt = dyn_cast<TypedInit>(MHS); 1421 if (!MHSt && !isa<UnsetInit>(MHS)) { 1422 Error(MHSLoc, "could not determine type of the child list in !dag"); 1423 return nullptr; 1424 } 1425 if (MHSt && !isa<ListRecTy>(MHSt->getType())) { 1426 Error(MHSLoc, Twine("expected list of children, got type '") + 1427 MHSt->getType()->getAsString() + "'"); 1428 return nullptr; 1429 } 1430 1431 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1432 if (!RHSt && !isa<UnsetInit>(RHS)) { 1433 Error(RHSLoc, "could not determine type of the name list in !dag"); 1434 return nullptr; 1435 } 1436 if (RHSt && StringRecTy::get()->getListTy() != RHSt->getType()) { 1437 Error(RHSLoc, Twine("expected list<string>, got type '") + 1438 RHSt->getType()->getAsString() + "'"); 1439 return nullptr; 1440 } 1441 1442 if (!MHSt && !RHSt) { 1443 Error(MHSLoc, 1444 "cannot have both unset children and unset names in !dag"); 1445 return nullptr; 1446 } 1447 break; 1448 } 1449 case tgtok::XIf: { 1450 RecTy *MHSTy = nullptr; 1451 RecTy *RHSTy = nullptr; 1452 1453 if (TypedInit *MHSt = dyn_cast<TypedInit>(MHS)) 1454 MHSTy = MHSt->getType(); 1455 if (BitsInit *MHSbits = dyn_cast<BitsInit>(MHS)) 1456 MHSTy = BitsRecTy::get(MHSbits->getNumBits()); 1457 if (isa<BitInit>(MHS)) 1458 MHSTy = BitRecTy::get(); 1459 1460 if (TypedInit *RHSt = dyn_cast<TypedInit>(RHS)) 1461 RHSTy = RHSt->getType(); 1462 if (BitsInit *RHSbits = dyn_cast<BitsInit>(RHS)) 1463 RHSTy = BitsRecTy::get(RHSbits->getNumBits()); 1464 if (isa<BitInit>(RHS)) 1465 RHSTy = BitRecTy::get(); 1466 1467 // For UnsetInit, it's typed from the other hand. 1468 if (isa<UnsetInit>(MHS)) 1469 MHSTy = RHSTy; 1470 if (isa<UnsetInit>(RHS)) 1471 RHSTy = MHSTy; 1472 1473 if (!MHSTy || !RHSTy) { 1474 TokError("could not get type for !if"); 1475 return nullptr; 1476 } 1477 1478 Type = resolveTypes(MHSTy, RHSTy); 1479 if (!Type) { 1480 TokError(Twine("inconsistent types '") + MHSTy->getAsString() + 1481 "' and '" + RHSTy->getAsString() + "' for !if"); 1482 return nullptr; 1483 } 1484 break; 1485 } 1486 case tgtok::XSubst: { 1487 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1488 if (!RHSt) { 1489 TokError("could not get type for !subst"); 1490 return nullptr; 1491 } 1492 Type = RHSt->getType(); 1493 break; 1494 } 1495 } 1496 return (TernOpInit::get(Code, LHS, MHS, RHS, Type))->Fold(CurRec); 1497 } 1498 1499 case tgtok::XCond: 1500 return ParseOperationCond(CurRec, ItemType); 1501 1502 case tgtok::XFoldl: { 1503 // Value ::= !foldl '(' Value ',' Value ',' Id ',' Id ',' Expr ')' 1504 Lex.Lex(); // eat the operation 1505 if (!consume(tgtok::l_paren)) { 1506 TokError("expected '(' after !foldl"); 1507 return nullptr; 1508 } 1509 1510 Init *StartUntyped = ParseValue(CurRec); 1511 if (!StartUntyped) 1512 return nullptr; 1513 1514 TypedInit *Start = dyn_cast<TypedInit>(StartUntyped); 1515 if (!Start) { 1516 TokError(Twine("could not get type of !foldl start: '") + 1517 StartUntyped->getAsString() + "'"); 1518 return nullptr; 1519 } 1520 1521 if (!consume(tgtok::comma)) { 1522 TokError("expected ',' in !foldl"); 1523 return nullptr; 1524 } 1525 1526 Init *ListUntyped = ParseValue(CurRec); 1527 if (!ListUntyped) 1528 return nullptr; 1529 1530 TypedInit *List = dyn_cast<TypedInit>(ListUntyped); 1531 if (!List) { 1532 TokError(Twine("could not get type of !foldl list: '") + 1533 ListUntyped->getAsString() + "'"); 1534 return nullptr; 1535 } 1536 1537 ListRecTy *ListType = dyn_cast<ListRecTy>(List->getType()); 1538 if (!ListType) { 1539 TokError(Twine("!foldl list must be a list, but is of type '") + 1540 List->getType()->getAsString()); 1541 return nullptr; 1542 } 1543 1544 if (Lex.getCode() != tgtok::comma) { 1545 TokError("expected ',' in !foldl"); 1546 return nullptr; 1547 } 1548 1549 if (Lex.Lex() != tgtok::Id) { // eat the ',' 1550 TokError("third argument of !foldl must be an identifier"); 1551 return nullptr; 1552 } 1553 1554 Init *A = StringInit::get(Lex.getCurStrVal()); 1555 if (CurRec && CurRec->getValue(A)) { 1556 TokError((Twine("left !foldl variable '") + A->getAsString() + 1557 "' already defined") 1558 .str()); 1559 return nullptr; 1560 } 1561 1562 if (Lex.Lex() != tgtok::comma) { // eat the id 1563 TokError("expected ',' in !foldl"); 1564 return nullptr; 1565 } 1566 1567 if (Lex.Lex() != tgtok::Id) { // eat the ',' 1568 TokError("fourth argument of !foldl must be an identifier"); 1569 return nullptr; 1570 } 1571 1572 Init *B = StringInit::get(Lex.getCurStrVal()); 1573 if (CurRec && CurRec->getValue(B)) { 1574 TokError((Twine("right !foldl variable '") + B->getAsString() + 1575 "' already defined") 1576 .str()); 1577 return nullptr; 1578 } 1579 1580 if (Lex.Lex() != tgtok::comma) { // eat the id 1581 TokError("expected ',' in !foldl"); 1582 return nullptr; 1583 } 1584 Lex.Lex(); // eat the ',' 1585 1586 // We need to create a temporary record to provide a scope for the 1587 // two variables. 1588 std::unique_ptr<Record> ParseRecTmp; 1589 Record *ParseRec = CurRec; 1590 if (!ParseRec) { 1591 ParseRecTmp = std::make_unique<Record>(".parse", ArrayRef<SMLoc>{}, Records); 1592 ParseRec = ParseRecTmp.get(); 1593 } 1594 1595 ParseRec->addValue(RecordVal(A, Start->getType(), false)); 1596 ParseRec->addValue(RecordVal(B, ListType->getElementType(), false)); 1597 Init *ExprUntyped = ParseValue(ParseRec); 1598 ParseRec->removeValue(A); 1599 ParseRec->removeValue(B); 1600 if (!ExprUntyped) 1601 return nullptr; 1602 1603 TypedInit *Expr = dyn_cast<TypedInit>(ExprUntyped); 1604 if (!Expr) { 1605 TokError("could not get type of !foldl expression"); 1606 return nullptr; 1607 } 1608 1609 if (Expr->getType() != Start->getType()) { 1610 TokError(Twine("!foldl expression must be of same type as start (") + 1611 Start->getType()->getAsString() + "), but is of type " + 1612 Expr->getType()->getAsString()); 1613 return nullptr; 1614 } 1615 1616 if (!consume(tgtok::r_paren)) { 1617 TokError("expected ')' in fold operator"); 1618 return nullptr; 1619 } 1620 1621 return FoldOpInit::get(Start, List, A, B, Expr, Start->getType()) 1622 ->Fold(CurRec); 1623 } 1624 } 1625 } 1626 1627 /// ParseOperatorType - Parse a type for an operator. This returns 1628 /// null on error. 1629 /// 1630 /// OperatorType ::= '<' Type '>' 1631 /// 1632 RecTy *TGParser::ParseOperatorType() { 1633 RecTy *Type = nullptr; 1634 1635 if (!consume(tgtok::less)) { 1636 TokError("expected type name for operator"); 1637 return nullptr; 1638 } 1639 1640 if (Lex.getCode() == tgtok::Code) 1641 TokError("the 'code' type is not allowed in bang operators; use 'string'"); 1642 1643 Type = ParseType(); 1644 1645 if (!Type) { 1646 TokError("expected type name for operator"); 1647 return nullptr; 1648 } 1649 1650 if (!consume(tgtok::greater)) { 1651 TokError("expected type name for operator"); 1652 return nullptr; 1653 } 1654 1655 return Type; 1656 } 1657 1658 /// Parse the !foreach and !filter operations. Return null on error. 1659 /// 1660 /// ForEach ::= !foreach(ID, list-or-dag, expr) => list<expr type> 1661 /// Filter ::= !foreach(ID, list, predicate) ==> list<list type> 1662 Init *TGParser::ParseOperationForEachFilter(Record *CurRec, RecTy *ItemType) { 1663 SMLoc OpLoc = Lex.getLoc(); 1664 tgtok::TokKind Operation = Lex.getCode(); 1665 Lex.Lex(); // eat the operation 1666 if (Lex.getCode() != tgtok::l_paren) { 1667 TokError("expected '(' after !foreach/!filter"); 1668 return nullptr; 1669 } 1670 1671 if (Lex.Lex() != tgtok::Id) { // eat the '(' 1672 TokError("first argument of !foreach/!filter must be an identifier"); 1673 return nullptr; 1674 } 1675 1676 Init *LHS = StringInit::get(Lex.getCurStrVal()); 1677 Lex.Lex(); // eat the ID. 1678 1679 if (CurRec && CurRec->getValue(LHS)) { 1680 TokError((Twine("iteration variable '") + LHS->getAsString() + 1681 "' is already defined") 1682 .str()); 1683 return nullptr; 1684 } 1685 1686 if (!consume(tgtok::comma)) { 1687 TokError("expected ',' in !foreach/!filter"); 1688 return nullptr; 1689 } 1690 1691 Init *MHS = ParseValue(CurRec); 1692 if (!MHS) 1693 return nullptr; 1694 1695 if (!consume(tgtok::comma)) { 1696 TokError("expected ',' in !foreach/!filter"); 1697 return nullptr; 1698 } 1699 1700 TypedInit *MHSt = dyn_cast<TypedInit>(MHS); 1701 if (!MHSt) { 1702 TokError("could not get type of !foreach/!filter list or dag"); 1703 return nullptr; 1704 } 1705 1706 RecTy *InEltType = nullptr; 1707 RecTy *ExprEltType = nullptr; 1708 bool IsDAG = false; 1709 1710 if (ListRecTy *InListTy = dyn_cast<ListRecTy>(MHSt->getType())) { 1711 InEltType = InListTy->getElementType(); 1712 if (ItemType) { 1713 if (ListRecTy *OutListTy = dyn_cast<ListRecTy>(ItemType)) { 1714 ExprEltType = (Operation == tgtok::XForEach) 1715 ? OutListTy->getElementType() 1716 : IntRecTy::get(); 1717 } else { 1718 Error(OpLoc, 1719 "expected value of type '" + 1720 Twine(ItemType->getAsString()) + 1721 "', but got list type"); 1722 return nullptr; 1723 } 1724 } 1725 } else if (DagRecTy *InDagTy = dyn_cast<DagRecTy>(MHSt->getType())) { 1726 if (Operation == tgtok::XFilter) { 1727 TokError("!filter must have a list argument"); 1728 return nullptr; 1729 } 1730 InEltType = InDagTy; 1731 if (ItemType && !isa<DagRecTy>(ItemType)) { 1732 Error(OpLoc, 1733 "expected value of type '" + Twine(ItemType->getAsString()) + 1734 "', but got dag type"); 1735 return nullptr; 1736 } 1737 IsDAG = true; 1738 } else { 1739 if (Operation == tgtok::XForEach) 1740 TokError("!foreach must have a list or dag argument"); 1741 else 1742 TokError("!filter must have a list argument"); 1743 return nullptr; 1744 } 1745 1746 // We need to create a temporary record to provide a scope for the 1747 // iteration variable. 1748 std::unique_ptr<Record> ParseRecTmp; 1749 Record *ParseRec = CurRec; 1750 if (!ParseRec) { 1751 ParseRecTmp = 1752 std::make_unique<Record>(".parse", ArrayRef<SMLoc>{}, Records); 1753 ParseRec = ParseRecTmp.get(); 1754 } 1755 1756 ParseRec->addValue(RecordVal(LHS, InEltType, false)); 1757 Init *RHS = ParseValue(ParseRec, ExprEltType); 1758 ParseRec->removeValue(LHS); 1759 if (!RHS) 1760 return nullptr; 1761 1762 if (!consume(tgtok::r_paren)) { 1763 TokError("expected ')' in !foreach/!filter"); 1764 return nullptr; 1765 } 1766 1767 RecTy *OutType = InEltType; 1768 if (Operation == tgtok::XForEach && !IsDAG) { 1769 TypedInit *RHSt = dyn_cast<TypedInit>(RHS); 1770 if (!RHSt) { 1771 TokError("could not get type of !foreach result expression"); 1772 return nullptr; 1773 } 1774 OutType = RHSt->getType()->getListTy(); 1775 } else if (Operation == tgtok::XFilter) { 1776 OutType = InEltType->getListTy(); 1777 } 1778 1779 return (TernOpInit::get((Operation == tgtok::XForEach) ? TernOpInit::FOREACH 1780 : TernOpInit::FILTER, 1781 LHS, MHS, RHS, OutType)) 1782 ->Fold(CurRec); 1783 } 1784 1785 Init *TGParser::ParseOperationCond(Record *CurRec, RecTy *ItemType) { 1786 Lex.Lex(); // eat the operation 'cond' 1787 1788 if (!consume(tgtok::l_paren)) { 1789 TokError("expected '(' after !cond operator"); 1790 return nullptr; 1791 } 1792 1793 // Parse through '[Case: Val,]+' 1794 SmallVector<Init *, 4> Case; 1795 SmallVector<Init *, 4> Val; 1796 while (true) { 1797 if (consume(tgtok::r_paren)) 1798 break; 1799 1800 Init *V = ParseValue(CurRec); 1801 if (!V) 1802 return nullptr; 1803 Case.push_back(V); 1804 1805 if (!consume(tgtok::colon)) { 1806 TokError("expected ':' following a condition in !cond operator"); 1807 return nullptr; 1808 } 1809 1810 V = ParseValue(CurRec, ItemType); 1811 if (!V) 1812 return nullptr; 1813 Val.push_back(V); 1814 1815 if (consume(tgtok::r_paren)) 1816 break; 1817 1818 if (!consume(tgtok::comma)) { 1819 TokError("expected ',' or ')' following a value in !cond operator"); 1820 return nullptr; 1821 } 1822 } 1823 1824 if (Case.size() < 1) { 1825 TokError("there should be at least 1 'condition : value' in the !cond operator"); 1826 return nullptr; 1827 } 1828 1829 // resolve type 1830 RecTy *Type = nullptr; 1831 for (Init *V : Val) { 1832 RecTy *VTy = nullptr; 1833 if (TypedInit *Vt = dyn_cast<TypedInit>(V)) 1834 VTy = Vt->getType(); 1835 if (BitsInit *Vbits = dyn_cast<BitsInit>(V)) 1836 VTy = BitsRecTy::get(Vbits->getNumBits()); 1837 if (isa<BitInit>(V)) 1838 VTy = BitRecTy::get(); 1839 1840 if (Type == nullptr) { 1841 if (!isa<UnsetInit>(V)) 1842 Type = VTy; 1843 } else { 1844 if (!isa<UnsetInit>(V)) { 1845 RecTy *RType = resolveTypes(Type, VTy); 1846 if (!RType) { 1847 TokError(Twine("inconsistent types '") + Type->getAsString() + 1848 "' and '" + VTy->getAsString() + "' for !cond"); 1849 return nullptr; 1850 } 1851 Type = RType; 1852 } 1853 } 1854 } 1855 1856 if (!Type) { 1857 TokError("could not determine type for !cond from its arguments"); 1858 return nullptr; 1859 } 1860 return CondOpInit::get(Case, Val, Type)->Fold(CurRec); 1861 } 1862 1863 /// ParseSimpleValue - Parse a tblgen value. This returns null on error. 1864 /// 1865 /// SimpleValue ::= IDValue 1866 /// SimpleValue ::= INTVAL 1867 /// SimpleValue ::= STRVAL+ 1868 /// SimpleValue ::= CODEFRAGMENT 1869 /// SimpleValue ::= '?' 1870 /// SimpleValue ::= '{' ValueList '}' 1871 /// SimpleValue ::= ID '<' ValueListNE '>' 1872 /// SimpleValue ::= '[' ValueList ']' 1873 /// SimpleValue ::= '(' IDValue DagArgList ')' 1874 /// SimpleValue ::= CONCATTOK '(' Value ',' Value ')' 1875 /// SimpleValue ::= ADDTOK '(' Value ',' Value ')' 1876 /// SimpleValue ::= SUBTOK '(' Value ',' Value ')' 1877 /// SimpleValue ::= SHLTOK '(' Value ',' Value ')' 1878 /// SimpleValue ::= SRATOK '(' Value ',' Value ')' 1879 /// SimpleValue ::= SRLTOK '(' Value ',' Value ')' 1880 /// SimpleValue ::= LISTCONCATTOK '(' Value ',' Value ')' 1881 /// SimpleValue ::= LISTSPLATTOK '(' Value ',' Value ')' 1882 /// SimpleValue ::= STRCONCATTOK '(' Value ',' Value ')' 1883 /// SimpleValue ::= COND '(' [Value ':' Value,]+ ')' 1884 /// 1885 Init *TGParser::ParseSimpleValue(Record *CurRec, RecTy *ItemType, 1886 IDParseMode Mode) { 1887 Init *R = nullptr; 1888 switch (Lex.getCode()) { 1889 default: TokError("Unknown or reserved token when parsing a value"); break; 1890 1891 case tgtok::TrueVal: 1892 R = IntInit::get(1); 1893 Lex.Lex(); 1894 break; 1895 case tgtok::FalseVal: 1896 R = IntInit::get(0); 1897 Lex.Lex(); 1898 break; 1899 case tgtok::IntVal: 1900 R = IntInit::get(Lex.getCurIntVal()); 1901 Lex.Lex(); 1902 break; 1903 case tgtok::BinaryIntVal: { 1904 auto BinaryVal = Lex.getCurBinaryIntVal(); 1905 SmallVector<Init*, 16> Bits(BinaryVal.second); 1906 for (unsigned i = 0, e = BinaryVal.second; i != e; ++i) 1907 Bits[i] = BitInit::get(BinaryVal.first & (1LL << i)); 1908 R = BitsInit::get(Bits); 1909 Lex.Lex(); 1910 break; 1911 } 1912 case tgtok::StrVal: { 1913 std::string Val = Lex.getCurStrVal(); 1914 Lex.Lex(); 1915 1916 // Handle multiple consecutive concatenated strings. 1917 while (Lex.getCode() == tgtok::StrVal) { 1918 Val += Lex.getCurStrVal(); 1919 Lex.Lex(); 1920 } 1921 1922 R = StringInit::get(Val); 1923 break; 1924 } 1925 case tgtok::CodeFragment: 1926 R = StringInit::get(Lex.getCurStrVal(), StringInit::SF_Code); 1927 Lex.Lex(); 1928 break; 1929 case tgtok::question: 1930 R = UnsetInit::get(); 1931 Lex.Lex(); 1932 break; 1933 case tgtok::Id: { 1934 SMLoc NameLoc = Lex.getLoc(); 1935 StringInit *Name = StringInit::get(Lex.getCurStrVal()); 1936 if (Lex.Lex() != tgtok::less) // consume the Id. 1937 return ParseIDValue(CurRec, Name, NameLoc, Mode); // Value ::= IDValue 1938 1939 // Value ::= ID '<' ValueListNE '>' 1940 if (Lex.Lex() == tgtok::greater) { 1941 TokError("expected non-empty value list"); 1942 return nullptr; 1943 } 1944 1945 // This is a CLASS<initvalslist> expression. This is supposed to synthesize 1946 // a new anonymous definition, deriving from CLASS<initvalslist> with no 1947 // body. 1948 Record *Class = Records.getClass(Name->getValue()); 1949 if (!Class) { 1950 Error(NameLoc, "Expected a class name, got '" + Name->getValue() + "'"); 1951 return nullptr; 1952 } 1953 1954 SmallVector<Init *, 8> Args; 1955 ParseValueList(Args, CurRec, Class); 1956 if (Args.empty()) return nullptr; 1957 1958 if (!consume(tgtok::greater)) { 1959 TokError("expected '>' at end of value list"); 1960 return nullptr; 1961 } 1962 1963 // Typecheck the template arguments list 1964 ArrayRef<Init *> ExpectedArgs = Class->getTemplateArgs(); 1965 if (ExpectedArgs.size() < Args.size()) { 1966 Error(NameLoc, 1967 "More template args specified than expected"); 1968 return nullptr; 1969 } 1970 1971 for (unsigned i = 0, e = ExpectedArgs.size(); i != e; ++i) { 1972 RecordVal *ExpectedArg = Class->getValue(ExpectedArgs[i]); 1973 if (i < Args.size()) { 1974 if (TypedInit *TI = dyn_cast<TypedInit>(Args[i])) { 1975 RecTy *ExpectedType = ExpectedArg->getType(); 1976 if (!TI->getType()->typeIsConvertibleTo(ExpectedType)) { 1977 Error(NameLoc, 1978 "Value specified for template argument #" + Twine(i) + " (" + 1979 ExpectedArg->getNameInitAsString() + ") is of type '" + 1980 TI->getType()->getAsString() + "', expected '" + 1981 ExpectedType->getAsString() + "': " + TI->getAsString()); 1982 return nullptr; 1983 } 1984 continue; 1985 } 1986 } else if (ExpectedArg->getValue()->isComplete()) 1987 continue; 1988 1989 Error(NameLoc, 1990 "Value not specified for template argument #" + Twine(i) + " (" + 1991 ExpectedArgs[i]->getAsUnquotedString() + ")"); 1992 return nullptr; 1993 } 1994 1995 return VarDefInit::get(Class, Args)->Fold(); 1996 } 1997 case tgtok::l_brace: { // Value ::= '{' ValueList '}' 1998 SMLoc BraceLoc = Lex.getLoc(); 1999 Lex.Lex(); // eat the '{' 2000 SmallVector<Init*, 16> Vals; 2001 2002 if (Lex.getCode() != tgtok::r_brace) { 2003 ParseValueList(Vals, CurRec); 2004 if (Vals.empty()) return nullptr; 2005 } 2006 if (!consume(tgtok::r_brace)) { 2007 TokError("expected '}' at end of bit list value"); 2008 return nullptr; 2009 } 2010 2011 SmallVector<Init *, 16> NewBits; 2012 2013 // As we parse { a, b, ... }, 'a' is the highest bit, but we parse it 2014 // first. We'll first read everything in to a vector, then we can reverse 2015 // it to get the bits in the correct order for the BitsInit value. 2016 for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 2017 // FIXME: The following two loops would not be duplicated 2018 // if the API was a little more orthogonal. 2019 2020 // bits<n> values are allowed to initialize n bits. 2021 if (BitsInit *BI = dyn_cast<BitsInit>(Vals[i])) { 2022 for (unsigned i = 0, e = BI->getNumBits(); i != e; ++i) 2023 NewBits.push_back(BI->getBit((e - i) - 1)); 2024 continue; 2025 } 2026 // bits<n> can also come from variable initializers. 2027 if (VarInit *VI = dyn_cast<VarInit>(Vals[i])) { 2028 if (BitsRecTy *BitsRec = dyn_cast<BitsRecTy>(VI->getType())) { 2029 for (unsigned i = 0, e = BitsRec->getNumBits(); i != e; ++i) 2030 NewBits.push_back(VI->getBit((e - i) - 1)); 2031 continue; 2032 } 2033 // Fallthrough to try convert this to a bit. 2034 } 2035 // All other values must be convertible to just a single bit. 2036 Init *Bit = Vals[i]->getCastTo(BitRecTy::get()); 2037 if (!Bit) { 2038 Error(BraceLoc, "Element #" + Twine(i) + " (" + Vals[i]->getAsString() + 2039 ") is not convertable to a bit"); 2040 return nullptr; 2041 } 2042 NewBits.push_back(Bit); 2043 } 2044 std::reverse(NewBits.begin(), NewBits.end()); 2045 return BitsInit::get(NewBits); 2046 } 2047 case tgtok::l_square: { // Value ::= '[' ValueList ']' 2048 Lex.Lex(); // eat the '[' 2049 SmallVector<Init*, 16> Vals; 2050 2051 RecTy *DeducedEltTy = nullptr; 2052 ListRecTy *GivenListTy = nullptr; 2053 2054 if (ItemType) { 2055 ListRecTy *ListType = dyn_cast<ListRecTy>(ItemType); 2056 if (!ListType) { 2057 TokError(Twine("Encountered a list when expecting a ") + 2058 ItemType->getAsString()); 2059 return nullptr; 2060 } 2061 GivenListTy = ListType; 2062 } 2063 2064 if (Lex.getCode() != tgtok::r_square) { 2065 ParseValueList(Vals, CurRec, nullptr, 2066 GivenListTy ? GivenListTy->getElementType() : nullptr); 2067 if (Vals.empty()) return nullptr; 2068 } 2069 if (!consume(tgtok::r_square)) { 2070 TokError("expected ']' at end of list value"); 2071 return nullptr; 2072 } 2073 2074 RecTy *GivenEltTy = nullptr; 2075 if (consume(tgtok::less)) { 2076 // Optional list element type 2077 GivenEltTy = ParseType(); 2078 if (!GivenEltTy) { 2079 // Couldn't parse element type 2080 return nullptr; 2081 } 2082 2083 if (!consume(tgtok::greater)) { 2084 TokError("expected '>' at end of list element type"); 2085 return nullptr; 2086 } 2087 } 2088 2089 // Check elements 2090 RecTy *EltTy = nullptr; 2091 for (Init *V : Vals) { 2092 TypedInit *TArg = dyn_cast<TypedInit>(V); 2093 if (TArg) { 2094 if (EltTy) { 2095 EltTy = resolveTypes(EltTy, TArg->getType()); 2096 if (!EltTy) { 2097 TokError("Incompatible types in list elements"); 2098 return nullptr; 2099 } 2100 } else { 2101 EltTy = TArg->getType(); 2102 } 2103 } 2104 } 2105 2106 if (GivenEltTy) { 2107 if (EltTy) { 2108 // Verify consistency 2109 if (!EltTy->typeIsConvertibleTo(GivenEltTy)) { 2110 TokError("Incompatible types in list elements"); 2111 return nullptr; 2112 } 2113 } 2114 EltTy = GivenEltTy; 2115 } 2116 2117 if (!EltTy) { 2118 if (!ItemType) { 2119 TokError("No type for list"); 2120 return nullptr; 2121 } 2122 DeducedEltTy = GivenListTy->getElementType(); 2123 } else { 2124 // Make sure the deduced type is compatible with the given type 2125 if (GivenListTy) { 2126 if (!EltTy->typeIsConvertibleTo(GivenListTy->getElementType())) { 2127 TokError(Twine("Element type mismatch for list: element type '") + 2128 EltTy->getAsString() + "' not convertible to '" + 2129 GivenListTy->getElementType()->getAsString()); 2130 return nullptr; 2131 } 2132 } 2133 DeducedEltTy = EltTy; 2134 } 2135 2136 return ListInit::get(Vals, DeducedEltTy); 2137 } 2138 case tgtok::l_paren: { // Value ::= '(' IDValue DagArgList ')' 2139 Lex.Lex(); // eat the '(' 2140 if (Lex.getCode() != tgtok::Id && Lex.getCode() != tgtok::XCast && 2141 Lex.getCode() != tgtok::question && Lex.getCode() != tgtok::XGetDagOp) { 2142 TokError("expected identifier in dag init"); 2143 return nullptr; 2144 } 2145 2146 Init *Operator = ParseValue(CurRec); 2147 if (!Operator) return nullptr; 2148 2149 // If the operator name is present, parse it. 2150 StringInit *OperatorName = nullptr; 2151 if (consume(tgtok::colon)) { 2152 if (Lex.getCode() != tgtok::VarName) { // eat the ':' 2153 TokError("expected variable name in dag operator"); 2154 return nullptr; 2155 } 2156 OperatorName = StringInit::get(Lex.getCurStrVal()); 2157 Lex.Lex(); // eat the VarName. 2158 } 2159 2160 SmallVector<std::pair<llvm::Init*, StringInit*>, 8> DagArgs; 2161 if (Lex.getCode() != tgtok::r_paren) { 2162 ParseDagArgList(DagArgs, CurRec); 2163 if (DagArgs.empty()) return nullptr; 2164 } 2165 2166 if (!consume(tgtok::r_paren)) { 2167 TokError("expected ')' in dag init"); 2168 return nullptr; 2169 } 2170 2171 return DagInit::get(Operator, OperatorName, DagArgs); 2172 } 2173 2174 case tgtok::XHead: 2175 case tgtok::XTail: 2176 case tgtok::XSize: 2177 case tgtok::XEmpty: 2178 case tgtok::XCast: 2179 case tgtok::XGetDagOp: // Value ::= !unop '(' Value ')' 2180 case tgtok::XIsA: 2181 case tgtok::XConcat: 2182 case tgtok::XDag: 2183 case tgtok::XADD: 2184 case tgtok::XSUB: 2185 case tgtok::XMUL: 2186 case tgtok::XNOT: 2187 case tgtok::XAND: 2188 case tgtok::XOR: 2189 case tgtok::XXOR: 2190 case tgtok::XSRA: 2191 case tgtok::XSRL: 2192 case tgtok::XSHL: 2193 case tgtok::XEq: 2194 case tgtok::XNe: 2195 case tgtok::XLe: 2196 case tgtok::XLt: 2197 case tgtok::XGe: 2198 case tgtok::XGt: 2199 case tgtok::XListConcat: 2200 case tgtok::XListSplat: 2201 case tgtok::XStrConcat: 2202 case tgtok::XInterleave: 2203 case tgtok::XSetDagOp: // Value ::= !binop '(' Value ',' Value ')' 2204 case tgtok::XIf: 2205 case tgtok::XCond: 2206 case tgtok::XFoldl: 2207 case tgtok::XForEach: 2208 case tgtok::XFilter: 2209 case tgtok::XSubst: { // Value ::= !ternop '(' Value ',' Value ',' Value ')' 2210 return ParseOperation(CurRec, ItemType); 2211 } 2212 } 2213 2214 return R; 2215 } 2216 2217 /// ParseValue - Parse a tblgen value. This returns null on error. 2218 /// 2219 /// Value ::= SimpleValue ValueSuffix* 2220 /// ValueSuffix ::= '{' BitList '}' 2221 /// ValueSuffix ::= '[' BitList ']' 2222 /// ValueSuffix ::= '.' ID 2223 /// 2224 Init *TGParser::ParseValue(Record *CurRec, RecTy *ItemType, IDParseMode Mode) { 2225 Init *Result = ParseSimpleValue(CurRec, ItemType, Mode); 2226 if (!Result) return nullptr; 2227 2228 // Parse the suffixes now if present. 2229 while (true) { 2230 switch (Lex.getCode()) { 2231 default: return Result; 2232 case tgtok::l_brace: { 2233 if (Mode == ParseNameMode) 2234 // This is the beginning of the object body. 2235 return Result; 2236 2237 SMLoc CurlyLoc = Lex.getLoc(); 2238 Lex.Lex(); // eat the '{' 2239 SmallVector<unsigned, 16> Ranges; 2240 ParseRangeList(Ranges); 2241 if (Ranges.empty()) return nullptr; 2242 2243 // Reverse the bitlist. 2244 std::reverse(Ranges.begin(), Ranges.end()); 2245 Result = Result->convertInitializerBitRange(Ranges); 2246 if (!Result) { 2247 Error(CurlyLoc, "Invalid bit range for value"); 2248 return nullptr; 2249 } 2250 2251 // Eat the '}'. 2252 if (!consume(tgtok::r_brace)) { 2253 TokError("expected '}' at end of bit range list"); 2254 return nullptr; 2255 } 2256 break; 2257 } 2258 case tgtok::l_square: { 2259 SMLoc SquareLoc = Lex.getLoc(); 2260 Lex.Lex(); // eat the '[' 2261 SmallVector<unsigned, 16> Ranges; 2262 ParseRangeList(Ranges); 2263 if (Ranges.empty()) return nullptr; 2264 2265 Result = Result->convertInitListSlice(Ranges); 2266 if (!Result) { 2267 Error(SquareLoc, "Invalid range for list slice"); 2268 return nullptr; 2269 } 2270 2271 // Eat the ']'. 2272 if (!consume(tgtok::r_square)) { 2273 TokError("expected ']' at end of list slice"); 2274 return nullptr; 2275 } 2276 break; 2277 } 2278 case tgtok::dot: { 2279 if (Lex.Lex() != tgtok::Id) { // eat the . 2280 TokError("expected field identifier after '.'"); 2281 return nullptr; 2282 } 2283 StringInit *FieldName = StringInit::get(Lex.getCurStrVal()); 2284 if (!Result->getFieldType(FieldName)) { 2285 TokError("Cannot access field '" + Lex.getCurStrVal() + "' of value '" + 2286 Result->getAsString() + "'"); 2287 return nullptr; 2288 } 2289 Result = FieldInit::get(Result, FieldName)->Fold(CurRec); 2290 Lex.Lex(); // eat field name 2291 break; 2292 } 2293 2294 case tgtok::paste: 2295 SMLoc PasteLoc = Lex.getLoc(); 2296 TypedInit *LHS = dyn_cast<TypedInit>(Result); 2297 if (!LHS) { 2298 Error(PasteLoc, "LHS of paste is not typed!"); 2299 return nullptr; 2300 } 2301 2302 // Check if it's a 'listA # listB' 2303 if (isa<ListRecTy>(LHS->getType())) { 2304 Lex.Lex(); // Eat the '#'. 2305 2306 assert(Mode == ParseValueMode && "encountered paste of lists in name"); 2307 2308 switch (Lex.getCode()) { 2309 case tgtok::colon: 2310 case tgtok::semi: 2311 case tgtok::l_brace: 2312 Result = LHS; // trailing paste, ignore. 2313 break; 2314 default: 2315 Init *RHSResult = ParseValue(CurRec, ItemType, ParseValueMode); 2316 if (!RHSResult) 2317 return nullptr; 2318 Result = BinOpInit::getListConcat(LHS, RHSResult); 2319 break; 2320 } 2321 break; 2322 } 2323 2324 // Create a !strconcat() operation, first casting each operand to 2325 // a string if necessary. 2326 if (LHS->getType() != StringRecTy::get()) { 2327 auto CastLHS = dyn_cast<TypedInit>( 2328 UnOpInit::get(UnOpInit::CAST, LHS, StringRecTy::get()) 2329 ->Fold(CurRec)); 2330 if (!CastLHS) { 2331 Error(PasteLoc, 2332 Twine("can't cast '") + LHS->getAsString() + "' to string"); 2333 return nullptr; 2334 } 2335 LHS = CastLHS; 2336 } 2337 2338 TypedInit *RHS = nullptr; 2339 2340 Lex.Lex(); // Eat the '#'. 2341 switch (Lex.getCode()) { 2342 case tgtok::colon: 2343 case tgtok::semi: 2344 case tgtok::l_brace: 2345 // These are all of the tokens that can begin an object body. 2346 // Some of these can also begin values but we disallow those cases 2347 // because they are unlikely to be useful. 2348 2349 // Trailing paste, concat with an empty string. 2350 RHS = StringInit::get(""); 2351 break; 2352 2353 default: 2354 Init *RHSResult = ParseValue(CurRec, nullptr, ParseNameMode); 2355 if (!RHSResult) 2356 return nullptr; 2357 RHS = dyn_cast<TypedInit>(RHSResult); 2358 if (!RHS) { 2359 Error(PasteLoc, "RHS of paste is not typed!"); 2360 return nullptr; 2361 } 2362 2363 if (RHS->getType() != StringRecTy::get()) { 2364 auto CastRHS = dyn_cast<TypedInit>( 2365 UnOpInit::get(UnOpInit::CAST, RHS, StringRecTy::get()) 2366 ->Fold(CurRec)); 2367 if (!CastRHS) { 2368 Error(PasteLoc, 2369 Twine("can't cast '") + RHS->getAsString() + "' to string"); 2370 return nullptr; 2371 } 2372 RHS = CastRHS; 2373 } 2374 2375 break; 2376 } 2377 2378 Result = BinOpInit::getStrConcat(LHS, RHS); 2379 break; 2380 } 2381 } 2382 } 2383 2384 /// ParseDagArgList - Parse the argument list for a dag literal expression. 2385 /// 2386 /// DagArg ::= Value (':' VARNAME)? 2387 /// DagArg ::= VARNAME 2388 /// DagArgList ::= DagArg 2389 /// DagArgList ::= DagArgList ',' DagArg 2390 void TGParser::ParseDagArgList( 2391 SmallVectorImpl<std::pair<llvm::Init*, StringInit*>> &Result, 2392 Record *CurRec) { 2393 2394 while (true) { 2395 // DagArg ::= VARNAME 2396 if (Lex.getCode() == tgtok::VarName) { 2397 // A missing value is treated like '?'. 2398 StringInit *VarName = StringInit::get(Lex.getCurStrVal()); 2399 Result.emplace_back(UnsetInit::get(), VarName); 2400 Lex.Lex(); 2401 } else { 2402 // DagArg ::= Value (':' VARNAME)? 2403 Init *Val = ParseValue(CurRec); 2404 if (!Val) { 2405 Result.clear(); 2406 return; 2407 } 2408 2409 // If the variable name is present, add it. 2410 StringInit *VarName = nullptr; 2411 if (Lex.getCode() == tgtok::colon) { 2412 if (Lex.Lex() != tgtok::VarName) { // eat the ':' 2413 TokError("expected variable name in dag literal"); 2414 Result.clear(); 2415 return; 2416 } 2417 VarName = StringInit::get(Lex.getCurStrVal()); 2418 Lex.Lex(); // eat the VarName. 2419 } 2420 2421 Result.push_back(std::make_pair(Val, VarName)); 2422 } 2423 if (!consume(tgtok::comma)) 2424 break; 2425 } 2426 } 2427 2428 /// ParseValueList - Parse a comma separated list of values, returning them as a 2429 /// vector. Note that this always expects to be able to parse at least one 2430 /// value. It returns an empty list if this is not possible. 2431 /// 2432 /// ValueList ::= Value (',' Value) 2433 /// 2434 void TGParser::ParseValueList(SmallVectorImpl<Init*> &Result, Record *CurRec, 2435 Record *ArgsRec, RecTy *EltTy) { 2436 RecTy *ItemType = EltTy; 2437 unsigned int ArgN = 0; 2438 if (ArgsRec && !EltTy) { 2439 ArrayRef<Init *> TArgs = ArgsRec->getTemplateArgs(); 2440 if (TArgs.empty()) { 2441 TokError("template argument provided to non-template class"); 2442 Result.clear(); 2443 return; 2444 } 2445 const RecordVal *RV = ArgsRec->getValue(TArgs[ArgN]); 2446 if (!RV) { 2447 errs() << "Cannot find template arg " << ArgN << " (" << TArgs[ArgN] 2448 << ")\n"; 2449 } 2450 assert(RV && "Template argument record not found??"); 2451 ItemType = RV->getType(); 2452 ++ArgN; 2453 } 2454 Result.push_back(ParseValue(CurRec, ItemType)); 2455 if (!Result.back()) { 2456 Result.clear(); 2457 return; 2458 } 2459 2460 while (consume(tgtok::comma)) { 2461 // ignore trailing comma for lists 2462 if (Lex.getCode() == tgtok::r_square) 2463 return; 2464 2465 if (ArgsRec && !EltTy) { 2466 ArrayRef<Init *> TArgs = ArgsRec->getTemplateArgs(); 2467 if (ArgN >= TArgs.size()) { 2468 TokError("too many template arguments"); 2469 Result.clear(); 2470 return; 2471 } 2472 const RecordVal *RV = ArgsRec->getValue(TArgs[ArgN]); 2473 assert(RV && "Template argument record not found??"); 2474 ItemType = RV->getType(); 2475 ++ArgN; 2476 } 2477 Result.push_back(ParseValue(CurRec, ItemType)); 2478 if (!Result.back()) { 2479 Result.clear(); 2480 return; 2481 } 2482 } 2483 } 2484 2485 /// ParseDeclaration - Read a declaration, returning the name of field ID, or an 2486 /// empty string on error. This can happen in a number of different context's, 2487 /// including within a def or in the template args for a def (which which case 2488 /// CurRec will be non-null) and within the template args for a multiclass (in 2489 /// which case CurRec will be null, but CurMultiClass will be set). This can 2490 /// also happen within a def that is within a multiclass, which will set both 2491 /// CurRec and CurMultiClass. 2492 /// 2493 /// Declaration ::= FIELD? Type ID ('=' Value)? 2494 /// 2495 Init *TGParser::ParseDeclaration(Record *CurRec, 2496 bool ParsingTemplateArgs) { 2497 // Read the field prefix if present. 2498 bool HasField = consume(tgtok::Field); 2499 2500 RecTy *Type = ParseType(); 2501 if (!Type) return nullptr; 2502 2503 if (Lex.getCode() != tgtok::Id) { 2504 TokError("Expected identifier in declaration"); 2505 return nullptr; 2506 } 2507 2508 std::string Str = Lex.getCurStrVal(); 2509 if (Str == "NAME") { 2510 TokError("'" + Str + "' is a reserved variable name"); 2511 return nullptr; 2512 } 2513 2514 SMLoc IdLoc = Lex.getLoc(); 2515 Init *DeclName = StringInit::get(Str); 2516 Lex.Lex(); 2517 2518 if (ParsingTemplateArgs) { 2519 if (CurRec) 2520 DeclName = QualifyName(*CurRec, CurMultiClass, DeclName, ":"); 2521 else 2522 assert(CurMultiClass); 2523 if (CurMultiClass) 2524 DeclName = QualifyName(CurMultiClass->Rec, CurMultiClass, DeclName, 2525 "::"); 2526 } 2527 2528 // Add the value. 2529 if (AddValue(CurRec, IdLoc, RecordVal(DeclName, IdLoc, Type, HasField))) 2530 return nullptr; 2531 2532 // If a value is present, parse it. 2533 if (consume(tgtok::equal)) { 2534 SMLoc ValLoc = Lex.getLoc(); 2535 Init *Val = ParseValue(CurRec, Type); 2536 if (!Val || 2537 SetValue(CurRec, ValLoc, DeclName, None, Val)) 2538 // Return the name, even if an error is thrown. This is so that we can 2539 // continue to make some progress, even without the value having been 2540 // initialized. 2541 return DeclName; 2542 } 2543 2544 return DeclName; 2545 } 2546 2547 /// ParseForeachDeclaration - Read a foreach declaration, returning 2548 /// the name of the declared object or a NULL Init on error. Return 2549 /// the name of the parsed initializer list through ForeachListName. 2550 /// 2551 /// ForeachDeclaration ::= ID '=' '{' RangeList '}' 2552 /// ForeachDeclaration ::= ID '=' RangePiece 2553 /// ForeachDeclaration ::= ID '=' Value 2554 /// 2555 VarInit *TGParser::ParseForeachDeclaration(Init *&ForeachListValue) { 2556 if (Lex.getCode() != tgtok::Id) { 2557 TokError("Expected identifier in foreach declaration"); 2558 return nullptr; 2559 } 2560 2561 Init *DeclName = StringInit::get(Lex.getCurStrVal()); 2562 Lex.Lex(); 2563 2564 // If a value is present, parse it. 2565 if (!consume(tgtok::equal)) { 2566 TokError("Expected '=' in foreach declaration"); 2567 return nullptr; 2568 } 2569 2570 RecTy *IterType = nullptr; 2571 SmallVector<unsigned, 16> Ranges; 2572 2573 switch (Lex.getCode()) { 2574 case tgtok::l_brace: { // '{' RangeList '}' 2575 Lex.Lex(); // eat the '{' 2576 ParseRangeList(Ranges); 2577 if (!consume(tgtok::r_brace)) { 2578 TokError("expected '}' at end of bit range list"); 2579 return nullptr; 2580 } 2581 break; 2582 } 2583 2584 default: { 2585 SMLoc ValueLoc = Lex.getLoc(); 2586 Init *I = ParseValue(nullptr); 2587 if (!I) 2588 return nullptr; 2589 2590 TypedInit *TI = dyn_cast<TypedInit>(I); 2591 if (TI && isa<ListRecTy>(TI->getType())) { 2592 ForeachListValue = I; 2593 IterType = cast<ListRecTy>(TI->getType())->getElementType(); 2594 break; 2595 } 2596 2597 if (TI) { 2598 if (ParseRangePiece(Ranges, TI)) 2599 return nullptr; 2600 break; 2601 } 2602 2603 std::string Type; 2604 if (TI) 2605 Type = (Twine("' of type '") + TI->getType()->getAsString()).str(); 2606 Error(ValueLoc, "expected a list, got '" + I->getAsString() + Type + "'"); 2607 if (CurMultiClass) { 2608 PrintNote({}, "references to multiclass template arguments cannot be " 2609 "resolved at this time"); 2610 } 2611 return nullptr; 2612 } 2613 } 2614 2615 2616 if (!Ranges.empty()) { 2617 assert(!IterType && "Type already initialized?"); 2618 IterType = IntRecTy::get(); 2619 std::vector<Init*> Values; 2620 for (unsigned R : Ranges) 2621 Values.push_back(IntInit::get(R)); 2622 ForeachListValue = ListInit::get(Values, IterType); 2623 } 2624 2625 if (!IterType) 2626 return nullptr; 2627 2628 return VarInit::get(DeclName, IterType); 2629 } 2630 2631 /// ParseTemplateArgList - Read a template argument list, which is a non-empty 2632 /// sequence of template-declarations in <>'s. If CurRec is non-null, these are 2633 /// template args for a def, which may or may not be in a multiclass. If null, 2634 /// these are the template args for a multiclass. 2635 /// 2636 /// TemplateArgList ::= '<' Declaration (',' Declaration)* '>' 2637 /// 2638 bool TGParser::ParseTemplateArgList(Record *CurRec) { 2639 assert(Lex.getCode() == tgtok::less && "Not a template arg list!"); 2640 Lex.Lex(); // eat the '<' 2641 2642 Record *TheRecToAddTo = CurRec ? CurRec : &CurMultiClass->Rec; 2643 2644 // Read the first declaration. 2645 Init *TemplArg = ParseDeclaration(CurRec, true/*templateargs*/); 2646 if (!TemplArg) 2647 return true; 2648 2649 TheRecToAddTo->addTemplateArg(TemplArg); 2650 2651 while (consume(tgtok::comma)) { 2652 // Read the following declarations. 2653 SMLoc Loc = Lex.getLoc(); 2654 TemplArg = ParseDeclaration(CurRec, true/*templateargs*/); 2655 if (!TemplArg) 2656 return true; 2657 2658 if (TheRecToAddTo->isTemplateArg(TemplArg)) 2659 return Error(Loc, "template argument with the same name has already been " 2660 "defined"); 2661 2662 TheRecToAddTo->addTemplateArg(TemplArg); 2663 } 2664 2665 if (!consume(tgtok::greater)) 2666 return TokError("expected '>' at end of template argument list"); 2667 return false; 2668 } 2669 2670 /// ParseBodyItem - Parse a single item at within the body of a def or class. 2671 /// 2672 /// BodyItem ::= Declaration ';' 2673 /// BodyItem ::= LET ID OptionalBitList '=' Value ';' 2674 /// BodyItem ::= Defvar 2675 bool TGParser::ParseBodyItem(Record *CurRec) { 2676 if (Lex.getCode() == tgtok::Defvar) 2677 return ParseDefvar(); 2678 2679 if (Lex.getCode() != tgtok::Let) { 2680 if (!ParseDeclaration(CurRec, false)) 2681 return true; 2682 2683 if (!consume(tgtok::semi)) 2684 return TokError("expected ';' after declaration"); 2685 return false; 2686 } 2687 2688 // LET ID OptionalRangeList '=' Value ';' 2689 if (Lex.Lex() != tgtok::Id) 2690 return TokError("expected field identifier after let"); 2691 2692 SMLoc IdLoc = Lex.getLoc(); 2693 StringInit *FieldName = StringInit::get(Lex.getCurStrVal()); 2694 Lex.Lex(); // eat the field name. 2695 2696 SmallVector<unsigned, 16> BitList; 2697 if (ParseOptionalBitList(BitList)) 2698 return true; 2699 std::reverse(BitList.begin(), BitList.end()); 2700 2701 if (!consume(tgtok::equal)) 2702 return TokError("expected '=' in let expression"); 2703 2704 RecordVal *Field = CurRec->getValue(FieldName); 2705 if (!Field) 2706 return TokError("Value '" + FieldName->getValue() + "' unknown!"); 2707 2708 RecTy *Type = Field->getType(); 2709 if (!BitList.empty() && isa<BitsRecTy>(Type)) { 2710 // When assigning to a subset of a 'bits' object, expect the RHS to have 2711 // the type of that subset instead of the type of the whole object. 2712 Type = BitsRecTy::get(BitList.size()); 2713 } 2714 2715 Init *Val = ParseValue(CurRec, Type); 2716 if (!Val) return true; 2717 2718 if (!consume(tgtok::semi)) 2719 return TokError("expected ';' after let expression"); 2720 2721 return SetValue(CurRec, IdLoc, FieldName, BitList, Val); 2722 } 2723 2724 /// ParseBody - Read the body of a class or def. Return true on error, false on 2725 /// success. 2726 /// 2727 /// Body ::= ';' 2728 /// Body ::= '{' BodyList '}' 2729 /// BodyList BodyItem* 2730 /// 2731 bool TGParser::ParseBody(Record *CurRec) { 2732 // If this is a null definition, just eat the semi and return. 2733 if (consume(tgtok::semi)) 2734 return false; 2735 2736 if (!consume(tgtok::l_brace)) 2737 return TokError("Expected ';' or '{' to start body"); 2738 2739 // An object body introduces a new scope for local variables. 2740 TGLocalVarScope *BodyScope = PushLocalScope(); 2741 2742 while (Lex.getCode() != tgtok::r_brace) 2743 if (ParseBodyItem(CurRec)) 2744 return true; 2745 2746 PopLocalScope(BodyScope); 2747 2748 // Eat the '}'. 2749 Lex.Lex(); 2750 return false; 2751 } 2752 2753 /// Apply the current let bindings to \a CurRec. 2754 /// \returns true on error, false otherwise. 2755 bool TGParser::ApplyLetStack(Record *CurRec) { 2756 for (SmallVectorImpl<LetRecord> &LetInfo : LetStack) 2757 for (LetRecord &LR : LetInfo) 2758 if (SetValue(CurRec, LR.Loc, LR.Name, LR.Bits, LR.Value)) 2759 return true; 2760 return false; 2761 } 2762 2763 bool TGParser::ApplyLetStack(RecordsEntry &Entry) { 2764 if (Entry.Rec) 2765 return ApplyLetStack(Entry.Rec.get()); 2766 2767 for (auto &E : Entry.Loop->Entries) { 2768 if (ApplyLetStack(E)) 2769 return true; 2770 } 2771 2772 return false; 2773 } 2774 2775 /// ParseObjectBody - Parse the body of a def or class. This consists of an 2776 /// optional ClassList followed by a Body. CurRec is the current def or class 2777 /// that is being parsed. 2778 /// 2779 /// ObjectBody ::= BaseClassList Body 2780 /// BaseClassList ::= /*empty*/ 2781 /// BaseClassList ::= ':' BaseClassListNE 2782 /// BaseClassListNE ::= SubClassRef (',' SubClassRef)* 2783 /// 2784 bool TGParser::ParseObjectBody(Record *CurRec) { 2785 // If there is a baseclass list, read it. 2786 if (consume(tgtok::colon)) { 2787 2788 // Read all of the subclasses. 2789 SubClassReference SubClass = ParseSubClassReference(CurRec, false); 2790 while (true) { 2791 // Check for error. 2792 if (!SubClass.Rec) return true; 2793 2794 // Add it. 2795 if (AddSubClass(CurRec, SubClass)) 2796 return true; 2797 2798 if (!consume(tgtok::comma)) 2799 break; 2800 SubClass = ParseSubClassReference(CurRec, false); 2801 } 2802 } 2803 2804 if (ApplyLetStack(CurRec)) 2805 return true; 2806 2807 return ParseBody(CurRec); 2808 } 2809 2810 /// ParseDef - Parse and return a top level or multiclass def, return the record 2811 /// corresponding to it. This returns null on error. 2812 /// 2813 /// DefInst ::= DEF ObjectName ObjectBody 2814 /// 2815 bool TGParser::ParseDef(MultiClass *CurMultiClass) { 2816 SMLoc DefLoc = Lex.getLoc(); 2817 assert(Lex.getCode() == tgtok::Def && "Unknown tok"); 2818 Lex.Lex(); // Eat the 'def' token. 2819 2820 // Parse ObjectName and make a record for it. 2821 std::unique_ptr<Record> CurRec; 2822 Init *Name = ParseObjectName(CurMultiClass); 2823 if (!Name) 2824 return true; 2825 2826 if (isa<UnsetInit>(Name)) 2827 CurRec = std::make_unique<Record>(Records.getNewAnonymousName(), DefLoc, Records, 2828 /*Anonymous=*/true); 2829 else 2830 CurRec = std::make_unique<Record>(Name, DefLoc, Records); 2831 2832 if (ParseObjectBody(CurRec.get())) 2833 return true; 2834 2835 return addEntry(std::move(CurRec)); 2836 } 2837 2838 /// ParseDefset - Parse a defset statement. 2839 /// 2840 /// Defset ::= DEFSET Type Id '=' '{' ObjectList '}' 2841 /// 2842 bool TGParser::ParseDefset() { 2843 assert(Lex.getCode() == tgtok::Defset); 2844 Lex.Lex(); // Eat the 'defset' token 2845 2846 DefsetRecord Defset; 2847 Defset.Loc = Lex.getLoc(); 2848 RecTy *Type = ParseType(); 2849 if (!Type) 2850 return true; 2851 if (!isa<ListRecTy>(Type)) 2852 return Error(Defset.Loc, "expected list type"); 2853 Defset.EltTy = cast<ListRecTy>(Type)->getElementType(); 2854 2855 if (Lex.getCode() != tgtok::Id) 2856 return TokError("expected identifier"); 2857 StringInit *DeclName = StringInit::get(Lex.getCurStrVal()); 2858 if (Records.getGlobal(DeclName->getValue())) 2859 return TokError("def or global variable of this name already exists"); 2860 2861 if (Lex.Lex() != tgtok::equal) // Eat the identifier 2862 return TokError("expected '='"); 2863 if (Lex.Lex() != tgtok::l_brace) // Eat the '=' 2864 return TokError("expected '{'"); 2865 SMLoc BraceLoc = Lex.getLoc(); 2866 Lex.Lex(); // Eat the '{' 2867 2868 Defsets.push_back(&Defset); 2869 bool Err = ParseObjectList(nullptr); 2870 Defsets.pop_back(); 2871 if (Err) 2872 return true; 2873 2874 if (!consume(tgtok::r_brace)) { 2875 TokError("expected '}' at end of defset"); 2876 return Error(BraceLoc, "to match this '{'"); 2877 } 2878 2879 Records.addExtraGlobal(DeclName->getValue(), 2880 ListInit::get(Defset.Elements, Defset.EltTy)); 2881 return false; 2882 } 2883 2884 /// ParseDefvar - Parse a defvar statement. 2885 /// 2886 /// Defvar ::= DEFVAR Id '=' Value ';' 2887 /// 2888 bool TGParser::ParseDefvar() { 2889 assert(Lex.getCode() == tgtok::Defvar); 2890 Lex.Lex(); // Eat the 'defvar' token 2891 2892 if (Lex.getCode() != tgtok::Id) 2893 return TokError("expected identifier"); 2894 StringInit *DeclName = StringInit::get(Lex.getCurStrVal()); 2895 if (CurLocalScope) { 2896 if (CurLocalScope->varAlreadyDefined(DeclName->getValue())) 2897 return TokError("local variable of this name already exists"); 2898 } else { 2899 if (Records.getGlobal(DeclName->getValue())) 2900 return TokError("def or global variable of this name already exists"); 2901 } 2902 2903 Lex.Lex(); 2904 if (!consume(tgtok::equal)) 2905 return TokError("expected '='"); 2906 2907 Init *Value = ParseValue(nullptr); 2908 if (!Value) 2909 return true; 2910 2911 if (!consume(tgtok::semi)) 2912 return TokError("expected ';'"); 2913 2914 if (CurLocalScope) 2915 CurLocalScope->addVar(DeclName->getValue(), Value); 2916 else 2917 Records.addExtraGlobal(DeclName->getValue(), Value); 2918 2919 return false; 2920 } 2921 2922 /// ParseForeach - Parse a for statement. Return the record corresponding 2923 /// to it. This returns true on error. 2924 /// 2925 /// Foreach ::= FOREACH Declaration IN '{ ObjectList '}' 2926 /// Foreach ::= FOREACH Declaration IN Object 2927 /// 2928 bool TGParser::ParseForeach(MultiClass *CurMultiClass) { 2929 SMLoc Loc = Lex.getLoc(); 2930 assert(Lex.getCode() == tgtok::Foreach && "Unknown tok"); 2931 Lex.Lex(); // Eat the 'for' token. 2932 2933 // Make a temporary object to record items associated with the for 2934 // loop. 2935 Init *ListValue = nullptr; 2936 VarInit *IterName = ParseForeachDeclaration(ListValue); 2937 if (!IterName) 2938 return TokError("expected declaration in for"); 2939 2940 if (!consume(tgtok::In)) 2941 return TokError("Unknown tok"); 2942 2943 // Create a loop object and remember it. 2944 Loops.push_back(std::make_unique<ForeachLoop>(Loc, IterName, ListValue)); 2945 2946 // A foreach loop introduces a new scope for local variables. 2947 TGLocalVarScope *ForeachScope = PushLocalScope(); 2948 2949 if (Lex.getCode() != tgtok::l_brace) { 2950 // FOREACH Declaration IN Object 2951 if (ParseObject(CurMultiClass)) 2952 return true; 2953 } else { 2954 SMLoc BraceLoc = Lex.getLoc(); 2955 // Otherwise, this is a group foreach. 2956 Lex.Lex(); // eat the '{'. 2957 2958 // Parse the object list. 2959 if (ParseObjectList(CurMultiClass)) 2960 return true; 2961 2962 if (!consume(tgtok::r_brace)) { 2963 TokError("expected '}' at end of foreach command"); 2964 return Error(BraceLoc, "to match this '{'"); 2965 } 2966 } 2967 2968 PopLocalScope(ForeachScope); 2969 2970 // Resolve the loop or store it for later resolution. 2971 std::unique_ptr<ForeachLoop> Loop = std::move(Loops.back()); 2972 Loops.pop_back(); 2973 2974 return addEntry(std::move(Loop)); 2975 } 2976 2977 /// ParseIf - Parse an if statement. 2978 /// 2979 /// If ::= IF Value THEN IfBody 2980 /// If ::= IF Value THEN IfBody ELSE IfBody 2981 /// 2982 bool TGParser::ParseIf(MultiClass *CurMultiClass) { 2983 SMLoc Loc = Lex.getLoc(); 2984 assert(Lex.getCode() == tgtok::If && "Unknown tok"); 2985 Lex.Lex(); // Eat the 'if' token. 2986 2987 // Make a temporary object to record items associated with the for 2988 // loop. 2989 Init *Condition = ParseValue(nullptr); 2990 if (!Condition) 2991 return true; 2992 2993 if (!consume(tgtok::Then)) 2994 return TokError("Unknown tok"); 2995 2996 // We have to be able to save if statements to execute later, and they have 2997 // to live on the same stack as foreach loops. The simplest implementation 2998 // technique is to convert each 'then' or 'else' clause *into* a foreach 2999 // loop, over a list of length 0 or 1 depending on the condition, and with no 3000 // iteration variable being assigned. 3001 3002 ListInit *EmptyList = ListInit::get({}, BitRecTy::get()); 3003 ListInit *SingletonList = ListInit::get({BitInit::get(1)}, BitRecTy::get()); 3004 RecTy *BitListTy = ListRecTy::get(BitRecTy::get()); 3005 3006 // The foreach containing the then-clause selects SingletonList if 3007 // the condition is true. 3008 Init *ThenClauseList = 3009 TernOpInit::get(TernOpInit::IF, Condition, SingletonList, EmptyList, 3010 BitListTy) 3011 ->Fold(nullptr); 3012 Loops.push_back(std::make_unique<ForeachLoop>(Loc, nullptr, ThenClauseList)); 3013 3014 if (ParseIfBody(CurMultiClass, "then")) 3015 return true; 3016 3017 std::unique_ptr<ForeachLoop> Loop = std::move(Loops.back()); 3018 Loops.pop_back(); 3019 3020 if (addEntry(std::move(Loop))) 3021 return true; 3022 3023 // Now look for an optional else clause. The if-else syntax has the usual 3024 // dangling-else ambiguity, and by greedily matching an else here if we can, 3025 // we implement the usual resolution of pairing with the innermost unmatched 3026 // if. 3027 if (consume(tgtok::ElseKW)) { 3028 // The foreach containing the else-clause uses the same pair of lists as 3029 // above, but this time, selects SingletonList if the condition is *false*. 3030 Init *ElseClauseList = 3031 TernOpInit::get(TernOpInit::IF, Condition, EmptyList, SingletonList, 3032 BitListTy) 3033 ->Fold(nullptr); 3034 Loops.push_back( 3035 std::make_unique<ForeachLoop>(Loc, nullptr, ElseClauseList)); 3036 3037 if (ParseIfBody(CurMultiClass, "else")) 3038 return true; 3039 3040 Loop = std::move(Loops.back()); 3041 Loops.pop_back(); 3042 3043 if (addEntry(std::move(Loop))) 3044 return true; 3045 } 3046 3047 return false; 3048 } 3049 3050 /// ParseIfBody - Parse the then-clause or else-clause of an if statement. 3051 /// 3052 /// IfBody ::= Object 3053 /// IfBody ::= '{' ObjectList '}' 3054 /// 3055 bool TGParser::ParseIfBody(MultiClass *CurMultiClass, StringRef Kind) { 3056 TGLocalVarScope *BodyScope = PushLocalScope(); 3057 3058 if (Lex.getCode() != tgtok::l_brace) { 3059 // A single object. 3060 if (ParseObject(CurMultiClass)) 3061 return true; 3062 } else { 3063 SMLoc BraceLoc = Lex.getLoc(); 3064 // A braced block. 3065 Lex.Lex(); // eat the '{'. 3066 3067 // Parse the object list. 3068 if (ParseObjectList(CurMultiClass)) 3069 return true; 3070 3071 if (!consume(tgtok::r_brace)) { 3072 TokError("expected '}' at end of '" + Kind + "' clause"); 3073 return Error(BraceLoc, "to match this '{'"); 3074 } 3075 } 3076 3077 PopLocalScope(BodyScope); 3078 return false; 3079 } 3080 3081 /// ParseClass - Parse a tblgen class definition. 3082 /// 3083 /// ClassInst ::= CLASS ID TemplateArgList? ObjectBody 3084 /// 3085 bool TGParser::ParseClass() { 3086 assert(Lex.getCode() == tgtok::Class && "Unexpected token!"); 3087 Lex.Lex(); 3088 3089 if (Lex.getCode() != tgtok::Id) 3090 return TokError("expected class name after 'class' keyword"); 3091 3092 Record *CurRec = Records.getClass(Lex.getCurStrVal()); 3093 if (CurRec) { 3094 // If the body was previously defined, this is an error. 3095 if (!CurRec->getValues().empty() || 3096 !CurRec->getSuperClasses().empty() || 3097 !CurRec->getTemplateArgs().empty()) 3098 return TokError("Class '" + CurRec->getNameInitAsString() + 3099 "' already defined"); 3100 } else { 3101 // If this is the first reference to this class, create and add it. 3102 auto NewRec = 3103 std::make_unique<Record>(Lex.getCurStrVal(), Lex.getLoc(), Records, 3104 /*Class=*/true); 3105 CurRec = NewRec.get(); 3106 Records.addClass(std::move(NewRec)); 3107 } 3108 Lex.Lex(); // eat the name. 3109 3110 // If there are template args, parse them. 3111 if (Lex.getCode() == tgtok::less) 3112 if (ParseTemplateArgList(CurRec)) 3113 return true; 3114 3115 return ParseObjectBody(CurRec); 3116 } 3117 3118 /// ParseLetList - Parse a non-empty list of assignment expressions into a list 3119 /// of LetRecords. 3120 /// 3121 /// LetList ::= LetItem (',' LetItem)* 3122 /// LetItem ::= ID OptionalRangeList '=' Value 3123 /// 3124 void TGParser::ParseLetList(SmallVectorImpl<LetRecord> &Result) { 3125 do { 3126 if (Lex.getCode() != tgtok::Id) { 3127 TokError("expected identifier in let definition"); 3128 Result.clear(); 3129 return; 3130 } 3131 3132 StringInit *Name = StringInit::get(Lex.getCurStrVal()); 3133 SMLoc NameLoc = Lex.getLoc(); 3134 Lex.Lex(); // Eat the identifier. 3135 3136 // Check for an optional RangeList. 3137 SmallVector<unsigned, 16> Bits; 3138 if (ParseOptionalRangeList(Bits)) { 3139 Result.clear(); 3140 return; 3141 } 3142 std::reverse(Bits.begin(), Bits.end()); 3143 3144 if (!consume(tgtok::equal)) { 3145 TokError("expected '=' in let expression"); 3146 Result.clear(); 3147 return; 3148 } 3149 3150 Init *Val = ParseValue(nullptr); 3151 if (!Val) { 3152 Result.clear(); 3153 return; 3154 } 3155 3156 // Now that we have everything, add the record. 3157 Result.emplace_back(Name, Bits, Val, NameLoc); 3158 } while (consume(tgtok::comma)); 3159 } 3160 3161 /// ParseTopLevelLet - Parse a 'let' at top level. This can be a couple of 3162 /// different related productions. This works inside multiclasses too. 3163 /// 3164 /// Object ::= LET LetList IN '{' ObjectList '}' 3165 /// Object ::= LET LetList IN Object 3166 /// 3167 bool TGParser::ParseTopLevelLet(MultiClass *CurMultiClass) { 3168 assert(Lex.getCode() == tgtok::Let && "Unexpected token"); 3169 Lex.Lex(); 3170 3171 // Add this entry to the let stack. 3172 SmallVector<LetRecord, 8> LetInfo; 3173 ParseLetList(LetInfo); 3174 if (LetInfo.empty()) return true; 3175 LetStack.push_back(std::move(LetInfo)); 3176 3177 if (!consume(tgtok::In)) 3178 return TokError("expected 'in' at end of top-level 'let'"); 3179 3180 TGLocalVarScope *LetScope = PushLocalScope(); 3181 3182 // If this is a scalar let, just handle it now 3183 if (Lex.getCode() != tgtok::l_brace) { 3184 // LET LetList IN Object 3185 if (ParseObject(CurMultiClass)) 3186 return true; 3187 } else { // Object ::= LETCommand '{' ObjectList '}' 3188 SMLoc BraceLoc = Lex.getLoc(); 3189 // Otherwise, this is a group let. 3190 Lex.Lex(); // eat the '{'. 3191 3192 // Parse the object list. 3193 if (ParseObjectList(CurMultiClass)) 3194 return true; 3195 3196 if (!consume(tgtok::r_brace)) { 3197 TokError("expected '}' at end of top level let command"); 3198 return Error(BraceLoc, "to match this '{'"); 3199 } 3200 } 3201 3202 PopLocalScope(LetScope); 3203 3204 // Outside this let scope, this let block is not active. 3205 LetStack.pop_back(); 3206 return false; 3207 } 3208 3209 /// ParseMultiClass - Parse a multiclass definition. 3210 /// 3211 /// MultiClassInst ::= MULTICLASS ID TemplateArgList? 3212 /// ':' BaseMultiClassList '{' MultiClassObject+ '}' 3213 /// MultiClassObject ::= DefInst 3214 /// MultiClassObject ::= MultiClassInst 3215 /// MultiClassObject ::= DefMInst 3216 /// MultiClassObject ::= LETCommand '{' ObjectList '}' 3217 /// MultiClassObject ::= LETCommand Object 3218 /// 3219 bool TGParser::ParseMultiClass() { 3220 assert(Lex.getCode() == tgtok::MultiClass && "Unexpected token"); 3221 Lex.Lex(); // Eat the multiclass token. 3222 3223 if (Lex.getCode() != tgtok::Id) 3224 return TokError("expected identifier after multiclass for name"); 3225 std::string Name = Lex.getCurStrVal(); 3226 3227 auto Result = 3228 MultiClasses.insert(std::make_pair(Name, 3229 std::make_unique<MultiClass>(Name, Lex.getLoc(),Records))); 3230 3231 if (!Result.second) 3232 return TokError("multiclass '" + Name + "' already defined"); 3233 3234 CurMultiClass = Result.first->second.get(); 3235 Lex.Lex(); // Eat the identifier. 3236 3237 // If there are template args, parse them. 3238 if (Lex.getCode() == tgtok::less) 3239 if (ParseTemplateArgList(nullptr)) 3240 return true; 3241 3242 bool inherits = false; 3243 3244 // If there are submulticlasses, parse them. 3245 if (consume(tgtok::colon)) { 3246 inherits = true; 3247 3248 // Read all of the submulticlasses. 3249 SubMultiClassReference SubMultiClass = 3250 ParseSubMultiClassReference(CurMultiClass); 3251 while (true) { 3252 // Check for error. 3253 if (!SubMultiClass.MC) return true; 3254 3255 // Add it. 3256 if (AddSubMultiClass(CurMultiClass, SubMultiClass)) 3257 return true; 3258 3259 if (!consume(tgtok::comma)) 3260 break; 3261 SubMultiClass = ParseSubMultiClassReference(CurMultiClass); 3262 } 3263 } 3264 3265 if (Lex.getCode() != tgtok::l_brace) { 3266 if (!inherits) 3267 return TokError("expected '{' in multiclass definition"); 3268 if (!consume(tgtok::semi)) 3269 return TokError("expected ';' in multiclass definition"); 3270 } else { 3271 if (Lex.Lex() == tgtok::r_brace) // eat the '{'. 3272 return TokError("multiclass must contain at least one def"); 3273 3274 // A multiclass body introduces a new scope for local variables. 3275 TGLocalVarScope *MulticlassScope = PushLocalScope(); 3276 3277 while (Lex.getCode() != tgtok::r_brace) { 3278 switch (Lex.getCode()) { 3279 default: 3280 return TokError("expected 'let', 'def', 'defm', 'defvar', 'foreach' " 3281 "or 'if' in multiclass body"); 3282 case tgtok::Let: 3283 case tgtok::Def: 3284 case tgtok::Defm: 3285 case tgtok::Defvar: 3286 case tgtok::Foreach: 3287 case tgtok::If: 3288 if (ParseObject(CurMultiClass)) 3289 return true; 3290 break; 3291 } 3292 } 3293 Lex.Lex(); // eat the '}'. 3294 3295 PopLocalScope(MulticlassScope); 3296 } 3297 3298 CurMultiClass = nullptr; 3299 return false; 3300 } 3301 3302 /// ParseDefm - Parse the instantiation of a multiclass. 3303 /// 3304 /// DefMInst ::= DEFM ID ':' DefmSubClassRef ';' 3305 /// 3306 bool TGParser::ParseDefm(MultiClass *CurMultiClass) { 3307 assert(Lex.getCode() == tgtok::Defm && "Unexpected token!"); 3308 Lex.Lex(); // eat the defm 3309 3310 Init *DefmName = ParseObjectName(CurMultiClass); 3311 if (!DefmName) 3312 return true; 3313 if (isa<UnsetInit>(DefmName)) { 3314 DefmName = Records.getNewAnonymousName(); 3315 if (CurMultiClass) 3316 DefmName = BinOpInit::getStrConcat( 3317 VarInit::get(QualifiedNameOfImplicitName(CurMultiClass), 3318 StringRecTy::get()), 3319 DefmName); 3320 } 3321 3322 if (Lex.getCode() != tgtok::colon) 3323 return TokError("expected ':' after defm identifier"); 3324 3325 // Keep track of the new generated record definitions. 3326 std::vector<RecordsEntry> NewEntries; 3327 3328 // This record also inherits from a regular class (non-multiclass)? 3329 bool InheritFromClass = false; 3330 3331 // eat the colon. 3332 Lex.Lex(); 3333 3334 SMLoc SubClassLoc = Lex.getLoc(); 3335 SubClassReference Ref = ParseSubClassReference(nullptr, true); 3336 3337 while (true) { 3338 if (!Ref.Rec) return true; 3339 3340 // To instantiate a multiclass, we need to first get the multiclass, then 3341 // instantiate each def contained in the multiclass with the SubClassRef 3342 // template parameters. 3343 MultiClass *MC = MultiClasses[std::string(Ref.Rec->getName())].get(); 3344 assert(MC && "Didn't lookup multiclass correctly?"); 3345 ArrayRef<Init*> TemplateVals = Ref.TemplateArgs; 3346 3347 // Verify that the correct number of template arguments were specified. 3348 ArrayRef<Init *> TArgs = MC->Rec.getTemplateArgs(); 3349 if (TArgs.size() < TemplateVals.size()) 3350 return Error(SubClassLoc, 3351 "more template args specified than multiclass expects"); 3352 3353 SubstStack Substs; 3354 for (unsigned i = 0, e = TArgs.size(); i != e; ++i) { 3355 if (i < TemplateVals.size()) { 3356 Substs.emplace_back(TArgs[i], TemplateVals[i]); 3357 } else { 3358 Init *Default = MC->Rec.getValue(TArgs[i])->getValue(); 3359 if (!Default->isComplete()) { 3360 return Error(SubClassLoc, 3361 "value not specified for template argument #" + 3362 Twine(i) + " (" + TArgs[i]->getAsUnquotedString() + 3363 ") of multiclass '" + MC->Rec.getNameInitAsString() + 3364 "'"); 3365 } 3366 Substs.emplace_back(TArgs[i], Default); 3367 } 3368 } 3369 3370 Substs.emplace_back(QualifiedNameOfImplicitName(MC), DefmName); 3371 3372 if (resolve(MC->Entries, Substs, CurMultiClass == nullptr, &NewEntries, 3373 &SubClassLoc)) 3374 return true; 3375 3376 if (!consume(tgtok::comma)) 3377 break; 3378 3379 if (Lex.getCode() != tgtok::Id) 3380 return TokError("expected identifier"); 3381 3382 SubClassLoc = Lex.getLoc(); 3383 3384 // A defm can inherit from regular classes (non-multiclass) as 3385 // long as they come in the end of the inheritance list. 3386 InheritFromClass = (Records.getClass(Lex.getCurStrVal()) != nullptr); 3387 3388 if (InheritFromClass) 3389 break; 3390 3391 Ref = ParseSubClassReference(nullptr, true); 3392 } 3393 3394 if (InheritFromClass) { 3395 // Process all the classes to inherit as if they were part of a 3396 // regular 'def' and inherit all record values. 3397 SubClassReference SubClass = ParseSubClassReference(nullptr, false); 3398 while (true) { 3399 // Check for error. 3400 if (!SubClass.Rec) return true; 3401 3402 // Get the expanded definition prototypes and teach them about 3403 // the record values the current class to inherit has 3404 for (auto &E : NewEntries) { 3405 // Add it. 3406 if (AddSubClass(E, SubClass)) 3407 return true; 3408 } 3409 3410 if (!consume(tgtok::comma)) 3411 break; 3412 SubClass = ParseSubClassReference(nullptr, false); 3413 } 3414 } 3415 3416 for (auto &E : NewEntries) { 3417 if (ApplyLetStack(E)) 3418 return true; 3419 3420 addEntry(std::move(E)); 3421 } 3422 3423 if (!consume(tgtok::semi)) 3424 return TokError("expected ';' at end of defm"); 3425 3426 return false; 3427 } 3428 3429 /// ParseObject 3430 /// Object ::= ClassInst 3431 /// Object ::= DefInst 3432 /// Object ::= MultiClassInst 3433 /// Object ::= DefMInst 3434 /// Object ::= LETCommand '{' ObjectList '}' 3435 /// Object ::= LETCommand Object 3436 /// Object ::= Defset 3437 /// Object ::= Defvar 3438 bool TGParser::ParseObject(MultiClass *MC) { 3439 switch (Lex.getCode()) { 3440 default: 3441 return TokError("Expected class, def, defm, defset, multiclass, let, " 3442 "foreach or if"); 3443 case tgtok::Let: return ParseTopLevelLet(MC); 3444 case tgtok::Def: return ParseDef(MC); 3445 case tgtok::Foreach: return ParseForeach(MC); 3446 case tgtok::If: return ParseIf(MC); 3447 case tgtok::Defm: return ParseDefm(MC); 3448 case tgtok::Defset: 3449 if (MC) 3450 return TokError("defset is not allowed inside multiclass"); 3451 return ParseDefset(); 3452 case tgtok::Defvar: 3453 return ParseDefvar(); 3454 case tgtok::Class: 3455 if (MC) 3456 return TokError("class is not allowed inside multiclass"); 3457 if (!Loops.empty()) 3458 return TokError("class is not allowed inside foreach loop"); 3459 return ParseClass(); 3460 case tgtok::MultiClass: 3461 if (!Loops.empty()) 3462 return TokError("multiclass is not allowed inside foreach loop"); 3463 return ParseMultiClass(); 3464 } 3465 } 3466 3467 /// ParseObjectList 3468 /// ObjectList :== Object* 3469 bool TGParser::ParseObjectList(MultiClass *MC) { 3470 while (isObjectStart(Lex.getCode())) { 3471 if (ParseObject(MC)) 3472 return true; 3473 } 3474 return false; 3475 } 3476 3477 bool TGParser::ParseFile() { 3478 Lex.Lex(); // Prime the lexer. 3479 if (ParseObjectList()) return true; 3480 3481 // If we have unread input at the end of the file, report it. 3482 if (Lex.getCode() == tgtok::Eof) 3483 return false; 3484 3485 return TokError("Unexpected input at top level"); 3486 } 3487 3488 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 3489 LLVM_DUMP_METHOD void RecordsEntry::dump() const { 3490 if (Loop) 3491 Loop->dump(); 3492 if (Rec) 3493 Rec->dump(); 3494 } 3495 3496 LLVM_DUMP_METHOD void ForeachLoop::dump() const { 3497 errs() << "foreach " << IterVar->getAsString() << " = " 3498 << ListValue->getAsString() << " in {\n"; 3499 3500 for (const auto &E : Entries) 3501 E.dump(); 3502 3503 errs() << "}\n"; 3504 } 3505 3506 LLVM_DUMP_METHOD void MultiClass::dump() const { 3507 errs() << "Record:\n"; 3508 Rec.dump(); 3509 3510 errs() << "Defs:\n"; 3511 for (const auto &E : Entries) 3512 E.dump(); 3513 } 3514 #endif 3515