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 crate::{
    generator::error_non_generator, handle_id_error, join_list_nf, GenOp, NormalForm, Normalize,
    Term,
};
use rand::SeedableRng;
use scop::Defs;
use weresocool_error::Error;

impl Normalize<Term> for GenOp {
    fn apply_to_normal_form(
        &self,
        input: &mut NormalForm,
        defs: &mut Defs<Term>,
    ) -> Result<(), Error> {
        match self {
            GenOp::Named { name, seed } => {
                let term = handle_id_error(name, defs)?;
                match term {
                    Term::Gen(gen) => {
                        gen.to_owned().set_seed(*seed);
                        gen.apply_to_normal_form(input, defs)
                    }
                    _ => Err(error_non_generator()),
                }
            }
            GenOp::Const { gen, seed } => {
                *input = join_list_nf(gen.to_owned().generate(
                    input,
                    gen.lcm_length(),
                    defs,
                    &mut SeedableRng::seed_from_u64(*seed),
                )?);
                Ok(())
            }
            GenOp::Taken { n, gen, seed } => {
                gen.to_owned().set_seed(*seed);
                *input = join_list_nf(gen.to_owned().generate_from_genop(input, Some(*n), defs)?);
                Ok(())
            }
        }
    }
}