Return table headers from the table cell level.

Source/modules/accessibility/AXTableCell.cpp

 /*
  * Copyright (C) 2008 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
@@ skipping 13 matching lines @@
  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 #include "modules/accessibility/AXTableCell.h"
 
 #include "core/layout/LayoutTableCell.h"
 #include "modules/accessibility/AXObjectCacheImpl.h"
-
+#include "modules/accessibility/AXTable.h"
 
 namespace blink {
 
 using namespace HTMLNames;
 
 AXTableCell::AXTableCell(RenderObject* renderer, AXObjectCacheImpl* axObjectCache)
     : AXRenderObject(renderer, axObjectCache)
 {
 }
 
@@ skipping 108 matching lines @@
 }
 
 AccessibilityRole AXTableCell::determineAccessibilityRole()
 {
     if (!isTableCell())
         return AXRenderObject::determineAccessibilityRole();
 
     return scanToDecideHeaderRole();
 }
 
+void AXTableCell::rowHeadersForCell(AccessibilityChildrenVector& headers)
+{
+    AXObject* parent = parentTable();
+    if (!parent || !parent->isAXTable())
+        return;
+
+    AXTable* parentTable = toAXTable(parent);
+
+    pair<unsigned, unsigned> rowRange;
+    rowIndexRange(rowRange);
+
+    pair<unsigned, unsigned> columnRange;
+    columnIndexRange(columnRange);
+
+    for (unsigned i = 0; i < columnRange.first; i++) {
+        AXTableCell* cell = parentTable->cellForColumnAndRow(i, rowRange.first);
+        if (!cell || !cell->isTableCell() || cell == this)
+            continue;
+        if (cell->scanToDecideHeaderRole() == RowHeaderRole)
+            headers.append(cell);
+    }
+}
+
+void AXTableCell::columnHeadersForCell(AccessibilityChildrenVector& headers)
+{
+    AXObject* parent = parentTable();
+    if (!parent || !parent->isAXTable())
+        return;
+
+    AXTable* parentTable = toAXTable(parent);
+
+    pair<unsigned, unsigned> rowRange;
+    rowIndexRange(rowRange);
+
+    pair<unsigned, unsigned> columnRange;
+    columnIndexRange(columnRange);
+
+    for (unsigned i = 0; i < rowRange.first; i++) {
+        AXTableCell* cell = parentTable->cellForColumnAndRow(columnRange.first, i);
+        if (!cell || !cell->isTableCell() || headers.contains(cell) || cell == this)
+            continue;
+        if (cell->scanToDecideHeaderRole() == ColumnHeaderRole)
+            headers.append(cell);
+    }
+}
+
 void AXTableCell::rowIndexRange(pair<unsigned, unsigned>& rowRange)
 {
     if (!m_renderer || !m_renderer->isTableCell())
         return;
 
     LayoutTableCell* layoutCell = toLayoutTableCell(m_renderer);
     rowRange.first = layoutCell->rowIndex();
     rowRange.second = layoutCell->rowSpan();
 
     // since our table might have multiple sections, we have to offset our row appropriately
@@ skipping 75 matching lines @@
         return 0;
 
     Node* cellElement = headerCell->node();
     if (!cellElement || !cellElement->hasTagName(thTag))
         return 0;
 
     return axObjectCache()->getOrCreate(headerCell);
 }
 
 } // namespace blink