ivy-vulkan 0.10.3

Low level vulkan abstractions for the Ivy game engine.
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
109
110
111
112

use crate::Result;
use std::slice;

use ash::vk::{DescriptorSet, ImageView, Sampler, ShaderStageFlags};

use super::DescriptorBuilder;

// Traits for types holding one or more descriptor sets for use in rendering.
pub trait IntoSet {
    /// Get the descriptor set for the current frame
    fn set(&self, current_frame: usize) -> DescriptorSet {
        *self
            .sets()
            .get(current_frame)
            .unwrap_or_else(|| &self.sets()[0])
    }
    // Retrieve descriptor sets for all frames. May be less than frames_in_flight if the same set is
    // to be used
    fn sets(&self) -> &[DescriptorSet];
}

impl IntoSet for DescriptorSet {
    fn set(&self, _: usize) -> DescriptorSet {
        *self
    }

    fn sets(&self) -> &[DescriptorSet] {
        slice::from_ref(self)
    }
}

impl IntoSet for Vec<DescriptorSet> {
    fn set(&self, current_frame: usize) -> DescriptorSet {
        self[current_frame]
    }

    fn sets(&self) -> &[DescriptorSet] {
        self
    }
}

pub trait DescriptorBindable {
    fn bind_resource<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
    ) -> Result<&'a mut DescriptorBuilder>;
}

impl DescriptorBindable for ImageView {
    fn bind_resource<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
    ) -> Result<&'a mut DescriptorBuilder> {
        Ok(builder.bind_image(binding, stage, *self))
    }
}

impl DescriptorBindable for Sampler {
    fn bind_resource<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
    ) -> Result<&'a mut DescriptorBuilder> {
        Ok(builder.bind_sampler(binding, stage, *self))
    }
}

/// Helper trait for binding a list of resources for multiple frames in flight
pub trait MultiDescriptorBindable {
    fn bind_resource_for<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
        current_frame: usize,
    ) -> Result<&'a mut DescriptorBuilder>;
}

impl<T> MultiDescriptorBindable for &[T]
where
    T: DescriptorBindable,
{
    fn bind_resource_for<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
        current_frame: usize,
    ) -> Result<&'a mut DescriptorBuilder> {
        self[current_frame].bind_resource(binding, stage, builder)
    }
}

impl<T> MultiDescriptorBindable for T
where
    T: DescriptorBindable,
{
    fn bind_resource_for<'a>(
        &self,
        binding: u32,
        stage: ShaderStageFlags,
        builder: &'a mut DescriptorBuilder,
        _: usize,
    ) -> Result<&'a mut DescriptorBuilder> {
        self.bind_resource(binding, stage, builder)
    }
}