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

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

Bug fixes

File size: 8.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;
39    friend class PabloBuilder;
[4227]40public:
[4206]41
[4432]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
[4870]55    inline static PabloBlock * Create(PabloFunction & function) noexcept {
56        return new PabloBlock(function.mSymbolTable);
[4210]57    }
58
[4871]59    inline static PabloBlock * Create(PabloBlock * const block) noexcept {
60        return new PabloBlock(block->mSymbolGenerator);
61    }
62
[4699]63    PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount);
[4214]64
[4432]65    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount);
66
[4699]67    PabloAST * createAdvance(PabloAST * expr, const Integer::Type shiftAmount, const std::string prefix);
[4438]68
69    PabloAST * createAdvance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
70
[4788]71    static inline Zeroes * createZeroes() {
[4797]72        return &mZeroes;
[4211]73    }
[4210]74
[4788]75    static inline Ones * createOnes() {
[4797]76        return &mOnes;
[4247]77    }
78
[4681]79    inline Call * createCall(Prototype * prototype, const std::vector<Var *> & args) {
80        return createCall(prototype, reinterpret_cast<const std::vector<PabloAST *> &>(args));
[4432]81    }
[4213]82
[4692]83    inline Call * createCall(Prototype * prototype, const std::vector<PabloAST *> & args) {
84        if (prototype == nullptr) {
85            throw std::runtime_error("Call object cannot be created with a Null prototype!");
86        }
87        if (args.size() != cast<Prototype>(prototype)->getNumOfParameters()) {
88            throw std::runtime_error("Invalid number of arguments passed into Call object!");
89        }
90        return createCall(static_cast<PabloAST *>(prototype), args);
91    }
92
[4657]93    Assign * createAssign(const std::string && prefix, PabloAST * expr);
[4252]94
[4432]95    inline Var * createVar(const std::string name) {
[4433]96        return createVar(getName(name, false));
[4432]97    }
[4254]98
[4602]99    inline Var * createVar(String * name) {
100        return createVar(cast<PabloAST>(name));
[4252]101    }
102
103    Next * createNext(Assign * assign, PabloAST * expr);
104
[4244]105    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2);
[4209]106
[4885]107    PabloAST * createAnd(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
108
[4886]109    And * createAnd(const unsigned reserved);
[4878]110
[4885]111    And * createAnd(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
112        return insertAtInsertionPoint(new And(begin, end, makeName("and_")));
113    }
[4438]114
[4244]115    PabloAST * createNot(PabloAST * expr);
[4209]116
[4438]117    PabloAST * createNot(PabloAST * expr, const std::string prefix);
118
[4244]119    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2);
[4209]120
[4885]121    PabloAST * createOr(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
122
123    Or * createOr(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
124        return insertAtInsertionPoint(new Or(begin, end, makeName("or_")));
125    }
126
[4886]127    Or * createOr(const unsigned reserved);
[4878]128
[4244]129    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2);
[4209]130
[4438]131    PabloAST * createXor(PabloAST * expr1, PabloAST * expr2, const std::string prefix);
132
[4885]133    Xor * createXor(std::vector<PabloAST *>::iterator begin, std::vector<PabloAST *>::iterator end) {
134        return insertAtInsertionPoint(new Xor(begin, end, makeName("xor_")));
135    }
136
[4886]137    Xor * createXor(const unsigned reserved);
138
[4255]139    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass);
[4207]140
[4438]141    PabloAST * createMatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix);
142
[4255]143    PabloAST * createScanThru(PabloAST * from, PabloAST * thru);
[4209]144
[4438]145    PabloAST * createScanThru(PabloAST * from, PabloAST * thru, const std::string prefix);
146
[4244]147    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr);
[4207]148
[4438]149    PabloAST * createSel(PabloAST * condition, PabloAST * trueExpr, PabloAST * falseExpr, const std::string prefix);
150
[4718]151    PabloAST * createCount(PabloAST * expr);
152   
153    PabloAST * createCount(PabloAST * expr, const std::string prefix);
154   
[4870]155    If * createIf(PabloAST * condition, const std::initializer_list<Assign *> definedVars, PabloBlock * body);
[4215]156
[4870]157    If * createIf(PabloAST * condition, const std::vector<Assign *> & definedVars, PabloBlock * body);
[4511]158
[4870]159    If * createIf(PabloAST * condition, std::vector<Assign *> && definedVars, PabloBlock * body);
[4511]160
[4870]161    While * createWhile(PabloAST * condition, const std::initializer_list<Next *> nextVars, PabloBlock * body);
[4215]162
[4870]163    While * createWhile(PabloAST * condition, const std::vector<Next *> & nextVars, PabloBlock * body);
[4641]164
[4870]165    While * createWhile(PabloAST * condition, std::vector<Next *> && nextVars, PabloBlock * body);
[4641]166
[4717]167    PabloAST * createMod64Advance(PabloAST * expr, const Integer::Type shiftAmount);
[4641]168
[4717]169    PabloAST * createMod64Advance(PabloAST * expr, PabloAST * shiftAmount);
170
171    PabloAST * createMod64Advance(PabloAST * expr, const Integer::Type shiftAmount, const std::string prefix);
172
173    PabloAST * createMod64Advance(PabloAST * expr, PabloAST * shiftAmount, const std::string prefix);
174
175    PabloAST * createMod64MatchStar(PabloAST * marker, PabloAST * charclass);
176
177    PabloAST * createMod64MatchStar(PabloAST * marker, PabloAST * charclass, const std::string prefix);
178
179    PabloAST * createMod64ScanThru(PabloAST * from, PabloAST * thru);
180
181    PabloAST * createMod64ScanThru(PabloAST * from, PabloAST * thru, const std::string prefix);
182
183
[4276]184    inline StatementList & statements() {
[4404]185        return *this;
[4276]186    }
187
[4268]188    inline const StatementList & statements() const {
[4404]189        return *this;
[4210]190    }
[4416]191
[4433]192    inline String * getName(const std::string name, const bool generated = true) const {
[4860]193        return mSymbolGenerator->get(name, generated);
[4410]194    }
[4416]195
[4433]196    inline String * makeName(const std::string prefix, const bool generated = true) const {
[4860]197        return mSymbolGenerator->make(prefix, generated);
[4410]198    }
[4416]199
[4699]200    inline Integer * getInteger(Integer::Type value) {
[4860]201        return mSymbolGenerator->getInteger(value);
[4443]202    }
203
[4521]204    inline PabloBlock * getParent() const {
205        return mParent;
206    }
[4540]207   
[4870]208    void setParent(PabloBlock * parent) {
209        mParent = parent;
210        // Add test to assert this block is in the same function.
211    }
212
[4650]213    void insert(Statement * const statement);
214
[4687]215    unsigned enumerateScopes(unsigned baseScopeIndex);
[4540]216   
[4687]217    inline unsigned getScopeIndex() const {
218        return mScopeIndex;
219    }
220   
[4870]221    void eraseFromParent(const bool recursively = false);
222
[4432]223    virtual ~PabloBlock();
224
[4404]225protected:
[4680]226
[4870]227    explicit PabloBlock(SymbolGenerator * symbolGenerator) noexcept;
[4276]228
[4438]229    PabloAST * renameNonNamedNode(PabloAST * expr, const std::string && prefix);
230
[4410]231    template<typename Type>
[4416]232    inline Type * insertAtInsertionPoint(Type * expr) {
233        if (isa<Statement>(expr)) {
[4611]234            if (LLVM_UNLIKELY(isa<If>(expr) || isa<While>(expr))) {
[4870]235                PabloBlock * const body = isa<If>(expr) ? cast<If>(expr)->getBody() : cast<While>(expr)->getBody();
236                body->setParent(this);
237                addUser(body);
[4611]238            }
[4416]239            insert(cast<Statement>(expr));
[4410]240        }
[4416]241        return expr;
[4410]242    }
[4692]243
[4602]244private:
[4410]245
[4692]246    Call * createCall(PabloAST * prototype, const std::vector<PabloAST *> &);
[4602]247
248    Var * createVar(PabloAST * name);
249
[4258]250private:       
[4797]251    static Zeroes                                       mZeroes;
252    static Ones                                         mOnes;
[4870]253    SymbolGenerator *                                   mSymbolGenerator; // TODO: need a better way of passing a symbol generator around
[4521]254    PabloBlock *                                        mParent;
[4687]255    unsigned                                            mScopeIndex;
[4197]256};
257
[4206]258}
259
[3850]260#endif // PS_PABLOS_H
Note: See TracBrowser for help on using the repository browser.