yuv2rgb.c 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. /*
  2. * software YUV to RGB converter
  3. *
  4. * Copyright (C) 2009 Konstantin Shishkov
  5. *
  6. * MMX/MMXEXT 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 "config.h"
  30. #include "libswscale/rgb2rgb.h"
  31. #include "libswscale/swscale.h"
  32. #include "libswscale/swscale_internal.h"
  33. #include "libavutil/attributes.h"
  34. #include "libavutil/x86/asm.h"
  35. #include "libavutil/x86/cpu.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 && HAVE_6REGS
  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 && HAVE_6REGS */
  54. // MMXEXT versions
  55. #if HAVE_MMXEXT_INLINE && HAVE_6REGS
  56. #undef RENAME
  57. #undef COMPILE_TEMPLATE_MMXEXT
  58. #define COMPILE_TEMPLATE_MMXEXT 1
  59. #define RENAME(a) a ## _mmxext
  60. #include "yuv2rgb_template.c"
  61. #endif /* HAVE_MMXEXT_INLINE && HAVE_6REGS */
  62. #endif /* HAVE_INLINE_ASM */
  63. av_cold SwsFunc ff_yuv2rgb_init_x86(SwsContext *c)
  64. {
  65. #if HAVE_MMX_INLINE && HAVE_6REGS
  66. int cpu_flags = av_get_cpu_flags();
  67. #if HAVE_MMXEXT_INLINE
  68. if (INLINE_MMXEXT(cpu_flags)) {
  69. switch (c->dstFormat) {
  70. case AV_PIX_FMT_RGB24:
  71. return yuv420_rgb24_mmxext;
  72. case AV_PIX_FMT_BGR24:
  73. return yuv420_bgr24_mmxext;
  74. }
  75. }
  76. #endif
  77. if (INLINE_MMX(cpu_flags)) {
  78. switch (c->dstFormat) {
  79. case AV_PIX_FMT_RGB32:
  80. if (c->srcFormat == AV_PIX_FMT_YUVA420P) {
  81. #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
  82. return yuva420_rgb32_mmx;
  83. #endif
  84. break;
  85. } else
  86. return yuv420_rgb32_mmx;
  87. case AV_PIX_FMT_BGR32:
  88. if (c->srcFormat == AV_PIX_FMT_YUVA420P) {
  89. #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
  90. return yuva420_bgr32_mmx;
  91. #endif
  92. break;
  93. } else
  94. return yuv420_bgr32_mmx;
  95. case AV_PIX_FMT_RGB24:
  96. return yuv420_rgb24_mmx;
  97. case AV_PIX_FMT_BGR24:
  98. return yuv420_bgr24_mmx;
  99. case AV_PIX_FMT_RGB565:
  100. return yuv420_rgb16_mmx;
  101. case AV_PIX_FMT_RGB555:
  102. return yuv420_rgb15_mmx;
  103. }
  104. }
  105. #endif /* HAVE_MMX_INLINE && HAVE_6REGS */
  106. return NULL;
  107. }