polars-utils 0.53.0

Private utils for the Polars DataFrame 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
56

use std::ops::{Add, AddAssign};

use num_traits::Num;

use crate::float::IsFloat;

#[derive(Debug, Clone)]
pub struct KahanSum<T> {
    sum: T,
    err: T,
}

impl<T: IsFloat + Num + Copy> KahanSum<T> {
    pub fn new(v: T) -> Self {
        KahanSum {
            sum: v,
            err: T::zero(),
        }
    }

    pub fn sum(&self) -> T {
        self.sum
    }
}

impl<T: Num> Default for KahanSum<T> {
    fn default() -> Self {
        KahanSum {
            sum: T::zero(),
            err: T::zero(),
        }
    }
}

impl<T: IsFloat + Num + AddAssign + Copy> AddAssign<T> for KahanSum<T> {
    fn add_assign(&mut self, rhs: T) {
        let y = rhs - self.err;
        let new_sum = self.sum + y;
        let new_err = (new_sum - self.sum) - y;
        self.sum = new_sum;
        if new_err.is_finite() {
            // Ensure err stays finite so we don't introduce NaNs through Inf - Inf.
            self.err = new_err;
        }
    }
}

impl<T: IsFloat + Num + AddAssign + Copy> Add<T> for KahanSum<T> {
    type Output = Self;

    fn add(self, rhs: T) -> Self::Output {
        let mut rv = self;
        rv += rhs;
        rv
    }
}