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

Last change on this file since 4995 was 4980, checked in by cameron, 3 years ago

Remove mod64 junk

File size: 9.2 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/pe_advance.h>
13#include <pablo/pe_lookahead.h>
14#include <pablo/pe_and.h>
15#include <pablo/pe_call.h>
16#include <pablo/pe_matchstar.h>
17#include <pablo/pe_next.h>
18#include <pablo/pe_not.h>
19#include <pablo/pe_ones.h>
20#include <pablo/pe_or.h>
21#include <pablo/pe_scanthru.h>
22#include <pablo/pe_sel.h>
23#include <pablo/pe_integer.h>
24#include <pablo/pe_string.h>
25#include <pablo/pe_var.h>
26#include <pablo/pe_xor.h>
27#include <pablo/pe_zeroes.h>
28#include <pablo/pe_count.h>
29#include <pablo/ps_assign.h>
30#include <pablo/ps_if.h>
31#include <pablo/ps_while.h>
32#include <pablo/function.h>
33#include <llvm/ADT/ArrayRef.h>
34#include <stdexcept>
35
36namespace pablo {
37
38class PabloBlock : public PabloAST, public StatementList {
39    friend class PabloAST;
40    friend class If;
41    friend class While;
42    friend class PabloBuilder;
43public:
44
45    static inline bool classof(const PabloBlock *) {
46        return true;
47    }
48    static inline bool classof(const Statement *) {
49        return false;
50    }
51    static inline bool classof(const PabloAST * e) {
52        return e->getClassTypeId() == ClassTypeId::Block;
53    }
54    static inline bool classof(const void *) {
55        return false;
56    }
57
58    inline static PabloBlock * Create(PabloFunction & function) noexcept {
59        return new PabloBlock(function.mSymbolTable);
60    }
61
62    inline static PabloBlock * Create(PabloBlock * const block) noexcept {
63        return new PabloBlock(block->mSymbolGenerator);
64    }
65
66    PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount);
67
68    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount);
69
70    PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount, const std::string prefix);
71
72    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
73
74    PabloAST * createLookahead(PabloAST * expr, const Integer::Type shiftAmount);
75
76    PabloAST * createLookahead(PabloAST * expr, PabloAST * shiftAmount);
77
78    PabloAST * createLookahead(PabloAST * expr, const Integer::Type shiftAmount, const std::string prefix);
79
80    PabloAST * createLookahead(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
81
82    static inline Zeroes * createZeroes() {
83        return &mZeroes;
84    }
85
86    static inline Ones * createOnes() {
87        return &mOnes;
88    }
89
90    inline Call * createCall(Prototype * prototype, const std::vector<Var *> & args) {
91        return createCall(prototype, reinterpret_cast<const std::vector<PabloAST *> &>(args));
92    }
93
94    inline Call * createCall(Prototype * prototype, const std::vector<PabloAST *> & args) {
95        if (prototype == nullptr) {
96            throw std::runtime_error("Call object cannot be created with a Null prototype!");
97        }
98        if (args.size() != cast<Prototype>(prototype)->getNumOfParameters()) {
99            throw std::runtime_error("Invalid number of arguments passed into Call object!");
100        }
101        return createCall(static_cast<PabloAST *>(prototype), args);
102    }
103
104    Assign * createAssign(const std::string && prefix, PabloAST * const expr);
105
106    inline Var * createVar(const std::string name) {
107        return createVar(getName(name, false));
108    }
109
110    inline Var * createVar(String * name) {
111        return createVar(cast<PabloAST>(name));
112    }
113
114    Next * createNext(Assign * assign, PabloAST * expr);
115
116    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2);
117
118    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
119
120    And * createAnd(const unsigned reserved);
121
122    And * createAnd(const unsigned reserved, const std::string prefix);
123
124    And * createAnd(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
125        return insertAtInsertionPoint(new And(begin, end, makeName("and_")));
126    }
127
128    And * createAnd(Variadic::iterator begin, Variadic::iterator end) {
129        return insertAtInsertionPoint(new And(begin, end, makeName("and_")));
130    }
131
132    PabloAST * createNot(PabloAST * expr);
133
134    PabloAST * createNot(PabloAST * expr, const std::string prefix);
135
136    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2);
137
138    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
139
140    Or * createOr(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
141        return insertAtInsertionPoint(new Or(begin, end, makeName("or_")));
142    }
143
144    Or * createOr(Variadic::iterator begin, Variadic::iterator end) {
145        return insertAtInsertionPoint(new Or(begin, end, makeName("or_")));
146    }
147
148    Or * createOr(const unsigned reserved);
149
150    Or * createOr(const unsigned reserved, const std::string prefix);
151
152    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2);
153
154    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
155
156    Xor * createXor(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
157        return insertAtInsertionPoint(new Xor(begin, end, makeName("xor_")));
158    }
159
160    Xor * createXor(Variadic::iterator begin, Variadic::iterator end) {
161        return insertAtInsertionPoint(new Xor(begin, end, makeName("xor_")));
162    }
163
164    Xor * createXor(const unsigned reserved);
165
166    Xor * createXor(const unsigned reserved, const std::string prefix);
167
168    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass);
169
170    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix);
171
172    PabloAST * createScanThru(PabloAST * from, PabloAST * thru);
173
174    PabloAST * createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix);
175
176    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr);
177
178    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix);
179
180    PabloAST * createCount(PabloAST * expr);
181   
182    PabloAST * createCount(PabloAST * expr, const std::string prefix);
183   
184    If * createIf(PabloAST * condition, const std::initializer_list<Assign *> definedVars, PabloBlock * body);
185
186    If * createIf(PabloAST * condition, const std::vector<Assign *> & definedVars, PabloBlock * body);
187
188    If * createIf(PabloAST * condition, std::vector<Assign *> && definedVars, PabloBlock * body);
189
190    While * createWhile(PabloAST * condition, const std::initializer_list<Next *> nextVars, PabloBlock * body);
191
192    While * createWhile(PabloAST * condition, const std::vector<Next *> & nextVars, PabloBlock * body);
193
194    While * createWhile(PabloAST * condition, std::vector<Next *> && nextVars, PabloBlock * body);
195
196    inline StatementList & statements() {
197        return *this;
198    }
199
200    inline const StatementList & statements() const {
201        return *this;
202    }
203
204    inline String * getName(const std::string name, const bool generated = true) const {
205        return mSymbolGenerator->get(name, generated);
206    }
207
208    inline String * makeName(const std::string prefix, const bool generated = true) const {
209        return mSymbolGenerator->make(prefix, generated);
210    }
211
212    inline Integer * getInteger(Integer::Type value) {
213        return mSymbolGenerator->getInteger(value);
214    }
215
216    inline PabloBlock * getParent() const {
217        return mParent;
218    }
219   
220    void setParent(PabloBlock * parent) {
221        mParent = parent;
222        // Add test to assert this block is in the same function.
223    }
224
225    void insert(Statement * const statement);
226
227    unsigned enumerateScopes(unsigned baseScopeIndex);
228   
229    inline unsigned getScopeIndex() const {
230        return mScopeIndex;
231    }
232   
233    void eraseFromParent(const bool recursively = false);
234
235    inline Statement * getBranch() const {
236        return mBranch;
237    }
238
239    virtual ~PabloBlock();
240
241protected:
242
243    explicit PabloBlock(SymbolGenerator * symbolGenerator) noexcept;
244
245    PabloAST * renameNonNamedNode(PabloAST * expr, const std::string && prefix);
246
247    template<typename Type>
248    inline Type * insertAtInsertionPoint(Type * expr) {
249        if (isa<Statement>(expr)) {
250            if (LLVM_UNLIKELY(isa<If>(expr) || isa<While>(expr))) {
251                PabloBlock * const body = isa<If>(expr) ? cast<If>(expr)->getBody() : cast<While>(expr)->getBody();
252                body->setParent(this);
253                addUser(body);
254            }
255            insert(cast<Statement>(expr));
256        }
257        return expr;
258    }
259
260    inline void setBranch(Statement * const branch) {
261        mBranch = branch;
262    }
263
264private:
265
266    Call * createCall(PabloAST * prototype, const std::vector<PabloAST *> &);
267
268    Var * createVar(PabloAST * name);
269
270private:       
271    static Zeroes                                       mZeroes;
272    static Ones                                         mOnes;
273    SymbolGenerator *                                   mSymbolGenerator; // TODO: need a better way of passing a symbol generator around
274    PabloBlock *                                        mParent;
275    Statement *                                         mBranch; // What statement branches into this scope block?
276    unsigned                                            mScopeIndex;
277};
278
279}
280
281#endif // PS_PABLOS_H
Note: See TracBrowser for help on using the repository browser.