Make NGInlineItemsBuilder construct whitespace-collapsed offset mapping

third_party/WebKit/Source/core/layout/ng/inline/ng_inline_items_builder.cc

 // Copyright 2016 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/layout/ng/inline/ng_inline_items_builder.h"
 
 #include "core/layout/LayoutObject.h"
+#include "core/layout/ng/api/ng_offset_mapping_builder.h"
 #include "core/layout/ng/inline/ng_inline_node.h"
 #include "core/style/ComputedStyle.h"
 
 namespace blink {
 
 NGInlineItemsBuilder::~NGInlineItemsBuilder() {
   DCHECK_EQ(0u, exits_.size());
   DCHECK_EQ(text_.length(), items_->IsEmpty() ? 0 : items_->back().EndOffset());
 }
 
@@ skipping 132 matching lines @@
   unsigned start_offset = text_.length();
   for (unsigned i = start; i < end;) {
     UChar c = string[i];
     if (c == kNewlineCharacter) {
       // LayoutBR does not set preserve_newline, but should be preserved.
       if (!i && end == 1 && layout_object && layout_object->IsBR()) {
         AppendForcedBreak(style, layout_object);
         return;
       }
 
-      if (last_collapsible_space_ == CollapsibleSpace::kNone)
+      if (last_collapsible_space_ == CollapsibleSpace::kNone) {
         text_.Append(kSpaceCharacter);
+        if (mapping_builder_)
+          mapping_builder_->AppendBijection(1);
+      } else {
+        if (mapping_builder_)
+          mapping_builder_->AppendCollapsed(1);
+      }
       last_collapsible_space_ = CollapsibleSpace::kNewline;
       i++;
       continue;
     }
 
     if (c == kSpaceCharacter || c == kTabulationCharacter) {
       if (last_collapsible_space_ == CollapsibleSpace::kNone) {
         text_.Append(kSpaceCharacter);
         last_collapsible_space_ = CollapsibleSpace::kSpace;
+        if (mapping_builder_)
+          mapping_builder_->AppendBijection(1);
+      } else {
+        if (mapping_builder_)
+          mapping_builder_->AppendCollapsed(1);
       }
       i++;
       continue;
     }
 
     if (last_collapsible_space_ == CollapsibleSpace::kNewline) {
       RemoveTrailingCollapsibleNewlineIfNeeded(&start_offset, string, i, style);
     }
 
     size_t end_of_non_space = string.Find(IsCollapsibleSpace, i + 1);
     if (end_of_non_space == kNotFound)
       end_of_non_space = string.length();
     text_.Append(string, i, end_of_non_space - i);
+    if (mapping_builder_)
+      mapping_builder_->AppendBijection(end_of_non_space - i);
     i = end_of_non_space;
     last_collapsible_space_ = CollapsibleSpace::kNone;
   }
 
   if (text_.length() > start_offset) {
     AppendItem(items_, NGInlineItem::kText, start_offset, text_.length(), style,
                layout_object);
   }
 }
 
@@ skipping 15 matching lines @@
     if (end == kNotFound)
       end = string.length();
     unsigned start_offset = text_.length();
     text_.Append(string, start, end - start);
     AppendItem(items_, NGInlineItem::kText, start_offset, text_.length(), style,
                layout_object);
     start = end;
   }
 
   last_collapsible_space_ = CollapsibleSpace::kNone;
+  if (mapping_builder_)
+    mapping_builder_->AppendBijection(string.length());
 }
 
 void NGInlineItemsBuilder::AppendWithPreservingNewlines(
     const String& string,
     const ComputedStyle* style,
     LayoutObject* layout_object) {
   for (unsigned start = 0; start < string.length();) {
     if (string[start] == kNewlineCharacter) {
       AppendForcedBreak(style, layout_object);
       start++;
       continue;
     }
 
     size_t end = string.find(kNewlineCharacter, start + 1);
     if (end == kNotFound)
       end = string.length();
     AppendWithWhiteSpaceCollapsing(string, start, end, style, layout_object);
     start = end;
   }
 }
 
 void NGInlineItemsBuilder::AppendForcedBreak(const ComputedStyle* style,
                                              LayoutObject* layout_object) {
   // Remove collapsible spaces immediately before a preserved newline.
   unsigned start_offset = text_.length();
   RemoveTrailingCollapsibleSpaceIfExists(&start_offset);
 
   Append(NGInlineItem::kControl, kNewlineCharacter, style, layout_object);
+  if (mapping_builder_)
+    mapping_builder_->AppendBijection(1);
 
   // Remove collapsible spaces immediately after a preserved newline.
   last_collapsible_space_ = CollapsibleSpace::kSpace;
 }
 
 void NGInlineItemsBuilder::Append(NGInlineItem::NGInlineItemType type,
                                   UChar character,
                                   const ComputedStyle* style,
                                   LayoutObject* layout_object) {
   DCHECK_NE(character, kSpaceCharacter);
@@ skipping 43 matching lines @@
 void NGInlineItemsBuilder::RemoveTrailingCollapsibleSpace(
     unsigned* next_start_offset) {
   DCHECK_NE(last_collapsible_space_, CollapsibleSpace::kNone);
   DCHECK(!text_.IsEmpty());
   DCHECK_EQ(text_[text_.length() - 1], kSpaceCharacter);
 
   unsigned new_size = text_.length() - 1;
   text_.Resize(new_size);
   last_collapsible_space_ = CollapsibleSpace::kNone;
 
+  if (mapping_builder_)
+    mapping_builder_->CollapseLastCharacter();
+
   if (*next_start_offset <= new_size)
     return;
   *next_start_offset = new_size;
 
   // Adjust the last item if the removed space is already appended.
   for (unsigned i = items_->size(); i > 0;) {
     NGInlineItem& item = (*items_)[--i];
     DCHECK_EQ(item.EndOffset(), new_size + 1);
     if (item.Type() == NGInlineItem::kText) {
       DCHECK_GE(item.Length(), 1u);
@@ skipping 102 matching lines @@
 }
 
 void NGInlineItemsBuilder::Exit(LayoutObject* node) {
   while (!exits_.IsEmpty() && exits_.back().node == node) {
     Append(NGInlineItem::kBidiControl, exits_.back().character);
     exits_.pop_back();
   }
 }
 
 }  // namespace blink