Initial commit of OpenSPARC T2 architecture model.

[OpenSPARC-T2-SAM] / sam-t2 / devtools / v9 / lib / python2.4 / test / test_ossaudiodev.py

from test import test_support
test_support.requires('audio')

from test.test_support import verbose, findfile, TestFailed, TestSkipped

import errno
import fcntl
import ossaudiodev
import os
import sys
import select
import sunaudio
import time
import audioop

# Arggh, AFMT_S16_NE not defined on all platforms -- seems to be a
# fairly recent addition to OSS.
try:
   from ossaudiodev import AFMT_S16_NE
except ImportError:
   if sys.byteorder == "little":
       AFMT_S16_NE = ossaudiodev.AFMT_S16_LE
   else:
       AFMT_S16_NE = ossaudiodev.AFMT_S16_BE


SND_FORMAT_MULAW_8 = 1

def read_sound_file(path):
   fp = open(path, 'rb')
   size, enc, rate, nchannels, extra = sunaudio.gethdr(fp)
   data = fp.read()
   fp.close()

   if enc != SND_FORMAT_MULAW_8:
       print "Expect .au file with 8-bit mu-law samples"
       return

   # Convert the data to 16-bit signed.
   data = audioop.ulaw2lin(data, 2)
   return (data, rate, 16, nchannels)


def play_sound_file(data, rate, ssize, nchannels):
   try:
       dsp = ossaudiodev.open('w')
   except IOError, msg:
       if msg[0] in (errno.EACCES, errno.ENODEV, errno.EBUSY):
           raise TestSkipped, msg
       raise TestFailed, msg

   # at least check that these methods can be invoked
   dsp.bufsize()
   dsp.obufcount()
   dsp.obuffree()
   dsp.getptr()
   dsp.fileno()

   # set parameters based on .au file headers
   dsp.setparameters(AFMT_S16_NE, nchannels, rate)
   t1 = time.time()
   print "playing test sound file..."
   dsp.write(data)
   dsp.close()
   t2 = time.time()
   print "elapsed time: %.1f sec" % (t2-t1)

def test_setparameters(dsp):
   # Two configurations for testing:
   #   config1 (8-bit, mono, 8 kHz) should work on even the most
   #      ancient and crufty sound card, but maybe not on special-
   #      purpose high-end hardware
   #   config2 (16-bit, stereo, 44.1kHz) should work on all but the
   #      most ancient and crufty hardware
   config1 = (ossaudiodev.AFMT_U8, 1, 8000)
   config2 = (AFMT_S16_NE, 2, 44100)

   for config in [config1, config2]:
       (fmt, channels, rate) = config
       if (dsp.setfmt(fmt) == fmt and
           dsp.channels(channels) == channels and
           dsp.speed(rate) == rate):
           break
   else:
       raise RuntimeError("unable to set audio sampling parameters: "
                          "you must have really weird audio hardware")

   # setparameters() should be able to set this configuration in
   # either strict or non-strict mode.
   result = dsp.setparameters(fmt, channels, rate, False)
   assert result == (fmt, channels, rate), \
          "setparameters%r: returned %r" % (config + result)
   result = dsp.setparameters(fmt, channels, rate, True)
   assert result == (fmt, channels, rate), \
          "setparameters%r: returned %r" % (config + result)

def test_bad_setparameters(dsp):

   # Now try some configurations that are presumably bogus: eg. 300
   # channels currently exceeds even Hollywood's ambitions, and
   # negative sampling rate is utter nonsense.  setparameters() should
   # accept these in non-strict mode, returning something other than
   # was requested, but should barf in strict mode.
   fmt = AFMT_S16_NE
   rate = 44100
   channels = 2
   for config in [(fmt, 300, rate),       # ridiculous nchannels
                  (fmt, -5, rate),        # impossible nchannels
                  (fmt, channels, -50),   # impossible rate
                 ]:
       (fmt, channels, rate) = config
       result = dsp.setparameters(fmt, channels, rate, False)
       assert result != config, \
              "setparameters: unexpectedly got requested configuration"

       try:
           result = dsp.setparameters(fmt, channels, rate, True)
           raise AssertionError("setparameters: expected OSSAudioError")
       except ossaudiodev.OSSAudioError, err:
           print "setparameters: got OSSAudioError as expected"

def test():
   (data, rate, ssize, nchannels) = read_sound_file(findfile('audiotest.au'))
   play_sound_file(data, rate, ssize, nchannels)

   dsp = ossaudiodev.open("w")
   try:
       test_setparameters(dsp)

       # Disabled because it fails under Linux 2.6 with ALSA's OSS
       # emulation layer.
       #test_bad_setparameters(dsp)
   finally:
       dsp.close()

test()