cidre 0.11.4

use crate::{arc, define_mtl, define_obj_type, mtl, ns, objc};

#[doc(alias = "MTLPatchType")]
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
#[repr(usize)]
pub enum PatchType {
    None = 0,
    Triangle = 1,
    Quad = 2,
}

define_obj_type!(
    /// An object that represents an attribute of a vertex function.
    #[doc(alias = "MTLVertexAttribute")]
    pub VertexAttr(ns::Id)
);

impl VertexAttr {
    #[objc::msg_send(name)]
    pub fn name(&self) -> arc::R<ns::String>;

    #[objc::msg_send(attributeIndex)]
    pub fn attr_index(&self) -> usize;

    #[objc::msg_send(attributeType)]
    pub fn attr_type(&self) -> mtl::DType;

    #[objc::msg_send(isActive)]
    pub fn is_active(&self) -> bool;

    #[objc::msg_send(isPatchData)]
    pub fn is_patch_data(&self) -> bool;

    #[objc::msg_send(isPatchControlPointData)]
    pub fn is_patch_ctrl_point_data(&self) -> bool;
}

define_obj_type!(
    /// An object that describes an attribute defined in the stage-in argument for a shader.
    #[doc(alias = "MTLAttribute")]
    pub Attr(ns::Id)
);

impl Attr {
    #[objc::msg_send(name)]
    pub fn name(&self) -> arc::R<ns::String>;

    #[objc::msg_send(attributeIndex)]
    pub fn attr_index(&self) -> usize;

    #[objc::msg_send(attributeType)]
    pub fn attr_type(&self) -> mtl::DType;

    #[objc::msg_send(isActive)]
    pub fn is_active(&self) -> bool;

    #[objc::msg_send(isPatchData)]
    pub fn is_patch_data(&self) -> bool;

    #[objc::msg_send(isPatchControlPointData)]
    pub fn is_patch_ctrl_point_data(&self) -> bool;
}

#[doc(alias = "MTLLanguageVersion")]
#[derive(Debug, PartialEq, Eq)]
#[repr(usize)]
pub enum LangVersion {
    #[doc(alias = "MTLLanguageVersion1_0")]
    _1_0 = (1 << 16),

    #[doc(alias = "MTLLanguageVersion1_1")]
    _1_1 = (1 << 16) + 1,
    #[doc(alias = "MTLLanguageVersion1_2")]
    _1_2 = (1 << 16) + 2,

    #[doc(alias = "MTLLanguageVersion2_0")]
    _2_0 = (2 << 16),

    #[doc(alias = "MTLLanguageVersion2_1")]
    _2_1 = (2 << 16) + 1,

    #[doc(alias = "MTLLanguageVersion2_2")]
    _2_2 = (2 << 16) + 2,

    #[doc(alias = "MTLLanguageVersion2_3")]
    _2_3 = (2 << 16) + 3,

    #[doc(alias = "MTLLanguageVersion2_4")]
    _2_4 = (2 << 16) + 4,

    #[doc(alias = "MTLLanguageVersion3_0")]
    _3_0 = (3 << 16),

    #[doc(alias = "MTLLanguageVersion3_1")]
    _3_1 = (3 << 16) + 1,

    #[doc(alias = "MTLLanguageVersion3_2")]
    _3_2 = (3 << 16) + 2,

    #[doc(alias = "MTLLanguageVersion4_0")]
    _4_0 = (4 << 16) + 0,
}

/// An enum to indicate if the compiler can perform optimizations for floating-point
/// arithmetic that may violate the IEEE 754 standard
#[doc(alias = "MTLMathMode")]
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
#[repr(isize)]
pub enum MathMode {
    /// Disables unsafe floating-point optimizations
    #[doc(alias = "MTLMathModeSafe")]
    Safe = 0,

    /// Allows aggressive, unsafe floating-point optimizations but preserves infs and nans
    #[doc(alias = "MTLMathModeRelaxed")]
    Relaxed = 1,

    /// Allows aggressive, unsafe floating-point optimizations
    #[doc(alias = "MTLMathModeFast")]
    Fast = 2,
}

/// An enum to indicate the default math functions for single precision floating-point
#[doc(alias = "MTLMathFloatingPointFunctions")]
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
#[repr(isize)]
pub enum MathFloatingPointFns {
    /// Sets the default math functions for single precision floating-point to the corresponding functions in `metal::fast` namespace
    Fast = 0,

    /// Sets the default math functions for single precision floating-point to the corresponding functions in 'metal::precise' namespace
    Precise = 1,
}

