Compare commits

..

5 commits

Author SHA1 Message Date
janis b8b9bf40a3
more command stuff 2026-04-12 20:23:45 +02:00
janis dde3093536
keep layout around in pipeline 2026-04-12 20:23:19 +02:00
janis 555244b221
more barriers 2026-04-12 15:44:59 +02:00
janis 7acda4d1fb
automatic barriers 2026-04-12 14:26:02 +02:00
janis 666e05bb8e
command impls 2026-04-12 10:33:59 +02:00
8 changed files with 1334 additions and 255 deletions

View file

@ -920,6 +920,10 @@ mod command_pools {
level: this.level,
})
}
pub fn raw(&self) -> vk::CommandBuffer {
self.handle()
}
}
pub struct CommandBufferFuture {

View file

@ -1,4 +1,9 @@
use std::{borrow::Cow, mem::ManuallyDrop, sync::Arc};
use std::{
borrow::Cow,
mem::ManuallyDrop,
ops::{Add, Sub},
sync::Arc,
};
use crate::{
device::{
@ -33,6 +38,22 @@ pub struct ImageDesc {
pub alloc_scheme: AllocationStrategy,
}
impl ImageDesc {
pub fn mip_level_range(&self) -> MipRange {
MipRange {
start: 0,
end: self.mip_levels,
}
}
pub fn layer_range(&self) -> MipRange {
MipRange {
start: 0,
end: self.array_layers,
}
}
}
impl std::hash::Hash for ImageDesc {
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.flags.hash(state);
@ -469,9 +490,9 @@ impl Image {
vk::ImageSubresourceRange::default()
.aspect_mask(desc.aspect)
.base_mip_level(desc.mip_range.start)
.level_count(desc.mip_range.count())
.level_count(desc.mip_range.len())
.base_array_layer(desc.layer_range.start)
.layer_count(desc.layer_range.count()),
.layer_count(desc.layer_range.len()),
);
let device = self.image.device();
@ -585,6 +606,185 @@ impl ImageViewDesc {
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct Extent {
pub width: u32,
pub height: u32,
pub depth: u32,
}
impl Extent {
pub fn as_offset(self) -> Offset {
Offset {
x: self.width as i32,
y: self.height as i32,
z: self.depth as i32,
}
}
pub fn from_offset(offset: Offset) -> Self {
Self {
width: offset.x as u32,
height: offset.y as u32,
depth: offset.z as u32,
}
}
pub fn min(self, other: Self) -> Self {
Self {
width: self.width.min(other.width),
height: self.height.min(other.height),
depth: self.depth.min(other.depth),
}
}
pub fn max(self, other: Self) -> Self {
Self {
width: self.width.max(other.width),
height: self.height.max(other.height),
depth: self.depth.max(other.depth),
}
}
}
impl Sub<Offset> for Extent {
type Output = Self;
fn sub(self, rhs: Offset) -> Self::Output {
Self {
width: self.width.saturating_sub(rhs.x as u32),
height: self.height.saturating_sub(rhs.y as u32),
depth: self.depth.saturating_sub(rhs.z as u32),
}
}
}
impl Add<Offset> for Extent {
type Output = Self;
fn add(self, rhs: Offset) -> Self::Output {
Self {
width: self.width.saturating_add(rhs.x as u32),
height: self.height.saturating_add(rhs.y as u32),
depth: self.depth.saturating_add(rhs.z as u32),
}
}
}
impl From<Extent> for vk::Extent3D {
fn from(extent: Extent) -> Self {
Self {
width: extent.width,
height: extent.height,
depth: extent.depth,
}
}
}
impl From<Extent> for (u32, u32, u32) {
fn from(extent: Extent) -> Self {
(extent.width, extent.height, extent.depth)
}
}
impl From<vk::Extent3D> for Extent {
fn from(extent: vk::Extent3D) -> Self {
Self {
width: extent.width,
height: extent.height,
depth: extent.depth,
}
}
}
impl From<(u32, u32, u32)> for Extent {
fn from((width, height, depth): (u32, u32, u32)) -> Self {
Self {
width,
height,
depth,
}
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct Offset {
pub x: i32,
pub y: i32,
pub z: i32,
}
impl Offset {
pub fn min(self, other: Self) -> Self {
Self {
x: self.x.min(other.x),
y: self.y.min(other.y),
z: self.z.min(other.z),
}
}
pub fn max(self, other: Self) -> Self {
Self {
x: self.x.max(other.x),
y: self.y.max(other.y),
z: self.z.max(other.z),
}
}
}
impl Add for Offset {
type Output = Self;
fn add(self, rhs: Self) -> Self::Output {
Self {
x: self.x.saturating_add(rhs.x),
y: self.y.saturating_add(rhs.y),
z: self.z.saturating_add(rhs.z),
}
}
}
impl Sub for Offset {
type Output = Self;
fn sub(self, rhs: Self) -> Self::Output {
Self {
x: self.x.saturating_sub(rhs.x),
y: self.y.saturating_sub(rhs.y),
z: self.z.saturating_sub(rhs.z),
}
}
}
impl From<Offset> for vk::Offset3D {
fn from(offset: Offset) -> Self {
Self {
x: offset.x,
y: offset.y,
z: offset.z,
}
}
}
impl From<Offset> for (i32, i32, i32) {
fn from(offset: Offset) -> Self {
(offset.x, offset.y, offset.z)
}
}
impl From<vk::Offset3D> for Offset {
fn from(offset: vk::Offset3D) -> Self {
Self {
x: offset.x,
y: offset.y,
z: offset.z,
}
}
}
impl From<(i32, i32, i32)> for Offset {
fn from((x, y, z): (i32, i32, i32)) -> Self {
Self { x, y, z }
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct MipRange {
start: u32,
@ -601,13 +801,57 @@ impl Default for MipRange {
}
impl MipRange {
pub fn count(&self) -> u32 {
pub fn len(&self) -> u32 {
self.end - self.start
}
pub fn start(&self) -> u32 {
self.start
}
pub fn end(&self) -> u32 {
self.end
}
pub fn range(&self, other: &Self) -> Self {
Self {
start: self.start.max(other.start).min(self.end),
end: self.end.min(other.end).max(self.start),
}
}
pub fn iter(&self) -> core::range::RangeIter<u32> {
self.into_iter()
}
pub fn fits_in(&self, total_mips: u32) -> bool {
self.start < total_mips && (self.end == vk::REMAINING_MIP_LEVELS || self.end <= total_mips)
}
pub fn intersects(&self, other: &Self) -> bool {
self.start < other.end && self.end > other.start
}
pub fn union(&self, other: &Self) -> Self {
Self {
start: self.start.min(other.start),
end: self.end.max(other.end),
}
}
}
impl IntoIterator for MipRange {
type Item = u32;
type IntoIter = core::range::RangeIter<u32>;
fn into_iter(self) -> Self::IntoIter {
core::range::Range {
start: self.start,
end: self.end,
}
.into_iter()
}
}
impl<R: core::ops::RangeBounds<u32>> From<R> for MipRange {
@ -1198,3 +1442,15 @@ impl From<vk::Format> for FormatClass {
}
}
}
pub fn format_to_aspect_mask(format: vk::Format) -> vk::ImageAspectFlags {
use vk::Format as F;
match format {
F::D16_UNORM | F::X8_D24_UNORM_PACK32 | F::D32_SFLOAT => vk::ImageAspectFlags::DEPTH,
F::D16_UNORM_S8_UINT | F::D24_UNORM_S8_UINT | F::D32_SFLOAT_S8_UINT => {
vk::ImageAspectFlags::DEPTH | vk::ImageAspectFlags::STENCIL
}
F::S8_UINT => vk::ImageAspectFlags::STENCIL,
_ => vk::ImageAspectFlags::COLOR,
}
}

View file

@ -2,6 +2,7 @@
negative_impls,
map_try_insert,
debug_closure_helpers,
core_io_borrowed_buf,
slice_partition_dedup
)]

View file

@ -19,6 +19,16 @@ pub struct ShaderStageDesc<'a> {
// specialization: Option<vk::SpecializationInfo>
}
impl ShaderStageDesc<'_> {
fn as_create_info(&'_ self) -> vk::PipelineShaderStageCreateInfo<'_> {
vk::PipelineShaderStageCreateInfo::default()
.module(self.module.raw())
.flags(self.flags)
.stage(self.stage)
.name(&self.entry)
}
}
#[derive(Debug, Default)]
pub struct DescriptorSetLayoutBindingDesc {
pub binding: u32,
@ -189,7 +199,7 @@ pub struct GraphicsPipelineDesc<'a> {
pub name: Option<Cow<'static, str>>,
pub shader_stages: &'a [ShaderStageDesc<'a>],
pub render_pass: Option<vk::RenderPass>,
pub layout: &'a PipelineLayout,
pub layout: Arc<PipelineLayout>,
pub subpass: Option<u32>,
pub base_pipeline: Option<Arc<Pipeline>>,
@ -544,6 +554,7 @@ impl ShaderModule {
pub struct Pipeline {
pipeline: DeviceObject<vk::Pipeline>,
bind_point: vk::PipelineBindPoint,
layout: Arc<PipelineLayout>,
}
impl<T: AsRef<DeviceInner>> ExternallyManagedObject<T> for vk::Pipeline {
@ -554,16 +565,6 @@ impl<T: AsRef<DeviceInner>> ExternallyManagedObject<T> for vk::Pipeline {
}
}
impl ShaderStageDesc<'_> {
fn as_create_info(&'_ self) -> vk::PipelineShaderStageCreateInfo<'_> {
vk::PipelineShaderStageCreateInfo::default()
.module(self.module.raw())
.flags(self.flags)
.stage(self.stage)
.name(&self.entry)
}
}
impl Pipeline {
pub fn new_compute(device: Device, desc: ComputePipelineDesc) -> crate::Result<Self> {
let info = &vk::ComputePipelineCreateInfo::default()
@ -593,6 +594,7 @@ impl Pipeline {
Ok(Self {
pipeline: DeviceObject::new_debug_named(device, pipeline, desc.name),
bind_point: vk::PipelineBindPoint::COMPUTE,
layout: desc.layout,
})
}
@ -761,6 +763,7 @@ impl Pipeline {
Ok(Self {
pipeline: DeviceObject::new_debug_named(device, pipeline, desc.name),
bind_point: vk::PipelineBindPoint::GRAPHICS,
layout: desc.layout,
})
}

View file

@ -1,4 +1,3 @@
#![allow(dead_code)]
//! This module defines the commands that can be recorded in the render
//! graph.
//! Commands can be allocated, and then recorded into a command
@ -13,78 +12,165 @@
//! recorded sequentially, in the order they were inserted into a command
//! buffer. During command insertion
use crate::render_graph::recorder::{CommandMeta, CommandRecorder, SideEffectMap};
use crate::{
device::DeviceOwned,
render_graph::recorder::{CommandRecorder, SideEffectMap},
};
use super::resources::*;
pub struct CopyBuffers {
pub src: Read<BufferSlice>,
pub dst: Write<BufferSlice>,
}
// pub struct CopyBuffers {
// pub src: Read<BufferSlice>,
// pub dst: Write<BufferSlice>,
// }
pub struct CopyTextures {
pub src: Read<TextureRegion>,
pub dst: Write<TextureRegion>,
}
// pub struct CopyTextures {
// pub src: Read<TextureRegion>,
// pub dst: Write<TextureRegion>,
// }
pub struct CopyBufferToTexture {
pub src: Read<BufferSlice>,
pub dst: Write<TextureRegion>,
}
// pub struct CopyBufferToTexture {
// pub src: Read<BufferSlice>,
// pub dst: Write<TextureRegion>,
// }
pub struct CopyTextureToBuffer {
pub src: Read<TextureRegion>,
pub dst: Write<BufferSlice>,
}
// pub struct CopyTextureToBuffer {
// pub src: Read<TextureRegion>,
// pub dst: Write<BufferSlice>,
// }
pub struct Copy<T: Resource, U: Resource> {
pub src: Read<T>,
pub dst: Write<U>,
pub src: T,
pub dst: U,
}
impl<T: Resource, U: Resource> Copy<T, U> {
fn side_effects_inner(&self, mut map: SideEffectMap) {
map.insert(
self.src.id(),
ResourceAccess {
range: self.src.into_access(),
access: vk::AccessFlags2::TRANSFER_READ,
},
self.src.side_effect(
map.reborrow(),
vk::AccessFlags2::TRANSFER_READ,
Some(vk::ImageLayout::TRANSFER_SRC_OPTIMAL),
);
map.insert(
self.dst.id(),
ResourceAccess {
range: self.dst.into_access(),
access: vk::AccessFlags2::TRANSFER_WRITE,
},
self.dst.side_effect(
map.reborrow(),
vk::AccessFlags2::TRANSFER_WRITE,
Some(vk::ImageLayout::TRANSFER_DST_OPTIMAL),
);
}
}
impl Command for Copy<BufferSlice, TextureRegion> {
impl Command for Copy<BufferRowSlice, TextureRegion> {
fn side_effects(&self, map: SideEffectMap) {
self.side_effects_inner(map)
}
fn apply(self, _recorder: &mut CommandRecorder) {
// CopyBufferToTexture {src, dst}.apply(recorder)
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
// it doesn't really make sense to copy the same data into multiple mip
// levels, considering the mip extents are different for each level.
debug_assert_eq!(self.dst.range.mip_levels.len(), 1);
let regions = &[vk::BufferImageCopy2::default()
.buffer_offset(self.src.row.offset)
.buffer_row_length(self.src.row.row_size)
.buffer_image_height(self.src.row.row_count)
.image_offset(self.dst.range.offset())
.image_extent(self.dst.range.extent())
.image_subresource(
vk::ImageSubresourceLayers::default()
.aspect_mask(self.dst.range.aspect)
.mip_level(self.dst.range.mip_levels.start())
.base_array_layer(self.dst.range.array_layers.start())
.layer_count(self.dst.range.array_layers.len()),
)];
let info = vk::CopyBufferToImageInfo2::default()
.src_buffer(recorder.get_buffer_handle(self.src.id()))
.dst_image(recorder.get_image_handle(self.dst.id()))
.dst_image_layout(recorder.get_image_layout(self.dst.id()))
.regions(regions);
unsafe {
dev.cmd_copy_buffer_to_image2(cmd.raw(), &info);
}
}
impl Command for Copy<TextureRegion, BufferSlice> {
}
impl Command for Copy<BufferRowSlices, TextureRegions> {
fn side_effects(&self, map: SideEffectMap) {
self.side_effects_inner(map)
}
fn apply(self, _recorder: &mut CommandRecorder) {
// CopyTextureToBuffer {src, dst}.apply(recorder)
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
let regions = &self
.src
.rows
.iter()
.zip(self.dst.ranges.iter())
.map(|(src_row, dst_range)| {
vk::BufferImageCopy2::default()
.buffer_offset(src_row.offset)
.buffer_row_length(src_row.row_size)
.buffer_image_height(src_row.row_count)
.image_offset(dst_range.offset())
.image_extent(dst_range.extent())
.image_subresource(
vk::ImageSubresourceLayers::default()
.aspect_mask(dst_range.aspect)
.mip_level(dst_range.mip_levels.start())
.base_array_layer(dst_range.array_layers.start())
.layer_count(dst_range.array_layers.len()),
)
})
.collect::<Vec<_>>();
let info = vk::CopyBufferToImageInfo2::default()
.src_buffer(recorder.get_buffer_handle(self.src.id()))
.dst_image(recorder.get_image_handle(self.dst.id()))
.dst_image_layout(recorder.get_image_layout(self.dst.id()))
.regions(regions);
unsafe {
dev.cmd_copy_buffer_to_image2(cmd.raw(), &info);
}
}
}
fn get_meta(&self) -> CommandMeta {
CommandMeta {
apply_command: |recorder, command_ptr, cursor| {
let command = unsafe { command_ptr.cast::<Self>().read() };
*cursor += std::mem::size_of::<Self>();
command.apply(recorder);
},
impl Command for Copy<TextureRegion, BufferRowSlice> {
fn side_effects(&self, map: SideEffectMap) {
self.side_effects_inner(map)
}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
let regions = &[vk::BufferImageCopy2::default()
.buffer_offset(self.dst.row.offset)
.buffer_row_length(self.dst.row.row_size)
.buffer_image_height(self.dst.row.row_count)
.image_offset(self.src.range.offset())
.image_extent(self.src.range.extent())
.image_subresource(
vk::ImageSubresourceLayers::default()
.aspect_mask(self.src.range.aspect)
.mip_level(self.src.range.mip_levels.start())
.base_array_layer(self.src.range.array_layers.start())
.layer_count(self.src.range.array_layers.len()),
)];
let info = vk::CopyImageToBufferInfo2::default()
.dst_buffer(recorder.get_buffer_handle(self.dst.id()))
.src_image(recorder.get_image_handle(self.src.id()))
.src_image_layout(recorder.get_image_layout(self.src.id()))
.regions(regions);
unsafe {
dev.cmd_copy_image_to_buffer2(cmd.raw(), &info);
}
}
}
@ -93,8 +179,39 @@ impl Command for Copy<TextureRegion, TextureRegion> {
self.side_effects_inner(map)
}
fn apply(self, _recorder: &mut CommandRecorder) {
// CopyTextures {src, dst}.apply(recorder)
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
let regions = &[vk::ImageCopy2::default()
.extent(self.src.range.extent())
.dst_offset(self.dst.range.offset())
.src_offset(self.src.range.offset())
.dst_subresource(
vk::ImageSubresourceLayers::default()
.aspect_mask(self.dst.range.aspect)
.mip_level(self.dst.range.mip_levels.start())
.base_array_layer(self.dst.range.array_layers.start())
.layer_count(self.dst.range.array_layers.len()),
)
.src_subresource(
vk::ImageSubresourceLayers::default()
.aspect_mask(self.src.range.aspect)
.mip_level(self.src.range.mip_levels.start())
.base_array_layer(self.src.range.array_layers.start())
.layer_count(self.src.range.array_layers.len()),
)];
let info = vk::CopyImageInfo2::default()
.dst_image(recorder.get_image_handle(self.dst.id()))
.dst_image_layout(recorder.get_image_layout(self.dst.id()))
.src_image(recorder.get_image_handle(self.src.id()))
.src_image_layout(recorder.get_image_layout(self.src.id()))
.regions(regions);
unsafe {
dev.cmd_copy_image2(cmd.raw(), &info);
}
}
}
impl Command for Copy<BufferSlice, BufferSlice> {
@ -102,8 +219,25 @@ impl Command for Copy<BufferSlice, BufferSlice> {
self.side_effects_inner(map)
}
fn apply(self, _recorder: &mut CommandRecorder) {
// CopyBuffers {src, dst}.apply(recorder)
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
debug_assert_eq!(self.src.range.size, self.dst.range.size);
let regions = &[vk::BufferCopy2::default()
.dst_offset(self.dst.range.offset)
.src_offset(self.src.range.offset)
.size(self.src.range.size)];
let info = vk::CopyBufferInfo2::default()
.dst_buffer(recorder.get_buffer_handle(self.dst.id()))
.src_buffer(recorder.get_buffer_handle(self.src.id()))
.regions(regions);
unsafe {
dev.cmd_copy_buffer2(cmd.raw(), &info);
}
}
}
@ -114,12 +248,10 @@ pub struct ClearTexture {
impl Command for ClearTexture {
fn side_effects(&self, mut map: SideEffectMap) {
map.insert(
self.dst.id(),
ResourceAccess {
range: self.dst.into_access(),
access: vk::AccessFlags2::TRANSFER_WRITE,
},
self.dst.side_effect(
map.reborrow(),
vk::AccessFlags2::TRANSFER_WRITE,
Some(vk::ImageLayout::TRANSFER_DST_OPTIMAL),
);
}
@ -133,53 +265,42 @@ pub struct UpdateBuffer {
impl Command for UpdateBuffer {
fn side_effects(&self, mut map: SideEffectMap) {
map.insert(
self.dst.id(),
ResourceAccess {
range: self.dst.into_access(),
access: vk::AccessFlags2::TRANSFER_WRITE,
},
);
self.dst
.side_effect(map.reborrow(), vk::AccessFlags2::TRANSFER_WRITE, None);
}
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct RenderPass {
pub color_attachments: Vec<Write<TextureRegion>>,
pub depth_attachment: Option<Write<TextureRegion>>,
pub stencil_attachment: Option<Write<TextureRegion>>,
pub struct BeginRendering {
pub color_attachments: Vec<TextureRegion>,
pub depth_attachment: Option<TextureRegion>,
pub stencil_attachment: Option<TextureRegion>,
pub area: (u32, u32),
pub layers: u32,
}
impl Command for RenderPass {
impl Command for BeginRendering {
fn side_effects(&self, mut map: SideEffectMap) {
for attachment in &self.color_attachments {
map.insert(
attachment.id(),
ResourceAccess {
range: attachment.into_access(),
access: vk::AccessFlags2::COLOR_ATTACHMENT_WRITE,
},
attachment.side_effect(
map.reborrow(),
vk::AccessFlags2::COLOR_ATTACHMENT_WRITE,
Some(vk::ImageLayout::COLOR_ATTACHMENT_OPTIMAL),
);
}
if let Some(depth) = &self.depth_attachment {
map.insert(
depth.id(),
ResourceAccess {
range: depth.into_access(),
access: vk::AccessFlags2::DEPTH_STENCIL_ATTACHMENT_WRITE,
},
depth.side_effect(
map.reborrow(),
vk::AccessFlags2::DEPTH_STENCIL_ATTACHMENT_WRITE,
Some(vk::ImageLayout::DEPTH_ATTACHMENT_OPTIMAL),
);
}
if let Some(stencil) = &self.stencil_attachment {
map.insert(
stencil.id(),
ResourceAccess {
range: stencil.into_access(),
access: vk::AccessFlags2::DEPTH_STENCIL_ATTACHMENT_WRITE,
},
stencil.side_effect(
map.reborrow(),
vk::AccessFlags2::DEPTH_STENCIL_ATTACHMENT_WRITE,
Some(vk::ImageLayout::STENCIL_ATTACHMENT_OPTIMAL),
);
}
}
@ -187,7 +308,17 @@ impl Command for RenderPass {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct BindPipeline(pub Pipeline);
pub struct EndRendering;
impl Command for EndRendering {
fn side_effects(&self, _map: SideEffectMap) {
// No resource access, but ends the render pass
}
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct BindPipeline;
impl Command for BindPipeline {
fn side_effects(&self, _map: SideEffectMap) {
@ -198,18 +329,16 @@ impl Command for BindPipeline {
}
pub struct BindVertexBuffers {
pub buffers: Vec<Read<BufferSlice>>,
pub buffers: Vec<BufferSlice>,
}
impl Command for BindVertexBuffers {
fn side_effects(&self, mut map: SideEffectMap) {
for buffer in &self.buffers {
map.insert(
buffer.id(),
ResourceAccess {
range: buffer.into_access(),
access: vk::AccessFlags2::VERTEX_ATTRIBUTE_READ,
},
buffer.side_effect(
map.reborrow(),
vk::AccessFlags2::VERTEX_ATTRIBUTE_READ,
None,
);
}
}
@ -217,25 +346,18 @@ impl Command for BindVertexBuffers {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct BindIndexBuffer(pub Read<BufferSlice>, pub IndexFormat);
pub struct BindIndexBuffer(pub BufferSlice, pub IndexFormat);
impl Command for BindIndexBuffer {
fn side_effects(&self, mut map: SideEffectMap) {
map.insert(
self.0.id(),
ResourceAccess {
range: self.0.into_access(),
access: vk::AccessFlags2::INDEX_READ,
},
);
self.0
.side_effect(map.reborrow(), vk::AccessFlags2::INDEX_READ, None);
}
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct BindDescriptorSets {
pub sets: Vec<DescriptorSet>,
}
pub struct BindDescriptorSets;
impl Command for BindDescriptorSets {
fn side_effects(&self, _map: SideEffectMap) {
@ -277,9 +399,7 @@ impl Command for SetScissor {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct PushConstants {
pub data: Box<[u8]>,
}
pub struct PushConstants;
impl Command for PushConstants {
fn side_effects(&self, _map: SideEffectMap) {
@ -305,11 +425,15 @@ pub struct DrawIndexedData {
}
pub struct DrawIndirectData {
pub indirect_buffer: Read<BufferSlice>,
pub indirect_buffer: BufferSlice,
pub count: u32,
pub stride: u32,
}
pub struct DrawIndexedIndirectData {
pub indirect_buffer: Read<BufferSlice>,
pub indirect_buffer: BufferSlice,
pub count: u32,
pub stride: u32,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
@ -321,39 +445,25 @@ pub enum IndexFormat {
pub struct Draw<T: IsDrawData> {
pub data: T,
pub vertex_buffers: Vec<Read<BufferSlice>>,
pub index_buffer: Option<(Read<BufferSlice>, IndexFormat)>,
pub count_buffer: Option<Read<BufferSlice>>,
pub vertex_buffers: Vec<BufferSlice>,
pub index_buffer: Option<(BufferSlice, IndexFormat)>,
}
impl<T: IsDrawData> Command for Draw<T> {
fn side_effects(&self, mut map: SideEffectMap) {
self.data.side_effects(map.reborrow());
for vertex_buffer in &self.vertex_buffers {
map.insert(
vertex_buffer.id(),
ResourceAccess {
range: vertex_buffer.into_access(),
access: vk::AccessFlags2::VERTEX_ATTRIBUTE_READ,
},
vertex_buffer.side_effect(
map.reborrow(),
vk::AccessFlags2::VERTEX_ATTRIBUTE_READ,
None,
);
}
if let Some((index_buffer, _)) = &self.index_buffer {
map.insert(
index_buffer.id(),
ResourceAccess {
range: index_buffer.into_access(),
access: vk::AccessFlags2::INDEX_READ,
},
);
}
if let Some(count_buffer) = &self.count_buffer {
map.insert(
count_buffer.id(),
ResourceAccess {
range: count_buffer.into_access(),
access: vk::AccessFlags2::INDIRECT_COMMAND_READ,
},
index_buffer.side_effect(
map.reborrow(),
vk::AccessFlags2::VERTEX_ATTRIBUTE_READ,
None,
);
}
}
@ -361,18 +471,19 @@ impl<T: IsDrawData> Command for Draw<T> {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct PipelineBarrier;
impl Command for PipelineBarrier {
fn side_effects(&self, _map: SideEffectMap) {}
fn apply(self, _recorder: &mut CommandRecorder) {
todo!()
}
}
pub trait Command: Sized {
fn side_effects(&self, map: SideEffectMap);
fn apply(self, recorder: &mut CommandRecorder);
fn get_meta(&self) -> CommandMeta {
CommandMeta {
apply_command: |recorder, command_ptr, cursor| {
let command = unsafe { command_ptr.cast::<Self>().read() };
*cursor += std::mem::size_of::<Self>();
command.apply(recorder);
},
}
}
}
mod sealed {
@ -383,34 +494,31 @@ mod sealed {
pub trait OutsideRenderPass {}
}
use ash::vk;
use sealed::*;
pub(super) use sealed::*;
impl IsDrawData for DrawData {}
impl IsDrawData for DrawIndexedData {}
impl IsDrawData for DrawIndirectData {
fn side_effects(&self, mut map: SideEffectMap) {
map.insert(
self.indirect_buffer.id(),
ResourceAccess {
range: self.indirect_buffer.into_access(),
access: vk::AccessFlags2::INDIRECT_COMMAND_READ,
},
self.indirect_buffer.side_effect(
map.reborrow(),
vk::AccessFlags2::INDIRECT_COMMAND_READ,
None,
);
}
}
impl IsDrawData for DrawIndexedIndirectData {
fn side_effects(&self, mut map: SideEffectMap) {
map.insert(
self.indirect_buffer.id(),
ResourceAccess {
range: self.indirect_buffer.into_access(),
access: vk::AccessFlags2::INDIRECT_COMMAND_READ,
},
self.indirect_buffer.side_effect(
map.reborrow(),
vk::AccessFlags2::INDIRECT_COMMAND_READ,
None,
);
}
}
impl OutsideRenderPass for RenderPass {}
impl OutsideRenderPass for BeginRendering {}
impl InsideRenderPass for EndRendering {}
impl<T: Resource, U: Resource> OutsideRenderPass for Copy<T, U> {}
impl OutsideRenderPass for ClearTexture {}
impl OutsideRenderPass for UpdateBuffer {}

View file

@ -1,6 +1,9 @@
#![allow(dead_code)]
mod legacy;
pub use legacy::*;
mod commands;
mod graph_builder;
mod recorder;
mod resources;

View file

@ -1,27 +1,96 @@
use std::{
collections::BTreeMap,
collections::{BTreeMap, BTreeSet},
mem::{MaybeUninit, size_of},
ptr::NonNull,
};
use crate::render_graph::{
commands::Command,
resources::{ResourceAccess, ResourceId},
use ash::vk;
use crate::{
commands::CommandBuffer,
images::ImageDesc,
render_graph::{
commands::{Command, InsideRenderPass, OutsideRenderPass},
resources::{ResourceAccess, ResourceId, TextureRegion, Write},
},
};
pub struct CommandRecorder;
pub struct RenderPassRecorder;
pub struct CommandMeta {
pub(crate) apply_command: unsafe fn(&mut CommandRecorder, NonNull<()>, cursor: &mut usize),
impl CommandRecorder {
pub fn cmd_buffer(&self) -> &CommandBuffer {
unimplemented!()
}
pub struct SideEffectMap<'a>(&'a mut BTreeMap<(ResourceId, u32), ResourceAccess>, u32);
pub fn get_buffer_handle(&self, _id: ResourceId) -> vk::Buffer {
unimplemented!()
}
pub fn get_image_desc(&self, _id: ResourceId) -> &ImageDesc {
unimplemented!()
}
pub fn get_image_handle(&self, _id: ResourceId) -> vk::Image {
unimplemented!()
}
pub fn get_image_layout(&self, _id: ResourceId) -> vk::ImageLayout {
unimplemented!()
}
pub fn get_image_view_handle(&self, _id: ResourceId) -> vk::ImageView {
unimplemented!()
}
}
struct CommandMeta {
apply_command: unsafe fn(&mut CommandRecorder, NonNull<()>, cursor: &mut usize),
}
enum VecOrSingle<T> {
Vec(Vec<T>),
Single(T),
}
impl<T> VecOrSingle<T> {
fn push(&mut self, value: T)
where
T: Default,
{
match self {
VecOrSingle::Vec(vec) => vec.push(value),
VecOrSingle::Single(existing) => {
let existing = std::mem::take(existing);
*self = VecOrSingle::Vec(vec![existing, value]);
}
}
}
fn iter(&self) -> core::slice::Iter<'_, T> {
match self {
VecOrSingle::Vec(items) => items.as_slice().iter(),
VecOrSingle::Single(item) => core::slice::from_ref(item).iter(),
}
}
}
#[derive(Default)]
struct SideEffects {
map: BTreeMap<(ResourceId, u32), VecOrSingle<ResourceAccess>>,
resources: BTreeSet<ResourceId>,
}
pub struct SideEffectMap<'a>(&'a mut SideEffects, u32);
impl<'a> SideEffectMap<'a> {
pub fn insert(&mut self, id: ResourceId, access: ResourceAccess) {
self.0.insert((id, self.1), access);
self.0.resources.insert(id);
self.0
.map
.entry((id, self.1))
.and_modify(|entry| entry.push(access))
.or_insert_with(|| VecOrSingle::Single(access));
}
pub fn reborrow(&mut self) -> SideEffectMap<'_> {
SideEffectMap(self.0, self.1)
}
@ -32,7 +101,23 @@ pub struct CommandList {
cursor: usize,
// each command that accesses a resource adds an entry to this map (id, command_index) -> access
// during command recording, we fold accesses to the same resource as commands are recorded, and when a command reads a resource, we check for any previous writes to be made available/visible
side_effects: BTreeMap<(ResourceId, u32), ResourceAccess>,
side_effects: SideEffects,
num_commands: u32,
}
#[must_use]
pub struct RenderPass<'a> {
cmd_list: &'a mut CommandList,
}
impl<'a> RenderPass<'a> {
pub fn finalise(self) {
self.cmd_list.push_inner(super::commands::EndRendering);
}
pub fn push<C: Command + InsideRenderPass>(&mut self, command: C) {
self.cmd_list.push_inner(command);
}
}
impl CommandList {
@ -40,16 +125,43 @@ impl CommandList {
Self {
command_bytes: Vec::new(),
cursor: 0,
side_effects: BTreeMap::new(),
side_effects: SideEffects::default(),
num_commands: 0,
}
}
pub fn push<C: Command>(&mut self, command: C) {
pub fn begin_rendering(
&mut self,
color_attachments: Vec<TextureRegion>,
depth_attachment: Option<TextureRegion>,
stencil_attachment: Option<TextureRegion>,
area: (u32, u32),
layers: u32,
) -> RenderPass<'_> {
self.push(super::commands::BeginRendering {
color_attachments,
depth_attachment,
stencil_attachment,
area,
layers,
});
RenderPass { cmd_list: self }
}
pub fn push<C: Command + OutsideRenderPass>(&mut self, command: C) {
self.push_inner(command);
}
fn push_inner<C: Command>(&mut self, command: C) {
#[repr(C, packed)]
struct Packed<C> {
meta: CommandMeta,
command: C,
}
command.side_effects(SideEffectMap(&mut self.side_effects, self.num_commands));
self.num_commands += 1;
let packed = Packed {
meta: CommandMeta {
apply_command: |recorder, command_ptr, cursor| {
@ -68,4 +180,139 @@ impl CommandList {
bytes.cast::<Packed<C>>().write_unaligned(packed);
}
}
fn fold_side_effects(&self) -> Vec<(u32, super::commands::PipelineBarrier)> {
// Buffer accesses can be granularly tracked by their offset and size;
// Image accesses are tracked by their subresource range, which means all regions of the same mip/array layer are considered the same subresource for the purposes of ImageMemoryBarriers
// But, for simplicity sake, and since buffers can already be created
// with specific sizes from device memory ranges, and aliased as
// required, we can just fold all buffer accesses together and collapse
// the size and offsets to min([offset, ..]), max([offset + size, ..])
// for all accesses to the same buffer.
// Generally, we want to be as granular as possible with barriers when
// it makes sense, but reduce the number of calls to
// vkCmdPipelineBarrier.
// Image accesses cannot be folded if they differ in: layout, access type, access stage/mask.
struct SubresourceState {
command_index: u32,
}
struct Barrier {
resource_id: ResourceId,
command_index: u32,
from: ResourceAccess,
to: usize,
}
// map to keep track of the current state of each subresource
let mut subresource_state: BTreeMap<ResourceId, Vec<(ResourceAccess, SubresourceState)>> =
Default::default();
let mut barriers = Vec::new();
for id in &self.side_effects.resources {
// check if this subresource has been accessed before, and if we can
// fold this access with the previous one
let entry = subresource_state.entry(*id).or_default();
for (command_index, accesses) in self
.side_effects
.map
.range((*id, 0)..=(*id, u32::MAX))
.map(|((_id, command_index), accesses)| (command_index, accesses))
{
for access in accesses.iter() {
let mut access = *access;
// iterate over the previous accesses to this subresource:
// - if we find an access with the same polarity, attempt to
// fold; a barrier already exists if it is needed.
// - if we find an access with a different polarity, we need to
// insert a barrier before the current command index.
// - if we find need for a layout transition, we need to insert
// a barrier before the current command index.
let mut folded = false;
let len = entry.len();
for (_i, (sub, _state)) in &mut entry.iter_mut().enumerate().rev() {
if !folded && sub.try_fold(&access) {
access = *sub;
folded = true;
} else if sub.conflicts(&access) {
// if this access conflicts with a previous one, we need
// to insert a barrier before the current command index.
// if we have managed to fold this access with a
// previous one, a barrier already exists.
// we know that `from` is correct, as the first
// conflicting access must be of opposite polarity (or a
// layout transition) and no more accesses can be folded
// into it.
// we store the index of `to`, as more accesses may be
// folded into it, expanding mips/array layers.
barriers.push(Barrier {
resource_id: *id,
command_index: *command_index,
from: *sub,
to: len,
});
// we can stop iterating over previous accesses, as we
// can no longer fold into any accesses before this
// conflict.
break;
}
}
if !folded {
// if we couldn't fold this access with any previous one, add it as a new entry
entry.push((
access,
SubresourceState {
command_index: *command_index,
},
));
}
}
}
}
// construct barriers
for barrier in barriers {
// stuff
}
todo!()
}
fn record_commands(&self, recorder: &mut CommandRecorder) {
let mut cursor = 0;
let mut barriers = self.fold_side_effects();
let mut command_index = 0;
while cursor < self.command_bytes.len() {
barriers
.drain(barriers.partition_point(|(idx, _)| *idx <= command_index)..)
.for_each(|(_, barrier)| {
barrier.apply(recorder);
});
unsafe {
let meta_ptr = self
.command_bytes
.as_ptr()
.add(cursor)
.cast::<CommandMeta>();
let meta = meta_ptr.read_unaligned();
let command_ptr = meta_ptr.add(1).cast::<()>().cast_mut();
(meta.apply_command)(recorder, NonNull::new_unchecked(command_ptr), &mut cursor);
}
command_index += 1;
}
}
}

View file

@ -1,13 +1,17 @@
use std::ops::Deref;
use ash::vk;
use crate::images::MipRange;
use crate::{images::MipRange, render_graph::recorder::SideEffectMap};
pub trait Resource: Sized + AsResourceRange {
pub trait Resource: Sized {
fn id(&self) -> ResourceId;
}
pub trait AsResourceRange {
fn into_access(&self) -> ResourceRange;
fn side_effect(
&self,
map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
);
}
mod impls {
@ -17,93 +21,240 @@ mod impls {
fn id(&self) -> ResourceId {
self.0
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
_layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: BufferRange::full().into(),
access,
},
);
}
}
impl Resource for BufferSlice {
fn id(&self) -> ResourceId {
self.buffer.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
_layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: self.range.into(),
access,
},
);
}
}
impl Resource for BufferSlices {
fn id(&self) -> ResourceId {
self.buffer.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
_layout: Option<vk::ImageLayout>,
) {
for range in &self.ranges {
map.insert(
self.id(),
ResourceAccess {
range: (*range).into(),
access,
},
);
}
}
}
impl Resource for BufferRowSlice {
fn id(&self) -> ResourceId {
self.buffer.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
_layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: BufferRange {
offset: self.row.offset,
size: (self.row.row_count as u64) * (self.row.row_size as u64),
}
.into(),
access,
},
);
}
}
impl Resource for BufferRowSlices {
fn id(&self) -> ResourceId {
self.buffer.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
_layout: Option<vk::ImageLayout>,
) {
for row in &self.rows {
map.insert(
self.id(),
ResourceAccess {
range: BufferRange {
offset: row.offset,
size: (row.row_count as u64) * (row.row_size as u64),
}
.into(),
access,
},
);
}
}
}
impl Resource for Texture {
fn id(&self) -> ResourceId {
self.0
}
}
impl Resource for BufferSlice {
fn id(&self) -> ResourceId {
self.buffer.id()
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: (TextureRange::full(), layout.unwrap_or_default()).into(),
access,
},
);
}
}
impl Resource for TextureRegion {
fn id(&self) -> ResourceId {
self.texture.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: (self.range, layout.unwrap_or_default()).into(),
access,
},
);
}
}
impl Resource for TextureRegions {
fn id(&self) -> ResourceId {
self.texture.id()
}
impl AsResourceRange for BufferSlice {
fn into_access(&self) -> ResourceRange {
ResourceRange::Buffer {
offset: self.offset,
size: self.size,
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
for range in &self.ranges {
map.insert(
self.id(),
ResourceAccess {
range: (*range, layout.unwrap_or_default()).into(),
access,
},
);
}
}
}
impl Resource for TextureView {
fn id(&self) -> ResourceId {
self.texture.id()
}
fn side_effect(
&self,
mut map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
map.insert(
self.id(),
ResourceAccess {
range: (self.range, layout.unwrap_or_default()).into(),
access,
},
);
}
}
impl AsResourceRange for TextureRegion {
fn into_access(&self) -> ResourceRange {
ResourceRange::Texture {
aspect: self.aspect,
mip_level: self.mip_level,
array_layers: self.array_layers,
origin: self.origin,
extent: self.extent,
}
}
}
impl AsResourceRange for Texture {
fn into_access(&self) -> ResourceRange {
ResourceRange::Texture {
aspect: vk::ImageAspectFlags::NONE,
mip_level: 0,
array_layers: MipRange::default(),
origin: (0, 0, 0),
extent: (u32::MAX, u32::MAX, u32::MAX),
}
}
}
impl AsResourceRange for Buffer {
fn into_access(&self) -> ResourceRange {
ResourceRange::Buffer {
offset: 0,
size: u64::MAX,
}
}
}
impl<T: AsResourceRange> AsResourceRange for Read<T> {
fn into_access(&self) -> ResourceRange {
self.0.into_access()
}
}
impl<T: AsResourceRange> AsResourceRange for Write<T> {
fn into_access(&self) -> ResourceRange {
self.0.into_access()
}
}
impl<T: AsResourceRange> AsResourceRange for ReadWrite<T> {
fn into_access(&self) -> ResourceRange {
self.0.into_access()
}
}
impl<T: Resource> Resource for Read<T> {
fn id(&self) -> ResourceId {
self.0.id()
}
fn side_effect(
&self,
map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
self.0.side_effect(map, access, layout);
}
}
impl<T: Resource> Resource for Write<T> {
fn id(&self) -> ResourceId {
self.0.id()
}
fn side_effect(
&self,
map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
self.0.side_effect(map, access, layout);
}
}
impl<T: Resource> Resource for ReadWrite<T> {
fn id(&self) -> ResourceId {
self.0.id()
}
fn side_effect(
&self,
map: SideEffectMap,
access: vk::AccessFlags2,
layout: Option<vk::ImageLayout>,
) {
self.0.side_effect(map, access, layout);
}
}
}
@ -111,13 +262,133 @@ pub struct Pipeline;
pub struct DescriptorSet;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct ResourceId(u32);
pub struct ResourceId(pub u32);
#[derive(Debug, Default, Clone, Copy)]
pub struct ResourceAccess {
pub range: ResourceRange,
pub access: vk::AccessFlags2,
}
impl ResourceAccess {
/// Attempts to fold another ResourceAccess into this one. Returns true if
/// successful, false if they are incompatible.
pub fn try_fold(&mut self, other: &Self) -> bool {
if self.access != other.access {
return false;
}
match (&mut self.range, &other.range) {
(ResourceRange::Buffer { .. }, ResourceRange::Texture { .. }) => false,
(ResourceRange::Texture { .. }, ResourceRange::Buffer { .. }) => false,
(
ResourceRange::Buffer { offset, size },
ResourceRange::Buffer {
offset: offset_b,
size: size_b,
},
) => {
// only fold if other lies within self
if *offset > *offset_b || *offset + *size < *offset_b + *size_b {
return false;
}
true
}
(
ResourceRange::Texture {
aspect,
mip_level,
array_layers,
layout,
..
},
ResourceRange::Texture {
aspect: aspect_b,
mip_level: mip_level_b,
array_layers: array_layers_b,
layout: layout_b,
..
},
) => {
if aspect != aspect_b
|| layout != layout_b
|| !mip_level.intersects(mip_level_b)
|| !array_layers.intersects(array_layers_b)
{
return false;
}
*mip_level = mip_level.union(mip_level_b);
*array_layers = array_layers.union(array_layers_b);
// let lower_left = Offset::from(*origin).min(Offset::from(*origin_b));
// let upper_right = (Offset::from(*origin) + Extent::from(*extent).as_offset())
// .max(Offset::from(*origin_b) + Extent::from(*extent_b).as_offset());
// *origin = lower_left.into();
// *extent = Extent::from_offset(upper_right - lower_left).into();
true
}
}
}
pub fn conflicts(&self, other: &Self) -> bool {
match (&self.range, &other.range) {
(ResourceRange::Buffer { .. }, ResourceRange::Texture { .. }) => false,
(ResourceRange::Texture { .. }, ResourceRange::Buffer { .. }) => false,
(
ResourceRange::Buffer { offset, size },
ResourceRange::Buffer {
offset: offset_b,
size: size_b,
},
) => {
// two reads don't conflict
// TODO: two writes may conflict if the user cares about the order of writes.
if access_flag_is_read(self.access) == access_flag_is_read(other.access) {
return false;
}
// must be a read/write conflict, check if subresources intersect
!(*offset > *offset_b + *size_b || *offset_b > *offset + *size)
}
(
ResourceRange::Texture {
aspect,
mip_level,
array_layers,
layout,
..
},
ResourceRange::Texture {
aspect: aspect_b,
mip_level: mip_level_b,
array_layers: array_layers_b,
layout: layout_b,
..
},
) => {
// layouts differing means we need a layout transition, which is a conflict
if layout != layout_b {
return true;
}
// two reads don't conflict
// TODO: two writes may conflict if the user cares about the order of writes.
if access_flag_is_read(self.access) == access_flag_is_read(other.access) {
return false;
}
// must be a read/write conflict, check if subresources intersect
aspect.intersects(*aspect_b)
&& (mip_level.intersects(mip_level_b)
|| array_layers.intersects(array_layers_b))
}
}
}
}
#[derive(Debug, Clone, Copy)]
pub enum ResourceRange {
Buffer {
offset: u64,
@ -125,29 +396,215 @@ pub enum ResourceRange {
},
Texture {
aspect: vk::ImageAspectFlags,
mip_level: u32,
mip_level: MipRange,
array_layers: MipRange,
origin: (u32, u32, u32),
origin: (i32, i32, i32),
extent: (u32, u32, u32),
layout: vk::ImageLayout,
},
}
pub struct Buffer(ResourceId);
pub struct Texture(ResourceId);
pub struct BufferSlice {
pub buffer: Buffer,
impl Default for ResourceRange {
fn default() -> Self {
ResourceRange::Buffer {
offset: 0,
size: u64::MAX,
}
}
}
#[derive(Debug, Default, Clone, Copy)]
pub struct BufferRange {
pub offset: u64,
pub size: u64,
}
impl BufferRange {
pub fn full() -> Self {
Self {
offset: 0,
size: u64::MAX,
}
}
}
#[derive(Debug, Default, Clone, Copy)]
pub struct BufferRows {
pub offset: u64,
pub row_count: u32,
pub row_size: u32,
}
#[derive(Debug, Default, Clone, Copy)]
pub struct TextureRange {
pub aspect: vk::ImageAspectFlags,
pub mip_levels: MipRange,
pub array_layers: MipRange,
pub origin: (i32, i32, i32),
pub extent: (u32, u32, u32),
}
impl TextureRange {
pub fn offset(&self) -> vk::Offset3D {
vk::Offset3D {
x: self.origin.0,
y: self.origin.1,
z: self.origin.2,
}
}
pub fn extent(&self) -> vk::Extent3D {
vk::Extent3D {
width: self.extent.0,
height: self.extent.1,
depth: self.extent.2,
}
}
}
impl From<(TextureRange, vk::ImageLayout)> for ResourceRange {
fn from((value, layout): (TextureRange, vk::ImageLayout)) -> Self {
ResourceRange::Texture {
aspect: value.aspect,
mip_level: value.mip_levels,
array_layers: value.array_layers,
origin: value.origin,
extent: value.extent,
layout,
}
}
}
impl From<BufferRange> for ResourceRange {
fn from(value: BufferRange) -> Self {
ResourceRange::Buffer {
offset: value.offset,
size: value.size,
}
}
}
impl TextureRange {
pub fn full() -> Self {
Self {
aspect: vk::ImageAspectFlags::COLOR
| vk::ImageAspectFlags::DEPTH
| vk::ImageAspectFlags::STENCIL,
mip_levels: MipRange::default(),
array_layers: MipRange::default(),
origin: (0, 0, 0),
extent: (u32::MAX, u32::MAX, u32::MAX),
}
}
}
pub struct Buffer(pub ResourceId);
impl Buffer {
pub fn full_slice(self) -> BufferSlice {
BufferSlice {
buffer: self,
range: BufferRange {
offset: 0,
size: u64::MAX,
},
}
}
}
pub struct Texture(pub ResourceId);
pub struct BufferSlice {
pub buffer: Buffer,
pub range: BufferRange,
}
pub struct BufferSlices {
pub buffer: Buffer,
pub ranges: Vec<BufferRange>,
}
pub struct BufferRowSlice {
pub buffer: Buffer,
pub row: BufferRows,
}
pub struct BufferRowSlices {
pub buffer: Buffer,
pub rows: Vec<BufferRows>,
}
pub struct TextureRegion {
pub texture: Texture,
pub aspect: vk::ImageAspectFlags,
pub mip_level: u32,
pub array_layers: MipRange,
pub origin: (u32, u32, u32),
pub extent: (u32, u32, u32),
pub range: TextureRange,
}
pub struct TextureRegions {
pub texture: Texture,
pub ranges: Vec<TextureRange>,
}
pub struct TextureView {
pub texture: Texture,
pub format: vk::Format,
pub mapping: vk::ComponentMapping,
pub range: TextureRange,
}
pub struct Read<T>(pub T);
pub struct Write<T>(pub T);
pub struct ReadWrite<T>(pub T);
impl<T> Deref for Read<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<T> Deref for Write<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<T> Deref for ReadWrite<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
pub fn access_flag_is_read(access: vk::AccessFlags2) -> bool {
access.intersects(
vk::AccessFlags2::INDIRECT_COMMAND_READ
| vk::AccessFlags2::INDEX_READ
| vk::AccessFlags2::VERTEX_ATTRIBUTE_READ
| vk::AccessFlags2::UNIFORM_READ
| vk::AccessFlags2::INPUT_ATTACHMENT_READ
| vk::AccessFlags2::SHADER_READ
| vk::AccessFlags2::COLOR_ATTACHMENT_READ
| vk::AccessFlags2::DEPTH_STENCIL_ATTACHMENT_READ
| vk::AccessFlags2::TRANSFER_READ
| vk::AccessFlags2::HOST_READ
| vk::AccessFlags2::MEMORY_READ
| vk::AccessFlags2::SHADER_SAMPLED_READ
| vk::AccessFlags2::SHADER_STORAGE_READ
| vk::AccessFlags2::VIDEO_DECODE_READ_KHR
| vk::AccessFlags2::VIDEO_ENCODE_READ_KHR
| vk::AccessFlags2::MICROMAP_READ_EXT
| vk::AccessFlags2::ACCELERATION_STRUCTURE_READ_KHR
| vk::AccessFlags2::FRAGMENT_DENSITY_MAP_READ_EXT
| vk::AccessFlags2::FRAGMENT_SHADING_RATE_ATTACHMENT_READ_KHR
| vk::AccessFlags2::CONDITIONAL_RENDERING_READ_EXT
| vk::AccessFlags2::SHADER_BINDING_TABLE_READ_KHR
| vk::AccessFlags2::DESCRIPTOR_BUFFER_READ_EXT
| vk::AccessFlags2::COLOR_ATTACHMENT_READ_NONCOHERENT_EXT
| vk::AccessFlags2::COMMAND_PREPROCESS_READ_NV
| vk::AccessFlags2::TRANSFORM_FEEDBACK_COUNTER_READ_EXT
| vk::AccessFlags2::INVOCATION_MASK_READ_HUAWEI,
)
}