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

Last change on this file since 4692 was 4692, checked in by nmedfort, 4 years ago

Temporary check in.

File size: 5.9 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::createNot(PabloAST * expr) {
85    MAKE_UNARY(createNot, PabloAST::ClassTypeId::Not, expr);
86    return result;
87}
88
89PabloAST * PabloBuilder::createNot(PabloAST * expr, const std::string prefix) {
90    MAKE_UNARY(createNot, PabloAST::ClassTypeId::Not, expr, prefix);
91    return result;
92}
93
94PabloAST * PabloBuilder::createAnd(PabloAST * expr1, PabloAST * expr2) {
95    if (expr1 < expr2) {
96        std::swap(expr1, expr2);
97    }
98    MAKE_BINARY(createAnd, PabloAST::ClassTypeId::And, expr1, expr2);
99    return result;
100}
101
102PabloAST * PabloBuilder::createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
103    if (expr1 < expr2) {
104        std::swap(expr1, expr2);
105    }
106    MAKE_BINARY(createAnd, PabloAST::ClassTypeId::And, expr1, expr2, prefix);
107    return result;
108}
109
110PabloAST * PabloBuilder::createOr(PabloAST * expr1, PabloAST * expr2) {
111    if (expr1 < expr2) {
112        std::swap(expr1, expr2);
113    }
114    MAKE_BINARY(createOr, PabloAST::ClassTypeId::Or, expr1, expr2);
115    return result;
116}
117
118PabloAST * PabloBuilder::createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
119    if (expr1 < expr2) {
120        std::swap(expr1, expr2);
121    }
122    MAKE_BINARY(createOr, PabloAST::ClassTypeId::Or, expr1, expr2, prefix);
123    return result;
124}
125
126PabloAST * PabloBuilder::createXor(PabloAST * expr1, PabloAST * expr2) {
127    if (expr1 < expr2) {
128        std::swap(expr1, expr2);
129    }
130    MAKE_BINARY(createXor, PabloAST::ClassTypeId::Xor, expr1, expr2);
131    return result;
132}
133
134PabloAST * PabloBuilder::createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix) {
135    if (expr1 < expr2) {
136        std::swap(expr1, expr2);
137    }
138    MAKE_BINARY(createXor, PabloAST::ClassTypeId::Xor, expr1, expr2, prefix);
139    return result;
140}
141
142PabloAST * PabloBuilder::createMatchStar(PabloAST * marker, PabloAST * charclass) {
143    MAKE_BINARY(createMatchStar, PabloAST::ClassTypeId::MatchStar, marker, charclass);
144    return result;
145}
146
147PabloAST * PabloBuilder::createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix) {
148    MAKE_BINARY(createMatchStar, PabloAST::ClassTypeId::MatchStar, marker, charclass, prefix);
149    return result;
150}
151
152PabloAST * PabloBuilder::createScanThru(PabloAST * from, PabloAST * thru) {
153    MAKE_BINARY(createScanThru, PabloAST::ClassTypeId::ScanThru, from, thru);
154    return result;
155}
156
157PabloAST * PabloBuilder::createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix) {
158    MAKE_BINARY(createScanThru, PabloAST::ClassTypeId::ScanThru, from, thru, prefix);
159    return result;
160}
161
162
163PabloAST * PabloBuilder::createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr) {
164    MAKE_TERNARY(createSel, PabloAST::ClassTypeId::Sel, condition, trueExpr, falseExpr);
165    return result;
166}
167
168PabloAST * PabloBuilder::createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix) {
169    MAKE_TERNARY(createSel, PabloAST::ClassTypeId::Sel, condition, trueExpr, falseExpr, prefix);
170    return result;
171}
172
173
174}
Note: See TracBrowser for help on using the repository browser.