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.

339 lines
23KB

  1. /*
  2. * Half-pel DSP functions
  3. *
  4. * Copyright (c) 2000, 2001 Fabrice Bellard
  5. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  6. *
  7. * gmc & q-pel & 32/64 bit based MC by Michael Niedermayer <michaelni@gmx.at>
  8. *
  9. * This file is part of Libav.
  10. *
  11. * Libav is free software; you can redistribute it and/or
  12. * modify it under the terms of the GNU Lesser General Public
  13. * License as published by the Free Software Foundation; either
  14. * version 2.1 of the License, or (at your option) any later version.
  15. *
  16. * Libav is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19. * Lesser General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU Lesser General Public
  22. * License along with Libav; if not, write to the Free Software
  23. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  24. */
  25. /**
  26. * @file
  27. * Half-pel DSP functions
  28. */
  29. #include "bit_depth_template.c"
  30. #include "hpel_template.c"
  31. #define PIXOP2(OPNAME, OP) \
  32. static inline void FUNC(OPNAME ## _no_rnd_pixels8_l2)(uint8_t *dst, \
  33. const uint8_t *src1, \
  34. const uint8_t *src2, \
  35. int dst_stride, \
  36. int src_stride1, \
  37. int src_stride2, \
  38. int h) \
  39. { \
  40. int i; \
  41. \
  42. for (i = 0; i < h; i++) { \
  43. pixel4 a, b; \
  44. a = AV_RN4P(&src1[i * src_stride1]); \
  45. b = AV_RN4P(&src2[i * src_stride2]); \
  46. OP(*((pixel4 *) &dst[i * dst_stride]), \
  47. no_rnd_avg_pixel4(a, b)); \
  48. a = AV_RN4P(&src1[i * src_stride1 + 4 * sizeof(pixel)]); \
  49. b = AV_RN4P(&src2[i * src_stride2 + 4 * sizeof(pixel)]); \
  50. OP(*((pixel4 *) &dst[i * dst_stride + 4 * sizeof(pixel)]), \
  51. no_rnd_avg_pixel4(a, b)); \
  52. } \
  53. } \
  54. \
  55. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_x2)(uint8_t *block, \
  56. const uint8_t *pixels, \
  57. ptrdiff_t line_size, \
  58. int h) \
  59. { \
  60. FUNC(OPNAME ## _no_rnd_pixels8_l2)(block, pixels, pixels + sizeof(pixel), \
  61. line_size, line_size, line_size, h); \
  62. } \
  63. \
  64. static inline void FUNCC(OPNAME ## _pixels8_x2)(uint8_t *block, \
  65. const uint8_t *pixels, \
  66. ptrdiff_t line_size, \
  67. int h) \
  68. { \
  69. FUNC(OPNAME ## _pixels8_l2)(block, pixels, pixels + sizeof(pixel), \
  70. line_size, line_size, line_size, h); \
  71. } \
  72. \
  73. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_y2)(uint8_t *block, \
  74. const uint8_t *pixels, \
  75. ptrdiff_t line_size, \
  76. int h) \
  77. { \
  78. FUNC(OPNAME ## _no_rnd_pixels8_l2)(block, pixels, pixels + line_size, \
  79. line_size, line_size, line_size, h); \
  80. } \
  81. \
  82. static inline void FUNCC(OPNAME ## _pixels8_y2)(uint8_t *block, \
  83. const uint8_t *pixels, \
  84. ptrdiff_t line_size, \
  85. int h) \
  86. { \
  87. FUNC(OPNAME ## _pixels8_l2)(block, pixels, pixels + line_size, \
  88. line_size, line_size, line_size, h); \
  89. } \
  90. \
  91. static inline void FUNCC(OPNAME ## _pixels4_x2)(uint8_t *block, \
  92. const uint8_t *pixels, \
  93. ptrdiff_t line_size, \
  94. int h) \
  95. { \
  96. FUNC(OPNAME ## _pixels4_l2)(block, pixels, pixels + sizeof(pixel), \
  97. line_size, line_size, line_size, h); \
  98. } \
  99. \
  100. static inline void FUNCC(OPNAME ## _pixels4_y2)(uint8_t *block, \
  101. const uint8_t *pixels, \
  102. ptrdiff_t line_size, \
  103. int h) \
  104. { \
  105. FUNC(OPNAME ## _pixels4_l2)(block, pixels, pixels + line_size, \
  106. line_size, line_size, line_size, h); \
  107. } \
  108. \
  109. static inline void FUNCC(OPNAME ## _pixels2_x2)(uint8_t *block, \
  110. const uint8_t *pixels, \
  111. ptrdiff_t line_size, \
  112. int h) \
  113. { \
  114. FUNC(OPNAME ## _pixels2_l2)(block, pixels, pixels + sizeof(pixel), \
  115. line_size, line_size, line_size, h); \
  116. } \
  117. \
  118. static inline void FUNCC(OPNAME ## _pixels2_y2)(uint8_t *block, \
  119. const uint8_t *pixels, \
  120. ptrdiff_t line_size, \
  121. int h) \
  122. { \
  123. FUNC(OPNAME ## _pixels2_l2)(block, pixels, pixels + line_size, \
  124. line_size, line_size, line_size, h); \
  125. } \
  126. \
  127. static inline void FUNCC(OPNAME ## _pixels2_xy2)(uint8_t *_block, \
  128. const uint8_t *_pixels, \
  129. ptrdiff_t line_size, \
  130. int h) \
  131. { \
  132. pixel *block = (pixel *) _block; \
  133. const pixel *pixels = (const pixel *) _pixels; \
  134. int i, a1, b1; \
  135. int a0 = pixels[0]; \
  136. int b0 = pixels[1] + 2; \
  137. \
  138. a0 += b0; \
  139. b0 += pixels[2]; \
  140. line_size /= sizeof(pixel); \
  141. pixels += line_size; \
  142. for (i = 0; i < h; i += 2) { \
  143. a1 = pixels[0]; \
  144. b1 = pixels[1]; \
  145. a1 += b1; \
  146. b1 += pixels[2]; \
  147. \
  148. block[0] = (a1 + a0) >> 2; /* FIXME non put */ \
  149. block[1] = (b1 + b0) >> 2; \
  150. \
  151. pixels += line_size; \
  152. block += line_size; \
  153. \
  154. a0 = pixels[0]; \
  155. b0 = pixels[1] + 2; \
  156. a0 += b0; \
  157. b0 += pixels[2]; \
  158. \
  159. block[0] = (a1 + a0) >> 2; \
  160. block[1] = (b1 + b0) >> 2; \
  161. pixels += line_size; \
  162. block += line_size; \
  163. } \
  164. } \
  165. \
  166. static inline void FUNCC(OPNAME ## _pixels4_xy2)(uint8_t *block, \
  167. const uint8_t *pixels, \
  168. ptrdiff_t line_size, \
  169. int h) \
  170. { \
  171. /* FIXME HIGH BIT DEPTH */ \
  172. int i; \
  173. const uint32_t a = AV_RN32(pixels); \
  174. const uint32_t b = AV_RN32(pixels + 1); \
  175. uint32_t l0 = (a & 0x03030303UL) + \
  176. (b & 0x03030303UL) + \
  177. 0x02020202UL; \
  178. uint32_t h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  179. ((b & 0xFCFCFCFCUL) >> 2); \
  180. uint32_t l1, h1; \
  181. \
  182. pixels += line_size; \
  183. for (i = 0; i < h; i += 2) { \
  184. uint32_t a = AV_RN32(pixels); \
  185. uint32_t b = AV_RN32(pixels + 1); \
  186. l1 = (a & 0x03030303UL) + \
  187. (b & 0x03030303UL); \
  188. h1 = ((a & 0xFCFCFCFCUL) >> 2) + \
  189. ((b & 0xFCFCFCFCUL) >> 2); \
  190. OP(*((uint32_t *) block), h0 + h1 + \
  191. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  192. pixels += line_size; \
  193. block += line_size; \
  194. a = AV_RN32(pixels); \
  195. b = AV_RN32(pixels + 1); \
  196. l0 = (a & 0x03030303UL) + \
  197. (b & 0x03030303UL) + \
  198. 0x02020202UL; \
  199. h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  200. ((b & 0xFCFCFCFCUL) >> 2); \
  201. OP(*((uint32_t *) block), h0 + h1 + \
  202. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  203. pixels += line_size; \
  204. block += line_size; \
  205. } \
  206. } \
  207. \
  208. static inline void FUNCC(OPNAME ## _pixels8_xy2)(uint8_t *block, \
  209. const uint8_t *pixels, \
  210. ptrdiff_t line_size, \
  211. int h) \
  212. { \
  213. /* FIXME HIGH BIT DEPTH */ \
  214. int j; \
  215. \
  216. for (j = 0; j < 2; j++) { \
  217. int i; \
  218. const uint32_t a = AV_RN32(pixels); \
  219. const uint32_t b = AV_RN32(pixels + 1); \
  220. uint32_t l0 = (a & 0x03030303UL) + \
  221. (b & 0x03030303UL) + \
  222. 0x02020202UL; \
  223. uint32_t h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  224. ((b & 0xFCFCFCFCUL) >> 2); \
  225. uint32_t l1, h1; \
  226. \
  227. pixels += line_size; \
  228. for (i = 0; i < h; i += 2) { \
  229. uint32_t a = AV_RN32(pixels); \
  230. uint32_t b = AV_RN32(pixels + 1); \
  231. l1 = (a & 0x03030303UL) + \
  232. (b & 0x03030303UL); \
  233. h1 = ((a & 0xFCFCFCFCUL) >> 2) + \
  234. ((b & 0xFCFCFCFCUL) >> 2); \
  235. OP(*((uint32_t *) block), h0 + h1 + \
  236. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  237. pixels += line_size; \
  238. block += line_size; \
  239. a = AV_RN32(pixels); \
  240. b = AV_RN32(pixels + 1); \
  241. l0 = (a & 0x03030303UL) + \
  242. (b & 0x03030303UL) + \
  243. 0x02020202UL; \
  244. h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  245. ((b & 0xFCFCFCFCUL) >> 2); \
  246. OP(*((uint32_t *) block), h0 + h1 + \
  247. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  248. pixels += line_size; \
  249. block += line_size; \
  250. } \
  251. pixels += 4 - line_size * (h + 1); \
  252. block += 4 - line_size * h; \
  253. } \
  254. } \
  255. \
  256. static inline void FUNCC(OPNAME ## _no_rnd_pixels8_xy2)(uint8_t *block, \
  257. const uint8_t *pixels, \
  258. ptrdiff_t line_size, \
  259. int h) \
  260. { \
  261. /* FIXME HIGH BIT DEPTH */ \
  262. int j; \
  263. \
  264. for (j = 0; j < 2; j++) { \
  265. int i; \
  266. const uint32_t a = AV_RN32(pixels); \
  267. const uint32_t b = AV_RN32(pixels + 1); \
  268. uint32_t l0 = (a & 0x03030303UL) + \
  269. (b & 0x03030303UL) + \
  270. 0x01010101UL; \
  271. uint32_t h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  272. ((b & 0xFCFCFCFCUL) >> 2); \
  273. uint32_t l1, h1; \
  274. \
  275. pixels += line_size; \
  276. for (i = 0; i < h; i += 2) { \
  277. uint32_t a = AV_RN32(pixels); \
  278. uint32_t b = AV_RN32(pixels + 1); \
  279. l1 = (a & 0x03030303UL) + \
  280. (b & 0x03030303UL); \
  281. h1 = ((a & 0xFCFCFCFCUL) >> 2) + \
  282. ((b & 0xFCFCFCFCUL) >> 2); \
  283. OP(*((uint32_t *) block), h0 + h1 + \
  284. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  285. pixels += line_size; \
  286. block += line_size; \
  287. a = AV_RN32(pixels); \
  288. b = AV_RN32(pixels + 1); \
  289. l0 = (a & 0x03030303UL) + \
  290. (b & 0x03030303UL) + \
  291. 0x01010101UL; \
  292. h0 = ((a & 0xFCFCFCFCUL) >> 2) + \
  293. ((b & 0xFCFCFCFCUL) >> 2); \
  294. OP(*((uint32_t *) block), h0 + h1 + \
  295. (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \
  296. pixels += line_size; \
  297. block += line_size; \
  298. } \
  299. pixels += 4 - line_size * (h + 1); \
  300. block += 4 - line_size * h; \
  301. } \
  302. } \
  303. \
  304. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_x2), \
  305. FUNCC(OPNAME ## _pixels8_x2), \
  306. 8 * sizeof(pixel)) \
  307. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_y2), \
  308. FUNCC(OPNAME ## _pixels8_y2), \
  309. 8 * sizeof(pixel)) \
  310. CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_xy2), \
  311. FUNCC(OPNAME ## _pixels8_xy2), \
  312. 8 * sizeof(pixel)) \
  313. av_unused CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16), \
  314. FUNCC(OPNAME ## _pixels8), \
  315. 8 * sizeof(pixel)) \
  316. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_x2), \
  317. FUNCC(OPNAME ## _no_rnd_pixels8_x2), \
  318. 8 * sizeof(pixel)) \
  319. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_y2), \
  320. FUNCC(OPNAME ## _no_rnd_pixels8_y2), \
  321. 8 * sizeof(pixel)) \
  322. CALL_2X_PIXELS(FUNCC(OPNAME ## _no_rnd_pixels16_xy2), \
  323. FUNCC(OPNAME ## _no_rnd_pixels8_xy2), \
  324. 8 * sizeof(pixel)) \
  325. #define op_avg(a, b) a = rnd_avg_pixel4(a, b)
  326. #define op_put(a, b) a = b
  327. #if BIT_DEPTH == 8
  328. #define put_no_rnd_pixels8_8_c put_pixels8_8_c
  329. PIXOP2(avg, op_avg)
  330. PIXOP2(put, op_put)
  331. #endif
  332. #undef op_avg
  333. #undef op_put