From 9ab8ec3d8afeb06aaaefbd90589a6ba9904330dd Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 20 Apr 2021 01:21:30 +0200
Subject: [PATCH] Add execution script

---
 src/execution.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++
 src/main.py      |  0
 2 files changed, 52 insertions(+)
 create mode 100644 src/execution.py
 mode change 100644 => 100755 src/main.py

diff --git a/src/execution.py b/src/execution.py
new file mode 100644
index 0000000..43c02dd
--- /dev/null
+++ b/src/execution.py
@@ -0,0 +1,52 @@
+from glob import glob
+from subprocess import run
+from sys import executable
+from numpy import mean, std
+from pandas import DataFrame
+
+
+def file_list(path):
+    file_list = []
+    for fname in glob(path):
+        file_list.append(fname)
+    return file_list
+
+
+def create_dataframes(datasets):
+    greedy = DataFrame(columns=["dataset", "distancia"])
+    local = DataFrame(columns=["dataset", "distancia"])
+    greedy["dataset"] = datasets
+    local["dataset"] = datasets
+    return greedy, local
+
+
+def process_output(results):
+    for line in results:
+        if line.startswith(bytes("Total distance:", encoding="utf-8")):
+            line_elements = line.split(sep=bytes(":", encoding="utf-8"))
+            distance = float(line_elements[1])
+    return distance
+
+
+def script_execution(filenames, greedy, local, iterations=3):
+    script = "src/main.py"
+    for dataset in filenames:
+        greedy_list = []
+        local_list = []
+        for _ in range(iterations):
+            greedy_cmd = run(
+                [executable, script, dataset, "greedy"], capture_output=True
+            ).stdout.splitlines()
+            greedy_list.append(greedy_cmd)
+            local_cmd = run(
+                [executable, script, dataset, "local"], capture_output=True
+            ).stdout.splitlines()
+            local_list.append(local_cmd)
+        greedy.loc[greedy["dataset"] == dataset, ["distancia"]] = mean(greedy_list)
+        local.loc[local["dataset"] == dataset, ["distancia"]] = mean(local_list)
+
+
+if __name__ == "__main__":
+    datasets = file_list(path="data/*.txt")
+    greedy, local = create_dataframes(datasets)
+    script_execution(filenames=datasets, greedy=greedy, local=local)
diff --git a/src/main.py b/src/main.py
old mode 100644
new mode 100755