OLD | NEW |
---|---|
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ | 11 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ |
12 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ | 12 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ |
13 | 13 |
14 #include "webrtc/base/basictypes.h" | |
15 #include "webrtc/base/random.h" | |
14 #include "webrtc/modules/include/module_common_types.h" | 16 #include "webrtc/modules/include/module_common_types.h" |
15 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" | 17 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" |
16 | 18 |
17 namespace webrtc { | 19 namespace webrtc { |
18 namespace test { | 20 namespace test { |
21 | |
22 // Needed to not clash with another webrtc::FrameGenerator. | |
23 namespace fec { | |
24 | |
19 struct RawRtpPacket : public ForwardErrorCorrection::Packet { | 25 struct RawRtpPacket : public ForwardErrorCorrection::Packet { |
20 WebRtcRTPHeader header; | 26 WebRtcRTPHeader header; |
21 }; | 27 }; |
22 } // namespace test | |
23 | |
24 const uint8_t kFecPayloadType = 96; | |
25 const uint8_t kRedPayloadType = 97; | |
26 const uint8_t kVp8PayloadType = 120; | |
27 | 28 |
28 class FrameGenerator { | 29 class FrameGenerator { |
29 public: | 30 public: |
30 FrameGenerator(); | 31 FrameGenerator(); |
31 | 32 |
32 void NewFrame(int num_packets); | 33 void NewFrame(int num_packets); |
33 | 34 |
34 uint16_t NextSeqNum(); | 35 uint16_t NextSeqNum(); |
35 | 36 |
36 test::RawRtpPacket* NextPacket(int offset, size_t length); | 37 RawRtpPacket* NextPacket(int offset, size_t length); |
37 | 38 |
38 // Creates a new RtpPacket with the RED header added to the packet. | 39 // Creates a new RtpPacket with the RED header added to the packet. |
39 test::RawRtpPacket* BuildMediaRedPacket(const test::RawRtpPacket* packet); | 40 RawRtpPacket* BuildMediaRedPacket(const RawRtpPacket* packet); |
40 | 41 |
41 // Creates a new RtpPacket with FEC payload and red header. Does this by | 42 // Creates a new RtpPacket with FEC payload and red header. Does this by |
42 // creating a new fake media RtpPacket, clears the marker bit and adds a RED | 43 // creating a new fake media RtpPacket, clears the marker bit and adds a RED |
43 // header. Finally replaces the payload with the content of |packet->data|. | 44 // header. Finally replaces the payload with the content of |packet->data|. |
44 test::RawRtpPacket* BuildFecRedPacket( | 45 RawRtpPacket* BuildFecRedPacket(const ForwardErrorCorrection::Packet* packet); |
45 const ForwardErrorCorrection::Packet* packet); | |
46 | 46 |
47 void SetRedHeader(ForwardErrorCorrection::Packet* red_packet, | 47 void SetRedHeader(ForwardErrorCorrection::Packet* red_packet, |
48 uint8_t payload_type, | 48 uint8_t payload_type, |
49 size_t header_length) const; | 49 size_t header_length) const; |
50 | 50 |
51 private: | 51 private: |
52 static void BuildRtpHeader(uint8_t* data, const RTPHeader* header); | 52 static void BuildRtpHeader(uint8_t* data, const RTPHeader* header); |
53 | 53 |
54 int num_packets_; | 54 int num_packets_; |
55 uint16_t seq_num_; | 55 uint16_t seq_num_; |
56 uint32_t timestamp_; | 56 uint32_t timestamp_; |
57 }; | 57 }; |
58 | |
59 class MediaPacketGenerator { | |
60 public: | |
61 MediaPacketGenerator(uint32_t min_packet_size, | |
62 uint32_t max_packet_size, | |
63 uint32_t ssrc, | |
64 Random* random) | |
65 : min_packet_size_(min_packet_size), | |
66 max_packet_size_(max_packet_size), | |
67 ssrc_(ssrc), | |
68 random_(random) {} | |
69 | |
70 // Construct the media packets, up to |num_media_packets| packets. | |
71 ForwardErrorCorrection::PacketList ConstructMediaPacketsSeqNum( | |
stefan-webrtc
2016/09/13 07:39:47
Does it have to be public?
brandtr
2016/09/13 09:27:14
Yep, see e.g. https://codereview.webrtc.org/226739
| |
72 int num_media_packets, | |
73 uint16_t start_seq_num); | |
74 ForwardErrorCorrection::PacketList ConstructMediaPackets( | |
75 int num_media_packets); | |
76 | |
77 uint16_t GetFecSeqNum(); | |
78 | |
79 private: | |
80 uint32_t min_packet_size_; | |
81 uint32_t max_packet_size_; | |
82 uint32_t ssrc_; | |
83 Random* random_; | |
84 | |
85 ForwardErrorCorrection::PacketList media_packets_; | |
86 uint16_t fec_seq_num_; | |
87 }; | |
88 | |
89 } // namespace fec | |
90 } // namespace test | |
58 } // namespace webrtc | 91 } // namespace webrtc |
59 | 92 |
60 #endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ | 93 #endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ |
OLD | NEW |