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.

414 lines
14KB

  1. /*
  2. * Copyright (c) 2009 David Conrad
  3. *
  4. * This file is part of Libav.
  5. *
  6. * Libav is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * Libav is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with Libav; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #include "libavutil/arm/asm.S"
  21. const vp3_idct_constants, align=4
  22. .short 64277, 60547, 54491, 46341, 36410, 25080, 12785
  23. endconst
  24. #define xC1S7 d0[0]
  25. #define xC2S6 d0[1]
  26. #define xC3S5 d0[2]
  27. #define xC4S4 d0[3]
  28. #define xC5S3 d1[0]
  29. #define xC6S2 d1[1]
  30. #define xC7S1 d1[2]
  31. .macro vp3_loop_filter
  32. vsubl.u8 q3, d18, d17
  33. vsubl.u8 q2, d16, d19
  34. vadd.i16 q1, q3, q3
  35. vadd.i16 q2, q2, q3
  36. vadd.i16 q0, q1, q2
  37. vrshr.s16 q0, q0, #3
  38. vmovl.u8 q9, d18
  39. vdup.u16 q15, r2
  40. vabs.s16 q1, q0
  41. vshr.s16 q0, q0, #15
  42. vqsub.u16 q2, q15, q1
  43. vqsub.u16 q3, q2, q1
  44. vsub.i16 q1, q2, q3
  45. veor q1, q1, q0
  46. vsub.i16 q0, q1, q0
  47. vaddw.u8 q2, q0, d17
  48. vsub.i16 q3, q9, q0
  49. vqmovun.s16 d0, q2
  50. vqmovun.s16 d1, q3
  51. .endm
  52. function ff_vp3_v_loop_filter_neon, export=1
  53. sub ip, r0, r1
  54. sub r0, r0, r1, lsl #1
  55. vld1.64 {d16}, [r0,:64], r1
  56. vld1.64 {d17}, [r0,:64], r1
  57. vld1.64 {d18}, [r0,:64], r1
  58. vld1.64 {d19}, [r0,:64], r1
  59. ldrb r2, [r2, #129*4]
  60. vp3_loop_filter
  61. vst1.64 {d0}, [ip,:64], r1
  62. vst1.64 {d1}, [ip,:64], r1
  63. bx lr
  64. endfunc
  65. function ff_vp3_h_loop_filter_neon, export=1
  66. sub ip, r0, #1
  67. sub r0, r0, #2
  68. vld1.32 {d16[]}, [r0], r1
  69. vld1.32 {d17[]}, [r0], r1
  70. vld1.32 {d18[]}, [r0], r1
  71. vld1.32 {d19[]}, [r0], r1
  72. vld1.32 {d16[1]}, [r0], r1
  73. vld1.32 {d17[1]}, [r0], r1
  74. vld1.32 {d18[1]}, [r0], r1
  75. vld1.32 {d19[1]}, [r0], r1
  76. ldrb r2, [r2, #129*4]
  77. vtrn.8 d16, d17
  78. vtrn.8 d18, d19
  79. vtrn.16 d16, d18
  80. vtrn.16 d17, d19
  81. vp3_loop_filter
  82. vtrn.8 d0, d1
  83. vst1.16 {d0[0]}, [ip], r1
  84. vst1.16 {d1[0]}, [ip], r1
  85. vst1.16 {d0[1]}, [ip], r1
  86. vst1.16 {d1[1]}, [ip], r1
  87. vst1.16 {d0[2]}, [ip], r1
  88. vst1.16 {d1[2]}, [ip], r1
  89. vst1.16 {d0[3]}, [ip], r1
  90. vst1.16 {d1[3]}, [ip], r1
  91. bx lr
  92. endfunc
  93. function vp3_idct_start_neon
  94. vpush {d8-d15}
  95. movrel r3, vp3_idct_constants
  96. vld1.64 {d0-d1}, [r3,:128]
  97. vld1.64 {d16-d19}, [r2,:128]!
  98. vld1.64 {d20-d23}, [r2,:128]!
  99. vld1.64 {d24-d27}, [r2,:128]!
  100. vadd.s16 q1, q8, q12
  101. vsub.s16 q8, q8, q12
  102. vld1.64 {d28-d31}, [r2,:128]!
  103. vp3_idct_core_neon:
  104. vmull.s16 q2, d18, xC1S7 // (ip[1] * C1) << 16
  105. vmull.s16 q3, d19, xC1S7
  106. vmull.s16 q4, d2, xC4S4 // ((ip[0] + ip[4]) * C4) << 16
  107. vmull.s16 q5, d3, xC4S4
  108. vmull.s16 q6, d16, xC4S4 // ((ip[0] - ip[4]) * C4) << 16
  109. vmull.s16 q7, d17, xC4S4
  110. vshrn.s32 d4, q2, #16
  111. vshrn.s32 d5, q3, #16
  112. vshrn.s32 d6, q4, #16
  113. vshrn.s32 d7, q5, #16
  114. vshrn.s32 d8, q6, #16
  115. vshrn.s32 d9, q7, #16
  116. vadd.s16 q12, q1, q3 // E = (ip[0] + ip[4]) * C4
  117. vadd.s16 q8, q8, q4 // F = (ip[0] - ip[4]) * C4
  118. vadd.s16 q1, q2, q9 // ip[1] * C1
  119. vmull.s16 q2, d30, xC1S7 // (ip[7] * C1) << 16
  120. vmull.s16 q3, d31, xC1S7
  121. vmull.s16 q4, d30, xC7S1 // (ip[7] * C7) << 16
  122. vmull.s16 q5, d31, xC7S1
  123. vmull.s16 q6, d18, xC7S1 // (ip[1] * C7) << 16
  124. vmull.s16 q7, d19, xC7S1
  125. vshrn.s32 d4, q2, #16
  126. vshrn.s32 d5, q3, #16
  127. vshrn.s32 d6, q4, #16 // ip[7] * C7
  128. vshrn.s32 d7, q5, #16
  129. vshrn.s32 d8, q6, #16 // ip[1] * C7
  130. vshrn.s32 d9, q7, #16
  131. vadd.s16 q2, q2, q15 // ip[7] * C1
  132. vadd.s16 q9, q1, q3 // A = ip[1] * C1 + ip[7] * C7
  133. vsub.s16 q15, q4, q2 // B = ip[1] * C7 - ip[7] * C1
  134. vmull.s16 q2, d22, xC5S3 // (ip[3] * C5) << 16
  135. vmull.s16 q3, d23, xC5S3
  136. vmull.s16 q4, d22, xC3S5 // (ip[3] * C3) << 16
  137. vmull.s16 q5, d23, xC3S5
  138. vmull.s16 q6, d26, xC5S3 // (ip[5] * C5) << 16
  139. vmull.s16 q7, d27, xC5S3
  140. vshrn.s32 d4, q2, #16
  141. vshrn.s32 d5, q3, #16
  142. vshrn.s32 d6, q4, #16
  143. vshrn.s32 d7, q5, #16
  144. vshrn.s32 d8, q6, #16
  145. vshrn.s32 d9, q7, #16
  146. vadd.s16 q3, q3, q11 // ip[3] * C3
  147. vadd.s16 q4, q4, q13 // ip[5] * C5
  148. vadd.s16 q1, q2, q11 // ip[3] * C5
  149. vadd.s16 q11, q3, q4 // C = ip[3] * C3 + ip[5] * C5
  150. vmull.s16 q2, d26, xC3S5 // (ip[5] * C3) << 16
  151. vmull.s16 q3, d27, xC3S5
  152. vmull.s16 q4, d20, xC2S6 // (ip[2] * C2) << 16
  153. vmull.s16 q5, d21, xC2S6
  154. vmull.s16 q6, d28, xC6S2 // (ip[6] * C6) << 16
  155. vmull.s16 q7, d29, xC6S2
  156. vshrn.s32 d4, q2, #16
  157. vshrn.s32 d5, q3, #16
  158. vshrn.s32 d6, q4, #16
  159. vshrn.s32 d7, q5, #16
  160. vshrn.s32 d8, q6, #16 // ip[6] * C6
  161. vshrn.s32 d9, q7, #16
  162. vadd.s16 q2, q2, q13 // ip[5] * C3
  163. vadd.s16 q3, q3, q10 // ip[2] * C2
  164. vsub.s16 q13, q2, q1 // D = ip[5] * C3 - ip[3] * C5
  165. vsub.s16 q1, q9, q11 // (A - C)
  166. vadd.s16 q11, q9, q11 // Cd = A + C
  167. vsub.s16 q9, q15, q13 // (B - D)
  168. vadd.s16 q13, q15, q13 // Dd = B + D
  169. vadd.s16 q15, q3, q4 // G = ip[2] * C2 + ip[6] * C6
  170. vmull.s16 q2, d2, xC4S4 // ((A - C) * C4) << 16
  171. vmull.s16 q3, d3, xC4S4
  172. vmull.s16 q4, d28, xC2S6 // (ip[6] * C2) << 16
  173. vmull.s16 q5, d29, xC2S6
  174. vmull.s16 q6, d20, xC6S2 // (ip[2] * C6) << 16
  175. vmull.s16 q7, d21, xC6S2
  176. vshrn.s32 d4, q2, #16
  177. vshrn.s32 d5, q3, #16
  178. vshrn.s32 d6, q4, #16
  179. vshrn.s32 d7, q5, #16
  180. vshrn.s32 d8, q6, #16 // ip[2] * C6
  181. vmull.s16 q5, d18, xC4S4 // ((B - D) * C4) << 16
  182. vmull.s16 q6, d19, xC4S4
  183. vshrn.s32 d9, q7, #16
  184. vadd.s16 q3, q3, q14 // ip[6] * C2
  185. vadd.s16 q10, q1, q2 // Ad = (A - C) * C4
  186. vsub.s16 q14, q4, q3 // H = ip[2] * C6 - ip[6] * C2
  187. bx lr
  188. endfunc
  189. .macro VP3_IDCT_END type
  190. function vp3_idct_end_\type\()_neon
  191. .ifc \type, col
  192. vdup.16 q0, r3
  193. vadd.s16 q12, q12, q0
  194. vadd.s16 q8, q8, q0
  195. .endif
  196. vshrn.s32 d2, q5, #16
  197. vshrn.s32 d3, q6, #16
  198. vadd.s16 q2, q12, q15 // Gd = E + G
  199. vadd.s16 q9, q1, q9 // (B - D) * C4
  200. vsub.s16 q12, q12, q15 // Ed = E - G
  201. vsub.s16 q3, q8, q10 // Fd = F - Ad
  202. vadd.s16 q10, q8, q10 // Add = F + Ad
  203. vadd.s16 q4, q9, q14 // Hd = Bd + H
  204. vsub.s16 q14, q9, q14 // Bdd = Bd - H
  205. vadd.s16 q8, q2, q11 // [0] = Gd + Cd
  206. vsub.s16 q15, q2, q11 // [7] = Gd - Cd
  207. vadd.s16 q9, q10, q4 // [1] = Add + Hd
  208. vsub.s16 q10, q10, q4 // [2] = Add - Hd
  209. vadd.s16 q11, q12, q13 // [3] = Ed + Dd
  210. vsub.s16 q12, q12, q13 // [4] = Ed - Dd
  211. .ifc \type, row
  212. vtrn.16 q8, q9
  213. .endif
  214. vadd.s16 q13, q3, q14 // [5] = Fd + Bdd
  215. vsub.s16 q14, q3, q14 // [6] = Fd - Bdd
  216. .ifc \type, row
  217. // 8x8 transpose
  218. vtrn.16 q10, q11
  219. vtrn.16 q12, q13
  220. vtrn.16 q14, q15
  221. vtrn.32 q8, q10
  222. vtrn.32 q9, q11
  223. vtrn.32 q12, q14
  224. vtrn.32 q13, q15
  225. vswp d17, d24
  226. vswp d19, d26
  227. vadd.s16 q1, q8, q12
  228. vswp d21, d28
  229. vsub.s16 q8, q8, q12
  230. vswp d23, d30
  231. .endif
  232. bx lr
  233. endfunc
  234. .endm
  235. VP3_IDCT_END row
  236. VP3_IDCT_END col
  237. function ff_vp3_idct_neon, export=1
  238. mov ip, lr
  239. mov r2, r0
  240. bl vp3_idct_start_neon
  241. bl vp3_idct_end_row_neon
  242. mov r3, #8
  243. bl vp3_idct_core_neon
  244. bl vp3_idct_end_col_neon
  245. mov lr, ip
  246. vpop {d8-d15}
  247. vshr.s16 q8, q8, #4
  248. vshr.s16 q9, q9, #4
  249. vshr.s16 q10, q10, #4
  250. vshr.s16 q11, q11, #4
  251. vshr.s16 q12, q12, #4
  252. vst1.64 {d16-d19}, [r0,:128]!
  253. vshr.s16 q13, q13, #4
  254. vshr.s16 q14, q14, #4
  255. vst1.64 {d20-d23}, [r0,:128]!
  256. vshr.s16 q15, q15, #4
  257. vst1.64 {d24-d27}, [r0,:128]!
  258. vst1.64 {d28-d31}, [r0,:128]!
  259. bx lr
  260. endfunc
  261. function ff_vp3_idct_put_neon, export=1
  262. mov ip, lr
  263. bl vp3_idct_start_neon
  264. bl vp3_idct_end_row_neon
  265. mov r3, #8
  266. add r3, r3, #2048 // convert signed pixel to unsigned
  267. bl vp3_idct_core_neon
  268. bl vp3_idct_end_col_neon
  269. mov lr, ip
  270. vpop {d8-d15}
  271. vqshrun.s16 d0, q8, #4
  272. vqshrun.s16 d1, q9, #4
  273. vqshrun.s16 d2, q10, #4
  274. vqshrun.s16 d3, q11, #4
  275. vst1.64 {d0}, [r0,:64], r1
  276. vqshrun.s16 d4, q12, #4
  277. vst1.64 {d1}, [r0,:64], r1
  278. vqshrun.s16 d5, q13, #4
  279. vst1.64 {d2}, [r0,:64], r1
  280. vqshrun.s16 d6, q14, #4
  281. vst1.64 {d3}, [r0,:64], r1
  282. vqshrun.s16 d7, q15, #4
  283. vst1.64 {d4}, [r0,:64], r1
  284. vst1.64 {d5}, [r0,:64], r1
  285. vst1.64 {d6}, [r0,:64], r1
  286. vst1.64 {d7}, [r0,:64], r1
  287. bx lr
  288. endfunc
  289. function ff_vp3_idct_add_neon, export=1
  290. mov ip, lr
  291. bl vp3_idct_start_neon
  292. bl vp3_idct_end_row_neon
  293. mov r3, #8
  294. bl vp3_idct_core_neon
  295. bl vp3_idct_end_col_neon
  296. mov lr, ip
  297. vpop {d8-d15}
  298. mov r2, r0
  299. vld1.64 {d0}, [r0,:64], r1
  300. vshr.s16 q8, q8, #4
  301. vld1.64 {d1}, [r0,:64], r1
  302. vshr.s16 q9, q9, #4
  303. vld1.64 {d2}, [r0,:64], r1
  304. vaddw.u8 q8, q8, d0
  305. vld1.64 {d3}, [r0,:64], r1
  306. vaddw.u8 q9, q9, d1
  307. vld1.64 {d4}, [r0,:64], r1
  308. vshr.s16 q10, q10, #4
  309. vld1.64 {d5}, [r0,:64], r1
  310. vshr.s16 q11, q11, #4
  311. vld1.64 {d6}, [r0,:64], r1
  312. vqmovun.s16 d0, q8
  313. vld1.64 {d7}, [r0,:64], r1
  314. vqmovun.s16 d1, q9
  315. vaddw.u8 q10, q10, d2
  316. vaddw.u8 q11, q11, d3
  317. vshr.s16 q12, q12, #4
  318. vshr.s16 q13, q13, #4
  319. vqmovun.s16 d2, q10
  320. vqmovun.s16 d3, q11
  321. vaddw.u8 q12, q12, d4
  322. vaddw.u8 q13, q13, d5
  323. vshr.s16 q14, q14, #4
  324. vshr.s16 q15, q15, #4
  325. vst1.64 {d0}, [r2,:64], r1
  326. vqmovun.s16 d4, q12
  327. vst1.64 {d1}, [r2,:64], r1
  328. vqmovun.s16 d5, q13
  329. vst1.64 {d2}, [r2,:64], r1
  330. vaddw.u8 q14, q14, d6
  331. vst1.64 {d3}, [r2,:64], r1
  332. vaddw.u8 q15, q15, d7
  333. vst1.64 {d4}, [r2,:64], r1
  334. vqmovun.s16 d6, q14
  335. vst1.64 {d5}, [r2,:64], r1
  336. vqmovun.s16 d7, q15
  337. vst1.64 {d6}, [r2,:64], r1
  338. vst1.64 {d7}, [r2,:64], r1
  339. bx lr
  340. endfunc
  341. function ff_vp3_idct_dc_add_neon, export=1
  342. ldrsh r2, [r2]
  343. mov r3, r0
  344. add r2, r2, #15
  345. vdup.16 q15, r2
  346. vshr.s16 q15, q15, #5
  347. vld1.8 {d0}, [r0,:64], r1
  348. vld1.8 {d1}, [r0,:64], r1
  349. vld1.8 {d2}, [r0,:64], r1
  350. vaddw.u8 q8, q15, d0
  351. vld1.8 {d3}, [r0,:64], r1
  352. vaddw.u8 q9, q15, d1
  353. vld1.8 {d4}, [r0,:64], r1
  354. vaddw.u8 q10, q15, d2
  355. vld1.8 {d5}, [r0,:64], r1
  356. vaddw.u8 q11, q15, d3
  357. vld1.8 {d6}, [r0,:64], r1
  358. vaddw.u8 q12, q15, d4
  359. vld1.8 {d7}, [r0,:64], r1
  360. vaddw.u8 q13, q15, d5
  361. vqmovun.s16 d0, q8
  362. vaddw.u8 q14, q15, d6
  363. vqmovun.s16 d1, q9
  364. vaddw.u8 q15, q15, d7
  365. vqmovun.s16 d2, q10
  366. vst1.8 {d0}, [r3,:64], r1
  367. vqmovun.s16 d3, q11
  368. vst1.8 {d1}, [r3,:64], r1
  369. vqmovun.s16 d4, q12
  370. vst1.8 {d2}, [r3,:64], r1
  371. vqmovun.s16 d5, q13
  372. vst1.8 {d3}, [r3,:64], r1
  373. vqmovun.s16 d6, q14
  374. vst1.8 {d4}, [r3,:64], r1
  375. vqmovun.s16 d7, q15
  376. vst1.8 {d5}, [r3,:64], r1
  377. vst1.8 {d6}, [r3,:64], r1
  378. vst1.8 {d7}, [r3,:64], r1
  379. bx lr
  380. endfunc