c33a9f9aa2
changed the behavior to import division and print to be consistent across all app and similar with python 3.x
74 lines
No EOL
2.1 KiB
Python
74 lines
No EOL
2.1 KiB
Python
# coding=utf-8
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__author__ = "Gina Häußge <osd@foosel.net>"
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
__copyright__ = "Copyright (C) 2014 The OctoPrint Project - Released under terms of the AGPLv3 License"
|
|
|
|
|
|
class TimeEstimationHelper(object):
|
|
|
|
STABLE_THRESHOLD = 0.1
|
|
STABLE_COUNTDOWN = 250
|
|
STABLE_ROLLING_WINDOW = 250
|
|
|
|
def __init__(self, rolling_window=None, countdown=None, threshold=None):
|
|
if rolling_window is None:
|
|
rolling_window = self.__class__.STABLE_ROLLING_WINDOW
|
|
if countdown is None:
|
|
countdown = self.__class__.STABLE_COUNTDOWN
|
|
if threshold is None:
|
|
threshold = self.__class__.STABLE_THRESHOLD
|
|
|
|
self._rolling_window = rolling_window
|
|
self._countdown = countdown
|
|
self._threshold = threshold
|
|
|
|
import collections
|
|
self._distances = collections.deque([], self._rolling_window)
|
|
self._totals = collections.deque([], self._rolling_window)
|
|
self._sum_total = 0
|
|
self._count = 0
|
|
self._stable_counter = None
|
|
|
|
def is_stable(self):
|
|
return self._stable_counter is not None and self._stable_counter >= self._countdown
|
|
|
|
def update(self, newEstimate):
|
|
old_average_total = self.average_total
|
|
|
|
self._sum_total += newEstimate
|
|
self._totals.append(newEstimate)
|
|
self._count += 1
|
|
|
|
if old_average_total:
|
|
self._distances.append(abs(self.average_total - old_average_total))
|
|
|
|
if -1.0 * self._threshold < self.average_distance < self._threshold:
|
|
if self._stable_counter is None:
|
|
self._stable_counter = 0
|
|
else:
|
|
self._stable_counter += 1
|
|
else:
|
|
self._stable_counter = None
|
|
|
|
@property
|
|
def average_total(self):
|
|
if not self._count:
|
|
return None
|
|
else:
|
|
return self._sum_total / self._count
|
|
|
|
@property
|
|
def average_total_rolling(self):
|
|
if not self._count or self._count < self._rolling_window:
|
|
return None
|
|
else:
|
|
return sum(self._totals) / len(self._totals)
|
|
|
|
@property
|
|
def average_distance(self):
|
|
if not self._count or self._count < self._rolling_window + 1:
|
|
return None
|
|
else:
|
|
return sum(self._distances) / len(self._distances) |