libavutil/float_dsp.c
d5a7229b
 /*
b0415287
  * Copyright 2005 Balatoni Denes
  * Copyright 2006 Loren Merritt
d5a7229b
  *
b0415287
  * This file is part of FFmpeg.
  *
  * FFmpeg is free software; you can redistribute it and/or
d5a7229b
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
b0415287
  * FFmpeg is distributed in the hope that it will be useful,
d5a7229b
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
b0415287
  * License along with FFmpeg; if not, write to the Free Software
d5a7229b
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
 #include "config.h"
a0b901a3
 #include "attributes.h"
d5a7229b
 #include "float_dsp.h"
a54a51cc
 #include "mem.h"
d5a7229b
 
 static void vector_fmul_c(float *dst, const float *src0, const float *src1,
                           int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] = src0[i] * src1[i];
 }
 
bb16a062
 static void vector_dmul_c(double *dst, const double *src0, const double *src1,
                           int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] = src0[i] * src1[i];
 }
 
cb5042d0
 static void vector_fmac_scalar_c(float *dst, const float *src, float mul,
                                  int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] += src[i] * mul;
 }
 
4dc2dd80
 static void vector_dmac_scalar_c(double *dst, const double *src, double mul,
                                  int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] += src[i] * mul;
 }
 
284ea790
 static void vector_fmul_scalar_c(float *dst, const float *src, float mul,
                                  int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] = src[i] * mul;
 }
 
ac7eb4cb
 static void vector_dmul_scalar_c(double *dst, const double *src, double mul,
                                  int len)
 {
     int i;
     for (i = 0; i < len; i++)
         dst[i] = src[i] * mul;
 }
 
e034cc6c
 static void vector_fmul_window_c(float *dst, const float *src0,
                                  const float *src1, const float *win, int len)
 {
     int i, j;
 
     dst  += len;
     win  += len;
     src0 += len;
 
     for (i = -len, j = len - 1; i < 0; i++, j--) {
         float s0 = src0[i];
         float s1 = src1[j];
         float wi = win[i];
         float wj = win[j];
         dst[i] = s0 * wj - s1 * wi;
         dst[j] = s0 * wi + s1 * wj;
     }
 }
 
55aa03b9
 static void vector_fmul_add_c(float *dst, const float *src0, const float *src1,
                               const float *src2, int len){
     int i;
 
     for (i = 0; i < len; i++)
         dst[i] = src0[i] * src1[i] + src2[i];
 }
 
42d32469
 static void vector_fmul_reverse_c(float *dst, const float *src0,
                                   const float *src1, int len)
 {
     int i;
 
     src1 += len-1;
     for (i = 0; i < len; i++)
         dst[i] = src0[i] * src1[-i];
 }
 
9d6e0ac6
 static void butterflies_float_c(float *av_restrict v1, float *av_restrict v2,
5959bfac
                                 int len)
 {
     int i;
 
     for (i = 0; i < len; i++) {
         float t = v1[i] - v2[i];
         v1[i] += v2[i];
         v2[i] = t;
     }
 }
 
d56668bd
 float avpriv_scalarproduct_float_c(const float *v1, const float *v2, int len)
 {
     float p = 0.0;
     int i;
 
     for (i = 0; i < len; i++)
         p += v1[i] * v2[i];
 
     return p;
 }
 
451b6a99
 av_cold AVFloatDSPContext *avpriv_float_dsp_alloc(int bit_exact)
d5a7229b
 {
451b6a99
     AVFloatDSPContext *fdsp = av_mallocz(sizeof(AVFloatDSPContext));
     if (!fdsp)
         return NULL;
 
d5a7229b
     fdsp->vector_fmul = vector_fmul_c;
bb16a062
     fdsp->vector_dmul = vector_dmul_c;
cb5042d0
     fdsp->vector_fmac_scalar = vector_fmac_scalar_c;
284ea790
     fdsp->vector_fmul_scalar = vector_fmul_scalar_c;
4dc2dd80
     fdsp->vector_dmac_scalar = vector_dmac_scalar_c;
ac7eb4cb
     fdsp->vector_dmul_scalar = vector_dmul_scalar_c;
e034cc6c
     fdsp->vector_fmul_window = vector_fmul_window_c;
55aa03b9
     fdsp->vector_fmul_add = vector_fmul_add_c;
42d32469
     fdsp->vector_fmul_reverse = vector_fmul_reverse_c;
5959bfac
     fdsp->butterflies_float = butterflies_float_c;
d56668bd
     fdsp->scalarproduct_float = avpriv_scalarproduct_float_c;
d5a7229b
 
46caba4a
     if (ARCH_AARCH64)
         ff_float_dsp_init_aarch64(fdsp);
     if (ARCH_ARM)
         ff_float_dsp_init_arm(fdsp);
     if (ARCH_PPC)
         ff_float_dsp_init_ppc(fdsp, bit_exact);
     if (ARCH_X86)
         ff_float_dsp_init_x86(fdsp);
cd1d4b51
     if (ARCH_MIPS)
         ff_float_dsp_init_mips(fdsp);
451b6a99
     return fdsp;
a54a51cc
 }