twblue/src/sound_lib/input.py

from ctypes import string_at
import platform
import wave

from .external.pybass import *
if platform.system() == 'Windows':
 from .external.pybasswasapi import *
from . import config
from .main import bass_call, bass_call_0

class Input (object):

 def __init__ (self, device=-1):
  bass_call(BASS_RecordInit, device)
  self._device = device
  self.config = config.BassConfig()

 def free(self):
  """Frees all resources used by the recording device."""
  return bass_call(BASS_RecordFree)

 def get_device(self):
    return bass_call_0(BASS_RecordGetDevice)

 def set_device(self, device):
  if device == self._device:
   return
  self.free()
  self.__init__(device=device)

 device = property(fget=get_device, fset=set_device)

 @staticmethod
 def get_device_names():
  """Convenience method that returns a list of device names that are considered
 valid by bass.
	
  Parameters: none.
  returns: list of devices, 0-indexed.
  """
  result = ['Default']
  info = BASS_DEVICEINFO()
  count = 0
  while BASS_RecordGetDeviceInfo(count, ctypes.byref(info)):
   if info.flags & BASS_DEVICE_ENABLED:
    retrieved = info.name
    if platform.system() == 'Windows':
     retrieved = retrieved.decode('mbcs')
    elif platform.system() == 'Darwin':
     retrieved = retrieved.decode('utf-8')
    retrieved = retrieved.replace('(', '').replace(')', '').strip()
    result.append(retrieved)
   count += 1
  return result

 def find_device_by_name(self, name):
  return self.get_device_names().index(name) - 1

 def find_default_device(self):
  return -1

 def find_user_provided_device(self, device_name):
  try:
   return self.find_device_by_name(device_name)
  except ValueError:
   return self.find_default_device()


class WASAPIInput(object):

 def __init__(self, device=-2, frequency=0, channels=0, flags=0, buffer=0.0, period=0.0, callback=None):
  if callback is None:
   callback = lambda buffer, length, user: True
  self.proc = WASAPIPROC(callback)
  bass_call(BASS_WASAPI_Init, device, frequency, channels, flags, buffer, period, self.proc, None)


 def free(self):
  bass_call(BASS_WASAPI_Free)

 def set_device(self, device):
  bass_call(BASS_WASAPI_SetDevice, device)

 def get_device(self):
  return bass_call_0(BASS_WASAPI_GetDevice)

 device = property(fget=get_device, fset=set_device)

 def start(self):
  return bass_call(BASS_WASAPI_Start)

 def stop(self, reset=False):
  return bass_call(BASS_WASAPI_Stop, reset)