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

Last change on this file since 5231 was 5231, checked in by cameron, 20 months ago

Use unsigned in preference to uint64_t for element counts and field widths

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