improve video fec

src/WBStreamRx.cpp

@@ -151,3 +151,7 @@ void WBStreamRx::reset_stream_stats() {
   m_n_input_bytes=0;
   m_n_input_packets=0;
 }
+
+void WBStreamRx::set_on_fec_block_done_cb(WBStreamRx::ON_BLOCK_DONE_CB cb) {
+  m_fec_decoder->m_block_done_cb=cb;
+}

src/WBStreamRx.h

@@ -72,6 +72,10 @@ class WBStreamRx {
   };
   FECRxStats2 get_latest_fec_stats();
   void reset_stream_stats();
+  // DIRTY AF !
+  typedef std::function<void(uint64_t block_idx,int n_fragments_total,int n_fragments_forwarded)>
+      ON_BLOCK_DONE_CB;
+  void set_on_fec_block_done_cb(ON_BLOCK_DONE_CB cb);
  private:
   const Options m_options;
   std::shared_ptr<WBTxRx> m_txrx;

src/fec/FECDecoder.cpp

@@ -71,6 +71,12 @@ void FECDecoder::rxQueuePopFront() {
           block.get_missing_primary_packets_readable());
     }
   }
+  if(m_block_done_cb){
+    auto& block = *rx_queue.front();
+    const int n_p_fragments=block.get_n_primary_fragments();
+    const int n_p_fragments_forwarded=block.get_n_forwarded_primary_fragments();
+    m_block_done_cb(block.getBlockIdx(),n_p_fragments,n_p_fragments_forwarded);
+  }
   rx_queue.pop_front();
 }
 

src/fec/FECDecoder.h

@@ -43,6 +43,10 @@ class FECDecoder {
       SEND_DECODED_PACKET;
   // WARNING: Don't forget to register this callback !
   SEND_DECODED_PACKET mSendDecodedPayloadCallback;
+  // Experimental
+  typedef std::function<void(uint64_t block_idx,int n_fragments_total,int n_fragments_forwarded)>
+      ON_BLOCK_DONE_CB;
+  ON_BLOCK_DONE_CB m_block_done_cb= nullptr;
   // A value too high doesn't really give much benefit and increases memory
   // usage
   const unsigned int RX_QUEUE_MAX_SIZE;

src/fec/RxBlock.cpp

@@ -162,4 +162,7 @@ std::string RxBlock::get_missing_primary_packets_readable() const {
 
 int RxBlock::get_n_primary_fragments() const {
   return m_n_primary_fragments_in_block;
-}
+}
+int RxBlock::get_n_forwarded_primary_fragments() const {
+  return nAlreadyForwardedPrimaryFragments;
+}

src/fec/RxBlock.h

@@ -89,6 +89,7 @@ class RxBlock {
   std::optional<int> get_missing_primary_packets() const;
   std::string get_missing_primary_packets_readable() const;
   int get_n_primary_fragments() const;
+  int get_n_forwarded_primary_fragments()const;
 
  private:
   // the block idx marks which block this element refers to