mitchellhansen/shade_runner
1
0
Fork 0
mirror of synced 2025-11-09 21:07:12 +00:00
Code Issues Releases Wiki Activity

descriptor layout

Browse Source
This commit is contained in:
Tom Gowan
2019-04-26 15:24:58 +10:00
parent 0755b4f807
commit f3346e48bb
6 changed files with 278 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/layouts.rs
View File

@@ -57,8 +57,16 @@ unsafe impl PipelineLayoutDesc for FragLayout {
fn num_bindings_in_set(&self, set: usize) -> Option<usize> {
self.layout_data.num_bindings.get(&set).map(|&i| i)
}
fn descriptor(&self, _set: usize, _binding: usize) -> Option<DescriptorDesc> {
None
fn descriptor(&self, set: usize, binding: usize) -> Option<DescriptorDesc> {
self.layout_data.descriptions.get(&set)
.and_then(|s|s.get(&binding))
.map(|desc| {
let mut desc = desc.clone();
dbg!(&self.stages);
desc.stages = self.stages.clone();
desc
})
}
fn num_push_constants_ranges(&self) -> usize {
0

58
src/reflection.rs
View File

@@ -1,8 +1,8 @@
use crate::sr;
use crate::srvk::SpirvTy;
use crate::srvk::{SpirvTy,DescriptorDescInfo};
use std::borrow::Cow;
use crate::vk::pipeline::shader::ShaderInterfaceDefEntry;
use crate::vk::descriptor::descriptor::ShaderStages;
use crate::vk::descriptor::descriptor::*;
use std::collections::HashMap;
use crate::CompiledShaders;
use crate::layouts::*;
@@ -16,11 +16,13 @@ pub struct ShaderInterfaces {
pub struct LayoutData {
pub num_sets: usize,
pub num_bindings: HashMap<usize, usize>,
pub descriptions: HashMap<usize, HashMap<usize, DescriptorDesc>>,
}
pub fn create_entry(shaders: &CompiledShaders) -> Entry {
let vertex_interfaces = create_interfaces(&shaders.vertex);
let fragment_interfaces = create_interfaces(&shaders.fragment);
let fragment_layout = create_layouts(&shaders.fragment);
let frag_input = FragInput{ inputs: fragment_interfaces.inputs };
let frag_output = FragOutput{ outputs: fragment_interfaces.outputs };
let frag_layout = FragLayout {
@@ -28,7 +30,7 @@ pub fn create_entry(shaders: &CompiledShaders) -> Entry {
fragment: true,
..ShaderStages::none()
},
layout_data: Default::default(),
layout_data: fragment_layout,
};
let vert_input = VertInput{ inputs: vertex_interfaces.inputs };
let vert_output = VertOutput{ outputs: vertex_interfaces.outputs };
@@ -88,3 +90,53 @@ fn create_interfaces(data: &[u32]) -> ShaderInterfaces {
})
.expect("failed to load module")
}
fn create_layouts(data: &[u32]) -> LayoutData {
sr::ShaderModule::load_u32_data(data)
.map(|m| {
m.enumerate_descriptor_sets(None)
.map(|sets| {
let num_sets = sets.len();
let num_bindings = sets
.iter()
.map(|i| {
dbg!(&i);
(i.set as usize, i.bindings.len())
})
.collect::<HashMap<usize, usize>>();
let descriptions = sets
.iter()
.map(|i| {
let desc = i.bindings.iter()
.map(|b| {
let info = DescriptorDescInfo{
descriptor_type: b.descriptor_type,
image: b.image,
};
let ty = SpirvTy::<DescriptorDescTy>::from(info).inner();
let stages = ShaderStages::none();
let d = DescriptorDesc {
ty,
array_count: b.count,
stages,
// TODO this is what vulkan_shaders does but I don't think
// it's correct
readonly: true,
};
(b.binding as usize, d)
})
.collect::<HashMap<usize, DescriptorDesc>>();
(i.set as usize, desc)
})
.collect::<HashMap<usize, HashMap<usize, DescriptorDesc>>>();
LayoutData {
num_sets,
num_bindings,
descriptions,
}
})
.expect("Failed to pass outputs")
})
.expect("failed to load module")
}

67
src/srvk.rs
View File

@@ -1,7 +1,6 @@
use crate::sr;
use crate::vk;
use vk::descriptor::descriptor::*;
use vk::pipeline::shader::ShaderInterfaceDefEntry;
use vk::format::Format;
pub struct SpirvTy<T> {
@@ -9,8 +8,8 @@ pub struct SpirvTy<T> {
}
pub struct DescriptorDescInfo {
descriptor_type: sr::types::ReflectDescriptorType,
image: sr::types::ReflectImageTraits,
pub descriptor_type: sr::types::ReflectDescriptorType,
pub image: sr::types::ReflectImageTraits,
}
impl<T> SpirvTy<T> {
@@ -56,7 +55,9 @@ impl From<sr::types::ReflectImageTraits> for SpirvTy<DescriptorImageDesc> {
let t = DescriptorImageDesc {
sampled: d.sampled != 0,
dimensions: SpirvTy::from(d.dim).inner(),
format: Some(SpirvTy::from(d.image_format).inner()),
// TODO figure out how to do format correctly
//format: Some(SpirvTy::from(d.image_format).inner()),
format: None,
multisampled: d.ms != 0,
array_layers: conv_array_layers(d.arrayed, d.depth),
};
@@ -66,13 +67,67 @@ impl From<sr::types::ReflectImageTraits> for SpirvTy<DescriptorImageDesc> {
impl From<sr::types::variable::ReflectDimension> for SpirvTy<DescriptorImageDescDimensions> {
fn from(d: sr::types::variable::ReflectDimension) -> Self {
unimplemented!()
use sr::types::variable::ReflectDimension::*;
use DescriptorImageDescDimensions::*;
let inner = match d {
Type1d => OneDimensional,
Type2d => TwoDimensional,
Type3d => ThreeDimensional,
sr::types::variable::ReflectDimension::Cube => DescriptorImageDescDimensions::Cube,
_ => unimplemented!(),
};
SpirvTy{ inner }
}
}
// I think this is wrong and currently is unused
impl From<sr::types::image::ReflectImageFormat> for SpirvTy<Format> {
fn from(d: sr::types::image::ReflectImageFormat) -> Self {
unimplemented!()
use sr::types::image::ReflectImageFormat::*;
use Format::*;
let inner = match d {
Undefined => unimplemented!(),
RGBA32_FLOAT => R32G32B32A32Sfloat,
RGBA16_FLOAT => R16G16B16A16Sfloat,
R32_FLOAT => R32Sfloat,
RGBA8 => unimplemented!(),
RGBA8_SNORM => R8G8B8A8Snorm,
RG32_FLOAT => R32G32Sfloat,
RG16_FLOAT => R16G16Sfloat,
R11G11B10_FLOAT => unimplemented!(),
R16_FLOAT => R16Sfloat,
RGBA16 => unimplemented!(),
RGB10A2 => unimplemented!(),
RG16 => unimplemented!(),
RG8 => unimplemented!(),
R16 => unimplemented!(),
R8 => unimplemented!(),
RGBA16_SNORM => R16G16B16A16Snorm,
RG16_SNORM => R16G16Snorm,
RG8_SNORM => R8G8Snorm,
R16_SNORM => R16Snorm,
R8_SNORM => R8Snorm,
RGBA32_INT => R32G32B32A32Sint,
RGBA16_INT => R16G16B16A16Sint,
RGBA8_INT => R8G8B8A8Sint,
R32_INT => R32Sint,
RG32_INT => R32G32Sint,
RG16_INT => R16G16Sint,
RG8_INT => R8G8Sint,
R16_INT => R16Sint,
R8_INT => R8Sint,
RGBA32_UINT => R32G32B32A32Uint,
RGBA16_UINT => R16G16B16A16Uint,
RGBA8_UINT => R8G8B8A8Uint,
R32_UINT => R32Uint,
RGB10A2_UINT => A2R10G10B10UintPack32,
RG32_UINT => R32G32Uint,
RG16_UINT => R16G16Uint,
RG8_UINT => R8G8Uint,
R16_UINT =>R16Uint,
R8_UINT =>R8Uint,
};
SpirvTy{ inner }
}
}