From: Christoph Egger <christoph@christoph-egger.org>
Date: Sun, 24 Apr 2016 13:20:48 +0000 (+0200)
Subject: Allow selecting Experiment via comandline parameter
X-Git-Url: https://git.siccegge.de//index.cgi?p=master%2Fcool-versuchung.git;a=commitdiff_plain;h=a94ca5b2cb1a6ec23247e81083ad3ed03ade10b6

Allow selecting Experiment via comandline parameter
---

diff --git a/cool.py b/cool.py
index 8b5568f..bab8482 100755
--- a/cool.py
+++ b/cool.py
@@ -7,6 +7,7 @@ from versuchung.execute import shell, CommandFailed
 
 import re
 import os
+import argparse
 
 def natural_key(string_):
     return [int(s) if s.isdigit() else s for s in re.split(r'(\d+)', string_)]
@@ -104,8 +105,19 @@ class GMULExperiment(SolverExperiment):
 
 
 if __name__ == "__main__":
-    import sys
-    experiment = CoolExperiment()
-    dirname = experiment(sys.argv)
+    parser = argparse.ArgumentParser(description="CTL Experiment runnter", add_help=False)
+    parser.add_argument("experiment", choices=['cool', 'treetab', 'gmul', 'comparison'])
 
+    local, unknown = parser.parse_known_args()
+
+    if local.experiment == 'cool':
+        experiment = CoolExperiment()
+
+    elif local.experiment == 'treetab':
+        experiment = TreeTabExperiment()
+
+    elif local.experiment == 'gmul':
+        experiment = GMULExperiment()
+
+    dirname = experiment(unknown)
     print(dirname)