virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

avfilter/vf_convolve: unbreak non-power of 2 width&height filtering

Browse Source 
Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Paul B Mahol 2017-12-23 16:27:39 +01:00
parent b943bc343d
commit b5958ff82e
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
libavfilter/vf_convolve.c
View File

@ -112,11 +112,13 @@ static int config_input_main(AVFilterLink *inlink)
for (i = 0; i < s->nb_planes; i++) { for (i = 0; i < s->nb_planes; i++) {
int w = s->planewidth[i]; int w = s->planewidth[i];
int h = s->planeheight[i]; int h = s->planeheight[i];
int n = FFMAX(w, h) * 10/9; int n = FFMAX(w, h);
n += n / 2;
for (fft_bits = 1; 1 << fft_bits < n; fft_bits++); for (fft_bits = 1; 1 << fft_bits < n; fft_bits++);
s->fft_bits[i] = fft_bits; s->fft_bits[i] = fft_bits + 1;
s->fft_len[i] = 1 << s->fft_bits[i]; s->fft_len[i] = 1 << s->fft_bits[i];
if (!(s->fft_hdata[i] = av_calloc(s->fft_len[i], s->fft_len[i] * sizeof(FFTComplex)))) if (!(s->fft_hdata[i] = av_calloc(s->fft_len[i], s->fft_len[i] * sizeof(FFTComplex))))
@ -173,7 +175,20 @@ static void fft_horizontal(ConvolveContext *s, FFTComplex *fft_hdata,
fft_hdata[y * n + x].im = 0; fft_hdata[y * n + x].im = 0;
} }
} }
for (; x < n / 2; x++) {
fft_hdata[y * n + x].re = 0;
fft_hdata[y * n + x].im = 0;
}
for (; x < n; x++) { for (; x < n; x++) {
fft_hdata[y * n + x].re = fft_hdata[y * n + n - x - 1].re;
fft_hdata[y * n + x].im = 0;
}
}
for (; y < n / 2; y++) {
for (x = 0; x < n; x++) {
fft_hdata[y * n + x].re = 0; fft_hdata[y * n + x].re = 0;
fft_hdata[y * n + x].im = 0; fft_hdata[y * n + x].im = 0;
} }
@ -181,7 +196,7 @@ static void fft_horizontal(ConvolveContext *s, FFTComplex *fft_hdata,
for (; y < n; y++) { for (; y < n; y++) {
for (x = 0; x < n; x++) { for (x = 0; x < n; x++) {
fft_hdata[y * n + x].re = 0; fft_hdata[y * n + x].re = fft_hdata[(n - y - 1) * n + x].re;
fft_hdata[y * n + x].im = 0; fft_hdata[y * n + x].im = 0;
} }
} }
@ -202,10 +217,7 @@ static void fft_vertical(ConvolveContext *s, FFTComplex *fft_hdata, FFTComplex *
fft_vdata[y * n + x].re = fft_hdata[x * n + y].re; fft_vdata[y * n + x].re = fft_hdata[x * n + y].re;
fft_vdata[y * n + x].im = fft_hdata[x * n + y].im; fft_vdata[y * n + x].im = fft_hdata[x * n + y].im;
} }
for (; x < n; x++) {
fft_vdata[y * n + x].re = 0;
fft_vdata[y * n + x].im = 0;
}
av_fft_permute(s->fft[plane], fft_vdata + y * n); av_fft_permute(s->fft[plane], fft_vdata + y * n);
av_fft_calc(s->fft[plane], fft_vdata + y * n); av_fft_calc(s->fft[plane], fft_vdata + y * n);
} }
@ -218,6 +230,7 @@ static void ifft_vertical(ConvolveContext *s, int n, int plane)
for (y = 0; y < n; y++) { for (y = 0; y < n; y++) {
av_fft_permute(s->ifft[plane], s->fft_vdata[plane] + y * n); av_fft_permute(s->ifft[plane], s->fft_vdata[plane] + y * n);
av_fft_calc(s->ifft[plane], s->fft_vdata[plane] + y * n); av_fft_calc(s->ifft[plane], s->fft_vdata[plane] + y * n);
for (x = 0; x < n; x++) { for (x = 0; x < n; x++) {
s->fft_hdata[plane][x * n + y].re = s->fft_vdata[plane][y * n + x].re; s->fft_hdata[plane][x * n + y].re = s->fft_vdata[plane][y * n + x].re;
s->fft_hdata[plane][x * n + y].im = s->fft_vdata[plane][y * n + x].im; s->fft_hdata[plane][x * n + y].im = s->fft_vdata[plane][y * n + x].im;