Some of the repetitive violations were suppressed.
3.21KiB; Python | 2018-01-28 18:59:31+01 | SLOC 76
1
#!/bin/env python
2
3
##
4
# Copyright(c) 2010-2015 Intel Corporation.
5
# Copyright(c) 2016-2018 Viosoft Corporation.
6
# All rights reserved.
7
#
8
# Redistribution and use in source and binary forms, with or without
9
# modification, are permitted provided that the following conditions
10
# are met:
11
#
12
#   * Redistributions of source code must retain the above copyright
13
#     notice, this list of conditions and the following disclaimer.
14
#   * Redistributions in binary form must reproduce the above copyright
15
#     notice, this list of conditions and the following disclaimer in
16
#     the documentation and/or other materials provided with the
17
#     distribution.
18
#   * Neither the name of Intel Corporation nor the names of its
19
#     contributors may be used to endorse or promote products derived
20
#     from this software without specific prior written permission.
21
#
22
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
25
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
26
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
27
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
28
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
30
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
32
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33
##
34
35 11
from decimal import *
36
from time import time
37
38
class Progress:
39
    def __init__(self, limit, fieldNames = [], overallETA = True):
40 1
        self._fieldNames = fieldNames;
41 1
        self._limit = limit;
42 1
        self._progress = 0;
43 1
        self._prevProgress = 0;
44 1
        self._prevTime = 0;
45 1
        self._progressSetCount = 0;
46 1
        self._time = 0;
47 1
        self._overallETA = overallETA;
48
49
    def setProgress(self, progress, fieldValues = []):
50 2
        self._fieldValues = fieldValues;
51 1
        if (self._overallETA == True):
52
            self._progress = progress
53 1
            self._time = time();
54 1
            if (self._progressSetCount == 0):
55
                self._prevProgress = self._progress;
56
                self._prevTime = self._time;
57
        else:
58
            self._prevProgress = self._progress;
59
            self._prevTime = self._time;
60
            self._progress = progress;
61
            self._time = time();
62
        self._progressSetCount += 1
63
64
    def incrProgress(self):
65
        self.setProgress(self._progress + 1);
66
67
    def toString(self):
68
        ret = ""
69
        ret += str(self._getETA()) + " seconds left"
70
        for f,v in zip(self._fieldNames, self._fieldValues):
71
            ret += ", %s=%s" % (str(f),str(v))
72
        return ret;
73
74
    def _getETA(self):
75 1
        if (self._progressSetCount < 2):
76
            return "N/A"
77
        diff = self._progress - self._prevProgress;
78
        t_diff = Decimal(self._time - self._prevTime);
79
        if (t_diff < 0.001 or diff <= 0):
80
            return "N/A"
81
        rate = Decimal(diff)/t_diff
82
        remaining = Decimal(self._limit - self._progress);
83
        return round(remaining/rate, 2);