src/tools/perf/measurements/screenshot.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 import os
   5
   6 from telemetry.page import page_test
   7 from telemetry.page import page_test
   8 from telemetry.value import scalar
   9
  10
  11 class Screenshot(page_test.PageTest):
  12   def __init__(self):
  13     super(Screenshot, self).__init__(
  14         action_name_to_run = 'RunPrepareForScreenshot',
  15         is_action_name_to_run_optional=True)
  16
  17   @classmethod
  18   def AddCommandLineArgs(cls, parser):
  19     parser.add_option('--png-outdir',
  20                       help='Output directory for the PNG files')
  21
  22   @classmethod
  23   def ProcessCommandLineArgs(cls, parser, args):
  24     if not args.png_outdir:
  25       parser.error('Please specify --png-outdir')
  26     cls._png_outdir = args.png_outdir
  27
  28   def ValidateAndMeasurePage(self, page, tab, results):
  29     if not tab.screenshot_supported:
  30       raise page_test.TestNotSupportedOnPlatformFailure(
  31           'Browser does not support screenshotting')
  32
  33     tab.WaitForDocumentReadyStateToBeComplete()
  34     screenshot = tab.Screenshot(60)
  35
  36     outpath = os.path.abspath(
  37         os.path.join(self._png_outdir, page.file_safe_name)) + '.png'
  38
  39     if os.path.exists(outpath):
  40       previous_mtime = os.path.getmtime(outpath)
  41     else:
  42       previous_mtime = -1
  43
  44     screenshot.WritePngFile(outpath)
  45
  46     saved_picture_count = 0
  47     if os.path.exists(outpath) and os.path.getmtime(outpath) > previous_mtime:
  48       saved_picture_count = 1
  49     results.AddValue(scalar.ScalarValue(
  50         results.current_page, 'saved_picture_count', 'count',
  51         saved_picture_count))