Make DOMSelection cache Range

third_party/WebKit/Source/core/editing/SelectionEditor.cpp

 /*
  * Copyright (C) 2004, 2008, 2009, 2010 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 32 matching lines @@
   m_selection = VisibleSelection();
   m_selectionInFlatTree = VisibleSelectionInFlatTree();
   if (!shouldAlwaysUseDirectionalSelection())
     return;
   m_selection.setIsDirectional(true);
   m_selectionInFlatTree.setIsDirectional(true);
 }
 
 void SelectionEditor::dispose() {
   resetLogicalRange();
+  clearDocumentCachedRange();
   clearVisibleSelection();
 }
 
 const Document& SelectionEditor::document() const {
   DCHECK(m_document);
   return *m_document;
 }
 
 template <>
 const VisibleSelection& SelectionEditor::visibleSelection<EditingStrategy>()
@@ skipping 15 matching lines @@
     return m_selectionInFlatTree;
   DCHECK_EQ(m_selectionInFlatTree.base().document(), document());
   return m_selectionInFlatTree;
 }
 
 void SelectionEditor::setVisibleSelection(
     const VisibleSelection& newSelection,
     FrameSelection::SetSelectionOptions options) {
   DCHECK(newSelection.isValidFor(document())) << newSelection;
   resetLogicalRange();
+  clearDocumentCachedRange();
+
   m_selection = newSelection;
   if (options & FrameSelection::DoNotAdjustInFlatTree) {
     m_selectionInFlatTree.setWithoutValidation(
         toPositionInFlatTree(m_selection.base()),
         toPositionInFlatTree(m_selection.extent()));
     return;
   }
 
   SelectionAdjuster::adjustSelectionInFlatTree(&m_selectionInFlatTree,
                                                m_selection);
 }
 
 void SelectionEditor::setVisibleSelection(
     const VisibleSelectionInFlatTree& newSelection,
     FrameSelection::SetSelectionOptions options) {
   DCHECK(newSelection.isValidFor(document())) << newSelection;
   DCHECK(!(options & FrameSelection::DoNotAdjustInFlatTree));
   resetLogicalRange();
+  clearDocumentCachedRange();
+
   m_selectionInFlatTree = newSelection;
   SelectionAdjuster::adjustSelectionInDOMTree(&m_selection,
                                               m_selectionInFlatTree);
 }
 
 void SelectionEditor::setWithoutValidation(const Position& base,
                                            const Position& extent) {
   resetLogicalRange();
   if (base.isNotNull())
     DCHECK_EQ(base.document(), document());
   if (extent.isNotNull())
     DCHECK_EQ(extent.document(), document());
+  clearDocumentCachedRange();
+
   m_selection.setWithoutValidation(base, extent);
   m_selectionInFlatTree.setWithoutValidation(toPositionInFlatTree(base),
                                              toPositionInFlatTree(extent));
 }
 
 void SelectionEditor::documentAttached(Document* document) {
   DCHECK(document);
   DCHECK(!m_document) << m_document;
   m_document = document;
 }
@@ skipping 29 matching lines @@
   return frame()->editor().behavior().shouldConsiderSelectionAsDirectional();
 }
 
 void SelectionEditor::updateIfNeeded() {
   DCHECK(m_selection.isValidFor(document())) << m_selection;
   DCHECK(m_selectionInFlatTree.isValidFor(document())) << m_selection;
   m_selection.updateIfNeeded();
   m_selectionInFlatTree.updateIfNeeded();
 }
 
+void SelectionEditor::cacheRangeOfDocument(Range* range) {
+  m_cachedRange = range;
+}
+
+Range* SelectionEditor::documentCachedRange() const {
+  return m_cachedRange;
+}
+
+void SelectionEditor::clearDocumentCachedRange() {
+  m_cachedRange = nullptr;
+}
+
 DEFINE_TRACE(SelectionEditor) {
   visitor->trace(m_document);
   visitor->trace(m_frame);
   visitor->trace(m_selection);
   visitor->trace(m_selectionInFlatTree);
   visitor->trace(m_logicalRange);
+  visitor->trace(m_cachedRange);
 }
 
 }  // namespace blink