source: icGREP/icgrep-devel/icgrep/pablo/builder.hpp @ 4699

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

Temporary check in.

File size: 6.9 KB
Line 
1#ifndef BUILDER_HPP
2#define BUILDER_HPP
3
4#include <pablo/codegenstate.h>
5#include <pablo/expression_map.hpp>
6
7namespace pablo {
8
9class PabloBuilder {
10public:
11
12    explicit PabloBuilder(PabloBlock & block) : mPb(&block), mParent(nullptr), mExprTable(nullptr) {}
13
14    explicit PabloBuilder(PabloBlock & block, PabloBuilder & parent) : mPb(&block), mParent(&parent), mExprTable(&(parent.mExprTable)) {}
15
16    PabloBuilder(PabloBuilder && builder) : mPb(builder.mPb), mParent(builder.mParent), mExprTable(std::move(builder.mExprTable)) {}
17
18    PabloBuilder & operator=(PabloBuilder) = delete;
19
20    PabloBuilder & operator=(PabloBuilder &) = delete;
21
22    PabloBuilder & operator=(PabloBuilder && builder) {
23        mPb = builder.mPb;
24        mParent = builder.mParent;
25        mExprTable = std::move(builder.mExprTable);
26        return *this;
27    }
28
29    inline static PabloBuilder Create(PabloBuilder & parent) noexcept {
30        return std::move(PabloBuilder(PabloBlock::Create(*(parent.mPb)), parent));
31    }
32
33    inline Zeroes * createZeroes() const {
34        return mPb->createZeroes();
35    }
36
37    inline Ones * createOnes() const {
38        return mPb->createOnes();
39    }
40
41    inline Var * createVar(const std::string name) {
42        return mPb->createVar(name);
43    }
44
45    inline Var * createVar(String * const name) {
46        return mPb->createVar(name);
47    }
48
49    inline Var * createVar(PabloAST * const name) {
50        return mPb->createVar(name);
51    }
52
53    inline Call * createCall(Prototype * prototype, const std::vector<Var *> & args) {
54        return createCall(prototype, reinterpret_cast<const std::vector<PabloAST *> &>(args));
55    }
56
57    Call * createCall(Prototype * prototype, const std::vector<PabloAST *> &vars);
58
59    Assign * createAssign(const std::string && prefix, PabloAST * expr) {
60        return mPb->createAssign(std::move(prefix), expr);
61    }
62
63    inline PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount) {
64        if (shiftAmount == 0) {
65            return expr;
66        }
67        return createAdvance(expr, mPb->getInteger(shiftAmount));
68    }
69
70    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount);
71
72    inline PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount, const std::string prefix) {
73        if (shiftAmount == 0) {
74            return expr;
75        }
76        return createAdvance(expr, mPb->getInteger(shiftAmount), prefix);
77    }
78
79    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
80
81    inline Next * createNext(Assign * assign, PabloAST * expr) {
82        return mPb->createNext(assign, expr);
83    }
84
85    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2);
86
87    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
88
89    PabloAST * createNot(PabloAST * expr);
90
91    PabloAST * createNot(PabloAST * expr, const std::string prefix);
92
93    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2);
94
95    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
96
97    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2);
98
99    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
100
101    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass);
102
103    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix);
104
105    PabloAST * createScanThru(PabloAST * from, PabloAST * thru);
106
107    PabloAST * createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix);
108
109    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr);
110
111    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix);
112
113    /// CreateIf Wrappers
114
115    inline If * createIf(PabloAST * condition, std::initializer_list<Assign *> definedVars, PabloBlock & body) {
116        return mPb->createIf(condition, std::move(definedVars), body);
117    }
118
119    inline If * createIf(PabloAST * condition, const std::vector<Assign *> & definedVars, PabloBlock & body) {
120        return mPb->createIf(condition, definedVars, body);
121    }
122
123    inline If * createIf(PabloAST * condition, std::vector<Assign *> && definedVars, PabloBlock & body) {
124        return mPb->createIf(condition, std::move(definedVars), body);
125    }
126
127    inline If * createIf(PabloAST * condition, std::initializer_list<Assign *> definedVars, PabloBuilder & builder) {
128        return mPb->createIf(condition, std::move(definedVars), *builder.mPb);
129    }
130
131    inline If * createIf(PabloAST * condition, const std::vector<Assign *> & definedVars, PabloBuilder & builder) {
132        return mPb->createIf(condition, definedVars, *builder.mPb);
133    }
134
135    inline If * createIf(PabloAST * condition, std::vector<Assign *> && definedVars, PabloBuilder & builder) {
136        return mPb->createIf(condition, std::move(definedVars), *builder.mPb);
137    }
138
139    /// CreateWhile Wrappers
140
141    inline While * createWhile(PabloAST * condition, const std::initializer_list<Next *> nextVars, PabloBlock & body) {
142        return mPb->createWhile(condition, nextVars, body);
143    }
144
145    inline While * createWhile(PabloAST * condition, const std::vector<Next *> & nextVars, PabloBlock & body) {
146        return mPb->createWhile(condition, nextVars, body);
147    }
148
149    inline While * createWhile(PabloAST * condition, std::vector<Next *> && nextVars, PabloBlock & body) {
150        return mPb->createWhile(condition, std::move(nextVars), body);
151    }
152
153    inline While * createWhile(PabloAST * condition, const std::initializer_list<Next *> nextVars, PabloBuilder & builder) {
154        return mPb->createWhile(condition, nextVars, *builder.mPb);
155    }
156
157    inline While * createWhile(PabloAST * condition, const std::vector<Next *> & nextVars, PabloBuilder & builder) {
158        return mPb->createWhile(condition, nextVars, *builder.mPb);
159    }
160
161    inline While * createWhile(PabloAST * condition, std::vector<Next *> && nextVars, PabloBuilder & builder) {
162        return mPb->createWhile(condition, std::move(nextVars), *builder.mPb);
163    }
164
165    inline Statement * front() const {
166        return mPb->front();
167    }
168
169    inline Statement * back() const {
170        return mPb->back();
171    }
172
173    inline String * getName(const std::string name, const bool generated = true) const {
174        return mPb->getName(std::move(name), generated);
175    }
176
177    inline String * makeName(const std::string prefix, const bool generated = true) const {
178        return mPb->makeName(std::move(prefix), generated);
179    }
180
181    inline Integer * getInteger(Integer::Type value) {
182        return mPb->getInteger(value);
183    }
184
185
186    inline Statement * getInsertPoint() const {
187        return mPb->getInsertPoint();
188    }
189
190    inline PabloBlock * getPabloBlock() {
191        return mPb;
192    }
193
194    inline PabloBuilder * getParent() {
195        return mParent;
196    }
197
198private:
199
200    PabloBlock *        mPb;
201    PabloBuilder *      mParent;
202    ExpressionTable     mExprTable;
203};
204
205
206}
207
208
209#endif // BUILDER_HPP
Note: See TracBrowser for help on using the repository browser.