Sets now match Hashes better, and can be Archived.

1 files changed, 31 insertions, 1 deletions

diff --git a/src/archive.h b/src/archive.h
index 24c3a67..b5477bb 100644
--- a/src/archive.h
+++ b/src/archive.h
@@ -8,6 +8,7 @@
 #include <list>
 #include "bu/hash.h"
 #include "bu/list.h"
+#include "bu/set.h"
 
 namespace Bu
 {
@@ -235,7 +236,36 @@ namespace Bu
                 }
 
                 return ar;
-        }       
+        }
+
+        template<typename value>
+        Archive &operator<<( Archive &ar, Set<value> &h )
+        {
+                ar << h.getSize();
+                for( typename Set<value>::iterator i = h.begin(); i != h.end(); i++ )
+                {
+                        ar << (*i);
+                }
+
+                return ar;
+        }
+
+        template<typename value>
+        Archive &operator>>( Archive &ar, Set<value> &h )
+        {
+                h.clear();
+                uint32_t nSize;
+                ar >> nSize;
+
+                for( uint32_t j = 0; j < nSize; j++ )
+                {
+                        value v;
+                        ar >> v;
+                        h.insert( v );
+                }
+
+                return ar;
+        }
 }
 
 #endif