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

// Copyright (c) 2016 The vulkano developers
// Licensed under the Apache License, Version 2.0
// <LICENSE-APACHE or
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT
// license <LICENSE-MIT or https://opensource.org/licenses/MIT>,
// at your option. All files in the project carrying such
// notice may not be copied, modified, or distributed except
// according to those terms.

use crate::descriptor::descriptor::DescriptorBufferDesc;
use crate::descriptor::descriptor::DescriptorDesc;
use crate::descriptor::descriptor::DescriptorDescTy;
use crate::descriptor::pipeline_layout::PipelineLayoutDesc;
use crate::descriptor::pipeline_layout::PipelineLayoutDescPcRange;
use fnv::FnvHashSet;

/// Transforms a `PipelineLayoutDesc`.
///
/// Used to adjust automatically inferred `PipelineLayoutDesc`s with information that cannot be inferred.
pub struct PipelineLayoutDescTweaks<T> {
    inner: T,
    dynamic_buffers: FnvHashSet<(usize, usize)>,
}

impl<T> PipelineLayoutDescTweaks<T>
where
    T: PipelineLayoutDesc,
{
    /// Describe a layout, ensuring that each `(set, binding)` in `dynamic_buffers` is a dynamic buffers.
    pub fn new<I>(inner: T, dynamic_buffers: I) -> Self
    where
        I: IntoIterator<Item = (usize, usize)>,
    {
        let dynamic_buffers = dynamic_buffers.into_iter().collect();
        for &(set, binding) in &dynamic_buffers {
            debug_assert!(
                inner
                    .descriptor(set, binding)
                    .map_or(false, |desc| match desc.ty {
                        DescriptorDescTy::Buffer(_) => true,
                        _ => false,
                    }),
                "tried to make the non-buffer descriptor at set {} binding {} a dynamic buffer",
                set,
                binding
            );
        }
        Self {
            inner,
            dynamic_buffers,
        }
    }
}

unsafe impl<T> PipelineLayoutDesc for PipelineLayoutDescTweaks<T>
where
    T: PipelineLayoutDesc,
{
    #[inline]
    fn num_sets(&self) -> usize {
        self.inner.num_sets()
    }

    #[inline]
    fn num_bindings_in_set(&self, set: usize) -> Option<usize> {
        self.inner.num_bindings_in_set(set)
    }

    #[inline]
    fn descriptor(&self, set: usize, binding: usize) -> Option<DescriptorDesc> {
        self.inner
            .descriptor(set, binding)
            .map(|desc| match desc.ty {
                DescriptorDescTy::Buffer(ref buffer_desc)
                    if self.dynamic_buffers.contains(&(set, binding)) =>
                {
                    DescriptorDesc {
                        ty: DescriptorDescTy::Buffer(DescriptorBufferDesc {
                            dynamic: Some(true),
                            ..*buffer_desc
                        }),
                        ..desc
                    }
                }
                _ => desc,
            })
    }

    #[inline]
    fn num_push_constants_ranges(&self) -> usize {
        self.inner.num_push_constants_ranges()
    }

    // TODO: needs tests
    #[inline]
    fn push_constants_range(&self, num: usize) -> Option<PipelineLayoutDescPcRange> {
        self.inner.push_constants_range(num)
    }
}