diff options
author | Morgan Deters <mdeters@gmail.com> | 2010-11-15 22:57:14 +0000 |
---|---|---|
committer | Morgan Deters <mdeters@gmail.com> | 2010-11-15 22:57:14 +0000 |
commit | 5e5956d492ab18b5b4d4bb51117ac760867a525d (patch) | |
tree | 0c151baa58810722288ad986dfa13123de273739 /src/printer/ast | |
parent | ec4e1bdba56565d6372cb19ded12c9cadc506870 (diff) |
Pretty-printer infrastructure created (in src/printer) and SMT-LIBv2 printer
implemented. This new infrastructure removes support for pretty-printing
(even in the AST language) an Expr with reference count 0. Previously,
this was supported in a few places internally to the expr package, for
example in NodeBuilder. (Now, a NodeBuilder cannot be prettyprinted, you
must extract the Node before printing it.)
Diffstat (limited to 'src/printer/ast')
-rw-r--r-- | src/printer/ast/ast_printer.cpp | 100 | ||||
-rw-r--r-- | src/printer/ast/ast_printer.h | 42 |
2 files changed, 142 insertions, 0 deletions
diff --git a/src/printer/ast/ast_printer.cpp b/src/printer/ast/ast_printer.cpp new file mode 100644 index 000000000..cd9b0cad5 --- /dev/null +++ b/src/printer/ast/ast_printer.cpp @@ -0,0 +1,100 @@ +/********************* */ +/*! \file ast_printer.cpp + ** \verbatim + ** Original author: mdeters + ** Major contributors: none + ** Minor contributors (to current version): none + ** This file is part of the CVC4 prototype. + ** Copyright (c) 2009, 2010 The Analysis of Computer Systems Group (ACSys) + ** Courant Institute of Mathematical Sciences + ** New York University + ** See the file COPYING in the top-level source directory for licensing + ** information.\endverbatim + ** + ** \brief The pretty-printer interface for the AST output language + ** + ** The pretty-printer interface for the AST output language. + **/ + +#include "printer/ast/ast_printer.h" +#include "util/language.h" // for LANG_AST +#include "expr/node_manager.h" // for VarNameAttr + +#include <iostream> + +using namespace std; + +namespace CVC4 { +namespace printer { +namespace ast { + +std::ostream& AstPrinter::toStream(std::ostream& out, TNode n, + int toDepth, bool types) const { + // null + if(n.getKind() == kind::NULL_EXPR) { + out << "null"; + return out; + } + + // variable + if(n.getMetaKind() == kind::metakind::VARIABLE) { + if(n.getKind() != kind::VARIABLE && + n.getKind() != kind::SORT_TYPE) { + out << n.getKind() << ':'; + } + + string s; + if(n.getAttribute(expr::VarNameAttr(), s)) { + out << s; + } else { + out << "var_" << n.getId(); + } + if(types) { + // print the whole type, but not *its* type + out << ":"; + n.getType().toStream(out, -1, false, language::output::LANG_AST); + } + + return out; + } + + out << '(' << n.getKind(); + if(n.getMetaKind() == kind::metakind::CONSTANT) { + // constant + out << ' '; + kind::metakind::NodeValueConstPrinter::toStream(out, n); + } else { + // operator + if(n.getMetaKind() == kind::metakind::PARAMETERIZED) { + out << ' '; + if(toDepth != 0) { + n.getOperator().toStream(out, toDepth < 0 ? toDepth : toDepth - 1, + types, language::output::LANG_AST); + } else { + out << "(...)"; + } + } + for(TNode::iterator i = n.begin(), + iend = n.end(); + i != iend; + ++i) { + if(i != iend) { + out << ' '; + } + if(toDepth != 0) { + (*i).toStream(out, toDepth < 0 ? toDepth : toDepth - 1, + types, language::output::LANG_AST); + } else { + out << "(...)"; + } + } + } + out << ')'; + + return out; +}/* AstPrinter::toStream() */ + +}/* CVC4::printer::ast namespace */ +}/* CVC4::printer namespace */ +}/* CVC4 namespace */ + diff --git a/src/printer/ast/ast_printer.h b/src/printer/ast/ast_printer.h new file mode 100644 index 000000000..0851aef6c --- /dev/null +++ b/src/printer/ast/ast_printer.h @@ -0,0 +1,42 @@ +/********************* */ +/*! \file ast_printer.h + ** \verbatim + ** Original author: mdeters + ** Major contributors: none + ** Minor contributors (to current version): none + ** This file is part of the CVC4 prototype. + ** Copyright (c) 2009, 2010 The Analysis of Computer Systems Group (ACSys) + ** Courant Institute of Mathematical Sciences + ** New York University + ** See the file COPYING in the top-level source directory for licensing + ** information.\endverbatim + ** + ** \brief The pretty-printer interface for the AST output language + ** + ** The pretty-printer interface for the AST output language. + **/ + +#include "cvc4_private.h" + +#ifndef __CVC4__PRINTER__AST_PRINTER_H +#define __CVC4__PRINTER__AST_PRINTER_H + +#include <iostream> + +#include "printer/printer.h" + +namespace CVC4 { +namespace printer { +namespace ast { + +class AstPrinter : public CVC4::Printer { +public: + std::ostream& toStream(std::ostream& out, TNode n, int toDepth, bool types) const; +};/* class AstPrinter */ + +}/* CVC4::printer::ast namespace */ +}/* CVC4::printer namespace */ +}/* CVC4 namespace */ + +#endif /* __CVC4__PRINTER__AST_PRINTER_H */ + |