#[doc(alias = "MTLLibraryType")]
#[derive(Debug, Eq, PartialEq, Copy, Clone)]
#[repr(isize)]
pub enum Type {
    /// A library that can create pipeline state objects.
    Executable = 0,

    /// A library that you can dynamically link to from other libraries.
    Dynamic = 1,
}

#[doc(alias = "MTLLibraryOptimizationLevel")]
#[derive(Default, Debug, Eq, PartialEq, Copy, Clone)]
#[repr(isize)]
pub enum OptimizationLevel {
    /// An optimization option for the Metal compiler that prioritizes runtime performance.
    #[default]
    Default = 0,

    /// An optimization option for the Metal compiler that prioritizes minimizing
    /// the size of its output binaries, which may also reduce compile time.
    Size = 1,
}

#[doc(alias = "MTLCompileSymbolVisibility")]
#[derive(Debug, Default, Eq, PartialEq, Copy, Clone)]
#[repr(isize)]
pub enum CompileSymbolVisibility {
    #[default]
    Default = 0,
    Hidden = 1,
}

define_obj_type!(
    /// Compilation settings for a Metal shader library.
    #[doc(alias = "MTLCompileOptions")]
    pub CompileOpts(ns::Id),
    MTL_COMPILE_OPTIONS
);

/// ```
/// use cidre::mtl;
///
/// let mut options = mtl::CompileOpts::new();
///
/// assert_eq!(true, options.fast_math_enabled());
/// options.set_fast_math_enabled(false);
/// assert_eq!(false, options.fast_math_enabled());
///
/// assert_ne!(options.lang_version(), mtl::LangVersion::_2_0);
///
/// options.set_lang_version(mtl::LangVersion::_2_4);
/// assert_eq!(options.lang_version(), mtl::LangVersion::_2_4);
///
/// ```
impl CompileOpts {
    #[objc::msg_send(fastMathEnabled)]
    pub fn fast_math_enabled(&self) -> bool;

    #[objc::msg_send(setFastMathEnabled:)]
    pub fn set_fast_math_enabled(&mut self, val: bool);

    #[objc::msg_send(mathMode)]
    #[objc::available(
        macos = 15.0,
        maccatalyst = 18.0,
        ios = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn math_mode(&self) -> MathMode;

    #[objc::msg_send(setMathMode:)]
    #[objc::available(
        macos = 15.0,
        maccatalyst = 18.0,
        ios = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn set_math_mode(&mut self, val: MathMode);

    #[objc::msg_send(mathFloatingPointFunctions)]
    #[objc::available(
        macos = 15.0,
        maccatalyst = 18.0,
        ios = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn math_floating_point_fns(&self) -> MathFloatingPointFns;

    #[objc::msg_send(setMathFloatingPointFunctions:)]
    #[objc::available(
        macos = 15.0,
        maccatalyst = 18.0,
        ios = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn set_math_floating_point_fns(&mut self, val: MathFloatingPointFns);

    #[objc::msg_send(languageVersion)]
    pub fn lang_version(&self) -> LangVersion;

    #[objc::msg_send(setLanguageVersion:)]
    pub fn set_lang_version(&mut self, val: LangVersion);

    #[objc::msg_send(compileSymbolVisibility)]
    pub fn compile_symbol_visibility(&self) -> CompileSymbolVisibility;

    #[objc::msg_send(setCompileSymbolVisibility:)]
    pub fn set_compile_symbol_visibility(&mut self, val: CompileSymbolVisibility);

    #[objc::msg_send(allowReferencingUndefinedSymbols)]
    pub fn allow_referencing_undefined_symbols(&self) -> bool;

    #[objc::msg_send(setAllowReferencingUndefinedSymbols:)]
    pub fn set_allow_referencing_undefined_symbols(&mut self, val: bool);

    #[objc::msg_send(maxTotalThreadsPerThreadgroup)]
    pub fn max_total_threads_per_threadgroup(&self) -> usize;

    #[objc::msg_send(setMaxTotalThreadsPerThreadgroup:)]
    pub fn set_max_total_threads_per_threadgroup(&mut self, val: usize);

    #[objc::msg_send(requiredThreadsPerThreadgroup)]
    #[objc::available(
        macos = 26.0,
        ios = 26.0,
        maccatalyst = 26.0,
        tvos = 26.0,
        visionos = 26.0
    )]
    pub fn required_threads_per_threadgroup(&self) -> mtl::Size;

    #[objc::msg_send(setRequiredThreadsPerThreadgroup:)]
    #[objc::available(
        macos = 26.0,
        ios = 26.0,
        maccatalyst = 26.0,
        tvos = 26.0,
        visionos = 26.0
    )]
    pub fn set_required_threads_per_threadgroup(&self, val: mtl::Size);

    #[objc::msg_send(enableLogging)]
    #[objc::available(
        macos = 15.0,
        ios = 18.0,
        maccatalyst = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn enable_logging(&self) -> bool;

    #[objc::msg_send(setEnableLogging:)]
    #[objc::available(
        macos = 15.0,
        ios = 18.0,
        maccatalyst = 18.0,
        tvos = 18.0,
        visionos = 2.0
    )]
    pub fn set_enable_logging(&mut self, val: bool);
}

