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
pub use super::mass_degeneration::MassDegeneration as ExtensionMassDegeneration;
pub use super::mass_extinction::MassExtinction as ExtensionMassExtinction;
pub use super::mass_genesis::MassGenesis as ExtensionMassGenesis;
pub use super::noop::Noop as ExtensionNoop;
pub use super::Extension;

use crate::genotype::EvolveGenotype;
use crate::strategy::evolve::{EvolveConfig, EvolveState};
use crate::strategy::StrategyReporter;
use rand::Rng;

#[derive(Clone, Debug)]
pub enum Wrapper {
    Noop(ExtensionNoop),
    MassExtinction(ExtensionMassExtinction),
    MassGenesis(ExtensionMassGenesis),
    MassDegeneration(ExtensionMassDegeneration),
}

impl Extension for Wrapper {
    fn call<G: EvolveGenotype, R: Rng, SR: StrategyReporter<Genotype = G>>(
        &mut self,
        genotype: &mut G,
        state: &mut EvolveState<G>,
        config: &EvolveConfig,
        reporter: &mut SR,
        rng: &mut R,
    ) {
        match self {
            Wrapper::Noop(extension) => extension.call(genotype, state, config, reporter, rng),
            Wrapper::MassExtinction(extension) => {
                extension.call(genotype, state, config, reporter, rng)
            }
            Wrapper::MassGenesis(extension) => {
                extension.call(genotype, state, config, reporter, rng)
            }
            Wrapper::MassDegeneration(extension) => {
                extension.call(genotype, state, config, reporter, rng)
            }
        }
    }
}

impl From<ExtensionNoop> for Wrapper {
    fn from(extension: ExtensionNoop) -> Self {
        Wrapper::Noop(extension)
    }
}
impl From<ExtensionMassExtinction> for Wrapper {
    fn from(extension: ExtensionMassExtinction) -> Self {
        Wrapper::MassExtinction(extension)
    }
}
impl From<ExtensionMassGenesis> for Wrapper {
    fn from(extension: ExtensionMassGenesis) -> Self {
        Wrapper::MassGenesis(extension)
    }
}
impl From<ExtensionMassDegeneration> for Wrapper {
    fn from(extension: ExtensionMassDegeneration) -> Self {
        Wrapper::MassDegeneration(extension)
    }
}