source: icGREP/icgrep-devel/icgrep/pablo/builder.cpp @ 4717

Last change on this file since 4717 was 4717, checked in by cameron, 4 years ago

Mod64Advance, Mod64MatchStar, Mod64ScanThru ops; -mod64-approximate command-line option

File size: 7.2 KB
Line 
1#include <pablo/builder.hpp>
2
3namespace pablo {
4
5#define MAKE_UNARY(NAME, TYPE, ARGS...) \
6struct __##NAME { \
7    inline PabloAST * operator()(PabloAST * arg) { \
8        return mPb->NAME(arg); \
9    } \
10    inline PabloAST * operator()(PabloAST * arg, const std::string name) { \
11        return mPb->NAME(arg, name); \
12    } \
13    inline __##NAME(PabloBlock * pb) : mPb(pb) {} \
14private: \
15    PabloBlock * mPb; \
16}; \
17__##NAME functor(mPb); \
18PabloAST * result = mExprTable.findUnaryOrCall(std::move(functor), TYPE, ARGS)
19
20#define MAKE_BINARY(NAME, TYPE, ARGS...) \
21struct __##NAME { \
22    inline PabloAST * operator()(PabloAST * arg1, PabloAST * arg2) { \
23        return mPb->NAME(arg1, arg2); \
24    } \
25    inline PabloAST * operator()(PabloAST * arg1, PabloAST * arg2, const std::string name) { \
26        return mPb->NAME(arg1, arg2, name); \
27    } \
28    inline __##NAME(PabloBlock * pb) : mPb(pb) {} \
29private: \
30    PabloBlock * mPb; \
31}; \
32__##NAME functor(mPb); \
33PabloAST * result = mExprTable.findBinaryOrCall(std::move(functor), TYPE, ARGS)
34
35
36#define MAKE_TERNARY(NAME, TYPE, ARGS...) \
37struct __##NAME { \
38    inline PabloAST * operator()(PabloAST * arg1, PabloAST * arg2, PabloAST * arg3) { \
39        return mPb->NAME(arg1, arg2, arg3); \
40    } \
41    inline PabloAST * operator()(PabloAST * arg1, PabloAST * arg2, PabloAST * arg3, const std::string name) { \
42        return mPb->NAME(arg1, arg2, arg3, name); \
43    } \
44    inline __##NAME(PabloBlock * pb) : mPb(pb) {} \
45private: \
46    PabloBlock * mPb; \
47}; \
48__##NAME functor(mPb); \
49PabloAST * result = mExprTable.findTernaryOrCall(std::move(functor), TYPE, ARGS)
50
51#define MAKE_VARIABLE(NAME, TYPE, ARGS...) \
52struct __##NAME { \
53    inline PabloAST * operator()(const std::vector<PabloAST *> & args, PabloAST * prototype) { \
54        return mPb->NAME(prototype, args); \
55    } \
56    inline __##NAME(PabloBlock * pb) : mPb(pb) {} \
57private: \
58    PabloBlock * mPb; \
59}; \
60__##NAME functor(mPb); \
61PabloAST * result = mExprTable.findVariableOrCall(std::move(functor), TYPE, ARGS)
62
63Call * PabloBuilder::createCall(Prototype * prototype, const std::vector<PabloAST *> & args) {
64    if (prototype == nullptr) {
65        throw std::runtime_error("Call object cannot be created with a Null prototype!");
66    }
67    if (args.size() != cast<Prototype>(prototype)->getNumOfParameters()) {
68        throw std::runtime_error("Invalid number of arguments passed into Call object!");
69    }
70    MAKE_VARIABLE(createCall, PabloAST::ClassTypeId::Call, prototype->getName(), args, prototype);
71    return cast<Call>(result);
72}
73
74PabloAST * PabloBuilder::createAdvance(PabloAST * expr, PabloAST * shiftAmount) {
75    MAKE_BINARY(createAdvance, PabloAST::ClassTypeId::Advance, expr, shiftAmount);
76    return result;
77}
78
79PabloAST * PabloBuilder::createAdvance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix) {
80    MAKE_BINARY(createAdvance, PabloAST::ClassTypeId::Advance, expr, shiftAmount, prefix);
81    return result;
82}
83
84PabloAST * PabloBuilder::createMod64Advance(PabloAST * expr, PabloAST * shiftAmount) {
85    MAKE_BINARY(createMod64Advance, PabloAST::ClassTypeId::Mod64Advance, expr, shiftAmount);
86    return result;
87}
88
89PabloAST * PabloBuilder::createMod64Advance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix) {
90    MAKE_BINARY(createMod64Advance, PabloAST::ClassTypeId::Mod64Advance, expr, shiftAmount, prefix);
91    return result;
92}
93
94PabloAST * PabloBuilder::createNot(PabloAST * expr) {
95    MAKE_UNARY(createNot, PabloAST::ClassTypeId::Not, expr);
96    return result;
97}
98
99PabloAST * PabloBuilder::createNot(PabloAST * expr, const std::string prefix) {
100    MAKE_UNARY(createNot, PabloAST::ClassTypeId::Not, expr, prefix);
101    return result;
102}
103
104PabloAST * PabloBuilder::createAnd(PabloAST * expr1, PabloAST * expr2) {
105    if (isa<Not>(expr1) || expr1 > expr2) {
106        std::swap(expr1, expr2);
107    }
108    MAKE_BINARY(createAnd, PabloAST::ClassTypeId::And, expr1, expr2);
109    return result;
110}
111
112PabloAST * PabloBuilder::createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
113    if (isa<Not>(expr1) || expr1 > expr2) {
114        std::swap(expr1, expr2);
115    }
116    MAKE_BINARY(createAnd, PabloAST::ClassTypeId::And, expr1, expr2, prefix);
117    return result;
118}
119
120PabloAST * PabloBuilder::createOr(PabloAST * expr1, PabloAST * expr2) {
121    if (expr1 > expr2) {
122        std::swap(expr1, expr2);
123    }
124    MAKE_BINARY(createOr, PabloAST::ClassTypeId::Or, expr1, expr2);
125    return result;
126}
127
128PabloAST * PabloBuilder::createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
129    if (expr1 > expr2) {
130        std::swap(expr1, expr2);
131    }
132    MAKE_BINARY(createOr, PabloAST::ClassTypeId::Or, expr1, expr2, prefix);
133    return result;
134}
135
136PabloAST * PabloBuilder::createXor(PabloAST * expr1, PabloAST * expr2) {
137    if (expr1 > expr2) {
138        std::swap(expr1, expr2);
139    }
140    MAKE_BINARY(createXor, PabloAST::ClassTypeId::Xor, expr1, expr2);
141    return result;
142}
143
144PabloAST * PabloBuilder::createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
145    if (expr1 > expr2) {
146        std::swap(expr1, expr2);
147    }
148    MAKE_BINARY(createXor, PabloAST::ClassTypeId::Xor, expr1, expr2, prefix);
149    return result;
150}
151
152PabloAST * PabloBuilder::createMatchStar(PabloAST * marker, PabloAST * charclass) {
153    MAKE_BINARY(createMatchStar, PabloAST::ClassTypeId::MatchStar, marker, charclass);
154    return result;
155}
156
157PabloAST * PabloBuilder::createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix) {
158    MAKE_BINARY(createMatchStar, PabloAST::ClassTypeId::MatchStar, marker, charclass, prefix);
159    return result;
160}
161
162PabloAST * PabloBuilder::createScanThru(PabloAST * from, PabloAST * thru) {
163    MAKE_BINARY(createScanThru, PabloAST::ClassTypeId::ScanThru, from, thru);
164    return result;
165}
166
167PabloAST * PabloBuilder::createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix) {
168    MAKE_BINARY(createScanThru, PabloAST::ClassTypeId::ScanThru, from, thru, prefix);
169    return result;
170}
171
172
173PabloAST * PabloBuilder::createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr) {
174    MAKE_TERNARY(createSel, PabloAST::ClassTypeId::Sel, condition, trueExpr, falseExpr);
175    return result;
176}
177
178PabloAST * PabloBuilder::createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix) {
179    MAKE_TERNARY(createSel, PabloAST::ClassTypeId::Sel, condition, trueExpr, falseExpr, prefix);
180    return result;
181}
182
183PabloAST * PabloBuilder::createMod64MatchStar(PabloAST * marker, PabloAST * charclass) {
184    MAKE_BINARY(createMod64MatchStar, PabloAST::ClassTypeId::Mod64MatchStar, marker, charclass);
185    return result;
186}
187
188PabloAST * PabloBuilder::createMod64MatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix) {
189    MAKE_BINARY(createMod64MatchStar, PabloAST::ClassTypeId::Mod64MatchStar, marker, charclass, prefix);
190    return result;
191}
192
193PabloAST * PabloBuilder::createMod64ScanThru(PabloAST * from, PabloAST * thru) {
194    MAKE_BINARY(createMod64ScanThru, PabloAST::ClassTypeId::Mod64ScanThru, from, thru);
195    return result;
196}
197
198PabloAST * PabloBuilder::createMod64ScanThru(PabloAST * from, PabloAST * thru, const std::string prefix) {
199    MAKE_BINARY(createMod64ScanThru, PabloAST::ClassTypeId::Mod64ScanThru, from, thru, prefix);
200    return result;
201}
202
203
204
205}
Note: See TracBrowser for help on using the repository browser.