WebCore/platform/graphics/cg/GraphicsContext3DCG.cpp

   1 /*
   2  * Copyright (C) 2010 Apple Inc. All rights reserved.
   3  * Copyright (C) 2010 Google Inc. All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #include "config.h"
  28
  29 #if ENABLE(3D_CANVAS)
  30
  31 #include "GraphicsContext3D.h"
  32
  33 #include "Image.h"
  34
  35 #include <CoreGraphics/CGBitmapContext.h>
  36 #include <CoreGraphics/CGContext.h>
  37 #include <CoreGraphics/CGImage.h>
  38
  39 namespace WebCore {
  40
  41 bool GraphicsContext3D::getImageData(Image* image,
  42                                      unsigned int format,
  43                                      unsigned int type,
  44                                      bool premultiplyAlpha,
  45                                      Vector<uint8_t>& outputVector)
  46 {
  47     if (!image)
  48         return false;
  49     CGImageRef cgImage = image->nativeImageForCurrentFrame();
  50     if (!cgImage)
  51         return false;
  52     int width = CGImageGetWidth(cgImage);
  53     int height = CGImageGetHeight(cgImage);
  54     // FIXME: we should get rid of this temporary copy where possible.
  55     int tempRowBytes = width * 4;
  56     Vector<uint8_t> tempVector;
  57     tempVector.resize(height * tempRowBytes);
  58     // Try to reuse the color space from the image to preserve its colors.
  59     // Some images use a color space (such as indexed) unsupported by the bitmap context.
  60     CGColorSpaceRef colorSpace = CGImageGetColorSpace(cgImage);
  61     bool releaseColorSpace = false;
  62     CGColorSpaceModel colorSpaceModel = CGColorSpaceGetModel(colorSpace);
  63     switch (colorSpaceModel) {
  64     case kCGColorSpaceModelMonochrome:
  65     case kCGColorSpaceModelRGB:
  66     case kCGColorSpaceModelCMYK:
  67     case kCGColorSpaceModelLab:
  68     case kCGColorSpaceModelDeviceN:
  69         break;
  70     default:
  71         colorSpace = CGColorSpaceCreateWithName(kCGColorSpaceGenericRGBLinear);
  72         releaseColorSpace = true;
  73         break;
  74     }
  75     CGContextRef tempContext = CGBitmapContextCreate(tempVector.data(),
  76                                                      width, height, 8, tempRowBytes,
  77                                                      colorSpace,
  78                                                      // FIXME: change this!
  79                                                      kCGImageAlphaPremultipliedLast);
  80     if (releaseColorSpace)
  81         CGColorSpaceRelease(colorSpace);
  82     if (!tempContext)
  83         return false;
  84     CGContextSetBlendMode(tempContext, kCGBlendModeCopy);
  85     CGContextDrawImage(tempContext,
  86                        CGRectMake(0, 0, static_cast<CGFloat>(width), static_cast<CGFloat>(height)),
  87                        cgImage);
  88     CGContextRelease(tempContext);
  89     // Pack the pixel data into the output vector.
  90     unsigned long componentsPerPixel, bytesPerComponent;
  91     if (!computeFormatAndTypeParameters(format, type, &componentsPerPixel, &bytesPerComponent))
  92         return false;
  93     int rowBytes = width * componentsPerPixel * bytesPerComponent;
  94     outputVector.resize(height * rowBytes);
  95     CGImageAlphaInfo info = CGImageGetAlphaInfo(cgImage);
  96     bool hasAlphaChannel = (info != kCGImageAlphaNone
  97                             && info != kCGImageAlphaNoneSkipLast
  98                             && info != kCGImageAlphaNoneSkipFirst);
  99     AlphaOp neededAlphaOp = kAlphaDoNothing;
 100     if (!premultiplyAlpha && hasAlphaChannel)
 101         // FIXME: must fetch the image data before the premultiplication step.
 102         neededAlphaOp = kAlphaDoUnmultiply;
 103     return packPixels(tempVector.data(), kSourceFormatRGBA8, width, height, 0,
 104                       format, type, neededAlphaOp, outputVector.data());
 105 }
 106
 107
 108 } // namespace WebCore
 109
 110 #endif // ENABLE(3D_CANVAS)