Merge remote-tracking branch 'qatar/master'
[ffmpeg.git] / libavcodec / ppc / fmtconvert_altivec.c
1 /*
2  * Copyright (c) 2006 Luca Barbato <lu_zero@gentoo.org>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include "libavcodec/fmtconvert.h"
22
23 #include "libavutil/ppc/util_altivec.h"
24 #include "libavutil/mem.h"
25 #include "dsputil_altivec.h"
26
27 static void int32_to_float_fmul_scalar_altivec(float *dst, const int *src,
28                                                float mul, int len)
29 {
30     union {
31         vector float v;
32         float s[4];
33     } mul_u;
34     int i;
35     vector float src1, src2, dst1, dst2, mul_v, zero;
36
37     zero = (vector float)vec_splat_u32(0);
38     mul_u.s[0] = mul;
39     mul_v = vec_splat(mul_u.v, 0);
40
41     for (i = 0; i < len; i += 8) {
42         src1 = vec_ctf(vec_ld(0,  src+i), 0);
43         src2 = vec_ctf(vec_ld(16, src+i), 0);
44         dst1 = vec_madd(src1, mul_v, zero);
45         dst2 = vec_madd(src2, mul_v, zero);
46         vec_st(dst1,  0, dst+i);
47         vec_st(dst2, 16, dst+i);
48     }
49 }
50
51
52 static vector signed short float_to_int16_one_altivec(const float *src)
53 {
54     vector float s0 = vec_ld(0, src);
55     vector float s1 = vec_ld(16, src);
56     vector signed int t0 = vec_cts(s0, 0);
57     vector signed int t1 = vec_cts(s1, 0);
58     return vec_packs(t0,t1);
59 }
60
61 static void float_to_int16_altivec(int16_t *dst, const float *src, long len)
62 {
63     int i;
64     vector signed short d0, d1, d;
65     vector unsigned char align;
66     if (((long)dst) & 15) { //FIXME
67         for (i = 0; i < len - 7; i += 8) {
68             d0 = vec_ld(0, dst+i);
69             d  = float_to_int16_one_altivec(src + i);
70             d1 = vec_ld(15, dst+i);
71             d1 = vec_perm(d1, d0, vec_lvsl(0, dst + i));
72             align = vec_lvsr(0, dst + i);
73             d0 = vec_perm(d1, d, align);
74             d1 = vec_perm(d, d1, align);
75             vec_st(d0,  0, dst + i);
76             vec_st(d1, 15, dst + i);
77         }
78     } else {
79         for (i = 0; i < len - 7; i += 8) {
80             d = float_to_int16_one_altivec(src + i);
81             vec_st(d, 0, dst + i);
82         }
83     }
84 }
85
86 static void float_to_int16_stride_altivec(int16_t *dst, const float *src,
87                                           long len, int stride)
88 {
89     int i, j;
90     vector signed short d, s;
91
92     for (i = 0; i < len - 7; i += 8) {
93         d = float_to_int16_one_altivec(src + i);
94         for (j = 0; j < 8; j++) {
95             s = vec_splat(d, j);
96             vec_ste(s, 0, dst);
97             dst += stride;
98         }
99     }
100 }
101
102 static void float_to_int16_interleave_altivec(int16_t *dst, const float **src,
103                                               long len, int channels)
104 {
105     int i;
106     vector signed short d0, d1, d2, c0, c1, t0, t1;
107     vector unsigned char align;
108
109     if (channels == 1)
110         float_to_int16_altivec(dst, src[0], len);
111     else {
112         if (channels == 2) {
113             if (((long)dst) & 15) {
114                 for (i = 0; i < len - 7; i += 8) {
115                     d0 = vec_ld(0,  dst + i);
116                     t0 = float_to_int16_one_altivec(src[0] + i);
117                     d1 = vec_ld(31, dst + i);
118                     t1 = float_to_int16_one_altivec(src[1] + i);
119                     c0 = vec_mergeh(t0, t1);
120                     c1 = vec_mergel(t0, t1);
121                     d2 = vec_perm(d1, d0, vec_lvsl(0, dst + i));
122                     align = vec_lvsr(0, dst + i);
123                     d0 = vec_perm(d2, c0, align);
124                     d1 = vec_perm(c0, c1, align);
125                     vec_st(d0,  0, dst + i);
126                     d0 = vec_perm(c1, d2, align);
127                     vec_st(d1, 15, dst + i);
128                     vec_st(d0, 31, dst + i);
129                     dst += 8;
130                 }
131             } else {
132                 for (i = 0; i < len - 7; i += 8) {
133                     t0 = float_to_int16_one_altivec(src[0] + i);
134                     t1 = float_to_int16_one_altivec(src[1] + i);
135                     d0 = vec_mergeh(t0, t1);
136                     d1 = vec_mergel(t0, t1);
137                     vec_st(d0,  0, dst + i);
138                     vec_st(d1, 16, dst + i);
139                     dst += 8;
140                 }
141             }
142         } else {
143             for (i = 0; i < channels; i++)
144                 float_to_int16_stride_altivec(dst + i, src[i], len, channels);
145         }
146     }
147 }
148
149 void ff_fmt_convert_init_altivec(FmtConvertContext *c, AVCodecContext *avctx)
150 {
151     c->int32_to_float_fmul_scalar = int32_to_float_fmul_scalar_altivec;
152     if (!(avctx->flags & CODEC_FLAG_BITEXACT)) {
153         c->float_to_int16 = float_to_int16_altivec;
154         c->float_to_int16_interleave = float_to_int16_interleave_altivec;
155     }
156 }