src/tests/audio_test_gui.py

#!/usr/bin/env python
#
# 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.
"""Script functions as a web app and wrapper for the cras_router program."""

import re
import subprocess
import logging
import cherrypy

# Node Format: [Stable_Id, ID, Vol, Plugged, L/R_swapped, Time, Type, Name]
ID_INDEX = 1
PLUGGED_INDEX = 3
TYPE_INDEX = 6
NAME_INDEX = 7


def get_plugged_nodes(plugged_nodes, lines, is_input):
  start_str = 'Input Nodes:' if is_input else 'Output Nodes:'
  end_str = 'Attached clients:' if is_input else 'Input Devices:'
  for i in range(lines.index(start_str) + 2,
                 lines.index(end_str)):
    node = filter(None, re.split(r'\s+|\*+', lines[i]))
    # check for nodes that are plugged nodes and loopback
    if node[PLUGGED_INDEX] == 'yes' and node[TYPE_INDEX][:4] != 'POST':
      key = node[TYPE_INDEX] + ' ' + node[NAME_INDEX]
      plugged_nodes[key] = node[ID_INDEX]


class CrasRouterTest(object):
  """Cherrypy class that builds and runs the HTML for audio testing tool."""

  @cherrypy.expose
  def index(self):
    """Builds up and displays the html for the audio testing tool.

    Returns:
      html that was built up based on plugged audio devices.
    """

    # Stop program if currently being run.
    if 'process' in cherrypy.session:
      print 'Existing process'
      # If return code is None process is still running
      if cherrypy.session['process'].poll() is None:
        print 'Killing existing process'
        cherrypy.session['process'].kill()
      else:
        print 'Process already finished'

    html = """<html>
              <head>
                <title>Audio Test</title>
              </head>
              <body style="background-color:lightgrey;">
                <font color="red">
                <h1>Audio Closed Loop Test</h1>
                <font style="color:rgb(100, 149, 237)">
                <h3>
                <form name="routerOptions" method="get"
                onsubmit="return validateForm()" action="start_test">
                  <h2>Input Type</h2>
              """
    dump = subprocess.check_output(['cras_test_client', '--dump_s'])
    if not dump:
      return 'Could not connect to server'
    dump_lines = dump.split('\n')
    input_plugged_nodes = {}
    get_plugged_nodes(input_plugged_nodes, dump_lines, True)
    for name, node_id in input_plugged_nodes.items():
      line = '<input type ="radio" name="input_type" value="'
      line += node_id + '">' +name + '<br>\n'
      html += line

    html += """<input type ="radio" id="i0" name="input_type"
               value="file">File<br>
                 <div id="input_file" style="display:none;">
                 Filename <input type="text" name="input_file"><br><br>
                 </div>
               <h2>Output Type</h2>"""
    output_plugged_nodes = {}
    get_plugged_nodes(output_plugged_nodes, dump_lines, False)
    for name, node_id in output_plugged_nodes.items():
      line = '<input type ="radio" name="output_type" value="'
      line = line + node_id + '">' +name + '<br>\n'
      html += line

    html += """<input type ="radio" name="output_type"
               value="file">File<br>
               <div id="output_file" style="display:none;">
               Filename <input type="text" name="output_file">
               </div><br>
               <h2>Sample Rate</h2>
               <input type="radio" name="rate" id="sample_rate1" value=48000
               checked>48,000 Hz<br>
               <input type="radio" name="rate" id="sample_rate0" value=44100>
               44,100 Hz<br><br>
               <button type="submit" onclick="onOff(this)">Test!</button>
               </h3>
               </form>
               </font>
               </body>
               </html>
               """
    javascript = """
                 <script>
                 /* Does basic error checking to make sure user doesn't
                  * give bad options to the router.
                  */
                 function validateForm(){
                    var input_type =
                        document.forms['routerOptions']['input_type'].value;
                    var output_type =
                        document.forms['routerOptions']['output_type'].value;
                    if (input_type == '' || output_type == '') {
                        alert('Please select an input and output type.');
                        return false;
                    }
                    if (input_type == 'file' && output_type == 'file') {
                        alert('Input and Output Types cannot both be files!');
                        return false;
                    }
                    //check if filename is valid
                    if (input_type == 'file') {
                        var input_file =
                          document.forms['routerOptions']['input_file'].value;
                        if (input_file == '') {
                            alert('Please enter a file name');
                            return false;
                        }
                    }
                    if (output_type == 'file') {
                        var output_file =
                          document.forms['routerOptions']['output_file'].value;
                        if (output_file == '') {
                            alert('Please enter a file name');
                            return false;
                        }
                    }
                }

                function show_filename(radio, file_elem) {
                    for(var i =0; i < radio.length; i++){
                        radio[i].onclick = function(){
                            if (this.value == 'file') {
                                file_elem.style.display = 'block';
                            } else {
                                file_elem.style.display = 'none';
                            }
                        }
                    }
                }
                /* Loops determine if filename field should be shown */
                var input_type_rad =
                    document.forms['routerOptions']['input_type'];
                var input_file_elem =
                    document.getElementById('input_file');
                var output_type_rad =
                    document.forms['routerOptions']['output_type'];
                var output_file_elem =
                    document.getElementById('output_file');
                show_filename(input_type_rad, input_file_elem);
                show_filename(output_type_rad, output_file_elem);
                </script>"""
    html += javascript
    return html

  @cherrypy.expose
  def start_test(self, input_type, output_type, input_file='',
                 output_file='', rate=48000):
    """Capture audio from the input_type and plays it back to the output_type.

    Args:
      input_type: Node id for the selected input or 'file' for files
      output_type: Node id for the selected output or 'file' for files
      input_file: Path of the input if 'file' is input type
      output_file: Path of the output if 'file' is output type
      rate: Sample rate for the test.

    Returns:
      html for the tesing in progress page.
    """
    print 'Beginning test'
    if input_type == 'file' or output_type == 'file':
        command = ['cras_test_client']
    else:
        command = ['cras_router']
    if input_type == 'file':
      command.append('--playback_file')
      command.append(str(input_file))
    else:
      set_input = ['cras_test_client', '--select_input', str(input_type)]
      if subprocess.check_call(set_input):
        print 'Error setting input'
    if output_type == 'file':
      command.append('--capture_file')
      command.append(str(output_file))
    else:
      set_output = ['cras_test_client', '--select_output', str(output_type)]
      if subprocess.check_call(set_output):
        print 'Error setting output'
    command.append('--rate')
    command.append(str(rate))
    print 'Running commmand: ' + str(command)
    p = subprocess.Popen(command)
    cherrypy.session['process'] = p
    return """
    <html>
    <head>
    <style type="text/css">
    body {
      background-color: #DC143C;
    }
    #test {
      color: white;
      text-align: center;
    }
    </style>
      <title>Running test</title>
    </head>
    <body>
      <div id="test">
        <h1>Test in progress</h1>
        <form action="index"><!--Go back to orginal page-->
          <button type="submit" id="stop">Click to stop</button>
        </form>
        <h2>Time Elapsed<br>
            <time id="elapsed_time">00:00</time>
        </h2>
        </div>
    </body>
    </html>
    <script type="text/javascript">
    var seconds = 0;
    var minutes = 0;
    var elapsed_time;
    var start_time = new Date().getTime();
    function secondPassed(){
      var time = new Date().getTime() - start_time;
      elapsed_time = Math.floor(time / 100) / 10;
      minutes = Math.floor(elapsed_time / 60);
      seconds = Math.floor(elapsed_time % 60);
      var seconds_str = (seconds < 10 ? '0' + seconds: '' + seconds);
      var minutes_str = (minutes < 10 ? '0' + minutes: '' + minutes);
      var time_passed = minutes_str + ':' + seconds_str;
      document.getElementById("elapsed_time").textContent = time_passed;
    }
    //have time tic every second
    var timer = setInterval(secondPassed, 1000);
    var stop = document.getElementById("stop");
    stop.onclick = function(){
      seconds = 0;
      minutes = 0;
      clearInterval(timer);
    }
    </script>"""

if __name__ == '__main__':
  conf = {
      '/': {
          'tools.sessions.on': True
      }
  }
  cherrypy.quickstart(CrasRouterTest(), '/', conf)