blaze-proc 1.0.0

Blaze internal/external proc macros
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

use derive_syn_parse::Parse;
use proc_macro2::{Ident};
use quote::{format_ident};
use syn::{Generics, Token, parse_quote_spanned, spanned::Spanned};
use super::{Type};

/*
    kernel void add (const ulong n, __global const float* rhs, __global const float* in, __global float* out) {
        for (ulong id = get_global_id(0); id<n; id += get_global_size(0)) {
            int two = (int)in[id];
            out[id] = in[id] + rhs[id];
        }
    }
*/

#[derive(Debug, Parse)]
pub struct Argument {
    pub mutability : Option<Token![mut]>,
    pub name: Ident,
    pub semi_token: Token![:],
    pub ty: Type,
}

impl Argument {
    pub fn ty (&self, generics: &mut Generics, lt: bool) -> syn::Type {
        let name = format_ident!("{}", self.name.to_string().to_uppercase());
        let (mutability, generify, ty) = self.ty.rustify(self.mutability.is_some(), &name);

        if let Some(imp) = generify {
            generics.params.push(imp);
            
            if lt {
                return match mutability {
                    true => parse_quote_spanned! { ty.span() => &'__env__ mut #ty },
                    false => parse_quote_spanned! { ty.span() => &'__env__ #ty }
                }
            }

            return match mutability {
                true => parse_quote_spanned! { ty.span() => &mut #ty },
                false => parse_quote_spanned! { ty.span() => &#ty }
            }
        }

        ty
    }
}