ppc: dsputil: more K&R formatting cosmetics
[ffmpeg.git] / libavcodec / ppc / gmc_altivec.c
1 /*
2  * GMC (Global Motion Compensation), AltiVec-enabled
3  *
4  * Copyright (c) 2003 Romain Dolbeau <romain@dolbeau.org>
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "libavutil/mem.h"
24 #include "libavutil/ppc/types_altivec.h"
25 #include "libavutil/ppc/util_altivec.h"
26 #include "dsputil_altivec.h"
27
28 /* AltiVec-enhanced gmc1. ATM this code assumes stride is a multiple of 8
29  * to preserve proper dst alignment. */
30 void ff_gmc1_altivec(uint8_t *dst /* align 8 */, uint8_t *src /* align1 */,
31                      int stride, int h, int x16, int y16, int rounder)
32 {
33     const DECLARE_ALIGNED(16, unsigned short, rounder_a) = rounder;
34     const DECLARE_ALIGNED(16, unsigned short, ABCD)[8] = {
35         (16 - x16) * (16 - y16), /* A */
36              (x16) * (16 - y16), /* B */
37         (16 - x16) * (y16),      /* C */
38              (x16) * (y16),      /* D */
39         0, 0, 0, 0               /* padding */
40     };
41     register const vector unsigned char vczero =
42         (const vector unsigned char) vec_splat_u8(0);
43     register const vector unsigned short vcsr8 =
44         (const vector unsigned short) vec_splat_u16(8);
45     register vector unsigned char dstv, dstv2, src_0, src_1,
46         srcvA, srcvB, srcvC, srcvD;
47     register vector unsigned short Av, Bv, Cv, Dv, rounderV,
48         tempA, tempB, tempC, tempD;
49     int i;
50     unsigned long dst_odd        = (unsigned long) dst & 0x0000000F;
51     unsigned long src_really_odd = (unsigned long) src & 0x0000000F;
52
53     tempA = vec_ld(0, (const unsigned short *) ABCD);
54     Av    = vec_splat(tempA, 0);
55     Bv    = vec_splat(tempA, 1);
56     Cv    = vec_splat(tempA, 2);
57     Dv    = vec_splat(tempA, 3);
58
59     rounderV = vec_splat((vec_u16) vec_lde(0, &rounder_a), 0);
60
61     /* we'll be able to pick-up our 9 char elements at src from those
62      * 32 bytes we load the first batch here, as inside the loop we can
63      * reuse 'src + stride' from one iteration as the 'src' of the next. */
64     src_0 = vec_ld(0, src);
65     src_1 = vec_ld(16, src);
66     srcvA = vec_perm(src_0, src_1, vec_lvsl(0, src));
67
68     if (src_really_odd != 0x0000000F)
69         /* If src & 0xF == 0xF, then (src + 1) is properly aligned
70          * on the second vector. */
71         srcvB = vec_perm(src_0, src_1, vec_lvsl(1, src));
72     else
73         srcvB = src_1;
74     srcvA = vec_mergeh(vczero, srcvA);
75     srcvB = vec_mergeh(vczero, srcvB);
76
77     for (i = 0; i < h; i++) {
78         dst_odd        =   (unsigned long) dst            & 0x0000000F;
79         src_really_odd = (((unsigned long) src) + stride) & 0x0000000F;
80
81         dstv = vec_ld(0, dst);
82
83         /* We'll be able to pick-up our 9 char elements at src + stride from
84          * those 32 bytes then reuse the resulting 2 vectors srvcC and srcvD
85          * as the next srcvA and srcvB. */
86         src_0 = vec_ld(stride +  0, src);
87         src_1 = vec_ld(stride + 16, src);
88         srcvC = vec_perm(src_0, src_1, vec_lvsl(stride + 0, src));
89
90         if (src_really_odd != 0x0000000F)
91             /* If src & 0xF == 0xF, then (src + 1) is properly aligned
92              * on the second vector. */
93             srcvD = vec_perm(src_0, src_1, vec_lvsl(stride + 1, src));
94         else
95             srcvD = src_1;
96
97         srcvC = vec_mergeh(vczero, srcvC);
98         srcvD = vec_mergeh(vczero, srcvD);
99
100         /* OK, now we (finally) do the math :-)
101          * Those four instructions replace 32 int muls & 32 int adds.
102          * Isn't AltiVec nice? */
103         tempA = vec_mladd((vector unsigned short) srcvA, Av, rounderV);
104         tempB = vec_mladd((vector unsigned short) srcvB, Bv, tempA);
105         tempC = vec_mladd((vector unsigned short) srcvC, Cv, tempB);
106         tempD = vec_mladd((vector unsigned short) srcvD, Dv, tempC);
107
108         srcvA = srcvC;
109         srcvB = srcvD;
110
111         tempD = vec_sr(tempD, vcsr8);
112
113         dstv2 = vec_pack(tempD, (vector unsigned short) vczero);
114
115         if (dst_odd)
116             dstv2 = vec_perm(dstv, dstv2, vcprm(0, 1, s0, s1));
117         else
118             dstv2 = vec_perm(dstv, dstv2, vcprm(s0, s1, 2, 3));
119
120         vec_st(dstv2, 0, dst);
121
122         dst += stride;
123         src += stride;
124     }
125 }