Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(252)

Side by Side Diff: remoting/host/video_frame_recorder.cc

Issue 468613002: Readability review. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Address review comments. Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright 2014 The Chromium Authors. All rights reserved. 1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "remoting/host/video_frame_recorder.h" 5 #include "remoting/host/video_frame_recorder.h"
6 6
7 #include "base/bind.h" 7 #include "base/bind.h"
8 #include "base/location.h" 8 #include "base/location.h"
9 #include "base/single_thread_task_runner.h" 9 #include "base/single_thread_task_runner.h"
10 #include "base/stl_util.h" 10 #include "base/stl_util.h"
11 #include "base/thread_task_runner_handle.h" 11 #include "base/thread_task_runner_handle.h"
12 #include "remoting/codec/video_encoder.h" 12 #include "remoting/codec/video_encoder.h"
13 #include "remoting/proto/video.pb.h" 13 #include "remoting/proto/video.pb.h"
14 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" 14 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
15 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" 15 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
16 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h" 16 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h"
17 17
18 namespace remoting { 18 namespace remoting {
19 19
20 static int64_t FrameContentSize(const webrtc::DesktopFrame* frame) { 20 static int64_t FrameContentSize(const webrtc::DesktopFrame* frame) {
21 DCHECK_GT(frame->stride(), 0); 21 DCHECK_GT(frame->stride(), 0);
22 return frame->stride() * frame->size().height(); 22 return frame->stride() * frame->size().height();
23 } 23 }
24 24
25 // VideoEncoder wrapper used to intercept frames passed to a real VideoEncoder. 25 // VideoEncoder wrapper used to intercept frames passed to a real VideoEncoder.
26 class VideoFrameRecorder::RecordingVideoEncoder : public VideoEncoder { 26 class VideoFrameRecorder::RecordingVideoEncoder : public VideoEncoder {
27 public: 27 public:
28 RecordingVideoEncoder(scoped_ptr<VideoEncoder> encoder, 28 RecordingVideoEncoder(scoped_ptr<VideoEncoder> encoder,
29 scoped_refptr<base::TaskRunner> recorder_task_runner, 29 scoped_refptr<base::TaskRunner> recorder_task_runner,
30 base::WeakPtr<VideoFrameRecorder> recorder) 30 base::WeakPtr<VideoFrameRecorder> recorder);
31 : encoder_(encoder.Pass()),
32 recorder_task_runner_(recorder_task_runner),
33 recorder_(recorder),
34 enable_recording_(false),
35 weak_factory_(this) {
36 DCHECK(encoder_);
37 DCHECK(recorder_task_runner_);
38 }
39 31
40 base::WeakPtr<RecordingVideoEncoder> AsWeakPtr() { 32 base::WeakPtr<RecordingVideoEncoder> AsWeakPtr();
41 return weak_factory_.GetWeakPtr();
42 }
43 33
44 void SetEnableRecording(bool enable_recording) { 34 void set_enable_recording(bool enable_recording) {
45 DCHECK(!encoder_task_runner_ || 35 DCHECK(!encoder_task_runner_ ||
46 encoder_task_runner_->BelongsToCurrentThread()); 36 encoder_task_runner_->BelongsToCurrentThread());
47 enable_recording_ = enable_recording; 37 enable_recording_ = enable_recording;
48 } 38 }
49 39
50 // remoting::VideoEncoder interface. 40 // remoting::VideoEncoder interface.
51 virtual void SetLosslessEncode(bool want_lossless) OVERRIDE { 41 virtual void SetLosslessEncode(bool want_lossless) OVERRIDE;
52 encoder_->SetLosslessEncode(want_lossless); 42 virtual void SetLosslessColor(bool want_lossless) OVERRIDE;
53 }
54 virtual void SetLosslessColor(bool want_lossless) OVERRIDE {
55 encoder_->SetLosslessColor(want_lossless);
56 }
57 virtual scoped_ptr<VideoPacket> Encode( 43 virtual scoped_ptr<VideoPacket> Encode(
58 const webrtc::DesktopFrame& frame) OVERRIDE { 44 const webrtc::DesktopFrame& frame) OVERRIDE;
59 // If this is the first Encode() then store the TaskRunner and inform the
60 // VideoFrameRecorder so it can post SetEnableRecording() on it.
61 if (!encoder_task_runner_) {
62 encoder_task_runner_ = base::ThreadTaskRunnerHandle::Get();
63 recorder_task_runner_->PostTask(FROM_HERE,
64 base::Bind(&VideoFrameRecorder::SetEncoderTaskRunner,
65 recorder_,
66 encoder_task_runner_));
67 }
68
69 DCHECK(encoder_task_runner_->BelongsToCurrentThread());
70
71 if (enable_recording_) {
72 // Copy the frame and post it to the VideoFrameRecorder to store.
73 scoped_ptr<webrtc::DesktopFrame> frame_copy(
74 new webrtc::BasicDesktopFrame(frame.size()));
75 *frame_copy->mutable_updated_region() = frame.updated_region();
76 frame_copy->set_dpi(frame.dpi());
77 frame_copy->CopyPixelsFrom(frame.data(),
78 frame.stride(),
79 webrtc::DesktopRect::MakeSize(frame.size()));
80 recorder_task_runner_->PostTask(FROM_HERE,
81 base::Bind(&VideoFrameRecorder::RecordFrame,
82 recorder_,
83 base::Passed(&frame_copy)));
84 }
85
86 return encoder_->Encode(frame);
87 }
88 45
89 private: 46 private:
90 scoped_ptr<VideoEncoder> encoder_; 47 scoped_ptr<VideoEncoder> encoder_;
91 scoped_refptr<base::TaskRunner> recorder_task_runner_; 48 scoped_refptr<base::TaskRunner> recorder_task_runner_;
92 base::WeakPtr<VideoFrameRecorder> recorder_; 49 base::WeakPtr<VideoFrameRecorder> recorder_;
93 50
94 bool enable_recording_; 51 bool enable_recording_;
95 scoped_refptr<base::SingleThreadTaskRunner> encoder_task_runner_; 52 scoped_refptr<base::SingleThreadTaskRunner> encoder_task_runner_;
96 53
97 base::WeakPtrFactory<RecordingVideoEncoder> weak_factory_; 54 base::WeakPtrFactory<RecordingVideoEncoder> weak_factory_;
98 55
99 DISALLOW_COPY_AND_ASSIGN(RecordingVideoEncoder); 56 DISALLOW_COPY_AND_ASSIGN(RecordingVideoEncoder);
100 }; 57 };
101 58
59 VideoFrameRecorder::RecordingVideoEncoder::RecordingVideoEncoder(
60 scoped_ptr<VideoEncoder> encoder,
61 scoped_refptr<base::TaskRunner> recorder_task_runner,
62 base::WeakPtr<VideoFrameRecorder> recorder)
63 : encoder_(encoder.Pass()),
64 recorder_task_runner_(recorder_task_runner),
65 recorder_(recorder),
66 enable_recording_(false),
67 weak_factory_(this) {
68 DCHECK(encoder_);
69 DCHECK(recorder_task_runner_);
70 }
71
72 base::WeakPtr<VideoFrameRecorder::RecordingVideoEncoder>
73 VideoFrameRecorder::RecordingVideoEncoder::AsWeakPtr() {
74 return weak_factory_.GetWeakPtr();
75 }
76
77 void VideoFrameRecorder::RecordingVideoEncoder::SetLosslessEncode(
78 bool want_lossless) {
79 encoder_->SetLosslessEncode(want_lossless);
80 }
81
82 void VideoFrameRecorder::RecordingVideoEncoder::SetLosslessColor(
83 bool want_lossless) {
84 encoder_->SetLosslessColor(want_lossless);
85 }
86
87 scoped_ptr<VideoPacket> VideoFrameRecorder::RecordingVideoEncoder::Encode(
88 const webrtc::DesktopFrame& frame) {
89 // If this is the first Encode() then store the TaskRunner and inform the
90 // VideoFrameRecorder so it can post set_enable_recording() on it.
91 if (!encoder_task_runner_) {
92 encoder_task_runner_ = base::ThreadTaskRunnerHandle::Get();
93 recorder_task_runner_->PostTask(FROM_HERE,
94 base::Bind(&VideoFrameRecorder::SetEncoderTaskRunner,
95 recorder_,
96 encoder_task_runner_));
97 }
98
99 DCHECK(encoder_task_runner_->BelongsToCurrentThread());
100
101 if (enable_recording_) {
102 // Copy the frame and post it to the VideoFrameRecorder to store.
103 scoped_ptr<webrtc::DesktopFrame> frame_copy(
104 new webrtc::BasicDesktopFrame(frame.size()));
105 *frame_copy->mutable_updated_region() = frame.updated_region();
106 frame_copy->set_dpi(frame.dpi());
107 frame_copy->CopyPixelsFrom(frame.data(),
108 frame.stride(),
109 webrtc::DesktopRect::MakeSize(frame.size()));
110 recorder_task_runner_->PostTask(FROM_HERE,
111 base::Bind(&VideoFrameRecorder::RecordFrame,
112 recorder_,
113 base::Passed(&frame_copy)));
114 }
115
116 return encoder_->Encode(frame);
117 }
118
102 VideoFrameRecorder::VideoFrameRecorder() 119 VideoFrameRecorder::VideoFrameRecorder()
103 : content_bytes_(0), 120 : content_bytes_(0),
104 max_content_bytes_(0), 121 max_content_bytes_(0),
105 enable_recording_(false), 122 enable_recording_(false),
106 weak_factory_(this) { 123 weak_factory_(this) {
107 } 124 }
108 125
109 VideoFrameRecorder::~VideoFrameRecorder() { 126 VideoFrameRecorder::~VideoFrameRecorder() {
110 DetachVideoEncoderWrapper(); 127 DetachVideoEncoderWrapper();
111 } 128 }
(...skipping 19 matching lines...) Expand all
131 // Immediately detach the wrapper from this recorder. 148 // Immediately detach the wrapper from this recorder.
132 weak_factory_.InvalidateWeakPtrs(); 149 weak_factory_.InvalidateWeakPtrs();
133 150
134 // Clean up any pending recorded frames. 151 // Clean up any pending recorded frames.
135 STLDeleteElements(&recorded_frames_); 152 STLDeleteElements(&recorded_frames_);
136 content_bytes_ = 0; 153 content_bytes_ = 0;
137 154
138 // Tell the wrapper to stop recording and posting frames to us. 155 // Tell the wrapper to stop recording and posting frames to us.
139 if (encoder_task_runner_) { 156 if (encoder_task_runner_) {
140 encoder_task_runner_->PostTask(FROM_HERE, 157 encoder_task_runner_->PostTask(FROM_HERE,
141 base::Bind(&RecordingVideoEncoder::SetEnableRecording, 158 base::Bind(&RecordingVideoEncoder::set_enable_recording,
142 recording_encoder_, false)); 159 recording_encoder_, false));
143 } 160 }
144 161
145 // Detach this recorder from the calling and encode threads. 162 // Detach this recorder from the calling and encode threads.
146 caller_task_runner_ = NULL; 163 caller_task_runner_ = NULL;
147 encoder_task_runner_ = NULL; 164 encoder_task_runner_ = NULL;
148 } 165 }
149 166
150 void VideoFrameRecorder::SetEnableRecording(bool enable_recording) { 167 void VideoFrameRecorder::SetEnableRecording(bool enable_recording) {
151 DCHECK(!caller_task_runner_ || caller_task_runner_->BelongsToCurrentThread()); 168 DCHECK(!caller_task_runner_ || caller_task_runner_->BelongsToCurrentThread());
152 169
153 if (enable_recording_ == enable_recording) { 170 if (enable_recording_ == enable_recording) {
154 return; 171 return;
155 } 172 }
156 enable_recording_ = enable_recording; 173 enable_recording_ = enable_recording;
157 174
158 if (encoder_task_runner_) { 175 if (encoder_task_runner_) {
159 encoder_task_runner_->PostTask(FROM_HERE, 176 encoder_task_runner_->PostTask(FROM_HERE,
160 base::Bind(&RecordingVideoEncoder::SetEnableRecording, 177 base::Bind(&RecordingVideoEncoder::set_enable_recording,
161 recording_encoder_, 178 recording_encoder_,
162 enable_recording_)); 179 enable_recording_));
163 } 180 }
164 } 181 }
165 182
166 void VideoFrameRecorder::SetMaxContentBytes(int64_t max_content_bytes) { 183 void VideoFrameRecorder::SetMaxContentBytes(int64_t max_content_bytes) {
167 DCHECK(!caller_task_runner_ || caller_task_runner_->BelongsToCurrentThread()); 184 DCHECK(!caller_task_runner_ || caller_task_runner_->BelongsToCurrentThread());
168 DCHECK_GE(max_content_bytes, 0); 185 DCHECK_GE(max_content_bytes, 0);
169 186
170 max_content_bytes_ = max_content_bytes; 187 max_content_bytes_ = max_content_bytes;
(...skipping 17 matching lines...) Expand all
188 scoped_refptr<base::TaskRunner> task_runner) { 205 scoped_refptr<base::TaskRunner> task_runner) {
189 DCHECK(caller_task_runner_->BelongsToCurrentThread()); 206 DCHECK(caller_task_runner_->BelongsToCurrentThread());
190 DCHECK(!encoder_task_runner_); 207 DCHECK(!encoder_task_runner_);
191 DCHECK(task_runner); 208 DCHECK(task_runner);
192 209
193 encoder_task_runner_ = task_runner; 210 encoder_task_runner_ = task_runner;
194 211
195 // If the caller already enabled recording, inform the recording encoder. 212 // If the caller already enabled recording, inform the recording encoder.
196 if (enable_recording_ && encoder_task_runner_) { 213 if (enable_recording_ && encoder_task_runner_) {
197 encoder_task_runner_->PostTask(FROM_HERE, 214 encoder_task_runner_->PostTask(FROM_HERE,
198 base::Bind(&RecordingVideoEncoder::SetEnableRecording, 215 base::Bind(&RecordingVideoEncoder::set_enable_recording,
199 recording_encoder_, 216 recording_encoder_,
200 enable_recording_)); 217 enable_recording_));
201 } 218 }
202 } 219 }
203 220
204 void VideoFrameRecorder::RecordFrame(scoped_ptr<webrtc::DesktopFrame> frame) { 221 void VideoFrameRecorder::RecordFrame(scoped_ptr<webrtc::DesktopFrame> frame) {
205 DCHECK(caller_task_runner_->BelongsToCurrentThread()); 222 DCHECK(caller_task_runner_->BelongsToCurrentThread());
206 223
207 int64_t frame_bytes = FrameContentSize(frame.get()); 224 int64_t frame_bytes = FrameContentSize(frame.get());
208 DCHECK_GE(frame_bytes, 0); 225 DCHECK_GE(frame_bytes, 0);
(...skipping 11 matching lines...) Expand all
220 if (content_bytes_ + frame_bytes > max_content_bytes_) { 237 if (content_bytes_ + frame_bytes > max_content_bytes_) {
221 return; 238 return;
222 } 239 }
223 240
224 // Store the frame and update the content byte count. 241 // Store the frame and update the content byte count.
225 recorded_frames_.push_back(frame.release()); 242 recorded_frames_.push_back(frame.release());
226 content_bytes_ += frame_bytes; 243 content_bytes_ += frame_bytes;
227 } 244 }
228 245
229 } // namespace remoting 246 } // namespace remoting
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698