Как установить формат пикселей декодирования в libavcodec?
-
11-12-2019 - |
Вопрос
Я декодирую видео через libavcodec, используя следующий код:
//Open input file
if(avformat_open_input(&ctx, filename, NULL, NULL)!=0)
return FALSE; // Couldn't open file
if(avformat_find_stream_info(ctx, NULL)<0)
return FALSE; // Couldn't find stream information
videoStream = -1;
//find video stream
for(i=0; i<ctx->nb_streams; i++)
{
if((ctx->streams[i])->codec->codec_type==AVMEDIA_TYPE_VIDEO)
{
videoStream=i;
break;
}
}
if (videoStream == -1)
return FALSE; // Didn't find a video stream
video_codec_ctx=ctx->streams[videoStream]->codec;
//find decoder
video_codec=avcodec_find_decoder(video_codec_ctx->codec_id);
if(video_codec==NULL)
return FALSE; // Codec not found
if(avcodec_open(video_codec_ctx, video_codec)<0)
return -1; // Could not open codec
video_frame=avcodec_alloc_frame();
scaled_frame=avcodec_alloc_frame();
static struct SwsContext *img_convert_ctx;
if(img_convert_ctx == NULL)
{
int w = video_codec_ctx->width;
int h = video_codec_ctx->height;
img_convert_ctx = sws_getContext(w, h,
video_codec_ctx->pix_fmt,
w, h, dst_pix_fmt, SWS_BICUBIC,
NULL, NULL, NULL);
if(img_convert_ctx == NULL) {
fprintf(stderr, "Cannot initialize the conversion context!\n");
return FALSE;
}
}
while(b_play)
{
if (av_read_frame(ctx, &packet) < 0)
{
break;
}
if(packet.stream_index==videoStream) {
// Decode video frame
avcodec_decode_video2(video_codec_ctx, video_frame, &frameFinished,
&packet);
// Did we get a video frame?
if(frameFinished)
{
if (video_codec_ctx->pix_fmt != dst_pix_fmt)
{
if (video_codec_ctx->pix_fmt != dst_pix_fmt)
sws_scale(img_convert_ctx, video_frame->data,
video_frame->linesize, 0,
video_codec_ctx->height,
scaled_frame->data, scaled_frame->linesize);
}
}
}
av_free_packet(&packet);
}
Код работает корректно, но необходимо конвертировать каждый кадр в нужный формат.Можно ли установить формат пикселей для декодирования, чтобы получить правильный формат без sws_scale?
Большое спасибо за ваши ответы.
Решение
ffmpeg
's AVCodec
каждый из экземпляров (объектов «фабрики» статического декодера) определяет массив поддерживаемых ими форматов пикселей, оканчивающийся значением -1.
А AVCodecContext
(экземпляр декодера) имеют указатель на функцию обратного вызова, называемый get_format
:это указатель на функцию в этой структуре.
Эта функция обратного вызова вызывается в какой-то момент инициализации кодека с помощью AVCodec
массив поддерживаемых форматов фабричного объекта, и обратный вызов должен выбрать один из форматов из этого массива (что-то вроде «выберите карту, любую карту») и вернуть это значение.Реализация этого по умолчанию get_format
обратный вызов — это функция, называемая avcodec_default_get_format
.(Это установлено avcodec_get_context_defaults2
).Эта функция по умолчанию довольно просто реализует логику «выбора формата»:он выбирает первый элемент массива, который не является форматом пикселей только с аппаратным ускорением.
Если вы хотите, чтобы кодек работал с другим форматом пикселей, вы можете установить свой собственный. get_format
обратный вызов в объект контекста.Однако обратный вызов должен возвращать одно из значений массива (например, выбор из меню).Он не может возвращать произвольное значение.Кодек будет поддерживать только те форматы, которые указаны в массиве.
Просмотрите множество доступных форматов и выберите лучший.Если вам повезет, это именно тот, который вам действительно нужен, и sws_scale
функции не придется выполнять преобразование формата пикселей.(Если, кроме того, вы не запрашиваете масштабирование или обрезку изображения, sws_scale должен распознать, что преобразование является ошибкой.)