Add CompositionMarkerList in preparation for DocumentMarkerController refactor

third_party/WebKit/Source/core/editing/markers/DocumentMarkerList.cpp

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "core/editing/markers/DocumentMarkerList.h"
+
+#include "core/editing/markers/DocumentMarkerController.h"
+
+namespace blink {
+
+namespace {
+
+bool endsBefore(size_t startOffset, const Member<DocumentMarker>& rhv) {
+  return startOffset < rhv->endOffset();
+}
+
+}  // namespace
+
+DocumentMarkerList::DocumentMarkerList(
+    DocumentMarkerController* documentMarkerController)
+    : m_documentMarkerController(documentMarkerController) {}
+
+bool DocumentMarkerList::isEditingMarkerList() const {
+  return false;
+}
+
+bool DocumentMarkerList::isSpellCheckMarkerList() const {
+  return false;
+}
+
+DocumentMarkerList::iterator DocumentMarkerList::begin() {
+  return m_markers.begin();
+}
+
+DocumentMarkerList::iterator DocumentMarkerList::end() {
+  return m_markers.end();
+}
+
+DocumentMarkerList::const_iterator DocumentMarkerList::begin() const {
+  return m_markers.begin();
+}
+
+DocumentMarkerList::const_iterator DocumentMarkerList::end() const {
+  return m_markers.end();
+}
+
+void DocumentMarkerList::appendMarkersToInputList(
+    DocumentMarkerVector* list) const {
+  for (Member<DocumentMarker> marker : m_markers) {
+    list->push_back(marker);
+  }
+}
+
+bool DocumentMarkerList::copyMarkers(unsigned startOffset,

[Xiaocheng, 2017/03/24 19:43:49]

With a second thought, I think DML should not depend on DMC.

So this function should take a |DML* dstList| parameter for receiving the copied
markers, instead of a |Node*|.

+                                     int length,
+                                     Node* dstNode,
+                                     int delta) const {
+  bool docDirty = false;
+  unsigned endOffset = startOffset + length - 1;
+
+  for (Member<DocumentMarker> marker : m_markers) {
+    // pin the marker to the specified range and apply the shift delta
+    if (marker->endOffset() >= startOffset &&
+        marker->startOffset() <= endOffset) {
+      docDirty = true;
+      if (marker->startOffset() < startOffset)
+        marker->setStartOffset(startOffset);
+      if (marker->endOffset() > endOffset)
+        marker->setEndOffset(endOffset);
+      marker->shiftOffsets(delta);
+
+      m_documentMarkerController->addMarker(dstNode, *marker);
+    }
+  }
+
+  return docDirty;
+}
+
+void DocumentMarkerList::removeMarkers(
+    unsigned startOffset,
+    int length,
+    bool shouldRemovePartiallyOverlappingMarkers,
+    bool* didRemoveMarker) {
+  unsigned endOffset = startOffset + length;
+  size_t markerIndex = 0;
+  if (markerListIsSorted()) {
+    markerIndex =
+        getPosOfFirstMarkerNotEndingBefore(startOffset) - m_markers.begin();
+  }
+
+  for (; markerIndex < m_markers.size();) {

[Xiaocheng, 2017/03/24 19:43:49]

nit: |while| seems better here.

+    DocumentMarker& marker = *m_markers.at(markerIndex);
+    if (!markerListIsSorted()) {
+      if (marker.endOffset() <= startOffset) {
+        ++markerIndex;
+        continue;
+      }
+    }
+
+    if (marker.startOffset() >= endOffset) {
+      if (markerListIsSorted())
+        break;
+      ++markerIndex;
+      continue;
+    }
+
+    // pitch the old marker
+    m_markers.remove(markerIndex);
+    *didRemoveMarker = true;
+
+    if (shouldRemovePartiallyOverlappingMarkers) {
+      // Stop here. Don't add resulting slices back.
+      continue;
+    }
+
+    // add either of the resulting slices that are left after removing target
+    if (startOffset > marker.startOffset()) {
+      DocumentMarker* newLeft = new DocumentMarker(marker);
+      newLeft->setEndOffset(startOffset);
+      if (markerListIsSorted()) {
+        m_markers.insert(markerIndex, *newLeft);
+        // Move to the marker after the inserted one.
+        ++markerIndex;
+      } else {
+        // For the unsorted case, we just stick the new marker at the end of the
+        // list. The loop will eventually run on it but that's not a problem
+        // since it's known to be outside the range being removed.
+        m_markers.push_back(*newLeft);
+      }
+    }
+    if (marker.endOffset() > endOffset) {
+      DocumentMarker* newRight = new DocumentMarker(marker);
+      newRight->setStartOffset(endOffset);
+      if (markerListIsSorted()) {
+        m_markers.insert(markerIndex, *newRight);
+        // Move to the marker after the inserted one.
+        ++markerIndex;
+      } else {
+        m_markers.push_back(*newRight);
+      }
+    }
+  }
+}
+
+bool DocumentMarkerList::shiftMarkers(unsigned offset,
+                                      unsigned oldLength,
+                                      unsigned newLength) {
+  bool didShift = false;
+  for (auto it = m_markers.begin(); it != m_markers.end(); ++it) {
+    DocumentMarker& marker = **it;
+
+    DocumentMarker::ShiftMarkerResult result =
+        marker.getShiftedMarkerPosition(offset, oldLength, newLength);
+    if (result.shouldRemoveMarker) {
+      m_markers.remove(it - m_markers.begin());
+      --it;
+
+      didShift = true;
+    } else if (result.newStartOffset != marker.startOffset() ||
+               result.newEndOffset != marker.endOffset()) {
+      marker.setStartOffset(result.newStartOffset);
+      marker.setEndOffset(result.newEndOffset);
+
+      didShift = true;
+    }
+  }
+  return didShift;
+}
+
+HeapVector<Member<DocumentMarker>>::iterator
+DocumentMarkerList::getPosOfFirstMarkerNotEndingBefore(size_t startOffset) {
+  return std::upper_bound(m_markers.begin(), m_markers.end(), startOffset,

[Xiaocheng, 2017/03/24 19:43:49]

Should DCHECK(markerListIsSorted());

+                          endsBefore);
+}
+
+DEFINE_TRACE(DocumentMarkerList) {
+  visitor->trace(m_documentMarkerController);
+  visitor->trace(m_markers);
+}
+
+}  // namespace blink