blob: 133157f0cc50ab89238b93e2b282daf1bff19697 [file] [log] [blame]
Chris Lattnerda272d12010-02-15 08:04:42 +00001//===- DAGISelMatcherEmitter.cpp - Matcher Emitter ------------------------===//
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 file contains code to generate C++ code a matcher.
11//
12//===----------------------------------------------------------------------===//
13
14#include "DAGISelMatcher.h"
15#include "CodeGenDAGPatterns.h"
Chris Lattner8e946be2010-02-21 03:22:59 +000016#include "Record.h"
Chris Lattnere609a512010-02-17 00:31:50 +000017#include "llvm/ADT/DenseMap.h"
Chris Lattnerda272d12010-02-15 08:04:42 +000018#include "llvm/ADT/SmallString.h"
Chris Lattner050a03d2010-02-16 07:21:10 +000019#include "llvm/ADT/StringMap.h"
Chris Lattnerda272d12010-02-15 08:04:42 +000020#include "llvm/Support/FormattedStream.h"
21using namespace llvm;
22
Chris Lattnerda272d12010-02-15 08:04:42 +000023enum {
Chris Lattner5be6e592010-02-17 00:39:26 +000024 CommentIndent = 30
Chris Lattnerda272d12010-02-15 08:04:42 +000025};
Chris Lattnerda272d12010-02-15 08:04:42 +000026
Chris Lattnere02ea542010-02-16 06:52:01 +000027namespace {
28class MatcherTableEmitter {
Chris Lattner050a03d2010-02-16 07:21:10 +000029 StringMap<unsigned> NodePredicateMap, PatternPredicateMap;
30 std::vector<std::string> NodePredicates, PatternPredicates;
Chris Lattnere609a512010-02-17 00:31:50 +000031
32 DenseMap<const ComplexPattern*, unsigned> ComplexPatternMap;
33 std::vector<const ComplexPattern*> ComplexPatterns;
Chris Lattner8e946be2010-02-21 03:22:59 +000034
35
36 DenseMap<Record*, unsigned> NodeXFormMap;
Chris Lattner4d0c9312010-03-01 01:54:19 +000037 std::vector<Record*> NodeXForms;
Chris Lattner8e946be2010-02-21 03:22:59 +000038
Chris Lattner09b9f392010-02-24 19:17:12 +000039 // Per opcode frequence count.
40 std::vector<unsigned> Histogram;
Chris Lattnere02ea542010-02-16 06:52:01 +000041public:
Chris Lattner984e1882010-02-21 06:30:04 +000042 MatcherTableEmitter() {}
Chris Lattnere02ea542010-02-16 06:52:01 +000043
Chris Lattnerb21ba712010-02-25 02:04:40 +000044 unsigned EmitMatcherList(const Matcher *N, unsigned Indent,
Chris Lattner8fbad242010-02-21 07:16:41 +000045 unsigned StartIdx, formatted_raw_ostream &OS);
Chris Lattner050a03d2010-02-16 07:21:10 +000046
Chris Lattner4d0c9312010-03-01 01:54:19 +000047 void EmitPredicateFunctions(const CodeGenDAGPatterns &CGP,
48 formatted_raw_ostream &OS);
Chris Lattner09b9f392010-02-24 19:17:12 +000049
50 void EmitHistogram(formatted_raw_ostream &OS);
Chris Lattnere02ea542010-02-16 06:52:01 +000051private:
Chris Lattnerd6c84722010-02-25 19:00:39 +000052 unsigned EmitMatcher(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
Chris Lattner984e1882010-02-21 06:30:04 +000053 formatted_raw_ostream &OS);
Chris Lattner050a03d2010-02-16 07:21:10 +000054
55 unsigned getNodePredicate(StringRef PredName) {
56 unsigned &Entry = NodePredicateMap[PredName];
57 if (Entry == 0) {
58 NodePredicates.push_back(PredName.str());
59 Entry = NodePredicates.size();
60 }
61 return Entry-1;
62 }
63 unsigned getPatternPredicate(StringRef PredName) {
64 unsigned &Entry = PatternPredicateMap[PredName];
65 if (Entry == 0) {
66 PatternPredicates.push_back(PredName.str());
67 Entry = PatternPredicates.size();
68 }
69 return Entry-1;
70 }
Chris Lattnere609a512010-02-17 00:31:50 +000071
72 unsigned getComplexPat(const ComplexPattern &P) {
73 unsigned &Entry = ComplexPatternMap[&P];
74 if (Entry == 0) {
75 ComplexPatterns.push_back(&P);
76 Entry = ComplexPatterns.size();
77 }
78 return Entry-1;
79 }
Chris Lattner8e946be2010-02-21 03:22:59 +000080
81 unsigned getNodeXFormID(Record *Rec) {
82 unsigned &Entry = NodeXFormMap[Rec];
83 if (Entry == 0) {
84 NodeXForms.push_back(Rec);
85 Entry = NodeXForms.size();
86 }
87 return Entry-1;
88 }
89
Chris Lattnere02ea542010-02-16 06:52:01 +000090};
91} // end anonymous namespace.
92
Chris Lattnerd6c84722010-02-25 19:00:39 +000093static unsigned GetVBRSize(unsigned Val) {
94 if (Val <= 127) return 1;
95
96 unsigned NumBytes = 0;
97 while (Val >= 128) {
98 Val >>= 7;
99 ++NumBytes;
100 }
101 return NumBytes+1;
102}
103
Chris Lattner5007e1e2010-02-23 00:59:59 +0000104/// EmitVBRValue - Emit the specified value as a VBR, returning the number of
105/// bytes emitted.
Chris Lattner53106542010-02-28 22:14:32 +0000106static uint64_t EmitVBRValue(uint64_t Val, raw_ostream &OS) {
Chris Lattner5007e1e2010-02-23 00:59:59 +0000107 if (Val <= 127) {
108 OS << Val << ", ";
109 return 1;
110 }
111
Chris Lattner53106542010-02-28 22:14:32 +0000112 uint64_t InVal = Val;
Chris Lattner5007e1e2010-02-23 00:59:59 +0000113 unsigned NumBytes = 0;
Chris Lattner860d4a72010-02-23 01:07:39 +0000114 while (Val >= 128) {
Chris Lattner5007e1e2010-02-23 00:59:59 +0000115 OS << (Val&127) << "|128,";
116 Val >>= 7;
117 ++NumBytes;
118 }
119 OS << Val << "/*" << InVal << "*/, ";
120 return NumBytes+1;
121}
122
Chris Lattnerda272d12010-02-15 08:04:42 +0000123/// EmitMatcherOpcodes - Emit bytes for the specified matcher and return
124/// the number of bytes emitted.
Chris Lattnere02ea542010-02-16 06:52:01 +0000125unsigned MatcherTableEmitter::
Chris Lattnerd6c84722010-02-25 19:00:39 +0000126EmitMatcher(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
127 formatted_raw_ostream &OS) {
Chris Lattnerda272d12010-02-15 08:04:42 +0000128 OS.PadToColumn(Indent*2);
129
130 switch (N->getKind()) {
Chris Lattnerd6c84722010-02-25 19:00:39 +0000131 case Matcher::Scope: {
132 const ScopeMatcher *SM = cast<ScopeMatcher>(N);
133 assert(SM->getNext() == 0 && "Shouldn't have next after scope");
134
135 unsigned StartIdx = CurrentIdx;
136
137 // Emit all of the children.
138 for (unsigned i = 0, e = SM->getNumChildren(); i != e; ++i) {
139 if (i == 0) {
140 OS << "OPC_Scope, ";
141 ++CurrentIdx;
142 } else {
143 OS << "/*" << CurrentIdx << "*/";
144 OS.PadToColumn(Indent*2) << "/*Scope*/ ";
145 }
146
147 // We need to encode the child and the offset of the failure code before
148 // emitting either of them. Handle this by buffering the output into a
149 // string while we get the size. Unfortunately, the offset of the
150 // children depends on the VBR size of the child, so for large children we
151 // have to iterate a bit.
152 SmallString<128> TmpBuf;
153 unsigned ChildSize = 0;
154 unsigned VBRSize = 0;
155 do {
156 VBRSize = GetVBRSize(ChildSize);
157
158 TmpBuf.clear();
159 raw_svector_ostream OS(TmpBuf);
160 formatted_raw_ostream FOS(OS);
161 ChildSize = EmitMatcherList(cast<ScopeMatcher>(N)->getChild(i),
162 Indent+1, CurrentIdx+VBRSize, FOS);
163 } while (GetVBRSize(ChildSize) != VBRSize);
164
165 assert(ChildSize != 0 && "Should not have a zero-sized child!");
166
167 CurrentIdx += EmitVBRValue(ChildSize, OS);
Chris Lattner5a56e5c2010-02-26 08:15:02 +0000168 OS << "/*->" << CurrentIdx+ChildSize << "*/";
169
170 if (i == 0)
171 OS.PadToColumn(CommentIndent) << "// " << SM->getNumChildren()
172 << " children in Scope";
173
Chris Lattnerd6c84722010-02-25 19:00:39 +0000174 OS << '\n' << TmpBuf.str();
175 CurrentIdx += ChildSize;
176 }
177
178 // Emit a zero as a sentinel indicating end of 'Scope'.
179 OS << "/*" << CurrentIdx << "*/";
180 OS.PadToColumn(Indent*2) << "0, /*End of Scope*/\n";
181 return CurrentIdx - StartIdx + 1;
182 }
183
Chris Lattnerb21ba712010-02-25 02:04:40 +0000184 case Matcher::RecordNode:
Chris Lattner845c0422010-02-18 22:03:03 +0000185 OS << "OPC_RecordNode,";
Chris Lattnerc96087b2010-02-17 01:03:09 +0000186 OS.PadToColumn(CommentIndent) << "// "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000187 << cast<RecordMatcher>(N)->getWhatFor() << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000188 return 1;
Chris Lattner19b5a752010-02-24 07:31:45 +0000189
Chris Lattnerb21ba712010-02-25 02:04:40 +0000190 case Matcher::RecordChild:
191 OS << "OPC_RecordChild" << cast<RecordChildMatcher>(N)->getChildNo()
Chris Lattner19b5a752010-02-24 07:31:45 +0000192 << ',';
193 OS.PadToColumn(CommentIndent) << "// "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000194 << cast<RecordChildMatcher>(N)->getWhatFor() << '\n';
Chris Lattner19b5a752010-02-24 07:31:45 +0000195 return 1;
Chris Lattner8e946be2010-02-21 03:22:59 +0000196
Chris Lattnerb21ba712010-02-25 02:04:40 +0000197 case Matcher::RecordMemRef:
Chris Lattner8e946be2010-02-21 03:22:59 +0000198 OS << "OPC_RecordMemRef,\n";
199 return 1;
200
Chris Lattnerb21ba712010-02-25 02:04:40 +0000201 case Matcher::CaptureFlagInput:
Chris Lattner8e946be2010-02-21 03:22:59 +0000202 OS << "OPC_CaptureFlagInput,\n";
203 return 1;
204
Chris Lattnerb21ba712010-02-25 02:04:40 +0000205 case Matcher::MoveChild:
206 OS << "OPC_MoveChild, " << cast<MoveChildMatcher>(N)->getChildNo() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000207 return 2;
208
Chris Lattnerb21ba712010-02-25 02:04:40 +0000209 case Matcher::MoveParent:
Chris Lattnerda272d12010-02-15 08:04:42 +0000210 OS << "OPC_MoveParent,\n";
211 return 1;
212
Chris Lattnerb21ba712010-02-25 02:04:40 +0000213 case Matcher::CheckSame:
Chris Lattnerda272d12010-02-15 08:04:42 +0000214 OS << "OPC_CheckSame, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000215 << cast<CheckSameMatcher>(N)->getMatchNumber() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000216 return 2;
217
Chris Lattnerb21ba712010-02-25 02:04:40 +0000218 case Matcher::CheckPatternPredicate: {
219 StringRef Pred = cast<CheckPatternPredicateMatcher>(N)->getPredicate();
Chris Lattner050a03d2010-02-16 07:21:10 +0000220 OS << "OPC_CheckPatternPredicate, " << getPatternPredicate(Pred) << ',';
221 OS.PadToColumn(CommentIndent) << "// " << Pred << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000222 return 2;
Chris Lattner050a03d2010-02-16 07:21:10 +0000223 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000224 case Matcher::CheckPredicate: {
225 StringRef Pred = cast<CheckPredicateMatcher>(N)->getPredicateName();
Chris Lattner050a03d2010-02-16 07:21:10 +0000226 OS << "OPC_CheckPredicate, " << getNodePredicate(Pred) << ',';
227 OS.PadToColumn(CommentIndent) << "// " << Pred << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000228 return 2;
Chris Lattner050a03d2010-02-16 07:21:10 +0000229 }
230
Chris Lattnerb21ba712010-02-25 02:04:40 +0000231 case Matcher::CheckOpcode:
Chris Lattnerda272d12010-02-15 08:04:42 +0000232 OS << "OPC_CheckOpcode, "
Chris Lattnera230f962010-02-27 21:48:43 +0000233 << cast<CheckOpcodeMatcher>(N)->getOpcode().getEnumName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000234 return 2;
235
Chris Lattnerb21ba712010-02-25 02:04:40 +0000236 case Matcher::CheckMultiOpcode: {
237 const CheckMultiOpcodeMatcher *CMO = cast<CheckMultiOpcodeMatcher>(N);
Chris Lattnera230f962010-02-27 21:48:43 +0000238 OS << "OPC_CheckMultiOpcode, " << CMO->getNumOpcodes() << ", ";
239 for (unsigned i = 0, e = CMO->getNumOpcodes(); i != e; ++i)
240 OS << CMO->getOpcode(i).getEnumName() << ", ";
Chris Lattner12a667c2010-02-22 22:30:37 +0000241 OS << '\n';
Chris Lattnera230f962010-02-27 21:48:43 +0000242 return 2 + CMO->getNumOpcodes();
Chris Lattner12a667c2010-02-22 22:30:37 +0000243 }
244
Chris Lattnerb21ba712010-02-25 02:04:40 +0000245 case Matcher::CheckType:
Chris Lattnerda272d12010-02-15 08:04:42 +0000246 OS << "OPC_CheckType, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000247 << getEnumName(cast<CheckTypeMatcher>(N)->getType()) << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000248 return 2;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000249 case Matcher::CheckChildType:
Chris Lattner23cfda72010-02-24 20:15:25 +0000250 OS << "OPC_CheckChild"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000251 << cast<CheckChildTypeMatcher>(N)->getChildNo() << "Type, "
252 << getEnumName(cast<CheckChildTypeMatcher>(N)->getType()) << ",\n";
Chris Lattner23cfda72010-02-24 20:15:25 +0000253 return 2;
254
Chris Lattner53106542010-02-28 22:14:32 +0000255 case Matcher::CheckInteger:
256 OS << "OPC_CheckInteger, ";
257 return 1+EmitVBRValue(cast<CheckIntegerMatcher>(N)->getValue(), OS);
Chris Lattnerb21ba712010-02-25 02:04:40 +0000258 case Matcher::CheckCondCode:
Chris Lattnerda272d12010-02-15 08:04:42 +0000259 OS << "OPC_CheckCondCode, ISD::"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000260 << cast<CheckCondCodeMatcher>(N)->getCondCodeName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000261 return 2;
262
Chris Lattnerb21ba712010-02-25 02:04:40 +0000263 case Matcher::CheckValueType:
Chris Lattnerda272d12010-02-15 08:04:42 +0000264 OS << "OPC_CheckValueType, MVT::"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000265 << cast<CheckValueTypeMatcher>(N)->getTypeName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000266 return 2;
267
Chris Lattnerb21ba712010-02-25 02:04:40 +0000268 case Matcher::CheckComplexPat: {
Chris Lattner5be6e592010-02-17 00:39:26 +0000269 const ComplexPattern &Pattern =
Chris Lattnerb21ba712010-02-25 02:04:40 +0000270 cast<CheckComplexPatMatcher>(N)->getPattern();
Chris Lattner5be6e592010-02-17 00:39:26 +0000271 OS << "OPC_CheckComplexPat, " << getComplexPat(Pattern) << ',';
Chris Lattner781f3592010-02-17 06:47:35 +0000272 OS.PadToColumn(CommentIndent) << "// " << Pattern.getSelectFunc();
273 OS << ": " << Pattern.getNumOperands() << " operands";
274 if (Pattern.hasProperty(SDNPHasChain))
275 OS << " + chain result and input";
276 OS << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000277 return 2;
Chris Lattner5be6e592010-02-17 00:39:26 +0000278 }
Chris Lattnerda272d12010-02-15 08:04:42 +0000279
Chris Lattner53106542010-02-28 22:14:32 +0000280 case Matcher::CheckAndImm:
281 OS << "OPC_CheckAndImm, ";
282 return 1+EmitVBRValue(cast<CheckAndImmMatcher>(N)->getValue(), OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000283
Chris Lattner53106542010-02-28 22:14:32 +0000284 case Matcher::CheckOrImm:
285 OS << "OPC_CheckOrImm, ";
286 return 1+EmitVBRValue(cast<CheckOrImmMatcher>(N)->getValue(), OS);
287
Chris Lattnerb21ba712010-02-25 02:04:40 +0000288 case Matcher::CheckFoldableChainNode:
Chris Lattner21390d72010-02-16 19:15:55 +0000289 OS << "OPC_CheckFoldableChainNode,\n";
Chris Lattnere39650a2010-02-16 06:10:58 +0000290 return 1;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000291 case Matcher::CheckChainCompatible:
Chris Lattner9a747f12010-02-17 06:23:39 +0000292 OS << "OPC_CheckChainCompatible, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000293 << cast<CheckChainCompatibleMatcher>(N)->getPreviousOp() << ",\n";
Chris Lattner9a747f12010-02-17 06:23:39 +0000294 return 2;
Chris Lattner845c0422010-02-18 22:03:03 +0000295
Chris Lattnerb21ba712010-02-25 02:04:40 +0000296 case Matcher::EmitInteger: {
297 int64_t Val = cast<EmitIntegerMatcher>(N)->getValue();
Chris Lattner53106542010-02-28 22:14:32 +0000298 OS << "OPC_EmitInteger, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000299 << getEnumName(cast<EmitIntegerMatcher>(N)->getVT()) << ", ";
Chris Lattner53106542010-02-28 22:14:32 +0000300 return 2+EmitVBRValue(Val, OS);
Chris Lattner906b4992010-02-19 07:49:56 +0000301 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000302 case Matcher::EmitStringInteger: {
303 const std::string &Val = cast<EmitStringIntegerMatcher>(N)->getValue();
Chris Lattner8e946be2010-02-21 03:22:59 +0000304 // These should always fit into one byte.
Chris Lattner53106542010-02-28 22:14:32 +0000305 OS << "OPC_EmitInteger, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000306 << getEnumName(cast<EmitStringIntegerMatcher>(N)->getVT()) << ", "
Chris Lattner8e946be2010-02-21 03:22:59 +0000307 << Val << ",\n";
308 return 3;
309 }
Chris Lattner906b4992010-02-19 07:49:56 +0000310
Chris Lattnerb21ba712010-02-25 02:04:40 +0000311 case Matcher::EmitRegister:
Chris Lattner906b4992010-02-19 07:49:56 +0000312 OS << "OPC_EmitRegister, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000313 << getEnumName(cast<EmitRegisterMatcher>(N)->getVT()) << ", ";
314 if (Record *R = cast<EmitRegisterMatcher>(N)->getReg())
Chris Lattner906b4992010-02-19 07:49:56 +0000315 OS << getQualifiedName(R) << ",\n";
316 else
317 OS << "0 /*zero_reg*/,\n";
318 return 3;
Chris Lattner8e946be2010-02-21 03:22:59 +0000319
Chris Lattnerb21ba712010-02-25 02:04:40 +0000320 case Matcher::EmitConvertToTarget:
Chris Lattner8e946be2010-02-21 03:22:59 +0000321 OS << "OPC_EmitConvertToTarget, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000322 << cast<EmitConvertToTargetMatcher>(N)->getSlot() << ",\n";
Chris Lattner8e946be2010-02-21 03:22:59 +0000323 return 2;
324
Chris Lattnerb21ba712010-02-25 02:04:40 +0000325 case Matcher::EmitMergeInputChains: {
326 const EmitMergeInputChainsMatcher *MN =
327 cast<EmitMergeInputChainsMatcher>(N);
Chris Lattner8e946be2010-02-21 03:22:59 +0000328 OS << "OPC_EmitMergeInputChains, " << MN->getNumNodes() << ", ";
329 for (unsigned i = 0, e = MN->getNumNodes(); i != e; ++i)
330 OS << MN->getNode(i) << ", ";
331 OS << '\n';
332 return 2+MN->getNumNodes();
333 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000334 case Matcher::EmitCopyToReg:
Chris Lattner8e946be2010-02-21 03:22:59 +0000335 OS << "OPC_EmitCopyToReg, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000336 << cast<EmitCopyToRegMatcher>(N)->getSrcSlot() << ", "
337 << getQualifiedName(cast<EmitCopyToRegMatcher>(N)->getDestPhysReg())
Chris Lattner8e946be2010-02-21 03:22:59 +0000338 << ",\n";
339 return 3;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000340 case Matcher::EmitNodeXForm: {
341 const EmitNodeXFormMatcher *XF = cast<EmitNodeXFormMatcher>(N);
Chris Lattner8e946be2010-02-21 03:22:59 +0000342 OS << "OPC_EmitNodeXForm, " << getNodeXFormID(XF->getNodeXForm()) << ", "
343 << XF->getSlot() << ',';
344 OS.PadToColumn(CommentIndent) << "// "<<XF->getNodeXForm()->getName()<<'\n';
345 return 3;
346 }
347
Chris Lattnere86097a2010-02-28 02:31:26 +0000348 case Matcher::EmitNode:
Chris Lattner9a215002010-02-28 20:55:18 +0000349 case Matcher::MorphNodeTo: {
Chris Lattnere86097a2010-02-28 02:31:26 +0000350 const EmitNodeMatcherCommon *EN = cast<EmitNodeMatcherCommon>(N);
Chris Lattner9a215002010-02-28 20:55:18 +0000351 OS << (isa<EmitNodeMatcher>(EN) ? "OPC_EmitNode" : "OPC_MorphNodeTo");
Chris Lattnere86097a2010-02-28 02:31:26 +0000352 OS << ", TARGET_OPCODE(" << EN->getOpcodeName() << "), 0";
Chris Lattner8e946be2010-02-21 03:22:59 +0000353
354 if (EN->hasChain()) OS << "|OPFL_Chain";
Chris Lattnerff7fb602010-02-28 21:53:42 +0000355 if (EN->hasInFlag()) OS << "|OPFL_FlagInput";
356 if (EN->hasOutFlag()) OS << "|OPFL_FlagOutput";
Chris Lattner8e946be2010-02-21 03:22:59 +0000357 if (EN->hasMemRefs()) OS << "|OPFL_MemRefs";
358 if (EN->getNumFixedArityOperands() != -1)
359 OS << "|OPFL_Variadic" << EN->getNumFixedArityOperands();
360 OS << ",\n";
361
362 OS.PadToColumn(Indent*2+4) << EN->getNumVTs() << "/*#VTs*/, ";
363 for (unsigned i = 0, e = EN->getNumVTs(); i != e; ++i)
364 OS << getEnumName(EN->getVT(i)) << ", ";
365
366 OS << EN->getNumOperands() << "/*#Ops*/, ";
Chris Lattner5007e1e2010-02-23 00:59:59 +0000367 unsigned NumOperandBytes = 0;
368 for (unsigned i = 0, e = EN->getNumOperands(); i != e; ++i) {
369 // We emit the operand numbers in VBR encoded format, in case the number
370 // is too large to represent with a byte.
371 NumOperandBytes += EmitVBRValue(EN->getOperand(i), OS);
372 }
Chris Lattner6281cda2010-02-28 02:41:25 +0000373
374 // Print the result #'s for EmitNode.
375 if (const EmitNodeMatcher *E = dyn_cast<EmitNodeMatcher>(EN)) {
Chris Lattnerccd23cc2010-02-28 23:00:47 +0000376 if (unsigned NumResults = EN->getNumVTs()) {
Chris Lattner6281cda2010-02-28 02:41:25 +0000377 OS.PadToColumn(CommentIndent) << "// Results = ";
378 unsigned First = E->getFirstResultSlot();
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000379 for (unsigned i = 0; i != NumResults; ++i)
Chris Lattner6281cda2010-02-28 02:41:25 +0000380 OS << "#" << First+i << " ";
Chris Lattner6281cda2010-02-28 02:41:25 +0000381 }
382 }
Chris Lattner8e946be2010-02-21 03:22:59 +0000383 OS << '\n';
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000384
Chris Lattner9a215002010-02-28 20:55:18 +0000385 if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(N)) {
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000386 OS.PadToColumn(Indent*2) << "// Src: "
387 << *SNT->getPattern().getSrcPattern() << '\n';
388 OS.PadToColumn(Indent*2) << "// Dst: "
389 << *SNT->getPattern().getDstPattern() << '\n';
390
391 }
392
Chris Lattner5007e1e2010-02-23 00:59:59 +0000393 return 6+EN->getNumVTs()+NumOperandBytes;
Chris Lattner8e946be2010-02-21 03:22:59 +0000394 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000395 case Matcher::MarkFlagResults: {
396 const MarkFlagResultsMatcher *CFR = cast<MarkFlagResultsMatcher>(N);
Chris Lattner02f73582010-02-24 05:33:42 +0000397 OS << "OPC_MarkFlagResults, " << CFR->getNumNodes() << ", ";
398 unsigned NumOperandBytes = 0;
399 for (unsigned i = 0, e = CFR->getNumNodes(); i != e; ++i)
400 NumOperandBytes += EmitVBRValue(CFR->getNode(i), OS);
401 OS << '\n';
402 return 2+NumOperandBytes;
403 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000404 case Matcher::CompleteMatch: {
405 const CompleteMatchMatcher *CM = cast<CompleteMatchMatcher>(N);
Chris Lattner77f2e272010-02-21 06:03:07 +0000406 OS << "OPC_CompleteMatch, " << CM->getNumResults() << ", ";
Chris Lattner5007e1e2010-02-23 00:59:59 +0000407 unsigned NumResultBytes = 0;
Chris Lattner77f2e272010-02-21 06:03:07 +0000408 for (unsigned i = 0, e = CM->getNumResults(); i != e; ++i)
Chris Lattner5007e1e2010-02-23 00:59:59 +0000409 NumResultBytes += EmitVBRValue(CM->getResult(i), OS);
Chris Lattner77f2e272010-02-21 06:03:07 +0000410 OS << '\n';
411 OS.PadToColumn(Indent*2) << "// Src: "
412 << *CM->getPattern().getSrcPattern() << '\n';
413 OS.PadToColumn(Indent*2) << "// Dst: "
414 << *CM->getPattern().getDstPattern() << '\n';
Chris Lattner5007e1e2010-02-23 00:59:59 +0000415 return 2 + NumResultBytes;
Chris Lattnerda272d12010-02-15 08:04:42 +0000416 }
Chris Lattner77f2e272010-02-21 06:03:07 +0000417 }
Chris Lattnerda272d12010-02-15 08:04:42 +0000418 assert(0 && "Unreachable");
419 return 0;
420}
421
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000422/// EmitMatcherList - Emit the bytes for the specified matcher subtree.
Chris Lattnere02ea542010-02-16 06:52:01 +0000423unsigned MatcherTableEmitter::
Chris Lattnerb21ba712010-02-25 02:04:40 +0000424EmitMatcherList(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
Chris Lattner984e1882010-02-21 06:30:04 +0000425 formatted_raw_ostream &OS) {
Chris Lattnerda272d12010-02-15 08:04:42 +0000426 unsigned Size = 0;
Chris Lattner8ef9c792010-02-18 02:49:24 +0000427 while (N) {
Chris Lattner09b9f392010-02-24 19:17:12 +0000428 if (unsigned(N->getKind()) >= Histogram.size())
429 Histogram.resize(N->getKind()+1);
430 Histogram[N->getKind()]++;
431
Chris Lattner8fbad242010-02-21 07:16:41 +0000432 OS << "/*" << CurrentIdx << "*/";
Chris Lattnerd6c84722010-02-25 19:00:39 +0000433 unsigned MatcherSize = EmitMatcher(N, Indent, CurrentIdx, OS);
Chris Lattner8fbad242010-02-21 07:16:41 +0000434 Size += MatcherSize;
435 CurrentIdx += MatcherSize;
Chris Lattnerda272d12010-02-15 08:04:42 +0000436
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000437 // If there are other nodes in this list, iterate to them, otherwise we're
Chris Lattnerda272d12010-02-15 08:04:42 +0000438 // done.
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000439 N = N->getNext();
Chris Lattnerda272d12010-02-15 08:04:42 +0000440 }
Chris Lattner8ef9c792010-02-18 02:49:24 +0000441 return Size;
Chris Lattnerda272d12010-02-15 08:04:42 +0000442}
443
Chris Lattner4d0c9312010-03-01 01:54:19 +0000444void MatcherTableEmitter::EmitPredicateFunctions(const CodeGenDAGPatterns &CGP,
445 formatted_raw_ostream &OS) {
Chris Lattnerb49985a2010-02-18 06:47:49 +0000446 // FIXME: Don't build off the DAGISelEmitter's predicates, emit them directly
447 // here into the case stmts.
448
Chris Lattnere609a512010-02-17 00:31:50 +0000449 // Emit pattern predicates.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000450 if (!PatternPredicates.empty()) {
451 OS << "bool CheckPatternPredicate(unsigned PredNo) const {\n";
452 OS << " switch (PredNo) {\n";
453 OS << " default: assert(0 && \"Invalid predicate in table?\");\n";
454 for (unsigned i = 0, e = PatternPredicates.size(); i != e; ++i)
455 OS << " case " << i << ": return " << PatternPredicates[i] << ";\n";
456 OS << " }\n";
457 OS << "}\n\n";
458 }
Chris Lattner4d0c9312010-03-01 01:54:19 +0000459
Chris Lattnere609a512010-02-17 00:31:50 +0000460 // Emit Node predicates.
Chris Lattner4d0c9312010-03-01 01:54:19 +0000461 // FIXME: Annoyingly, these are stored by name, which we never even emit. Yay?
462 StringMap<TreePattern*> PFsByName;
463
464 for (CodeGenDAGPatterns::pf_iterator I = CGP.pf_begin(), E = CGP.pf_end();
465 I != E; ++I)
466 PFsByName[I->first->getName()] = I->second;
467
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000468 if (!NodePredicates.empty()) {
Chris Lattner4d0c9312010-03-01 01:54:19 +0000469 OS << "bool CheckNodePredicate(SDNode *Node, unsigned PredNo) const {\n";
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000470 OS << " switch (PredNo) {\n";
471 OS << " default: assert(0 && \"Invalid predicate in table?\");\n";
Chris Lattner4d0c9312010-03-01 01:54:19 +0000472 for (unsigned i = 0, e = NodePredicates.size(); i != e; ++i) {
473 // FIXME: Storing this by name is horrible.
474 TreePattern *P =PFsByName[NodePredicates[i].substr(strlen("Predicate_"))];
475 assert(P && "Unknown name?");
476
477 // Emit the predicate code corresponding to this pattern.
478 std::string Code = P->getRecord()->getValueAsCode("Predicate");
479 assert(!Code.empty() && "No code in this predicate");
480 OS << " case " << i << ": { // " << NodePredicates[i] << '\n';
481 std::string ClassName;
482 if (P->getOnlyTree()->isLeaf())
483 ClassName = "SDNode";
484 else
485 ClassName =
486 CGP.getSDNodeInfo(P->getOnlyTree()->getOperator()).getSDClassName();
487 if (ClassName == "SDNode")
488 OS << " SDNode *N = Node;\n";
489 else
490 OS << " " << ClassName << "*N = cast<" << ClassName << ">(Node);\n";
491 OS << Code << "\n }\n";
492 }
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000493 OS << " }\n";
494 OS << "}\n\n";
495 }
Chris Lattnere609a512010-02-17 00:31:50 +0000496
497 // Emit CompletePattern matchers.
Chris Lattner8e946be2010-02-21 03:22:59 +0000498 // FIXME: This should be const.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000499 if (!ComplexPatterns.empty()) {
500 OS << "bool CheckComplexPattern(SDNode *Root, SDValue N,\n";
501 OS << " unsigned PatternNo, SmallVectorImpl<SDValue> &Result) {\n";
502 OS << " switch (PatternNo) {\n";
503 OS << " default: assert(0 && \"Invalid pattern # in table?\");\n";
504 for (unsigned i = 0, e = ComplexPatterns.size(); i != e; ++i) {
505 const ComplexPattern &P = *ComplexPatterns[i];
506 unsigned NumOps = P.getNumOperands();
Chris Lattner8e946be2010-02-21 03:22:59 +0000507
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000508 if (P.hasProperty(SDNPHasChain))
509 ++NumOps; // Get the chained node too.
510
511 OS << " case " << i << ":\n";
512 OS << " Result.resize(Result.size()+" << NumOps << ");\n";
513 OS << " return " << P.getSelectFunc();
Chris Lattner8e946be2010-02-21 03:22:59 +0000514
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000515 // FIXME: Temporary hack until old isel dies.
516 if (P.hasProperty(SDNPHasChain))
517 OS << "XXX";
518
519 OS << "(Root, N";
520 for (unsigned i = 0; i != NumOps; ++i)
521 OS << ", Result[Result.size()-" << (NumOps-i) << ']';
522 OS << ");\n";
523 }
524 OS << " }\n";
525 OS << "}\n\n";
Chris Lattnere609a512010-02-17 00:31:50 +0000526 }
Chris Lattner8e946be2010-02-21 03:22:59 +0000527
Chris Lattner4d0c9312010-03-01 01:54:19 +0000528
Chris Lattner8e946be2010-02-21 03:22:59 +0000529 // Emit SDNodeXForm handlers.
530 // FIXME: This should be const.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000531 if (!NodeXForms.empty()) {
532 OS << "SDValue RunSDNodeXForm(SDValue V, unsigned XFormNo) {\n";
533 OS << " switch (XFormNo) {\n";
534 OS << " default: assert(0 && \"Invalid xform # in table?\");\n";
535
536 // FIXME: The node xform could take SDValue's instead of SDNode*'s.
Chris Lattner4d0c9312010-03-01 01:54:19 +0000537 for (unsigned i = 0, e = NodeXForms.size(); i != e; ++i) {
538 const CodeGenDAGPatterns::NodeXForm &Entry =
539 CGP.getSDNodeTransform(NodeXForms[i]);
540
541 Record *SDNode = Entry.first;
542 const std::string &Code = Entry.second;
543
544 OS << " case " << i << ": { // " << NodeXForms[i]->getName() << '\n';
545
546 std::string ClassName = CGP.getSDNodeInfo(SDNode).getSDClassName();
547 if (ClassName == "SDNode")
548 OS << " SDNode *N = V.getNode();\n";
549 else
550 OS << " " << ClassName << " *N = cast<" << ClassName
551 << ">(V.getNode());\n";
552 OS << Code << "\n }\n";
553 }
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000554 OS << " }\n";
555 OS << "}\n\n";
556 }
Chris Lattner050a03d2010-02-16 07:21:10 +0000557}
558
Chris Lattner09b9f392010-02-24 19:17:12 +0000559void MatcherTableEmitter::EmitHistogram(formatted_raw_ostream &OS) {
560 OS << " // Opcode Histogram:\n";
561 for (unsigned i = 0, e = Histogram.size(); i != e; ++i) {
562 OS << " // #";
Chris Lattnerb21ba712010-02-25 02:04:40 +0000563 switch ((Matcher::KindTy)i) {
564 case Matcher::Scope: OS << "OPC_Scope"; break;
565 case Matcher::RecordNode: OS << "OPC_RecordNode"; break;
566 case Matcher::RecordChild: OS << "OPC_RecordChild"; break;
567 case Matcher::RecordMemRef: OS << "OPC_RecordMemRef"; break;
568 case Matcher::CaptureFlagInput: OS << "OPC_CaptureFlagInput"; break;
569 case Matcher::MoveChild: OS << "OPC_MoveChild"; break;
570 case Matcher::MoveParent: OS << "OPC_MoveParent"; break;
571 case Matcher::CheckSame: OS << "OPC_CheckSame"; break;
572 case Matcher::CheckPatternPredicate:
Chris Lattner09b9f392010-02-24 19:17:12 +0000573 OS << "OPC_CheckPatternPredicate"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000574 case Matcher::CheckPredicate: OS << "OPC_CheckPredicate"; break;
575 case Matcher::CheckOpcode: OS << "OPC_CheckOpcode"; break;
576 case Matcher::CheckMultiOpcode: OS << "OPC_CheckMultiOpcode"; break;
577 case Matcher::CheckType: OS << "OPC_CheckType"; break;
578 case Matcher::CheckChildType: OS << "OPC_CheckChildType"; break;
579 case Matcher::CheckInteger: OS << "OPC_CheckInteger"; break;
580 case Matcher::CheckCondCode: OS << "OPC_CheckCondCode"; break;
581 case Matcher::CheckValueType: OS << "OPC_CheckValueType"; break;
582 case Matcher::CheckComplexPat: OS << "OPC_CheckComplexPat"; break;
583 case Matcher::CheckAndImm: OS << "OPC_CheckAndImm"; break;
584 case Matcher::CheckOrImm: OS << "OPC_CheckOrImm"; break;
585 case Matcher::CheckFoldableChainNode:
Chris Lattner09b9f392010-02-24 19:17:12 +0000586 OS << "OPC_CheckFoldableChainNode"; break;
Chris Lattner499622b2010-02-25 02:09:00 +0000587 case Matcher::CheckChainCompatible: OS << "OPC_CheckChainCompatible"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000588 case Matcher::EmitInteger: OS << "OPC_EmitInteger"; break;
589 case Matcher::EmitStringInteger: OS << "OPC_EmitStringInteger"; break;
590 case Matcher::EmitRegister: OS << "OPC_EmitRegister"; break;
Chris Lattner499622b2010-02-25 02:09:00 +0000591 case Matcher::EmitConvertToTarget: OS << "OPC_EmitConvertToTarget"; break;
592 case Matcher::EmitMergeInputChains: OS << "OPC_EmitMergeInputChains"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000593 case Matcher::EmitCopyToReg: OS << "OPC_EmitCopyToReg"; break;
594 case Matcher::EmitNode: OS << "OPC_EmitNode"; break;
Chris Lattner9a215002010-02-28 20:55:18 +0000595 case Matcher::MorphNodeTo: OS << "OPC_MorphNodeTo"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000596 case Matcher::EmitNodeXForm: OS << "OPC_EmitNodeXForm"; break;
597 case Matcher::MarkFlagResults: OS << "OPC_MarkFlagResults"; break;
598 case Matcher::CompleteMatch: OS << "OPC_CompleteMatch"; break;
Chris Lattner09b9f392010-02-24 19:17:12 +0000599 }
600
601 OS.PadToColumn(40) << " = " << Histogram[i] << '\n';
602 }
603 OS << '\n';
604}
605
Chris Lattner050a03d2010-02-16 07:21:10 +0000606
Chris Lattner4d0c9312010-03-01 01:54:19 +0000607void llvm::EmitMatcherTable(const Matcher *TheMatcher,
608 const CodeGenDAGPatterns &CGP, raw_ostream &O) {
Chris Lattnerda272d12010-02-15 08:04:42 +0000609 formatted_raw_ostream OS(O);
610
611 OS << "// The main instruction selector code.\n";
Chris Lattnerc84edb72010-02-24 07:35:09 +0000612 OS << "SDNode *SelectCode(SDNode *N) {\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000613
Chris Lattner984e1882010-02-21 06:30:04 +0000614 MatcherTableEmitter MatcherEmitter;
Chris Lattnere02ea542010-02-16 06:52:01 +0000615
Chris Lattner8e946be2010-02-21 03:22:59 +0000616 OS << " // Opcodes are emitted as 2 bytes, TARGET_OPCODE handles this.\n";
617 OS << " #define TARGET_OPCODE(X) X & 255, unsigned(X) >> 8\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000618 OS << " static const unsigned char MatcherTable[] = {\n";
Chris Lattnerb21ba712010-02-25 02:04:40 +0000619 unsigned TotalSize = MatcherEmitter.EmitMatcherList(TheMatcher, 5, 0, OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000620 OS << " 0\n }; // Total Array size is " << (TotalSize+1) << " bytes\n\n";
Chris Lattner09b9f392010-02-24 19:17:12 +0000621
622 MatcherEmitter.EmitHistogram(OS);
623
Chris Lattner8e946be2010-02-21 03:22:59 +0000624 OS << " #undef TARGET_OPCODE\n";
Chris Lattnere02ea542010-02-16 06:52:01 +0000625 OS << " return SelectCodeCommon(N, MatcherTable,sizeof(MatcherTable));\n}\n";
Chris Lattner050a03d2010-02-16 07:21:10 +0000626 OS << "\n";
627
628 // Next up, emit the function for node and pattern predicates:
Chris Lattner4d0c9312010-03-01 01:54:19 +0000629 MatcherEmitter.EmitPredicateFunctions(CGP, OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000630}