# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# This file has been copied from
# https://chromium.googlesource.com/chromium/src/+/master/tools/perf/benchmarks/repaint.py
# and modified locally to support CT pagesets. Hopefully one day this file
# will live in telemetry codebase instead.

from benchmarks import silk_flags
from benchmarks import skpicture_printer
from measurements import smoothness
import page_sets
from telemetry import benchmark


class _Repaint(benchmark.Benchmark):
  @classmethod
  def AddBenchmarkCommandLineArgs(cls, parser):
    parser.add_option('--mode', type='string',
                      default='viewport',
                      help='Invalidation mode. '
                      'Supported values: fixed_size, layer, random, viewport.')
    parser.add_option('--width', type='int',
                      default=None,
                      help='Width of invalidations for fixed_size mode.')
    parser.add_option('--height', type='int',
                      default=None,
                      help='Height of invalidations for fixed_size mode.')
    parser.add_option('--page-set-name',  action='store', type='string')
    parser.add_option('--page-set-base-dir', action='store', type='string')

  def CreateUserStorySet(self, options):
    page_set_class = skpicture_printer._MatchPageSetName(
        options.page_set_name, options.page_set_base_dir)
    return page_set_class()

  def CreatePageTest(self, options):
    return smoothness.Repaint()


@benchmark.Disabled
class RepaintCTPages(_Repaint):
  test = smoothness.Repaint

  @classmethod
  def ProcessCommandLineArgs(cls, parser, args):
    if not args.page_set_name:
      parser.error('Please specify --page-set-name')
    if not args.page_set_base_dir:
      parser.error('Please specify --page-set-base-dir')

  def CreateStorySet(self, options):
    page_set_class = skpicture_printer._MatchPageSetName(
        options.page_set_name, options.page_set_base_dir)
    return page_set_class()
