Add option to TernaryTreeDictionary command line to show tree node de…

…pth histogram.
vt-middleware · Feb 11, 2019 · 4b1eb48 · 4b1eb48
1 parent 887734d
commit 4b1eb48
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/org/passay/dictionary/TernaryTreeDictionary.java b/src/main/java/org/passay/dictionary/TernaryTreeDictionary.java
@@ -165,6 +165,7 @@ public static void main(final String[] args)
       boolean partialSearch = false;
       boolean nearSearch = false;
       boolean print = false;
+      boolean stats = false;
 
       // operation parameters
       String word = null;
@@ -187,6 +188,8 @@ public static void main(final String[] args)
           distance = Integer.parseInt(args[++i]);
         } else if ("-p".equals(args[i])) {
           print = true;
+        } else if ("-st".equals(args[i])) {
+          stats = true;
         } else if ("-h".equals(args[i])) {
           throw new ArrayIndexOutOfBoundsException();
         } else {
@@ -228,6 +231,9 @@ public static void main(final String[] args)
             Arrays.asList(matches)));
       } else if (print) {
         dict.getTernaryTree().print(new PrintWriter(System.out, true));
+      } else if (stats) {
+        System.out.println("word path depths histogram:");
+        System.out.println(dict.getTernaryTree().getNodeStats());
       } else {
         throw new ArrayIndexOutOfBoundsException();
       }