#ifndef SelectorQuery_h
#define SelectorQuery_h
#include "CSSSelectorList.h"
#include "NodeList.h"
#include "SelectorCompiler.h"
#include <wtf/HashMap.h>
#include <wtf/Vector.h>
#include <wtf/text/AtomicStringHash.h>
#include <wtf/text/CString.h>
namespace WebCore {
typedef int ExceptionCode;
class CSSSelector;
class ContainerNode;
class Document;
class Element;
class Node;
class NodeList;
class SelectorDataList {
public:
explicit SelectorDataList(const CSSSelectorList&);
bool matches(Element&) const;
RefPtr<NodeList> queryAll(ContainerNode& rootNode) const;
Element* queryFirst(ContainerNode& rootNode) const;
private:
struct SelectorData {
SelectorData(const CSSSelector* selector, bool isFastCheckable)
: selector(selector)
#if ENABLE(CSS_SELECTOR_JIT) && CSS_SELECTOR_JIT_PROFILING
, m_compiledSelectorUseCount(0)
#endif
, isFastCheckable(isFastCheckable)
{
}
const CSSSelector* selector;
#if ENABLE(CSS_SELECTOR_JIT)
mutable JSC::MacroAssemblerCodeRef compiledSelectorCodeRef;
mutable SelectorCompilationStatus compilationStatus;
#if CSS_SELECTOR_JIT_PROFILING
~SelectorData()
{
if (compiledSelectorCodeRef.code().executableAddress())
dataLogF("SelectorData compiled selector %d \"%s\"\n", m_compiledSelectorUseCount, selector->selectorText().utf8().data());
}
mutable unsigned m_compiledSelectorUseCount;
void compiledSelectorUsed() const { m_compiledSelectorUseCount++; }
#endif
#endif // ENABLE(CSS_SELECTOR_JIT)
bool isFastCheckable;
};
bool selectorMatches(const SelectorData&, Element&, const ContainerNode& rootNode) const;
template <typename SelectorQueryTrait> void execute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeFastPathForIdSelector(const ContainerNode& rootNode, const SelectorData&, const CSSSelector* idSelector, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleTagNameSelectorData(const ContainerNode& rootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleClassNameSelectorData(const ContainerNode& rootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleSelectorData(const ContainerNode& rootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleMultiSelectorData(const ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
#if ENABLE(CSS_SELECTOR_JIT)
template <typename SelectorQueryTrait> void executeCompiledSimpleSelectorChecker(const ContainerNode& rootNode, SelectorCompiler::SimpleSelectorChecker, typename SelectorQueryTrait::OutputType&, const SelectorData&) const;
#endif // ENABLE(CSS_SELECTOR_JIT)
Vector<SelectorData> m_selectors;
mutable enum MatchType {
CompilableSingle,
CompilableSingleWithRootFilter,
CompiledSingle,
CompiledSingleWithRootFilter,
SingleSelector,
RightMostWithIdMatch,
TagNameMatch,
ClassNameMatch,
MultipleSelectorMatch,
} m_matchType;
};
class SelectorQuery {
WTF_MAKE_NONCOPYABLE(SelectorQuery);
WTF_MAKE_FAST_ALLOCATED;
public:
explicit SelectorQuery(CSSSelectorList&&);
bool matches(Element&) const;
RefPtr<NodeList> queryAll(ContainerNode& rootNode) const;
Element* queryFirst(ContainerNode& rootNode) const;
private:
CSSSelectorList m_selectorList;
SelectorDataList m_selectors;
};
class SelectorQueryCache {
WTF_MAKE_FAST_ALLOCATED;
public:
SelectorQuery* add(const String&, Document&, ExceptionCode&);
private:
HashMap<String, std::unique_ptr<SelectorQuery>> m_entries;
};
inline bool SelectorQuery::matches(Element& element) const
{
return m_selectors.matches(element);
}
inline RefPtr<NodeList> SelectorQuery::queryAll(ContainerNode& rootNode) const
{
return m_selectors.queryAll(rootNode);
}
inline Element* SelectorQuery::queryFirst(ContainerNode& rootNode) const
{
return m_selectors.queryFirst(rootNode);
}
}
#endif