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

Last change on this file since 4690 was 4687, checked in by cameron, 4 years ago

Carry Manager restructuring and bug fixes

File size: 6.3 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/ps_assign.h>
28#include <pablo/ps_if.h>
29#include <pablo/ps_while.h>
30#include <pablo/function.h>
31#include <llvm/ADT/ArrayRef.h>
32#include <stdexcept>
33
34namespace pablo {
35
36class PabloBlock : public PabloAST, public StatementList {
37    friend class PabloAST;
38    friend class PabloBuilder;
39public:
40
41    static inline bool classof(const PabloBlock *) {
42        return true;
43    }
44    static inline bool classof(const Statement *) {
45        return false;
46    }
47    static inline bool classof(const PabloAST * e) {
48        return e->getClassTypeId() == ClassTypeId::Block;
49    }
50    static inline bool classof(const void *) {
51        return false;
52    }
53
54    inline static PabloBlock & Create(SymbolGenerator & symbolGenerator) {
55        return *(new PabloBlock(symbolGenerator));
56    }
57
58    inline static PabloBlock & Create(PabloBlock & parent) {
59        return *(new PabloBlock(&parent));
60    }
61
62    PabloAST * createAdvance(PabloAST * expr, const Integer::integer_t shiftAmount);
63
64    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount);
65
66    PabloAST * createAdvance(PabloAST * expr, const Integer::integer_t shiftAmount, const std::string prefix);
67
68    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
69
70    inline Zeroes * createZeroes() const {
71        return mZeroes;
72    }
73
74    inline Ones * createOnes() const {
75        return mOnes;
76    }
77
78    inline Call * createCall(Prototype * prototype, const std::vector<Var *> & args) {
79        return createCall(prototype, reinterpret_cast<const std::vector<PabloAST *> &>(args));
80    }
81
82    Assign * createAssign(const std::string && prefix, PabloAST * expr);
83
84    inline Var * createVar(const std::string name) {
85        return createVar(getName(name, false));
86    }
87
88    inline Var * createVar(String * name) {
89        return createVar(cast<PabloAST>(name));
90    }
91
92    Next * createNext(Assign * assign, PabloAST * expr);
93
94    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2);
95
96    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
97
98    PabloAST * createNot(PabloAST * expr);
99
100    PabloAST * createNot(PabloAST * expr, const std::string prefix);
101
102    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2);
103
104    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
105
106    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2);
107
108    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
109
110    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass);
111
112    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix);
113
114    PabloAST * createScanThru(PabloAST * from, PabloAST * thru);
115
116    PabloAST * createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix);
117
118    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr);
119
120    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix);
121
122    If * createIf(PabloAST * condition, const std::initializer_list<Assign *> definedVars, PabloBlock & body);
123
124    If * createIf(PabloAST * condition, const std::vector<Assign *> & definedVars, PabloBlock & body);
125
126    If * createIf(PabloAST * condition, std::vector<Assign *> && definedVars, PabloBlock & body);
127
128    While * createWhile(PabloAST * condition, const std::initializer_list<Next *> nextVars, PabloBlock & body);
129
130    While * createWhile(PabloAST * condition, const std::vector<Next *> & nextVars, PabloBlock & body);
131
132    While * createWhile(PabloAST * condition, std::vector<Next *> && nextVars, PabloBlock & body);
133
134
135    inline StatementList & statements() {
136        return *this;
137    }
138
139    inline const StatementList & statements() const {
140        return *this;
141    }
142
143    inline String * getName(const std::string name, const bool generated = true) const {
144        return mSymbolGenerator.get(name, generated);
145    }
146
147    inline String * makeName(const std::string prefix, const bool generated = true) const {
148        return mSymbolGenerator.make(prefix, generated);
149    }
150
151    inline Integer * getInteger(Integer::integer_t value) {
152        return mSymbolGenerator.getInteger(value);
153    }
154
155    inline PabloBlock * getParent() const {
156        return mParent;
157    }
158   
159    void insert(Statement * const statement);
160
161    unsigned enumerateScopes(unsigned baseScopeIndex);
162   
163    inline unsigned getScopeIndex() const {
164        return mScopeIndex;
165    }
166   
167    virtual ~PabloBlock();
168
169protected:
170
171
172    PabloBlock(SymbolGenerator & symbolGenerator);
173
174    PabloBlock(PabloBlock * predecessor);
175
176    PabloAST * renameNonNamedNode(PabloAST * expr, const std::string && prefix);
177
178    template<typename Type>
179    inline Type * insertAtInsertionPoint(Type * expr) {
180        if (isa<Statement>(expr)) {
181            if (LLVM_UNLIKELY(isa<If>(expr) || isa<While>(expr))) {
182                PabloBlock & body = isa<If>(expr) ? cast<If>(expr)->getBody() : cast<While>(expr)->getBody();
183                this->addUser(&body);
184            }
185            insert(cast<Statement>(expr));
186        }
187        return expr;
188    }
189private:
190
191    Call * createCall(PabloAST * prototype, const std::vector<PabloAST *> & args);
192
193    Var * createVar(PabloAST * name);
194
195private:       
196    Zeroes * const                                      mZeroes;
197    Ones * const                                        mOnes;
198    SymbolGenerator &                                   mSymbolGenerator;
199    PabloBlock *                                        mParent;
200    unsigned                                            mScopeIndex;
201};
202
203}
204
205#endif // PS_PABLOS_H
Note: See TracBrowser for help on using the repository browser.