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

use crate::set::traits::Finite;
use num::{Float, FromPrimitive, Num};

/// # Example
/// ```
/// use math::{
///     interval::I64Interval,
///     iter::{
///         binned_interval_iter::{AggregateOp, IntoBinnedIntervalIter},
///         weighted_sum::WeightedSum,
///     },
///     partition::integer_interval_map::IntegerIntervalMap,
/// };
///
/// let bin_size = 5;
/// let mut interval_map = IntegerIntervalMap::new();
/// interval_map.aggregate(I64Interval::new(-1, 1), 2);
/// interval_map.aggregate(I64Interval::new(14, 17), -1);
///
/// // interval coordinates                        | value
/// // -1 | 0 1  |   ...   |        |              | +2
/// //    |      |   ...   |     14 | 15 16 17     | -1
/// //---------------------------------------------
/// // 0.4|| 0.8 ||  ...   || -0.2 || -0.6         | bin average
/// assert_eq!(
///     interval_map
///         .iter()
///         .into_binned_interval_iter(
///             bin_size,
///             AggregateOp::Average,
///             Box::new(|(&interval, &val)| (interval, val as f64))
///         )
///         .weighted_sum(),
///     2.0
/// );
/// ```
pub trait WeightedSum<Item, V>
where
    V: Copy + Num, {
    fn weighted_sum(self) -> V;
}

impl<I, P, Dtype> WeightedSum<(P, Dtype), Dtype> for I
where
    I: Iterator<Item = (P, Dtype)>,
    P: Finite,
    Dtype: Float + FromPrimitive,
{
    fn weighted_sum(self) -> Dtype {
        crate::stats::kahan_sigma(self, |(interval, value)| {
            Dtype::from_usize(interval.size())
                .expect("failed to convert from usize to Dtype")
                * value
        })
    }
}

#[cfg(test)]
mod tests {
    use crate::{interval::I64Interval, iter::weighted_sum::WeightedSum};

    #[test]
    fn test_weighted_sum() {
        let arr = vec![
            (I64Interval::new(2, 5), 2f32),
            (I64Interval::new(10, 25), 1.5),
            (I64Interval::new(1, 5), 0.5),
        ];
        // 4 * 2 + 16 * 1.5 + 5 * 0.5
        assert_eq!(arr.into_iter().weighted_sum(), 34.5);

        let arr = vec![
            (I64Interval::new(0, 9), -2f64),
            (I64Interval::new(9, 15), 1.5),
            (I64Interval::new(0, 20), 0.5),
        ];
        // 10 * (-2) + 7 * 1.5 + 21 * 0.5
        assert_eq!(arr.into_iter().weighted_sum(), 1.0);
    }
}