1 /*
2 * Copyright (c) 2017 Paul B Mahol
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
31
35 };
36
47 };
48
59 };
60
63
66
79
82
85
92
95
98
109
119
125
129 float c_re, float c_im,
130 float mag_totall, float mag_totalr,
131 float fl_phase, float fr_phase,
132 float bl_phase, float br_phase,
133 float sl_phase, float sr_phase,
134 float xl, float yl,
135 float xr, float yr,
136 int n);
138 float c_re, float c_im,
139 float lfe_re, float lfe_im,
140 float mag_totall, float mag_totalr,
141 float fl_phase, float fr_phase,
142 float bl_phase, float br_phase,
143 float sl_phase, float sr_phase,
144 float xl, float yl,
145 float xr, float yr,
146 int n);
148
152 {
156 };
157
161
165
170
174
179
183
184 return 0;
185 }
186
188 {
190
191 for (
int ch = 0; ch <
s->nb_in_channels &&
s->level_in >= 0.f; ch++)
192 s->input_levels[ch] =
s->level_in;
194
197 if (ch >= 0)
198 s->input_levels[ch] =
s->f_i[n];
199 }
200 }
201
203 {
205
206 for (
int ch = 0; ch <
s->nb_out_channels &&
s->level_out >= 0.f; ch++)
207 s->output_levels[ch] =
s->level_out;
209
212 if (ch >= 0)
213 s->output_levels[ch] =
s->f_o[n];
214 }
215 }
216
218 {
222
226 s->nb_in_channels =
inlink->ch_layout.nb_channels;
227
228 for (
int ch = 0; ch <
inlink->ch_layout.nb_channels; ch++) {
230
232 0,
s->win_size, &
scale, 0);
235 }
236
238 if (!
s->input_levels)
240
242
246
250
254
255 s->lowcut = 1.f *
s->lowcutf / (
inlink->sample_rate * 0.5) * (
s->win_size / 2);
256 s->highcut = 1.f *
s->highcutf / (
inlink->sample_rate * 0.5) * (
s->win_size / 2);
257
258 return 0;
259 }
260
262 {
266
271
273 float iscale = 1.f;
274
276 1,
s->win_size, &iscale, 0);
279 }
280
282 if (!
s->output_levels)
284
286
294 if (!
s->overlap_buffer || !
s->output || !
s->output_out || !
s->output_mag ||
295 !
s->output_ph || !
s->factors || !
s->sfactors)
297
298 s->rdft_size =
s->win_size / 2 + 1;
299
302 s->l_phase =
av_calloc(
s->rdft_size,
sizeof(*
s->l_phase));
303 s->r_phase =
av_calloc(
s->rdft_size,
sizeof(*
s->r_phase));
305 s->c_phase =
av_calloc(
s->rdft_size,
sizeof(*
s->c_phase));
306 s->mag_total =
av_calloc(
s->rdft_size,
sizeof(*
s->mag_total));
307 s->lfe_mag =
av_calloc(
s->rdft_size,
sizeof(*
s->lfe_mag));
308 s->lfe_phase =
av_calloc(
s->rdft_size,
sizeof(*
s->lfe_phase));
309 if (!
s->x_pos || !
s->y_pos || !
s->l_phase || !
s->r_phase || !
s->lfe_phase ||
310 !
s->c_phase || !
s->mag_total || !
s->lfe_mag || !
s->c_mag)
312
313 return 0;
314 }
315
317 {
318 return x * x;
319 }
320
322 {
324 }
325
326 #define MIN_MAG_SUM 0.00000001f
327
329 {
330 float reference,
r,
a;
331
333 return;
334
335 reference = angle *
M_PIf / 180.f;
338
340
343 else
345
347
350 }
351
353 {
355
357 return;
358
362 r = focus > 0.f ? 1.f -
powf(1.
f -
r, 1.
f + focus * 20.
f) :
powf(
r, 1.
f - focus * 20.
f);
366 }
367
369 {
374 }
375
376 static inline void get_lfe(
int output_lfe,
int n,
float lowcut,
float highcut,
377 float *lfe_mag, float c_mag, float *mag_total, int lfe_mode)
378 {
379 if (output_lfe && n < highcut) {
380 *lfe_mag = n < lowcut ? 1.f : .5f*(1.f+
cosf(
M_PIf*(lowcut-n)/(lowcut-highcut)));
381 *lfe_mag *= c_mag;
382 if (lfe_mode)
383 *mag_total -= *lfe_mag;
384 } else {
385 *lfe_mag = 0.f;
386 }
387 }
388
390 dst[2 * n ] = mag * cosf(ph); \
391 dst[2 * n + 1] = mag * sinf(ph);
392
394 {
396 float *
factor = (
float *)
s->factors->extended_data[ch];
397 const float f_x =
s->f_x[
sc_map[chan >= 0 ? chan : 0]];
398 const float f_y =
s->f_y[
sc_map[chan >= 0 ? chan : 0]];
399 const int rdft_size =
s->rdft_size;
400 const float *x =
s->x_pos;
401 const float *y =
s->y_pos;
402
403 switch (chan) {
405 for (int n = 0; n < rdft_size; n++)
407 break;
409 for (int n = 0; n < rdft_size; n++)
411 break;
413 for (int n = 0; n < rdft_size; n++)
415 break;
417 for (int n = 0; n < rdft_size; n++)
419 break;
421 for (int n = 0; n < rdft_size; n++)
423 break;
425 for (int n = 0; n < rdft_size; n++)
427 break;
429 for (int n = 0; n < rdft_size; n++)
431 break;
433 for (int n = 0; n < rdft_size; n++)
435 break;
437 for (int n = 0; n < rdft_size; n++)
439 break;
440 default:
441 for (int n = 0; n < rdft_size; n++)
443 break;
444 }
445 }
446
448 {
450 float *sfactor = (
float *)
s->sfactors->extended_data[ch];
451 float *
factor = (
float *)
s->factors->extended_data[ch];
452 float *omag = (
float *)
s->output_mag->extended_data[ch];
453 float *oph = (
float *)
s->output_ph->extended_data[ch];
454 float *
dst = (
float *)
s->output->extended_data[ch];
455 const int rdft_size =
s->rdft_size;
456 const float smooth =
s->smooth;
457
459 for (int n = 0; n < rdft_size; n++)
461
463 }
464
465 for (int n = 0; n < rdft_size; n++)
467
468 for (int n = 0; n < rdft_size; n++) {
469 const float mag = omag[n];
470 const float ph = oph[n];
471
473 }
474 }
475
477 {
479 float *omag = (
float *)
s->output_mag->extended_data[ch];
480 float *oph = (
float *)
s->output_ph->extended_data[ch];
481 const float *mag_total =
s->mag_total;
482 const int rdft_size =
s->rdft_size;
483 const float *c_phase =
s->c_phase;
484 const float *l_phase =
s->l_phase;
485 const float *r_phase =
s->r_phase;
486 const float *lfe_mag =
s->lfe_mag;
487 const float *c_mag =
s->c_mag;
488
489 switch (chan) {
491 memcpy(omag, c_mag, rdft_size * sizeof(*omag));
492 break;
494 memcpy(omag, lfe_mag, rdft_size * sizeof(*omag));
495 break;
503 memcpy(omag, mag_total, rdft_size * sizeof(*omag));
504 break;
505 default:
506 break;
507 }
508
509 switch (chan) {
513 memcpy(oph, c_phase, rdft_size * sizeof(*oph));
514 break;
518 memcpy(oph, l_phase, rdft_size * sizeof(*oph));
519 break;
523 memcpy(oph, r_phase, rdft_size * sizeof(*oph));
524 break;
525 default:
526 break;
527 }
528 }
529
531 {
534
536
538
540 }
541
543 {
546 float *omag = (
float *)
s->output_mag->extended_data[ch];
547 float *oph = (
float *)
s->output_ph->extended_data[ch];
548 const float *mag_total =
s->mag_total;
549 const float *lfe_phase =
s->lfe_phase;
550 const int rdft_size =
s->rdft_size;
551 const float *c_phase =
s->c_phase;
552 const float *l_phase =
s->l_phase;
553 const float *r_phase =
s->r_phase;
554 const float *lfe_mag =
s->lfe_mag;
555 const float *c_mag =
s->c_mag;
556
557 switch (chan) {
560 break;
561 default:
563 break;
564 }
565
566 switch (chan) {
568 memcpy(omag, c_mag, rdft_size * sizeof(*omag));
569 break;
571 memcpy(omag, lfe_mag, rdft_size * sizeof(*omag));
572 break;
580 memcpy(omag, mag_total, rdft_size * sizeof(*omag));
581 break;
582 default:
583 break;
584 }
585
586 switch (chan) {
588 memcpy(oph, lfe_phase, rdft_size * sizeof(*oph));
589 break;
592 memcpy(oph, c_phase, rdft_size * sizeof(*oph));
593 break;
597 memcpy(oph, l_phase, rdft_size * sizeof(*oph));
598 break;
602 memcpy(oph, r_phase, rdft_size * sizeof(*oph));
603 break;
604 default:
605 break;
606 }
607
609 }
610
612 {
615
616 switch (chan) {
619 break;
620 default:
622 break;
623 }
624
626
628 }
629
631 float c_re, float c_im,
632 float mag_totall, float mag_totalr,
633 float fl_phase, float fr_phase,
634 float bl_phase, float br_phase,
635 float sl_phase, float sr_phase,
636 float xl, float yl,
637 float xr, float yr,
638 int n)
639 {
640 float fl_mag, fr_mag, ls_mag, rs_mag, lb_mag, rb_mag;
641 float *dstc, *dstl, *dstr, *dstls, *dstrs, *dstlb, *dstrb, *dstlfe;
642 float lfe_mag, c_phase, mag_total = (mag_totall + mag_totalr) * 0.5
f;
644
645 dstl = (
float *)
s->output->extended_data[0];
646 dstr = (
float *)
s->output->extended_data[1];
647 dstc = (
float *)
s->output->extended_data[2];
648 dstlfe = (
float *)
s->output->extended_data[3];
649 dstlb = (
float *)
s->output->extended_data[4];
650 dstrb = (
float *)
s->output->extended_data[5];
651 dstls = (
float *)
s->output->extended_data[6];
652 dstrs = (
float *)
s->output->extended_data[7];
653
654 c_phase =
atan2f(c_im, c_re);
655
656 get_lfe(
s->output_lfe, n,
s->lowcut,
s->highcut, &lfe_mag, hypotf(c_re, c_im), &mag_total,
s->lfe_mode);
657
664
665 dstl[2 * n ] = fl_mag *
cosf(fl_phase);
666 dstl[2 * n + 1] = fl_mag *
sinf(fl_phase);
667
668 dstr[2 * n ] = fr_mag *
cosf(fr_phase);
669 dstr[2 * n + 1] = fr_mag *
sinf(fr_phase);
670
671 dstc[2 * n ] = c_re;
672 dstc[2 * n + 1] = c_im;
673
674 dstlfe[2 * n ] = lfe_mag *
cosf(c_phase);
675 dstlfe[2 * n + 1] = lfe_mag *
sinf(c_phase);
676
677 dstlb[2 * n ] = lb_mag *
cosf(bl_phase);
678 dstlb[2 * n + 1] = lb_mag *
sinf(bl_phase);
679
680 dstrb[2 * n ] = rb_mag *
cosf(br_phase);
681 dstrb[2 * n + 1] = rb_mag *
sinf(br_phase);
682
683 dstls[2 * n ] = ls_mag *
cosf(sl_phase);
684 dstls[2 * n + 1] = ls_mag *
sinf(sl_phase);
685
686 dstrs[2 * n ] = rs_mag *
cosf(sr_phase);
687 dstrs[2 * n + 1] = rs_mag *
sinf(sr_phase);
688 }
689
691 float c_re, float c_im,
692 float lfe_re, float lfe_im,
693 float mag_totall, float mag_totalr,
694 float fl_phase, float fr_phase,
695 float bl_phase, float br_phase,
696 float sl_phase, float sr_phase,
697 float xl, float yl,
698 float xr, float yr,
699 int n)
700 {
701 float fl_mag, fr_mag, ls_mag, rs_mag, lb_mag, rb_mag;
702 float *dstc, *dstl, *dstr, *dstls, *dstrs, *dstlb, *dstrb, *dstlfe;
704
705 dstl = (
float *)
s->output->extended_data[0];
706 dstr = (
float *)
s->output->extended_data[1];
707 dstc = (
float *)
s->output->extended_data[2];
708 dstlfe = (
float *)
s->output->extended_data[3];
709 dstlb = (
float *)
s->output->extended_data[4];
710 dstrb = (
float *)
s->output->extended_data[5];
711 dstls = (
float *)
s->output->extended_data[6];
712 dstrs = (
float *)
s->output->extended_data[7];
713
720
721 dstl[2 * n ] = fl_mag *
cosf(fl_phase);
722 dstl[2 * n + 1] = fl_mag *
sinf(fl_phase);
723
724 dstr[2 * n ] = fr_mag *
cosf(fr_phase);
725 dstr[2 * n + 1] = fr_mag *
sinf(fr_phase);
726
727 dstc[2 * n ] = c_re;
728 dstc[2 * n + 1] = c_im;
729
730 dstlfe[2 * n ] = lfe_re;
731 dstlfe[2 * n + 1] = lfe_im;
732
733 dstlb[2 * n ] = lb_mag *
cosf(bl_phase);
734 dstlb[2 * n + 1] = lb_mag *
sinf(bl_phase);
735
736 dstrb[2 * n ] = rb_mag *
cosf(br_phase);
737 dstrb[2 * n + 1] = rb_mag *
sinf(br_phase);
738
739 dstls[2 * n ] = ls_mag *
cosf(sl_phase);
740 dstls[2 * n + 1] = ls_mag *
sinf(sl_phase);
741
742 dstrs[2 * n ] = rs_mag *
cosf(sr_phase);
743 dstrs[2 * n + 1] = rs_mag *
sinf(sr_phase);
744 }
745
747 {
749 const float *srcl = (
const float *)
s->input->extended_data[0];
750 const float *srcr = (
const float *)
s->input->extended_data[1];
751 const int output_lfe =
s->output_lfe &&
s->create_lfe;
752 const int rdft_size =
s->rdft_size;
753 const int lfe_mode =
s->lfe_mode;
754 const float highcut =
s->highcut;
755 const float lowcut =
s->lowcut;
756 const float angle =
s->angle;
757 const float focus =
s->focus;
758 float *magtotal =
s->mag_total;
759 float *lfemag =
s->lfe_mag;
760 float *lphase =
s->l_phase;
761 float *rphase =
s->r_phase;
762 float *cphase =
s->c_phase;
763 float *cmag =
s->c_mag;
764 float *xpos =
s->x_pos;
765 float *ypos =
s->y_pos;
766
767 for (int n = 0; n < rdft_size; n++) {
768 float l_re = srcl[2 * n], r_re = srcr[2 * n];
769 float l_im = srcl[2 * n + 1], r_im = srcr[2 * n + 1];
770 float c_phase =
atan2f(l_im + r_im, l_re + r_re);
771 float l_mag = hypotf(l_re, l_im);
772 float r_mag = hypotf(r_re, r_im);
773 float mag_total = hypotf(l_mag, r_mag);
774 float l_phase =
atan2f(l_im, l_re);
775 float r_phase =
atan2f(r_im, r_re);
776 float phase_dif =
fabsf(l_phase - r_phase);
777 float mag_sum = l_mag + r_mag;
778 float c_mag = mag_sum * 0.5f;
779 float mag_dif, x, y;
780
782 mag_dif = (l_mag - r_mag) / mag_sum;
783 if (phase_dif >
M_PIf)
784 phase_dif = 2.f *
M_PIf - phase_dif;
785
789 get_lfe(output_lfe, n, lowcut, highcut, &lfemag[n], c_mag, &mag_total, lfe_mode);
790
791 xpos[n] = x;
792 ypos[n] = y;
793 lphase[n] = l_phase;
794 rphase[n] = r_phase;
795 cmag[n] = c_mag;
796 cphase[n] = c_phase;
797 magtotal[n] = mag_total;
798 }
799 }
800
802 {
804 const float *srcl = (
const float *)
s->input->extended_data[0];
805 const float *srcr = (
const float *)
s->input->extended_data[1];
806 const float *srclfe = (
const float *)
s->input->extended_data[2];
807 const int rdft_size =
s->rdft_size;
808 const float angle =
s->angle;
809 const float focus =
s->focus;
810 float *magtotal =
s->mag_total;
811 float *lfephase =
s->lfe_phase;
812 float *lfemag =
s->lfe_mag;
813 float *lphase =
s->l_phase;
814 float *rphase =
s->r_phase;
815 float *cphase =
s->c_phase;
816 float *cmag =
s->c_mag;
817 float *xpos =
s->x_pos;
818 float *ypos =
s->y_pos;
819
820 for (
int n = 0; n < rdft_size; n++) {
821 float l_re = srcl[2 * n], r_re = srcr[2 * n];
822 float l_im = srcl[2 * n + 1], r_im = srcr[2 * n + 1];
823 float lfe_re = srclfe[2 * n], lfe_im = srclfe[2 * n + 1];
824 float c_phase =
atan2f(l_im + r_im, l_re + r_re);
825 float l_mag = hypotf(l_re, l_im);
826 float r_mag = hypotf(r_re, r_im);
827 float lfe_mag = hypotf(lfe_re, lfe_im);
828 float lfe_phase =
atan2f(lfe_im, lfe_re);
829 float mag_total = hypotf(l_mag, r_mag);
830 float l_phase =
atan2f(l_im, l_re);
831 float r_phase =
atan2f(r_im, r_re);
832 float phase_dif =
fabsf(l_phase - r_phase);
833 float mag_sum = l_mag + r_mag;
834 float c_mag = mag_sum * 0.5f;
835 float mag_dif, x, y;
836
838 mag_dif = (l_mag - r_mag) / mag_sum;
839 if (phase_dif >
M_PIf)
840 phase_dif = 2.f *
M_PIf - phase_dif;
841
845
846 xpos[n] = x;
847 ypos[n] = y;
848 lphase[n] = l_phase;
849 rphase[n] = r_phase;
850 cmag[n] = c_mag;
851 cphase[n] = c_phase;
852 lfemag[n] = lfe_mag;
853 lfephase[n] = lfe_phase;
854 magtotal[n] = mag_total;
855 }
856 }
857
859 {
861 const float *srcl = (
const float *)
s->input->extended_data[0];
862 const float *srcr = (
const float *)
s->input->extended_data[1];
863 const float *srcc = (
const float *)
s->input->extended_data[2];
864 const int output_lfe =
s->output_lfe &&
s->create_lfe;
865 const int rdft_size =
s->rdft_size;
866 const int lfe_mode =
s->lfe_mode;
867 const float highcut =
s->highcut;
868 const float lowcut =
s->lowcut;
869 const float angle =
s->angle;
870 const float focus =
s->focus;
871 float *magtotal =
s->mag_total;
872 float *lfemag =
s->lfe_mag;
873 float *lphase =
s->l_phase;
874 float *rphase =
s->r_phase;
875 float *cphase =
s->c_phase;
876 float *cmag =
s->c_mag;
877 float *xpos =
s->x_pos;
878 float *ypos =
s->y_pos;
879
880 for (
int n = 0; n < rdft_size; n++) {
881 float l_re = srcl[2 * n], r_re = srcr[2 * n];
882 float l_im = srcl[2 * n + 1], r_im = srcr[2 * n + 1];
883 float c_re = srcc[2 * n], c_im = srcc[2 * n + 1];
884 float c_phase =
atan2f(c_im, c_re);
885 float c_mag = hypotf(c_re, c_im);
886 float l_mag = hypotf(l_re, l_im);
887 float r_mag = hypotf(r_re, r_im);
888 float mag_total = hypotf(l_mag, r_mag);
889 float l_phase =
atan2f(l_im, l_re);
890 float r_phase =
atan2f(r_im, r_re);
891 float phase_dif =
fabsf(l_phase - r_phase);
892 float mag_sum = l_mag + r_mag;
893 float mag_dif, x, y;
894
896 mag_dif = (l_mag - r_mag) / mag_sum;
897 if (phase_dif >
M_PIf)
898 phase_dif = 2.f *
M_PIf - phase_dif;
899
903 get_lfe(output_lfe, n, lowcut, highcut, &lfemag[n], c_mag, &mag_total, lfe_mode);
904
905 xpos[n] = x;
906 ypos[n] = y;
907 lphase[n] = l_phase;
908 rphase[n] = r_phase;
909 cmag[n] = c_mag;
910 cphase[n] = c_phase;
911 magtotal[n] = mag_total;
912 }
913 }
914
916 {
918 const int rdft_size =
s->rdft_size;
919 float *srcl, *srcr, *srcc, *srcsl, *srcsr;
920 int n;
921
922 srcl = (
float *)
s->input->extended_data[0];
923 srcr = (
float *)
s->input->extended_data[1];
924 srcc = (
float *)
s->input->extended_data[2];
925 srcsl = (
float *)
s->input->extended_data[3];
926 srcsr = (
float *)
s->input->extended_data[4];
927
928 for (n = 0; n < rdft_size; n++) {
929 float fl_re = srcl[2 * n], fr_re = srcr[2 * n];
930 float fl_im = srcl[2 * n + 1], fr_im = srcr[2 * n + 1];
931 float c_re = srcc[2 * n], c_im = srcc[2 * n + 1];
932 float sl_re = srcsl[2 * n], sl_im = srcsl[2 * n + 1];
933 float sr_re = srcsr[2 * n], sr_im = srcsr[2 * n + 1];
934 float fl_mag = hypotf(fl_re, fl_im);
935 float fr_mag = hypotf(fr_re, fr_im);
936 float fl_phase =
atan2f(fl_im, fl_re);
937 float fr_phase =
atan2f(fr_im, fr_re);
938 float sl_mag = hypotf(sl_re, sl_im);
939 float sr_mag = hypotf(sr_re, sr_im);
940 float sl_phase =
atan2f(sl_im, sl_re);
941 float sr_phase =
atan2f(sr_im, sr_re);
942 float phase_difl =
fabsf(fl_phase - sl_phase);
943 float phase_difr =
fabsf(fr_phase - sr_phase);
944 float magl_sum = fl_mag + sl_mag;
945 float magr_sum = fr_mag + sr_mag;
948 float mag_totall = hypotf(fl_mag, sl_mag);
949 float mag_totalr = hypotf(fr_mag, sr_mag);
950 float bl_phase =
atan2f(fl_im + sl_im, fl_re + sl_re);
951 float br_phase =
atan2f(fr_im + sr_im, fr_re + sr_re);
952 float xl, yl;
953 float xr, yr;
954
955 if (phase_difl >
M_PIf)
956 phase_difl = 2.f *
M_PIf - phase_difl;
957
958 if (phase_difr >
M_PIf)
959 phase_difr = 2.f *
M_PIf - phase_difr;
960
963
964 s->upmix_5_0(
ctx, c_re, c_im,
965 mag_totall, mag_totalr,
966 fl_phase, fr_phase,
967 bl_phase, br_phase,
968 sl_phase, sr_phase,
969 xl, yl, xr, yr, n);
970 }
971 }
972
974 {
976 const int rdft_size =
s->rdft_size;
977 float *srcl, *srcr, *srcc, *srclfe, *srcsl, *srcsr;
978 int n;
979
980 srcl = (
float *)
s->input->extended_data[0];
981 srcr = (
float *)
s->input->extended_data[1];
982 srcc = (
float *)
s->input->extended_data[2];
983 srclfe = (
float *)
s->input->extended_data[3];
984 srcsl = (
float *)
s->input->extended_data[4];
985 srcsr = (
float *)
s->input->extended_data[5];
986
987 for (n = 0; n < rdft_size; n++) {
988 float fl_re = srcl[2 * n], fr_re = srcr[2 * n];
989 float fl_im = srcl[2 * n + 1], fr_im = srcr[2 * n + 1];
990 float c_re = srcc[2 * n], c_im = srcc[2 * n + 1];
991 float lfe_re = srclfe[2 * n], lfe_im = srclfe[2 * n + 1];
992 float sl_re = srcsl[2 * n], sl_im = srcsl[2 * n + 1];
993 float sr_re = srcsr[2 * n], sr_im = srcsr[2 * n + 1];
994 float fl_mag = hypotf(fl_re, fl_im);
995 float fr_mag = hypotf(fr_re, fr_im);
996 float fl_phase =
atan2f(fl_im, fl_re);
997 float fr_phase =
atan2f(fr_im, fr_re);
998 float sl_mag = hypotf(sl_re, sl_im);
999 float sr_mag = hypotf(sr_re, sr_im);
1000 float sl_phase =
atan2f(sl_im, sl_re);
1001 float sr_phase =
atan2f(sr_im, sr_re);
1002 float phase_difl =
fabsf(fl_phase - sl_phase);
1003 float phase_difr =
fabsf(fr_phase - sr_phase);
1004 float magl_sum = fl_mag + sl_mag;
1005 float magr_sum = fr_mag + sr_mag;
1006 float mag_difl = magl_sum <
MIN_MAG_SUM ?
FFDIFFSIGN(fl_mag, sl_mag) : (fl_mag - sl_mag) / magl_sum;
1007 float mag_difr = magr_sum <
MIN_MAG_SUM ?
FFDIFFSIGN(fr_mag, sr_mag) : (fr_mag - sr_mag) / magr_sum;
1008 float mag_totall = hypotf(fl_mag, sl_mag);
1009 float mag_totalr = hypotf(fr_mag, sr_mag);
1010 float bl_phase =
atan2f(fl_im + sl_im, fl_re + sl_re);
1011 float br_phase =
atan2f(fr_im + sr_im, fr_re + sr_re);
1012 float xl, yl;
1013 float xr, yr;
1014
1015 if (phase_difl >
M_PIf)
1016 phase_difl = 2.f *
M_PIf - phase_difl;
1017
1018 if (phase_difr >
M_PIf)
1019 phase_difr = 2.f *
M_PIf - phase_difr;
1020
1023
1024 s->upmix_5_1(
ctx, c_re, c_im, lfe_re, lfe_im,
1025 mag_totall, mag_totalr,
1026 fl_phase, fr_phase,
1027 bl_phase, br_phase,
1028 sl_phase, sr_phase,
1029 xl, yl, xr, yr, n);
1030 }
1031 }
1032
1034 {
1036 const int rdft_size =
s->rdft_size;
1037 float *srcl, *srcr, *srcc, *srclfe, *srcbl, *srcbr;
1038 int n;
1039
1040 srcl = (
float *)
s->input->extended_data[0];
1041 srcr = (
float *)
s->input->extended_data[1];
1042 srcc = (
float *)
s->input->extended_data[2];
1043 srclfe = (
float *)
s->input->extended_data[3];
1044 srcbl = (
float *)
s->input->extended_data[4];
1045 srcbr = (
float *)
s->input->extended_data[5];
1046
1047 for (n = 0; n < rdft_size; n++) {
1048 float fl_re = srcl[2 * n], fr_re = srcr[2 * n];
1049 float fl_im = srcl[2 * n + 1], fr_im = srcr[2 * n + 1];
1050 float c_re = srcc[2 * n], c_im = srcc[2 * n + 1];
1051 float lfe_re = srclfe[2 * n], lfe_im = srclfe[2 * n + 1];
1052 float bl_re = srcbl[2 * n], bl_im = srcbl[2 * n + 1];
1053 float br_re = srcbr[2 * n], br_im = srcbr[2 * n + 1];
1054 float fl_mag = hypotf(fl_re, fl_im);
1055 float fr_mag = hypotf(fr_re, fr_im);
1056 float fl_phase =
atan2f(fl_im, fl_re);
1057 float fr_phase =
atan2f(fr_im, fr_re);
1058 float bl_mag = hypotf(bl_re, bl_im);
1059 float br_mag = hypotf(br_re, br_im);
1060 float bl_phase =
atan2f(bl_im, bl_re);
1061 float br_phase =
atan2f(br_im, br_re);
1062 float phase_difl =
fabsf(fl_phase - bl_phase);
1063 float phase_difr =
fabsf(fr_phase - br_phase);
1064 float magl_sum = fl_mag + bl_mag;
1065 float magr_sum = fr_mag + br_mag;
1066 float mag_difl = magl_sum <
MIN_MAG_SUM ?
FFDIFFSIGN(fl_mag, bl_mag) : (fl_mag - bl_mag) / magl_sum;
1067 float mag_difr = magr_sum <
MIN_MAG_SUM ?
FFDIFFSIGN(fr_mag, br_mag) : (fr_mag - br_mag) / magr_sum;
1068 float mag_totall = hypotf(fl_mag, bl_mag);
1069 float mag_totalr = hypotf(fr_mag, br_mag);
1070 float sl_phase =
atan2f(fl_im + bl_im, fl_re + bl_re);
1071 float sr_phase =
atan2f(fr_im + br_im, fr_re + br_re);
1072 float xl, yl;
1073 float xr, yr;
1074
1075 if (phase_difl >
M_PIf)
1076 phase_difl = 2.f *
M_PIf - phase_difl;
1077
1078 if (phase_difr >
M_PIf)
1079 phase_difr = 2.f *
M_PIf - phase_difr;
1080
1083
1084 s->upmix_5_1(
ctx, c_re, c_im, lfe_re, lfe_im,
1085 mag_totall, mag_totalr,
1086 fl_phase, fr_phase,
1087 bl_phase, br_phase,
1088 sl_phase, sr_phase,
1089 xl, yl, xr, yr, n);
1090 }
1091 }
1092
1094 {
1096
1097 if (
s->all_x >= 0.f)
1098 for (
int n = 0; n <
SC_NB; n++)
1099 s->f_x[n] =
s->all_x;
1101 if (
s->all_y >= 0.f)
1102 for (
int n = 0; n <
SC_NB; n++)
1103 s->f_y[n] =
s->all_y;
1105 }
1106
1108 {
1110 int64_t in_channel_layout, out_channel_layout;
1111 char in_name[128], out_name[128];
1112 float overlap;
1113
1114 if (
s->lowcutf >=
s->highcutf) {
1116 s->lowcutf,
s->highcutf);
1118 }
1119
1121 s->in_ch_layout.u.mask : 0;
1123 s->out_ch_layout.u.mask : 0;
1124
1127
1128 switch (out_channel_layout) {
1148 break;
1149 default:
1151 }
1152
1153 switch (in_channel_layout) {
1157 break;
1161 break;
1165 break;
1168 switch (out_channel_layout) {
1171 break;
1172 default:
1174 }
1175 break;
1178 switch (out_channel_layout) {
1181 break;
1182 default:
1184 }
1185 break;
1188 switch (out_channel_layout) {
1191 break;
1192 default:
1194 }
1195 break;
1196 default:
1201 in_name, out_name);
1203 }
1204
1205 s->window_func_lut =
av_calloc(
s->win_size,
sizeof(*
s->window_func_lut));
1206 if (!
s->window_func_lut)
1208
1210 if (
s->overlap == 1)
1211 s->overlap = overlap;
1212
1213 for (
int i = 0;
i <
s->win_size;
i++)
1214 s->window_func_lut[
i] =
sqrtf(
s->window_func_lut[
i] /
s->win_size);
1215 s->hop_size =
FFMAX(1,
s->win_size * (1. -
s->overlap));
1216
1217 {
1218 float max = 0.f, *temp_lut =
av_calloc(
s->win_size,
sizeof(*temp_lut));
1219 if (!temp_lut)
1221
1222 for (
int j = 0; j <
s->win_size; j +=
s->hop_size) {
1223 for (
int i = 0;
i <
s->win_size;
i++)
1224 temp_lut[(
i + j) %
s->win_size] +=
s->window_func_lut[
i];
1225 }
1226
1227 for (
int i = 0;
i <
s->win_size;
i++)
1230
1231 s->win_gain = 1.f / (
max *
sqrtf(
s->win_size));
1232 }
1233
1235
1236 return 0;
1237 }
1238
1240 {
1242 float *
src = (
float *)
s->input_in->extended_data[ch];
1243 float *
win = (
float *)
s->window->extended_data[ch];
1244 const float *window_func_lut =
s->window_func_lut;
1245 const int offset =
s->win_size -
s->hop_size;
1246 const float level_in =
s->input_levels[ch];
1247 const int win_size =
s->win_size;
1248
1252
1253 for (int n = 0; n < win_size; n++)
1254 win[n] =
src[n] * window_func_lut[n] * level_in;
1255
1256 s->tx_fn(
s->rdft[ch], (
float *)
s->input->extended_data[ch],
win,
sizeof(
float));
1257
1258 return 0;
1259 }
1260
1262 {
1266
1267 for (int ch = start; ch < end; ch++)
1269
1270 return 0;
1271 }
1272
1274 {
1276 const float level_out =
s->output_levels[ch] *
s->win_gain;
1277 const float *window_func_lut =
s->window_func_lut;
1278 const int win_size =
s->win_size;
1280
1281 dst = (
float *)
s->output_out->extended_data[ch];
1282 ptr = (
float *)
s->overlap_buffer->extended_data[ch];
1284
1285 memmove(
s->overlap_buffer->extended_data[ch],
1286 s->overlap_buffer->extended_data[ch] +
s->hop_size *
sizeof(
float),
1287 s->win_size *
sizeof(
float));
1288 memset(
s->overlap_buffer->extended_data[ch] +
s->win_size *
sizeof(
float),
1289 0,
s->hop_size *
sizeof(
float));
1290
1291 for (int n = 0; n < win_size; n++)
1292 ptr[n] +=
dst[n] * window_func_lut[n] * level_out;
1293
1294 ptr = (
float *)
s->overlap_buffer->extended_data[ch];
1295 dst = (
float *)
out->extended_data[ch];
1296 memcpy(
dst, ptr,
s->hop_size *
sizeof(
float));
1297
1298 return 0;
1299 }
1300
1302 {
1305 const int start = (
out->ch_layout.nb_channels * jobnr) / nb_jobs;
1306 const int end = (
out->ch_layout.nb_channels * (jobnr+1)) / nb_jobs;
1307
1308 for (int ch = start; ch < end; ch++) {
1312 }
1313
1314 return 0;
1315 }
1316
1318 {
1323
1327
1329
1333
1337
1340
1343 }
1344
1346 {
1353
1355
1359
1364
1367 return 0;
1368 }
1369
1372 return 0;
1373 }
1374
1376
1378 }
1379
1381 {
1383
1394
1395 for (
int ch = 0; ch <
s->nb_in_channels; ch++)
1397 for (
int ch = 0; ch <
s->nb_out_channels; ch++)
1404
1414 }
1415
1417 char *res,
int res_len,
int flags)
1418 {
1421
1425
1426 s->hop_size =
FFMAX(1,
s->win_size * (1. -
s->overlap));
1427
1431
1432 return 0;
1433 }
1434
1435 #define OFFSET(x) offsetof(AudioSurroundContext, x)
1436 #define FLAGS AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
1437 #define TFLAGS AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
1438
1449 {
"sub",
"subtract LFE channel with others", 0,
AV_OPT_TYPE_CONST, {.i64=1}, 0, 1,
TFLAGS, .unit =
"lfe_mode" },
1493 };
1494
1496
1498 {
1502 },
1503 };
1504
1506 {
1510 },
1511 };
1512
1517 .priv_class = &surround_class,
1526 };