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.

359 lines
12KB

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