Add FlexfecPacketGenerator. (pt. 9)

webrtc/modules/rtp_rtcp/source/fec_test_helper.h

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
@@ skipping 70 matching lines @@
 
   // Number of packets left to generate, in the current frame.
   size_t num_packets_;
 
  private:
   uint32_t ssrc_;
   uint16_t seq_num_;
   uint32_t timestamp_;
 };
 
+// This class generates media and FlexFEC packets for a single frame.
+class FlexfecPacketGenerator : public AugmentedPacketGenerator {
+ public:
+  FlexfecPacketGenerator(uint32_t media_ssrc, uint32_t flexfec_ssrc);
+
+  // Creates a new AugmentedPacket (with RTP headers) from a
+  // FlexFEC packet (without RTP headers).
+  std::unique_ptr<AugmentedPacket> BuildFlexfecPacket(
+      const ForwardErrorCorrection::Packet& packet);
+
+ private:
+  uint32_t flexfec_ssrc_;
+  uint16_t flexfec_seq_num_;
+  uint32_t flexfec_timestamp_;
+};
+
 // This class generates media and ULPFEC packets (both encapsulated in RED)
 // for a single frame.
 class UlpfecPacketGenerator : public AugmentedPacketGenerator {
  public:
   explicit UlpfecPacketGenerator(uint32_t ssrc);
 
   // Creates a new AugmentedPacket with the RED header added to the packet.
   static std::unique_ptr<AugmentedPacket> BuildMediaRedPacket(
       const AugmentedPacket& packet);
 
   // Creates a new AugmentedPacket with FEC payload and RED header. Does this by
   // creating a new fake media AugmentedPacket, clears the marker bit and adds a
   // RED header. Finally replaces the payload with the content of
   // |packet->data|.
   std::unique_ptr<AugmentedPacket> BuildUlpfecRedPacket(
       const ForwardErrorCorrection::Packet& packet);
 
  private:
   static void SetRedHeader(uint8_t payload_type,
                            size_t header_length,
                            AugmentedPacket* red_packet);
 };
 
 }  // namespace fec
 }  // namespace test
 }  // namespace webrtc
 
 #endif  // WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_