10 #include "libmpdemux/aviheader.h"
11 #include "libmpdemux/ms_hdr.h"
12 #include "stream/stream.h"
13 #include "libmpdemux/muxer.h"
16 #include "libaf/af_format.h"
17 #include "libaf/reorder_ch.h"
18 #include "libavcodec/avcodec.h"
19 #include "ffmpeg_files/intreadwrite.h"
21 static AVCodec
*lavc_acodec
;
22 static AVCodecContext
*lavc_actx
;
23 extern char *lavc_param_acodec
;
24 extern int lavc_param_abitrate
;
25 extern int lavc_param_atag
;
26 extern int lavc_param_audio_global_header
;
27 extern int avcodec_initialized
;
28 static int compressed_frame_size
= 0;
29 #include "libmpdemux/mp_taglists.h"
31 static int bind_lavc(audio_encoder_t
*encoder
, muxer_stream_t
*mux_a
)
33 mux_a
->wf
= malloc(sizeof(WAVEFORMATEX
)+lavc_actx
->extradata_size
+256);
34 mux_a
->wf
->wFormatTag
= lavc_param_atag
;
35 mux_a
->wf
->nChannels
= lavc_actx
->channels
;
36 mux_a
->wf
->nSamplesPerSec
= lavc_actx
->sample_rate
;
37 mux_a
->wf
->nAvgBytesPerSec
= (lavc_actx
->bit_rate
/ 8);
38 mux_a
->avg_rate
= lavc_actx
->bit_rate
;
39 mux_a
->h
.dwRate
= mux_a
->wf
->nAvgBytesPerSec
;
40 if(lavc_actx
->block_align
)
41 mux_a
->h
.dwSampleSize
= mux_a
->h
.dwScale
= lavc_actx
->block_align
;
44 mux_a
->h
.dwScale
= (mux_a
->wf
->nAvgBytesPerSec
* lavc_actx
->frame_size
)/ mux_a
->wf
->nSamplesPerSec
; /* for cbr */
46 if ((mux_a
->wf
->nAvgBytesPerSec
*
47 lavc_actx
->frame_size
) % mux_a
->wf
->nSamplesPerSec
)
49 mux_a
->h
.dwScale
= lavc_actx
->frame_size
;
50 mux_a
->h
.dwRate
= lavc_actx
->sample_rate
;
51 mux_a
->h
.dwSampleSize
= 0; // Blocksize not constant
54 mux_a
->h
.dwSampleSize
= 0;
56 if(mux_a
->h
.dwSampleSize
)
57 mux_a
->wf
->nBlockAlign
= mux_a
->h
.dwSampleSize
;
59 mux_a
->wf
->nBlockAlign
= 1;
60 mux_a
->h
.dwSuggestedBufferSize
= (encoder
->params
.audio_preload
*mux_a
->wf
->nAvgBytesPerSec
)/1000;
61 mux_a
->h
.dwSuggestedBufferSize
-= mux_a
->h
.dwSuggestedBufferSize
% mux_a
->wf
->nBlockAlign
;
63 switch(lavc_param_atag
)
65 case 0x11: /* imaadpcm */
66 mux_a
->wf
->wBitsPerSample
= 4;
67 mux_a
->wf
->cbSize
= 2;
68 AV_WL16(mux_a
->wf
+1, lavc_actx
->frame_size
);
71 mux_a
->wf
->cbSize
= 12;
72 mux_a
->wf
->wBitsPerSample
= 0; /* does not apply */
73 ((MPEGLAYER3WAVEFORMAT
*) (mux_a
->wf
))->wID
= 1;
74 ((MPEGLAYER3WAVEFORMAT
*) (mux_a
->wf
))->fdwFlags
= 2;
75 ((MPEGLAYER3WAVEFORMAT
*) (mux_a
->wf
))->nBlockSize
= mux_a
->wf
->nBlockAlign
;
76 ((MPEGLAYER3WAVEFORMAT
*) (mux_a
->wf
))->nFramesPerBlock
= 1;
77 ((MPEGLAYER3WAVEFORMAT
*) (mux_a
->wf
))->nCodecDelay
= 0;
80 mux_a
->wf
->wBitsPerSample
= 0; /* Unknown */
81 if (lavc_actx
->extradata
&& (lavc_actx
->extradata_size
> 0))
83 memcpy(mux_a
->wf
+1, lavc_actx
->extradata
, lavc_actx
->extradata_size
);
84 mux_a
->wf
->cbSize
= lavc_actx
->extradata_size
;
87 mux_a
->wf
->cbSize
= 0;
92 mux_a
->wf
= realloc(mux_a
->wf
, sizeof(WAVEFORMATEX
)+mux_a
->wf
->cbSize
);
94 encoder
->input_format
= AF_FORMAT_S16_NE
;
95 encoder
->min_buffer_size
= mux_a
->h
.dwSuggestedBufferSize
;
96 encoder
->max_buffer_size
= mux_a
->h
.dwSuggestedBufferSize
*2;
101 static int encode_lavc(audio_encoder_t
*encoder
, uint8_t *dest
, void *src
, int size
, int max_size
)
104 if ((encoder
->params
.channels
== 6 || encoder
->params
.channels
== 5) &&
105 (!strcmp(lavc_acodec
->name
,"ac3") ||
106 !strcmp(lavc_acodec
->name
,"libfaac"))) {
107 int isac3
= !strcmp(lavc_acodec
->name
,"ac3");
108 reorder_channel_nch(src
, AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT
,
109 isac3
? AF_CHANNEL_LAYOUT_LAVC_DEFAULT
110 : AF_CHANNEL_LAYOUT_AAC_DEFAULT
,
111 encoder
->params
.channels
,
114 n
= avcodec_encode_audio(lavc_actx
, dest
, size
, src
);
115 compressed_frame_size
= n
;
120 static int close_lavc(audio_encoder_t
*encoder
)
122 compressed_frame_size
= 0;
126 static int get_frame_size(audio_encoder_t
*encoder
)
128 int sz
= compressed_frame_size
;
129 compressed_frame_size
= 0;
133 int mpae_init_lavc(audio_encoder_t
*encoder
)
135 encoder
->params
.samples_per_frame
= encoder
->params
.sample_rate
;
136 encoder
->params
.bitrate
= encoder
->params
.sample_rate
* encoder
->params
.channels
* 2 * 8;
138 if(!lavc_param_acodec
)
140 mp_tmsg(MSGT_MENCODER
, MSGL_FATAL
, "Audio LAVC, Missing codec name!\n");
144 if(!avcodec_initialized
){
146 avcodec_register_all();
147 avcodec_initialized
=1;
150 lavc_acodec
= avcodec_find_encoder_by_name(lavc_param_acodec
);
153 mp_tmsg(MSGT_MENCODER
, MSGL_FATAL
, "Audio LAVC, couldn't find encoder for codec %s.\n", lavc_param_acodec
);
156 if(lavc_param_atag
== 0)
158 lavc_param_atag
= mp_av_codec_get_tag(mp_wav_taglists
, lavc_acodec
->id
);
161 mp_msg(MSGT_MENCODER
, MSGL_FATAL
, "Couldn't find wav tag for specified codec, exit\n");
166 lavc_actx
= avcodec_alloc_context();
167 if(lavc_actx
== NULL
)
169 mp_tmsg(MSGT_MENCODER
, MSGL_FATAL
, "Audio LAVC, couldn't allocate context!\n");
173 // put sample parameters
174 lavc_actx
->channels
= encoder
->params
.channels
;
175 lavc_actx
->sample_rate
= encoder
->params
.sample_rate
;
176 lavc_actx
->time_base
.num
= 1;
177 lavc_actx
->time_base
.den
= encoder
->params
.sample_rate
;
178 if(lavc_param_abitrate
<1000)
179 lavc_actx
->bit_rate
= encoder
->params
.bitrate
= lavc_param_abitrate
* 1000;
181 lavc_actx
->bit_rate
= encoder
->params
.bitrate
= lavc_param_abitrate
;
185 * Special case for adpcm_ima_wav.
186 * The bitrate is only dependent on samplerate.
187 * We have to known frame_size and block_align in advance,
188 * so I just copied the code from libavcodec/adpcm.c
190 * However, ms adpcm_ima_wav uses a block_align of 2048,
191 * lavc defaults to 1024
193 if(lavc_param_atag
== 0x11) {
195 int framesize
= (blkalign
- 4 * lavc_actx
->channels
) * 8 / (4 * lavc_actx
->channels
) + 1;
196 lavc_actx
->bit_rate
= lavc_actx
->sample_rate
*8*blkalign
/framesize
;
198 if((lavc_param_audio_global_header
&1)
199 /*|| (video_global_header==0 && (oc->oformat->flags & AVFMT_GLOBALHEADER))*/){
200 lavc_actx
->flags
|= CODEC_FLAG_GLOBAL_HEADER
;
202 if(lavc_param_audio_global_header
&2){
203 lavc_actx
->flags2
|= CODEC_FLAG2_LOCAL_HEADER
;
206 if(avcodec_open(lavc_actx
, lavc_acodec
) < 0)
208 mp_tmsg(MSGT_MENCODER
, MSGL_FATAL
, "Couldn't open codec %s, br=%d.\n", lavc_param_acodec
, lavc_param_abitrate
);
212 if(lavc_param_atag
== 0x11) {
213 lavc_actx
->block_align
= 2048;
214 lavc_actx
->frame_size
= (lavc_actx
->block_align
- 4 * lavc_actx
->channels
) * 8 / (4 * lavc_actx
->channels
) + 1;
217 encoder
->decode_buffer_size
= lavc_actx
->frame_size
* 2 * encoder
->params
.channels
;
218 while (encoder
->decode_buffer_size
< 1024) encoder
->decode_buffer_size
*= 2;
219 encoder
->bind
= bind_lavc
;
220 encoder
->get_frame_size
= get_frame_size
;
221 encoder
->encode
= encode_lavc
;
222 encoder
->close
= close_lavc
;