/src/ffmpeg/libavcodec/ac3enc_fixed.c
Line | Count | Source (jump to first uncovered line) |
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 | | * fixed-point AC-3 encoder. |
27 | | */ |
28 | | |
29 | 0 | #define AC3ENC_FLOAT 0 |
30 | | #include "audiodsp.h" |
31 | | #include "ac3enc.h" |
32 | | #include "codec_internal.h" |
33 | | #include "eac3enc.h" |
34 | | #include "kbdwin.h" |
35 | | |
36 | | static void sum_square_butterfly(AC3EncodeContext *s, int64_t sum[4], |
37 | | const int32_t *coef0, const int32_t *coef1, |
38 | | int len) |
39 | 0 | { |
40 | 0 | s->ac3dsp.sum_square_butterfly_int32(sum, coef0, coef1, len); |
41 | 0 | } |
42 | | |
43 | | /* |
44 | | * Clip MDCT coefficients to allowable range. |
45 | | */ |
46 | | static void clip_coefficients(AudioDSPContext *adsp, int32_t *coef, |
47 | | unsigned int len) |
48 | 0 | { |
49 | 0 | adsp->vector_clip_int32(coef, coef, COEF_MIN, COEF_MAX, len); |
50 | 0 | } |
51 | | |
52 | | |
53 | | /* |
54 | | * Calculate a single coupling coordinate. |
55 | | */ |
56 | | static CoefType calc_cpl_coord(CoefSumType energy_ch, CoefSumType energy_cpl) |
57 | 0 | { |
58 | 0 | if (energy_cpl <= COEF_MAX) { |
59 | 0 | return 1048576; |
60 | 0 | } else { |
61 | 0 | uint64_t coord = energy_ch / (energy_cpl >> 24); |
62 | 0 | uint32_t coord32 = FFMIN(coord, 1073741824); |
63 | 0 | coord32 = ff_sqrt(coord32) << 9; |
64 | 0 | return FFMIN(coord32, COEF_MAX); |
65 | 0 | } |
66 | 0 | } |
67 | | |
68 | | #include "ac3enc_template.c" |
69 | | |
70 | | /** |
71 | | * Initialize MDCT tables. |
72 | | * |
73 | | * @param s AC-3 encoder private context |
74 | | * @return 0 on success, negative error code on failure |
75 | | */ |
76 | | static av_cold int ac3_fixed_mdct_init(AVCodecContext *avctx, AC3EncodeContext *s) |
77 | 0 | { |
78 | 0 | float fwin[AC3_BLOCK_SIZE]; |
79 | 0 | const float scale = -1.0f; |
80 | |
|
81 | 0 | int32_t *iwin = s->mdct_window_fixed; |
82 | |
|
83 | 0 | ff_kbd_window_init(fwin, 5.0, AC3_BLOCK_SIZE); |
84 | 0 | for (int i = 0; i < AC3_BLOCK_SIZE; i++) |
85 | 0 | iwin[i] = lrintf(fwin[i] * (1 << 22)); |
86 | |
|
87 | 0 | s->fdsp = avpriv_alloc_fixed_dsp(avctx->flags & AV_CODEC_FLAG_BITEXACT); |
88 | 0 | if (!s->fdsp) |
89 | 0 | return AVERROR(ENOMEM); |
90 | | |
91 | 0 | return av_tx_init(&s->tx, &s->tx_fn, AV_TX_INT32_MDCT, 0, |
92 | 0 | AC3_BLOCK_SIZE, &scale, 0); |
93 | 0 | } |
94 | | |
95 | | |
96 | | static av_cold int ac3_fixed_encode_init(AVCodecContext *avctx) |
97 | 0 | { |
98 | 0 | AC3EncodeContext *s = avctx->priv_data; |
99 | 0 | int ret; |
100 | |
|
101 | 0 | s->fixed_point = 1; |
102 | 0 | s->encode_frame = encode_frame; |
103 | |
|
104 | 0 | ret = ac3_fixed_mdct_init(avctx, s); |
105 | 0 | if (ret < 0) |
106 | 0 | return ret; |
107 | | |
108 | 0 | return ff_ac3_encode_init(avctx); |
109 | 0 | } |
110 | | |
111 | | |
112 | | const FFCodec ff_ac3_fixed_encoder = { |
113 | | .p.name = "ac3_fixed", |
114 | | CODEC_LONG_NAME("ATSC A/52A (AC-3)"), |
115 | | .p.type = AVMEDIA_TYPE_AUDIO, |
116 | | .p.id = AV_CODEC_ID_AC3, |
117 | | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, |
118 | | .priv_data_size = sizeof(AC3EncodeContext), |
119 | | .init = ac3_fixed_encode_init, |
120 | | FF_CODEC_ENCODE_CB(ff_ac3_encode_frame), |
121 | | .close = ff_ac3_encode_close, |
122 | | .p.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_S32P, |
123 | | AV_SAMPLE_FMT_NONE }, |
124 | | .p.priv_class = &ff_ac3enc_class, |
125 | | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP, |
126 | | .p.supported_samplerates = ff_ac3_sample_rate_tab, |
127 | | .p.ch_layouts = ff_ac3_ch_layouts, |
128 | | .defaults = ff_ac3_enc_defaults, |
129 | | }; |