Replace WTF::move with WTFMove
[WebKit-https.git] / Source / WebCore / html / HTMLTableRowElement.cpp
index 5ec3bce..766bd0c 100644 (file)
 #include "HTMLTableRowElement.h"
 
 #include "ExceptionCode.h"
-#include "HTMLCollection.h"
+#include "GenericCachedHTMLCollection.h"
 #include "HTMLNames.h"
-#include "HTMLTableCellElement.h"
+#include "HTMLTableDataCellElement.h"
 #include "HTMLTableElement.h"
 #include "HTMLTableSectionElement.h"
 #include "NodeList.h"
+#include "NodeRareData.h"
 #include "Text.h"
 
 namespace WebCore {
 
 using namespace HTMLNames;
 
-HTMLTableRowElement::HTMLTableRowElement(const QualifiedName& tagName, Document* document)
+HTMLTableRowElement::HTMLTableRowElement(const QualifiedName& tagName, Document& document)
     : HTMLTablePartElement(tagName, document)
 {
     ASSERT(hasTagName(trTag));
 }
 
-PassRefPtr<HTMLTableRowElement> HTMLTableRowElement::create(Document* document)
+Ref<HTMLTableRowElement> HTMLTableRowElement::create(Document& document)
 {
-    return adoptRef(new HTMLTableRowElement(trTag, document));
+    return adoptRef(*new HTMLTableRowElement(trTag, document));
 }
 
-PassRefPtr<HTMLTableRowElement> HTMLTableRowElement::create(const QualifiedName& tagName, Document* document)
+Ref<HTMLTableRowElement> HTMLTableRowElement::create(const QualifiedName& tagName, Document& document)
 {
-    return adoptRef(new HTMLTableRowElement(tagName, document));
+    return adoptRef(*new HTMLTableRowElement(tagName, document));
 }
 
 int HTMLTableRowElement::rowIndex() const
 {
-    ContainerNode* table = parentNode();
-    if (!table)
+    auto* parent = parentNode();
+    if (!parent)
         return -1;
-    table = table->parentNode();
-    if (!table || !table->hasTagName(tableTag))
-        return -1;
-
-    // To match Firefox, the row indices work like this:
-    //   Rows from the first <thead> are numbered before all <tbody> rows.
-    //   Rows from the first <tfoot> are numbered after all <tbody> rows.
-    //   Rows from other <thead> and <tfoot> elements don't get row indices at all.
-
-    int rIndex = 0;
 
-    if (HTMLTableSectionElement* head = static_cast<HTMLTableElement*>(table)->tHead()) {
-        for (Node *row = head->firstChild(); row; row = row->nextSibling()) {
-            if (row == this)
-                return rIndex;
-            if (row->hasTagName(trTag))
-                ++rIndex;
-        }
-    }
-    
-    for (Node *node = table->firstChild(); node; node = node->nextSibling()) {
-        if (node->hasTagName(tbodyTag)) {
-            HTMLTableSectionElement* section = static_cast<HTMLTableSectionElement*>(node);
-            for (Node* row = section->firstChild(); row; row = row->nextSibling()) {
-                if (row == this)
-                    return rIndex;
-                if (row->hasTagName(trTag))
-                    ++rIndex;
-            }
-        }
+    HTMLTableElement* table;
+    if (is<HTMLTableElement>(*parent))
+        table = downcast<HTMLTableElement>(parent);
+    else {
+        if (!is<HTMLTableSectionElement>(*parent) || !is<HTMLTableElement>(parent->parentNode()))
+            return -1;
+        table = downcast<HTMLTableElement>(parent->parentNode());
     }
 
-    if (HTMLTableSectionElement* foot = static_cast<HTMLTableElement*>(table)->tFoot()) {
-        for (Node *row = foot->firstChild(); row; row = row->nextSibling()) {
-            if (row == this)
-                return rIndex;
-            if (row->hasTagName(trTag))
-                ++rIndex;
-        }
+    auto rows = table->rows();
+    unsigned length = rows->length();
+    for (unsigned i = 0; i < length; ++i) {
+        if (rows->item(i) == this)
+            return i;
     }
 
-    // We get here for rows that are in <thead> or <tfoot> sections other than the main header and footer.
     return -1;
 }
 
 int HTMLTableRowElement::sectionRowIndex() const
 {
-    int rIndex = 0;
-    const Node *n = this;
-    do {
-        n = n->previousSibling();
-        if (n && n->hasTagName(trTag))
-            rIndex++;
+    auto* parent = parentNode();
+    if (!parent)
+        return -1;
+
+    RefPtr<HTMLCollection> rows;
+    if (is<HTMLTableSectionElement>(*parent))
+        rows = downcast<HTMLTableSectionElement>(*parent).rows();
+    else if (is<HTMLTableElement>(*parent))
+        rows = downcast<HTMLTableElement>(*parent).rows();
+    else
+        return -1;
+
+    unsigned length = rows->length();
+    for (unsigned i = 0; i < length; ++i) {
+        if (rows->item(i) == this)
+            return i;
     }
-    while (n);
 
-    return rIndex;
+    return -1;
 }
 
-PassRefPtr<HTMLElement> HTMLTableRowElement::insertCell(int index, ExceptionCode& ec)
+RefPtr<HTMLTableCellElement> HTMLTableRowElement::insertCell(int index, ExceptionCode& ec)
 {
-    RefPtr<HTMLCollection> children = cells();
-    int numCells = children ? children->length() : 0;
+    Ref<HTMLCollection> children = cells();
+    int numCells = children->length();
     if (index < -1 || index > numCells) {
         ec = INDEX_SIZE_ERR;
-        return 0;
+        return nullptr;
     }
 
-    RefPtr<HTMLTableCellElement> cell = HTMLTableCellElement::create(tdTag, document());
+    auto cell = HTMLTableDataCellElement::create(document());
     if (index < 0 || index >= numCells)
-        appendChild(cell, ec);
+        appendChild(cell.copyRef(), ec);
     else {
         Node* n;
         if (index < 1)
             n = firstChild();
         else
             n = children->item(index);
-        insertBefore(cell, n, ec);
+        insertBefore(cell.copyRef(), n, ec);
     }
-    return cell.release();
+    return WTFMove(cell);
 }
 
 void HTMLTableRowElement::deleteCell(int index, ExceptionCode& ec)
 {
-    RefPtr<HTMLCollection> children = cells();
-    int numCells = children ? children->length() : 0;
+    Ref<HTMLCollection> children = cells();
+    int numCells = children->length();
     if (index == -1)
         index = numCells-1;
-    if (index >= 0 && index < numCells) {
-        RefPtr<Node> cell = children->item(index);
-        HTMLElement::removeChild(cell.get(), ec);
-    } else
+    if (index >= 0 && index < numCells)
+        HTMLElement::removeChild(*children->item(index), ec);
+    else
         ec = INDEX_SIZE_ERR;
 }
 
-PassRefPtr<HTMLCollection> HTMLTableRowElement::cells()
+Ref<HTMLCollection> HTMLTableRowElement::cells()
 {
-    return ensureCachedHTMLCollection(TRCells);
+    return ensureRareData().ensureNodeLists().addCachedCollection<GenericCachedHTMLCollection<CollectionTypeTraits<TRCells>::traversalType>>(*this, TRCells);
 }
 
 void HTMLTableRowElement::setCells(HTMLCollection*, ExceptionCode& ec)