356 lines
12 KiB
Python
356 lines
12 KiB
Python
|
import unittest
|
||
|
|
import struct
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
import pywintypes
|
||
|
|
import win32event, win32api
|
||
|
|
import os
|
||
|
|
from pywin32_testutil import str2bytes, TestSkipped
|
||
|
|
import win32com.directsound.directsound as ds
|
||
|
|
# next two lines are for for debugging:
|
||
|
|
# import win32com
|
||
|
|
# import directsound as ds
|
||
|
|
|
||
|
|
WAV_FORMAT_PCM = 1
|
||
|
|
WAV_HEADER_SIZE = struct.calcsize('<4sl4s4slhhllhh4sl')
|
||
|
|
|
||
|
|
def wav_header_unpack(data):
|
||
|
|
(riff, riffsize, wave, fmt, fmtsize, format, nchannels, samplespersecond,
|
||
|
|
datarate, blockalign, bitspersample, data, datalength) \
|
||
|
|
= struct.unpack('<4sl4s4slhhllhh4sl', data)
|
||
|
|
|
||
|
|
if riff != str2bytes('RIFF'):
|
||
|
|
raise ValueError('invalid wav header')
|
||
|
|
|
||
|
|
if fmtsize != 16 or fmt != str2bytes('fmt ') or str2bytes(data) != 'data':
|
||
|
|
# fmt chuck is not first chunk, directly followed by data chuck
|
||
|
|
# It is nowhere required that they are, it is just very common
|
||
|
|
raise ValueError('cannot understand wav header')
|
||
|
|
|
||
|
|
wfx = pywintypes.WAVEFORMATEX()
|
||
|
|
wfx.wFormatTag = format
|
||
|
|
wfx.nChannels = nchannels
|
||
|
|
wfx.nSamplesPerSec = samplespersecond
|
||
|
|
wfx.nAvgBytesPerSec = datarate
|
||
|
|
wfx.nBlockAlign = blockalign
|
||
|
|
wfx.wBitsPerSample = bitspersample
|
||
|
|
|
||
|
|
return wfx, datalength
|
||
|
|
|
||
|
|
def wav_header_pack(wfx, datasize):
|
||
|
|
return struct.pack('<4sl4s4slhhllhh4sl', 'RIFF', 36 + datasize,
|
||
|
|
'WAVE', 'fmt ', 16,
|
||
|
|
wfx.wFormatTag, wfx.nChannels, wfx.nSamplesPerSec,
|
||
|
|
wfx.nAvgBytesPerSec, wfx.nBlockAlign,
|
||
|
|
wfx.wBitsPerSample, 'data', datasize);
|
||
|
|
|
||
|
|
class WAVEFORMATTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'WAVEFORMATEX type'
|
||
|
|
w = pywintypes.WAVEFORMATEX()
|
||
|
|
self.failUnless(type(w) == pywintypes.WAVEFORMATEXType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'WAVEFORMATEX attribute access'
|
||
|
|
# A wav header for a soundfile from a CD should look like this...
|
||
|
|
w = pywintypes.WAVEFORMATEX()
|
||
|
|
w.wFormatTag = pywintypes.WAVE_FORMAT_PCM
|
||
|
|
w.nChannels = 2
|
||
|
|
w.nSamplesPerSec = 44100
|
||
|
|
w.nAvgBytesPerSec = 176400
|
||
|
|
w.nBlockAlign = 4
|
||
|
|
w.wBitsPerSample = 16
|
||
|
|
|
||
|
|
self.failUnless(w.wFormatTag == 1)
|
||
|
|
self.failUnless(w.nChannels == 2)
|
||
|
|
self.failUnless(w.nSamplesPerSec == 44100)
|
||
|
|
self.failUnless(w.nAvgBytesPerSec == 176400)
|
||
|
|
self.failUnless(w.nBlockAlign == 4)
|
||
|
|
self.failUnless(w.wBitsPerSample == 16)
|
||
|
|
|
||
|
|
class DSCAPSTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSCAPS type'
|
||
|
|
c = ds.DSCAPS()
|
||
|
|
self.failUnless(type(c) == ds.DSCAPSType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSCAPS attribute access'
|
||
|
|
c = ds.DSCAPS()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwMinSecondarySampleRate = 2
|
||
|
|
c.dwMaxSecondarySampleRate = 3
|
||
|
|
c.dwPrimaryBuffers = 4
|
||
|
|
c.dwMaxHwMixingAllBuffers = 5
|
||
|
|
c.dwMaxHwMixingStaticBuffers = 6
|
||
|
|
c.dwMaxHwMixingStreamingBuffers = 7
|
||
|
|
c.dwFreeHwMixingAllBuffers = 8
|
||
|
|
c.dwFreeHwMixingStaticBuffers = 9
|
||
|
|
c.dwFreeHwMixingStreamingBuffers = 10
|
||
|
|
c.dwMaxHw3DAllBuffers = 11
|
||
|
|
c.dwMaxHw3DStaticBuffers = 12
|
||
|
|
c.dwMaxHw3DStreamingBuffers = 13
|
||
|
|
c.dwFreeHw3DAllBuffers = 14
|
||
|
|
c.dwFreeHw3DStaticBuffers = 15
|
||
|
|
c.dwFreeHw3DStreamingBuffers = 16
|
||
|
|
c.dwTotalHwMemBytes = 17
|
||
|
|
c.dwFreeHwMemBytes = 18
|
||
|
|
c.dwMaxContigFreeHwMemBytes = 19
|
||
|
|
c.dwUnlockTransferRateHwBuffers = 20
|
||
|
|
c.dwPlayCpuOverheadSwBuffers = 21
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwMinSecondarySampleRate == 2)
|
||
|
|
self.failUnless(c.dwMaxSecondarySampleRate == 3)
|
||
|
|
self.failUnless(c.dwPrimaryBuffers == 4)
|
||
|
|
self.failUnless(c.dwMaxHwMixingAllBuffers == 5)
|
||
|
|
self.failUnless(c.dwMaxHwMixingStaticBuffers == 6)
|
||
|
|
self.failUnless(c.dwMaxHwMixingStreamingBuffers == 7)
|
||
|
|
self.failUnless(c.dwFreeHwMixingAllBuffers == 8)
|
||
|
|
self.failUnless(c.dwFreeHwMixingStaticBuffers == 9)
|
||
|
|
self.failUnless(c.dwFreeHwMixingStreamingBuffers == 10)
|
||
|
|
self.failUnless(c.dwMaxHw3DAllBuffers == 11)
|
||
|
|
self.failUnless(c.dwMaxHw3DStaticBuffers == 12)
|
||
|
|
self.failUnless(c.dwMaxHw3DStreamingBuffers == 13)
|
||
|
|
self.failUnless(c.dwFreeHw3DAllBuffers == 14)
|
||
|
|
self.failUnless(c.dwFreeHw3DStaticBuffers == 15)
|
||
|
|
self.failUnless(c.dwFreeHw3DStreamingBuffers == 16)
|
||
|
|
self.failUnless(c.dwTotalHwMemBytes == 17)
|
||
|
|
self.failUnless(c.dwFreeHwMemBytes == 18)
|
||
|
|
self.failUnless(c.dwMaxContigFreeHwMemBytes == 19)
|
||
|
|
self.failUnless(c.dwUnlockTransferRateHwBuffers == 20)
|
||
|
|
self.failUnless(c.dwPlayCpuOverheadSwBuffers == 21)
|
||
|
|
|
||
|
|
class DSBCAPSTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSBCAPS type'
|
||
|
|
c = ds.DSBCAPS()
|
||
|
|
self.failUnless(type(c) == ds.DSBCAPSType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSBCAPS attribute access'
|
||
|
|
c = ds.DSBCAPS()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwBufferBytes = 2
|
||
|
|
c.dwUnlockTransferRate = 3
|
||
|
|
c.dwPlayCpuOverhead = 4
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwBufferBytes == 2)
|
||
|
|
self.failUnless(c.dwUnlockTransferRate == 3)
|
||
|
|
self.failUnless(c.dwPlayCpuOverhead == 4)
|
||
|
|
|
||
|
|
class DSCCAPSTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSCCAPS type'
|
||
|
|
c = ds.DSCCAPS()
|
||
|
|
self.failUnless(type(c) == ds.DSCCAPSType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSCCAPS attribute access'
|
||
|
|
c = ds.DSCCAPS()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwFormats = 2
|
||
|
|
c.dwChannels = 4
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwFormats == 2)
|
||
|
|
self.failUnless(c.dwChannels == 4)
|
||
|
|
|
||
|
|
class DSCBCAPSTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSCBCAPS type'
|
||
|
|
c = ds.DSCBCAPS()
|
||
|
|
self.failUnless(type(c) == ds.DSCBCAPSType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSCBCAPS attribute access'
|
||
|
|
c = ds.DSCBCAPS()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwBufferBytes = 2
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwBufferBytes == 2)
|
||
|
|
|
||
|
|
class DSBUFFERDESCTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSBUFFERDESC type'
|
||
|
|
c = ds.DSBUFFERDESC()
|
||
|
|
self.failUnless(type(c) == ds.DSBUFFERDESCType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSBUFFERDESC attribute access'
|
||
|
|
c = ds.DSBUFFERDESC()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwBufferBytes = 2
|
||
|
|
c.lpwfxFormat = pywintypes.WAVEFORMATEX()
|
||
|
|
c.lpwfxFormat.wFormatTag = pywintypes.WAVE_FORMAT_PCM
|
||
|
|
c.lpwfxFormat.nChannels = 2
|
||
|
|
c.lpwfxFormat.nSamplesPerSec = 44100
|
||
|
|
c.lpwfxFormat.nAvgBytesPerSec = 176400
|
||
|
|
c.lpwfxFormat.nBlockAlign = 4
|
||
|
|
c.lpwfxFormat.wBitsPerSample = 16
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwBufferBytes == 2)
|
||
|
|
self.failUnless(c.lpwfxFormat.wFormatTag == 1)
|
||
|
|
self.failUnless(c.lpwfxFormat.nChannels == 2)
|
||
|
|
self.failUnless(c.lpwfxFormat.nSamplesPerSec == 44100)
|
||
|
|
self.failUnless(c.lpwfxFormat.nAvgBytesPerSec == 176400)
|
||
|
|
self.failUnless(c.lpwfxFormat.nBlockAlign == 4)
|
||
|
|
self.failUnless(c.lpwfxFormat.wBitsPerSample == 16)
|
||
|
|
|
||
|
|
def invalid_format(self, c):
|
||
|
|
c.lpwfxFormat = 17
|
||
|
|
|
||
|
|
def test_3_invalid_format(self):
|
||
|
|
'DSBUFFERDESC invalid lpwfxFormat assignment'
|
||
|
|
c = ds.DSBUFFERDESC()
|
||
|
|
self.failUnlessRaises(ValueError, self.invalid_format, c)
|
||
|
|
|
||
|
|
class DSCBUFFERDESCTest(unittest.TestCase):
|
||
|
|
def test_1_Type(self):
|
||
|
|
'DSCBUFFERDESC type'
|
||
|
|
c = ds.DSCBUFFERDESC()
|
||
|
|
self.failUnless(type(c) == ds.DSCBUFFERDESCType)
|
||
|
|
|
||
|
|
def test_2_Attr(self):
|
||
|
|
'DSCBUFFERDESC attribute access'
|
||
|
|
c = ds.DSCBUFFERDESC()
|
||
|
|
c.dwFlags = 1
|
||
|
|
c.dwBufferBytes = 2
|
||
|
|
c.lpwfxFormat = pywintypes.WAVEFORMATEX()
|
||
|
|
c.lpwfxFormat.wFormatTag = pywintypes.WAVE_FORMAT_PCM
|
||
|
|
c.lpwfxFormat.nChannels = 2
|
||
|
|
c.lpwfxFormat.nSamplesPerSec = 44100
|
||
|
|
c.lpwfxFormat.nAvgBytesPerSec = 176400
|
||
|
|
c.lpwfxFormat.nBlockAlign = 4
|
||
|
|
c.lpwfxFormat.wBitsPerSample = 16
|
||
|
|
|
||
|
|
self.failUnless(c.dwFlags == 1)
|
||
|
|
self.failUnless(c.dwBufferBytes == 2)
|
||
|
|
self.failUnless(c.lpwfxFormat.wFormatTag == 1)
|
||
|
|
self.failUnless(c.lpwfxFormat.nChannels == 2)
|
||
|
|
self.failUnless(c.lpwfxFormat.nSamplesPerSec == 44100)
|
||
|
|
self.failUnless(c.lpwfxFormat.nAvgBytesPerSec == 176400)
|
||
|
|
self.failUnless(c.lpwfxFormat.nBlockAlign == 4)
|
||
|
|
self.failUnless(c.lpwfxFormat.wBitsPerSample == 16)
|
||
|
|
|
||
|
|
def invalid_format(self, c):
|
||
|
|
c.lpwfxFormat = 17
|
||
|
|
|
||
|
|
def test_3_invalid_format(self):
|
||
|
|
'DSCBUFFERDESC invalid lpwfxFormat assignment'
|
||
|
|
c = ds.DSCBUFFERDESC()
|
||
|
|
self.failUnlessRaises(ValueError, self.invalid_format, c)
|
||
|
|
|
||
|
|
class DirectSoundTest(unittest.TestCase):
|
||
|
|
# basic tests - mostly just exercise the functions
|
||
|
|
|
||
|
|
def testEnumerate(self):
|
||
|
|
'''DirectSoundEnumerate() sanity tests'''
|
||
|
|
|
||
|
|
devices = ds.DirectSoundEnumerate()
|
||
|
|
# this might fail on machines without a sound card
|
||
|
|
self.failUnless(len(devices))
|
||
|
|
# if we have an entry, it must be a tuple of size 3
|
||
|
|
self.failUnless(len(devices[0]) == 3)
|
||
|
|
|
||
|
|
def testCreate(self):
|
||
|
|
'''DirectSoundCreate()'''
|
||
|
|
d = ds.DirectSoundCreate(None, None)
|
||
|
|
|
||
|
|
def testPlay(self):
|
||
|
|
'''Mesdames et Messieurs, la cour de Devin Dazzle'''
|
||
|
|
# look for the test file in various places
|
||
|
|
candidates = [
|
||
|
|
os.path.dirname(__file__),
|
||
|
|
os.path.dirname(sys.argv[0]),
|
||
|
|
# relative to 'testall.py' in the win32com test suite.
|
||
|
|
os.path.join(os.path.dirname(sys.argv[0]),
|
||
|
|
'../../win32comext/directsound/test'),
|
||
|
|
'.',
|
||
|
|
]
|
||
|
|
for candidate in candidates:
|
||
|
|
fname=os.path.join(candidate, "01-Intro.wav")
|
||
|
|
if os.path.isfile(fname):
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
raise TestSkipped("Can't find test .wav file to play")
|
||
|
|
|
||
|
|
f = open(fname, 'rb')
|
||
|
|
hdr = f.read(WAV_HEADER_SIZE)
|
||
|
|
wfx, size = wav_header_unpack(hdr)
|
||
|
|
|
||
|
|
d = ds.DirectSoundCreate(None, None)
|
||
|
|
d.SetCooperativeLevel(None, ds.DSSCL_PRIORITY)
|
||
|
|
|
||
|
|
sdesc = ds.DSBUFFERDESC()
|
||
|
|
sdesc.dwFlags = ds.DSBCAPS_STICKYFOCUS | ds.DSBCAPS_CTRLPOSITIONNOTIFY
|
||
|
|
sdesc.dwBufferBytes = size
|
||
|
|
sdesc.lpwfxFormat = wfx
|
||
|
|
|
||
|
|
buffer = d.CreateSoundBuffer(sdesc, None)
|
||
|
|
|
||
|
|
event = win32event.CreateEvent(None, 0, 0, None)
|
||
|
|
notify = buffer.QueryInterface(ds.IID_IDirectSoundNotify)
|
||
|
|
|
||
|
|
notify.SetNotificationPositions((ds.DSBPN_OFFSETSTOP, event))
|
||
|
|
|
||
|
|
buffer.Update(0, f.read(size))
|
||
|
|
|
||
|
|
buffer.Play(0)
|
||
|
|
|
||
|
|
win32event.WaitForSingleObject(event, -1)
|
||
|
|
|
||
|
|
class DirectSoundCaptureTest(unittest.TestCase):
|
||
|
|
# basic tests - mostly just exercise the functions
|
||
|
|
|
||
|
|
def testEnumerate(self):
|
||
|
|
'''DirectSoundCaptureEnumerate() sanity tests'''
|
||
|
|
|
||
|
|
devices = ds.DirectSoundCaptureEnumerate()
|
||
|
|
# this might fail on machines without a sound card
|
||
|
|
self.failUnless(len(devices))
|
||
|
|
# if we have an entry, it must be a tuple of size 3
|
||
|
|
self.failUnless(len(devices[0]) == 3)
|
||
|
|
|
||
|
|
def testCreate(self):
|
||
|
|
'''DirectSoundCreate()'''
|
||
|
|
d = ds.DirectSoundCaptureCreate(None, None)
|
||
|
|
|
||
|
|
def testRecord(self):
|
||
|
|
d = ds.DirectSoundCaptureCreate(None, None)
|
||
|
|
|
||
|
|
sdesc = ds.DSCBUFFERDESC()
|
||
|
|
sdesc.dwBufferBytes = 352800 # 2 seconds
|
||
|
|
sdesc.lpwfxFormat = pywintypes.WAVEFORMATEX()
|
||
|
|
sdesc.lpwfxFormat.wFormatTag = pywintypes.WAVE_FORMAT_PCM
|
||
|
|
sdesc.lpwfxFormat.nChannels = 2
|
||
|
|
sdesc.lpwfxFormat.nSamplesPerSec = 44100
|
||
|
|
sdesc.lpwfxFormat.nAvgBytesPerSec = 176400
|
||
|
|
sdesc.lpwfxFormat.nBlockAlign = 4
|
||
|
|
sdesc.lpwfxFormat.wBitsPerSample = 16
|
||
|
|
|
||
|
|
buffer = d.CreateCaptureBuffer(sdesc)
|
||
|
|
|
||
|
|
event = win32event.CreateEvent(None, 0, 0, None)
|
||
|
|
notify = buffer.QueryInterface(ds.IID_IDirectSoundNotify)
|
||
|
|
|
||
|
|
notify.SetNotificationPositions((ds.DSBPN_OFFSETSTOP, event))
|
||
|
|
|
||
|
|
buffer.Start(0)
|
||
|
|
|
||
|
|
win32event.WaitForSingleObject(event, -1)
|
||
|
|
event.Close()
|
||
|
|
|
||
|
|
data = buffer.Update(0, 352800)
|
||
|
|
fname=os.path.join(win32api.GetTempPath(), 'test_directsound_record.wav')
|
||
|
|
f = open(fname, 'wb')
|
||
|
|
f.write(wav_header_pack(sdesc.lpwfxFormat, 352800))
|
||
|
|
f.write(data)
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|