Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 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 | * |
| 14 | * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY |
| 15 | * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| 16 | * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| 17 | * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY |
| 18 | * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
| 19 | * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| 20 | * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| 21 | * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 22 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| 23 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 24 | */ |
| 25 | |
| 26 | #ifndef SelectorQuery_h |
| 27 | #define SelectorQuery_h |
| 28 | |
| 29 | #include "core/css/CSSSelectorList.h" |
Ben Murdoch | e69819b | 2013-07-17 14:56:49 +0100 | [diff] [blame] | 30 | #include "wtf/HashMap.h" |
| 31 | #include "wtf/Vector.h" |
| 32 | #include "wtf/text/AtomicStringHash.h" |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 33 | |
| 34 | namespace WebCore { |
| 35 | |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 36 | class CSSSelector; |
| 37 | class Document; |
| 38 | class Element; |
Ben Murdoch | df95704 | 2013-08-06 11:01:27 +0100 | [diff] [blame] | 39 | class ExceptionState; |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 40 | class Node; |
| 41 | class NodeList; |
Ben Murdoch | fff8884 | 2013-07-30 15:20:09 +0100 | [diff] [blame] | 42 | class SimpleNodeList; |
| 43 | class SpaceSplitString; |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 44 | |
| 45 | class SelectorDataList { |
| 46 | public: |
| 47 | void initialize(const CSSSelectorList&); |
| 48 | bool matches(Element*) const; |
| 49 | PassRefPtr<NodeList> queryAll(Node* rootNode) const; |
| 50 | PassRefPtr<Element> queryFirst(Node* rootNode) const; |
| 51 | |
| 52 | private: |
| 53 | struct SelectorData { |
| 54 | SelectorData(const CSSSelector* selector, bool isFastCheckable) : selector(selector), isFastCheckable(isFastCheckable) { } |
| 55 | const CSSSelector* selector; |
| 56 | bool isFastCheckable; |
| 57 | }; |
| 58 | |
Ben Murdoch | fff8884 | 2013-07-30 15:20:09 +0100 | [diff] [blame] | 59 | bool canUseFastQuery(Node* rootNode) const; |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 60 | bool selectorMatches(const SelectorData&, Element*, const Node*) const; |
Ben Murdoch | fff8884 | 2013-07-30 15:20:09 +0100 | [diff] [blame] | 61 | void collectElementsByClassName(Node* rootNode, const AtomicString& className, Vector<RefPtr<Node> >&) const; |
| 62 | Element* findElementByClassName(Node* rootNode, const AtomicString& className) const; |
| 63 | void collectElementsByTagName(Node* rootNode, const QualifiedName& tagName, Vector<RefPtr<Node> >&) const; |
| 64 | Element* findElementByTagName(Node* rootNode, const QualifiedName& tagName) const; |
| 65 | PassOwnPtr<SimpleNodeList> findTraverseRoots(Node* rootNode, bool& matchTraverseRoots) const; |
| 66 | void executeSlowQueryAll(Node* rootNode, Vector<RefPtr<Node> >& matchedElements) const; |
| 67 | void executeQueryAll(Node* rootNode, Vector<RefPtr<Node> >& matchedElements) const; |
| 68 | Node* findTraverseRoot(Node* rootNode, bool& matchTraverseRoot) const; |
| 69 | Element* executeSlowQueryFirst(Node* rootNode) const; |
| 70 | Element* executeQueryFirst(Node* rootNode) const; |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 71 | |
| 72 | Vector<SelectorData> m_selectors; |
| 73 | }; |
| 74 | |
| 75 | class SelectorQuery { |
| 76 | WTF_MAKE_NONCOPYABLE(SelectorQuery); |
| 77 | WTF_MAKE_FAST_ALLOCATED; |
| 78 | public: |
| 79 | explicit SelectorQuery(const CSSSelectorList&); |
| 80 | bool matches(Element*) const; |
| 81 | PassRefPtr<NodeList> queryAll(Node* rootNode) const; |
| 82 | PassRefPtr<Element> queryFirst(Node* rootNode) const; |
| 83 | private: |
| 84 | SelectorDataList m_selectors; |
| 85 | CSSSelectorList m_selectorList; |
| 86 | }; |
| 87 | |
| 88 | class SelectorQueryCache { |
| 89 | WTF_MAKE_FAST_ALLOCATED; |
| 90 | public: |
Ben Murdoch | df95704 | 2013-08-06 11:01:27 +0100 | [diff] [blame] | 91 | SelectorQuery* add(const AtomicString&, Document*, ExceptionState&); |
Torne (Richard Coles) | 53e740f | 2013-05-09 18:38:43 +0100 | [diff] [blame] | 92 | void invalidate(); |
| 93 | |
| 94 | private: |
| 95 | HashMap<AtomicString, OwnPtr<SelectorQuery> > m_entries; |
| 96 | }; |
| 97 | |
| 98 | } |
| 99 | |
| 100 | #endif |