1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*-
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "PrintTargetSkPDF.h"
#include "mozilla/gfx/2D.h"
#include "nsString.h"
#include <vector>
namespace mozilla::gfx {
PrintTargetSkPDF::PrintTargetSkPDF(const IntSize& aSize,
UniquePtr<SkWStream> aStream)
: PrintTarget(/* not using cairo_surface_t */ nullptr, aSize),
mOStream(std::move(aStream)),
mPageCanvas(nullptr),
mRefCanvas(nullptr) {}
PrintTargetSkPDF::~PrintTargetSkPDF() {
Finish(); // ensure stream is flushed
// Make sure mPDFDoc and mRefPDFDoc are destroyed before our member streams
// (which they wrap) are destroyed:
mPDFDoc = nullptr;
mRefPDFDoc = nullptr;
}
/* static */
already_AddRefed<PrintTargetSkPDF> PrintTargetSkPDF::CreateOrNull(
UniquePtr<SkWStream> aStream, const IntSize& aSizeInPoints) {
return do_AddRef(new PrintTargetSkPDF(aSizeInPoints, std::move(aStream)));
}
nsresult PrintTargetSkPDF::BeginPrinting(const nsAString& aTitle,
const nsAString& aPrintToFileName,
int32_t aStartPage, int32_t aEndPage) {
// We need to create the SkPDFDocument here rather than in CreateOrNull
// because it's only now that we are given aTitle which we want for the
// PDF metadata.
SkPDF::Metadata metadata;
metadata.fTitle = NS_ConvertUTF16toUTF8(aTitle).get();
metadata.fCreator = "Firefox";
SkTime::DateTime now;
SkTime::GetDateTime(&now);
metadata.fCreation = now;
metadata.fModified = now;
// SkDocument stores a non-owning raw pointer to aStream
mPDFDoc = SkPDF::MakeDocument(mOStream.get(), metadata);
return mPDFDoc ? NS_OK : NS_ERROR_FAILURE;
}
nsresult PrintTargetSkPDF::BeginPage() {
mPageCanvas = mPDFDoc->beginPage(mSize.width, mSize.height);
return !mPageCanvas ? NS_ERROR_FAILURE : PrintTarget::BeginPage();
}
nsresult PrintTargetSkPDF::EndPage() {
mPageCanvas = nullptr;
mPageDT = nullptr;
return PrintTarget::EndPage();
}
nsresult PrintTargetSkPDF::EndPrinting() {
mPDFDoc->close();
if (mRefPDFDoc) {
mRefPDFDoc->close();
}
mPageCanvas = nullptr;
mPageDT = nullptr;
return NS_OK;
}
void PrintTargetSkPDF::Finish() {
if (mIsFinished) {
return;
}
mOStream->flush();
PrintTarget::Finish();
}
already_AddRefed<DrawTarget> PrintTargetSkPDF::MakeDrawTarget(
const IntSize& aSize, DrawEventRecorder* aRecorder) {
if (aRecorder) {
return PrintTarget::MakeDrawTarget(aSize, aRecorder);
}
// MOZ_ASSERT(aSize == mSize, "Should mPageCanvas size match?");
if (!mPageCanvas) {
return nullptr;
}
mPageDT = Factory::CreateDrawTargetWithSkCanvas(mPageCanvas);
if (!mPageDT) {
mPageCanvas = nullptr;
return nullptr;
}
return do_AddRef(mPageDT);
}
already_AddRefed<DrawTarget> PrintTargetSkPDF::GetReferenceDrawTarget() {
if (!mRefDT) {
SkPDF::Metadata metadata;
// SkDocument stores a non-owning raw pointer to aStream
mRefPDFDoc = SkPDF::MakeDocument(&mRefOStream, metadata);
if (!mRefPDFDoc) {
return nullptr;
}
mRefCanvas = mRefPDFDoc->beginPage(mSize.width, mSize.height);
if (!mRefCanvas) {
return nullptr;
}
RefPtr<DrawTarget> dt = Factory::CreateDrawTargetWithSkCanvas(mRefCanvas);
if (!dt) {
return nullptr;
}
mRefDT = std::move(dt);
}
return do_AddRef(mRefDT);
}
} // namespace mozilla::gfx
|
