librashader/librashader-reflect/src/reflect/mod.rs

use crate::error::{SemanticsErrorKind, ShaderReflectError};
use crate::reflect::semantics::{SemanticMap, ShaderReflection, TextureSizeMeta, TextureSemantics, VariableMeta, VariableSemantics, TextureImage};
use rustc_hash::FxHashMap;

mod cross;
mod naga;
mod rspirv;
pub mod semantics;

pub trait ReflectShader {
    fn reflect(&self, options: &ReflectOptions) -> Result<ShaderReflection, ShaderReflectError>;
}

#[derive(Debug)]
pub enum UniformSemantic {
    Variable(SemanticMap<VariableSemantics>),
    Texture(SemanticMap<TextureSemantics>),
}

#[derive(Debug)]
pub struct ReflectOptions {
    pub pass_number: u32,
    pub uniform_semantics: FxHashMap<String, UniformSemantic>,
    pub non_uniform_semantics: FxHashMap<String, SemanticMap<TextureSemantics>>,
}

#[derive(Debug, Default)]
pub struct ReflectMeta {
    pub parameter_meta: FxHashMap<u32, VariableMeta>,
    pub variable_meta: FxHashMap<VariableSemantics, VariableMeta>,
    pub texture_meta: FxHashMap<SemanticMap<TextureSemantics>, TextureImage>,
    pub texture_size_meta: FxHashMap<SemanticMap<TextureSemantics>, TextureSizeMeta>,
}

pub fn builtin_uniform_semantics() -> FxHashMap<String, UniformSemantic> {
    let mut map = FxHashMap::default();

    map.insert(
        "MVP".into(),
        UniformSemantic::Variable(SemanticMap {
            semantics: VariableSemantics::MVP,
            index: 0,
        }),
    );

    map.insert(
        "OutputSize".into(),
        UniformSemantic::Variable(SemanticMap {
            semantics: VariableSemantics::Output,
            index: 0,
        }),
    );

    map.insert(
        "FinalViewportSize".into(),
        UniformSemantic::Variable(SemanticMap {
            semantics: VariableSemantics::FinalViewport,
            index: 0,
        }),
    );

    map.insert(
        "FrameCount".into(),
        UniformSemantic::Variable(SemanticMap {
            semantics: VariableSemantics::FrameCount,
            index: 0,
        }),
    );

    map.insert(
        "FrameDirection".into(),
        UniformSemantic::Variable(SemanticMap {
            semantics: VariableSemantics::FrameDirection,
            index: 0,
        }),
    );
    map
}