yuv2rgb.c 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. /*
  2. * software YUV to RGB converter
  3. *
  4. * Copyright (C) 2009 Konstantin Shishkov
  5. *
  6. * MMX/MMX2 template stuff (needed for fast movntq support),
  7. * 1,4,8bpp support and context / deglobalize stuff
  8. * by Michael Niedermayer (michaelni@gmx.at)
  9. *
  10. * This file is part of FFmpeg.
  11. *
  12. * FFmpeg is free software; you can redistribute it and/or
  13. * modify it under the terms of the GNU Lesser General Public
  14. * License as published by the Free Software Foundation; either
  15. * version 2.1 of the License, or (at your option) any later version.
  16. *
  17. * FFmpeg is distributed in the hope that it will be useful,
  18. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  20. * Lesser General Public License for more details.
  21. *
  22. * You should have received a copy of the GNU Lesser General Public
  23. * License along with FFmpeg; if not, write to the Free Software
  24. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  25. */
  26. #include <stdio.h>
  27. #include <stdlib.h>
  28. #include <inttypes.h>
  29. #include <assert.h>
  30. #include "config.h"
  31. #include "libswscale/rgb2rgb.h"
  32. #include "libswscale/swscale.h"
  33. #include "libswscale/swscale_internal.h"
  34. #include "libavutil/attributes.h"
  35. #include "libavutil/x86/asm.h"
  36. #include "libavutil/cpu.h"
  37. #if HAVE_INLINE_ASM
  38. #define DITHER1XBPP // only for MMX
  39. /* hope these constant values are cache line aligned */
  40. DECLARE_ASM_CONST(8, uint64_t, mmx_00ffw) = 0x00ff00ff00ff00ffULL;
  41. DECLARE_ASM_CONST(8, uint64_t, mmx_redmask) = 0xf8f8f8f8f8f8f8f8ULL;
  42. DECLARE_ASM_CONST(8, uint64_t, mmx_grnmask) = 0xfcfcfcfcfcfcfcfcULL;
  43. DECLARE_ASM_CONST(8, uint64_t, pb_e0) = 0xe0e0e0e0e0e0e0e0ULL;
  44. DECLARE_ASM_CONST(8, uint64_t, pb_03) = 0x0303030303030303ULL;
  45. DECLARE_ASM_CONST(8, uint64_t, pb_07) = 0x0707070707070707ULL;
  46. //MMX versions
  47. #if HAVE_MMX_INLINE
  48. #undef RENAME
  49. #undef COMPILE_TEMPLATE_MMXEXT
  50. #define COMPILE_TEMPLATE_MMXEXT 0
  51. #define RENAME(a) a ## _MMX
  52. #include "yuv2rgb_template.c"
  53. #endif /* HAVE_MMX_INLINE */
  54. //MMX2 versions
  55. #if HAVE_MMXEXT_INLINE
  56. #undef RENAME
  57. #undef COMPILE_TEMPLATE_MMXEXT
  58. #define COMPILE_TEMPLATE_MMXEXT 1
  59. #define RENAME(a) a ## _MMX2
  60. #include "yuv2rgb_template.c"
  61. #endif /* HAVE_MMXEXT_INLINE */
  62. #endif /* HAVE_INLINE_ASM */
  63. av_cold SwsFunc ff_yuv2rgb_init_mmx(SwsContext *c)
  64. {
  65. #if HAVE_INLINE_ASM
  66. int cpu_flags = av_get_cpu_flags();
  67. #if HAVE_MMXEXT_INLINE
  68. if (cpu_flags & AV_CPU_FLAG_MMXEXT) {
  69. switch (c->dstFormat) {
  70. case PIX_FMT_RGB24: return yuv420_rgb24_MMX2;
  71. case PIX_FMT_BGR24: return yuv420_bgr24_MMX2;
  72. }
  73. }
  74. #endif
  75. if (cpu_flags & AV_CPU_FLAG_MMX) {
  76. switch (c->dstFormat) {
  77. case PIX_FMT_RGB32:
  78. if (c->srcFormat == PIX_FMT_YUVA420P) {
  79. #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
  80. return yuva420_rgb32_MMX;
  81. #endif
  82. break;
  83. } else return yuv420_rgb32_MMX;
  84. case PIX_FMT_BGR32:
  85. if (c->srcFormat == PIX_FMT_YUVA420P) {
  86. #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
  87. return yuva420_bgr32_MMX;
  88. #endif
  89. break;
  90. } else return yuv420_bgr32_MMX;
  91. case PIX_FMT_RGB24: return yuv420_rgb24_MMX;
  92. case PIX_FMT_BGR24: return yuv420_bgr24_MMX;
  93. case PIX_FMT_RGB565: return yuv420_rgb16_MMX;
  94. case PIX_FMT_RGB555: return yuv420_rgb15_MMX;
  95. }
  96. }
  97. #endif /* HAVE_INLINE_ASM */
  98. return NULL;
  99. }