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) 2008 Mans Rullgard <mans@mansr.com>
  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. function ff_h264_idct_add_neon, export=1
  22. vld1.64 {d0-d3}, [r1,:128]
  23. vmov.i16 q15, #0
  24. vswp d1, d2
  25. vst1.16 {q15}, [r1,:128]!
  26. vadd.i16 d4, d0, d1
  27. vst1.16 {q15}, [r1,:128]!
  28. vshr.s16 q8, q1, #1
  29. vsub.i16 d5, d0, d1
  30. vadd.i16 d6, d2, d17
  31. vsub.i16 d7, d16, d3
  32. vadd.i16 q0, q2, q3
  33. vsub.i16 q1, q2, q3
  34. vtrn.16 d0, d1
  35. vtrn.16 d3, d2
  36. vtrn.32 d0, d3
  37. vtrn.32 d1, d2
  38. vadd.i16 d4, d0, d3
  39. vld1.32 {d18[0]}, [r0,:32], r2
  40. vswp d1, d3
  41. vshr.s16 q8, q1, #1
  42. vld1.32 {d19[1]}, [r0,:32], r2
  43. vsub.i16 d5, d0, d1
  44. vld1.32 {d18[1]}, [r0,:32], r2
  45. vadd.i16 d6, d16, d3
  46. vld1.32 {d19[0]}, [r0,:32], r2
  47. vsub.i16 d7, d2, d17
  48. sub r0, r0, r2, lsl #2
  49. vadd.i16 q0, q2, q3
  50. vsub.i16 q1, q2, q3
  51. vrshr.s16 q0, q0, #6
  52. vrshr.s16 q1, q1, #6
  53. vaddw.u8 q0, q0, d18
  54. vaddw.u8 q1, q1, d19
  55. vqmovun.s16 d0, q0
  56. vqmovun.s16 d1, q1
  57. vst1.32 {d0[0]}, [r0,:32], r2
  58. vst1.32 {d1[1]}, [r0,:32], r2
  59. vst1.32 {d0[1]}, [r0,:32], r2
  60. vst1.32 {d1[0]}, [r0,:32], r2
  61. sub r1, r1, #32
  62. bx lr
  63. endfunc
  64. function ff_h264_idct_dc_add_neon, export=1
  65. mov r3, #0
  66. vld1.16 {d2[],d3[]}, [r1,:16]
  67. strh r3, [r1]
  68. vrshr.s16 q1, q1, #6
  69. vld1.32 {d0[0]}, [r0,:32], r2
  70. vld1.32 {d0[1]}, [r0,:32], r2
  71. vaddw.u8 q2, q1, d0
  72. vld1.32 {d1[0]}, [r0,:32], r2
  73. vld1.32 {d1[1]}, [r0,:32], r2
  74. vaddw.u8 q1, q1, d1
  75. vqmovun.s16 d0, q2
  76. vqmovun.s16 d1, q1
  77. sub r0, r0, r2, lsl #2
  78. vst1.32 {d0[0]}, [r0,:32], r2
  79. vst1.32 {d0[1]}, [r0,:32], r2
  80. vst1.32 {d1[0]}, [r0,:32], r2
  81. vst1.32 {d1[1]}, [r0,:32], r2
  82. bx lr
  83. endfunc
  84. function ff_h264_idct_add16_neon, export=1
  85. push {r4-r8,lr}
  86. mov r4, r0
  87. mov r5, r1
  88. mov r1, r2
  89. mov r2, r3
  90. ldr r6, [sp, #24]
  91. movrel r7, scan8
  92. mov ip, #16
  93. 1: ldrb r8, [r7], #1
  94. ldr r0, [r5], #4
  95. ldrb r8, [r6, r8]
  96. subs r8, r8, #1
  97. blt 2f
  98. ldrsh lr, [r1]
  99. add r0, r0, r4
  100. it ne
  101. movne lr, #0
  102. cmp lr, #0
  103. ite ne
  104. adrne lr, X(ff_h264_idct_dc_add_neon) + CONFIG_THUMB
  105. adreq lr, X(ff_h264_idct_add_neon) + CONFIG_THUMB
  106. blx lr
  107. 2: subs ip, ip, #1
  108. add r1, r1, #32
  109. bne 1b
  110. pop {r4-r8,pc}
  111. endfunc
  112. function ff_h264_idct_add16intra_neon, export=1
  113. push {r4-r8,lr}
  114. mov r4, r0
  115. mov r5, r1
  116. mov r1, r2
  117. mov r2, r3
  118. ldr r6, [sp, #24]
  119. movrel r7, scan8
  120. mov ip, #16
  121. 1: ldrb r8, [r7], #1
  122. ldr r0, [r5], #4
  123. ldrb r8, [r6, r8]
  124. add r0, r0, r4
  125. cmp r8, #0
  126. ldrsh r8, [r1]
  127. iteet ne
  128. adrne lr, X(ff_h264_idct_add_neon) + CONFIG_THUMB
  129. adreq lr, X(ff_h264_idct_dc_add_neon) + CONFIG_THUMB
  130. cmpeq r8, #0
  131. blxne lr
  132. subs ip, ip, #1
  133. add r1, r1, #32
  134. bne 1b
  135. pop {r4-r8,pc}
  136. endfunc
  137. function ff_h264_idct_add8_neon, export=1
  138. push {r4-r10,lr}
  139. ldm r0, {r4,r9}
  140. add r5, r1, #16*4
  141. add r1, r2, #16*32
  142. mov r2, r3
  143. mov r10, r1
  144. ldr r6, [sp, #32]
  145. movrel r7, scan8+16
  146. mov r12, #0
  147. 1: ldrb r8, [r7, r12]
  148. ldr r0, [r5, r12, lsl #2]
  149. ldrb r8, [r6, r8]
  150. add r0, r0, r4
  151. add r1, r10, r12, lsl #5
  152. cmp r8, #0
  153. ldrsh r8, [r1]
  154. iteet ne
  155. adrne lr, X(ff_h264_idct_add_neon) + CONFIG_THUMB
  156. adreq lr, X(ff_h264_idct_dc_add_neon) + CONFIG_THUMB
  157. cmpeq r8, #0
  158. blxne lr
  159. add r12, r12, #1
  160. cmp r12, #4
  161. itt eq
  162. moveq r12, #16
  163. moveq r4, r9
  164. cmp r12, #20
  165. blt 1b
  166. pop {r4-r10,pc}
  167. endfunc
  168. .macro idct8x8_cols pass
  169. .if \pass == 0
  170. qa .req q2
  171. qb .req q14
  172. vshr.s16 q2, q10, #1
  173. vadd.i16 q0, q8, q12
  174. vld1.16 {q14-q15},[r1,:128]
  175. vst1.16 {q3}, [r1,:128]!
  176. vst1.16 {q3}, [r1,:128]!
  177. vsub.i16 q1, q8, q12
  178. vshr.s16 q3, q14, #1
  179. vsub.i16 q2, q2, q14
  180. vadd.i16 q3, q3, q10
  181. .else
  182. qa .req q14
  183. qb .req q2
  184. vtrn.32 q8, q10
  185. vtrn.16 q12, q13
  186. vtrn.32 q9, q11
  187. vtrn.32 q12, q2
  188. vtrn.32 q13, q15
  189. vswp d21, d4
  190. vshr.s16 q14, q10, #1
  191. vswp d17, d24
  192. vshr.s16 q3, q2, #1
  193. vswp d19, d26
  194. vadd.i16 q0, q8, q12
  195. vswp d23, d30
  196. vsub.i16 q1, q8, q12
  197. vsub.i16 q14, q14, q2
  198. vadd.i16 q3, q3, q10
  199. .endif
  200. vadd.i16 q10, q1, qa
  201. vsub.i16 q12, q1, qa
  202. vadd.i16 q8, q0, q3
  203. vsub.i16 qb, q0, q3
  204. vsub.i16 q0, q13, q11
  205. vadd.i16 q1, q15, q9
  206. vsub.i16 qa, q15, q9
  207. vadd.i16 q3, q13, q11
  208. vsub.i16 q0, q0, q15
  209. vsub.i16 q1, q1, q11
  210. vadd.i16 qa, qa, q13
  211. vadd.i16 q3, q3, q9
  212. vshr.s16 q9, q9, #1
  213. vshr.s16 q11, q11, #1
  214. vshr.s16 q13, q13, #1
  215. vshr.s16 q15, q15, #1
  216. vsub.i16 q0, q0, q15
  217. vsub.i16 q1, q1, q11
  218. vadd.i16 qa, qa, q13
  219. vadd.i16 q3, q3, q9
  220. vshr.s16 q9, q0, #2
  221. vshr.s16 q11, q1, #2
  222. vshr.s16 q13, qa, #2
  223. vshr.s16 q15, q3, #2
  224. vsub.i16 q3, q3, q9
  225. vsub.i16 qa, q11, qa
  226. vadd.i16 q1, q1, q13
  227. vadd.i16 q0, q0, q15
  228. .if \pass == 0
  229. vsub.i16 q15, q8, q3
  230. vadd.i16 q8, q8, q3
  231. vadd.i16 q9, q10, q2
  232. vsub.i16 q2, q10, q2
  233. vtrn.16 q8, q9
  234. vadd.i16 q10, q12, q1
  235. vtrn.16 q2, q15
  236. vadd.i16 q11, q14, q0
  237. vsub.i16 q13, q12, q1
  238. vtrn.16 q10, q11
  239. vsub.i16 q12, q14, q0
  240. .else
  241. vsub.i16 q15, q8, q3
  242. vadd.i16 q8, q8, q3
  243. vadd.i16 q9, q10, q14
  244. vsub.i16 q14, q10, q14
  245. vadd.i16 q10, q12, q1
  246. vsub.i16 q13, q12, q1
  247. vadd.i16 q11, q2, q0
  248. vsub.i16 q12, q2, q0
  249. .endif
  250. .unreq qa
  251. .unreq qb
  252. .endm
  253. function ff_h264_idct8_add_neon, export=1
  254. vmov.i16 q3, #0
  255. vld1.16 {q8-q9}, [r1,:128]
  256. vst1.16 {q3}, [r1,:128]!
  257. vst1.16 {q3}, [r1,:128]!
  258. vld1.16 {q10-q11},[r1,:128]
  259. vst1.16 {q3}, [r1,:128]!
  260. vst1.16 {q3}, [r1,:128]!
  261. vld1.16 {q12-q13},[r1,:128]
  262. vst1.16 {q3}, [r1,:128]!
  263. vst1.16 {q3}, [r1,:128]!
  264. idct8x8_cols 0
  265. idct8x8_cols 1
  266. mov r3, r0
  267. vrshr.s16 q8, q8, #6
  268. vld1.8 {d0}, [r0,:64], r2
  269. vrshr.s16 q9, q9, #6
  270. vld1.8 {d1}, [r0,:64], r2
  271. vrshr.s16 q10, q10, #6
  272. vld1.8 {d2}, [r0,:64], r2
  273. vrshr.s16 q11, q11, #6
  274. vld1.8 {d3}, [r0,:64], r2
  275. vrshr.s16 q12, q12, #6
  276. vld1.8 {d4}, [r0,:64], r2
  277. vrshr.s16 q13, q13, #6
  278. vld1.8 {d5}, [r0,:64], r2
  279. vrshr.s16 q14, q14, #6
  280. vld1.8 {d6}, [r0,:64], r2
  281. vrshr.s16 q15, q15, #6
  282. vld1.8 {d7}, [r0,:64], r2
  283. vaddw.u8 q8, q8, d0
  284. vaddw.u8 q9, q9, d1
  285. vaddw.u8 q10, q10, d2
  286. vqmovun.s16 d0, q8
  287. vaddw.u8 q11, q11, d3
  288. vqmovun.s16 d1, q9
  289. vaddw.u8 q12, q12, d4
  290. vqmovun.s16 d2, q10
  291. vst1.8 {d0}, [r3,:64], r2
  292. vaddw.u8 q13, q13, d5
  293. vqmovun.s16 d3, q11
  294. vst1.8 {d1}, [r3,:64], r2
  295. vaddw.u8 q14, q14, d6
  296. vqmovun.s16 d4, q12
  297. vst1.8 {d2}, [r3,:64], r2
  298. vaddw.u8 q15, q15, d7
  299. vqmovun.s16 d5, q13
  300. vst1.8 {d3}, [r3,:64], r2
  301. vqmovun.s16 d6, q14
  302. vqmovun.s16 d7, q15
  303. vst1.8 {d4}, [r3,:64], r2
  304. vst1.8 {d5}, [r3,:64], r2
  305. vst1.8 {d6}, [r3,:64], r2
  306. vst1.8 {d7}, [r3,:64], r2
  307. sub r1, r1, #128
  308. bx lr
  309. endfunc
  310. function ff_h264_idct8_dc_add_neon, export=1
  311. mov r3, #0
  312. vld1.16 {d30[],d31[]},[r1,:16]
  313. strh r3, [r1]
  314. vld1.32 {d0}, [r0,:64], r2
  315. vrshr.s16 q15, q15, #6
  316. vld1.32 {d1}, [r0,:64], r2
  317. vld1.32 {d2}, [r0,:64], r2
  318. vaddw.u8 q8, q15, d0
  319. vld1.32 {d3}, [r0,:64], r2
  320. vaddw.u8 q9, q15, d1
  321. vld1.32 {d4}, [r0,:64], r2
  322. vaddw.u8 q10, q15, d2
  323. vld1.32 {d5}, [r0,:64], r2
  324. vaddw.u8 q11, q15, d3
  325. vld1.32 {d6}, [r0,:64], r2
  326. vaddw.u8 q12, q15, d4
  327. vld1.32 {d7}, [r0,:64], r2
  328. vaddw.u8 q13, q15, d5
  329. vaddw.u8 q14, q15, d6
  330. vaddw.u8 q15, q15, d7
  331. vqmovun.s16 d0, q8
  332. vqmovun.s16 d1, q9
  333. vqmovun.s16 d2, q10
  334. vqmovun.s16 d3, q11
  335. sub r0, r0, r2, lsl #3
  336. vst1.32 {d0}, [r0,:64], r2
  337. vqmovun.s16 d4, q12
  338. vst1.32 {d1}, [r0,:64], r2
  339. vqmovun.s16 d5, q13
  340. vst1.32 {d2}, [r0,:64], r2
  341. vqmovun.s16 d6, q14
  342. vst1.32 {d3}, [r0,:64], r2
  343. vqmovun.s16 d7, q15
  344. vst1.32 {d4}, [r0,:64], r2
  345. vst1.32 {d5}, [r0,:64], r2
  346. vst1.32 {d6}, [r0,:64], r2
  347. vst1.32 {d7}, [r0,:64], r2
  348. bx lr
  349. endfunc
  350. function ff_h264_idct8_add4_neon, export=1
  351. push {r4-r8,lr}
  352. mov r4, r0
  353. mov r5, r1
  354. mov r1, r2
  355. mov r2, r3
  356. ldr r6, [sp, #24]
  357. movrel r7, scan8
  358. mov r12, #16
  359. 1: ldrb r8, [r7], #4
  360. ldr r0, [r5], #16
  361. ldrb r8, [r6, r8]
  362. subs r8, r8, #1
  363. blt 2f
  364. ldrsh lr, [r1]
  365. add r0, r0, r4
  366. it ne
  367. movne lr, #0
  368. cmp lr, #0
  369. ite ne
  370. adrne lr, X(ff_h264_idct8_dc_add_neon) + CONFIG_THUMB
  371. adreq lr, X(ff_h264_idct8_add_neon) + CONFIG_THUMB
  372. blx lr
  373. 2: subs r12, r12, #4
  374. add r1, r1, #128
  375. bne 1b
  376. pop {r4-r8,pc}
  377. endfunc
  378. const scan8
  379. .byte 4+ 1*8, 5+ 1*8, 4+ 2*8, 5+ 2*8
  380. .byte 6+ 1*8, 7+ 1*8, 6+ 2*8, 7+ 2*8
  381. .byte 4+ 3*8, 5+ 3*8, 4+ 4*8, 5+ 4*8
  382. .byte 6+ 3*8, 7+ 3*8, 6+ 4*8, 7+ 4*8
  383. .byte 4+ 6*8, 5+ 6*8, 4+ 7*8, 5+ 7*8
  384. .byte 6+ 6*8, 7+ 6*8, 6+ 7*8, 7+ 7*8
  385. .byte 4+ 8*8, 5+ 8*8, 4+ 9*8, 5+ 9*8
  386. .byte 6+ 8*8, 7+ 8*8, 6+ 9*8, 7+ 9*8
  387. .byte 4+11*8, 5+11*8, 4+12*8, 5+12*8
  388. .byte 6+11*8, 7+11*8, 6+12*8, 7+12*8
  389. .byte 4+13*8, 5+13*8, 4+14*8, 5+14*8
  390. .byte 6+13*8, 7+13*8, 6+14*8, 7+14*8
  391. endconst