intrapred_neon.c 37 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043
  1. /*
  2. * Copyright (c) 2014 The WebM project authors. All Rights Reserved.
  3. *
  4. * Use of this source code is governed by a BSD-style license
  5. * that can be found in the LICENSE file in the root of the source
  6. * tree. An additional intellectual property rights grant can be found
  7. * in the file PATENTS. All contributing project authors may
  8. * be found in the AUTHORS file in the root of the source tree.
  9. */
  10. #include <arm_neon.h>
  11. #include "./vpx_config.h"
  12. #include "./vpx_dsp_rtcd.h"
  13. #include "vpx/vpx_integer.h"
  14. //------------------------------------------------------------------------------
  15. // DC 4x4
  16. static INLINE uint16x4_t dc_sum_4(const uint8_t *ref) {
  17. const uint8x8_t ref_u8 = vld1_u8(ref);
  18. const uint16x4_t p0 = vpaddl_u8(ref_u8);
  19. return vpadd_u16(p0, p0);
  20. }
  21. static INLINE void dc_store_4x4(uint8_t *dst, ptrdiff_t stride,
  22. const uint8x8_t dc) {
  23. const uint8x8_t dc_dup = vdup_lane_u8(dc, 0);
  24. int i;
  25. for (i = 0; i < 4; ++i, dst += stride) {
  26. vst1_lane_u32((uint32_t *)dst, vreinterpret_u32_u8(dc_dup), 0);
  27. }
  28. }
  29. void vpx_dc_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  30. const uint8_t *above, const uint8_t *left) {
  31. const uint8x8_t a = vld1_u8(above);
  32. const uint8x8_t l = vld1_u8(left);
  33. const uint16x8_t al = vaddl_u8(a, l);
  34. uint16x4_t sum;
  35. uint8x8_t dc;
  36. sum = vpadd_u16(vget_low_u16(al), vget_low_u16(al));
  37. sum = vpadd_u16(sum, sum);
  38. dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 3));
  39. dc_store_4x4(dst, stride, dc);
  40. }
  41. void vpx_dc_left_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  42. const uint8_t *above, const uint8_t *left) {
  43. const uint16x4_t sum = dc_sum_4(left);
  44. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 2));
  45. (void)above;
  46. dc_store_4x4(dst, stride, dc);
  47. }
  48. void vpx_dc_top_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  49. const uint8_t *above, const uint8_t *left) {
  50. const uint16x4_t sum = dc_sum_4(above);
  51. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 2));
  52. (void)left;
  53. dc_store_4x4(dst, stride, dc);
  54. }
  55. void vpx_dc_128_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  56. const uint8_t *above, const uint8_t *left) {
  57. const uint8x8_t dc = vdup_n_u8(0x80);
  58. (void)above;
  59. (void)left;
  60. dc_store_4x4(dst, stride, dc);
  61. }
  62. //------------------------------------------------------------------------------
  63. // DC 8x8
  64. static INLINE uint16x4_t dc_sum_8(const uint8_t *ref) {
  65. const uint8x8_t ref_u8 = vld1_u8(ref);
  66. uint16x4_t sum = vpaddl_u8(ref_u8);
  67. sum = vpadd_u16(sum, sum);
  68. return vpadd_u16(sum, sum);
  69. }
  70. static INLINE void dc_store_8x8(uint8_t *dst, ptrdiff_t stride,
  71. const uint8x8_t dc) {
  72. const uint8x8_t dc_dup = vdup_lane_u8(dc, 0);
  73. int i;
  74. for (i = 0; i < 8; ++i, dst += stride) {
  75. vst1_u8(dst, dc_dup);
  76. }
  77. }
  78. void vpx_dc_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  79. const uint8_t *above, const uint8_t *left) {
  80. const uint8x8_t above_u8 = vld1_u8(above);
  81. const uint8x8_t left_u8 = vld1_u8(left);
  82. const uint8x16_t above_and_left = vcombine_u8(above_u8, left_u8);
  83. const uint16x8_t p0 = vpaddlq_u8(above_and_left);
  84. uint16x4_t sum = vadd_u16(vget_low_u16(p0), vget_high_u16(p0));
  85. uint8x8_t dc;
  86. sum = vpadd_u16(sum, sum);
  87. sum = vpadd_u16(sum, sum);
  88. dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 4));
  89. dc_store_8x8(dst, stride, dc);
  90. }
  91. void vpx_dc_left_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  92. const uint8_t *above, const uint8_t *left) {
  93. const uint16x4_t sum = dc_sum_8(left);
  94. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 3));
  95. (void)above;
  96. dc_store_8x8(dst, stride, dc);
  97. }
  98. void vpx_dc_top_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  99. const uint8_t *above, const uint8_t *left) {
  100. const uint16x4_t sum = dc_sum_8(above);
  101. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 3));
  102. (void)left;
  103. dc_store_8x8(dst, stride, dc);
  104. }
  105. void vpx_dc_128_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  106. const uint8_t *above, const uint8_t *left) {
  107. const uint8x8_t dc = vdup_n_u8(0x80);
  108. (void)above;
  109. (void)left;
  110. dc_store_8x8(dst, stride, dc);
  111. }
  112. //------------------------------------------------------------------------------
  113. // DC 16x16
  114. static INLINE uint16x4_t dc_sum_16(const uint8_t *ref) {
  115. const uint8x16_t ref_u8 = vld1q_u8(ref);
  116. const uint16x8_t p0 = vpaddlq_u8(ref_u8);
  117. uint16x4_t sum = vadd_u16(vget_low_u16(p0), vget_high_u16(p0));
  118. sum = vpadd_u16(sum, sum);
  119. return vpadd_u16(sum, sum);
  120. }
  121. static INLINE void dc_store_16x16(uint8_t *dst, ptrdiff_t stride,
  122. const uint8x8_t dc) {
  123. const uint8x16_t dc_dup = vdupq_lane_u8(dc, 0);
  124. int i;
  125. for (i = 0; i < 16; ++i, dst += stride) {
  126. vst1q_u8(dst, dc_dup);
  127. }
  128. }
  129. void vpx_dc_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  130. const uint8_t *above, const uint8_t *left) {
  131. const uint8x16_t ref0 = vld1q_u8(above);
  132. const uint8x16_t ref1 = vld1q_u8(left);
  133. const uint16x8_t p0 = vpaddlq_u8(ref0);
  134. const uint16x8_t p1 = vpaddlq_u8(ref1);
  135. const uint16x8_t p2 = vaddq_u16(p0, p1);
  136. uint16x4_t sum = vadd_u16(vget_low_u16(p2), vget_high_u16(p2));
  137. uint8x8_t dc;
  138. sum = vpadd_u16(sum, sum);
  139. sum = vpadd_u16(sum, sum);
  140. dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 5));
  141. dc_store_16x16(dst, stride, dc);
  142. }
  143. void vpx_dc_left_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  144. const uint8_t *above,
  145. const uint8_t *left) {
  146. const uint16x4_t sum = dc_sum_16(left);
  147. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 4));
  148. (void)above;
  149. dc_store_16x16(dst, stride, dc);
  150. }
  151. void vpx_dc_top_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  152. const uint8_t *above,
  153. const uint8_t *left) {
  154. const uint16x4_t sum = dc_sum_16(above);
  155. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 4));
  156. (void)left;
  157. dc_store_16x16(dst, stride, dc);
  158. }
  159. void vpx_dc_128_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  160. const uint8_t *above,
  161. const uint8_t *left) {
  162. const uint8x8_t dc = vdup_n_u8(0x80);
  163. (void)above;
  164. (void)left;
  165. dc_store_16x16(dst, stride, dc);
  166. }
  167. //------------------------------------------------------------------------------
  168. // DC 32x32
  169. static INLINE uint16x4_t dc_sum_32(const uint8_t *ref) {
  170. const uint8x16x2_t r = vld2q_u8(ref);
  171. const uint16x8_t p0 = vpaddlq_u8(r.val[0]);
  172. const uint16x8_t p1 = vpaddlq_u8(r.val[1]);
  173. const uint16x8_t p2 = vaddq_u16(p0, p1);
  174. uint16x4_t sum = vadd_u16(vget_low_u16(p2), vget_high_u16(p2));
  175. sum = vpadd_u16(sum, sum);
  176. return vpadd_u16(sum, sum);
  177. }
  178. static INLINE void dc_store_32x32(uint8_t *dst, ptrdiff_t stride,
  179. const uint8x8_t dc) {
  180. uint8x16x2_t dc_dup;
  181. int i;
  182. dc_dup.val[0] = dc_dup.val[1] = vdupq_lane_u8(dc, 0);
  183. for (i = 0; i < 32; ++i, dst += stride) {
  184. vst2q_u8(dst, dc_dup);
  185. }
  186. }
  187. void vpx_dc_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  188. const uint8_t *above, const uint8_t *left) {
  189. const uint8x16x2_t a = vld2q_u8(above);
  190. const uint8x16x2_t l = vld2q_u8(left);
  191. const uint16x8_t pa0 = vpaddlq_u8(a.val[0]);
  192. const uint16x8_t pl0 = vpaddlq_u8(l.val[0]);
  193. const uint16x8_t pa1 = vpaddlq_u8(a.val[1]);
  194. const uint16x8_t pl1 = vpaddlq_u8(l.val[1]);
  195. const uint16x8_t pa = vaddq_u16(pa0, pa1);
  196. const uint16x8_t pl = vaddq_u16(pl0, pl1);
  197. const uint16x8_t pal = vaddq_u16(pa, pl);
  198. uint16x4_t sum = vadd_u16(vget_low_u16(pal), vget_high_u16(pal));
  199. uint8x8_t dc;
  200. sum = vpadd_u16(sum, sum);
  201. sum = vpadd_u16(sum, sum);
  202. dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 6));
  203. dc_store_32x32(dst, stride, dc);
  204. }
  205. void vpx_dc_left_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  206. const uint8_t *above,
  207. const uint8_t *left) {
  208. const uint16x4_t sum = dc_sum_32(left);
  209. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 5));
  210. (void)above;
  211. dc_store_32x32(dst, stride, dc);
  212. }
  213. void vpx_dc_top_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  214. const uint8_t *above,
  215. const uint8_t *left) {
  216. const uint16x4_t sum = dc_sum_32(above);
  217. const uint8x8_t dc = vreinterpret_u8_u16(vrshr_n_u16(sum, 5));
  218. (void)left;
  219. dc_store_32x32(dst, stride, dc);
  220. }
  221. void vpx_dc_128_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  222. const uint8_t *above,
  223. const uint8_t *left) {
  224. const uint8x8_t dc = vdup_n_u8(0x80);
  225. (void)above;
  226. (void)left;
  227. dc_store_32x32(dst, stride, dc);
  228. }
  229. // -----------------------------------------------------------------------------
  230. void vpx_d45_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  231. const uint8_t *above, const uint8_t *left) {
  232. const uint8x8_t ABCDEFGH = vld1_u8(above);
  233. const uint64x1_t A1 = vshr_n_u64(vreinterpret_u64_u8(ABCDEFGH), 8);
  234. const uint64x1_t A2 = vshr_n_u64(vreinterpret_u64_u8(ABCDEFGH), 16);
  235. const uint8x8_t BCDEFGH0 = vreinterpret_u8_u64(A1);
  236. const uint8x8_t CDEFGH00 = vreinterpret_u8_u64(A2);
  237. const uint8x8_t avg1 = vhadd_u8(ABCDEFGH, CDEFGH00);
  238. const uint8x8_t avg2 = vrhadd_u8(avg1, BCDEFGH0);
  239. const uint64x1_t avg2_u64 = vreinterpret_u64_u8(avg2);
  240. const uint32x2_t r0 = vreinterpret_u32_u8(avg2);
  241. const uint32x2_t r1 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 8));
  242. const uint32x2_t r2 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 16));
  243. const uint32x2_t r3 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 24));
  244. (void)left;
  245. vst1_lane_u32((uint32_t *)(dst + 0 * stride), r0, 0);
  246. vst1_lane_u32((uint32_t *)(dst + 1 * stride), r1, 0);
  247. vst1_lane_u32((uint32_t *)(dst + 2 * stride), r2, 0);
  248. vst1_lane_u32((uint32_t *)(dst + 3 * stride), r3, 0);
  249. vst1_lane_u8(dst + 3 * stride + 3, ABCDEFGH, 7);
  250. }
  251. static INLINE void d45_store_8(uint8_t **dst, const ptrdiff_t stride,
  252. const uint8x8_t above_right, uint8x8_t *row) {
  253. *row = vext_u8(*row, above_right, 1);
  254. vst1_u8(*dst, *row);
  255. *dst += stride;
  256. }
  257. void vpx_d45_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  258. const uint8_t *above, const uint8_t *left) {
  259. const uint8x8_t A0 = vld1_u8(above);
  260. const uint8x8_t above_right = vdup_lane_u8(A0, 7);
  261. const uint8x8_t A1 = vext_u8(A0, above_right, 1);
  262. const uint8x8_t A2 = vext_u8(A0, above_right, 2);
  263. const uint8x8_t avg1 = vhadd_u8(A0, A2);
  264. uint8x8_t row = vrhadd_u8(avg1, A1);
  265. (void)left;
  266. vst1_u8(dst, row);
  267. dst += stride;
  268. d45_store_8(&dst, stride, above_right, &row);
  269. d45_store_8(&dst, stride, above_right, &row);
  270. d45_store_8(&dst, stride, above_right, &row);
  271. d45_store_8(&dst, stride, above_right, &row);
  272. d45_store_8(&dst, stride, above_right, &row);
  273. d45_store_8(&dst, stride, above_right, &row);
  274. vst1_u8(dst, above_right);
  275. }
  276. static INLINE void d45_store_16(uint8_t **dst, const ptrdiff_t stride,
  277. const uint8x16_t above_right, uint8x16_t *row) {
  278. *row = vextq_u8(*row, above_right, 1);
  279. vst1q_u8(*dst, *row);
  280. *dst += stride;
  281. }
  282. void vpx_d45_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  283. const uint8_t *above, const uint8_t *left) {
  284. const uint8x16_t A0 = vld1q_u8(above);
  285. const uint8x16_t above_right = vdupq_lane_u8(vget_high_u8(A0), 7);
  286. const uint8x16_t A1 = vextq_u8(A0, above_right, 1);
  287. const uint8x16_t A2 = vextq_u8(A0, above_right, 2);
  288. const uint8x16_t avg1 = vhaddq_u8(A0, A2);
  289. uint8x16_t row = vrhaddq_u8(avg1, A1);
  290. (void)left;
  291. vst1q_u8(dst, row);
  292. dst += stride;
  293. d45_store_16(&dst, stride, above_right, &row);
  294. d45_store_16(&dst, stride, above_right, &row);
  295. d45_store_16(&dst, stride, above_right, &row);
  296. d45_store_16(&dst, stride, above_right, &row);
  297. d45_store_16(&dst, stride, above_right, &row);
  298. d45_store_16(&dst, stride, above_right, &row);
  299. d45_store_16(&dst, stride, above_right, &row);
  300. d45_store_16(&dst, stride, above_right, &row);
  301. d45_store_16(&dst, stride, above_right, &row);
  302. d45_store_16(&dst, stride, above_right, &row);
  303. d45_store_16(&dst, stride, above_right, &row);
  304. d45_store_16(&dst, stride, above_right, &row);
  305. d45_store_16(&dst, stride, above_right, &row);
  306. d45_store_16(&dst, stride, above_right, &row);
  307. vst1q_u8(dst, above_right);
  308. }
  309. void vpx_d45_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  310. const uint8_t *above, const uint8_t *left) {
  311. const uint8x16_t A0_0 = vld1q_u8(above);
  312. const uint8x16_t A0_1 = vld1q_u8(above + 16);
  313. const uint8x16_t above_right = vdupq_lane_u8(vget_high_u8(A0_1), 7);
  314. const uint8x16_t A1_0 = vld1q_u8(above + 1);
  315. const uint8x16_t A1_1 = vld1q_u8(above + 17);
  316. const uint8x16_t A2_0 = vld1q_u8(above + 2);
  317. const uint8x16_t A2_1 = vld1q_u8(above + 18);
  318. const uint8x16_t avg_0 = vhaddq_u8(A0_0, A2_0);
  319. const uint8x16_t avg_1 = vhaddq_u8(A0_1, A2_1);
  320. uint8x16_t row_0 = vrhaddq_u8(avg_0, A1_0);
  321. uint8x16_t row_1 = vrhaddq_u8(avg_1, A1_1);
  322. int i;
  323. (void)left;
  324. vst1q_u8(dst, row_0);
  325. dst += 16;
  326. vst1q_u8(dst, row_1);
  327. dst += stride - 16;
  328. for (i = 0; i < 30; ++i) {
  329. row_0 = vextq_u8(row_0, row_1, 1);
  330. row_1 = vextq_u8(row_1, above_right, 1);
  331. vst1q_u8(dst, row_0);
  332. dst += 16;
  333. vst1q_u8(dst, row_1);
  334. dst += stride - 16;
  335. }
  336. vst1q_u8(dst, above_right);
  337. dst += 16;
  338. vst1q_u8(dst, row_1);
  339. }
  340. // -----------------------------------------------------------------------------
  341. void vpx_d135_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  342. const uint8_t *above, const uint8_t *left) {
  343. const uint8x8_t XA0123 = vld1_u8(above - 1);
  344. const uint8x8_t L0123 = vld1_u8(left);
  345. const uint8x8_t L3210 = vrev64_u8(L0123);
  346. const uint8x8_t L3210XA012 = vext_u8(L3210, XA0123, 4);
  347. const uint8x8_t L210XA0123 = vext_u8(L3210, XA0123, 5);
  348. const uint8x8_t L10XA0123_ =
  349. vreinterpret_u8_u64(vshr_n_u64(vreinterpret_u64_u8(L210XA0123), 8));
  350. const uint8x8_t avg1 = vhadd_u8(L10XA0123_, L3210XA012);
  351. const uint8x8_t avg2 = vrhadd_u8(avg1, L210XA0123);
  352. const uint64x1_t avg2_u64 = vreinterpret_u64_u8(avg2);
  353. const uint32x2_t r3 = vreinterpret_u32_u8(avg2);
  354. const uint32x2_t r2 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 8));
  355. const uint32x2_t r1 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 16));
  356. const uint32x2_t r0 = vreinterpret_u32_u64(vshr_n_u64(avg2_u64, 24));
  357. vst1_lane_u32((uint32_t *)dst, r0, 0);
  358. dst += stride;
  359. vst1_lane_u32((uint32_t *)dst, r1, 0);
  360. dst += stride;
  361. vst1_lane_u32((uint32_t *)dst, r2, 0);
  362. dst += stride;
  363. vst1_lane_u32((uint32_t *)dst, r3, 0);
  364. }
  365. void vpx_d135_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  366. const uint8_t *above, const uint8_t *left) {
  367. const uint8x8_t XA0123456 = vld1_u8(above - 1);
  368. const uint8x8_t A01234567 = vld1_u8(above);
  369. const uint8x8_t A1234567_ = vld1_u8(above + 1);
  370. const uint8x8_t L01234567 = vld1_u8(left);
  371. const uint8x8_t L76543210 = vrev64_u8(L01234567);
  372. const uint8x8_t L6543210X = vext_u8(L76543210, XA0123456, 1);
  373. const uint8x8_t L543210XA0 = vext_u8(L76543210, XA0123456, 2);
  374. const uint8x16_t L76543210XA0123456 = vcombine_u8(L76543210, XA0123456);
  375. const uint8x16_t L6543210XA01234567 = vcombine_u8(L6543210X, A01234567);
  376. const uint8x16_t L543210XA01234567_ = vcombine_u8(L543210XA0, A1234567_);
  377. const uint8x16_t avg = vhaddq_u8(L76543210XA0123456, L543210XA01234567_);
  378. const uint8x16_t row = vrhaddq_u8(avg, L6543210XA01234567);
  379. const uint8x8_t row_0 = vget_low_u8(row);
  380. const uint8x8_t row_1 = vget_high_u8(row);
  381. const uint8x8_t r0 = vext_u8(row_0, row_1, 7);
  382. const uint8x8_t r1 = vext_u8(row_0, row_1, 6);
  383. const uint8x8_t r2 = vext_u8(row_0, row_1, 5);
  384. const uint8x8_t r3 = vext_u8(row_0, row_1, 4);
  385. const uint8x8_t r4 = vext_u8(row_0, row_1, 3);
  386. const uint8x8_t r5 = vext_u8(row_0, row_1, 2);
  387. const uint8x8_t r6 = vext_u8(row_0, row_1, 1);
  388. vst1_u8(dst, r0);
  389. dst += stride;
  390. vst1_u8(dst, r1);
  391. dst += stride;
  392. vst1_u8(dst, r2);
  393. dst += stride;
  394. vst1_u8(dst, r3);
  395. dst += stride;
  396. vst1_u8(dst, r4);
  397. dst += stride;
  398. vst1_u8(dst, r5);
  399. dst += stride;
  400. vst1_u8(dst, r6);
  401. dst += stride;
  402. vst1_u8(dst, row_0);
  403. }
  404. static INLINE void d135_store_16x8(
  405. uint8_t **dst, const ptrdiff_t stride, const uint8x16_t row_0,
  406. const uint8x16_t row_1, const uint8x16_t row_2, const uint8x16_t row_3,
  407. const uint8x16_t row_4, const uint8x16_t row_5, const uint8x16_t row_6,
  408. const uint8x16_t row_7) {
  409. vst1q_u8(*dst, row_0);
  410. *dst += stride;
  411. vst1q_u8(*dst, row_1);
  412. *dst += stride;
  413. vst1q_u8(*dst, row_2);
  414. *dst += stride;
  415. vst1q_u8(*dst, row_3);
  416. *dst += stride;
  417. vst1q_u8(*dst, row_4);
  418. *dst += stride;
  419. vst1q_u8(*dst, row_5);
  420. *dst += stride;
  421. vst1q_u8(*dst, row_6);
  422. *dst += stride;
  423. vst1q_u8(*dst, row_7);
  424. *dst += stride;
  425. }
  426. void vpx_d135_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  427. const uint8_t *above, const uint8_t *left) {
  428. const uint8x16_t XA0123456789abcde = vld1q_u8(above - 1);
  429. const uint8x16_t A0123456789abcdef = vld1q_u8(above);
  430. const uint8x16_t A123456789abcdef_ = vld1q_u8(above + 1);
  431. const uint8x16_t L0123456789abcdef = vld1q_u8(left);
  432. const uint8x8_t L76543210 = vrev64_u8(vget_low_u8(L0123456789abcdef));
  433. const uint8x8_t Lfedcba98 = vrev64_u8(vget_high_u8(L0123456789abcdef));
  434. const uint8x16_t Lfedcba9876543210 = vcombine_u8(Lfedcba98, L76543210);
  435. const uint8x16_t Ledcba9876543210X =
  436. vextq_u8(Lfedcba9876543210, XA0123456789abcde, 1);
  437. const uint8x16_t Ldcba9876543210XA0 =
  438. vextq_u8(Lfedcba9876543210, XA0123456789abcde, 2);
  439. const uint8x16_t avg_0 = vhaddq_u8(Lfedcba9876543210, Ldcba9876543210XA0);
  440. const uint8x16_t avg_1 = vhaddq_u8(XA0123456789abcde, A123456789abcdef_);
  441. const uint8x16_t row_0 = vrhaddq_u8(avg_0, Ledcba9876543210X);
  442. const uint8x16_t row_1 = vrhaddq_u8(avg_1, A0123456789abcdef);
  443. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 15);
  444. const uint8x16_t r_1 = vextq_u8(row_0, row_1, 14);
  445. const uint8x16_t r_2 = vextq_u8(row_0, row_1, 13);
  446. const uint8x16_t r_3 = vextq_u8(row_0, row_1, 12);
  447. const uint8x16_t r_4 = vextq_u8(row_0, row_1, 11);
  448. const uint8x16_t r_5 = vextq_u8(row_0, row_1, 10);
  449. const uint8x16_t r_6 = vextq_u8(row_0, row_1, 9);
  450. const uint8x16_t r_7 = vcombine_u8(vget_high_u8(row_0), vget_low_u8(row_1));
  451. const uint8x16_t r_8 = vextq_u8(row_0, row_1, 7);
  452. const uint8x16_t r_9 = vextq_u8(row_0, row_1, 6);
  453. const uint8x16_t r_a = vextq_u8(row_0, row_1, 5);
  454. const uint8x16_t r_b = vextq_u8(row_0, row_1, 4);
  455. const uint8x16_t r_c = vextq_u8(row_0, row_1, 3);
  456. const uint8x16_t r_d = vextq_u8(row_0, row_1, 2);
  457. const uint8x16_t r_e = vextq_u8(row_0, row_1, 1);
  458. d135_store_16x8(&dst, stride, r_0, r_1, r_2, r_3, r_4, r_5, r_6, r_7);
  459. d135_store_16x8(&dst, stride, r_8, r_9, r_a, r_b, r_c, r_d, r_e, row_0);
  460. }
  461. static INLINE void d135_store_32x2(uint8_t **dst, const ptrdiff_t stride,
  462. const uint8x16_t row_0,
  463. const uint8x16_t row_1,
  464. const uint8x16_t row_2) {
  465. uint8_t *dst2 = *dst;
  466. vst1q_u8(dst2, row_1);
  467. dst2 += 16;
  468. vst1q_u8(dst2, row_2);
  469. dst2 += 16 * stride - 16;
  470. vst1q_u8(dst2, row_0);
  471. dst2 += 16;
  472. vst1q_u8(dst2, row_1);
  473. *dst += stride;
  474. }
  475. void vpx_d135_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  476. const uint8_t *above, const uint8_t *left) {
  477. const uint8x16_t LL0123456789abcdef = vld1q_u8(left + 16);
  478. const uint8x16_t LU0123456789abcdef = vld1q_u8(left);
  479. const uint8x8_t LL76543210 = vrev64_u8(vget_low_u8(LL0123456789abcdef));
  480. const uint8x8_t LU76543210 = vrev64_u8(vget_low_u8(LU0123456789abcdef));
  481. const uint8x8_t LLfedcba98 = vrev64_u8(vget_high_u8(LL0123456789abcdef));
  482. const uint8x8_t LUfedcba98 = vrev64_u8(vget_high_u8(LU0123456789abcdef));
  483. const uint8x16_t LLfedcba9876543210 = vcombine_u8(LLfedcba98, LL76543210);
  484. const uint8x16_t LUfedcba9876543210 = vcombine_u8(LUfedcba98, LU76543210);
  485. const uint8x16_t LLedcba9876543210Uf =
  486. vextq_u8(LLfedcba9876543210, LUfedcba9876543210, 1);
  487. const uint8x16_t LLdcba9876543210Ufe =
  488. vextq_u8(LLfedcba9876543210, LUfedcba9876543210, 2);
  489. const uint8x16_t avg_0 = vhaddq_u8(LLfedcba9876543210, LLdcba9876543210Ufe);
  490. const uint8x16_t row_0 = vrhaddq_u8(avg_0, LLedcba9876543210Uf);
  491. const uint8x16_t XAL0123456789abcde = vld1q_u8(above - 1);
  492. const uint8x16_t LUedcba9876543210X =
  493. vextq_u8(LUfedcba9876543210, XAL0123456789abcde, 1);
  494. const uint8x16_t LUdcba9876543210XA0 =
  495. vextq_u8(LUfedcba9876543210, XAL0123456789abcde, 2);
  496. const uint8x16_t avg_1 = vhaddq_u8(LUfedcba9876543210, LUdcba9876543210XA0);
  497. const uint8x16_t row_1 = vrhaddq_u8(avg_1, LUedcba9876543210X);
  498. const uint8x16_t AL0123456789abcdef = vld1q_u8(above);
  499. const uint8x16_t AL123456789abcdefg = vld1q_u8(above + 1);
  500. const uint8x16_t ALfR0123456789abcde = vld1q_u8(above + 15);
  501. const uint8x16_t AR0123456789abcdef = vld1q_u8(above + 16);
  502. const uint8x16_t AR123456789abcdef_ = vld1q_u8(above + 17);
  503. const uint8x16_t avg_2 = vhaddq_u8(XAL0123456789abcde, AL123456789abcdefg);
  504. const uint8x16_t row_2 = vrhaddq_u8(avg_2, AL0123456789abcdef);
  505. const uint8x16_t avg_3 = vhaddq_u8(ALfR0123456789abcde, AR123456789abcdef_);
  506. const uint8x16_t row_3 = vrhaddq_u8(avg_3, AR0123456789abcdef);
  507. {
  508. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 15);
  509. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 15);
  510. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 15);
  511. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  512. }
  513. {
  514. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 14);
  515. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 14);
  516. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 14);
  517. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  518. }
  519. {
  520. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 13);
  521. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 13);
  522. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 13);
  523. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  524. }
  525. {
  526. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 12);
  527. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 12);
  528. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 12);
  529. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  530. }
  531. {
  532. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 11);
  533. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 11);
  534. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 11);
  535. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  536. }
  537. {
  538. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 10);
  539. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 10);
  540. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 10);
  541. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  542. }
  543. {
  544. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 9);
  545. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 9);
  546. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 9);
  547. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  548. }
  549. {
  550. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 8);
  551. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 8);
  552. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 8);
  553. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  554. }
  555. {
  556. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 7);
  557. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 7);
  558. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 7);
  559. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  560. }
  561. {
  562. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 6);
  563. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 6);
  564. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 6);
  565. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  566. }
  567. {
  568. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 5);
  569. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 5);
  570. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 5);
  571. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  572. }
  573. {
  574. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 4);
  575. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 4);
  576. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 4);
  577. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  578. }
  579. {
  580. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 3);
  581. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 3);
  582. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 3);
  583. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  584. }
  585. {
  586. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 2);
  587. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 2);
  588. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 2);
  589. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  590. }
  591. {
  592. const uint8x16_t r_0 = vextq_u8(row_0, row_1, 1);
  593. const uint8x16_t r_1 = vextq_u8(row_1, row_2, 1);
  594. const uint8x16_t r_2 = vextq_u8(row_2, row_3, 1);
  595. d135_store_32x2(&dst, stride, r_0, r_1, r_2);
  596. }
  597. d135_store_32x2(&dst, stride, row_0, row_1, row_2);
  598. }
  599. // -----------------------------------------------------------------------------
  600. #if !HAVE_NEON_ASM
  601. void vpx_v_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  602. const uint8_t *above, const uint8_t *left) {
  603. const uint32_t d = *(const uint32_t *)above;
  604. int i;
  605. (void)left;
  606. for (i = 0; i < 4; i++, dst += stride) {
  607. *(uint32_t *)dst = d;
  608. }
  609. }
  610. void vpx_v_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  611. const uint8_t *above, const uint8_t *left) {
  612. const uint8x8_t d = vld1_u8(above);
  613. int i;
  614. (void)left;
  615. for (i = 0; i < 8; i++, dst += stride) {
  616. vst1_u8(dst, d);
  617. }
  618. }
  619. void vpx_v_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  620. const uint8_t *above, const uint8_t *left) {
  621. const uint8x16_t d = vld1q_u8(above);
  622. int i;
  623. (void)left;
  624. for (i = 0; i < 16; i++, dst += stride) {
  625. vst1q_u8(dst, d);
  626. }
  627. }
  628. void vpx_v_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  629. const uint8_t *above, const uint8_t *left) {
  630. const uint8x16_t d0 = vld1q_u8(above);
  631. const uint8x16_t d1 = vld1q_u8(above + 16);
  632. int i;
  633. (void)left;
  634. for (i = 0; i < 32; i++) {
  635. // Note: performance was worse using vst2q_u8 under gcc-4.9 & clang-3.8.
  636. // clang-3.8 unrolled the loop fully with no filler so the cause is likely
  637. // the latency of the instruction.
  638. vst1q_u8(dst, d0);
  639. dst += 16;
  640. vst1q_u8(dst, d1);
  641. dst += stride - 16;
  642. }
  643. }
  644. // -----------------------------------------------------------------------------
  645. void vpx_h_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  646. const uint8_t *above, const uint8_t *left) {
  647. const uint32x2_t zero = vdup_n_u32(0);
  648. const uint8x8_t left_u8 =
  649. vreinterpret_u8_u32(vld1_lane_u32((const uint32_t *)left, zero, 0));
  650. uint8x8_t d;
  651. (void)above;
  652. d = vdup_lane_u8(left_u8, 0);
  653. vst1_lane_u32((uint32_t *)dst, vreinterpret_u32_u8(d), 0);
  654. dst += stride;
  655. d = vdup_lane_u8(left_u8, 1);
  656. vst1_lane_u32((uint32_t *)dst, vreinterpret_u32_u8(d), 0);
  657. dst += stride;
  658. d = vdup_lane_u8(left_u8, 2);
  659. vst1_lane_u32((uint32_t *)dst, vreinterpret_u32_u8(d), 0);
  660. dst += stride;
  661. d = vdup_lane_u8(left_u8, 3);
  662. vst1_lane_u32((uint32_t *)dst, vreinterpret_u32_u8(d), 0);
  663. }
  664. void vpx_h_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  665. const uint8_t *above, const uint8_t *left) {
  666. const uint8x8_t left_u8 = vld1_u8(left);
  667. uint8x8_t d;
  668. (void)above;
  669. d = vdup_lane_u8(left_u8, 0);
  670. vst1_u8(dst, d);
  671. dst += stride;
  672. d = vdup_lane_u8(left_u8, 1);
  673. vst1_u8(dst, d);
  674. dst += stride;
  675. d = vdup_lane_u8(left_u8, 2);
  676. vst1_u8(dst, d);
  677. dst += stride;
  678. d = vdup_lane_u8(left_u8, 3);
  679. vst1_u8(dst, d);
  680. dst += stride;
  681. d = vdup_lane_u8(left_u8, 4);
  682. vst1_u8(dst, d);
  683. dst += stride;
  684. d = vdup_lane_u8(left_u8, 5);
  685. vst1_u8(dst, d);
  686. dst += stride;
  687. d = vdup_lane_u8(left_u8, 6);
  688. vst1_u8(dst, d);
  689. dst += stride;
  690. d = vdup_lane_u8(left_u8, 7);
  691. vst1_u8(dst, d);
  692. }
  693. static INLINE void h_store_16x8(uint8_t **dst, const ptrdiff_t stride,
  694. const uint8x8_t left) {
  695. const uint8x16_t row_0 = vdupq_lane_u8(left, 0);
  696. const uint8x16_t row_1 = vdupq_lane_u8(left, 1);
  697. const uint8x16_t row_2 = vdupq_lane_u8(left, 2);
  698. const uint8x16_t row_3 = vdupq_lane_u8(left, 3);
  699. const uint8x16_t row_4 = vdupq_lane_u8(left, 4);
  700. const uint8x16_t row_5 = vdupq_lane_u8(left, 5);
  701. const uint8x16_t row_6 = vdupq_lane_u8(left, 6);
  702. const uint8x16_t row_7 = vdupq_lane_u8(left, 7);
  703. vst1q_u8(*dst, row_0);
  704. *dst += stride;
  705. vst1q_u8(*dst, row_1);
  706. *dst += stride;
  707. vst1q_u8(*dst, row_2);
  708. *dst += stride;
  709. vst1q_u8(*dst, row_3);
  710. *dst += stride;
  711. vst1q_u8(*dst, row_4);
  712. *dst += stride;
  713. vst1q_u8(*dst, row_5);
  714. *dst += stride;
  715. vst1q_u8(*dst, row_6);
  716. *dst += stride;
  717. vst1q_u8(*dst, row_7);
  718. *dst += stride;
  719. }
  720. void vpx_h_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  721. const uint8_t *above, const uint8_t *left) {
  722. const uint8x16_t left_u8q = vld1q_u8(left);
  723. (void)above;
  724. h_store_16x8(&dst, stride, vget_low_u8(left_u8q));
  725. h_store_16x8(&dst, stride, vget_high_u8(left_u8q));
  726. }
  727. static INLINE void h_store_32x8(uint8_t **dst, const ptrdiff_t stride,
  728. const uint8x8_t left) {
  729. const uint8x16_t row_0 = vdupq_lane_u8(left, 0);
  730. const uint8x16_t row_1 = vdupq_lane_u8(left, 1);
  731. const uint8x16_t row_2 = vdupq_lane_u8(left, 2);
  732. const uint8x16_t row_3 = vdupq_lane_u8(left, 3);
  733. const uint8x16_t row_4 = vdupq_lane_u8(left, 4);
  734. const uint8x16_t row_5 = vdupq_lane_u8(left, 5);
  735. const uint8x16_t row_6 = vdupq_lane_u8(left, 6);
  736. const uint8x16_t row_7 = vdupq_lane_u8(left, 7);
  737. vst1q_u8(*dst, row_0); // Note clang-3.8 produced poor code w/vst2q_u8
  738. *dst += 16;
  739. vst1q_u8(*dst, row_0);
  740. *dst += stride - 16;
  741. vst1q_u8(*dst, row_1);
  742. *dst += 16;
  743. vst1q_u8(*dst, row_1);
  744. *dst += stride - 16;
  745. vst1q_u8(*dst, row_2);
  746. *dst += 16;
  747. vst1q_u8(*dst, row_2);
  748. *dst += stride - 16;
  749. vst1q_u8(*dst, row_3);
  750. *dst += 16;
  751. vst1q_u8(*dst, row_3);
  752. *dst += stride - 16;
  753. vst1q_u8(*dst, row_4);
  754. *dst += 16;
  755. vst1q_u8(*dst, row_4);
  756. *dst += stride - 16;
  757. vst1q_u8(*dst, row_5);
  758. *dst += 16;
  759. vst1q_u8(*dst, row_5);
  760. *dst += stride - 16;
  761. vst1q_u8(*dst, row_6);
  762. *dst += 16;
  763. vst1q_u8(*dst, row_6);
  764. *dst += stride - 16;
  765. vst1q_u8(*dst, row_7);
  766. *dst += 16;
  767. vst1q_u8(*dst, row_7);
  768. *dst += stride - 16;
  769. }
  770. void vpx_h_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  771. const uint8_t *above, const uint8_t *left) {
  772. int i;
  773. (void)above;
  774. for (i = 0; i < 2; i++, left += 16) {
  775. const uint8x16_t left_u8 = vld1q_u8(left);
  776. h_store_32x8(&dst, stride, vget_low_u8(left_u8));
  777. h_store_32x8(&dst, stride, vget_high_u8(left_u8));
  778. }
  779. }
  780. // -----------------------------------------------------------------------------
  781. static INLINE int16x8_t convert_u8_to_s16(uint8x8_t v) {
  782. return vreinterpretq_s16_u16(vmovl_u8(v));
  783. }
  784. void vpx_tm_predictor_4x4_neon(uint8_t *dst, ptrdiff_t stride,
  785. const uint8_t *above, const uint8_t *left) {
  786. const uint8x8_t top_left = vld1_dup_u8(above - 1);
  787. const uint8x8_t left_u8 = vld1_u8(left);
  788. const uint8x8_t above_u8 = vld1_u8(above);
  789. const int16x4_t left_s16 = vget_low_s16(convert_u8_to_s16(left_u8));
  790. int16x8_t sub, sum;
  791. uint32x2_t d;
  792. sub = vreinterpretq_s16_u16(vsubl_u8(above_u8, top_left));
  793. // Avoid vcombine_s16() which generates lots of redundant code with clang-3.8.
  794. sub = vreinterpretq_s16_s64(
  795. vdupq_lane_s64(vreinterpret_s64_s16(vget_low_s16(sub)), 0));
  796. sum = vcombine_s16(vdup_lane_s16(left_s16, 0), vdup_lane_s16(left_s16, 1));
  797. sum = vaddq_s16(sum, sub);
  798. d = vreinterpret_u32_u8(vqmovun_s16(sum));
  799. vst1_lane_u32((uint32_t *)dst, d, 0);
  800. dst += stride;
  801. vst1_lane_u32((uint32_t *)dst, d, 1);
  802. dst += stride;
  803. sum = vcombine_s16(vdup_lane_s16(left_s16, 2), vdup_lane_s16(left_s16, 3));
  804. sum = vaddq_s16(sum, sub);
  805. d = vreinterpret_u32_u8(vqmovun_s16(sum));
  806. vst1_lane_u32((uint32_t *)dst, d, 0);
  807. dst += stride;
  808. vst1_lane_u32((uint32_t *)dst, d, 1);
  809. }
  810. static INLINE void tm_8_kernel(uint8_t **dst, const ptrdiff_t stride,
  811. const int16x8_t left_dup, const int16x8_t sub) {
  812. const int16x8_t sum = vaddq_s16(left_dup, sub);
  813. const uint8x8_t d = vqmovun_s16(sum);
  814. vst1_u8(*dst, d);
  815. *dst += stride;
  816. }
  817. void vpx_tm_predictor_8x8_neon(uint8_t *dst, ptrdiff_t stride,
  818. const uint8_t *above, const uint8_t *left) {
  819. const uint8x8_t top_left = vld1_dup_u8(above - 1);
  820. const uint8x8_t above_u8 = vld1_u8(above);
  821. const uint8x8_t left_u8 = vld1_u8(left);
  822. const int16x8_t left_s16q = convert_u8_to_s16(left_u8);
  823. const int16x8_t sub = vreinterpretq_s16_u16(vsubl_u8(above_u8, top_left));
  824. int16x4_t left_s16d = vget_low_s16(left_s16q);
  825. int i;
  826. for (i = 0; i < 2; i++, left_s16d = vget_high_s16(left_s16q)) {
  827. int16x8_t left_dup;
  828. left_dup = vdupq_lane_s16(left_s16d, 0);
  829. tm_8_kernel(&dst, stride, left_dup, sub);
  830. left_dup = vdupq_lane_s16(left_s16d, 1);
  831. tm_8_kernel(&dst, stride, left_dup, sub);
  832. left_dup = vdupq_lane_s16(left_s16d, 2);
  833. tm_8_kernel(&dst, stride, left_dup, sub);
  834. left_dup = vdupq_lane_s16(left_s16d, 3);
  835. tm_8_kernel(&dst, stride, left_dup, sub);
  836. }
  837. }
  838. static INLINE void tm_16_kernel(uint8_t **dst, const ptrdiff_t stride,
  839. const int16x8_t left_dup, const int16x8_t sub0,
  840. const int16x8_t sub1) {
  841. const int16x8_t sum0 = vaddq_s16(left_dup, sub0);
  842. const int16x8_t sum1 = vaddq_s16(left_dup, sub1);
  843. const uint8x8_t d0 = vqmovun_s16(sum0);
  844. const uint8x8_t d1 = vqmovun_s16(sum1);
  845. vst1_u8(*dst, d0);
  846. *dst += 8;
  847. vst1_u8(*dst, d1);
  848. *dst += stride - 8;
  849. }
  850. void vpx_tm_predictor_16x16_neon(uint8_t *dst, ptrdiff_t stride,
  851. const uint8_t *above, const uint8_t *left) {
  852. const uint8x16_t top_left = vld1q_dup_u8(above - 1);
  853. const uint8x16_t above_u8 = vld1q_u8(above);
  854. const int16x8_t sub0 = vreinterpretq_s16_u16(
  855. vsubl_u8(vget_low_u8(above_u8), vget_low_u8(top_left)));
  856. const int16x8_t sub1 = vreinterpretq_s16_u16(
  857. vsubl_u8(vget_high_u8(above_u8), vget_high_u8(top_left)));
  858. int16x8_t left_dup;
  859. int i;
  860. for (i = 0; i < 2; i++, left += 8) {
  861. const uint8x8_t left_u8 = vld1_u8(left);
  862. const int16x8_t left_s16q = convert_u8_to_s16(left_u8);
  863. const int16x4_t left_low = vget_low_s16(left_s16q);
  864. const int16x4_t left_high = vget_high_s16(left_s16q);
  865. left_dup = vdupq_lane_s16(left_low, 0);
  866. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  867. left_dup = vdupq_lane_s16(left_low, 1);
  868. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  869. left_dup = vdupq_lane_s16(left_low, 2);
  870. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  871. left_dup = vdupq_lane_s16(left_low, 3);
  872. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  873. left_dup = vdupq_lane_s16(left_high, 0);
  874. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  875. left_dup = vdupq_lane_s16(left_high, 1);
  876. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  877. left_dup = vdupq_lane_s16(left_high, 2);
  878. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  879. left_dup = vdupq_lane_s16(left_high, 3);
  880. tm_16_kernel(&dst, stride, left_dup, sub0, sub1);
  881. }
  882. }
  883. static INLINE void tm_32_kernel(uint8_t **dst, const ptrdiff_t stride,
  884. const int16x8_t left_dup, const int16x8_t sub0,
  885. const int16x8_t sub1, const int16x8_t sub2,
  886. const int16x8_t sub3) {
  887. const int16x8_t sum0 = vaddq_s16(left_dup, sub0);
  888. const int16x8_t sum1 = vaddq_s16(left_dup, sub1);
  889. const int16x8_t sum2 = vaddq_s16(left_dup, sub2);
  890. const int16x8_t sum3 = vaddq_s16(left_dup, sub3);
  891. const uint8x8_t d0 = vqmovun_s16(sum0);
  892. const uint8x8_t d1 = vqmovun_s16(sum1);
  893. const uint8x8_t d2 = vqmovun_s16(sum2);
  894. const uint8x8_t d3 = vqmovun_s16(sum3);
  895. vst1q_u8(*dst, vcombine_u8(d0, d1));
  896. *dst += 16;
  897. vst1q_u8(*dst, vcombine_u8(d2, d3));
  898. *dst += stride - 16;
  899. }
  900. void vpx_tm_predictor_32x32_neon(uint8_t *dst, ptrdiff_t stride,
  901. const uint8_t *above, const uint8_t *left) {
  902. const uint8x16_t top_left = vld1q_dup_u8(above - 1);
  903. const uint8x16_t above_low = vld1q_u8(above);
  904. const uint8x16_t above_high = vld1q_u8(above + 16);
  905. const int16x8_t sub0 = vreinterpretq_s16_u16(
  906. vsubl_u8(vget_low_u8(above_low), vget_low_u8(top_left)));
  907. const int16x8_t sub1 = vreinterpretq_s16_u16(
  908. vsubl_u8(vget_high_u8(above_low), vget_high_u8(top_left)));
  909. const int16x8_t sub2 = vreinterpretq_s16_u16(
  910. vsubl_u8(vget_low_u8(above_high), vget_low_u8(top_left)));
  911. const int16x8_t sub3 = vreinterpretq_s16_u16(
  912. vsubl_u8(vget_high_u8(above_high), vget_high_u8(top_left)));
  913. int16x8_t left_dup;
  914. int i, j;
  915. for (j = 0; j < 4; j++, left += 8) {
  916. const uint8x8_t left_u8 = vld1_u8(left);
  917. const int16x8_t left_s16q = convert_u8_to_s16(left_u8);
  918. int16x4_t left_s16d = vget_low_s16(left_s16q);
  919. for (i = 0; i < 2; i++, left_s16d = vget_high_s16(left_s16q)) {
  920. left_dup = vdupq_lane_s16(left_s16d, 0);
  921. tm_32_kernel(&dst, stride, left_dup, sub0, sub1, sub2, sub3);
  922. left_dup = vdupq_lane_s16(left_s16d, 1);
  923. tm_32_kernel(&dst, stride, left_dup, sub0, sub1, sub2, sub3);
  924. left_dup = vdupq_lane_s16(left_s16d, 2);
  925. tm_32_kernel(&dst, stride, left_dup, sub0, sub1, sub2, sub3);
  926. left_dup = vdupq_lane_s16(left_s16d, 3);
  927. tm_32_kernel(&dst, stride, left_dup, sub0, sub1, sub2, sub3);
  928. }
  929. }
  930. }
  931. #endif // !HAVE_NEON_ASM