Home | History | Annotate | Download | only in thumbnails
      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
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/thumbnails/simple_thumbnail_crop.h"
      6 
      7 #include "base/basictypes.h"
      8 #include "base/strings/stringprintf.h"
      9 #include "chrome/browser/thumbnails/thumbnailing_context.h"
     10 #include "chrome/common/render_messages.h"
     11 #include "content/public/browser/notification_service.h"
     12 #include "content/public/browser/notification_types.h"
     13 #include "content/public/test/mock_render_process_host.h"
     14 #include "content/public/test/test_renderer_host.h"
     15 #include "skia/ext/platform_canvas.h"
     16 #include "testing/gtest/include/gtest/gtest.h"
     17 #include "third_party/skia/include/core/SkColorPriv.h"
     18 #include "ui/gfx/canvas.h"
     19 #include "ui/surface/transport_dib.h"
     20 
     21 using content::WebContents;
     22 using thumbnails::SimpleThumbnailCrop;
     23 
     24 typedef testing::Test SimpleThumbnailCropTest;
     25 
     26 TEST_F(SimpleThumbnailCropTest, CalculateBoringScore_Empty) {
     27   SkBitmap bitmap;
     28   EXPECT_DOUBLE_EQ(1.0, SimpleThumbnailCrop::CalculateBoringScore(bitmap));
     29 }
     30 
     31 TEST_F(SimpleThumbnailCropTest, CalculateBoringScore_SingleColor) {
     32   const gfx::Size kSize(20, 10);
     33   gfx::Canvas canvas(kSize, 1.0f, true);
     34   // Fill all pixels in black.
     35   canvas.FillRect(gfx::Rect(kSize), SK_ColorBLACK);
     36 
     37   SkBitmap bitmap =
     38       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
     39   // The thumbnail should deserve the highest boring score.
     40   EXPECT_DOUBLE_EQ(1.0, SimpleThumbnailCrop::CalculateBoringScore(bitmap));
     41 }
     42 
     43 TEST_F(SimpleThumbnailCropTest, CalculateBoringScore_TwoColors) {
     44   const gfx::Size kSize(20, 10);
     45 
     46   gfx::Canvas canvas(kSize, 1.0f, true);
     47   // Fill all pixels in black.
     48   canvas.FillRect(gfx::Rect(kSize), SK_ColorBLACK);
     49   // Fill the left half pixels in white.
     50   canvas.FillRect(gfx::Rect(0, 0, kSize.width() / 2, kSize.height()),
     51                   SK_ColorWHITE);
     52 
     53   SkBitmap bitmap =
     54       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
     55   ASSERT_EQ(kSize.width(), bitmap.width());
     56   ASSERT_EQ(kSize.height(), bitmap.height());
     57   // The thumbnail should be less boring because two colors are used.
     58   EXPECT_DOUBLE_EQ(0.5, SimpleThumbnailCrop::CalculateBoringScore(bitmap));
     59 }
     60 
     61 TEST_F(SimpleThumbnailCropTest, GetClippedBitmap_TallerThanWide) {
     62   // The input bitmap is vertically long.
     63   gfx::Canvas canvas(gfx::Size(40, 90), 1.0f, true);
     64   SkBitmap bitmap =
     65       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
     66 
     67   // The desired size is square.
     68   thumbnails::ClipResult clip_result = thumbnails::CLIP_RESULT_NOT_CLIPPED;
     69   SkBitmap clipped_bitmap = SimpleThumbnailCrop::GetClippedBitmap(
     70       bitmap, 10, 10, &clip_result);
     71   // The clipped bitmap should be square.
     72   EXPECT_EQ(40, clipped_bitmap.width());
     73   EXPECT_EQ(40, clipped_bitmap.height());
     74   // The input was taller than wide.
     75   EXPECT_EQ(thumbnails::CLIP_RESULT_TALLER_THAN_WIDE, clip_result);
     76 }
     77 
     78 TEST_F(SimpleThumbnailCropTest, GetClippedBitmap_WiderThanTall) {
     79   // The input bitmap is horizontally long.
     80   gfx::Canvas canvas(gfx::Size(70, 40), 1.0f, true);
     81   SkBitmap bitmap =
     82       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
     83 
     84   // The desired size is square.
     85   thumbnails::ClipResult clip_result = thumbnails::CLIP_RESULT_NOT_CLIPPED;
     86   SkBitmap clipped_bitmap = SimpleThumbnailCrop::GetClippedBitmap(
     87       bitmap, 10, 10, &clip_result);
     88   // The clipped bitmap should be square.
     89   EXPECT_EQ(40, clipped_bitmap.width());
     90   EXPECT_EQ(40, clipped_bitmap.height());
     91   // The input was wider than tall.
     92   EXPECT_EQ(thumbnails::CLIP_RESULT_WIDER_THAN_TALL, clip_result);
     93 }
     94 
     95 TEST_F(SimpleThumbnailCropTest, GetClippedBitmap_TooWiderThanTall) {
     96   // The input bitmap is horizontally very long.
     97   gfx::Canvas canvas(gfx::Size(90, 40), 1.0f, true);
     98   SkBitmap bitmap =
     99       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
    100 
    101   // The desired size is square.
    102   thumbnails::ClipResult clip_result = thumbnails::CLIP_RESULT_NOT_CLIPPED;
    103   SkBitmap clipped_bitmap = SimpleThumbnailCrop::GetClippedBitmap(
    104       bitmap, 10, 10, &clip_result);
    105   // The clipped bitmap should be square.
    106   EXPECT_EQ(40, clipped_bitmap.width());
    107   EXPECT_EQ(40, clipped_bitmap.height());
    108   // The input was wider than tall.
    109   EXPECT_EQ(thumbnails::CLIP_RESULT_MUCH_WIDER_THAN_TALL, clip_result);
    110 }
    111 
    112 TEST_F(SimpleThumbnailCropTest, GetClippedBitmap_NotClipped) {
    113   // The input bitmap is square.
    114   gfx::Canvas canvas(gfx::Size(40, 40), 1.0f, true);
    115   SkBitmap bitmap =
    116       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
    117 
    118   // The desired size is square.
    119   thumbnails::ClipResult clip_result = thumbnails::CLIP_RESULT_NOT_CLIPPED;
    120   SkBitmap clipped_bitmap = SimpleThumbnailCrop::GetClippedBitmap(
    121       bitmap, 10, 10, &clip_result);
    122   // The clipped bitmap should be square.
    123   EXPECT_EQ(40, clipped_bitmap.width());
    124   EXPECT_EQ(40, clipped_bitmap.height());
    125   // There was no need to clip.
    126   EXPECT_EQ(thumbnails::CLIP_RESULT_NOT_CLIPPED, clip_result);
    127 }
    128 
    129 TEST_F(SimpleThumbnailCropTest, GetClippedBitmap_NonSquareOutput) {
    130   // The input bitmap is square.
    131   gfx::Canvas canvas(gfx::Size(40, 40), 1.0f, true);
    132   SkBitmap bitmap =
    133       skia::GetTopDevice(*canvas.sk_canvas())->accessBitmap(false);
    134 
    135   // The desired size is horizontally long.
    136   thumbnails::ClipResult clip_result = thumbnails::CLIP_RESULT_NOT_CLIPPED;
    137   SkBitmap clipped_bitmap = SimpleThumbnailCrop::GetClippedBitmap(
    138       bitmap, 20, 10, &clip_result);
    139   // The clipped bitmap should have the same aspect ratio of the desired size.
    140   EXPECT_EQ(40, clipped_bitmap.width());
    141   EXPECT_EQ(20, clipped_bitmap.height());
    142   // The input was taller than wide.
    143   EXPECT_EQ(thumbnails::CLIP_RESULT_TALLER_THAN_WIDE, clip_result);
    144 }
    145 
    146 TEST_F(SimpleThumbnailCropTest, GetCanvasCopyInfo) {
    147   gfx::Size thumbnail_size(200, 120);
    148   float desired_aspect =
    149       static_cast<float>(thumbnail_size.width()) / thumbnail_size.height();
    150   scoped_refptr<thumbnails::ThumbnailingAlgorithm> algorithm(
    151       new SimpleThumbnailCrop(thumbnail_size));
    152   gfx::Rect clipping_rect_result;
    153   gfx::Size target_size_result;
    154 
    155   thumbnails::ClipResult clip_result = algorithm->GetCanvasCopyInfo(
    156       gfx::Size(400, 210),
    157       ui::SCALE_FACTOR_200P,
    158       &clipping_rect_result,
    159       &target_size_result);
    160   gfx::Size clipping_size = clipping_rect_result.size();
    161   float clip_aspect =
    162       static_cast<float>(clipping_size.width()) / clipping_size.height();
    163   EXPECT_EQ(thumbnails::CLIP_RESULT_WIDER_THAN_TALL, clip_result);
    164   EXPECT_EQ(thumbnail_size, target_size_result);
    165   EXPECT_NEAR(desired_aspect, clip_aspect, 0.01);
    166 
    167   clip_result = algorithm->GetCanvasCopyInfo(
    168       gfx::Size(600, 200),
    169       ui::SCALE_FACTOR_200P,
    170       &clipping_rect_result,
    171       &target_size_result);
    172   clipping_size = clipping_rect_result.size();
    173   clip_aspect =
    174       static_cast<float>(clipping_size.width()) / clipping_size.height();
    175   EXPECT_EQ(thumbnails::CLIP_RESULT_MUCH_WIDER_THAN_TALL, clip_result);
    176   EXPECT_EQ(thumbnail_size, target_size_result);
    177   EXPECT_NEAR(desired_aspect, clip_aspect, 0.01);
    178 
    179   clip_result = algorithm->GetCanvasCopyInfo(
    180       gfx::Size(300, 600),
    181       ui::SCALE_FACTOR_200P,
    182       &clipping_rect_result,
    183       &target_size_result);
    184   clipping_size = clipping_rect_result.size();
    185   clip_aspect =
    186       static_cast<float>(clipping_size.width()) / clipping_size.height();
    187   EXPECT_EQ(thumbnails::CLIP_RESULT_TALLER_THAN_WIDE, clip_result);
    188   EXPECT_EQ(thumbnail_size, target_size_result);
    189   EXPECT_NEAR(desired_aspect, clip_aspect, 0.01);
    190 
    191   clip_result = algorithm->GetCanvasCopyInfo(
    192       gfx::Size(200, 100),
    193       ui::SCALE_FACTOR_200P,
    194       &clipping_rect_result,
    195       &target_size_result);
    196   EXPECT_EQ(thumbnails::CLIP_RESULT_SOURCE_IS_SMALLER, clip_result);
    197   EXPECT_EQ(thumbnail_size, target_size_result);
    198 }
    199 
    200 TEST_F(SimpleThumbnailCropTest, GetClippingRect) {
    201   const gfx::Size desired_size(300, 200);
    202   thumbnails::ClipResult clip_result;
    203   // Try out 'microsource'.
    204   gfx::Rect clip_rect = SimpleThumbnailCrop::GetClippingRect(
    205       gfx::Size(300, 199), desired_size, &clip_result);
    206   EXPECT_EQ(thumbnails::CLIP_RESULT_SOURCE_IS_SMALLER, clip_result);
    207   EXPECT_EQ(gfx::Point(0, 0).ToString(), clip_rect.origin().ToString());
    208   EXPECT_EQ(desired_size.ToString(), clip_rect.size().ToString());
    209 
    210   // Portrait source.
    211   clip_rect = SimpleThumbnailCrop::GetClippingRect(
    212       gfx::Size(500, 1200), desired_size, &clip_result);
    213   EXPECT_EQ(thumbnails::CLIP_RESULT_TALLER_THAN_WIDE, clip_result);
    214   EXPECT_EQ(gfx::Point(0, 0).ToString(), clip_rect.origin().ToString());
    215   EXPECT_EQ(500, clip_rect.width());
    216   EXPECT_GE(1200, clip_rect.height());
    217 
    218   clip_rect = SimpleThumbnailCrop::GetClippingRect(
    219       gfx::Size(2000, 800), desired_size, &clip_result);
    220   EXPECT_TRUE(clip_result == thumbnails::CLIP_RESULT_WIDER_THAN_TALL ||
    221               clip_result == thumbnails::CLIP_RESULT_MUCH_WIDER_THAN_TALL);
    222   EXPECT_EQ(0, clip_rect.y());
    223   EXPECT_LT(0, clip_rect.x());
    224   EXPECT_GE(2000, clip_rect.width());
    225   EXPECT_EQ(800, clip_rect.height());
    226 
    227   clip_rect = SimpleThumbnailCrop::GetClippingRect(
    228       gfx::Size(900, 600), desired_size, &clip_result);
    229   EXPECT_EQ(thumbnails::CLIP_RESULT_NOT_CLIPPED, clip_result);
    230   EXPECT_EQ(gfx::Point(0, 0).ToString(), clip_rect.origin().ToString());
    231   EXPECT_EQ(gfx::Size(900, 600).ToString(), clip_rect.size().ToString());
    232 }
    233