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.

234 lines
13KB

  1. /*
  2. * Copyright (c) 2010 Jason Garrett-Glaser
  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/cpu.h"
  21. #include "libavcodec/h264pred.h"
  22. void ff_pred16x16_vertical_mmx (uint8_t *src, int stride);
  23. void ff_pred16x16_vertical_sse (uint8_t *src, int stride);
  24. void ff_pred16x16_horizontal_mmx (uint8_t *src, int stride);
  25. void ff_pred16x16_horizontal_mmxext(uint8_t *src, int stride);
  26. void ff_pred16x16_horizontal_ssse3 (uint8_t *src, int stride);
  27. void ff_pred16x16_dc_mmxext (uint8_t *src, int stride);
  28. void ff_pred16x16_dc_sse2 (uint8_t *src, int stride);
  29. void ff_pred16x16_dc_ssse3 (uint8_t *src, int stride);
  30. void ff_pred16x16_plane_h264_mmx (uint8_t *src, int stride);
  31. void ff_pred16x16_plane_h264_mmx2 (uint8_t *src, int stride);
  32. void ff_pred16x16_plane_h264_sse2 (uint8_t *src, int stride);
  33. void ff_pred16x16_plane_h264_ssse3 (uint8_t *src, int stride);
  34. void ff_pred16x16_plane_rv40_mmx (uint8_t *src, int stride);
  35. void ff_pred16x16_plane_rv40_mmx2 (uint8_t *src, int stride);
  36. void ff_pred16x16_plane_rv40_sse2 (uint8_t *src, int stride);
  37. void ff_pred16x16_plane_rv40_ssse3 (uint8_t *src, int stride);
  38. void ff_pred16x16_plane_svq3_mmx (uint8_t *src, int stride);
  39. void ff_pred16x16_plane_svq3_mmx2 (uint8_t *src, int stride);
  40. void ff_pred16x16_plane_svq3_sse2 (uint8_t *src, int stride);
  41. void ff_pred16x16_plane_svq3_ssse3 (uint8_t *src, int stride);
  42. void ff_pred16x16_tm_vp8_mmx (uint8_t *src, int stride);
  43. void ff_pred16x16_tm_vp8_mmxext (uint8_t *src, int stride);
  44. void ff_pred16x16_tm_vp8_sse2 (uint8_t *src, int stride);
  45. void ff_pred8x8_top_dc_mmxext (uint8_t *src, int stride);
  46. void ff_pred8x8_dc_rv40_mmxext (uint8_t *src, int stride);
  47. void ff_pred8x8_dc_mmxext (uint8_t *src, int stride);
  48. void ff_pred8x8_vertical_mmx (uint8_t *src, int stride);
  49. void ff_pred8x8_horizontal_mmx (uint8_t *src, int stride);
  50. void ff_pred8x8_horizontal_mmxext (uint8_t *src, int stride);
  51. void ff_pred8x8_horizontal_ssse3 (uint8_t *src, int stride);
  52. void ff_pred8x8_plane_mmx (uint8_t *src, int stride);
  53. void ff_pred8x8_plane_mmx2 (uint8_t *src, int stride);
  54. void ff_pred8x8_plane_sse2 (uint8_t *src, int stride);
  55. void ff_pred8x8_plane_ssse3 (uint8_t *src, int stride);
  56. void ff_pred8x8_tm_vp8_mmx (uint8_t *src, int stride);
  57. void ff_pred8x8_tm_vp8_mmxext (uint8_t *src, int stride);
  58. void ff_pred8x8_tm_vp8_sse2 (uint8_t *src, int stride);
  59. void ff_pred8x8_tm_vp8_ssse3 (uint8_t *src, int stride);
  60. void ff_pred8x8l_top_dc_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  61. void ff_pred8x8l_top_dc_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  62. void ff_pred8x8l_dc_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  63. void ff_pred8x8l_dc_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  64. void ff_pred8x8l_horizontal_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  65. void ff_pred8x8l_horizontal_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  66. void ff_pred8x8l_vertical_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  67. void ff_pred8x8l_vertical_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  68. void ff_pred8x8l_down_left_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  69. void ff_pred8x8l_down_left_sse2 (uint8_t *src, int has_topleft, int has_topright, int stride);
  70. void ff_pred8x8l_down_left_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  71. void ff_pred8x8l_down_right_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  72. void ff_pred8x8l_down_right_sse2 (uint8_t *src, int has_topleft, int has_topright, int stride);
  73. void ff_pred8x8l_down_right_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  74. void ff_pred8x8l_vertical_right_mmxext(uint8_t *src, int has_topleft, int has_topright, int stride);
  75. void ff_pred8x8l_vertical_right_sse2(uint8_t *src, int has_topleft, int has_topright, int stride);
  76. void ff_pred8x8l_vertical_right_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  77. void ff_pred8x8l_vertical_left_sse2(uint8_t *src, int has_topleft, int has_topright, int stride);
  78. void ff_pred8x8l_vertical_left_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  79. void ff_pred8x8l_horizontal_up_mmxext(uint8_t *src, int has_topleft, int has_topright, int stride);
  80. void ff_pred8x8l_horizontal_up_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  81. void ff_pred8x8l_horizontal_down_mmxext(uint8_t *src, int has_topleft, int has_topright, int stride);
  82. void ff_pred8x8l_horizontal_down_sse2(uint8_t *src, int has_topleft, int has_topright, int stride);
  83. void ff_pred8x8l_horizontal_down_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  84. void ff_pred4x4_dc_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  85. void ff_pred4x4_down_left_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  86. void ff_pred4x4_down_right_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  87. void ff_pred4x4_vertical_left_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  88. void ff_pred4x4_vertical_right_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  89. void ff_pred4x4_horizontal_up_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  90. void ff_pred4x4_horizontal_down_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  91. void ff_pred4x4_tm_vp8_mmx (uint8_t *src, const uint8_t *topright, int stride);
  92. void ff_pred4x4_tm_vp8_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  93. void ff_pred4x4_tm_vp8_ssse3 (uint8_t *src, const uint8_t *topright, int stride);
  94. void ff_pred4x4_vertical_vp8_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  95. void ff_h264_pred_init_x86(H264PredContext *h, int codec_id)
  96. {
  97. int mm_flags = av_get_cpu_flags();
  98. #if HAVE_YASM
  99. if (mm_flags & AV_CPU_FLAG_MMX) {
  100. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_mmx;
  101. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmx;
  102. h->pred8x8 [VERT_PRED8x8] = ff_pred8x8_vertical_mmx;
  103. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmx;
  104. if (codec_id == CODEC_ID_VP8) {
  105. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmx;
  106. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmx;
  107. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmx;
  108. } else {
  109. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx;
  110. if (codec_id == CODEC_ID_SVQ3) {
  111. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx;
  112. } else if (codec_id == CODEC_ID_RV40) {
  113. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx;
  114. } else {
  115. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx;
  116. }
  117. }
  118. }
  119. if (mm_flags & AV_CPU_FLAG_MMX2) {
  120. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmxext;
  121. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_mmxext;
  122. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmxext;
  123. #if CONFIG_GPL
  124. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_mmxext;
  125. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_mmxext;
  126. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_mmxext;
  127. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_mmxext;
  128. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_mmxext;
  129. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_mmxext;
  130. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_mmxext;
  131. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_mmxext;
  132. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_mmxext;
  133. h->pred4x4 [DIAG_DOWN_RIGHT_PRED] = ff_pred4x4_down_right_mmxext;
  134. h->pred4x4 [VERT_RIGHT_PRED ] = ff_pred4x4_vertical_right_mmxext;
  135. h->pred4x4 [HOR_DOWN_PRED ] = ff_pred4x4_horizontal_down_mmxext;
  136. #endif
  137. h->pred4x4 [DC_PRED ] = ff_pred4x4_dc_mmxext;
  138. #if CONFIG_GPL
  139. if (codec_id == CODEC_ID_VP8 || codec_id == CODEC_ID_H264)
  140. h->pred4x4 [DIAG_DOWN_LEFT_PRED ] = ff_pred4x4_down_left_mmxext;
  141. if (codec_id == CODEC_ID_SVQ3 || codec_id == CODEC_ID_H264)
  142. h->pred4x4 [VERT_LEFT_PRED ] = ff_pred4x4_vertical_left_mmxext;
  143. if (codec_id != CODEC_ID_RV40) {
  144. h->pred4x4 [HOR_UP_PRED ] = ff_pred4x4_horizontal_up_mmxext;
  145. }
  146. if (codec_id == CODEC_ID_SVQ3 || codec_id == CODEC_ID_H264) {
  147. h->pred8x8 [TOP_DC_PRED8x8 ] = ff_pred8x8_top_dc_mmxext;
  148. h->pred8x8 [DC_PRED8x8 ] = ff_pred8x8_dc_mmxext;
  149. }
  150. #endif
  151. if (codec_id == CODEC_ID_VP8) {
  152. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmxext;
  153. h->pred8x8 [DC_PRED8x8 ] = ff_pred8x8_dc_rv40_mmxext;
  154. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmxext;
  155. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmxext;
  156. h->pred4x4 [VERT_PRED ] = ff_pred4x4_vertical_vp8_mmxext;
  157. } else {
  158. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx2;
  159. if (codec_id == CODEC_ID_SVQ3) {
  160. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx2;
  161. } else if (codec_id == CODEC_ID_RV40) {
  162. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx2;
  163. } else {
  164. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx2;
  165. }
  166. }
  167. }
  168. if (mm_flags & AV_CPU_FLAG_SSE) {
  169. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_sse;
  170. }
  171. if (mm_flags & AV_CPU_FLAG_SSE2) {
  172. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_sse2;
  173. #if CONFIG_GPL
  174. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_sse2;
  175. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_sse2;
  176. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_sse2;
  177. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_sse2;
  178. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_sse2;
  179. #endif
  180. if (codec_id == CODEC_ID_VP8) {
  181. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_sse2;
  182. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_sse2;
  183. } else {
  184. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_sse2;
  185. if (codec_id == CODEC_ID_SVQ3) {
  186. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_sse2;
  187. } else if (codec_id == CODEC_ID_RV40) {
  188. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_sse2;
  189. } else {
  190. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_sse2;
  191. }
  192. }
  193. }
  194. if (mm_flags & AV_CPU_FLAG_SSSE3) {
  195. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_ssse3;
  196. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_ssse3;
  197. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_ssse3;
  198. #if CONFIG_GPL
  199. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_ssse3;
  200. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_ssse3;
  201. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_ssse3;
  202. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_ssse3;
  203. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_ssse3;
  204. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_ssse3;
  205. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_ssse3;
  206. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_ssse3;
  207. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_ssse3;
  208. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_ssse3;
  209. #endif
  210. if (codec_id == CODEC_ID_VP8) {
  211. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_ssse3;
  212. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_ssse3;
  213. } else {
  214. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_ssse3;
  215. if (codec_id == CODEC_ID_SVQ3) {
  216. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_ssse3;
  217. } else if (codec_id == CODEC_ID_RV40) {
  218. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_ssse3;
  219. } else {
  220. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_ssse3;
  221. }
  222. }
  223. }
  224. #endif
  225. }