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

import os

from autotest_lib.client.bin import test, utils
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib.cros import chrome


MEDIA_GVD_INIT_STATUS = 'Media.GpuVideoDecoderInitializeStatus'


class video_VideoDecodeAcceleration(test.test):
    """This test verifies VDA works in Chrome."""
    version = 1


    def run_once(self, video_file):
        """Tests whether VDA works by verifying histogram for the loaded video.

        @param video_file: Sample video file to be loaded in Chrome.
        """
        with chrome.Chrome() as cr:
            cr.browser.SetHTTPServerDirectories(self.bindir)
            video_url = cr.browser.http_server.UrlOf(
                    os.path.join(self.bindir, video_file))
            tab1 = cr.browser.tabs[0]
            tab1.Navigate(video_url)
            tab1.WaitForDocumentReadyStateToBeComplete()
            tab2 = cr.browser.tabs.New()

            # Waiting for histogram updated for the test video.
            def gpu_histogram_loaded():
                tab2.Navigate('chrome://histograms/%s' % MEDIA_GVD_INIT_STATUS)
                return tab2.EvaluateJavaScript(
                        'document.documentElement.innerText.search('
                        '\'%s\') != -1' % MEDIA_GVD_INIT_STATUS)

            utils.poll_for_condition(gpu_histogram_loaded,
                    exception=error.TestError(
                            'Histogram gpu status failed to load.'),
                            sleep_interval=1)
            if tab2.EvaluateJavaScript(
                    'document.documentElement.innerText.search('
                    '\'average = 0.0\') == -1'):
                raise error.TestError('Video decode acceleration not working.')