blob: dcd57776458a945beb15f9ebcf27fbf275f3b633 [file] [log] [blame]
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/remoting/fake_remoting_demuxer_stream_provider.h"
#include <vector>
#include "base/callback_helpers.h"
#include "media/base/decoder_buffer.h"
#include "media/base/media_util.h"
#include "testing/gtest/include/gtest/gtest.h"
using testing::_;
using testing::Invoke;
using testing::Return;
namespace media {
DummyDemuxerStream::DummyDemuxerStream(bool is_audio) {
type_ = is_audio ? DemuxerStream::AUDIO : DemuxerStream::VIDEO;
if (is_audio) {
audio_config_.Initialize(kCodecAAC, kSampleFormatS16, CHANNEL_LAYOUT_STEREO,
38400, std::vector<uint8_t>(), Unencrypted(),
base::TimeDelta(), 0);
} else {
gfx::Size size(640, 480);
gfx::Rect rect(0, 0, 640, 480);
video_config_.Initialize(kCodecH264, H264PROFILE_BASELINE,
PIXEL_FORMAT_I420, COLOR_SPACE_SD_REC601, size,
rect, size, std::vector<uint8_t>(), Unencrypted());
}
ON_CALL(*this, Read(_))
.WillByDefault(Invoke(this, &DummyDemuxerStream::FakeRead));
}
DummyDemuxerStream::~DummyDemuxerStream() = default;
void DummyDemuxerStream::FakeRead(const ReadCB& read_cb) {
if (buffer_queue_.empty()) {
// Silent return to simulate waiting for buffer available.
pending_read_cb_ = read_cb;
return;
}
scoped_refptr<::media::DecoderBuffer> buffer = buffer_queue_.front();
buffer_queue_.pop_front();
read_cb.Run(kOk, buffer);
}
AudioDecoderConfig DummyDemuxerStream::audio_decoder_config() {
return audio_config_;
}
VideoDecoderConfig DummyDemuxerStream::video_decoder_config() {
return video_config_;
}
DemuxerStream::Type DummyDemuxerStream::type() const {
return type_;
}
DemuxerStream::Liveness DummyDemuxerStream::liveness() const {
return LIVENESS_UNKNOWN;
}
bool DummyDemuxerStream::SupportsConfigChanges() {
return false;
}
VideoRotation DummyDemuxerStream::video_rotation() {
return VIDEO_ROTATION_0;
}
bool DummyDemuxerStream::enabled() const {
return false;
}
void DummyDemuxerStream::CreateFakeFrame(size_t size,
bool key_frame,
int pts_ms) {
std::vector<uint8_t> buffer(size);
// Assign each byte in the buffer its index mod 256.
for (size_t i = 0; i < size; ++i) {
buffer[i] = static_cast<uint8_t>(i & 0xFF);
}
base::TimeDelta pts = base::TimeDelta::FromMilliseconds(pts_ms);
// To DecoderBuffer
scoped_refptr<::media::DecoderBuffer> input_buffer =
::media::DecoderBuffer::CopyFrom(buffer.data(), size);
input_buffer->set_timestamp(pts);
input_buffer->set_is_key_frame(key_frame);
// Sends frame out if there is pending read callback. Otherwise, stores it
// in the buffer queue.
if (pending_read_cb_.is_null()) {
buffer_queue_.push_back(input_buffer);
} else {
base::ResetAndReturn(&pending_read_cb_).Run(kOk, input_buffer);
}
}
FakeRemotingDemuxerStreamProvider::FakeRemotingDemuxerStreamProvider()
: demuxer_stream_(new DummyDemuxerStream(true)) {}
FakeRemotingDemuxerStreamProvider::~FakeRemotingDemuxerStreamProvider() {}
DemuxerStream* FakeRemotingDemuxerStreamProvider::GetStream(
DemuxerStream::Type type) {
if (type == DemuxerStream::AUDIO)
return reinterpret_cast<DemuxerStream*>(demuxer_stream_.get());
return nullptr;
}
} // namespace media