blob: 279f036b364324137bcdc466ad88f9ff0a6de2e8 [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);
Chris Lattnereb669212010-03-01 06:59:22 +0000161 ChildSize = EmitMatcherList(SM->getChild(i), Indent+1,
162 CurrentIdx+VBRSize, FOS);
Chris Lattnerd6c84722010-02-25 19:00:39 +0000163 } 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 Lattner0cebe612010-03-01 02:24:17 +0000186 OS.PadToColumn(CommentIndent) << "// #"
187 << cast<RecordMatcher>(N)->getResultNo() << " = "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000188 << cast<RecordMatcher>(N)->getWhatFor() << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000189 return 1;
Chris Lattner19b5a752010-02-24 07:31:45 +0000190
Chris Lattnerb21ba712010-02-25 02:04:40 +0000191 case Matcher::RecordChild:
192 OS << "OPC_RecordChild" << cast<RecordChildMatcher>(N)->getChildNo()
Chris Lattner19b5a752010-02-24 07:31:45 +0000193 << ',';
Chris Lattner0cebe612010-03-01 02:24:17 +0000194 OS.PadToColumn(CommentIndent) << "// #"
195 << cast<RecordChildMatcher>(N)->getResultNo() << " = "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000196 << cast<RecordChildMatcher>(N)->getWhatFor() << '\n';
Chris Lattner19b5a752010-02-24 07:31:45 +0000197 return 1;
Chris Lattner8e946be2010-02-21 03:22:59 +0000198
Chris Lattnerb21ba712010-02-25 02:04:40 +0000199 case Matcher::RecordMemRef:
Chris Lattner8e946be2010-02-21 03:22:59 +0000200 OS << "OPC_RecordMemRef,\n";
201 return 1;
202
Chris Lattnerb21ba712010-02-25 02:04:40 +0000203 case Matcher::CaptureFlagInput:
Chris Lattner8e946be2010-02-21 03:22:59 +0000204 OS << "OPC_CaptureFlagInput,\n";
205 return 1;
206
Chris Lattnerb21ba712010-02-25 02:04:40 +0000207 case Matcher::MoveChild:
208 OS << "OPC_MoveChild, " << cast<MoveChildMatcher>(N)->getChildNo() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000209 return 2;
210
Chris Lattnerb21ba712010-02-25 02:04:40 +0000211 case Matcher::MoveParent:
Chris Lattnerda272d12010-02-15 08:04:42 +0000212 OS << "OPC_MoveParent,\n";
213 return 1;
214
Chris Lattnerb21ba712010-02-25 02:04:40 +0000215 case Matcher::CheckSame:
Chris Lattnerda272d12010-02-15 08:04:42 +0000216 OS << "OPC_CheckSame, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000217 << cast<CheckSameMatcher>(N)->getMatchNumber() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000218 return 2;
219
Chris Lattnerb21ba712010-02-25 02:04:40 +0000220 case Matcher::CheckPatternPredicate: {
221 StringRef Pred = cast<CheckPatternPredicateMatcher>(N)->getPredicate();
Chris Lattner050a03d2010-02-16 07:21:10 +0000222 OS << "OPC_CheckPatternPredicate, " << getPatternPredicate(Pred) << ',';
223 OS.PadToColumn(CommentIndent) << "// " << Pred << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000224 return 2;
Chris Lattner050a03d2010-02-16 07:21:10 +0000225 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000226 case Matcher::CheckPredicate: {
227 StringRef Pred = cast<CheckPredicateMatcher>(N)->getPredicateName();
Chris Lattner050a03d2010-02-16 07:21:10 +0000228 OS << "OPC_CheckPredicate, " << getNodePredicate(Pred) << ',';
229 OS.PadToColumn(CommentIndent) << "// " << Pred << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000230 return 2;
Chris Lattner050a03d2010-02-16 07:21:10 +0000231 }
232
Chris Lattnerb21ba712010-02-25 02:04:40 +0000233 case Matcher::CheckOpcode:
Chris Lattnerda272d12010-02-15 08:04:42 +0000234 OS << "OPC_CheckOpcode, "
Chris Lattnera230f962010-02-27 21:48:43 +0000235 << cast<CheckOpcodeMatcher>(N)->getOpcode().getEnumName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000236 return 2;
237
Chris Lattnereb669212010-03-01 06:59:22 +0000238 case Matcher::SwitchOpcode: {
239 unsigned StartIdx = CurrentIdx;
240 const SwitchOpcodeMatcher *SOM = cast<SwitchOpcodeMatcher>(N);
241 OS << "OPC_SwitchOpcode /*" << SOM->getNumCases() << " cases */, ";
242 ++CurrentIdx;
243
244 // For each case we emit the size, then the opcode, then the matcher.
245 for (unsigned i = 0, e = SOM->getNumCases(); i != e; ++i) {
246 // We need to encode the opcode and the offset of the case code before
247 // emitting the case code. Handle this by buffering the output into a
248 // string while we get the size. Unfortunately, the offset of the
249 // children depends on the VBR size of the child, so for large children we
250 // have to iterate a bit.
251 SmallString<128> TmpBuf;
252 unsigned ChildSize = 0;
253 unsigned VBRSize = 0;
254 do {
255 VBRSize = GetVBRSize(ChildSize);
256
257 TmpBuf.clear();
258 raw_svector_ostream OS(TmpBuf);
259 formatted_raw_ostream FOS(OS);
260 ChildSize = EmitMatcherList(SOM->getCaseMatcher(i),
261 Indent+1, CurrentIdx+VBRSize+1, FOS);
262 } while (GetVBRSize(ChildSize) != VBRSize);
263
264 assert(ChildSize != 0 && "Should not have a zero-sized child!");
265
266 if (i != 0)
267 OS.PadToColumn(Indent*2) << "/*SwitchOpcode*/ ";
268
269 // Emit the VBR.
270 CurrentIdx += EmitVBRValue(ChildSize, OS);
271
272 OS << " " << SOM->getCaseOpcode(i).getEnumName() << ",";
273 OS << "// ->" << CurrentIdx+ChildSize+1 << '\n';
274 ++CurrentIdx;
275 OS << TmpBuf.str();
276 CurrentIdx += ChildSize;
277 }
278
279 // Emit the final zero to terminate the switch.
280 OS.PadToColumn(Indent*2) << "0, // EndSwitchOpcode\n";
281 ++CurrentIdx;
282 return CurrentIdx-StartIdx;
283 }
284
Chris Lattnerfa342fa2010-03-01 07:17:40 +0000285 case Matcher::CheckType:
Chris Lattnerda272d12010-02-15 08:04:42 +0000286 OS << "OPC_CheckType, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000287 << getEnumName(cast<CheckTypeMatcher>(N)->getType()) << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000288 return 2;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000289 case Matcher::CheckChildType:
Chris Lattner23cfda72010-02-24 20:15:25 +0000290 OS << "OPC_CheckChild"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000291 << cast<CheckChildTypeMatcher>(N)->getChildNo() << "Type, "
292 << getEnumName(cast<CheckChildTypeMatcher>(N)->getType()) << ",\n";
Chris Lattner23cfda72010-02-24 20:15:25 +0000293 return 2;
294
Chris Lattner53106542010-02-28 22:14:32 +0000295 case Matcher::CheckInteger:
296 OS << "OPC_CheckInteger, ";
297 return 1+EmitVBRValue(cast<CheckIntegerMatcher>(N)->getValue(), OS);
Chris Lattnerb21ba712010-02-25 02:04:40 +0000298 case Matcher::CheckCondCode:
Chris Lattnerda272d12010-02-15 08:04:42 +0000299 OS << "OPC_CheckCondCode, ISD::"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000300 << cast<CheckCondCodeMatcher>(N)->getCondCodeName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000301 return 2;
302
Chris Lattnerb21ba712010-02-25 02:04:40 +0000303 case Matcher::CheckValueType:
Chris Lattnerda272d12010-02-15 08:04:42 +0000304 OS << "OPC_CheckValueType, MVT::"
Chris Lattnerb21ba712010-02-25 02:04:40 +0000305 << cast<CheckValueTypeMatcher>(N)->getTypeName() << ",\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000306 return 2;
307
Chris Lattnerb21ba712010-02-25 02:04:40 +0000308 case Matcher::CheckComplexPat: {
Chris Lattner5be6e592010-02-17 00:39:26 +0000309 const ComplexPattern &Pattern =
Chris Lattnerb21ba712010-02-25 02:04:40 +0000310 cast<CheckComplexPatMatcher>(N)->getPattern();
Chris Lattner5be6e592010-02-17 00:39:26 +0000311 OS << "OPC_CheckComplexPat, " << getComplexPat(Pattern) << ',';
Chris Lattner781f3592010-02-17 06:47:35 +0000312 OS.PadToColumn(CommentIndent) << "// " << Pattern.getSelectFunc();
313 OS << ": " << Pattern.getNumOperands() << " operands";
314 if (Pattern.hasProperty(SDNPHasChain))
315 OS << " + chain result and input";
316 OS << '\n';
Chris Lattnerda272d12010-02-15 08:04:42 +0000317 return 2;
Chris Lattner5be6e592010-02-17 00:39:26 +0000318 }
Chris Lattnerda272d12010-02-15 08:04:42 +0000319
Chris Lattner53106542010-02-28 22:14:32 +0000320 case Matcher::CheckAndImm:
321 OS << "OPC_CheckAndImm, ";
322 return 1+EmitVBRValue(cast<CheckAndImmMatcher>(N)->getValue(), OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000323
Chris Lattner53106542010-02-28 22:14:32 +0000324 case Matcher::CheckOrImm:
325 OS << "OPC_CheckOrImm, ";
326 return 1+EmitVBRValue(cast<CheckOrImmMatcher>(N)->getValue(), OS);
327
Chris Lattnerb21ba712010-02-25 02:04:40 +0000328 case Matcher::CheckFoldableChainNode:
Chris Lattner21390d72010-02-16 19:15:55 +0000329 OS << "OPC_CheckFoldableChainNode,\n";
Chris Lattnere39650a2010-02-16 06:10:58 +0000330 return 1;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000331 case Matcher::CheckChainCompatible:
Chris Lattner9a747f12010-02-17 06:23:39 +0000332 OS << "OPC_CheckChainCompatible, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000333 << cast<CheckChainCompatibleMatcher>(N)->getPreviousOp() << ",\n";
Chris Lattner9a747f12010-02-17 06:23:39 +0000334 return 2;
Chris Lattner845c0422010-02-18 22:03:03 +0000335
Chris Lattnerb21ba712010-02-25 02:04:40 +0000336 case Matcher::EmitInteger: {
337 int64_t Val = cast<EmitIntegerMatcher>(N)->getValue();
Chris Lattner53106542010-02-28 22:14:32 +0000338 OS << "OPC_EmitInteger, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000339 << getEnumName(cast<EmitIntegerMatcher>(N)->getVT()) << ", ";
Chris Lattner53106542010-02-28 22:14:32 +0000340 return 2+EmitVBRValue(Val, OS);
Chris Lattner906b4992010-02-19 07:49:56 +0000341 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000342 case Matcher::EmitStringInteger: {
343 const std::string &Val = cast<EmitStringIntegerMatcher>(N)->getValue();
Chris Lattner8e946be2010-02-21 03:22:59 +0000344 // These should always fit into one byte.
Chris Lattner53106542010-02-28 22:14:32 +0000345 OS << "OPC_EmitInteger, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000346 << getEnumName(cast<EmitStringIntegerMatcher>(N)->getVT()) << ", "
Chris Lattner8e946be2010-02-21 03:22:59 +0000347 << Val << ",\n";
348 return 3;
349 }
Chris Lattner906b4992010-02-19 07:49:56 +0000350
Chris Lattnerb21ba712010-02-25 02:04:40 +0000351 case Matcher::EmitRegister:
Chris Lattner906b4992010-02-19 07:49:56 +0000352 OS << "OPC_EmitRegister, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000353 << getEnumName(cast<EmitRegisterMatcher>(N)->getVT()) << ", ";
354 if (Record *R = cast<EmitRegisterMatcher>(N)->getReg())
Chris Lattner906b4992010-02-19 07:49:56 +0000355 OS << getQualifiedName(R) << ",\n";
356 else
357 OS << "0 /*zero_reg*/,\n";
358 return 3;
Chris Lattner8e946be2010-02-21 03:22:59 +0000359
Chris Lattnerb21ba712010-02-25 02:04:40 +0000360 case Matcher::EmitConvertToTarget:
Chris Lattner8e946be2010-02-21 03:22:59 +0000361 OS << "OPC_EmitConvertToTarget, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000362 << cast<EmitConvertToTargetMatcher>(N)->getSlot() << ",\n";
Chris Lattner8e946be2010-02-21 03:22:59 +0000363 return 2;
364
Chris Lattnerb21ba712010-02-25 02:04:40 +0000365 case Matcher::EmitMergeInputChains: {
366 const EmitMergeInputChainsMatcher *MN =
367 cast<EmitMergeInputChainsMatcher>(N);
Chris Lattner8e946be2010-02-21 03:22:59 +0000368 OS << "OPC_EmitMergeInputChains, " << MN->getNumNodes() << ", ";
369 for (unsigned i = 0, e = MN->getNumNodes(); i != e; ++i)
370 OS << MN->getNode(i) << ", ";
371 OS << '\n';
372 return 2+MN->getNumNodes();
373 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000374 case Matcher::EmitCopyToReg:
Chris Lattner8e946be2010-02-21 03:22:59 +0000375 OS << "OPC_EmitCopyToReg, "
Chris Lattnerb21ba712010-02-25 02:04:40 +0000376 << cast<EmitCopyToRegMatcher>(N)->getSrcSlot() << ", "
377 << getQualifiedName(cast<EmitCopyToRegMatcher>(N)->getDestPhysReg())
Chris Lattner8e946be2010-02-21 03:22:59 +0000378 << ",\n";
379 return 3;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000380 case Matcher::EmitNodeXForm: {
381 const EmitNodeXFormMatcher *XF = cast<EmitNodeXFormMatcher>(N);
Chris Lattner8e946be2010-02-21 03:22:59 +0000382 OS << "OPC_EmitNodeXForm, " << getNodeXFormID(XF->getNodeXForm()) << ", "
383 << XF->getSlot() << ',';
384 OS.PadToColumn(CommentIndent) << "// "<<XF->getNodeXForm()->getName()<<'\n';
385 return 3;
386 }
387
Chris Lattnere86097a2010-02-28 02:31:26 +0000388 case Matcher::EmitNode:
Chris Lattner9a215002010-02-28 20:55:18 +0000389 case Matcher::MorphNodeTo: {
Chris Lattnere86097a2010-02-28 02:31:26 +0000390 const EmitNodeMatcherCommon *EN = cast<EmitNodeMatcherCommon>(N);
Chris Lattner9a215002010-02-28 20:55:18 +0000391 OS << (isa<EmitNodeMatcher>(EN) ? "OPC_EmitNode" : "OPC_MorphNodeTo");
Chris Lattnere86097a2010-02-28 02:31:26 +0000392 OS << ", TARGET_OPCODE(" << EN->getOpcodeName() << "), 0";
Chris Lattner8e946be2010-02-21 03:22:59 +0000393
394 if (EN->hasChain()) OS << "|OPFL_Chain";
Chris Lattnerff7fb602010-02-28 21:53:42 +0000395 if (EN->hasInFlag()) OS << "|OPFL_FlagInput";
396 if (EN->hasOutFlag()) OS << "|OPFL_FlagOutput";
Chris Lattner8e946be2010-02-21 03:22:59 +0000397 if (EN->hasMemRefs()) OS << "|OPFL_MemRefs";
398 if (EN->getNumFixedArityOperands() != -1)
399 OS << "|OPFL_Variadic" << EN->getNumFixedArityOperands();
400 OS << ",\n";
401
402 OS.PadToColumn(Indent*2+4) << EN->getNumVTs() << "/*#VTs*/, ";
403 for (unsigned i = 0, e = EN->getNumVTs(); i != e; ++i)
404 OS << getEnumName(EN->getVT(i)) << ", ";
405
406 OS << EN->getNumOperands() << "/*#Ops*/, ";
Chris Lattner5007e1e2010-02-23 00:59:59 +0000407 unsigned NumOperandBytes = 0;
408 for (unsigned i = 0, e = EN->getNumOperands(); i != e; ++i) {
409 // We emit the operand numbers in VBR encoded format, in case the number
410 // is too large to represent with a byte.
411 NumOperandBytes += EmitVBRValue(EN->getOperand(i), OS);
412 }
Chris Lattner6281cda2010-02-28 02:41:25 +0000413
414 // Print the result #'s for EmitNode.
415 if (const EmitNodeMatcher *E = dyn_cast<EmitNodeMatcher>(EN)) {
Chris Lattnerccd23cc2010-02-28 23:00:47 +0000416 if (unsigned NumResults = EN->getNumVTs()) {
Chris Lattner6281cda2010-02-28 02:41:25 +0000417 OS.PadToColumn(CommentIndent) << "// Results = ";
418 unsigned First = E->getFirstResultSlot();
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000419 for (unsigned i = 0; i != NumResults; ++i)
Chris Lattner6281cda2010-02-28 02:41:25 +0000420 OS << "#" << First+i << " ";
Chris Lattner6281cda2010-02-28 02:41:25 +0000421 }
422 }
Chris Lattner8e946be2010-02-21 03:22:59 +0000423 OS << '\n';
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000424
Chris Lattner9a215002010-02-28 20:55:18 +0000425 if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(N)) {
Chris Lattnerc78f2a32010-02-28 20:49:53 +0000426 OS.PadToColumn(Indent*2) << "// Src: "
427 << *SNT->getPattern().getSrcPattern() << '\n';
428 OS.PadToColumn(Indent*2) << "// Dst: "
429 << *SNT->getPattern().getDstPattern() << '\n';
430
431 }
432
Chris Lattner5007e1e2010-02-23 00:59:59 +0000433 return 6+EN->getNumVTs()+NumOperandBytes;
Chris Lattner8e946be2010-02-21 03:22:59 +0000434 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000435 case Matcher::MarkFlagResults: {
436 const MarkFlagResultsMatcher *CFR = cast<MarkFlagResultsMatcher>(N);
Chris Lattner02f73582010-02-24 05:33:42 +0000437 OS << "OPC_MarkFlagResults, " << CFR->getNumNodes() << ", ";
438 unsigned NumOperandBytes = 0;
439 for (unsigned i = 0, e = CFR->getNumNodes(); i != e; ++i)
440 NumOperandBytes += EmitVBRValue(CFR->getNode(i), OS);
441 OS << '\n';
442 return 2+NumOperandBytes;
443 }
Chris Lattnerb21ba712010-02-25 02:04:40 +0000444 case Matcher::CompleteMatch: {
445 const CompleteMatchMatcher *CM = cast<CompleteMatchMatcher>(N);
Chris Lattner77f2e272010-02-21 06:03:07 +0000446 OS << "OPC_CompleteMatch, " << CM->getNumResults() << ", ";
Chris Lattner5007e1e2010-02-23 00:59:59 +0000447 unsigned NumResultBytes = 0;
Chris Lattner77f2e272010-02-21 06:03:07 +0000448 for (unsigned i = 0, e = CM->getNumResults(); i != e; ++i)
Chris Lattner5007e1e2010-02-23 00:59:59 +0000449 NumResultBytes += EmitVBRValue(CM->getResult(i), OS);
Chris Lattner77f2e272010-02-21 06:03:07 +0000450 OS << '\n';
451 OS.PadToColumn(Indent*2) << "// Src: "
452 << *CM->getPattern().getSrcPattern() << '\n';
453 OS.PadToColumn(Indent*2) << "// Dst: "
454 << *CM->getPattern().getDstPattern() << '\n';
Chris Lattner5007e1e2010-02-23 00:59:59 +0000455 return 2 + NumResultBytes;
Chris Lattnerda272d12010-02-15 08:04:42 +0000456 }
Chris Lattner77f2e272010-02-21 06:03:07 +0000457 }
Chris Lattnerda272d12010-02-15 08:04:42 +0000458 assert(0 && "Unreachable");
459 return 0;
460}
461
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000462/// EmitMatcherList - Emit the bytes for the specified matcher subtree.
Chris Lattnere02ea542010-02-16 06:52:01 +0000463unsigned MatcherTableEmitter::
Chris Lattnerb21ba712010-02-25 02:04:40 +0000464EmitMatcherList(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
Chris Lattner984e1882010-02-21 06:30:04 +0000465 formatted_raw_ostream &OS) {
Chris Lattnerda272d12010-02-15 08:04:42 +0000466 unsigned Size = 0;
Chris Lattner8ef9c792010-02-18 02:49:24 +0000467 while (N) {
Chris Lattner09b9f392010-02-24 19:17:12 +0000468 if (unsigned(N->getKind()) >= Histogram.size())
469 Histogram.resize(N->getKind()+1);
470 Histogram[N->getKind()]++;
471
Chris Lattner8fbad242010-02-21 07:16:41 +0000472 OS << "/*" << CurrentIdx << "*/";
Chris Lattnerd6c84722010-02-25 19:00:39 +0000473 unsigned MatcherSize = EmitMatcher(N, Indent, CurrentIdx, OS);
Chris Lattner8fbad242010-02-21 07:16:41 +0000474 Size += MatcherSize;
475 CurrentIdx += MatcherSize;
Chris Lattnerda272d12010-02-15 08:04:42 +0000476
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000477 // If there are other nodes in this list, iterate to them, otherwise we're
Chris Lattnerda272d12010-02-15 08:04:42 +0000478 // done.
Chris Lattnerbd8227f2010-02-18 02:53:41 +0000479 N = N->getNext();
Chris Lattnerda272d12010-02-15 08:04:42 +0000480 }
Chris Lattner8ef9c792010-02-18 02:49:24 +0000481 return Size;
Chris Lattnerda272d12010-02-15 08:04:42 +0000482}
483
Chris Lattner4d0c9312010-03-01 01:54:19 +0000484void MatcherTableEmitter::EmitPredicateFunctions(const CodeGenDAGPatterns &CGP,
485 formatted_raw_ostream &OS) {
Chris Lattnerb49985a2010-02-18 06:47:49 +0000486 // FIXME: Don't build off the DAGISelEmitter's predicates, emit them directly
487 // here into the case stmts.
488
Chris Lattnere609a512010-02-17 00:31:50 +0000489 // Emit pattern predicates.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000490 if (!PatternPredicates.empty()) {
491 OS << "bool CheckPatternPredicate(unsigned PredNo) const {\n";
492 OS << " switch (PredNo) {\n";
493 OS << " default: assert(0 && \"Invalid predicate in table?\");\n";
494 for (unsigned i = 0, e = PatternPredicates.size(); i != e; ++i)
495 OS << " case " << i << ": return " << PatternPredicates[i] << ";\n";
496 OS << " }\n";
497 OS << "}\n\n";
498 }
Chris Lattner4d0c9312010-03-01 01:54:19 +0000499
Chris Lattnere609a512010-02-17 00:31:50 +0000500 // Emit Node predicates.
Chris Lattner4d0c9312010-03-01 01:54:19 +0000501 // FIXME: Annoyingly, these are stored by name, which we never even emit. Yay?
502 StringMap<TreePattern*> PFsByName;
503
504 for (CodeGenDAGPatterns::pf_iterator I = CGP.pf_begin(), E = CGP.pf_end();
505 I != E; ++I)
506 PFsByName[I->first->getName()] = I->second;
507
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000508 if (!NodePredicates.empty()) {
Chris Lattner4d0c9312010-03-01 01:54:19 +0000509 OS << "bool CheckNodePredicate(SDNode *Node, unsigned PredNo) const {\n";
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000510 OS << " switch (PredNo) {\n";
511 OS << " default: assert(0 && \"Invalid predicate in table?\");\n";
Chris Lattner4d0c9312010-03-01 01:54:19 +0000512 for (unsigned i = 0, e = NodePredicates.size(); i != e; ++i) {
513 // FIXME: Storing this by name is horrible.
514 TreePattern *P =PFsByName[NodePredicates[i].substr(strlen("Predicate_"))];
515 assert(P && "Unknown name?");
516
517 // Emit the predicate code corresponding to this pattern.
518 std::string Code = P->getRecord()->getValueAsCode("Predicate");
519 assert(!Code.empty() && "No code in this predicate");
520 OS << " case " << i << ": { // " << NodePredicates[i] << '\n';
521 std::string ClassName;
522 if (P->getOnlyTree()->isLeaf())
523 ClassName = "SDNode";
524 else
525 ClassName =
526 CGP.getSDNodeInfo(P->getOnlyTree()->getOperator()).getSDClassName();
527 if (ClassName == "SDNode")
528 OS << " SDNode *N = Node;\n";
529 else
530 OS << " " << ClassName << "*N = cast<" << ClassName << ">(Node);\n";
531 OS << Code << "\n }\n";
532 }
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000533 OS << " }\n";
534 OS << "}\n\n";
535 }
Chris Lattnere609a512010-02-17 00:31:50 +0000536
537 // Emit CompletePattern matchers.
Chris Lattner8e946be2010-02-21 03:22:59 +0000538 // FIXME: This should be const.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000539 if (!ComplexPatterns.empty()) {
540 OS << "bool CheckComplexPattern(SDNode *Root, SDValue N,\n";
541 OS << " unsigned PatternNo, SmallVectorImpl<SDValue> &Result) {\n";
542 OS << " switch (PatternNo) {\n";
543 OS << " default: assert(0 && \"Invalid pattern # in table?\");\n";
544 for (unsigned i = 0, e = ComplexPatterns.size(); i != e; ++i) {
545 const ComplexPattern &P = *ComplexPatterns[i];
546 unsigned NumOps = P.getNumOperands();
Chris Lattner8e946be2010-02-21 03:22:59 +0000547
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000548 if (P.hasProperty(SDNPHasChain))
549 ++NumOps; // Get the chained node too.
550
551 OS << " case " << i << ":\n";
552 OS << " Result.resize(Result.size()+" << NumOps << ");\n";
553 OS << " return " << P.getSelectFunc();
Chris Lattner8e946be2010-02-21 03:22:59 +0000554
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000555 // FIXME: Temporary hack until old isel dies.
556 if (P.hasProperty(SDNPHasChain))
557 OS << "XXX";
558
559 OS << "(Root, N";
560 for (unsigned i = 0; i != NumOps; ++i)
561 OS << ", Result[Result.size()-" << (NumOps-i) << ']';
562 OS << ");\n";
563 }
564 OS << " }\n";
565 OS << "}\n\n";
Chris Lattnere609a512010-02-17 00:31:50 +0000566 }
Chris Lattner8e946be2010-02-21 03:22:59 +0000567
Chris Lattner4d0c9312010-03-01 01:54:19 +0000568
Chris Lattner8e946be2010-02-21 03:22:59 +0000569 // Emit SDNodeXForm handlers.
570 // FIXME: This should be const.
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000571 if (!NodeXForms.empty()) {
572 OS << "SDValue RunSDNodeXForm(SDValue V, unsigned XFormNo) {\n";
573 OS << " switch (XFormNo) {\n";
574 OS << " default: assert(0 && \"Invalid xform # in table?\");\n";
575
576 // FIXME: The node xform could take SDValue's instead of SDNode*'s.
Chris Lattner4d0c9312010-03-01 01:54:19 +0000577 for (unsigned i = 0, e = NodeXForms.size(); i != e; ++i) {
578 const CodeGenDAGPatterns::NodeXForm &Entry =
579 CGP.getSDNodeTransform(NodeXForms[i]);
580
581 Record *SDNode = Entry.first;
582 const std::string &Code = Entry.second;
583
584 OS << " case " << i << ": { // " << NodeXForms[i]->getName() << '\n';
585
586 std::string ClassName = CGP.getSDNodeInfo(SDNode).getSDClassName();
587 if (ClassName == "SDNode")
588 OS << " SDNode *N = V.getNode();\n";
589 else
590 OS << " " << ClassName << " *N = cast<" << ClassName
591 << ">(V.getNode());\n";
592 OS << Code << "\n }\n";
593 }
Chris Lattnerd20c0bc2010-02-28 22:57:03 +0000594 OS << " }\n";
595 OS << "}\n\n";
596 }
Chris Lattner050a03d2010-02-16 07:21:10 +0000597}
598
Chris Lattner09b9f392010-02-24 19:17:12 +0000599void MatcherTableEmitter::EmitHistogram(formatted_raw_ostream &OS) {
600 OS << " // Opcode Histogram:\n";
601 for (unsigned i = 0, e = Histogram.size(); i != e; ++i) {
602 OS << " // #";
Chris Lattnerb21ba712010-02-25 02:04:40 +0000603 switch ((Matcher::KindTy)i) {
604 case Matcher::Scope: OS << "OPC_Scope"; break;
605 case Matcher::RecordNode: OS << "OPC_RecordNode"; break;
606 case Matcher::RecordChild: OS << "OPC_RecordChild"; break;
607 case Matcher::RecordMemRef: OS << "OPC_RecordMemRef"; break;
608 case Matcher::CaptureFlagInput: OS << "OPC_CaptureFlagInput"; break;
609 case Matcher::MoveChild: OS << "OPC_MoveChild"; break;
610 case Matcher::MoveParent: OS << "OPC_MoveParent"; break;
611 case Matcher::CheckSame: OS << "OPC_CheckSame"; break;
612 case Matcher::CheckPatternPredicate:
Chris Lattner09b9f392010-02-24 19:17:12 +0000613 OS << "OPC_CheckPatternPredicate"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000614 case Matcher::CheckPredicate: OS << "OPC_CheckPredicate"; break;
615 case Matcher::CheckOpcode: OS << "OPC_CheckOpcode"; break;
Chris Lattnereb669212010-03-01 06:59:22 +0000616 case Matcher::SwitchOpcode: OS << "OPC_SwitchOpcode"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000617 case Matcher::CheckType: OS << "OPC_CheckType"; break;
618 case Matcher::CheckChildType: OS << "OPC_CheckChildType"; break;
619 case Matcher::CheckInteger: OS << "OPC_CheckInteger"; break;
620 case Matcher::CheckCondCode: OS << "OPC_CheckCondCode"; break;
621 case Matcher::CheckValueType: OS << "OPC_CheckValueType"; break;
622 case Matcher::CheckComplexPat: OS << "OPC_CheckComplexPat"; break;
623 case Matcher::CheckAndImm: OS << "OPC_CheckAndImm"; break;
624 case Matcher::CheckOrImm: OS << "OPC_CheckOrImm"; break;
625 case Matcher::CheckFoldableChainNode:
Chris Lattner09b9f392010-02-24 19:17:12 +0000626 OS << "OPC_CheckFoldableChainNode"; break;
Chris Lattner499622b2010-02-25 02:09:00 +0000627 case Matcher::CheckChainCompatible: OS << "OPC_CheckChainCompatible"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000628 case Matcher::EmitInteger: OS << "OPC_EmitInteger"; break;
629 case Matcher::EmitStringInteger: OS << "OPC_EmitStringInteger"; break;
630 case Matcher::EmitRegister: OS << "OPC_EmitRegister"; break;
Chris Lattner499622b2010-02-25 02:09:00 +0000631 case Matcher::EmitConvertToTarget: OS << "OPC_EmitConvertToTarget"; break;
632 case Matcher::EmitMergeInputChains: OS << "OPC_EmitMergeInputChains"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000633 case Matcher::EmitCopyToReg: OS << "OPC_EmitCopyToReg"; break;
634 case Matcher::EmitNode: OS << "OPC_EmitNode"; break;
Chris Lattner9a215002010-02-28 20:55:18 +0000635 case Matcher::MorphNodeTo: OS << "OPC_MorphNodeTo"; break;
Chris Lattnerb21ba712010-02-25 02:04:40 +0000636 case Matcher::EmitNodeXForm: OS << "OPC_EmitNodeXForm"; break;
637 case Matcher::MarkFlagResults: OS << "OPC_MarkFlagResults"; break;
638 case Matcher::CompleteMatch: OS << "OPC_CompleteMatch"; break;
Chris Lattner09b9f392010-02-24 19:17:12 +0000639 }
640
641 OS.PadToColumn(40) << " = " << Histogram[i] << '\n';
642 }
643 OS << '\n';
644}
645
Chris Lattner050a03d2010-02-16 07:21:10 +0000646
Chris Lattner4d0c9312010-03-01 01:54:19 +0000647void llvm::EmitMatcherTable(const Matcher *TheMatcher,
648 const CodeGenDAGPatterns &CGP, raw_ostream &O) {
Chris Lattnerda272d12010-02-15 08:04:42 +0000649 formatted_raw_ostream OS(O);
650
651 OS << "// The main instruction selector code.\n";
Chris Lattnerc84edb72010-02-24 07:35:09 +0000652 OS << "SDNode *SelectCode(SDNode *N) {\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000653
Chris Lattner984e1882010-02-21 06:30:04 +0000654 MatcherTableEmitter MatcherEmitter;
Chris Lattnere02ea542010-02-16 06:52:01 +0000655
Chris Lattner8e946be2010-02-21 03:22:59 +0000656 OS << " // Opcodes are emitted as 2 bytes, TARGET_OPCODE handles this.\n";
657 OS << " #define TARGET_OPCODE(X) X & 255, unsigned(X) >> 8\n";
Chris Lattnerda272d12010-02-15 08:04:42 +0000658 OS << " static const unsigned char MatcherTable[] = {\n";
Chris Lattnerb21ba712010-02-25 02:04:40 +0000659 unsigned TotalSize = MatcherEmitter.EmitMatcherList(TheMatcher, 5, 0, OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000660 OS << " 0\n }; // Total Array size is " << (TotalSize+1) << " bytes\n\n";
Chris Lattner09b9f392010-02-24 19:17:12 +0000661
662 MatcherEmitter.EmitHistogram(OS);
663
Chris Lattner8e946be2010-02-21 03:22:59 +0000664 OS << " #undef TARGET_OPCODE\n";
Chris Lattnere02ea542010-02-16 06:52:01 +0000665 OS << " return SelectCodeCommon(N, MatcherTable,sizeof(MatcherTable));\n}\n";
Chris Lattner050a03d2010-02-16 07:21:10 +0000666 OS << "\n";
667
668 // Next up, emit the function for node and pattern predicates:
Chris Lattner4d0c9312010-03-01 01:54:19 +0000669 MatcherEmitter.EmitPredicateFunctions(CGP, OS);
Chris Lattnerda272d12010-02-15 08:04:42 +0000670}