4e5ad2258c7583c7ae8f99a8263630e88e0f285d
[ffmpeg.git] / libavfilter / x86 / af_volume.asm
1 ;*****************************************************************************
2 ;* x86-optimized functions for volume filter
3 ;* Copyright (c) 2012 Justin Ruggles <justin.ruggles@gmail.com>
4 ;*
5 ;* This file is part of Libav.
6 ;*
7 ;* Libav is free software; you can redistribute it and/or
8 ;* modify it under the terms of the GNU Lesser General Public
9 ;* License as published by the Free Software Foundation; either
10 ;* version 2.1 of the License, or (at your option) any later version.
11 ;*
12 ;* Libav is distributed in the hope that it will be useful,
13 ;* but WITHOUT ANY WARRANTY; without even the implied warranty of
14 ;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 ;* Lesser General Public License for more details.
16 ;*
17 ;* You should have received a copy of the GNU Lesser General Public
18 ;* License along with Libav; if not, write to the Free Software
19 ;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 ;******************************************************************************
21
22 %include "libavutil/x86/x86util.asm"
23
24 SECTION_RODATA 32
25
26 pd_1_256:     times 4 dq 0x3F70000000000000
27 pd_int32_max: times 4 dq 0x41DFFFFFFFC00000
28 pw_1:         times 8 dw 1
29 pw_128:       times 8 dw 128
30 pq_128:       times 2 dq 128
31
32 SECTION_TEXT
33
34 ;------------------------------------------------------------------------------
35 ; void ff_scale_samples_s16(uint8_t *dst, const uint8_t *src, int len,
36 ;                           int volume)
37 ;------------------------------------------------------------------------------
38
39 INIT_XMM sse2
40 cglobal scale_samples_s16, 4,4,4, dst, src, len, volume
41     movd        m0, volumem
42     pshuflw     m0, m0, 0
43     punpcklwd   m0, [pw_1]
44     mova        m1, [pw_128]
45     lea       lenq, [lend*2-mmsize]
46 .loop:
47     ; dst[i] = av_clip_int16((src[i] * volume + 128) >> 8);
48     mova        m2, [srcq+lenq]
49     punpcklwd   m3, m2, m1
50     punpckhwd   m2, m1
51     pmaddwd     m3, m0
52     pmaddwd     m2, m0
53     psrad       m3, 8
54     psrad       m2, 8
55     packssdw    m3, m2
56     mova  [dstq+lenq], m3
57     sub       lenq, mmsize
58     jge .loop
59     REP_RET
60
61 ;------------------------------------------------------------------------------
62 ; void ff_scale_samples_s32(uint8_t *dst, const uint8_t *src, int len,
63 ;                           int volume)
64 ;------------------------------------------------------------------------------
65
66 %macro SCALE_SAMPLES_S32 0
67 cglobal scale_samples_s32, 4,4,4, dst, src, len, volume
68 %if ARCH_X86_32 && cpuflag(avx)
69     vbroadcastss   xmm2, volumem
70 %else
71     movd           xmm2, volumed
72     pshufd         xmm2, xmm2, 0
73 %endif
74     CVTDQ2PD         m2, xmm2
75     mulpd            m2, m2, [pd_1_256]
76     mova             m3, [pd_int32_max]
77     lea            lenq, [lend*4-mmsize]
78 .loop:
79     CVTDQ2PD         m0, [srcq+lenq         ]
80     CVTDQ2PD         m1, [srcq+lenq+mmsize/2]
81     mulpd            m0, m0, m2
82     mulpd            m1, m1, m2
83     minpd            m0, m0, m3
84     minpd            m1, m1, m3
85     cvtpd2dq       xmm0, m0
86     cvtpd2dq       xmm1, m1
87 %if cpuflag(avx)
88     vmovdqa [dstq+lenq         ], xmm0
89     vmovdqa [dstq+lenq+mmsize/2], xmm1
90 %else
91     movq    [dstq+lenq         ], xmm0
92     movq    [dstq+lenq+mmsize/2], xmm1
93 %endif
94     sub            lenq, mmsize
95     jge .loop
96     REP_RET
97 %endmacro
98
99 INIT_XMM sse2
100 %define CVTDQ2PD cvtdq2pd
101 SCALE_SAMPLES_S32
102 %define CVTDQ2PD vcvtdq2pd
103 INIT_YMM avx
104 SCALE_SAMPLES_S32
105 %undef CVTDQ2PD
106
107 ; NOTE: This is not bit-identical with the C version because it clips to
108 ;       [-INT_MAX, INT_MAX] instead of [INT_MIN, INT_MAX]
109
110 INIT_XMM ssse3, atom
111 cglobal scale_samples_s32, 4,4,8, dst, src, len, volume
112     movd        m4, volumem
113     pshufd      m4, m4, 0
114     mova        m5, [pq_128]
115     pxor        m6, m6
116     lea       lenq, [lend*4-mmsize]
117 .loop:
118     ; src[i] = av_clipl_int32((src[i] * volume + 128) >> 8);
119     mova        m7, [srcq+lenq]
120     pabsd       m3, m7
121     pshufd      m0, m3, q0100
122     pshufd      m1, m3, q0302
123     pmuludq     m0, m4
124     pmuludq     m1, m4
125     paddq       m0, m5
126     paddq       m1, m5
127     psrlq       m0, 7
128     psrlq       m1, 7
129     shufps      m2, m0, m1, q3131
130     shufps      m0, m0, m1, q2020
131     pcmpgtd     m2, m6
132     por         m0, m2
133     psrld       m0, 1
134     psignd      m0, m7
135     mova  [dstq+lenq], m0
136     sub       lenq, mmsize
137     jge .loop
138     REP_RET