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.

415 lines
14KB

  1. /*
  2. * Copyright (c) 2009 David Conrad
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg 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. * FFmpeg 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 FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #include "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. endfunc
  104. function vp3_idct_core_neon
  105. vmull.s16 q2, d18, xC1S7 // (ip[1] * C1) << 16
  106. vmull.s16 q3, d19, xC1S7
  107. vmull.s16 q4, d2, xC4S4 // ((ip[0] + ip[4]) * C4) << 16
  108. vmull.s16 q5, d3, xC4S4
  109. vmull.s16 q6, d16, xC4S4 // ((ip[0] - ip[4]) * C4) << 16
  110. vmull.s16 q7, d17, xC4S4
  111. vshrn.s32 d4, q2, #16
  112. vshrn.s32 d5, q3, #16
  113. vshrn.s32 d6, q4, #16
  114. vshrn.s32 d7, q5, #16
  115. vshrn.s32 d8, q6, #16
  116. vshrn.s32 d9, q7, #16
  117. vadd.s16 q12, q1, q3 // E = (ip[0] + ip[4]) * C4
  118. vadd.s16 q8, q8, q4 // F = (ip[0] - ip[4]) * C4
  119. vadd.s16 q1, q2, q9 // ip[1] * C1
  120. vmull.s16 q2, d30, xC1S7 // (ip[7] * C1) << 16
  121. vmull.s16 q3, d31, xC1S7
  122. vmull.s16 q4, d30, xC7S1 // (ip[7] * C7) << 16
  123. vmull.s16 q5, d31, xC7S1
  124. vmull.s16 q6, d18, xC7S1 // (ip[1] * C7) << 16
  125. vmull.s16 q7, d19, xC7S1
  126. vshrn.s32 d4, q2, #16
  127. vshrn.s32 d5, q3, #16
  128. vshrn.s32 d6, q4, #16 // ip[7] * C7
  129. vshrn.s32 d7, q5, #16
  130. vshrn.s32 d8, q6, #16 // ip[1] * C7
  131. vshrn.s32 d9, q7, #16
  132. vadd.s16 q2, q2, q15 // ip[7] * C1
  133. vadd.s16 q9, q1, q3 // A = ip[1] * C1 + ip[7] * C7
  134. vsub.s16 q15, q4, q2 // B = ip[1] * C7 - ip[7] * C1
  135. vmull.s16 q2, d22, xC5S3 // (ip[3] * C5) << 16
  136. vmull.s16 q3, d23, xC5S3
  137. vmull.s16 q4, d22, xC3S5 // (ip[3] * C3) << 16
  138. vmull.s16 q5, d23, xC3S5
  139. vmull.s16 q6, d26, xC5S3 // (ip[5] * C5) << 16
  140. vmull.s16 q7, d27, xC5S3
  141. vshrn.s32 d4, q2, #16
  142. vshrn.s32 d5, q3, #16
  143. vshrn.s32 d6, q4, #16
  144. vshrn.s32 d7, q5, #16
  145. vshrn.s32 d8, q6, #16
  146. vshrn.s32 d9, q7, #16
  147. vadd.s16 q3, q3, q11 // ip[3] * C3
  148. vadd.s16 q4, q4, q13 // ip[5] * C5
  149. vadd.s16 q1, q2, q11 // ip[3] * C5
  150. vadd.s16 q11, q3, q4 // C = ip[3] * C3 + ip[5] * C5
  151. vmull.s16 q2, d26, xC3S5 // (ip[5] * C3) << 16
  152. vmull.s16 q3, d27, xC3S5
  153. vmull.s16 q4, d20, xC2S6 // (ip[2] * C2) << 16
  154. vmull.s16 q5, d21, xC2S6
  155. vmull.s16 q6, d28, xC6S2 // (ip[6] * C6) << 16
  156. vmull.s16 q7, d29, xC6S2
  157. vshrn.s32 d4, q2, #16
  158. vshrn.s32 d5, q3, #16
  159. vshrn.s32 d6, q4, #16
  160. vshrn.s32 d7, q5, #16
  161. vshrn.s32 d8, q6, #16 // ip[6] * C6
  162. vshrn.s32 d9, q7, #16
  163. vadd.s16 q2, q2, q13 // ip[5] * C3
  164. vadd.s16 q3, q3, q10 // ip[2] * C2
  165. vsub.s16 q13, q2, q1 // D = ip[5] * C3 - ip[3] * C5
  166. vsub.s16 q1, q9, q11 // (A - C)
  167. vadd.s16 q11, q9, q11 // Cd = A + C
  168. vsub.s16 q9, q15, q13 // (B - D)
  169. vadd.s16 q13, q15, q13 // Dd = B + D
  170. vadd.s16 q15, q3, q4 // G = ip[2] * C2 + ip[6] * C6
  171. vmull.s16 q2, d2, xC4S4 // ((A - C) * C4) << 16
  172. vmull.s16 q3, d3, xC4S4
  173. vmull.s16 q4, d28, xC2S6 // (ip[6] * C2) << 16
  174. vmull.s16 q5, d29, xC2S6
  175. vmull.s16 q6, d20, xC6S2 // (ip[2] * C6) << 16
  176. vmull.s16 q7, d21, xC6S2
  177. vshrn.s32 d4, q2, #16
  178. vshrn.s32 d5, q3, #16
  179. vshrn.s32 d6, q4, #16
  180. vshrn.s32 d7, q5, #16
  181. vshrn.s32 d8, q6, #16 // ip[2] * C6
  182. vmull.s16 q5, d18, xC4S4 // ((B - D) * C4) << 16
  183. vmull.s16 q6, d19, xC4S4
  184. vshrn.s32 d9, q7, #16
  185. vadd.s16 q3, q3, q14 // ip[6] * C2
  186. vadd.s16 q10, q1, q2 // Ad = (A - C) * C4
  187. vsub.s16 q14, q4, q3 // H = ip[2] * C6 - ip[6] * C2
  188. bx lr
  189. endfunc
  190. .macro VP3_IDCT_END type
  191. function vp3_idct_end_\type\()_neon
  192. .ifc \type, col
  193. vdup.16 q0, r3
  194. vadd.s16 q12, q12, q0
  195. vadd.s16 q8, q8, q0
  196. .endif
  197. vshrn.s32 d2, q5, #16
  198. vshrn.s32 d3, q6, #16
  199. vadd.s16 q2, q12, q15 // Gd = E + G
  200. vadd.s16 q9, q1, q9 // (B - D) * C4
  201. vsub.s16 q12, q12, q15 // Ed = E - G
  202. vsub.s16 q3, q8, q10 // Fd = F - Ad
  203. vadd.s16 q10, q8, q10 // Add = F + Ad
  204. vadd.s16 q4, q9, q14 // Hd = Bd + H
  205. vsub.s16 q14, q9, q14 // Bdd = Bd - H
  206. vadd.s16 q8, q2, q11 // [0] = Gd + Cd
  207. vsub.s16 q15, q2, q11 // [7] = Gd - Cd
  208. vadd.s16 q9, q10, q4 // [1] = Add + Hd
  209. vsub.s16 q10, q10, q4 // [2] = Add - Hd
  210. vadd.s16 q11, q12, q13 // [3] = Ed + Dd
  211. vsub.s16 q12, q12, q13 // [4] = Ed - Dd
  212. .ifc \type, row
  213. vtrn.16 q8, q9
  214. .endif
  215. vadd.s16 q13, q3, q14 // [5] = Fd + Bdd
  216. vsub.s16 q14, q3, q14 // [6] = Fd - Bdd
  217. .ifc \type, row
  218. // 8x8 transpose
  219. vtrn.16 q10, q11
  220. vtrn.16 q12, q13
  221. vtrn.16 q14, q15
  222. vtrn.32 q8, q10
  223. vtrn.32 q9, q11
  224. vtrn.32 q12, q14
  225. vtrn.32 q13, q15
  226. vswp d17, d24
  227. vswp d19, d26
  228. vadd.s16 q1, q8, q12
  229. vswp d21, d28
  230. vsub.s16 q8, q8, q12
  231. vswp d23, d30
  232. .endif
  233. bx lr
  234. endfunc
  235. .endm
  236. VP3_IDCT_END row
  237. VP3_IDCT_END col
  238. function ff_vp3_idct_neon, export=1
  239. mov ip, lr
  240. mov r2, r0
  241. bl vp3_idct_start_neon
  242. bl vp3_idct_end_row_neon
  243. mov r3, #8
  244. bl vp3_idct_core_neon
  245. bl vp3_idct_end_col_neon
  246. mov lr, ip
  247. vpop {d8-d15}
  248. vshr.s16 q8, q8, #4
  249. vshr.s16 q9, q9, #4
  250. vshr.s16 q10, q10, #4
  251. vshr.s16 q11, q11, #4
  252. vshr.s16 q12, q12, #4
  253. vst1.64 {d16-d19}, [r0,:128]!
  254. vshr.s16 q13, q13, #4
  255. vshr.s16 q14, q14, #4
  256. vst1.64 {d20-d23}, [r0,:128]!
  257. vshr.s16 q15, q15, #4
  258. vst1.64 {d24-d27}, [r0,:128]!
  259. vst1.64 {d28-d31}, [r0,:128]!
  260. bx lr
  261. endfunc
  262. function ff_vp3_idct_put_neon, export=1
  263. mov ip, lr
  264. bl vp3_idct_start_neon
  265. bl vp3_idct_end_row_neon
  266. mov r3, #8
  267. add r3, r3, #2048 // convert signed pixel to unsigned
  268. bl vp3_idct_core_neon
  269. bl vp3_idct_end_col_neon
  270. mov lr, ip
  271. vpop {d8-d15}
  272. vqshrun.s16 d0, q8, #4
  273. vqshrun.s16 d1, q9, #4
  274. vqshrun.s16 d2, q10, #4
  275. vqshrun.s16 d3, q11, #4
  276. vst1.64 {d0}, [r0,:64], r1
  277. vqshrun.s16 d4, q12, #4
  278. vst1.64 {d1}, [r0,:64], r1
  279. vqshrun.s16 d5, q13, #4
  280. vst1.64 {d2}, [r0,:64], r1
  281. vqshrun.s16 d6, q14, #4
  282. vst1.64 {d3}, [r0,:64], r1
  283. vqshrun.s16 d7, q15, #4
  284. vst1.64 {d4}, [r0,:64], r1
  285. vst1.64 {d5}, [r0,:64], r1
  286. vst1.64 {d6}, [r0,:64], r1
  287. vst1.64 {d7}, [r0,:64], r1
  288. bx lr
  289. endfunc
  290. function ff_vp3_idct_add_neon, export=1
  291. mov ip, lr
  292. bl vp3_idct_start_neon
  293. bl vp3_idct_end_row_neon
  294. mov r3, #8
  295. bl vp3_idct_core_neon
  296. bl vp3_idct_end_col_neon
  297. mov lr, ip
  298. vpop {d8-d15}
  299. mov r2, r0
  300. vld1.64 {d0}, [r0,:64], r1
  301. vshr.s16 q8, q8, #4
  302. vld1.64 {d1}, [r0,:64], r1
  303. vshr.s16 q9, q9, #4
  304. vld1.64 {d2}, [r0,:64], r1
  305. vaddw.u8 q8, q8, d0
  306. vld1.64 {d3}, [r0,:64], r1
  307. vaddw.u8 q9, q9, d1
  308. vld1.64 {d4}, [r0,:64], r1
  309. vshr.s16 q10, q10, #4
  310. vld1.64 {d5}, [r0,:64], r1
  311. vshr.s16 q11, q11, #4
  312. vld1.64 {d6}, [r0,:64], r1
  313. vqmovun.s16 d0, q8
  314. vld1.64 {d7}, [r0,:64], r1
  315. vqmovun.s16 d1, q9
  316. vaddw.u8 q10, q10, d2
  317. vaddw.u8 q11, q11, d3
  318. vshr.s16 q12, q12, #4
  319. vshr.s16 q13, q13, #4
  320. vqmovun.s16 d2, q10
  321. vqmovun.s16 d3, q11
  322. vaddw.u8 q12, q12, d4
  323. vaddw.u8 q13, q13, d5
  324. vshr.s16 q14, q14, #4
  325. vshr.s16 q15, q15, #4
  326. vst1.64 {d0}, [r2,:64], r1
  327. vqmovun.s16 d4, q12
  328. vst1.64 {d1}, [r2,:64], r1
  329. vqmovun.s16 d5, q13
  330. vst1.64 {d2}, [r2,:64], r1
  331. vaddw.u8 q14, q14, d6
  332. vst1.64 {d3}, [r2,:64], r1
  333. vaddw.u8 q15, q15, d7
  334. vst1.64 {d4}, [r2,:64], r1
  335. vqmovun.s16 d6, q14
  336. vst1.64 {d5}, [r2,:64], r1
  337. vqmovun.s16 d7, q15
  338. vst1.64 {d6}, [r2,:64], r1
  339. vst1.64 {d7}, [r2,:64], r1
  340. bx lr
  341. endfunc
  342. function ff_vp3_idct_dc_add_neon, export=1
  343. ldrsh r2, [r2]
  344. mov r3, r0
  345. add r2, r2, #15
  346. vdup.16 q15, r2
  347. vshr.s16 q15, q15, #5
  348. vld1.8 {d0}, [r0,:64], r1
  349. vld1.8 {d1}, [r0,:64], r1
  350. vld1.8 {d2}, [r0,:64], r1
  351. vaddw.u8 q8, q15, d0
  352. vld1.8 {d3}, [r0,:64], r1
  353. vaddw.u8 q9, q15, d1
  354. vld1.8 {d4}, [r0,:64], r1
  355. vaddw.u8 q10, q15, d2
  356. vld1.8 {d5}, [r0,:64], r1
  357. vaddw.u8 q11, q15, d3
  358. vld1.8 {d6}, [r0,:64], r1
  359. vaddw.u8 q12, q15, d4
  360. vld1.8 {d7}, [r0,:64], r1
  361. vaddw.u8 q13, q15, d5
  362. vqmovun.s16 d0, q8
  363. vaddw.u8 q14, q15, d6
  364. vqmovun.s16 d1, q9
  365. vaddw.u8 q15, q15, d7
  366. vqmovun.s16 d2, q10
  367. vst1.8 {d0}, [r3,:64], r1
  368. vqmovun.s16 d3, q11
  369. vst1.8 {d1}, [r3,:64], r1
  370. vqmovun.s16 d4, q12
  371. vst1.8 {d2}, [r3,:64], r1
  372. vqmovun.s16 d5, q13
  373. vst1.8 {d3}, [r3,:64], r1
  374. vqmovun.s16 d6, q14
  375. vst1.8 {d4}, [r3,:64], r1
  376. vqmovun.s16 d7, q15
  377. vst1.8 {d5}, [r3,:64], r1
  378. vst1.8 {d6}, [r3,:64], r1
  379. vst1.8 {d7}, [r3,:64], r1
  380. bx lr
  381. endfunc