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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
# Copyright (c) 2016 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "Sorting algorithms" project.
# For details, see https://github.com/egor-tensin/sorting-algorithms.
# Distributed under the MIT License.
from enum import Enum
from numbers import Integral
from .input_kind import InputKind
from .plotter import PlotBuilder
from . import registry
from .timer import Timer
class TimeUnits(Enum):
SECONDS = 'seconds'
MILLISECONDS = 'milliseconds'
MICROSECONDS = 'microseconds'
def get_factor(self):
if self is TimeUnits.SECONDS:
return 1.
elif self is TimeUnits.MILLISECONDS:
return 1000.
elif self is TimeUnits.MICROSECONDS:
return 1000000.
else:
raise NotImplementedError('invalid time units: ' + str(self))
def __str__(self):
return self.value
class AlgorithmParameters:
def __init__(self, algorithm, min_len, max_len,
input_kind=InputKind.AVERAGE, iterations=1):
if isinstance(algorithm, str):
algorithm = registry.get(algorithm)
self.algorithm = algorithm
self.input_kind = input_kind
self._min_len = None
self._max_len = None
self.min_len = min_len
self.max_len = max_len
self._iterations = None
self.iterations = iterations
@property
def min_len(self):
return self._min_len
@min_len.setter
def min_len(self, val):
if not isinstance(val, Integral):
raise TypeError('must be an integral value')
val = int(val)
if val < 0:
raise ValueError('must be non-negative')
if self.max_len is not None and self.max_len < val:
raise ValueError('must not be greater than the maximum length')
self._min_len = val
@property
def max_len(self):
return self._max_len
@max_len.setter
def max_len(self, val):
if not isinstance(val, Integral):
raise TypeError('must be an integral value')
val = int(val)
if val < 0:
raise ValueError('must be non-negative')
if self.min_len is not None and self.min_len > val:
raise ValueError('must not be lesser than the minimum length')
self._max_len = val
@property
def iterations(self):
return self._iterations
@iterations.setter
def iterations(self, val):
if not isinstance(val, Integral):
raise TypeError('must be an integral value')
val = int(val)
if val < 1:
raise ValueError('must be positive')
self._iterations = val
def measure_running_time(self):
input_len_range = list(range(self.min_len, self.max_len + 1))
running_time = []
for input_len in input_len_range:
input_sample = self.algorithm.gen_input(input_len, self.input_kind)
input_copies = [list(input_sample) for _ in range(self.iterations)]
with Timer(running_time, self.iterations):
for i in range(self.iterations):
self.algorithm.function(input_copies[i])
return input_len_range, running_time
@staticmethod
def _format_plot_xlabel():
return 'Input length'
@staticmethod
def _format_plot_ylabel(units):
return 'Running time ({})'.format(units)
def _format_plot_title(self):
return '{}, {} case'.format(
self.algorithm.display_name, self.input_kind)
def _format_plot_suptitle(self):
return self.algorithm.display_name
@staticmethod
def _derive_time_units(ys):
max_y = max(ys)
if max_y > 0.1:
return TimeUnits.SECONDS
elif max_y > 0.0001:
return TimeUnits.MILLISECONDS
else:
return TimeUnits.MICROSECONDS
def plot_running_time(self, output_path=None):
xs, ys = self.measure_running_time()
units = self._derive_time_units(ys)
ys = [y * units.get_factor() for y in ys]
plot_builder = PlotBuilder()
plot_builder.show_grid()
plot_builder.set_xlabel(self._format_plot_xlabel())
plot_builder.set_ylabel(self._format_plot_ylabel(units))
#plot_builder.set_yticklabels_scientific()
plot_builder.set_title(self._format_plot_title())
plot_builder.plot(xs, ys)
if output_path is None:
plot_builder.show()
else:
plot_builder.save(output_path)
|
