use core::ops::Range;
use objc2::{Message, extern_protocol, msg_send, rc::Retained, runtime::ProtocolObject};
use objc2_foundation::{NSObjectProtocol, NSRange, NSString};
use crate::{MTLDevice, MTLResourceID};
extern_protocol!(
pub unsafe trait MTLResourceViewPool: NSObjectProtocol {
#[unsafe(method(baseResourceID))]
#[unsafe(method_family = none)]
fn base_resource_id(&self) -> MTLResourceID;
#[unsafe(method(resourceViewCount))]
#[unsafe(method_family = none)]
fn resource_view_count(&self) -> usize;
#[unsafe(method(device))]
#[unsafe(method_family = none)]
fn device(&self) -> Retained<ProtocolObject<dyn MTLDevice>>;
}
);
#[allow(unused)]
pub trait MTLResourceViewPoolExt: MTLResourceViewPool + Message {
fn label(&self) -> Option<String>
where
Self: Sized,
{
let label: Option<Retained<NSString>> = unsafe { msg_send![self, label] };
label.map(|label| label.to_string())
}
fn copy_resource_views_from_pool(
&self,
source_pool: &ProtocolObject<dyn MTLResourceViewPool>,
source_range: Range<usize>,
destination_index: usize,
) -> MTLResourceID
where
Self: Sized,
{
let source_range = NSRange::from(source_range);
unsafe {
msg_send![
self,
copyResourceViewsFromPool: source_pool,
sourceRange: source_range,
destinationIndex: destination_index
]
}
}
}
impl<T: MTLResourceViewPool + Message> MTLResourceViewPoolExt for T {}