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

app/tree_node_model.h

Index: app/tree_node_model.h
===================================================================
--- app/tree_node_model.h	(revision 29549)
+++ app/tree_node_model.h	(working copy)
@@ -104,6 +104,17 @@
     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());