Home | History | Annotate | Download | only in image
      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 "ui/gfx/image/image_skia_util_mac.h"
      6 
      7 #include <cmath>
      8 #include <limits>
      9 
     10 #import <AppKit/AppKit.h>
     11 
     12 #include "base/mac/mac_util.h"
     13 #include "base/mac/scoped_nsobject.h"
     14 #include "base/memory/scoped_ptr.h"
     15 #include "skia/ext/skia_utils_mac.h"
     16 #include "third_party/skia/include/core/SkBitmap.h"
     17 #include "ui/gfx/image/image_skia.h"
     18 
     19 namespace {
     20 
     21 // Returns NSImageRep whose pixel size most closely matches |desired_size|.
     22 NSImageRep* GetNSImageRepWithPixelSize(NSImage* image,
     23                                        NSSize desired_size) {
     24   float smallest_diff = std::numeric_limits<float>::max();
     25   NSImageRep* closest_match = nil;
     26   for (NSImageRep* image_rep in [image representations]) {
     27     float diff = std::abs(desired_size.width - [image_rep pixelsWide]) +
     28         std::abs(desired_size.height - [image_rep pixelsHigh]);
     29     if (diff < smallest_diff) {
     30       smallest_diff = diff;
     31       closest_match = image_rep;
     32     }
     33   }
     34   return closest_match;
     35 }
     36 
     37 // Returns true if NSImage has no representations
     38 bool IsNSImageEmpty(NSImage* image) {
     39   return ([image representations].count == 0);
     40 }
     41 
     42 }  // namespace
     43 
     44 namespace gfx {
     45 
     46 gfx::ImageSkia ImageSkiaFromNSImage(NSImage* image) {
     47   return ImageSkiaFromResizedNSImage(image, [image size]);
     48 }
     49 
     50 gfx::ImageSkia ImageSkiaFromResizedNSImage(NSImage* image,
     51                                            NSSize desired_size) {
     52   // Resize and convert to ImageSkia simultaneously to save on computation.
     53   // TODO(pkotwicz): Separate resizing NSImage and converting to ImageSkia.
     54   // Convert to ImageSkia by finding the most appropriate NSImageRep for
     55   // each supported scale factor and resizing if necessary.
     56 
     57   if (IsNSImageEmpty(image))
     58     return gfx::ImageSkia();
     59 
     60   std::vector<float> supported_scales = ImageSkia::GetSupportedScales();
     61 
     62   gfx::ImageSkia image_skia;
     63   for (size_t i = 0; i < supported_scales.size(); ++i) {
     64     float scale = supported_scales[i];
     65     NSSize desired_size_for_scale = NSMakeSize(desired_size.width * scale,
     66                                                desired_size.height * scale);
     67     NSImageRep* ns_image_rep = GetNSImageRepWithPixelSize(image,
     68         desired_size_for_scale);
     69 
     70     // TODO(dcheng): Should this function take a color space argument?
     71     SkBitmap bitmap(gfx::NSImageRepToSkBitmapWithColorSpace(ns_image_rep,
     72         desired_size_for_scale, false, base::mac::GetGenericRGBColorSpace()));
     73     if (bitmap.isNull())
     74       continue;
     75 
     76     image_skia.AddRepresentation(gfx::ImageSkiaRep(bitmap, scale));
     77   }
     78   return image_skia;
     79 }
     80 
     81 NSImage* NSImageFromImageSkia(const gfx::ImageSkia& image_skia) {
     82   if (image_skia.isNull())
     83     return nil;
     84 
     85   base::scoped_nsobject<NSImage> image([[NSImage alloc] init]);
     86   image_skia.EnsureRepsForSupportedScales();
     87   std::vector<gfx::ImageSkiaRep> image_reps = image_skia.image_reps();
     88   for (std::vector<gfx::ImageSkiaRep>::const_iterator it = image_reps.begin();
     89        it != image_reps.end(); ++it) {
     90     [image addRepresentation:
     91         gfx::SkBitmapToNSBitmapImageRep(it->sk_bitmap())];
     92   }
     93 
     94   [image setSize:NSMakeSize(image_skia.width(), image_skia.height())];
     95   return [image.release() autorelease];
     96 }
     97 
     98 NSImage* NSImageFromImageSkiaWithColorSpace(const gfx::ImageSkia& image_skia,
     99                                             CGColorSpaceRef color_space) {
    100   if (image_skia.isNull())
    101     return nil;
    102 
    103   base::scoped_nsobject<NSImage> image([[NSImage alloc] init]);
    104   image_skia.EnsureRepsForSupportedScales();
    105   std::vector<gfx::ImageSkiaRep> image_reps = image_skia.image_reps();
    106   for (std::vector<gfx::ImageSkiaRep>::const_iterator it = image_reps.begin();
    107        it != image_reps.end(); ++it) {
    108     [image addRepresentation:
    109         gfx::SkBitmapToNSBitmapImageRepWithColorSpace(it->sk_bitmap(),
    110                                                       color_space)];
    111   }
    112 
    113   [image setSize:NSMakeSize(image_skia.width(), image_skia.height())];
    114   return [image.release() autorelease];
    115 }
    116 
    117 }  // namespace gfx
    118