quantedge-ta 0.12.0

A streaming technical analysis library for Rust
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

use crate::internals::RingBuffer;

#[derive(Clone, Debug)]
pub(crate) struct RollingSum {
    buffer: RingBuffer,
    sum: f64,
}

impl RollingSum {
    pub(crate) fn new(length: usize) -> Self {
        Self {
            buffer: RingBuffer::new(length),
            sum: 0.0,
        }
    }

    pub(crate) fn push(&mut self, value: f64) -> Option<f64> {
        if let Some(old) = self.buffer.push(value) {
            self.sum = self.sum - old + value;

            Some(self.sum)
        } else {
            self.sum += value;

            None
        }
    }

    pub(crate) fn replace(&mut self, value: f64) -> Option<f64> {
        let old = self.buffer.replace(value);
        self.sum = self.sum - old + value;

        self.sum()
    }

    #[must_use]
    pub(crate) fn sum(&self) -> Option<f64> {
        if self.buffer.is_ready() {
            Some(self.sum)
        } else {
            None
        }
    }
}