OLD | NEW |
| (Empty) |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "printing/pdf_ps_metafile_linux.h" | |
6 | |
7 #include <stdio.h> | |
8 | |
9 #include <cairo.h> | |
10 #include <cairo-pdf.h> | |
11 #include <cairo-ps.h> | |
12 | |
13 #include "base/eintr_wrapper.h" | |
14 #include "base/file_descriptor_posix.h" | |
15 #include "base/file_util.h" | |
16 #include "base/logging.h" | |
17 #include "skia/ext/vector_platform_device_linux.h" | |
18 | |
19 namespace { | |
20 | |
21 // Tests if |surface| is valid. | |
22 bool IsSurfaceValid(cairo_surface_t* surface) { | |
23 return cairo_surface_status(surface) == CAIRO_STATUS_SUCCESS; | |
24 } | |
25 | |
26 // Tests if |context| is valid. | |
27 bool IsContextValid(cairo_t* context) { | |
28 return cairo_status(context) == CAIRO_STATUS_SUCCESS; | |
29 } | |
30 | |
31 // Destroys and resets |surface|. | |
32 void CleanUpSurface(cairo_surface_t** surface) { | |
33 if (*surface) { | |
34 cairo_surface_destroy(*surface); | |
35 *surface = NULL; | |
36 } | |
37 } | |
38 | |
39 // Destroys and resets |context|. | |
40 void CleanUpContext(cairo_t** context) { | |
41 if (*context) { | |
42 cairo_destroy(*context); | |
43 *context = NULL; | |
44 } | |
45 } | |
46 | |
47 // Callback function for Cairo to write PDF/PS stream. | |
48 // |dst_buffer| is actually a pointer of type `std::string*`. | |
49 cairo_status_t WriteCairoStream(void* dst_buffer, | |
50 const unsigned char* src_data, | |
51 unsigned int src_data_length) { | |
52 DCHECK(dst_buffer); | |
53 DCHECK(src_data); | |
54 DCHECK_GT(src_data_length, 0u); | |
55 | |
56 std::string* buffer = reinterpret_cast<std::string*>(dst_buffer); | |
57 buffer->append(reinterpret_cast<const char*>(src_data), src_data_length); | |
58 | |
59 return CAIRO_STATUS_SUCCESS; | |
60 } | |
61 | |
62 } // namespace | |
63 | |
64 namespace printing { | |
65 | |
66 PdfPsMetafile::PdfPsMetafile(const FileFormat& format) | |
67 : format_(format), | |
68 surface_(NULL), context_(NULL), | |
69 page_surface_(NULL), page_context_(NULL) { | |
70 } | |
71 | |
72 PdfPsMetafile::~PdfPsMetafile() { | |
73 // Releases all resources if we forgot to do so. | |
74 CleanUpAll(); | |
75 } | |
76 | |
77 bool PdfPsMetafile::Init() { | |
78 // We need to check at least these two members to ensure Init() has not been | |
79 // called before. Passing these two checks also implies that surface_, | |
80 // page_surface_, and page_context_ are NULL, and current_page_ is empty. | |
81 DCHECK(!context_); | |
82 DCHECK(all_pages_.empty()); | |
83 | |
84 // Creates an 1 by 1 Cairo surface for entire PDF/PS file. | |
85 // The size for each page will be overwritten later in StartPage(). | |
86 switch (format_) { | |
87 case PDF: { | |
88 surface_ = cairo_pdf_surface_create_for_stream(WriteCairoStream, | |
89 &all_pages_, 1, 1); | |
90 } | |
91 break; | |
92 | |
93 case PS: { | |
94 surface_ = cairo_ps_surface_create_for_stream(WriteCairoStream, | |
95 &all_pages_, 1, 1); | |
96 } | |
97 break; | |
98 | |
99 default: | |
100 NOTREACHED(); | |
101 return false; | |
102 } | |
103 | |
104 // Cairo always returns a valid pointer. | |
105 // Hence, we have to check if it points to a "nil" object. | |
106 if (!IsSurfaceValid(surface_)) { | |
107 DLOG(ERROR) << "Cannot create Cairo surface for PdfPsMetafile!"; | |
108 CleanUpSurface(&surface_); | |
109 return false; | |
110 } | |
111 | |
112 // Creates a context. | |
113 context_ = cairo_create(surface_); | |
114 if (!IsContextValid(context_)) { | |
115 DLOG(ERROR) << "Cannot create Cairo context for PdfPsMetafile!"; | |
116 CleanUpContext(&context_); | |
117 CleanUpSurface(&surface_); | |
118 return false; | |
119 } | |
120 | |
121 return true; | |
122 } | |
123 | |
124 bool PdfPsMetafile::Init(const void* src_buffer, size_t src_buffer_size) { | |
125 // We need to check at least these two members to ensure Init() has not been | |
126 // called before. Passing these two checks also implies that surface_, | |
127 // page_surface_, and page_context_ are NULL, and current_page_ is empty. | |
128 DCHECK(!context_); | |
129 DCHECK(all_pages_.empty()); | |
130 | |
131 if (src_buffer == NULL || src_buffer_size == 0) { | |
132 return false; | |
133 } | |
134 | |
135 all_pages_ = std::string(reinterpret_cast<const char*>(src_buffer), | |
136 src_buffer_size); | |
137 | |
138 return true; | |
139 } | |
140 | |
141 cairo_t* PdfPsMetafile::StartPage(double width_in_points, | |
142 double height_in_points) { | |
143 DCHECK(IsSurfaceValid(surface_)); | |
144 DCHECK(IsContextValid(context_)); | |
145 // Passing this check implies page_surface_ is NULL, and current_page_ is | |
146 // empty. | |
147 DCHECK(!page_context_); | |
148 DCHECK_GT(width_in_points, 0.); | |
149 DCHECK_GT(height_in_points, 0.); | |
150 | |
151 // Creates a target surface for the new page. | |
152 // Cairo 1.6.0 does NOT allow the first argument be NULL, | |
153 // but some newer versions do support NULL pointer. | |
154 switch (format_) { | |
155 case PDF: { | |
156 page_surface_ = cairo_pdf_surface_create_for_stream(WriteCairoStream, | |
157 ¤t_page_, | |
158 width_in_points, | |
159 height_in_points); | |
160 } | |
161 break; | |
162 | |
163 case PS: { | |
164 page_surface_ = cairo_ps_surface_create_for_stream(WriteCairoStream, | |
165 ¤t_page_, | |
166 width_in_points, | |
167 height_in_points); | |
168 } | |
169 break; | |
170 | |
171 default: | |
172 NOTREACHED(); | |
173 CleanUpAll(); | |
174 return NULL; | |
175 } | |
176 | |
177 // Cairo always returns a valid pointer. | |
178 // Hence, we have to check if it points to a "nil" object. | |
179 if (!IsSurfaceValid(page_surface_)) { | |
180 DLOG(ERROR) << "Cannot create Cairo surface for PdfPsMetafile!"; | |
181 CleanUpAll(); | |
182 return NULL; | |
183 } | |
184 | |
185 // Creates a context. | |
186 page_context_ = cairo_create(page_surface_); | |
187 if (!IsContextValid(page_context_)) { | |
188 DLOG(ERROR) << "Cannot create Cairo context for PdfPsMetafile!"; | |
189 CleanUpAll(); | |
190 return NULL; | |
191 } | |
192 | |
193 return page_context_; | |
194 } | |
195 | |
196 bool PdfPsMetafile::FinishPage(float shrink) { | |
197 DCHECK(IsSurfaceValid(surface_)); | |
198 DCHECK(IsContextValid(context_)); | |
199 DCHECK(IsSurfaceValid(page_surface_)); | |
200 DCHECK(IsContextValid(page_context_)); | |
201 DCHECK_GT(shrink, 0); | |
202 | |
203 // Flushes all rendering for current page. | |
204 cairo_surface_flush(page_surface_); | |
205 | |
206 // TODO(myhuang): Use real page settings. | |
207 // We hard-coded page settings here for testing purpose. | |
208 // The paper size is US Letter (8.5 in. by 11 in.). | |
209 // The default margins are: | |
210 // Left = 0.25 in. | |
211 // Right = 0.25 in. | |
212 // Top = 0.25 in. | |
213 // Bottom = 0.56 in. | |
214 const double kDPI = 72.0; // Dots (points) per inch. | |
215 const double kWidthInInch = 8.5; | |
216 const double kHeightInInch = 11.0; | |
217 const double kWidthInPoint = kWidthInInch * kDPI; | |
218 const double kHeightInPoint = kHeightInInch * kDPI; | |
219 switch (format_) { | |
220 case PDF: { | |
221 cairo_pdf_surface_set_size(surface_, kWidthInPoint, kHeightInPoint); | |
222 } | |
223 break; | |
224 | |
225 case PS: { | |
226 cairo_ps_surface_set_size(surface_, kWidthInPoint, kHeightInPoint); | |
227 } | |
228 break; | |
229 | |
230 default: | |
231 NOTREACHED(); | |
232 CleanUpAll(); | |
233 return false; | |
234 } | |
235 | |
236 // Checks if our surface is still valid after resizing. | |
237 if (!IsSurfaceValid(surface_)) { | |
238 DLOG(ERROR) << "Cannot resize Cairo surface for PdfPsMetafile!"; | |
239 CleanUpAll(); | |
240 return false; | |
241 } | |
242 | |
243 // Saves context's states. | |
244 cairo_save(context_); | |
245 // Copies current page onto the surface of final result. | |
246 // Margins are done by coordinates transformation. | |
247 // Please NOTE that we have to call cairo_scale() before we call | |
248 // cairo_set_source_surface(). | |
249 const double scale_factor = 1. / shrink; | |
250 cairo_scale(context_, scale_factor, scale_factor); | |
251 const double kLeftMarginInInch = 0.25; | |
252 const double kTopMarginInInch = 0.25; | |
253 const double kLeftMarginInPoint = kLeftMarginInInch * kDPI; | |
254 const double kTopMarginInPoint = kTopMarginInInch * kDPI; | |
255 const double kScaledLeftMarginInPoint = kLeftMarginInPoint * shrink; | |
256 const double kScaledTopMarginInPoint = kTopMarginInPoint * shrink; | |
257 cairo_set_source_surface(context_, | |
258 page_surface_, | |
259 kScaledLeftMarginInPoint, | |
260 kScaledTopMarginInPoint); | |
261 // In Cairo 1.6.0, if we use the following API, either the renderer will | |
262 // crash, or we will get an empty page. This might be a bug in Cairo. | |
263 // cairo_set_operator(context_, CAIRO_OPERATOR_SOURCE); | |
264 const double kRightMarginInInch = 0.25; | |
265 const double kBottomMarginInInch = 0.56; | |
266 const double kPrintableWidthInInch = | |
267 kWidthInInch - kLeftMarginInInch - kRightMarginInInch; | |
268 const double kPrintableHeightInInch = | |
269 kHeightInInch - kTopMarginInInch - kBottomMarginInInch; | |
270 const double kScaledPrintableWidthInPoint = | |
271 kPrintableWidthInInch * kDPI * shrink; | |
272 const double kScaledPrintableHeightInPoint = | |
273 kPrintableHeightInInch * kDPI * shrink; | |
274 cairo_rectangle(context_, | |
275 kScaledLeftMarginInPoint, | |
276 kScaledTopMarginInPoint, | |
277 kScaledPrintableWidthInPoint, | |
278 kScaledPrintableHeightInPoint); | |
279 cairo_fill(context_); | |
280 | |
281 // Finishes the duplication of current page. | |
282 cairo_show_page(context_); | |
283 cairo_surface_flush(surface_); | |
284 | |
285 // Destroys resources for current page. | |
286 CleanUpContext(&page_context_); | |
287 CleanUpSurface(&page_surface_); | |
288 current_page_.clear(); | |
289 | |
290 // Restores context's states. | |
291 cairo_restore(context_); | |
292 | |
293 return true; | |
294 } | |
295 | |
296 void PdfPsMetafile::Close() { | |
297 DCHECK(IsSurfaceValid(surface_)); | |
298 DCHECK(IsContextValid(context_)); | |
299 // Passing this check implies page_surface_ is NULL, and current_page_ is | |
300 // empty. | |
301 DCHECK(!page_context_); | |
302 | |
303 cairo_surface_finish(surface_); | |
304 DCHECK(!all_pages_.empty()); // Make sure we did get something. | |
305 | |
306 CleanUpContext(&context_); | |
307 CleanUpSurface(&surface_); | |
308 } | |
309 | |
310 unsigned int PdfPsMetafile::GetDataSize() const { | |
311 // We need to check at least these two members to ensure that either Init() | |
312 // has been called to initialize |all_pages_|, or metafile has been closed. | |
313 // Passing these two checks also implies that surface_, page_surface_, and | |
314 // page_context_ are NULL, and current_page_ is empty. | |
315 DCHECK(!context_); | |
316 DCHECK(!all_pages_.empty()); | |
317 | |
318 return all_pages_.size(); | |
319 } | |
320 | |
321 bool PdfPsMetafile::GetData(void* dst_buffer, size_t dst_buffer_size) const { | |
322 DCHECK(dst_buffer); | |
323 DCHECK_GT(dst_buffer_size, 0u); | |
324 // We need to check at least these two members to ensure that either Init() | |
325 // has been called to initialize |all_pages_|, or metafile has been closed. | |
326 // Passing these two checks also implies that surface_, page_surface_, and | |
327 // page_context_ are NULL, and current_page_ is empty. | |
328 DCHECK(!context_); | |
329 DCHECK(!all_pages_.empty()); | |
330 | |
331 size_t data_size = GetDataSize(); | |
332 if (dst_buffer_size > data_size) { | |
333 return false; | |
334 } | |
335 | |
336 memcpy(dst_buffer, all_pages_.data(), dst_buffer_size); | |
337 | |
338 return true; | |
339 } | |
340 | |
341 bool PdfPsMetafile::SaveTo(const base::FileDescriptor& fd) const { | |
342 // We need to check at least these two members to ensure that either Init() | |
343 // has been called to initialize |all_pages_|, or metafile has been closed. | |
344 // Passing these two checks also implies that surface_, page_surface_, and | |
345 // page_context_ are NULL, and current_page_ is empty. | |
346 DCHECK(!context_); | |
347 DCHECK(!all_pages_.empty()); | |
348 | |
349 if (fd.fd < 0) { | |
350 DLOG(ERROR) << "Invalid file descriptor!"; | |
351 return false; | |
352 } | |
353 | |
354 bool success = true; | |
355 if (file_util::WriteFileDescriptor(fd.fd, all_pages_.data(), | |
356 GetDataSize()) < 0) { | |
357 DLOG(ERROR) << "Failed to save file with fd " << fd.fd; | |
358 success = false; | |
359 } | |
360 | |
361 if (fd.auto_close) | |
362 HANDLE_EINTR(close(fd.fd)); | |
363 return success; | |
364 } | |
365 | |
366 void PdfPsMetafile::CleanUpAll() { | |
367 CleanUpContext(&context_); | |
368 CleanUpSurface(&surface_); | |
369 CleanUpContext(&page_context_); | |
370 CleanUpSurface(&page_surface_); | |
371 current_page_.clear(); | |
372 all_pages_.clear(); | |
373 skia::VectorPlatformDevice::ClearFontCache(); | |
374 } | |
375 | |
376 } // namespace printing | |
OLD | NEW |