cras/src/tests/audio_test_gui.py - platform/external/adhd - Git at Google

 #!/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)
	#!/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)