f456ec574d22c036cef2311c830ec2768ed6f26a
[WebKit-https.git] / WebCore / platform / graphics / cg / ImageBufferCG.cpp
1 /*
2  * Copyright (C) 2006 Nikolas Zimmermann <zimmermann@kde.org>
3  * Copyright (C) 2008 Apple, Inc
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 #include "ImageBuffer.h"
29
30 #include "GraphicsContext.h"
31 #include "ImageData.h"
32
33 #include <ApplicationServices/ApplicationServices.h>
34 #include <wtf/Assertions.h>
35
36 using namespace std;
37
38 namespace WebCore {
39
40 auto_ptr<ImageBuffer> ImageBuffer::create(const IntSize& size, bool grayScale)
41 {
42     if (size.width() < 0 || size.height() < 0)
43         return auto_ptr<ImageBuffer>();        
44     unsigned int bytesPerRow = size.width();
45     if (!grayScale) {
46         // Protect against overflow
47         if (bytesPerRow > 0x3FFFFFFF)
48             return auto_ptr<ImageBuffer>();            
49         bytesPerRow *= 4;
50     }
51
52     void* imageBuffer = fastCalloc(size.height(), bytesPerRow);
53     if (!imageBuffer)
54         return auto_ptr<ImageBuffer>();
55     
56     CGColorSpaceRef colorSpace = grayScale ? CGColorSpaceCreateDeviceGray() : CGColorSpaceCreateDeviceRGB();
57     CGContextRef cgContext = CGBitmapContextCreate(imageBuffer, size.width(), size.height(), 8, bytesPerRow,
58         colorSpace, grayScale ? kCGImageAlphaNone : kCGImageAlphaPremultipliedLast);
59     CGColorSpaceRelease(colorSpace);
60     if (!cgContext) {
61         fastFree(imageBuffer);
62         return auto_ptr<ImageBuffer>();
63     }
64
65     auto_ptr<GraphicsContext> context(new GraphicsContext(cgContext));
66     CGContextRelease(cgContext);
67     
68     return auto_ptr<ImageBuffer>(new ImageBuffer(imageBuffer, size, context));
69 }
70
71
72 ImageBuffer::ImageBuffer(void* imageData, const IntSize& size, auto_ptr<GraphicsContext> context)
73     : m_data(imageData)
74     , m_size(size)
75     , m_context(context.release())
76     , m_cgImage(0)
77 {
78 }
79
80 ImageBuffer::~ImageBuffer()
81 {
82     fastFree(m_data);
83     CGImageRelease(m_cgImage);
84 }
85
86 GraphicsContext* ImageBuffer::context() const
87 {
88     return m_context.get();
89 }
90
91 CGImageRef ImageBuffer::cgImage() const
92 {
93     // It's assumed that if cgImage() is called, the actual rendering to the
94     // contained GraphicsContext must be done, as we create the CGImageRef here.
95     if (!m_cgImage) {
96         ASSERT(context());
97         m_cgImage = CGBitmapContextCreateImage(context()->platformContext());
98     }
99
100     return m_cgImage;
101 }
102
103 PassRefPtr<ImageData> ImageBuffer::getImageData(const IntRect& rect) const
104 {
105     if (!m_data)
106         return 0;
107     
108     PassRefPtr<ImageData> result = ImageData::create(rect.width(), rect.height());
109     unsigned char* data = result->data()->data().data();
110     
111     if (rect.x() < 0 || rect.y() < 0 || (rect.x() + rect.width()) > m_size.width() || (rect.y() + rect.height()) > m_size.height())
112         memset(data, 0, result->data()->length());
113
114     int originx = rect.x();
115     int destx = 0;
116     if (originx < 0) {
117         destx = -originx;
118         originx = 0;
119     }
120     int endx = rect.x() + rect.width();
121     if (endx > m_size.width())
122         endx = m_size.width();
123     int numColumns = endx - originx;
124     
125     int originy = rect.y();
126     int desty = 0;
127     if (originy < 0) {
128         desty = -originy;
129         originy = 0;
130     }
131     int endy = rect.y() + rect.height();
132     if (endy > m_size.height())
133         endy = m_size.height();
134     int numRows = endy - originy;
135     
136     unsigned srcBytesPerRow = 4 * m_size.width();
137     unsigned destBytesPerRow = 4 * rect.width();
138     
139     // -originy to handle the accursed flipped y axis
140     unsigned char* srcRows = reinterpret_cast<unsigned char*>(m_data) + (m_size.height() - originy - 1) * srcBytesPerRow + originx * 4;
141     unsigned char* destRows = data + desty * destBytesPerRow + destx * 4;
142     for (int y = 0; y < numRows; ++y) {
143         for (int x = 0; x < numColumns; x++) {
144             if (unsigned char alpha = srcRows[3]) {
145                 destRows[0] = (srcRows[0] * 255) / alpha;
146                 destRows[1] = (srcRows[1] * 255) / alpha;
147                 destRows[2] = (srcRows[2] * 255) / alpha;
148                 destRows[3] = alpha;
149             } else {
150                 reinterpret_cast<uint32_t*>(destRows)[0] = reinterpret_cast<uint32_t*>(srcRows)[0];
151             }
152             destRows += 4;
153         }
154         srcRows -= srcBytesPerRow;
155     }
156     return result;
157 }
158
159 }