Index: webrtc/test/layer_filtering_transport.h |
diff --git a/webrtc/test/layer_filtering_transport.h b/webrtc/test/layer_filtering_transport.h |
index 32eb4c5a305c08d7889941ed5db33c7f25a5d68b..d1371cefb0eed8d5686f7a1d9b509aeb5665e063 100644 |
--- a/webrtc/test/layer_filtering_transport.h |
+++ b/webrtc/test/layer_filtering_transport.h |
@@ -10,18 +10,22 @@ |
#ifndef WEBRTC_TEST_LAYER_FILTERING_TRANSPORT_H_ |
#define WEBRTC_TEST_LAYER_FILTERING_TRANSPORT_H_ |
+#include <map> |
+#include <memory> |
+ |
#include "webrtc/call/call.h" |
#include "webrtc/test/direct_transport.h" |
#include "webrtc/test/fake_network_pipe.h" |
#include "webrtc/test/single_threaded_task_queue.h" |
-#include <map> |
- |
namespace webrtc { |
namespace test { |
-class LayerFilteringTransport : public test::DirectTransport { |
+class RtpFileWriter; |
+ |
+class LayerFilteringTransport : public test::DirectTransport, |
+ private PacketReceiver { |
public: |
LayerFilteringTransport(SingleThreadedTaskQueueForTesting* task_queue, |
const FakeNetworkPipe::Config& config, |
@@ -30,13 +34,20 @@ class LayerFilteringTransport : public test::DirectTransport { |
uint8_t vp9_video_payload_type, |
int selected_tl, |
int selected_sl, |
- const std::map<uint8_t, MediaType>& payload_type_map); |
+ const std::map<uint8_t, MediaType>& payload_type_map, |
+ std::unique_ptr<test::RtpFileWriter> rtp_file_writer); |
bool DiscardedLastPacket() const; |
bool SendRtp(const uint8_t* data, |
size_t length, |
const PacketOptions& options) override; |
+ void SetReceiver(PacketReceiver* receiver) override; |
private: |
+ DeliveryStatus DeliverPacket(MediaType media_type, |
+ const uint8_t* packet, |
+ size_t length, |
+ const PacketTime& packet_time) override; |
+ |
// Used to distinguish between VP8 and VP9. |
const uint8_t vp8_video_payload_type_; |
const uint8_t vp9_video_payload_type_; |
@@ -45,6 +56,10 @@ class LayerFilteringTransport : public test::DirectTransport { |
const int selected_tl_; |
const int selected_sl_; |
bool discarded_last_packet_; |
+ |
+ PacketReceiver* receiver_; |
+ const int64_t start_ms_; |
+ std::unique_ptr<test::RtpFileWriter> rtp_file_writer_; |
}; |
} // namespace test |