Revised sub-rectangle scaling for use in Chromoting.

media/base/yuv_convert_unittest.cc

 // Copyright (c) 2011 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 "base/base_paths.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "media/base/cpu_features.h"
 #include "media/base/djb2.h"
@@ skipping 361 matching lines @@
                           yuv_converted_bytes.get(),
                           yuv_converted_bytes.get() + kSourceUOffset,
                           yuv_converted_bytes.get() + kSourceVOffset,
                           kSourceWidth, kSourceHeight);
 
   uint32 yuy_hash = DJB2Hash(yuv_converted_bytes.get(), kYUV12Size,
                              kDJB2HashSeed);
   EXPECT_EQ(666823187u, yuy_hash);
 }
 
+
+TEST(YUVConvertTest, ScaleWithQuarterRect) {
+  // Read YUV reference data from file.
+  FilePath yuv_url;
+  EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &yuv_url));
+  yuv_url = yuv_url.Append(FILE_PATH_LITERAL("media"))
+                   .Append(FILE_PATH_LITERAL("test"))
+                   .Append(FILE_PATH_LITERAL("data"))
+                   .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv"));
+  const size_t size_of_yuv = kSourceYSize * 12 / 8;  // 12 bpp.
+  scoped_array<uint8> yuv_bytes(new uint8[size_of_yuv]);
+  EXPECT_EQ(static_cast<int>(size_of_yuv),
+            file_util::ReadFile(yuv_url,
+                                reinterpret_cast<char*>(yuv_bytes.get()),
+                                static_cast<int>(size_of_yuv)));
+
+  // Scale a frame of YUV to 32 bit ARGB.
+  const size_t size_of_rgb_scaled = kScaledWidth * kScaledHeight * kBpp;
+  scoped_array<uint8> rgb_source_bytes(new uint8[size_of_rgb_scaled]);
+
+  media::ScaleYUVToRGB32WithRect(
+      yuv_bytes.get(),                        // Y
+      yuv_bytes.get() + kSourceYSize,         // U
+      yuv_bytes.get() + kSourceYSize * 5 / 4, // V
+      rgb_source_bytes.get(),                 // Rgb output
+      kSourceWidth, kSourceHeight,            // Dimensions
+      kScaledWidth, kScaledHeight,            // Dimensions
+      0, 0, kScaledWidth, kScaledHeight,      // Dest rect
+      kSourceWidth,                           // YStride
+      kSourceWidth / 2,                       // UvStride
+      kScaledWidth * kBpp);                   // RgbStride
+
+  uint32 rgb_hash_full_rect = DJB2Hash(rgb_source_bytes.get(),
+                                       size_of_rgb_scaled,
+                                       kDJB2HashSeed);
+
+  // Scale only a partial rectangle.
+  media::ScaleYUVToRGB32WithRect(
+      yuv_bytes.get(),                          // Y
+      yuv_bytes.get() + kSourceYSize,           // U
+      yuv_bytes.get() + kSourceYSize * 5 / 4,   // V
+      rgb_source_bytes.get(),                   // Rgb output
+      kSourceWidth, kSourceHeight,              // Dimensions
+      kScaledWidth, kScaledHeight,              // Dimensions
+      kScaledWidth / 4,                         // Dest rect
+      kScaledHeight / 4,
+      (kScaledWidth * 3) / 4,
+      (kScaledHeight * 3) / 4,
+      kSourceWidth,                             // YStride
+      kSourceWidth / 2,                         // UvStride
+      kScaledWidth * kBpp);                     // RgbStride
+
+  uint32 rgb_hash_quarter_rect = DJB2Hash(rgb_source_bytes.get(),
+                                          size_of_rgb_scaled,
+                                          kDJB2HashSeed);
+  EXPECT_EQ(rgb_hash_full_rect, rgb_hash_quarter_rect);
+}
+
+TEST(YUVConvertTest, ScaleWithTinyRect) {
+  // Read YUV reference data from file.
+  FilePath yuv_url;
+  EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &yuv_url));
+  yuv_url = yuv_url.Append(FILE_PATH_LITERAL("media"))
+                   .Append(FILE_PATH_LITERAL("test"))
+                   .Append(FILE_PATH_LITERAL("data"))
+                   .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv"));
+  const size_t size_of_yuv = kSourceYSize * 12 / 8;  // 12 bpp.
+  scoped_array<uint8> yuv_bytes(new uint8[size_of_yuv]);
+  EXPECT_EQ(static_cast<int>(size_of_yuv),
+            file_util::ReadFile(yuv_url,
+                                reinterpret_cast<char*>(yuv_bytes.get()),
+                                static_cast<int>(size_of_yuv)));
+
+  // Scale a frame of YUV to 32 bit ARGB.
+  const size_t size_of_rgb_scaled = kScaledWidth * kScaledHeight * kBpp;
+  scoped_array<uint8> rgb_source_bytes(new uint8[size_of_rgb_scaled]);
+
+  media::ScaleYUVToRGB32WithRect(
+      yuv_bytes.get(),                          // Y
+      yuv_bytes.get() + kSourceYSize,           // U
+      yuv_bytes.get() + kSourceYSize * 5 / 4,   // V
+      rgb_source_bytes.get(),                   // Rgb output
+      kSourceWidth, kSourceHeight,              // Dimensions
+      kScaledWidth, kScaledHeight,              // Dimensions
+      0, 0, kScaledWidth, kScaledHeight,        // Dest rect
+      kSourceWidth,                             // YStride
+      kSourceWidth / 2,                         // UvStride
+      kScaledWidth * kBpp);                     // RgbStride
+
+  uint32 rgb_hash_full_rect = DJB2Hash(rgb_source_bytes.get(),
+                                       size_of_rgb_scaled,
+                                       kDJB2HashSeed);
+
+  // Scale only one pixel in the middle
+  media::ScaleYUVToRGB32WithRect(
+      yuv_bytes.get(),                          // Y
+      yuv_bytes.get() + kSourceYSize,           // U
+      yuv_bytes.get() + kSourceYSize * 5 / 4,   // V
+      rgb_source_bytes.get(),                   // Rgb output
+      kSourceWidth, kSourceHeight,              // Dimensions
+      kScaledWidth, kScaledHeight,              // Dimensions
+      kScaledWidth / 4,                         // Dest rect
+      kScaledHeight / 4,
+      kScaledWidth / 4 + 1,
+      kScaledHeight / 4 + 1,
+      kSourceWidth,                             // YStride
+      kSourceWidth / 2,                         // UvStride
+      kScaledWidth * kBpp);                     // RgbStride
+
+  uint32 rgb_hash_tiny_rect = DJB2Hash(rgb_source_bytes.get(),
+                                       size_of_rgb_scaled,
+                                       kDJB2HashSeed);
+  EXPECT_EQ(rgb_hash_full_rect, rgb_hash_tiny_rect);
+}
+
 #if !defined(ARCH_CPU_ARM_FAMILY)
 TEST(YUVConvertTest, RGB32ToYUV_SSE2_MatchReference) {
   if (!media::hasSSE2()) {
     LOG(WARNING) << "System doesn't support SSE2, test not executed.";
     return;
   }
 
   // Allocate all surfaces.
   scoped_array<uint8> yuv_bytes(new uint8[kYUV12Size]);
   scoped_array<uint8> rgb_bytes(new uint8[kRGBSize]);
@@ skipping 418 matching lines @@
                                    kSourceDx);
   media::EmptyRegisterState();
   EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(),
                       rgb_bytes_converted.get(),
                       kWidth * kBpp));
 }
 
 #endif  // defined(ARCH_CPU_X86_64)
 
 #endif  // defined(ARCH_CPU_X86_FAMILY)