| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <algorithm> | 5 #include <algorithm> |
| 6 | 6 |
| 7 #include "remoting/base/util.h" | 7 #include "remoting/base/util.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" | 9 #include "third_party/skia/include/core/SkRect.h" |
| 10 #include "third_party/skia/include/core/SkSize.h" |
| 10 | 11 |
| 11 static const int kWidth = 32 ; | 12 static const int kWidth = 32 ; |
| 12 static const int kHeight = 24 ; | 13 static const int kHeight = 24 ; |
| 13 static const int kBytesPerPixel = 4; | 14 static const int kBytesPerPixel = 4; |
| 14 static const int kYStride = kWidth; | 15 static const int kYStride = kWidth; |
| 15 static const int kUvStride = kWidth / 2; | 16 static const int kUvStride = kWidth / 2; |
| 16 static const int kRgbStride = kWidth * kBytesPerPixel; | 17 static const int kRgbStride = kWidth * kBytesPerPixel; |
| 17 static const uint32 kFillColor = 0xffffff; | 18 static const uint32 kFillColor = 0xffffff; |
| 18 | 19 |
| 19 namespace remoting { | 20 namespace remoting { |
| (...skipping 17 matching lines...) Expand all Loading... |
| 37 ~YuvToRgbTester() {} | 38 ~YuvToRgbTester() {} |
| 38 | 39 |
| 39 void ResetYuvBuffer() { | 40 void ResetYuvBuffer() { |
| 40 memset(yuv_buffer_.get(), 0, yuv_buffer_size_); | 41 memset(yuv_buffer_.get(), 0, yuv_buffer_size_); |
| 41 } | 42 } |
| 42 | 43 |
| 43 void ResetRgbBuffer() { | 44 void ResetRgbBuffer() { |
| 44 memset(rgb_buffer_.get(), 0, rgb_buffer_size_); | 45 memset(rgb_buffer_.get(), 0, rgb_buffer_size_); |
| 45 } | 46 } |
| 46 | 47 |
| 47 void FillRgbBuffer(const webrtc::DesktopRect& rect) { | 48 void FillRgbBuffer(const SkIRect& rect) { |
| 48 uint32* ptr = reinterpret_cast<uint32*>( | 49 uint32* ptr = reinterpret_cast<uint32*>( |
| 49 rgb_buffer_.get() + (rect.top() * kRgbStride) + | 50 rgb_buffer_.get() + (rect.top() * kRgbStride) + |
| 50 (rect.left() * kBytesPerPixel)); | 51 (rect.left() * kBytesPerPixel)); |
| 51 int width = rect.width(); | 52 int width = rect.width(); |
| 52 for (int height = rect.height(); height > 0; --height) { | 53 for (int height = rect.height(); height > 0; --height) { |
| 53 std::fill(ptr, ptr + width, kFillColor); | 54 std::fill(ptr, ptr + width, kFillColor); |
| 54 ptr += kRgbStride / kBytesPerPixel; | 55 ptr += kRgbStride / kBytesPerPixel; |
| 55 } | 56 } |
| 56 } | 57 } |
| 57 | 58 |
| 58 // Check the the desination buffer is filled within expected bounds. | 59 // Check the the desination buffer is filled within expected bounds. |
| 59 void CheckRgbBuffer(const webrtc::DesktopRect& rect) { | 60 void CheckRgbBuffer(const SkIRect& rect) { |
| 60 uint32* ptr = reinterpret_cast<uint32*>(rgb_buffer_.get()); | 61 uint32* ptr = reinterpret_cast<uint32*>(rgb_buffer_.get()); |
| 61 for (int y = 0; y < kHeight; ++y) { | 62 for (int y = 0; y < kHeight; ++y) { |
| 62 if (y < rect.top() || rect.bottom() <= y) { | 63 if (y < rect.top() || rect.bottom() <= y) { |
| 63 // The whole line should be intact. | 64 // The whole line should be intact. |
| 64 EXPECT_EQ((ptrdiff_t)kWidth, | 65 EXPECT_EQ((ptrdiff_t)kWidth, |
| 65 std::count(ptr, ptr + kWidth, 0u)); | 66 std::count(ptr, ptr + kWidth, 0u)); |
| 66 } else { | 67 } else { |
| 67 // The space before the painted rectangle should be intact. | 68 // The space before the painted rectangle should be intact. |
| 68 EXPECT_EQ((ptrdiff_t)rect.left(), | 69 EXPECT_EQ((ptrdiff_t)rect.left(), |
| 69 std::count(ptr, ptr + rect.left(), 0u)); | 70 std::count(ptr, ptr + rect.left(), 0u)); |
| 70 | 71 |
| 71 // All pixels of the target rectangle should be touched. | 72 // All pixels of the target rectangle should be touched. |
| 72 EXPECT_EQ(ptr + rect.right(), | 73 EXPECT_EQ(ptr + rect.right(), |
| 73 std::find(ptr + rect.left(), ptr + rect.right(), 0u)); | 74 std::find(ptr + rect.left(), ptr + rect.right(), 0u)); |
| 74 | 75 |
| 75 // The space after the painted rectangle should be intact. | 76 // The space after the painted rectangle should be intact. |
| 76 EXPECT_EQ((ptrdiff_t)kWidth - rect.right(), | 77 EXPECT_EQ((ptrdiff_t)kWidth - rect.right(), |
| 77 std::count(ptr + rect.right(), ptr + kWidth, 0u)); | 78 std::count(ptr + rect.right(), ptr + kWidth, 0u)); |
| 78 } | 79 } |
| 79 ptr += kRgbStride / kBytesPerPixel; | 80 ptr += kRgbStride / kBytesPerPixel; |
| 80 } | 81 } |
| 81 } | 82 } |
| 82 | 83 |
| 83 void RunTest(const webrtc::DesktopSize dest_size, | 84 void RunTest(const SkISize dest_size, const SkIRect& rect) { |
| 84 const webrtc::DesktopRect& rect) { | 85 ASSERT_TRUE(SkIRect::MakeSize(dest_size).contains(rect)); |
| 85 ASSERT_TRUE( | |
| 86 DoesRectContain(webrtc::DesktopRect::MakeSize(dest_size), rect)); | |
| 87 | 86 |
| 88 // Reset buffers. | 87 // Reset buffers. |
| 89 ResetYuvBuffer(); | 88 ResetYuvBuffer(); |
| 90 ResetRgbBuffer(); | 89 ResetRgbBuffer(); |
| 91 FillRgbBuffer(rect); | 90 FillRgbBuffer(rect); |
| 92 | 91 |
| 93 // RGB -> YUV | 92 // RGB -> YUV |
| 94 ConvertRGB32ToYUVWithRect(rgb_buffer_.get(), | 93 ConvertRGB32ToYUVWithRect(rgb_buffer_.get(), |
| 95 yplane_, | 94 yplane_, |
| 96 uplane_, | 95 uplane_, |
| 97 vplane_, | 96 vplane_, |
| 98 0, | 97 0, |
| 99 0, | 98 0, |
| 100 kWidth, | 99 kWidth, |
| 101 kHeight, | 100 kHeight, |
| 102 kRgbStride, | 101 kRgbStride, |
| 103 kYStride, | 102 kYStride, |
| 104 kUvStride); | 103 kUvStride); |
| 105 | 104 |
| 106 // Reset RGB buffer and do opposite conversion. | 105 // Reset RGB buffer and do opposite conversion. |
| 107 ResetRgbBuffer(); | 106 ResetRgbBuffer(); |
| 108 ConvertAndScaleYUVToRGB32Rect(yplane_, | 107 ConvertAndScaleYUVToRGB32Rect(yplane_, |
| 109 uplane_, | 108 uplane_, |
| 110 vplane_, | 109 vplane_, |
| 111 kYStride, | 110 kYStride, |
| 112 kUvStride, | 111 kUvStride, |
| 113 webrtc::DesktopSize(kWidth, kHeight), | 112 SkISize::Make(kWidth, kHeight), |
| 114 webrtc::DesktopRect::MakeWH(kWidth, kHeight), | 113 SkIRect::MakeWH(kWidth, kHeight), |
| 115 rgb_buffer_.get(), | 114 rgb_buffer_.get(), |
| 116 kRgbStride, | 115 kRgbStride, |
| 117 dest_size, | 116 dest_size, |
| 118 webrtc::DesktopRect::MakeSize(dest_size), | 117 SkIRect::MakeSize(dest_size), |
| 119 rect); | 118 rect); |
| 120 | 119 |
| 121 // Check if it worked out. | 120 // Check if it worked out. |
| 122 CheckRgbBuffer(rect); | 121 CheckRgbBuffer(rect); |
| 123 } | 122 } |
| 124 | 123 |
| 125 void TestBasicConversion() { | 124 void TestBasicConversion() { |
| 126 // Whole buffer. | 125 // Whole buffer. |
| 127 RunTest(webrtc::DesktopSize(kWidth, kHeight), | 126 RunTest(SkISize::Make(kWidth, kHeight), SkIRect::MakeWH(kWidth, kHeight)); |
| 128 webrtc::DesktopRect::MakeWH(kWidth, kHeight)); | |
| 129 } | 127 } |
| 130 | 128 |
| 131 private: | 129 private: |
| 132 size_t yuv_buffer_size_; | 130 size_t yuv_buffer_size_; |
| 133 scoped_ptr<uint8[]> yuv_buffer_; | 131 scoped_ptr<uint8[]> yuv_buffer_; |
| 134 uint8* yplane_; | 132 uint8* yplane_; |
| 135 uint8* uplane_; | 133 uint8* uplane_; |
| 136 uint8* vplane_; | 134 uint8* vplane_; |
| 137 | 135 |
| 138 size_t rgb_buffer_size_; | 136 size_t rgb_buffer_size_; |
| 139 scoped_ptr<uint8[]> rgb_buffer_; | 137 scoped_ptr<uint8[]> rgb_buffer_; |
| 140 | 138 |
| 141 DISALLOW_COPY_AND_ASSIGN(YuvToRgbTester); | 139 DISALLOW_COPY_AND_ASSIGN(YuvToRgbTester); |
| 142 }; | 140 }; |
| 143 | 141 |
| 144 TEST(YuvToRgbTest, BasicConversion) { | 142 TEST(YuvToRgbTest, BasicConversion) { |
| 145 YuvToRgbTester tester; | 143 YuvToRgbTester tester; |
| 146 tester.TestBasicConversion(); | 144 tester.TestBasicConversion(); |
| 147 } | 145 } |
| 148 | 146 |
| 149 TEST(YuvToRgbTest, Clipping) { | 147 TEST(YuvToRgbTest, Clipping) { |
| 150 YuvToRgbTester tester; | 148 YuvToRgbTester tester; |
| 151 | 149 |
| 152 webrtc::DesktopSize dest_size = webrtc::DesktopSize(kWidth, kHeight); | 150 SkISize dest_size = SkISize::Make(kWidth, kHeight); |
| 153 webrtc::DesktopRect rect = | 151 SkIRect rect = SkIRect::MakeLTRB(0, 0, kWidth - 1, kHeight - 1); |
| 154 webrtc::DesktopRect::MakeLTRB(0, 0, kWidth - 1, kHeight - 1); | |
| 155 for (int i = 0; i < 16; ++i) { | 152 for (int i = 0; i < 16; ++i) { |
| 156 webrtc::DesktopRect dest_rect = webrtc::DesktopRect::MakeLTRB( | 153 SkIRect dest_rect = rect; |
| 157 rect.left() + ((i & 1) ? 1 : 0), | 154 if ((i & 1) != 0) |
| 158 rect.top() + ((i & 2) ? 1 : 0), | 155 dest_rect.fLeft += 1; |
| 159 rect.right() + ((i & 4) ? 1 : 0), | 156 if ((i & 2) != 0) |
| 160 rect.bottom() + ((i & 8) ? 1 : 0)); | 157 dest_rect.fTop += 1; |
| 158 if ((i & 4) != 0) |
| 159 dest_rect.fRight += 1; |
| 160 if ((i & 8) != 0) |
| 161 dest_rect.fBottom += 1; |
| 161 | 162 |
| 162 tester.RunTest(dest_size, dest_rect); | 163 tester.RunTest(dest_size, dest_rect); |
| 163 } | 164 } |
| 164 } | 165 } |
| 165 | 166 |
| 166 TEST(YuvToRgbTest, ClippingAndScaling) { | 167 TEST(YuvToRgbTest, ClippingAndScaling) { |
| 167 YuvToRgbTester tester; | 168 YuvToRgbTester tester; |
| 168 | 169 |
| 169 webrtc::DesktopSize dest_size = | 170 SkISize dest_size = SkISize::Make(kWidth - 10, kHeight - 10); |
| 170 webrtc::DesktopSize(kWidth - 10, kHeight - 10); | 171 SkIRect rect = SkIRect::MakeLTRB(5, 5, kWidth - 11, kHeight - 11); |
| 171 webrtc::DesktopRect rect = | |
| 172 webrtc::DesktopRect::MakeLTRB(5, 5, kWidth - 11, kHeight - 11); | |
| 173 for (int i = 0; i < 16; ++i) { | 172 for (int i = 0; i < 16; ++i) { |
| 174 webrtc::DesktopRect dest_rect = webrtc::DesktopRect::MakeLTRB( | 173 SkIRect dest_rect = rect; |
| 175 rect.left() + ((i & 1) ? 1 : 0), | 174 if ((i & 1) != 0) |
| 176 rect.top() + ((i & 2) ? 1 : 0), | 175 dest_rect.fLeft += 1; |
| 177 rect.right() + ((i & 4) ? 1 : 0), | 176 if ((i & 2) != 0) |
| 178 rect.bottom() + ((i & 8) ? 1 : 0)); | 177 dest_rect.fTop += 1; |
| 178 if ((i & 4) != 0) |
| 179 dest_rect.fRight += 1; |
| 180 if ((i & 8) != 0) |
| 181 dest_rect.fBottom += 1; |
| 179 | 182 |
| 180 tester.RunTest(dest_size, dest_rect); | 183 tester.RunTest(dest_size, dest_rect); |
| 181 } | 184 } |
| 182 } | 185 } |
| 183 | 186 |
| 184 TEST(ReplaceLfByCrLfTest, Basic) { | 187 TEST(ReplaceLfByCrLfTest, Basic) { |
| 185 EXPECT_EQ("ab", ReplaceLfByCrLf("ab")); | 188 EXPECT_EQ("ab", ReplaceLfByCrLf("ab")); |
| 186 EXPECT_EQ("\r\nab", ReplaceLfByCrLf("\nab")); | 189 EXPECT_EQ("\r\nab", ReplaceLfByCrLf("\nab")); |
| 187 EXPECT_EQ("\r\nab\r\n", ReplaceLfByCrLf("\nab\n")); | 190 EXPECT_EQ("\r\nab\r\n", ReplaceLfByCrLf("\nab\n")); |
| 188 EXPECT_EQ("\r\nab\r\ncd", ReplaceLfByCrLf("\nab\ncd")); | 191 EXPECT_EQ("\r\nab\r\ncd", ReplaceLfByCrLf("\nab\ncd")); |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 278 EXPECT_FALSE(StringIsUtf8("\xfe\x80\x80\x80\x80\x80\x80", 7)); | 281 EXPECT_FALSE(StringIsUtf8("\xfe\x80\x80\x80\x80\x80\x80", 7)); |
| 279 EXPECT_FALSE(StringIsUtf8("\xff\x80\x80\x80\x80\x80\x80", 7)); | 282 EXPECT_FALSE(StringIsUtf8("\xff\x80\x80\x80\x80\x80\x80", 7)); |
| 280 | 283 |
| 281 // Invalid continuation byte | 284 // Invalid continuation byte |
| 282 EXPECT_FALSE(StringIsUtf8("\xc0\x00", 2)); | 285 EXPECT_FALSE(StringIsUtf8("\xc0\x00", 2)); |
| 283 EXPECT_FALSE(StringIsUtf8("\xc0\x40", 2)); | 286 EXPECT_FALSE(StringIsUtf8("\xc0\x40", 2)); |
| 284 EXPECT_FALSE(StringIsUtf8("\xc0\xc0", 2)); | 287 EXPECT_FALSE(StringIsUtf8("\xc0\xc0", 2)); |
| 285 } | 288 } |
| 286 | 289 |
| 287 } // namespace remoting | 290 } // namespace remoting |
| OLD | NEW |