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

// Copyright (c) 2016 The vulkano developers
// Licensed under the Apache License, Version 2.0
// <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT
// license <LICENSE-MIT or http://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 SafeDeref;
use image::ImageViewAccess;
use std::sync::Arc;
//use sync::AccessFlagBits;
//use sync::PipelineStages;

/// A list of attachments.
// TODO: rework this trait
pub unsafe trait AttachmentsList {
    fn num_attachments(&self) -> usize;

    fn as_image_view_access(&self, index: usize) -> Option<&dyn ImageViewAccess>;
}

unsafe impl<T> AttachmentsList for T
    where T: SafeDeref,
          T::Target: AttachmentsList
{
    #[inline]
    fn num_attachments(&self) -> usize {
        (**self).num_attachments()
    }

    #[inline]
    fn as_image_view_access(&self, index: usize) -> Option<&dyn ImageViewAccess> {
        (**self).as_image_view_access(index)
    }
}

unsafe impl AttachmentsList for () {
    #[inline]
    fn num_attachments(&self) -> usize {
        0
    }

    #[inline]
    fn as_image_view_access(&self, _: usize) -> Option<&dyn ImageViewAccess> {
        None
    }
}

unsafe impl AttachmentsList for Vec<Arc<dyn ImageViewAccess + Send + Sync>> {
    #[inline]
    fn num_attachments(&self) -> usize {
        self.len()
    }

    #[inline]
    fn as_image_view_access(&self, index: usize) -> Option<&dyn ImageViewAccess> {
        self.get(index).map(|v| &**v as &_)
    }
}

unsafe impl<A, B> AttachmentsList for (A, B)
    where A: AttachmentsList,
          B: ImageViewAccess
{
    #[inline]
    fn num_attachments(&self) -> usize {
        self.0.num_attachments() + 1
    }

    #[inline]
    fn as_image_view_access(&self, index: usize) -> Option<&dyn ImageViewAccess> {
        if index == self.0.num_attachments() {
            Some(&self.1)
        } else {
            self.0.as_image_view_access(index)
        }
    }
}