sys/lib/python/audiodev.py

   1 """Classes for manipulating audio devices (currently only for Sun and SGI)"""
   2
   3 __all__ = ["error","AudioDev"]
   4
   5 class error(Exception):
   6     pass
   7
   8 class Play_Audio_sgi:
   9     # Private instance variables
  10 ##      if 0: access frameratelist, nchannelslist, sampwidthlist, oldparams, \
  11 ##                params, config, inited_outrate, inited_width, \
  12 ##                inited_nchannels, port, converter, classinited: private
  13
  14     classinited = 0
  15     frameratelist = nchannelslist = sampwidthlist = None
  16
  17     def initclass(self):
  18         import AL
  19         self.frameratelist = [
  20                   (48000, AL.RATE_48000),
  21                   (44100, AL.RATE_44100),
  22                   (32000, AL.RATE_32000),
  23                   (22050, AL.RATE_22050),
  24                   (16000, AL.RATE_16000),
  25                   (11025, AL.RATE_11025),
  26                   ( 8000,  AL.RATE_8000),
  27                   ]
  28         self.nchannelslist = [
  29                   (1, AL.MONO),
  30                   (2, AL.STEREO),
  31                   (4, AL.QUADRO),
  32                   ]
  33         self.sampwidthlist = [
  34                   (1, AL.SAMPLE_8),
  35                   (2, AL.SAMPLE_16),
  36                   (3, AL.SAMPLE_24),
  37                   ]
  38         self.classinited = 1
  39
  40     def __init__(self):
  41         import al, AL
  42         if not self.classinited:
  43             self.initclass()
  44         self.oldparams = []
  45         self.params = [AL.OUTPUT_RATE, 0]
  46         self.config = al.newconfig()
  47         self.inited_outrate = 0
  48         self.inited_width = 0
  49         self.inited_nchannels = 0
  50         self.converter = None
  51         self.port = None
  52         return
  53
  54     def __del__(self):
  55         if self.port:
  56             self.stop()
  57         if self.oldparams:
  58             import al, AL
  59             al.setparams(AL.DEFAULT_DEVICE, self.oldparams)
  60             self.oldparams = []
  61
  62     def wait(self):
  63         if not self.port:
  64             return
  65         import time
  66         while self.port.getfilled() > 0:
  67             time.sleep(0.1)
  68         self.stop()
  69
  70     def stop(self):
  71         if self.port:
  72             self.port.closeport()
  73             self.port = None
  74         if self.oldparams:
  75             import al, AL
  76             al.setparams(AL.DEFAULT_DEVICE, self.oldparams)
  77             self.oldparams = []
  78
  79     def setoutrate(self, rate):
  80         for (raw, cooked) in self.frameratelist:
  81             if rate == raw:
  82                 self.params[1] = cooked
  83                 self.inited_outrate = 1
  84                 break
  85         else:
  86             raise error, 'bad output rate'
  87
  88     def setsampwidth(self, width):
  89         for (raw, cooked) in self.sampwidthlist:
  90             if width == raw:
  91                 self.config.setwidth(cooked)
  92                 self.inited_width = 1
  93                 break
  94         else:
  95             if width == 0:
  96                 import AL
  97                 self.inited_width = 0
  98                 self.config.setwidth(AL.SAMPLE_16)
  99                 self.converter = self.ulaw2lin
 100             else:
 101                 raise error, 'bad sample width'
 102
 103     def setnchannels(self, nchannels):
 104         for (raw, cooked) in self.nchannelslist:
 105             if nchannels == raw:
 106                 self.config.setchannels(cooked)
 107                 self.inited_nchannels = 1
 108                 break
 109         else:
 110             raise error, 'bad # of channels'
 111
 112     def writeframes(self, data):
 113         if not (self.inited_outrate and self.inited_nchannels):
 114             raise error, 'params not specified'
 115         if not self.port:
 116             import al, AL
 117             self.port = al.openport('Python', 'w', self.config)
 118             self.oldparams = self.params[:]
 119             al.getparams(AL.DEFAULT_DEVICE, self.oldparams)
 120             al.setparams(AL.DEFAULT_DEVICE, self.params)
 121         if self.converter:
 122             data = self.converter(data)
 123         self.port.writesamps(data)
 124
 125     def getfilled(self):
 126         if self.port:
 127             return self.port.getfilled()
 128         else:
 129             return 0
 130
 131     def getfillable(self):
 132         if self.port:
 133             return self.port.getfillable()
 134         else:
 135             return self.config.getqueuesize()
 136
 137     # private methods
 138 ##      if 0: access *: private
 139
 140     def ulaw2lin(self, data):
 141         import audioop
 142         return audioop.ulaw2lin(data, 2)
 143
 144 class Play_Audio_sun:
 145 ##      if 0: access outrate, sampwidth, nchannels, inited_outrate, inited_width, \
 146 ##                inited_nchannels, converter: private
 147
 148     def __init__(self):
 149         self.outrate = 0
 150         self.sampwidth = 0
 151         self.nchannels = 0
 152         self.inited_outrate = 0
 153         self.inited_width = 0
 154         self.inited_nchannels = 0
 155         self.converter = None
 156         self.port = None
 157         return
 158
 159     def __del__(self):
 160         self.stop()
 161
 162     def setoutrate(self, rate):
 163         self.outrate = rate
 164         self.inited_outrate = 1
 165
 166     def setsampwidth(self, width):
 167         self.sampwidth = width
 168         self.inited_width = 1
 169
 170     def setnchannels(self, nchannels):
 171         self.nchannels = nchannels
 172         self.inited_nchannels = 1
 173
 174     def writeframes(self, data):
 175         if not (self.inited_outrate and self.inited_width and self.inited_nchannels):
 176             raise error, 'params not specified'
 177         if not self.port:
 178             import sunaudiodev, SUNAUDIODEV
 179             self.port = sunaudiodev.open('w')
 180             info = self.port.getinfo()
 181             info.o_sample_rate = self.outrate
 182             info.o_channels = self.nchannels
 183             if self.sampwidth == 0:
 184                 info.o_precision = 8
 185                 self.o_encoding = SUNAUDIODEV.ENCODING_ULAW
 186                 # XXX Hack, hack -- leave defaults
 187             else:
 188                 info.o_precision = 8 * self.sampwidth
 189                 info.o_encoding = SUNAUDIODEV.ENCODING_LINEAR
 190                 self.port.setinfo(info)
 191         if self.converter:
 192             data = self.converter(data)
 193         self.port.write(data)
 194
 195     def wait(self):
 196         if not self.port:
 197             return
 198         self.port.drain()
 199         self.stop()
 200
 201     def stop(self):
 202         if self.port:
 203             self.port.flush()
 204             self.port.close()
 205             self.port = None
 206
 207     def getfilled(self):
 208         if self.port:
 209             return self.port.obufcount()
 210         else:
 211             return 0
 212
 213 ##    # Nobody remembers what this method does, and it's broken. :-(
 214 ##    def getfillable(self):
 215 ##        return BUFFERSIZE - self.getfilled()
 216
 217 def AudioDev():
 218     # Dynamically try to import and use a platform specific module.
 219     try:
 220         import al
 221     except ImportError:
 222         try:
 223             import sunaudiodev
 224             return Play_Audio_sun()
 225         except ImportError:
 226             try:
 227                 import Audio_mac
 228             except ImportError:
 229                 raise error, 'no audio device'
 230             else:
 231                 return Audio_mac.Play_Audio_mac()
 232     else:
 233         return Play_Audio_sgi()
 234
 235 def test(fn = None):
 236     import sys
 237     if sys.argv[1:]:
 238         fn = sys.argv[1]
 239     else:
 240         fn = 'f:just samples:just.aif'
 241     import aifc
 242     af = aifc.open(fn, 'r')
 243     print fn, af.getparams()
 244     p = AudioDev()
 245     p.setoutrate(af.getframerate())
 246     p.setsampwidth(af.getsampwidth())
 247     p.setnchannels(af.getnchannels())
 248     BUFSIZ = af.getframerate()/af.getsampwidth()/af.getnchannels()
 249     while 1:
 250         data = af.readframes(BUFSIZ)
 251         if not data: break
 252         print len(data)
 253         p.writeframes(data)
 254     p.wait()
 255
 256 if __name__ == '__main__':
 257     test()