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

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

First attempt to add arguments to Call objects. Currently non-functioning in Pablo_Compiler.

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