xref: /freebsd/contrib/llvm-project/llvm/utils/TableGen/DAGISelMatcherEmitter.cpp (revision 7fdf597e96a02165cfe22ff357b857d5fa15ed8a)
1 //===- DAGISelMatcherEmitter.cpp - Matcher Emitter ------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains code to generate C++ code for a matcher.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "Basic/SDNodeProperties.h"
14 #include "Common/CodeGenDAGPatterns.h"
15 #include "Common/CodeGenInstruction.h"
16 #include "Common/CodeGenRegisters.h"
17 #include "Common/CodeGenTarget.h"
18 #include "Common/DAGISelMatcher.h"
19 #include "llvm/ADT/DenseMap.h"
20 #include "llvm/ADT/MapVector.h"
21 #include "llvm/ADT/StringMap.h"
22 #include "llvm/ADT/TinyPtrVector.h"
23 #include "llvm/Support/CommandLine.h"
24 #include "llvm/Support/Format.h"
25 #include "llvm/Support/SourceMgr.h"
26 #include "llvm/TableGen/Error.h"
27 #include "llvm/TableGen/Record.h"
28 
29 using namespace llvm;
30 
31 enum {
32   IndexWidth = 6,
33   FullIndexWidth = IndexWidth + 4,
34   HistOpcWidth = 40,
35 };
36 
37 cl::OptionCategory DAGISelCat("Options for -gen-dag-isel");
38 
39 // To reduce generated source code size.
40 static cl::opt<bool> OmitComments("omit-comments",
41                                   cl::desc("Do not generate comments"),
42                                   cl::init(false), cl::cat(DAGISelCat));
43 
44 static cl::opt<bool> InstrumentCoverage(
45     "instrument-coverage",
46     cl::desc("Generates tables to help identify patterns matched"),
47     cl::init(false), cl::cat(DAGISelCat));
48 
49 namespace {
50 class MatcherTableEmitter {
51   const CodeGenDAGPatterns &CGP;
52 
53   SmallVector<unsigned, Matcher::HighestKind + 1> OpcodeCounts;
54 
55   std::vector<TreePattern *> NodePredicates;
56   std::vector<TreePattern *> NodePredicatesWithOperands;
57 
58   // We de-duplicate the predicates by code string, and use this map to track
59   // all the patterns with "identical" predicates.
60   MapVector<std::string, TinyPtrVector<TreePattern *>, StringMap<unsigned>>
61       NodePredicatesByCodeToRun;
62 
63   std::vector<std::string> PatternPredicates;
64 
65   std::vector<const ComplexPattern *> ComplexPatterns;
66 
67   DenseMap<Record *, unsigned> NodeXFormMap;
68   std::vector<Record *> NodeXForms;
69 
70   std::vector<std::string> VecIncludeStrings;
71   MapVector<std::string, unsigned, StringMap<unsigned>> VecPatterns;
72 
73   unsigned getPatternIdxFromTable(std::string &&P, std::string &&include_loc) {
74     const auto It = VecPatterns.find(P);
75     if (It == VecPatterns.end()) {
76       VecPatterns.insert(std::pair(std::move(P), VecPatterns.size()));
77       VecIncludeStrings.push_back(std::move(include_loc));
78       return VecIncludeStrings.size() - 1;
79     }
80     return It->second;
81   }
82 
83 public:
84   MatcherTableEmitter(const Matcher *TheMatcher, const CodeGenDAGPatterns &cgp)
85       : CGP(cgp), OpcodeCounts(Matcher::HighestKind + 1, 0) {
86     // Record the usage of ComplexPattern.
87     MapVector<const ComplexPattern *, unsigned> ComplexPatternUsage;
88     // Record the usage of PatternPredicate.
89     MapVector<StringRef, unsigned> PatternPredicateUsage;
90     // Record the usage of Predicate.
91     MapVector<TreePattern *, unsigned> PredicateUsage;
92 
93     // Iterate the whole MatcherTable once and do some statistics.
94     std::function<void(const Matcher *)> Statistic = [&](const Matcher *N) {
95       while (N) {
96         if (auto *SM = dyn_cast<ScopeMatcher>(N))
97           for (unsigned I = 0; I < SM->getNumChildren(); I++)
98             Statistic(SM->getChild(I));
99         else if (auto *SOM = dyn_cast<SwitchOpcodeMatcher>(N))
100           for (unsigned I = 0; I < SOM->getNumCases(); I++)
101             Statistic(SOM->getCaseMatcher(I));
102         else if (auto *STM = dyn_cast<SwitchTypeMatcher>(N))
103           for (unsigned I = 0; I < STM->getNumCases(); I++)
104             Statistic(STM->getCaseMatcher(I));
105         else if (auto *CPM = dyn_cast<CheckComplexPatMatcher>(N))
106           ++ComplexPatternUsage[&CPM->getPattern()];
107         else if (auto *CPPM = dyn_cast<CheckPatternPredicateMatcher>(N))
108           ++PatternPredicateUsage[CPPM->getPredicate()];
109         else if (auto *PM = dyn_cast<CheckPredicateMatcher>(N))
110           ++PredicateUsage[PM->getPredicate().getOrigPatFragRecord()];
111         N = N->getNext();
112       }
113     };
114     Statistic(TheMatcher);
115 
116     // Sort ComplexPatterns by usage.
117     std::vector<std::pair<const ComplexPattern *, unsigned>> ComplexPatternList(
118         ComplexPatternUsage.begin(), ComplexPatternUsage.end());
119     stable_sort(ComplexPatternList, [](const auto &A, const auto &B) {
120       return A.second > B.second;
121     });
122     for (const auto &ComplexPattern : ComplexPatternList)
123       ComplexPatterns.push_back(ComplexPattern.first);
124 
125     // Sort PatternPredicates by usage.
126     std::vector<std::pair<std::string, unsigned>> PatternPredicateList(
127         PatternPredicateUsage.begin(), PatternPredicateUsage.end());
128     stable_sort(PatternPredicateList, [](const auto &A, const auto &B) {
129       return A.second > B.second;
130     });
131     for (const auto &PatternPredicate : PatternPredicateList)
132       PatternPredicates.push_back(PatternPredicate.first);
133 
134     // Sort Predicates by usage.
135     // Merge predicates with same code.
136     for (const auto &Usage : PredicateUsage) {
137       TreePattern *TP = Usage.first;
138       TreePredicateFn Pred(TP);
139       NodePredicatesByCodeToRun[Pred.getCodeToRunOnSDNode()].push_back(TP);
140     }
141 
142     std::vector<std::pair<TreePattern *, unsigned>> PredicateList;
143     // Sum the usage.
144     for (auto &Predicate : NodePredicatesByCodeToRun) {
145       TinyPtrVector<TreePattern *> &TPs = Predicate.second;
146       stable_sort(TPs, [](const auto *A, const auto *B) {
147         return A->getRecord()->getName() < B->getRecord()->getName();
148       });
149       unsigned Uses = 0;
150       for (TreePattern *TP : TPs)
151         Uses += PredicateUsage[TP];
152 
153       // We only add the first predicate here since they are with the same code.
154       PredicateList.push_back({TPs[0], Uses});
155     }
156 
157     stable_sort(PredicateList, [](const auto &A, const auto &B) {
158       return A.second > B.second;
159     });
160     for (const auto &Predicate : PredicateList) {
161       TreePattern *TP = Predicate.first;
162       if (TreePredicateFn(TP).usesOperands())
163         NodePredicatesWithOperands.push_back(TP);
164       else
165         NodePredicates.push_back(TP);
166     }
167   }
168 
169   unsigned EmitMatcherList(const Matcher *N, const unsigned Indent,
170                            unsigned StartIdx, raw_ostream &OS);
171 
172   unsigned SizeMatcherList(Matcher *N, raw_ostream &OS);
173 
174   void EmitPredicateFunctions(raw_ostream &OS);
175 
176   void EmitHistogram(const Matcher *N, raw_ostream &OS);
177 
178   void EmitPatternMatchTable(raw_ostream &OS);
179 
180 private:
181   void EmitNodePredicatesFunction(const std::vector<TreePattern *> &Preds,
182                                   StringRef Decl, raw_ostream &OS);
183 
184   unsigned SizeMatcher(Matcher *N, raw_ostream &OS);
185 
186   unsigned EmitMatcher(const Matcher *N, const unsigned Indent,
187                        unsigned CurrentIdx, raw_ostream &OS);
188 
189   unsigned getNodePredicate(TreePredicateFn Pred) {
190     // We use the first predicate.
191     TreePattern *PredPat =
192         NodePredicatesByCodeToRun[Pred.getCodeToRunOnSDNode()][0];
193     return Pred.usesOperands()
194                ? llvm::find(NodePredicatesWithOperands, PredPat) -
195                      NodePredicatesWithOperands.begin()
196                : llvm::find(NodePredicates, PredPat) - NodePredicates.begin();
197   }
198 
199   unsigned getPatternPredicate(StringRef PredName) {
200     return llvm::find(PatternPredicates, PredName) - PatternPredicates.begin();
201   }
202   unsigned getComplexPat(const ComplexPattern &P) {
203     return llvm::find(ComplexPatterns, &P) - ComplexPatterns.begin();
204   }
205 
206   unsigned getNodeXFormID(Record *Rec) {
207     unsigned &Entry = NodeXFormMap[Rec];
208     if (Entry == 0) {
209       NodeXForms.push_back(Rec);
210       Entry = NodeXForms.size();
211     }
212     return Entry - 1;
213   }
214 };
215 } // end anonymous namespace.
216 
217 static std::string GetPatFromTreePatternNode(const TreePatternNode &N) {
218   std::string str;
219   raw_string_ostream Stream(str);
220   Stream << N;
221   return str;
222 }
223 
224 static unsigned GetVBRSize(unsigned Val) {
225   if (Val <= 127)
226     return 1;
227 
228   unsigned NumBytes = 0;
229   while (Val >= 128) {
230     Val >>= 7;
231     ++NumBytes;
232   }
233   return NumBytes + 1;
234 }
235 
236 /// EmitVBRValue - Emit the specified value as a VBR, returning the number of
237 /// bytes emitted.
238 static unsigned EmitVBRValue(uint64_t Val, raw_ostream &OS) {
239   if (Val <= 127) {
240     OS << Val << ", ";
241     return 1;
242   }
243 
244   uint64_t InVal = Val;
245   unsigned NumBytes = 0;
246   while (Val >= 128) {
247     OS << (Val & 127) << "|128,";
248     Val >>= 7;
249     ++NumBytes;
250   }
251   OS << Val;
252   if (!OmitComments)
253     OS << "/*" << InVal << "*/";
254   OS << ", ";
255   return NumBytes + 1;
256 }
257 
258 /// Emit the specified signed value as a VBR. To improve compression we encode
259 /// positive numbers shifted left by 1 and negative numbers negated and shifted
260 /// left by 1 with bit 0 set.
261 static unsigned EmitSignedVBRValue(uint64_t Val, raw_ostream &OS) {
262   if ((int64_t)Val >= 0)
263     Val = Val << 1;
264   else
265     Val = (-Val << 1) | 1;
266 
267   return EmitVBRValue(Val, OS);
268 }
269 
270 // This is expensive and slow.
271 static std::string getIncludePath(const Record *R) {
272   std::string str;
273   raw_string_ostream Stream(str);
274   auto Locs = R->getLoc();
275   SMLoc L;
276   if (Locs.size() > 1) {
277     // Get where the pattern prototype was instantiated
278     L = Locs[1];
279   } else if (Locs.size() == 1) {
280     L = Locs[0];
281   }
282   unsigned CurBuf = SrcMgr.FindBufferContainingLoc(L);
283   assert(CurBuf && "Invalid or unspecified location!");
284 
285   Stream << SrcMgr.getBufferInfo(CurBuf).Buffer->getBufferIdentifier() << ":"
286          << SrcMgr.FindLineNumber(L, CurBuf);
287   return str;
288 }
289 
290 /// This function traverses the matcher tree and sizes all the nodes
291 /// that are children of the three kinds of nodes that have them.
292 unsigned MatcherTableEmitter::SizeMatcherList(Matcher *N, raw_ostream &OS) {
293   unsigned Size = 0;
294   while (N) {
295     Size += SizeMatcher(N, OS);
296     N = N->getNext();
297   }
298   return Size;
299 }
300 
301 /// This function sizes the children of the three kinds of nodes that
302 /// have them. It does so by using special cases for those three
303 /// nodes, but sharing the code in EmitMatcher() for the other kinds.
304 unsigned MatcherTableEmitter::SizeMatcher(Matcher *N, raw_ostream &OS) {
305   unsigned Idx = 0;
306 
307   ++OpcodeCounts[N->getKind()];
308   switch (N->getKind()) {
309   // The Scope matcher has its kind, a series of child size + child,
310   // and a trailing zero.
311   case Matcher::Scope: {
312     ScopeMatcher *SM = cast<ScopeMatcher>(N);
313     assert(SM->getNext() == nullptr && "Scope matcher should not have next");
314     unsigned Size = 1; // Count the kind.
315     for (unsigned i = 0, e = SM->getNumChildren(); i != e; ++i) {
316       const unsigned ChildSize = SizeMatcherList(SM->getChild(i), OS);
317       assert(ChildSize != 0 && "Matcher cannot have child of size 0");
318       SM->getChild(i)->setSize(ChildSize);
319       Size += GetVBRSize(ChildSize) + ChildSize; // Count VBR and child size.
320     }
321     ++Size; // Count the zero sentinel.
322     return Size;
323   }
324 
325   // SwitchOpcode and SwitchType have their kind, a series of child size +
326   // opcode/type + child, and a trailing zero.
327   case Matcher::SwitchOpcode:
328   case Matcher::SwitchType: {
329     unsigned Size = 1; // Count the kind.
330     unsigned NumCases;
331     if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(N))
332       NumCases = SOM->getNumCases();
333     else
334       NumCases = cast<SwitchTypeMatcher>(N)->getNumCases();
335     for (unsigned i = 0, e = NumCases; i != e; ++i) {
336       Matcher *Child;
337       if (SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(N)) {
338         Child = SOM->getCaseMatcher(i);
339         Size += 2; // Count the child's opcode.
340       } else {
341         Child = cast<SwitchTypeMatcher>(N)->getCaseMatcher(i);
342         ++Size; // Count the child's type.
343       }
344       const unsigned ChildSize = SizeMatcherList(Child, OS);
345       assert(ChildSize != 0 && "Matcher cannot have child of size 0");
346       Child->setSize(ChildSize);
347       Size += GetVBRSize(ChildSize) + ChildSize; // Count VBR and child size.
348     }
349     ++Size; // Count the zero sentinel.
350     return Size;
351   }
352 
353   default:
354     // Employ the matcher emitter to size other matchers.
355     return EmitMatcher(N, 0, Idx, OS);
356   }
357   llvm_unreachable("Unreachable");
358 }
359 
360 static void BeginEmitFunction(raw_ostream &OS, StringRef RetType,
361                               StringRef Decl, bool AddOverride) {
362   OS << "#ifdef GET_DAGISEL_DECL\n";
363   OS << RetType << ' ' << Decl;
364   if (AddOverride)
365     OS << " override";
366   OS << ";\n"
367         "#endif\n"
368         "#if defined(GET_DAGISEL_BODY) || DAGISEL_INLINE\n";
369   OS << RetType << " DAGISEL_CLASS_COLONCOLON " << Decl << "\n";
370   if (AddOverride) {
371     OS << "#if DAGISEL_INLINE\n"
372           "  override\n"
373           "#endif\n";
374   }
375 }
376 
377 static void EndEmitFunction(raw_ostream &OS) {
378   OS << "#endif // GET_DAGISEL_BODY\n\n";
379 }
380 
381 void MatcherTableEmitter::EmitPatternMatchTable(raw_ostream &OS) {
382 
383   assert(isUInt<16>(VecPatterns.size()) &&
384          "Using only 16 bits to encode offset into Pattern Table");
385   assert(VecPatterns.size() == VecIncludeStrings.size() &&
386          "The sizes of Pattern and include vectors should be the same");
387 
388   BeginEmitFunction(OS, "StringRef", "getPatternForIndex(unsigned Index)",
389                     true /*AddOverride*/);
390   OS << "{\n";
391   OS << "static const char *PATTERN_MATCH_TABLE[] = {\n";
392 
393   for (const auto &It : VecPatterns) {
394     OS << "\"" << It.first << "\",\n";
395   }
396 
397   OS << "\n};";
398   OS << "\nreturn StringRef(PATTERN_MATCH_TABLE[Index]);";
399   OS << "\n}\n";
400   EndEmitFunction(OS);
401 
402   BeginEmitFunction(OS, "StringRef", "getIncludePathForIndex(unsigned Index)",
403                     true /*AddOverride*/);
404   OS << "{\n";
405   OS << "static const char *INCLUDE_PATH_TABLE[] = {\n";
406 
407   for (const auto &It : VecIncludeStrings) {
408     OS << "\"" << It << "\",\n";
409   }
410 
411   OS << "\n};";
412   OS << "\nreturn StringRef(INCLUDE_PATH_TABLE[Index]);";
413   OS << "\n}\n";
414   EndEmitFunction(OS);
415 }
416 
417 /// EmitMatcher - Emit bytes for the specified matcher and return
418 /// the number of bytes emitted.
419 unsigned MatcherTableEmitter::EmitMatcher(const Matcher *N,
420                                           const unsigned Indent,
421                                           unsigned CurrentIdx,
422                                           raw_ostream &OS) {
423   OS.indent(Indent);
424 
425   switch (N->getKind()) {
426   case Matcher::Scope: {
427     const ScopeMatcher *SM = cast<ScopeMatcher>(N);
428     unsigned StartIdx = CurrentIdx;
429 
430     // Emit all of the children.
431     for (unsigned i = 0, e = SM->getNumChildren(); i != e; ++i) {
432       if (i == 0) {
433         OS << "OPC_Scope, ";
434         ++CurrentIdx;
435       } else {
436         if (!OmitComments) {
437           OS << "/*" << format_decimal(CurrentIdx, IndexWidth) << "*/";
438           OS.indent(Indent) << "/*Scope*/ ";
439         } else
440           OS.indent(Indent);
441       }
442 
443       unsigned ChildSize = SM->getChild(i)->getSize();
444       unsigned VBRSize = EmitVBRValue(ChildSize, OS);
445       if (!OmitComments) {
446         OS << "/*->" << CurrentIdx + VBRSize + ChildSize << "*/";
447         if (i == 0)
448           OS << " // " << SM->getNumChildren() << " children in Scope";
449       }
450       OS << '\n';
451 
452       ChildSize = EmitMatcherList(SM->getChild(i), Indent + 1,
453                                   CurrentIdx + VBRSize, OS);
454       assert(ChildSize == SM->getChild(i)->getSize() &&
455              "Emitted child size does not match calculated size");
456       CurrentIdx += VBRSize + ChildSize;
457     }
458 
459     // Emit a zero as a sentinel indicating end of 'Scope'.
460     if (!OmitComments)
461       OS << "/*" << format_decimal(CurrentIdx, IndexWidth) << "*/";
462     OS.indent(Indent) << "0, ";
463     if (!OmitComments)
464       OS << "/*End of Scope*/";
465     OS << '\n';
466     return CurrentIdx - StartIdx + 1;
467   }
468 
469   case Matcher::RecordNode:
470     OS << "OPC_RecordNode,";
471     if (!OmitComments)
472       OS << " // #" << cast<RecordMatcher>(N)->getResultNo() << " = "
473          << cast<RecordMatcher>(N)->getWhatFor();
474     OS << '\n';
475     return 1;
476 
477   case Matcher::RecordChild:
478     OS << "OPC_RecordChild" << cast<RecordChildMatcher>(N)->getChildNo() << ',';
479     if (!OmitComments)
480       OS << " // #" << cast<RecordChildMatcher>(N)->getResultNo() << " = "
481          << cast<RecordChildMatcher>(N)->getWhatFor();
482     OS << '\n';
483     return 1;
484 
485   case Matcher::RecordMemRef:
486     OS << "OPC_RecordMemRef,\n";
487     return 1;
488 
489   case Matcher::CaptureGlueInput:
490     OS << "OPC_CaptureGlueInput,\n";
491     return 1;
492 
493   case Matcher::MoveChild: {
494     const auto *MCM = cast<MoveChildMatcher>(N);
495 
496     OS << "OPC_MoveChild";
497     // Handle the specialized forms.
498     if (MCM->getChildNo() >= 8)
499       OS << ", ";
500     OS << MCM->getChildNo() << ",\n";
501     return (MCM->getChildNo() >= 8) ? 2 : 1;
502   }
503 
504   case Matcher::MoveSibling: {
505     const auto *MSM = cast<MoveSiblingMatcher>(N);
506 
507     OS << "OPC_MoveSibling";
508     // Handle the specialized forms.
509     if (MSM->getSiblingNo() >= 8)
510       OS << ", ";
511     OS << MSM->getSiblingNo() << ",\n";
512     return (MSM->getSiblingNo() >= 8) ? 2 : 1;
513   }
514 
515   case Matcher::MoveParent:
516     OS << "OPC_MoveParent,\n";
517     return 1;
518 
519   case Matcher::CheckSame:
520     OS << "OPC_CheckSame, " << cast<CheckSameMatcher>(N)->getMatchNumber()
521        << ",\n";
522     return 2;
523 
524   case Matcher::CheckChildSame:
525     OS << "OPC_CheckChild" << cast<CheckChildSameMatcher>(N)->getChildNo()
526        << "Same, " << cast<CheckChildSameMatcher>(N)->getMatchNumber() << ",\n";
527     return 2;
528 
529   case Matcher::CheckPatternPredicate: {
530     StringRef Pred = cast<CheckPatternPredicateMatcher>(N)->getPredicate();
531     unsigned PredNo = getPatternPredicate(Pred);
532     if (PredNo > 255)
533       OS << "OPC_CheckPatternPredicateTwoByte, TARGET_VAL(" << PredNo << "),";
534     else if (PredNo < 8)
535       OS << "OPC_CheckPatternPredicate" << PredNo << ',';
536     else
537       OS << "OPC_CheckPatternPredicate, " << PredNo << ',';
538     if (!OmitComments)
539       OS << " // " << Pred;
540     OS << '\n';
541     return 2 + (PredNo > 255) - (PredNo < 8);
542   }
543   case Matcher::CheckPredicate: {
544     TreePredicateFn Pred = cast<CheckPredicateMatcher>(N)->getPredicate();
545     unsigned OperandBytes = 0;
546     unsigned PredNo = getNodePredicate(Pred);
547 
548     if (Pred.usesOperands()) {
549       unsigned NumOps = cast<CheckPredicateMatcher>(N)->getNumOperands();
550       OS << "OPC_CheckPredicateWithOperands, " << NumOps << "/*#Ops*/, ";
551       for (unsigned i = 0; i < NumOps; ++i)
552         OS << cast<CheckPredicateMatcher>(N)->getOperandNo(i) << ", ";
553       OperandBytes = 1 + NumOps;
554     } else {
555       if (PredNo < 8) {
556         OperandBytes = -1;
557         OS << "OPC_CheckPredicate" << PredNo << ", ";
558       } else
559         OS << "OPC_CheckPredicate, ";
560     }
561 
562     if (PredNo >= 8 || Pred.usesOperands())
563       OS << PredNo << ',';
564     if (!OmitComments)
565       OS << " // " << Pred.getFnName();
566     OS << '\n';
567     return 2 + OperandBytes;
568   }
569 
570   case Matcher::CheckOpcode:
571     OS << "OPC_CheckOpcode, TARGET_VAL("
572        << cast<CheckOpcodeMatcher>(N)->getOpcode().getEnumName() << "),\n";
573     return 3;
574 
575   case Matcher::SwitchOpcode:
576   case Matcher::SwitchType: {
577     unsigned StartIdx = CurrentIdx;
578 
579     unsigned NumCases;
580     if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(N)) {
581       OS << "OPC_SwitchOpcode ";
582       NumCases = SOM->getNumCases();
583     } else {
584       OS << "OPC_SwitchType ";
585       NumCases = cast<SwitchTypeMatcher>(N)->getNumCases();
586     }
587 
588     if (!OmitComments)
589       OS << "/*" << NumCases << " cases */";
590     OS << ", ";
591     ++CurrentIdx;
592 
593     // For each case we emit the size, then the opcode, then the matcher.
594     for (unsigned i = 0, e = NumCases; i != e; ++i) {
595       const Matcher *Child;
596       unsigned IdxSize;
597       if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(N)) {
598         Child = SOM->getCaseMatcher(i);
599         IdxSize = 2; // size of opcode in table is 2 bytes.
600       } else {
601         Child = cast<SwitchTypeMatcher>(N)->getCaseMatcher(i);
602         IdxSize = 1; // size of type in table is 1 byte.
603       }
604 
605       if (i != 0) {
606         if (!OmitComments)
607           OS << "/*" << format_decimal(CurrentIdx, IndexWidth) << "*/";
608         OS.indent(Indent);
609         if (!OmitComments)
610           OS << (isa<SwitchOpcodeMatcher>(N) ? "/*SwitchOpcode*/ "
611                                              : "/*SwitchType*/ ");
612       }
613 
614       unsigned ChildSize = Child->getSize();
615       CurrentIdx += EmitVBRValue(ChildSize, OS) + IdxSize;
616       if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(N))
617         OS << "TARGET_VAL(" << SOM->getCaseOpcode(i).getEnumName() << "),";
618       else
619         OS << getEnumName(cast<SwitchTypeMatcher>(N)->getCaseType(i)) << ',';
620       if (!OmitComments)
621         OS << "// ->" << CurrentIdx + ChildSize;
622       OS << '\n';
623 
624       ChildSize = EmitMatcherList(Child, Indent + 1, CurrentIdx, OS);
625       assert(ChildSize == Child->getSize() &&
626              "Emitted child size does not match calculated size");
627       CurrentIdx += ChildSize;
628     }
629 
630     // Emit the final zero to terminate the switch.
631     if (!OmitComments)
632       OS << "/*" << format_decimal(CurrentIdx, IndexWidth) << "*/";
633     OS.indent(Indent) << "0,";
634     if (!OmitComments)
635       OS << (isa<SwitchOpcodeMatcher>(N) ? " // EndSwitchOpcode"
636                                          : " // EndSwitchType");
637 
638     OS << '\n';
639     return CurrentIdx - StartIdx + 1;
640   }
641 
642   case Matcher::CheckType:
643     if (cast<CheckTypeMatcher>(N)->getResNo() == 0) {
644       MVT::SimpleValueType VT = cast<CheckTypeMatcher>(N)->getType();
645       switch (VT) {
646       case MVT::i32:
647       case MVT::i64:
648         OS << "OPC_CheckTypeI" << MVT(VT).getSizeInBits() << ",\n";
649         return 1;
650       default:
651         OS << "OPC_CheckType, " << getEnumName(VT) << ",\n";
652         return 2;
653       }
654     }
655     OS << "OPC_CheckTypeRes, " << cast<CheckTypeMatcher>(N)->getResNo() << ", "
656        << getEnumName(cast<CheckTypeMatcher>(N)->getType()) << ",\n";
657     return 3;
658 
659   case Matcher::CheckChildType: {
660     MVT::SimpleValueType VT = cast<CheckChildTypeMatcher>(N)->getType();
661     switch (VT) {
662     case MVT::i32:
663     case MVT::i64:
664       OS << "OPC_CheckChild" << cast<CheckChildTypeMatcher>(N)->getChildNo()
665          << "TypeI" << MVT(VT).getSizeInBits() << ",\n";
666       return 1;
667     default:
668       OS << "OPC_CheckChild" << cast<CheckChildTypeMatcher>(N)->getChildNo()
669          << "Type, " << getEnumName(VT) << ",\n";
670       return 2;
671     }
672   }
673 
674   case Matcher::CheckInteger: {
675     OS << "OPC_CheckInteger, ";
676     unsigned Bytes =
677         1 + EmitSignedVBRValue(cast<CheckIntegerMatcher>(N)->getValue(), OS);
678     OS << '\n';
679     return Bytes;
680   }
681   case Matcher::CheckChildInteger: {
682     OS << "OPC_CheckChild" << cast<CheckChildIntegerMatcher>(N)->getChildNo()
683        << "Integer, ";
684     unsigned Bytes = 1 + EmitSignedVBRValue(
685                              cast<CheckChildIntegerMatcher>(N)->getValue(), OS);
686     OS << '\n';
687     return Bytes;
688   }
689   case Matcher::CheckCondCode:
690     OS << "OPC_CheckCondCode, ISD::"
691        << cast<CheckCondCodeMatcher>(N)->getCondCodeName() << ",\n";
692     return 2;
693 
694   case Matcher::CheckChild2CondCode:
695     OS << "OPC_CheckChild2CondCode, ISD::"
696        << cast<CheckChild2CondCodeMatcher>(N)->getCondCodeName() << ",\n";
697     return 2;
698 
699   case Matcher::CheckValueType:
700     OS << "OPC_CheckValueType, "
701        << getEnumName(cast<CheckValueTypeMatcher>(N)->getVT()) << ",\n";
702     return 2;
703 
704   case Matcher::CheckComplexPat: {
705     const CheckComplexPatMatcher *CCPM = cast<CheckComplexPatMatcher>(N);
706     const ComplexPattern &Pattern = CCPM->getPattern();
707     unsigned PatternNo = getComplexPat(Pattern);
708     if (PatternNo < 8)
709       OS << "OPC_CheckComplexPat" << PatternNo << ", /*#*/"
710          << CCPM->getMatchNumber() << ',';
711     else
712       OS << "OPC_CheckComplexPat, /*CP*/" << PatternNo << ", /*#*/"
713          << CCPM->getMatchNumber() << ',';
714 
715     if (!OmitComments) {
716       OS << " // " << Pattern.getSelectFunc();
717       OS << ":$" << CCPM->getName();
718       for (unsigned i = 0, e = Pattern.getNumOperands(); i != e; ++i)
719         OS << " #" << CCPM->getFirstResult() + i;
720 
721       if (Pattern.hasProperty(SDNPHasChain))
722         OS << " + chain result";
723     }
724     OS << '\n';
725     return PatternNo < 8 ? 2 : 3;
726   }
727 
728   case Matcher::CheckAndImm: {
729     OS << "OPC_CheckAndImm, ";
730     unsigned Bytes =
731         1 + EmitVBRValue(cast<CheckAndImmMatcher>(N)->getValue(), OS);
732     OS << '\n';
733     return Bytes;
734   }
735 
736   case Matcher::CheckOrImm: {
737     OS << "OPC_CheckOrImm, ";
738     unsigned Bytes =
739         1 + EmitVBRValue(cast<CheckOrImmMatcher>(N)->getValue(), OS);
740     OS << '\n';
741     return Bytes;
742   }
743 
744   case Matcher::CheckFoldableChainNode:
745     OS << "OPC_CheckFoldableChainNode,\n";
746     return 1;
747 
748   case Matcher::CheckImmAllOnesV:
749     OS << "OPC_CheckImmAllOnesV,\n";
750     return 1;
751 
752   case Matcher::CheckImmAllZerosV:
753     OS << "OPC_CheckImmAllZerosV,\n";
754     return 1;
755 
756   case Matcher::EmitInteger: {
757     int64_t Val = cast<EmitIntegerMatcher>(N)->getValue();
758     MVT::SimpleValueType VT = cast<EmitIntegerMatcher>(N)->getVT();
759     unsigned OpBytes;
760     switch (VT) {
761     case MVT::i8:
762     case MVT::i16:
763     case MVT::i32:
764     case MVT::i64:
765       OpBytes = 1;
766       OS << "OPC_EmitInteger" << MVT(VT).getSizeInBits() << ", ";
767       break;
768     default:
769       OpBytes = 2;
770       OS << "OPC_EmitInteger, " << getEnumName(VT) << ", ";
771       break;
772     }
773     unsigned Bytes = OpBytes + EmitSignedVBRValue(Val, OS);
774     OS << '\n';
775     return Bytes;
776   }
777   case Matcher::EmitStringInteger: {
778     const std::string &Val = cast<EmitStringIntegerMatcher>(N)->getValue();
779     MVT::SimpleValueType VT = cast<EmitStringIntegerMatcher>(N)->getVT();
780     // These should always fit into 7 bits.
781     unsigned OpBytes;
782     switch (VT) {
783     case MVT::i32:
784       OpBytes = 1;
785       OS << "OPC_EmitStringInteger" << MVT(VT).getSizeInBits() << ", ";
786       break;
787     default:
788       OpBytes = 2;
789       OS << "OPC_EmitStringInteger, " << getEnumName(VT) << ", ";
790       break;
791     }
792     OS << Val << ",\n";
793     return OpBytes + 1;
794   }
795 
796   case Matcher::EmitRegister: {
797     const EmitRegisterMatcher *Matcher = cast<EmitRegisterMatcher>(N);
798     const CodeGenRegister *Reg = Matcher->getReg();
799     MVT::SimpleValueType VT = Matcher->getVT();
800     // If the enum value of the register is larger than one byte can handle,
801     // use EmitRegister2.
802     if (Reg && Reg->EnumValue > 255) {
803       OS << "OPC_EmitRegister2, " << getEnumName(VT) << ", ";
804       OS << "TARGET_VAL(" << getQualifiedName(Reg->TheDef) << "),\n";
805       return 4;
806     }
807     unsigned OpBytes;
808     switch (VT) {
809     case MVT::i32:
810     case MVT::i64:
811       OpBytes = 1;
812       OS << "OPC_EmitRegisterI" << MVT(VT).getSizeInBits() << ", ";
813       break;
814     default:
815       OpBytes = 2;
816       OS << "OPC_EmitRegister, " << getEnumName(VT) << ", ";
817       break;
818     }
819     if (Reg) {
820       OS << getQualifiedName(Reg->TheDef) << ",\n";
821     } else {
822       OS << "0 ";
823       if (!OmitComments)
824         OS << "/*zero_reg*/";
825       OS << ",\n";
826     }
827     return OpBytes + 1;
828   }
829 
830   case Matcher::EmitConvertToTarget: {
831     unsigned Slot = cast<EmitConvertToTargetMatcher>(N)->getSlot();
832     if (Slot < 8) {
833       OS << "OPC_EmitConvertToTarget" << Slot << ",\n";
834       return 1;
835     }
836     OS << "OPC_EmitConvertToTarget, " << Slot << ",\n";
837     return 2;
838   }
839 
840   case Matcher::EmitMergeInputChains: {
841     const EmitMergeInputChainsMatcher *MN =
842         cast<EmitMergeInputChainsMatcher>(N);
843 
844     // Handle the specialized forms OPC_EmitMergeInputChains1_0, 1_1, and 1_2.
845     if (MN->getNumNodes() == 1 && MN->getNode(0) < 3) {
846       OS << "OPC_EmitMergeInputChains1_" << MN->getNode(0) << ",\n";
847       return 1;
848     }
849 
850     OS << "OPC_EmitMergeInputChains, " << MN->getNumNodes() << ", ";
851     for (unsigned i = 0, e = MN->getNumNodes(); i != e; ++i)
852       OS << MN->getNode(i) << ", ";
853     OS << '\n';
854     return 2 + MN->getNumNodes();
855   }
856   case Matcher::EmitCopyToReg: {
857     const auto *C2RMatcher = cast<EmitCopyToRegMatcher>(N);
858     int Bytes = 3;
859     const CodeGenRegister *Reg = C2RMatcher->getDestPhysReg();
860     unsigned Slot = C2RMatcher->getSrcSlot();
861     if (Reg->EnumValue > 255) {
862       assert(isUInt<16>(Reg->EnumValue) && "not handled");
863       OS << "OPC_EmitCopyToRegTwoByte, " << Slot << ", "
864          << "TARGET_VAL(" << getQualifiedName(Reg->TheDef) << "),\n";
865       ++Bytes;
866     } else {
867       if (Slot < 8) {
868         OS << "OPC_EmitCopyToReg" << Slot << ", "
869            << getQualifiedName(Reg->TheDef) << ",\n";
870         --Bytes;
871       } else
872         OS << "OPC_EmitCopyToReg, " << Slot << ", "
873            << getQualifiedName(Reg->TheDef) << ",\n";
874     }
875 
876     return Bytes;
877   }
878   case Matcher::EmitNodeXForm: {
879     const EmitNodeXFormMatcher *XF = cast<EmitNodeXFormMatcher>(N);
880     OS << "OPC_EmitNodeXForm, " << getNodeXFormID(XF->getNodeXForm()) << ", "
881        << XF->getSlot() << ',';
882     if (!OmitComments)
883       OS << " // " << XF->getNodeXForm()->getName();
884     OS << '\n';
885     return 3;
886   }
887 
888   case Matcher::EmitNode:
889   case Matcher::MorphNodeTo: {
890     auto NumCoveredBytes = 0;
891     if (InstrumentCoverage) {
892       if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(N)) {
893         NumCoveredBytes = 3;
894         OS << "OPC_Coverage, ";
895         std::string src =
896             GetPatFromTreePatternNode(SNT->getPattern().getSrcPattern());
897         std::string dst =
898             GetPatFromTreePatternNode(SNT->getPattern().getDstPattern());
899         Record *PatRecord = SNT->getPattern().getSrcRecord();
900         std::string include_src = getIncludePath(PatRecord);
901         unsigned Offset =
902             getPatternIdxFromTable(src + " -> " + dst, std::move(include_src));
903         OS << "TARGET_VAL(" << Offset << "),\n";
904         OS.indent(FullIndexWidth + Indent);
905       }
906     }
907     const EmitNodeMatcherCommon *EN = cast<EmitNodeMatcherCommon>(N);
908     bool IsEmitNode = isa<EmitNodeMatcher>(EN);
909     OS << (IsEmitNode ? "OPC_EmitNode" : "OPC_MorphNodeTo");
910     bool CompressVTs = EN->getNumVTs() < 3;
911     bool CompressNodeInfo = false;
912     if (CompressVTs) {
913       OS << EN->getNumVTs();
914       if (!EN->hasChain() && !EN->hasInGlue() && !EN->hasOutGlue() &&
915           !EN->hasMemRefs() && EN->getNumFixedArityOperands() == -1) {
916         CompressNodeInfo = true;
917         OS << "None";
918       } else if (EN->hasChain() && !EN->hasInGlue() && !EN->hasOutGlue() &&
919                  !EN->hasMemRefs() && EN->getNumFixedArityOperands() == -1) {
920         CompressNodeInfo = true;
921         OS << "Chain";
922       } else if (!IsEmitNode && !EN->hasChain() && EN->hasInGlue() &&
923                  !EN->hasOutGlue() && !EN->hasMemRefs() &&
924                  EN->getNumFixedArityOperands() == -1) {
925         CompressNodeInfo = true;
926         OS << "GlueInput";
927       } else if (!IsEmitNode && !EN->hasChain() && !EN->hasInGlue() &&
928                  EN->hasOutGlue() && !EN->hasMemRefs() &&
929                  EN->getNumFixedArityOperands() == -1) {
930         CompressNodeInfo = true;
931         OS << "GlueOutput";
932       }
933     }
934 
935     const CodeGenInstruction &CGI = EN->getInstruction();
936     OS << ", TARGET_VAL(" << CGI.Namespace << "::" << CGI.TheDef->getName()
937        << ")";
938 
939     if (!CompressNodeInfo) {
940       OS << ", 0";
941       if (EN->hasChain())
942         OS << "|OPFL_Chain";
943       if (EN->hasInGlue())
944         OS << "|OPFL_GlueInput";
945       if (EN->hasOutGlue())
946         OS << "|OPFL_GlueOutput";
947       if (EN->hasMemRefs())
948         OS << "|OPFL_MemRefs";
949       if (EN->getNumFixedArityOperands() != -1)
950         OS << "|OPFL_Variadic" << EN->getNumFixedArityOperands();
951     }
952     OS << ",\n";
953 
954     OS.indent(FullIndexWidth + Indent + 4);
955     if (!CompressVTs) {
956       OS << EN->getNumVTs();
957       if (!OmitComments)
958         OS << "/*#VTs*/";
959       OS << ", ";
960     }
961     for (unsigned i = 0, e = EN->getNumVTs(); i != e; ++i)
962       OS << getEnumName(EN->getVT(i)) << ", ";
963 
964     OS << EN->getNumOperands();
965     if (!OmitComments)
966       OS << "/*#Ops*/";
967     OS << ", ";
968     unsigned NumOperandBytes = 0;
969     for (unsigned i = 0, e = EN->getNumOperands(); i != e; ++i)
970       NumOperandBytes += EmitVBRValue(EN->getOperand(i), OS);
971 
972     if (!OmitComments) {
973       // Print the result #'s for EmitNode.
974       if (const EmitNodeMatcher *E = dyn_cast<EmitNodeMatcher>(EN)) {
975         if (unsigned NumResults = EN->getNumVTs()) {
976           OS << " // Results =";
977           unsigned First = E->getFirstResultSlot();
978           for (unsigned i = 0; i != NumResults; ++i)
979             OS << " #" << First + i;
980         }
981       }
982       OS << '\n';
983 
984       if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(N)) {
985         OS.indent(FullIndexWidth + Indent)
986             << "// Src: " << SNT->getPattern().getSrcPattern()
987             << " - Complexity = " << SNT->getPattern().getPatternComplexity(CGP)
988             << '\n';
989         OS.indent(FullIndexWidth + Indent)
990             << "// Dst: " << SNT->getPattern().getDstPattern() << '\n';
991       }
992     } else
993       OS << '\n';
994 
995     return 4 + !CompressVTs + !CompressNodeInfo + EN->getNumVTs() +
996            NumOperandBytes + NumCoveredBytes;
997   }
998   case Matcher::CompleteMatch: {
999     const CompleteMatchMatcher *CM = cast<CompleteMatchMatcher>(N);
1000     auto NumCoveredBytes = 0;
1001     if (InstrumentCoverage) {
1002       NumCoveredBytes = 3;
1003       OS << "OPC_Coverage, ";
1004       std::string src =
1005           GetPatFromTreePatternNode(CM->getPattern().getSrcPattern());
1006       std::string dst =
1007           GetPatFromTreePatternNode(CM->getPattern().getDstPattern());
1008       Record *PatRecord = CM->getPattern().getSrcRecord();
1009       std::string include_src = getIncludePath(PatRecord);
1010       unsigned Offset =
1011           getPatternIdxFromTable(src + " -> " + dst, std::move(include_src));
1012       OS << "TARGET_VAL(" << Offset << "),\n";
1013       OS.indent(FullIndexWidth + Indent);
1014     }
1015     OS << "OPC_CompleteMatch, " << CM->getNumResults() << ", ";
1016     unsigned NumResultBytes = 0;
1017     for (unsigned i = 0, e = CM->getNumResults(); i != e; ++i)
1018       NumResultBytes += EmitVBRValue(CM->getResult(i), OS);
1019     OS << '\n';
1020     if (!OmitComments) {
1021       OS.indent(FullIndexWidth + Indent)
1022           << " // Src: " << CM->getPattern().getSrcPattern()
1023           << " - Complexity = " << CM->getPattern().getPatternComplexity(CGP)
1024           << '\n';
1025       OS.indent(FullIndexWidth + Indent)
1026           << " // Dst: " << CM->getPattern().getDstPattern();
1027     }
1028     OS << '\n';
1029     return 2 + NumResultBytes + NumCoveredBytes;
1030   }
1031   }
1032   llvm_unreachable("Unreachable");
1033 }
1034 
1035 /// This function traverses the matcher tree and emits all the nodes.
1036 /// The nodes have already been sized.
1037 unsigned MatcherTableEmitter::EmitMatcherList(const Matcher *N,
1038                                               const unsigned Indent,
1039                                               unsigned CurrentIdx,
1040                                               raw_ostream &OS) {
1041   unsigned Size = 0;
1042   while (N) {
1043     if (!OmitComments)
1044       OS << "/*" << format_decimal(CurrentIdx, IndexWidth) << "*/";
1045     unsigned MatcherSize = EmitMatcher(N, Indent, CurrentIdx, OS);
1046     Size += MatcherSize;
1047     CurrentIdx += MatcherSize;
1048 
1049     // If there are other nodes in this list, iterate to them, otherwise we're
1050     // done.
1051     N = N->getNext();
1052   }
1053   return Size;
1054 }
1055 
1056 void MatcherTableEmitter::EmitNodePredicatesFunction(
1057     const std::vector<TreePattern *> &Preds, StringRef Decl, raw_ostream &OS) {
1058   if (Preds.empty())
1059     return;
1060 
1061   BeginEmitFunction(OS, "bool", Decl, true /*AddOverride*/);
1062   OS << "{\n";
1063   OS << "  switch (PredNo) {\n";
1064   OS << "  default: llvm_unreachable(\"Invalid predicate in table?\");\n";
1065   for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
1066     // Emit the predicate code corresponding to this pattern.
1067     TreePredicateFn PredFn(Preds[i]);
1068     assert(!PredFn.isAlwaysTrue() && "No code in this predicate");
1069     std::string PredFnCodeStr = PredFn.getCodeToRunOnSDNode();
1070 
1071     OS << "  case " << i << ": {\n";
1072     for (auto *SimilarPred : NodePredicatesByCodeToRun[PredFnCodeStr])
1073       OS << "    // " << TreePredicateFn(SimilarPred).getFnName() << '\n';
1074     OS << PredFnCodeStr << "\n  }\n";
1075   }
1076   OS << "  }\n";
1077   OS << "}\n";
1078   EndEmitFunction(OS);
1079 }
1080 
1081 void MatcherTableEmitter::EmitPredicateFunctions(raw_ostream &OS) {
1082   // Emit pattern predicates.
1083   if (!PatternPredicates.empty()) {
1084     BeginEmitFunction(OS, "bool",
1085                       "CheckPatternPredicate(unsigned PredNo) const",
1086                       true /*AddOverride*/);
1087     OS << "{\n";
1088     OS << "  switch (PredNo) {\n";
1089     OS << "  default: llvm_unreachable(\"Invalid predicate in table?\");\n";
1090     for (unsigned i = 0, e = PatternPredicates.size(); i != e; ++i)
1091       OS << "  case " << i << ": return " << PatternPredicates[i] << ";\n";
1092     OS << "  }\n";
1093     OS << "}\n";
1094     EndEmitFunction(OS);
1095   }
1096 
1097   // Emit Node predicates.
1098   EmitNodePredicatesFunction(
1099       NodePredicates, "CheckNodePredicate(SDNode *Node, unsigned PredNo) const",
1100       OS);
1101   EmitNodePredicatesFunction(
1102       NodePredicatesWithOperands,
1103       "CheckNodePredicateWithOperands(SDNode *Node, unsigned PredNo, "
1104       "const SmallVectorImpl<SDValue> &Operands) const",
1105       OS);
1106 
1107   // Emit CompletePattern matchers.
1108   // FIXME: This should be const.
1109   if (!ComplexPatterns.empty()) {
1110     BeginEmitFunction(
1111         OS, "bool",
1112         "CheckComplexPattern(SDNode *Root, SDNode *Parent,\n"
1113         "      SDValue N, unsigned PatternNo,\n"
1114         "      SmallVectorImpl<std::pair<SDValue, SDNode *>> &Result)",
1115         true /*AddOverride*/);
1116     OS << "{\n";
1117     OS << "  unsigned NextRes = Result.size();\n";
1118     OS << "  switch (PatternNo) {\n";
1119     OS << "  default: llvm_unreachable(\"Invalid pattern # in table?\");\n";
1120     for (unsigned i = 0, e = ComplexPatterns.size(); i != e; ++i) {
1121       const ComplexPattern &P = *ComplexPatterns[i];
1122       unsigned NumOps = P.getNumOperands();
1123 
1124       if (P.hasProperty(SDNPHasChain))
1125         ++NumOps; // Get the chained node too.
1126 
1127       OS << "  case " << i << ":\n";
1128       if (InstrumentCoverage)
1129         OS << "  {\n";
1130       OS << "    Result.resize(NextRes+" << NumOps << ");\n";
1131       if (InstrumentCoverage)
1132         OS << "    bool Succeeded = " << P.getSelectFunc();
1133       else
1134         OS << "  return " << P.getSelectFunc();
1135 
1136       OS << "(";
1137       // If the complex pattern wants the root of the match, pass it in as the
1138       // first argument.
1139       if (P.hasProperty(SDNPWantRoot))
1140         OS << "Root, ";
1141 
1142       // If the complex pattern wants the parent of the operand being matched,
1143       // pass it in as the next argument.
1144       if (P.hasProperty(SDNPWantParent))
1145         OS << "Parent, ";
1146 
1147       OS << "N";
1148       for (unsigned i = 0; i != NumOps; ++i)
1149         OS << ", Result[NextRes+" << i << "].first";
1150       OS << ");\n";
1151       if (InstrumentCoverage) {
1152         OS << "    if (Succeeded)\n";
1153         OS << "       dbgs() << \"\\nCOMPLEX_PATTERN: " << P.getSelectFunc()
1154            << "\\n\" ;\n";
1155         OS << "    return Succeeded;\n";
1156         OS << "    }\n";
1157       }
1158     }
1159     OS << "  }\n";
1160     OS << "}\n";
1161     EndEmitFunction(OS);
1162   }
1163 
1164   // Emit SDNodeXForm handlers.
1165   // FIXME: This should be const.
1166   if (!NodeXForms.empty()) {
1167     BeginEmitFunction(OS, "SDValue",
1168                       "RunSDNodeXForm(SDValue V, unsigned XFormNo)",
1169                       true /*AddOverride*/);
1170     OS << "{\n";
1171     OS << "  switch (XFormNo) {\n";
1172     OS << "  default: llvm_unreachable(\"Invalid xform # in table?\");\n";
1173 
1174     // FIXME: The node xform could take SDValue's instead of SDNode*'s.
1175     for (unsigned i = 0, e = NodeXForms.size(); i != e; ++i) {
1176       const CodeGenDAGPatterns::NodeXForm &Entry =
1177           CGP.getSDNodeTransform(NodeXForms[i]);
1178 
1179       Record *SDNode = Entry.first;
1180       const std::string &Code = Entry.second;
1181 
1182       OS << "  case " << i << ": {  ";
1183       if (!OmitComments)
1184         OS << "// " << NodeXForms[i]->getName();
1185       OS << '\n';
1186 
1187       std::string ClassName =
1188           std::string(CGP.getSDNodeInfo(SDNode).getSDClassName());
1189       if (ClassName == "SDNode")
1190         OS << "    SDNode *N = V.getNode();\n";
1191       else
1192         OS << "    " << ClassName << " *N = cast<" << ClassName
1193            << ">(V.getNode());\n";
1194       OS << Code << "\n  }\n";
1195     }
1196     OS << "  }\n";
1197     OS << "}\n";
1198     EndEmitFunction(OS);
1199   }
1200 }
1201 
1202 static StringRef getOpcodeString(Matcher::KindTy Kind) {
1203   switch (Kind) {
1204   case Matcher::Scope:
1205     return "OPC_Scope";
1206   case Matcher::RecordNode:
1207     return "OPC_RecordNode";
1208   case Matcher::RecordChild:
1209     return "OPC_RecordChild";
1210   case Matcher::RecordMemRef:
1211     return "OPC_RecordMemRef";
1212   case Matcher::CaptureGlueInput:
1213     return "OPC_CaptureGlueInput";
1214   case Matcher::MoveChild:
1215     return "OPC_MoveChild";
1216   case Matcher::MoveSibling:
1217     return "OPC_MoveSibling";
1218   case Matcher::MoveParent:
1219     return "OPC_MoveParent";
1220   case Matcher::CheckSame:
1221     return "OPC_CheckSame";
1222   case Matcher::CheckChildSame:
1223     return "OPC_CheckChildSame";
1224   case Matcher::CheckPatternPredicate:
1225     return "OPC_CheckPatternPredicate";
1226   case Matcher::CheckPredicate:
1227     return "OPC_CheckPredicate";
1228   case Matcher::CheckOpcode:
1229     return "OPC_CheckOpcode";
1230   case Matcher::SwitchOpcode:
1231     return "OPC_SwitchOpcode";
1232   case Matcher::CheckType:
1233     return "OPC_CheckType";
1234   case Matcher::SwitchType:
1235     return "OPC_SwitchType";
1236   case Matcher::CheckChildType:
1237     return "OPC_CheckChildType";
1238   case Matcher::CheckInteger:
1239     return "OPC_CheckInteger";
1240   case Matcher::CheckChildInteger:
1241     return "OPC_CheckChildInteger";
1242   case Matcher::CheckCondCode:
1243     return "OPC_CheckCondCode";
1244   case Matcher::CheckChild2CondCode:
1245     return "OPC_CheckChild2CondCode";
1246   case Matcher::CheckValueType:
1247     return "OPC_CheckValueType";
1248   case Matcher::CheckComplexPat:
1249     return "OPC_CheckComplexPat";
1250   case Matcher::CheckAndImm:
1251     return "OPC_CheckAndImm";
1252   case Matcher::CheckOrImm:
1253     return "OPC_CheckOrImm";
1254   case Matcher::CheckFoldableChainNode:
1255     return "OPC_CheckFoldableChainNode";
1256   case Matcher::CheckImmAllOnesV:
1257     return "OPC_CheckImmAllOnesV";
1258   case Matcher::CheckImmAllZerosV:
1259     return "OPC_CheckImmAllZerosV";
1260   case Matcher::EmitInteger:
1261     return "OPC_EmitInteger";
1262   case Matcher::EmitStringInteger:
1263     return "OPC_EmitStringInteger";
1264   case Matcher::EmitRegister:
1265     return "OPC_EmitRegister";
1266   case Matcher::EmitConvertToTarget:
1267     return "OPC_EmitConvertToTarget";
1268   case Matcher::EmitMergeInputChains:
1269     return "OPC_EmitMergeInputChains";
1270   case Matcher::EmitCopyToReg:
1271     return "OPC_EmitCopyToReg";
1272   case Matcher::EmitNode:
1273     return "OPC_EmitNode";
1274   case Matcher::MorphNodeTo:
1275     return "OPC_MorphNodeTo";
1276   case Matcher::EmitNodeXForm:
1277     return "OPC_EmitNodeXForm";
1278   case Matcher::CompleteMatch:
1279     return "OPC_CompleteMatch";
1280   }
1281 
1282   llvm_unreachable("Unhandled opcode?");
1283 }
1284 
1285 void MatcherTableEmitter::EmitHistogram(const Matcher *M, raw_ostream &OS) {
1286   if (OmitComments)
1287     return;
1288 
1289   OS << "  // Opcode Histogram:\n";
1290   for (unsigned i = 0, e = OpcodeCounts.size(); i != e; ++i) {
1291     OS << "  // #"
1292        << left_justify(getOpcodeString((Matcher::KindTy)i), HistOpcWidth)
1293        << " = " << OpcodeCounts[i] << '\n';
1294   }
1295   OS << '\n';
1296 }
1297 
1298 void llvm::EmitMatcherTable(Matcher *TheMatcher, const CodeGenDAGPatterns &CGP,
1299                             raw_ostream &OS) {
1300   OS << "#if defined(GET_DAGISEL_DECL) && defined(GET_DAGISEL_BODY)\n";
1301   OS << "#error GET_DAGISEL_DECL and GET_DAGISEL_BODY cannot be both defined, ";
1302   OS << "undef both for inline definitions\n";
1303   OS << "#endif\n\n";
1304 
1305   // Emit a check for omitted class name.
1306   OS << "#ifdef GET_DAGISEL_BODY\n";
1307   OS << "#define LOCAL_DAGISEL_STRINGIZE(X) LOCAL_DAGISEL_STRINGIZE_(X)\n";
1308   OS << "#define LOCAL_DAGISEL_STRINGIZE_(X) #X\n";
1309   OS << "static_assert(sizeof(LOCAL_DAGISEL_STRINGIZE(GET_DAGISEL_BODY)) > 1,"
1310         "\n";
1311   OS << "   \"GET_DAGISEL_BODY is empty: it should be defined with the class "
1312         "name\");\n";
1313   OS << "#undef LOCAL_DAGISEL_STRINGIZE_\n";
1314   OS << "#undef LOCAL_DAGISEL_STRINGIZE\n";
1315   OS << "#endif\n\n";
1316 
1317   OS << "#if !defined(GET_DAGISEL_DECL) && !defined(GET_DAGISEL_BODY)\n";
1318   OS << "#define DAGISEL_INLINE 1\n";
1319   OS << "#else\n";
1320   OS << "#define DAGISEL_INLINE 0\n";
1321   OS << "#endif\n\n";
1322 
1323   OS << "#if !DAGISEL_INLINE\n";
1324   OS << "#define DAGISEL_CLASS_COLONCOLON GET_DAGISEL_BODY ::\n";
1325   OS << "#else\n";
1326   OS << "#define DAGISEL_CLASS_COLONCOLON\n";
1327   OS << "#endif\n\n";
1328 
1329   BeginEmitFunction(OS, "void", "SelectCode(SDNode *N)", false /*AddOverride*/);
1330   MatcherTableEmitter MatcherEmitter(TheMatcher, CGP);
1331 
1332   // First we size all the children of the three kinds of matchers that have
1333   // them. This is done by sharing the code in EmitMatcher(). but we don't
1334   // want to emit anything, so we turn off comments and use a null stream.
1335   bool SaveOmitComments = OmitComments;
1336   OmitComments = true;
1337   raw_null_ostream NullOS;
1338   unsigned TotalSize = MatcherEmitter.SizeMatcherList(TheMatcher, NullOS);
1339   OmitComments = SaveOmitComments;
1340 
1341   // Now that the matchers are sized, we can emit the code for them to the
1342   // final stream.
1343   OS << "{\n";
1344   OS << "  // Some target values are emitted as 2 bytes, TARGET_VAL handles\n";
1345   OS << "  // this.\n";
1346   OS << "  #define TARGET_VAL(X) X & 255, unsigned(X) >> 8\n";
1347   OS << "  static const unsigned char MatcherTable[] = {\n";
1348   TotalSize = MatcherEmitter.EmitMatcherList(TheMatcher, 1, 0, OS);
1349   OS << "    0\n  }; // Total Array size is " << (TotalSize + 1)
1350      << " bytes\n\n";
1351 
1352   MatcherEmitter.EmitHistogram(TheMatcher, OS);
1353 
1354   OS << "  #undef TARGET_VAL\n";
1355   OS << "  SelectCodeCommon(N, MatcherTable, sizeof(MatcherTable));\n";
1356   OS << "}\n";
1357   EndEmitFunction(OS);
1358 
1359   // Next up, emit the function for node and pattern predicates:
1360   MatcherEmitter.EmitPredicateFunctions(OS);
1361 
1362   if (InstrumentCoverage)
1363     MatcherEmitter.EmitPatternMatchTable(OS);
1364 
1365   // Clean up the preprocessor macros.
1366   OS << "\n";
1367   OS << "#ifdef DAGISEL_INLINE\n";
1368   OS << "#undef DAGISEL_INLINE\n";
1369   OS << "#endif\n";
1370   OS << "#ifdef DAGISEL_CLASS_COLONCOLON\n";
1371   OS << "#undef DAGISEL_CLASS_COLONCOLON\n";
1372   OS << "#endif\n";
1373   OS << "#ifdef GET_DAGISEL_DECL\n";
1374   OS << "#undef GET_DAGISEL_DECL\n";
1375   OS << "#endif\n";
1376   OS << "#ifdef GET_DAGISEL_BODY\n";
1377   OS << "#undef GET_DAGISEL_BODY\n";
1378   OS << "#endif\n";
1379 }
1380