summaryrefslogtreecommitdiff
path: root/src/printer/ast/ast_printer.cpp
diff options
context:
space:
mode:
authorMorgan Deters <mdeters@gmail.com>2010-11-15 22:57:14 +0000
committerMorgan Deters <mdeters@gmail.com>2010-11-15 22:57:14 +0000
commit5e5956d492ab18b5b4d4bb51117ac760867a525d (patch)
tree0c151baa58810722288ad986dfa13123de273739 /src/printer/ast/ast_printer.cpp
parentec4e1bdba56565d6372cb19ded12c9cadc506870 (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/ast_printer.cpp')
-rw-r--r--src/printer/ast/ast_printer.cpp100
1 files changed, 100 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 */
+
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback