fec_test_helper.h revision af5ffd5bb9e0afc886c11c4d8c8b0ee4403ab802
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 14e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#include "modules/interface/module_common_types.h" 15e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#include "modules/rtp_rtcp/source/forward_error_correction.h" 16e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 17e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgnamespace webrtc { 18e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 19e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgenum { kRtpHeaderSize = 12 }; 20e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgenum { kFecPayloadType = 96 }; 21e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgenum { kRedPayloadType = 97 }; 22e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgenum { kVp8PayloadType = 120 }; 23e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 24e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgtypedef ForwardErrorCorrection::Packet Packet; 25e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 26e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgstruct RtpPacket : public Packet { 27e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org WebRtcRTPHeader header; 28e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org}; 29e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 30e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.orgclass FrameGenerator { 31e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org public: 32e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org FrameGenerator(); 33e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 34e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org void NewFrame(int num_packets); 35e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 36e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint16_t NextSeqNum(); 37e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 38e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* NextPacket(int offset, size_t length); 39e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 40e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // Creates a new RtpPacket with the RED header added to the packet. 41e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* BuildMediaRedPacket(const RtpPacket* packet); 42e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 43e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // Creates a new RtpPacket with FEC payload and red header. Does this by 44e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // creating a new fake media RtpPacket, clears the marker bit and adds a RED 45e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org // header. Finally replaces the payload with the content of |packet->data|. 46e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org RtpPacket* BuildFecRedPacket(const Packet* packet); 47e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 48e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org void SetRedHeader(Packet* red_packet, uint8_t payload_type, 49e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org int header_length) const; 50e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 51e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org private: 52af5ffd5bb9e0afc886c11c4d8c8b0ee4403ab802stefan@webrtc.org static void BuildRtpHeader(uint8_t* data, const RTPHeader* header); 53e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 54e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org int num_packets_; 55e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint16_t seq_num_; 56e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org uint32_t timestamp_; 57e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org}; 58e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org} 59e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org 60e0d6fa4c66e025054e199a42f91a8b6d8af282b5stefan@webrtc.org#endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_FEC_TEST_HELPER_H_ 61