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

Last change on this file since 5217 was 5217, checked in by nmedfort, 3 years ago

Merged PabloFunction? and PabloKernel? classes. Updated projects where necessary.

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