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.

644 lines
27KB

  1. /*
  2. * DSP utils
  3. * Copyright (c) 2000, 2001 Fabrice Bellard
  4. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  5. *
  6. * gmc & q-pel & 32/64 bit based MC by Michael Niedermayer <michaelni@gmx.at>
  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. /**
  25. * @file
  26. * DSP utils
  27. */
  28. #include "bit_depth_template.c"
  29. static inline void FUNC(copy_block2)(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h)
  30. {
  31. int i;
  32. for(i=0; i<h; i++)
  33. {
  34. AV_WN2P(dst , AV_RN2P(src ));
  35. dst+=dstStride;
  36. src+=srcStride;
  37. }
  38. }
  39. static inline void FUNC(copy_block4)(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h)
  40. {
  41. int i;
  42. for(i=0; i<h; i++)
  43. {
  44. AV_WN4P(dst , AV_RN4P(src ));
  45. dst+=dstStride;
  46. src+=srcStride;
  47. }
  48. }
  49. static inline void FUNC(copy_block8)(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h)
  50. {
  51. int i;
  52. for(i=0; i<h; i++)
  53. {
  54. AV_WN4P(dst , AV_RN4P(src ));
  55. AV_WN4P(dst+4*sizeof(pixel), AV_RN4P(src+4*sizeof(pixel)));
  56. dst+=dstStride;
  57. src+=srcStride;
  58. }
  59. }
  60. static inline void FUNC(copy_block16)(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h)
  61. {
  62. int i;
  63. for(i=0; i<h; i++)
  64. {
  65. AV_WN4P(dst , AV_RN4P(src ));
  66. AV_WN4P(dst+ 4*sizeof(pixel), AV_RN4P(src+ 4*sizeof(pixel)));
  67. AV_WN4P(dst+ 8*sizeof(pixel), AV_RN4P(src+ 8*sizeof(pixel)));
  68. AV_WN4P(dst+12*sizeof(pixel), AV_RN4P(src+12*sizeof(pixel)));
  69. dst+=dstStride;
  70. src+=srcStride;
  71. }
  72. }
  73. /* draw the edges of width 'w' of an image of size width, height */
  74. //FIXME check that this is ok for mpeg4 interlaced
  75. static void FUNCC(draw_edges)(uint8_t *_buf, int _wrap, int width, int height, int w, int h, int sides)
  76. {
  77. pixel *buf = (pixel*)_buf;
  78. int wrap = _wrap / sizeof(pixel);
  79. pixel *ptr, *last_line;
  80. int i;
  81. /* left and right */
  82. ptr = buf;
  83. for(i=0;i<height;i++) {
  84. #if BIT_DEPTH > 8
  85. int j;
  86. for (j = 0; j < w; j++) {
  87. ptr[j-w] = ptr[0];
  88. ptr[j+width] = ptr[width-1];
  89. }
  90. #else
  91. memset(ptr - w, ptr[0], w);
  92. memset(ptr + width, ptr[width-1], w);
  93. #endif
  94. ptr += wrap;
  95. }
  96. /* top and bottom + corners */
  97. buf -= w;
  98. last_line = buf + (height - 1) * wrap;
  99. if (sides & EDGE_TOP)
  100. for(i = 0; i < h; i++)
  101. memcpy(buf - (i + 1) * wrap, buf, (width + w + w) * sizeof(pixel)); // top
  102. if (sides & EDGE_BOTTOM)
  103. for (i = 0; i < h; i++)
  104. memcpy(last_line + (i + 1) * wrap, last_line, (width + w + w) * sizeof(pixel)); // bottom
  105. }
  106. #define DCTELEM_FUNCS(dctcoef, suffix) \
  107. static void FUNCC(get_pixels ## suffix)(int16_t *restrict _block, \
  108. const uint8_t *_pixels, \
  109. int line_size) \
  110. { \
  111. const pixel *pixels = (const pixel *) _pixels; \
  112. dctcoef *restrict block = (dctcoef *) _block; \
  113. int i; \
  114. \
  115. /* read the pixels */ \
  116. for(i=0;i<8;i++) { \
  117. block[0] = pixels[0]; \
  118. block[1] = pixels[1]; \
  119. block[2] = pixels[2]; \
  120. block[3] = pixels[3]; \
  121. block[4] = pixels[4]; \
  122. block[5] = pixels[5]; \
  123. block[6] = pixels[6]; \
  124. block[7] = pixels[7]; \
  125. pixels += line_size / sizeof(pixel); \
  126. block += 8; \
  127. } \
  128. } \
  129. \
  130. static void FUNCC(add_pixels8 ## suffix)(uint8_t *restrict _pixels, \
  131. int16_t *_block, \
  132. int line_size) \
  133. { \
  134. int i; \
  135. pixel *restrict pixels = (pixel *restrict)_pixels; \
  136. dctcoef *block = (dctcoef*)_block; \
  137. line_size /= sizeof(pixel); \
  138. \
  139. for(i=0;i<8;i++) { \
  140. pixels[0] += block[0]; \
  141. pixels[1] += block[1]; \
  142. pixels[2] += block[2]; \
  143. pixels[3] += block[3]; \
  144. pixels[4] += block[4]; \
  145. pixels[5] += block[5]; \
  146. pixels[6] += block[6]; \
  147. pixels[7] += block[7]; \
  148. pixels += line_size; \
  149. block += 8; \
  150. } \
  151. } \
  152. \
  153. static void FUNCC(add_pixels4 ## suffix)(uint8_t *restrict _pixels, \
  154. int16_t *_block, \
  155. int line_size) \
  156. { \
  157. int i; \
  158. pixel *restrict pixels = (pixel *restrict)_pixels; \
  159. dctcoef *block = (dctcoef*)_block; \
  160. line_size /= sizeof(pixel); \
  161. \
  162. for(i=0;i<4;i++) { \
  163. pixels[0] += block[0]; \
  164. pixels[1] += block[1]; \
  165. pixels[2] += block[2]; \
  166. pixels[3] += block[3]; \
  167. pixels += line_size; \
  168. block += 4; \
  169. } \
  170. } \
  171. \
  172. static void FUNCC(clear_block ## suffix)(int16_t *block) \
  173. { \
  174. memset(block, 0, sizeof(dctcoef)*64); \
  175. } \
  176. \
  177. /** \
  178. * memset(blocks, 0, sizeof(int16_t)*6*64) \
  179. */ \
  180. static void FUNCC(clear_blocks ## suffix)(int16_t *blocks) \
  181. { \
  182. memset(blocks, 0, sizeof(dctcoef)*6*64); \
  183. }
  184. DCTELEM_FUNCS(int16_t, _16)
  185. #if BIT_DEPTH > 8
  186. DCTELEM_FUNCS(dctcoef, _32)
  187. #endif
  188. #include "hpel_template.c"
  189. #define PIXOP2(OPNAME, OP) \
  190. static inline void FUNC(OPNAME ## _no_rnd_pixels8_l2)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, int dst_stride, \
  191. int src_stride1, int src_stride2, int h){\
  192. int i;\
  193. for(i=0; i<h; i++){\
  194. pixel4 a,b;\
  195. a= AV_RN4P(&src1[i*src_stride1 ]);\
  196. b= AV_RN4P(&src2[i*src_stride2 ]);\
  197. OP(*((pixel4*)&dst[i*dst_stride ]), no_rnd_avg_pixel4(a, b));\
  198. a= AV_RN4P(&src1[i*src_stride1+4*sizeof(pixel)]);\
  199. b= AV_RN4P(&src2[i*src_stride2+4*sizeof(pixel)]);\
  200. OP(*((pixel4*)&dst[i*dst_stride+4*sizeof(pixel)]), no_rnd_avg_pixel4(a, b));\
  201. }\
  202. }\
  203. \
  204. static inline void FUNC(OPNAME ## _no_rnd_pixels16_l2)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, int dst_stride, \
  205. int src_stride1, int src_stride2, int h){\
  206. FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst , src1 , src2 , dst_stride, src_stride1, src_stride2, h);\
  207. FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, h);\
  208. }\
  209. \
  210. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  211. FUNC(OPNAME ## _no_rnd_pixels8_l2)(block, pixels, pixels+sizeof(pixel), line_size, line_size, line_size, h);\
  212. }\
  213. \
  214. static inline void FUNCC(OPNAME ## _pixels8_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  215. FUNC(OPNAME ## _pixels8_l2)(block, pixels, pixels+sizeof(pixel), line_size, line_size, line_size, h);\
  216. }\
  217. \
  218. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  219. FUNC(OPNAME ## _no_rnd_pixels8_l2)(block, pixels, pixels+line_size, line_size, line_size, line_size, h);\
  220. }\
  221. \
  222. static inline void FUNCC(OPNAME ## _pixels8_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  223. FUNC(OPNAME ## _pixels8_l2)(block, pixels, pixels+line_size, line_size, line_size, line_size, h);\
  224. }\
  225. \
  226. static inline void FUNC(OPNAME ## _pixels8_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
  227. int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
  228. /* FIXME HIGH BIT DEPTH */\
  229. int i;\
  230. for(i=0; i<h; i++){\
  231. uint32_t a, b, c, d, l0, l1, h0, h1;\
  232. a= AV_RN32(&src1[i*src_stride1]);\
  233. b= AV_RN32(&src2[i*src_stride2]);\
  234. c= AV_RN32(&src3[i*src_stride3]);\
  235. d= AV_RN32(&src4[i*src_stride4]);\
  236. l0= (a&0x03030303UL)\
  237. + (b&0x03030303UL)\
  238. + 0x02020202UL;\
  239. h0= ((a&0xFCFCFCFCUL)>>2)\
  240. + ((b&0xFCFCFCFCUL)>>2);\
  241. l1= (c&0x03030303UL)\
  242. + (d&0x03030303UL);\
  243. h1= ((c&0xFCFCFCFCUL)>>2)\
  244. + ((d&0xFCFCFCFCUL)>>2);\
  245. OP(*((uint32_t*)&dst[i*dst_stride]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  246. a= AV_RN32(&src1[i*src_stride1+4]);\
  247. b= AV_RN32(&src2[i*src_stride2+4]);\
  248. c= AV_RN32(&src3[i*src_stride3+4]);\
  249. d= AV_RN32(&src4[i*src_stride4+4]);\
  250. l0= (a&0x03030303UL)\
  251. + (b&0x03030303UL)\
  252. + 0x02020202UL;\
  253. h0= ((a&0xFCFCFCFCUL)>>2)\
  254. + ((b&0xFCFCFCFCUL)>>2);\
  255. l1= (c&0x03030303UL)\
  256. + (d&0x03030303UL);\
  257. h1= ((c&0xFCFCFCFCUL)>>2)\
  258. + ((d&0xFCFCFCFCUL)>>2);\
  259. OP(*((uint32_t*)&dst[i*dst_stride+4]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  260. }\
  261. }\
  262. \
  263. static inline void FUNCC(OPNAME ## _pixels4_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  264. FUNC(OPNAME ## _pixels4_l2)(block, pixels, pixels+sizeof(pixel), line_size, line_size, line_size, h);\
  265. }\
  266. \
  267. static inline void FUNCC(OPNAME ## _pixels4_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  268. FUNC(OPNAME ## _pixels4_l2)(block, pixels, pixels+line_size, line_size, line_size, line_size, h);\
  269. }\
  270. \
  271. static inline void FUNCC(OPNAME ## _pixels2_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  272. FUNC(OPNAME ## _pixels2_l2)(block, pixels, pixels+sizeof(pixel), line_size, line_size, line_size, h);\
  273. }\
  274. \
  275. static inline void FUNCC(OPNAME ## _pixels2_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){\
  276. FUNC(OPNAME ## _pixels2_l2)(block, pixels, pixels+line_size, line_size, line_size, line_size, h);\
  277. }\
  278. \
  279. static inline void FUNC(OPNAME ## _no_rnd_pixels8_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
  280. int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
  281. /* FIXME HIGH BIT DEPTH*/\
  282. int i;\
  283. for(i=0; i<h; i++){\
  284. uint32_t a, b, c, d, l0, l1, h0, h1;\
  285. a= AV_RN32(&src1[i*src_stride1]);\
  286. b= AV_RN32(&src2[i*src_stride2]);\
  287. c= AV_RN32(&src3[i*src_stride3]);\
  288. d= AV_RN32(&src4[i*src_stride4]);\
  289. l0= (a&0x03030303UL)\
  290. + (b&0x03030303UL)\
  291. + 0x01010101UL;\
  292. h0= ((a&0xFCFCFCFCUL)>>2)\
  293. + ((b&0xFCFCFCFCUL)>>2);\
  294. l1= (c&0x03030303UL)\
  295. + (d&0x03030303UL);\
  296. h1= ((c&0xFCFCFCFCUL)>>2)\
  297. + ((d&0xFCFCFCFCUL)>>2);\
  298. OP(*((uint32_t*)&dst[i*dst_stride]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  299. a= AV_RN32(&src1[i*src_stride1+4]);\
  300. b= AV_RN32(&src2[i*src_stride2+4]);\
  301. c= AV_RN32(&src3[i*src_stride3+4]);\
  302. d= AV_RN32(&src4[i*src_stride4+4]);\
  303. l0= (a&0x03030303UL)\
  304. + (b&0x03030303UL)\
  305. + 0x01010101UL;\
  306. h0= ((a&0xFCFCFCFCUL)>>2)\
  307. + ((b&0xFCFCFCFCUL)>>2);\
  308. l1= (c&0x03030303UL)\
  309. + (d&0x03030303UL);\
  310. h1= ((c&0xFCFCFCFCUL)>>2)\
  311. + ((d&0xFCFCFCFCUL)>>2);\
  312. OP(*((uint32_t*)&dst[i*dst_stride+4]), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  313. }\
  314. }\
  315. static inline void FUNC(OPNAME ## _pixels16_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
  316. int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
  317. FUNC(OPNAME ## _pixels8_l4)(dst , src1 , src2 , src3 , src4 , dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
  318. FUNC(OPNAME ## _pixels8_l4)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), src3+8*sizeof(pixel), src4+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
  319. }\
  320. static inline void FUNC(OPNAME ## _no_rnd_pixels16_l4)(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, const uint8_t *src3, const uint8_t *src4,\
  321. int dst_stride, int src_stride1, int src_stride2,int src_stride3,int src_stride4, int h){\
  322. FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst , src1 , src2 , src3 , src4 , dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
  323. FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst+8*sizeof(pixel), src1+8*sizeof(pixel), src2+8*sizeof(pixel), src3+8*sizeof(pixel), src4+8*sizeof(pixel), dst_stride, src_stride1, src_stride2, src_stride3, src_stride4, h);\
  324. }\
  325. \
  326. static inline void FUNCC(OPNAME ## _pixels2_xy2)(uint8_t *_block, const uint8_t *_pixels, int line_size, int h)\
  327. {\
  328. int i, a0, b0, a1, b1;\
  329. pixel *block = (pixel*)_block;\
  330. const pixel *pixels = (const pixel*)_pixels;\
  331. line_size /= sizeof(pixel);\
  332. a0= pixels[0];\
  333. b0= pixels[1] + 2;\
  334. a0 += b0;\
  335. b0 += pixels[2];\
  336. \
  337. pixels+=line_size;\
  338. for(i=0; i<h; i+=2){\
  339. a1= pixels[0];\
  340. b1= pixels[1];\
  341. a1 += b1;\
  342. b1 += pixels[2];\
  343. \
  344. block[0]= (a1+a0)>>2; /* FIXME non put */\
  345. block[1]= (b1+b0)>>2;\
  346. \
  347. pixels+=line_size;\
  348. block +=line_size;\
  349. \
  350. a0= pixels[0];\
  351. b0= pixels[1] + 2;\
  352. a0 += b0;\
  353. b0 += pixels[2];\
  354. \
  355. block[0]= (a1+a0)>>2;\
  356. block[1]= (b1+b0)>>2;\
  357. pixels+=line_size;\
  358. block +=line_size;\
  359. }\
  360. }\
  361. \
  362. static inline void FUNCC(OPNAME ## _pixels4_xy2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)\
  363. {\
  364. /* FIXME HIGH BIT DEPTH */\
  365. int i;\
  366. const uint32_t a= AV_RN32(pixels );\
  367. const uint32_t b= AV_RN32(pixels+1);\
  368. uint32_t l0= (a&0x03030303UL)\
  369. + (b&0x03030303UL)\
  370. + 0x02020202UL;\
  371. uint32_t h0= ((a&0xFCFCFCFCUL)>>2)\
  372. + ((b&0xFCFCFCFCUL)>>2);\
  373. uint32_t l1,h1;\
  374. \
  375. pixels+=line_size;\
  376. for(i=0; i<h; i+=2){\
  377. uint32_t a= AV_RN32(pixels );\
  378. uint32_t b= AV_RN32(pixels+1);\
  379. l1= (a&0x03030303UL)\
  380. + (b&0x03030303UL);\
  381. h1= ((a&0xFCFCFCFCUL)>>2)\
  382. + ((b&0xFCFCFCFCUL)>>2);\
  383. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  384. pixels+=line_size;\
  385. block +=line_size;\
  386. a= AV_RN32(pixels );\
  387. b= AV_RN32(pixels+1);\
  388. l0= (a&0x03030303UL)\
  389. + (b&0x03030303UL)\
  390. + 0x02020202UL;\
  391. h0= ((a&0xFCFCFCFCUL)>>2)\
  392. + ((b&0xFCFCFCFCUL)>>2);\
  393. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  394. pixels+=line_size;\
  395. block +=line_size;\
  396. }\
  397. }\
  398. \
  399. static inline void FUNCC(OPNAME ## _pixels8_xy2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)\
  400. {\
  401. /* FIXME HIGH BIT DEPTH */\
  402. int j;\
  403. for(j=0; j<2; j++){\
  404. int i;\
  405. const uint32_t a= AV_RN32(pixels );\
  406. const uint32_t b= AV_RN32(pixels+1);\
  407. uint32_t l0= (a&0x03030303UL)\
  408. + (b&0x03030303UL)\
  409. + 0x02020202UL;\
  410. uint32_t h0= ((a&0xFCFCFCFCUL)>>2)\
  411. + ((b&0xFCFCFCFCUL)>>2);\
  412. uint32_t l1,h1;\
  413. \
  414. pixels+=line_size;\
  415. for(i=0; i<h; i+=2){\
  416. uint32_t a= AV_RN32(pixels );\
  417. uint32_t b= AV_RN32(pixels+1);\
  418. l1= (a&0x03030303UL)\
  419. + (b&0x03030303UL);\
  420. h1= ((a&0xFCFCFCFCUL)>>2)\
  421. + ((b&0xFCFCFCFCUL)>>2);\
  422. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  423. pixels+=line_size;\
  424. block +=line_size;\
  425. a= AV_RN32(pixels );\
  426. b= AV_RN32(pixels+1);\
  427. l0= (a&0x03030303UL)\
  428. + (b&0x03030303UL)\
  429. + 0x02020202UL;\
  430. h0= ((a&0xFCFCFCFCUL)>>2)\
  431. + ((b&0xFCFCFCFCUL)>>2);\
  432. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  433. pixels+=line_size;\
  434. block +=line_size;\
  435. }\
  436. pixels+=4-line_size*(h+1);\
  437. block +=4-line_size*h;\
  438. }\
  439. }\
  440. \
  441. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_xy2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)\
  442. {\
  443. /* FIXME HIGH BIT DEPTH */\
  444. int j;\
  445. for(j=0; j<2; j++){\
  446. int i;\
  447. const uint32_t a= AV_RN32(pixels );\
  448. const uint32_t b= AV_RN32(pixels+1);\
  449. uint32_t l0= (a&0x03030303UL)\
  450. + (b&0x03030303UL)\
  451. + 0x01010101UL;\
  452. uint32_t h0= ((a&0xFCFCFCFCUL)>>2)\
  453. + ((b&0xFCFCFCFCUL)>>2);\
  454. uint32_t l1,h1;\
  455. \
  456. pixels+=line_size;\
  457. for(i=0; i<h; i+=2){\
  458. uint32_t a= AV_RN32(pixels );\
  459. uint32_t b= AV_RN32(pixels+1);\
  460. l1= (a&0x03030303UL)\
  461. + (b&0x03030303UL);\
  462. h1= ((a&0xFCFCFCFCUL)>>2)\
  463. + ((b&0xFCFCFCFCUL)>>2);\
  464. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  465. pixels+=line_size;\
  466. block +=line_size;\
  467. a= AV_RN32(pixels );\
  468. b= AV_RN32(pixels+1);\
  469. l0= (a&0x03030303UL)\
  470. + (b&0x03030303UL)\
  471. + 0x01010101UL;\
  472. h0= ((a&0xFCFCFCFCUL)>>2)\
  473. + ((b&0xFCFCFCFCUL)>>2);\
  474. OP(*((uint32_t*)block), h0+h1+(((l0+l1)>>2)&0x0F0F0F0FUL));\
  475. pixels+=line_size;\
  476. block +=line_size;\
  477. }\
  478. pixels+=4-line_size*(h+1);\
  479. block +=4-line_size*h;\
  480. }\
  481. }\
  482. \
  483. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_x2) , FUNCC(OPNAME ## _pixels8_x2) , 8*sizeof(pixel))\
  484. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_y2) , FUNCC(OPNAME ## _pixels8_y2) , 8*sizeof(pixel))\
  485. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_xy2), FUNCC(OPNAME ## _pixels8_xy2), 8*sizeof(pixel))\
  486. av_unused CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16) , FUNCC(OPNAME ## _pixels8) , 8*sizeof(pixel))\
  487. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_x2) , FUNCC(OPNAME ## _no_rnd_pixels8_x2) , 8*sizeof(pixel))\
  488. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_y2) , FUNCC(OPNAME ## _no_rnd_pixels8_y2) , 8*sizeof(pixel))\
  489. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_xy2), FUNCC(OPNAME ## _no_rnd_pixels8_xy2), 8*sizeof(pixel))\
  490. #define op_avg(a, b) a = rnd_avg_pixel4(a, b)
  491. #define op_put(a, b) a = b
  492. #if BIT_DEPTH == 8
  493. #define put_no_rnd_pixels8_8_c put_pixels8_8_c
  494. PIXOP2(avg, op_avg)
  495. PIXOP2(put, op_put)
  496. #endif
  497. #undef op_avg
  498. #undef op_put
  499. #define H264_CHROMA_MC(OPNAME, OP)\
  500. static void FUNCC(OPNAME ## h264_chroma_mc2)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  501. pixel *dst = (pixel*)_dst;\
  502. pixel *src = (pixel*)_src;\
  503. const int A=(8-x)*(8-y);\
  504. const int B=( x)*(8-y);\
  505. const int C=(8-x)*( y);\
  506. const int D=( x)*( y);\
  507. int i;\
  508. stride /= sizeof(pixel);\
  509. \
  510. assert(x<8 && y<8 && x>=0 && y>=0);\
  511. \
  512. if(D){\
  513. for(i=0; i<h; i++){\
  514. OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
  515. OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
  516. dst+= stride;\
  517. src+= stride;\
  518. }\
  519. }else{\
  520. const int E= B+C;\
  521. const int step= C ? stride : 1;\
  522. for(i=0; i<h; i++){\
  523. OP(dst[0], (A*src[0] + E*src[step+0]));\
  524. OP(dst[1], (A*src[1] + E*src[step+1]));\
  525. dst+= stride;\
  526. src+= stride;\
  527. }\
  528. }\
  529. }\
  530. \
  531. static void FUNCC(OPNAME ## h264_chroma_mc4)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  532. pixel *dst = (pixel*)_dst;\
  533. pixel *src = (pixel*)_src;\
  534. const int A=(8-x)*(8-y);\
  535. const int B=( x)*(8-y);\
  536. const int C=(8-x)*( y);\
  537. const int D=( x)*( y);\
  538. int i;\
  539. stride /= sizeof(pixel);\
  540. \
  541. assert(x<8 && y<8 && x>=0 && y>=0);\
  542. \
  543. if(D){\
  544. for(i=0; i<h; i++){\
  545. OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
  546. OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
  547. OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
  548. OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
  549. dst+= stride;\
  550. src+= stride;\
  551. }\
  552. }else{\
  553. const int E= B+C;\
  554. const int step= C ? stride : 1;\
  555. for(i=0; i<h; i++){\
  556. OP(dst[0], (A*src[0] + E*src[step+0]));\
  557. OP(dst[1], (A*src[1] + E*src[step+1]));\
  558. OP(dst[2], (A*src[2] + E*src[step+2]));\
  559. OP(dst[3], (A*src[3] + E*src[step+3]));\
  560. dst+= stride;\
  561. src+= stride;\
  562. }\
  563. }\
  564. }\
  565. \
  566. static void FUNCC(OPNAME ## h264_chroma_mc8)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  567. pixel *dst = (pixel*)_dst;\
  568. pixel *src = (pixel*)_src;\
  569. const int A=(8-x)*(8-y);\
  570. const int B=( x)*(8-y);\
  571. const int C=(8-x)*( y);\
  572. const int D=( x)*( y);\
  573. int i;\
  574. stride /= sizeof(pixel);\
  575. \
  576. assert(x<8 && y<8 && x>=0 && y>=0);\
  577. \
  578. if(D){\
  579. for(i=0; i<h; i++){\
  580. OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
  581. OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
  582. OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
  583. OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
  584. OP(dst[4], (A*src[4] + B*src[5] + C*src[stride+4] + D*src[stride+5]));\
  585. OP(dst[5], (A*src[5] + B*src[6] + C*src[stride+5] + D*src[stride+6]));\
  586. OP(dst[6], (A*src[6] + B*src[7] + C*src[stride+6] + D*src[stride+7]));\
  587. OP(dst[7], (A*src[7] + B*src[8] + C*src[stride+7] + D*src[stride+8]));\
  588. dst+= stride;\
  589. src+= stride;\
  590. }\
  591. }else{\
  592. const int E= B+C;\
  593. const int step= C ? stride : 1;\
  594. for(i=0; i<h; i++){\
  595. OP(dst[0], (A*src[0] + E*src[step+0]));\
  596. OP(dst[1], (A*src[1] + E*src[step+1]));\
  597. OP(dst[2], (A*src[2] + E*src[step+2]));\
  598. OP(dst[3], (A*src[3] + E*src[step+3]));\
  599. OP(dst[4], (A*src[4] + E*src[step+4]));\
  600. OP(dst[5], (A*src[5] + E*src[step+5]));\
  601. OP(dst[6], (A*src[6] + E*src[step+6]));\
  602. OP(dst[7], (A*src[7] + E*src[step+7]));\
  603. dst+= stride;\
  604. src+= stride;\
  605. }\
  606. }\
  607. }
  608. #define op_avg(a, b) a = (((a)+(((b) + 32)>>6)+1)>>1)
  609. #define op_put(a, b) a = (((b) + 32)>>6)
  610. H264_CHROMA_MC(put_ , op_put)
  611. H264_CHROMA_MC(avg_ , op_avg)
  612. #undef op_avg
  613. #undef op_put
  614. void FUNCC(ff_put_pixels8x8)(uint8_t *dst, uint8_t *src, int stride) {
  615. FUNCC(put_pixels8)(dst, src, stride, 8);
  616. }
  617. void FUNCC(ff_avg_pixels8x8)(uint8_t *dst, uint8_t *src, int stride) {
  618. FUNCC(avg_pixels8)(dst, src, stride, 8);
  619. }
  620. void FUNCC(ff_put_pixels16x16)(uint8_t *dst, uint8_t *src, int stride) {
  621. FUNCC(put_pixels16)(dst, src, stride, 16);
  622. }
  623. void FUNCC(ff_avg_pixels16x16)(uint8_t *dst, uint8_t *src, int stride) {
  624. FUNCC(avg_pixels16)(dst, src, stride, 16);
  625. }