c_load.py

import ctypes as ct
from numpy.ctypeslib import ndpointer
import platform
import os
import re

uname = platform.uname()[0]
match uname:
    case "Linux":
        library = "./libzerdax.so"
    case "Windows":
        # library = r".\libzerdax.dll"
        print("this program doesn't work on windows yet")
        exit(1)
    case _:
        print(f"unsuported operating system: {uname}")
        exit(1)
lib = ct.CDLL(library)
nthreads = os.cpu_count()

floaty = None
with open("c_filter.c", 'r') as file:
    for line in file:
        match = re.match(r"\s*#define\s+USE_DOUBLE\s+(\d+)\s*", line)
        if match:
            use_double_value = int(match.group(1))
            if use_double_value == 0:
                floaty = ct.c_float
            else:
                floaty = ct.c_double
            break
if floaty is None:
    print("Error finding #define USE_DOUBLE in c_filter.c")
    exit(1)


def lfilter():
    function = lib.filter

    function.restype = None
    function.argtypes = [ndpointer(floaty, flags="C_CONTIGUOUS"),
                         ndpointer(floaty, flags="C_CONTIGUOUS"),
                         ndpointer(floaty, flags="C_CONTIGUOUS"),
                         ct.c_size_t,
                         ct.c_int]
    return function


def segments_distance():
    function = lib.segments_distance

    function.restype = ct.c_int32
    function.argtypes = [ndpointer(ct.c_int32, flags="C_CONTIGUOUS"),
                         ndpointer(ct.c_int32, flags="C_CONTIGUOUS")]
    return function


def lines_bundle():
    function = lib.lines_bundle

    function.restype = ct.c_int32
    function.argtypes = [ndpointer(ct.c_int32, flags="C_CONTIGUOUS"),
                         ndpointer(ct.c_int32, flags="C_CONTIGUOUS"),
                         ct.c_int32, ct.c_int32]
    return function


segments_distance = segments_distance()
lfilter = lfilter()
lines_bundle = lines_bundle()