From 70d1c41ee4d44ee5ff76f2f290db87981974cac5 Mon Sep 17 00:00:00 2001
From: Linus Nordberg <linus@nordu.net>
Date: Sat, 26 Apr 2014 14:24:28 +0200
Subject: Don't export tree_version/1 but add size/1.

---
 src/ht.erl | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

(limited to 'src')

diff --git a/src/ht.erl b/src/ht.erl
index 1fb4dd5..3d7c02e 100644
--- a/src/ht.erl
+++ b/src/ht.erl
@@ -14,8 +14,7 @@
 -type inner() :: #inner{}.
 
 -export_type([head/0, tree/0, inner/0, leaf/0]).
--export([create/0, append/2, tree_hash/1, tree_version/1,
-        audit_path/2]).
+-export([create/0, append/2, tree_hash/1, size/1, audit_path/2]).
 
 %% Public interface.
 -spec create() -> head().
@@ -31,12 +30,9 @@ tree_hash(#head{tree=T}) ->
         #leaf{hash=H} -> H
     end.
 
-%% @doc Tree version number, i.e. number of leafs in tree. Note that
-%% this is set off by one (one higher) compared with the history tree
-%% version as explained by Crosby and Wallach.
--spec tree_version(head()) -> non_neg_integer().
-tree_version(#head{version=Ver}) ->
-    Ver.
+-spec size(head()) -> non_neg_integer().
+size(Head) ->
+    tree_version(Head).
 
 %% @doc Append Leaf to Head.
 %%
@@ -87,6 +83,13 @@ audit_path(Head, Index) ->
 %%%%%%%%%%%%%%%%%%%%
 %% Private functions.
 
+%% @doc Tree version number, i.e. number of leafs in tree. Note that
+%% this is set off by one (one higher) compared with the history tree
+%% version as explained by Crosby and Wallach.
+-spec tree_version(head()) -> non_neg_integer().
+tree_version(#head{version=Ver}) ->
+    Ver.
+
 -spec mkhead(non_neg_integer(), tree()) -> head();
             (head(), list()) -> head().
 mkhead(Version, Tree) when is_integer(Version) ->
-- 
cgit v1.1