librashader/librashader-capi/src/presets.rs

use std::ffi::{c_char, CStr, CString, OsStr};
use std::mem::{ManuallyDrop, MaybeUninit};
use std::panic::catch_unwind;
use std::path::Path;
use librashader::presets::ShaderPreset;
use crate::ffi::ffi_body;
use crate::ctypes::{libra_error_t, libra_shader_preset_t};
use crate::error::{assert_non_null, assert_some, LibrashaderError};
use safer_ffi::prelude::*;
use safer_ffi::ffi_export;
use safer_ffi::char_p::char_p_ref as CStrRef;

// extern_fn! {
//     /// SAFETY:
//     /// - filename is aligned and valid for reads.
//     fn load_preset(filename: *const c_char, out: *mut MaybeUninit<shader_preset_t>) {
//         assert_non_null!(filename, "filename");
//         assert_non_null!(out, "out");
//
//         let filename = unsafe {
//             CStr::from_ptr(filename)
//         };
//
//         let filename = filename.to_str()?;
//
//         println!("loading {filename}");
//         let preset = ShaderPreset::try_parse(filename)?;
//
//         unsafe {
//             out.write(MaybeUninit::new(ManuallyDrop::new(Box::new(preset))))
//         }
//     }
// }

/// Load a preset.
pub type PFN_lbr_load_preset = unsafe extern "C" fn (*const c_char, *mut MaybeUninit<libra_shader_preset_t>) -> libra_error_t;
#[no_mangle]
pub unsafe extern "C" fn libra_load_preset(filename: *const c_char, out: *mut MaybeUninit<libra_shader_preset_t>) -> libra_error_t {
    ffi_body!({
        assert_non_null!(filename);
        assert_non_null!(out);

        let filename = unsafe {
            CStr::from_ptr(filename)
        };

        let filename = filename.to_str()?;

        println!("loading {filename}");
        let preset = ShaderPreset::try_parse(filename)?;

        unsafe {
            out.write(MaybeUninit::new(ManuallyDrop::new(Some(Box::new(preset)))))
        }
    })
}

pub type PFN_lbr_preset_free = unsafe extern "C" fn (*mut libra_shader_preset_t) -> libra_error_t;

/// Free the preset.
#[no_mangle]
pub unsafe extern "C" fn libra_preset_free(preset: *mut libra_shader_preset_t) -> libra_error_t {
    ffi_body!({
        assert_non_null!(preset);
        unsafe {
            let preset_ptr = &mut *preset;
            ManuallyDrop::drop(preset_ptr);
            preset.write(ManuallyDrop::new(None));
        }
    })
}

pub type PFN_lbr_preset_set_param = unsafe extern "C" fn (*mut libra_shader_preset_t, *const c_char, f32) -> libra_error_t;
/// Set the value of the parameter in the preset.
#[no_mangle]
pub unsafe extern "C" fn libra_preset_set_param(preset: *mut libra_shader_preset_t,
                                                name: *const c_char, value: f32) -> libra_error_t {
    ffi_body!(|name|; mut |preset| {
        let name = unsafe {
            CStr::from_ptr(name)
        };

        let name = name.to_str()?;
        assert_some!(preset);
        let preset = preset.as_mut().unwrap();

        if let Some(param) = preset.parameters.iter_mut().find(|c| c.name == name) {
            param.value = value
        }
    })
}

pub type PFN_lbr_preset_get_param = unsafe extern "C" fn (*mut libra_shader_preset_t, *const c_char, *mut MaybeUninit<f32>) -> libra_error_t;

/// Get the value of the parameter as set in the preset.
#[no_mangle]
pub unsafe extern "C" fn libra_preset_get_param(preset: *mut libra_shader_preset_t,
                                                name: *const c_char, value: *mut MaybeUninit<f32>) -> libra_error_t {
    ffi_body!(|name, preset | {
        let name = unsafe {
            CStr::from_ptr(name)
        };

        let name = name.to_str()?;
        assert_some!(preset);
        let preset = preset.as_ref().unwrap();

        if let Some(param) = preset.parameters.iter().find(|c| c.name == name) {
            unsafe {
                value.write(MaybeUninit::new(param.value))
            }
        }
    })
}

pub type PFN_lbr_preset_print = unsafe extern "C" fn (*mut libra_shader_preset_t) -> libra_error_t;

/// Pretty print the shader preset.
#[no_mangle]
pub unsafe extern "C" fn libra_preset_print(preset: *mut libra_shader_preset_t) -> libra_error_t {
    ffi_body!(|preset| {
        assert_some!(preset);
        println!("{:#?}", preset.as_ref().unwrap());
    })
}


pub type PFN_lbr_preset_get_runtime_param_names = unsafe extern "C" fn (*mut libra_shader_preset_t, *mut MaybeUninit<f32>) -> libra_error_t;

/// Get a list of runtime parameter names.
///
/// The returned value can not currently be freed.
#[no_mangle]
pub unsafe extern "C" fn libra_preset_get_runtime_param_names(preset: *mut libra_shader_preset_t, mut value: MaybeUninit<*mut *const c_char>) -> libra_error_t {
    ffi_body!(|preset | {
        assert_some!(preset);
        let preset = preset.as_ref().unwrap();

        let iter = librashader::presets::get_parameter_meta(preset)?;
        let mut c_strings = Vec::new();
        for param in iter {
            let c_string = CString::new(param.id).map_err(|err| LibrashaderError::UnknownError(Box::new(err)))?;
            c_strings.push(c_string.into_raw().cast_const());
        }

        let (parts, _len, _cap) = c_strings.into_raw_parts();
        value.write(parts);
    })
}