1e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org/* 2e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 3e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * 4e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * Use of this source code is governed by a BSD-style license 5e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * that can be found in the LICENSE file in the root of the source 6e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * tree. An additional intellectual property rights grant can be found 7e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * in the file PATENTS. All contributing project authors may 8e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org * be found in the AUTHORS file in the root of the source tree. 9e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org */ 10e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 11e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ 12e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#define WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ 13e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 14ff761fba8274d93bd73e76c8b8a1f2d0776dd840Henrik Kjellander#include "webrtc/modules/include/module_common_types.h" 15527f6c62fc63d1f3829409a7822dc81983d1db86phoglund@webrtc.org#include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" 16e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 17e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgnamespace webrtc { 18e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 19d324546ced76d4e792338af4f7d02a5cd8819f92pkasting@chromium.orgconst uint8_t kFecPayloadType = 96; 20d324546ced76d4e792338af4f7d02a5cd8819f92pkasting@chromium.orgconst uint8_t kRedPayloadType = 97; 21d324546ced76d4e792338af4f7d02a5cd8819f92pkasting@chromium.orgconst uint8_t kVp8PayloadType = 120; 22e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 23e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgtypedef ForwardErrorCorrection::Packet Packet; 24e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 25e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgstruct RtpPacket : public Packet { 26e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org WebRtcRTPHeader header; 27e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org}; 28e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 29e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgclass FrameGenerator { 30e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org public: 31e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org FrameGenerator(); 32e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 33e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org void NewFrame(int num_packets); 34e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 35e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint16_t NextSeqNum(); 36e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 37e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* NextPacket(int offset, size_t length); 38e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 39e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // Creates a new RtpPacket with the RED header added to the packet. 40e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* BuildMediaRedPacket(const RtpPacket* packet); 41e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 42e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // Creates a new RtpPacket with FEC payload and red header. Does this by 43e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // creating a new fake media RtpPacket, clears the marker bit and adds a RED 44e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // header. Finally replaces the payload with the content of |packet->data|. 45e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* BuildFecRedPacket(const Packet* packet); 46e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 47e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org void SetRedHeader(Packet* red_packet, uint8_t payload_type, 484591fbd09f9cb6e83433c49a12dd8524c2806502pkasting@chromium.org size_t header_length) const; 49e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 50e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org private: 51af5ffd5bb9e0afc886c11c4d8c8b0ee4403ab802stefan@webrtc.org static void BuildRtpHeader(uint8_t* data, const RTPHeader* header); 52e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 53e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org int num_packets_; 54e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint16_t seq_num_; 55e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint32_t timestamp_; 56e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org}; 576a6f0893dd1e653410ba4b22e7f33947d15aeb65danilchap} // namespace webrtc 58e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 59e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ 60