Implemented encodeRootObject method

nicklockwood · Oct 31, 2012 · e278244 · e278244
1 parent 99a35a8
commit e278244
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/HRCoder/HRCoder.m b/HRCoder/HRCoder.m
@@ -274,8 +274,7 @@ - (id)initForWritingWithMutableData:(NSMutableData *)data
 - (id)archiveRootObject:(id)rootObject
 {
     [_knownObjects removeAllObjects];
-    if (rootObject) [_knownObjects setObject:rootObject forKey:HRCoderRootObjectKey];
-    [_stack setArray:[NSMutableArray arrayWithObject:[rootObject archivedObjectWithHRCoder:self]]];
+    [self encodeRootObject:rootObject];
     id plist = [_stack lastObject];
     [self finishEncoding];
     return plist;
@@ -342,6 +341,15 @@ - (void)encodeObject:(id)objv forKey:(NSString *)key
     if (object) [[_stack lastObject] setObject:object forKey:key];
 }
 
+- (void)encodeRootObject:(id)rootObject
+{
+    if (rootObject)
+    {
+        [_knownObjects setObject:rootObject forKey:HRCoderRootObjectKey];
+        [_stack setArray:[NSMutableArray arrayWithObject:[rootObject archivedObjectWithHRCoder:self]]];
+    }
+}
+
 - (void)encodeConditionalObject:(id)objv forKey:(NSString *)key
 {
     if ([[_knownObjects allValues] containsObject:objv])