kc3-lang/SDL/src/video/yuv2rgb/yuv_rgb_std_func.h
-
Show log
Commit
-
Hash : a6a4e27a
Author :
Date : 2017-11-12T22:51:12
Updated SDL's YUV support, many thanks to Adrien Descamps New functions get and set the YUV colorspace conversion mode: SDL_SetYUVConversionMode() SDL_GetYUVConversionMode() SDL_GetYUVConversionModeForResolution() SDL_ConvertPixels() converts between all supported RGB and YUV formats, with SSE acceleration for converting from planar YUV formats (YV12, NV12, etc) to common RGB/RGBA formats. Added a new test program, testyuv, to verify correctness and speed of YUV conversion functionality.
-
src/video/yuv2rgb/yuv_rgb_std_func.h
-
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
// Copyright 2016 Adrien Descamps // Distributed under BSD 3-Clause License /* You need to define the following macros before including this file: STD_FUNCTION_NAME YUV_FORMAT RGB_FORMAT */ #if RGB_FORMAT == RGB_FORMAT_RGB565 #define PACK_PIXEL(rgb_ptr) \ *(Uint16 *)rgb_ptr = \ ((((Uint16)clampU8(y_tmp+r_tmp)) << 8 ) & 0xF800) | \ ((((Uint16)clampU8(y_tmp+g_tmp)) << 3) & 0x07E0) | \ (((Uint16)clampU8(y_tmp+b_tmp)) >> 3); \ rgb_ptr += 2; \ #elif RGB_FORMAT == RGB_FORMAT_RGB24 #define PACK_PIXEL(rgb_ptr) \ rgb_ptr[0] = clampU8(y_tmp+r_tmp); \ rgb_ptr[1] = clampU8(y_tmp+g_tmp); \ rgb_ptr[2] = clampU8(y_tmp+b_tmp); \ rgb_ptr += 3; \ #elif RGB_FORMAT == RGB_FORMAT_RGBA #define PACK_PIXEL(rgb_ptr) \ *(Uint32 *)rgb_ptr = \ (((Uint32)clampU8(y_tmp+r_tmp)) << 24) | \ (((Uint32)clampU8(y_tmp+g_tmp)) << 16) | \ (((Uint32)clampU8(y_tmp+b_tmp)) << 8) | \ 0x000000FF; \ rgb_ptr += 4; \ #elif RGB_FORMAT == RGB_FORMAT_BGRA #define PACK_PIXEL(rgb_ptr) \ *(Uint32 *)rgb_ptr = \ (((Uint32)clampU8(y_tmp+b_tmp)) << 24) | \ (((Uint32)clampU8(y_tmp+g_tmp)) << 16) | \ (((Uint32)clampU8(y_tmp+r_tmp)) << 8) | \ 0x000000FF; \ rgb_ptr += 4; \ #elif RGB_FORMAT == RGB_FORMAT_ARGB #define PACK_PIXEL(rgb_ptr) \ *(Uint32 *)rgb_ptr = \ 0xFF000000 | \ (((Uint32)clampU8(y_tmp+r_tmp)) << 16) | \ (((Uint32)clampU8(y_tmp+g_tmp)) << 8) | \ (((Uint32)clampU8(y_tmp+b_tmp)) << 0); \ rgb_ptr += 4; \ #elif RGB_FORMAT == RGB_FORMAT_ABGR #define PACK_PIXEL(rgb_ptr) \ *(Uint32 *)rgb_ptr = \ 0xFF000000 | \ (((Uint32)clampU8(y_tmp+b_tmp)) << 16) | \ (((Uint32)clampU8(y_tmp+g_tmp)) << 8) | \ (((Uint32)clampU8(y_tmp+r_tmp)) << 0); \ rgb_ptr += 4; \ #else #error PACK_PIXEL unimplemented #endif void STD_FUNCTION_NAME( uint32_t width, uint32_t height, const uint8_t *Y, const uint8_t *U, const uint8_t *V, uint32_t Y_stride, uint32_t UV_stride, uint8_t *RGB, uint32_t RGB_stride, YCbCrType yuv_type) { const YUV2RGBParam *const param = &(YUV2RGB[yuv_type]); #if YUV_FORMAT == YUV_FORMAT_420 const int y_pixel_stride = 1; const int uv_pixel_stride = 1; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 2; #elif YUV_FORMAT == YUV_FORMAT_422 const int y_pixel_stride = 2; const int uv_pixel_stride = 4; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 1; #elif YUV_FORMAT == YUV_FORMAT_NV12 const int y_pixel_stride = 1; const int uv_pixel_stride = 2; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 2; #endif uint32_t x, y; for(y=0; y<(height-(uv_y_sample_interval-1)); y+=uv_y_sample_interval) { const uint8_t *y_ptr1=Y+y*Y_stride, *y_ptr2=Y+(y+1)*Y_stride, *u_ptr=U+(y/uv_y_sample_interval)*UV_stride, *v_ptr=V+(y/uv_y_sample_interval)*UV_stride; uint8_t *rgb_ptr1=RGB+y*RGB_stride, *rgb_ptr2=RGB+(y+1)*RGB_stride; for(x=0; x<(width-(uv_x_sample_interval-1)); x+=uv_x_sample_interval) { // Compute U and V contributions, common to the four pixels int32_t u_tmp = ((*u_ptr)-128); int32_t v_tmp = ((*v_ptr)-128); int32_t r_tmp = (v_tmp*param->v_r_factor); int32_t g_tmp = (u_tmp*param->u_g_factor + v_tmp*param->v_g_factor); int32_t b_tmp = (u_tmp*param->u_b_factor); // Compute the Y contribution for each pixel int32_t y_tmp = ((y_ptr1[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); y_tmp = ((y_ptr1[y_pixel_stride]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); if (uv_y_sample_interval > 1) { y_tmp = ((y_ptr2[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr2); y_tmp = ((y_ptr2[y_pixel_stride]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr2); } y_ptr1+=2*y_pixel_stride; y_ptr2+=2*y_pixel_stride; u_ptr+=2*uv_pixel_stride/uv_x_sample_interval; v_ptr+=2*uv_pixel_stride/uv_x_sample_interval; } /* Catch the last pixel, if needed */ if (uv_x_sample_interval == 2 && x == (width-1)) { // Compute U and V contributions, common to the four pixels int32_t u_tmp = ((*u_ptr)-128); int32_t v_tmp = ((*v_ptr)-128); int32_t r_tmp = (v_tmp*param->v_r_factor); int32_t g_tmp = (u_tmp*param->u_g_factor + v_tmp*param->v_g_factor); int32_t b_tmp = (u_tmp*param->u_b_factor); // Compute the Y contribution for each pixel int32_t y_tmp = ((y_ptr1[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); if (uv_y_sample_interval > 1) { y_tmp = ((y_ptr2[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr2); } } } /* Catch the last line, if needed */ if (uv_y_sample_interval == 2 && y == (height-1)) { const uint8_t *y_ptr1=Y+y*Y_stride, *u_ptr=U+(y/uv_y_sample_interval)*UV_stride, *v_ptr=V+(y/uv_y_sample_interval)*UV_stride; uint8_t *rgb_ptr1=RGB+y*RGB_stride; for(x=0; x<(width-(uv_x_sample_interval-1)); x+=uv_x_sample_interval) { // Compute U and V contributions, common to the four pixels int32_t u_tmp = ((*u_ptr)-128); int32_t v_tmp = ((*v_ptr)-128); int32_t r_tmp = (v_tmp*param->v_r_factor); int32_t g_tmp = (u_tmp*param->u_g_factor + v_tmp*param->v_g_factor); int32_t b_tmp = (u_tmp*param->u_b_factor); // Compute the Y contribution for each pixel int32_t y_tmp = ((y_ptr1[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); y_tmp = ((y_ptr1[y_pixel_stride]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); y_ptr1+=2*y_pixel_stride; u_ptr+=2*uv_pixel_stride/uv_x_sample_interval; v_ptr+=2*uv_pixel_stride/uv_x_sample_interval; } /* Catch the last pixel, if needed */ if (uv_x_sample_interval == 2 && x == (width-1)) { // Compute U and V contributions, common to the four pixels int32_t u_tmp = ((*u_ptr)-128); int32_t v_tmp = ((*v_ptr)-128); int32_t r_tmp = (v_tmp*param->v_r_factor); int32_t g_tmp = (u_tmp*param->u_g_factor + v_tmp*param->v_g_factor); int32_t b_tmp = (u_tmp*param->u_b_factor); // Compute the Y contribution for each pixel int32_t y_tmp = ((y_ptr1[0]-param->y_shift)*param->y_factor); PACK_PIXEL(rgb_ptr1); } } } #undef STD_FUNCTION_NAME #undef YUV_FORMAT #undef RGB_FORMAT #undef PACK_PIXEL