1 /*
2 * WavPack lossless audio encoder
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #define BITSTREAM_WRITER_LE
22
23 #include "libavutil/channel_layout.h"
24 #include "libavutil/intreadwrite.h"
25 #include "libavutil/opt.h"
26 #include "avcodec.h"
27 #include "codec_internal.h"
28 #include "encode.h"
29 #include "internal.h"
30 #include "put_bits.h"
31 #include "bytestream.h"
32 #include "wavpackenc.h"
33 #include "wavpack.h"
34
35 #define UPDATE_WEIGHT(weight, delta, source, result) \
36 if ((source) && (result)) { \
37 int32_t s = (int32_t) ((source) ^ (result)) >> 31; \
38 weight = ((delta) ^ s) + ((weight) - s); \
39 }
40
41 #define APPLY_WEIGHT_F(weight, sample) ((((((sample) & 0xffff) * (weight)) >> 9) + \
42 ((((sample) & ~0xffff) >> 9) * (weight)) + 1) >> 1)
43
44 #define APPLY_WEIGHT_I(weight, sample) (((weight) * (sample) + 512) >> 10)
45
46 #define APPLY_WEIGHT(weight, sample) ((sample) != (short) (sample) ? \
47 APPLY_WEIGHT_F(weight, sample) : APPLY_WEIGHT_I (weight, sample))
48
49 #define CLEAR(destin) memset(&destin, 0, sizeof(destin));
50
51 #define SHIFT_LSB 13
52 #define SHIFT_MASK (0x1FU << SHIFT_LSB)
53
54 #define MAG_LSB 18
55 #define MAG_MASK (0x1FU << MAG_LSB)
56
57 #define SRATE_LSB 23
58 #define SRATE_MASK (0xFU << SRATE_LSB)
59
60 #define EXTRA_TRY_DELTAS 1
61 #define EXTRA_ADJUST_DELTAS 2
62 #define EXTRA_SORT_FIRST 4
63 #define EXTRA_BRANCHES 8
64 #define EXTRA_SORT_LAST 16
65
66 typedef struct WavPackExtraInfo {
67 struct Decorr dps[MAX_TERMS];
68 int nterms, log_limit, gt16bit;
69 uint32_t best_bits;
70 } WavPackExtraInfo;
71
72 typedef struct WavPackWords {
73 int pend_data, holding_one, zeros_acc;
74 int holding_zero, pend_count;
75 WvChannel c[2];
76 } WavPackWords;
77
78 typedef struct WavPackEncodeContext {
79 AVClass *class;
80 AVCodecContext *avctx;
81 PutBitContext pb;
82 int block_samples;
83 int buffer_size;
84 int sample_index;
85 int stereo, stereo_in;
86 int ch_offset;
87
88 int32_t *samples[2];
89 int samples_size[2];
90
91 int32_t *sampleptrs[MAX_TERMS+2][2];
92 int sampleptrs_size[MAX_TERMS+2][2];
93
94 int32_t *temp_buffer[2][2];
95 int temp_buffer_size[2][2];
96
97 int32_t *best_buffer[2];
98 int best_buffer_size[2];
99
100 int32_t *js_left, *js_right;
101 int js_left_size, js_right_size;
102
103 int32_t *orig_l, *orig_r;
104 int orig_l_size, orig_r_size;
105
106 unsigned extra_flags;
107 int optimize_mono;
108 int decorr_filter;
109 int joint;
110 int num_branches;
111
112 uint32_t flags;
113 uint32_t crc_x;
114 WavPackWords w;
115
116 uint8_t int32_sent_bits, int32_zeros, int32_ones, int32_dups;
117 uint8_t float_flags, float_shift, float_max_exp, max_exp;
118 int32_t shifted_ones, shifted_zeros, shifted_both;
119 int32_t false_zeros, neg_zeros, ordata;
120
121 int num_terms, shift, joint_stereo, false_stereo;
122 int num_decorrs, num_passes, best_decorr, mask_decorr;
123 struct Decorr decorr_passes[MAX_TERMS];
124 const WavPackDecorrSpec *decorr_specs;
125 float delta_decay;
126 } WavPackEncodeContext;
127
wavpack_encode_init(AVCodecContext *avctx)128 static av_cold int wavpack_encode_init(AVCodecContext *avctx)
129 {
130 WavPackEncodeContext *s = avctx->priv_data;
131
132 s->avctx = avctx;
133
134 if (avctx->ch_layout.nb_channels > 255) {
135 av_log(avctx, AV_LOG_ERROR, "Invalid channel count: %d\n", avctx->ch_layout.nb_channels);
136 return AVERROR(EINVAL);
137 }
138
139 if (!avctx->frame_size) {
140 int block_samples;
141 if (!(avctx->sample_rate & 1))
142 block_samples = avctx->sample_rate / 2;
143 else
144 block_samples = avctx->sample_rate;
145
146 while (block_samples * avctx->ch_layout.nb_channels > WV_MAX_SAMPLES)
147 block_samples /= 2;
148
149 while (block_samples * avctx->ch_layout.nb_channels < 40000)
150 block_samples *= 2;
151 avctx->frame_size = block_samples;
152 } else if (avctx->frame_size && (avctx->frame_size < 128 ||
153 avctx->frame_size > WV_MAX_SAMPLES)) {
154 av_log(avctx, AV_LOG_ERROR, "invalid block size: %d\n", avctx->frame_size);
155 return AVERROR(EINVAL);
156 }
157
158 if (avctx->compression_level != FF_COMPRESSION_DEFAULT) {
159 if (avctx->compression_level >= 3) {
160 s->decorr_filter = 3;
161 s->num_passes = 9;
162 if (avctx->compression_level >= 8) {
163 s->num_branches = 4;
164 s->extra_flags = EXTRA_TRY_DELTAS|EXTRA_ADJUST_DELTAS|EXTRA_SORT_FIRST|EXTRA_SORT_LAST|EXTRA_BRANCHES;
165 } else if (avctx->compression_level >= 7) {
166 s->num_branches = 3;
167 s->extra_flags = EXTRA_TRY_DELTAS|EXTRA_ADJUST_DELTAS|EXTRA_SORT_FIRST|EXTRA_BRANCHES;
168 } else if (avctx->compression_level >= 6) {
169 s->num_branches = 2;
170 s->extra_flags = EXTRA_TRY_DELTAS|EXTRA_ADJUST_DELTAS|EXTRA_SORT_FIRST|EXTRA_BRANCHES;
171 } else if (avctx->compression_level >= 5) {
172 s->num_branches = 1;
173 s->extra_flags = EXTRA_TRY_DELTAS|EXTRA_ADJUST_DELTAS|EXTRA_SORT_FIRST|EXTRA_BRANCHES;
174 } else if (avctx->compression_level >= 4) {
175 s->num_branches = 1;
176 s->extra_flags = EXTRA_TRY_DELTAS|EXTRA_ADJUST_DELTAS|EXTRA_BRANCHES;
177 }
178 } else if (avctx->compression_level == 2) {
179 s->decorr_filter = 2;
180 s->num_passes = 4;
181 } else if (avctx->compression_level == 1) {
182 s->decorr_filter = 1;
183 s->num_passes = 2;
184 } else if (avctx->compression_level < 1) {
185 s->decorr_filter = 0;
186 s->num_passes = 0;
187 }
188 }
189
190 s->num_decorrs = decorr_filter_sizes[s->decorr_filter];
191 s->decorr_specs = decorr_filters[s->decorr_filter];
192
193 s->delta_decay = 2.0;
194
195 return 0;
196 }
197
shift_mono(int32_t *samples, int nb_samples, int shift)198 static void shift_mono(int32_t *samples, int nb_samples, int shift)
199 {
200 int i;
201 for (i = 0; i < nb_samples; i++)
202 samples[i] >>= shift;
203 }
204
shift_stereo(int32_t *left, int32_t *right, int nb_samples, int shift)205 static void shift_stereo(int32_t *left, int32_t *right,
206 int nb_samples, int shift)
207 {
208 int i;
209 for (i = 0; i < nb_samples; i++) {
210 left [i] >>= shift;
211 right[i] >>= shift;
212 }
213 }
214
215 #define FLOAT_SHIFT_ONES 1
216 #define FLOAT_SHIFT_SAME 2
217 #define FLOAT_SHIFT_SENT 4
218 #define FLOAT_ZEROS_SENT 8
219 #define FLOAT_NEG_ZEROS 0x10
220 #define FLOAT_EXCEPTIONS 0x20
221
222 #define get_mantissa(f) ((f) & 0x7fffff)
223 #define get_exponent(f) (((f) >> 23) & 0xff)
224 #define get_sign(f) (((f) >> 31) & 0x1)
225
process_float(WavPackEncodeContext *s, int32_t *sample)226 static void process_float(WavPackEncodeContext *s, int32_t *sample)
227 {
228 int32_t shift_count, value, f = *sample;
229
230 if (get_exponent(f) == 255) {
231 s->float_flags |= FLOAT_EXCEPTIONS;
232 value = 0x1000000;
233 shift_count = 0;
234 } else if (get_exponent(f)) {
235 shift_count = s->max_exp - get_exponent(f);
236 value = 0x800000 + get_mantissa(f);
237 } else {
238 shift_count = s->max_exp ? s->max_exp - 1 : 0;
239 value = get_mantissa(f);
240 }
241
242 if (shift_count < 25)
243 value >>= shift_count;
244 else
245 value = 0;
246
247 if (!value) {
248 if (get_exponent(f) || get_mantissa(f))
249 s->false_zeros++;
250 else if (get_sign(f))
251 s->neg_zeros++;
252 } else if (shift_count) {
253 int32_t mask = (1 << shift_count) - 1;
254
255 if (!(get_mantissa(f) & mask))
256 s->shifted_zeros++;
257 else if ((get_mantissa(f) & mask) == mask)
258 s->shifted_ones++;
259 else
260 s->shifted_both++;
261 }
262
263 s->ordata |= value;
264 *sample = get_sign(f) ? -value : value;
265 }
266
scan_float(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int nb_samples)267 static int scan_float(WavPackEncodeContext *s,
268 int32_t *samples_l, int32_t *samples_r,
269 int nb_samples)
270 {
271 uint32_t crc = 0xffffffffu;
272 int i;
273
274 s->shifted_ones = s->shifted_zeros = s->shifted_both = s->ordata = 0;
275 s->float_shift = s->float_flags = 0;
276 s->false_zeros = s->neg_zeros = 0;
277 s->max_exp = 0;
278
279 if (s->flags & WV_MONO_DATA) {
280 for (i = 0; i < nb_samples; i++) {
281 int32_t f = samples_l[i];
282 crc = crc * 27 + get_mantissa(f) * 9 + get_exponent(f) * 3 + get_sign(f);
283
284 if (get_exponent(f) > s->max_exp && get_exponent(f) < 255)
285 s->max_exp = get_exponent(f);
286 }
287 } else {
288 for (i = 0; i < nb_samples; i++) {
289 int32_t f;
290
291 f = samples_l[i];
292 crc = crc * 27 + get_mantissa(f) * 9 + get_exponent(f) * 3 + get_sign(f);
293 if (get_exponent(f) > s->max_exp && get_exponent(f) < 255)
294 s->max_exp = get_exponent(f);
295
296 f = samples_r[i];
297 crc = crc * 27 + get_mantissa(f) * 9 + get_exponent(f) * 3 + get_sign(f);
298
299 if (get_exponent(f) > s->max_exp && get_exponent(f) < 255)
300 s->max_exp = get_exponent(f);
301 }
302 }
303
304 s->crc_x = crc;
305
306 if (s->flags & WV_MONO_DATA) {
307 for (i = 0; i < nb_samples; i++)
308 process_float(s, &samples_l[i]);
309 } else {
310 for (i = 0; i < nb_samples; i++) {
311 process_float(s, &samples_l[i]);
312 process_float(s, &samples_r[i]);
313 }
314 }
315
316 s->float_max_exp = s->max_exp;
317
318 if (s->shifted_both)
319 s->float_flags |= FLOAT_SHIFT_SENT;
320 else if (s->shifted_ones && !s->shifted_zeros)
321 s->float_flags |= FLOAT_SHIFT_ONES;
322 else if (s->shifted_ones && s->shifted_zeros)
323 s->float_flags |= FLOAT_SHIFT_SAME;
324 else if (s->ordata && !(s->ordata & 1)) {
325 do {
326 s->float_shift++;
327 s->ordata >>= 1;
328 } while (!(s->ordata & 1));
329
330 if (s->flags & WV_MONO_DATA)
331 shift_mono(samples_l, nb_samples, s->float_shift);
332 else
333 shift_stereo(samples_l, samples_r, nb_samples, s->float_shift);
334 }
335
336 s->flags &= ~MAG_MASK;
337
338 while (s->ordata) {
339 s->flags += 1 << MAG_LSB;
340 s->ordata >>= 1;
341 }
342
343 if (s->false_zeros || s->neg_zeros)
344 s->float_flags |= FLOAT_ZEROS_SENT;
345
346 if (s->neg_zeros)
347 s->float_flags |= FLOAT_NEG_ZEROS;
348
349 return s->float_flags & (FLOAT_EXCEPTIONS | FLOAT_ZEROS_SENT |
350 FLOAT_SHIFT_SENT | FLOAT_SHIFT_SAME);
351 }
352
scan_int23(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int nb_samples)353 static void scan_int23(WavPackEncodeContext *s,
354 int32_t *samples_l, int32_t *samples_r,
355 int nb_samples)
356 {
357 uint32_t magdata = 0, ordata = 0, xordata = 0, anddata = ~0;
358 int i, total_shift = 0;
359
360 s->int32_sent_bits = s->int32_zeros = s->int32_ones = s->int32_dups = 0;
361
362 if (s->flags & WV_MONO_DATA) {
363 for (i = 0; i < nb_samples; i++) {
364 int32_t M = samples_l[i];
365
366 magdata |= (M < 0) ? ~M : M;
367 xordata |= M ^ -(M & 1);
368 anddata &= M;
369 ordata |= M;
370
371 if ((ordata & 1) && !(anddata & 1) && (xordata & 2))
372 return;
373 }
374 } else {
375 for (i = 0; i < nb_samples; i++) {
376 int32_t L = samples_l[i];
377 int32_t R = samples_r[i];
378
379 magdata |= (L < 0) ? ~L : L;
380 magdata |= (R < 0) ? ~R : R;
381 xordata |= L ^ -(L & 1);
382 xordata |= R ^ -(R & 1);
383 anddata &= L & R;
384 ordata |= L | R;
385
386 if ((ordata & 1) && !(anddata & 1) && (xordata & 2))
387 return;
388 }
389 }
390
391 s->flags &= ~MAG_MASK;
392
393 while (magdata) {
394 s->flags += 1 << MAG_LSB;
395 magdata >>= 1;
396 }
397
398 if (!(s->flags & MAG_MASK))
399 return;
400
401 if (!(ordata & 1)) {
402 do {
403 s->flags -= 1 << MAG_LSB;
404 s->int32_zeros++;
405 total_shift++;
406 ordata >>= 1;
407 } while (!(ordata & 1));
408 } else if (anddata & 1) {
409 do {
410 s->flags -= 1 << MAG_LSB;
411 s->int32_ones++;
412 total_shift++;
413 anddata >>= 1;
414 } while (anddata & 1);
415 } else if (!(xordata & 2)) {
416 do {
417 s->flags -= 1 << MAG_LSB;
418 s->int32_dups++;
419 total_shift++;
420 xordata >>= 1;
421 } while (!(xordata & 2));
422 }
423
424 if (total_shift) {
425 s->flags |= WV_INT32_DATA;
426
427 if (s->flags & WV_MONO_DATA)
428 shift_mono(samples_l, nb_samples, total_shift);
429 else
430 shift_stereo(samples_l, samples_r, nb_samples, total_shift);
431 }
432 }
433
scan_int32(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int nb_samples)434 static int scan_int32(WavPackEncodeContext *s,
435 int32_t *samples_l, int32_t *samples_r,
436 int nb_samples)
437 {
438 uint32_t magdata = 0, ordata = 0, xordata = 0, anddata = ~0;
439 uint32_t crc = 0xffffffffu;
440 int i, total_shift = 0;
441
442 s->int32_sent_bits = s->int32_zeros = s->int32_ones = s->int32_dups = 0;
443
444 if (s->flags & WV_MONO_DATA) {
445 for (i = 0; i < nb_samples; i++) {
446 int32_t M = samples_l[i];
447
448 crc = crc * 9 + (M & 0xffff) * 3 + ((M >> 16) & 0xffff);
449 magdata |= (M < 0) ? ~M : M;
450 xordata |= M ^ -(M & 1);
451 anddata &= M;
452 ordata |= M;
453 }
454 } else {
455 for (i = 0; i < nb_samples; i++) {
456 int32_t L = samples_l[i];
457 int32_t R = samples_r[i];
458
459 crc = crc * 9 + (L & 0xffff) * 3 + ((L >> 16) & 0xffff);
460 crc = crc * 9 + (R & 0xffff) * 3 + ((R >> 16) & 0xffff);
461 magdata |= (L < 0) ? ~L : L;
462 magdata |= (R < 0) ? ~R : R;
463 xordata |= L ^ -(L & 1);
464 xordata |= R ^ -(R & 1);
465 anddata &= L & R;
466 ordata |= L | R;
467 }
468 }
469
470 s->crc_x = crc;
471 s->flags &= ~MAG_MASK;
472
473 while (magdata) {
474 s->flags += 1 << MAG_LSB;
475 magdata >>= 1;
476 }
477
478 if (!((s->flags & MAG_MASK) >> MAG_LSB)) {
479 s->flags &= ~WV_INT32_DATA;
480 return 0;
481 }
482
483 if (!(ordata & 1))
484 do {
485 s->flags -= 1 << MAG_LSB;
486 s->int32_zeros++;
487 total_shift++;
488 ordata >>= 1;
489 } while (!(ordata & 1));
490 else if (anddata & 1)
491 do {
492 s->flags -= 1 << MAG_LSB;
493 s->int32_ones++;
494 total_shift++;
495 anddata >>= 1;
496 } while (anddata & 1);
497 else if (!(xordata & 2))
498 do {
499 s->flags -= 1 << MAG_LSB;
500 s->int32_dups++;
501 total_shift++;
502 xordata >>= 1;
503 } while (!(xordata & 2));
504
505 if (((s->flags & MAG_MASK) >> MAG_LSB) > 23) {
506 s->int32_sent_bits = (uint8_t)(((s->flags & MAG_MASK) >> MAG_LSB) - 23);
507 total_shift += s->int32_sent_bits;
508 s->flags &= ~MAG_MASK;
509 s->flags += 23 << MAG_LSB;
510 }
511
512 if (total_shift) {
513 s->flags |= WV_INT32_DATA;
514
515 if (s->flags & WV_MONO_DATA)
516 shift_mono(samples_l, nb_samples, total_shift);
517 else
518 shift_stereo(samples_l, samples_r, nb_samples, total_shift);
519 }
520
521 return s->int32_sent_bits;
522 }
523
store_weight(int weight)524 static int8_t store_weight(int weight)
525 {
526 weight = av_clip(weight, -1024, 1024);
527 if (weight > 0)
528 weight -= (weight + 64) >> 7;
529
530 return (weight + 4) >> 3;
531 }
532
restore_weight(int8_t weight)533 static int restore_weight(int8_t weight)
534 {
535 int result = 8 * weight;
536
537 if (result > 0)
538 result += (result + 64) >> 7;
539
540 return result;
541 }
542
log2s(int32_t value)543 static int log2s(int32_t value)
544 {
545 return (value < 0) ? -wp_log2(-value) : wp_log2(value);
546 }
547
decorr_mono(int32_t *in_samples, int32_t *out_samples, int nb_samples, struct Decorr *dpp, int dir)548 static void decorr_mono(int32_t *in_samples, int32_t *out_samples,
549 int nb_samples, struct Decorr *dpp, int dir)
550 {
551 int m = 0, i;
552
553 dpp->sumA = 0;
554
555 if (dir < 0) {
556 out_samples += (nb_samples - 1);
557 in_samples += (nb_samples - 1);
558 }
559
560 dpp->weightA = restore_weight(store_weight(dpp->weightA));
561
562 for (i = 0; i < MAX_TERM; i++)
563 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
564
565 if (dpp->value > MAX_TERM) {
566 while (nb_samples--) {
567 int32_t left, sam_A;
568
569 sam_A = ((3 - (dpp->value & 1)) * dpp->samplesA[0] - dpp->samplesA[1]) >> !(dpp->value & 1);
570
571 dpp->samplesA[1] = dpp->samplesA[0];
572 dpp->samplesA[0] = left = in_samples[0];
573
574 left -= APPLY_WEIGHT(dpp->weightA, sam_A);
575 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam_A, left);
576 dpp->sumA += dpp->weightA;
577 out_samples[0] = left;
578 in_samples += dir;
579 out_samples += dir;
580 }
581 } else if (dpp->value > 0) {
582 while (nb_samples--) {
583 int k = (m + dpp->value) & (MAX_TERM - 1);
584 int32_t left, sam_A;
585
586 sam_A = dpp->samplesA[m];
587 dpp->samplesA[k] = left = in_samples[0];
588 m = (m + 1) & (MAX_TERM - 1);
589
590 left -= APPLY_WEIGHT(dpp->weightA, sam_A);
591 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam_A, left);
592 dpp->sumA += dpp->weightA;
593 out_samples[0] = left;
594 in_samples += dir;
595 out_samples += dir;
596 }
597 }
598
599 if (m && dpp->value > 0 && dpp->value <= MAX_TERM) {
600 int32_t temp_A[MAX_TERM];
601
602 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
603
604 for (i = 0; i < MAX_TERM; i++) {
605 dpp->samplesA[i] = temp_A[m];
606 m = (m + 1) & (MAX_TERM - 1);
607 }
608 }
609 }
610
reverse_mono_decorr(struct Decorr *dpp)611 static void reverse_mono_decorr(struct Decorr *dpp)
612 {
613 if (dpp->value > MAX_TERM) {
614 int32_t sam_A;
615
616 if (dpp->value & 1)
617 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
618 else
619 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
620
621 dpp->samplesA[1] = dpp->samplesA[0];
622 dpp->samplesA[0] = sam_A;
623
624 if (dpp->value & 1)
625 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
626 else
627 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
628
629 dpp->samplesA[1] = sam_A;
630 } else if (dpp->value > 1) {
631 int i, j, k;
632
633 for (i = 0, j = dpp->value - 1, k = 0; k < dpp->value / 2; i++, j--, k++) {
634 i &= (MAX_TERM - 1);
635 j &= (MAX_TERM - 1);
636 dpp->samplesA[i] ^= dpp->samplesA[j];
637 dpp->samplesA[j] ^= dpp->samplesA[i];
638 dpp->samplesA[i] ^= dpp->samplesA[j];
639 }
640 }
641 }
642
643 #define count_bits(av) ((av) ? 32 - ff_clz(av) : 0)
644
log2sample(uint32_t v, int limit, uint32_t *result)645 static uint32_t log2sample(uint32_t v, int limit, uint32_t *result)
646 {
647 uint32_t dbits = count_bits(v);
648
649 if ((v += v >> 9) < (1 << 8)) {
650 *result += (dbits << 8) + ff_wp_log2_table[(v << (9 - dbits)) & 0xff];
651 } else {
652 *result += dbits = (dbits << 8) + ff_wp_log2_table[(v >> (dbits - 9)) & 0xff];
653
654 if (limit && dbits >= limit)
655 return 1;
656 }
657
658 return 0;
659 }
660
log2mono(int32_t *samples, int nb_samples, int limit)661 static uint32_t log2mono(int32_t *samples, int nb_samples, int limit)
662 {
663 uint32_t result = 0;
664 while (nb_samples--) {
665 if (log2sample(abs(*samples++), limit, &result))
666 return UINT32_MAX;
667 }
668 return result;
669 }
670
log2stereo(int32_t *samples_l, int32_t *samples_r, int nb_samples, int limit)671 static uint32_t log2stereo(int32_t *samples_l, int32_t *samples_r,
672 int nb_samples, int limit)
673 {
674 uint32_t result = 0;
675 while (nb_samples--) {
676 if (log2sample(abs(*samples_l++), limit, &result) ||
677 log2sample(abs(*samples_r++), limit, &result))
678 return UINT32_MAX;
679 }
680 return result;
681 }
682
decorr_mono_buffer(int32_t *samples, int32_t *outsamples, int nb_samples, struct Decorr *dpp, int tindex)683 static void decorr_mono_buffer(int32_t *samples, int32_t *outsamples,
684 int nb_samples, struct Decorr *dpp,
685 int tindex)
686 {
687 struct Decorr dp, *dppi = dpp + tindex;
688 int delta = dppi->delta, pre_delta, term = dppi->value;
689
690 if (delta == 7)
691 pre_delta = 7;
692 else if (delta < 2)
693 pre_delta = 3;
694 else
695 pre_delta = delta + 1;
696
697 CLEAR(dp);
698 dp.value = term;
699 dp.delta = pre_delta;
700 decorr_mono(samples, outsamples, FFMIN(2048, nb_samples), &dp, -1);
701 dp.delta = delta;
702
703 if (tindex == 0)
704 reverse_mono_decorr(&dp);
705 else
706 CLEAR(dp.samplesA);
707
708 memcpy(dppi->samplesA, dp.samplesA, sizeof(dp.samplesA));
709 dppi->weightA = dp.weightA;
710
711 if (delta == 0) {
712 dp.delta = 1;
713 decorr_mono(samples, outsamples, nb_samples, &dp, 1);
714 dp.delta = 0;
715 memcpy(dp.samplesA, dppi->samplesA, sizeof(dp.samplesA));
716 dppi->weightA = dp.weightA = dp.sumA / nb_samples;
717 }
718
719 decorr_mono(samples, outsamples, nb_samples, &dp, 1);
720 }
721
recurse_mono(WavPackEncodeContext *s, WavPackExtraInfo *info, int depth, int delta, uint32_t input_bits)722 static void recurse_mono(WavPackEncodeContext *s, WavPackExtraInfo *info,
723 int depth, int delta, uint32_t input_bits)
724 {
725 int term, branches = s->num_branches - depth;
726 int32_t *samples, *outsamples;
727 uint32_t term_bits[22], bits;
728
729 if (branches < 1 || depth + 1 == info->nterms)
730 branches = 1;
731
732 CLEAR(term_bits);
733 samples = s->sampleptrs[depth][0];
734 outsamples = s->sampleptrs[depth + 1][0];
735
736 for (term = 1; term <= 18; term++) {
737 if (term == 17 && branches == 1 && depth + 1 < info->nterms)
738 continue;
739
740 if (term > 8 && term < 17)
741 continue;
742
743 if (!s->extra_flags && (term > 4 && term < 17))
744 continue;
745
746 info->dps[depth].value = term;
747 info->dps[depth].delta = delta;
748 decorr_mono_buffer(samples, outsamples, s->block_samples, info->dps, depth);
749 bits = log2mono(outsamples, s->block_samples, info->log_limit);
750
751 if (bits < info->best_bits) {
752 info->best_bits = bits;
753 CLEAR(s->decorr_passes);
754 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * (depth + 1));
755 memcpy(s->sampleptrs[info->nterms + 1][0],
756 s->sampleptrs[depth + 1][0], s->block_samples * 4);
757 }
758
759 term_bits[term + 3] = bits;
760 }
761
762 while (depth + 1 < info->nterms && branches--) {
763 uint32_t local_best_bits = input_bits;
764 int best_term = 0, i;
765
766 for (i = 0; i < 22; i++)
767 if (term_bits[i] && term_bits[i] < local_best_bits) {
768 local_best_bits = term_bits[i];
769 best_term = i - 3;
770 }
771
772 if (!best_term)
773 break;
774
775 term_bits[best_term + 3] = 0;
776
777 info->dps[depth].value = best_term;
778 info->dps[depth].delta = delta;
779 decorr_mono_buffer(samples, outsamples, s->block_samples, info->dps, depth);
780
781 recurse_mono(s, info, depth + 1, delta, local_best_bits);
782 }
783 }
784
sort_mono(WavPackEncodeContext *s, WavPackExtraInfo *info)785 static void sort_mono(WavPackEncodeContext *s, WavPackExtraInfo *info)
786 {
787 int reversed = 1;
788 uint32_t bits;
789
790 while (reversed) {
791 int ri, i;
792
793 memcpy(info->dps, s->decorr_passes, sizeof(s->decorr_passes));
794 reversed = 0;
795
796 for (ri = 0; ri < info->nterms && s->decorr_passes[ri].value; ri++) {
797
798 if (ri + 1 >= info->nterms || !s->decorr_passes[ri+1].value)
799 break;
800
801 if (s->decorr_passes[ri].value == s->decorr_passes[ri+1].value) {
802 decorr_mono_buffer(s->sampleptrs[ri][0], s->sampleptrs[ri+1][0],
803 s->block_samples, info->dps, ri);
804 continue;
805 }
806
807 info->dps[ri ] = s->decorr_passes[ri+1];
808 info->dps[ri+1] = s->decorr_passes[ri ];
809
810 for (i = ri; i < info->nterms && s->decorr_passes[i].value; i++)
811 decorr_mono_buffer(s->sampleptrs[i][0], s->sampleptrs[i+1][0],
812 s->block_samples, info->dps, i);
813
814 bits = log2mono(s->sampleptrs[i][0], s->block_samples, info->log_limit);
815 if (bits < info->best_bits) {
816 reversed = 1;
817 info->best_bits = bits;
818 CLEAR(s->decorr_passes);
819 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
820 memcpy(s->sampleptrs[info->nterms + 1][0], s->sampleptrs[i][0],
821 s->block_samples * 4);
822 } else {
823 info->dps[ri ] = s->decorr_passes[ri];
824 info->dps[ri+1] = s->decorr_passes[ri+1];
825 decorr_mono_buffer(s->sampleptrs[ri][0], s->sampleptrs[ri+1][0],
826 s->block_samples, info->dps, ri);
827 }
828 }
829 }
830 }
831
delta_mono(WavPackEncodeContext *s, WavPackExtraInfo *info)832 static void delta_mono(WavPackEncodeContext *s, WavPackExtraInfo *info)
833 {
834 int lower = 0, delta, d;
835 uint32_t bits;
836
837 if (!s->decorr_passes[0].value)
838 return;
839 delta = s->decorr_passes[0].delta;
840
841 for (d = delta - 1; d >= 0; d--) {
842 int i;
843
844 for (i = 0; i < info->nterms && s->decorr_passes[i].value; i++) {
845 info->dps[i].value = s->decorr_passes[i].value;
846 info->dps[i].delta = d;
847 decorr_mono_buffer(s->sampleptrs[i][0], s->sampleptrs[i+1][0],
848 s->block_samples, info->dps, i);
849 }
850
851 bits = log2mono(s->sampleptrs[i][0], s->block_samples, info->log_limit);
852 if (bits >= info->best_bits)
853 break;
854
855 lower = 1;
856 info->best_bits = bits;
857 CLEAR(s->decorr_passes);
858 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
859 memcpy(s->sampleptrs[info->nterms + 1][0], s->sampleptrs[i][0],
860 s->block_samples * 4);
861 }
862
863 for (d = delta + 1; !lower && d <= 7; d++) {
864 int i;
865
866 for (i = 0; i < info->nterms && s->decorr_passes[i].value; i++) {
867 info->dps[i].value = s->decorr_passes[i].value;
868 info->dps[i].delta = d;
869 decorr_mono_buffer(s->sampleptrs[i][0], s->sampleptrs[i+1][0],
870 s->block_samples, info->dps, i);
871 }
872
873 bits = log2mono(s->sampleptrs[i][0], s->block_samples, info->log_limit);
874 if (bits >= info->best_bits)
875 break;
876
877 info->best_bits = bits;
878 CLEAR(s->decorr_passes);
879 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
880 memcpy(s->sampleptrs[info->nterms + 1][0], s->sampleptrs[i][0],
881 s->block_samples * 4);
882 }
883 }
884
allocate_buffers2(WavPackEncodeContext *s, int nterms)885 static int allocate_buffers2(WavPackEncodeContext *s, int nterms)
886 {
887 int i;
888
889 for (i = 0; i < nterms + 2; i++) {
890 av_fast_padded_malloc(&s->sampleptrs[i][0], &s->sampleptrs_size[i][0],
891 s->block_samples * 4);
892 if (!s->sampleptrs[i][0])
893 return AVERROR(ENOMEM);
894 if (!(s->flags & WV_MONO_DATA)) {
895 av_fast_padded_malloc(&s->sampleptrs[i][1], &s->sampleptrs_size[i][1],
896 s->block_samples * 4);
897 if (!s->sampleptrs[i][1])
898 return AVERROR(ENOMEM);
899 }
900 }
901
902 return 0;
903 }
904
allocate_buffers(WavPackEncodeContext *s)905 static int allocate_buffers(WavPackEncodeContext *s)
906 {
907 int i;
908
909 for (i = 0; i < 2; i++) {
910 av_fast_padded_malloc(&s->best_buffer[0], &s->best_buffer_size[0],
911 s->block_samples * 4);
912 if (!s->best_buffer[0])
913 return AVERROR(ENOMEM);
914
915 av_fast_padded_malloc(&s->temp_buffer[i][0], &s->temp_buffer_size[i][0],
916 s->block_samples * 4);
917 if (!s->temp_buffer[i][0])
918 return AVERROR(ENOMEM);
919 if (!(s->flags & WV_MONO_DATA)) {
920 av_fast_padded_malloc(&s->best_buffer[1], &s->best_buffer_size[1],
921 s->block_samples * 4);
922 if (!s->best_buffer[1])
923 return AVERROR(ENOMEM);
924
925 av_fast_padded_malloc(&s->temp_buffer[i][1], &s->temp_buffer_size[i][1],
926 s->block_samples * 4);
927 if (!s->temp_buffer[i][1])
928 return AVERROR(ENOMEM);
929 }
930 }
931
932 return 0;
933 }
934
analyze_mono(WavPackEncodeContext *s, int32_t *samples, int do_samples)935 static void analyze_mono(WavPackEncodeContext *s, int32_t *samples, int do_samples)
936 {
937 WavPackExtraInfo info;
938 int i;
939
940 info.log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
941 info.log_limit = FFMIN(6912, info.log_limit);
942
943 info.nterms = s->num_terms;
944
945 if (allocate_buffers2(s, s->num_terms))
946 return;
947
948 memcpy(info.dps, s->decorr_passes, sizeof(info.dps));
949 memcpy(s->sampleptrs[0][0], samples, s->block_samples * 4);
950
951 for (i = 0; i < info.nterms && info.dps[i].value; i++)
952 decorr_mono(s->sampleptrs[i][0], s->sampleptrs[i + 1][0],
953 s->block_samples, info.dps + i, 1);
954
955 info.best_bits = log2mono(s->sampleptrs[info.nterms][0], s->block_samples, 0) * 1;
956 memcpy(s->sampleptrs[info.nterms + 1][0], s->sampleptrs[i][0], s->block_samples * 4);
957
958 if (s->extra_flags & EXTRA_BRANCHES)
959 recurse_mono(s, &info, 0, (int) floor(s->delta_decay + 0.5),
960 log2mono(s->sampleptrs[0][0], s->block_samples, 0));
961
962 if (s->extra_flags & EXTRA_SORT_FIRST)
963 sort_mono(s, &info);
964
965 if (s->extra_flags & EXTRA_TRY_DELTAS) {
966 delta_mono(s, &info);
967
968 if ((s->extra_flags & EXTRA_ADJUST_DELTAS) && s->decorr_passes[0].value)
969 s->delta_decay = (float)((s->delta_decay * 2.0 + s->decorr_passes[0].delta) / 3.0);
970 else
971 s->delta_decay = 2.0;
972 }
973
974 if (s->extra_flags & EXTRA_SORT_LAST)
975 sort_mono(s, &info);
976
977 if (do_samples)
978 memcpy(samples, s->sampleptrs[info.nterms + 1][0], s->block_samples * 4);
979
980 for (i = 0; i < info.nterms; i++)
981 if (!s->decorr_passes[i].value)
982 break;
983
984 s->num_terms = i;
985 }
986
scan_word(WavPackEncodeContext *s, WvChannel *c, int32_t *samples, int nb_samples, int dir)987 static void scan_word(WavPackEncodeContext *s, WvChannel *c,
988 int32_t *samples, int nb_samples, int dir)
989 {
990 if (dir < 0)
991 samples += nb_samples - 1;
992
993 while (nb_samples--) {
994 uint32_t low, value = labs(samples[0]);
995
996 if (value < GET_MED(0)) {
997 DEC_MED(0);
998 } else {
999 low = GET_MED(0);
1000 INC_MED(0);
1001
1002 if (value - low < GET_MED(1)) {
1003 DEC_MED(1);
1004 } else {
1005 low += GET_MED(1);
1006 INC_MED(1);
1007
1008 if (value - low < GET_MED(2)) {
1009 DEC_MED(2);
1010 } else {
1011 INC_MED(2);
1012 }
1013 }
1014 }
1015 samples += dir;
1016 }
1017 }
1018
wv_mono(WavPackEncodeContext *s, int32_t *samples, int no_history, int do_samples)1019 static int wv_mono(WavPackEncodeContext *s, int32_t *samples,
1020 int no_history, int do_samples)
1021 {
1022 struct Decorr temp_decorr_pass, save_decorr_passes[MAX_TERMS] = {{0}};
1023 int nb_samples = s->block_samples;
1024 int buf_size = sizeof(int32_t) * nb_samples;
1025 uint32_t best_size = UINT32_MAX, size;
1026 int log_limit, pi, i, ret;
1027
1028 for (i = 0; i < nb_samples; i++)
1029 if (samples[i])
1030 break;
1031
1032 if (i == nb_samples) {
1033 CLEAR(s->decorr_passes);
1034 CLEAR(s->w);
1035 s->num_terms = 0;
1036 return 0;
1037 }
1038
1039 log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
1040 log_limit = FFMIN(6912, log_limit);
1041
1042 if ((ret = allocate_buffers(s)) < 0)
1043 return ret;
1044
1045 if (no_history || s->num_passes >= 7)
1046 s->best_decorr = s->mask_decorr = 0;
1047
1048 for (pi = 0; pi < s->num_passes;) {
1049 const WavPackDecorrSpec *wpds;
1050 int nterms, c, j;
1051
1052 if (!pi) {
1053 c = s->best_decorr;
1054 } else {
1055 if (s->mask_decorr == 0)
1056 c = 0;
1057 else
1058 c = (s->best_decorr & (s->mask_decorr - 1)) | s->mask_decorr;
1059
1060 if (c == s->best_decorr) {
1061 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1062 continue;
1063 }
1064 }
1065
1066 wpds = &s->decorr_specs[c];
1067 nterms = decorr_filter_nterms[s->decorr_filter];
1068
1069 while (1) {
1070 memcpy(s->temp_buffer[0][0], samples, buf_size);
1071 CLEAR(save_decorr_passes);
1072
1073 for (j = 0; j < nterms; j++) {
1074 CLEAR(temp_decorr_pass);
1075 temp_decorr_pass.delta = wpds->delta;
1076 temp_decorr_pass.value = wpds->terms[j];
1077
1078 if (temp_decorr_pass.value < 0)
1079 temp_decorr_pass.value = 1;
1080
1081 decorr_mono(s->temp_buffer[j&1][0], s->temp_buffer[~j&1][0],
1082 FFMIN(nb_samples, 2048), &temp_decorr_pass, -1);
1083
1084 if (j) {
1085 CLEAR(temp_decorr_pass.samplesA);
1086 } else {
1087 reverse_mono_decorr(&temp_decorr_pass);
1088 }
1089
1090 memcpy(save_decorr_passes + j, &temp_decorr_pass, sizeof(struct Decorr));
1091 decorr_mono(s->temp_buffer[j&1][0], s->temp_buffer[~j&1][0],
1092 nb_samples, &temp_decorr_pass, 1);
1093 }
1094
1095 size = log2mono(s->temp_buffer[j&1][0], nb_samples, log_limit);
1096 if (size != UINT32_MAX || !nterms)
1097 break;
1098 nterms >>= 1;
1099 }
1100
1101 if (size < best_size) {
1102 memcpy(s->best_buffer[0], s->temp_buffer[j&1][0], buf_size);
1103 memcpy(s->decorr_passes, save_decorr_passes, sizeof(struct Decorr) * MAX_TERMS);
1104 s->num_terms = nterms;
1105 s->best_decorr = c;
1106 best_size = size;
1107 }
1108
1109 if (pi++)
1110 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1111 }
1112
1113 if (s->extra_flags)
1114 analyze_mono(s, samples, do_samples);
1115 else if (do_samples)
1116 memcpy(samples, s->best_buffer[0], buf_size);
1117
1118 if (no_history || s->extra_flags) {
1119 CLEAR(s->w);
1120 scan_word(s, &s->w.c[0], s->best_buffer[0], nb_samples, -1);
1121 }
1122 return 0;
1123 }
1124
decorr_stereo(int32_t *in_left, int32_t *in_right, int32_t *out_left, int32_t *out_right, int nb_samples, struct Decorr *dpp, int dir)1125 static void decorr_stereo(int32_t *in_left, int32_t *in_right,
1126 int32_t *out_left, int32_t *out_right,
1127 int nb_samples, struct Decorr *dpp, int dir)
1128 {
1129 int m = 0, i;
1130
1131 dpp->sumA = dpp->sumB = 0;
1132
1133 if (dir < 0) {
1134 out_left += nb_samples - 1;
1135 out_right += nb_samples - 1;
1136 in_left += nb_samples - 1;
1137 in_right += nb_samples - 1;
1138 }
1139
1140 dpp->weightA = restore_weight(store_weight(dpp->weightA));
1141 dpp->weightB = restore_weight(store_weight(dpp->weightB));
1142
1143 for (i = 0; i < MAX_TERM; i++) {
1144 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
1145 dpp->samplesB[i] = wp_exp2(log2s(dpp->samplesB[i]));
1146 }
1147
1148 switch (dpp->value) {
1149 case 2:
1150 while (nb_samples--) {
1151 int32_t sam, tmp;
1152
1153 sam = dpp->samplesA[0];
1154 dpp->samplesA[0] = dpp->samplesA[1];
1155 out_left[0] = tmp = (dpp->samplesA[1] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1156 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1157 dpp->sumA += dpp->weightA;
1158
1159 sam = dpp->samplesB[0];
1160 dpp->samplesB[0] = dpp->samplesB[1];
1161 out_right[0] = tmp = (dpp->samplesB[1] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1162 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1163 dpp->sumB += dpp->weightB;
1164
1165 in_left += dir;
1166 out_left += dir;
1167 in_right += dir;
1168 out_right += dir;
1169 }
1170 break;
1171 case 17:
1172 while (nb_samples--) {
1173 int32_t sam, tmp;
1174
1175 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1176 dpp->samplesA[1] = dpp->samplesA[0];
1177 out_left[0] = tmp = (dpp->samplesA[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1178 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1179 dpp->sumA += dpp->weightA;
1180
1181 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1182 dpp->samplesB[1] = dpp->samplesB[0];
1183 out_right[0] = tmp = (dpp->samplesB[0] = in_right[0]) - APPLY_WEIGHT (dpp->weightB, sam);
1184 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1185 dpp->sumB += dpp->weightB;
1186
1187 in_left += dir;
1188 out_left += dir;
1189 in_right += dir;
1190 out_right += dir;
1191 }
1192 break;
1193 case 18:
1194 while (nb_samples--) {
1195 int32_t sam, tmp;
1196
1197 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
1198 dpp->samplesA[1] = dpp->samplesA[0];
1199 out_left[0] = tmp = (dpp->samplesA[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1200 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1201 dpp->sumA += dpp->weightA;
1202
1203 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
1204 dpp->samplesB[1] = dpp->samplesB[0];
1205 out_right[0] = tmp = (dpp->samplesB[0] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1206 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1207 dpp->sumB += dpp->weightB;
1208
1209 in_left += dir;
1210 out_left += dir;
1211 in_right += dir;
1212 out_right += dir;
1213 }
1214 break;
1215 default: {
1216 int k = dpp->value & (MAX_TERM - 1);
1217
1218 while (nb_samples--) {
1219 int32_t sam, tmp;
1220
1221 sam = dpp->samplesA[m];
1222 out_left[0] = tmp = (dpp->samplesA[k] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1223 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1224 dpp->sumA += dpp->weightA;
1225
1226 sam = dpp->samplesB[m];
1227 out_right[0] = tmp = (dpp->samplesB[k] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1228 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1229 dpp->sumB += dpp->weightB;
1230
1231 in_left += dir;
1232 out_left += dir;
1233 in_right += dir;
1234 out_right += dir;
1235 m = (m + 1) & (MAX_TERM - 1);
1236 k = (k + 1) & (MAX_TERM - 1);
1237 }
1238
1239 if (m) {
1240 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
1241 int k;
1242
1243 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
1244 memcpy(temp_B, dpp->samplesB, sizeof(dpp->samplesB));
1245
1246 for (k = 0; k < MAX_TERM; k++) {
1247 dpp->samplesA[k] = temp_A[m];
1248 dpp->samplesB[k] = temp_B[m];
1249 m = (m + 1) & (MAX_TERM - 1);
1250 }
1251 }
1252 break;
1253 }
1254 case -1:
1255 while (nb_samples--) {
1256 int32_t sam_A, sam_B, tmp;
1257
1258 sam_A = dpp->samplesA[0];
1259 out_left[0] = tmp = (sam_B = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam_A);
1260 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1261 dpp->sumA += dpp->weightA;
1262
1263 out_right[0] = tmp = (dpp->samplesA[0] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam_B);
1264 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1265 dpp->sumB += dpp->weightB;
1266
1267 in_left += dir;
1268 out_left += dir;
1269 in_right += dir;
1270 out_right += dir;
1271 }
1272 break;
1273 case -2:
1274 while (nb_samples--) {
1275 int32_t sam_A, sam_B, tmp;
1276
1277 sam_B = dpp->samplesB[0];
1278 out_right[0] = tmp = (sam_A = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam_B);
1279 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1280 dpp->sumB += dpp->weightB;
1281
1282 out_left[0] = tmp = (dpp->samplesB[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam_A);
1283 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1284 dpp->sumA += dpp->weightA;
1285
1286 in_left += dir;
1287 out_left += dir;
1288 in_right += dir;
1289 out_right += dir;
1290 }
1291 break;
1292 case -3:
1293 while (nb_samples--) {
1294 int32_t sam_A, sam_B, tmp;
1295
1296 sam_A = dpp->samplesA[0];
1297 sam_B = dpp->samplesB[0];
1298
1299 dpp->samplesA[0] = tmp = in_right[0];
1300 out_right[0] = tmp -= APPLY_WEIGHT(dpp->weightB, sam_B);
1301 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1302 dpp->sumB += dpp->weightB;
1303
1304 dpp->samplesB[0] = tmp = in_left[0];
1305 out_left[0] = tmp -= APPLY_WEIGHT(dpp->weightA, sam_A);
1306 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1307 dpp->sumA += dpp->weightA;
1308
1309 in_left += dir;
1310 out_left += dir;
1311 in_right += dir;
1312 out_right += dir;
1313 }
1314 break;
1315 }
1316 }
1317
reverse_decorr(struct Decorr *dpp)1318 static void reverse_decorr(struct Decorr *dpp)
1319 {
1320 if (dpp->value > MAX_TERM) {
1321 int32_t sam_A, sam_B;
1322
1323 if (dpp->value & 1) {
1324 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1325 sam_B = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1326 } else {
1327 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
1328 sam_B = (3 * dpp->samplesB[0] - dpp->samplesB[1]) >> 1;
1329 }
1330
1331 dpp->samplesA[1] = dpp->samplesA[0];
1332 dpp->samplesB[1] = dpp->samplesB[0];
1333 dpp->samplesA[0] = sam_A;
1334 dpp->samplesB[0] = sam_B;
1335
1336 if (dpp->value & 1) {
1337 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1338 sam_B = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1339 } else {
1340 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
1341 sam_B = (3 * dpp->samplesB[0] - dpp->samplesB[1]) >> 1;
1342 }
1343
1344 dpp->samplesA[1] = sam_A;
1345 dpp->samplesB[1] = sam_B;
1346 } else if (dpp->value > 1) {
1347 int i, j, k;
1348
1349 for (i = 0, j = dpp->value - 1, k = 0; k < dpp->value / 2; i++, j--, k++) {
1350 i &= (MAX_TERM - 1);
1351 j &= (MAX_TERM - 1);
1352 dpp->samplesA[i] ^= dpp->samplesA[j];
1353 dpp->samplesA[j] ^= dpp->samplesA[i];
1354 dpp->samplesA[i] ^= dpp->samplesA[j];
1355 dpp->samplesB[i] ^= dpp->samplesB[j];
1356 dpp->samplesB[j] ^= dpp->samplesB[i];
1357 dpp->samplesB[i] ^= dpp->samplesB[j];
1358 }
1359 }
1360 }
1361
decorr_stereo_quick(int32_t *in_left, int32_t *in_right, int32_t *out_left, int32_t *out_right, int nb_samples, struct Decorr *dpp)1362 static void decorr_stereo_quick(int32_t *in_left, int32_t *in_right,
1363 int32_t *out_left, int32_t *out_right,
1364 int nb_samples, struct Decorr *dpp)
1365 {
1366 int m = 0, i;
1367
1368 dpp->weightA = restore_weight(store_weight(dpp->weightA));
1369 dpp->weightB = restore_weight(store_weight(dpp->weightB));
1370
1371 for (i = 0; i < MAX_TERM; i++) {
1372 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
1373 dpp->samplesB[i] = wp_exp2(log2s(dpp->samplesB[i]));
1374 }
1375
1376 switch (dpp->value) {
1377 case 2:
1378 for (i = 0; i < nb_samples; i++) {
1379 int32_t sam, tmp;
1380
1381 sam = dpp->samplesA[0];
1382 dpp->samplesA[0] = dpp->samplesA[1];
1383 out_left[i] = tmp = (dpp->samplesA[1] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1384 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1385
1386 sam = dpp->samplesB[0];
1387 dpp->samplesB[0] = dpp->samplesB[1];
1388 out_right[i] = tmp = (dpp->samplesB[1] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1389 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1390 }
1391 break;
1392 case 17:
1393 for (i = 0; i < nb_samples; i++) {
1394 int32_t sam, tmp;
1395
1396 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1397 dpp->samplesA[1] = dpp->samplesA[0];
1398 out_left[i] = tmp = (dpp->samplesA[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1399 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1400
1401 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1402 dpp->samplesB[1] = dpp->samplesB[0];
1403 out_right[i] = tmp = (dpp->samplesB[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1404 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1405 }
1406 break;
1407 case 18:
1408 for (i = 0; i < nb_samples; i++) {
1409 int32_t sam, tmp;
1410
1411 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
1412 dpp->samplesA[1] = dpp->samplesA[0];
1413 out_left[i] = tmp = (dpp->samplesA[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1414 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1415
1416 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
1417 dpp->samplesB[1] = dpp->samplesB[0];
1418 out_right[i] = tmp = (dpp->samplesB[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1419 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1420 }
1421 break;
1422 default: {
1423 int k = dpp->value & (MAX_TERM - 1);
1424
1425 for (i = 0; i < nb_samples; i++) {
1426 int32_t sam, tmp;
1427
1428 sam = dpp->samplesA[m];
1429 out_left[i] = tmp = (dpp->samplesA[k] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1430 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1431
1432 sam = dpp->samplesB[m];
1433 out_right[i] = tmp = (dpp->samplesB[k] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1434 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1435
1436 m = (m + 1) & (MAX_TERM - 1);
1437 k = (k + 1) & (MAX_TERM - 1);
1438 }
1439
1440 if (m) {
1441 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
1442 int k;
1443
1444 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
1445 memcpy(temp_B, dpp->samplesB, sizeof(dpp->samplesB));
1446
1447 for (k = 0; k < MAX_TERM; k++) {
1448 dpp->samplesA[k] = temp_A[m];
1449 dpp->samplesB[k] = temp_B[m];
1450 m = (m + 1) & (MAX_TERM - 1);
1451 }
1452 }
1453 break;
1454 }
1455 case -1:
1456 for (i = 0; i < nb_samples; i++) {
1457 int32_t sam_A, sam_B, tmp;
1458
1459 sam_A = dpp->samplesA[0];
1460 out_left[i] = tmp = (sam_B = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
1461 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1462
1463 out_right[i] = tmp = (dpp->samplesA[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
1464 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1465 }
1466 break;
1467 case -2:
1468 for (i = 0; i < nb_samples; i++) {
1469 int32_t sam_A, sam_B, tmp;
1470
1471 sam_B = dpp->samplesB[0];
1472 out_right[i] = tmp = (sam_A = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
1473 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1474
1475 out_left[i] = tmp = (dpp->samplesB[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
1476 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1477 }
1478 break;
1479 case -3:
1480 for (i = 0; i < nb_samples; i++) {
1481 int32_t sam_A, sam_B, tmp;
1482
1483 sam_A = dpp->samplesA[0];
1484 sam_B = dpp->samplesB[0];
1485
1486 dpp->samplesA[0] = tmp = in_right[i];
1487 out_right[i] = tmp -= APPLY_WEIGHT_I(dpp->weightB, sam_B);
1488 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1489
1490 dpp->samplesB[0] = tmp = in_left[i];
1491 out_left[i] = tmp -= APPLY_WEIGHT_I(dpp->weightA, sam_A);
1492 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1493 }
1494 break;
1495 }
1496 }
1497
decorr_stereo_buffer(WavPackExtraInfo *info, int32_t *in_left, int32_t *in_right, int32_t *out_left, int32_t *out_right, int nb_samples, int tindex)1498 static void decorr_stereo_buffer(WavPackExtraInfo *info,
1499 int32_t *in_left, int32_t *in_right,
1500 int32_t *out_left, int32_t *out_right,
1501 int nb_samples, int tindex)
1502 {
1503 struct Decorr dp = {0}, *dppi = info->dps + tindex;
1504 int delta = dppi->delta, pre_delta;
1505 int term = dppi->value;
1506
1507 if (delta == 7)
1508 pre_delta = 7;
1509 else if (delta < 2)
1510 pre_delta = 3;
1511 else
1512 pre_delta = delta + 1;
1513
1514 dp.value = term;
1515 dp.delta = pre_delta;
1516 decorr_stereo(in_left, in_right, out_left, out_right,
1517 FFMIN(2048, nb_samples), &dp, -1);
1518 dp.delta = delta;
1519
1520 if (tindex == 0) {
1521 reverse_decorr(&dp);
1522 } else {
1523 CLEAR(dp.samplesA);
1524 CLEAR(dp.samplesB);
1525 }
1526
1527 memcpy(dppi->samplesA, dp.samplesA, sizeof(dp.samplesA));
1528 memcpy(dppi->samplesB, dp.samplesB, sizeof(dp.samplesB));
1529 dppi->weightA = dp.weightA;
1530 dppi->weightB = dp.weightB;
1531
1532 if (delta == 0) {
1533 dp.delta = 1;
1534 decorr_stereo(in_left, in_right, out_left, out_right, nb_samples, &dp, 1);
1535 dp.delta = 0;
1536 memcpy(dp.samplesA, dppi->samplesA, sizeof(dp.samplesA));
1537 memcpy(dp.samplesB, dppi->samplesB, sizeof(dp.samplesB));
1538 dppi->weightA = dp.weightA = dp.sumA / nb_samples;
1539 dppi->weightB = dp.weightB = dp.sumB / nb_samples;
1540 }
1541
1542 if (info->gt16bit)
1543 decorr_stereo(in_left, in_right, out_left, out_right,
1544 nb_samples, &dp, 1);
1545 else
1546 decorr_stereo_quick(in_left, in_right, out_left, out_right,
1547 nb_samples, &dp);
1548 }
1549
sort_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info)1550 static void sort_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info)
1551 {
1552 int reversed = 1;
1553 uint32_t bits;
1554
1555 while (reversed) {
1556 int ri, i;
1557
1558 memcpy(info->dps, s->decorr_passes, sizeof(s->decorr_passes));
1559 reversed = 0;
1560
1561 for (ri = 0; ri < info->nterms && s->decorr_passes[ri].value; ri++) {
1562
1563 if (ri + 1 >= info->nterms || !s->decorr_passes[ri+1].value)
1564 break;
1565
1566 if (s->decorr_passes[ri].value == s->decorr_passes[ri+1].value) {
1567 decorr_stereo_buffer(info,
1568 s->sampleptrs[ri ][0], s->sampleptrs[ri ][1],
1569 s->sampleptrs[ri+1][0], s->sampleptrs[ri+1][1],
1570 s->block_samples, ri);
1571 continue;
1572 }
1573
1574 info->dps[ri ] = s->decorr_passes[ri+1];
1575 info->dps[ri+1] = s->decorr_passes[ri ];
1576
1577 for (i = ri; i < info->nterms && s->decorr_passes[i].value; i++)
1578 decorr_stereo_buffer(info,
1579 s->sampleptrs[i ][0], s->sampleptrs[i ][1],
1580 s->sampleptrs[i+1][0], s->sampleptrs[i+1][1],
1581 s->block_samples, i);
1582
1583 bits = log2stereo(s->sampleptrs[i][0], s->sampleptrs[i][1],
1584 s->block_samples, info->log_limit);
1585
1586 if (bits < info->best_bits) {
1587 reversed = 1;
1588 info->best_bits = bits;
1589 CLEAR(s->decorr_passes);
1590 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
1591 memcpy(s->sampleptrs[info->nterms + 1][0],
1592 s->sampleptrs[i][0], s->block_samples * 4);
1593 memcpy(s->sampleptrs[info->nterms + 1][1],
1594 s->sampleptrs[i][1], s->block_samples * 4);
1595 } else {
1596 info->dps[ri ] = s->decorr_passes[ri ];
1597 info->dps[ri+1] = s->decorr_passes[ri+1];
1598 decorr_stereo_buffer(info,
1599 s->sampleptrs[ri ][0], s->sampleptrs[ri ][1],
1600 s->sampleptrs[ri+1][0], s->sampleptrs[ri+1][1],
1601 s->block_samples, ri);
1602 }
1603 }
1604 }
1605 }
1606
delta_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info)1607 static void delta_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info)
1608 {
1609 int lower = 0, delta, d, i;
1610 uint32_t bits;
1611
1612 if (!s->decorr_passes[0].value)
1613 return;
1614 delta = s->decorr_passes[0].delta;
1615
1616 for (d = delta - 1; d >= 0; d--) {
1617 for (i = 0; i < info->nterms && s->decorr_passes[i].value; i++) {
1618 info->dps[i].value = s->decorr_passes[i].value;
1619 info->dps[i].delta = d;
1620 decorr_stereo_buffer(info,
1621 s->sampleptrs[i ][0], s->sampleptrs[i ][1],
1622 s->sampleptrs[i+1][0], s->sampleptrs[i+1][1],
1623 s->block_samples, i);
1624 }
1625
1626 bits = log2stereo(s->sampleptrs[i][0], s->sampleptrs[i][1],
1627 s->block_samples, info->log_limit);
1628 if (bits >= info->best_bits)
1629 break;
1630 lower = 1;
1631 info->best_bits = bits;
1632 CLEAR(s->decorr_passes);
1633 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
1634 memcpy(s->sampleptrs[info->nterms + 1][0], s->sampleptrs[i][0],
1635 s->block_samples * 4);
1636 memcpy(s->sampleptrs[info->nterms + 1][1], s->sampleptrs[i][1],
1637 s->block_samples * 4);
1638 }
1639
1640 for (d = delta + 1; !lower && d <= 7; d++) {
1641 for (i = 0; i < info->nterms && s->decorr_passes[i].value; i++) {
1642 info->dps[i].value = s->decorr_passes[i].value;
1643 info->dps[i].delta = d;
1644 decorr_stereo_buffer(info,
1645 s->sampleptrs[i ][0], s->sampleptrs[i ][1],
1646 s->sampleptrs[i+1][0], s->sampleptrs[i+1][1],
1647 s->block_samples, i);
1648 }
1649
1650 bits = log2stereo(s->sampleptrs[i][0], s->sampleptrs[i][1],
1651 s->block_samples, info->log_limit);
1652
1653 if (bits < info->best_bits) {
1654 info->best_bits = bits;
1655 CLEAR(s->decorr_passes);
1656 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * i);
1657 memcpy(s->sampleptrs[info->nterms + 1][0],
1658 s->sampleptrs[i][0], s->block_samples * 4);
1659 memcpy(s->sampleptrs[info->nterms + 1][1],
1660 s->sampleptrs[i][1], s->block_samples * 4);
1661 }
1662 else
1663 break;
1664 }
1665 }
1666
recurse_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info, int depth, int delta, uint32_t input_bits)1667 static void recurse_stereo(WavPackEncodeContext *s, WavPackExtraInfo *info,
1668 int depth, int delta, uint32_t input_bits)
1669 {
1670 int term, branches = s->num_branches - depth;
1671 int32_t *in_left, *in_right, *out_left, *out_right;
1672 uint32_t term_bits[22], bits;
1673
1674 if (branches < 1 || depth + 1 == info->nterms)
1675 branches = 1;
1676
1677 CLEAR(term_bits);
1678 in_left = s->sampleptrs[depth ][0];
1679 in_right = s->sampleptrs[depth ][1];
1680 out_left = s->sampleptrs[depth + 1][0];
1681 out_right = s->sampleptrs[depth + 1][1];
1682
1683 for (term = -3; term <= 18; term++) {
1684 if (!term || (term > 8 && term < 17))
1685 continue;
1686
1687 if (term == 17 && branches == 1 && depth + 1 < info->nterms)
1688 continue;
1689
1690 if (term == -1 || term == -2)
1691 if (!(s->flags & WV_CROSS_DECORR))
1692 continue;
1693
1694 if (!s->extra_flags && (term > 4 && term < 17))
1695 continue;
1696
1697 info->dps[depth].value = term;
1698 info->dps[depth].delta = delta;
1699 decorr_stereo_buffer(info, in_left, in_right, out_left, out_right,
1700 s->block_samples, depth);
1701 bits = log2stereo(out_left, out_right, s->block_samples, info->log_limit);
1702
1703 if (bits < info->best_bits) {
1704 info->best_bits = bits;
1705 CLEAR(s->decorr_passes);
1706 memcpy(s->decorr_passes, info->dps, sizeof(info->dps[0]) * (depth + 1));
1707 memcpy(s->sampleptrs[info->nterms + 1][0], s->sampleptrs[depth + 1][0],
1708 s->block_samples * 4);
1709 memcpy(s->sampleptrs[info->nterms + 1][1], s->sampleptrs[depth + 1][1],
1710 s->block_samples * 4);
1711 }
1712
1713 term_bits[term + 3] = bits;
1714 }
1715
1716 while (depth + 1 < info->nterms && branches--) {
1717 uint32_t local_best_bits = input_bits;
1718 int best_term = 0, i;
1719
1720 for (i = 0; i < 22; i++)
1721 if (term_bits[i] && term_bits[i] < local_best_bits) {
1722 local_best_bits = term_bits[i];
1723 best_term = i - 3;
1724 }
1725
1726 if (!best_term)
1727 break;
1728
1729 term_bits[best_term + 3] = 0;
1730
1731 info->dps[depth].value = best_term;
1732 info->dps[depth].delta = delta;
1733 decorr_stereo_buffer(info, in_left, in_right, out_left, out_right,
1734 s->block_samples, depth);
1735
1736 recurse_stereo(s, info, depth + 1, delta, local_best_bits);
1737 }
1738 }
1739
analyze_stereo(WavPackEncodeContext *s, int32_t *in_left, int32_t *in_right, int do_samples)1740 static void analyze_stereo(WavPackEncodeContext *s,
1741 int32_t *in_left, int32_t *in_right,
1742 int do_samples)
1743 {
1744 WavPackExtraInfo info;
1745 int i;
1746
1747 info.gt16bit = ((s->flags & MAG_MASK) >> MAG_LSB) >= 16;
1748
1749 info.log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
1750 info.log_limit = FFMIN(6912, info.log_limit);
1751
1752 info.nterms = s->num_terms;
1753
1754 if (allocate_buffers2(s, s->num_terms))
1755 return;
1756
1757 memcpy(info.dps, s->decorr_passes, sizeof(info.dps));
1758 memcpy(s->sampleptrs[0][0], in_left, s->block_samples * 4);
1759 memcpy(s->sampleptrs[0][1], in_right, s->block_samples * 4);
1760
1761 for (i = 0; i < info.nterms && info.dps[i].value; i++)
1762 if (info.gt16bit)
1763 decorr_stereo(s->sampleptrs[i ][0], s->sampleptrs[i ][1],
1764 s->sampleptrs[i + 1][0], s->sampleptrs[i + 1][1],
1765 s->block_samples, info.dps + i, 1);
1766 else
1767 decorr_stereo_quick(s->sampleptrs[i ][0], s->sampleptrs[i ][1],
1768 s->sampleptrs[i + 1][0], s->sampleptrs[i + 1][1],
1769 s->block_samples, info.dps + i);
1770
1771 info.best_bits = log2stereo(s->sampleptrs[info.nterms][0], s->sampleptrs[info.nterms][1],
1772 s->block_samples, 0);
1773
1774 memcpy(s->sampleptrs[info.nterms + 1][0], s->sampleptrs[i][0], s->block_samples * 4);
1775 memcpy(s->sampleptrs[info.nterms + 1][1], s->sampleptrs[i][1], s->block_samples * 4);
1776
1777 if (s->extra_flags & EXTRA_BRANCHES)
1778 recurse_stereo(s, &info, 0, (int) floor(s->delta_decay + 0.5),
1779 log2stereo(s->sampleptrs[0][0], s->sampleptrs[0][1],
1780 s->block_samples, 0));
1781
1782 if (s->extra_flags & EXTRA_SORT_FIRST)
1783 sort_stereo(s, &info);
1784
1785 if (s->extra_flags & EXTRA_TRY_DELTAS) {
1786 delta_stereo(s, &info);
1787
1788 if ((s->extra_flags & EXTRA_ADJUST_DELTAS) && s->decorr_passes[0].value)
1789 s->delta_decay = (float)((s->delta_decay * 2.0 + s->decorr_passes[0].delta) / 3.0);
1790 else
1791 s->delta_decay = 2.0;
1792 }
1793
1794 if (s->extra_flags & EXTRA_SORT_LAST)
1795 sort_stereo(s, &info);
1796
1797 if (do_samples) {
1798 memcpy(in_left, s->sampleptrs[info.nterms + 1][0], s->block_samples * 4);
1799 memcpy(in_right, s->sampleptrs[info.nterms + 1][1], s->block_samples * 4);
1800 }
1801
1802 for (i = 0; i < info.nterms; i++)
1803 if (!s->decorr_passes[i].value)
1804 break;
1805
1806 s->num_terms = i;
1807 }
1808
wv_stereo(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int no_history, int do_samples)1809 static int wv_stereo(WavPackEncodeContext *s,
1810 int32_t *samples_l, int32_t *samples_r,
1811 int no_history, int do_samples)
1812 {
1813 struct Decorr temp_decorr_pass, save_decorr_passes[MAX_TERMS] = {{0}};
1814 int nb_samples = s->block_samples, ret;
1815 int buf_size = sizeof(int32_t) * nb_samples;
1816 int log_limit, force_js = 0, force_ts = 0, got_js = 0, pi, i;
1817 uint32_t best_size = UINT32_MAX, size;
1818
1819 for (i = 0; i < nb_samples; i++)
1820 if (samples_l[i] || samples_r[i])
1821 break;
1822
1823 if (i == nb_samples) {
1824 s->flags &= ~((uint32_t) WV_JOINT_STEREO);
1825 CLEAR(s->decorr_passes);
1826 CLEAR(s->w);
1827 s->num_terms = 0;
1828 return 0;
1829 }
1830
1831 log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
1832 log_limit = FFMIN(6912, log_limit);
1833
1834 if (s->joint != -1) {
1835 force_js = s->joint;
1836 force_ts = !s->joint;
1837 }
1838
1839 if ((ret = allocate_buffers(s)) < 0)
1840 return ret;
1841
1842 if (no_history || s->num_passes >= 7)
1843 s->best_decorr = s->mask_decorr = 0;
1844
1845 for (pi = 0; pi < s->num_passes;) {
1846 const WavPackDecorrSpec *wpds;
1847 int nterms, c, j;
1848
1849 if (!pi)
1850 c = s->best_decorr;
1851 else {
1852 if (s->mask_decorr == 0)
1853 c = 0;
1854 else
1855 c = (s->best_decorr & (s->mask_decorr - 1)) | s->mask_decorr;
1856
1857 if (c == s->best_decorr) {
1858 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1859 continue;
1860 }
1861 }
1862
1863 wpds = &s->decorr_specs[c];
1864 nterms = decorr_filter_nterms[s->decorr_filter];
1865
1866 while (1) {
1867 if (force_js || (wpds->joint_stereo && !force_ts)) {
1868 if (!got_js) {
1869 av_fast_padded_malloc(&s->js_left, &s->js_left_size, buf_size);
1870 av_fast_padded_malloc(&s->js_right, &s->js_right_size, buf_size);
1871 memcpy(s->js_left, samples_l, buf_size);
1872 memcpy(s->js_right, samples_r, buf_size);
1873
1874 for (i = 0; i < nb_samples; i++)
1875 s->js_right[i] += ((s->js_left[i] -= s->js_right[i]) >> 1);
1876 got_js = 1;
1877 }
1878
1879 memcpy(s->temp_buffer[0][0], s->js_left, buf_size);
1880 memcpy(s->temp_buffer[0][1], s->js_right, buf_size);
1881 } else {
1882 memcpy(s->temp_buffer[0][0], samples_l, buf_size);
1883 memcpy(s->temp_buffer[0][1], samples_r, buf_size);
1884 }
1885
1886 CLEAR(save_decorr_passes);
1887
1888 for (j = 0; j < nterms; j++) {
1889 CLEAR(temp_decorr_pass);
1890 temp_decorr_pass.delta = wpds->delta;
1891 temp_decorr_pass.value = wpds->terms[j];
1892
1893 if (temp_decorr_pass.value < 0 && !(s->flags & WV_CROSS_DECORR))
1894 temp_decorr_pass.value = -3;
1895
1896 decorr_stereo(s->temp_buffer[ j&1][0], s->temp_buffer[ j&1][1],
1897 s->temp_buffer[~j&1][0], s->temp_buffer[~j&1][1],
1898 FFMIN(2048, nb_samples), &temp_decorr_pass, -1);
1899
1900 if (j) {
1901 CLEAR(temp_decorr_pass.samplesA);
1902 CLEAR(temp_decorr_pass.samplesB);
1903 } else {
1904 reverse_decorr(&temp_decorr_pass);
1905 }
1906
1907 memcpy(save_decorr_passes + j, &temp_decorr_pass, sizeof(struct Decorr));
1908
1909 if (((s->flags & MAG_MASK) >> MAG_LSB) >= 16)
1910 decorr_stereo(s->temp_buffer[ j&1][0], s->temp_buffer[ j&1][1],
1911 s->temp_buffer[~j&1][0], s->temp_buffer[~j&1][1],
1912 nb_samples, &temp_decorr_pass, 1);
1913 else
1914 decorr_stereo_quick(s->temp_buffer[ j&1][0], s->temp_buffer[ j&1][1],
1915 s->temp_buffer[~j&1][0], s->temp_buffer[~j&1][1],
1916 nb_samples, &temp_decorr_pass);
1917 }
1918
1919 size = log2stereo(s->temp_buffer[j&1][0], s->temp_buffer[j&1][1],
1920 nb_samples, log_limit);
1921 if (size != UINT32_MAX || !nterms)
1922 break;
1923 nterms >>= 1;
1924 }
1925
1926 if (size < best_size) {
1927 memcpy(s->best_buffer[0], s->temp_buffer[j&1][0], buf_size);
1928 memcpy(s->best_buffer[1], s->temp_buffer[j&1][1], buf_size);
1929 memcpy(s->decorr_passes, save_decorr_passes, sizeof(struct Decorr) * MAX_TERMS);
1930 s->num_terms = nterms;
1931 s->best_decorr = c;
1932 best_size = size;
1933 }
1934
1935 if (pi++)
1936 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1937 }
1938
1939 if (force_js || (s->decorr_specs[s->best_decorr].joint_stereo && !force_ts))
1940 s->flags |= WV_JOINT_STEREO;
1941 else
1942 s->flags &= ~((uint32_t) WV_JOINT_STEREO);
1943
1944 if (s->extra_flags) {
1945 if (s->flags & WV_JOINT_STEREO) {
1946 analyze_stereo(s, s->js_left, s->js_right, do_samples);
1947
1948 if (do_samples) {
1949 memcpy(samples_l, s->js_left, buf_size);
1950 memcpy(samples_r, s->js_right, buf_size);
1951 }
1952 } else
1953 analyze_stereo(s, samples_l, samples_r, do_samples);
1954 } else if (do_samples) {
1955 memcpy(samples_l, s->best_buffer[0], buf_size);
1956 memcpy(samples_r, s->best_buffer[1], buf_size);
1957 }
1958
1959 if (s->extra_flags || no_history ||
1960 s->joint_stereo != s->decorr_specs[s->best_decorr].joint_stereo) {
1961 s->joint_stereo = s->decorr_specs[s->best_decorr].joint_stereo;
1962 CLEAR(s->w);
1963 scan_word(s, &s->w.c[0], s->best_buffer[0], nb_samples, -1);
1964 scan_word(s, &s->w.c[1], s->best_buffer[1], nb_samples, -1);
1965 }
1966 return 0;
1967 }
1968
encode_flush(WavPackEncodeContext *s)1969 static void encode_flush(WavPackEncodeContext *s)
1970 {
1971 WavPackWords *w = &s->w;
1972 PutBitContext *pb = &s->pb;
1973
1974 if (w->zeros_acc) {
1975 int cbits = count_bits(w->zeros_acc);
1976
1977 do {
1978 if (cbits > 31) {
1979 put_bits(pb, 31, 0x7FFFFFFF);
1980 cbits -= 31;
1981 } else {
1982 put_bits(pb, cbits, (1U << cbits) - 1);
1983 cbits = 0;
1984 }
1985 } while (cbits);
1986
1987 put_bits(pb, 1, 0);
1988
1989 while (w->zeros_acc > 1) {
1990 put_bits(pb, 1, w->zeros_acc & 1);
1991 w->zeros_acc >>= 1;
1992 }
1993
1994 w->zeros_acc = 0;
1995 }
1996
1997 if (w->holding_one) {
1998 if (w->holding_one >= 16) {
1999 int cbits;
2000
2001 put_bits(pb, 16, (1 << 16) - 1);
2002 put_bits(pb, 1, 0);
2003 w->holding_one -= 16;
2004 cbits = count_bits(w->holding_one);
2005
2006 do {
2007 if (cbits > 31) {
2008 put_bits(pb, 31, 0x7FFFFFFF);
2009 cbits -= 31;
2010 } else {
2011 put_bits(pb, cbits, (1U << cbits) - 1);
2012 cbits = 0;
2013 }
2014 } while (cbits);
2015
2016 put_bits(pb, 1, 0);
2017
2018 while (w->holding_one > 1) {
2019 put_bits(pb, 1, w->holding_one & 1);
2020 w->holding_one >>= 1;
2021 }
2022
2023 w->holding_zero = 0;
2024 } else {
2025 put_bits(pb, w->holding_one, (1 << w->holding_one) - 1);
2026 }
2027
2028 w->holding_one = 0;
2029 }
2030
2031 if (w->holding_zero) {
2032 put_bits(pb, 1, 0);
2033 w->holding_zero = 0;
2034 }
2035
2036 if (w->pend_count) {
2037 put_bits(pb, w->pend_count, w->pend_data);
2038 w->pend_data = w->pend_count = 0;
2039 }
2040 }
2041
wavpack_encode_sample(WavPackEncodeContext *s, WvChannel *c, int32_t sample)2042 static void wavpack_encode_sample(WavPackEncodeContext *s, WvChannel *c, int32_t sample)
2043 {
2044 WavPackWords *w = &s->w;
2045 uint32_t ones_count, low, high;
2046 int sign = sample < 0;
2047
2048 if (s->w.c[0].median[0] < 2 && !s->w.holding_zero && s->w.c[1].median[0] < 2) {
2049 if (w->zeros_acc) {
2050 if (sample)
2051 encode_flush(s);
2052 else {
2053 w->zeros_acc++;
2054 return;
2055 }
2056 } else if (sample) {
2057 put_bits(&s->pb, 1, 0);
2058 } else {
2059 CLEAR(s->w.c[0].median);
2060 CLEAR(s->w.c[1].median);
2061 w->zeros_acc = 1;
2062 return;
2063 }
2064 }
2065
2066 if (sign)
2067 sample = ~sample;
2068
2069 if (sample < (int32_t) GET_MED(0)) {
2070 ones_count = low = 0;
2071 high = GET_MED(0) - 1;
2072 DEC_MED(0);
2073 } else {
2074 low = GET_MED(0);
2075 INC_MED(0);
2076
2077 if (sample - low < GET_MED(1)) {
2078 ones_count = 1;
2079 high = low + GET_MED(1) - 1;
2080 DEC_MED(1);
2081 } else {
2082 low += GET_MED(1);
2083 INC_MED(1);
2084
2085 if (sample - low < GET_MED(2)) {
2086 ones_count = 2;
2087 high = low + GET_MED(2) - 1;
2088 DEC_MED(2);
2089 } else {
2090 ones_count = 2 + (sample - low) / GET_MED(2);
2091 low += (ones_count - 2) * GET_MED(2);
2092 high = low + GET_MED(2) - 1;
2093 INC_MED(2);
2094 }
2095 }
2096 }
2097
2098 if (w->holding_zero) {
2099 if (ones_count)
2100 w->holding_one++;
2101
2102 encode_flush(s);
2103
2104 if (ones_count) {
2105 w->holding_zero = 1;
2106 ones_count--;
2107 } else
2108 w->holding_zero = 0;
2109 } else
2110 w->holding_zero = 1;
2111
2112 w->holding_one = ones_count * 2;
2113
2114 if (high != low) {
2115 uint32_t maxcode = high - low, code = sample - low;
2116 int bitcount = count_bits(maxcode);
2117 uint32_t extras = (1 << bitcount) - maxcode - 1;
2118
2119 if (code < extras) {
2120 w->pend_data |= code << w->pend_count;
2121 w->pend_count += bitcount - 1;
2122 } else {
2123 w->pend_data |= ((code + extras) >> 1) << w->pend_count;
2124 w->pend_count += bitcount - 1;
2125 w->pend_data |= ((code + extras) & 1) << w->pend_count++;
2126 }
2127 }
2128
2129 w->pend_data |= ((int32_t) sign << w->pend_count++);
2130
2131 if (!w->holding_zero)
2132 encode_flush(s);
2133 }
2134
pack_int32(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int nb_samples)2135 static void pack_int32(WavPackEncodeContext *s,
2136 int32_t *samples_l, int32_t *samples_r,
2137 int nb_samples)
2138 {
2139 const int sent_bits = s->int32_sent_bits;
2140 PutBitContext *pb = &s->pb;
2141 int i, pre_shift;
2142
2143 pre_shift = s->int32_zeros + s->int32_ones + s->int32_dups;
2144
2145 if (!sent_bits)
2146 return;
2147
2148 if (s->flags & WV_MONO_DATA) {
2149 for (i = 0; i < nb_samples; i++) {
2150 put_sbits(pb, sent_bits, samples_l[i] >> pre_shift);
2151 }
2152 } else {
2153 for (i = 0; i < nb_samples; i++) {
2154 put_sbits(pb, sent_bits, samples_l[i] >> pre_shift);
2155 put_sbits(pb, sent_bits, samples_r[i] >> pre_shift);
2156 }
2157 }
2158 }
2159
pack_float_sample(WavPackEncodeContext *s, int32_t *sample)2160 static void pack_float_sample(WavPackEncodeContext *s, int32_t *sample)
2161 {
2162 const int max_exp = s->float_max_exp;
2163 PutBitContext *pb = &s->pb;
2164 int32_t value, shift_count;
2165
2166 if (get_exponent(*sample) == 255) {
2167 if (get_mantissa(*sample)) {
2168 put_bits(pb, 1, 1);
2169 put_bits(pb, 23, get_mantissa(*sample));
2170 } else {
2171 put_bits(pb, 1, 0);
2172 }
2173
2174 value = 0x1000000;
2175 shift_count = 0;
2176 } else if (get_exponent(*sample)) {
2177 shift_count = max_exp - get_exponent(*sample);
2178 value = 0x800000 + get_mantissa(*sample);
2179 } else {
2180 shift_count = max_exp ? max_exp - 1 : 0;
2181 value = get_mantissa(*sample);
2182 }
2183
2184 if (shift_count < 25)
2185 value >>= shift_count;
2186 else
2187 value = 0;
2188
2189 if (!value) {
2190 if (s->float_flags & FLOAT_ZEROS_SENT) {
2191 if (get_exponent(*sample) || get_mantissa(*sample)) {
2192 put_bits(pb, 1, 1);
2193 put_bits(pb, 23, get_mantissa(*sample));
2194
2195 if (max_exp >= 25)
2196 put_bits(pb, 8, get_exponent(*sample));
2197
2198 put_bits(pb, 1, get_sign(*sample));
2199 } else {
2200 put_bits(pb, 1, 0);
2201
2202 if (s->float_flags & FLOAT_NEG_ZEROS)
2203 put_bits(pb, 1, get_sign(*sample));
2204 }
2205 }
2206 } else if (shift_count) {
2207 if (s->float_flags & FLOAT_SHIFT_SENT) {
2208 put_sbits(pb, shift_count, get_mantissa(*sample));
2209 } else if (s->float_flags & FLOAT_SHIFT_SAME) {
2210 put_bits(pb, 1, get_mantissa(*sample) & 1);
2211 }
2212 }
2213 }
2214
pack_float(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, int nb_samples)2215 static void pack_float(WavPackEncodeContext *s,
2216 int32_t *samples_l, int32_t *samples_r,
2217 int nb_samples)
2218 {
2219 int i;
2220
2221 if (s->flags & WV_MONO_DATA) {
2222 for (i = 0; i < nb_samples; i++)
2223 pack_float_sample(s, &samples_l[i]);
2224 } else {
2225 for (i = 0; i < nb_samples; i++) {
2226 pack_float_sample(s, &samples_l[i]);
2227 pack_float_sample(s, &samples_r[i]);
2228 }
2229 }
2230 }
2231
decorr_stereo_pass2(struct Decorr *dpp, int32_t *samples_l, int32_t *samples_r, int nb_samples)2232 static void decorr_stereo_pass2(struct Decorr *dpp,
2233 int32_t *samples_l, int32_t *samples_r,
2234 int nb_samples)
2235 {
2236 int i, m, k;
2237
2238 switch (dpp->value) {
2239 case 17:
2240 for (i = 0; i < nb_samples; i++) {
2241 int32_t sam, tmp;
2242
2243 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
2244 dpp->samplesA[1] = dpp->samplesA[0];
2245 samples_l[i] = tmp = (dpp->samplesA[0] = samples_l[i]) - APPLY_WEIGHT(dpp->weightA, sam);
2246 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
2247
2248 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
2249 dpp->samplesB[1] = dpp->samplesB[0];
2250 samples_r[i] = tmp = (dpp->samplesB[0] = samples_r[i]) - APPLY_WEIGHT(dpp->weightB, sam);
2251 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
2252 }
2253 break;
2254 case 18:
2255 for (i = 0; i < nb_samples; i++) {
2256 int32_t sam, tmp;
2257
2258 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
2259 dpp->samplesA[1] = dpp->samplesA[0];
2260 samples_l[i] = tmp = (dpp->samplesA[0] = samples_l[i]) - APPLY_WEIGHT(dpp->weightA, sam);
2261 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
2262
2263 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
2264 dpp->samplesB[1] = dpp->samplesB[0];
2265 samples_r[i] = tmp = (dpp->samplesB[0] = samples_r[i]) - APPLY_WEIGHT(dpp->weightB, sam);
2266 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
2267 }
2268 break;
2269 default:
2270 for (m = 0, k = dpp->value & (MAX_TERM - 1), i = 0; i < nb_samples; i++) {
2271 int32_t sam, tmp;
2272
2273 sam = dpp->samplesA[m];
2274 samples_l[i] = tmp = (dpp->samplesA[k] = samples_l[i]) - APPLY_WEIGHT(dpp->weightA, sam);
2275 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
2276
2277 sam = dpp->samplesB[m];
2278 samples_r[i] = tmp = (dpp->samplesB[k] = samples_r[i]) - APPLY_WEIGHT(dpp->weightB, sam);
2279 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
2280
2281 m = (m + 1) & (MAX_TERM - 1);
2282 k = (k + 1) & (MAX_TERM - 1);
2283 }
2284 if (m) {
2285 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
2286
2287 memcpy(temp_A, dpp->samplesA, sizeof (dpp->samplesA));
2288 memcpy(temp_B, dpp->samplesB, sizeof (dpp->samplesB));
2289
2290 for (k = 0; k < MAX_TERM; k++) {
2291 dpp->samplesA[k] = temp_A[m];
2292 dpp->samplesB[k] = temp_B[m];
2293 m = (m + 1) & (MAX_TERM - 1);
2294 }
2295 }
2296 break;
2297 case -1:
2298 for (i = 0; i < nb_samples; i++) {
2299 int32_t sam_A, sam_B, tmp;
2300
2301 sam_A = dpp->samplesA[0];
2302 samples_l[i] = tmp = (sam_B = samples_l[i]) - APPLY_WEIGHT(dpp->weightA, sam_A);
2303 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
2304
2305 samples_r[i] = tmp = (dpp->samplesA[0] = samples_r[i]) - APPLY_WEIGHT(dpp->weightB, sam_B);
2306 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
2307 }
2308 break;
2309 case -2:
2310 for (i = 0; i < nb_samples; i++) {
2311 int32_t sam_A, sam_B, tmp;
2312
2313 sam_B = dpp->samplesB[0];
2314 samples_r[i] = tmp = (sam_A = samples_r[i]) - APPLY_WEIGHT(dpp->weightB, sam_B);
2315 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
2316
2317 samples_l[i] = tmp = (dpp->samplesB[0] = samples_l[i]) - APPLY_WEIGHT(dpp->weightA, sam_A);
2318 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
2319 }
2320 break;
2321 case -3:
2322 for (i = 0; i < nb_samples; i++) {
2323 int32_t sam_A, sam_B, tmp;
2324
2325 sam_A = dpp->samplesA[0];
2326 sam_B = dpp->samplesB[0];
2327
2328 dpp->samplesA[0] = tmp = samples_r[i];
2329 samples_r[i] = tmp -= APPLY_WEIGHT(dpp->weightB, sam_B);
2330 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
2331
2332 dpp->samplesB[0] = tmp = samples_l[i];
2333 samples_l[i] = tmp -= APPLY_WEIGHT(dpp->weightA, sam_A);
2334 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
2335 }
2336 break;
2337 }
2338 }
2339
2340 #define update_weight_d2(weight, delta, source, result) \
2341 if (source && result) \
2342 weight -= (((source ^ result) >> 29) & 4) - 2;
2343
2344 #define update_weight_clip_d2(weight, delta, source, result) \
2345 if (source && result) { \
2346 const int32_t s = (source ^ result) >> 31; \
2347 if ((weight = (weight ^ s) + (2 - s)) > 1024) weight = 1024; \
2348 weight = (weight ^ s) - s; \
2349 }
2350
decorr_stereo_pass_id2(struct Decorr *dpp, int32_t *samples_l, int32_t *samples_r, int nb_samples)2351 static void decorr_stereo_pass_id2(struct Decorr *dpp,
2352 int32_t *samples_l, int32_t *samples_r,
2353 int nb_samples)
2354 {
2355 int i, m, k;
2356
2357 switch (dpp->value) {
2358 case 17:
2359 for (i = 0; i < nb_samples; i++) {
2360 int32_t sam, tmp;
2361
2362 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
2363 dpp->samplesA[1] = dpp->samplesA[0];
2364 samples_l[i] = tmp = (dpp->samplesA[0] = samples_l[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
2365 update_weight_d2(dpp->weightA, dpp->delta, sam, tmp);
2366
2367 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
2368 dpp->samplesB[1] = dpp->samplesB[0];
2369 samples_r[i] = tmp = (dpp->samplesB[0] = samples_r[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
2370 update_weight_d2(dpp->weightB, dpp->delta, sam, tmp);
2371 }
2372 break;
2373 case 18:
2374 for (i = 0; i < nb_samples; i++) {
2375 int32_t sam, tmp;
2376
2377 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
2378 dpp->samplesA[1] = dpp->samplesA[0];
2379 samples_l[i] = tmp = (dpp->samplesA[0] = samples_l[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
2380 update_weight_d2(dpp->weightA, dpp->delta, sam, tmp);
2381
2382 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
2383 dpp->samplesB[1] = dpp->samplesB[0];
2384 samples_r[i] = tmp = (dpp->samplesB[0] = samples_r[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
2385 update_weight_d2(dpp->weightB, dpp->delta, sam, tmp);
2386 }
2387 break;
2388 default:
2389 for (m = 0, k = dpp->value & (MAX_TERM - 1), i = 0; i < nb_samples; i++) {
2390 int32_t sam, tmp;
2391
2392 sam = dpp->samplesA[m];
2393 samples_l[i] = tmp = (dpp->samplesA[k] = samples_l[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
2394 update_weight_d2(dpp->weightA, dpp->delta, sam, tmp);
2395
2396 sam = dpp->samplesB[m];
2397 samples_r[i] = tmp = (dpp->samplesB[k] = samples_r[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
2398 update_weight_d2(dpp->weightB, dpp->delta, sam, tmp);
2399
2400 m = (m + 1) & (MAX_TERM - 1);
2401 k = (k + 1) & (MAX_TERM - 1);
2402 }
2403
2404 if (m) {
2405 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
2406
2407 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
2408 memcpy(temp_B, dpp->samplesB, sizeof(dpp->samplesB));
2409
2410 for (k = 0; k < MAX_TERM; k++) {
2411 dpp->samplesA[k] = temp_A[m];
2412 dpp->samplesB[k] = temp_B[m];
2413 m = (m + 1) & (MAX_TERM - 1);
2414 }
2415 }
2416 break;
2417 case -1:
2418 for (i = 0; i < nb_samples; i++) {
2419 int32_t sam_A, sam_B, tmp;
2420
2421 sam_A = dpp->samplesA[0];
2422 samples_l[i] = tmp = (sam_B = samples_l[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
2423 update_weight_clip_d2(dpp->weightA, dpp->delta, sam_A, tmp);
2424
2425 samples_r[i] = tmp = (dpp->samplesA[0] = samples_r[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
2426 update_weight_clip_d2(dpp->weightB, dpp->delta, sam_B, tmp);
2427 }
2428 break;
2429 case -2:
2430 for (i = 0; i < nb_samples; i++) {
2431 int32_t sam_A, sam_B, tmp;
2432
2433 sam_B = dpp->samplesB[0];
2434 samples_r[i] = tmp = (sam_A = samples_r[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
2435 update_weight_clip_d2(dpp->weightB, dpp->delta, sam_B, tmp);
2436
2437 samples_l[i] = tmp = (dpp->samplesB[0] = samples_l[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
2438 update_weight_clip_d2(dpp->weightA, dpp->delta, sam_A, tmp);
2439 }
2440 break;
2441 case -3:
2442 for (i = 0; i < nb_samples; i++) {
2443 int32_t sam_A, sam_B, tmp;
2444
2445 sam_A = dpp->samplesA[0];
2446 sam_B = dpp->samplesB[0];
2447
2448 dpp->samplesA[0] = tmp = samples_r[i];
2449 samples_r[i] = tmp -= APPLY_WEIGHT_I(dpp->weightB, sam_B);
2450 update_weight_clip_d2(dpp->weightB, dpp->delta, sam_B, tmp);
2451
2452 dpp->samplesB[0] = tmp = samples_l[i];
2453 samples_l[i] = tmp -= APPLY_WEIGHT_I(dpp->weightA, sam_A);
2454 update_weight_clip_d2(dpp->weightA, dpp->delta, sam_A, tmp);
2455 }
2456 break;
2457 }
2458 }
2459
put_metadata_block(PutByteContext *pb, int flags, int size)2460 static void put_metadata_block(PutByteContext *pb, int flags, int size)
2461 {
2462 if (size & 1)
2463 flags |= WP_IDF_ODD;
2464
2465 bytestream2_put_byte(pb, flags);
2466 bytestream2_put_byte(pb, (size + 1) >> 1);
2467 }
2468
wavpack_encode_block(WavPackEncodeContext *s, int32_t *samples_l, int32_t *samples_r, uint8_t *out, int out_size)2469 static int wavpack_encode_block(WavPackEncodeContext *s,
2470 int32_t *samples_l, int32_t *samples_r,
2471 uint8_t *out, int out_size)
2472 {
2473 int block_size, start, end, data_size, tcount, temp, m = 0;
2474 int i, j, ret = 0, got_extra = 0, nb_samples = s->block_samples;
2475 uint32_t crc = 0xffffffffu;
2476 struct Decorr *dpp;
2477 PutByteContext pb;
2478
2479 if (s->flags & WV_MONO_DATA) {
2480 CLEAR(s->w);
2481 }
2482 if (!(s->flags & WV_MONO) && s->optimize_mono) {
2483 int32_t lor = 0, diff = 0;
2484
2485 for (i = 0; i < nb_samples; i++) {
2486 lor |= samples_l[i] | samples_r[i];
2487 diff |= samples_l[i] - samples_r[i];
2488
2489 if (lor && diff)
2490 break;
2491 }
2492
2493 if (i == nb_samples && lor && !diff) {
2494 s->flags &= ~(WV_JOINT_STEREO | WV_CROSS_DECORR);
2495 s->flags |= WV_FALSE_STEREO;
2496
2497 if (!s->false_stereo) {
2498 s->false_stereo = 1;
2499 s->num_terms = 0;
2500 CLEAR(s->w);
2501 }
2502 } else if (s->false_stereo) {
2503 s->false_stereo = 0;
2504 s->num_terms = 0;
2505 CLEAR(s->w);
2506 }
2507 }
2508
2509 if (s->flags & SHIFT_MASK) {
2510 int shift = (s->flags & SHIFT_MASK) >> SHIFT_LSB;
2511 int mag = (s->flags & MAG_MASK) >> MAG_LSB;
2512
2513 if (s->flags & WV_MONO_DATA)
2514 shift_mono(samples_l, nb_samples, shift);
2515 else
2516 shift_stereo(samples_l, samples_r, nb_samples, shift);
2517
2518 if ((mag -= shift) < 0)
2519 s->flags &= ~MAG_MASK;
2520 else
2521 s->flags -= (1 << MAG_LSB) * shift;
2522 }
2523
2524 if ((s->flags & WV_FLOAT_DATA) || (s->flags & MAG_MASK) >> MAG_LSB >= 24) {
2525 av_fast_padded_malloc(&s->orig_l, &s->orig_l_size, sizeof(int32_t) * nb_samples);
2526 memcpy(s->orig_l, samples_l, sizeof(int32_t) * nb_samples);
2527 if (!(s->flags & WV_MONO_DATA)) {
2528 av_fast_padded_malloc(&s->orig_r, &s->orig_r_size, sizeof(int32_t) * nb_samples);
2529 memcpy(s->orig_r, samples_r, sizeof(int32_t) * nb_samples);
2530 }
2531
2532 if (s->flags & WV_FLOAT_DATA)
2533 got_extra = scan_float(s, samples_l, samples_r, nb_samples);
2534 else
2535 got_extra = scan_int32(s, samples_l, samples_r, nb_samples);
2536 s->num_terms = 0;
2537 } else {
2538 scan_int23(s, samples_l, samples_r, nb_samples);
2539 if (s->shift != s->int32_zeros + s->int32_ones + s->int32_dups) {
2540 s->shift = s->int32_zeros + s->int32_ones + s->int32_dups;
2541 s->num_terms = 0;
2542 }
2543 }
2544
2545 if (!s->num_passes && !s->num_terms) {
2546 s->num_passes = 1;
2547
2548 if (s->flags & WV_MONO_DATA)
2549 ret = wv_mono(s, samples_l, 1, 0);
2550 else
2551 ret = wv_stereo(s, samples_l, samples_r, 1, 0);
2552
2553 s->num_passes = 0;
2554 }
2555 if (s->flags & WV_MONO_DATA) {
2556 for (i = 0; i < nb_samples; i++)
2557 crc += (crc << 1) + samples_l[i];
2558
2559 if (s->num_passes)
2560 ret = wv_mono(s, samples_l, !s->num_terms, 1);
2561 } else {
2562 for (i = 0; i < nb_samples; i++)
2563 crc += (crc << 3) + ((uint32_t)samples_l[i] << 1) + samples_l[i] + samples_r[i];
2564
2565 if (s->num_passes)
2566 ret = wv_stereo(s, samples_l, samples_r, !s->num_terms, 1);
2567 }
2568 if (ret < 0)
2569 return ret;
2570
2571 if (!s->ch_offset)
2572 s->flags |= WV_INITIAL_BLOCK;
2573
2574 s->ch_offset += 1 + !(s->flags & WV_MONO);
2575
2576 if (s->ch_offset == s->avctx->ch_layout.nb_channels)
2577 s->flags |= WV_FINAL_BLOCK;
2578
2579 bytestream2_init_writer(&pb, out, out_size);
2580 bytestream2_put_le32(&pb, MKTAG('w', 'v', 'p', 'k'));
2581 bytestream2_put_le32(&pb, 0);
2582 bytestream2_put_le16(&pb, 0x410);
2583 bytestream2_put_le16(&pb, 0);
2584 bytestream2_put_le32(&pb, 0);
2585 bytestream2_put_le32(&pb, s->sample_index);
2586 bytestream2_put_le32(&pb, nb_samples);
2587 bytestream2_put_le32(&pb, s->flags);
2588 bytestream2_put_le32(&pb, crc);
2589
2590 if (s->flags & WV_INITIAL_BLOCK &&
2591 s->avctx->ch_layout.order == AV_CHANNEL_ORDER_NATIVE &&
2592 s->avctx->ch_layout.u.mask != AV_CH_LAYOUT_MONO &&
2593 s->avctx->ch_layout.u.mask != AV_CH_LAYOUT_STEREO) {
2594 put_metadata_block(&pb, WP_ID_CHANINFO, 5);
2595 bytestream2_put_byte(&pb, s->avctx->ch_layout.nb_channels);
2596 bytestream2_put_le32(&pb, s->avctx->ch_layout.u.mask);
2597 bytestream2_put_byte(&pb, 0);
2598 }
2599
2600 if ((s->flags & SRATE_MASK) == SRATE_MASK) {
2601 put_metadata_block(&pb, WP_ID_SAMPLE_RATE, 3);
2602 bytestream2_put_le24(&pb, s->avctx->sample_rate);
2603 bytestream2_put_byte(&pb, 0);
2604 }
2605
2606 put_metadata_block(&pb, WP_ID_DECTERMS, s->num_terms);
2607 for (i = 0; i < s->num_terms; i++) {
2608 struct Decorr *dpp = &s->decorr_passes[i];
2609 bytestream2_put_byte(&pb, ((dpp->value + 5) & 0x1f) | ((dpp->delta << 5) & 0xe0));
2610 }
2611 if (s->num_terms & 1)
2612 bytestream2_put_byte(&pb, 0);
2613
2614 #define WRITE_DECWEIGHT(type) do { \
2615 temp = store_weight(type); \
2616 bytestream2_put_byte(&pb, temp); \
2617 type = restore_weight(temp); \
2618 } while (0)
2619
2620 bytestream2_put_byte(&pb, WP_ID_DECWEIGHTS);
2621 bytestream2_put_byte(&pb, 0);
2622 start = bytestream2_tell_p(&pb);
2623 for (i = s->num_terms - 1; i >= 0; --i) {
2624 struct Decorr *dpp = &s->decorr_passes[i];
2625
2626 if (store_weight(dpp->weightA) ||
2627 (!(s->flags & WV_MONO_DATA) && store_weight(dpp->weightB)))
2628 break;
2629 }
2630 tcount = i + 1;
2631 for (i = 0; i < s->num_terms; i++) {
2632 struct Decorr *dpp = &s->decorr_passes[i];
2633 if (i < tcount) {
2634 WRITE_DECWEIGHT(dpp->weightA);
2635 if (!(s->flags & WV_MONO_DATA))
2636 WRITE_DECWEIGHT(dpp->weightB);
2637 } else {
2638 dpp->weightA = dpp->weightB = 0;
2639 }
2640 }
2641 end = bytestream2_tell_p(&pb);
2642 out[start - 2] = WP_ID_DECWEIGHTS | (((end - start) & 1) ? WP_IDF_ODD: 0);
2643 out[start - 1] = (end - start + 1) >> 1;
2644 if ((end - start) & 1)
2645 bytestream2_put_byte(&pb, 0);
2646
2647 #define WRITE_DECSAMPLE(type) do { \
2648 temp = log2s(type); \
2649 type = wp_exp2(temp); \
2650 bytestream2_put_le16(&pb, temp); \
2651 } while (0)
2652
2653 bytestream2_put_byte(&pb, WP_ID_DECSAMPLES);
2654 bytestream2_put_byte(&pb, 0);
2655 start = bytestream2_tell_p(&pb);
2656 for (i = 0; i < s->num_terms; i++) {
2657 struct Decorr *dpp = &s->decorr_passes[i];
2658 if (i == 0) {
2659 if (dpp->value > MAX_TERM) {
2660 WRITE_DECSAMPLE(dpp->samplesA[0]);
2661 WRITE_DECSAMPLE(dpp->samplesA[1]);
2662 if (!(s->flags & WV_MONO_DATA)) {
2663 WRITE_DECSAMPLE(dpp->samplesB[0]);
2664 WRITE_DECSAMPLE(dpp->samplesB[1]);
2665 }
2666 } else if (dpp->value < 0) {
2667 WRITE_DECSAMPLE(dpp->samplesA[0]);
2668 WRITE_DECSAMPLE(dpp->samplesB[0]);
2669 } else {
2670 for (j = 0; j < dpp->value; j++) {
2671 WRITE_DECSAMPLE(dpp->samplesA[j]);
2672 if (!(s->flags & WV_MONO_DATA))
2673 WRITE_DECSAMPLE(dpp->samplesB[j]);
2674 }
2675 }
2676 } else {
2677 CLEAR(dpp->samplesA);
2678 CLEAR(dpp->samplesB);
2679 }
2680 }
2681 end = bytestream2_tell_p(&pb);
2682 out[start - 1] = (end - start) >> 1;
2683
2684 #define WRITE_CHAN_ENTROPY(chan) do { \
2685 for (i = 0; i < 3; i++) { \
2686 temp = wp_log2(s->w.c[chan].median[i]); \
2687 bytestream2_put_le16(&pb, temp); \
2688 s->w.c[chan].median[i] = wp_exp2(temp); \
2689 } \
2690 } while (0)
2691
2692 put_metadata_block(&pb, WP_ID_ENTROPY, 6 * (1 + (!(s->flags & WV_MONO_DATA))));
2693 WRITE_CHAN_ENTROPY(0);
2694 if (!(s->flags & WV_MONO_DATA))
2695 WRITE_CHAN_ENTROPY(1);
2696
2697 if (s->flags & WV_FLOAT_DATA) {
2698 put_metadata_block(&pb, WP_ID_FLOATINFO, 4);
2699 bytestream2_put_byte(&pb, s->float_flags);
2700 bytestream2_put_byte(&pb, s->float_shift);
2701 bytestream2_put_byte(&pb, s->float_max_exp);
2702 bytestream2_put_byte(&pb, 127);
2703 }
2704
2705 if (s->flags & WV_INT32_DATA) {
2706 put_metadata_block(&pb, WP_ID_INT32INFO, 4);
2707 bytestream2_put_byte(&pb, s->int32_sent_bits);
2708 bytestream2_put_byte(&pb, s->int32_zeros);
2709 bytestream2_put_byte(&pb, s->int32_ones);
2710 bytestream2_put_byte(&pb, s->int32_dups);
2711 }
2712
2713 if (s->flags & WV_MONO_DATA && !s->num_passes) {
2714 for (i = 0; i < nb_samples; i++) {
2715 int32_t code = samples_l[i];
2716
2717 for (tcount = s->num_terms, dpp = s->decorr_passes; tcount--; dpp++) {
2718 int32_t sam;
2719
2720 if (dpp->value > MAX_TERM) {
2721 if (dpp->value & 1)
2722 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
2723 else
2724 sam = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
2725
2726 dpp->samplesA[1] = dpp->samplesA[0];
2727 dpp->samplesA[0] = code;
2728 } else {
2729 sam = dpp->samplesA[m];
2730 dpp->samplesA[(m + dpp->value) & (MAX_TERM - 1)] = code;
2731 }
2732
2733 code -= APPLY_WEIGHT(dpp->weightA, sam);
2734 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, code);
2735 }
2736
2737 m = (m + 1) & (MAX_TERM - 1);
2738 samples_l[i] = code;
2739 }
2740 if (m) {
2741 for (tcount = s->num_terms, dpp = s->decorr_passes; tcount--; dpp++)
2742 if (dpp->value > 0 && dpp->value <= MAX_TERM) {
2743 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
2744 int k;
2745
2746 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
2747 memcpy(temp_B, dpp->samplesB, sizeof(dpp->samplesB));
2748
2749 for (k = 0; k < MAX_TERM; k++) {
2750 dpp->samplesA[k] = temp_A[m];
2751 dpp->samplesB[k] = temp_B[m];
2752 m = (m + 1) & (MAX_TERM - 1);
2753 }
2754 }
2755 }
2756 } else if (!s->num_passes) {
2757 if (s->flags & WV_JOINT_STEREO) {
2758 for (i = 0; i < nb_samples; i++)
2759 samples_r[i] += ((samples_l[i] -= samples_r[i]) >> 1);
2760 }
2761
2762 for (i = 0; i < s->num_terms; i++) {
2763 struct Decorr *dpp = &s->decorr_passes[i];
2764 if (((s->flags & MAG_MASK) >> MAG_LSB) >= 16 || dpp->delta != 2)
2765 decorr_stereo_pass2(dpp, samples_l, samples_r, nb_samples);
2766 else
2767 decorr_stereo_pass_id2(dpp, samples_l, samples_r, nb_samples);
2768 }
2769 }
2770
2771 bytestream2_put_byte(&pb, WP_ID_DATA | WP_IDF_LONG);
2772 init_put_bits(&s->pb, pb.buffer + 3, bytestream2_get_bytes_left_p(&pb));
2773 if (s->flags & WV_MONO_DATA) {
2774 for (i = 0; i < nb_samples; i++)
2775 wavpack_encode_sample(s, &s->w.c[0], s->samples[0][i]);
2776 } else {
2777 for (i = 0; i < nb_samples; i++) {
2778 wavpack_encode_sample(s, &s->w.c[0], s->samples[0][i]);
2779 wavpack_encode_sample(s, &s->w.c[1], s->samples[1][i]);
2780 }
2781 }
2782 encode_flush(s);
2783 flush_put_bits(&s->pb);
2784 data_size = put_bytes_output(&s->pb);
2785 bytestream2_put_le24(&pb, (data_size + 1) >> 1);
2786 bytestream2_skip_p(&pb, data_size);
2787 if (data_size & 1)
2788 bytestream2_put_byte(&pb, 0);
2789
2790 if (got_extra) {
2791 bytestream2_put_byte(&pb, WP_ID_EXTRABITS | WP_IDF_LONG);
2792 init_put_bits(&s->pb, pb.buffer + 7, bytestream2_get_bytes_left_p(&pb));
2793 if (s->flags & WV_FLOAT_DATA)
2794 pack_float(s, s->orig_l, s->orig_r, nb_samples);
2795 else
2796 pack_int32(s, s->orig_l, s->orig_r, nb_samples);
2797 flush_put_bits(&s->pb);
2798 data_size = put_bytes_output(&s->pb);
2799 bytestream2_put_le24(&pb, (data_size + 5) >> 1);
2800 bytestream2_put_le32(&pb, s->crc_x);
2801 bytestream2_skip_p(&pb, data_size);
2802 if (data_size & 1)
2803 bytestream2_put_byte(&pb, 0);
2804 }
2805
2806 block_size = bytestream2_tell_p(&pb);
2807 AV_WL32(out + 4, block_size - 8);
2808
2809 av_assert0(!bytestream2_get_eof(&pb));
2810
2811 return block_size;
2812 }
2813
fill_buffer(WavPackEncodeContext *s, const int8_t *src, int32_t *dst, int nb_samples)2814 static void fill_buffer(WavPackEncodeContext *s,
2815 const int8_t *src, int32_t *dst,
2816 int nb_samples)
2817 {
2818 int i;
2819
2820 #define COPY_SAMPLES(type, offset, shift) do { \
2821 const type *sptr = (const type *)src; \
2822 for (i = 0; i < nb_samples; i++) \
2823 dst[i] = (sptr[i] - offset) >> shift; \
2824 } while (0)
2825
2826 switch (s->avctx->sample_fmt) {
2827 case AV_SAMPLE_FMT_U8P:
2828 COPY_SAMPLES(int8_t, 0x80, 0);
2829 break;
2830 case AV_SAMPLE_FMT_S16P:
2831 COPY_SAMPLES(int16_t, 0, 0);
2832 break;
2833 case AV_SAMPLE_FMT_S32P:
2834 if (s->avctx->bits_per_raw_sample <= 24) {
2835 COPY_SAMPLES(int32_t, 0, 8);
2836 break;
2837 }
2838 case AV_SAMPLE_FMT_FLTP:
2839 memcpy(dst, src, nb_samples * 4);
2840 }
2841 }
2842
set_samplerate(WavPackEncodeContext *s)2843 static void set_samplerate(WavPackEncodeContext *s)
2844 {
2845 int i;
2846
2847 for (i = 0; i < 15; i++) {
2848 if (wv_rates[i] == s->avctx->sample_rate)
2849 break;
2850 }
2851
2852 s->flags = i << SRATE_LSB;
2853 }
2854
wavpack_encode_frame(AVCodecContext *avctx, AVPacket *avpkt, const AVFrame *frame, int *got_packet_ptr)2855 static int wavpack_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
2856 const AVFrame *frame, int *got_packet_ptr)
2857 {
2858 WavPackEncodeContext *s = avctx->priv_data;
2859 int buf_size, ret;
2860 uint8_t *buf;
2861
2862 s->block_samples = frame->nb_samples;
2863 av_fast_padded_malloc(&s->samples[0], &s->samples_size[0],
2864 sizeof(int32_t) * s->block_samples);
2865 if (!s->samples[0])
2866 return AVERROR(ENOMEM);
2867 if (avctx->ch_layout.nb_channels > 1) {
2868 av_fast_padded_malloc(&s->samples[1], &s->samples_size[1],
2869 sizeof(int32_t) * s->block_samples);
2870 if (!s->samples[1])
2871 return AVERROR(ENOMEM);
2872 }
2873
2874 buf_size = s->block_samples * avctx->ch_layout.nb_channels * 8
2875 + 200 * avctx->ch_layout.nb_channels /* for headers */;
2876 if ((ret = ff_alloc_packet(avctx, avpkt, buf_size)) < 0)
2877 return ret;
2878 buf = avpkt->data;
2879
2880 for (s->ch_offset = 0; s->ch_offset < avctx->ch_layout.nb_channels;) {
2881 set_samplerate(s);
2882
2883 switch (s->avctx->sample_fmt) {
2884 case AV_SAMPLE_FMT_S16P: s->flags |= 1; break;
2885 case AV_SAMPLE_FMT_S32P: s->flags |= 3 - (s->avctx->bits_per_raw_sample <= 24); break;
2886 case AV_SAMPLE_FMT_FLTP: s->flags |= 3 | WV_FLOAT_DATA;
2887 }
2888
2889 fill_buffer(s, frame->extended_data[s->ch_offset], s->samples[0], s->block_samples);
2890 if (avctx->ch_layout.nb_channels - s->ch_offset == 1) {
2891 s->flags |= WV_MONO;
2892 } else {
2893 s->flags |= WV_CROSS_DECORR;
2894 fill_buffer(s, frame->extended_data[s->ch_offset + 1], s->samples[1], s->block_samples);
2895 }
2896
2897 s->flags += (1 << MAG_LSB) * ((s->flags & 3) * 8 + 7);
2898
2899 if ((ret = wavpack_encode_block(s, s->samples[0], s->samples[1],
2900 buf, buf_size)) < 0)
2901 return ret;
2902
2903 buf += ret;
2904 buf_size -= ret;
2905 }
2906 s->sample_index += frame->nb_samples;
2907
2908 avpkt->pts = frame->pts;
2909 avpkt->size = buf - avpkt->data;
2910 avpkt->duration = ff_samples_to_time_base(avctx, frame->nb_samples);
2911 *got_packet_ptr = 1;
2912 return 0;
2913 }
2914
wavpack_encode_close(AVCodecContext *avctx)2915 static av_cold int wavpack_encode_close(AVCodecContext *avctx)
2916 {
2917 WavPackEncodeContext *s = avctx->priv_data;
2918 int i;
2919
2920 for (i = 0; i < MAX_TERMS + 2; i++) {
2921 av_freep(&s->sampleptrs[i][0]);
2922 av_freep(&s->sampleptrs[i][1]);
2923 s->sampleptrs_size[i][0] = s->sampleptrs_size[i][1] = 0;
2924 }
2925
2926 for (i = 0; i < 2; i++) {
2927 av_freep(&s->samples[i]);
2928 s->samples_size[i] = 0;
2929
2930 av_freep(&s->best_buffer[i]);
2931 s->best_buffer_size[i] = 0;
2932
2933 av_freep(&s->temp_buffer[i][0]);
2934 av_freep(&s->temp_buffer[i][1]);
2935 s->temp_buffer_size[i][0] = s->temp_buffer_size[i][1] = 0;
2936 }
2937
2938 av_freep(&s->js_left);
2939 av_freep(&s->js_right);
2940 s->js_left_size = s->js_right_size = 0;
2941
2942 av_freep(&s->orig_l);
2943 av_freep(&s->orig_r);
2944 s->orig_l_size = s->orig_r_size = 0;
2945
2946 return 0;
2947 }
2948
2949 #define OFFSET(x) offsetof(WavPackEncodeContext, x)
2950 #define FLAGS AV_OPT_FLAG_ENCODING_PARAM | AV_OPT_FLAG_AUDIO_PARAM
2951 static const AVOption options[] = {
2952 { "joint_stereo", "", OFFSET(joint), AV_OPT_TYPE_BOOL, {.i64=-1}, -1, 1, FLAGS },
2953 { "optimize_mono", "", OFFSET(optimize_mono), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS },
2954 { NULL },
2955 };
2956
2957 static const AVClass wavpack_encoder_class = {
2958 .class_name = "WavPack encoder",
2959 .item_name = av_default_item_name,
2960 .option = options,
2961 .version = LIBAVUTIL_VERSION_INT,
2962 };
2963
2964 const FFCodec ff_wavpack_encoder = {
2965 .p.name = "wavpack",
2966 .p.long_name = NULL_IF_CONFIG_SMALL("WavPack"),
2967 .p.type = AVMEDIA_TYPE_AUDIO,
2968 .p.id = AV_CODEC_ID_WAVPACK,
2969 .priv_data_size = sizeof(WavPackEncodeContext),
2970 .p.priv_class = &wavpack_encoder_class,
2971 .init = wavpack_encode_init,
2972 FF_CODEC_ENCODE_CB(wavpack_encode_frame),
2973 .close = wavpack_encode_close,
2974 .p.capabilities = AV_CODEC_CAP_SMALL_LAST_FRAME,
2975 .p.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_U8P,
2976 AV_SAMPLE_FMT_S16P,
2977 AV_SAMPLE_FMT_S32P,
2978 AV_SAMPLE_FMT_FLTP,
2979 AV_SAMPLE_FMT_NONE },
2980 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
2981 };
2982