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.

409 lines
15KB

  1. /*
  2. * XVID MPEG-4 VIDEO CODEC
  3. * - SSE2 inverse discrete cosine transform -
  4. *
  5. * Copyright(C) 2003 Pascal Massimino <skal@planet-d.net>
  6. *
  7. * Conversion to gcc syntax with modifications
  8. * by Alexander Strange <astrange@ithinksw.com>
  9. *
  10. * Originally from dct/x86_asm/fdct_sse2_skal.asm in Xvid.
  11. *
  12. * This file is part of Libav.
  13. *
  14. * Vertical pass is an implementation of the scheme:
  15. * Loeffler C., Ligtenberg A., and Moschytz C.S.:
  16. * Practical Fast 1D DCT Algorithm with Eleven Multiplications,
  17. * Proc. ICASSP 1989, 988-991.
  18. *
  19. * Horizontal pass is a double 4x4 vector/matrix multiplication,
  20. * (see also Intel's Application Note 922:
  21. * http://developer.intel.com/vtune/cbts/strmsimd/922down.htm
  22. * Copyright (C) 1999 Intel Corporation)
  23. *
  24. * More details at http://skal.planet-d.net/coding/dct.html
  25. *
  26. * Libav is free software; you can redistribute it and/or
  27. * modify it under the terms of the GNU Lesser General Public
  28. * License as published by the Free Software Foundation; either
  29. * version 2.1 of the License, or (at your option) any later version.
  30. *
  31. * Libav is distributed in the hope that it will be useful,
  32. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  33. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  34. * Lesser General Public License for more details.
  35. *
  36. * You should have received a copy of the GNU Lesser General Public License
  37. * along with Libav; if not, write to the Free Software Foundation,
  38. * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  39. */
  40. #include "libavutil/internal.h"
  41. #include "libavutil/mem.h"
  42. #include "libavutil/x86/asm.h"
  43. #include "idct_xvid.h"
  44. #include "idctdsp.h"
  45. #if HAVE_SSE2_INLINE
  46. /**
  47. * @file
  48. * @brief SSE2 idct compatible with xvidmmx
  49. */
  50. #define X8(x) x,x,x,x,x,x,x,x
  51. #define ROW_SHIFT 11
  52. #define COL_SHIFT 6
  53. DECLARE_ASM_CONST(16, int16_t, tan1)[] = {X8(13036)}; // tan( pi/16)
  54. DECLARE_ASM_CONST(16, int16_t, tan2)[] = {X8(27146)}; // tan(2pi/16) = sqrt(2)-1
  55. DECLARE_ASM_CONST(16, int16_t, tan3)[] = {X8(43790)}; // tan(3pi/16)-1
  56. DECLARE_ASM_CONST(16, int16_t, sqrt2)[]= {X8(23170)}; // 0.5/sqrt(2)
  57. DECLARE_ASM_CONST(8, uint8_t, m127)[] = {X8(127)};
  58. DECLARE_ASM_CONST(16, int16_t, iTab1)[] = {
  59. 0x4000, 0x539f, 0xc000, 0xac61, 0x4000, 0xdd5d, 0x4000, 0xdd5d,
  60. 0x4000, 0x22a3, 0x4000, 0x22a3, 0xc000, 0x539f, 0x4000, 0xac61,
  61. 0x3249, 0x11a8, 0x4b42, 0xee58, 0x11a8, 0x4b42, 0x11a8, 0xcdb7,
  62. 0x58c5, 0x4b42, 0xa73b, 0xcdb7, 0x3249, 0xa73b, 0x4b42, 0xa73b
  63. };
  64. DECLARE_ASM_CONST(16, int16_t, iTab2)[] = {
  65. 0x58c5, 0x73fc, 0xa73b, 0x8c04, 0x58c5, 0xcff5, 0x58c5, 0xcff5,
  66. 0x58c5, 0x300b, 0x58c5, 0x300b, 0xa73b, 0x73fc, 0x58c5, 0x8c04,
  67. 0x45bf, 0x187e, 0x6862, 0xe782, 0x187e, 0x6862, 0x187e, 0xba41,
  68. 0x7b21, 0x6862, 0x84df, 0xba41, 0x45bf, 0x84df, 0x6862, 0x84df
  69. };
  70. DECLARE_ASM_CONST(16, int16_t, iTab3)[] = {
  71. 0x539f, 0x6d41, 0xac61, 0x92bf, 0x539f, 0xd2bf, 0x539f, 0xd2bf,
  72. 0x539f, 0x2d41, 0x539f, 0x2d41, 0xac61, 0x6d41, 0x539f, 0x92bf,
  73. 0x41b3, 0x1712, 0x6254, 0xe8ee, 0x1712, 0x6254, 0x1712, 0xbe4d,
  74. 0x73fc, 0x6254, 0x8c04, 0xbe4d, 0x41b3, 0x8c04, 0x6254, 0x8c04
  75. };
  76. DECLARE_ASM_CONST(16, int16_t, iTab4)[] = {
  77. 0x4b42, 0x6254, 0xb4be, 0x9dac, 0x4b42, 0xd746, 0x4b42, 0xd746,
  78. 0x4b42, 0x28ba, 0x4b42, 0x28ba, 0xb4be, 0x6254, 0x4b42, 0x9dac,
  79. 0x3b21, 0x14c3, 0x587e, 0xeb3d, 0x14c3, 0x587e, 0x14c3, 0xc4df,
  80. 0x6862, 0x587e, 0x979e, 0xc4df, 0x3b21, 0x979e, 0x587e, 0x979e
  81. };
  82. DECLARE_ASM_CONST(16, int32_t, walkenIdctRounders)[] = {
  83. 65536, 65536, 65536, 65536,
  84. 3597, 3597, 3597, 3597,
  85. 2260, 2260, 2260, 2260,
  86. 1203, 1203, 1203, 1203,
  87. 120, 120, 120, 120,
  88. 512, 512, 512, 512
  89. };
  90. // Temporary storage before the column pass
  91. #define ROW1 "%%xmm6"
  92. #define ROW3 "%%xmm4"
  93. #define ROW5 "%%xmm5"
  94. #define ROW7 "%%xmm7"
  95. #define CLEAR_ODD(r) "pxor "r","r" \n\t"
  96. #define PUT_ODD(dst) "pshufhw $0x1B, %%xmm2, "dst" \n\t"
  97. #if ARCH_X86_64
  98. # define ROW0 "%%xmm8"
  99. # define REG0 ROW0
  100. # define ROW2 "%%xmm9"
  101. # define REG2 ROW2
  102. # define ROW4 "%%xmm10"
  103. # define REG4 ROW4
  104. # define ROW6 "%%xmm11"
  105. # define REG6 ROW6
  106. # define CLEAR_EVEN(r) CLEAR_ODD(r)
  107. # define PUT_EVEN(dst) PUT_ODD(dst)
  108. # define XMMS "%%xmm12"
  109. # define MOV_32_ONLY "#"
  110. # define SREG2 REG2
  111. # define TAN3 "%%xmm13"
  112. # define TAN1 "%%xmm14"
  113. #else
  114. # define ROW0 "(%0)"
  115. # define REG0 "%%xmm4"
  116. # define ROW2 "2*16(%0)"
  117. # define REG2 "%%xmm4"
  118. # define ROW4 "4*16(%0)"
  119. # define REG4 "%%xmm6"
  120. # define ROW6 "6*16(%0)"
  121. # define REG6 "%%xmm6"
  122. # define CLEAR_EVEN(r)
  123. # define PUT_EVEN(dst) \
  124. "pshufhw $0x1B, %%xmm2, %%xmm2 \n\t" \
  125. "movdqa %%xmm2, "dst" \n\t"
  126. # define XMMS "%%xmm2"
  127. # define MOV_32_ONLY "movdqa "
  128. # define SREG2 "%%xmm7"
  129. # define TAN3 "%%xmm0"
  130. # define TAN1 "%%xmm2"
  131. #endif
  132. #define ROUND(x) "paddd "MANGLE(x)
  133. #define JZ(reg, to) \
  134. "testl "reg","reg" \n\t" \
  135. "jz "to" \n\t"
  136. #define JNZ(reg, to) \
  137. "testl "reg","reg" \n\t" \
  138. "jnz "to" \n\t"
  139. #define TEST_ONE_ROW(src, reg, clear) \
  140. clear \
  141. "movq "src", %%mm1 \n\t" \
  142. "por 8+"src", %%mm1 \n\t" \
  143. "paddusb %%mm0, %%mm1 \n\t" \
  144. "pmovmskb %%mm1, "reg" \n\t"
  145. #define TEST_TWO_ROWS(row1, row2, reg1, reg2, clear1, clear2) \
  146. clear1 \
  147. clear2 \
  148. "movq "row1", %%mm1 \n\t" \
  149. "por 8+"row1", %%mm1 \n\t" \
  150. "movq "row2", %%mm2 \n\t" \
  151. "por 8+"row2", %%mm2 \n\t" \
  152. "paddusb %%mm0, %%mm1 \n\t" \
  153. "paddusb %%mm0, %%mm2 \n\t" \
  154. "pmovmskb %%mm1, "reg1" \n\t" \
  155. "pmovmskb %%mm2, "reg2" \n\t"
  156. ///IDCT pass on rows.
  157. #define iMTX_MULT(src, table, rounder, put) \
  158. "movdqa "src", %%xmm3 \n\t" \
  159. "movdqa %%xmm3, %%xmm0 \n\t" \
  160. "pshufd $0x11, %%xmm3, %%xmm1 \n\t" /* 4602 */ \
  161. "punpcklqdq %%xmm0, %%xmm0 \n\t" /* 0246 */ \
  162. "pmaddwd "table", %%xmm0 \n\t" \
  163. "pmaddwd 16+"table", %%xmm1 \n\t" \
  164. "pshufd $0xBB, %%xmm3, %%xmm2 \n\t" /* 5713 */ \
  165. "punpckhqdq %%xmm3, %%xmm3 \n\t" /* 1357 */ \
  166. "pmaddwd 32+"table", %%xmm2 \n\t" \
  167. "pmaddwd 48+"table", %%xmm3 \n\t" \
  168. "paddd %%xmm1, %%xmm0 \n\t" \
  169. "paddd %%xmm3, %%xmm2 \n\t" \
  170. rounder", %%xmm0 \n\t" \
  171. "movdqa %%xmm2, %%xmm3 \n\t" \
  172. "paddd %%xmm0, %%xmm2 \n\t" \
  173. "psubd %%xmm3, %%xmm0 \n\t" \
  174. "psrad $11, %%xmm2 \n\t" \
  175. "psrad $11, %%xmm0 \n\t" \
  176. "packssdw %%xmm0, %%xmm2 \n\t" \
  177. put \
  178. "1: \n\t"
  179. #define iLLM_HEAD \
  180. "movdqa "MANGLE(tan3)", "TAN3" \n\t" \
  181. "movdqa "MANGLE(tan1)", "TAN1" \n\t" \
  182. ///IDCT pass on columns.
  183. #define iLLM_PASS(dct) \
  184. "movdqa "TAN3", %%xmm1 \n\t" \
  185. "movdqa "TAN1", %%xmm3 \n\t" \
  186. "pmulhw %%xmm4, "TAN3" \n\t" \
  187. "pmulhw %%xmm5, %%xmm1 \n\t" \
  188. "paddsw %%xmm4, "TAN3" \n\t" \
  189. "paddsw %%xmm5, %%xmm1 \n\t" \
  190. "psubsw %%xmm5, "TAN3" \n\t" \
  191. "paddsw %%xmm4, %%xmm1 \n\t" \
  192. "pmulhw %%xmm7, %%xmm3 \n\t" \
  193. "pmulhw %%xmm6, "TAN1" \n\t" \
  194. "paddsw %%xmm6, %%xmm3 \n\t" \
  195. "psubsw %%xmm7, "TAN1" \n\t" \
  196. "movdqa %%xmm3, %%xmm7 \n\t" \
  197. "movdqa "TAN1", %%xmm6 \n\t" \
  198. "psubsw %%xmm1, %%xmm3 \n\t" \
  199. "psubsw "TAN3", "TAN1" \n\t" \
  200. "paddsw %%xmm7, %%xmm1 \n\t" \
  201. "paddsw %%xmm6, "TAN3" \n\t" \
  202. "movdqa %%xmm3, %%xmm6 \n\t" \
  203. "psubsw "TAN3", %%xmm3 \n\t" \
  204. "paddsw %%xmm6, "TAN3" \n\t" \
  205. "movdqa "MANGLE(sqrt2)", %%xmm4 \n\t" \
  206. "pmulhw %%xmm4, %%xmm3 \n\t" \
  207. "pmulhw %%xmm4, "TAN3" \n\t" \
  208. "paddsw "TAN3", "TAN3" \n\t" \
  209. "paddsw %%xmm3, %%xmm3 \n\t" \
  210. "movdqa "MANGLE(tan2)", %%xmm7 \n\t" \
  211. MOV_32_ONLY ROW2", "REG2" \n\t" \
  212. MOV_32_ONLY ROW6", "REG6" \n\t" \
  213. "movdqa %%xmm7, %%xmm5 \n\t" \
  214. "pmulhw "REG6", %%xmm7 \n\t" \
  215. "pmulhw "REG2", %%xmm5 \n\t" \
  216. "paddsw "REG2", %%xmm7 \n\t" \
  217. "psubsw "REG6", %%xmm5 \n\t" \
  218. MOV_32_ONLY ROW0", "REG0" \n\t" \
  219. MOV_32_ONLY ROW4", "REG4" \n\t" \
  220. MOV_32_ONLY" "TAN1", (%0) \n\t" \
  221. "movdqa "REG0", "XMMS" \n\t" \
  222. "psubsw "REG4", "REG0" \n\t" \
  223. "paddsw "XMMS", "REG4" \n\t" \
  224. "movdqa "REG4", "XMMS" \n\t" \
  225. "psubsw %%xmm7, "REG4" \n\t" \
  226. "paddsw "XMMS", %%xmm7 \n\t" \
  227. "movdqa "REG0", "XMMS" \n\t" \
  228. "psubsw %%xmm5, "REG0" \n\t" \
  229. "paddsw "XMMS", %%xmm5 \n\t" \
  230. "movdqa %%xmm5, "XMMS" \n\t" \
  231. "psubsw "TAN3", %%xmm5 \n\t" \
  232. "paddsw "XMMS", "TAN3" \n\t" \
  233. "movdqa "REG0", "XMMS" \n\t" \
  234. "psubsw %%xmm3, "REG0" \n\t" \
  235. "paddsw "XMMS", %%xmm3 \n\t" \
  236. MOV_32_ONLY" (%0), "TAN1" \n\t" \
  237. "psraw $6, %%xmm5 \n\t" \
  238. "psraw $6, "REG0" \n\t" \
  239. "psraw $6, "TAN3" \n\t" \
  240. "psraw $6, %%xmm3 \n\t" \
  241. "movdqa "TAN3", 1*16("dct") \n\t" \
  242. "movdqa %%xmm3, 2*16("dct") \n\t" \
  243. "movdqa "REG0", 5*16("dct") \n\t" \
  244. "movdqa %%xmm5, 6*16("dct") \n\t" \
  245. "movdqa %%xmm7, %%xmm0 \n\t" \
  246. "movdqa "REG4", %%xmm4 \n\t" \
  247. "psubsw %%xmm1, %%xmm7 \n\t" \
  248. "psubsw "TAN1", "REG4" \n\t" \
  249. "paddsw %%xmm0, %%xmm1 \n\t" \
  250. "paddsw %%xmm4, "TAN1" \n\t" \
  251. "psraw $6, %%xmm1 \n\t" \
  252. "psraw $6, %%xmm7 \n\t" \
  253. "psraw $6, "TAN1" \n\t" \
  254. "psraw $6, "REG4" \n\t" \
  255. "movdqa %%xmm1, ("dct") \n\t" \
  256. "movdqa "TAN1", 3*16("dct") \n\t" \
  257. "movdqa "REG4", 4*16("dct") \n\t" \
  258. "movdqa %%xmm7, 7*16("dct") \n\t"
  259. ///IDCT pass on columns, assuming rows 4-7 are zero.
  260. #define iLLM_PASS_SPARSE(dct) \
  261. "pmulhw %%xmm4, "TAN3" \n\t" \
  262. "paddsw %%xmm4, "TAN3" \n\t" \
  263. "movdqa %%xmm6, %%xmm3 \n\t" \
  264. "pmulhw %%xmm6, "TAN1" \n\t" \
  265. "movdqa %%xmm4, %%xmm1 \n\t" \
  266. "psubsw %%xmm1, %%xmm3 \n\t" \
  267. "paddsw %%xmm6, %%xmm1 \n\t" \
  268. "movdqa "TAN1", %%xmm6 \n\t" \
  269. "psubsw "TAN3", "TAN1" \n\t" \
  270. "paddsw %%xmm6, "TAN3" \n\t" \
  271. "movdqa %%xmm3, %%xmm6 \n\t" \
  272. "psubsw "TAN3", %%xmm3 \n\t" \
  273. "paddsw %%xmm6, "TAN3" \n\t" \
  274. "movdqa "MANGLE(sqrt2)", %%xmm4 \n\t" \
  275. "pmulhw %%xmm4, %%xmm3 \n\t" \
  276. "pmulhw %%xmm4, "TAN3" \n\t" \
  277. "paddsw "TAN3", "TAN3" \n\t" \
  278. "paddsw %%xmm3, %%xmm3 \n\t" \
  279. "movdqa "MANGLE(tan2)", %%xmm5 \n\t" \
  280. MOV_32_ONLY ROW2", "SREG2" \n\t" \
  281. "pmulhw "SREG2", %%xmm5 \n\t" \
  282. MOV_32_ONLY ROW0", "REG0" \n\t" \
  283. "movdqa "REG0", %%xmm6 \n\t" \
  284. "psubsw "SREG2", %%xmm6 \n\t" \
  285. "paddsw "REG0", "SREG2" \n\t" \
  286. MOV_32_ONLY" "TAN1", (%0) \n\t" \
  287. "movdqa "REG0", "XMMS" \n\t" \
  288. "psubsw %%xmm5, "REG0" \n\t" \
  289. "paddsw "XMMS", %%xmm5 \n\t" \
  290. "movdqa %%xmm5, "XMMS" \n\t" \
  291. "psubsw "TAN3", %%xmm5 \n\t" \
  292. "paddsw "XMMS", "TAN3" \n\t" \
  293. "movdqa "REG0", "XMMS" \n\t" \
  294. "psubsw %%xmm3, "REG0" \n\t" \
  295. "paddsw "XMMS", %%xmm3 \n\t" \
  296. MOV_32_ONLY" (%0), "TAN1" \n\t" \
  297. "psraw $6, %%xmm5 \n\t" \
  298. "psraw $6, "REG0" \n\t" \
  299. "psraw $6, "TAN3" \n\t" \
  300. "psraw $6, %%xmm3 \n\t" \
  301. "movdqa "TAN3", 1*16("dct") \n\t" \
  302. "movdqa %%xmm3, 2*16("dct") \n\t" \
  303. "movdqa "REG0", 5*16("dct") \n\t" \
  304. "movdqa %%xmm5, 6*16("dct") \n\t" \
  305. "movdqa "SREG2", %%xmm0 \n\t" \
  306. "movdqa %%xmm6, %%xmm4 \n\t" \
  307. "psubsw %%xmm1, "SREG2" \n\t" \
  308. "psubsw "TAN1", %%xmm6 \n\t" \
  309. "paddsw %%xmm0, %%xmm1 \n\t" \
  310. "paddsw %%xmm4, "TAN1" \n\t" \
  311. "psraw $6, %%xmm1 \n\t" \
  312. "psraw $6, "SREG2" \n\t" \
  313. "psraw $6, "TAN1" \n\t" \
  314. "psraw $6, %%xmm6 \n\t" \
  315. "movdqa %%xmm1, ("dct") \n\t" \
  316. "movdqa "TAN1", 3*16("dct") \n\t" \
  317. "movdqa %%xmm6, 4*16("dct") \n\t" \
  318. "movdqa "SREG2", 7*16("dct") \n\t"
  319. inline void ff_idct_xvid_sse2(short *block)
  320. {
  321. __asm__ volatile(
  322. "movq "MANGLE(m127)", %%mm0 \n\t"
  323. iMTX_MULT("(%0)", MANGLE(iTab1), ROUND(walkenIdctRounders), PUT_EVEN(ROW0))
  324. iMTX_MULT("1*16(%0)", MANGLE(iTab2), ROUND(walkenIdctRounders+1*16), PUT_ODD(ROW1))
  325. iMTX_MULT("2*16(%0)", MANGLE(iTab3), ROUND(walkenIdctRounders+2*16), PUT_EVEN(ROW2))
  326. TEST_TWO_ROWS("3*16(%0)", "4*16(%0)", "%%eax", "%%ecx", CLEAR_ODD(ROW3), CLEAR_EVEN(ROW4))
  327. JZ("%%eax", "1f")
  328. iMTX_MULT("3*16(%0)", MANGLE(iTab4), ROUND(walkenIdctRounders+3*16), PUT_ODD(ROW3))
  329. TEST_TWO_ROWS("5*16(%0)", "6*16(%0)", "%%eax", "%%edx", CLEAR_ODD(ROW5), CLEAR_EVEN(ROW6))
  330. TEST_ONE_ROW("7*16(%0)", "%%esi", CLEAR_ODD(ROW7))
  331. iLLM_HEAD
  332. ".p2align 4 \n\t"
  333. JNZ("%%ecx", "2f")
  334. JNZ("%%eax", "3f")
  335. JNZ("%%edx", "4f")
  336. JNZ("%%esi", "5f")
  337. iLLM_PASS_SPARSE("%0")
  338. "jmp 6f \n\t"
  339. "2: \n\t"
  340. iMTX_MULT("4*16(%0)", MANGLE(iTab1), "#", PUT_EVEN(ROW4))
  341. "3: \n\t"
  342. iMTX_MULT("5*16(%0)", MANGLE(iTab4), ROUND(walkenIdctRounders+4*16), PUT_ODD(ROW5))
  343. JZ("%%edx", "1f")
  344. "4: \n\t"
  345. iMTX_MULT("6*16(%0)", MANGLE(iTab3), ROUND(walkenIdctRounders+5*16), PUT_EVEN(ROW6))
  346. JZ("%%esi", "1f")
  347. "5: \n\t"
  348. iMTX_MULT("7*16(%0)", MANGLE(iTab2), ROUND(walkenIdctRounders+5*16), PUT_ODD(ROW7))
  349. #if ARCH_X86_32
  350. iLLM_HEAD
  351. #endif
  352. iLLM_PASS("%0")
  353. "6: \n\t"
  354. : "+r"(block)
  355. :
  356. : XMM_CLOBBERS("%xmm0" , "%xmm1" , "%xmm2" , "%xmm3" ,
  357. "%xmm4" , "%xmm5" , "%xmm6" , "%xmm7" ,)
  358. #if ARCH_X86_64
  359. XMM_CLOBBERS("%xmm8" , "%xmm9" , "%xmm10", "%xmm11",
  360. "%xmm12", "%xmm13", "%xmm14",)
  361. #endif
  362. "%eax", "%ecx", "%edx", "%esi", "memory"
  363. );
  364. }
  365. void ff_idct_xvid_sse2_put(uint8_t *dest, int line_size, short *block)
  366. {
  367. ff_idct_xvid_sse2(block);
  368. ff_put_pixels_clamped_mmx(block, dest, line_size);
  369. }
  370. void ff_idct_xvid_sse2_add(uint8_t *dest, int line_size, short *block)
  371. {
  372. ff_idct_xvid_sse2(block);
  373. ff_add_pixels_clamped_mmx(block, dest, line_size);
  374. }
  375. #endif /* HAVE_SSE2_INLINE */