Add a method to TreeNode to return total number of nodes in a subtree.

app/tree_node_model.h

 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef APP_TREE_NODE_MODEL_H_
 #define APP_TREE_NODE_MODEL_H_
 
 #include <algorithm>
 #include <vector>
 
@@ skipping 86 matching lines @@
     for (size_t i = 0; i < children_->size(); ++i)
       children_[i]->parent_ = NULL;
     children_->clear();
   }
 
   // Returns the number of children.
   int GetChildCount() const {
     return static_cast<int>(children_->size());
   }
 
+  // Returns the number of all nodes in teh subtree rooted at this node,
+  // including this node.
+  int GetTotalNodeCount() const {
+    int count = 1;  // Start with one to include the node itself.
+    for (size_t i = 0; i < children_->size()(); ++i) {

[sky, 2009/10/23 21:50:21]

()()->()

+      TreeNode<NodeType> *child = children_[i];
+      count += child->GetTotalNodeCount();
+    }
+    return count;
+  }
+
   // Returns a child by index.
   NodeType* GetChild(int index) {
     DCHECK(index >= 0 && index < GetChildCount());
     return children_[index];
   }
   const NodeType* GetChild(int index) const {
     DCHECK(index >= 0 && index < GetChildCount());
     return children_[index];
   }
 
@@ skipping 160 matching lines @@
   // The root.
   scoped_ptr<NodeType> root_;
 
   // The observer.
   TreeModelObserver* observer_;
 
   DISALLOW_COPY_AND_ASSIGN(TreeNodeModel);
 };
 
 #endif  // APP_TREE_NODE_MODEL_H_