2017-05-31 89 views
1

我正在使用FFMPEG库修剪视频文件。我这样做都是作为remux,没有编码或解码。由remux FFMPEG修剪不写关键帧

修剪当前可以与音频一起正常工作,但修剪后的视频数据显示为纯色,并且像素的小方块发生了变化。我相信这是因为我没有捕捉/编写关键帧。这是我的理解,av_seek_frame将寻求一个关键帧,这似乎并非如此..

如果需要,我可以解码,然后重新编码只是我查找后第一个视频帧?与重新编码每一帧相比,这可能是更多的代码,但速度是这里的主要问题,而不是复杂性。

谢谢你的帮助。此外,如果我误解了某些与视频文件有关的问题,我对此表示歉意,但我仍然对此不太了解。

实施例输出帧: enter image description here

代码,适于从设置有FFMPEG的REMUX例如:

const char *out_filename = "aaa.mp4"; 

FILE *fp = fdopen(fd, "r"); 
fseek(fp, 0, SEEK_SET); 

if (fp) { 

    // Build an ffmpeg file 
    char path[512]; 
    sprintf(path, "pipe:%d", fileno(fp)); 

    // Turn on verbosity 
    av_log_set_level(AV_LOG_DEBUG); 
    av_log_set_callback(avLogCallback); 


    av_register_all(); 
    avcodec_register_all(); 


    AVOutputFormat *ofmt = NULL; 
    AVFormatContext *ifmt_ctx = avformat_alloc_context(), *ofmt_ctx = NULL; 
    AVPacket pkt; 
    int ret, i; 


    if ((ret = avformat_open_input(&ifmt_ctx, path, av_find_input_format("mp4"), NULL)) < 0) { 
     LOG("Could not open input file '%s'", path); 
     goto end; 
    } 

    if ((ret = avformat_find_stream_info(ifmt_ctx, 0)) < 0) { 
     LOG("Failed to retrieve input stream information", ""); 
     goto end; 
    } 


    avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename); 
    if (!ofmt_ctx) { 
     LOG("Could not create output context\n"); 
     ret = AVERROR_UNKNOWN; 
     goto end; 
    } 

    ofmt = ofmt_ctx->oformat; 


    for (i = 0; i < ifmt_ctx->nb_streams; i++) { 
     AVStream *in_stream = ifmt_ctx->streams[i]; 
     AVStream *out_stream = avformat_new_stream(ofmt_ctx, NULL); 

     if (!out_stream) { 
      LOG("Failed allocating output stream\n"); 
      goto end; 
     } 

     ret = avcodec_parameters_copy(out_stream->codecpar, in_stream->codecpar); 
     if (ret < 0) { 
      LOG("Failed to copy context from input to output stream codec context\n"); 
      goto end; 
     } 
     out_stream->codecpar->codec_tag = 0; 
    } 

    if (!(ofmt->flags & AVFMT_NOFILE)) { 
     ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE); 
     if (ret < 0) { 
      LOG("Could not open output file '%s'", out_filename); 
      goto end; 
     } 
    } 

    ret = avformat_write_header(ofmt_ctx, NULL); 
    if (ret < 0) { 
     LOG("Error occurred when writing headers\n"); 
     goto end; 
    } 

    ret = av_seek_frame(ifmt_ctx, -1, from_seconds * AV_TIME_BASE, AVSEEK_FLAG_ANY); 
    if (ret < 0) { 
     LOG("Error seek\n"); 
     goto end; 
    } 

    int64_t *dts_start_from; 
    int64_t *pts_start_from; 
    dts_start_from = (int64_t *) malloc(sizeof(int64_t) * ifmt_ctx->nb_streams); 
    memset(dts_start_from, 0, sizeof(int64_t) * ifmt_ctx->nb_streams); 
    pts_start_from = (int64_t *) malloc(sizeof(int64_t) * ifmt_ctx->nb_streams); 
    memset(pts_start_from, 0, sizeof(int64_t) * ifmt_ctx->nb_streams); 

    while (1) { 
     AVStream *in_stream, *out_stream; 

     ret = av_read_frame(ifmt_ctx, &pkt); 
     LOG("while %d", ret); 
     LOG("Packet size: %d", pkt.size); 
     LOG("Packet stream: %d", pkt.stream_index); 
     if (ret < 0) 
      break; 

     in_stream = ifmt_ctx->streams[pkt.stream_index]; 
     out_stream = ofmt_ctx->streams[pkt.stream_index]; 

     if (av_q2d(in_stream->time_base) * pkt.pts > end_seconds) { 
      av_packet_unref(&pkt); 
      break; 
     } 

     if (dts_start_from[pkt.stream_index] == 0) { 
      dts_start_from[pkt.stream_index] = pkt.dts; 
      printf("dts_start_from: %s\n", av_ts_make_string((char[AV_TS_MAX_STRING_SIZE]){0},dts_start_from[pkt.stream_index])); 
     } 
     if (pts_start_from[pkt.stream_index] == 0) { 
      pts_start_from[pkt.stream_index] = pkt.pts; 
      printf("pts_start_from: %s\n", av_ts_make_string((char[AV_TS_MAX_STRING_SIZE]){0},pts_start_from[pkt.stream_index])); 
     } 

     /* copy packet */ 
     pkt.pts = ::av_rescale_q_rnd(pkt.pts - pts_start_from[pkt.stream_index], in_stream->time_base, out_stream->time_base, (AVRounding) (AV_ROUND_NEAR_INF | 
                                      AV_ROUND_PASS_MINMAX)); 
     pkt.dts = ::av_rescale_q_rnd(pkt.dts - dts_start_from[pkt.stream_index], in_stream->time_base, out_stream->time_base, (AVRounding) (AV_ROUND_NEAR_INF | 
                                      AV_ROUND_PASS_MINMAX)); 
     if (pkt.pts < 0) { 
      pkt.pts = 0; 
     } 
     if (pkt.dts < 0) { 
      pkt.dts = 0; 
     } 
     pkt.duration = (int) av_rescale_q((int64_t) pkt.duration, in_stream->time_base, out_stream->time_base); 
     pkt.pos = -1; 
     printf("\n"); 

     ret = av_interleaved_write_frame(ofmt_ctx, &pkt); 
     if (ret < 0) { 
      LOG("Error muxing packet\n"); 
      break; 
     } 
     av_packet_unref(&pkt); 
    } 
    free(dts_start_from); 
    free(pts_start_from); 

    av_write_trailer(ofmt_ctx); 

    end: 
    LOG("END"); 

    avformat_close_input(&ifmt_ctx); 

    // Close output 
    if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE)) 
     avio_closep(&ofmt_ctx->pb); 
    avformat_free_context(ofmt_ctx); 

    if (ret < 0 && ret != AVERROR_EOF) { 
     LOG("-- Error occurred: %s\n", av_err2str(ret)); 
     return 1; 
    } 
} 

回答

1

对于帧粒度remuxing(不只是开始在一个关键帧),至少是您需要对关键帧进行编码以替换您开始使用的非关键帧,但是如果该流使用B帧(对于使用h264/etc的mp4通用),那么它可能不够用。最安全的方法是重新编码你想要开始的帧,直到(但不包括)下一个关键帧,然后简单重复 - 从现有关键帧开始复制所有内容。

+0

所以为了确认我明白,在av_seek_frame调用之后,我应该重新编码所有收到的视频数据包,直到我得到一个关键帧,此时我可以恢复remux复制?谢谢。 – DweebsUnited

+0

我不知道'av_seek_frame'如何在当前的ffmpeg上工作,所以我不想给你错误的建议。如果它通过寻找上一个关键帧并解码到输出丢失的期望帧,然后我认为你是对的。如果没有,您可能需要手动回溯到上一个关键帧并丢弃已解码的帧,直到找到所需的帧。 –

+0

所以,事实证明,这个视频文件只是没有任何IDR帧,这意味着我正在remuxing正确。在包含IDR帧的视频上运行它完美无缺。 TIL扩展我的测试集,而不相信一个样本。感谢您的帮助。 – DweebsUnited