Add child and descendant const iterators
[WebKit-https.git] / Source / WebCore / html / HTMLFieldSetElement.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
4  *           (C) 2001 Dirk Mueller (mueller@kde.org)
5  * Copyright (C) 2004, 2005, 2006, 2010 Apple Inc. All rights reserved.
6  *           (C) 2006 Alexey Proskuryakov (ap@nypop.com)
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Library General Public
10  * License as published by the Free Software Foundation; either
11  * version 2 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Library General Public License for more details.
17  *
18  * You should have received a copy of the GNU Library General Public License
19  * along with this library; see the file COPYING.LIB.  If not, write to
20  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
21  * Boston, MA 02110-1301, USA.
22  *
23  */
24
25 #include "config.h"
26 #include "HTMLFieldSetElement.h"
27
28 #include "ChildIterator.h"
29 #include "DescendantIterator.h"
30 #include "HTMLCollection.h"
31 #include "HTMLLegendElement.h"
32 #include "HTMLNames.h"
33 #include "HTMLObjectElement.h"
34 #include "RenderFieldset.h"
35 #include <wtf/StdLibExtras.h>
36
37 namespace WebCore {
38
39 using namespace HTMLNames;
40
41 inline HTMLFieldSetElement::HTMLFieldSetElement(const QualifiedName& tagName, Document* document, HTMLFormElement* form)
42     : HTMLFormControlElement(tagName, document, form)
43     , m_documentVersion(0)
44 {
45     ASSERT(hasTagName(fieldsetTag));
46 }
47
48 PassRefPtr<HTMLFieldSetElement> HTMLFieldSetElement::create(const QualifiedName& tagName, Document* document, HTMLFormElement* form)
49 {
50     return adoptRef(new HTMLFieldSetElement(tagName, document, form));
51 }
52
53 void HTMLFieldSetElement::invalidateDisabledStateUnder(Element* base)
54 {
55     auto formControlDescendants = descendantsOfType<HTMLFormControlElement>(base);
56     for (auto control = formControlDescendants.begin(), end = formControlDescendants.end(); control != end; ++control)
57         control->ancestorDisabledStateWasChanged();
58 }
59
60 void HTMLFieldSetElement::disabledAttributeChanged()
61 {
62     // This element must be updated before the style of nodes in its subtree gets recalculated.
63     HTMLFormControlElement::disabledAttributeChanged();
64     invalidateDisabledStateUnder(this);
65 }
66
67 void HTMLFieldSetElement::childrenChanged(bool changedByParser, Node* beforeChange, Node* afterChange, int childCountDelta)
68 {
69     HTMLFormControlElement::childrenChanged(changedByParser, beforeChange, afterChange, childCountDelta);
70
71     auto legendChildren = childrenOfType<HTMLLegendElement>(this);
72     for (auto legend = legendChildren.begin(), end = legendChildren.end(); legend != end; ++legend)
73         invalidateDisabledStateUnder(&*legend);
74 }
75
76 bool HTMLFieldSetElement::supportsFocus() const
77 {
78     return HTMLElement::supportsFocus();
79 }
80
81 const AtomicString& HTMLFieldSetElement::formControlType() const
82 {
83     DEFINE_STATIC_LOCAL(const AtomicString, fieldset, ("fieldset", AtomicString::ConstructFromLiteral));
84     return fieldset;
85 }
86
87 RenderObject* HTMLFieldSetElement::createRenderer(RenderArena* arena, RenderStyle*)
88 {
89     return new (arena) RenderFieldset(this);
90 }
91
92 const HTMLLegendElement* HTMLFieldSetElement::legend() const
93 {
94     auto legendDescendants = descendantsOfType<HTMLLegendElement>(this);
95     auto firstLegend = legendDescendants.begin();
96     if (firstLegend != legendDescendants.end())
97         return &*firstLegend;
98     return nullptr;
99 }
100
101 PassRefPtr<HTMLCollection> HTMLFieldSetElement::elements()
102 {
103     return ensureCachedHTMLCollection(FormControls);
104 }
105
106 void HTMLFieldSetElement::refreshElementsIfNeeded() const
107 {
108     uint64_t docVersion = document()->domTreeVersion();
109     if (m_documentVersion == docVersion)
110         return;
111
112     m_documentVersion = docVersion;
113
114     m_associatedElements.clear();
115
116     for (Element* element = ElementTraversal::firstWithin(this); element; element = ElementTraversal::next(element, this)) {
117         if (element->hasTagName(objectTag)) {
118             m_associatedElements.append(static_cast<HTMLObjectElement*>(element));
119             continue;
120         }
121
122         if (!element->isFormControlElement())
123             continue;
124
125         m_associatedElements.append(static_cast<HTMLFormControlElement*>(element));
126     }
127 }
128
129 const Vector<FormAssociatedElement*>& HTMLFieldSetElement::associatedElements() const
130 {
131     refreshElementsIfNeeded();
132     return m_associatedElements;
133 }
134
135 unsigned HTMLFieldSetElement::length() const
136 {
137     refreshElementsIfNeeded();
138     unsigned len = 0;
139     for (unsigned i = 0; i < m_associatedElements.size(); ++i)
140         if (m_associatedElements[i]->isEnumeratable())
141             ++len;
142     return len;
143 }
144
145 } // namespace