mirror of https://github.com/NekoX-Dev/NekoX.git
309 lines
12 KiB
C++
309 lines
12 KiB
C++
//
|
|
// libtgvoip is free and unencumbered public domain software.
|
|
// For more information, see http://unlicense.org or the UNLICENSE file
|
|
// you should have received with this source code distribution.
|
|
//
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include "AudioInputAudioUnitOSX.h"
|
|
#include "../../logging.h"
|
|
#include "../../audio/Resampler.h"
|
|
#include "../../VoIPController.h"
|
|
|
|
#define BUFFER_SIZE 960
|
|
#define CHECK_AU_ERROR(res, msg) if(res!=noErr){ LOGE("input: " msg": OSStatus=%d", (int)res); failed=true; return; }
|
|
|
|
#define kOutputBus 0
|
|
#define kInputBus 1
|
|
|
|
using namespace tgvoip;
|
|
using namespace tgvoip::audio;
|
|
|
|
AudioInputAudioUnitLegacy::AudioInputAudioUnitLegacy(std::string deviceID) : AudioInput(deviceID){
|
|
remainingDataSize=0;
|
|
isRecording=false;
|
|
|
|
inBufferList.mBuffers[0].mData=malloc(10240);
|
|
inBufferList.mBuffers[0].mDataByteSize=10240;
|
|
inBufferList.mNumberBuffers=1;
|
|
|
|
OSStatus status;
|
|
AudioComponentDescription inputDesc={
|
|
.componentType = kAudioUnitType_Output, .componentSubType = kAudioUnitSubType_HALOutput, .componentFlags = 0, .componentFlagsMask = 0,
|
|
.componentManufacturer = kAudioUnitManufacturer_Apple
|
|
};
|
|
AudioComponent component=AudioComponentFindNext(NULL, &inputDesc);
|
|
status=AudioComponentInstanceNew(component, &unit);
|
|
CHECK_AU_ERROR(status, "Error creating AudioUnit");
|
|
|
|
UInt32 flag=0;
|
|
status = AudioUnitSetProperty(unit, kAudioOutputUnitProperty_EnableIO, kAudioUnitScope_Output, kOutputBus, &flag, sizeof(flag));
|
|
CHECK_AU_ERROR(status, "Error enabling AudioUnit output");
|
|
flag=1;
|
|
status = AudioUnitSetProperty(unit, kAudioOutputUnitProperty_EnableIO, kAudioUnitScope_Input, kInputBus, &flag, sizeof(flag));
|
|
CHECK_AU_ERROR(status, "Error enabling AudioUnit input");
|
|
|
|
SetCurrentDevice(deviceID);
|
|
|
|
CFRunLoopRef theRunLoop = NULL;
|
|
AudioObjectPropertyAddress propertyAddress = { kAudioHardwarePropertyRunLoop,
|
|
kAudioObjectPropertyScopeGlobal,
|
|
kAudioObjectPropertyElementMaster };
|
|
status = AudioObjectSetPropertyData(kAudioObjectSystemObject, &propertyAddress, 0, NULL, sizeof(CFRunLoopRef), &theRunLoop);
|
|
|
|
propertyAddress.mSelector = kAudioHardwarePropertyDefaultInputDevice;
|
|
propertyAddress.mScope = kAudioObjectPropertyScopeGlobal;
|
|
propertyAddress.mElement = kAudioObjectPropertyElementMaster;
|
|
AudioObjectAddPropertyListener(kAudioObjectSystemObject, &propertyAddress, AudioInputAudioUnitLegacy::DefaultDeviceChangedCallback, this);
|
|
|
|
AURenderCallbackStruct callbackStruct;
|
|
callbackStruct.inputProc = AudioInputAudioUnitLegacy::BufferCallback;
|
|
callbackStruct.inputProcRefCon=this;
|
|
status = AudioUnitSetProperty(unit, kAudioOutputUnitProperty_SetInputCallback, kAudioUnitScope_Global, kInputBus, &callbackStruct, sizeof(callbackStruct));
|
|
CHECK_AU_ERROR(status, "Error setting input buffer callback");
|
|
status=AudioUnitInitialize(unit);
|
|
CHECK_AU_ERROR(status, "Error initializing unit");
|
|
}
|
|
|
|
AudioInputAudioUnitLegacy::~AudioInputAudioUnitLegacy(){
|
|
AudioObjectPropertyAddress propertyAddress;
|
|
propertyAddress.mSelector = kAudioHardwarePropertyDefaultInputDevice;
|
|
propertyAddress.mScope = kAudioObjectPropertyScopeGlobal;
|
|
propertyAddress.mElement = kAudioObjectPropertyElementMaster;
|
|
AudioObjectRemovePropertyListener(kAudioObjectSystemObject, &propertyAddress, AudioInputAudioUnitLegacy::DefaultDeviceChangedCallback, this);
|
|
|
|
AudioUnitUninitialize(unit);
|
|
AudioComponentInstanceDispose(unit);
|
|
free(inBufferList.mBuffers[0].mData);
|
|
}
|
|
|
|
void AudioInputAudioUnitLegacy::Start(){
|
|
isRecording=true;
|
|
OSStatus status=AudioOutputUnitStart(unit);
|
|
CHECK_AU_ERROR(status, "Error starting AudioUnit");
|
|
}
|
|
|
|
void AudioInputAudioUnitLegacy::Stop(){
|
|
isRecording=false;
|
|
OSStatus status=AudioOutputUnitStart(unit);
|
|
CHECK_AU_ERROR(status, "Error stopping AudioUnit");
|
|
}
|
|
|
|
OSStatus AudioInputAudioUnitLegacy::BufferCallback(void *inRefCon, AudioUnitRenderActionFlags *ioActionFlags, const AudioTimeStamp *inTimeStamp, UInt32 inBusNumber, UInt32 inNumberFrames, AudioBufferList *ioData){
|
|
AudioInputAudioUnitLegacy* input=(AudioInputAudioUnitLegacy*) inRefCon;
|
|
input->inBufferList.mBuffers[0].mDataByteSize=10240;
|
|
AudioUnitRender(input->unit, ioActionFlags, inTimeStamp, inBusNumber, inNumberFrames, &input->inBufferList);
|
|
input->HandleBufferCallback(&input->inBufferList);
|
|
return noErr;
|
|
}
|
|
|
|
void AudioInputAudioUnitLegacy::HandleBufferCallback(AudioBufferList *ioData){
|
|
int i;
|
|
for(i=0;i<ioData->mNumberBuffers;i++){
|
|
AudioBuffer buf=ioData->mBuffers[i];
|
|
size_t len=buf.mDataByteSize;
|
|
if(hardwareSampleRate!=48000){
|
|
len=tgvoip::audio::Resampler::Convert((int16_t*)buf.mData, (int16_t*)(remainingData+remainingDataSize), buf.mDataByteSize/2, (10240-(buf.mDataByteSize+remainingDataSize))/2, 48000, hardwareSampleRate)*2;
|
|
}else{
|
|
assert(remainingDataSize+buf.mDataByteSize<10240);
|
|
memcpy(remainingData+remainingDataSize, buf.mData, buf.mDataByteSize);
|
|
}
|
|
remainingDataSize+=len;
|
|
while(remainingDataSize>=BUFFER_SIZE*2){
|
|
InvokeCallback((unsigned char*)remainingData, BUFFER_SIZE*2);
|
|
remainingDataSize-=BUFFER_SIZE*2;
|
|
if(remainingDataSize>0){
|
|
memmove(remainingData, remainingData+(BUFFER_SIZE*2), remainingDataSize);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void AudioInputAudioUnitLegacy::EnumerateDevices(std::vector<AudioInputDevice>& devs){
|
|
AudioObjectPropertyAddress propertyAddress = {
|
|
kAudioHardwarePropertyDevices,
|
|
kAudioObjectPropertyScopeGlobal,
|
|
kAudioObjectPropertyElementMaster
|
|
};
|
|
|
|
UInt32 dataSize = 0;
|
|
OSStatus status = AudioObjectGetPropertyDataSize(kAudioObjectSystemObject, &propertyAddress, 0, NULL, &dataSize);
|
|
if(kAudioHardwareNoError != status) {
|
|
LOGE("AudioObjectGetPropertyDataSize (kAudioHardwarePropertyDevices) failed: %i", status);
|
|
return;
|
|
}
|
|
|
|
UInt32 deviceCount = (UInt32)(dataSize / sizeof(AudioDeviceID));
|
|
|
|
|
|
AudioDeviceID *audioDevices = (AudioDeviceID*)(malloc(dataSize));
|
|
|
|
status = AudioObjectGetPropertyData(kAudioObjectSystemObject, &propertyAddress, 0, NULL, &dataSize, audioDevices);
|
|
if(kAudioHardwareNoError != status) {
|
|
LOGE("AudioObjectGetPropertyData (kAudioHardwarePropertyDevices) failed: %i", status);
|
|
free(audioDevices);
|
|
audioDevices = NULL;
|
|
return;
|
|
}
|
|
|
|
|
|
// Iterate through all the devices and determine which are input-capable
|
|
propertyAddress.mScope = kAudioDevicePropertyScopeInput;
|
|
for(UInt32 i = 0; i < deviceCount; ++i) {
|
|
// Query device UID
|
|
CFStringRef deviceUID = NULL;
|
|
dataSize = sizeof(deviceUID);
|
|
propertyAddress.mSelector = kAudioDevicePropertyDeviceUID;
|
|
status = AudioObjectGetPropertyData(audioDevices[i], &propertyAddress, 0, NULL, &dataSize, &deviceUID);
|
|
if(kAudioHardwareNoError != status) {
|
|
LOGE("AudioObjectGetPropertyData (kAudioDevicePropertyDeviceUID) failed: %i", status);
|
|
continue;
|
|
}
|
|
|
|
// Query device name
|
|
CFStringRef deviceName = NULL;
|
|
dataSize = sizeof(deviceName);
|
|
propertyAddress.mSelector = kAudioDevicePropertyDeviceNameCFString;
|
|
status = AudioObjectGetPropertyData(audioDevices[i], &propertyAddress, 0, NULL, &dataSize, &deviceName);
|
|
if(kAudioHardwareNoError != status) {
|
|
LOGE("AudioObjectGetPropertyData (kAudioDevicePropertyDeviceNameCFString) failed: %i", status);
|
|
continue;
|
|
}
|
|
|
|
// Determine if the device is an input device (it is an input device if it has input channels)
|
|
dataSize = 0;
|
|
propertyAddress.mSelector = kAudioDevicePropertyStreamConfiguration;
|
|
status = AudioObjectGetPropertyDataSize(audioDevices[i], &propertyAddress, 0, NULL, &dataSize);
|
|
if(kAudioHardwareNoError != status) {
|
|
LOGE("AudioObjectGetPropertyDataSize (kAudioDevicePropertyStreamConfiguration) failed: %i", status);
|
|
continue;
|
|
}
|
|
|
|
AudioBufferList *bufferList = (AudioBufferList*)(malloc(dataSize));
|
|
|
|
status = AudioObjectGetPropertyData(audioDevices[i], &propertyAddress, 0, NULL, &dataSize, bufferList);
|
|
if(kAudioHardwareNoError != status || 0 == bufferList->mNumberBuffers) {
|
|
if(kAudioHardwareNoError != status)
|
|
LOGE("AudioObjectGetPropertyData (kAudioDevicePropertyStreamConfiguration) failed: %i", status);
|
|
free(bufferList);
|
|
bufferList = NULL;
|
|
continue;
|
|
}
|
|
|
|
free(bufferList);
|
|
bufferList = NULL;
|
|
|
|
AudioInputDevice dev;
|
|
char buf[1024];
|
|
CFStringGetCString(deviceName, buf, 1024, kCFStringEncodingUTF8);
|
|
dev.displayName=std::string(buf);
|
|
CFStringGetCString(deviceUID, buf, 1024, kCFStringEncodingUTF8);
|
|
dev.id=std::string(buf);
|
|
if(dev.id.rfind("VPAUAggregateAudioDevice-0x")==0)
|
|
continue;
|
|
devs.push_back(dev);
|
|
}
|
|
|
|
free(audioDevices);
|
|
audioDevices = NULL;
|
|
}
|
|
|
|
void AudioInputAudioUnitLegacy::SetCurrentDevice(std::string deviceID){
|
|
UInt32 size=sizeof(AudioDeviceID);
|
|
AudioDeviceID inputDevice=0;
|
|
OSStatus status;
|
|
|
|
if(deviceID=="default"){
|
|
AudioObjectPropertyAddress propertyAddress;
|
|
propertyAddress.mSelector = kAudioHardwarePropertyDefaultInputDevice;
|
|
propertyAddress.mScope = kAudioObjectPropertyScopeGlobal;
|
|
propertyAddress.mElement = kAudioObjectPropertyElementMaster;
|
|
UInt32 propsize = sizeof(AudioDeviceID);
|
|
status = AudioObjectGetPropertyData(kAudioObjectSystemObject, &propertyAddress, 0, NULL, &propsize, &inputDevice);
|
|
CHECK_AU_ERROR(status, "Error getting default input device");
|
|
}else{
|
|
AudioObjectPropertyAddress propertyAddress = {
|
|
kAudioHardwarePropertyDevices,
|
|
kAudioObjectPropertyScopeGlobal,
|
|
kAudioObjectPropertyElementMaster
|
|
};
|
|
UInt32 dataSize = 0;
|
|
status = AudioObjectGetPropertyDataSize(kAudioObjectSystemObject, &propertyAddress, 0, NULL, &dataSize);
|
|
CHECK_AU_ERROR(status, "Error getting devices size");
|
|
UInt32 deviceCount = (UInt32)(dataSize / sizeof(AudioDeviceID));
|
|
AudioDeviceID audioDevices[deviceCount];
|
|
status = AudioObjectGetPropertyData(kAudioObjectSystemObject, &propertyAddress, 0, NULL, &dataSize, audioDevices);
|
|
CHECK_AU_ERROR(status, "Error getting device list");
|
|
for(UInt32 i = 0; i < deviceCount; ++i) {
|
|
// Query device UID
|
|
CFStringRef deviceUID = NULL;
|
|
dataSize = sizeof(deviceUID);
|
|
propertyAddress.mSelector = kAudioDevicePropertyDeviceUID;
|
|
status = AudioObjectGetPropertyData(audioDevices[i], &propertyAddress, 0, NULL, &dataSize, &deviceUID);
|
|
CHECK_AU_ERROR(status, "Error getting device uid");
|
|
char buf[1024];
|
|
CFStringGetCString(deviceUID, buf, 1024, kCFStringEncodingUTF8);
|
|
if(deviceID==buf){
|
|
LOGV("Found device for id %s", buf);
|
|
inputDevice=audioDevices[i];
|
|
break;
|
|
}
|
|
}
|
|
if(!inputDevice){
|
|
LOGW("Requested device not found, using default");
|
|
SetCurrentDevice("default");
|
|
return;
|
|
}
|
|
}
|
|
|
|
status =AudioUnitSetProperty(unit,
|
|
kAudioOutputUnitProperty_CurrentDevice,
|
|
kAudioUnitScope_Global,
|
|
kInputBus,
|
|
&inputDevice,
|
|
size);
|
|
CHECK_AU_ERROR(status, "Error setting input device");
|
|
|
|
AudioStreamBasicDescription hardwareFormat;
|
|
size=sizeof(hardwareFormat);
|
|
status=AudioUnitGetProperty(unit, kAudioUnitProperty_StreamFormat, kAudioUnitScope_Input, kInputBus, &hardwareFormat, &size);
|
|
CHECK_AU_ERROR(status, "Error getting hardware format");
|
|
hardwareSampleRate=hardwareFormat.mSampleRate;
|
|
|
|
AudioStreamBasicDescription desiredFormat={
|
|
.mSampleRate=hardwareFormat.mSampleRate, .mFormatID=kAudioFormatLinearPCM, .mFormatFlags=kAudioFormatFlagIsSignedInteger | kAudioFormatFlagIsPacked | kAudioFormatFlagsNativeEndian,
|
|
.mFramesPerPacket=1, .mChannelsPerFrame=1, .mBitsPerChannel=16, .mBytesPerPacket=2, .mBytesPerFrame=2
|
|
};
|
|
|
|
status=AudioUnitSetProperty(unit, kAudioUnitProperty_StreamFormat, kAudioUnitScope_Output, kInputBus, &desiredFormat, sizeof(desiredFormat));
|
|
CHECK_AU_ERROR(status, "Error setting format");
|
|
|
|
LOGD("Switched capture device, new sample rate %d", hardwareSampleRate);
|
|
|
|
this->currentDevice=deviceID;
|
|
|
|
AudioObjectPropertyAddress propertyAddress = {
|
|
kAudioDevicePropertyBufferFrameSize,
|
|
kAudioObjectPropertyScopeGlobal,
|
|
kAudioObjectPropertyElementMaster
|
|
};
|
|
size=4;
|
|
UInt32 bufferFrameSize;
|
|
status=AudioObjectGetPropertyData(inputDevice, &propertyAddress, 0, NULL, &size, &bufferFrameSize);
|
|
if(status==noErr){
|
|
estimatedDelay=bufferFrameSize/48;
|
|
LOGD("CoreAudio buffer size for output device is %u frames (%u ms)", bufferFrameSize, estimatedDelay);
|
|
}
|
|
}
|
|
|
|
OSStatus AudioInputAudioUnitLegacy::DefaultDeviceChangedCallback(AudioObjectID inObjectID, UInt32 inNumberAddresses, const AudioObjectPropertyAddress *inAddresses, void *inClientData){
|
|
LOGV("System default input device changed");
|
|
AudioInputAudioUnitLegacy* self=(AudioInputAudioUnitLegacy*)inClientData;
|
|
if(self->currentDevice=="default"){
|
|
self->SetCurrentDevice(self->currentDevice);
|
|
}
|
|
return noErr;
|
|
}
|