fune/gfx/thebes/PrintTargetSkPDF.cpp
Jonathan Watt 07bafd2423 Bug 1833244 p1. Create infrastructure to pass page dimensions to PrintTarget::BeginPage. r=dholbert,geckoview-reviewers,jonalmeida
OS print drivers/devices know nothing about page dimensions unless we tell
them. Previously, the physical page dimensions (including orientation) have
always been the same, so communicating their dimensions once at the start of
a print has been enough. In preparation for supporting different "physical"
page dimensions (in the immediate future only different page orientations) when
we save to PDF, we need to have the infrastructure to pass dimensions through
on a page-by-page basis. This patch adds that.

None of the PrintTarget subclasses do anything with this extra information yet,
but in a follow-up patch PrintTargetPDF will use this information to create
PDFs with mixed page orientations.

Differential Revision: https://phabricator.services.mozilla.com/D179423
2023-07-10 14:26:12 +00:00

127 lines
3.6 KiB
C++

/* -*- 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(const IntSize& aSizeInPoints) {
mPageCanvas = mPDFDoc->beginPage(mSize.width, mSize.height);
return !mPageCanvas ? NS_ERROR_FAILURE
: PrintTarget::BeginPage(aSizeInPoints);
}
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