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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162

use super::*;
use ash;

/// Simplified wrapper around `vkCmdPipelineBarrier`.
/// The mapping functions defined above are used to translate the passed in
/// barrier definitions into a set of pipeline stages and native Vulkan memory
/// barriers to be passed to `vkCmdPipelineBarrier`.
/// `command_buffer` is passed unmodified to `vkCmdPipelineBarrier`.
pub fn pipeline_barrier(
    device: &ash::Device,
    command_buffer: vk::CommandBuffer,
    global_barrier: Option<GlobalBarrier>,
    buffer_barriers: &[BufferBarrier],
    image_barriers: &[ImageBarrier],
) {
    let mut src_stage_mask = vk::PipelineStageFlags::TOP_OF_PIPE;
    let mut dst_stage_mask = vk::PipelineStageFlags::BOTTOM_OF_PIPE;

    // TODO: Optimize out the Vec heap allocations
    let mut vk_memory_barriers: Vec<vk::MemoryBarrier> = Vec::with_capacity(1);
    let mut vk_buffer_barriers: Vec<vk::BufferMemoryBarrier> =
        Vec::with_capacity(buffer_barriers.len());
    let mut vk_image_barriers: Vec<vk::ImageMemoryBarrier> =
        Vec::with_capacity(image_barriers.len());

    // Global memory barrier
    if let Some(ref barrier) = global_barrier {
        let (src_mask, dst_mask, barrier) = get_memory_barrier(barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_memory_barriers.push(barrier);
    }

    // Buffer memory barriers
    for buffer_barrier in buffer_barriers {
        let (src_mask, dst_mask, barrier) = get_buffer_memory_barrier(buffer_barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_buffer_barriers.push(barrier);
    }

    // Image memory barriers
    for image_barrier in image_barriers {
        let (src_mask, dst_mask, barrier) = get_image_memory_barrier(image_barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_image_barriers.push(barrier);
    }

    unsafe {
        device.cmd_pipeline_barrier(
            command_buffer,
            src_stage_mask,
            dst_stage_mask,
            vk::DependencyFlags::empty(),
            &vk_memory_barriers,
            &vk_buffer_barriers,
            &vk_image_barriers,
        );
    }
}

/// Wrapper around `vkCmdSetEvent`.
/// Sets an event when the accesses defined by `previous_accesses` are completed.
/// `command_buffer` and `event` are passed unmodified to `vkCmdSetEvent`.
pub fn set_event(
    device: &ash::Device,
    command_buffer: vk::CommandBuffer,
    event: vk::Event,
    previous_accesses: &[AccessType],
) {
    let mut stage_mask = vk::PipelineStageFlags::TOP_OF_PIPE;
    for previous_access in previous_accesses {
        let previous_info = get_access_info(*previous_access);
        stage_mask |= previous_info.stage_mask;
    }

    unsafe {
        device.cmd_set_event(command_buffer, event, stage_mask);
    }
}

/// Wrapper around `vkCmdResetEvent`.
/// Resets an event when the accesses defined by `previous_accesses` are completed.
/// `command_buffer` and `event` are passed unmodified to `vkCmdResetEvent`.
pub fn reset_event(
    device: &ash::Device,
    command_buffer: vk::CommandBuffer,
    event: vk::Event,
    previous_accesses: &[AccessType],
) {
    let mut stage_mask = vk::PipelineStageFlags::TOP_OF_PIPE;
    for previous_access in previous_accesses {
        let previous_info = get_access_info(*previous_access);
        stage_mask |= previous_info.stage_mask;
    }

    unsafe {
        device.cmd_reset_event(command_buffer, event, stage_mask);
    }
}

/// Simplified wrapper around `vkCmdWaitEvents`.
/// The mapping functions defined above are used to translate the passed in
/// barrier definitions into a set of pipeline stages and native Vulkan memory
/// barriers to be passed to `vkCmdPipelineBarrier`.
///
/// `commandBuffer` and `events` are passed unmodified to `vkCmdWaitEvents`.
pub fn wait_events(
    device: &ash::Device,
    command_buffer: vk::CommandBuffer,
    events: &[vk::Event],
    global_barrier: Option<GlobalBarrier>,
    buffer_barriers: &[BufferBarrier],
    image_barriers: &[ImageBarrier],
) {
    let mut src_stage_mask = vk::PipelineStageFlags::TOP_OF_PIPE;
    let mut dst_stage_mask = vk::PipelineStageFlags::BOTTOM_OF_PIPE;

    // TODO: Optimize out the Vec heap allocations
    let mut vk_memory_barriers: Vec<vk::MemoryBarrier> = Vec::with_capacity(1);
    let mut vk_buffer_barriers: Vec<vk::BufferMemoryBarrier> =
        Vec::with_capacity(buffer_barriers.len());
    let mut vk_image_barriers: Vec<vk::ImageMemoryBarrier> =
        Vec::with_capacity(image_barriers.len());

    // Global memory barrier
    if let Some(ref barrier) = global_barrier {
        let (src_mask, dst_mask, barrier) = get_memory_barrier(barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_memory_barriers.push(barrier);
    }

    // Buffer memory barriers
    for buffer_barrier in buffer_barriers {
        let (src_mask, dst_mask, barrier) = get_buffer_memory_barrier(buffer_barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_buffer_barriers.push(barrier);
    }

    // Image memory barriers
    for image_barrier in image_barriers {
        let (src_mask, dst_mask, barrier) = get_image_memory_barrier(image_barrier);
        src_stage_mask |= src_mask;
        dst_stage_mask |= dst_mask;
        vk_image_barriers.push(barrier);
    }

    unsafe {
        device.cmd_wait_events(
            command_buffer,
            &events,
            src_stage_mask,
            dst_stage_mask,
            &vk_memory_barriers,
            &vk_buffer_barriers,
            &vk_image_barriers,
        );
    }
}