Chromium Code Reviews| Index: LayoutTests/http/tests/media/resources/media-source/generate-config-change-tests.py |
| diff --git a/LayoutTests/http/tests/media/resources/media-source/generate-config-change-tests.py b/LayoutTests/http/tests/media/resources/media-source/generate-config-change-tests.py |
| new file mode 100755 |
| index 0000000000000000000000000000000000000000..7cb7c0f0cf6c91dd2c0512b33553b8758d486f27 |
| --- /dev/null |
| +++ b/LayoutTests/http/tests/media/resources/media-source/generate-config-change-tests.py |
| @@ -0,0 +1,211 @@ |
| +#!/usr/bin/python |
| +# Copyright 2013 Google Inc. All Rights Reserved. |
| +# |
| +# Licensed under the Apache License, Version 2.0 (the "License"); |
|
adamk
2013/06/18 21:54:23
Seems like this should use the standard Blink lice
acolwell GONE FROM CHROMIUM
2013/06/18 23:00:18
Oops. Done. This is all my own code. I just had an
|
| +# you may not use this file except in compliance with the License. |
| +# You may obtain a copy of the License at |
| +# |
| +# http://www.apache.org/licenses/LICENSE-2.0 |
| +# |
| +# Unless required by applicable law or agreed to in writing, software |
| +# distributed under the License is distributed on an "AS IS" BASIS, |
| +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| +# See the License for the specific language governing permissions and |
| +# limitations under the License. |
| + |
| +""" |
| +This is a script that generates the content and HTML files for Media Source |
| +codec config change LayoutTests. |
| +""" |
| +import json |
| +import os |
| + |
| +DURATION = 2 |
| +FORMATS = ['webm', 'mp4'] |
| +ENCODE_SETTINGS = [ |
| + ## Video-only files |
| + # Frame rate changes |
| + { 'fs': '320x240', 'fr': 24, 'kfr': 8, 'c': '#ff0000', 'vbr': 128, 'abr': 0, 'asr': 0, 'ach': 0, 'afreq': 0 }, |
| + { 'fs': '320x240', 'fr': 30, 'kfr': 10, 'c': '#ff0000', 'vbr': 128, 'abr': 0, 'asr': 0, 'ach': 0, 'afreq': 0 }, |
| + # Frame size change |
| + { 'fs': '640x480', 'fr': 30, 'kfr': 10, 'c': '#00ff00', 'vbr': 128, 'abr': 0, 'asr': 0, 'ach': 0, 'afreq': 0 }, |
| + # Bitrate change |
| + { 'fs': '320x240', 'fr': 30, 'kfr': 10, 'c': '#ff00ff', 'vbr': 256, 'abr': 0, 'asr': 0, 'ach': 0, 'afreq': 0 }, |
| + |
| + ## Audio-only files |
| + # Bitrate/Codebook changes |
| + { 'fs': '0x0', 'fr': 0, 'kfr': 0, 'c': '#000000', 'vbr': 0, 'abr': 128, 'asr': 44100, 'ach': 1, 'afreq': 2000 }, |
| + { 'fs': '0x0', 'fr': 0, 'kfr': 0, 'c': '#000000', 'vbr': 0, 'abr': 192, 'asr': 44100, 'ach': 1, 'afreq': 4000 }, |
| + |
| + ## Audio-Video files |
| + # Frame size change. |
| + { 'fs': '320x240', 'fr': 30, 'kfr': 10, 'c': '#ff0000', 'vbr': 256, 'abr': 128, 'asr': 44100, 'ach': 1, 'afreq': 2000 }, |
| + { 'fs': '640x480', 'fr': 30, 'kfr': 10, 'c': '#00ff00', 'vbr': 256, 'abr': 128, 'asr': 44100, 'ach': 1, 'afreq': 2000 }, |
| + # Audio bitrate change. |
| + { 'fs': '640x480', 'fr': 30, 'kfr': 10, 'c': '#00ff00', 'vbr': 256, 'abr': 192, 'asr': 44100, 'ach': 1, 'afreq': 4000 }, |
| + # Video bitrate change. |
| + { 'fs': '640x480', 'fr': 30, 'kfr': 10, 'c': '#00ffff', 'vbr': 512, 'abr': 128, 'asr': 44100, 'ach': 1, 'afreq': 2000 }, |
| +] |
| + |
| +CONFIG_CHANGE_TESTS = [ |
| + ["v-framerate", 0, 1, "Tests %s video-only frame rate changes."], |
| + ["v-framesize", 1, 2, "Tests %s video-only frame size changes."], |
| + ["v-bitrate", 1, 3, "Tests %s video-only bitrate changes."], |
| + ["a-bitrate", 4, 5, "Tests %s audio-only bitrate changes."], |
| + ["av-framesize", 6, 7, "Tests %s frame size changes in multiplexed content."], |
| + ["av-audio-bitrate", 7, 8, "Tests %s audio bitrate changes in multiplexed content."], |
| + ["av-video-bitrate", 7, 9, "Tests %s video bitrate changes in multiplexed content."] |
| +]; |
| + |
| +encoding_ids = [] |
| + |
| +def Run(cmdline): |
| + cmdlineStr = " ".join(cmdline) |
| + print cmdlineStr |
| + os.system(cmdlineStr) |
| + |
| +def GenerateManifest(filename, format, has_audio, has_video): |
| + |
| + codecInfo = { |
| + "mp4": { "audio": "mp4a.40.2", "video": "avc1.4D4001" }, |
| + "webm": { "audio": "vorbis", "video": "vp8" } |
| + } |
| + |
| + majorType = "audio" |
| + if has_video : |
| + majorType = "video" |
| + |
| + codecs = [] |
| + if has_video: |
| + codecs.append(codecInfo[format]["video"]); |
| + |
| + if has_audio: |
| + codecs.append(codecInfo[format]["audio"]); |
| + |
| + type = "%s/%s;codecs=\"%s\"" % (majorType, format, ",".join(codecs)) |
| + |
| + manifest = { 'url': media_filename, 'type' : type} |
| + |
| + f = open(filename, "wb") |
| + f.write(json.dumps(manifest, indent=4, separators=(',', ': '))) |
| + f.close() |
| + |
| +def GenerateTestHTML(format, config_change_tests, encoding_ids): |
| + template = """<!DOCTYPE html> |
| +<html> |
| + <head> |
| + <script src="/w3c/resources/testharness.js"></script> |
| + <script src="/w3c/resources/testharnessreport.js"></script> |
| + <script src="mediasource-util.js"></script> |
| + <script src="mediasource-config-changes.js"></script> |
| + <link rel="stylesheet" href="/w3c/resources/testharness.css"> |
| + </head> |
| + <body> |
| + <div id="log"></div> |
| + <script> |
| + mediasource_configchange_test("%(format)s", "%(idA)s", "%(idB)s", "%(description)s"); |
| + </script> |
| + </body> |
| +</html> |
| +""" |
| + for test_info in config_change_tests : |
| + filename = "../../media-source/mediasource-config-change-%s-%s.html" % (format, test_info[0]) |
| + html = template % {'format': format, |
| + 'idA': encoding_ids[test_info[1]], |
| + 'idB': encoding_ids[test_info[2]], |
| + 'description': test_info[3] % (format)} |
| + f = open(filename, "wb") |
| + f.write(html) |
| + f.close() |
| + |
| + |
| +for format in FORMATS: |
| + Run(["mkdir ",format]) |
| + |
| + for settings in ENCODE_SETTINGS: |
| + video_bitrate = settings['vbr'] |
| + has_video = (video_bitrate > 0) |
| + |
| + audio_bitrate = settings['abr'] |
| + has_audio = (audio_bitrate > 0) |
| + bitrate = video_bitrate + audio_bitrate |
| + |
| + frame_size = settings['fs'] |
| + frame_rate = settings['fr'] |
| + keyframe_rate = settings['kfr'] |
| + color = settings['c'] |
| + |
| + sample_rate = settings['asr'] |
| + channels = settings['ach'] |
| + frequency = settings['afreq'] |
| + |
| + cmdline = ["ffmpeg", "-y"] |
| + |
| + id_prefix = "" |
| + id_params = ""; |
| + if has_audio: |
| + id_prefix += "a" |
| + id_params += "-%sHz-%sch" % (sample_rate, channels) |
| + |
| + channel_layout = "FC" |
| + sin_func = "sin(%s*2*PI*t)" % frequency |
| + func = sin_func |
| + if channels == 2: |
| + channel_layout += "|BC" |
| + func += "|" + sin_func |
| + |
| + cmdline += ["-f", "lavfi", "-i", "aevalsrc=\"%s:s=%s:c=%s:d=%s\"" % (func, sample_rate, channel_layout, DURATION)] |
| + |
| + if has_video: |
| + id_prefix += "v" |
| + id_params += "-%s-%sfps-%skfr" % (frame_size, frame_rate, keyframe_rate) |
| + |
| + cmdline += ["-f", "lavfi", "-i", "color=%s:duration=%s:size=%s:rate=%s" % (color, DURATION, frame_size, frame_rate)] |
| + |
| + if has_audio: |
| + cmdline += ["-b:a", "%sk" % audio_bitrate] |
| + |
| + if has_video: |
| + cmdline += ["-b:v", "%sk" % video_bitrate] |
| + cmdline += ["-keyint_min", "%s" % keyframe_rate] |
| + cmdline += ["-g", "%s" % keyframe_rate] |
| + |
| + |
| + textOverlayInfo = "'drawtext=fontfile=Mono:fontsize=32:text=Time\\\\:\\\\ %{pts}" |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=32:text=Size\\\\:\\\\ %s" % (frame_size) |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=64:text=Bitrate\\\\:\\\\ %s" % (bitrate) |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=96:text=FrameRate\\\\:\\\\ %s" % (frame_rate) |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=128:text=KeyFrameRate\\\\:\\\\ %s" % (keyframe_rate) |
| + |
| + if has_audio: |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=160:text=SampleRate\\\\:\\\\ %s" % (sample_rate) |
| + textOverlayInfo += ",drawtext=fontfile=Mono:fontsize=32:y=192:text=Channels\\\\:\\\\ %s" % (channels) |
| + |
| + textOverlayInfo += "'" |
| + cmdline += ["-vf", textOverlayInfo] |
| + |
| + encoding_id = "%s-%sk%s" % (id_prefix, bitrate, id_params) |
| + |
| + if len(encoding_ids) < len(ENCODE_SETTINGS): |
| + encoding_ids.append(encoding_id) |
| + |
| + filename_base = "%s/test-%s" % (format, encoding_id) |
| + media_filename = filename_base + "." + format |
| + manifest_filename = filename_base + "-manifest.json" |
| + |
| + cmdline.append(media_filename) |
| + Run(cmdline) |
| + |
| + # Remux file so it conforms to MSE bytestream requirements. |
| + if format == "webm": |
| + tmp_filename = media_filename + ".tmp" |
| + Run(["mse_webm_remuxer", media_filename, tmp_filename]) |
| + Run(["mv", tmp_filename, media_filename]) |
| + elif format == "mp4": |
| + Run(["MP4Box", "-dash", "250", "-rap", media_filename]) |
| + Run(["mv", filename_base + "_dash.mp4", media_filename]) |
| + Run(["rm", filename_base + "_dash.mpd"]) |
| + |
| + GenerateManifest(manifest_filename, format, has_audio, has_video) |
| + |
| + GenerateTestHTML(format, CONFIG_CHANGE_TESTS, encoding_ids) |