Refactor DocumentMarkerController

third_party/WebKit/Source/core/editing/markers/RenderedTextMatchMarker.h

+// 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.
+
+#ifndef RenderedTextMatchMarker_h
+#define RenderedTextMatchMarker_h
+
+#include <utility>
+#include "core/editing/markers/DocumentMarker.h"
+#include "platform/geometry/LayoutRect.h"
+
+namespace blink {
+
+class RenderedTextMatchMarker : public DocumentMarker {

[Xiaocheng, 2017/03/21 03:59:44]

How about just naming it |TextMatchMarker|? "Rendered" seems unnecessary.

+ public:
+  static RenderedTextMatchMarker* create(const DocumentMarker& marker) {
+    DCHECK(marker.type() == DocumentMarker::TextMatch);
+    return new RenderedTextMatchMarker(marker);
+  }
+
+  bool isRendered() const { return m_state == State::ValidNotNull; }
+
+  void setRenderedRect(const LayoutRect& rect) {
+    if (m_state == State::ValidNotNull && rect == m_renderedRect)
+      return;
+    m_state = State::ValidNotNull;
+    m_renderedRect = rect;
+  }
+
+  const LayoutRect& renderedRect() const {
+    DCHECK_EQ(m_state, State::ValidNotNull);
+    return m_renderedRect;
+  }
+
+  void updateRenderedRectIfNeeded(const Node&);
+  void updateRenderedRect(const Node&);
+
+  void nullifyRenderedRect() {
+    m_state = State::ValidNull;
+    // Now |m_renderedRect| can not be accessed until |setRenderedRect| is
+    // called.
+  }
+
+  void invalidate() { m_state = State::Invalid; }
+  bool isValid() const { return m_state != State::Invalid; }
+
+  bool isRenderedTextMatch() const final;
+
+ private:
+  explicit RenderedTextMatchMarker(const DocumentMarker& marker)
+      : DocumentMarker(marker), m_state(State::Invalid) {}
+
+  enum class State { Invalid, ValidNull, ValidNotNull };
+
+  LayoutRect m_renderedRect;
+  State m_state;
+};
+
+DEFINE_TYPE_CASTS(RenderedTextMatchMarker,
+                  DocumentMarker,
+                  marker,
+                  marker->isRenderedTextMatch(),
+                  marker.isRenderedTextMatch());
+
+}  // namespace blink
+
+#endif  // RenderedTextMatchMarker_h