Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1 | from test import support |
| 2 | support.requires('audio') |
Guido van Rossum | 4507ec7 | 2003-02-14 19:29:22 +0000 | [diff] [blame] | 3 | |
Benjamin Peterson | e549ead | 2009-03-28 21:42:05 +0000 | [diff] [blame] | 4 | from test.support import findfile |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 5 | |
R. David Murray | a21e4ca | 2009-03-31 23:16:50 +0000 | [diff] [blame^] | 6 | ossaudiodev = support.import_module('ossaudiodev') |
| 7 | |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 8 | import errno |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 9 | import sys |
Brett Cannon | abd9a59 | 2008-07-25 17:56:47 +0000 | [diff] [blame] | 10 | import sunau |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 11 | import time |
| 12 | import audioop |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 13 | import unittest |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 14 | |
Greg Ward | 8a709b3 | 2003-06-03 00:32:44 +0000 | [diff] [blame] | 15 | # Arggh, AFMT_S16_NE not defined on all platforms -- seems to be a |
| 16 | # fairly recent addition to OSS. |
| 17 | try: |
| 18 | from ossaudiodev import AFMT_S16_NE |
| 19 | except ImportError: |
| 20 | if sys.byteorder == "little": |
| 21 | AFMT_S16_NE = ossaudiodev.AFMT_S16_LE |
| 22 | else: |
| 23 | AFMT_S16_NE = ossaudiodev.AFMT_S16_BE |
| 24 | |
| 25 | |
Greg Ward | 55a8790 | 2002-12-10 16:27:35 +0000 | [diff] [blame] | 26 | def read_sound_file(path): |
Brett Cannon | abd9a59 | 2008-07-25 17:56:47 +0000 | [diff] [blame] | 27 | with open(path, 'rb') as fp: |
| 28 | au = sunau.open(fp) |
| 29 | rate = au.getframerate() |
| 30 | nchannels = au.getnchannels() |
| 31 | encoding = au._encoding |
| 32 | fp.seek(0) |
| 33 | data = fp.read() |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 34 | |
Brett Cannon | abd9a59 | 2008-07-25 17:56:47 +0000 | [diff] [blame] | 35 | if encoding != sunau.AUDIO_FILE_ENCODING_MULAW_8: |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 36 | raise RuntimeError("Expect .au file with 8-bit mu-law samples") |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 37 | |
Greg Ward | 55a8790 | 2002-12-10 16:27:35 +0000 | [diff] [blame] | 38 | # Convert the data to 16-bit signed. |
| 39 | data = audioop.ulaw2lin(data, 2) |
| 40 | return (data, rate, 16, nchannels) |
| 41 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 42 | class OSSAudioDevTests(unittest.TestCase): |
Greg Ward | 55a8790 | 2002-12-10 16:27:35 +0000 | [diff] [blame] | 43 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 44 | def play_sound_file(self, data, rate, ssize, nchannels): |
| 45 | try: |
| 46 | dsp = ossaudiodev.open('w') |
| 47 | except IOError as msg: |
| 48 | if msg.args[0] in (errno.EACCES, errno.ENOENT, |
| 49 | errno.ENODEV, errno.EBUSY): |
Benjamin Peterson | e549ead | 2009-03-28 21:42:05 +0000 | [diff] [blame] | 50 | raise unittest.SkipTest(msg) |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 51 | raise |
| 52 | |
| 53 | # at least check that these methods can be invoked |
| 54 | dsp.bufsize() |
| 55 | dsp.obufcount() |
| 56 | dsp.obuffree() |
| 57 | dsp.getptr() |
| 58 | dsp.fileno() |
| 59 | |
| 60 | # Make sure the read-only attributes work. |
Antoine Pitrou | 7a5dc75 | 2008-08-17 00:38:32 +0000 | [diff] [blame] | 61 | self.failIf(dsp.closed) |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 62 | self.assertEqual(dsp.name, "/dev/dsp") |
| 63 | self.assertEqual(dsp.mode, "w", "bad dsp.mode: %r" % dsp.mode) |
| 64 | |
| 65 | # And make sure they're really read-only. |
| 66 | for attr in ('closed', 'name', 'mode'): |
| 67 | try: |
| 68 | setattr(dsp, attr, 42) |
Antoine Pitrou | 7a5dc75 | 2008-08-17 00:38:32 +0000 | [diff] [blame] | 69 | except (TypeError, AttributeError): |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 70 | pass |
| 71 | else: |
| 72 | self.fail("dsp.%s not read-only" % attr) |
| 73 | |
| 74 | # Compute expected running time of sound sample (in seconds). |
| 75 | expected_time = float(len(data)) / (ssize/8) / nchannels / rate |
| 76 | |
| 77 | # set parameters based on .au file headers |
| 78 | dsp.setparameters(AFMT_S16_NE, nchannels, rate) |
Antoine Pitrou | 7a5dc75 | 2008-08-17 00:38:32 +0000 | [diff] [blame] | 79 | self.assertTrue(abs(expected_time - 2.94) < 1e-2, expected_time) |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 80 | t1 = time.time() |
| 81 | dsp.write(data) |
| 82 | dsp.close() |
| 83 | t2 = time.time() |
| 84 | elapsed_time = t2 - t1 |
| 85 | |
| 86 | percent_diff = (abs(elapsed_time - expected_time) / expected_time) * 100 |
| 87 | self.failUnless(percent_diff <= 10.0, |
Guido van Rossum | e760631 | 2007-08-26 20:03:04 +0000 | [diff] [blame] | 88 | "elapsed time (%s) > 10%% off of expected time (%s)" % |
Neal Norwitz | 2bf4d5b | 2007-08-26 22:16:55 +0000 | [diff] [blame] | 89 | (elapsed_time, expected_time)) |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 90 | |
| 91 | def set_parameters(self, dsp): |
| 92 | # Two configurations for testing: |
| 93 | # config1 (8-bit, mono, 8 kHz) should work on even the most |
| 94 | # ancient and crufty sound card, but maybe not on special- |
| 95 | # purpose high-end hardware |
| 96 | # config2 (16-bit, stereo, 44.1kHz) should work on all but the |
| 97 | # most ancient and crufty hardware |
| 98 | config1 = (ossaudiodev.AFMT_U8, 1, 8000) |
| 99 | config2 = (AFMT_S16_NE, 2, 44100) |
| 100 | |
| 101 | for config in [config1, config2]: |
| 102 | (fmt, channels, rate) = config |
| 103 | if (dsp.setfmt(fmt) == fmt and |
| 104 | dsp.channels(channels) == channels and |
| 105 | dsp.speed(rate) == rate): |
| 106 | break |
| 107 | else: |
| 108 | raise RuntimeError("unable to set audio sampling parameters: " |
| 109 | "you must have really weird audio hardware") |
| 110 | |
| 111 | # setparameters() should be able to set this configuration in |
| 112 | # either strict or non-strict mode. |
| 113 | result = dsp.setparameters(fmt, channels, rate, False) |
| 114 | self.assertEqual(result, (fmt, channels, rate), |
| 115 | "setparameters%r: returned %r" % (config, result)) |
| 116 | |
| 117 | result = dsp.setparameters(fmt, channels, rate, True) |
| 118 | self.assertEqual(result, (fmt, channels, rate), |
| 119 | "setparameters%r: returned %r" % (config, result)) |
| 120 | |
| 121 | def set_bad_parameters(self, dsp): |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 122 | # Now try some configurations that are presumably bogus: eg. 300 |
| 123 | # channels currently exceeds even Hollywood's ambitions, and |
| 124 | # negative sampling rate is utter nonsense. setparameters() should |
| 125 | # accept these in non-strict mode, returning something other than |
| 126 | # was requested, but should barf in strict mode. |
| 127 | fmt = AFMT_S16_NE |
| 128 | rate = 44100 |
| 129 | channels = 2 |
| 130 | for config in [(fmt, 300, rate), # ridiculous nchannels |
| 131 | (fmt, -5, rate), # impossible nchannels |
| 132 | (fmt, channels, -50), # impossible rate |
| 133 | ]: |
| 134 | (fmt, channels, rate) = config |
| 135 | result = dsp.setparameters(fmt, channels, rate, False) |
| 136 | self.failIfEqual(result, config, |
| 137 | "unexpectedly got requested configuration") |
| 138 | |
| 139 | try: |
| 140 | result = dsp.setparameters(fmt, channels, rate, True) |
| 141 | except ossaudiodev.OSSAudioError as err: |
| 142 | pass |
| 143 | else: |
| 144 | self.fail("expected OSSAudioError") |
| 145 | |
| 146 | def test_playback(self): |
| 147 | sound_info = read_sound_file(findfile('audiotest.au')) |
| 148 | self.play_sound_file(*sound_info) |
| 149 | |
| 150 | def test_set_parameters(self): |
| 151 | dsp = ossaudiodev.open("w") |
| 152 | try: |
| 153 | self.set_parameters(dsp) |
| 154 | |
| 155 | # Disabled because it fails under Linux 2.6 with ALSA's OSS |
| 156 | # emulation layer. |
| 157 | #self.set_bad_parameters(dsp) |
| 158 | finally: |
| 159 | dsp.close() |
| 160 | self.failUnless(dsp.closed) |
| 161 | |
| 162 | |
| 163 | def test_main(): |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 164 | try: |
Greg Ward | 080c110 | 2003-05-29 00:23:17 +0000 | [diff] [blame] | 165 | dsp = ossaudiodev.open('w') |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 166 | except (ossaudiodev.error, IOError) as msg: |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 167 | if msg.args[0] in (errno.EACCES, errno.ENOENT, |
| 168 | errno.ENODEV, errno.EBUSY): |
Benjamin Peterson | e549ead | 2009-03-28 21:42:05 +0000 | [diff] [blame] | 169 | raise unittest.SkipTest(msg) |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 170 | raise |
Guido van Rossum | 23cfc98 | 2007-08-21 22:49:52 +0000 | [diff] [blame] | 171 | dsp.close() |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 172 | support.run_unittest(__name__) |
Greg Ward | 36dacfa | 2002-12-10 16:24:21 +0000 | [diff] [blame] | 173 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 174 | if __name__ == "__main__": |
| 175 | test_main() |