mirror of
https://github.com/nyanmisaka/ffmpeg-rockchip.git
synced 2025-10-27 02:41:54 +08:00
swscale: add gbr(a)p16 output support
This commit is contained in:
@@ -2021,6 +2021,81 @@ yuv2gbrp_full_X_c(SwsContext *c, const int16_t *lumFilter,
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
yuv2gbrp16_full_X_c(SwsContext *c, const int16_t *lumFilter,
|
||||
const int16_t **lumSrcx, int lumFilterSize,
|
||||
const int16_t *chrFilter, const int16_t **chrUSrcx,
|
||||
const int16_t **chrVSrcx, int chrFilterSize,
|
||||
const int16_t **alpSrc, uint8_t **dest,
|
||||
int dstW, int y)
|
||||
{
|
||||
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(c->dstFormat);
|
||||
int i;
|
||||
int hasAlpha = (desc->flags & AV_PIX_FMT_FLAG_ALPHA) && alpSrc;
|
||||
uint16_t **dest16 = (uint16_t**)dest;
|
||||
const int32_t **lumSrc = (const int32_t**)lumSrcx;
|
||||
const int32_t **chrUSrc = (const int32_t**)chrUSrcx;
|
||||
const int32_t **chrVSrc = (const int32_t**)chrVSrcx;
|
||||
int A = 0; // init to silence warning
|
||||
|
||||
for (i = 0; i < dstW; i++) {
|
||||
int j;
|
||||
int Y = -0x40000000;
|
||||
int U = -(128 << 23);
|
||||
int V = -(128 << 23);
|
||||
int R, G, B;
|
||||
|
||||
for (j = 0; j < lumFilterSize; j++)
|
||||
Y += lumSrc[j][i] * (unsigned)lumFilter[j];
|
||||
|
||||
for (j = 0; j < chrFilterSize; j++) {
|
||||
U += chrUSrc[j][i] * (unsigned)chrFilter[j];
|
||||
V += chrVSrc[j][i] * (unsigned)chrFilter[j];
|
||||
}
|
||||
|
||||
Y >>= 14;
|
||||
Y += 0x10000;
|
||||
U >>= 14;
|
||||
V >>= 14;
|
||||
|
||||
if (hasAlpha) {
|
||||
A = 1 << 18;
|
||||
|
||||
for (j = 0; j < lumFilterSize; j++)
|
||||
A += alpSrc[j][i] * lumFilter[j];
|
||||
|
||||
if (A & 0xF8000000)
|
||||
A = av_clip_uintp2(A, 27);
|
||||
}
|
||||
|
||||
Y -= c->yuv2rgb_y_offset;
|
||||
Y *= c->yuv2rgb_y_coeff;
|
||||
Y += 1 << 13;
|
||||
R = V * c->yuv2rgb_v2r_coeff;
|
||||
G = V * c->yuv2rgb_v2g_coeff + U * c->yuv2rgb_u2g_coeff;
|
||||
B = U * c->yuv2rgb_u2b_coeff;
|
||||
|
||||
R = av_clip_uintp2(Y + R, 30);
|
||||
G = av_clip_uintp2(Y + G, 30);
|
||||
B = av_clip_uintp2(Y + B, 30);
|
||||
|
||||
dest16[0][i] = G >> 14;
|
||||
dest16[1][i] = B >> 14;
|
||||
dest16[2][i] = R >> 14;
|
||||
if (hasAlpha)
|
||||
dest16[3][i] = A >> 11;
|
||||
}
|
||||
if ((!isBE(c->dstFormat)) != (!HAVE_BIGENDIAN)) {
|
||||
for (i = 0; i < dstW; i++) {
|
||||
dest16[0][i] = av_bswap16(dest16[0][i]);
|
||||
dest16[1][i] = av_bswap16(dest16[1][i]);
|
||||
dest16[2][i] = av_bswap16(dest16[2][i]);
|
||||
if (hasAlpha)
|
||||
dest16[3][i] = av_bswap16(dest16[3][i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
yuv2ya8_1_c(SwsContext *c, const int16_t *buf0,
|
||||
const int16_t *ubuf[2], const int16_t *vbuf[2],
|
||||
@@ -2402,8 +2477,6 @@ av_cold void ff_sws_init_output_funcs(SwsContext *c,
|
||||
case AV_PIX_FMT_GBRP12LE:
|
||||
case AV_PIX_FMT_GBRP14BE:
|
||||
case AV_PIX_FMT_GBRP14LE:
|
||||
case AV_PIX_FMT_GBRP16BE:
|
||||
case AV_PIX_FMT_GBRP16LE:
|
||||
case AV_PIX_FMT_GBRAP:
|
||||
case AV_PIX_FMT_GBRAP10BE:
|
||||
case AV_PIX_FMT_GBRAP10LE:
|
||||
@@ -2411,6 +2484,12 @@ av_cold void ff_sws_init_output_funcs(SwsContext *c,
|
||||
case AV_PIX_FMT_GBRAP12LE:
|
||||
*yuv2anyX = yuv2gbrp_full_X_c;
|
||||
break;
|
||||
case AV_PIX_FMT_GBRP16BE:
|
||||
case AV_PIX_FMT_GBRP16LE:
|
||||
case AV_PIX_FMT_GBRAP16BE:
|
||||
case AV_PIX_FMT_GBRAP16LE:
|
||||
*yuv2anyX = yuv2gbrp16_full_X_c;
|
||||
break;
|
||||
}
|
||||
if (!*yuv2packedX && !*yuv2anyX)
|
||||
goto YUV_PACKED;
|
||||
|
||||
Reference in New Issue
Block a user