| 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);
|
|
|