lpc_mmx.c 6.11 KB
Newer Older
1
/*
2
 * MMX optimized LPC DSP utils
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 * Copyright (c) 2007 Loren Merritt
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * 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.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * 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
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

22
#include "libavutil/x86_cpu.h"
23 24
#include "libavutil/cpu.h"
#include "libavcodec/lpc.h"
25

26 27
static void lpc_apply_welch_window_sse2(const int32_t *data, int len,
                                        double *w_data)
28 29 30
{
    double c = 2.0 / (len-1.0);
    int n2 = len>>1;
31 32
    x86_reg i = -n2*sizeof(int32_t);
    x86_reg j =  n2*sizeof(int32_t);
33
    __asm__ volatile(
34
        "movsd   %4,     %%xmm7                \n\t"
Alexander Strange's avatar
Alexander Strange committed
35 36 37 38 39
        "movapd  "MANGLE(ff_pd_1)", %%xmm6     \n\t"
        "movapd  "MANGLE(ff_pd_2)", %%xmm5     \n\t"
        "movlhps %%xmm7, %%xmm7                \n\t"
        "subpd   %%xmm5, %%xmm7                \n\t"
        "addsd   %%xmm6, %%xmm7                \n\t"
40 41
        "test    $1,     %5                    \n\t"
        "jz      2f                            \n\t"
42
#define WELCH(MOVPD, offset)\
Alexander Strange's avatar
Alexander Strange committed
43 44 45 46 47 48 49 50 51 52 53
        "1:                                    \n\t"\
        "movapd   %%xmm7,  %%xmm1              \n\t"\
        "mulpd    %%xmm1,  %%xmm1              \n\t"\
        "movapd   %%xmm6,  %%xmm0              \n\t"\
        "subpd    %%xmm1,  %%xmm0              \n\t"\
        "pshufd   $0x4e,   %%xmm0, %%xmm1      \n\t"\
        "cvtpi2pd (%3,%0), %%xmm2              \n\t"\
        "cvtpi2pd "#offset"*4(%3,%1), %%xmm3   \n\t"\
        "mulpd    %%xmm0,  %%xmm2              \n\t"\
        "mulpd    %%xmm1,  %%xmm3              \n\t"\
        "movapd   %%xmm2, (%2,%0,2)            \n\t"\
54
        MOVPD"    %%xmm3, "#offset"*8(%2,%1,2) \n\t"\
Alexander Strange's avatar
Alexander Strange committed
55 56 57 58
        "subpd    %%xmm5,  %%xmm7              \n\t"\
        "sub      $8,      %1                  \n\t"\
        "add      $8,      %0                  \n\t"\
        "jl 1b                                 \n\t"\
59

60
        WELCH("movupd", -1)
61 62
        "jmp 3f                                \n\t"
        "2:                                    \n\t"
63
        WELCH("movapd", -2)
64 65 66
        "3:                                    \n\t"
        :"+&r"(i), "+&r"(j)
        :"r"(w_data+n2), "r"(data+n2), "m"(c), "r"(len)
67 68
         XMM_CLOBBERS_ONLY("%xmm0", "%xmm1", "%xmm2", "%xmm3",
                                    "%xmm5", "%xmm6", "%xmm7")
69
    );
70 71 72
#undef WELCH
}

73
static void lpc_compute_autocorr_sse2(const double *data, int len, int lag,
74
                                      double *autoc)
75 76 77
{
    int j;

78 79
    if((x86_reg)data & 15)
        data++;
80 81

    for(j=0; j<lag; j+=2){
82
        x86_reg i = -len*sizeof(double);
83
        if(j == lag-2) {
84
            __asm__ volatile(
Alexander Strange's avatar
Alexander Strange committed
85 86 87 88
                "movsd    "MANGLE(ff_pd_1)", %%xmm0 \n\t"
                "movsd    "MANGLE(ff_pd_1)", %%xmm1 \n\t"
                "movsd    "MANGLE(ff_pd_1)", %%xmm2 \n\t"
                "1:                                 \n\t"
89 90 91
                "movapd   (%2,%0), %%xmm3           \n\t"
                "movupd -8(%3,%0), %%xmm4           \n\t"
                "movapd   (%3,%0), %%xmm5           \n\t"
Alexander Strange's avatar
Alexander Strange committed
92 93
                "mulpd     %%xmm3, %%xmm4           \n\t"
                "mulpd     %%xmm3, %%xmm5           \n\t"
94
                "mulpd -16(%3,%0), %%xmm3           \n\t"
Alexander Strange's avatar
Alexander Strange committed
95 96 97 98 99 100 101 102 103 104 105
                "addpd     %%xmm4, %%xmm1           \n\t"
                "addpd     %%xmm5, %%xmm0           \n\t"
                "addpd     %%xmm3, %%xmm2           \n\t"
                "add       $16,    %0               \n\t"
                "jl 1b                              \n\t"
                "movhlps   %%xmm0, %%xmm3           \n\t"
                "movhlps   %%xmm1, %%xmm4           \n\t"
                "movhlps   %%xmm2, %%xmm5           \n\t"
                "addsd     %%xmm3, %%xmm0           \n\t"
                "addsd     %%xmm4, %%xmm1           \n\t"
                "addsd     %%xmm5, %%xmm2           \n\t"
106 107 108 109
                "movsd     %%xmm0,   (%1)           \n\t"
                "movsd     %%xmm1,  8(%1)           \n\t"
                "movsd     %%xmm2, 16(%1)           \n\t"
                :"+&r"(i)
110
                :"r"(autoc+j), "r"(data+len), "r"(data+len-j)
111
                :"memory"
112 113
            );
        } else {
114
            __asm__ volatile(
Alexander Strange's avatar
Alexander Strange committed
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
                "movsd    "MANGLE(ff_pd_1)", %%xmm0 \n\t"
                "movsd    "MANGLE(ff_pd_1)", %%xmm1 \n\t"
                "1:                                 \n\t"
                "movapd   (%3,%0), %%xmm3           \n\t"
                "movupd -8(%4,%0), %%xmm4           \n\t"
                "mulpd     %%xmm3, %%xmm4           \n\t"
                "mulpd    (%4,%0), %%xmm3           \n\t"
                "addpd     %%xmm4, %%xmm1           \n\t"
                "addpd     %%xmm3, %%xmm0           \n\t"
                "add       $16,    %0               \n\t"
                "jl 1b                              \n\t"
                "movhlps   %%xmm0, %%xmm3           \n\t"
                "movhlps   %%xmm1, %%xmm4           \n\t"
                "addsd     %%xmm3, %%xmm0           \n\t"
                "addsd     %%xmm4, %%xmm1           \n\t"
                "movsd     %%xmm0, %1               \n\t"
                "movsd     %%xmm1, %2               \n\t"
132
                :"+&r"(i), "=m"(autoc[j]), "=m"(autoc[j+1])
133
                :"r"(data+len), "r"(data+len-j)
134 135 136 137
            );
        }
    }
}
138 139 140 141 142 143

av_cold void ff_lpc_init_x86(LPCContext *c)
{
    int mm_flags = av_get_cpu_flags();

    if (mm_flags & (AV_CPU_FLAG_SSE2|AV_CPU_FLAG_SSE2SLOW)) {
144 145
        c->lpc_apply_welch_window = lpc_apply_welch_window_sse2;
        c->lpc_compute_autocorr   = lpc_compute_autocorr_sse2;
146 147
    }
}