mtl-rs 0.1.11

use core::ptr::NonNull;

use objc2::{extern_protocol, runtime::ProtocolObject};

use super::{MTLCullMode, MTLDepthClipMode, MTLScissorRect, MTLTriangleFillMode, MTLViewport, MTLVisibilityResultMode};
use crate::{
    MTLBuffer, MTLCommandEncoder, MTLPrimitiveType, MTLTexture,
    depth_stencil::MTLDepthStencilState,
    render_pipeline::{MTLLogicalToPhysicalColorAttachmentMap, MTLRenderPipelineState},
};

extern_protocol!(
    /// Render command encoder interface.
    pub unsafe trait MTLRenderCommandEncoder: MTLCommandEncoder {
        #[unsafe(method(setRenderPipelineState:))]
        #[unsafe(method_family = none)]
        fn set_render_pipeline_state(
            &self,
            pipeline_state: &ProtocolObject<dyn MTLRenderPipelineState>,
        );

        /// Sets the depth and stencil state for the render command encoder.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(setDepthStencilState:))]
        #[unsafe(method_family = none)]
        fn set_depth_stencil_state(
            &self,
            depth_stencil_state: &ProtocolObject<dyn MTLDepthStencilState>,
        );

        #[unsafe(method(setViewport:))]
        #[unsafe(method_family = none)]
        fn set_viewport(
            &self,
            viewport: MTLViewport,
        );

        #[unsafe(method(setViewports:count:))]
        #[unsafe(method_family = none)]
        fn set_viewports(
            &self,
            viewports: NonNull<MTLViewport>,
            count: usize,
        );

        #[unsafe(method(setFrontFacingWinding:))]
        #[unsafe(method_family = none)]
        fn set_front_facing_winding(
            &self,
            winding: super::types::MTLWinding,
        );

        #[unsafe(method(setCullMode:))]
        #[unsafe(method_family = none)]
        fn set_cull_mode(
            &self,
            cull_mode: MTLCullMode,
        );

        #[unsafe(method(setDepthClipMode:))]
        #[unsafe(method_family = none)]
        fn set_depth_clip_mode(
            &self,
            mode: MTLDepthClipMode,
        );

        #[unsafe(method(setTriangleFillMode:))]
        #[unsafe(method_family = none)]
        fn set_triangle_fill_mode(
            &self,
            mode: MTLTriangleFillMode,
        );

        #[unsafe(method(setScissorRect:))]
        #[unsafe(method_family = none)]
        fn set_scissor_rect(
            &self,
            rect: MTLScissorRect,
        );

        #[unsafe(method(setScissorRects:count:))]
        #[unsafe(method_family = none)]
        fn set_scissor_rects(
            &self,
            rects: NonNull<MTLScissorRect>,
            count: usize,
        );

        #[unsafe(method(setBlendColorRed:green:blue:alpha:))]
        #[unsafe(method_family = none)]
        fn set_blend_color(
            &self,
            red: f32,
            green: f32,
            blue: f32,
            alpha: f32,
        );

        #[unsafe(method(setStencilReferenceValue:))]
        #[unsafe(method_family = none)]
        fn set_stencil_reference_value(
            &self,
            reference_value: u32,
        );

        #[unsafe(method(setStencilFrontReferenceValue:backReferenceValue:))]
        #[unsafe(method_family = none)]
        fn set_stencil_front_back_reference_value(
            &self,
            front: u32,
            back: u32,
        );

        #[unsafe(method(setVisibilityResultMode:offset:))]
        #[unsafe(method_family = none)]
        fn set_visibility_result_mode(
            &self,
            mode: MTLVisibilityResultMode,
            offset: usize,
        );

        #[unsafe(method(setColorAttachmentMap:))]
        #[unsafe(method_family = none)]
        fn set_color_attachment_map(
            &self,
            mapping: Option<&MTLLogicalToPhysicalColorAttachmentMap>,
        );

        /// Sets a buffer for the vertex shader function.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(setVertexBuffer:offset:atIndex:))]
        #[unsafe(method_family = none)]
        fn set_vertex_buffer(
            &self,
            buffer: Option<&ProtocolObject<dyn MTLBuffer>>,
            offset: usize,
            index: usize,
        );

        /// Sets inline data for the vertex shader function.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(setVertexBytes:length:atIndex:))]
        #[unsafe(method_family = none)]
        fn set_vertex_bytes(
            &self,
            bytes: NonNull<core::ffi::c_void>,
            length: usize,
            index: usize,
        );

        /// Sets a buffer for the fragment shader function.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(setFragmentBuffer:offset:atIndex:))]
        #[unsafe(method_family = none)]
        fn set_fragment_buffer(
            &self,
            buffer: Option<&ProtocolObject<dyn MTLBuffer>>,
            offset: usize,
            index: usize,
        );

        /// Sets a texture for the fragment shader function.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(setFragmentTexture:atIndex:))]
        #[unsafe(method_family = none)]
        fn set_fragment_texture(
            &self,
            texture: Option<&ProtocolObject<dyn MTLTexture>>,
            index: usize,
        );

        /// Encodes a draw command.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(drawPrimitives:vertexStart:vertexCount:))]
        #[unsafe(method_family = none)]
        fn draw_primitives(
            &self,
            primitive_type: MTLPrimitiveType,
            vertex_start: usize,
            vertex_count: usize,
        );

        /// Encodes an instanced draw command.
        ///
        /// Availability: macOS 10.11+, iOS 8.0+
        #[unsafe(method(drawPrimitives:vertexStart:vertexCount:instanceCount:baseInstance:))]
        #[unsafe(method_family = none)]
        fn draw_primitives_instanced(
            &self,
            primitive_type: MTLPrimitiveType,
            vertex_start: usize,
            vertex_count: usize,
            instance_count: usize,
            base_instance: usize,
        );
    }
);