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.

213 lines
11KB

  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_sse2 (uint8_t *src, int has_topleft, int has_topright, int stride);
  69. void ff_pred8x8l_down_left_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  70. void ff_pred8x8l_down_right_mmxext (uint8_t *src, int has_topleft, int has_topright, int stride);
  71. void ff_pred8x8l_down_right_sse2 (uint8_t *src, int has_topleft, int has_topright, int stride);
  72. void ff_pred8x8l_down_right_ssse3 (uint8_t *src, int has_topleft, int has_topright, int stride);
  73. void ff_pred8x8l_vertical_right_mmxext(uint8_t *src, int has_topleft, int has_topright, int stride);
  74. void ff_pred8x8l_vertical_right_sse2(uint8_t *src, int has_topleft, int has_topright, int stride);
  75. void ff_pred8x8l_vertical_right_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  76. void ff_pred8x8l_vertical_left_sse2(uint8_t *src, int has_topleft, int has_topright, int stride);
  77. void ff_pred8x8l_vertical_left_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  78. void ff_pred8x8l_horizontal_up_mmxext(uint8_t *src, int has_topleft, int has_topright, int stride);
  79. void ff_pred8x8l_horizontal_up_ssse3(uint8_t *src, int has_topleft, int has_topright, int stride);
  80. void ff_pred4x4_dc_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  81. void ff_pred4x4_down_left_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  82. void ff_pred4x4_tm_vp8_mmx (uint8_t *src, const uint8_t *topright, int stride);
  83. void ff_pred4x4_tm_vp8_mmxext (uint8_t *src, const uint8_t *topright, int stride);
  84. void ff_pred4x4_tm_vp8_ssse3 (uint8_t *src, const uint8_t *topright, int stride);
  85. void ff_pred4x4_vertical_vp8_mmxext(uint8_t *src, const uint8_t *topright, int stride);
  86. void ff_h264_pred_init_x86(H264PredContext *h, int codec_id)
  87. {
  88. int mm_flags = av_get_cpu_flags();
  89. #if HAVE_YASM
  90. if (mm_flags & AV_CPU_FLAG_MMX) {
  91. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_mmx;
  92. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmx;
  93. h->pred8x8 [VERT_PRED8x8] = ff_pred8x8_vertical_mmx;
  94. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmx;
  95. if (codec_id == CODEC_ID_VP8) {
  96. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmx;
  97. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmx;
  98. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmx;
  99. } else {
  100. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx;
  101. if (codec_id == CODEC_ID_SVQ3) {
  102. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx;
  103. } else if (codec_id == CODEC_ID_RV40) {
  104. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx;
  105. } else {
  106. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx;
  107. }
  108. }
  109. }
  110. if (mm_flags & AV_CPU_FLAG_MMX2) {
  111. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_mmxext;
  112. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_mmxext;
  113. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_mmxext;
  114. #if CONFIG_GPL
  115. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_mmxext;
  116. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_mmxext;
  117. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_mmxext;
  118. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_mmxext;
  119. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_mmxext;
  120. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_mmxext;
  121. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_mmxext;
  122. #endif
  123. h->pred4x4 [DC_PRED ] = ff_pred4x4_dc_mmxext;
  124. #if CONFIG_GPL
  125. if (codec_id == CODEC_ID_VP8 || codec_id == CODEC_ID_H264)
  126. h->pred4x4 [DIAG_DOWN_LEFT_PRED ] = ff_pred4x4_down_left_mmxext;
  127. if (codec_id == CODEC_ID_SVQ3 || codec_id == CODEC_ID_H264) {
  128. h->pred8x8 [TOP_DC_PRED8x8 ] = ff_pred8x8_top_dc_mmxext;
  129. h->pred8x8 [DC_PRED8x8 ] = ff_pred8x8_dc_mmxext;
  130. }
  131. #endif
  132. if (codec_id == CODEC_ID_VP8) {
  133. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_mmxext;
  134. h->pred8x8 [DC_PRED8x8 ] = ff_pred8x8_dc_rv40_mmxext;
  135. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_mmxext;
  136. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_mmxext;
  137. h->pred4x4 [VERT_PRED ] = ff_pred4x4_vertical_vp8_mmxext;
  138. } else {
  139. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_mmx2;
  140. if (codec_id == CODEC_ID_SVQ3) {
  141. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_mmx2;
  142. } else if (codec_id == CODEC_ID_RV40) {
  143. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_mmx2;
  144. } else {
  145. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_mmx2;
  146. }
  147. }
  148. }
  149. if (mm_flags & AV_CPU_FLAG_SSE) {
  150. h->pred16x16[VERT_PRED8x8] = ff_pred16x16_vertical_sse;
  151. }
  152. if (mm_flags & AV_CPU_FLAG_SSE2) {
  153. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_sse2;
  154. #if CONFIG_GPL
  155. h->pred8x8l [DIAG_DOWN_LEFT_PRED] = ff_pred8x8l_down_left_sse2;
  156. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_sse2;
  157. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_sse2;
  158. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_sse2;
  159. #endif
  160. if (codec_id == CODEC_ID_VP8) {
  161. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_tm_vp8_sse2;
  162. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_sse2;
  163. } else {
  164. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_sse2;
  165. if (codec_id == CODEC_ID_SVQ3) {
  166. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_sse2;
  167. } else if (codec_id == CODEC_ID_RV40) {
  168. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_sse2;
  169. } else {
  170. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_sse2;
  171. }
  172. }
  173. }
  174. if (mm_flags & AV_CPU_FLAG_SSSE3) {
  175. h->pred16x16[HOR_PRED8x8 ] = ff_pred16x16_horizontal_ssse3;
  176. h->pred16x16[DC_PRED8x8 ] = ff_pred16x16_dc_ssse3;
  177. h->pred8x8 [HOR_PRED8x8 ] = ff_pred8x8_horizontal_ssse3;
  178. #if CONFIG_GPL
  179. h->pred8x8l [TOP_DC_PRED ] = ff_pred8x8l_top_dc_ssse3;
  180. h->pred8x8l [DC_PRED ] = ff_pred8x8l_dc_ssse3;
  181. h->pred8x8l [HOR_PRED ] = ff_pred8x8l_horizontal_ssse3;
  182. h->pred8x8l [VERT_PRED ] = ff_pred8x8l_vertical_ssse3;
  183. h->pred8x8l [DIAG_DOWN_LEFT_PRED] = ff_pred8x8l_down_left_ssse3;
  184. h->pred8x8l [DIAG_DOWN_RIGHT_PRED] = ff_pred8x8l_down_right_ssse3;
  185. h->pred8x8l [VERT_RIGHT_PRED ] = ff_pred8x8l_vertical_right_ssse3;
  186. h->pred8x8l [VERT_LEFT_PRED ] = ff_pred8x8l_vertical_left_ssse3;
  187. h->pred8x8l [HOR_UP_PRED ] = ff_pred8x8l_horizontal_up_ssse3;
  188. #endif
  189. if (codec_id == CODEC_ID_VP8) {
  190. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_tm_vp8_ssse3;
  191. h->pred4x4 [TM_VP8_PRED ] = ff_pred4x4_tm_vp8_ssse3;
  192. } else {
  193. h->pred8x8 [PLANE_PRED8x8] = ff_pred8x8_plane_ssse3;
  194. if (codec_id == CODEC_ID_SVQ3) {
  195. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_svq3_ssse3;
  196. } else if (codec_id == CODEC_ID_RV40) {
  197. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_rv40_ssse3;
  198. } else {
  199. h->pred16x16[PLANE_PRED8x8] = ff_pred16x16_plane_h264_ssse3;
  200. }
  201. }
  202. }
  203. #endif
  204. }