Combine Fu-A subframes into complete h264 frame

src/rtp/rtp_video_receiver.cpp

@@ -0,0 +1,76 @@
+#include "rtp_video_receiver.h"
+
+#include "log.h"
+
+#define NV12_BUFFER_SIZE (1280 * 720 * 3 / 2)
+
+RtpVideoReceiver::RtpVideoReceiver() {}
+
+RtpVideoReceiver::~RtpVideoReceiver() {}
+
+void RtpVideoReceiver::InsertRtpPacket(RtpPacket& rtp_packet) {
+  if (NAL_UNIT_TYPE::NALU == rtp_packet.NalUnitType()) {
+    // compelete_video_frame_queue_.push(
+    //     VideoFrame(rtp_packet.Payload(), rtp_packet.Size()));
+    if (on_receive_complete_frame_) {
+      on_receive_complete_frame_(
+          VideoFrame(rtp_packet.Payload(), rtp_packet.Size()));
+    }
+  } else if (NAL_UNIT_TYPE::FU_A == rtp_packet.NalUnitType()) {
+    incomplete_frame_list_[rtp_packet.SequenceNumber()] = rtp_packet;
+    bool complete = CheckIsFrameCompleted(rtp_packet);
+  }
+}
+
+bool RtpVideoReceiver::CheckIsFrameCompleted(RtpPacket& rtp_packet) {
+  if (rtp_packet.FuAEnd()) {
+    size_t complete_frame_size = 0;
+    uint16_t end_seq = rtp_packet.SequenceNumber();
+    if (incomplete_frame_list_.size() == end_seq) {
+      return true;
+    }
+
+    while (end_seq--) {
+      auto it = incomplete_frame_list_.find(end_seq);
+      complete_frame_size += it->second.PayloadSize();
+      if (it == incomplete_frame_list_.end()) {
+        return false;
+      } else if (!it->second.FuAStart()) {
+        continue;
+      } else if (it->second.FuAStart()) {
+        if (!nv12_data_) {
+          nv12_data_ = new uint8_t[NV12_BUFFER_SIZE];
+        }
+
+        size_t complete_frame_size = 0;
+        for (size_t start = it->first; start <= rtp_packet.SequenceNumber();
+             start++) {
+          memcpy(nv12_data_ + complete_frame_size,
+                 incomplete_frame_list_[start].Payload(),
+                 incomplete_frame_list_[start].PayloadSize());
+
+          complete_frame_size += incomplete_frame_list_[start].PayloadSize();
+          incomplete_frame_list_.erase(start);
+        }
+
+        // compelete_video_frame_queue_.push(
+        //     VideoFrame(nv12_data_, complete_frame_size));
+
+        // LOG_ERROR("Size of compelete_video_frame_queue_ [{}]",
+        //           compelete_video_frame_queue_.size());
+
+        if (on_receive_complete_frame_) {
+          on_receive_complete_frame_(
+              VideoFrame(nv12_data_, complete_frame_size));
+        }
+        return true;
+      } else {
+        LOG_WARN("What happened?")
+        return false;
+      }
+    }
+
+    return true;
+  }
+  return false;
+}