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

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

Temporary check-in

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