reflect: move spirv_passes out of naga

This commit is contained in:
chyyran 2024-02-18 15:16:16 -05:00 committed by Ronny Chan
parent b75a614873
commit 22aa59b598
6 changed files with 14 additions and 17 deletions

View file

@ -1,6 +1,7 @@
use crate::error::ShaderCompileError; use crate::error::ShaderCompileError;
use librashader_preprocess::ShaderSource; use librashader_preprocess::ShaderSource;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
pub(crate) mod spirv_passes;
mod glslang; mod glslang;

View file

@ -0,0 +1,10 @@
pub mod link_input_outputs;
pub mod lower_samplers;
// Load SPIR-V as an rspirv module
pub(crate) fn load_module(words: &[u32]) -> rspirv::dr::Module {
let mut loader = rspirv::dr::Loader::new();
rspirv::binary::parse_words(words, &mut loader).unwrap();
let module = loader.module();
module
}

View file

@ -1,6 +1,5 @@
pub mod msl; pub mod msl;
pub mod spirv; pub mod spirv;
mod spirv_passes;
pub mod wgsl; pub mod wgsl;
use crate::error::{SemanticsErrorKind, ShaderReflectError}; use crate::error::{SemanticsErrorKind, ShaderReflectError};
@ -14,8 +13,8 @@ use rspirv::binary::Assemble;
use rspirv::dr::Builder; use rspirv::dr::Builder;
use rustc_hash::FxHashSet; use rustc_hash::FxHashSet;
use crate::front::spirv_passes::lower_samplers;
use crate::reflect::helper::{SemanticErrorBlame, TextureData, UboData}; use crate::reflect::helper::{SemanticErrorBlame, TextureData, UboData};
use crate::reflect::naga::spirv_passes::{link_input_outputs, lower_samplers};
use crate::reflect::semantics::{ use crate::reflect::semantics::{
BindingMeta, BindingStage, BufferReflection, MemberOffset, ShaderSemantics, TextureBinding, BindingMeta, BindingStage, BufferReflection, MemberOffset, ShaderSemantics, TextureBinding,
TextureSemanticMap, TextureSemantics, TextureSizeMeta, TypeInfo, UniformMemberBlock, TextureSemanticMap, TextureSemantics, TextureSizeMeta, TypeInfo, UniformMemberBlock,
@ -115,13 +114,6 @@ impl TryFrom<&SpirvCompilation> for NagaReflect {
type Error = ShaderReflectError; type Error = ShaderReflectError;
fn try_from(compile: &SpirvCompilation) -> Result<Self, Self::Error> { fn try_from(compile: &SpirvCompilation) -> Result<Self, Self::Error> {
fn load_module(words: &[u32]) -> rspirv::dr::Module {
let mut loader = rspirv::dr::Loader::new();
rspirv::binary::parse_words(words, &mut loader).unwrap();
let module = loader.module();
module
}
fn lower_fragment_shader(builder: &mut Builder) { fn lower_fragment_shader(builder: &mut Builder) {
let mut pass = lower_samplers::LowerCombinedImageSamplerPass::new(builder); let mut pass = lower_samplers::LowerCombinedImageSamplerPass::new(builder);
pass.ensure_op_type_sampler(); pass.ensure_op_type_sampler();
@ -134,20 +126,16 @@ impl TryFrom<&SpirvCompilation> for NagaReflect {
block_ctx_dump_prefix: None, block_ctx_dump_prefix: None,
}; };
let vertex = load_module(&compile.vertex); let vertex = crate::front::spirv_passes::load_module(&compile.vertex);
let fragment = load_module(&compile.fragment); let fragment = crate::front::spirv_passes::load_module(&compile.fragment);
let mut fragment = Builder::new_from_module(fragment); let mut fragment = Builder::new_from_module(fragment);
lower_fragment_shader(&mut fragment); lower_fragment_shader(&mut fragment);
let mut pass = link_input_outputs::LinkInputs::new(&vertex, &mut fragment);
pass.do_pass();
let vertex = vertex.assemble(); let vertex = vertex.assemble();
let fragment = fragment.module().assemble(); let fragment = fragment.module().assemble();
let vertex = naga::front::spv::parse_u8_slice(bytemuck::cast_slice(&vertex), &options)?; let vertex = naga::front::spv::parse_u8_slice(bytemuck::cast_slice(&vertex), &options)?;
let fragment = naga::front::spv::parse_u8_slice(bytemuck::cast_slice(&fragment), &options)?; let fragment = naga::front::spv::parse_u8_slice(bytemuck::cast_slice(&fragment), &options)?;
Ok(NagaReflect { vertex, fragment }) Ok(NagaReflect { vertex, fragment })

View file

@ -1,2 +0,0 @@
pub mod link_input_outputs;
pub mod lower_samplers;