| Index: remoting/test/test_video_renderer_unittest.cc
|
| diff --git a/remoting/test/test_video_renderer_unittest.cc b/remoting/test/test_video_renderer_unittest.cc
|
| index d30fd2c2400a9200d9e4a9b44caca5a3e337c793..ccfffe683343904dd3d6f6f5530c5eda0fd18b15 100644
|
| --- a/remoting/test/test_video_renderer_unittest.cc
|
| +++ b/remoting/test/test_video_renderer_unittest.cc
|
| @@ -16,6 +16,7 @@
|
| #include "remoting/codec/video_encoder_verbatim.h"
|
| #include "remoting/codec/video_encoder_vpx.h"
|
| #include "remoting/proto/video.pb.h"
|
| +#include "remoting/test/rgb_value.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
|
| #include "third_party/webrtc/modules/desktop_capture/desktop_region.h"
|
| @@ -86,11 +87,11 @@ class TestVideoRendererTest : public testing::Test {
|
| // matched.
|
| bool SendPacketAndWaitForMatch(scoped_ptr<VideoPacket> packet,
|
| const webrtc::DesktopRect& expected_rect,
|
| - uint32_t expected_average_color);
|
| + const RGBValue& expected_average_color);
|
|
|
| // Returns the average color value of pixels fall within |rect|.
|
| // NOTE: Callers should not release the objects.
|
| - uint32_t CalculateAverageColorValueForFrame(
|
| + RGBValue CalculateAverageColorValueForFrame(
|
| const webrtc::DesktopFrame* frame,
|
| const webrtc::DesktopRect& rect) const;
|
|
|
| @@ -168,7 +169,7 @@ void TestVideoRendererTest::TestVideoPacketProcessing(int screen_width,
|
| bool TestVideoRendererTest::SendPacketAndWaitForMatch(
|
| scoped_ptr<VideoPacket> packet,
|
| const webrtc::DesktopRect& expected_rect,
|
| - uint32_t expected_average_color) {
|
| + const RGBValue& expected_average_color) {
|
| DCHECK(!run_loop_ || !run_loop_->running());
|
| DCHECK(!timer_->IsRunning());
|
| run_loop_.reset(new base::RunLoop());
|
| @@ -224,7 +225,7 @@ void TestVideoRendererTest::TestImagePatternMatch(
|
|
|
| scoped_ptr<webrtc::DesktopFrame> frame =
|
| CreateDesktopFrameWithGradient(screen_width, screen_height);
|
| - uint32_t expected_average_color =
|
| + RGBValue expected_average_color =
|
| CalculateAverageColorValueForFrame(frame.get(), expected_rect);
|
| scoped_ptr<VideoPacket> packet = encoder_->Encode(*frame.get());
|
|
|
| @@ -236,19 +237,19 @@ void TestVideoRendererTest::TestImagePatternMatch(
|
| // e.g. (10, 127, 200) -> (138, 255, 73).
|
| // In this way, the error between expected color and true value is always
|
| // around 0.5.
|
| - int red_shift = (((expected_average_color >> 16) & 0xFF) + 128) % 255;
|
| - int green_shift = (((expected_average_color >> 8) & 0xFF) + 128) % 255;
|
| - int blue_shift = ((expected_average_color & 0xFF) + 128) % 255;
|
| + int red_shift = (expected_average_color.red + 128) % 255;
|
| + int green_shift = (expected_average_color.green + 128) % 255;
|
| + int blue_shift = (expected_average_color.blue + 128) % 255;
|
|
|
| - int expected_average_color_shift =
|
| - 0xFF000000 | (red_shift << 16) | (green_shift << 8) | blue_shift;
|
| + RGBValue expected_average_color_shift =
|
| + RGBValue(red_shift, green_shift, blue_shift);
|
|
|
| EXPECT_FALSE(SendPacketAndWaitForMatch(packet.Pass(), expected_rect,
|
| expected_average_color_shift));
|
| }
|
| }
|
|
|
| -uint32_t TestVideoRendererTest::CalculateAverageColorValueForFrame(
|
| +RGBValue TestVideoRendererTest::CalculateAverageColorValueForFrame(
|
| const webrtc::DesktopFrame* frame,
|
| const webrtc::DesktopRect& rect) const {
|
| int red_sum = 0;
|
| @@ -272,8 +273,8 @@ uint32_t TestVideoRendererTest::CalculateAverageColorValueForFrame(
|
| }
|
|
|
| int area = rect.width() * rect.height();
|
| - return 0xFF000000 | ((red_sum / area) << 16) | ((green_sum / area) << 8) |
|
| - (blue_sum / area);
|
| +
|
| + return RGBValue(red_sum / area, green_sum / area, blue_sum / area);
|
| }
|
|
|
| double TestVideoRendererTest::CalculateError(
|
| @@ -440,7 +441,7 @@ TEST_F(TestVideoRendererTest, VerifySetExpectedImagePattern) {
|
|
|
| // Since we don't care whether expected image pattern is matched or not in
|
| // this case, an expected color is chosen arbitrarily.
|
| - uint32_t black_color = 0xFF000000;
|
| + RGBValue black_color = RGBValue();
|
|
|
| // Set expected image pattern.
|
| test_video_renderer_->ExpectAverageColorInRect(
|
|
|