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
use std::ops::{Add, AddAssign, Sub, SubAssign};
pub struct Buffer<T> {
vals: Vec<T>,
len: usize,
pushes: usize,
index: usize,
sum: T,
}
impl<T> Buffer<T>
where
T: Default + Clone + Copy + Add + Sub + SubAssign + AddAssign,
{
pub fn new(len: usize) -> Self {
Buffer {
vals: vec![T::default(); len],
len,
pushes: 0,
index: 0,
sum: T::default(),
}
}
pub fn push(&mut self, val: T) {
if self.pushes >= self.len {
self.sum -= self.vals.get(self.index).expect("").clone();
}
self.sum += val;
self.vals[self.index] = val;
self.pushes += 1;
self.index += 1;
if self.index >= self.len {
self.index = 0;
}
}
pub fn qpush(&mut self, val: T) {
self.vals[self.index] = val;
self.index += 1;
if self.index >= self.len {
self.index = 0;
}
}
/**
get buffer value
# Examples
```
use new_york_utils::Buffer;
let mut buffer = Buffer::new(2);
buffer.push(1);
buffer.push(2);
assert_eq!(buffer.get_value(0), 2);
assert_eq!(buffer.get_value(1), 1);
buffer.push(3);
assert_eq!(buffer.get_value(0), 3);
assert_eq!(buffer.get_value(1), 2);
assert_eq!(buffer.get_value(2), 3);
```
*/
pub fn get_value(&self, ind: usize) -> T {
self.vals
.get((self.index + self.len - 1 + ind) % self.len)
.expect("")
.clone()
}
/**
get buffer value
# Examples
```
use new_york_utils::Buffer;
let mut buffer = Buffer::new(2);
buffer.push(1);
buffer.push(2);
assert_eq!(buffer.get_sum(), 3);
buffer.push(3);
assert_eq!(buffer.get_sum(), 5);
```
*/
pub fn get_sum(&self) -> T {
self.sum
}
}