Home | History | Annotate | Download | only in video 
      1 # Copyright (c) 2014 The Chromium OS 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 import argparse, common, datetime
      6 
      7 from autotest_lib.client.common_lib import file_utils
      8 from autotest_lib.client.common_lib.cros import chrome
      9 from autotest_lib.client.cros.video import media_test_factory, \
     10     sequence_generator
     11 
     12 
     13 def main():
     14     """
     15     Loads specified video in a chrome HTML5 player and collects its screenshots
     16     at specified time instances.
     17     """
     18 
     19     parser = argparse.ArgumentParser(
     20         formatter_class=argparse.RawDescriptionHelpFormatter,
     21 
     22         description='Capture images of a video.',
     23 
     24         epilog='''
     25         This tool supports image-comparison-based video glitch detection tests.
     26 
     27         Use this tool to capture golden/reference images that will be used to
     28         verify test images captured during a video playback.
     29 
     30         Run this tool directly on the device.
     31 
     32         Copy tool first into /usr/local/autotest/cros so that autotest's
     33         common.py is imported so as to resolve autotest references correctly.
     34 
     35         Output images will be placed under /tmp/test. Images will be saved as
     36         hh_mm_ss_mss.png denoting the moment in time it was captured.
     37 
     38         ''')
     39 
     40     parser.add_argument("name",
     41                         help="Name of video to use.")
     42 
     43     parser.add_argument("format",
     44                         choices=['mp4', 'webm'],
     45                         help="Video format to use.")
     46 
     47     parser.add_argument("definition",
     48                         choices=['480p', '720p', '1080p', '720p_1080p'],
     49                         help="Video definition to use.")
     50 
     51     parser.add_argument("--start",
     52                         default="00:01",
     53                         help="Time to start taking screenshots. (mm:ss)")
     54 
     55     parser.add_argument("--stop",
     56                         help="Time to stop taking screenshots. (mm:ss).")
     57 
     58     parser.add_argument("interval",
     59                         type=int,
     60                         help="Seconds between two successive captures.")
     61 
     62     args = parser.parse_args()
     63 
     64     time_format = '%M:%S'
     65 
     66     # Parse time arguments from user
     67     # Start time has a default argument of 01:00, parse right away
     68     # Parse stop time later as we don't know the length of the video,
     69     # the factory does
     70     tmp = datetime.datetime.strptime(args.start, time_format)
     71     start = datetime.timedelta(minutes=tmp.minute, seconds=tmp.second)
     72 
     73     with chrome.Chrome(init_network_controller=True) as cr:
     74         bindir = '/usr/local/autotest/cros/video'
     75 
     76         cr.browser.platform.SetHTTPServerDirectories(bindir)
     77 
     78         factory = media_test_factory.MediaTestFactory(
     79                       cr.browser.tabs[0],
     80                       cr.browser.platform.http_server,
     81                       bindir,
     82                       'dev',
     83                       args.name,
     84                       args.format,
     85                       args.definition)
     86 
     87         # if stop time is not specified, use the length of the video
     88         if args.stop is None:
     89             stop = factory.media_length
     90         else:
     91             tmp = datetime.datetime.strptime(args.stop, time_format)
     92             stop = datetime.timedelta(minutes=tmp.minute, seconds=tmp.second)
     93 
     94         file_utils.rm_dir_if_exists(factory.test_working_dir)
     95 
     96         file_utils.make_leaf_dir(factory.test_working_dir)
     97 
     98         seq = sequence_generator.generate_interval_sequence(start,
     99                                                             stop,
    100                                                             args.interval)
    101 
    102         collector = factory.make_video_screenshot_collector()
    103 
    104         collector.ensure_player_is_ready()
    105 
    106         collector.collect_multiple_screenshots(seq)
    107 
    108 
    109 if __name__ == '__main__':
    110     main()
    111