source: icGREP/icgrep-devel/icgrep/pablo/printer_pablos.cpp @ 4272

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

Generalized the writing of output variables by adding a 'flag' to the Assign nodes.

File size: 4.5 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#include "printer_pablos.h"
8
9
10//Regular Expressions
11#include <re/re_re.h>
12#include <re/re_cc.h>
13#include <re/re_start.h>
14#include <re/re_end.h>
15#include <re/re_seq.h>
16#include <re/re_name.h>
17
18//Pablo Expressions
19#include <pablo/pabloAST.h>
20#include <pablo/pe_advance.h>
21#include <pablo/pe_and.h>
22#include <pablo/pe_call.h>
23#include <pablo/pe_matchstar.h>
24#include <pablo/pe_not.h>
25#include <pablo/pe_or.h>
26#include <pablo/pe_scanthru.h>
27#include <pablo/pe_sel.h>
28#include <pablo/pe_var.h>
29#include <pablo/pe_xor.h>
30#include <pablo/ps_assign.h>
31#include <pablo/ps_if.h>
32#include <pablo/ps_while.h>
33#include <pablo/pe_zeroes.h>
34#include <pablo/pe_ones.h>
35#include <pablo/codegenstate.h>
36
37using namespace re;
38using namespace pablo;
39
40std::string StatementPrinter::PrintStmts(const PabloBlock & cg_state)
41{
42    std::string strOut = "[";
43
44    strOut = strOut.substr(0, strOut.length() - 1);
45    strOut += "],[";
46
47    strOut += Print_PB_PabloStmts(cg_state.statements());
48
49    strOut = strOut.substr(0, strOut.length() - 1);
50    strOut += "]";
51
52    return strOut;
53}
54
55std::string StatementPrinter::Print_PB_PabloStmts(const StatementList & stmts) {
56    std::string strOut = "";
57    for (const auto stmt : stmts) {
58        strOut += ShowPabloS(stmt);
59    }
60    return strOut;
61}
62
63std::string StatementPrinter::Print_CC_PabloStmts(const pablo::StatementList &stmts) {
64    std::string strOut = "Total Statements: " + std::to_string(stmts.size()) + "\n";
65    for (const auto stmt : stmts) {
66        strOut += ShowPabloS(stmt) + "\n";
67    }
68    return strOut;
69}
70
71std::string StatementPrinter::ShowPabloS(const PabloAST * stmt)
72{
73    if (const Assign * an = dyn_cast<const Assign>(stmt)) {
74        std::string result = "Assign('" + an->getName()->str() + "', " + ShowPabloAST(an->getExpr());
75        if (an->isOutputAssignment()) {
76            result += ", Output=" + std::to_string(an->getOutputIndex());
77        }
78        return result + "),";
79    }
80    else if (const Next * next = dyn_cast<const Next>(stmt)) {
81        return "Next(" + next->getName()->str() + ", " + ShowPabloAST(next->getExpr()) + ")";
82    }
83    else if (const If * ifstmt = dyn_cast<const If>(stmt)) {
84        return "If(" + ShowPabloAST(ifstmt->getCondition()) + ", " + Print_PB_PabloStmts(ifstmt->getBody()) + ")";
85    }
86    else if (const While * whl = dyn_cast<const While>(stmt)) {
87        return "While(" + ShowPabloAST(whl->getCondition()) + ", " + Print_PB_PabloStmts(whl->getBody()) + ")";
88    }
89    return "???";
90}
91
92std::string StatementPrinter::ShowPabloAST(const PabloAST *expr) {
93    if (isa<const Zeroes>(expr)) {
94        return "Zeroes";
95    }
96    else if (isa<const Ones>(expr)) {
97        return "Ones";
98    }
99    else if (const Call * pablo_call = dyn_cast<const Call>(expr)) {
100        return "Call '" + pablo_call->getCallee()->str() + "'";
101    }
102    else if (const Var * pablo_var = dyn_cast<const Var>(expr)) {
103        return "Var '" + pablo_var->getName()->str() + "' ";
104    }
105    else if (const And * pablo_and = dyn_cast<const And>(expr)) {
106        return "And(" + ShowPabloAST(pablo_and->getExpr1()) +", " + ShowPabloAST(pablo_and->getExpr2()) + ")";
107    }
108    else if (const Or * pablo_or = dyn_cast<const Or>(expr)) {
109        return "Or(" + ShowPabloAST(pablo_or->getExpr1()) + ", " + ShowPabloAST(pablo_or->getExpr2()) + ")";
110    }
111    else if (const Sel * pablo_sel = dyn_cast<const Sel>(expr)) {
112        return "((" + ShowPabloAST(pablo_sel->getCondition()) + "And " + ShowPabloAST(pablo_sel->getTrueExpr()) +
113                ")|(Not(" + ShowPabloAST(pablo_sel->getCondition()) + ") And " + ShowPabloAST(pablo_sel->getFalseExpr()) + ")";
114    }
115    else if (const Not * pablo_not = dyn_cast<const Not>(expr)) {
116        return "Not (" + ShowPabloAST(pablo_not->getExpr()) + ")";
117    }
118    else if (const Advance * adv = dyn_cast<const Advance>(expr)) {
119        return "Advance(" + ShowPabloAST(adv->getExpr()) + ", " + std::to_string(adv->getAdvanceAmount()) + ")";
120    }
121    else if (const MatchStar * mstar = dyn_cast<const MatchStar>(expr)) {
122        return "MatchStar (" + ShowPabloAST(mstar->getMarker()) + ", " + ShowPabloAST(mstar->getCharClass()) + ")";
123    }
124    else if (const ScanThru * sthru = dyn_cast<const ScanThru>(expr)) {
125        return "ScanThru (" + ShowPabloAST(sthru->getScanFrom()) + ", " + ShowPabloAST(sthru->getScanThru()) + ")";
126    }
127    return "???";
128}
129
130
Note: See TracBrowser for help on using the repository browser.