wesl 0.3.2

The WESL compiler
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

use std::collections::HashMap;

use itertools::Itertools;
use wgsl_parse::syntax::{
    Function, GlobalDeclaration, Ident, Import, TranslationUnit, TypeExpression,
};

use crate::{
    import::{absolute_resource, imported_resources, normalize_resource},
    visit::Visit,
    Mangler, Resource,
};

fn normalize_ty(
    ty: &TypeExpression,
    parent_res: &Resource,
    imports: &HashMap<Ident, (Resource, Ident)>,
) -> TypeExpression {
    let mut ty = ty.clone();
    if let Some(path) = &ty.path {
        let res = normalize_resource(path, parent_res, imports);
        ty.path = Some(res.path().to_path_buf());
    } else {
        ty.path = Some(parent_res.path().to_path_buf());
    }
    ty
}

pub(crate) fn mangle(wesl: &mut TranslationUnit, resource: &Resource, mangler: &impl Mangler) {
    let imports = imported_resources(&wesl.imports, resource);

    for decl in &mut wesl.global_declarations {
        if let GlobalDeclaration::Function(decl) = decl {
            // TODO: type aliases
            let sig = decl
                .parameters
                .iter()
                .map(|p| normalize_ty(&p.ty, resource, &imports)).collect_vec()
            let res = mangler.mangle_signature(decl.ident.name().as_str(), &sig);
        }
    }
}