Changeset 11356

Timestamp:

Oct 26, 2010, 7:30:35 PM (12 years ago)

Author:

charles

Message:

(trunk libT) #3678 "benc walking could be more efficient" -- fixed.

File:

• trunk/libtransmission/bencode.c (modified) (11 diffs)

trunk/libtransmission/bencode.c

@@ -942 +942 @@
 };
 
-static struct SaveNode*
-nodeNewDict( const tr_benc * val )
+static void
+nodeInitDict( struct SaveNode * node, const tr_benc * val )
 {
     int               i, j;
     int               nKeys;
-    struct SaveNode * node;
     struct KeyIndex * indices;
 
@@ -953 +952 @@
 
     nKeys = val->val.l.count / 2;
-    node = tr_new0( struct SaveNode, 1 );
     node->val = val;
     node->children = tr_new0( int, nKeys * 2 );
@@ -974 +972 @@
     assert( node->childCount == nKeys * 2 );
     tr_free( indices );
-    return node;
-}
-
-static struct SaveNode*
-nodeNewList( const tr_benc * val )
+}
+
+static void
+nodeInitList( struct SaveNode * node, const tr_benc * val )
 {
     int               i, n;
-    struct SaveNode * node;
 
     assert( tr_bencIsList( val ) );
 
     n = val->val.l.count;
-    node = tr_new0( struct SaveNode, 1 );
     node->val = val;
     node->childCount = n;
@@ -992 +987 @@
     for( i = 0; i < n; ++i ) /* a list's children don't need to be reordered */
         node->children[i] = i;
-
-    return node;
-}
-
-static struct SaveNode*
-nodeNewLeaf( const tr_benc * val )
-{
-    struct SaveNode * node;
-
+}
+
+static void
+nodeInitLeaf( struct SaveNode * node, const tr_benc * val )
+{
     assert( !isContainer( val ) );
 
-    node = tr_new0( struct SaveNode, 1 );
     node->val = val;
-    return node;
-}
-
-static struct SaveNode*
-nodeNew( const tr_benc * val )
-{
-    struct SaveNode * node;
-
-    if( tr_bencIsList( val ) )
-        node = nodeNewList( val );
-    else if( tr_bencIsDict( val ) )
-        node = nodeNewDict( val );
-    else
-        node = nodeNewLeaf( val );
-
-    return node;
+}
+
+static void
+nodeInit( struct SaveNode * node, const tr_benc * val )
+{
+    static const struct SaveNode INIT_NODE = { NULL, 0, 0, 0, NULL };
+    *node = INIT_NODE;
+
+         if( tr_bencIsList( val ) ) nodeInitList( node, val );
+    else if( tr_bencIsDict( val ) ) nodeInitDict( node, val );
+    else                            nodeInitLeaf( node, val );
 }
 
@@ -1046 +1031 @@
           void                   * user_data )
 {
-    tr_ptrArray stack = TR_PTR_ARRAY_INIT;
-
-    tr_ptrArrayAppend( &stack, nodeNew( top ) );
-
-    while( !tr_ptrArrayEmpty( &stack ) )
+    int stackSize = 0;
+    int stackAlloc = 64;
+    struct SaveNode * stack = tr_new( struct SaveNode, stackAlloc );
+
+    nodeInit( &stack[stackSize++], top );
+
+    while( stackSize > 0 )
     {
-        struct SaveNode * node = tr_ptrArrayBack( &stack );
+        struct SaveNode * node = &stack[stackSize-1];
         const tr_benc *   val;
 
@@ -1069 +1056 @@
             if( isContainer( node->val ) )
                 walkFuncs->containerEndFunc( node->val, user_data );
-            tr_ptrArrayPop( &stack );
+            --stackSize;
             tr_free( node->children );
-            tr_free( node );
             continue;
         }
@@ -1094 +1080 @@
 
                 case TR_TYPE_LIST:
-                    if( val != node->val )
-                        tr_ptrArrayAppend( &stack, nodeNew( val ) );
-                    else
+                    if( val == node->val )
                         walkFuncs->listBeginFunc( val, user_data );
+                    else {
+                        if( stackAlloc == stackSize ) {
+                            stackAlloc *= 2;
+                            stack = tr_renew( struct SaveNode, stack, stackAlloc );
+                        }
+                        nodeInit( &stack[stackSize++], val );
+                    }
                     break;
 
                 case TR_TYPE_DICT:
-                    if( val != node->val )
-                        tr_ptrArrayAppend( &stack, nodeNew( val ) );
-                    else
+                    if( val == node->val )
                         walkFuncs->dictBeginFunc( val, user_data );
+                    else {
+                        if( stackAlloc == stackSize ) {
+                            stackAlloc *= 2;
+                            stack = tr_renew( struct SaveNode, stack, stackAlloc );
+                        }
+                        nodeInit( &stack[stackSize++], val );
+                    }
                     break;
 
@@ -1114 +1110 @@
     }
 
-    tr_ptrArrayDestruct( &stack, NULL );
+    tr_free( stack );
 }
 
@@ -1192 +1188 @@
 
 static void
-freeDummyFunc( const tr_benc * val UNUSED,
-               void * buf          UNUSED  )
+freeDummyFunc( const tr_benc * val UNUSED, void * buf UNUSED  )
 {}
 
 static void
-freeStringFunc( const tr_benc * val,
-                void *          freeme )
+freeStringFunc( const tr_benc * val, void * unused UNUSED )
 {
     if( stringIsAlloced( val ) )
-        tr_ptrArrayAppend( freeme, val->val.s.str.ptr );
-}
-
-static void
-freeContainerBeginFunc( const tr_benc * val,
-                        void *          freeme )
-{
-    tr_ptrArrayAppend( freeme, val->val.l.vals );
+        tr_free( val->val.s.str.ptr );
+}
+
+static void
+freeContainerEndFunc( const tr_benc * val, void * unused UNUSED )
+{
+    tr_free( val->val.l.vals );
 }
 
@@ -1215 +1208 @@
                                                 freeDummyFunc,
                                                 freeStringFunc,
-                                                freeContainerBeginFunc,
-                                                freeContainerBeginFunc,
-                                                freeDummyFunc };
+                                                freeDummyFunc,
+                                                freeDummyFunc,
+                                                freeContainerEndFunc };
 
 void
@@ -1223 +1216 @@
 {
     if( isSomething( val ) )
-    {
-        tr_ptrArray a = TR_PTR_ARRAY_INIT;
-        bencWalk( val, &freeWalkFuncs, &a );
-        tr_ptrArrayDestruct( &a, tr_free );
-    }
+        bencWalk( val, &freeWalkFuncs, NULL );
 }