1 |
|
|
/* |
2 |
|
|
* The simplest AC-3 encoder |
3 |
|
|
* Copyright (c) 2000 Fabrice Bellard |
4 |
|
|
* Copyright (c) 2006-2010 Justin Ruggles <justin.ruggles@gmail.com> |
5 |
|
|
* Copyright (c) 2006-2010 Prakash Punnoor <prakash@punnoor.de> |
6 |
|
|
* |
7 |
|
|
* This file is part of FFmpeg. |
8 |
|
|
* |
9 |
|
|
* FFmpeg is free software; you can redistribute it and/or |
10 |
|
|
* modify it under the terms of the GNU Lesser General Public |
11 |
|
|
* License as published by the Free Software Foundation; either |
12 |
|
|
* version 2.1 of the License, or (at your option) any later version. |
13 |
|
|
* |
14 |
|
|
* FFmpeg is distributed in the hope that it will be useful, |
15 |
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
16 |
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
17 |
|
|
* Lesser General Public License for more details. |
18 |
|
|
* |
19 |
|
|
* You should have received a copy of the GNU Lesser General Public |
20 |
|
|
* License along with FFmpeg; if not, write to the Free Software |
21 |
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
22 |
|
|
*/ |
23 |
|
|
|
24 |
|
|
/** |
25 |
|
|
* @file |
26 |
|
|
* floating-point AC-3 encoder. |
27 |
|
|
*/ |
28 |
|
|
|
29 |
|
|
#define CONFIG_AC3ENC_FLOAT 1 |
30 |
|
|
#include "internal.h" |
31 |
|
|
#include "audiodsp.h" |
32 |
|
|
#include "ac3enc.h" |
33 |
|
|
#include "eac3enc.h" |
34 |
|
|
#include "kbdwin.h" |
35 |
|
|
|
36 |
|
|
|
37 |
|
|
#define AC3ENC_TYPE AC3ENC_TYPE_AC3 |
38 |
|
|
#include "ac3enc_opts_template.c" |
39 |
|
|
static const AVClass ac3enc_class = { |
40 |
|
|
.class_name = "AC-3 Encoder", |
41 |
|
|
.item_name = av_default_item_name, |
42 |
|
|
.option = ac3_options, |
43 |
|
|
.version = LIBAVUTIL_VERSION_INT, |
44 |
|
|
}; |
45 |
|
|
|
46 |
|
|
|
47 |
|
|
/* |
48 |
|
|
* Scale MDCT coefficients from float to 24-bit fixed-point. |
49 |
|
|
*/ |
50 |
|
546 |
static void scale_coefficients(AC3EncodeContext *s) |
51 |
|
|
{ |
52 |
|
546 |
int chan_size = AC3_MAX_COEFS * s->num_blocks; |
53 |
|
546 |
int cpl = s->cpl_on; |
54 |
|
546 |
s->ac3dsp.float_to_fixed24(s->fixed_coef_buffer + (chan_size * !cpl), |
55 |
|
546 |
s->mdct_coef_buffer + (chan_size * !cpl), |
56 |
|
546 |
chan_size * (s->channels + cpl)); |
57 |
|
546 |
} |
58 |
|
|
|
59 |
|
|
|
60 |
|
|
/* |
61 |
|
|
* Clip MDCT coefficients to allowable range. |
62 |
|
|
*/ |
63 |
|
3822 |
static void clip_coefficients(AudioDSPContext *adsp, float *coef, |
64 |
|
|
unsigned int len) |
65 |
|
|
{ |
66 |
|
3822 |
adsp->vector_clipf(coef, coef, len, COEF_MIN, COEF_MAX); |
67 |
|
3822 |
} |
68 |
|
|
|
69 |
|
|
|
70 |
|
|
/* |
71 |
|
|
* Calculate a single coupling coordinate. |
72 |
|
|
*/ |
73 |
|
30736 |
static CoefType calc_cpl_coord(CoefSumType energy_ch, CoefSumType energy_cpl) |
74 |
|
|
{ |
75 |
|
30736 |
float coord = 0.125; |
76 |
✓✗ |
30736 |
if (energy_cpl > 0) |
77 |
|
30736 |
coord *= sqrtf(energy_ch / energy_cpl); |
78 |
✗✓ |
30736 |
return FFMIN(coord, COEF_MAX); |
79 |
|
|
} |
80 |
|
|
|
81 |
|
13104 |
static void sum_square_butterfly(AC3EncodeContext *s, float sum[4], |
82 |
|
|
const float *coef0, const float *coef1, |
83 |
|
|
int len) |
84 |
|
|
{ |
85 |
|
13104 |
s->ac3dsp.sum_square_butterfly_float(sum, coef0, coef1, len); |
86 |
|
13104 |
} |
87 |
|
|
|
88 |
|
|
|
89 |
|
|
#include "ac3enc_template.c" |
90 |
|
|
|
91 |
|
|
|
92 |
|
|
/** |
93 |
|
|
* Finalize MDCT and free allocated memory. |
94 |
|
|
* |
95 |
|
|
* @param s AC-3 encoder private context |
96 |
|
|
*/ |
97 |
|
2 |
av_cold void ff_ac3_float_mdct_end(AC3EncodeContext *s) |
98 |
|
|
{ |
99 |
|
2 |
ff_mdct_end(&s->mdct); |
100 |
|
2 |
av_freep(&s->mdct_window); |
101 |
|
2 |
} |
102 |
|
|
|
103 |
|
|
|
104 |
|
|
/** |
105 |
|
|
* Initialize MDCT tables. |
106 |
|
|
* |
107 |
|
|
* @param s AC-3 encoder private context |
108 |
|
|
* @return 0 on success, negative error code on failure |
109 |
|
|
*/ |
110 |
|
2 |
av_cold int ff_ac3_float_mdct_init(AC3EncodeContext *s) |
111 |
|
|
{ |
112 |
|
|
float *window; |
113 |
|
|
int i, n, n2; |
114 |
|
|
|
115 |
|
2 |
n = 1 << 9; |
116 |
|
2 |
n2 = n >> 1; |
117 |
|
|
|
118 |
|
2 |
window = av_malloc_array(n, sizeof(*window)); |
119 |
✗✓ |
2 |
if (!window) { |
120 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Cannot allocate memory.\n"); |
121 |
|
|
return AVERROR(ENOMEM); |
122 |
|
|
} |
123 |
|
2 |
ff_kbd_window_init(window, 5.0, n2); |
124 |
✓✓ |
514 |
for (i = 0; i < n2; i++) |
125 |
|
512 |
window[n-1-i] = window[i]; |
126 |
|
2 |
s->mdct_window = window; |
127 |
|
|
|
128 |
|
2 |
return ff_mdct_init(&s->mdct, 9, 0, -2.0 / n); |
129 |
|
|
} |
130 |
|
|
|
131 |
|
|
|
132 |
|
2 |
av_cold int ff_ac3_float_encode_init(AVCodecContext *avctx) |
133 |
|
|
{ |
134 |
|
2 |
AC3EncodeContext *s = avctx->priv_data; |
135 |
|
2 |
s->fdsp = avpriv_float_dsp_alloc(avctx->flags & AV_CODEC_FLAG_BITEXACT); |
136 |
✗✓ |
2 |
if (!s->fdsp) |
137 |
|
|
return AVERROR(ENOMEM); |
138 |
|
2 |
return ff_ac3_encode_init(avctx); |
139 |
|
|
} |
140 |
|
|
|
141 |
|
|
AVCodec ff_ac3_encoder = { |
142 |
|
|
.name = "ac3", |
143 |
|
|
.long_name = NULL_IF_CONFIG_SMALL("ATSC A/52A (AC-3)"), |
144 |
|
|
.type = AVMEDIA_TYPE_AUDIO, |
145 |
|
|
.id = AV_CODEC_ID_AC3, |
146 |
|
|
.priv_data_size = sizeof(AC3EncodeContext), |
147 |
|
|
.init = ff_ac3_float_encode_init, |
148 |
|
|
.encode2 = ff_ac3_float_encode_frame, |
149 |
|
|
.close = ff_ac3_encode_close, |
150 |
|
|
.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_FLTP, |
151 |
|
|
AV_SAMPLE_FMT_NONE }, |
152 |
|
|
.priv_class = &ac3enc_class, |
153 |
|
|
.supported_samplerates = ff_ac3_sample_rate_tab, |
154 |
|
|
.channel_layouts = ff_ac3_channel_layouts, |
155 |
|
|
.defaults = ac3_defaults, |
156 |
|
|
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP, |
157 |
|
|
}; |