Add command line arguments

newptcai · Feb 25, 2020 · ce8b2ae · ce8b2ae
1 parent 6ef4eba
commit ce8b2ae
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/args.jl b/src/args.jl
@@ -41,6 +41,9 @@ function parse_commandline()
         "log-prod"
             action = :command
             help="simulation of log product of subtree sizes"
+        "size-power"
+            action = :command
+            help="simulation of sum of power of subtree sizes"
         "leaf"
             action = :command
             help="simulation of the number of leaves coutning"
@@ -72,5 +75,14 @@ function parse_commandline()
             arg_type=Int
             default=1
     end
+
+    @add_arg_table! setting["size-power"] begin
+        "--pow", "-p" 
+            dest_name="pow"
+            help="subtree size^pow"
+            arg_type=Real
+            default=0.5
+    end
+
     return parse_args(setting)
 end
diff --git a/src/main.jl b/src/main.jl
@@ -28,6 +28,8 @@ function main()
         sim = KcutSimulator(tree_ins, parsed_args["kcut"]["k"])
     elseif parsed_args["%COMMAND%"] == "log-prod"
         sim = LogProductSimulator(tree_ins, parsed_args["log-prod"]["pow"])
+    elseif parsed_args["%COMMAND%"] == "size-power"
+        sim = SizePowerSimulator(tree_ins, parsed_args["size-power"]["pow"])
     elseif parsed_args["%COMMAND%"] == "height"
         sim = HeightSimulator(tree_ins)
     elseif parsed_args["%COMMAND%"] == "total-path"