15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/icon_util.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 8a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles)#include "base/files/important_file_writer.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/resource_util.h" 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/scoped_gdi_object.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/scoped_hdc.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/image_operations.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/skia/include/core/SkBitmap.h" 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/gdi_util.h" 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/image/image.h" 19c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "ui/gfx/image/image_family.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/size.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct ScopedICONINFO : ICONINFO { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedICONINFO() { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hbmColor = NULL; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hbmMask = NULL; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ScopedICONINFO() { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (hbmColor) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteObject(hbmColor); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (hbmMask) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteObject(hbmMask); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 38c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Creates a new ImageFamily, |resized_image_family|, based on the images in 39c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |image_family|, but containing images of specific dimensions desirable for 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Windows icons. For each desired image dimension, it chooses the most 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// appropriate image for that size, and resizes it to the desired size. 42c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Returns true on success, false on failure. Failure can occur if 43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |image_family| is empty, all images in the family have size 0x0, or an image 44c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// has no allocated pixel data. 45c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |resized_image_family| must be empty. 46c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool BuildResizedImageFamily(const gfx::ImageFamily& image_family, 47c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) gfx::ImageFamily* resized_image_family) { 48c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(resized_image_family); 49c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(resized_image_family->empty()); 50c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 51c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) for (size_t i = 0; i < IconUtil::kNumIconDimensions; ++i) { 52c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int dimension = IconUtil::kIconDimensions[i]; 53c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) gfx::Size size(dimension, dimension); 54c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const gfx::Image* best = image_family.GetBest(size); 55c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!best || best->IsEmpty()) { 56c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Either |image_family| is empty, or all images have size 0x0. 57c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 58c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 59c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 60c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Optimize for the "Large icons" view in Windows Vista+. This view displays 61c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // icons at full size if only if there is a 256x256 (kLargeIconSize) image 62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // in the .ico file. Otherwise, it shrinks icons to 48x48 (kMediumIconSize). 63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (dimension > IconUtil::kMediumIconSize && 64c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) best->Width() <= IconUtil::kMediumIconSize && 65c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) best->Height() <= IconUtil::kMediumIconSize) { 66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // There is no source icon larger than 48x48, so do not create any 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // images larger than 48x48. kIconDimensions is sorted in ascending 68c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // order, so it is safe to break here. 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) break; 70c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 72c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (best->Size() == size) { 73c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) resized_image_family->Add(*best); 74c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } else { 75c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // There is no |dimension|x|dimension| source image. 76c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Resize this one to the desired size, and insert it. 77c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SkBitmap best_bitmap = best->AsBitmap(); 787d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // Only kARGB_8888 images are supported. 797d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // This will also filter out images with no pixels. 80116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (best_bitmap.colorType() != kN32_SkColorType) 81c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 82c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SkBitmap resized_bitmap = skia::ImageOperations::Resize( 83c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) best_bitmap, skia::ImageOperations::RESIZE_LANCZOS3, 84c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) dimension, dimension); 85c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) resized_image_family->Add(gfx::Image::CreateFrom1xBitmap(resized_bitmap)); 86c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 87c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 88c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return true; 89c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 90c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 91c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Creates a set of bitmaps from an image family. 92c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// All images smaller than 256x256 are converted to SkBitmaps, and inserted into 93c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |bitmaps| in order of aspect ratio (thinnest to widest), and then ascending 94c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// size order. If an image of exactly 256x256 is specified, it is converted into 95c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// PNG format and stored in |png_bytes|. Images with width or height larger than 96c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// 256 are ignored. 97c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |bitmaps| must be an empty vector, and not NULL. 98c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Returns true on success, false on failure. This fails if any image in 99c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// |image_family| is not a 32-bit ARGB image, or is otherwise invalid. 100c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool ConvertImageFamilyToBitmaps( 101c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const gfx::ImageFamily& image_family, 102c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) std::vector<SkBitmap>* bitmaps, 103c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_refptr<base::RefCountedMemory>* png_bytes) { 104c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(bitmaps != NULL); 105c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(bitmaps->empty()); 106c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 107c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) for (gfx::ImageFamily::const_iterator it = image_family.begin(); 108c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) it != image_family.end(); ++it) { 109c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const gfx::Image& image = *it; 110c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 111c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // All images should have one of the kIconDimensions sizes. 112c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_GT(image.Width(), 0); 113c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_LE(image.Width(), IconUtil::kLargeIconSize); 114c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_GT(image.Height(), 0); 115c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_LE(image.Height(), IconUtil::kLargeIconSize); 116c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 117c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SkBitmap bitmap = image.AsBitmap(); 118c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Only 32 bit ARGB bitmaps are supported. We also make sure the bitmap has 120c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // been properly initialized. 121c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SkAutoLockPixels bitmap_lock(bitmap); 122116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if ((bitmap.colorType() != kN32_SkColorType) || 123c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) (bitmap.getPixels() == NULL)) { 124c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 125c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 126c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 127c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Special case: Icons exactly 256x256 are stored in PNG format. 128c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (image.Width() == IconUtil::kLargeIconSize && 129c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) image.Height() == IconUtil::kLargeIconSize) { 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) *png_bytes = image.As1xPNGBytes(); 131c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } else { 132c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) bitmaps->push_back(bitmap); 133c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 134c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 135c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 136c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return true; 137c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 138c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The icon images appear in the icon file in same order in which their 142c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// corresponding dimensions appear in this array, so it is important to keep 143c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// this array sorted. Also note that the maximum icon image size we can handle 144c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// is 256 by 256. See: 145c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// http://msdn.microsoft.com/en-us/library/windows/desktop/aa511280.aspx#size 146c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const int IconUtil::kIconDimensions[] = { 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8, // Recommended by the MSDN as a nice to have icon size. 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10, // Used by the Shell (e.g. for shortcuts). 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 14, // Recommended by the MSDN as a nice to have icon size. 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 16, // Toolbar, Application and Shell icon sizes. 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 22, // Recommended by the MSDN as a nice to have icon size. 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 24, // Used by the Shell (e.g. for shortcuts). 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 32, // Toolbar, Dialog and Wizard icon size. 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 40, // Quick Launch. 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 48, // Alt+Tab icon size. 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 64, // Recommended by the MSDN as a nice to have icon size. 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 96, // Recommended by the MSDN as a nice to have icon size. 158c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 128, // Used by the Shell (e.g. for shortcuts). 159c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 256 // Used by Vista onwards for large icons. 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 162c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const size_t IconUtil::kNumIconDimensions = arraysize(kIconDimensions); 163c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const size_t IconUtil::kNumIconDimensionsUpToMediumSize = 9; 164c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HICON IconUtil::CreateHICONFromSkBitmap(const SkBitmap& bitmap) { 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only 32 bit ARGB bitmaps are supported. We also try to perform as many 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // validations as we can on the bitmap. 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkAutoLockPixels bitmap_lock(bitmap); 169116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if ((bitmap.colorType() != kN32_SkColorType) || 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (bitmap.width() <= 0) || (bitmap.height() <= 0) || 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (bitmap.getPixels() == NULL)) 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start by creating a DIB which we'll use later on in order to create 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the HICON. We use BITMAPV5HEADER since the bitmap we are about to convert 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // may contain an alpha channel and the V5 header allows us to specify the 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alpha mask for the DIB. 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BITMAPV5HEADER bitmap_header; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) InitializeBitmapHeader(&bitmap_header, bitmap.width(), bitmap.height()); 180c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 181c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch void* bits = NULL; 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HBITMAP dib; 183c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 184c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch { 185c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch base::win::ScopedGetDC hdc(NULL); 186c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch dib = ::CreateDIBSection(hdc, reinterpret_cast<BITMAPINFO*>(&bitmap_header), 187c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch DIB_RGB_COLORS, &bits, NULL, 0); 188c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch } 189c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch if (!dib || !bits) 190c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch return NULL; 191c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memcpy(bits, bitmap.getPixels(), bitmap.width() * bitmap.height() * 4); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Icons are generally created using an AND and XOR masks where the AND 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // specifies boolean transparency (the pixel is either opaque or 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // transparent) and the XOR mask contains the actual image pixels. If the XOR 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // mask bitmap has an alpha channel, the AND monochrome bitmap won't 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // actually be used for computing the pixel transparency. Even though all our 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bitmap has an alpha channel, Windows might not agree when all alpha values 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // are zero. So the monochrome bitmap is created with all pixels transparent 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // for this case. Otherwise, it is created with all pixels opaque. 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool bitmap_has_alpha_channel = PixelsHaveAlpha( 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<const uint32*>(bitmap.getPixels()), 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bitmap.width() * bitmap.height()); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<uint8[]> mask_bits; 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!bitmap_has_alpha_channel) { 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bytes per line with paddings to make it word alignment. 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bytes_per_line = (bitmap.width() + 0xF) / 16 * 2; 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t mask_bits_size = bytes_per_line * bitmap.height(); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mask_bits.reset(new uint8[mask_bits_size]); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(mask_bits.get()); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make all pixels transparent. 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memset(mask_bits.get(), 0xFF, mask_bits_size); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HBITMAP mono_bitmap = ::CreateBitmap(bitmap.width(), bitmap.height(), 1, 1, 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<LPVOID>(mask_bits.get())); 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(mono_bitmap); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ICONINFO icon_info; 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_info.fIcon = TRUE; 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_info.xHotspot = 0; 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_info.yHotspot = 0; 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_info.hbmMask = mono_bitmap; 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_info.hbmColor = dib; 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HICON icon = ::CreateIconIndirect(&icon_info); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteObject(dib); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteObject(mono_bitmap); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return icon; 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SkBitmap* IconUtil::CreateSkBitmapFromHICON(HICON icon, const gfx::Size& s) { 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start with validating parameters. 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!icon || s.IsEmpty()) 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedICONINFO icon_info; 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::GetIconInfo(icon, &icon_info)) 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!icon_info.fIcon) 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new SkBitmap(CreateSkBitmapFromHICONHelper(icon, s)); 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)scoped_ptr<SkBitmap> IconUtil::CreateSkBitmapFromIconResource(HMODULE module, 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int resource_id, 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int size) { 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_LE(size, kLargeIconSize); 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // For everything except the Vista+ 256x256 icons, use |LoadImage()|. 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (size != kLargeIconSize) { 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HICON icon_handle = 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) static_cast<HICON>(LoadImage(module, MAKEINTRESOURCE(resource_id), 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IMAGE_ICON, size, size, 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LR_DEFAULTCOLOR | LR_DEFAULTSIZE)); 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<SkBitmap> bitmap(IconUtil::CreateSkBitmapFromHICON(icon_handle)); 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DestroyIcon(icon_handle); 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return bitmap.Pass(); 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // For Vista+ 256x256 PNG icons, read the resource directly and find 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the corresponding icon entry to get its PNG bytes. 2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void* icon_dir_data = NULL; 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t icon_dir_size = 0; 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!base::win::GetResourceFromModule(module, resource_id, RT_GROUP_ICON, 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &icon_dir_data, &icon_dir_size)) { 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return scoped_ptr<SkBitmap>(); 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(icon_dir_data); 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_GE(icon_dir_size, sizeof(GRPICONDIR)); 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const GRPICONDIR* icon_dir = 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) reinterpret_cast<const GRPICONDIR*>(icon_dir_data); 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const GRPICONDIRENTRY* large_icon_entry = NULL; 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < icon_dir->idCount; ++i) { 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const GRPICONDIRENTRY* entry = &icon_dir->idEntries[i]; 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // 256x256 icons are stored with width and height set to 0. 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // See: http://en.wikipedia.org/wiki/ICO_(file_format) 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (entry->bWidth == 0 && entry->bHeight == 0) { 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) large_icon_entry = entry; 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break; 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!large_icon_entry) 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return scoped_ptr<SkBitmap>(); 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void* png_data = NULL; 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t png_size = 0; 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!base::win::GetResourceFromModule(module, large_icon_entry->nID, RT_ICON, 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &png_data, &png_size)) { 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return scoped_ptr<SkBitmap>(); 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(png_data); 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(png_size, large_icon_entry->dwBytesInRes); 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) gfx::Image image = gfx::Image::CreateFrom1xPNGBytes( 2995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) new base::RefCountedStaticMemory(png_data, png_size)); 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return scoped_ptr<SkBitmap>(new SkBitmap(image.AsBitmap())); 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SkBitmap* IconUtil::CreateSkBitmapFromHICON(HICON icon) { 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start with validating parameters. 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!icon) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedICONINFO icon_info; 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BITMAP bitmap_info = { 0 }; 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::GetIconInfo(icon, &icon_info)) 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::GetObject(icon_info.hbmMask, sizeof(bitmap_info), &bitmap_info)) 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) gfx::Size icon_size(bitmap_info.bmWidth, bitmap_info.bmHeight); 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new SkBitmap(CreateSkBitmapFromHICONHelper(icon, icon_size)); 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)HICON IconUtil::CreateCursorFromDIB(const gfx::Size& icon_size, 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const gfx::Point& hotspot, 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const void* dib_bits, 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t dib_size) { 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BITMAPINFO icon_bitmap_info = {0}; 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gfx::CreateBitmapHeader( 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) icon_size.width(), 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) icon_size.height(), 3292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) reinterpret_cast<BITMAPINFOHEADER*>(&icon_bitmap_info)); 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::win::ScopedGetDC dc(NULL); 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::win::ScopedCreateDC working_dc(CreateCompatibleDC(dc)); 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::win::ScopedGDIObject<HBITMAP> bitmap_handle( 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateDIBSection(dc, 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &icon_bitmap_info, 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DIB_RGB_COLORS, 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 0, 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 0, 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 0)); 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (dib_size > 0) { 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SetDIBits(0, 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bitmap_handle, 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 0, 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) icon_size.height(), 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) dib_bits, 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &icon_bitmap_info, 3472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DIB_RGB_COLORS); 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HBITMAP old_bitmap = reinterpret_cast<HBITMAP>( 3511320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SelectObject(working_dc.Get(), bitmap_handle)); 3521320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SetBkMode(working_dc.Get(), TRANSPARENT); 3531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SelectObject(working_dc.Get(), old_bitmap); 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::win::ScopedGDIObject<HBITMAP> mask( 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateBitmap(icon_size.width(), 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) icon_size.height(), 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1, 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1, 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NULL)); 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ICONINFO ii = {0}; 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ii.fIcon = FALSE; 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ii.xHotspot = hotspot.x(); 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ii.yHotspot = hotspot.y(); 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ii.hbmMask = mask; 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ii.hbmColor = bitmap_handle; 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return CreateIconIndirect(&ii); 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SkBitmap IconUtil::CreateSkBitmapFromHICONHelper(HICON icon, 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const gfx::Size& s) { 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(icon); 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!s.IsEmpty()); 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Allocating memory for the SkBitmap object. We are going to create an ARGB 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bitmap so we should set the configuration appropriately. 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkBitmap bitmap; 379a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) bitmap.allocN32Pixels(s.width(), s.height()); 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bitmap.eraseARGB(0, 0, 0, 0); 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkAutoLockPixels bitmap_lock(bitmap); 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Now we should create a DIB so that we can use ::DrawIconEx in order to 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // obtain the icon's image. 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BITMAPV5HEADER h; 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) InitializeBitmapHeader(&h, s.width(), s.height()); 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HDC hdc = ::GetDC(NULL); 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint32* bits; 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HBITMAP dib = ::CreateDIBSection(hdc, reinterpret_cast<BITMAPINFO*>(&h), 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DIB_RGB_COLORS, reinterpret_cast<void**>(&bits), NULL, 0); 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(dib); 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HDC dib_dc = CreateCompatibleDC(hdc); 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::ReleaseDC(NULL, hdc); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(dib_dc); 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HGDIOBJ old_obj = ::SelectObject(dib_dc, dib); 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Windows icons are defined using two different masks. The XOR mask, which 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // represents the icon image and an AND mask which is a monochrome bitmap 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // which indicates the transparency of each pixel. 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // To make things more complex, the icon image itself can be an ARGB bitmap 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and therefore contain an alpha channel which specifies the transparency 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // for each pixel. Unfortunately, there is no easy way to determine whether 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // or not a bitmap has an alpha channel and therefore constructing the bitmap 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // for the icon is nothing but straightforward. 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The idea is to read the AND mask but use it only if we know for sure that 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the icon image does not have an alpha channel. The only way to tell if the 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bitmap has an alpha channel is by looking through the pixels and checking 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // whether there are non-zero alpha bytes. 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start by drawing the AND mask into our DIB. 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t num_pixels = s.GetArea(); 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memset(bits, 0, num_pixels * 4); 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DrawIconEx(dib_dc, 0, 0, icon, s.width(), s.height(), 0, NULL, DI_MASK); 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Capture boolean opacity. We may not use it if we find out the bitmap has 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // an alpha channel. 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<bool[]> opaque(new bool[num_pixels]); 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < num_pixels; ++i) 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) opaque[i] = !bits[i]; 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Then draw the image itself which is really the XOR mask. 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memset(bits, 0, num_pixels * 4); 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DrawIconEx(dib_dc, 0, 0, icon, s.width(), s.height(), 0, NULL, DI_NORMAL); 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memcpy(bitmap.getPixels(), static_cast<void*>(bits), num_pixels * 4); 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Finding out whether the bitmap has an alpha channel. 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool bitmap_has_alpha_channel = PixelsHaveAlpha( 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<const uint32*>(bitmap.getPixels()), num_pixels); 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the bitmap does not have an alpha channel, we need to build it using 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the previously captured AND mask. Otherwise, we are done. 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!bitmap_has_alpha_channel) { 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint32* p = static_cast<uint32*>(bitmap.getPixels()); 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < num_pixels; ++p, ++i) { 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ((*p & 0xff000000), 0u); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (opaque[i]) 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *p |= 0xff000000; 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *p &= 0x00ffffff; 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::SelectObject(dib_dc, old_obj); 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteObject(dib); 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::DeleteDC(dib_dc); 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return bitmap; 4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 452c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// static 453c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool IconUtil::CreateIconFileFromImageFamily( 454c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const gfx::ImageFamily& image_family, 455c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::FilePath& icon_path) { 456c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Creating a set of bitmaps corresponding to the icon images we'll end up 457c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // storing in the icon file. Each bitmap is created by resizing the most 458c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // appropriate image from |image_family| to the desired size. 459c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) gfx::ImageFamily resized_image_family; 460c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!BuildResizedImageFamily(image_family, &resized_image_family)) 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 4622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 463c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) std::vector<SkBitmap> bitmaps; 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::RefCountedMemory> png_bytes; 465c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!ConvertImageFamilyToBitmaps(resized_image_family, &bitmaps, &png_bytes)) 466c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 468c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Guaranteed true because BuildResizedImageFamily will provide at least one 469c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // image < 256x256. 470c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(!bitmaps.empty()); 471c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) size_t bitmap_count = bitmaps.size(); // Not including PNG image. 472c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Including PNG image, if any. 473c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) size_t image_count = bitmap_count + (png_bytes.get() ? 1 : 0); 474c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Computing the total size of the buffer we need in order to store the 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // images in the desired icon format. 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t buffer_size = ComputeIconFileBufferSize(bitmaps); 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Account for the bytes needed for the PNG entry. 4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (png_bytes.get()) 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) buffer_size += sizeof(ICONDIRENTRY) + png_bytes->size(); 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Setting the information in the structures residing within the buffer. 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // First, we set the information which doesn't require iterating through the 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bitmap set and then we set the bitmap specific structures. In the latter 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // step we also copy the actual bits. 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<uint8> buffer(buffer_size); 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ICONDIR* icon_dir = reinterpret_cast<ICONDIR*>(&buffer[0]); 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idType = kResourceTypeIcon; 489c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) icon_dir->idCount = static_cast<WORD>(image_count); 490c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // - 1 because there is already one ICONDIRENTRY in ICONDIR. 491c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) size_t icon_dir_count = image_count - 1; 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t offset = sizeof(ICONDIR) + (sizeof(ICONDIRENTRY) * icon_dir_count); 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < bitmap_count; i++) { 4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ICONIMAGE* image = reinterpret_cast<ICONIMAGE*>(&buffer[offset]); 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_LT(offset, buffer_size); 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t icon_image_size = 0; 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetSingleIconImageInformation(bitmaps[i], i, icon_dir, image, offset, 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &icon_image_size); 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(icon_image_size, 0U); 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) offset += icon_image_size; 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Add the PNG entry, if necessary. 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (png_bytes.get()) { 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ICONDIRENTRY* entry = &icon_dir->idEntries[bitmap_count]; 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->bWidth = 0; 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->bHeight = 0; 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->wPlanes = 1; 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->wBitCount = 32; 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->dwBytesInRes = static_cast<DWORD>(png_bytes->size()); 5122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) entry->dwImageOffset = static_cast<DWORD>(offset); 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memcpy(&buffer[offset], png_bytes->front(), png_bytes->size()); 5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offset += png_bytes->size(); 5152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(offset, buffer_size); 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 519a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) std::string data(buffer.begin(), buffer.end()); 520a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) return base::ImportantFileWriter::WriteFileAtomically(icon_path, data); 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool IconUtil::PixelsHaveAlpha(const uint32* pixels, size_t num_pixels) { 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (const uint32* end = pixels + num_pixels; pixels != end; ++pixels) { 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((*pixels & 0xff000000) != 0) 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void IconUtil::InitializeBitmapHeader(BITMAPV5HEADER* header, int width, 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int height) { 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(header); 5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memset(header, 0, sizeof(BITMAPV5HEADER)); 5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Size = sizeof(BITMAPV5HEADER); 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note that icons are created using top-down DIBs so we must negate the 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // value used for the icon's height. 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Width = width; 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Height = -height; 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Planes = 1; 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Compression = BI_RGB; 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initializing the bitmap format to 32 bit ARGB. 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5BitCount = 32; 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5RedMask = 0x00FF0000; 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5GreenMask = 0x0000FF00; 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5BlueMask = 0x000000FF; 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5AlphaMask = 0xFF000000; 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Use the system color space. The default value is LCS_CALIBRATED_RGB, which 5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // causes us to crash if we don't specify the approprite gammas, etc. See 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // <http://msdn.microsoft.com/en-us/library/ms536531(VS.85).aspx> and 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // <http://b/1283121>. 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5CSType = LCS_WINDOWS_COLOR_SPACE; 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Use a valid value for bV5Intent as 0 is not a valid one. 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // <http://msdn.microsoft.com/en-us/library/dd183381(VS.85).aspx> 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header->bV5Intent = LCS_GM_IMAGES; 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void IconUtil::SetSingleIconImageInformation(const SkBitmap& bitmap, 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t index, 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ICONDIR* icon_dir, 5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ICONIMAGE* icon_image, 5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t image_offset, 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* image_byte_count) { 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(icon_dir != NULL); 5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(icon_image != NULL); 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(image_offset, 0U); 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(image_byte_count != NULL); 573c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_LT(bitmap.width(), kLargeIconSize); 574c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_LT(bitmap.height(), kLargeIconSize); 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start by computing certain image values we'll use later on. 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t xor_mask_size, bytes_in_resource; 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ComputeBitmapSizeComponents(bitmap, 5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &xor_mask_size, 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &bytes_in_resource); 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].bWidth = static_cast<BYTE>(bitmap.width()); 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].bHeight = static_cast<BYTE>(bitmap.height()); 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].wPlanes = 1; 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].wBitCount = 32; 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].dwBytesInRes = bytes_in_resource; 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_dir->idEntries[index].dwImageOffset = image_offset; 5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_image->icHeader.biSize = sizeof(BITMAPINFOHEADER); 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The width field in the BITMAPINFOHEADER structure accounts for the height 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // of both the AND mask and the XOR mask so we need to multiply the bitmap's 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // height by 2. The same does NOT apply to the width field. 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_image->icHeader.biHeight = bitmap.height() * 2; 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_image->icHeader.biWidth = bitmap.width(); 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_image->icHeader.biPlanes = 1; 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) icon_image->icHeader.biBitCount = 32; 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We use a helper function for copying to actual bits from the SkBitmap 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // object into the appropriate space in the buffer. We use a helper function 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // (rather than just copying the bits) because there is no way to specify the 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // orientation (bottom-up vs. top-down) of a bitmap residing in a .ico file. 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Thus, if we just copy the bits, we'll end up with a bottom up bitmap in 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the .ico file which will result in the icon being displayed upside down. 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The helper function copies the image into the buffer one scanline at a 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // time. 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note that we don't need to initialize the AND mask since the memory 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // allocated for the icon data buffer was initialized to zero. The icon we 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // create will therefore use an AND mask containing only zeros, which is OK 6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // because the underlying image has an alpha channel. An AND mask containing 6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // only zeros essentially means we'll initially treat all the pixels as 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // opaque. 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned char* image_addr = reinterpret_cast<unsigned char*>(icon_image); 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned char* xor_mask_addr = image_addr + sizeof(BITMAPINFOHEADER); 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CopySkBitmapBitsIntoIconBuffer(bitmap, xor_mask_addr, xor_mask_size); 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *image_byte_count = bytes_in_resource; 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void IconUtil::CopySkBitmapBitsIntoIconBuffer(const SkBitmap& bitmap, 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned char* buffer, 6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t buffer_size) { 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkAutoLockPixels bitmap_lock(bitmap); 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned char* bitmap_ptr = static_cast<unsigned char*>(bitmap.getPixels()); 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bitmap_size = bitmap.height() * bitmap.width() * 4; 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(buffer_size, bitmap_size); 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < bitmap_size; i += bitmap.width() * 4) { 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memcpy(buffer + bitmap_size - bitmap.width() * 4 - i, 6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bitmap_ptr + i, 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bitmap.width() * 4); 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t IconUtil::ComputeIconFileBufferSize(const std::vector<SkBitmap>& set) { 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!set.empty()); 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We start by counting the bytes for the structures that don't depend on the 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // number of icon images. Note that sizeof(ICONDIR) already accounts for a 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // single ICONDIRENTRY structure, which is why we subtract one from the 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // number of bitmaps. 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t total_buffer_size = sizeof(ICONDIR); 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bitmap_count = set.size(); 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_buffer_size += sizeof(ICONDIRENTRY) * (bitmap_count - 1); 643c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // May not have all icon sizes, but must have at least up to medium icon size. 644c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK_GE(bitmap_count, kNumIconDimensionsUpToMediumSize); 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add the bitmap specific structure sizes. 6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < bitmap_count; i++) { 6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t xor_mask_size, bytes_in_resource; 6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ComputeBitmapSizeComponents(set[i], 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &xor_mask_size, 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &bytes_in_resource); 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_buffer_size += bytes_in_resource; 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return total_buffer_size; 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void IconUtil::ComputeBitmapSizeComponents(const SkBitmap& bitmap, 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* xor_mask_size, 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* bytes_in_resource) { 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The XOR mask size is easy to calculate since we only deal with 32bpp 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // images. 6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *xor_mask_size = bitmap.width() * bitmap.height() * 4; 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Computing the AND mask is a little trickier since it is a monochrome 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bitmap (regardless of the number of bits per pixels used in the XOR mask). 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // There are two things we must make sure we do when computing the AND mask 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // size: 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 1. Make sure the right number of bytes is allocated for each AND mask 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // scan line in case the number of pixels in the image is not divisible by 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 8. For example, in a 15X15 image, 15 / 8 is one byte short of 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // containing the number of bits we need in order to describe a single 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // image scan line so we need to add a byte. Thus, we need 2 bytes instead 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // of 1 for each scan line. 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 2. Make sure each scan line in the AND mask is 4 byte aligned (so that the 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // total icon image has a 4 byte alignment). In the 15X15 image example 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // above, we can not use 2 bytes so we increase it to the next multiple of 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 4 which is 4. 6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Once we compute the size for a singe AND mask scan line, we multiply that 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // number by the image height in order to get the total number of bytes for 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the AND mask. Thus, for a 15X15 image, we need 15 * 4 which is 60 bytes 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // for the monochrome bitmap representing the AND mask. 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t and_line_length = (bitmap.width() + 7) >> 3; 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) and_line_length = (and_line_length + 3) & ~3; 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t and_mask_size = and_line_length * bitmap.height(); 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t masks_size = *xor_mask_size + and_mask_size; 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *bytes_in_resource = masks_size + sizeof(BITMAPINFOHEADER); 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 691