feat: added fatal error handler and interruption; RNG has to be disabled temporarily

Author: ntamas

poetry.lock

@@ -10,6 +10,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.10"
 numpy = "^1.23.3"
+pycapi = "^0.82.1"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.2.1"

pyproject.toml

@@ -6,7 +6,7 @@
 
 # fmt: off
 
-from ctypes import cdll, c_char_p, c_double, c_int, c_void_p, POINTER
+from ctypes import cdll, c_bool, c_char_p, c_double, c_int, c_void_p, POINTER
 from ctypes.util import find_library
 
 from .errors import handle_igraph_error_t
@@ -36,9 +36,11 @@ from .types import (
     igraph_arpack_options_t,
     igraph_bliss_info_t,
     igraph_error_handler_t,
+    igraph_fatal_handler_t,
     igraph_hrg_t,
     igraph_layout_drl_options_t,
     igraph_maxflow_stats_t,
+    igraph_interruption_handler_t,
     igraph_isocompat_t,
     igraph_plfit_result_t,
     igraph_rng_t,
@@ -382,11 +384,19 @@ igraph_destroy = _lib.igraph_destroy
 igraph_destroy.restype = None
 igraph_destroy.argtypes = [POINTER(igraph_t)]
 
-# Error handling
+# Error handling and interruptions
 
 igraph_set_error_handler = _lib.igraph_set_error_handler
 igraph_set_error_handler.restype = igraph_error_handler_t
 igraph_set_error_handler.argtypes = [igraph_error_handler_t]
 
+igraph_set_fatal_handler = _lib.igraph_set_fatal_handler
+igraph_set_fatal_handler.restype = igraph_fatal_handler_t
+igraph_set_fatal_handler.argtypes = [igraph_fatal_handler_t]
+
+igraph_set_interruption_handler = _lib.igraph_set_interruption_handler
+igraph_set_interruption_handler.restype = igraph_interruption_handler_t
+igraph_set_interruption_handler.argtypes = [igraph_interruption_handler_t]
+
 # The rest of this file is generated by Stimulus
 

src/codegen/internal_lib.py.in

@@ -11,5 +11,7 @@ def handle_igraph_error_t(code: igraph_error_t) -> None:
         error_state = _get_last_error_state()
         if error_state:
             error_state.raise_error()
+        elif code == 13:  # IGRAPH_INTERRUPTED; does not call error handler
+            raise KeyboardInterrupt
         else:
             raise RuntimeError(f"igraph returned error code {code}")

src/igraph_ctypes/_internal/errors.py

@@ -36,9 +36,11 @@
     igraph_arpack_options_t,
     igraph_bliss_info_t,
     igraph_error_handler_t,
+    igraph_fatal_handler_t,
     igraph_hrg_t,
     igraph_layout_drl_options_t,
     igraph_maxflow_stats_t,
+    igraph_interruption_handler_t,
     igraph_isocompat_t,
     igraph_plfit_result_t,
     igraph_rng_t,
@@ -382,12 +384,20 @@ def _load_igraph_c_library():
 igraph_destroy.restype = None
 igraph_destroy.argtypes = [POINTER(igraph_t)]
 
-# Error handling
+# Error handling and interruptions
 
 igraph_set_error_handler = _lib.igraph_set_error_handler
 igraph_set_error_handler.restype = igraph_error_handler_t
 igraph_set_error_handler.argtypes = [igraph_error_handler_t]
 
+igraph_set_fatal_handler = _lib.igraph_set_fatal_handler
+igraph_set_fatal_handler.restype = igraph_fatal_handler_t
+igraph_set_fatal_handler.argtypes = [igraph_fatal_handler_t]
+
+igraph_set_interruption_handler = _lib.igraph_set_interruption_handler
+igraph_set_interruption_handler.restype = igraph_interruption_handler_t
+igraph_set_interruption_handler.argtypes = [igraph_interruption_handler_t]
+
 # The rest of this file is generated by Stimulus
 
 # Set up aliases for all enum types