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

Last change on this file since 4276 was 4276, checked in by nmedfort, 5 years ago

Created an "insertion friendly" Pablo AST structure similar to LLVM; fixed engine builder bug when creating a unicode class.

File size: 3.9 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 PabloPrinter::print(const PabloBlock & block)
41{
42    std::string strOut = "[";
43
44    strOut = strOut.substr(0, strOut.length() - 1);
45    strOut += "],[";
46
47    strOut += print(block.statements());
48
49    strOut = strOut.substr(0, strOut.length() - 1);
50    strOut += "]";
51
52    return strOut;
53}
54
55std::string PabloPrinter::print(const StatementList & stmts) {
56    std::string strOut = "";
57    for (const Statement * stmt : stmts) {
58        strOut += print(stmt) + "\n";
59    }
60    return strOut;
61}
62
63std::string PabloPrinter::print(const Statement * stmt)
64{
65    if (const Assign * an = dyn_cast<const Assign>(stmt)) {
66        std::string result = "Assign('" + an->getName()->str() + "'," + print(an->getExpr());
67        if (an->isOutputAssignment()) {
68            result += ",Output=" + std::to_string(an->getOutputIndex());
69        }
70        return result + ")";
71    }
72    else if (const Next * next = dyn_cast<const Next>(stmt)) {
73        return "Next(" + next->getName()->str() + "," + print(next->getExpr()) + ")";
74    }
75    else if (const If * ifstmt = dyn_cast<const If>(stmt)) {
76        return "If(" + print(ifstmt->getCondition()) + "," + print(ifstmt->getBody()) + ")";
77    }
78    else if (const While * whl = dyn_cast<const While>(stmt)) {
79        return "While(" + print(whl->getCondition()) + "," + print(whl->getBody()) + ")";
80    }
81    return "???";
82}
83
84std::string PabloPrinter::print(const PabloAST *expr) {
85    if (isa<const Zeroes>(expr)) {
86        return "Zeroes";
87    }
88    else if (isa<const Ones>(expr)) {
89        return "Ones";
90    }
91    else if (const Call * pablo_call = dyn_cast<const Call>(expr)) {
92        return "Call '" + pablo_call->getCallee()->str() + "'";
93    }
94    else if (const Var * pablo_var = dyn_cast<const Var>(expr)) {
95        return "Var '" + pablo_var->getName()->str() + "' ";
96    }
97    else if (const And * pablo_and = dyn_cast<const And>(expr)) {
98        return "And(" + print(pablo_and->getExpr1()) +"," + print(pablo_and->getExpr2()) + ")";
99    }
100    else if (const Or * pablo_or = dyn_cast<const Or>(expr)) {
101        return "Or(" + print(pablo_or->getExpr1()) + "," + print(pablo_or->getExpr2()) + ")";
102    }
103    else if (const Sel * pablo_sel = dyn_cast<const Sel>(expr)) {
104        return "(" + print(pablo_sel->getCondition()) + " ? " + print(pablo_sel->getTrueExpr()) + " : " + print(pablo_sel->getFalseExpr()) + ")";
105    }
106    else if (const Not * pablo_not = dyn_cast<const Not>(expr)) {
107        return "Not (" + print(pablo_not->getExpr()) + ")";
108    }
109    else if (const Advance * adv = dyn_cast<const Advance>(expr)) {
110        return "Advance(" + print(adv->getExpr()) + ", " + std::to_string(adv->getAdvanceAmount()) + ")";
111    }
112    else if (const MatchStar * mstar = dyn_cast<const MatchStar>(expr)) {
113        return "MatchStar(" + print(mstar->getMarker()) + ", " + print(mstar->getCharClass()) + ")";
114    }
115    else if (const ScanThru * sthru = dyn_cast<const ScanThru>(expr)) {
116        return "ScanThru(" + print(sthru->getScanFrom()) + ", " + print(sthru->getScanThru()) + ")";
117    }
118    return "???";
119}
120
121
Note: See TracBrowser for help on using the repository browser.