AltiVec perf (take 2), plus a couple AltiVec functions by (Romain Dolbeau <dolbeau...
[ffmpeg.git] / libavcodec / ppc / mpegvideo_altivec.c
1 /*
2  * Copyright (c) 2002 Dieter Shirley
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17  */
18  
19 #include <stdlib.h>
20 #include <stdio.h>
21 #include "../dsputil.h"
22 #include "../mpegvideo.h"
23 #include "dsputil_altivec.h"
24
25 // Swaps two variables (used for altivec registers)
26 #define SWAP(a,b) \
27 do { \
28     __typeof__(a) swap_temp=a; \
29     a=b; \
30     b=swap_temp; \
31 } while (0)
32
33 // transposes a matrix consisting of four vectors with four elements each
34 #define TRANSPOSE4(a,b,c,d) \
35 do { \
36   __typeof__(a) _trans_ach = vec_mergeh(a, c); \
37   __typeof__(a) _trans_acl = vec_mergel(a, c); \
38   __typeof__(a) _trans_bdh = vec_mergeh(b, d); \
39   __typeof__(a) _trans_bdl = vec_mergel(b, d); \
40  \
41   a = vec_mergeh(_trans_ach, _trans_bdh); \
42   b = vec_mergel(_trans_ach, _trans_bdh); \
43   c = vec_mergeh(_trans_acl, _trans_bdl); \
44   d = vec_mergel(_trans_acl, _trans_bdl); \
45 } while (0)
46
47 #define TRANSPOSE8(a,b,c,d,e,f,g,h) \
48 do { \
49     __typeof__(a)  _A1, _B1, _C1, _D1, _E1, _F1, _G1, _H1; \
50     __typeof__(a)  _A2, _B2, _C2, _D2, _E2, _F2, _G2, _H2; \
51  \
52     _A1 = vec_mergeh (a, e); \
53     _B1 = vec_mergel (a, e); \
54     _C1 = vec_mergeh (b, f); \
55     _D1 = vec_mergel (b, f); \
56     _E1 = vec_mergeh (c, g); \
57     _F1 = vec_mergel (c, g); \
58     _G1 = vec_mergeh (d, h); \
59     _H1 = vec_mergel (d, h); \
60  \
61     _A2 = vec_mergeh (_A1, _E1); \
62     _B2 = vec_mergel (_A1, _E1); \
63     _C2 = vec_mergeh (_B1, _F1); \
64     _D2 = vec_mergel (_B1, _F1); \
65     _E2 = vec_mergeh (_C1, _G1); \
66     _F2 = vec_mergel (_C1, _G1); \
67     _G2 = vec_mergeh (_D1, _H1); \
68     _H2 = vec_mergel (_D1, _H1); \
69  \
70     a = vec_mergeh (_A2, _E2); \
71     b = vec_mergel (_A2, _E2); \
72     c = vec_mergeh (_B2, _F2); \
73     d = vec_mergel (_B2, _F2); \
74     e = vec_mergeh (_C2, _G2); \
75     f = vec_mergel (_C2, _G2); \
76     g = vec_mergeh (_D2, _H2); \
77     h = vec_mergel (_D2, _H2); \
78 } while (0)
79
80
81 // Loads a four-byte value (int or float) from the target address
82 // into every element in the target vector.  Only works if the
83 // target address is four-byte aligned (which should be always).
84 #define LOAD4(vec, address) \
85 { \
86     __typeof__(vec)* _load_addr = (__typeof__(vec)*)(address); \
87     vector unsigned char _perm_vec = vec_lvsl(0,(address)); \
88     vec = vec_ld(0, _load_addr); \
89     vec = vec_perm(vec, vec, _perm_vec); \
90     vec = vec_splat(vec, 0); \
91 }
92
93 int dct_quantize_altivec(MpegEncContext* s, 
94                         DCTELEM* data, int n,
95                         int qscale, int* overflow)
96 {
97     int lastNonZero;
98     vector float row0, row1, row2, row3, row4, row5, row6, row7;
99     vector float alt0, alt1, alt2, alt3, alt4, alt5, alt6, alt7;
100     const vector float zero = (const vector float)(0.0f);
101
102     // Load the data into the row/alt vectors
103     {
104         vector signed short data0, data1, data2, data3, data4, data5, data6, data7;
105
106         data0 = vec_ld(0, data);
107         data1 = vec_ld(16, data);
108         data2 = vec_ld(32, data);
109         data3 = vec_ld(48, data);
110         data4 = vec_ld(64, data);
111         data5 = vec_ld(80, data);
112         data6 = vec_ld(96, data);
113         data7 = vec_ld(112, data);
114
115         // Transpose the data before we start
116         TRANSPOSE8(data0, data1, data2, data3, data4, data5, data6, data7);
117
118         // load the data into floating point vectors.  We load
119         // the high half of each row into the main row vectors
120         // and the low half into the alt vectors.
121         row0 = vec_ctf(vec_unpackh(data0), 0);
122         alt0 = vec_ctf(vec_unpackl(data0), 0);
123         row1 = vec_ctf(vec_unpackh(data1), 0);
124         alt1 = vec_ctf(vec_unpackl(data1), 0);
125         row2 = vec_ctf(vec_unpackh(data2), 0);
126         alt2 = vec_ctf(vec_unpackl(data2), 0);
127         row3 = vec_ctf(vec_unpackh(data3), 0);
128         alt3 = vec_ctf(vec_unpackl(data3), 0);
129         row4 = vec_ctf(vec_unpackh(data4), 0);
130         alt4 = vec_ctf(vec_unpackl(data4), 0);
131         row5 = vec_ctf(vec_unpackh(data5), 0);
132         alt5 = vec_ctf(vec_unpackl(data5), 0);
133         row6 = vec_ctf(vec_unpackh(data6), 0);
134         alt6 = vec_ctf(vec_unpackl(data6), 0);
135         row7 = vec_ctf(vec_unpackh(data7), 0);
136         alt7 = vec_ctf(vec_unpackl(data7), 0);
137     }
138
139     // The following block could exist as a separate an altivec dct
140                 // function.  However, if we put it inline, the DCT data can remain
141                 // in the vector local variables, as floats, which we'll use during the
142                 // quantize step...
143     {
144         const vector float vec_0_298631336 = (vector float)(0.298631336f);
145         const vector float vec_0_390180644 = (vector float)(-0.390180644f);
146         const vector float vec_0_541196100 = (vector float)(0.541196100f);
147         const vector float vec_0_765366865 = (vector float)(0.765366865f);
148         const vector float vec_0_899976223 = (vector float)(-0.899976223f);
149         const vector float vec_1_175875602 = (vector float)(1.175875602f);
150         const vector float vec_1_501321110 = (vector float)(1.501321110f);
151         const vector float vec_1_847759065 = (vector float)(-1.847759065f);
152         const vector float vec_1_961570560 = (vector float)(-1.961570560f);
153         const vector float vec_2_053119869 = (vector float)(2.053119869f);
154         const vector float vec_2_562915447 = (vector float)(-2.562915447f);
155         const vector float vec_3_072711026 = (vector float)(3.072711026f);
156
157
158         int whichPass, whichHalf;
159
160         for(whichPass = 1; whichPass<=2; whichPass++)
161         {
162             for(whichHalf = 1; whichHalf<=2; whichHalf++)
163             {
164                 vector float tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
165                 vector float tmp10, tmp11, tmp12, tmp13;
166                 vector float z1, z2, z3, z4, z5;
167
168                 tmp0 = vec_add(row0, row7); // tmp0 = dataptr[0] + dataptr[7];
169                 tmp7 = vec_sub(row0, row7); // tmp7 = dataptr[0] - dataptr[7];
170                 tmp3 = vec_add(row3, row4); // tmp3 = dataptr[3] + dataptr[4];
171                 tmp4 = vec_sub(row3, row4); // tmp4 = dataptr[3] - dataptr[4];
172                 tmp1 = vec_add(row1, row6); // tmp1 = dataptr[1] + dataptr[6];
173                 tmp6 = vec_sub(row1, row6); // tmp6 = dataptr[1] - dataptr[6];
174                 tmp2 = vec_add(row2, row5); // tmp2 = dataptr[2] + dataptr[5];
175                 tmp5 = vec_sub(row2, row5); // tmp5 = dataptr[2] - dataptr[5];
176
177                 tmp10 = vec_add(tmp0, tmp3); // tmp10 = tmp0 + tmp3;
178                 tmp13 = vec_sub(tmp0, tmp3); // tmp13 = tmp0 - tmp3;
179                 tmp11 = vec_add(tmp1, tmp2); // tmp11 = tmp1 + tmp2;
180                 tmp12 = vec_sub(tmp1, tmp2); // tmp12 = tmp1 - tmp2;
181
182
183                 // dataptr[0] = (DCTELEM) ((tmp10 + tmp11) << PASS1_BITS);
184                 row0 = vec_add(tmp10, tmp11);
185
186                 // dataptr[4] = (DCTELEM) ((tmp10 - tmp11) << PASS1_BITS);
187                 row4 = vec_sub(tmp10, tmp11);
188
189
190                 // z1 = MULTIPLY(tmp12 + tmp13, FIX_0_541196100);
191                 z1 = vec_madd(vec_add(tmp12, tmp13), vec_0_541196100, (vector float)zero);
192
193                 // dataptr[2] = (DCTELEM) DESCALE(z1 + MULTIPLY(tmp13, FIX_0_765366865),
194                 //                 CONST_BITS-PASS1_BITS);
195                 row2 = vec_madd(tmp13, vec_0_765366865, z1);
196
197                 // dataptr[6] = (DCTELEM) DESCALE(z1 + MULTIPLY(tmp12, - FIX_1_847759065),
198                 //                 CONST_BITS-PASS1_BITS);
199                 row6 = vec_madd(tmp12, vec_1_847759065, z1);
200
201                 z1 = vec_add(tmp4, tmp7); // z1 = tmp4 + tmp7;
202                 z2 = vec_add(tmp5, tmp6); // z2 = tmp5 + tmp6;
203                 z3 = vec_add(tmp4, tmp6); // z3 = tmp4 + tmp6;
204                 z4 = vec_add(tmp5, tmp7); // z4 = tmp5 + tmp7;
205
206                 // z5 = MULTIPLY(z3 + z4, FIX_1_175875602); /* sqrt(2) * c3 */
207                 z5 = vec_madd(vec_add(z3, z4), vec_1_175875602, (vector float)zero);
208
209                 // z3 = MULTIPLY(z3, - FIX_1_961570560); /* sqrt(2) * (-c3-c5) */
210                 z3 = vec_madd(z3, vec_1_961570560, z5);
211
212                 // z4 = MULTIPLY(z4, - FIX_0_390180644); /* sqrt(2) * (c5-c3) */
213                 z4 = vec_madd(z4, vec_0_390180644, z5);
214
215                 // The following adds are rolled into the multiplies above
216                 // z3 = vec_add(z3, z5);  // z3 += z5;
217                 // z4 = vec_add(z4, z5);  // z4 += z5;
218
219                 // z2 = MULTIPLY(z2, - FIX_2_562915447); /* sqrt(2) * (-c1-c3) */
220                 // Wow!  It's actually more effecient to roll this multiply
221                 // into the adds below, even thought the multiply gets done twice!
222                 // z2 = vec_madd(z2, vec_2_562915447, (vector float)zero);
223
224                 // z1 = MULTIPLY(z1, - FIX_0_899976223); /* sqrt(2) * (c7-c3) */
225                 // Same with this one...
226                 // z1 = vec_madd(z1, vec_0_899976223, (vector float)zero);
227
228                 // tmp4 = MULTIPLY(tmp4, FIX_0_298631336); /* sqrt(2) * (-c1+c3+c5-c7) */
229                 // dataptr[7] = (DCTELEM) DESCALE(tmp4 + z1 + z3, CONST_BITS-PASS1_BITS);
230                 row7 = vec_madd(tmp4, vec_0_298631336, vec_madd(z1, vec_0_899976223, z3));
231
232                 // tmp5 = MULTIPLY(tmp5, FIX_2_053119869); /* sqrt(2) * ( c1+c3-c5+c7) */
233                 // dataptr[5] = (DCTELEM) DESCALE(tmp5 + z2 + z4, CONST_BITS-PASS1_BITS);
234                 row5 = vec_madd(tmp5, vec_2_053119869, vec_madd(z2, vec_2_562915447, z4));
235
236                 // tmp6 = MULTIPLY(tmp6, FIX_3_072711026); /* sqrt(2) * ( c1+c3+c5-c7) */
237                 // dataptr[3] = (DCTELEM) DESCALE(tmp6 + z2 + z3, CONST_BITS-PASS1_BITS);
238                 row3 = vec_madd(tmp6, vec_3_072711026, vec_madd(z2, vec_2_562915447, z3));
239
240                 // tmp7 = MULTIPLY(tmp7, FIX_1_501321110); /* sqrt(2) * ( c1+c3-c5-c7) */
241                 // dataptr[1] = (DCTELEM) DESCALE(tmp7 + z1 + z4, CONST_BITS-PASS1_BITS);
242                 row1 = vec_madd(z1, vec_0_899976223, vec_madd(tmp7, vec_1_501321110, z4));
243
244                 // Swap the row values with the alts.  If this is the first half,
245                 // this sets up the low values to be acted on in the second half.
246                 // If this is the second half, it puts the high values back in
247                 // the row values where they are expected to be when we're done.
248                 SWAP(row0, alt0);
249                 SWAP(row1, alt1);
250                 SWAP(row2, alt2);
251                 SWAP(row3, alt3);
252                 SWAP(row4, alt4);
253                 SWAP(row5, alt5);
254                 SWAP(row6, alt6);
255                 SWAP(row7, alt7);
256             }
257
258             if (whichPass == 1)
259             {
260                 // transpose the data for the second pass
261                  
262                 // First, block transpose the upper right with lower left.
263                 SWAP(row4, alt0);
264                 SWAP(row5, alt1);
265                 SWAP(row6, alt2);
266                 SWAP(row7, alt3);
267
268                 // Now, transpose each block of four
269                 TRANSPOSE4(row0, row1, row2, row3);
270                 TRANSPOSE4(row4, row5, row6, row7);
271                 TRANSPOSE4(alt0, alt1, alt2, alt3);
272                 TRANSPOSE4(alt4, alt5, alt6, alt7);
273             }
274         }
275     }
276
277     // used after quantise step
278     int oldBaseValue = 0;
279
280     // perform the quantise step, using the floating point data
281     // still in the row/alt registers
282     {
283         const int* biasAddr;
284         const vector signed int* qmat;
285         vector float bias, negBias;
286
287         if (s->mb_intra)
288         {
289             vector signed int baseVector;
290
291             // We must cache element 0 in the intra case
292             // (it needs special handling).
293             baseVector = vec_cts(vec_splat(row0, 0), 0);
294             vec_ste(baseVector, 0, &oldBaseValue);
295
296             qmat = (vector signed int*)s->q_intra_matrix[qscale];
297             biasAddr = &(s->intra_quant_bias);
298         }
299         else
300         {
301             qmat = (vector signed int*)s->q_inter_matrix[qscale];
302             biasAddr = &(s->inter_quant_bias);
303         }
304
305         // Load the bias vector (We add 0.5 to the bias so that we're
306                                 // rounding when we convert to int, instead of flooring.)
307         {
308             vector signed int biasInt;
309             const vector float negOneFloat = (vector float)(-1.0f);
310             LOAD4(biasInt, biasAddr);
311             bias = vec_ctf(biasInt, QUANT_BIAS_SHIFT);
312             negBias = vec_madd(bias, negOneFloat, zero);
313         }
314
315         {
316             vector float q0, q1, q2, q3, q4, q5, q6, q7;
317
318             q0 = vec_ctf(qmat[0], QMAT_SHIFT);
319             q1 = vec_ctf(qmat[2], QMAT_SHIFT);
320             q2 = vec_ctf(qmat[4], QMAT_SHIFT);
321             q3 = vec_ctf(qmat[6], QMAT_SHIFT);
322             q4 = vec_ctf(qmat[8], QMAT_SHIFT);
323             q5 = vec_ctf(qmat[10], QMAT_SHIFT);
324             q6 = vec_ctf(qmat[12], QMAT_SHIFT);
325             q7 = vec_ctf(qmat[14], QMAT_SHIFT);
326
327             row0 = vec_sel(vec_madd(row0, q0, negBias), vec_madd(row0, q0, bias),
328                     vec_cmpgt(row0, zero));
329             row1 = vec_sel(vec_madd(row1, q1, negBias), vec_madd(row1, q1, bias),
330                     vec_cmpgt(row1, zero));
331             row2 = vec_sel(vec_madd(row2, q2, negBias), vec_madd(row2, q2, bias),
332                     vec_cmpgt(row2, zero));
333             row3 = vec_sel(vec_madd(row3, q3, negBias), vec_madd(row3, q3, bias),
334                     vec_cmpgt(row3, zero));
335             row4 = vec_sel(vec_madd(row4, q4, negBias), vec_madd(row4, q4, bias),
336                     vec_cmpgt(row4, zero));
337             row5 = vec_sel(vec_madd(row5, q5, negBias), vec_madd(row5, q5, bias),
338                     vec_cmpgt(row5, zero));
339             row6 = vec_sel(vec_madd(row6, q6, negBias), vec_madd(row6, q6, bias),
340                     vec_cmpgt(row6, zero));
341             row7 = vec_sel(vec_madd(row7, q7, negBias), vec_madd(row7, q7, bias),
342                     vec_cmpgt(row7, zero));
343
344             q0 = vec_ctf(qmat[1], QMAT_SHIFT);
345             q1 = vec_ctf(qmat[3], QMAT_SHIFT);
346             q2 = vec_ctf(qmat[5], QMAT_SHIFT);
347             q3 = vec_ctf(qmat[7], QMAT_SHIFT);
348             q4 = vec_ctf(qmat[9], QMAT_SHIFT);
349             q5 = vec_ctf(qmat[11], QMAT_SHIFT);
350             q6 = vec_ctf(qmat[13], QMAT_SHIFT);
351             q7 = vec_ctf(qmat[15], QMAT_SHIFT);
352
353             alt0 = vec_sel(vec_madd(alt0, q0, negBias), vec_madd(alt0, q0, bias),
354                     vec_cmpgt(alt0, zero));
355             alt1 = vec_sel(vec_madd(alt1, q1, negBias), vec_madd(alt1, q1, bias),
356                     vec_cmpgt(alt1, zero));
357             alt2 = vec_sel(vec_madd(alt2, q2, negBias), vec_madd(alt2, q2, bias),
358                     vec_cmpgt(alt2, zero));
359             alt3 = vec_sel(vec_madd(alt3, q3, negBias), vec_madd(alt3, q3, bias),
360                     vec_cmpgt(alt3, zero));
361             alt4 = vec_sel(vec_madd(alt4, q4, negBias), vec_madd(alt4, q4, bias),
362                     vec_cmpgt(alt4, zero));
363             alt5 = vec_sel(vec_madd(alt5, q5, negBias), vec_madd(alt5, q5, bias),
364                     vec_cmpgt(alt5, zero));
365             alt6 = vec_sel(vec_madd(alt6, q6, negBias), vec_madd(alt6, q6, bias),
366                     vec_cmpgt(alt6, zero));
367             alt7 = vec_sel(vec_madd(alt7, q7, negBias), vec_madd(alt7, q7, bias),
368                     vec_cmpgt(alt7, zero));
369         }
370
371  
372     }
373
374     // Store the data back into the original block
375     {
376         vector signed short data0, data1, data2, data3, data4, data5, data6, data7;
377
378         data0 = vec_pack(vec_cts(row0, 0), vec_cts(alt0, 0));
379         data1 = vec_pack(vec_cts(row1, 0), vec_cts(alt1, 0));
380         data2 = vec_pack(vec_cts(row2, 0), vec_cts(alt2, 0));
381         data3 = vec_pack(vec_cts(row3, 0), vec_cts(alt3, 0));
382         data4 = vec_pack(vec_cts(row4, 0), vec_cts(alt4, 0));
383         data5 = vec_pack(vec_cts(row5, 0), vec_cts(alt5, 0));
384         data6 = vec_pack(vec_cts(row6, 0), vec_cts(alt6, 0));
385         data7 = vec_pack(vec_cts(row7, 0), vec_cts(alt7, 0));
386
387         {
388             // Clamp for overflow
389             vector signed int max_q_int, min_q_int;
390             vector signed short max_q, min_q;
391
392             LOAD4(max_q_int, &(s->max_qcoeff));
393             LOAD4(min_q_int, &(s->min_qcoeff));
394
395             max_q = vec_pack(max_q_int, max_q_int);
396             min_q = vec_pack(min_q_int, min_q_int);
397
398             data0 = vec_max(vec_min(data0, max_q), min_q);
399             data1 = vec_max(vec_min(data1, max_q), min_q);
400             data2 = vec_max(vec_min(data2, max_q), min_q);
401             data4 = vec_max(vec_min(data4, max_q), min_q);
402             data5 = vec_max(vec_min(data5, max_q), min_q);
403             data6 = vec_max(vec_min(data6, max_q), min_q);
404             data7 = vec_max(vec_min(data7, max_q), min_q);
405         }
406
407         vector bool char zero_01, zero_23, zero_45, zero_67;
408         vector signed char scanIndices_01, scanIndices_23, scanIndices_45, scanIndices_67;
409         vector signed char negOne = vec_splat_s8(-1);
410         vector signed char* scanPtr =
411                 (vector signed char*)(s->intra_scantable.inverse);
412
413         // Determine the largest non-zero index.
414         zero_01 = vec_pack(vec_cmpeq(data0, (vector short)zero),
415                 vec_cmpeq(data1, (vector short)zero));
416         zero_23 = vec_pack(vec_cmpeq(data2, (vector short)zero),
417                 vec_cmpeq(data3, (vector short)zero));
418         zero_45 = vec_pack(vec_cmpeq(data4, (vector short)zero),
419                 vec_cmpeq(data5, (vector short)zero));
420         zero_67 = vec_pack(vec_cmpeq(data6, (vector short)zero),
421                 vec_cmpeq(data7, (vector short)zero));
422
423         // 64 biggest values
424         scanIndices_01 = vec_sel(scanPtr[0], negOne, zero_01);
425         scanIndices_23 = vec_sel(scanPtr[1], negOne, zero_23);
426         scanIndices_45 = vec_sel(scanPtr[2], negOne, zero_45);
427         scanIndices_67 = vec_sel(scanPtr[3], negOne, zero_67);
428
429         // 32 largest values
430         scanIndices_01 = vec_max(scanIndices_01, scanIndices_23);
431         scanIndices_45 = vec_max(scanIndices_45, scanIndices_67);
432
433         // 16 largest values
434         scanIndices_01 = vec_max(scanIndices_01, scanIndices_45);
435
436         // 8 largest values
437         scanIndices_01 = vec_max(vec_mergeh(scanIndices_01, negOne),
438                 vec_mergel(scanIndices_01, negOne));
439
440         // 4 largest values
441         scanIndices_01 = vec_max(vec_mergeh(scanIndices_01, negOne),
442                 vec_mergel(scanIndices_01, negOne));
443
444         // 2 largest values
445         scanIndices_01 = vec_max(vec_mergeh(scanIndices_01, negOne),
446                 vec_mergel(scanIndices_01, negOne));
447
448         // largest value
449         scanIndices_01 = vec_max(vec_mergeh(scanIndices_01, negOne),
450                 vec_mergel(scanIndices_01, negOne));
451
452         scanIndices_01 = vec_splat(scanIndices_01, 0);
453
454         signed char lastNonZeroChar;
455
456         vec_ste(scanIndices_01, 0, &lastNonZeroChar);
457
458         lastNonZero = lastNonZeroChar;
459         
460         // While the data is still in vectors we check for the transpose IDCT permute
461         // and handle it using the vector unit if we can.  This is the permute used
462         // by the altivec idct, so it is common when using the altivec dct.
463
464         if ((lastNonZero > 0) && (s->idct_permutation_type == FF_TRANSPOSE_IDCT_PERM))
465         {
466             TRANSPOSE8(data0, data1, data2, data3, data4, data5, data6, data7);
467         }
468
469         vec_st(data0, 0, data);
470         vec_st(data1, 16, data);
471         vec_st(data2, 32, data);
472         vec_st(data3, 48, data);
473         vec_st(data4, 64, data);
474         vec_st(data5, 80, data);
475         vec_st(data6, 96, data);
476         vec_st(data7, 112, data);
477     }
478
479     // special handling of block[0]
480     if (s->mb_intra)
481     {
482         if (!s->h263_aic)
483         {
484             if (n < 4)
485                 oldBaseValue /= s->y_dc_scale;
486             else
487                 oldBaseValue /= s->c_dc_scale;
488         }
489
490         // Divide by 8, rounding the result
491         data[0] = (oldBaseValue + 4) >> 3;
492     }
493
494     // We handled the tranpose permutation above and we don't
495     // need to permute the "no" permutation case.
496     if ((lastNonZero > 0) &&
497         (s->idct_permutation_type != FF_TRANSPOSE_IDCT_PERM) &&
498         (s->idct_permutation_type != FF_NO_IDCT_PERM))
499     {
500         ff_block_permute(data, s->idct_permutation,
501                 s->intra_scantable.scantable, lastNonZero);
502     }
503
504     return lastNonZero;
505 }
506
507 /*
508   AltiVec version of dct_unquantize_h263
509   this code assumes `block' is 16 bytes-aligned
510 */
511 void dct_unquantize_h263_altivec(MpegEncContext *s, 
512                                  DCTELEM *block, int n, int qscale)
513 {
514 ALTIVEC_TBL_DECLARE(altivec_dct_unquantize_h263_num, 1);
515     int i, level, qmul, qadd;
516     int nCoeffs;
517     
518     assert(s->block_last_index[n]>=0);
519
520 ALTIVEC_TBL_START_COUNT(altivec_dct_unquantize_h263_num, 1);
521     
522     qadd = (qscale - 1) | 1;
523     qmul = qscale << 1;
524     
525     if (s->mb_intra) {
526         if (!s->h263_aic) {
527             if (n < 4) 
528                 block[0] = block[0] * s->y_dc_scale;
529             else
530                 block[0] = block[0] * s->c_dc_scale;
531         }else
532             qadd = 0;
533         i = 1;
534         nCoeffs= 63; //does not allways use zigzag table 
535     } else {
536         i = 0;
537         nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
538     }
539
540 #ifdef ALTIVEC_USE_REFERENCE_C_CODE
541     for(;i<=nCoeffs;i++) {
542         level = block[i];
543         if (level) {
544             if (level < 0) {
545                 level = level * qmul - qadd;
546             } else {
547                 level = level * qmul + qadd;
548             }
549             block[i] = level;
550         }
551     }
552 #else /* ALTIVEC_USE_REFERENCE_C_CODE */
553     {
554       register const vector short vczero = (const vector short)(0);
555       short __attribute__ ((aligned(16))) qmul8[] =
556           {
557             qmul, qmul, qmul, qmul,
558             qmul, qmul, qmul, qmul
559           };
560       short __attribute__ ((aligned(16))) qadd8[] =
561           {
562             qadd, qadd, qadd, qadd,
563             qadd, qadd, qadd, qadd
564           };
565       short __attribute__ ((aligned(16))) nqadd8[] =
566           {
567             -qadd, -qadd, -qadd, -qadd,
568             -qadd, -qadd, -qadd, -qadd
569           };
570       register vector short blockv, qmulv, qaddv, nqaddv, temp1;
571       register vector bool short blockv_null, blockv_neg;
572       register short backup_0 = block[0];
573       register int j = 0;
574       
575       qmulv = vec_ld(0, qmul8);
576       qaddv = vec_ld(0, qadd8);
577       nqaddv = vec_ld(0, nqadd8);
578
579 #if 0 // block *is* 16 bytes-aligned, it seems.
580       // first make sure block[j] is 16 bytes-aligned
581       for(j = 0; (j <= nCoeffs) && ((((unsigned long)block) + (j << 1)) & 0x0000000F) ; j++) {
582         level = block[j];
583         if (level) {
584           if (level < 0) {
585                 level = level * qmul - qadd;
586             } else {
587                 level = level * qmul + qadd;
588             }
589             block[j] = level;
590         }
591       }
592 #endif
593       
594       // vectorize all the 16 bytes-aligned blocks
595       // of 8 elements
596       for(; (j + 7) <= nCoeffs ; j+=8)
597       {
598         blockv = vec_ld(j << 1, block);
599         blockv_neg = vec_cmplt(blockv, vczero);
600         blockv_null = vec_cmpeq(blockv, vczero);
601         // choose between +qadd or -qadd as the third operand
602         temp1 = vec_sel(qaddv, nqaddv, blockv_neg);
603         // multiply & add (block{i,i+7} * qmul [+-] qadd)
604         temp1 = vec_mladd(blockv, qmulv, temp1);
605         // put 0 where block[{i,i+7} used to have 0
606         blockv = vec_sel(temp1, blockv, blockv_null);
607         vec_st(blockv, j << 1, block);
608       }
609
610       // if nCoeffs isn't a multiple of 8, finish the job
611       // using good old scalar units.
612       // (we could do it using a truncated vector,
613       // but I'm not sure it's worth the hassle)
614       for(; j <= nCoeffs ; j++) {
615         level = block[j];
616         if (level) {
617           if (level < 0) {
618                 level = level * qmul - qadd;
619             } else {
620                 level = level * qmul + qadd;
621             }
622             block[j] = level;
623         }
624       }
625       
626       if (i == 1)
627       { // cheat. this avoid special-casing the first iteration
628         block[0] = backup_0;
629       }
630     }
631 #endif /* ALTIVEC_USE_REFERENCE_C_CODE */
632
633 ALTIVEC_TBL_STOP_COUNT(altivec_dct_unquantize_h263_num, nCoeffs == 63);
634 }