Remove some unnecessary branches from LiveNodeList traversal
[WebKit-https.git] / Source / WebCore / dom / TagNodeList.h
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, 2007, 2008 Apple Inc. All rights reserved.
6  * Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies)
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 #ifndef TagNodeList_h
25 #define TagNodeList_h
26
27 #include "Element.h"
28 #include "LiveNodeList.h"
29 #include <wtf/text/AtomicString.h>
30
31 namespace WebCore {
32
33 // NodeList that limits to a particular tag.
34 class TagNodeList final : public CachedLiveNodeList<TagNodeList> {
35 public:
36     static PassRefPtr<TagNodeList> create(ContainerNode& rootNode, const AtomicString& namespaceURI, const AtomicString& localName)
37     {
38         ASSERT(namespaceURI != starAtom);
39         return adoptRef(new TagNodeList(rootNode, namespaceURI, localName));
40     }
41
42     static PassRefPtr<TagNodeList> create(ContainerNode& rootNode, Type type, const AtomicString& localName)
43     {
44         ASSERT_UNUSED(type, type == Type::TagNodeListType);
45         return adoptRef(new TagNodeList(rootNode, starAtom, localName));
46     }
47
48     virtual ~TagNodeList();
49
50     virtual bool nodeMatches(Element*) const override;
51
52 protected:
53     TagNodeList(ContainerNode& rootNode, const AtomicString& namespaceURI, const AtomicString& localName);
54
55     AtomicString m_namespaceURI;
56     AtomicString m_localName;
57 };
58
59 inline bool TagNodeList::nodeMatches(Element* element) const
60 {
61     // Implements http://dvcs.w3.org/hg/domcore/raw-file/tip/Overview.html#concept-getelementsbytagnamens
62     if (m_localName != starAtom && m_localName != element->localName())
63         return false;
64
65     return m_namespaceURI == starAtom || m_namespaceURI == element->namespaceURI();
66 }
67
68 class HTMLTagNodeList final : public CachedLiveNodeList<HTMLTagNodeList> {
69 public:
70     static PassRefPtr<HTMLTagNodeList> create(ContainerNode& rootNode, Type type, const AtomicString& localName)
71     {
72         ASSERT_UNUSED(type, type == Type::HTMLTagNodeListType);
73         return adoptRef(new HTMLTagNodeList(rootNode, localName));
74     }
75
76     virtual ~HTMLTagNodeList();
77
78     virtual bool nodeMatches(Element*) const override;
79
80 private:
81     HTMLTagNodeList(ContainerNode& rootNode, const AtomicString& localName);
82
83     AtomicString m_localName;
84     AtomicString m_loweredLocalName;
85 };
86
87 inline bool HTMLTagNodeList::nodeMatches(Element* element) const
88 {
89     // Implements http://dvcs.w3.org/hg/domcore/raw-file/tip/Overview.html#concept-getelementsbytagname
90     if (m_localName == starAtom)
91         return true;
92     const AtomicString& localName = element->isHTMLElement() ? m_loweredLocalName : m_localName;
93     return localName == element->localName();
94 }
95
96 } // namespace WebCore
97
98 #endif // TagNodeList_h