apedsp: move to llauddsp
[ffmpeg.git] / libavcodec / ppc / mpegvideo_altivec.c
1 /*
2  * Copyright (c) 2002 Dieter Shirley
3  *
4  * dct_unquantize_h263_altivec:
5  * Copyright (c) 2003 Romain Dolbeau <romain@dolbeau.org>
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 #include <stdlib.h>
25 #include <stdio.h>
26
27 #include "config.h"
28 #include "libavutil/attributes.h"
29 #include "libavutil/cpu.h"
30 #include "libavutil/ppc/types_altivec.h"
31 #include "libavutil/ppc/util_altivec.h"
32 #include "libavcodec/mpegvideo.h"
33
34 #if HAVE_ALTIVEC
35
36 /* AltiVec version of dct_unquantize_h263
37    this code assumes `block' is 16 bytes-aligned */
38 static void dct_unquantize_h263_altivec(MpegEncContext *s,
39                                  int16_t *block, int n, int qscale)
40 {
41     int i, level, qmul, qadd;
42     int nCoeffs;
43
44     assert(s->block_last_index[n]>=0);
45
46     qadd = (qscale - 1) | 1;
47     qmul = qscale << 1;
48
49     if (s->mb_intra) {
50         if (!s->h263_aic) {
51             if (n < 4)
52                 block[0] = block[0] * s->y_dc_scale;
53             else
54                 block[0] = block[0] * s->c_dc_scale;
55         }else
56             qadd = 0;
57         i = 1;
58         nCoeffs= 63; //does not always use zigzag table
59     } else {
60         i = 0;
61         nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
62     }
63
64     {
65         register const vector signed short vczero = (const vector signed short)vec_splat_s16(0);
66         DECLARE_ALIGNED(16, short, qmul8) = qmul;
67         DECLARE_ALIGNED(16, short, qadd8) = qadd;
68         register vector signed short blockv, qmulv, qaddv, nqaddv, temp1;
69         register vector bool short blockv_null, blockv_neg;
70         register short backup_0 = block[0];
71         register int j = 0;
72
73         qmulv = vec_splat((vec_s16)vec_lde(0, &qmul8), 0);
74         qaddv = vec_splat((vec_s16)vec_lde(0, &qadd8), 0);
75         nqaddv = vec_sub(vczero, qaddv);
76
77         // vectorize all the 16 bytes-aligned blocks
78         // of 8 elements
79         for(; (j + 7) <= nCoeffs ; j+=8) {
80             blockv = vec_ld(j << 1, block);
81             blockv_neg = vec_cmplt(blockv, vczero);
82             blockv_null = vec_cmpeq(blockv, vczero);
83             // choose between +qadd or -qadd as the third operand
84             temp1 = vec_sel(qaddv, nqaddv, blockv_neg);
85             // multiply & add (block{i,i+7} * qmul [+-] qadd)
86             temp1 = vec_mladd(blockv, qmulv, temp1);
87             // put 0 where block[{i,i+7} used to have 0
88             blockv = vec_sel(temp1, blockv, blockv_null);
89             vec_st(blockv, j << 1, block);
90         }
91
92         // if nCoeffs isn't a multiple of 8, finish the job
93         // using good old scalar units.
94         // (we could do it using a truncated vector,
95         // but I'm not sure it's worth the hassle)
96         for(; j <= nCoeffs ; j++) {
97             level = block[j];
98             if (level) {
99                 if (level < 0) {
100                     level = level * qmul - qadd;
101                 } else {
102                     level = level * qmul + qadd;
103                 }
104                 block[j] = level;
105             }
106         }
107
108         if (i == 1) {
109             // cheat. this avoid special-casing the first iteration
110             block[0] = backup_0;
111         }
112     }
113 }
114
115 #endif /* HAVE_ALTIVEC */
116
117 av_cold void ff_MPV_common_init_ppc(MpegEncContext *s)
118 {
119 #if HAVE_ALTIVEC
120     if (!(av_get_cpu_flags() & AV_CPU_FLAG_ALTIVEC))
121         return;
122
123     if ((s->avctx->dct_algo == FF_DCT_AUTO) ||
124         (s->avctx->dct_algo == FF_DCT_ALTIVEC)) {
125         s->dct_unquantize_h263_intra = dct_unquantize_h263_altivec;
126         s->dct_unquantize_h263_inter = dct_unquantize_h263_altivec;
127     }
128 #endif /* HAVE_ALTIVEC */
129 }