fix(avcodecDecoder): update pts on decode

Signed-off-by: pingkai <pingkai010@gmail.com>
alibaba · Feb 24, 2020 · b68426b · b68426b
1 parent 592e622
commit b68426b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/framework/base/media/AVAFPacket.cpp b/framework/base/media/AVAFPacket.cpp
@@ -69,7 +69,7 @@ int64_t AVAFPacket::getSize()
     return mpkt->size;
 }
 
-const AVPacket *AVAFPacket::ToAVPacket()
+AVPacket *AVAFPacket::ToAVPacket()
 {
     return mpkt;
 }
@@ -100,7 +100,7 @@ AVAFFrame::AVAFFrame(AVFrame **frame, IAFFrame::FrameType type) : mType(type)
 
 
 AVAFFrame::AVAFFrame(AVFrame *frame, FrameType type) : mAvFrame(av_frame_clone(frame)),
-    mType(type)
+                                                       mType(type)
 {
     assert(mAvFrame != nullptr);
     copyInfo();

diff --git a/framework/base/media/AVAFPacket.h b/framework/base/media/AVAFPacket.h
@@ -30,7 +30,7 @@ class AVAFPacket : public IAFPacket {
 
     int64_t getSize() override;
 
-    const AVPacket *ToAVPacket();
+    AVPacket *ToAVPacket();
 
     explicit operator AVPacket *();
 

diff --git a/framework/codec/avcodecDecoder.cpp b/framework/codec/avcodecDecoder.cpp
@@ -251,7 +251,7 @@ namespace Cicada {
     int avcodecDecoder::enqueue_decoder(unique_ptr<IAFPacket> &pPacket)
     {
         int ret;
-        const AVPacket *pkt = nullptr;
+        AVPacket *pkt = nullptr;
 
         if (pPacket) {
             auto *avAFPacket = dynamic_cast<AVAFPacket *>(pPacket.get());
@@ -261,6 +261,8 @@ namespace Cicada {
                 // TODO: tobe impl
             } else {
                 pkt = avAFPacket->ToAVPacket();
+                pkt->pts = pPacket->getInfo().pts;
+                pkt->dts = pPacket->getInfo().dts;
                 assert(pkt != nullptr);
             }
         }