tools/perf/benchmarks/skpicture_printer.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 from core import perf_benchmark
   6
   7 from telemetry import benchmark
   8 from telemetry.core import discover
   9 from telemetry.page import page_set
  10
  11 from measurements import skpicture_printer
  12
  13
  14 def _MatchPageSetName(page_set_name, page_set_base_dir):
  15   page_sets = []
  16   page_sets += discover.DiscoverClasses(page_set_base_dir, page_set_base_dir,
  17                                         page_set.PageSet,
  18                                         index_by_class_name=True).values()
  19   for p in page_sets:
  20     if page_set_name == p.Name():
  21       return p
  22   return None
  23
  24
  25 @benchmark.Disabled
  26 class SkpicturePrinter(perf_benchmark.PerfBenchmark):
  27   @classmethod
  28   def AddBenchmarkCommandLineArgs(cls, parser):
  29     parser.add_option('--page-set-name',  action='store', type='string')
  30     parser.add_option('--page-set-base-dir', action='store', type='string')
  31     parser.add_option('-s', '--skp-outdir',
  32                       help='Output directory for the SKP files')
  33   @classmethod
  34   def ProcessCommandLineArgs(cls, parser, args):
  35     if not args.page_set_name:
  36       parser.error('Please specify --page-set-name')
  37     if not args.page_set_base_dir:
  38       parser.error('Please specify --page-set-base-dir')
  39     if not args.skp_outdir:
  40       parser.error('Please specify --skp-outdir')
  41
  42   @classmethod
  43   def Name(cls):
  44     return 'skpicture_printer'
  45
  46   def CreatePageTest(self, options):
  47     return skpicture_printer.SkpicturePrinter(options.skp_outdir)
  48
  49   def CreatePageSet(self, options):
  50     page_set_class = _MatchPageSetName(options.page_set_name,
  51                                        options.page_set_base_dir)
  52     return page_set_class()