mirror of https://github.com/FFmpeg/FFmpeg.git
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.
407 lines
13 KiB
407 lines
13 KiB
#include "avutil.h" |
|
#include "swscale.h" |
|
#include "swscale_internal.h" |
|
|
|
#define YTABLE_MIN 384 |
|
|
|
/** |
|
* YUV -> RGB conversion matrixes (inverse of table 6.9 in MPEG2 standard) |
|
* |
|
* An YUV -> RGB conversion matrix is in the form |
|
* | 1 0 Rv | |
|
* | 1 Gu Gv | |
|
* | 1 Bu 0 | |
|
* |
|
* Inverse_Table_6_9 stores | Rv Bu Gv Gu | * 255/224*2^16. |
|
* \arg Maximum Rv value: 117570 |
|
* \arg Maximum Bu value: 138420 |
|
* \arg Maximum Gv + Gu value: 25642 + 53281 = 78923 |
|
* |
|
* These values are needed to allocate table_{r, g, b}. If you modify |
|
* this table, please update allocate_tables() accordingly |
|
*/ |
|
const int32_t Inverse_Table_6_9[8][4] = { |
|
{0, 0, 0, 0}, /* no sequence_display_extension */ |
|
{117500, 138420, -13985, -34933}, /* ITU-R Rec. 709 (1990) */ |
|
{0, 0, 0, 0}, /* unspecified */ |
|
{0, 0, 0, 0}, /* reserved */ |
|
{104480, 132820, -24811, -53150}, /* FCC */ |
|
{104570, 132210, -25642, -53281}, /* ITU-R Rec. 624-4 System B, G */ |
|
{104570, 132210, -25642, -53281}, /* SMPTE 170M */ |
|
{117570, 136230, -16892, -35552} /* SMPTE 240M (1987) */ |
|
}; |
|
|
|
|
|
/** |
|
* Dithering matrixes (these are bayer ordered dither matrixes |
|
* with some manual changes by Michael) |
|
*/ |
|
const uint8_t __attribute__((aligned(8))) dither_2x2_4[2][8]={ |
|
{ 1, 3, 1, 3, 1, 3, 1, 3, }, |
|
{ 2, 0, 2, 0, 2, 0, 2, 0, }, |
|
}; |
|
|
|
const uint8_t __attribute__((aligned(8))) dither_2x2_8[2][8]={ |
|
{ 6, 2, 6, 2, 6, 2, 6, 2, }, |
|
{ 0, 4, 0, 4, 0, 4, 0, 4, }, |
|
}; |
|
|
|
const uint8_t __attribute__((aligned(8))) dither_8x8_32[8][8]={ |
|
{ 17, 9, 23, 15, 16, 8, 22, 14, }, |
|
{ 5, 29, 3, 27, 4, 28, 2, 26, }, |
|
{ 21, 13, 19, 11, 20, 12, 18, 10, }, |
|
{ 0, 24, 6, 30, 1, 25, 7, 31, }, |
|
{ 16, 8, 22, 14, 17, 9, 23, 15, }, |
|
{ 4, 28, 2, 26, 5, 29, 3, 27, }, |
|
{ 20, 12, 18, 10, 21, 13, 19, 11, }, |
|
{ 1, 25, 7, 31, 0, 24, 6, 30, }, |
|
}; |
|
|
|
#if 0 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_64[8][8]={ |
|
{ 0, 48, 12, 60, 3, 51, 15, 63, }, |
|
{ 32, 16, 44, 28, 35, 19, 47, 31, }, |
|
{ 8, 56, 4, 52, 11, 59, 7, 55, }, |
|
{ 40, 24, 36, 20, 43, 27, 39, 23, }, |
|
{ 2, 50, 14, 62, 1, 49, 13, 61, }, |
|
{ 34, 18, 46, 30, 33, 17, 45, 29, }, |
|
{ 10, 58, 6, 54, 9, 57, 5, 53, }, |
|
{ 42, 26, 38, 22, 41, 25, 37, 21, }, |
|
}; |
|
#endif |
|
|
|
const uint8_t __attribute__((aligned(8))) dither_8x8_73[8][8]={ |
|
{ 0, 55, 14, 68, 3, 58, 17, 72, }, |
|
{ 37, 18, 50, 32, 40, 22, 54, 35, }, |
|
{ 9, 64, 5, 59, 13, 67, 8, 63, }, |
|
{ 46, 27, 41, 23, 49, 31, 44, 26, }, |
|
{ 2, 57, 16, 71, 1, 56, 15, 70, }, |
|
{ 39, 21, 52, 34, 38, 19, 51, 33, }, |
|
{ 11, 66, 7, 62, 10, 65, 6, 60, }, |
|
{ 48, 30, 43, 25, 47, 29, 42, 24, }, |
|
}; |
|
|
|
#if 0 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_128[8][8]={ |
|
{ 68, 36, 92, 60, 66, 34, 90, 58, }, |
|
{ 20, 116, 12, 108, 18, 114, 10, 106, }, |
|
{ 84, 52, 76, 44, 82, 50, 74, 42, }, |
|
{ 0, 96, 24, 120, 6, 102, 30, 126, }, |
|
{ 64, 32, 88, 56, 70, 38, 94, 62, }, |
|
{ 16, 112, 8, 104, 22, 118, 14, 110, }, |
|
{ 80, 48, 72, 40, 86, 54, 78, 46, }, |
|
{ 4, 100, 28, 124, 2, 98, 26, 122, }, |
|
}; |
|
#endif |
|
|
|
#if 1 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_220[8][8]={ |
|
{117, 62, 158, 103, 113, 58, 155, 100, }, |
|
{ 34, 199, 21, 186, 31, 196, 17, 182, }, |
|
{144, 89, 131, 76, 141, 86, 127, 72, }, |
|
{ 0, 165, 41, 206, 10, 175, 52, 217, }, |
|
{110, 55, 151, 96, 120, 65, 162, 107, }, |
|
{ 28, 193, 14, 179, 38, 203, 24, 189, }, |
|
{138, 83, 124, 69, 148, 93, 134, 79, }, |
|
{ 7, 172, 48, 213, 3, 168, 45, 210, }, |
|
}; |
|
#elif 1 |
|
// tries to correct a gamma of 1.5 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_220[8][8]={ |
|
{ 0, 143, 18, 200, 2, 156, 25, 215, }, |
|
{ 78, 28, 125, 64, 89, 36, 138, 74, }, |
|
{ 10, 180, 3, 161, 16, 195, 8, 175, }, |
|
{109, 51, 93, 38, 121, 60, 105, 47, }, |
|
{ 1, 152, 23, 210, 0, 147, 20, 205, }, |
|
{ 85, 33, 134, 71, 81, 30, 130, 67, }, |
|
{ 14, 190, 6, 171, 12, 185, 5, 166, }, |
|
{117, 57, 101, 44, 113, 54, 97, 41, }, |
|
}; |
|
#elif 1 |
|
// tries to correct a gamma of 2.0 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_220[8][8]={ |
|
{ 0, 124, 8, 193, 0, 140, 12, 213, }, |
|
{ 55, 14, 104, 42, 66, 19, 119, 52, }, |
|
{ 3, 168, 1, 145, 6, 187, 3, 162, }, |
|
{ 86, 31, 70, 21, 99, 39, 82, 28, }, |
|
{ 0, 134, 11, 206, 0, 129, 9, 200, }, |
|
{ 62, 17, 114, 48, 58, 16, 109, 45, }, |
|
{ 5, 181, 2, 157, 4, 175, 1, 151, }, |
|
{ 95, 36, 78, 26, 90, 34, 74, 24, }, |
|
}; |
|
#else |
|
// tries to correct a gamma of 2.5 |
|
const uint8_t __attribute__((aligned(8))) dither_8x8_220[8][8]={ |
|
{ 0, 107, 3, 187, 0, 125, 6, 212, }, |
|
{ 39, 7, 86, 28, 49, 11, 102, 36, }, |
|
{ 1, 158, 0, 131, 3, 180, 1, 151, }, |
|
{ 68, 19, 52, 12, 81, 25, 64, 17, }, |
|
{ 0, 119, 5, 203, 0, 113, 4, 195, }, |
|
{ 45, 9, 96, 33, 42, 8, 91, 30, }, |
|
{ 2, 172, 1, 144, 2, 165, 0, 137, }, |
|
{ 77, 23, 60, 15, 72, 21, 56, 14, }, |
|
}; |
|
#endif |
|
|
|
static int get_entry_size(int bpp) |
|
{ |
|
switch(bpp) { |
|
case 32: |
|
return 4; |
|
case 16: |
|
case 15: |
|
return 2; |
|
case 24: |
|
case 8: |
|
case 4: |
|
case 1: |
|
return 1; |
|
default: |
|
return -1; |
|
} |
|
} |
|
|
|
/** |
|
* Allocate table_r, table_g, and table_b |
|
* |
|
* For cache efficency reasons, these three tables are allocated |
|
* together, so that they are contiguous in memory |
|
* |
|
* table_r is indexed in the range |
|
* [-128 * 117570 / 76309, 255 + 127 * 117570 / 76309] = |
|
* [-197.21, 451.67] ---> [-198, 452] |
|
* table_b is indexed in the range |
|
* [-128 * 138420 / 76309, 255 + 127 * 138420 / 76309] = |
|
* [232.18, 485.37] ---> [-233, 486] |
|
* table_g is indexed in the range |
|
* [-128 * 78923 / 76309, 255 + 127 * 78923 / 76309] = |
|
* [-132.38, 386.35] ---> [-133, 387] |
|
* |
|
* Please look at the comments after Inverse_Table_6_9 to see where these |
|
* numbers are coming from. |
|
*/ |
|
static void *allocate_tables(uint8_t **table_r, uint8_t **table_g, uint8_t **table_b, int bpp) |
|
{ |
|
uint8_t *table; |
|
int entry_size; |
|
|
|
entry_size = get_entry_size(bpp); |
|
|
|
/* First allocate the memory... */ |
|
switch (bpp) { |
|
case 32: |
|
case 15: |
|
case 16: |
|
case 8: |
|
case 4: |
|
table = av_malloc((198 + 452 + 233 + 486 + 133 + 387) * entry_size); |
|
break; |
|
case 24: |
|
table = av_malloc(256 + 2 * 233); |
|
break; |
|
case 1: |
|
table = av_malloc (256 * 2); |
|
break; |
|
default: |
|
table = NULL; |
|
} |
|
if (table == NULL) { |
|
MSG_ERR("Cannot allocate memory for the YUV -> RGB tables!\n"); |
|
|
|
return NULL; |
|
} |
|
|
|
/* ...and then, assign the table_* value */ |
|
switch (bpp) { |
|
case 32: |
|
case 15: |
|
case 16: |
|
case 8: |
|
case 4: |
|
*table_r = table + 198 * entry_size; |
|
*table_b = table + (198 + 452 + 133 + 387 + 233) * entry_size; |
|
*table_g = table + (198 + 452 + 133) * entry_size; |
|
break; |
|
case 24: |
|
*table_r = *table_g = *table_b = table + 233; |
|
break; |
|
case 1: |
|
*table_g = table; |
|
*table_r = *table_b = NULL; |
|
break; |
|
} |
|
|
|
return table; |
|
} |
|
|
|
/** |
|
* Initialize the table_rV, table_gU[i], table_gV, and table_bU fields |
|
* in SwsContext |
|
* |
|
* @param inv_table the YUV -> RGB table (this is a line of Inverse_Table_6_9) |
|
* @param fullRange 0->MPEG YUV space 1->JPEG YUV space |
|
*/ |
|
int yuv2rgb_c_init_tables(SwsContext *c, const int inv_table[4], int fullRange, int brightness, int contrast, int saturation) |
|
{ |
|
int i; |
|
static uint8_t ytable[1024]; |
|
int64_t cy, oy; |
|
int64_t crv, cbu, cgu, cgv; |
|
int entry_size = 0; |
|
uint8_t *table_r, *table_g, *table_b; |
|
int value; |
|
|
|
if ((inv_table[0] == 0) || (inv_table[1] == 0) || (inv_table[2] == 0) || (inv_table[3] == 0)) { |
|
MSG_ERR("Invalid YUV ---> RGB table!\n"); |
|
|
|
return -1; |
|
} |
|
crv = inv_table[0]; |
|
cbu = inv_table[1]; |
|
cgu = inv_table[2]; |
|
cgv = inv_table[3]; |
|
if (fullRange) { |
|
cy = 1 << 16; |
|
oy = 0; |
|
} else { |
|
cy = ((1 << 16) * 255) / 219; |
|
oy= 16 << 16; |
|
} |
|
|
|
cy = (cy *contrast )>>16; |
|
crv= (crv*contrast * saturation)>>32; |
|
cbu= (cbu*contrast * saturation)>>32; |
|
cgu= (cgu*contrast * saturation)>>32; |
|
cgv= (cgv*contrast * saturation)>>32; |
|
oy -= 256*brightness; |
|
|
|
for (i = 0; i < 1024; i++) { |
|
value = (cy*(((i - YTABLE_MIN)<<16) - oy) + (1<<31))>>32; |
|
ytable[i] = clip_uint8(value); |
|
} |
|
|
|
entry_size = get_entry_size(fmt_depth(c->dstFormat)); |
|
av_free(c->yuvTable); |
|
c->yuvTable = allocate_tables(&table_r, &table_g, &table_b, fmt_depth(c->dstFormat)); |
|
if (c->yuvTable == NULL) { |
|
return -1; |
|
} |
|
|
|
switch (fmt_depth(c->dstFormat)) { |
|
case 32: |
|
for (i = -198; i < 256 + 197; i++) { |
|
value = ytable[i + YTABLE_MIN]; |
|
if (isBGR(c->dstFormat)) { |
|
value <<= 16; |
|
} |
|
((uint32_t *)table_r)[i] = value; |
|
} |
|
for (i = -133; i < 256 + 132; i++) { |
|
((uint32_t *)table_g)[i] = ytable[i + YTABLE_MIN] << 8; |
|
} |
|
for (i = -233; i < 256 + 232; i++) { |
|
value = ytable[i + YTABLE_MIN]; |
|
if (!isBGR(c->dstFormat)) { |
|
value <<= 16; |
|
} |
|
((uint32_t *)table_b)[i] = value; |
|
} |
|
break; |
|
|
|
case 24: |
|
for (i = -233; i < 256 + 232; i++) { |
|
((uint8_t * )table_b)[i] = ytable[i + YTABLE_MIN]; |
|
} |
|
break; |
|
|
|
case 15: |
|
case 16: |
|
for (i = -198; i < 256 + 197; i++) { |
|
value = ytable[i + YTABLE_MIN] >> 3; |
|
if (isBGR(c->dstFormat)) { |
|
value <<= ((fmt_depth(c->dstFormat) == 16) ? 11 : 10); |
|
} |
|
((uint16_t *)table_r)[i] = value; |
|
} |
|
for (i = -133; i < 256 + 132; i++) { |
|
value = ytable[i + YTABLE_MIN]; |
|
value >>= ((fmt_depth(c->dstFormat) == 16) ? 2 : 3); |
|
((uint16_t *)table_g)[i] = value << 5; |
|
} |
|
for (i = -233; i < 256 + 232; i++) { |
|
value = ytable[i + YTABLE_MIN] >> 3; |
|
if (!isBGR(c->dstFormat)) { |
|
value <<= ((fmt_depth(c->dstFormat) == 16) ? 11 : 10); |
|
} |
|
((uint16_t *)table_b)[i] = value; |
|
} |
|
break; |
|
case 8: |
|
for (i = -198; i < 256 + 197; i++) { |
|
value = (ytable[i + YTABLE_MIN - 16] + 18) / 36; |
|
if (isBGR(c->dstFormat)) { |
|
value <<= 5; |
|
} |
|
((uint8_t *)table_r)[i] = value; |
|
} |
|
for (i = -133; i < 256 + 132; i++) { |
|
value = (ytable[i + YTABLE_MIN - 16] + 18) / 36; |
|
if (!isBGR(c->dstFormat)) { |
|
value <<= 1; |
|
} |
|
((uint8_t *)table_g)[i] = value << 2; |
|
} |
|
for (i = -233; i < 256 + 232; i++) { |
|
value = (ytable[i + YTABLE_MIN - 37] + 43) / 85; |
|
if (!isBGR(c->dstFormat)) { |
|
value <<= 6; |
|
} |
|
((uint8_t *)table_b)[i] = value; |
|
} |
|
break; |
|
case 4: |
|
for (i = -198; i < 256 + 197; i++) { |
|
value = ytable[i + YTABLE_MIN - 110] >> 7; |
|
if (isBGR(c->dstFormat)) { |
|
value <<= 3; |
|
} |
|
((uint8_t *)table_r)[i] = value; |
|
} |
|
for (i = -133; i < 256 + 132; i++) { |
|
value = (ytable[i + YTABLE_MIN - 37]+ 43) / 85; |
|
((uint8_t *)table_g)[i] = value << 1; |
|
} |
|
for (i = -233; i < 256 + 232; i++) { |
|
value = ytable[i + YTABLE_MIN - 110] >> 7; |
|
if (!isBGR(c->dstFormat)) { |
|
value <<= 3; |
|
} |
|
((uint8_t *)table_b)[i] = value; |
|
} |
|
break; |
|
case 1: |
|
for (i = 0; i < 256 + 256; i++) { |
|
value = ytable[i + YTABLE_MIN - 110] >> 7; |
|
((uint8_t *)table_g)[i] = value; |
|
} |
|
break; |
|
default: |
|
MSG_ERR("%ibpp not supported by yuv2rgb\n", fmt_depth(c->dstFormat)); |
|
av_free(c->yuvTable); |
|
c->yuvTable = NULL; |
|
|
|
return -1; |
|
} |
|
|
|
for (i = 0; i < 256; i++) { |
|
c->table_rV[i] = table_r + |
|
entry_size * ROUNDED_DIV(crv * (i - 128), 76309); |
|
c->table_gU[i] = table_g + |
|
entry_size * ROUNDED_DIV(cgu * (i - 128), 76309); |
|
c->table_gV[i] = entry_size * ROUNDED_DIV(cgv * (i - 128), 76309); |
|
c->table_bU[i] = table_b + |
|
entry_size * ROUNDED_DIV(cbu * (i - 128), 76309); |
|
} |
|
|
|
return 0; |
|
}
|
|
|