hpt-macros 0.1.2

An internal library for generating helper functions for hpt
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 proc_macro2::TokenStream;

use crate::type_utils::TypeInfo;

pub(crate) fn __impl_into_scalar() -> TokenStream {
    let mut ret = proc_macro2::TokenStream::new();

    let types = [
        "bool",
        "bf16",
        "f16",
        "f32",
        "f64",
        "i8",
        "i16",
        "i32",
        "i64",
        "u8",
        "u16",
        "u32",
        "u64",
        "isize",
        "usize",
        "complex32",
        "complex64",
    ];

    for lhs in types.iter() {
        for rhs in types.iter() {
            let lhs_type = TypeInfo::new(lhs);
            let lhs_dtype = lhs_type.dtype;
            let rhs_type = TypeInfo::new(rhs);
            let rhs_dtype = rhs_type.dtype;
            let into_method =
                syn::Ident::new(&format!("to_{}", rhs), proc_macro2::Span::call_site());
            ret.extend(quote::quote! {
                impl Cast<#rhs_dtype> for #lhs_dtype {
                    fn cast(self) -> #rhs_dtype {
                        self.#into_method()
                    }
                }
            });
        }
    }

    ret.into()
}