define_obj_type!(
    /// An object that represents a public shader function in a Metal library.
    #[doc(alias = "MTLFunction")]
    pub Fn(ns::Id)
);

impl Fn {
    define_mtl!(set_label);

    #[objc::msg_send(device)]
    pub fn device(&self) -> &mtl::Device;

    #[objc::msg_send(label)]
    pub fn label(&self) -> Option<&ns::String>;

    #[objc::msg_send(name)]
    pub fn name(&self) -> arc::R<ns::String>;

    #[objc::msg_send(newArgumentEncoderWithBufferIndex:)]
    pub fn new_argument_enc_with_buf_index(&self, index: ns::UInteger) -> arc::R<mtl::ArgEncoder>;

    #[objc::msg_send(functionType)]
    pub fn fn_type(&self) -> mtl::FnType;

    #[objc::msg_send(patchType)]
    pub fn patch_type(&self) -> mtl::PatchType;

    #[objc::msg_send(patchControlPointCount)]
    pub fn patch_ctrl_point(&self) -> isize;

    #[objc::msg_send(vertexAttributes)]
    pub fn vertext_attrs(&self) -> Option<arc::R<ns::Array<mtl::VertexAttr>>>;

    #[objc::msg_send(stageInputAttributes)]
    pub fn stage_input_attrs(&self) -> Option<arc::R<ns::Array<mtl::Attr>>>;

    #[objc::msg_send(options)]
    pub fn opts(&self) -> mtl::FnOpts;
}

define_obj_type!(
    #[doc(alias = "MTLLibrary")]
    pub Lib(ns::Id)
);

unsafe impl Send for Lib {}

impl Lib {
    define_mtl!(set_label);

    #[objc::msg_send(device)]
    pub fn device(&self) -> arc::R<mtl::Device>;

    #[objc::msg_send(label)]
    pub fn label(&self) -> Option<arc::R<ns::String>>;

    /// The array contains ns::String objects, with the name of each function in library.
    #[objc::msg_send(functionNames)]
    pub fn fn_names(&self) -> arc::R<ns::Array<ns::String>>;

    #[objc::msg_send(newFunctionWithName:)]
    pub fn new_fn(&self, name: &ns::String) -> Option<arc::R<Fn>>;

    /// # Safety
    ///
    /// Use new_fn_const_values
    #[objc::msg_send(newFunctionWithName:constantValues:error:)]
    pub unsafe fn new_fn_with_consts_err<'ear>(
        &self,
        name: &ns::String,
        constant_values: &mtl::FnConstValues,
        error: *mut Option<&'ear ns::Error>,
    ) -> Option<arc::R<Fn>>;

    #[inline]
    pub fn new_fn_with_consts<'ear>(
        &self,
        name: &ns::String,
        constant_values: &mtl::FnConstValues,
    ) -> Result<arc::R<Fn>, &'ear ns::Error> {
        ns::if_none(|err| unsafe { Self::new_fn_with_consts_err(self, name, constant_values, err) })
    }

    #[objc::msg_send(newFunctionWithName:descriptor:error:)]
    pub unsafe fn new_fn_with_desc_err<'ar>(
        &self,
        name: &ns::String,
        descriptor: &mtl::FnDesc,
        error: *mut Option<&'ar ns::Error>,
    ) -> Option<arc::R<Fn>>;

    pub fn new_fn_with_desc<'ar>(
        &self,
        name: &ns::String,
        descriptor: &mtl::FnDesc,
    ) -> Result<arc::R<Fn>, &'ar ns::Error> {
        ns::if_none(|err| unsafe { Self::new_fn_with_desc_err(self, name, descriptor, err) })
    }

    #[objc::msg_send(type)]
    pub fn type_(&self) -> mtl::LibType;

    /// The install_name provided when this mtl::Lib was created.
    ///
    /// Always nil if the type of the library is not mtl::LibType::Dynamic.
    /// [read more](https://developer.apple.com/documentation/metal/mtllibrary/3554039-installname?language=objc)
    #[objc::msg_send(installName)]
    pub fn install_name(&self) -> Option<arc::R<ns::String>>;
}

