mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-06-25 03:22:07 +08:00
github-actions[bot]
f5e5d688c3
version: sunnypilot v0.9.9 release date: 2025-03-18T06:26:39 master commit: 8e5b656e654a7694c22f2b0b7bee0802db02b4af
25 lines
785 B
Python
25 lines
785 B
Python
class MovingAverage:
|
|
def __init__(self, window_size: int):
|
|
self.window_size: int = window_size
|
|
self.buffer: list[float] = [0.0] * window_size
|
|
self.index: int = 0
|
|
self.count: int = 0
|
|
self.sum: float = 0.0
|
|
|
|
def add_value(self, new_value: float):
|
|
# Update the sum: subtract the value being replaced and add the new value
|
|
self.sum -= self.buffer[self.index]
|
|
self.buffer[self.index] = new_value
|
|
self.sum += new_value
|
|
|
|
# Update the index in a circular manner
|
|
self.index = (self.index + 1) % self.window_size
|
|
|
|
# Track the number of added values (for partial windows)
|
|
self.count = min(self.count + 1, self.window_size)
|
|
|
|
def get_average(self) -> float:
|
|
if self.count == 0:
|
|
return float('nan')
|
|
return self.sum / self.count
|