franklin 0.1.4

A utility for generating images through evolution.
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

use crate::models::{Image, Pixel};

use super::CrossoverFunction;

/// This crossover function breeds specimens by taking the first half of the image from `first_image` and the second
/// half from `second_image`.
#[derive(Default)]
pub struct EqualHalfsCrossover;

impl CrossoverFunction for EqualHalfsCrossover {
    fn crossover(&mut self, first_image: &Image, second_image: &Image) -> Image {
        let size = first_image.pixels().len();

        let number_of_pixels_from_first_image = if size % 2 == 0 {
            size / 2
        } else {
            size / 2 + 1
        };

        let number_of_pixels_from_second_image = size / 2;

        let pixels = first_image
            .pixels()
            .iter()
            .take(number_of_pixels_from_first_image)
            .chain(
                second_image
                    .pixels()
                    .iter()
                    .skip(number_of_pixels_from_first_image)
                    .take(number_of_pixels_from_second_image),
            )
            .cloned()
            .collect::<Vec<Pixel>>();

        Image::new(first_image.height(), first_image.width(), pixels)
    }
}