Index: webrtc/test/frame_generator.cc |
diff --git a/webrtc/test/frame_generator.cc b/webrtc/test/frame_generator.cc |
index a6afce01d906ee20f78920c208a50f1e7fbe49c3..18376cca9b553726fb3e76058209d41080b8ada4 100644 |
--- a/webrtc/test/frame_generator.cc |
+++ b/webrtc/test/frame_generator.cc |
@@ -258,6 +258,7 @@ void FrameForwarder::AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink, |
rtc::CritScope lock(&crit_); |
RTC_DCHECK(!sink_ || sink_ == sink); |
sink_ = sink; |
+ current_sink_wants_ = wants; |
} |
void FrameForwarder::RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) { |
@@ -266,6 +267,16 @@ void FrameForwarder::RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) { |
sink_ = nullptr; |
} |
+rtc::VideoSinkWants FrameForwarder::current_sink_wants() const { |
+ rtc::CritScope lock(&crit_); |
+ return current_sink_wants_; |
+} |
+ |
+size_t FrameForwarder::number_of_sinks() const { |
+ rtc::CritScope lock(&crit_); |
+ return sink_ ? 1u : 0u; |
+} |
+ |
FrameGenerator* FrameGenerator::CreateChromaGenerator(size_t width, |
size_t height) { |
return new ChromaGenerator(width, height); |