Use unique_ptr to fix some memory leaks in Tablegen AsmMatcherEmitter.
[oota-llvm.git] / utils / TableGen / AsmMatcherEmitter.cpp
1 //===- AsmMatcherEmitter.cpp - Generate an assembly matcher ---------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This tablegen backend emits a target specifier matcher for converting parsed
11 // assembly operands in the MCInst structures. It also emits a matcher for
12 // custom operand parsing.
13 //
14 // Converting assembly operands into MCInst structures
15 // ---------------------------------------------------
16 //
17 // The input to the target specific matcher is a list of literal tokens and
18 // operands. The target specific parser should generally eliminate any syntax
19 // which is not relevant for matching; for example, comma tokens should have
20 // already been consumed and eliminated by the parser. Most instructions will
21 // end up with a single literal token (the instruction name) and some number of
22 // operands.
23 //
24 // Some example inputs, for X86:
25 //   'addl' (immediate ...) (register ...)
26 //   'add' (immediate ...) (memory ...)
27 //   'call' '*' %epc
28 //
29 // The assembly matcher is responsible for converting this input into a precise
30 // machine instruction (i.e., an instruction with a well defined encoding). This
31 // mapping has several properties which complicate matching:
32 //
33 //  - It may be ambiguous; many architectures can legally encode particular
34 //    variants of an instruction in different ways (for example, using a smaller
35 //    encoding for small immediates). Such ambiguities should never be
36 //    arbitrarily resolved by the assembler, the assembler is always responsible
37 //    for choosing the "best" available instruction.
38 //
39 //  - It may depend on the subtarget or the assembler context. Instructions
40 //    which are invalid for the current mode, but otherwise unambiguous (e.g.,
41 //    an SSE instruction in a file being assembled for i486) should be accepted
42 //    and rejected by the assembler front end. However, if the proper encoding
43 //    for an instruction is dependent on the assembler context then the matcher
44 //    is responsible for selecting the correct machine instruction for the
45 //    current mode.
46 //
47 // The core matching algorithm attempts to exploit the regularity in most
48 // instruction sets to quickly determine the set of possibly matching
49 // instructions, and the simplify the generated code. Additionally, this helps
50 // to ensure that the ambiguities are intentionally resolved by the user.
51 //
52 // The matching is divided into two distinct phases:
53 //
54 //   1. Classification: Each operand is mapped to the unique set which (a)
55 //      contains it, and (b) is the largest such subset for which a single
56 //      instruction could match all members.
57 //
58 //      For register classes, we can generate these subgroups automatically. For
59 //      arbitrary operands, we expect the user to define the classes and their
60 //      relations to one another (for example, 8-bit signed immediates as a
61 //      subset of 32-bit immediates).
62 //
63 //      By partitioning the operands in this way, we guarantee that for any
64 //      tuple of classes, any single instruction must match either all or none
65 //      of the sets of operands which could classify to that tuple.
66 //
67 //      In addition, the subset relation amongst classes induces a partial order
68 //      on such tuples, which we use to resolve ambiguities.
69 //
70 //   2. The input can now be treated as a tuple of classes (static tokens are
71 //      simple singleton sets). Each such tuple should generally map to a single
72 //      instruction (we currently ignore cases where this isn't true, whee!!!),
73 //      which we can emit a simple matcher for.
74 //
75 // Custom Operand Parsing
76 // ----------------------
77 //
78 //  Some targets need a custom way to parse operands, some specific instructions
79 //  can contain arguments that can represent processor flags and other kinds of
80 //  identifiers that need to be mapped to specific values in the final encoded
81 //  instructions. The target specific custom operand parsing works in the
82 //  following way:
83 //
84 //   1. A operand match table is built, each entry contains a mnemonic, an
85 //      operand class, a mask for all operand positions for that same
86 //      class/mnemonic and target features to be checked while trying to match.
87 //
88 //   2. The operand matcher will try every possible entry with the same
89 //      mnemonic and will check if the target feature for this mnemonic also
90 //      matches. After that, if the operand to be matched has its index
91 //      present in the mask, a successful match occurs. Otherwise, fallback
92 //      to the regular operand parsing.
93 //
94 //   3. For a match success, each operand class that has a 'ParserMethod'
95 //      becomes part of a switch from where the custom method is called.
96 //
97 //===----------------------------------------------------------------------===//
98
99 #include "CodeGenTarget.h"
100 #include "llvm/ADT/PointerUnion.h"
101 #include "llvm/ADT/STLExtras.h"
102 #include "llvm/ADT/SmallPtrSet.h"
103 #include "llvm/ADT/SmallVector.h"
104 #include "llvm/ADT/StringExtras.h"
105 #include "llvm/Support/CommandLine.h"
106 #include "llvm/Support/Debug.h"
107 #include "llvm/Support/ErrorHandling.h"
108 #include "llvm/TableGen/Error.h"
109 #include "llvm/TableGen/Record.h"
110 #include "llvm/TableGen/StringMatcher.h"
111 #include "llvm/TableGen/StringToOffsetTable.h"
112 #include "llvm/TableGen/TableGenBackend.h"
113 #include <cassert>
114 #include <cctype>
115 #include <map>
116 #include <set>
117 #include <sstream>
118 using namespace llvm;
119
120 #define DEBUG_TYPE "asm-matcher-emitter"
121
122 static cl::opt<std::string>
123 MatchPrefix("match-prefix", cl::init(""),
124             cl::desc("Only match instructions with the given prefix"));
125
126 namespace {
127 class AsmMatcherInfo;
128 struct SubtargetFeatureInfo;
129
130 // Register sets are used as keys in some second-order sets TableGen creates
131 // when generating its data structures. This means that the order of two
132 // RegisterSets can be seen in the outputted AsmMatcher tables occasionally, and
133 // can even affect compiler output (at least seen in diagnostics produced when
134 // all matches fail). So we use a type that sorts them consistently.
135 typedef std::set<Record*, LessRecordByID> RegisterSet;
136
137 class AsmMatcherEmitter {
138   RecordKeeper &Records;
139 public:
140   AsmMatcherEmitter(RecordKeeper &R) : Records(R) {}
141
142   void run(raw_ostream &o);
143 };
144
145 /// ClassInfo - Helper class for storing the information about a particular
146 /// class of operands which can be matched.
147 struct ClassInfo {
148   enum ClassInfoKind {
149     /// Invalid kind, for use as a sentinel value.
150     Invalid = 0,
151
152     /// The class for a particular token.
153     Token,
154
155     /// The (first) register class, subsequent register classes are
156     /// RegisterClass0+1, and so on.
157     RegisterClass0,
158
159     /// The (first) user defined class, subsequent user defined classes are
160     /// UserClass0+1, and so on.
161     UserClass0 = 1<<16
162   };
163
164   /// Kind - The class kind, which is either a predefined kind, or (UserClass0 +
165   /// N) for the Nth user defined class.
166   unsigned Kind;
167
168   /// SuperClasses - The super classes of this class. Note that for simplicities
169   /// sake user operands only record their immediate super class, while register
170   /// operands include all superclasses.
171   std::vector<ClassInfo*> SuperClasses;
172
173   /// Name - The full class name, suitable for use in an enum.
174   std::string Name;
175
176   /// ClassName - The unadorned generic name for this class (e.g., Token).
177   std::string ClassName;
178
179   /// ValueName - The name of the value this class represents; for a token this
180   /// is the literal token string, for an operand it is the TableGen class (or
181   /// empty if this is a derived class).
182   std::string ValueName;
183
184   /// PredicateMethod - The name of the operand method to test whether the
185   /// operand matches this class; this is not valid for Token or register kinds.
186   std::string PredicateMethod;
187
188   /// RenderMethod - The name of the operand method to add this operand to an
189   /// MCInst; this is not valid for Token or register kinds.
190   std::string RenderMethod;
191
192   /// ParserMethod - The name of the operand method to do a target specific
193   /// parsing on the operand.
194   std::string ParserMethod;
195
196   /// For register classes: the records for all the registers in this class.
197   RegisterSet Registers;
198
199   /// For custom match classes: the diagnostic kind for when the predicate fails.
200   std::string DiagnosticType;
201 public:
202   /// isRegisterClass() - Check if this is a register class.
203   bool isRegisterClass() const {
204     return Kind >= RegisterClass0 && Kind < UserClass0;
205   }
206
207   /// isUserClass() - Check if this is a user defined class.
208   bool isUserClass() const {
209     return Kind >= UserClass0;
210   }
211
212   /// isRelatedTo - Check whether this class is "related" to \p RHS. Classes
213   /// are related if they are in the same class hierarchy.
214   bool isRelatedTo(const ClassInfo &RHS) const {
215     // Tokens are only related to tokens.
216     if (Kind == Token || RHS.Kind == Token)
217       return Kind == Token && RHS.Kind == Token;
218
219     // Registers classes are only related to registers classes, and only if
220     // their intersection is non-empty.
221     if (isRegisterClass() || RHS.isRegisterClass()) {
222       if (!isRegisterClass() || !RHS.isRegisterClass())
223         return false;
224
225       RegisterSet Tmp;
226       std::insert_iterator<RegisterSet> II(Tmp, Tmp.begin());
227       std::set_intersection(Registers.begin(), Registers.end(),
228                             RHS.Registers.begin(), RHS.Registers.end(),
229                             II, LessRecordByID());
230
231       return !Tmp.empty();
232     }
233
234     // Otherwise we have two users operands; they are related if they are in the
235     // same class hierarchy.
236     //
237     // FIXME: This is an oversimplification, they should only be related if they
238     // intersect, however we don't have that information.
239     assert(isUserClass() && RHS.isUserClass() && "Unexpected class!");
240     const ClassInfo *Root = this;
241     while (!Root->SuperClasses.empty())
242       Root = Root->SuperClasses.front();
243
244     const ClassInfo *RHSRoot = &RHS;
245     while (!RHSRoot->SuperClasses.empty())
246       RHSRoot = RHSRoot->SuperClasses.front();
247
248     return Root == RHSRoot;
249   }
250
251   /// isSubsetOf - Test whether this class is a subset of \p RHS.
252   bool isSubsetOf(const ClassInfo &RHS) const {
253     // This is a subset of RHS if it is the same class...
254     if (this == &RHS)
255       return true;
256
257     // ... or if any of its super classes are a subset of RHS.
258     for (const ClassInfo *CI : SuperClasses)
259       if (CI->isSubsetOf(RHS))
260         return true;
261
262     return false;
263   }
264
265   /// operator< - Compare two classes.
266   bool operator<(const ClassInfo &RHS) const {
267     if (this == &RHS)
268       return false;
269
270     // Unrelated classes can be ordered by kind.
271     if (!isRelatedTo(RHS))
272       return Kind < RHS.Kind;
273
274     switch (Kind) {
275     case Invalid:
276       llvm_unreachable("Invalid kind!");
277
278     default:
279       // This class precedes the RHS if it is a proper subset of the RHS.
280       if (isSubsetOf(RHS))
281         return true;
282       if (RHS.isSubsetOf(*this))
283         return false;
284
285       // Otherwise, order by name to ensure we have a total ordering.
286       return ValueName < RHS.ValueName;
287     }
288   }
289 };
290
291 /// MatchableInfo - Helper class for storing the necessary information for an
292 /// instruction or alias which is capable of being matched.
293 struct MatchableInfo {
294   struct AsmOperand {
295     /// Token - This is the token that the operand came from.
296     StringRef Token;
297
298     /// The unique class instance this operand should match.
299     ClassInfo *Class;
300
301     /// The operand name this is, if anything.
302     StringRef SrcOpName;
303
304     /// The suboperand index within SrcOpName, or -1 for the entire operand.
305     int SubOpIdx;
306
307     /// Register record if this token is singleton register.
308     Record *SingletonReg;
309
310     explicit AsmOperand(StringRef T) : Token(T), Class(nullptr), SubOpIdx(-1),
311                                        SingletonReg(nullptr) {}
312   };
313
314   /// ResOperand - This represents a single operand in the result instruction
315   /// generated by the match.  In cases (like addressing modes) where a single
316   /// assembler operand expands to multiple MCOperands, this represents the
317   /// single assembler operand, not the MCOperand.
318   struct ResOperand {
319     enum {
320       /// RenderAsmOperand - This represents an operand result that is
321       /// generated by calling the render method on the assembly operand.  The
322       /// corresponding AsmOperand is specified by AsmOperandNum.
323       RenderAsmOperand,
324
325       /// TiedOperand - This represents a result operand that is a duplicate of
326       /// a previous result operand.
327       TiedOperand,
328
329       /// ImmOperand - This represents an immediate value that is dumped into
330       /// the operand.
331       ImmOperand,
332
333       /// RegOperand - This represents a fixed register that is dumped in.
334       RegOperand
335     } Kind;
336
337     union {
338       /// This is the operand # in the AsmOperands list that this should be
339       /// copied from.
340       unsigned AsmOperandNum;
341
342       /// TiedOperandNum - This is the (earlier) result operand that should be
343       /// copied from.
344       unsigned TiedOperandNum;
345
346       /// ImmVal - This is the immediate value added to the instruction.
347       int64_t ImmVal;
348
349       /// Register - This is the register record.
350       Record *Register;
351     };
352
353     /// MINumOperands - The number of MCInst operands populated by this
354     /// operand.
355     unsigned MINumOperands;
356
357     static ResOperand getRenderedOp(unsigned AsmOpNum, unsigned NumOperands) {
358       ResOperand X;
359       X.Kind = RenderAsmOperand;
360       X.AsmOperandNum = AsmOpNum;
361       X.MINumOperands = NumOperands;
362       return X;
363     }
364
365     static ResOperand getTiedOp(unsigned TiedOperandNum) {
366       ResOperand X;
367       X.Kind = TiedOperand;
368       X.TiedOperandNum = TiedOperandNum;
369       X.MINumOperands = 1;
370       return X;
371     }
372
373     static ResOperand getImmOp(int64_t Val) {
374       ResOperand X;
375       X.Kind = ImmOperand;
376       X.ImmVal = Val;
377       X.MINumOperands = 1;
378       return X;
379     }
380
381     static ResOperand getRegOp(Record *Reg) {
382       ResOperand X;
383       X.Kind = RegOperand;
384       X.Register = Reg;
385       X.MINumOperands = 1;
386       return X;
387     }
388   };
389
390   /// AsmVariantID - Target's assembly syntax variant no.
391   int AsmVariantID;
392
393   /// TheDef - This is the definition of the instruction or InstAlias that this
394   /// matchable came from.
395   Record *const TheDef;
396
397   /// DefRec - This is the definition that it came from.
398   PointerUnion<const CodeGenInstruction*, const CodeGenInstAlias*> DefRec;
399
400   const CodeGenInstruction *getResultInst() const {
401     if (DefRec.is<const CodeGenInstruction*>())
402       return DefRec.get<const CodeGenInstruction*>();
403     return DefRec.get<const CodeGenInstAlias*>()->ResultInst;
404   }
405
406   /// ResOperands - This is the operand list that should be built for the result
407   /// MCInst.
408   SmallVector<ResOperand, 8> ResOperands;
409
410   /// AsmString - The assembly string for this instruction (with variants
411   /// removed), e.g. "movsx $src, $dst".
412   std::string AsmString;
413
414   /// Mnemonic - This is the first token of the matched instruction, its
415   /// mnemonic.
416   StringRef Mnemonic;
417
418   /// AsmOperands - The textual operands that this instruction matches,
419   /// annotated with a class and where in the OperandList they were defined.
420   /// This directly corresponds to the tokenized AsmString after the mnemonic is
421   /// removed.
422   SmallVector<AsmOperand, 8> AsmOperands;
423
424   /// Predicates - The required subtarget features to match this instruction.
425   SmallVector<SubtargetFeatureInfo*, 4> RequiredFeatures;
426
427   /// ConversionFnKind - The enum value which is passed to the generated
428   /// convertToMCInst to convert parsed operands into an MCInst for this
429   /// function.
430   std::string ConversionFnKind;
431
432   /// If this instruction is deprecated in some form.
433   bool HasDeprecation;
434
435   MatchableInfo(const CodeGenInstruction &CGI)
436     : AsmVariantID(0), TheDef(CGI.TheDef), DefRec(&CGI),
437       AsmString(CGI.AsmString) {
438   }
439
440   MatchableInfo(const CodeGenInstAlias *Alias)
441     : AsmVariantID(0), TheDef(Alias->TheDef), DefRec(Alias),
442       AsmString(Alias->AsmString) {
443   }
444
445   // Two-operand aliases clone from the main matchable, but mark the second
446   // operand as a tied operand of the first for purposes of the assembler.
447   void formTwoOperandAlias(StringRef Constraint);
448
449   void initialize(const AsmMatcherInfo &Info,
450                   SmallPtrSetImpl<Record*> &SingletonRegisters,
451                   int AsmVariantNo, std::string &RegisterPrefix);
452
453   /// validate - Return true if this matchable is a valid thing to match against
454   /// and perform a bunch of validity checking.
455   bool validate(StringRef CommentDelimiter, bool Hack) const;
456
457   /// extractSingletonRegisterForAsmOperand - Extract singleton register,
458   /// if present, from specified token.
459   void
460   extractSingletonRegisterForAsmOperand(unsigned i, const AsmMatcherInfo &Info,
461                                         std::string &RegisterPrefix);
462
463   /// findAsmOperand - Find the AsmOperand with the specified name and
464   /// suboperand index.
465   int findAsmOperand(StringRef N, int SubOpIdx) const {
466     for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i)
467       if (N == AsmOperands[i].SrcOpName &&
468           SubOpIdx == AsmOperands[i].SubOpIdx)
469         return i;
470     return -1;
471   }
472
473   /// findAsmOperandNamed - Find the first AsmOperand with the specified name.
474   /// This does not check the suboperand index.
475   int findAsmOperandNamed(StringRef N) const {
476     for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i)
477       if (N == AsmOperands[i].SrcOpName)
478         return i;
479     return -1;
480   }
481
482   void buildInstructionResultOperands();
483   void buildAliasResultOperands();
484
485   /// operator< - Compare two matchables.
486   bool operator<(const MatchableInfo &RHS) const {
487     // The primary comparator is the instruction mnemonic.
488     if (Mnemonic != RHS.Mnemonic)
489       return Mnemonic < RHS.Mnemonic;
490
491     if (AsmOperands.size() != RHS.AsmOperands.size())
492       return AsmOperands.size() < RHS.AsmOperands.size();
493
494     // Compare lexicographically by operand. The matcher validates that other
495     // orderings wouldn't be ambiguous using \see couldMatchAmbiguouslyWith().
496     for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i) {
497       if (*AsmOperands[i].Class < *RHS.AsmOperands[i].Class)
498         return true;
499       if (*RHS.AsmOperands[i].Class < *AsmOperands[i].Class)
500         return false;
501     }
502
503     // Give matches that require more features higher precedence. This is useful
504     // because we cannot define AssemblerPredicates with the negation of
505     // processor features. For example, ARM v6 "nop" may be either a HINT or
506     // MOV. With v6, we want to match HINT. The assembler has no way to
507     // predicate MOV under "NoV6", but HINT will always match first because it
508     // requires V6 while MOV does not.
509     if (RequiredFeatures.size() != RHS.RequiredFeatures.size())
510       return RequiredFeatures.size() > RHS.RequiredFeatures.size();
511
512     return false;
513   }
514
515   /// couldMatchAmbiguouslyWith - Check whether this matchable could
516   /// ambiguously match the same set of operands as \p RHS (without being a
517   /// strictly superior match).
518   bool couldMatchAmbiguouslyWith(const MatchableInfo &RHS) const {
519     // The primary comparator is the instruction mnemonic.
520     if (Mnemonic != RHS.Mnemonic)
521       return false;
522
523     // The number of operands is unambiguous.
524     if (AsmOperands.size() != RHS.AsmOperands.size())
525       return false;
526
527     // Otherwise, make sure the ordering of the two instructions is unambiguous
528     // by checking that either (a) a token or operand kind discriminates them,
529     // or (b) the ordering among equivalent kinds is consistent.
530
531     // Tokens and operand kinds are unambiguous (assuming a correct target
532     // specific parser).
533     for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i)
534       if (AsmOperands[i].Class->Kind != RHS.AsmOperands[i].Class->Kind ||
535           AsmOperands[i].Class->Kind == ClassInfo::Token)
536         if (*AsmOperands[i].Class < *RHS.AsmOperands[i].Class ||
537             *RHS.AsmOperands[i].Class < *AsmOperands[i].Class)
538           return false;
539
540     // Otherwise, this operand could commute if all operands are equivalent, or
541     // there is a pair of operands that compare less than and a pair that
542     // compare greater than.
543     bool HasLT = false, HasGT = false;
544     for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i) {
545       if (*AsmOperands[i].Class < *RHS.AsmOperands[i].Class)
546         HasLT = true;
547       if (*RHS.AsmOperands[i].Class < *AsmOperands[i].Class)
548         HasGT = true;
549     }
550
551     return !(HasLT ^ HasGT);
552   }
553
554   void dump() const;
555
556 private:
557   void tokenizeAsmString(const AsmMatcherInfo &Info);
558 };
559
560 /// SubtargetFeatureInfo - Helper class for storing information on a subtarget
561 /// feature which participates in instruction matching.
562 struct SubtargetFeatureInfo {
563   /// \brief The predicate record for this feature.
564   Record *TheDef;
565
566   /// \brief An unique index assigned to represent this feature.
567   uint64_t Index;
568
569   SubtargetFeatureInfo(Record *D, uint64_t Idx) : TheDef(D), Index(Idx) {}
570
571   /// \brief The name of the enumerated constant identifying this feature.
572   std::string getEnumName() const {
573     return "Feature_" + TheDef->getName();
574   }
575
576   void dump() const {
577     errs() << getEnumName() << " " << Index << "\n";
578     TheDef->dump();
579   }
580 };
581
582 struct OperandMatchEntry {
583   unsigned OperandMask;
584   const MatchableInfo* MI;
585   ClassInfo *CI;
586
587   static OperandMatchEntry create(const MatchableInfo *mi, ClassInfo *ci,
588                                   unsigned opMask) {
589     OperandMatchEntry X;
590     X.OperandMask = opMask;
591     X.CI = ci;
592     X.MI = mi;
593     return X;
594   }
595 };
596
597
598 class AsmMatcherInfo {
599 public:
600   /// Tracked Records
601   RecordKeeper &Records;
602
603   /// The tablegen AsmParser record.
604   Record *AsmParser;
605
606   /// Target - The target information.
607   CodeGenTarget &Target;
608
609   /// The classes which are needed for matching.
610   std::vector<std::unique_ptr<ClassInfo>> Classes;
611
612   /// The information on the matchables to match.
613   std::vector<std::unique_ptr<MatchableInfo>> Matchables;
614
615   /// Info for custom matching operands by user defined methods.
616   std::vector<OperandMatchEntry> OperandMatchInfo;
617
618   /// Map of Register records to their class information.
619   typedef std::map<Record*, ClassInfo*, LessRecordByID> RegisterClassesTy;
620   RegisterClassesTy RegisterClasses;
621
622   /// Map of Predicate records to their subtarget information.
623   std::map<Record*, std::unique_ptr<SubtargetFeatureInfo>,
624            LessRecordByID> SubtargetFeatures;
625
626   /// Map of AsmOperandClass records to their class information.
627   std::map<Record*, ClassInfo*> AsmOperandClasses;
628
629 private:
630   /// Map of token to class information which has already been constructed.
631   std::map<std::string, ClassInfo*> TokenClasses;
632
633   /// Map of RegisterClass records to their class information.
634   std::map<Record*, ClassInfo*> RegisterClassClasses;
635
636 private:
637   /// getTokenClass - Lookup or create the class for the given token.
638   ClassInfo *getTokenClass(StringRef Token);
639
640   /// getOperandClass - Lookup or create the class for the given operand.
641   ClassInfo *getOperandClass(const CGIOperandList::OperandInfo &OI,
642                              int SubOpIdx);
643   ClassInfo *getOperandClass(Record *Rec, int SubOpIdx);
644
645   /// buildRegisterClasses - Build the ClassInfo* instances for register
646   /// classes.
647   void buildRegisterClasses(SmallPtrSetImpl<Record*> &SingletonRegisters);
648
649   /// buildOperandClasses - Build the ClassInfo* instances for user defined
650   /// operand classes.
651   void buildOperandClasses();
652
653   void buildInstructionOperandReference(MatchableInfo *II, StringRef OpName,
654                                         unsigned AsmOpIdx);
655   void buildAliasOperandReference(MatchableInfo *II, StringRef OpName,
656                                   MatchableInfo::AsmOperand &Op);
657
658 public:
659   AsmMatcherInfo(Record *AsmParser,
660                  CodeGenTarget &Target,
661                  RecordKeeper &Records);
662
663   /// buildInfo - Construct the various tables used during matching.
664   void buildInfo();
665
666   /// buildOperandMatchInfo - Build the necessary information to handle user
667   /// defined operand parsing methods.
668   void buildOperandMatchInfo();
669
670   /// getSubtargetFeature - Lookup or create the subtarget feature info for the
671   /// given operand.
672   SubtargetFeatureInfo *getSubtargetFeature(Record *Def) const {
673     assert(Def->isSubClassOf("Predicate") && "Invalid predicate type!");
674     const auto &I = SubtargetFeatures.find(Def);
675     return I == SubtargetFeatures.end() ? nullptr : I->second.get();
676   }
677
678   RecordKeeper &getRecords() const {
679     return Records;
680   }
681 };
682
683 } // End anonymous namespace
684
685 void MatchableInfo::dump() const {
686   errs() << TheDef->getName() << " -- " << "flattened:\"" << AsmString <<"\"\n";
687
688   for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i) {
689     const AsmOperand &Op = AsmOperands[i];
690     errs() << "  op[" << i << "] = " << Op.Class->ClassName << " - ";
691     errs() << '\"' << Op.Token << "\"\n";
692   }
693 }
694
695 static std::pair<StringRef, StringRef>
696 parseTwoOperandConstraint(StringRef S, ArrayRef<SMLoc> Loc) {
697   // Split via the '='.
698   std::pair<StringRef, StringRef> Ops = S.split('=');
699   if (Ops.second == "")
700     PrintFatalError(Loc, "missing '=' in two-operand alias constraint");
701   // Trim whitespace and the leading '$' on the operand names.
702   size_t start = Ops.first.find_first_of('$');
703   if (start == std::string::npos)
704     PrintFatalError(Loc, "expected '$' prefix on asm operand name");
705   Ops.first = Ops.first.slice(start + 1, std::string::npos);
706   size_t end = Ops.first.find_last_of(" \t");
707   Ops.first = Ops.first.slice(0, end);
708   // Now the second operand.
709   start = Ops.second.find_first_of('$');
710   if (start == std::string::npos)
711     PrintFatalError(Loc, "expected '$' prefix on asm operand name");
712   Ops.second = Ops.second.slice(start + 1, std::string::npos);
713   end = Ops.second.find_last_of(" \t");
714   Ops.first = Ops.first.slice(0, end);
715   return Ops;
716 }
717
718 void MatchableInfo::formTwoOperandAlias(StringRef Constraint) {
719   // Figure out which operands are aliased and mark them as tied.
720   std::pair<StringRef, StringRef> Ops =
721     parseTwoOperandConstraint(Constraint, TheDef->getLoc());
722
723   // Find the AsmOperands that refer to the operands we're aliasing.
724   int SrcAsmOperand = findAsmOperandNamed(Ops.first);
725   int DstAsmOperand = findAsmOperandNamed(Ops.second);
726   if (SrcAsmOperand == -1)
727     PrintFatalError(TheDef->getLoc(),
728                     "unknown source two-operand alias operand '" + Ops.first +
729                     "'.");
730   if (DstAsmOperand == -1)
731     PrintFatalError(TheDef->getLoc(),
732                     "unknown destination two-operand alias operand '" +
733                     Ops.second + "'.");
734
735   // Find the ResOperand that refers to the operand we're aliasing away
736   // and update it to refer to the combined operand instead.
737   for (unsigned i = 0, e = ResOperands.size(); i != e; ++i) {
738     ResOperand &Op = ResOperands[i];
739     if (Op.Kind == ResOperand::RenderAsmOperand &&
740         Op.AsmOperandNum == (unsigned)SrcAsmOperand) {
741       Op.AsmOperandNum = DstAsmOperand;
742       break;
743     }
744   }
745   // Remove the AsmOperand for the alias operand.
746   AsmOperands.erase(AsmOperands.begin() + SrcAsmOperand);
747   // Adjust the ResOperand references to any AsmOperands that followed
748   // the one we just deleted.
749   for (unsigned i = 0, e = ResOperands.size(); i != e; ++i) {
750     ResOperand &Op = ResOperands[i];
751     switch(Op.Kind) {
752     default:
753       // Nothing to do for operands that don't reference AsmOperands.
754       break;
755     case ResOperand::RenderAsmOperand:
756       if (Op.AsmOperandNum > (unsigned)SrcAsmOperand)
757         --Op.AsmOperandNum;
758       break;
759     case ResOperand::TiedOperand:
760       if (Op.TiedOperandNum > (unsigned)SrcAsmOperand)
761         --Op.TiedOperandNum;
762       break;
763     }
764   }
765 }
766
767 void MatchableInfo::initialize(const AsmMatcherInfo &Info,
768                                SmallPtrSetImpl<Record*> &SingletonRegisters,
769                                int AsmVariantNo, std::string &RegisterPrefix) {
770   AsmVariantID = AsmVariantNo;
771   AsmString =
772     CodeGenInstruction::FlattenAsmStringVariants(AsmString, AsmVariantNo);
773
774   tokenizeAsmString(Info);
775
776   // Compute the require features.
777   std::vector<Record*> Predicates =TheDef->getValueAsListOfDefs("Predicates");
778   for (unsigned i = 0, e = Predicates.size(); i != e; ++i)
779     if (SubtargetFeatureInfo *Feature =
780         Info.getSubtargetFeature(Predicates[i]))
781       RequiredFeatures.push_back(Feature);
782
783   // Collect singleton registers, if used.
784   for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i) {
785     extractSingletonRegisterForAsmOperand(i, Info, RegisterPrefix);
786     if (Record *Reg = AsmOperands[i].SingletonReg)
787       SingletonRegisters.insert(Reg);
788   }
789
790   const RecordVal *DepMask = TheDef->getValue("DeprecatedFeatureMask");
791   if (!DepMask)
792     DepMask = TheDef->getValue("ComplexDeprecationPredicate");
793
794   HasDeprecation =
795       DepMask ? !DepMask->getValue()->getAsUnquotedString().empty() : false;
796 }
797
798 /// tokenizeAsmString - Tokenize a simplified assembly string.
799 void MatchableInfo::tokenizeAsmString(const AsmMatcherInfo &Info) {
800   StringRef String = AsmString;
801   unsigned Prev = 0;
802   bool InTok = true;
803   for (unsigned i = 0, e = String.size(); i != e; ++i) {
804     switch (String[i]) {
805     case '[':
806     case ']':
807     case '*':
808     case '!':
809     case ' ':
810     case '\t':
811     case ',':
812       if (InTok) {
813         AsmOperands.push_back(AsmOperand(String.slice(Prev, i)));
814         InTok = false;
815       }
816       if (!isspace(String[i]) && String[i] != ',')
817         AsmOperands.push_back(AsmOperand(String.substr(i, 1)));
818       Prev = i + 1;
819       break;
820
821     case '\\':
822       if (InTok) {
823         AsmOperands.push_back(AsmOperand(String.slice(Prev, i)));
824         InTok = false;
825       }
826       ++i;
827       assert(i != String.size() && "Invalid quoted character");
828       AsmOperands.push_back(AsmOperand(String.substr(i, 1)));
829       Prev = i + 1;
830       break;
831
832     case '$': {
833       if (InTok) {
834         AsmOperands.push_back(AsmOperand(String.slice(Prev, i)));
835         InTok = false;
836       }
837
838       // If this isn't "${", treat like a normal token.
839       if (i + 1 == String.size() || String[i + 1] != '{') {
840         Prev = i;
841         break;
842       }
843
844       StringRef::iterator End = std::find(String.begin() + i, String.end(),'}');
845       assert(End != String.end() && "Missing brace in operand reference!");
846       size_t EndPos = End - String.begin();
847       AsmOperands.push_back(AsmOperand(String.slice(i, EndPos+1)));
848       Prev = EndPos + 1;
849       i = EndPos;
850       break;
851     }
852
853     case '.':
854       if (!Info.AsmParser->getValueAsBit("MnemonicContainsDot")) {
855         if (InTok)
856           AsmOperands.push_back(AsmOperand(String.slice(Prev, i)));
857         Prev = i;
858       }
859       InTok = true;
860       break;
861
862     default:
863       InTok = true;
864     }
865   }
866   if (InTok && Prev != String.size())
867     AsmOperands.push_back(AsmOperand(String.substr(Prev)));
868
869   // The first token of the instruction is the mnemonic, which must be a
870   // simple string, not a $foo variable or a singleton register.
871   if (AsmOperands.empty())
872     PrintFatalError(TheDef->getLoc(),
873                   "Instruction '" + TheDef->getName() + "' has no tokens");
874   Mnemonic = AsmOperands[0].Token;
875   if (Mnemonic.empty())
876     PrintFatalError(TheDef->getLoc(),
877                   "Missing instruction mnemonic");
878   // FIXME : Check and raise an error if it is a register.
879   if (Mnemonic[0] == '$')
880     PrintFatalError(TheDef->getLoc(),
881                     "Invalid instruction mnemonic '" + Mnemonic + "'!");
882
883   // Remove the first operand, it is tracked in the mnemonic field.
884   AsmOperands.erase(AsmOperands.begin());
885 }
886
887 bool MatchableInfo::validate(StringRef CommentDelimiter, bool Hack) const {
888   // Reject matchables with no .s string.
889   if (AsmString.empty())
890     PrintFatalError(TheDef->getLoc(), "instruction with empty asm string");
891
892   // Reject any matchables with a newline in them, they should be marked
893   // isCodeGenOnly if they are pseudo instructions.
894   if (AsmString.find('\n') != std::string::npos)
895     PrintFatalError(TheDef->getLoc(),
896                   "multiline instruction is not valid for the asmparser, "
897                   "mark it isCodeGenOnly");
898
899   // Remove comments from the asm string.  We know that the asmstring only
900   // has one line.
901   if (!CommentDelimiter.empty() &&
902       StringRef(AsmString).find(CommentDelimiter) != StringRef::npos)
903     PrintFatalError(TheDef->getLoc(),
904                   "asmstring for instruction has comment character in it, "
905                   "mark it isCodeGenOnly");
906
907   // Reject matchables with operand modifiers, these aren't something we can
908   // handle, the target should be refactored to use operands instead of
909   // modifiers.
910   //
911   // Also, check for instructions which reference the operand multiple times;
912   // this implies a constraint we would not honor.
913   std::set<std::string> OperandNames;
914   for (unsigned i = 0, e = AsmOperands.size(); i != e; ++i) {
915     StringRef Tok = AsmOperands[i].Token;
916     if (Tok[0] == '$' && Tok.find(':') != StringRef::npos)
917       PrintFatalError(TheDef->getLoc(),
918                       "matchable with operand modifier '" + Tok +
919                       "' not supported by asm matcher.  Mark isCodeGenOnly!");
920
921     // Verify that any operand is only mentioned once.
922     // We reject aliases and ignore instructions for now.
923     if (Tok[0] == '$' && !OperandNames.insert(Tok).second) {
924       if (!Hack)
925         PrintFatalError(TheDef->getLoc(),
926                         "ERROR: matchable with tied operand '" + Tok +
927                         "' can never be matched!");
928       // FIXME: Should reject these.  The ARM backend hits this with $lane in a
929       // bunch of instructions.  It is unclear what the right answer is.
930       DEBUG({
931         errs() << "warning: '" << TheDef->getName() << "': "
932                << "ignoring instruction with tied operand '"
933                << Tok << "'\n";
934       });
935       return false;
936     }
937   }
938
939   return true;
940 }
941
942 /// extractSingletonRegisterForAsmOperand - Extract singleton register,
943 /// if present, from specified token.
944 void MatchableInfo::
945 extractSingletonRegisterForAsmOperand(unsigned OperandNo,
946                                       const AsmMatcherInfo &Info,
947                                       std::string &RegisterPrefix) {
948   StringRef Tok = AsmOperands[OperandNo].Token;
949   if (RegisterPrefix.empty()) {
950     std::string LoweredTok = Tok.lower();
951     if (const CodeGenRegister *Reg = Info.Target.getRegisterByName(LoweredTok))
952       AsmOperands[OperandNo].SingletonReg = Reg->TheDef;
953     return;
954   }
955
956   if (!Tok.startswith(RegisterPrefix))
957     return;
958
959   StringRef RegName = Tok.substr(RegisterPrefix.size());
960   if (const CodeGenRegister *Reg = Info.Target.getRegisterByName(RegName))
961     AsmOperands[OperandNo].SingletonReg = Reg->TheDef;
962
963   // If there is no register prefix (i.e. "%" in "%eax"), then this may
964   // be some random non-register token, just ignore it.
965   return;
966 }
967
968 static std::string getEnumNameForToken(StringRef Str) {
969   std::string Res;
970
971   for (StringRef::iterator it = Str.begin(), ie = Str.end(); it != ie; ++it) {
972     switch (*it) {
973     case '*': Res += "_STAR_"; break;
974     case '%': Res += "_PCT_"; break;
975     case ':': Res += "_COLON_"; break;
976     case '!': Res += "_EXCLAIM_"; break;
977     case '.': Res += "_DOT_"; break;
978     case '<': Res += "_LT_"; break;
979     case '>': Res += "_GT_"; break;
980     default:
981       if ((*it >= 'A' && *it <= 'Z') ||
982           (*it >= 'a' && *it <= 'z') ||
983           (*it >= '0' && *it <= '9'))
984         Res += *it;
985       else
986         Res += "_" + utostr((unsigned) *it) + "_";
987     }
988   }
989
990   return Res;
991 }
992
993 ClassInfo *AsmMatcherInfo::getTokenClass(StringRef Token) {
994   ClassInfo *&Entry = TokenClasses[Token];
995
996   if (!Entry) {
997     Entry = new ClassInfo();
998     Entry->Kind = ClassInfo::Token;
999     Entry->ClassName = "Token";
1000     Entry->Name = "MCK_" + getEnumNameForToken(Token);
1001     Entry->ValueName = Token;
1002     Entry->PredicateMethod = "<invalid>";
1003     Entry->RenderMethod = "<invalid>";
1004     Entry->ParserMethod = "";
1005     Entry->DiagnosticType = "";
1006     Classes.push_back(std::unique_ptr<ClassInfo>(Entry));
1007   }
1008
1009   return Entry;
1010 }
1011
1012 ClassInfo *
1013 AsmMatcherInfo::getOperandClass(const CGIOperandList::OperandInfo &OI,
1014                                 int SubOpIdx) {
1015   Record *Rec = OI.Rec;
1016   if (SubOpIdx != -1)
1017     Rec = cast<DefInit>(OI.MIOperandInfo->getArg(SubOpIdx))->getDef();
1018   return getOperandClass(Rec, SubOpIdx);
1019 }
1020
1021 ClassInfo *
1022 AsmMatcherInfo::getOperandClass(Record *Rec, int SubOpIdx) {
1023   if (Rec->isSubClassOf("RegisterOperand")) {
1024     // RegisterOperand may have an associated ParserMatchClass. If it does,
1025     // use it, else just fall back to the underlying register class.
1026     const RecordVal *R = Rec->getValue("ParserMatchClass");
1027     if (!R || !R->getValue())
1028       PrintFatalError("Record `" + Rec->getName() +
1029         "' does not have a ParserMatchClass!\n");
1030
1031     if (DefInit *DI= dyn_cast<DefInit>(R->getValue())) {
1032       Record *MatchClass = DI->getDef();
1033       if (ClassInfo *CI = AsmOperandClasses[MatchClass])
1034         return CI;
1035     }
1036
1037     // No custom match class. Just use the register class.
1038     Record *ClassRec = Rec->getValueAsDef("RegClass");
1039     if (!ClassRec)
1040       PrintFatalError(Rec->getLoc(), "RegisterOperand `" + Rec->getName() +
1041                     "' has no associated register class!\n");
1042     if (ClassInfo *CI = RegisterClassClasses[ClassRec])
1043       return CI;
1044     PrintFatalError(Rec->getLoc(), "register class has no class info!");
1045   }
1046
1047
1048   if (Rec->isSubClassOf("RegisterClass")) {
1049     if (ClassInfo *CI = RegisterClassClasses[Rec])
1050       return CI;
1051     PrintFatalError(Rec->getLoc(), "register class has no class info!");
1052   }
1053
1054   if (!Rec->isSubClassOf("Operand"))
1055     PrintFatalError(Rec->getLoc(), "Operand `" + Rec->getName() +
1056                   "' does not derive from class Operand!\n");
1057   Record *MatchClass = Rec->getValueAsDef("ParserMatchClass");
1058   if (ClassInfo *CI = AsmOperandClasses[MatchClass])
1059     return CI;
1060
1061   PrintFatalError(Rec->getLoc(), "operand has no match class!");
1062 }
1063
1064 struct LessRegisterSet {
1065   bool operator() (const RegisterSet &LHS, const RegisterSet & RHS) const {
1066     // std::set<T> defines its own compariso "operator<", but it
1067     // performs a lexicographical comparison by T's innate comparison
1068     // for some reason. We don't want non-deterministic pointer
1069     // comparisons so use this instead.
1070     return std::lexicographical_compare(LHS.begin(), LHS.end(),
1071                                         RHS.begin(), RHS.end(),
1072                                         LessRecordByID());
1073   }
1074 };
1075
1076 void AsmMatcherInfo::
1077 buildRegisterClasses(SmallPtrSetImpl<Record*> &SingletonRegisters) {
1078   const std::vector<CodeGenRegister*> &Registers =
1079     Target.getRegBank().getRegisters();
1080   ArrayRef<CodeGenRegisterClass*> RegClassList =
1081     Target.getRegBank().getRegClasses();
1082
1083   typedef std::set<RegisterSet, LessRegisterSet> RegisterSetSet;
1084
1085   // The register sets used for matching.
1086   RegisterSetSet RegisterSets;
1087
1088   // Gather the defined sets.
1089   for (const CodeGenRegisterClass *RC : RegClassList)
1090     RegisterSets.insert(RegisterSet(RC->getOrder().begin(),
1091                                     RC->getOrder().end()));
1092
1093   // Add any required singleton sets.
1094   for (Record *Rec : SingletonRegisters) {
1095     RegisterSets.insert(RegisterSet(&Rec, &Rec + 1));
1096   }
1097
1098   // Introduce derived sets where necessary (when a register does not determine
1099   // a unique register set class), and build the mapping of registers to the set
1100   // they should classify to.
1101   std::map<Record*, RegisterSet> RegisterMap;
1102   for (const CodeGenRegister *CGR : Registers) {
1103     // Compute the intersection of all sets containing this register.
1104     RegisterSet ContainingSet;
1105
1106     for (const RegisterSet &RS : RegisterSets) {
1107       if (!RS.count(CGR->TheDef))
1108         continue;
1109
1110       if (ContainingSet.empty()) {
1111         ContainingSet = RS;
1112         continue;
1113       }
1114
1115       RegisterSet Tmp;
1116       std::swap(Tmp, ContainingSet);
1117       std::insert_iterator<RegisterSet> II(ContainingSet,
1118                                            ContainingSet.begin());
1119       std::set_intersection(Tmp.begin(), Tmp.end(), RS.begin(), RS.end(), II,
1120                             LessRecordByID());
1121     }
1122
1123     if (!ContainingSet.empty()) {
1124       RegisterSets.insert(ContainingSet);
1125       RegisterMap.insert(std::make_pair(CGR->TheDef, ContainingSet));
1126     }
1127   }
1128
1129   // Construct the register classes.
1130   std::map<RegisterSet, ClassInfo*, LessRegisterSet> RegisterSetClasses;
1131   unsigned Index = 0;
1132   for (const RegisterSet &RS : RegisterSets) {
1133     ClassInfo *CI = new ClassInfo();
1134     CI->Kind = ClassInfo::RegisterClass0 + Index;
1135     CI->ClassName = "Reg" + utostr(Index);
1136     CI->Name = "MCK_Reg" + utostr(Index);
1137     CI->ValueName = "";
1138     CI->PredicateMethod = ""; // unused
1139     CI->RenderMethod = "addRegOperands";
1140     CI->Registers = RS;
1141     // FIXME: diagnostic type.
1142     CI->DiagnosticType = "";
1143     Classes.push_back(std::unique_ptr<ClassInfo>(CI));
1144     RegisterSetClasses.insert(std::make_pair(RS, CI));
1145     ++Index;
1146   }
1147
1148   // Find the superclasses; we could compute only the subgroup lattice edges,
1149   // but there isn't really a point.
1150   for (const RegisterSet &RS : RegisterSets) {
1151     ClassInfo *CI = RegisterSetClasses[RS];
1152     for (const RegisterSet &RS2 : RegisterSets)
1153       if (RS != RS2 &&
1154           std::includes(RS2.begin(), RS2.end(), RS.begin(), RS.end(),
1155                         LessRecordByID()))
1156         CI->SuperClasses.push_back(RegisterSetClasses[RS2]);
1157   }
1158
1159   // Name the register classes which correspond to a user defined RegisterClass.
1160   for (const CodeGenRegisterClass *RC : RegClassList) {
1161     // Def will be NULL for non-user defined register classes.
1162     Record *Def = RC->getDef();
1163     if (!Def)
1164       continue;
1165     ClassInfo *CI = RegisterSetClasses[RegisterSet(RC->getOrder().begin(),
1166                                                    RC->getOrder().end())];
1167     if (CI->ValueName.empty()) {
1168       CI->ClassName = RC->getName();
1169       CI->Name = "MCK_" + RC->getName();
1170       CI->ValueName = RC->getName();
1171     } else
1172       CI->ValueName = CI->ValueName + "," + RC->getName();
1173
1174     RegisterClassClasses.insert(std::make_pair(Def, CI));
1175   }
1176
1177   // Populate the map for individual registers.
1178   for (std::map<Record*, RegisterSet>::iterator it = RegisterMap.begin(),
1179          ie = RegisterMap.end(); it != ie; ++it)
1180     RegisterClasses[it->first] = RegisterSetClasses[it->second];
1181
1182   // Name the register classes which correspond to singleton registers.
1183   for (Record *Rec : SingletonRegisters) {
1184     ClassInfo *CI = RegisterClasses[Rec];
1185     assert(CI && "Missing singleton register class info!");
1186
1187     if (CI->ValueName.empty()) {
1188       CI->ClassName = Rec->getName();
1189       CI->Name = "MCK_" + Rec->getName();
1190       CI->ValueName = Rec->getName();
1191     } else
1192       CI->ValueName = CI->ValueName + "," + Rec->getName();
1193   }
1194 }
1195
1196 void AsmMatcherInfo::buildOperandClasses() {
1197   std::vector<Record*> AsmOperands =
1198     Records.getAllDerivedDefinitions("AsmOperandClass");
1199
1200   // Pre-populate AsmOperandClasses map.
1201   for (Record *Rec : AsmOperands)
1202     AsmOperandClasses[Rec] = new ClassInfo();
1203
1204   unsigned Index = 0;
1205   for (Record *Rec : AsmOperands) {
1206     ClassInfo *CI = AsmOperandClasses[Rec];
1207     CI->Kind = ClassInfo::UserClass0 + Index;
1208
1209     ListInit *Supers = Rec->getValueAsListInit("SuperClasses");
1210     for (unsigned i = 0, e = Supers->getSize(); i != e; ++i) {
1211       DefInit *DI = dyn_cast<DefInit>(Supers->getElement(i));
1212       if (!DI) {
1213         PrintError(Rec->getLoc(), "Invalid super class reference!");
1214         continue;
1215       }
1216
1217       ClassInfo *SC = AsmOperandClasses[DI->getDef()];
1218       if (!SC)
1219         PrintError(Rec->getLoc(), "Invalid super class reference!");
1220       else
1221         CI->SuperClasses.push_back(SC);
1222     }
1223     CI->ClassName = Rec->getValueAsString("Name");
1224     CI->Name = "MCK_" + CI->ClassName;
1225     CI->ValueName = Rec->getName();
1226
1227     // Get or construct the predicate method name.
1228     Init *PMName = Rec->getValueInit("PredicateMethod");
1229     if (StringInit *SI = dyn_cast<StringInit>(PMName)) {
1230       CI->PredicateMethod = SI->getValue();
1231     } else {
1232       assert(isa<UnsetInit>(PMName) && "Unexpected PredicateMethod field!");
1233       CI->PredicateMethod = "is" + CI->ClassName;
1234     }
1235
1236     // Get or construct the render method name.
1237     Init *RMName = Rec->getValueInit("RenderMethod");
1238     if (StringInit *SI = dyn_cast<StringInit>(RMName)) {
1239       CI->RenderMethod = SI->getValue();
1240     } else {
1241       assert(isa<UnsetInit>(RMName) && "Unexpected RenderMethod field!");
1242       CI->RenderMethod = "add" + CI->ClassName + "Operands";
1243     }
1244
1245     // Get the parse method name or leave it as empty.
1246     Init *PRMName = Rec->getValueInit("ParserMethod");
1247     if (StringInit *SI = dyn_cast<StringInit>(PRMName))
1248       CI->ParserMethod = SI->getValue();
1249
1250     // Get the diagnostic type or leave it as empty.
1251     // Get the parse method name or leave it as empty.
1252     Init *DiagnosticType = Rec->getValueInit("DiagnosticType");
1253     if (StringInit *SI = dyn_cast<StringInit>(DiagnosticType))
1254       CI->DiagnosticType = SI->getValue();
1255
1256     Classes.push_back(std::unique_ptr<ClassInfo>(CI));
1257     ++Index;
1258   }
1259 }
1260
1261 AsmMatcherInfo::AsmMatcherInfo(Record *asmParser,
1262                                CodeGenTarget &target,
1263                                RecordKeeper &records)
1264   : Records(records), AsmParser(asmParser), Target(target) {
1265 }
1266
1267 /// buildOperandMatchInfo - Build the necessary information to handle user
1268 /// defined operand parsing methods.
1269 void AsmMatcherInfo::buildOperandMatchInfo() {
1270
1271   /// Map containing a mask with all operands indices that can be found for
1272   /// that class inside a instruction.
1273   typedef std::map<ClassInfo *, unsigned, less_ptr<ClassInfo>> OpClassMaskTy;
1274   OpClassMaskTy OpClassMask;
1275
1276   for (const auto &MI : Matchables) {
1277     OpClassMask.clear();
1278
1279     // Keep track of all operands of this instructions which belong to the
1280     // same class.
1281     for (unsigned i = 0, e = MI->AsmOperands.size(); i != e; ++i) {
1282       const MatchableInfo::AsmOperand &Op = MI->AsmOperands[i];
1283       if (Op.Class->ParserMethod.empty())
1284         continue;
1285       unsigned &OperandMask = OpClassMask[Op.Class];
1286       OperandMask |= (1 << i);
1287     }
1288
1289     // Generate operand match info for each mnemonic/operand class pair.
1290     for (const auto &OCM : OpClassMask) {
1291       unsigned OpMask = OCM.second;
1292       ClassInfo *CI = OCM.first;
1293       OperandMatchInfo.push_back(OperandMatchEntry::create(MI.get(), CI,
1294                                                            OpMask));
1295     }
1296   }
1297 }
1298
1299 void AsmMatcherInfo::buildInfo() {
1300   // Build information about all of the AssemblerPredicates.
1301   std::vector<Record*> AllPredicates =
1302     Records.getAllDerivedDefinitions("Predicate");
1303   for (unsigned i = 0, e = AllPredicates.size(); i != e; ++i) {
1304     Record *Pred = AllPredicates[i];
1305     // Ignore predicates that are not intended for the assembler.
1306     if (!Pred->getValueAsBit("AssemblerMatcherPredicate"))
1307       continue;
1308
1309     if (Pred->getName().empty())
1310       PrintFatalError(Pred->getLoc(), "Predicate has no name!");
1311
1312     uint64_t FeatureNo = SubtargetFeatures.size();
1313     SubtargetFeatures[Pred] =
1314       llvm::make_unique<SubtargetFeatureInfo>(Pred, FeatureNo);
1315     DEBUG(SubtargetFeatures[Pred]->dump());
1316     assert(FeatureNo < 64 && "Too many subtarget features!");
1317   }
1318
1319   // Parse the instructions; we need to do this first so that we can gather the
1320   // singleton register classes.
1321   SmallPtrSet<Record*, 16> SingletonRegisters;
1322   unsigned VariantCount = Target.getAsmParserVariantCount();
1323   for (unsigned VC = 0; VC != VariantCount; ++VC) {
1324     Record *AsmVariant = Target.getAsmParserVariant(VC);
1325     std::string CommentDelimiter =
1326       AsmVariant->getValueAsString("CommentDelimiter");
1327     std::string RegisterPrefix = AsmVariant->getValueAsString("RegisterPrefix");
1328     int AsmVariantNo = AsmVariant->getValueAsInt("Variant");
1329
1330     for (const CodeGenInstruction *CGI : Target.instructions()) {
1331
1332       // If the tblgen -match-prefix option is specified (for tblgen hackers),
1333       // filter the set of instructions we consider.
1334       if (!StringRef(CGI->TheDef->getName()).startswith(MatchPrefix))
1335         continue;
1336
1337       // Ignore "codegen only" instructions.
1338       if (CGI->TheDef->getValueAsBit("isCodeGenOnly"))
1339         continue;
1340
1341       std::unique_ptr<MatchableInfo> II(new MatchableInfo(*CGI));
1342
1343       II->initialize(*this, SingletonRegisters, AsmVariantNo, RegisterPrefix);
1344
1345       // Ignore instructions which shouldn't be matched and diagnose invalid
1346       // instruction definitions with an error.
1347       if (!II->validate(CommentDelimiter, true))
1348         continue;
1349
1350       Matchables.push_back(std::move(II));
1351     }
1352
1353     // Parse all of the InstAlias definitions and stick them in the list of
1354     // matchables.
1355     std::vector<Record*> AllInstAliases =
1356       Records.getAllDerivedDefinitions("InstAlias");
1357     for (unsigned i = 0, e = AllInstAliases.size(); i != e; ++i) {
1358       CodeGenInstAlias *Alias =
1359           new CodeGenInstAlias(AllInstAliases[i], AsmVariantNo, Target);
1360
1361       // If the tblgen -match-prefix option is specified (for tblgen hackers),
1362       // filter the set of instruction aliases we consider, based on the target
1363       // instruction.
1364       if (!StringRef(Alias->ResultInst->TheDef->getName())
1365             .startswith( MatchPrefix))
1366         continue;
1367
1368       std::unique_ptr<MatchableInfo> II(new MatchableInfo(Alias));
1369
1370       II->initialize(*this, SingletonRegisters, AsmVariantNo, RegisterPrefix);
1371
1372       // Validate the alias definitions.
1373       II->validate(CommentDelimiter, false);
1374
1375       Matchables.push_back(std::move(II));
1376     }
1377   }
1378
1379   // Build info for the register classes.
1380   buildRegisterClasses(SingletonRegisters);
1381
1382   // Build info for the user defined assembly operand classes.
1383   buildOperandClasses();
1384
1385   // Build the information about matchables, now that we have fully formed
1386   // classes.
1387   std::vector<std::unique_ptr<MatchableInfo>> NewMatchables;
1388   for (auto &II : Matchables) {
1389     // Parse the tokens after the mnemonic.
1390     // Note: buildInstructionOperandReference may insert new AsmOperands, so
1391     // don't precompute the loop bound.
1392     for (unsigned i = 0; i != II->AsmOperands.size(); ++i) {
1393       MatchableInfo::AsmOperand &Op = II->AsmOperands[i];
1394       StringRef Token = Op.Token;
1395
1396       // Check for singleton registers.
1397       if (Record *RegRecord = II->AsmOperands[i].SingletonReg) {
1398         Op.Class = RegisterClasses[RegRecord];
1399         assert(Op.Class && Op.Class->Registers.size() == 1 &&
1400                "Unexpected class for singleton register");
1401         continue;
1402       }
1403
1404       // Check for simple tokens.
1405       if (Token[0] != '$') {
1406         Op.Class = getTokenClass(Token);
1407         continue;
1408       }
1409
1410       if (Token.size() > 1 && isdigit(Token[1])) {
1411         Op.Class = getTokenClass(Token);
1412         continue;
1413       }
1414
1415       // Otherwise this is an operand reference.
1416       StringRef OperandName;
1417       if (Token[1] == '{')
1418         OperandName = Token.substr(2, Token.size() - 3);
1419       else
1420         OperandName = Token.substr(1);
1421
1422       if (II->DefRec.is<const CodeGenInstruction*>())
1423         buildInstructionOperandReference(II.get(), OperandName, i);
1424       else
1425         buildAliasOperandReference(II.get(), OperandName, Op);
1426     }
1427
1428     if (II->DefRec.is<const CodeGenInstruction*>()) {
1429       II->buildInstructionResultOperands();
1430       // If the instruction has a two-operand alias, build up the
1431       // matchable here. We'll add them in bulk at the end to avoid
1432       // confusing this loop.
1433       std::string Constraint =
1434         II->TheDef->getValueAsString("TwoOperandAliasConstraint");
1435       if (Constraint != "") {
1436         // Start by making a copy of the original matchable.
1437         std::unique_ptr<MatchableInfo> AliasII(new MatchableInfo(*II));
1438
1439         // Adjust it to be a two-operand alias.
1440         AliasII->formTwoOperandAlias(Constraint);
1441
1442         // Add the alias to the matchables list.
1443         NewMatchables.push_back(std::move(AliasII));
1444       }
1445     } else
1446       II->buildAliasResultOperands();
1447   }
1448   if (!NewMatchables.empty())
1449     std::move(NewMatchables.begin(), NewMatchables.end(),
1450               std::back_inserter(Matchables));
1451
1452   // Process token alias definitions and set up the associated superclass
1453   // information.
1454   std::vector<Record*> AllTokenAliases =
1455     Records.getAllDerivedDefinitions("TokenAlias");
1456   for (unsigned i = 0, e = AllTokenAliases.size(); i != e; ++i) {
1457     Record *Rec = AllTokenAliases[i];
1458     ClassInfo *FromClass = getTokenClass(Rec->getValueAsString("FromToken"));
1459     ClassInfo *ToClass = getTokenClass(Rec->getValueAsString("ToToken"));
1460     if (FromClass == ToClass)
1461       PrintFatalError(Rec->getLoc(),
1462                     "error: Destination value identical to source value.");
1463     FromClass->SuperClasses.push_back(ToClass);
1464   }
1465
1466   // Reorder classes so that classes precede super classes.
1467   std::sort(Classes.begin(), Classes.end(),
1468             [](const std::unique_ptr<ClassInfo> &a,
1469                const std::unique_ptr<ClassInfo> &b){
1470               return *a < *b;});
1471 }
1472
1473 /// buildInstructionOperandReference - The specified operand is a reference to a
1474 /// named operand such as $src.  Resolve the Class and OperandInfo pointers.
1475 void AsmMatcherInfo::
1476 buildInstructionOperandReference(MatchableInfo *II,
1477                                  StringRef OperandName,
1478                                  unsigned AsmOpIdx) {
1479   const CodeGenInstruction &CGI = *II->DefRec.get<const CodeGenInstruction*>();
1480   const CGIOperandList &Operands = CGI.Operands;
1481   MatchableInfo::AsmOperand *Op = &II->AsmOperands[AsmOpIdx];
1482
1483   // Map this token to an operand.
1484   unsigned Idx;
1485   if (!Operands.hasOperandNamed(OperandName, Idx))
1486     PrintFatalError(II->TheDef->getLoc(),
1487                     "error: unable to find operand: '" + OperandName + "'");
1488
1489   // If the instruction operand has multiple suboperands, but the parser
1490   // match class for the asm operand is still the default "ImmAsmOperand",
1491   // then handle each suboperand separately.
1492   if (Op->SubOpIdx == -1 && Operands[Idx].MINumOperands > 1) {
1493     Record *Rec = Operands[Idx].Rec;
1494     assert(Rec->isSubClassOf("Operand") && "Unexpected operand!");
1495     Record *MatchClass = Rec->getValueAsDef("ParserMatchClass");
1496     if (MatchClass && MatchClass->getValueAsString("Name") == "Imm") {
1497       // Insert remaining suboperands after AsmOpIdx in II->AsmOperands.
1498       StringRef Token = Op->Token; // save this in case Op gets moved
1499       for (unsigned SI = 1, SE = Operands[Idx].MINumOperands; SI != SE; ++SI) {
1500         MatchableInfo::AsmOperand NewAsmOp(Token);
1501         NewAsmOp.SubOpIdx = SI;
1502         II->AsmOperands.insert(II->AsmOperands.begin()+AsmOpIdx+SI, NewAsmOp);
1503       }
1504       // Replace Op with first suboperand.
1505       Op = &II->AsmOperands[AsmOpIdx]; // update the pointer in case it moved
1506       Op->SubOpIdx = 0;
1507     }
1508   }
1509
1510   // Set up the operand class.
1511   Op->Class = getOperandClass(Operands[Idx], Op->SubOpIdx);
1512
1513   // If the named operand is tied, canonicalize it to the untied operand.
1514   // For example, something like:
1515   //   (outs GPR:$dst), (ins GPR:$src)
1516   // with an asmstring of
1517   //   "inc $src"
1518   // we want to canonicalize to:
1519   //   "inc $dst"
1520   // so that we know how to provide the $dst operand when filling in the result.
1521   int OITied = -1;
1522   if (Operands[Idx].MINumOperands == 1)
1523     OITied = Operands[Idx].getTiedRegister();
1524   if (OITied != -1) {
1525     // The tied operand index is an MIOperand index, find the operand that
1526     // contains it.
1527     std::pair<unsigned, unsigned> Idx = Operands.getSubOperandNumber(OITied);
1528     OperandName = Operands[Idx.first].Name;
1529     Op->SubOpIdx = Idx.second;
1530   }
1531
1532   Op->SrcOpName = OperandName;
1533 }
1534
1535 /// buildAliasOperandReference - When parsing an operand reference out of the
1536 /// matching string (e.g. "movsx $src, $dst"), determine what the class of the
1537 /// operand reference is by looking it up in the result pattern definition.
1538 void AsmMatcherInfo::buildAliasOperandReference(MatchableInfo *II,
1539                                                 StringRef OperandName,
1540                                                 MatchableInfo::AsmOperand &Op) {
1541   const CodeGenInstAlias &CGA = *II->DefRec.get<const CodeGenInstAlias*>();
1542
1543   // Set up the operand class.
1544   for (unsigned i = 0, e = CGA.ResultOperands.size(); i != e; ++i)
1545     if (CGA.ResultOperands[i].isRecord() &&
1546         CGA.ResultOperands[i].getName() == OperandName) {
1547       // It's safe to go with the first one we find, because CodeGenInstAlias
1548       // validates that all operands with the same name have the same record.
1549       Op.SubOpIdx = CGA.ResultInstOperandIndex[i].second;
1550       // Use the match class from the Alias definition, not the
1551       // destination instruction, as we may have an immediate that's
1552       // being munged by the match class.
1553       Op.Class = getOperandClass(CGA.ResultOperands[i].getRecord(),
1554                                  Op.SubOpIdx);
1555       Op.SrcOpName = OperandName;
1556       return;
1557     }
1558
1559   PrintFatalError(II->TheDef->getLoc(),
1560                   "error: unable to find operand: '" + OperandName + "'");
1561 }
1562
1563 void MatchableInfo::buildInstructionResultOperands() {
1564   const CodeGenInstruction *ResultInst = getResultInst();
1565
1566   // Loop over all operands of the result instruction, determining how to
1567   // populate them.
1568   for (unsigned i = 0, e = ResultInst->Operands.size(); i != e; ++i) {
1569     const CGIOperandList::OperandInfo &OpInfo = ResultInst->Operands[i];
1570
1571     // If this is a tied operand, just copy from the previously handled operand.
1572     int TiedOp = -1;
1573     if (OpInfo.MINumOperands == 1)
1574       TiedOp = OpInfo.getTiedRegister();
1575     if (TiedOp != -1) {
1576       ResOperands.push_back(ResOperand::getTiedOp(TiedOp));
1577       continue;
1578     }
1579
1580     // Find out what operand from the asmparser this MCInst operand comes from.
1581     int SrcOperand = findAsmOperandNamed(OpInfo.Name);
1582     if (OpInfo.Name.empty() || SrcOperand == -1) {
1583       // This may happen for operands that are tied to a suboperand of a
1584       // complex operand.  Simply use a dummy value here; nobody should
1585       // use this operand slot.
1586       // FIXME: The long term goal is for the MCOperand list to not contain
1587       // tied operands at all.
1588       ResOperands.push_back(ResOperand::getImmOp(0));
1589       continue;
1590     }
1591
1592     // Check if the one AsmOperand populates the entire operand.
1593     unsigned NumOperands = OpInfo.MINumOperands;
1594     if (AsmOperands[SrcOperand].SubOpIdx == -1) {
1595       ResOperands.push_back(ResOperand::getRenderedOp(SrcOperand, NumOperands));
1596       continue;
1597     }
1598
1599     // Add a separate ResOperand for each suboperand.
1600     for (unsigned AI = 0; AI < NumOperands; ++AI) {
1601       assert(AsmOperands[SrcOperand+AI].SubOpIdx == (int)AI &&
1602              AsmOperands[SrcOperand+AI].SrcOpName == OpInfo.Name &&
1603              "unexpected AsmOperands for suboperands");
1604       ResOperands.push_back(ResOperand::getRenderedOp(SrcOperand + AI, 1));
1605     }
1606   }
1607 }
1608
1609 void MatchableInfo::buildAliasResultOperands() {
1610   const CodeGenInstAlias &CGA = *DefRec.get<const CodeGenInstAlias*>();
1611   const CodeGenInstruction *ResultInst = getResultInst();
1612
1613   // Loop over all operands of the result instruction, determining how to
1614   // populate them.
1615   unsigned AliasOpNo = 0;
1616   unsigned LastOpNo = CGA.ResultInstOperandIndex.size();
1617   for (unsigned i = 0, e = ResultInst->Operands.size(); i != e; ++i) {
1618     const CGIOperandList::OperandInfo *OpInfo = &ResultInst->Operands[i];
1619
1620     // If this is a tied operand, just copy from the previously handled operand.
1621     int TiedOp = -1;
1622     if (OpInfo->MINumOperands == 1)
1623       TiedOp = OpInfo->getTiedRegister();
1624     if (TiedOp != -1) {
1625       ResOperands.push_back(ResOperand::getTiedOp(TiedOp));
1626       continue;
1627     }
1628
1629     // Handle all the suboperands for this operand.
1630     const std::string &OpName = OpInfo->Name;
1631     for ( ; AliasOpNo <  LastOpNo &&
1632             CGA.ResultInstOperandIndex[AliasOpNo].first == i; ++AliasOpNo) {
1633       int SubIdx = CGA.ResultInstOperandIndex[AliasOpNo].second;
1634
1635       // Find out what operand from the asmparser that this MCInst operand
1636       // comes from.
1637       switch (CGA.ResultOperands[AliasOpNo].Kind) {
1638       case CodeGenInstAlias::ResultOperand::K_Record: {
1639         StringRef Name = CGA.ResultOperands[AliasOpNo].getName();
1640         int SrcOperand = findAsmOperand(Name, SubIdx);
1641         if (SrcOperand == -1)
1642           PrintFatalError(TheDef->getLoc(), "Instruction '" +
1643                         TheDef->getName() + "' has operand '" + OpName +
1644                         "' that doesn't appear in asm string!");
1645         unsigned NumOperands = (SubIdx == -1 ? OpInfo->MINumOperands : 1);
1646         ResOperands.push_back(ResOperand::getRenderedOp(SrcOperand,
1647                                                         NumOperands));
1648         break;
1649       }
1650       case CodeGenInstAlias::ResultOperand::K_Imm: {
1651         int64_t ImmVal = CGA.ResultOperands[AliasOpNo].getImm();
1652         ResOperands.push_back(ResOperand::getImmOp(ImmVal));
1653         break;
1654       }
1655       case CodeGenInstAlias::ResultOperand::K_Reg: {
1656         Record *Reg = CGA.ResultOperands[AliasOpNo].getRegister();
1657         ResOperands.push_back(ResOperand::getRegOp(Reg));
1658         break;
1659       }
1660       }
1661     }
1662   }
1663 }
1664
1665 static unsigned getConverterOperandID(const std::string &Name,
1666                                       SetVector<std::string> &Table,
1667                                       bool &IsNew) {
1668   IsNew = Table.insert(Name);
1669
1670   unsigned ID = IsNew ? Table.size() - 1 :
1671     std::find(Table.begin(), Table.end(), Name) - Table.begin();
1672
1673   assert(ID < Table.size());
1674
1675   return ID;
1676 }
1677
1678
1679 static void emitConvertFuncs(CodeGenTarget &Target, StringRef ClassName,
1680                              std::vector<std::unique_ptr<MatchableInfo>> &Infos,
1681                              raw_ostream &OS) {
1682   SetVector<std::string> OperandConversionKinds;
1683   SetVector<std::string> InstructionConversionKinds;
1684   std::vector<std::vector<uint8_t> > ConversionTable;
1685   size_t MaxRowLength = 2; // minimum is custom converter plus terminator.
1686
1687   // TargetOperandClass - This is the target's operand class, like X86Operand.
1688   std::string TargetOperandClass = Target.getName() + "Operand";
1689
1690   // Write the convert function to a separate stream, so we can drop it after
1691   // the enum. We'll build up the conversion handlers for the individual
1692   // operand types opportunistically as we encounter them.
1693   std::string ConvertFnBody;
1694   raw_string_ostream CvtOS(ConvertFnBody);
1695   // Start the unified conversion function.
1696   CvtOS << "void " << Target.getName() << ClassName << "::\n"
1697         << "convertToMCInst(unsigned Kind, MCInst &Inst, "
1698         << "unsigned Opcode,\n"
1699         << "                const OperandVector"
1700         << " &Operands) {\n"
1701         << "  assert(Kind < CVT_NUM_SIGNATURES && \"Invalid signature!\");\n"
1702         << "  const uint8_t *Converter = ConversionTable[Kind];\n"
1703         << "  Inst.setOpcode(Opcode);\n"
1704         << "  for (const uint8_t *p = Converter; *p; p+= 2) {\n"
1705         << "    switch (*p) {\n"
1706         << "    default: llvm_unreachable(\"invalid conversion entry!\");\n"
1707         << "    case CVT_Reg:\n"
1708         << "      static_cast<" << TargetOperandClass
1709         << "&>(*Operands[*(p + 1)]).addRegOperands(Inst, 1);\n"
1710         << "      break;\n"
1711         << "    case CVT_Tied:\n"
1712         << "      Inst.addOperand(Inst.getOperand(*(p + 1)));\n"
1713         << "      break;\n";
1714
1715   std::string OperandFnBody;
1716   raw_string_ostream OpOS(OperandFnBody);
1717   // Start the operand number lookup function.
1718   OpOS << "void " << Target.getName() << ClassName << "::\n"
1719        << "convertToMapAndConstraints(unsigned Kind,\n";
1720   OpOS.indent(27);
1721   OpOS << "const OperandVector &Operands) {\n"
1722        << "  assert(Kind < CVT_NUM_SIGNATURES && \"Invalid signature!\");\n"
1723        << "  unsigned NumMCOperands = 0;\n"
1724        << "  const uint8_t *Converter = ConversionTable[Kind];\n"
1725        << "  for (const uint8_t *p = Converter; *p; p+= 2) {\n"
1726        << "    switch (*p) {\n"
1727        << "    default: llvm_unreachable(\"invalid conversion entry!\");\n"
1728        << "    case CVT_Reg:\n"
1729        << "      Operands[*(p + 1)]->setMCOperandNum(NumMCOperands);\n"
1730        << "      Operands[*(p + 1)]->setConstraint(\"r\");\n"
1731        << "      ++NumMCOperands;\n"
1732        << "      break;\n"
1733        << "    case CVT_Tied:\n"
1734        << "      ++NumMCOperands;\n"
1735        << "      break;\n";
1736
1737   // Pre-populate the operand conversion kinds with the standard always
1738   // available entries.
1739   OperandConversionKinds.insert("CVT_Done");
1740   OperandConversionKinds.insert("CVT_Reg");
1741   OperandConversionKinds.insert("CVT_Tied");
1742   enum { CVT_Done, CVT_Reg, CVT_Tied };
1743
1744   for (auto &II : Infos) {
1745     // Check if we have a custom match function.
1746     std::string AsmMatchConverter =
1747       II->getResultInst()->TheDef->getValueAsString("AsmMatchConverter");
1748     if (!AsmMatchConverter.empty()) {
1749       std::string Signature = "ConvertCustom_" + AsmMatchConverter;
1750       II->ConversionFnKind = Signature;
1751
1752       // Check if we have already generated this signature.
1753       if (!InstructionConversionKinds.insert(Signature))
1754         continue;
1755
1756       // Remember this converter for the kind enum.
1757       unsigned KindID = OperandConversionKinds.size();
1758       OperandConversionKinds.insert("CVT_" +
1759                                     getEnumNameForToken(AsmMatchConverter));
1760
1761       // Add the converter row for this instruction.
1762       ConversionTable.push_back(std::vector<uint8_t>());
1763       ConversionTable.back().push_back(KindID);
1764       ConversionTable.back().push_back(CVT_Done);
1765
1766       // Add the handler to the conversion driver function.
1767       CvtOS << "    case CVT_"
1768             << getEnumNameForToken(AsmMatchConverter) << ":\n"
1769             << "      " << AsmMatchConverter << "(Inst, Operands);\n"
1770             << "      break;\n";
1771
1772       // FIXME: Handle the operand number lookup for custom match functions.
1773       continue;
1774     }
1775
1776     // Build the conversion function signature.
1777     std::string Signature = "Convert";
1778
1779     std::vector<uint8_t> ConversionRow;
1780
1781     // Compute the convert enum and the case body.
1782     MaxRowLength = std::max(MaxRowLength, II->ResOperands.size()*2 + 1 );
1783
1784     for (unsigned i = 0, e = II->ResOperands.size(); i != e; ++i) {
1785       const MatchableInfo::ResOperand &OpInfo = II->ResOperands[i];
1786
1787       // Generate code to populate each result operand.
1788       switch (OpInfo.Kind) {
1789       case MatchableInfo::ResOperand::RenderAsmOperand: {
1790         // This comes from something we parsed.
1791         const MatchableInfo::AsmOperand &Op =
1792           II->AsmOperands[OpInfo.AsmOperandNum];
1793
1794         // Registers are always converted the same, don't duplicate the
1795         // conversion function based on them.
1796         Signature += "__";
1797         std::string Class;
1798         Class = Op.Class->isRegisterClass() ? "Reg" : Op.Class->ClassName;
1799         Signature += Class;
1800         Signature += utostr(OpInfo.MINumOperands);
1801         Signature += "_" + itostr(OpInfo.AsmOperandNum);
1802
1803         // Add the conversion kind, if necessary, and get the associated ID
1804         // the index of its entry in the vector).
1805         std::string Name = "CVT_" + (Op.Class->isRegisterClass() ? "Reg" :
1806                                      Op.Class->RenderMethod);
1807         Name = getEnumNameForToken(Name);
1808
1809         bool IsNewConverter = false;
1810         unsigned ID = getConverterOperandID(Name, OperandConversionKinds,
1811                                             IsNewConverter);
1812
1813         // Add the operand entry to the instruction kind conversion row.
1814         ConversionRow.push_back(ID);
1815         ConversionRow.push_back(OpInfo.AsmOperandNum + 1);
1816
1817         if (!IsNewConverter)
1818           break;
1819
1820         // This is a new operand kind. Add a handler for it to the
1821         // converter driver.
1822         CvtOS << "    case " << Name << ":\n"
1823               << "      static_cast<" << TargetOperandClass
1824               << "&>(*Operands[*(p + 1)])." << Op.Class->RenderMethod
1825               << "(Inst, " << OpInfo.MINumOperands << ");\n"
1826               << "      break;\n";
1827
1828         // Add a handler for the operand number lookup.
1829         OpOS << "    case " << Name << ":\n"
1830              << "      Operands[*(p + 1)]->setMCOperandNum(NumMCOperands);\n";
1831
1832         if (Op.Class->isRegisterClass())
1833           OpOS << "      Operands[*(p + 1)]->setConstraint(\"r\");\n";
1834         else
1835           OpOS << "      Operands[*(p + 1)]->setConstraint(\"m\");\n";
1836         OpOS << "      NumMCOperands += " << OpInfo.MINumOperands << ";\n"
1837              << "      break;\n";
1838         break;
1839       }
1840       case MatchableInfo::ResOperand::TiedOperand: {
1841         // If this operand is tied to a previous one, just copy the MCInst
1842         // operand from the earlier one.We can only tie single MCOperand values.
1843         assert(OpInfo.MINumOperands == 1 && "Not a singular MCOperand");
1844         unsigned TiedOp = OpInfo.TiedOperandNum;
1845         assert(i > TiedOp && "Tied operand precedes its target!");
1846         Signature += "__Tie" + utostr(TiedOp);
1847         ConversionRow.push_back(CVT_Tied);
1848         ConversionRow.push_back(TiedOp);
1849         break;
1850       }
1851       case MatchableInfo::ResOperand::ImmOperand: {
1852         int64_t Val = OpInfo.ImmVal;
1853         std::string Ty = "imm_" + itostr(Val);
1854         Signature += "__" + Ty;
1855
1856         std::string Name = "CVT_" + Ty;
1857         bool IsNewConverter = false;
1858         unsigned ID = getConverterOperandID(Name, OperandConversionKinds,
1859                                             IsNewConverter);
1860         // Add the operand entry to the instruction kind conversion row.
1861         ConversionRow.push_back(ID);
1862         ConversionRow.push_back(0);
1863
1864         if (!IsNewConverter)
1865           break;
1866
1867         CvtOS << "    case " << Name << ":\n"
1868               << "      Inst.addOperand(MCOperand::CreateImm(" << Val << "));\n"
1869               << "      break;\n";
1870
1871         OpOS << "    case " << Name << ":\n"
1872              << "      Operands[*(p + 1)]->setMCOperandNum(NumMCOperands);\n"
1873              << "      Operands[*(p + 1)]->setConstraint(\"\");\n"
1874              << "      ++NumMCOperands;\n"
1875              << "      break;\n";
1876         break;
1877       }
1878       case MatchableInfo::ResOperand::RegOperand: {
1879         std::string Reg, Name;
1880         if (!OpInfo.Register) {
1881           Name = "reg0";
1882           Reg = "0";
1883         } else {
1884           Reg = getQualifiedName(OpInfo.Register);
1885           Name = "reg" + OpInfo.Register->getName();
1886         }
1887         Signature += "__" + Name;
1888         Name = "CVT_" + Name;
1889         bool IsNewConverter = false;
1890         unsigned ID = getConverterOperandID(Name, OperandConversionKinds,
1891                                             IsNewConverter);
1892         // Add the operand entry to the instruction kind conversion row.
1893         ConversionRow.push_back(ID);
1894         ConversionRow.push_back(0);
1895
1896         if (!IsNewConverter)
1897           break;
1898         CvtOS << "    case " << Name << ":\n"
1899               << "      Inst.addOperand(MCOperand::CreateReg(" << Reg << "));\n"
1900               << "      break;\n";
1901
1902         OpOS << "    case " << Name << ":\n"
1903              << "      Operands[*(p + 1)]->setMCOperandNum(NumMCOperands);\n"
1904              << "      Operands[*(p + 1)]->setConstraint(\"m\");\n"
1905              << "      ++NumMCOperands;\n"
1906              << "      break;\n";
1907       }
1908       }
1909     }
1910
1911     // If there were no operands, add to the signature to that effect
1912     if (Signature == "Convert")
1913       Signature += "_NoOperands";
1914
1915     II->ConversionFnKind = Signature;
1916
1917     // Save the signature. If we already have it, don't add a new row
1918     // to the table.
1919     if (!InstructionConversionKinds.insert(Signature))
1920       continue;
1921
1922     // Add the row to the table.
1923     ConversionTable.push_back(ConversionRow);
1924   }
1925
1926   // Finish up the converter driver function.
1927   CvtOS << "    }\n  }\n}\n\n";
1928
1929   // Finish up the operand number lookup function.
1930   OpOS << "    }\n  }\n}\n\n";
1931
1932   OS << "namespace {\n";
1933
1934   // Output the operand conversion kind enum.
1935   OS << "enum OperatorConversionKind {\n";
1936   for (unsigned i = 0, e = OperandConversionKinds.size(); i != e; ++i)
1937     OS << "  " << OperandConversionKinds[i] << ",\n";
1938   OS << "  CVT_NUM_CONVERTERS\n";
1939   OS << "};\n\n";
1940
1941   // Output the instruction conversion kind enum.
1942   OS << "enum InstructionConversionKind {\n";
1943   for (SetVector<std::string>::const_iterator
1944          i = InstructionConversionKinds.begin(),
1945          e = InstructionConversionKinds.end(); i != e; ++i)
1946     OS << "  " << *i << ",\n";
1947   OS << "  CVT_NUM_SIGNATURES\n";
1948   OS << "};\n\n";
1949
1950
1951   OS << "} // end anonymous namespace\n\n";
1952
1953   // Output the conversion table.
1954   OS << "static const uint8_t ConversionTable[CVT_NUM_SIGNATURES]["
1955      << MaxRowLength << "] = {\n";
1956
1957   for (unsigned Row = 0, ERow = ConversionTable.size(); Row != ERow; ++Row) {
1958     assert(ConversionTable[Row].size() % 2 == 0 && "bad conversion row!");
1959     OS << "  // " << InstructionConversionKinds[Row] << "\n";
1960     OS << "  { ";
1961     for (unsigned i = 0, e = ConversionTable[Row].size(); i != e; i += 2)
1962       OS << OperandConversionKinds[ConversionTable[Row][i]] << ", "
1963          << (unsigned)(ConversionTable[Row][i + 1]) << ", ";
1964     OS << "CVT_Done },\n";
1965   }
1966
1967   OS << "};\n\n";
1968
1969   // Spit out the conversion driver function.
1970   OS << CvtOS.str();
1971
1972   // Spit out the operand number lookup function.
1973   OS << OpOS.str();
1974 }
1975
1976 /// emitMatchClassEnumeration - Emit the enumeration for match class kinds.
1977 static void emitMatchClassEnumeration(CodeGenTarget &Target,
1978                                  std::vector<std::unique_ptr<ClassInfo>> &Infos,
1979                                  raw_ostream &OS) {
1980   OS << "namespace {\n\n";
1981
1982   OS << "/// MatchClassKind - The kinds of classes which participate in\n"
1983      << "/// instruction matching.\n";
1984   OS << "enum MatchClassKind {\n";
1985   OS << "  InvalidMatchClass = 0,\n";
1986   for (const auto &CI : Infos) {
1987     OS << "  " << CI->Name << ", // ";
1988     if (CI->Kind == ClassInfo::Token) {
1989       OS << "'" << CI->ValueName << "'\n";
1990     } else if (CI->isRegisterClass()) {
1991       if (!CI->ValueName.empty())
1992         OS << "register class '" << CI->ValueName << "'\n";
1993       else
1994         OS << "derived register class\n";
1995     } else {
1996       OS << "user defined class '" << CI->ValueName << "'\n";
1997     }
1998   }
1999   OS << "  NumMatchClassKinds\n";
2000   OS << "};\n\n";
2001
2002   OS << "}\n\n";
2003 }
2004
2005 /// emitValidateOperandClass - Emit the function to validate an operand class.
2006 static void emitValidateOperandClass(AsmMatcherInfo &Info,
2007                                      raw_ostream &OS) {
2008   OS << "static unsigned validateOperandClass(MCParsedAsmOperand &GOp, "
2009      << "MatchClassKind Kind) {\n";
2010   OS << "  " << Info.Target.getName() << "Operand &Operand = ("
2011      << Info.Target.getName() << "Operand&)GOp;\n";
2012
2013   // The InvalidMatchClass is not to match any operand.
2014   OS << "  if (Kind == InvalidMatchClass)\n";
2015   OS << "    return MCTargetAsmParser::Match_InvalidOperand;\n\n";
2016
2017   // Check for Token operands first.
2018   // FIXME: Use a more specific diagnostic type.
2019   OS << "  if (Operand.isToken())\n";
2020   OS << "    return isSubclass(matchTokenString(Operand.getToken()), Kind) ?\n"
2021      << "             MCTargetAsmParser::Match_Success :\n"
2022      << "             MCTargetAsmParser::Match_InvalidOperand;\n\n";
2023
2024   // Check the user classes. We don't care what order since we're only
2025   // actually matching against one of them.
2026   for (const auto &CI : Info.Classes) {
2027     if (!CI->isUserClass())
2028       continue;
2029
2030     OS << "  // '" << CI->ClassName << "' class\n";
2031     OS << "  if (Kind == " << CI->Name << ") {\n";
2032     OS << "    if (Operand." << CI->PredicateMethod << "())\n";
2033     OS << "      return MCTargetAsmParser::Match_Success;\n";
2034     if (!CI->DiagnosticType.empty())
2035       OS << "    return " << Info.Target.getName() << "AsmParser::Match_"
2036          << CI->DiagnosticType << ";\n";
2037     OS << "  }\n\n";
2038   }
2039
2040   // Check for register operands, including sub-classes.
2041   OS << "  if (Operand.isReg()) {\n";
2042   OS << "    MatchClassKind OpKind;\n";
2043   OS << "    switch (Operand.getReg()) {\n";
2044   OS << "    default: OpKind = InvalidMatchClass; break;\n";
2045   for (const auto &RC : Info.RegisterClasses)
2046     OS << "    case " << Info.Target.getName() << "::"
2047        << RC.first->getName() << ": OpKind = " << RC.second->Name
2048        << "; break;\n";
2049   OS << "    }\n";
2050   OS << "    return isSubclass(OpKind, Kind) ? "
2051      << "MCTargetAsmParser::Match_Success :\n                             "
2052      << "         MCTargetAsmParser::Match_InvalidOperand;\n  }\n\n";
2053
2054   // Generic fallthrough match failure case for operands that don't have
2055   // specialized diagnostic types.
2056   OS << "  return MCTargetAsmParser::Match_InvalidOperand;\n";
2057   OS << "}\n\n";
2058 }
2059
2060 /// emitIsSubclass - Emit the subclass predicate function.
2061 static void emitIsSubclass(CodeGenTarget &Target,
2062                            std::vector<std::unique_ptr<ClassInfo>> &Infos,
2063                            raw_ostream &OS) {
2064   OS << "/// isSubclass - Compute whether \\p A is a subclass of \\p B.\n";
2065   OS << "static bool isSubclass(MatchClassKind A, MatchClassKind B) {\n";
2066   OS << "  if (A == B)\n";
2067   OS << "    return true;\n\n";
2068
2069   std::string OStr;
2070   raw_string_ostream SS(OStr);
2071   unsigned Count = 0;
2072   SS << "  switch (A) {\n";
2073   SS << "  default:\n";
2074   SS << "    return false;\n";
2075   for (const auto &A : Infos) {
2076     std::vector<StringRef> SuperClasses;
2077     for (const auto &B : Infos) {
2078       if (A != B && A->isSubsetOf(*B))
2079         SuperClasses.push_back(B->Name);
2080     }
2081
2082     if (SuperClasses.empty())
2083       continue;
2084     ++Count;
2085
2086     SS << "\n  case " << A->Name << ":\n";
2087
2088     if (SuperClasses.size() == 1) {
2089       SS << "    return B == " << SuperClasses.back().str() << ";\n";
2090       continue;
2091     }
2092
2093     if (!SuperClasses.empty()) {
2094       SS << "    switch (B) {\n";
2095       SS << "    default: return false;\n";
2096       for (unsigned i = 0, e = SuperClasses.size(); i != e; ++i)
2097         SS << "    case " << SuperClasses[i].str() << ": return true;\n";
2098       SS << "    }\n";
2099     } else {
2100       // No case statement to emit
2101       SS << "    return false;\n";
2102     }
2103   }
2104   SS << "  }\n";
2105
2106   // If there were case statements emitted into the string stream, write them
2107   // to the output stream, otherwise write the default.
2108   if (Count)
2109     OS << SS.str();
2110   else
2111     OS << "  return false;\n";
2112
2113   OS << "}\n\n";
2114 }
2115
2116 /// emitMatchTokenString - Emit the function to match a token string to the
2117 /// appropriate match class value.
2118 static void emitMatchTokenString(CodeGenTarget &Target,
2119                                  std::vector<std::unique_ptr<ClassInfo>> &Infos,
2120                                  raw_ostream &OS) {
2121   // Construct the match list.
2122   std::vector<StringMatcher::StringPair> Matches;
2123   for (const auto &CI : Infos) {
2124     if (CI->Kind == ClassInfo::Token)
2125       Matches.push_back(StringMatcher::StringPair(CI->ValueName,
2126                                                   "return " + CI->Name + ";"));
2127   }
2128
2129   OS << "static MatchClassKind matchTokenString(StringRef Name) {\n";
2130
2131   StringMatcher("Name", Matches, OS).Emit();
2132
2133   OS << "  return InvalidMatchClass;\n";
2134   OS << "}\n\n";
2135 }
2136
2137 /// emitMatchRegisterName - Emit the function to match a string to the target
2138 /// specific register enum.
2139 static void emitMatchRegisterName(CodeGenTarget &Target, Record *AsmParser,
2140                                   raw_ostream &OS) {
2141   // Construct the match list.
2142   std::vector<StringMatcher::StringPair> Matches;
2143   const std::vector<CodeGenRegister*> &Regs =
2144     Target.getRegBank().getRegisters();
2145   for (unsigned i = 0, e = Regs.size(); i != e; ++i) {
2146     const CodeGenRegister *Reg = Regs[i];
2147     if (Reg->TheDef->getValueAsString("AsmName").empty())
2148       continue;
2149
2150     Matches.push_back(StringMatcher::StringPair(
2151                                      Reg->TheDef->getValueAsString("AsmName"),
2152                                      "return " + utostr(Reg->EnumValue) + ";"));
2153   }
2154
2155   OS << "static unsigned MatchRegisterName(StringRef Name) {\n";
2156
2157   StringMatcher("Name", Matches, OS).Emit();
2158
2159   OS << "  return 0;\n";
2160   OS << "}\n\n";
2161 }
2162
2163 static const char *getMinimalTypeForRange(uint64_t Range) {
2164   assert(Range <= 0xFFFFFFFFFFFFFFFFULL && "Enum too large");
2165   if (Range > 0xFFFFFFFFULL)
2166     return "uint64_t";
2167   if (Range > 0xFFFF)
2168     return "uint32_t";
2169   if (Range > 0xFF)
2170     return "uint16_t";
2171   return "uint8_t";
2172 }
2173
2174 static const char *getMinimalRequiredFeaturesType(const AsmMatcherInfo &Info) {
2175   uint64_t MaxIndex = Info.SubtargetFeatures.size();
2176   if (MaxIndex > 0)
2177     MaxIndex--;
2178   return getMinimalTypeForRange(1ULL << MaxIndex);
2179 }
2180
2181 /// emitSubtargetFeatureFlagEnumeration - Emit the subtarget feature flag
2182 /// definitions.
2183 static void emitSubtargetFeatureFlagEnumeration(AsmMatcherInfo &Info,
2184                                                 raw_ostream &OS) {
2185   OS << "// Flags for subtarget features that participate in "
2186      << "instruction matching.\n";
2187   OS << "enum SubtargetFeatureFlag : " << getMinimalRequiredFeaturesType(Info)
2188      << " {\n";
2189   for (const auto &SF : Info.SubtargetFeatures) {
2190     SubtargetFeatureInfo &SFI = *SF.second;
2191     OS << "  " << SFI.getEnumName() << " = (1ULL << " << SFI.Index << "),\n";
2192   }
2193   OS << "  Feature_None = 0\n";
2194   OS << "};\n\n";
2195 }
2196
2197 /// emitOperandDiagnosticTypes - Emit the operand matching diagnostic types.
2198 static void emitOperandDiagnosticTypes(AsmMatcherInfo &Info, raw_ostream &OS) {
2199   // Get the set of diagnostic types from all of the operand classes.
2200   std::set<StringRef> Types;
2201   for (std::map<Record*, ClassInfo*>::const_iterator
2202        I = Info.AsmOperandClasses.begin(),
2203        E = Info.AsmOperandClasses.end(); I != E; ++I) {
2204     if (!I->second->DiagnosticType.empty())
2205       Types.insert(I->second->DiagnosticType);
2206   }
2207
2208   if (Types.empty()) return;
2209
2210   // Now emit the enum entries.
2211   for (std::set<StringRef>::const_iterator I = Types.begin(), E = Types.end();
2212        I != E; ++I)
2213     OS << "  Match_" << *I << ",\n";
2214   OS << "  END_OPERAND_DIAGNOSTIC_TYPES\n";
2215 }
2216
2217 /// emitGetSubtargetFeatureName - Emit the helper function to get the
2218 /// user-level name for a subtarget feature.
2219 static void emitGetSubtargetFeatureName(AsmMatcherInfo &Info, raw_ostream &OS) {
2220   OS << "// User-level names for subtarget features that participate in\n"
2221      << "// instruction matching.\n"
2222      << "static const char *getSubtargetFeatureName(uint64_t Val) {\n";
2223   if (!Info.SubtargetFeatures.empty()) {
2224     OS << "  switch(Val) {\n";
2225     for (const auto &SF : Info.SubtargetFeatures) {
2226       SubtargetFeatureInfo &SFI = *SF.second;
2227       // FIXME: Totally just a placeholder name to get the algorithm working.
2228       OS << "  case " << SFI.getEnumName() << ": return \""
2229          << SFI.TheDef->getValueAsString("PredicateName") << "\";\n";
2230     }
2231     OS << "  default: return \"(unknown)\";\n";
2232     OS << "  }\n";
2233   } else {
2234     // Nothing to emit, so skip the switch
2235     OS << "  return \"(unknown)\";\n";
2236   }
2237   OS << "}\n\n";
2238 }
2239
2240 /// emitComputeAvailableFeatures - Emit the function to compute the list of
2241 /// available features given a subtarget.
2242 static void emitComputeAvailableFeatures(AsmMatcherInfo &Info,
2243                                          raw_ostream &OS) {
2244   std::string ClassName =
2245     Info.AsmParser->getValueAsString("AsmParserClassName");
2246
2247   OS << "uint64_t " << Info.Target.getName() << ClassName << "::\n"
2248      << "ComputeAvailableFeatures(uint64_t FB) const {\n";
2249   OS << "  uint64_t Features = 0;\n";
2250   for (const auto &SF : Info.SubtargetFeatures) {
2251     SubtargetFeatureInfo &SFI = *SF.second;
2252
2253     OS << "  if (";
2254     std::string CondStorage =
2255       SFI.TheDef->getValueAsString("AssemblerCondString");
2256     StringRef Conds = CondStorage;
2257     std::pair<StringRef,StringRef> Comma = Conds.split(',');
2258     bool First = true;
2259     do {
2260       if (!First)
2261         OS << " && ";
2262
2263       bool Neg = false;
2264       StringRef Cond = Comma.first;
2265       if (Cond[0] == '!') {
2266         Neg = true;
2267         Cond = Cond.substr(1);
2268       }
2269
2270       OS << "((FB & " << Info.Target.getName() << "::" << Cond << ")";
2271       if (Neg)
2272         OS << " == 0";
2273       else
2274         OS << " != 0";
2275       OS << ")";
2276
2277       if (Comma.second.empty())
2278         break;
2279
2280       First = false;
2281       Comma = Comma.second.split(',');
2282     } while (true);
2283
2284     OS << ")\n";
2285     OS << "    Features |= " << SFI.getEnumName() << ";\n";
2286   }
2287   OS << "  return Features;\n";
2288   OS << "}\n\n";
2289 }
2290
2291 static std::string GetAliasRequiredFeatures(Record *R,
2292                                             const AsmMatcherInfo &Info) {
2293   std::vector<Record*> ReqFeatures = R->getValueAsListOfDefs("Predicates");
2294   std::string Result;
2295   unsigned NumFeatures = 0;
2296   for (unsigned i = 0, e = ReqFeatures.size(); i != e; ++i) {
2297     SubtargetFeatureInfo *F = Info.getSubtargetFeature(ReqFeatures[i]);
2298
2299     if (!F)
2300       PrintFatalError(R->getLoc(), "Predicate '" + ReqFeatures[i]->getName() +
2301                     "' is not marked as an AssemblerPredicate!");
2302
2303     if (NumFeatures)
2304       Result += '|';
2305
2306     Result += F->getEnumName();
2307     ++NumFeatures;
2308   }
2309
2310   if (NumFeatures > 1)
2311     Result = '(' + Result + ')';
2312   return Result;
2313 }
2314
2315 static void emitMnemonicAliasVariant(raw_ostream &OS,const AsmMatcherInfo &Info,
2316                                      std::vector<Record*> &Aliases,
2317                                      unsigned Indent = 0,
2318                                   StringRef AsmParserVariantName = StringRef()){
2319   // Keep track of all the aliases from a mnemonic.  Use an std::map so that the
2320   // iteration order of the map is stable.
2321   std::map<std::string, std::vector<Record*> > AliasesFromMnemonic;
2322
2323   for (unsigned i = 0, e = Aliases.size(); i != e; ++i) {
2324     Record *R = Aliases[i];
2325     // FIXME: Allow AssemblerVariantName to be a comma separated list.
2326     std::string AsmVariantName = R->getValueAsString("AsmVariantName");
2327     if (AsmVariantName != AsmParserVariantName)
2328       continue;
2329     AliasesFromMnemonic[R->getValueAsString("FromMnemonic")].push_back(R);
2330   }
2331   if (AliasesFromMnemonic.empty())
2332     return;
2333
2334   // Process each alias a "from" mnemonic at a time, building the code executed
2335   // by the string remapper.
2336   std::vector<StringMatcher::StringPair> Cases;
2337   for (std::map<std::string, std::vector<Record*> >::iterator
2338        I = AliasesFromMnemonic.begin(), E = AliasesFromMnemonic.end();
2339        I != E; ++I) {
2340     const std::vector<Record*> &ToVec = I->second;
2341
2342     // Loop through each alias and emit code that handles each case.  If there
2343     // are two instructions without predicates, emit an error.  If there is one,
2344     // emit it last.
2345     std::string MatchCode;
2346     int AliasWithNoPredicate = -1;
2347
2348     for (unsigned i = 0, e = ToVec.size(); i != e; ++i) {
2349       Record *R = ToVec[i];
2350       std::string FeatureMask = GetAliasRequiredFeatures(R, Info);
2351
2352       // If this unconditionally matches, remember it for later and diagnose
2353       // duplicates.
2354       if (FeatureMask.empty()) {
2355         if (AliasWithNoPredicate != -1) {
2356           // We can't have two aliases from the same mnemonic with no predicate.
2357           PrintError(ToVec[AliasWithNoPredicate]->getLoc(),
2358                      "two MnemonicAliases with the same 'from' mnemonic!");
2359           PrintFatalError(R->getLoc(), "this is the other MnemonicAlias.");
2360         }
2361
2362         AliasWithNoPredicate = i;
2363         continue;
2364       }
2365       if (R->getValueAsString("ToMnemonic") == I->first)
2366         PrintFatalError(R->getLoc(), "MnemonicAlias to the same string");
2367
2368       if (!MatchCode.empty())
2369         MatchCode += "else ";
2370       MatchCode += "if ((Features & " + FeatureMask + ") == "+FeatureMask+")\n";
2371       MatchCode += "  Mnemonic = \"" +R->getValueAsString("ToMnemonic")+"\";\n";
2372     }
2373
2374     if (AliasWithNoPredicate != -1) {
2375       Record *R = ToVec[AliasWithNoPredicate];
2376       if (!MatchCode.empty())
2377         MatchCode += "else\n  ";
2378       MatchCode += "Mnemonic = \"" + R->getValueAsString("ToMnemonic")+"\";\n";
2379     }
2380
2381     MatchCode += "return;";
2382
2383     Cases.push_back(std::make_pair(I->first, MatchCode));
2384   }
2385   StringMatcher("Mnemonic", Cases, OS).Emit(Indent);
2386 }
2387
2388 /// emitMnemonicAliases - If the target has any MnemonicAlias<> definitions,
2389 /// emit a function for them and return true, otherwise return false.
2390 static bool emitMnemonicAliases(raw_ostream &OS, const AsmMatcherInfo &Info,
2391                                 CodeGenTarget &Target) {
2392   // Ignore aliases when match-prefix is set.
2393   if (!MatchPrefix.empty())
2394     return false;
2395
2396   std::vector<Record*> Aliases =
2397     Info.getRecords().getAllDerivedDefinitions("MnemonicAlias");
2398   if (Aliases.empty()) return false;
2399
2400   OS << "static void applyMnemonicAliases(StringRef &Mnemonic, "
2401     "uint64_t Features, unsigned VariantID) {\n";
2402   OS << "  switch (VariantID) {\n";
2403   unsigned VariantCount = Target.getAsmParserVariantCount();
2404   for (unsigned VC = 0; VC != VariantCount; ++VC) {
2405     Record *AsmVariant = Target.getAsmParserVariant(VC);
2406     int AsmParserVariantNo = AsmVariant->getValueAsInt("Variant");
2407     std::string AsmParserVariantName = AsmVariant->getValueAsString("Name");
2408     OS << "    case " << AsmParserVariantNo << ":\n";
2409     emitMnemonicAliasVariant(OS, Info, Aliases, /*Indent=*/2,
2410                              AsmParserVariantName);
2411     OS << "    break;\n";
2412   }
2413   OS << "  }\n";
2414
2415   // Emit aliases that apply to all variants.
2416   emitMnemonicAliasVariant(OS, Info, Aliases);
2417
2418   OS << "}\n\n";
2419
2420   return true;
2421 }
2422
2423 static void emitCustomOperandParsing(raw_ostream &OS, CodeGenTarget &Target,
2424                               const AsmMatcherInfo &Info, StringRef ClassName,
2425                               StringToOffsetTable &StringTable,
2426                               unsigned MaxMnemonicIndex) {
2427   unsigned MaxMask = 0;
2428   for (std::vector<OperandMatchEntry>::const_iterator it =
2429        Info.OperandMatchInfo.begin(), ie = Info.OperandMatchInfo.end();
2430        it != ie; ++it) {
2431     MaxMask |= it->OperandMask;
2432   }
2433
2434   // Emit the static custom operand parsing table;
2435   OS << "namespace {\n";
2436   OS << "  struct OperandMatchEntry {\n";
2437   OS << "    " << getMinimalRequiredFeaturesType(Info)
2438                << " RequiredFeatures;\n";
2439   OS << "    " << getMinimalTypeForRange(MaxMnemonicIndex)
2440                << " Mnemonic;\n";
2441   OS << "    " << getMinimalTypeForRange(Info.Classes.size())
2442                << " Class;\n";
2443   OS << "    " << getMinimalTypeForRange(MaxMask)
2444                << " OperandMask;\n\n";
2445   OS << "    StringRef getMnemonic() const {\n";
2446   OS << "      return StringRef(MnemonicTable + Mnemonic + 1,\n";
2447   OS << "                       MnemonicTable[Mnemonic]);\n";
2448   OS << "    }\n";
2449   OS << "  };\n\n";
2450
2451   OS << "  // Predicate for searching for an opcode.\n";
2452   OS << "  struct LessOpcodeOperand {\n";
2453   OS << "    bool operator()(const OperandMatchEntry &LHS, StringRef RHS) {\n";
2454   OS << "      return LHS.getMnemonic()  < RHS;\n";
2455   OS << "    }\n";
2456   OS << "    bool operator()(StringRef LHS, const OperandMatchEntry &RHS) {\n";
2457   OS << "      return LHS < RHS.getMnemonic();\n";
2458   OS << "    }\n";
2459   OS << "    bool operator()(const OperandMatchEntry &LHS,";
2460   OS << " const OperandMatchEntry &RHS) {\n";
2461   OS << "      return LHS.getMnemonic() < RHS.getMnemonic();\n";
2462   OS << "    }\n";
2463   OS << "  };\n";
2464
2465   OS << "} // end anonymous namespace.\n\n";
2466
2467   OS << "static const OperandMatchEntry OperandMatchTable["
2468      << Info.OperandMatchInfo.size() << "] = {\n";
2469
2470   OS << "  /* Operand List Mask, Mnemonic, Operand Class, Features */\n";
2471   for (std::vector<OperandMatchEntry>::const_iterator it =
2472        Info.OperandMatchInfo.begin(), ie = Info.OperandMatchInfo.end();
2473        it != ie; ++it) {
2474     const OperandMatchEntry &OMI = *it;
2475     const MatchableInfo &II = *OMI.MI;
2476
2477     OS << "  { ";
2478
2479     // Write the required features mask.
2480     if (!II.RequiredFeatures.empty()) {
2481       for (unsigned i = 0, e = II.RequiredFeatures.size(); i != e; ++i) {
2482         if (i) OS << "|";
2483         OS << II.RequiredFeatures[i]->getEnumName();
2484       }
2485     } else
2486       OS << "0";
2487
2488     // Store a pascal-style length byte in the mnemonic.
2489     std::string LenMnemonic = char(II.Mnemonic.size()) + II.Mnemonic.str();
2490     OS << ", " << StringTable.GetOrAddStringOffset(LenMnemonic, false)
2491        << " /* " << II.Mnemonic << " */, ";
2492
2493     OS << OMI.CI->Name;
2494
2495     OS << ", " << OMI.OperandMask;
2496     OS << " /* ";
2497     bool printComma = false;
2498     for (int i = 0, e = 31; i !=e; ++i)
2499       if (OMI.OperandMask & (1 << i)) {
2500         if (printComma)
2501           OS << ", ";
2502         OS << i;
2503         printComma = true;
2504       }
2505     OS << " */";
2506
2507     OS << " },\n";
2508   }
2509   OS << "};\n\n";
2510
2511   // Emit the operand class switch to call the correct custom parser for
2512   // the found operand class.
2513   OS << Target.getName() << ClassName << "::OperandMatchResultTy "
2514      << Target.getName() << ClassName << "::\n"
2515      << "tryCustomParseOperand(OperandVector"
2516      << " &Operands,\n                      unsigned MCK) {\n\n"
2517      << "  switch(MCK) {\n";
2518
2519   for (const auto &CI : Info.Classes) {
2520     if (CI->ParserMethod.empty())
2521       continue;
2522     OS << "  case " << CI->Name << ":\n"
2523        << "    return " << CI->ParserMethod << "(Operands);\n";
2524   }
2525
2526   OS << "  default:\n";
2527   OS << "    return MatchOperand_NoMatch;\n";
2528   OS << "  }\n";
2529   OS << "  return MatchOperand_NoMatch;\n";
2530   OS << "}\n\n";
2531
2532   // Emit the static custom operand parser. This code is very similar with
2533   // the other matcher. Also use MatchResultTy here just in case we go for
2534   // a better error handling.
2535   OS << Target.getName() << ClassName << "::OperandMatchResultTy "
2536      << Target.getName() << ClassName << "::\n"
2537      << "MatchOperandParserImpl(OperandVector"
2538      << " &Operands,\n                       StringRef Mnemonic) {\n";
2539
2540   // Emit code to get the available features.
2541   OS << "  // Get the current feature set.\n";
2542   OS << "  uint64_t AvailableFeatures = getAvailableFeatures();\n\n";
2543
2544   OS << "  // Get the next operand index.\n";
2545   OS << "  unsigned NextOpNum = Operands.size()-1;\n";
2546
2547   // Emit code to search the table.
2548   OS << "  // Search the table.\n";
2549   OS << "  std::pair<const OperandMatchEntry*, const OperandMatchEntry*>";
2550   OS << " MnemonicRange =\n";
2551   OS << "    std::equal_range(OperandMatchTable, OperandMatchTable+"
2552      << Info.OperandMatchInfo.size() << ", Mnemonic,\n"
2553      << "                     LessOpcodeOperand());\n\n";
2554
2555   OS << "  if (MnemonicRange.first == MnemonicRange.second)\n";
2556   OS << "    return MatchOperand_NoMatch;\n\n";
2557
2558   OS << "  for (const OperandMatchEntry *it = MnemonicRange.first,\n"
2559      << "       *ie = MnemonicRange.second; it != ie; ++it) {\n";
2560
2561   OS << "    // equal_range guarantees that instruction mnemonic matches.\n";
2562   OS << "    assert(Mnemonic == it->getMnemonic());\n\n";
2563
2564   // Emit check that the required features are available.
2565   OS << "    // check if the available features match\n";
2566   OS << "    if ((AvailableFeatures & it->RequiredFeatures) "
2567      << "!= it->RequiredFeatures) {\n";
2568   OS << "      continue;\n";
2569   OS << "    }\n\n";
2570
2571   // Emit check to ensure the operand number matches.
2572   OS << "    // check if the operand in question has a custom parser.\n";
2573   OS << "    if (!(it->OperandMask & (1 << NextOpNum)))\n";
2574   OS << "      continue;\n\n";
2575
2576   // Emit call to the custom parser method
2577   OS << "    // call custom parse method to handle the operand\n";
2578   OS << "    OperandMatchResultTy Result = ";
2579   OS << "tryCustomParseOperand(Operands, it->Class);\n";
2580   OS << "    if (Result != MatchOperand_NoMatch)\n";
2581   OS << "      return Result;\n";
2582   OS << "  }\n\n";
2583
2584   OS << "  // Okay, we had no match.\n";
2585   OS << "  return MatchOperand_NoMatch;\n";
2586   OS << "}\n\n";
2587 }
2588
2589 void AsmMatcherEmitter::run(raw_ostream &OS) {
2590   CodeGenTarget Target(Records);
2591   Record *AsmParser = Target.getAsmParser();
2592   std::string ClassName = AsmParser->getValueAsString("AsmParserClassName");
2593
2594   // Compute the information on the instructions to match.
2595   AsmMatcherInfo Info(AsmParser, Target, Records);
2596   Info.buildInfo();
2597
2598   // Sort the instruction table using the partial order on classes. We use
2599   // stable_sort to ensure that ambiguous instructions are still
2600   // deterministically ordered.
2601   std::stable_sort(Info.Matchables.begin(), Info.Matchables.end(),
2602                    [](const std::unique_ptr<MatchableInfo> &a,
2603                       const std::unique_ptr<MatchableInfo> &b){
2604                      return *a < *b;});
2605
2606   DEBUG_WITH_TYPE("instruction_info", {
2607       for (const auto &MI : Info.Matchables)
2608         MI->dump();
2609     });
2610
2611   // Check for ambiguous matchables.
2612   DEBUG_WITH_TYPE("ambiguous_instrs", {
2613     unsigned NumAmbiguous = 0;
2614     for (unsigned i = 0, e = Info.Matchables.size(); i != e; ++i) {
2615       for (unsigned j = i + 1; j != e; ++j) {
2616         const MatchableInfo &A = *Info.Matchables[i];
2617         const MatchableInfo &B = *Info.Matchables[j];
2618
2619         if (A.couldMatchAmbiguouslyWith(B)) {
2620           errs() << "warning: ambiguous matchables:\n";
2621           A.dump();
2622           errs() << "\nis incomparable with:\n";
2623           B.dump();
2624           errs() << "\n\n";
2625           ++NumAmbiguous;
2626         }
2627       }
2628     }
2629     if (NumAmbiguous)
2630       errs() << "warning: " << NumAmbiguous
2631              << " ambiguous matchables!\n";
2632   });
2633
2634   // Compute the information on the custom operand parsing.
2635   Info.buildOperandMatchInfo();
2636
2637   // Write the output.
2638
2639   // Information for the class declaration.
2640   OS << "\n#ifdef GET_ASSEMBLER_HEADER\n";
2641   OS << "#undef GET_ASSEMBLER_HEADER\n";
2642   OS << "  // This should be included into the middle of the declaration of\n";
2643   OS << "  // your subclasses implementation of MCTargetAsmParser.\n";
2644   OS << "  uint64_t ComputeAvailableFeatures(uint64_t FeatureBits) const;\n";
2645   OS << "  void convertToMCInst(unsigned Kind, MCInst &Inst, "
2646      << "unsigned Opcode,\n"
2647      << "                       const OperandVector "
2648      << "&Operands);\n";
2649   OS << "  void convertToMapAndConstraints(unsigned Kind,\n                ";
2650   OS << "           const OperandVector &Operands) override;\n";
2651   OS << "  bool mnemonicIsValid(StringRef Mnemonic, unsigned VariantID) override;\n";
2652   OS << "  unsigned MatchInstructionImpl(\n";
2653   OS.indent(27);
2654   OS << "const OperandVector &Operands,\n"
2655      << "                                MCInst &Inst,\n"
2656      << "                                uint64_t &ErrorInfo,"
2657      << " bool matchingInlineAsm,\n"
2658      << "                                unsigned VariantID = 0);\n";
2659
2660   if (Info.OperandMatchInfo.size()) {
2661     OS << "\n  enum OperandMatchResultTy {\n";
2662     OS << "    MatchOperand_Success,    // operand matched successfully\n";
2663     OS << "    MatchOperand_NoMatch,    // operand did not match\n";
2664     OS << "    MatchOperand_ParseFail   // operand matched but had errors\n";
2665     OS << "  };\n";
2666     OS << "  OperandMatchResultTy MatchOperandParserImpl(\n";
2667     OS << "    OperandVector &Operands,\n";
2668     OS << "    StringRef Mnemonic);\n";
2669
2670     OS << "  OperandMatchResultTy tryCustomParseOperand(\n";
2671     OS << "    OperandVector &Operands,\n";
2672     OS << "    unsigned MCK);\n\n";
2673   }
2674
2675   OS << "#endif // GET_ASSEMBLER_HEADER_INFO\n\n";
2676
2677   // Emit the operand match diagnostic enum names.
2678   OS << "\n#ifdef GET_OPERAND_DIAGNOSTIC_TYPES\n";
2679   OS << "#undef GET_OPERAND_DIAGNOSTIC_TYPES\n\n";
2680   emitOperandDiagnosticTypes(Info, OS);
2681   OS << "#endif // GET_OPERAND_DIAGNOSTIC_TYPES\n\n";
2682
2683
2684   OS << "\n#ifdef GET_REGISTER_MATCHER\n";
2685   OS << "#undef GET_REGISTER_MATCHER\n\n";
2686
2687   // Emit the subtarget feature enumeration.
2688   emitSubtargetFeatureFlagEnumeration(Info, OS);
2689
2690   // Emit the function to match a register name to number.
2691   // This should be omitted for Mips target
2692   if (AsmParser->getValueAsBit("ShouldEmitMatchRegisterName"))
2693     emitMatchRegisterName(Target, AsmParser, OS);
2694
2695   OS << "#endif // GET_REGISTER_MATCHER\n\n";
2696
2697   OS << "\n#ifdef GET_SUBTARGET_FEATURE_NAME\n";
2698   OS << "#undef GET_SUBTARGET_FEATURE_NAME\n\n";
2699
2700   // Generate the helper function to get the names for subtarget features.
2701   emitGetSubtargetFeatureName(Info, OS);
2702
2703   OS << "#endif // GET_SUBTARGET_FEATURE_NAME\n\n";
2704
2705   OS << "\n#ifdef GET_MATCHER_IMPLEMENTATION\n";
2706   OS << "#undef GET_MATCHER_IMPLEMENTATION\n\n";
2707
2708   // Generate the function that remaps for mnemonic aliases.
2709   bool HasMnemonicAliases = emitMnemonicAliases(OS, Info, Target);
2710
2711   // Generate the convertToMCInst function to convert operands into an MCInst.
2712   // Also, generate the convertToMapAndConstraints function for MS-style inline
2713   // assembly.  The latter doesn't actually generate a MCInst.
2714   emitConvertFuncs(Target, ClassName, Info.Matchables, OS);
2715
2716   // Emit the enumeration for classes which participate in matching.
2717   emitMatchClassEnumeration(Target, Info.Classes, OS);
2718
2719   // Emit the routine to match token strings to their match class.
2720   emitMatchTokenString(Target, Info.Classes, OS);
2721
2722   // Emit the subclass predicate routine.
2723   emitIsSubclass(Target, Info.Classes, OS);
2724
2725   // Emit the routine to validate an operand against a match class.
2726   emitValidateOperandClass(Info, OS);
2727
2728   // Emit the available features compute function.
2729   emitComputeAvailableFeatures(Info, OS);
2730
2731
2732   StringToOffsetTable StringTable;
2733
2734   size_t MaxNumOperands = 0;
2735   unsigned MaxMnemonicIndex = 0;
2736   bool HasDeprecation = false;
2737   for (const auto &MI : Info.Matchables) {
2738     MaxNumOperands = std::max(MaxNumOperands, MI->AsmOperands.size());
2739     HasDeprecation |= MI->HasDeprecation;
2740
2741     // Store a pascal-style length byte in the mnemonic.
2742     std::string LenMnemonic = char(MI->Mnemonic.size()) + MI->Mnemonic.str();
2743     MaxMnemonicIndex = std::max(MaxMnemonicIndex,
2744                         StringTable.GetOrAddStringOffset(LenMnemonic, false));
2745   }
2746
2747   OS << "static const char *const MnemonicTable =\n";
2748   StringTable.EmitString(OS);
2749   OS << ";\n\n";
2750
2751   // Emit the static match table; unused classes get initalized to 0 which is
2752   // guaranteed to be InvalidMatchClass.
2753   //
2754   // FIXME: We can reduce the size of this table very easily. First, we change
2755   // it so that store the kinds in separate bit-fields for each index, which
2756   // only needs to be the max width used for classes at that index (we also need
2757   // to reject based on this during classification). If we then make sure to
2758   // order the match kinds appropriately (putting mnemonics last), then we
2759   // should only end up using a few bits for each class, especially the ones
2760   // following the mnemonic.
2761   OS << "namespace {\n";
2762   OS << "  struct MatchEntry {\n";
2763   OS << "    " << getMinimalTypeForRange(MaxMnemonicIndex)
2764                << " Mnemonic;\n";
2765   OS << "    uint16_t Opcode;\n";
2766   OS << "    " << getMinimalTypeForRange(Info.Matchables.size())
2767                << " ConvertFn;\n";
2768   OS << "    " << getMinimalRequiredFeaturesType(Info)
2769                << " RequiredFeatures;\n";
2770   OS << "    " << getMinimalTypeForRange(Info.Classes.size())
2771                << " Classes[" << MaxNumOperands << "];\n";
2772   OS << "    StringRef getMnemonic() const {\n";
2773   OS << "      return StringRef(MnemonicTable + Mnemonic + 1,\n";
2774   OS << "                       MnemonicTable[Mnemonic]);\n";
2775   OS << "    }\n";
2776   OS << "  };\n\n";
2777
2778   OS << "  // Predicate for searching for an opcode.\n";
2779   OS << "  struct LessOpcode {\n";
2780   OS << "    bool operator()(const MatchEntry &LHS, StringRef RHS) {\n";
2781   OS << "      return LHS.getMnemonic() < RHS;\n";
2782   OS << "    }\n";
2783   OS << "    bool operator()(StringRef LHS, const MatchEntry &RHS) {\n";
2784   OS << "      return LHS < RHS.getMnemonic();\n";
2785   OS << "    }\n";
2786   OS << "    bool operator()(const MatchEntry &LHS, const MatchEntry &RHS) {\n";
2787   OS << "      return LHS.getMnemonic() < RHS.getMnemonic();\n";
2788   OS << "    }\n";
2789   OS << "  };\n";
2790
2791   OS << "} // end anonymous namespace.\n\n";
2792
2793   unsigned VariantCount = Target.getAsmParserVariantCount();
2794   for (unsigned VC = 0; VC != VariantCount; ++VC) {
2795     Record *AsmVariant = Target.getAsmParserVariant(VC);
2796     int AsmVariantNo = AsmVariant->getValueAsInt("Variant");
2797
2798     OS << "static const MatchEntry MatchTable" << VC << "[] = {\n";
2799
2800     for (const auto &MI : Info.Matchables) {
2801       if (MI->AsmVariantID != AsmVariantNo)
2802         continue;
2803
2804       // Store a pascal-style length byte in the mnemonic.
2805       std::string LenMnemonic = char(MI->Mnemonic.size()) + MI->Mnemonic.str();
2806       OS << "  { " << StringTable.GetOrAddStringOffset(LenMnemonic, false)
2807          << " /* " << MI->Mnemonic << " */, "
2808          << Target.getName() << "::"
2809          << MI->getResultInst()->TheDef->getName() << ", "
2810          << MI->ConversionFnKind << ", ";
2811
2812       // Write the required features mask.
2813       if (!MI->RequiredFeatures.empty()) {
2814         for (unsigned i = 0, e = MI->RequiredFeatures.size(); i != e; ++i) {
2815           if (i) OS << "|";
2816           OS << MI->RequiredFeatures[i]->getEnumName();
2817         }
2818       } else
2819         OS << "0";
2820
2821       OS << ", { ";
2822       for (unsigned i = 0, e = MI->AsmOperands.size(); i != e; ++i) {
2823         const MatchableInfo::AsmOperand &Op = MI->AsmOperands[i];
2824
2825         if (i) OS << ", ";
2826         OS << Op.Class->Name;
2827       }
2828       OS << " }, },\n";
2829     }
2830
2831     OS << "};\n\n";
2832   }
2833
2834   // A method to determine if a mnemonic is in the list.
2835   OS << "bool " << Target.getName() << ClassName << "::\n"
2836      << "mnemonicIsValid(StringRef Mnemonic, unsigned VariantID) {\n";
2837   OS << "  // Find the appropriate table for this asm variant.\n";
2838   OS << "  const MatchEntry *Start, *End;\n";
2839   OS << "  switch (VariantID) {\n";
2840   OS << "  default: // unreachable\n";
2841   for (unsigned VC = 0; VC != VariantCount; ++VC) {
2842     Record *AsmVariant = Target.getAsmParserVariant(VC);
2843     int AsmVariantNo = AsmVariant->getValueAsInt("Variant");
2844     OS << "  case " << AsmVariantNo << ": Start = std::begin(MatchTable" << VC
2845        << "); End = std::end(MatchTable" << VC << "); break;\n";
2846   }
2847   OS << "  }\n";
2848   OS << "  // Search the table.\n";
2849   OS << "  std::pair<const MatchEntry*, const MatchEntry*> MnemonicRange =\n";
2850   OS << "    std::equal_range(Start, End, Mnemonic, LessOpcode());\n";
2851   OS << "  return MnemonicRange.first != MnemonicRange.second;\n";
2852   OS << "}\n\n";
2853
2854   // Finally, build the match function.
2855   OS << "unsigned " << Target.getName() << ClassName << "::\n"
2856      << "MatchInstructionImpl(const OperandVector"
2857      << " &Operands,\n";
2858   OS << "                     MCInst &Inst,\n"
2859      << "uint64_t &ErrorInfo, bool matchingInlineAsm, unsigned VariantID) {\n";
2860
2861   OS << "  // Eliminate obvious mismatches.\n";
2862   OS << "  if (Operands.size() > " << (MaxNumOperands+1) << ") {\n";
2863   OS << "    ErrorInfo = " << (MaxNumOperands+1) << ";\n";
2864   OS << "    return Match_InvalidOperand;\n";
2865   OS << "  }\n\n";
2866
2867   // Emit code to get the available features.
2868   OS << "  // Get the current feature set.\n";
2869   OS << "  uint64_t AvailableFeatures = getAvailableFeatures();\n\n";
2870
2871   OS << "  // Get the instruction mnemonic, which is the first token.\n";
2872   OS << "  StringRef Mnemonic = ((" << Target.getName()
2873      << "Operand&)*Operands[0]).getToken();\n\n";
2874
2875   if (HasMnemonicAliases) {
2876     OS << "  // Process all MnemonicAliases to remap the mnemonic.\n";
2877     OS << "  applyMnemonicAliases(Mnemonic, AvailableFeatures, VariantID);\n\n";
2878   }
2879
2880   // Emit code to compute the class list for this operand vector.
2881   OS << "  // Some state to try to produce better error messages.\n";
2882   OS << "  bool HadMatchOtherThanFeatures = false;\n";
2883   OS << "  bool HadMatchOtherThanPredicate = false;\n";
2884   OS << "  unsigned RetCode = Match_InvalidOperand;\n";
2885   OS << "  uint64_t MissingFeatures = ~0ULL;\n";
2886   OS << "  // Set ErrorInfo to the operand that mismatches if it is\n";
2887   OS << "  // wrong for all instances of the instruction.\n";
2888   OS << "  ErrorInfo = ~0U;\n";
2889
2890   // Emit code to search the table.
2891   OS << "  // Find the appropriate table for this asm variant.\n";
2892   OS << "  const MatchEntry *Start, *End;\n";
2893   OS << "  switch (VariantID) {\n";
2894   OS << "  default: // unreachable\n";
2895   for (unsigned VC = 0; VC != VariantCount; ++VC) {
2896     Record *AsmVariant = Target.getAsmParserVariant(VC);
2897     int AsmVariantNo = AsmVariant->getValueAsInt("Variant");
2898     OS << "  case " << AsmVariantNo << ": Start = std::begin(MatchTable" << VC
2899        << "); End = std::end(MatchTable" << VC << "); break;\n";
2900   }
2901   OS << "  }\n";
2902   OS << "  // Search the table.\n";
2903   OS << "  std::pair<const MatchEntry*, const MatchEntry*> MnemonicRange =\n";
2904   OS << "    std::equal_range(Start, End, Mnemonic, LessOpcode());\n\n";
2905
2906   OS << "  // Return a more specific error code if no mnemonics match.\n";
2907   OS << "  if (MnemonicRange.first == MnemonicRange.second)\n";
2908   OS << "    return Match_MnemonicFail;\n\n";
2909
2910   OS << "  for (const MatchEntry *it = MnemonicRange.first, "
2911      << "*ie = MnemonicRange.second;\n";
2912   OS << "       it != ie; ++it) {\n";
2913
2914   OS << "    // equal_range guarantees that instruction mnemonic matches.\n";
2915   OS << "    assert(Mnemonic == it->getMnemonic());\n";
2916
2917   // Emit check that the subclasses match.
2918   OS << "    bool OperandsValid = true;\n";
2919   OS << "    for (unsigned i = 0; i != " << MaxNumOperands << "; ++i) {\n";
2920   OS << "      if (i + 1 >= Operands.size()) {\n";
2921   OS << "        OperandsValid = (it->Classes[i] == " <<"InvalidMatchClass);\n";
2922   OS << "        if (!OperandsValid) ErrorInfo = i + 1;\n";
2923   OS << "        break;\n";
2924   OS << "      }\n";
2925   OS << "      unsigned Diag = validateOperandClass(*Operands[i+1],\n";
2926   OS.indent(43);
2927   OS << "(MatchClassKind)it->Classes[i]);\n";
2928   OS << "      if (Diag == Match_Success)\n";
2929   OS << "        continue;\n";
2930   OS << "      // If the generic handler indicates an invalid operand\n";
2931   OS << "      // failure, check for a special case.\n";
2932   OS << "      if (Diag == Match_InvalidOperand) {\n";
2933   OS << "        Diag = validateTargetOperandClass(*Operands[i+1],\n";
2934   OS.indent(43);
2935   OS << "(MatchClassKind)it->Classes[i]);\n";
2936   OS << "        if (Diag == Match_Success)\n";
2937   OS << "          continue;\n";
2938   OS << "      }\n";
2939   OS << "      // If this operand is broken for all of the instances of this\n";
2940   OS << "      // mnemonic, keep track of it so we can report loc info.\n";
2941   OS << "      // If we already had a match that only failed due to a\n";
2942   OS << "      // target predicate, that diagnostic is preferred.\n";
2943   OS << "      if (!HadMatchOtherThanPredicate &&\n";
2944   OS << "          (it == MnemonicRange.first || ErrorInfo <= i+1)) {\n";
2945   OS << "        ErrorInfo = i+1;\n";
2946   OS << "        // InvalidOperand is the default. Prefer specificity.\n";
2947   OS << "        if (Diag != Match_InvalidOperand)\n";
2948   OS << "          RetCode = Diag;\n";
2949   OS << "      }\n";
2950   OS << "      // Otherwise, just reject this instance of the mnemonic.\n";
2951   OS << "      OperandsValid = false;\n";
2952   OS << "      break;\n";
2953   OS << "    }\n\n";
2954
2955   OS << "    if (!OperandsValid) continue;\n";
2956
2957   // Emit check that the required features are available.
2958   OS << "    if ((AvailableFeatures & it->RequiredFeatures) "
2959      << "!= it->RequiredFeatures) {\n";
2960   OS << "      HadMatchOtherThanFeatures = true;\n";
2961   OS << "      uint64_t NewMissingFeatures = it->RequiredFeatures & "
2962         "~AvailableFeatures;\n";
2963   OS << "      if (CountPopulation_64(NewMissingFeatures) <=\n"
2964         "          CountPopulation_64(MissingFeatures))\n";
2965   OS << "        MissingFeatures = NewMissingFeatures;\n";
2966   OS << "      continue;\n";
2967   OS << "    }\n";
2968   OS << "\n";
2969   OS << "    if (matchingInlineAsm) {\n";
2970   OS << "      Inst.setOpcode(it->Opcode);\n";
2971   OS << "      convertToMapAndConstraints(it->ConvertFn, Operands);\n";
2972   OS << "      return Match_Success;\n";
2973   OS << "    }\n\n";
2974   OS << "    // We have selected a definite instruction, convert the parsed\n"
2975      << "    // operands into the appropriate MCInst.\n";
2976   OS << "    convertToMCInst(it->ConvertFn, Inst, it->Opcode, Operands);\n";
2977   OS << "\n";
2978
2979   // Verify the instruction with the target-specific match predicate function.
2980   OS << "    // We have a potential match. Check the target predicate to\n"
2981      << "    // handle any context sensitive constraints.\n"
2982      << "    unsigned MatchResult;\n"
2983      << "    if ((MatchResult = checkTargetMatchPredicate(Inst)) !="
2984      << " Match_Success) {\n"
2985      << "      Inst.clear();\n"
2986      << "      RetCode = MatchResult;\n"
2987      << "      HadMatchOtherThanPredicate = true;\n"
2988      << "      continue;\n"
2989      << "    }\n\n";
2990
2991   // Call the post-processing function, if used.
2992   std::string InsnCleanupFn =
2993     AsmParser->getValueAsString("AsmParserInstCleanup");
2994   if (!InsnCleanupFn.empty())
2995     OS << "    " << InsnCleanupFn << "(Inst);\n";
2996
2997   if (HasDeprecation) {
2998     OS << "    std::string Info;\n";
2999     OS << "    if (MII.get(Inst.getOpcode()).getDeprecatedInfo(Inst, STI, Info)) {\n";
3000     OS << "      SMLoc Loc = ((" << Target.getName()
3001        << "Operand&)*Operands[0]).getStartLoc();\n";
3002     OS << "      getParser().Warning(Loc, Info, None);\n";
3003     OS << "    }\n";
3004   }
3005
3006   OS << "    return Match_Success;\n";
3007   OS << "  }\n\n";
3008
3009   OS << "  // Okay, we had no match.  Try to return a useful error code.\n";
3010   OS << "  if (HadMatchOtherThanPredicate || !HadMatchOtherThanFeatures)\n";
3011   OS << "    return RetCode;\n\n";
3012   OS << "  // Missing feature matches return which features were missing\n";
3013   OS << "  ErrorInfo = MissingFeatures;\n";
3014   OS << "  return Match_MissingFeature;\n";
3015   OS << "}\n\n";
3016
3017   if (Info.OperandMatchInfo.size())
3018     emitCustomOperandParsing(OS, Target, Info, ClassName, StringTable,
3019                              MaxMnemonicIndex);
3020
3021   OS << "#endif // GET_MATCHER_IMPLEMENTATION\n\n";
3022 }
3023
3024 namespace llvm {
3025
3026 void EmitAsmMatcher(RecordKeeper &RK, raw_ostream &OS) {
3027   emitSourceFileHeader("Assembly Matcher Source Fragment", OS);
3028   AsmMatcherEmitter(RK).run(OS);
3029 }
3030
3031 } // End llvm namespace