initial import
[vuplus_webkit] / Source / WebCore / html / HTMLFormCollection.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
4  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2010 Apple Inc. All rights reserved.
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public License
17  * along with this library; see the file COPYING.LIB.  If not, write to
18  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  *
21  */
22
23 #include "config.h"
24 #include "HTMLFormCollection.h"
25
26 #include "CollectionCache.h"
27 #include "HTMLFormControlElement.h"
28 #include "HTMLFormElement.h"
29 #include "HTMLImageElement.h"
30 #include "HTMLNames.h"
31
32 namespace WebCore {
33
34 using namespace HTMLNames;
35
36 // Since the collections are to be "live", we have to do the
37 // calculation every time if anything has changed.
38
39 inline CollectionCache* HTMLFormCollection::formCollectionInfo(HTMLFormElement* form)
40 {
41     if (!form->m_collectionCache)
42         form->m_collectionCache = adoptPtr(new CollectionCache);
43     return form->m_collectionCache.get();
44 }
45
46 HTMLFormCollection::HTMLFormCollection(PassRefPtr<HTMLFormElement> form)
47     : HTMLCollection(form.get(), OtherCollection, formCollectionInfo(form.get()))
48 {
49 }
50
51 PassRefPtr<HTMLFormCollection> HTMLFormCollection::create(PassRefPtr<HTMLFormElement> form)
52 {
53     return adoptRef(new HTMLFormCollection(form));
54 }
55
56 HTMLFormCollection::~HTMLFormCollection()
57 {
58 }
59
60 unsigned HTMLFormCollection::calcLength() const
61 {
62     return static_cast<HTMLFormElement*>(base())->length();
63 }
64
65 Node* HTMLFormCollection::item(unsigned index) const
66 {
67     resetCollectionInfo();
68
69     if (info()->current && info()->position == index)
70         return info()->current;
71
72     if (info()->hasLength && info()->length <= index)
73         return 0;
74
75     if (!info()->current || info()->position > index) {
76         info()->current = 0;
77         info()->position = 0;
78         info()->elementsArrayPosition = 0;
79     }
80
81     Vector<FormAssociatedElement*>& elementsArray = static_cast<HTMLFormElement*>(base())->m_associatedElements;
82     unsigned currentIndex = info()->position;
83     
84     for (unsigned i = info()->elementsArrayPosition; i < elementsArray.size(); i++) {
85         if (elementsArray[i]->isEnumeratable()) {
86             HTMLElement* element = toHTMLElement(elementsArray[i]);
87             if (index == currentIndex) {
88                 info()->position = index;
89                 info()->current = element;
90                 info()->elementsArrayPosition = i;
91                 return element;
92             }
93
94             currentIndex++;
95         }
96     }
97
98     return 0;
99 }
100
101 Element* HTMLFormCollection::getNamedItem(const QualifiedName& attrName, const AtomicString& name) const
102 {
103     info()->position = 0;
104     return getNamedFormItem(attrName, name, 0);
105 }
106
107 Element* HTMLFormCollection::getNamedFormItem(const QualifiedName& attrName, const String& name, int duplicateNumber) const
108 {
109     HTMLFormElement* form = static_cast<HTMLFormElement*>(base());
110
111     bool foundInputElements = false;
112     for (unsigned i = 0; i < form->m_associatedElements.size(); ++i) {
113         FormAssociatedElement* associatedElement = form->m_associatedElements[i];
114         HTMLElement* element = toHTMLElement(associatedElement);
115         if (associatedElement->isEnumeratable() && element->getAttribute(attrName) == name) {
116             foundInputElements = true;
117             if (!duplicateNumber)
118                 return element;
119             --duplicateNumber;
120         }
121     }
122
123     if (!foundInputElements) {
124         for (unsigned i = 0; i < form->m_imageElements.size(); ++i) {
125             HTMLImageElement* element = form->m_imageElements[i];
126             if (element->getAttribute(attrName) == name) {
127                 if (!duplicateNumber)
128                     return element;
129                 --duplicateNumber;
130             }
131         }
132     }
133
134     return 0;
135 }
136
137 Node* HTMLFormCollection::nextItem() const
138 {
139     return item(info()->position + 1);
140 }
141
142 Element* HTMLFormCollection::nextNamedItemInternal(const String &name) const
143 {
144     Element* retval = getNamedFormItem(m_idsDone ? nameAttr : idAttr, name, ++info()->position);
145     if (retval)
146         return retval;
147     if (m_idsDone) // we're done
148         return 0;
149     // After doing id, do name
150     m_idsDone = true;
151     return getNamedItem(nameAttr, name);
152 }
153
154 Node* HTMLFormCollection::namedItem(const AtomicString& name) const
155 {
156     // http://msdn.microsoft.com/workshop/author/dhtml/reference/methods/nameditem.asp
157     // This method first searches for an object with a matching id
158     // attribute. If a match is not found, the method then searches for an
159     // object with a matching name attribute, but only on those elements
160     // that are allowed a name attribute.
161     resetCollectionInfo();
162     m_idsDone = false;
163     info()->current = getNamedItem(idAttr, name);
164     if (info()->current)
165         return info()->current;
166     m_idsDone = true;
167     info()->current = getNamedItem(nameAttr, name);
168     return info()->current;
169 }
170
171 Node* HTMLFormCollection::nextNamedItem(const AtomicString& name) const
172 {
173     // The nextNamedItemInternal function can return the same item twice if it has
174     // both an id and name that are equal to the name parameter. So this function
175     // checks if we are on the nameAttr half of the iteration and skips over any
176     // that also have the same idAttributeName.
177     Element* impl = nextNamedItemInternal(name);
178     if (m_idsDone) {
179         while (impl && impl->getIdAttribute() == name)
180             impl = nextNamedItemInternal(name);
181     }
182     return impl;
183 }
184
185 void HTMLFormCollection::updateNameCache() const
186 {
187     if (info()->hasNameCache)
188         return;
189
190     HashSet<AtomicStringImpl*> foundInputElements;
191
192     HTMLFormElement* f = static_cast<HTMLFormElement*>(base());
193
194     for (unsigned i = 0; i < f->m_associatedElements.size(); ++i) {
195         FormAssociatedElement* associatedElement = f->m_associatedElements[i];
196         if (associatedElement->isEnumeratable()) {
197             HTMLElement* element = toHTMLElement(associatedElement);
198             const AtomicString& idAttrVal = element->getIdAttribute();
199             const AtomicString& nameAttrVal = element->getAttribute(nameAttr);
200             if (!idAttrVal.isEmpty()) {
201                 // add to id cache
202                 Vector<Element*>* idVector = info()->idCache.get(idAttrVal.impl());
203                 if (!idVector) {
204                     idVector = new Vector<Element*>;
205                     info()->idCache.add(idAttrVal.impl(), idVector);
206                 }
207                 idVector->append(element);
208                 foundInputElements.add(idAttrVal.impl());
209             }
210             if (!nameAttrVal.isEmpty() && idAttrVal != nameAttrVal) {
211                 // add to name cache
212                 Vector<Element*>* nameVector = info()->nameCache.get(nameAttrVal.impl());
213                 if (!nameVector) {
214                     nameVector = new Vector<Element*>;
215                     info()->nameCache.add(nameAttrVal.impl(), nameVector);
216                 }
217                 nameVector->append(element);
218                 foundInputElements.add(nameAttrVal.impl());
219             }
220         }
221     }
222
223     for (unsigned i = 0; i < f->m_imageElements.size(); ++i) {
224         HTMLImageElement* element = f->m_imageElements[i];
225         const AtomicString& idAttrVal = element->getIdAttribute();
226         const AtomicString& nameAttrVal = element->getAttribute(nameAttr);
227         if (!idAttrVal.isEmpty() && !foundInputElements.contains(idAttrVal.impl())) {
228             // add to id cache
229             Vector<Element*>* idVector = info()->idCache.get(idAttrVal.impl());
230             if (!idVector) {
231                 idVector = new Vector<Element*>;
232                 info()->idCache.add(idAttrVal.impl(), idVector);
233             }
234             idVector->append(element);
235         }
236         if (!nameAttrVal.isEmpty() && idAttrVal != nameAttrVal && !foundInputElements.contains(nameAttrVal.impl())) {
237             // add to name cache
238             Vector<Element*>* nameVector = info()->nameCache.get(nameAttrVal.impl());
239             if (!nameVector) {
240                 nameVector = new Vector<Element*>;
241                 info()->nameCache.add(nameAttrVal.impl(), nameVector);
242             }
243             nameVector->append(element);
244         }
245     }
246
247     info()->hasNameCache = true;
248 }
249
250 }