libavformat/libavcodec, предоставляющий неверный заголовок контейнера

Я использую libavcodec для кодирования потока в h264 и libavformat для его сохранения в формате mp4. Полученный контейнер имеет недопустимый заголовок, который можно воспроизвести в VLC, но не в любом другом проигрывателе.

Я обнаружил, что использование контейнера mp4 и кодека «mpeg4» создает действительный файл mp4, но использование кодека libx265 (HEVC) или кодека libx264 создает недопустимые файлы mp4.

Я могу использовать ffmpeg -i invalid.mp4 -vcodec copy valid.mp4 и получить файл почти такого же размера, но в правильном контейнере.

Примеры этих файлов находятся здесь: Битый файл и Повторно оплаченный файл [используйте ссылки для скачивания в правом верхнем углу, чтобы изучить]

Я использовал шестнадцатеричный редактор, чтобы увидеть различия в заголовках двух файлов, и недействительный файл на 1 байт меньше, чем действительный.

Код, который я использую для открытия контейнера и кодека, а также для записи заголовка, находится здесь:

AVOutputFormat *container_format;
AVFormatContext *container_format_context;
AVStream *video_stream;
int ret;

/* allocate the output media context */
avformat_alloc_output_context2(&container_format_context, NULL, NULL, out_file);
if (!container_format_context) {
    log(INFO, "Unable to determine container format from filename, exiting\n");
    exit(1);
}
else {
    log(INFO, "Using container %s\n", container_format_context->oformat->name);
}

if (!container_format_context) {
    log(ERROR, "Could not build container format context. Encoding failed.");
    exit(1);
}

container_format = container_format_context->oformat;

/* Pull codec based on name */
AVCodec* codec = avcodec_find_encoder_by_name(codec_name);
if (codec == NULL) {
    log(ERROR, "Failed to locate codec \"%s\".",
            codec_name);
    exit(1);
}

/* create stream */
video_stream = NULL;
video_stream = avformat_new_stream(container_format_context, codec);
if (!video_stream) {
    log(ERROR, "Could not allocate encoder stream. Cannot continue.\n");
    exit(1);
}
video_stream->id = container_format_context->nb_streams - 1;

video_stream->time_base = video_stream->codec->time_base = (AVRational) { 1, 25};

av_dump_format(container_format_context, 0, out_file, 1);



/* Retrieve encoding context */
AVCodecContext* avcodec_context = video_stream->codec;
if (avcodec_context == NULL) {
    log(ERROR, "Failed to allocate context for "
            "codec \"%s\".", codec_name);
    exit(1);
}


/* Init context with encoding parameters */
avcodec_context->bit_rate = bitrate;
avcodec_context->width = width;
avcodec_context->height = height;
avcodec_context->gop_size = 10;
avcodec_context->max_b_frames = 1;
avcodec_context->qmax = 31;
avcodec_context->qmin = 2;
avcodec_context->pix_fmt = AV_PIX_FMT_YUV420P;

av_dump_format(container_format_context, 0, out_file, 1);

/* Open codec for use */
if (avcodec_open2(avcodec_context, codec, NULL) < 0) {
    log(ERROR, "Failed to open codec \"%s\".", codec_name);
    exit(1);
}

/* Allocate corresponding frame */
AVFrame* frame = av_frame_alloc();
if (frame == NULL) {
    exit(1);
}

/* Copy necessary data for frame from avcodec_context */
frame->format = avcodec_context->pix_fmt;
frame->width = avcodec_context->width;
frame->height = avcodec_context->height;

/* Allocate actual backing data for frame */
if (av_image_alloc(frame->data, frame->linesize, frame->width,
            frame->height, frame->format, 32) < 0) {
    exit(1);
}

/* open the output file, if the container needs it */
if (!(container_format->flags & AVFMT_NOFILE)) {
    ret = avio_open(&container_format_context->pb, out_file, AVIO_FLAG_WRITE);
    if (ret < 0) {
        log(ERROR, "Error occurred while opening output file: %s\n",
                av_err2str(ret));
        exit(1);
    }
}

/* write the stream header, if needed */
ret = avformat_write_header(container_format_context, NULL);
if (ret < 0) {
    log(ERROR, "Error occurred while writing output file header: %s\n",
                av_err2str(ret));
}

Код для кодирования кадра находится здесь:

/* Init video packet */
AVPacket packet;
av_init_packet(&packet);

/* Request that encoder allocate data for packet */
packet.data = NULL;
packet.size = 0;

/* Write frame to video */
int got_data;
if (avcodec_encode_video2(avcontext, &packet, frame, &got_data) < 0) {
    log(WARNING, "Error encoding frame #%" PRId64,
            video_struct->next_pts);
    return -1;
}

/* Write corresponding data to file */
if (got_data) {
    if (packet.pts != AV_NOPTS_VALUE) {
        packet.pts = av_rescale_q(packet.pts, video_struct->output_stream->codec->time_base, video_struct->output_stream->time_base);
    }
    if (packet.dts != AV_NOPTS_VALUE) {
        packet.dts = av_rescale_q(packet.dts, video_struct->output_stream->codec->time_base, video_struct->output_stream->time_base);
    }
    write_packet(video_struct, &packet, packet.size);
    av_packet_unref(&packet);
}

И код для записи пакета в видеопоток:

static int write_packet(video_struct* video, void* data, int size) {

int ret;

/* use AVStream is not null, otherwise write to output fd */
AVPacket *pkt = (AVPacket*) data;
pkt->stream_index = video->output_stream->index;
ret = av_interleaved_write_frame(video->container_format_context, pkt);
if (ret != 0) {
    return -1;
}

/* Data was written successfully */
return ret;
}

person seanr8    schedule 13.12.2017    source источник
comment
Вы должны загрузить образец.   -  person aergistal    schedule 14.12.2017
comment
Вы уверены, что mpeg4 и h264 здесь одно и то же?   -  person the kamilz    schedule 14.12.2017
comment
h264 и mpeg4 не одно и то же, вот только код надо было вырезать из поста, буду редактировать. Я явно передаю libx264 в качестве аргумента командной строки. я отредактирую   -  person seanr8    schedule 14.12.2017
comment
Я добавил примеры файлов для вас, @aergistal   -  person seanr8    schedule 14.12.2017


Ответы (1)


Решил эту проблему. Проблема заключалась в том, что я не назначал контейнеру глобальные заголовки, если контейнер этого требовал. При назначении таких свойств, как высота, ширина, битрейт и т. д. для avcodec_context, я добавил

if (container_format_context->oformat->flags & AVFMT_GLOBALHEADER) {
    avcodec_context->flags |= CODEC_FLAG_GLOBAL_HEADER;
}

который, кажется, решил проблему.

person seanr8    schedule 14.12.2017