network: Implement function to count all nodes

[people/ms/libloc.git] / src / network.c

diff --git a/src/network.c b/src/network.c
index d34f881e1ac4f616c31fd464680e4790dac04228..7c7a819ba7fde5bc12c9ef74321e12c3080d8311 100644 (file)
--- a/src/network.c
+++ b/src/network.c
@@ -527,3 +527,19 @@ LOC_EXPORT size_t loc_network_tree_count_networks(struct loc_network_tree* tree)
 
        return counter;
 }
+
+static size_t __loc_network_tree_count_nodes(struct loc_network_tree_node* node) {
+       size_t counter = 1;
+
+       if (node->zero)
+               counter += __loc_network_tree_count_nodes(node->zero);
+
+       if (node->one)
+               counter += __loc_network_tree_count_nodes(node->one);
+
+       return counter;
+}
+
+LOC_EXPORT size_t loc_network_tree_count_nodes(struct loc_network_tree* tree) {
+       return __loc_network_tree_count_nodes(tree->root);
+}