layout/printing/nsPrintObject.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "nsPrintObject.h"
   8
   9 #include "nsIDocumentViewer.h"
  10 #include "nsContentUtils.h"  // for nsAutoScriptBlocker
  11 #include "nsIInterfaceRequestorUtils.h"
  12 #include "nsPIDOMWindow.h"
  13 #include "nsPresContext.h"
  14 #include "nsGkAtoms.h"
  15 #include "nsComponentManagerUtils.h"
  16 #include "nsIBaseWindow.h"
  17 #include "nsDocShell.h"
  18
  19 #include "mozilla/PresShell.h"
  20 #include "mozilla/dom/BrowsingContext.h"
  21 #include "mozilla/dom/Document.h"
  22 #include "mozilla/dom/Element.h"
  23
  24 using namespace mozilla;
  25 using mozilla::dom::BrowsingContext;
  26 using mozilla::dom::Document;
  27 using mozilla::dom::Element;
  28 using mozilla::dom::Selection;
  29
  30 //---------------------------------------------------
  31 //-- nsPrintObject Class Impl
  32 //---------------------------------------------------
  33 nsPrintObject::nsPrintObject(nsIDocShell& aDocShell, Document& aDoc,
  34                              nsPrintObject* aParent)
  35     : mDocShell(&aDocShell), mDocument(&aDoc), mParent(aParent) {
  36   MOZ_COUNT_CTOR(nsPrintObject);
  37   MOZ_ASSERT(aDoc.IsStaticDocument());
  38
  39   if (!aParent) {
  40     // We are a root nsPrintObject.
  41     // Ensure the document has no presentation.
  42     DestroyPresentation();
  43   } else {
  44     // We are a nested nsPrintObject.
  45     nsCOMPtr<nsPIDOMWindowOuter> window = aDoc.GetWindow();
  46     mContent = window->GetFrameElementInternal();
  47   }
  48 }
  49
  50 nsPrintObject::~nsPrintObject() {
  51   MOZ_COUNT_DTOR(nsPrintObject);
  52
  53   DestroyPresentation();
  54   mDocShell = nullptr;
  55   mTreeOwner = nullptr;  // mTreeOwner must be released after mDocShell;
  56 }
  57
  58 //------------------------------------------------------------------
  59 // Resets PO by destroying the presentation
  60 void nsPrintObject::DestroyPresentation() {
  61   if (mDocument) {
  62     if (RefPtr<PresShell> ps = mDocument->GetPresShell()) {
  63       MOZ_DIAGNOSTIC_ASSERT(!mPresShell || ps == mPresShell);
  64       mPresShell = nullptr;
  65       nsAutoScriptBlocker scriptBlocker;
  66       ps->EndObservingDocument();
  67       ps->Destroy();
  68     }
  69   }
  70   mPresShell = nullptr;
  71   mPresContext = nullptr;
  72   mViewManager = nullptr;
  73 }
  74
  75 void nsPrintObject::EnablePrinting(bool aEnable) {
  76   mPrintingIsEnabled = aEnable;
  77
  78   for (const UniquePtr<nsPrintObject>& kid : mKids) {
  79     kid->EnablePrinting(aEnable);
  80   }
  81 }
  82
  83 bool nsPrintObject::HasSelection() const {
  84   return mDocument && mDocument->GetProperty(nsGkAtoms::printselectionranges);
  85 }
  86
  87 void nsPrintObject::EnablePrintingSelectionOnly() {
  88   mPrintingIsEnabled = HasSelection();
  89
  90   for (const UniquePtr<nsPrintObject>& kid : mKids) {
  91     kid->EnablePrintingSelectionOnly();
  92   }
  93 }