Optimize collapsed border calculation (step 2)

third_party/WebKit/Source/core/layout/LayoutTableSection.cpp

 /*
  * Copyright (C) 1997 Martin Jones (mjones@kde.org)
  *           (C) 1997 Torben Weis (weis@kde.org)
  *           (C) 1998 Waldo Bastian (bastian@kde.org)
  *           (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  * Copyright (C) 2003, 2004, 2005, 2006, 2008, 2009, 2010, 2013 Apple Inc.
  *               All rights reserved.
  * Copyright (C) 2006 Alexey Proskuryakov (ap@nypop.com)
  *
@@ skipping 113 matching lines @@
   LayoutTableBoxComponent::StyleDidChange(diff, old_style);
   PropagateStyleToAnonymousChildren();
 
   if (!old_style)
     return;
 
   LayoutTable* table = this->Table();
   if (!table)
     return;
 
-  LayoutTableBoxComponent::InvalidateCollapsedBordersOnStyleChange(
-      *this, *table, diff, *old_style);
+  if (LayoutTableBoxComponent::NeedsInvalidateCollapsedBordersOnStyleChange(
+          *this, *table, diff, *old_style))
+    InvalidateCollapsedBordersOfAffectedCells();
 
   if (LayoutTableBoxComponent::DoCellsHaveDirtyWidth(*this, *table, diff,
                                                      *old_style))
     MarkAllCellsWidthsDirtyAndOrNeedsLayout(
         LayoutTable::kMarkDirtyAndNeedsLayout);
 }
 
 void LayoutTableSection::WillBeRemovedFromTree() {
   LayoutTableBoxComponent::WillBeRemovedFromTree();
 
@@ skipping 2019 matching lines @@
 
 bool LayoutTableSection::PaintedOutputOfObjectHasNoEffectRegardlessOfSize()
     const {
   // LayoutTableSection paints background from columns.
   if (Table()->HasColElements())
     return false;
   return LayoutTableBoxComponent::
       PaintedOutputOfObjectHasNoEffectRegardlessOfSize();
 }
 
+void LayoutTableSection::InvalidateCellCollapsedBordersIntersectingRow(
+    unsigned row_index) {
+  auto& row_struct = grid_[row_index].row;
+  for (auto& cell_struct : row_struct) {
+    if (cell_struct.in_col_span)
+      continue;
+    if (auto* cell = cell_struct.PrimaryCell())
+      cell->InvalidateCollapsedBorderValues();
+  }
+}
+
+void LayoutTableSection::InvalidateCollapsedBordersOfAffectedCells() {
+  auto* table = this->Table();
+  DCHECK(table->CollapseBorders());
+  if (table->NeedsInvalidateCollapsedBordersForAllCells())
+    return;
+
+  table->RecalcSectionsIfNeeded();
+  table->InvalidateCollapsedBorders();
+
+  if (NumRows()) {
+    // Invalidate cells intersecting the top row.
+    InvalidateCellCollapsedBordersIntersectingRow(0);
+    // Invalidate cells intersecting the bottom row.
+    if (NumRows() > 1)
+      InvalidateCellCollapsedBordersIntersectingRow(NumRows() - 1);
+
+    // Invalidate cell in start and end columns.
+    auto num_effective_columns = this->NumEffectiveColumns();
+    for (unsigned r = 0; r < NumRows(); ++r) {
+      if (auto* cell = PrimaryCellAt(r, 0))
+        cell->InvalidateCollapsedBorderValues();
+      if (auto* cell = PrimaryCellAt(r, num_effective_columns - 1))
+        cell->InvalidateCollapsedBorderValues();
+    }
+  }
+
+  // Invalidate cells in the row above the top row.
+  if (auto* section = table->SectionAbove(this, kSkipEmptySections)) {
+    section->InvalidateCellCollapsedBordersIntersectingRow(section->NumRows() -
+                                                           1);
+  }
+  // Invalidate cells in the row below the bottom row.
+  if (auto* section = table->SectionBelow(this, kSkipEmptySections)) {
+    section->InvalidateCellCollapsedBordersIntersectingRow(0);
+  }
+}
+
+LayoutTableCell* LayoutTableSection::PrimaryCellAboveInTable(
+    unsigned row,
+    unsigned effective_column) {
+  DCHECK(PrimaryCellAt(row, effective_column)->RowIndex() == row);
+  if (row)
+    return PrimaryCellAt(row - 1, effective_column);
+  if (auto* section_above = Table()->SectionAbove(this, kSkipEmptySections)) {
+    return section_above->PrimaryCellAt(section_above->NumRows() - 1,
+                                        effective_column);
+  }
+  return nullptr;
+}
+
+LayoutTableCell* LayoutTableSection::PrimaryCellBelowInTable(
+    unsigned row,
+    unsigned effective_column) {
+  DCHECK(PrimaryCellAt(row, effective_column)->RowIndex() == row);
+  unsigned row_below = row + PrimaryCellAt(row, effective_column)->RowSpan();
+  if (row_below < NumRows())
+    return PrimaryCellAt(row_below, effective_column);
+  if (auto* section_below = Table()->SectionBelow(this, kSkipEmptySections))
+    return section_below->PrimaryCellAt(0, effective_column);
+  return nullptr;
+}
+
 }  // namespace blink