Refactor HTMLCollection to be as fast as CachedLiveNodeList
[WebKit-https.git] / Source / WebCore / html / HTMLTableRowElement.cpp
1 /*
2  * Copyright (C) 1997 Martin Jones (mjones@kde.org)
3  *           (C) 1997 Torben Weis (weis@kde.org)
4  *           (C) 1998 Waldo Bastian (bastian@kde.org)
5  *           (C) 1999 Lars Knoll (knoll@kde.org)
6  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
7  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2010 Apple Inc. All rights reserved.
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Library General Public
11  * License as published by the Free Software Foundation; either
12  * version 2 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Library General Public License for more details.
18  *
19  * You should have received a copy of the GNU Library General Public License
20  * along with this library; see the file COPYING.LIB.  If not, write to
21  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
22  * Boston, MA 02110-1301, USA.
23  */
24
25 #include "config.h"
26 #include "HTMLTableRowElement.h"
27
28 #include "ExceptionCode.h"
29 #include "GenericCachedHTMLCollection.h"
30 #include "HTMLNames.h"
31 #include "HTMLTableCellElement.h"
32 #include "HTMLTableElement.h"
33 #include "HTMLTableSectionElement.h"
34 #include "NodeList.h"
35 #include "NodeRareData.h"
36 #include "Text.h"
37
38 namespace WebCore {
39
40 using namespace HTMLNames;
41
42 HTMLTableRowElement::HTMLTableRowElement(const QualifiedName& tagName, Document& document)
43     : HTMLTablePartElement(tagName, document)
44 {
45     ASSERT(hasTagName(trTag));
46 }
47
48 Ref<HTMLTableRowElement> HTMLTableRowElement::create(Document& document)
49 {
50     return adoptRef(*new HTMLTableRowElement(trTag, document));
51 }
52
53 Ref<HTMLTableRowElement> HTMLTableRowElement::create(const QualifiedName& tagName, Document& document)
54 {
55     return adoptRef(*new HTMLTableRowElement(tagName, document));
56 }
57
58 int HTMLTableRowElement::rowIndex() const
59 {
60     ContainerNode* table = parentNode();
61     if (!table)
62         return -1;
63     table = table->parentNode();
64     if (!is<HTMLTableElement>(table))
65         return -1;
66
67     // To match Firefox, the row indices work like this:
68     //   Rows from the first <thead> are numbered before all <tbody> rows.
69     //   Rows from the first <tfoot> are numbered after all <tbody> rows.
70     //   Rows from other <thead> and <tfoot> elements don't get row indices at all.
71
72     int rIndex = 0;
73
74     if (HTMLTableSectionElement* head = downcast<HTMLTableElement>(*table).tHead()) {
75         for (Node *row = head->firstChild(); row; row = row->nextSibling()) {
76             if (row == this)
77                 return rIndex;
78             if (row->hasTagName(trTag))
79                 ++rIndex;
80         }
81     }
82     
83     for (Node *node = table->firstChild(); node; node = node->nextSibling()) {
84         if (node->hasTagName(tbodyTag)) {
85             HTMLTableSectionElement& section = downcast<HTMLTableSectionElement>(*node);
86             for (Node* row = section.firstChild(); row; row = row->nextSibling()) {
87                 if (row == this)
88                     return rIndex;
89                 if (row->hasTagName(trTag))
90                     ++rIndex;
91             }
92         }
93     }
94
95     if (HTMLTableSectionElement* foot = downcast<HTMLTableElement>(*table).tFoot()) {
96         for (Node *row = foot->firstChild(); row; row = row->nextSibling()) {
97             if (row == this)
98                 return rIndex;
99             if (row->hasTagName(trTag))
100                 ++rIndex;
101         }
102     }
103
104     // We get here for rows that are in <thead> or <tfoot> sections other than the main header and footer.
105     return -1;
106 }
107
108 int HTMLTableRowElement::sectionRowIndex() const
109 {
110     int rIndex = 0;
111     const Node *n = this;
112     do {
113         n = n->previousSibling();
114         if (n && n->hasTagName(trTag))
115             rIndex++;
116     }
117     while (n);
118
119     return rIndex;
120 }
121
122 RefPtr<HTMLElement> HTMLTableRowElement::insertCell(int index, ExceptionCode& ec)
123 {
124     Ref<HTMLCollection> children = cells();
125     int numCells = children->length();
126     if (index < -1 || index > numCells) {
127         ec = INDEX_SIZE_ERR;
128         return 0;
129     }
130
131     RefPtr<HTMLTableCellElement> cell = HTMLTableCellElement::create(tdTag, document());
132     if (index < 0 || index >= numCells)
133         appendChild(cell, ec);
134     else {
135         Node* n;
136         if (index < 1)
137             n = firstChild();
138         else
139             n = children->item(index);
140         insertBefore(cell, n, ec);
141     }
142     return cell;
143 }
144
145 void HTMLTableRowElement::deleteCell(int index, ExceptionCode& ec)
146 {
147     Ref<HTMLCollection> children = cells();
148     int numCells = children->length();
149     if (index == -1)
150         index = numCells-1;
151     if (index >= 0 && index < numCells) {
152         RefPtr<Node> cell = children->item(index);
153         HTMLElement::removeChild(cell.get(), ec);
154     } else
155         ec = INDEX_SIZE_ERR;
156 }
157
158 Ref<HTMLCollection> HTMLTableRowElement::cells()
159 {
160     return ensureRareData().ensureNodeLists().addCachedCollection<GenericCachedHTMLCollection<CollectionTypeTraits<TRCells>::traversalType>>(*this, TRCells);
161 }
162
163 void HTMLTableRowElement::setCells(HTMLCollection*, ExceptionCode& ec)
164 {
165     ec = NO_MODIFICATION_ALLOWED_ERR;
166 }
167
168 }