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

Last change on this file since 4927 was 4927, checked in by nmedfort, 3 years ago

Bug fixes

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