~ruther/ctu-fee-eoa

ref: d8580bdc2c159bd7c925cfecb4a9d53921288722 ctu-fee-eoa/env/src/initializer/mod.rs -rw-r--r-- 2.3 KiB
d8580bdc — Rutherther feat: add evolution algorithm a month ago 
                                                                                
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use nalgebra::{allocator::Allocator, DefaultAllocator, Dim, OVector, Scalar, U1};
use rand::{rng, RngCore};

use crate::{binary_string::BinaryString, bounded::{Bounded, BoundedBinaryString}};

pub trait Initializer<D: Dim, T> {
    fn initialize_single(&mut self, size: D) -> T;
    fn initialize(&mut self, size: D, count: usize) -> impl Iterator<Item = T> {
        let size = size;
        (0..count).map(move |_| self.initialize_single(size))
    }
}

// Always initializes with zeros
pub struct ZeroInitializer;

impl ZeroInitializer {
    pub fn new() -> Self {
        Self {
        }
    }
}

impl<D> Initializer<D, BinaryString<D>> for ZeroInitializer
where
    D: Dim,
    DefaultAllocator: Allocator<D>
{
    fn initialize_single(&mut self, size: D) -> BinaryString<D> {
        BinaryString::<D>::from_ovector(
            <Self as Initializer<D, OVector<i8, D>>>::initialize_single(self, size)
        )
    }
}

impl<D, T> Initializer<D, OVector<T, D>> for ZeroInitializer
where
    T: Scalar + Default,
    D: Dim,
    DefaultAllocator: Allocator<D>
{
    fn initialize_single(&mut self, size: D) -> OVector<T, D> {
        OVector::<T, D>::from_element_generic(size, U1, Default::default())
    }
}

pub struct RandomInitializer<D: Dim, T> {
    rng: Box<dyn RngCore>,
    bounded: Box<dyn Bounded<D, Item = T>>
}

impl<T, D: Dim> RandomInitializer<D, T> {
    pub fn new(bounded: Box<dyn Bounded<D, Item = T>>) -> Self {
        Self {
            rng: Box::new(rand::rng()),
            bounded
        }
    }
}

impl<D: Dim> RandomInitializer<D, BinaryString<D>>
where
    D: Dim,
    DefaultAllocator: Allocator<D>
{
    pub fn new_binary() -> Self {
        Self {
            rng: Box::new(rand::rng()),
            bounded: Box::new(BoundedBinaryString::unbounded())
        }
    }
}

impl<D> Initializer<D, BinaryString<D>> for RandomInitializer<D, BinaryString<D>>
where
    D: Dim,
    DefaultAllocator: Allocator<D>
{
    fn initialize_single(&mut self, size: D) -> BinaryString<D> {
        self.bounded.next_random(size, &mut self.rng)
    }
}

impl<D, T> Initializer<D, OVector<T, D>> for RandomInitializer<D, OVector<T, D>>
where
    T: Scalar + Default,
    D: Dim,
    DefaultAllocator: Allocator<D>
{
    fn initialize_single(&mut self, size: D) -> OVector<T, D> {
        self.bounded.next_random(size, &mut self.rng)
    }
}