radiate-engines 1.3.0

Engines for the Radiate genetic algorithm library.
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55

#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use std::io;
use std::path::PathBuf;

pub trait FileWriter<T> {
    fn extension(&self) -> &str;
    fn write(&mut self, path: PathBuf, generation: &T) -> io::Result<()>;
}

pub trait FileReader<T> {
    fn read(&self, path: PathBuf) -> io::Result<T>;
}

pub struct JsonWriter;

#[cfg(feature = "serde")]
impl<T> FileWriter<T> for JsonWriter
where
    T: Serialize,
{
    fn extension(&self) -> &str {
        "json"
    }

    fn write(&mut self, path: PathBuf, generation: &T) -> io::Result<()> {
        if !path.parent().is_none_or(|p| p.exists()) {
            std::fs::create_dir_all(path.parent().unwrap()).map_err(|e| {
                io::Error::other(format!("Failed to create checkpoint directory: {}", e))
            })?;
        }

        let json = serde_json::to_string(generation)
            .map_err(|e| io::Error::other(format!("Failed to serialize checkpoint file: {}", e)))?;

        std::fs::write(path, json)
    }
}

pub struct JsonReader;

#[cfg(feature = "serde")]
impl<T> FileReader<T> for JsonReader
where
    T: for<'de> Deserialize<'de>,
{
    fn read(&self, path: PathBuf) -> io::Result<T> {
        let json = std::fs::read_to_string(path)?;
        let generation = serde_json::from_str(&json).map_err(|e| {
            io::Error::other(format!("Failed to deserialize checkpoint file: {}", e))
        })?;

        Ok(generation)
    }
}