[go: nahoru, domu]
Google Git
Sign in
chromium / chromium / src / 4163cc35e206688f75d30a5c6a9a77a93e6d0600 / . / printing / pdf_metafile_skia.cc
blob: c63e308cc54abbfb2b526a02c09aef63fe99beb3 [file] [log] [blame]
epoger@google.com0a4392a2012-03-23 17:50:19[diff] [blame]1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]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_metafile_skia.h"
6
brettw@chromium.org14c1c232013-06-11 17:52:44[diff] [blame]7#include "base/containers/hash_tables.h"
thestig22dfc4012014-09-05 08:29:44[diff] [blame]8#include "base/files/file_util.h"
vandebo@chromium.org67e16b392011-05-30 20:58:09[diff] [blame]9#include "base/metrics/histogram.h"
jschuh@chromium.orgcb154062014-01-17 03:32:40[diff] [blame]10#include "base/numerics/safe_conversions.h"
brettw@chromium.org2025d002012-11-14 20:54:35[diff] [blame]11#include "base/posix/eintr_wrapper.h"
enne@chromium.org584abe72013-05-18 09:40:26[diff] [blame]12#include "skia/ext/refptr.h"
halcanary5be808e2014-11-10 22:20:05[diff] [blame]13#include "skia/ext/vector_canvas.h"
reed@google.come195fbf52011-06-27 16:51:20[diff] [blame]14#include "third_party/skia/include/core/SkData.h"
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]15#include "third_party/skia/include/core/SkRefCnt.h"
vandebo@chromium.orga34f0122011-04-12 17:36:39[diff] [blame]16#include "third_party/skia/include/core/SkScalar.h"
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]17#include "third_party/skia/include/core/SkStream.h"
vandebo@chromium.org67e16b392011-05-30 20:58:09[diff] [blame]18#include "third_party/skia/include/core/SkTypeface.h"
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]19#include "third_party/skia/include/pdf/SkPDFDevice.h"
20#include "third_party/skia/include/pdf/SkPDFDocument.h"
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]21#include "ui/gfx/point.h"
22#include "ui/gfx/rect.h"
23#include "ui/gfx/size.h"
vandebo@chromium.org67e16b392011-05-30 20:58:09[diff] [blame]24
caryclark@chromium.orgb8d85bc2011-06-22 13:34:57[diff] [blame]25#if defined(OS_MACOSX)
26#include "printing/pdf_metafile_cg_mac.h"
27#endif
28
rvargas@chromium.org0daaebfe2014-03-15 00:09:05[diff] [blame]29#if defined(OS_POSIX)
30#include "base/file_descriptor_posix.h"
31#endif
32
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]33namespace printing {
34
35struct PdfMetafileSkiaData {
enne@chromium.org584abe72013-05-18 09:40:26[diff] [blame]36 skia::RefPtr<SkPDFDevice> current_page_;
halcanary5be808e2014-11-10 22:20:05[diff] [blame]37 skia::RefPtr<SkCanvas> current_page_canvas_;
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]38 SkPDFDocument pdf_doc_;
39 SkDynamicMemoryWStream pdf_stream_;
caryclark@chromium.orgb8d85bc2011-06-22 13:34:57[diff] [blame]40#if defined(OS_MACOSX)
41 PdfMetafileCg pdf_cg_;
42#endif
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]43};
44
45PdfMetafileSkia::~PdfMetafileSkia() {}
46
47bool PdfMetafileSkia::Init() {
48 return true;
49}
50bool PdfMetafileSkia::InitFromData(const void* src_buffer,
51 uint32 src_buffer_size) {
52 return data_->pdf_stream_.write(src_buffer, src_buffer_size);
53}
54
halcanary5be808e2014-11-10 22:20:05[diff] [blame]55bool PdfMetafileSkia::StartPage(const gfx::Size& page_size,
56 const gfx::Rect& content_area,
57 const float& scale_factor) {
58 DCHECK(!data_->current_page_canvas_);
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]59
vandebo@chromium.orga34f0122011-04-12 17:36:39[diff] [blame]60 // Adjust for the margins and apply the scale factor.
61 SkMatrix transform;
ctguil@chromium.org39892b92011-04-30 02:24:44[diff] [blame]62 transform.setTranslate(SkIntToScalar(content_area.x()),
63 SkIntToScalar(content_area.y()));
vandebo@chromium.orga34f0122011-04-12 17:36:39[diff] [blame]64 transform.preScale(SkFloatToScalar(scale_factor),
65 SkFloatToScalar(scale_factor));
66
ctguil@chromium.org39892b92011-04-30 02:24:44[diff] [blame]67 SkISize pdf_page_size = SkISize::Make(page_size.width(), page_size.height());
68 SkISize pdf_content_size =
69 SkISize::Make(content_area.width(), content_area.height());
halcanary5be808e2014-11-10 22:20:05[diff] [blame]70
71 data_->current_page_ = skia::AdoptRef(
72 new SkPDFDevice(pdf_page_size, pdf_content_size, transform));
73 data_->current_page_canvas_ =
74 skia::AdoptRef(new SkCanvas(data_->current_page_.get()));
75 return true;
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]76}
77
halcanary5be808e2014-11-10 22:20:05[diff] [blame]78skia::VectorCanvas* PdfMetafileSkia::GetVectorCanvasForNewPage(
79 const gfx::Size& page_size,
80 const gfx::Rect& content_area,
81 const float& scale_factor) {
82 if (!StartPage(page_size, content_area, scale_factor))
83 return nullptr;
84 return data_->current_page_canvas_.get();
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]85}
86
87bool PdfMetafileSkia::FinishPage() {
halcanary5be808e2014-11-10 22:20:05[diff] [blame]88 DCHECK(data_->current_page_canvas_);
89 DCHECK(data_->current_page_);
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]90
halcanary5be808e2014-11-10 22:20:05[diff] [blame]91 data_->current_page_canvas_.clear(); // Unref SkCanvas.
kmadhusu@chromium.org534c4fb2011-08-02 16:44:20[diff] [blame]92 data_->pdf_doc_.appendPage(data_->current_page_.get());
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]93 return true;
94}
95
96bool PdfMetafileSkia::FinishDocument() {
97 // Don't do anything if we've already set the data in InitFromData.
98 if (data_->pdf_stream_.getOffset())
99 return true;
100
halcanary5be808e2014-11-10 22:20:05[diff] [blame]101 if (data_->current_page_canvas_)
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]102 FinishPage();
vandebo@chromium.org67e16b392011-05-30 20:58:09[diff] [blame]103
enne@chromium.org584abe72013-05-18 09:40:26[diff] [blame]104 data_->current_page_.clear();
vandebo@chromium.org54ffd6f62011-06-02 17:29:05[diff] [blame]105
vandebo@chromium.org3cd37bc02013-12-09 20:52:11[diff] [blame]106 int font_counts[SkAdvancedTypefaceMetrics::kOther_Font + 2];
epoger@google.com0a4392a2012-03-23 17:50:19[diff] [blame]107 data_->pdf_doc_.getCountOfFontTypes(font_counts);
108 for (int type = 0;
vandebo@chromium.org3cd37bc02013-12-09 20:52:11[diff] [blame]109 type <= SkAdvancedTypefaceMetrics::kOther_Font + 1;
epoger@google.com0a4392a2012-03-23 17:50:19[diff] [blame]110 type++) {
111 for (int count = 0; count < font_counts[type]; count++) {
112 UMA_HISTOGRAM_ENUMERATION(
113 "PrintPreview.FontType", type,
vandebo@chromium.org3cd37bc02013-12-09 20:52:11[diff] [blame]114 SkAdvancedTypefaceMetrics::kOther_Font + 2);
vandebo@chromium.org54ffd6f62011-06-02 17:29:05[diff] [blame]115 }
vandebo@chromium.org67e16b392011-05-30 20:58:09[diff] [blame]116 }
117
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]118 return data_->pdf_doc_.emitPDF(&data_->pdf_stream_);
119}
120
121uint32 PdfMetafileSkia::GetDataSize() const {
jschuh@chromium.orgcb154062014-01-17 03:32:40[diff] [blame]122 return base::checked_cast<uint32>(data_->pdf_stream_.getOffset());
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]123}
124
125bool PdfMetafileSkia::GetData(void* dst_buffer,
126 uint32 dst_buffer_size) const {
127 if (dst_buffer_size < GetDataSize())
128 return false;
129
reed@google.come195fbf52011-06-27 16:51:20[diff] [blame]130 SkAutoDataUnref data(data_->pdf_stream_.copyToData());
reed@google.com24b86abf2012-07-12 12:32:32[diff] [blame]131 memcpy(dst_buffer, data->bytes(), dst_buffer_size);
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]132 return true;
133}
134
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]135gfx::Rect PdfMetafileSkia::GetPageBounds(unsigned int page_number) const {
136 // TODO(vandebo) add a method to get the page size for a given page to
137 // SkPDFDocument.
138 NOTIMPLEMENTED();
139 return gfx::Rect();
140}
141
142unsigned int PdfMetafileSkia::GetPageCount() const {
143 // TODO(vandebo) add a method to get the number of pages to SkPDFDocument.
144 NOTIMPLEMENTED();
145 return 0;
146}
147
148gfx::NativeDrawingContext PdfMetafileSkia::context() const {
149 NOTREACHED();
150 return NULL;
151}
152
153#if defined(OS_WIN)
154bool PdfMetafileSkia::Playback(gfx::NativeDrawingContext hdc,
155 const RECT* rect) const {
156 NOTREACHED();
157 return false;
158}
159
160bool PdfMetafileSkia::SafePlayback(gfx::NativeDrawingContext hdc) const {
161 NOTREACHED();
162 return false;
163}
164
caryclark@chromium.orgb8d85bc2011-06-22 13:34:57[diff] [blame]165#elif defined(OS_MACOSX)
166/* TODO(caryclark): The set up of PluginInstance::PrintPDFOutput may result in
167 rasterized output. Even if that flow uses PdfMetafileCg::RenderPage,
168 the drawing of the PDF into the canvas may result in a rasterized output.
169 PDFMetafileSkia::RenderPage should be not implemented as shown and instead
170 should do something like the following CL in PluginInstance::PrintPDFOutput:
171http://codereview.chromium.org/7200040/diff/1/webkit/plugins/ppapi/ppapi_plugin_instance.cc
172*/
173bool PdfMetafileSkia::RenderPage(unsigned int page_number,
174 CGContextRef context,
175 const CGRect rect,
thestig@chromium.orgb5cf844c2012-06-18 21:49:20[diff] [blame]176 const MacRenderPageParams& params) const {
caryclark@chromium.orgb8d85bc2011-06-22 13:34:57[diff] [blame]177 DCHECK_GT(data_->pdf_stream_.getOffset(), 0U);
reed@google.come195fbf52011-06-27 16:51:20[diff] [blame]178 if (data_->pdf_cg_.GetDataSize() == 0) {
179 SkAutoDataUnref data(data_->pdf_stream_.copyToData());
reed@google.com24b86abf2012-07-12 12:32:32[diff] [blame]180 data_->pdf_cg_.InitFromData(data->bytes(), data->size());
reed@google.come195fbf52011-06-27 16:51:20[diff] [blame]181 }
thestig@chromium.orgb5cf844c2012-06-18 21:49:20[diff] [blame]182 return data_->pdf_cg_.RenderPage(page_number, context, rect, params);
caryclark@chromium.orgb8d85bc2011-06-22 13:34:57[diff] [blame]183}
184#endif
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]185
halcanary5be808e2014-11-10 22:20:05[diff] [blame]186bool PdfMetafileSkia::SaveTo(base::File* file) const {
187 if (GetDataSize() == 0U)
188 return false;
189 SkAutoDataUnref data(data_->pdf_stream_.copyToData());
190 // TODO(halcanary): rewrite this function without extra data copy
191 // using SkStreamAsset.
192 const char* ptr = reinterpret_cast<const char*>(data->data());
193 int size = base::checked_cast<int>(data->size());
194 return file->WriteAtCurrentPos(ptr, size) == size;
195}
196
avayvod@chromium.orgb25f0032013-08-19 22:26:25[diff] [blame]197#if defined(OS_CHROMEOS) || defined(OS_ANDROID)
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]198bool PdfMetafileSkia::SaveToFD(const base::FileDescriptor& fd) const {
199 DCHECK_GT(data_->pdf_stream_.getOffset(), 0U);
200
201 if (fd.fd < 0) {
202 DLOG(ERROR) << "Invalid file descriptor!";
203 return false;
204 }
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]205 base::File file(fd.fd);
halcanary5be808e2014-11-10 22:20:05[diff] [blame]206 bool result = SaveTo(&file);
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]207 DLOG_IF(ERROR, !result) << "Failed to save file with fd " << fd.fd;
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]208
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]209 if (!fd.auto_close)
210 file.TakePlatformFile();
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]211 return result;
212}
213#endif
214
halcanary5be808e2014-11-10 22:20:05[diff] [blame]215PdfMetafileSkia::PdfMetafileSkia() : data_(new PdfMetafileSkiaData) {
vandebo@chromium.org19b9d3b2011-07-23 02:08:57[diff] [blame]216}
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]217
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]218scoped_ptr<PdfMetafileSkia> PdfMetafileSkia::GetMetafileForCurrentPage() {
219 scoped_ptr<PdfMetafileSkia> metafile;
vandebo@chromium.org19b9d3b2011-07-23 02:08:57[diff] [blame]220 SkPDFDocument pdf_doc(SkPDFDocument::kDraftMode_Flags);
reed@google.comb1e544f2011-07-21 14:52:25[diff] [blame]221 if (!pdf_doc.appendPage(data_->current_page_.get()))
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]222 return metafile.Pass();
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]223
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]224 SkDynamicMemoryWStream pdf_stream;
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]225 if (!pdf_doc.emitPDF(&pdf_stream))
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]226 return metafile.Pass();
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]227
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]228 SkAutoDataUnref data_copy(pdf_stream.copyToData());
229 if (data_copy->size() == 0)
230 return scoped_ptr<PdfMetafileSkia>();
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]231
vitalybuka5d129052014-09-12 09:19:59[diff] [blame]232 metafile.reset(new PdfMetafileSkia);
233 if (!metafile->InitFromData(data_copy->bytes(),
234 base::checked_cast<uint32>(data_copy->size()))) {
235 metafile.reset();
236 }
237 return metafile.Pass();
kmadhusu@chromium.org597516372011-07-01 05:10:44[diff] [blame]238}
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]239
vandebo@chromium.org8f879292011-04-08 00:21:20[diff] [blame]240} // namespace printing