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.

826 lines
26KB

  1. ;*****************************************************************************
  2. ;* SSE2-optimized HEVC deblocking code
  3. ;*****************************************************************************
  4. ;* Copyright (C) 2013 VTT
  5. ;*
  6. ;* Authors: Seppo Tomperi <seppo.tomperi@vtt.fi>
  7. ;*
  8. ;* This file is part of Libav.
  9. ;*
  10. ;* Libav is free software; you can redistribute it and/or
  11. ;* modify it under the terms of the GNU Lesser General Public
  12. ;* License as published by the Free Software Foundation; either
  13. ;* version 2.1 of the License, or (at your option) any later version.
  14. ;*
  15. ;* Libav is distributed in the hope that it will be useful,
  16. ;* but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. ;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. ;* Lesser General Public License for more details.
  19. ;*
  20. ;* You should have received a copy of the GNU Lesser General Public
  21. ;* License along with Libav; if not, write to the Free Software
  22. ;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. ;******************************************************************************
  24. %include "libavutil/x86/x86util.asm"
  25. SECTION_RODATA
  26. pw_pixel_max: times 8 dw ((1 << 10)-1)
  27. pw_m1: times 8 dw -1
  28. pw_m2: times 8 dw -2
  29. pd_1 : times 4 dd 1
  30. cextern pw_4
  31. cextern pw_8
  32. SECTION .text
  33. INIT_XMM sse2
  34. ; expands to [base],...,[base+7*stride]
  35. %define PASS8ROWS(base, base3, stride, stride3) \
  36. [base], [base+stride], [base+stride*2], [base3], \
  37. [base3+stride], [base3+stride*2], [base3+stride3], [base3+stride*4]
  38. ; in: 8 rows of 4 bytes in %4..%11
  39. ; out: 4 rows of 8 words in m0..m3
  40. %macro TRANSPOSE4x8B_LOAD 8
  41. movd m0, %1
  42. movd m2, %2
  43. movd m1, %3
  44. movd m3, %4
  45. punpcklbw m0, m2
  46. punpcklbw m1, m3
  47. punpcklwd m0, m1
  48. movd m4, %5
  49. movd m6, %6
  50. movd m5, %7
  51. movd m7, %8
  52. punpcklbw m4, m6
  53. punpcklbw m5, m7
  54. punpcklwd m4, m5
  55. punpckhdq m2, m0, m4
  56. punpckldq m0, m4
  57. pxor m5, m5
  58. punpckhbw m1, m0, m5
  59. punpcklbw m0, m5
  60. punpckhbw m3, m2, m5
  61. punpcklbw m2, m5
  62. %endmacro
  63. ; in: 4 rows of 8 words in m0..m3
  64. ; out: 8 rows of 4 bytes in %1..%8
  65. %macro TRANSPOSE8x4B_STORE 8
  66. packuswb m0, m0
  67. packuswb m1, m1
  68. packuswb m2, m2
  69. packuswb m3, m3
  70. punpcklbw m0, m1
  71. punpcklbw m2, m3
  72. punpckhwd m6, m0, m2
  73. punpcklwd m0, m2
  74. movd %1, m0
  75. pshufd m0, m0, 0x39
  76. movd %2, m0
  77. pshufd m0, m0, 0x39
  78. movd %3, m0
  79. pshufd m0, m0, 0x39
  80. movd %4, m0
  81. movd %5, m6
  82. pshufd m6, m6, 0x39
  83. movd %6, m6
  84. pshufd m6, m6, 0x39
  85. movd %7, m6
  86. pshufd m6, m6, 0x39
  87. movd %8, m6
  88. %endmacro
  89. ; in: 8 rows of 4 words in %4..%11
  90. ; out: 4 rows of 8 words in m0..m3
  91. %macro TRANSPOSE4x8W_LOAD 8
  92. movq m0, %1
  93. movq m2, %2
  94. movq m1, %3
  95. movq m3, %4
  96. punpcklwd m0, m2
  97. punpcklwd m1, m3
  98. punpckhdq m2, m0, m1
  99. punpckldq m0, m1
  100. movq m4, %5
  101. movq m6, %6
  102. movq m5, %7
  103. movq m7, %8
  104. punpcklwd m4, m6
  105. punpcklwd m5, m7
  106. punpckhdq m6, m4, m5
  107. punpckldq m4, m5
  108. punpckhqdq m1, m0, m4
  109. punpcklqdq m0, m4
  110. punpckhqdq m3, m2, m6
  111. punpcklqdq m2, m6
  112. %endmacro
  113. ; in: 4 rows of 8 words in m0..m3
  114. ; out: 8 rows of 4 words in %1..%8
  115. %macro TRANSPOSE8x4W_STORE 8
  116. pxor m5, m5; zeros reg
  117. CLIPW m0, m5, [pw_pixel_max]
  118. CLIPW m1, m5, [pw_pixel_max]
  119. CLIPW m2, m5, [pw_pixel_max]
  120. CLIPW m3, m5, [pw_pixel_max]
  121. punpckhwd m4, m0, m1
  122. punpcklwd m0, m1
  123. punpckhwd m5, m2, m3
  124. punpcklwd m2, m3
  125. punpckhdq m6, m0, m2
  126. punpckldq m0, m2
  127. movq %1, m0
  128. movhps %2, m0
  129. movq %3, m6
  130. movhps %4, m6
  131. punpckhdq m6, m4, m5
  132. punpckldq m4, m5
  133. movq %5, m4
  134. movhps %6, m4
  135. movq %7, m6
  136. movhps %8, m6
  137. %endmacro
  138. ; in: 8 rows of 8 bytes in %1..%8
  139. ; out: 8 rows of 8 words in m0..m7
  140. %macro TRANSPOSE8x8B_LOAD 8
  141. movq m7, %1
  142. movq m2, %2
  143. movq m1, %3
  144. movq m3, %4
  145. punpcklbw m7, m2
  146. punpcklbw m1, m3
  147. punpcklwd m3, m7, m1
  148. punpckhwd m7, m1
  149. movq m4, %5
  150. movq m6, %6
  151. movq m5, %7
  152. movq m15, %8
  153. punpcklbw m4, m6
  154. punpcklbw m5, m15
  155. punpcklwd m9, m4, m5
  156. punpckhwd m4, m5
  157. punpckldq m1, m3, m9; 0, 1
  158. punpckhdq m3, m9; 2, 3
  159. punpckldq m5, m7, m4; 4, 5
  160. punpckhdq m7, m4; 6, 7
  161. pxor m13, m13
  162. punpcklbw m0, m1, m13; 0 in 16 bit
  163. punpckhbw m1, m13; 1 in 16 bit
  164. punpcklbw m2, m3, m13; 2
  165. punpckhbw m3, m13; 3
  166. punpcklbw m4, m5, m13; 4
  167. punpckhbw m5, m13; 5
  168. punpcklbw m6, m7, m13; 6
  169. punpckhbw m7, m13; 7
  170. %endmacro
  171. ; in: 8 rows of 8 words in m0..m8
  172. ; out: 8 rows of 8 bytes in %1..%8
  173. %macro TRANSPOSE8x8B_STORE 8
  174. packuswb m0, m0
  175. packuswb m1, m1
  176. packuswb m2, m2
  177. packuswb m3, m3
  178. packuswb m4, m4
  179. packuswb m5, m5
  180. packuswb m6, m6
  181. packuswb m7, m7
  182. punpcklbw m0, m1
  183. punpcklbw m2, m3
  184. punpckhwd m8, m0, m2
  185. punpcklwd m0, m2
  186. punpcklbw m4, m5
  187. punpcklbw m6, m7
  188. punpckhwd m9, m4, m6
  189. punpcklwd m4, m6
  190. punpckhdq m10, m0, m4; 2, 3
  191. punpckldq m0, m4; 0, 1
  192. punpckldq m11, m8, m9; 4, 5
  193. punpckhdq m8, m9; 6, 7
  194. movq %1, m0
  195. movhps %2, m0
  196. movq %3, m10
  197. movhps %4, m10
  198. movq %5, m11
  199. movhps %6, m11
  200. movq %7, m8
  201. movhps %8, m8
  202. %endmacro
  203. ; in: 8 rows of 8 words in %1..%8
  204. ; out: 8 rows of 8 words in m0..m7
  205. %macro TRANSPOSE8x8W_LOAD 8
  206. movdqu m0, %1
  207. movdqu m1, %2
  208. movdqu m2, %3
  209. movdqu m3, %4
  210. movdqu m4, %5
  211. movdqu m5, %6
  212. movdqu m6, %7
  213. movdqu m7, %8
  214. TRANSPOSE8x8W 0, 1, 2, 3, 4, 5, 6, 7, 8
  215. %endmacro
  216. ; in: 8 rows of 8 words in m0..m8
  217. ; out: 8 rows of 8 words in %1..%8
  218. %macro TRANSPOSE8x8W_STORE 8
  219. TRANSPOSE8x8W 0, 1, 2, 3, 4, 5, 6, 7, 8
  220. pxor m8, m8
  221. CLIPW m0, m8, [pw_pixel_max]
  222. CLIPW m1, m8, [pw_pixel_max]
  223. CLIPW m2, m8, [pw_pixel_max]
  224. CLIPW m3, m8, [pw_pixel_max]
  225. CLIPW m4, m8, [pw_pixel_max]
  226. CLIPW m5, m8, [pw_pixel_max]
  227. CLIPW m6, m8, [pw_pixel_max]
  228. CLIPW m7, m8, [pw_pixel_max]
  229. movdqu %1, m0
  230. movdqu %2, m1
  231. movdqu %3, m2
  232. movdqu %4, m3
  233. movdqu %5, m4
  234. movdqu %6, m5
  235. movdqu %7, m6
  236. movdqu %8, m7
  237. %endmacro
  238. ; in: %2 clobbered
  239. ; out: %1
  240. ; mask in m11
  241. ; clobbers m10
  242. %macro MASKED_COPY 2
  243. pand %2, m11 ; and mask
  244. pandn m10, m11, %1; and -mask
  245. por %2, m10
  246. mova %1, %2
  247. %endmacro
  248. ; in: %2 clobbered
  249. ; out: %1
  250. ; mask in %3, will be clobbered
  251. %macro MASKED_COPY2 3
  252. pand %2, %3 ; and mask
  253. pandn %3, %1; and -mask
  254. por %2, %3
  255. mova %1, %2
  256. %endmacro
  257. ALIGN 16
  258. ; input in m0 ... m3 and tcs in r2. Output in m1 and m2
  259. %macro CHROMA_DEBLOCK_BODY 1
  260. psubw m4, m2, m1; q0 - p0
  261. psubw m5, m0, m3; p1 - q1
  262. psllw m4, 2; << 2
  263. paddw m5, m4;
  264. ;tc calculations
  265. movd m6, [r2]; tc0
  266. add r2, 4;
  267. punpcklwd m6, m6
  268. movd m7, [r2]; tc1
  269. punpcklwd m7, m7
  270. shufps m6, m7, 0; tc0, tc1
  271. pmullw m4, m6, [pw_m1]; -tc0, -tc1
  272. ;end tc calculations
  273. paddw m5, [pw_4]; +4
  274. psraw m5, 3; >> 3
  275. %if %1 > 8
  276. psllw m4, %1-8; << (BIT_DEPTH - 8)
  277. psllw m6, %1-8; << (BIT_DEPTH - 8)
  278. %endif
  279. pmaxsw m5, m4
  280. pminsw m5, m6
  281. paddw m1, m5; p0 + delta0
  282. psubw m2, m5; q0 - delta0
  283. %endmacro
  284. ; input in m0 ... m7, beta in r2 tcs in r3. Output in m1...m6
  285. %macro LUMA_DEBLOCK_BODY 2
  286. psllw m9, m2, 1; *2
  287. psubw m10, m1, m9
  288. paddw m10, m3
  289. ABS1 m10, m11 ; 0dp0, 0dp3 , 1dp0, 1dp3
  290. psllw m9, m5, 1; *2
  291. psubw m11, m6, m9
  292. paddw m11, m4
  293. ABS1 m11, m13 ; 0dq0, 0dq3 , 1dq0, 1dq3
  294. ;beta calculations
  295. %if %1 > 8
  296. shl betaq, %1 - 8
  297. %endif
  298. movd m13, betad
  299. SPLATW m13, m13, 0
  300. ;end beta calculations
  301. paddw m9, m10, m11; 0d0, 0d3 , 1d0, 1d3
  302. pshufhw m14, m9, q0033 ;0b00001111; 0d3 0d3 0d0 0d0 in high
  303. pshuflw m14, m14, q0033 ;0b00001111; 1d3 1d3 1d0 1d0 in low
  304. pshufhw m9, m9, q3300 ;0b11110000; 0d0 0d0 0d3 0d3
  305. pshuflw m9, m9, q3300 ;0b11110000; 1d0 1d0 1d3 1d3
  306. paddw m14, m9; 0d0+0d3, 1d0+1d3
  307. ;compare
  308. pcmpgtw m15, m13, m14
  309. movmskps r13, m15 ;filtering mask 0d0 + 0d3 < beta0 (bit 2 or 3) , 1d0 + 1d3 < beta1 (bit 0 or 1)
  310. test r13, r13
  311. je .bypassluma
  312. ;weak / strong decision compare to beta_2
  313. psraw m15, m13, 2; beta >> 2
  314. psllw m8, m9, 1;
  315. pcmpgtw m15, m8; (d0 << 1) < beta_2, (d3 << 1) < beta_2
  316. movmskps r14, m15;
  317. ;end weak / strong decision
  318. ; weak filter nd_p/q calculation
  319. pshufd m8, m10, 0x31
  320. psrld m8, 16
  321. paddw m8, m10
  322. movd r7d, m8
  323. and r7, 0xffff; 1dp0 + 1dp3
  324. pshufd m8, m8, 0x4E
  325. movd r8d, m8
  326. and r8, 0xffff; 0dp0 + 0dp3
  327. pshufd m8, m11, 0x31
  328. psrld m8, 16
  329. paddw m8, m11
  330. movd r9d, m8
  331. and r9, 0xffff; 1dq0 + 1dq3
  332. pshufd m8, m8, 0x4E
  333. movd r10d, m8
  334. and r10, 0xffff; 0dq0 + 0dq3
  335. ; end calc for weak filter
  336. ; filtering mask
  337. mov r11, r13
  338. shr r11, 3
  339. movd m15, r11d
  340. and r13, 1
  341. movd m11, r13d
  342. shufps m11, m15, 0
  343. shl r11, 1
  344. or r13, r11
  345. pcmpeqd m11, [pd_1]; filtering mask
  346. ;decide between strong and weak filtering
  347. ;tc25 calculations
  348. mov r11d, [tcq];
  349. %if %1 > 8
  350. shl r11, %1 - 8
  351. %endif
  352. movd m8, r11d; tc0
  353. add tcq, 4;
  354. mov r3d, [tcq];
  355. %if %1 > 8
  356. shl r3, %1 - 8
  357. %endif
  358. movd m9, r3d; tc1
  359. add r11d, r3d; tc0 + tc1
  360. jz .bypassluma
  361. punpcklwd m8, m8
  362. punpcklwd m9, m9
  363. shufps m8, m9, 0; tc0, tc1
  364. mova m9, m8
  365. psllw m8, 2; tc << 2
  366. pavgw m8, m9; tc25 = ((tc * 5 + 1) >> 1)
  367. ;end tc25 calculations
  368. ;----beta_3 comparison-----
  369. psubw m12, m0, m3; p3 - p0
  370. ABS1 m12, m14; abs(p3 - p0)
  371. psubw m15, m7, m4; q3 - q0
  372. ABS1 m15, m14; abs(q3 - q0)
  373. paddw m12, m15; abs(p3 - p0) + abs(q3 - q0)
  374. pshufhw m12, m12, 0xf0 ;0b11110000;
  375. pshuflw m12, m12, 0xf0 ;0b11110000;
  376. psraw m13, 3; beta >> 3
  377. pcmpgtw m13, m12;
  378. movmskps r11, m13;
  379. and r14, r11; strong mask , beta_2 and beta_3 comparisons
  380. ;----beta_3 comparison end-----
  381. ;----tc25 comparison---
  382. psubw m12, m3, m4; p0 - q0
  383. ABS1 m12, m14; abs(p0 - q0)
  384. pshufhw m12, m12, 0xf0 ;0b11110000;
  385. pshuflw m12, m12, 0xf0 ;0b11110000;
  386. pcmpgtw m8, m12; tc25 comparisons
  387. movmskps r11, m8;
  388. and r14, r11; strong mask, beta_2, beta_3 and tc25 comparisons
  389. ;----tc25 comparison end---
  390. mov r11, r14;
  391. shr r11, 1;
  392. and r14, r11; strong mask, bits 2 and 0
  393. pmullw m14, m9, [pw_m2]; -tc * 2
  394. paddw m9, m9
  395. and r14, 5; 0b101
  396. mov r11, r14; strong mask
  397. shr r14, 2;
  398. movd m12, r14d; store to xmm for mask generation
  399. shl r14, 1
  400. and r11, 1
  401. movd m10, r11d; store to xmm for mask generation
  402. or r14, r11; final strong mask, bits 1 and 0
  403. jz .weakfilter
  404. shufps m10, m12, 0
  405. pcmpeqd m10, [pd_1]; strong mask
  406. mova m13, [pw_4]; 4 in every cell
  407. pand m11, m10; combine filtering mask and strong mask
  408. paddw m12, m2, m3; p1 + p0
  409. paddw m12, m4; p1 + p0 + q0
  410. mova m10, m12; copy
  411. paddw m12, m12; 2*p1 + 2*p0 + 2*q0
  412. paddw m12, m1; p2 + 2*p1 + 2*p0 + 2*q0
  413. paddw m12, m5; p2 + 2*p1 + 2*p0 + 2*q0 + q1
  414. paddw m12, m13; p2 + 2*p1 + 2*p0 + 2*q0 + q1 + 4
  415. psraw m12, 3; ((p2 + 2*p1 + 2*p0 + 2*q0 + q1 + 4) >> 3)
  416. psubw m12, m3; ((p2 + 2*p1 + 2*p0 + 2*q0 + q1 + 4) >> 3) - p0
  417. pmaxsw m12, m14
  418. pminsw m12, m9; av_clip( , -2 * tc, 2 * tc)
  419. paddw m12, m3; p0'
  420. paddw m15, m1, m10; p2 + p1 + p0 + q0
  421. psrlw m13, 1; 2 in every cell
  422. paddw m15, m13; p2 + p1 + p0 + q0 + 2
  423. psraw m15, 2; (p2 + p1 + p0 + q0 + 2) >> 2
  424. psubw m15, m2;((p2 + p1 + p0 + q0 + 2) >> 2) - p1
  425. pmaxsw m15, m14
  426. pminsw m15, m9; av_clip( , -2 * tc, 2 * tc)
  427. paddw m15, m2; p1'
  428. paddw m8, m1, m0; p3 + p2
  429. paddw m8, m8; 2*p3 + 2*p2
  430. paddw m8, m1; 2*p3 + 3*p2
  431. paddw m8, m10; 2*p3 + 3*p2 + p1 + p0 + q0
  432. paddw m13, m13
  433. paddw m8, m13; 2*p3 + 3*p2 + p1 + p0 + q0 + 4
  434. psraw m8, 3; (2*p3 + 3*p2 + p1 + p0 + q0 + 4) >> 3
  435. psubw m8, m1; ((2*p3 + 3*p2 + p1 + p0 + q0 + 4) >> 3) - p2
  436. pmaxsw m8, m14
  437. pminsw m8, m9; av_clip( , -2 * tc, 2 * tc)
  438. paddw m8, m1; p2'
  439. MASKED_COPY m1, m8
  440. paddw m8, m3, m4; p0 + q0
  441. paddw m8, m5; p0 + q0 + q1
  442. paddw m8, m8; 2*p0 + 2*q0 + 2*q1
  443. paddw m8, m2; p1 + 2*p0 + 2*q0 + 2*q1
  444. paddw m8, m6; p1 + 2*p0 + 2*q0 + 2*q1 + q2
  445. paddw m8, m13; p1 + 2*p0 + 2*q0 + 2*q1 + q2 + 4
  446. psraw m8, 3; (p1 + 2*p0 + 2*q0 + 2*q1 + q2 + 4) >>3
  447. psubw m8, m4;
  448. pmaxsw m8, m14
  449. pminsw m8, m9; av_clip( , -2 * tc, 2 * tc)
  450. paddw m8, m4; q0'
  451. MASKED_COPY m2, m15
  452. paddw m15, m3, m4; p0 + q0
  453. paddw m15, m5; p0 + q0 + q1
  454. mova m10, m15;
  455. paddw m15, m6; p0 + q0 + q1 + q2
  456. psrlw m13, 1; 2 in every cell
  457. paddw m15, m13; p0 + q0 + q1 + q2 + 2
  458. psraw m15, 2; (p0 + q0 + q1 + q2 + 2) >> 2
  459. psubw m15, m5; ((p0 + q0 + q1 + q2 + 2) >> 2) - q1
  460. pmaxsw m15, m14
  461. pminsw m15, m9; av_clip( , -2 * tc, 2 * tc)
  462. paddw m15, m5; q1'
  463. paddw m13, m7; q3 + 2
  464. paddw m13, m6; q3 + q2 + 2
  465. paddw m13, m13; 2*q3 + 2*q2 + 4
  466. paddw m13, m6; 2*q3 + 3*q2 + 4
  467. paddw m13, m10; 2*q3 + 3*q2 + q1 + q0 + p0 + 4
  468. psraw m13, 3; (2*q3 + 3*q2 + q1 + q0 + p0 + 4) >> 3
  469. psubw m13, m6; ((2*q3 + 3*q2 + q1 + q0 + p0 + 4) >> 3) - q2
  470. pmaxsw m13, m14
  471. pminsw m13, m9; av_clip( , -2 * tc, 2 * tc)
  472. paddw m13, m6; q2'
  473. MASKED_COPY m6, m13
  474. MASKED_COPY m5, m15
  475. MASKED_COPY m4, m8
  476. MASKED_COPY m3, m12
  477. .weakfilter:
  478. not r14; strong mask -> weak mask
  479. and r14, r13; final weak filtering mask, bits 0 and 1
  480. jz .store
  481. ; weak filtering mask
  482. mov r11, r14
  483. shr r11, 1
  484. movd m12, r11d
  485. and r14, 1
  486. movd m11, r14d
  487. shufps m11, m12, 0
  488. pcmpeqd m11, [pd_1]; filtering mask
  489. mov r13, betaq
  490. shr r13, 1;
  491. add betaq, r13
  492. shr betaq, 3; ((beta + (beta >> 1)) >> 3))
  493. mova m13, [pw_8]
  494. psubw m12, m4, m3 ; q0 - p0
  495. psllw m10, m12, 3; 8 * (q0 - p0)
  496. paddw m12, m10 ; 9 * (q0 - p0)
  497. psubw m10, m5, m2 ; q1 - p1
  498. psllw m8, m10, 1; 2 * ( q1 - p1 )
  499. paddw m10, m8; 3 * ( q1 - p1 )
  500. psubw m12, m10; 9 * (q0 - p0) - 3 * ( q1 - p1 )
  501. paddw m12, m13; + 8
  502. psraw m12, 4; >> 4 , delta0
  503. PABSW m13, m12; abs(delta0)
  504. psllw m10, m9, 2; 8 * tc
  505. paddw m10, m9; 10 * tc
  506. pcmpgtw m10, m13
  507. pand m11, m10
  508. psraw m9, 1; tc * 2 -> tc
  509. psraw m14, 1; -tc * 2 -> -tc
  510. pmaxsw m12, m14
  511. pminsw m12, m9; av_clip(delta0, -tc, tc)
  512. psraw m9, 1; tc -> tc / 2
  513. pmullw m14, m9, [pw_m1]; -tc / 2
  514. pavgw m15, m1, m3; (p2 + p0 + 1) >> 1
  515. psubw m15, m2; ((p2 + p0 + 1) >> 1) - p1
  516. paddw m15, m12; ((p2 + p0 + 1) >> 1) - p1 + delta0
  517. psraw m15, 1; (((p2 + p0 + 1) >> 1) - p1 + delta0) >> 1
  518. pmaxsw m15, m14
  519. pminsw m15, m9; av_clip(deltap1, -tc/2, tc/2)
  520. paddw m15, m2; p1'
  521. ;beta calculations
  522. movd m10, betad
  523. SPLATW m10, m10, 0
  524. movd m13, r7d; 1dp0 + 1dp3
  525. movd m8, r8d; 0dp0 + 0dp3
  526. punpcklwd m8, m8
  527. punpcklwd m13, m13
  528. shufps m13, m8, 0;
  529. pcmpgtw m8, m10, m13
  530. pand m8, m11
  531. ;end beta calculations
  532. MASKED_COPY2 m2, m15, m8; write p1'
  533. pavgw m8, m6, m4; (q2 + q0 + 1) >> 1
  534. psubw m8, m5; ((q2 + q0 + 1) >> 1) - q1
  535. psubw m8, m12; ((q2 + q0 + 1) >> 1) - q1 - delta0)
  536. psraw m8, 1; ((q2 + q0 + 1) >> 1) - q1 - delta0) >> 1
  537. pmaxsw m8, m14
  538. pminsw m8, m9; av_clip(deltaq1, -tc/2, tc/2)
  539. paddw m8, m5; q1'
  540. movd m13, r9d;
  541. movd m15, r10d;
  542. punpcklwd m15, m15
  543. punpcklwd m13, m13
  544. shufps m13, m15, 0; dq0 + dq3
  545. pcmpgtw m10, m13; compare to ((beta+(beta>>1))>>3)
  546. pand m10, m11
  547. MASKED_COPY2 m5, m8, m10; write q1'
  548. paddw m15, m3, m12 ; p0 + delta0
  549. MASKED_COPY m3, m15
  550. psubw m8, m4, m12 ; q0 - delta0
  551. MASKED_COPY m4, m8
  552. %endmacro
  553. INIT_XMM sse2
  554. ;-----------------------------------------------------------------------------
  555. ; void ff_hevc_v_loop_filter_chroma(uint8_t *_pix, ptrdiff_t _stride, int *_tc,
  556. ; uint8_t *_no_p, uint8_t *_no_q);
  557. ;-----------------------------------------------------------------------------
  558. cglobal hevc_v_loop_filter_chroma_8, 3, 6, 8
  559. sub r0, 2
  560. lea r5, [3 * r1]
  561. mov r4, r0
  562. add r0, r5
  563. TRANSPOSE4x8B_LOAD PASS8ROWS(r4, r0, r1, r5)
  564. CHROMA_DEBLOCK_BODY 8
  565. TRANSPOSE8x4B_STORE PASS8ROWS(r4, r0, r1, r5)
  566. RET
  567. cglobal hevc_v_loop_filter_chroma_10, 3, 6, 8
  568. sub r0, 4
  569. lea r5, [3 * r1]
  570. mov r4, r0
  571. add r0, r5
  572. TRANSPOSE4x8W_LOAD PASS8ROWS(r4, r0, r1, r5)
  573. CHROMA_DEBLOCK_BODY 10
  574. TRANSPOSE8x4W_STORE PASS8ROWS(r4, r0, r1, r5)
  575. RET
  576. ;-----------------------------------------------------------------------------
  577. ; void ff_hevc_h_loop_filter_chroma(uint8_t *_pix, ptrdiff_t _stride, int *_tc,
  578. ; uint8_t *_no_p, uint8_t *_no_q);
  579. ;-----------------------------------------------------------------------------
  580. cglobal hevc_h_loop_filter_chroma_8, 3, 6, 8
  581. mov r5, r0; pix
  582. sub r5, r1
  583. sub r5, r1
  584. movh m0, [r5]; p1
  585. movh m1, [r5 + r1]; p0
  586. movh m2, [r0]; q0
  587. movh m3, [r0 + r1]; q1
  588. pxor m5, m5; zeros reg
  589. punpcklbw m0, m5
  590. punpcklbw m1, m5
  591. punpcklbw m2, m5
  592. punpcklbw m3, m5
  593. CHROMA_DEBLOCK_BODY 8
  594. packuswb m1, m2
  595. movh [r5 + r1], m1
  596. movhps [r0], m1
  597. RET
  598. cglobal hevc_h_loop_filter_chroma_10, 3, 6, 8
  599. mov r5, r0; pix
  600. sub r5, r1
  601. sub r5, r1
  602. movdqu m0, [r5]; p1
  603. movdqu m1, [r5+r1]; p0
  604. movdqu m2, [r0]; q0
  605. movdqu m3, [r0 + r1]; q1
  606. CHROMA_DEBLOCK_BODY 10
  607. pxor m5, m5; zeros reg
  608. CLIPW m1, m5, [pw_pixel_max]
  609. CLIPW m2, m5, [pw_pixel_max]
  610. movdqu [r5 + r1], m1
  611. movdqu [r0], m2
  612. RET
  613. %if ARCH_X86_64
  614. INIT_XMM ssse3
  615. ;-----------------------------------------------------------------------------
  616. ; void ff_hevc_v_loop_filter_luma(uint8_t *_pix, ptrdiff_t _stride, int beta,
  617. ; int *_tc, uint8_t *_no_p, uint8_t *_no_q);
  618. ;-----------------------------------------------------------------------------
  619. cglobal hevc_v_loop_filter_luma_8, 4, 15, 16, pix, stride, beta, tc
  620. sub r0, 4
  621. lea r5, [3 * r1]
  622. mov r6, r0
  623. add r0, r5
  624. TRANSPOSE8x8B_LOAD PASS8ROWS(r6, r0, r1, r5)
  625. LUMA_DEBLOCK_BODY 8, v
  626. .store:
  627. TRANSPOSE8x8B_STORE PASS8ROWS(r6, r0, r1, r5)
  628. .bypassluma:
  629. RET
  630. cglobal hevc_v_loop_filter_luma_10, 4, 15, 16, pix, stride, beta, tc
  631. sub pixq, 8
  632. lea r5, [3 * strideq]
  633. mov r6, pixq
  634. add pixq, r5
  635. TRANSPOSE8x8W_LOAD PASS8ROWS(r6, pixq, strideq, r5)
  636. LUMA_DEBLOCK_BODY 10, v
  637. .store:
  638. TRANSPOSE8x8W_STORE PASS8ROWS(r6, r0, r1, r5)
  639. .bypassluma:
  640. RET
  641. ;-----------------------------------------------------------------------------
  642. ; void ff_hevc_h_loop_filter_luma(uint8_t *_pix, ptrdiff_t _stride, int beta,
  643. ; int *_tc, uint8_t *_no_p, uint8_t *_no_q);
  644. ;-----------------------------------------------------------------------------
  645. cglobal hevc_h_loop_filter_luma_8, 4, 15, 16, pix, stride, beta, tc, count, pix0, src3stride
  646. lea src3strideq, [3 * strideq]
  647. mov pix0q, pixq
  648. sub pix0q, src3strideq
  649. sub pix0q, strideq
  650. movdqu m0, [pix0q]; p3
  651. movdqu m1, [pix0q + strideq]; p2
  652. movdqu m2, [pix0q + 2 * strideq]; p1
  653. movdqu m3, [pix0q + src3strideq]; p0
  654. movdqu m4, [pixq]; q0
  655. movdqu m5, [pixq + strideq]; q1
  656. movdqu m6, [pixq + 2 * strideq]; q2
  657. movdqu m7, [pixq + src3strideq]; q3
  658. pxor m8, m8
  659. punpcklbw m0, m8
  660. punpcklbw m1, m8
  661. punpcklbw m2, m8
  662. punpcklbw m3, m8
  663. punpcklbw m4, m8
  664. punpcklbw m5, m8
  665. punpcklbw m6, m8
  666. punpcklbw m7, m8
  667. LUMA_DEBLOCK_BODY 8, h
  668. .store:
  669. packuswb m1, m2
  670. packuswb m3, m4
  671. packuswb m5, m6
  672. movh [r5 + r1], m1
  673. movhps [r5 + 2 * r1], m1
  674. movh [r5 + r6], m3
  675. movhps [r0 ], m3
  676. movh [r0 + r1], m5
  677. movhps [r0 + 2 * r1], m5
  678. .bypassluma:
  679. RET
  680. cglobal hevc_h_loop_filter_luma_10, 4, 15, 16, pix, stride, beta, tc, count, pix0, src3stride
  681. lea src3strideq, [3 * strideq]
  682. mov pix0q, pixq
  683. sub pix0q, src3strideq
  684. sub pix0q, strideq
  685. movdqu m0, [pix0q]; p3
  686. movdqu m1, [pix0q + strideq]; p2
  687. movdqu m2, [pix0q + 2 * strideq]; p1
  688. movdqu m3, [pix0q + src3strideq]; p0
  689. movdqu m4, [pixq]; q0
  690. movdqu m5, [pixq + strideq]; q1
  691. movdqu m6, [pixq + 2 * strideq]; q2
  692. movdqu m7, [pixq + src3strideq]; q3
  693. LUMA_DEBLOCK_BODY 10, h
  694. .store:
  695. pxor m8, m8; zeros reg
  696. CLIPW m1, m8, [pw_pixel_max]
  697. CLIPW m2, m8, [pw_pixel_max]
  698. CLIPW m3, m8, [pw_pixel_max]
  699. CLIPW m4, m8, [pw_pixel_max]
  700. CLIPW m5, m8, [pw_pixel_max]
  701. CLIPW m6, m8, [pw_pixel_max]
  702. movdqu [pix0q + strideq], m1; p2
  703. movdqu [pix0q + 2 * strideq], m2; p1
  704. movdqu [pix0q + src3strideq], m3; p0
  705. movdqu [pixq ], m4; q0
  706. movdqu [pixq + strideq], m5; q1
  707. movdqu [pixq + 2 * strideq], m6; q2
  708. .bypassluma:
  709. RET
  710. %endif