зеркало из https://github.com/mozilla/gecko-dev.git
150 строки
3.7 KiB
C++
150 строки
3.7 KiB
C++
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 4 -*-
|
|
* 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 {
|
|
namespace gfx {
|
|
|
|
PrintTargetSkPDF::PrintTargetSkPDF(const IntSize& aSize,
|
|
UniquePtr<SkWStream> aStream)
|
|
: PrintTarget(/* not using cairo_surface_t */ nullptr, aSize)
|
|
, mOStream(std::move(aStream))
|
|
, mPageCanvas(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.
|
|
|
|
SkDocument::PDFMetadata metadata;
|
|
metadata.fTitle = NS_ConvertUTF16toUTF8(aTitle).get();
|
|
metadata.fCreator = "Firefox";
|
|
SkTime::DateTime now;
|
|
SkTime::GetDateTime(&now);
|
|
metadata.fCreation.fEnabled = true;
|
|
metadata.fCreation.fDateTime = now;
|
|
metadata.fModified.fEnabled = true;
|
|
metadata.fModified.fDateTime = now;
|
|
|
|
// SkDocument stores a non-owning raw pointer to aStream
|
|
mPDFDoc = SkDocument::MakePDF(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) {
|
|
SkDocument::PDFMetadata metadata;
|
|
// SkDocument stores a non-owning raw pointer to aStream
|
|
mRefPDFDoc = SkDocument::MakePDF(&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 = dt.forget();
|
|
}
|
|
|
|
return do_AddRef(mRefDT);
|
|
}
|
|
|
|
} // namespace gfx
|
|
} // namespace mozilla
|