ecrs 0.1.0-beta.4

Evolutionary computation tools & algorithms
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

use crate::pso::probe::Probe;
use crate::pso::swarm::Swarm;

pub struct AggregatedProbe {
    probes: Vec<Box<dyn Probe>>,
}

impl AggregatedProbe {
    pub fn new() -> AggregatedProbe {
        AggregatedProbe { probes: vec![] }
    }

    pub fn from_probes(probes: Vec<Box<dyn Probe>>) -> AggregatedProbe {
        AggregatedProbe { probes }
    }

    pub fn add_probe<Pr: Probe + 'static>(mut self, probe: Pr) -> AggregatedProbe {
        self.probes.push(Box::new(probe));
        self
    }
}

impl Probe for AggregatedProbe {
    fn on_begin(&mut self, swarm: &Swarm) {
        for probe in self.probes.iter_mut() {
            probe.on_begin(swarm);
        }
    }

    fn on_end(&mut self, swarm: &Swarm, generation: usize) {
        for probe in self.probes.iter_mut() {
            probe.on_end(swarm, generation);
        }
    }

    fn on_new_generation(&mut self, swarm: &Swarm, generation: usize) {
        for probe in self.probes.iter_mut() {
            probe.on_new_generation(swarm, generation);
        }
    }
}