FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/wavpackenc.c
Date: 2022-07-07 01:21:54
Exec Total Coverage
Lines: 792 1747 45.3%
Branches: 552 1443 38.3%

Line Branch Exec Source
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
128 2 static av_cold int wavpack_encode_init(AVCodecContext *avctx)
129 {
130 2 WavPackEncodeContext *s = avctx->priv_data;
131
132 2 s->avctx = avctx;
133
134
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 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
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
2 if (!avctx->frame_size) {
140 int block_samples;
141
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
2 if (!(avctx->sample_rate & 1))
142 2 block_samples = avctx->sample_rate / 2;
143 else
144 block_samples = avctx->sample_rate;
145
146
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 while (block_samples * avctx->ch_layout.nb_channels > WV_MAX_SAMPLES)
147 block_samples /= 2;
148
149
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
3 while (block_samples * avctx->ch_layout.nb_channels < 40000)
150 1 block_samples *= 2;
151 2 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
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
2 if (avctx->compression_level != FF_COMPRESSION_DEFAULT) {
159
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 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
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 } else if (avctx->compression_level == 2) {
179 s->decorr_filter = 2;
180 s->num_passes = 4;
181
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 } else if (avctx->compression_level == 1) {
182 1 s->decorr_filter = 1;
183 1 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 2 s->num_decorrs = decorr_filter_sizes[s->decorr_filter];
191 2 s->decorr_specs = decorr_filters[s->decorr_filter];
192
193 2 s->delta_decay = 2.0;
194
195 2 return 0;
196 }
197
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
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
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
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
353 13 static void scan_int23(WavPackEncodeContext *s,
354 int32_t *samples_l, int32_t *samples_r,
355 int nb_samples)
356 {
357 13 uint32_t magdata = 0, ordata = 0, xordata = 0, anddata = ~0;
358 13 int i, total_shift = 0;
359
360 13 s->int32_sent_bits = s->int32_zeros = s->int32_ones = s->int32_dups = 0;
361
362
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (s->flags & WV_MONO_DATA) {
363
1/2
✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
9 for (i = 0; i < nb_samples; i++) {
364 9 int32_t M = samples_l[i];
365
366
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
9 magdata |= (M < 0) ? ~M : M;
367 9 xordata |= M ^ -(M & 1);
368 9 anddata &= M;
369 9 ordata |= M;
370
371
4/6
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
9 if ((ordata & 1) && !(anddata & 1) && (xordata & 2))
372 1 return;
373 }
374 } else {
375
1/2
✓ Branch 0 taken 41 times.
✗ Branch 1 not taken.
41 for (i = 0; i < nb_samples; i++) {
376 41 int32_t L = samples_l[i];
377 41 int32_t R = samples_r[i];
378
379
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 37 times.
41 magdata |= (L < 0) ? ~L : L;
380
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 37 times.
41 magdata |= (R < 0) ? ~R : R;
381 41 xordata |= L ^ -(L & 1);
382 41 xordata |= R ^ -(R & 1);
383 41 anddata &= L & R;
384 41 ordata |= L | R;
385
386
6/6
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 10 times.
✓ Branch 4 taken 12 times.
✓ Branch 5 taken 3 times.
41 if ((ordata & 1) && !(anddata & 1) && (xordata & 2))
387 12 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
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
524 624 static int8_t store_weight(int weight)
525 {
526 624 weight = av_clip(weight, -1024, 1024);
527
2/2
✓ Branch 0 taken 267 times.
✓ Branch 1 taken 357 times.
624 if (weight > 0)
528 267 weight -= (weight + 64) >> 7;
529
530 624 return (weight + 4) >> 3;
531 }
532
533 606 static int restore_weight(int8_t weight)
534 {
535 606 int result = 8 * weight;
536
537
2/2
✓ Branch 0 taken 252 times.
✓ Branch 1 taken 354 times.
606 if (result > 0)
538 252 result += (result + 64) >> 7;
539
540 606 return result;
541 }
542
543 3922 static int log2s(int32_t value)
544 {
545
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 3907 times.
3922 return (value < 0) ? -wp_log2(-value) : wp_log2(value);
546 }
547
548 4 static void decorr_mono(int32_t *in_samples, int32_t *out_samples,
549 int nb_samples, struct Decorr *dpp, int dir)
550 {
551 4 int m = 0, i;
552
553 4 dpp->sumA = 0;
554
555
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
4 if (dir < 0) {
556 2 out_samples += (nb_samples - 1);
557 2 in_samples += (nb_samples - 1);
558 }
559
560 4 dpp->weightA = restore_weight(store_weight(dpp->weightA));
561
562
2/2
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 4 times.
36 for (i = 0; i < MAX_TERM; i++)
563 32 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
564
565
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
4 if (dpp->value > MAX_TERM) {
566
2/2
✓ Branch 0 taken 92296 times.
✓ Branch 1 taken 4 times.
92300 while (nb_samples--) {
567 int32_t left, sam_A;
568
569 92296 sam_A = ((3 - (dpp->value & 1)) * dpp->samplesA[0] - dpp->samplesA[1]) >> !(dpp->value & 1);
570
571 92296 dpp->samplesA[1] = dpp->samplesA[0];
572 92296 dpp->samplesA[0] = left = in_samples[0];
573
574
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 92296 times.
92296 left -= APPLY_WEIGHT(dpp->weightA, sam_A);
575
4/4
✓ Branch 0 taken 92286 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 92236 times.
✓ Branch 3 taken 50 times.
92296 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam_A, left);
576 92296 dpp->sumA += dpp->weightA;
577 92296 out_samples[0] = left;
578 92296 in_samples += dir;
579 92296 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
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
4 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 4 }
610
611 1 static void reverse_mono_decorr(struct Decorr *dpp)
612 {
613
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (dpp->value > MAX_TERM) {
614 int32_t sam_A;
615
616
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (dpp->value & 1)
617 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
618 else
619 1 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
620
621 1 dpp->samplesA[1] = dpp->samplesA[0];
622 1 dpp->samplesA[0] = sam_A;
623
624
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (dpp->value & 1)
625 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
626 else
627 1 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
628
629 1 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 1 }
642
643 #define count_bits(av) ((av) ? 32 - ff_clz(av) : 0)
644
645 1102500 static uint32_t log2sample(uint32_t v, int limit, uint32_t *result)
646 {
647
2/2
✓ Branch 0 taken 773442 times.
✓ Branch 1 taken 329058 times.
1102500 uint32_t dbits = count_bits(v);
648
649
2/2
✓ Branch 0 taken 959899 times.
✓ Branch 1 taken 142601 times.
1102500 if ((v += v >> 9) < (1 << 8)) {
650 959899 *result += (dbits << 8) + ff_wp_log2_table[(v << (9 - dbits)) & 0xff];
651 } else {
652 142601 *result += dbits = (dbits << 8) + ff_wp_log2_table[(v >> (dbits - 9)) & 0xff];
653
654
2/4
✓ Branch 0 taken 142601 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 142601 times.
142601 if (limit && dbits >= limit)
655 return 1;
656 }
657
658 1102500 return 0;
659 }
660
661 1 static uint32_t log2mono(int32_t *samples, int nb_samples, int limit)
662 {
663 1 uint32_t result = 0;
664
2/2
✓ Branch 0 taken 44100 times.
✓ Branch 1 taken 1 times.
44101 while (nb_samples--) {
665
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 44100 times.
44100 if (log2sample(abs(*samples++), limit, &result))
666 return UINT32_MAX;
667 }
668 1 return result;
669 }
670
671 24 static uint32_t log2stereo(int32_t *samples_l, int32_t *samples_r,
672 int nb_samples, int limit)
673 {
674 24 uint32_t result = 0;
675
2/2
✓ Branch 0 taken 529200 times.
✓ Branch 1 taken 24 times.
529224 while (nb_samples--) {
676
2/4
✓ Branch 1 taken 529200 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 529200 times.
1058400 if (log2sample(abs(*samples_l++), limit, &result) ||
677 529200 log2sample(abs(*samples_r++), limit, &result))
678 return UINT32_MAX;
679 }
680 24 return result;
681 }
682
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
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
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
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
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
905 13 static int allocate_buffers(WavPackEncodeContext *s)
906 {
907 int i;
908
909
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 13 times.
39 for (i = 0; i < 2; i++) {
910 26 av_fast_padded_malloc(&s->best_buffer[0], &s->best_buffer_size[0],
911 26 s->block_samples * 4);
912
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26 times.
26 if (!s->best_buffer[0])
913 return AVERROR(ENOMEM);
914
915 26 av_fast_padded_malloc(&s->temp_buffer[i][0], &s->temp_buffer_size[i][0],
916 26 s->block_samples * 4);
917
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26 times.
26 if (!s->temp_buffer[i][0])
918 return AVERROR(ENOMEM);
919
2/2
✓ Branch 0 taken 24 times.
✓ Branch 1 taken 2 times.
26 if (!(s->flags & WV_MONO_DATA)) {
920 24 av_fast_padded_malloc(&s->best_buffer[1], &s->best_buffer_size[1],
921 24 s->block_samples * 4);
922
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
24 if (!s->best_buffer[1])
923 return AVERROR(ENOMEM);
924
925 24 av_fast_padded_malloc(&s->temp_buffer[i][1], &s->temp_buffer_size[i][1],
926 24 s->block_samples * 4);
927
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
24 if (!s->temp_buffer[i][1])
928 return AVERROR(ENOMEM);
929 }
930 }
931
932 13 return 0;
933 }
934
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
987 3 static void scan_word(WavPackEncodeContext *s, WvChannel *c,
988 int32_t *samples, int nb_samples, int dir)
989 {
990
1/2
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
3 if (dir < 0)
991 3 samples += nb_samples - 1;
992
993
2/2
✓ Branch 0 taken 88200 times.
✓ Branch 1 taken 3 times.
88203 while (nb_samples--) {
994 88200 uint32_t low, value = labs(samples[0]);
995
996
2/2
✓ Branch 0 taken 68946 times.
✓ Branch 1 taken 19254 times.
88200 if (value < GET_MED(0)) {
997 68946 DEC_MED(0);
998 } else {
999 19254 low = GET_MED(0);
1000 19254 INC_MED(0);
1001
1002
2/2
✓ Branch 0 taken 13482 times.
✓ Branch 1 taken 5772 times.
19254 if (value - low < GET_MED(1)) {
1003 13482 DEC_MED(1);
1004 } else {
1005 5772 low += GET_MED(1);
1006 5772 INC_MED(1);
1007
1008
2/2
✓ Branch 0 taken 3975 times.
✓ Branch 1 taken 1797 times.
5772 if (value - low < GET_MED(2)) {
1009 3975 DEC_MED(2);
1010 } else {
1011 1797 INC_MED(2);
1012 }
1013 }
1014 }
1015 88200 samples += dir;
1016 }
1017 3 }
1018
1019 1 static int wv_mono(WavPackEncodeContext *s, int32_t *samples,
1020 int no_history, int do_samples)
1021 {
1022 1 struct Decorr temp_decorr_pass, save_decorr_passes[MAX_TERMS] = {{0}};
1023 1 int nb_samples = s->block_samples;
1024 1 int buf_size = sizeof(int32_t) * nb_samples;
1025 1 uint32_t best_size = UINT32_MAX, size;
1026 int log_limit, pi, i, ret;
1027
1028
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 for (i = 0; i < nb_samples; i++)
1029
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (samples[i])
1030 1 break;
1031
1032
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (i == nb_samples) {
1033 CLEAR(s->decorr_passes);
1034 CLEAR(s->w);
1035 s->num_terms = 0;
1036 return 0;
1037 }
1038
1039 1 log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
1040 1 log_limit = FFMIN(6912, log_limit);
1041
1042
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 if ((ret = allocate_buffers(s)) < 0)
1043 return ret;
1044
1045
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1 if (no_history || s->num_passes >= 7)
1046 1 s->best_decorr = s->mask_decorr = 0;
1047
1048
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
2 for (pi = 0; pi < s->num_passes;) {
1049 const WavPackDecorrSpec *wpds;
1050 int nterms, c, j;
1051
1052
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (!pi) {
1053 1 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 1 wpds = &s->decorr_specs[c];
1067 1 nterms = decorr_filter_nterms[s->decorr_filter];
1068
1069 while (1) {
1070 1 memcpy(s->temp_buffer[0][0], samples, buf_size);
1071 1 CLEAR(save_decorr_passes);
1072
1073
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
3 for (j = 0; j < nterms; j++) {
1074 2 CLEAR(temp_decorr_pass);
1075 2 temp_decorr_pass.delta = wpds->delta;
1076 2 temp_decorr_pass.value = wpds->terms[j];
1077
1078
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (temp_decorr_pass.value < 0)
1079 temp_decorr_pass.value = 1;
1080
1081 2 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
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
2 if (j) {
1085 1 CLEAR(temp_decorr_pass.samplesA);
1086 } else {
1087 1 reverse_mono_decorr(&temp_decorr_pass);
1088 }
1089
1090 2 memcpy(save_decorr_passes + j, &temp_decorr_pass, sizeof(struct Decorr));
1091 2 decorr_mono(s->temp_buffer[j&1][0], s->temp_buffer[~j&1][0],
1092 nb_samples, &temp_decorr_pass, 1);
1093 }
1094
1095 1 size = log2mono(s->temp_buffer[j&1][0], nb_samples, log_limit);
1096
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1 if (size != UINT32_MAX || !nterms)
1097 break;
1098 nterms >>= 1;
1099 }
1100
1101
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (size < best_size) {
1102 1 memcpy(s->best_buffer[0], s->temp_buffer[j&1][0], buf_size);
1103 1 memcpy(s->decorr_passes, save_decorr_passes, sizeof(struct Decorr) * MAX_TERMS);
1104 1 s->num_terms = nterms;
1105 1 s->best_decorr = c;
1106 1 best_size = size;
1107 }
1108
1109
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (pi++)
1110 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1111 }
1112
1113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (s->extra_flags)
1114 analyze_mono(s, samples, do_samples);
1115
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 else if (do_samples)
1116 memcpy(samples, s->best_buffer[0], buf_size);
1117
1118
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1 if (no_history || s->extra_flags) {
1119 1 CLEAR(s->w);
1120 1 scan_word(s, &s->w.c[0], s->best_buffer[0], nb_samples, -1);
1121 }
1122 1 return 0;
1123 }
1124
1125 120 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 120 int m = 0, i;
1130
1131 120 dpp->sumA = dpp->sumB = 0;
1132
1133
1/2
✓ Branch 0 taken 120 times.
✗ Branch 1 not taken.
120 if (dir < 0) {
1134 120 out_left += nb_samples - 1;
1135 120 out_right += nb_samples - 1;
1136 120 in_left += nb_samples - 1;
1137 120 in_right += nb_samples - 1;
1138 }
1139
1140 120 dpp->weightA = restore_weight(store_weight(dpp->weightA));
1141 120 dpp->weightB = restore_weight(store_weight(dpp->weightB));
1142
1143
2/2
✓ Branch 0 taken 960 times.
✓ Branch 1 taken 120 times.
1080 for (i = 0; i < MAX_TERM; i++) {
1144 960 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
1145 960 dpp->samplesB[i] = wp_exp2(log2s(dpp->samplesB[i]));
1146 }
1147
1148
7/7
✓ Branch 0 taken 21 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 25 times.
✓ Branch 3 taken 17 times.
✓ Branch 4 taken 3 times.
✓ Branch 5 taken 13 times.
✓ Branch 6 taken 1 times.
120 switch (dpp->value) {
1149 21 case 2:
1150
2/2
✓ Branch 0 taken 43008 times.
✓ Branch 1 taken 21 times.
43029 while (nb_samples--) {
1151 int32_t sam, tmp;
1152
1153 43008 sam = dpp->samplesA[0];
1154 43008 dpp->samplesA[0] = dpp->samplesA[1];
1155
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 43008 times.
43008 out_left[0] = tmp = (dpp->samplesA[1] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1156
4/4
✓ Branch 0 taken 24962 times.
✓ Branch 1 taken 18046 times.
✓ Branch 2 taken 23419 times.
✓ Branch 3 taken 1543 times.
43008 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1157 43008 dpp->sumA += dpp->weightA;
1158
1159 43008 sam = dpp->samplesB[0];
1160 43008 dpp->samplesB[0] = dpp->samplesB[1];
1161
2/2
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 42976 times.
43008 out_right[0] = tmp = (dpp->samplesB[1] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1162
4/4
✓ Branch 0 taken 36818 times.
✓ Branch 1 taken 6190 times.
✓ Branch 2 taken 34741 times.
✓ Branch 3 taken 2077 times.
43008 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1163 43008 dpp->sumB += dpp->weightB;
1164
1165 43008 in_left += dir;
1166 43008 out_left += dir;
1167 43008 in_right += dir;
1168 43008 out_right += dir;
1169 }
1170 21 break;
1171 40 case 17:
1172
2/2
✓ Branch 0 taken 81920 times.
✓ Branch 1 taken 40 times.
81960 while (nb_samples--) {
1173 int32_t sam, tmp;
1174
1175 81920 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1176 81920 dpp->samplesA[1] = dpp->samplesA[0];
1177
2/2
✓ Branch 0 taken 3023 times.
✓ Branch 1 taken 78897 times.
81920 out_left[0] = tmp = (dpp->samplesA[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1178
4/4
✓ Branch 0 taken 64957 times.
✓ Branch 1 taken 16963 times.
✓ Branch 2 taken 62571 times.
✓ Branch 3 taken 2386 times.
81920 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1179 81920 dpp->sumA += dpp->weightA;
1180
1181 81920 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1182 81920 dpp->samplesB[1] = dpp->samplesB[0];
1183
2/2
✓ Branch 0 taken 4028 times.
✓ Branch 1 taken 77892 times.
81920 out_right[0] = tmp = (dpp->samplesB[0] = in_right[0]) - APPLY_WEIGHT (dpp->weightB, sam);
1184
4/4
✓ Branch 0 taken 80670 times.
✓ Branch 1 taken 1250 times.
✓ Branch 2 taken 76508 times.
✓ Branch 3 taken 4162 times.
81920 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1185 81920 dpp->sumB += dpp->weightB;
1186
1187 81920 in_left += dir;
1188 81920 out_left += dir;
1189 81920 in_right += dir;
1190 81920 out_right += dir;
1191 }
1192 40 break;
1193 25 case 18:
1194
2/2
✓ Branch 0 taken 51200 times.
✓ Branch 1 taken 25 times.
51225 while (nb_samples--) {
1195 int32_t sam, tmp;
1196
1197 51200 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
1198 51200 dpp->samplesA[1] = dpp->samplesA[0];
1199
2/2
✓ Branch 0 taken 671 times.
✓ Branch 1 taken 50529 times.
51200 out_left[0] = tmp = (dpp->samplesA[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1200
4/4
✓ Branch 0 taken 40226 times.
✓ Branch 1 taken 10974 times.
✓ Branch 2 taken 39108 times.
✓ Branch 3 taken 1118 times.
51200 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1201 51200 dpp->sumA += dpp->weightA;
1202
1203 51200 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
1204 51200 dpp->samplesB[1] = dpp->samplesB[0];
1205
2/2
✓ Branch 0 taken 671 times.
✓ Branch 1 taken 50529 times.
51200 out_right[0] = tmp = (dpp->samplesB[0] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1206
4/4
✓ Branch 0 taken 49966 times.
✓ Branch 1 taken 1234 times.
✓ Branch 2 taken 48092 times.
✓ Branch 3 taken 1874 times.
51200 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1207 51200 dpp->sumB += dpp->weightB;
1208
1209 51200 in_left += dir;
1210 51200 out_left += dir;
1211 51200 in_right += dir;
1212 51200 out_right += dir;
1213 }
1214 25 break;
1215 17 default: {
1216 17 int k = dpp->value & (MAX_TERM - 1);
1217
1218
2/2
✓ Branch 0 taken 34816 times.
✓ Branch 1 taken 17 times.
34833 while (nb_samples--) {
1219 int32_t sam, tmp;
1220
1221 34816 sam = dpp->samplesA[m];
1222
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 34816 times.
34816 out_left[0] = tmp = (dpp->samplesA[k] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam);
1223
4/4
✓ Branch 0 taken 22208 times.
✓ Branch 1 taken 12608 times.
✓ Branch 2 taken 20184 times.
✓ Branch 3 taken 2024 times.
34816 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1224 34816 dpp->sumA += dpp->weightA;
1225
1226 34816 sam = dpp->samplesB[m];
1227
2/2
✓ Branch 0 taken 55 times.
✓ Branch 1 taken 34761 times.
34816 out_right[0] = tmp = (dpp->samplesB[k] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam);
1228
4/4
✓ Branch 0 taken 28242 times.
✓ Branch 1 taken 6574 times.
✓ Branch 2 taken 26274 times.
✓ Branch 3 taken 1968 times.
34816 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1229 34816 dpp->sumB += dpp->weightB;
1230
1231 34816 in_left += dir;
1232 34816 out_left += dir;
1233 34816 in_right += dir;
1234 34816 out_right += dir;
1235 34816 m = (m + 1) & (MAX_TERM - 1);
1236 34816 k = (k + 1) & (MAX_TERM - 1);
1237 }
1238
1239
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
17 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 17 break;
1253 }
1254 3 case -1:
1255
2/2
✓ Branch 0 taken 6144 times.
✓ Branch 1 taken 3 times.
6147 while (nb_samples--) {
1256 int32_t sam_A, sam_B, tmp;
1257
1258 6144 sam_A = dpp->samplesA[0];
1259
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6144 times.
6144 out_left[0] = tmp = (sam_B = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam_A);
1260
8/10
✓ Branch 0 taken 5896 times.
✓ Branch 1 taken 248 times.
✓ Branch 2 taken 5359 times.
✓ Branch 3 taken 537 times.
✓ Branch 4 taken 2246 times.
✓ Branch 5 taken 3113 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 2246 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 3113 times.
6144 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1261 6144 dpp->sumA += dpp->weightA;
1262
1263
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6144 times.
6144 out_right[0] = tmp = (dpp->samplesA[0] = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam_B);
1264
8/10
✓ Branch 0 taken 5706 times.
✓ Branch 1 taken 438 times.
✓ Branch 2 taken 2685 times.
✓ Branch 3 taken 3021 times.
✓ Branch 4 taken 750 times.
✓ Branch 5 taken 1935 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 750 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 1935 times.
6144 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1265 6144 dpp->sumB += dpp->weightB;
1266
1267 6144 in_left += dir;
1268 6144 out_left += dir;
1269 6144 in_right += dir;
1270 6144 out_right += dir;
1271 }
1272 3 break;
1273 13 case -2:
1274
2/2
✓ Branch 0 taken 26624 times.
✓ Branch 1 taken 13 times.
26637 while (nb_samples--) {
1275 int32_t sam_A, sam_B, tmp;
1276
1277 26624 sam_B = dpp->samplesB[0];
1278
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 26594 times.
26624 out_right[0] = tmp = (sam_A = in_right[0]) - APPLY_WEIGHT(dpp->weightB, sam_B);
1279
8/10
✓ Branch 0 taken 26025 times.
✓ Branch 1 taken 599 times.
✓ Branch 2 taken 25110 times.
✓ Branch 3 taken 915 times.
✓ Branch 4 taken 11467 times.
✓ Branch 5 taken 13643 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 11467 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 13643 times.
26624 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1280 26624 dpp->sumB += dpp->weightB;
1281
1282
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 26594 times.
26624 out_left[0] = tmp = (dpp->samplesB[0] = in_left[0]) - APPLY_WEIGHT(dpp->weightA, sam_A);
1283
9/10
✓ Branch 0 taken 26027 times.
✓ Branch 1 taken 597 times.
✓ Branch 2 taken 16895 times.
✓ Branch 3 taken 9132 times.
✓ Branch 4 taken 3818 times.
✓ Branch 5 taken 13077 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 3818 times.
✓ Branch 8 taken 920 times.
✓ Branch 9 taken 12157 times.
26624 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1284 26624 dpp->sumA += dpp->weightA;
1285
1286 26624 in_left += dir;
1287 26624 out_left += dir;
1288 26624 in_right += dir;
1289 26624 out_right += dir;
1290 }
1291 13 break;
1292 1 case -3:
1293
2/2
✓ Branch 0 taken 2048 times.
✓ Branch 1 taken 1 times.
2049 while (nb_samples--) {
1294 int32_t sam_A, sam_B, tmp;
1295
1296 2048 sam_A = dpp->samplesA[0];
1297 2048 sam_B = dpp->samplesB[0];
1298
1299 2048 dpp->samplesA[0] = tmp = in_right[0];
1300
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2048 times.
2048 out_right[0] = tmp -= APPLY_WEIGHT(dpp->weightB, sam_B);
1301
8/10
✓ Branch 0 taken 2026 times.
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 1938 times.
✓ Branch 3 taken 88 times.
✓ Branch 4 taken 976 times.
✓ Branch 5 taken 962 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 976 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 962 times.
2048 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1302 2048 dpp->sumB += dpp->weightB;
1303
1304 2048 dpp->samplesB[0] = tmp = in_left[0];
1305
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2048 times.
2048 out_left[0] = tmp -= APPLY_WEIGHT(dpp->weightA, sam_A);
1306
8/10
✓ Branch 0 taken 1952 times.
✓ Branch 1 taken 96 times.
✓ Branch 2 taken 1932 times.
✓ Branch 3 taken 20 times.
✓ Branch 4 taken 955 times.
✓ Branch 5 taken 977 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 955 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 977 times.
2048 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1307 2048 dpp->sumA += dpp->weightA;
1308
1309 2048 in_left += dir;
1310 2048 out_left += dir;
1311 2048 in_right += dir;
1312 2048 out_right += dir;
1313 }
1314 1 break;
1315 }
1316 120 }
1317
1318 24 static void reverse_decorr(struct Decorr *dpp)
1319 {
1320
2/2
✓ Branch 0 taken 23 times.
✓ Branch 1 taken 1 times.
24 if (dpp->value > MAX_TERM) {
1321 int32_t sam_A, sam_B;
1322
1323
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 11 times.
23 if (dpp->value & 1) {
1324 12 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1325 12 sam_B = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1326 } else {
1327 11 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
1328 11 sam_B = (3 * dpp->samplesB[0] - dpp->samplesB[1]) >> 1;
1329 }
1330
1331 23 dpp->samplesA[1] = dpp->samplesA[0];
1332 23 dpp->samplesB[1] = dpp->samplesB[0];
1333 23 dpp->samplesA[0] = sam_A;
1334 23 dpp->samplesB[0] = sam_B;
1335
1336
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 11 times.
23 if (dpp->value & 1) {
1337 12 sam_A = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1338 12 sam_B = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1339 } else {
1340 11 sam_A = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
1341 11 sam_B = (3 * dpp->samplesB[0] - dpp->samplesB[1]) >> 1;
1342 }
1343
1344 23 dpp->samplesA[1] = sam_A;
1345 23 dpp->samplesB[1] = sam_B;
1346
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 } else if (dpp->value > 1) {
1347 int i, j, k;
1348
1349
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
2 for (i = 0, j = dpp->value - 1, k = 0; k < dpp->value / 2; i++, j--, k++) {
1350 1 i &= (MAX_TERM - 1);
1351 1 j &= (MAX_TERM - 1);
1352 1 dpp->samplesA[i] ^= dpp->samplesA[j];
1353 1 dpp->samplesA[j] ^= dpp->samplesA[i];
1354 1 dpp->samplesA[i] ^= dpp->samplesA[j];
1355 1 dpp->samplesB[i] ^= dpp->samplesB[j];
1356 1 dpp->samplesB[j] ^= dpp->samplesB[i];
1357 1 dpp->samplesB[i] ^= dpp->samplesB[j];
1358 }
1359 }
1360 24 }
1361
1362 120 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 120 int m = 0, i;
1367
1368 120 dpp->weightA = restore_weight(store_weight(dpp->weightA));
1369 120 dpp->weightB = restore_weight(store_weight(dpp->weightB));
1370
1371
2/2
✓ Branch 0 taken 960 times.
✓ Branch 1 taken 120 times.
1080 for (i = 0; i < MAX_TERM; i++) {
1372 960 dpp->samplesA[i] = wp_exp2(log2s(dpp->samplesA[i]));
1373 960 dpp->samplesB[i] = wp_exp2(log2s(dpp->samplesB[i]));
1374 }
1375
1376
7/7
✓ Branch 0 taken 21 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 25 times.
✓ Branch 3 taken 17 times.
✓ Branch 4 taken 3 times.
✓ Branch 5 taken 13 times.
✓ Branch 6 taken 1 times.
120 switch (dpp->value) {
1377 21 case 2:
1378
2/2
✓ Branch 0 taken 463050 times.
✓ Branch 1 taken 21 times.
463071 for (i = 0; i < nb_samples; i++) {
1379 int32_t sam, tmp;
1380
1381 463050 sam = dpp->samplesA[0];
1382 463050 dpp->samplesA[0] = dpp->samplesA[1];
1383 463050 out_left[i] = tmp = (dpp->samplesA[1] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1384
4/4
✓ Branch 0 taken 263976 times.
✓ Branch 1 taken 199074 times.
✓ Branch 2 taken 242436 times.
✓ Branch 3 taken 21540 times.
463050 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1385
1386 463050 sam = dpp->samplesB[0];
1387 463050 dpp->samplesB[0] = dpp->samplesB[1];
1388 463050 out_right[i] = tmp = (dpp->samplesB[1] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1389
4/4
✓ Branch 0 taken 400796 times.
✓ Branch 1 taken 62254 times.
✓ Branch 2 taken 380033 times.
✓ Branch 3 taken 20763 times.
463050 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1390 }
1391 21 break;
1392 40 case 17:
1393
2/2
✓ Branch 0 taken 882000 times.
✓ Branch 1 taken 40 times.
882040 for (i = 0; i < nb_samples; i++) {
1394 int32_t sam, tmp;
1395
1396 882000 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
1397 882000 dpp->samplesA[1] = dpp->samplesA[0];
1398 882000 out_left[i] = tmp = (dpp->samplesA[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1399
4/4
✓ Branch 0 taken 693986 times.
✓ Branch 1 taken 188014 times.
✓ Branch 2 taken 659217 times.
✓ Branch 3 taken 34769 times.
882000 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1400
1401 882000 sam = 2 * dpp->samplesB[0] - dpp->samplesB[1];
1402 882000 dpp->samplesB[1] = dpp->samplesB[0];
1403 882000 out_right[i] = tmp = (dpp->samplesB[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1404
4/4
✓ Branch 0 taken 868910 times.
✓ Branch 1 taken 13090 times.
✓ Branch 2 taken 828745 times.
✓ Branch 3 taken 40165 times.
882000 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1405 }
1406 40 break;
1407 25 case 18:
1408
2/2
✓ Branch 0 taken 551250 times.
✓ Branch 1 taken 25 times.
551275 for (i = 0; i < nb_samples; i++) {
1409 int32_t sam, tmp;
1410
1411 551250 sam = dpp->samplesA[0] + ((dpp->samplesA[0] - dpp->samplesA[1]) >> 1);
1412 551250 dpp->samplesA[1] = dpp->samplesA[0];
1413 551250 out_left[i] = tmp = (dpp->samplesA[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1414
4/4
✓ Branch 0 taken 429384 times.
✓ Branch 1 taken 121866 times.
✓ Branch 2 taken 413959 times.
✓ Branch 3 taken 15425 times.
551250 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1415
1416 551250 sam = dpp->samplesB[0] + ((dpp->samplesB[0] - dpp->samplesB[1]) >> 1);
1417 551250 dpp->samplesB[1] = dpp->samplesB[0];
1418 551250 out_right[i] = tmp = (dpp->samplesB[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1419
4/4
✓ Branch 0 taken 540079 times.
✓ Branch 1 taken 11171 times.
✓ Branch 2 taken 523574 times.
✓ Branch 3 taken 16505 times.
551250 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1420 }
1421 25 break;
1422 17 default: {
1423 17 int k = dpp->value & (MAX_TERM - 1);
1424
1425
2/2
✓ Branch 0 taken 374850 times.
✓ Branch 1 taken 17 times.
374867 for (i = 0; i < nb_samples; i++) {
1426 int32_t sam, tmp;
1427
1428 374850 sam = dpp->samplesA[m];
1429 374850 out_left[i] = tmp = (dpp->samplesA[k] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam);
1430
4/4
✓ Branch 0 taken 238577 times.
✓ Branch 1 taken 136273 times.
✓ Branch 2 taken 215368 times.
✓ Branch 3 taken 23209 times.
374850 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, tmp);
1431
1432 374850 sam = dpp->samplesB[m];
1433 374850 out_right[i] = tmp = (dpp->samplesB[k] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam);
1434
4/4
✓ Branch 0 taken 307328 times.
✓ Branch 1 taken 67522 times.
✓ Branch 2 taken 285909 times.
✓ Branch 3 taken 21419 times.
374850 UPDATE_WEIGHT(dpp->weightB, dpp->delta, sam, tmp);
1435
1436 374850 m = (m + 1) & (MAX_TERM - 1);
1437 374850 k = (k + 1) & (MAX_TERM - 1);
1438 }
1439
1440
1/2
✓ Branch 0 taken 17 times.
✗ Branch 1 not taken.
17 if (m) {
1441 int32_t temp_A[MAX_TERM], temp_B[MAX_TERM];
1442 int k;
1443
1444 17 memcpy(temp_A, dpp->samplesA, sizeof(dpp->samplesA));
1445 17 memcpy(temp_B, dpp->samplesB, sizeof(dpp->samplesB));
1446
1447
2/2
✓ Branch 0 taken 136 times.
✓ Branch 1 taken 17 times.
153 for (k = 0; k < MAX_TERM; k++) {
1448 136 dpp->samplesA[k] = temp_A[m];
1449 136 dpp->samplesB[k] = temp_B[m];
1450 136 m = (m + 1) & (MAX_TERM - 1);
1451 }
1452 }
1453 17 break;
1454 }
1455 3 case -1:
1456
2/2
✓ Branch 0 taken 66150 times.
✓ Branch 1 taken 3 times.
66153 for (i = 0; i < nb_samples; i++) {
1457 int32_t sam_A, sam_B, tmp;
1458
1459 66150 sam_A = dpp->samplesA[0];
1460 66150 out_left[i] = tmp = (sam_B = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
1461
8/10
✓ Branch 0 taken 62663 times.
✓ Branch 1 taken 3487 times.
✓ Branch 2 taken 56065 times.
✓ Branch 3 taken 6598 times.
✓ Branch 4 taken 28021 times.
✓ Branch 5 taken 28044 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 28021 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 28044 times.
66150 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1462
1463 66150 out_right[i] = tmp = (dpp->samplesA[0] = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
1464
8/10
✓ Branch 0 taken 61753 times.
✓ Branch 1 taken 4397 times.
✓ Branch 2 taken 17252 times.
✓ Branch 3 taken 44501 times.
✓ Branch 4 taken 8568 times.
✓ Branch 5 taken 8684 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 8568 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 8684 times.
66150 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1465 }
1466 3 break;
1467 13 case -2:
1468
2/2
✓ Branch 0 taken 286650 times.
✓ Branch 1 taken 13 times.
286663 for (i = 0; i < nb_samples; i++) {
1469 int32_t sam_A, sam_B, tmp;
1470
1471 286650 sam_B = dpp->samplesB[0];
1472 286650 out_right[i] = tmp = (sam_A = in_right[i]) - APPLY_WEIGHT_I(dpp->weightB, sam_B);
1473
9/10
✓ Branch 0 taken 280614 times.
✓ Branch 1 taken 6036 times.
✓ Branch 2 taken 274117 times.
✓ Branch 3 taken 6497 times.
✓ Branch 4 taken 132987 times.
✓ Branch 5 taken 141130 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 132987 times.
✓ Branch 8 taken 7601 times.
✓ Branch 9 taken 133529 times.
286650 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1474
1475 286650 out_left[i] = tmp = (dpp->samplesB[0] = in_left[i]) - APPLY_WEIGHT_I(dpp->weightA, sam_A);
1476
9/10
✓ Branch 0 taken 280065 times.
✓ Branch 1 taken 6585 times.
✓ Branch 2 taken 102274 times.
✓ Branch 3 taken 177791 times.
✓ Branch 4 taken 46542 times.
✓ Branch 5 taken 55732 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 46542 times.
✓ Branch 8 taken 10080 times.
✓ Branch 9 taken 45652 times.
286650 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1477 }
1478 13 break;
1479 1 case -3:
1480
2/2
✓ Branch 0 taken 22050 times.
✓ Branch 1 taken 1 times.
22051 for (i = 0; i < nb_samples; i++) {
1481 int32_t sam_A, sam_B, tmp;
1482
1483 22050 sam_A = dpp->samplesA[0];
1484 22050 sam_B = dpp->samplesB[0];
1485
1486 22050 dpp->samplesA[0] = tmp = in_right[i];
1487 22050 out_right[i] = tmp -= APPLY_WEIGHT_I(dpp->weightB, sam_B);
1488
8/10
✓ Branch 0 taken 21647 times.
✓ Branch 1 taken 403 times.
✓ Branch 2 taken 20357 times.
✓ Branch 3 taken 1290 times.
✓ Branch 4 taken 10187 times.
✓ Branch 5 taken 10170 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 10187 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 10170 times.
22050 UPDATE_WEIGHT_CLIP(dpp->weightB, dpp->delta, sam_B, tmp);
1489
1490 22050 dpp->samplesB[0] = tmp = in_left[i];
1491 22050 out_left[i] = tmp -= APPLY_WEIGHT_I(dpp->weightA, sam_A);
1492
8/10
✓ Branch 0 taken 20719 times.
✓ Branch 1 taken 1331 times.
✓ Branch 2 taken 20361 times.
✓ Branch 3 taken 358 times.
✓ Branch 4 taken 10201 times.
✓ Branch 5 taken 10160 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 10201 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 10160 times.
22050 UPDATE_WEIGHT_CLIP(dpp->weightA, dpp->delta, sam_A, tmp);
1493 }
1494 1 break;
1495 }
1496 120 }
1497
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
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
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
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
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
1809 12 static int wv_stereo(WavPackEncodeContext *s,
1810 int32_t *samples_l, int32_t *samples_r,
1811 int no_history, int do_samples)
1812 {
1813 12 struct Decorr temp_decorr_pass, save_decorr_passes[MAX_TERMS] = {{0}};
1814 12 int nb_samples = s->block_samples, ret;
1815 12 int buf_size = sizeof(int32_t) * nb_samples;
1816 12 int log_limit, force_js = 0, force_ts = 0, got_js = 0, pi, i;
1817 12 uint32_t best_size = UINT32_MAX, size;
1818
1819
1/2
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
12 for (i = 0; i < nb_samples; i++)
1820
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
12 if (samples_l[i] || samples_r[i])
1821 break;
1822
1823
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 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 12 log_limit = (((s->flags & MAG_MASK) >> MAG_LSB) + 4) * 256;
1832 12 log_limit = FFMIN(6912, log_limit);
1833
1834
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 if (s->joint != -1) {
1835 force_js = s->joint;
1836 force_ts = !s->joint;
1837 }
1838
1839
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
12 if ((ret = allocate_buffers(s)) < 0)
1840 return ret;
1841
1842
3/4
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 11 times.
12 if (no_history || s->num_passes >= 7)
1843 1 s->best_decorr = s->mask_decorr = 0;
1844
1845
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 12 times.
37 for (pi = 0; pi < s->num_passes;) {
1846 const WavPackDecorrSpec *wpds;
1847 int nterms, c, j;
1848
1849
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 13 times.
25 if (!pi)
1850 12 c = s->best_decorr;
1851 else {
1852
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 11 times.
13 if (s->mask_decorr == 0)
1853 2 c = 0;
1854 else
1855 11 c = (s->best_decorr & (s->mask_decorr - 1)) | s->mask_decorr;
1856
1857
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (c == s->best_decorr) {
1858
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1859 1 continue;
1860 }
1861 }
1862
1863 24 wpds = &s->decorr_specs[c];
1864 24 nterms = decorr_filter_nterms[s->decorr_filter];
1865
1866 while (1) {
1867
4/6
✓ Branch 0 taken 24 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 21 times.
✓ Branch 4 taken 3 times.
✗ Branch 5 not taken.
24 if (force_js || (wpds->joint_stereo && !force_ts)) {
1868
1/2
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
3 if (!got_js) {
1869 3 av_fast_padded_malloc(&s->js_left, &s->js_left_size, buf_size);
1870 3 av_fast_padded_malloc(&s->js_right, &s->js_right_size, buf_size);
1871 3 memcpy(s->js_left, samples_l, buf_size);
1872 3 memcpy(s->js_right, samples_r, buf_size);
1873
1874
2/2
✓ Branch 0 taken 66150 times.
✓ Branch 1 taken 3 times.
66153 for (i = 0; i < nb_samples; i++)
1875 66150 s->js_right[i] += ((s->js_left[i] -= s->js_right[i]) >> 1);
1876 3 got_js = 1;
1877 }
1878
1879 3 memcpy(s->temp_buffer[0][0], s->js_left, buf_size);
1880 3 memcpy(s->temp_buffer[0][1], s->js_right, buf_size);
1881 } else {
1882 21 memcpy(s->temp_buffer[0][0], samples_l, buf_size);
1883 21 memcpy(s->temp_buffer[0][1], samples_r, buf_size);
1884 }
1885
1886 24 CLEAR(save_decorr_passes);
1887
1888
2/2
✓ Branch 0 taken 120 times.
✓ Branch 1 taken 24 times.
144 for (j = 0; j < nterms; j++) {
1889 120 CLEAR(temp_decorr_pass);
1890 120 temp_decorr_pass.delta = wpds->delta;
1891 120 temp_decorr_pass.value = wpds->terms[j];
1892
1893
3/4
✓ Branch 0 taken 17 times.
✓ Branch 1 taken 103 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 17 times.
120 if (temp_decorr_pass.value < 0 && !(s->flags & WV_CROSS_DECORR))
1894 temp_decorr_pass.value = -3;
1895
1896 120 decorr_stereo(s->temp_buffer[ j&1][0], s->temp_buffer[ j&1][1],
1897 120 s->temp_buffer[~j&1][0], s->temp_buffer[~j&1][1],
1898 FFMIN(2048, nb_samples), &temp_decorr_pass, -1);
1899
1900
2/2
✓ Branch 0 taken 96 times.
✓ Branch 1 taken 24 times.
120 if (j) {
1901 96 CLEAR(temp_decorr_pass.samplesA);
1902 96 CLEAR(temp_decorr_pass.samplesB);
1903 } else {
1904 24 reverse_decorr(&temp_decorr_pass);
1905 }
1906
1907 120 memcpy(save_decorr_passes + j, &temp_decorr_pass, sizeof(struct Decorr));
1908
1909
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 120 times.
120 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 120 decorr_stereo_quick(s->temp_buffer[ j&1][0], s->temp_buffer[ j&1][1],
1915 120 s->temp_buffer[~j&1][0], s->temp_buffer[~j&1][1],
1916 nb_samples, &temp_decorr_pass);
1917 }
1918
1919 24 size = log2stereo(s->temp_buffer[j&1][0], s->temp_buffer[j&1][1],
1920 nb_samples, log_limit);
1921
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
24 if (size != UINT32_MAX || !nterms)
1922 break;
1923 nterms >>= 1;
1924 }
1925
1926
2/2
✓ Branch 0 taken 18 times.
✓ Branch 1 taken 6 times.
24 if (size < best_size) {
1927 18 memcpy(s->best_buffer[0], s->temp_buffer[j&1][0], buf_size);
1928 18 memcpy(s->best_buffer[1], s->temp_buffer[j&1][1], buf_size);
1929 18 memcpy(s->decorr_passes, save_decorr_passes, sizeof(struct Decorr) * MAX_TERMS);
1930 18 s->num_terms = nterms;
1931 18 s->best_decorr = c;
1932 18 best_size = size;
1933 }
1934
1935
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 12 times.
24 if (pi++)
1936
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 1 times.
12 s->mask_decorr = s->mask_decorr ? ((s->mask_decorr << 1) & (s->num_decorrs - 1)) : 1;
1937 }
1938
1939
2/6
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 12 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
12 if (force_js || (s->decorr_specs[s->best_decorr].joint_stereo && !force_ts))
1940 s->flags |= WV_JOINT_STEREO;
1941 else
1942 12 s->flags &= ~((uint32_t) WV_JOINT_STEREO);
1943
1944
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 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
1/2
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
12 } else if (do_samples) {
1955 12 memcpy(samples_l, s->best_buffer[0], buf_size);
1956 12 memcpy(samples_r, s->best_buffer[1], buf_size);
1957 }
1958
1959
3/4
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✓ Branch 3 taken 1 times.
12 if (s->extra_flags || no_history ||
1960
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
11 s->joint_stereo != s->decorr_specs[s->best_decorr].joint_stereo) {
1961 1 s->joint_stereo = s->decorr_specs[s->best_decorr].joint_stereo;
1962 1 CLEAR(s->w);
1963 1 scan_word(s, &s->w.c[0], s->best_buffer[0], nb_samples, -1);
1964 1 scan_word(s, &s->w.c[1], s->best_buffer[1], nb_samples, -1);
1965 }
1966 12 return 0;
1967 }
1968
1969 573304 static void encode_flush(WavPackEncodeContext *s)
1970 {
1971 573304 WavPackWords *w = &s->w;
1972 573304 PutBitContext *pb = &s->pb;
1973
1974
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 573304 times.
573304 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, (1 << 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
2/2
✓ Branch 0 taken 125252 times.
✓ Branch 1 taken 448052 times.
573304 if (w->holding_one) {
1998
2/2
✓ Branch 0 taken 81 times.
✓ Branch 1 taken 125171 times.
125252 if (w->holding_one >= 16) {
1999 int cbits;
2000
2001 81 put_bits(pb, 16, (1 << 16) - 1);
2002 81 put_bits(pb, 1, 0);
2003 81 w->holding_one -= 16;
2004
2/2
✓ Branch 0 taken 77 times.
✓ Branch 1 taken 4 times.
81 cbits = count_bits(w->holding_one);
2005
2006 do {
2007
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
81 if (cbits > 31) {
2008 put_bits(pb, 31, 0x7FFFFFFF);
2009 cbits -= 31;
2010 } else {
2011 81 put_bits(pb, cbits, (1 << cbits) - 1);
2012 81 cbits = 0;
2013 }
2014
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
81 } while (cbits);
2015
2016 81 put_bits(pb, 1, 0);
2017
2018
2/2
✓ Branch 0 taken 371 times.
✓ Branch 1 taken 81 times.
452 while (w->holding_one > 1) {
2019 371 put_bits(pb, 1, w->holding_one & 1);
2020 371 w->holding_one >>= 1;
2021 }
2022
2023 81 w->holding_zero = 0;
2024 } else {
2025 125171 put_bits(pb, w->holding_one, (1 << w->holding_one) - 1);
2026 }
2027
2028 125252 w->holding_one = 0;
2029 }
2030
2031
2/2
✓ Branch 0 taken 307729 times.
✓ Branch 1 taken 265575 times.
573304 if (w->holding_zero) {
2032 307729 put_bits(pb, 1, 0);
2033 307729 w->holding_zero = 0;
2034 }
2035
2036
2/2
✓ Branch 0 taken 573300 times.
✓ Branch 1 taken 4 times.
573304 if (w->pend_count) {
2037 573300 put_bits(pb, w->pend_count, w->pend_data);
2038 573300 w->pend_data = w->pend_count = 0;
2039 }
2040 573304 }
2041
2042 573300 static void wavpack_encode_sample(WavPackEncodeContext *s, WvChannel *c, int32_t sample)
2043 {
2044 573300 WavPackWords *w = &s->w;
2045 uint32_t ones_count, low, high;
2046 573300 int sign = sample < 0;
2047
2048
5/6
✓ Branch 0 taken 224776 times.
✓ Branch 1 taken 348524 times.
✓ Branch 2 taken 112330 times.
✓ Branch 3 taken 112446 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 112330 times.
573300 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
2/2
✓ Branch 0 taken 203451 times.
✓ Branch 1 taken 369849 times.
573300 if (sign)
2067 203451 sample = ~sample;
2068
2069
2/2
✓ Branch 0 taken 447048 times.
✓ Branch 1 taken 126252 times.
573300 if (sample < (int32_t) GET_MED(0)) {
2070 447048 ones_count = low = 0;
2071 447048 high = GET_MED(0) - 1;
2072 447048 DEC_MED(0);
2073 } else {
2074 126252 low = GET_MED(0);
2075 126252 INC_MED(0);
2076
2077
2/2
✓ Branch 0 taken 89886 times.
✓ Branch 1 taken 36366 times.
126252 if (sample - low < GET_MED(1)) {
2078 89886 ones_count = 1;
2079 89886 high = low + GET_MED(1) - 1;
2080 89886 DEC_MED(1);
2081 } else {
2082 36366 low += GET_MED(1);
2083 36366 INC_MED(1);
2084
2085
2/2
✓ Branch 0 taken 25626 times.
✓ Branch 1 taken 10740 times.
36366 if (sample - low < GET_MED(2)) {
2086 25626 ones_count = 2;
2087 25626 high = low + GET_MED(2) - 1;
2088 25626 DEC_MED(2);
2089 } else {
2090 10740 ones_count = 2 + (sample - low) / GET_MED(2);
2091 10740 low += (ones_count - 2) * GET_MED(2);
2092 10740 high = low + GET_MED(2) - 1;
2093 10740 INC_MED(2);
2094 }
2095 }
2096 }
2097
2098
2/2
✓ Branch 0 taken 307801 times.
✓ Branch 1 taken 265499 times.
573300 if (w->holding_zero) {
2099
2/2
✓ Branch 0 taken 42311 times.
✓ Branch 1 taken 265490 times.
307801 if (ones_count)
2100 42311 w->holding_one++;
2101
2102 307801 encode_flush(s);
2103
2104
2/2
✓ Branch 0 taken 42311 times.
✓ Branch 1 taken 265490 times.
307801 if (ones_count) {
2105 42311 w->holding_zero = 1;
2106 42311 ones_count--;
2107 } else
2108 265490 w->holding_zero = 0;
2109 } else
2110 265499 w->holding_zero = 1;
2111
2112 573300 w->holding_one = ones_count * 2;
2113
2114
2/2
✓ Branch 0 taken 417675 times.
✓ Branch 1 taken 155625 times.
573300 if (high != low) {
2115 417675 uint32_t maxcode = high - low, code = sample - low;
2116
1/2
✓ Branch 0 taken 417675 times.
✗ Branch 1 not taken.
417675 int bitcount = count_bits(maxcode);
2117 417675 uint32_t extras = (1 << bitcount) - maxcode - 1;
2118
2119
2/2
✓ Branch 0 taken 151217 times.
✓ Branch 1 taken 266458 times.
417675 if (code < extras) {
2120 151217 w->pend_data |= code << w->pend_count;
2121 151217 w->pend_count += bitcount - 1;
2122 } else {
2123 266458 w->pend_data |= ((code + extras) >> 1) << w->pend_count;
2124 266458 w->pend_count += bitcount - 1;
2125 266458 w->pend_data |= ((code + extras) & 1) << w->pend_count++;
2126 }
2127 }
2128
2129 573300 w->pend_data |= ((int32_t) sign << w->pend_count++);
2130
2131
2/2
✓ Branch 0 taken 265490 times.
✓ Branch 1 taken 307810 times.
573300 if (!w->holding_zero)
2132 265490 encode_flush(s);
2133 }
2134
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
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
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
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
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
2460 26 static void put_metadata_block(PutByteContext *pb, int flags, int size)
2461 {
2462
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 14 times.
26 if (size & 1)
2463 12 flags |= WP_IDF_ODD;
2464
2465 26 bytestream2_put_byte(pb, flags);
2466 26 bytestream2_put_byte(pb, (size + 1) >> 1);
2467 26 }
2468
2469 13 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 13 int block_size, start, end, data_size, tcount, temp, m = 0;
2474 13 int i, j, ret = 0, got_extra = 0, nb_samples = s->block_samples;
2475 13 uint32_t crc = 0xffffffffu;
2476 struct Decorr *dpp;
2477 PutByteContext pb;
2478
2479
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (s->flags & WV_MONO_DATA) {
2480 1 CLEAR(s->w);
2481 }
2482
3/4
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 12 times.
13 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
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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
2/4
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 13 times.
13 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 13 scan_int23(s, samples_l, samples_r, nb_samples);
2539
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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
3/4
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
13 if (!s->num_passes && !s->num_terms) {
2546 1 s->num_passes = 1;
2547
2548
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (s->flags & WV_MONO_DATA)
2549 1 ret = wv_mono(s, samples_l, 1, 0);
2550 else
2551 ret = wv_stereo(s, samples_l, samples_r, 1, 0);
2552
2553 1 s->num_passes = 0;
2554 }
2555
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (s->flags & WV_MONO_DATA) {
2556
2/2
✓ Branch 0 taken 44100 times.
✓ Branch 1 taken 1 times.
44101 for (i = 0; i < nb_samples; i++)
2557 44100 crc += (crc << 1) + samples_l[i];
2558
2559
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 if (s->num_passes)
2560 ret = wv_mono(s, samples_l, !s->num_terms, 1);
2561 } else {
2562
2/2
✓ Branch 0 taken 264600 times.
✓ Branch 1 taken 12 times.
264612 for (i = 0; i < nb_samples; i++)
2563 264600 crc += (crc << 3) + ((uint32_t)samples_l[i] << 1) + samples_l[i] + samples_r[i];
2564
2565
1/2
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
12 if (s->num_passes)
2566 12 ret = wv_stereo(s, samples_l, samples_r, !s->num_terms, 1);
2567 }
2568
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 if (ret < 0)
2569 return ret;
2570
2571
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (!s->ch_offset)
2572 13 s->flags |= WV_INITIAL_BLOCK;
2573
2574
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 s->ch_offset += 1 + !(s->flags & WV_MONO);
2575
2576
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (s->ch_offset == s->avctx->ch_layout.nb_channels)
2577 13 s->flags |= WV_FINAL_BLOCK;
2578
2579 13 bytestream2_init_writer(&pb, out, out_size);
2580 13 bytestream2_put_le32(&pb, MKTAG('w', 'v', 'p', 'k'));
2581 13 bytestream2_put_le32(&pb, 0);
2582 13 bytestream2_put_le16(&pb, 0x410);
2583 13 bytestream2_put_le16(&pb, 0);
2584 13 bytestream2_put_le32(&pb, 0);
2585 13 bytestream2_put_le32(&pb, s->sample_index);
2586 13 bytestream2_put_le32(&pb, nb_samples);
2587 13 bytestream2_put_le32(&pb, s->flags);
2588 13 bytestream2_put_le32(&pb, crc);
2589
2590
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (s->flags & WV_INITIAL_BLOCK &&
2591
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 s->avctx->ch_layout.order == AV_CHANNEL_ORDER_NATIVE &&
2592
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 s->avctx->ch_layout.u.mask != AV_CH_LAYOUT_MONO &&
2593
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 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
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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 13 put_metadata_block(&pb, WP_ID_DECTERMS, s->num_terms);
2607
2/2
✓ Branch 0 taken 62 times.
✓ Branch 1 taken 13 times.
75 for (i = 0; i < s->num_terms; i++) {
2608 62 struct Decorr *dpp = &s->decorr_passes[i];
2609 62 bytestream2_put_byte(&pb, ((dpp->value + 5) & 0x1f) | ((dpp->delta << 5) & 0xe0));
2610 }
2611
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 if (s->num_terms & 1)
2612 12 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 13 bytestream2_put_byte(&pb, WP_ID_DECWEIGHTS);
2621 13 bytestream2_put_byte(&pb, 0);
2622 13 start = bytestream2_tell_p(&pb);
2623
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 for (i = s->num_terms - 1; i >= 0; --i) {
2624 13 struct Decorr *dpp = &s->decorr_passes[i];
2625
2626
2/2
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 8 times.
13 if (store_weight(dpp->weightA) ||
2627
2/4
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 5 times.
5 (!(s->flags & WV_MONO_DATA) && store_weight(dpp->weightB)))
2628 break;
2629 }
2630 13 tcount = i + 1;
2631
2/2
✓ Branch 0 taken 62 times.
✓ Branch 1 taken 13 times.
75 for (i = 0; i < s->num_terms; i++) {
2632 62 struct Decorr *dpp = &s->decorr_passes[i];
2633
1/2
✓ Branch 0 taken 62 times.
✗ Branch 1 not taken.
62 if (i < tcount) {
2634 62 WRITE_DECWEIGHT(dpp->weightA);
2635
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 2 times.
62 if (!(s->flags & WV_MONO_DATA))
2636 60 WRITE_DECWEIGHT(dpp->weightB);
2637 } else {
2638 dpp->weightA = dpp->weightB = 0;
2639 }
2640 }
2641 13 end = bytestream2_tell_p(&pb);
2642 13 out[start - 2] = WP_ID_DECWEIGHTS | (((end - start) & 1) ? WP_IDF_ODD: 0);
2643 13 out[start - 1] = (end - start + 1) >> 1;
2644
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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 13 bytestream2_put_byte(&pb, WP_ID_DECSAMPLES);
2654 13 bytestream2_put_byte(&pb, 0);
2655 13 start = bytestream2_tell_p(&pb);
2656
2/2
✓ Branch 0 taken 62 times.
✓ Branch 1 taken 13 times.
75 for (i = 0; i < s->num_terms; i++) {
2657 62 struct Decorr *dpp = &s->decorr_passes[i];
2658
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 49 times.
62 if (i == 0) {
2659
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (dpp->value > MAX_TERM) {
2660 13 WRITE_DECSAMPLE(dpp->samplesA[0]);
2661 13 WRITE_DECSAMPLE(dpp->samplesA[1]);
2662
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 if (!(s->flags & WV_MONO_DATA)) {
2663 12 WRITE_DECSAMPLE(dpp->samplesB[0]);
2664 12 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 49 CLEAR(dpp->samplesA);
2678 49 CLEAR(dpp->samplesB);
2679 }
2680 }
2681 13 end = bytestream2_tell_p(&pb);
2682 13 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
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 put_metadata_block(&pb, WP_ID_ENTROPY, 6 * (1 + (!(s->flags & WV_MONO_DATA))));
2693
2/2
✓ Branch 3 taken 39 times.
✓ Branch 4 taken 13 times.
52 WRITE_CHAN_ENTROPY(0);
2694
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 if (!(s->flags & WV_MONO_DATA))
2695
2/2
✓ Branch 3 taken 36 times.
✓ Branch 4 taken 12 times.
48 WRITE_CHAN_ENTROPY(1);
2696
2697
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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
3/4
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
13 if (s->flags & WV_MONO_DATA && !s->num_passes) {
2714
2/2
✓ Branch 0 taken 44100 times.
✓ Branch 1 taken 1 times.
44101 for (i = 0; i < nb_samples; i++) {
2715 44100 int32_t code = samples_l[i];
2716
2717
2/2
✓ Branch 0 taken 88200 times.
✓ Branch 1 taken 44100 times.
132300 for (tcount = s->num_terms, dpp = s->decorr_passes; tcount--; dpp++) {
2718 int32_t sam;
2719
2720
1/2
✓ Branch 0 taken 88200 times.
✗ Branch 1 not taken.
88200 if (dpp->value > MAX_TERM) {
2721
2/2
✓ Branch 0 taken 44100 times.
✓ Branch 1 taken 44100 times.
88200 if (dpp->value & 1)
2722 44100 sam = 2 * dpp->samplesA[0] - dpp->samplesA[1];
2723 else
2724 44100 sam = (3 * dpp->samplesA[0] - dpp->samplesA[1]) >> 1;
2725
2726 88200 dpp->samplesA[1] = dpp->samplesA[0];
2727 88200 dpp->samplesA[0] = code;
2728 } else {
2729 sam = dpp->samplesA[m];
2730 dpp->samplesA[(m + dpp->value) & (MAX_TERM - 1)] = code;
2731 }
2732
2733
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 88200 times.
88200 code -= APPLY_WEIGHT(dpp->weightA, sam);
2734
4/4
✓ Branch 0 taken 88192 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 88150 times.
✓ Branch 3 taken 42 times.
88200 UPDATE_WEIGHT(dpp->weightA, dpp->delta, sam, code);
2735 }
2736
2737 44100 m = (m + 1) & (MAX_TERM - 1);
2738 44100 samples_l[i] = code;
2739 }
2740
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
1 if (m) {
2741
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
3 for (tcount = s->num_terms, dpp = s->decorr_passes; tcount--; dpp++)
2742
2/4
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
2 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
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 } 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 13 bytestream2_put_byte(&pb, WP_ID_DATA | WP_IDF_LONG);
2772 13 init_put_bits(&s->pb, pb.buffer + 3, bytestream2_get_bytes_left_p(&pb));
2773
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (s->flags & WV_MONO_DATA) {
2774
2/2
✓ Branch 0 taken 44100 times.
✓ Branch 1 taken 1 times.
44101 for (i = 0; i < nb_samples; i++)
2775 44100 wavpack_encode_sample(s, &s->w.c[0], s->samples[0][i]);
2776 } else {
2777
2/2
✓ Branch 0 taken 264600 times.
✓ Branch 1 taken 12 times.
264612 for (i = 0; i < nb_samples; i++) {
2778 264600 wavpack_encode_sample(s, &s->w.c[0], s->samples[0][i]);
2779 264600 wavpack_encode_sample(s, &s->w.c[1], s->samples[1][i]);
2780 }
2781 }
2782 13 encode_flush(s);
2783 13 flush_put_bits(&s->pb);
2784 13 data_size = put_bytes_output(&s->pb);
2785 13 bytestream2_put_le24(&pb, (data_size + 1) >> 1);
2786 13 bytestream2_skip_p(&pb, data_size);
2787
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 6 times.
13 if (data_size & 1)
2788 7 bytestream2_put_byte(&pb, 0);
2789
2790
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 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 13 block_size = bytestream2_tell_p(&pb);
2807 13 AV_WL32(out + 4, block_size - 8);
2808
2809
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 13 times.
13 av_assert0(!bytestream2_get_eof(&pb));
2810
2811 13 return block_size;
2812 }
2813
2814 25 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
1/5
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
25 switch (s->avctx->sample_fmt) {
2827 case AV_SAMPLE_FMT_U8P:
2828 COPY_SAMPLES(int8_t, 0x80, 0);
2829 break;
2830 25 case AV_SAMPLE_FMT_S16P:
2831
2/2
✓ Branch 0 taken 573300 times.
✓ Branch 1 taken 25 times.
573325 COPY_SAMPLES(int16_t, 0, 0);
2832 25 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 25 }
2842
2843 13 static void set_samplerate(WavPackEncodeContext *s)
2844 {
2845 int i;
2846
2847
1/2
✓ Branch 0 taken 130 times.
✗ Branch 1 not taken.
130 for (i = 0; i < 15; i++) {
2848
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 117 times.
130 if (wv_rates[i] == s->avctx->sample_rate)
2849 13 break;
2850 }
2851
2852 13 s->flags = i << SRATE_LSB;
2853 13 }
2854
2855 13 static int wavpack_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
2856 const AVFrame *frame, int *got_packet_ptr)
2857 {
2858 13 WavPackEncodeContext *s = avctx->priv_data;
2859 int buf_size, ret;
2860 uint8_t *buf;
2861
2862 13 s->block_samples = frame->nb_samples;
2863 13 av_fast_padded_malloc(&s->samples[0], &s->samples_size[0],
2864 13 sizeof(int32_t) * s->block_samples);
2865
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 if (!s->samples[0])
2866 return AVERROR(ENOMEM);
2867
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
13 if (avctx->ch_layout.nb_channels > 1) {
2868 12 av_fast_padded_malloc(&s->samples[1], &s->samples_size[1],
2869 12 sizeof(int32_t) * s->block_samples);
2870
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
12 if (!s->samples[1])
2871 return AVERROR(ENOMEM);
2872 }
2873
2874 13 buf_size = s->block_samples * avctx->ch_layout.nb_channels * 8
2875 13 + 200 * avctx->ch_layout.nb_channels /* for headers */;
2876
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 13 times.
13 if ((ret = ff_alloc_packet(avctx, avpkt, buf_size)) < 0)
2877 return ret;
2878 13 buf = avpkt->data;
2879
2880
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 13 times.
26 for (s->ch_offset = 0; s->ch_offset < avctx->ch_layout.nb_channels;) {
2881 13 set_samplerate(s);
2882
2883
1/4
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
13 switch (s->avctx->sample_fmt) {
2884 13 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 13 fill_buffer(s, frame->extended_data[s->ch_offset], s->samples[0], s->block_samples);
2890
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 12 times.
13 if (avctx->ch_layout.nb_channels - s->ch_offset == 1) {
2891 1 s->flags |= WV_MONO;
2892 } else {
2893 12 s->flags |= WV_CROSS_DECORR;
2894 12 fill_buffer(s, frame->extended_data[s->ch_offset + 1], s->samples[1], s->block_samples);
2895 }
2896
2897 13 s->flags += (1 << MAG_LSB) * ((s->flags & 3) * 8 + 7);
2898
2899
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 13 times.
13 if ((ret = wavpack_encode_block(s, s->samples[0], s->samples[1],
2900 buf, buf_size)) < 0)
2901 return ret;
2902
2903 13 buf += ret;
2904 13 buf_size -= ret;
2905 }
2906 13 s->sample_index += frame->nb_samples;
2907
2908 13 avpkt->pts = frame->pts;
2909 13 avpkt->size = buf - avpkt->data;
2910 13 avpkt->duration = ff_samples_to_time_base(avctx, frame->nb_samples);
2911 13 *got_packet_ptr = 1;
2912 13 return 0;
2913 }
2914
2915 2 static av_cold int wavpack_encode_close(AVCodecContext *avctx)
2916 {
2917 2 WavPackEncodeContext *s = avctx->priv_data;
2918 int i;
2919
2920
2/2
✓ Branch 0 taken 36 times.
✓ Branch 1 taken 2 times.
38 for (i = 0; i < MAX_TERMS + 2; i++) {
2921 36 av_freep(&s->sampleptrs[i][0]);
2922 36 av_freep(&s->sampleptrs[i][1]);
2923 36 s->sampleptrs_size[i][0] = s->sampleptrs_size[i][1] = 0;
2924 }
2925
2926
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 2 times.
6 for (i = 0; i < 2; i++) {
2927 4 av_freep(&s->samples[i]);
2928 4 s->samples_size[i] = 0;
2929
2930 4 av_freep(&s->best_buffer[i]);
2931 4 s->best_buffer_size[i] = 0;
2932
2933 4 av_freep(&s->temp_buffer[i][0]);
2934 4 av_freep(&s->temp_buffer[i][1]);
2935 4 s->temp_buffer_size[i][0] = s->temp_buffer_size[i][1] = 0;
2936 }
2937
2938 2 av_freep(&s->js_left);
2939 2 av_freep(&s->js_right);
2940 2 s->js_left_size = s->js_right_size = 0;
2941
2942 2 av_freep(&s->orig_l);
2943 2 av_freep(&s->orig_r);
2944 2 s->orig_l_size = s->orig_r_size = 0;
2945
2946 2 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