# Copyright 2016 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
import random
import re
import subprocess
import time

from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error
from autotest_lib.client.cros.audio import audio_helper


class audio_CrasOutputStress(test.test):
    """Checks if output buffer will drift to super high level."""
    version = 2
    _MAX_OUTPUT_STREAMS = 3
    _LOOP_COUNT = 300
    _OUTPUT_BUFFER_LEVEL = '.*?FILL_AUDIO.*?hw_level.*?(\d+).*?'
    _BUFFER_DRIFT_CRITERIA = 4096

    def run_once(self):
        """
        Repeatedly add output streams of random configurations and
        remove them to verify if output buffer level would drift.
        """
        self._output_streams = []
        self._rates = ['48000', '44100']
        self._block_sizes = ['512', '1024']

        loop_count = 0
        while loop_count < self._LOOP_COUNT:

            # 1 for adding stream, 0 for removing stream
            add = random.randint(0, 1)
            if len(self._output_streams) == 0:
                add = 1
            elif len(self._output_streams) == self._MAX_OUTPUT_STREAMS:
                add = 0

            if add == 1:
                cmd = ['cras_test_client', '--playback_file', '/dev/zero',
                       '--rate', self._rates[random.randint(0, 1)],
                       '--block_size', self._block_sizes[random.randint(0, 1)]]
                proc = subprocess.Popen(cmd)
                self._output_streams.append(proc)
            else:
                self._output_streams[0].kill()
                self._output_streams.remove(self._output_streams[0])
                time.sleep(0.1)
            loop_count += 1

        # Get the buffer level.
        buffer_level = self._get_buffer_level()
        log_file = os.path.join(self.resultsdir, "audio_diagnostics.txt")
        with open(log_file, 'w') as f:
            f.write(audio_helper.get_audio_diagnostics())

        # Clean up all streams.
        while len(self._output_streams) > 0:
            self._output_streams[0].kill()
            self._output_streams.remove(self._output_streams[0])

        if buffer_level > self._BUFFER_DRIFT_CRITERIA:
            raise error.TestFail('Buffer level %d drift too high' %
                                 buffer_level)

    def _get_buffer_level(self):
        """Gets a rough number about current buffer level.

        @returns: The current buffer level.

        """
        proc = subprocess.Popen(['cras_test_client', '--dump_a'], stdout=subprocess.PIPE)
        output, err = proc.communicate()
        buffer_level = 0
        for line in output.split('\n'):
            search = re.match(self._OUTPUT_BUFFER_LEVEL, line)
            if search:
                tmp = int(search.group(1))
                if tmp > buffer_level:
                    buffer_level = tmp
        return buffer_level