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

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

Back up check in. Memory leaks should be fixed.

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