// Copyright 2014 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/codec/video_encoder_helper.h" #include #include "remoting/proto/video.pb.h" #include "testing/gtest/include/gtest/gtest.h" #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" using webrtc::BasicDesktopFrame; using webrtc::DesktopRect; using webrtc::DesktopRegion; using webrtc::DesktopSize; using webrtc::DesktopVector; namespace remoting { TEST(VideoEncoderHelperTest, PropagatesCommonFields) { BasicDesktopFrame frame(DesktopSize(32, 32)); frame.set_dpi(DesktopVector(96, 97)); frame.set_capture_time_ms(20); frame.mutable_updated_region()->SetRect(DesktopRect::MakeLTRB(0, 0, 16, 16)); VideoEncoderHelper helper; std::unique_ptr packet(helper.CreateVideoPacket(frame)); ASSERT_TRUE(packet->has_format()); EXPECT_FALSE(packet->format().has_encoding()); EXPECT_TRUE(packet->format().has_screen_width()); EXPECT_TRUE(packet->format().has_screen_height()); EXPECT_TRUE(packet->format().has_x_dpi()); EXPECT_TRUE(packet->format().has_y_dpi()); EXPECT_EQ(1, packet->dirty_rects().size()); } TEST(VideoEncoderHelperTest, ZeroDpi) { BasicDesktopFrame frame(DesktopSize(32, 32)); // DPI is zero unless explicitly set. VideoEncoderHelper helper; std::unique_ptr packet(helper.CreateVideoPacket(frame)); // Packet should have a format containing the screen dimensions only. ASSERT_TRUE(packet->has_format()); EXPECT_TRUE(packet->format().has_screen_width()); EXPECT_TRUE(packet->format().has_screen_height()); EXPECT_FALSE(packet->format().has_x_dpi()); EXPECT_FALSE(packet->format().has_y_dpi()); } TEST(VideoEncoderHelperTest, NoScreenSizeIfUnchanged) { BasicDesktopFrame frame(DesktopSize(32, 32)); // Set DPI so that the packet will have a format, with DPI but no size. frame.set_dpi(DesktopVector(96, 97)); VideoEncoderHelper helper; std::unique_ptr packet(helper.CreateVideoPacket(frame)); packet = helper.CreateVideoPacket(frame); ASSERT_TRUE(packet->has_format()); EXPECT_FALSE(packet->format().has_screen_width()); EXPECT_FALSE(packet->format().has_screen_height()); EXPECT_TRUE(packet->format().has_x_dpi()); EXPECT_TRUE(packet->format().has_y_dpi()); } TEST(VideoEncoderHelperTest, ScreenSizeWhenChanged) { VideoEncoderHelper helper; // Process the same frame twice, so the helper knows the current size, and // to trigger suppression of the size field due to the size not changing. BasicDesktopFrame frame1(DesktopSize(32, 32)); std::unique_ptr packet(helper.CreateVideoPacket(frame1)); packet = helper.CreateVideoPacket(frame1); // Process a different-sized frame to trigger size to be emitted. BasicDesktopFrame frame2(DesktopSize(48, 48)); packet = helper.CreateVideoPacket(frame2); ASSERT_TRUE(packet->has_format()); EXPECT_TRUE(packet->format().has_screen_width()); EXPECT_TRUE(packet->format().has_screen_height()); } } // namespace remoting