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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

// Traits
pub use crate::traits::{Configurable, Plotable, Saveable};
use core::fmt::Display;
use core::ops::Add;

// Structs
use average::Variance;

pub mod comparison;

pub use comparison::ProcessErrors;

/// Indexed collection of values with a given error.
///
/// # Examples
///
/// Quick plot.
/// ```no_run
/// use preexplorer::prelude::*;
/// let domain = (0..10).map(|i: i32| i.pow(2));
/// let image = domain.clone().map(|i| i..10 + i);
/// pre::ProcessError::new(domain, image).plot("my_identifier").unwrap();
/// ```
///
/// Compare ``ProcessError`` structs.
/// ```no_run
/// use preexplorer::prelude::*;
/// pre::ProcessErrors::new(vec![
///     pre::ProcessError::new((5..15), (0..10).map(|i| i..10 + i)),
///     pre::ProcessError::new((5..15), (0..10).map(|i| i..20 + i)),
///     ])
///     .plot("my_identifier").unwrap();
/// ```
#[derive(Debug, PartialEq, Clone)]
#[cfg_attr(feature = "use-serde", derive(serde::Serialize, serde::Deserialize))]
pub struct ProcessError<T>
where
    T: Display + Clone,
{
    domain: Vec<T>,
    image: Vec<(f64, f64)>,
    config: crate::configuration::Configuration,
}

impl<T> ProcessError<T>
where
    T: Display + Clone,
{
    /// Constructs a new ``ProcessError<T>``.
    ///
    /// # Examples
    ///
    /// From a complicated computation.
    /// ```
    /// use preexplorer::prelude::*;
    /// let data = (0..10).map(|i| i..10 + i);
    /// let pro_err = pre::ProcessError::new((0..10), data);
    /// ```
    pub fn new<I, J, K, S>(domain: I, image: J) -> ProcessError<T>
    where
        I: IntoIterator<Item = T>,
        J: IntoIterator<Item = K>,
        K: IntoIterator<Item = S>,
        S: Into<f64>,
    {
        let domain: Vec<T> = domain.into_iter().collect();
        let image: Vec<(f64, f64)> = image
            .into_iter()
            .map(|k| {
                let v: Variance = k.into_iter().map(|s| s.into()).collect();
                (v.mean(), v.error())
            })
            .collect();
        let config = crate::configuration::Configuration::default();

        ProcessError {
            domain,
            image,
            config,
        }
    }
}

impl<T> Add for ProcessError<T>
where
    T: Display + Clone,
{
    type Output = crate::ProcessErrors<T>;

    fn add(self, other: crate::ProcessError<T>) -> crate::ProcessErrors<T> {
        let mut cmp = self.into();
        cmp += other;
        cmp
    }
}

impl<T> Configurable for ProcessError<T>
where
    T: Display + Clone,
{
    fn configuration_mut(&mut self) -> &mut crate::configuration::Configuration {
        &mut self.config
    }
    fn configuration(&self) -> &crate::configuration::Configuration {
        &self.config
    }
}

impl<T> Saveable for ProcessError<T>
where
    T: Display + Clone,
{
    fn plotable_data(&self) -> String {
        // Initial warning
        if self.domain.is_empty() {
            eprintln!("Warning: There is no data.");
        }

        let mut raw_data = String::new();
        for (time, (value, error)) in self.domain.clone().into_iter().zip(self.image.clone()) {
            raw_data.push_str(&format!("{}\t{}\t{}\n", time, value, error));
        }
        raw_data
    }
}

impl<T> Plotable for ProcessError<T>
where
    T: Display + Clone,
{
    fn plot_script(&self) -> String {
        let mut gnuplot_script = self.opening_plot_script();

        let dashtype = self.dashtype().unwrap_or(1);

        gnuplot_script += &format!(
            "plot {:?} using 1:2 with {} dashtype {}, \"\" using 1:($2+$3):($2-$3) with filledcurves fs transparent solid 0.5 linecolor rgb \"dark-grey\"\n",
            self.data_path(),
            self.style(),
            dashtype,
        );
        gnuplot_script += &self.ending_plot_script();

        gnuplot_script
    }
}