rgb2rgb.c 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. /*
  2. * software RGB to RGB converter
  3. * pluralize by software PAL8 to RGB converter
  4. * software YUV to YUV converter
  5. * software YUV to RGB converter
  6. * Written by Nick Kurshev.
  7. * palette & YUV & runtime CPU stuff by Michael (michaelni@gmx.at)
  8. *
  9. * This file is part of FFmpeg.
  10. *
  11. * FFmpeg is free software; you can redistribute it and/or
  12. * modify it under the terms of the GNU Lesser General Public
  13. * License as published by the Free Software Foundation; either
  14. * version 2.1 of the License, or (at your option) any later version.
  15. *
  16. * FFmpeg is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19. * Lesser General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU Lesser General Public
  22. * License along with FFmpeg; if not, write to the Free Software
  23. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  24. */
  25. #include <stdint.h>
  26. #include "config.h"
  27. #include "libavutil/x86_cpu.h"
  28. #include "libavutil/cpu.h"
  29. #include "libavutil/bswap.h"
  30. #include "libswscale/rgb2rgb.h"
  31. #include "libswscale/swscale.h"
  32. #include "libswscale/swscale_internal.h"
  33. DECLARE_ASM_CONST(8, uint64_t, mmx_ff) = 0x00000000000000FFULL;
  34. DECLARE_ASM_CONST(8, uint64_t, mmx_null) = 0x0000000000000000ULL;
  35. DECLARE_ASM_CONST(8, uint64_t, mmx_one) = 0xFFFFFFFFFFFFFFFFULL;
  36. DECLARE_ASM_CONST(8, uint64_t, mask32b) = 0x000000FF000000FFULL;
  37. DECLARE_ASM_CONST(8, uint64_t, mask32g) = 0x0000FF000000FF00ULL;
  38. DECLARE_ASM_CONST(8, uint64_t, mask32r) = 0x00FF000000FF0000ULL;
  39. DECLARE_ASM_CONST(8, uint64_t, mask32a) = 0xFF000000FF000000ULL;
  40. DECLARE_ASM_CONST(8, uint64_t, mask32) = 0x00FFFFFF00FFFFFFULL;
  41. DECLARE_ASM_CONST(8, uint64_t, mask3216br) = 0x00F800F800F800F8ULL;
  42. DECLARE_ASM_CONST(8, uint64_t, mask3216g) = 0x0000FC000000FC00ULL;
  43. DECLARE_ASM_CONST(8, uint64_t, mask3215g) = 0x0000F8000000F800ULL;
  44. DECLARE_ASM_CONST(8, uint64_t, mul3216) = 0x2000000420000004ULL;
  45. DECLARE_ASM_CONST(8, uint64_t, mul3215) = 0x2000000820000008ULL;
  46. DECLARE_ASM_CONST(8, uint64_t, mask24b) = 0x00FF0000FF0000FFULL;
  47. DECLARE_ASM_CONST(8, uint64_t, mask24g) = 0xFF0000FF0000FF00ULL;
  48. DECLARE_ASM_CONST(8, uint64_t, mask24r) = 0x0000FF0000FF0000ULL;
  49. DECLARE_ASM_CONST(8, uint64_t, mask24l) = 0x0000000000FFFFFFULL;
  50. DECLARE_ASM_CONST(8, uint64_t, mask24h) = 0x0000FFFFFF000000ULL;
  51. DECLARE_ASM_CONST(8, uint64_t, mask24hh) = 0xffff000000000000ULL;
  52. DECLARE_ASM_CONST(8, uint64_t, mask24hhh) = 0xffffffff00000000ULL;
  53. DECLARE_ASM_CONST(8, uint64_t, mask24hhhh) = 0xffffffffffff0000ULL;
  54. DECLARE_ASM_CONST(8, uint64_t, mask15b) = 0x001F001F001F001FULL; /* 00000000 00011111 xxB */
  55. DECLARE_ASM_CONST(8, uint64_t, mask15rg) = 0x7FE07FE07FE07FE0ULL; /* 01111111 11100000 RGx */
  56. DECLARE_ASM_CONST(8, uint64_t, mask15s) = 0xFFE0FFE0FFE0FFE0ULL;
  57. DECLARE_ASM_CONST(8, uint64_t, mask15g) = 0x03E003E003E003E0ULL;
  58. DECLARE_ASM_CONST(8, uint64_t, mask15r) = 0x7C007C007C007C00ULL;
  59. #define mask16b mask15b
  60. DECLARE_ASM_CONST(8, uint64_t, mask16g) = 0x07E007E007E007E0ULL;
  61. DECLARE_ASM_CONST(8, uint64_t, mask16r) = 0xF800F800F800F800ULL;
  62. DECLARE_ASM_CONST(8, uint64_t, red_16mask) = 0x0000f8000000f800ULL;
  63. DECLARE_ASM_CONST(8, uint64_t, green_16mask) = 0x000007e0000007e0ULL;
  64. DECLARE_ASM_CONST(8, uint64_t, blue_16mask) = 0x0000001f0000001fULL;
  65. DECLARE_ASM_CONST(8, uint64_t, red_15mask) = 0x00007c0000007c00ULL;
  66. DECLARE_ASM_CONST(8, uint64_t, green_15mask) = 0x000003e0000003e0ULL;
  67. DECLARE_ASM_CONST(8, uint64_t, blue_15mask) = 0x0000001f0000001fULL;
  68. DECLARE_ASM_CONST(8, uint64_t, mul15_mid) = 0x4200420042004200ULL;
  69. DECLARE_ASM_CONST(8, uint64_t, mul15_hi) = 0x0210021002100210ULL;
  70. DECLARE_ASM_CONST(8, uint64_t, mul16_mid) = 0x2080208020802080ULL;
  71. #define RGB2YUV_SHIFT 8
  72. #define BY ((int)( 0.098*(1<<RGB2YUV_SHIFT)+0.5))
  73. #define BV ((int)(-0.071*(1<<RGB2YUV_SHIFT)+0.5))
  74. #define BU ((int)( 0.439*(1<<RGB2YUV_SHIFT)+0.5))
  75. #define GY ((int)( 0.504*(1<<RGB2YUV_SHIFT)+0.5))
  76. #define GV ((int)(-0.368*(1<<RGB2YUV_SHIFT)+0.5))
  77. #define GU ((int)(-0.291*(1<<RGB2YUV_SHIFT)+0.5))
  78. #define RY ((int)( 0.257*(1<<RGB2YUV_SHIFT)+0.5))
  79. #define RV ((int)( 0.439*(1<<RGB2YUV_SHIFT)+0.5))
  80. #define RU ((int)(-0.148*(1<<RGB2YUV_SHIFT)+0.5))
  81. //Note: We have C, MMX, MMX2, 3DNOW versions, there is no 3DNOW + MMX2 one.
  82. #define COMPILE_TEMPLATE_MMX2 0
  83. #define COMPILE_TEMPLATE_AMD3DNOW 0
  84. #define COMPILE_TEMPLATE_SSE2 0
  85. //MMX versions
  86. #undef RENAME
  87. #define RENAME(a) a ## _MMX
  88. #include "rgb2rgb_template.c"
  89. //MMX2 versions
  90. #undef RENAME
  91. #undef COMPILE_TEMPLATE_MMX2
  92. #define COMPILE_TEMPLATE_MMX2 1
  93. #define RENAME(a) a ## _MMX2
  94. #include "rgb2rgb_template.c"
  95. //SSE2 versions
  96. #undef RENAME
  97. #undef COMPILE_TEMPLATE_SSE2
  98. #define COMPILE_TEMPLATE_SSE2 1
  99. #define RENAME(a) a ## _SSE2
  100. #include "rgb2rgb_template.c"
  101. //3DNOW versions
  102. #undef RENAME
  103. #undef COMPILE_TEMPLATE_MMX2
  104. #undef COMPILE_TEMPLATE_SSE2
  105. #undef COMPILE_TEMPLATE_AMD3DNOW
  106. #define COMPILE_TEMPLATE_MMX2 0
  107. #define COMPILE_TEMPLATE_SSE2 0
  108. #define COMPILE_TEMPLATE_AMD3DNOW 1
  109. #define RENAME(a) a ## _3DNOW
  110. #include "rgb2rgb_template.c"
  111. /*
  112. RGB15->RGB16 original by Strepto/Astral
  113. ported to gcc & bugfixed : A'rpi
  114. MMX2, 3DNOW optimization by Nick Kurshev
  115. 32-bit C version, and and&add trick by Michael Niedermayer
  116. */
  117. void rgb2rgb_init_x86(void)
  118. {
  119. int cpu_flags = av_get_cpu_flags();
  120. if (cpu_flags & AV_CPU_FLAG_MMX)
  121. rgb2rgb_init_MMX();
  122. if (HAVE_AMD3DNOW && cpu_flags & AV_CPU_FLAG_3DNOW)
  123. rgb2rgb_init_3DNOW();
  124. if (HAVE_MMX2 && cpu_flags & AV_CPU_FLAG_MMX2)
  125. rgb2rgb_init_MMX2();
  126. if (HAVE_SSE && cpu_flags & AV_CPU_FLAG_SSE2)
  127. rgb2rgb_init_SSE2();
  128. }