optirustic 1.2.3

This crate moved to https://github.com/s-simoncelli/nsga-rs
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
94
95
96
97
98
99
100
101
102
103
104
105

pub use distance::Distance;
pub use hv_wfg::HyperVolumeWhile2012;
pub use hypervolume::{AllHyperVolumeFileData, HyperVolume, HyperVolumeFileData};
pub use hypervolume_2d::HyperVolume2D;
pub use hypervolume_fonseca_2006::HyperVolumeFonseca2006;

mod distance;
mod hv_wfg;
mod hypervolume;
mod hypervolume_2d;
mod hypervolume_fonseca_2006;

#[cfg(test)]
pub(crate) mod test_utils {
    use std::env;
    use std::error::Error;
    use std::fs::read_to_string;
    use std::path::Path;

    #[derive(Debug)]
    /// Pagmo's test data.
    pub(crate) struct TestData {
        /// The reference point to use in the test.
        pub(crate) reference_point: Vec<f64>,
        /// The objective values to use in the test.
        pub(crate) objective_values: Vec<Vec<f64>>,
        /// The expected hyper-volume for the front.
        pub(crate) hyper_volume: f64,
    }

    /// Parse a fle containing Pagmo test data.
    ///
    /// return: `Vec<(Vec<Vec<f64>>, Vec<f64>)>` A vector with a set of tests.
    pub(crate) fn parse_pagmo_test_data_file(
        filename: &str,
    ) -> Result<Vec<TestData>, Box<dyn Error>> {
        let test_path = Path::new(&env::current_dir().unwrap())
            .join("src")
            .join("metrics")
            .join("test_data");

        // data for one test
        let mut reference_point: Vec<f64> = vec![];
        let mut objective_values: Vec<Vec<f64>> = vec![];
        let mut hyper_volume: f64;

        let mut all_test_data: Vec<TestData> = vec![];
        let mut data_line: usize = 0;
        let mut total_points: usize = 0;

        for (li, line) in read_to_string(test_path.join(filename))?
            .lines()
            .enumerate()
        {
            if li == 0 {
                continue;
            }

            // skip number of objectives
            if data_line == 0 {
                data_line += 1;
                continue;
            } else if data_line == 1 {
                total_points = line.to_string().parse::<usize>()?;
                data_line += 1;
                continue;
            } else if data_line == 2 {
                reference_point = line
                    .split(' ')
                    .map(|v| v.to_string().parse::<f64>())
                    .collect::<Result<Vec<f64>, _>>()?;
                data_line += 1;
                continue;
            } else if total_points + 3 > data_line {
                // collect objectives
                let point = line
                    .split(' ')
                    .map(|v| v.to_string().parse::<f64>())
                    .collect::<Result<Vec<f64>, _>>()?;
                objective_values.push(point);
                data_line += 1;
                continue;
            } else if total_points + 3 == data_line {
                // get expected hyper-volume
                hyper_volume = line.to_string().parse::<f64>()?;

                // collect test data
                all_test_data.push(TestData {
                    reference_point,
                    objective_values: objective_values.clone(),
                    hyper_volume,
                });

                // reset
                data_line = 0;
                total_points = 0;
                reference_point = vec![];
                objective_values = vec![];
                continue;
            }
        }

        Ok(all_test_data)
    }
}