pub type ErrorDomain = ns::ErrorDomain;

impl ErrorDomain {
    pub fn lib() -> &'static ErrorDomain {
        unsafe { MTLLibraryErrorDomain }
    }
}

#[doc(alias = "MTLFunctionType")]
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
#[repr(usize)]
pub enum FnType {
    /// A vertex shader, usable for a mtl::RenderPipelineState.
    Vertex = 1,
    Fragment = 2,
    Kernel = 3,
    Visible = 5,
    Intersection = 6,
    Mesh = 7,
    Object = 8,
}

#[doc(alias = "MTLLibraryError")]
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
#[repr(usize)]
pub enum Error {
    Unsupported = 1,
    Internal = 2,
    CompileFailure = 3,
    CompileWarning = 4,
    FunctionNotFound = 5,
    FileNotFound = 6,
}

impl PartialEq<isize> for Error {
    fn eq(&self, other: &isize) -> bool {
        *self as isize == *other
    }
}

#[link(name = "Metal", kind = "framework")]
unsafe extern "C" {
    static MTLLibraryErrorDomain: &'static ErrorDomain;
}

#[link(name = "mtl", kind = "static")]
unsafe extern "C" {
    static MTL_COMPILE_OPTIONS: &'static objc::Class<CompileOpts>;
}

#[cfg(test)]
mod tests {
    use crate::{api, blocks, mtl, ns, objc::ar_pool};

    #[test]
    fn foo() {
        let device = mtl::Device::sys_default().unwrap();
        let src = ns::str!(c"kernel void function_a() {}");

        let mut ch = blocks::ResultCh::new2(move |lib, error| {
            println!("nice!!! {:?} {:?}", lib, error);
        });
        device.new_lib_with_src_ch(src, None, &mut ch);
    }

    #[test]
    fn function_names() {
        let device = mtl::Device::sys_default().unwrap();

        let src = ns::str!(c"kernel void function_a() {}; void function_b() {}");
        let lib = device.new_lib_with_src_blocking(src, None).unwrap();
        let names = lib.fn_names();
        assert_eq!(1, names.len());
        let n = names.get(0).unwrap();

        let expected_name = ns::str!(c"function_a");

        assert!(n.eq(expected_name));
    }

    #[test]
    fn error_basics() {
        ar_pool(|| {
            let device = mtl::Device::sys_default().unwrap();

            let src = ns::str!(c"vid function_a() {}");
            let err = device.new_lib_with_src_blocking(src, None).unwrap_err();

            assert_eq!(mtl::LibError::CompileFailure, err.code());
        })
    }

    #[test]
    fn new_function_with_name() {
        let device = mtl::Device::sys_default().unwrap();

        let src = ns::str!(c"kernel void function_a() {}");
        let lib = device.new_lib_with_src_blocking(src, None).unwrap();

        let fn_name = ns::str!(c"function_a");
        let func = lib.new_fn(fn_name).unwrap();
        let name = func.name();
        assert!(fn_name.is_equal(&name));
        assert_eq!(func.opts(), Default::default());
    }

    #[test]
    fn new_function_with_name_constant_values() {
        let device = mtl::Device::sys_default().unwrap();

        let src = ns::str!(c"kernel void function_a() {}");
        let lib = device.new_lib_with_src_blocking(src, None).unwrap();

        let fn_name = ns::str!(c"function_a");
        let const_values = mtl::FnConstValues::new();
        let func = lib.new_fn_with_consts(fn_name, &const_values).unwrap();
        let name = func.name();
        assert!(fn_name.is_equal(&name));
    }

    #[test]
    fn compile_opts() {
        let mut opts = mtl::CompileOpts::new();

        assert_eq!(true, opts.fast_math_enabled());
        opts.set_fast_math_enabled(false);
        assert_eq!(false, opts.fast_math_enabled());

        assert_ne!(opts.lang_version(), mtl::LangVersion::_2_0);

        opts.set_lang_version(mtl::LangVersion::_2_4);
        assert_eq!(opts.lang_version(), mtl::LangVersion::_2_4);
    }

    #[test]
    fn install_name() {
        let device = mtl::Device::sys_default().unwrap();
        let src = ns::str!(c"kernel void function_a() {}");
        let lib = device.new_lib_with_src_blocking(src, None).unwrap();

        if api::macos_available("26.0") {
            assert_eq!(lib.install_name().unwrap().as_ref(), "default.metallib")
        } else {
            assert!(lib.install_name().is_none());
        }
        assert_eq!(mtl::LibType::Executable, lib.type_());
    }
}