mtl-rs 0.1.8

Rust bindings for Apple's Metal API
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use objc2::{Message, extern_protocol, msg_send, rc::Retained, runtime::ProtocolObject};
use objc2_foundation::{NSObjectProtocol, NSString};

use crate::{MTLFunction, MTLFunctionLogDebugLocation, MTLFunctionLogType};

extern_protocol!(
    /// Function log information (from `MTLFunctionLog`).
    ///
    /// Availability: macOS 11.0+, iOS 14.0+
    pub unsafe trait MTLFunctionLog: NSObjectProtocol {
        #[unsafe(method(r#type))]
        #[unsafe(method_family = none)]
        fn r#type(&self) -> MTLFunctionLogType;

        #[unsafe(method(function))]
        #[unsafe(method_family = none)]
        fn function(&self) -> Option<Retained<ProtocolObject<dyn MTLFunction>>>;

        #[unsafe(method(debugLocation))]
        #[unsafe(method_family = none)]
        fn debug_location(&self) -> Option<Retained<ProtocolObject<dyn MTLFunctionLogDebugLocation>>>;
    }
);

#[allow(unused)]
pub trait MTLFunctionLogExt: MTLFunctionLog + Message {
    fn encoder_label(&self) -> Option<String>;
}

impl MTLFunctionLogExt for ProtocolObject<dyn MTLFunctionLog> {
    fn encoder_label(&self) -> Option<String> {
        let s: Option<Retained<NSString>> = unsafe { msg_send![self, encoderLabel] };
        s.map(|v| v.to_string())
    }
}