H264转MP4
  sdKhIpc8YPFM 2023年11月05日 26 0


函数体

void h264toMp4(const char *fileName, const char *savePath)
{
  int frame_index=0;//统计帧数
  int inVStreamIndex=-1,outVStreamIndex=-1;//输入输出视频流在文件中的索引位置

  //const char inVFileName[1000];
  //strcpy(inVFileName, fileName);

  //char* str1 = strtok(inVFileName, "/");
  //char* fileTmpName;
  //printf("%s\n", str1);

  //int k = 0;
  //char outFile[1000] = { 0 };

  //if (savePath != NULL) {
  //  char outTmpFile[1000] = { 0 };

  //  while (str1 != NULL)
  //  {
  //    str1 = strtok(NULL, "/");
  //    printf("%s\n", str1);
  //    if (str1 != NULL) {
  //      fileTmpName = str1;
  //    }
  //  }

  //  k = strlen(fileTmpName);
  //  memcpy(outTmpFile, fileTmpName, k - 4);
  //  strcat(outTmpFile, "mp4");
  //  strcat(outFile, savePath);
  //  strcat(outFile, "/");
  //  strcat(outFile, outTmpFile);
  //}
  //else 
  //{
  //  k = strlen(fileName);
  //  memcpy(outFile, fileName, k - 4);
  //  strcat(outFile, "mp4");
  //}

  const char *outFileName = savePath;

  AVFormatContext *inVFmtCtx=NULL,*outFmtCtx=NULL;
  AVCodecParameters *codecPara=NULL;
  AVStream *outVStream=NULL;
  const AVCodec *outCodec=NULL;
  AVCodecContext *outCodecCtx=NULL;
  AVCodecParameters *outCodecPara=NULL;
  AVPacket *pkt=av_packet_alloc();

  do{
    //======================输入部分============================//
    //打开输入文件
    if(avformat_open_input(&inVFmtCtx, fileName,NULL,NULL)<0){
      printf("Cannot open input file.\n");
      break;
    }

    //查找输入文件中的流
    if(avformat_find_stream_info(inVFmtCtx,NULL)<0){
      printf("Cannot find stream info in input file.\n");
      break;
    }

    //查找视频流在文件中的位置
    for(size_t i=0;i<inVFmtCtx->nb_streams;i++){
      if(inVFmtCtx->streams[i]->codecpar->codec_type==AVMEDIA_TYPE_VIDEO){
        inVStreamIndex=(int)i;
        break;
      }
    }

    codecPara = inVFmtCtx->streams[inVStreamIndex]->codecpar;//输入视频流的编码参数

    printf("===============Input information========>\n");
    av_dump_format(inVFmtCtx, 0, fileName, 0);
    printf("===============Input information========<\n");

    //=====================输出部分=========================//
    //打开输出文件并填充格式数据
    if(avformat_alloc_output_context2(&outFmtCtx,NULL,NULL,outFileName)<0){
      printf("Cannot alloc output file context.\n");
      break;
    }

    //打开输出文件并填充数据
    if(avio_open(&outFmtCtx->pb,outFileName,AVIO_FLAG_READ_WRITE)<0){
      printf("output file open failed.\n");
      break;
    }

    //在输出的mp4文件中创建一条视频流
    outVStream = avformat_new_stream(outFmtCtx,NULL);
    if(!outVStream){
      printf("Failed allocating output stream.\n");
      break;
    }
    outVStream->time_base.den=30;
    outVStream->time_base.num=1;
    outVStreamIndex=outVStream->index;

    //查找编码器
    outCodec = avcodec_find_encoder(codecPara->codec_id);
    if(outCodec==NULL){
      printf("Cannot find any encoder.\n");
      break;
    }

    //从输入的h264编码器数据复制一份到输出文件的编码器中
    outCodecCtx=avcodec_alloc_context3(outCodec);
    outCodecPara = outFmtCtx->streams[outVStream->index]->codecpar;
    if(avcodec_parameters_copy(outCodecPara,codecPara)<0){
      printf("Cannot copy codec para.\n");
      break;
    }
    if(avcodec_parameters_to_context(outCodecCtx,outCodecPara)<0){
      printf("Cannot alloc codec ctx from para.\n");
      break;
    }
    outCodecCtx->time_base.den = 30;
    outCodecCtx->time_base.num=1;

    //打开输出文件需要的编码器
    if(avcodec_open2(outCodecCtx,outCodec,NULL)<0){
      printf("Cannot open output codec.\n");
      break;
    }

    printf("============Output Information=============>\n");
    av_dump_format(outFmtCtx,0,outFileName,1);
    printf("============Output Information=============<\n");

    //写入文件头
    if(avformat_write_header(outFmtCtx,NULL)<0){
      printf("Cannot write header to file.\n");
      return;
    }

    //===============编码部分===============//
    while(av_read_frame(inVFmtCtx,pkt)>=0){//循环读取每一帧直到读完
      if(pkt->stream_index==inVStreamIndex){//确保处理的是视频流
        //FIXME:No PTS (Example: Raw H.264)
        //Simple Write PTS
        //如果当前处理帧的显示时间戳为0或者没有等等不是正常值
        if(pkt->pts==AV_NOPTS_VALUE){
          printf("frame_index:%d\n", frame_index);
          //Write PTS
          //Duration between 2 frames (us)
          //Parameters
          pkt->pts = (double)((timeArr[frame_index] - timeArr[0]) * AV_TIME_BASE);
          pkt->dts = pkt->pts;
          if (frame_index != number - 1) {
            pkt->duration = ((timeArr[frame_index+1] - timeArr[frame_index]) * AV_TIME_BASE);
          }
          else {
            pkt->duration = 400000;
          }
          frame_index++;
        }
        AVRational time_base;
        time_base.num = 1;
        time_base.den = AV_TIME_BASE;
        //Convert PTS/DTS
        pkt->pts = av_rescale_q_rnd(pkt->pts, time_base, outVStream->time_base, (enum AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt->dts = av_rescale_q_rnd(pkt->dts, time_base, outVStream->time_base, (enum AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt->duration = av_rescale_q(pkt->duration, time_base, outVStream->time_base);
        pkt->pos = -1;
        pkt->stream_index = outVStreamIndex;
        printf("Write 1 Packet. size:%5d\tpts:%I64d\n", pkt->size, pkt->pts);
        //Write
        if (av_interleaved_write_frame(outFmtCtx, pkt) < 0) {
          printf("Error muxing packet\n");

          break;
        }
        av_packet_unref(pkt);
      }
    }

    av_write_trailer(outFmtCtx);
  }while(0);

  //=================释放所有指针=======================
  av_packet_free(&pkt);
  avformat_close_input(&inVFmtCtx);
  avformat_close_input(&outFmtCtx);
  avcodec_free_context(&outCodecCtx);

}


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
  8Tw5Riv1mGFK   2024年05月01日   80   0   0 C++
  BYaHC1OPAeY4   2024年05月08日   56   0   0 C++
  yZdUbUDB8h5t   2024年04月29日   59   0   0 C++
  yZdUbUDB8h5t   2024年05月05日   43   0   0 C++
  oXKBKZoQY2lx   2024年05月17日   57   0   0 C++
sdKhIpc8YPFM