blob: a7cb216bc6aeda3b60c88c5d3abf6923cf88210e [file] [log] [blame]
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001/*
kjellander1afca732016-02-07 20:46:45 -08002 * Copyright (c) 2004 The WebRTC project authors. All Rights Reserved.
henrike@webrtc.org28e20752013-07-10 00:45:36 +00003 *
kjellander1afca732016-02-07 20:46:45 -08004 * 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.
henrike@webrtc.org28e20752013-07-10 00:45:36 +00009 */
10
kjellandera96e2d72016-02-04 23:52:28 -080011#ifndef WEBRTC_MEDIA_BASE_TESTUTILS_H_
12#define WEBRTC_MEDIA_BASE_TESTUTILS_H_
henrike@webrtc.org28e20752013-07-10 00:45:36 +000013
14#include <string>
15#include <vector>
wu@webrtc.orgcadf9042013-08-30 21:24:16 +000016
kjellandera96e2d72016-02-04 23:52:28 -080017#include "webrtc/media/base/mediachannel.h"
18#include "webrtc/media/base/videocapturer.h"
19#include "webrtc/media/base/videocommon.h"
Edward Lemurc20978e2017-07-06 19:44:34 +020020#include "webrtc/rtc_base/arraysize.h"
21#include "webrtc/rtc_base/basictypes.h"
22#include "webrtc/rtc_base/sigslot.h"
23#include "webrtc/rtc_base/window.h"
henrike@webrtc.org28e20752013-07-10 00:45:36 +000024
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +000025namespace rtc {
jbauchf1f87202016-03-30 06:43:37 -070026class ByteBufferReader;
27class ByteBufferWriter;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000028class StreamInterface;
29}
30
nisseacd935b2016-11-11 03:55:13 -080031namespace webrtc {
32class VideoFrame;
33}
34
henrike@webrtc.org28e20752013-07-10 00:45:36 +000035namespace cricket {
36
37// Returns size of 420 image with rounding on chroma for odd sizes.
38#define I420_SIZE(w, h) (w * h + (((w + 1) / 2) * ((h + 1) / 2)) * 2)
39// Returns size of ARGB image.
40#define ARGB_SIZE(w, h) (w * h * 4)
41
42template <class T> inline std::vector<T> MakeVector(const T a[], size_t s) {
43 return std::vector<T>(a, a + s);
44}
tfarina5237aaf2015-11-10 23:44:30 -080045#define MAKE_VECTOR(a) cricket::MakeVector(a, arraysize(a))
henrike@webrtc.org28e20752013-07-10 00:45:36 +000046
47struct RtpDumpPacket;
48class RtpDumpWriter;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000049
50struct RawRtpPacket {
jbauchf1f87202016-03-30 06:43:37 -070051 void WriteToByteBuffer(uint32_t in_ssrc, rtc::ByteBufferWriter* buf) const;
52 bool ReadFromByteBuffer(rtc::ByteBufferReader* buf);
henrike@webrtc.org28e20752013-07-10 00:45:36 +000053 // Check if this packet is the same as the specified packet except the
54 // sequence number and timestamp, which should be the same as the specified
55 // parameters.
Peter Boström0c4e06b2015-10-07 12:23:21 +020056 bool SameExceptSeqNumTimestampSsrc(const RawRtpPacket& packet,
57 uint16_t seq,
58 uint32_t ts,
59 uint32_t ssc) const;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000060 int size() const { return 28; }
61
Peter Boström0c4e06b2015-10-07 12:23:21 +020062 uint8_t ver_to_cc;
63 uint8_t m_to_pt;
64 uint16_t sequence_number;
65 uint32_t timestamp;
66 uint32_t ssrc;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000067 char payload[16];
68};
69
70struct RawRtcpPacket {
jbauchf1f87202016-03-30 06:43:37 -070071 void WriteToByteBuffer(rtc::ByteBufferWriter* buf) const;
72 bool ReadFromByteBuffer(rtc::ByteBufferReader* buf);
henrike@webrtc.org28e20752013-07-10 00:45:36 +000073 bool EqualsTo(const RawRtcpPacket& packet) const;
74
Peter Boström0c4e06b2015-10-07 12:23:21 +020075 uint8_t ver_to_count;
76 uint8_t type;
77 uint16_t length;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000078 char payload[16];
79};
80
henrike@webrtc.org28e20752013-07-10 00:45:36 +000081// Test helper for testing VideoCapturer implementations.
nisse79246972016-08-23 05:50:09 -070082class VideoCapturerListener
83 : public sigslot::has_slots<>,
nisseacd935b2016-11-11 03:55:13 -080084 public rtc::VideoSinkInterface<webrtc::VideoFrame> {
henrike@webrtc.org28e20752013-07-10 00:45:36 +000085 public:
86 explicit VideoCapturerListener(VideoCapturer* cap);
nisse79246972016-08-23 05:50:09 -070087 ~VideoCapturerListener();
henrike@webrtc.org28e20752013-07-10 00:45:36 +000088
89 CaptureState last_capture_state() const { return last_capture_state_; }
90 int frame_count() const { return frame_count_; }
henrike@webrtc.org28e20752013-07-10 00:45:36 +000091 int frame_width() const { return frame_width_; }
92 int frame_height() const { return frame_height_; }
henrike@webrtc.org28e20752013-07-10 00:45:36 +000093 bool resolution_changed() const { return resolution_changed_; }
94
95 void OnStateChange(VideoCapturer* capturer, CaptureState state);
nisseacd935b2016-11-11 03:55:13 -080096 void OnFrame(const webrtc::VideoFrame& frame) override;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000097
98 private:
nisse79246972016-08-23 05:50:09 -070099 VideoCapturer* capturer_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000100 CaptureState last_capture_state_;
101 int frame_count_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000102 int frame_width_;
103 int frame_height_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000104 bool resolution_changed_;
105};
106
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000107class VideoMediaErrorCatcher : public sigslot::has_slots<> {
108 public:
109 VideoMediaErrorCatcher() : ssrc_(0), error_(VideoMediaChannel::ERROR_NONE) { }
Peter Boström0c4e06b2015-10-07 12:23:21 +0200110 uint32_t ssrc() const { return ssrc_; }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000111 VideoMediaChannel::Error error() const { return error_; }
Peter Boström0c4e06b2015-10-07 12:23:21 +0200112 void OnError(uint32_t ssrc, VideoMediaChannel::Error error) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000113 ssrc_ = ssrc;
114 error_ = error;
115 }
116 private:
Peter Boström0c4e06b2015-10-07 12:23:21 +0200117 uint32_t ssrc_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000118 VideoMediaChannel::Error error_;
119};
120
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000121// Checks whether |codecs| contains |codec|; checks using Codec::Matches().
122template <class C>
123bool ContainsMatchingCodec(const std::vector<C>& codecs, const C& codec) {
124 typename std::vector<C>::const_iterator it;
125 for (it = codecs.begin(); it != codecs.end(); ++it) {
126 if (it->Matches(codec)) {
127 return true;
128 }
129 }
130 return false;
131}
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000132
133// Create Simulcast StreamParams with given |ssrcs| and |cname|.
Peter Boström0c4e06b2015-10-07 12:23:21 +0200134cricket::StreamParams CreateSimStreamParams(const std::string& cname,
135 const std::vector<uint32_t>& ssrcs);
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000136// Create Simulcast stream with given |ssrcs| and |rtx_ssrcs|.
137// The number of |rtx_ssrcs| must match number of |ssrcs|.
138cricket::StreamParams CreateSimWithRtxStreamParams(
Peter Boström0c4e06b2015-10-07 12:23:21 +0200139 const std::string& cname,
140 const std::vector<uint32_t>& ssrcs,
141 const std::vector<uint32_t>& rtx_ssrcs);
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000142
brandtr9688e382016-11-22 00:59:48 -0800143// Create StreamParams with single primary SSRC and corresponding FlexFEC SSRC.
144cricket::StreamParams CreatePrimaryWithFecFrStreamParams(
145 const std::string& cname,
146 uint32_t primary_ssrc,
147 uint32_t flexfec_ssrc);
148
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000149} // namespace cricket
150
kjellandera96e2d72016-02-04 23:52:28 -0800151#endif // WEBRTC_MEDIA_BASE_TESTUTILS_H_