OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2013 Google Inc. | 2 * Copyright 2013 Google Inc. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
6 */ | 6 */ |
7 | 7 |
8 #include "SkDocument.h" | 8 #include "SkDocument.h" |
9 #include "SkPDFCanon.h" | 9 #include "SkPDFCanon.h" |
10 #include "SkPDFDocument.h" | 10 #include "SkPDFDocument.h" |
11 #include "SkPDFDevice.h" | 11 #include "SkPDFDevice.h" |
12 | 12 |
| 13 namespace { |
13 class SkDocument_PDF : public SkDocument { | 14 class SkDocument_PDF : public SkDocument { |
14 public: | 15 public: |
15 SkDocument_PDF(SkWStream* stream, | 16 SkDocument_PDF(SkWStream* stream, |
16 void (*doneProc)(SkWStream*, bool), | 17 void (*doneProc)(SkWStream*, bool), |
17 SkScalar rasterDpi) | 18 SkScalar rasterDpi) |
18 : SkDocument(stream, doneProc) | 19 : SkDocument(stream, doneProc) |
19 , fDoc(SkNEW(SkPDFDocument)) | |
20 , fRasterDpi(rasterDpi) {} | 20 , fRasterDpi(rasterDpi) {} |
21 | 21 |
22 virtual ~SkDocument_PDF() { | 22 virtual ~SkDocument_PDF() { |
23 // subclasses must call close() in their destructors | 23 // subclasses must call close() in their destructors |
24 this->close(); | 24 this->close(); |
25 } | 25 } |
26 | 26 |
27 protected: | 27 protected: |
28 virtual SkCanvas* onBeginPage(SkScalar width, SkScalar height, | 28 virtual SkCanvas* onBeginPage(SkScalar width, SkScalar height, |
29 const SkRect& trimBox) SK_OVERRIDE { | 29 const SkRect& trimBox) SK_OVERRIDE { |
30 SkASSERT(!fCanvas.get()); | 30 SkASSERT(!fCanvas.get()); |
31 SkASSERT(!fDevice.get()); | |
32 | 31 |
33 SkISize pageSize = SkISize::Make( | 32 SkISize pageSize = SkISize::Make( |
34 SkScalarRoundToInt(width), SkScalarRoundToInt(height)); | 33 SkScalarRoundToInt(width), SkScalarRoundToInt(height)); |
35 fDevice.reset(SkPDFDevice::Create(pageSize, fRasterDpi, &fCanon)); | 34 SkAutoTUnref<SkPDFDevice> device( |
36 fCanvas.reset(SkNEW_ARGS(SkCanvas, (fDevice))); | 35 SkPDFDevice::Create(pageSize, fRasterDpi, &fCanon)); |
| 36 fCanvas.reset(SkNEW_ARGS(SkCanvas, (device.get()))); |
| 37 fPageDevices.push(device.detach()); |
37 fCanvas->clipRect(trimBox); | 38 fCanvas->clipRect(trimBox); |
38 fCanvas->translate(trimBox.x(), trimBox.y()); | 39 fCanvas->translate(trimBox.x(), trimBox.y()); |
39 return fCanvas.get(); | 40 return fCanvas.get(); |
40 } | 41 } |
41 | 42 |
42 void onEndPage() SK_OVERRIDE { | 43 void onEndPage() SK_OVERRIDE { |
43 SkASSERT(fCanvas.get()); | 44 SkASSERT(fCanvas.get()); |
44 SkASSERT(fDevice.get()); | |
45 | |
46 fCanvas->flush(); | 45 fCanvas->flush(); |
47 fDoc->appendPage(fDevice.get()); | |
48 | |
49 fCanvas.reset(NULL); | 46 fCanvas.reset(NULL); |
50 fDevice.reset(NULL); | |
51 } | 47 } |
52 | 48 |
53 bool onClose(SkWStream* stream) SK_OVERRIDE { | 49 bool onClose(SkWStream* stream) SK_OVERRIDE { |
54 SkASSERT(!fCanvas.get()); | 50 SkASSERT(!fCanvas.get()); |
55 SkASSERT(!fDevice.get()); | |
56 | 51 |
57 bool success = fDoc->emitPDF(stream); | 52 bool success = SkPDFDocument::EmitPDF(fPageDevices, stream); |
58 fDoc.free(); | 53 fPageDevices.unrefAll(); |
59 fCanon.reset(); | 54 fCanon.reset(); |
60 return success; | 55 return success; |
61 } | 56 } |
62 | 57 |
63 void onAbort() SK_OVERRIDE { | 58 void onAbort() SK_OVERRIDE { |
64 fDoc.free(); | 59 fPageDevices.unrefAll(); |
65 fCanon.reset(); | 60 fCanon.reset(); |
66 } | 61 } |
67 | 62 |
68 private: | 63 private: |
69 SkPDFCanon fCanon; | 64 SkPDFCanon fCanon; |
70 SkAutoTDelete<SkPDFDocument> fDoc; | 65 SkTDArray<SkPDFDevice*> fPageDevices; |
71 SkAutoTUnref<SkPDFDevice> fDevice; | |
72 SkAutoTUnref<SkCanvas> fCanvas; | 66 SkAutoTUnref<SkCanvas> fCanvas; |
73 SkScalar fRasterDpi; | 67 SkScalar fRasterDpi; |
74 }; | 68 }; |
75 | 69 } // namespace |
76 /////////////////////////////////////////////////////////////////////////////// | 70 /////////////////////////////////////////////////////////////////////////////// |
77 | 71 |
78 SkDocument* SkDocument::CreatePDF(SkWStream* stream, SkScalar dpi) { | 72 SkDocument* SkDocument::CreatePDF(SkWStream* stream, SkScalar dpi) { |
79 return stream ? SkNEW_ARGS(SkDocument_PDF, (stream, NULL, dpi)) : NULL; | 73 return stream ? SkNEW_ARGS(SkDocument_PDF, (stream, NULL, dpi)) : NULL; |
80 } | 74 } |
81 | 75 |
82 static void delete_wstream(SkWStream* stream, bool aborted) { | 76 static void delete_wstream(SkWStream* stream, bool aborted) { |
83 SkDELETE(stream); | 77 SkDELETE(stream); |
84 } | 78 } |
85 | 79 |
86 SkDocument* SkDocument::CreatePDF(const char path[], SkScalar dpi) { | 80 SkDocument* SkDocument::CreatePDF(const char path[], SkScalar dpi) { |
87 SkFILEWStream* stream = SkNEW_ARGS(SkFILEWStream, (path)); | 81 SkFILEWStream* stream = SkNEW_ARGS(SkFILEWStream, (path)); |
88 if (!stream->isValid()) { | 82 if (!stream->isValid()) { |
89 SkDELETE(stream); | 83 SkDELETE(stream); |
90 return NULL; | 84 return NULL; |
91 } | 85 } |
92 return SkNEW_ARGS(SkDocument_PDF, (stream, delete_wstream, dpi)); | 86 return SkNEW_ARGS(SkDocument_PDF, (stream, delete_wstream, dpi)); |
93 } | 87 } |
OLD | NEW |