OLD | NEW |
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 "media/cast/receiver/video_decoder.h" | 5 #include "media/cast/receiver/video_decoder.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
9 #include "base/json/json_reader.h" | 9 #include "base/json/json_reader.h" |
10 #include "base/location.h" | 10 #include "base/location.h" |
(...skipping 12 matching lines...) Expand all Loading... |
23 namespace media { | 23 namespace media { |
24 namespace cast { | 24 namespace cast { |
25 | 25 |
26 // Base class that handles the common problem of detecting dropped frames, and | 26 // Base class that handles the common problem of detecting dropped frames, and |
27 // then invoking the Decode() method implemented by the subclasses to convert | 27 // then invoking the Decode() method implemented by the subclasses to convert |
28 // the encoded payload data into a usable video frame. | 28 // the encoded payload data into a usable video frame. |
29 class VideoDecoder::ImplBase | 29 class VideoDecoder::ImplBase |
30 : public base::RefCountedThreadSafe<VideoDecoder::ImplBase> { | 30 : public base::RefCountedThreadSafe<VideoDecoder::ImplBase> { |
31 public: | 31 public: |
32 ImplBase(const scoped_refptr<CastEnvironment>& cast_environment, | 32 ImplBase(const scoped_refptr<CastEnvironment>& cast_environment, |
33 transport::Codec codec) | 33 Codec codec) |
34 : cast_environment_(cast_environment), | 34 : cast_environment_(cast_environment), |
35 codec_(codec), | 35 codec_(codec), |
36 cast_initialization_status_(STATUS_VIDEO_UNINITIALIZED), | 36 cast_initialization_status_(STATUS_VIDEO_UNINITIALIZED), |
37 seen_first_frame_(false) {} | 37 seen_first_frame_(false) {} |
38 | 38 |
39 CastInitializationStatus InitializationResult() const { | 39 CastInitializationStatus InitializationResult() const { |
40 return cast_initialization_status_; | 40 return cast_initialization_status_; |
41 } | 41 } |
42 | 42 |
43 void DecodeFrame(scoped_ptr<transport::EncodedFrame> encoded_frame, | 43 void DecodeFrame(scoped_ptr<EncodedFrame> encoded_frame, |
44 const DecodeFrameCallback& callback) { | 44 const DecodeFrameCallback& callback) { |
45 DCHECK_EQ(cast_initialization_status_, STATUS_VIDEO_INITIALIZED); | 45 DCHECK_EQ(cast_initialization_status_, STATUS_VIDEO_INITIALIZED); |
46 | 46 |
47 COMPILE_ASSERT(sizeof(encoded_frame->frame_id) == sizeof(last_frame_id_), | 47 COMPILE_ASSERT(sizeof(encoded_frame->frame_id) == sizeof(last_frame_id_), |
48 size_of_frame_id_types_do_not_match); | 48 size_of_frame_id_types_do_not_match); |
49 bool is_continuous = true; | 49 bool is_continuous = true; |
50 if (seen_first_frame_) { | 50 if (seen_first_frame_) { |
51 const uint32 frames_ahead = encoded_frame->frame_id - last_frame_id_; | 51 const uint32 frames_ahead = encoded_frame->frame_id - last_frame_id_; |
52 if (frames_ahead > 1) { | 52 if (frames_ahead > 1) { |
53 RecoverBecauseFramesWereDropped(); | 53 RecoverBecauseFramesWereDropped(); |
(...skipping 16 matching lines...) Expand all Loading... |
70 protected: | 70 protected: |
71 friend class base::RefCountedThreadSafe<ImplBase>; | 71 friend class base::RefCountedThreadSafe<ImplBase>; |
72 virtual ~ImplBase() {} | 72 virtual ~ImplBase() {} |
73 | 73 |
74 virtual void RecoverBecauseFramesWereDropped() {} | 74 virtual void RecoverBecauseFramesWereDropped() {} |
75 | 75 |
76 // Note: Implementation of Decode() is allowed to mutate |data|. | 76 // Note: Implementation of Decode() is allowed to mutate |data|. |
77 virtual scoped_refptr<VideoFrame> Decode(uint8* data, int len) = 0; | 77 virtual scoped_refptr<VideoFrame> Decode(uint8* data, int len) = 0; |
78 | 78 |
79 const scoped_refptr<CastEnvironment> cast_environment_; | 79 const scoped_refptr<CastEnvironment> cast_environment_; |
80 const transport::Codec codec_; | 80 const Codec codec_; |
81 | 81 |
82 // Subclass' ctor is expected to set this to STATUS_VIDEO_INITIALIZED. | 82 // Subclass' ctor is expected to set this to STATUS_VIDEO_INITIALIZED. |
83 CastInitializationStatus cast_initialization_status_; | 83 CastInitializationStatus cast_initialization_status_; |
84 | 84 |
85 private: | 85 private: |
86 bool seen_first_frame_; | 86 bool seen_first_frame_; |
87 uint32 last_frame_id_; | 87 uint32 last_frame_id_; |
88 | 88 |
89 DISALLOW_COPY_AND_ASSIGN(ImplBase); | 89 DISALLOW_COPY_AND_ASSIGN(ImplBase); |
90 }; | 90 }; |
91 | 91 |
92 class VideoDecoder::Vp8Impl : public VideoDecoder::ImplBase { | 92 class VideoDecoder::Vp8Impl : public VideoDecoder::ImplBase { |
93 public: | 93 public: |
94 explicit Vp8Impl(const scoped_refptr<CastEnvironment>& cast_environment) | 94 explicit Vp8Impl(const scoped_refptr<CastEnvironment>& cast_environment) |
95 : ImplBase(cast_environment, transport::CODEC_VIDEO_VP8) { | 95 : ImplBase(cast_environment, CODEC_VIDEO_VP8) { |
96 if (ImplBase::cast_initialization_status_ != STATUS_VIDEO_UNINITIALIZED) | 96 if (ImplBase::cast_initialization_status_ != STATUS_VIDEO_UNINITIALIZED) |
97 return; | 97 return; |
98 | 98 |
99 vpx_codec_dec_cfg_t cfg = {0}; | 99 vpx_codec_dec_cfg_t cfg = {0}; |
100 // TODO(miu): Revisit this for typical multi-core desktop use case. This | 100 // TODO(miu): Revisit this for typical multi-core desktop use case. This |
101 // feels like it should be 4 or 8. | 101 // feels like it should be 4 or 8. |
102 cfg.threads = 1; | 102 cfg.threads = 1; |
103 | 103 |
104 DCHECK(vpx_codec_get_caps(vpx_codec_vp8_dx()) & VPX_CODEC_CAP_POSTPROC); | 104 DCHECK(vpx_codec_get_caps(vpx_codec_vp8_dx()) & VPX_CODEC_CAP_POSTPROC); |
105 if (vpx_codec_dec_init(&context_, | 105 if (vpx_codec_dec_init(&context_, |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
166 vpx_codec_ctx_t context_; | 166 vpx_codec_ctx_t context_; |
167 | 167 |
168 DISALLOW_COPY_AND_ASSIGN(Vp8Impl); | 168 DISALLOW_COPY_AND_ASSIGN(Vp8Impl); |
169 }; | 169 }; |
170 | 170 |
171 #ifndef OFFICIAL_BUILD | 171 #ifndef OFFICIAL_BUILD |
172 // A fake video decoder that always output 2x2 black frames. | 172 // A fake video decoder that always output 2x2 black frames. |
173 class VideoDecoder::FakeImpl : public VideoDecoder::ImplBase { | 173 class VideoDecoder::FakeImpl : public VideoDecoder::ImplBase { |
174 public: | 174 public: |
175 explicit FakeImpl(const scoped_refptr<CastEnvironment>& cast_environment) | 175 explicit FakeImpl(const scoped_refptr<CastEnvironment>& cast_environment) |
176 : ImplBase(cast_environment, transport::CODEC_VIDEO_FAKE), | 176 : ImplBase(cast_environment, CODEC_VIDEO_FAKE), |
177 last_decoded_id_(-1) { | 177 last_decoded_id_(-1) { |
178 if (ImplBase::cast_initialization_status_ != STATUS_VIDEO_UNINITIALIZED) | 178 if (ImplBase::cast_initialization_status_ != STATUS_VIDEO_UNINITIALIZED) |
179 return; | 179 return; |
180 ImplBase::cast_initialization_status_ = STATUS_VIDEO_INITIALIZED; | 180 ImplBase::cast_initialization_status_ = STATUS_VIDEO_INITIALIZED; |
181 } | 181 } |
182 | 182 |
183 private: | 183 private: |
184 virtual ~FakeImpl() {} | 184 virtual ~FakeImpl() {} |
185 | 185 |
186 virtual scoped_refptr<VideoFrame> Decode(uint8* data, int len) OVERRIDE { | 186 virtual scoped_refptr<VideoFrame> Decode(uint8* data, int len) OVERRIDE { |
(...skipping 20 matching lines...) Expand all Loading... |
207 } | 207 } |
208 | 208 |
209 int last_decoded_id_; | 209 int last_decoded_id_; |
210 | 210 |
211 DISALLOW_COPY_AND_ASSIGN(FakeImpl); | 211 DISALLOW_COPY_AND_ASSIGN(FakeImpl); |
212 }; | 212 }; |
213 #endif | 213 #endif |
214 | 214 |
215 VideoDecoder::VideoDecoder( | 215 VideoDecoder::VideoDecoder( |
216 const scoped_refptr<CastEnvironment>& cast_environment, | 216 const scoped_refptr<CastEnvironment>& cast_environment, |
217 transport::Codec codec) | 217 Codec codec) |
218 : cast_environment_(cast_environment) { | 218 : cast_environment_(cast_environment) { |
219 switch (codec) { | 219 switch (codec) { |
220 #ifndef OFFICIAL_BUILD | 220 #ifndef OFFICIAL_BUILD |
221 case transport::CODEC_VIDEO_FAKE: | 221 case CODEC_VIDEO_FAKE: |
222 impl_ = new FakeImpl(cast_environment); | 222 impl_ = new FakeImpl(cast_environment); |
223 break; | 223 break; |
224 #endif | 224 #endif |
225 case transport::CODEC_VIDEO_VP8: | 225 case CODEC_VIDEO_VP8: |
226 impl_ = new Vp8Impl(cast_environment); | 226 impl_ = new Vp8Impl(cast_environment); |
227 break; | 227 break; |
228 case transport::CODEC_VIDEO_H264: | 228 case CODEC_VIDEO_H264: |
229 // TODO(miu): Need implementation. | 229 // TODO(miu): Need implementation. |
230 NOTIMPLEMENTED(); | 230 NOTIMPLEMENTED(); |
231 break; | 231 break; |
232 default: | 232 default: |
233 NOTREACHED() << "Unknown or unspecified codec."; | 233 NOTREACHED() << "Unknown or unspecified codec."; |
234 break; | 234 break; |
235 } | 235 } |
236 } | 236 } |
237 | 237 |
238 VideoDecoder::~VideoDecoder() {} | 238 VideoDecoder::~VideoDecoder() {} |
239 | 239 |
240 CastInitializationStatus VideoDecoder::InitializationResult() const { | 240 CastInitializationStatus VideoDecoder::InitializationResult() const { |
241 if (impl_) | 241 if (impl_) |
242 return impl_->InitializationResult(); | 242 return impl_->InitializationResult(); |
243 return STATUS_UNSUPPORTED_VIDEO_CODEC; | 243 return STATUS_UNSUPPORTED_VIDEO_CODEC; |
244 } | 244 } |
245 | 245 |
246 void VideoDecoder::DecodeFrame( | 246 void VideoDecoder::DecodeFrame( |
247 scoped_ptr<transport::EncodedFrame> encoded_frame, | 247 scoped_ptr<EncodedFrame> encoded_frame, |
248 const DecodeFrameCallback& callback) { | 248 const DecodeFrameCallback& callback) { |
249 DCHECK(encoded_frame.get()); | 249 DCHECK(encoded_frame.get()); |
250 DCHECK(!callback.is_null()); | 250 DCHECK(!callback.is_null()); |
251 if (!impl_ || impl_->InitializationResult() != STATUS_VIDEO_INITIALIZED) { | 251 if (!impl_ || impl_->InitializationResult() != STATUS_VIDEO_INITIALIZED) { |
252 callback.Run(make_scoped_refptr<VideoFrame>(NULL), false); | 252 callback.Run(make_scoped_refptr<VideoFrame>(NULL), false); |
253 return; | 253 return; |
254 } | 254 } |
255 cast_environment_->PostTask(CastEnvironment::VIDEO, | 255 cast_environment_->PostTask(CastEnvironment::VIDEO, |
256 FROM_HERE, | 256 FROM_HERE, |
257 base::Bind(&VideoDecoder::ImplBase::DecodeFrame, | 257 base::Bind(&VideoDecoder::ImplBase::DecodeFrame, |
258 impl_, | 258 impl_, |
259 base::Passed(&encoded_frame), | 259 base::Passed(&encoded_frame), |
260 callback)); | 260 callback)); |
261 } | 261 } |
262 | 262 |
263 } // namespace cast | 263 } // namespace cast |
264 } // namespace media | 264 } // namespace media |
OLD | NEW |