gfx/thebes/PrintTargetSkPDF.cpp

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