Merge global_ into global_options.

Author: michelp

suitesparse_graphblas/api/__init__.py

@@ -25,7 +25,6 @@
     container,
     context,
     descriptor,
-    global_,
     global_options,
     grb_type,
     indexbinaryop,

suitesparse_graphblas/api/global_.py

@@ -74,3 +74,74 @@ def global_option_get_char(field):
             f"GxB_Global_Option_get_CHAR failed with info={info}"
         )
     return ffi.string(val[0]).decode()
+
+
+def global_get_int32(field):
+    """Get a global property as an int32.
+
+    >>> isinstance(global_get_int32(lib.GxB_NTHREADS), int)
+    True
+
+    """
+    val = ffi.new("int32_t*")
+    info = lib.GrB_Global_get_INT32(lib.GrB_GLOBAL, val, field)
+    if info != lib.GrB_SUCCESS:
+        raise _error_code_lookup.get(info, RuntimeError)(
+            f"GrB_Global_get_INT32 failed with info={info}"
+        )
+    return val[0]
+
+
+def global_set_int32(field, value):
+    """Set a global property from an int32.
+
+    >>> nthreads = global_get_int32(lib.GxB_NTHREADS)
+    >>> global_set_int32(lib.GxB_NTHREADS, nthreads)
+
+    """
+    info = lib.GrB_Global_set_INT32(lib.GrB_GLOBAL, ffi.cast("int32_t", value), field)
+    if info != lib.GrB_SUCCESS:
+        raise _error_code_lookup.get(info, RuntimeError)(
+            f"GrB_Global_set_INT32 failed with info={info}"
+        )
+
+
+def global_get_size(field):
+    """Get a global property as a size_t.
+
+    >>> isinstance(global_get_size(lib.GrB_NAME), int)
+    True
+
+    """
+    val = ffi.new("size_t*")
+    info = lib.GrB_Global_get_SIZE(lib.GrB_GLOBAL, val, field)
+    if info != lib.GrB_SUCCESS:
+        raise _error_code_lookup.get(info, RuntimeError)(
+            f"GrB_Global_get_SIZE failed with info={info}"
+        )
+    return val[0]
+
+
+def global_get_string(field):
+    """Get a global property as a string.
+
+    >>> 'SuiteSparse:GraphBLAS' in global_get_string(lib.GrB_NAME)
+    True
+
+    """
+    val = ffi.new("char[256]")
+    info = lib.GrB_Global_get_String(lib.GrB_GLOBAL, val, field)
+    if info != lib.GrB_SUCCESS:
+        raise _error_code_lookup.get(info, RuntimeError)(
+            f"GrB_Global_get_String failed with info={info}"
+        )
+    return ffi.string(val).decode()
+
+
+def global_set_string(field, value):
+    """Set a global property from a string."""
+    info = lib.GrB_Global_set_String(lib.GrB_GLOBAL, value.encode(), field)
+    if info != lib.GrB_SUCCESS:
+        raise _error_code_lookup.get(info, RuntimeError)(
+            f"GrB_Global_set_String failed with info={info}"
+        )

suitesparse_graphblas/api/global_options.py

@@ -6,7 +6,6 @@ def test_run_doctests():
         container,
         context,
         descriptor,
-        global_,
         global_options,
         grb_type,
         indexbinaryop,
@@ -38,6 +37,5 @@ def test_run_doctests():
         descriptor,
         selectop,
         container,
-        global_,
     ):
         doctest.testmod(mod, optionflags=doctest.ELLIPSIS, raise_on_error=True)