data_bucket_derive 0.3.16

DataBucket is container for WorkTable
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

use proc_macro2::TokenStream;
use quote::quote;
use syn::ItemStruct;

pub struct Generator {
    pub struct_def: ItemStruct,
}

impl Generator {
    pub fn gen_impl(&self) -> TokenStream {
        let struct_ident = &self.struct_def.ident;

        let mut num = 0;
        let sum = self
            .struct_def
            .fields
            .iter()
            .map(|f| {
                if let Some(i) = &f.ident {
                    quote! {
                        self.#i.aligned_size()
                    }
                } else {
                    let i = syn::Index::from(num);
                    num += 1;
                    quote! {
                        self.#i.aligned_size()
                    }
                }
            })
            .collect::<Vec<_>>();

        let align_check = self
            .struct_def
            .fields
            .iter()
            .map(|f| {
                let t = &f.ty;
                quote! {
                    if #t::align() == Some(8) {
                        return Some(8)
                    }
                }
            })
            .collect::<Vec<_>>();

        quote! {
            impl SizeMeasurable for #struct_ident {
                fn aligned_size(&self) -> usize {
                    let len = #(#sum+)* 0;
                    align(len)
                }
                fn align() -> Option<usize> {
                    #(#align_check)*
                    None
                }
            }
        }
    }
}