From 1dcd8433f52de0112affa735246dbe2d27ea58b2 Mon Sep 17 00:00:00 2001
From: imp_rayjay <32894715+rayjay214@users.noreply.github.com>
Date: Thu, 12 Oct 2023 16:35:27 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0mjpeg=E8=B4=9F=E8=BD=BD?=
 =?UTF-8?q?=E7=9A=84mp4=E6=96=87=E4=BB=B6=E7=82=B9=E6=92=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Record/MP4Demuxer.cpp | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/src/Record/MP4Demuxer.cpp b/src/Record/MP4Demuxer.cpp
index 02c66976..c63847f9 100644
--- a/src/Record/MP4Demuxer.cpp
+++ b/src/Record/MP4Demuxer.cpp
@@ -1,4 +1,4 @@
-/*
+/*
  * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
  *
  * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
@@ -16,8 +16,10 @@
 #include "Extension/AAC.h"
 #include "Extension/G711.h"
 #include "Extension/Opus.h"
-using namespace toolkit;
+#include "Extension/JPEG.h"
+
 using namespace std;
+using namespace toolkit;
 
 namespace mediakit {
 
@@ -105,8 +107,9 @@ void MP4Demuxer::onVideoTrack(uint32_t track, uint8_t object, int width, int hei
                     video->inputFrame(std::make_shared<H264FrameNoCacheAble>((char *)config, size, 0, 0,4));
                 }
             }
-        }
             break;
+        }
+
         case MOV_OBJECT_HEVC: {
             auto video = std::make_shared<H265Track>();
             _track_to_codec.emplace(track,video);
@@ -120,11 +123,16 @@ void MP4Demuxer::onVideoTrack(uint32_t track, uint8_t object, int width, int hei
                     video->inputFrame(std::make_shared<H265FrameNoCacheAble>((char *) config, size, 0, 0,4));
                 }
             }
+            break;
         }
+
+        case MOV_OBJECT_JPEG: {
+            auto video = std::make_shared<JPEGTrack>();
+            _track_to_codec.emplace(track,video);
             break;
-        default:
-            WarnL << "不支持该编码类型的MP4,已忽略:" << getObjectName(object);
-            break;
+        }
+
+        default: WarnL << "不支持该编码类型的MP4,已忽略:" << getObjectName(object); break;
     }
 }
 
@@ -243,6 +251,11 @@ Frame::Ptr MP4Demuxer::makeFrame(uint32_t track_id, const Buffer::Ptr &buf, int6
             break;
         }
 
+        case CodecJPEG: {
+            ret = std::make_shared<JPEGFrame>(buf, (uint64_t)dts, 0, DATA_OFFSET);
+            break;
+        }
+
         case CodecAAC: {
             AACTrack::Ptr track = dynamic_pointer_cast<AACTrack>(it->second);
             assert(track);
@@ -283,4 +296,4 @@ uint64_t MP4Demuxer::getDurationMS() const {
 }
 
 }//namespace mediakit
-#endif// ENABLE_MP4
\ No newline at end of file
+#endif// ENABLE_MP4