From a94ca5b2cb1a6ec23247e81083ad3ed03ade10b6 Mon Sep 17 00:00:00 2001
From: Christoph Egger <christoph@christoph-egger.org>
Date: Sun, 24 Apr 2016 15:20:48 +0200
Subject: [PATCH] Allow selecting Experiment via comandline parameter

---
 cool.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

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)
-- 
2.39.5