Repo created

TMessagesProj/jni/voip/libtgvoip/os/android/AudioInputOpenSLES.cpp

@@ -0,0 +1,137 @@
+//
+// 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 <assert.h>
+
+#include "AudioInputOpenSLES.h"
+#include "../../logging.h"
+#include "OpenSLEngineWrapper.h"
+
+#define CHECK_SL_ERROR(res, msg) if(res!=SL_RESULT_SUCCESS){ LOGE(msg); return; }
+#define BUFFER_SIZE 960 // 20 ms
+
+using namespace tgvoip;
+using namespace tgvoip::audio;
+
+unsigned int AudioInputOpenSLES::nativeBufferSize;
+
+AudioInputOpenSLES::AudioInputOpenSLES(){
+	slEngine=OpenSLEngineWrapper::CreateEngine();
+
+	LOGI("Native buffer size is %u samples", nativeBufferSize);
+	if(nativeBufferSize<BUFFER_SIZE && BUFFER_SIZE % nativeBufferSize!=0){
+		LOGE("20ms is not divisible by native buffer size!!");
+	}else if(nativeBufferSize>BUFFER_SIZE && nativeBufferSize%BUFFER_SIZE!=0){
+		LOGE("native buffer size is not multiple of 20ms!!");
+		nativeBufferSize+=nativeBufferSize%BUFFER_SIZE;
+	}
+	if(nativeBufferSize==BUFFER_SIZE)
+		nativeBufferSize*=2;
+	LOGI("Adjusted native buffer size is %u", nativeBufferSize);
+
+	buffer=(int16_t*)calloc(BUFFER_SIZE, sizeof(int16_t));
+	nativeBuffer=(int16_t*)calloc((size_t) nativeBufferSize, sizeof(int16_t));
+	slRecorderObj=NULL;
+}
+
+AudioInputOpenSLES::~AudioInputOpenSLES(){
+	//Stop();
+	(*slBufferQueue)->Clear(slBufferQueue);
+	(*slRecorderObj)->Destroy(slRecorderObj);
+	slRecorderObj=NULL;
+	slRecorder=NULL;
+	slBufferQueue=NULL;
+	slEngine=NULL;
+	OpenSLEngineWrapper::DestroyEngine();
+	free(buffer);
+	buffer=NULL;
+	free(nativeBuffer);
+	nativeBuffer=NULL;
+}
+
+void AudioInputOpenSLES::BufferCallback(SLAndroidSimpleBufferQueueItf bq, void *context){
+	((AudioInputOpenSLES*)context)->HandleSLCallback();
+}
+
+void AudioInputOpenSLES::Configure(uint32_t sampleRate, uint32_t bitsPerSample, uint32_t channels){
+	assert(slRecorderObj==NULL);
+	SLDataLocator_IODevice loc_dev = {SL_DATALOCATOR_IODEVICE,
+									  SL_IODEVICE_AUDIOINPUT,
+									  SL_DEFAULTDEVICEID_AUDIOINPUT, NULL};
+	SLDataSource audioSrc = {&loc_dev, NULL};
+	SLDataLocator_AndroidSimpleBufferQueue loc_bq =
+			{SL_DATALOCATOR_ANDROIDSIMPLEBUFFERQUEUE, 1};
+	SLDataFormat_PCM format_pcm = {SL_DATAFORMAT_PCM, channels, sampleRate*1000,
+								   SL_PCMSAMPLEFORMAT_FIXED_16, SL_PCMSAMPLEFORMAT_FIXED_16,
+								   channels==2 ? (SL_SPEAKER_FRONT_LEFT | SL_SPEAKER_FRONT_RIGHT) : SL_SPEAKER_FRONT_CENTER, SL_BYTEORDER_LITTLEENDIAN};
+	SLDataSink audioSnk = {&loc_bq, &format_pcm};
+
+	const SLInterfaceID id[2] = {SL_IID_ANDROIDSIMPLEBUFFERQUEUE, SL_IID_ANDROIDCONFIGURATION};
+	const SLboolean req[2] = {SL_BOOLEAN_TRUE, SL_BOOLEAN_TRUE};
+	SLresult result = (*slEngine)->CreateAudioRecorder(slEngine, &slRecorderObj, &audioSrc, &audioSnk, 2, id, req);
+	CHECK_SL_ERROR(result, "Error creating recorder");
+
+	SLAndroidConfigurationItf recorderConfig;
+	result = (*slRecorderObj)->GetInterface(slRecorderObj, SL_IID_ANDROIDCONFIGURATION, &recorderConfig);
+	SLint32 streamType = SL_ANDROID_RECORDING_PRESET_VOICE_RECOGNITION;
+	result = (*recorderConfig)->SetConfiguration(recorderConfig, SL_ANDROID_KEY_RECORDING_PRESET, &streamType, sizeof(SLint32));
+
+	result=(*slRecorderObj)->Realize(slRecorderObj, SL_BOOLEAN_FALSE);
+	CHECK_SL_ERROR(result, "Error realizing recorder");
+
+	result=(*slRecorderObj)->GetInterface(slRecorderObj, SL_IID_RECORD, &slRecorder);
+	CHECK_SL_ERROR(result, "Error getting recorder interface");
+
+	result=(*slRecorderObj)->GetInterface(slRecorderObj, SL_IID_ANDROIDSIMPLEBUFFERQUEUE, &slBufferQueue);
+	CHECK_SL_ERROR(result, "Error getting buffer queue");
+
+	result=(*slBufferQueue)->RegisterCallback(slBufferQueue, AudioInputOpenSLES::BufferCallback, this);
+	CHECK_SL_ERROR(result, "Error setting buffer queue callback");
+
+	(*slBufferQueue)->Enqueue(slBufferQueue, nativeBuffer, nativeBufferSize*sizeof(int16_t));
+}
+
+void AudioInputOpenSLES::Start(){
+	SLresult result=(*slRecorder)->SetRecordState(slRecorder, SL_RECORDSTATE_RECORDING);
+	CHECK_SL_ERROR(result, "Error starting record");
+}
+
+void AudioInputOpenSLES::Stop(){
+	SLresult result=(*slRecorder)->SetRecordState(slRecorder, SL_RECORDSTATE_STOPPED);
+	CHECK_SL_ERROR(result, "Error stopping record");
+}
+
+
+void AudioInputOpenSLES::HandleSLCallback(){
+	//SLmillisecond pMsec = 0;
+	//(*slRecorder)->GetPosition(slRecorder, &pMsec);
+	//LOGI("Callback! pos=%lu", pMsec);
+	//InvokeCallback((unsigned char*)buffer, BUFFER_SIZE*sizeof(int16_t));
+	//fwrite(nativeBuffer, 1, nativeBufferSize*2, test);
+
+	if(nativeBufferSize==BUFFER_SIZE){
+		//LOGV("nativeBufferSize==BUFFER_SIZE");
+		InvokeCallback((unsigned char *) nativeBuffer, BUFFER_SIZE*sizeof(int16_t));
+	}else if(nativeBufferSize<BUFFER_SIZE){
+		//LOGV("nativeBufferSize<BUFFER_SIZE");
+		if(positionInBuffer>=BUFFER_SIZE){
+			InvokeCallback((unsigned char *) buffer, BUFFER_SIZE*sizeof(int16_t));
+			positionInBuffer=0;
+		}
+		memcpy(((unsigned char*)buffer)+positionInBuffer*2, nativeBuffer, (size_t)nativeBufferSize*2);
+		positionInBuffer+=nativeBufferSize;
+	}else if(nativeBufferSize>BUFFER_SIZE){
+		//LOGV("nativeBufferSize>BUFFER_SIZE");
+		for(unsigned int offset=0;offset<nativeBufferSize;offset+=BUFFER_SIZE){
+			InvokeCallback(((unsigned char *) nativeBuffer)+offset*2, BUFFER_SIZE*sizeof(int16_t));
+		}
+	}
+
+	(*slBufferQueue)->Enqueue(slBufferQueue, nativeBuffer, nativeBufferSize*sizeof(int16_t));
+}
+