123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185 |
- // Copyright 2015 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 "remoting/client/software_video_renderer.h"
- #include <stdint.h>
- #include <memory>
- #include <utility>
- #include <vector>
- #include "base/bind.h"
- #include "base/run_loop.h"
- #include "base/test/task_environment.h"
- #include "base/threading/thread.h"
- #include "remoting/client/client_context.h"
- #include "remoting/codec/video_encoder_verbatim.h"
- #include "remoting/proto/video.pb.h"
- #include "remoting/protocol/frame_consumer.h"
- #include "remoting/protocol/session_config.h"
- #include "testing/gtest/include/gtest/gtest.h"
- #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
- using webrtc::DesktopFrame;
- namespace remoting {
- namespace {
- const int kFrameWidth = 200;
- const int kFrameHeight = 200;
- class TestFrameConsumer : public protocol::FrameConsumer {
- public:
- TestFrameConsumer() = default;
- ~TestFrameConsumer() override = default;
- std::unique_ptr<DesktopFrame> WaitForNextFrame(
- base::OnceClosure* out_done_callback) {
- EXPECT_TRUE(thread_checker_.CalledOnValidThread());
- frame_run_loop_ = std::make_unique<base::RunLoop>();
- frame_run_loop_->Run();
- frame_run_loop_.reset();
- *out_done_callback = std::move(last_frame_done_callback_);
- return std::move(last_frame_);
- }
- // FrameConsumer interface.
- std::unique_ptr<DesktopFrame> AllocateFrame(
- const webrtc::DesktopSize& size) override {
- EXPECT_TRUE(thread_checker_.CalledOnValidThread());
- return std::make_unique<webrtc::BasicDesktopFrame>(size);
- }
- void DrawFrame(std::unique_ptr<DesktopFrame> frame,
- base::OnceClosure done) override {
- EXPECT_TRUE(thread_checker_.CalledOnValidThread());
- last_frame_ = std::move(frame);
- last_frame_done_callback_ = std::move(done);
- frame_run_loop_->Quit();
- }
- PixelFormat GetPixelFormat() override {
- EXPECT_TRUE(thread_checker_.CalledOnValidThread());
- return FORMAT_BGRA;
- }
- private:
- base::ThreadChecker thread_checker_;
- std::unique_ptr<base::RunLoop> frame_run_loop_;
- std::unique_ptr<DesktopFrame> last_frame_;
- base::OnceClosure last_frame_done_callback_;
- };
- std::unique_ptr<DesktopFrame> CreateTestFrame(int index) {
- std::unique_ptr<DesktopFrame> frame(new webrtc::BasicDesktopFrame(
- webrtc::DesktopSize(kFrameWidth, kFrameHeight)));
- for (int y = 0; y < kFrameHeight; y++) {
- for (int x = 0; x < kFrameWidth; x++) {
- uint8_t* out = frame->data() + x * DesktopFrame::kBytesPerPixel +
- y * frame->stride();
- out[0] = index + x + y * kFrameWidth;
- out[1] = index + x + y * kFrameWidth + 1;
- out[2] = index + x + y * kFrameWidth + 2;
- out[3] = 0;
- }
- }
- if (index == 0) {
- frame->mutable_updated_region()->SetRect(
- webrtc::DesktopRect::MakeWH(kFrameWidth, kFrameHeight));
- } else {
- frame->mutable_updated_region()->SetRect(
- webrtc::DesktopRect::MakeWH(index, index));
- }
- return frame;
- }
- // Returns true when frames a and b are equivalent.
- bool CompareFrames(const DesktopFrame& a, const DesktopFrame& b) {
- if (!a.size().equals(b.size()) ||
- !a.updated_region().Equals(b.updated_region())) {
- return false;
- }
- for (webrtc::DesktopRegion::Iterator i(a.updated_region()); !i.IsAtEnd();
- i.Advance()) {
- for (int row = i.rect().top(); row < i.rect().bottom(); ++row) {
- if (memcmp(a.data() + a.stride() * row +
- i.rect().left() * DesktopFrame::kBytesPerPixel,
- b.data() + b.stride() * row +
- i.rect().left() * DesktopFrame::kBytesPerPixel,
- i.rect().width() * DesktopFrame::kBytesPerPixel) != 0) {
- return false;
- }
- }
- }
- return true;
- }
- // Helper to set value at |out| to 1.
- void SetTrue(int* out) {
- *out = 1;
- }
- } // namespace
- class SoftwareVideoRendererTest : public ::testing::Test {
- public:
- SoftwareVideoRendererTest() : context_(nullptr) {
- context_.Start();
- renderer_ = std::make_unique<SoftwareVideoRenderer>(&frame_consumer_);
- renderer_->Initialize(context_, nullptr);
- renderer_->OnSessionConfig(
- *protocol::SessionConfig::ForTestWithVerbatimVideo());
- }
- protected:
- base::test::SingleThreadTaskEnvironment task_environment_;
- ClientContext context_;
- TestFrameConsumer frame_consumer_;
- std::unique_ptr<SoftwareVideoRenderer> renderer_;
- VideoEncoderVerbatim encoder_;
- };
- TEST_F(SoftwareVideoRendererTest, DecodeFrame) {
- const int kFrameCount = 5;
- std::vector<std::unique_ptr<DesktopFrame>> test_frames;
- // std::vector<bool> doesn't allow to get pointer to individual values, so
- // int needs to be used instead.
- std::vector<int> callback_called(kFrameCount);
- for (int frame_index = 0; frame_index < kFrameCount; frame_index++) {
- test_frames.push_back(CreateTestFrame(frame_index));
- callback_called[frame_index] = 0;
- renderer_->ProcessVideoPacket(
- encoder_.Encode(*test_frames[frame_index]),
- base::BindOnce(&SetTrue, &(callback_called[frame_index])));
- }
- for (int frame_index = 0; frame_index < kFrameCount; frame_index++) {
- base::OnceClosure done_callback;
- std::unique_ptr<DesktopFrame> decoded_frame =
- frame_consumer_.WaitForNextFrame(&done_callback);
- EXPECT_FALSE(callback_called[frame_index]);
- std::move(done_callback).Run();
- EXPECT_TRUE(callback_called[frame_index]);
- EXPECT_TRUE(CompareFrames(*test_frames[frame_index], *decoded_frame));
- }
- }
- } // namespace remoting
|