2007-01-17 Eric Seidel <eric@webkit.org>
[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::setNativeData(NativeBytePtr data, bool allDataReceived)
57 {
58     if (allDataReceived && !m_document && data) {
59         CGDataProviderRef dataProvider = CGDataProviderCreateWithCFData(data);
60         m_document = CGPDFDocumentCreateWithProvider(dataProvider);
61         CGDataProviderRelease(dataProvider);
62         setCurrentPage(0);
63     }
64     return m_document; // return true if size is available
65 }
66
67 void PDFDocumentImage::adjustCTM(GraphicsContext* context) const
68 {
69     // rotate the crop box and calculate bounding box
70     float sina = sinf(-m_rotation);
71     float cosa = cosf(-m_rotation);
72     float width = m_cropBox.width();
73     float height = m_cropBox.height();
74
75     // calculate rotated x and y edges of the corp box. if they're negative, it means part of the image has
76     // been rotated outside of the bounds and we need to shift over the image so it lies inside the bounds again
77     CGPoint rx = CGPointMake(width * cosa, width * sina);
78     CGPoint ry = CGPointMake(-height * sina, height * cosa);
79
80     // adjust so we are at the crop box origin
81     const CGFloat zero = 0;
82     CGContextTranslateCTM(context->platformContext(), floorf(-min(zero, min(rx.x, ry.x))), floorf(-min(zero, min(rx.y, ry.y))));
83
84     // rotate -ve to remove rotation
85     CGContextRotateCTM(context->platformContext(), -m_rotation);
86
87     // shift so we are completely within media box
88     CGContextTranslateCTM(context->platformContext(), m_mediaBox.x() - m_cropBox.x(), m_mediaBox.y() - m_cropBox.y());
89 }
90
91 void PDFDocumentImage::setCurrentPage(int page)
92 {
93     if (!m_document)
94         return;
95
96     if (page == m_currentPage)
97         return;
98
99     if (!(page >= 0 && page < pageCount()))
100         return;
101
102     m_currentPage = page;
103
104     CGPDFPageRef cgPage = CGPDFDocumentGetPage(m_document, page + 1);
105
106     // get media box (guaranteed)
107     m_mediaBox = CGPDFPageGetBoxRect(cgPage, kCGPDFMediaBox);
108
109     // get crop box (not always there). if not, use media box
110     CGRect r = CGPDFPageGetBoxRect(cgPage, kCGPDFCropBox);
111     if (!CGRectIsEmpty(r))
112         m_cropBox = r;
113     else
114         m_cropBox = m_mediaBox;
115
116     // get page rotation angle
117     m_rotation = CGPDFPageGetRotationAngle(cgPage) * M_PI / 180.0; // to radians
118 }
119
120 int PDFDocumentImage::pageCount() const
121 {
122     return m_document ? CGPDFDocumentGetNumberOfPages(m_document) : 0;
123 }
124
125 void PDFDocumentImage::draw(GraphicsContext* context, const FloatRect& dstRect, const FloatRect& srcRect, CompositeOperator op)
126 {
127     if (!m_document || m_currentPage == -1)
128         return;
129
130     context->save();
131
132     context->setCompositeOperation(op);
133
134     float hScale = dstRect.width() / srcRect.width();
135     float vScale = dstRect.height() / srcRect.height();
136
137     // Scale and translate so the document is rendered in the correct location,
138     // including accounting for the fact that a GraphicsContext is always flipped
139     // and doing appropriate flipping.
140     CGContextTranslateCTM(context->platformContext(), dstRect.x() - srcRect.x() * hScale, dstRect.y() - srcRect.y() * vScale);
141     CGContextScaleCTM(context->platformContext(), hScale, vScale);
142     CGContextScaleCTM(context->platformContext(), 1, -1);
143     CGContextTranslateCTM(context->platformContext(), 0, -dstRect.height());
144     CGContextClipToRect(context->platformContext(), CGRectIntegral(srcRect));
145
146     // Rotate translate image into position according to doc properties.
147     adjustCTM(context);
148
149     // Media box may have non-zero origin which we ignore. Pass 1 for the page number.
150     CGContextDrawPDFDocument(context->platformContext(), FloatRect(FloatPoint(), size()),
151         m_document, m_currentPage + 1);
152
153     context->restore();
154 }
155
156 }
157
158 #endif // PLATFORM(CG)