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