Abbreviate (def (fn ...))

git-svn-id: http://svn.drobilla.net/resp/resp@381 ad02d1e2-f140-0410-9f75-f8b11f17cedd

1 files changed, 17 insertions, 7 deletions

diff --git a/src/pprint.cpp b/src/pprint.cpp
index 0ea80ef..718e43b 100644
--- a/src/pprint.cpp
+++ b/src/pprint.cpp
@@ -137,14 +137,24 @@ print_to(ostream& out, const AST* ast, unsigned indent, CEnv* cenv, bool types)
 		}
 
 		if (form == "def") {
-			out << (*i++); // Print symbol
-			unsigned child_indent = types ? indent + 2 : indent;
-			if (types) {
-				print_annotation(out, tup->list_ref(2), indent + head_width + 1, cenv, true);
-				newline(out, child_indent);
+			if (tup->rrst() && is_form(tup->frrst(), "fn")) {
+				// Abreviate (def (fn (...) ...))
+				out << "(" << (*i++) << " ";
+				const ATuple* const fn   = tup->frrst()->as_tuple();
+				const ATuple* const prot = fn->frst()->as_tuple();
+				print_list_one_line(out, prot, prot->begin(), indent + 7, cenv, types, types);
+				newline(out, indent + 2);
+				print_list(out, fn, fn->iter_at(2), indent + 2, cenv, types, false);
+			} else {
+				const unsigned child_indent = types ? indent + 2 : indent;
+				out << (*i++); // Print symbol
+				if (types) {
+					print_annotation(out, tup->list_ref(2), indent + head_width + 1, cenv, true);
+					newline(out, child_indent);
+				}
+				out << " ";
+				print_to(out, *i++, child_indent, cenv, types);
 			}
-			out << " ";
-			print_to(out, *i++, child_indent, cenv, types);
 			out << ")";
 
 		} else if (form == "def-type") {