From 0877d011948b8f902e4b9124c625d5199b271f31 Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Sun, 26 Jan 2014 16:38:41 -0800
Subject: [PATCH] Add stream operator for parse states

---
 spec/stream_methods.h        | 13 +++++++++++++
 src/compiler/parse_table.cpp | 19 +++++++++++++++++++
 src/compiler/parse_table.h   |  2 ++
 3 files changed, 34 insertions(+)

diff --git a/spec/stream_methods.h b/spec/stream_methods.h
index 36879e4e..83dbe636 100644
--- a/spec/stream_methods.h
+++ b/spec/stream_methods.h
@@ -5,6 +5,7 @@
 #include <set>
 #include <unordered_set>
 #include <unordered_map>
+#include <vector>
 
 using std::cout;
 
@@ -21,6 +22,18 @@ namespace std {
         return stream << ">";
     }
     
+    template<typename T>
+    inline std::ostream& operator<<(std::ostream &stream, const std::vector<T> &vector) {
+        stream << std::string("#<vector: ");
+        bool started = false;
+        for (auto item : vector) {
+            if (started) stream << std::string(", ");
+            stream << item;
+            started = true;
+        }
+        return stream << ">";
+    }
+    
     template<typename T>
     inline std::ostream& operator<<(std::ostream &stream, const std::set<T> &set) {
         stream << std::string("#<set: ");
diff --git a/src/compiler/parse_table.cpp b/src/compiler/parse_table.cpp
index bc265fdc..4aaa83fd 100644
--- a/src/compiler/parse_table.cpp
+++ b/src/compiler/parse_table.cpp
@@ -59,6 +59,25 @@ namespace tree_sitter {
         return result;
     }
     
+    ostream& operator<<(ostream &stream, const ParseState &state) {
+        stream << string("#<parse_state ");
+        bool started1 = false;
+        for (auto pair : state.actions) {
+            if (started1) stream << string(", ");
+            stream << pair.first << string(" => #<set: ");
+            bool started2 = false;
+            for (auto action : pair.second) {
+                if (started2) stream << string(", ");
+                stream << action;
+                started2 = true;
+            }
+            stream << string(">");
+            started1 = true;
+        }
+        stream << string(">");
+        return stream;
+    }
+    
     // Table
     size_t ParseTable::add_state() {
         states.push_back(ParseState());
diff --git a/src/compiler/parse_table.h b/src/compiler/parse_table.h
index f9efcd92..82225bd6 100644
--- a/src/compiler/parse_table.h
+++ b/src/compiler/parse_table.h
@@ -54,6 +54,8 @@ namespace tree_sitter {
         size_t lex_state_index;
     };
     
+    std::ostream& operator<<(std::ostream &stream, const ParseState &state);
+    
     class ParseTable {
     public:
         size_t add_state();