mev 0.1.0

Metal Et Vulkan abstraction
Documentation 
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::generic::{ArgumentGroupLayout, ArgumentKind, ArgumentsSealed};

use super::{shader::Bindings, ComputeCommandEncoder, RenderCommandEncoder};

pub trait Arguments: 'static {
    const LAYOUT: ArgumentGroupLayout<'static>;

    fn bind_render(&self, group: u32, encoder: &mut RenderCommandEncoder);
    fn bind_compute(&self, group: u32, encoder: &mut ComputeCommandEncoder);
}

impl<T> ArgumentsSealed for T where T: Arguments {}
impl<T> crate::generic::Arguments for T
where
    T: Arguments,
{
    const LAYOUT: ArgumentGroupLayout<'static> = T::LAYOUT;

    #[inline(always)]
    fn bind_render(&self, group: u32, encoder: &mut RenderCommandEncoder) {
        Arguments::bind_render(self, group, encoder)
    }

    #[inline(always)]
    fn bind_compute(&self, group: u32, encoder: &mut ComputeCommandEncoder) {
        Arguments::bind_compute(self, group, encoder)
    }
}

fn non_zero_group_no_bindings() -> ! {
    panic!(
        "Attempt to bind non-zero group to a pipeline stage with shader compiled from Metal Shading Language.
        This use-case is not supported right now."
    );
}

#[doc(hidden)]
pub trait ArgumentsField<T>: 'static {
    const KIND: ArgumentKind;
    const SIZE: usize;

    fn bind_vertex(&self, slot: u32, encoder: &metal::RenderCommandEncoderRef);
    fn bind_fragment(&self, slot: u32, encoder: &metal::RenderCommandEncoderRef);
    fn bind_compute(&self, slot: u32, encoder: &metal::ComputeCommandEncoderRef);

    #[inline]
    fn bind_vertex_argument(
        &self,
        group: u32,
        index: u32,
        bindings: Option<&Bindings>,
        encoder: &metal::RenderCommandEncoderRef,
    ) {
        match bindings {
            Some(bindings) => {
                let slot = bindings.groups[group as usize].bindings[index as usize];
                self.bind_vertex(slot.into(), encoder);
            }
            None if group == 0 => self.bind_vertex(index, encoder),
            None => non_zero_group_no_bindings(),
        }
    }

    #[inline]
    fn bind_fragment_argument(
        &self,
        group: u32,
        index: u32,
        bindings: Option<&Bindings>,
        encoder: &metal::RenderCommandEncoderRef,
    ) {
        match bindings {
            Some(bindings) => {
                let slot = bindings.groups[group as usize].bindings[index as usize];
                self.bind_fragment(slot.into(), encoder);
            }
            None if group == 0 => self.bind_fragment(index, encoder),
            None => non_zero_group_no_bindings(),
        }
    }

    #[inline]
    fn bind_compute_argument(
        &self,
        group: u32,
        index: u32,
        bindings: Option<&Bindings>,
        encoder: &metal::ComputeCommandEncoderRef,
    ) {
        match bindings {
            Some(bindings) => {
                let slot = bindings.groups[group as usize].bindings[index as usize];
                self.bind_compute(slot.into(), encoder);
            }
            None if group == 0 => self.bind_compute(index, encoder),
            None => non_zero_group_no_bindings(),
        }
    }
}

impl<T, F> crate::generic::ArgumentsField<T> for F
where
    T: ArgumentsSealed,
    F: ArgumentsField<T> + ArgumentsSealed,
{
    const KIND: ArgumentKind = F::KIND;
    const SIZE: usize = F::SIZE;
}