1 files changed, 115 insertions, 0 deletions
diff --git a/test/utils/screendisplay.py b/test/utils/screendisplay.py
new file mode 100644
index 00000000..7c7bfde1
--- /dev/null
+++ b/test/utils/screendisplay.py
@@ -0,0 +1,115 @@
+import time
+import signal
+from cosmetics import getTerminalSize, colors
+
+
+global_state = {"termSize": getTerminalSize(),
+                "message": "",
+                "pstate": []}
+
+
+def print_conf(conf, value):
+    "Print a configuration line"
+    return colors["green"] + conf + ": " + colors["end"] + str(value)
+
+
+def clr_screen():
+    "Update the screen to display some information"
+
+    # Header
+    to_print = []
+    to_print.append(" " * (global_state["termSize"][0] / 2 - 10) + colors[
+                    "blue"] + "Miasm2 Regression tests" + colors["end"])
+    to_print.append("")
+    to_print.append("=" * global_state["termSize"][0])
+    to_print.append("")
+    to_print.append(print_conf("Current mode", "Multiprocessing"))
+    to_print.append(print_conf("Nb CPU detected", global_state["cpu_c"]))
+    to_print.append("")
+    to_print.append("=" * global_state["termSize"][0])
+    to_print.append("")
+    test_done = 0
+    test_failed = 0
+    message = global_state["message"] + "\n"
+    for v in global_state["pstate"]:
+        if v["status"] != "running":
+            test_done += 1
+            if v["status"] != 0:
+                test_failed += 1
+                cmd_line = " ".join(v["test"].command_line)
+                message += colors["red"] + "FAIL:" + colors["end"] + cmd_line
+                message += "\n" + v["message"] + "\n"
+
+    to_print.append(print_conf("Success rate", "%d/%d" %
+                    (test_done - test_failed, test_done)))
+    printed_time = time.strftime(
+        "%M:%S", time.gmtime(time.time() - global_state["init_time"]))
+    to_print.append(print_conf("Cumulated time", printed_time))
+    to_print.append("")
+    to_print.append("=" * global_state["termSize"][0])
+
+    cur = "\n".join(to_print)
+    cur += "\n"
+
+    # Message
+    cur += message
+    print cur
+    already_printed = cur.count("\n")
+
+    # Current state
+    current_job = []
+    for process in global_state["pstate"]:
+        if process["status"] == "running":
+            current_job.append(process)
+    print "\n" * (global_state["termSize"][1] - already_printed - 3 - len(current_job))
+
+    for job in current_job:
+        command_line = " ".join(job["test"].command_line)
+        base_dir = job["test"].base_dir.upper()
+        s = "[" + colors["lightcyan"] + command_line + colors["end"]
+        s_end = base_dir
+        cur_time = time.strftime(
+            "%M:%Ss", time.gmtime(time.time() - job["init_time"]))
+        l = len(command_line) + len(s_end) + 4 + len(str(cur_time)) + 2
+        s_end += "    " + colors["blue"] + cur_time + colors["end"] + "]"
+        print "%s%s%s" % (s, " " * (global_state["termSize"][0] - l), s_end)
+
+
+def on_signal(sig1, sig2):
+    "Update view every second"
+    clr_screen()
+    signal.alarm(1)
+
+
+def init(cpu_c):
+    """Initialize global state
+    @cpu_c: number of cpu (for conf displaying)
+    """
+    # Init global_state
+    global_state["cpu_c"] = cpu_c
+    global_state["init_time"] = time.time()
+
+    # Launch view updater
+    signal.signal(signal.SIGALRM, on_signal)
+    signal.alarm(1)
+
+
+def task_done(test, error):
+    "Report a test done"
+    for task in global_state["pstate"]:
+        if task["test"] == test:
+            if error is not None:
+                task["status"] = -1
+                task["message"] = error
+            else:
+                task["status"] = 0
+            break
+    clr_screen()
+
+
+def task_new(test):
+    "Report a new test"
+    global_state["pstate"].append({"status": "running",
+                                   "test": test,
+                                   "init_time": time.time()})
+    clr_screen()