Lib/audiodev.py

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