rafx-plugins 0.0.16

Rendering framework built on an extensible asset pipeline
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

// This code is auto-generated by the shader processor.

#[allow(unused_imports)]
use rafx::RafxResult;

#[allow(unused_imports)]
use rafx::framework::{
    DescriptorSetAllocator, DescriptorSetArc, DescriptorSetBindings, DescriptorSetInitializer,
    DescriptorSetWriter, DescriptorSetWriterContext, DynDescriptorSet, ImageViewResource,
    ResourceArc,
};

#[derive(Copy, Clone, Debug)]
#[repr(C)]
pub struct ArgsStd140 {
    pub inverse_projection: [[f32; 4]; 4], // +0 (size: 64)
    pub inverse_view: [[f32; 4]; 4],       // +64 (size: 64)
} // 128 bytes

impl Default for ArgsStd140 {
    fn default() -> Self {
        ArgsStd140 {
            inverse_projection: <[[f32; 4]; 4]>::default(),
            inverse_view: <[[f32; 4]; 4]>::default(),
        }
    }
}

pub type ArgsUniform = ArgsStd140;

pub const SMP_DESCRIPTOR_SET_INDEX: usize = 0;
pub const SMP_DESCRIPTOR_BINDING_INDEX: usize = 0;
pub const SKYBOX_TEX_DESCRIPTOR_SET_INDEX: usize = 0;
pub const SKYBOX_TEX_DESCRIPTOR_BINDING_INDEX: usize = 1;
pub const UNIFORM_BUFFER_DESCRIPTOR_SET_INDEX: usize = 0;
pub const UNIFORM_BUFFER_DESCRIPTOR_BINDING_INDEX: usize = 2;

pub struct DescriptorSet0Args<'a> {
    pub skybox_tex: &'a ResourceArc<ImageViewResource>,
    pub uniform_buffer: &'a ArgsUniform,
}

impl<'a> DescriptorSetInitializer<'a> for DescriptorSet0Args<'a> {
    type Output = DescriptorSet0;

    fn create_dyn_descriptor_set(
        descriptor_set: DynDescriptorSet,
        args: Self,
    ) -> Self::Output {
        let mut descriptor = DescriptorSet0(descriptor_set);
        descriptor.set_args(args);
        descriptor
    }

    fn create_descriptor_set(
        descriptor_set_allocator: &mut DescriptorSetAllocator,
        descriptor_set: DynDescriptorSet,
        args: Self,
    ) -> RafxResult<DescriptorSetArc> {
        let mut descriptor = Self::create_dyn_descriptor_set(descriptor_set, args);
        descriptor.0.flush(descriptor_set_allocator)?;
        Ok(descriptor.0.descriptor_set().clone())
    }
}

impl<'a> DescriptorSetWriter<'a> for DescriptorSet0Args<'a> {
    fn write_to(
        descriptor_set: &mut DescriptorSetWriterContext,
        args: Self,
    ) {
        descriptor_set.set_image(SKYBOX_TEX_DESCRIPTOR_BINDING_INDEX as u32, args.skybox_tex);
        descriptor_set.set_buffer_data(
            UNIFORM_BUFFER_DESCRIPTOR_BINDING_INDEX as u32,
            args.uniform_buffer,
        );
    }
}

pub struct DescriptorSet0(pub DynDescriptorSet);

impl DescriptorSet0 {
    pub fn set_args_static(
        descriptor_set: &mut DynDescriptorSet,
        args: DescriptorSet0Args,
    ) {
        descriptor_set.set_image(SKYBOX_TEX_DESCRIPTOR_BINDING_INDEX as u32, args.skybox_tex);
        descriptor_set.set_buffer_data(
            UNIFORM_BUFFER_DESCRIPTOR_BINDING_INDEX as u32,
            args.uniform_buffer,
        );
    }

    pub fn set_args(
        &mut self,
        args: DescriptorSet0Args,
    ) {
        self.set_skybox_tex(args.skybox_tex);
        self.set_uniform_buffer(args.uniform_buffer);
    }

    pub fn set_skybox_tex(
        &mut self,
        skybox_tex: &ResourceArc<ImageViewResource>,
    ) {
        self.0
            .set_image(SKYBOX_TEX_DESCRIPTOR_BINDING_INDEX as u32, skybox_tex);
    }

    pub fn set_uniform_buffer(
        &mut self,
        uniform_buffer: &ArgsUniform,
    ) {
        self.0.set_buffer_data(
            UNIFORM_BUFFER_DESCRIPTOR_BINDING_INDEX as u32,
            uniform_buffer,
        );
    }

    pub fn flush(
        &mut self,
        descriptor_set_allocator: &mut DescriptorSetAllocator,
    ) -> RafxResult<()> {
        self.0.flush(descriptor_set_allocator)
    }
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn test_struct_args_std140() {
        assert_eq!(std::mem::size_of::<ArgsStd140>(), 128);
        assert_eq!(std::mem::size_of::<[[f32; 4]; 4]>(), 64);
        assert_eq!(std::mem::align_of::<[[f32; 4]; 4]>(), 4);
        assert_eq!(memoffset::offset_of!(ArgsStd140, inverse_projection), 0);
        assert_eq!(std::mem::size_of::<[[f32; 4]; 4]>(), 64);
        assert_eq!(std::mem::align_of::<[[f32; 4]; 4]>(), 4);
        assert_eq!(memoffset::offset_of!(ArgsStd140, inverse_view), 64);
    }
}