annotate apedec.c @ 8006:c7c1e85d14bc libavcodec

Rename variables to clarify the channel coupling element and corresponding target channel element. Patch by Alex Converse (alex converse gmail com)
author superdump
date Mon, 06 Oct 2008 16:22:11 +0000
parents c15b1e83d27d
children f11197441364
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
1 /*
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
2 * Monkey's Audio lossless audio decoder
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
3 * Copyright (c) 2007 Benjamin Zores <ben@geexbox.org>
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
4 * based upon libdemac from Dave Chapman.
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
5 *
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
6 * This file is part of FFmpeg.
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
7 *
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
8 * FFmpeg is free software; you can redistribute it and/or
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
9 * modify it under the terms of the GNU Lesser General Public
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
10 * License as published by the Free Software Foundation; either
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
11 * version 2.1 of the License, or (at your option) any later version.
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
12 *
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
13 * FFmpeg is distributed in the hope that it will be useful,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
16 * Lesser General Public License for more details.
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
17 *
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
18 * You should have received a copy of the GNU Lesser General Public
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
19 * License along with FFmpeg; if not, write to the Free Software
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
21 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
22
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
23 #define ALT_BITSTREAM_READER_LE
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
24 #include "avcodec.h"
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
25 #include "dsputil.h"
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
26 #include "bitstream.h"
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
27 #include "bytestream.h"
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
28
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
29 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
30 * @file apedec.c
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
31 * Monkey's Audio lossless audio decoder
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
32 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
33
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
34 #define BLOCKS_PER_LOOP 4608
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
35 #define MAX_CHANNELS 2
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
36 #define MAX_BYTESPERSAMPLE 3
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
37
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
38 #define APE_FRAMECODE_MONO_SILENCE 1
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
39 #define APE_FRAMECODE_STEREO_SILENCE 3
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
40 #define APE_FRAMECODE_PSEUDO_STEREO 4
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
41
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
42 #define HISTORY_SIZE 512
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
43 #define PREDICTOR_ORDER 8
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
44 /** Total size of all predictor histories */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
45 #define PREDICTOR_SIZE 50
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
46
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
47 #define YDELAYA (18 + PREDICTOR_ORDER*4)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
48 #define YDELAYB (18 + PREDICTOR_ORDER*3)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
49 #define XDELAYA (18 + PREDICTOR_ORDER*2)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
50 #define XDELAYB (18 + PREDICTOR_ORDER)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
51
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
52 #define YADAPTCOEFFSA 18
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
53 #define XADAPTCOEFFSA 14
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
54 #define YADAPTCOEFFSB 10
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
55 #define XADAPTCOEFFSB 5
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
56
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
57 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
58 * Possible compression levels
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
59 * @{
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
60 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
61 enum APECompressionLevel {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
62 COMPRESSION_LEVEL_FAST = 1000,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
63 COMPRESSION_LEVEL_NORMAL = 2000,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
64 COMPRESSION_LEVEL_HIGH = 3000,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
65 COMPRESSION_LEVEL_EXTRA_HIGH = 4000,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
66 COMPRESSION_LEVEL_INSANE = 5000
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
67 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
68 /** @} */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
69
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
70 #define APE_FILTER_LEVELS 3
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
71
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
72 /** Filter orders depending on compression level */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
73 static const uint16_t ape_filter_orders[5][APE_FILTER_LEVELS] = {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
74 { 0, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
75 { 16, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
76 { 64, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
77 { 32, 256, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
78 { 16, 256, 1280 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
79 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
80
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
81 /** Filter fraction bits depending on compression level */
6482
258f569d30ca smaller tables
michael
parents: 6444
diff changeset
82 static const uint8_t ape_filter_fracbits[5][APE_FILTER_LEVELS] = {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
83 { 0, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
84 { 11, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
85 { 11, 0, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
86 { 10, 13, 0 },
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
87 { 11, 13, 15 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
88 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
89
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
90
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
91 /** Filters applied to the decoded data */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
92 typedef struct APEFilter {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
93 int16_t *coeffs; ///< actual coefficients used in filtering
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
94 int16_t *adaptcoeffs; ///< adaptive filter coefficients used for correcting of actual filter coefficients
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
95 int16_t *historybuffer; ///< filter memory
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
96 int16_t *delay; ///< filtered values
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
97
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
98 int avg;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
99 } APEFilter;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
100
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
101 typedef struct APERice {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
102 uint32_t k;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
103 uint32_t ksum;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
104 } APERice;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
105
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
106 typedef struct APERangecoder {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
107 uint32_t low; ///< low end of interval
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
108 uint32_t range; ///< length of interval
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
109 uint32_t help; ///< bytes_to_follow resp. intermediate value
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
110 unsigned int buffer; ///< buffer for input/output
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
111 } APERangecoder;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
112
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
113 /** Filter histories */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
114 typedef struct APEPredictor {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
115 int32_t *buf;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
116
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
117 int32_t lastA[2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
118
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
119 int32_t filterA[2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
120 int32_t filterB[2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
121
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
122 int32_t coeffsA[2][4]; ///< adaption coefficients
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
123 int32_t coeffsB[2][5]; ///< adaption coefficients
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
124 int32_t historybuffer[HISTORY_SIZE + PREDICTOR_SIZE];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
125 } APEPredictor;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
126
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
127 /** Decoder context */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
128 typedef struct APEContext {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
129 AVCodecContext *avctx;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
130 DSPContext dsp;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
131 int channels;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
132 int samples; ///< samples left to decode in current frame
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
133
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
134 int fileversion; ///< codec version, very important in decoding process
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
135 int compression_level; ///< compression levels
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
136 int fset; ///< which filter set to use (calculated from compression level)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
137 int flags; ///< global decoder flags
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
138
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
139 uint32_t CRC; ///< frame CRC
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
140 int frameflags; ///< frame flags
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
141 int currentframeblocks; ///< samples (per channel) in current frame
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
142 int blocksdecoded; ///< count of decoded samples in current frame
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
143 APEPredictor predictor; ///< predictor used for final reconstruction
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
144
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
145 int32_t decoded0[BLOCKS_PER_LOOP]; ///< decoded data for the first channel
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
146 int32_t decoded1[BLOCKS_PER_LOOP]; ///< decoded data for the second channel
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
147
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
148 int16_t* filterbuf[APE_FILTER_LEVELS]; ///< filter memory
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
149
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
150 APERangecoder rc; ///< rangecoder used to decode actual values
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
151 APERice riceX; ///< rice code parameters for the second channel
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
152 APERice riceY; ///< rice code parameters for the first channel
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
153 APEFilter filters[APE_FILTER_LEVELS][2]; ///< filters used for reconstruction
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
154
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
155 uint8_t *data; ///< current frame data
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
156 uint8_t *data_end; ///< frame data end
6223
michael
parents: 6202
diff changeset
157 const uint8_t *ptr; ///< current position in frame data
michael
parents: 6202
diff changeset
158 const uint8_t *last_ptr; ///< position where last 4608-sample block ended
6443
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
159
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
160 int error;
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
161 } APEContext;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
162
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
163 // TODO: dsputilize
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
164
6517
48759bfbd073 Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents: 6482
diff changeset
165 static av_cold int ape_decode_init(AVCodecContext * avctx)
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
166 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
167 APEContext *s = avctx->priv_data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
168 int i;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
169
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
170 if (avctx->extradata_size != 6) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
171 av_log(avctx, AV_LOG_ERROR, "Incorrect extradata\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
172 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
173 }
7823
4525dcd81357 Bump Major version, this commit is almost just renaming bits_per_sample to
michael
parents: 7723
diff changeset
174 if (avctx->bits_per_coded_sample != 16) {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
175 av_log(avctx, AV_LOG_ERROR, "Only 16-bit samples are supported\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
176 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
177 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
178 if (avctx->channels > 2) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
179 av_log(avctx, AV_LOG_ERROR, "Only mono and stereo is supported\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
180 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
181 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
182 s->avctx = avctx;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
183 s->channels = avctx->channels;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
184 s->fileversion = AV_RL16(avctx->extradata);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
185 s->compression_level = AV_RL16(avctx->extradata + 2);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
186 s->flags = AV_RL16(avctx->extradata + 4);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
187
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
188 av_log(avctx, AV_LOG_DEBUG, "Compression Level: %d - Flags: %d\n", s->compression_level, s->flags);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
189 if (s->compression_level % 1000 || s->compression_level > COMPRESSION_LEVEL_INSANE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
190 av_log(avctx, AV_LOG_ERROR, "Incorrect compression level %d\n", s->compression_level);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
191 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
192 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
193 s->fset = s->compression_level / 1000 - 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
194 for (i = 0; i < APE_FILTER_LEVELS; i++) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
195 if (!ape_filter_orders[s->fset][i])
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
196 break;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
197 s->filterbuf[i] = av_malloc((ape_filter_orders[s->fset][i] * 3 + HISTORY_SIZE) * 4);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
198 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
199
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
200 dsputil_init(&s->dsp, avctx);
7451
85ab7655ad4d Modify all codecs to report their supported input and output sample format(s).
pross
parents: 7203
diff changeset
201 avctx->sample_fmt = SAMPLE_FMT_S16;
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
202 return 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
203 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
204
6517
48759bfbd073 Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents: 6482
diff changeset
205 static av_cold int ape_decode_close(AVCodecContext * avctx)
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
206 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
207 APEContext *s = avctx->priv_data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
208 int i;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
209
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
210 for (i = 0; i < APE_FILTER_LEVELS; i++)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
211 av_freep(&s->filterbuf[i]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
212
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
213 return 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
214 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
215
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
216 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
217 * @defgroup rangecoder APE range decoder
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
218 * @{
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
219 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
220
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
221 #define CODE_BITS 32
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
222 #define TOP_VALUE ((unsigned int)1 << (CODE_BITS-1))
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
223 #define SHIFT_BITS (CODE_BITS - 9)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
224 #define EXTRA_BITS ((CODE_BITS-2) % 8 + 1)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
225 #define BOTTOM_VALUE (TOP_VALUE >> 8)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
226
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
227 /** Start the decoder */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
228 static inline void range_start_decoding(APEContext * ctx)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
229 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
230 ctx->rc.buffer = bytestream_get_byte(&ctx->ptr);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
231 ctx->rc.low = ctx->rc.buffer >> (8 - EXTRA_BITS);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
232 ctx->rc.range = (uint32_t) 1 << EXTRA_BITS;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
233 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
234
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
235 /** Perform normalization */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
236 static inline void range_dec_normalize(APEContext * ctx)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
237 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
238 while (ctx->rc.range <= BOTTOM_VALUE) {
6442
28e45bf84973 Prevent segfault due to reading over the end of the input buffer.
michael
parents: 6223
diff changeset
239 ctx->rc.buffer <<= 8;
28e45bf84973 Prevent segfault due to reading over the end of the input buffer.
michael
parents: 6223
diff changeset
240 if(ctx->ptr < ctx->data_end)
28e45bf84973 Prevent segfault due to reading over the end of the input buffer.
michael
parents: 6223
diff changeset
241 ctx->rc.buffer += *ctx->ptr;
28e45bf84973 Prevent segfault due to reading over the end of the input buffer.
michael
parents: 6223
diff changeset
242 ctx->ptr++;
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
243 ctx->rc.low = (ctx->rc.low << 8) | ((ctx->rc.buffer >> 1) & 0xFF);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
244 ctx->rc.range <<= 8;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
245 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
246 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
247
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
248 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
249 * Calculate culmulative frequency for next symbol. Does NO update!
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
250 * @param tot_f is the total frequency or (code_value)1<<shift
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
251 * @return the culmulative frequency
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
252 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
253 static inline int range_decode_culfreq(APEContext * ctx, int tot_f)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
254 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
255 range_dec_normalize(ctx);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
256 ctx->rc.help = ctx->rc.range / tot_f;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
257 return ctx->rc.low / ctx->rc.help;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
258 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
259
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
260 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
261 * Decode value with given size in bits
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
262 * @param shift number of bits to decode
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
263 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
264 static inline int range_decode_culshift(APEContext * ctx, int shift)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
265 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
266 range_dec_normalize(ctx);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
267 ctx->rc.help = ctx->rc.range >> shift;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
268 return ctx->rc.low / ctx->rc.help;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
269 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
270
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
271
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
272 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
273 * Update decoding state
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
274 * @param sy_f the interval length (frequency of the symbol)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
275 * @param lt_f the lower end (frequency sum of < symbols)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
276 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
277 static inline void range_decode_update(APEContext * ctx, int sy_f, int lt_f)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
278 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
279 ctx->rc.low -= ctx->rc.help * lt_f;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
280 ctx->rc.range = ctx->rc.help * sy_f;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
281 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
282
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
283 /** Decode n bits (n <= 16) without modelling */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
284 static inline int range_decode_bits(APEContext * ctx, int n)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
285 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
286 int sym = range_decode_culshift(ctx, n);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
287 range_decode_update(ctx, 1, sym);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
288 return sym;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
289 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
290
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
291
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
292 #define MODEL_ELEMENTS 64
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
293
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
294 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
295 * Fixed probabilities for symbols in Monkey Audio version 3.97
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
296 */
6482
258f569d30ca smaller tables
michael
parents: 6444
diff changeset
297 static const uint16_t counts_3970[22] = {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
298 0, 14824, 28224, 39348, 47855, 53994, 58171, 60926,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
299 62682, 63786, 64463, 64878, 65126, 65276, 65365, 65419,
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
300 65450, 65469, 65480, 65487, 65491, 65493,
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
301 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
302
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
303 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
304 * Probability ranges for symbols in Monkey Audio version 3.97
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
305 */
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
306 static const uint16_t counts_diff_3970[21] = {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
307 14824, 13400, 11124, 8507, 6139, 4177, 2755, 1756,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
308 1104, 677, 415, 248, 150, 89, 54, 31,
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
309 19, 11, 7, 4, 2,
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
310 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
311
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
312 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
313 * Fixed probabilities for symbols in Monkey Audio version 3.98
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
314 */
6482
258f569d30ca smaller tables
michael
parents: 6444
diff changeset
315 static const uint16_t counts_3980[22] = {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
316 0, 19578, 36160, 48417, 56323, 60899, 63265, 64435,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
317 64971, 65232, 65351, 65416, 65447, 65466, 65476, 65482,
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
318 65485, 65488, 65490, 65491, 65492, 65493,
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
319 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
320
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
321 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
322 * Probability ranges for symbols in Monkey Audio version 3.98
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
323 */
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
324 static const uint16_t counts_diff_3980[21] = {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
325 19578, 16582, 12257, 7906, 4576, 2366, 1170, 536,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
326 261, 119, 65, 31, 19, 10, 6, 3,
6444
3572cc5bc8ff Remove unused elements from tables.
michael
parents: 6443
diff changeset
327 3, 2, 1, 1, 1,
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
328 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
329
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
330 /**
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
331 * Decode symbol
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
332 * @param counts probability range start position
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
333 * @param count_diffs probability range widths
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
334 */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
335 static inline int range_get_symbol(APEContext * ctx,
6482
258f569d30ca smaller tables
michael
parents: 6444
diff changeset
336 const uint16_t counts[],
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
337 const uint16_t counts_diff[])
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
338 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
339 int symbol, cf;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
340
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
341 cf = range_decode_culshift(ctx, 16);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
342
6443
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
343 if(cf > 65492){
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
344 symbol= cf - 65535 + 63;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
345 range_decode_update(ctx, 1, cf);
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
346 if(cf > 65535)
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
347 ctx->error=1;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
348 return symbol;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
349 }
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
350 /* figure out the symbol inefficiently; a binary search would be much better */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
351 for (symbol = 0; counts[symbol + 1] <= cf; symbol++);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
352
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
353 range_decode_update(ctx, counts_diff[symbol], counts[symbol]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
354
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
355 return symbol;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
356 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
357 /** @} */ // group rangecoder
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
358
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
359 static inline void update_rice(APERice *rice, int x)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
360 {
7914
c15b1e83d27d Correct wrong lower limit and condition used in APE decoder
kostya
parents: 7823
diff changeset
361 int lim = rice->k ? (1 << (rice->k + 4)) : 0;
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
362 rice->ksum += ((x + 1) / 2) - ((rice->ksum + 16) >> 5);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
363
7914
c15b1e83d27d Correct wrong lower limit and condition used in APE decoder
kostya
parents: 7823
diff changeset
364 if (rice->ksum < lim)
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
365 rice->k--;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
366 else if (rice->ksum >= (1 << (rice->k + 5)))
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
367 rice->k++;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
368 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
369
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
370 static inline int ape_decode_value(APEContext * ctx, APERice *rice)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
371 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
372 int x, overflow;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
373
7723
20b105281e87 While APE changed container format in 3.98, frequency tables for range coding
kostya
parents: 7451
diff changeset
374 if (ctx->fileversion < 3990) {
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
375 int tmpk;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
376
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
377 overflow = range_get_symbol(ctx, counts_3970, counts_diff_3970);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
378
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
379 if (overflow == (MODEL_ELEMENTS - 1)) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
380 tmpk = range_decode_bits(ctx, 5);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
381 overflow = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
382 } else
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
383 tmpk = (rice->k < 1) ? 0 : rice->k - 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
384
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
385 if (tmpk <= 16)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
386 x = range_decode_bits(ctx, tmpk);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
387 else {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
388 x = range_decode_bits(ctx, 16);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
389 x |= (range_decode_bits(ctx, tmpk - 16) << 16);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
390 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
391 x += overflow << tmpk;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
392 } else {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
393 int base, pivot;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
394
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
395 pivot = rice->ksum >> 5;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
396 if (pivot == 0)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
397 pivot = 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
398
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
399 overflow = range_get_symbol(ctx, counts_3980, counts_diff_3980);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
400
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
401 if (overflow == (MODEL_ELEMENTS - 1)) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
402 overflow = range_decode_bits(ctx, 16) << 16;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
403 overflow |= range_decode_bits(ctx, 16);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
404 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
405
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
406 base = range_decode_culfreq(ctx, pivot);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
407 range_decode_update(ctx, 1, base);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
408
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
409 x = base + overflow * pivot;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
410 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
411
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
412 update_rice(rice, x);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
413
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
414 /* Convert to signed */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
415 if (x & 1)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
416 return (x >> 1) + 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
417 else
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
418 return -(x >> 1);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
419 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
420
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
421 static void entropy_decode(APEContext * ctx, int blockstodecode, int stereo)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
422 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
423 int32_t *decoded0 = ctx->decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
424 int32_t *decoded1 = ctx->decoded1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
425
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
426 ctx->blocksdecoded = blockstodecode;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
427
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
428 if (ctx->frameflags & APE_FRAMECODE_STEREO_SILENCE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
429 /* We are pure silence, just memset the output buffer. */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
430 memset(decoded0, 0, blockstodecode * sizeof(int32_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
431 memset(decoded1, 0, blockstodecode * sizeof(int32_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
432 } else {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
433 while (blockstodecode--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
434 *decoded0++ = ape_decode_value(ctx, &ctx->riceY);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
435 if (stereo)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
436 *decoded1++ = ape_decode_value(ctx, &ctx->riceX);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
437 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
438 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
439
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
440 if (ctx->blocksdecoded == ctx->currentframeblocks)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
441 range_dec_normalize(ctx); /* normalize to use up all bytes */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
442 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
443
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
444 static void init_entropy_decoder(APEContext * ctx)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
445 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
446 /* Read the CRC */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
447 ctx->CRC = bytestream_get_be32(&ctx->ptr);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
448
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
449 /* Read the frame flags if they exist */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
450 ctx->frameflags = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
451 if ((ctx->fileversion > 3820) && (ctx->CRC & 0x80000000)) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
452 ctx->CRC &= ~0x80000000;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
453
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
454 ctx->frameflags = bytestream_get_be32(&ctx->ptr);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
455 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
456
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
457 /* Keep a count of the blocks decoded in this frame */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
458 ctx->blocksdecoded = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
459
5963
80103098c797 spelling
vitor
parents: 5673
diff changeset
460 /* Initialize the rice structs */
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
461 ctx->riceX.k = 10;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
462 ctx->riceX.ksum = (1 << ctx->riceX.k) * 16;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
463 ctx->riceY.k = 10;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
464 ctx->riceY.ksum = (1 << ctx->riceY.k) * 16;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
465
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
466 /* The first 8 bits of input are ignored. */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
467 ctx->ptr++;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
468
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
469 range_start_decoding(ctx);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
470 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
471
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
472 static const int32_t initial_coeffs[4] = {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
473 360, 317, -109, 98
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
474 };
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
475
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
476 static void init_predictor_decoder(APEContext * ctx)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
477 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
478 APEPredictor *p = &ctx->predictor;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
479
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
480 /* Zero the history buffers */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
481 memset(p->historybuffer, 0, PREDICTOR_SIZE * sizeof(int32_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
482 p->buf = p->historybuffer;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
483
5966
1275461ae6f8 Fix some spelling mistakes.
diego
parents: 5963
diff changeset
484 /* Initialize and zero the coefficients */
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
485 memcpy(p->coeffsA[0], initial_coeffs, sizeof(initial_coeffs));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
486 memcpy(p->coeffsA[1], initial_coeffs, sizeof(initial_coeffs));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
487 memset(p->coeffsB, 0, sizeof(p->coeffsB));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
488
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
489 p->filterA[0] = p->filterA[1] = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
490 p->filterB[0] = p->filterB[1] = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
491 p->lastA[0] = p->lastA[1] = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
492 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
493
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
494 /** Get inverse sign of integer (-1 for positive, 1 for negative and 0 for zero) */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
495 static inline int APESIGN(int32_t x) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
496 return (x < 0) - (x > 0);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
497 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
498
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
499 static int predictor_update_filter(APEPredictor *p, const int decoded, const int filter, const int delayA, const int delayB, const int adaptA, const int adaptB)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
500 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
501 int32_t predictionA, predictionB;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
502
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
503 p->buf[delayA] = p->lastA[filter];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
504 p->buf[adaptA] = APESIGN(p->buf[delayA]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
505 p->buf[delayA - 1] = p->buf[delayA] - p->buf[delayA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
506 p->buf[adaptA - 1] = APESIGN(p->buf[delayA - 1]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
507
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
508 predictionA = p->buf[delayA ] * p->coeffsA[filter][0] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
509 p->buf[delayA - 1] * p->coeffsA[filter][1] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
510 p->buf[delayA - 2] * p->coeffsA[filter][2] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
511 p->buf[delayA - 3] * p->coeffsA[filter][3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
512
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
513 /* Apply a scaled first-order filter compression */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
514 p->buf[delayB] = p->filterA[filter ^ 1] - ((p->filterB[filter] * 31) >> 5);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
515 p->buf[adaptB] = APESIGN(p->buf[delayB]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
516 p->buf[delayB - 1] = p->buf[delayB] - p->buf[delayB - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
517 p->buf[adaptB - 1] = APESIGN(p->buf[delayB - 1]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
518 p->filterB[filter] = p->filterA[filter ^ 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
519
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
520 predictionB = p->buf[delayB ] * p->coeffsB[filter][0] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
521 p->buf[delayB - 1] * p->coeffsB[filter][1] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
522 p->buf[delayB - 2] * p->coeffsB[filter][2] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
523 p->buf[delayB - 3] * p->coeffsB[filter][3] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
524 p->buf[delayB - 4] * p->coeffsB[filter][4];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
525
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
526 p->lastA[filter] = decoded + ((predictionA + (predictionB >> 1)) >> 10);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
527 p->filterA[filter] = p->lastA[filter] + ((p->filterA[filter] * 31) >> 5);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
528
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
529 if (!decoded) // no need updating filter coefficients
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
530 return p->filterA[filter];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
531
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
532 if (decoded > 0) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
533 p->coeffsA[filter][0] -= p->buf[adaptA ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
534 p->coeffsA[filter][1] -= p->buf[adaptA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
535 p->coeffsA[filter][2] -= p->buf[adaptA - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
536 p->coeffsA[filter][3] -= p->buf[adaptA - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
537
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
538 p->coeffsB[filter][0] -= p->buf[adaptB ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
539 p->coeffsB[filter][1] -= p->buf[adaptB - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
540 p->coeffsB[filter][2] -= p->buf[adaptB - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
541 p->coeffsB[filter][3] -= p->buf[adaptB - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
542 p->coeffsB[filter][4] -= p->buf[adaptB - 4];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
543 } else {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
544 p->coeffsA[filter][0] += p->buf[adaptA ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
545 p->coeffsA[filter][1] += p->buf[adaptA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
546 p->coeffsA[filter][2] += p->buf[adaptA - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
547 p->coeffsA[filter][3] += p->buf[adaptA - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
548
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
549 p->coeffsB[filter][0] += p->buf[adaptB ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
550 p->coeffsB[filter][1] += p->buf[adaptB - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
551 p->coeffsB[filter][2] += p->buf[adaptB - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
552 p->coeffsB[filter][3] += p->buf[adaptB - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
553 p->coeffsB[filter][4] += p->buf[adaptB - 4];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
554 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
555 return p->filterA[filter];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
556 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
557
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
558 static void predictor_decode_stereo(APEContext * ctx, int count)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
559 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
560 int32_t predictionA, predictionB;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
561 APEPredictor *p = &ctx->predictor;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
562 int32_t *decoded0 = ctx->decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
563 int32_t *decoded1 = ctx->decoded1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
564
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
565 while (count--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
566 /* Predictor Y */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
567 predictionA = predictor_update_filter(p, *decoded0, 0, YDELAYA, YDELAYB, YADAPTCOEFFSA, YADAPTCOEFFSB);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
568 predictionB = predictor_update_filter(p, *decoded1, 1, XDELAYA, XDELAYB, XADAPTCOEFFSA, XADAPTCOEFFSB);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
569 *(decoded0++) = predictionA;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
570 *(decoded1++) = predictionB;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
571
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
572 /* Combined */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
573 p->buf++;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
574
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
575 /* Have we filled the history buffer? */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
576 if (p->buf == p->historybuffer + HISTORY_SIZE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
577 memmove(p->historybuffer, p->buf, PREDICTOR_SIZE * sizeof(int32_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
578 p->buf = p->historybuffer;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
579 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
580 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
581 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
582
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
583 static void predictor_decode_mono(APEContext * ctx, int count)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
584 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
585 APEPredictor *p = &ctx->predictor;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
586 int32_t *decoded0 = ctx->decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
587 int32_t predictionA, currentA, A;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
588
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
589 currentA = p->lastA[0];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
590
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
591 while (count--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
592 A = *decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
593
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
594 p->buf[YDELAYA] = currentA;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
595 p->buf[YDELAYA - 1] = p->buf[YDELAYA] - p->buf[YDELAYA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
596
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
597 predictionA = p->buf[YDELAYA ] * p->coeffsA[0][0] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
598 p->buf[YDELAYA - 1] * p->coeffsA[0][1] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
599 p->buf[YDELAYA - 2] * p->coeffsA[0][2] +
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
600 p->buf[YDELAYA - 3] * p->coeffsA[0][3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
601
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
602 currentA = A + (predictionA >> 10);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
603
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
604 p->buf[YADAPTCOEFFSA] = APESIGN(p->buf[YDELAYA ]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
605 p->buf[YADAPTCOEFFSA - 1] = APESIGN(p->buf[YDELAYA - 1]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
606
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
607 if (A > 0) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
608 p->coeffsA[0][0] -= p->buf[YADAPTCOEFFSA ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
609 p->coeffsA[0][1] -= p->buf[YADAPTCOEFFSA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
610 p->coeffsA[0][2] -= p->buf[YADAPTCOEFFSA - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
611 p->coeffsA[0][3] -= p->buf[YADAPTCOEFFSA - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
612 } else if (A < 0) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
613 p->coeffsA[0][0] += p->buf[YADAPTCOEFFSA ];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
614 p->coeffsA[0][1] += p->buf[YADAPTCOEFFSA - 1];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
615 p->coeffsA[0][2] += p->buf[YADAPTCOEFFSA - 2];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
616 p->coeffsA[0][3] += p->buf[YADAPTCOEFFSA - 3];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
617 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
618
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
619 p->buf++;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
620
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
621 /* Have we filled the history buffer? */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
622 if (p->buf == p->historybuffer + HISTORY_SIZE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
623 memmove(p->historybuffer, p->buf, PREDICTOR_SIZE * sizeof(int32_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
624 p->buf = p->historybuffer;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
625 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
626
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
627 p->filterA[0] = currentA + ((p->filterA[0] * 31) >> 5);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
628 *(decoded0++) = p->filterA[0];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
629 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
630
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
631 p->lastA[0] = currentA;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
632 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
633
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
634 static void do_init_filter(APEFilter *f, int16_t * buf, int order)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
635 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
636 f->coeffs = buf;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
637 f->historybuffer = buf + order;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
638 f->delay = f->historybuffer + order * 2;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
639 f->adaptcoeffs = f->historybuffer + order;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
640
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
641 memset(f->historybuffer, 0, (order * 2) * sizeof(int16_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
642 memset(f->coeffs, 0, order * sizeof(int16_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
643 f->avg = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
644 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
645
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
646 static void init_filter(APEContext * ctx, APEFilter *f, int16_t * buf, int order)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
647 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
648 do_init_filter(&f[0], buf, order);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
649 do_init_filter(&f[1], buf + order * 3 + HISTORY_SIZE, order);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
650 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
651
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
652 static inline void do_apply_filter(APEContext * ctx, int version, APEFilter *f, int32_t *data, int count, int order, int fracbits)
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
653 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
654 int res;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
655 int absres;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
656
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
657 while (count--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
658 /* round fixedpoint scalar product */
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
659 res = (ctx->dsp.scalarproduct_int16(f->delay - order, f->coeffs, order, 0) + (1 << (fracbits - 1))) >> fracbits;
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
660
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
661 if (*data < 0)
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
662 ctx->dsp.add_int16(f->coeffs, f->adaptcoeffs - order, order);
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
663 else if (*data > 0)
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
664 ctx->dsp.sub_int16(f->coeffs, f->adaptcoeffs - order, order);
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
665
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
666 res += *data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
667
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
668 *data++ = res;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
669
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
670 /* Update the output history */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
671 *f->delay++ = av_clip_int16(res);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
672
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
673 if (version < 3980) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
674 /* Version ??? to < 3.98 files (untested) */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
675 f->adaptcoeffs[0] = (res == 0) ? 0 : ((res >> 28) & 8) - 4;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
676 f->adaptcoeffs[-4] >>= 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
677 f->adaptcoeffs[-8] >>= 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
678 } else {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
679 /* Version 3.98 and later files */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
680
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
681 /* Update the adaption coefficients */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
682 absres = (res < 0 ? -res : res);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
683
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
684 if (absres > (f->avg * 3))
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
685 *f->adaptcoeffs = ((res >> 25) & 64) - 32;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
686 else if (absres > (f->avg * 4) / 3)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
687 *f->adaptcoeffs = ((res >> 26) & 32) - 16;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
688 else if (absres > 0)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
689 *f->adaptcoeffs = ((res >> 27) & 16) - 8;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
690 else
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
691 *f->adaptcoeffs = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
692
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
693 f->avg += (absres - f->avg) / 16;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
694
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
695 f->adaptcoeffs[-1] >>= 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
696 f->adaptcoeffs[-2] >>= 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
697 f->adaptcoeffs[-8] >>= 1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
698 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
699
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
700 f->adaptcoeffs++;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
701
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
702 /* Have we filled the history buffer? */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
703 if (f->delay == f->historybuffer + HISTORY_SIZE + (order * 2)) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
704 memmove(f->historybuffer, f->delay - (order * 2),
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
705 (order * 2) * sizeof(int16_t));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
706 f->delay = f->historybuffer + order * 2;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
707 f->adaptcoeffs = f->historybuffer + order;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
708 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
709 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
710 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
711
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
712 static void apply_filter(APEContext * ctx, APEFilter *f,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
713 int32_t * data0, int32_t * data1,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
714 int count, int order, int fracbits)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
715 {
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
716 do_apply_filter(ctx, ctx->fileversion, &f[0], data0, count, order, fracbits);
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
717 if (data1)
7203
87b1dfb5a98d Add several vector functions used by Monkey's Audio decoder to dsputil
kostya
parents: 7040
diff changeset
718 do_apply_filter(ctx, ctx->fileversion, &f[1], data1, count, order, fracbits);
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
719 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
720
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
721 static void ape_apply_filters(APEContext * ctx, int32_t * decoded0,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
722 int32_t * decoded1, int count)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
723 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
724 int i;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
725
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
726 for (i = 0; i < APE_FILTER_LEVELS; i++) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
727 if (!ape_filter_orders[ctx->fset][i])
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
728 break;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
729 apply_filter(ctx, ctx->filters[i], decoded0, decoded1, count, ape_filter_orders[ctx->fset][i], ape_filter_fracbits[ctx->fset][i]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
730 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
731 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
732
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
733 static void init_frame_decoder(APEContext * ctx)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
734 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
735 int i;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
736 init_entropy_decoder(ctx);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
737 init_predictor_decoder(ctx);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
738
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
739 for (i = 0; i < APE_FILTER_LEVELS; i++) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
740 if (!ape_filter_orders[ctx->fset][i])
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
741 break;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
742 init_filter(ctx, ctx->filters[i], ctx->filterbuf[i], ape_filter_orders[ctx->fset][i]);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
743 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
744 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
745
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
746 static void ape_unpack_mono(APEContext * ctx, int count)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
747 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
748 int32_t left;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
749 int32_t *decoded0 = ctx->decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
750 int32_t *decoded1 = ctx->decoded1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
751
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
752 if (ctx->frameflags & APE_FRAMECODE_STEREO_SILENCE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
753 entropy_decode(ctx, count, 0);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
754 /* We are pure silence, so we're done. */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
755 av_log(ctx->avctx, AV_LOG_DEBUG, "pure silence mono\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
756 return;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
757 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
758
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
759 entropy_decode(ctx, count, 0);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
760 ape_apply_filters(ctx, decoded0, NULL, count);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
761
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
762 /* Now apply the predictor decoding */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
763 predictor_decode_mono(ctx, count);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
764
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
765 /* Pseudo-stereo - just copy left channel to right channel */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
766 if (ctx->channels == 2) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
767 while (count--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
768 left = *decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
769 *(decoded1++) = *(decoded0++) = left;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
770 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
771 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
772 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
773
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
774 static void ape_unpack_stereo(APEContext * ctx, int count)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
775 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
776 int32_t left, right;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
777 int32_t *decoded0 = ctx->decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
778 int32_t *decoded1 = ctx->decoded1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
779
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
780 if (ctx->frameflags & APE_FRAMECODE_STEREO_SILENCE) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
781 /* We are pure silence, so we're done. */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
782 av_log(ctx->avctx, AV_LOG_DEBUG, "pure silence stereo\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
783 return;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
784 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
785
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
786 entropy_decode(ctx, count, 1);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
787 ape_apply_filters(ctx, decoded0, decoded1, count);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
788
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
789 /* Now apply the predictor decoding */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
790 predictor_decode_stereo(ctx, count);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
791
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
792 /* Decorrelate and scale to output depth */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
793 while (count--) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
794 left = *decoded1 - (*decoded0 / 2);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
795 right = left + *decoded0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
796
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
797 *(decoded0++) = left;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
798 *(decoded1++) = right;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
799 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
800 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
801
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
802 static int ape_decode_frame(AVCodecContext * avctx,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
803 void *data, int *data_size,
6223
michael
parents: 6202
diff changeset
804 const uint8_t * buf, int buf_size)
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
805 {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
806 APEContext *s = avctx->priv_data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
807 int16_t *samples = data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
808 int nblocks;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
809 int i, n;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
810 int blockstodecode;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
811 int bytes_used;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
812
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
813 if (buf_size == 0 && !s->samples) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
814 *data_size = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
815 return 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
816 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
817
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
818 /* should not happen but who knows */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
819 if (BLOCKS_PER_LOOP * 2 * avctx->channels > *data_size) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
820 av_log (avctx, AV_LOG_ERROR, "Packet size is too big to be handled in lavc! (max is %d where you have %d)\n", *data_size, s->samples * 2 * avctx->channels);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
821 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
822 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
823
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
824 if(!s->samples){
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
825 s->data = av_realloc(s->data, (buf_size + 3) & ~3);
6223
michael
parents: 6202
diff changeset
826 s->dsp.bswap_buf((uint32_t*)s->data, (const uint32_t*)buf, buf_size >> 2);
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
827 s->ptr = s->last_ptr = s->data;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
828 s->data_end = s->data + buf_size;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
829
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
830 nblocks = s->samples = bytestream_get_be32(&s->ptr);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
831 n = bytestream_get_be32(&s->ptr);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
832 if(n < 0 || n > 3){
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
833 av_log(avctx, AV_LOG_ERROR, "Incorrect offset passed\n");
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
834 s->data = NULL;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
835 return -1;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
836 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
837 s->ptr += n;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
838
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
839 s->currentframeblocks = nblocks;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
840 buf += 4;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
841 if (s->samples <= 0) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
842 *data_size = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
843 return buf_size;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
844 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
845
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
846 memset(s->decoded0, 0, sizeof(s->decoded0));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
847 memset(s->decoded1, 0, sizeof(s->decoded1));
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
848
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
849 /* Initialize the frame decoder */
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
850 init_frame_decoder(s);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
851 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
852
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
853 if (!s->data) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
854 *data_size = 0;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
855 return buf_size;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
856 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
857
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
858 nblocks = s->samples;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
859 blockstodecode = FFMIN(BLOCKS_PER_LOOP, nblocks);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
860
6443
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
861 s->error=0;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
862
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
863 if ((s->channels == 1) || (s->frameflags & APE_FRAMECODE_PSEUDO_STEREO))
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
864 ape_unpack_mono(s, blockstodecode);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
865 else
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
866 ape_unpack_stereo(s, blockstodecode);
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
867
6443
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
868 if(s->error || s->ptr > s->data_end){
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
869 s->samples=0;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
870 av_log(avctx, AV_LOG_ERROR, "Error decoding frame\n");
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
871 return -1;
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
872 }
e3adb7e96812 Detect and prevent reading over the end of counts_*. We pass the error
michael
parents: 6442
diff changeset
873
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
874 for (i = 0; i < blockstodecode; i++) {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
875 *samples++ = s->decoded0[i];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
876 if(s->channels == 2)
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
877 *samples++ = s->decoded1[i];
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
878 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
879
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
880 s->samples -= blockstodecode;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
881
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
882 *data_size = blockstodecode * 2 * s->channels;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
883 bytes_used = s->samples ? s->ptr - s->last_ptr : buf_size;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
884 s->last_ptr = s->ptr;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
885 return bytes_used;
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
886 }
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
887
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
888 AVCodec ape_decoder = {
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
889 "ape",
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
890 CODEC_TYPE_AUDIO,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
891 CODEC_ID_APE,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
892 sizeof(APEContext),
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
893 ape_decode_init,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
894 NULL,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
895 ape_decode_close,
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
896 ape_decode_frame,
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 6712
diff changeset
897 .long_name = NULL_IF_CONFIG_SMALL("Monkey's Audio"),
5673
9810f0bbacb2 Monkey Audio decoder
kostya
parents:
diff changeset
898 };