From aa122695e53bb2c39175e2d04f626b2cd6c2e911 Mon Sep 17 00:00:00 2001
From: Alon Zakai <alonzakai@gmail.com>
Date: Tue, 3 Nov 2015 10:18:14 -0800
Subject: refactor pretty printing code

---
 src/pretty_printing.h | 70 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 src/pretty_printing.h

(limited to 'src/pretty_printing.h')

diff --git a/src/pretty_printing.h b/src/pretty_printing.h
new file mode 100644
index 000000000..71a729625
--- /dev/null
+++ b/src/pretty_printing.h
@@ -0,0 +1,70 @@
+
+#include <ostream>
+
+#include "colors.h"
+
+std::ostream &doIndent(std::ostream &o, unsigned indent) {
+  for (unsigned i = 0; i < indent; i++) {
+    o << "  ";
+  }
+  return o;
+}
+
+std::ostream &incIndent(std::ostream &o, unsigned& indent) {
+  o << '\n';
+  indent++;
+  return o; 
+}
+
+std::ostream &decIndent(std::ostream &o, unsigned& indent) {
+  indent--;
+  doIndent(o, indent);
+  return o << ')';
+}
+
+std::ostream &prepareMajorColor(std::ostream &o) {
+  Colors::red(o);
+  Colors::bold(o);
+  return o;
+}
+
+std::ostream &prepareColor(std::ostream &o) {
+  Colors::magenta(o);
+  Colors::bold(o);
+  return o;
+}
+
+std::ostream &prepareMinorColor(std::ostream &o) {
+  Colors::orange(o);
+  return o;
+}
+
+std::ostream &restoreNormalColor(std::ostream &o) {
+  Colors::normal(o);
+  return o;
+}
+
+std::ostream& printText(std::ostream &o, const char *str) {
+  o << '"';
+  Colors::green(o);
+  o << str;
+  Colors::normal(o);
+  return o << '"';
+}
+
+std::ostream& printOpening(std::ostream &o, const char *str, bool major=false) {
+  o << '(';
+  major ? prepareMajorColor(o) : prepareColor(o);
+  o << str;
+  restoreNormalColor(o);
+  return o;
+}
+
+std::ostream& printMinorOpening(std::ostream &o, const char *str) {
+  o << '(';
+  prepareMinorColor(o);
+  o << str;
+  restoreNormalColor(o);
+  return o;
+}
+
-- 
cgit v1.2.3