Add HandleStreamStopInfo helper

jp9000 · Sep 24, 2014 · 3caa30c · 3caa30c
1 parent 8919fcb
commit 3caa30c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/Source/OBS.h b/Source/OBS.h
@@ -637,6 +637,16 @@ class OBS
 
     //---------------------------------------------------
 
+    struct StopInfo
+    {
+        DWORD time = (DWORD)-1;
+        bool timeSeen = false;
+        std::function<void()> func;
+    };
+    bool HandleStreamStopInfo(StopInfo &, PacketType, const VideoSegment&);
+
+    //---------------------------------------------------
+
     std::unique_ptr<NetworkStream> network;
 
     //---------------------------------------------------

diff --git a/Source/OBSVideoCapture.cpp b/Source/OBSVideoCapture.cpp
@@ -222,6 +222,25 @@ void OBS::SendFrame(VideoSegment &curSegment, QWORD firstFrameTime)
     }
 }
 
+bool OBS::HandleStreamStopInfo(OBS::StopInfo &info, PacketType type, const VideoSegment& segment)
+{
+    if (type == PacketType_Audio || !info.func)
+        return false;
+
+    if (segment.pts < info.time)
+        return false;
+
+    if (!info.timeSeen)
+    {
+        info.timeSeen = true;
+        return false;
+    }
+
+    info.func();
+    info = {};
+    return true;
+}
+
 bool OBS::ProcessFrame(FrameProcessInfo &frameInfo)
 {
     List<DataPacket> videoPackets;