shader-sense 1.3.1

Library for runtime shader validation and symbol inspection
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use std::path::Path;

use crate::{
    position::ShaderRange,
    symbols::{
        symbol_parser::{get_name, ShaderSymbolListBuilder, SymbolTreeParser},
        symbols::{
            ShaderMember, ShaderParameter, ShaderScope, ShaderSymbol, ShaderSymbolData,
            ShaderSymbolMode, ShaderSymbolRuntime,
        },
    },
};

pub fn get_wgsl_parsers() -> Vec<Box<dyn SymbolTreeParser>> {
    vec![]
}

#[allow(dead_code)] // For now, dont pollute Wgsl as its not ready
struct WgslStructTreeParser {}

impl SymbolTreeParser for WgslStructTreeParser {
    fn get_query(&self) -> String {
        r#"(struct_declaration
            name: (identifier) @struct.type
            ((struct_member
                (variable_identifier_declaration
                    name: (identifier) @struct.param.type
                    type: (type_declaration) @struct.param.decl
                )
            )(",")?)*
        )"#
        .into()
    }
    fn process_match(
        &self,
        symbol_match: &tree_sitter::QueryMatch,
        file_path: &Path,
        shader_content: &str,
        scopes: &Vec<ShaderScope>,
        symbols: &mut ShaderSymbolListBuilder,
    ) {
        let label_node = symbol_match.captures[0].node;
        let range = ShaderRange::from(label_node.range());
        let scope_stack = self.compute_scope_stack(&scopes, &range);
        let struct_name: String = get_name(shader_content, symbol_match.captures[0].node).into();
        symbols.add_type(ShaderSymbol {
            label: struct_name.clone(),
            requirement: None,
            data: ShaderSymbolData::Struct {
                constructors: vec![], // Constructor in wgsl ?
                members: symbol_match.captures[1..]
                    .chunks(2)
                    .map(|w| ShaderMember {
                        context: struct_name.clone(),
                        parameters: ShaderParameter {
                            ty: get_name(shader_content, w[1].node).into(),
                            label: get_name(shader_content, w[0].node).into(),
                            count: None,
                            description: "".into(),
                            range: Some(ShaderRange::from(ShaderRange::from(w[0].node.range()))),
                        },
                    })
                    .collect(),
                methods: vec![],
            },
            // TODO: compute scope
            mode: ShaderSymbolMode::Runtime(ShaderSymbolRuntime::new(
                file_path.into(),
                range,
                None,
                scope_stack,
            )),
        });
    }
}