From 9573e934db25453b1dc2577fb6fed1d3fe7031b2 Mon Sep 17 00:00:00 2001
From: Mohammad Saknini <mohammad.saknini@smail.inf.h-brs.de>
Date: Sat, 18 Nov 2023 22:47:10 +0100
Subject: [PATCH] Added windows supported version for the function "timeout"
 that uses threads instead of Signal

---
 popper/util.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/popper/util.py b/popper/util.py
index 65088db6..d22ed3da 100644
--- a/popper/util.py
+++ b/popper/util.py
@@ -11,6 +11,7 @@
 from contextlib import contextmanager
 from .core import Literal
 from math import comb
+import threading
 
 clingo.script.enable_python()
 
@@ -73,6 +74,9 @@ class TimeoutError(Exception):
 
     def handler(signum, frame):
         raise TimeoutError()
+    
+    if not (hasattr(signal, 'SIGALRM') and hasattr(signal, 'alarm')):
+        return _windows_timeout(settings, func, args, kwargs, timeout_duration)
 
     # set the timeout handler
     signal.signal(signal.SIGALRM, handler)
@@ -92,6 +96,36 @@ def handler(signum, frame):
 
     return result
 
+def _windows_timeout(settings, func, args=(), kwargs={}, timeout_duration=1):
+    """
+    A replacement for the `timeout` function that works on Windows, since the `signal` module is not fully supported on Windows.
+    """
+    result = None
+
+    def target():
+        nonlocal result
+        try:
+            result = func(*args, **kwargs)
+        except Exception as e:
+            result = e
+
+    thread = threading.Thread(target=target)
+    thread.start()
+    thread.join(timeout_duration)
+
+    if thread.is_alive():
+        # If the thread is still alive after the timeout_duration, it means it timed out
+        thread.join()  # Ensure the thread terminates properly
+        settings.logger.warn(f'TIMEOUT OF {int(settings.timeout)} SECONDS EXCEEDED')
+        # raise TimeoutError(f"Execution of {func.__name__} timed out after {timeout_duration} seconds")
+        return result
+
+    if isinstance(result, Exception):
+        # If the function raised an exception, re-raise it here
+        raise result
+
+    return result
+
 def load_kbpath(kbpath):
     def fix_path(filename):
         full_filename = os.path.join(kbpath, filename)
@@ -587,7 +621,7 @@ def load_types(settings):
 def bias_order(settings, max_size):
 
     if not (settings.no_bias or settings.order_space):
-        return [(size_literals, settings.max_vars, settings.max_rules, None) for size_literals in range(1, max_size+1)]
+        return [(size_literals, settings.max_vars, settings.max_rules, None) for size_literals in range(1, max_size)]
 
     # if settings.search_order is None:
     ret = []