OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "media/cast/sender/size_adaptable_video_encoder_base.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/location.h" | |
9 #include "media/base/video_frame.h" | |
10 | |
11 namespace media { | |
12 namespace cast { | |
13 | |
14 SizeAdaptableVideoEncoderBase::SizeAdaptableVideoEncoderBase( | |
15 const scoped_refptr<CastEnvironment>& cast_environment, | |
16 const VideoSenderConfig& video_config, | |
17 const StatusChangeCallback& status_change_cb) | |
18 : cast_environment_(cast_environment), | |
19 video_config_(video_config), | |
20 status_change_cb_(status_change_cb), | |
21 frames_in_encoder_(0), | |
22 last_frame_id_(kStartFrameId), | |
23 weak_factory_(this) { | |
24 cast_environment_->PostTask( | |
25 CastEnvironment::MAIN, | |
26 FROM_HERE, | |
27 base::Bind(status_change_cb_, STATUS_INITIALIZED)); | |
28 } | |
29 | |
30 SizeAdaptableVideoEncoderBase::~SizeAdaptableVideoEncoderBase() { | |
31 weak_factory_.InvalidateWeakPtrs(); | |
hubbe
2015/02/11 00:47:54
This is kind of weird.
Why invalidate weak pointer
miu
2015/02/11 02:14:23
Done. Consolidated into DestroyCurrentEncoder(),
| |
32 DestroyCurrentEncoder(); | |
33 } | |
34 | |
35 bool SizeAdaptableVideoEncoderBase::EncodeVideoFrame( | |
36 const scoped_refptr<media::VideoFrame>& video_frame, | |
37 const base::TimeTicks& reference_time, | |
38 const FrameEncodedCallback& frame_encoded_callback) { | |
39 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
40 | |
41 const gfx::Size frame_size = video_frame->visible_rect().size(); | |
42 if (frame_size.IsEmpty()) { | |
43 DVLOG(1) << "Rejecting empty video frame."; | |
44 return false; | |
45 } | |
46 if (frame_size != current_encoder_frame_size_ || !current_encoder_) { | |
47 VLOG(1) << "Dropping frame (of size " << frame_size.ToString() | |
48 << ") until a replacement encoder is spun-up. Current encoder " | |
49 "handles frames of size " | |
50 << current_encoder_frame_size_.ToString() << '.'; | |
51 MaybeSpawnReplacementEncoder(frame_size); | |
52 return false; | |
53 } | |
54 | |
55 const bool is_frame_accepted = current_encoder_->EncodeVideoFrame( | |
56 video_frame, | |
57 reference_time, | |
58 base::Bind(&SizeAdaptableVideoEncoderBase::OnEncodedVideoFrame, | |
59 weak_factory_.GetWeakPtr(), | |
60 frame_encoded_callback)); | |
61 if (is_frame_accepted) | |
62 ++frames_in_encoder_; | |
63 return is_frame_accepted; | |
64 } | |
65 | |
66 void SizeAdaptableVideoEncoderBase::SetBitRate(int new_bit_rate) { | |
67 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
68 video_config_.start_bitrate = new_bit_rate; | |
69 if (current_encoder_) | |
70 current_encoder_->SetBitRate(new_bit_rate); | |
71 } | |
72 | |
73 void SizeAdaptableVideoEncoderBase::GenerateKeyFrame() { | |
74 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
75 if (current_encoder_) | |
76 current_encoder_->GenerateKeyFrame(); | |
77 } | |
78 | |
79 void SizeAdaptableVideoEncoderBase::LatestFrameIdToReference(uint32 frame_id) { | |
80 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
81 if (current_encoder_) | |
82 current_encoder_->LatestFrameIdToReference(frame_id); | |
83 } | |
84 | |
85 scoped_ptr<VideoFrameFactory> | |
86 SizeAdaptableVideoEncoderBase::CreateVideoFrameFactory() { | |
87 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
88 return nullptr; | |
89 } | |
90 | |
91 void SizeAdaptableVideoEncoderBase::EmitFrames() { | |
92 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
93 if (current_encoder_) | |
94 current_encoder_->EmitFrames(); | |
95 } | |
96 | |
97 StatusChangeCallback | |
98 SizeAdaptableVideoEncoderBase::CreateEncoderStatusChangeCallback() { | |
99 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
100 return base::Bind(&SizeAdaptableVideoEncoderBase::OnEncoderStatusChange, | |
101 weak_factory_.GetWeakPtr()); | |
102 } | |
103 | |
104 void SizeAdaptableVideoEncoderBase::OnEncoderReplaced( | |
105 VideoEncoder* replacement_encoder) {} | |
106 | |
107 void SizeAdaptableVideoEncoderBase::DestroyCurrentEncoder() { | |
108 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
109 current_encoder_.reset(); | |
110 } | |
111 | |
112 void SizeAdaptableVideoEncoderBase::MaybeSpawnReplacementEncoder( | |
hubbe
2015/02/11 00:47:54
I don't like "Maybe"
I think "Try" or "" would be
miu
2015/02/11 02:14:23
Done.
| |
113 const gfx::Size& size_needed) { | |
114 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
115 | |
116 // If prior frames are still encoding in the current encoder, let them finish | |
117 // first. | |
118 if (frames_in_encoder_ > 0) { | |
119 current_encoder_->EmitFrames(); | |
120 return; | |
121 } | |
122 | |
123 if (next_encoder_.get()) | |
124 return; // Already spawned. | |
125 | |
126 // Prevent future callbacks from |current_encoder_| and destroy it. | |
127 weak_factory_.InvalidateWeakPtrs(); | |
128 DestroyCurrentEncoder(); | |
129 | |
130 OnEncoderStatusChange(STATUS_CODEC_REINIT_PENDING); | |
131 next_encoder_frame_size_ = size_needed; | |
132 VLOG(1) << "Creating replacement video encoder (for frame size change from " | |
133 << current_encoder_frame_size_.ToString() << " to " | |
134 << next_encoder_frame_size_.ToString() << ")."; | |
135 next_encoder_ = CreateReplacementEncoder().Pass(); | |
136 DCHECK(next_encoder_.get()); | |
137 } | |
138 | |
139 void SizeAdaptableVideoEncoderBase::OnEncoderStatusChange( | |
140 OperationalStatus status) { | |
141 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
142 if (next_encoder_.get() && status == STATUS_INITIALIZED) { | |
143 // Begin using the replacement encoder. | |
144 current_encoder_ = next_encoder_.Pass(); | |
145 current_encoder_frame_size_ = next_encoder_frame_size_; | |
146 OnEncoderReplaced(current_encoder_.get()); | |
147 } | |
148 status_change_cb_.Run(status); | |
149 } | |
150 | |
151 void SizeAdaptableVideoEncoderBase::OnEncodedVideoFrame( | |
152 const FrameEncodedCallback& frame_encoded_callback, | |
153 scoped_ptr<EncodedFrame> encoded_frame) { | |
154 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
155 --frames_in_encoder_; | |
156 DCHECK_GE(frames_in_encoder_, 0); | |
157 last_frame_id_ = encoded_frame->frame_id; | |
158 frame_encoded_callback.Run(encoded_frame.Pass()); | |
159 } | |
160 | |
161 } // namespace cast | |
162 } // namespace media | |
OLD | NEW |