The Bu::Archive class now supports arbitrary, named properties via variants.

2 files changed, 22 insertions, 1 deletions

diff --git a/src/archive.h b/src/archive.h
index 7b74999..bd85113 100644
--- a/src/archive.h
+++ b/src/archive.h
@@ -12,6 +12,7 @@
 #include "bu/archivebase.h"
 #include "bu/hash.h"
 #include "bu/util.h"
+#include "bu/variant.h"
 
 namespace Bu
 {
@@ -109,11 +110,28 @@ namespace Bu
                  */
                 void readID( const void *ptr, uint32_t id );
 
+                template<typename t>
+                void setProp( const Bu::FString &sId, const t &val )
+                {
+                        if( !hProps.has( sId ) )
+                        {
+                                hProps.insert( sId, Variant() );
+                        }
+                        hProps.get( sId ) = val;
+                }
+
+                template<typename t>
+                t getProp( const Bu::FString &sId )
+                {
+                        return hProps.get( sId );
+                }
+
         private:
                 Stream &rStream;
                 uint32_t nNextID;
                 Hash<uint32_t,uint32_t> hPtrID;
                 Hash<uint32_t,List<void **> > hPtrDest;
+                Hash<Bu::FString, Variant> hProps;
         };
 }
 
diff --git a/src/tests/archive.cpp b/src/tests/archive.cpp
index a0bc78a..08e3b17 100644
--- a/src/tests/archive.cpp
+++ b/src/tests/archive.cpp
@@ -13,12 +13,15 @@ using namespace Bu;
 
 int main()
 {
-        File f("test.dat", File::Write );
+        File f("test.dat", File::WriteNew );
         Archive ar( f, Archive::save );
 
         Bu::FString s("Hello there");
         ar << s;
 
+        ar.setProp("hi", 45 );
+        printf("Hi: %d", ar.getProp<int>("hi") );
+
         return 0;
 }