ec98703cc8dd57d75a472b53e9715940d8f8458c
[WebKit-https.git] / WebCore / platform / graphics / cg / PDFDocumentImage.cpp
1 /*
2  * Copyright (C) 2004, 2005, 2006 Apple Computer, Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #define _USE_MATH_DEFINES 1
27 #include "config.h"
28 #include "PDFDocumentImage.h"
29
30 #if PLATFORM(CG)
31
32 #include "GraphicsContext.h"
33
34 using namespace std;
35
36 namespace WebCore {
37
38 PDFDocumentImage::PDFDocumentImage()
39     : Image(0) // PDFs don't animate
40     , m_document(0)
41     , m_rotation(0.0f)
42     , m_currentPage(-1)
43 {
44 }
45
46 PDFDocumentImage::~PDFDocumentImage()
47 {
48     CGPDFDocumentRelease(m_document);
49 }
50
51 IntSize PDFDocumentImage::size() const
52 {
53     return IntSize((int)m_mediaBox.size().width(), (int)m_mediaBox.size().height());
54 }
55
56 bool PDFDocumentImage::dataChanged(bool allDataReceived)
57 {
58     if (allDataReceived && !m_document) {
59 #if PLATFORM(MAC)
60         // On Mac the NSData inside the SharedBuffer can be secretly appended to without the SharedBuffer's knowledge.  We use SharedBuffer's ability
61         // to wrap itself in an NSData to get around this, ensuring that ImageIO is really looking at the SharedBuffer.
62         CFDataRef data = (CFDataRef)m_data->createNSData();
63 #else
64         // If no NSData is available, then we know SharedBuffer will always just be a vector.  That means no secret changes can occur to it behind the
65         // scenes.  We use CFDataCreateWithBytesNoCopy in that case.
66         CFDataRef data = CFDataCreateWithBytesNoCopy(0, reinterpret_cast<const UInt8*>(m_data->data()), m_data->size(), kCFAllocatorNull);
67 #endif
68         CGDataProviderRef dataProvider = CGDataProviderCreateWithCFData(data);
69         CFRelease(data);
70         m_document = CGPDFDocumentCreateWithProvider(dataProvider);
71         CGDataProviderRelease(dataProvider);
72         setCurrentPage(0);
73     }
74     return m_document; // return true if size is available
75 }
76
77 void PDFDocumentImage::adjustCTM(GraphicsContext* context) const
78 {
79     // rotate the crop box and calculate bounding box
80     float sina = sinf(-m_rotation);
81     float cosa = cosf(-m_rotation);
82     float width = m_cropBox.width();
83     float height = m_cropBox.height();
84
85     // calculate rotated x and y edges of the corp box. if they're negative, it means part of the image has
86     // been rotated outside of the bounds and we need to shift over the image so it lies inside the bounds again
87     CGPoint rx = CGPointMake(width * cosa, width * sina);
88     CGPoint ry = CGPointMake(-height * sina, height * cosa);
89
90     // adjust so we are at the crop box origin
91     const CGFloat zero = 0;
92     CGContextTranslateCTM(context->platformContext(), floorf(-min(zero, min(rx.x, ry.x))), floorf(-min(zero, min(rx.y, ry.y))));
93
94     // rotate -ve to remove rotation
95     CGContextRotateCTM(context->platformContext(), -m_rotation);
96
97     // shift so we are completely within media box
98     CGContextTranslateCTM(context->platformContext(), m_mediaBox.x() - m_cropBox.x(), m_mediaBox.y() - m_cropBox.y());
99 }
100
101 void PDFDocumentImage::setCurrentPage(int page)
102 {
103     if (!m_document)
104         return;
105
106     if (page == m_currentPage)
107         return;
108
109     if (!(page >= 0 && page < pageCount()))
110         return;
111
112     m_currentPage = page;
113
114     CGPDFPageRef cgPage = CGPDFDocumentGetPage(m_document, page + 1);
115
116     // get media box (guaranteed)
117     m_mediaBox = CGPDFPageGetBoxRect(cgPage, kCGPDFMediaBox);
118
119     // get crop box (not always there). if not, use media box
120     CGRect r = CGPDFPageGetBoxRect(cgPage, kCGPDFCropBox);
121     if (!CGRectIsEmpty(r))
122         m_cropBox = r;
123     else
124         m_cropBox = m_mediaBox;
125
126     // get page rotation angle
127     m_rotation = CGPDFPageGetRotationAngle(cgPage) * M_PI / 180.0; // to radians
128 }
129
130 int PDFDocumentImage::pageCount() const
131 {
132     return m_document ? CGPDFDocumentGetNumberOfPages(m_document) : 0;
133 }
134
135 void PDFDocumentImage::draw(GraphicsContext* context, const FloatRect& dstRect, const FloatRect& srcRect, CompositeOperator op)
136 {
137     if (!m_document || m_currentPage == -1)
138         return;
139
140     context->save();
141
142     context->setCompositeOperation(op);
143
144     float hScale = dstRect.width() / srcRect.width();
145     float vScale = dstRect.height() / srcRect.height();
146
147     // Scale and translate so the document is rendered in the correct location,
148     // including accounting for the fact that a GraphicsContext is always flipped
149     // and doing appropriate flipping.
150     CGContextTranslateCTM(context->platformContext(), dstRect.x() - srcRect.x() * hScale, dstRect.y() - srcRect.y() * vScale);
151     CGContextScaleCTM(context->platformContext(), hScale, vScale);
152     CGContextScaleCTM(context->platformContext(), 1, -1);
153     CGContextTranslateCTM(context->platformContext(), 0, -dstRect.height());
154     CGContextClipToRect(context->platformContext(), CGRectIntegral(srcRect));
155
156     // Rotate translate image into position according to doc properties.
157     adjustCTM(context);
158
159     // Media box may have non-zero origin which we ignore. Pass 1 for the page number.
160     CGContextDrawPDFDocument(context->platformContext(), FloatRect(FloatPoint(), size()),
161         m_document, m_currentPage + 1);
162
163     context->restore();
164 }
165
166 }
167
168 #endif // PLATFORM(CG)