source: icGREP/icgrep-devel/icgrep/pablo/codegenstate.h @ 5298

Last change on this file since 5298 was 5285, checked in by nmedfort, 2 years ago

Start of work to simplify kernel writing. Removed generateDoBlockLogic method.

File size: 9.9 KB
Line 
1/*
2 *  Copyright (c) 2014 International Characters.
3 *  This software is licensed to the public under the Open Software License 3.0.
4 *  icgrep is a trademark of International Characters.
5 */
6
7#ifndef PS_PABLOS_H
8#define PS_PABLOS_H
9
10#include <pablo/pabloAST.h>
11#include <pablo/pablo_kernel.h>
12#include <pablo/pe_integer.h>
13namespace llvm { class Type; }
14namespace llvm { class raw_ostream; }
15namespace pablo { class Add; }
16namespace pablo { class Advance; }
17namespace pablo { class And; }
18namespace pablo { class Assign; }
19namespace pablo { class AtEOF; }
20namespace pablo { class Branch; }
21namespace pablo { class If; }
22namespace pablo { class While; }
23namespace pablo { class Count; }
24namespace pablo { class Extract; }
25namespace pablo { class InFile; }
26namespace pablo { class LessThan; }
27namespace pablo { class Lookahead; }
28namespace pablo { class MatchStar; }
29namespace pablo { class Not; }
30namespace pablo { class Ones; }
31namespace pablo { class Or; }
32namespace pablo { class PabloKernel; }
33namespace pablo { class ScanThru; }
34namespace pablo { class Sel; }
35namespace pablo { class String; }
36namespace pablo { class Subtract; }
37namespace pablo { class Var; }
38namespace pablo { class Xor; }
39namespace pablo { class Zeroes; }
40
41
42namespace pablo {
43
44class PabloBlock : public PabloAST, public StatementList {
45    friend class PabloAST;
46    friend class Branch;
47    friend class PabloBuilder;
48    friend class PabloKernel;
49public:
50
51    static inline bool classof(const PabloBlock *) {
52        return true;
53    }
54    static inline bool classof(const Statement *) {
55        return false;
56    }
57    static inline bool classof(const PabloAST * e) {
58        return e->getClassTypeId() == ClassTypeId::Block;
59    }
60    static inline bool classof(const void *) {
61        return false;
62    }
63
64    static PabloBlock * Create(PabloKernel * const parent) noexcept;
65
66    Advance * createAdvance(PabloAST * expr, PabloAST * shiftAmount) {
67        return createAdvance(expr, shiftAmount, nullptr);
68    }
69
70    Advance * createAdvance(PabloAST * expr, PabloAST * shiftAmount, const llvm::StringRef & prefix) {
71        return createAdvance(expr, shiftAmount, makeName(prefix));
72    }
73
74    Advance * createAdvance(PabloAST * expr, PabloAST * shiftAmount, String * name);
75
76    Lookahead * createLookahead(PabloAST * expr, PabloAST * shiftAmount) {
77        return createLookahead(expr, shiftAmount, nullptr);
78    }
79
80    Lookahead * createLookahead(PabloAST * expr, PabloAST * shiftAmount, const llvm::StringRef & prefix) {
81        return createLookahead(expr, shiftAmount, makeName(prefix));
82    }
83
84    Lookahead * createLookahead(PabloAST * expr, PabloAST * shiftAmount, String * name);
85
86    inline Zeroes * createZeroes(llvm::Type * const type = nullptr) {
87        return mParent->getNullValue(type);
88    }
89
90    inline Ones * createOnes(llvm::Type * const type = nullptr) {
91        return mParent->getAllOnesValue(type);
92    }
93
94    Not * createNot(PabloAST * expr) {
95        return createNot(expr, nullptr);
96    }
97
98    Not * createNot(PabloAST * expr, const llvm::StringRef & prefix) {
99        return createNot(expr, makeName(prefix));
100    }
101
102    Not * createNot(PabloAST * expr, String * name);
103
104    inline Var * createVar(const llvm::StringRef & name, llvm::Type * const type = nullptr) {
105        return createVar(makeName(name), type);
106    }
107
108    inline Var * createVar(String * name, llvm::Type * const type = nullptr) {
109        return createVar(reinterpret_cast<PabloAST *>(name), type);
110    }
111
112    Count * createCount(PabloAST * expr);
113
114    Count * createCount(PabloAST * expr, const llvm::StringRef & prefix);
115
116    InFile * createInFile(PabloAST * expr) {
117        return createInFile(expr, nullptr);
118    }
119
120    InFile * createInFile(PabloAST * expr, const llvm::StringRef & prefix) {
121        return createInFile(expr, makeName(prefix));
122    }
123
124    InFile * createInFile(PabloAST * expr, String * name);
125
126    AtEOF * createAtEOF(PabloAST * expr) {
127        return createAtEOF(expr, nullptr);
128    }
129
130    AtEOF * createAtEOF(PabloAST * expr, const llvm::StringRef & prefix) {
131        return createAtEOF(expr, makeName(prefix));
132    }
133
134    AtEOF * createAtEOF(PabloAST * expr, String * name);
135
136    inline Extract * createExtract(PabloAST * array, PabloAST * index) {
137        return createExtract(array, index, nullptr);
138    }
139
140    Extract * createExtract(PabloAST * array, PabloAST * index, const llvm::StringRef & prefix) {
141        return createExtract(array, index, makeName(prefix));
142    }
143
144    Extract * createExtract(PabloAST * array, const int64_t index) {
145        return createExtract(array, getInteger(index), nullptr);
146    }
147
148    Extract * createExtract(PabloAST * array, const int64_t index, const llvm::StringRef & prefix) {
149        return createExtract(array, getInteger(index), makeName(prefix));
150    }
151
152    Extract * createExtract(PabloAST * array, PabloAST * index, String * name);
153
154    Assign * createAssign(PabloAST * const var, PabloAST * const value);
155
156    And * createAnd(PabloAST * expr1, PabloAST * expr2) {
157        return createAnd(expr1, expr2, nullptr);
158    }
159
160    And * createAnd(PabloAST * expr1, PabloAST * expr2, const llvm::StringRef & prefix) {
161        return createAnd(expr1, expr2, makeName(prefix));
162    }
163
164    And * createAnd(PabloAST * expr1, PabloAST * expr2, String * name);
165
166    And * createAnd(llvm::Type * const type, const unsigned reserved) {
167        return createAnd(type, reserved, nullptr);
168    }
169
170    And * createAnd(llvm::Type * const type, const unsigned reserved, String * name);
171
172    Or * createOr(PabloAST * expr1, PabloAST * expr2) {
173        return createOr(expr1, expr2, nullptr);
174    }
175
176    Or * createOr(PabloAST * expr1, PabloAST * expr2, const llvm::StringRef & prefix) {
177        return createOr(expr1, expr2, makeName(prefix));
178    }
179
180    Or * createOr(PabloAST * expr1, PabloAST * expr2, String * name);
181
182    Or * createOr(llvm::Type * const type, const unsigned reserved) {
183        return createOr(type, reserved, nullptr);
184    }
185
186    Or * createOr(llvm::Type * const type, const unsigned reserved, String * name);
187
188    Xor * createXor(PabloAST * expr1, PabloAST * expr2) {
189        return createXor(expr1, expr2, nullptr);
190    }
191
192    Xor * createXor(PabloAST * expr1, PabloAST * expr2, const llvm::StringRef & prefix) {
193        return createXor(expr1, expr2, makeName(prefix));
194    }
195
196    Xor * createXor(PabloAST * expr1, PabloAST * expr2, String * name);
197
198    Xor * createXor(llvm::Type * const type, const unsigned reserved) {
199        return createXor(type, reserved, nullptr);
200    }
201
202    Xor * createXor(llvm::Type * const type, const unsigned reserved, String * name);
203
204    Sel * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr) {
205        return createSel(condition, trueExpr, falseExpr, nullptr);
206    }
207
208    Sel * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const llvm::StringRef & prefix) {
209        return createSel(condition, trueExpr, falseExpr, makeName(prefix));
210    }
211
212    Sel * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, String * name);
213
214    Add * createAdd(PabloAST * expr1, PabloAST * expr2);
215
216    Subtract * createSubtract(PabloAST * expr1, PabloAST * expr2);
217
218    LessThan * createLessThan(PabloAST * expr1, PabloAST * expr2);
219
220    MatchStar * createMatchStar(PabloAST * marker, PabloAST * charclass) {
221        return createMatchStar(marker, charclass, nullptr);
222    }
223
224    MatchStar * createMatchStar(PabloAST * marker, PabloAST * charclass, const llvm::StringRef & prefix) {
225        return createMatchStar(marker, charclass, makeName(prefix));
226    }
227
228    MatchStar * createMatchStar(PabloAST * marker, PabloAST * charclass, String * name);
229
230    ScanThru * createScanThru(PabloAST * from, PabloAST * thru) {
231        return createScanThru(from, thru, nullptr);
232    }
233
234    ScanThru * createScanThru(PabloAST * from, PabloAST * thru, const llvm::StringRef & prefix) {
235        return createScanThru(from, thru, makeName(prefix));
236    }
237
238    ScanThru * createScanThru(PabloAST * from, PabloAST * thru, String * name);
239
240    If * createIf(PabloAST * condition, PabloBlock * body);
241
242    While * createWhile(PabloAST * condition, PabloBlock * body);
243
244    llvm::Type * getStreamTy(const unsigned FieldWidth = 1) {
245        return mParent->getStreamTy(FieldWidth);
246    }
247   
248    llvm::Type * getStreamSetTy(const unsigned NumElements = 1, const unsigned FieldWidth = 1) {
249        return mParent->getStreamSetTy(NumElements, FieldWidth);
250    }
251   
252    PabloBlock * getPredecessor() const;
253
254    inline PabloKernel * getParent() const {
255        return mParent;
256    }
257
258    void insert(Statement * const statement);
259
260    inline void setScopeIndex(const unsigned index) {
261        mScopeIndex = index;
262    }
263
264    inline unsigned getScopeIndex() const {
265        return mScopeIndex;
266    }
267   
268    void eraseFromParent(const bool recursively = false);
269
270    inline Branch * getBranch() const {
271        return mBranch;
272    }
273
274    inline void setBranch(Branch * const branch) {
275        mBranch = branch;
276    }
277
278    inline String * makeName(const llvm::StringRef & prefix) const {
279        return mParent->makeName(prefix);
280    }
281
282    inline Integer * getInteger(const int64_t value) const {
283        return mParent->getInteger(value);
284    }
285
286    void print(llvm::raw_ostream & O, const bool expandNested = true) const;
287
288    virtual ~PabloBlock() {}
289
290protected:
291
292    explicit PabloBlock(PabloKernel * const parent, Allocator & allocator) noexcept
293    : PabloAST(PabloAST::ClassTypeId::Block, nullptr, allocator)
294    , mParent(parent)
295    , mBranch(nullptr)
296    , mScopeIndex(0)
297    , mAllocator(allocator) {
298
299    }
300
301    template<typename Type>
302    inline Type * insertAtInsertionPoint(Type * expr) {
303        if (llvm::isa<Statement>(expr)) {
304            insert(llvm::cast<Statement>(expr));
305        }
306        return expr;
307    }
308
309    Var * createVar(PabloAST * name, llvm::Type * const type);
310
311private:       
312    PabloKernel * const         mParent;
313    Branch *                    mBranch;
314    unsigned                    mScopeIndex;
315    Allocator &                 mAllocator;
316};
317
318}
319
320#endif // PS_PABLOS_H
Note: See TracBrowser for help on using the repository browser.