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.

230 lines
13KB

  1. /*
  2. * Copyright (c) 2010 Jason Garrett-Glaser
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg 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. * FFmpeg 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 FFmpeg; 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, const int bit_depth)
  96. {
  97. int mm_flags = av_get_cpu_flags();
  98. const int high_depth = bit_depth > 8;
  99. if (high_depth)
  100. return;
  101. #if HAVE_YASM
  102. if (mm_flags & AV_CPU_FLAG_MMX) {
  103. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_mmx;
  104. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmx;
  105. h->pred8x8 [VERT_PRED8x8] = ff_pred8x8_vertical_mmx;
  106. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmx;
  107. if (codec_id == CODEC_ID_VP8) {
  108. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmx;
  109. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmx;
  110. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmx;
  111. } else {
  112. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx;
  113. if (codec_id == CODEC_ID_SVQ3) {
  114. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx;
  115. } else if (codec_id == CODEC_ID_RV40) {
  116. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx;
  117. } else {
  118. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx;
  119. }
  120. }
  121. }
  122. if (mm_flags & AV_CPU_FLAG_MMX2) {
  123. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmxext;
  124. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_mmxext;
  125. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmxext;
  126. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_mmxext;
  127. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_mmxext;
  128. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_mmxext;
  129. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_mmxext;
  130. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_mmxext;
  131. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_mmxext;
  132. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_mmxext;
  133. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_mmxext;
  134. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_mmxext;
  135. h->pred4x4 [DIAG_DOWN_RIGHT_PRED] = ff_pred4x4_down_right_mmxext;
  136. h->pred4x4 [VERT_RIGHT_PRED ] = ff_pred4x4_vertical_right_mmxext;
  137. h->pred4x4 [HOR_DOWN_PRED ] = ff_pred4x4_horizontal_down_mmxext;
  138. h->pred4x4 [DC_PRED ] = ff_pred4x4_dc_mmxext;
  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. if (codec_id == CODEC_ID_VP8) {
  151. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmxext;
  152. h->pred8x8 [DC_PRED8x8 ] = ff_pred8x8_dc_rv40_mmxext;
  153. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmxext;
  154. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmxext;
  155. h->pred4x4 [VERT_PRED ] = ff_pred4x4_vertical_vp8_mmxext;
  156. } else {
  157. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx2;
  158. if (codec_id == CODEC_ID_SVQ3) {
  159. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx2;
  160. } else if (codec_id == CODEC_ID_RV40) {
  161. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx2;
  162. } else {
  163. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx2;
  164. }
  165. }
  166. }
  167. if (mm_flags & AV_CPU_FLAG_SSE) {
  168. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_sse;
  169. }
  170. if (mm_flags & AV_CPU_FLAG_SSE2) {
  171. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_sse2;
  172. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_sse2;
  173. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_sse2;
  174. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_sse2;
  175. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_sse2;
  176. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_sse2;
  177. if (codec_id == CODEC_ID_VP8) {
  178. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_sse2;
  179. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_sse2;
  180. } else {
  181. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_sse2;
  182. if (codec_id == CODEC_ID_SVQ3) {
  183. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_sse2;
  184. } else if (codec_id == CODEC_ID_RV40) {
  185. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_sse2;
  186. } else {
  187. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_sse2;
  188. }
  189. }
  190. }
  191. if (mm_flags & AV_CPU_FLAG_SSSE3) {
  192. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_ssse3;
  193. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_ssse3;
  194. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_ssse3;
  195. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_ssse3;
  196. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_ssse3;
  197. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_ssse3;
  198. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_ssse3;
  199. h->pred8x8l [DIAG_DOWN_LEFT_PRED ] = ff_pred8x8l_down_left_ssse3;
  200. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_ssse3;
  201. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_ssse3;
  202. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_ssse3;
  203. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_ssse3;
  204. h->pred8x8l [HOR_DOWN_PRED ] = ff_pred8x8l_horizontal_down_ssse3;
  205. if (codec_id == CODEC_ID_VP8) {
  206. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_ssse3;
  207. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_ssse3;
  208. } else {
  209. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_ssse3;
  210. if (codec_id == CODEC_ID_SVQ3) {
  211. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_ssse3;
  212. } else if (codec_id == CODEC_ID_RV40) {
  213. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_ssse3;
  214. } else {
  215. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_ssse3;
  216. }
  217. }
  218. }
  219. #endif
  220. }