You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

329 lines
11KB

  1. /*
  2. * VC-1 and WMV3 decoder - DSP functions AltiVec-optimized
  3. * Copyright (c) 2006 Konstantin Shishkov
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg 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. * FFmpeg 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 FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "libavcodec/dsputil.h"
  22. #include "util_altivec.h"
  23. // main steps of 8x8 transform
  24. #define STEP8(s0, s1, s2, s3, s4, s5, s6, s7, vec_rnd) \
  25. do { \
  26. t0 = vec_sl(vec_add(s0, s4), vec_2); \
  27. t0 = vec_add(vec_sl(t0, vec_1), t0); \
  28. t0 = vec_add(t0, vec_rnd); \
  29. t1 = vec_sl(vec_sub(s0, s4), vec_2); \
  30. t1 = vec_add(vec_sl(t1, vec_1), t1); \
  31. t1 = vec_add(t1, vec_rnd); \
  32. t2 = vec_add(vec_sl(s6, vec_2), vec_sl(s6, vec_1)); \
  33. t2 = vec_add(t2, vec_sl(s2, vec_4)); \
  34. t3 = vec_add(vec_sl(s2, vec_2), vec_sl(s2, vec_1)); \
  35. t3 = vec_sub(t3, vec_sl(s6, vec_4)); \
  36. t4 = vec_add(t0, t2); \
  37. t5 = vec_add(t1, t3); \
  38. t6 = vec_sub(t1, t3); \
  39. t7 = vec_sub(t0, t2); \
  40. \
  41. t0 = vec_sl(vec_add(s1, s3), vec_4); \
  42. t0 = vec_add(t0, vec_sl(s5, vec_3)); \
  43. t0 = vec_add(t0, vec_sl(s7, vec_2)); \
  44. t0 = vec_add(t0, vec_sub(s5, s3)); \
  45. \
  46. t1 = vec_sl(vec_sub(s1, s5), vec_4); \
  47. t1 = vec_sub(t1, vec_sl(s7, vec_3)); \
  48. t1 = vec_sub(t1, vec_sl(s3, vec_2)); \
  49. t1 = vec_sub(t1, vec_add(s1, s7)); \
  50. \
  51. t2 = vec_sl(vec_sub(s7, s3), vec_4); \
  52. t2 = vec_add(t2, vec_sl(s1, vec_3)); \
  53. t2 = vec_add(t2, vec_sl(s5, vec_2)); \
  54. t2 = vec_add(t2, vec_sub(s1, s7)); \
  55. \
  56. t3 = vec_sl(vec_sub(s5, s7), vec_4); \
  57. t3 = vec_sub(t3, vec_sl(s3, vec_3)); \
  58. t3 = vec_add(t3, vec_sl(s1, vec_2)); \
  59. t3 = vec_sub(t3, vec_add(s3, s5)); \
  60. \
  61. s0 = vec_add(t4, t0); \
  62. s1 = vec_add(t5, t1); \
  63. s2 = vec_add(t6, t2); \
  64. s3 = vec_add(t7, t3); \
  65. s4 = vec_sub(t7, t3); \
  66. s5 = vec_sub(t6, t2); \
  67. s6 = vec_sub(t5, t1); \
  68. s7 = vec_sub(t4, t0); \
  69. }while(0)
  70. #define SHIFT_HOR8(s0, s1, s2, s3, s4, s5, s6, s7) \
  71. do { \
  72. s0 = vec_sra(s0, vec_3); \
  73. s1 = vec_sra(s1, vec_3); \
  74. s2 = vec_sra(s2, vec_3); \
  75. s3 = vec_sra(s3, vec_3); \
  76. s4 = vec_sra(s4, vec_3); \
  77. s5 = vec_sra(s5, vec_3); \
  78. s6 = vec_sra(s6, vec_3); \
  79. s7 = vec_sra(s7, vec_3); \
  80. }while(0)
  81. #define SHIFT_VERT8(s0, s1, s2, s3, s4, s5, s6, s7) \
  82. do { \
  83. s0 = vec_sra(s0, vec_7); \
  84. s1 = vec_sra(s1, vec_7); \
  85. s2 = vec_sra(s2, vec_7); \
  86. s3 = vec_sra(s3, vec_7); \
  87. s4 = vec_sra(vec_add(s4, vec_1s), vec_7); \
  88. s5 = vec_sra(vec_add(s5, vec_1s), vec_7); \
  89. s6 = vec_sra(vec_add(s6, vec_1s), vec_7); \
  90. s7 = vec_sra(vec_add(s7, vec_1s), vec_7); \
  91. }while(0)
  92. /* main steps of 4x4 transform */
  93. #define STEP4(s0, s1, s2, s3, vec_rnd) \
  94. do { \
  95. t1 = vec_add(vec_sl(s0, vec_4), s0); \
  96. t1 = vec_add(t1, vec_rnd); \
  97. t2 = vec_add(vec_sl(s2, vec_4), s2); \
  98. t0 = vec_add(t1, t2); \
  99. t1 = vec_sub(t1, t2); \
  100. t3 = vec_sl(vec_sub(s3, s1), vec_1); \
  101. t3 = vec_add(t3, vec_sl(t3, vec_2)); \
  102. t2 = vec_add(t3, vec_sl(s1, vec_5)); \
  103. t3 = vec_add(t3, vec_sl(s3, vec_3)); \
  104. t3 = vec_add(t3, vec_sl(s3, vec_2)); \
  105. s0 = vec_add(t0, t2); \
  106. s1 = vec_sub(t1, t3); \
  107. s2 = vec_add(t1, t3); \
  108. s3 = vec_sub(t0, t2); \
  109. }while (0)
  110. #define SHIFT_HOR4(s0, s1, s2, s3) \
  111. s0 = vec_sra(s0, vec_3); \
  112. s1 = vec_sra(s1, vec_3); \
  113. s2 = vec_sra(s2, vec_3); \
  114. s3 = vec_sra(s3, vec_3);
  115. #define SHIFT_VERT4(s0, s1, s2, s3) \
  116. s0 = vec_sra(s0, vec_7); \
  117. s1 = vec_sra(s1, vec_7); \
  118. s2 = vec_sra(s2, vec_7); \
  119. s3 = vec_sra(s3, vec_7);
  120. /** Do inverse transform on 8x8 block
  121. */
  122. static void vc1_inv_trans_8x8_altivec(DCTELEM block[64])
  123. {
  124. vector signed short src0, src1, src2, src3, src4, src5, src6, src7;
  125. vector signed int s0, s1, s2, s3, s4, s5, s6, s7;
  126. vector signed int s8, s9, sA, sB, sC, sD, sE, sF;
  127. vector signed int t0, t1, t2, t3, t4, t5, t6, t7;
  128. const vector signed int vec_64 = vec_sl(vec_splat_s32(4), vec_splat_u32(4));
  129. const vector unsigned int vec_7 = vec_splat_u32(7);
  130. const vector unsigned int vec_4 = vec_splat_u32(4);
  131. const vector signed int vec_4s = vec_splat_s32(4);
  132. const vector unsigned int vec_3 = vec_splat_u32(3);
  133. const vector unsigned int vec_2 = vec_splat_u32(2);
  134. const vector signed int vec_1s = vec_splat_s32(1);
  135. const vector unsigned int vec_1 = vec_splat_u32(1);
  136. src0 = vec_ld( 0, block);
  137. src1 = vec_ld( 16, block);
  138. src2 = vec_ld( 32, block);
  139. src3 = vec_ld( 48, block);
  140. src4 = vec_ld( 64, block);
  141. src5 = vec_ld( 80, block);
  142. src6 = vec_ld( 96, block);
  143. src7 = vec_ld(112, block);
  144. TRANSPOSE8(src0, src1, src2, src3, src4, src5, src6, src7);
  145. s0 = vec_unpackl(src0);
  146. s1 = vec_unpackl(src1);
  147. s2 = vec_unpackl(src2);
  148. s3 = vec_unpackl(src3);
  149. s4 = vec_unpackl(src4);
  150. s5 = vec_unpackl(src5);
  151. s6 = vec_unpackl(src6);
  152. s7 = vec_unpackl(src7);
  153. s8 = vec_unpackh(src0);
  154. s9 = vec_unpackh(src1);
  155. sA = vec_unpackh(src2);
  156. sB = vec_unpackh(src3);
  157. sC = vec_unpackh(src4);
  158. sD = vec_unpackh(src5);
  159. sE = vec_unpackh(src6);
  160. sF = vec_unpackh(src7);
  161. STEP8(s0, s1, s2, s3, s4, s5, s6, s7, vec_4s);
  162. SHIFT_HOR8(s0, s1, s2, s3, s4, s5, s6, s7);
  163. STEP8(s8, s9, sA, sB, sC, sD, sE, sF, vec_4s);
  164. SHIFT_HOR8(s8, s9, sA, sB, sC, sD, sE, sF);
  165. src0 = vec_pack(s8, s0);
  166. src1 = vec_pack(s9, s1);
  167. src2 = vec_pack(sA, s2);
  168. src3 = vec_pack(sB, s3);
  169. src4 = vec_pack(sC, s4);
  170. src5 = vec_pack(sD, s5);
  171. src6 = vec_pack(sE, s6);
  172. src7 = vec_pack(sF, s7);
  173. TRANSPOSE8(src0, src1, src2, src3, src4, src5, src6, src7);
  174. s0 = vec_unpackl(src0);
  175. s1 = vec_unpackl(src1);
  176. s2 = vec_unpackl(src2);
  177. s3 = vec_unpackl(src3);
  178. s4 = vec_unpackl(src4);
  179. s5 = vec_unpackl(src5);
  180. s6 = vec_unpackl(src6);
  181. s7 = vec_unpackl(src7);
  182. s8 = vec_unpackh(src0);
  183. s9 = vec_unpackh(src1);
  184. sA = vec_unpackh(src2);
  185. sB = vec_unpackh(src3);
  186. sC = vec_unpackh(src4);
  187. sD = vec_unpackh(src5);
  188. sE = vec_unpackh(src6);
  189. sF = vec_unpackh(src7);
  190. STEP8(s0, s1, s2, s3, s4, s5, s6, s7, vec_64);
  191. SHIFT_VERT8(s0, s1, s2, s3, s4, s5, s6, s7);
  192. STEP8(s8, s9, sA, sB, sC, sD, sE, sF, vec_64);
  193. SHIFT_VERT8(s8, s9, sA, sB, sC, sD, sE, sF);
  194. src0 = vec_pack(s8, s0);
  195. src1 = vec_pack(s9, s1);
  196. src2 = vec_pack(sA, s2);
  197. src3 = vec_pack(sB, s3);
  198. src4 = vec_pack(sC, s4);
  199. src5 = vec_pack(sD, s5);
  200. src6 = vec_pack(sE, s6);
  201. src7 = vec_pack(sF, s7);
  202. vec_st(src0, 0, block);
  203. vec_st(src1, 16, block);
  204. vec_st(src2, 32, block);
  205. vec_st(src3, 48, block);
  206. vec_st(src4, 64, block);
  207. vec_st(src5, 80, block);
  208. vec_st(src6, 96, block);
  209. vec_st(src7,112, block);
  210. }
  211. /** Do inverse transform on 8x4 part of block
  212. */
  213. static void vc1_inv_trans_8x4_altivec(uint8_t *dest, int stride, DCTELEM *block)
  214. {
  215. vector signed short src0, src1, src2, src3, src4, src5, src6, src7;
  216. vector signed int s0, s1, s2, s3, s4, s5, s6, s7;
  217. vector signed int s8, s9, sA, sB, sC, sD, sE, sF;
  218. vector signed int t0, t1, t2, t3, t4, t5, t6, t7;
  219. const vector signed int vec_64 = vec_sl(vec_splat_s32(4), vec_splat_u32(4));
  220. const vector unsigned int vec_7 = vec_splat_u32(7);
  221. const vector unsigned int vec_5 = vec_splat_u32(5);
  222. const vector unsigned int vec_4 = vec_splat_u32(4);
  223. const vector signed int vec_4s = vec_splat_s32(4);
  224. const vector unsigned int vec_3 = vec_splat_u32(3);
  225. const vector unsigned int vec_2 = vec_splat_u32(2);
  226. const vector unsigned int vec_1 = vec_splat_u32(1);
  227. vector unsigned char tmp;
  228. vector signed short tmp2, tmp3;
  229. vector unsigned char perm0, perm1, p0, p1, p;
  230. src0 = vec_ld( 0, block);
  231. src1 = vec_ld( 16, block);
  232. src2 = vec_ld( 32, block);
  233. src3 = vec_ld( 48, block);
  234. src4 = vec_ld( 64, block);
  235. src5 = vec_ld( 80, block);
  236. src6 = vec_ld( 96, block);
  237. src7 = vec_ld(112, block);
  238. TRANSPOSE8(src0, src1, src2, src3, src4, src5, src6, src7);
  239. s0 = vec_unpackl(src0);
  240. s1 = vec_unpackl(src1);
  241. s2 = vec_unpackl(src2);
  242. s3 = vec_unpackl(src3);
  243. s4 = vec_unpackl(src4);
  244. s5 = vec_unpackl(src5);
  245. s6 = vec_unpackl(src6);
  246. s7 = vec_unpackl(src7);
  247. s8 = vec_unpackh(src0);
  248. s9 = vec_unpackh(src1);
  249. sA = vec_unpackh(src2);
  250. sB = vec_unpackh(src3);
  251. sC = vec_unpackh(src4);
  252. sD = vec_unpackh(src5);
  253. sE = vec_unpackh(src6);
  254. sF = vec_unpackh(src7);
  255. STEP8(s0, s1, s2, s3, s4, s5, s6, s7, vec_4s);
  256. SHIFT_HOR8(s0, s1, s2, s3, s4, s5, s6, s7);
  257. STEP8(s8, s9, sA, sB, sC, sD, sE, sF, vec_4s);
  258. SHIFT_HOR8(s8, s9, sA, sB, sC, sD, sE, sF);
  259. src0 = vec_pack(s8, s0);
  260. src1 = vec_pack(s9, s1);
  261. src2 = vec_pack(sA, s2);
  262. src3 = vec_pack(sB, s3);
  263. src4 = vec_pack(sC, s4);
  264. src5 = vec_pack(sD, s5);
  265. src6 = vec_pack(sE, s6);
  266. src7 = vec_pack(sF, s7);
  267. TRANSPOSE8(src0, src1, src2, src3, src4, src5, src6, src7);
  268. s0 = vec_unpackh(src0);
  269. s1 = vec_unpackh(src1);
  270. s2 = vec_unpackh(src2);
  271. s3 = vec_unpackh(src3);
  272. s8 = vec_unpackl(src0);
  273. s9 = vec_unpackl(src1);
  274. sA = vec_unpackl(src2);
  275. sB = vec_unpackl(src3);
  276. STEP4(s0, s1, s2, s3, vec_64);
  277. SHIFT_VERT4(s0, s1, s2, s3);
  278. STEP4(s8, s9, sA, sB, vec_64);
  279. SHIFT_VERT4(s8, s9, sA, sB);
  280. src0 = vec_pack(s0, s8);
  281. src1 = vec_pack(s1, s9);
  282. src2 = vec_pack(s2, sA);
  283. src3 = vec_pack(s3, sB);
  284. p0 = vec_lvsl (0, dest);
  285. p1 = vec_lvsl (stride, dest);
  286. p = vec_splat_u8 (-1);
  287. perm0 = vec_mergeh (p, p0);
  288. perm1 = vec_mergeh (p, p1);
  289. #define ADD(dest,src,perm) \
  290. /* *(uint64_t *)&tmp = *(uint64_t *)dest; */ \
  291. tmp = vec_ld (0, dest); \
  292. tmp2 = (vector signed short)vec_perm (tmp, vec_splat_u8(0), perm); \
  293. tmp3 = vec_adds (tmp2, src); \
  294. tmp = vec_packsu (tmp3, tmp3); \
  295. vec_ste ((vector unsigned int)tmp, 0, (unsigned int *)dest); \
  296. vec_ste ((vector unsigned int)tmp, 4, (unsigned int *)dest);
  297. ADD (dest, src0, perm0) dest += stride;
  298. ADD (dest, src1, perm1) dest += stride;
  299. ADD (dest, src2, perm0) dest += stride;
  300. ADD (dest, src3, perm1)
  301. }
  302. void vc1dsp_init_altivec(DSPContext* dsp, AVCodecContext *avctx) {
  303. dsp->vc1_inv_trans_8x8 = vc1_inv_trans_8x8_altivec;
  304. dsp->vc1_inv_trans_8x4 = vc1_inv_trans_8x4_altivec;
  305. }