1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/* $Id: HBMp3Encoder.cpp,v 1.5 2003/08/24 20:50:49 titer Exp $ */
#include "HBCommon.h"
#include "HBMp3Encoder.h"
#include "HBManager.h"
#include "HBFifo.h"
#include <lame/lame.h>
HBMp3Encoder::HBMp3Encoder( HBManager * manager, HBAudioInfo * audioInfo )
: HBThread( "mp3encoder" )
{
fManager = manager;
fAudioInfo = audioInfo;
fRawBuffer = NULL;
}
void HBMp3Encoder::DoWork()
{
while( !fAudioInfo->fRawFifo->Size() )
{
snooze( 5000 );
}
/* The idea is to have exactly one mp3 frame (i.e. 1152 samples) by
output buffer. As we are resampling from fInSampleRate to
fOutSampleRate, we will give ( 1152 * fInSampleRate ) /
( 2 * fOutSampleRate ) to libmp3lame so we are sure we will
never get more than 1 frame at a time */
uint32_t count = ( 1152 * fAudioInfo->fInSampleRate ) /
( 2 * fAudioInfo->fOutSampleRate );
fLeftSamples = (float*) malloc( count * sizeof( float ) );
fRightSamples = (float*) malloc( count * sizeof( float ) );
/* Init libmp3lame */
lame_global_flags * globalFlags = lame_init();
lame_set_in_samplerate( globalFlags, fAudioInfo->fInSampleRate );
lame_set_out_samplerate( globalFlags, fAudioInfo->fOutSampleRate );
lame_set_brate( globalFlags, fAudioInfo->fOutBitrate );
lame_init_params( globalFlags );
HBBuffer * mp3Buffer;
int ret;
while( !fDie )
{
/* Get new samples */
if( !GetSamples( count ) )
{
break;
}
mp3Buffer = new HBBuffer( LAME_MAXMP3BUFFER );
ret = lame_encode_buffer_float( globalFlags,
fLeftSamples, fRightSamples,
count, mp3Buffer->fData,
mp3Buffer->fSize );
if( ret < 0 )
{
/* Something wrong happened */
Log( "HBMp3Encoder : lame_encode_buffer_float() failed (%d)", ret );
delete mp3Buffer;
fManager->Error();
break;
}
else if( !ret )
{
delete mp3Buffer;
}
else if( ret > 0 )
{
/* We got something, send it to the muxer */
mp3Buffer->fSize = ret;
mp3Buffer->fKeyFrame = true;
fAudioInfo->fMp3Fifo->Push( mp3Buffer );
}
}
}
bool HBMp3Encoder::GetSamples( uint32_t count )
{
uint32_t samplesNb = 0;
while( samplesNb < count )
{
if( !fRawBuffer )
{
if( !( fRawBuffer = fAudioInfo->fRawFifo->Pop() ) )
{
return false;
}
fPosInBuffer = 0;
}
int willCopy = MIN( count - samplesNb, 6 * 256 - fPosInBuffer );
memcpy( fLeftSamples + samplesNb,
(float*) fRawBuffer->fData + fPosInBuffer,
willCopy * sizeof( float ) );
memcpy( fRightSamples + samplesNb,
(float*) fRawBuffer->fData + 6 * 256 + fPosInBuffer,
willCopy * sizeof( float ) );
samplesNb += willCopy;
fPosInBuffer += willCopy;
if( fPosInBuffer == 6 * 256 )
{
delete fRawBuffer;
fRawBuffer = NULL;
}
}
return true;
}
|
