GraphicsContext3DCG.cpp [plain text]
#include "config.h"
#if ENABLE(3D_CANVAS)
#include "GraphicsContext3D.h"
#include "Image.h"
#include <CoreGraphics/CGBitmapContext.h>
#include <CoreGraphics/CGContext.h>
#include <CoreGraphics/CGImage.h>
namespace WebCore {
bool GraphicsContext3D::getImageData(Image* image,
Vector<uint8_t>& outputVector,
bool premultiplyAlpha,
bool* hasAlphaChannel,
AlphaOp* neededAlphaOp,
unsigned int* format)
{
if (!image)
return false;
CGImageRef cgImage = image->nativeImageForCurrentFrame();
if (!cgImage)
return false;
int width = CGImageGetWidth(cgImage);
int height = CGImageGetHeight(cgImage);
int rowBytes = width * 4;
CGImageAlphaInfo info = CGImageGetAlphaInfo(cgImage);
*hasAlphaChannel = (info != kCGImageAlphaNone
&& info != kCGImageAlphaNoneSkipLast
&& info != kCGImageAlphaNoneSkipFirst);
if (!premultiplyAlpha && *hasAlphaChannel)
*neededAlphaOp = kAlphaDoUnmultiply;
*format = RGBA;
outputVector.resize(height * rowBytes);
CGColorSpaceRef colorSpace = CGImageGetColorSpace(cgImage);
bool releaseColorSpace = false;
CGColorSpaceModel colorSpaceModel = CGColorSpaceGetModel(colorSpace);
switch (colorSpaceModel) {
case kCGColorSpaceModelMonochrome:
case kCGColorSpaceModelRGB:
case kCGColorSpaceModelCMYK:
case kCGColorSpaceModelLab:
case kCGColorSpaceModelDeviceN:
break;
default:
colorSpace = CGColorSpaceCreateWithName(kCGColorSpaceGenericRGBLinear);
releaseColorSpace = true;
break;
}
CGContextRef tmpContext = CGBitmapContextCreate(outputVector.data(),
width, height, 8, rowBytes,
colorSpace,
kCGImageAlphaPremultipliedLast);
if (releaseColorSpace)
CGColorSpaceRelease(colorSpace);
if (!tmpContext)
return false;
CGContextSetBlendMode(tmpContext, kCGBlendModeCopy);
CGContextDrawImage(tmpContext,
CGRectMake(0, 0, static_cast<CGFloat>(width), static_cast<CGFloat>(height)),
cgImage);
CGContextRelease(tmpContext);
return true;
}
}
#endif // ENABLE(3D_CANVAS)