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.

903 lines
39KB

  1. /*
  2. * MMX and SSE2 optimized snow DSP utils
  3. * Copyright (c) 2005-2006 Robert Edele <yartrebo@earthlink.net>
  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 "libavutil/cpu.h"
  22. #include "libavutil/x86_cpu.h"
  23. #include "libavcodec/avcodec.h"
  24. #include "libavcodec/snow.h"
  25. #include "libavcodec/dwt.h"
  26. #include "dsputil_mmx.h"
  27. #if HAVE_INLINE_ASM
  28. static void ff_snow_horizontal_compose97i_sse2(IDWTELEM *b, IDWTELEM *temp, int width){
  29. const int w2= (width+1)>>1;
  30. const int w_l= (width>>1);
  31. const int w_r= w2 - 1;
  32. int i;
  33. { // Lift 0
  34. IDWTELEM * const ref = b + w2 - 1;
  35. IDWTELEM b_0 = b[0]; //By allowing the first entry in b[0] to be calculated twice
  36. // (the first time erroneously), we allow the SSE2 code to run an extra pass.
  37. // The savings in code and time are well worth having to store this value and
  38. // calculate b[0] correctly afterwards.
  39. i = 0;
  40. __asm__ volatile(
  41. "pcmpeqd %%xmm7, %%xmm7 \n\t"
  42. "pcmpeqd %%xmm3, %%xmm3 \n\t"
  43. "psllw $1, %%xmm3 \n\t"
  44. "paddw %%xmm7, %%xmm3 \n\t"
  45. "psllw $13, %%xmm3 \n\t"
  46. ::);
  47. for(; i<w_l-15; i+=16){
  48. __asm__ volatile(
  49. "movdqu (%1), %%xmm1 \n\t"
  50. "movdqu 16(%1), %%xmm5 \n\t"
  51. "movdqu 2(%1), %%xmm2 \n\t"
  52. "movdqu 18(%1), %%xmm6 \n\t"
  53. "paddw %%xmm1, %%xmm2 \n\t"
  54. "paddw %%xmm5, %%xmm6 \n\t"
  55. "paddw %%xmm7, %%xmm2 \n\t"
  56. "paddw %%xmm7, %%xmm6 \n\t"
  57. "pmulhw %%xmm3, %%xmm2 \n\t"
  58. "pmulhw %%xmm3, %%xmm6 \n\t"
  59. "paddw (%0), %%xmm2 \n\t"
  60. "paddw 16(%0), %%xmm6 \n\t"
  61. "movdqa %%xmm2, (%0) \n\t"
  62. "movdqa %%xmm6, 16(%0) \n\t"
  63. :: "r"(&b[i]), "r"(&ref[i])
  64. : "memory"
  65. );
  66. }
  67. snow_horizontal_compose_lift_lead_out(i, b, b, ref, width, w_l, 0, W_DM, W_DO, W_DS);
  68. b[0] = b_0 - ((W_DM * 2 * ref[1]+W_DO)>>W_DS);
  69. }
  70. { // Lift 1
  71. IDWTELEM * const dst = b+w2;
  72. i = 0;
  73. for(; (((x86_reg)&dst[i]) & 0x1F) && i<w_r; i++){
  74. dst[i] = dst[i] - (b[i] + b[i + 1]);
  75. }
  76. for(; i<w_r-15; i+=16){
  77. __asm__ volatile(
  78. "movdqu (%1), %%xmm1 \n\t"
  79. "movdqu 16(%1), %%xmm5 \n\t"
  80. "movdqu 2(%1), %%xmm2 \n\t"
  81. "movdqu 18(%1), %%xmm6 \n\t"
  82. "paddw %%xmm1, %%xmm2 \n\t"
  83. "paddw %%xmm5, %%xmm6 \n\t"
  84. "movdqa (%0), %%xmm0 \n\t"
  85. "movdqa 16(%0), %%xmm4 \n\t"
  86. "psubw %%xmm2, %%xmm0 \n\t"
  87. "psubw %%xmm6, %%xmm4 \n\t"
  88. "movdqa %%xmm0, (%0) \n\t"
  89. "movdqa %%xmm4, 16(%0) \n\t"
  90. :: "r"(&dst[i]), "r"(&b[i])
  91. : "memory"
  92. );
  93. }
  94. snow_horizontal_compose_lift_lead_out(i, dst, dst, b, width, w_r, 1, W_CM, W_CO, W_CS);
  95. }
  96. { // Lift 2
  97. IDWTELEM * const ref = b+w2 - 1;
  98. IDWTELEM b_0 = b[0];
  99. i = 0;
  100. __asm__ volatile(
  101. "psllw $15, %%xmm7 \n\t"
  102. "pcmpeqw %%xmm6, %%xmm6 \n\t"
  103. "psrlw $13, %%xmm6 \n\t"
  104. "paddw %%xmm7, %%xmm6 \n\t"
  105. ::);
  106. for(; i<w_l-15; i+=16){
  107. __asm__ volatile(
  108. "movdqu (%1), %%xmm0 \n\t"
  109. "movdqu 16(%1), %%xmm4 \n\t"
  110. "movdqu 2(%1), %%xmm1 \n\t"
  111. "movdqu 18(%1), %%xmm5 \n\t" //FIXME try aligned reads and shifts
  112. "paddw %%xmm6, %%xmm0 \n\t"
  113. "paddw %%xmm6, %%xmm4 \n\t"
  114. "paddw %%xmm7, %%xmm1 \n\t"
  115. "paddw %%xmm7, %%xmm5 \n\t"
  116. "pavgw %%xmm1, %%xmm0 \n\t"
  117. "pavgw %%xmm5, %%xmm4 \n\t"
  118. "psubw %%xmm7, %%xmm0 \n\t"
  119. "psubw %%xmm7, %%xmm4 \n\t"
  120. "psraw $1, %%xmm0 \n\t"
  121. "psraw $1, %%xmm4 \n\t"
  122. "movdqa (%0), %%xmm1 \n\t"
  123. "movdqa 16(%0), %%xmm5 \n\t"
  124. "paddw %%xmm1, %%xmm0 \n\t"
  125. "paddw %%xmm5, %%xmm4 \n\t"
  126. "psraw $2, %%xmm0 \n\t"
  127. "psraw $2, %%xmm4 \n\t"
  128. "paddw %%xmm1, %%xmm0 \n\t"
  129. "paddw %%xmm5, %%xmm4 \n\t"
  130. "movdqa %%xmm0, (%0) \n\t"
  131. "movdqa %%xmm4, 16(%0) \n\t"
  132. :: "r"(&b[i]), "r"(&ref[i])
  133. : "memory"
  134. );
  135. }
  136. snow_horizontal_compose_liftS_lead_out(i, b, b, ref, width, w_l);
  137. b[0] = b_0 + ((2 * ref[1] + W_BO-1 + 4 * b_0) >> W_BS);
  138. }
  139. { // Lift 3
  140. IDWTELEM * const src = b+w2;
  141. i = 0;
  142. for(; (((x86_reg)&temp[i]) & 0x1F) && i<w_r; i++){
  143. temp[i] = src[i] - ((-W_AM*(b[i] + b[i+1]))>>W_AS);
  144. }
  145. for(; i<w_r-7; i+=8){
  146. __asm__ volatile(
  147. "movdqu 2(%1), %%xmm2 \n\t"
  148. "movdqu 18(%1), %%xmm6 \n\t"
  149. "paddw (%1), %%xmm2 \n\t"
  150. "paddw 16(%1), %%xmm6 \n\t"
  151. "movdqu (%0), %%xmm0 \n\t"
  152. "movdqu 16(%0), %%xmm4 \n\t"
  153. "paddw %%xmm2, %%xmm0 \n\t"
  154. "paddw %%xmm6, %%xmm4 \n\t"
  155. "psraw $1, %%xmm2 \n\t"
  156. "psraw $1, %%xmm6 \n\t"
  157. "paddw %%xmm0, %%xmm2 \n\t"
  158. "paddw %%xmm4, %%xmm6 \n\t"
  159. "movdqa %%xmm2, (%2) \n\t"
  160. "movdqa %%xmm6, 16(%2) \n\t"
  161. :: "r"(&src[i]), "r"(&b[i]), "r"(&temp[i])
  162. : "memory"
  163. );
  164. }
  165. snow_horizontal_compose_lift_lead_out(i, temp, src, b, width, w_r, 1, -W_AM, W_AO+1, W_AS);
  166. }
  167. {
  168. snow_interleave_line_header(&i, width, b, temp);
  169. for (; (i & 0x3E) != 0x3E; i-=2){
  170. b[i+1] = temp[i>>1];
  171. b[i] = b[i>>1];
  172. }
  173. for (i-=62; i>=0; i-=64){
  174. __asm__ volatile(
  175. "movdqa (%1), %%xmm0 \n\t"
  176. "movdqa 16(%1), %%xmm2 \n\t"
  177. "movdqa 32(%1), %%xmm4 \n\t"
  178. "movdqa 48(%1), %%xmm6 \n\t"
  179. "movdqa (%1), %%xmm1 \n\t"
  180. "movdqa 16(%1), %%xmm3 \n\t"
  181. "movdqa 32(%1), %%xmm5 \n\t"
  182. "movdqa 48(%1), %%xmm7 \n\t"
  183. "punpcklwd (%2), %%xmm0 \n\t"
  184. "punpcklwd 16(%2), %%xmm2 \n\t"
  185. "punpcklwd 32(%2), %%xmm4 \n\t"
  186. "punpcklwd 48(%2), %%xmm6 \n\t"
  187. "movdqa %%xmm0, (%0) \n\t"
  188. "movdqa %%xmm2, 32(%0) \n\t"
  189. "movdqa %%xmm4, 64(%0) \n\t"
  190. "movdqa %%xmm6, 96(%0) \n\t"
  191. "punpckhwd (%2), %%xmm1 \n\t"
  192. "punpckhwd 16(%2), %%xmm3 \n\t"
  193. "punpckhwd 32(%2), %%xmm5 \n\t"
  194. "punpckhwd 48(%2), %%xmm7 \n\t"
  195. "movdqa %%xmm1, 16(%0) \n\t"
  196. "movdqa %%xmm3, 48(%0) \n\t"
  197. "movdqa %%xmm5, 80(%0) \n\t"
  198. "movdqa %%xmm7, 112(%0) \n\t"
  199. :: "r"(&(b)[i]), "r"(&(b)[i>>1]), "r"(&(temp)[i>>1])
  200. : "memory"
  201. );
  202. }
  203. }
  204. }
  205. static void ff_snow_horizontal_compose97i_mmx(IDWTELEM *b, IDWTELEM *temp, int width){
  206. const int w2= (width+1)>>1;
  207. const int w_l= (width>>1);
  208. const int w_r= w2 - 1;
  209. int i;
  210. { // Lift 0
  211. IDWTELEM * const ref = b + w2 - 1;
  212. i = 1;
  213. b[0] = b[0] - ((W_DM * 2 * ref[1]+W_DO)>>W_DS);
  214. __asm__ volatile(
  215. "pcmpeqw %%mm7, %%mm7 \n\t"
  216. "pcmpeqw %%mm3, %%mm3 \n\t"
  217. "psllw $1, %%mm3 \n\t"
  218. "paddw %%mm7, %%mm3 \n\t"
  219. "psllw $13, %%mm3 \n\t"
  220. ::);
  221. for(; i<w_l-7; i+=8){
  222. __asm__ volatile(
  223. "movq (%1), %%mm2 \n\t"
  224. "movq 8(%1), %%mm6 \n\t"
  225. "paddw 2(%1), %%mm2 \n\t"
  226. "paddw 10(%1), %%mm6 \n\t"
  227. "paddw %%mm7, %%mm2 \n\t"
  228. "paddw %%mm7, %%mm6 \n\t"
  229. "pmulhw %%mm3, %%mm2 \n\t"
  230. "pmulhw %%mm3, %%mm6 \n\t"
  231. "paddw (%0), %%mm2 \n\t"
  232. "paddw 8(%0), %%mm6 \n\t"
  233. "movq %%mm2, (%0) \n\t"
  234. "movq %%mm6, 8(%0) \n\t"
  235. :: "r"(&b[i]), "r"(&ref[i])
  236. : "memory"
  237. );
  238. }
  239. snow_horizontal_compose_lift_lead_out(i, b, b, ref, width, w_l, 0, W_DM, W_DO, W_DS);
  240. }
  241. { // Lift 1
  242. IDWTELEM * const dst = b+w2;
  243. i = 0;
  244. for(; i<w_r-7; i+=8){
  245. __asm__ volatile(
  246. "movq (%1), %%mm2 \n\t"
  247. "movq 8(%1), %%mm6 \n\t"
  248. "paddw 2(%1), %%mm2 \n\t"
  249. "paddw 10(%1), %%mm6 \n\t"
  250. "movq (%0), %%mm0 \n\t"
  251. "movq 8(%0), %%mm4 \n\t"
  252. "psubw %%mm2, %%mm0 \n\t"
  253. "psubw %%mm6, %%mm4 \n\t"
  254. "movq %%mm0, (%0) \n\t"
  255. "movq %%mm4, 8(%0) \n\t"
  256. :: "r"(&dst[i]), "r"(&b[i])
  257. : "memory"
  258. );
  259. }
  260. snow_horizontal_compose_lift_lead_out(i, dst, dst, b, width, w_r, 1, W_CM, W_CO, W_CS);
  261. }
  262. { // Lift 2
  263. IDWTELEM * const ref = b+w2 - 1;
  264. i = 1;
  265. b[0] = b[0] + (((2 * ref[1] + W_BO) + 4 * b[0]) >> W_BS);
  266. __asm__ volatile(
  267. "psllw $15, %%mm7 \n\t"
  268. "pcmpeqw %%mm6, %%mm6 \n\t"
  269. "psrlw $13, %%mm6 \n\t"
  270. "paddw %%mm7, %%mm6 \n\t"
  271. ::);
  272. for(; i<w_l-7; i+=8){
  273. __asm__ volatile(
  274. "movq (%1), %%mm0 \n\t"
  275. "movq 8(%1), %%mm4 \n\t"
  276. "movq 2(%1), %%mm1 \n\t"
  277. "movq 10(%1), %%mm5 \n\t"
  278. "paddw %%mm6, %%mm0 \n\t"
  279. "paddw %%mm6, %%mm4 \n\t"
  280. "paddw %%mm7, %%mm1 \n\t"
  281. "paddw %%mm7, %%mm5 \n\t"
  282. "pavgw %%mm1, %%mm0 \n\t"
  283. "pavgw %%mm5, %%mm4 \n\t"
  284. "psubw %%mm7, %%mm0 \n\t"
  285. "psubw %%mm7, %%mm4 \n\t"
  286. "psraw $1, %%mm0 \n\t"
  287. "psraw $1, %%mm4 \n\t"
  288. "movq (%0), %%mm1 \n\t"
  289. "movq 8(%0), %%mm5 \n\t"
  290. "paddw %%mm1, %%mm0 \n\t"
  291. "paddw %%mm5, %%mm4 \n\t"
  292. "psraw $2, %%mm0 \n\t"
  293. "psraw $2, %%mm4 \n\t"
  294. "paddw %%mm1, %%mm0 \n\t"
  295. "paddw %%mm5, %%mm4 \n\t"
  296. "movq %%mm0, (%0) \n\t"
  297. "movq %%mm4, 8(%0) \n\t"
  298. :: "r"(&b[i]), "r"(&ref[i])
  299. : "memory"
  300. );
  301. }
  302. snow_horizontal_compose_liftS_lead_out(i, b, b, ref, width, w_l);
  303. }
  304. { // Lift 3
  305. IDWTELEM * const src = b+w2;
  306. i = 0;
  307. for(; i<w_r-7; i+=8){
  308. __asm__ volatile(
  309. "movq 2(%1), %%mm2 \n\t"
  310. "movq 10(%1), %%mm6 \n\t"
  311. "paddw (%1), %%mm2 \n\t"
  312. "paddw 8(%1), %%mm6 \n\t"
  313. "movq (%0), %%mm0 \n\t"
  314. "movq 8(%0), %%mm4 \n\t"
  315. "paddw %%mm2, %%mm0 \n\t"
  316. "paddw %%mm6, %%mm4 \n\t"
  317. "psraw $1, %%mm2 \n\t"
  318. "psraw $1, %%mm6 \n\t"
  319. "paddw %%mm0, %%mm2 \n\t"
  320. "paddw %%mm4, %%mm6 \n\t"
  321. "movq %%mm2, (%2) \n\t"
  322. "movq %%mm6, 8(%2) \n\t"
  323. :: "r"(&src[i]), "r"(&b[i]), "r"(&temp[i])
  324. : "memory"
  325. );
  326. }
  327. snow_horizontal_compose_lift_lead_out(i, temp, src, b, width, w_r, 1, -W_AM, W_AO+1, W_AS);
  328. }
  329. {
  330. snow_interleave_line_header(&i, width, b, temp);
  331. for (; (i & 0x1E) != 0x1E; i-=2){
  332. b[i+1] = temp[i>>1];
  333. b[i] = b[i>>1];
  334. }
  335. for (i-=30; i>=0; i-=32){
  336. __asm__ volatile(
  337. "movq (%1), %%mm0 \n\t"
  338. "movq 8(%1), %%mm2 \n\t"
  339. "movq 16(%1), %%mm4 \n\t"
  340. "movq 24(%1), %%mm6 \n\t"
  341. "movq (%1), %%mm1 \n\t"
  342. "movq 8(%1), %%mm3 \n\t"
  343. "movq 16(%1), %%mm5 \n\t"
  344. "movq 24(%1), %%mm7 \n\t"
  345. "punpcklwd (%2), %%mm0 \n\t"
  346. "punpcklwd 8(%2), %%mm2 \n\t"
  347. "punpcklwd 16(%2), %%mm4 \n\t"
  348. "punpcklwd 24(%2), %%mm6 \n\t"
  349. "movq %%mm0, (%0) \n\t"
  350. "movq %%mm2, 16(%0) \n\t"
  351. "movq %%mm4, 32(%0) \n\t"
  352. "movq %%mm6, 48(%0) \n\t"
  353. "punpckhwd (%2), %%mm1 \n\t"
  354. "punpckhwd 8(%2), %%mm3 \n\t"
  355. "punpckhwd 16(%2), %%mm5 \n\t"
  356. "punpckhwd 24(%2), %%mm7 \n\t"
  357. "movq %%mm1, 8(%0) \n\t"
  358. "movq %%mm3, 24(%0) \n\t"
  359. "movq %%mm5, 40(%0) \n\t"
  360. "movq %%mm7, 56(%0) \n\t"
  361. :: "r"(&b[i]), "r"(&b[i>>1]), "r"(&temp[i>>1])
  362. : "memory"
  363. );
  364. }
  365. }
  366. }
  367. #if HAVE_7REGS
  368. #define snow_vertical_compose_sse2_load_add(op,r,t0,t1,t2,t3)\
  369. ""op" ("r",%%"REG_d"), %%"t0" \n\t"\
  370. ""op" 16("r",%%"REG_d"), %%"t1" \n\t"\
  371. ""op" 32("r",%%"REG_d"), %%"t2" \n\t"\
  372. ""op" 48("r",%%"REG_d"), %%"t3" \n\t"
  373. #define snow_vertical_compose_sse2_load(r,t0,t1,t2,t3)\
  374. snow_vertical_compose_sse2_load_add("movdqa",r,t0,t1,t2,t3)
  375. #define snow_vertical_compose_sse2_add(r,t0,t1,t2,t3)\
  376. snow_vertical_compose_sse2_load_add("paddw",r,t0,t1,t2,t3)
  377. #define snow_vertical_compose_r2r_sub(s0,s1,s2,s3,t0,t1,t2,t3)\
  378. "psubw %%"s0", %%"t0" \n\t"\
  379. "psubw %%"s1", %%"t1" \n\t"\
  380. "psubw %%"s2", %%"t2" \n\t"\
  381. "psubw %%"s3", %%"t3" \n\t"
  382. #define snow_vertical_compose_sse2_store(w,s0,s1,s2,s3)\
  383. "movdqa %%"s0", ("w",%%"REG_d") \n\t"\
  384. "movdqa %%"s1", 16("w",%%"REG_d") \n\t"\
  385. "movdqa %%"s2", 32("w",%%"REG_d") \n\t"\
  386. "movdqa %%"s3", 48("w",%%"REG_d") \n\t"
  387. #define snow_vertical_compose_sra(n,t0,t1,t2,t3)\
  388. "psraw $"n", %%"t0" \n\t"\
  389. "psraw $"n", %%"t1" \n\t"\
  390. "psraw $"n", %%"t2" \n\t"\
  391. "psraw $"n", %%"t3" \n\t"
  392. #define snow_vertical_compose_r2r_add(s0,s1,s2,s3,t0,t1,t2,t3)\
  393. "paddw %%"s0", %%"t0" \n\t"\
  394. "paddw %%"s1", %%"t1" \n\t"\
  395. "paddw %%"s2", %%"t2" \n\t"\
  396. "paddw %%"s3", %%"t3" \n\t"
  397. #define snow_vertical_compose_r2r_pmulhw(s0,s1,s2,s3,t0,t1,t2,t3)\
  398. "pmulhw %%"s0", %%"t0" \n\t"\
  399. "pmulhw %%"s1", %%"t1" \n\t"\
  400. "pmulhw %%"s2", %%"t2" \n\t"\
  401. "pmulhw %%"s3", %%"t3" \n\t"
  402. #define snow_vertical_compose_sse2_move(s0,s1,s2,s3,t0,t1,t2,t3)\
  403. "movdqa %%"s0", %%"t0" \n\t"\
  404. "movdqa %%"s1", %%"t1" \n\t"\
  405. "movdqa %%"s2", %%"t2" \n\t"\
  406. "movdqa %%"s3", %%"t3" \n\t"
  407. static void ff_snow_vertical_compose97i_sse2(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5, int width){
  408. x86_reg i = width;
  409. while(i & 0x1F)
  410. {
  411. i--;
  412. b4[i] -= (W_DM*(b3[i] + b5[i])+W_DO)>>W_DS;
  413. b3[i] -= (W_CM*(b2[i] + b4[i])+W_CO)>>W_CS;
  414. b2[i] += (W_BM*(b1[i] + b3[i])+4*b2[i]+W_BO)>>W_BS;
  415. b1[i] += (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
  416. }
  417. i+=i;
  418. __asm__ volatile (
  419. "jmp 2f \n\t"
  420. "1: \n\t"
  421. snow_vertical_compose_sse2_load("%4","xmm0","xmm2","xmm4","xmm6")
  422. snow_vertical_compose_sse2_add("%6","xmm0","xmm2","xmm4","xmm6")
  423. "pcmpeqw %%xmm0, %%xmm0 \n\t"
  424. "pcmpeqw %%xmm2, %%xmm2 \n\t"
  425. "paddw %%xmm2, %%xmm2 \n\t"
  426. "paddw %%xmm0, %%xmm2 \n\t"
  427. "psllw $13, %%xmm2 \n\t"
  428. snow_vertical_compose_r2r_add("xmm0","xmm0","xmm0","xmm0","xmm1","xmm3","xmm5","xmm7")
  429. snow_vertical_compose_r2r_pmulhw("xmm2","xmm2","xmm2","xmm2","xmm1","xmm3","xmm5","xmm7")
  430. snow_vertical_compose_sse2_add("%5","xmm1","xmm3","xmm5","xmm7")
  431. snow_vertical_compose_sse2_store("%5","xmm1","xmm3","xmm5","xmm7")
  432. snow_vertical_compose_sse2_load("%4","xmm0","xmm2","xmm4","xmm6")
  433. snow_vertical_compose_sse2_add("%3","xmm1","xmm3","xmm5","xmm7")
  434. snow_vertical_compose_r2r_sub("xmm1","xmm3","xmm5","xmm7","xmm0","xmm2","xmm4","xmm6")
  435. snow_vertical_compose_sse2_store("%4","xmm0","xmm2","xmm4","xmm6")
  436. "pcmpeqw %%xmm7, %%xmm7 \n\t"
  437. "pcmpeqw %%xmm5, %%xmm5 \n\t"
  438. "psllw $15, %%xmm7 \n\t"
  439. "psrlw $13, %%xmm5 \n\t"
  440. "paddw %%xmm7, %%xmm5 \n\t"
  441. snow_vertical_compose_r2r_add("xmm5","xmm5","xmm5","xmm5","xmm0","xmm2","xmm4","xmm6")
  442. "movq (%2,%%"REG_d"), %%xmm1 \n\t"
  443. "movq 8(%2,%%"REG_d"), %%xmm3 \n\t"
  444. "paddw %%xmm7, %%xmm1 \n\t"
  445. "paddw %%xmm7, %%xmm3 \n\t"
  446. "pavgw %%xmm1, %%xmm0 \n\t"
  447. "pavgw %%xmm3, %%xmm2 \n\t"
  448. "movq 16(%2,%%"REG_d"), %%xmm1 \n\t"
  449. "movq 24(%2,%%"REG_d"), %%xmm3 \n\t"
  450. "paddw %%xmm7, %%xmm1 \n\t"
  451. "paddw %%xmm7, %%xmm3 \n\t"
  452. "pavgw %%xmm1, %%xmm4 \n\t"
  453. "pavgw %%xmm3, %%xmm6 \n\t"
  454. snow_vertical_compose_r2r_sub("xmm7","xmm7","xmm7","xmm7","xmm0","xmm2","xmm4","xmm6")
  455. snow_vertical_compose_sra("1","xmm0","xmm2","xmm4","xmm6")
  456. snow_vertical_compose_sse2_add("%3","xmm0","xmm2","xmm4","xmm6")
  457. snow_vertical_compose_sra("2","xmm0","xmm2","xmm4","xmm6")
  458. snow_vertical_compose_sse2_add("%3","xmm0","xmm2","xmm4","xmm6")
  459. snow_vertical_compose_sse2_store("%3","xmm0","xmm2","xmm4","xmm6")
  460. snow_vertical_compose_sse2_add("%1","xmm0","xmm2","xmm4","xmm6")
  461. snow_vertical_compose_sse2_move("xmm0","xmm2","xmm4","xmm6","xmm1","xmm3","xmm5","xmm7")
  462. snow_vertical_compose_sra("1","xmm0","xmm2","xmm4","xmm6")
  463. snow_vertical_compose_r2r_add("xmm1","xmm3","xmm5","xmm7","xmm0","xmm2","xmm4","xmm6")
  464. snow_vertical_compose_sse2_add("%2","xmm0","xmm2","xmm4","xmm6")
  465. snow_vertical_compose_sse2_store("%2","xmm0","xmm2","xmm4","xmm6")
  466. "2: \n\t"
  467. "sub $64, %%"REG_d" \n\t"
  468. "jge 1b \n\t"
  469. :"+d"(i)
  470. :"r"(b0),"r"(b1),"r"(b2),"r"(b3),"r"(b4),"r"(b5));
  471. }
  472. #define snow_vertical_compose_mmx_load_add(op,r,t0,t1,t2,t3)\
  473. ""op" ("r",%%"REG_d"), %%"t0" \n\t"\
  474. ""op" 8("r",%%"REG_d"), %%"t1" \n\t"\
  475. ""op" 16("r",%%"REG_d"), %%"t2" \n\t"\
  476. ""op" 24("r",%%"REG_d"), %%"t3" \n\t"
  477. #define snow_vertical_compose_mmx_load(r,t0,t1,t2,t3)\
  478. snow_vertical_compose_mmx_load_add("movq",r,t0,t1,t2,t3)
  479. #define snow_vertical_compose_mmx_add(r,t0,t1,t2,t3)\
  480. snow_vertical_compose_mmx_load_add("paddw",r,t0,t1,t2,t3)
  481. #define snow_vertical_compose_mmx_store(w,s0,s1,s2,s3)\
  482. "movq %%"s0", ("w",%%"REG_d") \n\t"\
  483. "movq %%"s1", 8("w",%%"REG_d") \n\t"\
  484. "movq %%"s2", 16("w",%%"REG_d") \n\t"\
  485. "movq %%"s3", 24("w",%%"REG_d") \n\t"
  486. #define snow_vertical_compose_mmx_move(s0,s1,s2,s3,t0,t1,t2,t3)\
  487. "movq %%"s0", %%"t0" \n\t"\
  488. "movq %%"s1", %%"t1" \n\t"\
  489. "movq %%"s2", %%"t2" \n\t"\
  490. "movq %%"s3", %%"t3" \n\t"
  491. static void ff_snow_vertical_compose97i_mmx(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5, int width){
  492. x86_reg i = width;
  493. while(i & 15)
  494. {
  495. i--;
  496. b4[i] -= (W_DM*(b3[i] + b5[i])+W_DO)>>W_DS;
  497. b3[i] -= (W_CM*(b2[i] + b4[i])+W_CO)>>W_CS;
  498. b2[i] += (W_BM*(b1[i] + b3[i])+4*b2[i]+W_BO)>>W_BS;
  499. b1[i] += (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
  500. }
  501. i+=i;
  502. __asm__ volatile(
  503. "jmp 2f \n\t"
  504. "1: \n\t"
  505. snow_vertical_compose_mmx_load("%4","mm1","mm3","mm5","mm7")
  506. snow_vertical_compose_mmx_add("%6","mm1","mm3","mm5","mm7")
  507. "pcmpeqw %%mm0, %%mm0 \n\t"
  508. "pcmpeqw %%mm2, %%mm2 \n\t"
  509. "paddw %%mm2, %%mm2 \n\t"
  510. "paddw %%mm0, %%mm2 \n\t"
  511. "psllw $13, %%mm2 \n\t"
  512. snow_vertical_compose_r2r_add("mm0","mm0","mm0","mm0","mm1","mm3","mm5","mm7")
  513. snow_vertical_compose_r2r_pmulhw("mm2","mm2","mm2","mm2","mm1","mm3","mm5","mm7")
  514. snow_vertical_compose_mmx_add("%5","mm1","mm3","mm5","mm7")
  515. snow_vertical_compose_mmx_store("%5","mm1","mm3","mm5","mm7")
  516. snow_vertical_compose_mmx_load("%4","mm0","mm2","mm4","mm6")
  517. snow_vertical_compose_mmx_add("%3","mm1","mm3","mm5","mm7")
  518. snow_vertical_compose_r2r_sub("mm1","mm3","mm5","mm7","mm0","mm2","mm4","mm6")
  519. snow_vertical_compose_mmx_store("%4","mm0","mm2","mm4","mm6")
  520. "pcmpeqw %%mm7, %%mm7 \n\t"
  521. "pcmpeqw %%mm5, %%mm5 \n\t"
  522. "psllw $15, %%mm7 \n\t"
  523. "psrlw $13, %%mm5 \n\t"
  524. "paddw %%mm7, %%mm5 \n\t"
  525. snow_vertical_compose_r2r_add("mm5","mm5","mm5","mm5","mm0","mm2","mm4","mm6")
  526. "movq (%2,%%"REG_d"), %%mm1 \n\t"
  527. "movq 8(%2,%%"REG_d"), %%mm3 \n\t"
  528. "paddw %%mm7, %%mm1 \n\t"
  529. "paddw %%mm7, %%mm3 \n\t"
  530. "pavgw %%mm1, %%mm0 \n\t"
  531. "pavgw %%mm3, %%mm2 \n\t"
  532. "movq 16(%2,%%"REG_d"), %%mm1 \n\t"
  533. "movq 24(%2,%%"REG_d"), %%mm3 \n\t"
  534. "paddw %%mm7, %%mm1 \n\t"
  535. "paddw %%mm7, %%mm3 \n\t"
  536. "pavgw %%mm1, %%mm4 \n\t"
  537. "pavgw %%mm3, %%mm6 \n\t"
  538. snow_vertical_compose_r2r_sub("mm7","mm7","mm7","mm7","mm0","mm2","mm4","mm6")
  539. snow_vertical_compose_sra("1","mm0","mm2","mm4","mm6")
  540. snow_vertical_compose_mmx_add("%3","mm0","mm2","mm4","mm6")
  541. snow_vertical_compose_sra("2","mm0","mm2","mm4","mm6")
  542. snow_vertical_compose_mmx_add("%3","mm0","mm2","mm4","mm6")
  543. snow_vertical_compose_mmx_store("%3","mm0","mm2","mm4","mm6")
  544. snow_vertical_compose_mmx_add("%1","mm0","mm2","mm4","mm6")
  545. snow_vertical_compose_mmx_move("mm0","mm2","mm4","mm6","mm1","mm3","mm5","mm7")
  546. snow_vertical_compose_sra("1","mm0","mm2","mm4","mm6")
  547. snow_vertical_compose_r2r_add("mm1","mm3","mm5","mm7","mm0","mm2","mm4","mm6")
  548. snow_vertical_compose_mmx_add("%2","mm0","mm2","mm4","mm6")
  549. snow_vertical_compose_mmx_store("%2","mm0","mm2","mm4","mm6")
  550. "2: \n\t"
  551. "sub $32, %%"REG_d" \n\t"
  552. "jge 1b \n\t"
  553. :"+d"(i)
  554. :"r"(b0),"r"(b1),"r"(b2),"r"(b3),"r"(b4),"r"(b5));
  555. }
  556. #endif //HAVE_7REGS
  557. #define snow_inner_add_yblock_sse2_header \
  558. IDWTELEM * * dst_array = sb->line + src_y;\
  559. x86_reg tmp;\
  560. __asm__ volatile(\
  561. "mov %7, %%"REG_c" \n\t"\
  562. "mov %6, %2 \n\t"\
  563. "mov %4, %%"REG_S" \n\t"\
  564. "pxor %%xmm7, %%xmm7 \n\t" /* 0 */\
  565. "pcmpeqd %%xmm3, %%xmm3 \n\t"\
  566. "psllw $15, %%xmm3 \n\t"\
  567. "psrlw $12, %%xmm3 \n\t" /* FRAC_BITS >> 1 */\
  568. "1: \n\t"\
  569. "mov %1, %%"REG_D" \n\t"\
  570. "mov (%%"REG_D"), %%"REG_D" \n\t"\
  571. "add %3, %%"REG_D" \n\t"
  572. #define snow_inner_add_yblock_sse2_start_8(out_reg1, out_reg2, ptr_offset, s_offset)\
  573. "mov "PTR_SIZE"*"ptr_offset"(%%"REG_a"), %%"REG_d"; \n\t"\
  574. "movq (%%"REG_d"), %%"out_reg1" \n\t"\
  575. "movq (%%"REG_d", %%"REG_c"), %%"out_reg2" \n\t"\
  576. "punpcklbw %%xmm7, %%"out_reg1" \n\t"\
  577. "punpcklbw %%xmm7, %%"out_reg2" \n\t"\
  578. "movq "s_offset"(%%"REG_S"), %%xmm0 \n\t"\
  579. "movq "s_offset"+16(%%"REG_S"), %%xmm4 \n\t"\
  580. "punpcklbw %%xmm7, %%xmm0 \n\t"\
  581. "punpcklbw %%xmm7, %%xmm4 \n\t"\
  582. "pmullw %%xmm0, %%"out_reg1" \n\t"\
  583. "pmullw %%xmm4, %%"out_reg2" \n\t"
  584. #define snow_inner_add_yblock_sse2_start_16(out_reg1, out_reg2, ptr_offset, s_offset)\
  585. "mov "PTR_SIZE"*"ptr_offset"(%%"REG_a"), %%"REG_d"; \n\t"\
  586. "movq (%%"REG_d"), %%"out_reg1" \n\t"\
  587. "movq 8(%%"REG_d"), %%"out_reg2" \n\t"\
  588. "punpcklbw %%xmm7, %%"out_reg1" \n\t"\
  589. "punpcklbw %%xmm7, %%"out_reg2" \n\t"\
  590. "movq "s_offset"(%%"REG_S"), %%xmm0 \n\t"\
  591. "movq "s_offset"+8(%%"REG_S"), %%xmm4 \n\t"\
  592. "punpcklbw %%xmm7, %%xmm0 \n\t"\
  593. "punpcklbw %%xmm7, %%xmm4 \n\t"\
  594. "pmullw %%xmm0, %%"out_reg1" \n\t"\
  595. "pmullw %%xmm4, %%"out_reg2" \n\t"
  596. #define snow_inner_add_yblock_sse2_accum_8(ptr_offset, s_offset) \
  597. snow_inner_add_yblock_sse2_start_8("xmm2", "xmm6", ptr_offset, s_offset)\
  598. "paddusw %%xmm2, %%xmm1 \n\t"\
  599. "paddusw %%xmm6, %%xmm5 \n\t"
  600. #define snow_inner_add_yblock_sse2_accum_16(ptr_offset, s_offset) \
  601. snow_inner_add_yblock_sse2_start_16("xmm2", "xmm6", ptr_offset, s_offset)\
  602. "paddusw %%xmm2, %%xmm1 \n\t"\
  603. "paddusw %%xmm6, %%xmm5 \n\t"
  604. #define snow_inner_add_yblock_sse2_end_common1\
  605. "add $32, %%"REG_S" \n\t"\
  606. "add %%"REG_c", %0 \n\t"\
  607. "add %%"REG_c", "PTR_SIZE"*3(%%"REG_a");\n\t"\
  608. "add %%"REG_c", "PTR_SIZE"*2(%%"REG_a");\n\t"\
  609. "add %%"REG_c", "PTR_SIZE"*1(%%"REG_a");\n\t"\
  610. "add %%"REG_c", (%%"REG_a") \n\t"
  611. #define snow_inner_add_yblock_sse2_end_common2\
  612. "jnz 1b \n\t"\
  613. :"+m"(dst8),"+m"(dst_array),"=&r"(tmp)\
  614. :\
  615. "rm"((x86_reg)(src_x<<1)),"m"(obmc),"a"(block),"m"(b_h),"m"(src_stride):\
  616. "%"REG_c"","%"REG_S"","%"REG_D"","%"REG_d"");
  617. #define snow_inner_add_yblock_sse2_end_8\
  618. "sal $1, %%"REG_c" \n\t"\
  619. "addl $"PTR_SIZE"*2, %1 \n\t"\
  620. snow_inner_add_yblock_sse2_end_common1\
  621. "sar $1, %%"REG_c" \n\t"\
  622. "sub $2, %2 \n\t"\
  623. snow_inner_add_yblock_sse2_end_common2
  624. #define snow_inner_add_yblock_sse2_end_16\
  625. "addl $"PTR_SIZE"*1, %1 \n\t"\
  626. snow_inner_add_yblock_sse2_end_common1\
  627. "dec %2 \n\t"\
  628. snow_inner_add_yblock_sse2_end_common2
  629. static void inner_add_yblock_bw_8_obmc_16_bh_even_sse2(const uint8_t *obmc, const x86_reg obmc_stride, uint8_t * * block, int b_w, x86_reg b_h,
  630. int src_x, int src_y, x86_reg src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  631. snow_inner_add_yblock_sse2_header
  632. snow_inner_add_yblock_sse2_start_8("xmm1", "xmm5", "3", "0")
  633. snow_inner_add_yblock_sse2_accum_8("2", "8")
  634. snow_inner_add_yblock_sse2_accum_8("1", "128")
  635. snow_inner_add_yblock_sse2_accum_8("0", "136")
  636. "mov %0, %%"REG_d" \n\t"
  637. "movdqa (%%"REG_D"), %%xmm0 \n\t"
  638. "movdqa %%xmm1, %%xmm2 \n\t"
  639. "punpckhwd %%xmm7, %%xmm1 \n\t"
  640. "punpcklwd %%xmm7, %%xmm2 \n\t"
  641. "paddd %%xmm2, %%xmm0 \n\t"
  642. "movdqa 16(%%"REG_D"), %%xmm2 \n\t"
  643. "paddd %%xmm1, %%xmm2 \n\t"
  644. "paddd %%xmm3, %%xmm0 \n\t"
  645. "paddd %%xmm3, %%xmm2 \n\t"
  646. "mov %1, %%"REG_D" \n\t"
  647. "mov "PTR_SIZE"(%%"REG_D"), %%"REG_D";\n\t"
  648. "add %3, %%"REG_D" \n\t"
  649. "movdqa (%%"REG_D"), %%xmm4 \n\t"
  650. "movdqa %%xmm5, %%xmm6 \n\t"
  651. "punpckhwd %%xmm7, %%xmm5 \n\t"
  652. "punpcklwd %%xmm7, %%xmm6 \n\t"
  653. "paddd %%xmm6, %%xmm4 \n\t"
  654. "movdqa 16(%%"REG_D"), %%xmm6 \n\t"
  655. "paddd %%xmm5, %%xmm6 \n\t"
  656. "paddd %%xmm3, %%xmm4 \n\t"
  657. "paddd %%xmm3, %%xmm6 \n\t"
  658. "psrad $8, %%xmm0 \n\t" /* FRAC_BITS. */
  659. "psrad $8, %%xmm2 \n\t" /* FRAC_BITS. */
  660. "packssdw %%xmm2, %%xmm0 \n\t"
  661. "packuswb %%xmm7, %%xmm0 \n\t"
  662. "movq %%xmm0, (%%"REG_d") \n\t"
  663. "psrad $8, %%xmm4 \n\t" /* FRAC_BITS. */
  664. "psrad $8, %%xmm6 \n\t" /* FRAC_BITS. */
  665. "packssdw %%xmm6, %%xmm4 \n\t"
  666. "packuswb %%xmm7, %%xmm4 \n\t"
  667. "movq %%xmm4, (%%"REG_d",%%"REG_c");\n\t"
  668. snow_inner_add_yblock_sse2_end_8
  669. }
  670. static void inner_add_yblock_bw_16_obmc_32_sse2(const uint8_t *obmc, const x86_reg obmc_stride, uint8_t * * block, int b_w, x86_reg b_h,
  671. int src_x, int src_y, x86_reg src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  672. snow_inner_add_yblock_sse2_header
  673. snow_inner_add_yblock_sse2_start_16("xmm1", "xmm5", "3", "0")
  674. snow_inner_add_yblock_sse2_accum_16("2", "16")
  675. snow_inner_add_yblock_sse2_accum_16("1", "512")
  676. snow_inner_add_yblock_sse2_accum_16("0", "528")
  677. "mov %0, %%"REG_d" \n\t"
  678. "psrlw $4, %%xmm1 \n\t"
  679. "psrlw $4, %%xmm5 \n\t"
  680. "paddw (%%"REG_D"), %%xmm1 \n\t"
  681. "paddw 16(%%"REG_D"), %%xmm5 \n\t"
  682. "paddw %%xmm3, %%xmm1 \n\t"
  683. "paddw %%xmm3, %%xmm5 \n\t"
  684. "psraw $4, %%xmm1 \n\t" /* FRAC_BITS. */
  685. "psraw $4, %%xmm5 \n\t" /* FRAC_BITS. */
  686. "packuswb %%xmm5, %%xmm1 \n\t"
  687. "movdqu %%xmm1, (%%"REG_d") \n\t"
  688. snow_inner_add_yblock_sse2_end_16
  689. }
  690. #define snow_inner_add_yblock_mmx_header \
  691. IDWTELEM * * dst_array = sb->line + src_y;\
  692. x86_reg tmp;\
  693. __asm__ volatile(\
  694. "mov %7, %%"REG_c" \n\t"\
  695. "mov %6, %2 \n\t"\
  696. "mov %4, %%"REG_S" \n\t"\
  697. "pxor %%mm7, %%mm7 \n\t" /* 0 */\
  698. "pcmpeqd %%mm3, %%mm3 \n\t"\
  699. "psllw $15, %%mm3 \n\t"\
  700. "psrlw $12, %%mm3 \n\t" /* FRAC_BITS >> 1 */\
  701. "1: \n\t"\
  702. "mov %1, %%"REG_D" \n\t"\
  703. "mov (%%"REG_D"), %%"REG_D" \n\t"\
  704. "add %3, %%"REG_D" \n\t"
  705. #define snow_inner_add_yblock_mmx_start(out_reg1, out_reg2, ptr_offset, s_offset, d_offset)\
  706. "mov "PTR_SIZE"*"ptr_offset"(%%"REG_a"), %%"REG_d"; \n\t"\
  707. "movd "d_offset"(%%"REG_d"), %%"out_reg1" \n\t"\
  708. "movd "d_offset"+4(%%"REG_d"), %%"out_reg2" \n\t"\
  709. "punpcklbw %%mm7, %%"out_reg1" \n\t"\
  710. "punpcklbw %%mm7, %%"out_reg2" \n\t"\
  711. "movd "s_offset"(%%"REG_S"), %%mm0 \n\t"\
  712. "movd "s_offset"+4(%%"REG_S"), %%mm4 \n\t"\
  713. "punpcklbw %%mm7, %%mm0 \n\t"\
  714. "punpcklbw %%mm7, %%mm4 \n\t"\
  715. "pmullw %%mm0, %%"out_reg1" \n\t"\
  716. "pmullw %%mm4, %%"out_reg2" \n\t"
  717. #define snow_inner_add_yblock_mmx_accum(ptr_offset, s_offset, d_offset) \
  718. snow_inner_add_yblock_mmx_start("mm2", "mm6", ptr_offset, s_offset, d_offset)\
  719. "paddusw %%mm2, %%mm1 \n\t"\
  720. "paddusw %%mm6, %%mm5 \n\t"
  721. #define snow_inner_add_yblock_mmx_mix(read_offset, write_offset)\
  722. "mov %0, %%"REG_d" \n\t"\
  723. "psrlw $4, %%mm1 \n\t"\
  724. "psrlw $4, %%mm5 \n\t"\
  725. "paddw "read_offset"(%%"REG_D"), %%mm1 \n\t"\
  726. "paddw "read_offset"+8(%%"REG_D"), %%mm5 \n\t"\
  727. "paddw %%mm3, %%mm1 \n\t"\
  728. "paddw %%mm3, %%mm5 \n\t"\
  729. "psraw $4, %%mm1 \n\t"\
  730. "psraw $4, %%mm5 \n\t"\
  731. "packuswb %%mm5, %%mm1 \n\t"\
  732. "movq %%mm1, "write_offset"(%%"REG_d") \n\t"
  733. #define snow_inner_add_yblock_mmx_end(s_step)\
  734. "add $"s_step", %%"REG_S" \n\t"\
  735. "add %%"REG_c", "PTR_SIZE"*3(%%"REG_a");\n\t"\
  736. "add %%"REG_c", "PTR_SIZE"*2(%%"REG_a");\n\t"\
  737. "add %%"REG_c", "PTR_SIZE"*1(%%"REG_a");\n\t"\
  738. "add %%"REG_c", (%%"REG_a") \n\t"\
  739. "add"OPSIZE " $"PTR_SIZE"*1, %1 \n\t"\
  740. "add %%"REG_c", %0 \n\t"\
  741. "dec %2 \n\t"\
  742. "jnz 1b \n\t"\
  743. :"+m"(dst8),"+m"(dst_array),"=&r"(tmp)\
  744. :\
  745. "rm"((x86_reg)(src_x<<1)),"m"(obmc),"a"(block),"m"(b_h),"m"(src_stride):\
  746. "%"REG_c"","%"REG_S"","%"REG_D"","%"REG_d"");
  747. static void inner_add_yblock_bw_8_obmc_16_mmx(const uint8_t *obmc, const x86_reg obmc_stride, uint8_t * * block, int b_w, x86_reg b_h,
  748. int src_x, int src_y, x86_reg src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  749. snow_inner_add_yblock_mmx_header
  750. snow_inner_add_yblock_mmx_start("mm1", "mm5", "3", "0", "0")
  751. snow_inner_add_yblock_mmx_accum("2", "8", "0")
  752. snow_inner_add_yblock_mmx_accum("1", "128", "0")
  753. snow_inner_add_yblock_mmx_accum("0", "136", "0")
  754. snow_inner_add_yblock_mmx_mix("0", "0")
  755. snow_inner_add_yblock_mmx_end("16")
  756. }
  757. static void inner_add_yblock_bw_16_obmc_32_mmx(const uint8_t *obmc, const x86_reg obmc_stride, uint8_t * * block, int b_w, x86_reg b_h,
  758. int src_x, int src_y, x86_reg src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  759. snow_inner_add_yblock_mmx_header
  760. snow_inner_add_yblock_mmx_start("mm1", "mm5", "3", "0", "0")
  761. snow_inner_add_yblock_mmx_accum("2", "16", "0")
  762. snow_inner_add_yblock_mmx_accum("1", "512", "0")
  763. snow_inner_add_yblock_mmx_accum("0", "528", "0")
  764. snow_inner_add_yblock_mmx_mix("0", "0")
  765. snow_inner_add_yblock_mmx_start("mm1", "mm5", "3", "8", "8")
  766. snow_inner_add_yblock_mmx_accum("2", "24", "8")
  767. snow_inner_add_yblock_mmx_accum("1", "520", "8")
  768. snow_inner_add_yblock_mmx_accum("0", "536", "8")
  769. snow_inner_add_yblock_mmx_mix("16", "8")
  770. snow_inner_add_yblock_mmx_end("32")
  771. }
  772. static void ff_snow_inner_add_yblock_sse2(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
  773. int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  774. if (b_w == 16)
  775. inner_add_yblock_bw_16_obmc_32_sse2(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  776. else if (b_w == 8 && obmc_stride == 16) {
  777. if (!(b_h & 1))
  778. inner_add_yblock_bw_8_obmc_16_bh_even_sse2(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  779. else
  780. inner_add_yblock_bw_8_obmc_16_mmx(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  781. } else
  782. ff_snow_inner_add_yblock(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  783. }
  784. static void ff_snow_inner_add_yblock_mmx(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
  785. int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8){
  786. if (b_w == 16)
  787. inner_add_yblock_bw_16_obmc_32_mmx(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  788. else if (b_w == 8 && obmc_stride == 16)
  789. inner_add_yblock_bw_8_obmc_16_mmx(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  790. else
  791. ff_snow_inner_add_yblock(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
  792. }
  793. #endif /* HAVE_INLINE_ASM */
  794. void ff_dwt_init_x86(DWTContext *c)
  795. {
  796. #if HAVE_INLINE_ASM
  797. int mm_flags = av_get_cpu_flags();
  798. if (mm_flags & AV_CPU_FLAG_MMX) {
  799. if(mm_flags & AV_CPU_FLAG_SSE2 & 0){
  800. c->horizontal_compose97i = ff_snow_horizontal_compose97i_sse2;
  801. #if HAVE_7REGS
  802. c->vertical_compose97i = ff_snow_vertical_compose97i_sse2;
  803. #endif
  804. c->inner_add_yblock = ff_snow_inner_add_yblock_sse2;
  805. }
  806. else{
  807. if(mm_flags & AV_CPU_FLAG_MMX2){
  808. c->horizontal_compose97i = ff_snow_horizontal_compose97i_mmx;
  809. #if HAVE_7REGS
  810. c->vertical_compose97i = ff_snow_vertical_compose97i_mmx;
  811. #endif
  812. }
  813. c->inner_add_yblock = ff_snow_inner_add_yblock_mmx;
  814. }
  815. }
  816. #endif /* HAVE_INLINE_ASM */
  817. }