diffsol-c 0.4.0

A diffsol wrapper featuring runtime scalar/matrix/solver types and a C API
Documentation
use std::os::raw::c_char;
use std::ptr;

use crate::c_invalid_arg;
use crate::jit::JitBackendType;

#[cfg(feature = "diffsl-cranelift")]
const JIT_BACKEND_CRANELIFT: &[u8] = b"cranelift\0";
#[cfg(feature = "diffsl-llvm")]
const JIT_BACKEND_LLVM: &[u8] = b"llvm\0";

pub(crate) fn jit_backend_from_i32(value: i32) -> Option<JitBackendType> {
    match value {
        #[cfg(feature = "diffsl-cranelift")]
        0 => Some(JitBackendType::Cranelift),
        #[cfg(all(feature = "diffsl-cranelift", feature = "diffsl-llvm"))]
        1 => Some(JitBackendType::Llvm),
        #[cfg(all(not(feature = "diffsl-cranelift"), feature = "diffsl-llvm"))]
        0 => Some(JitBackendType::Llvm),
        _ => None,
    }
}

#[allow(dead_code)]
pub(crate) fn jit_backend_to_i32(value: JitBackendType) -> i32 {
    match value {
        #[cfg(feature = "diffsl-cranelift")]
        JitBackendType::Cranelift => 0,
        #[cfg(all(feature = "diffsl-cranelift", feature = "diffsl-llvm"))]
        JitBackendType::Llvm => 1,
        #[cfg(all(not(feature = "diffsl-cranelift"), feature = "diffsl-llvm"))]
        JitBackendType::Llvm => 0,
    }
}

/// Return the number of JIT backends compiled into this library.
///
/// # Safety
/// This function is safe to call from C. It does not dereference any
/// caller-provided pointers.
#[unsafe(no_mangle)]
pub unsafe extern "C" fn diffsol_jit_backend_type_count() -> usize {
    usize::from(cfg!(feature = "diffsl-cranelift")) + usize::from(cfg!(feature = "diffsl-llvm"))
}

/// Return whether a JIT backend enum value is valid.
///
/// # Safety
/// This function is safe to call from C. It does not dereference any
/// caller-provided pointers.
#[unsafe(no_mangle)]
pub unsafe extern "C" fn diffsol_jit_backend_type_is_valid(value: i32) -> i32 {
    if jit_backend_from_i32(value).is_some() {
        1
    } else {
        c_invalid_arg!("invalid jit_backend_type");
        0
    }
}

/// Return the name of a JIT backend enum value.
///
/// # Safety
/// The returned pointer is borrowed from static storage owned by this library
/// and must not be freed or mutated by the caller.
#[unsafe(no_mangle)]
pub unsafe extern "C" fn diffsol_jit_backend_type_name(value: i32) -> *const c_char {
    match jit_backend_from_i32(value) {
        #[cfg(feature = "diffsl-cranelift")]
        Some(JitBackendType::Cranelift) => JIT_BACKEND_CRANELIFT.as_ptr() as *const c_char,
        #[cfg(feature = "diffsl-llvm")]
        Some(JitBackendType::Llvm) => JIT_BACKEND_LLVM.as_ptr() as *const c_char,
        None => {
            c_invalid_arg!("invalid jit_backend_type");
            ptr::null()
        }
    }
}