| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/remoting/fake_remoting_demuxer_stream_provider.h" | 5 #include "media/remoting/fake_demuxer_stream_provider.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/callback_helpers.h" | 9 #include "base/callback_helpers.h" |
| 10 #include "media/base/decoder_buffer.h" | 10 #include "media/base/decoder_buffer.h" |
| 11 #include "media/base/media_util.h" | 11 #include "media/base/media_util.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 | 13 |
| 14 using testing::_; | 14 using testing::_; |
| 15 using testing::Invoke; | 15 using testing::Invoke; |
| 16 using testing::Return; | 16 using testing::Return; |
| 17 | 17 |
| 18 namespace media { | 18 namespace media { |
| 19 namespace remoting { |
| 19 | 20 |
| 20 DummyDemuxerStream::DummyDemuxerStream(bool is_audio) { | 21 FakeDemuxerStream::FakeDemuxerStream(bool is_audio) { |
| 21 type_ = is_audio ? DemuxerStream::AUDIO : DemuxerStream::VIDEO; | 22 type_ = is_audio ? DemuxerStream::AUDIO : DemuxerStream::VIDEO; |
| 22 if (is_audio) { | 23 if (is_audio) { |
| 23 audio_config_.Initialize(kCodecAAC, kSampleFormatS16, CHANNEL_LAYOUT_STEREO, | 24 audio_config_.Initialize(kCodecAAC, kSampleFormatS16, CHANNEL_LAYOUT_STEREO, |
| 24 38400, std::vector<uint8_t>(), Unencrypted(), | 25 38400, std::vector<uint8_t>(), Unencrypted(), |
| 25 base::TimeDelta(), 0); | 26 base::TimeDelta(), 0); |
| 26 } else { | 27 } else { |
| 27 gfx::Size size(640, 480); | 28 gfx::Size size(640, 480); |
| 28 gfx::Rect rect(0, 0, 640, 480); | 29 gfx::Rect rect(0, 0, 640, 480); |
| 29 video_config_.Initialize(kCodecH264, H264PROFILE_BASELINE, | 30 video_config_.Initialize(kCodecH264, H264PROFILE_BASELINE, |
| 30 PIXEL_FORMAT_I420, COLOR_SPACE_SD_REC601, size, | 31 PIXEL_FORMAT_I420, COLOR_SPACE_SD_REC601, size, |
| 31 rect, size, std::vector<uint8_t>(), Unencrypted()); | 32 rect, size, std::vector<uint8_t>(), Unencrypted()); |
| 32 } | 33 } |
| 33 ON_CALL(*this, Read(_)) | 34 ON_CALL(*this, Read(_)) |
| 34 .WillByDefault(Invoke(this, &DummyDemuxerStream::FakeRead)); | 35 .WillByDefault(Invoke(this, &FakeDemuxerStream::FakeRead)); |
| 35 } | 36 } |
| 36 | 37 |
| 37 DummyDemuxerStream::~DummyDemuxerStream() = default; | 38 FakeDemuxerStream::~FakeDemuxerStream() = default; |
| 38 | 39 |
| 39 void DummyDemuxerStream::FakeRead(const ReadCB& read_cb) { | 40 void FakeDemuxerStream::FakeRead(const ReadCB& read_cb) { |
| 40 if (buffer_queue_.empty()) { | 41 if (buffer_queue_.empty()) { |
| 41 // Silent return to simulate waiting for buffer available. | 42 // Silent return to simulate waiting for buffer available. |
| 42 pending_read_cb_ = read_cb; | 43 pending_read_cb_ = read_cb; |
| 43 return; | 44 return; |
| 44 } | 45 } |
| 45 scoped_refptr<::media::DecoderBuffer> buffer = buffer_queue_.front(); | 46 scoped_refptr<DecoderBuffer> buffer = buffer_queue_.front(); |
| 46 buffer_queue_.pop_front(); | 47 buffer_queue_.pop_front(); |
| 47 read_cb.Run(kOk, buffer); | 48 read_cb.Run(kOk, buffer); |
| 48 } | 49 } |
| 49 | 50 |
| 50 AudioDecoderConfig DummyDemuxerStream::audio_decoder_config() { | 51 AudioDecoderConfig FakeDemuxerStream::audio_decoder_config() { |
| 51 return audio_config_; | 52 return audio_config_; |
| 52 } | 53 } |
| 53 | 54 |
| 54 VideoDecoderConfig DummyDemuxerStream::video_decoder_config() { | 55 VideoDecoderConfig FakeDemuxerStream::video_decoder_config() { |
| 55 return video_config_; | 56 return video_config_; |
| 56 } | 57 } |
| 57 | 58 |
| 58 DemuxerStream::Type DummyDemuxerStream::type() const { | 59 DemuxerStream::Type FakeDemuxerStream::type() const { |
| 59 return type_; | 60 return type_; |
| 60 } | 61 } |
| 61 | 62 |
| 62 DemuxerStream::Liveness DummyDemuxerStream::liveness() const { | 63 DemuxerStream::Liveness FakeDemuxerStream::liveness() const { |
| 63 return LIVENESS_UNKNOWN; | 64 return LIVENESS_UNKNOWN; |
| 64 } | 65 } |
| 65 | 66 |
| 66 bool DummyDemuxerStream::SupportsConfigChanges() { | 67 bool FakeDemuxerStream::SupportsConfigChanges() { |
| 67 return false; | 68 return false; |
| 68 } | 69 } |
| 69 | 70 |
| 70 VideoRotation DummyDemuxerStream::video_rotation() { | 71 VideoRotation FakeDemuxerStream::video_rotation() { |
| 71 return VIDEO_ROTATION_0; | 72 return VIDEO_ROTATION_0; |
| 72 } | 73 } |
| 73 | 74 |
| 74 bool DummyDemuxerStream::enabled() const { | 75 bool FakeDemuxerStream::enabled() const { |
| 75 return false; | 76 return false; |
| 76 } | 77 } |
| 77 | 78 |
| 78 void DummyDemuxerStream::CreateFakeFrame(size_t size, | 79 void FakeDemuxerStream::CreateFakeFrame(size_t size, |
| 79 bool key_frame, | 80 bool key_frame, |
| 80 int pts_ms) { | 81 int pts_ms) { |
| 81 std::vector<uint8_t> buffer(size); | 82 std::vector<uint8_t> buffer(size); |
| 82 // Assign each byte in the buffer its index mod 256. | 83 // Assign each byte in the buffer its index mod 256. |
| 83 for (size_t i = 0; i < size; ++i) { | 84 for (size_t i = 0; i < size; ++i) { |
| 84 buffer[i] = static_cast<uint8_t>(i & 0xFF); | 85 buffer[i] = static_cast<uint8_t>(i & 0xFF); |
| 85 } | 86 } |
| 86 base::TimeDelta pts = base::TimeDelta::FromMilliseconds(pts_ms); | 87 base::TimeDelta pts = base::TimeDelta::FromMilliseconds(pts_ms); |
| 87 | 88 |
| 88 // To DecoderBuffer | 89 // To DecoderBuffer |
| 89 scoped_refptr<::media::DecoderBuffer> input_buffer = | 90 scoped_refptr<DecoderBuffer> input_buffer = |
| 90 ::media::DecoderBuffer::CopyFrom(buffer.data(), size); | 91 DecoderBuffer::CopyFrom(buffer.data(), size); |
| 91 input_buffer->set_timestamp(pts); | 92 input_buffer->set_timestamp(pts); |
| 92 input_buffer->set_is_key_frame(key_frame); | 93 input_buffer->set_is_key_frame(key_frame); |
| 93 | 94 |
| 94 // Sends frame out if there is pending read callback. Otherwise, stores it | 95 // Sends frame out if there is pending read callback. Otherwise, stores it |
| 95 // in the buffer queue. | 96 // in the buffer queue. |
| 96 if (pending_read_cb_.is_null()) { | 97 if (pending_read_cb_.is_null()) { |
| 97 buffer_queue_.push_back(input_buffer); | 98 buffer_queue_.push_back(input_buffer); |
| 98 } else { | 99 } else { |
| 99 base::ResetAndReturn(&pending_read_cb_).Run(kOk, input_buffer); | 100 base::ResetAndReturn(&pending_read_cb_).Run(kOk, input_buffer); |
| 100 } | 101 } |
| 101 } | 102 } |
| 102 | 103 |
| 103 FakeRemotingDemuxerStreamProvider::FakeRemotingDemuxerStreamProvider() | 104 FakeDemuxerStreamProvider::FakeDemuxerStreamProvider() |
| 104 : demuxer_stream_(new DummyDemuxerStream(true)) {} | 105 : demuxer_stream_(new FakeDemuxerStream(true)) {} |
| 105 | 106 |
| 106 FakeRemotingDemuxerStreamProvider::~FakeRemotingDemuxerStreamProvider() {} | 107 FakeDemuxerStreamProvider::~FakeDemuxerStreamProvider() {} |
| 107 | 108 |
| 108 DemuxerStream* FakeRemotingDemuxerStreamProvider::GetStream( | 109 DemuxerStream* FakeDemuxerStreamProvider::GetStream(DemuxerStream::Type type) { |
| 109 DemuxerStream::Type type) { | |
| 110 if (type == DemuxerStream::AUDIO) | 110 if (type == DemuxerStream::AUDIO) |
| 111 return reinterpret_cast<DemuxerStream*>(demuxer_stream_.get()); | 111 return reinterpret_cast<DemuxerStream*>(demuxer_stream_.get()); |
| 112 return nullptr; | 112 return nullptr; |
| 113 } | 113 } |
| 114 | 114 |
| 115 } // namespace remoting |
| 115 } // namespace media | 116 } // namespace media |
| OLD | NEW |