Implement a simple version of SelectorQuery.

third_party/WebKit/Source/core/dom/SelectorQuery.h

 /*
  * Copyright (C) 2011, 2013 Apple Inc. All rights reserved.
  * Copyright (C) 2014 Samsung Electronics. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
@@ skipping 17 matching lines @@
 #define SelectorQuery_h
 
 #include "core/css/CSSSelectorList.h"
 #include "platform/heap/Handle.h"
 #include "wtf/HashMap.h"
 #include "wtf/Vector.h"
 #include "wtf/text/AtomicStringHash.h"
 
 namespace blink {
 
-class CSSSelector;
 class ContainerNode;
 class Document;
 class Element;
 class ExceptionState;
+class SelectorMatcher;
 template <typename NodeType> class StaticNodeTypeList;
 typedef StaticNodeTypeList<Element> StaticElementList;
 
-class SelectorDataList {
-    DISALLOW_NEW();
-public:
-    void initialize(const CSSSelectorList&);
-    bool matches(Element&) const;
-    Element* closest(Element&) const;
-    PassRefPtrWillBeRawPtr<StaticElementList> queryAll(ContainerNode& rootNode) const;
-    PassRefPtrWillBeRawPtr<Element> queryFirst(ContainerNode& rootNode) const;
-
-private:
-    bool canUseFastQuery(const ContainerNode& rootNode) const;
-    bool selectorMatches(const CSSSelector&, Element&, const ContainerNode&) const;
-
-    template <typename SelectorQueryTrait>
-    void collectElementsByClassName(ContainerNode& rootNode, const AtomicString& className, typename SelectorQueryTrait::OutputType&) const;
-    template <typename SelectorQueryTrait>
-    void collectElementsByTagName(ContainerNode& rootNode, const QualifiedName& tagName, typename SelectorQueryTrait::OutputType&) const;
-
-    template <typename SelectorQueryTrait>
-    void findTraverseRootsAndExecute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-
-    enum MatchTraverseRootState { DoesNotMatchTraverseRoots, MatchesTraverseRoots };
-    template <typename SelectorQueryTrait>
-    void executeForTraverseRoot(const CSSSelector&, ContainerNode* traverseRoot, MatchTraverseRootState, ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-    template <typename SelectorQueryTrait, typename SimpleElementListType>
-    void executeForTraverseRoots(const CSSSelector&, SimpleElementListType& traverseRoots, MatchTraverseRootState, ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-
-    template <typename SelectorQueryTrait>
-    bool selectorListMatches(ContainerNode& rootNode, Element&, typename SelectorQueryTrait::OutputType&) const;
-    template <typename SelectorQueryTrait>
-    void executeSlow(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-    template <typename SelectorQueryTrait>
-    void executeSlowTraversingShadowTree(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-    template <typename SelectorQueryTrait>
-    void execute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
-    const CSSSelector* selectorForIdLookup(const CSSSelector&) const;
-
-    Vector<const CSSSelector*> m_selectors;
-    bool m_usesDeepCombinatorOrShadowPseudo : 1;
-    bool m_needsUpdatedDistribution : 1;
-};
-
 class SelectorQuery {
     WTF_MAKE_NONCOPYABLE(SelectorQuery);
     USING_FAST_MALLOC(SelectorQuery);
 public:
     static PassOwnPtr<SelectorQuery> adopt(CSSSelectorList&);
+    ~SelectorQuery();
 
     bool matches(Element&) const;
     Element* closest(Element&) const;
     PassRefPtrWillBeRawPtr<StaticElementList> queryAll(ContainerNode& rootNode) const;
-    PassRefPtrWillBeRawPtr<Element> queryFirst(ContainerNode& rootNode) const;
+    Element* queryFirst(ContainerNode& rootNode) const;
 private:
     explicit SelectorQuery(CSSSelectorList&);
 
-    SelectorDataList m_selectors;
-    CSSSelectorList m_selectorList;
+    OwnPtr<SelectorMatcher> m_matcher;
 };
 
 class SelectorQueryCache {
     USING_FAST_MALLOC(SelectorQueryCache);
 public:
     SelectorQuery* add(const AtomicString&, const Document&, ExceptionState&);
     void invalidate();
 
 private:
     HashMap<AtomicString, OwnPtr<SelectorQuery>> m_entries;
 };
 
 }
 
 #endif