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.

250 lines
11KB

  1. /*
  2. * RV30 decoder motion compensation functions
  3. * Copyright (c) 2007 Konstantin Shishkov
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file rv30dsp.c
  23. * RV30 decoder motion compensation functions
  24. */
  25. #include "avcodec.h"
  26. #include "dsputil.h"
  27. #define RV30_LOWPASS(OPNAME, OP) \
  28. static av_unused void OPNAME ## rv30_tpel8_h_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, const int C1, const int C2){\
  29. const int h=8;\
  30. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;\
  31. int i;\
  32. for(i=0; i<h; i++)\
  33. {\
  34. OP(dst[0], -(src[-1]+src[2]) + src[0]*C1 + src[1]*C2);\
  35. OP(dst[1], -(src[ 0]+src[3]) + src[1]*C1 + src[2]*C2);\
  36. OP(dst[2], -(src[ 1]+src[4]) + src[2]*C1 + src[3]*C2);\
  37. OP(dst[3], -(src[ 2]+src[5]) + src[3]*C1 + src[4]*C2);\
  38. OP(dst[4], -(src[ 3]+src[6]) + src[4]*C1 + src[5]*C2);\
  39. OP(dst[5], -(src[ 4]+src[7]) + src[5]*C1 + src[6]*C2);\
  40. OP(dst[6], -(src[ 5]+src[8]) + src[6]*C1 + src[7]*C2);\
  41. OP(dst[7], -(src[ 6]+src[9]) + src[7]*C1 + src[8]*C2);\
  42. dst+=dstStride;\
  43. src+=srcStride;\
  44. }\
  45. }\
  46. \
  47. static void OPNAME ## rv30_tpel8_v_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, const int C1, const int C2){\
  48. const int w=8;\
  49. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;\
  50. int i;\
  51. for(i=0; i<w; i++)\
  52. {\
  53. const int srcA= src[-1*srcStride];\
  54. const int src0= src[0 *srcStride];\
  55. const int src1= src[1 *srcStride];\
  56. const int src2= src[2 *srcStride];\
  57. const int src3= src[3 *srcStride];\
  58. const int src4= src[4 *srcStride];\
  59. const int src5= src[5 *srcStride];\
  60. const int src6= src[6 *srcStride];\
  61. const int src7= src[7 *srcStride];\
  62. const int src8= src[8 *srcStride];\
  63. const int src9= src[9 *srcStride];\
  64. OP(dst[0*dstStride], -(srcA+src2) + src0*C1 + src1*C2);\
  65. OP(dst[1*dstStride], -(src0+src3) + src1*C1 + src2*C2);\
  66. OP(dst[2*dstStride], -(src1+src4) + src2*C1 + src3*C2);\
  67. OP(dst[3*dstStride], -(src2+src5) + src3*C1 + src4*C2);\
  68. OP(dst[4*dstStride], -(src3+src6) + src4*C1 + src5*C2);\
  69. OP(dst[5*dstStride], -(src4+src7) + src5*C1 + src6*C2);\
  70. OP(dst[6*dstStride], -(src5+src8) + src6*C1 + src7*C2);\
  71. OP(dst[7*dstStride], -(src6+src9) + src7*C1 + src8*C2);\
  72. dst++;\
  73. src++;\
  74. }\
  75. }\
  76. \
  77. static void OPNAME ## rv30_tpel8_h3_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  78. const int h=8+2;\
  79. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;\
  80. int i;\
  81. for(i=0; i<h; i++)\
  82. {\
  83. OP(dst[0], 6*src[0]+9*src[1]+src[2]);\
  84. OP(dst[1], 6*src[1]+9*src[2]+src[3]);\
  85. OP(dst[2], 6*src[2]+9*src[3]+src[4]);\
  86. OP(dst[3], 6*src[3]+9*src[4]+src[5]);\
  87. OP(dst[4], 6*src[4]+9*src[5]+src[6]);\
  88. OP(dst[5], 6*src[5]+9*src[6]+src[7]);\
  89. OP(dst[6], 6*src[6]+9*src[7]+src[8]);\
  90. OP(dst[7], 6*src[7]+9*src[8]+src[9]);\
  91. dst+=dstStride;\
  92. src+=srcStride;\
  93. }\
  94. }\
  95. \
  96. static void OPNAME ## rv30_tpel8_v3_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  97. const int w=8;\
  98. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;\
  99. int i;\
  100. for(i=0; i<w; i++)\
  101. {\
  102. const int src0= src[0 *srcStride];\
  103. const int src1= src[1 *srcStride];\
  104. const int src2= src[2 *srcStride];\
  105. const int src3= src[3 *srcStride];\
  106. const int src4= src[4 *srcStride];\
  107. const int src5= src[5 *srcStride];\
  108. const int src6= src[6 *srcStride];\
  109. const int src7= src[7 *srcStride];\
  110. const int src8= src[8 *srcStride];\
  111. const int src9= src[9 *srcStride];\
  112. OP(dst[0*dstStride], 6*src0 + 9*src1 + src2);\
  113. OP(dst[1*dstStride], 6*src1 + 9*src2 + src3);\
  114. OP(dst[2*dstStride], 6*src2 + 9*src3 + src4);\
  115. OP(dst[3*dstStride], 6*src3 + 9*src4 + src5);\
  116. OP(dst[4*dstStride], 6*src4 + 9*src5 + src6);\
  117. OP(dst[5*dstStride], 6*src5 + 9*src6 + src7);\
  118. OP(dst[6*dstStride], 6*src6 + 9*src7 + src8);\
  119. OP(dst[7*dstStride], 6*src7 + 9*src8 + src9);\
  120. dst ++;\
  121. src ++;\
  122. }\
  123. }\
  124. \
  125. static void OPNAME ## rv30_tpel8_hv_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  126. uint8_t half[8*10];\
  127. put_rv30_tpel8_h3_lowpass(half, src, 8, srcStride);\
  128. OPNAME ## rv30_tpel8_v3_lowpass(dst, half, dstStride, 8);\
  129. }\
  130. \
  131. static void OPNAME ## rv30_tpel16_v_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, const int C1, const int C2){\
  132. OPNAME ## rv30_tpel8_v_lowpass(dst , src , dstStride, srcStride, C1, C2);\
  133. OPNAME ## rv30_tpel8_v_lowpass(dst+8, src+8, dstStride, srcStride, C1, C2);\
  134. src += 8*srcStride;\
  135. dst += 8*dstStride;\
  136. OPNAME ## rv30_tpel8_v_lowpass(dst , src , dstStride, srcStride, C1, C2);\
  137. OPNAME ## rv30_tpel8_v_lowpass(dst+8, src+8, dstStride, srcStride, C1, C2);\
  138. }\
  139. \
  140. static void OPNAME ## rv30_tpel16_h_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, const int C1, const int C2){\
  141. OPNAME ## rv30_tpel8_h_lowpass(dst , src , dstStride, srcStride, C1, C2);\
  142. OPNAME ## rv30_tpel8_h_lowpass(dst+8, src+8, dstStride, srcStride, C1, C2);\
  143. src += 8*srcStride;\
  144. dst += 8*dstStride;\
  145. OPNAME ## rv30_tpel8_h_lowpass(dst , src , dstStride, srcStride, C1, C2);\
  146. OPNAME ## rv30_tpel8_h_lowpass(dst+8, src+8, dstStride, srcStride, C1, C2);\
  147. }\
  148. \
  149. static void OPNAME ## rv30_tpel16_hv_lowpass(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  150. OPNAME ## rv30_tpel8_hv_lowpass(dst , src , dstStride, srcStride);\
  151. OPNAME ## rv30_tpel8_hv_lowpass(dst+8, src+8, dstStride, srcStride);\
  152. src += 8*srcStride;\
  153. dst += 8*dstStride;\
  154. OPNAME ## rv30_tpel8_hv_lowpass(dst , src , dstStride, srcStride);\
  155. OPNAME ## rv30_tpel8_hv_lowpass(dst+8, src+8, dstStride, srcStride);\
  156. }\
  157. \
  158. #define RV30_MC(OPNAME, SIZE) \
  159. static void OPNAME ## rv30_tpel ## SIZE ## _mc10_c(uint8_t *dst, uint8_t *src, int stride){\
  160. OPNAME ## rv30_tpel ## SIZE ## _h_lowpass(dst, src, stride, stride, 12, 6);\
  161. }\
  162. \
  163. static void OPNAME ## rv30_tpel ## SIZE ## _mc20_c(uint8_t *dst, uint8_t *src, int stride){\
  164. OPNAME ## rv30_tpel ## SIZE ## _h_lowpass(dst, src, stride, stride, 6, 12);\
  165. }\
  166. \
  167. static void OPNAME ## rv30_tpel ## SIZE ## _mc01_c(uint8_t *dst, uint8_t *src, int stride){\
  168. OPNAME ## rv30_tpel ## SIZE ## _v_lowpass(dst, src, stride, stride, 12, 6);\
  169. }\
  170. \
  171. static void OPNAME ## rv30_tpel ## SIZE ## _mc02_c(uint8_t *dst, uint8_t *src, int stride){\
  172. OPNAME ## rv30_tpel ## SIZE ## _v_lowpass(dst, src, stride, stride, 6, 12);\
  173. }\
  174. \
  175. static void OPNAME ## rv30_tpel ## SIZE ## _mc11_c(uint8_t *dst, uint8_t *src, int stride){\
  176. uint8_t half[SIZE*SIZE];\
  177. put_rv30_tpel ## SIZE ## _h_lowpass(half, src, SIZE, stride, 12, 6);\
  178. OPNAME ## rv30_tpel ## SIZE ## _v_lowpass(dst, src, stride, stride, 12, 6);\
  179. }\
  180. \
  181. static void OPNAME ## rv30_tpel ## SIZE ## _mc12_c(uint8_t *dst, uint8_t *src, int stride){\
  182. uint8_t half[SIZE*SIZE];\
  183. put_rv30_tpel ## SIZE ## _h_lowpass(half, src, SIZE, stride, 12, 6);\
  184. OPNAME ## rv30_tpel ## SIZE ## _v_lowpass(dst, src, stride, stride, 6, 12);\
  185. }\
  186. \
  187. static void OPNAME ## rv30_tpel ## SIZE ## _mc21_c(uint8_t *dst, uint8_t *src, int stride){\
  188. uint8_t half[SIZE*SIZE];\
  189. put_rv30_tpel ## SIZE ## _h_lowpass(half, src, SIZE, stride, 6, 12);\
  190. OPNAME ## rv30_tpel ## SIZE ## _v_lowpass(dst, src, stride, stride, 12, 6);\
  191. }\
  192. \
  193. static void OPNAME ## rv30_tpel ## SIZE ## _mc22_c(uint8_t *dst, uint8_t *src, int stride){\
  194. OPNAME ## rv30_tpel ## SIZE ## _hv_lowpass(dst, src, stride, stride);\
  195. }\
  196. \
  197. #define op_avg(a, b) a = (((a)+cm[((b) + 8)>>4]+1)>>1)
  198. #define op_put(a, b) a = cm[((b) + 8)>>4]
  199. RV30_LOWPASS(put_ , op_put)
  200. RV30_LOWPASS(avg_ , op_avg)
  201. RV30_MC(put_, 8)
  202. RV30_MC(put_, 16)
  203. RV30_MC(avg_, 8)
  204. RV30_MC(avg_, 16)
  205. void ff_rv30dsp_init(DSPContext* c, AVCodecContext *avctx) {
  206. c->put_rv30_tpel_pixels_tab[0][ 0] = c->put_h264_qpel_pixels_tab[0][0];
  207. c->put_rv30_tpel_pixels_tab[0][ 1] = put_rv30_tpel16_mc10_c;
  208. c->put_rv30_tpel_pixels_tab[0][ 2] = put_rv30_tpel16_mc20_c;
  209. c->put_rv30_tpel_pixels_tab[0][ 4] = put_rv30_tpel16_mc01_c;
  210. c->put_rv30_tpel_pixels_tab[0][ 5] = put_rv30_tpel16_mc11_c;
  211. c->put_rv30_tpel_pixels_tab[0][ 6] = put_rv30_tpel16_mc21_c;
  212. c->put_rv30_tpel_pixels_tab[0][ 8] = put_rv30_tpel16_mc02_c;
  213. c->put_rv30_tpel_pixels_tab[0][ 9] = put_rv30_tpel16_mc12_c;
  214. c->put_rv30_tpel_pixels_tab[0][10] = put_rv30_tpel16_mc22_c;
  215. c->avg_rv30_tpel_pixels_tab[0][ 0] = c->avg_h264_qpel_pixels_tab[0][0];
  216. c->avg_rv30_tpel_pixels_tab[0][ 1] = avg_rv30_tpel16_mc10_c;
  217. c->avg_rv30_tpel_pixels_tab[0][ 2] = avg_rv30_tpel16_mc20_c;
  218. c->avg_rv30_tpel_pixels_tab[0][ 4] = avg_rv30_tpel16_mc01_c;
  219. c->avg_rv30_tpel_pixels_tab[0][ 5] = avg_rv30_tpel16_mc11_c;
  220. c->avg_rv30_tpel_pixels_tab[0][ 6] = avg_rv30_tpel16_mc21_c;
  221. c->avg_rv30_tpel_pixels_tab[0][ 8] = avg_rv30_tpel16_mc02_c;
  222. c->avg_rv30_tpel_pixels_tab[0][ 9] = avg_rv30_tpel16_mc12_c;
  223. c->avg_rv30_tpel_pixels_tab[0][10] = avg_rv30_tpel16_mc22_c;
  224. c->put_rv30_tpel_pixels_tab[1][ 0] = c->put_h264_qpel_pixels_tab[1][0];
  225. c->put_rv30_tpel_pixels_tab[1][ 1] = put_rv30_tpel8_mc10_c;
  226. c->put_rv30_tpel_pixels_tab[1][ 2] = put_rv30_tpel8_mc20_c;
  227. c->put_rv30_tpel_pixels_tab[1][ 4] = put_rv30_tpel8_mc01_c;
  228. c->put_rv30_tpel_pixels_tab[1][ 5] = put_rv30_tpel8_mc11_c;
  229. c->put_rv30_tpel_pixels_tab[1][ 6] = put_rv30_tpel8_mc21_c;
  230. c->put_rv30_tpel_pixels_tab[1][ 8] = put_rv30_tpel8_mc02_c;
  231. c->put_rv30_tpel_pixels_tab[1][ 9] = put_rv30_tpel8_mc12_c;
  232. c->put_rv30_tpel_pixels_tab[1][10] = put_rv30_tpel8_mc22_c;
  233. c->avg_rv30_tpel_pixels_tab[1][ 0] = c->avg_h264_qpel_pixels_tab[1][0];
  234. c->avg_rv30_tpel_pixels_tab[1][ 1] = avg_rv30_tpel8_mc10_c;
  235. c->avg_rv30_tpel_pixels_tab[1][ 2] = avg_rv30_tpel8_mc20_c;
  236. c->avg_rv30_tpel_pixels_tab[1][ 4] = avg_rv30_tpel8_mc01_c;
  237. c->avg_rv30_tpel_pixels_tab[1][ 5] = avg_rv30_tpel8_mc11_c;
  238. c->avg_rv30_tpel_pixels_tab[1][ 6] = avg_rv30_tpel8_mc21_c;
  239. c->avg_rv30_tpel_pixels_tab[1][ 8] = avg_rv30_tpel8_mc02_c;
  240. c->avg_rv30_tpel_pixels_tab[1][ 9] = avg_rv30_tpel8_mc12_c;
  241. c->avg_rv30_tpel_pixels_tab[1][10] = avg_rv30_tpel8_mc22_c;
  242. }