avfilter/af_afir: use segment pointer

1 files changed, 20 insertions, 19 deletions

diff --git a/libavfilter/af_afir.c b/libavfilter/af_afir.c
index dc32b92dc7..76ef21098a 100644
--- a/libavfilter/af_afir.c
+++ b/libavfilter/af_afir.c
@@ -59,42 +59,43 @@ static void fcmul_add_c(float *sum, const float *t, const float *c, ptrdiff_t le
 static int fir_channel(AVFilterContext *ctx, void *arg, int ch, int nb_jobs)
 {
     AudioFIRContext *s = ctx->priv;
+    AudioFIRSegment *seg = &s->seg;
     const float *src = (const float *)s->in[0]->extended_data[ch];
-    float *sum = (float *)s->seg.sum->extended_data[ch];
+    float *sum = (float *)seg->sum->extended_data[ch];
     AVFrame *out = arg;
     float *block, *dst, *ptr;
     int n, i, j;
 
-    memset(sum, 0, sizeof(*sum) * s->seg.fft_length);
-    block = (float *)s->seg.block->extended_data[ch] + s->seg.part_index * s->seg.block_size;
-    memset(block, 0, sizeof(*block) * s->seg.fft_length);
+    memset(sum, 0, sizeof(*sum) * seg->fft_length);
+    block = (float *)seg->block->extended_data[ch] + seg->part_index * seg->block_size;
+    memset(block, 0, sizeof(*block) * seg->fft_length);
 
     s->fdsp->vector_fmul_scalar(block, src, s->dry_gain, FFALIGN(out->nb_samples, 4));
     emms_c();
 
-    av_rdft_calc(s->seg.rdft[ch], block);
-    block[2 * s->seg.part_size] = block[1];
+    av_rdft_calc(seg->rdft[ch], block);
+    block[2 * seg->part_size] = block[1];
     block[1] = 0;
 
-    j = s->seg.part_index;
+    j = seg->part_index;
 
-    for (i = 0; i < s->seg.nb_partitions; i++) {
-        const int coffset = i * s->seg.coeff_size;
-        const float *block = (const float *)s->seg.block->extended_data[ch] + j * s->seg.block_size;
-        const FFTComplex *coeff = s->seg.coeff[ch * !s->one2many] + coffset;
+    for (i = 0; i < seg->nb_partitions; i++) {
+        const int coffset = i * seg->coeff_size;
+        const float *block = (const float *)seg->block->extended_data[ch] + j * seg->block_size;
+        const FFTComplex *coeff = seg->coeff[ch * !s->one2many] + coffset;
 
-        s->fcmul_add(sum, block, (const float *)coeff, s->seg.part_size);
+        s->fcmul_add(sum, block, (const float *)coeff, seg->part_size);
 
         if (j == 0)
-            j = s->seg.nb_partitions;
+            j = seg->nb_partitions;
         j--;
     }
 
-    sum[1] = sum[2 * s->seg.part_size];
-    av_rdft_calc(s->seg.irdft[ch], sum);
+    sum[1] = sum[2 * seg->part_size];
+    av_rdft_calc(seg->irdft[ch], sum);
 
-    dst = (float *)s->seg.buffer->extended_data[ch];
-    for (n = 0; n < s->seg.part_size; n++) {
+    dst = (float *)seg->buffer->extended_data[ch];
+    for (n = 0; n < seg->part_size; n++) {
         dst[n] += sum[n];
     }
 
@@ -102,8 +103,8 @@ static int fir_channel(AVFilterContext *ctx, void *arg, int ch, int nb_jobs)
     s->fdsp->vector_fmul_scalar(ptr, dst, s->wet_gain, FFALIGN(out->nb_samples, 4));
     emms_c();
 
-    dst = (float *)s->seg.buffer->extended_data[ch];
-    memcpy(dst, sum + s->seg.part_size, s->seg.part_size * sizeof(*dst));
+    dst = (float *)seg->buffer->extended_data[ch];
+    memcpy(dst, sum + seg->part_size, seg->part_size * sizeof(*dst));
 
     return 0;
 }