Properly use data when creating a bitmap platform canvas

skia/ext/bitmap_platform_device_mac.cc

 // Copyright (c) 2006-2008 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 "skia/ext/bitmap_platform_device_mac.h"
 
 #include <time.h>
 
 #include "base/ref_counted.h"
 #include "skia/ext/skia_utils_mac.h"
@@ skipping 26 matching lines @@
     if (overflow > 0) {
       *size -= overflow;
     }
   } else {
     // Fill up available size.
     *size = available_size - *position;
   }
   return true;
 }
 
+static CGContextRef CGContextForData(void* data, int width, int height) {
+  CGColorSpaceRef color_space =
+      CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB);
+  // Allocate a bitmap context with 4 components per pixel (BGRA).  Apple
+  // recommends these flags for improved CG performance.
+  CGContextRef context =
+      CGBitmapContextCreate(data, width, height, 8, width * 4,
+                            color_space,
+                            kCGImageAlphaPremultipliedFirst |
+                                kCGBitmapByteOrder32Host);
+  CGColorSpaceRelease(color_space);
+
+  if (!context)
+    return NULL;
+
+  // Change the coordinate system to match WebCore's
+  CGContextTranslateCTM(context, 0, height);
+  CGContextScaleCTM(context, 1.0, -1.0);
+
+  return context;
+}
+
 }  // namespace
 
 class BitmapPlatformDevice::BitmapPlatformDeviceData : public SkRefCnt {
  public:
   explicit BitmapPlatformDeviceData(CGContextRef bitmap);
 
   // Create/destroy CoreGraphics context for our bitmap data.
   CGContextRef GetBitmapContext() {
     LoadConfig();
     return bitmap_context_;
@@ skipping 89 matching lines @@
   t.setTranslateX(-t.getTranslateX());
   t.setTranslateY(-t.getTranslateY());
   LoadClippingRegionToCGContext(bitmap_context_, clip_region_, t);
 }
 
 
 // We use this static factory function instead of the regular constructor so
 // that we can create the pixel data before calling the constructor. This is
 // required so that we can call the base class' constructor with the pixel
 // data.
-BitmapPlatformDevice* BitmapPlatformDevice::Create(CGContextRef context,
-                                                   int width,
-                                                   int height,
-                                                   bool is_opaque) {
+BitmapPlatformDevice* BitmapPlatformDevice::CreateWithContext(
+    CGContextRef context, int width, int height, bool is_opaque) {
   SkBitmap bitmap;
   bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height);
   if (bitmap.allocPixels() != true)
     return NULL;
-  void* data = bitmap.getPixels();
+
+  void* data = NULL;
+  if (context) {
+    data = CGBitmapContextGetData(context);
+    bitmap.setPixels(data);
+  } else {
+    data = bitmap.getPixels();
+  }
 
   // Note: The Windows implementation clears the Bitmap later on.
   // This bears mentioning since removal of this line makes the
   // unit tests only fail periodically (or when MallocPreScribble is set).
   bitmap.eraseARGB(0, 0, 0, 0);
 
   bitmap.setIsOpaque(is_opaque);
 
+#ifndef NDEBUG
   if (is_opaque) {
-#ifndef NDEBUG
     // To aid in finding bugs, we set the background color to something
     // obviously wrong so it will be noticable when it is not cleared
     bitmap.eraseARGB(255, 0, 255, 128);  // bright bluish green
+  }
 #endif
-  }
 
-  if (!context) {
-    CGColorSpaceRef color_space =
-      CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB);
-    // allocate a bitmap context with 4 components per pixel (BGRA). Apple
-    // recommends these flags for improved CG performance.
-    context =
-      CGBitmapContextCreate(data, width, height, 8, width*4,
-                            color_space,
-                            kCGImageAlphaPremultipliedFirst |
-                                kCGBitmapByteOrder32Host);
-
-    // Change the coordinate system to match WebCore's
-    CGContextTranslateCTM(context, 0, height);
-    CGContextScaleCTM(context, 1.0, -1.0);
-    CGColorSpaceRelease(color_space);
-  }
+  if (!context)
+    context = CGContextForData(data, width, height);
 
   // The device object will take ownership of the graphics context.
   return new BitmapPlatformDevice(
       new BitmapPlatformDeviceData(context), bitmap);
 }
 
+BitmapPlatformDevice* BitmapPlatformDevice::CreateWithData(uint8_t* data,
+                                                           int width,
+                                                           int height,
+                                                           bool is_opaque) {
+  CGContextRef context = NULL;
+  if (data)
+    context = CGContextForData(data, width, height);
+
+  return CreateWithContext(context, width, height, is_opaque);
+}
+
 // The device will own the bitmap, which corresponds to also owning the pixel
 // data. Therefore, we do not transfer ownership to the SkDevice's bitmap.
 BitmapPlatformDevice::BitmapPlatformDevice(
     BitmapPlatformDeviceData* data, const SkBitmap& bitmap)
     : PlatformDevice(bitmap),
       data_(data) {
 }
 
 // The copy constructor just adds another reference to the underlying data.
 // We use a const cast since the default Skia definitions don't define the
@@ skipping 86 matching lines @@
     for (int i = 0; i < height; i++) {
       size_t offset = (i + bitmap_start_y) * row_words + bitmap_start_x;
       for (int j = 0; j < width; j++) {
         adjustor(data + offset + j);
       }
     }
   }
 }
 
 }  // namespace skia