From a8b447be656145c9ba2a2d8319a10ae8f726de1f Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 09 十月 2020 18:20:46 +0800
Subject: [PATCH] h264 mp4

---
 csrc/ffmpeg/format/FormatOut.cpp |   29 ++++++++++++++++++++---------
 1 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/csrc/ffmpeg/format/FormatOut.cpp b/csrc/ffmpeg/format/FormatOut.cpp
index 2994153..4d6ce60 100644
--- a/csrc/ffmpeg/format/FormatOut.cpp
+++ b/csrc/ffmpeg/format/FormatOut.cpp
@@ -38,6 +38,7 @@
 	FormatOut::~FormatOut()
 	{
 		clear();
+
 	}
 
     void FormatOut::clear(){
@@ -293,7 +294,7 @@
             logIt("Failed to copy context from input to output stream codec context\n");
             return false;
         }
-        out_stream->codecpar->codec_tag = out_stream->codec->codec_tag = 0;
+        out_stream->codecpar->codec_tag = 0;
         
         if(ctx_->oformat->flags & AVFMT_GLOBALHEADER)
             out_stream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
@@ -349,7 +350,7 @@
             flag = writeHeader(&avdic);
             av_dict_free(&avdic);
         }
-        
+
         return flag;
     }
     
@@ -399,7 +400,7 @@
         int out_idx = -1;
         std::vector<AVStream*> in_streams{in_v_stream_, in_a_stream_};
         for (auto i : in_streams){
-            if (i->index == pkt->stream_index){
+            if (i && (i->index == pkt->stream_index)){
                 if (i->codecpar->codec_type == AVMEDIA_TYPE_VIDEO){
                     out_idx = v_idx_;
                     in_stream = i;
@@ -434,13 +435,19 @@
             pkt->pts = pkt->dts = pkt->duration * time_stamp; 
 
         }
+        // logIt("BEFORE in stream timebase %d:%d, out timebase %d:%d, 
+        //         pts: %lld, dts: %lld, duration: %lld", 
+        //     in_stream->time_base.num, in_stream->time_base.den,
+        //     out_stream->time_base.num, out_stream->time_base.den,
+        //     pkt->pts, pkt->dts, pkt->duration);
+
     }
 
     bool FormatOut::writeFrame(AVPacket *pkt, const int64_t &frame_cnt,
                               bool interleaved/* = true*/){
 
         adjustPTS(pkt, frame_cnt);
-        auto ret = writeFrame2(pkt, interleaved);
+        auto ret = writeFrameInternal(pkt, interleaved);
         if (!ret){
             logIt("write to file failed, pkt.pts: %lld, dts: %lld, frame count: %d",
                     pkt->pts, pkt->dts, frame_cnt);
@@ -448,16 +455,15 @@
         return ret;
     }
 
-    bool FormatOut::writeFrame2(AVPacket *pkt, bool interleaved){
-        
+    static bool write_frame(AVFormatContext *ctx, AVPacket *pkt, bool interleaved){
         int ret = 0;
         if(interleaved){
-            ret = av_interleaved_write_frame(ctx_, pkt);
+            ret = av_interleaved_write_frame(ctx, pkt);
         }else{
             // returns 1 if flushed and there is no more data to flush
-            ret = av_write_frame(ctx_, pkt);
+            ret = av_write_frame(ctx, pkt);
         }
-    
+
         if(ret < -22 || ret == 0){
             return true;
         }
@@ -465,6 +471,11 @@
         return false;
     }
 
+    bool FormatOut::writeFrameInternal(AVPacket *pkt, bool interleaved){
+
+        return write_frame(ctx_, pkt, interleaved);
+    }
+
     bool FormatOut::writeTrailer(){
         const int ret = av_write_trailer(ctx_);
         if(ret != 0)

--
Gitblit v1.8.0