mirror of
https://github.com/librempeg/librempeg
synced 2024-11-23 03:28:27 +00:00
avutil/hwcontext_cuda: fix YUV420P cuda_get_buffer
Regression since ece068a771
.
Signed-off-by: Marton Balint <cus@passwd.hu>
This commit is contained in:
parent
2dbe936bf7
commit
1b2471039b
@ -159,10 +159,11 @@ static int cuda_get_buffer(AVHWFramesContext *ctx, AVFrame *frame)
|
||||
return res;
|
||||
|
||||
// YUV420P is a special case.
|
||||
// Nvenc expects the U/V planes in swapped order from how ffmpeg expects them.
|
||||
// Nvenc expects the U/V planes in swapped order from how ffmpeg expects them, also chroma is half-aligned
|
||||
if (ctx->sw_format == AV_PIX_FMT_YUV420P) {
|
||||
FFSWAP(uint8_t*, frame->data[1], frame->data[2]);
|
||||
FFSWAP(int, frame->linesize[1], frame->linesize[2]);
|
||||
frame->linesize[1] = frame->linesize[2] = frame->linesize[0] / 2;
|
||||
frame->data[2] = frame->data[1];
|
||||
frame->data[1] = frame->data[2] + frame->linesize[2] * ctx->height / 2;
|
||||
}
|
||||
|
||||
frame->format = AV_PIX_FMT_CUDA;
|
||||
|
Loading…
Reference in New Issue
Block a user