Compare commits

...

2 commits

Author SHA1 Message Date
janis 933b4a5979
compiles! 2026-04-12 21:12:00 +02:00
janis a297dd94da
idk? 2026-04-12 21:03:58 +02:00
7 changed files with 117 additions and 40 deletions

View file

@ -631,7 +631,6 @@ pub fn egui_pass(
let record: Box<RecordFn> = Box::new({
let pipeline = egui_state.pipeline.clone();
let pipeline_layout = egui_state.pipeline_layout.clone();
let descriptor_set = egui_state.descriptor_set;
let screen_rect = egui.screen_rect();
@ -733,7 +732,7 @@ pub fn egui_pass(
cmd.bind_indices(indices.raw(), 0, vk::IndexType::UINT32);
cmd.bind_vertex_buffers(&[vertices.raw()], &[0]);
cmd.push_constants(
&pipeline_layout,
pipeline.layout(),
vk::ShaderStageFlags::VERTEX,
0,
bytemuck::cast_slice(
@ -741,7 +740,7 @@ pub fn egui_pass(
),
);
cmd.bind_descriptor_sets(
&pipeline_layout,
pipeline.layout(),
vk::PipelineBindPoint::GRAPHICS,
&[descriptor_set],
);

View file

@ -690,7 +690,6 @@ pub struct EguiState {
descriptor_set: vk::DescriptorSet,
#[allow(unused)]
descriptor_layout: pipeline::DescriptorSetLayout,
pipeline_layout: Arc<pipeline::PipelineLayout>,
pipeline: Arc<pipeline::Pipeline>,
}
@ -789,7 +788,7 @@ impl EguiState {
Result::Ok((descriptor_pool, descriptor_layout, sets))
})?;
let pipeline_layout = pipeline::PipelineLayout::new(
let pipeline_layout = Arc::new(pipeline::PipelineLayout::new(
device.clone(),
pipeline::PipelineLayoutDesc {
descriptor_set_layouts: &[&descriptor_layout],
@ -800,7 +799,7 @@ impl EguiState {
}],
name: Some("egui-pipeline-layout".into()),
},
)?;
)?);
let frag_shader = pipeline::ShaderModule::new_from_path(
device.clone(),
@ -831,7 +830,7 @@ impl EguiState {
},
],
render_pass: None,
layout: &pipeline_layout,
layout: pipeline_layout,
subpass: None,
base_pipeline: None,
vertex_input: Some(pipeline::VertexInputState {
@ -915,7 +914,6 @@ impl EguiState {
descriptor_layout,
descriptor_set: sets[0],
pipeline: Arc::new(pipeline),
pipeline_layout: Arc::new(pipeline_layout),
})
}

View file

@ -773,6 +773,9 @@ impl Pipeline {
pub fn bind_point(&self) -> vk::PipelineBindPoint {
self.bind_point
}
pub fn layout(&self) -> &Arc<PipelineLayout> {
&self.layout
}
}
pub(crate) mod pipeline_cache {

View file

@ -14,6 +14,7 @@
use crate::{
device::DeviceOwned,
pipeline::Pipeline,
render_graph::recorder::{CommandRecorder, SideEffectMap},
};
@ -318,18 +319,26 @@ impl Command for EndRendering {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct BindPipeline;
pub struct BindPipeline<'cmd>(pub &'cmd Pipeline);
impl Command for BindPipeline {
impl<'cmd> Command for BindPipeline<'cmd> {
fn side_effects(&self, _map: SideEffectMap) {
// No resource access, but affects the pipeline state
}
fn apply(self, _recorder: &mut CommandRecorder) {}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
unsafe {
dev.cmd_bind_pipeline(cmd.raw(), self.0.bind_point(), self.0.raw());
}
}
}
pub struct BindVertexBuffers {
pub buffers: Vec<BufferSlice>,
pub first_binding: u32,
}
impl Command for BindVertexBuffers {
@ -343,7 +352,26 @@ impl Command for BindVertexBuffers {
}
}
fn apply(self, _recorder: &mut CommandRecorder) {}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
let buffers = self
.buffers
.iter()
.map(|buffer| recorder.get_buffer_handle(buffer.id()))
.collect::<Vec<_>>();
let offsets = self
.buffers
.iter()
.map(|buffer| buffer.range.offset)
.collect::<Vec<_>>();
unsafe {
dev.cmd_bind_vertex_buffers(cmd.raw(), self.first_binding, &buffers, &offsets);
}
}
}
pub struct BindIndexBuffer(pub BufferSlice, pub IndexFormat);
@ -354,17 +382,50 @@ impl Command for BindIndexBuffer {
.side_effect(map.reborrow(), vk::AccessFlags2::INDEX_READ, None);
}
fn apply(self, _recorder: &mut CommandRecorder) {}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
let buffer = recorder.get_buffer_handle(self.0.id());
let index_type = match self.1 {
IndexFormat::Uint8 => vk::IndexType::UINT8_KHR,
IndexFormat::Uint16 => vk::IndexType::UINT16,
IndexFormat::Uint32 => vk::IndexType::UINT32,
};
unsafe {
dev.cmd_bind_index_buffer(cmd.raw(), buffer, self.0.range.offset, index_type);
}
}
}
pub struct BindDescriptorSets;
pub struct BindDescriptorSets {
pub first_set: u32,
pub pipeline_layout: vk::PipelineLayout,
pub bind_point: vk::PipelineBindPoint,
pub sets: Vec<vk::DescriptorSet>,
}
impl Command for BindDescriptorSets {
fn side_effects(&self, _map: SideEffectMap) {
// No resource access, but affects the descriptor set state
}
fn apply(self, _recorder: &mut CommandRecorder) {}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
unsafe {
dev.cmd_bind_descriptor_sets(
cmd.raw(),
self.bind_point,
self.pipeline_layout,
self.first_set,
&self.sets,
&[],
);
}
}
}
pub struct SetViewport {
@ -399,14 +460,32 @@ impl Command for SetScissor {
fn apply(self, _recorder: &mut CommandRecorder) {}
}
pub struct PushConstants;
pub struct PushConstants {
pub pipeline_layout: vk::PipelineLayout,
pub stage_flags: vk::ShaderStageFlags,
pub offset: u32,
pub data: Vec<u8>,
}
impl Command for PushConstants {
fn side_effects(&self, _map: SideEffectMap) {
// No resource access, but affects the push constant state
}
fn apply(self, _recorder: &mut CommandRecorder) {}
fn apply(self, recorder: &mut CommandRecorder) {
let cmd = recorder.cmd_buffer();
let dev = &cmd.device().raw;
unsafe {
dev.cmd_push_constants(
cmd.raw(),
self.pipeline_layout,
self.stage_flags,
self.offset,
&self.data,
);
}
}
}
pub struct DrawData {
@ -523,8 +602,8 @@ impl<T: Resource, U: Resource> OutsideRenderPass for Copy<T, U> {}
impl OutsideRenderPass for ClearTexture {}
impl OutsideRenderPass for UpdateBuffer {}
impl InsideRenderPass for BindPipeline {}
impl OutsideRenderPass for BindPipeline {}
impl InsideRenderPass for BindPipeline<'_> {}
impl OutsideRenderPass for BindPipeline<'_> {}
impl InsideRenderPass for BindVertexBuffers {}
impl OutsideRenderPass for BindVertexBuffers {}
impl InsideRenderPass for BindIndexBuffer {}

View file

@ -1,5 +1,6 @@
use std::{
collections::{BTreeMap, BTreeSet},
marker::PhantomData,
mem::{MaybeUninit, size_of},
ptr::NonNull,
};
@ -96,37 +97,40 @@ impl<'a> SideEffectMap<'a> {
}
}
pub struct CommandList {
pub struct CommandList<'cmd> {
command_bytes: Vec<MaybeUninit<u8>>,
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: SideEffects,
num_commands: u32,
_pd: PhantomData<fn(&'cmd ())>,
}
#[must_use]
pub struct RenderPass<'a> {
cmd_list: &'a mut CommandList,
pub struct RenderPass<'cmd, 'a> {
cmd_list: &'a mut CommandList<'cmd>,
}
impl<'a> RenderPass<'a> {
impl<'cmd, 'a> RenderPass<'cmd, 'a> {
pub fn finalise(self) {
self.cmd_list.push_inner(super::commands::EndRendering);
}
pub fn push<C: Command + InsideRenderPass>(&mut self, command: C) {
pub fn push<C: Command + InsideRenderPass + 'cmd>(&mut self, command: C) {
self.cmd_list.push_inner(command);
}
}
impl CommandList {
impl<'cmd> CommandList<'cmd> {
pub fn new() -> Self {
Self {
command_bytes: Vec::new(),
cursor: 0,
side_effects: SideEffects::default(),
num_commands: 0,
_pd: PhantomData,
}
}
@ -137,7 +141,7 @@ impl CommandList {
stencil_attachment: Option<TextureRegion>,
area: (u32, u32),
layers: u32,
) -> RenderPass<'_> {
) -> RenderPass<'cmd, '_> {
self.push(super::commands::BeginRendering {
color_attachments,
depth_attachment,
@ -148,7 +152,7 @@ impl CommandList {
RenderPass { cmd_list: self }
}
pub fn push<C: Command + OutsideRenderPass>(&mut self, command: C) {
pub fn push<C: Command + OutsideRenderPass + 'cmd>(&mut self, command: C) {
self.push_inner(command);
}

View file

@ -258,9 +258,6 @@ mod impls {
}
}
pub struct Pipeline;
pub struct DescriptorSet;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct ResourceId(pub u32);

View file

@ -26,7 +26,6 @@ pub struct Wireframe {
num_indices: u32,
colors: Arc<Buffer>,
pipeline: Arc<pipeline::Pipeline>,
layout: Arc<pipeline::PipelineLayout>,
}
impl Wireframe {
@ -168,7 +167,7 @@ impl Wireframe {
Arc::new(sync::Fence::from_pool(&dev.pools.fences, None)?),
)?;
let (pipeline, layout) = Self::create_pipeline(dev.clone())?;
let pipeline = Self::create_pipeline(dev.clone())?;
future.await;
@ -177,13 +176,12 @@ impl Wireframe {
indices: Arc::new(indices),
colors: Arc::new(colors),
pipeline: Arc::new(pipeline),
layout: Arc::new(layout),
num_indices,
})
}
fn create_pipeline(device: Device) -> Result<(pipeline::Pipeline, pipeline::PipelineLayout)> {
let pipeline_layout = pipeline::PipelineLayout::new(
fn create_pipeline(device: Device) -> Result<pipeline::Pipeline> {
let pipeline_layout = Arc::new(pipeline::PipelineLayout::new(
device.clone(),
pipeline::PipelineLayoutDesc {
descriptor_set_layouts: &[],
@ -194,7 +192,7 @@ impl Wireframe {
}],
name: Some("wireframe-pipeline-layout".into()),
},
)?;
)?);
let shader = pipeline::ShaderModule::new_from_path(
device.clone(),
@ -221,7 +219,7 @@ impl Wireframe {
},
],
render_pass: None,
layout: &pipeline_layout,
layout: pipeline_layout,
subpass: None,
base_pipeline: None,
vertex_input: Some(pipeline::VertexInputState {
@ -301,7 +299,7 @@ impl Wireframe {
},
)?;
Ok((pipeline, pipeline_layout))
Ok(pipeline)
}
pub fn pass(&self, rg: &mut RenderGraph, target: GraphResourceId) -> Result<()> {
@ -315,7 +313,6 @@ impl Wireframe {
let num_indices = self.num_indices;
let pipeline = self.pipeline.clone();
let layout = self.layout.clone();
move |ctx: &RenderContext| -> Result<()> {
let target = ctx.get_image(target).unwrap();
@ -361,7 +358,7 @@ impl Wireframe {
cmd.bind_indices(indices.raw(), 0, vk::IndexType::UINT32);
cmd.bind_vertex_buffers(&[positions.raw(), colors.raw()], &[0, 0]);
cmd.push_constants(
&layout,
&pipeline.layout(),
vk::ShaderStageFlags::VERTEX,
0,
bytemuck::cast_slice(&[to_clip]),