1 /*
2 * This file is part of MPlayer.
3 *
4 * MPlayer is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * MPlayer is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License along
15 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17 */
18
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <inttypes.h>
22 #include <unistd.h>
23 #include <string.h>
24 #include <sys/types.h>
25 #include "m_option.h"
26 #include "mp_msg.h"
27 #include "libmpdemux/aviheader.h"
28 #include "libaf/af_format.h"
29 #include "libaf/reorder_ch.h"
30 #include "libmpdemux/ms_hdr.h"
31 #include "stream/stream.h"
32 #include "libmpdemux/muxer.h"
33 #include <faac.h>
34 #include "ae.h"
35 #include "ae_faac.h"
36
37
38 static faacEncHandle faac;
39 static faacEncConfigurationPtr config = NULL;
40 static int
41 param_bitrate = 128,
42 param_quality = 0,
43 param_object_type = 1,
44 param_mpeg = 2,
45 param_tns = 0,
46 param_raw = 0,
47 param_cutoff = 0,
48 param_format = 16,
49 param_debug = 0;
50
51 static int enc_frame_size = 0, divisor;
52 static unsigned long samples_input, max_bytes_output;
53 static unsigned char *decoder_specific_buffer = NULL;
54 static unsigned long decoder_specific_len = 0;
55
56 const m_option_t faacopts_conf[] = {
57 {"br", ¶m_bitrate, CONF_TYPE_INT, 0, 0, 0, NULL},
58 {"quality", ¶m_quality, CONF_TYPE_INT, CONF_RANGE, 0, 1000, NULL},
59 {"object", ¶m_object_type, CONF_TYPE_INT, CONF_RANGE, 1, 4, NULL},
60 {"mpeg", ¶m_mpeg, CONF_TYPE_INT, CONF_RANGE, 2, 4, NULL},
61 {"tns", ¶m_tns, CONF_TYPE_FLAG, 0, 0, 1, NULL},
62 {"cutoff", ¶m_cutoff, CONF_TYPE_INT, 0, 0, 0, NULL},
63 {"format", ¶m_format, CONF_TYPE_INT, 0, 0, 0, NULL},
64 {"raw", ¶m_raw, CONF_TYPE_FLAG, 0, 0, 1, NULL},
65 {"debug", ¶m_debug, CONF_TYPE_INT, CONF_RANGE, 0, 100000000, NULL},
66 {NULL, NULL, 0, 0, 0, 0, NULL}
67 };
68
69
bind_faac(audio_encoder_t * encoder,muxer_stream_t * mux_a)70 static int bind_faac(audio_encoder_t *encoder, muxer_stream_t *mux_a)
71 {
72 mux_a->wf = calloc(1, sizeof(*mux_a->wf) + decoder_specific_len + 256);
73 mux_a->wf->wFormatTag = 0x706D;
74 mux_a->wf->nChannels = encoder->params.channels;
75 mux_a->h.dwSampleSize=0; // VBR
76 mux_a->h.dwRate=encoder->params.sample_rate;
77 mux_a->h.dwScale=encoder->params.samples_per_frame;
78 mux_a->wf->nSamplesPerSec=mux_a->h.dwRate;
79 mux_a->wf->nAvgBytesPerSec = encoder->params.bitrate / 8;
80
81 mux_a->wf->nBlockAlign = mux_a->h.dwScale;
82 mux_a->h.dwSuggestedBufferSize = (encoder->params.audio_preload*mux_a->wf->nAvgBytesPerSec)/1000;
83 mux_a->h.dwSuggestedBufferSize -= mux_a->h.dwSuggestedBufferSize % mux_a->wf->nBlockAlign;
84
85 mux_a->wf->cbSize = decoder_specific_len;
86 mux_a->wf->wBitsPerSample = 0; /* does not apply */
87 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->wID = 1;
88 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->fdwFlags = 2;
89 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nBlockSize = mux_a->wf->nBlockAlign;
90 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nFramesPerBlock = 1;
91 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nCodecDelay = 0;
92
93 // Fix allocation
94 mux_a->wf = realloc(mux_a->wf, sizeof(*mux_a->wf)+mux_a->wf->cbSize);
95
96 if(config->inputFormat == FAAC_INPUT_FLOAT)
97 encoder->input_format = AF_FORMAT_FLOAT_NE;
98 else if(config->inputFormat == FAAC_INPUT_32BIT)
99 encoder->input_format = AF_FORMAT_S32_NE;
100 else
101 encoder->input_format = AF_FORMAT_S16_NE;
102
103 encoder->min_buffer_size = mux_a->h.dwSuggestedBufferSize;
104 encoder->max_buffer_size = mux_a->h.dwSuggestedBufferSize*2;
105
106 if(decoder_specific_buffer && decoder_specific_len)
107 memcpy(mux_a->wf + 1, decoder_specific_buffer, decoder_specific_len);
108
109 return 1;
110 }
111
get_frame_size(audio_encoder_t * encoder)112 static int get_frame_size(audio_encoder_t *encoder)
113 {
114 int sz = enc_frame_size;
115 enc_frame_size = 0;
116 return sz;
117 }
118
encode_faac(audio_encoder_t * encoder,uint8_t * dest,void * src,int len,int max_size)119 static int encode_faac(audio_encoder_t *encoder, uint8_t *dest, void *src, int len, int max_size)
120 {
121 if (encoder->params.channels >= 5)
122 reorder_channel_nch(src, AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT,
123 AF_CHANNEL_LAYOUT_AAC_DEFAULT,
124 encoder->params.channels,
125 len / divisor, divisor);
126
127 // len is divided by the number of bytes per sample
128 enc_frame_size = faacEncEncode(faac, (int32_t*) src, len / divisor, dest, max_size);
129
130 return enc_frame_size;
131 }
132
close_faac(audio_encoder_t * encoder)133 static int close_faac(audio_encoder_t *encoder)
134 {
135 return 1;
136 }
137
mpae_init_faac(audio_encoder_t * encoder)138 int mpae_init_faac(audio_encoder_t *encoder)
139 {
140 if(encoder->params.channels < 1 || encoder->params.channels > 6 || (param_mpeg != 2 && param_mpeg != 4))
141 {
142 mp_msg(MSGT_MENCODER, MSGL_FATAL, "AE_FAAC, unsupported number of channels: %d, or mpeg version: %d, exit\n", encoder->params.channels, param_mpeg);
143 return 0;
144 }
145
146 faac = faacEncOpen(encoder->params.sample_rate, encoder->params.channels, &samples_input, &max_bytes_output);
147 if(!faac)
148 {
149 mp_msg(MSGT_MENCODER, MSGL_FATAL, "AE_FAAC, couldn't init, exit\n");
150 return 0;
151 }
152 mp_msg(MSGT_MENCODER, MSGL_V, "AE_FAAC, sample_input: %lu, max_bytes_output: %lu\n", samples_input, max_bytes_output);
153 config = faacEncGetCurrentConfiguration(faac);
154 if(!config)
155 {
156 mp_msg(MSGT_MENCODER, MSGL_FATAL, "AE_FAAC, couldn't get init configuration, exit\n");
157 return 0;
158 }
159
160 param_bitrate *= 1000;
161 if(param_quality)
162 config->quantqual = param_quality;
163 else
164 config->bitRate = param_bitrate / encoder->params.channels;
165
166 if(param_format==33)
167 {
168 config->inputFormat = FAAC_INPUT_FLOAT;
169 divisor = 4;
170 }
171 else if(param_format==32)
172 {
173 config->inputFormat = FAAC_INPUT_32BIT;
174 divisor = 4;
175 }
176 else
177 {
178 config->inputFormat = FAAC_INPUT_16BIT;
179 divisor = 2;
180 }
181 config->outputFormat = param_raw ? 0 : 1; // 1 is ADTS
182 config->aacObjectType = param_object_type;
183 if(MAIN==0) config->aacObjectType--;
184 config->mpegVersion = (param_mpeg == 4 ? MPEG4 : MPEG2);
185 config->useTns = param_tns;
186 //Do not set allowMidside for API compatibility with faac,
187 //see https://github.com/knik0/faac/issues/8 for details
188 //config->allowMidside = 1;
189 config->shortctl = SHORTCTL_NORMAL;
190 param_cutoff = param_cutoff ? param_cutoff : encoder->params.sample_rate / 2;
191 if(param_cutoff > encoder->params.sample_rate / 2)
192 param_cutoff = encoder->params.sample_rate / 2;
193 config->bandWidth = param_cutoff;
194 if(encoder->params.channels == 6)
195 config->useLfe = 1;
196
197 if(!faacEncSetConfiguration(faac, config))
198 {
199 mp_msg(MSGT_MENCODER, MSGL_FATAL, "AE_FAAC, counldn't set specified parameters, exiting\n");
200 return 0;
201 }
202
203 if(param_raw)
204 faacEncGetDecoderSpecificInfo(faac, &decoder_specific_buffer, &decoder_specific_len);
205 else
206 decoder_specific_len = 0;
207
208 encoder->params.bitrate = param_bitrate;
209 encoder->params.samples_per_frame = 1024;
210 encoder->decode_buffer_size = divisor * samples_input; //samples * 16 bits_per_sample
211
212 encoder->bind = bind_faac;
213 encoder->get_frame_size = get_frame_size;
214 encoder->encode = encode_faac;
215 encoder->close = close_faac;
216
217 return 1;
218 }
219