1 # Copyright 2016 Google Inc. 2 # 3 # Use of this source code is governed by a BSD-style license that can be 4 # found in the LICENSE file. 5 6 """Parses an skpbench result from a line of output text.""" 7 8 from __future__ import print_function 9 import re 10 import sys 11 12 class BenchResult: 13 FLOAT_REGEX = '[-+]?(\d+(\.\d*)?|\.\d+)([eE][-+]?\d+)?' 14 PATTERN = re.compile('^(?P<accum_pad> *)' 15 '(?P<accum>' + FLOAT_REGEX + ')' 16 '(?P<median_pad> +)' 17 '(?P<median>' + FLOAT_REGEX + ')' 18 '(?P<max_pad> +)' 19 '(?P<max>' + FLOAT_REGEX + ')' 20 '(?P<min_pad> +)' 21 '(?P<min>' + FLOAT_REGEX + ')' 22 '(?P<stddev_pad> +)' 23 '(?P<stddev>' + FLOAT_REGEX + '%)' 24 '(?P<samples_pad> +)' 25 '(?P<samples>\d+)' 26 '(?P<sample_ms_pad> +)' 27 '(?P<sample_ms>\d+)' 28 '(?P<clock_pad> +)' 29 '(?P<clock>[cg]pu)' 30 '(?P<metric_pad> +)' 31 '(?P<metric>ms|fps)' 32 '(?P<config_pad> +)' 33 '(?P<config>[^\s]+)' 34 '(?P<bench_pad> +)' 35 '(?P<bench>[^\s]+)$') 36 37 @classmethod 38 def match(cls, text): 39 match = cls.PATTERN.search(text) 40 return cls(match) if match else None 41 42 def __init__(self, match): 43 self.accum = float(match.group('accum')) 44 self.median = float(match.group('median')) 45 self.max = float(match.group('max')) 46 self.min = float(match.group('min')) 47 self.stddev = float(match.group('stddev')[:-1]) # Drop '%' sign. 48 self.samples = int(match.group('samples')) 49 self.sample_ms = int(match.group('sample_ms')) 50 self.clock = match.group('clock') 51 self.metric = match.group('metric') 52 self.config = match.group('config') 53 self.bench = match.group('bench') 54 self._match = match 55 56 def get_string(self, name): 57 return self._match.group(name) 58 59 def format(self, config_suffix=None): 60 if not config_suffix or config_suffix == '': 61 return self._match.group(0) 62 else: 63 values = list() 64 for name in ['accum', 'median', 'max', 'min', 'stddev', 65 'samples', 'sample_ms', 'clock', 'metric', 'config']: 66 values.append(self.get_string(name + '_pad')) 67 values.append(self.get_string(name)) 68 values.append(config_suffix) 69 bench_pad = self.get_string('bench_pad') 70 values.append(bench_pad[min(len(config_suffix), len(bench_pad) - 1):]) 71 values.append(self.get_string('bench')) 72 return ''.join(values) 73
