Improve printing of ASTs
This commit is contained in:
parent
84a13dc576
commit
ecc2c8f464
3 changed files with 58 additions and 16 deletions
|
@ -16,6 +16,47 @@ std::ostream & operator << (std::ostream & str, Expr & e)
|
||||||
return str;
|
return str;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void showString(std::ostream & str, const string & s)
|
||||||
|
{
|
||||||
|
str << '"';
|
||||||
|
for (auto c : (string) s)
|
||||||
|
if (c == '"' || c == '\\' || c == '$') str << "\\" << c;
|
||||||
|
else if (c == '\n') str << "\\n";
|
||||||
|
else if (c == '\r') str << "\\r";
|
||||||
|
else if (c == '\t') str << "\\t";
|
||||||
|
else str << c;
|
||||||
|
str << '"';
|
||||||
|
}
|
||||||
|
|
||||||
|
static void showId(std::ostream & str, const string & s)
|
||||||
|
{
|
||||||
|
assert(!s.empty());
|
||||||
|
if (s == "if")
|
||||||
|
str << '"' << s << '"';
|
||||||
|
else {
|
||||||
|
char c = s[0];
|
||||||
|
if (!((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || c == '_')) {
|
||||||
|
showString(str, s);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
for (auto c : s)
|
||||||
|
if (!((c >= 'a' && c <= 'z') ||
|
||||||
|
(c >= 'A' && c <= 'Z') ||
|
||||||
|
(c >= '0' && c <= '9') ||
|
||||||
|
c == '_' || c == '\'' || c == '-')) {
|
||||||
|
showString(str, s);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
str << s;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::ostream & operator << (std::ostream & str, const Symbol & sym)
|
||||||
|
{
|
||||||
|
showId(str, *sym.s);
|
||||||
|
return str;
|
||||||
|
}
|
||||||
|
|
||||||
void Expr::show(std::ostream & str)
|
void Expr::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
abort();
|
abort();
|
||||||
|
@ -28,7 +69,7 @@ void ExprInt::show(std::ostream & str)
|
||||||
|
|
||||||
void ExprString::show(std::ostream & str)
|
void ExprString::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "\"" << s << "\""; // !!! escaping
|
showString(str, s);
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprPath::show(std::ostream & str)
|
void ExprPath::show(std::ostream & str)
|
||||||
|
@ -44,12 +85,12 @@ void ExprVar::show(std::ostream & str)
|
||||||
void ExprSelect::show(std::ostream & str)
|
void ExprSelect::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "(" << *e << ")." << showAttrPath(attrPath);
|
str << "(" << *e << ")." << showAttrPath(attrPath);
|
||||||
if (def) str << " or " << *def;
|
if (def) str << " or (" << *def << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprOpHasAttr::show(std::ostream & str)
|
void ExprOpHasAttr::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "(" << *e << ") ? " << showAttrPath(attrPath);
|
str << "((" << *e << ") ? " << showAttrPath(attrPath) << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprAttrs::show(std::ostream & str)
|
void ExprAttrs::show(std::ostream & str)
|
||||||
|
@ -85,6 +126,10 @@ void ExprLambda::show(std::ostream & str)
|
||||||
str << i->name;
|
str << i->name;
|
||||||
if (i->def) str << " ? " << *i->def;
|
if (i->def) str << " ? " << *i->def;
|
||||||
}
|
}
|
||||||
|
if (formals->ellipsis) {
|
||||||
|
if (!first) str << ", ";
|
||||||
|
str << "...";
|
||||||
|
}
|
||||||
str << " }";
|
str << " }";
|
||||||
if (!arg.empty()) str << " @ ";
|
if (!arg.empty()) str << " @ ";
|
||||||
}
|
}
|
||||||
|
@ -94,23 +139,24 @@ void ExprLambda::show(std::ostream & str)
|
||||||
|
|
||||||
void ExprLet::show(std::ostream & str)
|
void ExprLet::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "let ";
|
str << "(let ";
|
||||||
foreach (ExprAttrs::AttrDefs::iterator, i, attrs->attrs)
|
foreach (ExprAttrs::AttrDefs::iterator, i, attrs->attrs)
|
||||||
if (i->second.inherited)
|
if (i->second.inherited) {
|
||||||
str << "inherit " << i->first << "; ";
|
str << "inherit " << i->first << "; ";
|
||||||
|
}
|
||||||
else
|
else
|
||||||
str << i->first << " = " << *i->second.e << "; ";
|
str << i->first << " = " << *i->second.e << "; ";
|
||||||
str << "in " << *body;
|
str << "in " << *body << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprWith::show(std::ostream & str)
|
void ExprWith::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "with " << *attrs << "; " << *body;
|
str << "(with " << *attrs << "; " << *body << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprIf::show(std::ostream & str)
|
void ExprIf::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "if " << *cond << " then " << *then << " else " << *else_;
|
str << "(if " << *cond << " then " << *then << " else " << *else_ << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprAssert::show(std::ostream & str)
|
void ExprAssert::show(std::ostream & str)
|
||||||
|
@ -120,16 +166,18 @@ void ExprAssert::show(std::ostream & str)
|
||||||
|
|
||||||
void ExprOpNot::show(std::ostream & str)
|
void ExprOpNot::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
str << "! " << *e;
|
str << "(! " << *e << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprConcatStrings::show(std::ostream & str)
|
void ExprConcatStrings::show(std::ostream & str)
|
||||||
{
|
{
|
||||||
bool first = true;
|
bool first = true;
|
||||||
|
str << "(";
|
||||||
foreach (vector<Expr *>::iterator, i, *es) {
|
foreach (vector<Expr *>::iterator, i, *es) {
|
||||||
if (first) first = false; else str << " + ";
|
if (first) first = false; else str << " + ";
|
||||||
str << **i;
|
str << **i;
|
||||||
}
|
}
|
||||||
|
str << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
void ExprPos::show(std::ostream & str)
|
void ExprPos::show(std::ostream & str)
|
||||||
|
|
|
@ -280,7 +280,7 @@ struct ExprOpNot : Expr
|
||||||
Expr##name(const Pos & pos, Expr * e1, Expr * e2) : pos(pos), e1(e1), e2(e2) { }; \
|
Expr##name(const Pos & pos, Expr * e1, Expr * e2) : pos(pos), e1(e1), e2(e2) { }; \
|
||||||
void show(std::ostream & str) \
|
void show(std::ostream & str) \
|
||||||
{ \
|
{ \
|
||||||
str << *e1 << " " s " " << *e2; \
|
str << "(" << *e1 << " " s " " << *e2 << ")"; \
|
||||||
} \
|
} \
|
||||||
void bindVars(const StaticEnv & env) \
|
void bindVars(const StaticEnv & env) \
|
||||||
{ \
|
{ \
|
||||||
|
|
|
@ -58,12 +58,6 @@ public:
|
||||||
friend std::ostream & operator << (std::ostream & str, const Symbol & sym);
|
friend std::ostream & operator << (std::ostream & str, const Symbol & sym);
|
||||||
};
|
};
|
||||||
|
|
||||||
inline std::ostream & operator << (std::ostream & str, const Symbol & sym)
|
|
||||||
{
|
|
||||||
str << *sym.s;
|
|
||||||
return str;
|
|
||||||
}
|
|
||||||
|
|
||||||
class SymbolTable
|
class SymbolTable
|
||||||
{
|
{
|
||||||
private:
|
private:
|
||||||
|
|
Loading…
Reference in a new issue