Home | History | Annotate | Download | only in source 
      1 /*
      2  *  Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
      3  *
      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
      6  *  tree. An additional intellectual property rights grant can be found
      7  *  in the file PATENTS.  All contributing project authors may
      8  *  be found in the AUTHORS file in the root of the source tree.
      9  */
     10 
     11 // This file contains the class RtpFormatVp8TestHelper. The class is
     12 // responsible for setting up a fake VP8 bitstream according to the
     13 // RTPVideoHeaderVP8 header, and partition information. After initialization,
     14 // an RTPFragmentationHeader is provided so that the tester can create a
     15 // packetizer. The packetizer can then be provided to this helper class, which
     16 // will then extract all packets and compare to the expected outcome.
     17 
     18 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP8_TEST_HELPER_H_
     19 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP8_TEST_HELPER_H_
     20 
     21 #include "webrtc/base/constructormagic.h"
     22 #include "webrtc/modules/include/module_common_types.h"
     23 #include "webrtc/modules/rtp_rtcp/source/rtp_format_vp8.h"
     24 #include "webrtc/typedefs.h"
     25 
     26 namespace webrtc {
     27 
     28 namespace test {
     29 
     30 class RtpFormatVp8TestHelper {
     31  public:
     32   explicit RtpFormatVp8TestHelper(const RTPVideoHeaderVP8* hdr);
     33   ~RtpFormatVp8TestHelper();
     34   bool Init(const size_t* partition_sizes, size_t num_partitions);
     35   void GetAllPacketsAndCheck(RtpPacketizerVp8* packetizer,
     36                              const size_t* expected_sizes,
     37                              const int* expected_part,
     38                              const bool* expected_frag_start,
     39                              size_t expected_num_packets);
     40 
     41   uint8_t* payload_data() const { return payload_data_; }
     42   size_t payload_size() const { return payload_size_; }
     43   RTPFragmentationHeader* fragmentation() const { return fragmentation_; }
     44   size_t buffer_size() const { return buffer_size_; }
     45   void set_sloppy_partitioning(bool value) { sloppy_partitioning_ = value; }
     46 
     47  private:
     48   void CheckHeader(bool frag_start);
     49   void CheckPictureID();
     50   void CheckTl0PicIdx();
     51   void CheckTIDAndKeyIdx();
     52   void CheckPayload(size_t payload_end);
     53   void CheckLast(bool last) const;
     54   void CheckPacket(size_t send_bytes, size_t expect_bytes, bool last,
     55                    bool frag_start);
     56 
     57   uint8_t* payload_data_;
     58   uint8_t* buffer_;
     59   uint8_t* data_ptr_;
     60   RTPFragmentationHeader* fragmentation_;
     61   const RTPVideoHeaderVP8* hdr_info_;
     62   int payload_start_;
     63   size_t payload_size_;
     64   size_t buffer_size_;
     65   bool sloppy_partitioning_;
     66   bool inited_;
     67 
     68   RTC_DISALLOW_COPY_AND_ASSIGN(RtpFormatVp8TestHelper);
     69 };
     70 
     71 }  // namespace test
     72 
     73 }  // namespace webrtc
     74 
     75 #endif  // WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP8_TEST_HELPER_H_
     76