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