<https://webkit.org/b/120013> Tighten up logic in HTMLTableRowsCollection
[WebKit-https.git] / Source / WebCore / html / HTMLTableRowsCollection.cpp
1 /*
2  * Copyright (C) 2008, 2011, 2012 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer.
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution.
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #include "config.h"
30 #include "HTMLTableRowsCollection.h"
31
32 #include "HTMLNames.h"
33 #include "HTMLTableElement.h"
34 #include "HTMLTableRowElement.h"
35
36 namespace WebCore {
37
38 using namespace HTMLNames;
39
40 #if ASSERT_DISABLED
41
42 static inline void assertRowIsInTable(HTMLTableRowElement*)
43 {
44 }
45
46 #else
47
48 // The HTMLCollection caching mechanism along with the code in this class will
49 // guarantee that this row is an immediate child of either the table, a thead,
50 // a tbody, or a tfoot.
51 static inline void assertRowIsInTable(HTMLTableRowElement* row)
52 {
53     if (!row)
54         return;
55     ContainerNode* parent = row->parentNode();
56     ASSERT(parent);
57     if (parent->hasTagName(tableTag))
58         return;
59     ASSERT(parent->hasTagName(theadTag) || parent->hasTagName(tbodyTag) || parent->hasTagName(tfootTag));
60     ContainerNode* grandparent = parent->parentNode();
61     ASSERT(grandparent);
62     ASSERT(grandparent->hasTagName(tableTag));
63 }
64
65 #endif
66
67 static inline bool isInSection(HTMLTableRowElement* row, const QualifiedName& sectionTag)
68 {
69     // Because we know that the parent is a table or a section, all of which are in the HTML
70     // namespace, it's OK to do the faster hasLocalName here instead of the more typical hasTagName,
71     // since we don't need the check for the HTML namespace.
72     return toElement(row->parentNode())->hasLocalName(sectionTag);
73 }
74
75 HTMLTableRowElement* HTMLTableRowsCollection::rowAfter(HTMLTableElement* table, HTMLTableRowElement* previous)
76 {
77     assertRowIsInTable(previous);
78
79     Node* child = 0;
80
81     // Start by looking for the next row in this section.
82     // Continue only if there is none.
83     if (previous && previous->parentNode() != table) {
84         for (child = previous->nextSibling(); child; child = child->nextSibling()) {
85             if (child->hasTagName(trTag))
86                 return static_cast<HTMLTableRowElement*>(child);
87         }
88     }
89
90     // If still looking at head sections, find the first row in the next head section.
91     if (!previous)
92         child = table->firstChild();
93     else if (isInSection(previous, theadTag))
94         child = previous->parentNode()->nextSibling();
95     for (; child; child = child->nextSibling()) {
96         if (child->hasTagName(theadTag)) {
97             for (Node* grandchild = child->firstChild(); grandchild; grandchild = grandchild->nextSibling()) {
98                 if (grandchild->hasTagName(trTag))
99                     return static_cast<HTMLTableRowElement*>(grandchild);
100             }
101         }
102     }
103
104     // If still looking at top level and bodies, find the next row in top level or the first in the next body section.
105     if (!previous || isInSection(previous, theadTag))
106         child = table->firstChild();
107     else if (previous->parentNode() == table)
108         child = previous->nextSibling();
109     else if (isInSection(previous, tbodyTag))
110         child = previous->parentNode()->nextSibling();
111     for (; child; child = child->nextSibling()) {
112         if (child->hasTagName(trTag))
113             return static_cast<HTMLTableRowElement*>(child);
114         if (child->hasTagName(tbodyTag)) {
115             for (Node* grandchild = child->firstChild(); grandchild; grandchild = grandchild->nextSibling()) {
116                 if (grandchild->hasTagName(trTag))
117                     return static_cast<HTMLTableRowElement*>(grandchild);
118             }
119         }
120     }
121
122     // Find the first row in the next foot section.
123     if (!previous || !isInSection(previous, tfootTag))
124         child = table->firstChild();
125     else
126         child = previous->parentNode()->nextSibling();
127     for (; child; child = child->nextSibling()) {
128         if (child->hasTagName(tfootTag)) {
129             for (Node* grandchild = child->firstChild(); grandchild; grandchild = grandchild->nextSibling()) {
130                 if (grandchild->hasTagName(trTag))
131                     return static_cast<HTMLTableRowElement*>(grandchild);
132             }
133         }
134     }
135
136     return 0;
137 }
138
139 HTMLTableRowElement* HTMLTableRowsCollection::lastRow(HTMLTableElement* table)
140 {
141     for (Node* child = table->lastChild(); child; child = child->previousSibling()) {
142         if (child->hasTagName(tfootTag)) {
143             for (Node* grandchild = child->lastChild(); grandchild; grandchild = grandchild->previousSibling()) {
144                 if (grandchild->hasTagName(trTag))
145                     return static_cast<HTMLTableRowElement*>(grandchild);
146             }
147         }
148     }
149
150     for (Node* child = table->lastChild(); child; child = child->previousSibling()) {
151         if (child->hasTagName(trTag))
152             return static_cast<HTMLTableRowElement*>(child);
153         if (child->hasTagName(tbodyTag)) {
154             for (Node* grandchild = child->lastChild(); grandchild; grandchild = grandchild->previousSibling()) {
155                 if (grandchild->hasTagName(trTag))
156                     return static_cast<HTMLTableRowElement*>(grandchild);
157             }
158         }
159     }
160
161     for (Node* child = table->lastChild(); child; child = child->previousSibling()) {
162         if (child->hasTagName(theadTag)) {
163             for (Node* grandchild = child->lastChild(); grandchild; grandchild = grandchild->previousSibling()) {
164                 if (grandchild->hasTagName(trTag))
165                     return static_cast<HTMLTableRowElement*>(grandchild);
166             }
167         }
168     }
169
170     return 0;
171 }
172
173 // Must call get() on the table in case that argument is compiled before dereferencing the
174 // table to get at the collection cache. Order of argument evaluation is undefined and can
175 // differ between compilers.
176 HTMLTableRowsCollection::HTMLTableRowsCollection(Node* table)
177     : HTMLCollection(table, TableRows, OverridesItemAfter)
178 {
179     ASSERT(isHTMLTableElement(table));
180 }
181
182 PassRefPtr<HTMLTableRowsCollection> HTMLTableRowsCollection::create(Node* table, CollectionType)
183 {
184     return adoptRef(new HTMLTableRowsCollection(table));
185 }
186
187 Element* HTMLTableRowsCollection::virtualItemAfter(unsigned& offsetInArray, Element* previous) const
188 {
189     ASSERT_UNUSED(offsetInArray, !offsetInArray);
190     ASSERT(!previous || (previous->isHTMLElement() && toHTMLElement(previous)->hasLocalName(trTag)));
191     return rowAfter(toHTMLTableElement(ownerNode()), static_cast<HTMLTableRowElement*>(previous));
192 }
193
194 }