Make DOMSelection cache Range

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

Index: third_party/WebKit/Source/core/editing/SelectionEditor.cpp
diff --git a/third_party/WebKit/Source/core/editing/SelectionEditor.cpp b/third_party/WebKit/Source/core/editing/SelectionEditor.cpp
index 8206ecf15d3debae364272601af153a1bff8911d..64e10319c1d8794f8ae76ce3f59c7774c4dde921 100644
--- a/third_party/WebKit/Source/core/editing/SelectionEditor.cpp
+++ b/third_party/WebKit/Source/core/editing/SelectionEditor.cpp
@@ -25,6 +25,7 @@
 
 #include "core/editing/SelectionEditor.h"
 
+#include "core/editing/DOMSelection.h"
 #include "core/editing/EditingUtilities.h"
 #include "core/editing/Editor.h"
 #include "core/editing/SelectionAdjuster.h"
@@ -80,11 +81,32 @@ SelectionEditor::visibleSelection<EditingInFlatTreeStrategy>() const {
   return m_selectionInFlatTree;
 }
 
+static void markRangeDirty(TreeScope* treeScope) {
+  while (treeScope) {

[yosin_UTC9, 2017/02/07 10:04:29]

Better to use for-statement;

for (TreeScope* runner = treeScope; runner; runner = runner->parentTreeScope())
{
  if (!runner->hasSeelection())
    continue;
  treeScope()->getSelection()->markRangeDirty();
}

+    if (treeScope->hasSelection())
+      treeScope->getSelection()->markRangeDirty();
+    treeScope = treeScope->parentTreeScope();
+  }
+}
+
+static void markDOMSelectionDirty(const VisibleSelection& selection) {
+  if (selection.isNone())
+    return;
+
+  markRangeDirty(&selection.start().anchorNode()->treeScope());
+  if (selection.isCaret())
+    return;
+
+  markRangeDirty(&selection.end().anchorNode()->treeScope());
+}
+
 void SelectionEditor::setVisibleSelection(
     const VisibleSelection& newSelection,
     FrameSelection::SetSelectionOptions options) {
   DCHECK(newSelection.isValidFor(document())) << newSelection;
   resetLogicalRange();
+  markDOMSelectionDirty(m_selection);
+
   m_selection = newSelection;
   if (options & FrameSelection::DoNotAdjustInFlatTree) {
     m_selectionInFlatTree.setWithoutValidation(
@@ -95,6 +117,8 @@ void SelectionEditor::setVisibleSelection(
 
   SelectionAdjuster::adjustSelectionInFlatTree(&m_selectionInFlatTree,
                                                m_selection);
+
+  markDOMSelectionDirty(m_selection);
 }
 
 void SelectionEditor::setVisibleSelection(
@@ -103,9 +127,13 @@ void SelectionEditor::setVisibleSelection(
   DCHECK(newSelection.isValidFor(document())) << newSelection;
   DCHECK(!(options & FrameSelection::DoNotAdjustInFlatTree));
   resetLogicalRange();
+  markDOMSelectionDirty(m_selection);
+
   m_selectionInFlatTree = newSelection;
   SelectionAdjuster::adjustSelectionInDOMTree(&m_selection,
                                               m_selectionInFlatTree);
+
+  markDOMSelectionDirty(m_selection);
 }
 
 void SelectionEditor::setWithoutValidation(const Position& base,
@@ -115,9 +143,12 @@ void SelectionEditor::setWithoutValidation(const Position& base,
     DCHECK_EQ(base.document(), document());
   if (extent.isNotNull())
     DCHECK_EQ(extent.document(), document());
+  markDOMSelectionDirty(m_selection);
+
   m_selection.setWithoutValidation(base, extent);
   m_selectionInFlatTree.setWithoutValidation(toPositionInFlatTree(base),
                                              toPositionInFlatTree(extent));
+  markDOMSelectionDirty(m_selection);
 }
 
 void SelectionEditor::documentAttached(Document* document) {