Add AccelerationStructurePassDescriptor

Also renames "accelerator_structure" module for consistency

src/acceleration_structure_pass.rs

@@ -0,0 +1,108 @@
+use super::{CounterSampleBufferRef, NSUInteger};
+
+/// See <https://developer.apple.com/documentation/metal/mtlaccelerationstructurepassdescriptor>
+pub enum MTLAccelerationStructurePassDescriptor {}
+
+foreign_obj_type! {
+    type CType = MTLAccelerationStructurePassDescriptor;
+    pub struct AccelerationStructurePassDescriptor;
+}
+
+impl AccelerationStructurePassDescriptor {
+    /// Creates a default compute pass descriptor with no attachments.
+    pub fn new<'a>() -> &'a AccelerationStructurePassDescriptorRef {
+        unsafe {
+            msg_send![
+                class!(MTLAccelerationStructurePassDescriptor),
+                accelerationStructurePassDescriptor
+            ]
+        }
+    }
+}
+
+impl AccelerationStructurePassDescriptorRef {
+    pub fn sample_buffer_attachments(
+        &self,
+    ) -> &AccelerationStructurePassSampleBufferAttachmentDescriptorArrayRef {
+        unsafe { msg_send![self, sampleBufferAttachments] }
+    }
+}
+
+/// See <https://developer.apple.com/documentation/metal/mtlaccelerationstructurepasssamplebufferattachmentdescriptorarray>
+pub enum MTLAccelerationStructurePassSampleBufferAttachmentDescriptorArray {}
+
+foreign_obj_type! {
+    type CType = MTLAccelerationStructurePassSampleBufferAttachmentDescriptorArray;
+    pub struct AccelerationStructurePassSampleBufferAttachmentDescriptorArray;
+}
+
+impl AccelerationStructurePassSampleBufferAttachmentDescriptorArrayRef {
+    pub fn object_at(
+        &self,
+        index: NSUInteger,
+    ) -> Option<&AccelerationStructurePassSampleBufferAttachmentDescriptorRef> {
+        unsafe { msg_send![self, objectAtIndexedSubscript: index] }
+    }
+
+    pub fn set_object_at(
+        &self,
+        index: NSUInteger,
+        attachment: Option<&AccelerationStructurePassSampleBufferAttachmentDescriptorRef>,
+    ) {
+        unsafe {
+            msg_send![self, setObject:attachment
+                     atIndexedSubscript:index]
+        }
+    }
+}
+
+/// See <https://developer.apple.com/documentation/metal/mtlaccelerationstructurepasssamplebufferattachmentdescriptor>
+pub enum MTLAccelerationStructurePassSampleBufferAttachmentDescriptor {}
+
+foreign_obj_type! {
+    type CType = MTLAccelerationStructurePassSampleBufferAttachmentDescriptor;
+    pub struct AccelerationStructurePassSampleBufferAttachmentDescriptor;
+}
+
+impl AccelerationStructurePassSampleBufferAttachmentDescriptor {
+    pub fn new() -> Self {
+        let class = class!(MTLAccelerationStructurePassSampleBufferAttachmentDescriptor);
+        unsafe { msg_send![class, new] }
+    }
+}
+
+impl AccelerationStructurePassSampleBufferAttachmentDescriptorRef {
+    pub fn sample_buffer(&self) -> &CounterSampleBufferRef {
+        unsafe { msg_send![self, sampleBuffer] }
+    }
+
+    pub fn set_sample_buffer(&self, sample_buffer: &CounterSampleBufferRef) {
+        unsafe { msg_send![self, setSampleBuffer: sample_buffer] }
+    }
+
+    pub fn start_of_encoder_sample_index(&self) -> NSUInteger {
+        unsafe { msg_send![self, startOfEncoderSampleIndex] }
+    }
+
+    pub fn set_start_of_encoder_sample_index(&self, start_of_encoder_sample_index: NSUInteger) {
+        unsafe {
+            msg_send![
+                self,
+                setStartOfEncoderSampleIndex: start_of_encoder_sample_index
+            ]
+        }
+    }
+
+    pub fn end_of_encoder_sample_index(&self) -> NSUInteger {
+        unsafe { msg_send![self, endOfEncoderSampleIndex] }
+    }
+
+    pub fn set_end_of_encoder_sample_index(&self, end_of_encoder_sample_index: NSUInteger) {
+        unsafe {
+            msg_send![
+                self,
+                setEndOfEncoderSampleIndex: end_of_encoder_sample_index
+            ]
+        }
+    }
+}

src/commandbuffer.rs

@@ -154,6 +154,13 @@ impl CommandBufferRef {
         unsafe { msg_send![self, accelerationStructureCommandEncoder] }
     }
 
+    pub fn acceleration_structure_command_encoder_with_descriptor(
+        &self,
+        descriptor: &AccelerationStructurePassDescriptorRef,
+    ) -> &AccelerationStructureCommandEncoderRef {
+        unsafe { msg_send![self, accelerationStructureCommandEncoderWithDescriptor: descriptor] }
+    }
+
     pub fn encode_signal_event(&self, event: &EventRef, new_value: u64) {
         unsafe {
             msg_send![self,

src/device.rs

@@ -2109,10 +2109,7 @@ impl DeviceRef {
         unsafe { msg_send![self, accelerationStructureSizesWithDescriptor: desc] }
     }
 
-    pub fn new_acceleration_structure_with_size(
-        &self,
-        size: NSUInteger,
-    ) -> accelerator_structure::AccelerationStructure {
+    pub fn new_acceleration_structure_with_size(&self, size: NSUInteger) -> AccelerationStructure {
         unsafe { msg_send![self, newAccelerationStructureWithSize: size] }
     }
 

src/encoder.rs

@@ -418,7 +418,7 @@ impl RenderCommandEncoderRef {
     pub fn set_vertex_acceleration_structure(
         &self,
         index: NSUInteger,
-        accel: Option<&accelerator_structure::AccelerationStructureRef>,
+        accel: Option<&AccelerationStructureRef>,
     ) {
         unsafe {
             msg_send![
@@ -868,7 +868,7 @@ impl RenderCommandEncoderRef {
     pub fn set_fragment_acceleration_structure(
         &self,
         index: NSUInteger,
-        accel: Option<&accelerator_structure::AccelerationStructureRef>,
+        accel: Option<&AccelerationStructureRef>,
     ) {
         unsafe {
             msg_send![
@@ -1815,7 +1815,7 @@ impl ComputeCommandEncoderRef {
     pub fn set_acceleration_structure(
         &self,
         index: NSUInteger,
-        accel: Option<&accelerator_structure::AccelerationStructureRef>,
+        accel: Option<&AccelerationStructureRef>,
     ) {
         unsafe {
             msg_send![

src/lib.rs

@@ -536,7 +536,8 @@ impl MetalLayerRef {
     }
 }
 
-mod accelerator_structure;
+mod acceleration_structure;
+mod acceleration_structure_pass;
 mod argument;
 mod blitpass;
 mod buffer;
@@ -567,7 +568,8 @@ mod vertexdescriptor;
 
 #[rustfmt::skip]
 pub use {
-    accelerator_structure::*,
+    acceleration_structure::*,
+    acceleration_structure_pass::*,
     argument::*,
     blitpass::*,
     buffer::*,