1 /*
2 * Copyright (c) 2012 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 #include "common_video/include/incoming_video_stream.h"
12
13 #include <memory>
14 #include <utility>
15
16 #include "absl/types/optional.h"
17 #include "common_video/video_render_frames.h"
18 #include "rtc_base/checks.h"
19 #include "rtc_base/trace_event.h"
20
21 namespace webrtc {
22
IncomingVideoStream(TaskQueueFactory * task_queue_factory,int32_t delay_ms,rtc::VideoSinkInterface<VideoFrame> * callback)23 IncomingVideoStream::IncomingVideoStream(
24 TaskQueueFactory* task_queue_factory,
25 int32_t delay_ms,
26 rtc::VideoSinkInterface<VideoFrame>* callback)
27 : render_buffers_(delay_ms),
28 callback_(callback),
29 incoming_render_queue_(task_queue_factory->CreateTaskQueue(
30 "IncomingVideoStream",
31 TaskQueueFactory::Priority::HIGH)) {}
32
~IncomingVideoStream()33 IncomingVideoStream::~IncomingVideoStream() {
34 RTC_DCHECK(main_thread_checker_.IsCurrent());
35 }
36
OnFrame(const VideoFrame & video_frame)37 void IncomingVideoStream::OnFrame(const VideoFrame& video_frame) {
38 TRACE_EVENT0("webrtc", "IncomingVideoStream::OnFrame");
39 RTC_CHECK_RUNS_SERIALIZED(&decoder_race_checker_);
40 RTC_DCHECK(!incoming_render_queue_.IsCurrent());
41 // TODO(srte): Using video_frame = std::move(video_frame) would move the frame
42 // into the lambda instead of copying it, but it doesn't work unless we change
43 // OnFrame to take its frame argument by value instead of const reference.
44 incoming_render_queue_.PostTask([this, video_frame = video_frame]() mutable {
45 RTC_DCHECK(incoming_render_queue_.IsCurrent());
46 if (render_buffers_.AddFrame(std::move(video_frame)) == 1)
47 Dequeue();
48 });
49 }
50
Dequeue()51 void IncomingVideoStream::Dequeue() {
52 TRACE_EVENT0("webrtc", "IncomingVideoStream::Dequeue");
53 RTC_DCHECK(incoming_render_queue_.IsCurrent());
54 absl::optional<VideoFrame> frame_to_render = render_buffers_.FrameToRender();
55 if (frame_to_render)
56 callback_->OnFrame(*frame_to_render);
57
58 if (render_buffers_.HasPendingFrames()) {
59 uint32_t wait_time = render_buffers_.TimeToNextFrameRelease();
60 incoming_render_queue_.PostDelayedTask([this]() { Dequeue(); }, wait_time);
61 }
62 }
63
64 } // namespace webrtc
65