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

use crate::pso::probe::{Probe, ProbingPolicy};
use crate::pso::swarm::Swarm;

pub struct PolicyDrivenProbe {
    probe: Box<dyn Probe>,
    policy: Box<dyn ProbingPolicy>,
}

impl PolicyDrivenProbe {
    pub fn new(probe: Box<dyn Probe>, policy: Box<dyn ProbingPolicy>) -> PolicyDrivenProbe {
        PolicyDrivenProbe { probe, policy }
    }
}

impl Probe for PolicyDrivenProbe {
    fn on_begin(&mut self, swarm: &Swarm) {
        if self.policy.on_begin() {
            self.probe.on_begin(swarm);
        }
    }

    fn on_end(&mut self, swarm: &Swarm, generation: usize) {
        if self.policy.on_end(generation) {
            self.probe.on_new_generation(swarm, generation);
        }
        self.probe.on_end(swarm, generation);
    }

    fn on_new_generation(&mut self, swarm: &Swarm, generation: usize) {
        if self.policy.on_new_generation(generation) {
            self.probe.on_new_generation(swarm, generation);
        }
    }
}