int universal_speex_encode(void* handle, const void* pAudioBuffer,
unsigned cbAudioSamples, int* rSamplesConsumed,
void* pCodedData, unsigned cbMaxCodedData,
int* pcbCodedSize, unsigned* pbSendNow)
{
struct speex_codec_data_encoder *mpSpeexEnc =
(struct speex_codec_data_encoder *)handle;
assert(handle != NULL);
memcpy(&mpSpeexEnc->mpBuffer[mpSpeexEnc->mBufferLoad], pAudioBuffer,
SIZE_OF_SAMPLE * cbAudioSamples);
mpSpeexEnc->mBufferLoad = mpSpeexEnc->mBufferLoad+cbAudioSamples;
assert(mpSpeexEnc->mBufferLoad <= mpSpeexEnc->mNumSamplesPerFrame);
// Check for necessary number of samples
if(mpSpeexEnc->mBufferLoad == mpSpeexEnc->mNumSamplesPerFrame)
{
SpeexBits bits;
// Wrap our buffer to speex bits structure
speex_bits_init_buffer(&bits, pCodedData, cbMaxCodedData);
// Preprocess data if requested
if(mpSpeexEnc->mDoPreprocess)
speex_preprocess(mpSpeexEnc->mpPreprocessState, mpSpeexEnc->mpBuffer, NULL);
// Encode frame and append terminator
speex_encode_int(mpSpeexEnc->mpEncoderState, mpSpeexEnc->mpBuffer, &bits);
speex_bits_insert_terminator(&bits);
// Tell that we've produced packet
*pbSendNow = TRUE;
*pcbCodedSize = speex_bits_nbytes(&bits);
// Reset the buffer count.
mpSpeexEnc->mBufferLoad = 0;
}
else
{
*pbSendNow = FALSE;
*pcbCodedSize = 0;
}
*rSamplesConsumed = cbAudioSamples;
return RPLG_SUCCESS;
}
void gviSpursSpeexEncode(SpursSpeexTaskOutput *spuTaskOut)
{
short *inBuffer;
float *speexBuffer;
char *outBuffer;
unsigned int i;
spuTaskOut->mSpeexEncodedFrameSize = 0;
spuTaskOut->mSpeexInitialized = 1;
spuTaskOut->mSpeexSamplesPerFrame = 0;
spuTaskOut->mSpeexReturnCode = 0;
spuTaskOut->mSpeexOutBufferSize = 0;
speexBuffer = (float *)memalign(16, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(float));
inBuffer = (short *)memalign(16, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(short));
outBuffer = (char *)memalign(16, gviSpursSpeexTaskDesc.mOutputBufferSize);
memset(speexBuffer, 0, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(float));
memset(inBuffer, 0, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(short));
memset(outBuffer, 0, gviSpursSpeexTaskDesc.mOutputBufferSize);
cellDmaGet(inBuffer, (uint64_t)gviSpursSpeexTaskDesc.mInputBuffer, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(short), DMA_TAG(1), 0,0);
cellDmaWaitTagStatusAll(DMA_MASK(1));
// convert the input to floats for encoding
for(i = 0 ; i < gviSpursSpeexTaskDesc.mInputBufferSize ; i++)
speexBuffer[i] = inBuffer[i];
// (re)initialize the bits struct
speex_bits_init_buffer(&gviSpursSpeexBits,gviSpursSpeexBitsBuffer,sizeof(gviSpursSpeexBitsBuffer));
// flush the bits
speex_bits_reset(&gviSpursSpeexBits);
// encode the frame
speex_encode(gviSpursSpeexStateBuffer, speexBuffer, &gviSpursSpeexBits);
// write the bits to the output
spuTaskOut->mSpeexOutBufferSize = speex_bits_write(&gviSpursSpeexBits, (char *)outBuffer, gviSpursSpeexTaskDesc.mEncodedFrameSize);
//spuDebugPrintf("[Speex][SPU] transferring data back, output size should be: %d\n", gviSpursSpeexTaskDesc.mOutputBufferSize>16?gviSpursSpeexTaskDesc.mOutputBufferSize:16);
cellDmaPut(outBuffer, (uint64_t)gviSpursSpeexTaskDesc.mOutputBuffer, gviSpursSpeexTaskDesc.mOutputBufferSize, DMA_TAG(1), 0, 0);
cellDmaWaitTagStatusAll(DMA_MASK(1));
//spuDebugPrintf("[Speex][SPU] done transferring data back\n");
free(speexBuffer);
free(inBuffer);
free(outBuffer);
spuTaskOut->mSpeexReturnCode = 0;
}
void gviSpursSpeexDecodeAdd(SpursSpeexTaskOutput *spuTaskOut)
{
char *inBuffer;
float *speexBuffer;
short *outBuffer;
int rcode;
unsigned int i;
//spuDebugPrintf("[Speex][SPU] allocating buffers for decoding\n");
speexBuffer = (float *)memalign(16, gviSpursSpeexTaskDesc.mOutputBufferSize * sizeof(float));
outBuffer = (short *)memalign(16, gviSpursSpeexTaskDesc.mOutputBufferSize * sizeof(short));
inBuffer = (char *)memalign(16, gviSpursSpeexTaskDesc.mInputBufferSize);
memset(speexBuffer, 0, gviSpursSpeexTaskDesc.mOutputBufferSize * sizeof(float));
memset(outBuffer, 0, gviSpursSpeexTaskDesc.mOutputBufferSize);
memset(inBuffer, 0, gviSpursSpeexTaskDesc.mInputBufferSize * sizeof(short));
//spuDebugPrintf("[Speex][SPU] done allocating, getting input data, inbuffer size: %d\n", gSpuSampleTaskDesc.mInputBufferSize);
cellDmaGet(inBuffer, (uint64_t)gviSpursSpeexTaskDesc.mInputBuffer, gviSpursSpeexTaskDesc.mInputBufferSize, DMA_TAG(1), 0,0);
cellDmaWaitTagStatusAll(DMA_MASK(1));
// spuDebugPrintf("[Speex][SPU] done getting input data, preparing for speex to decode\n");
// read the data into the bits
// (re)initialize the bits struct
speex_bits_init_buffer(&gviSpursSpeexBits,gviSpursSpeexBitsBuffer,sizeof(gviSpursSpeexBitsBuffer));
speex_bits_read_from(&gviSpursSpeexBits, (char *)inBuffer, gviSpursSpeexTaskDesc.mEncodedFrameSize);
// decode it
rcode = speex_decode((void *)gviSpursSpeexStateBuffer, &gviSpursSpeexBits, speexBuffer);
assert(rcode == 0);
//spuDebugPrintf("[Speex][SPU] done with speex decode\n");
// convert the output from floats
for(i = 0 ; i < gviSpursSpeexTaskDesc.mOutputBufferSize ; i++)
outBuffer[i] = (short)speexBuffer[i];
//spuDebugPrintf("[Speex][SPU] transferring data back\n");
cellDmaPut(outBuffer, (uint64_t)gviSpursSpeexTaskDesc.mOutputBuffer, gviSpursSpeexTaskDesc.mOutputBufferSize * sizeof(short), DMA_TAG(1), 0, 0);
cellDmaWaitTagStatusAll(DMA_MASK(1));
//spuDebugPrintf("[Speex][SPU] done transferring data back\n");
free(speexBuffer);
free(inBuffer);
free(outBuffer);
spuTaskOut->mSpeexReturnCode = 0;
}
void gviSpursSpeexEncoderInitialize(SpursSpeexTaskOutput *spuOutput)
{
void *gviSpeexEncoderState;
int sampleRate = gviSpursSpeexTaskDesc.mSamplesPerSecond;
int quality = gviSpursSpeexTaskDesc.mQuality;
int rate;
int bitsPerFrame;
//spuDebugPrintf("[Speex][SPU] sample rate: %d, quality: %d\n", sampleRate, quality);
// create a new encoder state
if (sampleRate == GVRate_8KHz)
gviSpeexEncoderState = speex_encoder_init(&speex_nb_mode, gviSpursSpeexStateBuffer);
else if (sampleRate == GVRate_16KHz)
gviSpeexEncoderState = speex_encoder_init(&speex_wb_mode, gviSpursSpeexStateBuffer);
else
{
//spuDebugPrintf("[Speex][SPU] Initializing Speex failed\n");
spuOutput->mSpeexReturnCode = -2;
return;
}
if(!gviSpeexEncoderState)
{
//spuDebugPrintf("[Speex][SPU] Initializing Speex failed\n");
spuOutput->mSpeexReturnCode = -3;
return;
}
//spuDebugPrintf("[Speex][SPU] Done getting speex mode\n");
//spuDebugPrintf("[Speex][SPU] encoder state addr: 0x%8x\n", gviSpeexEncoderState);
// set the sampling rate
speex_encoder_ctl(gviSpeexEncoderState, SPEEX_SET_SAMPLING_RATE, &sampleRate);
// Get the samples per frame setting.
speex_encoder_ctl(gviSpeexEncoderState, SPEEX_GET_FRAME_SIZE, &spuOutput->mSpeexSamplesPerFrame);
// set the quality
speex_encoder_ctl(gviSpeexEncoderState, SPEEX_SET_QUALITY, &quality);
// (re)initialize the bits struct
speex_bits_init_buffer(&gviSpursSpeexBits,gviSpursSpeexBitsBuffer,sizeof(gviSpursSpeexBitsBuffer));
//spuDebugPrintf("[Speex][SPU] speex bits addr: 0x%8x\n", gviSpeexBits);
// get the bit rate
speex_encoder_ctl(gviSpeexEncoderState, SPEEX_GET_BITRATE, &rate);
//spuDebugPrintf("[Speex][SPU] Done with customizing options for speex\n");
// convert to bits per frame
bitsPerFrame = (rate / (sampleRate / spuOutput->mSpeexSamplesPerFrame));
// convert to bytes per frame and store, round up to allocate more space than needed.
spuOutput->mSpeexEncodedFrameSize = (bitsPerFrame / 8);
if (bitsPerFrame % 8)
spuOutput->mSpeexEncodedFrameSize++;
// we're now initialized
spuOutput->mSpeexInitialized = 1;
//spuDebugPrintf("[Speex][SPU] Done with initing speex\n");
spuOutput->mSpeexReturnCode = 0;
}
