random-constructible-derive 0.13.0

Provides a derive macro for the random-constructible crate which is used for creating random instances of enums with weighted probabilities
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

// ---------------- [ File: random-constructible-derive/src/collect_tuple_field_context.rs ]
crate::ix!();

#[derive(Builder,Getters)]
#[getset(get = "pub")]
#[builder(setter(into))]
pub struct TupleContext {
    provider_types:    Vec<Type>,
    rand_bounds:       Vec<TokenStream2>,
    inits_random:      Vec<TokenStream2>,
    inits_uniform:     Vec<TokenStream2>,

    #[cfg(feature="env")]
    inits_random_env:  Vec<TokenStream2>,

    #[cfg(feature="env")]
    inits_uniform_env: Vec<TokenStream2>,

    field_types:       Vec<Type>,
}

pub fn collect_tuple_field_context(fields: &FieldsUnnamed) -> TupleContext {
    let mut ctx = TupleContext {
        provider_types:    Vec::new(),
        rand_bounds:       Vec::new(),
        inits_random:      Vec::new(),
        inits_uniform:     Vec::new(),

        #[cfg(feature="env")]
        inits_random_env:  Vec::new(),

        #[cfg(feature="env")]
        inits_uniform_env: Vec::new(),

        field_types:       Vec::new(),
    };

    for field in &fields.unnamed {
        let spec   = ParsedFieldSpec::from_syn_field(field);
        let tokens = spec.build_generation_tokens();

        append_ts(&mut ctx.inits_random,      tokens.random());
        append_ts(&mut ctx.inits_uniform,     tokens.uniform());

        #[cfg(feature="env")]
        append_ts(&mut ctx.inits_random_env,  tokens.random_env());

        #[cfg(feature="env")]
        append_ts(&mut ctx.inits_uniform_env, tokens.uniform_env());

        ctx.provider_types.extend(tokens.provider_types().iter().cloned());
        ctx.rand_bounds.push(tokens.rand_bound().clone());
        ctx.field_types.push(spec.ty().clone());
    }

    ctx
}

#[cfg(not(feature="env"))]
pub fn generate_env_helpers_tuple(
    name:     &Ident,
    generics: &Generics,
    c:        &TupleContext,

) -> TokenStream2 {
    quote! { }
}

#[cfg(feature="env")]
pub fn generate_env_helpers_tuple(
    name:     &Ident,
    generics: &Generics,
    c:        &TupleContext,

) -> TokenStream2 {

    if provider_types_contain_primitive(c.provider_types()) {
        return quote!{};
    }

    let TupleContext {
        provider_types,
        inits_random_env,
        inits_uniform_env,
        ..
    } = c;

    quote! {
        impl #generics #name #generics {
            pub fn random_with_env<ENV>() -> Self
            where #( ENV : RandConstructProbabilityMapProvider<#provider_types>, )* {
                Self( #( #inits_random_env ),* )
            }
            pub fn random_uniform_with_env<ENV>() -> Self
            where #( ENV : RandConstructProbabilityMapProvider<#provider_types>, )* {
                Self( #( #inits_uniform_env ),* )
            }
        }
    }
}

pub fn append_ts(target: &mut Vec<TokenStream2>, ts: &TokenStream2) {
    target.push(ts.clone());
}

pub fn generate_rand_impl_tuple(
    name: &Ident,
    generics: &Generics,
    c: &TupleContext,
) -> TokenStream2 {
    let TupleContext {
        rand_bounds,
        inits_random,
        inits_uniform,
        field_types,
        ..
    } = c;

    quote! {
        impl #generics RandConstruct for #name #generics
        where #(#rand_bounds,)* {
            fn random()  -> Self { Self( #( #inits_random ),* ) }
            fn uniform() -> Self { Self( #( #inits_uniform ),* ) }
            fn random_with_rng<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
                Self( #( <#field_types as RandConstruct>::random_with_rng(rng) ),* )
            }
        }
    }
}