skpbench

tools/skpbench/_benchresult.py

+# Copyright 2016 Google Inc.
+#
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+'''Parses an skpbench result from a line of output text.'''
+
+from __future__ import print_function
+import re
+import sys
+
+class BenchResult:
+  FLOAT_REGEX = '[-+]?(\d+(\.\d*)?|\.\d+)([eE][-+]?\d+)?'
+  PATTERN = re.compile('^(?P<median_pad> *)'
+                       '(?P<median>' + FLOAT_REGEX + ')'
+                       '(?P<accum_pad> +)'
+                       '(?P<accum>' + FLOAT_REGEX + ')'
+                       '(?P<max_pad> +)'
+                       '(?P<max>' + FLOAT_REGEX + ')'
+                       '(?P<min_pad> +)'
+                       '(?P<min>' + FLOAT_REGEX + ')'
+                       '(?P<stddev_pad> +)'
+                       '(?P<stddev>' + FLOAT_REGEX + '%)'
+                       '(?P<metric_pad> +)'
+                       '(?P<metric>ms|fps)'
+                       '(?P<samples_pad> +)'
+                       '(?P<samples>\d+)'
+                       '(?P<sample_ms_pad> +)'
+                       '(?P<sample_ms>\d+)'
+                       '(?P<config_pad> +)'
+                       '(?P<config>[^\s]+)'
+                       '(?P<bench_pad> +)'
+                       '(?P<bench>[^\s]+)$')
+
+  @classmethod
+  def match(cls, text):
+    match = cls.PATTERN.search(text)
+    return cls(match) if match else None
+
+  def __init__(self, match):
+    self.median = float(match.group('median'))
+    self.accum = float(match.group('accum'))
+    self.max = float(match.group('max'))
+    self.min = float(match.group('min'))
+    self.stddev = float(match.group('stddev')[:-1]) # Drop '%' sign.
+    self.metric = match.group('metric')
+    self.samples = int(match.group('samples'))
+    self.sample_ms = int(match.group('sample_ms'))
+    self.config = match.group('config')
+    self.bench = match.group('bench')
+    self._match = match
+
+  def get_string(self, name):
+    return self._match.group(name)
+
+  def print_values(self, config_suffix=None, outfile=sys.stdout):
+    if not config_suffix or config_suffix == '':
+      print(self._match.group(0), file=outfile)
+    else:
+      values = list()
+      for name in ['median', 'accum', 'max', 'min', 'stddev',
+                   'metric', 'samples', 'sample_ms', 'config']:
+        values.append(self.get_string(name + '_pad'))
+        values.append(self.get_string(name))
+      values.append(config_suffix)
+      bench_pad = self.get_string('bench_pad')
+      values.append(bench_pad[min(len(config_suffix), len(bench_pad) - 1):])
+      values.append(self.get_string('bench'))
+      print(''.join(values), file=outfile)